X-Git-Url: http://git.meshlink.io/?a=blobdiff_plain;f=test%2Futils.c;h=f1f1c7700334e20ba1080b5a2e850c779203bbba;hb=1de30bc05683aade10634f2d357dcc2379bfcc42;hp=9a9bd211fcef7b85a5ee7904062e30f305009134;hpb=99ab0b58b4a20473670b442981eeed417dde1582;p=meshlink

diff --git a/test/utils.c b/test/utils.c
index 9a9bd211..f1f1c770 100644
--- a/test/utils.c
+++ b/test/utils.c
@@ -1,25 +1,41 @@
 #define _GNU_SOURCE 1
 
+#ifdef NDEBUG
+#undef NDEBUG
+#endif
+
 #include <stdio.h>
 #include <stdlib.h>
 #include <pthread.h>
 #include <assert.h>
 #include <errno.h>
+#include <time.h>
 
 #include "utils.h"
 
+void init_sync_flag(struct sync_flag *s) {
+	assert(pthread_mutex_init(&s->mutex, NULL) == 0);
+	assert(pthread_cond_init(&s->cond, NULL) == 0);
+}
+
 void set_sync_flag(struct sync_flag *s, bool value) {
-	pthread_mutex_lock(&s->mutex);
+	assert(pthread_mutex_lock(&s->mutex) == 0);
 	s->flag = value;
-	pthread_cond_broadcast(&s->cond);
-	pthread_mutex_unlock(&s->mutex);
+	assert(pthread_cond_broadcast(&s->cond) == 0);
+	assert(pthread_mutex_unlock(&s->mutex) == 0);
+}
+
+void reset_sync_flag(struct sync_flag *s) {
+	assert(pthread_mutex_lock(&s->mutex) == 0);
+	s->flag = false;
+	assert(pthread_mutex_unlock(&s->mutex) == 0);
 }
 
 bool check_sync_flag(struct sync_flag *s) {
 	bool flag;
-	pthread_mutex_lock(&s->mutex);
+	assert(pthread_mutex_lock(&s->mutex) == 0);
 	flag = s->flag;
-	pthread_mutex_unlock(&s->mutex);
+	assert(pthread_mutex_unlock(&s->mutex) == 0);
 	return flag;
 }
 
@@ -28,7 +44,7 @@ bool wait_sync_flag(struct sync_flag *s, int seconds) {
 	clock_gettime(CLOCK_REALTIME, &timeout);
 	timeout.tv_sec += seconds;
 
-	pthread_mutex_lock(&s->mutex);
+	assert(pthread_mutex_lock(&s->mutex) == 0);
 
 	while(!s->flag) {
 		if(!pthread_cond_timedwait(&s->cond, &s->mutex, &timeout) || errno != EINTR) {
@@ -36,48 +52,77 @@ bool wait_sync_flag(struct sync_flag *s, int seconds) {
 		}
 	}
 
-	pthread_mutex_unlock(&s->mutex);
+	assert(pthread_mutex_unlock(&s->mutex) == 0);
 
 	return s->flag;
 }
 
+void link_meshlink_pair(meshlink_handle_t *a, meshlink_handle_t *b) {
+	// Import and export both side's data
+
+	assert(meshlink_set_canonical_address(a, meshlink_get_self(a), "localhost", NULL));
+	assert(meshlink_set_canonical_address(b, meshlink_get_self(b), "localhost", NULL));
+
+	char *data = meshlink_export(a);
+	assert(data);
+	assert(meshlink_import(b, data));
+	free(data);
+
+	data = meshlink_export(b);
+	assert(data);
+	assert(meshlink_import(a, data));
+	free(data);
+}
+
 void open_meshlink_pair(meshlink_handle_t **pa, meshlink_handle_t **pb, const char *prefix) {
 	// Create two new MeshLink instances
 
 	*pa = *pb = NULL;
 
 	char *a_name, *b_name;
-	int ret_val;
-	(void)ret_val;
 
-	ret_val = asprintf(&a_name, "%s_conf.1", prefix);
+	assert(asprintf(&a_name, "%s_conf.1", prefix) > 0);
 	assert(a_name);
 
-	ret_val = asprintf(&b_name, "%s_conf.2", prefix);
+	assert(asprintf(&b_name, "%s_conf.2", prefix) > 0);
 	assert(b_name);
 
+	assert(meshlink_destroy(a_name));
+	assert(meshlink_destroy(b_name));
+
 	meshlink_handle_t *a = meshlink_open(a_name, "a", prefix, DEV_CLASS_BACKBONE);
 	assert(a);
 
 	meshlink_handle_t *b = meshlink_open(b_name, "b", prefix, DEV_CLASS_BACKBONE);
 	assert(b);
 
+	free(a_name);
+	free(b_name);
+
 	meshlink_enable_discovery(a, false);
 	meshlink_enable_discovery(b, false);
 
-	// Import and export both side's data
+	link_meshlink_pair(a, b);
+
+	*pa = a;
+	*pb = b;
+}
 
-	meshlink_add_address(a, "localhost");
+void open_meshlink_pair_ephemeral(meshlink_handle_t **pa, meshlink_handle_t **pb, const char *prefix) {
+	// Create two new MeshLink instances
 
-	char *data = meshlink_export(a);
-	assert(data);
-	assert(meshlink_import(b, data));
-	free(data);
+	*pa = *pb = NULL;
 
-	data = meshlink_export(b);
-	assert(data);
-	assert(meshlink_import(a, data));
-	free(data);
+	meshlink_handle_t *a = meshlink_open_ephemeral("a", prefix, DEV_CLASS_BACKBONE);
+	meshlink_handle_t *b = meshlink_open_ephemeral("b", prefix, DEV_CLASS_BACKBONE);
+
+	assert(a);
+	assert(b);
+
+	meshlink_enable_discovery(a, false);
+	meshlink_enable_discovery(b, false);
+
+	link_meshlink_pair(a, b);
 
 	*pa = a;
 	*pb = b;
@@ -87,17 +132,20 @@ void open_meshlink_pair(meshlink_handle_t **pa, meshlink_handle_t **pb, const ch
 static void pair_status_cb(meshlink_handle_t *mesh, meshlink_node_t *node, bool reachable) {
 	(void)node;
 
-	set_sync_flag(mesh->priv, reachable);
+	if(reachable) {
+		set_sync_flag(mesh->priv, true);
+	}
 }
 
 void start_meshlink_pair(meshlink_handle_t *a, meshlink_handle_t *b) {
 	struct sync_flag pair_status = {.flag = false};
+	init_sync_flag(&pair_status);
 
 	a->priv = &pair_status;
 	meshlink_set_node_status_cb(a, pair_status_cb);
 
-	meshlink_start(a);
-	meshlink_start(b);
+	assert(meshlink_start(a));
+	assert(meshlink_start(b));
 
 	assert(wait_sync_flag(&pair_status, 5));
 
@@ -110,21 +158,34 @@ void stop_meshlink_pair(meshlink_handle_t *a, meshlink_handle_t *b) {
 	meshlink_stop(b);
 }
 
-void close_meshlink_pair(meshlink_handle_t *a, meshlink_handle_t *b, const char *prefix) {
-	int ret_val;
-	(void)ret_val;
+void close_meshlink_pair(meshlink_handle_t *a, meshlink_handle_t *b) {
 	meshlink_close(a);
 	meshlink_close(b);
+}
 
-	if(prefix) {
-		char *a_name, *b_name;
+void log_cb(meshlink_handle_t *mesh, meshlink_log_level_t level, const char *text) {
+	static const char *levelstr[] = {
+		[MESHLINK_DEBUG] = "DEBUG",
+		[MESHLINK_INFO] = "INFO",
+		[MESHLINK_WARNING] = "WARNING",
+		[MESHLINK_ERROR] = "ERROR",
+		[MESHLINK_CRITICAL] = "CRITICAL",
+	};
 
-		ret_val = asprintf(&a_name, "%s_conf.1", prefix);
-		assert(a_name);
-		assert(meshlink_destroy(a_name));
+	static struct timespec ts0;
+	struct timespec ts;
 
-		ret_val = asprintf(&b_name, "%s_conf.2", prefix);
-		assert(b_name);
-		assert(meshlink_destroy(b_name));
+	clock_gettime(CLOCK_MONOTONIC, &ts);
+
+	if(ts0.tv_sec == 0) {
+		ts0 = ts;
 	}
+
+	float diff = (ts.tv_sec - ts0.tv_sec) + (ts.tv_nsec - ts0.tv_nsec) * 1e-9;
+
+	fprintf(stderr, "%7.3f (%s) [%s] %s\n",
+	        diff,
+	        mesh ? mesh->name : "",
+	        levelstr[level],
+	        text);
 }