403 lines
11 KiB
C++
403 lines
11 KiB
C++
//***************************************************************************
|
|
|
|
//
|
|
|
|
// MAINDLL.CPP
|
|
|
|
//
|
|
|
|
// Purpose: Contains DLL entry points. Also has code that controls
|
|
|
|
// when the DLL can be unloaded by tracking the number of
|
|
|
|
// objects and locks.
|
|
|
|
//
|
|
|
|
// Copyright (c) 1996-2001 Microsoft Corporation, All Rights Reserved
|
|
//
|
|
//***************************************************************************
|
|
|
|
#include <windows.h>
|
|
#include <snmptempl.h>
|
|
#include <snmpmt.h>
|
|
#include <process.h>
|
|
#include <objbase.h>
|
|
#include <olectl.h>
|
|
#include <wbemidl.h>
|
|
#include <instpath.h>
|
|
#include <snmpevt.h>
|
|
#include <snmpthrd.h>
|
|
#include <snmplog.h>
|
|
#include <snmpcont.h>
|
|
#include <snmpcl.h>
|
|
#include <snmptype.h>
|
|
#include <snmpobj.h>
|
|
#include "classfac.h"
|
|
#include "wndtime.h"
|
|
#include "rmon.h"
|
|
#include "trrtprov.h"
|
|
#include "guids.h"
|
|
|
|
//OK we need this one
|
|
HINSTANCE g_hInst=NULL;
|
|
|
|
//***************************************************************************
|
|
//
|
|
// LibMain32
|
|
//
|
|
// Purpose: Entry point for DLL. Good place for initialization.
|
|
// Return: TRUE if OK.
|
|
//***************************************************************************
|
|
|
|
BOOL APIENTRY DllMain (
|
|
|
|
HINSTANCE hInstance,
|
|
ULONG ulReason ,
|
|
LPVOID pvReserved
|
|
)
|
|
{
|
|
g_hInst=hInstance;
|
|
|
|
BOOL status = TRUE ;
|
|
|
|
if ( DLL_PROCESS_DETACH == ulReason )
|
|
{
|
|
status = TRUE ;
|
|
}
|
|
else if ( DLL_PROCESS_ATTACH == ulReason )
|
|
{
|
|
DisableThreadLibraryCalls(hInstance); // 158024
|
|
status = TRUE ;
|
|
}
|
|
else if ( DLL_THREAD_DETACH == ulReason )
|
|
{
|
|
status = TRUE ;
|
|
}
|
|
else if ( DLL_THREAD_ATTACH == ulReason )
|
|
{
|
|
status = TRUE ;
|
|
}
|
|
|
|
return TRUE ;
|
|
}
|
|
|
|
//***************************************************************************
|
|
//
|
|
// DllGetClassObject
|
|
//
|
|
// Purpose: Called by Ole when some client wants a a class factory. Return
|
|
// one only if it is the sort of class this DLL supports.
|
|
//
|
|
//***************************************************************************
|
|
|
|
STDAPI DllGetClassObject (
|
|
|
|
REFCLSID rclsid ,
|
|
REFIID riid,
|
|
void **ppv
|
|
)
|
|
{
|
|
HRESULT status = S_OK ;
|
|
|
|
if ( rclsid == CLSID_CTraceRouteLocatorClassFactory )
|
|
{
|
|
CTraceRouteLocatorClassFactory *lpunk = new CTraceRouteLocatorClassFactory ;
|
|
if ( lpunk == NULL )
|
|
{
|
|
status = E_OUTOFMEMORY ;
|
|
}
|
|
else
|
|
{
|
|
status = lpunk->QueryInterface ( riid , ppv ) ;
|
|
if ( FAILED ( status ) )
|
|
{
|
|
delete lpunk ;
|
|
}
|
|
else
|
|
{
|
|
}
|
|
}
|
|
}
|
|
else if ( rclsid == CLSID_CTraceRouteProvClassFactory )
|
|
{
|
|
CTraceRouteProvClassFactory *lpunk = new CTraceRouteProvClassFactory ;
|
|
if ( lpunk == NULL )
|
|
{
|
|
status = E_OUTOFMEMORY ;
|
|
}
|
|
else
|
|
{
|
|
status = lpunk->QueryInterface ( riid , ppv ) ;
|
|
if ( FAILED ( status ) )
|
|
{
|
|
delete lpunk ;
|
|
}
|
|
else
|
|
{
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
status = CLASS_E_CLASSNOTAVAILABLE ;
|
|
}
|
|
|
|
return status ;
|
|
}
|
|
|
|
//***************************************************************************
|
|
//
|
|
// DllCanUnloadNow
|
|
//
|
|
// Purpose: Called periodically by Ole in order to determine if the
|
|
// DLL can be freed.//
|
|
// Return: TRUE if there are no objects in use and the class factory
|
|
// isn't locked.
|
|
//***************************************************************************
|
|
|
|
STDAPI DllCanUnloadNow ()
|
|
{
|
|
|
|
/*
|
|
* Place code in critical section
|
|
*/
|
|
|
|
BOOL unload = ( CTraceRouteLocatorClassFactory :: s_LocksInProgress || CTraceRouteLocatorClassFactory :: s_ObjectsInProgress ) ;
|
|
unload = ! unload ;
|
|
|
|
return unload ? ResultFromScode ( S_OK ) : ResultFromScode ( S_FALSE ) ;
|
|
}
|
|
|
|
//Strings used during self registeration
|
|
|
|
#define REG_FORMAT_STR L"%s\\%s"
|
|
#define VER_IND_STR L"VersionIndependentProgID"
|
|
#define NOT_INTERT_STR L"NotInsertable"
|
|
#define INPROC32_STR L"InprocServer32"
|
|
#define INPROC_STR L"InprocServer"
|
|
#define PROGID_STR L"ProgID"
|
|
#define THREADING_MODULE_STR L"ThreadingModel"
|
|
#define APARTMENT_STR L"Both"
|
|
|
|
#define CLSID_STR L"CLSID\\"
|
|
#define CLSID2_STR L"CLSID"
|
|
#define CVER_STR L"CurVer"
|
|
|
|
#define LOCATOR_NAME_STR L"Microsoft WBEM Trace Route Locator"
|
|
#define LOCATOR_STR L"OLEMS.SNMP.PROPERTY.LOCATOR"
|
|
#define LOCATOR_VER_STR L"OLEMS.SNMP.PROPERTY.LOCATOR.0"
|
|
|
|
#define PROVIDER_NAME_STR L"Microsoft WBEM Trace Route Provider"
|
|
#define PROVIDER_STR L"OLEMS.SNMP.PROPERTY.PROVIDER"
|
|
#define PROVIDER_VER_STR L"OLEMS.SNMP.PROPERTY.PROVIDER.0"
|
|
|
|
/***************************************************************************
|
|
* SetKeyAndValue
|
|
*
|
|
* Purpose:
|
|
* Private helper function for DllRegisterServer that creates
|
|
* a key, sets a value, and closes that key.
|
|
*
|
|
* Parameters:
|
|
* pszKey LPTSTR to the ame of the key
|
|
* pszSubkey LPTSTR ro the name of a subkey
|
|
* pszValue LPTSTR to the value to store
|
|
*
|
|
* Return Value:
|
|
* BOOL TRUE if successful, FALSE otherwise.
|
|
***************************************************************************/
|
|
|
|
BOOL SetKeyAndValue(wchar_t* pszKey, wchar_t* pszSubkey, wchar_t* pszValueName, wchar_t* pszValue)
|
|
{
|
|
HKEY hKey;
|
|
wchar_t szKey[256];
|
|
|
|
wcscpy(szKey, pszKey);
|
|
|
|
if (NULL!=pszSubkey)
|
|
{
|
|
wcscat(szKey, L"\\");
|
|
wcscat(szKey, pszSubkey);
|
|
}
|
|
|
|
if (ERROR_SUCCESS!=RegCreateKeyEx(HKEY_CLASSES_ROOT
|
|
, szKey, 0, NULL, REG_OPTION_NON_VOLATILE
|
|
, KEY_ALL_ACCESS, NULL, &hKey, NULL))
|
|
return FALSE;
|
|
|
|
if (NULL!=pszValue)
|
|
{
|
|
if (ERROR_SUCCESS != RegSetValueEx(hKey, pszValueName, 0, REG_SZ, (BYTE *)pszValue
|
|
, (lstrlen(pszValue)+1)*sizeof(wchar_t)))
|
|
return FALSE;
|
|
}
|
|
RegCloseKey(hKey);
|
|
return TRUE;
|
|
}
|
|
|
|
/***************************************************************************
|
|
* DllRegisterServer
|
|
*
|
|
* Purpose:
|
|
* Instructs the server to create its own registry entries
|
|
*
|
|
* Parameters:
|
|
* None
|
|
*
|
|
* Return Value:
|
|
* HRESULT NOERROR if registration successful, error
|
|
* otherwise.
|
|
***************************************************************************/
|
|
|
|
STDAPI InstanceDllRegisterServer()
|
|
{
|
|
wchar_t szModule[512];
|
|
GetModuleFileName(g_hInst,(wchar_t*)szModule, sizeof(szModule)/sizeof(wchar_t));
|
|
|
|
wchar_t szLocatorClassID[128];
|
|
wchar_t szLocatorCLSIDClassID[128];
|
|
|
|
int iRet = StringFromGUID2(CLSID_CTraceRouteLocatorClassFactory,szLocatorClassID, 128);
|
|
|
|
wcscpy(szLocatorCLSIDClassID,CLSID_STR);
|
|
wcscat(szLocatorCLSIDClassID,szLocatorClassID);
|
|
|
|
//Create entries under CLSID
|
|
if (FALSE ==SetKeyAndValue(szLocatorCLSIDClassID, NULL, NULL, LOCATOR_NAME_STR))
|
|
return SELFREG_E_CLASS;
|
|
if (FALSE ==SetKeyAndValue(szLocatorCLSIDClassID, PROGID_STR, NULL, LOCATOR_VER_STR))
|
|
return SELFREG_E_CLASS;
|
|
if (FALSE ==SetKeyAndValue(szLocatorCLSIDClassID, VER_IND_STR, NULL, LOCATOR_STR))
|
|
return SELFREG_E_CLASS;
|
|
if (FALSE ==SetKeyAndValue(szLocatorCLSIDClassID, NOT_INTERT_STR, NULL, NULL))
|
|
return SELFREG_E_CLASS;
|
|
|
|
if (FALSE ==SetKeyAndValue(szLocatorCLSIDClassID, INPROC32_STR, NULL,szModule))
|
|
return SELFREG_E_CLASS;
|
|
if (FALSE ==SetKeyAndValue(szLocatorCLSIDClassID, INPROC32_STR,THREADING_MODULE_STR, APARTMENT_STR))
|
|
return SELFREG_E_CLASS;
|
|
|
|
wchar_t szProviderClassID[128];
|
|
wchar_t szProviderCLSIDClassID[128];
|
|
|
|
iRet = StringFromGUID2(CLSID_CTraceRouteProvClassFactory,szProviderClassID, 128);
|
|
|
|
wcscpy(szProviderCLSIDClassID,CLSID_STR);
|
|
wcscat(szProviderCLSIDClassID,szProviderClassID);
|
|
|
|
//Create entries under CLSID
|
|
if (FALSE ==SetKeyAndValue(szProviderCLSIDClassID, NULL, NULL, PROVIDER_NAME_STR))
|
|
return SELFREG_E_CLASS;
|
|
if (FALSE ==SetKeyAndValue(szProviderCLSIDClassID, PROGID_STR, NULL, PROVIDER_VER_STR))
|
|
return SELFREG_E_CLASS;
|
|
if (FALSE ==SetKeyAndValue(szProviderCLSIDClassID, VER_IND_STR, NULL, PROVIDER_STR))
|
|
return SELFREG_E_CLASS;
|
|
if (FALSE ==SetKeyAndValue(szProviderCLSIDClassID, NOT_INTERT_STR, NULL, NULL))
|
|
return SELFREG_E_CLASS;
|
|
|
|
if (FALSE ==SetKeyAndValue(szProviderCLSIDClassID, INPROC32_STR, NULL,szModule))
|
|
return SELFREG_E_CLASS;
|
|
if (FALSE ==SetKeyAndValue(szProviderCLSIDClassID, INPROC32_STR,THREADING_MODULE_STR, APARTMENT_STR))
|
|
return SELFREG_E_CLASS;
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
|
|
STDAPI DllRegisterServer()
|
|
{
|
|
return InstanceDllRegisterServer () ;
|
|
}
|
|
|
|
/***************************************************************************
|
|
* DllUnregisterServer
|
|
*
|
|
* Purpose:
|
|
* Instructs the server to remove its own registry entries
|
|
*
|
|
* Parameters:
|
|
* None
|
|
*
|
|
* Return Value:
|
|
* HRESULT NOERROR if registration successful, error
|
|
* otherwise.
|
|
***************************************************************************/
|
|
|
|
STDAPI InstanceDllUnregisterServer(void)
|
|
{
|
|
wchar_t szTemp[128];
|
|
|
|
wchar_t szLocatorClassID[128];
|
|
wchar_t szLocatorCLSIDClassID[128];
|
|
|
|
int iRet = StringFromGUID2(CLSID_CTraceRouteLocatorClassFactory,szLocatorClassID, 128);
|
|
|
|
wcscpy(szLocatorCLSIDClassID,CLSID_STR);
|
|
wcscat(szLocatorCLSIDClassID,szLocatorClassID);
|
|
|
|
//Delete ProgID keys
|
|
|
|
RegDeleteKey(HKEY_CLASSES_ROOT, LOCATOR_STR);
|
|
|
|
//Delete VersionIndependentProgID keys
|
|
|
|
RegDeleteKey(HKEY_CLASSES_ROOT, LOCATOR_VER_STR);
|
|
|
|
//Delete entries under CLSID
|
|
|
|
wsprintf(szTemp, REG_FORMAT_STR, szLocatorCLSIDClassID, PROGID_STR);
|
|
RegDeleteKey(HKEY_CLASSES_ROOT, szTemp);
|
|
|
|
wsprintf(szTemp, REG_FORMAT_STR, szLocatorCLSIDClassID, VER_IND_STR);
|
|
RegDeleteKey(HKEY_CLASSES_ROOT, szTemp);
|
|
|
|
wsprintf(szTemp, REG_FORMAT_STR, szLocatorCLSIDClassID, NOT_INTERT_STR);
|
|
RegDeleteKey(HKEY_CLASSES_ROOT, szTemp);
|
|
|
|
wsprintf(szTemp, REG_FORMAT_STR,szLocatorCLSIDClassID, INPROC32_STR);
|
|
RegDeleteKey(HKEY_CLASSES_ROOT, szTemp);
|
|
|
|
RegDeleteKey(HKEY_CLASSES_ROOT, szLocatorCLSIDClassID);
|
|
|
|
wchar_t szProviderClassID[128];
|
|
wchar_t szProviderCLSIDClassID[128];
|
|
|
|
iRet = StringFromGUID2(CLSID_CTraceRouteProvClassFactory,szProviderClassID, 128);
|
|
|
|
wcscpy(szProviderCLSIDClassID,CLSID_STR);
|
|
wcscat(szProviderCLSIDClassID,szLocatorClassID);
|
|
|
|
//Delete ProgID keys
|
|
|
|
RegDeleteKey(HKEY_CLASSES_ROOT, PROVIDER_STR);
|
|
|
|
//Delete VersionIndependentProgID keys
|
|
|
|
RegDeleteKey(HKEY_CLASSES_ROOT, PROVIDER_VER_STR);
|
|
|
|
//Delete entries under CLSID
|
|
|
|
wsprintf(szTemp, REG_FORMAT_STR, szProviderCLSIDClassID, PROGID_STR);
|
|
RegDeleteKey(HKEY_CLASSES_ROOT, szTemp);
|
|
|
|
wsprintf(szTemp, REG_FORMAT_STR, szProviderCLSIDClassID, VER_IND_STR);
|
|
RegDeleteKey(HKEY_CLASSES_ROOT, szTemp);
|
|
|
|
wsprintf(szTemp, REG_FORMAT_STR, szProviderCLSIDClassID, NOT_INTERT_STR);
|
|
RegDeleteKey(HKEY_CLASSES_ROOT, szTemp);
|
|
|
|
wsprintf(szTemp, REG_FORMAT_STR,szProviderCLSIDClassID, INPROC32_STR);
|
|
RegDeleteKey(HKEY_CLASSES_ROOT, szTemp);
|
|
|
|
RegDeleteKey(HKEY_CLASSES_ROOT, szProviderCLSIDClassID);
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
|
|
STDAPI DllUnregisterServer(void)
|
|
{
|
|
return InstanceDllUnregisterServer () ;
|
|
}
|