windows-server-2003/inetsrv/query/admin/snapin.cxx

//+-------------------------------------------------------------------------
//
//  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;
}