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.
 
 
 
 
 
 

3726 lines
113 KiB

//+-------------------------------------------------------------------------
//
// Microsoft Windows
// Copyright (C) Microsoft Corporation, 1996 - 2001.
//
// File: Snapin.cxx
//
// Contents: MMC snapin for CI.
//
// History: 26-Nov-1996 KyleP Created
// 20-Jan-1999 SLarimor Modified rescan interface to include
// Full and Incremental options separatly
//
//--------------------------------------------------------------------------
#include <pch.cxx>
#pragma hdrstop
#include <ISReg.hxx>
#include <CIARes.h>
#include <snapin.hxx>
#include <dataobj.hxx>
#include <propsht.hxx>
#include <callback.hxx>
#include <cmddlg.hxx>
#include <classid.hxx>
#include <catadmin.hxx>
#include <catalog.hxx>
#include <ntverp.h>
//
// Global data
//
//
// NOTE: Any new menus added to this array must be at the same location as
// the value of "lCommandID", which is the third member of the
// struct CONTEXTMENUITEM ie. comidAddScope = 0 below.
// Additionally, you should add another Win4Assert in the
// "CCISnapinData::Initialize" method below.
CONTEXTMENUITEM aContextMenus[] = {
// Add scope
{ STRINGRESOURCE( srCMScope ),
STRINGRESOURCE( srCMScopeHelp ),
comidAddScope,
CCM_INSERTIONPOINTID_PRIMARY_NEW,
MF_ENABLED,
0 },
// Add catalog
{ STRINGRESOURCE( srCMAddCatalog ),
STRINGRESOURCE( srCMAddCatalogHelp ),
comidAddCatalog,
CCM_INSERTIONPOINTID_PRIMARY_NEW,
MF_ENABLED,
0 },
// Rescan Full scope
{ STRINGRESOURCE( srCMRescanFull ),
STRINGRESOURCE( srCMRescanFullHelp ),
comidRescanFullScope,
CCM_INSERTIONPOINTID_PRIMARY_TASK,
MF_ENABLED,
0 },
// Force merge
{ STRINGRESOURCE( srCMMerge ),
STRINGRESOURCE( srCMMergeHelp ),
comidMergeCatalog,
CCM_INSERTIONPOINTID_PRIMARY_TASK,
MF_ENABLED,
0 },
// Start CI
{ STRINGRESOURCE( srCMStartCI ),
STRINGRESOURCE( srCMStartCIHelp ),
comidStartCI,
CCM_INSERTIONPOINTID_PRIMARY_TASK,
MF_ENABLED,
0 },
// Stop CI
{ STRINGRESOURCE( srCMStopCI ),
STRINGRESOURCE( srCMStopCIHelp ),
comidStopCI,
CCM_INSERTIONPOINTID_PRIMARY_TASK,
MF_ENABLED,
0 },
// Pause CI
{ STRINGRESOURCE( srCMPauseCI ),
STRINGRESOURCE( srCMPauseCIHelp ),
comidPauseCI,
CCM_INSERTIONPOINTID_PRIMARY_TASK,
MF_ENABLED,
0 },
// Refresh properties list
{ STRINGRESOURCE( srRefreshProperties ),
STRINGRESOURCE( srRefreshPropertiesHelp ),
comidRefreshProperties,
CCM_INSERTIONPOINTID_PRIMARY_TASK,
MF_ENABLED,
0 },
// Empty catalog
{ STRINGRESOURCE( srCMEmptyCatalog ),
STRINGRESOURCE( srCMEmptyCatalogHelp ),
comidEmptyCatalog,
CCM_INSERTIONPOINTID_PRIMARY_TASK,
MF_ENABLED,
0 },
// Tune performance
{ STRINGRESOURCE( srCMTunePerformance ),
STRINGRESOURCE( srCMTunePerformanceHelp ),
comidTunePerfCITop,
CCM_INSERTIONPOINTID_PRIMARY_TASK,
MF_ENABLED,
0 },
// Duplication of the commonly used menus
// at the top of the menu.
// Start CI
{ STRINGRESOURCE( srCMStartCI ),
STRINGRESOURCE( srCMStartCIHelp ),
comidStartCITop,
CCM_INSERTIONPOINTID_PRIMARY_TOP,
MF_ENABLED,
0 },
// Stop CI
{ STRINGRESOURCE( srCMStopCI ),
STRINGRESOURCE( srCMStopCIHelp ),
comidStopCITop,
CCM_INSERTIONPOINTID_PRIMARY_TOP,
MF_ENABLED,
0 },
// Pause CI
{ STRINGRESOURCE( srCMPauseCI ),
STRINGRESOURCE( srCMPauseCIHelp ),
comidPauseCITop,
CCM_INSERTIONPOINTID_PRIMARY_TOP,
MF_ENABLED,
0 },
// Rescan Incremental scope
{ STRINGRESOURCE( srCMRescanIncremental ),
STRINGRESOURCE( srCMRescanIncrementalHelp ),
comidRescanIncrementalScope,
CCM_INSERTIONPOINTID_PRIMARY_TASK,
MF_ENABLED,
0 },
};
static DWORD aIds[] =
{
IDOK, HIDP_OK,
IDCANCEL, HIDP_CANCEL,
// Catalog properties
IDDI_FILTER_UNKNOWN, HIDP_GENERATION_FILTER_UNKNOWN,
IDDI_CHARACTERIZATION, HIDP_GENERATION_GENERATE_CHARACTERIZATION,
IDDI_CHARSIZE_STATIC, HIDP_GENERATION_MAXIMUM_SIZE,
IDDI_CHARACTERIZATION_SIZE, HIDP_GENERATION_MAXIMUM_SIZE,
IDDI_SPIN_CHARACTERIZATION, HIDP_GENERATION_MAXIMUM_SIZE,
IDDI_SELECT_SIZE, HIDP_LOCATION_SIZE,
IDDI_SIZE, HIDP_LOCATION_SIZE,
IDDI_SELECT_PATH2, HIDP_CATALOG_LOCATION,
IDDI_PATH, HIDP_CATALOG_LOCATION,
IDDI_SELECT_PROPCACHE_SIZE, HIDP_PROPCACHE_SIZE,
IDDI_PROPCACHE_SIZE, HIDP_PROPCACHE_SIZE,
IDDI_VSERVER_STATIC, HIDP_WEB_VSERVER,
IDDI_VIRTUAL_SERVER, HIDP_WEB_VSERVER,
IDDI_NNTP_STATIC, HIDP_WEB_NNTPSERVER,
IDDI_NNTP_SERVER, HIDP_WEB_NNTPSERVER,
IDDI_AUTO_ALIAS, HIDP_ALIAS_NETWORK_SHARES,
IDDI_INHERIT1, HIDP_SETTINGS_INHERIT1,
IDDI_INHERIT2, HIDP_SETTINGS_INHERIT2,
IDDI_GROUP_INHERIT, HIDP_INHERIT,
IDDI_SELECT_CATNAME, HIDP_CATALOG_NAME,
IDDI_CATNAME, HIDP_CATALOG_NAME,
// New catalog
IDDI_SELECT_CATPATH, HIDP_LOCATION_LOCATION,
IDDI_CATPATH, HIDP_LOCATION_LOCATION,
IDDI_SELECT_CATNAME2, HIDP_LOCATION_NAME,
IDDI_CATNAME2, HIDP_LOCATION_NAME,
IDDI_BROWSE, HIDP_LOCATION_BROWSE,
// Property dialog box
IDDI_SELECT_PROPSET, HIDP_PROPERTY_SET,
IDDI_PROPSET, HIDP_PROPERTY_SET,
IDDI_SELECT_PROPERTY, HIDP_PROPERTY_PROPERTY,
IDDI_PROPERTY, HIDP_PROPERTY_PROPERTY,
IDDI_CACHED, HIDP_PROPERTY_CACHED,
IDDI_SELECT_DATATYPE, HIDP_PROPERTY_DATATYPE,
IDDI_DATATYPE, HIDP_PROPERTY_DATATYPE,
IDDI_SELECT_CACHEDSIZE, HIDP_PROPERTY_SIZE,
IDDI_CACHEDSIZE, HIDP_PROPERTY_SIZE,
IDDI_SPIN_CACHEDSIZE, HIDP_PROPERTY_SIZE,
IDDI_SELECT_STORAGELEVEL, HIDP_PROPERTY_STORAGELEVEL,
IDDI_STORAGELEVEL, HIDP_PROPERTY_STORAGELEVEL,
//IDDI_COMPNAME, ,
IDDI_LOCAL_COMPUTER, HIDP_CONNECT_LOCAL,
IDDI_REMOTE_COMPUTER, HIDP_CONNECT_ANOTHER,
IDDI_COMPNAME, HIDP_CONNECT_ANOTHER,
// New directory
IDDI_SELECT_PATH, HIDP_SCOPE_PATH,
IDDI_DIRPATH, HIDP_SCOPE_PATH,
IDDI_BROWSE, HIDP_SCOPE_BROWSE,
IDDI_SELECT_ALIAS, HIDP_SCOPE_ALIAS,
IDDI_ALIAS, HIDP_SCOPE_ALIAS,
IDDI_SELECT_USER_NAME, HIDP_SCOPE_USER_NAME,
IDDI_USER_NAME, HIDP_SCOPE_USER_NAME,
IDDI_SELECT_PASSWORD, HIDP_SCOPE_PASSWORD,
IDDI_PASSWORD, HIDP_SCOPE_PASSWORD,
IDDI_INCLUDE, HIDP_SCOPE_INCLUDE,
IDDI_EXCLUDE, HIDP_SCOPE_EXCLUDE,
IDDI_ACCOUNT_INFORMATION, HIDP_ACCOUNT_INFORMATION,
IDDI_INCLUSION, HIDP_INCLUSION,
// Performance Tuning
IDDI_DEDICATED, HIDP_DEDICATED,
IDDI_USEDOFTEN, HIDP_USEDOFTEN,
IDDI_USEDOCCASIONALLY, HIDP_USEDOCCASIONALLY,
IDDI_NEVERUSED, HIDP_NEVERUSED,
IDDI_CUSTOMIZE, HIDP_CUSTOMIZE,
IDDI_ADVANCED, HIDP_ADVANCED_CONFIG,
IDDI_SELECT_INDEXING, HIDP_INDEXING_PERFORMANCE,
IDDI_SLIDER_INDEXING, HIDP_INDEXING_PERFORMANCE,
IDDI_SELECT_QUERYING, HIDP_QUERY_PERFORMANCE,
IDDI_SLIDER_QUERYING, HIDP_QUERY_PERFORMANCE,
0, 0 };
MMCBUTTON aContextButtons[] =
{
// Start CI
{ comidStartCIButton, // 0
comidStartCITop,
TBSTATE_ENABLED,
TBSTYLE_BUTTON,
STRINGRESOURCE( srCMStartCI ),
STRINGRESOURCE( srCMStartCIHelp )
},
// Stop CI
{ comidStopCIButton, // 1
comidStopCITop,
TBSTATE_ENABLED,
TBSTYLE_BUTTON,
STRINGRESOURCE( srCMStopCI ),
STRINGRESOURCE( srCMStopCIHelp )
},
// Pause CI
{ comidPauseCIButton, // 1
comidPauseCITop,
TBSTATE_ENABLED,
TBSTYLE_BUTTON,
STRINGRESOURCE( srCMPauseCI ),
STRINGRESOURCE( srCMPauseCIHelp )
}
};
//
// Registry constants
//
WCHAR const wszSnapinPath[] = L"Software\\Microsoft\\MMC\\SnapIns\\";
WCHAR const wszTriedEnable[] = L"TriedEnable";
//
// Global variables
//
extern long gulcInstances;
extern CStaticMutexSem gmtxTimer;
HINSTANCE ghInstance;
DECLARE_INFOLEVEL(cia)
//
// Function prototypes
//
SCODE StartMenu( IContextMenuCallback * piCallback, BOOL fTop = TRUE );
SCODE StopMenu( IContextMenuCallback * piCallback, BOOL fTop = TRUE );
SCODE PauseMenu( IContextMenuCallback * piCallback, BOOL fTop = TRUE );
SCODE DisabledMenu( IContextMenuCallback * piCallback, BOOL fTop = TRUE );
SCODE SetStartStopMenu( IContextMenuCallback * piCallback, BOOL fTop = TRUE );
BOOL WINAPI DllMain( HANDLE hInstance, DWORD dwReason, LPVOID dwReserved )
{
BOOL fRetval = TRUE;
TRANSLATE_EXCEPTIONS;
TRY
{
if ( dwReason == DLL_PROCESS_ATTACH )
{
ghInstance = (HINSTANCE)hInstance;
gmtxTimer.Init();
InitStrings( ghInstance );
}
}
CATCH( CException, e )
{
// About the only thing this could be is STATUS_NO_MEMORY which
// can be thrown by InitializeCriticalSection.
ciaDebugOut(( DEB_ERROR,
"CIADMIN: Exception %#x in DllMain\n",
e.GetErrorCode()));
#if CIDBG == 1 // for debugging NTRAID 340297
if (e.GetErrorCode() == STATUS_NO_MEMORY)
DbgPrint( "CIADMIN: STATUS_NO_MEMORY exception in DllMain\n");
else
DbgPrint( "CIADMIN: ??? Exception in DllMain\n");
#endif // CIDBG == 1
fRetval = FALSE;
}
END_CATCH
UNTRANSLATE_EXCEPTIONS;
return fRetval;
}
// Detect if we are running on a server or a workstation
BOOL IsNTServer()
{
BOOL fServer = FALSE;
CRegAccess reg( RTL_REGISTRY_CONTROL,
L"ProductOptions" );
WCHAR awcProductType[ MAX_PATH ];
reg.Get( L"ProductType",
awcProductType,
sizeof awcProductType / sizeof WCHAR );
// winnt, pdc/bdc, server.
// note: 4.0 bdcs are LanmanNt, 5.0+ are spec'ed to be LansecNt
if ( !_wcsicmp( awcProductType, L"WinNt" ) )
fServer = FALSE;
else if ( ( !_wcsicmp( awcProductType, L"LanmanNt" ) ) ||
( !_wcsicmp( awcProductType, L"LansecNt" ) ) ||
( !_wcsicmp( awcProductType, L"ServerNt" ) ) )
fServer = TRUE;
return fServer;
} //IsNTServer
//+-------------------------------------------------------------------------
//
// Method: CCISnapinData::QueryInterface
//
// Synopsis: Switch from one interface to another
//
// History: 26-Nov-1996 KyleP Created
//
//--------------------------------------------------------------------------
SCODE STDMETHODCALLTYPE CCISnapinData::QueryInterface( REFIID riid,
void ** ppvObject )
{
ciaDebugOut(( DEB_ITRACE, "CCISnapin::QueryInterface " ));
SCODE sc = S_OK;
if ( 0 == ppvObject )
return E_INVALIDARG;
if ( IID_IComponentData == riid )
*ppvObject = (IUnknown *)(IComponentData *)this;
else if ( IID_IExtendPropertySheet == riid )
*ppvObject = (IUnknown *)(IExtendPropertySheet *)this;
else if ( IID_IExtendContextMenu == riid )
*ppvObject = (IUnknown *)(IExtendContextMenu *)this;
else if ( IID_IPersistStream == riid )
*ppvObject = (IUnknown *)(IPersistStream *)this;
else if ( IID_ISnapinAbout == riid )
*ppvObject = (IUnknown *)(ISnapinAbout *)this;
else if ( IID_ISnapinHelp == riid )
*ppvObject = (IUnknown *)(ISnapinHelp *)this;
else if ( IID_IExtendControlbar == riid )
*ppvObject = (IUnknown *)(IExtendControlbar *)this;
else if ( IID_IUnknown == riid )
*ppvObject = (IUnknown *)(IComponentData *)this;
else
sc = E_NOINTERFACE;
if ( SUCCEEDED( sc ) )
AddRef();
return sc;
} //QueryInterface
//+-------------------------------------------------------------------------
//
// Method: CCISnapinData::AddRef
//
// Synopsis: Increment ref count
//
// History: 26-Nov-1996 KyleP Created
//
//--------------------------------------------------------------------------
ULONG STDMETHODCALLTYPE CCISnapinData::AddRef()
{
ciaDebugOut(( DEB_ITRACE, "CCISnapinData::AddRef\n" ));
return InterlockedIncrement( &_uRefs );
}
//+-------------------------------------------------------------------------
//
// Method: CCISnapinData::Release
//
// Synopsis: Deccrement ref count
//
// History: 26-Nov-1996 KyleP Created
//
//--------------------------------------------------------------------------
ULONG STDMETHODCALLTYPE CCISnapinData::Release()
{
ciaDebugOut(( DEB_ITRACE, "CCISnapinData::Release\n" ));
unsigned long uTmp = InterlockedDecrement( &_uRefs );
if ( 0 == uTmp )
delete this;
return(uTmp);
}
SCODE STDMETHODCALLTYPE CCISnapinData::Initialize( IUnknown * pUnk )
{
ciaDebugOut(( DEB_ITRACE, "CCISnapinData::Initialize\n" ));
//
// Parameter checking.
//
if ( 0 == pUnk )
return E_INVALIDARG;
// Ensure that contex menus are in correct location in menu array.
Win4Assert( comidAddScope == aContextMenus[comidAddScope].lCommandID );
Win4Assert( comidAddCatalog == aContextMenus[comidAddCatalog].lCommandID );
Win4Assert( comidRescanFullScope == aContextMenus[comidRescanFullScope].lCommandID );
Win4Assert( comidMergeCatalog == aContextMenus[comidMergeCatalog].lCommandID );
Win4Assert( comidStartCI == aContextMenus[comidStartCI].lCommandID );
Win4Assert( comidStopCI == aContextMenus[comidStopCI].lCommandID );
Win4Assert( comidPauseCI == aContextMenus[comidPauseCI].lCommandID );
Win4Assert( comidRefreshProperties == aContextMenus[comidRefreshProperties].lCommandID );
Win4Assert( comidEmptyCatalog == aContextMenus[comidEmptyCatalog].lCommandID );
Win4Assert( comidTunePerfCITop == aContextMenus[comidTunePerfCITop].lCommandID );
Win4Assert( comidStartCITop == aContextMenus[comidStartCITop].lCommandID );
Win4Assert( comidStopCITop == aContextMenus[comidStopCITop].lCommandID );
Win4Assert( comidPauseCITop == aContextMenus[comidPauseCITop].lCommandID );
Win4Assert( comidRescanIncrementalScope == aContextMenus[comidRescanIncrementalScope].lCommandID );
TRANSLATE_EXCEPTIONS;
SCODE sc;
TRY
{
do
{
//
// Collect interfaces
//
sc = pUnk->QueryInterface( IID_IConsole, (void **)&_pFrame );
if ( FAILED(sc) )
break;
sc = _pFrame->GetMainWindow(&_hFrameWindow);
if ( FAILED(sc) )
break;
//sc = _pFrame->QueryInterface( IID_IHeaderCtrl, (void **)&_pHeader );
//
//if ( FAILED(sc) )
// break;
sc = _pFrame->QueryInterface(IID_IConsoleNameSpace, (void **)&_pScopePane );
if ( FAILED(sc) )
break;
_rootNode.Init( _pScopePane );
_Catalogs.Init( _pScopePane );
//
// Initialize resources
//
IImageList * pImageList;
sc = _pFrame->QueryScopeImageList( &pImageList );
if ( FAILED(sc) )
break;
sc = InitImageResources( pImageList );
pImageList->Release();
if ( FAILED(sc) )
break;
}
while ( FALSE );
}
CATCH( CException, e )
{
sc = e.GetErrorCode();
ciaDebugOut(( DEB_ERROR, "Exception 0x%x caught in CCISnapinData::Initialize\n", sc ));
}
END_CATCH
UNTRANSLATE_EXCEPTIONS;
//
// Damage control
//
if ( FAILED(sc) )
{
if ( 0 != _pScopePane )
{
_pScopePane->Release();
_pScopePane = 0;
}
if ( 0 != _pFrame )
{
_pFrame->Release();
_pFrame = 0;
}
}
return sc;
}
SCODE STDMETHODCALLTYPE CCISnapinData::CreateComponent( IComponent * * ppComponent )
{
TRANSLATE_EXCEPTIONS;
SCODE sc = S_OK;
TRY
{
CCISnapin * p = new CCISnapin( *this, _pChild );
*ppComponent = p;
_pChild = p;
}
CATCH( CException, e )
{
sc = E_FAIL;
}
END_CATCH
UNTRANSLATE_EXCEPTIONS;
return sc;
}
SCODE STDMETHODCALLTYPE CCISnapinData::Notify( IDataObject * pDO,
MMC_NOTIFY_TYPE event,
LPARAM arg,
LPARAM param )
{
ciaDebugOut(( DEB_ITRACE,
"CCISnapinData::Notify (pDO = 0x%x, event = 0x%x)\n",
pDO, event ));
TRANSLATE_EXCEPTIONS;
SCODE sc = S_OK;
TRY
{
switch ( event )
{
case MMCN_EXPAND:
ShowFolder( (CCIAdminDO *)pDO, // Cookie
(BOOL)arg, // TRUE --> Expand, FALSE --> Contract
(HSCOPEITEM)param ); // Scope item selected
break;
case MMCN_DELETE:
RemoveCatalog( (CCIAdminDO *)pDO );
break;
case MMCN_REMOVE_CHILDREN:
if ( _rootNode.IsParent( (HSCOPEITEM)arg ) )
{
Win4Assert( _fIsExtension );
sc = _Catalogs.ReInit();
if ( SUCCEEDED(sc) )
sc = _rootNode.Delete();
_fIsInitialized = FALSE;
}
break;
default:
ciaDebugOut(( DEB_WARN, "unhandled notify 0x%x\n", event ));
break;
}
}
CATCH( CException, e )
{
ciaDebugOut(( DEB_ERROR, "Exception 0x%x in CCISnapinData::Notify\n", e.GetErrorCode() ));
sc = E_UNEXPECTED;
}
END_CATCH
UNTRANSLATE_EXCEPTIONS;
return sc;
}
SCODE STDMETHODCALLTYPE CCISnapinData::Destroy()
{
ciaDebugOut(( DEB_ITRACE, "CCISnapinData::Destroy\n" ));
if ( 0 != _pFrame )
{
_pFrame->Release();
_pFrame = 0;
_pScopePane->Release();
_pScopePane = 0;
for ( CCISnapin * pCurrent = _pChild;
0 != pCurrent;
pCurrent = pCurrent->Next() )
{
pCurrent->ClearSnapinData();
}
// It's MMC's responsibility to destroy IConsole pointer we give them
// I checked with Gautam in mmc team and confirmed this.
_pChild = 0;
}
return S_OK;
}
SCODE STDMETHODCALLTYPE CCISnapinData::QueryDataObject( MMC_COOKIE cookie,
DATA_OBJECT_TYPES type,
IDataObject * * ppDataObject )
{
//ciaDebugOut(( DEB_ITRACE,
// "CCISnapinData::QueryDataObject (cookie = 0x%x, type = 0x%x)\n",
// cookie, type ));
//
// Minimal parameter validation
//
if ( 0 == ppDataObject )
return E_INVALIDARG;
//
// Create a data object
//
TRANSLATE_EXCEPTIONS;
SCODE sc = S_OK;
TRY
{
*ppDataObject = new CCIAdminDO( cookie, type, _fIsExtension ? 0 : _Catalogs.GetMachine() );
}
CATCH( CException, e )
{
sc = E_UNEXPECTED;
}
END_CATCH
UNTRANSLATE_EXCEPTIONS;
return sc;
}
SCODE STDMETHODCALLTYPE CCISnapinData::GetDisplayInfo( SCOPEDATAITEM * pScopeDataItem )
{
//ciaDebugOut(( DEB_ITRACE, "CCISnapinData::GetDisplayInfo( 0x%x )\n", pScopeDataItem ));
if (pScopeDataItem == NULL)
return E_POINTER;
TRANSLATE_EXCEPTIONS;
SCODE sc = E_UNEXPECTED;
TRY
{
PCIObjectType * pType = (PCIObjectType *)pScopeDataItem->lParam;
if ( 0 == pType )
return sc;
switch ( pType->Type() )
{
case PCIObjectType::RootNode:
{
if ( pScopeDataItem->mask == SDI_STR )
{
pScopeDataItem->displayname = (WCHAR *)GetRootDisplay();
sc = S_OK;
}
break;
}
case PCIObjectType::Catalog:
{
CCatalog * pCat = (CCatalog *)pScopeDataItem->lParam;
if ( pScopeDataItem->mask == SDI_STR )
{
pScopeDataItem->displayname = (WCHAR *)pCat->GetCat( TRUE );
sc = S_OK;
}
break;
}
case PCIObjectType::Intermediate_Scope:
{
if ( pScopeDataItem->mask == SDI_STR )
{
pScopeDataItem->displayname = STRINGRESOURCE( srNodeDirectories );
sc = S_OK;
}
break;
}
case PCIObjectType::Intermediate_Properties:
{
if ( pScopeDataItem->mask == SDI_STR )
{
pScopeDataItem->displayname = STRINGRESOURCE( srNodeProperties );
sc = S_OK;
}
break;
}
case PCIObjectType::Intermediate_UnfilteredURL:
{
if ( pScopeDataItem->mask == SDI_STR )
{
pScopeDataItem->displayname = STRINGRESOURCE( srNodeUnfiltered );
sc = S_OK;
}
break;
}
case PCIObjectType::Property:
case PCIObjectType::Directory:
default:
Win4Assert( !"Oops!" );
break;
}
}
CATCH( CException, e )
{
ciaDebugOut(( DEB_ERROR, "Exception 0x%x in CCISnapinData::GetDisplayInfo\n", e.GetErrorCode() ));
sc = E_UNEXPECTED;
}
END_CATCH
UNTRANSLATE_EXCEPTIONS;
return sc;
}
SCODE STDMETHODCALLTYPE CCISnapinData::CompareObjects( IDataObject * lpDataObjectA,
IDataObject * lpDataObjectB )
{
ciaDebugOut(( DEB_ITRACE, "CCISnapinData::CompareObjects( 0x%x, 0x%x )\n", lpDataObjectA, lpDataObjectB ));
if ( *((CCIAdminDO *)lpDataObjectA) == *((CCIAdminDO *)lpDataObjectB) )
return S_OK;
else
return S_FALSE;
}
SCODE STDMETHODCALLTYPE CCISnapinData::AddMenuItems( IDataObject * piDataObject,
IContextMenuCallback * piCallback,
long * pInsertionAllowed )
{
if ( 0 == piDataObject )
return E_UNEXPECTED;
if ( 0 == (*pInsertionAllowed & CCM_INSERTIONALLOWED_TOP ) )
{
ciaDebugOut(( DEB_WARN, "Menu Insertion not allowed.\n" ));
return S_OK;
}
TRANSLATE_EXCEPTIONS;
SCODE sc = S_OK;
TRY
{
CCIAdminDO * pDO = (CCIAdminDO *)piDataObject;
ciaDebugOut(( DEB_ITRACE, "CCISnapinData::AddMenuItems (cookie = 0x%x, type = 0x%x)\n",
pDO->Cookie(), pDO->Type() ));
if ( pDO->Type() == CCT_RESULT )
{
if ( SUCCEEDED(sc) && pDO->IsACatalog() )
{
sc = piCallback->AddItem( &aContextMenus[comidMergeCatalog] );
}
else if ( SUCCEEDED(sc) && pDO->IsADirectory() )
{
CMachineAdmin MachineAdmin( _Catalogs.IsLocalMachine() ? 0 : _Catalogs.GetMachine() );
if (MachineAdmin.IsCIStarted())
{
// Add both Full and Incremental rescan options to menu.
sc = piCallback->AddItem( &aContextMenus[comidRescanFullScope] );
if ( SUCCEEDED(sc) )
sc = piCallback->AddItem( &aContextMenus[comidRescanIncrementalScope] );
}
}
else
ciaDebugOut(( DEB_WARN, "No menu items for (cookie = 0x%x, type = 0x%x)\n",
pDO->Cookie(), pDO->Type() ));
}
else if ( SUCCEEDED(sc) && pDO->Type() == CCT_SCOPE )
{
if ( (pDO->IsRoot() || pDO->IsStandAloneRoot()) )
{
if (*pInsertionAllowed & CCM_INSERTIONALLOWED_TASK )
{
//
// Adjust state accordingly.
//
CMachineAdmin MachineAdmin( _Catalogs.IsLocalMachine() ? 0 : _Catalogs.GetMachine() );
if ( MachineAdmin.IsCIStarted() )
{
sc = StartMenu( piCallback );
if (SUCCEEDED(sc))
{
aContextMenus[comidTunePerfCITop].fFlags = MF_ENABLED;
sc = piCallback->AddItem( &aContextMenus[comidTunePerfCITop] );
}
}
else if ( MachineAdmin.IsCIPaused() )
{
sc = PauseMenu( piCallback );
if (SUCCEEDED(sc))
{
aContextMenus[comidTunePerfCITop].fFlags = MF_ENABLED;
sc = piCallback->AddItem( &aContextMenus[comidTunePerfCITop] );
}
}
else
{
Win4Assert( MachineAdmin.IsCIStopped() );
sc = StopMenu( piCallback );
if (SUCCEEDED(sc))
{
aContextMenus[comidTunePerfCITop].fFlags = MF_ENABLED;
sc = piCallback->AddItem( &aContextMenus[comidTunePerfCITop] );
}
}
}
if (SUCCEEDED(sc) && (*pInsertionAllowed & CCM_INSERTIONALLOWED_NEW ) )
{
sc = piCallback->AddItem( &aContextMenus[comidAddCatalog] );
}
}
else if ( pDO->IsACatalog() )
{
if (SUCCEEDED(sc) && (*pInsertionAllowed & CCM_INSERTIONALLOWED_NEW ) )
{
sc = piCallback->AddItem( &aContextMenus[comidAddScope] );
}
if (SUCCEEDED(sc) && (*pInsertionAllowed & CCM_INSERTIONALLOWED_TASK ) )
{
CMachineAdmin MachineAdmin( _Catalogs.IsLocalMachine() ? 0 : _Catalogs.GetMachine() );
//
// Only append individual catalog control if service is running.
//
if ( MachineAdmin.IsCIStarted() || MachineAdmin.IsCIPaused() )
{
// Menu item to trigger clean up of the catalog
aContextMenus[comidEmptyCatalog].fFlags = MF_GRAYED;
piCallback->AddItem( &aContextMenus[comidEmptyCatalog] );
aContextMenus[comidMergeCatalog].fFlags = MF_ENABLED;
XPtr<CCatalogAdmin> xCat( MachineAdmin.QueryCatalogAdmin( pDO->GetCatalog()->GetCat(TRUE) ) );
if ( xCat->IsStarted() )
{
aContextMenus[comidMergeCatalog].fFlags = MF_ENABLED;
sc = StartMenu( piCallback, FALSE );
}
else if ( xCat->IsPaused() )
{
aContextMenus[comidMergeCatalog].fFlags = MF_GRAYED;
sc = PauseMenu( piCallback, FALSE );
}
else
{
Win4Assert( xCat->IsStopped() );
aContextMenus[comidMergeCatalog].fFlags = MF_GRAYED;
sc = StopMenu( piCallback, FALSE );
}
}
else
{
// Menu item to trigger clean up of the catalog
aContextMenus[comidEmptyCatalog].fFlags = MF_ENABLED;
piCallback->AddItem( &aContextMenus[comidEmptyCatalog] );
aContextMenus[comidMergeCatalog].fFlags = MF_GRAYED;
sc = DisabledMenu( piCallback, FALSE );
}
if (SUCCEEDED(sc))
{
sc = piCallback->AddItem( &aContextMenus[comidMergeCatalog] );
}
}
}
else if ( pDO->IsADirectoryIntermediate() && (*pInsertionAllowed & CCM_INSERTIONALLOWED_NEW ) )
{
if (SUCCEEDED(sc))
{
sc = piCallback->AddItem( &aContextMenus[comidAddScope] );
}
}
else if ( pDO->IsAPropertyIntermediate() && (*pInsertionAllowed & CCM_INSERTIONALLOWED_TASK ) )
{
if (SUCCEEDED(sc))
{
sc = piCallback->AddItem( &aContextMenus[comidRefreshProperties] );
}
}
else
{
ciaDebugOut(( DEB_WARN, "No menu items for (cookie = 0x%x, type = 0x%x)\n",
pDO->Cookie(), pDO->Type() ));
}
}
else
{
ciaDebugOut(( DEB_WARN, "No menu items for (cookie = 0x%x, type = 0x%x)\n",
pDO->Cookie(), pDO->Type() ));
}
}
CATCH( CException, e )
{
ciaDebugOut(( DEB_ERROR, "Exception 0x%x in CCISnapinData::AddMenuItems\n", e.GetErrorCode() ));
sc = E_UNEXPECTED;
}
END_CATCH
UNTRANSLATE_EXCEPTIONS;
return sc;
}
SCODE STDMETHODCALLTYPE CCISnapinData::Command( long lCommandID,
IDataObject * piDataObject )
{
if ( 0 == piDataObject )
return E_POINTER;
TRANSLATE_EXCEPTIONS;
SCODE sc = S_OK;
TRY
{
CCIAdminDO * pDO = (CCIAdminDO *)piDataObject;
switch ( lCommandID )
{
case comidRescanFullScope:
{
CScope * pScope = pDO->GetScope();
Win4Assert( 0 != pScope );
//
// Make sure the user wants to do a full rescan.
//
WCHAR awcMsg[MAX_PATH];
WCHAR awcTemp[2 * MAX_PATH];
LoadString( ghInstance, MSG_RESCAN_FULL_SCOPE_EXPLAIN, awcMsg, sizeof(awcMsg) / sizeof(WCHAR) );
wsprintf( awcTemp, awcMsg, pScope->GetPath() );
int iResult;
// Pop up YES/NO box
sc = _pFrame->MessageBox( awcTemp,
STRINGRESOURCE( srMsgRescanFull ),
MB_YESNO | /* MB_HELP | */
MB_ICONWARNING | MB_DEFBUTTON2 |
MB_APPLMODAL,
&iResult );
if ( SUCCEEDED(sc) )
{
switch ( iResult )
{
// Rescan( TRUE ) means "Full scan"
case IDYES:
{
pScope->Rescan( TRUE );
break;
}
case IDNO:
{
// Do nothing.
break;
}
/* Help is not being used...
case IDHELP:
{
DisplayHelp( _hFrameWindow, HIDD_RESCAN );
break;
}
*/
default:
break;
}
}
break;
}
case comidRescanIncrementalScope:
{
CScope * pScope = pDO->GetScope();
Win4Assert( 0 != pScope );
//
// Make sure the user wants to do an incremental rescan.
//
WCHAR awcMsg[MAX_PATH];
WCHAR awcTemp[2 * MAX_PATH];
LoadString( ghInstance, MSG_RESCAN_INCREMENTAL_SCOPE_EXPLAIN, awcMsg, sizeof(awcMsg) / sizeof(WCHAR) );
wsprintf( awcTemp, awcMsg, pScope->GetPath() );
int iResult;
// Pop up YES/NO box
sc = _pFrame->MessageBox( awcTemp,
STRINGRESOURCE( srMsgRescanIncremental ),
MB_YESNO | /* MB_HELP | */
MB_ICONWARNING | MB_DEFBUTTON2 |
MB_APPLMODAL,
&iResult );
if ( SUCCEEDED(sc) )
{
switch ( iResult )
{
// Rescan( FALSE ) means "Incremental scan"
case IDYES:
{
pScope->Rescan( FALSE );
break;
}
case IDNO:
{
// Do nothing.
break;
}
/* Help is not being used...
case IDHELP:
{
DisplayHelp( _hFrameWindow, HIDD_RESCAN );
break;
}
*/
default:
break;
}
}
break;
}
case comidRefreshProperties:
{
// The user has explicitly requested a refresh of the list.
// Delete the old list and recreate it.
CCatalog * pcat = pDO->GetCatalog();
BOOL fClearedYet = FALSE;
for ( CCISnapin * pCurrent = _pChild;
0 != pCurrent;
pCurrent = pCurrent->Next() )
{
if ( CCISnapin::Properties == pCurrent->GetCurrentView() &&
pCurrent->GetCurrentCatalog() == pcat )
{
if ( !fClearedYet )
{
pcat->ClearProperties( pCurrent->ResultPane() );
fClearedYet = TRUE;
}
pCurrent->ResultPane()->DeleteAllRsltItems();
pcat->DisplayProperties( TRUE, pCurrent->ResultPane() );
}
}
break;
}
case comidEmptyCatalog:
{
// NOTE: The service must be stopped before we can do this. Deleting
// files under an active catalog leads to unpredictable results.
int iResult;
sc = _pFrame->MessageBox( STRINGRESOURCE( srMsgEmptyCatalogPrompt ),
STRINGRESOURCE( srMsgEmptyCatalogAsk ),
MB_YESNO | /* MB_HELP | */
MB_ICONWARNING | MB_DEFBUTTON2 |
MB_APPLMODAL,
&iResult );
if ( SUCCEEDED(sc) )
{
switch ( iResult )
{
case IDYES:
{
CMachineAdmin MachineAdmin( _Catalogs.IsLocalMachine() ? 0 : _Catalogs.GetMachine() );
XPtr<CCatalogAdmin> xCat( MachineAdmin.QueryCatalogAdmin( pDO->GetCatalog()->GetCat(TRUE) ) );
//xCat->EmptyThisCatalog();
MachineAdmin.RemoveCatalogFiles(pDO->GetCatalog()->GetCat(TRUE));
break;
}
case IDNO:
// Do nothing.
break;
default:
break;
}
}
break;
}
case comidAddScope:
{
CCatalog * pCat = pDO->GetCatalog();
Win4Assert( 0 != pCat );
INT_PTR err = DialogBoxParam( ghInstance, // Application instance
MAKEINTRESOURCE( IDD_ADD_SCOPE ), // Dialog box
_hFrameWindow, // main window
AddScopeDlg, // Dialog box function
(LPARAM)pCat ); // User parameter
if ( -1 == err )
THROW( CException() );
Refresh(); // Update all result pane(s)
break;
}
case comidMergeCatalog:
{
CCatalog * pCat = pDO->GetCatalog();
Win4Assert( 0 != pCat );
//
// Make sure the user wants to remove scope.
//
int iResult;
sc = _pFrame->MessageBox( STRINGRESOURCE( srMsgMerge ),
STRINGRESOURCE( srMsgMerge ),
MB_YESNO | /* MB_HELP | */
MB_ICONWARNING | MB_DEFBUTTON2 |
MB_APPLMODAL,
&iResult );
if ( SUCCEEDED(sc) )
{
switch ( iResult )
{
case IDYES:
{
pCat->Merge();
break;
}
case IDNO:
case IDCANCEL:
// Do nothing.
break;
/* Help is not being used...
case IDHELP:
{
DisplayHelp( _hFrameWindow, HIDD_MERGE_CATALOG );
break;
}
*/
default:
break;
}
}
break;
}
case comidAddCatalog:
{
Win4Assert( pDO->IsRoot() || pDO->IsStandAloneRoot() );
INT_PTR err = DialogBoxParam( ghInstance, // Application instance
MAKEINTRESOURCE( IDD_ADD_CATALOG ), // Dialog box
_hFrameWindow, // main frame window
AddCatalogDlg, // Dialog box function
(LPARAM)&_Catalogs ); // User parameter
if ( -1 == err )
THROW( CException() );
// Ensure that only and all active catalogs are listed.
_Catalogs.UpdateActiveState();
Refresh(); // Update all result pane(s)
break;
}
case comidStartCI:
case comidStartCITop:
{
CMachineAdmin MachineAdmin( _Catalogs.IsLocalMachine() ? 0 : _Catalogs.GetMachine() );
if ( pDO->IsRoot() || pDO->IsStandAloneRoot() )
{
MaybeEnableCI( MachineAdmin );
MachineAdmin.StartCI();
// Ensure that only and all active catalogs are listed.
_Catalogs.UpdateActiveState();
_Catalogs.SetSnapinData( this );
}
else
{
Win4Assert( pDO->IsACatalog() );
XPtr<CCatalogAdmin> xCat( MachineAdmin.QueryCatalogAdmin( pDO->GetCatalog()->GetCat(TRUE) ) );
xCat->Start();
// We are using the top-level buttons to control individual catalogs. Accordingly, they
// should be updated to reflect the selected catalog's state.
// The toolbar may not be created yet, so check if it is non-zero
if ( xCat->IsStarted() && _pChild->_xToolbar.GetPointer() )
{
// This is more efficient than calling CCISnapinData::SetButtonState
for ( CCISnapin * pCurrent = _pChild;
0 != pCurrent;
pCurrent = pCurrent->Next() )
{
pCurrent->_xToolbar->SetButtonState(comidStartCITop, ENABLED, FALSE);
pCurrent->_xToolbar->SetButtonState(comidStopCITop, ENABLED, TRUE);
pCurrent->_xToolbar->SetButtonState(comidPauseCITop, ENABLED, TRUE);
}
}
}
break;
}
case comidStopCI:
case comidStopCITop:
{
CMachineAdmin MachineAdmin( _Catalogs.IsLocalMachine() ? 0 : _Catalogs.GetMachine() );
if ( pDO->IsRoot() || pDO->IsStandAloneRoot() )
{
MachineAdmin.StopCI();
// The toolbar may not be created yet, so check if it is non-zero
if ( MachineAdmin.IsCIStopped() && _pChild->_xToolbar.GetPointer() )
{
for ( CCISnapin * pCurrent = _pChild;
0 != pCurrent;
pCurrent = pCurrent->Next() )
{
pCurrent->_xToolbar->SetButtonState(comidStartCITop, ENABLED, TRUE);
pCurrent->_xToolbar->SetButtonState(comidStopCITop, ENABLED, FALSE);
pCurrent->_xToolbar->SetButtonState(comidPauseCITop, ENABLED, FALSE);
}
}
// Ensure that only and all active catalogs are listed.
_Catalogs.UpdateActiveState();
}
else
{
Win4Assert( pDO->IsACatalog() );
XPtr<CCatalogAdmin> xCat( MachineAdmin.QueryCatalogAdmin( pDO->GetCatalog()->GetCat(TRUE) ) );
xCat->Stop();
// The toolbar may not be created yet, so check if it is non-zero
if ( xCat->IsStopped() && _pChild->_xToolbar.GetPointer() )
{
for ( CCISnapin * pCurrent = _pChild;
0 != pCurrent;
pCurrent = pCurrent->Next() )
{
pCurrent->_xToolbar->SetButtonState(comidStartCITop, ENABLED, TRUE);
pCurrent->_xToolbar->SetButtonState(comidStopCITop, ENABLED, FALSE);
pCurrent->_xToolbar->SetButtonState(comidPauseCITop, ENABLED, FALSE);
}
}
}
break;
}
case comidPauseCI:
case comidPauseCITop:
{
CMachineAdmin MachineAdmin( _Catalogs.IsLocalMachine() ? 0 : _Catalogs.GetMachine() );
if ( pDO->IsRoot() || pDO->IsStandAloneRoot() )
{
MachineAdmin.PauseCI();
// The toolbar may not be created yet, so check if it is non-zero
if ( MachineAdmin.IsCIPaused() && _pChild->_xToolbar.GetPointer() )
{
for ( CCISnapin * pCurrent = _pChild;
0 != pCurrent;
pCurrent = pCurrent->Next() )
{
pCurrent->_xToolbar->SetButtonState(comidStartCITop, ENABLED, TRUE);
pCurrent->_xToolbar->SetButtonState(comidStopCITop, ENABLED, TRUE);
pCurrent->_xToolbar->SetButtonState(comidPauseCITop, ENABLED, FALSE);
}
}
// Ensure that only and all active catalogs are listed.
_Catalogs.UpdateActiveState();
}
else
{
Win4Assert( pDO->IsACatalog() );
XPtr<CCatalogAdmin> xCat( MachineAdmin.QueryCatalogAdmin( pDO->GetCatalog()->GetCat(TRUE) ) );
xCat->Pause();
// The toolbar may bot be created yet, so check if it is non-zero
if ( xCat->IsPaused() && _pChild->_xToolbar.GetPointer() )
{
for ( CCISnapin * pCurrent = _pChild;
0 != pCurrent;
pCurrent = pCurrent->Next() )
{
pCurrent->_xToolbar->SetButtonState(comidStartCITop, ENABLED, TRUE);
pCurrent->_xToolbar->SetButtonState(comidStopCITop, ENABLED, TRUE);
pCurrent->_xToolbar->SetButtonState(comidPauseCITop, ENABLED, FALSE);
}
}
}
break;
}
case comidTunePerfCITop:
{
Win4Assert( pDO->IsRoot() || pDO->IsStandAloneRoot() );
if ( IsNTServer() )
{
DialogBoxParam( ghInstance, // Application instance
MAKEINTRESOURCE( IDD_USAGE_ON_SERVER ), // Dialog box
_hFrameWindow, // main frame window
SrvTunePerfDlg, // Dialog box function
(LPARAM)&_Catalogs ); // User parameter
}
else
{
DialogBoxParam( ghInstance, // Application instance
MAKEINTRESOURCE( IDD_USAGE_ON_WORKSTATION ), // Dialog box
_hFrameWindow, // main frame window
WksTunePerfDlg, // Dialog box function
(LPARAM)&_Catalogs ); // User parameter
}
// Ensure that only and all active catalogs are listed.
_Catalogs.UpdateActiveState();
Refresh(); // Update all result pane(s)
break;
}
default:
sc = E_UNEXPECTED;
break;
}
}
CATCH( CException, e )
{
ciaDebugOut(( DEB_ERROR, "Exception 0x%x in CCISnapinData::Command\n", e.GetErrorCode() ));
sc = E_UNEXPECTED;
}
END_CATCH
UNTRANSLATE_EXCEPTIONS;
return sc;
}
//+---------------------------------------------------------------------------
//
// Member: CCISnapinData::GetClassID, public
//
// Synopsis: Identifies class of storage
//
// Arguments: [pClassID] -- Class written here.
//
// Returns: S_OK
//
// History: 14-Jul-97 KyleP Created
//
//----------------------------------------------------------------------------
SCODE CCISnapinData::GetClassID( CLSID * pClassID )
{
*pClassID = guidCISnapin;
return S_OK;
}
//+---------------------------------------------------------------------------
//
// Member: CCISnapinData::IsDirty, public
//
// Returns: TRUE if snapin data has not been saved.
//
// History: 14-Jul-97 KyleP Created
//
//----------------------------------------------------------------------------
SCODE CCISnapinData::IsDirty()
{
return (_fDirty) ? S_OK : S_FALSE;
}
//+---------------------------------------------------------------------------
//
// Member: CCISnapinData::Load, public
//
// Synopsis: Load persistent state
//
// Arguments: [pStm] -- State stored in stream.
//
// Returns: S_OK on successful load
//
// History: 14-Jul-97 KyleP Created
//
//----------------------------------------------------------------------------
SCODE CCISnapinData::Load( IStream * pStm )
{
//
// Read in machine name.
//
ULONG cbRead = 0;
ULONG cc = 0;
SCODE sc = pStm->Read( &cc, sizeof(cc), &cbRead );
if ( S_OK != sc || cbRead != sizeof(cc) )
{
ciaDebugOut(( DEB_ERROR, "Could not load catalog name: 0x%x\n", sc ));
return E_FAIL;
}
XGrowable<WCHAR> xwcsMachine( cc );
sc = pStm->Read( xwcsMachine.Get(), cc * sizeof(WCHAR), &cbRead );
if ( S_OK != sc || cbRead != cc * sizeof(WCHAR) )
{
ciaDebugOut(( DEB_ERROR, "Could not load catalog name: 0x%x\n", sc ));
return E_FAIL;
}
_Catalogs.SetMachine( xwcsMachine.Get() );
_fDirty = FALSE;
return S_OK;
}
//+---------------------------------------------------------------------------
//
// Member: CCISnapinData::Save, public
//
// Synopsis: Save persistent state
//
// Arguments: [pStm] -- Stream into which state can be stored.
// [fClearDirty] -- TRUE if dirty bit should be cleared
//
// Returns: S_OK if save succeeded.
//
// History: 14-Jul-97 KyleP Created
//
//----------------------------------------------------------------------------
SCODE CCISnapinData::Save( IStream * pStm, BOOL fClearDirty )
{
ULONG cbWritten;
ULONG cc = wcslen( _Catalogs.GetMachine() ) + 1;
SCODE sc = pStm->Write( &cc, sizeof(cc), &cbWritten );
if ( S_OK != sc || cbWritten != sizeof(cc) )
{
ciaDebugOut(( DEB_ERROR, "Could not save catalog name: 0x%x\n", sc ));
return E_FAIL;
}
sc = pStm->Write( _Catalogs.GetMachine(), cc * sizeof(WCHAR), &cbWritten );
if ( S_OK != sc || cbWritten != cc * sizeof(WCHAR) )
{
ciaDebugOut(( DEB_ERROR, "Could not save catalog name: 0x%x\n", sc ));
return E_FAIL;
}
if ( fClearDirty )
_fDirty = FALSE;
return S_OK;
}
//+---------------------------------------------------------------------------
//
// Member: CCISnapinData::GetSizeMax, public
//
// Synopsis: Computes size of persistent state
//
// Arguments: [pcbSize] -- Size returned here.
//
// Returns: S_OK
//
// History: 14-Jul-97 KyleP Created
//
//----------------------------------------------------------------------------
SCODE CCISnapinData::GetSizeMax( ULARGE_INTEGER * pcbSize )
{
pcbSize->HighPart = 0;
pcbSize->LowPart = sizeof(ULONG) +
sizeof(WCHAR) * (wcslen(_Catalogs.GetMachine()) + 1);
return S_OK;
}
//+-------------------------------------------------------------------------
//
// Method: CCISnapinData::CCISnapinData
//
// Synopsis: Constructor
//
// History: 26-Nov-1996 KyleP Created
//
//--------------------------------------------------------------------------
CCISnapinData::CCISnapinData( )
: _pFrame( 0 ),
_pScopePane( 0 ),
_pChild( 0 ),
_uRefs( 1 ),
_fDirty( TRUE ),
_fIsInitialized( FALSE ),
_fIsExtension( FALSE ),
_fTriedEnable( FALSE ),
_notifHandle( 0 ),
_fURLDeselected( FALSE )
{
InterlockedIncrement( &gulcInstances );
}
//+-------------------------------------------------------------------------
//
// Method: CCISnapinData::~CCISnapinData
//
// Synopsis: Destructor
//
// History: 26-Nov-1996 KyleP Created
//
//--------------------------------------------------------------------------
CCISnapinData::~CCISnapinData()
{
ciaDebugOut(( DEB_ITRACE, "CCISnapinData::~CCISnapinData\n" ));
Win4Assert( 0 == _pFrame );
Win4Assert( 0 == _pScopePane );
// Tell catalogs that snapindata is no longer valid!
_Catalogs.SetSnapinData(0);
for ( CCISnapin * pCurrent = _pChild;
0 != pCurrent;
pCurrent = pCurrent->Next() )
{
pCurrent->ClearSnapinData();
}
InterlockedDecrement( &gulcInstances );
}
void CCISnapinData::Refresh()
{
_Catalogs.DisplayScope( 0xFFFFFFFF );
for ( CCISnapin * pCurrent = _pChild;
0 != pCurrent;
pCurrent = pCurrent->Next() )
{
pCurrent->Refresh();
}
}
//+-------------------------------------------------------------------------
//
// Method: CCISnapin::QueryInterface
//
// Synopsis: Switch from one interface to another
//
// History: 26-Nov-1996 KyleP Created
//
//--------------------------------------------------------------------------
SCODE STDMETHODCALLTYPE CCISnapin::QueryInterface( REFIID riid,
void ** ppvObject )
{
ciaDebugOut(( DEB_ITRACE, "CCISnapin::QueryInterface\n" ));
SCODE sc = S_OK;
if ( 0 == ppvObject )
return E_INVALIDARG;
if ( IID_IComponent == riid )
*ppvObject = (IUnknown *)(IComponent *)this;
else if ( IID_IExtendPropertySheet == riid )
*ppvObject = (IUnknown *)(IExtendPropertySheet *)this;
else if ( IID_IExtendContextMenu == riid )
*ppvObject = (IUnknown *)(IExtendContextMenu *)this;
else if ( IID_IExtendControlbar == riid )
*ppvObject = (IUnknown *)(IExtendControlbar *)this;
else if ( IID_IUnknown == riid )
*ppvObject = (IUnknown *)(IComponent *)this;
else
sc = E_NOINTERFACE;
if ( SUCCEEDED( sc ) )
AddRef();
return sc;
} //QueryInterface
//+-------------------------------------------------------------------------
//
// Method: CCISnapin::AddRef
//
// Synopsis: Increment ref count
//
// History: 26-Nov-1996 KyleP Created
//
//--------------------------------------------------------------------------
ULONG STDMETHODCALLTYPE CCISnapin::AddRef()
{
ciaDebugOut(( DEB_ITRACE, "CCISnapin::AddRef\n" ));
return InterlockedIncrement( &_uRefs );
}
//+-------------------------------------------------------------------------
//
// Method: CCISnapin::Release
//
// Synopsis: Deccrement ref count
//
// History: 26-Nov-1996 KyleP Created
//
//--------------------------------------------------------------------------
ULONG STDMETHODCALLTYPE CCISnapin::Release()
{
ciaDebugOut(( DEB_ITRACE, "CCISnapin::Release\n" ));
unsigned long uTmp = InterlockedDecrement( &_uRefs );
if ( 0 == uTmp )
delete this;
return(uTmp);
}
SCODE STDMETHODCALLTYPE CCISnapin::Initialize( IConsole * lpFrame )
{
ciaDebugOut(( DEB_ITRACE, "CCISnapin::Initialize\n" ));
//
// Parameter checking.
//
if ( 0 == lpFrame )
return E_INVALIDARG;
_pFrame = lpFrame;
_pFrame->AddRef();
SCODE sc;
do
{
//
// Collect interfaces
//
sc = _pFrame->QueryInterface( IID_IHeaderCtrl, (void **)&_pHeader );
if ( FAILED(sc) )
break;
sc = _pFrame->QueryInterface( IID_IResultData, (void **)&_pResultPane );
if ( FAILED(sc) )
break;
sc = _pFrame->QueryConsoleVerb( &_pConsoleVerb );
if ( FAILED(sc) )
break;
sc = _pFrame->QueryInterface( IID_IDisplayHelp, (void **)&_pDisplayHelp );
if ( FAILED(sc) )
break;
//
// Initialize resources
//
sc = _pFrame->QueryResultImageList( &_pImageList );
if ( FAILED(sc) )
break;
sc = _pFrame->GetMainWindow(&_hFrameWindow);
if ( FAILED(sc) )
break;
//
// Initialize catalogs
//
// _SnapinData.GetCatalogs().InitHeader( _pHeader );
}
while ( FALSE );
//
// Damage control
//
if ( FAILED(sc) )
{
if ( 0 != _pImageList )
{
_pImageList->Release();
_pImageList = 0;
}
if ( 0 != _pResultPane )
{
_pResultPane->Release();
_pResultPane = 0;
}
if ( 0 != _pConsoleVerb )
{
_pConsoleVerb->Release();
_pConsoleVerb = 0;
}
if ( 0 != _pHeader )
{
_pHeader->Release();
_pHeader = 0;
}
if ( 0 != _pFrame )
{
_pFrame->Release();
_pFrame = 0;
}
}
return sc;
}
SCODE STDMETHODCALLTYPE CCISnapin::Notify( IDataObject * pDataObject,
MMC_NOTIFY_TYPE event,
LPARAM arg,
LPARAM param )
{
ciaDebugOut(( DEB_ITRACE, "CCISnapin::Notify (pDO = 0x%x, event = 0x%x)\n",
pDataObject, event ));
CCIAdminDO * pDO = (CCIAdminDO *)pDataObject;
TRANSLATE_EXCEPTIONS;
SCODE sc = S_OK;
TRY
{
switch ( event )
{
case MMCN_SHOW:
ShowItem( pDO, // Cookie
(BOOL)arg, // TRUE --> Select, FALSE --> Deselect
(HSCOPEITEM)param ); // Scope item selected
break;
case MMCN_SELECT:
if ( (CCIAdminDO *) -2 != pDO )
EnableStandardVerbs( pDO );
break;
/*
case MMCN_COLUMN_CLICK:
// No need to sort on these columns in this view
sc = _pResultPane->Sort(arg, RSI_DESCENDING, 0);
break;
*/
case MMCN_DBLCLICK:
if (pDO->Type() == CCT_SCOPE)
{
// This is an undocumented feature. Return S_FALSE and that will
// cause the appropriate node in the scope pane to automagically expand
// and select the right node. Figured this out after spending a couple of
// hours trying to find a documented way of doing this...
// Other components use this undocumented feature.
sc = S_FALSE;
}
else if (pDO->IsADirectory())
{
CScope * pScope = pDO->GetScope();
Win4Assert( 0 != pScope );
DialogBoxParam( ghInstance, // Application instance
MAKEINTRESOURCE( IDD_ADD_SCOPE ), // Dialog box
_hFrameWindow, // main window
ModifyScopeDlg, // Dialog box function
(LPARAM)pScope ); // User parameter
Refresh(); // Update all result pane(s)
}
else if (pDO->IsAProperty())
{
XPtr<CPropertyPropertySheet1> xPropSheet(
new CPropertyPropertySheet1( ghInstance,
_pSnapinData->NotifyHandle(),
pDO->GetProperty(),
pDO->GetCatalog()
));
LPCPROPSHEETPAGE psp = xPropSheet->GetPropSheet();
PROPSHEETHEADER psh;
psh.dwSize = sizeof (PROPSHEETHEADER);
psh.dwFlags = PSH_PROPSHEETPAGE;
psh.hwndParent = _hFrameWindow;
psh.hInstance = ghInstance;
psh.pszIcon = NULL;
psh.pszCaption = pDO->GetProperty()->GetPropSet();
psh.nPages = 1;
psh.ppsp = psp;
PropertySheet(&psh);
xPropSheet.Acquire();
}
break;
case MMCN_PROPERTY_CHANGE:
{
PCIObjectType * pType = (PCIObjectType *)param;
if ( pType->Type() == PCIObjectType::Property )
{
ciaDebugOut(( DEB_ITRACE, "PROPERTY CHANGE\n" ));
HRESULTITEM hItem;
sc = _pResultPane->FindItemByLParam( param, &hItem );
if ( SUCCEEDED(sc) )
{
RESULTDATAITEM rdi;
ZeroMemory(&rdi, sizeof(rdi));
rdi.mask = RDI_IMAGE;
rdi.itemID = hItem;
rdi.nImage = ICON_MODIFIED_PROPERTY;
sc = _pResultPane->SetItem( &rdi );
}
}
break;
} // case
case MMCN_DELETE:
if ( pDO->IsACatalog() )
_pSnapinData->RemoveCatalog( pDO );
else
{
Win4Assert( pDO->IsADirectory() );
RemoveScope( pDO );
}
break;
case MMCN_HELP:
{
LPOLESTR lpHelpFile;
if ( S_OK == _pSnapinData->GetHelpTopic2(&lpHelpFile) )
_pDisplayHelp->ShowTopic(lpHelpFile);
}
break;
case MMCN_CONTEXTHELP:
{
LPOLESTR lpHelpFile;
if ( S_OK == _pSnapinData->GetHelpTopic2(&lpHelpFile) )
_pDisplayHelp->ShowTopic(lpHelpFile);
}
break;
case MMCN_SNAPINHELP:
{
LPOLESTR lpHelpFile;
if ( S_OK == _pSnapinData->GetHelpTopic2(&lpHelpFile) )
_pDisplayHelp->ShowTopic(lpHelpFile);
}
break;
} // switch
}
CATCH( CException, e )
{
ciaDebugOut(( DEB_ERROR, "Exception 0x%x in CCISnapin::Notify\n", e.GetErrorCode() ));
sc = E_UNEXPECTED;
}
END_CATCH
UNTRANSLATE_EXCEPTIONS;
return sc;
}
SCODE STDMETHODCALLTYPE CCISnapin::Destroy( MMC_COOKIE cookie )
{
ciaDebugOut(( DEB_ITRACE, "CCISnapin::Destroy\n" ));
if ( 0 != _pFrame )
{
_pFrame->Release();
_pFrame = 0;
_pHeader->Release();
_pHeader = 0;
_pResultPane->Release();
_pResultPane = 0;
_pImageList->Release();
_pImageList = 0;
_pConsoleVerb->Release();
_pConsoleVerb = 0;
_pDisplayHelp->Release();
_pDisplayHelp = 0;
}
return S_OK;
}
SCODE STDMETHODCALLTYPE CCISnapin::QueryDataObject( MMC_COOKIE cookie,
DATA_OBJECT_TYPES type,
IDataObject * * ppDataObject )
{
return _pSnapinData->QueryDataObject( cookie, type, ppDataObject );
}
SCODE STDMETHODCALLTYPE CCISnapin::GetResultViewType( MMC_COOKIE cookie,
BSTR * ppViewType,
long * pViewOptions )
{
ciaDebugOut(( DEB_ITRACE, "CCISnapin::GetResultViewType (cookie = 0x%x)\n", cookie ));
CIntermediate *pIntermediate = (CIntermediate *) cookie;
if ( 0 != pIntermediate &&
PCIObjectType::Intermediate_UnfilteredURL == pIntermediate->Type() )
{
Win4Assert(pIntermediate);
// To display a URL
WCHAR wszSysPath[MAX_PATH + 1];
WCHAR wszPath[MAX_PATH + 1];
GetSystemWindowsDirectory(wszSysPath, MAX_PATH);
wcscpy(wszPath, L"file://");
wcscat(wszPath, wszSysPath);
wcscat(wszPath, L"\\Help\\ciadmin.htm#machine=");
wcscat(wszPath, _pSnapinData->GetCatalogs().GetMachine());
wcscat(wszPath, L",catalog=");
wcscat(wszPath, pIntermediate->GetCatalog().GetCat(TRUE));
MakeOLESTR(ppViewType, wszPath);
*pViewOptions = MMC_VIEW_OPTIONS_NOLISTVIEWS ;
return S_OK;
}
else
{
//
// S_FALSE --> Use listview
//
return S_FALSE;
}
}
SCODE STDMETHODCALLTYPE CCISnapin::GetDisplayInfo( RESULTDATAITEM * pResult )
{
//ciaDebugOut(( DEB_ITRACE, "CCISnapin::GetDisplayInfo (itemID = 0x%x, bScopeItem = %s, lParam = 0x%x)\n",
// pResult->itemID, pResult->bScopeItem ? "TRUE" : "FALSE", pResult->lParam ));
if ( 0 == pResult )
return E_POINTER;
TRANSLATE_EXCEPTIONS;
SCODE sc = S_OK;
TRY
{
if ( pResult->mask & RDI_STR )
{
PCIObjectType * pType = (PCIObjectType *)pResult->lParam;
switch ( pType->Type() )
{
case PCIObjectType::RootNode:
{
switch ( pResult->nCol )
{
case 0:
pResult->str = (WCHAR *)_pSnapinData->GetRootDisplay();
break;
case 1:
pResult->str = (WCHAR *)_pSnapinData->GetType();
break;
case 2:
pResult->str = (WCHAR *)_pSnapinData->GetDescription();
break;
default:
Win4Assert(!"How did we get here?");
pResult->str = L"";
break;
}
if ( pResult->mask & RDI_IMAGE && 0 == pResult->nCol )
pResult->nImage = ICON_APP;
break;
}
case PCIObjectType::Catalog:
{
CCatalog * pCat = (CCatalog *)pResult->lParam;
pCat->GetDisplayInfo( pResult );
break;
}
case PCIObjectType::Directory:
{
CScope * pScope = (CScope *)pResult->lParam;
pScope->GetDisplayInfo( pResult );
break;
}
case PCIObjectType::Property:
{
CCachedProperty * pProperty = (CCachedProperty *)pResult->lParam;
pProperty->GetDisplayInfo( pResult );
break;
}
case PCIObjectType::Intermediate_Scope:
{
if ( 0 == pResult->nCol )
pResult->str = STRINGRESOURCE( srNodeDirectories );
else
pResult->str = L"";
if ( pResult->mask & RDI_IMAGE && 0 == pResult->nCol )
pResult->nImage = ICON_FOLDER;
break;
}
case PCIObjectType::Intermediate_Properties:
{
if ( 0 == pResult->nCol )
pResult->str = STRINGRESOURCE( srNodeProperties );
else
pResult->str = L"";
if ( pResult->mask & RDI_IMAGE && 0 == pResult->nCol )
pResult->nImage = ICON_FOLDER;
break;
}
case PCIObjectType::Intermediate_UnfilteredURL:
{
if ( 0 == pResult->nCol )
pResult->str = STRINGRESOURCE( srNodeUnfiltered );
else
pResult->str = L"";
if ( pResult->mask & RDI_IMAGE && 0 == pResult->nCol )
pResult->nImage = ICON_URL;
break;
}
}
}
else
sc = E_UNEXPECTED;
}
CATCH( CException, e )
{
ciaDebugOut(( DEB_ERROR, "Exception 0x%x in CCISnapinData::GetDisplayInfo\n", e.GetErrorCode() ));
sc = E_UNEXPECTED;
}
END_CATCH
UNTRANSLATE_EXCEPTIONS;
return sc;
}
SCODE STDMETHODCALLTYPE CCISnapin::CompareObjects( IDataObject * lpDataObjectA,
IDataObject * lpDataObjectB )
{
return _pSnapinData->CompareObjects( lpDataObjectA, lpDataObjectB );
}
SCODE STDMETHODCALLTYPE CCISnapin::CreatePropertyPages( IPropertySheetCallback * lpProvider,
LONG_PTR handle,
IDataObject * lpIDataObject )
{
return _pSnapinData->CreatePropertyPages( lpProvider, handle, lpIDataObject );
}
SCODE STDMETHODCALLTYPE CCISnapinData::CreatePropertyPages( IPropertySheetCallback * lpProvider,
LONG_PTR handle,
IDataObject * lpIDataObject )
{
SCODE sc = S_OK;
_notifHandle = handle;
TRY
{
CCIAdminDO * pDO = (CCIAdminDO *)lpIDataObject;
CCatalog * pCat = pDO->GetCatalog();
CScope * pScope = pDO->GetScope();
CCachedProperty * pProperty = pDO->GetProperty();
if ( pDO->Type() == CCT_SNAPIN_MANAGER )
{
ciaDebugOut(( DEB_ITRACE, "CCISnapin::CreatePropertyPages (object is Catalogs)\n" ));
XPtr<CIndexSrvPropertySheet0> xFoo( new CIndexSrvPropertySheet0( ghInstance, handle, &_Catalogs ));
if (S_OK == lpProvider->AddPage( xFoo->GetHandle()) )
xFoo.Acquire();
}
else if ( 0 != pProperty )
{
ciaDebugOut(( DEB_ITRACE, "CCISnapin::CreatePropertyPages (object is property %ws)\n",
pProperty->GetProperty() ));
XPtr<CPropertyPropertySheet1> xFoo( new CPropertyPropertySheet1( ghInstance, handle, pProperty, pCat ));
if ( S_OK == lpProvider->AddPage( xFoo->GetHandle()) )
xFoo.Acquire();
}
else if ( 0 != pScope )
{
ciaDebugOut(( DEB_ITRACE, "CCISnapin::CreatePropertyPages (object is scope %ws)\n",
pScope->GetPath() ));
}
//
// NOTE: The following has to be last, because you can derive a pCat from the
// preceding choices.
//
else if ( 0 != pCat )
{
ciaDebugOut(( DEB_ITRACE, "CCISnapin::CreatePropertyPages (object is catalog %ws)\n",
pCat->GetCat( TRUE ) ));
XPtr<CCatalogBasicPropertySheet> xCat1( new CCatalogBasicPropertySheet( ghInstance, handle, pCat ));
if (S_OK == lpProvider->AddPage( xCat1->GetHandle()) )
xCat1.Acquire();
XPtr<CIndexSrvPropertySheet2> xCP2( new CIndexSrvPropertySheet2( ghInstance, handle, pCat ));
if ( S_OK == lpProvider->AddPage( xCP2->GetHandle()) )
xCP2.Acquire();
XPtr<CIndexSrvPropertySheet1> xIS1( new CIndexSrvPropertySheet1( ghInstance, handle, pCat ));
if (S_OK == lpProvider->AddPage( xIS1->GetHandle()) )
xIS1.Acquire();
}
else if ( pDO->IsRoot() || pDO->IsStandAloneRoot() )
{
ciaDebugOut(( DEB_ITRACE, "CCISnapin::CreatePropertyPages (object is root)\n" ));
XPtr<CIndexSrvPropertySheet1> xIS1( new CIndexSrvPropertySheet1( ghInstance, handle, &_Catalogs) );
if (S_OK == lpProvider->AddPage( xIS1->GetHandle()) )
xIS1.Acquire();
XPtr<CIndexSrvPropertySheet2> xIS2( new CIndexSrvPropertySheet2( ghInstance, handle, &_Catalogs) );
if (S_OK == lpProvider->AddPage( xIS2->GetHandle()) )
xIS2.Acquire();
}
else
{
ciaDebugOut(( DEB_ITRACE, "CCISnapin::CreatePropertyPages. Invalid call. (cookie = 0x%x, type = 0x%x)\n",
((CCIAdminDO *)lpIDataObject)->Cookie(),
((CCIAdminDO *)lpIDataObject)->Type() ));
}
}
CATCH( CException, e )
{
ciaDebugOut(( DEB_ERROR, "CCISnapin::CreatePropertyPages: Caught error 0x%x\n", e.GetErrorCode() ));
sc = GetOleError( e );
}
END_CATCH
return sc;
}
SCODE STDMETHODCALLTYPE CCISnapin::QueryPagesFor( IDataObject * lpDataObject )
{
return _pSnapinData->QueryPagesFor( lpDataObject );
}
SCODE STDMETHODCALLTYPE CCISnapin::SetControlbar( LPCONTROLBAR pControlbar)
{
// Notes: This implementation is based on the MMC
// sample Step4.
if (0 == pControlbar)
{
_xControlbar.Free();
return S_OK;
}
SCODE sc = S_OK;
// cache the incoming pointer and AddRef it
_xControlbar.Set(pControlbar);
_xControlbar->AddRef();
// If we haven't yet created a toolbar, create now
if (0 == _xToolbar.GetPointer())
{
Win4Assert(0 == _xBmpToolbar.Get());
sc = pControlbar->Create(TOOLBAR, this, _xToolbar.GetIUPointer());
if (SUCCEEDED(sc))
{
_xBmpToolbar.Set(LoadBitmap(ghInstance, MAKEINTRESOURCE(BMP_TOOLBAR_SMALL)));
sc = (_xBmpToolbar.Get() ? S_OK : E_FAIL);
}
if (SUCCEEDED(sc))
sc = _xToolbar->AddBitmap(sizeof aContextButtons / sizeof aContextButtons[0],
_xBmpToolbar.Get(), 16, 16, RGB(255, 0, 255));
if (SUCCEEDED(sc))
sc = _xToolbar->AddButtons(sizeof aContextButtons / sizeof aContextButtons[0],
aContextButtons);
}
return sc;
}
SCODE STDMETHODCALLTYPE CCISnapin::ControlbarNotify( MMC_NOTIFY_TYPE event,
LPARAM arg,
LPARAM param)
{
Win4Assert(_xControlbar.GetPointer() &&
_xToolbar.GetPointer() &&
_xBmpToolbar.Get());
if (MMCN_SELECT == event && !((BOOL)LOWORD(arg)) && _pSnapinData->IsURLDeselected() )
return E_POINTER;
SCODE sc = S_OK;
BOOL fPaused = FALSE;
BOOL fStopped = FALSE;
BOOL fStarted = FALSE;
TRY
{
if (MMCN_SELECT == event)
{
LPDATAOBJECT pDataObject = (LPDATAOBJECT) param;
if( NULL == pDataObject )
return S_FALSE;
// Completely random MMC behavior (apparently).
if ( -2 == param )
return S_FALSE;
CCIAdminDO * pDO = (CCIAdminDO *)pDataObject;
BOOL bScope = (BOOL) LOWORD(arg);
BOOL bSelect = (BOOL) HIWORD(arg);
ciaDebugOut((DEB_ITRACE,
"select event: scope: %d, selection %d, lparam: 0x%x\n",
bScope, bSelect, param));
CMachineAdmin MachineAdmin( _pSnapinData->GetCatalogs().IsLocalMachine() ?
0 : _pSnapinData->GetCatalogs().GetMachine() );
if (bScope) // scope item selected
{
if (pDO->IsRoot() || pDO->IsStandAloneRoot() )
{
sc = _xControlbar->Attach(TOOLBAR, _xToolbar.GetPointer());
if ( SUCCEEDED(sc) )
{
if ( MachineAdmin.IsCIStarted() )
fStarted = TRUE;
else if ( MachineAdmin.IsCIPaused() )
fPaused = TRUE;
else
{
Win4Assert( MachineAdmin.IsCIStopped() );
fStopped = TRUE;
}
}
}
else
{
sc = _xControlbar->Detach(_xToolbar.GetPointer());
// If the URL on scope pane is deselected, remember that
_pSnapinData->SetURLDeselected( pDO->IsURLIntermediate() && bSelect == FALSE );
}
}
// result list item selected
else
{
if ( pDO->IsACatalog() )
{
XPtr<CCatalogAdmin> xCat( MachineAdmin.QueryCatalogAdmin( pDO->GetCatalog()->GetCat(TRUE) ) );
if ( xCat->IsStarted() )
fStarted = TRUE;
else if ( xCat->IsPaused() )
fPaused = TRUE;
else
{
Win4Assert( xCat->IsStopped() );
fStopped = TRUE;
}
}
else if ( pDO->IsRoot() || pDO->IsStandAloneRoot() )
{
sc = _xControlbar->Attach(TOOLBAR, _xToolbar.GetPointer());
if ( SUCCEEDED(sc) )
{
if ( MachineAdmin.IsCIStarted() )
fStarted = TRUE;
else if ( MachineAdmin.IsCIPaused() )
fPaused = TRUE;
else
{
Win4Assert( MachineAdmin.IsCIStopped() );
fStopped = TRUE;
}
}
}
}
Win4Assert( _xToolbar.GetPointer() );
if (fStarted)
{
_xToolbar->SetButtonState(comidStartCITop, ENABLED, FALSE);
_xToolbar->SetButtonState(comidStopCITop, ENABLED, TRUE);
_xToolbar->SetButtonState(comidPauseCITop, ENABLED, TRUE);
}
else if (fStopped)
{
_xToolbar->SetButtonState(comidStartCITop, ENABLED, TRUE);
_xToolbar->SetButtonState(comidStopCITop, ENABLED, FALSE);
_xToolbar->SetButtonState(comidPauseCITop, ENABLED, FALSE);
}
else if (fPaused)
{
_xToolbar->SetButtonState(comidStartCITop, ENABLED, TRUE);
_xToolbar->SetButtonState(comidStopCITop, ENABLED, TRUE);
_xToolbar->SetButtonState(comidPauseCITop, ENABLED, FALSE);
}
}
else if (MMCN_BTN_CLICK == event)
{
Win4Assert( comidStartCITop == param ||
comidStopCITop == param ||
comidPauseCITop == param );
LPDATAOBJECT pDataObject = (LPDATAOBJECT) arg;
CCIAdminDO TempDO( 0, CCT_SCOPE, 0 );
if( NULL == pDataObject )
pDataObject = (LPDATAOBJECT)&TempDO;
return Command( (long)param, pDataObject );
}
}
CATCH( CException, e )
{
sc = E_FAIL;
}
END_CATCH
return sc;
}
SCODE STDMETHODCALLTYPE CCISnapinData::QueryPagesFor( IDataObject * lpDataObject )
{
if ( 0 == lpDataObject || _fURLDeselected )
return E_POINTER;
SCODE sc = S_OK;
TRY
{
// NOTE: Attempt to open the service and if that fails, we'll be unable to
// add property pages. That's expected behavior
XPtr<CMachineAdmin> xMachineAdmin(new CMachineAdmin( _Catalogs.GetMachine() ));
CCIAdminDO * pDO = (CCIAdminDO *)lpDataObject;
ciaDebugOut(( DEB_ITRACE, "CCISnapin::QueryPagesFor (cookie = 0x%x, type = 0x%x)\n",
pDO->Cookie(), pDO->Type() ));
if ( pDO->Type() == CCT_SNAPIN_MANAGER )
sc = S_OK;
if ( pDO->IsRoot() || pDO->IsStandAloneRoot() || pDO->IsACatalog() || pDO->IsAProperty())
{
sc = S_OK;
}
else
{
ciaDebugOut(( DEB_WARN, "No property pages for (cookie = 0x%x, type = 0x%x)\n",
pDO->Cookie(), pDO->Type() ));
sc = S_FALSE;
}
}
CATCH( CException, e )
{
ciaDebugOut(( DEB_ERROR, "CCISnapin::QueryPagesFor: Caught error 0x%x\n", e.GetErrorCode() ));
sc = GetOleError( e );
}
END_CATCH
return sc;
}
SCODE STDMETHODCALLTYPE CCISnapin::AddMenuItems( IDataObject * piDataObject,
IContextMenuCallback * piCallback,
long * pInsertionAllowed )
{
EnableStandardVerbs( piDataObject );
return _pSnapinData->AddMenuItems( piDataObject, piCallback, pInsertionAllowed );
}
SCODE STDMETHODCALLTYPE CCISnapin::Command( long lCommandID,
IDataObject * piDataObject )
{
return _pSnapinData->Command( lCommandID, piDataObject );
}
//
// ISnapinAbout methods
//
//+-------------------------------------------------------------------------
//
// Method: CCISnapinData::GetSnapinDescription
//
// Synopsis: Get description about indexing service.
//
// History: 02-Feb-1998 KrishanN Added Header
//
//--------------------------------------------------------------------------
SCODE STDMETHODCALLTYPE CCISnapinData::GetSnapinDescription(LPOLESTR *lpDescription)
{
WCHAR szStr[1024];
wsprintf(szStr, L"%s\r\n%s", STRINGRESOURCE( srProductDescription ),
STRINGRESOURCE( srVendorCopyright ));
return MakeOLESTR(lpDescription, szStr);
}
//+-------------------------------------------------------------------------
//
// Method: CCISnapinData::GetProvider
//
// Synopsis: Get provider of index server.
//
// History: 02-Feb-1998 KrishanN Added Header
//
//--------------------------------------------------------------------------
SCODE STDMETHODCALLTYPE CCISnapinData::GetProvider(LPOLESTR *lpName)
{
WCHAR szStr[1024];
wsprintf(szStr, L"%S, %s", VER_COMPANYNAME_STR, STRINGRESOURCE( srVendorName ));
return MakeOLESTR(lpName, szStr);
}
//+-------------------------------------------------------------------------
//
// Method: CCISnapinData::GetSnapinVersion
//
// Synopsis: Get version of index server.
//
// History: 02-Feb-1998 KrishanN Added Header
//
//--------------------------------------------------------------------------
SCODE STDMETHODCALLTYPE CCISnapinData::GetSnapinVersion(LPOLESTR *lpVersion)
{
WCHAR szStr[1024];
wsprintf(szStr, L"%S", VER_PRODUCTVERSION_STR);
return MakeOLESTR(lpVersion, szStr);
}
//+-------------------------------------------------------------------------
//
// Method: CCISnapinData::GetSnapinImage
//
// Synopsis: Get image of index server.
//
// History: 02-Feb-1998 KrishanN Added Header
//
//--------------------------------------------------------------------------
SCODE STDMETHODCALLTYPE CCISnapinData::GetSnapinImage(HICON *phAppIcon)
{
*phAppIcon = LoadIcon(ghInstance, MAKEINTRESOURCE(ICON_ABOUT));
return (NULL == *phAppIcon) ? E_FAIL : S_OK;
}
//+-------------------------------------------------------------------------
//
// Method: CCISnapinData::GetStaticFolderImage
//
// Synopsis: Get static folder images for use with Index Server.
//
// History: 02-Feb-1998 KrishanN Added Header
//
//--------------------------------------------------------------------------
SCODE STDMETHODCALLTYPE CCISnapinData::GetStaticFolderImage(
HBITMAP *hSmallImage,
HBITMAP *hSmallImageOpen,
HBITMAP *hLargeImage,
COLORREF *cMask)
{
*hSmallImage = *hSmallImageOpen = *hLargeImage = NULL;
*hSmallImage = LoadBitmap(ghInstance, MAKEINTRESOURCE(BMP_SMALL_CLOSED_FOLDER));
*hSmallImageOpen = LoadBitmap(ghInstance, MAKEINTRESOURCE(BMP_SMALL_OPEN_FOLDER));
*hLargeImage = LoadBitmap(ghInstance, MAKEINTRESOURCE(BMP_LARGE_CLOSED_FOLDER));
*cMask = RGB(255, 0, 255);
if (NULL == *hSmallImage || NULL == *hSmallImageOpen || NULL == *hLargeImage)
{
if (*hSmallImage)
DeleteObject(*hSmallImage);
if (*hSmallImageOpen)
DeleteObject(*hSmallImageOpen);
if (*hLargeImage)
DeleteObject(*hLargeImage);
return E_FAIL;
}
return S_OK;
}
//+-------------------------------------------------------------------------
//
// Method: CCISnapin::CCISnapin
//
// Synopsis: Constructor
//
// History: 26-Nov-1996 KyleP Created
//
//--------------------------------------------------------------------------
CCISnapin::CCISnapin( CCISnapinData & SnapinData, CCISnapin * pNext )
: _pFrame( 0 ),
_pHeader( 0 ),
_pResultPane( 0 ),
_pImageList( 0 ),
_pConsoleVerb( 0 ),
_pDisplayHelp( 0 ),
_CurrentView( CCISnapin::Nothing ),
_pSnapinData( & SnapinData ),
_pNext( pNext ),
_uRefs( 1 )
{
}
//+-------------------------------------------------------------------------
//
// Method: CCISnapin::~CCISnapin
//
// Synopsis: Destructor
//
// History: 26-Nov-1996 KyleP Created
//
//--------------------------------------------------------------------------
CCISnapin::~CCISnapin()
{
ciaDebugOut(( DEB_ITRACE, "CCISnapin::~CCISnapin\n" ));
if ( 0 != _pSnapinData )
{
if ( _pSnapinData->_pChild == this )
{
_pSnapinData->_pChild = _pNext;
}
else
{
for ( CCISnapin * pCurrent = _pSnapinData->_pChild;
0 != pCurrent && 0 != pCurrent->Next();
pCurrent = pCurrent->Next() )
{
if ( pCurrent->Next() == this )
{
pCurrent->Link( pCurrent->Next()->Next() );
break;
}
}
}
}
Win4Assert( 0 == _pFrame );
Win4Assert( 0 == _pHeader );
Win4Assert( 0 == _pResultPane );
Win4Assert( 0 == _pImageList );
Win4Assert( 0 == _pConsoleVerb );
}
//+-------------------------------------------------------------------------
//
// Function: GetMachineName
//
// Synopsis: gets machine name to administer.
//
// Arguments: [pDO] -- dataobject pointer
// [pwszMachineName] -- out buffer
// [ cc ] -- buffer size in wchars
//
// Returns: none. throws upon fatal errors (out of memory).
//
// History: 01-Jul-1998 mohamedn created
// 31-Aug-1998 KyleP Support DNS names
//
//--------------------------------------------------------------------------
void GetMachineName(LPDATAOBJECT pDO, XGrowable<WCHAR,MAX_COMPUTERNAME_LENGTH> & xwszMachineName)
{
Win4Assert( pDO );
STGMEDIUM stgmedium = { TYMED_HGLOBAL, NULL };
FORMATETC formatetc = { (CLIPFORMAT)CCIAdminDO::GetMachineNameCF(), NULL,
DVASPECT_CONTENT, -1, TYMED_HGLOBAL
};
// Allocate memory for the stream
//
// What if the computer name is > 512 bytes long?
//
stgmedium.hGlobal = GlobalAlloc(GMEM_SHARE, 512);
XGlobalAllocMem xhGlobal(stgmedium.hGlobal);
WCHAR * pwszTmpBuf = NULL;
// Attempt to get data from the object
do
{
if (stgmedium.hGlobal == NULL)
break;
if (FAILED(pDO->GetDataHere(&formatetc, &stgmedium)))
break;
pwszTmpBuf = (WCHAR *)stgmedium.hGlobal;
if ( pwszTmpBuf == NULL || *pwszTmpBuf == L'' )
{
xwszMachineName[0] = L'.';
xwszMachineName[1] = 0;
}
else
{
unsigned cc = wcslen(pwszTmpBuf) + 1;
xwszMachineName.SetSize( cc );
RtlCopyMemory( xwszMachineName.Get(), pwszTmpBuf, cc * sizeof(WCHAR) );
}
} while (FALSE);
}
//+-------------------------------------------------------------------------
//
// Method: CCISnapin::ShowFolder, private
//
// Synopsis: Called when folder is selected. Displays result in
// scope pane.
//
// History: 27-Nov-1996 KyleP Created
//
//--------------------------------------------------------------------------
void CCISnapinData::ShowFolder( CCIAdminDO * pDO, BOOL fExpanded, HSCOPEITEM hScopeItem )
{
ciaDebugOut(( DEB_ITRACE, "CCISnapinData::ShowFolder (fExpanded = %d, hScopeItem = 0x%x)\n",
fExpanded, hScopeItem ));
Win4Assert( pDO );
//
// Only do something on expansion
//
if ( fExpanded )
{
if ( !_fIsInitialized && pDO->IsStandAloneRoot() )
{
_fIsExtension = FALSE;
}
else if ( !_fIsInitialized )
{
_fIsExtension = TRUE;
}
//
// we're stand alone.
//
if ( !_fIsExtension )
{
_fIsInitialized = TRUE;
if ( pDO->IsStandAloneRoot() )
{
_Catalogs.DisplayScope( hScopeItem );
}
else if ( pDO->IsACatalog() )
{
pDO->GetCatalog()->DisplayIntermediate( _pScopePane );
}
}
else if ( _fIsExtension && !_fIsInitialized )
{
//
// we're an extension
//
XGrowable<WCHAR,MAX_COMPUTERNAME_LENGTH> xwcsMachine;
GetMachineName( pDO, xwcsMachine );
_Catalogs.SetMachine( xwcsMachine.Get() );
SetRootDisplay();
_rootNode.Display(hScopeItem);
_fIsInitialized = TRUE;
}
else if ( _fIsExtension && pDO->IsRoot() )
{
Win4Assert( _fIsInitialized );
_Catalogs.DisplayScope( _rootNode.GethScopeItem() );
}
else if ( _fIsExtension && pDO->IsACatalog() )
{
Win4Assert( _fIsInitialized );
pDO->GetCatalog()->DisplayIntermediate( _pScopePane );
}
}
}
//+-------------------------------------------------------------------------
//
// Method: CCISnapin::SetRootDisplay
//
// Synopsis: sets display name string, type string, and description string
// for static root node
//
// History: 7/1/98 mohamedn created
// 9/29/98 KrishnaN Added type and description.
//
//--------------------------------------------------------------------------
void CCISnapinData::SetRootDisplay()
{
if ( _fIsExtension )
{
unsigned cc = wcslen( STRINGRESOURCE(srIndexServerCmpManage) + 1 );
_xwcsTitle.SetSize(cc);
wcscpy( _xwcsTitle.Get(), STRINGRESOURCE(srIndexServerCmpManage) );
cc = wcslen( STRINGRESOURCE(srType) + 1);
_xwcsType.SetSize(cc);
wcscpy( _xwcsType.Get(), STRINGRESOURCE(srType) );
cc = wcslen( STRINGRESOURCE(srProductDescription) + 1);
_xwcsDescription.SetSize(cc);
wcscpy( _xwcsDescription.Get(), STRINGRESOURCE(srProductDescription) );
}
else
{
unsigned cc = wcslen( STRINGRESOURCE(srIndexServer) ) + 1;
_xwcsTitle.SetSize(cc);
wcscpy( _xwcsTitle.Get(), STRINGRESOURCE(srIndexServer) );
if ( _Catalogs.IsLocalMachine() )
{
cc += wcslen( STRINGRESOURCE(srLM) );
_xwcsTitle.SetSize( cc );
wcscat( _xwcsTitle.Get(), STRINGRESOURCE(srLM) );
}
else
{
cc += wcslen( _Catalogs.GetMachine() );
cc += 2; // the UNC slashes
_xwcsTitle.SetSize( cc );
wcscat( _xwcsTitle.Get(), L"\\\\" );
wcscat( _xwcsTitle.Get(), _Catalogs.GetMachine() );
}
_xwcsType.Free();
_xwcsDescription.Free();
}
}
void CCISnapin::EnableStandardVerbs( IDataObject * piDataObject )
{
SCODE sc = QueryPagesFor( piDataObject );
if ( S_OK == sc )
{
_pConsoleVerb->SetVerbState(MMC_VERB_PROPERTIES, ENABLED, TRUE);
}
else
_pConsoleVerb->SetVerbState(MMC_VERB_PROPERTIES, HIDDEN, TRUE);
if (!_pSnapinData->IsURLDeselected() )
{
CCIAdminDO * pDO = (CCIAdminDO *)piDataObject;
if ( ( pDO->IsADirectory() && !pDO->GetScope()->IsVirtual() && !pDO->GetScope()->IsShadowAlias() ) ||
pDO->IsACatalog() )
{
_pConsoleVerb->SetVerbState(MMC_VERB_DELETE, ENABLED, TRUE);
}
else
{
_pConsoleVerb->SetVerbState(MMC_VERB_DELETE, HIDDEN, TRUE);
}
}
}
//+-------------------------------------------------------------------------
//
// Method: CCISnapin::ShowItem, private
//
// Synopsis: Called when folder is selected. Displays result in
// result pane.
//
// History: 27-Nov-1996 KyleP Created
//
//--------------------------------------------------------------------------
void CCISnapin::ShowItem( CCIAdminDO * pDO, BOOL fExpanded, HSCOPEITEM hScopeItem )
{
ciaDebugOut(( DEB_ITRACE, "CCISnapin::ShowItem (fExpanded = %d, hScopeItem = 0x%x)\n",
fExpanded, hScopeItem ));
if ( fExpanded )
{
SCODE sc = InitImageResources( _pImageList );
if ( FAILED(sc) )
{
ciaDebugOut(( DEB_ERROR, "InitImageResources returned 0x%x\n", sc ));
THROW( CException( sc ) );
}
if ( pDO->IsRoot() || pDO->IsStandAloneRoot() )
{
if ( !_CatalogsHeader.IsInitialized() )
{
_pSnapinData->GetCatalogs().InitHeader( _CatalogsHeader );
//
// Start background polling.
//
_pSnapinData->GetCatalogs().Display( TRUE );
}
_CatalogsHeader.Display( _pHeader );
_CurrentView = CCISnapin::Catalogs;
}
else if ( pDO->IsADirectoryIntermediate() )
{
CCatalog * pCat = pDO->GetCatalog();
if ( !_CatalogScopeHeader.IsInitialized() )
pCat->InitScopeHeader( _CatalogScopeHeader );
_CatalogScopeHeader.Display( _pHeader );
pCat->DisplayScopes( TRUE, _pResultPane );
_CurrentView = CCISnapin::Scopes;
_CurrentCatalog = pCat;
}
else if ( pDO->IsAPropertyIntermediate() )
{
CCatalog * pCat = pDO->GetCatalog();
if ( !_CatalogPropertyHeader.IsInitialized() )
pCat->InitPropertyHeader( _CatalogPropertyHeader );
_CatalogPropertyHeader.Display( _pHeader );
pCat->DisplayProperties( TRUE, _pResultPane );
_CurrentView = CCISnapin::Properties;
_CurrentCatalog = pCat;
}
else
_CurrentView = CCISnapin::Nothing;
}
else
{
switch ( _CurrentView )
{
case CCISnapin::Catalogs:
Win4Assert( pDO->IsRoot() || pDO->IsStandAloneRoot() );
_CatalogsHeader.Update( _pHeader );
break;
case CCISnapin::Scopes:
Win4Assert( pDO->IsADirectoryIntermediate() );
_CatalogScopeHeader.Update( _pHeader );
break;
case CCISnapin::Properties:
Win4Assert( pDO->IsAPropertyIntermediate() );
_CatalogPropertyHeader.Update( _pHeader );
break;
}
_CurrentView = CCISnapin::Nothing;
}
}
void CCISnapin::Refresh()
{
switch ( _CurrentView )
{
case CCISnapin::Scopes:
_CurrentCatalog->DisplayScopes( FALSE, _pResultPane );
break;
case CCISnapin::Properties:
_CurrentCatalog->DisplayProperties( FALSE, _pResultPane );
break;
case CCISnapin::Catalogs:
case CCISnapin::Nothing:
default:
break;
}
}
void CCISnapin::RemoveScope( CCIAdminDO * pDO )
{
Win4Assert( pDO->IsADirectory() && !pDO->GetScope()->IsVirtual() );
CScope * pScope = pDO->GetScope();
Win4Assert( 0 != pScope );
//
// Make sure the user wants to remove scope.
//
WCHAR awcMsg[MAX_PATH];
WCHAR awcTemp[2 * MAX_PATH];
LoadString( ghInstance, MSG_REMOVE_SCOPE, awcMsg, sizeof(awcMsg) / sizeof(WCHAR) );
wsprintf( awcTemp, awcMsg, pScope->GetPath() );
LoadString( ghInstance, MSG_REMOVE_SCOPE_TITLE, awcMsg, sizeof(awcMsg) / sizeof(WCHAR) );
int iResult;
SCODE sc = _pFrame->MessageBox( awcTemp,
awcMsg,
MB_YESNO | /* MB_HELP | */
MB_ICONWARNING | MB_DEFBUTTON2 |
MB_APPLMODAL,
&iResult );
if ( SUCCEEDED(sc) )
{
switch ( iResult )
{
case IDYES:
{
CCatalog & cat = pScope->GetCatalog();
cat.RemoveScope( pScope );
Refresh(); // Update all result pane(s)
break;
}
case IDNO:
// Do nothing.
break;
/* Help is not being used...
case IDHELP:
{
// NTRAID#DB-NTBUG9-83341-2000/07/31-dlee Need online help for several Indexing Service admin dialogs
DisplayHelp( _hFrameWindow, HIDD_REMOVE_SCOPE );
break;
}
*/
default:
break;
}
}
}
void CCISnapinData::RemoveCatalog( CCIAdminDO * pDO )
{
CCatalog * pCat = pDO->GetCatalog();
Win4Assert( 0 != pCat );
//
// Make sure we can perform the operation right now.
//
CMachineAdmin MachineAdmin( _Catalogs.IsLocalMachine() ? 0 : _Catalogs.GetMachine() );
if ( MachineAdmin.IsCIStarted() )
{
int iResult;
SCODE sc = _pFrame->MessageBox( STRINGRESOURCE( srMsgCantDeleteCatalog ),
STRINGRESOURCE( srMsgDeleteCatalogTitle ),
MB_OK | /* MB_HELP | */
MB_ICONWARNING | MB_APPLMODAL,
&iResult );
if ( SUCCEEDED(sc) )
{
switch ( iResult )
{
case IDOK:
case IDCANCEL:
// Do nothing.
break;
/* Help is not being used
case IDHELP:
{
DisplayHelp( _hFrameWindow, HIDD_REMOVE_CATALOG );
break;
}
*/
default:
break;
}
}
}
else
{
int iResult;
SCODE sc = _pFrame->MessageBox( STRINGRESOURCE( srMsgDeleteCatalog ),
STRINGRESOURCE( srMsgDeleteCatalogAsk ),
MB_YESNO | /* MB_HELP | */
MB_ICONWARNING | MB_DEFBUTTON2 |
MB_APPLMODAL,
&iResult );
if ( SUCCEEDED(sc) )
{
switch ( iResult )
{
case IDYES:
{
if ( FAILED(_Catalogs.RemoveCatalog( pCat )) )
_pFrame->MessageBox( STRINGRESOURCE( srMsgCatalogPartialDeletion ),
STRINGRESOURCE( srMsgDeleteCatalogTitle ),
MB_OK | /* MB_HELP | */
MB_ICONWARNING | MB_APPLMODAL,
&iResult);
Refresh(); // Update all result pane(s)
break;
}
case IDNO:
case IDCANCEL:
// Do nothing.
break;
/* Help is not being used...
case IDHELP:
{
DisplayHelp( _hFrameWindow, HIDD_REMOVE_CATALOG );
break;
}
*/
default:
break;
}
}
}
}
SCODE STDMETHODCALLTYPE CCISnapinData::GetHelpTopic( LPOLESTR *lpCompiledHelpFile)
{
if (0 == lpCompiledHelpFile)
return E_POINTER;
WCHAR awc[ MAX_PATH + 1];
const UINT cwcMax = sizeof awc / sizeof WCHAR;
UINT cwc = GetSystemWindowsDirectory( awc, cwcMax );
if ( 0 == cwc || cwc > cwcMax )
return E_UNEXPECTED;
if ( L'\\' != awc[ cwc-1] )
wcscat( awc, L"\\" );
// ixhelp.hlp is placed in %windir%\help
wcscat( awc, L"help\\\\is.chm" );
return MakeOLESTR(lpCompiledHelpFile, awc);
}
SCODE STDMETHODCALLTYPE CCISnapinData::GetLinkedTopics( LPOLESTR *lpCompiledHelpFiles)
{
if (0 == lpCompiledHelpFiles)
return E_POINTER;
WCHAR awc[ MAX_PATH + 1];
const UINT cwcMax = sizeof awc / sizeof WCHAR;
UINT cwc = GetSystemWindowsDirectory( awc, cwcMax );
if ( 0 == cwc || cwc > cwcMax )
return E_UNEXPECTED;
if ( L'\\' != awc[ cwc-1] )
wcscat( awc, L"\\" );
wcscat( awc, L"help\\\\isconcepts.chm" );
return MakeOLESTR(lpCompiledHelpFiles, awc);
}
SCODE STDMETHODCALLTYPE CCISnapinData::GetHelpTopic2( LPOLESTR *lpCompiledHelpFile)
{
if (0 == lpCompiledHelpFile)
return E_POINTER;
WCHAR awc[ MAX_PATH + 1];
const UINT cwcMax = sizeof awc / sizeof WCHAR;
UINT cwc = GetSystemWindowsDirectory( awc, cwcMax );
if ( 0 == cwc || cwc > cwcMax )
return E_UNEXPECTED;
if ( L'\\' != awc[ cwc-1] )
wcscat( awc, L"\\" );
wcscat( awc, L"help\\\\isconcepts.chm::/sag_INDEXtopnode.htm" );
return MakeOLESTR(lpCompiledHelpFile, awc);
}
SCODE InitImageResources( IImageList * pImageList )
{
HBITMAP hbmpSmall = LoadBitmap( ghInstance, MAKEINTRESOURCE( BMP_SMALL_ICONS ) );
if ( 0 == hbmpSmall )
return E_FAIL;
HBITMAP hbmpLarge = LoadBitmap( ghInstance, MAKEINTRESOURCE( BMP_LARGE_ICONS ) );
if ( 0 == hbmpLarge )
{
DeleteObject( hbmpSmall );
return E_FAIL;
}
SCODE sc = pImageList->ImageListSetStrip( (LONG_PTR *)hbmpSmall,
(LONG_PTR *)hbmpLarge,
0,
RGB( 255, 0, 255 ) );
DeleteObject( hbmpSmall );
DeleteObject( hbmpLarge );
return sc;
}
//+-------------------------------------------------------------------------
//
// Function: DisplayHelp
//
// Synopsis: Displays context sensitive help
//
// Arguments: [hwnd] -- The parent window handle
// [dwID] -- The help context identifier
//
// Returns: TRUE if successful
//
// History: 20-Sep-1997 dlee Created
// 14-Sep-1998 KrishnaN Handle only context sensitive help.
// Regular help is handled by MMC help.
// This fixes 214619.
//
//--------------------------------------------------------------------------
BOOL DisplayHelp( HWND hwnd, DWORD dwID )
{
if (0 != dwID) // Display context-sensitive help
return DisplayHelp( hwnd, dwID, HELP_CONTEXT );
else
return TRUE; // Don't display regular help
} //DisplayHelp
//+-------------------------------------------------------------------------
//
// Function: DisplayHelp
//
// Synopsis: Displays context sensitive help
//
// Arguments: [hwnd] -- The parent window handle
// [dwID] -- The help context identifier
//
// Returns: TRUE if successful
//
// History: 20-Sep-1997 dlee Created
//
//--------------------------------------------------------------------------
BOOL DisplayHelp( HWND hwnd, DWORD dwID, UINT uCommand )
{
WCHAR awc[ MAX_PATH ];
const UINT cwcMax = sizeof awc / sizeof WCHAR;
UINT cwc = GetSystemWindowsDirectory( awc, cwcMax );
if ( 0 == cwc || cwc > cwcMax )
return FALSE;
if ( L'\\' != awc[ cwc-1] )
wcscat( awc, L"\\" );
// ixhelp.hlp is placed in %windir%\help
wcscat( awc, L"help\\\\ixhelp.hlp" );
return WinHelp( hwnd, awc, uCommand, dwID );
} //DisplayHelp
//+-------------------------------------------------------------------------
//
// Function: DisplayPopupHelp
//
// Synopsis: Displays context sensitive help as a popup
//
// Arguments: [hwnd] -- The parent window handle
// [dwID] -- The help context identifier
//
// Returns: TRUE if successful
//
// History: 11-May-1998 KrishnaN Created
//
//--------------------------------------------------------------------------
BOOL DisplayPopupHelp( HWND hwnd, DWORD dwHelpType )
{
return DisplayHelp( hwnd, (DWORD) (DWORD_PTR) aIds, dwHelpType);
} //DisplayPopupHelp
SCODE MakeOLESTR(LPOLESTR *lpBuffer, WCHAR const * pwszText)
{
if (0 == lpBuffer)
return E_INVALIDARG;
ULONG uLen = wcslen(pwszText);
uLen++; // string terminator
uLen = sizeof(WCHAR) * uLen;
*lpBuffer = (LPOLESTR)CoTaskMemAlloc(uLen);
if (*lpBuffer)
{
RtlCopyMemory(*lpBuffer, pwszText, uLen);
return S_OK;
}
else
return E_OUTOFMEMORY;
}
//+-------------------------------------------------------------------------
//
// Method: CCISnapinData::MaybeEnableCI, private
//
// Synopsis: Prompt user to set service to automatic start
//
// Arguments: [MachineAdmin] -- Machine administration object
//
// History: 07-Jul-1998 KyleP Created
//
//--------------------------------------------------------------------------
void CCISnapinData::MaybeEnableCI( CMachineAdmin & MachineAdmin )
{
if ( _Catalogs.IsLocalMachine() && !_fTriedEnable )
{
//
// Have we tried before?
//
if ( MachineAdmin.IsCIEnabled() )
{
_fTriedEnable = TRUE;
}
else
{
WCHAR wcTemp[ (sizeof(wszSnapinPath) + sizeof(wszCISnapin)) / sizeof(WCHAR) + 1];
wcscpy( wcTemp, wszSnapinPath );
wcscat( wcTemp, wszCISnapin );
CWin32RegAccess reg( HKEY_LOCAL_MACHINE, wcTemp );
BOOL fTry = FALSE;
if ( reg.Ok() )
{
DWORD dwVal;
BOOL fOk = reg.Get( wszTriedEnable, dwVal );
if ( fOk )
{
fTry = (0 == dwVal);
}
else
{
if ( reg.GetLastError() == ERROR_FILE_NOT_FOUND )
fTry = TRUE;
}
}
if ( fTry )
{
int iResult;
SCODE sc = _pFrame->MessageBox( STRINGRESOURCE( srMsgEnableCI ),
STRINGRESOURCE( srMsgEnableCITitle ),
MB_YESNO | /* MB_HELP | */
MB_ICONQUESTION | MB_DEFBUTTON1 |
MB_APPLMODAL,
&iResult );
if ( SUCCEEDED(sc) )
{
switch ( iResult )
{
case IDYES:
MachineAdmin.EnableCI();
break;
default:
break;
}
Win4Assert( reg.Ok() );
reg.Set( wszTriedEnable, 1 );
_fTriedEnable = TRUE;
}
}
else
_fTriedEnable = TRUE;
}
}
}
void CCISnapinData::SetButtonState( int idCommand, MMC_BUTTON_STATE nState, BOOL bState )
{
for ( CCISnapin * pCurrent = _pChild;
0 != pCurrent;
pCurrent = pCurrent->Next() )
{
if ( !pCurrent->_xToolbar.IsNull() )
pCurrent->_xToolbar->SetButtonState( idCommand, nState, bState );
}
}
//+-------------------------------------------------------------------------
//
// Function: StartMenu
//
// Synopsis: Sets context menu for service started state.
//
// Arguments: [piCallback] -- Context menu callback routine
// [fTop] -- TRUE if menu items should be on top of list
// in addition to under Task section.
//
// History: 07-Jul-1998 KyleP Created
//
//--------------------------------------------------------------------------
SCODE StartMenu( IContextMenuCallback * piCallback, BOOL fTop )
{
aContextMenus[comidStartCI].fFlags =
aContextMenus[comidStartCITop].fFlags = MF_GRAYED;
aContextMenus[comidStopCI].fFlags =
aContextMenus[comidStopCITop].fFlags = MF_ENABLED;
aContextMenus[comidPauseCI].fFlags =
aContextMenus[comidPauseCITop].fFlags = MF_ENABLED;
return SetStartStopMenu( piCallback, fTop );
}
//+-------------------------------------------------------------------------
//
// Function: StopMenu
//
// Synopsis: Sets context menu for service stopped state.
//
// Arguments: [piCallback] -- Context menu callback routine
// [fTop] -- TRUE if menu items should be on top of list
// in addition to under Task section.
//
// History: 07-Jul-1998 KyleP Created
//
//--------------------------------------------------------------------------
SCODE StopMenu( IContextMenuCallback * piCallback, BOOL fTop )
{
aContextMenus[comidStartCI].fFlags =
aContextMenus[comidStartCITop].fFlags = MF_ENABLED;
aContextMenus[comidStopCI].fFlags =
aContextMenus[comidStopCITop].fFlags = MF_GRAYED;
aContextMenus[comidPauseCI].fFlags =
aContextMenus[comidPauseCITop].fFlags = MF_GRAYED;
return SetStartStopMenu( piCallback, fTop );
}
//+-------------------------------------------------------------------------
//
// Function: PauseMenu
//
// Synopsis: Sets context menu for service paused state.
//
// Arguments: [piCallback] -- Context menu callback routine
// [fTop] -- TRUE if menu items should be on top of list
// in addition to under Task section.
//
// History: 07-Jul-1998 KyleP Created
//
//--------------------------------------------------------------------------
SCODE PauseMenu( IContextMenuCallback * piCallback, BOOL fTop )
{
aContextMenus[comidStartCI].fFlags =
aContextMenus[comidStartCITop].fFlags = MF_ENABLED;
aContextMenus[comidStopCI].fFlags =
aContextMenus[comidStopCITop].fFlags = MF_ENABLED;
aContextMenus[comidPauseCI].fFlags =
aContextMenus[comidPauseCITop].fFlags = MF_GRAYED;
return SetStartStopMenu( piCallback, fTop );
}
//+-------------------------------------------------------------------------
//
// Function: DisabledMenu
//
// Synopsis: Disables start/stop/pause menu items
//
// Arguments: [piCallback] -- Context menu callback routine
// [fTop] -- TRUE if menu items should be on top of list
// in addition to under Task section.
//
// History: 07-Jul-1998 KyleP Created
//
//--------------------------------------------------------------------------
SCODE DisabledMenu( IContextMenuCallback * piCallback, BOOL fTop )
{
aContextMenus[comidStartCI].fFlags =
aContextMenus[comidStartCITop].fFlags = MF_GRAYED;
aContextMenus[comidStopCI].fFlags =
aContextMenus[comidStopCITop].fFlags = MF_GRAYED;
aContextMenus[comidPauseCI].fFlags =
aContextMenus[comidPauseCITop].fFlags = MF_GRAYED;
return SetStartStopMenu( piCallback, fTop );
}
//+-------------------------------------------------------------------------
//
// Function: SetStartStopMenu, private
//
// Synopsis: Worker routine to call menu callback and set menu items.
//
// Arguments: [piCallback] -- Context menu callback routine
// [fTop] -- TRUE if menu items should be on top of list
// in addition to under Task section.
//
// History: 07-Jul-1998 KyleP Created
//
//--------------------------------------------------------------------------
SCODE SetStartStopMenu( IContextMenuCallback * piCallback, BOOL fTop )
{
SCODE sc = S_OK;
if (SUCCEEDED(sc) && fTop)
piCallback->AddItem( &aContextMenus[comidStartCITop] );
if (SUCCEEDED(sc) && fTop)
sc = piCallback->AddItem( &aContextMenus[comidStopCITop] );
if (SUCCEEDED(sc) && fTop)
sc = piCallback->AddItem( &aContextMenus[comidPauseCITop] );
if (SUCCEEDED(sc))
sc = piCallback->AddItem( &aContextMenus[comidStartCI] );
if (SUCCEEDED(sc))
sc = piCallback->AddItem( &aContextMenus[comidStopCI] );
if (SUCCEEDED(sc))
sc = piCallback->AddItem( &aContextMenus[comidPauseCI] );
return sc;
}