#include "conf.h"
#include "connection.h"
+#include "devtools.h"
#include "ecdsa.h"
#include "edge.h"
#include "graph.h"
return send_request(mesh, c, NULL, "%d %s %d.%d %s", ID, mesh->self->name, PROT_MAJOR, PROT_MINOR, mesh->appname);
}
-static bool finalize_invitation(meshlink_handle_t *mesh, connection_t *c, const void *data, uint16_t len) {
- if(len != 32) {
- logger(mesh, MESHLINK_ERROR, "Received invalid key from invited node %s!\n", c->name);
- return false;
- }
-
+static bool commit_invitation(meshlink_handle_t *mesh, connection_t *c, const void *data) {
// Create a new node
node_t *n = new_node();
n->name = xstrdup(c->name);
n->devclass = DEV_CLASS_UNKNOWN;
n->ecdsa = ecdsa_set_public_key(data);
n->submesh = c->submesh;
- n->status.dirty = true;
+
+ // Remember its current address
+ node_add_recent_address(mesh, n, &c->address);
+
+ if(!node_write_config(mesh, n) || !config_sync(mesh, "current")) {
+ logger(mesh, MESHLINK_ERROR, "Error writing configuration file for invited node %s!\n", c->name);
+ free_node(n);
+ return false;
+
+ }
+
node_add(mesh, n);
- // TODO: immediately write the config file?
logger(mesh, MESHLINK_INFO, "Key successfully received from %s", c->name);
return true;
}
-static bool receive_invitation_sptps(void *handle, uint8_t type, const void *data, uint16_t len) {
- connection_t *c = handle;
- meshlink_handle_t *mesh = c->mesh;
-
- if(type == 128) {
- return true;
- }
-
- if(type == 1 && c->status.invitation_used) {
- return finalize_invitation(mesh, c, data, len);
- }
-
- if(type != 0 || len != 18 || c->status.invitation_used) {
- return false;
- }
-
+static bool process_invitation(meshlink_handle_t *mesh, connection_t *c, const void *data) {
// Recover the filename from the cookie and the key
char *fingerprint = ecdsa_get_base64_public_key(mesh->invitation_key);
char hash[64];
config_t config;
- if(!invitation_read(mesh, cookie, &config)) {
+ if(!invitation_read(mesh, "current", cookie, &config, mesh->config_key)) {
logger(mesh, MESHLINK_ERROR, "Error while trying to read invitation file\n");
return false;
}
char *submesh_name = packmsg_get_str_dup(&in);
if(!strcmp(submesh_name, CORE_MESH)) {
+ free(submesh_name);
c->submesh = NULL;
} else {
if(!check_id(submesh_name)) {
logger(mesh, MESHLINK_ERROR, "Invalid invitation file %s\n", cookie);
- abort();
+ free(submesh_name);
return false;
}
c->submesh = lookup_or_create_submesh(mesh, submesh_name);
+ free(submesh_name);
if(!c->submesh) {
+ logger(mesh, MESHLINK_ERROR, "Unknown submesh in invitation file %s\n", cookie);
return false;
}
}
+ if(mesh->inviter_commits_first && !commit_invitation(mesh, c, (const char *)data + 18)) {
+ return false;
+ }
+
+ if(mesh->inviter_commits_first) {
+ devtool_set_inviter_commits_first(true);
+ }
+
// Send the node the contents of the invitation file
sptps_send_record(&c->sptps, 0, config.buf, config.len);
+ config_free(&config);
+
c->status.invitation_used = true;
logger(mesh, MESHLINK_INFO, "Invitation %s successfully sent to %s", cookie, c->name);
return true;
}
+static bool receive_invitation_sptps(void *handle, uint8_t type, const void *data, uint16_t len) {
+ connection_t *c = handle;
+ meshlink_handle_t *mesh = c->mesh;
+
+ if(type == SPTPS_HANDSHAKE) {
+ // The peer should send its cookie first.
+ return true;
+ }
+
+ if(mesh->inviter_commits_first) {
+ if(type == 2 && len == 18 + 32 && !c->status.invitation_used) {
+ return process_invitation(mesh, c, data);
+ }
+ } else {
+ if(type == 0 && len == 18 && !c->status.invitation_used) {
+ return process_invitation(mesh, c, data);
+ } else if(type == 1 && len == 32 && c->status.invitation_used) {
+ return commit_invitation(mesh, c, data);
+ }
+ }
+
+ return false;
+}
+
bool id_h(meshlink_handle_t *mesh, connection_t *c, const char *request) {
+ assert(request);
+ assert(*request);
+
char name[MAX_STRING_SIZE];
if(sscanf(request, "%*d " MAX_STRING " %d.%d", name, &c->protocol_major, &c->protocol_minor) < 2) {
}
if(n->status.blacklisted) {
- logger(mesh, MESHLINK_EPEER, "Peer %s is blacklisted", c->name);
+ logger(mesh, MESHLINK_WARNING, "Peer %s is blacklisted", c->name);
return false;
}
- node_read_public_key(mesh, n);
-
- if(!ecdsa_active(n->ecdsa)) {
+ if(!node_read_public_key(mesh, n)) {
logger(mesh, MESHLINK_ERROR, "No key known for peer %s", c->name);
if(n->status.reachable && !n->status.waitingforkey) {
}
bool send_ack(meshlink_handle_t *mesh, connection_t *c) {
-
- /* Check some options */
-
- if(mesh->self->options & OPTION_PMTU_DISCOVERY) {
- c->options |= OPTION_PMTU_DISCOVERY;
- }
-
- return send_request(mesh, c, NULL, "%d %s %d %x", ACK, mesh->myport, mesh->devclass, (c->options & 0xffffff) | (PROT_MINOR << 24));
+ return send_request(mesh, c, NULL, "%d %s %d %x", ACK, mesh->myport, mesh->devclass, OPTION_PMTU_DISCOVERY | (PROT_MINOR << 24));
}
static void send_everything(meshlink_handle_t *mesh, connection_t *c) {
}
bool ack_h(meshlink_handle_t *mesh, connection_t *c, const char *request) {
+ assert(request);
+ assert(*request);
+
char hisport[MAX_STRING_SIZE];
int devclass;
uint32_t options;
return false;
}
- if(devclass < 0 || devclass > _DEV_CLASS_MAX) {
+ if(devclass < 0 || devclass >= DEV_CLASS_COUNT) {
logger(mesh, MESHLINK_ERROR, "Got bad %s from %s: %s", "ACK", c->name, "devclass invalid");
return false;
}
n->connection->outgoing = NULL;
}
+ /* Remove the edge before terminating the connection, to prevent a graph update. */
+ edge_del(mesh, n->connection->edge);
+ n->connection->edge = NULL;
+
terminate_connection(mesh, n->connection, false);
- /* Run graph algorithm to keep things in sync */
- graph(mesh);
}
}
n->devclass = devclass;
n->status.dirty = true;
- n->last_successfull_connection = time(NULL);
+ n->last_successfull_connection = mesh->loop.now.tv_sec;
n->connection = c;
c->node = n;
- if(!(c->options & options & OPTION_PMTU_DISCOVERY)) {
- c->options &= ~OPTION_PMTU_DISCOVERY;
- options &= ~OPTION_PMTU_DISCOVERY;
- }
-
- c->options |= options;
-
/* Activate this connection */
c->allow_request = ALL;
/* Create an edge_t for this connection */
- assert(devclass >= 0 && devclass <= _DEV_CLASS_MAX);
+ assert(devclass >= 0 && devclass < DEV_CLASS_COUNT);
c->edge = new_edge();
c->edge->from = mesh->self;
c->edge->to = n;
sockaddrcpy_setport(&c->edge->address, &c->address, atoi(hisport));
- c->edge->weight = dev_class_traits[devclass].edge_weight;
+ c->edge->weight = mesh->dev_class_traits[devclass].edge_weight;
c->edge->connection = c;
- c->edge->options = c->options;
edge_add(mesh, c->edge);