simplify and improve double-free check
authorRich Felker <dalias@aerifal.cx>
Mon, 15 Aug 2011 05:59:15 +0000 (01:59 -0400)
committerRich Felker <dalias@aerifal.cx>
Mon, 15 Aug 2011 05:59:15 +0000 (01:59 -0400)
a valid mmapped block will have an even (actually aligned) "extra"
field, whereas a freed chunk on the heap will always have an in-use
neighbor.

this fixes a potential bug if mmap ever allocated memory below the
main program/brk (in which case it would be wrongly-detected as a
double-free by the old code) and allows the double-free check to work
for donated memory outside of the brk area (or, in the future,
secondary heap zones if support for their creation is added).

src/malloc/malloc.c

index db4287e..0888afa 100644 (file)
@@ -394,7 +394,7 @@ void *realloc(void *p, size_t n)
                size_t oldlen = n0 + extra;
                size_t newlen = n + extra;
                /* Crash on realloc of freed chunk */
                size_t oldlen = n0 + extra;
                size_t newlen = n + extra;
                /* Crash on realloc of freed chunk */
-               if ((uintptr_t)base < mal.brk) *(volatile char *)0=0;
+               if (extra & 1) *(volatile char *)0=0;
                if (newlen < PAGE_SIZE && (new = malloc(n))) {
                        memcpy(new, p, n-OVERHEAD);
                        free(p);
                if (newlen < PAGE_SIZE && (new = malloc(n))) {
                        memcpy(new, p, n-OVERHEAD);
                        free(p);
@@ -457,7 +457,7 @@ void free(void *p)
                char *base = (char *)self - extra;
                size_t len = CHUNK_SIZE(self) + extra;
                /* Crash on double free */
                char *base = (char *)self - extra;
                size_t len = CHUNK_SIZE(self) + extra;
                /* Crash on double free */
-               if ((uintptr_t)base < mal.brk) *(volatile char *)0=0;
+               if (extra & 1) *(volatile char *)0=0;
                __munmap(base, len);
                return;
        }
                __munmap(base, len);
                return;
        }