Unit tests for new functions in log.c

4 files changed, 138 insertions, 2 deletions

diff --git a/src/or/config.c b/src/or/config.c
index c62a25a7a..fc033ce1b 100644
--- a/src/or/config.c
+++ b/src/or/config.c
@@ -1142,8 +1142,6 @@ options_act_reversible(const or_options_t *old_options, char **msg)
     goto rollback;
   }
 
-  tor_log_update_sigsafe_err_fds();
-
  commit:
   r = 0;
   if (logs_marked) {
@@ -1153,6 +1151,7 @@ options_act_reversible(const or_options_t *old_options, char **msg)
     add_callback_log(severity, control_event_logmsg);
     control_adjust_event_log_severity();
     tor_free(severity);
+    tor_log_update_sigsafe_err_fds();
   }
   if (get_min_log_level() >= LOG_INFO &&
       get_min_log_level() != old_min_log_level) {
diff --git a/src/test/include.am b/src/test/include.am
index fb704d718..ffbfefba1 100644
--- a/src/test/include.am
+++ b/src/test/include.am
@@ -27,6 +27,7 @@ src_test_test_SOURCES = \
 	src/test/test_data.c \
 	src/test/test_dir.c \
 	src/test/test_introduce.c \
+	src/test/test_logging.c \
 	src/test/test_microdesc.c \
 	src/test/test_options.c \
 	src/test/test_pt.c \
diff --git a/src/test/test.c b/src/test/test.c
index 60fbfb132..ae4ab4462 100644
--- a/src/test/test.c
+++ b/src/test/test.c
@@ -2136,6 +2136,8 @@ extern struct testcase_t circuitlist_tests[];
 extern struct testcase_t circuitmux_tests[];
 extern struct testcase_t cell_queue_tests[];
 extern struct testcase_t options_tests[];
+extern struct testcase_t logging_tests[];
+extern struct testcase_t backtrace_tests[];
 
 static struct testgroup_t testgroups[] = {
   { "", test_array },
@@ -2144,6 +2146,7 @@ static struct testgroup_t testgroups[] = {
   { "crypto/", crypto_tests },
   { "container/", container_tests },
   { "util/", util_tests },
+  { "util/logging/", logging_tests },
   { "cellfmt/", cell_format_tests },
   { "cellqueue/", cell_queue_tests },
   { "dir/", dir_tests },
diff --git a/src/test/test_logging.c b/src/test/test_logging.c
new file mode 100644
index 000000000..4111a46eb
--- /dev/null
+++ b/src/test/test_logging.c
@@ -0,0 +1,133 @@
+/* Copyright (c) 2013, The Tor Project, Inc. */
+/* See LICENSE for licensing information */
+
+#include "orconfig.h"
+#include "or.h"
+#include "torlog.h"
+#include "test.h"
+
+static void
+dummy_cb_fn(int severity, uint32_t domain, const char *msg)
+{
+  (void)severity; (void)domain; (void)msg;
+}
+
+static void
+test_get_sigsafe_err_fds(void *arg)
+{
+  const int *fds;
+  int n;
+  log_severity_list_t include_bug, no_bug, no_bug2;
+  (void) arg;
+  init_logging();
+
+  n = tor_log_get_sigsafe_err_fds(&fds);
+  tt_int_op(n, ==, 1);
+  tt_int_op(fds[0], ==, STDERR_FILENO);
+
+  set_log_severity_config(LOG_WARN, LOG_ERR, &include_bug);
+  set_log_severity_config(LOG_WARN, LOG_ERR, &no_bug);
+  no_bug.masks[0] &= ~(LD_BUG|LD_GENERAL);
+  set_log_severity_config(LOG_INFO, LOG_NOTICE, &no_bug2);
+
+  /* Add some logs; make sure the output is as expected. */
+  mark_logs_temp();
+  add_stream_log(&include_bug, "dummy-1", 3);
+  add_stream_log(&no_bug, "dummy-2", 4);
+  add_stream_log(&no_bug2, "dummy-3", 5);
+  add_callback_log(&include_bug, dummy_cb_fn);
+  close_temp_logs();
+  tor_log_update_sigsafe_err_fds();
+
+  n = tor_log_get_sigsafe_err_fds(&fds);
+  tt_int_op(n, ==, 2);
+  tt_int_op(fds[0], ==, STDERR_FILENO);
+  tt_int_op(fds[1], ==, 3);
+
+  /* Allow STDOUT to replace STDERR. */
+  add_stream_log(&include_bug, "dummy-4", STDOUT_FILENO);
+  tor_log_update_sigsafe_err_fds();
+  n = tor_log_get_sigsafe_err_fds(&fds);
+  tt_int_op(n, ==, 2);
+  tt_int_op(fds[0], ==, 3);
+  tt_int_op(fds[1], ==, STDOUT_FILENO);
+
+  /* But don't allow it to replace explicit STDERR. */
+  add_stream_log(&include_bug, "dummy-5", STDERR_FILENO);
+  tor_log_update_sigsafe_err_fds();
+  n = tor_log_get_sigsafe_err_fds(&fds);
+  tt_int_op(n, ==, 3);
+  tt_int_op(fds[0], ==, STDERR_FILENO);
+  tt_int_op(fds[1], ==, STDOUT_FILENO);
+  tt_int_op(fds[2], ==, 3);
+
+  /* Don't overflow the array. */
+  {
+    int i;
+    for (i=5; i<20; ++i) {
+      add_stream_log(&include_bug, "x-dummy", i);
+    }
+  }
+  tor_log_update_sigsafe_err_fds();
+  n = tor_log_get_sigsafe_err_fds(&fds);
+  tt_int_op(n, ==, 8);
+
+ done:
+  ;
+}
+
+static void
+test_sigsafe_err(void *arg)
+{
+  const char *fn=get_fname("sigsafe_err_log");
+  char *content=NULL;
+  log_severity_list_t include_bug;
+  smartlist_t *lines = smartlist_new();
+  (void)arg;
+
+  set_log_severity_config(LOG_WARN, LOG_ERR, &include_bug);
+
+  init_logging();
+  mark_logs_temp();
+  add_file_log(&include_bug, fn);
+  tor_log_update_sigsafe_err_fds();
+  close_temp_logs();
+
+  close(STDERR_FILENO);
+  log_err(LD_BUG, "Say, this isn't too cool.");
+  tor_log_err_sigsafe("Minimal.\n", NULL);
+  tor_log_err_sigsafe("Testing any ",
+                      "attempt to manually log ",
+                      "from a signal.\n",
+                      NULL);
+  mark_logs_temp();
+  close_temp_logs();
+  close(STDERR_FILENO);
+  content = read_file_to_str(fn, 0, NULL);
+
+  tt_assert(content != NULL);
+  tor_split_lines(lines, content, (int)strlen(content));
+  tt_int_op(smartlist_len(lines), >=, 5);
+
+  if (strstr(smartlist_get(lines, 0), "opening new log file"))
+    smartlist_del_keeporder(lines, 0);
+  tt_assert(strstr(smartlist_get(lines, 0), "Say, this isn't too cool"));
+  /* Next line is blank. */
+  tt_assert(!strcmpstart(smartlist_get(lines, 1), "=============="));
+  tt_assert(!strcmpstart(smartlist_get(lines, 2), "Minimal."));
+  /* Next line is blank. */
+  tt_assert(!strcmpstart(smartlist_get(lines, 3), "=============="));
+  tt_str_op(smartlist_get(lines, 4), ==,
+            "Testing any attempt to manually log from a signal.");
+
+ done:
+  tor_free(content);
+  smartlist_free(lines);
+}
+
+struct testcase_t logging_tests[] = {
+  { "sigsafe_err_fds", test_get_sigsafe_err_fds, TT_FORK, NULL, NULL },
+  { "sigsafe_err", test_sigsafe_err, TT_FORK, NULL, NULL },
+  END_OF_TESTCASES
+};
+