- leaks
[oweals/gnunet.git] / src / mesh / test_mesh_api.c
index 08a3b1c3cd9d4a33e69e1051bf1468ec76a9b9d1..dcac19085ef4348d716ea38252ecc72f512c5386 100644 (file)
  * @brief test mesh api: dummy test of callbacks
  * @author Bartlomiej Polot
  */
-
 #include "platform.h"
 #include "gnunet_util_lib.h"
+#include "gnunet_testing_lib-new.h"
 #include "gnunet_dht_service.h"
-#include "gnunet_mesh_service_new.h"
+#include "gnunet_mesh_service.h"
 
-#define VERBOSE 1
-#define VERBOSE_ARM 0
+static struct GNUNET_MESH_Handle *mesh;
 
-static struct GNUNET_MESH_MessageHandler        handlers[] = {{NULL, 0, 0}};
-static struct GNUNET_OS_Process                 *arm_pid;
-static struct GNUNET_MESH_Handle                *mesh;
-static int                                      result;
-GNUNET_SCHEDULER_TaskIdentifier                 abort_task;
-GNUNET_SCHEDULER_TaskIdentifier                 test_task;
+static struct GNUNET_MESH_Tunnel *t;
 
-static void
-do_shutdown (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
+static int result;
+
+static GNUNET_SCHEDULER_TaskIdentifier abort_task;
+
+
+/**
+ * Function is called whenever a message is received.
+ *
+ * @param cls closure (set from GNUNET_MESH_connect)
+ * @param tunnel connection to the other end
+ * @param tunnel_ctx place to store local state associated with the tunnel
+ * @param sender who sent the message
+ * @param message the actual message
+ * @param atsi performance data for the connection
+ * @return GNUNET_OK to keep the connection open,
+ *         GNUNET_SYSERR to close it (signal serious error)
+ */
+static int
+callback (void *cls, struct GNUNET_MESH_Tunnel *tunnel, void **tunnel_ctx,
+          const struct GNUNET_PeerIdentity *sender,
+          const struct GNUNET_MessageHeader *message,
+          const struct GNUNET_ATS_Information *atsi)
 {
-    if (0 != abort_task) {
-        GNUNET_SCHEDULER_cancel(abort_task);
-    }
-    if (NULL != mesh) {
-        GNUNET_MESH_disconnect (mesh);
-    }
-    if (0 != GNUNET_OS_process_kill (arm_pid, SIGTERM)) {
-        GNUNET_log_strerror (GNUNET_ERROR_TYPE_WARNING, "kill");
-    }
-    GNUNET_assert (GNUNET_OK == GNUNET_OS_process_wait (arm_pid));
-    GNUNET_OS_process_close (arm_pid);
+  return GNUNET_OK;
 }
 
+
+static struct GNUNET_MESH_MessageHandler handlers[] = { 
+  { &callback, 1, 0 },
+  { NULL, 0, 0 }
+};
+
+
 static void
-do_abort (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
+do_shutdown (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 {
-    if (0 != test_task) {
-        GNUNET_SCHEDULER_cancel(test_task);
-    }
-    result = GNUNET_SYSERR;
-    abort_task = 0;
-    do_shutdown(cls, tc);
+  if (NULL != t)
+  {
+    GNUNET_MESH_tunnel_destroy (t);
+  }
+  if (0 != abort_task)
+  {
+    GNUNET_SCHEDULER_cancel (abort_task);
+  }
+  if (NULL != mesh)
+  {
+    GNUNET_MESH_disconnect (mesh);
+  }
 }
 
+
 static void
-test (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
+do_abort (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 {
-    struct GNUNET_CONFIGURATION_Handle  *cfg = cls;
-    GNUNET_MESH_ApplicationType         app;
-
-    test_task = (GNUNET_SCHEDULER_TaskIdentifier) 0;
-    app = (GNUNET_MESH_ApplicationType) 0;
-    mesh = GNUNET_MESH_connect(cfg, NULL, NULL, handlers, &app);
-    if(NULL == mesh) {
-        GNUNET_log(GNUNET_ERROR_TYPE_ERROR, "Couldn't connect to mesh :(\n");
-        return;
-    } else {
-        GNUNET_log(GNUNET_ERROR_TYPE_DEBUG, "YAY! CONNECTED TO MESH :D\n");
-    }
-
-    GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_relative_multiply(
-                                    GNUNET_TIME_UNIT_SECONDS, 1),
-                                    &do_shutdown,
-                                    NULL);
+  result = GNUNET_SYSERR;
+  abort_task = 0;
+  do_shutdown (cls, tc);
 }
 
 
 static void
-run (void *cls,
-     char *const *args,
-     const char *cfgfile, const struct GNUNET_CONFIGURATION_Handle *cfg)
+run (void *cls, 
+     const struct GNUNET_CONFIGURATION_Handle *cfg,
+     struct GNUNET_TESTING_Peer *peer)
 {
-    arm_pid = GNUNET_OS_start_process (NULL, NULL,
-                                       "gnunet-service-arm",
-                                       "gnunet-service-arm",
-#if VERBOSE_ARM
-                                       "-L", "DEBUG",
-#endif
-                                       "-c", "test_mesh.conf",
-                                       NULL);
-
-    abort_task = GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_relative_multiply(
-                                                GNUNET_TIME_UNIT_SECONDS, 5),
-                                                &do_abort,
-                                                NULL);
-    test_task = GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_relative_multiply(
-                                                GNUNET_TIME_UNIT_SECONDS, 1),
-                                                &test,
-                                                (void *)cfg);
+  static const GNUNET_MESH_ApplicationType app[] =
+    { 1, 2, 3, 4, 5, 6, 7, 8, 0 };
 
+  mesh = GNUNET_MESH_connect (cfg, NULL, NULL, NULL, handlers, app);
+  if (NULL == mesh)
+  {
+    GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "test: Couldn't connect to mesh :(\n");
+    result = GNUNET_SYSERR;
+    return;
+  }
+  else
+  {
+    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "test: YAY! CONNECTED TO MESH :D\n");
+  }
+  t = GNUNET_MESH_tunnel_create (mesh, NULL, NULL, NULL, NULL);
+  GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_relative_multiply
+                                (GNUNET_TIME_UNIT_SECONDS, 5), &do_shutdown,
+                                NULL);
+  abort_task =
+      GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_relative_multiply
+                                    (GNUNET_TIME_UNIT_SECONDS, 20), &do_abort,
+                                    NULL);
 }
 
 
 int
 main (int argc, char *argv[])
 {
-    int ret;
-    char *const argv2[] = {"test-mesh-api",
-        "-c", "test_mesh.conf",
-#if VERBOSE
-        "-L", "DEBUG",
-#endif
-        NULL
-    };
-    struct GNUNET_GETOPT_CommandLineOption options[] = {
-        GNUNET_GETOPT_OPTION_END
-    };
-    GNUNET_log_setup ("test-mesh-api",
-#if VERBOSE
-                    "DEBUG",
-#else
-                    "WARNING",
-#endif
-                    NULL);
-    ret = GNUNET_PROGRAM_run ((sizeof (argv2) / sizeof (char *)) - 1,
-                        argv2, "test-mesh-api", "nohelp",
-                        options, &run, NULL);
-
-    if ( GNUNET_OK != ret ) {
-        GNUNET_log(GNUNET_ERROR_TYPE_WARNING,
-                   "run failed with error code %d\n", ret);
-        return 1;
-    }
-    if ( GNUNET_SYSERR == result ) {
-        GNUNET_log(GNUNET_ERROR_TYPE_WARNING,
-                   "test failed\n");
-        return 1;
-    }
-    return 0;
+  result = GNUNET_OK;
+  if (0 != GNUNET_TESTING_peer_run ("test-mesh-api",
+                                   "test_mesh.conf",
+                                   &run, NULL))
+    return 1;
+  return (result == GNUNET_OK) ? 0 : 1;
 }
+
+/* end of test_mesh_api.c */
+