C语言获取线程启动地址源代码

1. 
   
2. #define   WIN32_LEAN_AND_MEAN   
   
3.   #define   _WIN32_WINNT   0x400   
   
4.   #include   <stdio.h>   
   
5.   #include   <tchar.h>   
   
6.   #include   <locale.h>   
   
7.   #include   <windows.h>   
   
8.   #include   <psapi.h>   
   
9.   #include   <Tlhelp32.h>   
   
10.    
    
11.   #pragma   comment   (lib,   "psapi.lib")   
    
12.    
    
13.   //   
    
14.   //   Thread   Information   Classes   
    
15.   //   
    
16.    
    
17.   typedef   enum   _THREADINFOCLASS   {   
    
18.           ThreadBasicInformation,   
    
19.           ThreadTimes,   
    
20.           ThreadPriority,   
    
21.           ThreadBasePriority,   
    
22.           ThreadAffinityMask,   
    
23.           ThreadImpersonationToken,   
    
24.           ThreadDescriptorTableEntry,   
    
25.           ThreadEnableAlignmentFaultFixup,   
    
26.           ThreadEventPair_Reusable,   
    
27.           ThreadQuerySetWin32StartAddress,   
    
28.           ThreadZeroTlsCell,   
    
29.           ThreadPerformanceCount,   
    
30.           ThreadAmILastThread,   
    
31.           ThreadIdealProcessor,   
    
32.           ThreadPriorityBoost,   
    
33.           ThreadSetTlsArrayAddress,   
    
34.           ThreadIsIoPending,   
    
35.           ThreadHideFromDebugger,   
    
36.           ThreadBreakOnTermination,   
    
37.           MaxThreadInfoClass   
    
38.           }   THREADINFOCLASS;   
    
39.    
    
40.   typedef   struct   _CLIENT_ID   {   
    
41.           HANDLE   UniqueProcess;   
    
42.           HANDLE   UniqueThread;   
    
43.   }   CLIENT_ID;   
    
44.   typedef   CLIENT_ID   *PCLIENT_ID;   
    
45.    
    
46.   typedef   struct   _THREAD_BASIC_INFORMATION   {   //   Information   Class   0   
    
47.           LONG           ExitStatus;   
    
48.           PVOID         TebBaseAddress;   
    
49.           CLIENT_ID   ClientId;   
    
50.           LONG   AffinityMask;   
    
51.           LONG   Priority;   
    
52.           LONG   BasePriority;   
    
53.   }   THREAD_BASIC_INFORMATION,   *PTHREAD_BASIC_INFORMATION;   
    
54.    
    
55.   extern   "C"   LONG   (__stdcall   *ZwQueryInformationThread)   (   
    
56.           IN   HANDLE   ThreadHandle,   
    
57.           IN   THREADINFOCLASS   ThreadInformationClass,   
    
58.           OUT   PVOID   ThreadInformation,   
    
59.           IN   ULONG   ThreadInformationLength,   
    
60.           OUT   PULONG   ReturnLength   OPTIONAL   
    
61.           )   =   NULL;   
    
62.    
    
63.    
    
64.   extern   "C"   LONG   (__stdcall   *RtlNtStatusToDosError)   (   
    
65.           IN     ULONG   status)   =   NULL;
    
66. 
    
67. BOOL   ShowThreadInfo   (DWORD   tid)   
    
68.           {   
    
69.                   THREAD_BASIC_INFORMATION         tbi;   
    
70.                   PVOID                                               startaddr;   
    
71.                   LONG                                                 status;   
    
72.                   HANDLE                                             thread,   process;   
    
73.                     
    
74.                   thread   =   ::OpenThread   (THREAD_ALL_ACCESS,   FALSE,   tid);   
    
75.                   if   (thread   ==   NULL)   
    
76.                           return   FALSE;   
    
77.    
    
78.                   status   =   ZwQueryInformationThread   (thread,     
    
79.                           ThreadQuerySetWin32StartAddress,     
    
80.                           &startaddr,     
    
81.                           sizeof   (startaddr),     
    
82.                           NULL);   
    
83.    
    
84.                   if   (status   <   0)   
    
85.                   {   
    
86.                           CloseHandle   (thread);   
    
87.                           SetLastError   (RtlNtStatusToDosError   (status));   
    
88.                           return   FALSE;   
    
89.                   };   
    
90.    
    
91.                   _tprintf   (TEXT   ("线程   %08x   的起始地址为   %p\n"),     
    
92.                           tid,     
    
93.                           startaddr);   
    
94.    
    
95.                   status   =   ZwQueryInformationThread   (thread,     
    
96.                           ThreadBasicInformation,     
    
97.                           &tbi,     
    
98.                           sizeof   (tbi),     
    
99.                           NULL);   
    
100.    
     
101.                   if   (status   <   0)   
     
102.                   {   
     
103.                           CloseHandle   (thread);   
     
104.                           SetLastError   (RtlNtStatusToDosError   (status));   
     
105.                           return   FALSE;   
     
106.                   };   
     
107.    
     
108.                   _tprintf   (TEXT   ("线程   %08x   所在进程ID为   %08x\n"),     
     
109.                           tid,     
     
110.                           (DWORD)tbi.ClientId.UniqueProcess);   
     
111.    
     
112.                   process   =   ::OpenProcess   (PROCESS_ALL_ACCESS,     
     
113.                           FALSE,     
     
114.                           (DWORD)tbi.ClientId.UniqueProcess);   
     
115.    
     
116.                   if   (process   ==   NULL)   
     
117.                   {   
     
118.                           DWORD   error   =   ::GetLastError   ();   
     
119.                           CloseHandle   (thread);   
     
120.                           SetLastError   (error);   
     
121.                           return   FALSE;   
     
122.                   };   
     
123.    
     
124.                   TCHAR   modname   [0x100];   
     
125.                   ::GetModuleFileNameEx   (process,   NULL,   modname,   0x100);   
     
126.    
     
127.                   _tprintf   (TEXT   ("线程   %08x   所在进程映象为   %s\n"),     
     
128.                           tid,     
     
129.                           modname);   
     
130.    
     
131.                   GetMappedFileName(process,     
     
132.                           startaddr,     
     
133.                           modname,     
     
134.                           0x100);   
     
135.    
     
136.                   _tprintf   (TEXT   ("线程   %08x   可执行代码所在模块为   %s\n"),     
     
137.                           tid,     
     
138.                           modname);   
     
139.    
     
140.                   CloseHandle   (process);   
     
141.                   CloseHandle   (thread);   
     
142.                   return   TRUE;   
     
143.           };   
     
144.    
     
145.   int   main   (void)   
     
146.           {   
     
147.                   setlocale   (LC_ALL,   ".ACP");   
     
148.    
     
149.                   HINSTANCE   hNTDLL   =   ::GetModuleHandle   (TEXT   ("ntdll"));   
     
150.    
     
151.                   (FARPROC&)ZwQueryInformationThread     =   
     
152.                           ::GetProcAddress   (hNTDLL,   "ZwQueryInformationThread");   
     
153.    
     
154.                   (FARPROC&)RtlNtStatusToDosError           =   
     
155.                           ::GetProcAddress   (hNTDLL,   "RtlNtStatusToDosError");   
     
156.    
     
157.                   HANDLE   h   =   CreateToolhelp32Snapshot   (TH32CS_SNAPTHREAD,   0);   
     
158.                   THREADENTRY32   te;   
     
159.                   te.dwSize   =   sizeof   (te);   
     
160.                   if   (Thread32First   (h,   &te))   
     
161.                   {   
     
162.                           do   
     
163.                           {   
     
164.                                   if   (ShowThreadInfo   (te.th32ThreadID))   
     
165.                                   {   
     
166.                                   }   
     
167.                                   else   
     
168.                                   {   
     
169.                                           _tprintf   (TEXT("无法获得线程   %08x   的相关信息，错误代码为   %d\n"),     
     
170.                                                   te.th32ThreadID,   GetLastError   ());   
     
171.                                   };   
     
172.                           }   while   (Thread32Next   (h,   &te));   
     
173.                   };   
     
174.                   CloseHandle   (h);   
     
175.           };  
     

复制代码