]> git.meshlink.io Git - meshlink/blob - src/meshlink.c
Modify meshlink configuration base file structre
[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 "crypto.h"
24 #include "ecdsagen.h"
25 #include "logger.h"
26 #include "meshlink_internal.h"
27 #include "netutl.h"
28 #include "node.h"
29 #include "submesh.h"
30 #include "packmsg.h"
31 #include "prf.h"
32 #include "protocol.h"
33 #include "route.h"
34 #include "sockaddr.h"
35 #include "utils.h"
36 #include "xalloc.h"
37 #include "ed25519/sha512.h"
38 #include "discovery.h"
39 #include "devtools.h"
40
41 #ifndef MSG_NOSIGNAL
42 #define MSG_NOSIGNAL 0
43 #endif
44 __thread meshlink_errno_t meshlink_errno;
45 meshlink_log_cb_t global_log_cb;
46 meshlink_log_level_t global_log_level;
47
48 typedef bool (*search_node_by_condition_t)(const node_t *, const void *);
49
50 static int rstrip(char *value) {
51         int len = strlen(value);
52
53         while(len && strchr("\t\r\n ", value[len - 1])) {
54                 value[--len] = 0;
55         }
56
57         return len;
58 }
59
60 static void get_canonical_address(node_t *n, char **hostname, char **port) {
61         if(!n->canonical_address) {
62                 return;
63         }
64
65         *hostname = xstrdup(n->canonical_address);
66         char *space = strchr(*hostname, ' ');
67
68         if(space) {
69                 *space++ = 0;
70                 *port = xstrdup(space);
71         }
72 }
73
74 static bool is_valid_hostname(const char *hostname) {
75         if(!*hostname) {
76                 return false;
77         }
78
79         for(const char *p = hostname; *p; p++) {
80                 if(!(isalnum(*p) || *p == '-' || *p == '.' || *p == ':')) {
81                         return false;
82                 }
83         }
84
85         return true;
86 }
87
88 static bool is_valid_port(const char *port) {
89         if(!*port) {
90                 return false;
91         }
92
93         if(isdigit(*port)) {
94                 char *end;
95                 unsigned long int result = strtoul(port, &end, 10);
96                 return result && result < 65536 && !*end;
97         }
98
99         for(const char *p = port; *p; p++) {
100                 if(!(isalnum(*p) || *p == '-')) {
101                         return false;
102                 }
103         }
104
105         return true;
106 }
107
108 static void set_timeout(int sock, int timeout) {
109 #ifdef _WIN32
110         DWORD tv = timeout;
111 #else
112         struct timeval tv;
113         tv.tv_sec = timeout / 1000;
114         tv.tv_usec = (timeout - tv.tv_sec * 1000) * 1000;
115 #endif
116         setsockopt(sock, SOL_SOCKET, SO_RCVTIMEO, &tv, sizeof(tv));
117         setsockopt(sock, SOL_SOCKET, SO_SNDTIMEO, &tv, sizeof(tv));
118 }
119
120 struct socket_in_netns_params {
121         int domain;
122         int type;
123         int protocol;
124         int netns;
125         int fd;
126 };
127
128 #ifdef HAVE_SETNS
129 static void *socket_in_netns_thread(void *arg) {
130         struct socket_in_netns_params *params = arg;
131
132         if(setns(params->netns, CLONE_NEWNET) == -1) {
133                 meshlink_errno = MESHLINK_EINVAL;
134                 return NULL;
135         }
136
137         params->fd = socket(params->domain, params->type, params->protocol);
138
139         return NULL;
140 }
141 #endif // HAVE_SETNS
142
143 static int socket_in_netns(int domain, int type, int protocol, int netns) {
144         if(netns == -1) {
145                 return socket(domain, type, protocol);
146         }
147
148 #ifdef HAVE_SETNS
149         struct socket_in_netns_params params = {domain, type, protocol, netns, -1};
150
151         pthread_t thr;
152
153         if(pthread_create(&thr, NULL, socket_in_netns_thread, &params) == 0) {
154                 pthread_join(thr, NULL);
155         }
156
157         return params.fd;
158 #else
159         return -1;
160 #endif // HAVE_SETNS
161
162 }
163
164 // Find out what local address a socket would use if we connect to the given address.
165 // We do this using connect() on a UDP socket, so the kernel has to resolve the address
166 // of both endpoints, but this will actually not send any UDP packet.
167 static bool getlocaladdr(char *destaddr, struct sockaddr *sn, socklen_t *sl, int netns) {
168         struct addrinfo *rai = NULL;
169         const struct addrinfo hint = {
170                 .ai_family = AF_UNSPEC,
171                 .ai_socktype = SOCK_DGRAM,
172                 .ai_protocol = IPPROTO_UDP,
173         };
174
175         if(getaddrinfo(destaddr, "80", &hint, &rai) || !rai) {
176                 return false;
177         }
178
179         int sock = socket_in_netns(rai->ai_family, rai->ai_socktype, rai->ai_protocol, netns);
180
181         if(sock == -1) {
182                 freeaddrinfo(rai);
183                 return false;
184         }
185
186         if(connect(sock, rai->ai_addr, rai->ai_addrlen) && !sockwouldblock(errno)) {
187                 closesocket(sock);
188                 freeaddrinfo(rai);
189                 return false;
190         }
191
192         freeaddrinfo(rai);
193
194         if(getsockname(sock, sn, sl)) {
195                 closesocket(sock);
196                 return false;
197         }
198
199         closesocket(sock);
200         return true;
201 }
202
203 static bool getlocaladdrname(char *destaddr, char *host, socklen_t hostlen, int netns) {
204         struct sockaddr_storage sn;
205         socklen_t sl = sizeof(sn);
206
207         if(!getlocaladdr(destaddr, (struct sockaddr *)&sn, &sl, netns)) {
208                 return false;
209         }
210
211         if(getnameinfo((struct sockaddr *)&sn, sl, host, hostlen, NULL, 0, NI_NUMERICHOST | NI_NUMERICSERV)) {
212                 return false;
213         }
214
215         return true;
216 }
217
218 char *meshlink_get_external_address(meshlink_handle_t *mesh) {
219         return meshlink_get_external_address_for_family(mesh, AF_UNSPEC);
220 }
221
222 char *meshlink_get_external_address_for_family(meshlink_handle_t *mesh, int family) {
223         char *hostname = NULL;
224
225         logger(mesh, MESHLINK_DEBUG, "Trying to discover externally visible hostname...\n");
226         struct addrinfo *ai = str2addrinfo("meshlink.io", "80", SOCK_STREAM);
227         static const char request[] = "GET http://www.meshlink.io/host.cgi HTTP/1.0\r\n\r\n";
228         char line[256];
229
230         for(struct addrinfo *aip = ai; aip; aip = aip->ai_next) {
231                 if(family != AF_UNSPEC && aip->ai_family != family) {
232                         continue;
233                 }
234
235                 int s = socket_in_netns(aip->ai_family, aip->ai_socktype, aip->ai_protocol, mesh->netns);
236
237                 if(s >= 0) {
238                         set_timeout(s, 5000);
239
240                         if(connect(s, aip->ai_addr, aip->ai_addrlen)) {
241                                 closesocket(s);
242                                 s = -1;
243                         }
244                 }
245
246                 if(s >= 0) {
247                         send(s, request, sizeof(request) - 1, 0);
248                         int len = recv(s, line, sizeof(line) - 1, MSG_WAITALL);
249
250                         if(len > 0) {
251                                 line[len] = 0;
252
253                                 if(line[len - 1] == '\n') {
254                                         line[--len] = 0;
255                                 }
256
257                                 char *p = strrchr(line, '\n');
258
259                                 if(p && p[1]) {
260                                         hostname = xstrdup(p + 1);
261                                 }
262                         }
263
264                         closesocket(s);
265
266                         if(hostname) {
267                                 break;
268                         }
269                 }
270         }
271
272         if(ai) {
273                 freeaddrinfo(ai);
274         }
275
276         // Check that the hostname is reasonable
277         if(hostname && !is_valid_hostname(hostname)) {
278                 free(hostname);
279                 hostname = NULL;
280         }
281
282         if(!hostname) {
283                 meshlink_errno = MESHLINK_ERESOLV;
284         }
285
286         return hostname;
287 }
288
289 char *meshlink_get_local_address_for_family(meshlink_handle_t *mesh, int family) {
290         (void)mesh;
291
292         // Determine address of the local interface used for outgoing connections.
293         char localaddr[NI_MAXHOST];
294         bool success = false;
295
296         if(family == AF_INET) {
297                 success = getlocaladdrname("93.184.216.34", localaddr, sizeof(localaddr), mesh->netns);
298         } else if(family == AF_INET6) {
299                 success = getlocaladdrname("2606:2800:220:1:248:1893:25c8:1946", localaddr, sizeof(localaddr), mesh->netns);
300         }
301
302         if(!success) {
303                 meshlink_errno = MESHLINK_ENETWORK;
304                 return NULL;
305         }
306
307         return xstrdup(localaddr);
308 }
309
310 void remove_duplicate_hostnames(char *host[], char *port[], int n) {
311         for(int i = 0; i < n; i++) {
312                 if(!host[i]) {
313                         continue;
314                 }
315
316                 // Ignore duplicate hostnames
317                 bool found = false;
318
319                 for(int j = 0; j < i; j++) {
320                         if(!host[j]) {
321                                 continue;
322                         }
323
324                         if(strcmp(host[i], host[j])) {
325                                 continue;
326                         }
327
328                         if(strcmp(port[i], port[j])) {
329                                 continue;
330                         }
331
332                         found = true;
333                         break;
334                 }
335
336                 if(found) {
337                         free(host[i]);
338                         free(port[i]);
339                         host[i] = NULL;
340                         port[i] = NULL;
341                         continue;
342                 }
343         }
344 }
345
346 // This gets the hostname part for use in invitation URLs
347 static char *get_my_hostname(meshlink_handle_t *mesh, uint32_t flags) {
348         char *hostname[4] = {NULL};
349         char *port[4] = {NULL};
350         char *hostport = NULL;
351
352         if(!(flags & (MESHLINK_INVITE_LOCAL | MESHLINK_INVITE_PUBLIC))) {
353                 flags |= MESHLINK_INVITE_LOCAL | MESHLINK_INVITE_PUBLIC;
354         }
355
356         if(!(flags & (MESHLINK_INVITE_IPV4 | MESHLINK_INVITE_IPV6))) {
357                 flags |= MESHLINK_INVITE_IPV4 | MESHLINK_INVITE_IPV6;
358         }
359
360         // Add local addresses if requested
361         if(flags & MESHLINK_INVITE_LOCAL) {
362                 if(flags & MESHLINK_INVITE_IPV4) {
363                         hostname[0] = meshlink_get_local_address_for_family(mesh, AF_INET);
364                 }
365
366                 if(flags & MESHLINK_INVITE_IPV6) {
367                         hostname[1] = meshlink_get_local_address_for_family(mesh, AF_INET6);
368                 }
369         }
370
371         // Add public/canonical addresses if requested
372         if(flags & MESHLINK_INVITE_PUBLIC) {
373                 // Try the CanonicalAddress first
374                 get_canonical_address(mesh->self, &hostname[2], &port[2]);
375
376                 if(!hostname[2]) {
377                         if(flags & MESHLINK_INVITE_IPV4) {
378                                 hostname[2] = meshlink_get_external_address_for_family(mesh, AF_INET);
379                         }
380
381                         if(flags & MESHLINK_INVITE_IPV6) {
382                                 hostname[3] = meshlink_get_external_address_for_family(mesh, AF_INET6);
383                         }
384                 }
385         }
386
387         for(int i = 0; i < 4; i++) {
388                 // Ensure we always have a port number
389                 if(hostname[i] && !port[i]) {
390                         port[i] = xstrdup(mesh->myport);
391                 }
392         }
393
394         remove_duplicate_hostnames(hostname, port, 4);
395
396         if(!(flags & MESHLINK_INVITE_NUMERIC)) {
397                 for(int i = 0; i < 4; i++) {
398                         if(!hostname[i]) {
399                                 continue;
400                         }
401
402                         // Convert what we have to a sockaddr
403                         struct addrinfo *ai_in, *ai_out;
404                         struct addrinfo hint = {
405                                 .ai_family = AF_UNSPEC,
406                                 .ai_flags = AI_NUMERICSERV,
407                                 .ai_socktype = SOCK_STREAM,
408                         };
409                         int err = getaddrinfo(hostname[i], port[i], &hint, &ai_in);
410
411                         if(err || !ai_in) {
412                                 continue;
413                         }
414
415                         // Convert it to a hostname
416                         char resolved_host[NI_MAXHOST];
417                         char resolved_port[NI_MAXSERV];
418                         err = getnameinfo(ai_in->ai_addr, ai_in->ai_addrlen, resolved_host, sizeof resolved_host, resolved_port, sizeof resolved_port, NI_NUMERICSERV);
419
420                         if(err) {
421                                 freeaddrinfo(ai_in);
422                                 continue;
423                         }
424
425                         // Convert the hostname back to a sockaddr
426                         hint.ai_family = ai_in->ai_family;
427                         err = getaddrinfo(resolved_host, resolved_port, &hint, &ai_out);
428
429                         if(err || !ai_out) {
430                                 freeaddrinfo(ai_in);
431                                 continue;
432                         }
433
434                         // Check if it's still the same sockaddr
435                         if(ai_in->ai_addrlen != ai_out->ai_addrlen || memcmp(ai_in->ai_addr, ai_out->ai_addr, ai_in->ai_addrlen)) {
436                                 freeaddrinfo(ai_in);
437                                 freeaddrinfo(ai_out);
438                                 continue;
439                         }
440
441                         // Yes: replace the hostname with the resolved one
442                         free(hostname[i]);
443                         hostname[i] = xstrdup(resolved_host);
444
445                         freeaddrinfo(ai_in);
446                         freeaddrinfo(ai_out);
447                 }
448         }
449
450         // Remove duplicates again, since IPv4 and IPv6 addresses might map to the same hostname
451         remove_duplicate_hostnames(hostname, port, 4);
452
453         // Concatenate all unique address to the hostport string
454         for(int i = 0; i < 4; i++) {
455                 if(!hostname[i]) {
456                         continue;
457                 }
458
459                 // Ensure we have the same addresses in our own host config file.
460                 char *tmphostport;
461                 xasprintf(&tmphostport, "%s %s", hostname[i], port[i]);
462                 /// TODO: FIX
463                 //config_add_string(&mesh->config, "Address", tmphostport);
464                 free(tmphostport);
465
466                 // Append the address to the hostport string
467                 char *newhostport;
468                 xasprintf(&newhostport, (strchr(hostname[i], ':') ? "%s%s[%s]:%s" : "%s%s%s:%s"), hostport ? hostport : "", hostport ? "," : "", hostname[i], port[i]);
469                 free(hostport);
470                 hostport = newhostport;
471
472                 free(hostname[i]);
473                 free(port[i]);
474         }
475
476         return hostport;
477 }
478
479 static bool try_bind(int port) {
480         struct addrinfo *ai = NULL;
481         struct addrinfo hint = {
482                 .ai_flags = AI_PASSIVE,
483                 .ai_family = AF_UNSPEC,
484                 .ai_socktype = SOCK_STREAM,
485                 .ai_protocol = IPPROTO_TCP,
486         };
487
488         char portstr[16];
489         snprintf(portstr, sizeof(portstr), "%d", port);
490
491         if(getaddrinfo(NULL, portstr, &hint, &ai) || !ai) {
492                 return false;
493         }
494
495         //while(ai) {
496         for(struct addrinfo *aip = ai; aip; aip = aip->ai_next) {
497                 int fd = socket(aip->ai_family, SOCK_STREAM, IPPROTO_TCP);
498
499                 if(!fd) {
500                         freeaddrinfo(ai);
501                         return false;
502                 }
503
504                 int result = bind(fd, aip->ai_addr, aip->ai_addrlen);
505                 closesocket(fd);
506
507                 if(result) {
508                         freeaddrinfo(ai);
509                         return false;
510                 }
511         }
512
513         freeaddrinfo(ai);
514         return true;
515 }
516
517 int check_port(meshlink_handle_t *mesh) {
518         for(int i = 0; i < 1000; i++) {
519                 int port = 0x1000 + (rand() & 0x7fff);
520
521                 if(try_bind(port)) {
522                         free(mesh->myport);
523                         xasprintf(&mesh->myport, "%d", port);
524                         return port;
525                 }
526         }
527
528         meshlink_errno = MESHLINK_ENETWORK;
529         logger(mesh, MESHLINK_DEBUG, "Could not find any available network port.\n");
530         return 0;
531 }
532
533 static bool write_main_config_files(meshlink_handle_t *mesh) {
534         if(!mesh->confbase) {
535                 return true;
536         }
537
538         uint8_t buf[4096];
539
540         /* Write the main config file */
541         packmsg_output_t out = {buf, sizeof buf};
542
543         packmsg_add_uint32(&out, MESHLINK_CONFIG_VERSION);
544         packmsg_add_str(&out, mesh->name);
545         packmsg_add_bin(&out, ecdsa_get_private_key(mesh->private_key), 96);
546         packmsg_add_bin(&out, ecdsa_get_private_key(mesh->invitation_key), 96);
547         packmsg_add_uint16(&out, atoi(mesh->myport));
548
549         if(!packmsg_output_ok(&out)) {
550                 return false;
551         }
552
553         config_t config = {buf, packmsg_output_size(&out, buf)};
554
555         if(!main_config_write(mesh, "current", &config, mesh->config_key)) {
556                 return false;
557         }
558
559         /* Write our own host config file */
560         if(!node_write_config(mesh, mesh->self)) {
561                 return false;
562         }
563
564         return true;
565 }
566
567 static bool finalize_join(meshlink_handle_t *mesh, const void *buf, uint16_t len) {
568         packmsg_input_t in = {buf, len};
569         uint32_t version = packmsg_get_uint32(&in);
570
571         if(version != MESHLINK_INVITATION_VERSION) {
572                 logger(mesh, MESHLINK_ERROR, "Invalid invitation version!\n");
573                 return false;
574         }
575
576         char *name = packmsg_get_str_dup(&in);
577         packmsg_skip_element(&in); /* submesh */
578         int32_t devclass = packmsg_get_int32(&in);
579         uint32_t count = packmsg_get_array(&in);
580
581         if(!name) {
582                 logger(mesh, MESHLINK_DEBUG, "No Name found in invitation!\n");
583                 return false;
584         }
585
586         if(!check_id(name)) {
587                 logger(mesh, MESHLINK_DEBUG, "Invalid Name found in invitation: %s!\n", name);
588                 free(name);
589                 return false;
590         }
591
592         if(!count) {
593                 logger(mesh, MESHLINK_ERROR, "Incomplete invitation file!\n");
594                 free(name);
595                 return false;
596         }
597
598         free(mesh->name);
599         free(mesh->self->name);
600         mesh->name = name;
601         mesh->self->name = xstrdup(name);
602         mesh->self->devclass = devclass;
603
604         // Initialize configuration directory
605         if(!config_init(mesh, "current")) {
606                 return false;
607         }
608
609         if(!write_main_config_files(mesh)) {
610                 return false;
611         }
612
613         // Write host config files
614         while(count--) {
615                 const void *data;
616                 uint32_t len = packmsg_get_bin_raw(&in, &data);
617
618                 if(!len) {
619                         logger(mesh, MESHLINK_ERROR, "Incomplete invitation file!\n");
620                         return false;
621                 }
622
623                 packmsg_input_t in2 = {data, len};
624                 uint32_t version = packmsg_get_uint32(&in2);
625                 char *name = packmsg_get_str_dup(&in2);
626
627                 if(!packmsg_input_ok(&in2) || version != MESHLINK_CONFIG_VERSION || !check_id(name)) {
628                         free(name);
629                         packmsg_input_invalidate(&in);
630                         break;
631                 }
632
633                 if(!check_id(name)) {
634                         free(name);
635                         break;
636                 }
637
638                 if(!strcmp(name, mesh->name)) {
639                         logger(mesh, MESHLINK_DEBUG, "Secondary chunk would overwrite our own host config file.\n");
640                         free(name);
641                         meshlink_errno = MESHLINK_EPEER;
642                         return false;
643                 }
644
645                 node_t *n = new_node();
646                 n->name = name;
647
648                 config_t config = {data, len};
649
650                 if(!node_read_from_config(mesh, n, &config)) {
651                         free_node(n);
652                         logger(mesh, MESHLINK_ERROR, "Invalid host config file in invitation file!\n");
653                         meshlink_errno = MESHLINK_EPEER;
654                         return false;
655                 }
656
657                 node_add(mesh, n);
658
659                 if(!config_write(mesh, "current", n->name, &config, mesh->config_key)) {
660                         return false;
661                 }
662         }
663
664         sptps_send_record(&(mesh->sptps), 1, ecdsa_get_public_key(mesh->private_key), 32);
665
666         logger(mesh, MESHLINK_DEBUG, "Configuration stored in: %s\n", mesh->confbase);
667
668         return true;
669 }
670
671 static bool invitation_send(void *handle, uint8_t type, const void *data, size_t len) {
672         (void)type;
673         meshlink_handle_t *mesh = handle;
674         const char *ptr = data;
675
676         while(len) {
677                 int result = send(mesh->sock, ptr, len, 0);
678
679                 if(result == -1 && errno == EINTR) {
680                         continue;
681                 } else if(result <= 0) {
682                         return false;
683                 }
684
685                 ptr += result;
686                 len -= result;
687         }
688
689         return true;
690 }
691
692 static bool invitation_receive(void *handle, uint8_t type, const void *msg, uint16_t len) {
693         meshlink_handle_t *mesh = handle;
694
695         switch(type) {
696         case SPTPS_HANDSHAKE:
697                 return sptps_send_record(&(mesh->sptps), 0, mesh->cookie, sizeof(mesh)->cookie);
698
699         case 0:
700                 return finalize_join(mesh, msg, len);
701
702         case 1:
703                 logger(mesh, MESHLINK_DEBUG, "Invitation succesfully accepted.\n");
704                 shutdown(mesh->sock, SHUT_RDWR);
705                 mesh->success = true;
706                 break;
707
708         default:
709                 return false;
710         }
711
712         return true;
713 }
714
715 static bool recvline(meshlink_handle_t *mesh, size_t len) {
716         char *newline = NULL;
717
718         if(!mesh->sock) {
719                 abort();
720         }
721
722         while(!(newline = memchr(mesh->buffer, '\n', mesh->blen))) {
723                 int result = recv(mesh->sock, mesh->buffer + mesh->blen, sizeof(mesh)->buffer - mesh->blen, 0);
724
725                 if(result == -1 && errno == EINTR) {
726                         continue;
727                 } else if(result <= 0) {
728                         return false;
729                 }
730
731                 mesh->blen += result;
732         }
733
734         if((size_t)(newline - mesh->buffer) >= len) {
735                 return false;
736         }
737
738         len = newline - mesh->buffer;
739
740         memcpy(mesh->line, mesh->buffer, len);
741         mesh->line[len] = 0;
742         memmove(mesh->buffer, newline + 1, mesh->blen - len - 1);
743         mesh->blen -= len + 1;
744
745         return true;
746 }
747 static bool sendline(int fd, char *format, ...) {
748         static char buffer[4096];
749         char *p = buffer;
750         int blen = 0;
751         va_list ap;
752
753         va_start(ap, format);
754         blen = vsnprintf(buffer, sizeof(buffer), format, ap);
755         va_end(ap);
756
757         if(blen < 1 || (size_t)blen >= sizeof(buffer)) {
758                 return false;
759         }
760
761         buffer[blen] = '\n';
762         blen++;
763
764         while(blen) {
765                 int result = send(fd, p, blen, MSG_NOSIGNAL);
766
767                 if(result == -1 && errno == EINTR) {
768                         continue;
769                 } else if(result <= 0) {
770                         return false;
771                 }
772
773                 p += result;
774                 blen -= result;
775         }
776
777         return true;
778 }
779
780 static const char *errstr[] = {
781         [MESHLINK_OK] = "No error",
782         [MESHLINK_EINVAL] = "Invalid argument",
783         [MESHLINK_ENOMEM] = "Out of memory",
784         [MESHLINK_ENOENT] = "No such node",
785         [MESHLINK_EEXIST] = "Node already exists",
786         [MESHLINK_EINTERNAL] = "Internal error",
787         [MESHLINK_ERESOLV] = "Could not resolve hostname",
788         [MESHLINK_ESTORAGE] = "Storage error",
789         [MESHLINK_ENETWORK] = "Network error",
790         [MESHLINK_EPEER] = "Error communicating with peer",
791         [MESHLINK_ENOTSUP] = "Operation not supported",
792         [MESHLINK_EBUSY] = "MeshLink instance already in use",
793 };
794
795 const char *meshlink_strerror(meshlink_errno_t err) {
796         if((int)err < 0 || err >= sizeof(errstr) / sizeof(*errstr)) {
797                 return "Invalid error code";
798         }
799
800         return errstr[err];
801 }
802
803 static bool ecdsa_keygen(meshlink_handle_t *mesh) {
804         logger(mesh, MESHLINK_DEBUG, "Generating ECDSA keypairs:\n");
805
806         mesh->private_key = ecdsa_generate();
807         mesh->invitation_key = ecdsa_generate();
808
809         if(!mesh->private_key || !mesh->invitation_key) {
810                 logger(mesh, MESHLINK_DEBUG, "Error during key generation!\n");
811                 meshlink_errno = MESHLINK_EINTERNAL;
812                 return false;
813         }
814
815         logger(mesh, MESHLINK_DEBUG, "Done.\n");
816
817         return true;
818 }
819
820 static struct timeval idle(event_loop_t *loop, void *data) {
821         (void)loop;
822         meshlink_handle_t *mesh = data;
823         struct timeval t, tmin = {3600, 0};
824
825         for splay_each(node_t, n, mesh->nodes) {
826                 if(!n->utcp) {
827                         continue;
828                 }
829
830                 t = utcp_timeout(n->utcp);
831
832                 if(timercmp(&t, &tmin, <)) {
833                         tmin = t;
834                 }
835         }
836
837         return tmin;
838 }
839
840 // Get our local address(es) by simulating connecting to an Internet host.
841 static void add_local_addresses(meshlink_handle_t *mesh) {
842         struct sockaddr_storage sn;
843         socklen_t sl = sizeof(sn);
844
845         // IPv4 example.org
846
847         if(getlocaladdr("93.184.216.34", (struct sockaddr *)&sn, &sl, mesh->netns)) {
848                 ((struct sockaddr_in *)&sn)->sin_port = ntohs(atoi(mesh->myport));
849                 meshlink_hint_address(mesh, (meshlink_node_t *)mesh->self, (struct sockaddr *)&sn);
850         }
851
852         // IPv6 example.org
853
854         sl = sizeof(sn);
855
856         if(getlocaladdr("2606:2800:220:1:248:1893:25c8:1946", (struct sockaddr *)&sn, &sl, mesh->netns)) {
857                 ((struct sockaddr_in6 *)&sn)->sin6_port = ntohs(atoi(mesh->myport));
858                 meshlink_hint_address(mesh, (meshlink_node_t *)mesh->self, (struct sockaddr *)&sn);
859         }
860 }
861
862 static bool meshlink_setup(meshlink_handle_t *mesh) {
863         if(!config_init(mesh, "current")) {
864                 logger(mesh, MESHLINK_ERROR, "Could not set up configuration in %s/current: %s\n", mesh->confbase, strerror(errno));
865                 meshlink_errno = MESHLINK_ESTORAGE;
866                 return false;
867         }
868
869         if(!ecdsa_keygen(mesh)) {
870                 meshlink_errno = MESHLINK_EINTERNAL;
871                 return false;
872         }
873
874         if(check_port(mesh) == 0) {
875                 meshlink_errno = MESHLINK_ENETWORK;
876                 return false;
877         }
878
879         /* Create a node for ourself */
880
881         mesh->self = new_node();
882         mesh->self->name = xstrdup(mesh->name);
883         mesh->self->devclass = mesh->devclass;
884         mesh->self->ecdsa = ecdsa_set_public_key(ecdsa_get_public_key(mesh->private_key));
885
886         if(!write_main_config_files(mesh)) {
887                 logger(mesh, MESHLINK_ERROR, "Could not write main config files into %s/current: %s\n", mesh->confbase, strerror(errno));
888                 meshlink_errno = MESHLINK_ESTORAGE;
889                 return false;
890         }
891
892         if(!main_config_lock(mesh)) {
893                 logger(NULL, MESHLINK_ERROR, "Cannot lock main config file\n");
894                 meshlink_errno = MESHLINK_ESTORAGE;
895                 return false;
896         }
897
898         return true;
899 }
900
901 static bool meshlink_read_config(meshlink_handle_t *mesh) {
902         // Open the configuration file and lock it
903         if(!main_config_lock(mesh)) {
904                 logger(NULL, MESHLINK_ERROR, "Cannot lock main config file\n");
905                 meshlink_errno = MESHLINK_ESTORAGE;
906                 return false;
907         }
908
909         config_t config;
910
911         if(!main_config_read(mesh, "current", &config, mesh->config_key)) {
912                 logger(NULL, MESHLINK_ERROR, "Could not read main configuration file!");
913                 return false;
914         }
915
916         packmsg_input_t in = {config.buf, config.len};
917         const void *private_key;
918         const void *invitation_key;
919
920         uint32_t version = packmsg_get_uint32(&in);
921         char *name = packmsg_get_str_dup(&in);
922         uint32_t private_key_len = packmsg_get_bin_raw(&in, &private_key);
923         uint32_t invitation_key_len = packmsg_get_bin_raw(&in, &invitation_key);
924         uint16_t myport = packmsg_get_uint16(&in);
925
926         if(!packmsg_done(&in) || version != MESHLINK_CONFIG_VERSION || private_key_len != 96 || invitation_key_len != 96) {
927                 logger(NULL, MESHLINK_ERROR, "Error parsing main configuration file!");
928                 free(name);
929                 config_free(&config);
930                 return false;
931         }
932
933 #if 0
934
935         // TODO: check this?
936         if(mesh->name && strcmp(mesh->name, name)) {
937                 logger(NULL, MESHLINK_ERROR, "Configuration is for a different name (%s)!", name);
938                 meshlink_errno = MESHLINK_ESTORAGE;
939                 free(name);
940                 config_free(&config);
941                 return false;
942         }
943
944 #endif
945
946         free(mesh->name);
947         mesh->name = name;
948         xasprintf(&mesh->myport, "%u", myport);
949         mesh->private_key = ecdsa_set_private_key(private_key);
950         mesh->invitation_key = ecdsa_set_private_key(invitation_key);
951         config_free(&config);
952
953         /* Create a node for ourself and read our host configuration file */
954
955         mesh->self = new_node();
956         mesh->self->name = xstrdup(name);
957         mesh->self->devclass = mesh->devclass;
958
959         if(!node_read_public_key(mesh, mesh->self)) {
960                 logger(NULL, MESHLINK_ERROR, "Could not read our host configuration file!");
961                 meshlink_errno = MESHLINK_ESTORAGE;
962                 free_node(mesh->self);
963                 mesh->self = NULL;
964                 return false;
965         }
966
967         return true;
968 }
969
970 #ifdef HAVE_SETNS
971 static void *setup_network_in_netns_thread(void *arg) {
972         meshlink_handle_t *mesh = arg;
973
974         if(setns(mesh->netns, CLONE_NEWNET) != 0) {
975                 return NULL;
976         }
977
978         bool success = setup_network(mesh);
979         add_local_addresses(mesh);
980         return success ? arg : NULL;
981 }
982 #endif // HAVE_SETNS
983
984 meshlink_open_params_t *meshlink_open_params_init(const char *confbase, const char *name, const char *appname, dev_class_t devclass) {
985         if(!confbase || !*confbase) {
986                 logger(NULL, MESHLINK_ERROR, "No confbase given!\n");
987                 meshlink_errno = MESHLINK_EINVAL;
988                 return NULL;
989         }
990
991         if(!appname || !*appname) {
992                 logger(NULL, MESHLINK_ERROR, "No appname given!\n");
993                 meshlink_errno = MESHLINK_EINVAL;
994                 return NULL;
995         }
996
997         if(strchr(appname, ' ')) {
998                 logger(NULL, MESHLINK_ERROR, "Invalid appname given!\n");
999                 meshlink_errno = MESHLINK_EINVAL;
1000                 return NULL;
1001         }
1002
1003         if(!name || !*name) {
1004                 logger(NULL, MESHLINK_ERROR, "No name given!\n");
1005                 //return NULL;
1006         } else { //check name only if there is a name != NULL
1007                 if(!check_id(name)) {
1008                         logger(NULL, MESHLINK_ERROR, "Invalid name given!\n");
1009                         meshlink_errno = MESHLINK_EINVAL;
1010                         return NULL;
1011                 }
1012         }
1013
1014         if((int)devclass < 0 || devclass > _DEV_CLASS_MAX) {
1015                 logger(NULL, MESHLINK_ERROR, "Invalid devclass given!\n");
1016                 meshlink_errno = MESHLINK_EINVAL;
1017                 return NULL;
1018         }
1019
1020         meshlink_open_params_t *params = xzalloc(sizeof * params);
1021
1022         params->confbase = xstrdup(confbase);
1023         params->name = xstrdup(name);
1024         params->appname = xstrdup(appname);
1025         params->devclass = devclass;
1026         params->netns = -1;
1027
1028         return params;
1029 }
1030
1031 bool meshlink_open_params_set_netns(meshlink_open_params_t *params, int netns) {
1032         if(!params) {
1033                 meshlink_errno = MESHLINK_EINVAL;
1034                 return false;
1035         }
1036
1037         params->netns = netns;
1038
1039         return true;
1040 }
1041
1042 bool meshlink_open_params_set_storage_key(meshlink_open_params_t *params, const void *key, size_t keylen) {
1043         if(!params) {
1044                 meshlink_errno = MESHLINK_EINVAL;
1045                 return false;
1046         }
1047
1048         if((!key && keylen) || (key && !keylen)) {
1049                 logger(NULL, MESHLINK_ERROR, "Invalid key length!\n");
1050                 meshlink_errno = MESHLINK_EINVAL;
1051                 return false;
1052         }
1053
1054         params->key = key;
1055         params->keylen = keylen;
1056
1057         return true;
1058 }
1059
1060 bool meshlink_encrypted_key_rotate(meshlink_handle_t *mesh, const void *new_key, size_t new_keylen) {
1061
1062         // While copying old config files to new config files
1063         devtool_keyrotate_probe(1);
1064         // After completed creating new config files in confbase/new/
1065         devtool_keyrotate_probe(2);
1066         // Rename confbase/current to confbase/old/
1067         devtool_keyrotate_probe(3);
1068         // Rename confbase/new/ to confbase/current
1069         devtool_keyrotate_probe(4);
1070         // Before deleting old sub-directory
1071         devtool_keyrotate_probe(5);
1072
1073         return false;
1074 }
1075
1076 void meshlink_open_params_free(meshlink_open_params_t *params) {
1077         if(!params) {
1078                 meshlink_errno = MESHLINK_EINVAL;
1079                 return;
1080         }
1081
1082         free(params->confbase);
1083         free(params->name);
1084         free(params->appname);
1085
1086         free(params);
1087 }
1088
1089 meshlink_handle_t *meshlink_open(const char *confbase, const char *name, const char *appname, dev_class_t devclass) {
1090         if(!confbase || !*confbase) {
1091                 logger(NULL, MESHLINK_ERROR, "No confbase given!\n");
1092                 meshlink_errno = MESHLINK_EINVAL;
1093                 return NULL;
1094         }
1095
1096         /* Create a temporary struct on the stack, to avoid allocating and freeing one. */
1097         meshlink_open_params_t params;
1098         memset(&params, 0, sizeof(params));
1099
1100         params.confbase = (char *)confbase;
1101         params.name = (char *)name;
1102         params.appname = (char *)appname;
1103         params.devclass = devclass;
1104         params.netns = -1;
1105
1106         return meshlink_open_ex(&params);
1107 }
1108
1109 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) {
1110         if(!confbase || !*confbase) {
1111                 logger(NULL, MESHLINK_ERROR, "No confbase given!\n");
1112                 meshlink_errno = MESHLINK_EINVAL;
1113                 return NULL;
1114         }
1115
1116         /* Create a temporary struct on the stack, to avoid allocating and freeing one. */
1117         meshlink_open_params_t params = {NULL};
1118
1119         params.confbase = (char *)confbase;
1120         params.name = (char *)name;
1121         params.appname = (char *)appname;
1122         params.devclass = devclass;
1123         params.netns = -1;
1124
1125         if(!meshlink_open_params_set_storage_key(&params, key, keylen)) {
1126                 return false;
1127         }
1128
1129         return meshlink_open_ex(&params);
1130 }
1131
1132 meshlink_handle_t *meshlink_open_ephemeral(const char *name, const char *appname, dev_class_t devclass) {
1133         /* Create a temporary struct on the stack, to avoid allocating and freeing one. */
1134         meshlink_open_params_t params = {NULL};
1135
1136         params.name = (char *)name;
1137         params.appname = (char *)appname;
1138         params.devclass = devclass;
1139         params.netns = -1;
1140
1141         return meshlink_open_ex(&params);
1142 }
1143
1144 meshlink_handle_t *meshlink_open_ex(const meshlink_open_params_t *params) {
1145         // Validate arguments provided by the application
1146         bool usingname = false;
1147
1148         logger(NULL, MESHLINK_DEBUG, "meshlink_open called\n");
1149
1150         if(!params->appname || !*params->appname) {
1151                 logger(NULL, MESHLINK_ERROR, "No appname given!\n");
1152                 meshlink_errno = MESHLINK_EINVAL;
1153                 return NULL;
1154         }
1155
1156         if(strchr(params->appname, ' ')) {
1157                 logger(NULL, MESHLINK_ERROR, "Invalid appname given!\n");
1158                 meshlink_errno = MESHLINK_EINVAL;
1159                 return NULL;
1160         }
1161
1162         if(!params->name || !*params->name) {
1163                 logger(NULL, MESHLINK_ERROR, "No name given!\n");
1164                 //return NULL;
1165         } else { //check name only if there is a name != NULL
1166
1167                 if(!check_id(params->name)) {
1168                         logger(NULL, MESHLINK_ERROR, "Invalid name given!\n");
1169                         meshlink_errno = MESHLINK_EINVAL;
1170                         return NULL;
1171                 } else {
1172                         usingname = true;
1173                 }
1174         }
1175
1176         if((int)params->devclass < 0 || params->devclass > _DEV_CLASS_MAX) {
1177                 logger(NULL, MESHLINK_ERROR, "Invalid devclass given!\n");
1178                 meshlink_errno = MESHLINK_EINVAL;
1179                 return NULL;
1180         }
1181
1182         if((params->key && !params->keylen) || (!params->key && params->keylen)) {
1183                 logger(NULL, MESHLINK_ERROR, "Invalid key length!\n");
1184                 meshlink_errno = MESHLINK_EINVAL;
1185                 return NULL;
1186         }
1187
1188         meshlink_handle_t *mesh = xzalloc(sizeof(meshlink_handle_t));
1189
1190         if(params->confbase) {
1191                 mesh->confbase = xstrdup(params->confbase);
1192         }
1193
1194         mesh->appname = xstrdup(params->appname);
1195         mesh->devclass = params->devclass;
1196         mesh->discovery = true;
1197         mesh->invitation_timeout = 604800; // 1 week
1198         mesh->netns = params->netns;
1199         mesh->submeshes = NULL;
1200
1201         if(usingname) {
1202                 mesh->name = xstrdup(params->name);
1203         }
1204
1205         // Hash the key
1206         if(params->key) {
1207                 mesh->config_key = xmalloc(CHACHA_POLY1305_KEYLEN);
1208
1209                 if(!prf(params->key, params->keylen, "MeshLink configuration key", 26, mesh->config_key, CHACHA_POLY1305_KEYLEN)) {
1210                         logger(NULL, MESHLINK_ERROR, "Error creating configuration key!\n");
1211                         meshlink_close(mesh);
1212                         meshlink_errno = MESHLINK_EINTERNAL;
1213                         return NULL;
1214                 }
1215         }
1216
1217         // initialize mutex
1218         pthread_mutexattr_t attr;
1219         pthread_mutexattr_init(&attr);
1220         pthread_mutexattr_settype(&attr, PTHREAD_MUTEX_RECURSIVE);
1221         pthread_mutex_init(&(mesh->mesh_mutex), &attr);
1222
1223         mesh->threadstarted = false;
1224         event_loop_init(&mesh->loop);
1225         mesh->loop.data = mesh;
1226
1227         meshlink_queue_init(&mesh->outpacketqueue);
1228
1229         // If no configuration exists yet, create it.
1230
1231         if(!meshlink_confbase_exists(mesh)) {
1232                 if(!meshlink_setup(mesh)) {
1233                         logger(NULL, MESHLINK_ERROR, "Cannot create initial configuration\n");
1234                         meshlink_close(mesh);
1235                         return NULL;
1236                 }
1237         } else {
1238                 if(!meshlink_read_config(mesh)) {
1239                         logger(NULL, MESHLINK_ERROR, "Cannot read main configuration\n");
1240                         meshlink_close(mesh);
1241                         return NULL;
1242                 }
1243         }
1244
1245 #ifdef HAVE_MINGW
1246         struct WSAData wsa_state;
1247         WSAStartup(MAKEWORD(2, 2), &wsa_state);
1248 #endif
1249
1250         // Setup up everything
1251         // TODO: we should not open listening sockets yet
1252
1253         bool success = false;
1254
1255         if(mesh->netns != -1) {
1256 #ifdef HAVE_SETNS
1257                 pthread_t thr;
1258
1259                 if(pthread_create(&thr, NULL, setup_network_in_netns_thread, mesh) == 0) {
1260                         void *retval = NULL;
1261                         success = pthread_join(thr, &retval) == 0 && retval;
1262                 }
1263
1264 #else
1265                 meshlink_errno = MESHLINK_EINTERNAL;
1266                 return NULL;
1267
1268 #endif // HAVE_SETNS
1269         } else {
1270                 success = setup_network(mesh);
1271                 add_local_addresses(mesh);
1272         }
1273
1274         if(!success) {
1275                 meshlink_close(mesh);
1276                 meshlink_errno = MESHLINK_ENETWORK;
1277                 return NULL;
1278         }
1279
1280         add_local_addresses(mesh);
1281         node_write_config(mesh, mesh->self);
1282
1283         idle_set(&mesh->loop, idle, mesh);
1284
1285         logger(NULL, MESHLINK_DEBUG, "meshlink_open returning\n");
1286         return mesh;
1287 }
1288
1289 meshlink_submesh_t *meshlink_submesh_open(meshlink_handle_t  *mesh, const char *submesh) {
1290         meshlink_submesh_t *s = NULL;
1291
1292         if(!mesh) {
1293                 logger(NULL, MESHLINK_ERROR, "No mesh handle given!\n");
1294                 meshlink_errno = MESHLINK_EINVAL;
1295                 return NULL;
1296         }
1297
1298         if(!submesh || !*submesh) {
1299                 logger(NULL, MESHLINK_ERROR, "No submesh name given!\n");
1300                 meshlink_errno = MESHLINK_EINVAL;
1301                 return NULL;
1302         }
1303
1304         //lock mesh->nodes
1305         pthread_mutex_lock(&(mesh->mesh_mutex));
1306
1307         s = (meshlink_submesh_t *)create_submesh(mesh, submesh);
1308
1309         pthread_mutex_unlock(&(mesh->mesh_mutex));
1310
1311         return s;
1312 }
1313
1314 static void *meshlink_main_loop(void *arg) {
1315         meshlink_handle_t *mesh = arg;
1316
1317         if(mesh->netns != -1) {
1318 #ifdef HAVE_SETNS
1319
1320                 if(setns(mesh->netns, CLONE_NEWNET) != 0) {
1321                         return NULL;
1322                 }
1323
1324 #else
1325                 return NULL;
1326 #endif // HAVE_SETNS
1327         }
1328
1329         pthread_mutex_lock(&(mesh->mesh_mutex));
1330
1331         logger(mesh, MESHLINK_DEBUG, "Starting main_loop...\n");
1332         main_loop(mesh);
1333         logger(mesh, MESHLINK_DEBUG, "main_loop returned.\n");
1334
1335         pthread_mutex_unlock(&(mesh->mesh_mutex));
1336         return NULL;
1337 }
1338
1339 bool meshlink_start(meshlink_handle_t *mesh) {
1340         assert(mesh->self);
1341         assert(mesh->private_key);
1342
1343         if(!mesh) {
1344                 meshlink_errno = MESHLINK_EINVAL;
1345                 return false;
1346         }
1347
1348         logger(mesh, MESHLINK_DEBUG, "meshlink_start called\n");
1349
1350         pthread_mutex_lock(&(mesh->mesh_mutex));
1351
1352         if(mesh->threadstarted) {
1353                 logger(mesh, MESHLINK_DEBUG, "thread was already running\n");
1354                 pthread_mutex_unlock(&(mesh->mesh_mutex));
1355                 return true;
1356         }
1357
1358         if(mesh->listen_socket[0].tcp.fd < 0) {
1359                 logger(mesh, MESHLINK_ERROR, "Listening socket not open\n");
1360                 meshlink_errno = MESHLINK_ENETWORK;
1361                 return false;
1362         }
1363
1364         mesh->thedatalen = 0;
1365
1366         // TODO: open listening sockets first
1367
1368         //Check that a valid name is set
1369         if(!mesh->name) {
1370                 logger(mesh, MESHLINK_DEBUG, "No name given!\n");
1371                 meshlink_errno = MESHLINK_EINVAL;
1372                 pthread_mutex_unlock(&(mesh->mesh_mutex));
1373                 return false;
1374         }
1375
1376         init_outgoings(mesh);
1377
1378         // Start the main thread
1379
1380         event_loop_start(&mesh->loop);
1381
1382         if(pthread_create(&mesh->thread, NULL, meshlink_main_loop, mesh) != 0) {
1383                 logger(mesh, MESHLINK_DEBUG, "Could not start thread: %s\n", strerror(errno));
1384                 memset(&mesh->thread, 0, sizeof(mesh)->thread);
1385                 meshlink_errno = MESHLINK_EINTERNAL;
1386                 event_loop_stop(&mesh->loop);
1387                 pthread_mutex_unlock(&(mesh->mesh_mutex));
1388                 return false;
1389         }
1390
1391         mesh->threadstarted = true;
1392
1393 #if HAVE_CATTA
1394
1395         if(mesh->discovery) {
1396                 discovery_start(mesh);
1397         }
1398
1399 #endif
1400
1401         assert(mesh->self->ecdsa);
1402         assert(!memcmp((uint8_t *)mesh->self->ecdsa + 64, (uint8_t *)mesh->private_key + 64, 32));
1403
1404
1405         pthread_mutex_unlock(&(mesh->mesh_mutex));
1406         return true;
1407 }
1408
1409 void meshlink_stop(meshlink_handle_t *mesh) {
1410         if(!mesh) {
1411                 meshlink_errno = MESHLINK_EINVAL;
1412                 return;
1413         }
1414
1415         pthread_mutex_lock(&(mesh->mesh_mutex));
1416         logger(mesh, MESHLINK_DEBUG, "meshlink_stop called\n");
1417
1418 #if HAVE_CATTA
1419
1420         // Stop discovery
1421         if(mesh->discovery) {
1422                 discovery_stop(mesh);
1423         }
1424
1425 #endif
1426
1427         // Shut down the main thread
1428         event_loop_stop(&mesh->loop);
1429
1430         // Send ourselves a UDP packet to kick the event loop
1431         for(int i = 0; i < mesh->listen_sockets; i++) {
1432                 sockaddr_t sa;
1433                 socklen_t salen = sizeof(sa.sa);
1434
1435                 if(getsockname(mesh->listen_socket[i].udp.fd, &sa.sa, &salen) == -1) {
1436                         logger(mesh, MESHLINK_ERROR, "System call `%s' failed: %s", "getsockname", sockstrerror(sockerrno));
1437                         continue;
1438                 }
1439
1440                 if(sendto(mesh->listen_socket[i].udp.fd, "", 1, MSG_NOSIGNAL, &sa.sa, salen) == -1) {
1441                         logger(mesh, MESHLINK_ERROR, "Could not send a UDP packet to ourself: %s", sockstrerror(sockerrno));
1442                 }
1443         }
1444
1445         if(mesh->threadstarted) {
1446                 // Wait for the main thread to finish
1447                 pthread_mutex_unlock(&(mesh->mesh_mutex));
1448                 pthread_join(mesh->thread, NULL);
1449                 pthread_mutex_lock(&(mesh->mesh_mutex));
1450
1451                 mesh->threadstarted = false;
1452         }
1453
1454         // Close all metaconnections
1455         if(mesh->connections) {
1456                 for(list_node_t *node = mesh->connections->head, *next; node; node = next) {
1457                         next = node->next;
1458                         connection_t *c = node->data;
1459                         c->outgoing = NULL;
1460                         terminate_connection(mesh, c, false);
1461                 }
1462         }
1463
1464         exit_outgoings(mesh);
1465
1466         // Write out any changed node config files
1467         if(mesh->nodes) {
1468                 for splay_each(node_t, n, mesh->nodes) {
1469                         if(n->status.dirty) {
1470                                 node_write_config(mesh, n);
1471                                 n->status.dirty = false;
1472                         }
1473                 }
1474         }
1475
1476         pthread_mutex_unlock(&(mesh->mesh_mutex));
1477 }
1478
1479 void meshlink_close(meshlink_handle_t *mesh) {
1480         if(!mesh) {
1481                 meshlink_errno = MESHLINK_EINVAL;
1482                 return;
1483         }
1484
1485         // stop can be called even if mesh has not been started
1486         meshlink_stop(mesh);
1487
1488         // lock is not released after this
1489         pthread_mutex_lock(&(mesh->mesh_mutex));
1490
1491         // Close and free all resources used.
1492
1493         close_network_connections(mesh);
1494
1495         logger(mesh, MESHLINK_INFO, "Terminating");
1496
1497         event_loop_exit(&mesh->loop);
1498
1499 #ifdef HAVE_MINGW
1500
1501         if(mesh->confbase) {
1502                 WSACleanup();
1503         }
1504
1505 #endif
1506
1507         ecdsa_free(mesh->invitation_key);
1508
1509         if(mesh->netns != -1) {
1510                 close(mesh->netns);
1511         }
1512
1513         free(mesh->name);
1514         free(mesh->appname);
1515         free(mesh->confbase);
1516         free(mesh->config_key);
1517         ecdsa_free(mesh->private_key);
1518         pthread_mutex_destroy(&(mesh->mesh_mutex));
1519
1520         main_config_unlock(mesh);
1521
1522         memset(mesh, 0, sizeof(*mesh));
1523
1524         free(mesh);
1525 }
1526
1527 bool meshlink_destroy(const char *confbase) {
1528         if(!confbase) {
1529                 meshlink_errno = MESHLINK_EINVAL;
1530                 return false;
1531         }
1532
1533         if(!config_destroy(confbase, "current")) {
1534                 logger(NULL, MESHLINK_ERROR, "Cannot remove confbase sub-directories %s: %s\n", confbase, strerror(errno));
1535                 return false;
1536         }
1537
1538         config_destroy(confbase, "new");
1539         config_destroy(confbase, "old");
1540
1541         if(rmdir(confbase) && errno != ENOENT) {
1542                 logger(NULL, MESHLINK_ERROR, "Cannot remove directory %s: %s\n", confbase, strerror(errno));
1543                 meshlink_errno = MESHLINK_ESTORAGE;
1544                 return false;
1545         }
1546
1547         return true;
1548 }
1549
1550 void meshlink_set_receive_cb(meshlink_handle_t *mesh, meshlink_receive_cb_t cb) {
1551         if(!mesh) {
1552                 meshlink_errno = MESHLINK_EINVAL;
1553                 return;
1554         }
1555
1556         pthread_mutex_lock(&(mesh->mesh_mutex));
1557         mesh->receive_cb = cb;
1558         pthread_mutex_unlock(&(mesh->mesh_mutex));
1559 }
1560
1561 void meshlink_set_connection_try_cb(meshlink_handle_t *mesh, meshlink_connection_try_cb_t cb) {
1562         if(!mesh) {
1563                 meshlink_errno = MESHLINK_EINVAL;
1564                 return;
1565         }
1566
1567         pthread_mutex_lock(&(mesh->mesh_mutex));
1568         mesh->connection_try_cb = cb;
1569         pthread_mutex_unlock(&(mesh->mesh_mutex));
1570 }
1571
1572 void meshlink_set_node_status_cb(meshlink_handle_t *mesh, meshlink_node_status_cb_t cb) {
1573         if(!mesh) {
1574                 meshlink_errno = MESHLINK_EINVAL;
1575                 return;
1576         }
1577
1578         pthread_mutex_lock(&(mesh->mesh_mutex));
1579         mesh->node_status_cb = cb;
1580         pthread_mutex_unlock(&(mesh->mesh_mutex));
1581 }
1582
1583 void meshlink_set_node_duplicate_cb(meshlink_handle_t *mesh, meshlink_node_duplicate_cb_t cb) {
1584         if(!mesh) {
1585                 meshlink_errno = MESHLINK_EINVAL;
1586                 return;
1587         }
1588
1589         pthread_mutex_lock(&(mesh->mesh_mutex));
1590         mesh->node_duplicate_cb = cb;
1591         pthread_mutex_unlock(&(mesh->mesh_mutex));
1592 }
1593
1594 void meshlink_set_log_cb(meshlink_handle_t *mesh, meshlink_log_level_t level, meshlink_log_cb_t cb) {
1595         if(mesh) {
1596                 pthread_mutex_lock(&(mesh->mesh_mutex));
1597                 mesh->log_cb = cb;
1598                 mesh->log_level = cb ? level : 0;
1599                 pthread_mutex_unlock(&(mesh->mesh_mutex));
1600         } else {
1601                 global_log_cb = cb;
1602                 global_log_level = cb ? level : 0;
1603         }
1604 }
1605
1606 bool meshlink_send(meshlink_handle_t *mesh, meshlink_node_t *destination, const void *data, size_t len) {
1607         meshlink_packethdr_t *hdr;
1608
1609         // Validate arguments
1610         if(!mesh || !destination || len >= MAXSIZE - sizeof(*hdr)) {
1611                 meshlink_errno = MESHLINK_EINVAL;
1612                 return false;
1613         }
1614
1615         if(!len) {
1616                 return true;
1617         }
1618
1619         if(!data) {
1620                 meshlink_errno = MESHLINK_EINVAL;
1621                 return false;
1622         }
1623
1624         node_t *n = (node_t *)destination;
1625
1626         if(n->status.blacklisted) {
1627                 logger(mesh, MESHLINK_ERROR, "Node %s blacklisted, dropping packet\n", n->name);
1628                 return false;
1629         }
1630
1631         // Prepare the packet
1632         vpn_packet_t *packet = malloc(sizeof(*packet));
1633
1634         if(!packet) {
1635                 meshlink_errno = MESHLINK_ENOMEM;
1636                 return false;
1637         }
1638
1639         packet->probe = false;
1640         packet->tcp = false;
1641         packet->len = len + sizeof(*hdr);
1642
1643         hdr = (meshlink_packethdr_t *)packet->data;
1644         memset(hdr, 0, sizeof(*hdr));
1645         // leave the last byte as 0 to make sure strings are always
1646         // null-terminated if they are longer than the buffer
1647         strncpy((char *)hdr->destination, destination->name, (sizeof(hdr)->destination) - 1);
1648         strncpy((char *)hdr->source, mesh->self->name, (sizeof(hdr)->source) - 1);
1649
1650         memcpy(packet->data + sizeof(*hdr), data, len);
1651
1652         // Queue it
1653         if(!meshlink_queue_push(&mesh->outpacketqueue, packet)) {
1654                 free(packet);
1655                 meshlink_errno = MESHLINK_ENOMEM;
1656                 return false;
1657         }
1658
1659         // Notify event loop
1660         signal_trigger(&(mesh->loop), &(mesh->datafromapp));
1661
1662         return true;
1663 }
1664
1665 void meshlink_send_from_queue(event_loop_t *loop, meshlink_handle_t *mesh) {
1666         (void)loop;
1667         vpn_packet_t *packet = meshlink_queue_pop(&mesh->outpacketqueue);
1668
1669         if(!packet) {
1670                 return;
1671         }
1672
1673         mesh->self->in_packets++;
1674         mesh->self->in_bytes += packet->len;
1675         route(mesh, mesh->self, packet);
1676 }
1677
1678 ssize_t meshlink_get_pmtu(meshlink_handle_t *mesh, meshlink_node_t *destination) {
1679         if(!mesh || !destination) {
1680                 meshlink_errno = MESHLINK_EINVAL;
1681                 return -1;
1682         }
1683
1684         pthread_mutex_lock(&(mesh->mesh_mutex));
1685
1686         node_t *n = (node_t *)destination;
1687
1688         if(!n->status.reachable) {
1689                 pthread_mutex_unlock(&(mesh->mesh_mutex));
1690                 return 0;
1691
1692         } else if(n->mtuprobes > 30 && n->minmtu) {
1693                 pthread_mutex_unlock(&(mesh->mesh_mutex));
1694                 return n->minmtu;
1695         } else {
1696                 pthread_mutex_unlock(&(mesh->mesh_mutex));
1697                 return MTU;
1698         }
1699 }
1700
1701 char *meshlink_get_fingerprint(meshlink_handle_t *mesh, meshlink_node_t *node) {
1702         if(!mesh || !node) {
1703                 meshlink_errno = MESHLINK_EINVAL;
1704                 return NULL;
1705         }
1706
1707         pthread_mutex_lock(&(mesh->mesh_mutex));
1708
1709         node_t *n = (node_t *)node;
1710
1711         if(!node_read_public_key(mesh, n) || !n->ecdsa) {
1712                 meshlink_errno = MESHLINK_EINTERNAL;
1713                 pthread_mutex_unlock(&(mesh->mesh_mutex));
1714                 return false;
1715         }
1716
1717         char *fingerprint = ecdsa_get_base64_public_key(n->ecdsa);
1718
1719         if(!fingerprint) {
1720                 meshlink_errno = MESHLINK_EINTERNAL;
1721         }
1722
1723         pthread_mutex_unlock(&(mesh->mesh_mutex));
1724         return fingerprint;
1725 }
1726
1727 meshlink_node_t *meshlink_get_self(meshlink_handle_t *mesh) {
1728         if(!mesh) {
1729                 meshlink_errno = MESHLINK_EINVAL;
1730                 return NULL;
1731         }
1732
1733         return (meshlink_node_t *)mesh->self;
1734 }
1735
1736 meshlink_node_t *meshlink_get_node(meshlink_handle_t *mesh, const char *name) {
1737         if(!mesh || !name) {
1738                 meshlink_errno = MESHLINK_EINVAL;
1739                 return NULL;
1740         }
1741
1742         meshlink_node_t *node = NULL;
1743
1744         pthread_mutex_lock(&(mesh->mesh_mutex));
1745         node = (meshlink_node_t *)lookup_node(mesh, (char *)name); // TODO: make lookup_node() use const
1746         pthread_mutex_unlock(&(mesh->mesh_mutex));
1747         return node;
1748 }
1749
1750 meshlink_submesh_t *meshlink_get_submesh(meshlink_handle_t *mesh, const char *name) {
1751         if(!mesh || !name) {
1752                 meshlink_errno = MESHLINK_EINVAL;
1753                 return NULL;
1754         }
1755
1756         meshlink_submesh_t *submesh = NULL;
1757
1758         pthread_mutex_lock(&(mesh->mesh_mutex));
1759         submesh = (meshlink_submesh_t *)lookup_submesh(mesh, name);
1760         pthread_mutex_unlock(&(mesh->mesh_mutex));
1761         return submesh;
1762 }
1763
1764 meshlink_node_t **meshlink_get_all_nodes(meshlink_handle_t *mesh, meshlink_node_t **nodes, size_t *nmemb) {
1765         if(!mesh || !nmemb || (*nmemb && !nodes)) {
1766                 meshlink_errno = MESHLINK_EINVAL;
1767                 return NULL;
1768         }
1769
1770         meshlink_node_t **result;
1771
1772         //lock mesh->nodes
1773         pthread_mutex_lock(&(mesh->mesh_mutex));
1774
1775         *nmemb = mesh->nodes->count;
1776         result = realloc(nodes, *nmemb * sizeof(*nodes));
1777
1778         if(result) {
1779                 meshlink_node_t **p = result;
1780
1781                 for splay_each(node_t, n, mesh->nodes) {
1782                         *p++ = (meshlink_node_t *)n;
1783                 }
1784         } else {
1785                 *nmemb = 0;
1786                 free(nodes);
1787                 meshlink_errno = MESHLINK_ENOMEM;
1788         }
1789
1790         pthread_mutex_unlock(&(mesh->mesh_mutex));
1791
1792         return result;
1793 }
1794
1795 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) {
1796         meshlink_node_t **result;
1797
1798         pthread_mutex_lock(&(mesh->mesh_mutex));
1799
1800         *nmemb = 0;
1801
1802         for splay_each(node_t, n, mesh->nodes) {
1803                 if(true == search_node(n, condition)) {
1804                         *nmemb = *nmemb + 1;
1805                 }
1806         }
1807
1808         if(*nmemb == 0) {
1809                 free(nodes);
1810                 pthread_mutex_unlock(&(mesh->mesh_mutex));
1811                 return NULL;
1812         }
1813
1814         result = realloc(nodes, *nmemb * sizeof(*nodes));
1815
1816         if(result) {
1817                 meshlink_node_t **p = result;
1818
1819                 for splay_each(node_t, n, mesh->nodes) {
1820                         if(true == search_node(n, condition)) {
1821                                 *p++ = (meshlink_node_t *)n;
1822                         }
1823                 }
1824         } else {
1825                 *nmemb = 0;
1826                 free(nodes);
1827                 meshlink_errno = MESHLINK_ENOMEM;
1828         }
1829
1830         pthread_mutex_unlock(&(mesh->mesh_mutex));
1831
1832         return result;
1833 }
1834
1835 static bool search_node_by_dev_class(const node_t *node, const void *condition) {
1836         dev_class_t *devclass = (dev_class_t *)condition;
1837
1838         if(*devclass == (dev_class_t)node->devclass) {
1839                 return true;
1840         }
1841
1842         return false;
1843 }
1844
1845 static bool search_node_by_submesh(const node_t *node, const void *condition) {
1846         if(condition == node->submesh) {
1847                 return true;
1848         }
1849
1850         return false;
1851 }
1852
1853 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) {
1854         if(!mesh || ((int)devclass < 0) || (devclass > _DEV_CLASS_MAX) || !nmemb) {
1855                 meshlink_errno = MESHLINK_EINVAL;
1856                 return NULL;
1857         }
1858
1859         return meshlink_get_all_nodes_by_condition(mesh, &devclass, nodes, nmemb, search_node_by_dev_class);
1860 }
1861
1862 meshlink_node_t **meshlink_get_all_nodes_by_submesh(meshlink_handle_t *mesh, meshlink_submesh_t *submesh, meshlink_node_t **nodes, size_t *nmemb) {
1863         if(!mesh || !submesh || !nmemb) {
1864                 meshlink_errno = MESHLINK_EINVAL;
1865                 return NULL;
1866         }
1867
1868         return meshlink_get_all_nodes_by_condition(mesh, submesh, nodes, nmemb, search_node_by_submesh);
1869 }
1870
1871 dev_class_t meshlink_get_node_dev_class(meshlink_handle_t *mesh, meshlink_node_t *node) {
1872         if(!mesh || !node) {
1873                 meshlink_errno = MESHLINK_EINVAL;
1874                 return -1;
1875         }
1876
1877         dev_class_t devclass;
1878
1879         pthread_mutex_lock(&(mesh->mesh_mutex));
1880
1881         devclass = ((node_t *)node)->devclass;
1882
1883         pthread_mutex_unlock(&(mesh->mesh_mutex));
1884
1885         return devclass;
1886 }
1887
1888 meshlink_submesh_t *meshlink_get_node_submesh(meshlink_handle_t *mesh, meshlink_node_t *node) {
1889         if(!mesh || !node) {
1890                 meshlink_errno = MESHLINK_EINVAL;
1891                 return NULL;
1892         }
1893
1894         node_t *n = (node_t *)node;
1895
1896         meshlink_submesh_t *s;
1897
1898         s = (meshlink_submesh_t *)n->submesh;
1899
1900         return s;
1901 }
1902
1903 bool meshlink_sign(meshlink_handle_t *mesh, const void *data, size_t len, void *signature, size_t *siglen) {
1904         if(!mesh || !data || !len || !signature || !siglen) {
1905                 meshlink_errno = MESHLINK_EINVAL;
1906                 return false;
1907         }
1908
1909         if(*siglen < MESHLINK_SIGLEN) {
1910                 meshlink_errno = MESHLINK_EINVAL;
1911                 return false;
1912         }
1913
1914         pthread_mutex_lock(&(mesh->mesh_mutex));
1915
1916         if(!ecdsa_sign(mesh->private_key, data, len, signature)) {
1917                 meshlink_errno = MESHLINK_EINTERNAL;
1918                 pthread_mutex_unlock(&(mesh->mesh_mutex));
1919                 return false;
1920         }
1921
1922         *siglen = MESHLINK_SIGLEN;
1923         pthread_mutex_unlock(&(mesh->mesh_mutex));
1924         return true;
1925 }
1926
1927 bool meshlink_verify(meshlink_handle_t *mesh, meshlink_node_t *source, const void *data, size_t len, const void *signature, size_t siglen) {
1928         if(!mesh || !data || !len || !signature) {
1929                 meshlink_errno = MESHLINK_EINVAL;
1930                 return false;
1931         }
1932
1933         if(siglen != MESHLINK_SIGLEN) {
1934                 meshlink_errno = MESHLINK_EINVAL;
1935                 return false;
1936         }
1937
1938         pthread_mutex_lock(&(mesh->mesh_mutex));
1939
1940         bool rval = false;
1941
1942         struct node_t *n = (struct node_t *)source;
1943
1944         if(!node_read_public_key(mesh, n)) {
1945                 meshlink_errno = MESHLINK_EINTERNAL;
1946                 rval = false;
1947         } else {
1948                 rval = ecdsa_verify(((struct node_t *)source)->ecdsa, data, len, signature);
1949         }
1950
1951         pthread_mutex_unlock(&(mesh->mesh_mutex));
1952         return rval;
1953 }
1954
1955 static bool refresh_invitation_key(meshlink_handle_t *mesh) {
1956         pthread_mutex_lock(&(mesh->mesh_mutex));
1957
1958         size_t count = invitation_purge_old(mesh, time(NULL) - mesh->invitation_timeout);
1959
1960         if(!count) {
1961                 // TODO: Update invitation key if necessary?
1962         }
1963
1964         pthread_mutex_unlock(&(mesh->mesh_mutex));
1965
1966         return mesh->invitation_key;
1967 }
1968
1969 bool meshlink_set_canonical_address(meshlink_handle_t *mesh, meshlink_node_t *node, const char *address, const char *port) {
1970         if(!mesh || !node || !address) {
1971                 meshlink_errno = MESHLINK_EINVAL;
1972                 return false;
1973         }
1974
1975         if(!is_valid_hostname(address)) {
1976                 logger(mesh, MESHLINK_DEBUG, "Invalid character in address: %s\n", address);
1977                 meshlink_errno = MESHLINK_EINVAL;
1978                 return false;
1979         }
1980
1981         if(port && !is_valid_port(port)) {
1982                 logger(mesh, MESHLINK_DEBUG, "Invalid character in port: %s\n", address);
1983                 meshlink_errno = MESHLINK_EINVAL;
1984                 return false;
1985         }
1986
1987         char *canonical_address;
1988
1989         if(port) {
1990                 xasprintf(&canonical_address, "%s %s", address, port);
1991         } else {
1992                 canonical_address = xstrdup(address);
1993         }
1994
1995         pthread_mutex_lock(&(mesh->mesh_mutex));
1996
1997         node_t *n = (node_t *)node;
1998         free(n->canonical_address);
1999         n->canonical_address = canonical_address;
2000         node_write_config(mesh, n);
2001
2002         pthread_mutex_unlock(&(mesh->mesh_mutex));
2003
2004         return true;
2005 }
2006
2007 bool meshlink_add_address(meshlink_handle_t *mesh, const char *address) {
2008         return meshlink_set_canonical_address(mesh, (meshlink_node_t *)mesh->self, address, NULL);
2009 }
2010
2011 bool meshlink_add_external_address(meshlink_handle_t *mesh) {
2012         if(!mesh) {
2013                 meshlink_errno = MESHLINK_EINVAL;
2014                 return false;
2015         }
2016
2017         char *address = meshlink_get_external_address(mesh);
2018
2019         if(!address) {
2020                 return false;
2021         }
2022
2023         bool rval = meshlink_add_address(mesh, address);
2024         free(address);
2025
2026         return rval;
2027 }
2028
2029 int meshlink_get_port(meshlink_handle_t *mesh) {
2030         if(!mesh) {
2031                 meshlink_errno = MESHLINK_EINVAL;
2032                 return -1;
2033         }
2034
2035         if(!mesh->myport) {
2036                 meshlink_errno = MESHLINK_EINTERNAL;
2037                 return -1;
2038         }
2039
2040         return atoi(mesh->myport);
2041 }
2042
2043 bool meshlink_set_port(meshlink_handle_t *mesh, int port) {
2044         if(!mesh || port < 0 || port >= 65536 || mesh->threadstarted) {
2045                 meshlink_errno = MESHLINK_EINVAL;
2046                 return false;
2047         }
2048
2049         if(mesh->myport && port == atoi(mesh->myport)) {
2050                 return true;
2051         }
2052
2053         if(!try_bind(port)) {
2054                 meshlink_errno = MESHLINK_ENETWORK;
2055                 return false;
2056         }
2057
2058         devtool_trybind_probe();
2059
2060         bool rval = false;
2061
2062         pthread_mutex_lock(&(mesh->mesh_mutex));
2063
2064         if(mesh->threadstarted) {
2065                 meshlink_errno = MESHLINK_EINVAL;
2066                 goto done;
2067         }
2068
2069         free(mesh->myport);
2070         xasprintf(&mesh->myport, "%d", port);
2071
2072         /* Write meshlink.conf with the updated port number */
2073         write_main_config_files(mesh);
2074
2075         /* Close down the network. This also deletes mesh->self. */
2076         close_network_connections(mesh);
2077
2078         /* Recreate mesh->self. */
2079         mesh->self = new_node();
2080         mesh->self->name = xstrdup(mesh->name);
2081         mesh->self->devclass = mesh->devclass;
2082         xasprintf(&mesh->myport, "%d", port);
2083
2084         if(!node_read_public_key(mesh, mesh->self)) {
2085                 logger(NULL, MESHLINK_ERROR, "Could not read our host configuration file!");
2086                 meshlink_errno = MESHLINK_ESTORAGE;
2087                 free_node(mesh->self);
2088                 mesh->self = NULL;
2089         } else if(!setup_network(mesh)) {
2090                 meshlink_errno = MESHLINK_ENETWORK;
2091         } else {
2092                 rval = true;
2093         }
2094
2095 done:
2096         pthread_mutex_unlock(&(mesh->mesh_mutex));
2097
2098         return rval && meshlink_get_port(mesh) == port;
2099 }
2100
2101 void meshlink_set_invitation_timeout(meshlink_handle_t *mesh, int timeout) {
2102         mesh->invitation_timeout = timeout;
2103 }
2104
2105 char *meshlink_invite_ex(meshlink_handle_t *mesh, meshlink_submesh_t *submesh, const char *name, uint32_t flags) {
2106         meshlink_submesh_t *s = NULL;
2107
2108         if(!mesh) {
2109                 meshlink_errno = MESHLINK_EINVAL;
2110                 return NULL;
2111         }
2112
2113         if(submesh) {
2114                 s = (meshlink_submesh_t *)lookup_submesh(mesh, submesh->name);
2115
2116                 if(s != submesh) {
2117                         logger(mesh, MESHLINK_DEBUG, "Invalid SubMesh Handle.\n");
2118                         meshlink_errno = MESHLINK_EINVAL;
2119                         return NULL;
2120                 }
2121         } else {
2122                 s = (meshlink_submesh_t *)mesh->self->submesh;
2123         }
2124
2125         pthread_mutex_lock(&(mesh->mesh_mutex));
2126
2127         // Check validity of the new node's name
2128         if(!check_id(name)) {
2129                 logger(mesh, MESHLINK_DEBUG, "Invalid name for node.\n");
2130                 meshlink_errno = MESHLINK_EINVAL;
2131                 pthread_mutex_unlock(&(mesh->mesh_mutex));
2132                 return NULL;
2133         }
2134
2135         // Ensure no host configuration file with that name exists
2136         if(config_exists(mesh, "current", name)) {
2137                 logger(mesh, MESHLINK_DEBUG, "A host config file for %s already exists!\n", name);
2138                 meshlink_errno = MESHLINK_EEXIST;
2139                 pthread_mutex_unlock(&(mesh->mesh_mutex));
2140                 return NULL;
2141         }
2142
2143         // Ensure no other nodes know about this name
2144         if(meshlink_get_node(mesh, name)) {
2145                 logger(mesh, MESHLINK_DEBUG, "A node with name %s is already known!\n", name);
2146                 meshlink_errno = MESHLINK_EEXIST;
2147                 pthread_mutex_unlock(&(mesh->mesh_mutex));
2148                 return NULL;
2149         }
2150
2151         // Get the local address
2152         char *address = get_my_hostname(mesh, flags);
2153
2154         if(!address) {
2155                 logger(mesh, MESHLINK_DEBUG, "No Address known for ourselves!\n");
2156                 meshlink_errno = MESHLINK_ERESOLV;
2157                 pthread_mutex_unlock(&(mesh->mesh_mutex));
2158                 return NULL;
2159         }
2160
2161         if(!refresh_invitation_key(mesh)) {
2162                 meshlink_errno = MESHLINK_EINTERNAL;
2163                 pthread_mutex_unlock(&(mesh->mesh_mutex));
2164                 return NULL;
2165         }
2166
2167         char hash[64];
2168
2169         // Create a hash of the key.
2170         char *fingerprint = ecdsa_get_base64_public_key(mesh->invitation_key);
2171         sha512(fingerprint, strlen(fingerprint), hash);
2172         b64encode_urlsafe(hash, hash, 18);
2173
2174         // Create a random cookie for this invitation.
2175         char cookie[25];
2176         randomize(cookie, 18);
2177
2178         // Create a filename that doesn't reveal the cookie itself
2179         char buf[18 + strlen(fingerprint)];
2180         char cookiehash[64];
2181         memcpy(buf, cookie, 18);
2182         memcpy(buf + 18, fingerprint, sizeof(buf) - 18);
2183         sha512(buf, sizeof(buf), cookiehash);
2184         b64encode_urlsafe(cookiehash, cookiehash, 18);
2185
2186         b64encode_urlsafe(cookie, cookie, 18);
2187
2188         free(fingerprint);
2189
2190         /* Construct the invitation file */
2191         uint8_t outbuf[4096];
2192         packmsg_output_t inv = {outbuf, sizeof(outbuf)};
2193
2194         packmsg_add_uint32(&inv, MESHLINK_INVITATION_VERSION);
2195         packmsg_add_str(&inv, name);
2196         packmsg_add_str(&inv, s ? s->name : CORE_MESH);
2197         packmsg_add_int32(&inv, DEV_CLASS_UNKNOWN); /* TODO: allow this to be set by inviter? */
2198
2199         /* TODO: Add several host config files to bootstrap connections.
2200          * Note: make sure we only add config files of nodes that are in the core mesh or the same submesh,
2201          * and are not blacklisted.
2202          */
2203         config_t configs[5] = {NULL};
2204         int count = 0;
2205
2206         if(config_read(mesh, "current", mesh->self->name, &configs[count], mesh->config_key)) {
2207                 count++;
2208         }
2209
2210         /* Append host config files to the invitation file */
2211         packmsg_add_array(&inv, count);
2212
2213         for(int i = 0; i < count; i++) {
2214                 packmsg_add_bin(&inv, configs[i].buf, configs[i].len);
2215                 config_free(&configs[i]);
2216         }
2217
2218         config_t config = {outbuf, packmsg_output_size(&inv, outbuf)};
2219
2220         if(!invitation_write(mesh, "current", cookiehash, &config, mesh->config_key)) {
2221                 logger(mesh, MESHLINK_DEBUG, "Could not create invitation file %s: %s\n", cookiehash, strerror(errno));
2222                 meshlink_errno = MESHLINK_ESTORAGE;
2223                 pthread_mutex_unlock(&(mesh->mesh_mutex));
2224                 return NULL;
2225         }
2226
2227         // Create an URL from the local address, key hash and cookie
2228         char *url;
2229         xasprintf(&url, "%s/%s%s", address, hash, cookie);
2230         free(address);
2231
2232         pthread_mutex_unlock(&(mesh->mesh_mutex));
2233         return url;
2234 }
2235
2236 char *meshlink_invite(meshlink_handle_t *mesh, meshlink_submesh_t *submesh, const char *name) {
2237         return meshlink_invite_ex(mesh, submesh, name, 0);
2238 }
2239
2240 bool meshlink_join(meshlink_handle_t *mesh, const char *invitation) {
2241         if(!mesh || !invitation) {
2242                 meshlink_errno = MESHLINK_EINVAL;
2243                 return false;
2244         }
2245
2246         pthread_mutex_lock(&(mesh->mesh_mutex));
2247
2248         //Before doing meshlink_join make sure we are not connected to another mesh
2249         if(mesh->threadstarted) {
2250                 logger(mesh, MESHLINK_DEBUG, "Already connected to a mesh\n");
2251                 meshlink_errno = MESHLINK_EINVAL;
2252                 pthread_mutex_unlock(&(mesh->mesh_mutex));
2253                 return false;
2254         }
2255
2256         //TODO: think of a better name for this variable, or of a different way to tokenize the invitation URL.
2257         char copy[strlen(invitation) + 1];
2258         strcpy(copy, invitation);
2259
2260         // Split the invitation URL into a list of hostname/port tuples, a key hash and a cookie.
2261
2262         char *slash = strchr(copy, '/');
2263
2264         if(!slash) {
2265                 goto invalid;
2266         }
2267
2268         *slash++ = 0;
2269
2270         if(strlen(slash) != 48) {
2271                 goto invalid;
2272         }
2273
2274         char *address = copy;
2275         char *port = NULL;
2276
2277         if(!b64decode(slash, mesh->hash, 18) || !b64decode(slash + 24, mesh->cookie, 18)) {
2278                 goto invalid;
2279         }
2280
2281         // Generate a throw-away key for the invitation.
2282         ecdsa_t *key = ecdsa_generate();
2283
2284         if(!key) {
2285                 meshlink_errno = MESHLINK_EINTERNAL;
2286                 pthread_mutex_unlock(&(mesh->mesh_mutex));
2287                 return false;
2288         }
2289
2290         char *b64key = ecdsa_get_base64_public_key(key);
2291         char *comma;
2292         mesh->sock = -1;
2293
2294         while(address && *address) {
2295                 // We allow commas in the address part to support multiple addresses in one invitation URL.
2296                 comma = strchr(address, ',');
2297
2298                 if(comma) {
2299                         *comma++ = 0;
2300                 }
2301
2302                 // Split of the port
2303                 port = strrchr(address, ':');
2304
2305                 if(!port) {
2306                         goto invalid;
2307                 }
2308
2309                 *port++ = 0;
2310
2311                 // IPv6 address are enclosed in brackets, per RFC 3986
2312                 if(*address == '[') {
2313                         address++;
2314                         char *bracket = strchr(address, ']');
2315
2316                         if(!bracket) {
2317                                 goto invalid;
2318                         }
2319
2320                         *bracket++ = 0;
2321
2322                         if(*bracket) {
2323                                 goto invalid;
2324                         }
2325                 }
2326
2327                 // Connect to the meshlink daemon mentioned in the URL.
2328                 struct addrinfo *ai = str2addrinfo(address, port, SOCK_STREAM);
2329
2330                 if(ai) {
2331                         for(struct addrinfo *aip = ai; aip; aip = aip->ai_next) {
2332                                 mesh->sock = socket_in_netns(aip->ai_family, aip->ai_socktype, aip->ai_protocol, mesh->netns);
2333
2334                                 if(mesh->sock == -1) {
2335                                         logger(mesh, MESHLINK_DEBUG, "Could not open socket: %s\n", strerror(errno));
2336                                         meshlink_errno = MESHLINK_ENETWORK;
2337                                         continue;
2338                                 }
2339
2340                                 set_timeout(mesh->sock, 5000);
2341
2342                                 if(connect(mesh->sock, aip->ai_addr, aip->ai_addrlen)) {
2343                                         logger(mesh, MESHLINK_DEBUG, "Could not connect to %s port %s: %s\n", address, port, strerror(errno));
2344                                         meshlink_errno = MESHLINK_ENETWORK;
2345                                         closesocket(mesh->sock);
2346                                         mesh->sock = -1;
2347                                         continue;
2348                                 }
2349                         }
2350
2351                         freeaddrinfo(ai);
2352                 } else {
2353                         meshlink_errno = MESHLINK_ERESOLV;
2354                 }
2355
2356                 if(mesh->sock != -1 || !comma) {
2357                         break;
2358                 }
2359
2360                 address = comma;
2361         }
2362
2363         if(mesh->sock == -1) {
2364                 pthread_mutex_unlock(&mesh->mesh_mutex);
2365                 return false;
2366         }
2367
2368         logger(mesh, MESHLINK_DEBUG, "Connected to %s port %s...\n", address, port);
2369
2370         // Tell him we have an invitation, and give him our throw-away key.
2371
2372         mesh->blen = 0;
2373
2374         if(!sendline(mesh->sock, "0 ?%s %d.%d %s", b64key, PROT_MAJOR, PROT_MINOR, mesh->appname)) {
2375                 logger(mesh, MESHLINK_DEBUG, "Error sending request to %s port %s: %s\n", address, port, strerror(errno));
2376                 closesocket(mesh->sock);
2377                 meshlink_errno = MESHLINK_ENETWORK;
2378                 pthread_mutex_unlock(&(mesh->mesh_mutex));
2379                 return false;
2380         }
2381
2382         free(b64key);
2383
2384         char hisname[4096] = "";
2385         int code, hismajor, hisminor = 0;
2386
2387         if(!recvline(mesh, sizeof(mesh)->line) || sscanf(mesh->line, "%d %s %d.%d", &code, hisname, &hismajor, &hisminor) < 3 || code != 0 || hismajor != PROT_MAJOR || !check_id(hisname) || !recvline(mesh, sizeof(mesh)->line) || !rstrip(mesh->line) || sscanf(mesh->line, "%d ", &code) != 1 || code != ACK || strlen(mesh->line) < 3) {
2388                 logger(mesh, MESHLINK_DEBUG, "Cannot read greeting from peer\n");
2389                 closesocket(mesh->sock);
2390                 meshlink_errno = MESHLINK_ENETWORK;
2391                 pthread_mutex_unlock(&(mesh->mesh_mutex));
2392                 return false;
2393         }
2394
2395         // Check if the hash of the key he gave us matches the hash in the URL.
2396         char *fingerprint = mesh->line + 2;
2397         char hishash[64];
2398
2399         if(sha512(fingerprint, strlen(fingerprint), hishash)) {
2400                 logger(mesh, MESHLINK_DEBUG, "Could not create hash\n%s\n", mesh->line + 2);
2401                 meshlink_errno = MESHLINK_EINTERNAL;
2402                 pthread_mutex_unlock(&(mesh->mesh_mutex));
2403                 return false;
2404         }
2405
2406         if(memcmp(hishash, mesh->hash, 18)) {
2407                 logger(mesh, MESHLINK_DEBUG, "Peer has an invalid key!\n%s\n", mesh->line + 2);
2408                 meshlink_errno = MESHLINK_EPEER;
2409                 pthread_mutex_unlock(&(mesh->mesh_mutex));
2410                 return false;
2411
2412         }
2413
2414         ecdsa_t *hiskey = ecdsa_set_base64_public_key(fingerprint);
2415
2416         if(!hiskey) {
2417                 meshlink_errno = MESHLINK_EINTERNAL;
2418                 pthread_mutex_unlock(&(mesh->mesh_mutex));
2419                 return false;
2420         }
2421
2422         // Start an SPTPS session
2423         if(!sptps_start(&mesh->sptps, mesh, true, false, key, hiskey, meshlink_invitation_label, sizeof(meshlink_invitation_label), invitation_send, invitation_receive)) {
2424                 meshlink_errno = MESHLINK_EINTERNAL;
2425                 pthread_mutex_unlock(&(mesh->mesh_mutex));
2426                 return false;
2427         }
2428
2429         // Feed rest of input buffer to SPTPS
2430         if(!sptps_receive_data(&mesh->sptps, mesh->buffer, mesh->blen)) {
2431                 meshlink_errno = MESHLINK_EPEER;
2432                 pthread_mutex_unlock(&(mesh->mesh_mutex));
2433                 return false;
2434         }
2435
2436         int len;
2437
2438         while((len = recv(mesh->sock, mesh->line, sizeof(mesh)->line, 0))) {
2439                 if(len < 0) {
2440                         if(errno == EINTR) {
2441                                 continue;
2442                         }
2443
2444                         logger(mesh, MESHLINK_DEBUG, "Error reading data from %s port %s: %s\n", address, port, strerror(errno));
2445                         meshlink_errno = MESHLINK_ENETWORK;
2446                         pthread_mutex_unlock(&(mesh->mesh_mutex));
2447                         return false;
2448                 }
2449
2450                 if(!sptps_receive_data(&mesh->sptps, mesh->line, len)) {
2451                         meshlink_errno = MESHLINK_EPEER;
2452                         pthread_mutex_unlock(&(mesh->mesh_mutex));
2453                         return false;
2454                 }
2455         }
2456
2457         sptps_stop(&mesh->sptps);
2458         ecdsa_free(hiskey);
2459         ecdsa_free(key);
2460         closesocket(mesh->sock);
2461
2462         if(!mesh->success) {
2463                 logger(mesh, MESHLINK_DEBUG, "Connection closed by peer, invitation cancelled.\n");
2464                 meshlink_errno = MESHLINK_EPEER;
2465                 pthread_mutex_unlock(&(mesh->mesh_mutex));
2466                 return false;
2467         }
2468
2469         pthread_mutex_unlock(&(mesh->mesh_mutex));
2470         return true;
2471
2472 invalid:
2473         logger(mesh, MESHLINK_DEBUG, "Invalid invitation URL\n");
2474         meshlink_errno = MESHLINK_EINVAL;
2475         pthread_mutex_unlock(&(mesh->mesh_mutex));
2476         return false;
2477 }
2478
2479 char *meshlink_export(meshlink_handle_t *mesh) {
2480         if(!mesh) {
2481                 meshlink_errno = MESHLINK_EINVAL;
2482                 return NULL;
2483         }
2484
2485         // Create a config file on the fly.
2486
2487         uint8_t buf[4096];
2488         packmsg_output_t out = {buf, sizeof(buf)};
2489         packmsg_add_uint32(&out, MESHLINK_CONFIG_VERSION);
2490         packmsg_add_str(&out, mesh->name);
2491         packmsg_add_str(&out, CORE_MESH);
2492
2493         pthread_mutex_lock(&(mesh->mesh_mutex));
2494
2495         packmsg_add_int32(&out, mesh->self->devclass);
2496         packmsg_add_bool(&out, mesh->self->status.blacklisted);
2497         packmsg_add_bin(&out, ecdsa_get_public_key(mesh->private_key), 32);
2498         packmsg_add_str(&out, mesh->self->canonical_address ? mesh->self->canonical_address : "");
2499
2500         uint32_t count = 0;
2501
2502         for(uint32_t i = 0; i < 5; i++) {
2503                 if(mesh->self->recent[i].sa.sa_family) {
2504                         count++;
2505                 } else {
2506                         break;
2507                 }
2508         }
2509
2510         packmsg_add_array(&out, count);
2511
2512         for(uint32_t i = 0; i < count; i++) {
2513                 packmsg_add_sockaddr(&out, &mesh->self->recent[i]);
2514         }
2515
2516         pthread_mutex_unlock(&(mesh->mesh_mutex));
2517
2518         if(!packmsg_output_ok(&out)) {
2519                 logger(mesh, MESHLINK_DEBUG, "Error creating export data\n");
2520                 meshlink_errno = MESHLINK_EINTERNAL;
2521                 return NULL;
2522         }
2523
2524         // Prepare a base64-encoded packmsg array containing our config file
2525
2526         uint32_t len = packmsg_output_size(&out, buf);
2527         uint32_t len2 = ((len + 4) * 4) / 3 + 4;
2528         uint8_t *buf2 = xmalloc(len2);
2529         packmsg_output_t out2 = {buf2, len2};
2530         packmsg_add_array(&out2, 1);
2531         packmsg_add_bin(&out2, buf, packmsg_output_size(&out, buf));
2532
2533         if(!packmsg_output_ok(&out2)) {
2534                 logger(mesh, MESHLINK_DEBUG, "Error creating export data\n");
2535                 meshlink_errno = MESHLINK_EINTERNAL;
2536                 free(buf2);
2537                 return NULL;
2538         }
2539
2540         b64encode_urlsafe(buf2, (char *)buf2, packmsg_output_size(&out2, buf2));
2541
2542         return (char *)buf2;
2543 }
2544
2545 bool meshlink_import(meshlink_handle_t *mesh, const char *data) {
2546         if(!mesh || !data) {
2547                 meshlink_errno = MESHLINK_EINVAL;
2548                 return false;
2549         }
2550
2551         size_t datalen = strlen(data);
2552         uint8_t *buf = xmalloc(datalen);
2553         int buflen = b64decode(data, buf, datalen);
2554
2555         if(!buflen) {
2556                 logger(mesh, MESHLINK_DEBUG, "Invalid data\n");
2557                 meshlink_errno = MESHLINK_EPEER;
2558                 return false;
2559         }
2560
2561         packmsg_input_t in = {buf, buflen};
2562         uint32_t count = packmsg_get_array(&in);
2563
2564         if(!count) {
2565                 logger(mesh, MESHLINK_DEBUG, "Invalid data\n");
2566                 meshlink_errno = MESHLINK_EPEER;
2567                 return false;
2568         }
2569
2570         pthread_mutex_lock(&(mesh->mesh_mutex));
2571
2572         while(count--) {
2573                 const void *data;
2574                 uint32_t len = packmsg_get_bin_raw(&in, &data);
2575
2576                 if(!len) {
2577                         break;
2578                 }
2579
2580                 packmsg_input_t in2 = {data, len};
2581                 uint32_t version = packmsg_get_uint32(&in2);
2582                 char *name = packmsg_get_str_dup(&in2);
2583
2584                 if(!packmsg_input_ok(&in2) || version != MESHLINK_CONFIG_VERSION || !check_id(name)) {
2585                         free(name);
2586                         packmsg_input_invalidate(&in);
2587                         break;
2588                 }
2589
2590                 if(!check_id(name)) {
2591                         free(name);
2592                         break;
2593                 }
2594
2595                 node_t *n = lookup_node(mesh, name);
2596
2597                 if(n) {
2598                         logger(mesh, MESHLINK_DEBUG, "Node %s already exists, not importing\n", name);
2599                         free(name);
2600                         continue;
2601                 }
2602
2603                 n = new_node();
2604                 n->name = name;
2605
2606                 config_t config = {data, len};
2607
2608                 if(!node_read_from_config(mesh, n, &config)) {
2609                         free_node(n);
2610                         packmsg_input_invalidate(&in);
2611                         break;
2612                 }
2613
2614                 config_write(mesh, "current", n->name, &config, mesh->config_key);
2615                 node_add(mesh, n);
2616         }
2617
2618         pthread_mutex_unlock(&(mesh->mesh_mutex));
2619
2620         if(!packmsg_done(&in)) {
2621                 logger(mesh, MESHLINK_ERROR, "Invalid data\n");
2622                 meshlink_errno = MESHLINK_EPEER;
2623                 return false;
2624         }
2625
2626         return true;
2627 }
2628
2629 void meshlink_blacklist(meshlink_handle_t *mesh, meshlink_node_t *node) {
2630         if(!mesh || !node) {
2631                 meshlink_errno = MESHLINK_EINVAL;
2632                 return;
2633         }
2634
2635         pthread_mutex_lock(&(mesh->mesh_mutex));
2636
2637         node_t *n;
2638         n = (node_t *)node;
2639
2640         if(n == mesh->self) {
2641                 logger(mesh, MESHLINK_ERROR, "%s blacklisting itself?\n", node->name);
2642                 meshlink_errno = MESHLINK_EINVAL;
2643                 pthread_mutex_unlock(&(mesh->mesh_mutex));
2644                 return;
2645         }
2646
2647         if(n->status.blacklisted) {
2648                 logger(mesh, MESHLINK_DEBUG, "Node %s already blacklisted\n", node->name);
2649                 pthread_mutex_unlock(&(mesh->mesh_mutex));
2650                 return;
2651         }
2652
2653         n->status.blacklisted = true;
2654         node_write_config(mesh, n);
2655         logger(mesh, MESHLINK_DEBUG, "Blacklisted %s.\n", node->name);
2656
2657         //Immediately terminate any connections we have with the blacklisted node
2658         for list_each(connection_t, c, mesh->connections) {
2659                 if(c->node == n) {
2660                         terminate_connection(mesh, c, c->status.active);
2661                 }
2662         }
2663
2664         utcp_abort_all_connections(n->utcp);
2665
2666         n->mtu = 0;
2667         n->minmtu = 0;
2668         n->maxmtu = MTU;
2669         n->mtuprobes = 0;
2670         n->status.udp_confirmed = false;
2671
2672         if(n->status.reachable) {
2673                 update_node_status(mesh, n);
2674         }
2675
2676         pthread_mutex_unlock(&(mesh->mesh_mutex));
2677 }
2678
2679 void meshlink_whitelist(meshlink_handle_t *mesh, meshlink_node_t *node) {
2680         if(!mesh || !node) {
2681                 meshlink_errno = MESHLINK_EINVAL;
2682                 return;
2683         }
2684
2685         pthread_mutex_lock(&(mesh->mesh_mutex));
2686
2687         node_t *n = (node_t *)node;
2688
2689         if(!n->status.blacklisted) {
2690                 logger(mesh, MESHLINK_DEBUG, "Node %s was already whitelisted\n", node->name);
2691                 meshlink_errno = MESHLINK_EINVAL;
2692                 pthread_mutex_unlock(&(mesh->mesh_mutex));
2693                 return;
2694         }
2695
2696         n->status.blacklisted = false;
2697         node_write_config(mesh, n);
2698
2699         pthread_mutex_unlock(&(mesh->mesh_mutex));
2700         return;
2701 }
2702
2703 void meshlink_set_default_blacklist(meshlink_handle_t *mesh, bool blacklist) {
2704         mesh->default_blacklist = blacklist;
2705 }
2706
2707 /* Hint that a hostname may be found at an address
2708  * See header file for detailed comment.
2709  */
2710 void meshlink_hint_address(meshlink_handle_t *mesh, meshlink_node_t *node, const struct sockaddr *addr) {
2711         if(!mesh || !node || !addr) {
2712                 meshlink_errno = EINVAL;
2713                 return;
2714         }
2715
2716         pthread_mutex_lock(&(mesh->mesh_mutex));
2717
2718         node_t *n = (node_t *)node;
2719         memmove(n->recent + 1, n->recent, 4 * sizeof(*n->recent));
2720         memcpy(n->recent, addr, SALEN(*addr));
2721         node_write_config(mesh, n);
2722
2723         pthread_mutex_unlock(&(mesh->mesh_mutex));
2724         // @TODO do we want to fire off a connection attempt right away?
2725 }
2726
2727 static bool channel_pre_accept(struct utcp *utcp, uint16_t port) {
2728         (void)port;
2729         node_t *n = utcp->priv;
2730         meshlink_handle_t *mesh = n->mesh;
2731         return mesh->channel_accept_cb;
2732 }
2733
2734 static ssize_t channel_recv(struct utcp_connection *connection, const void *data, size_t len) {
2735         meshlink_channel_t *channel = connection->priv;
2736
2737         if(!channel) {
2738                 abort();
2739         }
2740
2741         node_t *n = channel->node;
2742         meshlink_handle_t *mesh = n->mesh;
2743
2744         if(n->status.destroyed) {
2745                 meshlink_channel_close(mesh, channel);
2746         } else if(channel->receive_cb) {
2747                 channel->receive_cb(mesh, channel, data, len);
2748         }
2749
2750         return len;
2751 }
2752
2753 static void channel_accept(struct utcp_connection *utcp_connection, uint16_t port) {
2754         node_t *n = utcp_connection->utcp->priv;
2755
2756         if(!n) {
2757                 abort();
2758         }
2759
2760         meshlink_handle_t *mesh = n->mesh;
2761
2762         if(!mesh->channel_accept_cb) {
2763                 return;
2764         }
2765
2766         meshlink_channel_t *channel = xzalloc(sizeof(*channel));
2767         channel->node = n;
2768         channel->c = utcp_connection;
2769
2770         if(mesh->channel_accept_cb(mesh, channel, port, NULL, 0)) {
2771                 utcp_accept(utcp_connection, channel_recv, channel);
2772         } else {
2773                 free(channel);
2774         }
2775 }
2776
2777 static ssize_t channel_send(struct utcp *utcp, const void *data, size_t len) {
2778         node_t *n = utcp->priv;
2779
2780         if(n->status.destroyed) {
2781                 return -1;
2782         }
2783
2784         meshlink_handle_t *mesh = n->mesh;
2785         return meshlink_send(mesh, (meshlink_node_t *)n, data, len) ? (ssize_t)len : -1;
2786 }
2787
2788 void meshlink_set_channel_receive_cb(meshlink_handle_t *mesh, meshlink_channel_t *channel, meshlink_channel_receive_cb_t cb) {
2789         if(!mesh || !channel) {
2790                 meshlink_errno = MESHLINK_EINVAL;
2791                 return;
2792         }
2793
2794         channel->receive_cb = cb;
2795 }
2796
2797 static void channel_receive(meshlink_handle_t *mesh, meshlink_node_t *source, const void *data, size_t len) {
2798         (void)mesh;
2799         node_t *n = (node_t *)source;
2800
2801         if(!n->utcp) {
2802                 abort();
2803         }
2804
2805         utcp_recv(n->utcp, data, len);
2806 }
2807
2808 static void channel_poll(struct utcp_connection *connection, size_t len) {
2809         meshlink_channel_t *channel = connection->priv;
2810
2811         if(!channel) {
2812                 abort();
2813         }
2814
2815         node_t *n = channel->node;
2816         meshlink_handle_t *mesh = n->mesh;
2817
2818         if(channel->poll_cb) {
2819                 channel->poll_cb(mesh, channel, len);
2820         }
2821 }
2822
2823 void meshlink_set_channel_poll_cb(meshlink_handle_t *mesh, meshlink_channel_t *channel, meshlink_channel_poll_cb_t cb) {
2824         (void)mesh;
2825         channel->poll_cb = cb;
2826         utcp_set_poll_cb(channel->c, cb ? channel_poll : NULL);
2827 }
2828
2829 void meshlink_set_channel_accept_cb(meshlink_handle_t *mesh, meshlink_channel_accept_cb_t cb) {
2830         if(!mesh) {
2831                 meshlink_errno = MESHLINK_EINVAL;
2832                 return;
2833         }
2834
2835         pthread_mutex_lock(&mesh->mesh_mutex);
2836         mesh->channel_accept_cb = cb;
2837         mesh->receive_cb = channel_receive;
2838
2839         for splay_each(node_t, n, mesh->nodes) {
2840                 if(!n->utcp && n != mesh->self) {
2841                         n->utcp = utcp_init(channel_accept, channel_pre_accept, channel_send, n);
2842                 }
2843         }
2844
2845         pthread_mutex_unlock(&mesh->mesh_mutex);
2846 }
2847
2848 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) {
2849         if(data || len) {
2850                 abort();        // TODO: handle non-NULL data
2851         }
2852
2853         if(!mesh || !node) {
2854                 meshlink_errno = MESHLINK_EINVAL;
2855                 return NULL;
2856         }
2857
2858         node_t *n = (node_t *)node;
2859
2860         if(!n->utcp) {
2861                 n->utcp = utcp_init(channel_accept, channel_pre_accept, channel_send, n);
2862                 mesh->receive_cb = channel_receive;
2863
2864                 if(!n->utcp) {
2865                         meshlink_errno = errno == ENOMEM ? MESHLINK_ENOMEM : MESHLINK_EINTERNAL;
2866                         return NULL;
2867                 }
2868         }
2869
2870         if(n->status.blacklisted) {
2871                 logger(mesh, MESHLINK_ERROR, "Cannot open a channel with blacklisted node\n");
2872                 return NULL;
2873         }
2874
2875         meshlink_channel_t *channel = xzalloc(sizeof(*channel));
2876         channel->node = n;
2877         channel->receive_cb = cb;
2878         channel->c = utcp_connect_ex(n->utcp, port, channel_recv, channel, flags);
2879
2880         if(!channel->c) {
2881                 meshlink_errno = errno == ENOMEM ? MESHLINK_ENOMEM : MESHLINK_EINTERNAL;
2882                 free(channel);
2883                 return NULL;
2884         }
2885
2886         return channel;
2887 }
2888
2889 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) {
2890         return meshlink_channel_open_ex(mesh, node, port, cb, data, len, MESHLINK_CHANNEL_TCP);
2891 }
2892
2893 void meshlink_channel_shutdown(meshlink_handle_t *mesh, meshlink_channel_t *channel, int direction) {
2894         if(!mesh || !channel) {
2895                 meshlink_errno = MESHLINK_EINVAL;
2896                 return;
2897         }
2898
2899         utcp_shutdown(channel->c, direction);
2900 }
2901
2902 void meshlink_channel_close(meshlink_handle_t *mesh, meshlink_channel_t *channel) {
2903         if(!mesh || !channel) {
2904                 meshlink_errno = MESHLINK_EINVAL;
2905                 return;
2906         }
2907
2908         utcp_close(channel->c);
2909         free(channel);
2910 }
2911
2912 ssize_t meshlink_channel_send(meshlink_handle_t *mesh, meshlink_channel_t *channel, const void *data, size_t len) {
2913         if(!mesh || !channel) {
2914                 meshlink_errno = MESHLINK_EINVAL;
2915                 return -1;
2916         }
2917
2918         if(!len) {
2919                 return 0;
2920         }
2921
2922         if(!data) {
2923                 meshlink_errno = MESHLINK_EINVAL;
2924                 return -1;
2925         }
2926
2927         // TODO: more finegrained locking.
2928         // Ideally we want to put the data into the UTCP connection's send buffer.
2929         // Then, preferably only if there is room in the receiver window,
2930         // kick the meshlink thread to go send packets.
2931
2932         pthread_mutex_lock(&mesh->mesh_mutex);
2933         ssize_t retval = utcp_send(channel->c, data, len);
2934         pthread_mutex_unlock(&mesh->mesh_mutex);
2935
2936         if(retval < 0) {
2937                 meshlink_errno = MESHLINK_ENETWORK;
2938         }
2939
2940         return retval;
2941 }
2942
2943 uint32_t meshlink_channel_get_flags(meshlink_handle_t *mesh, meshlink_channel_t *channel) {
2944         if(!mesh || !channel) {
2945                 meshlink_errno = MESHLINK_EINVAL;
2946                 return -1;
2947         }
2948
2949         return channel->c->flags;
2950 }
2951
2952 size_t meshlink_channel_get_sendq(meshlink_handle_t *mesh, meshlink_channel_t *channel) {
2953         if(!mesh || !channel) {
2954                 meshlink_errno = MESHLINK_EINVAL;
2955                 return -1;
2956         }
2957
2958         return utcp_get_sendq(channel->c);
2959 }
2960
2961 size_t meshlink_channel_get_recvq(meshlink_handle_t *mesh, meshlink_channel_t *channel) {
2962         if(!mesh || !channel) {
2963                 meshlink_errno = MESHLINK_EINVAL;
2964                 return -1;
2965         }
2966
2967         return utcp_get_recvq(channel->c);
2968 }
2969
2970 void update_node_status(meshlink_handle_t *mesh, node_t *n) {
2971         if(n->status.reachable && mesh->channel_accept_cb && !n->utcp) {
2972                 n->utcp = utcp_init(channel_accept, channel_pre_accept, channel_send, n);
2973         }
2974
2975         if(mesh->node_status_cb) {
2976                 mesh->node_status_cb(mesh, (meshlink_node_t *)n, n->status.reachable && !n->status.blacklisted);
2977         }
2978 }
2979
2980 void handle_duplicate_node(meshlink_handle_t *mesh, node_t *n) {
2981         if(!mesh->node_duplicate_cb || n->status.duplicate) {
2982                 return;
2983         }
2984
2985         n->status.duplicate = true;
2986         mesh->node_duplicate_cb(mesh, (meshlink_node_t *)n);
2987 }
2988
2989 void meshlink_enable_discovery(meshlink_handle_t *mesh, bool enable) {
2990 #if HAVE_CATTA
2991
2992         if(!mesh) {
2993                 meshlink_errno = MESHLINK_EINVAL;
2994                 return;
2995         }
2996
2997         pthread_mutex_lock(&mesh->mesh_mutex);
2998
2999         if(mesh->discovery == enable) {
3000                 goto end;
3001         }
3002
3003         if(mesh->threadstarted) {
3004                 if(enable) {
3005                         discovery_start(mesh);
3006                 } else {
3007                         discovery_stop(mesh);
3008                 }
3009         }
3010
3011         mesh->discovery = enable;
3012
3013 end:
3014         pthread_mutex_unlock(&mesh->mesh_mutex);
3015 #else
3016         (void)mesh;
3017         (void)enable;
3018         meshlink_errno = MESHLINK_ENOTSUP;
3019 #endif
3020 }
3021
3022 static void __attribute__((constructor)) meshlink_init(void) {
3023         crypto_init();
3024         unsigned int seed;
3025         randomize(&seed, sizeof(seed));
3026         srand(seed);
3027 }
3028
3029 static void __attribute__((destructor)) meshlink_exit(void) {
3030         crypto_exit();
3031 }
3032
3033 /// Device class traits
3034 dev_class_traits_t dev_class_traits[_DEV_CLASS_MAX + 1] = {
3035         { .min_connects = 3, .max_connects = 10000, .edge_weight = 1 }, // DEV_CLASS_BACKBONE
3036         { .min_connects = 3, .max_connects = 100, .edge_weight = 3 },   // DEV_CLASS_STATIONARY
3037         { .min_connects = 3, .max_connects = 3, .edge_weight = 6 },             // DEV_CLASS_PORTABLE
3038         { .min_connects = 1, .max_connects = 1, .edge_weight = 9 },             // DEV_CLASS_UNKNOWN
3039 };