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.
 
 
 
 
 
 

1002 lines
27 KiB

/**********************************************************************/
/** Microsoft Windows/NT **/
/** Copyright(c) Microsoft Corporation, 1997 - 2002 **/
/**********************************************************************/
/*
provider.cpp
Filter node handler
FILE HISTORY:
*/
#include "stdafx.h"
#include "server.h"
#include "FltrNode.h"
#include "SpdUtil.h"
#include "fltrpp.h"
#include "srchfltr.h"
/*---------------------------------------------------------------------------
Class CFilterHandler implementation
---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------
Constructor and destructor
Description
Author: NSun
---------------------------------------------------------------------------*/
CFilterHandler::CFilterHandler
(
ITFSComponentData * pComponentData
) : CIpsmHandler(pComponentData),
m_FltrType(FILTER_TYPE_ANY) //by default we display both transport and tunnel filters
{
AFX_MANAGE_STATE(AfxGetStaticModuleState());
}
CFilterHandler::~CFilterHandler()
{
}
/*!--------------------------------------------------------------------------
CFilterHandler::InitializeNode
Initializes node specific data
Author: NSun
---------------------------------------------------------------------------*/
HRESULT
CFilterHandler::InitializeNode
(
ITFSNode * pNode
)
{
AFX_MANAGE_STATE(AfxGetStaticModuleState( ));
CString strTemp;
strTemp.LoadString(IDS_FILTER_NODE);
SetDisplayName(strTemp);
// Make the node immediately visible
pNode->SetVisibilityState(TFS_VIS_SHOW);
pNode->SetData(TFS_DATA_COOKIE, (LPARAM) pNode);
pNode->SetData(TFS_DATA_IMAGEINDEX, ICON_IDX_FOLDER_CLOSED);
pNode->SetData(TFS_DATA_OPENIMAGEINDEX, ICON_IDX_FOLDER_OPEN);
pNode->SetData(TFS_DATA_USER, (LPARAM) this);
pNode->SetData(TFS_DATA_TYPE, IPSECMON_FILTER);
pNode->SetData(TFS_DATA_SCOPE_LEAF_NODE, TRUE);
SetColumnStringIDs(&aColumns[IPSECMON_FILTER][0]);
SetColumnWidths(&aColumnWidths[IPSECMON_FILTER][0]);
return hrOK;
}
/*---------------------------------------------------------------------------
CFilterHandler::GetImageIndex
-
Author: NSun
---------------------------------------------------------------------------*/
int
CFilterHandler::GetImageIndex(BOOL bOpenImage)
{
int nIndex = -1;
return nIndex;
}
/*---------------------------------------------------------------------------
Overridden base handler functions
---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------
CFilterHandler::OnAddMenuItems
Adds context menu items for the SA scope pane node
Author: NSun
---------------------------------------------------------------------------*/
STDMETHODIMP
CFilterHandler::OnAddMenuItems
(
ITFSNode * pNode,
LPCONTEXTMENUCALLBACK pContextMenuCallback,
LPDATAOBJECT lpDataObject,
DATA_OBJECT_TYPES type,
DWORD dwType,
long * pInsertionAllowed
)
{
AFX_MANAGE_STATE(AfxGetStaticModuleState( ));
LONG fFlags = 0, fLoadingFlags = 0;
HRESULT hr = S_OK;
CString strMenuItem;
if (type == CCT_SCOPE)
{
if (*pInsertionAllowed & CCM_INSERTIONALLOWED_TOP)
{
}
}
return hr;
}
/*!--------------------------------------------------------------------------
CFilterHandler::AddMenuItems
Adds context menu items for virtual list box (result pane) items
Author: NSun
---------------------------------------------------------------------------*/
STDMETHODIMP
CFilterHandler::AddMenuItems
(
ITFSComponent * pComponent,
MMC_COOKIE cookie,
LPDATAOBJECT pDataObject,
LPCONTEXTMENUCALLBACK pContextMenuCallback,
long * pInsertionAllowed
)
{
AFX_MANAGE_STATE(AfxGetStaticModuleState( ));
HRESULT hr = hrOK;
CString strMenuItem;
SPINTERNAL spInternal;
LONG fFlags = 0;
spInternal = ExtractInternalFormat(pDataObject);
// virtual listbox notifications come to the handler of the node that is selected.
// check to see if this notification is for a virtual listbox item or this SA
// node itself.
if (*pInsertionAllowed & CCM_INSERTIONALLOWED_VIEW)
{
//load and view menu items here
}
if (*pInsertionAllowed & CCM_INSERTIONALLOWED_VIEW)
{
strMenuItem.LoadString(IDS_VIEW_ALL_FLTR);
hr = LoadAndAddMenuItem( pContextMenuCallback,
strMenuItem,
IDS_VIEW_ALL_FLTR,
CCM_INSERTIONPOINTID_PRIMARY_VIEW,
(FILTER_TYPE_ANY == m_FltrType) ? MF_CHECKED : 0 );
strMenuItem.LoadString(IDS_VIEW_TRANSPORT_FLTR);
hr = LoadAndAddMenuItem( pContextMenuCallback,
strMenuItem,
IDS_VIEW_TRANSPORT_FLTR,
CCM_INSERTIONPOINTID_PRIMARY_VIEW,
(FILTER_TYPE_TRANSPORT == m_FltrType) ? MF_CHECKED : 0 );
strMenuItem.LoadString(IDS_VIEW_TUNNEL_FLTR);
hr = LoadAndAddMenuItem( pContextMenuCallback,
strMenuItem,
IDS_VIEW_TUNNEL_FLTR,
CCM_INSERTIONPOINTID_PRIMARY_VIEW,
(FILTER_TYPE_TUNNEL == m_FltrType) ? MF_CHECKED : 0 );
ASSERT( SUCCEEDED(hr) );
}
return hr;
}
/*!--------------------------------------------------------------------------
CFilterHandler::OnRefresh
Default implementation for the refresh functionality
Author: NSun
---------------------------------------------------------------------------*/
HRESULT
CFilterHandler::OnRefresh
(
ITFSNode * pNode,
LPDATAOBJECT pDataObject,
DWORD dwType,
LPARAM arg,
LPARAM param
)
{
HRESULT hr = S_OK;
int i = 0;
SPIConsole spConsole;
CORg(CHandler::OnRefresh(pNode, pDataObject, dwType, arg, param));
CORg(m_spSpdInfo->EnumQmFilters());
m_spSpdInfo->ChangeQmFilterViewType(m_FltrType);
i = m_spSpdInfo->GetQmFilterCountOfCurrentViewType();
// now notify the virtual listbox
CORg ( m_spNodeMgr->GetConsole(&spConsole) );
CORg ( spConsole->UpdateAllViews(pDataObject, i, RESULT_PANE_SET_VIRTUAL_LB_SIZE));
Error:
return hr;
}
/*---------------------------------------------------------------------------
CFilterHandler::OnCommand
Handles context menu commands for SA scope pane node
Author: NSun
---------------------------------------------------------------------------*/
STDMETHODIMP
CFilterHandler::OnCommand
(
ITFSNode * pNode,
long nCommandId,
DATA_OBJECT_TYPES type,
LPDATAOBJECT pDataObject,
DWORD dwType
)
{
AFX_MANAGE_STATE(AfxGetStaticModuleState( ));
//handle the scope context menu commands here
return S_OK;
}
/*!--------------------------------------------------------------------------
CFilterHandler::Command
Handles context menu commands for virtual listbox items
Author: NSun
---------------------------------------------------------------------------*/
STDMETHODIMP
CFilterHandler::Command
(
ITFSComponent * pComponent,
MMC_COOKIE cookie,
int nCommandID,
LPDATAOBJECT pDataObject
)
{
AFX_MANAGE_STATE(AfxGetStaticModuleState( ));
HRESULT hr = S_OK;
SPITFSNode spNode;
m_spResultNodeMgr->FindNode(cookie, &spNode);
FILTER_TYPE NewFltrType = m_FltrType;
// handle result context menu and view menus here
switch (nCommandID)
{
case IDS_VIEW_ALL_FLTR:
NewFltrType = FILTER_TYPE_ANY;
break;
case IDS_VIEW_TRANSPORT_FLTR:
NewFltrType = FILTER_TYPE_TRANSPORT;
break;
case IDS_VIEW_TUNNEL_FLTR:
NewFltrType = FILTER_TYPE_TUNNEL;
break;
default:
break;
}
//Update the views if a different view is selected.
if (NewFltrType != m_FltrType)
{
m_FltrType = NewFltrType;
UpdateViewType(spNode, m_FltrType);
}
return hr;
}
/*!--------------------------------------------------------------------------
CFilterHandler::HasPropertyPages
Implementation of ITFSNodeHandler::HasPropertyPages
NOTE: the root node handler has to over-ride this function to
handle the snapin manager property page (wizard) case!!!
Author: KennT
---------------------------------------------------------------------------*/
STDMETHODIMP
CFilterHandler::HasPropertyPages
(
ITFSNode * pNode,
LPDATAOBJECT pDataObject,
DATA_OBJECT_TYPES type,
DWORD dwType
)
{
AFX_MANAGE_STATE(AfxGetStaticModuleState());
return hrFalse;
}
/*---------------------------------------------------------------------------
CFilterHandler::CreatePropertyPages
Description
Author: NSun
---------------------------------------------------------------------------*/
STDMETHODIMP
CFilterHandler::CreatePropertyPages
(
ITFSNode * pNode,
LPPROPERTYSHEETCALLBACK lpSA,
LPDATAOBJECT pDataObject,
LONG_PTR handle,
DWORD dwType
)
{
AFX_MANAGE_STATE(AfxGetStaticModuleState());
DWORD dwError;
DWORD dwDynDnsFlags;
//
// Create the property page
//
SPIComponentData spComponentData;
m_spNodeMgr->GetComponentData(&spComponentData);
//CServerProperties * pServerProp = new CServerProperties(pNode, spComponentData, m_spTFSCompData, NULL);
//
// Object gets deleted when the page is destroyed
//
Assert(lpSA != NULL);
//return pServerProp->CreateModelessSheet(lpSA, handle);
return hrFalse;
}
/*---------------------------------------------------------------------------
CFilterHandler::OnPropertyChange
Description
Author: NSun
---------------------------------------------------------------------------*/
HRESULT
CFilterHandler::OnPropertyChange
(
ITFSNode * pNode,
LPDATAOBJECT pDataobject,
DWORD dwType,
LPARAM arg,
LPARAM lParam
)
{
AFX_MANAGE_STATE(AfxGetStaticModuleState( ));
//CServerProperties * pServerProp = reinterpret_cast<CServerProperties *>(lParam);
LONG_PTR changeMask = 0;
// tell the property page to do whatever now that we are back on the
// main thread
//pServerProp->OnPropertyChange(TRUE, &changeMask);
//pServerProp->AcknowledgeNotify();
if (changeMask)
pNode->ChangeNode(changeMask);
return hrOK;
}
/*---------------------------------------------------------------------------
CFilterHandler::OnExpand
Handles enumeration of a scope item
Author: NSun
---------------------------------------------------------------------------*/
HRESULT
CFilterHandler::OnExpand
(
ITFSNode * pNode,
LPDATAOBJECT pDataObject,
DWORD dwType,
LPARAM arg,
LPARAM param
)
{
HRESULT hr = hrOK;
if (m_bExpanded)
return hr;
// do the default handling
CORg (CIpsmHandler::OnExpand(pNode, pDataObject, dwType, arg, param));
Error:
return hr;
}
/*!--------------------------------------------------------------------------
CFilterHandler::OnResultSelect
Handles the MMCN_SELECT notifcation
Author: NSun
---------------------------------------------------------------------------*/
HRESULT
CFilterHandler::OnResultSelect
(
ITFSComponent * pComponent,
LPDATAOBJECT pDataObject,
MMC_COOKIE cookie,
LPARAM arg,
LPARAM lParam
)
{
AFX_MANAGE_STATE(AfxGetStaticModuleState( ));
HRESULT hr = hrOK;
SPINTERNAL spInternal;
SPIConsole spConsole;
SPIConsoleVerb spConsoleVerb;
SPITFSNode spNode;
BOOL bStates[ARRAYLEN(g_ConsoleVerbs)];
int i;
LONG_PTR dwNodeType;
BOOL fSelect = HIWORD(arg);
IConsole2 *pConsole2 = NULL;
//get pConsole2 for writing to the status bar, if fails not to worry
pComponent->GetConsole(&pConsole2);
// virtual listbox notifications come to the handler of the node that is selected.
// check to see if this notification is for a virtual listbox item or the active
// registrations node itself.
CORg (pComponent->GetConsoleVerb(&spConsoleVerb));
m_verbDefault = MMC_VERB_OPEN;
if (!fSelect)
{
if(pConsole2) {
pConsole2->SetStatusText(_T(""));
pConsole2->Release();
}
return hr;
}
if (m_spSpdInfo)
{
DWORD dwInitInfo;
dwInitInfo=m_spSpdInfo->GetInitInfo();
if (!(dwInitInfo & MON_QM_FILTER)) {
CORg(m_spSpdInfo->EnumQmFilters());
m_spSpdInfo->SetInitInfo(dwInitInfo | MON_QM_FILTER);
//m_spSpdInfo->SetInitInfo(dwInitInfo | MON_QM_SP_FILTER);
}
m_spSpdInfo->SetActiveInfo(MON_QM_FILTER);
// Get the current count
i = m_spSpdInfo->GetQmFilterCountOfCurrentViewType();
// now notify the virtual listbox
CORg ( m_spNodeMgr->GetConsole(&spConsole) );
CORg ( spConsole->UpdateAllViews(pDataObject, i, RESULT_PANE_SET_VIRTUAL_LB_SIZE) );
if(pConsole2)
{
CString strTemp;
WCHAR szTemp[20];
wsprintf(szTemp, L"%d", i);
AfxFormatString1(strTemp, IDS_STATUS_NUM_ITEMS, szTemp);
pConsole2->SetStatusText((LPOLESTR) (LPCTSTR) strTemp);
pConsole2->Release();
}
}
// now update the verbs...
spInternal = ExtractInternalFormat(pDataObject);
Assert(spInternal);
if (spInternal->HasVirtualIndex())
{
//TODO add to here if we want to have some result console verbs
// we gotta do special stuff for the virtual index items
dwNodeType = IPSECMON_FILTER_ITEM;
for (i = 0; i < ARRAYLEN(g_ConsoleVerbs); bStates[i++] = FALSE);
//enable the "properties" menu
bStates[MMC_VERB_PROPERTIES & 0x000F] = TRUE;
m_verbDefault = MMC_VERB_PROPERTIES;
}
else
{
// enable/disable delete depending if the node supports it
CORg (m_spNodeMgr->FindNode(cookie, &spNode));
dwNodeType = spNode->GetData(TFS_DATA_TYPE);
for (i = 0; i < ARRAYLEN(g_ConsoleVerbs); bStates[i++] = TRUE);
//hide "delete" context menu
bStates[MMC_VERB_DELETE & 0x000F] = FALSE;
bStates[MMC_VERB_REFRESH & 0x000F] = TRUE;
}
EnableVerbs(spConsoleVerb, g_ConsoleVerbStates[dwNodeType], bStates);
COM_PROTECT_ERROR_LABEL;
return hr;
}
/*!--------------------------------------------------------------------------
CFilterHandler::OnDelete
The base handler calls this when MMC sends a MMCN_DELETE for a
scope pane item. We just call our delete command handler.
Author: NSun
---------------------------------------------------------------------------*/
HRESULT
CFilterHandler::OnDelete
(
ITFSNode * pNode,
LPARAM arg,
LPARAM lParam
)
{
return S_FALSE;
}
/*!--------------------------------------------------------------------------
CFilterHandler::HasPropertyPages
Handle the result notification
Author: NSun
---------------------------------------------------------------------------*/
STDMETHODIMP
CFilterHandler::HasPropertyPages(
ITFSComponent *pComponent,
MMC_COOKIE cookie,
LPDATAOBJECT pDataObject)
{
return hrOK;
}
/*!--------------------------------------------------------------------------
CFilterHandler::HasPropertyPages
Handle the result notification. Create the filter property sheet
Author: NSun
---------------------------------------------------------------------------*/
STDMETHODIMP CFilterHandler::CreatePropertyPages
(
ITFSComponent * pComponent,
MMC_COOKIE cookie,
LPPROPERTYSHEETCALLBACK lpProvider,
LPDATAOBJECT pDataObject,
LONG_PTR handle
)
{
AFX_MANAGE_STATE(AfxGetStaticModuleState());
HRESULT hr = hrOK;
SPINTERNAL spInternal;
SPITFSNode spNode;
int nIndex;
SPIComponentData spComponentData;
CFilterInfo FltrInfo;
CFilterProperties * pFilterProp;
Assert(m_spNodeMgr);
CORg( m_spNodeMgr->FindNode(cookie, &spNode) );
CORg( m_spNodeMgr->GetComponentData(&spComponentData) );
spInternal = ExtractInternalFormat(pDataObject);
// virtual listbox notifications come to the handler of the node that is selected.
// assert that this notification is for a virtual listbox item
Assert(spInternal);
if (!spInternal->HasVirtualIndex())
return hr;
nIndex = spInternal->GetVirtualIndex();
CORg(m_spSpdInfo->GetFilterInfo(nIndex, &FltrInfo));
pFilterProp = new CFilterProperties(
spNode,
spComponentData,
m_spTFSCompData,
&FltrInfo,
m_spSpdInfo,
NULL);
hr = pFilterProp->CreateModelessSheet(lpProvider, handle);
COM_PROTECT_ERROR_LABEL;
return hr;
}
/*---------------------------------------------------------------------------
CFilterHandler::OnGetResultViewType
Return the result view that this node is going to support
Author: NSun
---------------------------------------------------------------------------*/
HRESULT
CFilterHandler::OnGetResultViewType
(
ITFSComponent * pComponent,
MMC_COOKIE cookie,
LPOLESTR * ppViewType,
long * pViewOptions
)
{
if (cookie != NULL)
{
*pViewOptions = MMC_VIEW_OPTIONS_OWNERDATALIST;
}
return S_FALSE;
}
/*---------------------------------------------------------------------------
CFilterHandler::GetVirtualImage
Returns the image index for virtual listbox items
Author: NSun
---------------------------------------------------------------------------*/
int
CFilterHandler::GetVirtualImage
(
int nIndex
)
{
return ICON_IDX_FILTER;
}
/*---------------------------------------------------------------------------
CFilterHandler::GetVirtualString
returns a pointer to the string for virtual listbox items
Author: NSun
---------------------------------------------------------------------------*/
LPCWSTR
CFilterHandler::GetVirtualString
(
int nIndex,
int nCol
)
{
HRESULT hr = S_OK;
static CString strTemp;
strTemp.Empty();
if (nCol >= DimensionOf(aColumns[IPSECMON_FILTER]))
return NULL;
CFilterInfo filter;
CORg(m_spSpdInfo->GetFilterInfo(nIndex, &filter));
switch (aColumns[IPSECMON_FILTER][nCol])
{
case IDS_COL_FLTR_NAME:
{
strTemp = filter.m_stName;
if( filter.m_dwFlags & IPSEC_QM_POLICY_DEFAULT_POLICY )
{
AfxFormatString1(strTemp, IDS_POL_DEFAULT_RESPONSE, (LPCTSTR) filter.m_stName);
}
return strTemp;
break;
}
case IDS_COL_FLTR_SRC:
AddressToString(filter.m_SrcAddr, &strTemp);
return strTemp;
break;
case IDS_COL_FLTR_DEST:
AddressToString(filter.m_DesAddr, &strTemp);
return strTemp;
break;
case IDS_COL_FLTR_SRC_PORT:
PortToString(filter.m_SrcPort, &strTemp);
return strTemp;
break;
case IDS_COL_FLTR_DEST_PORT:
PortToString(filter.m_DesPort, &strTemp);
return strTemp;
break;
case IDS_COL_FLTR_SRC_TNL:
TnlEpToString(filter.m_FilterType,
filter.m_MyTnlAddr,
&strTemp
);
return strTemp;
break;
case IDS_COL_FLTR_DEST_TNL:
TnlEpToString(filter.m_FilterType,
filter.m_PeerTnlAddr,
&strTemp
);
return strTemp;
break;
case IDS_COL_FLTR_PROT:
ProtocolToString(filter.m_Protocol, &strTemp);
return strTemp;
break;
case IDS_COL_QM_POLICY:
strTemp = filter.m_stPolicyName;
return strTemp;
break;
case IDS_COL_IF_TYPE:
InterfaceTypeToString(filter.m_InterfaceType, &strTemp);
return strTemp;
break;
case IDS_COL_FLTR_OUT_FLAG:
FilterFlagToString(filter.m_OutboundFilterAction, &strTemp);
return strTemp;
break;
case IDS_COL_FLTR_IN_FLAG:
FilterFlagToString(filter.m_InboundFilterAction, &strTemp);
return strTemp;
break;
default:
Panic0("CFilterHandler::GetVirtualString - Unknown column!\n");
break;
}
COM_PROTECT_ERROR_LABEL;
return NULL;
}
/*---------------------------------------------------------------------------
CFilterHandler::CacheHint
MMC tells us which items it will need before it requests things
Author: NSun
---------------------------------------------------------------------------*/
STDMETHODIMP
CFilterHandler::CacheHint
(
int nStartIndex,
int nEndIndex
)
{
HRESULT hr = hrOK;;
Trace2("CacheHint - Start %d, End %d\n", nStartIndex, nEndIndex);
return hr;
}
/*---------------------------------------------------------------------------
CFilterHandler::SortItems
We are responsible for sorting of virtual listbox items
Author: NSun
---------------------------------------------------------------------------*/
STDMETHODIMP
CFilterHandler::SortItems
(
int nColumn,
DWORD dwSortOptions,
LPARAM lUserParam
)
{
AFX_MANAGE_STATE(AfxGetStaticModuleState());
HRESULT hr = S_OK;
if (nColumn >= DimensionOf(aColumns[IPSECMON_FILTER]))
return E_INVALIDARG;
BEGIN_WAIT_CURSOR
DWORD dwIndexType = aColumns[IPSECMON_FILTER][nColumn];
hr = m_spSpdInfo->SortFilters(dwIndexType, dwSortOptions);
END_WAIT_CURSOR
return hr;
}
/*!--------------------------------------------------------------------------
CFilterHandler::OnResultUpdateView
Implementation of ITFSResultHandler::OnResultUpdateView
Author: NSun
---------------------------------------------------------------------------*/
HRESULT CFilterHandler::OnResultUpdateView
(
ITFSComponent *pComponent,
LPDATAOBJECT pDataObject,
LPARAM data,
LONG_PTR hint
)
{
HRESULT hr = hrOK;
SPITFSNode spSelectedNode;
pComponent->GetSelectedNode(&spSelectedNode);
if (spSelectedNode == NULL)
return S_OK; // no selection for our IComponentData
if ( hint == IPSECMON_UPDATE_STATUS )
{
SPINTERNAL spInternal = ExtractInternalFormat(pDataObject);
ITFSNode * pNode = reinterpret_cast<ITFSNode *>(spInternal->m_cookie);
SPITFSNode spSelectedNode;
pComponent->GetSelectedNode(&spSelectedNode);
if (pNode == spSelectedNode)
{
// if we are the selected node, then we need to update
SPIResultData spResultData;
CORg (pComponent->GetResultData(&spResultData));
CORg (spResultData->SetItemCount((int) data, MMCLV_UPDATE_NOSCROLL));
}
}
else
{
// we don't handle this message, let the base class do it.
return CIpsmHandler::OnResultUpdateView(pComponent, pDataObject, data, hint);
}
COM_PROTECT_ERROR_LABEL;
return hr;
}
/*!--------------------------------------------------------------------------
CFilterHandler::LoadColumns
Set the correct column header and then call the base class
Author: NSun
---------------------------------------------------------------------------*/
HRESULT
CFilterHandler::LoadColumns
(
ITFSComponent * pComponent,
MMC_COOKIE cookie,
LPARAM arg,
LPARAM lParam
)
{
//set column info
return CIpsmHandler::LoadColumns(pComponent, cookie, arg, lParam);
}
/*---------------------------------------------------------------------------
Command handlers
---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------
CFilterHandler::OnDelete
Removes a service SA
Author: NSun
---------------------------------------------------------------------------*/
HRESULT
CFilterHandler::OnDelete
(
ITFSNode * pNode
)
{
HRESULT hr = S_FALSE;
return hr;
}
/*---------------------------------------------------------------------------
CFilterHandler::UpdateStatus
-
Author: NSun
---------------------------------------------------------------------------*/
HRESULT
CFilterHandler::UpdateStatus
(
ITFSNode * pNode
)
{
HRESULT hr = hrOK;
SPIComponentData spComponentData;
SPIConsole spConsole;
IDataObject * pDataObject;
SPIDataObject spDataObject;
int i = 0;
Trace0("CFilterHandler::UpdateStatus - Updating status for Filter");
// force the listbox to update. We do this by setting the count and
// telling it to invalidate the data
CORg(m_spNodeMgr->GetComponentData(&spComponentData));
CORg(m_spNodeMgr->GetConsole(&spConsole));
// grab a data object to use
CORg(spComponentData->QueryDataObject((MMC_COOKIE) pNode, CCT_RESULT, &pDataObject) );
spDataObject = pDataObject;
m_spSpdInfo->ChangeQmFilterViewType(m_FltrType);
i = m_spSpdInfo->GetQmFilterCountOfCurrentViewType();
CORg(spConsole->UpdateAllViews(pDataObject, i, IPSECMON_UPDATE_STATUS));
COM_PROTECT_ERROR_LABEL;
return hr;
}
/*---------------------------------------------------------------------------
Misc functions
---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------
CFilterHandler::InitData
Initializes data for this node
Author: NSun
---------------------------------------------------------------------------*/
HRESULT
CFilterHandler::InitData
(
ISpdInfo * pSpdInfo
)
{
m_spSpdInfo.Set(pSpdInfo);
return hrOK;
}
HRESULT
CFilterHandler::UpdateViewType(ITFSNode * pNode, FILTER_TYPE NewFltrType)
{
// clear the listbox then set the size
HRESULT hr = hrOK;
SPIComponentData spCompData;
SPIConsole spConsole;
IDataObject* pDataObject;
SPIDataObject spDataObject;
LONG_PTR command;
int i;
COM_PROTECT_TRY
{
m_FltrType = NewFltrType;
//tell the spddb to update its index manager for QM filter
m_spSpdInfo->ChangeQmFilterViewType(m_FltrType);
i = m_spSpdInfo->GetQmFilterCountOfCurrentViewType();
m_spNodeMgr->GetComponentData(&spCompData);
CORg ( spCompData->QueryDataObject((MMC_COOKIE) pNode, CCT_RESULT, &pDataObject) );
spDataObject = pDataObject;
CORg ( m_spNodeMgr->GetConsole(&spConsole) );
//update the result pane virtual list
CORg ( spConsole->UpdateAllViews(spDataObject, i, RESULT_PANE_CLEAR_VIRTUAL_LB) );
COM_PROTECT_ERROR_LABEL;
}
COM_PROTECT_CATCH
return hr;
}