mDNSResponder-320.10.80.tar.gz

[apple/mdnsresponder.git] / mDNSWindows / VPCDetect.cpp

diff --git a/mDNSWindows/VPCDetect.cpp b/mDNSWindows/VPCDetect.cpp
index a5723063e5ff21d70bcf6b1445b6f671390af896..3df7c141ccadbabc2666f458d5356ca196811f81 100755 (executable)
--- a/mDNSWindows/VPCDetect.cpp
+++ b/mDNSWindows/VPCDetect.cpp
@@ -1,32 +1,19 @@
-/*
+/* -*- Mode: C; tab-width: 4 -*-
+ *

  * Copyright (c) 2002-2004 Apple Computer, Inc. All rights reserved.
  *
  * 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.
  * limitations under the License.

- * 
- * @APPLE_LICENSE_HEADER_END@
-
-    Change History (most recent first):
-    
-$Log: VPCDetect.cpp,v $
-Revision 1.1  2005/11/27 20:21:16  herscher
-<rdar://problem/4210580> Workaround Virtual PC bug that incorrectly modifies incoming mDNS packets
-
-*/
+ */

 
 #define _WIN32_DCOM
 #include "VPCDetect.h"
 
 #define _WIN32_DCOM
 #include "VPCDetect.h"


@@ -40,55 +27,75 @@ static BOOL g_doneCheck = FALSE;
 static BOOL g_isVPC            = 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;
 {
        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);
     // 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 );
        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 );
 
                       
     // 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.
 
  
     // 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 );
     
     // 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);
     
 
     // 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
        {
 
        do
        {


@@ -114,7 +121,7 @@ IsVPCRunning()
 
                                if (wcscmp( wstring, L"microsoft corporation" ) == 0 )
                                {
 
                                if (wcscmp( wstring, L"microsoft corporation" ) == 0 )
                                {

-                                       g_isVPC = true;
+                                       *inVirtualPC = TRUE;

                                }
                        }
                
                                }
                        }
                


@@ -139,15 +146,20 @@ exit:
        CoUninitialize();
        }
 
        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;

 }
 }