///////////////////////////////////////////////////////////////////////////////
/* 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
*/
///////////////////////////////////////////////////////////////////////////////
HRESULT
DllRegisterServer(
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
*/
///////////////////////////////////////////////////////////////////////////////
HRESULT
DllUnregisterServer(
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
*/
///////////////////////////////////////////////////////////////////////////////
HRESULT
OnProcessAttach(
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
*/
///////////////////////////////////////////////////////////////////////////////
HRESULT
OnProcessDetach(
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;
}