windows-nt/Source/XPSP1/NT/inetsrv/iis/svcs/loadbal/cnfg/lbxbf.hxx

/*++

Copyright (c) 1998  Microsoft Corporation

Module Name:

    xbf.hxx

Abstract:

    Classes to handle extensible buffers

Author:

    Philippe Choquier (phillich)

--*/

#if !defined( _IISXBF_HXX )
#define _IISXBF_HXX

#if !defined(dllexp)
#define dllexp __declspec( dllexport )
#endif

#define INDEX_ERROR     0xffffffff

//
// Extensible buffer class
//
// Use GetBuff() & GetUsed() to retrieve buffer ptr & length
//

class CStoreXBF
{
public:
    dllexp CStoreXBF( UINT cG = sizeof(DWORD) ) { m_pBuff = NULL; m_cAllocBuff = 0; m_cUsedBuff = 0; m_cGrain = cG; }
    dllexp ~CStoreXBF() { Reset(); }
    dllexp VOID Reset()
        {
            if ( m_pBuff )
            {
                LocalFree( m_pBuff );
                m_pBuff = NULL;
                m_cAllocBuff = 0;
                m_cUsedBuff = 0;
            }
        }
    dllexp VOID Clear()
        {
            m_cUsedBuff = 0;
        }
    dllexp BOOL DecreaseUse( DWORD dwD )
        {
            if ( dwD <= m_cUsedBuff )
            {
                m_cUsedBuff -= dwD;
                return TRUE;
            }
            return FALSE;
        }
    dllexp BOOL Append( LPWSTR pszB )
        {
            return Append( (LPBYTE)pszB, wcslen(pszB)*sizeof(WCHAR) );
        }
    dllexp BOOL AppendZ( LPWSTR pszB )
        {
            return Append( (LPBYTE)pszB, (wcslen(pszB)+1)*sizeof(WCHAR) );
        }
    dllexp BOOL Append( DWORD dwB )
        {
            return Append( (LPBYTE)&dwB, (DWORD)sizeof(DWORD) );
        }
    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;
        }
    dllexp BOOL Need( DWORD dwNeed );

    //
    // Get ptr to buffer
    //

    dllexp LPBYTE GetBuff() { return m_pBuff; }

    //
    // Get count of bytes in buffer
    //

    dllexp DWORD GetUsed() { return m_cUsedBuff; }
    dllexp BOOL SetUsed( DWORD dw ) 
        { if ( dw <= m_cAllocBuff ) { m_cUsedBuff = dw; return TRUE; } else return FALSE; }
    dllexp BOOL Save( HANDLE hFile );
    dllexp BOOL Load( HANDLE hFile );

private:
    LPBYTE  m_pBuff;
    DWORD   m_cAllocBuff;
    DWORD   m_cGrain;
    DWORD   m_cUsedBuff;
} ;

typedef CStoreXBF XBF;

//
// extensible array of LPVOID
//

class CPtrXBF : public CStoreXBF
{
public:
    dllexp CPtrXBF() : CStoreXBF( sizeof(LPVOID) ) {}
    dllexp DWORD GetNbPtr() { return GetUsed()/sizeof(LPVOID); }
    dllexp DWORD AddPtr( LPVOID pV);
    dllexp DWORD InsertPtr( DWORD iBefore, LPVOID pV );
    dllexp LPVOID GetPtr( DWORD i ) { return ((LPVOID*)GetBuff())[i]; }
    dllexp LPVOID GetPtrAddr( DWORD i ) { return ((LPVOID*)GetBuff())+i; }
    dllexp BOOL SetPtr( DWORD i, LPVOID pV ) { ((LPVOID*)GetBuff())[i] = pV; return TRUE; }
    dllexp BOOL DeletePtr( DWORD i );
    dllexp BOOL Unserialize( LPBYTE* ppB, LPDWORD pC, DWORD cNbEntry );
    dllexp BOOL Serialize( CStoreXBF* pX );
} ;


//
// string storage class
//

class CAllocString {
public:
    dllexp CAllocString() { m_pStr = NULL; }
    dllexp ~CAllocString() { Reset(); }
    dllexp VOID Reset() { if ( m_pStr != NULL ) {LocalFree( m_pStr ); m_pStr = NULL;} }
    dllexp BOOL Set( LPWSTR pS );
    dllexp BOOL Append( LPWSTR pS );
    dllexp BOOL Unserialize( LPBYTE* ppb, LPDWORD pc );
    dllexp BOOL Serialize( CStoreXBF* pX );
    dllexp LPWSTR Get() { return m_pStr; }
private:
    LPWSTR   m_pStr;
} ;


//
// binary object, contains ptr & size
//

class CBlob {
public:
    dllexp CBlob() { m_pStr = NULL; m_cStr = 0; }
    dllexp ~CBlob() { Reset(); }
    dllexp VOID Reset() { if ( m_pStr != NULL ) {LocalFree( m_pStr ); m_pStr = NULL; m_cStr = 0;} }
    dllexp BOOL Set( LPBYTE pStr, DWORD cStr );
    dllexp BOOL InitSet( LPBYTE pStr, DWORD cStr );
    dllexp LPBYTE Get( LPDWORD pc ) { *pc = m_cStr; return m_pStr; }
    dllexp BOOL Unserialize( LPBYTE* ppB, LPDWORD pC );
    dllexp BOOL Serialize( CStoreXBF* pX );

private:
    LPBYTE  m_pStr;
    DWORD   m_cStr;
} ;


//
// extensible array of strings
//

class CStrPtrXBF : public CPtrXBF {
public:
    dllexp ~CStrPtrXBF();
    dllexp DWORD AddEntry( LPWSTR pS );
    dllexp DWORD InsertEntry( DWORD iBefore, LPWSTR pS );
    dllexp DWORD GetNbEntry() { return GetNbPtr(); }
    dllexp LPWSTR GetEntry( DWORD i ) { return ((CAllocString*)GetPtrAddr(i))->Get(); }
    dllexp BOOL SetEntry( DWORD i, LPWSTR pS ) { return ((CAllocString*)GetPtrAddr(i))->Set( pS ); }
    dllexp BOOL DeleteEntry( DWORD i );
    dllexp BOOL Unserialize( LPBYTE* ppB, LPDWORD pC, DWORD cNbEntry );
    dllexp BOOL Serialize( CStoreXBF* pX );
} ;


//
// extensible array of binary object
// ptr & size are stored for each entry
//

class CBlobXBF : public CStoreXBF {
public:
    dllexp CBlobXBF() : CStoreXBF( sizeof(CBlob) ) {}
    dllexp ~CBlobXBF();
    dllexp VOID Reset();
    dllexp DWORD AddEntry( LPBYTE pS, DWORD cS );
    dllexp DWORD InsertEntry( DWORD iBefore, LPBYTE pS, DWORD cS );
    dllexp DWORD GetNbEntry() { return GetUsed()/sizeof(CBlob); }
    dllexp CBlob* GetBlob( DWORD i )
        { return (CBlob*)(GetBuff()+i*sizeof(CBlob)); }
    dllexp BOOL GetEntry( DWORD i, LPBYTE* ppB, LPDWORD pcB ) 
        { 
            if ( i < (GetUsed()/sizeof(CBlob)) )
            {
                *ppB = GetBlob(i)->Get( pcB );
                return TRUE;
            }
            return FALSE;
        }
    dllexp BOOL SetEntry( DWORD i, LPBYTE pS, DWORD cS ) 
        { return GetBlob(i)->Set( pS, cS ); }
    dllexp BOOL DeleteEntry( DWORD i );
    dllexp BOOL Unserialize( LPBYTE* ppB, LPDWORD pC, DWORD cNbEntry );
    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