--- /dev/null
+/* -*- Mode: C; tab-width: 4 -*-
+ *
+ * Copyright (c) 2002-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 <Secret.h>
+#include "RegistrationPage.h"
+#include "resource.h"
+
+#include "ConfigPropertySheet.h"
+extern "C"
+{
+#include <ClientCommon.h>
+}
+#include <WinServices.h>
+
+#define MAX_KEY_LENGTH 255
+
+
+IMPLEMENT_DYNCREATE(CRegistrationPage, CPropertyPage)
+
+//---------------------------------------------------------------------------------------------------------------------------
+// CRegistrationPage::CRegistrationPage
+//---------------------------------------------------------------------------------------------------------------------------
+
+CRegistrationPage::CRegistrationPage()
+:
+ CPropertyPage(CRegistrationPage::IDD),
+ m_ignoreChanges( false ),
+ m_hostnameSetupKey( NULL ),
+ m_registrationSetupKey( NULL ),
+ m_statusKey( NULL )
+{
+ //{{AFX_DATA_INIT(CRegistrationPage)
+ //}}AFX_DATA_INIT
+
+ OSStatus err;
+
+ err = RegCreateKeyEx( HKEY_LOCAL_MACHINE, kServiceParametersNode L"\\DynDNS\\Setup\\Hostnames", 0,
+ NULL, REG_OPTION_NON_VOLATILE, KEY_READ|KEY_WRITE|KEY_WOW64_32KEY, NULL, &m_hostnameSetupKey, NULL );
+ check_noerr( err );
+
+ err = RegCreateKeyEx( HKEY_LOCAL_MACHINE, kServiceParametersNode L"\\DynDNS\\Setup\\" kServiceDynDNSRegistrationDomains, 0,
+ NULL, REG_OPTION_NON_VOLATILE, KEY_READ|KEY_WRITE|KEY_WOW64_32KEY, NULL, &m_registrationSetupKey, NULL );
+ check_noerr( err );
+
+ err = RegCreateKeyEx( HKEY_LOCAL_MACHINE, kServiceParametersNode L"\\DynDNS\\State\\Hostnames", 0,
+ NULL, REG_OPTION_NON_VOLATILE, KEY_READ|KEY_WRITE|KEY_WOW64_32KEY, NULL, &m_statusKey, NULL );
+ check_noerr( err );
+
+
+}
+
+CRegistrationPage::~CRegistrationPage()
+{
+ if ( m_hostnameSetupKey )
+ {
+ RegCloseKey( m_hostnameSetupKey );
+ m_hostnameSetupKey = NULL;
+ }
+
+ if ( m_registrationSetupKey )
+ {
+ RegCloseKey( m_registrationSetupKey );
+ m_registrationSetupKey = NULL;
+ }
+
+ if ( m_statusKey )
+ {
+ RegCloseKey( m_statusKey );
+ m_statusKey = NULL;
+ }
+}
+
+
+//---------------------------------------------------------------------------------------------------------------------------
+// CRegistrationPage::DoDataExchange
+//---------------------------------------------------------------------------------------------------------------------------
+
+void CRegistrationPage::DoDataExchange(CDataExchange* pDX)
+{
+ CPropertyPage::DoDataExchange(pDX);
+ //{{AFX_DATA_MAP(CRegistrationPage)
+ //}}AFX_DATA_MAP
+ DDX_Control(pDX, IDC_HOSTNAME, m_hostnameControl);
+ DDX_Control(pDX, IDC_USERNAME, m_usernameControl);
+ DDX_Control(pDX, IDC_PASSWORD, m_passwordControl);
+ DDX_Control(pDX, IDC_ADVERTISE_SERVICES, m_advertiseServices);
+}
+
+BEGIN_MESSAGE_MAP(CRegistrationPage, CPropertyPage)
+ //{{AFX_MSG_MAP(CRegistrationPage)
+ //}}AFX_MSG_MAP
+ ON_EN_CHANGE(IDC_HOSTNAME, OnEnChangeHostname)
+ ON_EN_CHANGE(IDC_USERNAME, OnEnChangeUsername)
+ ON_EN_CHANGE(IDC_PASSWORD, OnEnChangePassword)
+ ON_BN_CLICKED(IDC_ADVERTISE_SERVICES, OnBnClickedAdvertiseServices)
+END_MESSAGE_MAP()
+
+
+//---------------------------------------------------------------------------------------------------------------------------
+// CRegistrationPage::OnEnChangedHostname
+//---------------------------------------------------------------------------------------------------------------------------
+
+void CRegistrationPage::OnEnChangeHostname()
+{
+ if ( !m_ignoreChanges )
+ {
+ SetModified( TRUE );
+ }
+}
+
+
+//---------------------------------------------------------------------------------------------------------------------------
+// CRegistrationPage::OnEnChangedUsername
+//---------------------------------------------------------------------------------------------------------------------------
+
+void CRegistrationPage::OnEnChangeUsername()
+{
+ if ( !m_ignoreChanges )
+ {
+ SetModified( TRUE );
+ }
+}
+
+
+//---------------------------------------------------------------------------------------------------------------------------
+// CRegistrationPage::OnEnChangedPassword
+//---------------------------------------------------------------------------------------------------------------------------
+
+void CRegistrationPage::OnEnChangePassword()
+{
+ if ( !m_ignoreChanges )
+ {
+ SetModified( TRUE );
+ }
+}
+
+
+//---------------------------------------------------------------------------------------------------------------------------
+// CRegistrationPage::OnBnClickedAdvertiseServices
+//---------------------------------------------------------------------------------------------------------------------------
+
+void CRegistrationPage::OnBnClickedAdvertiseServices()
+{
+ if ( !m_ignoreChanges )
+ {
+ SetModified( TRUE );
+ }
+}
+
+
+//---------------------------------------------------------------------------------------------------------------------------
+// CRegistrationPage::SetModified
+//---------------------------------------------------------------------------------------------------------------------------
+
+void CRegistrationPage::SetModified( BOOL bChanged )
+{
+ m_modified = bChanged ? true : false;
+
+ CPropertyPage::SetModified( bChanged );
+}
+
+
+//---------------------------------------------------------------------------------------------------------------------------
+// CRegistrationPage::OnSetActive
+//---------------------------------------------------------------------------------------------------------------------------
+
+BOOL
+CRegistrationPage::OnSetActive()
+{
+ TCHAR name[kDNSServiceMaxDomainName + 1];
+ DWORD nameLen = ( kDNSServiceMaxDomainName + 1 ) * sizeof( TCHAR );
+ DWORD err;
+
+ BOOL b = CPropertyPage::OnSetActive();
+
+ m_ignoreChanges = true;
+ m_modified = FALSE;
+
+ if ( m_hostnameSetupKey )
+ {
+ err = RegQueryValueEx( m_hostnameSetupKey, L"", NULL, NULL, (LPBYTE) name, &nameLen );
+
+ if ( !err )
+ {
+ char hostnameUTF8[ 256 ];
+ char outDomain[ 256 ];
+ char outUsername[ 256 ];
+ char outPassword[ 256 ];
+ CString hostname = name;
+ CString username;
+ CString password;
+
+ m_hostnameControl.SetWindowText( hostname );
+
+ StringObjectToUTF8String( hostname, hostnameUTF8, sizeof( hostnameUTF8 ) );
+
+ if ( LsaGetSecret( hostnameUTF8, outDomain, sizeof( outDomain ) / sizeof( TCHAR ), outUsername, sizeof( outUsername ) / sizeof( TCHAR ), outPassword, sizeof( outPassword ) / sizeof( TCHAR ) ) )
+ {
+ username = outUsername;
+ m_usernameControl.SetWindowText( username );
+
+ password = outPassword;
+ m_passwordControl.SetWindowText( password );
+ }
+ }
+ }
+
+ m_advertiseServices.SetCheck( 0 );
+
+ if ( m_registrationSetupKey )
+ {
+ HKEY subKey = NULL;
+ DWORD dwSize;
+ DWORD enabled = 0;
+ TCHAR subKeyName[MAX_KEY_LENGTH];
+ DWORD cSubKeys = 0;
+ DWORD cbMaxSubKey;
+ DWORD cchMaxClass;
+ OSStatus err;
+
+ err = RegQueryInfoKey( m_registrationSetupKey, NULL, NULL, NULL, &cSubKeys, &cbMaxSubKey, &cchMaxClass, NULL, NULL, NULL, NULL, NULL );
+ if ( !err )
+ {
+ if ( cSubKeys > 0 )
+ {
+ dwSize = MAX_KEY_LENGTH;
+
+ err = RegEnumKeyEx( m_registrationSetupKey, 0, subKeyName, &dwSize, NULL, NULL, NULL, NULL );
+ if ( !err )
+ {
+ err = RegOpenKey( m_registrationSetupKey, subKeyName, &subKey );
+ if ( !err )
+ {
+ dwSize = sizeof( DWORD );
+ err = RegQueryValueEx( subKey, L"Enabled", NULL, NULL, (LPBYTE) &enabled, &dwSize );
+ if ( !err && enabled )
+ {
+ m_advertiseServices.SetCheck( enabled );
+ }
+
+ RegCloseKey( subKey );
+ subKey = NULL;
+ }
+ }
+ }
+ }
+ }
+
+ m_ignoreChanges = false;
+
+ return b;
+}
+
+
+//---------------------------------------------------------------------------------------------------------------------------
+// CRegistrationPage::OnOK
+//---------------------------------------------------------------------------------------------------------------------------
+
+void
+CRegistrationPage::OnOK()
+{
+ if ( m_modified )
+ {
+ Commit();
+ }
+}
+
+
+//---------------------------------------------------------------------------------------------------------------------------
+// CRegistrationPage::Commit
+//---------------------------------------------------------------------------------------------------------------------------
+
+void
+CRegistrationPage::Commit()
+{
+ CString hostname;
+ char hostnameUTF8[ 256 ];
+ CString username;
+ char usernameUTF8[ 256 ];
+ CString password;
+ char passwordUTF8[ 256 ];
+ DWORD enabled = 1;
+ BOOL secret = FALSE;
+ DWORD err;
+
+ m_hostnameControl.GetWindowText( hostname );
+ hostname.MakeLower();
+ hostname.TrimRight( '.' );
+ StringObjectToUTF8String( hostname, hostnameUTF8, sizeof( hostnameUTF8 ) );
+
+ m_usernameControl.GetWindowText( username );
+ m_passwordControl.GetWindowText( password );
+
+ if ( username.GetLength() && password.GetLength() )
+ {
+ StringObjectToUTF8String( username, usernameUTF8, sizeof( usernameUTF8 ) );
+ StringObjectToUTF8String( password, passwordUTF8, sizeof( passwordUTF8 ) );
+ secret = TRUE;
+ }
+
+ if ( m_hostnameSetupKey != NULL )
+ {
+ err = RegSetValueEx( m_hostnameSetupKey, L"", 0, REG_SZ, (LPBYTE) (LPCTSTR) hostname, ( hostname.GetLength() + 1 ) * sizeof( TCHAR ) );
+ require_noerr( err, exit );
+
+ err = RegSetValueEx( m_hostnameSetupKey, L"Enabled", 0, REG_DWORD, (LPBYTE) &enabled, sizeof( DWORD ) );
+ require_noerr( err, exit );
+
+ if ( secret )
+ {
+ LsaSetSecret( hostnameUTF8, usernameUTF8, passwordUTF8 );
+ }
+ }
+
+ if ( m_registrationSetupKey != NULL )
+ {
+ TCHAR subKeyName[MAX_KEY_LENGTH];
+ DWORD cSubKeys = 0;
+ DWORD cbMaxSubKey;
+ DWORD cchMaxClass;
+ DWORD dwSize;
+ int i;
+ OSStatus err = kNoErr;
+
+ // First, remove all the entries that are there
+
+ err = RegQueryInfoKey( m_registrationSetupKey, NULL, NULL, NULL, &cSubKeys, &cbMaxSubKey, &cchMaxClass, NULL, NULL, NULL, NULL, NULL );
+ if ( !err )
+ {
+ for ( i = 0; i < (int) cSubKeys; i++ )
+ {
+ dwSize = MAX_KEY_LENGTH;
+
+ err = RegEnumKeyEx( m_registrationSetupKey, 0, subKeyName, &dwSize, NULL, NULL, NULL, NULL );
+ require_noerr( err, exit );
+
+ err = RegDeleteKey( m_registrationSetupKey, subKeyName );
+ require_noerr( err, exit );
+ }
+ }
+
+ if ( m_advertiseServices.GetCheck() )
+ {
+ const char * domainUTF8;
+ CString domain;
+ char label[ 64 ];
+ HKEY subKey = NULL;
+
+ domainUTF8 = GetNextLabel( hostnameUTF8, label );
+ domain = domainUTF8;
+
+ err = RegCreateKeyEx( m_registrationSetupKey, domain, 0,
+ NULL, REG_OPTION_NON_VOLATILE, KEY_READ|KEY_WRITE|KEY_WOW64_32KEY, NULL, &subKey, NULL );
+ if ( !err )
+ {
+ err = RegSetValueEx( subKey, L"Enabled", 0, REG_DWORD, (LPBYTE) &enabled, sizeof( DWORD ) );
+ check_noerr( err );
+
+ RegCloseKey( subKey );
+ subKey = NULL;
+ }
+
+ if ( secret )
+ {
+ LsaSetSecret( domainUTF8, usernameUTF8, passwordUTF8 );
+ }
+ }
+ }
+
+exit:
+
+ return;
+}
projects / apple / mdnsresponder.git / blobdiff