Microsoft/Windows2000
Microsoft/Windows2000
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity
45f4a58588
Windows2000/private/shell/shdocvw/sccls.cpp
Microsoft 45f4a58588 First commit
2020-09-30 17:12:32 +02:00

637 lines
18 KiB
C++
Raw Blame History

#define DONT_USE_ATL
#include "priv.h"
#include "sccls.h"
#include "atl.h"
#include <ntverp.h>
#include <shdguid.h>
#include <shguidp.h> // for CLSID_CDocObjectFolder
#include "ishcut.h"
#include "reload.h" // for URLDL_WNDCLASS
#include "inetnot.h" // CWinInetNotify_szWindowClass
#define MLUI_INIT
#include <mluisupp.h>
STDAPI_(void) InitURLIDs(UINT uPlatform); // from shdocfl.cpp
LONG g_cRefThisDll = 0; // per-instance
CRITICAL_SECTION g_csDll = {0}; // per-instance
HINSTANCE g_hinst = NULL;
EXTERN_C HANDLE g_hMutexHistory = NULL;
BOOL g_fRunningOnNT = FALSE;
BOOL g_bNT5Upgrade = FALSE;
BOOL g_bRunOnNT5 = FALSE;
BOOL g_bRunOnMemphis = FALSE;
BOOL g_fRunOnFE = FALSE;
BOOL g_bInDllInstall;
UINT g_uiACP = CP_ACP;
DWORD g_dwStopWatchMode = 0; // Shell perf automation
// Is Mirroring enabled
BOOL g_bMirroredOS = FALSE;
#ifdef WINDOWS_ME
// This is needed for BiDi localized win95 RTL stuff
BOOL g_bBiDiW95Loc = FALSE;
#endif // WINDOWS_ME
EXTERN_C HANDLE g_hSemBrowserCount;
HPALETTE g_hpalHalftone = NULL;
EXTERN_C const GUID CLSID_MsgBand ;
HRESULT CMsgBand_CreateInstance(IUnknown *punkOuter, IUnknown **ppunk, LPCOBJECTINFO poi);
EXTERN_C void FreeDynamicLibraries();
// This array holds information needed for ClassFacory.
// OLEMISC_ flags are used by shembed and shocx.
// BUGBUG: this table should be ordered in most-to-least used order
CF_TABLE_BEGIN( g_ObjectInfo )
CF_TABLE_ENTRY( &CLSID_CDocObjectFolder, CDocObjectFolder_CreateInstance,
COCREATEONLY),
CF_TABLE_ENTRY_NOFLAGS( &CLSID_CBaseBrowser, CBaseBrowser2_CreateInstance,
COCREATEONLY_NOFLAGS, OIF_ALLOWAGGREGATION),
CF_TABLE_ENTRY( &CLSID_CURLFolder, CInternetFolder_CreateInstance,
COCREATEONLY),
CF_TABLE_ENTRY( &CLSID_Internet, CInternetFolder_CreateInstance,
COCREATEONLY),
CF_TABLE_ENTRY( &CLSID_CacheFolder, CacheFolder_CreateInstance,
COCREATEONLY),
CF_TABLE_ENTRY( &CLSID_HistFolder, HistFolder_CreateInstance,
COCREATEONLY),
CF_TABLE_ENTRY_ALL( &CLSID_WebBrowser, CWebBrowserOC_CreateInstance,
&IID_IWebBrowser2, &DIID_DWebBrowserEvents2, VERSION_2,
OLEMISC_SETCLIENTSITEFIRST|OLEMISC_ACTIVATEWHENVISIBLE|OLEMISC_RECOMPOSEONRESIZE|OLEMISC_CANTLINKINSIDE|OLEMISC_INSIDEOUT,OIF_ALLOWAGGREGATION),
CF_TABLE_ENTRY( &CLSID_CUrlHistory, CUrlHistory_CreateInstance,
COCREATEONLY),
CF_TABLE_ENTRY( &CLSID_CURLSearchHook, CURLSearchHook_CreateInstance,
COCREATEONLY),
CF_TABLE_ENTRY_ALL( &CLSID_WebBrowser_V1, CWebBrowserOC_CreateInstance,
&IID_IWebBrowser, &DIID_DWebBrowserEvents, VERSION_1,
OLEMISC_SETCLIENTSITEFIRST|OLEMISC_ACTIVATEWHENVISIBLE|OLEMISC_RECOMPOSEONRESIZE|OLEMISC_CANTLINKINSIDE|OLEMISC_INSIDEOUT,OIF_ALLOWAGGREGATION),
CF_TABLE_ENTRY( &CLSID_CStubBindStatusCallback, CStubBSC_CreateInstance,
COCREATEONLY),
CF_TABLE_ENTRY( &CLSID_ShellUIHelper, CShellUIHelper_CreateInstance,
COCREATEONLY),
CF_TABLE_ENTRY( &CLSID_InternetShortcut, CIntShcut_CreateInstance,
COCREATEONLY),
#ifdef ENABLE_CHANNELS
CF_TABLE_ENTRY_ALL( &CLSID_ChannelOC, ChannelOC_CreateInstance,
NULL,NULL,0,
OLEMISC_ACTIVATEWHENVISIBLE|OLEMISC_CANTLINKINSIDE|OLEMISC_INSIDEOUT,
OIF_ALLOWAGGREGATION),
#endif // ENABLE_CHANNELS
#ifndef NO_SPLASHSCREEN
CF_TABLE_ENTRY( &CLSID_IESplashScreen, CIESplashScreen_CreateInstance,
COCREATEONLY),
#endif
CF_TABLE_ENTRY( &CLSID_WinListShellProc, CWinListShellProc_CreateInstance,
COCREATEONLY),
CF_TABLE_ENTRY( &CLSID_CDFCopyHook, CCDFCopyHook_CreateInstance,
COCREATEONLY),
CF_TABLE_ENTRY( &CLSID_InternetCacheCleaner, CInternetCacheCleaner_CreateInstance,
COCREATEONLY),
CF_TABLE_ENTRY( &CLSID_OfflinePagesCacheCleaner, COfflinePagesCacheCleaner_CreateInstance,
COCREATEONLY),
CF_TABLE_ENTRY( &CLSID_TaskbarList, TaskbarList_CreateInstance,
COCREATEONLY),
CF_TABLE_ENTRY( &CLSID_DocFileInfoTip, CDocFileInfoTip_CreateInstance,
COCREATEONLY),
CF_TABLE_ENTRY( &CLSID_DocHostUIHandler, CDocHostUIHandler_CreateInstance,
COCREATEONLY),
CF_TABLE_ENTRY( &CLSID_ToolbarExtBand, CToolbarExtBand_CreateInstance,
COCREATEONLY),
CF_TABLE_ENTRY( &CLSID_ToolbarExtExec, CToolbarExtExec_CreateInstance,
COCREATEONLY),
CF_TABLE_ENTRY( &CLSID_NSCTree, CNscTree_CreateInstance,
COCREATEONLY),
CF_TABLE_ENTRY( &CLSID_FavBand, CFavBand_CreateInstance,
COCREATEONLY),
CF_TABLE_ENTRY( &CLSID_HistBand, CHistBand_CreateInstance,
COCREATEONLY),
CF_TABLE_END( g_ObjectInfo )
// constructor for CObjectInfo.
CObjectInfo::CObjectInfo(CLSID const* pclsidin, LPFNCREATEOBJINSTANCE pfnCreatein, IID const* piidIn,
IID const* piidEventsIn, long lVersionIn, DWORD dwOleMiscFlagsIn,
DWORD dwClassFactFlagsIn)
{
pclsid = pclsidin;
pfnCreateInstance = pfnCreatein;
piid = piidIn;
piidEvents = piidEventsIn;
lVersion = lVersionIn;
dwOleMiscFlags = dwOleMiscFlagsIn;
dwClassFactFlags = dwClassFactFlagsIn;
}
// static class factory (no allocs!)
STDMETHODIMP CClassFactory::QueryInterface(REFIID riid, void **ppvObj)
{
if (IsEqualIID(riid, IID_IClassFactory) || IsEqualIID(riid, IID_IUnknown))
{
*ppvObj = (void *)GET_ICLASSFACTORY(this);
DllAddRef();
return NOERROR;
}
*ppvObj = NULL;
return E_NOINTERFACE;
}
STDMETHODIMP_(ULONG) CClassFactory::AddRef()
{
DllAddRef();
return 2;
}
STDMETHODIMP_(ULONG) CClassFactory::Release()
{
DllRelease();
return 1;
}
STDMETHODIMP CClassFactory::CreateInstance(IUnknown *punkOuter, REFIID riid, void **ppv)
{
*ppv = NULL;
if (punkOuter && !IsEqualIID(riid, IID_IUnknown))
{
// It is technically illegal to aggregate an object and request
// any interface other than IUnknown. Enforce this.
return CLASS_E_NOAGGREGATION;
}
else
{
LPOBJECTINFO pthisobj = (LPOBJECTINFO)this;
if (punkOuter && !(pthisobj->dwClassFactFlags & OIF_ALLOWAGGREGATION))
return CLASS_E_NOAGGREGATION;
IUnknown *punk;
HRESULT hres = pthisobj->pfnCreateInstance(punkOuter, &punk, pthisobj);
if (SUCCEEDED(hres))
{
hres = punk->QueryInterface(riid, ppv);
punk->Release();
}
ASSERT(FAILED(hres) ? *ppv == NULL : TRUE);
return hres;
}
}
STDMETHODIMP CClassFactory::LockServer(BOOL fLock)
{
if (fLock)
DllAddRef();
else
DllRelease();
TraceMsg(DM_TRACE, "sccls: LockServer(%s) to %d", fLock ? TEXT("LOCK") : TEXT("UNLOCK"), g_cRefThisDll);
return S_OK;
}
VOID SHShouldRegisterActxprxy(void)
{
// published, so invariant
const TCHAR szIIDIOCT[] = TEXT("Interface\\{b722bccb-4e68-101b-a2bc-00aa00404770}\\ProxyStubClsid32");
const TCHAR szIIDIHT[] = TEXT("Interface\\{79eac9c4-baf9-11ce-8c82-00aa004ba90b}\\ProxyStubClsid32");
const TCHAR szACTXPS[] = TEXT("{B8DA6310-E19B-11D0-933C-00A0C90DCAA9}");
TCHAR szValueIOCT[39];
TCHAR szValueIHT[39];
DWORD dwType;
DWORD cbValue;
long rcIOCT, rcIHT;
// Compare the retrieved value with our typelib id.
cbValue = sizeof( szValueIOCT );
rcIOCT = SHGetValue( HKEY_CLASSES_ROOT, szIIDIOCT, NULL, &dwType, szValueIOCT, &cbValue );
cbValue = sizeof( szValueIHT );
rcIHT = SHGetValue( HKEY_CLASSES_ROOT, szIIDIHT, NULL, &dwType, szValueIHT, &cbValue );
if ( rcIOCT // IOleCommandTarget Proxy/Stub CLSID key missing
|| rcIHT // IHlinkTarget Proxy/Stub CLSID key missing
|| StrCmpI( szValueIOCT, szACTXPS ) != 0 // IOleCommandTarget Proxy/Stub CLSID key wrong
|| StrCmpI( szValueIHT, szACTXPS ) != 0 ) // IHlinkTarget Proxy/Stub CLSID key wrong
{
HINSTANCE hinst = LoadLibrary(TEXT("ACTXPRXY.DLL"));
if (hinst)
{
typedef HRESULT (WINAPI * REGSERVERPROC)(void);
REGSERVERPROC pfn = (REGSERVERPROC) GetProcAddress(hinst, "DllRegisterServer");
if (pfn)
pfn();
FreeLibrary(hinst);
}
}
}
VOID SHCheckRegistry(void)
{
// VBE has a bug where they destroy the interface registration information of any control
// hosted in a VBE user form. Check the registry here. Only do this once.
// 17-Nov-97 [alanau/terrylu] Added a check for the IOleCommandTarget Proxy/Stub handler
static BOOL fNeedToCheckRegistry = TRUE;
if( fNeedToCheckRegistry )
{
LONG rc;
// This is published, so is invariant
const TCHAR szIIDWB[] = TEXT("Interface\\{EAB22AC1-30C1-11CF-A7EB-0000C05BAE0B}\\Typelib");
const TCHAR szIIDIOCT[] = TEXT("Interface\\{b722bccb-4e68-101b-a2bc-00aa00404770}\\ProxyStubClsid32");
// This is published, so is invariant
const TCHAR szLIB[] = TEXT("{EAB22AC0-30C1-11CF-A7EB-0000C05BAE0B}");
const TCHAR szACTXLIB[] = TEXT("{B8DA6310-E19B-11D0-933C-00A0C90DCAA9}");
TCHAR szValue[39];
DWORD dwType;
DWORD cbValue;
fNeedToCheckRegistry = FALSE;
cbValue = sizeof( szValue );
rc = SHGetValue( HKEY_CLASSES_ROOT, szIIDWB, NULL, &dwType, szValue, &cbValue );
if ( rc == ERROR_SUCCESS )
{
// Compare the retrieved value with our typelib id.
if ( StrCmpI( szValue, szLIB ) != 0 )
{
// If different, we need to explicitly register our typelib
SHRegisterTypeLib();
}
}
SHShouldRegisterActxprxy();
}
}
STDAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, void **ppv)
{
extern IClassFactory *CInstClassFactory_Create(const CLSID *pInstID);
TraceMsg(TF_SHDLIFE, "DllGetClassObject called with riid=%x (%x)", riid, &riid);
SHCheckRegistry();
if (IsEqualIID(riid, IID_IClassFactory) || IsEqualIID(riid, IID_IUnknown))
{
LPCOBJECTINFO pcls;
for (pcls = g_ObjectInfo; pcls->pclsid; pcls++)
{
if (IsEqualGUID(rclsid, *(pcls->pclsid)))
{
*ppv = (void*)pcls;
DllAddRef(); // class factory holds DLL ref count
return NOERROR;
}
}
// Try the ATL class factory
if (SUCCEEDED(AtlGetClassObject(rclsid, riid, ppv)))
return NOERROR;
// not found, see if it's an 'instance' (code + initialization)
if (*ppv = (void*) CInstClassFactory_Create(&rclsid))
{
// if so, we've now got a 'stub loader' class factory (w/ refcnt=1)
return NOERROR;
}
}
*ppv = NULL;
return CLASS_E_CLASSNOTAVAILABLE;
}
STDAPI DllCanUnloadNow(void)
{
if (0 != g_cRefThisDll || 0 != AtlGetLockCount())
return S_FALSE;
TraceMsg(DM_TRACE, "DllCanUnloadNow returning S_OK (bye, bye...)");
return S_OK;
}
// DllGetVersion - New for IE 4.0
// All we have to do is declare this puppy and CCDllGetVersion does the rest
DLLVER_SINGLEBINARY(VER_PRODUCTVERSION_DW, VER_PRODUCTBUILD_QFE);
#ifdef DEBUG
EXTERN_C
DWORD g_TlsMem = 0xffffffff;
#endif
void InitNFCtl()
{
INITCOMMONCONTROLSEX icc;
icc.dwSize = sizeof(INITCOMMONCONTROLSEX);
icc.dwICC = ICC_NATIVEFNTCTL_CLASS;
InitCommonControlsEx(&icc);
}
// Table of all window classes we register so we can unregister them
// at DLL unload.
const LPCTSTR c_rgszClasses[] = {
c_szViewClass, // dochost.cpp
URLDL_WNDCLASS, // reload.cpp
c_szShellEmbedding, // embed.cpp
TEXT("CIESplashScreen"), // splash.cpp
CWinInetNotify_szWindowClass, // inetnot.cpp
OCHOST_CLASS, // occtrl.cpp
};
// Since we are single-binary, we have to play it safe and do
// this cleanup (needed only on NT, but harmless on Win95).
#define UnregisterWindowClasses() \
SHUnregisterClasses(HINST_THISDLL, c_rgszClasses, ARRAYSIZE(c_rgszClasses))
extern HMODULE g_hmodShfolder;
// IEUNIX - This function should be moved to some file used to create
// shdocvw.dll. While compiling for DLLs mainsoft will #define DllMain
// to the appropriate function being called in generated *_init.c
#if defined(MAINWIN)
STDAPI_(BOOL) DllMain_Internal(HINSTANCE hDll, DWORD dwReason, void *fProcessUnload)
#else
STDAPI_(BOOL) DllMain(HINSTANCE hDll, DWORD dwReason, void *fProcessUnload)
#endif
{
if (dwReason == DLL_PROCESS_ATTACH)
{
AtlInit(hDll);
DisableThreadLibraryCalls(hDll); // perf
g_hinst = hDll;
InitializeCriticalSection(&g_csDll);
MLLoadResources(g_hinst, TEXT("shdoclc.dll"));
// Don't put it under #ifdef DEBUG
CcshellGetDebugFlags();
#ifdef DEBUG
g_TlsMem = TlsAlloc();
if (IsFlagSet(g_dwBreakFlags, BF_ONLOADED))
{
TraceMsg(TF_ALWAYS, "SHDOCVW.DLL has just loaded");
DEBUG_BREAK;
}
#endif
g_fRunningOnNT = IsOS(OS_NT);
if (g_fRunningOnNT)
g_bRunOnNT5 = IsOS(OS_NT5);
else
g_bRunOnMemphis = IsOS(OS_MEMPHIS);
g_fRunOnFE = GetSystemMetrics(SM_DBCSENABLED);
g_uiACP = GetACP();
// Check if the mirroring APIs exist on the current
// platform.
g_bMirroredOS = IS_MIRRORING_ENABLED();
#ifdef WINDOWS_ME
// Check to see if running on BiDi localized Win95
g_bBiDiW95Loc = IsBiDiLocalizedWin95(FALSE);
#endif // WINDOWS_ME
InitNFCtl();
// See if perfmode is enabled
g_dwStopWatchMode = StopWatchMode();
// Cache a palette handle for use throughout shdocvw
g_hpalHalftone = SHCreateShellPalette( NULL );
SHCheckRegistry();
}
else if (dwReason == DLL_PROCESS_DETACH)
{
AtlTerm();
MLFreeResources(g_hinst);
if (g_hMutexHistory)
{
CloseHandle(g_hMutexHistory);
g_hMutexHistory = NULL;
}
if (g_hSemBrowserCount)
CloseHandle(g_hSemBrowserCount);
if (g_hpalHalftone)
DeletePalette( g_hpalHalftone );
if (g_hiconSSL)
DestroyIcon(g_hiconSSL);
if (g_hiconOffline)
DestroyIcon(g_hiconOffline);
if (g_hiconPrinter)
DestroyIcon(g_hiconPrinter);
if (fProcessUnload == NULL)
{
// DLL being unloaded, FreeLibrary() (vs process shutdown)
// at process shutdown time we can't make call outs since
// we don't know if those DLLs will still be loaded!
CUrlHistory_CleanUp();
if (g_psfInternet)
{
// Atomic Release for a C pgm.
IShellFolder *psfInternet = g_psfInternet;
g_psfInternet = NULL;
psfInternet->Release();
}
if (g_hmodShfolder)
{
FreeLibrary(g_hmodShfolder);
}
UnregisterWindowClasses();
FreeDynamicLibraries();
}
DeleteCriticalSection(&g_csDll);
}
return TRUE;
}
STDAPI_(void) DllAddRef(void)
{
InterlockedIncrement(&g_cRefThisDll);
ASSERT(g_cRefThisDll < 1000); // reasonable upper limit
}
STDAPI_(void) DestroyHdpaHooks(); // implemented in url.cpp
STDAPI_(void) DllRelease(void)
{
InterlockedDecrement(&g_cRefThisDll);
ASSERT(g_cRefThisDll >= 0); // don't underflow
if (g_cRefThisDll == 0)
DestroyHdpaHooks();
}
STDAPI_(UINT) WhichPlatformFORWARD()
{
return WhichPlatform();
}
// IEUNIX
// CoCreateInstance is #defined as IECreateInstance #ifdef __cplusplus,
// so I #undef it here to prevent the recursive call.
// On Windows it works, because this file is C file.
#ifdef CoCreateInstance
#undef CoCreateInstance
#endif
HRESULT IECreateInstance(REFCLSID rclsid, IUnknown *pUnkOuter, DWORD dwClsContext, REFIID riid, void **ppv)
{
LPCOBJECTINFO pcls;
#ifndef NO_MARSHALLING
if (dwClsContext == CLSCTX_INPROC_SERVER)
{
#else
if (dwClsContext & CLSCTX_INPROC_SERVER)
{
#endif
for (pcls = g_ObjectInfo; pcls->pclsid; pcls++)
{
// Note that we do pointer comparison (instead of IsEuqalGUID)
if (&rclsid == pcls->pclsid)
{
// const -> non-const expclit casting (this is OK)
IClassFactory* pcf = GET_ICLASSFACTORY(pcls);
return pcf->CreateInstance(pUnkOuter, riid, ppv);
}
}
}
return CoCreateInstance(rclsid, pUnkOuter, dwClsContext, riid, ppv);
}
#ifdef DEBUG
// In DEBUG, make sure every class we register lives in the c_rgszClasses
// table so we can clean up properly at DLL unload. NT does not automatically
// unregister classes when a DLL unloads, so we have to do it manually.
STDAPI_(BOOL) WINAPI SHRegisterClassD(CONST WNDCLASS *pwc)
{
int i;
for (i = 0; i < ARRAYSIZE(c_rgszClasses); i++) {
if (StrCmpI(c_rgszClasses[i], pwc->lpszClassName) == 0) {
return RealSHRegisterClass(pwc);
}
}
AssertMsg(0, TEXT("Class %s needs to be added to the c_rgszClasses list"), pwc->lpszClassName);
return 0;
}
STDAPI_(ATOM) WINAPI RegisterClassD(CONST WNDCLASS *pwc)
{
int i;
for (i = 0; i < ARRAYSIZE(c_rgszClasses); i++) {
if (StrCmpI(c_rgszClasses[i], pwc->lpszClassName) == 0) {
return RealRegisterClass(pwc);
}
}
AssertMsg(0, TEXT("Class %s needs to be added to the c_rgszClasses list"), pwc->lpszClassName);
return 0;
}
// In DEBUG, send FindWindow through a wrapper that ensures that the
// critical section is not taken. FindWindow'ing for a window title
// sends inter-thread WM_GETTEXT messages, which is not obvious.
STDAPI_(HWND) FindWindowD(LPCTSTR lpClassName, LPCTSTR lpWindowName)
{
return FindWindowExD(NULL, NULL, lpClassName, lpWindowName);
}
STDAPI_(HWND) FindWindowExD(HWND hwndParent, HWND hwndChildAfter, LPCTSTR lpClassName, LPCTSTR lpWindowName)
{
if (lpWindowName) {
ASSERTNONCRITICAL;
}
return RealFindWindowEx(hwndParent, hwndChildAfter, lpClassName, lpWindowName);
}
#endif // DEBUG
Reference in New Issue View Git Blame Copy Permalink