|
|
///////////////////////////////////////////////////////////////////////////////
/* File: dskquota.cpp
Description: Contains standard functions for an OLE component server DLL.
DllMain DllGetClassObject DllCanUnloadNow
Revision History:
Date Description Programmer -------- --------------------------------------------------- ---------- 05/22/96 Initial creation. BrianAu 12/10/96 Moved to free-threading OLE apartment model. BrianAu*////////////////////////////////////////////////////////////////////////////////
#include "pch.h" // PCH
#pragma hdrstop
#define INITGUIDS // Define GUIDs.
#include "dskquota.h"
#include "guidsp.h" // Private GUIDs.
#include <gpedit.h> // For GUIDs.
#include "factory.h" // Class factory declarations.
#include "sidcache.h" // SID/Name cache.
#include "registry.h"
//
// Verify that build is UNICODE.
//
#if !defined(UNICODE)
# error This module must be compiled UNICODE.
#endif
HINSTANCE g_hInstDll = NULL; // DLL instance handle.
LONG g_cRefThisDll = 0; // DLL reference count.
LONG g_cLockThisDll = 0; // DLL lock count.
///////////////////////////////////////////////////////////////////////////////
/* Function: DllGetClassObject
Description: Creates instance of DiskQuotaControlClassFactory.
Arguments: rclsid - Reference to class ID that identifies the type of object that the class factory will be asked to create.
riid - Reference to interface ID on the class factory object.
ppvOut - Destination location for class factory object pointer after instantiation.
Returns: NOERROR - Success. E_OUTOFMEMORY - Can't create class factory object. E_NOINTERFACE - Interface not supported. E_INVALIDARG - ppvOut arg is NULL. CLASS_E_CLASSNOTAVAILABLE - Class factory not available.
Revision History:
Date Description Programmer -------- --------------------------------------------------- ---------- 05/22/96 Initial creation. BrianAu*////////////////////////////////////////////////////////////////////////////////
STDAPI DllGetClassObject( REFCLSID rclsid, REFIID riid, LPVOID *ppvOut ){ DBGTRACE((DM_COM, DL_HIGH, TEXT("DllGetClassObject"))); DBGPRINTIID(DM_COM, DL_HIGH, (REFIID)rclsid); DBGPRINTIID(DM_COM, DL_HIGH, riid);
HRESULT hr = CLASS_E_CLASSNOTAVAILABLE;
if (NULL == ppvOut) return E_INVALIDARG;
*ppvOut = NULL;
try { if (IsEqualIID(rclsid, CLSID_DiskQuotaControl)) { DiskQuotaControlClassFactory *pClassFactory = NULL;
pClassFactory = new DiskQuotaControlClassFactory; hr = pClassFactory->QueryInterface(riid, ppvOut); if (FAILED(hr)) { delete pClassFactory; } } } catch(CAllocException& e) { DBGERROR((TEXT("Insufficient memory exception"))); hr = E_OUTOFMEMORY; } return hr;}
///////////////////////////////////////////////////////////////////////////////
/* Function: DllCanUnloadNow
Description: Called by OLE to determine if DLL can be unloaded.
Arguments: None.
Returns: S_FALSE - Can't unload. Ref count or lock count are > 0. S_OK - OK to unload. Ref count and lock count are 0.
Revision History:
Date Description Programmer -------- --------------------------------------------------- ---------- 05/22/96 Initial creation. BrianAu*////////////////////////////////////////////////////////////////////////////////
STDAPI DllCanUnloadNow( VOID ){ DBGTRACE((DM_COM, DL_HIGH, TEXT("DllCanUnloadNow"))); DBGPRINT((DM_COM, DL_HIGH, TEXT("\tRefCnt = %d LockCnt = %d"), g_cRefThisDll, g_cLockThisDll));
#ifdef DEBUG_PROCESS_DETACH
//
// This code will force the DLL to be unloaded so that we can
// test OnProcessDetach(). Otherwise, OLE will wait 10 minutes
// after DllCanUnloadNow returns S_OK to call FreeLibrary.
// Note however that the process will AV when OLE finally gets
// around to freeing the library. Therefore, we don't leave
// this code active in free or check builds; only private debug
// builds. Must explicitly define DEBUG_PROCESS_DETACH.
//
if (0 == g_cRefThisDll && 0 == g_cLockThisDll) { HINSTANCE hMod; static INT iExeThisPath = 0; if (0 == iExeThisPath++) { hMod = LoadLibrary(TEXT("dskquota.dll")); if (NULL != hMod) { DBGASSERT((g_hInstDll == hMod)); FreeLibrary(hMod); FreeLibrary(hMod); } } }#endif // DEBUG_PROCESS_DETACH
return (0 == g_cRefThisDll && 0 == g_cLockThisDll) ? S_OK : S_FALSE;}
///////////////////////////////////////////////////////////////////////////////
/* Function: DllRegisterServer
Description: Create the necessary registry entries for dskquota.dll to operate properly. This is called by REGSVR32.EXE.
Arguments: None.
Returns: S_OK - Succeeded. SELFREG_E_CLASS - Failed to create one of the registry entries.
Revision History:
Date Description Programmer -------- --------------------------------------------------- ---------- 08/18/97 Initial creation. BrianAu*////////////////////////////////////////////////////////////////////////////////
HRESULTDllRegisterServer( VOID ){ DBGTRACE((DM_COM, DL_HIGH, TEXT("DllRegisterServer"))); HRESULT hr = CallRegInstall(g_hInstDll, "RegDll");
if (FAILED(hr)) { hr = SELFREG_E_CLASS; } return hr;}
///////////////////////////////////////////////////////////////////////////////
/* Function: DllUnregisterServer
Description: Remove the necessary registry entries for dskquota.dll. This is called by REGSVR32.EXE.
Arguments: None.
Returns: S_OK - Succeeded. SELFREG_E_CLASS - Failed to remove the CLSID entry.
Revision History:
Date Description Programmer -------- --------------------------------------------------- ---------- 08/18/97 Initial creation. BrianAu*////////////////////////////////////////////////////////////////////////////////
HRESULTDllUnregisterServer( VOID ){ DBGTRACE((DM_COM, DL_HIGH, TEXT("DllUnregisterServer"))); HRESULT hr = CallRegInstall(g_hInstDll, "UnregDll");
if (FAILED(hr)) { hr = SELFREG_E_CLASS; } return hr;}
///////////////////////////////////////////////////////////////////////////////
/* Function: OnProcessAttach
Description: Handles all tasks associated with a process attaching to the DLL.
Try to keep processing time to a minimum.
Arguments: hInstDll - The DLL instance handle passed to DllMain.
Returns: NOERROR - Success. E_FAIL - Something failed.
Revision History:
Date Description Programmer -------- --------------------------------------------------- ---------- 08/09/96 Initial creation. BrianAu*////////////////////////////////////////////////////////////////////////////////
HRESULTOnProcessAttach( HINSTANCE hInstDll ){ DBGTRACE((DM_COM, DL_HIGH, TEXT("OnProcessAttach"))); HRESULT hr = NOERROR;
//
// Start IceCAP profiling.
//
ICAP_START_ALL;
#if DBG
DBGMODULE(TEXT("DSKQUOTA")); // Name of module displayed with messages.
RegKey key(HKEY_LOCAL_MACHINE, REGSTR_KEY_DISKQUOTA); if (SUCCEEDED(key.Open(KEY_READ))) { DebugRegParams dp; if (SUCCEEDED(key.GetValue(REGSTR_VAL_DEBUGPARAMS, (LPBYTE)&dp, sizeof(dp)))) { DBGPRINTMASK(dp.PrintMask); DBGPRINTLEVEL(dp.PrintLevel); DBGPRINTVERBOSE(dp.PrintVerbose); DBGTRACEMASK(dp.TraceMask); DBGTRACELEVEL(dp.TraceLevel); DBGTRACEVERBOSE(dp.TraceVerbose); DBGTRACEONEXIT(dp.TraceOnExit); } }#endif // DBG
g_hInstDll = hInstDll; DisableThreadLibraryCalls(hInstDll);
return hr;}
///////////////////////////////////////////////////////////////////////////////
/* Function: OnProcessDetach
Description: Handles all tasks associated with a process detaching from the DLL.
Arguments: None.
Returns: NOERROR - Success. Revision History:
Date Description Programmer -------- --------------------------------------------------- ---------- 08/09/96 Initial creation. BrianAu*////////////////////////////////////////////////////////////////////////////////
HRESULTOnProcessDetach( VOID ){ DBGTRACE((DM_COM, DL_HIGH, TEXT("OnProcessAttach")));
HRESULT hr = NOERROR;
#ifdef DEBUG_PROCESS_DETACH
LoadLibrary(TEXT("dskquota.dll"));#endif
SidNameCache_Destroy();
//
// Stop IceCAP profiling.
//
ICAP_STOP_ALL;
return hr;}
///////////////////////////////////////////////////////////////////////////////
/* Function: DllMain
Description: Main entry point for OLE component server.
Arguments: hInstDll - Instance handle of DLL
fdwReason - Reason DllMain is being called. Can be at Process attach/ detach or Thread attach/detach.
lpdwReserved - Reserved.
Returns: TRUE - Successful initialization. FALSE - Failed initialization.
Revision History:
Date Description Programmer -------- --------------------------------------------------- ---------- 05/22/96 Initial creation. BrianAu 08/09/96 Moved code associated with process attach and BrianAu detach out to separate functions.*////////////////////////////////////////////////////////////////////////////////
BOOL WINAPI DllMain( HINSTANCE hInstDll, DWORD fdwReason, LPVOID lpvReserved ){ DBGTRACE((DM_COM, DL_HIGH, TEXT("DllMain"))); BOOL bResult = FALSE;
switch(fdwReason) { case DLL_PROCESS_ATTACH: DBGPRINT((DM_COM, DL_HIGH, TEXT("DLL_PROCESS_ATTACH"))); bResult = SUCCEEDED(OnProcessAttach(hInstDll)); break;
case DLL_THREAD_ATTACH: case DLL_THREAD_DETACH: bResult = TRUE; break;
case DLL_PROCESS_DETACH: DBGPRINT((DM_COM, DL_HIGH, TEXT("DLL_PROCESS_DETACH"))); bResult = SUCCEEDED(OnProcessDetach()); break; }
return bResult;}
|
