1 files changed, 140 insertions, 0 deletions
diff --git a/src/jemalloc_cpp.cpp b/src/jemalloc_cpp.cpp
new file mode 100644
index 0000000..4c5756b
--- /dev/null
+++ b/src/jemalloc_cpp.cpp
@@ -0,0 +1,140 @@
+#include <mutex>
+#include <new>
+
+#define	JEMALLOC_CPP_CPP_
+#include "jemalloc/internal/jemalloc_internal.h"
+
+// All operators in this file are exported.
+
+// Possibly alias hidden versions of malloc and sdallocx to avoid an extra plt
+// thunk?
+//
+// extern __typeof (sdallocx) sdallocx_int
+//  __attribute ((alias ("sdallocx"),
+//		visibility ("hidden")));
+//
+// ... but it needs to work with jemalloc namespaces.
+
+void	*operator new(std::size_t size);
+void	*operator new[](std::size_t size);
+void	*operator new(std::size_t size, const std::nothrow_t&) noexcept;
+void	*operator new[](std::size_t size, const std::nothrow_t&) noexcept;
+void	operator delete(void* ptr) noexcept;
+void	operator delete[](void* ptr) noexcept;
+void	operator delete(void* ptr, const std::nothrow_t&) noexcept;
+void	operator delete[](void* ptr, const std::nothrow_t&) noexcept;
+
+#if __cpp_sized_deallocation >= 201309
+/* C++14's sized-delete operators. */
+void	operator delete(void* ptr, std::size_t size) noexcept;
+void	operator delete[](void* ptr, std::size_t size) noexcept;
+#endif
+
+
+template <bool IsNoExcept>
+JEMALLOC_INLINE
+void *
+newImpl(std::size_t size) noexcept(IsNoExcept)
+{
+	void* ptr = je_malloc(size);
+	if (likely(ptr != nullptr))
+		return (ptr);
+
+	while (ptr == nullptr) {
+		std::new_handler handler;
+		// GCC-4.8 and clang 4.0 do not have std::get_new_handler.
+		{
+			static std::mutex mtx;
+			std::lock_guard<std::mutex> lock(mtx);
+
+			handler = std::set_new_handler(nullptr);
+			std::set_new_handler(handler);
+		}
+		if (handler == nullptr)
+			break;
+
+		try {
+			handler();
+		} catch (const std::bad_alloc&) {
+			break;
+		}
+
+		ptr = je_malloc(size);
+	}
+
+	if (ptr == nullptr && !IsNoExcept)
+		std::__throw_bad_alloc();
+	return (ptr);
+}
+
+void *
+operator new(std::size_t size)
+{
+
+	return (newImpl<false>(size));
+}
+
+void *
+operator new[](std::size_t size)
+{
+
+	return (newImpl<false>(size));
+}
+
+void *
+operator new(std::size_t size, const std::nothrow_t&) noexcept
+{
+
+	return (newImpl<true>(size));
+}
+
+void *
+operator new[](std::size_t size, const std::nothrow_t&) noexcept
+{
+
+  return (newImpl<true>(size));
+}
+
+void
+operator delete(void* ptr) noexcept
+{
+
+	je_free(ptr);
+}
+
+void
+operator delete[](void* ptr) noexcept
+{
+
+	je_free(ptr);
+}
+
+void
+operator delete(void* ptr, const std::nothrow_t&) noexcept
+{
+
+	je_free(ptr);
+}
+
+void operator delete[](void* ptr, const std::nothrow_t&) noexcept
+{
+
+	je_free(ptr);
+}
+
+#if __cpp_sized_deallocation >= 201309
+
+void
+operator delete(void* ptr, std::size_t size) noexcept
+{
+
+	je_sdallocx(ptr, size, /*flags=*/0);
+}
+
+void operator delete[](void* ptr, std::size_t size) noexcept
+{
+
+	je_sdallocx(ptr, size, /*flags=*/0);
+}
+
+#endif  // __cpp_sized_deallocation