Leaked source code of windows server 2003
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.
 
 
 
 
 
 

445 lines
11 KiB

//+-------------------------------------------------------------------------
//
// Microsoft Windows
// Copyright (C) Microsoft Corporation, 1992 - 1993.
//
// File: utils.hxx
//
// Contents: Useful classes in implementing properties.
//
// Classes: CPropSetName - Buffer which converts fmtids->names
// CSubPropertyName - Buffer which converts VT_STREAM etc -> name
//
// Functions: DfpStatusToHResult - map NTSTATUS to HRESULT
// Probe - probe memory range to generate GP fault.
//
// History: 1-Mar-95 BillMo Created.
// 31-Mar-97 Danl Removed VerifyCommitFlags. Use definition
// in stg\h\docfilep.hxx
// 10-Mar-98 MikeHill Added IsOriginalPropVariantType.
// 06-May-98 MikeHill Use CoTaskMem rather than new/delete.
// 5/18/98 MikeHill
// - Moved IsOriginalPropVariantType to utils.cxx.
// 6/11/98 MikeHill
// - Add default constructor for CPropSetName.
// - Add CStringize (used for dbg outs).
//
// Notes:
//
// Codework:
//
//--------------------------------------------------------------------------
#include "prophdr.hxx"
#include "stdio.h" // _snprintf
//+-------------------------------------------------------------------------
//
// Misc functions and defines
//
//--------------------------------------------------------------------------
#ifndef _UTILS_HXX_
#define _UTILS_HXX_
#include "prophdr.hxx"
#include <olechar.h> // ocs* routines
#if DBG
#define STACKELEMS 3
#else
#define STACKELEMS 64
#endif
// These are defined in h\docfilep.hxx as STGM_DENY and STGM_RDWR
// But props doesn't include that file and we like these names better.
//
#define STGM_SHARE_MASK ( STGM_SHARE_DENY_NONE \
| STGM_SHARE_DENY_READ \
| STGM_SHARE_DENY_WRITE \
| STGM_SHARE_EXCLUSIVE )
#define STGM_RDWR_MASK (STGM_READ | STGM_WRITE | STGM_READWRITE)
// This function is functionally equivalent to the macro in Inside OLE,
// except that it it also returns the resulting ref count, which is
// useful in Asserts.
// This needs to be template to compile. Can't up cast to a reference.
//
template< class T_IUnknown >
RELEASE_INTERFACE( T_IUnknown *(&punk) )
{
ULONG ul = 0;
if( NULL != punk )
ul = punk->Release();
punk = NULL;
return( ul );
}
extern HRESULT NtStatusToScode(NTSTATUS);
inline HRESULT DfpNtStatusToHResult(NTSTATUS nts)
{
if (NT_SUCCESS(nts))
return S_OK;
if ((nts & 0xF0000000) == 0x80000000)
return nts;
else
return(NtStatusToScode(nts));
}
//+-------------------------------------------------------------------------
//
// Class: CPropSetName
//
// Purpose: Wrap buffer to convert
//
// Interface:
//
//
//
//
//
// Notes:
//
//--------------------------------------------------------------------------
class CPropSetName
{
public:
CPropSetName(REFFMTID rfmtid);
inline CPropSetName();
inline const OLECHAR * GetPropSetName()
{
return(_oszName);
}
private:
OLECHAR _oszName[CWCSTORAGENAME];
};
inline CPropSetName::CPropSetName()
{
_oszName[0] = OLESTR('\0');
}
class CStackBuffer
{
public:
CStackBuffer(BYTE *pbStackBuf,
ULONG ulElementSize,
ULONG cStackElements);
~CStackBuffer();
HRESULT Init(ULONG cElements);
protected:
BYTE * _pbHeapBuf;
ULONG _cElements;
private:
BYTE * _pbStackBuf;
ULONG _cbElement;
};
inline CStackBuffer::CStackBuffer( BYTE *pbStackBuf,
ULONG cbElement,
ULONG cElements)
: _pbStackBuf(pbStackBuf),
_pbHeapBuf(pbStackBuf),
_cbElement(cbElement),
_cElements(cElements)
{
}
inline CStackBuffer::~CStackBuffer()
{
if (_pbHeapBuf != _pbStackBuf)
{
CoTaskMemFree( _pbHeapBuf );
}
}
template < class t_ElementType, int t_cElements = STACKELEMS >
class TStackBuffer : public CStackBuffer
{
public:
inline TStackBuffer() : CStackBuffer( (BYTE*)_rg, sizeof(_rg[0]), sizeof(_rg)/sizeof(_rg[0]) )
{}
inline t_ElementType * GetBuf()
{
return((t_ElementType*)_pbHeapBuf);
}
inline t_ElementType& operator[]( ULONG i )
{
DfpAssert( i < _cElements );
return( GetBuf()[ i ] );
}
inline operator t_ElementType*()
{
return( GetBuf() );
}
inline ULONG Count() const
{
return( _cElements );
}
private:
t_ElementType _rg[ t_cElements ];
}; // template TStackBuffer
typedef TStackBuffer<PROPID> CStackPropIdArray;
typedef TStackBuffer<OLECHAR*> CStackOSZArray;
typedef TStackBuffer<PROPVARIANT> CStackPropVarArray;
HRESULT ValidateInRGPROPVARIANT( ULONG cpspec, const PROPVARIANT rgpropvar[] );
HRESULT ValidateOutRGPROPVARIANT( ULONG cpspec, PROPVARIANT rgpropvar[] );
HRESULT ValidateInRGLPOLESTR( ULONG cpropid, const OLECHAR* const rglpwstrName[] );
HRESULT ValidateOutRGLPOLESTR( ULONG cpropid, LPOLESTR rglpwstrName[] );
void * AllocAndCopy(ULONG cb, void * pvData, HRESULT *phr = NULL);
inline BOOL
GrfModeIsWriteable( DWORD grfMode )
{
return( (STGM_WRITE & grfMode) || (STGM_READWRITE & grfMode) );
}
inline BOOL
GrfModeIsReadable( DWORD grfMode )
{
return( !(STGM_WRITE & grfMode) );
}
BOOL IsOriginalPropVariantType( VARTYPE vt );
BOOL IsVariantType( VARTYPE vt );
inline BOOL
IsSupportedVarType( VARTYPE vt )
{
// Check for unsupported combinations of the high nibble
if( VT_RESERVED & vt )
return( FALSE );
if( (VT_ARRAY & vt) && (VT_VECTOR & vt) )
return( FALSE );
// Check the type
return( IsOriginalPropVariantType( vt ) // NT4 PROPVARIANT
||
IsVariantType( vt ) // Supported VARIANT
/*
|| // Allow arrays of (U)LONGLONGs
(VT_ARRAY|VT_I8) == vt || (VT_ARRAY|VT_UI8) == vt
*/
||
VT_VERSIONED_STREAM == vt // New to NT5
||
(VT_VECTOR|VT_I1) == vt ); // New to NT5
}
// Debug routine to get the ref-count of an interface.
// *** Not thread accurate ***
#if DBG
inline ULONG
GetRefCount( IUnknown *punk )
{
punk->AddRef();
return( punk->Release() );
}
#endif
//+----------------------------------------------------------------------------
//
// CStringize
//
// Convert from various data types into an Ansi string in _sz.
//
// E.g. CStringize(rfmtid).
//
// For types that are used in multiple ways (e.g. grfMode & grfFlags are
// both DWORDs), a structure is used to allow for overloading.
//
// E.g. CStringize(SGrfMode(grfMode))
//
//+----------------------------------------------------------------------------
#if DBG
struct SGrfMode
{
DWORD grfMode;
SGrfMode( DWORD grf )
{
grfMode = grf;
}
};
struct SGrfFlags
{
DWORD grfFlags;
SGrfFlags( DWORD grf )
{
grfFlags = grf;
}
};
class CStringize
{
private:
CHAR _sz[ MAX_PATH ];
public:
inline operator const char*() const;
public:
inline CStringize( const GUID &guid );
inline CStringize( const SGrfMode &sgrfMode );
inline CStringize( const SGrfFlags &sgrfFlags );
}; // CStringize
inline
CStringize::operator const char*() const
{
return( _sz );
}
inline
CStringize::CStringize( const GUID &guid )
{
if( NULL == &guid )
{
StringCbCopyA( _sz, sizeof(_sz), "<NULL>" );
}
else
{
StringCbPrintfA( _sz, sizeof(_sz),
"{%08X-%04X-%04X-%02X%02X%02X%02X%02X%02X%02X%02X}",
guid.Data1, guid.Data2, guid.Data3,
guid.Data4[0], guid.Data4[1], guid.Data4[2], guid.Data4[3],
guid.Data4[4], guid.Data4[5], guid.Data4[6], guid.Data4[7] );
}
}
inline
CStringize::CStringize( const SGrfMode &sgrfMode )
{
_sz[0] = '\0';
DWORD grfMode = sgrfMode.grfMode;
if (grfMode & STGM_TRANSACTED)
StringCbCatA(_sz, sizeof(_sz), "STGM_TRANSACTED | ");
else
StringCbCatA(_sz, sizeof(_sz), "STGM_DIRECT | ");
if (grfMode & STGM_SIMPLE)
StringCbCatA(_sz, sizeof(_sz), "STGM_SIMPLE | ");
switch (grfMode & 3)
{
case STGM_READ:
StringCbCatA(_sz, sizeof(_sz), "STGM_READ |");
break;
case STGM_WRITE:
StringCbCatA(_sz, sizeof(_sz), "STGM_WRITE |");
break;
case STGM_READWRITE:
StringCbCatA(_sz, sizeof(_sz), "STGM_READWRITE |");
break;
default:
StringCbCatA(_sz, sizeof(_sz), "BAD grfMode |");
break;
}
switch (grfMode & 0x70)
{
case STGM_SHARE_DENY_NONE:
StringCbCatA(_sz, sizeof(_sz), "STGM_SHARE_DENY_NONE |");
break;
case STGM_SHARE_DENY_READ:
StringCbCatA(_sz, sizeof(_sz), "STGM_SHARE_DENY_READ |");
break;
case STGM_SHARE_DENY_WRITE:
StringCbCatA(_sz, sizeof(_sz), "STGM_SHARE_DENY_WRITE |");
break;
case STGM_SHARE_EXCLUSIVE:
StringCbCatA(_sz, sizeof(_sz), "STGM_SHARE_EXCLUSIVE |");
break;
default:
StringCbCatA(_sz, sizeof(_sz), "BAD grfMode | ");
break;
}
if (grfMode & STGM_PRIORITY)
StringCbCatA(_sz, sizeof(_sz), "STGM_PRIORITY | ");
if (grfMode & STGM_DELETEONRELEASE)
StringCbCatA(_sz, sizeof(_sz), "STGM_DELETEONRELEASE | ");
if (grfMode & STGM_NOSCRATCH)
StringCbCatA(_sz, sizeof(_sz), "STGM_NOSCRATCH | ");
if (grfMode & STGM_CREATE)
StringCbCatA(_sz, sizeof(_sz), "STGM_CREATE | ");
if (grfMode & STGM_CONVERT)
StringCbCatA(_sz, sizeof(_sz), "STGM_CONVERT | ");
if (grfMode & STGM_FAILIFTHERE)
StringCbCatA(_sz, sizeof(_sz), "STGM_FAILIFTHERE | ");
}
inline
CStringize::CStringize( const SGrfFlags &sgrfFlags )
{
StringCbCopyA(_sz, sizeof(_sz), "PROPSETFLAG_DEFAULT |");
if (sgrfFlags.grfFlags & PROPSETFLAG_NONSIMPLE)
StringCbCatA(_sz, sizeof(_sz), "PROPSETFLAG_NONSIMPLE |");
if (sgrfFlags.grfFlags & PROPSETFLAG_ANSI)
StringCbCatA(_sz, sizeof(_sz), "PROPSETFLAG_ANSI |");
}
#endif // #if DBG
#endif // _UTILS_HXX_