将boost :: thread-> native_handle()翻译为XP ThreadId

时间:2013-05-23 09:08:14

标签: winapi boost-thread

我设法通过 GetThreadId(HANDLE) boost :: thread 中取出 native_handle()中的Windows ThreadId >。遗憾的是,这个调用在Windows XP上不可用,在搜索后我找到了解决方案,通过 Thread32First() Thread32Next()遍历所有线程,为XP提供后备支持。 WINAPI的功能。

这确实有效,但我的问题是我目前只能识别我的进程的线程...我现在不知道如何匹配 native_handle() / 从循环遍历中使用适当的 THREADENTRY32 从一侧处理

THREADENTRY32 te32;
//...
do { 
    if( te32.th32OwnerProcessID == GetCurrentProcessId() ) {
        DWORD threadId = te32.th32ThreadID;
        printf( "\n     THREAD ID      = 0x%08X", te32.th32ThreadID ); 
    }
} while( Thread32Next(hThreadSnap, &te32 ) );

任何人都可以帮助我吗?如何将boost :: thread-> native_handle()转换为WindowsXP上的ThreadId?

非常感谢!

1 个答案:

答案 0 :(得分:2)

将循环中的每个线程ID传递给OpenThread(),直到找到匹配的HANDLE。例如:

HANDLE hBoostThread = ...; // from boost::thread->native_handle()
DWORD dwBoostThreadID = 0;

THREADENTRY32 te32;
//...
do
{ 
    if( te32.th32OwnerProcessID == GetCurrentProcessId() )
    {
        HANDLE hThread = OpenThread(THREAD_ALL_ACCESS, FALSE, te32.th32ThreadID);
        if (hThread != NULL)
        {
            if (hThread == hBoostThread)
            {
                CloseHandle(hThread);
                dwBoostThreadID = te32.th32ThreadID; 
                break;
            }
            CloseHandle(hThread);
        }
    }
}
while( Thread32Next(hThreadSnap, &te32 ) );

为了更好的衡量,您可以将此包装在一个函数中,只要GetThreadId()本身不可用,您就可以调用它,这样您的代码就不需要知道差异,例如:

DWORD WINAPI MyGetThreadId(HANDLE Thread)
{
    THREADENTRY32 te32;

    HANDLE hThreadSnap = CreateToolhelp32Snapshot(TH32CS_SNAPTHREAD, 0);
    if (hThreadSnap == INVALID_HANDLE_VALUE)
        return 0;

    if (Thread32First(hThreadSnap, &te32))
    {
        do
        { 
            HANDLE hOpenThread = OpenThread(THREAD_ALL_ACCESS, FALSE, te32.th32ThreadID);
            if (hOpenThread != NULL)
            {
                if (hOpenThread == Thread)
                {
                    CloseHandle(hOpenThread);
                    CloseHandle(hThreadSnap);
                    return te32.th32ThreadID; 
                }

                CloseHandle(hOpenThread);
            }
        }
        while( Thread32Next(hThreadSnap, &te32 ) );
    }

    CloseHandle(hThreadSnap);
    return 0;
}

typedef DWORD (WINAPI *LPFN_GTID)(HANDLE);

LPFN_GTID lpGetThreadId = (LPFN_GTID) GetProcAddress(GetModuleHandle("kernel32"), "GetThreadId");
if (!lpGetThreadId)
    lpGetThreadId = &MyGetThreadId;

DWORD dwThreadID = lpGetThreadId((HANDLE) boost::thread->native_handle());

话虽如此,更好的选择是直接查询目标线程本身以获取自己的ID,而不是试图手动搜索它:

typedef long (WINAPI *LPFN_NTQIT)(HANDLE thread, int infoclass, void *buf, long size, long *used);

typedef struct _THREAD_BASIC_INFORMATION
{
    ULONG ExitStatus;
    void* TebBaseAddress;
    ULONG UniqueProcessId;
    ULONG UniqueThreadId;
    ULONG AffinityMask;
    ULONG BasePriority;
    ULONG DiffProcessPriority;
} THREAD_BASIC_INFORMATION;

DWORD WINAPI MyGetThreadId(HANDLE Thread)
{
    DWORD dwThreadId = 0;

    HMODULE hLib = LoadLibrary("ntdll.dll");
    if (hLib != NULL)
    {
        LPFN_NTQIT lpNtQueryInformationThread = (LPFN_NTQIT) GetProcAddress(hLib, "NtQueryInformationThread");

        if (lpNtQueryInformationThread != NULL)
        {
            THREAD_BASIC_INFORMATION tbi = {0};
            ULONG used = 0;

            if (lpNtQueryInformationThread(Thread, 0, &tbi, sizeof(tbi), &used) == 0)
                dwThreadId = tbi.UniqueThreadId;
        }

        FreeLibrary(hLib);
    }

    return dwThreadId;
}