使用WMI获取正在运行的进程

1. #define _WIN32_DCOM 
3. #include "stdafx.h" 
4. #include <iostream> 
5. #include <comdef.h> 
6. #include <Wbemidl.h> 
7. #include <Wbemcli.h>
8. #include <conio.h> 
11. #pragma comment(lib, "wbemuuid.lib") 
13. using namespace std; 
15. const char * processName = "System";
17. BOOL ManageWMI(); 
18. int _tmain(int argc, _TCHAR* argv[]) 
19. { 
20.     if(ManageWMI()) printf("%WMI Error!"); 
21.     _getch(); 
22.     return 0; 
23. }
25. int FindProcess(IWbemServices *pSvc, const char * processName)
26. {
27.     HRESULT hres;
29.     // Step 6: -------------------------------------------------- 
30.     // Use the IWbemServices pointer to make requests of WMI ---- 
32.     // For example, get the name of the operating system 
33.     IEnumWbemClassObject* pEnumerator = NULL; 
34.     hres = pSvc->ExecQuery(bstr_t("WQL"), bstr_t("SELECT * FROM Win32_Process WHERE Description = 'System'"), WBEM_FLAG_FORWARD_ONLY | WBEM_FLAG_RETURN_IMMEDIATELY, NULL, &pEnumerator); 
36.     if (FAILED(hres)) 
37.     { 
38.         cout << "Query for Process Configuration failed." 
39.             << " Error code = 0x" 
40.             << hex << hres << endl; 
41.         pSvc->Release(); 
42.         return 1; // Program has failed. 
43.     } 
45.     // Step 7: ------------------------------------------------- 
46.     // Get the data from the query in step 6 ------------------- 
48.     IWbemClassObject *pclsObj; 
49.     ULONG uReturn = 0; 
51.     int processId = 0;
52.     while (pEnumerator) 
53.     { 
54.         HRESULT hr = pEnumerator->Next(WBEM_INFINITE, 1, &pclsObj, &uReturn); 
56.         if(0 == uReturn) 
57.         { 
58.             break; 
59.         } 
61.         VARIANT vtProp; 
62.         VariantInit(&vtProp); 
64.         hr = pclsObj->Get(L"ProcessId", 0, &vtProp, 0, 0); 
65.         processId = vtProp.intVal;
67.         VariantClear(&vtProp); 
68.     } 
70.     pEnumerator->Release(); 
71.     pclsObj->Release(); 
73.     return processId;
74. }
76. BOOL ManageWMI() 
77. { 
78.     HRESULT hres; 
80.     // Step 1: -------------------------------------------------- 
81.     // Initialize COM. ------------------------------------------ 
83.     hres = CoInitializeEx(0, COINIT_MULTITHREADED); 
84.     if (FAILED(hres)) 
85.     { 
86.         cout << "Failed to initialize COM library. Error code = 0x" 
87.             << hex << hres << endl; 
88.         return 1; // Program has failed. 
89.     } 
91.     // Step 2: -------------------------------------------------- 
92.     // Set general COM security levels -------------------------- 
93.     // Note: If you are using Windows 2000, you need to specify - 
94.     // the default authentication credentials for a user by using 
95.     // a SOLE_AUTHENTICATION_LIST structure in the pAuthList ---- 
96.     // parameter of CoInitializeSecurity ------------------------ 
98.     hres = CoInitializeSecurity( 
99.         NULL, 
100.         -1, // COM authentication 
101.         NULL, // Authentication services 
102.         NULL, // Reserved 
103.         RPC_C_AUTHN_LEVEL_DEFAULT, // Default authentication 
104.         RPC_C_IMP_LEVEL_IMPERSONATE, // Default Impersonation 
105.         NULL, // Authentication info 
106.         EOAC_NONE, // Additional capabilities 
107.         NULL // Reserved 
108.         ); 
111.     if (FAILED(hres)) 
112.     { 
113.         cout << "Failed to initialize security. Error code = 0x" 
114.             << hex << hres << endl; 
115.         CoUninitialize(); 
116.         return 1; // Program has failed. 
117.     } 
119.     // Step 3: --------------------------------------------------- 
120.     // Obtain the initial locator to WMI ------------------------- 
122.     IWbemLocator *pLoc = NULL; 
124.     hres = CoCreateInstance( 
125.         CLSID_WbemLocator, 
126.         0, 
127.         CLSCTX_INPROC_SERVER, 
128.         IID_IWbemLocator, (LPVOID *) &pLoc); 
130.     if (FAILED(hres)) 
131.     { 
132.         cout << "Failed to create IWbemLocator object." 
133.             << " Err code = 0x" 
134.             << hex << hres << endl; 
135.         CoUninitialize(); 
136.         return 1; // Program has failed. 
137.     } 
139.     // Step 4: ----------------------------------------------------- 
140.     // Connect to WMI through the IWbemLocator::ConnectServer method 
142.     IWbemServices *pSvc = NULL; 
144.     // Connect to the root/cimv2 namespace with 
145.     // the current user and obtain pointer pSvc 
146.     // to make IWbemServices calls. 
147.     hres = pLoc->ConnectServer( 
148.         _bstr_t(L"ROOT//CIMV2"), // Object path of WMI namespace 
149.         NULL, // User name. NULL = current user 
150.         NULL, // User password. NULL = current 
151.         0, // Locale. NULL indicates current 
152.         NULL, // Security flags. 
153.         0, // Authority (e.g. Kerberos) 
154.         0, // Context object 
155.         &pSvc // pointer to IWbemServices proxy 
156.         ); 
158.     if (FAILED(hres)) 
159.     { 
160.         cout << "Could not connect. Error code = 0x" 
161.             << hex << hres << endl; 
162.         pLoc->Release(); 
163.         CoUninitialize(); 
164.         return 1; // Program has failed. 
165.     } 
167.     cout << "Connected to ROOT//CIMV2 WMI namespace" << endl; 
170.     // Step 5: -------------------------------------------------- 
171.     // Set security levels on the proxy ------------------------- 
173.     hres = CoSetProxyBlanket( 
174.         pSvc, // Indicates the proxy to set 
175.         RPC_C_AUTHN_WINNT, // RPC_C_AUTHN_xxx 
176.         RPC_C_AUTHZ_NONE, // RPC_C_AUTHZ_xxx 
177.         NULL, // Server principal name 
178.         RPC_C_AUTHN_LEVEL_CALL, // RPC_C_AUTHN_LEVEL_xxx 
179.         RPC_C_IMP_LEVEL_IMPERSONATE, // RPC_C_IMP_LEVEL_xxx 
180.         NULL, // client identity 
181.         EOAC_NONE // proxy capabilities 
182.         ); 
184.     if (FAILED(hres)) 
185.     { 
186.         cout << "Could not set proxy blanket. Error code = 0x" 
187.             << hex << hres << endl; 
188.         pSvc->Release(); 
189.         pLoc->Release(); 
190.         CoUninitialize(); 
191.         return 1; // Program has failed. 
192.     } 
194.     int processId = FindProcess(pSvc, processName);
195.     if(0 == processId)
196.     {
197.         pLoc->Release(); 
198.         CoUninitialize(); 
199.         cout << "process not found" << endl;
200.         return 1;
201.     }
203.     // Cleanup 
204.     // ======== 
206.     pSvc->Release(); 
207.     pLoc->Release(); 
208.     CoUninitialize(); 
210.     return 0; // Program successfully completed. 
212. }