X-Git-Url: http://git.meshlink.io/?p=meshlink;a=blobdiff_plain;f=test%2Finvite-join.c;h=831cac4143f8cd3fe31b193a6e3a6783a448ee1e;hp=b69eb3d87a267179cc560d212da2706dc6fd6ca5;hb=3febbb4453f5c6b0286e7e3c68a5dbcec975b6fd;hpb=cdb3dada645394bb1e0d0bf7816291fbd9a2a4a0 diff --git a/test/invite-join.c b/test/invite-join.c index b69eb3d8..831cac41 100644 --- a/test/invite-join.c +++ b/test/invite-join.c @@ -1,70 +1,44 @@ +#ifdef NDEBUG +#undef NDEBUG +#endif + #include #include #include #include #include +#include #include "meshlink.h" +#include "utils.h" -volatile bool baz_reachable = false; - -void log_cb(meshlink_handle_t *mesh, meshlink_log_level_t level, const char *text) { - static struct timeval tv0; - struct timeval tv; - - if(tv0.tv_sec == 0) { - gettimeofday(&tv0, NULL); - } - - gettimeofday(&tv, NULL); - fprintf(stderr, "%u.%.03u ", (unsigned int)(tv.tv_sec - tv0.tv_sec), (unsigned int)tv.tv_usec / 1000); - - if(mesh) { - fprintf(stderr, "(%s) ", mesh->name); - } - - fprintf(stderr, "[%d] %s\n", level, text); -} +static struct sync_flag baz_reachable; -void status_cb(meshlink_handle_t *mesh, meshlink_node_t *node, bool reachable) { +static void status_cb(meshlink_handle_t *mesh, meshlink_node_t *node, bool reachable) { (void)mesh; - if(!strcmp(node->name, "baz")) { - baz_reachable = reachable; + if(reachable && !strcmp(node->name, "baz")) { + set_sync_flag(&baz_reachable, true); } } int main() { meshlink_set_log_cb(NULL, MESHLINK_DEBUG, log_cb); + assert(meshlink_destroy("invite_join_conf.1")); + assert(meshlink_destroy("invite_join_conf.2")); + assert(meshlink_destroy("invite_join_conf.3")); + // Open thee new meshlink instance. meshlink_handle_t *mesh1 = meshlink_open("invite_join_conf.1", "foo", "invite-join", DEV_CLASS_BACKBONE); - - if(!mesh1) { - fprintf(stderr, "Could not initialize configuration for foo\n"); - return 1; - } - - meshlink_set_log_cb(mesh1, MESHLINK_DEBUG, log_cb); + assert(mesh1); meshlink_handle_t *mesh2 = meshlink_open("invite_join_conf.2", "bar", "invite-join", DEV_CLASS_BACKBONE); - - if(!mesh2) { - fprintf(stderr, "Could not initialize configuration for bar\n"); - return 1; - } - - meshlink_set_log_cb(mesh2, MESHLINK_DEBUG, log_cb); + assert(mesh2); meshlink_handle_t *mesh3 = meshlink_open("invite_join_conf.3", "quux", "invite-join", DEV_CLASS_BACKBONE); - - if(!mesh3) { - fprintf(stderr, "Could not initialize configuration for quux\n"); - return 1; - } - - meshlink_set_log_cb(mesh3, MESHLINK_DEBUG, log_cb); + assert(mesh3); // Disable local discovery. @@ -72,61 +46,29 @@ int main() { meshlink_enable_discovery(mesh2, false); meshlink_enable_discovery(mesh3, false); - // Start the first instance and have it generate invitations. + // Have the first instance generate invitations. meshlink_set_node_status_cb(mesh1, status_cb); - if(!meshlink_start(mesh1)) { - fprintf(stderr, "Foo could not start\n"); - return 1; - } - - meshlink_add_address(mesh1, "localhost"); + assert(meshlink_set_canonical_address(mesh1, meshlink_get_self(mesh1), "localhost", NULL)); char *baz_url = meshlink_invite(mesh1, NULL, "baz"); - - if(!baz_url) { - fprintf(stderr, "Foo could not generate an invitation for baz\n"); - return 1; - } + assert(baz_url); char *quux_url = meshlink_invite(mesh1, NULL, "quux"); - - if(!quux_url) { - fprintf(stderr, "Foo could not generate an invitation for quux\n"); - return 1; - } - - fprintf(stderr, "Invitation URL for baz: %s\n", baz_url); - fprintf(stderr, "Invitation URL for quux: %s\n", quux_url); + assert(quux_url); // Have the second instance join the first. - if(!meshlink_join(mesh2, baz_url)) { - fprintf(stderr, "Baz could not join foo's mesh\n"); - return 1; - } - - free(baz_url); + assert(meshlink_start(mesh1)); - if(!meshlink_start(mesh2)) { - fprintf(stderr, "Baz could not start\n"); - return 1; - } + assert(meshlink_join(mesh2, baz_url)); + assert(meshlink_start(mesh2)); // Wait for the two to connect. - for(int i = 0; i < 60; i++) { - sleep(1); + assert(wait_sync_flag(&baz_reachable, 20)); - if(baz_reachable) { - break; - } - } - - if(!baz_reachable) { - fprintf(stderr, "Baz not reachable for foo after 20 seconds\n"); - return 1; - } + // Wait for UDP communication to become possible. int pmtu = meshlink_get_pmtu(mesh1, meshlink_get_node(mesh1, "baz")); @@ -135,20 +77,67 @@ int main() { pmtu = meshlink_get_pmtu(mesh1, meshlink_get_node(mesh1, "baz")); } - if(!pmtu) { - fprintf(stderr, "UDP communication with baz not possible after 10 seconds\n"); - return 1; - } + assert(pmtu); + + // Check that an invitation cannot be used twice + + assert(!meshlink_join(mesh3, baz_url)); + free(baz_url); // Check that nodes cannot join with expired invitations meshlink_set_invitation_timeout(mesh1, 0); - if(meshlink_join(mesh3, quux_url)) { - fprintf(stderr, "Quux could join foo's mesh using an outdated invitation\n"); - return 1; + assert(!meshlink_join(mesh3, quux_url)); + free(quux_url); + + // Check that existing nodes cannot join another mesh + + char *corge_url = meshlink_invite(mesh3, NULL, "corge"); + assert(corge_url); + + assert(meshlink_start(mesh3)); + + meshlink_stop(mesh2); + + assert(!meshlink_join(mesh2, corge_url)); + 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 = false; + + for(int i = 0; !success && 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"); + assert(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)); free(quux_url); // Clean up. @@ -156,6 +145,4 @@ int main() { meshlink_close(mesh3); meshlink_close(mesh2); meshlink_close(mesh1); - - return 0; }