]> git.saurik.com Git - apple/mdnsresponder.git/blobdiff - mDNSWindows/ControlPanel/ThirdPage.cpp
mDNSResponder-107.tar.gz
[apple/mdnsresponder.git] / mDNSWindows / ControlPanel / ThirdPage.cpp
diff --git a/mDNSWindows/ControlPanel/ThirdPage.cpp b/mDNSWindows/ControlPanel/ThirdPage.cpp
new file mode 100755 (executable)
index 0000000..bfb56bd
--- /dev/null
@@ -0,0 +1,456 @@
+/*
+ * Copyright (c) 2002-2004 Apple Computer, Inc. All rights reserved.
+ *
+ * @APPLE_LICENSE_HEADER_START@
+ * 
+ * This file contains Original Code and/or Modifications of Original Code
+ * as defined in and that are subject to the Apple Public Source License
+ * Version 2.0 (the 'License'). You may not use this file except in
+ * compliance with the License. Please obtain a copy of the License at
+ * http://www.opensource.apple.com/apsl/ and read it before using this
+ * file.
+ * 
+ * The Original Code and all software distributed under the License are
+ * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
+ * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
+ * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
+ * Please see the License for the specific language governing rights and
+ * limitations under the License.
+ * 
+ * @APPLE_LICENSE_HEADER_END@
+
+    Change History (most recent first):
+
+$Log: ThirdPage.cpp,v $
+Revision 1.3  2005/03/07 18:27:42  shersche
+<rdar://problem/4037940> Fix problem when ControlPanel commits changes to the browse domain list
+
+Revision 1.2  2005/03/03 19:55:22  shersche
+<rdar://problem/4034481> ControlPanel source code isn't saving CVS log info
+
+
+*/
+
+#include "ThirdPage.h"
+#include "resource.h"
+
+#include "ConfigPropertySheet.h"
+#include "SharedSecret.h"
+
+#include <WinServices.h>
+    
+#define MAX_KEY_LENGTH 255\r
+
+
+IMPLEMENT_DYNCREATE(CThirdPage, CPropertyPage)
+
+
+//---------------------------------------------------------------------------------------------------------------------------
+//     CThirdPage::CThirdPage
+//---------------------------------------------------------------------------------------------------------------------------
+
+CThirdPage::CThirdPage()
+:
+       CPropertyPage(CThirdPage::IDD)
+{
+       //{{AFX_DATA_INIT(CThirdPage)
+       //}}AFX_DATA_INIT
+
+       m_firstTime = true;
+}
+
+
+//---------------------------------------------------------------------------------------------------------------------------
+//     CThirdPage::~CThirdPage
+//---------------------------------------------------------------------------------------------------------------------------
+
+CThirdPage::~CThirdPage()
+{
+}
+
+
+//---------------------------------------------------------------------------------------------------------------------------
+//     CThirdPage::DoDataExchange
+//---------------------------------------------------------------------------------------------------------------------------
+
+void CThirdPage::DoDataExchange(CDataExchange* pDX)
+{
+       CPropertyPage::DoDataExchange(pDX);
+       //{{AFX_DATA_MAP(CThirdPage)
+       //}}AFX_DATA_MAP
+       DDX_Control(pDX, IDC_BROWSE_LIST, m_browseListCtrl);\r
+       DDX_Control(pDX, IDC_REMOVE_BROWSE_DOMAIN, m_removeButton);\r
+}
+
+BEGIN_MESSAGE_MAP(CThirdPage, CPropertyPage)
+       //{{AFX_MSG_MAP(CThirdPage)
+       //}}AFX_MSG_MAP
+       ON_BN_CLICKED(IDC_ADD_BROWSE_DOMAIN, OnBnClickedAddBrowseDomain)\r
+       ON_BN_CLICKED(IDC_REMOVE_BROWSE_DOMAIN, OnBnClickedRemoveBrowseDomain)\r
+       ON_NOTIFY(LVN_ITEMCHANGED, IDC_BROWSE_LIST, OnLvnItemchangedBrowseList)\r
+END_MESSAGE_MAP()
+
+
+//---------------------------------------------------------------------------------------------------------------------------
+//     CThirdPage::SetModified
+//---------------------------------------------------------------------------------------------------------------------------
+
+void CThirdPage::SetModified( BOOL bChanged )
+{
+       m_modified = bChanged;
+
+       CPropertyPage::SetModified( bChanged );
+}
+
+
+//---------------------------------------------------------------------------------------------------------------------------
+//     CThirdPage::OnSetActive
+//---------------------------------------------------------------------------------------------------------------------------
+
+BOOL
+CThirdPage::OnSetActive()
+{
+       CConfigPropertySheet    *       psheet;
+       HKEY                                            key = NULL;
+       HKEY                                            subKey = NULL;
+       DWORD                                           dwSize;
+       DWORD                                           enabled;
+       DWORD                                           err;
+       TCHAR                                           subKeyName[MAX_KEY_LENGTH];\r
+       DWORD                                           cSubKeys = 0;\r
+       DWORD                                           cbMaxSubKey;\r
+       DWORD                                           cchMaxClass;\r
+       int                                                     nIndex;\r
+    DWORD                                              i; 
+       BOOL                                            b = CPropertyPage::OnSetActive();
+
+       psheet = reinterpret_cast<CConfigPropertySheet*>(GetParent());
+       require_quiet( psheet, exit );
+       
+       m_modified = FALSE;
+
+       if ( m_firstTime )
+       {
+               m_browseListCtrl.SetExtendedStyle((m_browseListCtrl.GetStyle() & (~LVS_EX_GRIDLINES))|LVS_EX_CHECKBOXES | LVS_EX_FULLROWSELECT);
+
+               m_browseListCtrl.InsertColumn(0, L"", LVCFMT_LEFT, 20 );
+               m_browseListCtrl.InsertColumn(1, L"", LVCFMT_LEFT, 345);
+
+               m_firstTime = false;
+       }
+
+       m_initialized = false;
+
+       // Clear out what's there
+
+       m_browseListCtrl.DeleteAllItems();
+
+       // Now populate the browse domain box
+
+       err = RegCreateKey( HKEY_LOCAL_MACHINE, L"SYSTEM\\CurrentControlSet\\Services\\" kServiceName L"\\Parameters\\DynDNS\\Setup\\" kServiceDynDNSBrowseDomains, &key );
+       require_noerr( err, exit );
+
+       // Get information about this node\r
+\r
+    err = RegQueryInfoKey( key, NULL, NULL, NULL, &cSubKeys, &cbMaxSubKey, &cchMaxClass, NULL, NULL, NULL, NULL, NULL );       \r
+       require_noerr( err, exit );
+\r
+       for ( i = 0; i < cSubKeys; i++)\r
+       {       \r
+               dwSize = MAX_KEY_LENGTH;\r
+            \r
+               err = RegEnumKeyEx( key, i, subKeyName, &dwSize, NULL, NULL, NULL, NULL );\r
+               require_noerr( err, exit );\r
+\r
+               err = RegOpenKey( key, subKeyName, &subKey );\r
+               require_noerr( err, exit );\r
+\r
+               dwSize = sizeof( DWORD );\r
+               err = RegQueryValueEx( subKey, L"Enabled", NULL, NULL, (LPBYTE) &enabled, &dwSize );\r
+               require_noerr( err, exit );\r
+\r
+               nIndex = m_browseListCtrl.InsertItem( m_browseListCtrl.GetItemCount(), L"");
+               m_browseListCtrl.SetItemText( nIndex, 1, subKeyName );\r
+               m_browseListCtrl.SetCheck( nIndex, enabled );\r
+\r
+               RegCloseKey( subKey );\r
+               subKey = NULL;\r
+    }\r
+\r
+       m_browseListCtrl.SortItems( SortFunc, (DWORD_PTR) this );\r
+\r
+       m_removeButton.EnableWindow( FALSE );\r
+exit:
+
+       if ( subKey )
+       {
+               RegCloseKey( subKey );
+       }
+
+       if ( key )
+       {
+               RegCloseKey( key );
+       }
+
+       m_initialized = true;
+
+       return b;
+}
+\r
\r
+
+//---------------------------------------------------------------------------------------------------------------------------
+//     CThirdPage::OnOK
+//---------------------------------------------------------------------------------------------------------------------------
+
+void
+CThirdPage::OnOK()
+{
+       if ( m_modified )
+       {
+               Commit();
+       }
+}
+
+
+
+//---------------------------------------------------------------------------------------------------------------------------
+//     CThirdPage::Commit
+//---------------------------------------------------------------------------------------------------------------------------
+
+void
+CThirdPage::Commit()
+{
+       HKEY            key             = NULL;
+       HKEY            subKey  = NULL;
+       TCHAR           subKeyName[MAX_KEY_LENGTH];\r
+       DWORD           cSubKeys = 0;\r
+       DWORD           cbMaxSubKey;\r
+       DWORD           cchMaxClass;
+       DWORD           dwSize;
+       int                     i;
+       DWORD           err;
+
+       err = RegCreateKey( HKEY_LOCAL_MACHINE, L"SYSTEM\\CurrentControlSet\\Services\\" kServiceName L"\\Parameters\\DynDNS\\Setup\\" kServiceDynDNSBrowseDomains, &key );
+       require_noerr( err, exit );
+
+       // First, remove all the entries that are there\r
+\r
+    err = RegQueryInfoKey( key, NULL, NULL, NULL, &cSubKeys, &cbMaxSubKey, &cchMaxClass, NULL, NULL, NULL, NULL, NULL );       \r
+       require_noerr( err, exit );
+\r
+       for ( i = 0; i < (int) cSubKeys; i++ )\r
+       {       \r
+               dwSize = MAX_KEY_LENGTH;\r
+            \r
+               err = RegEnumKeyEx( key, 0, subKeyName, &dwSize, NULL, NULL, NULL, NULL );\r
+               require_noerr( err, exit );\r
+                       \r
+               err = RegDeleteKey( key, subKeyName );\r
+               require_noerr( err, exit );\r
+       }
+
+       // Now re-populate
+
+       for ( i = 0; i < m_browseListCtrl.GetItemCount(); i++ )
+       {
+               DWORD enabled = (DWORD) m_browseListCtrl.GetCheck( i );
+
+               err = RegCreateKey( key, m_browseListCtrl.GetItemText( i, 1 ), &subKey );
+               require_noerr( err, exit );
+
+               err = RegSetValueEx( subKey, L"Enabled", NULL, REG_DWORD, (LPBYTE) &enabled, sizeof( enabled ) );
+               require_noerr( err, exit );
+
+               RegCloseKey( subKey );
+               subKey = NULL;
+       }
+       
+exit:
+
+       if ( subKey )
+       {
+               RegCloseKey( subKey );
+       }
+
+       if ( key )
+       {
+               RegCloseKey( key );
+       }
+}
+
+
+
+//---------------------------------------------------------------------------------------------------------------------------
+//     CThirdPage::OnBnClickedAddBrowseDomain
+//---------------------------------------------------------------------------------------------------------------------------
+
+void
+CThirdPage::OnBnClickedAddBrowseDomain()
+{
+       CAddBrowseDomain dlg( GetParent() );
+
+       if ( ( dlg.DoModal() == IDOK ) && ( dlg.m_text.GetLength() > 0 ) )
+       {
+               int nIndex;
+
+               nIndex = m_browseListCtrl.InsertItem( m_browseListCtrl.GetItemCount(), L"");
+               m_browseListCtrl.SetItemText( nIndex, 1, dlg.m_text );\r
+               m_browseListCtrl.SetCheck( nIndex, 1 );\r
+\r
+               m_browseListCtrl.SortItems( SortFunc, (DWORD_PTR) this );
+
+               m_browseListCtrl.Invalidate();
+
+               SetModified( TRUE );
+       }
+}
+
+
+//---------------------------------------------------------------------------------------------------------------------------
+//     CThirdPage::OnBnClickedRemoveBrowseDomain
+//---------------------------------------------------------------------------------------------------------------------------
+
+void
+CThirdPage::OnBnClickedRemoveBrowseDomain()
+{
+       UINT    selectedCount = m_browseListCtrl.GetSelectedCount();\r
+       int             nItem = -1;\r
+       UINT    i;\r
+\r
+       // Update all of the selected items.\r
+\r
+       for ( i = 0; i < selectedCount; i++ )\r
+       {\r
+               nItem = m_browseListCtrl.GetNextItem( -1, LVNI_SELECTED );\r
+               check( nItem != -1 );\r
+\r
+               m_browseListCtrl.DeleteItem( nItem );\r
+\r
+               SetModified( TRUE );\r
+       }\r
+\r
+       m_removeButton.EnableWindow( FALSE );\r
+}
+
+
+void\r
+CThirdPage::OnLvnItemchangedBrowseList(NMHDR *pNMHDR, LRESULT *pResult)\r
+{\r
+       if ( m_browseListCtrl.GetSelectedCount() )\r
+       {\r
+               m_removeButton.EnableWindow( TRUE );\r
+       }\r
+\r
+       if ( m_initialized )\r
+       {\r
+               NM_LISTVIEW * pNMListView = (NM_LISTVIEW*)pNMHDR; \r
+        \r
+               BOOL bPrevState = (BOOL) ( ( ( pNMListView->uOldState & LVIS_STATEIMAGEMASK ) >> 12 ) - 1 ); \r
+        \r
+               if ( bPrevState < 0 )\r
+               {\r
+                       bPrevState = 0;\r
+               }\r
+\r
+\r
+               BOOL bChecked = ( BOOL ) ( ( ( pNMListView->uNewState & LVIS_STATEIMAGEMASK ) >> 12) - 1 ); \r
+        \r
+               if ( bChecked < 0 )\r
+               {\r
+                       bChecked = 0;\r
+               }\r
+\r
+               if ( bPrevState != bChecked )\r
+               {\r
+                       SetModified( TRUE );\r
+               }\r
+       }\r
+\r
+       *pResult = 0;\r
+}
+
+
+
+int CALLBACK \r
+CThirdPage::SortFunc(LPARAM lParam1, LPARAM lParam2, LPARAM lParamSort)\r
+{\r
+       CString str1;\r
+       CString str2;\r
+       int             ret = 0;\r
+\r
+       CThirdPage * self = reinterpret_cast<CThirdPage*>( lParamSort );
+       require_quiet( self, exit );
+\r
+       str1 = self->m_browseListCtrl.GetItemText( (int) lParam1, 1 );\r
+       str2 = self->m_browseListCtrl.GetItemText( (int) lParam2, 1 );\r
+\r
+       ret = str1.Compare( str2 );\r
+\r
+exit:\r
+\r
+       return ret;\r
+}
+\r
+\r
+// CAddBrowseDomain dialog\r
+\r
+IMPLEMENT_DYNAMIC(CAddBrowseDomain, CDialog)\r
+CAddBrowseDomain::CAddBrowseDomain(CWnd* pParent /*=NULL*/)\r
+       : CDialog(CAddBrowseDomain::IDD, pParent)\r
+{\r
+}\r
+\r
+CAddBrowseDomain::~CAddBrowseDomain()\r
+{\r
+}\r
+\r
+void CAddBrowseDomain::DoDataExchange(CDataExchange* pDX)\r
+{\r
+       CDialog::DoDataExchange(pDX);\r
+       DDX_Control(pDX, IDC_COMBO1, m_comboBox);\r
+}\r
+\r
+\r
+BOOL\r
+CAddBrowseDomain::OnInitDialog()\r
+{\r
+       CConfigPropertySheet    *       psheet;\r
+       CConfigPropertySheet::StringList::iterator              it;\r
+       \r
+       BOOL b = CDialog::OnInitDialog();\r
+\r
+       psheet = reinterpret_cast<CConfigPropertySheet*>(GetParent());
+       require_quiet( psheet, exit );\r
+\r
+       for ( it = psheet->m_browseDomains.begin(); it != psheet->m_browseDomains.end(); it++ )\r
+       {\r
+               CString text = *it;\r
+\r
+               if ( m_comboBox.FindStringExact( -1, *it ) == CB_ERR )
+               {
+                       m_comboBox.AddString( *it );
+               }\r
+       }\r
+\r
+exit:\r
+\r
+       return b;\r
+}\r
+\r
+\r
+void\r
+CAddBrowseDomain::OnOK()\r
+{\r
+       m_comboBox.GetWindowText( m_text );\r
+\r
+       CDialog::OnOK();\r
+}\r
+\r
+\r
+\r
+BEGIN_MESSAGE_MAP(CAddBrowseDomain, CDialog)\r
+END_MESSAGE_MAP()\r
+\r