424 lines
10 KiB
C++
424 lines
10 KiB
C++
|
/**********************************************************************/
|
||
|
|
/** Microsoft Windows/NT **/
|
||
|
|
/** Copyright(c) Microsoft Corporation, 1997 - 1999 **/
|
||
|
|
/**********************************************************************/
|
||
|
|
|
||
|
|
/*
|
||
|
|
nodes.cpp
|
||
|
|
Replication partner leaf node
|
||
|
|
|
||
|
|
FILE HISTORY:
|
||
|
|
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include "stdafx.h"
|
||
|
|
#include "nodes.h"
|
||
|
|
|
||
|
|
#include "repprtpp.h"
|
||
|
|
#include "pushtrig.h"
|
||
|
|
#include "fndrcdlg.h"
|
||
|
|
#include "resource.h"
|
||
|
|
|
||
|
|
CString g_strPartnerTypePushPull;
|
||
|
|
CString g_strPartnerTypePush;
|
||
|
|
CString g_strPartnerTypePull;
|
||
|
|
CString g_strPartnerTypeUnknown;
|
||
|
|
|
||
|
|
CReplicationPartner::CReplicationPartner(
|
||
|
|
ITFSComponentData *pCompData,
|
||
|
|
CWinsServerObj *pObj
|
||
|
|
): CWinsHandler(pCompData),
|
||
|
|
m_strServerName(pObj->GetNetBIOSName()),
|
||
|
|
m_strIPAddress(pObj->GetstrIPAddress())
|
||
|
|
{
|
||
|
|
m_bExpanded = FALSE;
|
||
|
|
m_Server = *pObj;
|
||
|
|
m_nState = loaded;
|
||
|
|
m_verbDefault = MMC_VERB_PROPERTIES;
|
||
|
|
|
||
|
|
if (g_strPartnerTypePushPull.IsEmpty())
|
||
|
|
{
|
||
|
|
g_strPartnerTypePushPull.LoadString(IDS_PUSHPULL);
|
||
|
|
g_strPartnerTypePush.LoadString(IDS_PUSH);
|
||
|
|
g_strPartnerTypePull.LoadString(IDS_PULL);
|
||
|
|
g_strPartnerTypeUnknown.LoadString(IDS_NONE);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
HRESULT
|
||
|
|
CReplicationPartner::InitializeNode
|
||
|
|
(
|
||
|
|
ITFSNode * pNode
|
||
|
|
)
|
||
|
|
{
|
||
|
|
AFX_MANAGE_STATE(AfxGetStaticModuleState( ));
|
||
|
|
|
||
|
|
SetDisplayName(m_strServerName);
|
||
|
|
|
||
|
|
if (m_Server.IsPull() && m_Server.IsPush())
|
||
|
|
m_strType = g_strPartnerTypePushPull;
|
||
|
|
else if(m_Server.IsPull() && !m_Server.IsPush())
|
||
|
|
m_strType = g_strPartnerTypePull;
|
||
|
|
else if(!m_Server.IsPull() && m_Server.IsPush())
|
||
|
|
m_strType = g_strPartnerTypePush;
|
||
|
|
else
|
||
|
|
m_strType = g_strPartnerTypeUnknown;
|
||
|
|
|
||
|
|
// Make the node immediately visible
|
||
|
|
pNode->SetVisibilityState(TFS_VIS_SHOW);
|
||
|
|
pNode->SetData(TFS_DATA_COOKIE, (LPARAM) pNode);
|
||
|
|
pNode->SetData(TFS_DATA_USER, (LPARAM) this);
|
||
|
|
pNode->SetData(TFS_DATA_TYPE, WINSSNAP_REPLICATION_PARTNER);
|
||
|
|
pNode->SetData(TFS_DATA_IMAGEINDEX, ICON_IDX_PARTNER);
|
||
|
|
pNode->SetData(TFS_DATA_OPENIMAGEINDEX, ICON_IDX_PARTNER);
|
||
|
|
|
||
|
|
SetColumnStringIDs(&aColumns[WINSSNAP_REPLICATION_PARTNER][0]);
|
||
|
|
SetColumnWidths(&aColumnWidths[WINSSNAP_REPLICATION_PARTNER][0]);
|
||
|
|
|
||
|
|
return hrOK;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
/*!--------------------------------------------------------------------------
|
||
|
|
CActiveRegistration::AddMenuItems
|
||
|
|
Implementation of ITFSResultHandler::AddMenuItems
|
||
|
|
Author: EricDav
|
||
|
|
---------------------------------------------------------------------------*/
|
||
|
|
STDMETHODIMP
|
||
|
|
CReplicationPartner::AddMenuItems
|
||
|
|
(
|
||
|
|
ITFSComponent * pComponent,
|
||
|
|
MMC_COOKIE cookie,
|
||
|
|
LPDATAOBJECT pDataObject,
|
||
|
|
LPCONTEXTMENUCALLBACK pContextMenuCallback,
|
||
|
|
long * pInsertionAllowed
|
||
|
|
)
|
||
|
|
{
|
||
|
|
AFX_MANAGE_STATE(AfxGetStaticModuleState());
|
||
|
|
|
||
|
|
HRESULT hr;
|
||
|
|
|
||
|
|
CString strMenuItem;
|
||
|
|
BOOL bEnablePush, bEnablePull;
|
||
|
|
|
||
|
|
bEnablePush = bEnablePull = FALSE;
|
||
|
|
|
||
|
|
if (m_strType.Compare(g_strPartnerTypePushPull) == 0)
|
||
|
|
{
|
||
|
|
bEnablePush = bEnablePull = TRUE;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
if (m_strType.Compare(g_strPartnerTypePush) == 0)
|
||
|
|
{
|
||
|
|
bEnablePush = TRUE;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
if (m_strType.Compare(g_strPartnerTypePull) == 0)
|
||
|
|
{
|
||
|
|
bEnablePull = TRUE;
|
||
|
|
}
|
||
|
|
|
||
|
|
// these menu items go in the new menu,
|
||
|
|
// only visible from scope pane
|
||
|
|
|
||
|
|
if (*pInsertionAllowed & CCM_INSERTIONALLOWED_TOP)
|
||
|
|
{
|
||
|
|
if (bEnablePush)
|
||
|
|
{
|
||
|
|
strMenuItem.LoadString(IDS_REP_SEND_PUSH_TRIGGER);
|
||
|
|
hr = LoadAndAddMenuItem( pContextMenuCallback,
|
||
|
|
strMenuItem,
|
||
|
|
IDS_REP_SEND_PUSH_TRIGGER,
|
||
|
|
CCM_INSERTIONPOINTID_PRIMARY_TOP,
|
||
|
|
0 );
|
||
|
|
ASSERT( SUCCEEDED(hr) );
|
||
|
|
}
|
||
|
|
|
||
|
|
if (bEnablePull)
|
||
|
|
{
|
||
|
|
strMenuItem.LoadString(IDS_REP_SEND_PULL_TRIGGER);
|
||
|
|
hr = LoadAndAddMenuItem( pContextMenuCallback,
|
||
|
|
strMenuItem,
|
||
|
|
IDS_REP_SEND_PULL_TRIGGER,
|
||
|
|
CCM_INSERTIONPOINTID_PRIMARY_TOP,
|
||
|
|
0 );
|
||
|
|
ASSERT( SUCCEEDED(hr) );
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return hrOK;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
/*---------------------------------------------------------------------------
|
||
|
|
Overridden base handler functions
|
||
|
|
---------------------------------------------------------------------------*/
|
||
|
|
|
||
|
|
/*!--------------------------------------------------------------------------
|
||
|
|
CActiveRegistration::GetString
|
||
|
|
Implementation of ITFSNodeHandler::GetString
|
||
|
|
Author: KennT
|
||
|
|
---------------------------------------------------------------------------*/
|
||
|
|
STDMETHODIMP_(LPCTSTR)
|
||
|
|
CReplicationPartner::GetString
|
||
|
|
(
|
||
|
|
ITFSComponent * pComponent,
|
||
|
|
MMC_COOKIE cookie,
|
||
|
|
int nCol
|
||
|
|
)
|
||
|
|
{
|
||
|
|
if (nCol == 0 || nCol == -1)
|
||
|
|
return GetDisplayName();
|
||
|
|
switch(nCol)
|
||
|
|
{
|
||
|
|
case -1:
|
||
|
|
case 0:
|
||
|
|
return GetDisplayName();
|
||
|
|
case 1:
|
||
|
|
return m_strIPAddress;
|
||
|
|
case 2:
|
||
|
|
return m_strType;
|
||
|
|
case 3:
|
||
|
|
return m_strReplicationTime;
|
||
|
|
default:
|
||
|
|
return NULL;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
/*---------------------------------------------------------------------------
|
||
|
|
CReplicationPartner::Command
|
||
|
|
Description
|
||
|
|
Author: EricDav
|
||
|
|
---------------------------------------------------------------------------*/
|
||
|
|
STDMETHODIMP
|
||
|
|
CReplicationPartner::Command
|
||
|
|
(
|
||
|
|
ITFSComponent * pComponent,
|
||
|
|
MMC_COOKIE cookie,
|
||
|
|
int nCommandID,
|
||
|
|
LPDATAOBJECT pDataObject
|
||
|
|
)
|
||
|
|
{
|
||
|
|
HRESULT hr;
|
||
|
|
|
||
|
|
AFX_MANAGE_STATE(AfxGetStaticModuleState( ));
|
||
|
|
|
||
|
|
SPITFSNode spNode;
|
||
|
|
m_spNodeMgr->FindNode(cookie, &spNode);
|
||
|
|
|
||
|
|
// get the server node now
|
||
|
|
SPITFSNode spRepNode, spServerNode;
|
||
|
|
spNode->GetParent(&spRepNode);
|
||
|
|
spRepNode->GetParent(&spServerNode);
|
||
|
|
|
||
|
|
CWinsServerHandler *pServer = GETHANDLER(CWinsServerHandler, spServerNode);
|
||
|
|
|
||
|
|
//
|
||
|
|
// Object gets deleted when the page is destroyed
|
||
|
|
//
|
||
|
|
switch (nCommandID)
|
||
|
|
{
|
||
|
|
case IDS_REP_SEND_PUSH_TRIGGER:
|
||
|
|
hr = OnSendPushTrigger(pServer);
|
||
|
|
break;
|
||
|
|
|
||
|
|
case IDS_REP_SEND_PULL_TRIGGER:
|
||
|
|
hr = OnSendPullTrigger(pServer);
|
||
|
|
break;
|
||
|
|
|
||
|
|
default:
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
return hr;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
/*---------------------------------------------------------------------------
|
||
|
|
CActiveRegistration::CreatePropertyPages
|
||
|
|
Description
|
||
|
|
Author: EricDav
|
||
|
|
---------------------------------------------------------------------------*/
|
||
|
|
STDMETHODIMP
|
||
|
|
CReplicationPartner::CreatePropertyPages
|
||
|
|
(
|
||
|
|
ITFSComponent * pComponent,
|
||
|
|
MMC_COOKIE cookie,
|
||
|
|
LPPROPERTYSHEETCALLBACK lpProvider,
|
||
|
|
LPDATAOBJECT pDataObject,
|
||
|
|
LONG_PTR handle
|
||
|
|
)
|
||
|
|
{
|
||
|
|
AFX_MANAGE_STATE(AfxGetStaticModuleState());
|
||
|
|
|
||
|
|
HRESULT hr = hrOK;
|
||
|
|
|
||
|
|
SPITFSNode spNode;
|
||
|
|
m_spNodeMgr->FindNode(cookie, &spNode);
|
||
|
|
|
||
|
|
//
|
||
|
|
// Object gets deleted when the page is destroyed
|
||
|
|
//
|
||
|
|
SPIComponentData spComponentData;
|
||
|
|
m_spNodeMgr->GetComponentData(&spComponentData);
|
||
|
|
|
||
|
|
CReplicationPartnerProperties * pProp =
|
||
|
|
new CReplicationPartnerProperties(spNode, spComponentData, m_spTFSCompData, NULL);
|
||
|
|
pProp->m_pageGeneral.m_uImage = (UINT) spNode->GetData(TFS_DATA_IMAGEINDEX);
|
||
|
|
pProp->SetServer(&m_Server);
|
||
|
|
|
||
|
|
Assert(lpProvider != NULL);
|
||
|
|
|
||
|
|
return pProp->CreateModelessSheet(lpProvider, handle);
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
/*---------------------------------------------------------------------------
|
||
|
|
Command handlers
|
||
|
|
---------------------------------------------------------------------------*/
|
||
|
|
|
||
|
|
/*---------------------------------------------------------------------------
|
||
|
|
CReplicationPartner::ToIPAddressString()
|
||
|
|
Returns a CString IP address
|
||
|
|
---------------------------------------------------------------------------*/
|
||
|
|
CString
|
||
|
|
CReplicationPartner::ToIPAddressString()
|
||
|
|
{
|
||
|
|
CString strIP;
|
||
|
|
|
||
|
|
DWORD dwFirst = GETIP_FIRST(m_dwIPAddress);
|
||
|
|
DWORD dwSecond = GETIP_SECOND(m_dwIPAddress);
|
||
|
|
DWORD dwThird = GETIP_THIRD(m_dwIPAddress);
|
||
|
|
DWORD dwLast = GETIP_FOURTH(m_dwIPAddress);
|
||
|
|
|
||
|
|
// wrap it into CString object
|
||
|
|
TCHAR szStr[20];
|
||
|
|
|
||
|
|
_itot(dwFirst, szStr, 10);
|
||
|
|
CString strTemp(szStr);
|
||
|
|
strIP = strTemp + _T(".");
|
||
|
|
|
||
|
|
_itot(dwSecond, szStr, 10);
|
||
|
|
strTemp = (CString)szStr;
|
||
|
|
strIP += strTemp + _T(".");
|
||
|
|
|
||
|
|
_itot(dwThird, szStr, 10);
|
||
|
|
strTemp = (CString)szStr;
|
||
|
|
strIP += strTemp + _T(".");
|
||
|
|
|
||
|
|
_itot(dwLast, szStr, 10);
|
||
|
|
strTemp = (CString)szStr;
|
||
|
|
strIP += strTemp;
|
||
|
|
|
||
|
|
return strIP;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
/*---------------------------------------------------------------------------
|
||
|
|
CReplicationPartner::OnResultPropertyChange
|
||
|
|
Base class override
|
||
|
|
---------------------------------------------------------------------------*/
|
||
|
|
HRESULT
|
||
|
|
CReplicationPartner::OnResultPropertyChange
|
||
|
|
(
|
||
|
|
ITFSComponent * pComponent,
|
||
|
|
LPDATAOBJECT pDataObject,
|
||
|
|
MMC_COOKIE cookie,
|
||
|
|
LPARAM arg,
|
||
|
|
LPARAM param
|
||
|
|
)
|
||
|
|
{
|
||
|
|
HRESULT hr = hrOK;
|
||
|
|
AFX_MANAGE_STATE(AfxGetStaticModuleState( ));
|
||
|
|
|
||
|
|
SPITFSNode spNode;
|
||
|
|
m_spNodeMgr->FindNode(cookie, &spNode);
|
||
|
|
|
||
|
|
CReplicationPartnerProperties * pProp = reinterpret_cast<CReplicationPartnerProperties *>(param);
|
||
|
|
|
||
|
|
LONG_PTR changeMask = 0;
|
||
|
|
|
||
|
|
// tell the property page to do whatever now that we are back on the
|
||
|
|
// main thread
|
||
|
|
pProp->OnPropertyChange(TRUE, &changeMask);
|
||
|
|
|
||
|
|
pProp->AcknowledgeNotify();
|
||
|
|
|
||
|
|
if (changeMask)
|
||
|
|
spNode->ChangeNode(changeMask);
|
||
|
|
|
||
|
|
return hr;
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
/*---------------------------------------------------------------------------
|
||
|
|
CReplicationPartner::OnSendPushTrigger()
|
||
|
|
Sends Push replication trigger
|
||
|
|
---------------------------------------------------------------------------*/
|
||
|
|
HRESULT
|
||
|
|
CReplicationPartner::OnSendPushTrigger(CWinsServerHandler *pServer)
|
||
|
|
{
|
||
|
|
HRESULT hr = hrOK;
|
||
|
|
DWORD err = ERROR_SUCCESS;
|
||
|
|
|
||
|
|
CPushTrig cPushDlg;
|
||
|
|
|
||
|
|
if (cPushDlg.DoModal() != IDOK)
|
||
|
|
return hr;
|
||
|
|
|
||
|
|
err = ::SendTrigger(pServer->GetBinding(),
|
||
|
|
(LONG) m_Server.GetIpAddress(),
|
||
|
|
TRUE,
|
||
|
|
cPushDlg.GetPropagate());
|
||
|
|
|
||
|
|
if (err == ERROR_SUCCESS)
|
||
|
|
{
|
||
|
|
AfxMessageBox(IDS_REPL_QUEUED, MB_ICONINFORMATION);
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
WinsMessageBox(err);
|
||
|
|
}
|
||
|
|
|
||
|
|
return HRESULT_FROM_WIN32(err);
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
/*---------------------------------------------------------------------------
|
||
|
|
CReplicationPartner::OnSendPullTrigger()
|
||
|
|
Sends Pull replication trigger
|
||
|
|
---------------------------------------------------------------------------*/
|
||
|
|
HRESULT
|
||
|
|
CReplicationPartner::OnSendPullTrigger(CWinsServerHandler *pServer)
|
||
|
|
{
|
||
|
|
HRESULT hr = hrOK;
|
||
|
|
DWORD err = ERROR_SUCCESS;
|
||
|
|
|
||
|
|
CPullTrig cPullTrig;
|
||
|
|
|
||
|
|
if (cPullTrig.DoModal() != IDOK)
|
||
|
|
return hr;
|
||
|
|
|
||
|
|
err = ::SendTrigger(pServer->GetBinding(),
|
||
|
|
(LONG) m_Server.GetIpAddress(),
|
||
|
|
FALSE,
|
||
|
|
FALSE);
|
||
|
|
|
||
|
|
if (err == ERROR_SUCCESS)
|
||
|
|
{
|
||
|
|
AfxMessageBox(IDS_REPL_QUEUED, MB_ICONINFORMATION);
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
::WinsMessageBox(err);
|
||
|
|
}
|
||
|
|
|
||
|
|
return HRESULT_FROM_WIN32(err);
|
||
|
|
}
|
||
|
|
|
||
|
|
|