MinHook test

#include <windows.h>
#include "MinHook.h"

#if defined _M_X64
#pragma comment(lib, "libMinHook.x64.lib")
#elif defined _M_IX86
#pragma comment(lib, "libMinHook.x86.lib")
#endif

#pragma comment(lib, "winmm.lib") // for timeGetTime

// Random 100 functions
void *pUser32Functions[] = {
    ShowWindow,
    SendMessage,
    AnimateWindow,
    GetClipboardData,
    SetCursorPos,
    GetForegroundWindow,
    MessageBox,
    BringWindowToTop,
    DispatchMessage,
    MessageBeep,
    IsWindow,
    CloseWindow,
    mouse_event,
    PrintWindow,
    GetDC,
    WaitMessage,
    ShowCursor,
    IsIconic,
    CharUpper,
    ReleaseCapture,
    RemoveMenu,
    MoveWindow,
    GetMessage,
    PostMessage,
    GetWindowThreadProcessId,
    SetWindowPos,
    DestroyWindow,
    OpenDesktop,
    SetActiveWindow,
    GetSystemMenu,
    SetTimer,
    ChangeDisplaySettings,
    ReleaseDC,
    FindWindow,
    FlashWindow,
    SetParent,
    PeekMessage,
    GetWindow,
    SetWindowText,
    GetCursor,
    OpenClipboard,
    GetClassName,
    SetFocus,
    EnumWindows,
    EnumDisplaySettings,
    BeginPaint,
    GetAncestor,
    CallWindowProc,
    GetWindowInfo,
    GetCursorInfo,
    LoadString,
    GetParent,
    RedrawWindow,
    GetSystemMetrics,
    RegisterClassEx,
    DefWindowProc,
    ClientToScreen,
    GetProp,
    SwitchDesktop,
    SetForegroundWindow,
    BlockInput,
    ShowScrollBar,
    keybd_event,
    GetMonitorInfo,
    CloseDesktop,
    CallNextHookEx,
    SendMessageTimeout,
    ExitWindowsEx,
    OpenInputDesktop,
    GetCursorPos,
    CharNext,
    KillTimer,
    InsertMenu,
    HideCaret,
    ScreenToClient,
    GetDlgItemText,
    LockWindowUpdate,
    GetClassInfo,
    GetWindowPlacement,
    UpdateWindow,
    GetLastInputInfo,
    GetMenuItemCount,
    FillRect,
    RegisterClass,
    GetClientRect,
    CreateDesktop,
    GetSubMenu,
    SetClipboardData,
    RegisterDeviceNotification,
    CharToOem,
    RegisterHotKey,
    GetWindowText,
    LoadImage,
    GetMenuString,
    RegisterWindowMessage,
    EnableMenuItem,
    TileWindows,
    GetScrollPos,
    GetFocus,
    TrackPopupMenu,
    NULL
};

// Foo detour function
void WINAPI DetourFunc()
{
    __debugbreak();
}

// Foo thread
DWORD WINAPI ThreadProc(LPVOID lpParameter)
{
    Sleep(INFINITE);
    return 0;
}

int WINAPI WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine, int nShowCmd)
{
    int i;
    void *pFoo;
    DWORD dwTickCount;
    DWORD dwt1, dwt2;
    DWORD dwt3, dwt4;
    char szMessage[256];

    // Initialize MinHook
    if (MH_Initialize() != MH_OK)
    {
        return 1;
    }

    // Create hooks
    for(i=0; pUser32Functions[i] != NULL; i++)
    {
        if (MH_CreateHook(pUser32Functions[i], &DetourFunc,
            (void**)(&pFoo)) != MH_OK)
        {
            return 1;
        }
    }

    //////////////////////////////////////////////////////////////////////////
    // Test #1

    // Enable all using MH_EnableHook
    dwTickCount = timeGetTime();

    for(i=0; pUser32Functions[i] != NULL; i++)
    {
        if (MH_EnableHook(pUser32Functions[i]) != MH_OK)
        {
            return 1;
        }
    }

    dwt1 = timeGetTime() - dwTickCount;

    // Disable all
    MH_DisableAllHooks();

    // Enable all using MH_EnableAllHooks
    dwTickCount = timeGetTime();

    MH_EnableAllHooks();

    dwt2 = timeGetTime() - dwTickCount;

    // Disable all
    MH_DisableAllHooks();

    //////////////////////////////////////////////////////////////////////////
    // Test #2

    // Create 100 threads
    for(i=0; i<100; i++)
        CloseHandle(CreateThread(NULL, 0, ThreadProc, NULL, 0, NULL));

    // Enable all using MH_EnableHook
    dwTickCount = timeGetTime();

    for(i=0; pUser32Functions[i] != NULL; i++)
    {
        if (MH_EnableHook(pUser32Functions[i]) != MH_OK)
        {
            return 1;
        }
    }

    dwt3 = timeGetTime() - dwTickCount;

    // Disable all
    MH_DisableAllHooks();

    // Enable all using MH_EnableAllHooks
    dwTickCount = timeGetTime();

    MH_EnableAllHooks();

    dwt4 = timeGetTime() - dwTickCount;

    // Disable all
    MH_DisableAllHooks();

    //////////////////////////////////////////////////////////////////////////
    // Done

    // Uninitialize MinHook
    if (MH_Uninitialize() != MH_OK)
    {
        return 1;
    }

    // And the winner is...
    wsprintf(szMessage,
        "No threads:\n"
        "MH_EnableHook: %u ms\n"
        "MH_EnableAllHooks: %u ms\n"
        "\n"
        "100 threads:\n"
        "MH_EnableHook: %u ms\n"
        "MH_EnableAllHooks: %u ms",
        dwt1, dwt2, dwt3, dwt4);

    MessageBox(NULL, szMessage, "Timing", MB_OK);

    ExitProcess(0);
}