util as a directoryethomson/util_as_a_directory

1 files changed, 43 insertions, 0 deletions

diff --git a/util/alloc.c b/util/alloc.c
new file mode 100644
index 000000000..51c4d8029
--- /dev/null
+++ b/util/alloc.c
@@ -0,0 +1,43 @@
+/*
+ * Copyright (C) the libgit2 contributors. All rights reserved.
+ *
+ * This file is part of libgit2, distributed under the GNU GPL v2 with
+ * a Linking Exception. For full terms see the included COPYING file.
+ */
+
+#include "alloc.h"
+
+#include "allocators/stdalloc.h"
+#include "allocators/win32_crtdbg.h"
+
+git_allocator git__allocator;
+
+static int setup_default_allocator(void)
+{
+#if defined(GIT_MSVC_CRTDBG)
+	return git_win32_crtdbg_init_allocator(&git__allocator);
+#else
+	return git_stdalloc_init_allocator(&git__allocator);
+#endif
+}
+
+int git_allocator_global_init(void)
+{
+	/*
+	 * We don't want to overwrite any allocator which has been set before
+	 * the init function is called.
+	 */
+	if (git__allocator.gmalloc != NULL)
+		return 0;
+
+	return setup_default_allocator();
+}
+
+int git_allocator_setup(git_allocator *allocator)
+{
+	if (!allocator)
+		return setup_default_allocator();
+
+	memcpy(&git__allocator, allocator, sizeof(*allocator));
+	return 0;
+}