#include "precomp.h"
#include <strsafe.h>
#include <initguid.h>
#include <objbase.h>
#include <wbemcli.h>
#include <debug.h>

Include dependency graph for wmi.c:

Macros
#define	COBJMACROS

#define	NDEBUG

Functions
BOOL	CheckOsVersion (_In_ PNS_OSVERSIONCHECK pfnOsVersionCheck)

static HRESULT	QueryOperatingSystemInfo (_In_ IWbemServices *Services)

static HRESULT	QueryProcessorInfo (_In_ IWbemServices *Services)

HRESULT	GetWmiVersionInfo (VOID)

Variables
UINT	VersionInfoArchitecture

UINT	VersionInfoOsProductSuite

UINT	VersionInfoOsType

WCHAR	VersionInfoVersion [MAX_PATH]

WCHAR	VersionInfoBuildNumber [MAX_PATH]

WCHAR	VersionInfoServicePackMajorVersion [MAX_PATH]

WCHAR	VersionInfoServicePackMinorVersion [MAX_PATH]

Macro Definition Documentation

◆ COBJMACROS

#define COBJMACROS

Definition at line 10 of file wmi.c.

◆ NDEBUG

#define NDEBUG

Definition at line 19 of file wmi.c.

Function Documentation

◆ CheckOsVersion()

BOOL CheckOsVersion ( _In_ PNS_OSVERSIONCHECK pfnOsVersionCheck )

Definition at line 35 of file wmi.c.

{
    DPRINT("CheckOsVersion(%p)\n", pfnOsVersionCheck);
 
    if (pfnOsVersionCheck == NULL)
        return TRUE;
 
    return pfnOsVersionCheck(VersionInfoOsType,
                             VersionInfoOsProductSuite,
                             VersionInfoVersion,
                             VersionInfoBuildNumber,
                             VersionInfoServicePackMajorVersion,
                             VersionInfoServicePackMinorVersion,
                             VersionInfoArchitecture,
                             0);
}

Referenced by InterpretCommand(), PrintCommandHelp(), PrintContext(), PrintShortGroupCommands(), and PrintSubcontexts().

◆ GetWmiVersionInfo()

HRESULT GetWmiVersionInfo ( VOID )

Definition at line 246 of file wmi.c.

{
    BSTR Path = NULL;
    IWbemLocator *Locator = NULL;
    IWbemServices *Services = NULL;
    HRESULT hr;
 
    CoInitialize(NULL);
 
    CoInitializeSecurity(NULL, -1, NULL, NULL, RPC_C_AUTHN_LEVEL_DEFAULT,
                         RPC_C_IMP_LEVEL_IMPERSONATE, NULL, EOAC_NONE, NULL);
    hr = CoCreateInstance(&CLSID_WbemLocator, NULL, CLSCTX_INPROC_SERVER,
                          &IID_IWbemLocator, (void **)&Locator);
    if (hr != S_OK)
    {
        DPRINT1("Can't create instance of WbemLocator\n");
        return hr;
    }
 
    Path = SysAllocString(L"ROOT\\CIMV2");
    if (Path == NULL)
    {
        hr = E_OUTOFMEMORY;
        goto done;
    }
 
    hr = IWbemLocator_ConnectServer(Locator, Path, NULL, NULL, NULL, 0, NULL, NULL, &Services);
    if (hr != S_OK)
    {
        DPRINT1("Failed to get IWbemServices interface %08x\n", hr);
        goto done;
    }
 
    hr = CoSetProxyBlanket((IUnknown *)Services, RPC_C_AUTHN_WINNT, RPC_C_AUTHZ_NONE, NULL,
                           RPC_C_AUTHN_LEVEL_CALL, RPC_C_IMP_LEVEL_IMPERSONATE, NULL, EOAC_NONE);
    if (hr != S_OK)
    {
        DPRINT1("Failed to set proxy blanket %08x\n", hr);
        goto done;
    }
 
    hr = QueryOperatingSystemInfo(Services);
    if (hr == S_OK)
        hr = QueryProcessorInfo(Services);
 
done:
    if (Path)
        SysFreeString(Path);
 
    if (Services)
        IWbemServices_Release(Services);
 
    if (Locator)
        IWbemLocator_Release(Locator);
 
    CoUninitialize();
 
    return hr;
}

Referenced by wmain().

◆ QueryOperatingSystemInfo()

static HRESULT QueryOperatingSystemInfo ( _In_ IWbemServices * Services )

static

Definition at line 55 of file wmi.c.

{
    IEnumWbemClassObject *Result = NULL;
    IWbemClassObject *Object = NULL;
    BSTR Wql = NULL;
    BSTR Query = NULL;
    BSTR Name;
    CIMTYPE Type;
    ULONG Count;
    VARIANT Value;
    HRESULT hr;
 
    Wql = SysAllocString(L"WQL");
    if (Wql == NULL)
        return E_OUTOFMEMORY;
    
    Query = SysAllocString(L"SELECT * FROM Win32_OperatingSystem");
    if (Query == NULL)
    {
        hr = E_OUTOFMEMORY;
        goto done;
    }
 
    hr = IWbemServices_ExecQuery(Services, Wql, Query, 0, NULL, &Result);
    if (hr !=  S_OK)
    {
        DPRINT1("IWbemServices_ExecQuery failed %08x\n", hr);
        goto done;
    }
 
    hr = IEnumWbemClassObject_Next(Result, 10000, 1, &Object, &Count);
    if (hr != S_OK)
    {
        DPRINT1("IEnumWbemClassObject_Next failed %08x\n", hr);
        goto done;
    }
 
    hr = IWbemClassObject_BeginEnumeration(Object, 0);
    if (hr != S_OK)
    {
        DPRINT1("IWbemClassObject_BeginEnumeration returned %08x\n", hr);
        goto done;
    }
 
    while (IWbemClassObject_Next(Object, 0, &Name, &Value, &Type, NULL) == S_OK)
    {
        DPRINT("Name: %S\n", Name);
        if (_wcsicmp(Name, L"BuildNumber") == 0)
        {
            DPRINT("BuildNumber %S\n", V_BSTR(&Value));
            StringCbCopyW(VersionInfoBuildNumber, sizeof(VersionInfoBuildNumber), V_BSTR(&Value));
        }
        else if (_wcsicmp(Name, L"OSProductSuite") == 0)
        {
            DPRINT("OSProductSuite 0x%x\n", V_UINT(&Value));
            VersionInfoOsProductSuite = V_UINT(&Value);
        }
        else if (_wcsicmp(Name, L"OSType") == 0)
        {
            DPRINT("OSType %u\n", V_UINT(&Value));
            VersionInfoOsType = V_UINT(&Value);
        }
        else if (_wcsicmp(Name, L"ServicePackMajorVersion") == 0)
        {
            DPRINT("ServicePackMajorVersion %hu\n", V_UINT(&Value));
            _swprintf(VersionInfoServicePackMajorVersion, L"%hu", V_UINT(&Value));
        }
        else if (_wcsicmp(Name, L"ServicePackMinorVersion") == 0)
        {
            DPRINT("ServicePackMinorVersion %hu\n", V_UINT(&Value));
            _swprintf(VersionInfoServicePackMinorVersion, L"%hu", V_UINT(&Value));
        }
        else if (_wcsicmp(Name, L"Version") == 0)
        {
            DPRINT("Version %S\n", V_BSTR(&Value));
            StringCbCopyW(VersionInfoVersion, sizeof(VersionInfoVersion), V_BSTR(&Value));
        }
 
        SysFreeString(Name);
        VariantClear(&Value);
    }
 
    hr = IWbemClassObject_EndEnumeration(Object);
    if (hr != S_OK)
    {
        DPRINT1("got %08x\n", hr);
        goto done;
    }
 
done:
    if (Object)
        IWbemClassObject_Release(Object);
 
    if (Result)
        IEnumWbemClassObject_Release(Result);
 
    if (Query)
        SysFreeString(Query);
 
    if (Wql)
        SysFreeString(Wql);
 
    return hr;
}

Referenced by GetWmiVersionInfo().

◆ QueryProcessorInfo()

static HRESULT QueryProcessorInfo ( _In_ IWbemServices * Services )

static

Definition at line 164 of file wmi.c.

{
    IEnumWbemClassObject *Result = NULL;
    IWbemClassObject *Object = NULL;
    BSTR Wql = NULL;
    BSTR Query = NULL;
    BSTR Name;
    CIMTYPE Type;
    ULONG Count;
    VARIANT Value;
    HRESULT hr;
 
    Wql = SysAllocString(L"WQL");
    if (Wql == NULL)
        return E_OUTOFMEMORY;
    
    Query = SysAllocString(L"SELECT * FROM Win32_Processor");
    if (Query == NULL)
    {
        hr = E_OUTOFMEMORY;
        goto done;
    }
 
    hr = IWbemServices_ExecQuery(Services, Wql, Query, 0, NULL, &Result);
    if (hr !=  S_OK)
    {
        DPRINT1("IWbemServices_ExecQuery failed %08x\n", hr);
        goto done;
    }
 
    hr = IEnumWbemClassObject_Next(Result, 10000, 1, &Object, &Count);
    if (hr != S_OK)
    {
        DPRINT1("IEnumWbemClassObject_Next failed %08x\n", hr);
        goto done;
    }
 
    hr = IWbemClassObject_BeginEnumeration(Object, 0);
    if (hr != S_OK)
    {
        DPRINT1("IWbemClassObject_BeginEnumeration returned %08x\n", hr);
        goto done;
    }
 
    while (IWbemClassObject_Next(Object, 0, &Name, &Value, &Type, NULL) == S_OK)
    {
        DPRINT("Name: %S\n", Name);
        if (_wcsicmp(Name, L"Architecture") == 0)
        {
            DPRINT("Architecture %u\n", V_UINT(&Value));
            VersionInfoArchitecture = V_UINT(&Value);
        }
 
        SysFreeString(Name);
        VariantClear(&Value);
    }
 
    hr = IWbemClassObject_EndEnumeration(Object);
    if (hr != S_OK)
    {
        DPRINT1("got %08x\n", hr);
        goto done;
    }
 
done:
    if (Object)
        IWbemClassObject_Release(Object);
 
    if (Result)
        IEnumWbemClassObject_Release(Result);
 
    if (Query)
        SysFreeString(Query);
 
    if (Wql)
        SysFreeString(Wql);
 
    return hr;
}