//***************************************************************************

//

//  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 () ;
}