mirror of
https://github.com/ToyB-Chan/minimal-heap.git
synced 2026-07-13 21:51:16 +02:00
fixed some bugs
This commit is contained in:
@@ -17,7 +17,7 @@
|
||||
#endif
|
||||
|
||||
#ifndef MIHP_MEMSET
|
||||
#define MIHP_MEMSET(Ptr, Size, Val) for(size_t i = 0; i < Size; ++i) *((char*)Ptr + i) = Val;
|
||||
#define MIHP_MEMSET(Ptr, Size, Val) do { for(size_t i = 0; i < Size; ++i) *((char*)Ptr + i) = Val; } while(0);
|
||||
#endif
|
||||
/* */
|
||||
|
||||
@@ -62,7 +62,7 @@ typedef struct _MIHP_HeapConfig
|
||||
|
||||
void* HeapLock; // Does not need to be reentred safe, can be NULL to disable locking
|
||||
MIHP_LockHeapFn* LockHeapFn; // Can be NULL if HeapLock is NULL
|
||||
MIHP_LockHeapFn* UnlockHeapFn; // Can be NULL if HeapLock is NULL
|
||||
MIHP_UnlockHeapFn* UnlockHeapFn; // Can be NULL if HeapLock is NULL
|
||||
} MIHP_HeapConfig;
|
||||
|
||||
typedef struct _MIHP_HeapStats
|
||||
@@ -73,7 +73,6 @@ typedef struct _MIHP_HeapStats
|
||||
size_t TotalSize;
|
||||
} MIHP_HeapStats;
|
||||
|
||||
// This structs size must be multiple of 8
|
||||
typedef struct _MIHP_HeapMemoryAreaHeader
|
||||
{
|
||||
size_t AreaSize;
|
||||
@@ -86,6 +85,7 @@ typedef struct _MIHP_HeapMemoryAreaHeader
|
||||
struct _MIHP_HeapSegmentHeader* FirstSegment;
|
||||
struct _MIHP_HeapSegmentHeader* LastSuccessfulAllocationSegment;
|
||||
|
||||
// TODO: Change this to an InstigatingHeapUuid (opaque ptr)
|
||||
const struct _MIHP_Heap* InstigatingHeap; // Heap that created this area. May not always be the owning heap (in case of a merge) or safe to deference!
|
||||
|
||||
uint32_t Checksum; // Checksum must be the last member !
|
||||
@@ -96,7 +96,6 @@ typedef struct _MIHP_AllocationCustomMetadata
|
||||
uint8_t data[16];
|
||||
} MIHP_AllocationCustomMetadata;
|
||||
|
||||
// This structs size must be multiple of 8
|
||||
typedef struct _MIHP_HeapSegmentHeader
|
||||
{
|
||||
size_t SegmentSize;
|
||||
@@ -109,7 +108,6 @@ typedef struct _MIHP_HeapSegmentHeader
|
||||
struct _MIHP_AllocationCustomMetadata CustomMetadata;
|
||||
|
||||
uint32_t Checksum; // Checksum must be the last member !
|
||||
|
||||
} MIHP_HeapSegmentHeader;
|
||||
|
||||
typedef struct _MIHP_HeapCorruptionInfo
|
||||
|
||||
@@ -1,5 +1,5 @@
|
||||
#define MIHP_MIN(A, B) (A) < (B) ? (A) : (B)
|
||||
#define MIHP_MAX(A, B) (A) > (B) ? (A) : (B)
|
||||
#define MIHP_MIN(A, B) ((A) < (B) ? (A) : (B))
|
||||
#define MIHP_MAX(A, B) ((A) > (B) ? (A) : (B))
|
||||
|
||||
#define MIHP_IS_PO2(Val) (Val != 0 && (Val & (Val - 1)) == 0)
|
||||
|
||||
@@ -66,6 +66,9 @@ bool MIHP_InitializeHeap(MIHP_Heap* heap, MIHP_HeapConfig config)
|
||||
|
||||
heap->Config = config;
|
||||
heap->FirstArea = MIHP_CreateHeapMemoryArea(heap, config.MinMemoryAreaSize);
|
||||
if (heap->FirstArea == NULL)
|
||||
return false;
|
||||
|
||||
heap->LastSuccessfulAllocationArea = heap->FirstArea;
|
||||
|
||||
heap->FirstArea->Checksum = MIHP_GenerateHeapMemoryAreaChecksum(heap, heap->FirstArea);
|
||||
@@ -315,6 +318,8 @@ bool MIHP_Free(MIHP_Heap* heap, void* ptr)
|
||||
|
||||
if (heap->Config.OnHeapCorruptionDetectedFn != NULL)
|
||||
heap->Config.OnHeapCorruptionDetectedFn(heap, corruptionInfo);
|
||||
|
||||
MIHP_UnlockHeap(heap);
|
||||
return false;
|
||||
}
|
||||
|
||||
@@ -494,6 +499,7 @@ size_t MIHP_GetPtrAllocationSize(MIHP_Heap* heap, void* ptr)
|
||||
MIHP_HeapSegmentHeader* segment = (MIHP_HeapSegmentHeader*)((char*)ptr - segmentHeaderSize);
|
||||
MIHP_ValidateHeapSegmentHeader(heap, segment);
|
||||
|
||||
MIHP_UnlockHeap(heap);
|
||||
return segment->OccupiedSize;
|
||||
}
|
||||
|
||||
|
||||
@@ -74,8 +74,6 @@ int main()
|
||||
memcpy(&returnPtr, &outMetadata.data, sizeof(void*));
|
||||
|
||||
MIHP_ASSERT(ptr == returnPtr);
|
||||
|
||||
__debugbreak();
|
||||
#endif
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user