|
|
// Mdsp.cpp : Implementation of DLL Exports.
// Note: Proxy/Stub Information
// To merge the proxy/stub code into the object DLL, add the file
// dlldatax.c to the project. Make sure precompiled headers
// are turned off for this file, and add _MERGE_PROXYSTUB to the
// defines for the project.
//
// If you are not running WinNT4.0 or Win95 with DCOM, then you
// need to remove the following define from dlldatax.c
// #define _WIN32_WINNT 0x0400
//
// Further, if you are running MIDL without /Oicf switch, you also
// need to remove the following define from dlldatax.c.
// #define USE_STUBLESS_PROXY
//
// Modify the custom build rule for Mdsp.idl by adding the following
// files to the Outputs.
// Mdsp_p.c
// dlldata.c
// To build a separate proxy/stub DLL,
// run nmake -f Mdspps.mk in the project directory.
#include "stdafx.h"
#include "resource.h"
#include "initguid.h"
#include "MsPMSP.h"
#include "dlldatax.h"
#include "MsPMSP_i.c"
#include "MDServiceProvider.h"
//#include "MDSPEnumDevice.h"
//#include "MDSPDevice.h"
//#include "MDSPEnumStorage.h"
//#include "MDSPStorage.h"
//#include "MDSPStorageGlobals.h"
#include "MdspDefs.h"
#include "scserver.h"
#ifdef _MERGE_PROXYSTUB
extern "C" HINSTANCE hProxyDll; #endif
CComModule _Module; DWORD g_dwStartDrive; HINSTANCE g_hinstance; MDSPNOTIFYINFO g_NotifyInfo[MDSP_MAX_DEVICE_OBJ]; MDSPGLOBALDEVICEINFO g_GlobalDeviceInfo[MDSP_MAX_DEVICE_OBJ]; WCHAR g_wcsBackslash[2] = { (WCHAR)0x5c, NULL }; CHAR g_szBackslash[2] = {(CHAR)0x5c, NULL }; CComMultiThreadModel::AutoCriticalSection g_CriticalSection; CSecureChannelServer *g_pAppSCServer=NULL; BOOL g_bIsWinNT; BEGIN_OBJECT_MAP(ObjectMap) OBJECT_ENTRY(CLSID_MDServiceProvider, CMDServiceProvider) // OBJECT_ENTRY(CLSID_MDSPEnumDevice, CMDSPEnumDevice)
// OBJECT_ENTRY(CLSID_MDSPDevice, CMDSPDevice)
// OBJECT_ENTRY(CLSID_MDSPEnumFormatSupport, CMDSPEnumFormatSupport)
// OBJECT_ENTRY(CLSID_MDSPEnumStorage, CMDSPEnumStorage)
// OBJECT_ENTRY(CLSID_MDSPStorage, CMDSPStorage)
// OBJECT_ENTRY(CLSID_MDSPStorageGlobals, CMDSPStorageGlobals)
END_OBJECT_MAP()
/////////////////////////////////////////////////////////////////////////////
// DLL Entry Point
extern "C" BOOL WINAPI DllMain(HINSTANCE hInstance, DWORD dwReason, LPVOID lpReserved) { lpReserved; #ifdef _MERGE_PROXYSTUB
if (!PrxDllMain(hInstance, dwReason, lpReserved)) return FALSE; #endif
if (dwReason == DLL_PROCESS_ATTACH) { _Module.Init(ObjectMap, hInstance); DisableThreadLibraryCalls(hInstance); } else if (dwReason == DLL_PROCESS_DETACH) _Module.Term();
g_hinstance = hInstance; return TRUE; // ok
}
/////////////////////////////////////////////////////////////////////////////
// Used to determine whether the DLL can be unloaded by OLE
STDAPI DllCanUnloadNow(void) { #ifdef _MERGE_PROXYSTUB
if (PrxDllCanUnloadNow() != S_OK) return S_FALSE; #endif
return (_Module.GetLockCount()==0) ? S_OK : S_FALSE; }
/////////////////////////////////////////////////////////////////////////////
// Returns a class factory to create an object of the requested type
STDAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID* ppv) { #ifdef _MERGE_PROXYSTUB
if (PrxDllGetClassObject(rclsid, riid, ppv) == S_OK) return S_OK; #endif
return _Module.GetClassObject(rclsid, riid, ppv); }
/////////////////////////////////////////////////////////////////////////////
// DllRegisterServer - Adds entries to the system registry
STDAPI DllRegisterServer(void) { #ifdef _MERGE_PROXYSTUB
HRESULT hRes = PrxDllRegisterServer(); if (FAILED(hRes)) return hRes; #endif
char szTemp[MAX_PATH]; HKEY hKey;
if( !RegCreateKeyEx(HKEY_LOCAL_MACHINE, STR_MDSPREG, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_READ | KEY_WRITE, NULL, &hKey, NULL) ) { strcpy(szTemp, STR_MDSPPROGID); RegSetValueEx( hKey, "ProgID", 0, REG_SZ, (LPBYTE)szTemp, sizeof(szTemp)); #ifdef MDSP_TEMP
DWORD dwSD=1; RegSetValueEx( hKey, "StartDrive", 0, REG_DWORD, (LPBYTE)&dwSD, sizeof(DWORD)); #endif
RegCloseKey(hKey); // registers object, typelib and all interfaces in typelib
return _Module.RegisterServer(TRUE); } else return REGDB_E_WRITEREGDB;
}
/////////////////////////////////////////////////////////////////////////////
// DllUnregisterServer - Removes entries from the system registry
STDAPI DllUnregisterServer(void) { #ifdef _MERGE_PROXYSTUB
PrxDllUnregisterServer(); #endif
_Module.UnregisterServer();
RegDeleteKey(HKEY_LOCAL_MACHINE, STR_MDSPREG);
return S_OK; }
|