在《WMI技术介绍和应用——Event Provider》和《WMI技术介绍和应用——接收事件》中,我们展现了如何处理和事件相关的WMI知识。而《WMI技术介绍和应用——接收事件》一文则主要讲解了如何查询事件,这种查询是在我们进程存在时发生的,一旦我们进程不存在了,这种查询也无法执行。我们将这种行为称为消费事件,我们执行查询的进程叫做“事件临时消费者”。相对应的,WMI还存在“事件永久消费者”,它并不寄生我们的编写的进程中。本文主要讲解“事件永久消费者”的编写方法。(转载请指明出于breaksoftware的csdn博客)
由于VS2005没有这种模板,所以我便使用事件提供者模板生成了一个工程,并对这个工程进行改造。我们先从mof文件出发
由于我们不需要注册事件提供者,所以我们删除instance of __EventProviderRegistration内容。
我们先要声明一个事件消费类
class TestEventConsumer : __EventConsumer
{
[key] string name;
[write] string value;
};
实例化该类
instance of TestEventConsumer as $Consumer
{
name = "TestEventConsumer_Name1";
value = "TestEventConsumer_Value";
};
再注册一个事件消费者提供者实例
instance of __EventConsumerProviderRegistration
{
Provider = $EventConsumer;
ConsumerClassNames = {"TestEventConsumer"};
};
ConsumerClassNames是一个数组,它记录了事件消费者名。
然后要实例化一个筛选器
instance of __EventFilter as $Filter
{
Name = "TestEventFilter";
QueryLanguage = "WQL";
Query = "SELECT * FROM __InstanceCreationEvent WITHIN 1 WHERE TargetInstance ISA 'Win32_USBCOntrollerDevice'";
EventNamespace = "\\\\.\\root\\CIMV2";
};
其中定义了筛选器的名字、查询的语言、查询的命令和需要查询的命名空间。如果没有指定EventNamespace,则使用该mof中默认的命名空间。在本例中,我们要监控USB设备的创建,故要监控CIMV2空间。这儿需要注意的是,我们可以Query内部事件或者外部事件。
最后我们将筛选器绑定到消费者上。
instance of __FilterToConsumerBinding
{
Consumer = $Consumer;
Filter = $Filter;
};
如此,我们便将mof文件构建好了。
我们再回到cpp文件中,我们首先要定义一个事件消费者Sink
class CEventConsumerSink:
public CComObjectRootEx<CComMultiThreadModel>,
public IWbemUnboundObjectSink
{
public:
CEventConsumerSink() {}
~CEventConsumerSink()
{
}
BEGIN_COM_MAP(CEventConsumerSink)
COM_INTERFACE_ENTRY(IWbemUnboundObjectSink)
END_COM_MAP()
// IWbemUnboundObjectSink
public:
STDMETHOD(IndicateToConsumer)(IWbemClassObject* pLogicalConsumer, long lNumObjects, IWbemClassObject** apObjects);
};
我们要实现IndicateToConsumer方法,将事件传到到该方法中处理
STDMETHODIMP CEventConsumerSink::IndicateToConsumer(
IWbemClassObject* pLogicalConsumer, long lNumObjects, IWbemClassObject** apObjects)
{
ObjectLock lock(this);
HRESULT hr = WBEM_S_NO_ERROR;
CComVariant varClass;
hr = pLogicalConsumer->Get(CComBSTR("__CLASS"), 0, &varClass, 0, 0);
if (0 == _wcsicmp(V_BSTR(&varClass), L"TestEventConsumer")) {
for (long lIndex = 0; lIndex < lNumObjects; lIndex++) {
CComVariant varEventType;
hr = apObjects[lIndex]->Get(CComBSTR("TargetInstance"), 0, &varEventType, 0, 0);
CComQIPtr<IWbemClassObject> spEvent = V_UNKNOWN(&varEventType);
CComVariant varEventTypeClass;
hr = spEvent->Get(CComBSTR("__CLASS"), 0, &varEventTypeClass, 0, 0);
OutputTrace("CEventConsumerSink::IndicateToConsumer");
}
}
else {
hr = WBEM_E_NOT_FOUND;
}
return hr;
}
这段逻辑,我们获取事件名称,并检测其是否是我们需要监控的事件。如果是,则遍历事件数组——发来的事件是一批的。
然后我们改造主类。事件消费者提供者不需要继承IWbemInstProviderImpl,但是要继承IWbemEventConsumerProvider。于是我们在模板生成的代码中将其他无关的方法去掉
class ATL_NO_VTABLE CEventConsumer :
public CComObjectRootEx<CComMultiThreadModel>,
public CComCoClass<CEventConsumer, &CLSID_EventConsumer>,
public IWbemProviderInit,
public IWbemEventConsumerProvider
{
public:
CEventConsumer(){
}
~CEventConsumer(){
}
DECLARE_REGISTRY_RESOURCEID(IDR_EVENTCONSUMER)
DECLARE_NOT_AGGREGATABLE(CEventConsumer)
BEGIN_COM_MAP(CEventConsumer)
COM_INTERFACE_ENTRY(IWbemProviderInit)
COM_INTERFACE_ENTRY(IWbemEventConsumerProvider)
END_COM_MAP()
//IWbemProviderInit
HRESULT STDMETHODCALLTYPE Initialize(
__in_opt LPWSTR pszUser,
LONG lFlags,
__in LPWSTR pszNamespace,
__in_opt LPWSTR pszLocale,
IWbemServices *pNamespace,
IWbemContext *pCtx,
IWbemProviderInitSink *pInitSink);
// IWbemEventConsumerProvider
HRESULT STDMETHODCALLTYPE FindConsumer(
/* [in] */ IWbemClassObject *pLogicalConsumer,
/* [out] */ IWbemUnboundObjectSink **ppConsumer);
};
Initialize中删除模板生成的多余代码就可以了,我们主要要实现的就是FindConsumer方法
STDMETHODIMP CEventConsumer::FindConsumer(
/* [in] */ IWbemClassObject *pLogicalConsumer,
/* [out] */ IWbemUnboundObjectSink **ppConsumer)
{
HRESULT hr = WBEM_E_NOT_FOUND;
*ppConsumer = NULL;
CComVariant varClass;
hr = pLogicalConsumer->Get(L"__Class", 0, &varClass, 0, 0);
if (0 == _wcsicmp(V_BSTR(&varClass), L"TestEventConsumer")) {
CComObject<CEventConsumerSink>* pEventSink = NULL;
hr = CComObject<CEventConsumerSink>::CreateInstance(&pEventSink);
if (FAILED(hr)) {
return hr;
}
hr = pEventSink->QueryInterface(IID_IWbemUnboundObjectSink, (LPVOID*)ppConsumer);
if (FAILED(hr)) {
return hr;
}
}
else {
hr = WBEM_E_NOT_FOUND;
}
return hr;
}
这段逻辑我们在检测完消费者类是否正确后,就去获取消费者指针。
如此Event Consumer Provider便搭建起来了。