077c9d
commit ebe544bf6e8eec35e754fd49efb027c6f161b6cb
077c9d
Author: Istvan Kurucsai <pistukem@gmail.com>
077c9d
Date:   Thu Dec 20 23:30:07 2018 -0500
077c9d
077c9d
    malloc: Add more integrity checks to mremap_chunk.
077c9d
    
077c9d
    * malloc/malloc.c (mremap_chunk): Additional checks.
077c9d
077c9d
diff --git a/malloc/malloc.c b/malloc/malloc.c
077c9d
index eb6a8ff33c0c313b..4df5cb4862a7b854 100644
077c9d
--- a/malloc/malloc.c
077c9d
+++ b/malloc/malloc.c
077c9d
@@ -2856,16 +2856,22 @@ mremap_chunk (mchunkptr p, size_t new_size)
077c9d
   char *cp;
077c9d
 
077c9d
   assert (chunk_is_mmapped (p));
077c9d
-  assert (((size + offset) & (GLRO (dl_pagesize) - 1)) == 0);
077c9d
+
077c9d
+  uintptr_t block = (uintptr_t) p - offset;
077c9d
+  uintptr_t mem = (uintptr_t) chunk2mem(p);
077c9d
+  size_t total_size = offset + size;
077c9d
+  if (__glibc_unlikely ((block | total_size) & (pagesize - 1)) != 0
077c9d
+      || __glibc_unlikely (!powerof2 (mem & (pagesize - 1))))
077c9d
+    malloc_printerr("mremap_chunk(): invalid pointer");
077c9d
 
077c9d
   /* Note the extra SIZE_SZ overhead as in mmap_chunk(). */
077c9d
   new_size = ALIGN_UP (new_size + offset + SIZE_SZ, pagesize);
077c9d
 
077c9d
   /* No need to remap if the number of pages does not change.  */
077c9d
-  if (size + offset == new_size)
077c9d
+  if (total_size == new_size)
077c9d
     return p;
077c9d
 
077c9d
-  cp = (char *) __mremap ((char *) p - offset, size + offset, new_size,
077c9d
+  cp = (char *) __mremap ((char *) block, total_size, new_size,
077c9d
                           MREMAP_MAYMOVE);
077c9d
 
077c9d
   if (cp == MAP_FAILED)