code.vuplus.com Git - vuplus_xbmc/blob - xbmc/win32/WMIInterface.cpp

   1 /*
   2  *      Copyright (C) 2005-2013 Team XBMC
   3  *      http://www.xbmc.org
   4  *
   5  *  This Program is free software; you can redistribute it and/or modify
   6  *  it under the terms of the GNU General Public License as published by
   7  *  the Free Software Foundation; either version 2, or (at your option)
   8  *  any later version.
   9  *
  10  *  This Program is distributed in the hope that it will be useful,
  11  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  13  *  GNU General Public License for more details.
  14  *
  15  *  You should have received a copy of the GNU General Public License
  16  *  along with XBMC; see the file COPYING.  If not, see
  17  *  <http://www.gnu.org/licenses/>.
  18  *
  19  */
  20
  21 #include "stdafx.h"
  22 #include "WMIInterface.h"
  23 #include "../Util.h"
  24
  25 using namespace std;
  26
  27
  28
  29 CWIN32Wmi::CWIN32Wmi(void)
  30 {
  31   bconnected = false;
  32   pclsObj = NULL;
  33   Connect();
  34 }
  35
  36 CWIN32Wmi::~CWIN32Wmi(void)
  37 {
  38   Release();
  39 }
  40
  41 bool CWIN32Wmi::Connect()
  42 {
  43   // Initialize COM. ------------------------------------------
  44
  45   hres =  CoInitializeEx(0, COINIT_MULTITHREADED);
  46   if (FAILED(hres))
  47   {
  48       return false;                  // Program has failed.
  49   }
  50
  51         hres =  CoInitializeSecurity(
  52         NULL,
  53         -1,                          // COM authentication
  54         NULL,                        // Authentication services
  55         NULL,                        // Reserved
  56         RPC_C_AUTHN_LEVEL_DEFAULT,   // Default authentication
  57         RPC_C_IMP_LEVEL_IMPERSONATE, // Default Impersonation
  58         NULL,                        // Authentication info
  59         EOAC_NONE,                   // Additional capabilities
  60         NULL                         // Reserved
  61         );
  62
  63         if (FAILED(hres))
  64   {
  65       return false;                    // Program has failed.
  66   }
  67
  68         pLoc = NULL;
  69
  70   hres = CoCreateInstance(
  71       CLSID_WbemLocator,
  72       0,
  73       CLSCTX_INPROC_SERVER,
  74       IID_IWbemLocator, (LPVOID *) &pLoc);
  75
  76   if (FAILED(hres))
  77   {
  78       return false;                 // Program has failed.
  79   }
  80
  81         pSvc = NULL;
  82
  83   // Connect to the root\cimv2 namespace with
  84   // the current user and obtain pointer pSvc
  85   // to make IWbemServices calls.
  86   hres = pLoc->ConnectServer(
  87        _bstr_t(L"ROOT\\CIMV2"), // Object path of WMI namespace
  88        NULL,                    // User name. NULL = current user
  89        NULL,                    // User password. NULL = current
  90        0,                       // Locale. NULL indicates current
  91        NULL,                    // Security flags.
  92        0,                       // Authority (e.g. Kerberos)
  93        0,                       // Context object
  94        &pSvc                    // pointer to IWbemServices proxy
  95        );
  96
  97   if (FAILED(hres))
  98   {
  99       pLoc->Release();
 100       CoUninitialize();
 101       return false;                // Program has failed.
 102   }
 103
 104         hres = CoSetProxyBlanket(
 105        pSvc,                        // Indicates the proxy to set
 106        RPC_C_AUTHN_WINNT,           // RPC_C_AUTHN_xxx
 107        RPC_C_AUTHZ_NONE,            // RPC_C_AUTHZ_xxx
 108        NULL,                        // Server principal name
 109        RPC_C_AUTHN_LEVEL_CALL,      // RPC_C_AUTHN_LEVEL_xxx
 110        RPC_C_IMP_LEVEL_IMPERSONATE, // RPC_C_IMP_LEVEL_xxx
 111        NULL,                        // client identity
 112        EOAC_NONE                    // proxy capabilities
 113     );
 114
 115     if (FAILED(hres))
 116     {
 117         pSvc->Release();
 118         pLoc->Release();
 119         CoUninitialize();
 120         return false;               // Program has failed.
 121     }
 122
 123                 pEnumerator = NULL;
 124
 125     bconnected = true;
 126     return true;
 127 }
 128
 129 void CWIN32Wmi::Release()
 130 {
 131   if(pSvc != NULL)
 132     pSvc->Release();
 133   if(pLoc != NULL)
 134     pLoc->Release();
 135   if(pEnumerator != NULL)
 136     pEnumerator->Release();
 137   if(pclsObj != NULL)
 138     pclsObj->Release();
 139
 140   CoUninitialize();
 141
 142   bconnected = false;
 143   pSvc = NULL;
 144   pLoc = NULL;
 145   pEnumerator = NULL;
 146   pclsObj = NULL;
 147 }
 148
 149 void CWIN32Wmi::testquery()
 150 {
 151   hres = pSvc->ExecQuery(
 152       bstr_t("WQL"),
 153       //bstr_t("SELECT * FROM Win32_NetworkAdapterConfiguration WHERE IPEnabled=TRUE"),
 154       //bstr_t("SELECT * FROM Win32_NetworkAdapter WHERE PhysicalAdapter=TRUE"),
 155       bstr_t("SELECT * FROM Win32_NetworkAdapter WHERE Description='Atheros AR5008X Wireless Network Adapter'"),
 156       WBEM_FLAG_FORWARD_ONLY | WBEM_FLAG_RETURN_IMMEDIATELY,
 157       NULL,
 158       &pEnumerator);
 159
 160   if (FAILED(hres))
 161   {
 162       return;               // Program has failed.
 163   }
 164   ULONG uReturn = 0;
 165
 166   while (pEnumerator)
 167   {
 168       HRESULT hr = pEnumerator->Next(WBEM_INFINITE, 1,
 169           &pclsObj, &uReturn);
 170
 171       if(0 == uReturn)
 172       {
 173           break;
 174       }
 175
 176       VARIANT vtProp;
 177       VariantInit(&vtProp);
 178
 179       // Get the value of the Name property
 180       //hr = pclsObj->Get(L"Description", 0, &vtProp, 0, 0);
 181
 182       vtProp.bstrVal = bstr_t("192.168.1.209");
 183       hr = pclsObj->Put(L"IPAddress",0,&vtProp,0);
 184       VariantClear(&vtProp);
 185                         //iCpu++;
 186   }
 187         pclsObj->Release();
 188   pclsObj = NULL;
 189 }
 190
 191 std::vector<CStdString> CWIN32Wmi::GetWMIStrVector(CStdString& strQuery, CStdStringW& strProperty)
 192 {
 193   std::vector<CStdString> strResult;
 194   pEnumerator = NULL;
 195   pclsObj = NULL;
 196
 197   if(!bconnected)
 198     return strResult;
 199
 200   hres = pSvc->ExecQuery(
 201       bstr_t("WQL"),
 202       bstr_t(strQuery.c_str()),
 203       WBEM_FLAG_FORWARD_ONLY | WBEM_FLAG_RETURN_IMMEDIATELY,
 204       NULL,
 205       &pEnumerator);
 206
 207   if (FAILED(hres))
 208   {
 209       return strResult;               // Program has failed.
 210   }
 211   ULONG uReturn = 0;
 212
 213   while (pEnumerator)
 214   {
 215       HRESULT hr = pEnumerator->Next(WBEM_INFINITE, 1,
 216           &pclsObj, &uReturn);
 217
 218       if(0 == uReturn)
 219       {
 220           break;
 221       }
 222
 223       VARIANT vtProp;
 224       VariantInit(&vtProp);
 225
 226       hr = pclsObj->Get(strProperty.c_str(), 0, &vtProp, 0, 0);
 227       strResult.push_back(vtProp.bstrVal);
 228       VariantClear(&vtProp);
 229   }
 230   if(pEnumerator != NULL)
 231     pEnumerator->Release();
 232   pEnumerator = NULL;
 233   if(pclsObj != NULL)
 234           pclsObj->Release();
 235   pclsObj = NULL;
 236   return strResult;
 237 }
 238
 239 CStdString CWIN32Wmi::GetWMIString(CStdString& strQuery, CStdStringW& strProperty)
 240 {
 241   return GetWMIStrVector(strQuery, strProperty)[0];
 242 }