利用NtQuerySystemInformation函數遍歷進程,遍歷線程,獲取線程掛起或運行狀態

版權聲明：專注於計算機網絡安全學習 https://blog.csdn.net/u011672712/article/details/51586030

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241

#define NT_SUCCESS(Status) ((NTSTATUS)(Status) >= 0) #define STATUS_SUCCESS              ((NTSTATUS) 0x00000000) #define SystemProcessesAndThreadsInformation    5 // 功能號 #define NTAPI    __stdcall   // 線程狀態的枚舉常量 typedef  enum  _THREAD_STATE {      StateInitialized ,  // 初始化狀態      StateReady ,  // 准備狀態      StateRunning ,  // 運行狀態      StateStandby ,  //       StateTerminated , //關閉      StateWait ,  // 等待      StateTransition ,  // 切換???      StateUnknown }THREAD_STATE;   // 線程處於等待的原因的枚舉常量 typedef  enum  _KWAIT_REASON {      Executive ,      FreePage ,      PageIn ,      PoolAllocation ,      DelayExecution ,      Suspended ,      UserRequest ,      WrExecutive ,      WrFreePage ,      WrPageIn ,      WrPoolAllocation ,      WrDelayExecution ,      WrSuspended ,      WrUserRequest ,      WrEventPair ,      WrQueue ,      WrLpcReceive ,      WrLpcReply ,      WrVirtualMemory ,      WrPageOut ,      WrRendezvous ,      Spare2 ,      Spare3 ,      Spare4 ,      Spare5 ,      Spare6 ,      WrKernel ,      MaximumWaitReason }KWAIT_REASON;         typedef  LONG    NTSTATUS; typedef  LONG     KPRIORITY;   typedef  struct  _CLIENT_ID {      DWORD         UniqueProcess;      DWORD         UniqueThread; } CLIENT_ID , *PCLIENT_ID;     typedef  struct  _VM_COUNTERS {      SIZE_T         PeakVirtualSize;      SIZE_T         VirtualSize;      ULONG          PageFaultCount;      SIZE_T         PeakWorkingSetSize;      SIZE_T         WorkingSetSize;      SIZE_T         QuotaPeakPagedPoolUsage;      SIZE_T         QuotaPagedPoolUsage;      SIZE_T         QuotaPeakNonPagedPoolUsage;      SIZE_T         QuotaNonPagedPoolUsage;      SIZE_T         PagefileUsage;      SIZE_T         PeakPagefileUsage; } VM_COUNTERS;   // 線程信息結構體 typedef  struct  _SYSTEM_THREAD_INFORMATION {      LARGE_INTEGER   KernelTime;      LARGE_INTEGER   UserTime;      LARGE_INTEGER   CreateTime;      ULONG            WaitTime;      PVOID            StartAddress;      CLIENT_ID       ClientId;      KPRIORITY       Priority;      KPRIORITY       BasePriority;      ULONG            ContextSwitchCount;      LONG             State; // 狀態,是THREAD_STATE枚舉類型中的一個值      LONG             WaitReason; //等待原因, KWAIT_REASON中的一個值 } SYSTEM_THREAD_INFORMATION , *PSYSTEM_THREAD_INFORMATION;     typedef  struct  _UNICODE_STRING {      USHORT  Length;      USHORT  MaximumLength;      PWSTR   Buffer; } UNICODE_STRING , *PUNICODE_STRING;     // 進程信息結構體 typedef  struct  _SYSTEM_PROCESS_INFORMATION {      ULONG             NextEntryDelta;  // 指向下一個結構體的指針      ULONG             ThreadCount;  // 本進程的總線程數      ULONG             Reserved1[ 6 ];  // 保留      LARGE_INTEGER    CreateTime;  // 進程的創建時間      LARGE_INTEGER    UserTime;  // 在用戶層的使用時間      LARGE_INTEGER    KernelTime;  // 在內核層的使用時間      UNICODE_STRING   ProcessName;  // 進程名      KPRIORITY        BasePriority;  //       ULONG             ProcessId;  // 進程ID      ULONG             InheritedFromProcessId;      ULONG             HandleCount;  // 進程的句柄總數      ULONG             Reserved2[ 2 ];  // 保留      VM_COUNTERS      VmCounters;      IO_COUNTERS      IoCounters;      SYSTEM_THREAD_INFORMATION Threads[ 5 ];  // 子線程信息數組 }SYSTEM_PROCESS_INFORMATION , *PSYSTEM_PROCESS_INFORMATION;     // NtQuerySystemInformation 函數的原型 // 由於該沒有導出,所以得自己定義函數的原型 typedef  DWORD (WINAPI* PQUERYSYSTEM)( UINT  ,  PVOID  ,  DWORD  , PDWORD);   /* ※※* 函數名 : IsThreadSuspend ※※* 功  能 : 獲取線程是被是否被掛起 ※※* 返回值 : int 1： 表示線程被掛起 , 0： 表示線程正常。 -1 未知狀態 ※※* 形  參 : DWORD dwThreadID 要驗證的線程的ID ※※* 形  參 : wchar_t * szProcessName 表示線程所在的進程名 */ int  IsThreadSuspend( DWORD  dwProcessID ,  DWORD  dwThreadID) {      int  nRet = 0;      NTSTATUS Status = 0;          PQUERYSYSTEM NtQuerySystemInformation = NULL;      PSYSTEM_PROCESS_INFORMATION pInfo = { 0 };        // 獲取函數地址      NtQuerySystemInformation = (PQUERYSYSTEM)          GetProcAddress(LoadLibrary(L "ntdll.dll" ) ,  "NtQuerySystemInformation" );          DWORD    dwSize = 0;      // 獲取信息所需的緩沖區大小      Status = NtQuerySystemInformation(SystemProcessesAndThreadsInformation , //要獲取的信息的類型                                        NULL ,  // 用於接收信息的緩沖區                                        0 ,   // 緩沖區大小                                        &dwSize                                        );      // 申請緩沖區      char* pBuff =  new  char [ dwSize ]; 　　　　pInfo = (PSYSTEM_PROCESS_INFORMATION)pBuff ;      if (pInfo == NULL)          return  -1;      // 再次調用函數, 獲取信息      Status = NtQuerySystemInformation(SystemProcessesAndThreadsInformation , //要獲取的信息的類型                                        pInfo ,  // 用於接收信息的緩沖區                                        dwSize ,   // 緩沖區大小                                        &dwSize                                        );      if (!NT_SUCCESS(Status)) { /*如果函數執行失敗*/          printf ( "失敗\n" );          delete [] pInfo;          return  -1;      }        // 遍歷結構體,找到對應的進程      while (1) {          // 判斷是否還有下一個進程          if (pInfo->NextEntryDelta == 0)              break ;            // 判斷是否找到了ID          if (pInfo->ProcessId == dwProcessID) {                // 找到該進程下的對應的線程,也就是遍歷所有線程              for ( DWORD  i = 0; i < pInfo->ThreadCount ; i++) {                  if (pInfo->Threads[ i ].ClientId.UniqueThread == dwThreadID) {  //找到線程                       //如果線程被掛起                      if (pInfo->Threads[ i ].State == StateWait                         && pInfo->Threads[ i ].WaitReason == Suspended) {                          nRet = 1;                          break ;                      }                  }              }              break ;          }          // 迭代到下一個節點          pInfo = (PSYSTEM_PROCESS_INFORMATION)((( PUCHAR )pInfo) + pInfo->NextEntryDelta);      }        delete [] pBuff;      return  nRet; }       /* ※※* 函數名 : IsThreadAlive ※※* 功  能 :  判斷線程是否被終止 , 如果終止返回FALSE,如果還活着返回TRUE ※※* 返回值 : BOOL ※※* 形  參 : DWORD dwThreadID 線程ID */ BOOL  IsThreadAlive( DWORD  dwThreadID) {      BOOL  bRet = FALSE;      DWORD  ExitCode = 0;        HANDLE  hThread = OpenThread(THREAD_QUERY_INFORMATION , FALSE , dwThreadID);      if (hThread != NULL) {          if (GetExitCodeThread(hThread , &ExitCode)) {              if (ExitCode == STILL_ACTIVE)                  bRet = TRUE;          }          CloseHandle(hThread);      }      return  bRet; }           int  main() {      BOOL  bRET = IsThreadSuspend(2640 , 1111);      if (bRET) {          printf ( " 2320線程被掛起了！" );      }

 

jpg轉 rar