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_set_canonical_address(mesh1, meshlink_get_self(mesh1), "localhost", NULL));
 164
 165         assert(meshlink_start(mesh1));
 166         quux_url = meshlink_invite(mesh1, NULL, "quux");
 167         assert(quux_url);
 168
 169         // The old port should not be in the invitation URL
 170
 171         char portstr[10];
 172         snprintf(portstr, sizeof(portstr), ":%d", oldport);
 173         assert(!strstr(quux_url, portstr));
 174
 175         // The new port should be in the invitation URL
 176
 177         snprintf(portstr, sizeof(portstr), ":%d", newport);
 178         assert(strstr(quux_url, portstr));
 179
 180         // The invitation should work
 181
 182         assert(meshlink_join(mesh3, quux_url));
 183         free(quux_url);
 184
 185         // Check that adding duplicate addresses get removed correctly
 186
 187         assert(meshlink_add_invitation_address(mesh1, "localhost", portstr + 1));
 188         corge_url = meshlink_invite(mesh1, NULL, "corge");
 189         assert(corge_url);
 190         char *localhost = strstr(corge_url, "localhost");
 191         assert(localhost);
 192         assert(!strstr(localhost + 1, "localhost"));
 193         free(corge_url);
 194
 195         // Check that resetting and adding multiple, different invitation address works
 196
 197         meshlink_clear_invitation_addresses(mesh1);
 198         assert(meshlink_add_invitation_address(mesh1, "1.invalid.", "12345"));
 199         assert(meshlink_add_invitation_address(mesh1, "2.invalid.", NULL));
 200         assert(meshlink_add_invitation_address(mesh1, "3.invalid.", NULL));
 201         assert(meshlink_add_invitation_address(mesh1, "4.invalid.", NULL));
 202         assert(meshlink_add_invitation_address(mesh1, "5.invalid.", NULL));
 203         char *grault_url = meshlink_invite(mesh1, NULL, "grault");
 204         assert(grault_url);
 205         localhost = strstr(grault_url, "localhost");
 206         assert(localhost);
 207         char *invalid1 = strstr(grault_url, "1.invalid.:12345");
 208         assert(invalid1);
 209         char *invalid5 = strstr(grault_url, "5.invalid.");
 210         assert(invalid5);
 211
 212         // Check that explicitly added invitation addresses come before others, in the order they were specified.
 213
 214         assert(invalid1 < invalid5);
 215         assert(invalid5 < localhost);
 216         free(grault_url);
 217
 218         // Check inviting nodes into a submesh
 219
 220         assert(!meshlink_get_node_submesh(mesh1, meshlink_get_self(mesh1)));
 221
 222         meshlink_handle_t *mesh4 = meshlink_open("invite_join_conf.4", "four", "invite-join", DEV_CLASS_BACKBONE);
 223         meshlink_handle_t *mesh5 = meshlink_open("invite_join_conf.5", "five", "invite-join", DEV_CLASS_BACKBONE);
 224         meshlink_handle_t *mesh6 = meshlink_open("invite_join_conf.6", "six", "invite-join", DEV_CLASS_BACKBONE);
 225         assert(mesh4);
 226         assert(mesh5);
 227         assert(mesh6);
 228
 229         meshlink_enable_discovery(mesh4, false);
 230         meshlink_enable_discovery(mesh5, false);
 231         meshlink_enable_discovery(mesh6, false);
 232
 233         meshlink_submesh_t *submesh1 = meshlink_submesh_open(mesh1, "submesh1");
 234         meshlink_submesh_t *submesh2 = meshlink_submesh_open(mesh1, "submesh2");
 235         assert(submesh1);
 236         assert(submesh2);
 237
 238         char *four_url = meshlink_invite(mesh1, submesh1, mesh4->name);
 239         char *five_url = meshlink_invite(mesh1, submesh1, mesh5->name);
 240         char *six_url = meshlink_invite(mesh1, submesh2, mesh6->name);
 241         assert(four_url);
 242         assert(five_url);
 243         assert(six_url);
 244
 245         assert(meshlink_join(mesh4, four_url));
 246         assert(meshlink_join(mesh5, five_url));
 247         assert(meshlink_join(mesh6, six_url));
 248
 249         free(four_url);
 250         free(five_url);
 251         free(six_url);
 252
 253         assert(meshlink_start(mesh2));
 254         assert(meshlink_start(mesh4));
 255         assert(meshlink_start(mesh5));
 256         assert(meshlink_start(mesh6));
 257
 258         // Check that each node knows in which submesh it is
 259
 260         meshlink_submesh_t *mesh4_submesh = meshlink_get_node_submesh(mesh4, meshlink_get_self(mesh4));
 261         meshlink_submesh_t *mesh5_submesh = meshlink_get_node_submesh(mesh4, meshlink_get_self(mesh5));
 262         meshlink_submesh_t *mesh6_submesh = meshlink_get_node_submesh(mesh6, meshlink_get_self(mesh6));
 263         assert(mesh4_submesh);
 264         assert(mesh5_submesh);
 265         assert(mesh6_submesh);
 266         assert(!strcmp(mesh4_submesh->name, "submesh1"));
 267         assert(!strcmp(mesh5_submesh->name, "submesh1"));
 268         assert(!strcmp(mesh6_submesh->name, "submesh2"));
 269
 270         // Wait for nodes to connect, and check that foo sees the right submeshes
 271
 272         sleep(2);
 273         meshlink_node_t *mesh1_four = meshlink_get_node(mesh1, mesh4->name);
 274         meshlink_node_t *mesh1_six = meshlink_get_node(mesh1, mesh6->name);
 275         assert(meshlink_get_node_submesh(mesh1, meshlink_get_self(mesh1)) == NULL);
 276         assert(meshlink_get_node_submesh(mesh1, mesh1_four) == submesh1);
 277         assert(meshlink_get_node_submesh(mesh1, mesh1_six) == submesh2);
 278
 279         // Check that the new invitees still have the right submesh information
 280
 281         meshlink_node_t *mesh4_four = meshlink_get_node(mesh4, mesh4->name);
 282         meshlink_node_t *mesh4_five = meshlink_get_node(mesh4, mesh5->name);
 283         meshlink_node_t *mesh6_six = meshlink_get_node(mesh6, mesh6->name);
 284         assert(meshlink_get_node_submesh(mesh4, mesh4_four) == mesh4_submesh);
 285         assert(meshlink_get_node_submesh(mesh4, mesh4_five) == mesh4_submesh);
 286         assert(meshlink_get_node_submesh(mesh6, mesh6_six) == mesh6_submesh);
 287
 288         // Check that bar can see all the nodes in submeshes and vice versa
 289
 290         assert(meshlink_get_node(mesh2, mesh4->name));
 291         assert(meshlink_get_node(mesh2, mesh5->name));
 292         assert(meshlink_get_node(mesh2, mesh6->name));
 293         assert(meshlink_get_node(mesh4, mesh2->name));
 294         assert(meshlink_get_node(mesh5, mesh2->name));
 295         assert(meshlink_get_node(mesh6, mesh2->name));
 296
 297         // Check that four and five can see each other
 298
 299         assert(meshlink_get_node(mesh4, mesh5->name));
 300         assert(meshlink_get_node(mesh5, mesh4->name));
 301
 302         // Check that the nodes in different submeshes cannot see each other
 303
 304         assert(!meshlink_get_node(mesh4, mesh6->name));
 305         assert(!meshlink_get_node(mesh5, mesh6->name));
 306         assert(!meshlink_get_node(mesh6, mesh4->name));
 307         assert(!meshlink_get_node(mesh6, mesh5->name));
 308
 309         // Check that bar sees the right submesh information for the nodes in submeshes
 310
 311         meshlink_submesh_t *mesh2_four_submesh = meshlink_get_node_submesh(mesh2, meshlink_get_node(mesh2, mesh4->name));
 312         meshlink_submesh_t *mesh2_five_submesh = meshlink_get_node_submesh(mesh2, meshlink_get_node(mesh2, mesh5->name));
 313         meshlink_submesh_t *mesh2_six_submesh = meshlink_get_node_submesh(mesh2, meshlink_get_node(mesh2, mesh6->name));
 314         assert(mesh2_four_submesh);
 315         assert(mesh2_five_submesh);
 316         assert(mesh2_six_submesh);
 317         assert(!strcmp(mesh2_four_submesh->name, "submesh1"));
 318         assert(!strcmp(mesh2_five_submesh->name, "submesh1"));
 319         assert(!strcmp(mesh2_six_submesh->name, "submesh2"));
 320
 321         // Test case #2: check invalid parameters
 322
 323         meshlink_handle_t *mesh7 = meshlink_open("invite_join_conf.7", "seven", "invite-join", DEV_CLASS_BACKBONE);
 324         assert(mesh7);
 325         meshlink_enable_discovery(mesh7, false);
 326         char *seven_url = meshlink_invite(mesh1, NULL, "seven");
 327         assert(seven_url);
 328
 329         meshlink_errno = MESHLINK_OK;
 330         assert(!meshlink_invite(NULL, NULL, "seven"));
 331         assert(meshlink_errno == MESHLINK_EINVAL);
 332
 333         meshlink_errno = MESHLINK_OK;
 334         assert(!meshlink_invite(mesh1, NULL, NULL));
 335         assert(meshlink_errno == MESHLINK_EINVAL);
 336
 337         meshlink_errno = MESHLINK_OK;
 338         assert(!meshlink_invite(mesh1, NULL, ""));
 339         assert(meshlink_errno == MESHLINK_EINVAL);
 340
 341         meshlink_errno = MESHLINK_OK;
 342         assert(!meshlink_join(NULL, seven_url));
 343         assert(meshlink_errno == MESHLINK_EINVAL);
 344
 345         meshlink_errno = MESHLINK_OK;
 346         assert(!meshlink_join(mesh7, NULL));
 347         assert(meshlink_errno == MESHLINK_EINVAL);
 348
 349         meshlink_errno = MESHLINK_OK;
 350         assert(!meshlink_join(mesh7, ""));
 351         assert(meshlink_errno == MESHLINK_EINVAL);
 352
 353         // Test case #3 and #4: check persistence of inviter and invitee
 354
 355         assert(meshlink_join(mesh7, seven_url));
 356         free(seven_url);
 357         meshlink_close(mesh1);
 358         meshlink_stop(mesh2);
 359         meshlink_stop(mesh3);
 360         meshlink_stop(mesh4);
 361         meshlink_stop(mesh5);
 362         meshlink_stop(mesh6);
 363         meshlink_close(mesh7);
 364         mesh1 = meshlink_open("invite_join_conf.1", "foo", "invite-join", DEV_CLASS_BACKBONE);
 365         mesh7 = meshlink_open("invite_join_conf.7", "seven", "invite-join", DEV_CLASS_BACKBONE);
 366         assert(mesh1);
 367         assert(mesh7);
 368         meshlink_enable_discovery(mesh1, false);
 369         meshlink_enable_discovery(mesh7, false);
 370         meshlink_set_node_status_cb(mesh1, status_cb);
 371         assert(meshlink_start(mesh1));
 372         assert(meshlink_start(mesh7));
 373         assert(wait_sync_flag(&seven_reachable, 5));
 374         meshlink_stop(mesh7);
 375
 376         // Test case #6 and #7: check invalid inviter_commits_first combinations
 377
 378         meshlink_handle_t *mesh8 = meshlink_open("invite_join_conf.8", "eight", "invite-join", DEV_CLASS_BACKBONE);
 379         assert(mesh8);
 380         meshlink_enable_discovery(mesh8, false);
 381         char *eight_url = meshlink_invite(mesh1, NULL, "eight");
 382         assert(eight_url);
 383         meshlink_set_inviter_commits_first(mesh1, true);
 384         meshlink_set_inviter_commits_first(mesh8, false);
 385         assert(!meshlink_join(mesh8, eight_url));
 386         free(eight_url);
 387
 388         eight_url = meshlink_invite(mesh1, NULL, "eight");
 389         meshlink_set_inviter_commits_first(mesh1, false);
 390         meshlink_set_inviter_commits_first(mesh8, true);
 391         assert(!meshlink_join(mesh8, eight_url));
 392         free(eight_url);
 393
 394         // Test case #5: test invitee committing first scenario
 395
 396         eight_url = meshlink_invite(mesh1, NULL, "eight");
 397         meshlink_set_inviter_commits_first(mesh1, false);
 398         meshlink_set_inviter_commits_first(mesh8, false);
 399         devtool_set_inviter_commits_first = invitee_commits_first_cb;
 400         assert(meshlink_join(mesh8, eight_url));
 401         free(eight_url);
 402         assert(wait_sync_flag(&commits_first_flag, 5));
 403
 404         // Test case #6: test inviter committing first scenario
 405
 406         meshlink_handle_t *mesh9 = meshlink_open("invite_join_conf.9", "nine", "invite-join", DEV_CLASS_BACKBONE);
 407         assert(mesh9);
 408         meshlink_enable_discovery(mesh9, false);
 409         char *nine_url = meshlink_invite(mesh1, NULL, "nine");
 410         meshlink_set_inviter_commits_first(mesh1, true);
 411         meshlink_set_inviter_commits_first(mesh9, true);
 412         devtool_set_inviter_commits_first = inviter_commits_first_cb;
 413         reset_sync_flag(&commits_first_flag);
 414         assert(meshlink_join(mesh9, nine_url));
 415         free(nine_url);
 416         assert(wait_sync_flag(&commits_first_flag, 5));
 417
 418         // Clean up.
 419
 420         meshlink_close(mesh9);
 421         meshlink_close(mesh8);
 422         meshlink_close(mesh7);
 423         meshlink_close(mesh6);
 424         meshlink_close(mesh5);
 425         meshlink_close(mesh4);
 426         meshlink_close(mesh3);
 427         meshlink_close(mesh2);
 428         meshlink_close(mesh1);
 429 }