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 } 166
