137 lines
3.3 KiB
C++
137 lines
3.3 KiB
C++
|
/*++
|
|||
|
|
|
|||
|
|
Copyright (c) 1998 Microsoft Corporation
|
|||
|
|
|
|||
|
|
Module Name:
|
|||
|
|
|
|||
|
|
mmfile.hxx
|
|||
|
|
|
|||
|
|
Abstract:
|
|||
|
|
|
|||
|
|
This file contains definitions for mmfile.cxx
|
|||
|
|
Generic shared memory allocator.
|
|||
|
|
|
|||
|
|
Author:
|
|||
|
|
|
|||
|
|
Adriaan Canter (adriaanc) 01-Aug-1998
|
|||
|
|
|
|||
|
|
--*/
|
|||
|
|
#ifndef MMFILE_HXX
|
|||
|
|
#define MMFILE_HXX
|
|||
|
|
|
|||
|
|
#define MAKE_MUTEX_NAME 0
|
|||
|
|
#define MAKE_MAP_NAME 1
|
|||
|
|
|
|||
|
|
// BUGBUG dynamic map name.
|
|||
|
|
#define SZ_MAPNAME "SSPIDIGESTMAP:"
|
|||
|
|
#define SIG_CMMF 'FMMC'
|
|||
|
|
|
|||
|
|
#define MMF_SIG_SZ "Digest Cred Cache Ver 0.9"
|
|||
|
|
#define MMF_SIG_SIZE sizeof(MMF_SIG_SZ)
|
|||
|
|
|
|||
|
|
#define PAGE_SIZE 4096
|
|||
|
|
|
|||
|
|
// BUGBUG reverse sigs or make up new ones.
|
|||
|
|
#define SIG_FREE 0xbadf00d
|
|||
|
|
#define SIG_ALLOC 0xdeadbeef
|
|||
|
|
|
|||
|
|
#define NUM_BITS_IN_BYTE 8
|
|||
|
|
#define NUM_BITS_IN_DWORD (sizeof(DWORD) * NUM_BITS_IN_BYTE)
|
|||
|
|
|
|||
|
|
#define MAX_ENTRY_SIZE PAGE_SIZE
|
|||
|
|
#define MAX_HEADER_ARRAY_SIZE 8
|
|||
|
|
|
|||
|
|
// Useful macros.
|
|||
|
|
// BUGBUG - precedence, parens.
|
|||
|
|
|
|||
|
|
#define ROUNDUPTOPOWEROF2(bytesize, powerof2)(((bytesize) + (powerof2) - 1) & ~((powerof2) - 1))
|
|||
|
|
#define ROUNDUPBLOCKS(bytesize) ((bytesize + _cbEntry-1) & ~(_cbEntry-1))
|
|||
|
|
#define ROUNDUPDWORD(bytesize) ((bytesize + sizeof(DWORD)-1) & ~(sizeof(DWORD)-1))
|
|||
|
|
#define ROUNDUPPAGE(bytesize) ((bytesize + PAGE_SIZE-1) & ~(PAGE_SIZE-1))
|
|||
|
|
|
|||
|
|
#define NUMBLOCKS(bytesize, blocksize) (bytesize / blocksize)
|
|||
|
|
|
|||
|
|
|
|||
|
|
//--------------------------------------------------------------------
|
|||
|
|
// struct MEMMAP_HEADER
|
|||
|
|
//--------------------------------------------------------------------
|
|||
|
|
typedef struct MEMMAP_HEADER
|
|||
|
|
{
|
|||
|
|
TCHAR szSig[MMF_SIG_SIZE];
|
|||
|
|
DWORD nEntries;
|
|||
|
|
DWORD dwHeaderData[MAX_HEADER_ARRAY_SIZE * sizeof(DWORD)];
|
|||
|
|
} *LPMEMMAP_HEADER;
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
//--------------------------------------------------------------------
|
|||
|
|
// struct MAP_ENTRY
|
|||
|
|
//--------------------------------------------------------------------
|
|||
|
|
typedef struct MAP_ENTRY
|
|||
|
|
{
|
|||
|
|
DWORD dwSig;
|
|||
|
|
DWORD nBlocks;
|
|||
|
|
} *LPMAP_ENTRY;
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
//--------------------------------------------------------------------
|
|||
|
|
// class CMMFile
|
|||
|
|
//--------------------------------------------------------------------
|
|||
|
|
class CMMFile
|
|||
|
|
{
|
|||
|
|
protected:
|
|||
|
|
|
|||
|
|
DWORD _dwSig;
|
|||
|
|
DWORD _dwStatus;
|
|||
|
|
|
|||
|
|
LPMEMMAP_HEADER _pHeader;
|
|||
|
|
LPDWORD _pBitMap;
|
|||
|
|
LPBYTE _pHeap;
|
|||
|
|
|
|||
|
|
DWORD _cbHeap;
|
|||
|
|
DWORD _cbEntry;
|
|||
|
|
DWORD _cbBitMap;
|
|||
|
|
DWORD _cbTotal;
|
|||
|
|
|
|||
|
|
DWORD _nBitMapDwords;
|
|||
|
|
DWORD _nMaxEntries;
|
|||
|
|
|
|||
|
|
HANDLE _hFile;
|
|||
|
|
|
|||
|
|
BOOL CheckNextNBits(DWORD& nArrayIndex, DWORD& dwStartMask,
|
|||
|
|
DWORD nBitsRequired, DWORD& nBitsFound);
|
|||
|
|
|
|||
|
|
BOOL SetNextNBits(DWORD nIdx, DWORD dwMask,
|
|||
|
|
DWORD nBitsRequired);
|
|||
|
|
|
|||
|
|
DWORD GetAndSetNextFreeEntry(DWORD nBitsRequired);
|
|||
|
|
|
|||
|
|
|
|||
|
|
VOID ResetEntryData(LPMAP_ENTRY Entry,
|
|||
|
|
DWORD dwResetValue, DWORD nBlocks);
|
|||
|
|
|
|||
|
|
|
|||
|
|
public:
|
|||
|
|
|
|||
|
|
CMMFile(DWORD cbHeap, DWORD cbEntry);
|
|||
|
|
~CMMFile();
|
|||
|
|
|
|||
|
|
DWORD Init();
|
|||
|
|
DWORD DeInit();
|
|||
|
|
|
|||
|
|
LPDWORD GetHeaderData(DWORD dwHeaderIndex);
|
|||
|
|
VOID SetHeaderData(DWORD dwHeaderIndex, DWORD dwHeaderValue);
|
|||
|
|
LPMAP_ENTRY AllocateEntry(DWORD cbBytes);
|
|||
|
|
BOOL ReAllocateEntry(LPMAP_ENTRY pEntry, DWORD cbBytes);
|
|||
|
|
BOOL FreeEntry(LPMAP_ENTRY Entry);
|
|||
|
|
DWORD_PTR GetMapPtr();
|
|||
|
|
DWORD GetStatus();
|
|||
|
|
static DWORD MakeUserObjectName(LPSTR szName, LPDWORD pcbName, DWORD dwFlags);
|
|||
|
|
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
#endif
|
|||
|
|
|