git.meshlink.io Git - meshlink/blob - test/invite-join.c

   1 #ifdef NDEBUG
   2 #undef NDEBUG
   3 #endif
   4
   5 #include <stdio.h>
   6 #include <unistd.h>
   7 #include <stdlib.h>
   8 #include <string.h>
   9 #include <sys/time.h>
  10 #include <assert.h>
  11 #include <errno.h>
  12
  13 #include "meshlink.h"
  14 #include "devtools.h"
  15 #include "utils.h"
  16
  17 static struct sync_flag baz_reachable;
  18 static struct sync_flag seven_reachable;
  19 static struct sync_flag commits_first_flag;
  20
  21 static void status_cb(meshlink_handle_t *mesh, meshlink_node_t *node, bool reachable) {
  22         (void)mesh;
  23
  24         if(reachable && !strcmp(node->name, "baz")) {
  25                 set_sync_flag(&baz_reachable, true);
  26         }
  27
  28         if(reachable && !strcmp(node->name, "seven")) {
  29                 set_sync_flag(&seven_reachable, true);
  30         }
  31 }
  32
  33 static void invitee_commits_first_cb(bool inviter_first) {
  34         // Check that eight has committed foo's host config file, but foo hasn't committed eight's
  35         assert(access("invite_join_conf.8/current/hosts/foo", F_OK) == 0);
  36         assert(access("invite_join_conf.1/current/hosts/eight", F_OK) == -1 && errno == ENOENT);
  37         set_sync_flag(&commits_first_flag, !inviter_first);
  38 }
  39
  40 static void inviter_commits_first_cb(bool inviter_first) {
  41         // Check that foo has committed nine's host config file, but nine hasn't committed foo's
  42         assert(access("invite_join_conf.1/current/hosts/nine", F_OK) == 0);
  43         assert(access("invite_join_conf.9/current/hosts/foo", F_OK) == -1 && errno == ENOENT);
  44         set_sync_flag(&commits_first_flag, inviter_first);
  45 }
  46
  47 int main(void) {
  48         init_sync_flag(&baz_reachable);
  49         init_sync_flag(&seven_reachable);
  50         init_sync_flag(&commits_first_flag);
  51
  52         meshlink_set_log_cb(NULL, MESHLINK_DEBUG, log_cb);
  53
  54         assert(meshlink_destroy("invite_join_conf.1"));
  55         assert(meshlink_destroy("invite_join_conf.2"));
  56         assert(meshlink_destroy("invite_join_conf.3"));
  57         assert(meshlink_destroy("invite_join_conf.4"));
  58         assert(meshlink_destroy("invite_join_conf.5"));
  59         assert(meshlink_destroy("invite_join_conf.6"));
  60         assert(meshlink_destroy("invite_join_conf.7"));
  61         assert(meshlink_destroy("invite_join_conf.8"));
  62         assert(meshlink_destroy("invite_join_conf.9"));
  63
  64         // Open thee new meshlink instance.
  65
  66         meshlink_handle_t *mesh1 = meshlink_open("invite_join_conf.1", "foo", "invite-join", DEV_CLASS_BACKBONE);
  67         assert(mesh1);
  68
  69         meshlink_handle_t *mesh2 = meshlink_open("invite_join_conf.2", "bar", "invite-join", DEV_CLASS_BACKBONE);
  70         assert(mesh2);
  71
  72         meshlink_handle_t *mesh3 = meshlink_open("invite_join_conf.3", "quux", "invite-join", DEV_CLASS_BACKBONE);
  73         assert(mesh3);
  74
  75         // Disable local discovery.
  76
  77         meshlink_enable_discovery(mesh1, false);
  78         meshlink_enable_discovery(mesh2, false);
  79         meshlink_enable_discovery(mesh3, false);
  80
  81         // Have the first instance generate invitations.
  82
  83         meshlink_set_node_status_cb(mesh1, status_cb);
  84
  85         assert(meshlink_set_canonical_address(mesh1, meshlink_get_self(mesh1), "localhost", NULL));
  86
  87         char *baz_url = meshlink_invite(mesh1, NULL, "baz");
  88         assert(baz_url);
  89
  90         char *quux_url = meshlink_invite(mesh1, NULL, "quux");
  91         assert(quux_url);
  92
  93         // Check that the second instances cannot join if it is already started
  94
  95         assert(meshlink_start(mesh1));
  96         assert(meshlink_start(mesh2));
  97         meshlink_errno = MESHLINK_OK;
  98         assert(!meshlink_join(mesh2, baz_url));
  99         assert(meshlink_errno = MESHLINK_EINVAL);
 100
 101         // Have the second instance join the first.
 102
 103         meshlink_stop(mesh2);
 104         assert(meshlink_join(mesh2, baz_url));
 105         assert(meshlink_start(mesh2));
 106
 107         // Wait for the two to connect.
 108
 109         assert(wait_sync_flag(&baz_reachable, 20));
 110
 111         // Wait for UDP communication to become possible.
 112
 113         int pmtu = meshlink_get_pmtu(mesh1, meshlink_get_node(mesh1, "baz"));
 114
 115         for(int i = 0; i < 10 && !pmtu; i++) {
 116                 sleep(1);
 117                 pmtu = meshlink_get_pmtu(mesh1, meshlink_get_node(mesh1, "baz"));
 118         }
 119
 120         assert(pmtu);
 121
 122         // Check that an invitation cannot be used twice
 123
 124         assert(!meshlink_join(mesh3, baz_url));
 125         free(baz_url);
 126
 127         // Check that nodes cannot join with expired invitations
 128
 129         meshlink_set_invitation_timeout(mesh1, 0);
 130
 131         assert(!meshlink_join(mesh3, quux_url));
 132         free(quux_url);
 133
 134         // Check that existing nodes cannot join another mesh
 135
 136         char *corge_url = meshlink_invite(mesh3, NULL, "corge");
 137         assert(corge_url);
 138
 139         assert(meshlink_start(mesh3));
 140
 141         meshlink_stop(mesh2);
 142
 143         assert(!meshlink_join(mesh2, corge_url));
 144         free(corge_url);
 145
 146         // Check that invitations work correctly after changing ports
 147
 148         meshlink_set_invitation_timeout(mesh1, 86400);
 149         meshlink_stop(mesh1);
 150         meshlink_stop(mesh3);
 151
 152         int oldport = meshlink_get_port(mesh1);
 153         bool success = false;
 154
 155         for(int i = 0; !success && i < 100; i++) {
 156                 success = meshlink_set_port(mesh1, 0x9000 + rand() % 0x1000);
 157         }
 158
 159         assert(success);
 160         int newport = meshlink_get_port(mesh1);
 161         assert(oldport != newport);
 162
 163         assert(meshlink_start(mesh1));
 164         quux_url = meshlink_invite(mesh1, NULL, "quux");
 165         assert(quux_url);
 166
 167         // The old port should not be in the invitation URL
 168
 169         char portstr[10];
 170         snprintf(portstr, sizeof(portstr), ":%d", oldport);
 171         assert(!strstr(quux_url, portstr));
 172
 173         // The new port should be in the invitation URL
 174
 175         snprintf(portstr, sizeof(portstr), ":%d", newport);
 176         assert(strstr(quux_url, portstr));
 177
 178         // The invitation should work
 179
 180         assert(meshlink_join(mesh3, quux_url));
 181         free(quux_url);
 182
 183         // Check that adding duplicate addresses get removed correctly
 184
 185         assert(meshlink_add_invitation_address(mesh1, "localhost", portstr + 1));
 186         corge_url = meshlink_invite(mesh1, NULL, "corge");
 187         assert(corge_url);
 188         char *localhost = strstr(corge_url, "localhost");
 189         assert(localhost);
 190         assert(!strstr(localhost + 1, "localhost"));
 191         free(corge_url);
 192
 193         // Check that resetting and adding multiple, different invitation address works
 194
 195         meshlink_clear_invitation_addresses(mesh1);
 196         assert(meshlink_add_invitation_address(mesh1, "1.invalid.", "12345"));
 197         assert(meshlink_add_invitation_address(mesh1, "2.invalid.", NULL));
 198         assert(meshlink_add_invitation_address(mesh1, "3.invalid.", NULL));
 199         assert(meshlink_add_invitation_address(mesh1, "4.invalid.", NULL));
 200         assert(meshlink_add_invitation_address(mesh1, "5.invalid.", NULL));
 201         char *grault_url = meshlink_invite(mesh1, NULL, "grault");
 202         assert(grault_url);
 203         localhost = strstr(grault_url, "localhost");
 204         assert(localhost);
 205         char *invalid1 = strstr(grault_url, "1.invalid.:12345");
 206         assert(invalid1);
 207         char *invalid5 = strstr(grault_url, "5.invalid.");
 208         assert(invalid5);
 209
 210         // Check that explicitly added invitation addresses come before others, in the order they were specified.
 211
 212         assert(invalid1 < invalid5);
 213         assert(invalid5 < localhost);
 214         free(grault_url);
 215
 216         // Check inviting nodes into a submesh
 217
 218         assert(!meshlink_get_node_submesh(mesh1, meshlink_get_self(mesh1)));
 219
 220         meshlink_handle_t *mesh4 = meshlink_open("invite_join_conf.4", "four", "invite-join", DEV_CLASS_BACKBONE);
 221         meshlink_handle_t *mesh5 = meshlink_open("invite_join_conf.5", "five", "invite-join", DEV_CLASS_BACKBONE);
 222         meshlink_handle_t *mesh6 = meshlink_open("invite_join_conf.6", "six", "invite-join", DEV_CLASS_BACKBONE);
 223         assert(mesh4);
 224         assert(mesh5);
 225         assert(mesh6);
 226
 227         meshlink_enable_discovery(mesh4, false);
 228         meshlink_enable_discovery(mesh5, false);
 229         meshlink_enable_discovery(mesh6, false);
 230
 231         meshlink_submesh_t *submesh1 = meshlink_submesh_open(mesh1, "submesh1");
 232         meshlink_submesh_t *submesh2 = meshlink_submesh_open(mesh1, "submesh2");
 233         assert(submesh1);
 234         assert(submesh2);
 235
 236         char *four_url = meshlink_invite(mesh1, submesh1, mesh4->name);
 237         char *five_url = meshlink_invite(mesh1, submesh1, mesh5->name);
 238         char *six_url = meshlink_invite(mesh1, submesh2, mesh6->name);
 239         assert(four_url);
 240         assert(five_url);
 241         assert(six_url);
 242
 243         assert(meshlink_join(mesh4, four_url));
 244         assert(meshlink_join(mesh5, five_url));
 245         assert(meshlink_join(mesh6, six_url));
 246
 247         free(four_url);
 248         free(five_url);
 249         free(six_url);
 250
 251         assert(meshlink_start(mesh2));
 252         assert(meshlink_start(mesh4));
 253         assert(meshlink_start(mesh5));
 254         assert(meshlink_start(mesh6));
 255
 256         // Check that each node knows in which submesh it is
 257
 258         meshlink_submesh_t *mesh4_submesh = meshlink_get_node_submesh(mesh4, meshlink_get_self(mesh4));
 259         meshlink_submesh_t *mesh5_submesh = meshlink_get_node_submesh(mesh4, meshlink_get_self(mesh5));
 260         meshlink_submesh_t *mesh6_submesh = meshlink_get_node_submesh(mesh6, meshlink_get_self(mesh6));
 261         assert(mesh4_submesh);
 262         assert(mesh5_submesh);
 263         assert(mesh6_submesh);
 264         assert(!strcmp(mesh4_submesh->name, "submesh1"));
 265         assert(!strcmp(mesh5_submesh->name, "submesh1"));
 266         assert(!strcmp(mesh6_submesh->name, "submesh2"));
 267
 268         // Wait for nodes to connect, and check that foo sees the right submeshes
 269
 270         sleep(2);
 271         meshlink_node_t *mesh1_four = meshlink_get_node(mesh1, mesh4->name);
 272         meshlink_node_t *mesh1_six = meshlink_get_node(mesh1, mesh6->name);
 273         assert(meshlink_get_node_submesh(mesh1, meshlink_get_self(mesh1)) == NULL);
 274         assert(meshlink_get_node_submesh(mesh1, mesh1_four) == submesh1);
 275         assert(meshlink_get_node_submesh(mesh1, mesh1_six) == submesh2);
 276
 277         // Check that the new invitees still have the right submesh information
 278
 279         meshlink_node_t *mesh4_four = meshlink_get_node(mesh4, mesh4->name);
 280         meshlink_node_t *mesh4_five = meshlink_get_node(mesh4, mesh5->name);
 281         meshlink_node_t *mesh6_six = meshlink_get_node(mesh6, mesh6->name);
 282         assert(meshlink_get_node_submesh(mesh4, mesh4_four) == mesh4_submesh);
 283         assert(meshlink_get_node_submesh(mesh4, mesh4_five) == mesh4_submesh);
 284         assert(meshlink_get_node_submesh(mesh6, mesh6_six) == mesh6_submesh);
 285
 286         // Check that bar can see all the nodes in submeshes and vice versa
 287
 288         assert(meshlink_get_node(mesh2, mesh4->name));
 289         assert(meshlink_get_node(mesh2, mesh5->name));
 290         assert(meshlink_get_node(mesh2, mesh6->name));
 291         assert(meshlink_get_node(mesh4, mesh2->name));
 292         assert(meshlink_get_node(mesh5, mesh2->name));
 293         assert(meshlink_get_node(mesh6, mesh2->name));
 294
 295         // Check that four and five can see each other
 296
 297         assert(meshlink_get_node(mesh4, mesh5->name));
 298         assert(meshlink_get_node(mesh5, mesh4->name));
 299
 300         // Check that the nodes in different submeshes cannot see each other
 301
 302         assert(!meshlink_get_node(mesh4, mesh6->name));
 303         assert(!meshlink_get_node(mesh5, mesh6->name));
 304         assert(!meshlink_get_node(mesh6, mesh4->name));
 305         assert(!meshlink_get_node(mesh6, mesh5->name));
 306
 307         // Check that bar sees the right submesh information for the nodes in submeshes
 308
 309         meshlink_submesh_t *mesh2_four_submesh = meshlink_get_node_submesh(mesh2, meshlink_get_node(mesh2, mesh4->name));
 310         meshlink_submesh_t *mesh2_five_submesh = meshlink_get_node_submesh(mesh2, meshlink_get_node(mesh2, mesh5->name));
 311         meshlink_submesh_t *mesh2_six_submesh = meshlink_get_node_submesh(mesh2, meshlink_get_node(mesh2, mesh6->name));
 312         assert(mesh2_four_submesh);
 313         assert(mesh2_five_submesh);
 314         assert(mesh2_six_submesh);
 315         assert(!strcmp(mesh2_four_submesh->name, "submesh1"));
 316         assert(!strcmp(mesh2_five_submesh->name, "submesh1"));
 317         assert(!strcmp(mesh2_six_submesh->name, "submesh2"));
 318
 319         // Test case #2: check invalid parameters
 320
 321         meshlink_handle_t *mesh7 = meshlink_open("invite_join_conf.7", "seven", "invite-join", DEV_CLASS_BACKBONE);
 322         assert(mesh7);
 323         meshlink_enable_discovery(mesh7, false);
 324         char *seven_url = meshlink_invite(mesh1, NULL, "seven");
 325         assert(seven_url);
 326
 327         meshlink_errno = MESHLINK_OK;
 328         assert(!meshlink_invite(NULL, NULL, "seven"));
 329         assert(meshlink_errno == MESHLINK_EINVAL);
 330
 331         meshlink_errno = MESHLINK_OK;
 332         assert(!meshlink_invite(mesh1, NULL, NULL));
 333         assert(meshlink_errno == MESHLINK_EINVAL);
 334
 335         meshlink_errno = MESHLINK_OK;
 336         assert(!meshlink_invite(mesh1, NULL, ""));
 337         assert(meshlink_errno == MESHLINK_EINVAL);
 338
 339         meshlink_errno = MESHLINK_OK;
 340         assert(!meshlink_join(NULL, seven_url));
 341         assert(meshlink_errno == MESHLINK_EINVAL);
 342
 343         meshlink_errno = MESHLINK_OK;
 344         assert(!meshlink_join(mesh7, NULL));
 345         assert(meshlink_errno == MESHLINK_EINVAL);
 346
 347         meshlink_errno = MESHLINK_OK;
 348         assert(!meshlink_join(mesh7, ""));
 349         assert(meshlink_errno == MESHLINK_EINVAL);
 350
 351         // Test case #3 and #4: check persistence of inviter and invitee
 352
 353         assert(meshlink_join(mesh7, seven_url));
 354         free(seven_url);
 355         meshlink_close(mesh1);
 356         meshlink_stop(mesh2);
 357         meshlink_stop(mesh3);
 358         meshlink_stop(mesh4);
 359         meshlink_stop(mesh5);
 360         meshlink_stop(mesh6);
 361         meshlink_close(mesh7);
 362         mesh1 = meshlink_open("invite_join_conf.1", "foo", "invite-join", DEV_CLASS_BACKBONE);
 363         mesh7 = meshlink_open("invite_join_conf.7", "seven", "invite-join", DEV_CLASS_BACKBONE);
 364         assert(mesh1);
 365         assert(mesh7);
 366         meshlink_enable_discovery(mesh1, false);
 367         meshlink_enable_discovery(mesh7, false);
 368         meshlink_set_node_status_cb(mesh1, status_cb);
 369         assert(meshlink_start(mesh1));
 370         assert(meshlink_start(mesh7));
 371         assert(wait_sync_flag(&seven_reachable, 5));
 372         meshlink_stop(mesh7);
 373
 374         // Test case #6 and #7: check invalid inviter_commits_first combinations
 375
 376         meshlink_handle_t *mesh8 = meshlink_open("invite_join_conf.8", "eight", "invite-join", DEV_CLASS_BACKBONE);
 377         assert(mesh8);
 378         meshlink_enable_discovery(mesh8, false);
 379         char *eight_url = meshlink_invite(mesh1, NULL, "eight");
 380         assert(eight_url);
 381         meshlink_set_inviter_commits_first(mesh1, true);
 382         meshlink_set_inviter_commits_first(mesh8, false);
 383         assert(!meshlink_join(mesh8, eight_url));
 384         free(eight_url);
 385
 386         eight_url = meshlink_invite(mesh1, NULL, "eight");
 387         meshlink_set_inviter_commits_first(mesh1, false);
 388         meshlink_set_inviter_commits_first(mesh8, true);
 389         assert(!meshlink_join(mesh8, eight_url));
 390         free(eight_url);
 391
 392         // Test case #5: test invitee committing first scenario
 393
 394         eight_url = meshlink_invite(mesh1, NULL, "eight");
 395         meshlink_set_inviter_commits_first(mesh1, false);
 396         meshlink_set_inviter_commits_first(mesh8, false);
 397         devtool_set_inviter_commits_first = invitee_commits_first_cb;
 398         assert(meshlink_join(mesh8, eight_url));
 399         free(eight_url);
 400         assert(wait_sync_flag(&commits_first_flag, 5));
 401
 402         // Test case #6: test inviter committing first scenario
 403
 404         meshlink_handle_t *mesh9 = meshlink_open("invite_join_conf.9", "nine", "invite-join", DEV_CLASS_BACKBONE);
 405         assert(mesh9);
 406         meshlink_enable_discovery(mesh9, false);
 407         char *nine_url = meshlink_invite(mesh1, NULL, "nine");
 408         meshlink_set_inviter_commits_first(mesh1, true);
 409         meshlink_set_inviter_commits_first(mesh9, true);
 410         devtool_set_inviter_commits_first = inviter_commits_first_cb;
 411         reset_sync_flag(&commits_first_flag);
 412         assert(meshlink_join(mesh9, nine_url));
 413         free(nine_url);
 414         assert(wait_sync_flag(&commits_first_flag, 5));
 415
 416         // Clean up.
 417
 418         meshlink_close(mesh9);
 419         meshlink_close(mesh8);
 420         meshlink_close(mesh7);
 421         meshlink_close(mesh6);
 422         meshlink_close(mesh5);
 423         meshlink_close(mesh4);
 424         meshlink_close(mesh3);
 425         meshlink_close(mesh2);
 426         meshlink_close(mesh1);
 427 }