#define _WIN32_DCOM
#define UNICODE
#include <iostream>
using namespace std;
#include <comdef.h>
#include <Wbemidl.h>
# pragma comment(lib, "wbemuuid.lib")
# pragma comment(lib, "credui.lib")
# pragma comment(lib, "comsuppw.lib")
#include <wincred.h>
#include <strsafe.h>
#include <math.h>
#include <cmath>

HRESULT hres;
IWbemLocator *pLoc;
IWbemServices *pSvc;

int __cdecl main(int argc, char **argv)
{
    // Step 1: --------------------------------------------------
    // Initialize COM. ------------------------------------------

	hres =  CoInitializeEx(0, COINIT_MULTITHREADED); 
    if (FAILED(hres))
    {
        cout << "Failed to initialize COM library. Error code = 0x" << hex << hres << endl;
        return 1;                  // Program has failed.
    }

    // Step 2: --------------------------------------------------
    // Set general COM security levels --------------------------

    hres =  CoInitializeSecurity(
        NULL, 
        -1,                          // COM authentication
        NULL,                        // Authentication services
        NULL,                        // Reserved
        RPC_C_AUTHN_LEVEL_DEFAULT,   // Default authentication 
        RPC_C_IMP_LEVEL_IDENTIFY,    // Default Impersonation  
        NULL,                        // Authentication info
        EOAC_NONE,                   // Additional capabilities 
        NULL                         // Reserved
        );

    if (FAILED(hres))
    {
        cout << "Failed to initialize security. Error code = 0x" << hex << hres << endl;
        CoUninitialize();
        return 1;                    // Program has failed.
    }
    
    // Step 3: ---------------------------------------------------
    // Obtain the initial locator to WMI -------------------------

    pLoc = NULL;

    hres = CoCreateInstance(
        CLSID_WbemLocator,             
        0, 
        CLSCTX_INPROC_SERVER, 
        IID_IWbemLocator, (LPVOID *) &pLoc);
 
    if (FAILED(hres))
    {
        cout << "Failed to create IWbemLocator object." << " Err code = 0x" << hex << hres << endl;
        CoUninitialize();
        return 1;                 // Program has failed.
    }

    // Step 4: -----------------------------------------------------
    // Connect to WMI through the IWbemLocator::ConnectServer method

    pSvc = NULL;
   
    hres = pLoc->ConnectServer(
        _bstr_t(L"\\\\178.207.59.214\\root\\cimv2"),
        NULL,                              // User name
        NULL,                              // User password
        NULL,                              // Locale             
        NULL,                              // Security flags
        NULL,                              // Authority        
        NULL,                              // Context object 
        &pSvc                              // IWbemServices proxy
        );
    
    if (FAILED(hres))
    {
        cout << "Could not connect. Error code = 0x" << hex << hres << endl;
        pLoc->Release();     
        CoUninitialize();
        return 1;                // Program has failed.
    }
    
    // Step 6: --------------------------------------------------
    // Set security levels on a WMI connection ------------------

    hres = CoSetProxyBlanket(
       pSvc,                           // Indicates the proxy to set
       RPC_C_AUTHN_DEFAULT,            // RPC_C_AUTHN_xxx
       RPC_C_AUTHZ_DEFAULT,            // RPC_C_AUTHZ_xxx
       COLE_DEFAULT_PRINCIPAL,         // Server principal name 
       RPC_C_AUTHN_LEVEL_PKT_PRIVACY,  // 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))
    {
        cout << "Could not set proxy blanket. Error code = 0x" << hex << hres << endl;
        pSvc->Release();
        pLoc->Release();     
        CoUninitialize();
        return 1;               // Program has failed.
    }

    // Step 7: --------------------------------------------------
    // Use the IWbemServices pointer to make requests of WMI ----
    IEnumWbemClassObject* pEnumerator = NULL;
    hres = pSvc->ExecQuery(
        bstr_t("WQL"), 
		bstr_t("SELECT Name, ProcessId, Caption, VirtualSize from Win32_Process"),
        WBEM_FLAG_FORWARD_ONLY | WBEM_FLAG_RETURN_IMMEDIATELY, 
        NULL,
        &pEnumerator);
    
    if (FAILED(hres))
    {
        cout << "Query for operating system name failed." << " Error code = 0x" << hex << hres << endl;
        pSvc->Release();
        pLoc->Release();
        CoUninitialize();
        return 1;               // Program has failed.
    }

    // Step 8: -------------------------------------------------
    // Secure the enumerator proxy
    hres = CoSetProxyBlanket(
        pEnumerator,                    // Indicates the proxy to set
        RPC_C_AUTHN_DEFAULT,            // RPC_C_AUTHN_xxx
        RPC_C_AUTHZ_DEFAULT,            // RPC_C_AUTHZ_xxx
        COLE_DEFAULT_PRINCIPAL,         // Server principal name 
        RPC_C_AUTHN_LEVEL_PKT_PRIVACY,  // 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))
    {
        cout << "Could not set proxy blanket on enumerator. Error code = 0x" << hex << hres << endl;
        pEnumerator->Release();
        pSvc->Release();
        pLoc->Release();     
        CoUninitialize();
        return 1;               // Program has failed.
    }

    // Step 9: -------------------------------------------------
    // Get the data from the query in step 7 -------------------
	
    IWbemClassObject *pclsObj = NULL;
    ULONG uReturn = 0;

    while (pEnumerator)
    {
        HRESULT hr = pEnumerator->Next(WBEM_INFINITE, 1, &pclsObj, &uReturn);

        if(0 == uReturn)
        {
            break;
        }

        VARIANT vtProp1;
		VARIANT vtProp2;
		VARIANT vtProp3;
		VARIANT vtProp4;
		VARIANT vtProp5;
		char ProcessId;
		char ProcessStr;
		BSTR MethodName = SysAllocString(L"Create");
		BSTR ClassName = SysAllocString(L"Win32_Process");

		hr = pclsObj->Get(L"Name", 0, &vtProp1, 0, 0);
		wcout << " Name " << vtProp1.bstrVal << endl;
		VariantClear(&vtProp1);

		hr = pclsObj->Get(L"ProcessId", 0, &vtProp2, 0, 0);
		wcout << " ProcessId " << vtProp2.uintVal << endl;
		//ProcessId.Format( "%u", vtProp2.bstrVal);
		VariantClear(&vtProp2);

		hr = pclsObj->Get(L"Caption", 0, &vtProp3, 0, 0);
		wcout << " Caption " << vtProp3.bstrVal << endl;
		VariantClear(&vtProp3);

		hr = pclsObj->Get(L"WorkingSetSize", 0, &vtProp4, 0, 0);
		wcout << " WorkingSetSize " << vtProp4.uintVal << endl;
		//long memUsage = _wtol(V_BSTR(&vtProp4));	
		//wcout << ", MemUsage: " << (memUsage/1024) << " K" << endl; 
		//wcout << " WorkingSetSize " << (_wtoi64(vtProp4.bstrVal)/1024)/1024 << endl;
		/*VariantClear(&vtProp4);

		BSTR MethodName = SysAllocString(L"GetOwner");
		BSTR ClassName = SysAllocString(L"Win32_Process");
		
		IWbemClassObject* pInParamsDefinition = NULL;
		IWbemClassObject* pOutParamsDefinition = NULL;
		hr = pclsObj->GetMethod(MethodName, 0, &pInParamsDefinition, &pOutParamsDefinition);

		IWbemClassObject* pClassInstance = NULL;
		//hr = pInParamsDefinition->SpawnInstance(0, &pClassInstance);

		// Execute Method
		IWbemClassObject* pOutParams = NULL;
		hr = pSvc->ExecMethod(ClassName, MethodName, 0, NULL, NULL, &pOutParams, NULL);


		wcout << " VirtualSize " << pOutParams << endl;*/
		//VariantClear(&vtProp5);
		 
		
		IWbemClassObject* pClass;
		IWbemClassObject* pwcrGetOwnerIn = NULL;  
		IWbemClassObject* pwcrGetOwnerOut = NULL; 
		IWbemClassObject* pOutParams=0;
		hr = pSvc->GetObject(_bstr_t("Win32_Process"), 0, NULL, &pClass, NULL);            // - Get Service Object 
		hr = pClass->GetMethod(_bstr_t("GetOwner"), 0, &pwcrGetOwnerIn, &pwcrGetOwnerOut);    // - Get Service Method

		//ProcessStr.Format( "Win32_Process.Handle=%s", ProcessId); // - Prepare Method Parameters 
		//WMIHandle = service->ExecMethod(_bstr_t(ProcessStr), _bstr_t(L"GetOwner"), 0, NULL, pwcrGetOwnerIn, &pOutParams, NULL);

		WCHAR path[] = L"Win32_Process.Handle=1234"; 
		hr = pClass->ExecMethod(_bstr_t(path), _bstr_t(L"GetOwner"), 0, NULL, NULL, &pOutParams, NULL);
		//_variant_t user;
		//hr = pOutParams->Get( L"User", 0, &user, NULL, 0 );
		//_variant_t domain;
		//hr = pOutParams->Get( L"Domain", 0, &domain, NULL, 0 );
		
        pclsObj->Release();
        pclsObj = NULL;
    }

    // Cleanup
    pEnumerator->Release();

    if( pclsObj )
    {
        pclsObj->Release();
    }

	Sleep(100000);

    return 0;   // Program successfully completed.
}