You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
229 lines
7.0 KiB
229 lines
7.0 KiB
//+-------------------------------------------------------------------------
|
|
//
|
|
// Microsoft Windows
|
|
//
|
|
// Copyright (C) Microsoft Corporation, 1999 - 1999
|
|
//
|
|
// File: cic.cpp
|
|
//
|
|
//--------------------------------------------------------------------------
|
|
|
|
// cic.cpp : Implementation of DLL Exports.
|
|
|
|
|
|
// Note: Proxy/Stub Information
|
|
// To build a separate proxy/stub DLL,
|
|
// run nmake -f cicps.mk in the project directory.
|
|
|
|
#include "stdafx.h"
|
|
#include "resource.h"
|
|
#include "initguid.h"
|
|
#include "cic.h"
|
|
|
|
#include "cic_i.c"
|
|
#include "MMCCtrl.h"
|
|
#include "MMCTask.h"
|
|
#include "MMClpi.h"
|
|
#include "ListPad.h"
|
|
#include "SysColorCtrl.h"
|
|
|
|
CComModule _Module;
|
|
|
|
BEGIN_OBJECT_MAP(ObjectMap)
|
|
OBJECT_ENTRY(CLSID_MMCCtrl, CMMCCtrl)
|
|
OBJECT_ENTRY(CLSID_MMCTask, CMMCTask)
|
|
OBJECT_ENTRY(CLSID_MMCListPadInfo, CMMCListPadInfo)
|
|
OBJECT_ENTRY(CLSID_ListPad, CListPad)
|
|
OBJECT_ENTRY(CLSID_SysColorCtrl, CSysColorCtrl)
|
|
END_OBJECT_MAP()
|
|
|
|
// cut from ndmgr_i.c (yuck) !!!
|
|
const IID IID_ITaskPadHost = {0x4f7606d0,0x5568,0x11d1,{0x9f,0xea,0x00,0x60,0x08,0x32,0xdb,0x4a}};
|
|
|
|
#ifdef DBG
|
|
CTraceTag tagCicGetClassObject(TEXT("Cic"), TEXT("DllGetClassObject"));
|
|
#endif
|
|
|
|
|
|
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
// DLL Entry Point
|
|
|
|
extern "C"
|
|
BOOL WINAPI DllMain(HINSTANCE hInstance, DWORD dwReason, LPVOID /*lpReserved*/)
|
|
{
|
|
if (dwReason == DLL_PROCESS_ATTACH)
|
|
{
|
|
_Module.Init(ObjectMap, hInstance);
|
|
DisableThreadLibraryCalls(hInstance);
|
|
}
|
|
else if (dwReason == DLL_PROCESS_DETACH)
|
|
_Module.Term();
|
|
return TRUE; // ok
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
// Used to determine whether the DLL can be unloaded by OLE
|
|
|
|
STDAPI DllCanUnloadNow(void)
|
|
{
|
|
return (_Module.GetLockCount()==0) ? S_OK : S_FALSE;
|
|
}
|
|
|
|
|
|
//***************************************************************************
|
|
//
|
|
// ScGetSystemWow64Directory
|
|
//
|
|
// PURPOSE: Calls GetSystemWow64DirectoryW using a late bind, to avoid
|
|
// requiring the XP version of kernel32.dll
|
|
//
|
|
// PARAMETERS:
|
|
// LPTSTR lpBuffer :
|
|
// UINT uSize :
|
|
//
|
|
// RETURNS:
|
|
// SC
|
|
//
|
|
//****************************************************************************
|
|
SC ScGetSystemWow64Directory(LPTSTR lpBuffer, UINT uSize )
|
|
{
|
|
DECLARE_SC(sc, TEXT("ScGetSystemWow64Directory"));
|
|
|
|
sc = ScCheckPointers(lpBuffer);
|
|
if(sc)
|
|
return sc;
|
|
|
|
HMODULE hmod = GetModuleHandle (_T("kernel32.dll"));
|
|
if (hmod == NULL)
|
|
return (sc = E_FAIL);
|
|
|
|
UINT (WINAPI* pfnGetSystemWow64Directory)(LPTSTR, UINT);
|
|
(FARPROC&)pfnGetSystemWow64Directory = GetProcAddress (hmod, "GetSystemWow64DirectoryW");
|
|
|
|
sc = ScCheckPointers(pfnGetSystemWow64Directory, E_FAIL);
|
|
if(sc)
|
|
return sc;
|
|
|
|
if ((pfnGetSystemWow64Directory)(lpBuffer, uSize) == 0)
|
|
return (sc = E_FAIL);
|
|
|
|
return sc;
|
|
}
|
|
|
|
//***************************************************************************
|
|
//
|
|
// DllGetClassObject
|
|
//
|
|
// PURPOSE: Returns a class factory to create an object of the requested type
|
|
// For security reasons, these COM objects can only be instantiated
|
|
// within the context of MMC.EXE. If they are instantiated by any
|
|
// other host, such as IE, they will fail.
|
|
//
|
|
// PARAMETERS:
|
|
// REFCLSID rclsid :
|
|
// REFIID riid :
|
|
// LPVOID* ppv :
|
|
//
|
|
// RETURNS:
|
|
// STDAPI - S_OK if the call succeeds
|
|
//
|
|
//
|
|
//****************************************************************************
|
|
STDAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID* ppv)
|
|
{
|
|
DECLARE_SC(sc, TEXT("CIC.DLL:DllGetClassObject"));
|
|
|
|
TCHAR szFileName[MAX_PATH] = {0};
|
|
DWORD cchFilename = MAX_PATH;
|
|
|
|
// 1. Get the filename for the .exe associated with the process
|
|
DWORD dw = GetModuleFileName(GetModuleHandle(NULL), szFileName, cchFilename);
|
|
|
|
Trace(tagCicGetClassObject, TEXT("Process Filename: %s"), szFileName);
|
|
|
|
if(0==dw)
|
|
return sc.FromLastError().ToHr();
|
|
|
|
// 2. Build the path to where MMC.EXE should be
|
|
const int cchMMCPathName = MAX_PATH;
|
|
TCHAR szMMCPathName[cchMMCPathName] = {0};
|
|
|
|
UINT nLength = GetSystemDirectory(szMMCPathName, cchMMCPathName);
|
|
if(0==nLength)
|
|
return (sc = E_FAIL).ToHr();
|
|
|
|
LPCTSTR szMMC = TEXT("\\MMC.EXE");
|
|
|
|
sc = StringCchCat(szMMCPathName, cchMMCPathName, szMMC);
|
|
if(sc)
|
|
return sc.ToHr();
|
|
|
|
// 3. Canonicalize by converting both paths to long path names
|
|
const DWORD cchLongPath1 = MAX_PATH;
|
|
const DWORD cchLongPath2 = MAX_PATH;
|
|
TCHAR szLongPath1[cchLongPath1], szLongPath2[cchLongPath2];
|
|
|
|
DWORD dw1 = GetLongPathName(szMMCPathName, szLongPath1, cchLongPath1);
|
|
if(0==dw1)
|
|
return sc.FromLastError().ToHr();
|
|
|
|
DWORD dw2 = GetLongPathName(szFileName, szLongPath2, cchLongPath2);
|
|
if(0==dw2)
|
|
return sc.FromLastError().ToHr();
|
|
|
|
// 4. Compare (case-insensitive) both parts to ensure that they are the same.
|
|
// If they are not, some other .exe is trying to instantiate an object. Do
|
|
// not allow this.
|
|
Trace(tagCicGetClassObject, TEXT("Comparing %s to %s"), szLongPath1, szLongPath2);
|
|
if(0 != _tcsicmp(szLongPath1, szLongPath2))
|
|
{
|
|
// try one more test (in case this is a 64-bit machine) - check for the SysWow64 directory
|
|
const int cchMMCSysWow64PathName = MAX_PATH;
|
|
TCHAR szMMCSysWow64PathName[cchMMCSysWow64PathName] = {0};
|
|
|
|
sc = ScGetSystemWow64Directory(szMMCSysWow64PathName, cchMMCSysWow64PathName);
|
|
if(sc)
|
|
return sc.ToHr();
|
|
|
|
sc = StringCchCat(szMMCSysWow64PathName, cchMMCSysWow64PathName, szMMC);
|
|
if(sc)
|
|
return sc.ToHr();
|
|
|
|
const DWORD cchLongPathSysWow64 = MAX_PATH;
|
|
TCHAR szLongPathSysWow64[cchLongPathSysWow64] = {0};
|
|
|
|
DWORD dw3 = GetLongPathName(szMMCSysWow64PathName, szLongPathSysWow64, cchLongPathSysWow64);
|
|
if(0==dw3)
|
|
return sc.FromLastError().ToHr();
|
|
|
|
Trace(tagCicGetClassObject, TEXT("Comparing %s to %s"), szLongPathSysWow64, szLongPath2);
|
|
if(0 != _tcsicmp(szLongPath2, szLongPathSysWow64))
|
|
{
|
|
Trace(tagCicGetClassObject, TEXT("Invalid exe - must be %s or %s. Did not instantiate object."), szMMCPathName, szMMCSysWow64PathName);
|
|
return (sc = CLASS_E_CLASSNOTAVAILABLE).ToHr();
|
|
}
|
|
}
|
|
|
|
return _Module.GetClassObject(rclsid, riid, ppv);
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
// DllRegisterServer - Adds entries to the system registry
|
|
|
|
STDAPI DllRegisterServer(void)
|
|
{
|
|
// registers object, typelib and all interfaces in typelib
|
|
return _Module.RegisterServer(TRUE);
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
// DllUnregisterServer - Removes entries from the system registry
|
|
|
|
STDAPI DllUnregisterServer(void)
|
|
{
|
|
_Module.UnregisterServer();
|
|
return S_OK;
|
|
}
|
|
|
|
|