3,245
社区成员
发帖
与我相关
我的任务
分享
BOOL lantest::ManageWMI(char *str)
{
HRESULT hres;
char szTmp[40960];
//http://www.189works.com/article-9608-1.html
// Step 1: --------------------------------------------------
// Initialize COM. ------------------------------------------
hres = CoInitializeEx(NULL, COINIT_MULTITHREADED);
// Step 2: --------------------------------------------------
// Set general COM security levels --------------------------
// Note: If you are using Windows 2000, you need to specify -
// the default authentication credentials for a user by using
// a SOLE_AUTHENTICATION_LIST structure in the pAuthList ----
// parameter of CoInitializeSecurity ------------------------
hres = CoInitializeSecurity(
NULL,
-1, // COM authentication
NULL, // Authentication services
NULL, // Reserved
RPC_C_AUTHN_LEVEL_DEFAULT, // Default authentication
RPC_C_IMP_LEVEL_IMPERSONATE, // Default Impersonation
NULL, // Authentication info
EOAC_NONE, // Additional capabilities
NULL // Reserved
);
if (FAILED(hres)) {
sprintf(str,"Failed to initialize security. Error code = 0x%X/r/n",hres);
CoUninitialize();
return 1; // Program has failed.
}
// Step 3: ---------------------------------------------------
// Obtain the initial locator to WMI -------------------------
IWbemLocator *pLoc = NULL;
hres = CoCreateInstance( CLSID_WbemLocator, 0, CLSCTX_INPROC_SERVER, IID_IWbemLocator, (LPVOID *) &pLoc);
if (FAILED(hres)) {
sprintf(str,"Failed to create IWbemLocator object. Err code = 0x%X/r/n",hres);
CoUninitialize();
return 1; // Program has failed.
}
// Step 4: -----------------------------------------------------
// Connect to WMI through the IWbemLocator::ConnectServer method
IWbemServices *pSvc = NULL;
// Connect to the root/cimv2 namespace with
// the current user and obtain pointer pSvc
// to make IWbemServices calls.
hres = pLoc->ConnectServer(
_bstr_t(L"root\\cimv2"), // Object path of WMI namespace
NULL, // User name. NULL = current user
NULL, // User password. NULL = current
0, // Locale. NULL indicates current
NULL, // Security flags.
0, // Authority (e.g. Kerberos)
0, // Context object
&pSvc // pointer to IWbemServices proxy
);
if (FAILED(hres)) {
sprintf(str,"Could not connect. Error code = 0x%X/r/n",hres);
pLoc->Release();
CoUninitialize();
return 1; // Program has failed.
}
sprintf(str,"Connected to ROOT//CIMV2 WMI namespace");
// Step 5: --------------------------------------------------
// Set security levels on the proxy -------------------------
hres = CoSetProxyBlanket(
pSvc, // Indicates the proxy to set
RPC_C_AUTHN_WINNT, // RPC_C_AUTHN_xxx
RPC_C_AUTHZ_NONE, // RPC_C_AUTHZ_xxx
NULL, // Server principal name
RPC_C_AUTHN_LEVEL_CALL, // RPC_C_AUTHN_LEVEL_xxx
RPC_C_IMP_LEVEL_IMPERSONATE, // RPC_C_IMP_LEVEL_xxx
NULL, // client identity
EOAC_NONE // proxy capabilities
);
if (FAILED(hres)) {
sprintf(str,"Could not set proxy blanket. Error code = 0x%X/r/n",hres);
pSvc->Release();
pLoc->Release();
CoUninitialize();
return 1; // Program has failed.
}
// Step 6: --------------------------------------------------
// Use the IWbemServices pointer to make requests of WMI ----
// For example, get the name of the operating system
IEnumWbemClassObject* pEnumerator = NULL;
IWbemClassObject *pclsObj;
ULONG uReturn = 0;
// get baseboard's infomation
hres = pSvc->ExecQuery(
bstr_t("WQL"),
//bstr_t("SELECT * FROM Win32_BaseBoard"),// WHERE IPEnabled = 'TRUE'"),
//bstr_t("SELECT * FROM Win32_Process"),
//bstr_t("SELECT * FROM Win32_NetworkAdapter"),
bstr_t("SELECT * FROM Win32_NetworkAdapterConfiguration"),
WBEM_FLAG_FORWARD_ONLY | WBEM_FLAG_RETURN_IMMEDIATELY,
NULL,
&pEnumerator);
if (FAILED(hres)) {
sprintf(str,"Query for BaseBoard's infomation failed. Error code = 0x%X/r/n", hres);
pSvc->Release();
pLoc->Release();
CoUninitialize();
return 1; // Program has failed.
}
uReturn = 0;
BSTR strClassProp = SysAllocString(L"__CLASS");
//http://msdn.microsoft.com/en-us/library/windows/desktop/aa394216(v=vs.85).aspx
//http://blog.csdn.net/wangjieest/article/details/7320572
while (pEnumerator)
{
HRESULT hr = pEnumerator->Next(WBEM_INFINITE, 1, &pclsObj, &uReturn);
if(0 == uReturn)
{
break;
}
VARIANT vtProp;
VariantInit(&vtProp);
hr = pclsObj->Get(L"IPXMediaType", 0, &vtProp, 0, 0);
//hr = pclsObj->Get(L"Name", 0, &vtProp, 0, 0);//当前运行进程
if (vtProp.bstrVal)
{
NdisMediaConnectStatus++;
}
VariantClear(&vtProp);
}
if (pSvc)
pSvc->Release();
if (pLoc)
pLoc->Release();
if (pEnumerator)
pEnumerator->Release();
if (pclsObj)
pclsObj->Release();
CoUninitialize();
return 0; // Program successfully completed.
}