Add xtalloc_reference.

Yet another talloc wrapper that should come in handy.

diff --git a/glcpp.h b/glcpp.h
index 503731b85b3b8e7ef0df104f64dd6dc1a2b6d48c..6171ce8b4a05a9b13fc641878a58010190c5369a 100644 (file)
--- a/glcpp.h
+++ b/glcpp.h
@@ -164,4 +164,10 @@ xtalloc_strndup (const void *t, const char *p, size_t n);
 char *
 xtalloc_asprintf (const void *t, const char *fmt, ...);
 
+void *
+_xtalloc_reference_loc (const void *context,
+                       const void *ptr, const char *location);
+
+#define xtalloc_reference(ctx, ptr) (_TALLOC_TYPEOF(ptr))_xtalloc_reference_loc((ctx),(ptr), __location__)
+
 #endif

diff --git a/xtalloc.c b/xtalloc.c
index e52d12ac6b260d1be098d2bd0b5d6abac276f3a2..656ac2d6cb5bff989f62462fff0e147432e643d2 100644 (file)
--- a/xtalloc.c
+++ b/xtalloc.c
@@ -82,3 +82,18 @@ xtalloc_asprintf (const void *t, const char *fmt, ...)
        va_end(ap);
        return ret;
 }
+
+void *
+_xtalloc_reference_loc (const void *context,
+                       const void *ptr, const char *location)
+{
+       void *ret;
+
+       ret = _talloc_reference_loc (context, ptr, location);
+       if (ret == NULL) {
+               fprintf (stderr, "Out of memory.\n");
+               exit (1);
+       }
+
+       return ret;
+}