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