ek0 / htp Goto Github PK

void CreateFileWHook(HTPContext* ctx)
{
}


void CreateFileWPostHook(HTPContext* ctx)
{
}

void NtCreateFileHook(HTPContext* ctx)
{
}


void NtCreateFilePostHook(HTPContext* ctx)
{
}

VOID MyThreadFunction(VOID)
{
    Sleep(1);
    HANDLE hFile = CreateFile("roux.bmp",
            GENERIC_WRITE,
            0,
            NULL,
            CREATE_NEW,
            FILE_ATTRIBUTE_NORMAL,
            NULL);
    Sleep(1);
    printf("[%d] Bye bye\n", GetCurrentThreadId());
}

/*
....
*/
int main(int argc, char** argv)
{
    HTPHandle handle;

    HTPInit(&handle);
    SetupInlineHook(&handle, "kernel32.dll", "CreateFileW", CreateFileWHook, CreateFileWPostHook);
    SetupInlineHook(&handle, "ntdll.dll", "NtCreateFile", NtCreateFileHook, NtCreateFilePostHook);

    std::queue<HANDLE> thqueue;
    for (int i = 0; i < 0x2000; i++) {
        DWORD dwThreadId;
        HANDLE hThread = CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE)MyThreadFunction, NULL, 0, &dwThreadId);
        if (hThread != NULL) {
            thqueue.push(hThread);
        }
    }
    while (!thqueue.empty()) {
        WaitForSingleObject(thqueue.front(), INFINITE);
        thqueue.pop();
    }
    printf("GOOD!\n");
}