able/windows-nt
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity
windows-nt/Source/XPSP1/NT/inetsrv/iis/ui/admin/mmc/cinetmgr.cpp
CryptoAlgo Inc b3cac27891 Add source files
2020-09-26 16:20:57 +08:00

7193 lines
138 KiB
C++
Raw Blame History

/*++
Copyright (c) 1994-1998 Microsoft Corporation
Module Name :
cinetmgr.cpp
Abstract:
Snapin object
Author:
Ronald Meijer (ronaldm)
Project:
Internet Services Manager
Revision History:
--*/
//
// Include Files
//
#include "stdafx.h"
#include "inetmgr.h"
#include "cinetmgr.h"
#include "connects.h"
#include "dataobj.h"
#include "afxdlgs.h"
#include "constr.h"
#include "metaback.h"
#include "shutdown.h"
#include <shlwapi.h>
#if !MSDEV_BUILD
//
// An odd difference between vc++ and sdk environments.
//
#include <atlimpl.cpp>
#endif !MSDEV_BUILD
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif
//
// Image background colour for the toolbar buttons
//
#define RGB_BK_IMAGES (RGB(255,0,255)) // purple
#define ARRAYLEN(x) (sizeof(x) / sizeof((x)[0]))
static HRESULT
GetSnapinHelpFile(LPOLESTR * lpCompiledHelpFile);
//
// Toolbar Definition. String IDs for menu and tooltip text will be resolved at initialization
//
static MMCBUTTON SnapinButtons[] =
{
{ IDM_CONNECT - 1, IDM_CONNECT, TBSTATE_ENABLED, TBSTYLE_BUTTON, (BSTR)IDS_MENU_CONNECT, (BSTR)IDS_MENU_TT_CONNECT },
// { IDM_DISCOVER - 1, IDM_DISCOVER, TBSTATE_ENABLED, TBSTYLE_BUTTON, (BSTR)IDS_MENU_DISCOVER, (BSTR)IDS_MENU_TT_CONNECT },
{ 0, 0, TBSTATE_ENABLED, TBSTYLE_SEP, _T(" "), _T("") },
{ IDM_START - 1, IDM_START, TBSTATE_ENABLED, TBSTYLE_BUTTON, (BSTR)IDS_MENU_START, (BSTR)IDS_MENU_TT_START },
{ IDM_STOP - 1, IDM_STOP, TBSTATE_ENABLED, TBSTYLE_BUTTON, (BSTR)IDS_MENU_STOP, (BSTR)IDS_MENU_TT_STOP },
{ IDM_PAUSE - 1, IDM_PAUSE, TBSTATE_ENABLED, TBSTYLE_BUTTON, (BSTR)IDS_MENU_PAUSE, (BSTR)IDS_MENU_TT_PAUSE },
{ 0, 0, TBSTATE_ENABLED, TBSTYLE_SEP, _T(" "), _T("") },
//
// Add-on tools come here
//
};
#define NUM_BUTTONS (ARRAYLEN(SnapinButtons))
#define NUM_BITMAPS (5)
//
// Name of our taskpad group
//
const LPCTSTR g_cszTaskGroup = _T("CMTP1");
template <class TYPE>
TYPE * Extract(
IN LPDATAOBJECT lpDataObject,
IN unsigned int cf,
IN int len = -1
)
/*++
Routine Description:
Template function to extract information of type TYPE from the data object
Arguments:
LPDATAOBJECT lpDataObject : Data object to extract data from
unsigned int cf : Clipboard format describing the info type
Return Value:
Pointer to type TYPE
--*/
{
ASSERT(lpDataObject != NULL);
TYPE * p = NULL;
STGMEDIUM stgmedium = { TYMED_HGLOBAL, NULL };
FORMATETC formatetc =
{
(CLIPFORMAT)cf,
NULL,
DVASPECT_CONTENT,
-1,
TYMED_HGLOBAL
};
//
// Allocate memory for the stream
//
if (len < 0)
{
len = sizeof(TYPE);
}
stgmedium.hGlobal = GlobalAlloc(GMEM_SHARE, len);
do
{
if (stgmedium.hGlobal == NULL)
{
break;
}
HRESULT hr = lpDataObject->GetDataHere(&formatetc, &stgmedium);
if (FAILED(hr))
{
break;
}
p = (TYPE *)stgmedium.hGlobal;
if (p == NULL)
{
break;
}
}
while (FALSE);
return p;
}
//
// Data object extraction helpers
//
// <<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
/*
BOOL
IsMMCMultiSelectDataObject(
IN LPDATAOBJECT lpDataObject
)
{
if (lpDataObject == NULL)
{
return FALSE;
}
static UINT s_cf = 0;
if (s_cf == 0)
{
//
// Multi-select clipboard format not registered -- do it now
//
s_cf = RegisterClipboardFormat(CCF_MMC_MULTISELECT_DATAOBJECT);
}
FORMATETC fmt = {s_cf, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL};
return (lpDataObject->QueryGetData(&fmt) == S_OK);
}
*/
CLSID *
ExtractClassID(
IN LPDATAOBJECT lpDataObject
)
{
return Extract<CLSID>(lpDataObject, CDataObject::m_cfCoClass);
}
GUID *
ExtractNodeType(
IN LPDATAOBJECT lpDataObject
)
{
return Extract<GUID>(lpDataObject, CDataObject::m_cfNodeType);
}
wchar_t *
ExtractMachineName(
IN LPDATAOBJECT lpDataObject
)
{
wchar_t * lpszMachineName = Extract<wchar_t>(
lpDataObject,
CDataObject::m_cfISMMachineName,
(MAX_PATH + 1) * sizeof(wchar_t)
);
if (lpszMachineName == NULL)
{
//
// This is an extension -- grab the computer management
// name instead.
//
lpszMachineName = Extract<wchar_t>(
lpDataObject,
CDataObject::m_cfMyComputMachineName,
(MAX_PATH + 1) * sizeof(wchar_t)
);
}
return lpszMachineName;
}
INTERNAL *
ExtractInternalFormat(
IN LPDATAOBJECT lpDataObject
)
{
ASSERT(lpDataObject != NULL);
return Extract<INTERNAL>(lpDataObject, CDataObject::m_cfInternal);
}
//
// IEnumTask Implementation
//
// <<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
CEnumTasks::CEnumTasks()
/*++
Routine Description:
Constructor
Arguments:
None
Return Value:
N/A
--*/
: m_pObject(NULL)
{
}
CEnumTasks::~CEnumTasks()
/*++
Routine Description:
Destructor
Arguments:
None
Return Value:
None
--*/
{
}
HRESULT
CEnumTasks::Next(
IN ULONG celt,
IN MMC_TASK * rgelt,
IN ULONG * pceltFetched
)
/*++
Routine Description:
Add next task to taskpad
Arguments:
ULONG celt,
MMC_TASK * rgelt,
ULONG * pceltFetched
Return Value:
HRESULT
--*/
{
//
// Callee fills MMC_TASK elements (via CoTaskMemAlloc)
//
ASSERT(!IsBadWritePtr(rgelt, celt * sizeof(MMC_TASK)));
if (m_pObject == NULL)
{
//
// Must be at the snap-in's root handle
//
return S_FALSE;
}
//
// celt will actually always only be 1
//
ASSERT(celt == 1);
HRESULT hr = S_FALSE;
for (ULONG i = 0; i < celt; ++i)
{
MMC_TASK * task = &rgelt[i];
hr = m_pObject->AddNextTaskpadItem(task);
if (FAILED(hr))
{
if (pceltFetched)
{
*pceltFetched = i;
}
break;
}
}
//
// If we get here all is well
//
if(pceltFetched)
{
*pceltFetched = celt;
}
return hr;
}
HRESULT
CEnumTasks::Skip(
IN ULONG celt
)
/*++
Routine Description:
Skip the task index
Arguments:
ULONG celt : Number of tasks to skip
Return Value:
HRESULT
--*/
{
return E_NOTIMPL;
}
HRESULT
CEnumTasks::Reset()
/*++
Routine Description:
Result the taskpad enumeration index
Arguments:
None
Return Value:
HRESULT
--*/
{
return E_NOTIMPL;
}
HRESULT
CEnumTasks::Clone(
IN IEnumTASK ** ppenum
)
/*++
Routine Description:
Close a task -- obsolete, no longer supported by mmc
Arguments:
IEnumTASK ** ppenum : Source task to clone
Return Value:
HRESULT
--*/
{
return E_NOTIMPL;
}
HRESULT
CEnumTasks::Init(
IN IDataObject * pdo,
IN LPOLESTR szTaskGroup
)
/*++
Routine Description:
Here is where we see what taskpad we are providing tasks for.
In our case we know that we only have one taskpad.
The string we test for is "CMTP1". This was the string following
the '#' that we passed in GetResultViewType.
Arguments:
IDataObject * pdo : Data object
LPOLESTR szTaskGroup : Taskpad group name
Return Value:
HRESULT
--*/
{
//
// Return ok if we can handle data object and group.
//
if (!lstrcmp(szTaskGroup, g_cszTaskGroup))
{
//
// CODEWORK: How about a helper for this!
//
INTERNAL * pInternal = ExtractInternalFormat(pdo);
if (pInternal == NULL)
{
//
// Extension
//
return S_OK;
}
m_pObject = (CIISObject *)pInternal->m_cookie;
return S_OK;
}
//
// Should never happen
//
ASSERT(FALSE);
return S_FALSE;
}
//
// CSnapin class
//
// <<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
DEBUG_DECLARE_INSTANCE_COUNTER(CSnapin);
long CSnapin::lDataObjectRefCount = 0;
CSnapin::CSnapin()
/*++
Routine Description:
Constructor
Arguments:
None
Return Value:
N/A
--*/
: m_pConsole(NULL),
m_pHeader(NULL),
m_pResult(NULL),
m_pImageResult(NULL),
m_pComponentData(NULL),
m_pToolbar(NULL),
m_pControlbar(NULL),
m_pbmpToolbar(NULL),
m_pConsoleVerb(NULL),
m_oblResultItems(),
m_strlRef(),
m_fTaskView(FALSE),
m_fSettingsChanged(FALSE),
m_fIsExtension(FALSE),
m_fWinSockInit(FALSE)
{
AFX_MANAGE_STATE(::AfxGetStaticModuleState());
DEBUG_INCREMENT_INSTANCE_COUNTER(CSnapin);
#ifdef _DEBUG
dbg_cRef = 0;
#endif // _DEBUG
}
CSnapin::~CSnapin()
/*++
Routine Description:
Destructor
Arguments:
N/A
Return Value:
N/A
--*/
{
#ifdef _DEBUG
ASSERT(dbg_cRef == 0);
#endif // _DEBUG
DEBUG_DECREMENT_INSTANCE_COUNTER(CSnapin);
SAFE_RELEASE(m_pToolbar);
SAFE_RELEASE(m_pControlbar);
//
// Make sure the interfaces have been released
//
ASSERT(m_pConsole == NULL);
ASSERT(m_pHeader == NULL);
ASSERT(m_pToolbar == NULL);
if (m_pbmpToolbar)
{
m_pbmpToolbar->DeleteObject();
delete m_pbmpToolbar;
}
//
// Should have been removed via notification, but just in case:
//
ASSERT(m_oblResultItems.GetCount() == 0);
m_oblResultItems.RemoveAll();
}
LPTSTR
CSnapin::StringReferenceFromResourceID(
IN UINT nID
)
/*++
Routine Description:
Load a string from the resource segment, add it to the internal
string table, and return a pointer to it. The pointer will be
valid for the entire scope of the CSnapin object
Arguments
UINT nID : Resource ID
Return Value:
A pointer to the string.
--*/
{
CString str;
VERIFY(str.LoadString(nID));
m_strlRef.AddTail(str);
CString & strRef = m_strlRef.GetAt(m_strlRef.GetTailPosition());
return (LPTSTR)(LPCTSTR)strRef;
}
BOOL
CSnapin::IsEnumerating(
IN LPDATAOBJECT lpDataObject
)
/*++
Routine Description:
Return TRUE if we are enumerating our main folder
Arguments:
LPDATAOBJECT lpDataObject : Data object
Return Value:
TRUE if we are enumerating our main folder, FALSE if not
--*/
{
BOOL bResult = FALSE;
ASSERT(lpDataObject);
GUID * nodeType = ExtractNodeType(lpDataObject);
//
// Is this my main node (static folder node type)
//
bResult = ::IsEqualGUID(*nodeType, cInternetRootNode);
//
// Free resources
//
FREE_DATA(nodeType);
return bResult;
}
//
// CSnapin's IComponent implementation
//
// <<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
STDMETHODIMP
CSnapin::GetResultViewType(
IN MMC_COOKIE cookie,
OUT BSTR * ppViewType,
OUT long * pViewOptions
)
/*++
Routine Description:
Tell MMC what our result view looks like
Arguments:
MMC_COOKIE cookie : Currently selected cookie
BSTR * ppViewType : Return view type here
long * pViewOptions : View options
Return Value:
S_FALSE to use default view type, S_OK indicates the
view type is returned in *ppViewType
--*/
{
//
// Taskpads not supported for beta2
//
#if TASKPADS_SUPPORTED
if (m_fTaskView)
{
//
// We will use the default DHTML provided by MMC. It actually
// resides as a resource inside MMC.EXE. We just get the path
// to it and use that.
//
// The one piece of magic here is the text following the '#'. That
// is the special way we have of identifying they taskpad we are
// talking about. Here we say we are wanting to show a taskpad
// that we refer to as "CMTP1". We will actually see this string
// pass back to us later. If someone is extending our taskpad,
// they also need to know what this secret string is.
//
*pViewOptions = MMC_VIEW_OPTIONS_NONE;
CString strResURL;
HRESULT hr = BuildResURL(strResURL, NULL);
if (FAILED(hr))
{
return hr;
}
strResURL += _T("/default.htm#");
strResURL += g_cszTaskGroup;
TRACEEOLID("Taskpad URL is " << strResURL);
*ppViewType = CoTaskDupString((LPCOLESTR)strResURL);
return S_OK;
}
#endif // TASKPADS_SUPPORTED
//
// Use default view
//
*pViewOptions = MMC_VIEW_OPTIONS_USEFONTLINKING;
//*pViewOptions = MMC_VIEW_OPTIONS_USEFONTLINKING
// | MMC_VIEW_OPTIONS_LEXICAL_SORT;
return S_FALSE;
}
STDMETHODIMP
CSnapin::Initialize(
IN LPCONSOLE lpConsole
)
/*++
Routine Description:
Initialize the console
Arguments"
LPCONSOLE lpConsole : Pointer to console
Return Value:
HRESULT
--*/
{
ASSERT(lpConsole != NULL);
::AfxEnableControlContainer();
//
// Initialize IISUI extension DLL
//
#ifndef _COMSTATIC
//
// Initialize IISUI extension DLL
//
InitIISUIDll();
#endif // _COMSTATIC
//
// Initialise winsock
//
// ISSUE: If proxy is installed, this might take a long time
// if the proxy server is in a hung state.
//
WSADATA wsaData;
m_fWinSockInit = (::WSAStartup(MAKEWORD(1, 1), &wsaData) == 0);
//
// Save the IConsole pointer
//
m_pConsole = lpConsole;
m_pConsole->AddRef();
//
// Load resource strings
//
LoadResources();
//
// QI for a IHeaderCtrl
//
HRESULT hr = m_pConsole->QueryInterface(
IID_IHeaderCtrl,
(void **)&m_pHeader
);
//
// Give the console the header control interface pointer
//
if (SUCCEEDED(hr))
{
m_pConsole->SetHeader(m_pHeader);
}
m_pConsole->QueryInterface(
IID_IResultData,
(void **)&m_pResult
);
m_pConsole->QueryResultImageList(&m_pImageResult);
m_pConsole->QueryConsoleVerb(&m_pConsoleVerb);
//
// Resolve tool tips texts
//
CString str;
static BOOL fInitialised = FALSE;
if (!fInitialised)
{
AFX_MANAGE_STATE(::AfxGetStaticModuleState());
for (int i = 0; i < NUM_BUTTONS; ++i)
{
if (SnapinButtons[i].idCommand != 0)
{
SnapinButtons[i].lpButtonText = StringReferenceFromResourceID(
(UINT)PtrToUlong(SnapinButtons[i].lpButtonText));
SnapinButtons[i].lpTooltipText = StringReferenceFromResourceID(
(UINT)PtrToUlong(SnapinButtons[i].lpTooltipText));
}
}
fInitialised = TRUE;
}
return S_OK;
}
STDMETHODIMP
CSnapin::Notify(
IN LPDATAOBJECT lpDataObject,
IN MMC_NOTIFY_TYPE event,
IN LPARAM arg,
IN LPARAM param
)
/*++
Routine Description:
Notification handler
Arguments:
LPDATAOBJECT lpDataObject : Data object
MMC_NOTIFY_TYPE event : Event
LPARAM arg : Argument
LPARAM param : Parameter
Return Value:
HRESULT
--*/
{
HRESULT hr = S_OK;
INTERNAL * pInternal = NULL;
MMC_COOKIE cookie;
switch(event)
{
case MMCN_REFRESH:
case MMCN_RENAME:
//
// Delegate it to the IComponentData
//
ASSERT(m_pComponentData != NULL);
hr = m_pComponentData->Notify(lpDataObject, event, arg, param);
case MMCN_PROPERTY_CHANGE:
hr = OnPropertyChange(lpDataObject);
break;
case MMCN_DELETE:
//
// Let IComponentData do the deletion, then refresh the parent object
//
ASSERT(m_pComponentData != NULL);
hr = m_pComponentData->Notify(lpDataObject, event, arg, param);
m_pComponentData->Notify(lpDataObject, MMCN_REFRESH, arg, param);
case MMCN_VIEW_CHANGE:
hr = OnUpdateView(lpDataObject);
break;
case MMCN_SNAPINHELP:
case MMCN_CONTEXTHELP:
{
LPOLESTR pCompiledHelpFile = NULL;
if (SUCCEEDED(hr = GetSnapinHelpFile(&pCompiledHelpFile)))
{
IDisplayHelp * pdh;
if (SUCCEEDED(hr = m_pConsole->QueryInterface(
IID_IDisplayHelp, (void **)&pdh)))
{
CString topic = PathFindFileName(pCompiledHelpFile);
topic += _T("::/iint1.htm");
LPTSTR p = topic.GetBuffer(topic.GetLength());
hr = pdh->ShowTopic(p);
topic.ReleaseBuffer();
pdh->Release();
}
}
if (FAILED(hr))
{
CError err(hr);
err.MessageBoxOnFailure();
}
CoTaskMemFree(pCompiledHelpFile);
}
break;
default:
//
// don't process the internal format if the dataobject is null.
//
if (lpDataObject == NULL)
{
return S_OK;
}
INTERNAL * pInternal = ExtractInternalFormat(lpDataObject);
if (pInternal == NULL)
{
//
// Extension
//
TRACEEOLID("This is an (CSnapin) EXTENSION");
cookie = NULL;
CIISObject::m_fIsExtension = m_fIsExtension = TRUE;
}
else
{
cookie = pInternal->m_cookie;
FREE_DATA(pInternal);
}
switch(event)
{
case MMCN_CLICK:
case MMCN_DBLCLICK:
hr = OnResultItemClkOrDblClk(cookie, (event == MMCN_DBLCLICK));
break;
case MMCN_ADD_IMAGES:
OnAddImages(cookie, arg, param);
break;
case MMCN_SHOW:
hr = OnShow(cookie, arg, param);
break;
case MMCN_ACTIVATE:
hr = OnActivate(cookie, arg, param);
break;
case MMCN_MINIMIZED:
hr = OnMinimize(cookie, arg, param);
break;
case MMCN_BTN_CLICK:
break;
case MMCN_SELECT:
HandleStandardVerbs(arg, lpDataObject);
break;
case MMCN_COLUMN_CLICK:
break;
case MMCN_MENU_BTNCLICK:
break;
default:
ASSERT(FALSE); // Handle new messages
hr = E_UNEXPECTED;
break;
}
}
return hr;
}
STDMETHODIMP
CSnapin::Destroy(
IN MMC_COOKIE cookie
)
/*++
Routine Description:
Destruction handler
Arguments:
MMC_COOKIE cookie : Currently selected cookie
Return Value:
HRESULT
--*/
{
//
// Release the interfaces that we QI'ed
//
AFX_MANAGE_STATE(AfxGetStaticModuleState());
if (m_pConsole != NULL)
{
//
// Tell the console to release the header control interface
//
m_pConsole->SetHeader(NULL);
SAFE_RELEASE_SETTONULL(m_pHeader);
SAFE_RELEASE_SETTONULL(m_pResult);
SAFE_RELEASE_SETTONULL(m_pImageResult);
SAFE_RELEASE_SETTONULL(m_pConsole);
SAFE_RELEASE_SETTONULL(m_pComponentData);
SAFE_RELEASE_SETTONULL(m_pConsoleVerb);
SAFE_RELEASE_SETTONULL(m_pToolbar);
SAFE_RELEASE_SETTONULL(m_pControlbar);
}
//
// Terminate use of the WinSock routines.
//
if (m_fWinSockInit)
{
::WSACleanup();
m_fWinSockInit = FALSE;
}
//
// Everything OK
//
return S_OK;
}
STDMETHODIMP
CSnapin::QueryDataObject(
IN MMC_COOKIE cookie,
IN DATA_OBJECT_TYPES type,
OUT LPDATAOBJECT * ppDataObject
)
/*++
Routine Description:
Get dataobject info
Arguments:
MMC_COOKIE cookie : cookie
DATA_OBJECT_TYPES : Data object type
LPDATAOBJECT * : Returns the data object
Return Value:
HRESULT
--*/
{
//
// Delegate it to the IComponentData
//
ASSERT(m_pComponentData != NULL);
return m_pComponentData->QueryDataObject(cookie, type, ppDataObject);
}
void
CSnapin::LoadResources()
/*++
Routine Description:
Load resource belong to the main snapin
Arguments:
None
Return Value:
None
--*/
{
//
// Load strings from resources
//
}
//
// IExtendPropertySheet Implementation
//
// <<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
STDMETHODIMP
CSnapin::CreatePropertyPages(
IN LPPROPERTYSHEETCALLBACK lpProvider,
IN LONG_PTR handle,
IN LPDATAOBJECT lpDataObject
)
/*++
Routine Description:
Create the property pages for the given object
Arguments:
LPPROPERTYSHEETCALLBACK lpProvider : Provider
LONG_PTR handle : Handle
LPDATAOBJECT lpDataObject : Data object
Return Value:
HRESULT
--*/
{
return static_cast<CComponentDataImpl *>(m_pComponentData)->CreatePropertyPages(
lpProvider,
handle,
lpDataObject
);
}
STDMETHODIMP
CSnapin::QueryPagesFor(
IN LPDATAOBJECT lpDataObject
)
/*++
Routine Description:
Check to see if a property sheet should be brought up for this data
object
Arguments:
LPDATAOBJECT lpDataObject : Data object
Return Value:
S_OK, if properties may be brought up for this item, S_FALSE otherwise
--*/
{
return static_cast<CComponentDataImpl *>(m_pComponentData)->QueryPagesFor(
lpDataObject
);
}
HRESULT
CSnapin::InitializeHeaders(
IN MMC_COOKIE cookie
)
/*++
Routine Description:
Initialize the result view headers for this cookie
Arguments:
MMC_COOKIE cookie : Current cookie
Return Value:
HRESULT
--*/
{
CIISObject * pObject = (CIISObject *)cookie;
//
// The result pane is always one level behind the scope pane
// for a folder object
//
if (pObject == NULL)
{
CIISMachine::InitializeHeaders(m_pHeader);
}
else
{
pObject->InitializeChildHeaders(m_pHeader);
}
return S_OK;
}
class CImage : public CObjectPlus
/*++
Class Description:
A structure to tie a bitmap together with a background
colour mask
Public Interface:
CImage : Constructor
~CImage : Destructor
Notes: This image owns the bitmap and will delete it upon
destruction.
--*/
{
//
// Constructor/Destructor
//
public:
CImage(
IN CBitmap * pbmp,
IN COLORREF rgb
)
: m_pbmp(pbmp),
m_rgb(rgb)
{
}
~CImage()
{
ASSERT(m_pbmp != NULL);
m_pbmp->DeleteObject();
delete m_pbmp;
}
//
// Access
//
public:
//
// Get the bitmap object
//
CBitmap & GetBitmap() { return *m_pbmp; }
//
// Get the background colour definition
//
COLORREF & GetBkColor() { return m_rgb; }
private:
CBitmap * m_pbmp;
COLORREF m_rgb;
};
//
// Global oblists of service bitmaps -- used by scope and result side
//
CObListPlus g_obl16x16;
CObListPlus g_obl32x32;
//
// Add-on-tools definitions
//
CObListPlus g_oblAddOnTools;
HRESULT
CSnapin::InitializeBitmaps(
IN MMC_COOKIE cookie
)
/*++
Routine Description:
Initialize the service bitmaps
Arguments:
MMC_COOKIE cookie : Currently selected cookie (CIISObject *)
Return Value:
HRESULT
--*/
{
ASSERT(m_pImageResult != NULL);
CBitmap bmp16x16,
bmp32x32,
bmpMgr16x16,
bmpMgr32x32;
{
//
// Load the bitmaps from the dll
//
AFX_MANAGE_STATE(::AfxGetStaticModuleState());
VERIFY(bmp16x16.LoadBitmap(IDB_VIEWS16));
VERIFY(bmp32x32.LoadBitmap(IDB_VIEWS32));
VERIFY(bmpMgr16x16.LoadBitmap(IDB_INETMGR16));
VERIFY(bmpMgr32x32.LoadBitmap(IDB_INETMGR16));
}
//
// Set the images
//
HRESULT hr = m_pImageResult->ImageListSetStrip(
(LONG_PTR *)(HBITMAP)bmp16x16,
(LONG_PTR *)(HBITMAP)bmp32x32,
0,
RGB_BK_IMAGES
);
ASSERT(hr == S_OK);
//
// Add on inetmgr bitmap
//
hr = m_pImageResult->ImageListSetStrip(
(LONG_PTR *)(HBITMAP)bmpMgr16x16,
(LONG_PTR *)(HBITMAP)bmpMgr32x32,
BMP_INETMGR,
RGB_BK_IMAGES
);
//
// Add the ones from the service config DLLs
//
POSITION pos16x16 = g_obl16x16.GetHeadPosition();
POSITION pos32x32 = g_obl32x32.GetHeadPosition();
int i = BMP_SERVICE;
while (pos16x16 && pos32x32)
{
CImage * pimg16x16 = (CImage *)g_obl16x16.GetNext(pos16x16);
CImage * pimg32x32 = (CImage *)g_obl32x32.GetNext(pos32x32);
ASSERT(pimg16x16 && pimg32x32);
hr = m_pImageResult->ImageListSetStrip(
(LONG_PTR *)(HBITMAP)pimg16x16->GetBitmap(),
(LONG_PTR *)(HBITMAP)pimg32x32->GetBitmap(),
i++,
pimg16x16->GetBkColor()
);
ASSERT(hr == S_OK);
}
return hr;
}
STDMETHODIMP
CSnapin::GetDisplayInfo(
IN LPRESULTDATAITEM lpResultDataItem
)
/*++
Routine Description:
Get display info for result item
Arguments:
LPRESULTDATAITEM lpResultDataItem : Address of result data tiem
Return Value:
HRESULT
--*/
{
wchar_t * szString = _T("");
ASSERT(lpResultDataItem != NULL);
static int nImage = -1;
static CString str;
if (lpResultDataItem)
{
CIISObject * pObject = (CIISObject *)lpResultDataItem->lParam;
ASSERT(pObject != NULL);
if (pObject != NULL && pObject->IsValidObject())
{
pObject->GetResultDisplayInfo(lpResultDataItem->nCol, str, nImage);
lpResultDataItem->str = (LPTSTR)(LPCTSTR)str;
lpResultDataItem->nImage = nImage;
return S_OK;
}
}
return S_FALSE;
}
//
// IExtendContextMenu Implementation
//
// <<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<</
STDMETHODIMP
CSnapin::AddMenuItems(
IN LPDATAOBJECT lpDataObject,
IN LPCONTEXTMENUCALLBACK lpContextMenuCallback,
IN long * pInsertionAllowed
)
/*++
Routine Description:
Add menu items -- pass it on the the component date
Arguments:
LPDATAOBJECT lpDataObject : Data object
LPCONTEXTMENUCALLBACK lpContextMenuCallback : Context menu
long * pInsertionAllowed : TRUE if insertion is allowed
Return Value:
HRESULT
--*/
{
//
// Cut support for taskpads in beta2
//
#if TASKPADS_SUPPORTED
if (*pInsertionAllowed & CCM_INSERTIONALLOWED_VIEW)
{
//
// Add View Menu Items
//
CIISObject::AddMenuItemByCommand(
lpContextMenuCallback,
IDM_VIEW_TASKPAD,
m_fTaskView ? MF_CHECKED : 0
);
}
#endif // TASKPADS_SUPPORTED
//
// Pass it on to CComponentDataImpl
//
return static_cast<CComponentDataImpl *>(m_pComponentData)->AddMenuItems(
lpDataObject,
lpContextMenuCallback,
pInsertionAllowed
);
}
STDMETHODIMP
CSnapin::Command(
IN long nCommandID,
IN LPDATAOBJECT lpDataObject
)
/*++
Routine Description:
Command handler -- pass it on the component data
Arguments:
long nCommandID : Command ID
LPDTATAOBJECT lpDataObject : Data object
Return Value:
HRESULT
--*/
{
HRESULT hr = S_OK;
INTERNAL * pInternal = NULL;
switch(nCommandID)
{
case -1:
//
// Built-in views -- everything handled.
//
m_fTaskView = FALSE;
m_fSettingsChanged = TRUE;
break;
#if TASKPADS_SUPPORTED
//
// No taskpad support in beta2
//
case IDM_VIEW_TASKPAD:
//
// Handle view change
//
m_fTaskView = !m_fTaskView;
m_fSettingsChanged = TRUE;
//
// Reselect current scope item to force view change
//
pInternal = ExtractInternalFormat(lpDataObject);
if (pInternal != NULL)
{
CIISObject * pObject = (CIISObject *)pInternal->m_cookie;
FREE_DATA(pInternal);
if (pObject)
{
m_pConsole->SelectScopeItem(pObject->GetScopeHandle());
}
else
{
//
// Must be the root item
//
m_pConsole->SelectScopeItem(
static_cast<CComponentDataImpl *>(m_pComponentData)->GetRootHandle()
);
}
}
break;
#endif // TASKPADS_SUPPORTED
default:
//
// Pass it on to CComponentDataImpl
//
hr = static_cast<CComponentDataImpl *>(m_pComponentData)->Command(
nCommandID,
lpDataObject
);
}
return hr;
}
//
// ITaskPad implementation
//
// <<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
HRESULT
CSnapin::TaskNotify(
IN IDataObject * pdo,
IN VARIANT * pvarg,
IN VARIANT * pvparam
)
/*++
Routine Description:
Handle notification from taskpad
Arguments:
IDataObject * pdo,
VARIANT * pvarg,
VARIANT * pvparam
Return Value:
HRESULT
--*/
{
if (pvarg->vt == VT_I4)
{
//
// Pass it on to CComponentDataImpl
//
return static_cast<CComponentDataImpl *>(m_pComponentData)->Command(
pvarg->lVal,
pdo
);
}
return S_OK;
}
HRESULT
CSnapin::GetTitle(
IN LPOLESTR szGroup,
OUT LPOLESTR * lpszTitle
)
/*++
Routine Description:
Get the title of the taskpad
Arguments:
LPOLESTR szGroup : Group name
LPOLESTR * szTitle : Returns title
Return Value:
HRESULT
--*/
{
OLECHAR sztitle[] = L"IIS TaskPad";
*lpszTitle = CoTaskDupString(sztitle);
return *lpszTitle != NULL ? S_OK : E_OUTOFMEMORY;
}
HRESULT
CSnapin::GetDescriptiveText(
IN LPOLESTR szGroup,
OUT LPOLESTR * lpszDescriptiveText
)
/*++
Routine Description:
Get the descriptive text
Arguments:
LPOLESTR szGroup : Group name
LPOLESTR * szTitle : Returns title
Return Value:
HRESULT
--*/
{
return E_NOTIMPL;
}
HRESULT
CSnapin::GetBanner(
IN LPOLESTR szGroup,
OUT LPOLESTR * pszBitmapResource
)
/*++
Routine Description:
Get the banner resource
Arguments:
LPOLESTR szGroup : Group name
LPOLESTR * pszBitmapResource : Returns bitmap resource
Return Value:
HRESULT
--*/
{
CString strResURL;
HRESULT hr = BuildResURL(strResURL);
if (FAILED(hr))
{
return hr;
}
strResURL += _T("/img\\ntbanner.gif");
TRACEEOLID(strResURL);
*pszBitmapResource = CoTaskDupString((LPCOLESTR)strResURL);
return *pszBitmapResource != NULL ? S_OK : E_OUTOFMEMORY;
}
HRESULT
CSnapin::GetBackground(
IN LPOLESTR szGroup,
OUT MMC_TASK_DISPLAY_OBJECT * pTDO
)
/*++
Routine Description:
Get the background resource
Arguments:
LPOLESTR szGroup : Group name
LPOLESTR * pszBitmapResource : Returns bitmap resource
Return Value:
HRESULT
--*/
{
return E_NOTIMPL;
}
HRESULT
CSnapin::EnumTasks(
IN IDataObject * pdo,
IN LPOLESTR szTaskGroup,
OUT IEnumTASK ** ppEnumTASK
)
/*++
Routine Description:
Enumerate the tasks on the taskpad
Arguments:
IDataObject * pdo : Data object selected;
LPOLESTR szTaskGroup, : Taskgroup name
IEnumTASK ** ppEnumTASK : Returns enumtask interface
Return Value:
HRESULT
--*/
{
HRESULT hr = S_OK;
try
{
ASSERT(ppEnumTASK != NULL);
CComObject<CEnumTasks>* pObject;
CComObject<CEnumTasks>::CreateInstance(&pObject);
ASSERT(pObject != NULL);
VERIFY(SUCCEEDED(pObject->Init(pdo, szTaskGroup)));
hr = pObject->QueryInterface(
IID_IEnumTASK,
reinterpret_cast<void **>(ppEnumTASK)
);
}
catch(CMemoryException * e)
{
e->Delete();
hr = E_OUTOFMEMORY;
}
return hr;
}
//
// IExtendControlbar implementation
//
// <<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
STDMETHODIMP
CSnapin::SetControlbar(
IN LPCONTROLBAR lpControlbar
)
/*++
Routine Description:
Set the control bar
Arguments:
LPCONTROLBAR lpControlbar : Pointer to control bar
Return Value:
HRESULT
--*/
{
if (lpControlbar != NULL)
{
//
// Hold on to the controlbar interface.
//
if (m_pControlbar != NULL)
{
m_pControlbar->Release();
}
m_pControlbar = lpControlbar;
m_pControlbar->AddRef();
HRESULT hr = S_FALSE;
//
// Create the Toolbar
//
if (!m_pToolbar)
{
hr = m_pControlbar->Create(TOOLBAR, this, (LPUNKNOWN *)&m_pToolbar);
ASSERT(SUCCEEDED(hr));
m_pbmpToolbar = new CBitmap;
{
//
// Add the bitmap
//
AFX_MANAGE_STATE(AfxGetStaticModuleState());
m_pbmpToolbar->LoadBitmap(IDB_TOOLBAR);
}
//
// Add 16x16 bitmaps
//
hr = m_pToolbar->AddBitmap(
NUM_BITMAPS,
(HBITMAP)*m_pbmpToolbar,
16,
16,
TB_COLORMASK
);
ASSERT(SUCCEEDED(hr));
//
// Add the buttons to the toolbar
//
hr = m_pToolbar->AddButtons(NUM_BUTTONS, SnapinButtons);
ASSERT(SUCCEEDED(hr));
//
// Now add the add-on tools
//
POSITION pos = g_oblAddOnTools.GetHeadPosition();
//
// Toolbar buttons were created using buttonface
// background colour
//
COLORREF rgbMask = ::GetSysColor(COLOR_BTNFACE);
while (pos != NULL)
{
CISMShellExecutable * pTool =
(CISMShellExecutable *)g_oblAddOnTools.GetNext(pos);
ASSERT(pTool != NULL);
if (pTool->ShowInToolBar())
{
if (pTool->InitializedOK())
{
//
// Add 16x16 image
//
hr = m_pToolbar->AddBitmap(
1,
pTool->GetBitmap(),
16,
16,
rgbMask
);
ASSERT(SUCCEEDED(hr));
hr = m_pToolbar->AddButtons(1, pTool->GetButton());
ASSERT(SUCCEEDED(hr));
}
}
}
}
}
else
{
SAFE_RELEASE_SETTONULL(m_pControlbar);
}
return S_OK;
}
CISMShellExecutable *
CSnapin::GetCommandAt(
IN CObListPlus & obl,
IN int nIndex
)
/*++
Routine Description:
Get the add-on tool at the given index.
Arguments:
int nIndex : Index where to look for the add-on tool
Return Value:
Shell Executable object pointer, or NULL if the index was not valid
--*/
{
if (nIndex < 0 || nIndex >= obl.GetCount())
{
TRACEEOLID("Invalid tool index requested");
return NULL;
}
return (CISMShellExecutable *)obl.GetAt(obl.FindIndex(nIndex));
}
HRESULT
CSnapin::SetToolbarStates(
IN MMC_COOKIE cookie
)
/*++
Routine Description:
Set toolbar states based on current selection
Arguments:
MMC_COOKIE cookie : Currently selected scope item cookie (CIISObject *)
Return Value:
HRESULT
--*/
{
CIISObject * pObject = (CIISObject *)cookie;
if (m_pToolbar)
{
m_pToolbar->SetButtonState(
IDM_CONNECT,
ENABLED,
!CIISObject::m_fIsExtension
);
m_pToolbar->SetButtonState(
IDM_PAUSE,
ENABLED,
pObject && pObject->IsPausable()
);
m_pToolbar->SetButtonState(
IDM_START,
ENABLED,
pObject && pObject->IsStartable()
);
m_pToolbar->SetButtonState(
IDM_STOP,
ENABLED,
pObject && pObject->IsStoppable()
);
m_pToolbar->SetButtonState(
IDM_PAUSE,
BUTTONPRESSED,
pObject && pObject->IsPaused()
);
}
return S_OK;
}
HRESULT
CSnapin::OnButtonClick(
IN LPDATAOBJECT lpDataObject,
IN long lID
)
/*++
Routine Description:
Handle toolbar button click
Arguments:
LPDATAOBJECT lpDataObject : Current data object
LONG lID : Button ID pressed
Return Value:
HRESULT
--*/
{
//
// Check to see if the button ID is in the add-on tools
// range
//
if (lID >= IDM_TOOLBAR)
{
INTERNAL * pInternal = ExtractInternalFormat(lpDataObject);
CIISObject * pObject = pInternal
? (CIISObject *)pInternal->m_cookie
: NULL;
//
// Button ID was in the add-on tools range. Match
// up the ID with the add-on tool, and execute same.
//
CISMShellExecutable * pTool = GetCommandAt(
g_oblAddOnTools,
lID - IDM_TOOLBAR
);
ASSERT(pTool != NULL);
if (pTool != NULL)
{
CError err;
if (pObject == NULL)
{
//
// Nothing selected, execute with no parameters
//
err = pTool->Execute();
}
else
{
//
// Pass server/service to the add-on tool
//
LPCTSTR lpstrServer = pObject->GetMachineName();
LPCTSTR lpstrService = pObject->GetServiceName();
err = pTool->Execute(lpstrServer, lpstrService);
}
if (err.Failed())
{
AFX_MANAGE_STATE(AfxGetStaticModuleState());
err.MessageBox();
}
}
FREE_DATA(pInternal);
}
else
{
//
// Otherwise, it maps to a menu command, pass it on to the
// component data
//
ASSERT(m_pComponentData != NULL);
((CComponentDataImpl *)m_pComponentData)->Command(lID, lpDataObject);
}
return S_OK;
}
void
CSnapin::HandleStandardVerbs(
IN LPARAM arg,
IN LPDATAOBJECT lpDataObject
)
/*++
Routine Description:
Set the standard verb states based on current selection
Arguments:
LPARAM arg : Argument
LPDATAOBJECT lpDataObject : Selected data object
Return Value:
None
--*/
{
if (lpDataObject == NULL || m_pConsoleVerb == NULL)
{
return;
}
CIISObject * pObject = NULL;
INTERNAL * pInternal = ExtractInternalFormat(lpDataObject);
if (pInternal != NULL)
{
pObject = (CIISObject *)pInternal->m_cookie;
}
m_pConsoleVerb->SetVerbState(
MMC_VERB_RENAME,
ENABLED,
pObject && pObject->IsRenamable()
);
m_pConsoleVerb->SetVerbState(MMC_VERB_COPY, HIDDEN, TRUE);
m_pConsoleVerb->SetVerbState(MMC_VERB_PASTE, HIDDEN, TRUE);
m_pConsoleVerb->SetVerbState(
MMC_VERB_DELETE,
ENABLED,
pObject && pObject->IsDeletable()
);
m_pConsoleVerb->SetVerbState(
MMC_VERB_REFRESH,
ENABLED,
pObject && pObject->IsRefreshable()
);
#ifdef MMC_PAGES
BOOL fConfig = pObject && pObject->IsMMCConfigurable();
m_pConsoleVerb->SetVerbState(MMC_VERB_PROPERTIES, ENABLED,fConfig);
if (pObject && pObject->IsLeafNode())
{
m_pConsoleVerb->SetDefaultVerb(MMC_VERB_PROPERTIES);
}
else
{
m_pConsoleVerb->SetDefaultVerb(MMC_VERB_OPEN);
}
#else
m_pConsoleVerb->SetVerbState(MMC_VERB_PROPERTIES, HIDDEN, TRUE);
#endif
}
void
CSnapin::HandleToolbar(
IN LPARAM arg,
IN LPARAM param
)
/*++
Routine Description:
Handle toolbar states
Arguments:
LPARAM arg : Selection change notification argument
LPARAM param : Selection change notification parameter
Return Value:
None.
--*/
{
INTERNAL * pInternal = NULL;
CIISObject * pObject = NULL;
HRESULT hr;
LPDATAOBJECT lpDataObject = NULL;
BOOL bScope = (BOOL)LOWORD(arg);
BOOL bSelect = (BOOL)HIWORD(arg);
if (bScope)
{
lpDataObject = (LPDATAOBJECT)param;
if (lpDataObject != NULL)
{
pInternal = ExtractInternalFormat(lpDataObject);
}
if (pInternal == NULL)
{
return;
}
pObject = (CIISObject *)pInternal->m_cookie;
//
// Attach the toolbars to the window
//
if (m_pControlbar != NULL)
{
hr = m_pControlbar->Attach(TOOLBAR, (LPUNKNOWN)m_pToolbar);
ASSERT(SUCCEEDED(hr));
}
}
else
{
//
// Result Pane -- disable all if nothing selected.
//
if (bSelect)
{
lpDataObject = (LPDATAOBJECT)param;
if (lpDataObject != NULL)
{
pInternal = ExtractInternalFormat(lpDataObject);
}
if (pInternal == NULL)
{
return;
}
pObject = (CIISObject *)pInternal->m_cookie;
//
// Attach the toolbars to the window
//
hr = m_pControlbar->Attach(TOOLBAR, (LPUNKNOWN)m_pToolbar);
ASSERT(SUCCEEDED(hr));
}
}
if (pInternal)
{
FREE_DATA(pInternal);
}
SetToolbarStates((MMC_COOKIE)pObject);
}
STDMETHODIMP
CSnapin::ControlbarNotify(
IN MMC_NOTIFY_TYPE event,
IN LPARAM arg,
IN LPARAM param
)
/*++
Routine Description:
Handle control bar notification
Arguments:
MMC_NOTIFY_TYPE event : Notification event
LPARAM arg : argument as needed
LPARAM param : Parameter ad needed
Return Value:
HRESULT
--*/
{
HRESULT hr = S_OK;
switch (event)
{
case MMCN_BTN_CLICK:
//
// Note for MMC: it seems to be that arg and param
// should be reversed here. The MMCN_SELECT
// casts the dataobject to the param (which makes
// more sense anyway).
//
TRACEEOLID("CSnapin::ControlbarNotify - MMCN_BTN_CLICK");
OnButtonClick((LPDATAOBJECT)arg, (long)PtrToUlong((PVOID)param));
break;
case MMCN_SELECT:
TRACEEOLID("CSnapin::ControlbarNotify - MMCN_SEL_CHANGE");
HandleToolbar(arg, param);
break;
case MMCN_HELP:
break; // New
default:
//
// Unhandled event
//
ASSERT(FALSE);
}
return hr;
}
STDMETHODIMP
CSnapin::CompareObjects(
IN LPDATAOBJECT lpDataObjectA,
IN LPDATAOBJECT lpDataObjectB
)
/*++
Routine Description:
Compare two data objects. This method is used to see if a property
sheet for the given data object is already open
Arguments:
LPDATAOBJECT lpDataObjectA : A data object
LPDATAOBJECT lpDataObjectB : B data object
Return Value:
S_OK if they match, S_FALSE otherwise
--*/
{
//
// Delegate it to the IComponentData
//
ASSERT(m_pComponentData != NULL);
return m_pComponentData->CompareObjects(lpDataObjectA, lpDataObjectB);
}
/* virtual */
HRESULT
CSnapin::Compare(
IN RDCOMPARE * prdc,
OUT int * pnResult
)
/*++
Routine Description:
Compare method
Arguments:
RDCOMPARE * prdc : Compare structure
int * pnResult : Returns result
Return Value:
HRESULT
--*/
{
if (!pnResult || !prdc || !prdc->prdch1->cookie || !prdc->prdch2->cookie)
{
ASSERT(FALSE);
return E_POINTER;
}
CIISObject * pObjectA = (CIISObject *)prdc->prdch1->cookie;
CIISObject * pObjectB = (CIISObject *)prdc->prdch2->cookie;
*pnResult = pObjectA->Compare(prdc->nColumn, pObjectB);
return S_OK;
}
//
// IPersistStream interface members
//
// <<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
STDMETHODIMP
CSnapin::GetClassID(
OUT CLSID * pClassID
)
/*++
Routine Description:
Get class ID
Arguments:
CLSID * pClassID : Returns class ID
Return Value:
HRESULT
--*/
{
ASSERT(pClassID != NULL);
//
// Copy the CLSID for this snapin
//
*pClassID = CLSID_Snapin;
return E_NOTIMPL;
}
STDMETHODIMP
CSnapin::IsDirty()
/*++
Routine Description:
Checks to see if the snapin's persistence stream is dirty
Arguments:
None
Return Value:
S_OK if the stream needs to be updated, S_FALSE otherwise
--*/
{
return m_fSettingsChanged ? S_OK : S_FALSE;
}
STDMETHODIMP
CSnapin::Load(
IN IStream * pStm
)
/*++
Routine Description:
Load the persisted information
Arguments:
IStream * pStm : Persistence stream
Return Value:
HRESULT
--*/
{
ASSERT(pStm != NULL);
//
// Verify a simple signature
//
DWORD dw = 0x1234;
DWORD dw2;
DWORD cBytesRead;
HRESULT hr = S_OK;
do
{
hr = pStm->Read(&dw2, sizeof(dw2), &cBytesRead);
if (FAILED(hr))
{
break;
}
ASSERT(cBytesRead == sizeof(dw2) && dw2 == dw);
//
// Read settings
//
hr = pStm->Read(
&m_fTaskView,
sizeof(m_fTaskView),
&cBytesRead
);
if (cBytesRead == 0)
{
//
// Old style console file
//
TRACEEOLID("Warning: old-style console file encountered");
m_fTaskView = FALSE;
}
else
{
ASSERT(cBytesRead == sizeof(m_fTaskView));
}
}
while(FALSE);
#if !TASKPADS_SUPPORTED
m_fTaskView = FALSE;
#endif // TASKPADS_SUPPORTED
return hr;
}
STDMETHODIMP
CSnapin::Save(
IN IStream * pStm,
IN BOOL fClearDirty
)
/*++
Routine Description:
Save persistence information
Arguments:
IStream * pStm : Persistence stream
BOOL fClearDirty : TRUE to clear the dirty flag
Return Value:
HRESULT
--*/
{
ASSERT(pStm != NULL);
//
// Write a simple signature
//
DWORD dw = 0x1234;
DWORD cBytesWritten;
HRESULT hr = STG_E_CANTSAVE;
do
{
hr = pStm->Write(&dw, sizeof(dw), &cBytesWritten);
if (FAILED(hr))
{
break;
}
ASSERT(cBytesWritten == sizeof(dw));
hr = pStm->Write(&m_fTaskView, sizeof(m_fTaskView), &cBytesWritten);
ASSERT(cBytesWritten == sizeof(m_fTaskView));
}
while(FALSE);
return hr;
}
STDMETHODIMP
CSnapin::GetSizeMax(
ULARGE_INTEGER * pcbSize
)
/*++
Routine Description:
Get max size of persistence information
Arguments:
ULARGE_INTEGER * pcbSize : Returns the size
Return Value:
HRESULT
--*/
{
ASSERT(pcbSize != NULL);
//
// Set the size of the string to be saved
//
pcbSize->QuadPart = (ULONGLONG)sizeof(DWORD) + sizeof(m_fTaskView);
return S_OK;
}
STDMETHODIMP
CSnapin::GetHelpTopic(LPOLESTR *lpCompiledHelpFile)
{
return GetSnapinHelpFile(lpCompiledHelpFile);
}
//
// IComponentData implementation
//
// <<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
DEBUG_DECLARE_INSTANCE_COUNTER(CComponentDataImpl);
CComponentDataImpl::CComponentDataImpl()
/*++
Routine Description:
Contructor
Arguments:
None
Return Value:
N/A
--*/
: m_pScope(NULL),
m_pConsole(NULL),
m_hIISRoot(NULL),
m_fIsCacheDirty(FALSE),
m_fIsExtension(FALSE),
m_strlCachedServers(),
m_ullDiscoveryMask((ULONGLONG)0L)
{
AFX_MANAGE_STATE(::AfxGetStaticModuleState());
DEBUG_INCREMENT_INSTANCE_COUNTER(CComponentDataImpl);
EmptyServerList();
#ifdef _DEBUG
m_cDataObjects = 0;
#endif
}
CComponentDataImpl::~CComponentDataImpl()
/*++
Routine Description:
Destructor
Arguments:
N/A
Return Value:
N/A
--*/
{
SAFE_RELEASE(m_pScope);
SAFE_RELEASE(m_pConsole);
DEBUG_DECREMENT_INSTANCE_COUNTER(CComponentDataImpl);
//
// Some snap-in is hanging on to data objects.
// If they access, it will crash!!!
//
ASSERT(m_cDataObjects == 0);
}
STDMETHODIMP
CComponentDataImpl::Initialize(
IN LPUNKNOWN pUnknown
)
/*++
Routine Description:
Initalize the component data
Arguments:
LPUNKNOWN pUnknown : Pointer to IUnknown implementation
Return Value:
HRESULT
--*/
{
AFX_MANAGE_STATE(AfxGetStaticModuleState( ));
ASSERT(pUnknown != NULL);
HRESULT hr;
//
// MMC should only call ::Initialize once!
//
ASSERT(m_pScope == NULL);
pUnknown->QueryInterface(IID_IConsoleNameSpace, (void **)&m_pScope);
CIISObject::AttachScopeView(m_pScope);
//
// add the images for the scope tree
//
LPIMAGELIST lpScopeImage;
hr = pUnknown->QueryInterface(IID_IConsole2, (void **)&m_pConsole);
ASSERT(hr == S_OK);
hr = m_pConsole->QueryScopeImageList(&lpScopeImage);
ASSERT(hr == S_OK);
//
// Load the config services DLLs
//
GetServicesDLL();
//
// Get the add-on tools
//
GetToolMenu();
//
// Get the DLLs which contain computer property pages
//
GetISMMachinePages();
CBitmap bmp16x16,
bmp32x32,
bmpMgr16x16,
bmpMgr32x32;
//
// Load the bitmaps from the dll
//
// Must AFX_MANAGE_STATE here
//
{
AFX_MANAGE_STATE(::AfxGetStaticModuleState());
VERIFY(bmp16x16.LoadBitmap(IDB_VIEWS16));
VERIFY(bmp32x32.LoadBitmap(IDB_VIEWS32));
VERIFY(bmpMgr16x16.LoadBitmap(IDB_INETMGR16));
VERIFY(bmpMgr32x32.LoadBitmap(IDB_INETMGR16));
}
//
// Set the images
//
hr = lpScopeImage->ImageListSetStrip(
(LONG_PTR *)(HBITMAP)bmp16x16,
(LONG_PTR *)(HBITMAP)bmp32x32,
0,
RGB_BK_IMAGES
);
ASSERT(hr == S_OK);
//
// Add on inetmgr bitmap
//
hr = lpScopeImage->ImageListSetStrip(
(LONG_PTR *)(HBITMAP)bmpMgr16x16,
(LONG_PTR *)(HBITMAP)bmpMgr32x32,
BMP_INETMGR,
RGB_BK_IMAGES
);
ASSERT(hr == S_OK);
//
// Add the ones from the service config DLL's
//
POSITION pos16x16 = g_obl16x16.GetHeadPosition();
POSITION pos32x32 = g_obl32x32.GetHeadPosition();
int i = BMP_SERVICE;
while (pos16x16 && pos32x32)
{
CImage * pimg16x16 = (CImage *)g_obl16x16.GetNext(pos16x16);
CImage * pimg32x32 = (CImage *)g_obl32x32.GetNext(pos32x32);
ASSERT(pimg16x16 && pimg32x32);
hr = lpScopeImage->ImageListSetStrip(
(LONG_PTR *)(HBITMAP)pimg16x16->GetBitmap(),
(LONG_PTR *)(HBITMAP)pimg32x32->GetBitmap(),
i++,
pimg16x16->GetBkColor()
);
ASSERT(hr == S_OK);
}
lpScopeImage->Release();
return hr;
}
STDMETHODIMP
CComponentDataImpl::CreateComponent(
IN LPCOMPONENT * ppComponent
)
/*++
Routine Description:
Create component
Arguments:
LPCOMPONENT * ppComponent : Address if COMPONENT
Return Value:
HRESULT
--*/
{
ASSERT(ppComponent != NULL);
CComObject<CSnapin>* pObject;
CComObject<CSnapin>::CreateInstance(&pObject);
ASSERT(pObject != NULL);
//
// Store IComponentData
//
pObject->SetIComponentData(this);
return pObject->QueryInterface(IID_IComponent, (void **)ppComponent);
}
BOOL
CComponentDataImpl::FindOpenPropSheetOnNodeAndDescendants(
IN LPPROPERTYSHEETPROVIDER piPropertySheetProvider,
IN MMC_COOKIE cookie
)
/*++
Routine Description:
Starting with the current node, check to see if an open
property sheet exists. Keep looking through descendants also.
As soon as an open property sheet is found, return TRUE.
Arguments:
MMC_COOKIE cookie : Cookie for currently selected item
Return Value:
TRUE if an open property sheet is found, FALSE if not.
Notes:
Function is called recursively
--*/
{
HSCOPEITEM hScopeItem = NULL;
CIISObject * pObject = (CIISObject *)cookie;
if (pObject != NULL)
{
hScopeItem = pObject->GetScopeHandle();
}
//
// Check the current node
//
HRESULT hr = piPropertySheetProvider->FindPropertySheet(
(MMC_COOKIE)hScopeItem,
NULL,
NULL
);
if (hr == S_OK)
{
//
// Found a sheet
//
return TRUE;
}
//
// Now check the descendants for open sheets
//
ASSERT(m_pScope != NULL);
//
// Look for machine object off the root
//
HSCOPEITEM hChildItem;
hr = m_pScope->GetChildItem(hScopeItem, &hChildItem, &cookie);
while (hr == S_OK && hChildItem != NULL)
{
CIISObject * pObject = (CIISObject *)cookie;
if (FindOpenPropSheetOnNodeAndDescendants(
piPropertySheetProvider,
cookie
))
{
//
// Found a sheet
//
return TRUE;
}
//
// Advance to next child of the same parent.
//
hr = m_pScope->GetNextItem(hChildItem, &hChildItem, &cookie);
}
//
// Not found
//
return FALSE;
}
/* INTRINSA suppress=null_pointers, uninitialized */
STDMETHODIMP
CComponentDataImpl::Notify(
IN LPDATAOBJECT lpDataObject,
IN MMC_NOTIFY_TYPE event,
IN LPARAM arg,
IN LPARAM param
)
/*++
Routine Description:
Notification message handler
Arguments:
LPDATAOBJECT lpDataObject : Selected item
MMC_NOTIFY_TYPE event : Notification message
LPARAM arg : Notification Argument
LPARAM param : Notification Parameter
Return Value:
HRESULT
--*/
{
if (event == MMCN_PROPERTY_CHANGE)
{
return OnProperties(param);
}
ASSERT(m_pScope != NULL);
HRESULT hr = S_OK;
INTERNAL * pInternal = ExtractInternalFormat(lpDataObject);
MMC_COOKIE cookie;
if (pInternal == NULL)
{
TRACEEOLID("Extension (CComponentDataImpl) Snapin");
CIISObject::m_fIsExtension = m_fIsExtension = TRUE;
cookie = NULL;
}
else
{
cookie = pInternal->m_cookie;
FREE_DATA(pInternal);
}
switch(event)
{
case MMCN_REFRESH:
RefreshIISObject((CIISObject *)cookie, TRUE, arg);
break;
case MMCN_DELETE:
DeleteObject((CIISObject *)cookie);
break;
case MMCN_RENAME:
hr = OnRename(cookie, arg, param);
break;
case MMCN_REMOVE_CHILDREN:
hr = OnRemoveChildren(arg);
break;
case MMCN_EXPAND:
hr = OnExpand(lpDataObject, arg, param);
break;
case MMCN_SELECT:
hr = OnSelect(cookie, arg, param);
break;
case MMCN_CONTEXTMENU:
hr = OnContextMenu(cookie, arg, param);
break;
default:
break;
}
return hr;
}
STDMETHODIMP
CComponentDataImpl::Destroy()
/*++
Routine Description:
Destroy component data
Arguments:
None
Return Value:
HRESULT
--*/
{
SAFE_RELEASE_SETTONULL(m_pScope);
SAFE_RELEASE_SETTONULL(m_pConsole);
m_hIISRoot = NULL;
//
// Scope items will clean themselves up
//
return S_OK;
}
STDMETHODIMP
CComponentDataImpl::QueryDataObject(
IN MMC_COOKIE cookie,
IN DATA_OBJECT_TYPES type,
OUT LPDATAOBJECT * ppDataObject
)
/*++
Routine Description:
Query data object
Arguments:
MMC_COOKIE cookie : Private data (i.e. a CIISObject *)
DATA_OBJECT_TYPES type : Data object type
LPDATAOBJECT * ppDataObject : Returns LPDATAOBJECT
Return Value:
HRESULT
--*/
{
ASSERT(ppDataObject != NULL);
CComObject<CDataObject>* pObject;
CComObject<CDataObject>::CreateInstance(&pObject);
ASSERT(pObject != NULL);
if (pObject == NULL)
return E_UNEXPECTED;
//
// Save cookie and type for delayed rendering
//
pObject->SetType(type);
pObject->SetCookie(cookie);
#ifdef _DEBUG
pObject->SetComponentData(this);
#endif
//
// Store the coclass with the data object
//
pObject->SetClsid(GetCoClassID());
return pObject->QueryInterface(IID_IDataObject, (void **)ppDataObject);
}
//
// IExtendPropertySheet Implementation
//
// <<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
STDMETHODIMP
CComponentDataImpl::CreatePropertyPages(
IN LPPROPERTYSHEETCALLBACK lpProvider,
IN LONG_PTR handle,
IN LPDATAOBJECT lpDataObject
)
/*++
Routine Description:
Create the property pages for the given object
Arguments:
LPPROPERTYSHEETCALLBACK lpProvider : Provider
LONG_PTR handle : Handle.
LPDATAOBJECT lpDataObject : Data object
Return Value:
HRESULT
--*/
{
#ifdef MMC_PAGES
CIISObject * pObject = NULL;
if (lpDataObject != NULL)
{
INTERNAL * pInternal = ExtractInternalFormat(lpDataObject);
if (pInternal)
{
pObject = (CIISObject *)pInternal->m_cookie;
FREE_DATA(pInternal);
}
}
ASSERT(pObject && pObject->IsMMCConfigurable());
CError err(pObject->ConfigureMMC(lpProvider, (LPARAM)pObject, handle));
//
// ISSUE: MMC silently fails on this error. Perhaps it should
// write the error message?
//
if (err.Failed())
{
AFX_MANAGE_STATE(AfxGetStaticModuleState());
err.MessageBox();
}
return err;
#else
return E_NOTIMPL;
#endif // MMC_PAGES
}
STDMETHODIMP
CComponentDataImpl::QueryPagesFor(
IN LPDATAOBJECT lpDataObject
)
/*++
Routine Description:
Check to see if a property sheet should be brought up for this data
object
Arguments:
LPDATAOBJECT lpDataObject : Data object
Return Value:
S_OK, if properties may be brought up for this item, S_FALSE otherwise
--*/
{
#ifdef MMC_PAGES
CIISObject * pObject = NULL;
if (lpDataObject != NULL)
{
INTERNAL * pInternal = ExtractInternalFormat(lpDataObject);
if (pInternal)
{
pObject = (CIISObject *)pInternal->m_cookie;
FREE_DATA(pInternal);
}
}
return pObject && pObject->IsMMCConfigurable() ? S_OK : S_FALSE;
#else
return S_FALSE;
#endif // MMC_PAGES
}
//
// IPersistStream interface members
//
// <<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
STDMETHODIMP
CComponentDataImpl::GetClassID(
OUT CLSID * pClassID
)
/*++
Routine Description:
Return the objects class ID
Arguments:
CLSID pClassID : Returns the class ID
Return Value:
HRESULT
--*/
{
ASSERT(pClassID != NULL);
//
// Copy the CLSID for this snapin
//
*pClassID = CLSID_Snapin;
return S_OK;
}
STDMETHODIMP
CComponentDataImpl::IsDirty()
/*++
Routine Description:
Determine if we need to save the cache
Arguments:
None
Return Value:
S_OK if the stream needs to be updated, S_FALSE otherwise
--*/
{
if (m_fIsExtension)
{
//
// Extension to computer management -- no private cache
//
return S_FALSE;
}
//
// Primary snapin
//
return IsCacheDirty() ? S_OK : S_FALSE;
}
STDMETHODIMP
CComponentDataImpl::Load(
IN IStream * pStm
)
/*++
Routine Description:
Load the persisted information
Arguments:
IStream * pStm : Persistence stream
Return Value:
HRESULT
--*/
{
ASSERT(pStm);
DWORD cch;
DWORD cBytesRead;
LPTSTR lpstr;
if (m_fIsExtension)
{
//
// Extension to computer management -- no private cache
//
return S_OK;
}
//
// First read the size of the string array
//
HRESULT hr = pStm->Read(&cch, sizeof(cch), &cBytesRead);
if (FAILED(hr))
{
return E_FAIL;
}
//
// Virgin cache
//
if (cBytesRead != sizeof(cch) || cch == 0)
{
return S_OK;
}
lpstr = AllocTString(cch);
if (lpstr == NULL)
{
return E_FAIL;
}
hr = pStm->Read(lpstr, cch * sizeof(TCHAR), &cBytesRead);
ASSERT(SUCCEEDED(hr) && cBytesRead == cch * sizeof(TCHAR));
if (FAILED(hr))
{
return E_FAIL;
}
CStringList strl;
ConvertDoubleNullListToStringList(lpstr, strl);
for (POSITION pos = strl.GetHeadPosition(); pos!= NULL; )
{
CString & str = strl.GetNext(pos);
AddServerToCache(str, FALSE);
}
FreeMem(lpstr);
return hr;
}
STDMETHODIMP
CComponentDataImpl::Save(
IN IStream * pStm,
IN BOOL fClearDirty
)
/*++
Routine Description:
Save persistence information
Arguments:
IStream * pStm : Persistence stream
BOOL fClearDirty : TRUE to clear the dirty flag
Return Value:
HRESULT
--*/
{
ASSERT(pStm);
if (m_fIsExtension)
{
//
// Extension to computer management -- no private cache
//
return S_OK;
}
//
// Flatten the cache
//
DWORD cch;
LPTSTR lpstr;
DWORD cBytesWritten;
HRESULT hr = S_OK;
DWORD err = ConvertStringListToDoubleNullList(
GetCachedServers(),
cch,
lpstr
);
if (err == ERROR_SUCCESS)
{
//
// First write the total size
//
hr = pStm->Write(&cch, sizeof(cch), &cBytesWritten);
ASSERT(SUCCEEDED(hr) && cBytesWritten == sizeof(cch));
if (FAILED(hr))
{
return STG_E_CANTSAVE;
}
//
// Now write the body.
//
hr = pStm->Write(lpstr, cch * sizeof(TCHAR), &cBytesWritten);
ASSERT(SUCCEEDED(hr) && cBytesWritten == cch * sizeof(TCHAR));
if (FAILED(hr))
{
return STG_E_CANTSAVE;
}
if (fClearDirty)
{
ClearCacheDirty();
}
FreeMem(lpstr);
}
return S_OK;
}
STDMETHODIMP
CComponentDataImpl::GetSizeMax(
OUT ULARGE_INTEGER * pcbSize
)
/*++
Routine Description:
Get max size of persistence information
Arguments:
ULARGE_INTEGER * pcbSize : Returns the size
Return Value:
HRESULT
--*/
{
ASSERT(pcbSize);
//
// Set the size of the string to be saved
//
pcbSize->QuadPart = (ULONGLONG)0;
return S_OK;
}
//
// IExtendContextMenu implementation
//
// <<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
STDMETHODIMP
CComponentDataImpl::AddMenuItems(
IN LPDATAOBJECT lpDataObject,
IN LPCONTEXTMENUCALLBACK lpContextMenuCallback,
IN long * pInsertionAllowed
)
/*++
Routine Description:
Add menu items to the right-click context menu
Arguments:
LPDATAOBJECT pDataObject : Select
LPCONTEXTMENUCALLBACK pContextMenuCallback : Context menu callback function
long * pInsertionAllowed : ???
Return Value:
HRESULT
--*/
{
if (!(*pInsertionAllowed & CCM_INSERTIONALLOWED_TOP))
{
//
// Nothing to add to the action menu.
//
return S_OK;
}
INTERNAL * pInternal = ExtractInternalFormat(lpDataObject);
if (pInternal == NULL)
{
//
// Extensions not supported yet.
//
ASSERT(FALSE);
return S_OK;
}
CIISObject * pObject = (CIISObject *)pInternal->m_cookie;
FREE_DATA(pInternal);
if (pObject == NULL)
{
//
// Must be the static root, add connect menu item only.
//
CIISObject::AddMenuItemByCommand(lpContextMenuCallback, IDM_CONNECT);
return S_OK;
}
return pObject->AddMenuItems(lpContextMenuCallback);
}
BOOL
CComponentDataImpl::DoChangeState(
IN CIISObject * pObject,
IN int nNewState
)
/*++
Routine Description:
Change the state of the selected object
Arguments:
CIISObject * pObject : Selected object
int nNewState : Desired new state
Return Value:
TRUE for success, FALSE for failure.
Notes:
In case of failure, this method will already have displayed
an error message with the cause
--*/
{
ASSERT(pObject != NULL);
ASSERT(pObject->IsControllable());
int nNumRunningChange = pObject->IsRunning()
? -1
: 0;
CError err;
//
// Temporarily override some messages
//
TEMP_ERROR_OVERRIDE(ERROR_BAD_DEV_TYPE, IDS_CLUSTER_ENABLED);
TEMP_ERROR_OVERRIDE(ERROR_INVALID_PARAMETER, IDS_CANT_START_SERVICE);
{
//
// Needed for CWaitCursor
//
AFX_MANAGE_STATE(::AfxGetStaticModuleState());
CWaitCursor wait;
err = pObject->ChangeState(nNewState);
}
if (err.Failed())
{
AFX_MANAGE_STATE(AfxGetStaticModuleState());
err.MessageBox();
}
//
// Refresh regardless
//
pObject->RefreshDisplayInfo();
nNumRunningChange += pObject->IsRunning()
? +1
: 0;
AddToNumRunning(nNumRunningChange);
//UpdateStatusBarNumbers();
return err.Succeeded();
}
void
CComponentDataImpl::OnMetaBackRest(
IN CIISObject * pObject
)
/*++
Routine Description:
Perform metabase backup/restore
Arguments:
CIISObject * pObject : Selected object (should be machine node)
Return Value:
None
--*/
{
AFX_MANAGE_STATE(AfxGetStaticModuleState());
ASSERT(pObject->QueryGUID() == cMachineNode);
CBackupDlg dlg(pObject->GetMachineName());
dlg.DoModal();
if (dlg.HasChangedMetabase())
{
RefreshIISObject(pObject, TRUE);
}
}
void
CComponentDataImpl::OnIISShutDown(
IN CIISObject * pObject
)
/*++
Routine Description:
Bring up IIS shutdown/restart dialogs
Arguments:
CIISObject * pObject : Selected object (should be machine node)
Return Value:
None
--*/
{
AFX_MANAGE_STATE(AfxGetStaticModuleState());
ASSERT(pObject->QueryGUID() == cMachineNode);
LPCTSTR lpszMachineName = pObject->GetMachineName();
//
// If a property sheet is open for this item, don't
// allow deletion.
//
LPPROPERTYSHEETPROVIDER piPropertySheetProvider = NULL;
HRESULT hr = m_pConsole->QueryInterface(
IID_IPropertySheetProvider,
(void **)&piPropertySheetProvider
);
if (FindOpenPropSheetOnNodeAndDescendants(
piPropertySheetProvider,
(ULONG_PTR)pObject
))
{
//
// Already had properties open
//
::AfxMessageBox(IDS_PROP_OPEN_SHUTDOWN);
return;
}
CIISShutdownDlg dlg(lpszMachineName);
dlg.DoModal();
if (dlg.ServicesWereRestarted())
{
//
// Rebind all metabase handles on this server
//
CServerInfo * pServerInfo;
CObListIter obli(m_oblServers);
while (pServerInfo = (CServerInfo *)obli.Next())
{
if (!::lstrcmpi(pServerInfo->QueryServerName(), lpszMachineName))
{
TRACEEOLID(
"Rebinding against "
<< pServerInfo->GetShortName()
<< " on "
<< lpszMachineName
);
pServerInfo->ISMRebind();
}
}
//
// Now do a refresh on the computer node. Since we've forced
// the rebinding here, we should not get the disconnect warning.
//
RefreshIISObject(pObject, TRUE);
}
}
void
CComponentDataImpl::OnConnectOne()
/*++
Routine Description:
Connect to a single server
Arguments:
None
Return Value:
None
--*/
{
AFX_MANAGE_STATE(AfxGetStaticModuleState());
TEMP_ERROR_OVERRIDE(EPT_S_NOT_REGISTERED, IDS_ERR_RPC_NA);
TEMP_ERROR_OVERRIDE(RPC_S_SERVER_UNAVAILABLE, IDS_ERR_RPC_NA);
TEMP_ERROR_OVERRIDE(RPC_S_UNKNOWN_IF, IDS_ERR_INTERFACE);
TEMP_ERROR_OVERRIDE(RPC_S_PROCNUM_OUT_OF_RANGE, IDS_ERR_INTERFACE);
ConnectServerDlg dlg;
if (dlg.DoModal() == IDOK)
{
//
// Clean up name.
//
CString strServerName(dlg.QueryServerName());
int cServices = 0;
//
// The function will report the errors
//
CError err;
{
CWaitCursor wait;
err = AddServerToList(
TRUE, // Cache
FALSE, // Handle errors
strServerName,
cServices
);
}
if (err.Failed())
{
err.MessageBoxFormat(
IDS_ERROR_CONNECTING,
MB_OK,
NO_HELP_CONTEXT,
(LPCTSTR)strServerName
);
}
else if (cServices == 0)
{
//
// No errors, but no services found
//
::AfxMessageBox(IDS_NO_SERVICE);
}
}
}
void
CComponentDataImpl::DoConfigure(
IN CIISObject * pObject
)
/*++
Routine Description:
Configure the given iis object
Arguments:
CIISObject * pObject : Object to be configured
Return Value:
None
--*/
{
CError err;
//
// We need a window handle for this, but MMC won't
// give us one. Fortunately, we know MMC to be
// an MFC app, so we can sleazily grab it anyway.
//
CWnd * pMainWnd = NULL;
{
HWND hwnd;
err = m_pConsole->GetMainWindow(&hwnd);
if (err.Succeeded())
{
pMainWnd = CWnd::FromHandle(hwnd);
}
}
ASSERT(pMainWnd != NULL);
CWnd wnd;
if (pMainWnd == NULL)
{
//
// No main window in MMC? Use NULL handle
//
pMainWnd = &wnd;
}
ASSERT(m_pConsole);
ASSERT(pObject->IsConfigurable());
err = pObject->Configure(pMainWnd);
AFX_MANAGE_STATE(::AfxGetStaticModuleState());
if (!err.MessageBoxOnFailure())
{
//
// Refresh and display the entry
//
OnProperties((LPARAM)pObject);
}
}
BOOL
CComponentDataImpl::DeleteObject(
IN CIISObject * pObject
)
/*++
Routine Description:
Delete the given iisobject. Ask for confirmation first.
Arguments:
CIISObject * pObject : Object to be deleted
Return Value:
TRUE if the item was deleted successfully,
FALSE otherwise.
--*/
{
AFX_MANAGE_STATE(::AfxGetStaticModuleState());
BOOL fReturn = FALSE;
//
// If a property sheet is open for this item, don't
// allow deletion.
//
LPPROPERTYSHEETPROVIDER piPropertySheetProvider = NULL;
HRESULT hr = m_pConsole->QueryInterface(
IID_IPropertySheetProvider,
(void **)&piPropertySheetProvider
);
if (FindOpenPropSheetOnNodeAndDescendants(
piPropertySheetProvider,
(ULONG_PTR)pObject
))
{
//
// Already had properties open
//
::AfxMessageBox(IDS_PROP_OPEN);
}
else
{
CError err;
// Before we go and delete this baby,
// let's check if it's got a Cluster property set.
if (pObject->IsClusterEnabled())
{
::AfxMessageBox(IDS_CLUSTER_ENABLED_2);
}
else
{
if (!pObject->HandleUI() || NoYesMessageBox(IDS_CONFIRM_DELETE))
{
CWaitCursor wait;
err = pObject->Delete();
if (pObject->HandleUI())
{
//
// Only complain if we're to handle the error messages.
// In e.g. the file system, explorer handles all error
// messages
//
err.MessageBoxOnFailure();
}
if (err.Succeeded())
{
ASSERT(m_pScope);
//
// Delete the item from the view, but be careful that
// result item nodes store a scope handle, but which
// actually refers to the _parent_'s scope handle.
//
if (!pObject->ScopeHandleIsParent())
{
m_pScope->DeleteItem(pObject->GetScopeHandle(), TRUE);
delete pObject;
}
fReturn = TRUE;
}
}
}
}
piPropertySheetProvider->Release();
return fReturn;
}
void
CComponentDataImpl::DisconnectItem(
IN CIISObject * pObject
)
/*++
Routine Description:
Delete the given iisobject. Ask for confirmation first.
Arguments:
CIISObject * pObject : Object to be deleted
Return Value:
None
--*/
{
AFX_MANAGE_STATE(::AfxGetStaticModuleState());
//
// If a property sheet is open for this item, don't
// allow deletion.
//
LPPROPERTYSHEETPROVIDER piPropertySheetProvider = NULL;
HRESULT hr = m_pConsole->QueryInterface(
IID_IPropertySheetProvider,
(void **)&piPropertySheetProvider
);
if (FindOpenPropSheetOnNodeAndDescendants(
piPropertySheetProvider,
(ULONG_PTR)pObject
))
{
//
// Already had properties open
//
::AfxMessageBox(IDS_PROP_OPEN);
}
else
{
CString strMachine(pObject->GetMachineName());
CString str, str2;
VERIFY(str.LoadString(IDS_CONFIRM_DISCONNECT));
str2.Format(str, strMachine);
if (NoYesMessageBox(str2))
{
//
// Remove from Cache and oblist
//
CError err(RemoveServerFromList(TRUE, strMachine));
if (!err.MessageBoxOnFailure())
{
ASSERT(m_pScope);
m_pScope->DeleteItem(pObject->GetScopeHandle(), TRUE);
delete pObject;
}
}
}
piPropertySheetProvider->Release();
}
STDMETHODIMP
CComponentDataImpl::Command(
IN long nCommandID,
IN LPDATAOBJECT lpDataObject
)
/*++
Routine Description:
Command handler
Arguments:
long nCommandID : Command ID
LPDATAOBJECT lpDataObject : Selected Data object
Return Value:
HRESULT
--*/
{
if (nCommandID == IDM_CONNECT)
{
//
// This is the only case that doesn't require a selected
// data object.
//
OnConnectOne();
return S_OK;
}
/*
if (IsMMCMultiSelectDataObject(lpDataObject))
{
//
// Do something for multi-select?
//
TRACEEOLID("Multiple selection");
}
*/
INTERNAL * pInternal = lpDataObject
? ExtractInternalFormat(lpDataObject)
: NULL;
if (pInternal == NULL)
{
return S_OK;
}
CIISObject * pObject = (CIISObject *)pInternal->m_cookie;
FREE_DATA(pInternal);
CError err;
switch (nCommandID)
{
case IDM_METABACKREST:
OnMetaBackRest(pObject);
break;
case IDM_SHUTDOWN:
OnIISShutDown(pObject);
break;
case IDM_DISCONNECT:
DisconnectItem(pObject);
break;
case IDM_CONFIGURE:
DoConfigure(pObject);
break;
case IDM_STOP:
DoChangeState(pObject, INetServiceStopped);
m_pConsole->UpdateAllViews(lpDataObject, 0L, (LONG_PTR)pObject);
break;
case IDM_START:
DoChangeState(pObject, INetServiceRunning);
m_pConsole->UpdateAllViews(lpDataObject, 0L, (LONG_PTR)pObject);
break;
case IDM_PAUSE:
DoChangeState(
pObject,
pObject->IsPaused() ? INetServiceRunning : INetServicePaused
);
m_pConsole->UpdateAllViews(lpDataObject, 0L, (LONG_PTR)pObject);
break;
case IDM_EXPLORE:
pObject->Explore();
break;
case IDM_OPEN:
pObject->Open();
break;
case IDM_BROWSE:
pObject->Browse();
break;
case IDM_TASK_SECURITY_WIZARD:
{
//
// Launch the security wizard
//
err = pObject->SecurityWizard();
err.MessageBoxOnFailure();
}
break;
case IDM_NEW_INSTANCE:
{
//
// Executed from another instance of the same type
//
ISMINSTANCEINFO ii;
CServerInfo * pServerInfo = pObject->GetServerInfo();
ASSERT(pServerInfo);
err = pServerInfo->AddInstance(&ii, sizeof(ii));
if (err.Succeeded())
{
CIISInstance * pInstance = new CIISInstance(&ii, pServerInfo);
//
// Add the new instance grouped with the service type,
// and select it
//
BOOL fNext;
HSCOPEITEM hParent = FindServerInfoParent(
GetRootHandle(),
pServerInfo
);
ASSERT(hParent != NULL);
HSCOPEITEM hSibling = FindNextInstanceSibling(
hParent,
pInstance,
&fNext
);
HSCOPEITEM hItem = AddIISObject(
hParent,
pInstance,
hSibling,
fNext
);
//
// hItem could return NULL if the hParent
// was not yet expanded.
//
if (hItem)
{
m_pConsole->SelectScopeItem(hItem);
}
}
}
break;
case IDM_NEW_VROOT:
CIISChildNode * pChild;
err = pObject->AddChildNode(pChild);
if (err.Succeeded())
{
//
// Insert prior to file/dir nodes, and select the item
//
HSCOPEITEM hItem = AddIISObject(
pObject->GetScopeHandle(),
pChild,
FindNextVDirSibling(pObject->GetScopeHandle(), pChild)
);
//
// hItem could return NULL if the parent object
// was not yet expanded.
//
if (hItem)
{
m_pConsole->SelectScopeItem(hItem);
}
}
break;
default:
//
// Now try to get to the service that we're
// supposed to create a new instance for.
//
if (nCommandID >= IDM_NEW_EX_INSTANCE)
{
int nID = nCommandID - IDM_NEW_EX_INSTANCE;
CNewInstanceCmd * pcmd =
(CNewInstanceCmd *)m_oblNewInstanceCmds.Index(nID);
ASSERT(pcmd != NULL);
if (pcmd)
{
ISMINSTANCEINFO ii;
CServerInfo * pServerInfo = FindServerInfo(
pObject->GetMachineName(),
pcmd->GetServiceInfo()
);
if (pServerInfo == NULL)
{
AFX_MANAGE_STATE(::AfxGetStaticModuleState());
::AfxMessageBox(IDS_ERR_SERVICE_NOT_INSTALLED);
break;
}
ASSERT(pServerInfo);
err = pServerInfo->AddInstance(&ii, sizeof(ii));
if (err.Succeeded())
{
//
// Add and select the item
//
CIISInstance * pInstance = new CIISInstance(
&ii,
pServerInfo
);
BOOL fNext;
HSCOPEITEM hSibling = FindNextInstanceSibling(
pObject->GetScopeHandle(),
pInstance,
&fNext
);
HSCOPEITEM hItem = AddIISObject(
pObject->GetScopeHandle(),
pInstance,
hSibling,
fNext
);
m_pConsole->SelectScopeItem(hItem);
}
}
}
else
{
//
// Unknown command!
//
ASSERT(FALSE);
}
break;
}
return S_OK;
}
//
// Notification handlers for IComponentData
//
// <<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
HRESULT
CComponentDataImpl::OnAdd(
IN MMC_COOKIE cookie,
IN LPARAM arg,
IN LPARAM param
)
/*++
Routine Description:
Add handler
Arguments:
MMC_COOKIE cookie : Scope item cookie (CIISObject *)
LPARAM arg : Argument
LPARAM param : Parameter
Return Value:
HRESULT
--*/
{
return E_UNEXPECTED;
}
HRESULT
CComponentDataImpl::OnDelete(
IN MMC_COOKIE cookie
)
/*++
Routine Description:
Handle deletion of the given scope item (CIISObject *)
Arguments:
MMC_COOKIE cookie : Casts to a CIISObject *
Return Value:
HRESULT
--*/
{
CIISObject * pObject = (CIISObject *)cookie;
if (pObject == NULL)
{
return S_FALSE;
}
delete pObject;
return S_OK;
}
HRESULT
CComponentDataImpl::OnRename(
IN MMC_COOKIE cookie,
IN LPARAM arg,
IN LPARAM param
)
/*++
Routine Description:
Rename notification handler
Arguments:
MMC_COOKIE cookie : Currently selected cookie (CIISObject *)
LPARAM arg : Notification argument
LPARAM param : Notification parameter
Return Value:
HRESULT
--*/
{
CIISObject * pObject = (CIISObject *)cookie;
if (pObject == NULL)
{
//
// Can't rename this one
//
return S_FALSE;
}
if (!arg)
{
//
// Check to see if we're renamable
//
return pObject->IsRenamable() ? S_OK : S_FALSE;
}
TEMP_ERROR_OVERRIDE(ERROR_ALREADY_EXISTS, IDS_ERR_DUP_VROOT);
CError err;
//
// Do an actual rename
//
LPCTSTR lpstrNewName = (LPCTSTR)param;
err = pObject->Rename(lpstrNewName);
if (err.Succeeded())
{
//
// Force a refresh on the children
//
pObject->DirtyChildren();
}
else
{
if (pObject->HandleUI())
{
AFX_MANAGE_STATE(::AfxGetStaticModuleState());
err.MessageBox();
}
}
return err.Succeeded() ? S_OK : S_FALSE;
}
HRESULT
CComponentDataImpl::OnRemoveChildren(
IN LPARAM arg
)
/*++
Routine Description:
'Remove Children' notification handler
Arguments:
LPARAM arg : Notification argument
Return Value:
HRESULT
--*/
{
m_strlCachedServers.RemoveAll();
return S_OK;
}
HRESULT
CComponentDataImpl::OnExpand(
LPDATAOBJECT lpDataObject,
LPARAM arg,
LPARAM param
)
/*++
Routine Description:
Expand notification handler
Arguments:
LPDATAOBJECT lpDataObject : Currently selected cookie (CIISObject *)
LPARAM arg : Notification argument
LPARAM param : Notification parameter
Return Value:
HRESULT
--*/
{
if (arg)
{
//
// Did Initialize get called?
//
ASSERT(m_pScope != NULL);
EnumerateScopePane(lpDataObject, param);
}
return S_OK;
}
HRESULT
CComponentDataImpl::OnSelect(
IN MMC_COOKIE cookie,
IN LPARAM arg,
IN LPARAM param
)
/*++
Routine Description:
Selection notification handler
Arguments:
MMC_COOKIE cookie : Currently selected cookie (CIISObject *)
LPARAM arg : Notification argument
LPARAM param : Notification parameter
Return Value:
HRESULT
--*/
{
return E_UNEXPECTED;
}
HRESULT
CComponentDataImpl::OnContextMenu(
IN MMC_COOKIE cookie,
IN LPARAM arg,
IN LPARAM param
)
/*++
Routine Description:
Context Menu notification handler
Arguments:
MMC_COOKIE cookie : Currently selected cookie (CIISObject *)
LPARAM arg : Notification argument
LPARAM param : Notification parameter
Return Value:
HRESULT
--*/
{
return S_OK;
}
HRESULT
CComponentDataImpl::OnProperties(
IN LPARAM param
)
/*++
Routine Description:
Properties change notification handler
Arguments:
LPARAM arg : Notification argument (CIISObject *)
Return Value:
HRESULT
--*/
{
if (param == NULL)
{
return S_OK;
}
CIISObject * pObject = (CIISObject *)param;
ASSERT(pObject != NULL);
if (pObject != NULL)
{
CString strOldPath(pObject->QueryPhysicalPath());
CString strOldRedirect(pObject->QueryRedirPath());
BOOL fOldChildRedirOnly = pObject->IsChildOnlyRedir();
//
// Refresh the data to see if either the physical or redirection
// path has changed. If it has, refresh the child objects
//
CError err(pObject->RefreshData());
//
// Determine if the file system needs to be refreshed
//
BOOL fRefreshFileSystem = err.Succeeded()
&& pObject->SupportsChildren()
&& pObject->ChildrenExpanded()
&& pObject->SupportsFileSystem()
&& strOldPath.CompareNoCase(pObject->QueryPhysicalPath()) != 0;
//
// Determine if everything needs to be refreshed
//
BOOL fFullRefresh = err.Succeeded()
&& pObject->SupportsChildren()
&& pObject->ChildrenExpanded()
&& (strOldRedirect.CompareNoCase(pObject->QueryRedirPath()) != 0
|| fOldChildRedirOnly != pObject->IsChildOnlyRedir());
TRACEEOLID("Refresh files: "
<< fRefreshFileSystem
<< " Full Refresh: "
<< fFullRefresh
);
RefreshIISObject(pObject, fFullRefresh);
if (!fFullRefresh && fRefreshFileSystem)
{
//
// Not a full refresh -- the file system only.
//
CString strPhysicalPath, strMetaRoot;
pObject->BuildPhysicalPath(strPhysicalPath);
pObject->BuildFullPath(strMetaRoot, FALSE);
//
// Note: we can't use ExpandIISObject, because we're only
// replacing specific nodes, not the entire subtree
//
AddFileSystem(
pObject->GetScopeHandle(),
strPhysicalPath,
strMetaRoot,
pObject->FindOwnerInstance(),
GET_DIRECTORIES,
DELETE_CURRENT_DIR_TREE
);
//
// Mark this node to indicate that the child nodes
// have been added.
//
pObject->CleanChildren();
}
//
// Re-enumerate the result side if the current item is
// selected.
//
if ((fRefreshFileSystem || fFullRefresh) && pObject->IsScopeSelected())
{
ASSERT(m_pConsole);
m_pConsole->SelectScopeItem(pObject->GetScopeHandle());
}
}
return S_OK;
}
CServerInfo *
CComponentDataImpl::FindServerInfo(
IN LPCTSTR lpstrMachine,
IN CServiceInfo * pServiceInfo
)
/*++
Routine Description:
Find specific server info (i.e. machine name / service info
combination type)
Arguments:
LPCTSTR lpstrMachine : Machine name
CServiceInfo * pServiceInfo : Service info
Return Value:
Server Info object pointer, or NULL
--*/
{
CServerInfo * pServerInfo;
CObListIter obli(m_oblServers);
//
// Search is sequential, because we don't foresee more then a few
// of these objects in the cache
//
while (pServerInfo = (CServerInfo *)obli.Next())
{
if (!::lstrcmpi(pServerInfo->QueryServerName(), lpstrMachine)
&& pServerInfo->GetServiceInfo() == pServiceInfo)
{
//
// Found it
//
return pServerInfo;
}
}
return NULL;
}
void
CComponentDataImpl::RefreshIISObject(
IN CIISObject * pObject,
IN BOOL fExpandTree,
IN HSCOPEITEM pParent
)
/*++
Routine Description:
Refresh object, and optionally re-enumerate its display
Arguments:
CIISObject * pObject : Object to be refreshed
BOOL fExpandTree : TRUE to expand its tree
Return Value:
None
--*/
{
ASSERT(pObject != NULL);
CError err;
CServerInfo * pServer = pObject->GetServerInfo();
CMetaInterface * pInterface = NULL;
if (pServer)
{
pInterface = GetMetaKeyFromHandle(pServer->GetHandle());
}
if (pInterface)
{
BEGIN_ASSURE_BINDING_SECTION
err = pObject->RefreshData();
END_ASSURE_BINDING_SECTION(err, pInterface, RPC_S_SERVER_UNAVAILABLE);
}
else
{
err = pObject->RefreshData();
}
if (err.Failed())
{
AFX_MANAGE_STATE(::AfxGetStaticModuleState());
err.MessageBox();
return;
}
pObject->RefreshDisplayInfo();
//
// Reenumerate its children if requested to do so and its necessary
//
if (fExpandTree)
{
if (pObject->ChildrenExpanded())
{
HSCOPEITEM hNode = pObject->GetScopeHandle();
if (KillChildren(
hNode,
IDS_PROP_OPEN_REFRESH,
DELETE_EVERYTHING,
DONT_CONTINUE_ON_OPEN_SHEET
))
{
pObject->DirtyChildren();
ExpandIISObject(hNode, pObject);
}
}
//
// Re-enumerate the result side, if the current item is
// selected.
//
if (pObject->IsScopeSelected())
{
ASSERT(m_pConsole);
m_pConsole->SelectScopeItem(pObject->GetScopeHandle());
}
}
}
void
CComponentDataImpl::LoadDynamicExtensions(
IN HSCOPEITEM hParent,
IN CIISObject * pObject
)
/*++
Routine Description:
Load dynamic snap-in extensions for the given node type.
Arguments:
None
Return Value:
None (if it fails, it always fails silently)
--*/
{
//
// We'll look here (on the server) to see if we need to load
// extension snap-ins.
//
const TCHAR SERVICES_KEY[] = SZ_REMOTEIISEXT;
if (pObject == NULL)
{
//
// This should never happen, right?
//
ASSERT(FALSE);
return;
}
CString str, strKey;
//
// Get base path for the given node type
//
strKey.Format(
_T("%s\\%s"),
SERVICES_KEY,
GUIDToCString(pObject->QueryGUID(), str)
);
TRACEEOLID(
pObject->GetMachineName() <<
" Attempting to dynamically load extensions for " <<
strKey);
CError err;
CRMCRegKey rkExtensions(REG_KEY, strKey, KEY_READ, pObject->GetMachineName());
CComQIPtr<IConsoleNameSpace2, &IID_IConsoleNameSpace2> pIConsoleNameSpace2
= m_pConsole;
if (pIConsoleNameSpace2 && rkExtensions.Ok())
{
DWORD dwType;
GUID guidExtension;
CRMCRegValueIter rkValues(rkExtensions);
//
// Loop through, and attempt to add each extension found in the
// registry
//
while (rkValues.Next(&str, &dwType) == ERROR_SUCCESS)
{
TRACEEOLID("Found dynamic extension: " << str);
err = ::CLSIDFromString((LPTSTR)(LPCTSTR)str, &guidExtension);
if (err.Succeeded())
{
err = pIConsoleNameSpace2->AddExtension(hParent, &guidExtension);
}
TRACEEOLID("DynaLoad returned " << err);
}
}
}
void
CComponentDataImpl::ExpandIISObject(
IN HSCOPEITEM hParent,
IN CIISObject * pObject,
IN LPTSTR lpszMachineName
)
/*++
Routine Description:
Expand the given IIS object tree
Arguments:
HSCOPEITEM hParent : Handle to parent item
CIISObject * pObject : IISObject to expand
Return Value:
None
--*/
{
CError err;
if (pObject != NULL)
{
ASSERT(hParent == pObject->GetScopeHandle());
}
//
// make sure we QI'ed for the interface
//
ASSERT(m_pScope != NULL);
AFX_MANAGE_STATE(::AfxGetStaticModuleState());
{
CWaitCursor wait;
if (m_hIISRoot == NULL)
{
//
// Store the IIS root node for future use
//
m_hIISRoot = hParent;
}
if (pObject == NULL)
{
//
// Static root node -- populate with computers.
//
// This is done only once per session
//
ASSERT(GetRootHandle() == hParent);
TRACEEOLID("Inserting static root node");
AddCachedServersToView();
CServerInfo * pServerInfo = NULL;
CObListIter obli(m_oblServers);
if (m_fIsExtension)
{
//
// Since we're extending the computer management
// snap-in, only add the one server info parent
// object (only one computer, after all).
// This is also necessary, because for some reason
// I can't get the child item nodes of my parent,
// even if I've added them myself, and so we'd get
// duplicates.
//
if (lpszMachineName != NULL && *lpszMachineName != 0)
{
CServerInfo * p;
while (p = (CServerInfo *)obli.Next())
{
if (!::lstrcmpi(p->QueryServerName(), lpszMachineName))
{
pServerInfo = p;
break;
}
}
}
else
pServerInfo = (CServerInfo *)obli.Next();
ASSERT(pServerInfo != NULL);
if (pServerInfo)
{
AddServerInfoParent(hParent, pServerInfo);
}
}
else
{
//
// We're the primary snap-in, add all the server info
// parent objects (computers) to the view
//
while (pServerInfo = (CServerInfo *)obli.Next())
{
if (pServerInfo->IsServiceSelected())
{
//
// Add each item in the tree
//
AddServerInfoParent(hParent, pServerInfo);
}
}
}
}
else
{
if (!pObject->ChildrenExpanded())
{
//
// Delete whatever children there may be
//
if (pObject->QueryGUID() == cMachineNode)
{
CIISMachine * pMachine = (CIISMachine *)pObject;
CServerInfo * pServerInfo;
CObListIter obli(m_oblServers);
while (pServerInfo = (CServerInfo *)obli.Next())
{
if (pServerInfo->MatchServerName(pMachine->GetMachineName())
&& pServerInfo->IsServiceSelected())
{
//
// Add each item in the tree
//
AddServerInfo(hParent, pServerInfo, FALSE);
}
}
}
CString strMetaPath;
if (pObject->SupportsChildren())
{
pObject->BuildFullPath(strMetaPath, FALSE);
//
// Expand the children off the root
//
AddVirtualRoots(
hParent,
strMetaPath,
pObject->FindOwnerInstance()
);
}
if (pObject->SupportsFileSystem())
{
if (strMetaPath.IsEmpty())
{
pObject->BuildFullPath(strMetaPath, FALSE);
}
//
// Expand file system objects
//
CString strPhysicalPath;
pObject->BuildPhysicalPath(strPhysicalPath);
AddFileSystem(
hParent,
strPhysicalPath,
strMetaPath,
pObject->FindOwnerInstance(),
GET_DIRECTORIES,
DONT_DELETE_CURRENT_DIR_TREE
);
}
//
// Now load the dynamic extensions
//
LoadDynamicExtensions(hParent, pObject);
//
// Mark this node to indicate that the child nodes
// have been added.
//
pObject->CleanChildren();
}
}
}
err.MessageBoxOnFailure();
}
void
CComponentDataImpl::EnumerateScopePane(
IN LPDATAOBJECT lpDataObject,
IN HSCOPEITEM hParent
)
/*++
Routine Description:
Handle expansion of hParent scope item.
Arguments:
LPDATAOBJECT lpDataObject : Selected data object
HSCOPEITEM hParent : Scope handle of parent item or NULL
Return Value:
None
--*/
{
ASSERT(lpDataObject != NULL);
INTERNAL * pInternal = ExtractInternalFormat(lpDataObject);
MMC_COOKIE cookie = 0L;
LPTSTR lpszMachine = NULL;
if (pInternal == NULL)
{
//
// Not mine -- must be an extension; Get the machine name
//
ASSERT(m_fIsExtension);
lpszMachine = ExtractMachineName(lpDataObject);
TRACEEOLID(lpszMachine);
CString strServerName = PURE_COMPUTER_NAME(lpszMachine);
if (strServerName.IsEmpty())
{
//
// MyComputer reports "" for the computer name.
// This means the local machine is indicated
//
DWORD dwSize = MAX_SERVERNAME_LEN;
if (::GetComputerName(strServerName.GetBuffer(dwSize + 1), &dwSize))
{
strServerName.ReleaseBuffer();
}
}
//
// Since we're an extension, the cache will not be
// loaded from the persistence stream, and we can
// therefore guarantee that this will be the only
// item in the cache.
//
AddServerToCache(strServerName, FALSE);
}
else
{
cookie = pInternal->m_cookie;
FREE_DATA(pInternal);
}
#ifdef _DEBUG
CIISObject * pObject = (CIISObject *)cookie;
if (pObject)
{
ASSERT(hParent == pObject->GetScopeHandle());
}
#endif // _DEBUG
ExpandIISObject(hParent, (CIISObject *)cookie, lpszMachine);
}
STDMETHODIMP
CComponentDataImpl::GetDisplayInfo(
IN LPSCOPEDATAITEM lpScopeDataItem
)
/*++
Routine Description:
Get display info (text, bitmap) for the selected scope item
Arguments:
LPSCOPEDATAITEM lpScopeDataItem : Selected item
Return Value:
HRESULT
--*/
{
ASSERT(lpScopeDataItem != NULL);
if (lpScopeDataItem == NULL)
{
return E_POINTER;
}
static CString strText;
CIISObject * pObject = (CIISObject *)lpScopeDataItem->lParam;
ASSERT(lpScopeDataItem->mask & SDI_STR);
pObject->GetDisplayText(strText);
lpScopeDataItem->displayname = (LPTSTR)(LPCTSTR)strText;
ASSERT(lpScopeDataItem->displayname != NULL);
return S_OK;
}
STDMETHODIMP
CComponentDataImpl::CompareObjects(
IN LPDATAOBJECT lpDataObjectA,
IN LPDATAOBJECT lpDataObjectB
)
/*++
Routine Description:
Compare two data objects. This is used by MMC to determine whether a
given node has a property sheet already open for it.
Arguments:
LPDATAOBJECT lpDataObjectA : Data object 1
LPDATAOBJECT lpDataObjectB : Data object 2
Return Value:
S_OK if the objects are identical, S_FALSE otherwise
--*/
{
if (lpDataObjectA == NULL || lpDataObjectB == NULL)
{
return E_POINTER;
}
//
// Make sure both data object are mine
//
INTERNAL * pA;
INTERNAL * pB;
HRESULT hr = S_FALSE;
pA = ExtractInternalFormat(lpDataObjectA);
pB = ExtractInternalFormat(lpDataObjectA);
if (pA != NULL && pB != NULL)
{
//hr = (*pA == *pB) ? S_OK : S_FALSE;
return S_OK;
}
FREE_DATA(pA);
FREE_DATA(pB);
return hr;
}
void
CComponentDataImpl::GetISMMachinePages()
/*++
Routine Description:
Load the names of the DLL providing ISM machine property
page extentions
Arguments:
Return Value:
--*/
{
/* OBSOLETE
OBSOLETE
OBSOLETE
OBSOLETE
OBSOLETE
CString strValueName;
DWORD dwValueType;
CRMCRegKey rkMachine(REG_KEY, SZ_ADDONMACHINEPAGES, KEY_READ);
if (rkMachine.Ok())
{
return;
}
CRMCRegValueIter rvi(rkMachine);
CIISMachine::AttachPages(&m_oblISMMachinePages);
try
{
while (rvi.Next(&strValueName, &dwValueType) == ERROR_SUCCESS)
{
CString strValue;
rkMachine.QueryValue(strValueName, strValue);
TRACEEOLID("Registering machine pages in " << strValue);
m_oblISMMachinePages.AddTail(new CISMMachinePageExt(strValue));
}
}
catch(CException * e)
{
TRACEEOLID("!!!exception building ISM machine page list");
e->ReportError();
e->Delete();
}
*/
}
void
CComponentDataImpl::ConvertBitmapFormats(
IN CBitmap & bmpSource,
OUT CBitmap & bmp16x16,
OUT CBitmap & bmp32x32
)
/*++
Routine Description:
Convert a ISM service config bitmap to 16x16 and 32x32 exactly.
Downlevel services only included a single sized bitmap (usually
16x16) for use in ISM. We need a small one and a large one,
so we expand as needed.
Arguments:
IN CBitmap & bmpSource : Source bitmap
OUT CBitmap & bmp16x16 : 16x16 output bitmap
OUT CBitmap & bmp32x32 : 32x32 output bitmap
Return Value:
None
--*/
{
CDC dcImage;
CDC dc16x16;
CDC dc32x32;
VERIFY(dcImage.CreateCompatibleDC(NULL));
VERIFY(dc16x16.CreateCompatibleDC(NULL));
VERIFY(dc32x32.CreateCompatibleDC(NULL));
CBitmap * pOld = dcImage.SelectObject(&bmpSource);
BITMAP bm;
VERIFY(bmpSource.GetObject(sizeof(bm), &bm));
VERIFY(bmp16x16.CreateBitmap(16, 16, bm.bmPlanes, bm.bmBitsPixel, NULL));
VERIFY(bmp32x32.CreateBitmap(32, 32, bm.bmPlanes, bm.bmBitsPixel, NULL));
CBitmap * pOld16x16 = dc16x16.SelectObject(&bmp16x16);
CBitmap * pOld32x32 = dc32x32.SelectObject(&bmp32x32);
dc16x16.StretchBlt(0, 0, 16, 16, &dcImage, 0, 0, bm.bmWidth, bm.bmHeight, SRCCOPY);
dc32x32.StretchBlt(0, 0, 32, 32, &dcImage, 0, 0, bm.bmWidth, bm.bmHeight, SRCCOPY);
dc16x16.SelectObject(pOld16x16);
dc32x32.SelectObject(pOld32x32);
dcImage.SelectObject(pOld);
}
BOOL
CComponentDataImpl::VerifyBitmapSize(
IN HBITMAP hBitmap,
IN LONG nHeight,
IN LONG nWidth
)
/*++
Routine Description:
Verify the given bitmap is of the right size
Arguments:
HBITMAP hBitmap : Bitmap handle
LONG nHeight : Height the bitmap should be
LONG nWidth : Width the bitmap should be
Return Value:
TRUE if the bitmap is of the right size, FALSE otherwise
--*/
{
BITMAP bm;
int cb = GetObject(hBitmap, sizeof(BITMAP), &bm);
return (cb == sizeof(BITMAP)
&& bm.bmWidth == nWidth
&& bm.bmHeight == nHeight
);
}
BOOL
CComponentDataImpl::GetBitmapParms(
IN CServiceInfo * pServiceInfo,
IN BMP_TYPES bmpt,
OUT CBitmap *& pbmp16x16,
OUT CBitmap *& pbmp32x32,
OUT COLORREF & rgbMask
)
/*++
Routine Description:
Get bitmap information from service info object
Arguments:
CServiceInfo * pServiceInfo : Service info
BMP_TYPES bmpt : Type of info requested
COLORREF & rgbMask : Returns background mask
Return Value:
None.
--*/
{
ASSERT(pServiceInfo != NULL);
UINT nID16x16 = 0;
UINT nID32x32 = 0;
HINSTANCE hMod = NULL;
if (pServiceInfo->InitializedOK())
{
switch(bmpt)
{
case BMT_BUTTON:
nID16x16 = pServiceInfo->QueryButtonBitmapID();
nID32x32 = 0;
rgbMask = pServiceInfo->QueryButtonBkMask();
break;
case BMT_SERVICE:
nID16x16 = pServiceInfo->QueryServiceBitmapID();
nID32x32 = pServiceInfo->QueryLargeServiceBitmapID();
rgbMask = pServiceInfo->QueryServiceBkMask();
break;
case BMT_VROOT:
ASSERT(pServiceInfo->SupportsChildren());
nID16x16 = pServiceInfo->QueryChildBitmapID();
nID32x32 = pServiceInfo->QueryLargeChildBitmapID();
rgbMask = pServiceInfo->QueryChildBkMask();
break;
}
if (nID16x16 != 0)
{
hMod = pServiceInfo->QueryInstanceHandle();
}
else
{
//
// No bitmap provided by the service DLL, provide one from our
// own resource segment.
//
nID16x16 = IDB_UNKNOWN;
nID32x32 = 0;
hMod = ::AfxGetResourceHandle();
rgbMask = TB_COLORMASK;
}
}
else
{
//
// Add a disabled dummy button for a service
// that didn't load.
//
nID16x16 = IDB_NOTLOADED;
nID32x32 = 0;
hMod = ::AfxGetResourceHandle();
rgbMask = TB_COLORMASK;
}
if (nID16x16 == 0)
{
return FALSE;
}
pbmp16x16 = new CBitmap;
pbmp32x32 = new CBitmap;
if (pbmp16x16 == NULL || pbmp32x32 == NULL)
{
return FALSE;
}
HINSTANCE hOld = ::AfxGetResourceHandle();
::AfxSetResourceHandle(hMod);
if (nID32x32 != 0)
{
//
// Have explicit large and small images
//
VERIFY(pbmp16x16->LoadBitmap(nID16x16));
VERIFY(pbmp32x32->LoadBitmap(nID32x32));
//
// Check to make sure they're the right size
//
if (!VerifyBitmapSize((HBITMAP)*pbmp16x16, 16, 16) ||
!VerifyBitmapSize((HBITMAP)*pbmp32x32, 32, 32))
{
ASSERT(FALSE);
TRACEEOLID("Bogus bitmap size provided by service bitmap");
//
// Synthesize based on small image.
//
delete pbmp32x32;
CBitmap * pTmp = pbmp16x16;
ConvertBitmapFormats(*pTmp, *pbmp16x16, *pbmp32x32);
delete pTmp;
}
}
else
{
//
// Only have one image. Synthesize small and large from this
// image.
//
CBitmap bmp;
VERIFY(bmp.LoadBitmap(nID16x16));
//
// Convert to 16x16 and 32x32 image
//
ConvertBitmapFormats(bmp, *pbmp16x16, *pbmp32x32);
}
::AfxSetResourceHandle(hOld);
return TRUE;
}
void
CComponentDataImpl::GetToolMenu()
/*++
Routine Description:
Load the add-on tools
Arguments:
None.
Return Value:
None.
--*/
{
CString strValueName;
DWORD dwValueType;
int cTools = 0;
CRMCRegKey rkMachine(REG_KEY, SZ_ADDONTOOLS, KEY_READ);
if (!rkMachine.Ok())
{
//
// No registry entry
//
return;
}
CRMCRegValueIter rvi(rkMachine);
static BOOL fInitialised = FALSE;
if (fInitialised)
{
TRACEEOLID("Toolbar already initialised");
return;
}
try
{
int nButton = IDM_TOOLBAR;
while (rvi.Next(&strValueName, &dwValueType) == ERROR_SUCCESS)
{
CString strValue;
BOOL fExpanded;
rkMachine.QueryValue(
strValueName,
strValue,
EXPANSION_ON,
&fExpanded
);
TRACEEOLID("Adding tool: " << strValueName);
TRACEEOLID("From Path: " << strValue);
TRACEEOLID("Expansion: " << fExpanded);
CISMShellExecutable * pNewAddOnTool = new CISMShellExecutable(
strValue,
nButton - 1,
nButton
);
if (!pNewAddOnTool->HasBitmap())
{
TRACEEOLID("Tossing useless toolbar item");
delete pNewAddOnTool;
continue;
}
g_oblAddOnTools.AddTail(pNewAddOnTool);
++nButton;
}
}
catch(CException * e)
{
TRACEEOLID("!!!exception building tool menu");
e->ReportError();
e->Delete();
}
fInitialised = TRUE;
}
void
CComponentDataImpl::MatchupSuperDLLs()
/*++
Routine Description:
Match up all dlls with superceed dlls
Arguments:
None
Return Value:
None
--*/
{
POSITION pos = m_oblServices.GetHeadPosition();
while(pos)
{
CServiceInfo * pService = (CServiceInfo *)m_oblServices.GetNext(pos);
ASSERT(pService != NULL);
if (pService->RequiresSuperDll())
{
//
// Match up the super DLL
//
POSITION pos2 = m_oblServices.GetHeadPosition();
while (pos2)
{
CServiceInfo * pService2 =
(CServiceInfo *)m_oblServices.GetNext(pos2);
if (pService2->IsSuperDllFor(pService))
{
pService->AssignSuperDll(pService2);
break;
}
}
ASSERT(pService->HasSuperDll());
}
}
}
void
CComponentDataImpl::GetServicesDLL()
/*++
Routine Description:
Load the add-on services.
Arguments:
None
Return Value:
None
--*/
{
int cServices = 0;
HRESULT hr = S_OK;
LPIMAGELIST lpScopeImage = NULL;
LPIMAGELIST lpResultImage = NULL;
ASSERT(m_pConsole);
hr = m_pConsole->QueryScopeImageList(&lpScopeImage);
ASSERT(SUCCEEDED(hr));
hr = m_pConsole->QueryResultImageList(&lpResultImage);
ASSERT(SUCCEEDED(hr));
//
// Run through the list of installed services,
// load its associated cfg dll, and build up
// a discovery mask for each service.
//
CString strValueName;
DWORD dwValueType;
CRMCRegKey rkMachine(REG_KEY, SZ_ADDONSERVICES, KEY_READ);
if (rkMachine.Ok())
{
CRMCRegValueIter rvi(rkMachine);
CIISMachine::AttachNewInstanceCmds(&m_oblNewInstanceCmds);
try
{
//
// Now load the services
//
// AFX_MANAGE_STATE required to load service bitmaps
// from the dlls
//
AFX_MANAGE_STATE(::AfxGetStaticModuleState());
int nBmpIndex = BMP_SERVICE;
while (rvi.Next(&strValueName, &dwValueType) == ERROR_SUCCESS)
{
//
// Expand environment variables in path if present.
//
CString strValue;
BOOL fExpanded;
rkMachine.QueryValue(
strValueName,
strValue,
EXPANSION_ON,
&fExpanded
);
CServiceInfo * pServiceInfo = NULL;
{
CWaitCursor wait;
TRACEEOLID("Adding service DLL: " << strValue);
pServiceInfo = new CServiceInfo(cServices, strValue);
}
CError err(pServiceInfo->QueryReturnCode());
if (err.Failed())
{
if (err.Win32Error() == ERROR_INVALID_PARAMETER)
{
//
// The ERROR_INVALID_PARAMETER error return code
// gets sent when the info buffer provided is too
// small for the configuration DLL
//
::AfxMessageBox(
IDS_VERSION_INCOMPATIBLE,
MB_OK | MB_ICONEXCLAMATION
);
}
else
{
err.MessageBoxFormat(
IDS_ERR_NO_LOAD,
MB_OK | MB_ICONEXCLAMATION,
NO_HELP_CONTEXT,
(LPCTSTR)pServiceInfo->QueryDllName()
);
}
//
// Don't add it to the list
//
delete pServiceInfo;
continue;
}
AddServiceToList(pServiceInfo);
//
// If this service use inetsloc discovery,
// add it to the mask.
//
if (pServiceInfo->UseInetSlocDiscover())
{
m_ullDiscoveryMask |= pServiceInfo->QueryDiscoveryMask();
}
//
// Add a bitmap representing the service
// to the image list
//
CBitmap * pbmp16x16 = NULL;
CBitmap * pbmp32x32 = NULL;
COLORREF rgbMask;
if (GetBitmapParms(
pServiceInfo,
BMT_SERVICE,
pbmp16x16,
pbmp32x32,
rgbMask
))
{
g_obl16x16.AddTail(new CImage(pbmp16x16, rgbMask));
g_obl32x32.AddTail(new CImage(pbmp32x32, rgbMask));
pServiceInfo->SetBitmapIndex(nBmpIndex++);
}
//
// Add to the 'new instance' menu commands
//
if (pServiceInfo->SupportsInstances())
{
//
// Add to new instances menu object
//
m_oblNewInstanceCmds.AddTail(new CNewInstanceCmd(pServiceInfo));
}
if (pServiceInfo->SupportsChildren())
{
if (GetBitmapParms(
pServiceInfo,
BMT_VROOT,
pbmp16x16,
pbmp32x32,
rgbMask
))
{
g_obl16x16.AddTail(new CImage(pbmp16x16, rgbMask));
g_obl32x32.AddTail(new CImage(pbmp32x32, rgbMask));
pServiceInfo->SetChildBitmapIndex(nBmpIndex++);
}
}
++cServices;
}
MatchupSuperDLLs();
}
catch(CException * e)
{
TRACEEOLID("Exception loading library");
e->ReportError();
e->Delete();
}
}
if (cServices == 0)
{
//
// No services installed
//
CString str;
VERIFY(str.LoadString(IDS_NO_SERVICES_INSTALLED));
AfxMessageBox(str);
}
}
BOOL
CComponentDataImpl::RemoveServerFromCache(
IN LPCTSTR lpstrServer
)
/*++
Routine Description:
Remove machine from cache.
Arguments:
LPCTSTR lpstrServer : computer name to be removed from cache
Return Value:
None
--*/
{
CStringList & strList = GetCachedServers();
TRACEEOLID("Removing " << lpstrServer << " from cache");
POSITION posOld;
POSITION pos = strList.GetHeadPosition();
int nResult;
while(pos)
{
posOld = pos;
CString & str = strList.GetNext(pos);
nResult = str.CompareNoCase(lpstrServer);
if (nResult == 0)
{
strList.RemoveAt(posOld);
SetCacheDirty(TRUE);
return TRUE;
}
if (nResult > 0)
{
//
// We're not going to find it.
//
break;
}
}
//
// Didn't exist
//
ASSERT(FALSE && "Attempting to remove non-existent server from cache");
return FALSE;
}
void
CComponentDataImpl::AddServerToCache(
IN LPCTSTR lpstrServer,
IN BOOL fSetCacheDirty OPTIONAL
)
/*++
Routine Description:
Add machine name to the cache
Arguments:
LPCTSTR lpstrServer : computer name to be added to the cache.
BOOL fSetCacheDirty : TRUE to dirty the cache
Return Value:
None
--*/
{
CStringList & strList = GetCachedServers();
TRACEEOLID("Adding " << lpstrServer << " to cache");
CString strServer(lpstrServer);
POSITION posOld;
POSITION pos = strList.GetHeadPosition();
int nResult;
while(pos)
{
posOld = pos;
CString & str = strList.GetNext(pos);
nResult = str.CompareNoCase(strServer);
if (nResult == 0)
{
//
// Already existed in the case, ignore
//
return;
}
else if (nResult > 0)
{
//
// Found the proper place
//
strList.InsertBefore(posOld, strServer);
if (fSetCacheDirty)
{
SetCacheDirty();
}
return;
}
}
//
// Didn't exist yet, so add to list here
//
strList.AddTail(strServer);
if (fSetCacheDirty)
{
SetCacheDirty();
}
}
static BOOL
GetCommandLineServer(LPTSTR * pStr)
{
BOOL bRes = FALSE;
LPTSTR pCmdLine = GetCommandLine();
int n;
LPTSTR * pArgv = CommandLineToArgvW(pCmdLine, &n);
*pStr = NULL;
if (pArgv != NULL)
{
TCHAR szCmd[] = _T("/SERVER:");
int len = sizeof(szCmd) / sizeof(TCHAR) - 1;
for (int i = 0; i < n; i++)
{
if (CSTR_EQUAL == CompareString(LOCALE_SYSTEM_DEFAULT,
NORM_IGNORECASE, pArgv[i], len, szCmd, len))
{
LPTSTR p = pArgv[i] + len;
int count = 0;
while (*p != _T(' ') && *p != 0)
{
p++;
count++;
}
*pStr = (LPTSTR)LocalAlloc(LPTR, sizeof(TCHAR) * (count + 1));
if (*pStr != NULL)
{
lstrcpyn(*pStr, pArgv[i] + len, count + 1);
bRes = TRUE;
}
break;
}
}
GlobalFree(pArgv);
}
return bRes;
}
void
CComponentDataImpl::AddCachedServersToView()
/*++
Routine Description:
Move the cached servers to the scope view.
Arguments:
None
Return Value:
None
--*/
{
CStringList & strlList = GetCachedServers();
BOOL bCmdLine = FALSE;
LPTSTR pCmdLine = NULL;
if (GetCommandLineServer(&pCmdLine))
{
bCmdLine = TRUE;
EmptyServerList();
AddServerToCache(pCmdLine, FALSE);
if (pCmdLine != NULL)
LocalFree(pCmdLine);
}
else if (strlList.IsEmpty())
{
//
// Nothing pre-selected or cached.
// Add the local machine.
//
CString str;
DWORD dwSize = MAX_SERVERNAME_LEN;
if (::GetComputerName(str.GetBuffer(dwSize + 1), &dwSize))
{
//
// Add local machine, though don't persist this later
//
str.ReleaseBuffer();
AddServerToCache(str, FALSE);
}
}
//
// Now add everything cached to the current
// view
//
CError err;
for(POSITION pos = strlList.GetHeadPosition(); pos != NULL; )
{
CString & strMachine = strlList.GetNext(pos);
int cServices;
err = AddServerToList(
FALSE,
bCmdLine ? TRUE : FALSE,
strMachine,
cServices,
m_oblServices
);
TRACEEOLID("adding " << strMachine << " to the view returned error code " << err);
if (err.Failed())
{
//
// Temporarily map RPC errors to friendlier error message
//
TEMP_ERROR_OVERRIDE(EPT_S_NOT_REGISTERED, IDS_ERR_RPC_NA);
TEMP_ERROR_OVERRIDE(RPC_S_SERVER_UNAVAILABLE, IDS_ERR_RPC_NA);
TEMP_ERROR_OVERRIDE(RPC_S_UNKNOWN_IF, IDS_ERR_INTERFACE);
TEMP_ERROR_OVERRIDE(RPC_S_PROCNUM_OUT_OF_RANGE, IDS_ERR_INTERFACE);
AFX_MANAGE_STATE(::AfxGetStaticModuleState());
//
// Give the option of removing from cache
//
if (err.MessageBoxFormat(
IDS_ERROR_CONNECTING_CACHE,
MB_YESNO | MB_DEFBUTTON1,
NO_HELP_CONTEXT,
(LPCTSTR)strMachine) != IDYES)
{
VERIFY(RemoveServerFromCache(strMachine));
}
}
else if (cServices == 0)
{
//
// No errors, but no services found
//
::AfxMessageBox(IDS_NO_SERVICE);
}
}
}
CServiceInfo *
CComponentDataImpl::GetServiceAt(
IN int nIndex
)
/*++
Routine Description:
Get the service object at the given index.
Arguments:
int nIndex : Index where to look for the service info object
Return Value:
Service info pointer, or NULL if the index was not valid
--*/
{
if (nIndex < 0 || nIndex >= m_oblServices.GetCount())
{
TRACEEOLID("Invalid service index requested");
return NULL;
}
return (CServiceInfo *)m_oblServices.GetAt(m_oblServices.FindIndex(nIndex));
}
void
CComponentDataImpl::EmptyServerList()
/*++
Routine Description:
Empty server list
Arguments:
None
Return Value:
None
--*/
{
m_oblServers.RemoveAll();
m_cServers = m_cServicesRunning = 0;
}
DWORD
CComponentDataImpl::AddServerToList(
IN BOOL fCache,
IN LPINET_SERVER_INFO lpServerInfo,
IN OUT CObListPlus & oblServices
)
/*++
Routine Description:
Add a service object for each service discovered
to be belonging to this server.
Arguments:
BOOL fCache : TRUE to cache
LPINET_SERVER_INFO lpServerInfo : Discovery information (from inetsloc)
CObListPlus & oblServices : List of installed services
Return Value:
Error return code
--*/
{
TRACEEOLID("For Server " << lpServerInfo->ServerName);
CServerInfo * pServerInfo;
DWORD err = ERROR_SUCCESS;
for (DWORD j = 0; j < lpServerInfo->Services.NumServices; ++j)
{
LPINET_SERVICE_INFO lpServiceInfo = lpServerInfo->Services.Services[j];
try
{
//
// Attempt to create a server info block
//
pServerInfo = new CServerInfo(
lpServerInfo->ServerName,
lpServiceInfo,
oblServices
);
if (pServerInfo->IsConfigurable())
{
TRACEEOLID("Adding " << (DWORD)lpServiceInfo->ServiceMask);
if (!AddToList(fCache, pServerInfo))
{
TRACEEOLID("It already existed in the list");
delete pServerInfo;
}
}
else
{
//
// Toss it
//
TRACEEOLID("Tossing " << (DWORD)lpServiceInfo->ServiceMask);
delete pServerInfo;
}
}
catch(CMemoryException * e)
{
TRACEEOLID("AddServerList: memory exception");
err = ERROR_NOT_ENOUGH_MEMORY;
e->Delete();
}
}
return err;
}
DWORD
CComponentDataImpl::AddServerToList(
IN BOOL fCache,
IN BOOL fHandleErrors,
IN CString & strServerName,
OUT int & cServices,
IN OUT CObListPlus & oblServices
)
/*++
Routine Description:
Add a service object for each service running
on the machine listed above.
Arguments:
BOOL fCache : TRUE to cache the server
BOOL fHandleErrors : TRUE to display error messages, FALSE to abort
on error
CString & strServerName : Name of this server
int & cServices : # Services added
CObListPlus & oblServices : List of installed services
Return Value:
Error return code
--*/
{
TEMP_ERROR_OVERRIDE(EPT_S_NOT_REGISTERED, IDS_ERR_RPC_NA);
TEMP_ERROR_OVERRIDE(RPC_S_SERVER_UNAVAILABLE, IDS_ERR_RPC_NA);
TEMP_ERROR_OVERRIDE(RPC_S_UNKNOWN_IF, IDS_ERR_INTERFACE);
TEMP_ERROR_OVERRIDE(RPC_S_PROCNUM_OUT_OF_RANGE, IDS_ERR_INTERFACE);
//
// Loop through the services, and find out which ones
// are installed on the target machine, if any.
//
CObListIter obli(oblServices);
CServiceInfo * psi;
cServices = 0;
CError err;
ISMSERVERINFO ServerInfo;
CServerInfo::CleanServerName(strServerName);
//
// See if we can make contact with the machine
//
if (!DoesServerExist(strServerName))
{
//
// No, quit right here
//
err = RPC_S_SERVER_UNAVAILABLE;
if (fHandleErrors)
{
err.MessageBox();
}
return err.Win32Error();
}
while (psi = (CServiceInfo *)obli.Next())
{
int cErrors = 0;
if (psi->InitializedOK())
{
TRACEEOLID("Trying: " << psi->GetShortName());
ServerInfo.dwSize = sizeof(ServerInfo);
{
AFX_MANAGE_STATE(::AfxGetStaticModuleState());
CWaitCursor wait;
err = psi->ISMQueryServerInfo(
strServerName,
&ServerInfo
);
}
if (err.Win32Error() == ERROR_SERVICE_DOES_NOT_EXIST)
{
TRACEEOLID("Service not installed -- acceptable response");
err.Reset();
}
else if (err.Win32Error() == ERROR_SERVICE_START_HANG)
{
TRACEEOLID("Service is hanging -- ignore silently");
err.Reset();
}
else if (err.Succeeded())
{
//
// Yes, this service is running on this
// machine.
//
++cServices;
//
// Add to list
//
try
{
CServerInfo * pNewServer = new CServerInfo(
strServerName,
&ServerInfo,
psi
);
if (!AddToList(fCache, pNewServer, TRUE))
{
TRACEEOLID("It already existed in the list");
delete pNewServer;
}
}
catch(CMemoryException * e)
{
TRACEEOLID("AddServerList: memory exception");
err = ERROR_NOT_ENOUGH_MEMORY;
e->Delete();
}
}
if (err.Failed())
{
if (!fHandleErrors)
{
//
// Let the calling process handle the errors,
// we're stopping.
//
break;
}
++cErrors;
//
// Display error about the service
//
AFX_MANAGE_STATE(::AfxGetStaticModuleState());
err.MessageBoxFormat(
IDS_ERR_ENUMERATE_SVC,
MB_OK,
NO_HELP_CONTEXT,
(LPCTSTR)psi->GetShortName(),
(LPCTSTR)strServerName
);
//
// Optionally cancel here on no response.
//
//break;
}
}
}
return err.Win32Error();
}
DWORD
CComponentDataImpl::RemoveServerFromList(
IN BOOL fCache,
IN CString & strServerName
)
/*++
Routine Description:
Remove each service in the list belonging to the given computer name.
Arguments:
CString & strServerName : Name of this server
Return Value:
Error return code
--*/
{
CServerInfo::CleanServerName(strServerName);
CServerInfo * pEntry;
POSITION pos1, pos2;
pos1 = m_oblServers.GetHeadPosition();
while(pos1)
{
pos2 = pos1;
pEntry = (CServerInfo *)m_oblServers.GetNext(pos1);
if (pEntry->MatchServerName(strServerName))
{
m_oblServers.RemoveAt(pos2);
}
}
if (fCache)
{
RemoveServerFromCache(strServerName);
}
return ERROR_SUCCESS;
}
void
CComponentDataImpl::Refresh()
/*++
Routine Description:
Refresh the server list
Arguments:
None
Return Value:
None
--*/
{
//
// AFX_MANAGE_STATE required for wait cursor
//
AFX_MANAGE_STATE(::AfxGetStaticModuleState());
POSITION pos;
CWaitCursor wait;
CServerInfo * pEntry;
for(pos = m_oblServers.GetHeadPosition(); pos != NULL; /**/ )
{
pEntry = (CServerInfo *)m_oblServers.GetNext(pos);
int oldState = pEntry->QueryServiceState();
if (pEntry->Refresh() == ERROR_SUCCESS)
{
if (oldState != pEntry->QueryServiceState())
{
//
// Take away this service from the total
// running count if it was part of it
// before, and (re-) add it if it's currently
// running.
//
if (oldState == INetServiceRunning)
{
--m_cServicesRunning;
}
if (pEntry->IsServiceRunning())
{
++m_cServicesRunning;
}
}
//UpdateAllViews(NULL, HINT_REFRESHITEM, pEntry);
}
}
}
BOOL
CComponentDataImpl::AddToList(
IN BOOL fCache,
IN CServerInfo * pServerInfo,
IN BOOL fSelect
)
/*++
Routine Description:
Add the service to the list if it didn't exist already,
otherwise refresh the info if it did exist. Return
TRUE if the service was added, FALSE, if already
existed and was refreshed.
Arguments:
CServerInfo * pServerInfo : Server to add
BOOL fSelect : If TRUE, select the newly added item
Return Value:
TRUE if added, FALSE otherwise
--*/
{
POSITION pos;
BOOL fFoundService = FALSE;
BOOL fFoundServer = FALSE;
BOOL fServiceAdded = FALSE;
//int nAddHint = fSelect ? HINT_ADDITEM_SELECT : HINT_ADDITEM;
CServerInfo * pEntry;
for( pos = m_oblServers.GetHeadPosition();
pos != NULL;
m_oblServers.GetNext(pos)
)
{
pEntry = (CServerInfo *)m_oblServers.GetAt(pos);
if (pEntry->CompareByServer(pServerInfo) == 0)
{
fFoundServer = TRUE;
//
// Found the server, also the service?
//
if (pEntry->CompareByService(pServerInfo) == 0)
{
//
// Yes, the service was found also -- update the information
// if the service state has changed.
//
TRACEEOLID("Entry Already Existed");
fFoundService = TRUE;
if (pEntry->QueryServiceState()
!= pServerInfo->QueryServiceState())
{
TRACEEOLID("Service State has changed -- refreshing");
//
// Decrement the running counter if this service
// was part of that counter. The counter will be
// re-added if the service is still running
//
if (pEntry->IsServiceRunning())
{
--m_cServicesRunning;
}
*pEntry = *pServerInfo;
if (pEntry->IsServiceRunning())
{
++m_cServicesRunning;
}
//UpdateAllViews(NULL, HINT_REFRESHITEM, pEntry);
}
break;
}
}
else
{
if (fFoundServer)
{
//
// Server name no longer matches, but did match
// the last time, so we didn't find the service.
// Insert it at the end of the services belonging
// to this guy.
//
TRACEEOLID("Found new service belonging to known server");
m_oblServers.InsertBefore(pos, pServerInfo);
fServiceAdded = TRUE; // Don't add again.
if (pServerInfo->IsServiceRunning())
{
++m_cServicesRunning;
}
//UpdateAllViews(NULL, nAddHint, pServerInfo);
break;
}
}
}
if (!fFoundService && !fServiceAdded)
{
//
// Came to the end of the list without
// finding the service -- add a new one
// at the end.
//
TRACEEOLID("Adding new entry to tail");
m_oblServers.AddTail(pServerInfo);
if (pServerInfo->IsServiceRunning())
{
++m_cServicesRunning;
}
if (!fFoundServer)
{
++m_cServers;
if (fCache)
{
AddServerToCache(pServerInfo->QueryServerName(), TRUE);
}
}
if (fCache && GetRootHandle() != NULL)
{
//
// View has already been expanded -- add it here.
//
if (pServerInfo->IsServiceSelected())
{
AddServerInfoParent(GetRootHandle(), pServerInfo);
}
}
}
return !fFoundService;
}
static HRESULT
GetSnapinHelpFile(LPOLESTR * lpCompiledHelpFile)
{
if (lpCompiledHelpFile == NULL)
return E_INVALIDARG;
CString strFilePath, strWindowsPath, strBuffer;
AFX_MANAGE_STATE(AfxGetStaticModuleState());
// Use system API to get windows directory.
UINT uiResult = GetWindowsDirectory(strWindowsPath.GetBuffer(MAX_PATH),
MAX_PATH);
strWindowsPath.ReleaseBuffer();
if (uiResult <= 0 || uiResult > MAX_PATH)
{
return E_FAIL;
}
if (!strFilePath.LoadString(IDS_HELPFILE))
{
return E_FAIL;
}
strBuffer = strWindowsPath;
strBuffer += _T('\\');
strBuffer += strFilePath;
*lpCompiledHelpFile
= reinterpret_cast<LPOLESTR>(CoTaskMemAlloc((strBuffer.GetLength() + 1)
* sizeof(_TCHAR)));
if (*lpCompiledHelpFile == NULL)
return E_OUTOFMEMORY;
USES_CONVERSION;
_tcscpy(*lpCompiledHelpFile, T2OLE((LPTSTR)(LPCTSTR)strBuffer));
return S_OK;
}
STDMETHODIMP
CComponentDataImpl::GetHelpTopic(LPOLESTR *lpCompiledHelpFile)
{
return GetSnapinHelpFile(lpCompiledHelpFile);
}
//
// About implementation
//
// <<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
DEBUG_DECLARE_INSTANCE_COUNTER(CSnapinAboutImpl);
CSnapinAboutImpl::CSnapinAboutImpl()
/*++
Routine Description:
Constructor
Arguments:
None
Return Value:
N/A
--*/
{
}
CSnapinAboutImpl::~CSnapinAboutImpl()
/*++
Routine Description:
Destructor
Arguments:
N/A
Return Value:
N/A
--*/
{
DEBUG_DECREMENT_INSTANCE_COUNTER(CSnapinAboutImpl);
}
HRESULT
CSnapinAboutImpl::AboutHelper(
IN UINT nID,
OUT LPOLESTR * lpPtr
)
/*++
Routine Description:
"About" helper function
Arguments:
UINT nID : String resource ID
LPOLESTR * lpPtr : Return buffer for the string
Return Value:
HRESULT
--*/
{
if (lpPtr == NULL)
{
return E_POINTER;
}
CString s;
VERIFY(::LoadString(
_Module.GetModuleInstance(),
nID,
s.GetBuffer(255),
255));
s.ReleaseBuffer();
*lpPtr = (LPOLESTR)CoTaskMemAlloc((s.GetLength() + 1) * sizeof(wchar_t));
if (*lpPtr == NULL)
{
return E_OUTOFMEMORY;
}
lstrcpy(*lpPtr, (LPCTSTR)s);
return S_OK;
}
STDMETHODIMP
CSnapinAboutImpl::GetSnapinDescription(
OUT LPOLESTR * lpDescription
)
/*++
Routine Description:
Get the snapin description
Arguments:
LPOLESTR * lpDescription : String return buffer
Return Value:
HRESULT
--*/
{
return AboutHelper(IDS_SNAPIN_DESC, lpDescription);
}
STDMETHODIMP
CSnapinAboutImpl::GetProvider(
OUT LPOLESTR * lpName
)
/*++
Routine Description:
Get the snapin provider string
Arguments:
LPOLESTR * lpName : String return buffer
Return Value:
HRESULT
--*/
{
return AboutHelper(IDS_COMPANY, lpName);
}
STDMETHODIMP
CSnapinAboutImpl::GetSnapinVersion(
IN LPOLESTR * lpVersion
)
/*++
Routine Description:
Get the snapin version string:
Arguments:
LPOLESTR * lpVersion : Version string
Return Value:
HRESULT
--*/
{
return AboutHelper(IDS_VERSION, lpVersion);
}
STDMETHODIMP
CSnapinAboutImpl::GetSnapinImage(
IN HICON * hAppIcon
)
/*++
Routine Description:
Get the icon for this snapin
Arguments:
HICON * hAppIcon : Return handle to the icon
Return Value:
HRESULT
--*/
{
if (hAppIcon == NULL)
{
return E_POINTER;
}
*hAppIcon = LoadIcon(
_Module.GetModuleInstance(),
MAKEINTRESOURCE(IDI_ICON));
ASSERT(*hAppIcon != NULL);
return (*hAppIcon != NULL) ? S_OK : E_FAIL;
}
STDMETHODIMP
CSnapinAboutImpl::GetStaticFolderImage(
OUT HBITMAP * phSmallImage,
OUT HBITMAP * phSmallImageOpen,
OUT HBITMAP * phLargeImage,
OUT COLORREF * prgbMask
)
/*++
Routine Description:
Get the static folder image
Arguments:
HBITMAP * phSmallImage : Small folder
HBITMAP * phSmallImageOpen : Small open folder
HBITMAP * phLargeImage : Large image
COLORREF * prgbMask : Mask
Return Value:
HRESULT
--*/
{
if (!phSmallImage || !phSmallImageOpen || !phLargeImage || !prgbMask)
{
return HRESULT_FROM_WIN32(ERROR_INVALID_PARAMETER);
}
*phSmallImage = (HBITMAP)LoadImage(
_Module.GetModuleInstance(),
MAKEINTRESOURCE(IDB_INETMGR16),
IMAGE_BITMAP,
0,
0,
LR_DEFAULTCOLOR
);
*phSmallImageOpen = (HBITMAP)LoadImage(
_Module.GetModuleInstance(),
MAKEINTRESOURCE(IDB_INETMGR16),
IMAGE_BITMAP,
0,
0,
LR_DEFAULTCOLOR
);
*phLargeImage = (HBITMAP)LoadImage(
_Module.GetModuleInstance(),
MAKEINTRESOURCE(IDB_INETMGR32),
IMAGE_BITMAP,
0,
0,
LR_DEFAULTCOLOR
);
*prgbMask = RGB_BK_IMAGES;
return *phSmallImage && *phLargeImage ? S_OK : E_FAIL;
}
Reference in a new issue View git blame Copy permalink