diff mem.c @ 803:236a184bf6f0

Merge
author Michael Pavone <pavone@retrodev.com>
date Sun, 26 Jul 2015 16:51:03 -0700
parents c3e3a0d734e2
children 9f149f0e98b7
line wrap: on
line diff
--- a/mem.c	Sun Jul 26 16:48:25 2015 -0700
+++ b/mem.c	Sun Jul 26 16:51:03 2015 -0700
@@ -8,33 +8,30 @@
 #include <stdint.h>
 #include <stdlib.h>
 #include <unistd.h>
-#include "mem.h"
-
-/*
-void * alloc_code(size_t *size)
-{
-	*size += PAGE_SIZE - (*size & (PAGE_SIZE - 1));
-	return mmap(NULL, *size, PROT_EXEC | PROT_READ | PROT_WRITE, MAP_PRIVATE | MAP_ANONYMOUS, -1, 0);
-}
-*/
+#include <errno.h>
+#include <stdio.h>
 
-/*
-void * alloc_code(size_t *size)
-{
-	char * ret = malloc(*size);
-	char * base = (char *)(((intptr_t)ret) & (~(PAGE_SIZE-1)));
-	mprotect(base, (ret + *size) - base, PROT_EXEC | PROT_READ | PROT_WRITE);
-	return ret;
-}
-*/
+#include "mem.h"
+#ifndef MAP_ANONYMOUS
+#define MAP_ANONYMOUS MAP_ANON
+#endif
+
+#ifndef MAP_32BIT
+#define MAP_32BIT 0
+#endif
 
 void * alloc_code(size_t *size)
 {
+	//start at the 1GB mark to allow plenty of room for sbrk based malloc implementations
+	//while still keeping well within 32-bit displacement range for calling code compiled into the executable
+	static uint8_t *next = (uint8_t *)0x40000000;
 	*size += PAGE_SIZE - (*size & (PAGE_SIZE - 1));
-	void * ret = sbrk(*size);
-	if (ret == ((void *)-1)) {
+	uint8_t *ret = mmap(NULL, *size, PROT_EXEC | PROT_READ | PROT_WRITE, MAP_PRIVATE | MAP_ANONYMOUS | MAP_32BIT, -1, 0);
+	if (ret == MAP_FAILED) {
+		perror("alloc_code");
 		return NULL;
 	}
-	mprotect(ret, *size, PROT_EXEC | PROT_READ | PROT_WRITE);
+	next = ret + *size;
 	return ret;
 }
+