1 files changed, 25 insertions, 0 deletions
diff --git a/mem.c b/mem.c
index c3f03a4..cc07985 100644
--- a/mem.c
+++ b/mem.c
@@ -1,10 +1,35 @@
 #include <sys/mman.h>
 #include <stddef.h>
+#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);
 }
+*/
 
+/*
+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;
+}
+*/
+
+void * alloc_code(size_t *size)
+{
+	*size += PAGE_SIZE - (*size & (PAGE_SIZE - 1));
+	void * ret = sbrk(*size);
+	if (ret == ((void *)-1)) {
+		return NULL;
+	}
+	mprotect(ret, *size, PROT_EXEC | PROT_READ | PROT_WRITE);
+	return ret;
+}