157 lines
3.9 KiB
C++
157 lines
3.9 KiB
C++
/*++
|
|
|
|
Copyright (C) 1996-2001 Microsoft Corporation
|
|
|
|
Module Name:
|
|
|
|
Abstract:
|
|
|
|
History:
|
|
|
|
--*/
|
|
|
|
|
|
#ifndef __SMRTMRSH_H__
|
|
#define __SMRTMRSH_H__
|
|
|
|
#include <wbemcli.h>
|
|
#include <wstlallc.h>
|
|
#include <wmimsg.h>
|
|
#include <comutl.h>
|
|
#include <sync.h>
|
|
#include <unk.h>
|
|
#include <map>
|
|
|
|
#define HASHSIZE 16
|
|
|
|
class CHash
|
|
{
|
|
BYTE m_achHash[HASHSIZE];
|
|
|
|
public:
|
|
|
|
CHash( PBYTE pHash ) { memcpy( m_achHash, pHash, HASHSIZE ); }
|
|
|
|
bool operator < ( const CHash& rHash ) const
|
|
{
|
|
return memcmp( m_achHash, rHash.m_achHash, HASHSIZE ) < 0;
|
|
}
|
|
|
|
};
|
|
|
|
|
|
/***************************************************************************
|
|
CSmartObjectMarshaler
|
|
****************************************************************************/
|
|
|
|
class CSmartObjectMarshaler
|
|
: public CUnkBase< IWmiObjectMarshal, &IID_IWmiObjectMarshal >
|
|
{
|
|
CCritSec m_cs;
|
|
std::map<CHash, BOOL, std::less<CHash>, wbem_allocator<BOOL> > m_SentMap;
|
|
|
|
HRESULT GetMaxMarshalSize( IWbemClassObject* pObj,
|
|
LPCWSTR wszNamespace,
|
|
DWORD dwFlags,
|
|
ULONG* pulSize );
|
|
|
|
HRESULT InternalPack( IWbemClassObject* pObj,
|
|
LPCWSTR wszNamespace,
|
|
DWORD dwFlags,
|
|
ULONG cBuff,
|
|
PBYTE pBuff,
|
|
ULONG* pcUsed );
|
|
public:
|
|
|
|
CSmartObjectMarshaler( CLifeControl* pCtl, IUnknown* pUnk = NULL )
|
|
: CUnkBase< IWmiObjectMarshal, &IID_IWmiObjectMarshal >( pCtl )
|
|
{
|
|
}
|
|
|
|
STDMETHOD(Pack)( IWbemClassObject* pObj,
|
|
LPCWSTR wszNamespace,
|
|
DWORD dwFlags,
|
|
ULONG cBuff,
|
|
PBYTE pBuff,
|
|
ULONG* pcUsed );
|
|
|
|
STDMETHOD(Unpack)( ULONG cBuff,
|
|
PBYTE pBuff,
|
|
DWORD dwFlags,
|
|
IWbemClassObject** pObj,
|
|
ULONG* pcUsed )
|
|
{
|
|
return WBEM_E_NOT_SUPPORTED;
|
|
}
|
|
|
|
STDMETHOD(Flush)();
|
|
};
|
|
|
|
/***************************************************************************
|
|
CSmartObjectUnmarshaler
|
|
****************************************************************************/
|
|
|
|
class CSmartObjectUnmarshaler
|
|
: public CUnkBase< IWmiObjectMarshal, &IID_IWmiObjectMarshal >
|
|
{
|
|
CCritSec m_cs;
|
|
|
|
ULONG m_ulMaxCacheSize;
|
|
ULONG m_ulCacheSize;
|
|
|
|
CWbemPtr<IMarshal> m_pUnmrsh;
|
|
CWbemPtr<IWbemLocator> m_pLocator;
|
|
CWbemPtr<IWbemClassObject> m_pEmptyClass;
|
|
|
|
struct CacheRecord
|
|
{
|
|
DWORD m_dwClassSize;
|
|
DWORD m_dwLastUsedTime;
|
|
CWbemPtr<IWbemClassObject> m_pClassPart;
|
|
};
|
|
|
|
typedef std::map< CHash,
|
|
CacheRecord,
|
|
std::less<CHash>,
|
|
wbem_allocator<CacheRecord> > ClassPartMap;
|
|
|
|
ClassPartMap m_Cache;
|
|
|
|
void MakeRoomInCache( DWORD dwSize );
|
|
|
|
HRESULT EnsureInitialized();
|
|
|
|
HRESULT CacheClassPart(PBYTE pHash, DWORD dwSize, IWbemClassObject* pInst);
|
|
|
|
HRESULT FindClassPart( PBYTE pClassPartHash,
|
|
LPCWSTR wszClassPath,
|
|
IWbemClassObject** ppClassPart );
|
|
|
|
public:
|
|
|
|
CSmartObjectUnmarshaler( CLifeControl* pCtl, IUnknown* pUnk = NULL )
|
|
: CUnkBase< IWmiObjectMarshal, &IID_IWmiObjectMarshal >( pCtl ),
|
|
m_ulMaxCacheSize(0x500000), m_ulCacheSize(0) { }
|
|
|
|
STDMETHOD(Pack)( IWbemClassObject* pObj,
|
|
LPCWSTR wszNamespace,
|
|
DWORD dwFlags,
|
|
ULONG cBuff,
|
|
PBYTE pBuff,
|
|
ULONG* pcUsed )
|
|
{
|
|
return WBEM_E_NOT_SUPPORTED;
|
|
}
|
|
|
|
STDMETHOD(Unpack)( ULONG cBuff,
|
|
PBYTE pBuff,
|
|
DWORD dwFlags,
|
|
IWbemClassObject** pObj,
|
|
ULONG* pcUsed );
|
|
|
|
STDMETHOD(Flush)();
|
|
};
|
|
|
|
#endif __SMRTMRSH_H__
|
|
|