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