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.
1920 lines
53 KiB
1920 lines
53 KiB
/*++
|
|
Module Name:
|
|
|
|
CusTop.cpp
|
|
|
|
Abstract:
|
|
|
|
This module contains the Implementation of CCustomTopology.
|
|
This class displays the Customize Topology Dialog.
|
|
|
|
*/
|
|
|
|
#include "stdafx.h"
|
|
#include "CusTop.h"
|
|
#include "frsAdv.h"
|
|
#include "utils.h"
|
|
#include "dfshelp.h"
|
|
#include "ldaputils.h"
|
|
|
|
int g_FRS_CUSTOP_Last_SortColumn = 1;
|
|
#define NUM_OF_FRS_CUSTOP_COLUMNS 5
|
|
|
|
RSTOPOLOGYPREF_STRING g_TopologyPref[] = {
|
|
{FRS_RSTOPOLOGYPREF_RING, IDS_FRSPROP_RING},
|
|
{FRS_RSTOPOLOGYPREF_HUBSPOKE, IDS_FRSPROP_HUBSPOKE},
|
|
{FRS_RSTOPOLOGYPREF_FULLMESH, IDS_FRSPROP_FULLMESH},
|
|
{FRS_RSTOPOLOGYPREF_CUSTOM, IDS_FRSPROP_CUSTOM}
|
|
};
|
|
|
|
// sort member list based on m_bstrSite
|
|
struct CusTopMemberCompareSite : greater<CCusTopMember*>
|
|
{
|
|
bool operator()(const CCusTopMember *pMem1, const CCusTopMember *pMem2) const
|
|
{ return (lstrcmpi(pMem1->m_bstrSite, pMem2->m_bstrSite) > 0); }
|
|
};
|
|
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// CCustomTopology
|
|
//
|
|
|
|
CCustomTopology::CCustomTopology()
|
|
{
|
|
}
|
|
|
|
CCustomTopology::~CCustomTopology()
|
|
{
|
|
}
|
|
|
|
HRESULT CCustomTopology::_GetMemberList()
|
|
{
|
|
FreeCusTopMembers(&m_MemberList);
|
|
|
|
RETURN_INVALIDARG_IF_NULL((IReplicaSet *)m_piReplicaSet);
|
|
|
|
VARIANT var;
|
|
VariantInit(&var);
|
|
HRESULT hr = m_piReplicaSet->GetMemberListEx(&var);
|
|
RETURN_IF_FAILED(hr);
|
|
|
|
if (V_VT(&var) != (VT_ARRAY | VT_VARIANT))
|
|
return E_INVALIDARG;
|
|
|
|
SAFEARRAY *psa_1 = V_ARRAY(&var);
|
|
if (!psa_1) // no member at all
|
|
return hr;
|
|
|
|
long lLowerBound_1 = 0;
|
|
long lUpperBound_1 = 0;
|
|
long lCount_1 = 0;
|
|
SafeArrayGetLBound(psa_1, 1, &lLowerBound_1);
|
|
SafeArrayGetUBound(psa_1, 1, &lUpperBound_1);
|
|
lCount_1 = lUpperBound_1 - lLowerBound_1 + 1;
|
|
|
|
VARIANT HUGEP *pArray_1;
|
|
SafeArrayAccessData(psa_1, (void HUGEP **) &pArray_1);
|
|
|
|
for (long i = 0; i < lCount_1; i++)
|
|
{
|
|
if (V_VT(&(pArray_1[i])) != (VT_ARRAY | VT_VARIANT))
|
|
{
|
|
hr = E_INVALIDARG;
|
|
break;
|
|
}
|
|
|
|
SAFEARRAY *psa_0 = V_ARRAY(&(pArray_1[i]));
|
|
if (!psa_0)
|
|
{
|
|
hr = E_INVALIDARG;
|
|
break;
|
|
}
|
|
|
|
long lLowerBound_0 = 0;
|
|
long lUpperBound_0 = 0;
|
|
long lCount_0 = 0;
|
|
SafeArrayGetLBound(psa_0, 1, &lLowerBound_0);
|
|
SafeArrayGetUBound(psa_0, 1, &lUpperBound_0);
|
|
lCount_0 = lUpperBound_0 - lLowerBound_0 + 1;
|
|
if (NUM_OF_FRSMEMBER_ATTRS != lCount_0)
|
|
{
|
|
hr = E_INVALIDARG;
|
|
break;
|
|
}
|
|
|
|
VARIANT HUGEP *pArray_0;
|
|
SafeArrayAccessData(psa_0, (void HUGEP **) &pArray_0);
|
|
|
|
do {
|
|
CCusTopMember* pNew = new CCusTopMember;
|
|
BREAK_OUTOFMEMORY_IF_NULL(pNew, &hr);
|
|
|
|
hr = pNew->Init(
|
|
pArray_0[2].bstrVal, //bstrMemberDN,
|
|
pArray_0[4].bstrVal, //bstrServer,
|
|
pArray_0[6].bstrVal //bstrSite
|
|
);
|
|
|
|
if (SUCCEEDED(hr))
|
|
m_MemberList.push_back(pNew);
|
|
else
|
|
delete pNew;
|
|
|
|
} while (0);
|
|
|
|
SafeArrayUnaccessData(psa_0);
|
|
}
|
|
|
|
SafeArrayUnaccessData(psa_1);
|
|
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
//
|
|
// sort member list on m_bstrServer alphabetically
|
|
//
|
|
hr = _SortMemberList();
|
|
}
|
|
|
|
if (FAILED(hr))
|
|
FreeCusTopMembers(&m_MemberList);
|
|
|
|
SafeArrayDestroy(psa_1); // it should free psa_0 as well
|
|
|
|
return hr;
|
|
}
|
|
|
|
HRESULT CCustomTopology::_GetConnectionList()
|
|
{
|
|
FreeCusTopConnections(&m_ConnectionList);
|
|
|
|
RETURN_INVALIDARG_IF_NULL((IReplicaSet *)m_piReplicaSet);
|
|
|
|
VARIANT var;
|
|
VariantInit(&var);
|
|
HRESULT hr = m_piReplicaSet->GetConnectionListEx(&var);
|
|
RETURN_IF_FAILED(hr);
|
|
|
|
if (V_VT(&var) != (VT_ARRAY | VT_VARIANT))
|
|
return E_INVALIDARG;
|
|
|
|
SAFEARRAY *psa_1 = V_ARRAY(&var);
|
|
if (!psa_1) // no connection at all
|
|
return hr;
|
|
|
|
long lLowerBound_1 = 0;
|
|
long lUpperBound_1 = 0;
|
|
long lCount_1 = 0;
|
|
SafeArrayGetLBound(psa_1, 1, &lLowerBound_1);
|
|
SafeArrayGetUBound(psa_1, 1, &lUpperBound_1);
|
|
lCount_1 = lUpperBound_1 - lLowerBound_1 + 1;
|
|
|
|
VARIANT HUGEP *pArray_1;
|
|
SafeArrayAccessData(psa_1, (void HUGEP **) &pArray_1);
|
|
|
|
for (long i = 0; i < lCount_1; i++)
|
|
{
|
|
if (V_VT(&(pArray_1[i])) != (VT_ARRAY | VT_VARIANT))
|
|
{
|
|
hr = E_INVALIDARG;
|
|
break;
|
|
}
|
|
|
|
SAFEARRAY *psa_0 = V_ARRAY(&(pArray_1[i]));
|
|
if (!psa_0)
|
|
{
|
|
hr = E_INVALIDARG;
|
|
break;
|
|
}
|
|
|
|
long lLowerBound_0 = 0;
|
|
long lUpperBound_0 = 0;
|
|
long lCount_0 = 0;
|
|
SafeArrayGetLBound(psa_0, 1, &lLowerBound_0);
|
|
SafeArrayGetUBound(psa_0, 1, &lUpperBound_0);
|
|
lCount_0 = lUpperBound_0 - lLowerBound_0 + 1;
|
|
if (NUM_OF_FRSCONNECTION_ATTRS != lCount_0)
|
|
{
|
|
hr = E_INVALIDARG;
|
|
break;
|
|
}
|
|
|
|
VARIANT HUGEP *pArray_0;
|
|
SafeArrayAccessData(psa_0, (void HUGEP **) &pArray_0);
|
|
|
|
do {
|
|
CComBSTR bstrFromServer;
|
|
CComBSTR bstrFromSite;
|
|
hr = _GetMemberDNInfo(pArray_0[1].bstrVal, &bstrFromServer, &bstrFromSite);
|
|
BREAK_IF_FAILED(hr);
|
|
|
|
CComBSTR bstrToServer;
|
|
CComBSTR bstrToSite;
|
|
hr = _GetMemberDNInfo(pArray_0[2].bstrVal, &bstrToServer, &bstrToSite);
|
|
BREAK_IF_FAILED(hr);
|
|
|
|
BOOL bSyncImmediately = (BOOL)(NTDSCONN_IGNORE_SCHEDULE(pArray_0[4].lVal));
|
|
Connection_Priority nPriority = PRIORITY_LOW;
|
|
switch ((pArray_0[4].lVal & FRSCONN_PRIORITY_MASK) >> 28)
|
|
{
|
|
case 1:
|
|
case 2:
|
|
nPriority = PRIORITY_HIGH;
|
|
break;
|
|
case 3:
|
|
case 4:
|
|
nPriority = PRIORITY_MEDIUM;
|
|
break;
|
|
default:
|
|
nPriority = PRIORITY_LOW;
|
|
break;
|
|
}
|
|
|
|
CCusTopConnection* pNew = new CCusTopConnection;
|
|
BREAK_OUTOFMEMORY_IF_NULL(pNew, &hr);
|
|
hr = pNew->Init(
|
|
pArray_0[1].bstrVal, //bstrFromDN,
|
|
bstrFromServer,
|
|
bstrFromSite,
|
|
pArray_0[2].bstrVal, //bstrToDN,
|
|
bstrToServer,
|
|
bstrToSite,
|
|
(BOOL)(pArray_0[3].lVal), //bEnable
|
|
CONNECTION_OPTYPE_OTHERS,
|
|
bSyncImmediately,
|
|
nPriority
|
|
);
|
|
|
|
if (SUCCEEDED(hr))
|
|
m_ConnectionList.push_back(pNew);
|
|
else
|
|
delete pNew;
|
|
|
|
} while (0);
|
|
|
|
SafeArrayUnaccessData(psa_0);
|
|
}
|
|
|
|
SafeArrayUnaccessData(psa_1);
|
|
|
|
if (FAILED(hr))
|
|
FreeCusTopConnections(&m_ConnectionList);
|
|
|
|
SafeArrayDestroy(psa_1);
|
|
|
|
return hr;
|
|
}
|
|
|
|
void CCustomTopology::_Reset()
|
|
{
|
|
m_bstrTopologyPref.Empty();
|
|
m_bstrHubMemberDN.Empty();
|
|
|
|
FreeCusTopMembers(&m_MemberList);
|
|
FreeCusTopConnections(&m_ConnectionList);
|
|
|
|
m_piReplicaSet = NULL;
|
|
}
|
|
|
|
HRESULT CCustomTopology::put_ReplicaSet
|
|
(
|
|
IReplicaSet* i_piReplicaSet
|
|
)
|
|
{
|
|
RETURN_INVALIDARG_IF_NULL(i_piReplicaSet);
|
|
|
|
_Reset();
|
|
|
|
m_piReplicaSet = i_piReplicaSet;
|
|
|
|
HRESULT hr = S_OK;
|
|
|
|
do {
|
|
hr = m_piReplicaSet->get_TopologyPref(&m_bstrTopologyPref);
|
|
BREAK_IF_FAILED(hr);
|
|
|
|
hr = m_piReplicaSet->get_HubMemberDN(&m_bstrHubMemberDN);
|
|
BREAK_IF_FAILED(hr);
|
|
|
|
hr = _GetMemberList();
|
|
BREAK_IF_FAILED(hr);
|
|
|
|
hr = _GetConnectionList();
|
|
BREAK_IF_FAILED(hr);
|
|
} while (0);
|
|
|
|
if (FAILED(hr))
|
|
_Reset();
|
|
|
|
return hr;
|
|
}
|
|
|
|
int __cdecl CusTopMemberCompareServer(const void *arg1, const void *arg2 )
|
|
{
|
|
return lstrcmpi(
|
|
(*(CCusTopMember**)arg1)->m_bstrServer,
|
|
(*(CCusTopMember**)arg2)->m_bstrServer
|
|
);
|
|
}
|
|
|
|
HRESULT CCustomTopology::_SortMemberList()
|
|
{
|
|
HRESULT hr = S_OK;
|
|
int cMembers = m_MemberList.size();
|
|
if (2 > cMembers)
|
|
return hr;
|
|
|
|
CCusTopMember** ppMember = (CCusTopMember **)calloc(cMembers, sizeof(CCusTopMember *));
|
|
RETURN_OUTOFMEMORY_IF_NULL(ppMember);
|
|
|
|
int i = 0;
|
|
for (CCusTopMemberList::iterator it = m_MemberList.begin(); it != m_MemberList.end(); it++, i++)
|
|
{
|
|
ppMember[i] = (*it);
|
|
}
|
|
|
|
qsort((void *)ppMember, cMembers, sizeof(CCusTopMember *), CusTopMemberCompareServer);
|
|
|
|
m_MemberList.clear(); // without deleting the object
|
|
for (i = 0; i < cMembers; i++)
|
|
{
|
|
m_MemberList.push_back(ppMember[i]);
|
|
}
|
|
|
|
free((void *)ppMember);
|
|
|
|
return hr;
|
|
}
|
|
|
|
HRESULT CCustomTopology::_GetMemberDNInfo(
|
|
IN BSTR i_bstrMemberDN,
|
|
OUT BSTR* o_pbstrServer,
|
|
OUT BSTR* o_pbstrSite
|
|
)
|
|
{
|
|
RETURN_INVALIDARG_IF_NULL(i_bstrMemberDN);
|
|
RETURN_INVALIDARG_IF_NULL(o_pbstrServer);
|
|
RETURN_INVALIDARG_IF_NULL(o_pbstrSite);
|
|
|
|
for (CCusTopMemberList::iterator i = m_MemberList.begin(); i != m_MemberList.end(); i++)
|
|
{
|
|
if (!lstrcmpi(i_bstrMemberDN, (*i)->m_bstrMemberDN))
|
|
break;
|
|
}
|
|
|
|
if (i == m_MemberList.end())
|
|
return S_FALSE;
|
|
|
|
*o_pbstrServer = (*i)->m_bstrServer.Copy();
|
|
RETURN_OUTOFMEMORY_IF_NULL(*o_pbstrServer);
|
|
|
|
*o_pbstrSite = (*i)->m_bstrSite.Copy();
|
|
if (!*o_pbstrSite)
|
|
{
|
|
SysFreeString(*o_pbstrServer);
|
|
return E_OUTOFMEMORY;
|
|
}
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT CCustomTopology::_GetHubMember(
|
|
OUT CCusTopMember** o_ppHubMember
|
|
)
|
|
{
|
|
RETURN_INVALIDARG_IF_NULL(o_ppHubMember);
|
|
|
|
int index = SendDlgItemMessage(IDC_FRS_CUSTOP_HUBSERVER, CB_GETCURSEL, 0, 0);
|
|
int len = SendDlgItemMessage(IDC_FRS_CUSTOP_HUBSERVER, CB_GETLBTEXTLEN, index, 0);
|
|
PTSTR pszServer = (PTSTR)calloc(len + 1, sizeof(TCHAR));
|
|
RETURN_OUTOFMEMORY_IF_NULL(pszServer);
|
|
|
|
SendDlgItemMessage(IDC_FRS_CUSTOP_HUBSERVER, CB_GETLBTEXT, index, (LPARAM)pszServer);
|
|
|
|
CCusTopMemberList::iterator i;
|
|
for (i = m_MemberList.begin(); i != m_MemberList.end(); i++)
|
|
{
|
|
if (!lstrcmpi(pszServer, (*i)->m_bstrServer))
|
|
break;
|
|
}
|
|
|
|
free(pszServer);
|
|
|
|
if (i == m_MemberList.end())
|
|
return E_INVALIDARG;
|
|
|
|
*o_ppHubMember = (*i);
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
LRESULT CCustomTopology::OnInitDialog
|
|
(
|
|
UINT uMsg,
|
|
WPARAM wParam,
|
|
LPARAM lParam,
|
|
BOOL& bHandled
|
|
)
|
|
{
|
|
int i = 0;
|
|
int nControlID = 0;
|
|
|
|
//
|
|
// set IDC_FRS_CUSTOP_TOPOLOGYPREF
|
|
//
|
|
nControlID = IDC_FRS_CUSTOP_TOPOLOGYPREF;
|
|
for (i = 0; i < 4; i++)
|
|
{
|
|
CComBSTR bstrTopologyPref;
|
|
LoadStringFromResource(g_TopologyPref[i].nStringID, &bstrTopologyPref);
|
|
SendDlgItemMessage(nControlID, CB_INSERTSTRING, i, (LPARAM)(BSTR)bstrTopologyPref);
|
|
if (!lstrcmpi(m_bstrTopologyPref, g_TopologyPref[i].pszTopologyPref))
|
|
{
|
|
SendDlgItemMessage(nControlID, CB_SETCURSEL, i, 0);
|
|
::EnableWindow(GetDlgItem(IDC_FRS_CUSTOP_REBUILD), (0 != lstrcmpi(m_bstrTopologyPref, FRS_RSTOPOLOGYPREF_CUSTOM)));
|
|
}
|
|
}
|
|
|
|
//
|
|
// set IDC_FRS_CUSTOP_HUBSERVER
|
|
//
|
|
nControlID = IDC_FRS_CUSTOP_HUBSERVER;
|
|
int index = 0;
|
|
CCusTopMemberList::iterator itMem;
|
|
for (i = 0, itMem = m_MemberList.begin(); itMem != m_MemberList.end(); i++, itMem++)
|
|
{
|
|
SendDlgItemMessage(nControlID, CB_INSERTSTRING, i, (LPARAM)(BSTR)(*itMem)->m_bstrServer);
|
|
if (!lstrcmpi(m_bstrHubMemberDN, (*itMem)->m_bstrMemberDN))
|
|
index = i;
|
|
}
|
|
SendDlgItemMessage(nControlID, CB_SETCURSEL, index, 0);
|
|
|
|
if (lstrcmpi(m_bstrTopologyPref, FRS_RSTOPOLOGYPREF_HUBSPOKE))
|
|
{
|
|
MyShowWindow(GetDlgItem(IDC_FRS_CUSTOP_HUBSERVER_LABEL), FALSE);
|
|
MyShowWindow(GetDlgItem(IDC_FRS_CUSTOP_HUBSERVER), FALSE);
|
|
}
|
|
|
|
//
|
|
// set IDC_FRS_CUSTOP_CONNECTIONS
|
|
//
|
|
nControlID = IDC_FRS_CUSTOP_CONNECTIONS;
|
|
HWND hwndControl = GetDlgItem(nControlID);
|
|
AddLVColumns(hwndControl, IDS_FRS_CUSTOP_COL_ENABLE, NUM_OF_FRS_CUSTOP_COLUMNS);
|
|
ListView_SetExtendedListViewStyle(hwndControl, LVS_EX_FULLROWSELECT | LVS_EX_CHECKBOXES);
|
|
|
|
CCusTopConnectionList::iterator itConn;
|
|
for (itConn = m_ConnectionList.begin(); itConn != m_ConnectionList.end(); itConn++)
|
|
_InsertConnection(*itConn);
|
|
|
|
ListView_SortItems( hwndControl,
|
|
ConnectionsListCompareProc,
|
|
(LPARAM)g_FRS_CUSTOP_Last_SortColumn);
|
|
|
|
_EnableButtonsForConnectionList();
|
|
|
|
return TRUE; // Let the system set the focus
|
|
}
|
|
|
|
void CCustomTopology::_EnableButtonsForConnectionList()
|
|
{
|
|
//
|
|
// enable New/Delete/Schedule/Advanced buttons accordingly
|
|
//
|
|
HWND hwndList = GetDlgItem(IDC_FRS_CUSTOP_CONNECTIONS);
|
|
::EnableWindow(GetDlgItem(IDC_FRS_CUSTOP_ADVANCED), (ListView_GetItemCount(hwndList) >= 1));
|
|
|
|
int nCount = ListView_GetSelectedCount(hwndList);
|
|
::EnableWindow(GetDlgItem(IDC_FRS_CUSTOP_SCHEDULE), (nCount >= 1));
|
|
|
|
int index = SendDlgItemMessage(IDC_FRS_CUSTOP_TOPOLOGYPREF, CB_GETCURSEL, 0, 0);
|
|
if (3 == index) // bCustomTopology
|
|
{
|
|
::EnableWindow(GetDlgItem(IDC_FRS_CUSTOP_CONNECTIONS_NEW), TRUE);
|
|
::EnableWindow(GetDlgItem(IDC_FRS_CUSTOP_CONNECTIONS_DELETE), (nCount >= 1));
|
|
} else
|
|
{
|
|
::EnableWindow(GetDlgItem(IDC_FRS_CUSTOP_CONNECTIONS_NEW), FALSE);
|
|
::EnableWindow(GetDlgItem(IDC_FRS_CUSTOP_CONNECTIONS_DELETE), FALSE);
|
|
}
|
|
}
|
|
|
|
/*++
|
|
This function is called when a user clicks the ? in the top right of a property sheet
|
|
and then clciks a control, or when they hit F1 in a control.
|
|
--*/
|
|
LRESULT CCustomTopology::OnCtxHelp(
|
|
IN UINT i_uMsg,
|
|
IN WPARAM i_wParam,
|
|
IN LPARAM i_lParam,
|
|
IN OUT BOOL& io_bHandled
|
|
)
|
|
{
|
|
LPHELPINFO lphi = (LPHELPINFO) i_lParam;
|
|
if (!lphi || lphi->iContextType != HELPINFO_WINDOW || lphi->iCtrlId < 0)
|
|
return FALSE;
|
|
|
|
::WinHelp((HWND)(lphi->hItemHandle),
|
|
DFS_CTX_HELP_FILE,
|
|
HELP_WM_HELP,
|
|
(DWORD_PTR)(PVOID)g_aHelpIDs_IDD_FRS_CUSTOP);
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
/*++
|
|
This function handles "What's This" help when a user right clicks the control
|
|
--*/
|
|
LRESULT CCustomTopology::OnCtxMenuHelp(
|
|
IN UINT i_uMsg,
|
|
IN WPARAM i_wParam,
|
|
IN LPARAM i_lParam,
|
|
IN OUT BOOL& io_bHandled
|
|
)
|
|
{
|
|
::WinHelp((HWND)i_wParam,
|
|
DFS_CTX_HELP_FILE,
|
|
HELP_CONTEXTMENU,
|
|
(DWORD_PTR)(PVOID)g_aHelpIDs_IDD_FRS_CUSTOP);
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
BOOL CCustomTopology::_EnableRebuild()
|
|
{
|
|
BOOL bSameTopologyPref = FALSE;
|
|
int index = SendDlgItemMessage(IDC_FRS_CUSTOP_TOPOLOGYPREF, CB_GETCURSEL, 0, 0);
|
|
if (lstrcmpi(FRS_RSTOPOLOGYPREF_CUSTOM, g_TopologyPref[index].pszTopologyPref) &&
|
|
!lstrcmpi(m_bstrTopologyPref, g_TopologyPref[index].pszTopologyPref))
|
|
{
|
|
bSameTopologyPref = TRUE;
|
|
}
|
|
if (!bSameTopologyPref || 0 != lstrcmpi(m_bstrTopologyPref, FRS_RSTOPOLOGYPREF_HUBSPOKE))
|
|
return bSameTopologyPref;
|
|
|
|
BOOL bSameHub = FALSE;
|
|
CCusTopMember* pHubMember = NULL;
|
|
HRESULT hr = _GetHubMember(&pHubMember);
|
|
if (SUCCEEDED(hr))
|
|
bSameHub = (!lstrcmpi(m_bstrHubMemberDN, pHubMember->m_bstrMemberDN));
|
|
|
|
return bSameHub;
|
|
}
|
|
|
|
LRESULT CCustomTopology::OnTopologyPref
|
|
(
|
|
WORD wNotifyCode,
|
|
WORD wID,
|
|
HWND hWndCtl,
|
|
BOOL& bHandled
|
|
)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
|
|
if (CBN_SELCHANGE == wNotifyCode)
|
|
{
|
|
int index = SendDlgItemMessage(wID, CB_GETCURSEL, 0, 0);
|
|
BOOL bCmdShow = (1 == index);
|
|
MyShowWindow(GetDlgItem(IDC_FRS_CUSTOP_HUBSERVER_LABEL), bCmdShow);
|
|
MyShowWindow(GetDlgItem(IDC_FRS_CUSTOP_HUBSERVER), bCmdShow);
|
|
|
|
if (1 == index)
|
|
{
|
|
CCusTopMember* pHubMember = NULL;
|
|
hr = _GetHubMember(&pHubMember);
|
|
if (SUCCEEDED(hr))
|
|
hr = _RebuildConnections(FRS_RSTOPOLOGYPREF_HUBSPOKE, pHubMember);
|
|
} else
|
|
hr = _RebuildConnections(g_TopologyPref[index].pszTopologyPref, NULL);
|
|
|
|
_EnableButtonsForConnectionList();
|
|
|
|
BOOL bSameTopology = _EnableRebuild();
|
|
::EnableWindow(GetDlgItem(IDC_FRS_CUSTOP_REBUILD), bSameTopology);
|
|
}
|
|
|
|
return (SUCCEEDED(hr));
|
|
}
|
|
|
|
LRESULT CCustomTopology::OnHubServer
|
|
(
|
|
WORD wNotifyCode,
|
|
WORD wID,
|
|
HWND hWndCtl,
|
|
BOOL& bHandled
|
|
)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
|
|
if (CBN_SELCHANGE == wNotifyCode)
|
|
{
|
|
CCusTopMember* pHubMember = NULL;
|
|
hr = _GetHubMember(&pHubMember);
|
|
if (SUCCEEDED(hr))
|
|
hr = _RebuildConnections(FRS_RSTOPOLOGYPREF_HUBSPOKE, pHubMember);
|
|
|
|
BOOL bSameTopology = _EnableRebuild();
|
|
::EnableWindow(GetDlgItem(IDC_FRS_CUSTOP_REBUILD), bSameTopology);
|
|
}
|
|
|
|
return (SUCCEEDED(hr));
|
|
}
|
|
|
|
LRESULT CCustomTopology::OnRebuild
|
|
(
|
|
WORD wNotifyCode,
|
|
WORD wID,
|
|
HWND hWndCtl,
|
|
BOOL& bHandled
|
|
)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
|
|
if (!lstrcmpi(m_bstrTopologyPref, FRS_RSTOPOLOGYPREF_HUBSPOKE))
|
|
{
|
|
CCusTopMember* pHubMember = NULL;
|
|
hr = _GetHubMember(&pHubMember);
|
|
if (SUCCEEDED(hr))
|
|
hr = _RebuildConnections(m_bstrTopologyPref, pHubMember);
|
|
} else
|
|
hr = _RebuildConnections(m_bstrTopologyPref, NULL);
|
|
|
|
return (SUCCEEDED(hr));
|
|
}
|
|
|
|
HRESULT CCustomTopology::_RebuildConnections(
|
|
IN BSTR i_bstrTopologyPref,
|
|
IN CCusTopMember* i_pHubMember)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
|
|
//
|
|
// delete all existing connections from list and view
|
|
//
|
|
CCusTopConnectionList::iterator it = m_ConnectionList.begin();
|
|
while (it != m_ConnectionList.end())
|
|
{
|
|
CCusTopConnectionList::iterator itConn = it++;
|
|
|
|
if (CONNECTION_OPTYPE_ADD == (*itConn)->m_opType)
|
|
{
|
|
delete (*itConn);
|
|
m_ConnectionList.erase(itConn);
|
|
} else
|
|
{
|
|
(*itConn)->m_opType = CONNECTION_OPTYPE_DEL;
|
|
}
|
|
}
|
|
|
|
ListView_DeleteAllItems(GetDlgItem(IDC_FRS_CUSTOP_CONNECTIONS));
|
|
|
|
//
|
|
// re-create the connections as specified
|
|
//
|
|
if (m_MemberList.size() == 1)
|
|
return hr;
|
|
|
|
CCusTopMemberList::iterator n1;
|
|
CCusTopMemberList::iterator n2;
|
|
CCusTopConnection Conn;
|
|
if (!lstrcmpi(i_bstrTopologyPref, FRS_RSTOPOLOGYPREF_RING))
|
|
{
|
|
//
|
|
// sort member list, such that members on the same site will be neighbors
|
|
//
|
|
m_MemberList.sort(CusTopMemberCompareSite());
|
|
|
|
CCusTopMemberList::iterator head;
|
|
|
|
head = n1 = m_MemberList.begin();
|
|
while (n1 != m_MemberList.end())
|
|
{
|
|
n2 = n1++;
|
|
if (n1 == m_MemberList.end())
|
|
{
|
|
if (m_MemberList.size() == 2)
|
|
break;
|
|
|
|
n1 = head;
|
|
}
|
|
|
|
hr = Conn.Init((*n1)->m_bstrMemberDN, (*n1)->m_bstrServer, (*n1)->m_bstrSite,
|
|
(*n2)->m_bstrMemberDN, (*n2)->m_bstrServer, (*n2)->m_bstrSite,
|
|
TRUE, CONNECTION_OPTYPE_ADD);
|
|
BREAK_IF_FAILED(hr);
|
|
hr = _AddToConnectionListAndView(&Conn);
|
|
BREAK_IF_FAILED(hr);
|
|
|
|
hr = Conn.Init((*n2)->m_bstrMemberDN, (*n2)->m_bstrServer, (*n2)->m_bstrSite,
|
|
(*n1)->m_bstrMemberDN, (*n1)->m_bstrServer, (*n1)->m_bstrSite,
|
|
TRUE, CONNECTION_OPTYPE_ADD);
|
|
BREAK_IF_FAILED(hr);
|
|
hr = _AddToConnectionListAndView(&Conn);
|
|
BREAK_IF_FAILED(hr);
|
|
|
|
if (n1 == head)
|
|
break;
|
|
}
|
|
} else if (!lstrcmpi(i_bstrTopologyPref, FRS_RSTOPOLOGYPREF_HUBSPOKE))
|
|
{
|
|
for (n1 = m_MemberList.begin(); n1 != m_MemberList.end(); n1++)
|
|
{
|
|
if (!lstrcmpi((*n1)->m_bstrMemberDN, i_pHubMember->m_bstrMemberDN))
|
|
continue;
|
|
|
|
hr = Conn.Init((*n1)->m_bstrMemberDN, (*n1)->m_bstrServer, (*n1)->m_bstrSite,
|
|
i_pHubMember->m_bstrMemberDN, i_pHubMember->m_bstrServer, i_pHubMember->m_bstrSite,
|
|
TRUE, CONNECTION_OPTYPE_ADD);
|
|
BREAK_IF_FAILED(hr);
|
|
hr = _AddToConnectionListAndView(&Conn);
|
|
BREAK_IF_FAILED(hr);
|
|
|
|
hr = Conn.Init(i_pHubMember->m_bstrMemberDN, i_pHubMember->m_bstrServer, i_pHubMember->m_bstrSite,
|
|
(*n1)->m_bstrMemberDN, (*n1)->m_bstrServer, (*n1)->m_bstrSite,
|
|
TRUE, CONNECTION_OPTYPE_ADD);
|
|
BREAK_IF_FAILED(hr);
|
|
hr = _AddToConnectionListAndView(&Conn);
|
|
BREAK_IF_FAILED(hr);
|
|
}
|
|
} else if (!lstrcmpi(i_bstrTopologyPref, FRS_RSTOPOLOGYPREF_FULLMESH))
|
|
{
|
|
for (n1 = m_MemberList.begin(); n1 != m_MemberList.end(); n1++)
|
|
{
|
|
for (n2 = m_MemberList.begin(); n2 != m_MemberList.end(); n2++)
|
|
{
|
|
if (!lstrcmpi((*n1)->m_bstrMemberDN, (*n2)->m_bstrMemberDN))
|
|
continue;
|
|
|
|
hr = Conn.Init((*n1)->m_bstrMemberDN, (*n1)->m_bstrServer, (*n1)->m_bstrSite,
|
|
(*n2)->m_bstrMemberDN, (*n2)->m_bstrServer, (*n2)->m_bstrSite,
|
|
TRUE, CONNECTION_OPTYPE_ADD);
|
|
BREAK_IF_FAILED(hr);
|
|
hr = _AddToConnectionListAndView(&Conn);
|
|
BREAK_IF_FAILED(hr);
|
|
}
|
|
BREAK_IF_FAILED(hr);
|
|
}
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
HRESULT CCustomTopology::_SetConnectionState(CCusTopConnection *pConn)
|
|
{
|
|
HWND hwnd = GetDlgItem(IDC_FRS_CUSTOP_CONNECTIONS);
|
|
int nIndex = ListView_GetNextItem(hwnd, -1, LVNI_ALL);
|
|
while (-1 != nIndex)
|
|
{
|
|
if (pConn == (CCusTopConnection *)GetListViewItemData(hwnd, nIndex))
|
|
break;
|
|
|
|
nIndex = ListView_GetNextItem(hwnd, nIndex, LVNI_ALL);
|
|
}
|
|
|
|
if (-1 != nIndex)
|
|
{
|
|
ListView_SetCheckState(hwnd, nIndex, pConn->m_bStateNew);
|
|
ListView_Update(hwnd, nIndex);
|
|
}
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT CCustomTopology::_InsertConnection(CCusTopConnection *pConn)
|
|
{
|
|
RETURN_INVALIDARG_IF_NULL(pConn);
|
|
|
|
HWND hwndControl = GetDlgItem(IDC_FRS_CUSTOP_CONNECTIONS);
|
|
|
|
LVITEM lvItem;
|
|
|
|
lvItem.mask = LVIF_TEXT | LVIF_PARAM;
|
|
lvItem.lParam = (LPARAM)pConn;
|
|
lvItem.pszText = _T("");
|
|
lvItem.iSubItem = 0;
|
|
int iItemIndex = ListView_InsertItem(hwndControl, &lvItem);
|
|
|
|
lvItem.mask = LVIF_TEXT;
|
|
lvItem.iItem = iItemIndex;
|
|
lvItem.pszText = pConn->m_bstrFromServer;
|
|
lvItem.iSubItem = 1;
|
|
ListView_SetItem(hwndControl, &lvItem);
|
|
|
|
lvItem.mask = LVIF_TEXT;
|
|
lvItem.iItem = iItemIndex;
|
|
lvItem.pszText = pConn->m_bstrToServer;
|
|
lvItem.iSubItem = 2;
|
|
ListView_SetItem(hwndControl, &lvItem);
|
|
|
|
lvItem.mask = LVIF_TEXT;
|
|
lvItem.iItem = iItemIndex;
|
|
lvItem.pszText = pConn->m_bstrFromSite;
|
|
lvItem.iSubItem = 3;
|
|
ListView_SetItem(hwndControl, &lvItem);
|
|
|
|
lvItem.mask = LVIF_TEXT;
|
|
lvItem.iItem = iItemIndex;
|
|
lvItem.pszText = pConn->m_bstrToSite;
|
|
lvItem.iSubItem = 4;
|
|
ListView_SetItem(hwndControl, &lvItem);
|
|
ListView_SetCheckState(hwndControl, iItemIndex, pConn->m_bStateNew);
|
|
|
|
ListView_Update(hwndControl, iItemIndex);
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
LRESULT CCustomTopology::OnConnectionsNew
|
|
(
|
|
WORD wNotifyCode,
|
|
WORD wID,
|
|
HWND hWndCtl,
|
|
BOOL& bHandled
|
|
)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
|
|
CNewConnections NewConnDlg;
|
|
|
|
// Building the RING topology might have sorted the members on sites,
|
|
// we need to re-sort it on server alphabetically before we give it to the New Connections dialog
|
|
_SortMemberList();
|
|
|
|
hr = NewConnDlg.Initialize(&m_MemberList);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
hr = NewConnDlg.DoModal();
|
|
if (S_OK == hr)
|
|
{
|
|
CCusTopConnectionList* pNewConnectionList = NULL;
|
|
hr = NewConnDlg.get_NewConnections(&pNewConnectionList);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
HWND hwnd = GetDlgItem(IDC_FRS_CUSTOP_CONNECTIONS);
|
|
int nCount = ListView_GetItemCount(hwnd);
|
|
|
|
CCusTopConnectionList::iterator it;
|
|
for (it = pNewConnectionList->begin(); it != pNewConnectionList->end(); it++)
|
|
{
|
|
hr = _AddToConnectionListAndView(*it);
|
|
BREAK_IF_FAILED(hr);
|
|
}
|
|
|
|
if (ListView_GetItemCount(hwnd) > nCount)
|
|
ListView_SortItems(hwnd, ConnectionsListCompareProc, (LPARAM)g_FRS_CUSTOP_Last_SortColumn);
|
|
}
|
|
}
|
|
}
|
|
|
|
// if FAILED, display msg?
|
|
|
|
return (SUCCEEDED(hr));
|
|
}
|
|
|
|
LRESULT CCustomTopology::OnConnectionsDelete
|
|
(
|
|
WORD wNotifyCode,
|
|
WORD wID,
|
|
HWND hWndCtl,
|
|
BOOL& bHandled
|
|
)
|
|
{
|
|
CCusTopConnection* pConn;
|
|
HWND hwnd = GetDlgItem(IDC_FRS_CUSTOP_CONNECTIONS);
|
|
int nIndex = ListView_GetNextItem(hwnd, -1, LVNI_ALL | LVNI_SELECTED);
|
|
while (-1 != nIndex &&
|
|
(pConn = (CCusTopConnection *)GetListViewItemData(hwnd, nIndex)))
|
|
{
|
|
_RemoveFromConnectionList(pConn);
|
|
|
|
ListView_DeleteItem(hwnd, nIndex);
|
|
|
|
nIndex = ListView_GetNextItem(hwnd, -1, LVNI_ALL | LVNI_SELECTED);
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
HRESULT CCustomTopology::_InitScheduleOnSelectedConnections()
|
|
{
|
|
HRESULT hr = S_OK;
|
|
int nIndex = -1;
|
|
CCusTopConnection* pConn = NULL;
|
|
HWND hwnd = GetDlgItem(IDC_FRS_CUSTOP_CONNECTIONS);
|
|
|
|
while ( -1 != (nIndex = ListView_GetNextItem(hwnd, nIndex, LVNI_ALL | LVNI_SELECTED)) &&
|
|
NULL != (pConn = (CCusTopConnection *)GetListViewItemData(hwnd, nIndex)))
|
|
{
|
|
if (!pConn->m_pScheduleNew)
|
|
{
|
|
if (pConn->m_pScheduleOld)
|
|
{
|
|
hr = CopySchedule(pConn->m_pScheduleOld, &pConn->m_pScheduleNew);
|
|
RETURN_IF_FAILED(hr);
|
|
} else
|
|
{
|
|
if (CONNECTION_OPTYPE_OTHERS == pConn->m_opType)
|
|
{
|
|
//
|
|
// read schedule on an existing connection for the very first time
|
|
//
|
|
VARIANT var;
|
|
VariantInit(&var);
|
|
hr = m_piReplicaSet->GetConnectionScheduleEx(pConn->m_bstrFromMemberDN, pConn->m_bstrToMemberDN, &var);
|
|
RETURN_IF_FAILED(hr);
|
|
|
|
hr = VariantToSchedule(&var, &pConn->m_pScheduleOld);
|
|
|
|
VariantClear(&var);
|
|
|
|
if (SUCCEEDED(hr))
|
|
hr = CopySchedule(pConn->m_pScheduleOld, &pConn->m_pScheduleNew);
|
|
|
|
RETURN_IF_FAILED(hr);
|
|
} else
|
|
{ // must be ADD operation
|
|
hr = GetDefaultSchedule(&pConn->m_pScheduleNew);
|
|
RETURN_IF_FAILED(hr);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
HRESULT CCustomTopology::_UpdateScheduleOnSelectedConnections(IN SCHEDULE* i_pSchedule)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
int nIndex = -1;
|
|
CCusTopConnection* pConn = NULL;
|
|
HWND hwnd = GetDlgItem(IDC_FRS_CUSTOP_CONNECTIONS);
|
|
|
|
while ( -1 != (nIndex = ListView_GetNextItem(hwnd, nIndex, LVNI_ALL | LVNI_SELECTED)) &&
|
|
NULL != (pConn = (CCusTopConnection *)GetListViewItemData(hwnd, nIndex)))
|
|
{
|
|
if (pConn->m_pScheduleNew)
|
|
{
|
|
free(pConn->m_pScheduleNew);
|
|
pConn->m_pScheduleNew = NULL;
|
|
}
|
|
|
|
hr = CopySchedule(i_pSchedule, &pConn->m_pScheduleNew);
|
|
BREAK_IF_FAILED(hr);
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
LRESULT CCustomTopology::OnSchedule
|
|
(
|
|
WORD wNotifyCode,
|
|
WORD wID,
|
|
HWND hWndCtl,
|
|
BOOL& bHandled
|
|
)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
|
|
HWND hwnd = GetDlgItem(IDC_FRS_CUSTOP_CONNECTIONS);
|
|
int nCount = ListView_GetSelectedCount(hwnd);
|
|
if (nCount < 1)
|
|
{
|
|
DisplayMessageBoxWithOK(IDS_FRS_CUSTOP_NOSELECTION);
|
|
return FALSE;
|
|
}
|
|
|
|
do {
|
|
//
|
|
// get schedule info on each selected connections
|
|
//
|
|
hr = _InitScheduleOnSelectedConnections();
|
|
BREAK_IF_FAILED(hr);
|
|
|
|
//
|
|
// get schedule of the first selected item
|
|
//
|
|
int nIndex = ListView_GetNextItem(hwnd, -1, LVNI_ALL | LVNI_SELECTED);
|
|
if (-1 == nIndex)
|
|
{
|
|
hr = E_INVALIDARG;
|
|
break;
|
|
}
|
|
|
|
CCusTopConnection* pConn = (CCusTopConnection *)GetListViewItemData(hwnd, nIndex);
|
|
if (!pConn)
|
|
{
|
|
hr = E_INVALIDARG;
|
|
break;
|
|
}
|
|
|
|
SCHEDULE* pSchedule = NULL;
|
|
hr = CopySchedule(pConn->m_pScheduleNew, &pSchedule);
|
|
BREAK_IF_FAILED(hr);
|
|
|
|
hr = InvokeScheduleDlg(m_hWnd, pSchedule);
|
|
|
|
if (S_OK == hr)
|
|
hr = _UpdateScheduleOnSelectedConnections(pSchedule);
|
|
|
|
free(pSchedule);
|
|
|
|
} while (0);
|
|
|
|
if (FAILED(hr))
|
|
DisplayMessageBoxForHR(hr);
|
|
|
|
return (SUCCEEDED(hr));
|
|
}
|
|
|
|
LRESULT CCustomTopology::OnAdvanced
|
|
(
|
|
WORD wNotifyCode,
|
|
WORD wID,
|
|
HWND hWndCtl,
|
|
BOOL& bHandled
|
|
)
|
|
{
|
|
HWND hwnd = GetDlgItem(IDC_FRS_CUSTOP_CONNECTIONS);
|
|
|
|
//
|
|
// get ToServer of the first selected connection
|
|
//
|
|
PTSTR pszToServer = NULL;
|
|
int nIndex = ListView_GetNextItem(hwnd, -1, LVNI_ALL | LVNI_SELECTED);
|
|
if (-1 != nIndex)
|
|
{
|
|
CCusTopConnection* pConn = (CCusTopConnection *)GetListViewItemData(hwnd, nIndex);
|
|
if (pConn)
|
|
{
|
|
pszToServer = pConn->m_bstrToServer;
|
|
}
|
|
}
|
|
|
|
// Building the RING topology might have sorted the members on sites,
|
|
// we need to re-sort it on server alphabetically before we give it to the Advanced dialog
|
|
_SortMemberList();
|
|
|
|
CFRSAdvanced frsAdvancedDlg;
|
|
HRESULT hr = frsAdvancedDlg.Initialize(&m_MemberList, &m_ConnectionList, pszToServer);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
hr = frsAdvancedDlg.DoModal();
|
|
}
|
|
|
|
if (FAILED(hr))
|
|
DisplayMessageBoxForHR(hr);
|
|
|
|
return (SUCCEEDED(hr));
|
|
}
|
|
|
|
LRESULT CCustomTopology::OnOK
|
|
(
|
|
WORD wNotifyCode,
|
|
WORD wID,
|
|
HWND hWndCtl,
|
|
BOOL& bHandled
|
|
)
|
|
{
|
|
CWaitCursor wait;
|
|
BOOL bValidInput = FALSE;
|
|
int idString = 0;
|
|
HRESULT hr = S_OK;
|
|
|
|
do {
|
|
//
|
|
// if changed, update TopologyPref
|
|
//
|
|
int index = SendDlgItemMessage(IDC_FRS_CUSTOP_TOPOLOGYPREF, CB_GETCURSEL, 0, 0);
|
|
if (0 != lstrcmpi(m_bstrTopologyPref, g_TopologyPref[index].pszTopologyPref))
|
|
{
|
|
hr = m_piReplicaSet->put_TopologyPref(g_TopologyPref[index].pszTopologyPref);
|
|
BREAK_IF_FAILED(hr);
|
|
}
|
|
|
|
//
|
|
// if changed, update HubServer
|
|
//
|
|
if (!lstrcmpi(m_bstrTopologyPref, FRS_RSTOPOLOGYPREF_HUBSPOKE) &&
|
|
0 != lstrcmpi(g_TopologyPref[index].pszTopologyPref, FRS_RSTOPOLOGYPREF_HUBSPOKE))
|
|
{
|
|
hr = m_piReplicaSet->put_HubMemberDN(NULL);
|
|
BREAK_IF_FAILED(hr);
|
|
} else if (!lstrcmpi(g_TopologyPref[index].pszTopologyPref, FRS_RSTOPOLOGYPREF_HUBSPOKE))
|
|
{
|
|
CCusTopMember* pHubMember = NULL;
|
|
hr = _GetHubMember(&pHubMember);
|
|
BREAK_IF_FAILED(hr);
|
|
|
|
if (0 != lstrcmpi(m_bstrHubMemberDN, pHubMember->m_bstrMemberDN))
|
|
hr = m_piReplicaSet->put_HubMemberDN(pHubMember->m_bstrMemberDN);
|
|
|
|
BREAK_IF_FAILED(hr);
|
|
}
|
|
|
|
//
|
|
// if changed, update connections
|
|
//
|
|
CCusTopConnection* pConn = NULL;
|
|
HWND hwnd = GetDlgItem(IDC_FRS_CUSTOP_CONNECTIONS);
|
|
index = -1;
|
|
while (-1 != (index = ListView_GetNextItem(hwnd, index, LVNI_ALL)))
|
|
{
|
|
pConn = (CCusTopConnection *)GetListViewItemData(hwnd, index);
|
|
if (pConn)
|
|
pConn->m_bStateNew = ListView_GetCheckState(hwnd, index);
|
|
}
|
|
|
|
hr = _MakeConnections();
|
|
BREAK_IF_FAILED(hr);
|
|
|
|
bValidInput = TRUE;
|
|
|
|
} while (0);
|
|
|
|
if (FAILED(hr))
|
|
{
|
|
DisplayMessageBoxForHR(hr);
|
|
return FALSE;
|
|
} else if (bValidInput)
|
|
{
|
|
EndDialog(S_OK);
|
|
return TRUE;
|
|
}
|
|
else
|
|
{
|
|
if (idString)
|
|
DisplayMessageBoxWithOK(idString);
|
|
return FALSE;
|
|
}
|
|
}
|
|
|
|
LRESULT CCustomTopology::OnCancel
|
|
(
|
|
WORD wNotifyCode,
|
|
WORD wID,
|
|
HWND hWndCtl,
|
|
BOOL& bHandled
|
|
)
|
|
{
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Called OnCancel. Ends the dialog with S_FALSE;
|
|
|
|
*/
|
|
EndDialog(S_FALSE);
|
|
return(true);
|
|
}
|
|
|
|
int CALLBACK ConnectionsListCompareProc(
|
|
IN LPARAM lParam1,
|
|
IN LPARAM lParam2,
|
|
IN LPARAM lParamColumn)
|
|
{
|
|
CCusTopConnection* pItem1 = (CCusTopConnection *)lParam1;
|
|
CCusTopConnection* pItem2 = (CCusTopConnection *)lParam2;
|
|
int iResult = 0;
|
|
|
|
if (pItem1 && pItem2)
|
|
{
|
|
g_FRS_CUSTOP_Last_SortColumn = lParamColumn;
|
|
|
|
switch (lParamColumn)
|
|
{
|
|
case 0: // Sort by bStateNew.
|
|
iResult = pItem1->m_bStateNew - pItem2->m_bStateNew;
|
|
break;
|
|
case 1: // Sort by From Server.
|
|
iResult = lstrcmpi(pItem1->m_bstrFromServer, pItem2->m_bstrFromServer);
|
|
break;
|
|
case 2: // Sort by To Server.
|
|
iResult = lstrcmpi(pItem1->m_bstrToServer, pItem2->m_bstrToServer);
|
|
break;
|
|
case 3: // Sort by From Site.
|
|
iResult = lstrcmpi(pItem1->m_bstrFromSite, pItem2->m_bstrFromSite);
|
|
break;
|
|
case 4: // Sort by To Site.
|
|
iResult = lstrcmpi(pItem1->m_bstrToSite, pItem2->m_bstrToSite);
|
|
break;
|
|
default:
|
|
iResult = 0;
|
|
break;
|
|
}
|
|
}
|
|
|
|
return(iResult);
|
|
}
|
|
|
|
LRESULT
|
|
CCustomTopology::OnNotify(
|
|
IN UINT i_uMsg,
|
|
IN WPARAM i_wParam,
|
|
IN LPARAM i_lParam,
|
|
IN OUT BOOL& io_bHandled
|
|
)
|
|
{
|
|
NM_LISTVIEW* pNMListView = (NM_LISTVIEW*)i_lParam;
|
|
io_bHandled = FALSE; // So that the base class gets this notify too
|
|
|
|
if (IDC_FRS_CUSTOP_CONNECTIONS == pNMListView->hdr.idFrom)
|
|
{
|
|
HWND hwndList = GetDlgItem(IDC_FRS_CUSTOP_CONNECTIONS);
|
|
if (LVN_COLUMNCLICK == pNMListView->hdr.code)
|
|
{
|
|
// sort items
|
|
ListView_SortItems( hwndList,
|
|
ConnectionsListCompareProc,
|
|
(LPARAM)(pNMListView->iSubItem));
|
|
io_bHandled = TRUE;
|
|
} else if (LVN_ITEMCHANGED == pNMListView->hdr.code)
|
|
{
|
|
_EnableButtonsForConnectionList();
|
|
}
|
|
}
|
|
|
|
return io_bHandled;
|
|
}
|
|
|
|
//
|
|
// Update the connection objects in the DS
|
|
//
|
|
HRESULT CCustomTopology::_MakeConnections()
|
|
{
|
|
HRESULT hr = S_OK;
|
|
|
|
CCusTopConnectionList::iterator it;
|
|
for (it = m_ConnectionList.begin(); it != m_ConnectionList.end(); it++)
|
|
{
|
|
switch ((*it)->m_opType)
|
|
{
|
|
case CONNECTION_OPTYPE_ADD:
|
|
hr = m_piReplicaSet->AddConnection(
|
|
(*it)->m_bstrFromMemberDN,
|
|
(*it)->m_bstrToMemberDN,
|
|
(*it)->m_bStateNew,
|
|
(*it)->m_bSyncImmediatelyNew,
|
|
(*it)->m_nPriorityNew,
|
|
NULL
|
|
);
|
|
BREAK_IF_FAILED(hr);
|
|
|
|
if ((*it)->m_pScheduleNew)
|
|
{
|
|
VARIANT var;
|
|
VariantInit(&var);
|
|
hr = ScheduleToVariant((*it)->m_pScheduleNew, &var);
|
|
BREAK_IF_FAILED(hr);
|
|
|
|
hr = m_piReplicaSet->SetConnectionScheduleEx(
|
|
(*it)->m_bstrFromMemberDN,
|
|
(*it)->m_bstrToMemberDN,
|
|
&var);
|
|
VariantClear(&var);
|
|
}
|
|
break;
|
|
|
|
case CONNECTION_OPTYPE_DEL:
|
|
hr = m_piReplicaSet->RemoveConnectionEx(
|
|
(*it)->m_bstrFromMemberDN,
|
|
(*it)->m_bstrToMemberDN
|
|
);
|
|
break;
|
|
|
|
default:
|
|
if ((*it)->m_bStateNew != (*it)->m_bStateOld)
|
|
{
|
|
hr = m_piReplicaSet->EnableConnectionEx(
|
|
(*it)->m_bstrFromMemberDN,
|
|
(*it)->m_bstrToMemberDN,
|
|
(*it)->m_bStateNew
|
|
);
|
|
BREAK_IF_FAILED(hr);
|
|
}
|
|
|
|
if ((*it)->m_bSyncImmediatelyNew != (*it)->m_bSyncImmediatelyOld ||
|
|
(*it)->m_nPriorityNew != (*it)->m_nPriorityOld)
|
|
{
|
|
hr = m_piReplicaSet->SetConnectionOptionsEx(
|
|
(*it)->m_bstrFromMemberDN,
|
|
(*it)->m_bstrToMemberDN,
|
|
(*it)->m_bSyncImmediatelyNew,
|
|
(*it)->m_nPriorityNew
|
|
);
|
|
BREAK_IF_FAILED(hr);
|
|
}
|
|
|
|
if (S_OK == CompareSchedules((*it)->m_pScheduleNew, (*it)->m_pScheduleOld))
|
|
break; // no change on shcedule
|
|
|
|
if ((*it)->m_pScheduleNew)
|
|
{
|
|
VARIANT var;
|
|
VariantInit(&var);
|
|
hr = ScheduleToVariant((*it)->m_pScheduleNew, &var);
|
|
BREAK_IF_FAILED(hr);
|
|
|
|
hr = m_piReplicaSet->SetConnectionScheduleEx(
|
|
(*it)->m_bstrFromMemberDN,
|
|
(*it)->m_bstrToMemberDN,
|
|
&var);
|
|
VariantClear(&var);
|
|
}
|
|
|
|
break;
|
|
}
|
|
|
|
BREAK_IF_FAILED(hr);
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
//////////////////////////////////////////////////////////
|
|
//
|
|
//
|
|
|
|
void FreeCusTopMembers(CCusTopMemberList* pList)
|
|
{
|
|
if (pList && !pList->empty())
|
|
{
|
|
for (CCusTopMemberList::iterator i = pList->begin(); i != pList->end(); i++)
|
|
delete (*i);
|
|
|
|
pList->clear();
|
|
}
|
|
}
|
|
|
|
void FreeCusTopConnections(CCusTopConnectionList* pList)
|
|
{
|
|
if (pList && !pList->empty())
|
|
{
|
|
for (CCusTopConnectionList::iterator i = pList->begin(); i != pList->end(); i++)
|
|
delete (*i);
|
|
|
|
pList->clear();
|
|
}
|
|
}
|
|
|
|
//////////////////////////////////////////////////////////
|
|
//
|
|
// CCusTopMember
|
|
//
|
|
CCusTopMember::~CCusTopMember()
|
|
{
|
|
_Reset();
|
|
}
|
|
|
|
HRESULT CCusTopMember::Init(BSTR bstrMemberDN, BSTR bstrServer, BSTR bstrSite)
|
|
{
|
|
RETURN_INVALIDARG_IF_NULL(bstrMemberDN);
|
|
RETURN_INVALIDARG_IF_NULL(bstrServer);
|
|
RETURN_INVALIDARG_IF_NULL(bstrSite);
|
|
|
|
_Reset();
|
|
|
|
HRESULT hr = S_OK;
|
|
do {
|
|
m_bstrMemberDN = bstrMemberDN;
|
|
BREAK_OUTOFMEMORY_IF_NULL((BSTR)m_bstrMemberDN, &hr);
|
|
m_bstrServer = bstrServer;
|
|
BREAK_OUTOFMEMORY_IF_NULL((BSTR)m_bstrServer, &hr);
|
|
m_bstrSite = bstrSite;
|
|
BREAK_OUTOFMEMORY_IF_NULL((BSTR)m_bstrSite, &hr);
|
|
} while (0);
|
|
|
|
if (FAILED(hr))
|
|
_Reset();
|
|
|
|
return hr;
|
|
}
|
|
|
|
void CCusTopMember::_Reset()
|
|
{
|
|
m_bstrMemberDN.Empty();
|
|
m_bstrServer.Empty();
|
|
m_bstrSite.Empty();
|
|
}
|
|
|
|
//////////////////////////////////////////////////////////
|
|
//
|
|
// CCusTopConnection
|
|
//
|
|
CCusTopConnection::CCusTopConnection()
|
|
{
|
|
m_bStateNew = m_bStateOld= TRUE;
|
|
m_bSyncImmediatelyNew = m_bSyncImmediatelyOld = FALSE;
|
|
m_nPriorityNew = m_nPriorityOld = PRIORITY_LOW;
|
|
m_pScheduleNew = m_pScheduleOld = NULL;
|
|
m_opType = CONNECTION_OPTYPE_OTHERS;
|
|
}
|
|
|
|
CCusTopConnection::~CCusTopConnection()
|
|
{
|
|
_Reset();
|
|
}
|
|
|
|
HRESULT CCusTopConnection::Init(
|
|
BSTR bstrFromMemberDN, BSTR bstrFromServer, BSTR bstrFromSite,
|
|
BSTR bstrToMemberDN, BSTR bstrToServer, BSTR bstrToSite,
|
|
BOOL bState, // = TRUE
|
|
CONNECTION_OPTYPE opType, // = CONNECTION_OPTYPE_OTHERS
|
|
BOOL bSyncImmediately, // = FALSE,
|
|
Connection_Priority nPriority // = PRIORITY_LOW
|
|
)
|
|
{
|
|
RETURN_INVALIDARG_IF_NULL(bstrFromMemberDN);
|
|
RETURN_INVALIDARG_IF_NULL(bstrFromServer);
|
|
RETURN_INVALIDARG_IF_NULL(bstrFromSite);
|
|
|
|
RETURN_INVALIDARG_IF_NULL(bstrToMemberDN);
|
|
RETURN_INVALIDARG_IF_NULL(bstrToServer);
|
|
RETURN_INVALIDARG_IF_NULL(bstrToSite);
|
|
|
|
_Reset();
|
|
|
|
HRESULT hr = S_OK;
|
|
do {
|
|
m_bstrFromMemberDN = bstrFromMemberDN;
|
|
BREAK_OUTOFMEMORY_IF_NULL((BSTR)m_bstrFromMemberDN, &hr);
|
|
m_bstrFromServer = bstrFromServer;
|
|
BREAK_OUTOFMEMORY_IF_NULL((BSTR)m_bstrFromServer, &hr);
|
|
m_bstrFromSite = bstrFromSite;
|
|
BREAK_OUTOFMEMORY_IF_NULL((BSTR)m_bstrFromSite, &hr);
|
|
|
|
m_bstrToMemberDN = bstrToMemberDN;
|
|
BREAK_OUTOFMEMORY_IF_NULL((BSTR)m_bstrToMemberDN, &hr);
|
|
m_bstrToServer = bstrToServer;
|
|
BREAK_OUTOFMEMORY_IF_NULL((BSTR)m_bstrToServer, &hr);
|
|
m_bstrToSite = bstrToSite;
|
|
BREAK_OUTOFMEMORY_IF_NULL((BSTR)m_bstrToSite, &hr);
|
|
|
|
m_bStateNew = m_bStateOld = bState;
|
|
|
|
m_opType = opType;
|
|
|
|
m_pScheduleOld = m_pScheduleNew = NULL;
|
|
|
|
m_bSyncImmediatelyNew = m_bSyncImmediatelyOld = bSyncImmediately;
|
|
|
|
m_nPriorityNew = m_nPriorityOld = nPriority;
|
|
|
|
} while (0);
|
|
|
|
if (FAILED(hr))
|
|
_Reset();
|
|
|
|
return hr;
|
|
}
|
|
|
|
void CCusTopConnection::_Reset()
|
|
{
|
|
m_bstrFromMemberDN.Empty();
|
|
m_bstrFromServer.Empty();
|
|
m_bstrFromSite.Empty();
|
|
m_bstrToMemberDN.Empty();
|
|
m_bstrToServer.Empty();
|
|
m_bstrToSite.Empty();
|
|
m_bStateOld = TRUE;
|
|
m_bStateNew = TRUE;
|
|
m_opType = CONNECTION_OPTYPE_OTHERS;
|
|
m_bSyncImmediatelyNew = m_bSyncImmediatelyOld = FALSE;
|
|
m_nPriorityNew = m_nPriorityOld = PRIORITY_LOW;
|
|
|
|
if (m_pScheduleOld)
|
|
{
|
|
free(m_pScheduleOld);
|
|
m_pScheduleOld = NULL;
|
|
}
|
|
|
|
if (m_pScheduleNew)
|
|
{
|
|
free(m_pScheduleNew);
|
|
m_pScheduleNew = NULL;
|
|
}
|
|
}
|
|
|
|
HRESULT CCusTopConnection::Copy(CCusTopConnection* pConn)
|
|
{
|
|
if (!pConn || !(pConn->m_bstrFromMemberDN) || !*(pConn->m_bstrFromMemberDN))
|
|
return E_INVALIDARG;
|
|
|
|
_Reset();
|
|
|
|
HRESULT hr = S_OK;
|
|
do {
|
|
m_bstrFromMemberDN = pConn->m_bstrFromMemberDN;
|
|
BREAK_OUTOFMEMORY_IF_NULL((BSTR)m_bstrFromMemberDN, &hr);
|
|
m_bstrFromServer = pConn->m_bstrFromServer;
|
|
BREAK_OUTOFMEMORY_IF_NULL((BSTR)m_bstrFromServer, &hr);
|
|
m_bstrFromSite = pConn->m_bstrFromSite;
|
|
BREAK_OUTOFMEMORY_IF_NULL((BSTR)m_bstrFromSite, &hr);
|
|
|
|
m_bstrToMemberDN = pConn->m_bstrToMemberDN;
|
|
BREAK_OUTOFMEMORY_IF_NULL((BSTR)m_bstrToMemberDN, &hr);
|
|
m_bstrToServer = pConn->m_bstrToServer;
|
|
BREAK_OUTOFMEMORY_IF_NULL((BSTR)m_bstrToServer, &hr);
|
|
m_bstrToSite = pConn->m_bstrToSite;
|
|
BREAK_OUTOFMEMORY_IF_NULL((BSTR)m_bstrToSite, &hr);
|
|
|
|
m_bStateOld = pConn->m_bStateOld;
|
|
m_bStateNew = pConn->m_bStateNew;
|
|
|
|
m_bSyncImmediatelyOld = pConn->m_bSyncImmediatelyOld;
|
|
m_bSyncImmediatelyNew = pConn->m_bSyncImmediatelyNew;
|
|
|
|
m_nPriorityOld = pConn->m_nPriorityOld;
|
|
m_nPriorityNew = pConn->m_nPriorityNew;
|
|
|
|
m_opType = pConn->m_opType;
|
|
|
|
if (pConn->m_pScheduleOld)
|
|
{
|
|
hr = CopySchedule(pConn->m_pScheduleOld, &m_pScheduleOld);
|
|
BREAK_IF_FAILED(hr);
|
|
}
|
|
|
|
if (pConn->m_pScheduleNew)
|
|
{
|
|
hr = CopySchedule(pConn->m_pScheduleNew, &m_pScheduleNew);
|
|
BREAK_IF_FAILED(hr);
|
|
}
|
|
} while (0);
|
|
|
|
if (FAILED(hr))
|
|
_Reset();
|
|
|
|
return hr;
|
|
}
|
|
|
|
HRESULT CCustomTopology::_AddToConnectionListAndView(CCusTopConnection* pConn)
|
|
{
|
|
RETURN_INVALIDARG_IF_NULL(pConn);
|
|
|
|
BOOL bFound = FALSE;
|
|
|
|
CCusTopConnectionList::iterator it;
|
|
for (it = m_ConnectionList.begin(); it != m_ConnectionList.end(); it++)
|
|
{
|
|
if (!lstrcmpi((*it)->m_bstrFromMemberDN, pConn->m_bstrFromMemberDN) &&
|
|
!lstrcmpi((*it)->m_bstrToMemberDN, pConn->m_bstrToMemberDN))
|
|
{
|
|
bFound = TRUE;
|
|
break;
|
|
}
|
|
}
|
|
|
|
HRESULT hr = S_OK;
|
|
if (!bFound)
|
|
{
|
|
CCusTopConnection* pNew = new CCusTopConnection;
|
|
RETURN_OUTOFMEMORY_IF_NULL(pNew);
|
|
|
|
hr = pNew->Copy(pConn);
|
|
if (FAILED(hr))
|
|
delete pNew;
|
|
else
|
|
{
|
|
m_ConnectionList.push_back(pNew);
|
|
|
|
hr = _InsertConnection(pNew);
|
|
}
|
|
} else
|
|
{
|
|
(*it)->m_bStateNew = TRUE;
|
|
(*it)->m_bSyncImmediatelyNew = FALSE;
|
|
(*it)->m_nPriorityNew = PRIORITY_LOW;
|
|
if ((*it)->m_opType == CONNECTION_OPTYPE_DEL)
|
|
{
|
|
(*it)->m_opType = CONNECTION_OPTYPE_OTHERS;
|
|
hr = _InsertConnection(*it);
|
|
} else
|
|
{
|
|
(*it)->m_opType = CONNECTION_OPTYPE_OTHERS;
|
|
hr = _SetConnectionState(*it);
|
|
}
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
HRESULT CCustomTopology::_RemoveFromConnectionList(CCusTopConnection* pConn)
|
|
{
|
|
RETURN_INVALIDARG_IF_NULL(pConn);
|
|
|
|
BOOL bFound = FALSE;
|
|
|
|
CCusTopConnectionList::iterator it;
|
|
for (it = m_ConnectionList.begin(); it != m_ConnectionList.end(); it++)
|
|
{
|
|
if (!lstrcmpi((*it)->m_bstrFromMemberDN, pConn->m_bstrFromMemberDN) &&
|
|
!lstrcmpi((*it)->m_bstrToMemberDN, pConn->m_bstrToMemberDN))
|
|
{
|
|
bFound = TRUE;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (it != m_ConnectionList.end())
|
|
{
|
|
if (CONNECTION_OPTYPE_ADD == (*it)->m_opType)
|
|
{
|
|
delete (*it);
|
|
m_ConnectionList.erase(it);
|
|
} else
|
|
{
|
|
(*it)->m_opType = CONNECTION_OPTYPE_DEL;
|
|
}
|
|
}
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// CNewConnections
|
|
//
|
|
|
|
CNewConnections::CNewConnections() : m_pMemberList(NULL)
|
|
{
|
|
}
|
|
|
|
CNewConnections::~CNewConnections()
|
|
{
|
|
FreeCusTopConnections(&m_NewConnectionList);
|
|
}
|
|
|
|
|
|
HRESULT CNewConnections::Initialize
|
|
(
|
|
CCusTopMemberList* i_pMemberList
|
|
)
|
|
{
|
|
RETURN_INVALIDARG_IF_NULL(i_pMemberList);
|
|
m_pMemberList = i_pMemberList;
|
|
return S_OK;
|
|
}
|
|
|
|
#define NUM_OF_FRS_NEWCONN_COLUMNS 2
|
|
|
|
LRESULT CNewConnections::OnInitDialog
|
|
(
|
|
UINT uMsg,
|
|
WPARAM wParam,
|
|
LPARAM lParam,
|
|
BOOL& bHandled
|
|
)
|
|
{
|
|
int nControlID[] = {IDC_FRS_NEWCONN_FROM, IDC_FRS_NEWCONN_TO};
|
|
int nColStringID[] = {IDS_FRS_NEWCONN_COL_FROMSERVER, IDS_FRS_NEWCONN_COL_TOSERVER};
|
|
|
|
HWND hwndControl = NULL;
|
|
CCusTopMemberList::iterator it;
|
|
|
|
for (int i = 0; i < 2; i++)
|
|
{
|
|
hwndControl = GetDlgItem(nControlID[i]);
|
|
AddLVColumns(hwndControl, nColStringID[i], NUM_OF_FRS_NEWCONN_COLUMNS);
|
|
ListView_SetExtendedListViewStyle(hwndControl, LVS_EX_FULLROWSELECT);
|
|
|
|
for (it = m_pMemberList->begin(); it != m_pMemberList->end(); it++)
|
|
{
|
|
LVITEM lvItem;
|
|
|
|
lvItem.mask = LVIF_TEXT | LVIF_PARAM;
|
|
lvItem.lParam = (LPARAM)(*it);
|
|
lvItem.pszText = (*it)->m_bstrServer;
|
|
lvItem.iSubItem = 0;
|
|
int iItemIndex = ListView_InsertItem(hwndControl, &lvItem);
|
|
|
|
lvItem.mask = LVIF_TEXT;
|
|
lvItem.iItem = iItemIndex;
|
|
lvItem.pszText = (*it)->m_bstrSite;
|
|
lvItem.iSubItem = 1;
|
|
ListView_SetItem(hwndControl, &lvItem);
|
|
}
|
|
}
|
|
|
|
return TRUE; // Let the system set the focus
|
|
}
|
|
|
|
/*++
|
|
This function is called when a user clicks the ? in the top right of a property sheet
|
|
and then clciks a control, or when they hit F1 in a control.
|
|
--*/
|
|
LRESULT CNewConnections::OnCtxHelp(
|
|
IN UINT i_uMsg,
|
|
IN WPARAM i_wParam,
|
|
IN LPARAM i_lParam,
|
|
IN OUT BOOL& io_bHandled
|
|
)
|
|
{
|
|
LPHELPINFO lphi = (LPHELPINFO) i_lParam;
|
|
if (!lphi || lphi->iContextType != HELPINFO_WINDOW || lphi->iCtrlId < 0)
|
|
return FALSE;
|
|
|
|
::WinHelp((HWND)(lphi->hItemHandle),
|
|
DFS_CTX_HELP_FILE,
|
|
HELP_WM_HELP,
|
|
(DWORD_PTR)(PVOID)g_aHelpIDs_IDD_FRS_NEWCONN);
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
/*++
|
|
This function handles "What's This" help when a user right clicks the control
|
|
--*/
|
|
LRESULT CNewConnections::OnCtxMenuHelp(
|
|
IN UINT i_uMsg,
|
|
IN WPARAM i_wParam,
|
|
IN LPARAM i_lParam,
|
|
IN OUT BOOL& io_bHandled
|
|
)
|
|
{
|
|
::WinHelp((HWND)i_wParam,
|
|
DFS_CTX_HELP_FILE,
|
|
HELP_CONTEXTMENU,
|
|
(DWORD_PTR)(PVOID)g_aHelpIDs_IDD_FRS_NEWCONN);
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
HRESULT CNewConnections::get_NewConnections(CCusTopConnectionList** ppConnectionList)
|
|
{
|
|
RETURN_INVALIDARG_IF_NULL(ppConnectionList);
|
|
*ppConnectionList = &m_NewConnectionList;
|
|
return S_OK;
|
|
}
|
|
|
|
LRESULT CNewConnections::OnOK
|
|
(
|
|
WORD wNotifyCode,
|
|
WORD wID,
|
|
HWND hWndCtl,
|
|
BOOL& bHandled
|
|
)
|
|
{
|
|
BOOL bValidInput = FALSE;
|
|
int idString = 0;
|
|
HRESULT hr = S_OK;
|
|
|
|
do {
|
|
idString = IDS_FRS_NEWCONN_NOSELECTION;
|
|
|
|
//
|
|
// get all selected From servers
|
|
//
|
|
CCusTopMember* pMember;
|
|
CCusTopMemberList fromMemberList;
|
|
HWND hwndFrom = GetDlgItem(IDC_FRS_NEWCONN_FROM);
|
|
int nIndexFrom = ListView_GetNextItem(hwndFrom, -1, LVNI_ALL | LVNI_SELECTED);
|
|
while (-1 != nIndexFrom &&
|
|
(pMember = (CCusTopMember *)GetListViewItemData(hwndFrom, nIndexFrom)))
|
|
{
|
|
fromMemberList.push_back(pMember);
|
|
nIndexFrom = ListView_GetNextItem(hwndFrom, nIndexFrom, LVNI_ALL | LVNI_SELECTED);
|
|
}
|
|
if (fromMemberList.empty())
|
|
break;
|
|
|
|
//
|
|
// get all selected To servers
|
|
//
|
|
CCusTopMemberList toMemberList;
|
|
HWND hwndTo = GetDlgItem(IDC_FRS_NEWCONN_TO);
|
|
int nIndexTo = ListView_GetNextItem(hwndTo, -1, LVNI_ALL | LVNI_SELECTED);
|
|
while (-1 != nIndexTo &&
|
|
(pMember = (CCusTopMember *)GetListViewItemData(hwndTo, nIndexTo)))
|
|
{
|
|
toMemberList.push_back(pMember);
|
|
nIndexTo = ListView_GetNextItem(hwndTo, nIndexTo, LVNI_ALL | LVNI_SELECTED);
|
|
}
|
|
if (toMemberList.empty())
|
|
break;
|
|
|
|
//
|
|
// init the list
|
|
//
|
|
FreeCusTopConnections(&m_NewConnectionList);
|
|
|
|
//
|
|
// build the connection list
|
|
//
|
|
CCusTopMemberList::iterator from, to;
|
|
for (from = fromMemberList.begin(); from != fromMemberList.end(); from++)
|
|
{
|
|
for (to = toMemberList.begin(); to != toMemberList.end(); to++)
|
|
{
|
|
if (!lstrcmpi((*from)->m_bstrServer, (*to)->m_bstrServer))
|
|
continue;
|
|
|
|
CCusTopConnection* pNew = new CCusTopConnection;
|
|
BREAK_OUTOFMEMORY_IF_NULL(pNew, &hr);
|
|
|
|
hr = pNew->Init(
|
|
(*from)->m_bstrMemberDN, (*from)->m_bstrServer, (*from)->m_bstrSite,
|
|
(*to)->m_bstrMemberDN, (*to)->m_bstrServer, (*to)->m_bstrSite,
|
|
TRUE, CONNECTION_OPTYPE_ADD);
|
|
BREAK_IF_FAILED(hr);
|
|
|
|
m_NewConnectionList.push_back(pNew);
|
|
}
|
|
BREAK_IF_FAILED(hr);
|
|
}
|
|
|
|
if (SUCCEEDED(hr))
|
|
bValidInput = TRUE;
|
|
|
|
} while (0);
|
|
|
|
if (FAILED(hr))
|
|
{
|
|
DisplayMessageBoxForHR(hr);
|
|
return FALSE;
|
|
} else if (bValidInput)
|
|
{
|
|
EndDialog(S_OK);
|
|
return TRUE;
|
|
}
|
|
else
|
|
{
|
|
if (idString)
|
|
DisplayMessageBoxWithOK(idString);
|
|
return FALSE;
|
|
}
|
|
}
|
|
|
|
LRESULT CNewConnections::OnCancel
|
|
(
|
|
WORD wNotifyCode,
|
|
WORD wID,
|
|
HWND hWndCtl,
|
|
BOOL& bHandled
|
|
)
|
|
{
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Called OnCancel. Ends the dialog with S_FALSE;
|
|
|
|
*/
|
|
EndDialog(S_FALSE);
|
|
return(true);
|
|
}
|
|
|
|
int CALLBACK MembersListCompareProc(
|
|
IN LPARAM lParam1,
|
|
IN LPARAM lParam2,
|
|
IN LPARAM lParamColumn)
|
|
{
|
|
CCusTopMember* pItem1 = (CCusTopMember *)lParam1;
|
|
CCusTopMember* pItem2 = (CCusTopMember *)lParam2;
|
|
int iResult = 0;
|
|
|
|
if (pItem1 && pItem2)
|
|
{
|
|
switch( lParamColumn)
|
|
{
|
|
case 0: // Sort by Server.
|
|
iResult = lstrcmpi(pItem1->m_bstrServer, pItem2->m_bstrServer);
|
|
break;
|
|
case 1: // Sort by Site.
|
|
iResult = lstrcmpi(pItem1->m_bstrSite, pItem2->m_bstrSite);
|
|
break;
|
|
default:
|
|
iResult = 0;
|
|
break;
|
|
}
|
|
}
|
|
|
|
return(iResult);
|
|
}
|
|
|
|
LRESULT
|
|
CNewConnections::OnNotify(
|
|
IN UINT i_uMsg,
|
|
IN WPARAM i_wParam,
|
|
IN LPARAM i_lParam,
|
|
IN OUT BOOL& io_bHandled
|
|
)
|
|
{
|
|
NM_LISTVIEW* pNMListView = (NM_LISTVIEW*)i_lParam;
|
|
io_bHandled = FALSE; // So that the base class gets this notify too
|
|
|
|
if (IDC_FRS_NEWCONN_FROM == pNMListView->hdr.idFrom ||
|
|
IDC_FRS_NEWCONN_TO == pNMListView->hdr.idFrom)
|
|
{
|
|
HWND hwndList = GetDlgItem(pNMListView->hdr.idFrom);
|
|
if (LVN_COLUMNCLICK == pNMListView->hdr.code)
|
|
{
|
|
// sort items
|
|
ListView_SortItems( hwndList,
|
|
MembersListCompareProc,
|
|
(LPARAM)(pNMListView->iSubItem));
|
|
io_bHandled = TRUE;
|
|
}
|
|
}
|
|
|
|
return io_bHandled;
|
|
}
|