a

a/admin/activec/base/classreg.cpp

@@ -0,0 +1,221 @@
+/*--------------------------------------------------------------------------*
+ *
+ *  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();
+}