archive
/
windows-server-2003
mirror of https://github.com/selfrender/Windows-Server-2003
2
0
Fork 0
Code Issues Projects Releases Wiki Activity
Leaked source code of windows server 2003
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
4 Commits
1 Branch
0 Tags
1.5 GiB
 
 
 
 
 
 
Branch: master
Branches Tags
${ item.name }
Create tag ${ searchTerm }
Create branch ${ searchTerm }
from 'master'
${ noResults }
windows-server-2003/admin/snapin/dnsmgr/nspage.cpp

1845 lines
50 KiB
Raw Permalink Blame History

//+-------------------------------------------------------------------------
//
// Microsoft Windows
//
// Copyright (C) Microsoft Corporation, 1998 - 1999
//
// File: nspage.cpp
//
//--------------------------------------------------------------------------
#include "preDNSsn.h"
#include <SnapBase.h>
#include "resource.h"
#include "dnsutil.h"
#include "DNSSnap.h"
#include "snapdata.h"
#include "server.h"
#include "domain.h"
#include "record.h"
#include "zone.h"
#include "nspage.h"
#include "uiutil.h"
#include "ipeditor.h"
#include "browser.h"
///////////////////////////////////////////////////////////////////////////////
// CDNS_NS_RecordDialog
class CDNS_NS_RecordDialog : public CPropertyPage
{
public:
CDNS_NS_RecordDialog(CDNSNameServersPropertyPage* pNSPage, BOOL bNew);
CDNS_NS_RecordDialog(CDNSNameServersWizardPage* pNSWiz, BOOL bNew);
~CDNS_NS_RecordDialog();
INT_PTR DoModalSheet(HWND parent = 0);
// data
BOOL m_bDirty;
CDNSRecordNodeEditInfo* m_pNSInfo;
protected:
virtual int OnCreate(LPCREATESTRUCT lpCreateStruct);
virtual BOOL OnInitDialog();
virtual BOOL OnApply();
afx_msg void OnServerNameChange();
afx_msg void OnBrowse();
afx_msg void OnQuery();
BOOL OnHelpInfo(HELPINFO* pHelpInfo);
DECLARE_MESSAGE_MAP()
private:
class CARecordAddressesIPEditor : public CIPEditor
{
public:
CARecordAddressesIPEditor(CDNS_NS_RecordDialog* pNSRecordDialog)
{ m_pNSRecordDialog = pNSRecordDialog;}
void SetIpAddresses(CDNSRecordNodeEditInfo* pNSInfo);
BOOL GetIpAddresses(CDNSRecordNodeEditInfo* pNSInfo);
protected:
virtual void OnChangeData();
private:
CDNS_NS_RecordDialog* m_pNSRecordDialog;
};
CARecordAddressesIPEditor m_RecordAddressesEditor;
BOOL m_bNew;
CDNSNameServersPropertyPage* m_pNSPage;
CDNSNameServersWizardPage* m_pNSWiz;
CPropertyPageBase* GetPage()
{
if (m_pNSPage != NULL)
{
return m_pNSPage;
}
return m_pNSWiz;
}
HWND m_hWndOKButton;
HWND m_hWndQueryButton;
CDNSTTLControl* GetTTLCtrl() { return (CDNSTTLControl*)GetDlgItem(IDC_TTLEDIT);}
CEdit* GetServerEdit() { return (CEdit*)GetDlgItem(IDC_SERVER_NAME_EDIT); }
CDNS_NS_Record* GetNSRecord()
{
ASSERT( m_pNSInfo != NULL);
ASSERT( m_pNSInfo->m_pRecord->m_wType == DNS_TYPE_NS);
return (CDNS_NS_Record*)m_pNSInfo->m_pRecord;
}
void GetNSServerName(CString& szNameNode);
void SyncUIButtons();
void EnableTTLCtrl(BOOL bShow);
friend class CARecordAddressesIPEditor;
};
INT_PTR CDNS_NS_RecordDialog::DoModalSheet(HWND parent)
{
/* NOTE : The first call to this may cause a first-chance exception. Excerpt from MSDN January 2000.
Note The first time a property page is created from its corresponding dialog resource,
it may cause a first-chance exception. This is a result of the property page changing
the style of the dialog resource to the required style prior to creating the page. Because
resources are generally read-only, this causes an exception. The exception is handled by
the system, and a copy of the modified resource is made automatically by the system. The
first-chance exception can thus be ignored.
Since this exception must be handled by the operating system, do not wrap calls to
CPropertySheet::DoModal with a C++ try/catch block in which the catch handles all exceptions,
for example, catch (...). This will handle the exception intended for the operating system,
causing unpredictable behavior. Using C++ exception handling with specific exception types
or using structured exception handling where the Access Violation exception is passed through
to the operating system is safe, however.
*/
AFX_MANAGE_STATE(AfxGetStaticModuleState());
CThemeContextActivator activator;
CString szTitle;
szTitle.LoadString(m_bNew ? IDS_NEW_RECORD_TITLE : IDS_EDIT_RECORD_TITLE);
CPropertySheet hostSheet;
hostSheet.m_psh.dwFlags |= PSH_NOAPPLYNOW;
hostSheet.m_psh.pszCaption = (LPCTSTR)szTitle;
hostSheet.m_psh.hwndParent = parent;
hostSheet.AddPage(this);
INT_PTR iRes = hostSheet.DoModal();
GetPage()->GetHolder()->PopDialogHWnd();
return iRes;
}
void CDNS_NS_RecordDialog::CARecordAddressesIPEditor::OnChangeData()
{
m_pNSRecordDialog->SyncUIButtons();
}
void CDNS_NS_RecordDialog::CARecordAddressesIPEditor::
SetIpAddresses(CDNSRecordNodeEditInfo* pNSInfo)
{
Clear();
ASSERT(pNSInfo != NULL);
INT_PTR nArraySize = pNSInfo->m_pEditInfoList->GetCount();
if (nArraySize == 0)
return;
DWORD* pArr = (DWORD*)malloc(nArraySize*sizeof(DWORD));
if (!pArr)
{
return;
}
int k=0;
POSITION pos;
for( pos = pNSInfo->m_pEditInfoList->GetHeadPosition(); pos != NULL; )
{
CDNSRecordNodeEditInfo* pARecordInfo = pNSInfo->m_pEditInfoList->GetNext(pos);
ASSERT(pARecordInfo != NULL);
if (pARecordInfo->m_action != CDNSRecordNodeEditInfo::remove)
{
ASSERT(pARecordInfo->m_pRecord != NULL);
ASSERT(pARecordInfo->m_pRecord->m_wType == DNS_TYPE_A);
CDNS_A_Record* pARecord = (CDNS_A_Record*)pARecordInfo->m_pRecord;
pArr[k++] = pARecord->m_ipAddress;
}
}
AddAddresses(pArr, k);
if (pArr)
{
free(pArr);
}
}
BOOL CDNS_NS_RecordDialog::CARecordAddressesIPEditor::
GetIpAddresses(CDNSRecordNodeEditInfo* pNSInfo)
{
BOOL bDirty = FALSE;
int nArraySize = GetCount();
// if the count of IP addresses is zero,
// we mark the NS record as slated for removal
if (nArraySize == 0)
pNSInfo->m_action = CDNSRecordNodeEditInfo::remove;
// read the IP addresses from IP editor, if any
DWORD* pArr = (nArraySize >0) ? (DWORD*)malloc(nArraySize*sizeof(DWORD)) : NULL;
if (!pArr)
{
return FALSE;
}
int nFilled = 0;
if (nArraySize > 0)
GetAddresses(pArr, nArraySize, &nFilled);
ASSERT(nFilled == nArraySize);
ASSERT(pNSInfo->m_pRecord != NULL);
ASSERT(pNSInfo->m_pRecord->GetType() == DNS_TYPE_NS);
CDNS_NS_Record* pNSRecord = (CDNS_NS_Record*)pNSInfo->m_pRecord;
CDNSRecordNodeEditInfoList* pNSInfoList = pNSInfo->m_pEditInfoList;
CDNSRecordNodeEditInfoList NSInfoRemoveList;
POSITION pos;
// walk the list of A records, to mark the ones to be deleted,
// if nArraySize == 0, all of them well be marked for removal
for( pos = pNSInfoList->GetHeadPosition(); pos != NULL; )
{
CDNSRecordNodeEditInfo* pCurrentInfo = pNSInfoList->GetNext(pos);
ASSERT(pCurrentInfo->m_pRecordNode != NULL);
ASSERT(pCurrentInfo->m_pRecord != NULL);
ASSERT(pCurrentInfo->m_pRecord->GetType() == DNS_TYPE_A);
CDNS_A_Record* pARecord = (CDNS_A_Record*)pCurrentInfo->m_pRecord;
BOOL bFound = FALSE;
for (int k=0; k<nArraySize; k++)
{
if (pARecord->m_ipAddress == pArr[k])
{
bFound = TRUE;
break;
}
}
if (!bFound)
{
bDirty = TRUE;
if (pCurrentInfo->m_bExisting)
{
pCurrentInfo->m_action = CDNSRecordNodeEditInfo::remove; // mark as deleted
}
else
{
NSInfoRemoveList.AddTail(pCurrentInfo);
}
}
} // for
// This gives NSInfoRemoveList ownership of all memory management for all nodes
// removed from the pNSInfoList
POSITION listPos = NSInfoRemoveList.GetHeadPosition();
while (listPos != NULL)
{
CDNSRecordNodeEditInfo* pCurrentInfo = NSInfoRemoveList.GetNext(listPos);
ASSERT(pCurrentInfo != NULL);
POSITION removePos = pNSInfoList->Find(pCurrentInfo);
pNSInfoList->RemoveAt(removePos);
}
// Remove and delete all nodes that were removed from the pNSInfoList
NSInfoRemoveList.RemoveAllNodes();
// walk the list of addresses, to look for matching A records to add
// if nArraySize == 0, loop skipped, nothing to add
for (int k=0; k<nArraySize; k++)
{
BOOL bFound = FALSE;
for( pos = pNSInfoList->GetHeadPosition(); pos != NULL; )
{
CDNSRecordNodeEditInfo* pCurrentInfo = pNSInfoList->GetNext(pos);
ASSERT(pCurrentInfo->m_pRecordNode != NULL);
ASSERT(pCurrentInfo->m_pRecord != NULL);
ASSERT(pCurrentInfo->m_pRecord->GetType() == DNS_TYPE_A);
CDNS_A_Record* pARecord = (CDNS_A_Record*)pCurrentInfo->m_pRecord;
if (pARecord->m_ipAddress == pArr[k])
{
bFound = TRUE;
if (pCurrentInfo->m_action == CDNSRecordNodeEditInfo::remove) // we got it already, resuscitate it
{
bDirty = TRUE;
if(pCurrentInfo->m_bExisting)
pCurrentInfo->m_action = CDNSRecordNodeEditInfo::edit;
else
pCurrentInfo->m_action = CDNSRecordNodeEditInfo::add;
}
break;
}
}
if (!bFound)
{
// A record not found, need to create one
CDNSRecordNodeEditInfo* pNewInfo = new CDNSRecordNodeEditInfo;
if (pNewInfo)
{
CDNS_A_RecordNode* pARecordNode = new CDNS_A_RecordNode;
if (pARecordNode)
{
pNewInfo->CreateFromNewRecord(pARecordNode);
CDNS_A_Record* pARecord = (CDNS_A_Record*)pNewInfo->m_pRecord;
pARecord->m_ipAddress = pArr[k];
pNewInfo->m_pRecordNode->m_bAtTheNode = FALSE;
pNewInfo->m_pRecordNode->SetRecordName(pNSRecord->m_szNameNode, FALSE /*bAtTheNode*/);
// inherit the TTL of the NS record
pNewInfo->m_pRecord->m_dwTtlSeconds = pNSInfo->m_pRecord->m_dwTtlSeconds;
pNSInfoList->AddTail(pNewInfo);
bDirty = TRUE;
}
else
{
delete pNewInfo;
}
}
}
} // for
if (pArr)
{
free(pArr);
}
return bDirty;
}
BEGIN_MESSAGE_MAP(CDNS_NS_RecordDialog, CPropertyPage)
ON_WM_CREATE()
ON_EN_CHANGE(IDC_SERVER_NAME_EDIT, OnServerNameChange)
ON_BN_CLICKED(IDC_BROWSE_BUTTON, OnBrowse)
ON_BN_CLICKED(IDC_QUERY_BUTTON, OnQuery)
ON_WM_HELPINFO()
END_MESSAGE_MAP()
CDNS_NS_RecordDialog::CDNS_NS_RecordDialog(CDNSNameServersPropertyPage* pNSPage, BOOL bNew)
: CPropertyPage(IDD_RR_NS_EDIT),
m_RecordAddressesEditor(this)
{
ASSERT(pNSPage != NULL);
m_pNSPage = pNSPage;
m_pNSWiz = NULL;
m_bNew = bNew;
m_bDirty = FALSE;
m_pNSInfo = NULL;
m_hWndOKButton = m_hWndQueryButton = NULL;
}
CDNS_NS_RecordDialog::CDNS_NS_RecordDialog(CDNSNameServersWizardPage* pNSWiz, BOOL bNew)
: CPropertyPage(IDD_RR_NS_EDIT),
m_RecordAddressesEditor(this)
{
ASSERT(pNSWiz != NULL);
m_pNSPage = NULL;
m_pNSWiz = pNSWiz;
m_bNew = bNew;
m_bDirty = FALSE;
m_pNSInfo = NULL;
m_hWndOKButton = m_hWndQueryButton = NULL;
}
CDNS_NS_RecordDialog::~CDNS_NS_RecordDialog()
{
}
int CDNS_NS_RecordDialog::OnCreate(LPCREATESTRUCT lpCreateStruct)
{
int res = CPropertyPage::OnCreate(lpCreateStruct);
DWORD dwStyle = ::GetWindowLong(::GetParent(GetSafeHwnd()), GWL_EXSTYLE);
dwStyle |= WS_EX_CONTEXTHELP; // force the [?] button
::SetWindowLong(::GetParent(GetSafeHwnd()), GWL_EXSTYLE, dwStyle);
return res;
}
BOOL CDNS_NS_RecordDialog::OnHelpInfo(HELPINFO* pHelpInfo)
{
CComponentDataObject* pComponentData = GetPage()->GetHolder()->GetComponentData();
ASSERT(pComponentData != NULL);
pComponentData->OnDialogContextHelp(m_nIDHelp, pHelpInfo);
return TRUE;
}
void CDNS_NS_RecordDialog::EnableTTLCtrl(BOOL bShow)
{
CDNSTTLControl* pCtrl = GetTTLCtrl();
ASSERT(pCtrl != NULL);
pCtrl->EnableWindow(bShow);
pCtrl->ShowWindow(bShow);
CWnd* pWnd = GetDlgItem(IDC_STATIC_TTL);
ASSERT(pWnd != NULL);
pWnd->EnableWindow(bShow);
pWnd->ShowWindow(bShow);
CWnd* pLabelWnd = GetDlgItem(IDC_TTL_LABEL);
ASSERT(pLabelWnd != NULL);
pLabelWnd->EnableWindow(bShow);
pLabelWnd->ShowWindow(bShow);
}
BOOL CDNS_NS_RecordDialog::OnInitDialog()
{
CPropertyPage::OnInitDialog();
ASSERT(m_pNSInfo != NULL);
ASSERT(m_pNSInfo->m_pRecord != NULL);
ASSERT(m_hWnd != NULL);
ASSERT(::IsWindow(m_hWnd));
HWND hParent = ::GetParent(m_hWnd);
ASSERT(hParent);
GetPage()->GetHolder()->PushDialogHWnd(hParent);
//
// OK button on the sheet
//
m_hWndOKButton = ::GetDlgItem(hParent, IDOK);
ASSERT(::IsWindow(m_hWndOKButton));
//
// query button handle
//
m_hWndQueryButton = :: GetDlgItem(m_hWnd, IDC_QUERY_BUTTON);
ASSERT(::IsWindow(m_hWndQueryButton));
//
// initialize IP editor
//
VERIFY(m_RecordAddressesEditor.Initialize(this,
GetParent(),
IDC_BUTTON_UP,
IDC_BUTTON_DOWN,
IDC_BUTTON_ADD,
IDC_BUTTON_REMOVE,
IDC_IPEDIT,
IDC_LIST));
//
// Load Data in the UI
//
m_RecordAddressesEditor.SetIpAddresses(m_pNSInfo);
GetServerEdit()->LimitText(MAX_DNS_NAME_LEN);
GetServerEdit()->SetWindowText(GetNSRecord()->m_szNameNode);
GetTTLCtrl()->SetTTL(m_pNSInfo->m_pRecord->m_dwTtlSeconds);
// need to decide if we want to show the TTL control
CDNSRootData* pRootData = (CDNSRootData*)GetPage()->GetHolder()->GetComponentData()->GetRootData();
ASSERT(pRootData != NULL);
BOOL bEnableTTLCtrl;
if (m_pNSPage != NULL)
{
bEnableTTLCtrl = m_pNSPage->HasMeaningfulTTL() && pRootData->IsAdvancedView();
}
else
{
bEnableTTLCtrl = m_pNSWiz->HasMeaningfulTTL() && pRootData->IsAdvancedView();
}
EnableTTLCtrl(bEnableTTLCtrl);
return TRUE; // return TRUE unless you set the focus to a control
}
void CDNS_NS_RecordDialog::GetNSServerName(CString& szNameNode)
{
GetServerEdit()->GetWindowText(szNameNode);
szNameNode.TrimLeft();
szNameNode.TrimRight();
if (szNameNode[szNameNode.GetLength()-1] != TEXT('.'))
szNameNode += TEXT('.');
}
void CDNS_NS_RecordDialog::SyncUIButtons()
{
CString szServerName;
GetServerEdit()->GetWindowText(szServerName);
szServerName.TrimLeft();
szServerName.TrimRight();
DWORD dwNameChecking = 0;
if (m_pNSWiz)
{
dwNameChecking = m_pNSWiz->GetDomainNode()->GetServerNode()->GetNameCheckFlag();
}
else if (m_pNSPage)
{
dwNameChecking = m_pNSPage->GetDomainNode()->GetServerNode()->GetNameCheckFlag();
}
//
// Enable OK button if it is a valid name
//
BOOL bIsValidName = (0 == ValidateDnsNameAgainstServerFlags(szServerName,
DnsNameDomain,
dwNameChecking));
::EnableWindow(m_hWndOKButton, bIsValidName &&
m_RecordAddressesEditor. GetCount() > 0);
::EnableWindow(m_hWndQueryButton, bIsValidName);
}
BOOL CDNS_NS_RecordDialog::OnApply()
{
ASSERT(m_pNSInfo->m_pRecord != NULL);
CString szNameNode;
GetNSServerName(szNameNode);
// Make sure the new name is an FQDN or else we could have
// a broken delegation
CString szNSName;
GetServerEdit()->GetWindowText(szNSName);
if (szNSName.Find(L".") == -1)
{
DNSMessageBox(IDS_MSG_RECORD_WARNING_NS_NOT_FQDN, MB_OK);
return false;
}
// compare (case insensitive) with old name to see if it changed,
// NOTICE: CDNSDomainNode::UpdateARecordsOfNSInfoHelper() will then
// take care of regenerating the list of A records
m_bDirty = _wcsicmp((LPCWSTR)szNameNode,
(LPCWSTR)GetNSRecord()->m_szNameNode);
if (m_bDirty)
GetNSRecord()->m_szNameNode = szNameNode;
// update list of IP addresses
if (m_bNew)
{
// the dialog is used to create a new entry
ASSERT(!m_pNSInfo->m_bExisting);
m_pNSInfo->m_action = CDNSRecordNodeEditInfo::add;
}
else
{
// the dialog is used to edit
if (m_pNSInfo->m_bExisting)
{
// an existing entry
m_pNSInfo->m_action = CDNSRecordNodeEditInfo::edit;
}
else
{
// a newly created entry, edited before committing
m_pNSInfo->m_action = CDNSRecordNodeEditInfo::add;
}
}
// this call migth mark the info as remove, if no IP addresses are found
if (m_RecordAddressesEditor.GetIpAddresses(m_pNSInfo))
m_bDirty = TRUE;
DWORD dwTTL;
GetTTLCtrl()->GetTTL(&dwTTL);
if (m_pNSInfo->m_pRecord->m_dwTtlSeconds != dwTTL)
{
m_bDirty = TRUE;
m_pNSInfo->m_pRecord->m_dwTtlSeconds = dwTTL;
// Need to change the TTL on all associated A records
CDNSRecordNodeEditInfoList* pNSInfoList = m_pNSInfo->m_pEditInfoList;
for(POSITION pos = pNSInfoList->GetHeadPosition(); pos != NULL; )
{
CDNSRecordNodeEditInfo* pCurrentInfo = pNSInfoList->GetNext(pos);
ASSERT(pCurrentInfo->m_pRecordNode != NULL);
ASSERT(pCurrentInfo->m_pRecord != NULL);
// if slated for removal, don't bother to change
if (pCurrentInfo->m_action != CDNSRecordNodeEditInfo::remove)
{
pCurrentInfo->m_pRecord->m_dwTtlSeconds = m_pNSInfo->m_pRecord->m_dwTtlSeconds;
// if already marked "add" or "edit", leave as is,
// but if unchanged, need to mark as "edit"
if (pCurrentInfo->m_action == CDNSRecordNodeEditInfo::unchanged)
pCurrentInfo->m_action = CDNSRecordNodeEditInfo::edit;
}
}
}
if (m_pNSInfo->m_action == CDNSRecordNodeEditInfo::remove)
{
if (IDNO == DNSMessageBox(IDS_MSG_RECORD_WARNING_NS_NO_IP, MB_YESNO))
return true;
}
return true;
}
void CDNS_NS_RecordDialog::OnServerNameChange()
{
SyncUIButtons();
}
void CDNS_NS_RecordDialog::OnBrowse()
{
AFX_MANAGE_STATE(AfxGetStaticModuleState());
FIX_THREAD_STATE_MFC_BUG();
CThemeContextActivator activator;
CComponentDataObject* pComponentDataObject = GetPage()->GetHolder()->GetComponentData();
CDNSBrowserDlg dlg(pComponentDataObject, GetPage()->GetHolder(), RECORD_A_AND_CNAME);
if (IDOK == dlg.DoModal())
{
GetServerEdit()->SetWindowText(dlg.GetSelectionString());
//
// if it is an A record, add the IP address to the IP editor
//
CDNSRecordNodeBase* pRecordNode = reinterpret_cast<CDNSRecordNodeBase*>(dlg.GetSelection());
if ((pRecordNode != NULL) && (pRecordNode->GetType() == DNS_TYPE_A))
{
DWORD ip = ((CDNS_A_RecordNode*)pRecordNode)->GetIPAddress();
m_RecordAddressesEditor.AddAddresses(&ip,1);
}
}
}
void CDNS_NS_RecordDialog::OnQuery()
{
CWaitCursor cursor;
CDNSRecordNodeEditInfo tempNSInfo; // test
CString szNameNode;
GetNSServerName(szNameNode);
CDNSServerNode* pServerNode;
if (m_pNSPage != NULL)
{
pServerNode = m_pNSPage->GetDomainNode()->GetServerNode();
}
else
{
pServerNode = m_pNSWiz->GetDomainNode()->GetServerNode();
}
LPCWSTR lpszZoneName = NULL;
CDNSZoneNode* pZoneNode = NULL;
if (m_pNSPage != NULL)
{
pZoneNode = m_pNSPage->GetDomainNode()->GetZoneNode();
}
else
{
pZoneNode = m_pNSWiz->GetDomainNode()->GetZoneNode();
}
if (pZoneNode != NULL)
{
lpszZoneName = pZoneNode->GetFullName();
}
ASSERT(pServerNode != NULL);
CComponentDataObject* pComponentDataObject =
GetPage()->GetHolder()->GetComponentData();
CDNSRootData* pRootData = (CDNSRootData*)pComponentDataObject->GetRootData();
ASSERT(pRootData != NULL);
DWORD cAddrCount;
PIP_ADDRESS pipAddrs;
pServerNode->GetListenAddressesInfo(&cAddrCount, &pipAddrs);
if (cAddrCount == 0)
{
// listening on all addresses
pServerNode->GetServerAddressesInfo(&cAddrCount, &pipAddrs);
}
CDNSDomainNode::FindARecordsFromNSInfo(pServerNode->GetRPCName(),
pServerNode->GetVersion(),
cAddrCount, pipAddrs,
lpszZoneName,
szNameNode,
tempNSInfo.m_pEditInfoList,
pRootData->IsAdvancedView());
if (tempNSInfo.m_pEditInfoList->GetCount() > 0)
{
// update the list only if we have valid data
m_RecordAddressesEditor.SetIpAddresses(&tempNSInfo);
}
else
{
DNSMessageBox(IDS_MSG_RECORD_NS_RESOLVE_IP, MB_OK | MB_ICONERROR);
}
}
///////////////////////////////////////////////////////////////////////////////
// CNSListCtrl
BEGIN_MESSAGE_MAP(CNSListCtrl, CListCtrl)
END_MESSAGE_MAP()
void CNSListCtrl::Initialize()
{
// get size of control to help set the column widths
CRect controlRect;
GetClientRect(controlRect);
// get width of control, width of potential scrollbar, width needed for sub-item
// string
int controlWidth = controlRect.Width();
int scrollThumbWidth = ::GetSystemMetrics(SM_CXHTHUMB);
// clean net width
int nNetControlWidth = controlWidth - scrollThumbWidth - 12 * ::GetSystemMetrics(SM_CXBORDER);
// fields widths
int nWidth1 = 3 * nNetControlWidth / 4;
int nWidth2 = nNetControlWidth - nWidth1;
//
// set up columns
CString szHeaders;
{
AFX_MANAGE_STATE(AfxGetStaticModuleState());
szHeaders.LoadString(IDS_NSPAGE_LISTVIEW_HEADERS);
}
ASSERT(!szHeaders.IsEmpty());
LPWSTR lpszArr[2];
UINT n;
ParseNewLineSeparatedString(szHeaders.GetBuffer(1), lpszArr, &n);
szHeaders.ReleaseBuffer();
ASSERT(n == 2);
InsertColumn(1, lpszArr[0], LVCFMT_LEFT, nWidth1, 1);
InsertColumn(2, lpszArr[1], LVCFMT_LEFT, nWidth2 + 28, 2);
}
int CNSListCtrl::GetSelection()
{
return GetNextItem(-1, LVIS_SELECTED);
}
void CNSListCtrl::SetSelection(int nSel)
{
VERIFY(SetItemState(nSel, LVIS_SELECTED | LVIS_FOCUSED, LVIS_SELECTED | LVIS_FOCUSED));
/*
VERIFY(SetItem(nSel, // nItem
0, // nSubItem
LVIF_STATE, // nMask
NULL, // lpszItem
0, // nImage
LVIS_SELECTED | LVIS_FOCUSED, // nState
LVIS_SELECTED | LVIS_FOCUSED, // nStateMask
NULL // lParam
));
*/
}
void CNSListCtrl::UpdateNSRecordEntry(int nItemIndex)
{
CDNSRecordNodeEditInfo* pNSInfo = (CDNSRecordNodeEditInfo*)GetItemData(nItemIndex);
VERIFY(SetItem(nItemIndex, // nItem
0, // nSubItem
LVIF_TEXT, // nMask
((CDNS_NS_Record*)pNSInfo->m_pRecord)->m_szNameNode, // lpszItem
0, // nImage
0, // nState
0, // nStateMask
NULL // lParam
));
CString szTemp;
GetIPAddressString(pNSInfo, szTemp);
SetItemText(nItemIndex, 1, szTemp);
}
CDNSRecordNodeEditInfo* CNSListCtrl::GetSelectionEditInfo()
{
int nSel = GetSelection();
if (nSel == -1)
return NULL; // no selection
return (CDNSRecordNodeEditInfo*)GetItemData(nSel);
}
void CNSListCtrl::BuildIPAddrDisplayString(CDNSRecordNodeEditInfo* pNSInfo, CString& szDisplayData)
{
USES_CONVERSION;
// need to chain the IP addresses in a single string
CString szTemp;
szTemp.GetBuffer(20); // length of an IP string
szTemp.ReleaseBuffer();
szDisplayData.GetBuffer(static_cast<int>(20*pNSInfo->m_pEditInfoList->GetCount()));
szDisplayData.ReleaseBuffer();
POSITION pos;
for( pos = pNSInfo->m_pEditInfoList->GetHeadPosition(); pos != NULL; )
{
CDNSRecordNodeEditInfo* pARecordInfo = pNSInfo->m_pEditInfoList->GetNext(pos);
if (pARecordInfo->m_action != CDNSRecordNodeEditInfo::remove)
{
ASSERT(pARecordInfo != NULL);
ASSERT(pARecordInfo->m_pRecord != NULL);
ASSERT(pARecordInfo->m_pRecord->m_wType == DNS_TYPE_A);
CDNS_A_Record* pARecord = (CDNS_A_Record*)pARecordInfo->m_pRecord;
szDisplayData += _T("[");
FormatIpAddress(szTemp, pARecord->m_ipAddress);
szDisplayData += szTemp;
if (pARecordInfo->m_bFromDnsQuery)
{
szDisplayData += _T("*");
}
szDisplayData += _T("] ");
}
}
}
BOOL CNSListCtrl::InsertNSRecordEntry(CDNSRecordNodeEditInfo* pNSInfo, int nItemIndex)
{
ASSERT(pNSInfo != NULL);
ASSERT( (pNSInfo->m_action == CDNSRecordNodeEditInfo::unchanged) ||
(pNSInfo->m_action == CDNSRecordNodeEditInfo::add) );
ASSERT(pNSInfo->m_pRecord != NULL);
ASSERT(pNSInfo->m_pRecordNode != NULL);
ASSERT(pNSInfo->m_pRecordNode->m_bAtTheNode);
BOOL bAlreadyExists = FALSE;
//
// First check to see if its already there
//
for (int idx = 0; idx < GetItemCount(); idx++)
{
CDNSRecordNodeEditInfo* pIdxInfo = reinterpret_cast<CDNSRecordNodeEditInfo*>(GetItemData(idx));
ASSERT(pIdxInfo != NULL);
if (pIdxInfo == NULL)
{
continue;
}
CDNS_NS_Record* pNSRecord = reinterpret_cast<CDNS_NS_Record*>(pIdxInfo->m_pRecord);
ASSERT(pNSRecord != NULL);
if (pNSRecord == NULL)
{
continue;
}
//
// Adding trailing '.' if not already present
//
CString szUINSName = pNSRecord->m_szNameNode;
CString szNewNSName = ((CDNS_NS_Record*)pNSInfo->m_pRecord)->m_szNameNode;
if (szUINSName[szUINSName.GetLength() - 1] != L'.')
{
szUINSName += L".";
}
if (szNewNSName[szNewNSName.GetLength() - 1] != L'.')
{
szNewNSName += L".";
}
//
// if it exists, just update the existing one
//
if (_wcsicmp(szNewNSName, szUINSName) == 0)
{
bAlreadyExists = TRUE;
//
// Merge the A record lists together
//
POSITION newPos = pNSInfo->m_pEditInfoList->GetHeadPosition();
while (newPos != NULL)
{
CDNSRecordNodeEditInfo* pAInfo = pNSInfo->m_pEditInfoList->GetNext(newPos);
CDNS_A_Record* pARecord = reinterpret_cast<CDNS_A_Record*>(pAInfo->m_pRecord);
ASSERT(pARecord != NULL);
if (pARecord == NULL)
{
continue;
}
BOOL bARecordExists = FALSE;
POSITION IdxPos = pIdxInfo->m_pEditInfoList->GetHeadPosition();
while (IdxPos != NULL)
{
CDNSRecordNodeEditInfo* pIdxAInfo = pIdxInfo->m_pEditInfoList->GetNext(IdxPos);
CDNS_A_Record* pIdxARecord = reinterpret_cast<CDNS_A_Record*>(pIdxAInfo->m_pRecord);
ASSERT(pIdxARecord != NULL);
if (pIdxARecord == NULL)
{
continue;
}
if (pIdxARecord->m_ipAddress == pARecord->m_ipAddress)
{
bARecordExists = TRUE;
break;
}
}
if (!bARecordExists)
{
//
// Add the A record since it doesn't already exist in the list
//
pIdxInfo->m_pEditInfoList->AddTail(pAInfo);
pIdxInfo->m_action = CDNSRecordNodeEditInfo::edit;
UpdateNSRecordEntry(idx);
}
}
}
}
if (!bAlreadyExists)
{
CString szTemp;
GetIPAddressString(pNSInfo, szTemp);
InsertItemHelper(nItemIndex, pNSInfo,
((CDNS_NS_Record*)pNSInfo->m_pRecord)->m_szNameNode,
(LPCTSTR)szTemp);
//
// Added new item so return TRUE;
//
return TRUE;
}
//
// Updated an existing item so return FALSE
//
return FALSE;
}
void CNSListCtrl::InsertItemHelper(int nIndex, CDNSRecordNodeEditInfo* pNSInfo,
LPCTSTR lpszName, LPCTSTR lpszValue)
{
UINT nState = 0;
if (nIndex == 0 )
nState = LVIS_SELECTED | LVIS_FOCUSED; // have at least one item, select it
VERIFY(-1 != InsertItem(LVIF_TEXT | LVIF_PARAM, nIndex,
lpszName, nState, 0, 0, (LPARAM)pNSInfo));
SetItemText(nIndex, 1, lpszValue);
}
void CNSListCtrl::GetIPAddressString(CDNSRecordNodeEditInfo* pNSInfo, CString& sz)
{
if (pNSInfo->m_pEditInfoList->GetCount() > 0)
{
BuildIPAddrDisplayString(pNSInfo, sz);
}
else
sz.LoadString(IDS_UNKNOWN);
}
///////////////////////////////////////////////////////////////////////////////
// CDNSNameServersPropertyPage
BEGIN_MESSAGE_MAP(CDNSNameServersPropertyPage, CPropertyPageBase)
ON_BN_CLICKED(IDC_ADD_NS_BUTTON, OnAddButton)
ON_BN_CLICKED(IDC_REMOVE_NS_BUTTON, OnRemoveButton)
ON_BN_CLICKED(IDC_EDIT_NS_BUTTON, OnEditButton)
ON_NOTIFY(LVN_ITEMCHANGED, IDC_NS_LIST, OnListItemChanged)
END_MESSAGE_MAP()
CDNSNameServersPropertyPage::CDNSNameServersPropertyPage(UINT nIDTemplate, UINT nIDCaption)
: CPropertyPageBase(nIDTemplate, nIDCaption)
{
m_pDomainNode = NULL;
m_pCloneInfoList = new CDNSRecordNodeEditInfoList;
m_bReadOnly = FALSE;
m_bMeaningfulTTL = TRUE;
}
CDNSNameServersPropertyPage::~CDNSNameServersPropertyPage()
{
delete m_pCloneInfoList;
}
BOOL CDNSNameServersPropertyPage::WriteNSRecordNodesList()
{
ASSERT(!m_bReadOnly);
ASSERT(m_pCloneInfoList != NULL);
CDNSDomainNode* pDomainNode = GetDomainNode();
return pDomainNode->UpdateNSRecordNodesInfo(m_pCloneInfoList, GetHolder()->GetComponentData());
}
BOOL CDNSNameServersPropertyPage::OnWriteNSRecordNodesListError()
{
ASSERT(!m_bReadOnly);
BOOL bSuccess = TRUE;
// loop for each NS record
POSITION pos;
for( pos = m_pCloneInfoList->GetHeadPosition(); pos != NULL; )
{
CDNSRecordNodeEditInfo* pCurrentInfo = m_pCloneInfoList->GetNext(pos);
if (pCurrentInfo->m_dwErr != 0)
{
if (pCurrentInfo->m_dwErr == DNS_ERROR_RECORD_ALREADY_EXISTS)
{
// ignore if the NS record us already there
pCurrentInfo->m_dwErr = 0;
}
else if (pCurrentInfo->m_dwErr == RPC_S_SERVER_UNAVAILABLE)
{
// Couldn't contact the server. Give a single error message
// and then break out.
DNSErrorDialog(pCurrentInfo->m_dwErr, L"");
return FALSE;
}
else
{
bSuccess = FALSE;
ASSERT(pCurrentInfo->m_pRecord->GetType() == DNS_TYPE_NS);
CString szNSMsg;
szNSMsg.Format(_T("Failure to write NS record <%s>"),
(((CDNS_NS_Record*)pCurrentInfo->m_pRecord))->m_szNameNode);
DNSErrorDialog(pCurrentInfo->m_dwErr,szNSMsg);
}
}
// loop for each related A record
CDNSRecordNodeEditInfoList* pARecordInfoList = pCurrentInfo->m_pEditInfoList;
ASSERT(pARecordInfoList != NULL);
POSITION posA;
for( posA = pARecordInfoList->GetHeadPosition(); posA != NULL; )
{
CDNSRecordNodeEditInfo* pARecordCurrentInfo = pARecordInfoList->GetNext(posA);
if (pARecordCurrentInfo->m_dwErr != 0)
{
ASSERT(pARecordCurrentInfo->m_pRecord->GetType() == DNS_TYPE_A);
CString szTemp;
FormatIpAddress(szTemp, (((CDNS_A_Record*)pARecordCurrentInfo->m_pRecord))->m_ipAddress);
CString szAMsg;
szAMsg.Format(_T("Failure to write A record <%s>, IP Address %s"),
(((CDNS_NS_Record*)pARecordCurrentInfo->m_pRecord))->m_szNameNode,
(LPCWSTR)szTemp );
DNSErrorDialog(pCurrentInfo->m_dwErr,szAMsg);
}
}
}
return bSuccess;
}
void CDNSNameServersPropertyPage::OnListItemChanged(NMHDR*, LRESULT*)
{
EnableEditorButtons(m_listCtrl.GetSelection());
}
BOOL CDNSNameServersPropertyPage::OnApply()
{
if (m_bReadOnly)
return TRUE;
if (!IsDirty())
return TRUE;
DNS_STATUS err = GetHolder()->NotifyConsole(this);
if ( (err != 0) && OnWriteNSRecordNodesListError() )
{
err = 0; // error was handled and it was not fatal
}
if (err == 0)
{
// refresh data from the zone/domain
LoadUIData();
SetDirty(FALSE);
}
return (err == 0);
}
BOOL CDNSNameServersPropertyPage::OnPropertyChange(BOOL, long*)
{
if (m_bReadOnly)
{
ASSERT(FALSE); // sould not happen
return FALSE;
}
ASSERT(m_pCloneInfoList != NULL);
if (m_pCloneInfoList == NULL)
return FALSE;
BOOL bRes = WriteNSRecordNodesList();
if (!bRes)
GetHolder()->SetError(static_cast<DWORD>(-1)); // something went wrong, error code will be per item
return bRes; // update flag
}
void CDNSNameServersPropertyPage::OnAddButton()
{
if (m_bReadOnly)
{
ASSERT(FALSE); // sould not happen
return;
}
AFX_MANAGE_STATE(AfxGetStaticModuleState());
FIX_THREAD_STATE_MFC_BUG();
ASSERT(m_pCloneInfoList != NULL);
// provide subdialog to add record, for the time being can have just
// dialog to enter the host name
// create an item in the list of changes
// TODO: first check if we can recycle some old stuff in the list
// create new data
CDNSRecordNodeEditInfo* pNewInfo = new CDNSRecordNodeEditInfo;
if (!pNewInfo)
{
return;
}
pNewInfo->m_action = CDNSRecordNodeEditInfo::add;
CDNS_NS_RecordNode* pNSRecordNode = new CDNS_NS_RecordNode;
if (!pNSRecordNode)
{
delete pNewInfo;
return;
}
pNewInfo->CreateFromNewRecord(pNSRecordNode);
// NS records are ALWAYS at the node
CDNSDomainNode* pDomainNode = GetDomainNode();
CDNSRootData* pRootData = (CDNSRootData*)(GetHolder()->GetComponentData()->GetRootData());
ASSERT(pRootData != NULL);
// set name and type flag
pNewInfo->m_pRecordNode->SetRecordName(pDomainNode->GetDisplayName(), TRUE /*bAtTheNode */);
pNewInfo->m_pRecordNode->SetFlagsDown(TN_FLAG_DNS_RECORD_FULL_NAME, !pRootData->IsAdvancedView());
// set TTL
pNewInfo->m_pRecord->m_dwTtlSeconds = pDomainNode->GetDefaultTTL();
CDNS_NS_RecordDialog dlg(this,TRUE);
dlg.m_pNSInfo = pNewInfo;
if (IDOK == dlg.DoModalSheet(m_hWnd) && pNewInfo->m_action == CDNSRecordNodeEditInfo::add)
{
//
// add to the list view (at the end)
//
int nCount = m_listCtrl.GetItemCount();
if (m_listCtrl.InsertNSRecordEntry(pNewInfo, nCount))
{
//
// create entry into the record info list
//
m_pCloneInfoList->AddTail(pNewInfo);
//
// set selection and button state on the last inserted
//
m_listCtrl.SetSelection(nCount);
EnableEditorButtons(nCount);
//
// notify count change
//
OnCountChange(nCount+1); // added one
}
//
// set dirty flag. It was either a new record or an update of an old one
//
SetDirty(TRUE);
}
else
{
delete pNewInfo->m_pRecordNode;
pNewInfo->m_pRecordNode = NULL;
delete pNewInfo->m_pRecord;
pNewInfo->m_pRecord = NULL;
delete pNewInfo;
}
}
void CDNSNameServersPropertyPage::OnEditButton()
{
if (m_bReadOnly)
{
ASSERT(FALSE); // sould not happen
return;
}
AFX_MANAGE_STATE(AfxGetStaticModuleState());
FIX_THREAD_STATE_MFC_BUG();
ASSERT(m_pCloneInfoList != NULL);
// get the selection and bring up the dialog with the host name for editing
int nSel = m_listCtrl.GetSelection();
ASSERT(nSel != -1);
if (nSel == -1)
return; // should not happen
CDNSRecordNodeEditInfo* pNSInfo = (CDNSRecordNodeEditInfo*)m_listCtrl.GetItemData(nSel);
ASSERT(pNSInfo != NULL);
CDNS_NS_RecordDialog dlg(this, FALSE);
ASSERT(pNSInfo->m_pRecord->GetType() == DNS_TYPE_NS);
dlg.m_pNSInfo = pNSInfo;
if (IDOK == dlg.DoModalSheet() && dlg.m_bDirty)
{
if (pNSInfo->m_action == CDNSRecordNodeEditInfo::add)
{
// this was a new entry that was edited after creation
// but before committing the change
ASSERT(!pNSInfo->m_bExisting);
// update the listview
m_listCtrl.UpdateNSRecordEntry(nSel);
}
else if (pNSInfo->m_action == CDNSRecordNodeEditInfo::edit)
{
// this was an existing entry that was changed
ASSERT(pNSInfo->m_bExisting);
// update the listview
m_listCtrl.UpdateNSRecordEntry(nSel);
}
else
{
// there were no IP addresses, so mark the item for removal
ASSERT(pNSInfo->m_action == CDNSRecordNodeEditInfo::remove);
OnRemoveButton();
}
// set dirty flag
SetDirty(TRUE);
}
}
void CDNSNameServersPropertyPage::OnRemoveButton()
{
if (m_bReadOnly)
{
ASSERT(FALSE); // sould not happen
return;
}
int nSel = m_listCtrl.GetSelection();
if (nSel == -1)
{
ASSERT(FALSE);
return; // should not happen
}
//
// save focus to restore afterwards, if needed
//
CWnd* pWndFocusOld = CWnd::GetFocus();
ASSERT(pWndFocusOld != NULL);
//
// got a selection, delete from listview
//
CDNSRecordNodeEditInfo* pNSInfo = (CDNSRecordNodeEditInfo*)m_listCtrl.GetItemData(nSel);
ASSERT(pNSInfo != NULL);
m_listCtrl.DeleteItem(nSel);
//
// we lost the selection, set it again
//
int nNewCount = m_listCtrl.GetItemCount();
if (nNewCount == nSel)
{
//
// last item in the list was deleted, move selection up
//
nSel--;
}
if (nSel != -1)
{
m_listCtrl.SetSelection(nSel);
ASSERT(m_listCtrl.GetSelection() == nSel);
}
//
// if there are no items left, need to disable the Edit and Remove buttons
//
if (nNewCount == 0)
{
CButton* pRemoveButton = GetRemoveButton();
//
// need to shift focus before disabling buttons
//
CButton* pAddButton = GetAddButton();
pAddButton->SetFocus();
// avoid to have the OK button on the sheet to become the default button
pRemoveButton->SendMessage(BM_SETSTYLE, BS_PUSHBUTTON, MAKELPARAM(TRUE, 0));
pAddButton->SendMessage(BM_SETSTYLE, BS_DEFPUSHBUTTON,
MAKELPARAM(/*redraw flag*/ TRUE, 0));
EnableEditorButtons(nSel); // this will disable both Edit and Remove
}
ASSERT(CWnd::GetFocus());
if (pNSInfo->m_action == CDNSRecordNodeEditInfo::add)
{
//
// mark the item action as none since it was just added without being applied
//
pNSInfo->m_action = CDNSRecordNodeEditInfo::none;
}
else
{
// mark the item as deleted in the list of changes
pNSInfo->m_action = CDNSRecordNodeEditInfo::remove;
}
// set dirty flag, removed a record
SetDirty(TRUE);
// notify count change
OnCountChange(nNewCount);
}
BOOL CDNSNameServersPropertyPage::OnInitDialog()
{
CPropertyPageBase::OnInitDialog();
// controls initialization
VERIFY(m_listCtrl.SubclassDlgItem(IDC_NS_LIST, this));
m_listCtrl.Initialize();
// load the data
LoadUIData();
// set button state
if (m_bReadOnly)
{
EnableButtons(FALSE);
}
else
{
// set selection to first item in the list, if there
int nSel = (m_listCtrl.GetItemCount()>0) ? 0 : -1;
EnableEditorButtons(nSel);
}
return TRUE; // return TRUE unless you set the focus to a control
// EXCEPTION: OCX Property Pages should return FALSE
}
void CDNSNameServersPropertyPage::LoadUIData()
{
m_pCloneInfoList->RemoveAllNodes();
ReadRecordNodesList(); // read from source
FillNsListView();
}
void CDNSNameServersPropertyPage::FillNsListView()
{
ASSERT(m_pCloneInfoList != NULL);
m_listCtrl.DeleteAllItems();
// loop through the list of NS records and insert
POSITION pos;
int itemIndex = 0;
for( pos = m_pCloneInfoList->GetHeadPosition(); pos != NULL; )
{
CDNSRecordNodeEditInfo* pCurrentInfo = m_pCloneInfoList->GetNext(pos);
if (m_listCtrl.InsertNSRecordEntry(pCurrentInfo, itemIndex))
{
if (itemIndex == 0)
m_listCtrl.SetSelection(0);
itemIndex++;
}
}
}
void CDNSNameServersPropertyPage::EnableEditorButtons(int nListBoxSel)
{
if (m_bReadOnly)
return;
// must have item selected to remove or add
GetRemoveButton()->EnableWindow(nListBoxSel != -1);
GetEditButton()->EnableWindow(nListBoxSel != -1);
}
void CDNSNameServersPropertyPage::EnableButtons(BOOL bEnable)
{
GetAddButton()->EnableWindow(bEnable);
GetRemoveButton()->EnableWindow(bEnable);
GetEditButton()->EnableWindow(bEnable);
}
//////////////////////////////////////////////////////////////////////////////
///////////////////////////////////////////////////////////////////////////////
// CDNSNameServersWizardPage
BEGIN_MESSAGE_MAP(CDNSNameServersWizardPage, CPropertyPageBase)
ON_BN_CLICKED(IDC_ADD_NS_BUTTON, OnAddButton)
ON_BN_CLICKED(IDC_REMOVE_NS_BUTTON, OnRemoveButton)
ON_BN_CLICKED(IDC_EDIT_NS_BUTTON, OnEditButton)
ON_NOTIFY(LVN_ITEMCHANGED, IDC_NS_LIST, OnListItemChanged)
END_MESSAGE_MAP()
CDNSNameServersWizardPage::CDNSNameServersWizardPage(UINT nIDTemplate)
: CPropertyPageBase(nIDTemplate)
{
InitWiz97(FALSE,IDS_SERVWIZ_ROOTHINTS_TITLE,IDS_SERVWIZ_ROOTHINTS_SUBTITLE);
m_pDomainNode = NULL;
m_pCloneInfoList = new CDNSRecordNodeEditInfoList;
m_bReadOnly = FALSE;
m_bMeaningfulTTL = TRUE;
}
CDNSNameServersWizardPage::~CDNSNameServersWizardPage()
{
delete m_pCloneInfoList;
}
BOOL CDNSNameServersWizardPage::WriteNSRecordNodesList()
{
ASSERT(!m_bReadOnly);
ASSERT(m_pCloneInfoList != NULL);
CDNSDomainNode* pDomainNode = GetDomainNode();
return pDomainNode->UpdateNSRecordNodesInfo(m_pCloneInfoList, GetHolder()->GetComponentData());
}
BOOL CDNSNameServersWizardPage::OnWriteNSRecordNodesListError()
{
ASSERT(!m_bReadOnly);
BOOL bSuccess = TRUE;
// loop for each NS record
POSITION pos;
for( pos = m_pCloneInfoList->GetHeadPosition(); pos != NULL; )
{
CDNSRecordNodeEditInfo* pCurrentInfo = m_pCloneInfoList->GetNext(pos);
if (pCurrentInfo->m_dwErr != 0)
{
if (pCurrentInfo->m_dwErr == DNS_ERROR_RECORD_ALREADY_EXISTS)
{
// ignore if the NS record us already there
pCurrentInfo->m_dwErr = 0;
}
else
{
bSuccess = FALSE;
ASSERT(pCurrentInfo->m_pRecord->GetType() == DNS_TYPE_NS);
CString szNSMsg;
szNSMsg.Format(_T("Failure to write NS record <%s>"),
(((CDNS_NS_Record*)pCurrentInfo->m_pRecord))->m_szNameNode);
DNSErrorDialog(pCurrentInfo->m_dwErr,szNSMsg);
}
}
// loop for each related A record
CDNSRecordNodeEditInfoList* pARecordInfoList = pCurrentInfo->m_pEditInfoList;
ASSERT(pARecordInfoList != NULL);
POSITION posA;
for( posA = pARecordInfoList->GetHeadPosition(); posA != NULL; )
{
CDNSRecordNodeEditInfo* pARecordCurrentInfo = pARecordInfoList->GetNext(posA);
if (pARecordCurrentInfo->m_dwErr != 0)
{
ASSERT(pARecordCurrentInfo->m_pRecord->GetType() == DNS_TYPE_A);
CString szTemp;
FormatIpAddress(szTemp, (((CDNS_A_Record*)pARecordCurrentInfo->m_pRecord))->m_ipAddress);
CString szAMsg;
szAMsg.Format(_T("Failure to write A record <%s>, IP Address %s"),
(((CDNS_NS_Record*)pARecordCurrentInfo->m_pRecord))->m_szNameNode,
(LPCWSTR)szTemp );
DNSErrorDialog(pCurrentInfo->m_dwErr,szAMsg);
}
}
}
return bSuccess;
}
BOOL CDNSNameServersWizardPage::OnApply()
{
if (m_bReadOnly)
return TRUE;
if (!IsDirty())
return TRUE;
DNS_STATUS err = GetHolder()->NotifyConsole(this);
if ( (err != 0) && OnWriteNSRecordNodesListError() )
{
err = 0; // error was handled and it was not fatal
}
if (err == 0)
{
// refresh data from the zone/domain
LoadUIData();
SetDirty(FALSE);
}
else
{
::SetLastError(err);
}
return (err == 0);
}
BOOL CDNSNameServersWizardPage::OnPropertyChange(BOOL, long*)
{
if (m_bReadOnly)
{
ASSERT(FALSE); // sould not happen
return FALSE;
}
ASSERT(m_pCloneInfoList != NULL);
if (m_pCloneInfoList == NULL)
return FALSE;
BOOL bRes = WriteNSRecordNodesList();
if (!bRes)
GetHolder()->SetError(static_cast<DWORD>(-1)); // something went wrong, error code will be per item
return bRes; // update flag
}
void CDNSNameServersWizardPage::OnListItemChanged(NMHDR*, LRESULT*)
{
EnableEditorButtons(m_listCtrl.GetSelection());
}
void CDNSNameServersWizardPage::OnAddButton()
{
if (m_bReadOnly)
{
ASSERT(FALSE); // sould not happen
return;
}
AFX_MANAGE_STATE(AfxGetStaticModuleState());
FIX_THREAD_STATE_MFC_BUG();
ASSERT(m_pCloneInfoList != NULL);
// provide subdialog to add record, for the time being can have just
// dialog to enter the host name
// create an item in the list of changes
// TODO: first check if we can recycle some old stuff in the list
// create new data
CDNSRecordNodeEditInfo* pNewInfo = new CDNSRecordNodeEditInfo;
if (!pNewInfo)
{
return;
}
pNewInfo->m_action = CDNSRecordNodeEditInfo::add;
CDNS_NS_RecordNode* pNSRecordNode = new CDNS_NS_RecordNode;
if (!pNSRecordNode)
{
delete pNewInfo;
return;
}
pNewInfo->CreateFromNewRecord(pNSRecordNode);
// NS records are ALWAYS at the node
CDNSDomainNode* pDomainNode = GetDomainNode();
CDNSRootData* pRootData = (CDNSRootData*)(GetHolder()->GetComponentData()->GetRootData());
ASSERT(pRootData != NULL);
// set name and type flag
pNewInfo->m_pRecordNode->SetRecordName(pDomainNode->GetDisplayName(), TRUE /*bAtTheNode */);
pNewInfo->m_pRecordNode->SetFlagsDown(TN_FLAG_DNS_RECORD_FULL_NAME, !pRootData->IsAdvancedView());
// set TTL
pNewInfo->m_pRecord->m_dwTtlSeconds = pDomainNode->GetDefaultTTL();
CDNS_NS_RecordDialog dlg(this,TRUE);
dlg.m_pNSInfo = pNewInfo;
if (IDOK == dlg.DoModalSheet() && pNewInfo->m_action == CDNSRecordNodeEditInfo::add)
{
//
// add to the list view (at the end)
//
int nCount = m_listCtrl.GetItemCount();
if (m_listCtrl.InsertNSRecordEntry(pNewInfo, nCount))
{
//
// create entry into the record info list
//
m_pCloneInfoList->AddTail(pNewInfo);
//
// set selection and button state on the last inserted
//
m_listCtrl.SetSelection(nCount);
EnableEditorButtons(nCount);
//
// notify count change
//
OnCountChange(nCount+1); // added one
}
//
// set dirty flag, it is a new record
//
SetDirty(TRUE);
}
else
{
delete pNewInfo->m_pRecordNode;
pNewInfo->m_pRecordNode = NULL;
delete pNewInfo->m_pRecord;
pNewInfo->m_pRecord = NULL;
delete pNewInfo;
}
}
void CDNSNameServersWizardPage::OnEditButton()
{
if (m_bReadOnly)
{
ASSERT(FALSE); // sould not happen
return;
}
AFX_MANAGE_STATE(AfxGetStaticModuleState());
FIX_THREAD_STATE_MFC_BUG();
ASSERT(m_pCloneInfoList != NULL);
// get the selection and bring up the dialog with the host name for editing
int nSel = m_listCtrl.GetSelection();
ASSERT(nSel != -1);
if (nSel == -1)
return; // should not happen
CDNSRecordNodeEditInfo* pNSInfo = (CDNSRecordNodeEditInfo*)m_listCtrl.GetItemData(nSel);
ASSERT(pNSInfo != NULL);
CDNS_NS_RecordDialog dlg(this, FALSE);
ASSERT(pNSInfo->m_pRecord->GetType() == DNS_TYPE_NS);
dlg.m_pNSInfo = pNSInfo;
if (IDOK == dlg.DoModalSheet() && dlg.m_bDirty)
{
if (pNSInfo->m_action == CDNSRecordNodeEditInfo::add)
{
// this was a new entry that was edited after creation
// but before committing the change
ASSERT(!pNSInfo->m_bExisting);
// update the listview
m_listCtrl.UpdateNSRecordEntry(nSel);
}
else if (pNSInfo->m_action == CDNSRecordNodeEditInfo::edit)
{
// this was an existing entry that was changed
ASSERT(pNSInfo->m_bExisting);
// update the listview
m_listCtrl.UpdateNSRecordEntry(nSel);
}
else
{
// there were no IP addresses, so mark the item for removal
ASSERT(pNSInfo->m_action == CDNSRecordNodeEditInfo::remove);
OnRemoveButton();
}
// set dirty flag
SetDirty(TRUE);
}
}
void CDNSNameServersWizardPage::OnRemoveButton()
{
if (m_bReadOnly)
{
ASSERT(FALSE); // sould not happen
return;
}
int nSel = m_listCtrl.GetSelection();
if (nSel == -1)
{
ASSERT(FALSE);
return; // should not happen
}
//
// save focus to restore afterwards, if needed
//
CWnd* pWndFocusOld = CWnd::GetFocus();
ASSERT(pWndFocusOld != NULL);
//
// got a selection, delete from listview
//
CDNSRecordNodeEditInfo* pNSInfo = (CDNSRecordNodeEditInfo*)m_listCtrl.GetItemData(nSel);
ASSERT(pNSInfo != NULL);
m_listCtrl.DeleteItem(nSel);
//
// we lost the selection, set it again
//
int nNewCount = m_listCtrl.GetItemCount();
if (nNewCount == nSel)
{
//
// last item in the list was deleted, move selection up
//
nSel--;
}
if (nSel != -1)
{
m_listCtrl.SetSelection(nSel);
ASSERT(m_listCtrl.GetSelection() == nSel);
}
//
// if there are no items left, need to disable the Edit and Remove buttons
//
if (nNewCount == 0)
{
CWnd* pCurrentFocusCtrl = CWnd::GetFocus();
CButton* pRemoveButton = GetRemoveButton();
CButton* pEditButton = GetEditButton();
//
// need to shift focus before disabling buttons
//
if ( (pCurrentFocusCtrl == pRemoveButton) ||
(pCurrentFocusCtrl == pEditButton) )
{
CButton* pAddButton = GetAddButton();
pAddButton->SetFocus();
// avoid to have the OK button on the sheet to become the default button
pAddButton->SendMessage(BM_SETSTYLE, BS_DEFPUSHBUTTON,
MAKELPARAM(/*redraw flag*/ TRUE, 0));
}
EnableEditorButtons(nSel); // this will disable both Edit and Remove
}
ASSERT(CWnd::GetFocus());
if (pNSInfo->m_action == CDNSRecordNodeEditInfo::add)
{
//
// mark the item action as none since the item hasn't been added yet anyways
//
pNSInfo->m_action = CDNSRecordNodeEditInfo::none;
}
else
{
// mark the item as deleted in the list of changes
pNSInfo->m_action = CDNSRecordNodeEditInfo::remove;
}
// set dirty flag, removed a record
SetDirty(TRUE);
// notify count change
OnCountChange(nNewCount);
}
BOOL CDNSNameServersWizardPage::OnInitDialog()
{
CPropertyPageBase::OnInitDialog();
// controls initialization
VERIFY(m_listCtrl.SubclassDlgItem(IDC_NS_LIST, this));
m_listCtrl.Initialize();
// load the data
LoadUIData();
// set button state
if (m_bReadOnly)
{
EnableButtons(FALSE);
}
else
{
// set selection to first item in the list, if there
int nSel = (m_listCtrl.GetItemCount()>0) ? 0 : -1;
EnableEditorButtons(nSel);
}
return TRUE; // return TRUE unless you set the focus to a control
// EXCEPTION: OCX Property Pages should return FALSE
}
void CDNSNameServersWizardPage::LoadUIData()
{
m_pCloneInfoList->RemoveAllNodes();
ReadRecordNodesList(); // read from source
FillNsListView();
}
void CDNSNameServersWizardPage::FillNsListView()
{
ASSERT(m_pCloneInfoList != NULL);
m_listCtrl.DeleteAllItems();
// loop through the list of NS records and insert
POSITION pos;
int itemIndex = 0;
for( pos = m_pCloneInfoList->GetHeadPosition(); pos != NULL; )
{
CDNSRecordNodeEditInfo* pCurrentInfo = m_pCloneInfoList->GetNext(pos);
if (m_listCtrl.InsertNSRecordEntry(pCurrentInfo, itemIndex))
{
if (itemIndex == 0)
m_listCtrl.SetSelection(0);
itemIndex++;
}
}
}
void CDNSNameServersWizardPage::EnableEditorButtons(int nListBoxSel)
{
if (m_bReadOnly)
return;
// must have item selected to remove or add
GetRemoveButton()->EnableWindow(nListBoxSel != -1);
GetEditButton()->EnableWindow(nListBoxSel != -1);
}
void CDNSNameServersWizardPage::EnableButtons(BOOL bEnable)
{
GetAddButton()->EnableWindow(bEnable);
GetRemoveButton()->EnableWindow(bEnable);
GetEditButton()->EnableWindow(bEnable);
}