mirror of
https://github.com/jart/cosmopolitan.git
synced 2025-01-31 19:43:32 +00:00
f4f4caab0e
I wanted a tiny scriptable meltdown proof way to run userspace programs and visualize how program execution impacts memory. It helps to explain how things like Actually Portable Executable works. It can show you how the GCC generated code is going about manipulating matrices and more. I didn't feel fully comfortable with Qemu and Bochs because I'm not smart enough to understand them. I wanted something like gVisor but with much stronger levels of assurances. I wanted a single binary that'll run, on all major operating systems with an embedded GPL barrier ZIP filesystem that is tiny enough to transpile to JavaScript and run in browsers too. https://justine.storage.googleapis.com/emulator625.mp4
218 lines
7.6 KiB
C
218 lines
7.6 KiB
C
/*-*- mode:c;indent-tabs-mode:nil;c-basic-offset:2;tab-width:8;coding:utf-8 -*-│
|
|
│vi: set net ft=c ts=2 sts=2 sw=2 fenc=utf-8 :vi│
|
|
╞══════════════════════════════════════════════════════════════════════════════╡
|
|
│ Copyright 2020 Justine Alexandra Roberts Tunney │
|
|
│ │
|
|
│ This program is free software; you can redistribute it and/or modify │
|
|
│ it under the terms of the GNU General Public License as published by │
|
|
│ the Free Software Foundation; version 2 of the License. │
|
|
│ │
|
|
│ This program is distributed in the hope that it will be useful, but │
|
|
│ WITHOUT ANY WARRANTY; without even the implied warranty of │
|
|
│ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU │
|
|
│ General Public License for more details. │
|
|
│ │
|
|
│ You should have received a copy of the GNU General Public License │
|
|
│ along with this program; if not, write to the Free Software │
|
|
│ Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA │
|
|
│ 02110-1301 USA │
|
|
╚─────────────────────────────────────────────────────────────────────────────*/
|
|
#include "libc/log/check.h"
|
|
#include "libc/macros.h"
|
|
#include "libc/mem/mem.h"
|
|
#include "libc/str/str.h"
|
|
#include "libc/sysv/errfuns.h"
|
|
#include "tool/build/lib/memory.h"
|
|
#include "tool/build/lib/pml4t.h"
|
|
|
|
static int64_t MakeAddress(unsigned short a[4]) {
|
|
uint64_t x;
|
|
x = 0;
|
|
x |= a[0];
|
|
x <<= 9;
|
|
x |= a[1];
|
|
x <<= 9;
|
|
x |= a[2];
|
|
x <<= 9;
|
|
x |= a[3];
|
|
x <<= 12;
|
|
return SignExtendAddr(x);
|
|
}
|
|
|
|
static uint64_t *GetPageTable(pml4t_t p, long i, void *NewPhysicalPage(void)) {
|
|
uint64_t *res;
|
|
DCHECK_ALIGNED(4096, p);
|
|
DCHECK(0 <= i && i < 512);
|
|
if (IsValidPage(p[i])) {
|
|
res = UnmaskPageAddr(p[i]);
|
|
} else if ((res = NewPhysicalPage())) {
|
|
DCHECK_ALIGNED(4096, res);
|
|
p[i] = MaskPageAddr(res) | 0b11;
|
|
}
|
|
return res;
|
|
}
|
|
|
|
/**
|
|
* Maps virtual page region to system memory region.
|
|
*
|
|
* @param pml4t is root of 48-bit page tables
|
|
* @param v is fixed page-aligned virtual address, rounded down
|
|
* @param r is real memory address, rounded down
|
|
* @param n is number of bytes needed, rounded up
|
|
* @return 0 on success, or -1 w/ errno
|
|
* @note existing pages are overwritten
|
|
*/
|
|
int RegisterPml4t(pml4t_t pml4t, int64_t v, int64_t r, size_t n,
|
|
void *NewPhysicalPage(void)) {
|
|
unsigned i, j, k, l;
|
|
uint64_t *pdpt, *pdt, *pd, u;
|
|
if (!n) return 0;
|
|
u = ROUNDDOWN(r, 4096);
|
|
n = ROUNDUP(n, 4096) >> 12;
|
|
i = (v >> 39) & 511;
|
|
j = (v >> 30) & 511;
|
|
k = (v >> 21) & 511;
|
|
l = (v >> 12) & 511;
|
|
if (u + n > 0x800000000000) return eoverflow();
|
|
if (r + n > 0x800000000000) return eoverflow();
|
|
for (; i < 512; ++i) {
|
|
if (!(pdpt = GetPageTable(pml4t, i, NewPhysicalPage))) return -1;
|
|
for (; j < 512; ++j) {
|
|
if (!(pdt = GetPageTable(pdpt, j, NewPhysicalPage))) return -1;
|
|
for (; k < 512; ++k) {
|
|
if (!(pd = GetPageTable(pdt, k, NewPhysicalPage))) return -1;
|
|
for (; l < 512; ++l) {
|
|
pd[l] = MaskPageAddr(u) | 0b11;
|
|
if (!--n) return 0;
|
|
u += 4096;
|
|
}
|
|
l = 0;
|
|
}
|
|
k = 0;
|
|
}
|
|
j = 0;
|
|
}
|
|
return enomem();
|
|
}
|
|
|
|
/**
|
|
* Locates free memory range.
|
|
*
|
|
* @param hint specifies signedness and around where to start searching
|
|
* @return virtual page address with size bytes free, or -1 w/ errno
|
|
*/
|
|
int64_t FindPml4t(pml4t_t pml4t, int64_t hint, uint64_t size,
|
|
void *NewPhysicalPage(void)) {
|
|
int64_t res;
|
|
unsigned short a[4], b[4];
|
|
uint64_t *pdpt, *pdt, *pd, have;
|
|
if (!size) return einval();
|
|
have = 0;
|
|
size = ROUNDUP(size, 4096) >> 12;
|
|
b[0] = a[0] = (hint >> 39) & 511;
|
|
b[1] = a[1] = (hint >> 30) & 511;
|
|
b[2] = a[2] = (hint >> 21) & 511;
|
|
a[3] = 0;
|
|
for (; b[0] < 512; ++b[0]) {
|
|
if (!(pdpt = GetPageTable(pml4t, b[0], NewPhysicalPage))) return -1;
|
|
for (; b[1] < 512; ++b[1]) {
|
|
if (!(pdt = GetPageTable(pdpt, b[1], NewPhysicalPage))) return -1;
|
|
for (; b[2] < 512; ++b[2]) {
|
|
if (!IsValidPage(pdt[b[2]])) {
|
|
if ((have += 512) >= size) {
|
|
return MakeAddress(a);
|
|
}
|
|
} else if (size < 0x200) {
|
|
pd = UnmaskPageAddr(pdt[b[2]]);
|
|
for (b[3] = 0; b[3] < 512; ++b[3]) {
|
|
if (!IsValidPage(pd[b[3]])) {
|
|
if ((have += 1) >= size) {
|
|
return MakeAddress(a);
|
|
}
|
|
} else {
|
|
have = 0;
|
|
a[0] = b[0];
|
|
a[1] = b[1];
|
|
a[2] = b[2];
|
|
a[3] = b[3];
|
|
if ((a[3] += 1) == 512) {
|
|
a[3] = 0;
|
|
if ((a[2] += 1) == 512) {
|
|
a[2] = 0;
|
|
if ((a[1] += 1) == 512) {
|
|
a[1] = 0;
|
|
a[0] += 1;
|
|
if (a[0] == 256 || a[0] == 512) {
|
|
return eoverflow();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
a[2] = 0;
|
|
}
|
|
a[1] = 0;
|
|
}
|
|
return enomem();
|
|
}
|
|
|
|
/**
|
|
* Unmaps pages and frees page tables.
|
|
*/
|
|
int FreePml4t(pml4t_t pml4t, int64_t addr, uint64_t size,
|
|
void FreePhysicalPageTable(void *),
|
|
int FreePhysicalPages(void *, size_t)) {
|
|
int rc;
|
|
char *pages;
|
|
uint64_t i, *pdpt, *pdt, *pd;
|
|
unsigned short r, s[4], a[4], R[2][2] = {{256, 512}, {0, 256}};
|
|
a[0] = addr >> 39;
|
|
a[1] = addr >> 30;
|
|
a[2] = addr >> 21;
|
|
a[3] = addr >> 12;
|
|
size = ROUNDUP(size, 4096) >> 12;
|
|
for (rc = r = 0; r < ARRAYLEN(R); ++r) {
|
|
for (a[0] &= 511; size && R[r][0] <= a[0] && a[0] < R[r][1]; ++a[0]) {
|
|
if (!IsValidPage(pml4t[a[0]])) continue;
|
|
pdpt = UnmaskPageAddr(pml4t[a[0]]);
|
|
for (s[1] = (a[1] &= 511); size && a[1] < 512; ++a[1]) {
|
|
if (!IsValidPage(pdpt[a[1]])) continue;
|
|
pdt = UnmaskPageAddr(pdpt[a[1]]);
|
|
for (s[2] = (a[2] &= 511); size && a[2] < 512; ++a[2]) {
|
|
if (!IsValidPage(pdt[a[2]])) continue;
|
|
pd = UnmaskPageAddr(pdt[a[2]]);
|
|
for (s[3] = (a[3] &= 511); size && a[3] < 512; ++a[3]) {
|
|
if (IsValidPage(pd[a[3]])) {
|
|
pages = UnmaskPageAddr(pd[a[3]]);
|
|
pd[a[3]] = 0;
|
|
for (i = 1; i + 1 < size && a[3] + i < 512; ++i) {
|
|
if (!IsValidPage(pd[a[3] + i])) break;
|
|
if (UnmaskPageAddr(pd[a[3] + i]) != pages + i * 4096) break;
|
|
pd[a[3] + i] = 0;
|
|
}
|
|
FreePhysicalPages(pages, i * 4096);
|
|
a[3] += i - 1;
|
|
size -= i;
|
|
}
|
|
}
|
|
if (s[3] == 0 && a[3] == 512) {
|
|
FreePhysicalPageTable(pd);
|
|
pdt[a[2]] = 0;
|
|
}
|
|
}
|
|
if (s[2] == 0 && a[2] == 512) {
|
|
FreePhysicalPageTable(pdt);
|
|
pdpt[a[1]] = 0;
|
|
}
|
|
}
|
|
if (s[1] == 0 && a[1] == 512) {
|
|
FreePhysicalPageTable(pdpt);
|
|
pml4t[a[0]] = 0;
|
|
}
|
|
}
|
|
}
|
|
return 0;
|
|
}
|