128 lines
3.5 KiB
C++
128 lines
3.5 KiB
C++
#pragma option push -b -a8 -pc -A- /*P_O_Push*/
|
|
//
|
|
// MethodContext.h -- declaration of MethodContext class
|
|
//
|
|
// Copyright 1998 - 1999 Microsoft Corporation
|
|
//
|
|
//
|
|
//=================================================================
|
|
|
|
#if _MSC_VER > 1000
|
|
#pragma once
|
|
#endif
|
|
|
|
#ifndef _METHOD_CONTEXT_H__
|
|
#define _METHOD_CONTEXT_H__
|
|
|
|
#include "ThreadBase.h"
|
|
#include "refptrcollection.h"
|
|
#ifdef PROVIDER_INSTRUMENTATION
|
|
#include "StopWatch.h"
|
|
#endif
|
|
|
|
class CInstance;
|
|
class Provider;
|
|
class MethodContext;
|
|
|
|
typedef HRESULT (WINAPI * LPProviderInstanceCallback)(Provider* pProvider, CInstance* pInstance, MethodContext* pContext, void* pUserData);
|
|
|
|
//////////////////////////////////////////////////////
|
|
//
|
|
// STRUCT MethodContext
|
|
//
|
|
// a little something to make sure we can keep our threads from getting tangled
|
|
// idea is that there is one MethodContext for each request from CIMOM or another provider
|
|
// pointers are passed around.
|
|
//////////////////////////////////////////////////////
|
|
class POLARITY MethodContext : public CThreadBase
|
|
{
|
|
public:
|
|
MethodContext(IWbemContext __RPC_FAR* piContext);
|
|
~MethodContext();
|
|
|
|
virtual HRESULT Commit(CInstance *pInstance) = 0;
|
|
virtual IWbemContext __RPC_FAR* GetIWBEMContext();
|
|
|
|
ULONG AddRef(void);
|
|
ULONG Release(void);
|
|
virtual void QueryPostProcess(void);
|
|
|
|
bool SetStatusObject(IWbemClassObject* pObj);
|
|
IWbemClassObject __RPC_FAR* GetStatusObject();
|
|
|
|
#ifdef PROVIDER_INSTRUMENTATION
|
|
StopWatch* pStopWatch;
|
|
#endif
|
|
|
|
private:
|
|
IWbemContext __RPC_FAR* m_pContext;
|
|
IWbemClassObject __RPC_FAR* m_pStatusObject;
|
|
};
|
|
|
|
// for queries and suchlike that originate in CIMOM
|
|
class ExternalMethodContext : public MethodContext
|
|
{
|
|
public:
|
|
ExternalMethodContext(IWbemObjectSink __RPC_FAR* pResponseHandler,
|
|
IWbemContext __RPC_FAR* pContext,
|
|
void* pReserved = NULL
|
|
);
|
|
|
|
HRESULT Commit(CInstance *pInstance);
|
|
virtual void QueryPostProcess(void);
|
|
|
|
ULONG AddRef(void);
|
|
ULONG Release(void);
|
|
|
|
private:
|
|
IWbemObjectSink __RPC_FAR* m_pResponseHandler;
|
|
void * m_pReserved;
|
|
};
|
|
|
|
// for queries and suchlike that come from within.
|
|
// contains a list of objects returned.
|
|
class InternalMethodContext : public MethodContext
|
|
{
|
|
public:
|
|
InternalMethodContext(TRefPointerCollection<CInstance>* pList ,
|
|
IWbemContext __RPC_FAR* pContext);
|
|
~InternalMethodContext();
|
|
|
|
HRESULT Commit(CInstance *pInstance);
|
|
|
|
ULONG AddRef(void);
|
|
ULONG Release(void);
|
|
|
|
private:
|
|
TRefPointerCollection<CInstance>* m_pInstances;
|
|
};
|
|
|
|
// for queries and suchlike that come from within.
|
|
// "Asynch" is a bit of a misnomer - but it does help support
|
|
// asynchronous calls, in that each instance committed is routed
|
|
// to a callback function supplied by the requester
|
|
class InternalMethodContextAsynch : public MethodContext
|
|
{
|
|
public:
|
|
InternalMethodContextAsynch(Provider* pThat,
|
|
LPProviderInstanceCallback pCallback,
|
|
IWbemContext __RPC_FAR* pContext,
|
|
MethodContext* pUsersContext,
|
|
void* pUserData);
|
|
~InternalMethodContextAsynch();
|
|
|
|
HRESULT Commit(CInstance *pInstance);
|
|
|
|
ULONG AddRef(void);
|
|
ULONG Release(void);
|
|
|
|
private:
|
|
Provider* m_pThat;
|
|
LPProviderInstanceCallback m_pCallback;
|
|
void* m_pUserData;
|
|
MethodContext* m_pUsersContext;
|
|
};
|
|
|
|
#endif
|
|
#pragma option pop /*P_O_Pop*/
|