523 lines
15 KiB
C++
523 lines
15 KiB
C++
/////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// THIS CODE AND INFORMATION IS PROVIDED "AS IS" WITHOUT WARRANTY OF
|
|
// ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED TO
|
|
// THE IMPLIED WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A
|
|
// PARTICULAR PURPOSE.
|
|
//
|
|
// Copyright 1998 Microsoft Corporation. All Rights Reserved.
|
|
//
|
|
// Author: Scott Roberts, Microsoft Developer Support - Internet Client SDK
|
|
//
|
|
// Portions of this code were taken from the bandobj sample that comes
|
|
// with the Internet Client SDK for Internet Explorer 4.0x
|
|
//
|
|
// BLFrame.cpp : Contains DLLMain and standard COM object functions.
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
#include "pch.hxx"
|
|
|
|
#include <shlwapi.h>
|
|
#include <shlwapip.h>
|
|
|
|
#define ARRAYSIZE(x) (sizeof(x)/sizeof(x[0]))
|
|
|
|
#include <atlimpl.cpp>
|
|
#include <atlctl.cpp>
|
|
#include <atlwinx.cpp>
|
|
|
|
#include <ole2.h>
|
|
#include <comcat.h>
|
|
#include <olectl.h>
|
|
#include "ClsFact.h"
|
|
#include "resource.h"
|
|
#include "msoert.h"
|
|
#include "shared.h"
|
|
// #include "demand.h"
|
|
|
|
static const char c_szShlwapiDll[] = "shlwapi.dll";
|
|
static const char c_szDllGetVersion[] = "DllGetVersion";
|
|
|
|
// This part is only done once
|
|
// If you need to use the GUID in another file, just include Guid.h
|
|
//#pragma data_seg(".text")
|
|
#define INITGUID
|
|
#include <initguid.h>
|
|
#include <shlguid.h>
|
|
#include "Guid.h"
|
|
//#pragma data_seg()
|
|
|
|
// HINSTANCE LoadLangDll(HINSTANCE hInstCaller, LPCSTR szDllName, BOOL fNT);
|
|
const TCHAR c_szBlCtlResDll[] = "iecontlc.dll";
|
|
|
|
extern "C" BOOL WINAPI DllMain(HINSTANCE, DWORD, LPVOID);
|
|
BOOL RegisterServer(CLSID, LPTSTR);
|
|
BOOL RegisterComCat(CLSID, CATID);
|
|
BOOL UnRegisterServer(CLSID);
|
|
BOOL UnRegisterComCat(CLSID, CATID);
|
|
|
|
HINSTANCE g_hLocRes;
|
|
HINSTANCE g_OrgInst = NULL;
|
|
LONG g_cDllRefCount;
|
|
CComModule _Module;
|
|
|
|
BEGIN_OBJECT_MAP(ObjectMap)
|
|
OBJECT_ENTRY(CLSID_IEMsgAb, CIEMsgAb)
|
|
END_OBJECT_MAP()
|
|
|
|
typedef struct
|
|
{
|
|
HKEY hRootKey;
|
|
LPTSTR szSubKey; // TCHAR szSubKey[MAX_PATH];
|
|
LPTSTR lpszValueName;
|
|
LPTSTR szData; // TCHAR szData[MAX_PATH];
|
|
|
|
} DOREGSTRUCT, *LPDOREGSTRUCT;
|
|
|
|
typedef HINSTANCE (STDAPICALLTYPE *PFNMLLOADLIBARY)(LPCSTR lpLibFileName, HMODULE hModule, DWORD dwCrossCodePage);
|
|
|
|
HINSTANCE IELoadLangDll(HINSTANCE hInstCaller, LPCSTR szDllName, BOOL fNT)
|
|
{
|
|
char szPath[MAX_PATH];
|
|
HINSTANCE hinstShlwapi;
|
|
PFNMLLOADLIBARY pfn;
|
|
DLLGETVERSIONPROC pfnVersion;
|
|
int iEnd;
|
|
DLLVERSIONINFO info;
|
|
HINSTANCE hInst = NULL;
|
|
|
|
hinstShlwapi = LoadLibrary(c_szShlwapiDll);
|
|
if (hinstShlwapi != NULL)
|
|
{
|
|
pfnVersion = (DLLGETVERSIONPROC)GetProcAddress(hinstShlwapi, c_szDllGetVersion);
|
|
if (pfnVersion != NULL)
|
|
{
|
|
info.cbSize = sizeof(DLLVERSIONINFO);
|
|
if (SUCCEEDED(pfnVersion(&info)))
|
|
{
|
|
if (info.dwMajorVersion >= 5)
|
|
{
|
|
pfn = (PFNMLLOADLIBARY)GetProcAddress(hinstShlwapi, MAKEINTRESOURCE(377));
|
|
if (pfn != NULL)
|
|
hInst = pfn(szDllName, hInstCaller, (ML_CROSSCODEPAGE));
|
|
}
|
|
}
|
|
}
|
|
|
|
FreeLibrary(hinstShlwapi);
|
|
}
|
|
|
|
if ((NULL == hInst) && (GetModuleFileName(hInstCaller, szPath, ARRAYSIZE(szPath))))
|
|
{
|
|
PathRemoveFileSpec(szPath);
|
|
iEnd = lstrlen(szPath);
|
|
szPath[iEnd++] = '\\';
|
|
StrCpyN(&szPath[iEnd], szDllName, ARRAYSIZE(szPath)-iEnd);
|
|
hInst = LoadLibrary(szPath);
|
|
}
|
|
|
|
AssertSz(hInst, "Failed to LoadLibrary Lang Dll");
|
|
|
|
return(hInst);
|
|
}
|
|
// CComModule _Module;
|
|
|
|
IMalloc *g_pMalloc=NULL;
|
|
|
|
extern "C" BOOL WINAPI DllMain(HINSTANCE hInstance, DWORD dwReason, LPVOID lpReserved)
|
|
{
|
|
switch(dwReason)
|
|
{
|
|
case DLL_PROCESS_ATTACH:
|
|
CoGetMalloc(1, &g_pMalloc);
|
|
// Get Resources from Lang DLL
|
|
g_OrgInst = hInstance;
|
|
g_hLocRes = IELoadLangDll(hInstance, c_szBlCtlResDll, TRUE);
|
|
if(g_hLocRes == NULL)
|
|
{
|
|
_ASSERT(FALSE);
|
|
return FALSE;
|
|
}
|
|
_Module.Init(ObjectMap, hInstance);
|
|
// InitDemandLoadedLibs();
|
|
DisableThreadLibraryCalls(hInstance);
|
|
break;
|
|
|
|
case DLL_PROCESS_DETACH:
|
|
// FreeDemandLoadedLibs();
|
|
_Module.Term();
|
|
SafeRelease(g_pMalloc);
|
|
break;
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
STDAPI DllCanUnloadNow(void)
|
|
{
|
|
return (g_cDllRefCount ? S_FALSE : S_OK);
|
|
}
|
|
|
|
STDAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID* ppv)
|
|
{
|
|
*ppv = NULL;
|
|
HRESULT hr = E_FAIL;
|
|
|
|
// If we don't support this classid, return the proper error code
|
|
if (!IsEqualCLSID(rclsid, CLSID_BLHost) && !IsEqualCLSID(rclsid, CLSID_IEMsgAb) )
|
|
return CLASS_E_CLASSNOTAVAILABLE;
|
|
|
|
if(IsEqualCLSID(rclsid, CLSID_BLHost))
|
|
{
|
|
// Create a CClassFactory object and check it for validity
|
|
CClassFactory* pClassFactory = new CClassFactory(rclsid);
|
|
if (NULL == pClassFactory)
|
|
return E_OUTOFMEMORY;
|
|
|
|
// QI for the requested interface
|
|
hr = pClassFactory->QueryInterface(riid, ppv);
|
|
|
|
// Call Release to decement the ref count - creating the object set it to one
|
|
// and QueryInterface incremented it - since its being used externally (not by
|
|
// us), we only want the ref count to be 1
|
|
pClassFactory->Release();
|
|
}
|
|
else if(IsEqualCLSID(rclsid, CLSID_IEMsgAb))
|
|
return _Module.GetClassObject(rclsid, riid, ppv);
|
|
|
|
return hr;
|
|
}
|
|
|
|
STDAPI DllRegisterServer(void)
|
|
{
|
|
TCHAR szTitle[256];
|
|
// Register the explorer bar object.
|
|
if(!ANSI_AthLoadString(idsTitle, szTitle, ARRAYSIZE(szTitle)))
|
|
{
|
|
_ASSERT(FALSE);
|
|
szTitle[0] = '\0';
|
|
}
|
|
|
|
if (!RegisterServer(CLSID_BLHost, szTitle))
|
|
return SELFREG_E_CLASS;
|
|
|
|
// Register the Component Categories for the explorer bar object.
|
|
RegisterComCat(CLSID_BLHost, CATID_InfoBand); // ignore error in this case
|
|
|
|
// registers IEMsgAb object, typelib and all interfaces in typelib
|
|
_Module.RegisterServer(TRUE);
|
|
return S_OK;
|
|
}
|
|
|
|
STDAPI DllUnregisterServer(void)
|
|
{
|
|
// Register the Component Categories for the explorer bar object.
|
|
UnRegisterComCat(CLSID_BLHost, CATID_InfoBand); // ignore error in this case
|
|
|
|
// Register the explorer bar object.
|
|
if (!UnRegisterServer(CLSID_BLHost))
|
|
return SELFREG_E_CLASS;
|
|
|
|
_Module.UnregisterServer();
|
|
return S_OK;
|
|
}
|
|
|
|
BOOL RegisterServer(CLSID clsid, LPTSTR lpszTitle)
|
|
{
|
|
int i;
|
|
HKEY hKey;
|
|
LRESULT lResult;
|
|
DWORD dwDisp;
|
|
TCHAR szSubKey[MAX_PATH];
|
|
TCHAR szCLSID[MAX_PATH];
|
|
TCHAR szBlFrameCLSID[MAX_PATH];
|
|
TCHAR szModule[MAX_PATH];
|
|
LPWSTR pwsz;
|
|
|
|
// Get the CLSID in string form
|
|
StringFromIID(clsid, &pwsz);
|
|
|
|
if (pwsz)
|
|
{
|
|
#ifdef UNICODE
|
|
StrCpyN(szBlFrameCLSID, pwsz, ARRAYSIZE(szBlFrameCLSID));
|
|
#else
|
|
WideCharToMultiByte(CP_ACP, 0, pwsz, -1, szBlFrameCLSID,
|
|
ARRAYSIZE(szBlFrameCLSID), NULL, NULL);
|
|
#endif
|
|
|
|
// Free the string
|
|
LPMALLOC pMalloc;
|
|
|
|
CoGetMalloc(1, &pMalloc);
|
|
pMalloc->Free(pwsz);
|
|
|
|
pMalloc->Release();
|
|
}
|
|
|
|
// Get this app's path and file name
|
|
GetModuleFileName(g_OrgInst, szModule, ARRAYSIZE(szModule));
|
|
|
|
DOREGSTRUCT ClsidEntries[] =
|
|
{
|
|
HKEY_CLASSES_ROOT, TEXT("CLSID\\%s"),
|
|
NULL, lpszTitle,
|
|
HKEY_CLASSES_ROOT, TEXT("CLSID\\%s\\InprocServer32"),
|
|
NULL, szModule,
|
|
HKEY_CLASSES_ROOT, TEXT("CLSID\\%s\\InprocServer32"),
|
|
TEXT("ThreadingModel"), TEXT("Apartment"),
|
|
NULL, NULL, NULL, NULL
|
|
};
|
|
|
|
// Register the CLSID entries
|
|
for (i = 0; ClsidEntries[i].hRootKey; i++)
|
|
{
|
|
// Create the sub key string - for this case, insert
|
|
// the file extension
|
|
//
|
|
wnsprintf(szSubKey, ARRAYSIZE(szSubKey), ClsidEntries[i].szSubKey, szBlFrameCLSID);
|
|
|
|
lResult = RegCreateKeyEx(ClsidEntries[i].hRootKey, szSubKey,
|
|
0, NULL, REG_OPTION_NON_VOLATILE,
|
|
KEY_WRITE, NULL, &hKey, &dwDisp);
|
|
|
|
if (ERROR_SUCCESS == lResult)
|
|
{
|
|
TCHAR szData[MAX_PATH];
|
|
|
|
// If necessary, create the value string
|
|
wnsprintf(szData, ARRAYSIZE(szData), ClsidEntries[i].szData, szModule);
|
|
|
|
lResult = RegSetValueEx(hKey, ClsidEntries[i].lpszValueName,
|
|
0, REG_SZ, (LPBYTE)szData,
|
|
lstrlen(szData) + 1);
|
|
RegCloseKey(hKey);
|
|
|
|
if (ERROR_SUCCESS != lResult)
|
|
return FALSE;
|
|
}
|
|
else
|
|
{
|
|
return FALSE;
|
|
}
|
|
}
|
|
|
|
// Create the Registry key entry and values for the
|
|
// toolbar button.
|
|
//
|
|
StringFromIID(CLSID_BlFrameButton, &pwsz);
|
|
|
|
if (!pwsz)
|
|
return TRUE;
|
|
|
|
#ifdef UNICODE
|
|
StrCpyN(szCLSID, pwsz, ARRAYSIZE(szCLSID));
|
|
#else
|
|
WideCharToMultiByte(CP_ACP, 0, pwsz, -1, szCLSID,
|
|
ARRAYSIZE(szCLSID), NULL, NULL);
|
|
#endif
|
|
|
|
// Free the string
|
|
LPMALLOC pMalloc;
|
|
|
|
CoGetMalloc(1, &pMalloc);
|
|
pMalloc->Free(pwsz);
|
|
pMalloc->Release();
|
|
|
|
wnsprintf(szSubKey, ARRAYSIZE(szSubKey), "Software\\Microsoft\\Internet Explorer\\Extensions\\%s",
|
|
szCLSID);
|
|
|
|
lResult = RegCreateKeyEx(HKEY_LOCAL_MACHINE, szSubKey,
|
|
0, NULL, REG_OPTION_NON_VOLATILE,
|
|
KEY_WRITE, NULL, &hKey, &dwDisp);
|
|
|
|
if (ERROR_SUCCESS == lResult)
|
|
{
|
|
TCHAR szData[MAX_PATH];
|
|
|
|
// Create the value strings
|
|
//
|
|
// Register the explorer bar object.
|
|
|
|
|
|
OSVERSIONINFO OSInfo;
|
|
BOOL fWhistler = FALSE;
|
|
OSInfo.dwOSVersionInfoSize = sizeof(OSVERSIONINFO);
|
|
|
|
GetVersionEx(&OSInfo);
|
|
if((OSInfo.dwPlatformId == VER_PLATFORM_WIN32_NT) && (OSInfo.dwMajorVersion >= 5) &&
|
|
(OSInfo.dwMinorVersion >= 1))
|
|
fWhistler = TRUE;
|
|
|
|
#ifdef NEED // Bug 28254
|
|
StrCpyN(szData, "Yes", ARRAYSIZE(szData));
|
|
RegSetValueEx(hKey, TEXT("Default Visible"),
|
|
0, REG_SZ, (LPBYTE)szData,
|
|
lstrlen(szData) + 1);
|
|
#endif
|
|
TCHAR szPath[MAX_PATH];
|
|
HMODULE hModule;
|
|
|
|
hModule = GetModuleHandle(TEXT("iecont.dll"));
|
|
GetModuleFileName(hModule, szPath, MAX_PATH);
|
|
|
|
wnsprintf(szData, ARRAYSIZE(szData), "%s,%d", szPath, fWhistler ? IDI_WHISTICON : IDI_HOTICON);
|
|
RegSetValueEx(hKey, TEXT("HotIcon"),
|
|
0, REG_SZ, (LPBYTE)szData,
|
|
lstrlen(szData) + 1);
|
|
|
|
wnsprintf(szData, ARRAYSIZE(szData), "%s,%d", szPath, fWhistler ? IDI_WHISTICON : IDI_ICON);
|
|
RegSetValueEx(hKey, TEXT("Icon"),
|
|
0, REG_SZ, (LPBYTE)szData,
|
|
lstrlen(szData) + 1);
|
|
|
|
StrCpyN(szData, "{E0DD6CAB-2D10-11D2-8F1A-0000F87ABD16}", ARRAYSIZE(szData));
|
|
RegSetValueEx(hKey, TEXT("CLSID"),
|
|
0, REG_SZ, (LPBYTE)szData,
|
|
lstrlen(szData) + 1);
|
|
|
|
wnsprintf(szData, ARRAYSIZE(szData), "%s", szBlFrameCLSID);
|
|
RegSetValueEx(hKey, TEXT("BandCLSID"),
|
|
0, REG_SZ, (LPBYTE)szData,
|
|
lstrlen(szData) + 1);
|
|
|
|
// SHGetWebFolderFilePath(TEXT("iecontlc.dll"), szPath, ARRAYSIZE(szPath));
|
|
wnsprintf(szData, ARRAYSIZE(szData), "@iecontlc.dll,-%d",idsButtontext);
|
|
RegSetValueEx(hKey, TEXT("ButtonText"),
|
|
0, REG_SZ, (LPBYTE)szData,
|
|
lstrlen(szData) + 1);
|
|
|
|
wnsprintf(szData, ARRAYSIZE(szData), "@iecontlc.dll,-%d",idsTitle);
|
|
RegSetValueEx(hKey, TEXT("MenuText"),
|
|
0, REG_SZ, (LPBYTE)szData,
|
|
lstrlen(szData) + 1);
|
|
|
|
RegCloseKey(hKey);
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
BOOL RegisterComCat(CLSID clsid, CATID CatID)
|
|
{
|
|
ICatRegister* pcr;
|
|
HRESULT hr = S_OK ;
|
|
|
|
CoInitialize(NULL);
|
|
|
|
hr = CoCreateInstance(CLSID_StdComponentCategoriesMgr,
|
|
NULL,
|
|
CLSCTX_INPROC_SERVER,
|
|
IID_ICatRegister,
|
|
(LPVOID*)&pcr);
|
|
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
hr = pcr->RegisterClassImplCategories(clsid, 1, &CatID);
|
|
pcr->Release();
|
|
}
|
|
|
|
CoUninitialize();
|
|
|
|
return SUCCEEDED(hr);
|
|
}
|
|
|
|
BOOL UnRegisterServer(CLSID clsid)
|
|
{
|
|
TCHAR szSubKey[MAX_PATH];
|
|
TCHAR szCLSID[MAX_PATH];
|
|
LPWSTR pwsz;
|
|
|
|
// Get the CLSID in string form
|
|
StringFromIID(clsid, &pwsz);
|
|
|
|
if (pwsz)
|
|
{
|
|
#ifdef UNICODE
|
|
StrCpyN(szCLSID, pwsz, ARRAYSIZE(szCLSID));
|
|
#else
|
|
WideCharToMultiByte(CP_ACP, 0, pwsz, -1, szCLSID,
|
|
ARRAYSIZE(szCLSID), NULL, NULL);
|
|
#endif
|
|
|
|
// Free the string
|
|
LPMALLOC pMalloc;
|
|
|
|
CoGetMalloc(1, &pMalloc);
|
|
|
|
pMalloc->Free(pwsz);
|
|
pMalloc->Release();
|
|
}
|
|
|
|
DOREGSTRUCT ClsidEntries[] =
|
|
{
|
|
HKEY_CLASSES_ROOT, TEXT("CLSID\\%s\\InprocServer32"),
|
|
NULL, NULL,
|
|
//
|
|
// Remove the "Implemented Categories" key, just in case
|
|
// UnRegisterClassImplCategories does not remove it
|
|
//
|
|
HKEY_CLASSES_ROOT, TEXT("CLSID\\%s\\Implemented Categories"),
|
|
NULL, NULL,
|
|
HKEY_CLASSES_ROOT, TEXT("CLSID\\%s"), NULL, NULL,
|
|
NULL, NULL, NULL, NULL
|
|
};
|
|
|
|
// Delete the CLSID entries
|
|
for (int i = 0; ClsidEntries[i].hRootKey; i++)
|
|
{
|
|
wnsprintf(szSubKey, ARRAYSIZE(szSubKey), ClsidEntries[i].szSubKey, szCLSID);
|
|
RegDeleteKey(ClsidEntries[i].hRootKey, szSubKey);
|
|
}
|
|
|
|
// Delete the button information
|
|
//
|
|
StringFromIID(CLSID_BlFrameButton, &pwsz);
|
|
|
|
if (!pwsz)
|
|
return TRUE;
|
|
|
|
#ifdef UNICODE
|
|
StrCpyN(szCLSID, pwsz, ARRAYSIZE(szCLSID));
|
|
#else
|
|
WideCharToMultiByte(CP_ACP, 0, pwsz, -1, szCLSID,
|
|
ARRAYSIZE(szCLSID), NULL, NULL);
|
|
#endif
|
|
|
|
// Free the string
|
|
LPMALLOC pMalloc;
|
|
|
|
CoGetMalloc(1, &pMalloc);
|
|
pMalloc->Free(pwsz);
|
|
pMalloc->Release();
|
|
|
|
wnsprintf(szSubKey, ARRAYSIZE(szSubKey), "Software\\Microsoft\\Internet Explorer\\Extensions\\%s", szCLSID);
|
|
RegDeleteKey(HKEY_LOCAL_MACHINE, szSubKey);
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
BOOL UnRegisterComCat(CLSID clsid, CATID CatID)
|
|
{
|
|
ICatRegister* pcr;
|
|
HRESULT hr = S_OK ;
|
|
|
|
CoInitialize(NULL);
|
|
|
|
hr = CoCreateInstance(CLSID_StdComponentCategoriesMgr,
|
|
NULL,
|
|
CLSCTX_INPROC_SERVER,
|
|
IID_ICatRegister,
|
|
(LPVOID*)&pcr);
|
|
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
hr = pcr->UnRegisterClassImplCategories(clsid, 1, &CatID);
|
|
pcr->Release();
|
|
}
|
|
|
|
CoUninitialize();
|
|
|
|
return SUCCEEDED(hr);
|
|
}
|
|
|