1 files changed, 14 insertions, 0 deletions

diff --git a/src/transport/test_transport_api.c b/src/transport/test_transport_api.c
index a5afc5fe4..817077f58 100644
--- a/src/transport/test_transport_api.c
+++ b/src/transport/test_transport_api.c
@@ -62,6 +62,10 @@ static struct GNUNET_SCHEDULER_Handle *sched;
 
 static int ok;
 
+static int is_tcp;
+
+static int is_udp;
+
 #if VERBOSE
 #define OKPP do { ok++; fprintf (stderr, "Now at stage %u at %s:%u\n", ok, __FILE__, __LINE__); } while (0)
 #else
@@ -283,6 +287,16 @@ main (int argc, char *argv[])
 {
   int ret;
 
+  if (strstr(argv[0], "test_transport_api_tcp") == 0)
+    {
+      is_tcp = GNUNET_YES;
+    }
+  else if (strstr(argv[0], "test_transport_api_udp") == 0)
+    {
+      is_udp = GNUNET_NO;
+    }
+
+
   GNUNET_log_setup ("test-transport-api",
 #if VERBOSE
                     "DEBUG",