You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
1103 lines
22 KiB
1103 lines
22 KiB
//***************************************************************************
|
|
|
|
//
|
|
|
|
// File:
|
|
|
|
//
|
|
|
|
// Module: MS SNMP Provider
|
|
|
|
//
|
|
|
|
// Purpose:
|
|
|
|
//
|
|
|
|
// Copyright (c) 1997-2001 Microsoft Corporation, All Rights Reserved
|
|
//
|
|
//***************************************************************************
|
|
|
|
#include "precomp.h"
|
|
#include <provexpt.h>
|
|
#include <snmpstd.h>
|
|
#include <snmpmt.h>
|
|
#include <snmptempl.h>
|
|
#include <objbase.h>
|
|
|
|
// this redefines the DEFINE_GUID() macro to do allocation.
|
|
//
|
|
#include <initguid.h>
|
|
#ifndef INITGUID
|
|
#define INITGUID
|
|
#endif
|
|
|
|
#include <wbemidl.h>
|
|
#include <snmpcont.h>
|
|
#include <instpath.h>
|
|
#include <snmpevt.h>
|
|
#include <snmpthrd.h>
|
|
#include <snmplog.h>
|
|
#include <snmpcl.h>
|
|
#include <snmptype.h>
|
|
#include <snmpobj.h>
|
|
#include <smir.h>
|
|
#include <classfac.h>
|
|
#include "clasprov.h"
|
|
#include "propprov.h"
|
|
#include "guids.h"
|
|
|
|
#include <notify.h>
|
|
|
|
#include <evtdefs.h>
|
|
#include <evtthrd.h>
|
|
#include <evtmap.h>
|
|
#include <evtprov.h>
|
|
|
|
|
|
LONG CClasProvClassFactory :: objectsInProgress = 0 ;
|
|
LONG CClasProvClassFactory :: locksInProgress = 0 ;
|
|
|
|
LONG CPropProvClassFactory :: objectsInProgress = 0 ;
|
|
LONG CPropProvClassFactory :: locksInProgress = 0 ;
|
|
|
|
LONG CSNMPEventProviderClassFactory :: objectsInProgress = 0 ;
|
|
LONG CSNMPEventProviderClassFactory :: locksInProgress = 0 ;
|
|
|
|
extern CEventProviderThread* g_pProvThrd;
|
|
extern CEventProviderWorkerThread* g_pWorkerThread;
|
|
extern CRITICAL_SECTION s_ProviderCriticalSection ;
|
|
|
|
extern void ProviderStartup () ;
|
|
extern void ProviderClosedown () ;
|
|
|
|
//***************************************************************************
|
|
//
|
|
// CClasProvClassFactory::CClasProvClassFactory
|
|
// CClasProvClassFactory::~CClasProvClassFactory
|
|
//
|
|
// Constructor Parameters:
|
|
// None
|
|
//***************************************************************************
|
|
|
|
CClasProvClassFactory :: CClasProvClassFactory ()
|
|
{
|
|
InterlockedIncrement ( & objectsInProgress ) ;
|
|
m_referenceCount = 0 ;
|
|
}
|
|
|
|
CClasProvClassFactory::~CClasProvClassFactory ()
|
|
{
|
|
InterlockedDecrement ( & objectsInProgress ) ;
|
|
}
|
|
|
|
//***************************************************************************
|
|
//
|
|
// CClasProvClassFactory::QueryInterface
|
|
// CClasProvClassFactory::AddRef
|
|
// CClasProvClassFactory::Release
|
|
//
|
|
// Purpose: Standard Ole routines needed for all interfaces
|
|
//
|
|
//***************************************************************************
|
|
|
|
STDMETHODIMP CClasProvClassFactory::QueryInterface (
|
|
|
|
REFIID iid ,
|
|
LPVOID FAR *iplpv
|
|
)
|
|
{
|
|
SetStructuredExceptionHandler seh;
|
|
|
|
try
|
|
{
|
|
*iplpv = NULL ;
|
|
|
|
if ( iid == IID_IUnknown )
|
|
{
|
|
*iplpv = ( LPVOID ) this ;
|
|
}
|
|
else if ( iid == IID_IClassFactory )
|
|
{
|
|
*iplpv = ( LPVOID ) this ;
|
|
}
|
|
|
|
if ( *iplpv )
|
|
{
|
|
( ( LPUNKNOWN ) *iplpv )->AddRef () ;
|
|
|
|
return ResultFromScode ( S_OK ) ;
|
|
}
|
|
else
|
|
{
|
|
return ResultFromScode ( E_NOINTERFACE ) ;
|
|
}
|
|
}
|
|
catch(Structured_Exception e_SE)
|
|
{
|
|
return E_UNEXPECTED;
|
|
}
|
|
catch(Heap_Exception e_HE)
|
|
{
|
|
return E_OUTOFMEMORY;
|
|
}
|
|
catch(...)
|
|
{
|
|
return E_UNEXPECTED;
|
|
}
|
|
}
|
|
|
|
|
|
STDMETHODIMP_( ULONG ) CClasProvClassFactory :: AddRef ()
|
|
{
|
|
SetStructuredExceptionHandler seh;
|
|
|
|
try
|
|
{
|
|
return InterlockedIncrement ( & m_referenceCount ) ;
|
|
}
|
|
catch(Structured_Exception e_SE)
|
|
{
|
|
return 0;
|
|
}
|
|
catch(Heap_Exception e_HE)
|
|
{
|
|
return 0;
|
|
}
|
|
catch(...)
|
|
{
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
STDMETHODIMP_(ULONG) CClasProvClassFactory :: Release ()
|
|
{
|
|
SetStructuredExceptionHandler seh;
|
|
|
|
try
|
|
{
|
|
LONG ref ;
|
|
if ( ( ref = InterlockedDecrement ( & m_referenceCount ) ) == 0 )
|
|
{
|
|
delete this ;
|
|
return 0 ;
|
|
}
|
|
else
|
|
{
|
|
return ref ;
|
|
}
|
|
}
|
|
catch(Structured_Exception e_SE)
|
|
{
|
|
return 0;
|
|
}
|
|
catch(Heap_Exception e_HE)
|
|
{
|
|
return 0;
|
|
}
|
|
catch(...)
|
|
{
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
//***************************************************************************
|
|
//
|
|
// CClasProvClassFactory::CreateInstance
|
|
//
|
|
// Purpose: Instantiates a Provider object returning an interface pointer.
|
|
//
|
|
// Parameters:
|
|
// pUnkOuter LPUNKNOWN to the controlling IUnknown if we are
|
|
// being used in an aggregation.
|
|
// riid REFIID identifying the interface the caller
|
|
// desires to have for the new object.
|
|
// ppvObj PPVOID in which to store the desired
|
|
// interface pointer for the new object.
|
|
//
|
|
// Return Value:
|
|
// HRESULT NOERROR if successful, otherwise E_NOINTERFACE
|
|
// if we cannot support the requested interface.
|
|
//***************************************************************************
|
|
|
|
STDMETHODIMP CClasProvClassFactory :: CreateInstance (
|
|
|
|
LPUNKNOWN pUnkOuter ,
|
|
REFIID riid ,
|
|
LPVOID FAR * ppvObject
|
|
)
|
|
{
|
|
SetStructuredExceptionHandler seh;
|
|
|
|
try
|
|
{
|
|
HRESULT status = S_OK ;
|
|
|
|
if ( pUnkOuter )
|
|
{
|
|
status = CLASS_E_NOAGGREGATION ;
|
|
}
|
|
else
|
|
{
|
|
IWbemServices *lpunk = ( IWbemServices * ) new CImpClasProv ;
|
|
if ( lpunk == NULL )
|
|
{
|
|
status = E_OUTOFMEMORY ;
|
|
}
|
|
else
|
|
{
|
|
status = lpunk->QueryInterface ( riid , ppvObject ) ;
|
|
if ( FAILED ( status ) )
|
|
{
|
|
delete lpunk ;
|
|
}
|
|
else
|
|
{
|
|
}
|
|
}
|
|
}
|
|
|
|
return status ;
|
|
}
|
|
catch(Structured_Exception e_SE)
|
|
{
|
|
return E_UNEXPECTED;
|
|
}
|
|
catch(Heap_Exception e_HE)
|
|
{
|
|
return E_OUTOFMEMORY;
|
|
}
|
|
}
|
|
|
|
//***************************************************************************
|
|
//
|
|
// CClasProvClassFactory::LockServer
|
|
//
|
|
// Purpose:
|
|
// Increments or decrements the lock count of the DLL. If the
|
|
// lock count goes to zero and there are no objects, the DLL
|
|
// is allowed to unload. See DllCanUnloadNow.
|
|
//
|
|
// Parameters:
|
|
// fLock BOOL specifying whether to increment or
|
|
// decrement the lock count.
|
|
//
|
|
// Return Value:
|
|
// HRESULT NOERROR always.
|
|
//***************************************************************************
|
|
|
|
STDMETHODIMP CClasProvClassFactory :: LockServer ( BOOL fLock )
|
|
{
|
|
SetStructuredExceptionHandler seh;
|
|
|
|
try
|
|
{
|
|
/*
|
|
* Place code in critical section
|
|
*/
|
|
|
|
if ( fLock )
|
|
{
|
|
InterlockedIncrement ( & locksInProgress ) ;
|
|
}
|
|
else
|
|
{
|
|
InterlockedDecrement ( & locksInProgress ) ;
|
|
}
|
|
|
|
return S_OK ;
|
|
}
|
|
catch(Structured_Exception e_SE)
|
|
{
|
|
return E_UNEXPECTED;
|
|
}
|
|
catch(Heap_Exception e_HE)
|
|
{
|
|
return E_OUTOFMEMORY;
|
|
}
|
|
catch(...)
|
|
{
|
|
return E_UNEXPECTED;
|
|
}
|
|
}
|
|
|
|
//***************************************************************************
|
|
//
|
|
// CPropProvClassFactory::CPropProvClassFactory
|
|
// CPropProvClassFactory::~CPropProvClassFactory
|
|
//
|
|
// Constructor Parameters:
|
|
// None
|
|
//***************************************************************************
|
|
|
|
CPropProvClassFactory :: CPropProvClassFactory ()
|
|
{
|
|
InterlockedIncrement ( & objectsInProgress ) ;
|
|
m_referenceCount = 0 ;
|
|
}
|
|
|
|
CPropProvClassFactory::~CPropProvClassFactory ()
|
|
{
|
|
InterlockedDecrement ( & objectsInProgress ) ;
|
|
}
|
|
|
|
//***************************************************************************
|
|
//
|
|
// CPropProvClassFactory::QueryInterface
|
|
// CPropProvClassFactory::AddRef
|
|
// CPropProvClassFactory::Release
|
|
//
|
|
// Purpose: Standard Ole routines needed for all interfaces
|
|
//
|
|
//***************************************************************************
|
|
|
|
STDMETHODIMP CPropProvClassFactory::QueryInterface (
|
|
|
|
REFIID iid ,
|
|
LPVOID FAR *iplpv
|
|
)
|
|
{
|
|
SetStructuredExceptionHandler seh;
|
|
|
|
try
|
|
{
|
|
*iplpv = NULL ;
|
|
|
|
if ( iid == IID_IUnknown )
|
|
{
|
|
*iplpv = ( LPVOID ) this ;
|
|
}
|
|
else if ( iid == IID_IClassFactory )
|
|
{
|
|
*iplpv = ( LPVOID ) this ;
|
|
}
|
|
|
|
if ( *iplpv )
|
|
{
|
|
( ( LPUNKNOWN ) *iplpv )->AddRef () ;
|
|
|
|
return ResultFromScode ( S_OK ) ;
|
|
}
|
|
else
|
|
{
|
|
return ResultFromScode ( E_NOINTERFACE ) ;
|
|
}
|
|
}
|
|
catch(Structured_Exception e_SE)
|
|
{
|
|
return E_UNEXPECTED;
|
|
}
|
|
catch(Heap_Exception e_HE)
|
|
{
|
|
return E_OUTOFMEMORY;
|
|
}
|
|
catch(...)
|
|
{
|
|
return E_UNEXPECTED;
|
|
}
|
|
}
|
|
|
|
|
|
STDMETHODIMP_( ULONG ) CPropProvClassFactory :: AddRef ()
|
|
{
|
|
SetStructuredExceptionHandler seh;
|
|
|
|
try
|
|
{
|
|
return InterlockedIncrement ( & m_referenceCount ) ;
|
|
}
|
|
catch(Structured_Exception e_SE)
|
|
{
|
|
return 0;
|
|
}
|
|
catch(Heap_Exception e_HE)
|
|
{
|
|
return 0;
|
|
}
|
|
catch(...)
|
|
{
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
STDMETHODIMP_(ULONG) CPropProvClassFactory :: Release ()
|
|
{
|
|
SetStructuredExceptionHandler seh;
|
|
|
|
try
|
|
{
|
|
LONG ref ;
|
|
if ( ( ref = InterlockedDecrement ( & m_referenceCount ) ) == 0 )
|
|
{
|
|
delete this ;
|
|
return 0 ;
|
|
}
|
|
else
|
|
{
|
|
return ref ;
|
|
}
|
|
}
|
|
catch(Structured_Exception e_SE)
|
|
{
|
|
return 0;
|
|
}
|
|
catch(Heap_Exception e_HE)
|
|
{
|
|
return 0;
|
|
}
|
|
catch(...)
|
|
{
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
//***************************************************************************
|
|
//
|
|
// CPropProvClassFactory::CreateInstance
|
|
//
|
|
// Purpose: Instantiates a Provider object returning an interface pointer.
|
|
//
|
|
// Parameters:
|
|
// pUnkOuter LPUNKNOWN to the controlling IUnknown if we are
|
|
// being used in an aggregation.
|
|
// riid REFIID identifying the interface the caller
|
|
// desires to have for the new object.
|
|
// ppvObj PPVOID in which to store the desired
|
|
// interface pointer for the new object.
|
|
//
|
|
// Return Value:
|
|
// HRESULT NOERROR if successful, otherwise E_NOINTERFACE
|
|
// if we cannot support the requested interface.
|
|
//***************************************************************************
|
|
|
|
STDMETHODIMP CPropProvClassFactory :: CreateInstance (
|
|
|
|
LPUNKNOWN pUnkOuter ,
|
|
REFIID riid ,
|
|
LPVOID FAR * ppvObject
|
|
)
|
|
{
|
|
SetStructuredExceptionHandler seh;
|
|
|
|
try
|
|
{
|
|
HRESULT status = S_OK ;
|
|
|
|
if ( pUnkOuter )
|
|
{
|
|
status = CLASS_E_NOAGGREGATION ;
|
|
}
|
|
else
|
|
{
|
|
IWbemServices *lpunk = ( IWbemServices * ) new CImpPropProv ;
|
|
if ( lpunk == NULL )
|
|
{
|
|
status = E_OUTOFMEMORY ;
|
|
}
|
|
else
|
|
{
|
|
status = lpunk->QueryInterface ( riid , ppvObject ) ;
|
|
if ( FAILED ( status ) )
|
|
{
|
|
delete lpunk ;
|
|
}
|
|
else
|
|
{
|
|
}
|
|
}
|
|
}
|
|
|
|
return status ;
|
|
}
|
|
catch(Structured_Exception e_SE)
|
|
{
|
|
return E_UNEXPECTED;
|
|
}
|
|
catch(Heap_Exception e_HE)
|
|
{
|
|
return E_OUTOFMEMORY;
|
|
}
|
|
catch(...)
|
|
{
|
|
return E_UNEXPECTED;
|
|
}
|
|
}
|
|
|
|
//***************************************************************************
|
|
//
|
|
// CPropProvClassFactory::LockServer
|
|
//
|
|
// Purpose:
|
|
// Increments or decrements the lock count of the DLL. If the
|
|
// lock count goes to zero and there are no objects, the DLL
|
|
// is allowed to unload. See DllCanUnloadNow.
|
|
//
|
|
// Parameters:
|
|
// fLock BOOL specifying whether to increment or
|
|
// decrement the lock count.
|
|
//
|
|
// Return Value:
|
|
// HRESULT NOERROR always.
|
|
//***************************************************************************
|
|
|
|
STDMETHODIMP CPropProvClassFactory :: LockServer ( BOOL fLock )
|
|
{
|
|
SetStructuredExceptionHandler seh;
|
|
|
|
try
|
|
{
|
|
/*
|
|
* Place code in critical section
|
|
*/
|
|
|
|
if ( fLock )
|
|
{
|
|
InterlockedIncrement ( & locksInProgress ) ;
|
|
}
|
|
else
|
|
{
|
|
InterlockedDecrement ( & locksInProgress ) ;
|
|
}
|
|
|
|
return S_OK ;
|
|
}
|
|
catch(Structured_Exception e_SE)
|
|
{
|
|
return E_UNEXPECTED;
|
|
}
|
|
catch(Heap_Exception e_HE)
|
|
{
|
|
return E_OUTOFMEMORY;
|
|
}
|
|
catch(...)
|
|
{
|
|
return E_UNEXPECTED;
|
|
}
|
|
}
|
|
|
|
//***************************************************************************
|
|
//
|
|
// CSNMPEventProviderClassFactory::CSNMPEventProviderClassFactory
|
|
// CSNMPEventProviderClassFactory::~CSNMPEventProviderClassFactory
|
|
//
|
|
// Constructor Parameters:
|
|
// None
|
|
//***************************************************************************
|
|
|
|
CSNMPEventProviderClassFactory :: CSNMPEventProviderClassFactory ()
|
|
{
|
|
m_referenceCount = 0 ;
|
|
}
|
|
|
|
CSNMPEventProviderClassFactory::~CSNMPEventProviderClassFactory ()
|
|
{
|
|
}
|
|
|
|
//***************************************************************************
|
|
//
|
|
// CSNMPEventProviderClassFactory::QueryInterface
|
|
// CSNMPEventProviderClassFactory::AddRef
|
|
// CSNMPEventProviderClassFactory::Release
|
|
//
|
|
// Purpose: Standard Ole routines needed for all interfaces
|
|
//
|
|
//***************************************************************************
|
|
|
|
STDMETHODIMP CSNMPEventProviderClassFactory::QueryInterface (
|
|
|
|
REFIID iid ,
|
|
LPVOID FAR *iplpv
|
|
)
|
|
{
|
|
SetStructuredExceptionHandler seh;
|
|
|
|
try
|
|
{
|
|
*iplpv = NULL ;
|
|
|
|
if ( iid == IID_IUnknown )
|
|
{
|
|
*iplpv = ( LPVOID ) this ;
|
|
}
|
|
else if ( iid == IID_IClassFactory )
|
|
{
|
|
*iplpv = ( LPVOID ) this ;
|
|
}
|
|
|
|
if ( *iplpv )
|
|
{
|
|
( ( LPUNKNOWN ) *iplpv )->AddRef () ;
|
|
|
|
return ResultFromScode ( S_OK ) ;
|
|
}
|
|
else
|
|
{
|
|
return ResultFromScode ( E_NOINTERFACE ) ;
|
|
}
|
|
}
|
|
catch(Structured_Exception e_SE)
|
|
{
|
|
return E_UNEXPECTED;
|
|
}
|
|
catch(Heap_Exception e_HE)
|
|
{
|
|
return E_OUTOFMEMORY;
|
|
}
|
|
catch(...)
|
|
{
|
|
return E_UNEXPECTED;
|
|
}
|
|
}
|
|
|
|
|
|
STDMETHODIMP_( ULONG ) CSNMPEventProviderClassFactory :: AddRef ()
|
|
{
|
|
SetStructuredExceptionHandler seh;
|
|
|
|
try
|
|
{
|
|
InterlockedIncrement(&objectsInProgress);
|
|
return InterlockedIncrement ( &m_referenceCount ) ;
|
|
}
|
|
catch(Structured_Exception e_SE)
|
|
{
|
|
return 0;
|
|
}
|
|
catch(Heap_Exception e_HE)
|
|
{
|
|
return 0;
|
|
}
|
|
catch(...)
|
|
{
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
STDMETHODIMP_(ULONG) CSNMPEventProviderClassFactory :: Release ()
|
|
{
|
|
SetStructuredExceptionHandler seh;
|
|
|
|
try
|
|
{
|
|
LONG ref ;
|
|
|
|
if ( ( ref = InterlockedDecrement ( & m_referenceCount ) ) == 0 )
|
|
{
|
|
delete this ;
|
|
InterlockedDecrement(&objectsInProgress);
|
|
return 0 ;
|
|
}
|
|
else
|
|
{
|
|
InterlockedDecrement(&objectsInProgress);
|
|
return ref ;
|
|
}
|
|
}
|
|
catch(Structured_Exception e_SE)
|
|
{
|
|
return 0;
|
|
}
|
|
catch(Heap_Exception e_HE)
|
|
{
|
|
return 0;
|
|
}
|
|
catch(...)
|
|
{
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
//***************************************************************************
|
|
//
|
|
// CSNMPEventProviderClassFactory::LockServer
|
|
//
|
|
// Purpose:
|
|
// Increments or decrements the lock count of the DLL. If the
|
|
// lock count goes to zero and there are no objects, the DLL
|
|
// is allowed to unload. See DllCanUnloadNow.
|
|
//
|
|
// Parameters:
|
|
// fLock BOOL specifying whether to increment or
|
|
// decrement the lock count.
|
|
//
|
|
// Return Value:
|
|
// HRESULT NOERROR always.
|
|
//***************************************************************************
|
|
|
|
STDMETHODIMP CSNMPEventProviderClassFactory :: LockServer ( BOOL fLock )
|
|
{
|
|
SetStructuredExceptionHandler seh;
|
|
|
|
try
|
|
{
|
|
if ( fLock )
|
|
{
|
|
InterlockedIncrement ( & locksInProgress ) ;
|
|
}
|
|
else
|
|
{
|
|
InterlockedDecrement ( & locksInProgress ) ;
|
|
}
|
|
|
|
return S_OK ;
|
|
}
|
|
catch(Structured_Exception e_SE)
|
|
{
|
|
return E_UNEXPECTED;
|
|
}
|
|
catch(Heap_Exception e_HE)
|
|
{
|
|
return E_OUTOFMEMORY;
|
|
}
|
|
catch(...)
|
|
{
|
|
return E_UNEXPECTED;
|
|
}
|
|
}
|
|
|
|
//***************************************************************************
|
|
//
|
|
// CSNMPEncapEventProviderClassFactory::CreateInstance
|
|
//
|
|
// Purpose: Instantiates a Provider object returning an interface pointer.
|
|
//
|
|
// Parameters:
|
|
// pUnkOuter LPUNKNOWN to the controlling IUnknown if we are
|
|
// being used in an aggregation.
|
|
// riid REFIID identifying the interface the caller
|
|
// desires to have for the new object.
|
|
// ppvObj PPVOID in which to store the desired
|
|
// interface pointer for the new object.
|
|
//
|
|
// Return Value:
|
|
// HRESULT NOERROR if successful, otherwise E_NOINTERFACE
|
|
// if we cannot support the requested interface.
|
|
//***************************************************************************
|
|
|
|
STDMETHODIMP CSNMPEncapEventProviderClassFactory :: CreateInstance(LPUNKNOWN pUnkOuter ,
|
|
REFIID riid,
|
|
LPVOID FAR * ppvObject
|
|
)
|
|
{
|
|
SetStructuredExceptionHandler seh;
|
|
|
|
BOOL bEnteredCritSec = FALSE;
|
|
BOOL bContinue = FALSE;
|
|
|
|
try
|
|
{
|
|
HRESULT status = E_FAIL;
|
|
|
|
if ( pUnkOuter )
|
|
{
|
|
status = CLASS_E_NOAGGREGATION;
|
|
}
|
|
else
|
|
{
|
|
EnterCriticalSection ( & s_ProviderCriticalSection ) ;
|
|
bEnteredCritSec = TRUE;
|
|
|
|
if (NULL == g_pProvThrd)
|
|
{
|
|
ProviderStartup () ;
|
|
|
|
SnmpThreadObject :: Startup () ;
|
|
try
|
|
{
|
|
SnmpDebugLog :: Startup () ;
|
|
}
|
|
catch ( ... )
|
|
{
|
|
// we do not want to left s_Reference count up
|
|
SnmpThreadObject :: Closedown () ;
|
|
ProviderClosedown () ;
|
|
|
|
throw;
|
|
}
|
|
|
|
BOOL bStatus = FALSE;
|
|
try
|
|
{
|
|
bStatus = SnmpClassLibrary :: Startup () ;
|
|
}
|
|
catch ( ... )
|
|
{
|
|
// we do not want to left s_Reference count up
|
|
SnmpDebugLog :: Closedown () ;
|
|
SnmpThreadObject :: Closedown () ;
|
|
ProviderClosedown () ;
|
|
|
|
throw;
|
|
}
|
|
|
|
if ( bStatus == FALSE )
|
|
{
|
|
SnmpDebugLog :: Closedown () ;
|
|
SnmpThreadObject :: Closedown () ;
|
|
|
|
ProviderClosedown () ;
|
|
}
|
|
else
|
|
{
|
|
try
|
|
{
|
|
g_pWorkerThread = new CEventProviderWorkerThread;
|
|
g_pWorkerThread->BeginThread();
|
|
g_pWorkerThread->WaitForStartup();
|
|
g_pWorkerThread->CreateServerWrap () ;
|
|
g_pProvThrd = new CEventProviderThread;
|
|
|
|
bContinue = TRUE;
|
|
}
|
|
catch ( ... )
|
|
{
|
|
if ( g_pWorkerThread )
|
|
{
|
|
g_pWorkerThread->SignalThreadShutdown();
|
|
g_pWorkerThread = NULL;
|
|
}
|
|
|
|
g_pProvThrd = NULL;
|
|
|
|
SnmpClassLibrary :: Closedown () ;
|
|
SnmpDebugLog :: Closedown () ;
|
|
SnmpThreadObject :: Closedown () ;
|
|
ProviderClosedown () ;
|
|
|
|
throw;
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
bContinue = TRUE;
|
|
}
|
|
|
|
if ( bContinue )
|
|
{
|
|
CTrapEventProvider* prov = new CTrapEventProvider(CMapToEvent::EMappingType::ENCAPSULATED_MAPPER, g_pProvThrd);
|
|
|
|
LeaveCriticalSection ( & s_ProviderCriticalSection ) ;
|
|
bEnteredCritSec = FALSE;
|
|
|
|
status = prov->QueryInterface (riid, ppvObject);
|
|
|
|
if (NOERROR != status)
|
|
{
|
|
delete prov;
|
|
}
|
|
}
|
|
|
|
if ( bEnteredCritSec )
|
|
{
|
|
LeaveCriticalSection ( & s_ProviderCriticalSection ) ;
|
|
bEnteredCritSec = FALSE;
|
|
}
|
|
}
|
|
|
|
return status ;
|
|
}
|
|
catch(Structured_Exception e_SE)
|
|
{
|
|
if ( bEnteredCritSec )
|
|
{
|
|
LeaveCriticalSection ( & s_ProviderCriticalSection ) ;
|
|
bEnteredCritSec = FALSE;
|
|
}
|
|
|
|
return E_UNEXPECTED;
|
|
}
|
|
catch(Heap_Exception e_HE)
|
|
{
|
|
if ( bEnteredCritSec )
|
|
{
|
|
LeaveCriticalSection ( & s_ProviderCriticalSection ) ;
|
|
bEnteredCritSec = FALSE;
|
|
}
|
|
|
|
return E_OUTOFMEMORY;
|
|
}
|
|
catch(...)
|
|
{
|
|
if ( bEnteredCritSec )
|
|
{
|
|
LeaveCriticalSection ( & s_ProviderCriticalSection ) ;
|
|
bEnteredCritSec = FALSE;
|
|
}
|
|
|
|
return E_UNEXPECTED;
|
|
}
|
|
}
|
|
|
|
//***************************************************************************
|
|
//
|
|
// CSNMPRefEventProviderClassFactory::CreateInstance
|
|
//
|
|
// Purpose: Instantiates a Provider object returning an interface pointer.
|
|
//
|
|
// Parameters:
|
|
// pUnkOuter LPUNKNOWN to the controlling IUnknown if we are
|
|
// being used in an aggregation.
|
|
// riid REFIID identifying the interface the caller
|
|
// desires to have for the new object.
|
|
// ppvObj PPVOID in which to store the desired
|
|
// interface pointer for the new object.
|
|
//
|
|
// Return Value:
|
|
// HRESULT NOERROR if successful, otherwise E_NOINTERFACE
|
|
// if we cannot support the requested interface.
|
|
//***************************************************************************
|
|
|
|
STDMETHODIMP CSNMPRefEventProviderClassFactory :: CreateInstance(LPUNKNOWN pUnkOuter ,
|
|
REFIID riid,
|
|
LPVOID FAR * ppvObject
|
|
)
|
|
{
|
|
SetStructuredExceptionHandler seh;
|
|
|
|
BOOL bEnteredCritSec = FALSE;
|
|
BOOL bContinue = FALSE;
|
|
|
|
try
|
|
{
|
|
HRESULT status = E_FAIL;
|
|
|
|
if ( pUnkOuter )
|
|
{
|
|
status = CLASS_E_NOAGGREGATION;
|
|
}
|
|
else
|
|
{
|
|
EnterCriticalSection ( & s_ProviderCriticalSection ) ;
|
|
bEnteredCritSec = TRUE;
|
|
|
|
if (NULL == g_pProvThrd)
|
|
{
|
|
ProviderStartup () ;
|
|
|
|
SnmpThreadObject :: Startup () ;
|
|
try
|
|
{
|
|
SnmpDebugLog :: Startup () ;
|
|
}
|
|
catch ( ... )
|
|
{
|
|
// we do not want to left s_Reference count up
|
|
SnmpThreadObject :: Closedown () ;
|
|
ProviderClosedown () ;
|
|
|
|
throw;
|
|
}
|
|
|
|
BOOL bStatus = FALSE;
|
|
try
|
|
{
|
|
bStatus = SnmpClassLibrary :: Startup () ;
|
|
}
|
|
catch ( ... )
|
|
{
|
|
// we do not want to left s_Reference count up
|
|
SnmpDebugLog :: Closedown () ;
|
|
SnmpThreadObject :: Closedown () ;
|
|
ProviderClosedown () ;
|
|
|
|
throw;
|
|
}
|
|
|
|
if ( bStatus == FALSE )
|
|
{
|
|
SnmpDebugLog :: Closedown () ;
|
|
SnmpThreadObject :: Closedown () ;
|
|
|
|
ProviderClosedown () ;
|
|
}
|
|
else
|
|
{
|
|
try
|
|
{
|
|
g_pWorkerThread = new CEventProviderWorkerThread;
|
|
g_pWorkerThread->BeginThread();
|
|
g_pWorkerThread->WaitForStartup();
|
|
g_pWorkerThread->CreateServerWrap () ;
|
|
g_pProvThrd = new CEventProviderThread;
|
|
|
|
bContinue = TRUE;
|
|
}
|
|
catch ( ... )
|
|
{
|
|
if ( g_pWorkerThread )
|
|
{
|
|
g_pWorkerThread->SignalThreadShutdown();
|
|
g_pWorkerThread = NULL;
|
|
}
|
|
|
|
SnmpClassLibrary :: Closedown () ;
|
|
SnmpDebugLog :: Closedown () ;
|
|
SnmpThreadObject :: Closedown () ;
|
|
|
|
ProviderClosedown () ;
|
|
|
|
throw;
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
bContinue = TRUE;
|
|
}
|
|
|
|
if ( bContinue )
|
|
{
|
|
CTrapEventProvider* prov = new CTrapEventProvider(CMapToEvent::EMappingType::REFERENT_MAPPER, g_pProvThrd);
|
|
|
|
LeaveCriticalSection ( & s_ProviderCriticalSection ) ;
|
|
bEnteredCritSec = FALSE;
|
|
|
|
status = prov->QueryInterface (riid, ppvObject);
|
|
|
|
if (NOERROR != status)
|
|
{
|
|
delete prov;
|
|
}
|
|
}
|
|
|
|
if ( bEnteredCritSec )
|
|
{
|
|
LeaveCriticalSection ( & s_ProviderCriticalSection ) ;
|
|
bEnteredCritSec = FALSE;
|
|
}
|
|
}
|
|
|
|
return status ;
|
|
}
|
|
catch(Structured_Exception e_SE)
|
|
{
|
|
if ( bEnteredCritSec )
|
|
{
|
|
LeaveCriticalSection ( & s_ProviderCriticalSection ) ;
|
|
bEnteredCritSec = FALSE;
|
|
}
|
|
|
|
return E_UNEXPECTED;
|
|
}
|
|
catch(Heap_Exception e_HE)
|
|
{
|
|
if ( bEnteredCritSec )
|
|
{
|
|
LeaveCriticalSection ( & s_ProviderCriticalSection ) ;
|
|
bEnteredCritSec = FALSE;
|
|
}
|
|
|
|
return E_OUTOFMEMORY;
|
|
}
|
|
catch(...)
|
|
{
|
|
if ( bEnteredCritSec )
|
|
{
|
|
LeaveCriticalSection ( & s_ProviderCriticalSection ) ;
|
|
bEnteredCritSec = FALSE;
|
|
}
|
|
|
|
return E_UNEXPECTED;
|
|
}
|
|
}
|