/***************************************************************************
*
* Copyright (C) 2001-2002 Microsoft Corporation. All Rights Reserved.
*
* File: dpnhpastdllmain.cpp
*
* Content: DPNHPAST DLL entry points.
*
* History:
* Date By Reason
* ======== ======== =========
* 04/16/01 VanceO Split DPNATHLP into DPNHUPNP and DPNHPAST.
*
***************************************************************************/
#include "dpnhpasti.h"
//=============================================================================
// External globals
//=============================================================================
volatile LONG g_lOutstandingInterfaceCount = 0; // number of outstanding interfaces
DNCRITICAL_SECTION g_csGlobalsLock; // lock protecting all of the following globals
CBilink g_blNATHelpPASTObjs; // bilink of all the NATHelpPAST interface objects
DWORD g_dwHoldRand; // current random number sequence
DWORD g_dwPASTICSMode; // whether using PAST ICS is enabled or not
DWORD g_dwPASTPFWMode; // whether using PAST PFW is enabled or not
INT g_iUnicastTTL = 1; // unicast TTL value, or 0 to use default set by OS; normally we use 1
DWORD g_dwDefaultGatewayV4 = INADDR_BROADCAST; // default to broadcasting when searching for PAST server
DWORD g_dwSubnetMaskV4 = 0x00FFFFFF; // = 0xFFFFFF00 in Intel order = 255.255.255.0, a class C network
DWORD g_dwNoActiveNotifyPollInterval = 25000; // start by polling every 25 seconds
DWORD g_dwMinUpdateServerStatusInterval = 1000; // don't hit the network more often than every 1 second
DWORD g_dwPollIntervalBackoff = 30000; // backoff an additional 0 to 30 seconds if no network changes occur
DWORD g_dwMaxPollInterval = 300000; // don't go more than 5 minutes without polling
BOOL g_fKeepPollingForRemoteGateway = FALSE; // whether to continue to search for new Internet gateway devices if none were found during startup
DWORD g_dwReusePortTime = 60000; // how long to keep using the same port for polling remote Internet gateway devices (default is 1 minute)
DWORD g_dwCacheLifeFound = 30000; // how long to cache QueryAddress results where the address was found
DWORD g_dwCacheLifeNotFound = 30000; // how long to cache QueryAddress results where the address was not found
//=============================================================================
// Defines
//=============================================================================
#define REGKEY_VALUE_GUID L"Guid"
#define REGKEY_VALUE_DIRECTPLAY8PRIORITY L"DirectPlay8Priority"
#define REGKEY_VALUE_DIRECTPLAY8INITFLAGS L"DirectPlay8InitFlags"
#define REGKEY_VALUE_PASTICSMODE L"PASTICSMode"
#define REGKEY_VALUE_PASTPFWMODE L"PASTPFWMode"
#define REGKEY_VALUE_UNICASTTTL L"UnicastTTL"
#define REGKEY_VALUE_DEFAULTGATEWAYV4 L"GatewayV4"
#define REGKEY_VALUE_SUBNETMASKV4 L"SubnetMaskV4"
#define REGKEY_VALUE_NOACTIVENOTIFYPOLLINTERVAL L"NoActiveNotifyPollInterval"
#define REGKEY_VALUE_MINUPDATESERVERSTATUSINTERVAL L"MinUpdateServerStatusInterval"
#define REGKEY_VALUE_POLLINTERVALBACKOFF L"PollIntervalBackoff"
#define REGKEY_VALUE_MAXPOLLINTERVAL L"MaxPollInterval"
#define REGKEY_VALUE_KEEPPOLLINGFORREMOTEGATEWAY L"KeepPollingForRemoteGateway"
#define REGKEY_VALUE_REUSEPORTTIME L"ReusePortTime"
#define REGKEY_VALUE_CACHELIFEFOUND L"CacheLifeFound"
#define REGKEY_VALUE_CACHELIFENOTFOUND L"CacheLifeNotFound"
#define DEFAULT_DIRECTPLAY8PRIORITY 2
#define DEFAULT_DIRECTPLAY8INITFLAGS DPNHINITIALIZE_DISABLELOCALFIREWALLSUPPORT
//=============================================================================
// Local prototypes
//=============================================================================
BOOL InitializeProcessGlobals(void);
void CleanupProcessGlobals(void);
void InitializeGlobalRand(const DWORD dwSeed);
#undef DPF_MODNAME
#define DPF_MODNAME "DllMain"
//=============================================================================
// DllMain
//-----------------------------------------------------------------------------
//
// Description: DLL entry point.
//
// Arguments:
// HANDLE hDllInst - Handle to this DLL module.
// DWORD dwReason - Reason for calling this function.
// LPVOID lpvReserved - Reserved.
//
// Returns: TRUE if all goes well, FALSE otherwise.
//=============================================================================
BOOL WINAPI DllMain(HANDLE hDllInst,
DWORD dwReason,
LPVOID lpvReserved)
{
BOOL fResult = TRUE;
switch (dwReason)
{
case DLL_PROCESS_ATTACH:
{
DPFX(DPFPREP, 2, "====> ENTER: DLLMAIN(%p): Process Attach: %08lx, tid=%08lx",
DllMain, GetCurrentProcessId(), GetCurrentThreadId());
//
// Ignore thread attach/detach messages.
//
DisableThreadLibraryCalls((HMODULE) hDllInst);
//
// Attempt to initialize the OS abstraction layer.
//
if (DNOSIndirectionInit(0))
{
//
// Attempt to initialize process-global items.
//
if (! InitializeProcessGlobals())
{
DPFX(DPFPREP, 0, "Failed to initialize globals!");
DNOSIndirectionDeinit();
fResult = FALSE;
}
}
else
{
DPFX(DPFPREP, 0, "Failed to initialize OS indirection layer!");
fResult = FALSE;
}
break;
}
case DLL_THREAD_ATTACH:
{
//
// Ignore.
//
break;
}
case DLL_THREAD_DETACH:
{
//
// Ignore.
//
break;
}
case DLL_PROCESS_DETACH:
{
DPFX(DPFPREP, 2, "====> EXIT: DLLMAIN(%p): Process Detach %08lx, tid=%08lx",
DllMain, GetCurrentProcessId(), GetCurrentThreadId());
CleanupProcessGlobals();
DNOSIndirectionDeinit();
break;
}
default:
{
DNASSERT(FALSE);
break;
}
}
return fResult;
} // DllMain
#ifndef DPNBUILD_NOCOMREGISTER
#undef DPF_MODNAME
#define DPF_MODNAME "DllRegisterServer"
//=============================================================================
// DllRegisterServer
//-----------------------------------------------------------------------------
//
// Description: Registers the DirectPlay NAT Helper PAST COM object.
//
// Arguments: None.
//
// Returns: HRESULT
// S_OK - Successfully unregistered DirectPlay NAT Helper PAST.
// E_FAIL - Failed unregistering DirectPlay NAT Helper PAST.
//=============================================================================
HRESULT WINAPI DllRegisterServer(void)
{
CRegistry RegObject;
//
// Register this COM object CLSID.
//
if (! CRegistry::Register(L"DirectPlayNATHelperPAST.1",
L"DirectPlay NAT Helper PAST Object",
L"dpnhpast.dll",
&CLSID_DirectPlayNATHelpPAST,
L"DirectPlayNATHelperPAST"))
{
DPFX(DPFPREP, 0, "Could not register DirectPlay NAT Helper PAST object!");
return E_FAIL;
}
//
// Write this object's GUID and DirectPlay8 availability to the registry.
//
if (! RegObject.Open(HKEY_LOCAL_MACHINE, DIRECTPLAYNATHELP_REGKEY L"\\" REGKEY_COMPONENTSUBKEY, FALSE, TRUE))
{
DPFX(DPFPREP, 0, "Couldn't create DirectPlay NAT Helper PAST key!");
return E_FAIL;
}
if (! RegObject.WriteGUID(REGKEY_VALUE_GUID, CLSID_DirectPlayNATHelpPAST))
{
DPFX(DPFPREP, 0, "Couldn't write GUID to registry!");
return E_FAIL;
}
if (! RegObject.WriteDWORD(REGKEY_VALUE_DIRECTPLAY8PRIORITY, DEFAULT_DIRECTPLAY8PRIORITY))
{
DPFX(DPFPREP, 0, "Couldn't write DirectPlay8 priority to registry!");
return E_FAIL;
}
if (! RegObject.WriteDWORD(REGKEY_VALUE_DIRECTPLAY8INITFLAGS, DEFAULT_DIRECTPLAY8INITFLAGS))
{
DPFX(DPFPREP, 0, "Couldn't write DirectPlay8 init flags to registry!");
return E_FAIL;
}
RegObject.Close();
return S_OK;
} // DllRegisterServer
#undef DPF_MODNAME
#define DPF_MODNAME "DllUnregisterServer"
//=============================================================================
// DllUnregisterServer
//-----------------------------------------------------------------------------
//
// Description: Unregisters the DirectPlay NAT Helper PAST COM object.
//
// Arguments: None.
//
// Returns: HRESULT
// S_OK - Successfully unregistered DirectPlay NAT Helper PAST.
// E_FAIL - Failed unregistering DirectPlay NAT Helper PAST.
//=============================================================================
STDAPI DllUnregisterServer(void)
{
CRegistry RegObject;
//
// Unregister the class.
//
if (! CRegistry::UnRegister(&CLSID_DirectPlayNATHelpPAST))
{
DPFX(DPFPREP, 0, "Failed to unregister DirectPlay NAT Helper PAST object!");
return E_FAIL;
}
//
// Try removing all the subitems we registered.
//
if (! RegObject.Open(HKEY_LOCAL_MACHINE, DIRECTPLAYNATHELP_REGKEY L"\\" REGKEY_COMPONENTSUBKEY, FALSE, FALSE))
{
DPFX(DPFPREP, 0, "Couldn't open DirectPlay NAT Helper key! Ignoring.");
}
else
{
if (! RegObject.DeleteValue(REGKEY_VALUE_GUID))
{
DPFX(DPFPREP, 0, "Couldn't delete GUID registry value! Ignoring.");
}
if (! RegObject.DeleteValue(REGKEY_VALUE_DIRECTPLAY8PRIORITY))
{
DPFX(DPFPREP, 0, "Couldn't delete DirectPlay8 priority registry value! Ignoring.");
}
if (! RegObject.DeleteValue(REGKEY_VALUE_DIRECTPLAY8INITFLAGS))
{
DPFX(DPFPREP, 0, "Couldn't delete DirectPlay8 init flags registry value! Ignoring.");
}
RegObject.Close();
}
return S_OK;
} // DllUnregisterServer
#endif // !DPNBUILD_NOCOMREGISTER
#ifndef WINCE
#undef DPF_MODNAME
#define DPF_MODNAME "DirectPlayNATHelpCreate"
//=============================================================================
// DirectPlayNATHelpCreate
//-----------------------------------------------------------------------------
//
// Description: Creates an IDirectPlayNATHelp interface object.
//
// Arguments:
// GUID * pIID - Pointer to IDirectPlayNATHelp interface GUID.
// void ** ppvInterface - Place to store pointer to interface object
// created.
//
// Returns: HRESULT
// DPNH_OK - Creating the object was successful.
// DPNHERR_INVALIDPOINTER - The destination pointer is invalid.
// DPNHERR_OUTOFMEMORY - Not enough memory to create the object.
// E_NOINTERFACE - The requested interface was invalid.
//=============================================================================
HRESULT WINAPI DirectPlayNATHelpCreate(const GUID * pIID, void ** ppvInterface)
{
HRESULT hr;
DPFX(DPFPREP, 2, "Parameters: (0x%p, 0x%p)", pIID, ppvInterface);
hr = DoCreateInstance(NULL, // no class factory object necessary
NULL, // ?
CLSID_DirectPlayNATHelpPAST, // DirectPlayNATHelp class
(*pIID), // requested interface
ppvInterface); // place to store interface
DPFX(DPFPREP, 2, "Returning: [0x%lx]", hr);
return hr;
} // DirectPlayNATHelpCreate
#endif // !WINCE
#undef DPF_MODNAME
#define DPF_MODNAME "InitializeProcessGlobals"
//=============================================================================
// InitializeProcessGlobals
//-----------------------------------------------------------------------------
//
// Description: Initialize global items needed for the DLL to operate.
//
// Arguments: None.
//
// Returns: TRUE if successful, FALSE if an error occurred.
//=============================================================================
BOOL InitializeProcessGlobals(void)
{
BOOL fReturn = TRUE;
if (! DNInitializeCriticalSection(&g_csGlobalsLock))
{
fReturn = FALSE;
}
//
// Don't allow critical section reentry.
//
DebugSetCriticalSectionRecursionCount(&g_csGlobalsLock, 0);
g_blNATHelpPASTObjs.Initialize();
//
// Seed the random number generator with the current time.
//
InitializeGlobalRand(GETTIMESTAMP());
return fReturn;
} // InitializeProcessGlobals
#undef DPF_MODNAME
#define DPF_MODNAME "CleanupProcessGlobals"
//=============================================================================
// CleanupProcessGlobals
//-----------------------------------------------------------------------------
//
// Description: Releases global items used by DLL.
//
// Arguments: None.
//
// Returns: None.
//=============================================================================
void CleanupProcessGlobals(void)
{
CBilink * pBilink;
CNATHelpPAST * pNATHelpPAST;
if (! g_blNATHelpPASTObjs.IsEmpty())
{
//
// This assert is far more descriptive than hitting one of those in the
// cleanup code that complain about flags incorrectly being set.
//
DNASSERT(! "DPNHPAST.DLL unloading without all objects having been released! The caller's DirectPlayNATHelpPAST cleanup code needs to be fixed!");
//
// Force close all the objects still outstanding.
//
pBilink = g_blNATHelpPASTObjs.GetNext();
while (pBilink != &g_blNATHelpPASTObjs)
{
pNATHelpPAST = NATHELPPAST_FROM_BILINK(pBilink);
pBilink = pBilink->GetNext();
DPFX(DPFPREP, 0, "Forcefully releasing object 0x%p!", pNATHelpPAST);
pNATHelpPAST->Close(0); // ignore error
//
// Forcefully remove it from the list and delete it instead of
// using pNATHelpPAST->Release().
//
pNATHelpPAST->m_blList.RemoveFromList();
pNATHelpPAST->UninitializeObject();
delete pNATHelpPAST;
}
}
DNDeleteCriticalSection(&g_csGlobalsLock);
} // CleanupProcessGlobals
#undef DPF_MODNAME
#define DPF_MODNAME "ReadRegistrySettings"
//=============================================================================
// ReadRegistrySettings
//-----------------------------------------------------------------------------
//
// Description: Reads registry settings to override behavior of this DLL and to
// turn on some debugging features.
//
// Arguments: None.
//
// Returns: None.
//=============================================================================
void ReadRegistrySettings(void)
{
CRegistry RegObject;
DWORD dwNewValue;
BOOL fNewValue;
//
// Try opening the registry key.
//
if (RegObject.Open(HKEY_LOCAL_MACHINE, DIRECTPLAYNATHELP_REGKEY L"\\" REGKEY_COMPONENTSUBKEY) != FALSE)
{
//
// Lock out other interfaces from modifying the globals simultaneously.
//
DNEnterCriticalSection(&g_csGlobalsLock);
//
// If we successfully read a new mode, save it.
//
if (RegObject.ReadDWORD(REGKEY_VALUE_PASTICSMODE, &dwNewValue))
{
g_dwPASTICSMode = dwNewValue;
DPFX(DPFPREP, 1, "Using PAST ICS mode %u.", g_dwPASTICSMode);
}
//
// If we successfully read a new mode, save it.
//
if (RegObject.ReadDWORD(REGKEY_VALUE_PASTPFWMODE, &dwNewValue))
{
g_dwPASTPFWMode = dwNewValue;
DPFX(DPFPREP, 1, "Using PAST PFW mode %u.", g_dwPASTPFWMode);
}
//
// If we successfully read a new value, save it.
//
if (RegObject.ReadDWORD(REGKEY_VALUE_UNICASTTTL, &dwNewValue))
{
g_iUnicastTTL = dwNewValue;
if (g_iUnicastTTL != 0)
{
DPFX(DPFPREP, 1, "Using unicast TTL of %i.", g_iUnicastTTL);
}
else
{
DPFX(DPFPREP, 1, "Using OS default unicast TTL.");
}
}
//
// If we successfully read a new default gateway, save it.
//
if (RegObject.ReadDWORD(REGKEY_VALUE_DEFAULTGATEWAYV4, &dwNewValue))
{
g_dwDefaultGatewayV4 = dwNewValue;
DPFX(DPFPREP, 1, "Using default gateway 0x%08lx.", g_dwDefaultGatewayV4);
}
//
// If we successfully read a new mask, save it.
//
if (RegObject.ReadDWORD(REGKEY_VALUE_SUBNETMASKV4, &dwNewValue))
{
g_dwSubnetMaskV4 = dwNewValue;
DPFX(DPFPREP, 1, "Using subnet mask 0x%08lx.", g_dwSubnetMaskV4);
}
//
// If we successfully read a new interval, save it.
//
if (RegObject.ReadDWORD(REGKEY_VALUE_NOACTIVENOTIFYPOLLINTERVAL, &dwNewValue))
{
g_dwNoActiveNotifyPollInterval = dwNewValue;
DPFX(DPFPREP, 1, "Using no-active-notify recommended poll interval %u ms.", g_dwNoActiveNotifyPollInterval);
}
//
// If we successfully read a new interval, save it.
//
if (RegObject.ReadDWORD(REGKEY_VALUE_MINUPDATESERVERSTATUSINTERVAL, &dwNewValue))
{
g_dwMinUpdateServerStatusInterval = dwNewValue;
DPFX(DPFPREP, 1, "Using minimum update-server-status interval %u ms.", g_dwMinUpdateServerStatusInterval);
}
//
// If we successfully read a new value, save it.
//
if (RegObject.ReadDWORD(REGKEY_VALUE_POLLINTERVALBACKOFF, &dwNewValue))
{
if (dwNewValue != 0)
{
g_dwPollIntervalBackoff = dwNewValue;
DPFX(DPFPREP, 1, "Using poll interval backoff between 0 and %u ms.",
g_dwPollIntervalBackoff);
}
else
{
DPFX(DPFPREP, 0, "Ignoring invalid poll interval backoff setting, using default between 0 and %u ms!",
g_dwPollIntervalBackoff);
}
}
//
// If we successfully read a new interval, save it.
//
if (RegObject.ReadDWORD(REGKEY_VALUE_MAXPOLLINTERVAL, &dwNewValue))
{
//
// Make sure the value is greater than the starting value.
//
if (dwNewValue >= g_dwNoActiveNotifyPollInterval)
{
g_dwMaxPollInterval = dwNewValue;
DPFX(DPFPREP, 1, "Using max poll interval of %u ms.",
g_dwMaxPollInterval);
}
else
{
g_dwMaxPollInterval = g_dwNoActiveNotifyPollInterval;
DPFX(DPFPREP, 0, "Ignoring max poll interval of %u ms, the starting value is %u ms.",
g_dwMaxPollInterval);
}
}
else
{
//
// Make sure the max poll interval default value is greater than
// the starting value because we may have read in a new
// g_dwNoActiveNotifyPollInterval that makes the default
// g_dwMaxPollInterval invalid.
//
if (g_dwMaxPollInterval < g_dwNoActiveNotifyPollInterval)
{
g_dwMaxPollInterval = g_dwNoActiveNotifyPollInterval;
DPFX(DPFPREP, 0, "Resetting max poll interval to %u ms so as to meet starting value.",
g_dwMaxPollInterval);
}
}
//
// If we successfully read a new boolean, save it.
//
if (RegObject.ReadBOOL(REGKEY_VALUE_KEEPPOLLINGFORREMOTEGATEWAY, &fNewValue))
{
g_fKeepPollingForRemoteGateway = fNewValue;
if (g_fKeepPollingForRemoteGateway)
{
DPFX(DPFPREP, 1, "Will continue to poll for remote gateways.");
}
else
{
//
// This is actually default behavior, but print out a statement
// anyway.
//
DPFX(DPFPREP, 1, "Continually polling for remote gateways is disallowed by registry key.");
}
}
//
// If we successfully read a new value, save it.
//
if (RegObject.ReadDWORD(REGKEY_VALUE_REUSEPORTTIME, &dwNewValue))
{
g_dwReusePortTime = dwNewValue;
DPFX(DPFPREP, 1, "Reusing remote gateway discovery ports for %u ms.",
g_dwReusePortTime);
}
//
// If we successfully read a new value, save it.
//
if (RegObject.ReadDWORD(REGKEY_VALUE_CACHELIFEFOUND, &dwNewValue))
{
g_dwCacheLifeFound = dwNewValue;
DPFX(DPFPREP, 1, "Caching found addresses for %u ms.",
g_dwCacheLifeFound);
}
//
// If we successfully read a new value, save it.
//
if (RegObject.ReadDWORD(REGKEY_VALUE_CACHELIFENOTFOUND, &dwNewValue))
{
g_dwCacheLifeNotFound = dwNewValue;
DPFX(DPFPREP, 1, "Caching not-found addresses for %u ms.",
g_dwCacheLifeNotFound);
}
//
// Okay, we're done. Drop the lock.
//
DNLeaveCriticalSection(&g_csGlobalsLock);
//
// Done reading registry.
//
RegObject.Close();
}
} // ReadRegistrySettings
#undef DPF_MODNAME
#define DPF_MODNAME "InitializeGlobalRand"
//=============================================================================
// InitializeGlobalRand
//-----------------------------------------------------------------------------
//
// Description: Initializes the fallback global psuedo-random number
// generator, using the given seed value.
//
// Arguments:
// DWORD dwSeed - Seed to use.
//
// Returns: None.
//=============================================================================
void InitializeGlobalRand(const DWORD dwSeed)
{
//
// We don't need to hold a lock, since this should only be done once,
// during initialization time.
//
g_dwHoldRand = dwSeed;
} // InitializeGlobalRand
#undef DPF_MODNAME
#define DPF_MODNAME "GetGlobalRand"
//=============================================================================
// GetGlobalRand
//-----------------------------------------------------------------------------
//
// Description: Generates a pseudo-random DWORD.
//
// Arguments: None.
//
// Returns: Pseudo-random number.
//=============================================================================
DWORD GetGlobalRand(void)
{
HCRYPTPROV hCryptProv;
DWORD dwResult;
WORD wResult1;
WORD wResult2;
#ifdef DBG
DWORD dwError;
#endif // DBG
if (CryptAcquireContext(&hCryptProv, NULL, NULL, PROV_RSA_FULL, CRYPT_VERIFYCONTEXT))
{
if (CryptGenRandom(hCryptProv, sizeof(dwResult), (BYTE*) (&dwResult)))
{
CryptReleaseContext(hCryptProv, 0);
return dwResult;
}
#ifdef DBG
else
{
dwError = GetLastError();
DPFX(DPFPREP, 0, "Crypto couldn't generate random number (err = %u)!",
dwError);
}
#endif // DBG
CryptReleaseContext(hCryptProv, 0);
}
#ifdef DBG
else
{
dwError = GetLastError();
DPFX(DPFPREP, 0, "Couldn't acquire crypto provider context (err = %u)!",
dwError);
}
#endif // DBG
//
// We couldn't use the crypto API to generate a random number, so make
// our own based off the C run time source.
//
DNEnterCriticalSection(&g_csGlobalsLock);
g_dwHoldRand = ((g_dwHoldRand * 214013L + 2531011L) >> 16) & 0x7fff;
wResult1 = (WORD) g_dwHoldRand;
g_dwHoldRand = ((g_dwHoldRand * 214013L + 2531011L) >> 16) & 0x7fff;
wResult2 = (WORD) g_dwHoldRand;
DNLeaveCriticalSection(&g_csGlobalsLock);
return MAKELONG(wResult1, wResult2);
} // GetGlobalRand
#undef DPF_MODNAME
#define DPF_MODNAME "DoCreateInstance"
//=============================================================================
// DoCreateInstance
//-----------------------------------------------------------------------------
//
// Description: Creates an instance of an interface. Required by the general
// purpose class factory functions.
//
// Arguments:
// LPCLASSFACTORY This - Pointer to class factory.
// LPUNKNOWN pUnkOuter - Pointer to unknown interface.
// REFCLSID rclsid - Reference of GUID of desired interface.
// REFIID riid - Reference to another GUID?
// LPVOID * ppvObj - Pointer to pointer to interface.
//
// Returns: HRESULT
//=============================================================================
HRESULT DoCreateInstance(LPCLASSFACTORY This,
LPUNKNOWN pUnkOuter,
REFCLSID rclsid,
REFIID riid,
LPVOID * ppvObj)
{
HRESULT hr;
BOOL fNotCreatedWithCOM;
CNATHelpPAST * pNATHelpPAST = NULL;
DNASSERT(ppvObj != NULL);
if (! IsEqualCLSID(rclsid, CLSID_DirectPlayNATHelpPAST))
{
//
// This shouldn't happen if they called IClassFactory::CreateObject
// correctly.
//
DNASSERT(FALSE);
//
// Return an error.
//
hr = E_UNEXPECTED;
goto Failure;
}
//
// If the class factory pointer is NULL, then we were called by the
// DirectPlayNATHelpCreate function.
//
if (This == NULL)
{
fNotCreatedWithCOM = TRUE;
}
else
{
fNotCreatedWithCOM = FALSE;
}
//
// Create the object instance.
//
pNATHelpPAST = new CNATHelpPAST(fNotCreatedWithCOM);
if (pNATHelpPAST == NULL)
{
hr = E_OUTOFMEMORY;
goto Failure;
}
//
// Initialize the base object (which might fail).
//
hr = pNATHelpPAST->InitializeObject();
if (hr != S_OK)
{
DPFX(DPFPREP, 0, "Couldn't initialize object!");
delete pNATHelpPAST;
pNATHelpPAST = NULL;
goto Failure;
}
//
// Add it to the global list.
//
DNEnterCriticalSection(&g_csGlobalsLock);
pNATHelpPAST->m_blList.InsertBefore(&g_blNATHelpPASTObjs);
g_lOutstandingInterfaceCount++; // update count so DllCanUnloadNow works correctly
DNLeaveCriticalSection(&g_csGlobalsLock);
//
// Get the right interface for the caller and bump the refcount.
//
hr = pNATHelpPAST->QueryInterface(riid, ppvObj);
if (hr != S_OK)
{
goto Failure;
}
Exit:
//
// Release the local reference to the object. If this function was
// successful, there's still a reference in ppvObj.
//
if (pNATHelpPAST != NULL)
{
pNATHelpPAST->Release();
pNATHelpPAST = NULL;
}
return hr;
Failure:
//
// Make sure we don't hand back a pointer.
//
(*ppvObj) = NULL;
goto Exit;
} // DoCreateInstance
#undef DPF_MODNAME
#define DPF_MODNAME "IsClassImplemented"
//=============================================================================
// IsClassImplemented
//-----------------------------------------------------------------------------
//
// Description: Determine if a class is implemented in this DLL. Required by
// the general purpose class factory functions.
//
// Arguments:
// REFCLSID rclsid - Reference to class GUID.
//
// Returns: BOOL
// TRUE - This DLL implements the class.
// FALSE - This DLL doesn't implement the class.
//=============================================================================
BOOL IsClassImplemented(REFCLSID rclsid)
{
return (IsEqualCLSID(rclsid, CLSID_DirectPlayNATHelpPAST));
} // IsClassImplemented