mirror of https://github.com/tongzx/nt5src
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.
520 lines
14 KiB
520 lines
14 KiB
/*++
|
|
|
|
Copyright (c) 1998 Microsoft Corporation
|
|
|
|
Module Name:
|
|
|
|
cpropbag.h
|
|
|
|
Abstract:
|
|
|
|
This module contains the definition of the
|
|
generic property bag class
|
|
|
|
Author:
|
|
|
|
Keith Lau ([email protected])
|
|
|
|
Revision History:
|
|
|
|
keithlau 06/30/98 created
|
|
|
|
--*/
|
|
|
|
#ifndef _CPROPBAG_H_
|
|
#define _CPROPBAG_H_
|
|
|
|
|
|
#include "filehc.h"
|
|
#include "mailmsg.h"
|
|
#include "cmmtypes.h"
|
|
#include "cleanback.h"
|
|
|
|
//
|
|
//Logging facilities
|
|
//
|
|
#include "inetcom.h"
|
|
#include "logtype.h"
|
|
|
|
|
|
/***************************************************************************/
|
|
// Definitions
|
|
//
|
|
|
|
#define GENERIC_PTABLE_INSTANCE_SIGNATURE_VALID ((DWORD)'PTGv')
|
|
|
|
|
|
/***************************************************************************/
|
|
// CMailMsgPropertyBag
|
|
//
|
|
|
|
//
|
|
// Disable warning about using the this pointer in the constructor
|
|
// (CCleanBack only saves the pointer, so this is safe)
|
|
//
|
|
#pragma warning( disable: 4355 )
|
|
|
|
class CMailMsgPropertyBag :
|
|
public IMailMsgPropertyBag,
|
|
public CCleanBack
|
|
{
|
|
public:
|
|
|
|
CMailMsgPropertyBag() :
|
|
CCleanBack((IUnknown *)(IMailMsgPropertyBag *)this),
|
|
m_bmBlockManager(NULL),
|
|
m_ptProperties(
|
|
PTT_PROPERTY_TABLE,
|
|
GENERIC_PTABLE_INSTANCE_SIGNATURE_VALID,
|
|
&m_bmBlockManager,
|
|
&m_InstanceInfo,
|
|
CompareProperty,
|
|
NULL,
|
|
NULL
|
|
)
|
|
{
|
|
m_lRefCount = 1;
|
|
|
|
// Copy the default instance into our instance
|
|
MoveMemory(
|
|
&m_InstanceInfo,
|
|
&s_DefaultInstanceInfo,
|
|
sizeof(PROPERTY_TABLE_INSTANCE));
|
|
}
|
|
|
|
~CMailMsgPropertyBag()
|
|
{
|
|
//
|
|
// Call all registered callbacks BEFORE destroying member
|
|
// variables (so that properties will still be accessible)
|
|
//
|
|
CallCallBacks();
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE QueryInterface(REFIID riid, LPVOID *ppvObj)
|
|
{
|
|
if (riid == IID_IUnknown)
|
|
*ppvObj = (IUnknown *)(IMailMsgPropertyBag *)this;
|
|
else if (riid == IID_IMailMsgPropertyBag)
|
|
*ppvObj = (IMailMsgPropertyBag *)this;
|
|
else if (riid == IID_IMailMsgRegisterCleanupCallback)
|
|
*ppvObj = (IMailMsgRegisterCleanupCallback *)this;
|
|
else
|
|
return(E_NOINTERFACE);
|
|
AddRef();
|
|
return(S_OK);
|
|
}
|
|
|
|
unsigned long STDMETHODCALLTYPE AddRef()
|
|
{
|
|
return(InterlockedIncrement(&m_lRefCount));
|
|
}
|
|
|
|
unsigned long STDMETHODCALLTYPE Release()
|
|
{
|
|
LONG lTemp = InterlockedDecrement(&m_lRefCount);
|
|
if (!lTemp)
|
|
{
|
|
// Extra releases are bad!
|
|
_ASSERT(lTemp);
|
|
}
|
|
return(lTemp);
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE PutProperty(
|
|
DWORD dwPropID,
|
|
DWORD cbLength,
|
|
LPBYTE pbValue
|
|
)
|
|
{
|
|
GLOBAL_PROPERTY_ITEM piItem;
|
|
piItem.idProp = dwPropID;
|
|
return(m_ptProperties.PutProperty(
|
|
(LPVOID)&dwPropID,
|
|
(LPPROPERTY_ITEM)&piItem,
|
|
cbLength,
|
|
pbValue));
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE GetProperty(
|
|
DWORD dwPropID,
|
|
DWORD cbLength,
|
|
DWORD *pcbLength,
|
|
LPBYTE pbValue
|
|
)
|
|
{
|
|
GLOBAL_PROPERTY_ITEM piItem;
|
|
return(m_ptProperties.GetPropertyItemAndValue(
|
|
(LPVOID)&dwPropID,
|
|
(LPPROPERTY_ITEM)&piItem,
|
|
cbLength,
|
|
pcbLength,
|
|
pbValue));
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE PutStringA(
|
|
DWORD dwPropID,
|
|
LPCSTR pszValue
|
|
)
|
|
{
|
|
return(PutProperty(dwPropID, pszValue?strlen(pszValue)+1:0, (LPBYTE)pszValue));
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE GetStringA(
|
|
DWORD dwPropID,
|
|
DWORD cchLength,
|
|
LPSTR pszValue
|
|
)
|
|
{
|
|
DWORD dwLength;
|
|
return(GetProperty(dwPropID, cchLength, &dwLength, (LPBYTE)pszValue));
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE PutStringW(
|
|
DWORD dwPropID,
|
|
LPCWSTR pszValue
|
|
)
|
|
{
|
|
return(PutProperty(dwPropID, pszValue?(wcslen(pszValue)+1)*sizeof(WCHAR):0, (LPBYTE)pszValue));
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE GetStringW(
|
|
DWORD dwPropID,
|
|
DWORD cchLength,
|
|
LPWSTR pszValue
|
|
)
|
|
{
|
|
DWORD dwLength;
|
|
return(GetProperty(dwPropID, cchLength*sizeof(WCHAR), &dwLength, (LPBYTE)pszValue));
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE PutDWORD(
|
|
DWORD dwPropID,
|
|
DWORD dwValue
|
|
)
|
|
{
|
|
return(PutProperty(dwPropID, sizeof(DWORD), (LPBYTE)&dwValue));
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE GetDWORD(
|
|
DWORD dwPropID,
|
|
DWORD *pdwValue
|
|
)
|
|
{
|
|
DWORD dwLength;
|
|
return(GetProperty(dwPropID, sizeof(DWORD), &dwLength, (LPBYTE)pdwValue));
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE PutBool(
|
|
DWORD dwPropID,
|
|
DWORD dwValue
|
|
)
|
|
{
|
|
dwValue = dwValue ? 1 : 0;
|
|
return(PutProperty(dwPropID, sizeof(DWORD), (LPBYTE)&dwValue));
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE GetBool(
|
|
DWORD dwPropID,
|
|
DWORD *pdwValue
|
|
)
|
|
{
|
|
HRESULT hrRes;
|
|
DWORD dwLength;
|
|
|
|
hrRes = GetProperty(dwPropID, sizeof(DWORD), &dwLength, (LPBYTE)pdwValue);
|
|
if (pdwValue)
|
|
*pdwValue = *pdwValue ? 1 : 0;
|
|
return (hrRes);
|
|
}
|
|
|
|
private:
|
|
|
|
// The specific compare function for this type of property table
|
|
static HRESULT CompareProperty(
|
|
LPVOID pvPropKey,
|
|
LPPROPERTY_ITEM pItem
|
|
);
|
|
|
|
private:
|
|
|
|
// Usage count
|
|
LONG m_lRefCount;
|
|
|
|
// Property table instance
|
|
PROPERTY_TABLE_INSTANCE m_InstanceInfo;
|
|
static PROPERTY_TABLE_INSTANCE s_DefaultInstanceInfo;
|
|
|
|
// IMailMsgProperties is an instance of CPropertyTable
|
|
CPropertyTable m_ptProperties;
|
|
|
|
// An instance of the block memory manager
|
|
CBlockManager m_bmBlockManager;
|
|
|
|
};
|
|
|
|
//
|
|
// Restore original warning settings
|
|
//
|
|
#pragma warning ( default: 4355 )
|
|
|
|
/***************************************************************************/
|
|
// CMailMsgLoggingPropertyBag
|
|
//
|
|
|
|
class __declspec(uuid("58f9a2d2-21ca-11d2-aa6b-00c04fa35b82")) CMailMsgLoggingPropertyBag :
|
|
public CMailMsgPropertyBag,
|
|
public IMailMsgLoggingPropertyBag
|
|
{
|
|
public:
|
|
|
|
CMailMsgLoggingPropertyBag()
|
|
{
|
|
m_lRefCount = 1;
|
|
m_pvLogHandle = NULL;
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE PutProperty(
|
|
DWORD dwPropID,
|
|
DWORD cbLength,
|
|
LPBYTE pbValue
|
|
)
|
|
{
|
|
HRESULT hrRes = S_OK;
|
|
|
|
m_rwLock.ExclusiveLock();
|
|
hrRes = CMailMsgPropertyBag::PutProperty(
|
|
dwPropID,
|
|
cbLength,
|
|
pbValue);
|
|
m_rwLock.ExclusiveUnlock();
|
|
return(hrRes);
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE GetProperty(
|
|
DWORD dwPropID,
|
|
DWORD cbLength,
|
|
DWORD *pcbLength,
|
|
LPBYTE pbValue
|
|
)
|
|
{
|
|
HRESULT hrRes = S_OK;
|
|
|
|
m_rwLock.ExclusiveLock();
|
|
hrRes = CMailMsgPropertyBag::GetProperty(
|
|
dwPropID,
|
|
cbLength,
|
|
pcbLength,
|
|
pbValue);
|
|
m_rwLock.ExclusiveUnlock();
|
|
return(hrRes);
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE PutStringA(
|
|
DWORD dwPropID,
|
|
LPCSTR pszValue
|
|
)
|
|
{
|
|
return(PutProperty(dwPropID, pszValue?strlen(pszValue)+1:0, (LPBYTE)pszValue));
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE GetStringA(
|
|
DWORD dwPropID,
|
|
DWORD cchLength,
|
|
LPSTR pszValue
|
|
)
|
|
{
|
|
DWORD dwLength;
|
|
return(GetProperty(dwPropID, cchLength, &dwLength, (LPBYTE)pszValue));
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE PutStringW(
|
|
DWORD dwPropID,
|
|
LPCWSTR pszValue
|
|
)
|
|
{
|
|
return(PutProperty(dwPropID, pszValue?(wcslen(pszValue)+1)*sizeof(WCHAR):0, (LPBYTE)pszValue));
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE GetStringW(
|
|
DWORD dwPropID,
|
|
DWORD cchLength,
|
|
LPWSTR pszValue
|
|
)
|
|
{
|
|
DWORD dwLength;
|
|
return(GetProperty(dwPropID, cchLength*sizeof(WCHAR), &dwLength, (LPBYTE)pszValue));
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE PutDWORD(
|
|
DWORD dwPropID,
|
|
DWORD dwValue
|
|
)
|
|
{
|
|
return(PutProperty(dwPropID, sizeof(DWORD), (LPBYTE)&dwValue));
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE GetDWORD(
|
|
DWORD dwPropID,
|
|
DWORD *pdwValue
|
|
)
|
|
{
|
|
DWORD dwLength;
|
|
return(GetProperty(dwPropID, sizeof(DWORD), &dwLength, (LPBYTE)pdwValue));
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE PutBool(
|
|
DWORD dwPropID,
|
|
DWORD dwValue
|
|
)
|
|
{
|
|
dwValue = dwValue ? 1 : 0;
|
|
return(PutProperty(dwPropID, sizeof(DWORD), (LPBYTE)&dwValue));
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE GetBool(
|
|
DWORD dwPropID,
|
|
DWORD *pdwValue
|
|
)
|
|
{
|
|
HRESULT hrRes;
|
|
DWORD dwLength;
|
|
|
|
hrRes = GetProperty(dwPropID, sizeof(DWORD), &dwLength, (LPBYTE)pdwValue);
|
|
if (pdwValue)
|
|
*pdwValue = *pdwValue ? 1 : 0;
|
|
return (hrRes);
|
|
}
|
|
|
|
HRESULT SetLogging(
|
|
LPVOID pvLogHandle
|
|
)
|
|
{
|
|
if (!pvLogHandle)
|
|
return(E_POINTER);
|
|
m_pvLogHandle = pvLogHandle;
|
|
return(S_OK);
|
|
}
|
|
|
|
static void SetInetLogInfoField(
|
|
LPCSTR pszInput,
|
|
LPSTR *ppszOutput,
|
|
DWORD *pdwOutput
|
|
)
|
|
{
|
|
if (pszInput)
|
|
{
|
|
*ppszOutput = (LPSTR) pszInput;
|
|
if (pdwOutput)
|
|
*pdwOutput = lstrlen(pszInput);
|
|
}
|
|
}
|
|
|
|
unsigned long STDMETHODCALLTYPE AddRef()
|
|
{
|
|
return(InterlockedIncrement(&m_lRefCount));
|
|
}
|
|
|
|
unsigned long STDMETHODCALLTYPE Release()
|
|
{
|
|
LONG lTemp = InterlockedDecrement(&m_lRefCount);
|
|
if (!lTemp)
|
|
{
|
|
// Extra releases are bad!
|
|
_ASSERT(lTemp);
|
|
}
|
|
return(lTemp);
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE QueryInterface(REFIID riid, LPVOID *ppvObj)
|
|
{
|
|
if (riid == IID_IUnknown)
|
|
*ppvObj = (IUnknown *)(IMailMsgLoggingPropertyBag *)this;
|
|
else if (riid == IID_IMailMsgLoggingPropertyBag)
|
|
*ppvObj = (IMailMsgLoggingPropertyBag *)this;
|
|
else if (riid == __uuidof(CMailMsgLoggingPropertyBag))
|
|
*ppvObj = (CMailMsgLoggingPropertyBag *)this;
|
|
else if (riid == IID_IMailMsgPropertyBag)
|
|
*ppvObj = (IMailMsgPropertyBag *)
|
|
((IMailMsgLoggingPropertyBag *)this);
|
|
else if (riid == IID_IMailMsgRegisterCleanupCallback)
|
|
*ppvObj = (IMailMsgRegisterCleanupCallback *)this;
|
|
else
|
|
return(E_NOINTERFACE);
|
|
AddRef();
|
|
return(S_OK);
|
|
}
|
|
|
|
static DWORD LoggingHelper(
|
|
LPVOID pvLogHandle,
|
|
const INETLOG_INFORMATION *pLogInformation
|
|
);
|
|
|
|
HRESULT STDMETHODCALLTYPE WriteToLog(
|
|
LPCSTR pszClientHostName,
|
|
LPCSTR pszClientUserName,
|
|
LPCSTR pszServerAddress,
|
|
LPCSTR pszOperation,
|
|
LPCSTR pszTarget,
|
|
LPCSTR pszParameters,
|
|
LPCSTR pszVersion,
|
|
DWORD dwBytesSent,
|
|
DWORD dwBytesReceived,
|
|
DWORD dwProcessingTimeMS,
|
|
DWORD dwWin32Status,
|
|
DWORD dwProtocolStatus,
|
|
DWORD dwPort,
|
|
LPCSTR pszHTTPHeader
|
|
)
|
|
{
|
|
INETLOG_INFORMATION info;
|
|
DWORD dwRes;
|
|
|
|
memset(&info,0,sizeof(info));
|
|
SetInetLogInfoField(pszClientHostName,&info.pszClientHostName,&info.cbClientHostName);
|
|
SetInetLogInfoField(pszClientUserName,&info.pszClientUserName,NULL);
|
|
SetInetLogInfoField(pszServerAddress,&info.pszServerAddress,NULL);
|
|
SetInetLogInfoField(pszOperation,&info.pszOperation,&info.cbOperation);
|
|
SetInetLogInfoField(pszTarget,&info.pszTarget,&info.cbTarget);
|
|
SetInetLogInfoField(pszParameters,&info.pszParameters,NULL);
|
|
SetInetLogInfoField(pszVersion,&info.pszVersion,NULL);
|
|
info.dwBytesSent = dwBytesSent;
|
|
info.dwBytesRecvd = dwBytesReceived;
|
|
info.msTimeForProcessing = dwProcessingTimeMS;
|
|
info.dwWin32Status = dwWin32Status;
|
|
info.dwProtocolStatus = dwProtocolStatus;
|
|
info.dwPort = dwPort;
|
|
SetInetLogInfoField(pszHTTPHeader,&info.pszHTTPHeader,&info.cbHTTPHeaderSize);
|
|
dwRes = LoggingHelper(m_pvLogHandle,&info);
|
|
return(S_OK);
|
|
}
|
|
|
|
|
|
private:
|
|
|
|
// Usage count
|
|
LONG m_lRefCount;
|
|
LPVOID m_pvLogHandle;
|
|
CShareLockNH m_rwLock;
|
|
|
|
};
|
|
|
|
|
|
|
|
// =================================================================
|
|
// Compare function
|
|
//
|
|
|
|
inline HRESULT CMailMsgPropertyBag::CompareProperty(
|
|
LPVOID pvPropKey,
|
|
LPPROPERTY_ITEM pItem
|
|
)
|
|
{
|
|
if (*(PROP_ID *)pvPropKey == ((LPGLOBAL_PROPERTY_ITEM)pItem)->idProp)
|
|
return(S_OK);
|
|
return(STG_E_UNKNOWN);
|
|
}
|
|
|
|
|
|
|
|
#endif
|