1 files changed, 62 insertions, 3 deletions

diff --git a/src/setup/gnunet-setup-transport-test.c b/src/setup/gnunet-setup-transport-test.c
index f351ddf5..dd2f5674 100644
--- a/src/setup/gnunet-setup-transport-test.c
+++ b/src/setup/gnunet-setup-transport-test.c
@@ -1,6 +1,6 @@
 /*
      This file is part of GNUnet.
-     Copyright (C) 2010, 2011, 2014 GNUnet e.V.
+     Copyright (C) 2010, 2011, 2014, 2017 GNUnet e.V.
 
      GNUnet is free software; you can redistribute it and/or modify
      it under the terms of the GNU General Public License as published
@@ -30,9 +30,13 @@
 /**
  * How long do we wait for the NAT test to report success?
  */
-#define TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 15)
+#define TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 2)
 
 
+/**
+ * Data structure we keep for NAT tests that run asynchronously until
+ * #TIMEOUT or shutdown.
+ */
 struct TestContext
 {
 
@@ -42,6 +46,16 @@ struct TestContext
   struct GNUNET_NAT_AUTO_Test *tst;
 
   /**
+   * Timeout task.
+   */
+  struct GNUNET_SCHEDULER_Task *tt;
+
+  /**
+   * Shutdown task.
+   */
+  struct GNUNET_SCHEDULER_Task *st;
+
+  /**
    * Name of widget to show on success.
    */
   const char *success_image;
@@ -61,7 +75,8 @@ struct TestContext
  * @param result #GNUNET_YES on success
  */
 static void
-display_test_result (struct TestContext *tc, int result)
+display_test_result (struct TestContext *tc,
+                     int result)
 {
   GObject *w;
 
@@ -93,6 +108,41 @@ display_test_result (struct TestContext *tc, int result)
 
 
 /**
+ * Task run on timeout.
+ *
+ * @param cls the `struct TestContext`
+ */
+static void
+timeout_task (void *cls)
+{
+  struct TestContext *tc = cls;
+
+  tc->tt = NULL;
+  GNUNET_SCHEDULER_cancel (tc->st);
+  tc->st = NULL;
+  display_test_result (tc,
+                       GNUNET_SYSERR);
+}
+
+
+/**
+ * Task run on shutdown.
+ *
+ * @param cls the `struct TestContext`
+ */
+static void
+shutdown_task (void *cls)
+{
+  struct TestContext *tc = cls;
+
+  tc->st = NULL;
+  GNUNET_SCHEDULER_cancel (tc->tt);
+  GNUNET_NAT_AUTO_test_stop (tc->tst);
+  GNUNET_free (tc);
+}
+
+
+/**
  * Function called by NAT on success.
  * Clean up and update GUI (with success).
  *
@@ -105,6 +155,10 @@ result_callback (void *cls,
 {
   struct TestContext *tc = cls;
 
+  GNUNET_SCHEDULER_cancel (tc->tt);
+  tc->tt = NULL;
+  GNUNET_SCHEDULER_cancel (tc->st);
+  tc->st = NULL;
   display_test_result (tc,
                        (GNUNET_NAT_ERROR_SUCCESS == result)
                        ? GNUNET_OK
@@ -137,6 +191,11 @@ GNUNET_setup_transport_test (const char *section_name,
   gtk_widget_hide (w);
   GNUNET_assert (NULL != cfg);
   GNUNET_RESOLVER_connect (cfg);
+  tc->tt = GNUNET_SCHEDULER_add_delayed (TIMEOUT,
+                                         &timeout_task,
+                                         tc);
+  tc->st = GNUNET_SCHEDULER_add_shutdown (&shutdown_task,
+                                          tc);
   tc->tst = GNUNET_NAT_AUTO_test_start (cfg,
                                         proto,
                                         section_name,