211 lines
4.2 KiB
C++
211 lines
4.2 KiB
C++
//***************************************************************************
|
|
|
|
//
|
|
|
|
// EVTPROV.CPP
|
|
|
|
//
|
|
|
|
// Module: WBEM MS SNMP EVENT PROVIDER
|
|
|
|
//
|
|
|
|
// Purpose: Contains the class factory. This creates objects when
|
|
|
|
// connections are requested.
|
|
|
|
//
|
|
|
|
// Copyright (c) 1997-2001 Microsoft Corporation, All Rights Reserved
|
|
//
|
|
//***************************************************************************
|
|
|
|
#include <windows.h>
|
|
#include <snmptempl.h>
|
|
#include <snmpmt.h>
|
|
#include <typeinfo.h>
|
|
#include <process.h>
|
|
#include <objbase.h>
|
|
#include <stdio.h>
|
|
#include <wbemidl.h>
|
|
#include <snmplog.h>
|
|
#include <snmpcl.h>
|
|
#include <snmpcont.h>
|
|
#include <snmptype.h>
|
|
#include <snmpauto.h>
|
|
#include <snmpevt.h>
|
|
#include <snmpthrd.h>
|
|
#include <snmpobj.h>
|
|
#include <classfac.h>
|
|
#include <evtdefs.h>
|
|
#include <evtthrd.h>
|
|
#include <evtmap.h>
|
|
#include <evtprov.h>
|
|
|
|
STDMETHODIMP CTrapEventProvider::Initialize (
|
|
LPWSTR pszUser,
|
|
LONG lFlags,
|
|
LPWSTR pszNamespace,
|
|
LPWSTR pszLocale,
|
|
IWbemServices *pCIMOM, // For anybody
|
|
IWbemContext *pCtx,
|
|
IWbemProviderInitSink *pInitSink // For init signals
|
|
)
|
|
{
|
|
DebugMacro9(
|
|
SnmpDebugLog :: s_SnmpDebugLog->Write (
|
|
__FILE__,__LINE__,
|
|
L"Entering CTrapEventProvider::Initialize\n");
|
|
)
|
|
|
|
m_pNamespace = pCIMOM;
|
|
m_pNamespace->AddRef();
|
|
pInitSink->SetStatus ( WBEM_S_INITIALIZED , 0 );
|
|
|
|
|
|
DebugMacro9(
|
|
SnmpDebugLog :: s_SnmpDebugLog->Write (
|
|
__FILE__,__LINE__,
|
|
L"Leaving CTrapEventProvider::Initialize with SUCCEEDED\n");
|
|
)
|
|
|
|
return WBEM_NO_ERROR;
|
|
}
|
|
|
|
|
|
STDMETHODIMP CTrapEventProvider::ProvideEvents(IWbemObjectSink* pSink, LONG lFlags)
|
|
{
|
|
DebugMacro9(
|
|
SnmpDebugLog :: s_SnmpDebugLog->Write (
|
|
__FILE__,__LINE__,
|
|
L"Entering CTrapEventProvider::ProvideEvents\n");
|
|
)
|
|
|
|
m_pEventSink = pSink;
|
|
m_pEventSink->AddRef();
|
|
|
|
if (!m_thrd->Register(this))
|
|
{
|
|
|
|
DebugMacro9(
|
|
SnmpDebugLog :: s_SnmpDebugLog->Write (
|
|
__FILE__,__LINE__,
|
|
L"Leaving CTrapEventProvider::ProvideEvents with FAILED\n");
|
|
)
|
|
|
|
return WBEM_E_FAILED;
|
|
}
|
|
|
|
DebugMacro9(
|
|
SnmpDebugLog :: s_SnmpDebugLog->Write (
|
|
__FILE__,__LINE__,
|
|
L"Leaving CNTEventProvider::ProvideEvents with SUCCEEDED\n");
|
|
)
|
|
|
|
return WBEM_NO_ERROR;
|
|
}
|
|
|
|
|
|
CTrapEventProvider::~CTrapEventProvider()
|
|
{
|
|
m_thrd->UnRegister(this);
|
|
}
|
|
|
|
|
|
CTrapEventProvider::CTrapEventProvider(DWORD mapperType, CEventProviderThread* thrd)
|
|
{
|
|
m_thrd = thrd;
|
|
m_MapType = mapperType;
|
|
m_ref = 0;
|
|
}
|
|
|
|
|
|
IWbemServices* CTrapEventProvider::GetNamespace()
|
|
{
|
|
m_pNamespace->AddRef();
|
|
return m_pNamespace;
|
|
}
|
|
|
|
IWbemObjectSink* CTrapEventProvider::GetEventSink()
|
|
{
|
|
m_pEventSink->AddRef();
|
|
return m_pEventSink;
|
|
}
|
|
void CTrapEventProvider::ReleaseAll()
|
|
{
|
|
//release dependencies
|
|
m_pNamespace->Release();
|
|
m_pEventSink->Release();
|
|
|
|
if ( 0 != InterlockedDecrement(&m_ref) )
|
|
{
|
|
return;
|
|
}
|
|
|
|
delete this;
|
|
InterlockedDecrement(&(CSNMPEventProviderClassFactory::objectsInProgress));
|
|
return;
|
|
}
|
|
|
|
void CTrapEventProvider::AddRefAll()
|
|
{
|
|
//addref dependencies
|
|
m_pNamespace->AddRef();
|
|
m_pEventSink->AddRef();
|
|
|
|
InterlockedIncrement(&(CSNMPEventProviderClassFactory::objectsInProgress));
|
|
InterlockedIncrement ( &m_ref ) ;
|
|
}
|
|
|
|
STDMETHODIMP_( ULONG ) CTrapEventProvider::AddRef()
|
|
{
|
|
InterlockedIncrement(&(CSNMPEventProviderClassFactory::objectsInProgress));
|
|
return InterlockedIncrement ( &m_ref ) ;
|
|
}
|
|
|
|
STDMETHODIMP_(ULONG) CTrapEventProvider::Release()
|
|
{
|
|
long ret;
|
|
|
|
if ( 0 != (ret = InterlockedDecrement(&m_ref)) )
|
|
{
|
|
InterlockedDecrement(&(CSNMPEventProviderClassFactory::objectsInProgress));
|
|
return ret;
|
|
}
|
|
|
|
delete this;
|
|
InterlockedDecrement(&(CSNMPEventProviderClassFactory::objectsInProgress));
|
|
return 0;
|
|
}
|
|
|
|
STDMETHODIMP CTrapEventProvider::QueryInterface(REFIID riid, PVOID* ppv)
|
|
{
|
|
*ppv = NULL;
|
|
|
|
if (IID_IUnknown == riid)
|
|
{
|
|
*ppv= (IWbemEventProvider*)this;
|
|
}
|
|
else if (IID_IWbemEventProvider == riid)
|
|
{
|
|
*ppv=(IWbemEventProvider*)this;
|
|
}
|
|
else if (IID_IWbemProviderInit == riid)
|
|
{
|
|
*ppv=(IWbemProviderInit*)this;
|
|
}
|
|
|
|
if (NULL==*ppv)
|
|
{
|
|
return E_NOINTERFACE;
|
|
}
|
|
|
|
//AddRef any interface we'll return.
|
|
((LPUNKNOWN)*ppv)->AddRef();
|
|
return NOERROR;
|
|
}
|
|
|
|
|
|
|
|
|