2020-08-25 11:23:25 +00:00
|
|
|
#ifndef COSMOPOLITAN_LIBC_RUNTIME_MEMTRACK_H_
|
|
|
|
#define COSMOPOLITAN_LIBC_RUNTIME_MEMTRACK_H_
|
2021-01-17 03:18:37 +00:00
|
|
|
#include "libc/dce.h"
|
2021-03-01 07:42:35 +00:00
|
|
|
#include "libc/macros.internal.h"
|
2021-01-17 03:18:37 +00:00
|
|
|
#include "libc/nt/enum/version.h"
|
2020-10-11 04:18:53 +00:00
|
|
|
#include "libc/runtime/runtime.h"
|
2020-08-25 11:23:25 +00:00
|
|
|
#if !(__ASSEMBLER__ + __LINKER__ + 0)
|
|
|
|
COSMOPOLITAN_C_START_
|
|
|
|
|
2021-02-04 01:48:59 +00:00
|
|
|
#define _kAutomapStart 0x0000100080000000 /* asan can't spread its poison */
|
|
|
|
#define _kAutomapSize 0x00000fff80000000 /* beyond the above mem address */
|
2021-01-17 03:18:37 +00:00
|
|
|
#define _kFixedmapStart 0x0000200000000000
|
|
|
|
|
|
|
|
/*
|
|
|
|
* TODO: Why can't we allocate addresses above 4GB on Windows 7 x64?
|
|
|
|
* https://github.com/jart/cosmopolitan/issues/19
|
|
|
|
*/
|
|
|
|
#define MEMTRACK_ADDRESS(NORMAL, WIN7) \
|
|
|
|
(!(IsWindows() && NtGetVersion() < kNtVersionWindows10) ? NORMAL : WIN7)
|
|
|
|
#define kAutomapStart MEMTRACK_ADDRESS(_kAutomapStart, 0x10000000)
|
|
|
|
#define kAutomapSize MEMTRACK_ADDRESS(_kAutomapSize, 0x40000000)
|
2021-10-08 15:11:51 +00:00
|
|
|
#define kFixedmapStart MEMTRACK_ADDRESS(_kFixedmapStart, 0x50000000)
|
2020-08-25 11:23:25 +00:00
|
|
|
|
2021-09-04 20:20:47 +00:00
|
|
|
struct MemoryInterval {
|
|
|
|
int x;
|
|
|
|
int y;
|
|
|
|
long h;
|
|
|
|
int prot;
|
|
|
|
int flags;
|
|
|
|
};
|
|
|
|
|
2020-08-25 11:23:25 +00:00
|
|
|
struct MemoryIntervals {
|
2021-09-04 20:20:47 +00:00
|
|
|
long i, n;
|
|
|
|
struct MemoryInterval *p;
|
|
|
|
struct MemoryInterval s[OPEN_MAX];
|
2020-08-25 11:23:25 +00:00
|
|
|
};
|
|
|
|
|
2021-08-06 21:12:11 +00:00
|
|
|
extern hidden struct MemoryIntervals _mmi;
|
2020-08-25 11:23:25 +00:00
|
|
|
|
2021-08-06 21:12:11 +00:00
|
|
|
bool AreMemoryIntervalsOk(const struct MemoryIntervals *) nosideeffect hidden;
|
|
|
|
void PrintMemoryIntervals(int, const struct MemoryIntervals *) hidden;
|
|
|
|
int TrackMemoryInterval(struct MemoryIntervals *, int, int, long, int,
|
|
|
|
int) hidden;
|
2020-08-25 11:23:25 +00:00
|
|
|
int ReleaseMemoryIntervals(struct MemoryIntervals *, int, int,
|
2021-08-06 21:12:11 +00:00
|
|
|
void (*)(struct MemoryIntervals *, int, int)) hidden;
|
|
|
|
void ReleaseMemoryNt(struct MemoryIntervals *, int, int) hidden;
|
|
|
|
int UntrackMemoryIntervals(void *, size_t) hidden;
|
2020-08-25 11:23:25 +00:00
|
|
|
|
2021-09-28 05:58:51 +00:00
|
|
|
static inline noasan unsigned FindMemoryInterval(
|
|
|
|
const struct MemoryIntervals *mm, int x) {
|
|
|
|
unsigned l, m, r;
|
|
|
|
l = 0;
|
|
|
|
r = mm->i;
|
|
|
|
while (l < r) {
|
|
|
|
m = (l + r) >> 1;
|
|
|
|
if (mm->p[m].y < x) {
|
|
|
|
l = m + 1;
|
|
|
|
} else {
|
|
|
|
r = m;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return l;
|
|
|
|
}
|
|
|
|
|
2020-08-25 11:23:25 +00:00
|
|
|
COSMOPOLITAN_C_END_
|
|
|
|
#endif /* !(__ASSEMBLER__ + __LINKER__ + 0) */
|
|
|
|
#endif /* COSMOPOLITAN_LIBC_RUNTIME_MEMTRACK_H_ */
|