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.
155 lines
5.3 KiB
155 lines
5.3 KiB
|
|
|
|
//***************************************************************************
|
|
//
|
|
// TASK.H
|
|
//
|
|
// raymcc 23-Apr-00 First oversimplified draft for Whistler
|
|
// raymcc 18-Mar-02 Security review
|
|
//
|
|
//***************************************************************************
|
|
|
|
#ifndef _WMITASK_H_
|
|
#define _WMITASK_H_
|
|
|
|
#define CORE_TASK_TYPE(x) (x & 0xFF)
|
|
#include <context.h>
|
|
|
|
// forward
|
|
class CAsyncReq;
|
|
class CProviderSink;
|
|
class CStatusSink;
|
|
|
|
struct STaskProvider
|
|
{
|
|
IWbemServices *m_pProv;
|
|
CProviderSink *m_pProvSink;
|
|
STaskProvider() { m_pProv = 0; m_pProvSink = 0; }
|
|
~STaskProvider();
|
|
|
|
HRESULT Cancel( CStatusSink* pStatusSink );
|
|
HRESULT ExecCancelOnNewRequest ( IWbemServices* pProv, CProviderSink* pSink, CStatusSink* pStatusSink ) ;
|
|
};
|
|
|
|
|
|
class CWmiTask : public _IWmiCoreHandle
|
|
{
|
|
friend struct STaskProvider;
|
|
|
|
private:
|
|
ULONG m_uRefCount;
|
|
ULONG m_uTaskType;
|
|
ULONG m_uTaskStatus;
|
|
ULONG m_uTaskId;
|
|
HRESULT m_hResult ;
|
|
BOOL m_bAccountedForThrottling ;
|
|
BOOL m_bCancelledDueToThrottling ;
|
|
CFlexArray m_aTaskProviders; // Array of STaskProvider structs
|
|
IWbemObjectSink * m_pAsyncClientSink; // Used for cross-ref purposes only
|
|
CStdSink * m_pReqSink; // The CStdSink pointer for each request
|
|
LONG m_uMemoryUsage;
|
|
ULONG m_uTotalSleepTime;
|
|
ULONG m_uCancelState;
|
|
ULONG m_uLastSleepTime;
|
|
HANDLE m_hCompletion ;
|
|
HANDLE m_hTimer;
|
|
CWbemContext* m_pMainCtx;
|
|
CFlexArray m_aArbitratees;
|
|
CCritSec m_csTask;
|
|
CWbemNamespace *m_pNs;
|
|
DWORD m_uStartTime;
|
|
CStatusSink * m_pStatusSink;
|
|
CAsyncReq_RemoveNotifySink * m_pReqCancelNotSink;
|
|
CFlexArray m_aTaskProvStorage;
|
|
CFlexArray m_aArbitrateesStorage;
|
|
CAsyncReq * m_pReqDoNotUse; // this is just a pointer copy for the debugger
|
|
|
|
|
|
CWmiTask( );
|
|
~CWmiTask();
|
|
|
|
CWmiTask(const CWmiTask &Src){};
|
|
CWmiTask & operator=(const CWmiTask &Src){};
|
|
|
|
static CStaticCritSec m_TaskCs;
|
|
|
|
public:
|
|
|
|
ULONG STDMETHODCALLTYPE AddRef();
|
|
ULONG STDMETHODCALLTYPE Release();
|
|
|
|
HRESULT STDMETHODCALLTYPE QueryInterface( IN REFIID riid, OUT LPVOID *ppvObj);
|
|
CWbemContext *GetCtx() { return m_pMainCtx; }
|
|
|
|
HRESULT STDMETHODCALLTYPE GetHandleType(ULONG *puType);
|
|
|
|
static CWmiTask *CreateTask( ) ;
|
|
|
|
ULONG GetTaskType() { return m_uTaskType; }
|
|
|
|
ULONG GetTaskStatus() { return m_uTaskStatus; }
|
|
HANDLE GetTimerHandle() { return m_hTimer; }
|
|
|
|
void RemoveTaskStatusBit(ULONG uMask) { m_uTaskStatus &= !uMask; }
|
|
|
|
VOID SetCancelledState ( BOOL bState ) { m_bCancelledDueToThrottling = bState ; }
|
|
BOOL GetCancelledState ( ) { return m_bCancelledDueToThrottling ; }
|
|
|
|
HRESULT SignalCancellation ( ) ;
|
|
HRESULT ReleaseArbitratees (BOOL bIsShutdown = FALSE ) ;
|
|
HRESULT SetTaskResult ( HRESULT hRes ) ;
|
|
HRESULT AddTaskProv(STaskProvider *);
|
|
BOOL IsESSNamespace ( );
|
|
BOOL IsProviderNamespace ( );
|
|
|
|
HRESULT GetFinalizer( _IWmiFinalizer **ppFnz );
|
|
BOOL IsAccountedForThrottling ( ) { return m_bAccountedForThrottling ; }
|
|
VOID SetAccountedForThrottling ( BOOL bSet ) { m_bAccountedForThrottling = bSet ; }
|
|
|
|
HRESULT HasMatchingSink(void *Test, IN REFIID riid);
|
|
HRESULT Cancel( HRESULT hRes = WBEM_E_CALL_CANCELLED );
|
|
HRESULT GetPrimaryTask ( _IWmiCoreHandle** pPTask );
|
|
|
|
HRESULT AddArbitratee(ULONG uFlags, _IWmiArbitratee* pArbitratee);
|
|
HRESULT RemoveArbitratee(ULONG uFlags, _IWmiArbitratee* pArbitratee);
|
|
|
|
HRESULT GetArbitratedQuery( ULONG uFlags, _IWmiArbitratedQuery** ppArbitratedQuery );
|
|
|
|
HRESULT GetMemoryUsage ( ULONG* uMemUsage ){ *uMemUsage = m_uMemoryUsage; return WBEM_S_NO_ERROR; } // SEC:REVIEWED 2002-03-22 : Needs ptr check
|
|
HRESULT UpdateMemoryUsage ( LONG lDelta ) ;
|
|
|
|
HRESULT GetTotalSleepTime ( ULONG* uSleepTime ){ *uSleepTime = m_uTotalSleepTime; return WBEM_S_NO_ERROR; } // SEC:REVIEWED 2002-03-22 : Needs ptr check
|
|
HRESULT UpdateTotalSleepTime ( ULONG uSleepTime ) ;
|
|
|
|
HRESULT GetCancelState ( ULONG* uCancelState ){ *uCancelState = m_uCancelState; return WBEM_S_NO_ERROR; } // SEC:REVIEWED 2002-03-22 : Needs ptr check
|
|
HRESULT SetCancelState ( ULONG uCancelState ){ m_uCancelState = uCancelState; return WBEM_S_NO_ERROR; }
|
|
|
|
HRESULT SetLastSleepTime ( ULONG uSleep ){ m_uLastSleepTime = uSleep; return WBEM_S_NO_ERROR; }
|
|
|
|
HRESULT CreateTimerEvent ( );
|
|
|
|
HRESULT SetArbitrateesOperationResult ( ULONG, HRESULT );
|
|
|
|
HRESULT Initialize(
|
|
IN CWbemNamespace *pNs,
|
|
IN ULONG uTaskType,
|
|
IN IWbemContext *pCtx,
|
|
IN IWbemObjectSink *pClientSinkCopy,
|
|
IN CAsyncReq *pReq);
|
|
|
|
HRESULT AssertProviderRef(IWbemServices *pProv);
|
|
HRESULT RetractProviderRef(IWbemServices *pProv);
|
|
|
|
ULONG GetTaskId() { return m_uTaskId; }
|
|
|
|
HRESULT Dump(FILE* f); // Debug only
|
|
HRESULT SetRequestSink(CStdSink *pSnk);
|
|
|
|
CWbemNamespace* GetNamespace ( ) { return m_pNs; }
|
|
};
|
|
|
|
typedef CWmiTask *PCWmiTask;
|
|
|
|
|
|
#endif
|
|
|