WMI C++

WMI가 정말 흥미로워서 공부를 해보았는데 공부한건 천천히 쓸게요.

WMI MSDN 을 보시려면 이곳을 클릭하세요.

코드 부분입니다.

wmi.h

#pragma once
#define _WIN32_DCOM
 
#include <comdef.h>
#include <WbemIdl.h>
#pragma comment(lib, "wbemuuid.lib")
 
#include <iostream>
using namespace std;
 
class Wmi
{
private:
    HRESULT hRes;
    IWbemLocator *pLoc;
    IWbemServices *pSvc;
    IEnumWbemClassObject *pEnum;
    IWbemClassObject *pclsObj;
    int WmiComInit();
    int WmiComSecurity();
    int WmiComCreate();
    int WmiComConnect();
    int WmiComProxy();
public:
    Wmi();
    ~Wmi();
    int WmiComExcute(bstr_t, const wchar_t*);
};

wmi.cpp

#include "wmi.h"
 
Wmi::Wmi()
{
    WmiComInit();
    WmiComSecurity();
    WmiComCreate();
    WmiComConnect();
    WmiComProxy();
}
Wmi::~Wmi()
{
    pSvc->Release();
    pLoc->Release();
    pEnum->Release();
    CoUninitialize();
}
 
int Wmi::WmiComInit()
{
    hRes = CoInitializeEx(0, COINIT_MULTITHREADED);
    if (FAILED(hRes))
    {
        cout << "CoInitializeEx func err : " << hex << hRes << endl;
        return 1;
    }
 
    return 0;
}
 
int Wmi::WmiComSecurity()
{
    hRes = CoInitializeSecurity(
        NULL,
        -1,
        NULL,
        NULL,
        RPC_C_AUTHN_LEVEL_DEFAULT,
        RPC_C_IMP_LEVEL_IMPERSONATE,
        NULL,
        EOAC_NONE,
        NULL);
    if (FAILED(hRes))
    {
        cout << "CoInitializeSecurity Func err : " << hex << hRes << endl;
        CoUninitialize();
        return 1;
    }
    return 0;
}
 
int Wmi::WmiComCreate()
{
    pLoc = NULL;
 
    hRes = CoCreateInstance(
        CLSID_WbemLocator,
        0,
        CLSCTX_INPROC_SERVER,
        IID_IWbemLocator,
        (LPVOID *)&pLoc);
    if (FAILED(hRes))
    {
        cout << "CoCreateInstance Func err : " << hex << hRes << endl;
        CoUninitialize();
        return 1;
    }
    return 0;
}
 
int Wmi::WmiComConnect()
{
    pSvc = NULL;
 
    hRes = pLoc->ConnectServer(
        _bstr_t(L"ROOT\\CIMV2"),
        NULL,
        NULL,
        0,
        NULL,
        0,
        0,
        &pSvc);
    if (FAILED(hRes))
    {
        cout << "ConnectServer Func err : " << hex << hRes << endl;
        pLoc->Release();
        CoUninitialize();
        return 1;
    }
    return 0;
}
 
int Wmi::WmiComProxy()
{
    hRes = CoSetProxyBlanket(
        pSvc,
        RPC_C_AUTHN_WINNT,
        RPC_C_AUTHZ_NONE,
        NULL,
        RPC_C_AUTHN_LEVEL_CALL,
        RPC_C_IMP_LEVEL_IMPERSONATE,
        NULL,
        EOAC_NONE);
    if (FAILED(hRes))
    {
        cout << "CoSetProxyBlanket Func err : " << hex << hRes << endl;
        pSvc->Release();
        pLoc->Release();
        CoUninitialize();
        return 1;
    }
    return 0;
}
 
//윗 부분까지는 공통부분이고 이 부분부터는 수정을 하셔서 사용하셔됩니다.
int Wmi::WmiComExcute(bstr_t query,const wchar_t *object)
{
    pEnum = NULL;
 
    hRes = pSvc->ExecQuery(
        bstr_t("WQL"),
        query,
        WBEM_FLAG_FORWARD_ONLY | WBEM_FLAG_RETURN_IMMEDIATELY,
        NULL,
        &pEnum);
    if (FAILED(hRes))
    {
        cout << "ExecQuery Func err : " << hex << hRes << endl;
        pSvc->Release();
        pLoc->Release();
        CoUninitialize();
        return 1;
    }
 
    pclsObj = NULL;
    ULONG uRet = 0;
 
    while (pEnum)
    {
        HRESULT hr = pEnum->Next(WBEM_INFINITE, 1, &pclsObj, &uRet);
        if (0 == uRet)
            break;
 
        VARIANT vtProp;
 
        hr = pclsObj->Get(object, 0, &vtProp, 0, 0);
        VariantClear(&vtProp);
 
        //example
        wcout << vtProp.bstrVal << endl;
 
        pclsObj->Release();
    }
 
    return 0;
}

main.cpp

#include "wmi.h"
 
int main()
{
    Wmi t;
 
    //query ex) SELECT * FROM Win32_LogicalDisk
    //object ex) Name
    t.WmiComExcute("SELECT * FROM Win32_LogicalDisk", L"Name");
}

호출하실 땐 그냥 WmiComExcute 만 호출하시면 됩니다.

첫번째 파라미터에는 쿼리문을 넣어주시면 되고 두번째는 클래스 인자가 들어가는데 이건 가변 인자로 만들려다 사용되는게 그때 그때 달라서 임시로 wchar_t 형으로 넣었어요.