Fix SHOW_MEMORY for gcc and glibc on linux

ConflictSet.cpp

@@ -47,7 +47,9 @@ limitations under the License.
 #if __has_builtin(__builtin_assume)
 #define assume(e) __builtin_assume(e)
 #else
-#define assume(e) __attribute__((assume(e)))
+#define assume(e)                                                              \
+  if (!(e))                                                                    \
+  __builtin_unreachable()
 #endif
 #else
 #define assume assert

Internal.h

@@ -43,6 +43,8 @@ operator<=>(const std::span<const uint8_t> &lhs,
 #if SHOW_MEMORY
 #ifdef __APPLE__
 #include <malloc/malloc.h>
+#else
+#include <malloc.h>
 #endif
 inline int64_t mallocBytes = 0;
 inline int64_t peakMallocBytes = 0;
@@ -50,17 +52,22 @@ inline int64_t peakMallocBytes = 0;
 
 // malloc that aborts on OOM and thus always returns a non-null pointer
 __attribute__((always_inline)) inline void *safe_malloc(size_t s) {
+  void *p = malloc(s);
+  if (p == nullptr) {
+    abort();
+  }
 #if SHOW_MEMORY
+#ifdef __APPLE__
   mallocBytes += s;
+#else
+  mallocBytes += malloc_usable_size(p);
+#endif
   if (mallocBytes > peakMallocBytes) {
     peakMallocBytes = mallocBytes;
   }
 #endif
-  if (void *p = malloc(s)) {
   return p;
 }
-  abort();
-}
 
 // There's nothing safer about this than free. Only called safe_free for
 // symmetry with safe_malloc.
@@ -68,6 +75,8 @@ __attribute__((always_inline)) inline void safe_free(void *p) {
 #if SHOW_MEMORY
 #ifdef __APPLE__
   mallocBytes -= malloc_size(p);
+#else
+  mallocBytes -= malloc_usable_size(p);
 #endif
 #endif
   free(p);