|
|
/*--------------------------------------------------------------------------*
* * Microsoft Windows * Copyright (C) Microsoft Corporation, 1992 - 00 * * File: classreg.cpp * * Contents: Class registration code * * History: 3-Feb-2000 jeffro Created * *--------------------------------------------------------------------------*/
#include "stdafx.h"
#ifdef DBG
CTraceTag tagDllRegistration (_T("MMC Dll Registration"), _T("MMC Dll Registration"));#endif //DBG
/*+-------------------------------------------------------------------------*
* szObjScript * * Standard registration script template for all objects. '%' characters * that we want to end up in the final script must be doubled (i.e. "%%") * because this string is used as a format string for sprintf. sprintf * will convert "%%" to "%" during formatting. *--------------------------------------------------------------------------*/
static const WCHAR szObjScript[] = L"HKCR" L"\n" L"{" L"\n" L" %%VProgID%% = s '%%VClassName%%'" L"\n" L" {" L"\n" L" CLSID = s '%%VCLSID%%'" L"\n" L" }" L"\n" L" %%VVersionIndependentProgID%% = s '%%VClassName%%'" L"\n" L" {" L"\n" L" CLSID = s '%%VCLSID%%'" L"\n" L" CurVer = s '%%VProgID%%'" L"\n" L" }" L"\n" L" NoRemove CLSID" L"\n" L" {" L"\n" L" ForceRemove %%VCLSID%% = s '%%VClassName%%'" L"\n" L" {" L"\n" L" ProgID = s '%%VProgID%%'" L"\n" L" VersionIndependentProgID = s '%%VVersionIndependentProgID%%'" L"\n" L" %%ServerType%% = s '%%VFileName%%'" L"\n" L" {" L"\n" L" val ThreadingModel = s 'Apartment'" L"\n" L" }" L"\n" L" %s" /* szCtlScript substituted here if necessary */ L"\n" L" }" L"\n" L" }" L"\n" L"}";
/*+-------------------------------------------------------------------------*
* szCtlScript * * Additional registration script elements for controls. Note that '%' * characters we want to end up in the final script DO NOT need to be * doubled, because this string is used as a sprintf replacement parameter * (which are substituted as-is) and not the format string (where "%%"'s * are converted to "%"). *--------------------------------------------------------------------------*/
static const WCHAR szCtlScript[] = L" ForceRemove 'Programmable'" L"\n" L" ForceRemove 'Control'" L"\n" L" ForceRemove 'ToolboxBitmap32' = s '%VFileName%, %VBitmapID%'" L"\n" L" 'MiscStatus' = s '0'" L"\n" L" {" L"\n" L" '1' = s '131473'" L"\n" L" }" L"\n" L" 'TypeLib' = s '%VLIBID%'" L"\n" L" 'Version' = s '%VVersion%'";
/*+-------------------------------------------------------------------------*
* MMCUpdateRegistry * * Registers a COM object or control. This function typically isn't used * directly, but indirectly via DECLARE_MMC_OBJECT_REGISTRATION or * DECLARE_MMC_CONTROL_REGISTRATION. * * This function uses a class (ATL::CRegObject) that ATL only documents * indirectly. Search MSDN for "StringRegister" to find sketchy details. *--------------------------------------------------------------------------*/
HRESULT WINAPI MMCUpdateRegistry ( BOOL bRegister, // I:register or unregister?
const CObjectRegParams* pObjParams, // I:object registration parameters
const CControlRegParams* pCtlParams) // I:control registration parameters (optional)
{ DECLARE_SC(sc, TEXT("MMCUpdateRegistry"));
/*
* validate required inputs */ sc = ScCheckPointers (pObjParams, E_FAIL); if(sc) return sc.ToHr();
/*
* string-ify the CLSID */ CCoTaskMemPtr<WCHAR> spszClsid; sc = StringFromCLSID (pObjParams->m_clsid, &spszClsid); if (sc) return sc.ToHr();
#ifdef _ATL_NAMESPACE_BUG_FIXED
::ATL::CRegObject ro; // hack around nested namespace bug in ATL30
#else
/*
* specify the standard object substitution parameters for CRegObject */ ::ATL::ATL::CRegObject ro; // hack around nested namespace bug in ATL30
#endif
_ATL_REGMAP_ENTRY rgObjEntries[] = { { L"VCLSID", spszClsid }, { L"VFileName", pObjParams->m_strModuleName.data() }, { L"VClassName", pObjParams->m_strClassName.data() }, { L"VProgID", pObjParams->m_strProgID.data() }, { L"VVersionIndependentProgID", pObjParams->m_strVersionIndependentProgID.data() }, { L"ServerType", pObjParams->m_strServerType.data() }, };
#ifdef DBG
std::wstring strReplacements;#endif
for (int i = 0; i < countof (rgObjEntries); i++) { sc = ro.AddReplacement (rgObjEntries[i].szKey, rgObjEntries[i].szData); if (sc) return (sc.ToHr());
AddReplacementTrace (strReplacements, rgObjEntries[i].szKey, rgObjEntries[i].szData); }
/*
* if we're registering a control, add its substitution parameters for CRegObject */ if (pCtlParams != NULL) { /*
* string-ify the LIBID */ CCoTaskMemPtr<WCHAR> spszLibid; sc = StringFromCLSID (pCtlParams->m_libid, &spszLibid); if (sc) return (sc.ToHr());
_ATL_REGMAP_ENTRY rgCtlEntries[] = { { L"VLIBID", spszLibid }, { L"VBitmapID", pCtlParams->m_strToolboxBitmapID.data() }, { L"VVersion", pCtlParams->m_strVersion.data() }, };
for (int i = 0; i < countof (rgCtlEntries); i++) { sc = ro.AddReplacement (rgCtlEntries[i].szKey, rgCtlEntries[i].szData); if (sc) return (sc.ToHr()); AddReplacementTrace (strReplacements, rgCtlEntries[i].szKey, rgCtlEntries[i].szData); } }
/*
* format the registration script */ WCHAR szRegScript[countof(szObjScript) + countof(szCtlScript)]; sc = StringCchPrintfW(szRegScript, countof(szRegScript), szObjScript, (pCtlParams != NULL) ? szCtlScript : L""); if (sc) return sc.ToHr();
USES_CONVERSION; Trace (tagDllRegistration, _T("Registration script:\n%s"), W2T(szRegScript)); Trace (tagDllRegistration, W2CT(strReplacements.data()));
/*
* (un)register! */ sc = (bRegister) ? ro.StringRegister (szRegScript) : ro.StringUnregister (szRegScript);
if (sc) return sc.ToHr();
// change the module path to the absolute one, if we know it
if ( bRegister && pObjParams->m_strModulePath.length() != 0 ) { // format class ID key.
tstring strKey = tstring(_T("CLSID\\")) + W2CT( spszClsid ); strKey += tstring(_T("\\")) + W2CT( pObjParams->m_strServerType.c_str() );
// see what type of value we need to put
DWORD dwValueType = CModulePath::PlatformSupports_REG_EXPAND_SZ_Values() ? REG_EXPAND_SZ : REG_SZ;
CRegKey keyServer; LONG lRet = keyServer.Open(HKEY_CLASSES_ROOT, strKey.c_str() , KEY_WRITE); if (lRet == ERROR_SUCCESS) { RegSetValueEx( keyServer, NULL, 0, dwValueType, (CONST BYTE *)( W2CT( pObjParams->m_strModulePath.c_str() ) ), (pObjParams->m_strModulePath.length() + 1) * sizeof(TCHAR) ); } }
return sc.ToHr();}
|
