memheap.h

Go to the documentation of this file.

#pragma once
#include <wut.h>
#include "spinlock.h"
#include "memlist.h"

#ifdef __cplusplus
extern "C" {
#endif

typedef struct MEMHeapHeader MEMHeapHeader;

typedef enum MEMHeapFillType
{
   MEM_HEAP_FILL_TYPE_UNUSED    = 0,
   MEM_HEAP_FILL_TYPE_ALLOCATED = 1,
   MEM_HEAP_FILL_TYPE_FREED     = 2,
} MEMHeapFillType;

typedef enum MEMHeapTag
{
  MEM_BLOCK_HEAP_TAG      = 0x424C4B48u,
  MEM_EXPANDED_HEAP_TAG   = 0x45585048u,
  MEM_FRAME_HEAP_TAG      = 0x46524D48u,
  MEM_UNIT_HEAP_TAG       = 0x554E5448u,
  MEM_USER_HEAP_TAG       = 0x55535248u,
} MEMHeapTag;

typedef enum MEMHeapFlags
{
  MEM_HEAP_FLAG_ZERO_ALLOCATED  = 1 << 0,
  MEM_HEAP_FLAG_DEBUG_MODE      = 1 << 1,
  MEM_HEAP_FLAG_USE_LOCK        = 1 << 2,
} MEMHeapFlags;

struct MEMHeapHeader
{
   MEMHeapTag tag;

   MEMMemoryLink link;

   MEMMemoryList list;

   void *dataStart;

   void *dataEnd;

   OSSpinLock lock;

   uint32_t flags;

   UNKNOWN(0x0C);
};
CHECK_OFFSET(MEMHeapHeader, 0x00, tag);
CHECK_OFFSET(MEMHeapHeader, 0x04, link);
CHECK_OFFSET(MEMHeapHeader, 0x0C, list);
CHECK_OFFSET(MEMHeapHeader, 0x18, dataStart);
CHECK_OFFSET(MEMHeapHeader, 0x1C, dataEnd);
CHECK_OFFSET(MEMHeapHeader, 0x20, lock);
CHECK_OFFSET(MEMHeapHeader, 0x30, flags);
CHECK_SIZE(MEMHeapHeader, 0x40);

void
MEMDumpHeap(MEMHeapHeader *heap);

MEMHeapHeader *
MEMFindContainHeap(void *block);

uint32_t
MEMGetFillValForHeap(MEMHeapFillType type);

void
MEMSetFillValForHeap(MEMHeapFillType type,
                     uint32_t value);

#ifdef __cplusplus
}
#endif