Update to rra-c-util 5.3 and C TAP Harness 3.0

[kerberos/krb5-strength.git] / util / messages.c

diff --git a/util/messages.c b/util/messages.c
index 769166fc4c01b4cd0cb33a13092ce129d0b205e4..4dac8236ca030cc3e5eee756ad4e3e1a3925eb21 100644 (file)
--- a/util/messages.c
+++ b/util/messages.c
@@ -159,6 +159,31 @@ HANDLER_FUNCTION(warn)
 HANDLER_FUNCTION(die)
 
 
+/*
+ * Reset all handlers back to the defaults and free all allocated memory.
+ * This is primarily useful for programs that undergo comprehensive memory
+ * allocation analysis.
+ */
+void
+message_handlers_reset(void)
+{
+    free(debug_handlers);
+    debug_handlers = NULL;
+    if (notice_handlers != stdout_handlers) {
+        free(notice_handlers);
+        notice_handlers = stdout_handlers;
+    }
+    if (warn_handlers != stderr_handlers) {
+        free(warn_handlers);
+        warn_handlers = stderr_handlers;
+    }
+    if (die_handlers != stderr_handlers) {
+        free(die_handlers);
+        die_handlers = stderr_handlers;
+    }
+}
+
+
 /*
  * Print a message to stdout, supporting message_program_name.
  */