273 lines
7.5 KiB
C++
273 lines
7.5 KiB
C++
/*++
|
|
|
|
Copyright (c) 1996 Microsoft Corporation
|
|
|
|
Module Name:
|
|
|
|
xbf.hxx
|
|
|
|
Abstract:
|
|
|
|
Helper classes used to handle IIS client cert wildcard mapping
|
|
see iis\svcs\iismap\iiscmr.cxx for implementation
|
|
|
|
Author:
|
|
|
|
Philippe Choquier (phillich) 17-oct-1996
|
|
|
|
--*/
|
|
|
|
#if !defined( _IISXBF_HXX )
|
|
#define _IISXBF_HXX
|
|
|
|
#ifndef IISMAP_DLLEXP
|
|
# ifdef DLL_IMPLEMENTATION
|
|
# define IISMAP_DLLEXP __declspec(IISMAP_DLLEXPort)
|
|
# ifdef IMPLEMENTATION_EXPORT
|
|
# define IRTL_EXPIMP
|
|
# else
|
|
# undef IRTL_EXPIMP
|
|
# endif
|
|
# else // !DLL_IMPLEMENTATION
|
|
# define IISMAP_DLLEXP __declspec(dllimport)
|
|
# define IRTL_EXPIMP extern
|
|
# endif // !DLL_IMPLEMENTATION
|
|
#endif // !IISMAP_DLLEXP
|
|
|
|
//
|
|
// Extensible buffer class
|
|
//
|
|
// Use GetBuff() & GetUsed() to retrieve buffer ptr & length
|
|
//
|
|
|
|
class CStoreXBF
|
|
{
|
|
public:
|
|
IISMAP_DLLEXP CStoreXBF( UINT cG = sizeof(DWORD) ) { m_pBuff = NULL; m_cAllocBuff = 0; m_cUsedBuff = 0; m_cGrain = cG; }
|
|
IISMAP_DLLEXP ~CStoreXBF() { Reset(); }
|
|
IISMAP_DLLEXP VOID Reset()
|
|
{
|
|
if ( m_pBuff )
|
|
{
|
|
LocalFree( m_pBuff );
|
|
m_pBuff = NULL;
|
|
m_cAllocBuff = 0;
|
|
m_cUsedBuff = 0;
|
|
}
|
|
}
|
|
IISMAP_DLLEXP VOID Clear()
|
|
{
|
|
m_cUsedBuff = 0;
|
|
}
|
|
IISMAP_DLLEXP BOOL DecreaseUse( DWORD dwD )
|
|
{
|
|
if ( dwD <= m_cUsedBuff )
|
|
{
|
|
m_cUsedBuff -= dwD;
|
|
return TRUE;
|
|
}
|
|
return FALSE;
|
|
}
|
|
IISMAP_DLLEXP BOOL Append( LPSTR pszB )
|
|
{
|
|
return Append( (LPBYTE)pszB, (DWORD)strlen(pszB) );
|
|
}
|
|
IISMAP_DLLEXP BOOL AppendZ( LPSTR pszB )
|
|
{
|
|
return Append( (LPBYTE)pszB, (DWORD)strlen(pszB)+1 );
|
|
}
|
|
IISMAP_DLLEXP BOOL Append( DWORD dwB )
|
|
{
|
|
return Append( (LPBYTE)&dwB, (DWORD)sizeof(DWORD) );
|
|
}
|
|
IISMAP_DLLEXP BOOL Append( LPBYTE pB, DWORD cB )
|
|
{
|
|
DWORD dwNeed = m_cUsedBuff + cB;
|
|
if ( Need( dwNeed ) )
|
|
{
|
|
memcpy( m_pBuff + m_cUsedBuff, pB, cB );
|
|
m_cUsedBuff += cB;
|
|
return TRUE;
|
|
}
|
|
return FALSE;
|
|
}
|
|
IISMAP_DLLEXP BOOL Need( DWORD dwNeed );
|
|
|
|
//
|
|
// Get ptr to buffer
|
|
//
|
|
|
|
IISMAP_DLLEXP LPBYTE GetBuff() { return m_pBuff; }
|
|
|
|
//
|
|
// Get count of bytes in buffer
|
|
//
|
|
|
|
IISMAP_DLLEXP DWORD GetUsed() { return m_cUsedBuff; }
|
|
IISMAP_DLLEXP BOOL Save( HANDLE hFile );
|
|
IISMAP_DLLEXP BOOL Load( HANDLE hFile );
|
|
|
|
private:
|
|
LPBYTE m_pBuff;
|
|
DWORD m_cAllocBuff;
|
|
DWORD m_cGrain;
|
|
DWORD m_cUsedBuff;
|
|
} ;
|
|
|
|
|
|
//
|
|
// extensible array of DWORDS
|
|
// Added to make win64 stuff work
|
|
//
|
|
class CPtrDwordXBF : public CStoreXBF
|
|
{
|
|
public:
|
|
IISMAP_DLLEXP CPtrDwordXBF() : CStoreXBF( sizeof(DWORD) ) {}
|
|
IISMAP_DLLEXP DWORD GetNbPtr() { return GetUsed()/sizeof(DWORD); }
|
|
IISMAP_DLLEXP DWORD AddPtr(DWORD dwEntry);
|
|
IISMAP_DLLEXP DWORD InsertPtr(DWORD iBefore, DWORD dwEntry);
|
|
IISMAP_DLLEXP LPDWORD GetPtr( DWORD i ) { return ( (LPDWORD*) GetBuff())[i]; }
|
|
IISMAP_DLLEXP BOOL SetPtr( DWORD i, DWORD pV ) { ((LPDWORD)GetBuff())[i] = pV; return TRUE; }
|
|
IISMAP_DLLEXP BOOL DeletePtr( DWORD i );
|
|
IISMAP_DLLEXP BOOL Unserialize( LPBYTE* ppB, LPDWORD pC, DWORD cNbEntry );
|
|
IISMAP_DLLEXP BOOL Serialize( CStoreXBF* pX );
|
|
} ;
|
|
|
|
|
|
//
|
|
// extensible array of LPVOID
|
|
//
|
|
|
|
class CPtrXBF : public CStoreXBF
|
|
{
|
|
public:
|
|
IISMAP_DLLEXP CPtrXBF() : CStoreXBF( sizeof(LPVOID) ) {}
|
|
IISMAP_DLLEXP DWORD GetNbPtr() { return GetUsed()/sizeof(LPVOID); }
|
|
IISMAP_DLLEXP DWORD AddPtr( LPVOID pV);
|
|
IISMAP_DLLEXP DWORD InsertPtr( DWORD iBefore, LPVOID pV );
|
|
IISMAP_DLLEXP LPVOID GetPtr( DWORD i ) { return ((LPVOID*)GetBuff())[i]; }
|
|
IISMAP_DLLEXP LPVOID GetPtrAddr( DWORD i ) { return ((LPVOID*)GetBuff())+i; }
|
|
IISMAP_DLLEXP BOOL SetPtr( DWORD i, LPVOID pV ) { ((LPVOID*)GetBuff())[i] = pV; return TRUE; }
|
|
IISMAP_DLLEXP BOOL DeletePtr( DWORD i );
|
|
IISMAP_DLLEXP BOOL Unserialize( LPBYTE* ppB, LPDWORD pC, DWORD cNbEntry );
|
|
IISMAP_DLLEXP BOOL Serialize( CStoreXBF* pX );
|
|
} ;
|
|
|
|
|
|
//
|
|
// string storage class
|
|
//
|
|
|
|
class CAllocString {
|
|
public:
|
|
IISMAP_DLLEXP CAllocString() { m_pStr = NULL; }
|
|
IISMAP_DLLEXP ~CAllocString() { Reset(); }
|
|
IISMAP_DLLEXP VOID Reset() { if ( m_pStr != NULL ) {LocalFree( m_pStr ); m_pStr = NULL;} }
|
|
IISMAP_DLLEXP BOOL Set( LPSTR pS );
|
|
IISMAP_DLLEXP BOOL Append( LPSTR pS );
|
|
IISMAP_DLLEXP BOOL Unserialize( LPBYTE* ppb, LPDWORD pc );
|
|
IISMAP_DLLEXP BOOL Serialize( CStoreXBF* pX );
|
|
IISMAP_DLLEXP LPSTR Get() { return m_pStr; }
|
|
private:
|
|
LPSTR m_pStr;
|
|
} ;
|
|
|
|
|
|
//
|
|
// binary object, contains ptr & size
|
|
//
|
|
|
|
class CBlob {
|
|
public:
|
|
IISMAP_DLLEXP CBlob() { m_pStr = NULL; m_cStr = 0; }
|
|
IISMAP_DLLEXP ~CBlob() { Reset(); }
|
|
IISMAP_DLLEXP VOID Reset() { if ( m_pStr != NULL ) {LocalFree( m_pStr ); m_pStr = NULL; m_cStr = 0;} }
|
|
IISMAP_DLLEXP BOOL Set( LPBYTE pStr, DWORD cStr );
|
|
IISMAP_DLLEXP BOOL InitSet( LPBYTE pStr, DWORD cStr );
|
|
IISMAP_DLLEXP LPBYTE Get( LPDWORD pc ) { *pc = m_cStr; return m_pStr; }
|
|
IISMAP_DLLEXP BOOL Unserialize( LPBYTE* ppB, LPDWORD pC );
|
|
IISMAP_DLLEXP BOOL Serialize( CStoreXBF* pX );
|
|
|
|
private:
|
|
LPBYTE m_pStr;
|
|
DWORD m_cStr;
|
|
} ;
|
|
|
|
|
|
//
|
|
// extensible array of strings
|
|
//
|
|
|
|
class CStrPtrXBF : public CPtrXBF {
|
|
public:
|
|
IISMAP_DLLEXP ~CStrPtrXBF();
|
|
IISMAP_DLLEXP DWORD AddEntry( LPSTR pS );
|
|
IISMAP_DLLEXP DWORD InsertEntry( DWORD iBefore, LPSTR pS );
|
|
IISMAP_DLLEXP DWORD GetNbEntry() { return GetNbPtr(); }
|
|
IISMAP_DLLEXP LPSTR GetEntry( DWORD i ) { return ((CAllocString*)GetPtrAddr(i))->Get(); }
|
|
IISMAP_DLLEXP BOOL SetEntry( DWORD i, LPSTR pS ) { return ((CAllocString*)GetPtrAddr(i))->Set( pS ); }
|
|
IISMAP_DLLEXP BOOL DeleteEntry( DWORD i );
|
|
IISMAP_DLLEXP BOOL Unserialize( LPBYTE* ppB, LPDWORD pC, DWORD cNbEntry );
|
|
IISMAP_DLLEXP BOOL Serialize( CStoreXBF* pX );
|
|
} ;
|
|
|
|
|
|
//
|
|
// extensible array of binary object
|
|
// ptr & size are stored for each entry
|
|
//
|
|
|
|
class CBlobXBF : public CStoreXBF {
|
|
public:
|
|
IISMAP_DLLEXP CBlobXBF() : CStoreXBF( sizeof(CBlob) ) {}
|
|
IISMAP_DLLEXP ~CBlobXBF();
|
|
IISMAP_DLLEXP VOID Reset();
|
|
IISMAP_DLLEXP DWORD AddEntry( LPBYTE pS, DWORD cS );
|
|
IISMAP_DLLEXP DWORD InsertEntry( DWORD iBefore, LPSTR pS, DWORD cS );
|
|
IISMAP_DLLEXP DWORD GetNbEntry() { return GetUsed()/sizeof(CBlob); }
|
|
IISMAP_DLLEXP CBlob* GetBlob( DWORD i )
|
|
{ return (CBlob*)(GetBuff()+i*sizeof(CBlob)); }
|
|
IISMAP_DLLEXP BOOL GetEntry( DWORD i, LPBYTE* ppB, LPDWORD pcB )
|
|
{
|
|
if ( i < (GetUsed()/sizeof(CBlob)) )
|
|
{
|
|
*ppB = GetBlob(i)->Get( pcB );
|
|
return TRUE;
|
|
}
|
|
return FALSE;
|
|
}
|
|
IISMAP_DLLEXP BOOL SetEntry( DWORD i, LPBYTE pS, DWORD cS )
|
|
{ return GetBlob(i)->Set( pS, cS ); }
|
|
IISMAP_DLLEXP BOOL DeleteEntry( DWORD i );
|
|
IISMAP_DLLEXP BOOL Unserialize( LPBYTE* ppB, LPDWORD pC, DWORD cNbEntry );
|
|
IISMAP_DLLEXP BOOL Serialize( CStoreXBF* pX );
|
|
} ;
|
|
|
|
BOOL
|
|
Unserialize(
|
|
LPBYTE* ppB,
|
|
LPDWORD pC,
|
|
LPDWORD pU
|
|
);
|
|
|
|
BOOL
|
|
Unserialize(
|
|
LPBYTE* ppB,
|
|
LPDWORD pC,
|
|
LPBOOL pU
|
|
);
|
|
|
|
BOOL
|
|
Serialize(
|
|
CStoreXBF* pX,
|
|
DWORD dw
|
|
);
|
|
|
|
BOOL
|
|
Serialize(
|
|
CStoreXBF* pX,
|
|
BOOL f
|
|
);
|
|
|
|
#endif
|