Merge pull request #756 from libgit2/cmn/memory-allocator

Set a custom allocator to redirect via the ruby runtime

Author: carlosmn

ext/rugged/rugged.c

@@ -604,7 +604,8 @@ void Init_rugged(void)
 	 */
 	rb_define_const(rb_mRugged, "SORT_REVERSE", INT2FIX(GIT_SORT_REVERSE));
 
-	/* Initialize libgit2 */
+	/* Set the allocator and initialize libgit2 */
+	rugged_set_allocator();
 	git_libgit2_init();
 
 	/* Hook a global object to cleanup the library

ext/rugged/rugged.h

@@ -20,6 +20,8 @@
 #include <ruby/encoding.h>
 #endif
 
+#include <ruby/util.h>
+
 #include <assert.h>
 #include <git2.h>
 #include <git2/odb_backend.h>
@@ -173,4 +175,6 @@ typedef struct _rugged_backend {
   int (* refdb_backend)(git_refdb_backend **backend_out, struct _rugged_backend *backend, const char* path);
 } rugged_backend;
 
+extern void rugged_set_allocator(void);
+
 #endif

ext/rugged/rugged_allocator.c

@@ -0,0 +1,89 @@
+/*
+ * Copyright (C) the Rugged contributors.  All rights reserved.
+ *
+ * This file is part of Rugged, distributed under the MIT license.
+ * For full terms see the included LICENSE file.
+ */
+
+#include "rugged.h"
+#include <git2/sys/alloc.h>
+
+static void *rugged_gmalloc(size_t n, const char *file, int line)
+{
+	return xmalloc(n);
+}
+
+static void *rugged_gcalloc(size_t nelem, size_t elsize, const char *file, int line)
+{
+	return xcalloc(nelem, elsize);
+}
+
+static char *rugged_gstrdup(const char *str, const char *file, int line)
+{
+	return ruby_strdup(str);
+}
+
+static char *rugged_gstrndup(const char *str, size_t n, const char *file, int line)
+{
+	size_t len;
+	char *newstr;
+
+	len = strlen(str);
+	if (len < n)
+		n = len;
+
+	newstr = xmalloc(n+1);
+	memcpy(newstr, str, n);
+	newstr[n] = '\0';
+
+	return newstr;
+}
+
+static char *rugged_gsubstrdup(const char *str, size_t n, const char *file, int line)
+{
+	char *newstr;
+
+	newstr = xmalloc(n+1);
+	memcpy(newstr, str, n);
+	newstr[n] = '\0';
+
+	return newstr;
+}
+
+static void *rugged_grealloc(void *ptr, size_t size, const char *file, int line)
+{
+	return xrealloc(ptr, size);
+}
+
+static void *rugged_greallocarray(void *ptr, size_t nelem, size_t elsize, const char *file, int line)
+{
+	return xrealloc2(ptr, nelem, elsize);
+}
+
+static void *rugged_gmallocarray(size_t nelem, size_t elsize, const char *file, int line)
+{
+	return xmalloc2(nelem, elsize);
+}
+
+static void rugged_gfree(void *ptr)
+{
+	xfree(ptr);
+}
+
+void rugged_set_allocator(void)
+{
+	git_allocator allocator;
+
+	allocator.gmalloc = rugged_gmalloc;
+	allocator.gcalloc = rugged_gcalloc;
+	allocator.gstrdup = rugged_gstrdup;
+	allocator.gstrndup = rugged_gstrndup;
+	allocator.gstrndup = rugged_gstrndup;
+	allocator.gsubstrdup = rugged_gsubstrdup;
+	allocator.grealloc = rugged_grealloc;
+	allocator.greallocarray = rugged_greallocarray;
+	allocator.gmallocarray = rugged_gmallocarray;
+	allocator.gfree = rugged_gfree;
+
+	git_libgit2_opts(GIT_OPT_SET_ALLOCATOR, &allocator);
+}