]> git.meshlink.io Git - meshlink/blobdiff - src/meshlink.c
Make sure the SPTSP connection get set up properly during a join.
[meshlink] / src / meshlink.c
index 78c75abfe3aa897cfb3b7a0bba0d539fe3f39b4a..718b07a183d2d0449cdadfbef6c8881c9956d46b 100644 (file)
@@ -32,9 +32,11 @@ typedef struct {
 #include "crypto.h"
 #include "ecdsagen.h"
 #include "meshlink_internal.h"
+#include "netutl.h"
 #include "node.h"
 #include "protocol.h"
 #include "route.h"
+#include "utils.h"
 #include "xalloc.h"
 #include "ed25519/sha512.h"
 
@@ -117,6 +119,14 @@ static bool fcopy(FILE *out, const char *filename) {
        fclose(in);
        return true;
 }
+
+static int rstrip(char *value) {
+       int len = strlen(value);
+       while(len && strchr("\t\r\n ", value[len - 1]))
+               value[--len] = 0;
+       return len;
+}
+
 static void scan_for_hostname(const char *filename, char **hostname, char **port) {
        char line[4096];
        if(!filename || (*hostname && *port))
@@ -164,8 +174,9 @@ static char *get_my_hostname(meshlink_handle_t* mesh) {
        char *port = NULL;
        char *hostport = NULL;
        char *name = mesh->self->name;
-       char filename[PATH_MAX];
+       char filename[PATH_MAX] = "";
        char line[4096];
+       FILE *f;
 
        // Use first Address statement in own host config file
        snprintf(filename,PATH_MAX, "%s" SLASH "hosts" SLASH "%s", mesh->confbase, name);
@@ -261,14 +272,12 @@ again:
        hostname = xstrdup(line);
 
 save:
-       if(filename) {
-               FILE *f = fopen(filename, "a");
-               if(f) {
-                       fprintf(f, "\nAddress = %s\n", hostname);
-                       fclose(f);
-               } else {
-                       fprintf(stderr, "Could not append Address to %s: %s\n", filename, strerror(errno));
-               }
+       f = fopen(filename, "a");
+       if(f) {
+               fprintf(f, "\nAddress = %s\n", hostname);
+               fclose(f);
+       } else {
+               fprintf(stderr, "Could not append Address to %s: %s\n", filename, strerror(errno));
        }
 
 done:
@@ -346,6 +355,63 @@ static FILE *fopenmask(const char *filename, const char *mode, mode_t perms) {
        return f;
 }
 
+static bool try_bind(int port) {
+       struct addrinfo *ai = NULL;
+       struct addrinfo hint = {
+               .ai_flags = AI_PASSIVE,
+               .ai_family = AF_UNSPEC,
+               .ai_socktype = SOCK_STREAM,
+               .ai_protocol = IPPROTO_TCP,
+       };
+
+       char portstr[16];
+       snprintf(portstr, sizeof portstr, "%d", port);
+
+       if(getaddrinfo(NULL, portstr, &hint, &ai) || !ai)
+               return false;
+
+       while(ai) {
+               int fd = socket(ai->ai_family, SOCK_STREAM, IPPROTO_TCP);
+               if(!fd)
+                       return false;
+               int result = bind(fd, ai->ai_addr, ai->ai_addrlen);
+               closesocket(fd);
+               if(result)
+                       return false;
+               ai = ai->ai_next;
+       }
+
+       return true;
+}
+
+static int check_port(meshlink_handle_t *mesh) {
+       if(try_bind(655))
+               return 655;
+
+       fprintf(stderr, "Warning: could not bind to port 655.\n");
+
+       for(int i = 0; i < 100; i++) {
+               int port = 0x1000 + (rand() & 0x7fff);
+               if(try_bind(port)) {
+                       char filename[PATH_MAX];
+                       snprintf(filename, sizeof filename, "%s" SLASH "hosts" SLASH "%s", mesh->confbase, mesh->name);
+                       FILE *f = fopen(filename, "a");
+                       if(!f) {
+                               fprintf(stderr, "Please change MeshLink's Port manually.\n");
+                               return 0;
+                       }
+
+                       fprintf(f, "Port = %d\n", port);
+                       fclose(f);
+                       fprintf(stderr, "MeshLink will instead listen on port %d.\n", port);
+                       return port;
+               }
+       }
+
+       fprintf(stderr, "Please change MeshLink's Port manually.\n");
+       return 0;
+}
+
 static bool finalize_join(meshlink_handle_t *mesh) {
        char *name = xstrdup(get_value(mesh->data, "Name"));
        if(!name) {
@@ -507,7 +573,7 @@ static bool finalize_join(meshlink_handle_t *mesh) {
 
        ecdsa_free(key);
 
-       check_port(name);
+       check_port(mesh);
 
        fprintf(stderr, "Configuration stored in: %s\n", mesh->confbase);
 
@@ -612,13 +678,6 @@ static bool sendline(int fd, char *format, ...) {
 
        return true;
 }
-int rstrip(char *value) {
-       int len = strlen(value);
-       while(len && strchr("\t\r\n ", value[len - 1]))
-               value[--len] = 0;
-       return len;
-}
-
 
 static const char *errstr[] = {
        [MESHLINK_OK] = "No error",
@@ -679,63 +738,6 @@ static bool ecdsa_keygen(meshlink_handle_t *mesh) {
        return true;
 }
 
-static bool try_bind(int port) {
-       struct addrinfo *ai = NULL;
-       struct addrinfo hint = {
-               .ai_flags = AI_PASSIVE,
-               .ai_family = AF_UNSPEC,
-               .ai_socktype = SOCK_STREAM,
-               .ai_protocol = IPPROTO_TCP,
-       };
-
-       char portstr[16];
-       snprintf(portstr, sizeof portstr, "%d", port);
-
-       if(getaddrinfo(NULL, portstr, &hint, &ai) || !ai)
-               return false;
-
-       while(ai) {
-               int fd = socket(ai->ai_family, SOCK_STREAM, IPPROTO_TCP);
-               if(!fd)
-                       return false;
-               int result = bind(fd, ai->ai_addr, ai->ai_addrlen);
-               closesocket(fd);
-               if(result)
-                       return false;
-               ai = ai->ai_next;
-       }
-
-       return true;
-}
-
-int check_port(meshlink_handle_t *mesh) {
-       if(try_bind(655))
-               return 655;
-
-       fprintf(stderr, "Warning: could not bind to port 655.\n");
-
-       for(int i = 0; i < 100; i++) {
-               int port = 0x1000 + (rand() & 0x7fff);
-               if(try_bind(port)) {
-                       char filename[PATH_MAX];
-                       snprintf(filename, sizeof filename, "%s" SLASH "hosts" SLASH "%s", mesh->confbase, mesh->name);
-                       FILE *f = fopen(filename, "a");
-                       if(!f) {
-                               fprintf(stderr, "Please change MeshLink's Port manually.\n");
-                               return 0;
-                       }
-
-                       fprintf(f, "Port = %d\n", port);
-                       fclose(f);
-                       fprintf(stderr, "MeshLink will instead listen on port %d.\n", port);
-                       return port;
-               }
-       }
-
-       fprintf(stderr, "Please change MeshLink's Port manually.\n");
-       return 0;
-}
-
 static bool meshlink_setup(meshlink_handle_t *mesh) {
 
        if(mkdir(mesh->confbase, 0777) && errno != EEXIST) {
@@ -911,8 +913,7 @@ bool meshlink_send(meshlink_handle_t *mesh, meshlink_node_t *destination, const
 }
 
 meshlink_node_t *meshlink_get_node(meshlink_handle_t *mesh, const char *name) {
-       return (meshlink_node_t *)lookup_node(mesh, name);
-       return NULL;
+       return (meshlink_node_t *)lookup_node(mesh, (char *)name); // TODO: make lookup_node() use const
 }
 
 size_t meshlink_get_all_nodes(meshlink_handle_t *mesh, meshlink_node_t **nodes, size_t nmemb) {
@@ -927,46 +928,10 @@ bool meshlink_verify(meshlink_handle_t *mesh, meshlink_node_t *source, const cha
        return false;
 }
 
-char *meshlink_invite(meshlink_handle_t *mesh, const char *name) {
-       // Check validity of the new node's name
-       if(!check_id(name)) {
-               fprintf(stderr, "Invalid name for node.\n");
-               return NULL;
-       }
-
-       // Ensure no host configuration file with that name exists
-       char filename [PATH_MAX];
-       snprintf(filename,PATH_MAX, "%s" SLASH "hosts" SLASH "%s", mesh->confbase, name);
-       if(!access(filename, F_OK)) {
-               fprintf(stderr, "A host config file for %s already exists!\n", name);
-               return NULL;
-       }
-
-       // If a daemon is running, ensure no other nodes know about this name
-
-       //TODO: original tinc code connects to tincd cli and makes this check. How we want to implement this here ?
-       //bool found = false;
-       //if(connect_tincd(false)) {
-       //      sendline(fd, "%d %d", CONTROL, REQ_DUMP_NODES);
-
-       //      while(recvline(fd, line, sizeof line)) {
-       //              char node[4096];
-       //              int code, req;
-       //              if(sscanf(line, "%d %d %s", &code, &req, node) != 3)
-       //                      break;
-       //              if(!strcmp(node, name))
-       //                      found = true;
-       //      }
-
-       //      if(found) {
-       //              fprintf(stderr, "A node with name %s is already known!\n", name);
-       //              return 1;
-       //      }
-       //}
-
-       char hash[64];
+static bool refresh_invitation_key(meshlink_handle_t *mesh) {
+       char filename[PATH_MAX];
 
-       snprintf(filename,PATH_MAX, "%s" SLASH "invitations", mesh->confbase);
+       snprintf(filename, sizeof filename, "%s" SLASH "invitations", mesh->confbase);
        if(mkdir(filename, 0700) && errno != EEXIST) {
                fprintf(stderr, "Could not create directory %s: %s\n", filename, strerror(errno));
                return NULL;
@@ -991,7 +956,7 @@ char *meshlink_invite(meshlink_handle_t *mesh, const char *name) {
                struct stat st;
                snprintf(invname,PATH_MAX, "%s" SLASH "%s", filename, ent->d_name);
                if(!stat(invname, &st)) {
-                       if(deadline < st.st_mtime)
+                       if(mesh->invitation_key && deadline < st.st_mtime)
                                count++;
                        else
                                unlink(invname);
@@ -1004,17 +969,24 @@ char *meshlink_invite(meshlink_handle_t *mesh, const char *name) {
        if(errno) {
                fprintf(stderr, "Error while reading directory %s: %s\n", filename, strerror(errno));
                closedir(dir);
-               return NULL;
+               return false;
        }
 
        closedir(dir);
 
-       ecdsa_t *key;
-       snprintf(filename,PATH_MAX, "%s" SLASH "invitations" SLASH "ecdsa_key.priv", mesh->confbase);
+       snprintf(filename, sizeof filename, "%s" SLASH "invitations" SLASH "ecdsa_key.priv", mesh->confbase);
 
        // Remove the key if there are no outstanding invitations.
-       if(!count)
+       if(!count) {
                unlink(filename);
+               if(mesh->invitation_key) {
+                       ecdsa_free(mesh->invitation_key);
+                       mesh->invitation_key = NULL;
+               }
+       }
+
+       if(mesh->invitation_key)
+               return true;
 
        // Create a new key if necessary.
        FILE *f = fopen(filename, "r");
@@ -1024,8 +996,8 @@ char *meshlink_invite(meshlink_handle_t *mesh, const char *name) {
                        return NULL;
                }
 
-               key = ecdsa_generate();
-               if(!key) {
+               mesh->invitation_key = ecdsa_generate();
+               if(!mesh->invitation_key) {
                        fprintf(stderr, "Could not generate a new key!\n");
                        return NULL;
                }
@@ -1035,23 +1007,46 @@ char *meshlink_invite(meshlink_handle_t *mesh, const char *name) {
                        return NULL;
                }
                chmod(filename, 0600);
-               ecdsa_write_pem_private_key(key, f);
+               ecdsa_write_pem_private_key(mesh->invitation_key, f);
                fclose(f);
-               //TODO: handle this in meshlink
-               //if(connect_tincd(false))
-               //      sendline(fd, "%d %d", CONTROL, REQ_RELOAD);
        } else {
-               key = ecdsa_read_pem_private_key(f);
+               mesh->invitation_key = ecdsa_read_pem_private_key(f);
                fclose(f);
-               if(!key)
+               if(!mesh->invitation_key)
                        fprintf(stderr, "Could not read private key from %s\n", filename);
        }
 
-       if(!key)
+       return mesh->invitation_key;
+}
+
+char *meshlink_invite(meshlink_handle_t *mesh, const char *name) {
+       // Check validity of the new node's name
+       if(!check_id(name)) {
+               fprintf(stderr, "Invalid name for node.\n");
+               return NULL;
+       }
+
+       // Ensure no host configuration file with that name exists
+       char filename [PATH_MAX];
+       snprintf(filename,PATH_MAX, "%s" SLASH "hosts" SLASH "%s", mesh->confbase, name);
+       if(!access(filename, F_OK)) {
+               fprintf(stderr, "A host config file for %s already exists!\n", name);
+               return NULL;
+       }
+
+       // Ensure no other nodes know about this name
+       if(meshlink_get_node(mesh, name)) {
+               fprintf(stderr, "A node with name %s is already known!\n", name);
+               return NULL;
+       }
+
+       if(!refresh_invitation_key(mesh))
                return NULL;
 
+       char hash[64];
+
        // Create a hash of the key.
-       char *fingerprint = ecdsa_get_base64_public_key(key);
+       char *fingerprint = ecdsa_get_base64_public_key(mesh->invitation_key);
        sha512(fingerprint, strlen(fingerprint), hash);
        b64encode_urlsafe(hash, hash, 18);
 
@@ -1076,7 +1071,7 @@ char *meshlink_invite(meshlink_handle_t *mesh, const char *name) {
                fprintf(stderr, "Could not create invitation file %s: %s\n", filename, strerror(errno));
                return NULL;
        }
-       f = fdopen(ifd, "w");
+       FILE *f = fdopen(ifd, "w");
        if(!f)
                abort();
 
@@ -1121,27 +1116,13 @@ char *meshlink_invite(meshlink_handle_t *mesh, const char *name) {
 }
 
 bool meshlink_join(meshlink_handle_t *mesh, const char *invitation) {
+       //TODO: think of a better name for this variable, or of a different way to tokenize the invitation URL.
+       char copy[strlen(invitation) + 1];
+       strcpy(copy, invitation);
 
+       // Split the invitation URL into hostname, port, key hash and cookie.
 
-       // Make sure confbase exists and is accessible.
-       if(mkdir(mesh->confbase, 0777) && errno != EEXIST) {
-               fprintf(stderr, "Could not create directory %s: %s\n", mesh->confbase, strerror(errno));
-               return 1;
-       }
-
-       if(access(mesh->confbase, R_OK | W_OK | X_OK)) {
-               fprintf(stderr, "No permission to write in directory %s: %s\n", mesh->confbase, strerror(errno));
-               return 1;
-       }
-
-       // TODO: Either remove or reintroduce netname in meshlink
-       // If a netname or explicit configuration directory is specified, check for an existing meshlink.conf.
-       //if((mesh->netname || confbasegiven) && !access(meshlink_conf, F_OK)) {
-       //      fprintf(stderr, "Configuration file %s already exists!\n", meshlink_conf);
-       //      return 1;
-       //}
-
-       char *slash = strchr(invitation, '/');
+       char *slash = strchr(copy, '/');
        if(!slash)
                goto invalid;
 
@@ -1150,7 +1131,7 @@ bool meshlink_join(meshlink_handle_t *mesh, const char *invitation) {
        if(strlen(slash) != 48)
                goto invalid;
 
-       char *address = invitation;
+       char *address = copy;
        char *port = NULL;
        if(*address == '[') {
                address++;
@@ -1175,38 +1156,37 @@ bool meshlink_join(meshlink_handle_t *mesh, const char *invitation) {
        // Generate a throw-away key for the invitation.
        ecdsa_t *key = ecdsa_generate();
        if(!key)
-               return 1;
+               return false;
 
        char *b64key = ecdsa_get_base64_public_key(key);
 
        // Connect to the meshlink daemon mentioned in the URL.
        struct addrinfo *ai = str2addrinfo(address, port, SOCK_STREAM);
        if(!ai)
-               return 1;
+               return false;
 
        mesh->sock = socket(ai->ai_family, ai->ai_socktype, ai->ai_protocol);
        if(mesh->sock <= 0) {
                fprintf(stderr, "Could not open socket: %s\n", strerror(errno));
-               return 1;
+               return false;
        }
 
        if(connect(mesh->sock, ai->ai_addr, ai->ai_addrlen)) {
                fprintf(stderr, "Could not connect to %s port %s: %s\n", address, port, strerror(errno));
                closesocket(mesh->sock);
-               return 1;
+               return false;
        }
 
        fprintf(stderr, "Connected to %s port %s...\n", address, port);
 
        // Tell him we have an invitation, and give him our throw-away key.
-       int len = snprintf(invitation, sizeof invitation, "0 ?%s %d.%d\n", b64key, PROT_MAJOR, PROT_MINOR);
-       if(len <= 0 || len >= sizeof invitation)
-               abort();
+
+       mesh->blen = 0;
 
        if(!sendline(mesh->sock, "0 ?%s %d.%d", b64key, PROT_MAJOR, 1)) {
                fprintf(stderr, "Error sending request to %s port %s: %s\n", address, port, strerror(errno));
                closesocket(mesh->sock);
-               return 1;
+               return false;
        }
 
        char hisname[4096] = "";
@@ -1215,44 +1195,46 @@ bool meshlink_join(meshlink_handle_t *mesh, const char *invitation) {
        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) {
                fprintf(stderr, "Cannot read greeting from peer\n");
                closesocket(mesh->sock);
-               return 1;
+               return false;
        }
 
        // Check if the hash of the key he gave us matches the hash in the URL.
        char *fingerprint = mesh->line + 2;
        char hishash[64];
-       if(!sha512(fingerprint, strlen(fingerprint), hishash)) {
+       if(sha512(fingerprint, strlen(fingerprint), hishash)) {
                fprintf(stderr, "Could not create hash\n%s\n", mesh->line + 2);
-               return 1;
+               return false;
        }
        if(memcmp(hishash, mesh->hash, 18)) {
                fprintf(stderr, "Peer has an invalid key!\n%s\n", mesh->line + 2);
-               return 1;
+               return false;
 
        }
 
        ecdsa_t *hiskey = ecdsa_set_base64_public_key(fingerprint);
        if(!hiskey)
-               return 1;
+               return false;
 
        // Start an SPTPS session
        if(!sptps_start(&mesh->sptps, mesh, true, false, key, hiskey, "meshlink invitation", 15, invitation_send, invitation_receive))
-               return 1;
+               return false;
 
        // Feed rest of input buffer to SPTPS
        if(!sptps_receive_data(&mesh->sptps, mesh->buffer, mesh->blen))
-               return 1;
+               return false;
+
+       int len;
 
        while((len = recv(mesh->sock, mesh->line, sizeof mesh->line, 0))) {
                if(len < 0) {
                        if(errno == EINTR)
                                continue;
                        fprintf(stderr, "Error reading data from %s port %s: %s\n", address, port, strerror(errno));
-                       return 1;
+                       return false;
                }
 
                if(!sptps_receive_data(&mesh->sptps, mesh->line, len))
-                       return 1;
+                       return false;
        }
 
        sptps_stop(&mesh->sptps);