105 lines
2.7 KiB
C++
105 lines
2.7 KiB
C++
// SystemStatusListener.cpp: implementation of the CSystemStatusListener class.
|
|
//
|
|
//////////////////////////////////////////////////////////////////////
|
|
|
|
#include "stdafx.h"
|
|
#include "snapin.h"
|
|
#include "SystemStatusListener.h"
|
|
#include "HMSystemStatus.h"
|
|
#include "EventManager.h"
|
|
|
|
#ifdef _DEBUG
|
|
#undef THIS_FILE
|
|
static char THIS_FILE[]=__FILE__;
|
|
#define new DEBUG_NEW
|
|
#endif
|
|
|
|
IMPLEMENT_DYNCREATE(CSystemStatusListener,CWbemEventListener)
|
|
|
|
//////////////////////////////////////////////////////////////////////
|
|
// Construction/Destruction
|
|
//////////////////////////////////////////////////////////////////////
|
|
|
|
CSystemStatusListener::CSystemStatusListener()
|
|
{
|
|
SetEventQuery(IDS_STRING_STATUS_EVENTQUERY);
|
|
}
|
|
|
|
CSystemStatusListener::~CSystemStatusListener()
|
|
{
|
|
Destroy();
|
|
}
|
|
|
|
//////////////////////////////////////////////////////////////////////
|
|
// Event Processing Members
|
|
//////////////////////////////////////////////////////////////////////
|
|
|
|
HRESULT CSystemStatusListener::ProcessEventClassObject(IWbemClassObject* pClassObject)
|
|
{
|
|
ASSERT(pClassObject);
|
|
if( pClassObject == NULL )
|
|
{
|
|
return E_FAIL;
|
|
}
|
|
|
|
HRESULT hr = S_OK;
|
|
|
|
CWbemClassObject StatusObject;
|
|
|
|
if( ! CHECKHRESULT(hr = StatusObject.Create(pClassObject)) )
|
|
{
|
|
return hr;
|
|
}
|
|
|
|
StatusObject.SetMachineName(GetObjectPtr()->GetSystemName());
|
|
|
|
EvtGetEventManager()->ProcessEvent(&StatusObject);
|
|
|
|
return hr;
|
|
}
|
|
|
|
inline HRESULT CSystemStatusListener::OnSetStatus(long lFlags, HRESULT hResult, BSTR strParam, IWbemClassObject __RPC_FAR *pObjParam)
|
|
{
|
|
TRACEX(_T("CSystemStatusListener::OnSetStatus\n"));
|
|
TRACEARGn(lFlags);
|
|
TRACEARGn(hResult);
|
|
TRACEARGs(strParam);
|
|
TRACEARGn(pObjParam);
|
|
|
|
if( lFlags == 0L && hResult != S_OK )
|
|
{
|
|
if( hResult == WBEM_E_CALL_CANCELLED )
|
|
return WBEM_NO_ERROR;
|
|
|
|
GetObjectPtr()->SetState(HMS_NODATA,true,true);
|
|
|
|
CRuleEvent* pUnknownEvent = new CRuleEvent;
|
|
|
|
// create the GUID
|
|
GUID ChildGuid;
|
|
CoCreateGuid(&ChildGuid);
|
|
|
|
OLECHAR szGuid[GUID_CCH];
|
|
::StringFromGUID2(ChildGuid, szGuid, GUID_CCH);
|
|
pUnknownEvent->m_sStatusGuid = OLE2CT(szGuid);
|
|
pUnknownEvent->m_iState = 6;
|
|
pUnknownEvent->m_sName.LoadString(IDS_STRING_CONNECTION);
|
|
pUnknownEvent->m_sSystemName = GetObjectPtr()->GetSystemName();
|
|
CTime time = CTime::GetCurrentTime();
|
|
time.GetAsSystemTime(pUnknownEvent->m_st);
|
|
|
|
CnxGetErrorString(hResult,GetObjectPtr()->GetSystemName(),pUnknownEvent->m_sMessage);
|
|
|
|
EvtGetEventManager()->ProcessUnknownEvent(GetObjectPtr()->GetSystemName(),pUnknownEvent);
|
|
GetObjectPtr()->UpdateStatus();
|
|
|
|
}
|
|
else if( lFlags >= 1 && hResult == S_OK )
|
|
{
|
|
//GetObjectPtr()->Refresh();
|
|
}
|
|
|
|
|
|
return WBEM_NO_ERROR;
|
|
}
|