mDNSWindows/ControlPanel/FirstPage.cpp

   1 /*
   2  * Copyright (c) 2002-2004 Apple Computer, Inc. All rights reserved.
   3  *
   4  * @APPLE_LICENSE_HEADER_START@
   5  *
   6  * This file contains Original Code and/or Modifications of Original Code
   7  * as defined in and that are subject to the Apple Public Source License
   8  * Version 2.0 (the 'License'). You may not use this file except in
   9  * compliance with the License. Please obtain a copy of the License at
  10  * http://www.opensource.apple.com/apsl/ and read it before using this
  11  * file.
  12  *
  13  * The Original Code and all software distributed under the License are
  14  * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
  15  * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
  16  * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
  17  * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
  18  * Please see the License for the specific language governing rights and
  19  * limitations under the License.
  20  *
  21  * @APPLE_LICENSE_HEADER_END@
  22
  23     Change History (most recent first):
  24
  25 $Log: FirstPage.cpp,v $
  26 Revision 1.5  2005/10/05 20:46:50  herscher
  27 <rdar://problem/4192011> Move Wide-Area preferences to another part of the registry so they don't removed during an update-install.
  28
  29 Revision 1.4  2005/04/05 03:52:14  shersche
  30 <rdar://problem/4066485> Registering with shared secret key doesn't work. Additionally, mDNSResponder wasn't dynamically re-reading it's DynDNS setup after setting a shared secret key.
  31
  32 Revision 1.3  2005/03/07 18:27:42  shersche
  33 <rdar://problem/4037940> Fix problem when ControlPanel commits changes to the browse domain list
  34
  35 Revision 1.2  2005/03/03 19:55:22  shersche
  36 <rdar://problem/4034481> ControlPanel source code isn't saving CVS log info
  37
  38
  39 */
  40
  41 #include "FirstPage.h"
  42 #include "resource.h"
  43
  44 #include "ConfigPropertySheet.h"
  45 #include "SharedSecret.h"
  46
  47 #define MAX_KEY_LENGTH 255
  48
  49
  50 IMPLEMENT_DYNCREATE(CFirstPage, CPropertyPage)
  51
  52 //---------------------------------------------------------------------------------------------------------------------------
  53 //      CFirstPage::CFirstPage
  54 //---------------------------------------------------------------------------------------------------------------------------
  55
  56 CFirstPage::CFirstPage()
  57 :
  58         CPropertyPage(CFirstPage::IDD),
  59         m_ignoreHostnameChange( false ),
  60         m_statusKey( NULL ),
  61         m_setupKey( NULL )
  62 {
  63         //{{AFX_DATA_INIT(CFirstPage)
  64         //}}AFX_DATA_INIT
  65
  66         OSStatus err;
  67
  68         err = RegCreateKey( HKEY_LOCAL_MACHINE, kServiceParametersNode L"\\DynDNS\\State\\Hostnames", &m_statusKey );
  69         check_noerr( err );
  70
  71         err = RegCreateKey( HKEY_LOCAL_MACHINE, kServiceParametersNode L"\\DynDNS\\Setup\\Hostnames", &m_setupKey );
  72         check_noerr( err );
  73 }
  74
  75 CFirstPage::~CFirstPage()
  76 {
  77         if ( m_statusKey )
  78         {
  79                 RegCloseKey( m_statusKey );
  80                 m_statusKey = NULL;
  81         }
  82
  83         if ( m_setupKey )
  84         {
  85                 RegCloseKey( m_setupKey );
  86                 m_setupKey = NULL;
  87         }
  88 }
  89
  90
  91 //---------------------------------------------------------------------------------------------------------------------------
  92 //      CFirstPage::DoDataExchange
  93 //---------------------------------------------------------------------------------------------------------------------------
  94
  95 void CFirstPage::DoDataExchange(CDataExchange* pDX)
  96 {
  97         CPropertyPage::DoDataExchange(pDX);
  98         //{{AFX_DATA_MAP(CFirstPage)
  99         //}}AFX_DATA_MAP
 100         DDX_Control(pDX, IDC_EDIT1, m_hostnameControl);
 101         DDX_Control(pDX, IDC_FAILURE, m_failureIcon);
 102         DDX_Control(pDX, IDC_SUCCESS, m_successIcon);
 103 }
 104
 105 BEGIN_MESSAGE_MAP(CFirstPage, CPropertyPage)
 106         //{{AFX_MSG_MAP(CFirstPage)
 107         //}}AFX_MSG_MAP
 108         ON_BN_CLICKED(IDC_BUTTON1, OnBnClickedSharedSecret)
 109         ON_EN_CHANGE(IDC_EDIT1, OnEnChangeHostname)
 110 END_MESSAGE_MAP()
 111
 112
 113 //---------------------------------------------------------------------------------------------------------------------------
 114 //      CFirstPage::OnEnChangedHostname
 115 //---------------------------------------------------------------------------------------------------------------------------
 116
 117 void CFirstPage::OnEnChangeHostname()
 118 {
 119         if ( !m_ignoreHostnameChange )
 120         {
 121                 SetModified( TRUE );
 122         }
 123 }
 124
 125
 126 //---------------------------------------------------------------------------------------------------------------------------
 127 //      CFirstPage::OnBnClickedSharedSecret
 128 //---------------------------------------------------------------------------------------------------------------------------
 129
 130 void CFirstPage::OnBnClickedSharedSecret()
 131 {
 132         CString name;
 133
 134         m_hostnameControl.GetWindowText( name );
 135
 136         CSharedSecret dlg;
 137
 138         dlg.m_key = name;
 139
 140         if ( dlg.DoModal() == IDOK )
 141         {
 142                 DWORD           wakeup = 0;
 143                 DWORD           dwSize = sizeof( DWORD );
 144                 OSStatus        err;
 145
 146                 dlg.Commit( name );
 147
 148                 // We have now updated the secret, however the system service
 149                 // doesn't know about it yet.  So we're going to update the
 150                 // registry with a dummy value which will cause the system
 151                 // service to re-initialize it's DynDNS setup
 152                 //
 153
 154                 RegQueryValueEx( m_setupKey, L"Wakeup", NULL, NULL, (LPBYTE) &wakeup, &dwSize );
 155
 156                 wakeup++;
 157
 158                 err = RegSetValueEx( m_setupKey, L"Wakeup", 0, REG_DWORD, (LPBYTE) &wakeup, sizeof( DWORD ) );
 159                 require_noerr( err, exit );
 160         }
 161
 162 exit:
 163
 164         return;
 165 }
 166
 167
 168 //---------------------------------------------------------------------------------------------------------------------------
 169 //      CFirstPage::SetModified
 170 //---------------------------------------------------------------------------------------------------------------------------
 171
 172 void CFirstPage::SetModified( BOOL bChanged )
 173 {
 174         m_modified = bChanged ? true : false;
 175
 176         CPropertyPage::SetModified( bChanged );
 177 }
 178
 179
 180 //---------------------------------------------------------------------------------------------------------------------------
 181 //      CFirstPage::OnSetActive
 182 //---------------------------------------------------------------------------------------------------------------------------
 183
 184 BOOL
 185 CFirstPage::OnSetActive()
 186 {
 187         TCHAR   name[kDNSServiceMaxDomainName + 1];
 188         DWORD   nameLen = ( kDNSServiceMaxDomainName + 1 ) * sizeof( TCHAR );
 189         DWORD   err;
 190
 191         BOOL b = CPropertyPage::OnSetActive();
 192
 193         m_modified = FALSE;
 194
 195         if ( m_setupKey )
 196         {
 197                 err = RegQueryValueEx( m_setupKey, L"", NULL, NULL, (LPBYTE) name, &nameLen );
 198
 199                 if ( !err )
 200                 {
 201                         m_ignoreHostnameChange = true;
 202                         m_hostnameControl.SetWindowText( name );
 203                         m_ignoreHostnameChange = false;
 204                 }
 205         }
 206
 207         // Check the status of this hostname
 208
 209         err = CheckStatus();
 210         check_noerr( err );
 211
 212         return b;
 213 }
 214
 215
 216 //---------------------------------------------------------------------------------------------------------------------------
 217 //      CFirstPage::OnOK
 218 //---------------------------------------------------------------------------------------------------------------------------
 219
 220 void
 221 CFirstPage::OnOK()
 222 {
 223         if ( m_modified )
 224         {
 225                 Commit();
 226         }
 227 }
 228
 229
 230 //---------------------------------------------------------------------------------------------------------------------------
 231 //      CFirstPage::Commit
 232 //---------------------------------------------------------------------------------------------------------------------------
 233
 234 void
 235 CFirstPage::Commit()
 236 {
 237         DWORD   enabled = 1;
 238         CString name;
 239         DWORD   err;
 240
 241         m_hostnameControl.GetWindowText( name );
 242
 243         // Convert to lower case
 244
 245         name.MakeLower();
 246
 247         // Remove trailing dot
 248
 249         name.TrimRight( '.' );
 250
 251         err = RegSetValueEx( m_setupKey, L"", 0, REG_SZ, (LPBYTE) (LPCTSTR) name, ( name.GetLength() + 1 ) * sizeof( TCHAR ) );
 252         require_noerr( err, exit );
 253
 254         err = RegSetValueEx( m_setupKey, L"Enabled", 0, REG_DWORD, (LPBYTE) &enabled, sizeof( DWORD ) );
 255         require_noerr( err, exit );
 256
 257 exit:
 258
 259         return;
 260 }
 261
 262
 263 //---------------------------------------------------------------------------------------------------------------------------
 264 //      CFirstPage::CheckStatus
 265 //---------------------------------------------------------------------------------------------------------------------------
 266
 267 OSStatus
 268 CFirstPage::CheckStatus()
 269 {
 270         DWORD           status = 0;
 271         DWORD           dwSize = sizeof( DWORD );
 272         OSStatus        err;
 273
 274         // Get the status field
 275
 276         err = RegQueryValueEx( m_statusKey, L"Status", NULL, NULL, (LPBYTE) &status, &dwSize );
 277         require_noerr( err, exit );
 278
 279         ShowStatus( status );
 280
 281 exit:
 282
 283         return kNoErr;
 284 }
 285
 286
 287 //---------------------------------------------------------------------------------------------------------------------------
 288 //      CFirstPage::ShowStatus
 289 //---------------------------------------------------------------------------------------------------------------------------
 290
 291 void
 292 CFirstPage::ShowStatus( DWORD status )
 293 {
 294         if ( status )
 295         {
 296                 m_failureIcon.ShowWindow( SW_HIDE );
 297                 m_successIcon.ShowWindow( SW_SHOW );
 298         }
 299         else
 300         {
 301                 m_failureIcon.ShowWindow( SW_SHOW );
 302                 m_successIcon.ShowWindow( SW_HIDE );
 303         }
 304 }
 305
 306
 307 //---------------------------------------------------------------------------------------------------------------------------
 308 //      CFirstPage::OnRegistryChanged
 309 //---------------------------------------------------------------------------------------------------------------------------
 310
 311 void
 312 CFirstPage::OnRegistryChanged()
 313 {
 314         CheckStatus();
 315 }