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