Implement dynstr_set_malloc_error_handler.

2 files changed, 19 insertions, 0 deletions

diff --git a/dynstr.c b/dynstr.c
index 41cb6ef..28804bf 100644
--- a/dynstr.c
+++ b/dynstr.c
@@ -23,6 +23,15 @@ dynstr_error_handler *dynstr_get_malloc_error_handler(void)
     return error_handler;
 }
 
+void dynstr_set_malloc_error_handler(dynstr_error_handler *handler)
+{
+    error_handler = handler;
+}
+
 void dynstr_malloc_error_indicate(int error, size_t size, void *oldptr)
 {
 }
+
+void dynstr_malloc_error_abort(int error, size_t size, void *oldptr)
+{
+}
diff --git a/unittests.c b/unittests.c
index 0f43982..fd30081 100644
--- a/unittests.c
+++ b/unittests.c
@@ -22,6 +22,15 @@ static int test_default_error_handler_is_indicate(void)
 }
 
 
+static int test_sets_error_handler(void)
+{
+    dynstr_set_malloc_error_handler(dynstr_malloc_error_abort);
+    FAIL_UNLESS_EQUAL(dynstr_get_malloc_error_handler(),
+                      dynstr_malloc_error_abort);
+    return true;
+}
+
+
 struct test {
     const char *name;
     int (*test)(void);
@@ -32,6 +41,7 @@ struct test {
 
 static const struct test tests[] = {
     TEST(test_default_error_handler_is_indicate),
+    TEST(test_sets_error_handler),
 };
 static const int num_tests = sizeof(tests) / sizeof(tests[0]);