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