120 lines
3.2 KiB
C
120 lines
3.2 KiB
C
#include <stdio.h>
|
|
#include <stddef.h>
|
|
#include <stdlib.h>
|
|
#include <inttypes.h>
|
|
#include <time.h>
|
|
#include <sys/mman.h>
|
|
#include <socks/types.h>
|
|
#include <socks/memblock.h>
|
|
|
|
/* we're working with 4MiB of simulated system RAM */
|
|
#define MEMORY_SIZE_MB 128
|
|
|
|
#define ALLOC_START_MB 16
|
|
#define ALLOC_END_MB 18
|
|
|
|
struct mem_map_region {
|
|
phys_addr_t base;
|
|
phys_addr_t limit;
|
|
enum { REGION_FREE, REGION_RESERVED } status;
|
|
};
|
|
|
|
static struct mem_map_region mem_map[] = {
|
|
{ .base = 0x00000000, .limit = 0x0000ffff, .status = REGION_RESERVED },
|
|
{ .base = 0x00010000, .limit = 0x0004ffff, .status = REGION_FREE },
|
|
{ .base = 0x00050000, .limit = 0x0005ffff, .status = REGION_RESERVED },
|
|
{ .base = 0x00060000, .limit = 0x000affff, .status = REGION_FREE },
|
|
};
|
|
|
|
/* virtual address of where system memory is mapped */
|
|
static void *system_memory = NULL;
|
|
|
|
#define MEMPTR(offset) ((uintptr_t)system_memory + (offset))
|
|
#define MB_TO_BYTES(v) ((size_t)(v) * 0x100000)
|
|
|
|
#define PHYS_TO_VIRT(p) ((void *)((uintptr_t)system_memory + (p)))
|
|
#define VIRT_TO_PHYS(p) ((void *)((p) - (uintptr_t)system_memory))
|
|
|
|
int main(int argc, const char **argv)
|
|
{
|
|
srand(time(NULL));
|
|
system_memory = mmap(
|
|
NULL,
|
|
MB_TO_BYTES(MEMORY_SIZE_MB),
|
|
PROT_READ | PROT_WRITE,
|
|
MAP_PRIVATE | MAP_ANONYMOUS,
|
|
-1, 0);
|
|
|
|
if (system_memory == MAP_FAILED) {
|
|
perror("mmap");
|
|
fprintf(stderr, "cannot allocate simulated system RAM buffer\n");
|
|
return -1;
|
|
}
|
|
|
|
phys_addr_t pmem_base = UINTPTR_MAX, pmem_limit = 0;
|
|
size_t nr_mem_map_entries = sizeof mem_map / sizeof mem_map[0];
|
|
|
|
for (size_t i = 0; i < nr_mem_map_entries; i++) {
|
|
if (mem_map[i].base < pmem_base) {
|
|
pmem_base = mem_map[i].base;
|
|
}
|
|
|
|
if (mem_map[i].limit > pmem_limit) {
|
|
pmem_limit = mem_map[i].limit;
|
|
}
|
|
}
|
|
|
|
memblock_add(pmem_base, pmem_limit);
|
|
|
|
for (size_t i = 0; i < nr_mem_map_entries; i++) {
|
|
if (mem_map[i].status == REGION_RESERVED) {
|
|
memblock_reserve(mem_map[i].base, mem_map[i].limit);
|
|
}
|
|
}
|
|
|
|
printf("allocated %u MiB (0x%zx bytes) of memory to act as system RAM at %p\n", MEMORY_SIZE_MB, MB_TO_BYTES(MEMORY_SIZE_MB), system_memory);
|
|
|
|
uintptr_t voffset = (uintptr_t)system_memory;
|
|
|
|
memblock_init(MB_TO_BYTES(ALLOC_START_MB) + voffset, MB_TO_BYTES(ALLOC_END_MB) + voffset, voffset);
|
|
|
|
printf("memblock heap initialised in 0x%zx-0x%zx\n", MB_TO_BYTES(ALLOC_START_MB), MB_TO_BYTES(ALLOC_END_MB));
|
|
|
|
memblock_add(0, MB_TO_BYTES(MEMORY_SIZE_MB));
|
|
|
|
for (int i = 0; i < 8; i++) {
|
|
int size = 512 + (rand() % 16384);
|
|
|
|
phys_addr_t alloc = memblock_alloc_phys(size);
|
|
printf("allocated %d bytes at 0x%" PRIxPTR "\n", size, alloc);
|
|
}
|
|
|
|
|
|
printf("memory regions:\n");
|
|
|
|
memblock_iter_t it;
|
|
for_each_mem_range(&it, 0, 0x100000) {
|
|
printf("\t%08" PRIxPTR "-%08" PRIxPTR "\n",
|
|
it.it_base,
|
|
it.it_limit);
|
|
}
|
|
|
|
printf("reserved regions:\n");
|
|
for_each_reserved_mem_range(&it, 0, 0x100000) {
|
|
printf("\t%08" PRIxPTR "-%08" PRIxPTR " (%s)\n",
|
|
it.it_base,
|
|
it.it_limit,
|
|
it.it_status == MEMBLOCK_ALLOC ? "allocated" : "reserved");
|
|
}
|
|
|
|
printf("free regions:\n");
|
|
for_each_free_mem_range(&it, 0, ULLONG_MAX) {
|
|
printf("\t%08" PRIxPTR "-%08" PRIxPTR "\n",
|
|
it.it_base,
|
|
it.it_limit);
|
|
}
|
|
|
|
munmap(system_memory, MB_TO_BYTES(MEMORY_SIZE_MB));
|
|
return 0;
|
|
}
|