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
IsVPCRunning(BOOL * inVirtualPC)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