1 files changed, 11 insertions, 22 deletions

diff --git a/src/setup/gnunet-setup-transport-test.c b/src/setup/gnunet-setup-transport-test.c
index 6c21777e..f351ddf5 100644
--- a/src/setup/gnunet-setup-transport-test.c
+++ b/src/setup/gnunet-setup-transport-test.c
@@ -25,7 +25,7 @@
  */
 #include "gnunet-setup.h"
 #include <gnunet/gnunet_resolver_service.h>
-#include <gnunet/gnunet_nat_lib.h>
+#include <gnunet/gnunet_nat_auto_service.h>
 
 /**
  * How long do we wait for the NAT test to report success?
@@ -39,7 +39,7 @@ struct TestContext
   /**
    * Handle to the active NAT test.
    */
-  struct GNUNET_NAT_Test *tst;
+  struct GNUNET_NAT_AUTO_Test *tst;
 
   /**
    * Name of widget to show on success.
@@ -85,7 +85,7 @@ display_test_result (struct TestContext *tc, int result)
   }
   if (NULL != tc->tst)
   {
-    GNUNET_NAT_test_stop (tc->tst);
+    GNUNET_NAT_AUTO_test_stop (tc->tst);
     tc->tst = NULL;
   }
   GNUNET_free (tc);
@@ -117,31 +117,19 @@ result_callback (void *cls,
  * transport configuration.
  *
  * @param section_name section with the port numbers
- * @param is_tcp #GNUNET_YES for TCP, #GNUNET_NO for UDP
+ * @param proto IPPROTO_TCP or IPPROTO_UDP
  * @param success_image image to show on success
  * @param failure_image image to show on failure
  */
 void
-GNUNET_setup_transport_test (const char *section_name, int is_tcp,
+GNUNET_setup_transport_test (const char *section_name,
+                             int proto,
                              const char *success_image,
                              const char *failure_image)
 {
   struct TestContext *tc;
-  unsigned long long bnd_port;
-  unsigned long long adv_port;
   GtkWidget *w;
 
-  if (GNUNET_OK !=
-      GNUNET_CONFIGURATION_get_value_number (cfg, section_name, "PORT",
-                                             &bnd_port))
-  {
-    GNUNET_break (0);
-    return;
-  }
-  if (GNUNET_OK !=
-      GNUNET_CONFIGURATION_get_value_number (cfg, section_name,
-                                             "ADVERTISED_PORT", &adv_port))
-    adv_port = bnd_port;
   tc = GNUNET_new (struct TestContext);
   tc->success_image = success_image;
   tc->failure_image = failure_image;
@@ -149,10 +137,11 @@ GNUNET_setup_transport_test (const char *section_name, int is_tcp,
   gtk_widget_hide (w);
   GNUNET_assert (NULL != cfg);
   GNUNET_RESOLVER_connect (cfg);
-  tc->tst =
-      GNUNET_NAT_test_start (cfg, is_tcp, (uint16_t) bnd_port,
-                             (uint16_t) adv_port, TIMEOUT,
-                             &result_callback, tc);
+  tc->tst = GNUNET_NAT_AUTO_test_start (cfg,
+                                        proto,
+                                        section_name,
+                                        &result_callback,
+                                        tc);
 }
 
 /* end of gnunet-setup-transport-test.c */