]> git.meshlink.io Git - meshlink/blob - src/protocol_auth.c
Merge branch 'discovery' of chicago.everbase.net:meshlink/meshlink into everbase
[meshlink] / src / protocol_auth.c
1 /*
2     protocol_auth.c -- handle the meta-protocol, authentication
3     Copyright (C) 2014 Guus Sliepen <guus@meshlink.io>
4
5     This program is free software; you can redistribute it and/or modify
6     it under the terms of the GNU General Public License as published by
7     the Free Software Foundation; either version 2 of the License, or
8     (at your option) any later version.
9
10     This program is distributed in the hope that it will be useful,
11     but WITHOUT ANY WARRANTY; without even the implied warranty of
12     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13     GNU General Public License for more details.
14
15     You should have received a copy of the GNU General Public License along
16     with this program; if not, write to the Free Software Foundation, Inc.,
17     51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
18 */
19
20 #include "system.h"
21
22 #include "conf.h"
23 #include "connection.h"
24 #include "ecdsa.h"
25 #include "edge.h"
26 #include "graph.h"
27 #include "logger.h"
28 #include "meshlink_internal.h"
29 #include "meta.h"
30 #include "net.h"
31 #include "netutl.h"
32 #include "node.h"
33 #include "prf.h"
34 #include "protocol.h"
35 #include "sptps.h"
36 #include "utils.h"
37 #include "xalloc.h"
38 #include "ed25519/sha512.h"
39
40 static bool send_proxyrequest(meshlink_handle_t *mesh, connection_t *c) {
41         switch(mesh->proxytype) {
42                 case PROXY_HTTP: {
43                         char *host;
44                         char *port;
45
46                         sockaddr2str(&c->address, &host, &port);
47                         send_request(mesh, c, "CONNECT %s:%s HTTP/1.1\r\n\r", host, port);
48                         free(host);
49                         free(port);
50                         return true;
51                 }
52                 case PROXY_SOCKS4: {
53                         if(c->address.sa.sa_family != AF_INET) {
54                                 logger(mesh, MESHLINK_ERROR, "Cannot connect to an IPv6 host through a SOCKS 4 proxy!");
55                                 return false;
56                         }
57                         char s4req[9 + (mesh->proxyuser ? strlen(mesh->proxyuser) : 0)];
58                         s4req[0] = 4;
59                         s4req[1] = 1;
60                         memcpy(s4req + 2, &c->address.in.sin_port, 2);
61                         memcpy(s4req + 4, &c->address.in.sin_addr, 4);
62                         if(mesh->proxyuser)
63                                 memcpy(s4req + 8, mesh->proxyuser, strlen(mesh->proxyuser));
64                         s4req[sizeof s4req - 1] = 0;
65                         c->tcplen = 8;
66                         return send_meta(mesh, c, s4req, sizeof s4req);
67                 }
68                 case PROXY_SOCKS5: {
69                         int len = 3 + 6 + (c->address.sa.sa_family == AF_INET ? 4 : 16);
70                         c->tcplen = 2;
71                         if(mesh->proxypass)
72                                 len += 3 + strlen(mesh->proxyuser) + strlen(mesh->proxypass);
73                         char s5req[len];
74                         int i = 0;
75                         s5req[i++] = 5;
76                         s5req[i++] = 1;
77                         if(mesh->proxypass) {
78                                 s5req[i++] = 2;
79                                 s5req[i++] = 1;
80                                 s5req[i++] = strlen(mesh->proxyuser);
81                                 memcpy(s5req + i, mesh->proxyuser, strlen(mesh->proxyuser));
82                                 i += strlen(mesh->proxyuser);
83                                 s5req[i++] = strlen(mesh->proxypass);
84                                 memcpy(s5req + i, mesh->proxypass, strlen(mesh->proxypass));
85                                 i += strlen(mesh->proxypass);
86                                 c->tcplen += 2;
87                         } else {
88                                 s5req[i++] = 0;
89                         }
90                         s5req[i++] = 5;
91                         s5req[i++] = 1;
92                         s5req[i++] = 0;
93                         if(c->address.sa.sa_family == AF_INET) {
94                                 s5req[i++] = 1;
95                                 memcpy(s5req + i, &c->address.in.sin_addr, 4);
96                                 i += 4;
97                                 memcpy(s5req + i, &c->address.in.sin_port, 2);
98                                 i += 2;
99                                 c->tcplen += 10;
100                         } else if(c->address.sa.sa_family == AF_INET6) {
101                                 s5req[i++] = 3;
102                                 memcpy(s5req + i, &c->address.in6.sin6_addr, 16);
103                                 i += 16;
104                                 memcpy(s5req + i, &c->address.in6.sin6_port, 2);
105                                 i += 2;
106                                 c->tcplen += 22;
107                         } else {
108                                 logger(mesh, MESHLINK_ERROR, "Address family %hx not supported for SOCKS 5 proxies!", c->address.sa.sa_family);
109                                 return false;
110                         }
111                         if(i > len)
112                                 abort();
113                         return send_meta(mesh, c, s5req, sizeof s5req);
114                 }
115                 case PROXY_SOCKS4A:
116                         logger(mesh, MESHLINK_ERROR, "Proxy type not implemented yet");
117                         return false;
118                 case PROXY_EXEC:
119                         return true;
120                 default:
121                         logger(mesh, MESHLINK_ERROR, "Unknown proxy type");
122                         return false;
123         }
124 }
125
126 bool send_id(meshlink_handle_t *mesh, connection_t *c) {
127         gettimeofday(&c->start, NULL);
128
129         int minor = mesh->self->connection->protocol_minor;
130
131         if(mesh->proxytype && c->outgoing)
132                 if(!send_proxyrequest(mesh, c))
133                         return false;
134
135         return send_request(mesh, c, "%d %s %d.%d", ID, mesh->self->connection->name, mesh->self->connection->protocol_major, minor);
136 }
137
138 static bool finalize_invitation(meshlink_handle_t *mesh, connection_t *c, const void *data, uint16_t len) {
139         if(strchr(data, '\n')) {
140                 logger(mesh, MESHLINK_ERROR, "Received invalid key from invited node %s (%s)!\n", c->name, c->hostname);
141                 return false;
142         }
143
144         // Create a new host config file
145         char filename[PATH_MAX];
146         snprintf(filename, sizeof filename, "%s" SLASH "hosts" SLASH "%s", mesh->confbase, c->name);
147         if(!access(filename, F_OK)) {
148                 logger(mesh, MESHLINK_ERROR, "Host config file for %s (%s) already exists!\n", c->name, c->hostname);
149                 return false;
150         }
151
152         FILE *f = fopen(filename, "w");
153         if(!f) {
154                 logger(mesh, MESHLINK_ERROR, "Error trying to create %s: %s\n", filename, strerror(errno));
155                 return false;
156         }
157
158         fprintf(f, "ECDSAPublicKey = %s\n", (const char *)data);
159         fclose(f);
160
161         logger(mesh, MESHLINK_INFO, "Key succesfully received from %s (%s)", c->name, c->hostname);
162
163         //TODO: callback to application to inform of an accepted invitation
164
165         sptps_send_record(&c->sptps, 2, data, 0);
166
167         load_all_nodes(mesh);
168
169         return true;
170 }
171
172 static bool receive_invitation_sptps(void *handle, uint8_t type, const void *data, uint16_t len) {
173         connection_t *c = handle;
174         meshlink_handle_t *mesh = c->mesh;
175
176         if(type == 128)
177                 return true;
178
179         if(type == 1 && c->status.invitation_used)
180                 return finalize_invitation(mesh, c, data, len);
181
182         if(type != 0 || len != 18 || c->status.invitation_used)
183                 return false;
184
185         // Recover the filename from the cookie and the key
186         char *fingerprint = ecdsa_get_base64_public_key(mesh->invitation_key);
187         char hash[64];
188         char hashbuf[18 + strlen(fingerprint)];
189         char cookie[25];
190         memcpy(hashbuf, data, 18);
191         memcpy(hashbuf + 18, fingerprint, sizeof hashbuf - 18);
192         sha512(hashbuf, sizeof hashbuf, hash);
193         b64encode_urlsafe(hash, cookie, 18);
194         free(fingerprint);
195
196         char filename[PATH_MAX], usedname[PATH_MAX];
197         snprintf(filename, sizeof filename, "%s" SLASH "invitations" SLASH "%s", mesh->confbase, cookie);
198         snprintf(usedname, sizeof usedname, "%s" SLASH "invitations" SLASH "%s.used", mesh->confbase, cookie);
199
200         // Atomically rename the invitation file
201         if(rename(filename, usedname)) {
202                 if(errno == ENOENT)
203                         logger(mesh, MESHLINK_ERROR, "Peer %s tried to use non-existing invitation %s\n", c->hostname, cookie);
204                 else
205                         logger(mesh, MESHLINK_ERROR, "Error trying to rename invitation %s\n", cookie);
206                 return false;
207         }
208
209         // Open the renamed file
210         FILE *f = fopen(usedname, "r");
211         if(!f) {
212                 logger(mesh, MESHLINK_ERROR, "Error trying to open invitation %s\n", cookie);
213                 return false;
214         }
215
216         // Read the new node's Name from the file
217         char buf[1024];
218         fgets(buf, sizeof buf, f);
219         if(*buf)
220                 buf[strlen(buf) - 1] = 0;
221
222         len = strcspn(buf, " \t=");
223         char *name = buf + len;
224         name += strspn(name, " \t");
225         if(*name == '=') {
226                 name++;
227                 name += strspn(name, " \t");
228         }
229         buf[len] = 0;
230
231         if(!*buf || !*name || strcasecmp(buf, "Name") || !check_id(name)) {
232                 logger(mesh, MESHLINK_ERROR, "Invalid invitation file %s\n", cookie);
233                 fclose(f);
234                 return false;
235         }
236
237         free(c->name);
238         c->name = xstrdup(name);
239
240         // Send the node the contents of the invitation file
241         rewind(f);
242         size_t result;
243         while((result = fread(buf, 1, sizeof buf, f)))
244                 sptps_send_record(&c->sptps, 0, buf, result);
245         sptps_send_record(&c->sptps, 1, buf, 0);
246         fclose(f);
247         unlink(usedname);
248
249         c->status.invitation_used = true;
250
251         logger(mesh, MESHLINK_INFO, "Invitation %s succesfully sent to %s (%s)", cookie, c->name, c->hostname);
252         return true;
253 }
254
255 bool id_h(meshlink_handle_t *mesh, connection_t *c, const char *request) {
256         char name[MAX_STRING_SIZE];
257
258         if(sscanf(request, "%*d " MAX_STRING " %d.%d", name, &c->protocol_major, &c->protocol_minor) < 2) {
259                 logger(mesh, MESHLINK_ERROR, "Got bad %s from %s (%s)", "ID", c->name,
260                            c->hostname);
261                 return false;
262         }
263
264         /* Check if this is an invitation  */
265
266         if(name[0] == '?') {
267                 if(!mesh->invitation_key) {
268                         logger(mesh, MESHLINK_ERROR, "Got invitation from %s but we don't have an invitation key", c->hostname);
269                         return false;
270                 }
271
272                 c->ecdsa = ecdsa_set_base64_public_key(name + 1);
273                 if(!c->ecdsa) {
274                         logger(mesh, MESHLINK_ERROR, "Got bad invitation from %s", c->hostname);
275                         return false;
276                 }
277
278                 c->status.invitation = true;
279                 char *mykey = ecdsa_get_base64_public_key(mesh->invitation_key);
280                 if(!mykey)
281                         return false;
282                 if(!send_request(mesh, c, "%d %s", ACK, mykey))
283                         return false;
284                 free(mykey);
285
286                 c->protocol_minor = 2;
287                 c->allow_request = 1;
288
289                 return sptps_start(&c->sptps, c, false, false, mesh->invitation_key, c->ecdsa, "meshlink invitation", 15, send_meta_sptps, receive_invitation_sptps);
290         }
291
292         /* Check if identity is a valid name */
293
294         if(!check_id(name)) {
295                 logger(mesh, MESHLINK_ERROR, "Got bad %s from %s (%s): %s", "ID", c->name,
296                            c->hostname, "invalid name");
297                 return false;
298         }
299
300         /* If this is an outgoing connection, make sure we are connected to the right host */
301
302         if(c->outgoing) {
303                 if(strcmp(c->name, name)) {
304                         logger(mesh, MESHLINK_ERROR, "Peer %s is %s instead of %s", c->hostname, name,
305                                    c->name);
306                         return false;
307                 }
308         } else {
309                 if(c->name)
310                         free(c->name);
311                 c->name = xstrdup(name);
312         }
313
314         /* Check if version matches */
315
316         if(c->protocol_major != mesh->self->connection->protocol_major) {
317                 logger(mesh, MESHLINK_ERROR, "Peer %s (%s) uses incompatible version %d.%d",
318                         c->name, c->hostname, c->protocol_major, c->protocol_minor);
319                 return false;
320         }
321
322         if(!c->config_tree) {
323                 init_configuration(&c->config_tree);
324
325                 if(!read_host_config(mesh, c->config_tree, c->name)) {
326                         logger(mesh, MESHLINK_ERROR, "Peer %s had unknown identity (%s)", c->hostname, c->name);
327                         return false;
328                 }
329
330                 read_ecdsa_public_key(mesh, c);
331         } else {
332                 if(c->protocol_minor && !ecdsa_active(c->ecdsa))
333                         c->protocol_minor = 1;
334         }
335
336         /* Forbid version rollback for nodes whose ECDSA key we know */
337
338         if(ecdsa_active(c->ecdsa) && c->protocol_minor < 2) {
339                 logger(mesh, MESHLINK_ERROR, "Peer %s (%s) tries to roll back protocol version to %d.%d",
340                         c->name, c->hostname, c->protocol_major, c->protocol_minor);
341                 return false;
342         }
343
344         c->allow_request = ACK;
345         char label[25 + strlen(mesh->self->name) + strlen(c->name)];
346
347         if(c->outgoing)
348                 snprintf(label, sizeof label, "meshlink TCP key expansion %s %s", mesh->self->name, c->name);
349         else
350                 snprintf(label, sizeof label, "meshlink TCP key expansion %s %s", c->name, mesh->self->name);
351
352         return sptps_start(&c->sptps, c, c->outgoing, false, mesh->self->connection->ecdsa, c->ecdsa, label, sizeof label, send_meta_sptps, receive_meta_sptps);
353 }
354
355 bool send_ack(meshlink_handle_t *mesh, connection_t *c) {
356         /* ACK message contains rest of the information the other end needs
357            to create node_t and edge_t structures. */
358
359         struct timeval now;
360
361         /* Estimate weight */
362
363         gettimeofday(&now, NULL);
364         c->estimated_weight = (now.tv_sec - c->start.tv_sec) * 1000 + (now.tv_usec - c->start.tv_usec) / 1000;
365
366         /* Check some options */
367
368         if(mesh->self->options & OPTION_PMTU_DISCOVERY)
369                 c->options |= OPTION_PMTU_DISCOVERY;
370
371         return send_request(mesh, c, "%d %s %d %x", ACK, mesh->myport, c->estimated_weight, (c->options & 0xffffff) | (PROT_MINOR << 24));
372 }
373
374 static void send_everything(meshlink_handle_t *mesh, connection_t *c) {
375         /* Send all known subnets and edges */
376
377         for splay_each(node_t, n, mesh->nodes) {
378                 for splay_each(edge_t, e, n->edge_tree)
379                         send_add_edge(mesh, c, e);
380         }
381 }
382
383 bool ack_h(meshlink_handle_t *mesh, connection_t *c, const char *request) {
384         char hisport[MAX_STRING_SIZE];
385         char *hisaddress;
386         int weight;
387         uint32_t options;
388         node_t *n;
389
390         if(sscanf(request, "%*d " MAX_STRING " %d %x", hisport, &weight, &options) != 3) {
391                 logger(mesh, MESHLINK_ERROR, "Got bad %s from %s (%s)", "ACK", c->name,
392                            c->hostname);
393                 return false;
394         }
395
396         /* Check if we already have a node_t for him */
397
398         n = lookup_node(mesh, c->name);
399
400         if(!n) {
401                 n = new_node();
402                 n->name = xstrdup(c->name);
403                 node_add(mesh, n);
404         } else {
405                 if(n->connection) {
406                         /* Oh dear, we already have a connection to this node. */
407                         logger(mesh, MESHLINK_DEBUG, "Established a second connection with %s (%s), closing old connection", n->connection->name, n->connection->hostname);
408
409                         if(n->connection->outgoing) {
410                                 if(c->outgoing)
411                                         logger(mesh, MESHLINK_WARNING, "Two outgoing connections to the same node!");
412                                 else
413                                         c->outgoing = n->connection->outgoing;
414
415                                 n->connection->outgoing = NULL;
416                         }
417
418                         terminate_connection(mesh, n->connection, false);
419                         /* Run graph algorithm to keep things in sync */
420                         graph(mesh);
421                 }
422         }
423
424         n->connection = c;
425         c->node = n;
426         if(!(c->options & options & OPTION_PMTU_DISCOVERY)) {
427                 c->options &= ~OPTION_PMTU_DISCOVERY;
428                 options &= ~OPTION_PMTU_DISCOVERY;
429         }
430         c->options |= options;
431
432         /* Activate this connection */
433
434         c->allow_request = ALL;
435         c->status.active = true;
436
437         logger(mesh, MESHLINK_INFO, "Connection with %s (%s) activated", c->name,
438                            c->hostname);
439
440         /* Send him everything we know */
441
442         send_everything(mesh, c);
443
444         /* Create an edge_t for this connection */
445
446         c->edge = new_edge();
447         c->edge->from = mesh->self;
448         c->edge->to = n;
449         sockaddr2str(&c->address, &hisaddress, NULL);
450         c->edge->address = str2sockaddr(hisaddress, hisport);
451         free(hisaddress);
452         c->edge->weight = (weight + c->estimated_weight) / 2;
453         c->edge->connection = c;
454         c->edge->options = c->options;
455
456         edge_add(mesh, c->edge);
457
458         /* Notify everyone of the new edge */
459
460         send_add_edge(mesh, mesh->everyone, c->edge);
461
462         /* Run MST and SSSP algorithms */
463
464         graph(mesh);
465
466         return true;
467 }