git.meshlink.io Git - meshlink/blob - src/meshlink.c

   1 /*
   2     meshlink.c -- Implementation of the MeshLink API.
   3     Copyright (C) 2014-2018 Guus Sliepen <guus@meshlink.io>
   4
   5     This program is free software; you can redistribute it and/or modify
   6     it under the terms of the GNU General Public License as published by
   7     the Free Software Foundation; either version 2 of the License, or
   8     (at your option) any later version.
   9
  10     This program is distributed in the hope that it will be useful,
  11     but WITHOUT ANY WARRANTY; without even the implied warranty of
  12     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13     GNU General Public License for more details.
  14
  15     You should have received a copy of the GNU General Public License along
  16     with this program; if not, write to the Free Software Foundation, Inc.,
  17     51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  18 */
  19
  20 #include "system.h"
  21 #include <pthread.h>
  22
  23 #include "adns.h"
  24 #include "crypto.h"
  25 #include "ecdsagen.h"
  26 #include "logger.h"
  27 #include "meshlink_internal.h"
  28 #include "net.h"
  29 #include "netutl.h"
  30 #include "node.h"
  31 #include "submesh.h"
  32 #include "packmsg.h"
  33 #include "prf.h"
  34 #include "protocol.h"
  35 #include "route.h"
  36 #include "sockaddr.h"
  37 #include "utils.h"
  38 #include "xalloc.h"
  39 #include "ed25519/sha512.h"
  40 #include "discovery.h"
  41 #include "devtools.h"
  42 #include "graph.h"
  43
  44 #ifndef MSG_NOSIGNAL
  45 #define MSG_NOSIGNAL 0
  46 #endif
  47 __thread meshlink_errno_t meshlink_errno;
  48 meshlink_log_cb_t global_log_cb;
  49 meshlink_log_level_t global_log_level;
  50
  51 typedef bool (*search_node_by_condition_t)(const node_t *, const void *);
  52
  53 static int rstrip(char *value) {
  54         int len = strlen(value);
  55
  56         while(len && strchr("\t\r\n ", value[len - 1])) {
  57                 value[--len] = 0;
  58         }
  59
  60         return len;
  61 }
  62
  63 static void get_canonical_address(node_t *n, char **hostname, char **port) {
  64         if(!n->canonical_address) {
  65                 return;
  66         }
  67
  68         *hostname = xstrdup(n->canonical_address);
  69         char *space = strchr(*hostname, ' ');
  70
  71         if(space) {
  72                 *space++ = 0;
  73                 *port = xstrdup(space);
  74         }
  75 }
  76
  77 static bool is_valid_hostname(const char *hostname) {
  78         if(!*hostname) {
  79                 return false;
  80         }
  81
  82         for(const char *p = hostname; *p; p++) {
  83                 if(!(isalnum(*p) || *p == '-' || *p == '.' || *p == ':')) {
  84                         return false;
  85                 }
  86         }
  87
  88         return true;
  89 }
  90
  91 static bool is_valid_port(const char *port) {
  92         if(!*port) {
  93                 return false;
  94         }
  95
  96         if(isdigit(*port)) {
  97                 char *end;
  98                 unsigned long int result = strtoul(port, &end, 10);
  99                 return result && result < 65536 && !*end;
 100         }
 101
 102         for(const char *p = port; *p; p++) {
 103                 if(!(isalnum(*p) || *p == '-')) {
 104                         return false;
 105                 }
 106         }
 107
 108         return true;
 109 }
 110
 111 static void set_timeout(int sock, int timeout) {
 112 #ifdef _WIN32
 113         DWORD tv = timeout;
 114 #else
 115         struct timeval tv;
 116         tv.tv_sec = timeout / 1000;
 117         tv.tv_usec = (timeout - tv.tv_sec * 1000) * 1000;
 118 #endif
 119         setsockopt(sock, SOL_SOCKET, SO_RCVTIMEO, &tv, sizeof(tv));
 120         setsockopt(sock, SOL_SOCKET, SO_SNDTIMEO, &tv, sizeof(tv));
 121 }
 122
 123 struct socket_in_netns_params {
 124         int domain;
 125         int type;
 126         int protocol;
 127         int netns;
 128         int fd;
 129 };
 130
 131 #ifdef HAVE_SETNS
 132 static void *socket_in_netns_thread(void *arg) {
 133         struct socket_in_netns_params *params = arg;
 134
 135         if(setns(params->netns, CLONE_NEWNET) == -1) {
 136                 meshlink_errno = MESHLINK_EINVAL;
 137                 return NULL;
 138         }
 139
 140         params->fd = socket(params->domain, params->type, params->protocol);
 141
 142         return NULL;
 143 }
 144 #endif // HAVE_SETNS
 145
 146 static int socket_in_netns(int domain, int type, int protocol, int netns) {
 147         if(netns == -1) {
 148                 return socket(domain, type, protocol);
 149         }
 150
 151 #ifdef HAVE_SETNS
 152         struct socket_in_netns_params params = {domain, type, protocol, netns, -1};
 153
 154         pthread_t thr;
 155
 156         if(pthread_create(&thr, NULL, socket_in_netns_thread, &params) == 0) {
 157                 assert(pthread_join(thr, NULL) == 0);
 158         }
 159
 160         return params.fd;
 161 #else
 162         return -1;
 163 #endif // HAVE_SETNS
 164
 165 }
 166
 167 // Find out what local address a socket would use if we connect to the given address.
 168 // We do this using connect() on a UDP socket, so the kernel has to resolve the address
 169 // of both endpoints, but this will actually not send any UDP packet.
 170 static bool getlocaladdr(const char *destaddr, sockaddr_t *sa, socklen_t *salen, int netns) {
 171         struct addrinfo *rai = NULL;
 172         const struct addrinfo hint = {
 173                 .ai_family = AF_UNSPEC,
 174                 .ai_socktype = SOCK_DGRAM,
 175                 .ai_protocol = IPPROTO_UDP,
 176                 .ai_flags = AI_NUMERICHOST | AI_NUMERICSERV,
 177         };
 178
 179         if(getaddrinfo(destaddr, "80", &hint, &rai) || !rai) {
 180                 return false;
 181         }
 182
 183         int sock = socket_in_netns(rai->ai_family, rai->ai_socktype, rai->ai_protocol, netns);
 184
 185         if(sock == -1) {
 186                 freeaddrinfo(rai);
 187                 return false;
 188         }
 189
 190         if(connect(sock, rai->ai_addr, rai->ai_addrlen) && !sockwouldblock(errno)) {
 191                 closesocket(sock);
 192                 freeaddrinfo(rai);
 193                 return false;
 194         }
 195
 196         freeaddrinfo(rai);
 197
 198         if(getsockname(sock, &sa->sa, salen)) {
 199                 closesocket(sock);
 200                 return false;
 201         }
 202
 203         closesocket(sock);
 204         return true;
 205 }
 206
 207 static bool getlocaladdrname(const char *destaddr, char *host, socklen_t hostlen, int netns) {
 208         sockaddr_t sa;
 209         socklen_t salen = sizeof(sa);
 210
 211         if(!getlocaladdr(destaddr, &sa, &salen, netns)) {
 212                 return false;
 213         }
 214
 215         if(getnameinfo(&sa.sa, salen, host, hostlen, NULL, 0, NI_NUMERICHOST | NI_NUMERICSERV)) {
 216                 return false;
 217         }
 218
 219         return true;
 220 }
 221
 222 char *meshlink_get_external_address(meshlink_handle_t *mesh) {
 223         return meshlink_get_external_address_for_family(mesh, AF_UNSPEC);
 224 }
 225
 226 char *meshlink_get_external_address_for_family(meshlink_handle_t *mesh, int family) {
 227         const char *url = mesh->external_address_url;
 228
 229         if(!url) {
 230                 url = "http://meshlink.io/host.cgi";
 231         }
 232
 233         /* Find the hostname part between the slashes */
 234         if(strncmp(url, "http://", 7)) {
 235                 abort();
 236                 meshlink_errno = MESHLINK_EINTERNAL;
 237                 return NULL;
 238         }
 239
 240         const char *begin = url + 7;
 241
 242         const char *end = strchr(begin, '/');
 243
 244         if(!end) {
 245                 end = begin + strlen(begin);
 246         }
 247
 248         /* Make a copy */
 249         char host[end - begin + 1];
 250         strncpy(host, begin, end - begin);
 251         host[end - begin] = 0;
 252
 253         char *port = strchr(host, ':');
 254
 255         if(port) {
 256                 *port++ = 0;
 257         }
 258
 259         logger(mesh, MESHLINK_DEBUG, "Trying to discover externally visible hostname...\n");
 260         struct addrinfo *ai = adns_blocking_request(mesh, xstrdup(host), xstrdup(port ? port : "80"), SOCK_STREAM, 5);
 261         char line[256];
 262         char *hostname = NULL;
 263
 264         for(struct addrinfo *aip = ai; aip; aip = aip->ai_next) {
 265                 if(family != AF_UNSPEC && aip->ai_family != family) {
 266                         continue;
 267                 }
 268
 269                 int s = socket_in_netns(aip->ai_family, aip->ai_socktype, aip->ai_protocol, mesh->netns);
 270
 271                 if(s >= 0) {
 272                         set_timeout(s, 5000);
 273
 274                         if(connect(s, aip->ai_addr, aip->ai_addrlen)) {
 275                                 closesocket(s);
 276                                 s = -1;
 277                         }
 278                 }
 279
 280                 if(s >= 0) {
 281                         send(s, "GET ", 4, 0);
 282                         send(s, url, strlen(url), 0);
 283                         send(s, " HTTP/1.0\r\n\r\n", 13, 0);
 284                         int len = recv(s, line, sizeof(line) - 1, MSG_WAITALL);
 285
 286                         if(len > 0) {
 287                                 line[len] = 0;
 288
 289                                 if(line[len - 1] == '\n') {
 290                                         line[--len] = 0;
 291                                 }
 292
 293                                 char *p = strrchr(line, '\n');
 294
 295                                 if(p && p[1]) {
 296                                         hostname = xstrdup(p + 1);
 297                                 }
 298                         }
 299
 300                         closesocket(s);
 301
 302                         if(hostname) {
 303                                 break;
 304                         }
 305                 }
 306         }
 307
 308         if(ai) {
 309                 freeaddrinfo(ai);
 310         }
 311
 312         // Check that the hostname is reasonable
 313         if(hostname && !is_valid_hostname(hostname)) {
 314                 free(hostname);
 315                 hostname = NULL;
 316         }
 317
 318         if(!hostname) {
 319                 meshlink_errno = MESHLINK_ERESOLV;
 320         }
 321
 322         return hostname;
 323 }
 324
 325 static bool is_localaddr(sockaddr_t *sa) {
 326         switch(sa->sa.sa_family) {
 327         case AF_INET:
 328                 return *(uint8_t *)(&sa->in.sin_addr.s_addr) == 127;
 329
 330         case AF_INET6: {
 331                 uint16_t first = sa->in6.sin6_addr.s6_addr[0] << 8 | sa->in6.sin6_addr.s6_addr[1];
 332                 return first == 0 || (first & 0xffc0) == 0xfe80;
 333         }
 334
 335         default:
 336                 return false;
 337         }
 338 }
 339
 340 char *meshlink_get_local_address_for_family(meshlink_handle_t *mesh, int family) {
 341         (void)mesh;
 342
 343         // Determine address of the local interface used for outgoing connections.
 344         char localaddr[NI_MAXHOST];
 345         bool success = false;
 346
 347         if(family == AF_INET) {
 348                 success = getlocaladdrname("93.184.216.34", localaddr, sizeof(localaddr), mesh->netns);
 349         } else if(family == AF_INET6) {
 350                 success = getlocaladdrname("2606:2800:220:1:248:1893:25c8:1946", localaddr, sizeof(localaddr), mesh->netns);
 351         }
 352
 353 #ifdef HAVE_GETIFADDRS
 354
 355         if(!success) {
 356                 struct ifaddrs *ifa = NULL;
 357                 getifaddrs(&ifa);
 358
 359                 for(struct ifaddrs *ifap = ifa; ifap; ifap = ifap->ifa_next) {
 360                         sockaddr_t *sa = (sockaddr_t *)ifap->ifa_addr;
 361
 362                         if(sa->sa.sa_family != family) {
 363                                 continue;
 364                         }
 365
 366                         if(is_localaddr(sa)) {
 367                                 continue;
 368                         }
 369
 370                         if(!getnameinfo(&sa->sa, SALEN(sa->sa), localaddr, sizeof(localaddr), NULL, 0, NI_NUMERICHOST | NI_NUMERICSERV)) {
 371                                 success = true;
 372                                 break;
 373                         }
 374                 }
 375
 376                 freeifaddrs(ifa);
 377         }
 378
 379 #endif
 380
 381         if(!success) {
 382                 meshlink_errno = MESHLINK_ENETWORK;
 383                 return NULL;
 384         }
 385
 386         return xstrdup(localaddr);
 387 }
 388
 389 static void remove_duplicate_hostnames(char *host[], char *port[], int n) {
 390         for(int i = 0; i < n; i++) {
 391                 if(!host[i]) {
 392                         continue;
 393                 }
 394
 395                 // Ignore duplicate hostnames
 396                 bool found = false;
 397
 398                 for(int j = 0; j < i; j++) {
 399                         if(!host[j]) {
 400                                 continue;
 401                         }
 402
 403                         if(strcmp(host[i], host[j])) {
 404                                 continue;
 405                         }
 406
 407                         if(strcmp(port[i], port[j])) {
 408                                 continue;
 409                         }
 410
 411                         found = true;
 412                         break;
 413                 }
 414
 415                 if(found || !is_valid_hostname(host[i])) {
 416                         free(host[i]);
 417                         free(port[i]);
 418                         host[i] = NULL;
 419                         port[i] = NULL;
 420                         continue;
 421                 }
 422         }
 423 }
 424
 425 // This gets the hostname part for use in invitation URLs
 426 static char *get_my_hostname(meshlink_handle_t *mesh, uint32_t flags) {
 427         int count = 4 + (mesh->invitation_addresses ? mesh->invitation_addresses->count : 0);
 428         int n = 0;
 429         char *hostname[count];
 430         char *port[count];
 431         char *hostport = NULL;
 432
 433         memset(hostname, 0, sizeof(hostname));
 434         memset(port, 0, sizeof(port));
 435
 436         if(!(flags & (MESHLINK_INVITE_LOCAL | MESHLINK_INVITE_PUBLIC))) {
 437                 flags |= MESHLINK_INVITE_LOCAL | MESHLINK_INVITE_PUBLIC;
 438         }
 439
 440         if(!(flags & (MESHLINK_INVITE_IPV4 | MESHLINK_INVITE_IPV6))) {
 441                 flags |= MESHLINK_INVITE_IPV4 | MESHLINK_INVITE_IPV6;
 442         }
 443
 444         // Add all explicitly set invitation addresses
 445         if(mesh->invitation_addresses) {
 446                 for list_each(char, combo, mesh->invitation_addresses) {
 447                         hostname[n] = xstrdup(combo);
 448                         char *slash = strrchr(hostname[n], '/');
 449
 450                         if(slash) {
 451                                 *slash = 0;
 452                                 port[n] = xstrdup(slash + 1);
 453                         }
 454
 455                         n++;
 456                 }
 457         }
 458
 459         // Add local addresses if requested
 460         if(flags & MESHLINK_INVITE_LOCAL) {
 461                 if(flags & MESHLINK_INVITE_IPV4) {
 462                         hostname[n++] = meshlink_get_local_address_for_family(mesh, AF_INET);
 463                 }
 464
 465                 if(flags & MESHLINK_INVITE_IPV6) {
 466                         hostname[n++] = meshlink_get_local_address_for_family(mesh, AF_INET6);
 467                 }
 468         }
 469
 470         // Add public/canonical addresses if requested
 471         if(flags & MESHLINK_INVITE_PUBLIC) {
 472                 // Try the CanonicalAddress first
 473                 get_canonical_address(mesh->self, &hostname[n], &port[n]);
 474
 475                 if(!hostname[n] && count == 4) {
 476                         if(flags & MESHLINK_INVITE_IPV4) {
 477                                 hostname[n++] = meshlink_get_external_address_for_family(mesh, AF_INET);
 478                         }
 479
 480                         if(flags & MESHLINK_INVITE_IPV6) {
 481                                 hostname[n++] = meshlink_get_external_address_for_family(mesh, AF_INET6);
 482                         }
 483                 } else {
 484                         n++;
 485                 }
 486         }
 487
 488         for(int i = 0; i < n; i++) {
 489                 // Ensure we always have a port number
 490                 if(hostname[i] && !port[i]) {
 491                         port[i] = xstrdup(mesh->myport);
 492                 }
 493         }
 494
 495         remove_duplicate_hostnames(hostname, port, n);
 496
 497         // Resolve the hostnames
 498         for(int i = 0; i < n; i++) {
 499                 if(!hostname[i]) {
 500                         continue;
 501                 }
 502
 503                 // Convert what we have to a sockaddr
 504                 struct addrinfo *ai_in = adns_blocking_request(mesh, xstrdup(hostname[i]), xstrdup(port[i]), SOCK_STREAM, 5);
 505
 506                 if(!ai_in) {
 507                         continue;
 508                 }
 509
 510                 // Remember the address(es)
 511                 for(struct addrinfo *aip = ai_in; aip; aip = aip->ai_next) {
 512                         node_add_recent_address(mesh, mesh->self, (sockaddr_t *)aip->ai_addr);
 513                 }
 514
 515                 freeaddrinfo(ai_in);
 516                 continue;
 517         }
 518
 519         // Remove duplicates again, since IPv4 and IPv6 addresses might map to the same hostname
 520         remove_duplicate_hostnames(hostname, port, n);
 521
 522         // Concatenate all unique address to the hostport string
 523         for(int i = 0; i < n; i++) {
 524                 if(!hostname[i]) {
 525                         continue;
 526                 }
 527
 528                 // Append the address to the hostport string
 529                 char *newhostport;
 530                 xasprintf(&newhostport, (strchr(hostname[i], ':') ? "%s%s[%s]:%s" : "%s%s%s:%s"), hostport ? hostport : "", hostport ? "," : "", hostname[i], port[i]);
 531                 free(hostport);
 532                 hostport = newhostport;
 533
 534                 free(hostname[i]);
 535                 free(port[i]);
 536         }
 537
 538         return hostport;
 539 }
 540
 541 static bool try_bind(meshlink_handle_t *mesh, int port) {
 542         struct addrinfo *ai = NULL;
 543         struct addrinfo hint = {
 544                 .ai_flags = AI_PASSIVE,
 545                 .ai_family = AF_UNSPEC,
 546                 .ai_socktype = SOCK_STREAM,
 547                 .ai_protocol = IPPROTO_TCP,
 548         };
 549
 550         char portstr[16];
 551         snprintf(portstr, sizeof(portstr), "%d", port);
 552
 553         if(getaddrinfo(NULL, portstr, &hint, &ai) || !ai) {
 554                 return false;
 555         }
 556
 557         bool success = false;
 558
 559         for(struct addrinfo *aip = ai; aip; aip = aip->ai_next) {
 560                 /* Try to bind to TCP. */
 561
 562                 int tcp_fd = setup_tcp_listen_socket(mesh, aip);
 563
 564                 if(tcp_fd == -1) {
 565                         if(errno == EADDRINUSE) {
 566                                 /* If this port is in use for any address family, avoid it. */
 567                                 success = false;
 568                                 break;
 569                         } else {
 570                                 continue;
 571                         }
 572                 }
 573
 574                 /* If TCP worked, then we require that UDP works as well. */
 575
 576                 int udp_fd = setup_udp_listen_socket(mesh, aip);
 577
 578                 if(udp_fd == -1) {
 579                         closesocket(tcp_fd);
 580                         success = false;
 581                         break;
 582                 }
 583
 584                 closesocket(tcp_fd);
 585                 closesocket(udp_fd);
 586                 success = true;
 587         }
 588
 589         freeaddrinfo(ai);
 590         return success;
 591 }
 592
 593 int check_port(meshlink_handle_t *mesh) {
 594         for(int i = 0; i < 1000; i++) {
 595                 int port = 0x1000 + prng(mesh, 0x8000);
 596
 597                 if(try_bind(mesh, port)) {
 598                         free(mesh->myport);
 599                         xasprintf(&mesh->myport, "%d", port);
 600                         return port;
 601                 }
 602         }
 603
 604         meshlink_errno = MESHLINK_ENETWORK;
 605         logger(mesh, MESHLINK_DEBUG, "Could not find any available network port.\n");
 606         return 0;
 607 }
 608
 609 static bool write_main_config_files(meshlink_handle_t *mesh) {
 610         if(!mesh->confbase) {
 611                 return true;
 612         }
 613
 614         uint8_t buf[4096];
 615
 616         /* Write the main config file */
 617         packmsg_output_t out = {buf, sizeof buf};
 618
 619         packmsg_add_uint32(&out, MESHLINK_CONFIG_VERSION);
 620         packmsg_add_str(&out, mesh->name);
 621         packmsg_add_bin(&out, ecdsa_get_private_key(mesh->private_key), 96);
 622         packmsg_add_bin(&out, ecdsa_get_private_key(mesh->invitation_key), 96);
 623         packmsg_add_uint16(&out, atoi(mesh->myport));
 624
 625         if(!packmsg_output_ok(&out)) {
 626                 return false;
 627         }
 628
 629         config_t config = {buf, packmsg_output_size(&out, buf)};
 630
 631         if(!main_config_write(mesh, "current", &config, mesh->config_key)) {
 632                 return false;
 633         }
 634
 635         /* Write our own host config file */
 636         if(!node_write_config(mesh, mesh->self)) {
 637                 return false;
 638         }
 639
 640         return true;
 641 }
 642
 643 typedef struct {
 644         meshlink_handle_t *mesh;
 645         int sock;
 646         char cookie[18 + 32];
 647         char hash[18];
 648         bool success;
 649         sptps_t sptps;
 650         char *data;
 651         size_t thedatalen;
 652         size_t blen;
 653         char line[4096];
 654         char buffer[4096];
 655 } join_state_t;
 656
 657 static bool finalize_join(join_state_t *state, const void *buf, uint16_t len) {
 658         meshlink_handle_t *mesh = state->mesh;
 659         packmsg_input_t in = {buf, len};
 660         uint32_t version = packmsg_get_uint32(&in);
 661
 662         if(version != MESHLINK_INVITATION_VERSION) {
 663                 logger(mesh, MESHLINK_ERROR, "Invalid invitation version!\n");
 664                 return false;
 665         }
 666
 667         char *name = packmsg_get_str_dup(&in);
 668         char *submesh_name = packmsg_get_str_dup(&in);
 669         dev_class_t devclass = packmsg_get_int32(&in);
 670         uint32_t count = packmsg_get_array(&in);
 671
 672         if(!name || !check_id(name)) {
 673                 logger(mesh, MESHLINK_DEBUG, "No valid Name found in invitation!\n");
 674                 free(name);
 675                 free(submesh_name);
 676                 return false;
 677         }
 678
 679         if(!submesh_name || (strcmp(submesh_name, CORE_MESH) && !check_id(submesh_name))) {
 680                 logger(mesh, MESHLINK_DEBUG, "No valid Submesh found in invitation!\n");
 681                 free(name);
 682                 free(submesh_name);
 683                 return false;
 684         }
 685
 686         if(!count) {
 687                 logger(mesh, MESHLINK_ERROR, "Incomplete invitation file!\n");
 688                 free(name);
 689                 free(submesh_name);
 690                 return false;
 691         }
 692
 693         free(mesh->name);
 694         free(mesh->self->name);
 695         mesh->name = name;
 696         mesh->self->name = xstrdup(name);
 697         mesh->self->submesh = strcmp(submesh_name, CORE_MESH) ? lookup_or_create_submesh(mesh, submesh_name) : NULL;
 698         free(submesh_name);
 699         mesh->self->devclass = devclass == DEV_CLASS_UNKNOWN ? mesh->devclass : devclass;
 700
 701         // Initialize configuration directory
 702         if(!config_init(mesh, "current")) {
 703                 return false;
 704         }
 705
 706         if(!write_main_config_files(mesh)) {
 707                 return false;
 708         }
 709
 710         // Write host config files
 711         for(uint32_t i = 0; i < count; i++) {
 712                 const void *data;
 713                 uint32_t data_len = packmsg_get_bin_raw(&in, &data);
 714
 715                 if(!data_len) {
 716                         logger(mesh, MESHLINK_ERROR, "Incomplete invitation file!\n");
 717                         return false;
 718                 }
 719
 720                 packmsg_input_t in2 = {data, data_len};
 721                 uint32_t version2 = packmsg_get_uint32(&in2);
 722                 char *name2 = packmsg_get_str_dup(&in2);
 723
 724                 if(!packmsg_input_ok(&in2) || version2 != MESHLINK_CONFIG_VERSION || !check_id(name2)) {
 725                         free(name2);
 726                         packmsg_input_invalidate(&in);
 727                         break;
 728                 }
 729
 730                 if(!check_id(name2)) {
 731                         free(name2);
 732                         break;
 733                 }
 734
 735                 if(!strcmp(name2, mesh->name)) {
 736                         logger(mesh, MESHLINK_DEBUG, "Secondary chunk would overwrite our own host config file.\n");
 737                         free(name2);
 738                         meshlink_errno = MESHLINK_EPEER;
 739                         return false;
 740                 }
 741
 742                 node_t *n = new_node();
 743                 n->name = name2;
 744
 745                 config_t config = {data, data_len};
 746
 747                 if(!node_read_from_config(mesh, n, &config)) {
 748                         free_node(n);
 749                         logger(mesh, MESHLINK_ERROR, "Invalid host config file in invitation file!\n");
 750                         meshlink_errno = MESHLINK_EPEER;
 751                         return false;
 752                 }
 753
 754                 if(i == 0) {
 755                         /* The first host config file is of the inviter itself;
 756                          * remember the address we are currently using for the invitation connection.
 757                          */
 758                         sockaddr_t sa;
 759                         socklen_t salen = sizeof(sa);
 760
 761                         if(getpeername(state->sock, &sa.sa, &salen) == 0) {
 762                                 node_add_recent_address(mesh, n, &sa);
 763                         }
 764                 }
 765
 766                 /* Clear the reachability times, since we ourself have never seen these nodes yet */
 767                 n->last_reachable = 0;
 768                 n->last_unreachable = 0;
 769
 770                 if(!node_write_config(mesh, n)) {
 771                         free_node(n);
 772                         return false;
 773                 }
 774
 775                 node_add(mesh, n);
 776         }
 777
 778         /* Ensure the configuration directory metadata is on disk */
 779         if(!config_sync(mesh, "current") || !sync_path(mesh->confbase)) {
 780                 return false;
 781         }
 782
 783         if(!mesh->inviter_commits_first) {
 784                 devtool_set_inviter_commits_first(false);
 785         }
 786
 787         sptps_send_record(&state->sptps, 1, ecdsa_get_public_key(mesh->private_key), 32);
 788
 789         logger(mesh, MESHLINK_DEBUG, "Configuration stored in: %s\n", mesh->confbase);
 790
 791         return true;
 792 }
 793
 794 static bool invitation_send(void *handle, uint8_t type, const void *data, size_t len) {
 795         (void)type;
 796         join_state_t *state = handle;
 797         const char *ptr = data;
 798
 799         while(len) {
 800                 int result = send(state->sock, ptr, len, 0);
 801
 802                 if(result == -1 && errno == EINTR) {
 803                         continue;
 804                 } else if(result <= 0) {
 805                         return false;
 806                 }
 807
 808                 ptr += result;
 809                 len -= result;
 810         }
 811
 812         return true;
 813 }
 814
 815 static bool invitation_receive(void *handle, uint8_t type, const void *msg, uint16_t len) {
 816         join_state_t *state = handle;
 817         meshlink_handle_t *mesh = state->mesh;
 818
 819         if(mesh->inviter_commits_first) {
 820                 switch(type) {
 821                 case SPTPS_HANDSHAKE:
 822                         return sptps_send_record(&state->sptps, 2, state->cookie, 18 + 32);
 823
 824                 case 1:
 825                         break;
 826
 827                 case 0:
 828                         if(!finalize_join(state, msg, len)) {
 829                                 return false;
 830                         }
 831
 832                         logger(mesh, MESHLINK_DEBUG, "Invitation successfully accepted.\n");
 833                         shutdown(state->sock, SHUT_RDWR);
 834                         state->success = true;
 835                         break;
 836
 837                 default:
 838                         return false;
 839                 }
 840         } else {
 841                 switch(type) {
 842                 case SPTPS_HANDSHAKE:
 843                         return sptps_send_record(&state->sptps, 0, state->cookie, 18);
 844
 845                 case 0:
 846                         return finalize_join(state, msg, len);
 847
 848                 case 1:
 849                         logger(mesh, MESHLINK_DEBUG, "Invitation successfully accepted.\n");
 850                         shutdown(state->sock, SHUT_RDWR);
 851                         state->success = true;
 852                         break;
 853
 854                 default:
 855                         return false;
 856                 }
 857         }
 858
 859         return true;
 860 }
 861
 862 static bool recvline(join_state_t *state) {
 863         char *newline = NULL;
 864
 865         while(!(newline = memchr(state->buffer, '\n', state->blen))) {
 866                 int result = recv(state->sock, state->buffer + state->blen, sizeof(state)->buffer - state->blen, 0);
 867
 868                 if(result == -1 && errno == EINTR) {
 869                         continue;
 870                 } else if(result <= 0) {
 871                         return false;
 872                 }
 873
 874                 state->blen += result;
 875         }
 876
 877         if((size_t)(newline - state->buffer) >= sizeof(state->line)) {
 878                 return false;
 879         }
 880
 881         size_t len = newline - state->buffer;
 882
 883         memcpy(state->line, state->buffer, len);
 884         state->line[len] = 0;
 885         memmove(state->buffer, newline + 1, state->blen - len - 1);
 886         state->blen -= len + 1;
 887
 888         return true;
 889 }
 890
 891 static bool sendline(int fd, const char *format, ...) {
 892         char buffer[4096];
 893         char *p = buffer;
 894         int blen = 0;
 895         va_list ap;
 896
 897         va_start(ap, format);
 898         blen = vsnprintf(buffer, sizeof(buffer), format, ap);
 899         va_end(ap);
 900
 901         if(blen < 1 || (size_t)blen >= sizeof(buffer)) {
 902                 return false;
 903         }
 904
 905         buffer[blen] = '\n';
 906         blen++;
 907
 908         while(blen) {
 909                 int result = send(fd, p, blen, MSG_NOSIGNAL);
 910
 911                 if(result == -1 && errno == EINTR) {
 912                         continue;
 913                 } else if(result <= 0) {
 914                         return false;
 915                 }
 916
 917                 p += result;
 918                 blen -= result;
 919         }
 920
 921         return true;
 922 }
 923
 924 static const char *errstr[] = {
 925         [MESHLINK_OK] = "No error",
 926         [MESHLINK_EINVAL] = "Invalid argument",
 927         [MESHLINK_ENOMEM] = "Out of memory",
 928         [MESHLINK_ENOENT] = "No such node",
 929         [MESHLINK_EEXIST] = "Node already exists",
 930         [MESHLINK_EINTERNAL] = "Internal error",
 931         [MESHLINK_ERESOLV] = "Could not resolve hostname",
 932         [MESHLINK_ESTORAGE] = "Storage error",
 933         [MESHLINK_ENETWORK] = "Network error",
 934         [MESHLINK_EPEER] = "Error communicating with peer",
 935         [MESHLINK_ENOTSUP] = "Operation not supported",
 936         [MESHLINK_EBUSY] = "MeshLink instance already in use",
 937         [MESHLINK_EBLACKLISTED] = "Node is blacklisted",
 938 };
 939
 940 const char *meshlink_strerror(meshlink_errno_t err) {
 941         if((int)err < 0 || err >= sizeof(errstr) / sizeof(*errstr)) {
 942                 return "Invalid error code";
 943         }
 944
 945         return errstr[err];
 946 }
 947
 948 static bool ecdsa_keygen(meshlink_handle_t *mesh) {
 949         logger(mesh, MESHLINK_DEBUG, "Generating ECDSA keypairs:\n");
 950
 951         mesh->private_key = ecdsa_generate();
 952         mesh->invitation_key = ecdsa_generate();
 953
 954         if(!mesh->private_key || !mesh->invitation_key) {
 955                 logger(mesh, MESHLINK_DEBUG, "Error during key generation!\n");
 956                 meshlink_errno = MESHLINK_EINTERNAL;
 957                 return false;
 958         }
 959
 960         logger(mesh, MESHLINK_DEBUG, "Done.\n");
 961
 962         return true;
 963 }
 964
 965 static bool timespec_lt(const struct timespec *a, const struct timespec *b) {
 966         if(a->tv_sec == b->tv_sec) {
 967                 return a->tv_nsec < b->tv_nsec;
 968         } else {
 969                 return a->tv_sec < b->tv_sec;
 970         }
 971 }
 972
 973 static struct timespec idle(event_loop_t *loop, void *data) {
 974         (void)loop;
 975         meshlink_handle_t *mesh = data;
 976         struct timespec t, tmin = {3600, 0};
 977
 978         for splay_each(node_t, n, mesh->nodes) {
 979                 if(!n->utcp) {
 980                         continue;
 981                 }
 982
 983                 t = utcp_timeout(n->utcp);
 984
 985                 if(timespec_lt(&t, &tmin)) {
 986                         tmin = t;
 987                 }
 988         }
 989
 990         return tmin;
 991 }
 992
 993 // Get our local address(es) by simulating connecting to an Internet host.
 994 static void add_local_addresses(meshlink_handle_t *mesh) {
 995         sockaddr_t sa;
 996         sa.storage.ss_family = AF_UNKNOWN;
 997         socklen_t salen = sizeof(sa);
 998
 999         // IPv4 example.org
1000
1001         if(getlocaladdr("93.184.216.34", &sa, &salen, mesh->netns)) {
1002                 sa.in.sin_port = ntohs(atoi(mesh->myport));
1003                 node_add_recent_address(mesh, mesh->self, &sa);
1004         }
1005
1006         // IPv6 example.org
1007
1008         salen = sizeof(sa);
1009
1010         if(getlocaladdr("2606:2800:220:1:248:1893:25c8:1946", &sa, &salen, mesh->netns)) {
1011                 sa.in6.sin6_port = ntohs(atoi(mesh->myport));
1012                 node_add_recent_address(mesh, mesh->self, &sa);
1013         }
1014 }
1015
1016 static bool meshlink_setup(meshlink_handle_t *mesh) {
1017         if(!config_destroy(mesh->confbase, "new")) {
1018                 logger(mesh, MESHLINK_ERROR, "Could not delete configuration in %s/new: %s\n", mesh->confbase, strerror(errno));
1019                 meshlink_errno = MESHLINK_ESTORAGE;
1020                 return false;
1021         }
1022
1023         if(!config_destroy(mesh->confbase, "old")) {
1024                 logger(mesh, MESHLINK_ERROR, "Could not delete configuration in %s/old: %s\n", mesh->confbase, strerror(errno));
1025                 meshlink_errno = MESHLINK_ESTORAGE;
1026                 return false;
1027         }
1028
1029         if(!config_init(mesh, "current")) {
1030                 logger(mesh, MESHLINK_ERROR, "Could not set up configuration in %s/current: %s\n", mesh->confbase, strerror(errno));
1031                 meshlink_errno = MESHLINK_ESTORAGE;
1032                 return false;
1033         }
1034
1035         if(!ecdsa_keygen(mesh)) {
1036                 meshlink_errno = MESHLINK_EINTERNAL;
1037                 return false;
1038         }
1039
1040         if(check_port(mesh) == 0) {
1041                 meshlink_errno = MESHLINK_ENETWORK;
1042                 return false;
1043         }
1044
1045         /* Create a node for ourself */
1046
1047         mesh->self = new_node();
1048         mesh->self->name = xstrdup(mesh->name);
1049         mesh->self->devclass = mesh->devclass;
1050         mesh->self->ecdsa = ecdsa_set_public_key(ecdsa_get_public_key(mesh->private_key));
1051         mesh->self->session_id = mesh->session_id;
1052
1053         if(!write_main_config_files(mesh)) {
1054                 logger(mesh, MESHLINK_ERROR, "Could not write main config files into %s/current: %s\n", mesh->confbase, strerror(errno));
1055                 meshlink_errno = MESHLINK_ESTORAGE;
1056                 return false;
1057         }
1058
1059         /* Ensure the configuration directory metadata is on disk */
1060         if(!config_sync(mesh, "current")) {
1061                 return false;
1062         }
1063
1064         return true;
1065 }
1066
1067 static bool meshlink_read_config(meshlink_handle_t *mesh) {
1068         config_t config;
1069
1070         if(!main_config_read(mesh, "current", &config, mesh->config_key)) {
1071                 logger(NULL, MESHLINK_ERROR, "Could not read main configuration file!");
1072                 return false;
1073         }
1074
1075         packmsg_input_t in = {config.buf, config.len};
1076         const void *private_key;
1077         const void *invitation_key;
1078
1079         uint32_t version = packmsg_get_uint32(&in);
1080         char *name = packmsg_get_str_dup(&in);
1081         uint32_t private_key_len = packmsg_get_bin_raw(&in, &private_key);
1082         uint32_t invitation_key_len = packmsg_get_bin_raw(&in, &invitation_key);
1083         uint16_t myport = packmsg_get_uint16(&in);
1084
1085         if(!packmsg_done(&in) || version != MESHLINK_CONFIG_VERSION || private_key_len != 96 || invitation_key_len != 96) {
1086                 logger(NULL, MESHLINK_ERROR, "Error parsing main configuration file!");
1087                 free(name);
1088                 config_free(&config);
1089                 return false;
1090         }
1091
1092         if(mesh->name && strcmp(mesh->name, name)) {
1093                 logger(NULL, MESHLINK_ERROR, "Configuration is for a different name (%s)!", name);
1094                 meshlink_errno = MESHLINK_ESTORAGE;
1095                 free(name);
1096                 config_free(&config);
1097                 return false;
1098         }
1099
1100         free(mesh->name);
1101         mesh->name = name;
1102         xasprintf(&mesh->myport, "%u", myport);
1103         mesh->private_key = ecdsa_set_private_key(private_key);
1104         mesh->invitation_key = ecdsa_set_private_key(invitation_key);
1105         config_free(&config);
1106
1107         /* Create a node for ourself and read our host configuration file */
1108
1109         mesh->self = new_node();
1110         mesh->self->name = xstrdup(name);
1111         mesh->self->devclass = mesh->devclass;
1112         mesh->self->session_id = mesh->session_id;
1113
1114         if(!node_read_public_key(mesh, mesh->self)) {
1115                 logger(NULL, MESHLINK_ERROR, "Could not read our host configuration file!");
1116                 meshlink_errno = MESHLINK_ESTORAGE;
1117                 free_node(mesh->self);
1118                 mesh->self = NULL;
1119                 return false;
1120         }
1121
1122         return true;
1123 }
1124
1125 #ifdef HAVE_SETNS
1126 static void *setup_network_in_netns_thread(void *arg) {
1127         meshlink_handle_t *mesh = arg;
1128
1129         if(setns(mesh->netns, CLONE_NEWNET) != 0) {
1130                 return NULL;
1131         }
1132
1133         bool success = setup_network(mesh);
1134         return success ? arg : NULL;
1135 }
1136 #endif // HAVE_SETNS
1137
1138 meshlink_open_params_t *meshlink_open_params_init(const char *confbase, const char *name, const char *appname, dev_class_t devclass) {
1139         if(!confbase || !*confbase) {
1140                 logger(NULL, MESHLINK_ERROR, "No confbase given!\n");
1141                 meshlink_errno = MESHLINK_EINVAL;
1142                 return NULL;
1143         }
1144
1145         if(!appname || !*appname) {
1146                 logger(NULL, MESHLINK_ERROR, "No appname given!\n");
1147                 meshlink_errno = MESHLINK_EINVAL;
1148                 return NULL;
1149         }
1150
1151         if(strchr(appname, ' ')) {
1152                 logger(NULL, MESHLINK_ERROR, "Invalid appname given!\n");
1153                 meshlink_errno = MESHLINK_EINVAL;
1154                 return NULL;
1155         }
1156
1157         if(name && !check_id(name)) {
1158                 logger(NULL, MESHLINK_ERROR, "Invalid name given!\n");
1159                 meshlink_errno = MESHLINK_EINVAL;
1160                 return NULL;
1161         }
1162
1163         if(devclass < 0 || devclass >= DEV_CLASS_COUNT) {
1164                 logger(NULL, MESHLINK_ERROR, "Invalid devclass given!\n");
1165                 meshlink_errno = MESHLINK_EINVAL;
1166                 return NULL;
1167         }
1168
1169         meshlink_open_params_t *params = xzalloc(sizeof * params);
1170
1171         params->confbase = xstrdup(confbase);
1172         params->name = name ? xstrdup(name) : NULL;
1173         params->appname = xstrdup(appname);
1174         params->devclass = devclass;
1175         params->netns = -1;
1176
1177         return params;
1178 }
1179
1180 bool meshlink_open_params_set_netns(meshlink_open_params_t *params, int netns) {
1181         if(!params) {
1182                 meshlink_errno = MESHLINK_EINVAL;
1183                 return false;
1184         }
1185
1186         params->netns = netns;
1187
1188         return true;
1189 }
1190
1191 bool meshlink_open_params_set_storage_key(meshlink_open_params_t *params, const void *key, size_t keylen) {
1192         if(!params) {
1193                 meshlink_errno = MESHLINK_EINVAL;
1194                 return false;
1195         }
1196
1197         if((!key && keylen) || (key && !keylen)) {
1198                 logger(NULL, MESHLINK_ERROR, "Invalid key length!\n");
1199                 meshlink_errno = MESHLINK_EINVAL;
1200                 return false;
1201         }
1202
1203         params->key = key;
1204         params->keylen = keylen;
1205
1206         return true;
1207 }
1208
1209 bool meshlink_encrypted_key_rotate(meshlink_handle_t *mesh, const void *new_key, size_t new_keylen) {
1210         if(!mesh || !new_key || !new_keylen) {
1211                 logger(mesh, MESHLINK_ERROR, "Invalid arguments given!\n");
1212                 meshlink_errno = MESHLINK_EINVAL;
1213                 return false;
1214         }
1215
1216         assert(pthread_mutex_lock(&mesh->mutex) == 0);
1217
1218         // Create hash for the new key
1219         void *new_config_key;
1220         new_config_key = xmalloc(CHACHA_POLY1305_KEYLEN);
1221
1222         if(!prf(new_key, new_keylen, "MeshLink configuration key", 26, new_config_key, CHACHA_POLY1305_KEYLEN)) {
1223                 logger(mesh, MESHLINK_ERROR, "Error creating new configuration key!\n");
1224                 meshlink_errno = MESHLINK_EINTERNAL;
1225                 assert(pthread_mutex_unlock(&mesh->mutex) == 0);
1226                 return false;
1227         }
1228
1229         // Copy contents of the "current" confbase sub-directory to "new" confbase sub-directory with the new key
1230
1231         if(!config_copy(mesh, "current", mesh->config_key, "new", new_config_key)) {
1232                 logger(mesh, MESHLINK_ERROR, "Could not set up configuration in %s/old: %s\n", mesh->confbase, strerror(errno));
1233                 meshlink_errno = MESHLINK_ESTORAGE;
1234                 assert(pthread_mutex_unlock(&mesh->mutex) == 0);
1235                 return false;
1236         }
1237
1238         devtool_keyrotate_probe(1);
1239
1240         // Rename confbase/current/ to confbase/old
1241
1242         if(!config_rename(mesh, "current", "old")) {
1243                 logger(mesh, MESHLINK_ERROR, "Cannot rename %s/current to %s/old\n", mesh->confbase, mesh->confbase);
1244                 meshlink_errno = MESHLINK_ESTORAGE;
1245                 assert(pthread_mutex_unlock(&mesh->mutex) == 0);
1246                 return false;
1247         }
1248
1249         devtool_keyrotate_probe(2);
1250
1251         // Rename confbase/new/ to confbase/current
1252
1253         if(!config_rename(mesh, "new", "current")) {
1254                 logger(mesh, MESHLINK_ERROR, "Cannot rename %s/new to %s/current\n", mesh->confbase, mesh->confbase);
1255                 meshlink_errno = MESHLINK_ESTORAGE;
1256                 assert(pthread_mutex_unlock(&mesh->mutex) == 0);
1257                 return false;
1258         }
1259
1260         devtool_keyrotate_probe(3);
1261
1262         // Cleanup the "old" confbase sub-directory
1263
1264         if(!config_destroy(mesh->confbase, "old")) {
1265                 assert(pthread_mutex_unlock(&mesh->mutex) == 0);
1266                 return false;
1267         }
1268
1269         // Change the mesh handle key with new key
1270
1271         free(mesh->config_key);
1272         mesh->config_key = new_config_key;
1273
1274         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
1275
1276         return true;
1277 }
1278
1279 void meshlink_open_params_free(meshlink_open_params_t *params) {
1280         if(!params) {
1281                 meshlink_errno = MESHLINK_EINVAL;
1282                 return;
1283         }
1284
1285         free(params->confbase);
1286         free(params->name);
1287         free(params->appname);
1288
1289         free(params);
1290 }
1291
1292 /// Device class traits
1293 static const dev_class_traits_t default_class_traits[DEV_CLASS_COUNT] = {
1294         { .pingtimeout = 5, .pinginterval = 60, .min_connects = 3, .max_connects = 10000, .edge_weight = 1 }, // DEV_CLASS_BACKBONE
1295         { .pingtimeout = 5, .pinginterval = 60, .min_connects = 3, .max_connects = 100, .edge_weight = 3 },   // DEV_CLASS_STATIONARY
1296         { .pingtimeout = 5, .pinginterval = 60, .min_connects = 3, .max_connects = 3, .edge_weight = 6 },     // DEV_CLASS_PORTABLE
1297         { .pingtimeout = 5, .pinginterval = 60, .min_connects = 1, .max_connects = 1, .edge_weight = 9 },     // DEV_CLASS_UNKNOWN
1298 };
1299
1300 meshlink_handle_t *meshlink_open(const char *confbase, const char *name, const char *appname, dev_class_t devclass) {
1301         if(!confbase || !*confbase) {
1302                 logger(NULL, MESHLINK_ERROR, "No confbase given!\n");
1303                 meshlink_errno = MESHLINK_EINVAL;
1304                 return NULL;
1305         }
1306
1307         /* Create a temporary struct on the stack, to avoid allocating and freeing one. */
1308         meshlink_open_params_t params;
1309         memset(&params, 0, sizeof(params));
1310
1311         params.confbase = (char *)confbase;
1312         params.name = (char *)name;
1313         params.appname = (char *)appname;
1314         params.devclass = devclass;
1315         params.netns = -1;
1316
1317         return meshlink_open_ex(&params);
1318 }
1319
1320 meshlink_handle_t *meshlink_open_encrypted(const char *confbase, const char *name, const char *appname, dev_class_t devclass, const void *key, size_t keylen) {
1321         if(!confbase || !*confbase) {
1322                 logger(NULL, MESHLINK_ERROR, "No confbase given!\n");
1323                 meshlink_errno = MESHLINK_EINVAL;
1324                 return NULL;
1325         }
1326
1327         /* Create a temporary struct on the stack, to avoid allocating and freeing one. */
1328         meshlink_open_params_t params;
1329         memset(&params, 0, sizeof(params));
1330
1331         params.confbase = (char *)confbase;
1332         params.name = (char *)name;
1333         params.appname = (char *)appname;
1334         params.devclass = devclass;
1335         params.netns = -1;
1336
1337         if(!meshlink_open_params_set_storage_key(&params, key, keylen)) {
1338                 return false;
1339         }
1340
1341         return meshlink_open_ex(&params);
1342 }
1343
1344 meshlink_handle_t *meshlink_open_ephemeral(const char *name, const char *appname, dev_class_t devclass) {
1345         if(!name) {
1346                 logger(NULL, MESHLINK_ERROR, "No name given!\n");
1347                 meshlink_errno = MESHLINK_EINVAL;
1348                 return NULL;
1349         }
1350
1351         if(!check_id(name)) {
1352                 logger(NULL, MESHLINK_ERROR, "Invalid name given!\n");
1353                 meshlink_errno = MESHLINK_EINVAL;
1354                 return NULL;
1355         }
1356
1357         if(!appname || !*appname) {
1358                 logger(NULL, MESHLINK_ERROR, "No appname given!\n");
1359                 meshlink_errno = MESHLINK_EINVAL;
1360                 return NULL;
1361         }
1362
1363         if(strchr(appname, ' ')) {
1364                 logger(NULL, MESHLINK_ERROR, "Invalid appname given!\n");
1365                 meshlink_errno = MESHLINK_EINVAL;
1366                 return NULL;
1367         }
1368
1369         if(devclass < 0 || devclass >= DEV_CLASS_COUNT) {
1370                 logger(NULL, MESHLINK_ERROR, "Invalid devclass given!\n");
1371                 meshlink_errno = MESHLINK_EINVAL;
1372                 return NULL;
1373         }
1374
1375         /* Create a temporary struct on the stack, to avoid allocating and freeing one. */
1376         meshlink_open_params_t params;
1377         memset(&params, 0, sizeof(params));
1378
1379         params.name = (char *)name;
1380         params.appname = (char *)appname;
1381         params.devclass = devclass;
1382         params.netns = -1;
1383
1384         return meshlink_open_ex(&params);
1385 }
1386
1387 meshlink_handle_t *meshlink_open_ex(const meshlink_open_params_t *params) {
1388         logger(NULL, MESHLINK_DEBUG, "meshlink_open called\n");
1389
1390         // Validate arguments provided by the application
1391         if(!params->appname || !*params->appname) {
1392                 logger(NULL, MESHLINK_ERROR, "No appname given!\n");
1393                 meshlink_errno = MESHLINK_EINVAL;
1394                 return NULL;
1395         }
1396
1397         if(strchr(params->appname, ' ')) {
1398                 logger(NULL, MESHLINK_ERROR, "Invalid appname given!\n");
1399                 meshlink_errno = MESHLINK_EINVAL;
1400                 return NULL;
1401         }
1402
1403         if(params->name && !check_id(params->name)) {
1404                 logger(NULL, MESHLINK_ERROR, "Invalid name given!\n");
1405                 meshlink_errno = MESHLINK_EINVAL;
1406                 return NULL;
1407         }
1408
1409         if(params->devclass < 0 || params->devclass >= DEV_CLASS_COUNT) {
1410                 logger(NULL, MESHLINK_ERROR, "Invalid devclass given!\n");
1411                 meshlink_errno = MESHLINK_EINVAL;
1412                 return NULL;
1413         }
1414
1415         if((params->key && !params->keylen) || (!params->key && params->keylen)) {
1416                 logger(NULL, MESHLINK_ERROR, "Invalid key length!\n");
1417                 meshlink_errno = MESHLINK_EINVAL;
1418                 return NULL;
1419         }
1420
1421         meshlink_handle_t *mesh = xzalloc(sizeof(meshlink_handle_t));
1422
1423         if(params->confbase) {
1424                 mesh->confbase = xstrdup(params->confbase);
1425         }
1426
1427         mesh->appname = xstrdup(params->appname);
1428         mesh->devclass = params->devclass;
1429         mesh->discovery = true;
1430         mesh->invitation_timeout = 604800; // 1 week
1431         mesh->netns = params->netns;
1432         mesh->submeshes = NULL;
1433         mesh->log_cb = global_log_cb;
1434         mesh->log_level = global_log_level;
1435         mesh->packet = xmalloc(sizeof(vpn_packet_t));
1436
1437         randomize(&mesh->prng_state, sizeof(mesh->prng_state));
1438
1439         do {
1440                 randomize(&mesh->session_id, sizeof(mesh->session_id));
1441         } while(mesh->session_id == 0);
1442
1443         memcpy(mesh->dev_class_traits, default_class_traits, sizeof(default_class_traits));
1444
1445         mesh->name = params->name ? xstrdup(params->name) : NULL;
1446
1447         // Hash the key
1448         if(params->key) {
1449                 mesh->config_key = xmalloc(CHACHA_POLY1305_KEYLEN);
1450
1451                 if(!prf(params->key, params->keylen, "MeshLink configuration key", 26, mesh->config_key, CHACHA_POLY1305_KEYLEN)) {
1452                         logger(NULL, MESHLINK_ERROR, "Error creating configuration key!\n");
1453                         meshlink_close(mesh);
1454                         meshlink_errno = MESHLINK_EINTERNAL;
1455                         return NULL;
1456                 }
1457         }
1458
1459         // initialize mutexes and conds
1460         pthread_mutexattr_t attr;
1461         assert(pthread_mutexattr_init(&attr) == 0);
1462         assert(pthread_mutexattr_settype(&attr, PTHREAD_MUTEX_RECURSIVE) == 0);
1463         assert(pthread_mutex_init(&mesh->mutex, &attr) == 0);
1464         assert(pthread_cond_init(&mesh->cond, NULL) == 0);
1465
1466         assert(pthread_mutex_init(&mesh->discovery_mutex, NULL) == 0);
1467         assert(pthread_cond_init(&mesh->discovery_cond, NULL) == 0);
1468
1469         assert(pthread_cond_init(&mesh->adns_cond, NULL) == 0);
1470
1471         mesh->threadstarted = false;
1472         event_loop_init(&mesh->loop);
1473         mesh->loop.data = mesh;
1474
1475         meshlink_queue_init(&mesh->outpacketqueue);
1476
1477         // Atomically lock the configuration directory.
1478         if(!main_config_lock(mesh)) {
1479                 meshlink_close(mesh);
1480                 return NULL;
1481         }
1482
1483         // If no configuration exists yet, create it.
1484
1485         if(!meshlink_confbase_exists(mesh)) {
1486                 if(!mesh->name) {
1487                         logger(NULL, MESHLINK_ERROR, "No configuration files found!\n");
1488                         meshlink_close(mesh);
1489                         meshlink_errno = MESHLINK_ESTORAGE;
1490                         return NULL;
1491                 }
1492
1493                 if(!meshlink_setup(mesh)) {
1494                         logger(NULL, MESHLINK_ERROR, "Cannot create initial configuration\n");
1495                         meshlink_close(mesh);
1496                         return NULL;
1497                 }
1498         } else {
1499                 if(!meshlink_read_config(mesh)) {
1500                         logger(NULL, MESHLINK_ERROR, "Cannot read main configuration\n");
1501                         meshlink_close(mesh);
1502                         return NULL;
1503                 }
1504         }
1505
1506 #ifdef HAVE_MINGW
1507         struct WSAData wsa_state;
1508         WSAStartup(MAKEWORD(2, 2), &wsa_state);
1509 #endif
1510
1511         // Setup up everything
1512         // TODO: we should not open listening sockets yet
1513
1514         bool success = false;
1515
1516         if(mesh->netns != -1) {
1517 #ifdef HAVE_SETNS
1518                 pthread_t thr;
1519
1520                 if(pthread_create(&thr, NULL, setup_network_in_netns_thread, mesh) == 0) {
1521                         void *retval = NULL;
1522                         success = pthread_join(thr, &retval) == 0 && retval;
1523                 }
1524
1525 #else
1526                 meshlink_errno = MESHLINK_EINTERNAL;
1527                 return NULL;
1528
1529 #endif // HAVE_SETNS
1530         } else {
1531                 success = setup_network(mesh);
1532         }
1533
1534         if(!success) {
1535                 meshlink_close(mesh);
1536                 meshlink_errno = MESHLINK_ENETWORK;
1537                 return NULL;
1538         }
1539
1540         add_local_addresses(mesh);
1541
1542         if(!node_write_config(mesh, mesh->self)) {
1543                 logger(NULL, MESHLINK_ERROR, "Cannot update configuration\n");
1544                 return NULL;
1545         }
1546
1547         idle_set(&mesh->loop, idle, mesh);
1548
1549         logger(NULL, MESHLINK_DEBUG, "meshlink_open returning\n");
1550         return mesh;
1551 }
1552
1553 meshlink_submesh_t *meshlink_submesh_open(meshlink_handle_t  *mesh, const char *submesh) {
1554         meshlink_submesh_t *s = NULL;
1555
1556         if(!mesh) {
1557                 logger(NULL, MESHLINK_ERROR, "No mesh handle given!\n");
1558                 meshlink_errno = MESHLINK_EINVAL;
1559                 return NULL;
1560         }
1561
1562         if(!submesh || !*submesh) {
1563                 logger(NULL, MESHLINK_ERROR, "No submesh name given!\n");
1564                 meshlink_errno = MESHLINK_EINVAL;
1565                 return NULL;
1566         }
1567
1568         //lock mesh->nodes
1569         assert(pthread_mutex_lock(&mesh->mutex) == 0);
1570
1571         s = (meshlink_submesh_t *)create_submesh(mesh, submesh);
1572
1573         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
1574
1575         return s;
1576 }
1577
1578 static void *meshlink_main_loop(void *arg) {
1579         meshlink_handle_t *mesh = arg;
1580
1581         if(mesh->netns != -1) {
1582 #ifdef HAVE_SETNS
1583
1584                 if(setns(mesh->netns, CLONE_NEWNET) != 0) {
1585                         assert(pthread_cond_signal(&mesh->cond) == 0);
1586                         return NULL;
1587                 }
1588
1589 #else
1590                 assert(pthread_cond_signal(&mesh->cond) == 0);
1591                 return NULL;
1592 #endif // HAVE_SETNS
1593         }
1594
1595 #if HAVE_CATTA
1596
1597         if(mesh->discovery) {
1598                 discovery_start(mesh);
1599         }
1600
1601 #endif
1602
1603         assert(pthread_mutex_lock(&mesh->mutex) == 0);
1604
1605         logger(mesh, MESHLINK_DEBUG, "Starting main_loop...\n");
1606         assert(pthread_cond_broadcast(&mesh->cond) == 0);
1607         main_loop(mesh);
1608         logger(mesh, MESHLINK_DEBUG, "main_loop returned.\n");
1609
1610         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
1611
1612 #if HAVE_CATTA
1613
1614         // Stop discovery
1615         if(mesh->discovery) {
1616                 discovery_stop(mesh);
1617         }
1618
1619 #endif
1620
1621         return NULL;
1622 }
1623
1624 bool meshlink_start(meshlink_handle_t *mesh) {
1625         if(!mesh) {
1626                 meshlink_errno = MESHLINK_EINVAL;
1627                 return false;
1628         }
1629
1630         logger(mesh, MESHLINK_DEBUG, "meshlink_start called\n");
1631
1632         assert(pthread_mutex_lock(&mesh->mutex) == 0);
1633
1634         assert(mesh->self);
1635         assert(mesh->private_key);
1636         assert(mesh->self->ecdsa);
1637         assert(!memcmp((uint8_t *)mesh->self->ecdsa + 64, (uint8_t *)mesh->private_key + 64, 32));
1638
1639         if(mesh->threadstarted) {
1640                 logger(mesh, MESHLINK_DEBUG, "thread was already running\n");
1641                 assert(pthread_mutex_unlock(&mesh->mutex) == 0);
1642                 return true;
1643         }
1644
1645         if(mesh->listen_socket[0].tcp.fd < 0) {
1646                 logger(mesh, MESHLINK_ERROR, "Listening socket not open\n");
1647                 meshlink_errno = MESHLINK_ENETWORK;
1648                 return false;
1649         }
1650
1651         // TODO: open listening sockets first
1652
1653         //Check that a valid name is set
1654         if(!mesh->name) {
1655                 logger(mesh, MESHLINK_DEBUG, "No name given!\n");
1656                 meshlink_errno = MESHLINK_EINVAL;
1657                 assert(pthread_mutex_unlock(&mesh->mutex) == 0);
1658                 return false;
1659         }
1660
1661         init_outgoings(mesh);
1662         init_adns(mesh);
1663
1664         // Start the main thread
1665
1666         event_loop_start(&mesh->loop);
1667
1668         // Ensure we have a decent amount of stack space. Musl's default of 80 kB is too small.
1669         pthread_attr_t attr;
1670         assert(pthread_attr_init(&attr) == 0);
1671         assert(pthread_attr_setstacksize(&attr, 1024 * 1024) == 0);
1672
1673         if(pthread_create(&mesh->thread, &attr, meshlink_main_loop, mesh) != 0) {
1674                 logger(mesh, MESHLINK_DEBUG, "Could not start thread: %s\n", strerror(errno));
1675                 memset(&mesh->thread, 0, sizeof(mesh)->thread);
1676                 meshlink_errno = MESHLINK_EINTERNAL;
1677                 event_loop_stop(&mesh->loop);
1678                 assert(pthread_mutex_unlock(&mesh->mutex) == 0);
1679                 return false;
1680         }
1681
1682         assert(pthread_cond_wait(&mesh->cond, &mesh->mutex) == 0);
1683         mesh->threadstarted = true;
1684
1685         // Ensure we are considered reachable
1686         graph(mesh);
1687
1688         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
1689         return true;
1690 }
1691
1692 void meshlink_stop(meshlink_handle_t *mesh) {
1693         if(!mesh) {
1694                 meshlink_errno = MESHLINK_EINVAL;
1695                 return;
1696         }
1697
1698         assert(pthread_mutex_lock(&mesh->mutex) == 0);
1699         logger(mesh, MESHLINK_DEBUG, "meshlink_stop called\n");
1700
1701         // Shut down the main thread
1702         event_loop_stop(&mesh->loop);
1703
1704         // Send ourselves a UDP packet to kick the event loop
1705         for(int i = 0; i < mesh->listen_sockets; i++) {
1706                 sockaddr_t sa;
1707                 socklen_t salen = sizeof(sa);
1708
1709                 if(getsockname(mesh->listen_socket[i].udp.fd, &sa.sa, &salen) == -1) {
1710                         logger(mesh, MESHLINK_ERROR, "System call `%s' failed: %s", "getsockname", sockstrerror(sockerrno));
1711                         continue;
1712                 }
1713
1714                 if(sendto(mesh->listen_socket[i].udp.fd, "", 1, MSG_NOSIGNAL, &sa.sa, salen) == -1) {
1715                         logger(mesh, MESHLINK_ERROR, "Could not send a UDP packet to ourself: %s", sockstrerror(sockerrno));
1716                 }
1717         }
1718
1719         if(mesh->threadstarted) {
1720                 // Wait for the main thread to finish
1721                 assert(pthread_mutex_unlock(&mesh->mutex) == 0);
1722                 assert(pthread_join(mesh->thread, NULL) == 0);
1723                 assert(pthread_mutex_lock(&mesh->mutex) == 0);
1724
1725                 mesh->threadstarted = false;
1726         }
1727
1728         // Close all metaconnections
1729         if(mesh->connections) {
1730                 for(list_node_t *node = mesh->connections->head, *next; node; node = next) {
1731                         next = node->next;
1732                         connection_t *c = node->data;
1733                         c->outgoing = NULL;
1734                         terminate_connection(mesh, c, false);
1735                 }
1736         }
1737
1738         exit_adns(mesh);
1739         exit_outgoings(mesh);
1740
1741         // Ensure we are considered unreachable
1742         if(mesh->nodes) {
1743                 graph(mesh);
1744         }
1745
1746         // Try to write out any changed node config files, ignore errors at this point.
1747         if(mesh->nodes) {
1748                 for splay_each(node_t, n, mesh->nodes) {
1749                         if(n->status.dirty) {
1750                                 n->status.dirty = !node_write_config(mesh, n);
1751                         }
1752                 }
1753         }
1754
1755         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
1756 }
1757
1758 void meshlink_close(meshlink_handle_t *mesh) {
1759         if(!mesh) {
1760                 meshlink_errno = MESHLINK_EINVAL;
1761                 return;
1762         }
1763
1764         // stop can be called even if mesh has not been started
1765         meshlink_stop(mesh);
1766
1767         // lock is not released after this
1768         assert(pthread_mutex_lock(&mesh->mutex) == 0);
1769
1770         // Close and free all resources used.
1771
1772         close_network_connections(mesh);
1773
1774         logger(mesh, MESHLINK_INFO, "Terminating");
1775
1776         event_loop_exit(&mesh->loop);
1777
1778 #ifdef HAVE_MINGW
1779
1780         if(mesh->confbase) {
1781                 WSACleanup();
1782         }
1783
1784 #endif
1785
1786         ecdsa_free(mesh->invitation_key);
1787
1788         if(mesh->netns != -1) {
1789                 close(mesh->netns);
1790         }
1791
1792         for(vpn_packet_t *packet; (packet = meshlink_queue_pop(&mesh->outpacketqueue));) {
1793                 free(packet);
1794         }
1795
1796         meshlink_queue_exit(&mesh->outpacketqueue);
1797
1798         free(mesh->name);
1799         free(mesh->appname);
1800         free(mesh->confbase);
1801         free(mesh->config_key);
1802         free(mesh->external_address_url);
1803         free(mesh->packet);
1804         ecdsa_free(mesh->private_key);
1805
1806         if(mesh->invitation_addresses) {
1807                 list_delete_list(mesh->invitation_addresses);
1808         }
1809
1810         main_config_unlock(mesh);
1811
1812         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
1813         assert(pthread_mutex_destroy(&mesh->mutex) == 0);
1814
1815         memset(mesh, 0, sizeof(*mesh));
1816
1817         free(mesh);
1818 }
1819
1820 bool meshlink_destroy(const char *confbase) {
1821         if(!confbase) {
1822                 meshlink_errno = MESHLINK_EINVAL;
1823                 return false;
1824         }
1825
1826         /* Exit early if the confbase directory itself doesn't exist */
1827         if(access(confbase, F_OK) && errno == ENOENT) {
1828                 return true;
1829         }
1830
1831         /* Take the lock the same way meshlink_open() would. */
1832         char lockfilename[PATH_MAX];
1833         snprintf(lockfilename, sizeof(lockfilename), "%s" SLASH "meshlink.lock", confbase);
1834
1835         FILE *lockfile = fopen(lockfilename, "w+");
1836
1837         if(!lockfile) {
1838                 logger(NULL, MESHLINK_ERROR, "Could not open lock file %s: %s", lockfilename, strerror(errno));
1839                 meshlink_errno = MESHLINK_ESTORAGE;
1840                 return false;
1841         }
1842
1843 #ifdef FD_CLOEXEC
1844         fcntl(fileno(lockfile), F_SETFD, FD_CLOEXEC);
1845 #endif
1846
1847 #ifdef HAVE_MINGW
1848         // TODO: use _locking()?
1849 #else
1850
1851         if(flock(fileno(lockfile), LOCK_EX | LOCK_NB) != 0) {
1852                 logger(NULL, MESHLINK_ERROR, "Configuration directory %s still in use\n", lockfilename);
1853                 fclose(lockfile);
1854                 meshlink_errno = MESHLINK_EBUSY;
1855                 return false;
1856         }
1857
1858 #endif
1859
1860         if(!config_destroy(confbase, "current") || !config_destroy(confbase, "new") || !config_destroy(confbase, "old")) {
1861                 logger(NULL, MESHLINK_ERROR, "Cannot remove sub-directories in %s: %s\n", confbase, strerror(errno));
1862                 return false;
1863         }
1864
1865         if(unlink(lockfilename)) {
1866                 logger(NULL, MESHLINK_ERROR, "Cannot remove lock file %s: %s\n", lockfilename, strerror(errno));
1867                 fclose(lockfile);
1868                 meshlink_errno = MESHLINK_ESTORAGE;
1869                 return false;
1870         }
1871
1872         fclose(lockfile);
1873
1874         if(!sync_path(confbase)) {
1875                 logger(NULL, MESHLINK_ERROR, "Cannot sync directory %s: %s\n", confbase, strerror(errno));
1876                 meshlink_errno = MESHLINK_ESTORAGE;
1877                 return false;
1878         }
1879
1880         return true;
1881 }
1882
1883 void meshlink_set_receive_cb(meshlink_handle_t *mesh, meshlink_receive_cb_t cb) {
1884         if(!mesh) {
1885                 meshlink_errno = MESHLINK_EINVAL;
1886                 return;
1887         }
1888
1889         assert(pthread_mutex_lock(&mesh->mutex) == 0);
1890         mesh->receive_cb = cb;
1891         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
1892 }
1893
1894 void meshlink_set_connection_try_cb(meshlink_handle_t *mesh, meshlink_connection_try_cb_t cb) {
1895         if(!mesh) {
1896                 meshlink_errno = MESHLINK_EINVAL;
1897                 return;
1898         }
1899
1900         assert(pthread_mutex_lock(&mesh->mutex) == 0);
1901         mesh->connection_try_cb = cb;
1902         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
1903 }
1904
1905 void meshlink_set_node_status_cb(meshlink_handle_t *mesh, meshlink_node_status_cb_t cb) {
1906         if(!mesh) {
1907                 meshlink_errno = MESHLINK_EINVAL;
1908                 return;
1909         }
1910
1911         assert(pthread_mutex_lock(&mesh->mutex) == 0);
1912         mesh->node_status_cb = cb;
1913         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
1914 }
1915
1916 void meshlink_set_node_pmtu_cb(meshlink_handle_t *mesh, meshlink_node_pmtu_cb_t cb) {
1917         if(!mesh) {
1918                 meshlink_errno = MESHLINK_EINVAL;
1919                 return;
1920         }
1921
1922         assert(pthread_mutex_lock(&mesh->mutex) == 0);
1923         mesh->node_pmtu_cb = cb;
1924         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
1925 }
1926
1927 void meshlink_set_node_duplicate_cb(meshlink_handle_t *mesh, meshlink_node_duplicate_cb_t cb) {
1928         if(!mesh) {
1929                 meshlink_errno = MESHLINK_EINVAL;
1930                 return;
1931         }
1932
1933         assert(pthread_mutex_lock(&mesh->mutex) == 0);
1934         mesh->node_duplicate_cb = cb;
1935         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
1936 }
1937
1938 void meshlink_set_log_cb(meshlink_handle_t *mesh, meshlink_log_level_t level, meshlink_log_cb_t cb) {
1939         if(mesh) {
1940                 assert(pthread_mutex_lock(&mesh->mutex) == 0);
1941                 mesh->log_cb = cb;
1942                 mesh->log_level = cb ? level : 0;
1943                 assert(pthread_mutex_unlock(&mesh->mutex) == 0);
1944         } else {
1945                 global_log_cb = cb;
1946                 global_log_level = cb ? level : 0;
1947         }
1948 }
1949
1950 void meshlink_set_error_cb(struct meshlink_handle *mesh, meshlink_error_cb_t cb) {
1951         if(!mesh) {
1952                 meshlink_errno = MESHLINK_EINVAL;
1953                 return;
1954         }
1955
1956         assert(pthread_mutex_lock(&mesh->mutex) == 0);
1957         mesh->error_cb = cb;
1958         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
1959 }
1960
1961 static bool prepare_packet(meshlink_handle_t *mesh, meshlink_node_t *destination, const void *data, size_t len, vpn_packet_t *packet) {
1962         meshlink_packethdr_t *hdr;
1963
1964         if(len > MAXSIZE - sizeof(*hdr)) {
1965                 meshlink_errno = MESHLINK_EINVAL;
1966                 return false;
1967         }
1968
1969         node_t *n = (node_t *)destination;
1970
1971         if(n->status.blacklisted) {
1972                 logger(mesh, MESHLINK_ERROR, "Node %s blacklisted, dropping packet\n", n->name);
1973                 meshlink_errno = MESHLINK_EBLACKLISTED;
1974                 return false;
1975         }
1976
1977         // Prepare the packet
1978         packet->probe = false;
1979         packet->tcp = false;
1980         packet->len = len + sizeof(*hdr);
1981
1982         hdr = (meshlink_packethdr_t *)packet->data;
1983         memset(hdr, 0, sizeof(*hdr));
1984         // leave the last byte as 0 to make sure strings are always
1985         // null-terminated if they are longer than the buffer
1986         strncpy((char *)hdr->destination, destination->name, sizeof(hdr->destination) - 1);
1987         strncpy((char *)hdr->source, mesh->self->name, sizeof(hdr->source) - 1);
1988
1989         memcpy(packet->data + sizeof(*hdr), data, len);
1990
1991         return true;
1992 }
1993
1994 static bool meshlink_send_immediate(meshlink_handle_t *mesh, meshlink_node_t *destination, const void *data, size_t len) {
1995         assert(mesh);
1996         assert(destination);
1997         assert(data);
1998         assert(len);
1999
2000         // Prepare the packet
2001         if(!prepare_packet(mesh, destination, data, len, mesh->packet)) {
2002                 return false;
2003         }
2004
2005         // Send it immediately
2006         route(mesh, mesh->self, mesh->packet);
2007
2008         return true;
2009 }
2010
2011 bool meshlink_send(meshlink_handle_t *mesh, meshlink_node_t *destination, const void *data, size_t len) {
2012         // Validate arguments
2013         if(!mesh || !destination) {
2014                 meshlink_errno = MESHLINK_EINVAL;
2015                 return false;
2016         }
2017
2018         if(!len) {
2019                 return true;
2020         }
2021
2022         if(!data) {
2023                 meshlink_errno = MESHLINK_EINVAL;
2024                 return false;
2025         }
2026
2027         // Prepare the packet
2028         vpn_packet_t *packet = malloc(sizeof(*packet));
2029
2030         if(!packet) {
2031                 meshlink_errno = MESHLINK_ENOMEM;
2032                 return false;
2033         }
2034
2035         if(!prepare_packet(mesh, destination, data, len, packet)) {
2036                 free(packet);
2037                 return false;
2038         }
2039
2040         // Queue it
2041         if(!meshlink_queue_push(&mesh->outpacketqueue, packet)) {
2042                 free(packet);
2043                 meshlink_errno = MESHLINK_ENOMEM;
2044                 return false;
2045         }
2046
2047         logger(mesh, MESHLINK_DEBUG, "Adding packet of %zu bytes to packet queue", len);
2048
2049         // Notify event loop
2050         signal_trigger(&mesh->loop, &mesh->datafromapp);
2051
2052         return true;
2053 }
2054
2055 void meshlink_send_from_queue(event_loop_t *loop, void *data) {
2056         (void)loop;
2057         meshlink_handle_t *mesh = data;
2058
2059         logger(mesh, MESHLINK_DEBUG, "Flushing the packet queue");
2060
2061         for(vpn_packet_t *packet; (packet = meshlink_queue_pop(&mesh->outpacketqueue));) {
2062                 logger(mesh, MESHLINK_DEBUG, "Removing packet of %d bytes from packet queue", packet->len);
2063                 mesh->self->in_packets++;
2064                 mesh->self->in_bytes += packet->len;
2065                 route(mesh, mesh->self, packet);
2066                 free(packet);
2067         }
2068 }
2069
2070 ssize_t meshlink_get_pmtu(meshlink_handle_t *mesh, meshlink_node_t *destination) {
2071         if(!mesh || !destination) {
2072                 meshlink_errno = MESHLINK_EINVAL;
2073                 return -1;
2074         }
2075
2076         assert(pthread_mutex_lock(&mesh->mutex) == 0);
2077
2078         node_t *n = (node_t *)destination;
2079
2080         if(!n->status.reachable) {
2081                 assert(pthread_mutex_unlock(&mesh->mutex) == 0);
2082                 return 0;
2083
2084         } else if(n->mtuprobes > 30 && n->minmtu) {
2085                 assert(pthread_mutex_unlock(&mesh->mutex) == 0);
2086                 return n->minmtu;
2087         } else {
2088                 assert(pthread_mutex_unlock(&mesh->mutex) == 0);
2089                 return MTU;
2090         }
2091 }
2092
2093 char *meshlink_get_fingerprint(meshlink_handle_t *mesh, meshlink_node_t *node) {
2094         if(!mesh || !node) {
2095                 meshlink_errno = MESHLINK_EINVAL;
2096                 return NULL;
2097         }
2098
2099         assert(pthread_mutex_lock(&mesh->mutex) == 0);
2100
2101         node_t *n = (node_t *)node;
2102
2103         if(!node_read_public_key(mesh, n) || !n->ecdsa) {
2104                 meshlink_errno = MESHLINK_EINTERNAL;
2105                 assert(pthread_mutex_unlock(&mesh->mutex) == 0);
2106                 return false;
2107         }
2108
2109         char *fingerprint = ecdsa_get_base64_public_key(n->ecdsa);
2110
2111         if(!fingerprint) {
2112                 meshlink_errno = MESHLINK_EINTERNAL;
2113         }
2114
2115         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
2116         return fingerprint;
2117 }
2118
2119 meshlink_node_t *meshlink_get_self(meshlink_handle_t *mesh) {
2120         if(!mesh) {
2121                 meshlink_errno = MESHLINK_EINVAL;
2122                 return NULL;
2123         }
2124
2125         return (meshlink_node_t *)mesh->self;
2126 }
2127
2128 meshlink_node_t *meshlink_get_node(meshlink_handle_t *mesh, const char *name) {
2129         if(!mesh || !name) {
2130                 meshlink_errno = MESHLINK_EINVAL;
2131                 return NULL;
2132         }
2133
2134         node_t *n = NULL;
2135
2136         assert(pthread_mutex_lock(&mesh->mutex) == 0);
2137         n = lookup_node(mesh, (char *)name); // TODO: make lookup_node() use const
2138         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
2139
2140         if(!n) {
2141                 meshlink_errno = MESHLINK_ENOENT;
2142         }
2143
2144         return (meshlink_node_t *)n;
2145 }
2146
2147 meshlink_submesh_t *meshlink_get_submesh(meshlink_handle_t *mesh, const char *name) {
2148         if(!mesh || !name) {
2149                 meshlink_errno = MESHLINK_EINVAL;
2150                 return NULL;
2151         }
2152
2153         meshlink_submesh_t *submesh = NULL;
2154
2155         assert(pthread_mutex_lock(&mesh->mutex) == 0);
2156         submesh = (meshlink_submesh_t *)lookup_submesh(mesh, name);
2157         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
2158
2159         if(!submesh) {
2160                 meshlink_errno = MESHLINK_ENOENT;
2161         }
2162
2163         return submesh;
2164 }
2165
2166 meshlink_node_t **meshlink_get_all_nodes(meshlink_handle_t *mesh, meshlink_node_t **nodes, size_t *nmemb) {
2167         if(!mesh || !nmemb || (*nmemb && !nodes)) {
2168                 meshlink_errno = MESHLINK_EINVAL;
2169                 return NULL;
2170         }
2171
2172         meshlink_node_t **result;
2173
2174         //lock mesh->nodes
2175         assert(pthread_mutex_lock(&mesh->mutex) == 0);
2176
2177         *nmemb = mesh->nodes->count;
2178         result = realloc(nodes, *nmemb * sizeof(*nodes));
2179
2180         if(result) {
2181                 meshlink_node_t **p = result;
2182
2183                 for splay_each(node_t, n, mesh->nodes) {
2184                         *p++ = (meshlink_node_t *)n;
2185                 }
2186         } else {
2187                 *nmemb = 0;
2188                 free(nodes);
2189                 meshlink_errno = MESHLINK_ENOMEM;
2190         }
2191
2192         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
2193
2194         return result;
2195 }
2196
2197 static meshlink_node_t **meshlink_get_all_nodes_by_condition(meshlink_handle_t *mesh, const void *condition, meshlink_node_t **nodes, size_t *nmemb, search_node_by_condition_t search_node) {
2198         meshlink_node_t **result;
2199
2200         assert(pthread_mutex_lock(&mesh->mutex) == 0);
2201
2202         *nmemb = 0;
2203
2204         for splay_each(node_t, n, mesh->nodes) {
2205                 if(search_node(n, condition)) {
2206                         ++*nmemb;
2207                 }
2208         }
2209
2210         if(*nmemb == 0) {
2211                 free(nodes);
2212                 assert(pthread_mutex_unlock(&mesh->mutex) == 0);
2213                 return NULL;
2214         }
2215
2216         result = realloc(nodes, *nmemb * sizeof(*nodes));
2217
2218         if(result) {
2219                 meshlink_node_t **p = result;
2220
2221                 for splay_each(node_t, n, mesh->nodes) {
2222                         if(search_node(n, condition)) {
2223                                 *p++ = (meshlink_node_t *)n;
2224                         }
2225                 }
2226         } else {
2227                 *nmemb = 0;
2228                 free(nodes);
2229                 meshlink_errno = MESHLINK_ENOMEM;
2230         }
2231
2232         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
2233
2234         return result;
2235 }
2236
2237 static bool search_node_by_dev_class(const node_t *node, const void *condition) {
2238         dev_class_t *devclass = (dev_class_t *)condition;
2239
2240         if(*devclass == (dev_class_t)node->devclass) {
2241                 return true;
2242         }
2243
2244         return false;
2245 }
2246
2247 static bool search_node_by_submesh(const node_t *node, const void *condition) {
2248         if(condition == node->submesh) {
2249                 return true;
2250         }
2251
2252         return false;
2253 }
2254
2255 struct time_range {
2256         time_t start;
2257         time_t end;
2258 };
2259
2260 static bool search_node_by_last_reachable(const node_t *node, const void *condition) {
2261         const struct time_range *range = condition;
2262         time_t start = node->last_reachable;
2263         time_t end = node->last_unreachable;
2264
2265         if(end < start) {
2266                 end = time(NULL);
2267
2268                 if(end < start) {
2269                         start = end;
2270                 }
2271         }
2272
2273         if(range->end >= range->start) {
2274                 return start <= range->end && end >= range->start;
2275         } else {
2276                 return start > range->start || end < range->end;
2277         }
2278 }
2279
2280 meshlink_node_t **meshlink_get_all_nodes_by_dev_class(meshlink_handle_t *mesh, dev_class_t devclass, meshlink_node_t **nodes, size_t *nmemb) {
2281         if(!mesh || devclass < 0 || devclass >= DEV_CLASS_COUNT || !nmemb) {
2282                 meshlink_errno = MESHLINK_EINVAL;
2283                 return NULL;
2284         }
2285
2286         return meshlink_get_all_nodes_by_condition(mesh, &devclass, nodes, nmemb, search_node_by_dev_class);
2287 }
2288
2289 meshlink_node_t **meshlink_get_all_nodes_by_submesh(meshlink_handle_t *mesh, meshlink_submesh_t *submesh, meshlink_node_t **nodes, size_t *nmemb) {
2290         if(!mesh || !submesh || !nmemb) {
2291                 meshlink_errno = MESHLINK_EINVAL;
2292                 return NULL;
2293         }
2294
2295         return meshlink_get_all_nodes_by_condition(mesh, submesh, nodes, nmemb, search_node_by_submesh);
2296 }
2297
2298 meshlink_node_t **meshlink_get_all_nodes_by_last_reachable(meshlink_handle_t *mesh, time_t start, time_t end, meshlink_node_t **nodes, size_t *nmemb) {
2299         if(!mesh || !nmemb) {
2300                 meshlink_errno = MESHLINK_EINVAL;
2301                 return NULL;
2302         }
2303
2304         struct time_range range = {start, end};
2305
2306         return meshlink_get_all_nodes_by_condition(mesh, &range, nodes, nmemb, search_node_by_last_reachable);
2307 }
2308
2309 dev_class_t meshlink_get_node_dev_class(meshlink_handle_t *mesh, meshlink_node_t *node) {
2310         if(!mesh || !node) {
2311                 meshlink_errno = MESHLINK_EINVAL;
2312                 return -1;
2313         }
2314
2315         dev_class_t devclass;
2316
2317         assert(pthread_mutex_lock(&mesh->mutex) == 0);
2318
2319         devclass = ((node_t *)node)->devclass;
2320
2321         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
2322
2323         return devclass;
2324 }
2325
2326 meshlink_submesh_t *meshlink_get_node_submesh(meshlink_handle_t *mesh, meshlink_node_t *node) {
2327         if(!mesh || !node) {
2328                 meshlink_errno = MESHLINK_EINVAL;
2329                 return NULL;
2330         }
2331
2332         node_t *n = (node_t *)node;
2333
2334         meshlink_submesh_t *s;
2335
2336         s = (meshlink_submesh_t *)n->submesh;
2337
2338         return s;
2339 }
2340
2341 bool meshlink_get_node_reachability(struct meshlink_handle *mesh, struct meshlink_node *node, time_t *last_reachable, time_t *last_unreachable) {
2342         if(!mesh || !node) {
2343                 meshlink_errno = MESHLINK_EINVAL;
2344                 return NULL;
2345         }
2346
2347         node_t *n = (node_t *)node;
2348         bool reachable;
2349
2350         assert(pthread_mutex_lock(&mesh->mutex) == 0);
2351         reachable = n->status.reachable && !n->status.blacklisted;
2352
2353         if(last_reachable) {
2354                 *last_reachable = n->last_reachable;
2355         }
2356
2357         if(last_unreachable) {
2358                 *last_unreachable = n->last_unreachable;
2359         }
2360
2361         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
2362
2363         return reachable;
2364 }
2365
2366 bool meshlink_sign(meshlink_handle_t *mesh, const void *data, size_t len, void *signature, size_t *siglen) {
2367         if(!mesh || !data || !len || !signature || !siglen) {
2368                 meshlink_errno = MESHLINK_EINVAL;
2369                 return false;
2370         }
2371
2372         if(*siglen < MESHLINK_SIGLEN) {
2373                 meshlink_errno = MESHLINK_EINVAL;
2374                 return false;
2375         }
2376
2377         assert(pthread_mutex_lock(&mesh->mutex) == 0);
2378
2379         if(!ecdsa_sign(mesh->private_key, data, len, signature)) {
2380                 meshlink_errno = MESHLINK_EINTERNAL;
2381                 assert(pthread_mutex_unlock(&mesh->mutex) == 0);
2382                 return false;
2383         }
2384
2385         *siglen = MESHLINK_SIGLEN;
2386         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
2387         return true;
2388 }
2389
2390 bool meshlink_verify(meshlink_handle_t *mesh, meshlink_node_t *source, const void *data, size_t len, const void *signature, size_t siglen) {
2391         if(!mesh || !source || !data || !len || !signature) {
2392                 meshlink_errno = MESHLINK_EINVAL;
2393                 return false;
2394         }
2395
2396         if(siglen != MESHLINK_SIGLEN) {
2397                 meshlink_errno = MESHLINK_EINVAL;
2398                 return false;
2399         }
2400
2401         assert(pthread_mutex_lock(&mesh->mutex) == 0);
2402
2403         bool rval = false;
2404
2405         struct node_t *n = (struct node_t *)source;
2406
2407         if(!node_read_public_key(mesh, n)) {
2408                 meshlink_errno = MESHLINK_EINTERNAL;
2409                 rval = false;
2410         } else {
2411                 rval = ecdsa_verify(((struct node_t *)source)->ecdsa, data, len, signature);
2412         }
2413
2414         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
2415         return rval;
2416 }
2417
2418 static bool refresh_invitation_key(meshlink_handle_t *mesh) {
2419         assert(pthread_mutex_lock(&mesh->mutex) == 0);
2420
2421         size_t count = invitation_purge_old(mesh, time(NULL) - mesh->invitation_timeout);
2422
2423         if(!count) {
2424                 // TODO: Update invitation key if necessary?
2425         }
2426
2427         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
2428
2429         return mesh->invitation_key;
2430 }
2431
2432 bool meshlink_set_canonical_address(meshlink_handle_t *mesh, meshlink_node_t *node, const char *address, const char *port) {
2433         if(!mesh || !node || !address) {
2434                 meshlink_errno = MESHLINK_EINVAL;
2435                 return false;
2436         }
2437
2438         if(!is_valid_hostname(address)) {
2439                 logger(mesh, MESHLINK_DEBUG, "Invalid character in address: %s", address);
2440                 meshlink_errno = MESHLINK_EINVAL;
2441                 return false;
2442         }
2443
2444         if((node_t *)node != mesh->self && !port) {
2445                 logger(mesh, MESHLINK_DEBUG, "Missing port number!");
2446                 meshlink_errno = MESHLINK_EINVAL;
2447                 return false;
2448
2449         }
2450
2451         if(port && !is_valid_port(port)) {
2452                 logger(mesh, MESHLINK_DEBUG, "Invalid character in port: %s", address);
2453                 meshlink_errno = MESHLINK_EINVAL;
2454                 return false;
2455         }
2456
2457         char *canonical_address;
2458
2459         if(port) {
2460                 xasprintf(&canonical_address, "%s %s", address, port);
2461         } else {
2462                 canonical_address = xstrdup(address);
2463         }
2464
2465         assert(pthread_mutex_lock(&mesh->mutex) == 0);
2466
2467         node_t *n = (node_t *)node;
2468         free(n->canonical_address);
2469         n->canonical_address = canonical_address;
2470
2471         if(!node_write_config(mesh, n)) {
2472                 assert(pthread_mutex_unlock(&mesh->mutex) == 0);
2473                 return false;
2474         }
2475
2476         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
2477
2478         return config_sync(mesh, "current");
2479 }
2480
2481 bool meshlink_add_invitation_address(struct meshlink_handle *mesh, const char *address, const char *port) {
2482         if(!mesh || !address) {
2483                 meshlink_errno = MESHLINK_EINVAL;
2484                 return false;
2485         }
2486
2487         if(!is_valid_hostname(address)) {
2488                 logger(mesh, MESHLINK_DEBUG, "Invalid character in address: %s\n", address);
2489                 meshlink_errno = MESHLINK_EINVAL;
2490                 return false;
2491         }
2492
2493         if(port && !is_valid_port(port)) {
2494                 logger(mesh, MESHLINK_DEBUG, "Invalid character in port: %s\n", address);
2495                 meshlink_errno = MESHLINK_EINVAL;
2496                 return false;
2497         }
2498
2499         char *combo;
2500
2501         if(port) {
2502                 xasprintf(&combo, "%s/%s", address, port);
2503         } else {
2504                 combo = xstrdup(address);
2505         }
2506
2507         assert(pthread_mutex_lock(&mesh->mutex) == 0);
2508
2509         if(!mesh->invitation_addresses) {
2510                 mesh->invitation_addresses = list_alloc((list_action_t)free);
2511         }
2512
2513         list_insert_tail(mesh->invitation_addresses, combo);
2514         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
2515
2516         return true;
2517 }
2518
2519 void meshlink_clear_invitation_addresses(struct meshlink_handle *mesh) {
2520         if(!mesh) {
2521                 meshlink_errno = MESHLINK_EINVAL;
2522                 return;
2523         }
2524
2525         assert(pthread_mutex_lock(&mesh->mutex) == 0);
2526
2527         if(mesh->invitation_addresses) {
2528                 list_delete_list(mesh->invitation_addresses);
2529                 mesh->invitation_addresses = NULL;
2530         }
2531
2532         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
2533 }
2534
2535 bool meshlink_add_address(meshlink_handle_t *mesh, const char *address) {
2536         return meshlink_set_canonical_address(mesh, (meshlink_node_t *)mesh->self, address, NULL);
2537 }
2538
2539 bool meshlink_add_external_address(meshlink_handle_t *mesh) {
2540         if(!mesh) {
2541                 meshlink_errno = MESHLINK_EINVAL;
2542                 return false;
2543         }
2544
2545         char *address = meshlink_get_external_address(mesh);
2546
2547         if(!address) {
2548                 return false;
2549         }
2550
2551         bool rval = meshlink_set_canonical_address(mesh, (meshlink_node_t *)mesh->self, address, NULL);
2552         free(address);
2553
2554         return rval;
2555 }
2556
2557 int meshlink_get_port(meshlink_handle_t *mesh) {
2558         if(!mesh) {
2559                 meshlink_errno = MESHLINK_EINVAL;
2560                 return -1;
2561         }
2562
2563         if(!mesh->myport) {
2564                 meshlink_errno = MESHLINK_EINTERNAL;
2565                 return -1;
2566         }
2567
2568         int port;
2569
2570         assert(pthread_mutex_lock(&mesh->mutex) == 0);
2571         port = atoi(mesh->myport);
2572         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
2573
2574         return port;
2575 }
2576
2577 bool meshlink_set_port(meshlink_handle_t *mesh, int port) {
2578         if(!mesh || port < 0 || port >= 65536 || mesh->threadstarted) {
2579                 meshlink_errno = MESHLINK_EINVAL;
2580                 return false;
2581         }
2582
2583         if(mesh->myport && port == atoi(mesh->myport)) {
2584                 return true;
2585         }
2586
2587         if(!try_bind(mesh, port)) {
2588                 meshlink_errno = MESHLINK_ENETWORK;
2589                 return false;
2590         }
2591
2592         devtool_trybind_probe();
2593
2594         bool rval = false;
2595
2596         assert(pthread_mutex_lock(&mesh->mutex) == 0);
2597
2598         if(mesh->threadstarted) {
2599                 meshlink_errno = MESHLINK_EINVAL;
2600                 goto done;
2601         }
2602
2603         free(mesh->myport);
2604         xasprintf(&mesh->myport, "%d", port);
2605
2606         /* Close down the network. This also deletes mesh->self. */
2607         close_network_connections(mesh);
2608
2609         /* Recreate mesh->self. */
2610         mesh->self = new_node();
2611         mesh->self->name = xstrdup(mesh->name);
2612         mesh->self->devclass = mesh->devclass;
2613         mesh->self->session_id = mesh->session_id;
2614         xasprintf(&mesh->myport, "%d", port);
2615
2616         if(!node_read_public_key(mesh, mesh->self)) {
2617                 logger(NULL, MESHLINK_ERROR, "Could not read our host configuration file!");
2618                 meshlink_errno = MESHLINK_ESTORAGE;
2619                 free_node(mesh->self);
2620                 mesh->self = NULL;
2621                 goto done;
2622         } else if(!setup_network(mesh)) {
2623                 meshlink_errno = MESHLINK_ENETWORK;
2624                 goto done;
2625         }
2626
2627         /* Rebuild our own list of recent addresses */
2628         memset(mesh->self->recent, 0, sizeof(mesh->self->recent));
2629         add_local_addresses(mesh);
2630
2631         /* Write meshlink.conf with the updated port number */
2632         write_main_config_files(mesh);
2633
2634         rval = config_sync(mesh, "current");
2635
2636 done:
2637         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
2638
2639         return rval && meshlink_get_port(mesh) == port;
2640 }
2641
2642 void meshlink_set_invitation_timeout(meshlink_handle_t *mesh, int timeout) {
2643         mesh->invitation_timeout = timeout;
2644 }
2645
2646 char *meshlink_invite_ex(meshlink_handle_t *mesh, meshlink_submesh_t *submesh, const char *name, uint32_t flags) {
2647         meshlink_submesh_t *s = NULL;
2648
2649         if(!mesh) {
2650                 meshlink_errno = MESHLINK_EINVAL;
2651                 return NULL;
2652         }
2653
2654         if(submesh) {
2655                 s = (meshlink_submesh_t *)lookup_submesh(mesh, submesh->name);
2656
2657                 if(s != submesh) {
2658                         logger(mesh, MESHLINK_DEBUG, "Invalid SubMesh Handle.\n");
2659                         meshlink_errno = MESHLINK_EINVAL;
2660                         return NULL;
2661                 }
2662         } else {
2663                 s = (meshlink_submesh_t *)mesh->self->submesh;
2664         }
2665
2666         assert(pthread_mutex_lock(&mesh->mutex) == 0);
2667
2668         // Check validity of the new node's name
2669         if(!check_id(name)) {
2670                 logger(mesh, MESHLINK_ERROR, "Invalid name for node.\n");
2671                 meshlink_errno = MESHLINK_EINVAL;
2672                 assert(pthread_mutex_unlock(&mesh->mutex) == 0);
2673                 return NULL;
2674         }
2675
2676         // Ensure no host configuration file with that name exists
2677         if(config_exists(mesh, "current", name)) {
2678                 logger(mesh, MESHLINK_ERROR, "A host config file for %s already exists!\n", name);
2679                 meshlink_errno = MESHLINK_EEXIST;
2680                 assert(pthread_mutex_unlock(&mesh->mutex) == 0);
2681                 return NULL;
2682         }
2683
2684         // Ensure no other nodes know about this name
2685         if(lookup_node(mesh, name)) {
2686                 logger(mesh, MESHLINK_ERROR, "A node with name %s is already known!\n", name);
2687                 meshlink_errno = MESHLINK_EEXIST;
2688                 assert(pthread_mutex_unlock(&mesh->mutex) == 0);
2689                 return NULL;
2690         }
2691
2692         // Get the local address
2693         char *address = get_my_hostname(mesh, flags);
2694
2695         if(!address) {
2696                 logger(mesh, MESHLINK_ERROR, "No Address known for ourselves!\n");
2697                 meshlink_errno = MESHLINK_ERESOLV;
2698                 assert(pthread_mutex_unlock(&mesh->mutex) == 0);
2699                 return NULL;
2700         }
2701
2702         if(!refresh_invitation_key(mesh)) {
2703                 meshlink_errno = MESHLINK_EINTERNAL;
2704                 assert(pthread_mutex_unlock(&mesh->mutex) == 0);
2705                 return NULL;
2706         }
2707
2708         // If we changed our own host config file, write it out now
2709         if(mesh->self->status.dirty) {
2710                 if(!node_write_config(mesh, mesh->self)) {
2711                         logger(mesh, MESHLINK_ERROR, "Could not write our own host config file!\n");
2712                         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
2713                         return NULL;
2714                 }
2715         }
2716
2717         char hash[64];
2718
2719         // Create a hash of the key.
2720         char *fingerprint = ecdsa_get_base64_public_key(mesh->invitation_key);
2721         sha512(fingerprint, strlen(fingerprint), hash);
2722         b64encode_urlsafe(hash, hash, 18);
2723
2724         // Create a random cookie for this invitation.
2725         char cookie[25];
2726         randomize(cookie, 18);
2727
2728         // Create a filename that doesn't reveal the cookie itself
2729         char buf[18 + strlen(fingerprint)];
2730         char cookiehash[64];
2731         memcpy(buf, cookie, 18);
2732         memcpy(buf + 18, fingerprint, sizeof(buf) - 18);
2733         sha512(buf, sizeof(buf), cookiehash);
2734         b64encode_urlsafe(cookiehash, cookiehash, 18);
2735
2736         b64encode_urlsafe(cookie, cookie, 18);
2737
2738         free(fingerprint);
2739
2740         /* Construct the invitation file */
2741         uint8_t outbuf[4096];
2742         packmsg_output_t inv = {outbuf, sizeof(outbuf)};
2743
2744         packmsg_add_uint32(&inv, MESHLINK_INVITATION_VERSION);
2745         packmsg_add_str(&inv, name);
2746         packmsg_add_str(&inv, s ? s->name : CORE_MESH);
2747         packmsg_add_int32(&inv, DEV_CLASS_UNKNOWN); /* TODO: allow this to be set by inviter? */
2748
2749         /* TODO: Add several host config files to bootstrap connections.
2750          * Note: make sure we only add config files of nodes that are in the core mesh or the same submesh,
2751          * and are not blacklisted.
2752          */
2753         config_t configs[5];
2754         memset(configs, 0, sizeof(configs));
2755         int count = 0;
2756
2757         if(config_read(mesh, "current", mesh->self->name, &configs[count], mesh->config_key)) {
2758                 count++;
2759         }
2760
2761         /* Append host config files to the invitation file */
2762         packmsg_add_array(&inv, count);
2763
2764         for(int i = 0; i < count; i++) {
2765                 packmsg_add_bin(&inv, configs[i].buf, configs[i].len);
2766                 config_free(&configs[i]);
2767         }
2768
2769         config_t config = {outbuf, packmsg_output_size(&inv, outbuf)};
2770
2771         if(!invitation_write(mesh, "current", cookiehash, &config, mesh->config_key)) {
2772                 logger(mesh, MESHLINK_DEBUG, "Could not create invitation file %s: %s\n", cookiehash, strerror(errno));
2773                 meshlink_errno = MESHLINK_ESTORAGE;
2774                 assert(pthread_mutex_unlock(&mesh->mutex) == 0);
2775                 return NULL;
2776         }
2777
2778         // Create an URL from the local address, key hash and cookie
2779         char *url;
2780         xasprintf(&url, "%s/%s%s", address, hash, cookie);
2781         free(address);
2782
2783         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
2784         return url;
2785 }
2786
2787 char *meshlink_invite(meshlink_handle_t *mesh, meshlink_submesh_t *submesh, const char *name) {
2788         return meshlink_invite_ex(mesh, submesh, name, 0);
2789 }
2790
2791 bool meshlink_join(meshlink_handle_t *mesh, const char *invitation) {
2792         if(!mesh || !invitation) {
2793                 meshlink_errno = MESHLINK_EINVAL;
2794                 return false;
2795         }
2796
2797         join_state_t state = {
2798                 .mesh = mesh,
2799                 .sock = -1,
2800         };
2801
2802         ecdsa_t *key = NULL;
2803         ecdsa_t *hiskey = NULL;
2804
2805         //TODO: think of a better name for this variable, or of a different way to tokenize the invitation URL.
2806         char copy[strlen(invitation) + 1];
2807
2808         assert(pthread_mutex_lock(&mesh->mutex) == 0);
2809
2810         //Before doing meshlink_join make sure we are not connected to another mesh
2811         if(mesh->threadstarted) {
2812                 logger(mesh, MESHLINK_ERROR, "Cannot join while started\n");
2813                 meshlink_errno = MESHLINK_EINVAL;
2814                 goto exit;
2815         }
2816
2817         // Refuse to join a mesh if we are already part of one. We are part of one if we know at least one other node.
2818         if(mesh->nodes->count > 1) {
2819                 logger(mesh, MESHLINK_ERROR, "Already part of an existing mesh\n");
2820                 meshlink_errno = MESHLINK_EINVAL;
2821                 goto exit;
2822         }
2823
2824         strcpy(copy, invitation);
2825
2826         // Split the invitation URL into a list of hostname/port tuples, a key hash and a cookie.
2827
2828         char *slash = strchr(copy, '/');
2829
2830         if(!slash) {
2831                 goto invalid;
2832         }
2833
2834         *slash++ = 0;
2835
2836         if(strlen(slash) != 48) {
2837                 goto invalid;
2838         }
2839
2840         char *address = copy;
2841         char *port = NULL;
2842
2843         if(!b64decode(slash, state.hash, 18) || !b64decode(slash + 24, state.cookie, 18)) {
2844                 goto invalid;
2845         }
2846
2847         if(mesh->inviter_commits_first) {
2848                 memcpy(state.cookie + 18, ecdsa_get_public_key(mesh->private_key), 32);
2849         }
2850
2851         // Generate a throw-away key for the invitation.
2852         key = ecdsa_generate();
2853
2854         if(!key) {
2855                 meshlink_errno = MESHLINK_EINTERNAL;
2856                 goto exit;
2857         }
2858
2859         char *b64key = ecdsa_get_base64_public_key(key);
2860         char *comma;
2861
2862         while(address && *address) {
2863                 // We allow commas in the address part to support multiple addresses in one invitation URL.
2864                 comma = strchr(address, ',');
2865
2866                 if(comma) {
2867                         *comma++ = 0;
2868                 }
2869
2870                 // Split of the port
2871                 port = strrchr(address, ':');
2872
2873                 if(!port) {
2874                         goto invalid;
2875                 }
2876
2877                 *port++ = 0;
2878
2879                 // IPv6 address are enclosed in brackets, per RFC 3986
2880                 if(*address == '[') {
2881                         address++;
2882                         char *bracket = strchr(address, ']');
2883
2884                         if(!bracket) {
2885                                 goto invalid;
2886                         }
2887
2888                         *bracket++ = 0;
2889
2890                         if(*bracket) {
2891                                 goto invalid;
2892                         }
2893                 }
2894
2895                 // Connect to the meshlink daemon mentioned in the URL.
2896                 struct addrinfo *ai = adns_blocking_request(mesh, xstrdup(address), xstrdup(port), SOCK_STREAM, 5);
2897
2898                 if(ai) {
2899                         for(struct addrinfo *aip = ai; aip; aip = aip->ai_next) {
2900                                 state.sock = socket_in_netns(aip->ai_family, SOCK_STREAM, IPPROTO_TCP, mesh->netns);
2901
2902                                 if(state.sock == -1) {
2903                                         logger(mesh, MESHLINK_DEBUG, "Could not open socket: %s\n", strerror(errno));
2904                                         meshlink_errno = MESHLINK_ENETWORK;
2905                                         continue;
2906                                 }
2907
2908                                 set_timeout(state.sock, 5000);
2909
2910                                 if(connect(state.sock, aip->ai_addr, aip->ai_addrlen)) {
2911                                         logger(mesh, MESHLINK_DEBUG, "Could not connect to %s port %s: %s\n", address, port, strerror(errno));
2912                                         meshlink_errno = MESHLINK_ENETWORK;
2913                                         closesocket(state.sock);
2914                                         state.sock = -1;
2915                                         continue;
2916                                 }
2917
2918                                 break;
2919                         }
2920
2921                         freeaddrinfo(ai);
2922                 } else {
2923                         meshlink_errno = MESHLINK_ERESOLV;
2924                 }
2925
2926                 if(state.sock != -1 || !comma) {
2927                         break;
2928                 }
2929
2930                 address = comma;
2931         }
2932
2933         if(state.sock == -1) {
2934                 goto exit;
2935         }
2936
2937         logger(mesh, MESHLINK_DEBUG, "Connected to %s port %s...\n", address, port);
2938
2939         // Tell him we have an invitation, and give him our throw-away key.
2940
2941         state.blen = 0;
2942
2943         if(!sendline(state.sock, "0 ?%s %d.%d %s", b64key, PROT_MAJOR, PROT_MINOR, mesh->appname)) {
2944                 logger(mesh, MESHLINK_DEBUG, "Error sending request to %s port %s: %s\n", address, port, strerror(errno));
2945                 meshlink_errno = MESHLINK_ENETWORK;
2946                 goto exit;
2947         }
2948
2949         free(b64key);
2950
2951         char hisname[4096] = "";
2952         int code, hismajor, hisminor = 0;
2953
2954         if(!recvline(&state) || sscanf(state.line, "%d %s %d.%d", &code, hisname, &hismajor, &hisminor) < 3 || code != 0 || hismajor != PROT_MAJOR || !check_id(hisname) || !recvline(&state) || !rstrip(state.line) || sscanf(state.line, "%d ", &code) != 1 || code != ACK || strlen(state.line) < 3) {
2955                 logger(mesh, MESHLINK_DEBUG, "Cannot read greeting from peer\n");
2956                 meshlink_errno = MESHLINK_ENETWORK;
2957                 goto exit;
2958         }
2959
2960         // Check if the hash of the key he gave us matches the hash in the URL.
2961         char *fingerprint = state.line + 2;
2962         char hishash[64];
2963
2964         if(sha512(fingerprint, strlen(fingerprint), hishash)) {
2965                 logger(mesh, MESHLINK_DEBUG, "Could not create hash\n%s\n", state.line + 2);
2966                 meshlink_errno = MESHLINK_EINTERNAL;
2967                 goto exit;
2968         }
2969
2970         if(memcmp(hishash, state.hash, 18)) {
2971                 logger(mesh, MESHLINK_DEBUG, "Peer has an invalid key!\n%s\n", state.line + 2);
2972                 meshlink_errno = MESHLINK_EPEER;
2973                 goto exit;
2974         }
2975
2976         hiskey = ecdsa_set_base64_public_key(fingerprint);
2977
2978         if(!hiskey) {
2979                 meshlink_errno = MESHLINK_EINTERNAL;
2980                 goto exit;
2981         }
2982
2983         // Start an SPTPS session
2984         if(!sptps_start(&state.sptps, &state, true, false, key, hiskey, meshlink_invitation_label, sizeof(meshlink_invitation_label), invitation_send, invitation_receive)) {
2985                 meshlink_errno = MESHLINK_EINTERNAL;
2986                 goto exit;
2987         }
2988
2989         // Feed rest of input buffer to SPTPS
2990         if(!sptps_receive_data(&state.sptps, state.buffer, state.blen)) {
2991                 meshlink_errno = MESHLINK_EPEER;
2992                 goto exit;
2993         }
2994
2995         ssize_t len;
2996         logger(mesh, MESHLINK_DEBUG, "Starting invitation recv loop: %d %zu\n", state.sock, sizeof(state.line));
2997
2998         while((len = recv(state.sock, state.line, sizeof(state.line), 0))) {
2999                 if(len < 0) {
3000                         if(errno == EINTR) {
3001                                 continue;
3002                         }
3003
3004                         logger(mesh, MESHLINK_DEBUG, "Error reading data from %s port %s: %s\n", address, port, strerror(errno));
3005                         meshlink_errno = MESHLINK_ENETWORK;
3006                         goto exit;
3007                 }
3008
3009                 if(!sptps_receive_data(&state.sptps, state.line, len)) {
3010                         meshlink_errno = MESHLINK_EPEER;
3011                         goto exit;
3012                 }
3013         }
3014
3015         if(!state.success) {
3016                 logger(mesh, MESHLINK_DEBUG, "Connection closed by peer, invitation cancelled.\n");
3017                 meshlink_errno = MESHLINK_EPEER;
3018                 goto exit;
3019         }
3020
3021         sptps_stop(&state.sptps);
3022         ecdsa_free(hiskey);
3023         ecdsa_free(key);
3024         closesocket(state.sock);
3025
3026         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
3027         return true;
3028
3029 invalid:
3030         logger(mesh, MESHLINK_DEBUG, "Invalid invitation URL\n");
3031         meshlink_errno = MESHLINK_EINVAL;
3032 exit:
3033         sptps_stop(&state.sptps);
3034         ecdsa_free(hiskey);
3035         ecdsa_free(key);
3036
3037         if(state.sock != -1) {
3038                 closesocket(state.sock);
3039         }
3040
3041         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
3042         return false;
3043 }
3044
3045 char *meshlink_export(meshlink_handle_t *mesh) {
3046         if(!mesh) {
3047                 meshlink_errno = MESHLINK_EINVAL;
3048                 return NULL;
3049         }
3050
3051         // Create a config file on the fly.
3052
3053         uint8_t buf[4096];
3054         packmsg_output_t out = {buf, sizeof(buf)};
3055         packmsg_add_uint32(&out, MESHLINK_CONFIG_VERSION);
3056         packmsg_add_str(&out, mesh->name);
3057         packmsg_add_str(&out, CORE_MESH);
3058
3059         assert(pthread_mutex_lock(&mesh->mutex) == 0);
3060
3061         packmsg_add_int32(&out, mesh->self->devclass);
3062         packmsg_add_bool(&out, mesh->self->status.blacklisted);
3063         packmsg_add_bin(&out, ecdsa_get_public_key(mesh->private_key), 32);
3064
3065         if(mesh->self->canonical_address && !strchr(mesh->self->canonical_address, ' ')) {
3066                 char *canonical_address = NULL;
3067                 xasprintf(&canonical_address, "%s %s", mesh->self->canonical_address, mesh->myport);
3068                 packmsg_add_str(&out, canonical_address);
3069                 free(canonical_address);
3070         } else {
3071                 packmsg_add_str(&out, mesh->self->canonical_address ? mesh->self->canonical_address : "");
3072         }
3073
3074         uint32_t count = 0;
3075
3076         for(uint32_t i = 0; i < MAX_RECENT; i++) {
3077                 if(mesh->self->recent[i].sa.sa_family) {
3078                         count++;
3079                 } else {
3080                         break;
3081                 }
3082         }
3083
3084         packmsg_add_array(&out, count);
3085
3086         for(uint32_t i = 0; i < count; i++) {
3087                 packmsg_add_sockaddr(&out, &mesh->self->recent[i]);
3088         }
3089
3090         packmsg_add_int64(&out, 0);
3091         packmsg_add_int64(&out, 0);
3092
3093         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
3094
3095         if(!packmsg_output_ok(&out)) {
3096                 logger(mesh, MESHLINK_DEBUG, "Error creating export data\n");
3097                 meshlink_errno = MESHLINK_EINTERNAL;
3098                 return NULL;
3099         }
3100
3101         // Prepare a base64-encoded packmsg array containing our config file
3102
3103         uint32_t len = packmsg_output_size(&out, buf);
3104         uint32_t len2 = ((len + 4) * 4) / 3 + 4;
3105         uint8_t *buf2 = xmalloc(len2);
3106         packmsg_output_t out2 = {buf2, len2};
3107         packmsg_add_array(&out2, 1);
3108         packmsg_add_bin(&out2, buf, packmsg_output_size(&out, buf));
3109
3110         if(!packmsg_output_ok(&out2)) {
3111                 logger(mesh, MESHLINK_DEBUG, "Error creating export data\n");
3112                 meshlink_errno = MESHLINK_EINTERNAL;
3113                 free(buf2);
3114                 return NULL;
3115         }
3116
3117         b64encode_urlsafe(buf2, (char *)buf2, packmsg_output_size(&out2, buf2));
3118
3119         return (char *)buf2;
3120 }
3121
3122 bool meshlink_import(meshlink_handle_t *mesh, const char *data) {
3123         if(!mesh || !data) {
3124                 meshlink_errno = MESHLINK_EINVAL;
3125                 return false;
3126         }
3127
3128         size_t datalen = strlen(data);
3129         uint8_t *buf = xmalloc(datalen);
3130         int buflen = b64decode(data, buf, datalen);
3131
3132         if(!buflen) {
3133                 logger(mesh, MESHLINK_DEBUG, "Invalid data\n");
3134                 meshlink_errno = MESHLINK_EPEER;
3135                 return false;
3136         }
3137
3138         packmsg_input_t in = {buf, buflen};
3139         uint32_t count = packmsg_get_array(&in);
3140
3141         if(!count) {
3142                 logger(mesh, MESHLINK_DEBUG, "Invalid data\n");
3143                 meshlink_errno = MESHLINK_EPEER;
3144                 return false;
3145         }
3146
3147         assert(pthread_mutex_lock(&mesh->mutex) == 0);
3148
3149         while(count--) {
3150                 const void *data2;
3151                 uint32_t len2 = packmsg_get_bin_raw(&in, &data2);
3152
3153                 if(!len2) {
3154                         break;
3155                 }
3156
3157                 packmsg_input_t in2 = {data2, len2};
3158                 uint32_t version = packmsg_get_uint32(&in2);
3159                 char *name = packmsg_get_str_dup(&in2);
3160
3161                 if(!packmsg_input_ok(&in2) || version != MESHLINK_CONFIG_VERSION || !check_id(name)) {
3162                         free(name);
3163                         packmsg_input_invalidate(&in);
3164                         break;
3165                 }
3166
3167                 if(!check_id(name)) {
3168                         free(name);
3169                         break;
3170                 }
3171
3172                 node_t *n = lookup_node(mesh, name);
3173
3174                 if(n) {
3175                         logger(mesh, MESHLINK_DEBUG, "Node %s already exists, not importing\n", name);
3176                         free(name);
3177                         continue;
3178                 }
3179
3180                 n = new_node();
3181                 n->name = name;
3182
3183                 config_t config = {data2, len2};
3184
3185                 if(!node_read_from_config(mesh, n, &config)) {
3186                         free_node(n);
3187                         packmsg_input_invalidate(&in);
3188                         break;
3189                 }
3190
3191                 /* Clear the reachability times, since we ourself have never seen these nodes yet */
3192                 n->last_reachable = 0;
3193                 n->last_unreachable = 0;
3194
3195                 if(!node_write_config(mesh, n)) {
3196                         free_node(n);
3197                         return false;
3198                 }
3199
3200                 node_add(mesh, n);
3201         }
3202
3203         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
3204
3205         free(buf);
3206
3207         if(!packmsg_done(&in)) {
3208                 logger(mesh, MESHLINK_ERROR, "Invalid data\n");
3209                 meshlink_errno = MESHLINK_EPEER;
3210                 return false;
3211         }
3212
3213         if(!config_sync(mesh, "current")) {
3214                 return false;
3215         }
3216
3217         return true;
3218 }
3219
3220 static bool blacklist(meshlink_handle_t *mesh, node_t *n) {
3221         if(n == mesh->self) {
3222                 logger(mesh, MESHLINK_ERROR, "%s blacklisting itself?\n", n->name);
3223                 meshlink_errno = MESHLINK_EINVAL;
3224                 return false;
3225         }
3226
3227         if(n->status.blacklisted) {
3228                 logger(mesh, MESHLINK_DEBUG, "Node %s already blacklisted\n", n->name);
3229                 return true;
3230         }
3231
3232         n->status.blacklisted = true;
3233
3234         /* Immediately shut down any connections we have with the blacklisted node.
3235          * We can't call terminate_connection(), because we might be called from a callback function.
3236          */
3237         for list_each(connection_t, c, mesh->connections) {
3238                 if(c->node == n) {
3239                         shutdown(c->socket, SHUT_RDWR);
3240                 }
3241         }
3242
3243         utcp_abort_all_connections(n->utcp);
3244
3245         n->mtu = 0;
3246         n->minmtu = 0;
3247         n->maxmtu = MTU;
3248         n->mtuprobes = 0;
3249         n->status.udp_confirmed = false;
3250
3251         if(n->status.reachable) {
3252                 n->last_unreachable = time(NULL);
3253         }
3254
3255         /* Graph updates will suppress status updates for blacklisted nodes, so we need to
3256          * manually call the status callback if necessary.
3257          */
3258         if(n->status.reachable && mesh->node_status_cb) {
3259                 mesh->node_status_cb(mesh, (meshlink_node_t *)n, false);
3260         }
3261
3262         return node_write_config(mesh, n) && config_sync(mesh, "current");
3263 }
3264
3265 bool meshlink_blacklist(meshlink_handle_t *mesh, meshlink_node_t *node) {
3266         if(!mesh || !node) {
3267                 meshlink_errno = MESHLINK_EINVAL;
3268                 return false;
3269         }
3270
3271         assert(pthread_mutex_lock(&mesh->mutex) == 0);
3272
3273         if(!blacklist(mesh, (node_t *)node)) {
3274                 assert(pthread_mutex_unlock(&mesh->mutex) == 0);
3275                 return false;
3276         }
3277
3278         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
3279
3280         logger(mesh, MESHLINK_DEBUG, "Blacklisted %s.\n", node->name);
3281         return true;
3282 }
3283
3284 bool meshlink_blacklist_by_name(meshlink_handle_t *mesh, const char *name) {
3285         if(!mesh || !name) {
3286                 meshlink_errno = MESHLINK_EINVAL;
3287                 return false;
3288         }
3289
3290         assert(pthread_mutex_lock(&mesh->mutex) == 0);
3291
3292         node_t *n = lookup_node(mesh, (char *)name);
3293
3294         if(!n) {
3295                 n = new_node();
3296                 n->name = xstrdup(name);
3297                 node_add(mesh, n);
3298         }
3299
3300         if(!blacklist(mesh, (node_t *)n)) {
3301                 assert(pthread_mutex_unlock(&mesh->mutex) == 0);
3302                 return false;
3303         }
3304
3305         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
3306
3307         logger(mesh, MESHLINK_DEBUG, "Blacklisted %s.\n", name);
3308         return true;
3309 }
3310
3311 static bool whitelist(meshlink_handle_t *mesh, node_t *n) {
3312         if(n == mesh->self) {
3313                 logger(mesh, MESHLINK_ERROR, "%s whitelisting itself?\n", n->name);
3314                 meshlink_errno = MESHLINK_EINVAL;
3315                 return false;
3316         }
3317
3318         if(!n->status.blacklisted) {
3319                 logger(mesh, MESHLINK_DEBUG, "Node %s was already whitelisted\n", n->name);
3320                 return true;
3321         }
3322
3323         n->status.blacklisted = false;
3324
3325         if(n->status.reachable) {
3326                 n->last_reachable = time(NULL);
3327                 update_node_status(mesh, n);
3328         }
3329
3330         return node_write_config(mesh, n) && config_sync(mesh, "current");
3331 }
3332
3333 bool meshlink_whitelist(meshlink_handle_t *mesh, meshlink_node_t *node) {
3334         if(!mesh || !node) {
3335                 meshlink_errno = MESHLINK_EINVAL;
3336                 return false;
3337         }
3338
3339         assert(pthread_mutex_lock(&mesh->mutex) == 0);
3340
3341         if(!whitelist(mesh, (node_t *)node)) {
3342                 assert(pthread_mutex_unlock(&mesh->mutex) == 0);
3343                 return false;
3344         }
3345
3346         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
3347
3348         logger(mesh, MESHLINK_DEBUG, "Whitelisted %s.\n", node->name);
3349         return true;
3350 }
3351
3352 bool meshlink_whitelist_by_name(meshlink_handle_t *mesh, const char *name) {
3353         if(!mesh || !name) {
3354                 meshlink_errno = MESHLINK_EINVAL;
3355                 return false;
3356         }
3357
3358         assert(pthread_mutex_lock(&mesh->mutex) == 0);
3359
3360         node_t *n = lookup_node(mesh, (char *)name);
3361
3362         if(!n) {
3363                 n = new_node();
3364                 n->name = xstrdup(name);
3365                 node_add(mesh, n);
3366         }
3367
3368         if(!whitelist(mesh, (node_t *)n)) {
3369                 assert(pthread_mutex_unlock(&mesh->mutex) == 0);
3370                 return false;
3371         }
3372
3373         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
3374
3375         logger(mesh, MESHLINK_DEBUG, "Whitelisted %s.\n", name);
3376         return true;
3377 }
3378
3379 void meshlink_set_default_blacklist(meshlink_handle_t *mesh, bool blacklist) {
3380         mesh->default_blacklist = blacklist;
3381 }
3382
3383 bool meshlink_forget_node(meshlink_handle_t *mesh, meshlink_node_t *node) {
3384         if(!mesh || !node) {
3385                 meshlink_errno = MESHLINK_EINVAL;
3386                 return false;
3387         }
3388
3389         node_t *n = (node_t *)node;
3390
3391         assert(pthread_mutex_lock(&mesh->mutex) == 0);
3392
3393         /* Check that the node is not reachable */
3394         if(n->status.reachable || n->connection) {
3395                 assert(pthread_mutex_unlock(&mesh->mutex) == 0);
3396                 logger(mesh, MESHLINK_WARNING, "Could not forget %s: still reachable", n->name);
3397                 return false;
3398         }
3399
3400         /* Check that we don't have any active UTCP connections */
3401         if(n->utcp && utcp_is_active(n->utcp)) {
3402                 assert(pthread_mutex_unlock(&mesh->mutex) == 0);
3403                 logger(mesh, MESHLINK_WARNING, "Could not forget %s: active UTCP connections", n->name);
3404                 return false;
3405         }
3406
3407         /* Check that we have no active connections to this node */
3408         for list_each(connection_t, c, mesh->connections) {
3409                 if(c->node == n) {
3410                         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
3411                         logger(mesh, MESHLINK_WARNING, "Could not forget %s: active connection", n->name);
3412                         return false;
3413                 }
3414         }
3415
3416         /* Remove any pending outgoings to this node */
3417         if(mesh->outgoings) {
3418                 for list_each(outgoing_t, outgoing, mesh->outgoings) {
3419                         if(outgoing->node == n) {
3420                                 list_delete_node(mesh->outgoings, list_node);
3421                         }
3422                 }
3423         }
3424
3425         /* Delete the config file for this node */
3426         if(!config_delete(mesh, "current", n->name)) {
3427                 assert(pthread_mutex_unlock(&mesh->mutex) == 0);
3428                 return false;
3429         }
3430
3431         /* Delete the node struct and any remaining edges referencing this node */
3432         node_del(mesh, n);
3433
3434         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
3435
3436         return config_sync(mesh, "current");
3437 }
3438
3439 /* Hint that a hostname may be found at an address
3440  * See header file for detailed comment.
3441  */
3442 void meshlink_hint_address(meshlink_handle_t *mesh, meshlink_node_t *node, const struct sockaddr *addr) {
3443         if(!mesh || !node || !addr) {
3444                 meshlink_errno = EINVAL;
3445                 return;
3446         }
3447
3448         assert(pthread_mutex_lock(&mesh->mutex) == 0);
3449
3450         node_t *n = (node_t *)node;
3451
3452         if(node_add_recent_address(mesh, n, (sockaddr_t *)addr)) {
3453                 if(!node_write_config(mesh, n)) {
3454                         logger(mesh, MESHLINK_DEBUG, "Could not update %s\n", n->name);
3455                 }
3456         }
3457
3458         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
3459         // @TODO do we want to fire off a connection attempt right away?
3460 }
3461
3462 static bool channel_pre_accept(struct utcp *utcp, uint16_t port) {
3463         (void)port;
3464         node_t *n = utcp->priv;
3465         meshlink_handle_t *mesh = n->mesh;
3466         return mesh->channel_accept_cb;
3467 }
3468
3469 /* Finish one AIO buffer, return true if the channel is still open. */
3470 static bool aio_finish_one(meshlink_handle_t *mesh, meshlink_channel_t *channel, meshlink_aio_buffer_t **head) {
3471         meshlink_aio_buffer_t *aio = *head;
3472         *head = aio->next;
3473
3474         if(channel->c) {
3475                 channel->in_callback = true;
3476
3477                 if(aio->data) {
3478                         if(aio->cb.buffer) {
3479                                 aio->cb.buffer(mesh, channel, aio->data, aio->done, aio->priv);
3480                         }
3481                 } else {
3482                         if(aio->cb.fd) {
3483                                 aio->cb.fd(mesh, channel, aio->fd, aio->done, aio->priv);
3484                         }
3485                 }
3486
3487                 channel->in_callback = false;
3488
3489                 if(!channel->c) {
3490                         free(aio);
3491                         free(channel);
3492                         return false;
3493                 }
3494         }
3495
3496         free(aio);
3497         return true;
3498 }
3499
3500 /* Finish all AIO buffers, return true if the channel is still open. */
3501 static bool aio_abort(meshlink_handle_t *mesh, meshlink_channel_t *channel, meshlink_aio_buffer_t **head) {
3502         while(*head) {
3503                 if(!aio_finish_one(mesh, channel, head)) {
3504                         return false;
3505                 }
3506         }
3507
3508         return true;
3509 }
3510
3511 static ssize_t channel_recv(struct utcp_connection *connection, const void *data, size_t len) {
3512         meshlink_channel_t *channel = connection->priv;
3513
3514         if(!channel) {
3515                 abort();
3516         }
3517
3518         node_t *n = channel->node;
3519         meshlink_handle_t *mesh = n->mesh;
3520
3521         if(n->status.destroyed) {
3522                 meshlink_channel_close(mesh, channel);
3523                 return len;
3524         }
3525
3526         const char *p = data;
3527         size_t left = len;
3528
3529         while(channel->aio_receive) {
3530                 if(!len) {
3531                         /* This receive callback signalled an error, abort all outstanding AIO buffers. */
3532                         if(!aio_abort(mesh, channel, &channel->aio_receive)) {
3533                                 return len;
3534                         }
3535
3536                         break;
3537                 }
3538
3539                 meshlink_aio_buffer_t *aio = channel->aio_receive;
3540                 size_t todo = aio->len - aio->done;
3541
3542                 if(todo > left) {
3543                         todo = left;
3544                 }
3545
3546                 if(aio->data) {
3547                         memcpy((char *)aio->data + aio->done, p, todo);
3548                 } else {
3549                         ssize_t result = write(aio->fd, p, todo);
3550
3551                         if(result <= 0) {
3552                                 if(result < 0 && errno == EINTR) {
3553                                         continue;
3554                                 }
3555
3556                                 /* Writing to fd failed, cancel just this AIO buffer. */
3557                                 logger(mesh, MESHLINK_ERROR, "Writing to AIO fd %d failed: %s", aio->fd, strerror(errno));
3558
3559                                 if(!aio_finish_one(mesh, channel, &channel->aio_receive)) {
3560                                         return len;
3561                                 }
3562
3563                                 continue;
3564                         }
3565
3566                         todo = result;
3567                 }
3568
3569                 aio->done += todo;
3570                 p += todo;
3571                 left -= todo;
3572
3573                 if(aio->done == aio->len) {
3574                         if(!aio_finish_one(mesh, channel, &channel->aio_receive)) {
3575                                 return len;
3576                         }
3577                 }
3578
3579                 if(!left) {
3580                         return len;
3581                 }
3582         }
3583
3584         if(channel->receive_cb) {
3585                 channel->receive_cb(mesh, channel, p, left);
3586         }
3587
3588         return len;
3589 }
3590
3591 static void channel_accept(struct utcp_connection *utcp_connection, uint16_t port) {
3592         node_t *n = utcp_connection->utcp->priv;
3593
3594         if(!n) {
3595                 abort();
3596         }
3597
3598         meshlink_handle_t *mesh = n->mesh;
3599
3600         if(!mesh->channel_accept_cb) {
3601                 return;
3602         }
3603
3604         meshlink_channel_t *channel = xzalloc(sizeof(*channel));
3605         channel->node = n;
3606         channel->c = utcp_connection;
3607
3608         if(mesh->channel_accept_cb(mesh, channel, port, NULL, 0)) {
3609                 utcp_accept(utcp_connection, channel_recv, channel);
3610         } else {
3611                 free(channel);
3612         }
3613 }
3614
3615 static void channel_retransmit(struct utcp_connection *utcp_connection) {
3616         node_t *n = utcp_connection->utcp->priv;
3617         meshlink_handle_t *mesh = n->mesh;
3618
3619         if(n->mtuprobes == 31) {
3620                 timeout_set(&mesh->loop, &n->mtutimeout, &(struct timespec) {
3621                         0, 0
3622                 });
3623         }
3624 }
3625
3626 static ssize_t channel_send(struct utcp *utcp, const void *data, size_t len) {
3627         node_t *n = utcp->priv;
3628
3629         if(n->status.destroyed) {
3630                 return -1;
3631         }
3632
3633         meshlink_handle_t *mesh = n->mesh;
3634         return meshlink_send_immediate(mesh, (meshlink_node_t *)n, data, len) ? (ssize_t)len : -1;
3635 }
3636
3637 void meshlink_set_channel_receive_cb(meshlink_handle_t *mesh, meshlink_channel_t *channel, meshlink_channel_receive_cb_t cb) {
3638         if(!mesh || !channel) {
3639                 meshlink_errno = MESHLINK_EINVAL;
3640                 return;
3641         }
3642
3643         channel->receive_cb = cb;
3644 }
3645
3646 static void channel_receive(meshlink_handle_t *mesh, meshlink_node_t *source, const void *data, size_t len) {
3647         (void)mesh;
3648         node_t *n = (node_t *)source;
3649
3650         if(!n->utcp) {
3651                 abort();
3652         }
3653
3654         utcp_recv(n->utcp, data, len);
3655 }
3656
3657 static void channel_poll(struct utcp_connection *connection, size_t len) {
3658         meshlink_channel_t *channel = connection->priv;
3659
3660         if(!channel) {
3661                 abort();
3662         }
3663
3664         node_t *n = channel->node;
3665         meshlink_handle_t *mesh = n->mesh;
3666
3667         while(channel->aio_send) {
3668                 if(!len) {
3669                         /* This poll callback signalled an error, abort all outstanding AIO buffers. */
3670                         if(!aio_abort(mesh, channel, &channel->aio_send)) {
3671                                 return;
3672                         }
3673
3674                         break;
3675                 }
3676
3677                 /* We have at least one AIO buffer. Send as much as possible from the buffers. */
3678                 meshlink_aio_buffer_t *aio = channel->aio_send;
3679                 size_t todo = aio->len - aio->done;
3680                 ssize_t sent;
3681
3682                 if(todo > len) {
3683                         todo = len;
3684                 }
3685
3686                 if(aio->data) {
3687                         sent = utcp_send(connection, (char *)aio->data + aio->done, todo);
3688                 } else {
3689                         /* Limit the amount we read at once to avoid stack overflows */
3690                         if(todo > 65536) {
3691                                 todo = 65536;
3692                         }
3693
3694                         char buf[todo];
3695                         ssize_t result = read(aio->fd, buf, todo);
3696
3697                         if(result > 0) {
3698                                 todo = result;
3699                                 sent = utcp_send(connection, buf, todo);
3700                         } else {
3701                                 if(result < 0 && errno == EINTR) {
3702                                         continue;
3703                                 }
3704
3705                                 /* Reading from fd failed, cancel just this AIO buffer. */
3706                                 if(result != 0) {
3707                                         logger(mesh, MESHLINK_ERROR, "Reading from AIO fd %d failed: %s", aio->fd, strerror(errno));
3708                                 }
3709
3710                                 if(!aio_finish_one(mesh, channel, &channel->aio_send)) {
3711                                         return;
3712                                 }
3713
3714                                 continue;
3715                         }
3716                 }
3717
3718                 if(sent != (ssize_t)todo) {
3719                         /* We should never get a partial send at this point */
3720                         assert(sent <= 0);
3721
3722                         /* Sending failed, abort all outstanding AIO buffers and send a poll callback. */
3723                         if(!aio_abort(mesh, channel, &channel->aio_send)) {
3724                                 return;
3725                         }
3726
3727                         len = 0;
3728                         break;
3729                 }
3730
3731                 aio->done += sent;
3732                 len -= sent;
3733
3734                 /* If we didn't finish this buffer, exit early. */
3735                 if(aio->done < aio->len) {
3736                         return;
3737                 }
3738
3739                 /* Signal completion of this buffer, and go to the next one. */
3740                 if(!aio_finish_one(mesh, channel, &channel->aio_send)) {
3741                         return;
3742                 }
3743
3744                 if(!len) {
3745                         return;
3746                 }
3747         }
3748
3749         if(channel->poll_cb) {
3750                 channel->poll_cb(mesh, channel, len);
3751         } else {
3752                 utcp_set_poll_cb(connection, NULL);
3753         }
3754 }
3755
3756 void meshlink_set_channel_poll_cb(meshlink_handle_t *mesh, meshlink_channel_t *channel, meshlink_channel_poll_cb_t cb) {
3757         if(!mesh || !channel) {
3758                 meshlink_errno = MESHLINK_EINVAL;
3759                 return;
3760         }
3761
3762         assert(pthread_mutex_lock(&mesh->mutex) == 0);
3763         channel->poll_cb = cb;
3764         utcp_set_poll_cb(channel->c, (cb || channel->aio_send) ? channel_poll : NULL);
3765         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
3766 }
3767
3768 void meshlink_set_channel_accept_cb(meshlink_handle_t *mesh, meshlink_channel_accept_cb_t cb) {
3769         if(!mesh) {
3770                 meshlink_errno = MESHLINK_EINVAL;
3771                 return;
3772         }
3773
3774         assert(pthread_mutex_lock(&mesh->mutex) == 0);
3775         mesh->channel_accept_cb = cb;
3776         mesh->receive_cb = channel_receive;
3777
3778         for splay_each(node_t, n, mesh->nodes) {
3779                 if(!n->utcp && n != mesh->self) {
3780                         n->utcp = utcp_init(channel_accept, channel_pre_accept, channel_send, n);
3781                         utcp_set_mtu(n->utcp, n->mtu - sizeof(meshlink_packethdr_t));
3782                         utcp_set_retransmit_cb(n->utcp, channel_retransmit);
3783                 }
3784         }
3785
3786         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
3787 }
3788
3789 void meshlink_set_channel_sndbuf(meshlink_handle_t *mesh, meshlink_channel_t *channel, size_t size) {
3790         (void)mesh;
3791
3792         if(!channel) {
3793                 meshlink_errno = MESHLINK_EINVAL;
3794                 return;
3795         }
3796
3797         assert(pthread_mutex_lock(&mesh->mutex) == 0);
3798         utcp_set_sndbuf(channel->c, size);
3799         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
3800 }
3801
3802 void meshlink_set_channel_rcvbuf(meshlink_handle_t *mesh, meshlink_channel_t *channel, size_t size) {
3803         (void)mesh;
3804
3805         if(!channel) {
3806                 meshlink_errno = MESHLINK_EINVAL;
3807                 return;
3808         }
3809
3810         assert(pthread_mutex_lock(&mesh->mutex) == 0);
3811         utcp_set_rcvbuf(channel->c, size);
3812         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
3813 }
3814
3815 meshlink_channel_t *meshlink_channel_open_ex(meshlink_handle_t *mesh, meshlink_node_t *node, uint16_t port, meshlink_channel_receive_cb_t cb, const void *data, size_t len, uint32_t flags) {
3816         if(data && len) {
3817                 abort();        // TODO: handle non-NULL data
3818         }
3819
3820         if(!mesh || !node) {
3821                 meshlink_errno = MESHLINK_EINVAL;
3822                 return NULL;
3823         }
3824
3825         assert(pthread_mutex_lock(&mesh->mutex) == 0);
3826
3827         node_t *n = (node_t *)node;
3828
3829         if(!n->utcp) {
3830                 n->utcp = utcp_init(channel_accept, channel_pre_accept, channel_send, n);
3831                 utcp_set_mtu(n->utcp, n->mtu - sizeof(meshlink_packethdr_t));
3832                 utcp_set_retransmit_cb(n->utcp, channel_retransmit);
3833                 mesh->receive_cb = channel_receive;
3834
3835                 if(!n->utcp) {
3836                         meshlink_errno = errno == ENOMEM ? MESHLINK_ENOMEM : MESHLINK_EINTERNAL;
3837                         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
3838                         return NULL;
3839                 }
3840         }
3841
3842         if(n->status.blacklisted) {
3843                 logger(mesh, MESHLINK_ERROR, "Cannot open a channel with blacklisted node\n");
3844                 meshlink_errno = MESHLINK_EBLACKLISTED;
3845                 assert(pthread_mutex_unlock(&mesh->mutex) == 0);
3846                 return NULL;
3847         }
3848
3849         meshlink_channel_t *channel = xzalloc(sizeof(*channel));
3850         channel->node = n;
3851         channel->receive_cb = cb;
3852
3853         if(data && !len) {
3854                 channel->priv = (void *)data;
3855         }
3856
3857         channel->c = utcp_connect_ex(n->utcp, port, channel_recv, channel, flags);
3858
3859         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
3860
3861         if(!channel->c) {
3862                 meshlink_errno = errno == ENOMEM ? MESHLINK_ENOMEM : MESHLINK_EINTERNAL;
3863                 free(channel);
3864                 return NULL;
3865         }
3866
3867         return channel;
3868 }
3869
3870 meshlink_channel_t *meshlink_channel_open(meshlink_handle_t *mesh, meshlink_node_t *node, uint16_t port, meshlink_channel_receive_cb_t cb, const void *data, size_t len) {
3871         return meshlink_channel_open_ex(mesh, node, port, cb, data, len, MESHLINK_CHANNEL_TCP);
3872 }
3873
3874 void meshlink_channel_shutdown(meshlink_handle_t *mesh, meshlink_channel_t *channel, int direction) {
3875         if(!mesh || !channel) {
3876                 meshlink_errno = MESHLINK_EINVAL;
3877                 return;
3878         }
3879
3880         assert(pthread_mutex_lock(&mesh->mutex) == 0);
3881         utcp_shutdown(channel->c, direction);
3882         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
3883 }
3884
3885 void meshlink_channel_close(meshlink_handle_t *mesh, meshlink_channel_t *channel) {
3886         if(!mesh || !channel) {
3887                 meshlink_errno = MESHLINK_EINVAL;
3888                 return;
3889         }
3890
3891         assert(pthread_mutex_lock(&mesh->mutex) == 0);
3892
3893         if(channel->c) {
3894                 utcp_close(channel->c);
3895                 channel->c = NULL;
3896
3897                 /* Clean up any outstanding AIO buffers. */
3898                 aio_abort(mesh, channel, &channel->aio_send);
3899                 aio_abort(mesh, channel, &channel->aio_receive);
3900         }
3901
3902         if(!channel->in_callback) {
3903                 free(channel);
3904         }
3905
3906         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
3907 }
3908
3909 ssize_t meshlink_channel_send(meshlink_handle_t *mesh, meshlink_channel_t *channel, const void *data, size_t len) {
3910         if(!mesh || !channel) {
3911                 meshlink_errno = MESHLINK_EINVAL;
3912                 return -1;
3913         }
3914
3915         if(!len) {
3916                 return 0;
3917         }
3918
3919         if(!data) {
3920                 meshlink_errno = MESHLINK_EINVAL;
3921                 return -1;
3922         }
3923
3924         // TODO: more finegrained locking.
3925         // Ideally we want to put the data into the UTCP connection's send buffer.
3926         // Then, preferably only if there is room in the receiver window,
3927         // kick the meshlink thread to go send packets.
3928
3929         ssize_t retval;
3930
3931         assert(pthread_mutex_lock(&mesh->mutex) == 0);
3932
3933         /* Disallow direct calls to utcp_send() while we still have AIO active. */
3934         if(channel->aio_send) {
3935                 retval = 0;
3936         } else {
3937                 retval = utcp_send(channel->c, data, len);
3938         }
3939
3940         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
3941
3942         if(retval < 0) {
3943                 meshlink_errno = MESHLINK_ENETWORK;
3944         }
3945
3946         return retval;
3947 }
3948
3949 bool meshlink_channel_aio_send(meshlink_handle_t *mesh, meshlink_channel_t *channel, const void *data, size_t len, meshlink_aio_cb_t cb, void *priv) {
3950         if(!mesh || !channel) {
3951                 meshlink_errno = MESHLINK_EINVAL;
3952                 return false;
3953         }
3954
3955         if(!len || !data) {
3956                 meshlink_errno = MESHLINK_EINVAL;
3957                 return false;
3958         }
3959
3960         meshlink_aio_buffer_t *aio = xzalloc(sizeof(*aio));
3961         aio->data = data;
3962         aio->len = len;
3963         aio->cb.buffer = cb;
3964         aio->priv = priv;
3965
3966         assert(pthread_mutex_lock(&mesh->mutex) == 0);
3967
3968         /* Append the AIO buffer descriptor to the end of the chain */
3969         meshlink_aio_buffer_t **p = &channel->aio_send;
3970
3971         while(*p) {
3972                 p = &(*p)->next;
3973         }
3974
3975         *p = aio;
3976
3977         /* Ensure the poll callback is set, and call it right now to push data if possible */
3978         utcp_set_poll_cb(channel->c, channel_poll);
3979         size_t todo = MIN(len, utcp_get_rcvbuf_free(channel->c));
3980
3981         if(todo) {
3982                 channel_poll(channel->c, todo);
3983         }
3984
3985         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
3986
3987         return true;
3988 }
3989
3990 bool meshlink_channel_aio_fd_send(meshlink_handle_t *mesh, meshlink_channel_t *channel, int fd, size_t len, meshlink_aio_fd_cb_t cb, void *priv) {
3991         if(!mesh || !channel) {
3992                 meshlink_errno = MESHLINK_EINVAL;
3993                 return false;
3994         }
3995
3996         if(!len || fd == -1) {
3997                 meshlink_errno = MESHLINK_EINVAL;
3998                 return false;
3999         }
4000
4001         meshlink_aio_buffer_t *aio = xzalloc(sizeof(*aio));
4002         aio->fd = fd;
4003         aio->len = len;
4004         aio->cb.fd = cb;
4005         aio->priv = priv;
4006
4007         assert(pthread_mutex_lock(&mesh->mutex) == 0);
4008
4009         /* Append the AIO buffer descriptor to the end of the chain */
4010         meshlink_aio_buffer_t **p = &channel->aio_send;
4011
4012         while(*p) {
4013                 p = &(*p)->next;
4014         }
4015
4016         *p = aio;
4017
4018         /* Ensure the poll callback is set, and call it right now to push data if possible */
4019         utcp_set_poll_cb(channel->c, channel_poll);
4020         size_t left = utcp_get_rcvbuf_free(channel->c);
4021
4022         if(left) {
4023                 channel_poll(channel->c, left);
4024         }
4025
4026         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
4027
4028         return true;
4029 }
4030
4031 bool meshlink_channel_aio_receive(meshlink_handle_t *mesh, meshlink_channel_t *channel, const void *data, size_t len, meshlink_aio_cb_t cb, void *priv) {
4032         if(!mesh || !channel) {
4033                 meshlink_errno = MESHLINK_EINVAL;
4034                 return false;
4035         }
4036
4037         if(!len || !data) {
4038                 meshlink_errno = MESHLINK_EINVAL;
4039                 return false;
4040         }
4041
4042         meshlink_aio_buffer_t *aio = xzalloc(sizeof(*aio));
4043         aio->data = data;
4044         aio->len = len;
4045         aio->cb.buffer = cb;
4046         aio->priv = priv;
4047
4048         assert(pthread_mutex_lock(&mesh->mutex) == 0);
4049
4050         /* Append the AIO buffer descriptor to the end of the chain */
4051         meshlink_aio_buffer_t **p = &channel->aio_receive;
4052
4053         while(*p) {
4054                 p = &(*p)->next;
4055         }
4056
4057         *p = aio;
4058
4059         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
4060
4061         return true;
4062 }
4063
4064 bool meshlink_channel_aio_fd_receive(meshlink_handle_t *mesh, meshlink_channel_t *channel, int fd, size_t len, meshlink_aio_fd_cb_t cb, void *priv) {
4065         if(!mesh || !channel) {
4066                 meshlink_errno = MESHLINK_EINVAL;
4067                 return false;
4068         }
4069
4070         if(!len || fd == -1) {
4071                 meshlink_errno = MESHLINK_EINVAL;
4072                 return false;
4073         }
4074
4075         meshlink_aio_buffer_t *aio = xzalloc(sizeof(*aio));
4076         aio->fd = fd;
4077         aio->len = len;
4078         aio->cb.fd = cb;
4079         aio->priv = priv;
4080
4081         assert(pthread_mutex_lock(&mesh->mutex) == 0);
4082
4083         /* Append the AIO buffer descriptor to the end of the chain */
4084         meshlink_aio_buffer_t **p = &channel->aio_receive;
4085
4086         while(*p) {
4087                 p = &(*p)->next;
4088         }
4089
4090         *p = aio;
4091
4092         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
4093
4094         return true;
4095 }
4096
4097 uint32_t meshlink_channel_get_flags(meshlink_handle_t *mesh, meshlink_channel_t *channel) {
4098         if(!mesh || !channel) {
4099                 meshlink_errno = MESHLINK_EINVAL;
4100                 return -1;
4101         }
4102
4103         return channel->c->flags;
4104 }
4105
4106 size_t meshlink_channel_get_sendq(meshlink_handle_t *mesh, meshlink_channel_t *channel) {
4107         if(!mesh || !channel) {
4108                 meshlink_errno = MESHLINK_EINVAL;
4109                 return -1;
4110         }
4111
4112         return utcp_get_sendq(channel->c);
4113 }
4114
4115 size_t meshlink_channel_get_recvq(meshlink_handle_t *mesh, meshlink_channel_t *channel) {
4116         if(!mesh || !channel) {
4117                 meshlink_errno = MESHLINK_EINVAL;
4118                 return -1;
4119         }
4120
4121         return utcp_get_recvq(channel->c);
4122 }
4123
4124 size_t meshlink_channel_get_mss(meshlink_handle_t *mesh, meshlink_channel_t *channel) {
4125         if(!mesh || !channel) {
4126                 meshlink_errno = MESHLINK_EINVAL;
4127                 return -1;
4128         }
4129
4130         return utcp_get_mss(channel->node->utcp);
4131 }
4132
4133 void meshlink_set_node_channel_timeout(meshlink_handle_t *mesh, meshlink_node_t *node, int timeout) {
4134         if(!mesh || !node) {
4135                 meshlink_errno = MESHLINK_EINVAL;
4136                 return;
4137         }
4138
4139         node_t *n = (node_t *)node;
4140
4141         assert(pthread_mutex_lock(&mesh->mutex) == 0);
4142
4143         if(!n->utcp) {
4144                 n->utcp = utcp_init(channel_accept, channel_pre_accept, channel_send, n);
4145                 utcp_set_mtu(n->utcp, n->mtu - sizeof(meshlink_packethdr_t));
4146                 utcp_set_retransmit_cb(n->utcp, channel_retransmit);
4147         }
4148
4149         utcp_set_user_timeout(n->utcp, timeout);
4150
4151         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
4152 }
4153
4154 void update_node_status(meshlink_handle_t *mesh, node_t *n) {
4155         if(n->status.reachable && mesh->channel_accept_cb && !n->utcp) {
4156                 n->utcp = utcp_init(channel_accept, channel_pre_accept, channel_send, n);
4157                 utcp_set_mtu(n->utcp, n->mtu - sizeof(meshlink_packethdr_t));
4158                 utcp_set_retransmit_cb(n->utcp, channel_retransmit);
4159         }
4160
4161         if(mesh->node_status_cb) {
4162                 mesh->node_status_cb(mesh, (meshlink_node_t *)n, n->status.reachable && !n->status.blacklisted);
4163         }
4164
4165         if(mesh->node_pmtu_cb) {
4166                 mesh->node_pmtu_cb(mesh, (meshlink_node_t *)n, n->minmtu);
4167         }
4168 }
4169
4170 void update_node_pmtu(meshlink_handle_t *mesh, node_t *n) {
4171         utcp_set_mtu(n->utcp, (n->minmtu > MINMTU ? n->minmtu : MINMTU) - sizeof(meshlink_packethdr_t));
4172
4173         if(mesh->node_pmtu_cb && !n->status.blacklisted) {
4174                 mesh->node_pmtu_cb(mesh, (meshlink_node_t *)n, n->minmtu);
4175         }
4176 }
4177
4178 void handle_duplicate_node(meshlink_handle_t *mesh, node_t *n) {
4179         if(!mesh->node_duplicate_cb || n->status.duplicate) {
4180                 return;
4181         }
4182
4183         n->status.duplicate = true;
4184         mesh->node_duplicate_cb(mesh, (meshlink_node_t *)n);
4185 }
4186
4187 void meshlink_enable_discovery(meshlink_handle_t *mesh, bool enable) {
4188 #if HAVE_CATTA
4189
4190         if(!mesh) {
4191                 meshlink_errno = MESHLINK_EINVAL;
4192                 return;
4193         }
4194
4195         assert(pthread_mutex_lock(&mesh->mutex) == 0);
4196
4197         if(mesh->discovery == enable) {
4198                 goto end;
4199         }
4200
4201         if(mesh->threadstarted) {
4202                 if(enable) {
4203                         discovery_start(mesh);
4204                 } else {
4205                         discovery_stop(mesh);
4206                 }
4207         }
4208
4209         mesh->discovery = enable;
4210
4211 end:
4212         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
4213 #else
4214         (void)mesh;
4215         (void)enable;
4216         meshlink_errno = MESHLINK_ENOTSUP;
4217 #endif
4218 }
4219
4220 void meshlink_set_dev_class_timeouts(meshlink_handle_t *mesh, dev_class_t devclass, int pinginterval, int pingtimeout) {
4221         if(!mesh || devclass < 0 || devclass >= DEV_CLASS_COUNT) {
4222                 meshlink_errno = EINVAL;
4223                 return;
4224         }
4225
4226         if(pinginterval < 1 || pingtimeout < 1 || pingtimeout > pinginterval) {
4227                 meshlink_errno = EINVAL;
4228                 return;
4229         }
4230
4231         assert(pthread_mutex_lock(&mesh->mutex) == 0);
4232         mesh->dev_class_traits[devclass].pinginterval = pinginterval;
4233         mesh->dev_class_traits[devclass].pingtimeout = pingtimeout;
4234         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
4235 }
4236
4237 void meshlink_set_dev_class_fast_retry_period(meshlink_handle_t *mesh, dev_class_t devclass, int fast_retry_period) {
4238         if(!mesh || devclass < 0 || devclass >= DEV_CLASS_COUNT) {
4239                 meshlink_errno = EINVAL;
4240                 return;
4241         }
4242
4243         if(fast_retry_period < 0) {
4244                 meshlink_errno = EINVAL;
4245                 return;
4246         }
4247
4248         assert(pthread_mutex_lock(&mesh->mutex) == 0);
4249         mesh->dev_class_traits[devclass].fast_retry_period = fast_retry_period;
4250         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
4251 }
4252
4253 extern void meshlink_set_inviter_commits_first(struct meshlink_handle *mesh, bool inviter_commits_first) {
4254         if(!mesh) {
4255                 meshlink_errno = EINVAL;
4256                 return;
4257         }
4258
4259         assert(pthread_mutex_lock(&mesh->mutex) == 0);
4260         mesh->inviter_commits_first = inviter_commits_first;
4261         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
4262 }
4263
4264 void meshlink_set_external_address_discovery_url(struct meshlink_handle *mesh, const char *url) {
4265         if(!mesh) {
4266                 meshlink_errno = EINVAL;
4267                 return;
4268         }
4269
4270         if(url && (strncmp(url, "http://", 7) || strchr(url, ' '))) {
4271                 meshlink_errno = EINVAL;
4272                 return;
4273         }
4274
4275         assert(pthread_mutex_lock(&mesh->mutex) == 0);
4276         free(mesh->external_address_url);
4277         mesh->external_address_url = url ? xstrdup(url) : NULL;
4278         assert(pthread_mutex_unlock(&mesh->mutex) == 0);
4279 }
4280
4281 void meshlink_set_scheduling_granularity(struct meshlink_handle *mesh, long granularity) {
4282         if(!mesh || granularity < 0) {
4283                 meshlink_errno = EINVAL;
4284                 return;
4285         }
4286
4287         utcp_set_clock_granularity(granularity);
4288 }
4289
4290 void handle_network_change(meshlink_handle_t *mesh, bool online) {
4291         (void)online;
4292
4293         if(!mesh->connections || !mesh->loop.running) {
4294                 return;
4295         }
4296
4297         retry(mesh);
4298 }
4299
4300 void call_error_cb(meshlink_handle_t *mesh, meshlink_errno_t cb_errno) {
4301         // We should only call the callback function if we are in the background thread.
4302         if(!mesh->error_cb) {
4303                 return;
4304         }
4305
4306         if(!mesh->threadstarted) {
4307                 return;
4308         }
4309
4310         if(mesh->thread == pthread_self()) {
4311                 mesh->error_cb(mesh, cb_errno);
4312         }
4313 }
4314
4315 static void __attribute__((constructor)) meshlink_init(void) {
4316         crypto_init();
4317         utcp_set_clock_granularity(10000);
4318 }
4319
4320 static void __attribute__((destructor)) meshlink_exit(void) {
4321         crypto_exit();
4322 }