311 lines
7.5 KiB
C++
311 lines
7.5 KiB
C++
|
/////////////////////////////////////////////////////////////////////////////
|
||
|
//
|
||
|
// Copyright (c) 1998-2000 <company name>
|
||
|
//
|
||
|
// Module Name:
|
||
|
// DummyEx.cpp
|
||
|
//
|
||
|
// Abstract:
|
||
|
// Implementation of the CDummyApp class and DLL initialization
|
||
|
// routines.
|
||
|
//
|
||
|
// Author:
|
||
|
// <name> (<e-mail name>) Mmmm DD, 1998
|
||
|
//
|
||
|
// Revision History:
|
||
|
//
|
||
|
// Notes:
|
||
|
// NOTE: You must use the MIDL compiler from NT 4.0,
|
||
|
// version 3.00.44 or greater
|
||
|
//
|
||
|
/////////////////////////////////////////////////////////////////////////////
|
||
|
|
||
|
#include "stdafx.h"
|
||
|
#include <initguid.h>
|
||
|
#include <CluAdmEx.h>
|
||
|
#include "DummyEx.h"
|
||
|
#include "ExtObj.h"
|
||
|
#include "BasePage.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;
|
||
|
|
||
|
#pragma warning(disable : 4701) // local variable may be used without having been initialized
|
||
|
#include <atlimpl.cpp>
|
||
|
#pragma warning(default : 4701)
|
||
|
|
||
|
BEGIN_OBJECT_MAP(ObjectMap)
|
||
|
OBJECT_ENTRY(CLSID_CoDummyEx, CExtObject)
|
||
|
END_OBJECT_MAP()
|
||
|
|
||
|
/////////////////////////////////////////////////////////////////////////////
|
||
|
// Global Function Prototypes
|
||
|
/////////////////////////////////////////////////////////////////////////////
|
||
|
|
||
|
STDAPI DllCanUnloadNow(void);
|
||
|
STDAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID* ppv);
|
||
|
STDAPI DllRegisterServer(void);
|
||
|
STDAPI DllUnregisterServer(void);
|
||
|
STDAPI DllRegisterCluAdminExtension(IN HCLUSTER hcluster);
|
||
|
STDAPI DllUnregisterCluAdminExtension(IN HCLUSTER hcluster);
|
||
|
|
||
|
/////////////////////////////////////////////////////////////////////////////
|
||
|
// class CDummyApp
|
||
|
/////////////////////////////////////////////////////////////////////////////
|
||
|
|
||
|
class CDummyApp : public CWinApp
|
||
|
{
|
||
|
public:
|
||
|
virtual BOOL InitInstance();
|
||
|
virtual int ExitInstance();
|
||
|
};
|
||
|
|
||
|
/////////////////////////////////////////////////////////////////////////////
|
||
|
// The one and only CDummyApp object
|
||
|
|
||
|
CDummyApp theApp;
|
||
|
|
||
|
/////////////////////////////////////////////////////////////////////////////
|
||
|
//++
|
||
|
//
|
||
|
// CDummyApp::InitInstance
|
||
|
//
|
||
|
// Routine Description:
|
||
|
// Initialize this instance of the application.
|
||
|
//
|
||
|
// Arguments:
|
||
|
// None.
|
||
|
//
|
||
|
// Return Value:
|
||
|
// Any return codes from CWinApp::InitInstance().
|
||
|
//
|
||
|
//--
|
||
|
/////////////////////////////////////////////////////////////////////////////
|
||
|
BOOL CDummyApp::InitInstance(void)
|
||
|
{
|
||
|
_Module.Init(ObjectMap, m_hInstance);
|
||
|
return CWinApp::InitInstance();
|
||
|
|
||
|
} //*** CDummyApp::InitInstance()
|
||
|
|
||
|
/////////////////////////////////////////////////////////////////////////////
|
||
|
//++
|
||
|
//
|
||
|
// CDummyApp::ExitInstance
|
||
|
//
|
||
|
// Routine Description:
|
||
|
// Deinitialize this instance of the application.
|
||
|
//
|
||
|
// Arguments:
|
||
|
// None.
|
||
|
//
|
||
|
// Return Value:
|
||
|
// Any return codes from CWinApp::ExitInstance().
|
||
|
//
|
||
|
//--
|
||
|
/////////////////////////////////////////////////////////////////////////////
|
||
|
int CDummyApp::ExitInstance(void)
|
||
|
{
|
||
|
_Module.Term();
|
||
|
return CWinApp::ExitInstance();
|
||
|
|
||
|
} //*** CDummyApp::ExitInstance()
|
||
|
|
||
|
/////////////////////////////////////////////////////////////////////////////
|
||
|
//++
|
||
|
//
|
||
|
// 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 _cch;
|
||
|
TCHAR _szError[512];
|
||
|
|
||
|
_cch = FormatMessage(
|
||
|
FORMAT_MESSAGE_FROM_SYSTEM,
|
||
|
NULL,
|
||
|
dwError,
|
||
|
MAKELANGID(LANG_NEUTRAL, SUBLANG_NEUTRAL),
|
||
|
_szError,
|
||
|
sizeof(_szError) / sizeof(TCHAR),
|
||
|
0
|
||
|
);
|
||
|
if (_cch == 0)
|
||
|
{
|
||
|
// Format the NT status code from NTDLL since this hasn't been
|
||
|
// integrated into the system yet.
|
||
|
_cch = 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: error formatting status code from system
|
||
|
|
||
|
if (_cch > 0)
|
||
|
{
|
||
|
rstrError = _szError;
|
||
|
} // if: no error
|
||
|
else
|
||
|
{
|
||
|
|
||
|
#ifdef _DEBUG
|
||
|
|
||
|
DWORD _sc = GetLastError();
|
||
|
|
||
|
TRACE(_T("FormatError() - Error 0x%08.8x formatting string for error code 0x%08.8x\n"), _sc, dwError);
|
||
|
|
||
|
#endif
|
||
|
|
||
|
rstrError.Format(_T("Error 0x%08.8x"), dwError);
|
||
|
|
||
|
} // else: error formatting the message
|
||
|
|
||
|
} //*** FormatError()
|
||
|
|
||
|
/////////////////////////////////////////////////////////////////////////////
|
||
|
// 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;
|
||
|
|
||
|
} //*** DllCanUnloadNow()
|
||
|
|
||
|
/////////////////////////////////////////////////////////////////////////////
|
||
|
// 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);
|
||
|
|
||
|
} //*** DllGetClassObject()
|
||
|
|
||
|
/////////////////////////////////////////////////////////////////////////////
|
||
|
// 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;
|
||
|
|
||
|
} //*** DllRegisterServer()
|
||
|
|
||
|
/////////////////////////////////////////////////////////////////////////////
|
||
|
// DllUnregisterServer - Removes entries from the system registry
|
||
|
|
||
|
STDAPI DllUnregisterServer(void)
|
||
|
{
|
||
|
HRESULT hRes = S_OK;
|
||
|
_Module.UnregisterServer();
|
||
|
return hRes;
|
||
|
|
||
|
} //*** DllUnregisterServer()
|
||
|
|
||
|
/////////////////////////////////////////////////////////////////////////////
|
||
|
//++
|
||
|
//
|
||
|
// 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;
|
||
|
HRESULT hrReturn = S_OK;
|
||
|
LPCWSTR pwszResTypes = g_wszResourceTypeNames;
|
||
|
|
||
|
while (*pwszResTypes != L'\0')
|
||
|
{
|
||
|
wprintf(L" %s\n", pwszResTypes);
|
||
|
hr = RegisterCluAdminResourceTypeExtension(
|
||
|
hCluster,
|
||
|
pwszResTypes,
|
||
|
&CLSID_CoDummyEx
|
||
|
);
|
||
|
if (hr != S_OK)
|
||
|
hrReturn = hr;
|
||
|
pwszResTypes += lstrlenW(pwszResTypes) + 1;
|
||
|
} // while: more resource types
|
||
|
|
||
|
return hrReturn;
|
||
|
|
||
|
} //*** 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;
|
||
|
HRESULT hrReturn = S_OK;
|
||
|
LPCWSTR pwszResTypes = g_wszResourceTypeNames;
|
||
|
|
||
|
while (*pwszResTypes != L'\0')
|
||
|
{
|
||
|
wprintf(L" %s\n", pwszResTypes);
|
||
|
hr = UnregisterCluAdminResourceTypeExtension(
|
||
|
hCluster,
|
||
|
pwszResTypes,
|
||
|
&CLSID_CoDummyEx
|
||
|
);
|
||
|
if (hr != S_OK)
|
||
|
hrReturn = hr;
|
||
|
pwszResTypes += lstrlenW(pwszResTypes) + 1;
|
||
|
} // while: more resource types
|
||
|
|
||
|
return hrReturn;
|
||
|
|
||
|
} //*** DllUnregisterCluAdminExtension()
|