|
1 | 1 | #include <assert.h> |
2 | | -#include <phnt.h> |
| 2 | +// #include <phnt.h> |
3 | 3 | #include <malloc.h> |
| 4 | +#include <windows.h> |
| 5 | + |
| 6 | +NTSYSAPI |
| 7 | +void* NTAPI RtlGetCurrentPeb(VOID); |
| 8 | + |
| 9 | +NTSYSAPI |
| 10 | +PVOID |
| 11 | +NTAPI |
| 12 | +RtlAllocateHeap(_In_ HANDLE HeapHandle, _In_opt_ ULONG Flags, _In_ SIZE_T Size); |
| 13 | + |
| 14 | +NTSYSAPI |
| 15 | +PVOID |
| 16 | +NTAPI |
| 17 | +RtlReAllocateHeap(_In_ HANDLE HeapHandle, _In_ ULONG Flags, _Frees_ptr_opt_ PVOID BaseAddress, _In_ SIZE_T Size); |
| 18 | + |
| 19 | +NTSYSAPI |
| 20 | +ULONG |
| 21 | +NTAPI |
| 22 | +RtlFreeHeap(_In_ HANDLE HeapHandle, _In_opt_ ULONG Flags, _Frees_ptr_opt_ _Post_invalid_ PVOID BaseAddress); |
| 23 | + |
| 24 | +static HANDLE GetHeapHandle() |
| 25 | +{ |
| 26 | + HANDLE heap; |
| 27 | + memcpy(&heap, (char*)RtlGetCurrentPeb() + 0x30, sizeof(HANDLE)); |
| 28 | + return heap; |
| 29 | +} |
4 | 30 |
|
5 | 31 | void* __cdecl malloc(size_t size) |
6 | 32 | { |
7 | | - HANDLE heap = RtlGetCurrentPeb()->ProcessHeap; |
8 | | - return RtlAllocateHeap(heap, HEAP_ZERO_MEMORY, size); |
| 33 | + return RtlAllocateHeap(GetHeapHandle(), HEAP_ZERO_MEMORY, size); |
9 | 34 | } |
10 | 35 |
|
11 | 36 | #ifndef _DEBUG |
12 | 37 | void* __cdecl _expand(void* block, size_t size) |
13 | 38 | { |
14 | | - HANDLE heap = RtlGetCurrentPeb()->ProcessHeap; |
15 | | - return RtlReAllocateHeap(heap, HEAP_ZERO_MEMORY, block, size); |
| 39 | + return RtlReAllocateHeap(GetHeapHandle(), HEAP_ZERO_MEMORY, block, size); |
16 | 40 | } |
17 | 41 | #endif |
18 | 42 |
|
19 | 43 | void __cdecl free(void* block) |
20 | 44 | { |
21 | | - HANDLE heap = RtlGetCurrentPeb()->ProcessHeap; |
22 | | - RtlFreeHeap(heap, 0, block); |
| 45 | + RtlFreeHeap(GetHeapHandle(), 0, block); |
23 | 46 | } |
24 | 47 |
|
25 | 48 | void* __cdecl calloc(size_t num, size_t size) |
|
0 commit comments