mDNSWindows/ControlPanel/ControlPanelExe.cpp

   1 /* -*- Mode: C; tab-width: 4 -*-
   2  *
   3  * Copyright (c) 2002-2007 Apple 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
  19 #include "ControlPanelExe.h"
  20 #include "ConfigDialog.h"
  21 #include "ConfigPropertySheet.h"
  22 #include "resource.h"
  23
  24 #include <DebugServices.h>
  25 #include "loclibrary.h"
  26
  27
  28 #ifdef _DEBUG
  29 #define new DEBUG_NEW
  30 #undef THIS_FILE
  31 static char THIS_FILE[] = __FILE__;
  32 #endif
  33
  34 #ifndef HeapEnableTerminationOnCorruption
  35 #       define HeapEnableTerminationOnCorruption (HEAP_INFORMATION_CLASS) 1
  36 #endif
  37
  38
  39 // Stash away pointers to our resource DLLs
  40
  41 static HINSTANCE g_nonLocalizedResources        = NULL;
  42 static HINSTANCE g_localizedResources           = NULL;
  43
  44
  45 HINSTANCE       GetNonLocalizedResources()
  46 {
  47         return g_nonLocalizedResources;
  48 }
  49
  50
  51 HINSTANCE       GetLocalizedResources()
  52 {
  53         return g_localizedResources;
  54 }
  55
  56
  57 //---------------------------------------------------------------------------------------------------------------------------
  58 //      Static Declarations
  59 //---------------------------------------------------------------------------------------------------------------------------
  60 DEFINE_GUID(CLSID_ControlPanel,
  61
  62 0x1207552c, 0xe59, 0x4d9f, 0x85, 0x54, 0xf1, 0xf8, 0x6, 0xcd, 0x7f, 0xa9);
  63
  64 static LPCTSTR g_controlPanelGUID                       =       TEXT( "{1207552C-0E59-4d9f-8554-F1F806CD7FA9}" );
  65 static LPCTSTR g_controlPanelName                       =       TEXT( "Bonjour" );
  66 static LPCTSTR g_controlPanelCanonicalName      =       TEXT( "Apple.Bonjour" );
  67 static LPCTSTR g_controlPanelCategory           =       TEXT( "3,8" );
  68
  69 static CCPApp theApp;
  70
  71 //===========================================================================================================================
  72 //      MyRegDeleteKey
  73 //===========================================================================================================================
  74
  75 DEBUG_LOCAL OSStatus MyRegDeleteKey( HKEY hKeyRoot, LPTSTR lpSubKey )
  76 {
  77     LPTSTR lpEnd;
  78     OSStatus err;
  79     DWORD dwSize;
  80     TCHAR szName[MAX_PATH];
  81     HKEY hKey;
  82     FILETIME ftWrite;
  83
  84     // First, see if we can delete the key without having to recurse.
  85
  86     err = RegDeleteKey( hKeyRoot, lpSubKey );
  87
  88     if ( !err )
  89         {
  90                 goto exit;
  91         }
  92
  93     err = RegOpenKeyEx( hKeyRoot, lpSubKey, 0, KEY_READ, &hKey );
  94         require_noerr( err, exit );
  95
  96     // Check for an ending slash and add one if it is missing.
  97
  98     lpEnd = lpSubKey + lstrlen(lpSubKey);
  99
 100     if ( *( lpEnd - 1 ) != TEXT( '\\' ) )
 101     {
 102         *lpEnd =  TEXT('\\');
 103         lpEnd++;
 104         *lpEnd =  TEXT('\0');
 105     }
 106
 107     // Enumerate the keys
 108
 109     dwSize = MAX_PATH;
 110     err = RegEnumKeyEx(hKey, 0, szName, &dwSize, NULL, NULL, NULL, &ftWrite);
 111
 112     if ( !err )
 113     {
 114         do
 115                 {
 116             lstrcpy (lpEnd, szName);
 117
 118             if ( !MyRegDeleteKey( hKeyRoot, lpSubKey ) )
 119                         {
 120                 break;
 121             }
 122
 123             dwSize = MAX_PATH;
 124
 125             err = RegEnumKeyEx( hKey, 0, szName, &dwSize, NULL, NULL, NULL, &ftWrite );
 126
 127         }
 128                 while ( !err );
 129     }
 130
 131     lpEnd--;
 132     *lpEnd = TEXT('\0');
 133
 134     RegCloseKey( hKey );
 135
 136     // Try again to delete the key.
 137
 138     err = RegDeleteKey(hKeyRoot, lpSubKey);
 139         require_noerr( err, exit );
 140
 141 exit:
 142
 143         return err;
 144 }
 145
 146
 147
 148 //---------------------------------------------------------------------------------------------------------------------------
 149 //      CCPApp::CCPApp
 150 //---------------------------------------------------------------------------------------------------------------------------
 151 IMPLEMENT_DYNAMIC(CCPApp, CWinApp);
 152
 153 CCPApp::CCPApp()
 154 {
 155         debug_initialize( kDebugOutputTypeWindowsEventLog, "DNS-SD Control Panel", GetModuleHandle( NULL ) );
 156         debug_set_property( kDebugPropertyTagPrintLevel, kDebugLevelInfo );
 157 }
 158
 159
 160 //---------------------------------------------------------------------------------------------------------------------------
 161 //      CCPApp::~CCPApp
 162 //---------------------------------------------------------------------------------------------------------------------------
 163
 164 CCPApp::~CCPApp()
 165 {
 166 }
 167
 168
 169 void
 170 CCPApp::Register( LPCTSTR inClsidString, LPCTSTR inName, LPCTSTR inCanonicalName, LPCTSTR inCategory, LPCTSTR inLocalizedName, LPCTSTR inInfoTip, LPCTSTR inIconPath, LPCTSTR inExePath )
 171 {
 172         typedef struct  RegistryBuilder         RegistryBuilder;
 173
 174         struct  RegistryBuilder
 175         {
 176                 HKEY            rootKey;
 177                 LPCTSTR         subKey;
 178                 LPCTSTR         valueName;
 179                 DWORD           valueType;
 180                 LPCTSTR         data;
 181         };
 182
 183         OSStatus                        err;
 184         size_t                          n;
 185         size_t                          i;
 186         HKEY                            key;
 187         TCHAR                           keyName[ MAX_PATH ];
 188         RegistryBuilder         entries[] =
 189         {
 190                 { HKEY_LOCAL_MACHINE,   TEXT( "SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Explorer\\ControlPanel\\NameSpace\\%s" ),  NULL,                                                                   REG_SZ,         inName },
 191                 { HKEY_CLASSES_ROOT,    TEXT( "CLSID\\%s" ),                                                                                                                                                    NULL,                                                                   NULL,           NULL },
 192                 { HKEY_CLASSES_ROOT,    TEXT( "CLSID\\%s" ),                                                                                                                                                    TEXT( "System.ApplicationName" ),               REG_SZ,         inCanonicalName },
 193                 { HKEY_CLASSES_ROOT,    TEXT( "CLSID\\%s" ),                                                                                                                                                    TEXT( "System.ControlPanel.Category" ), REG_SZ,         inCategory },
 194                 { HKEY_CLASSES_ROOT,    TEXT( "CLSID\\%s" ),                                                                                                                                                    TEXT( "LocalizedString" ),                              REG_SZ,         inLocalizedName },
 195                 { HKEY_CLASSES_ROOT,    TEXT( "CLSID\\%s" ),                                                                                                                                                    TEXT( "InfoTip" ),                                              REG_SZ,         inInfoTip },
 196                 { HKEY_CLASSES_ROOT,    TEXT( "CLSID\\%s\\DefaultIcon" ),                                                                                                                               NULL,                                                                   REG_SZ,         inIconPath },
 197                 { HKEY_CLASSES_ROOT,    TEXT( "CLSID\\%s\\Shell" ),                                                                                                                                             NULL,                                                                   NULL,           NULL },
 198                 { HKEY_CLASSES_ROOT,    TEXT( "CLSID\\%s\\Shell\\Open" ),                                                                                                                               NULL,                                                                   NULL,           NULL },
 199                 { HKEY_CLASSES_ROOT,    TEXT( "CLSID\\%s\\Shell\\Open\\Command" ),                                                                                                              NULL,                                                                   REG_SZ,         inExePath }
 200         };
 201         DWORD                           size;
 202
 203         // Register the registry entries.
 204
 205         n = sizeof_array( entries );
 206         for( i = 0; i < n; ++i )
 207         {
 208                 wsprintf( keyName, entries[ i ].subKey, inClsidString );
 209                 err = RegCreateKeyEx( entries[ i ].rootKey, keyName, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &key, NULL );
 210                 require_noerr( err, exit );
 211
 212                 if ( entries[ i ].data )
 213                 {
 214                         size = (DWORD)( ( lstrlen( entries[ i ].data ) + 1 ) * sizeof( TCHAR ) );
 215                         err = RegSetValueEx( key, entries[ i ].valueName, 0, entries[ i ].valueType, (LPBYTE) entries[ i ].data, size );
 216                         require_noerr( err, exit );
 217                 }
 218
 219                 RegCloseKey( key );
 220         }
 221
 222 exit:
 223         return;
 224 }
 225
 226
 227 //-----------------------------------------------------------
 228 //      CCPApp::Unregister
 229 //-----------------------------------------------------------
 230 void
 231 CCPApp::Unregister( LPCTSTR clsidString )
 232 {
 233         TCHAR keyName[ MAX_PATH * 2 ];
 234
 235         wsprintf( keyName, L"SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Explorer\\ControlPanel\\NameSpace\\%s", clsidString );
 236         MyRegDeleteKey( HKEY_LOCAL_MACHINE, keyName );
 237
 238         wsprintf( keyName, L"CLSID\\%s", clsidString );
 239         MyRegDeleteKey( HKEY_CLASSES_ROOT, keyName );
 240 }
 241
 242
 243
 244 //-----------------------------------------------------------
 245 //      CCPApp::InitInstance
 246 //-----------------------------------------------------------
 247
 248 BOOL
 249 CCPApp::InitInstance()
 250 {
 251         CCommandLineInfo        commandLine;
 252         wchar_t                         resource[MAX_PATH];
 253         CString                         errorMessage;
 254         CString                         errorCaption;
 255         int                                     res;
 256         OSStatus                        err = kNoErr;
 257
 258         HeapSetInformation( NULL, HeapEnableTerminationOnCorruption, NULL, 0 );
 259
 260         //
 261         // initialize the debugging framework
 262         //
 263         debug_initialize( kDebugOutputTypeWindowsDebugger, "ControlPanel", NULL );
 264         debug_set_property( kDebugPropertyTagPrintLevel, kDebugLevelTrace );
 265
 266         // Before we load the resources, let's load the error string
 267
 268         errorMessage.LoadString( IDS_REINSTALL );
 269         errorCaption.LoadString( IDS_REINSTALL_CAPTION );
 270
 271         res = PathForResource( NULL, L"ControlPanelResources.dll", resource, MAX_PATH );
 272         err = translate_errno( res != 0, kUnknownErr, kUnknownErr );
 273         require_noerr( err, exit );
 274
 275         g_nonLocalizedResources = LoadLibrary( resource );
 276         translate_errno( g_nonLocalizedResources, GetLastError(), kUnknownErr );
 277         require_noerr( err, exit );
 278
 279         res = PathForResource( NULL, L"ControlPanelLocalized.dll", resource, MAX_PATH );
 280         err = translate_errno( res != 0, kUnknownErr, kUnknownErr );
 281         require_noerr( err, exit );
 282
 283         g_localizedResources = LoadLibrary( resource );
 284         translate_errno( g_localizedResources, GetLastError(), kUnknownErr );
 285         require_noerr( err, exit );
 286
 287         AfxSetResourceHandle( g_localizedResources );
 288
 289         // InitCommonControls() is required on Windows XP if an application
 290         // manifest specifies use of ComCtl32.dll version 6 or later to enable
 291         // visual styles.  Otherwise, any window creation will fail.
 292
 293         InitCommonControls();
 294
 295         CWinApp::InitInstance();
 296
 297         AfxEnableControlContainer();
 298
 299         ParseCommandLine( commandLine );
 300
 301         if ( commandLine.m_nShellCommand == CCommandLineInfo::AppRegister )
 302         {
 303                 CString         localizedName;
 304                 CString         toolTip;
 305                 TCHAR           iconPath[ MAX_PATH + 12 ]       = TEXT( "" );
 306                 TCHAR           exePath[ MAX_PATH ]                     = TEXT( "" );
 307                 DWORD           nChars;
 308                 OSStatus        err;
 309
 310                 nChars = GetModuleFileName( NULL, exePath, sizeof_array( exePath ) );
 311
 312                 err = translate_errno( nChars > 0, (OSStatus) GetLastError(), kUnknownErr );
 313
 314                 require_noerr( err, exit );
 315
 316                 wsprintf( iconPath, L"%s,-%d", exePath, IDR_APPLET );
 317
 318                 localizedName.LoadString( IDS_APPLET_NAME );
 319                 toolTip.LoadString( IDS_APPLET_TOOLTIP );
 320
 321                 Register( g_controlPanelGUID, g_controlPanelName, g_controlPanelCanonicalName, g_controlPanelCategory, localizedName, toolTip, iconPath, exePath );
 322         }
 323         else if ( commandLine.m_nShellCommand == CCommandLineInfo::AppUnregister )
 324         {
 325                 Unregister( g_controlPanelGUID );
 326         }
 327         else
 328         {
 329                 CString                                 name;
 330                 CConfigPropertySheet    dlg;
 331
 332                 name.LoadString( IDR_APPLET );
 333                 dlg.Construct( name, NULL, 0 );
 334
 335                 m_pMainWnd = &dlg;
 336
 337                 try
 338                 {
 339                         INT_PTR nResponse = dlg.DoModal();
 340
 341                         if (nResponse == IDOK)
 342                         {
 343                                 // TODO: Place code here to handle when the dialog is
 344                                 //  dismissed with OK
 345                         }
 346                         else if (nResponse == IDCANCEL)
 347                         {
 348                                 // TODO: Place code here to handle when the dialog is
 349                                 //  dismissed with Cancel
 350                         }
 351                 }
 352                 catch (...)
 353                 {
 354                         MessageBox(NULL, L"", L"", MB_OK|MB_ICONEXCLAMATION);
 355                 }
 356         }
 357
 358         if ( err )
 359         {
 360                 MessageBox( NULL, L"", L"", MB_ICONERROR | MB_OK );
 361         }
 362
 363 exit:
 364
 365         if ( err )
 366         {
 367                 MessageBox( NULL, errorMessage, errorCaption, MB_ICONERROR | MB_OK );
 368         }
 369
 370         // Since the dialog has been closed, return FALSE so that we exit the
 371         //  application, rather than start the application's message pump.
 372         return FALSE;
 373 }