Check size is correct if specified

src/snmalloc/ds_core/defines.h

@@ -166,29 +166,46 @@ namespace snmalloc
 
 namespace snmalloc
 {
+  /**
+   * Forward declaration so that this can be called before the pal header is
+   * included.
+   */
+  template<size_t BufferSize = 1024, typename... Args>
+  [[noreturn]] inline void report_fatal_error(Args... args);
+
+  /**
+   * Forward declaration so that this can be called before the pal header is
+   * included.
+   */
+  template<size_t BufferSize = 1024, typename... Args>
+  inline void message(Args... args);
+
   template<typename... Args>
   SNMALLOC_FAST_PATH_INLINE void UNUSED(Args&&...)
   {}
 
-  inline SNMALLOC_FAST_PATH void check_client_error(const char* const str)
+  template<typename... Args>
+  inline SNMALLOC_FAST_PATH void
+  check_client_error(const char* const str, Args... args)
   {
     //[[clang::musttail]]
-    return snmalloc::error(str);
+    return snmalloc::report_fatal_error(str, args...);
   }
 
+  template<typename... Args>
   inline SNMALLOC_FAST_PATH void
-  check_client_impl(bool test, const char* const str)
+  check_client_impl(bool test, const char* const str, Args... args)
   {
     if (SNMALLOC_UNLIKELY(!test))
     {
-      if constexpr (DEBUG)
+      if constexpr (!DEBUG)
       {
-        UNUSED(str);
+        UNUSED(str, args...);
         SNMALLOC_FAST_FAIL();
       }
       else
       {
-        check_client_error(str);
+        check_client_error(str, args...);
       }
     }
   }
@@ -198,25 +215,11 @@ namespace snmalloc
 #else
   static constexpr bool CHECK_CLIENT = false;
 #endif
-
-  /**
-   * Forward declaration so that this can be called before the pal header is
-   * included.
-   */
-  template<size_t BufferSize = 1024, typename... Args>
-  [[noreturn]] inline void report_fatal_error(Args... args);
-
-  /**
-   * Forward declaration so that this can be called before the pal header is
-   * included.
-   */
-  template<size_t BufferSize = 1024, typename... Args>
-  inline void message(Args... args);
 } // namespace snmalloc
 
 #ifdef SNMALLOC_CHECK_CLIENT
-#  define snmalloc_check_client(test, str) \
-    snmalloc::check_client_impl(test, str)
+#  define snmalloc_check_client(test, str, ...) \
+    snmalloc::check_client_impl(test, str, ##__VA_ARGS__)
 #else
-#  define snmalloc_check_client(test, str)
+#  define snmalloc_check_client(test, str, ...)
 #endif

src/snmalloc/mem/localalloc.h

@@ -675,16 +675,32 @@ namespace snmalloc
 #endif
     }
 
+    void check_size(void* p, size_t size)
+    {
+#ifdef SNMALLOC_CHECK_CLIENT
+      size = size == 0 ? 1 : size;
+      auto sc = size_to_sizeclass_full(size);
+      auto pm_sc =
+        Backend::Pagemap::get_metaentry(address_cast(p)).get_sizeclass();
+      auto rsize = sizeclass_full_to_size(sc);
+      auto pm_size = sizeclass_full_to_size(pm_sc);
+      snmalloc_check_client(
+        sc == pm_sc, "Dealloc rounded size mismatch: {} != {}", rsize, pm_size);
+#else
+      UNUSED(p, size);
+#endif
+    }
+
     SNMALLOC_FAST_PATH void dealloc(void* p, size_t s)
     {
-      UNUSED(s);
+      check_size(p, s);
       dealloc(p);
     }
 
     template<size_t size>
     SNMALLOC_FAST_PATH void dealloc(void* p)
     {
-      UNUSED(size);
+      check_size(p, size);
       dealloc(p);
     }
 

src/snmalloc/mem/sizeclasstable.h

@@ -120,6 +120,11 @@ namespace snmalloc
     {
       return value == 0;
     }
+
+    constexpr bool operator==(sizeclass_t other)
+    {
+      return value == other.value;
+    }
   };
 
   using sizeclass_compress_t = uint8_t;