expanding the page range id feature to e able to distinguish better

 page states.

Author: devnexen

src/snmalloc/pal/pal.h

@@ -78,6 +78,21 @@ namespace snmalloc
     Pal::error(str);
   }
 
+  template<StateMem state_mem>
+  inline void pagetype(char *buf, size_t len)
+  {
+    ssize_t sz;
+    if constexpr (state_mem == Unused)
+      sz = snprintf(buf, len, "snmalloc (Unused)");
+    else if constexpr (state_mem == Allocated)
+      sz = snprintf(buf, len, "snmalloc (Allocated)");
+    else if constexpr (state_mem == Pagemap)
+      sz = snprintf(buf, len, "snmalloc (Pagemap)");
+    else if constexpr (state_mem == Metadata)
+      sz = snprintf(buf, len, "snmalloc (Metadata)");
+    buf[sz] = 0;
+  }
+
   // Used to keep Superslab metadata committed.
   static constexpr size_t OS_PAGE_SIZE = Pal::page_size;
 

src/snmalloc/pal/pal_consts.h

@@ -94,6 +94,14 @@ namespace snmalloc
     YesZero
   };
 
+  enum StateMem
+  {
+    Unused,
+    Allocated,
+    Pagemap,
+    Metadata
+  };
+
   /**
    * Default Tag ID for the Apple class
    */

src/snmalloc/pal/pal_linux.h

@@ -35,35 +35,54 @@ namespace snmalloc
      */
     static constexpr int default_mmap_flags = MAP_NORESERVE;
 
-    static void* reserve(size_t size) noexcept
+    /**
+     * Not generalizing this handler purposely as in other platforms
+     * mechanisms could differ greatly.
+     */
+    template<StateMem state_mem>
+    static void pageid(void* p, size_t size) noexcept
     {
-      void* p = PALPOSIX<PALLinux>::reserve(size);
-      if (p)
-      {
-        madvise(p, size, MADV_DONTDUMP);
 #  ifdef SNMALLOC_PAGEID
 #    ifndef PR_SET_VMA
 #      define PR_SET_VMA 0x53564d41
 #      define PR_SET_VMA_ANON_NAME 0
 #    endif
 
-        /**
-         *
-         * If the kernel is set with CONFIG_ANON_VMA_NAME
-         * the reserved pages would appear as follow
-         *
-         * 7fa5f0ceb000-7fa5f0e00000 rw-p 00000000 00:00 0 [anon:snmalloc]
-         * 7fa5f0e00000-7fa5f1800000 rw-p 00000000 00:00 0 [anon:snmalloc]
-         *
-         */
-
-        prctl(
-          PR_SET_VMA,
-          PR_SET_VMA_ANON_NAME,
-          (unsigned long)p,
-          size,
-          (unsigned long)"snmalloc");
+      /**
+       *
+       * If the kernel is set with CONFIG_ANON_VMA_NAME
+       * the reserved pages would appear as follow
+       *
+       * 7fa5f0ceb000-7fa5f0e00000 rw-p 00000000 00:00 0 [anon:snmalloc
+       * (<type>)] 7fa5f0e00000-7fa5f1800000 rw-p 00000000 00:00 0
+       * [anon:snmalloc (<type>)]
+       *
+       * The 80 buffer limit is specific to this syscall.
+       */
+
+      char buf[80] = {0};
+      pagetype<state_mem>(buf, sizeof(buf));
+
+      prctl(
+        PR_SET_VMA,
+        PR_SET_VMA_ANON_NAME,
+        (unsigned long)p,
+        size,
+        (unsigned long)buf);
+#  else
+      UNUSED(p);
+      UNUSED(size);
 #  endif
+    }
+
+    template<StateMem state_mem = Unused>
+    static void* reserve(size_t size) noexcept
+    {
+      void* p = PALPOSIX<PALLinux>::reserve<state_mem>(size);
+      if (p)
+      {
+        madvise(p, size, MADV_DONTDUMP);
+        pageid<state_mem>(p, size);
       }
       return p;
     }
@@ -131,11 +150,12 @@ namespace snmalloc
     /**
      * Notify platform that we will be using these pages.
      */
-    template<ZeroMem zero_mem>
+    template<ZeroMem zero_mem, StateMem state_mem = Allocated>
     static void notify_using(void* p, size_t size) noexcept
     {
       PALPOSIX<PALLinux>::notify_using<zero_mem>(p, size);
       madvise(p, size, MADV_DODUMP);
+      pageid<state_mem>(p, size);
     }
 
     static uint64_t get_entropy64()

src/snmalloc/pal/pal_posix.h

@@ -221,7 +221,7 @@ namespace snmalloc
      * are also zeroing the pages in which case we call the platform's `zero`
      * function, or we have initially mapped the pages as PROT_NONE.
      */
-    template<ZeroMem zero_mem>
+    template<ZeroMem zero_mem, StateMem state_mem = Allocated>
     static void notify_using(void* p, size_t size) noexcept
     {
       SNMALLOC_ASSERT(
@@ -316,6 +316,7 @@ namespace snmalloc
      * POSIX does not define a portable interface for specifying alignment
      * greater than a page.
      */
+    template<StateMem state_mem = Unused>
     static void* reserve(size_t size) noexcept
     {
       // If enforcing access, map pages initially as None, and then

src/snmalloc/pal/pal_windows.h

@@ -198,6 +198,7 @@ namespace snmalloc
     }
 #  endif
 
+    template<StateMem state_mem = Unused>
     static void* reserve(size_t size) noexcept
     {
       void* ret = VirtualAlloc(nullptr, size, MEM_RESERVE, PAGE_READWRITE);