mDNSWindows/VPCDetect.cpp

   1 /* -*- Mode: C; tab-width: 4 -*-
   2  *
   3  * Copyright (c) 2002-2004 Apple Computer, Inc. All rights reserved.
   4  *
   5  * Licensed under the Apache License, Version 2.0 (the "License");
   6  * you may not use this file except in compliance with the License.
   7  * You may obtain a copy of the License at
   8  *
   9  *     http://www.apache.org/licenses/LICENSE-2.0
  10  *
  11  * Unless required by applicable law or agreed to in writing, software
  12  * distributed under the License is distributed on an "AS IS" BASIS,
  13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14  * See the License for the specific language governing permissions and
  15  * limitations under the License.
  16  */
  17
  18 #define _WIN32_DCOM
  19 #include "VPCDetect.h"
  20 #include "DebugServices.h"
  21 #include <comdef.h>
  22 #include <Wbemidl.h>
  23
  24 # pragma comment(lib, "wbemuuid.lib")
  25
  26 static BOOL g_doneCheck = FALSE;
  27 static BOOL g_isVPC             = FALSE;
  28
  29
  30 mStatus
  31 IsVPCRunning( BOOL * inVirtualPC )
  32 {
  33         IWbemLocator                    *       pLoc            = 0;
  34         IWbemServices                   *       pSvc            = 0;
  35     IEnumWbemClassObject        *       pEnumerator = NULL;
  36         bool                                            coInit          = false;
  37         HRESULT                                         hres;
  38         SC_HANDLE                                       scm                     = NULL;
  39         SC_HANDLE                                       service         = NULL;
  40         SERVICE_STATUS                          status;
  41         mStatus                                         err;
  42         BOOL                                            ok          = TRUE;
  43
  44         // Initialize flag
  45
  46         *inVirtualPC = FALSE;
  47
  48         // Find out if WMI is running
  49
  50         scm = OpenSCManager( 0, 0, SC_MANAGER_CONNECT );
  51         err = translate_errno( scm, (OSStatus) GetLastError(), kOpenErr );
  52         require_noerr( err, exit );
  53
  54         service = OpenService( scm, TEXT( "winmgmt" ), SERVICE_QUERY_STATUS );
  55         err = translate_errno( service, (OSStatus) GetLastError(), kNotFoundErr );
  56         require_noerr( err, exit );
  57
  58         ok = QueryServiceStatus( service, &status );
  59         err = translate_errno( ok, (OSStatus) GetLastError(), kAuthenticationErr );
  60         require_noerr( err, exit );
  61         require_action( status.dwCurrentState == SERVICE_RUNNING, exit, err = kUnknownErr );
  62
  63     // Initialize COM.
  64
  65         hres = CoInitializeEx(0, COINIT_MULTITHREADED);
  66         require_action( SUCCEEDED( hres ), exit, err = kUnknownErr );
  67         coInit = true;
  68
  69         // Initialize Security
  70
  71         hres =  CoInitializeSecurity(NULL, -1, NULL, NULL, RPC_C_AUTHN_LEVEL_DEFAULT, RPC_C_IMP_LEVEL_IMPERSONATE, NULL, EOAC_NONE, NULL );
  72         require_action( SUCCEEDED( hres ), exit, err = kUnknownErr );
  73
  74
  75     // Obtain the initial locator to Windows Management on a particular host computer.
  76
  77     hres = CoCreateInstance( CLSID_WbemLocator, 0, CLSCTX_INPROC_SERVER, IID_IWbemLocator, (LPVOID *) &pLoc );
  78         require_action( SUCCEEDED( hres ), exit, err = kUnknownErr );
  79
  80     // Connect to the root\cimv2 namespace with the
  81     // current user and obtain pointer pSvc
  82     // to make IWbemServices calls.
  83
  84         hres = pLoc->ConnectServer( _bstr_t(L"ROOT\\CIMV2"), NULL, NULL, 0, WBEM_FLAG_CONNECT_USE_MAX_WAIT, 0, 0, &pSvc );
  85         require_action( SUCCEEDED( hres ), exit, err = kUnknownErr );
  86
  87     // Set the IWbemServices proxy so that impersonation
  88     // of the user (client) occurs.
  89
  90         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 );
  91         require_action( SUCCEEDED( hres ), exit, err = kUnknownErr );
  92
  93     // Use the IWbemServices pointer to make requests of WMI.
  94     // Make requests here:
  95
  96         hres = pSvc->ExecQuery( bstr_t("WQL"), bstr_t("SELECT * from Win32_BaseBoard"), WBEM_FLAG_FORWARD_ONLY | WBEM_FLAG_RETURN_IMMEDIATELY, NULL, &pEnumerator);
  97
  98         require_action( SUCCEEDED( hres ), exit, err = kUnknownErr );
  99
 100         do
 101         {
 102                 IWbemClassObject* pInstance = NULL;
 103                 ULONG dwCount = NULL;
 104
 105                 hres = pEnumerator->Next( WBEM_INFINITE, 1, &pInstance, &dwCount);
 106
 107                 if ( pInstance )
 108                 {
 109                         VARIANT v;
 110                         BSTR strClassProp = SysAllocString(L"Manufacturer");
 111                         HRESULT hr;
 112
 113                         hr = pInstance->Get(strClassProp, 0, &v, 0, 0);
 114                         SysFreeString(strClassProp);
 115
 116                         // check the HRESULT to see if the action succeeded.
 117
 118                         if (SUCCEEDED(hr) && (V_VT(&v) == VT_BSTR))
 119                         {
 120                                 wchar_t * wstring = wcslwr( V_BSTR( &v ) );
 121
 122                                 if (wcscmp( wstring, L"microsoft corporation" ) == 0 )
 123                                 {
 124                                         *inVirtualPC = TRUE;
 125                                 }
 126                         }
 127
 128                         VariantClear(&v);
 129                 }
 130         } while (hres == WBEM_S_NO_ERROR);
 131
 132 exit:
 133
 134         if ( pSvc != NULL )
 135         {
 136         pSvc->Release();
 137         }
 138
 139         if ( pLoc != NULL )
 140         {
 141         pLoc->Release();
 142         }
 143
 144         if ( coInit )
 145         {
 146         CoUninitialize();
 147         }
 148
 149         if ( service )
 150         {
 151                 CloseServiceHandle( service );
 152         }
 153
 154         if ( scm )
 155         {
 156                 CloseServiceHandle( scm );
 157         }
 158
 159         if ( *inVirtualPC )
 160         {
 161                 dlog( kDebugLevelTrace, "Virtual PC detected" );
 162         }
 163
 164         return err;
 165 }