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