X-Git-Url: https://git.saurik.com/apple/mdnsresponder.git/blobdiff_plain/051c1f9c2d4470907e1dc28b63d18f2bcfba1bb6..67c8f8a10700c05d2460d60f5927f23cb5cb9241:/mDNSWindows/VPCDetect.cpp diff --git a/mDNSWindows/VPCDetect.cpp b/mDNSWindows/VPCDetect.cpp index a572306..c0f4d01 100755 --- a/mDNSWindows/VPCDetect.cpp +++ b/mDNSWindows/VPCDetect.cpp @@ -1,28 +1,28 @@ -/* +/* -*- Mode: C; tab-width: 4 -*- + * * Copyright (c) 2002-2004 Apple Computer, Inc. All rights reserved. * - * @APPLE_LICENSE_HEADER_START@ + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at * - * This file contains Original Code and/or Modifications of Original Code - * as defined in and that are subject to the Apple Public Source License - * Version 2.0 (the 'License'). You may not use this file except in - * compliance with the License. Please obtain a copy of the License at - * http://www.opensource.apple.com/apsl/ and read it before using this - * file. + * http://www.apache.org/licenses/LICENSE-2.0 * - * The Original Code and all software distributed under the License are - * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER - * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES, - * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY, - * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT. - * Please see the License for the specific language governing rights and + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and * limitations under the License. - * - * @APPLE_LICENSE_HEADER_END@ Change History (most recent first): $Log: VPCDetect.cpp,v $ +Revision 1.3 2006/08/14 23:25:20 cheshire +Re-licensed mDNSResponder daemon source code under Apache License, Version 2.0 + +Revision 1.2 2006/02/26 19:31:05 herscher + Bonjour For Windows takes 90 seconds to start. This was caused by a bad interaction between the VirtualPC check, and the removal of the WMI dependency. The problem was fixed by: 1) checking to see if WMI is running before trying to talk to it. 2) Retrying the VirtualPC check every 10 seconds upon failure, stopping after 10 unsuccessful tries. + Revision 1.1 2005/11/27 20:21:16 herscher Workaround Virtual PC bug that incorrectly modifies incoming mDNS packets @@ -40,55 +40,75 @@ static BOOL g_doneCheck = FALSE; static BOOL g_isVPC = FALSE; -BOOL -IsVPCRunning() +mStatus +IsVPCRunning( BOOL * inVirtualPC ) { IWbemLocator * pLoc = 0; IWbemServices * pSvc = 0; IEnumWbemClassObject * pEnumerator = NULL; bool coInit = false; HRESULT hres; - - // Short circuit if we've already done this - - require_action_quiet( !g_doneCheck, exit, g_doneCheck = TRUE ); - + SC_HANDLE scm = NULL; + SC_HANDLE service = NULL; + SERVICE_STATUS status; + mStatus err; + BOOL ok = TRUE; + + // Initialize flag + + *inVirtualPC = FALSE; + + // Find out if WMI is running + + scm = OpenSCManager( 0, 0, SC_MANAGER_CONNECT ); + err = translate_errno( scm, (OSStatus) GetLastError(), kOpenErr ); + require_noerr( err, exit ); + + service = OpenService( scm, TEXT( "winmgmt" ), SERVICE_QUERY_STATUS ); + err = translate_errno( service, (OSStatus) GetLastError(), kNotFoundErr ); + require_noerr( err, exit ); + + ok = QueryServiceStatus( service, &status ); + err = translate_errno( ok, (OSStatus) GetLastError(), kAuthenticationErr ); + require_noerr( err, exit ); + require_action( status.dwCurrentState == SERVICE_RUNNING, exit, err = kUnknownErr ); + // Initialize COM. hres = CoInitializeEx(0, COINIT_MULTITHREADED); - require_action( SUCCEEDED( hres ), exit, g_isVPC = false ); + require_action( SUCCEEDED( hres ), exit, err = kUnknownErr ); coInit = true; // Initialize Security hres = CoInitializeSecurity(NULL, -1, NULL, NULL, RPC_C_AUTHN_LEVEL_DEFAULT, RPC_C_IMP_LEVEL_IMPERSONATE, NULL, EOAC_NONE, NULL ); - require_action( SUCCEEDED( hres ), exit, g_isVPC = false ); + require_action( SUCCEEDED( hres ), exit, err = kUnknownErr ); // Obtain the initial locator to Windows Management on a particular host computer. hres = CoCreateInstance( CLSID_WbemLocator, 0, CLSCTX_INPROC_SERVER, IID_IWbemLocator, (LPVOID *) &pLoc ); - require_action( SUCCEEDED( hres ), exit, g_isVPC = false ); + require_action( SUCCEEDED( hres ), exit, err = kUnknownErr ); // Connect to the root\cimv2 namespace with the // current user and obtain pointer pSvc // to make IWbemServices calls. - hres = pLoc->ConnectServer( _bstr_t(L"ROOT\\CIMV2"), NULL, NULL, 0, NULL, 0, 0, &pSvc ); - require_action( SUCCEEDED( hres ), exit, g_isVPC = false ); + hres = pLoc->ConnectServer( _bstr_t(L"ROOT\\CIMV2"), NULL, NULL, 0, WBEM_FLAG_CONNECT_USE_MAX_WAIT, 0, 0, &pSvc ); + require_action( SUCCEEDED( hres ), exit, err = kUnknownErr ); // Set the IWbemServices proxy so that impersonation // of the user (client) occurs. 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 ); - require_action( SUCCEEDED( hres ), exit, g_isVPC = false ); + require_action( SUCCEEDED( hres ), exit, err = kUnknownErr ); // Use the IWbemServices pointer to make requests of WMI. // Make requests here: hres = pSvc->ExecQuery( bstr_t("WQL"), bstr_t("SELECT * from Win32_BaseBoard"), WBEM_FLAG_FORWARD_ONLY | WBEM_FLAG_RETURN_IMMEDIATELY, NULL, &pEnumerator); - require_action( SUCCEEDED( hres ), exit, g_isVPC = false ); + require_action( SUCCEEDED( hres ), exit, err = kUnknownErr ); do { @@ -114,7 +134,7 @@ IsVPCRunning() if (wcscmp( wstring, L"microsoft corporation" ) == 0 ) { - g_isVPC = true; + *inVirtualPC = TRUE; } } @@ -139,15 +159,20 @@ exit: CoUninitialize(); } - if ( !g_doneCheck ) + if ( service ) { - g_doneCheck = TRUE; + CloseServiceHandle( service ); + } - if ( g_isVPC ) - { - dlog( kDebugLevelTrace, "Virtual PC detected" ); - } + if ( scm ) + { + CloseServiceHandle( scm ); + } + + if ( *inVirtualPC ) + { + dlog( kDebugLevelTrace, "Virtual PC detected" ); } - return g_isVPC; + return err; }