munmap() extra pages during realloc()

1 files changed, 6 insertions, 1 deletions

diff --git a/src/mapalloc.c b/src/mapalloc.c
index afd8443..035332f 100644
--- a/src/mapalloc.c
+++ b/src/mapalloc.c
@@ -172,7 +172,12 @@ void *MA_realloc(void *ptr, size_t n)
 
 	if (n < (b->allocated - (PAGESIZE * 2))) {
 		b->used = n;
-		/* TODO: munmap() and mprotect() as necessary */
+		char *over = (char*)ptr + b->used + (PAGESIZE - (PAGESIZE % b->used));
+		if (over != b->over) {
+			mprotect(over, PAGESIZE, PROT_NONE);
+			munmap(over + PAGESIZE, (char*)b->over - over - PAGESIZE);
+			b->over = over;
+		}
 		return ptr;
 	}