]> git.saurik.com Git - apple/mdnsresponder.git/blobdiff - Clients/ExplorerPlugin/ExplorerPlugin.cpp
mDNSResponder-171.4.tar.gz
[apple/mdnsresponder.git] / Clients / ExplorerPlugin / ExplorerPlugin.cpp
index 464051abba8c31fc53b5f0c9123297b9aaee3921..c2c661b0a849f8ec8db36c68713d8858430d5eed 100644 (file)
@@ -1,28 +1,31 @@
-/*
+/* -*- Mode: C; tab-width: 4 -*-
+ *
  * Copyright (c) 2003-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: ExplorerPlugin.cpp,v $
+Revision 1.9  2006/08/14 23:24:00  cheshire
+Re-licensed mDNSResponder daemon source code under Apache License, Version 2.0
+
+Revision 1.8  2005/06/30 18:01:54  shersche
+<rdar://problem/4130635> Cause IE to rebuild cache so we don't have to reboot following an install.
+
+Revision 1.7  2005/02/23 02:00:45  shersche
+<rdar://problem/4014479> Delete all the registry entries when component is unregistered
+
 Revision 1.6  2005/01/25 17:56:45  shersche
 <rdar://problem/3911084> Load resource DLLs, get icons and bitmaps from resource DLLs
 Bug #: 3911084
@@ -101,6 +104,8 @@ DEBUG_LOCAL void            MFCDLLThreadDetach( HINSTANCE inInstance );
 
 DEBUG_LOCAL OSStatus   RegisterServer( HINSTANCE inInstance, CLSID inCLSID, LPCTSTR inName );
 DEBUG_LOCAL OSStatus   RegisterCOMCategory( CLSID inCLSID, CATID inCategoryID, BOOL inRegister );
+DEBUG_LOCAL OSStatus   UnregisterServer( CLSID inCLSID );
+DEBUG_LOCAL OSStatus   MyRegDeleteKey( HKEY hKeyRoot, LPTSTR lpSubKey );
 
 // Stash away pointers to our resource DLLs
 
@@ -120,6 +125,14 @@ GetLocalizedResources()
        return g_localizedResources;
 }
 
+// This is the class GUID for an undocumented hook into IE that will allow us to register
+// and have IE notice our new ExplorerBar without rebooting.
+// {8C7461EF-2B13-11d2-BE35-3078302C2030}
+
+DEFINE_GUID(CLSID_CompCatCacheDaemon, 
+0x8C7461EF, 0x2b13, 0x11d2, 0xbe, 0x35, 0x30, 0x78, 0x30, 0x2c, 0x20, 0x30);
+
+
 #if 0
 #pragma mark == Globals ==
 #endif
@@ -243,9 +256,10 @@ exit:
 
 STDAPI DllRegisterServer( void )
 {
-       HRESULT         err;
-       BOOL            ok;
-       CString         s;
+       IRunnableTask * pTask = NULL;
+       HRESULT                 err;
+       BOOL                    ok;
+       CString                 s;
        
        dlog( kDebugLevelTrace, "DllRegisterServer\n" );
        
@@ -257,7 +271,16 @@ STDAPI     DllRegisterServer( void )
        
        err = RegisterCOMCategory( CLSID_ExplorerBar, CATID_InfoBand, TRUE );
        require_noerr( err, exit );
-       
+
+       // <rdar://problem/4130635> Clear IE cache so it will rebuild the cache when it runs next.  This
+       // will allow us to install and not reboot
+
+       err = CoCreateInstance(CLSID_CompCatCacheDaemon, NULL, CLSCTX_INPROC, IID_IRunnableTask, (void**) &pTask);
+       require_noerr( err, exit );
+
+       pTask->Run();
+       pTask->Release();
+
 exit:
        return( err );
 }
@@ -274,6 +297,9 @@ STDAPI      DllUnregisterServer( void )
        
        err = RegisterCOMCategory( CLSID_ExplorerBar, CATID_InfoBand, FALSE );
        require_noerr( err, exit );
+
+       err = UnregisterServer( CLSID_ExplorerBar );
+       require_noerr( err, exit );
        
 exit:
        return( err );
@@ -590,3 +616,138 @@ DEBUG_LOCAL OSStatus      RegisterCOMCategory( CLSID inCLSID, CATID inCategoryID, BOO
 exit:
        return( err );
 }
+
+
+//===========================================================================================================================
+//     UnregisterServer
+//===========================================================================================================================
+
+DEBUG_LOCAL OSStatus   UnregisterServer( CLSID inCLSID )
+{
+       OSStatus                        err = 0;
+       LPWSTR                          clsidWideString;
+       TCHAR                           clsidString[ 64 ];
+       HKEY                            key;
+       TCHAR                           keyName[ MAX_PATH * 2 ];
+       OSVERSIONINFO           versionInfo;
+
+       // Convert the CLSID to a string based on the encoding of this code (ANSI or Unicode).
+       
+       err = StringFromIID( inCLSID, &clsidWideString );
+       require_noerr( err, exit );
+       require_action( clsidWideString, exit, err = kNoMemoryErr );
+       
+       #ifdef UNICODE
+               lstrcpyn( clsidString, clsidWideString, sizeof_array( clsidString ) );
+               CoTaskMemFree( clsidWideString );
+       #else
+               nChars = WideCharToMultiByte( CP_ACP, 0, clsidWideString, -1, clsidString, sizeof_array( clsidString ), NULL, NULL );
+               err = translate_errno( nChars > 0, (OSStatus) GetLastError(), kUnknownErr );
+               CoTaskMemFree( clsidWideString );
+               require_noerr( err, exit );
+       #endif
+
+       wsprintf( keyName, L"CLSID\\%s", clsidString );
+       MyRegDeleteKey( HKEY_CLASSES_ROOT, keyName );
+       
+       // If running on NT, de-register the extension as approved.
+       
+       versionInfo.dwOSVersionInfoSize = sizeof( versionInfo );
+       GetVersionEx( &versionInfo );
+       if( versionInfo.dwPlatformId == VER_PLATFORM_WIN32_NT )
+       {
+               lstrcpyn( keyName, TEXT( "SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Shell Extensions\\Approved" ), sizeof_array( keyName ) );
+               err = RegCreateKeyEx( HKEY_LOCAL_MACHINE, keyName, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &key, NULL );
+               require_noerr( err, exit );
+
+               RegDeleteValue( key, clsidString );
+
+               err = RegCloseKey( key );
+               require_noerr( err, exit );
+       }
+
+       // de-register toolbar button
+
+       lstrcpyn( keyName, TEXT( "SOFTWARE\\Microsoft\\Internet Explorer\\Extensions\\{7F9DB11C-E358-4ca6-A83D-ACC663939424}"), sizeof_array( keyName ) );
+       MyRegDeleteKey( HKEY_LOCAL_MACHINE, keyName );
+       
+exit:
+       return( err );
+}
+
+
+
+//===========================================================================================================================
+//     MyRegDeleteKey
+//===========================================================================================================================
+
+DEBUG_LOCAL OSStatus MyRegDeleteKey( HKEY hKeyRoot, LPTSTR lpSubKey )
+{
+    LPTSTR lpEnd;
+    OSStatus err;
+    DWORD dwSize;
+    TCHAR szName[MAX_PATH];
+    HKEY hKey;
+    FILETIME ftWrite;
+
+    // First, see if we can delete the key without having to recurse.
+
+    err = RegDeleteKey( hKeyRoot, lpSubKey );
+
+    if ( !err )
+       {
+               goto exit;
+       }
+
+    err = RegOpenKeyEx( hKeyRoot, lpSubKey, 0, KEY_READ, &hKey );
+       require_noerr( err, exit );
+
+    // Check for an ending slash and add one if it is missing.
+
+    lpEnd = lpSubKey + lstrlen(lpSubKey);
+
+    if ( *( lpEnd - 1 ) != TEXT( '\\' ) ) 
+    {
+        *lpEnd =  TEXT('\\');
+        lpEnd++;
+        *lpEnd =  TEXT('\0');
+    }
+
+    // Enumerate the keys
+
+    dwSize = MAX_PATH;
+    err = RegEnumKeyEx(hKey, 0, szName, &dwSize, NULL, NULL, NULL, &ftWrite);
+
+    if ( !err ) 
+    {
+        do
+               {
+            lstrcpy (lpEnd, szName);
+
+            if ( !MyRegDeleteKey( hKeyRoot, lpSubKey ) )
+                       {
+                break;
+            }
+
+            dwSize = MAX_PATH;
+
+            err = RegEnumKeyEx( hKey, 0, szName, &dwSize, NULL, NULL, NULL, &ftWrite );
+
+        }
+               while ( !err );
+    }
+
+    lpEnd--;
+    *lpEnd = TEXT('\0');
+
+    RegCloseKey( hKey );
+
+    // Try again to delete the key.
+
+    err = RegDeleteKey(hKeyRoot, lpSubKey);
+       require_noerr( err, exit );
+
+exit:
+
+       return err;
+}