mDNSWindows/ControlPanel/RegistrationPage.cpp

   1 /* -*- Mode: C; tab-width: 4 -*-
   2  *
   3  * Copyright (c) 2002-2004 Apple Computer, 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 #include <Secret.h>
  19 #include "RegistrationPage.h"
  20 #include "resource.h"
  21
  22 #include "ConfigPropertySheet.h"
  23 extern "C"
  24 {
  25 #include <ClientCommon.h>
  26 }
  27 #include <WinServices.h>
  28
  29 #define MAX_KEY_LENGTH 255
  30
  31
  32 IMPLEMENT_DYNCREATE(CRegistrationPage, CPropertyPage)
  33
  34 //---------------------------------------------------------------------------------------------------------------------------
  35 //      CRegistrationPage::CRegistrationPage
  36 //---------------------------------------------------------------------------------------------------------------------------
  37
  38 CRegistrationPage::CRegistrationPage()
  39 :
  40         CPropertyPage(CRegistrationPage::IDD),
  41         m_ignoreChanges( false ),
  42         m_hostnameSetupKey( NULL ),
  43         m_registrationSetupKey( NULL ),
  44         m_statusKey( NULL )
  45 {
  46         //{{AFX_DATA_INIT(CRegistrationPage)
  47         //}}AFX_DATA_INIT
  48
  49         OSStatus err;
  50
  51         err = RegCreateKeyEx( HKEY_LOCAL_MACHINE, kServiceParametersNode L"\\DynDNS\\Setup\\Hostnames", 0,
  52                               NULL, REG_OPTION_NON_VOLATILE, KEY_READ|KEY_WRITE, NULL, &m_hostnameSetupKey, NULL );
  53         check_noerr( err );
  54
  55         err = RegCreateKeyEx( HKEY_LOCAL_MACHINE, kServiceParametersNode L"\\DynDNS\\Setup\\" kServiceDynDNSRegistrationDomains, 0,
  56                               NULL, REG_OPTION_NON_VOLATILE, KEY_READ|KEY_WRITE, NULL, &m_registrationSetupKey, NULL );
  57         check_noerr( err );
  58
  59         err = RegCreateKeyEx( HKEY_LOCAL_MACHINE, kServiceParametersNode L"\\DynDNS\\State\\Hostnames", 0,
  60                               NULL, REG_OPTION_NON_VOLATILE, KEY_READ|KEY_WRITE, NULL, &m_statusKey, NULL );
  61         check_noerr( err );
  62
  63
  64 }
  65
  66 CRegistrationPage::~CRegistrationPage()
  67 {
  68         if ( m_hostnameSetupKey )
  69         {
  70                 RegCloseKey( m_hostnameSetupKey );
  71                 m_hostnameSetupKey = NULL;
  72         }
  73
  74         if ( m_registrationSetupKey )
  75         {
  76                 RegCloseKey( m_registrationSetupKey );
  77                 m_registrationSetupKey = NULL;
  78         }
  79
  80         if ( m_statusKey )
  81         {
  82                 RegCloseKey( m_statusKey );
  83                 m_statusKey = NULL;
  84         }
  85 }
  86
  87
  88 //---------------------------------------------------------------------------------------------------------------------------
  89 //      CRegistrationPage::DoDataExchange
  90 //---------------------------------------------------------------------------------------------------------------------------
  91
  92 void CRegistrationPage::DoDataExchange(CDataExchange* pDX)
  93 {
  94         CPropertyPage::DoDataExchange(pDX);
  95         //{{AFX_DATA_MAP(CRegistrationPage)
  96         //}}AFX_DATA_MAP
  97         DDX_Control(pDX, IDC_HOSTNAME, m_hostnameControl);
  98         DDX_Control(pDX, IDC_USERNAME, m_usernameControl);
  99         DDX_Control(pDX, IDC_PASSWORD, m_passwordControl);
 100         DDX_Control(pDX, IDC_ADVERTISE_SERVICES, m_advertiseServices);
 101 }
 102
 103 BEGIN_MESSAGE_MAP(CRegistrationPage, CPropertyPage)
 104         //{{AFX_MSG_MAP(CRegistrationPage)
 105         //}}AFX_MSG_MAP
 106         ON_EN_CHANGE(IDC_HOSTNAME, OnEnChangeHostname)
 107         ON_EN_CHANGE(IDC_USERNAME, OnEnChangeUsername)
 108         ON_EN_CHANGE(IDC_PASSWORD, OnEnChangePassword)
 109         ON_BN_CLICKED(IDC_ADVERTISE_SERVICES, OnBnClickedAdvertiseServices)
 110 END_MESSAGE_MAP()
 111
 112
 113 //---------------------------------------------------------------------------------------------------------------------------
 114 //      CRegistrationPage::OnEnChangedHostname
 115 //---------------------------------------------------------------------------------------------------------------------------
 116
 117 void CRegistrationPage::OnEnChangeHostname()
 118 {
 119         if ( !m_ignoreChanges )
 120         {
 121                 SetModified( TRUE );
 122         }
 123 }
 124
 125
 126 //---------------------------------------------------------------------------------------------------------------------------
 127 //      CRegistrationPage::OnEnChangedUsername
 128 //---------------------------------------------------------------------------------------------------------------------------
 129
 130 void CRegistrationPage::OnEnChangeUsername()
 131 {
 132         if ( !m_ignoreChanges )
 133         {
 134                 SetModified( TRUE );
 135         }
 136 }
 137
 138
 139 //---------------------------------------------------------------------------------------------------------------------------
 140 //      CRegistrationPage::OnEnChangedPassword
 141 //---------------------------------------------------------------------------------------------------------------------------
 142
 143 void CRegistrationPage::OnEnChangePassword()
 144 {
 145         if ( !m_ignoreChanges )
 146         {
 147                 SetModified( TRUE );
 148         }
 149 }
 150
 151
 152 //---------------------------------------------------------------------------------------------------------------------------
 153 //      CRegistrationPage::OnBnClickedAdvertiseServices
 154 //---------------------------------------------------------------------------------------------------------------------------
 155
 156 void CRegistrationPage::OnBnClickedAdvertiseServices()
 157 {
 158         if ( !m_ignoreChanges )
 159         {
 160                 SetModified( TRUE );
 161         }
 162 }
 163
 164
 165 //---------------------------------------------------------------------------------------------------------------------------
 166 //      CRegistrationPage::SetModified
 167 //---------------------------------------------------------------------------------------------------------------------------
 168
 169 void CRegistrationPage::SetModified( BOOL bChanged )
 170 {
 171         m_modified = bChanged ? true : false;
 172
 173         CPropertyPage::SetModified( bChanged );
 174 }
 175
 176
 177 //---------------------------------------------------------------------------------------------------------------------------
 178 //      CRegistrationPage::OnSetActive
 179 //---------------------------------------------------------------------------------------------------------------------------
 180
 181 BOOL
 182 CRegistrationPage::OnSetActive()
 183 {
 184         TCHAR   name[kDNSServiceMaxDomainName + 1];
 185         DWORD   nameLen = ( kDNSServiceMaxDomainName + 1 ) * sizeof( TCHAR );
 186         DWORD   err;
 187
 188         BOOL b = CPropertyPage::OnSetActive();
 189
 190         m_ignoreChanges = true;
 191         m_modified = FALSE;
 192
 193         if ( m_hostnameSetupKey )
 194         {
 195                 err = RegQueryValueEx( m_hostnameSetupKey, L"", NULL, NULL, (LPBYTE) name, &nameLen );
 196
 197                 if ( !err )
 198                 {
 199                         char    hostnameUTF8[ 256 ];
 200                         char    outDomain[ 256 ];
 201                         char    outUsername[ 256 ];
 202                         char    outPassword[ 256 ];
 203                         CString hostname = name;
 204                         CString username;
 205                         CString password;
 206
 207                         m_hostnameControl.SetWindowText( hostname );
 208
 209                         StringObjectToUTF8String( hostname, hostnameUTF8, sizeof( hostnameUTF8 ) );
 210
 211                         if ( LsaGetSecret( hostnameUTF8, outDomain, sizeof( outDomain ) / sizeof( TCHAR ), outUsername, sizeof( outUsername ) / sizeof( TCHAR ), outPassword, sizeof( outPassword ) / sizeof( TCHAR ) ) )
 212                         {
 213                                 username = outUsername;
 214                                 m_usernameControl.SetWindowText( username );
 215
 216                                 password = outPassword;
 217                                 m_passwordControl.SetWindowText( password );
 218                         }
 219                 }
 220         }
 221
 222         m_advertiseServices.SetCheck( 0 );
 223
 224         if ( m_registrationSetupKey )
 225         {
 226                 HKEY            subKey = NULL;
 227                 DWORD           dwSize;
 228                 DWORD           enabled = 0;
 229                 TCHAR           subKeyName[MAX_KEY_LENGTH];
 230                 DWORD           cSubKeys = 0;
 231                 DWORD           cbMaxSubKey;
 232                 DWORD           cchMaxClass;
 233                 OSStatus        err;
 234
 235                 err = RegQueryInfoKey( m_registrationSetupKey, NULL, NULL, NULL, &cSubKeys, &cbMaxSubKey, &cchMaxClass, NULL, NULL, NULL, NULL, NULL );
 236                 if ( !err )
 237                 {
 238                         if ( cSubKeys > 0 )
 239                         {
 240                                 dwSize = MAX_KEY_LENGTH;
 241
 242                                 err = RegEnumKeyEx( m_registrationSetupKey, 0, subKeyName, &dwSize, NULL, NULL, NULL, NULL );
 243                                 if ( !err )
 244                                 {
 245                                         err = RegOpenKey( m_registrationSetupKey, subKeyName, &subKey );
 246                                         if ( !err )
 247                                         {
 248                                                 dwSize = sizeof( DWORD );
 249                                                 err = RegQueryValueEx( subKey, L"Enabled", NULL, NULL, (LPBYTE) &enabled, &dwSize );
 250                                                 if ( !err && enabled )
 251                                                 {
 252                                                         m_advertiseServices.SetCheck( enabled );
 253                                                 }
 254
 255                                                 RegCloseKey( subKey );
 256                                                 subKey = NULL;
 257                                         }
 258                                 }
 259                         }
 260                 }
 261         }
 262
 263         m_ignoreChanges = false;
 264
 265         return b;
 266 }
 267
 268
 269 //---------------------------------------------------------------------------------------------------------------------------
 270 //      CRegistrationPage::OnOK
 271 //---------------------------------------------------------------------------------------------------------------------------
 272
 273 void
 274 CRegistrationPage::OnOK()
 275 {
 276         if ( m_modified )
 277         {
 278                 Commit();
 279         }
 280 }
 281
 282
 283 //---------------------------------------------------------------------------------------------------------------------------
 284 //      CRegistrationPage::Commit
 285 //---------------------------------------------------------------------------------------------------------------------------
 286
 287 void
 288 CRegistrationPage::Commit()
 289 {
 290         CString hostname;
 291         char    hostnameUTF8[ 256 ];
 292         CString username;
 293         char    usernameUTF8[ 256 ];
 294         CString password;
 295         char    passwordUTF8[ 256 ];
 296         DWORD   enabled = 1;
 297         BOOL    secret = FALSE;
 298         DWORD   err;
 299
 300         m_hostnameControl.GetWindowText( hostname );
 301         hostname.MakeLower();
 302         hostname.TrimRight( '.' );
 303         StringObjectToUTF8String( hostname, hostnameUTF8, sizeof( hostnameUTF8 ) );
 304
 305         m_usernameControl.GetWindowText( username );
 306         m_passwordControl.GetWindowText( password );
 307
 308         if ( username.GetLength() && password.GetLength() )
 309         {
 310                 StringObjectToUTF8String( username, usernameUTF8, sizeof( usernameUTF8 ) );
 311                 StringObjectToUTF8String( password, passwordUTF8, sizeof( passwordUTF8 ) );
 312                 secret = TRUE;
 313         }
 314
 315         if ( m_hostnameSetupKey != NULL )
 316         {
 317                 err = RegSetValueEx( m_hostnameSetupKey, L"", 0, REG_SZ, (LPBYTE) (LPCTSTR) hostname, ( hostname.GetLength() + 1 ) * sizeof( TCHAR ) );
 318                 require_noerr( err, exit );
 319
 320                 err = RegSetValueEx( m_hostnameSetupKey, L"Enabled", 0, REG_DWORD, (LPBYTE) &enabled, sizeof( DWORD ) );
 321                 require_noerr( err, exit );
 322
 323                 if ( secret )
 324                 {
 325                         LsaSetSecret( hostnameUTF8, usernameUTF8, passwordUTF8 );
 326                 }
 327         }
 328
 329         if ( m_registrationSetupKey != NULL )
 330         {
 331                 TCHAR           subKeyName[MAX_KEY_LENGTH];
 332                 DWORD           cSubKeys = 0;
 333                 DWORD           cbMaxSubKey;
 334                 DWORD           cchMaxClass;
 335                 DWORD           dwSize;
 336                 int                     i;
 337                 OSStatus        err = kNoErr;
 338
 339                 // First, remove all the entries that are there
 340
 341                 err = RegQueryInfoKey( m_registrationSetupKey, NULL, NULL, NULL, &cSubKeys, &cbMaxSubKey, &cchMaxClass, NULL, NULL, NULL, NULL, NULL );
 342                 if ( !err )
 343                 {
 344                         for ( i = 0; i < (int) cSubKeys; i++ )
 345                         {
 346                                 dwSize = MAX_KEY_LENGTH;
 347
 348                                 err = RegEnumKeyEx( m_registrationSetupKey, 0, subKeyName, &dwSize, NULL, NULL, NULL, NULL );
 349                                 require_noerr( err, exit );
 350
 351                                 err = RegDeleteKey( m_registrationSetupKey, subKeyName );
 352                                 require_noerr( err, exit );
 353                         }
 354                 }
 355
 356                 if ( m_advertiseServices.GetCheck() )
 357                 {
 358                         const char      * domainUTF8;
 359                         CString           domain;
 360                         char              label[ 64 ];
 361                         HKEY              subKey = NULL;
 362
 363                         domainUTF8      = GetNextLabel( hostnameUTF8, label );
 364                         domain          = domainUTF8;
 365
 366                         err = RegCreateKeyEx( m_registrationSetupKey, domain, 0,
 367                                                                          NULL, REG_OPTION_NON_VOLATILE, KEY_READ|KEY_WRITE, NULL, &subKey, NULL );
 368                         if ( !err )
 369                         {
 370                                 err = RegSetValueEx( subKey, L"Enabled", 0, REG_DWORD, (LPBYTE) &enabled, sizeof( DWORD ) );
 371                                 check_noerr( err );
 372
 373                                 RegCloseKey( subKey );
 374                                 subKey = NULL;
 375                         }
 376
 377                         if ( secret )
 378                         {
 379                                 LsaSetSecret( domainUTF8, usernameUTF8, passwordUTF8 );
 380                         }
 381                 }
 382         }
 383
 384 exit:
 385
 386         return;
 387 }