From 7a9502973d4c20bd5ac962a9b6e5869494990025 Mon Sep 17 00:00:00 2001
From: daan <daanl@outlook.com>
Date: Thu, 16 Jan 2020 15:57:11 -0800
Subject: [PATCH] rearrange STL allocator code: remove pragma, ifdef for C++11

---
 ide/vs2019/mimalloc.vcxproj |  2 +-
 include/mimalloc-internal.h |  5 ++++
 include/mimalloc.h          | 46 +++++++++++++++----------------------
 src/alloc.c                 |  7 +-----
 4 files changed, 25 insertions(+), 35 deletions(-)

diff --git a/ide/vs2019/mimalloc.vcxproj b/ide/vs2019/mimalloc.vcxproj
index f59de292..037e380d 100644
--- a/ide/vs2019/mimalloc.vcxproj
+++ b/ide/vs2019/mimalloc.vcxproj
@@ -111,7 +111,7 @@
   </ItemDefinitionGroup>
   <ItemDefinitionGroup Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">
     <ClCompile>
-      <WarningLevel>Level2</WarningLevel>
+      <WarningLevel>Level4</WarningLevel>
       <Optimization>Disabled</Optimization>
       <SDLCheck>true</SDLCheck>
       <ConformanceMode>true</ConformanceMode>
diff --git a/include/mimalloc-internal.h b/include/mimalloc-internal.h
index a9391a40..500764ed 100644
--- a/include/mimalloc-internal.h
+++ b/include/mimalloc-internal.h
@@ -167,6 +167,11 @@ bool        _mi_page_is_valid(mi_page_t* page);
 
 // Overflow detecting multiply
 static inline bool mi_mul_overflow(size_t count, size_t size, size_t* total) {
+  // quick check for the case where count is one (common for C++ allocators)
+  if (count==1) {
+    *total = size;
+    return false;
+  }
 #if __has_builtin(__builtin_umul_overflow) || __GNUC__ >= 5
 #include <limits.h>   // UINT_MAX, ULONG_MAX
 #if (SIZE_MAX == UINT_MAX)
diff --git a/include/mimalloc.h b/include/mimalloc.h
index 29481c80..4c5b0cad 100644
--- a/include/mimalloc.h
+++ b/include/mimalloc.h
@@ -73,8 +73,6 @@ terms of the MIT license. A copy of the license can be found in the file
 #include <stdbool.h>    // bool
 
 #ifdef __cplusplus
-#include <type_traits> // true_type
-
 extern "C" {
 #endif
 
@@ -337,41 +335,33 @@ mi_decl_export void* mi_new_aligned_nothrow(size_t n, size_t alignment) mi_attr_
 }
 #endif
 
+// ---------------------------------------------------------------------------------------------
+// Implement the C++ std::allocator interface for use in STL containers.
+// (note: see `mimalloc-new-delete.h` for overriding the new/delete operators globally)
+// ---------------------------------------------------------------------------------------------
 #ifdef __cplusplus
 
-// ------------------------------------------------------
-// STL allocator - an extension to hook mimalloc into STL 
-// containers in place of std::allocator.
-// ------------------------------------------------------
+#if (__cplusplus >= 201103L) || (_MSC_VER > 1900)  // C++11
+#include <type_traits> // true_type
+#endif
 
-#pragma warning(disable: 4100)
-template <class T>
-struct mi_stl_allocator {
+template<class T> struct mi_stl_allocator {
   typedef T value_type;
-  
+#if (__cplusplus >= 201103L) || (_MSC_VER > 1900)  // C++11
   using propagate_on_container_copy_assignment = std::true_type;
   using propagate_on_container_move_assignment = std::true_type;
   using propagate_on_container_swap = std::true_type;
   using is_always_equal = std::true_type;
-
-  mi_stl_allocator() noexcept {}
-  mi_stl_allocator(const mi_stl_allocator& other) noexcept {}
-  template <class U>
-  mi_stl_allocator(const mi_stl_allocator<U>& other) noexcept {}
-
-  T* allocate(size_t n, const void* hint = 0) {
-    return (T*)mi_mallocn(n, sizeof(T));
-  }
-
-  void deallocate(T* p, size_t n) {
-    mi_free(p);
-  }
+#endif
+  mi_stl_allocator() mi_attr_noexcept {}
+  mi_stl_allocator(const mi_stl_allocator& other) mi_attr_noexcept { (void)other; }
+  template<class U> mi_stl_allocator(const mi_stl_allocator<U>& other) mi_attr_noexcept { (void)other; }
+  T* allocate(size_t n, const void* hint = 0) { (void)hint; return (T*)mi_mallocn(n, sizeof(T)); }
+  void deallocate(T* p, size_t n)             { mi_free_size(p,n); }
 };
 
-template <class T1, class T2>
-bool operator==(const mi_stl_allocator<T1>& lhs, const mi_stl_allocator<T2>& rhs) noexcept { return true; }
-template <class T1, class T2>
-bool operator!=(const mi_stl_allocator<T1>& lhs, const mi_stl_allocator<T2>& rhs) noexcept { return false; }
-#endif
+template<class T1,class T2> bool operator==(const mi_stl_allocator<T1>& lhs, const mi_stl_allocator<T2>& rhs) mi_attr_noexcept { (void)lhs; (void)rhs; return true; }
+template<class T1,class T2> bool operator!=(const mi_stl_allocator<T1>& lhs, const mi_stl_allocator<T2>& rhs) mi_attr_noexcept { (void)lhs; (void)rhs; return false; }
+#endif // __cplusplus
 
 #endif
diff --git a/src/alloc.c b/src/alloc.c
index be63f86a..d66c629b 100644
--- a/src/alloc.c
+++ b/src/alloc.c
@@ -443,12 +443,7 @@ mi_decl_allocator void* mi_calloc(size_t count, size_t size) mi_attr_noexcept {
 // Uninitialized `calloc`
 extern mi_decl_allocator void* mi_heap_mallocn(mi_heap_t* heap, size_t count, size_t size) mi_attr_noexcept {
   size_t total;
-  if (count==1) {
-    total = size;
-  }
-  else if (mi_mul_overflow(count, size, &total)) {
-    return NULL;
-  }
+  if (mi_mul_overflow(count, size, &total)) return NULL;
   return mi_heap_malloc(heap, total);
 }