|
|
/////////////////////////////////////////////////////////////////////////////
//
// Copyright (c) 1996-1997 Microsoft Corporation
//
// Module Name:
// IISClEx4.cpp
//
// Abstract:
// Implementation of the CIISCluExApp class and DLL initialization
// routines.
//
// Author:
// David Potter (davidp) June 28, 1996
//
// Revision History:
//
// Notes:
//
/////////////////////////////////////////////////////////////////////////////
#include "stdafx.h"
#include "resource.h"
#include <initguid.h>
#include <CluAdmEx.h>
#include "IISClEx4.h"
#include "ExtObj.h"
#include "RegExt.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;#endif
#define IID_DEFINED
#include "ExtObjID_i.c"
CComModule _Module;
#include <atlimpl.cpp>
BEGIN_OBJECT_MAP(ObjectMap) OBJECT_ENTRY(CLSID_CoIISClEx4, CExtObject)END_OBJECT_MAP()
/////////////////////////////////////////////////////////////////////////////
// Global Function Prototypes
/////////////////////////////////////////////////////////////////////////////
STDAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID * ppv);STDAPI DllCanUnloadNow(void);STDAPI DllRegisterServer(void);STDAPI DllUnregisterServer(void);HRESULT HrDeleteKey(IN const CString & rstrKey);STDAPI DllGetCluAdminExtensionCaps( OUT DWORD * pdwCaps, OUT CLSID * pclsid, OUT LPWSTR pwszResTypeNames, IN OUT DWORD * pcchResTypeNames );
/////////////////////////////////////////////////////////////////////////////
// class CIISCluExApp
/////////////////////////////////////////////////////////////////////////////
class CIISCluExApp : public CWinApp{public: virtual BOOL InitInstance(); virtual int ExitInstance();
private: TCHAR szHelpPath[MAX_PATH+1];};
CIISCluExApp theApp;
BOOL CIISCluExApp::InitInstance(){ _Module.Init(ObjectMap, m_hInstance);
//
// Setup the help file
//
if (GetWindowsDirectory(szHelpPath, MAX_PATH)) { lstrcat(szHelpPath,_T("\\help\\iishelp\\iis\\winhelp\\iisclex4.hlp")); m_pszHelpFilePath = szHelpPath; } return CWinApp::InitInstance();}
int CIISCluExApp::ExitInstance(){ m_pszHelpFilePath = NULL; _Module.Term(); return CWinApp::ExitInstance();}
/////////////////////////////////////////////////////////////////////////////
// Used to determine whether the DLL can be unloaded by OLE
STDAPI DllCanUnloadNow(void){ AFX_MANAGE_STATE(AfxGetStaticModuleState()); return (AfxDllCanUnloadNow() && _Module.GetLockCount()==0) ? S_OK : S_FALSE;}
/////////////////////////////////////////////////////////////////////////////
// Returns a class factory to create an object of the requested type
STDAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID* ppv){ return _Module.GetClassObject(rclsid, riid, ppv);}
/////////////////////////////////////////////////////////////////////////////
// DllRegisterServer - Adds entries to the system registry
STDAPI DllRegisterServer(void){ HRESULT hRes = S_OK; // registers object, typelib and all interfaces in typelib
hRes = _Module.RegisterServer(FALSE /*bRegTypeLib*/); return hRes;}
/////////////////////////////////////////////////////////////////////////////
// DllUnregisterServer - Adds entries to the system registry
STDAPI DllUnregisterServer(void){ HRESULT hRes = S_OK; _Module.UnregisterServer(); return hRes;}
/////////////////////////////////////////////////////////////////////////////
//++
//
// FormatError
//
// Routine Description:
// Format an error.
//
// Arguments:
// rstrError [OUT] String in which to return the error message.
// dwError [IN] Error code to format.
//
// Return Value:
// None.
//
//--
/////////////////////////////////////////////////////////////////////////////
void FormatError(CString & rstrError, DWORD dwError){ DWORD dwResult; TCHAR szError[256];
dwResult = ::FormatMessage( FORMAT_MESSAGE_FROM_SYSTEM, NULL, dwError, MAKELANGID(LANG_NEUTRAL, SUBLANG_NEUTRAL), szError, sizeof(szError) / sizeof(TCHAR), 0 ); if (dwResult == 0) { // Format the NT status code from NTDLL since this hasn't been
// integrated into the system yet.
dwResult = ::FormatMessage( FORMAT_MESSAGE_FROM_HMODULE | FORMAT_MESSAGE_IGNORE_INSERTS, ::GetModuleHandle(_T("NTDLL.DLL")), dwError, MAKELANGID(LANG_NEUTRAL, SUBLANG_NEUTRAL), szError, sizeof(szError) / sizeof(TCHAR), 0 ); if (dwResult == 0) { // Format the NT status code from CLUSAPI. This is necessary
// for the cases where cluster messages haven't been added to
// the system message file yet.
dwResult = ::FormatMessage( FORMAT_MESSAGE_FROM_HMODULE | FORMAT_MESSAGE_IGNORE_INSERTS, ::GetModuleHandle(_T("CLUSAPI.DLL")), dwError, MAKELANGID(LANG_NEUTRAL, SUBLANG_NEUTRAL), szError, sizeof(szError) / sizeof(TCHAR), 0 ); } // if: error formatting status code from NTDLL
} // if: error formatting status code from system
if (dwResult != 0) rstrError = szError; else { TRACE(_T("FormatError() - Error 0x%08.8x formatting string for error code 0x%08.8x\n"), dwResult, dwError); rstrError.Format(_T("Error 0x%08.8x")); } // else: error formatting the message
} //*** FormatError()
/////////////////////////////////////////////////////////////////////////////
//++
//
// DllGetCluAdminExtensionCaps
//
// Routine Description:
// Returns the CLSID supported by this extension.
//
// Arguments:
// pdwCaps [OUT] DWORD in which to return extension capabilities.
// pclsid [OUT] Place in which to return the CLSID.
// pwszResTypeNames [OUT] Buffer in which to return the resource type
// names supported by this extension. Each name is
// null-terminated. Two nulls follow the last name.
// pcchResTypeNames [IN OUT] On input, contains the number of characters
// available in the output buffer, including the
// null-terminators. On output, contains the
// total number of characters written, not
// including the null-terminator.
//
// Return Value:
// S_OK Capabilities returned successfully.
// E_INVALIDARG Invalid argument specified.
//
//--
/////////////////////////////////////////////////////////////////////////////
STDAPI DllGetCluAdminExtensionCaps( OUT DWORD * pdwCaps, OUT CLSID * pclsid, OUT LPWSTR pwszResTypeNames, IN OUT DWORD * pcchResTypeNames ){ AFX_MANAGE_STATE(AfxGetStaticModuleState());
// Validate arguments.
if ((pdwCaps == NULL) || (pclsid == NULL) || (pwszResTypeNames == NULL) || (pcchResTypeNames == NULL) || (*pcchResTypeNames < 1) ) return E_INVALIDARG;
// Set capabilities flags.
*pdwCaps = 0 //| CLUADMEX_CAPS_RESOURCE_PAGES
;
// Copy the CLSID to the caller's buffer.
CopyMemory(pclsid, &CLSID_CoIISClEx4, sizeof(CLSID));
// Return the resource type names we support.
{ DWORD cchCopy = min(g_cchResourceTypeNames, *pcchResTypeNames); CopyMemory(pwszResTypeNames, g_wszResourceTypeNames, cchCopy * sizeof(WCHAR)); *pcchResTypeNames = cchCopy; } // Return he resource type names we support
return S_OK;
} //*** DllGetCluAdminExtensionCaps()
/////////////////////////////////////////////////////////////////////////////
//++
//
// DllRegisterCluAdminExtension
//
// Routine Description:
// Register the extension with the cluster database.
//
// Arguments:
// hCluster [IN] Handle to the cluster to modify.
//
// Return Value:
// S_OK Extension registered successfully.
// Win32 error code if another failure occurred.
//
//--
/////////////////////////////////////////////////////////////////////////////
STDAPI DllRegisterCluAdminExtension(IN HCLUSTER hCluster){ HRESULT hr = S_OK; LPCWSTR pwszResTypes = g_wszResourceTypeNames;
while (*pwszResTypes != L'\0') { hr = RegisterCluAdminResourceTypeExtension( hCluster, pwszResTypes, &CLSID_CoIISClEx4 ); if (hr != S_OK) break;
wprintf(L" %s\n", pwszResTypes); pwszResTypes += lstrlenW(pwszResTypes) + 1; } // while: more resource types
return hr;
} //*** DllRegisterCluAdminExtension()
/////////////////////////////////////////////////////////////////////////////
//++
//
// DllUnregisterCluAdminExtension
//
// Routine Description:
// Unregister the extension with the cluster database.
//
// Arguments:
// hCluster [IN] Handle to the cluster to modify.
//
// Return Value:
// S_OK Extension unregistered successfully.
// Win32 error code if another failure occurred.
//
//--
/////////////////////////////////////////////////////////////////////////////
STDAPI DllUnregisterCluAdminExtension(IN HCLUSTER hCluster){ HRESULT hr = S_OK; LPCWSTR pwszResTypes = g_wszResourceTypeNames;
while (*pwszResTypes != L'\0') { wprintf(L" %s\n", pwszResTypes); hr = UnregisterCluAdminResourceTypeExtension( hCluster, pwszResTypes, &CLSID_CoIISClEx4 ); if (hr != S_OK) break; pwszResTypes += lstrlenW(pwszResTypes) + 1; } // while: more resource types
return hr;
} //*** DllUnregisterCluAdminExtension()
|
