mirror of
https://git.rtems.org/rtems-libbsd/
synced 2025-06-14 12:49:48 +08:00

The sources can be obtained via: http://opensource.apple.com/tarballs/mDNSResponder/mDNSResponder-544.tar.gz
601 lines
18 KiB
C++
601 lines
18 KiB
C++
/* -*- Mode: C; tab-width: 4 -*-
|
|
*
|
|
* Copyright (c) 2003-2004 Apple Computer, Inc. All rights reserved.
|
|
*
|
|
* 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
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* 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.
|
|
*/
|
|
|
|
#include "StdAfx.h"
|
|
|
|
// The following 2 includes have to be in this order and INITGUID must be defined here, before including the file
|
|
// that specifies the GUID(s), and nowhere else. The reason for this is that initguid.h doesn't provide separate
|
|
// define and declare macros for GUIDs so you have to #define INITGUID in the single file where you want to define
|
|
// your GUID then in all the other files that just need the GUID declared, INITGUID must not be defined.
|
|
|
|
#define INITGUID
|
|
#include <initguid.h>
|
|
#include "ExplorerPlugin.h"
|
|
|
|
#include <comcat.h>
|
|
#include <Shlwapi.h>
|
|
|
|
#include "CommonServices.h"
|
|
#include "DebugServices.h"
|
|
|
|
#include "ClassFactory.h"
|
|
#include "Resource.h"
|
|
|
|
#include "loclibrary.h"
|
|
|
|
// MFC Debugging
|
|
|
|
#ifdef _DEBUG
|
|
#define new DEBUG_NEW
|
|
#undef THIS_FILE
|
|
static char THIS_FILE[] = __FILE__;
|
|
#endif
|
|
|
|
#if 0
|
|
#pragma mark == Prototypes ==
|
|
#endif
|
|
|
|
//===========================================================================================================================
|
|
// Prototypes
|
|
//===========================================================================================================================
|
|
|
|
// Utilities
|
|
|
|
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
|
|
|
|
static HINSTANCE g_nonLocalizedResources = NULL;
|
|
static CString g_nonLocalizedResourcesName;
|
|
static HINSTANCE g_localizedResources = NULL;
|
|
|
|
HINSTANCE
|
|
GetNonLocalizedResources()
|
|
{
|
|
return g_nonLocalizedResources;
|
|
}
|
|
|
|
HINSTANCE
|
|
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
|
|
|
|
//===========================================================================================================================
|
|
// Globals
|
|
//===========================================================================================================================
|
|
|
|
HINSTANCE gInstance = NULL;
|
|
int gDLLRefCount = 0;
|
|
CExplorerPluginApp gApp;
|
|
|
|
#if 0
|
|
#pragma mark -
|
|
#pragma mark == DLL Exports ==
|
|
#endif
|
|
|
|
//===========================================================================================================================
|
|
// CExplorerPluginApp::CExplorerPluginApp
|
|
//===========================================================================================================================
|
|
|
|
IMPLEMENT_DYNAMIC(CExplorerPluginApp, CWinApp);
|
|
|
|
CExplorerPluginApp::CExplorerPluginApp()
|
|
{
|
|
}
|
|
|
|
|
|
//===========================================================================================================================
|
|
// CExplorerPluginApp::~CExplorerPluginApp
|
|
//===========================================================================================================================
|
|
|
|
CExplorerPluginApp::~CExplorerPluginApp()
|
|
{
|
|
}
|
|
|
|
|
|
//===========================================================================================================================
|
|
// CExplorerPluginApp::InitInstance
|
|
//===========================================================================================================================
|
|
|
|
BOOL
|
|
CExplorerPluginApp::InitInstance()
|
|
{
|
|
wchar_t resource[MAX_PATH];
|
|
OSStatus err;
|
|
int res;
|
|
HINSTANCE inInstance;
|
|
|
|
inInstance = AfxGetInstanceHandle();
|
|
gInstance = inInstance;
|
|
|
|
debug_initialize( kDebugOutputTypeWindowsEventLog, "DNSServices Bar", inInstance );
|
|
debug_set_property( kDebugPropertyTagPrintLevel, kDebugLevelTrace );
|
|
dlog( kDebugLevelTrace, "\nCCPApp::InitInstance\n" );
|
|
|
|
res = PathForResource( inInstance, L"ExplorerPluginResources.dll", resource, MAX_PATH );
|
|
|
|
err = translate_errno( res != 0, kUnknownErr, kUnknownErr );
|
|
require_noerr( err, exit );
|
|
|
|
g_nonLocalizedResources = LoadLibrary( resource );
|
|
translate_errno( g_nonLocalizedResources, GetLastError(), kUnknownErr );
|
|
require_noerr( err, exit );
|
|
|
|
g_nonLocalizedResourcesName = resource;
|
|
|
|
res = PathForResource( inInstance, L"ExplorerPluginLocalized.dll", resource, MAX_PATH );
|
|
err = translate_errno( res != 0, kUnknownErr, kUnknownErr );
|
|
require_noerr( err, exit );
|
|
|
|
g_localizedResources = LoadLibrary( resource );
|
|
translate_errno( g_localizedResources, GetLastError(), kUnknownErr );
|
|
require_noerr( err, exit );
|
|
|
|
AfxSetResourceHandle( g_localizedResources );
|
|
|
|
exit:
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
|
|
//===========================================================================================================================
|
|
// CExplorerPluginApp::ExitInstance
|
|
//===========================================================================================================================
|
|
|
|
int
|
|
CExplorerPluginApp::ExitInstance()
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
|
|
|
|
//===========================================================================================================================
|
|
// DllCanUnloadNow
|
|
//===========================================================================================================================
|
|
|
|
STDAPI DllCanUnloadNow( void )
|
|
{
|
|
dlog( kDebugLevelTrace, "DllCanUnloadNow (refCount=%d)\n", gDLLRefCount );
|
|
|
|
return( gDLLRefCount == 0 );
|
|
}
|
|
|
|
//===========================================================================================================================
|
|
// DllGetClassObject
|
|
//===========================================================================================================================
|
|
|
|
STDAPI DllGetClassObject( REFCLSID inCLSID, REFIID inIID, LPVOID *outResult )
|
|
{
|
|
HRESULT err;
|
|
BOOL ok;
|
|
ClassFactory * factory;
|
|
|
|
dlog( kDebugLevelTrace, "DllGetClassObject\n" );
|
|
|
|
*outResult = NULL;
|
|
|
|
// Check if the class ID is supported.
|
|
|
|
ok = IsEqualCLSID( inCLSID, CLSID_ExplorerBar );
|
|
require_action_quiet( ok, exit, err = CLASS_E_CLASSNOTAVAILABLE );
|
|
|
|
// Create the ClassFactory object.
|
|
|
|
factory = NULL;
|
|
try
|
|
{
|
|
factory = new ClassFactory( inCLSID );
|
|
}
|
|
catch( ... )
|
|
{
|
|
// Do not let exception escape.
|
|
}
|
|
require_action( factory, exit, err = E_OUTOFMEMORY );
|
|
|
|
// Query for the specified interface. Release the factory since QueryInterface retains it.
|
|
|
|
err = factory->QueryInterface( inIID, outResult );
|
|
factory->Release();
|
|
|
|
exit:
|
|
return( err );
|
|
}
|
|
|
|
//===========================================================================================================================
|
|
// DllRegisterServer
|
|
//===========================================================================================================================
|
|
|
|
STDAPI DllRegisterServer( void )
|
|
{
|
|
IRunnableTask * pTask = NULL;
|
|
HRESULT err;
|
|
BOOL ok;
|
|
CString s;
|
|
|
|
dlog( kDebugLevelTrace, "DllRegisterServer\n" );
|
|
|
|
ok = s.LoadString( IDS_NAME );
|
|
require_action( ok, exit, err = E_UNEXPECTED );
|
|
|
|
err = RegisterServer( gInstance, CLSID_ExplorerBar, s );
|
|
require_noerr( err, exit );
|
|
|
|
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 );
|
|
}
|
|
|
|
//===========================================================================================================================
|
|
// DllUnregisterServer
|
|
//===========================================================================================================================
|
|
|
|
STDAPI DllUnregisterServer( void )
|
|
{
|
|
HRESULT err;
|
|
|
|
dlog( kDebugLevelTrace, "DllUnregisterServer\n" );
|
|
|
|
err = RegisterCOMCategory( CLSID_ExplorerBar, CATID_InfoBand, FALSE );
|
|
require_noerr( err, exit );
|
|
|
|
err = UnregisterServer( CLSID_ExplorerBar );
|
|
require_noerr( err, exit );
|
|
|
|
exit:
|
|
return( err );
|
|
}
|
|
|
|
|
|
#if 0
|
|
#pragma mark -
|
|
#pragma mark == Utilities ==
|
|
#endif
|
|
|
|
//===========================================================================================================================
|
|
// RegisterServer
|
|
//===========================================================================================================================
|
|
|
|
DEBUG_LOCAL OSStatus RegisterServer( HINSTANCE inInstance, CLSID inCLSID, LPCTSTR inName )
|
|
{
|
|
typedef struct RegistryBuilder RegistryBuilder;
|
|
struct RegistryBuilder
|
|
{
|
|
HKEY rootKey;
|
|
LPCTSTR subKey;
|
|
LPCTSTR valueName;
|
|
LPCTSTR data;
|
|
};
|
|
|
|
OSStatus err;
|
|
LPWSTR clsidWideString;
|
|
TCHAR clsidString[ 64 ];
|
|
DWORD nChars;
|
|
size_t n;
|
|
size_t i;
|
|
HKEY key;
|
|
TCHAR keyName[ MAX_PATH ];
|
|
TCHAR moduleName[ MAX_PATH ] = TEXT( "" );
|
|
TCHAR data[ MAX_PATH ];
|
|
RegistryBuilder entries[] =
|
|
{
|
|
{ HKEY_CLASSES_ROOT, TEXT( "CLSID\\%s" ), NULL, inName },
|
|
{ HKEY_CLASSES_ROOT, TEXT( "CLSID\\%s\\InprocServer32" ), NULL, moduleName },
|
|
{ HKEY_CLASSES_ROOT, TEXT( "CLSID\\%s\\InprocServer32" ), TEXT( "ThreadingModel" ), TEXT( "Apartment" ) }
|
|
};
|
|
DWORD size;
|
|
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
|
|
|
|
// Register the CLSID entries.
|
|
|
|
nChars = GetModuleFileName( inInstance, moduleName, sizeof_array( moduleName ) );
|
|
err = translate_errno( nChars > 0, (OSStatus) GetLastError(), kUnknownErr );
|
|
require_noerr( err, exit );
|
|
|
|
n = sizeof_array( entries );
|
|
for( i = 0; i < n; ++i )
|
|
{
|
|
wsprintf( keyName, entries[ i ].subKey, clsidString );
|
|
err = RegCreateKeyEx( entries[ i ].rootKey, keyName, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &key, NULL );
|
|
require_noerr( err, exit );
|
|
|
|
size = (DWORD)( ( lstrlen( entries[ i ].data ) + 1 ) * sizeof( TCHAR ) );
|
|
err = RegSetValueEx( key, entries[ i ].valueName, 0, REG_SZ, (LPBYTE) entries[ i ].data, size );
|
|
RegCloseKey( key );
|
|
require_noerr( err, exit );
|
|
}
|
|
|
|
// If running on NT, 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 );
|
|
|
|
lstrcpyn( data, inName, sizeof_array( data ) );
|
|
size = (DWORD)( ( lstrlen( data ) + 1 ) * sizeof( TCHAR ) );
|
|
err = RegSetValueEx( key, clsidString, 0, REG_SZ, (LPBYTE) data, size );
|
|
RegCloseKey( key );
|
|
}
|
|
|
|
// register toolbar button
|
|
lstrcpyn( keyName, TEXT( "SOFTWARE\\Microsoft\\Internet Explorer\\Extensions\\{7F9DB11C-E358-4ca6-A83D-ACC663939424}"), sizeof_array( keyName ) );
|
|
err = RegCreateKeyEx( HKEY_LOCAL_MACHINE, keyName, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &key, NULL );
|
|
require_noerr( err, exit );
|
|
|
|
lstrcpyn( data, L"Yes", sizeof_array( data ) );
|
|
size = (DWORD)( ( lstrlen( data ) + 1 ) * sizeof( TCHAR ) );
|
|
RegSetValueEx( key, L"Default Visible", 0, REG_SZ, (LPBYTE) data, size );
|
|
|
|
lstrcpyn( data, inName, sizeof_array( data ) );
|
|
size = (DWORD)( ( lstrlen( data ) + 1 ) * sizeof( TCHAR ) );
|
|
RegSetValueEx( key, L"ButtonText", 0, REG_SZ, (LPBYTE) data, size );
|
|
|
|
lstrcpyn( data, L"{E0DD6CAB-2D10-11D2-8F1A-0000F87ABD16}", sizeof_array( data ) );
|
|
size = (DWORD)( ( lstrlen( data ) + 1 ) * sizeof( TCHAR ) );
|
|
RegSetValueEx( key, L"CLSID", 0, REG_SZ, (LPBYTE) data, size );
|
|
|
|
lstrcpyn( data, clsidString, sizeof_array( data ) );
|
|
size = (DWORD)( ( lstrlen( data ) + 1 ) * sizeof( TCHAR ) );
|
|
RegSetValueEx( key, L"BandCLSID", 0, REG_SZ, (LPBYTE) data, size );
|
|
|
|
// check if we're running XP or later
|
|
if ( ( versionInfo.dwPlatformId == VER_PLATFORM_WIN32_NT ) &&
|
|
( versionInfo.dwMajorVersion == 5 ) &&
|
|
( versionInfo.dwMinorVersion >= 1 ) )
|
|
{
|
|
wsprintf( data, L"%s,%d", (LPCTSTR) g_nonLocalizedResourcesName, IDI_BUTTON_XP );
|
|
size = (DWORD)( ( lstrlen( data ) + 1 ) * sizeof( TCHAR ) );
|
|
RegSetValueEx( key, L"Icon", 0, REG_SZ, (LPBYTE) data, size);
|
|
|
|
wsprintf( data, L"%s,%d", (LPCTSTR) g_nonLocalizedResourcesName, IDI_BUTTON_XP );
|
|
size = (DWORD)( ( lstrlen( data ) + 1 ) * sizeof( TCHAR ) );
|
|
RegSetValueEx( key, L"HotIcon", 0, REG_SZ, (LPBYTE) data, size);
|
|
}
|
|
else
|
|
{
|
|
wsprintf( data, L"%s,%d", (LPCTSTR) g_nonLocalizedResourcesName, IDI_BUTTON_2K );
|
|
size = (DWORD)( ( lstrlen( data ) + 1 ) * sizeof( TCHAR ) );
|
|
RegSetValueEx( key, L"Icon", 0, REG_SZ, (LPBYTE) data, size);
|
|
|
|
wsprintf( data, L"%s,%d", (LPCTSTR) g_nonLocalizedResourcesName, IDI_BUTTON_2K );
|
|
size = (DWORD)( ( lstrlen( data ) + 1 ) * sizeof( TCHAR ) );
|
|
RegSetValueEx( key, L"HotIcon", 0, REG_SZ, (LPBYTE) data, size);
|
|
}
|
|
|
|
RegCloseKey( key );
|
|
|
|
exit:
|
|
return( err );
|
|
}
|
|
|
|
//===========================================================================================================================
|
|
// RegisterCOMCategory
|
|
//===========================================================================================================================
|
|
|
|
DEBUG_LOCAL OSStatus RegisterCOMCategory( CLSID inCLSID, CATID inCategoryID, BOOL inRegister )
|
|
{
|
|
HRESULT err;
|
|
ICatRegister * cat;
|
|
|
|
err = CoInitialize( NULL );
|
|
require( SUCCEEDED( err ), exit );
|
|
|
|
err = CoCreateInstance( CLSID_StdComponentCategoriesMgr, NULL, CLSCTX_INPROC_SERVER, IID_ICatRegister, (LPVOID *) &cat );
|
|
check( SUCCEEDED( err ) );
|
|
if( SUCCEEDED( err ) )
|
|
{
|
|
if( inRegister )
|
|
{
|
|
err = cat->RegisterClassImplCategories( inCLSID, 1, &inCategoryID );
|
|
check_noerr( err );
|
|
}
|
|
else
|
|
{
|
|
err = cat->UnRegisterClassImplCategories( inCLSID, 1, &inCategoryID );
|
|
check_noerr( err );
|
|
}
|
|
cat->Release();
|
|
}
|
|
CoUninitialize();
|
|
|
|
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;
|
|
}
|