142 lines
3.1 KiB
C
142 lines
3.1 KiB
C
#define ALLOCATESTRUCT 0x0
|
|
#define REALLOCATESTRUCT 0x1
|
|
#define FREESTRUCT 0x2
|
|
#define CREATESTRUCT 0x3
|
|
#define DESTROYSTRUCT 0x4
|
|
#define COALESCESTRUCT 0x5
|
|
|
|
#define SPACEBETWEEN 0x18
|
|
|
|
#define CHUNK(x) (heap->chunks[x])
|
|
|
|
#define NEXTADDR(x) (PVOID)(*((ULONG *)(&(CHUNK(i).addr))) + CHUNK(i).size + SPACEBETWEEN)
|
|
|
|
#define NULLNODE 0xffffffff
|
|
|
|
#define MODEL 1
|
|
#define LOG 2
|
|
#define RUNNING 4
|
|
|
|
// Space between chunks on vista
|
|
|
|
// Dont use a direct access list here because looping should
|
|
// be faster for our number of heaps (I think!)
|
|
struct HeapState {
|
|
BYTE state;
|
|
ULONG numHeaps;
|
|
ULONG hPoolListLen;
|
|
HANDLE hLogFile;
|
|
struct HPool *heaps;
|
|
|
|
};
|
|
|
|
struct HPool {
|
|
PVOID base;
|
|
ULONG numChunks;
|
|
ULONG chunkListLen;
|
|
ULONG flags;
|
|
ULONG reserve;
|
|
ULONG commit;
|
|
BOOLEAN lock;
|
|
ULONG *map;
|
|
struct HeapChunk *chunks;
|
|
ULONG inUseHead;
|
|
ULONG lastInUse;
|
|
BOOLEAN inUse;
|
|
};
|
|
|
|
struct LookAsideList {
|
|
DWORD placeHolder;
|
|
};
|
|
|
|
struct HeapCache {
|
|
ULONG NumBuckets;
|
|
unsigned __int8 *pBitmap;
|
|
ULONG **pBuckets;
|
|
};
|
|
|
|
struct HeapChunk {
|
|
PVOID addr;
|
|
PVOID heapHandle;
|
|
ULONG size;
|
|
ULONG flags;
|
|
ULONG nextBucket;
|
|
ULONG nextInUse;
|
|
ULONG nextFreeListChunk;
|
|
BOOLEAN free;
|
|
BOOLEAN inUse;
|
|
};
|
|
|
|
struct AllocateStruct {
|
|
BYTE type;
|
|
PVOID heapHandle;
|
|
ULONG flags;
|
|
ULONG size;
|
|
PVOID ret;
|
|
PVOID caller;
|
|
};
|
|
|
|
struct ReallocateStruct {
|
|
BYTE type;
|
|
PVOID heapHandle;
|
|
ULONG flags;
|
|
PVOID memoryPointer;
|
|
ULONG size;
|
|
PVOID ret;
|
|
PVOID caller;
|
|
};
|
|
|
|
struct FreeStruct {
|
|
BYTE type;
|
|
PVOID heapHandle;
|
|
ULONG flags;
|
|
PVOID memoryPointer;
|
|
PVOID ret;
|
|
PVOID caller;
|
|
};
|
|
|
|
struct CreateStruct {
|
|
BYTE type;
|
|
ULONG flags;
|
|
PVOID base;
|
|
ULONG reserve;
|
|
ULONG commit;
|
|
BOOLEAN lock;
|
|
PVOID RtlHeapParams; // Wont get this info back now - maybe later
|
|
PVOID ret; // if we think we really need it?
|
|
|
|
};
|
|
|
|
struct DestroyStruct {
|
|
BYTE type;
|
|
PVOID heapHandle;
|
|
NTSTATUS ret;
|
|
};
|
|
|
|
struct CoalesceStruct {
|
|
BYTE type;
|
|
PVOID heapHandle;
|
|
ULONG arg2;
|
|
ULONG arg3;
|
|
ULONG arg4;
|
|
PVOID ret;
|
|
};
|
|
|
|
void initializeHeapModel(struct HeapState *);
|
|
void heapAllocate(struct HeapState *heapModel, struct AllocateStruct *aStruct);
|
|
void logAllocate(struct HeapState *heapModel, struct AllocateStruct *aStruct);
|
|
|
|
void heapReallocate(struct HeapState *heapModel, struct ReallocateStruct *aStruct);
|
|
void logReallocate(struct HeapState *heapModel, struct ReallocateStruct *aStruct);
|
|
|
|
void heapFree(struct HeapState *heapModel, struct FreeStruct *fStruct);
|
|
void logFree(struct HeapState *heapModel, struct FreeStruct *fStruct);
|
|
|
|
void heapCreate(struct HeapState *heapModel, struct CreateStruct *cStruct);
|
|
void heapDestroy(struct HeapState *heapModel, struct DestroyStruct *dStruct);
|
|
void heapCoalesce(struct HeapState *heapModel, struct CoalesceStruct *cfbStruct);
|
|
struct HPool *getHeap(struct HeapState *heapModel, PVOID heapHandle);
|
|
struct HeapChunk *getChunk(struct HPool *heap, PVOID memoryPointer, ULONG inAfter);
|
|
int FindOffsetForChunk(struct HPool *heap, PVOID memoryPointer); //quickly match a (heap, chunkAddress) into an offset in heap.chunks
|
|
|