X-Git-Url: http://git.meshlink.io/?a=blobdiff_plain;f=test%2Finvite-join.c;h=f76dc536fffeb57da76f7e69400ebec5924c8ba0;hb=15b02e0766ce4fdbf604489fdb1f55c9a0cce850;hp=c9ebea6043d2a39b05f02c9e163711eec028a61b;hpb=fa05f996c5500c056a36c1d43e33a407f876643c;p=meshlink

diff --git a/test/invite-join.c b/test/invite-join.c
index c9ebea60..f76dc536 100644
--- a/test/invite-join.c
+++ b/test/invite-join.c
@@ -2,6 +2,8 @@
 #include <unistd.h>
 #include <stdlib.h>
 #include <string.h>
+#include <sys/time.h>
+#include <assert.h>
 
 #include "meshlink.h"
 
@@ -105,8 +107,6 @@ int main() {
 		return 1;
 	}
 
-	free(baz_url);
-
 	if(!meshlink_start(mesh2)) {
 		fprintf(stderr, "Baz could not start\n");
 		return 1;
@@ -139,6 +139,15 @@ int main() {
 		return 1;
 	}
 
+	// Check that an invitation cannot be used twice
+
+	if(meshlink_join(mesh3, baz_url)) {
+		fprintf(stderr, "Quux could join foo's mesh using an already used invitation\n");
+		return 1;
+	}
+
+	free(baz_url);
+
 	// Check that nodes cannot join with expired invitations
 
 	meshlink_set_invitation_timeout(mesh1, 0);
@@ -150,6 +159,67 @@ int main() {
 
 	free(quux_url);
 
+	// Check that existing nodes cannot join another mesh
+
+	char *corge_url = meshlink_invite(mesh3, NULL, "corge");
+
+	if(!corge_url) {
+		fprintf(stderr, "Quux could not generate an invitation for corge\n");
+		return 1;
+	}
+
+	fprintf(stderr, "Invitation URL for corge: %s\n", corge_url);
+
+	if(!meshlink_start(mesh3)) {
+		fprintf(stderr, "Quux could not start\n");
+		return 1;
+	}
+
+	meshlink_stop(mesh2);
+
+	if(meshlink_join(mesh2, corge_url)) {
+		fprintf(stderr, "Bar could join twice\n");
+		return 1;
+	}
+
+	free(corge_url);
+
+	// Check that invitations work correctly after changing ports
+
+	meshlink_set_invitation_timeout(mesh1, 86400);
+	meshlink_stop(mesh1);
+	meshlink_stop(mesh3);
+
+	int oldport = meshlink_get_port(mesh1);
+	bool success;
+
+	for(int i = 0; i < 100; i++) {
+		success = meshlink_set_port(mesh1, 0x9000 + rand() % 0x1000);
+	}
+
+	assert(success);
+	int newport = meshlink_get_port(mesh1);
+	assert(oldport != newport);
+
+	assert(meshlink_start(mesh1));
+	quux_url = meshlink_invite(mesh1, NULL, "quux");
+	fprintf(stderr, "Invitation URL for quux: %s\n", quux_url);
+
+	// The old port should not be in the invitation URL
+
+	char portstr[10];
+	snprintf(portstr, sizeof(portstr), ":%d", oldport);
+	assert(!strstr(quux_url, portstr));
+
+	// The new port should be in the invitation URL
+
+	snprintf(portstr, sizeof(portstr), ":%d", newport);
+	assert(strstr(quux_url, portstr));
+
+	// The invitation should work
+
+	assert(meshlink_join(mesh3, quux_url));
+
 	// Clean up.
 
 	meshlink_close(mesh3);