slept.cpp

微软提供的截取Win32 API函数的开发包和例子detours-src-1.2.rar

//////////////////////////////////////////////////////////////////////
//
//	Module:		slept.dll
//	Author:		Galen C. Hunt
//	Copyright:	1996-1999, Microsoft Corporation
//
//	Version 1.2. (Build 35)
//

#define WIN32
#define NT

#include <windows.h>
#include "detours.h"
#include "slept.h"

LONG dwSlept = 0;

DETOUR_TRAMPOLINE(VOID WINAPI UntimedSleep(DWORD dwMilliseconds), Sleep);
 
VOID WINAPI TimedSleep(DWORD dwMilliseconds)
{
    DWORD dwBeg = GetTickCount();
    UntimedSleep(dwMilliseconds);
    DWORD dwEnd = GetTickCount();
    
    InterlockedExchangeAdd(&dwSlept, dwEnd - dwBeg);
}

DWORD WINAPI GetSleptTicks(VOID)
{
    return dwSlept;
}

BOOL WINAPI DllMain(HINSTANCE hinst, DWORD dwReason, LPVOID reserved)
{
    if (dwReason == DLL_PROCESS_ATTACH)
        DetourFunctionWithTrampoline((PBYTE)UntimedSleep, (PBYTE)TimedSleep);
    else if (dwReason == DLL_PROCESS_DETACH)
        DetourRemoveWithTrampoline((PBYTE)UntimedSleep, (PBYTE)TimedSleep);
    return TRUE;
}
//
///////////////////////////////////////////////////////////////// End of File.