--- /dev/null
+
+#include "meshlink_internal.h"
+#include "discovery.h"
+
+#include <pthread.h>
+
+#include <avahi-core/core.h>
+#include <avahi-core/lookup.h>
+#include <avahi-core/publish.h>
+#include <avahi-common/simple-watch.h>
+#include <avahi-common/malloc.h>
+#include <avahi-common/alternative.h>
+#include <avahi-common/error.h>
+
+#define MESHLINK_MDNS_SERVICE_TYPE "_meshlink._tcp"
+//#define MESHLINK_MDNS_SERVICE_NAME "Meshlink"
+#define MESHLINK_MDNS_FINGERPRINT_KEY "fingerprint"
+
+static void discovery_resolve_callback(
+ AvahiSServiceResolver *resolver,
+ AVAHI_GCC_UNUSED AvahiIfIndex interface,
+ AVAHI_GCC_UNUSED AvahiProtocol protocol,
+ AvahiResolverEvent event,
+ const char *name,
+ const char *type,
+ const char *domain,
+ const char *host_name,
+ const AvahiAddress *address,
+ uint16_t port,
+ AvahiStringList *txt,
+ AvahiLookupResultFlags flags,
+ AVAHI_GCC_UNUSED void* userdata)
+{
+
+ meshlink_handle_t *mesh = userdata;
+
+ /* Called whenever a service has been resolved successfully or timed out */
+
+ switch (event)
+ {
+ case AVAHI_RESOLVER_FAILURE:
+ fprintf(stderr, "(Resolver) Failed to resolve service '%s' of type '%s' in domain '%s': %s\n", name, type, domain, avahi_strerror(avahi_server_errno(mesh->avahi_server)));
+ break;
+
+ case AVAHI_RESOLVER_FOUND:
+ {
+ char straddr[AVAHI_ADDRESS_STR_MAX], *strtxt;
+
+ fprintf(stderr, "(Resolver) Service '%s' of type '%s' in domain '%s':\n", name, type, domain);
+
+ avahi_address_snprint(straddr, sizeof(straddr), address);
+ strtxt = avahi_string_list_to_string(txt);
+ fprintf(stderr,
+ "\t%s:%u (%s)\n"
+ "\tTXT=%s\n"
+ "\tcookie is %u\n"
+ "\tis_local: %i\n"
+ "\twide_area: %i\n"
+ "\tmulticast: %i\n"
+ "\tcached: %i\n",
+ host_name, port, straddr,
+ strtxt,
+ avahi_string_list_get_service_cookie(txt),
+ !!(flags & AVAHI_LOOKUP_RESULT_LOCAL),
+ !!(flags & AVAHI_LOOKUP_RESULT_WIDE_AREA),
+ !!(flags & AVAHI_LOOKUP_RESULT_MULTICAST),
+ !!(flags & AVAHI_LOOKUP_RESULT_CACHED));
+ avahi_free(strtxt);
+
+ // retrieve fingerprint
+ AvahiStringList *fgli = avahi_string_list_find(txt, MESHLINK_MDNS_FINGERPRINT_KEY);
+ meshlink_node_t *node = meshlink_get_node(mesh, name);
+
+ fprintf(stderr, "%p, %p, %s, %s\n", fgli, node, avahi_string_list_get_text(fgli), meshlink_get_fingerprint(mesh, node));
+
+ if( node && fgli && strcmp(avahi_string_list_get_text(fgli)+strlen(MESHLINK_MDNS_FINGERPRINT_KEY)+1, meshlink_get_fingerprint(mesh, node)) == 0 )
+ {
+ fprintf(stderr, "Node %s is part of the mesh network - updating ip address.\n", node->name);
+ }
+ else
+ {
+ fprintf(stderr, "Node %s is not part of the mesh network - ignoring ip address.\n", node->name);
+ }
+ }
+ }
+
+ avahi_s_service_resolver_free(resolver);
+}
+
+static void discovery_entry_group_callback(AvahiServer *server, AvahiSEntryGroup *group, AvahiEntryGroupState state, AVAHI_GCC_UNUSED void *userdata)
+{
+ meshlink_handle_t *mesh = userdata;
+
+ /* Called whenever the entry group state changes */
+ switch (state)
+ {
+ case AVAHI_ENTRY_GROUP_ESTABLISHED:
+ /* The entry group has been established successfully */
+ fprintf(stderr, "Service '%s' successfully established.\n", /*MESHLINK_MDNS_SERVICE_NAME*/ mesh->name);
+ break;
+
+ case AVAHI_ENTRY_GROUP_COLLISION:
+ fprintf(stderr, "Service name collision '%s'\n", /*MESHLINK_MDNS_SERVICE_NAME*/ mesh->name);
+ break;
+
+ case AVAHI_ENTRY_GROUP_FAILURE :
+ fprintf(stderr, "Entry group failure: %s\n", avahi_strerror(avahi_server_errno(mesh->avahi_server)));
+
+ /* Some kind of failure happened while we were registering our services */
+ avahi_simple_poll_quit(mesh->avahi_poll);
+ break;
+
+ case AVAHI_ENTRY_GROUP_UNCOMMITED:
+ case AVAHI_ENTRY_GROUP_REGISTERING:
+ ;
+ }
+}
+
+
+static void discovery_create_services(meshlink_handle_t *mesh)
+{
+ fprintf(stderr, "Adding service '%s'\n", /*MESHLINK_MDNS_SERVICE_NAME*/ mesh->name);
+
+ /* If this is the first time we're called, let's create a new entry group */
+ if (!mesh->avahi_group)
+ if (!(mesh->avahi_group = avahi_s_entry_group_new(mesh->avahi_server, discovery_entry_group_callback, mesh))) {
+ fprintf(stderr, "avahi_entry_group_new() failed: %s\n", avahi_strerror(avahi_server_errno(mesh->avahi_server)));
+ goto fail;
+ }
+
+ /* Create some random TXT data */
+ char fingerprint[1024] = "";
+ snprintf(fingerprint, sizeof(fingerprint), "%s=%s", MESHLINK_MDNS_FINGERPRINT_KEY, meshlink_get_fingerprint(mesh, meshlink_get_node(mesh, mesh->name)));
+
+ /* Add the service for IPP */
+ int ret = 0;
+ if ((ret = avahi_server_add_service(mesh->avahi_server, mesh->avahi_group, AVAHI_IF_UNSPEC, AVAHI_PROTO_UNSPEC, 0, /*MESHLINK_MDNS_SERVICE_NAME*/ mesh->name, MESHLINK_MDNS_SERVICE_TYPE, NULL, NULL, mesh->myport ? atoi(mesh->myport) : 655, fingerprint, NULL)) < 0) {
+ fprintf(stderr, "Failed to add _ipp._tcp service: %s\n", avahi_strerror(ret));
+ goto fail;
+ }
+
+ /* Tell the server to register the service */
+ if ((ret = avahi_s_entry_group_commit(mesh->avahi_group)) < 0) {
+ fprintf(stderr, "Failed to commit entry_group: %s\n", avahi_strerror(ret));
+ goto fail;
+ }
+
+ return;
+
+fail:
+ avahi_simple_poll_quit(mesh->avahi_poll);
+}
+
+static void discovery_server_callback(AvahiServer *server, AvahiServerState state, AVAHI_GCC_UNUSED void * userdata)
+{
+ meshlink_handle_t *mesh = userdata;
+
+ switch (state)
+ {
+ case AVAHI_SERVER_RUNNING:
+ /* The serve has startup successfully and registered its host
+ * name on the network, so it's time to create our services */
+ if (!mesh->avahi_group)
+ discovery_create_services(mesh);
+ break;
+
+ case AVAHI_SERVER_COLLISION:
+ /* A host name collision happened. Let's do nothing */
+ break;
+
+ case AVAHI_SERVER_REGISTERING:
+ /* Let's drop our registered services. When the server is back
+ * in AVAHI_SERVER_RUNNING state we will register them
+ * again with the new host name. */
+ //if (mesh->avahi_group)
+ // avahi_s_entry_group_reset(mesh->avahi_group);
+ break;
+
+ case AVAHI_SERVER_FAILURE:
+ /* Terminate on failure */
+ fprintf(stderr, "Server failure: %s\n", avahi_strerror(avahi_server_errno(mesh->avahi_server)));
+ avahi_simple_poll_quit(mesh->avahi_poll);
+ break;
+
+ case AVAHI_SERVER_INVALID:
+ break;
+ }
+}
+
+static void discovery_browse_callback(
+ AvahiSServiceBrowser *browser,
+ AvahiIfIndex interface,
+ AvahiProtocol protocol,
+ AvahiBrowserEvent event,
+ const char *name,
+ const char *type,
+ const char *domain,
+ AVAHI_GCC_UNUSED AvahiLookupResultFlags flags,
+ void* userdata)
+{
+ meshlink_handle_t *mesh = userdata;
+
+ /* Called whenever a new services becomes available on the LAN or is removed from the LAN */
+
+ switch (event)
+ {
+ case AVAHI_BROWSER_FAILURE:
+ fprintf(stderr, "(Browser) %s\n", avahi_strerror(avahi_server_errno(mesh->avahi_server)));
+ avahi_simple_poll_quit(mesh->avahi_poll);
+ return;
+
+ case AVAHI_BROWSER_NEW:
+ fprintf(stderr, "(Browser) NEW: service '%s' of type '%s' in domain '%s'\n", name, type, domain);
+ /* We ignore the returned resolver object. In the callback
+ function we free it. If the server is terminated before
+ the callback function is called the server will free
+ the resolver for us. */
+ if (!(avahi_s_service_resolver_new(mesh->avahi_server, interface, protocol, name, type, domain, AVAHI_PROTO_UNSPEC, 0, discovery_resolve_callback, mesh)))
+ fprintf(stderr, "Failed to resolve service '%s': %s\n", name, avahi_strerror(avahi_server_errno(mesh->avahi_server)));
+ break;
+
+ case AVAHI_BROWSER_REMOVE:
+ fprintf(stderr, "(Browser) REMOVE: service '%s' of type '%s' in domain '%s'\n", name, type, domain);
+ break;
+
+ case AVAHI_BROWSER_ALL_FOR_NOW:
+ case AVAHI_BROWSER_CACHE_EXHAUSTED:
+ fprintf(stderr, "(Browser) %s\n", event == AVAHI_BROWSER_CACHE_EXHAUSTED ? "CACHE_EXHAUSTED" : "ALL_FOR_NOW");
+ break;
+ }
+}
+
+static void *discovery_loop(void *arg)
+{
+ meshlink_handle_t *mesh = arg;
+
+ avahi_simple_poll_loop(mesh->avahi_poll);
+
+ return NULL;
+}
+
+bool discovery_start(meshlink_handle_t *mesh)
+{
+ // Allocate discovery loop object
+ if (!(mesh->avahi_poll = avahi_simple_poll_new())) {
+ fprintf(stderr, "Failed to create discovery poll object.\n");
+ goto fail;
+ }
+
+ // Let's set the host name for this server.
+ AvahiServerConfig config;
+ avahi_server_config_init(&config);
+ config.host_name = avahi_strdup(mesh->name);
+ config.publish_workstation = 0;
+
+ /* Allocate a new server */
+ int error;
+ mesh->avahi_server = avahi_server_new(avahi_simple_poll_get(mesh->avahi_poll), &config, discovery_server_callback, mesh, &error);
+
+ /* Free the configuration data */
+ avahi_server_config_free(&config);
+
+ /* Check wether creating the server object succeeded */
+ if (!mesh->avahi_server) {
+ fprintf(stderr, "Failed to create discovery server: %s\n", avahi_strerror(error));
+ goto fail;
+ }
+
+ // Create the service browser
+ if (!(mesh->avahi_browser = avahi_s_service_browser_new(mesh->avahi_server, AVAHI_IF_UNSPEC, AVAHI_PROTO_UNSPEC, MESHLINK_MDNS_SERVICE_TYPE, NULL, 0, discovery_browse_callback, mesh))) {
+ fprintf(stderr, "Failed to create discovery service browser: %s\n", avahi_strerror(avahi_server_errno(mesh->avahi_server)));
+ goto fail;
+ }
+
+ // Start the discovery thread
+ if(pthread_create(&mesh->discovery_thread, NULL, discovery_loop, mesh) != 0) {
+ fprintf(stderr, "Could not start discovery thread: %s\n", strerror(errno));
+ memset(&mesh->discovery_thread, 0, sizeof mesh->discovery_thread);
+ goto fail;
+ }
+
+ mesh->discovery_threadstarted = true;
+
+ return true;
+
+fail:
+ if (mesh->avahi_browser)
+ avahi_s_service_browser_free(mesh->avahi_browser);
+
+ if (mesh->avahi_server)
+ avahi_server_free(mesh->avahi_server);
+
+ if (mesh->avahi_poll)
+ avahi_simple_poll_free(mesh->avahi_poll);
+
+ return false;
+}
+
+void discovery_stop(meshlink_handle_t *mesh)
+{
+ // @TODO: Shut down
+ avahi_simple_poll_quit(mesh->avahi_poll);
+
+ // Wait for the discovery thread to finish
+
+ pthread_join(mesh->discovery_thread, NULL);
+
+ // Clean up resources
+ if (mesh->avahi_browser)
+ avahi_s_service_browser_free(mesh->avahi_browser);
+
+ if (mesh->avahi_server)
+ avahi_server_free(mesh->avahi_server);
+
+ if (mesh->avahi_poll)
+ avahi_simple_poll_free(mesh->avahi_poll);
+}
#include "utils.h"
#include "xalloc.h"
#include "ed25519/sha512.h"
+#include "discovery.h"
#ifndef MSG_NOSIGNAL
#define MSG_NOSIGNAL 0
static const char *errstr[] = {
[MESHLINK_OK] = "No error",
+ [MESHLINK_EINVAL] = "Invalid argument",
[MESHLINK_ENOMEM] = "Out of memory",
[MESHLINK_ENOENT] = "No such node",
+ [MESHLINK_EEXIST] = "Node already exists",
+ [MESHLINK_EINTERNAL] = "Internal error",
+ [MESHLINK_ERESOLV] = "Could not resolve hostname",
+ [MESHLINK_ESTORAGE] = "Storage error",
+ [MESHLINK_ENETWORK] = "Network error",
+ [MESHLINK_EPEER] = "Error communicating with peer",
};
const char *meshlink_strerror(meshlink_errno_t err) {
if(!(key = ecdsa_generate())) {
fprintf(stderr, "Error during key generation!\n");
+ meshlink_errno = MESHLINK_EINTERNAL;
return false;
} else
fprintf(stderr, "Done.\n");
snprintf(privname, sizeof privname, "%s" SLASH "ecdsa_key.priv", mesh->confbase);
f = fopen(privname, "w");
- if(!f)
+ if(!f) {
+ meshlink_errno = MESHLINK_ESTORAGE;
return false;
+ }
#ifdef HAVE_FCHMOD
fchmod(fileno(f), 0600);
fprintf(stderr, "Error writing private key!\n");
ecdsa_free(key);
fclose(f);
+ meshlink_errno = MESHLINK_EINTERNAL;
return false;
}
fclose(f);
-
snprintf(pubname, sizeof pubname, "%s" SLASH "hosts" SLASH "%s", mesh->confbase, mesh->name);
f = fopen(pubname, "a");
- if(!f)
+ if(!f) {
+ meshlink_errno = MESHLINK_ESTORAGE;
return false;
+ }
char *pubkey = ecdsa_get_base64_public_key(key);
fprintf(f, "ECDSAPublicKey = %s\n", pubkey);
static bool meshlink_setup(meshlink_handle_t *mesh) {
if(mkdir(mesh->confbase, 0777) && errno != EEXIST) {
fprintf(stderr, "Could not create directory %s: %s\n", mesh->confbase, strerror(errno));
+ meshlink_errno = MESHLINK_ESTORAGE;
return false;
}
if(mkdir(filename, 0777) && errno != EEXIST) {
fprintf(stderr, "Could not create directory %s: %s\n", filename, strerror(errno));
+ meshlink_errno = MESHLINK_ESTORAGE;
return false;
}
if(!access(filename, F_OK)) {
fprintf(stderr, "Configuration file %s already exists!\n", filename);
+ meshlink_errno = MESHLINK_EEXIST;
return false;
}
FILE *f = fopen(filename, "w");
if(!f) {
fprintf(stderr, "Could not create file %s: %s\n", filename, strerror(errno));
- return 1;
+ meshlink_errno = MESHLINK_ESTORAGE;
+ return false;
}
fprintf(f, "Name = %s\n", mesh->name);
fclose(f);
- if(!ecdsa_keygen(mesh))
+ if(!ecdsa_keygen(mesh)) {
+ meshlink_errno = MESHLINK_EINTERNAL;
return false;
+ }
check_port(mesh);
if(!confbase || !*confbase) {
fprintf(stderr, "No confbase given!\n");
+ meshlink_errno = MESHLINK_EINVAL;
return NULL;
}
if(!check_id(name)) {
fprintf(stderr, "Invalid name given!\n");
+ meshlink_errno = MESHLINK_EINVAL;
return NULL;
} else { usingname = true;}
}
if(access(filename, R_OK)) {
if(errno == ENOENT) {
// If not, create it
- meshlink_setup(mesh);
+ if(!meshlink_setup(mesh)) {
+ // meshlink_errno is set by meshlink_setup()
+ return NULL;
+ }
} else {
fprintf(stderr, "Cannot not read from %s: %s\n", filename, strerror(errno));
- return meshlink_close(mesh), NULL;
+ meshlink_close(mesh);
+ meshlink_errno = MESHLINK_ESTORAGE;
+ return NULL;
}
}
init_configuration(&mesh->config);
- if(!read_server_config(mesh))
- return meshlink_close(mesh), NULL;
+ if(!read_server_config(mesh)) {
+ meshlink_close(mesh);
+ meshlink_errno = MESHLINK_ESTORAGE;
+ return NULL;
+ };
#ifdef HAVE_MINGW
struct WSAData wsa_state;
// Setup up everything
// TODO: we should not open listening sockets yet
- if(!setup_network(mesh))
- return meshlink_close(mesh), NULL;
+ if(!setup_network(mesh)) {
+ meshlink_close(mesh);
+ meshlink_errno = MESHLINK_ENETWORK;
+ return NULL;
+ }
return mesh;
}
}
bool meshlink_start(meshlink_handle_t *mesh) {
- if(!mesh)
+ if(!mesh) {
+ meshlink_errno = MESHLINK_EINVAL;
return false;
+ }
// TODO: open listening sockets first
//Check that a valid name is set
if(!mesh->name ) {
fprintf(stderr, "No name given!\n");
+ meshlink_errno = MESHLINK_EINVAL;
return false;
}
if(pthread_create(&mesh->thread, NULL, meshlink_main_loop, mesh) != 0) {
fprintf(stderr, "Could not start thread: %s\n", strerror(errno));
memset(&mesh->thread, 0, sizeof mesh->thread);
+ meshlink_errno = MESHLINK_EINTERNAL;
return false;
}
mesh->threadstarted=true;
+ // Start discovery
+ if(!discovery_start(mesh))
+ return false;
+
return true;
}
void meshlink_stop(meshlink_handle_t *mesh) {
- if(!mesh)
+ if(!mesh) {
+ meshlink_errno = MESHLINK_EINVAL;
return;
+ }
+
+ // Stop discovery
+ discovery_stop(mesh);
// Shut down the listening sockets to signal the main thread to shut down
}
void meshlink_close(meshlink_handle_t *mesh) {
- if(!mesh || !mesh->confbase)
+ if(!mesh || !mesh->confbase) {
+ meshlink_errno = MESHLINK_EINVAL;
return;
+ }
// Close and free all resources used.
}
void meshlink_set_receive_cb(meshlink_handle_t *mesh, meshlink_receive_cb_t cb) {
- if(!mesh)
+ if(!mesh) {
+ meshlink_errno = MESHLINK_EINVAL;
return;
+ }
+
mesh->receive_cb = cb;
}
void meshlink_set_node_status_cb(meshlink_handle_t *mesh, meshlink_node_status_cb_t cb) {
- if(!mesh)
+ if(!mesh) {
+ meshlink_errno = MESHLINK_EINVAL;
return;
+ }
+
mesh->node_status_cb = cb;
}
void meshlink_set_log_cb(meshlink_handle_t *mesh, meshlink_log_level_t level, meshlink_log_cb_t cb) {
- if(!mesh)
+ if(!mesh) {
+ meshlink_errno = MESHLINK_EINVAL;
return;
+ }
+
mesh->log_cb = cb;
mesh->log_level = level;
}
bool meshlink_send(meshlink_handle_t *mesh, meshlink_node_t *destination, const void *data, size_t len) {
- if(!mesh || !destination)
+ if(!mesh || !destination) {
+ meshlink_errno = MESHLINK_EINVAL;
return false;
+ }
+
if(!len)
return true;
- if(!data)
+
+ if(!data) {
+ meshlink_errno = MESHLINK_EINVAL;
return false;
+ }
/* If there is no outgoing list yet, create one. */
}
ssize_t meshlink_get_pmtu(meshlink_handle_t *mesh, meshlink_node_t *destination) {
- if(!mesh || !destination)
+ if(!mesh || !destination) {
+ meshlink_errno = MESHLINK_EINVAL;
return -1;
+ }
node_t *n = (node_t *)destination;
if(!n->status.reachable)
return MTU;
}
+char *meshlink_get_fingerprint(meshlink_handle_t *mesh, meshlink_node_t *node) {
+ if(!mesh || !node) {
+ meshlink_errno = MESHLINK_EINVAL;
+ return NULL;
+ }
+
+ node_t *n = (node_t *)node;
+
+ if(!node_read_ecdsa_public_key(mesh, n) || !n->ecdsa) {
+ meshlink_errno = MESHLINK_EINTERNAL;
+ return false;
+ }
+
+ char *fingerprint = ecdsa_get_base64_public_key(n->ecdsa);
+
+ if(!fingerprint)
+ meshlink_errno = MESHLINK_EINTERNAL;
+
+ return fingerprint;
+}
+
meshlink_node_t *meshlink_get_node(meshlink_handle_t *mesh, const char *name) {
- if(!mesh || !name)
+ if(!mesh || !name) {
+ meshlink_errno = MESHLINK_EINVAL;
return NULL;
+ }
+
return (meshlink_node_t *)lookup_node(mesh, (char *)name); // TODO: make lookup_node() use const
}
meshlink_node_t **meshlink_get_all_nodes(meshlink_handle_t *mesh, meshlink_node_t **nodes, size_t *nmemb) {
- if(!mesh || (nmemb && !nodes))
+ if(!mesh || !nmemb || (*nmemb && !nodes)) {
+ meshlink_errno = MESHLINK_EINVAL;
return NULL;
+ }
- meshlink_node_t **result, **p;
+ meshlink_node_t **result;
//lock mesh->nodes
pthread_mutex_lock(&(mesh->nodes_mutex));
result = realloc(nodes, *nmemb * sizeof *nodes);
if(result) {
+ meshlink_node_t **p = result;
for splay_each(node_t, n, mesh->nodes)
*p++ = (meshlink_node_t *)n;
} else {
}
bool meshlink_sign(meshlink_handle_t *mesh, const void *data, size_t len, void *signature, size_t *siglen) {
- if(!mesh || !data || !len || !signature || !siglen)
+ if(!mesh || !data || !len || !signature || !siglen) {
+ meshlink_errno = MESHLINK_EINVAL;
return false;
- if(*siglen < MESHLINK_SIGLEN)
+ }
+
+ if(*siglen < MESHLINK_SIGLEN) {
+ meshlink_errno = MESHLINK_EINVAL;
return false;
- if(!ecdsa_sign(mesh->self->connection->ecdsa, data, len, signature))
+ }
+
+ if(!ecdsa_sign(mesh->self->connection->ecdsa, data, len, signature)) {
+ meshlink_errno = MESHLINK_EINTERNAL;
return false;
+ }
+
*siglen = MESHLINK_SIGLEN;
return true;
}
bool meshlink_verify(meshlink_handle_t *mesh, meshlink_node_t *source, const void *data, size_t len, const void *signature, size_t siglen) {
- if(!mesh || !data || !len || !signature)
+ if(!mesh || !data || !len || !signature) {
+ meshlink_errno = MESHLINK_EINVAL;
return false;
- if(siglen != MESHLINK_SIGLEN)
+ }
+
+ if(siglen != MESHLINK_SIGLEN) {
+ meshlink_errno = MESHLINK_EINVAL;
return false;
+ }
+
struct node_t *n = (struct node_t *)source;
node_read_ecdsa_public_key(mesh, n);
- if(!n->ecdsa)
+ if(!n->ecdsa) {
+ meshlink_errno = MESHLINK_EINTERNAL;
return false;
+ }
+
return ecdsa_verify(((struct node_t *)source)->ecdsa, data, len, signature);
}
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));
+ meshlink_errno = MESHLINK_ESTORAGE;
return false;
}
DIR *dir = opendir(filename);
if(!dir) {
fprintf(stderr, "Could not read directory %s: %s\n", filename, strerror(errno));
+ meshlink_errno = MESHLINK_ESTORAGE;
return false;
}
if(errno) {
fprintf(stderr, "Error while reading directory %s: %s\n", filename, strerror(errno));
closedir(dir);
+ meshlink_errno = MESHLINK_ESTORAGE;
return false;
}
if(!f) {
if(errno != ENOENT) {
fprintf(stderr, "Could not read %s: %s\n", filename, strerror(errno));
+ meshlink_errno = MESHLINK_ESTORAGE;
return false;
}
mesh->invitation_key = ecdsa_generate();
if(!mesh->invitation_key) {
fprintf(stderr, "Could not generate a new key!\n");
+ meshlink_errno = MESHLINK_EINTERNAL;
return false;
}
f = fopen(filename, "w");
if(!f) {
fprintf(stderr, "Could not write %s: %s\n", filename, strerror(errno));
+ meshlink_errno = MESHLINK_ESTORAGE;
return false;
}
chmod(filename, 0600);
} else {
mesh->invitation_key = ecdsa_read_pem_private_key(f);
fclose(f);
- if(!mesh->invitation_key)
+ if(!mesh->invitation_key) {
fprintf(stderr, "Could not read private key from %s\n", filename);
+ meshlink_errno = MESHLINK_ESTORAGE;
+ }
}
return mesh->invitation_key;
}
bool meshlink_add_address(meshlink_handle_t *mesh, const char *address) {
- if(!mesh || !address)
+ if(!mesh || !address) {
+ meshlink_errno = MESHLINK_EINVAL;
return false;
+ }
for(const char *p = address; *p; p++) {
if(isalnum(*p) || *p == '-' || *p == '.' || *p == ':')
continue;
fprintf(stderr, "Invalid character in address: %s\n", address);
+ meshlink_errno = MESHLINK_EINVAL;
return false;
}
}
char *meshlink_invite(meshlink_handle_t *mesh, const char *name) {
- if(!mesh)
- return false;
+ if(!mesh) {
+ meshlink_errno = MESHLINK_EINVAL;
+ return NULL;
+ }
// Check validity of the new node's name
if(!check_id(name)) {
fprintf(stderr, "Invalid name for node.\n");
+ meshlink_errno = MESHLINK_EINVAL;
return NULL;
}
snprintf(filename, sizeof filename, "%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);
+ meshlink_errno = MESHLINK_EEXIST;
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);
+ meshlink_errno = MESHLINK_EEXIST;
return NULL;
}
char *address = get_my_hostname(mesh);
if(!address) {
fprintf(stderr, "No Address known for ourselves!\n");
+ meshlink_errno = MESHLINK_ERESOLV;
return NULL;
}
- if(!refresh_invitation_key(mesh))
+ if(!refresh_invitation_key(mesh)) {
+ meshlink_errno = MESHLINK_EINTERNAL;
return NULL;
+ }
char hash[64];
int ifd = open(filename, O_RDWR | O_CREAT | O_EXCL, 0600);
if(!ifd) {
fprintf(stderr, "Could not create invitation file %s: %s\n", filename, strerror(errno));
+ meshlink_errno = MESHLINK_ESTORAGE;
return NULL;
}
FILE *f = fdopen(ifd, "w");
fclose(tc);
} else {
fprintf(stderr, "Could not create %s: %s\n", filename, strerror(errno));
+ meshlink_errno = MESHLINK_ESTORAGE;
return NULL;
}
}
bool meshlink_join(meshlink_handle_t *mesh, const char *invitation) {
- if(!mesh || !invitation)
+ if(!mesh || !invitation) {
+ meshlink_errno = MESHLINK_EINVAL;
return false;
+ }
//TODO: think of a better name for this variable, or of a different way to tokenize the invitation URL.
char copy[strlen(invitation) + 1];
// Generate a throw-away key for the invitation.
ecdsa_t *key = ecdsa_generate();
- if(!key)
+ if(!key) {
+ meshlink_errno = MESHLINK_EINTERNAL;
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)
+ if(!ai) {
+ meshlink_errno = MESHLINK_ERESOLV;
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));
freeaddrinfo(ai);
+ meshlink_errno = MESHLINK_ENETWORK;
return false;
}
fprintf(stderr, "Could not connect to %s port %s: %s\n", address, port, strerror(errno));
closesocket(mesh->sock);
freeaddrinfo(ai);
+ meshlink_errno = MESHLINK_ENETWORK;
return false;
}
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);
+ meshlink_errno = MESHLINK_ENETWORK;
return false;
}
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);
+ meshlink_errno = MESHLINK_ENETWORK;
return false;
}
char hishash[64];
if(sha512(fingerprint, strlen(fingerprint), hishash)) {
fprintf(stderr, "Could not create hash\n%s\n", mesh->line + 2);
+ meshlink_errno = MESHLINK_EINTERNAL;
return false;
}
if(memcmp(hishash, mesh->hash, 18)) {
fprintf(stderr, "Peer has an invalid key!\n%s\n", mesh->line + 2);
+ meshlink_errno = MESHLINK_EPEER;
return false;
}
ecdsa_t *hiskey = ecdsa_set_base64_public_key(fingerprint);
- if(!hiskey)
+ if(!hiskey) {
+ meshlink_errno = MESHLINK_EINTERNAL;
return false;
+ }
// Start an SPTPS session
- if(!sptps_start(&mesh->sptps, mesh, true, false, key, hiskey, "meshlink invitation", 15, invitation_send, invitation_receive))
+ if(!sptps_start(&mesh->sptps, mesh, true, false, key, hiskey, "meshlink invitation", 15, invitation_send, invitation_receive)) {
+ meshlink_errno = MESHLINK_EINTERNAL;
return false;
+ }
// Feed rest of input buffer to SPTPS
- if(!sptps_receive_data(&mesh->sptps, mesh->buffer, mesh->blen))
+ if(!sptps_receive_data(&mesh->sptps, mesh->buffer, mesh->blen)) {
+ meshlink_errno = MESHLINK_EPEER;
return false;
+ }
int len;
if(errno == EINTR)
continue;
fprintf(stderr, "Error reading data from %s port %s: %s\n", address, port, strerror(errno));
+ meshlink_errno = MESHLINK_ENETWORK;
return false;
}
- if(!sptps_receive_data(&mesh->sptps, mesh->line, len))
+ if(!sptps_receive_data(&mesh->sptps, mesh->line, len)) {
+ meshlink_errno = MESHLINK_EPEER;
return false;
+ }
}
sptps_stop(&mesh->sptps);
if(!mesh->success) {
fprintf(stderr, "Connection closed by peer, invitation cancelled.\n");
+ meshlink_errno = MESHLINK_EPEER;
return false;
}
invalid:
fprintf(stderr, "Invalid invitation URL or you are already connected to a Mesh ?\n");
+ meshlink_errno = MESHLINK_EINVAL;
return false;
}
char *meshlink_export(meshlink_handle_t *mesh) {
- if(!mesh)
+ if(!mesh) {
+ meshlink_errno = MESHLINK_EINVAL;
return NULL;
+ }
char filename[PATH_MAX];
snprintf(filename, sizeof filename, "%s" SLASH "hosts" SLASH "%s", mesh->confbase, mesh->self->name);
FILE *f = fopen(filename, "r");
if(!f) {
fprintf(stderr, "Could not open %s: %s\n", filename, strerror(errno));
+ meshlink_errno = MESHLINK_ESTORAGE;
return NULL;
}
if(fread(buf + len - fsize - 1, fsize, 1, f) != 1) {
fprintf(stderr, "Error reading from %s: %s\n", filename, strerror(errno));
fclose(f);
+ meshlink_errno = MESHLINK_ESTORAGE;
return NULL;
}
}
bool meshlink_import(meshlink_handle_t *mesh, const char *data) {
- if(!mesh || !data)
+ if(!mesh || !data) {
+ meshlink_errno = MESHLINK_EINVAL;
return false;
+ }
if(strncmp(data, "Name = ", 7)) {
fprintf(stderr, "Invalid data\n");
+ meshlink_errno = MESHLINK_EPEER;
return false;
}
char *end = strchr(data + 7, '\n');
if(!end) {
fprintf(stderr, "Invalid data\n");
+ meshlink_errno = MESHLINK_EPEER;
return false;
}
name[len] = 0;
if(!check_id(name)) {
fprintf(stderr, "Invalid Name\n");
+ meshlink_errno = MESHLINK_EPEER;
return false;
}
snprintf(filename, sizeof filename, "%s" SLASH "hosts" SLASH "%s", mesh->confbase, name);
if(!access(filename, F_OK)) {
fprintf(stderr, "File %s already exists, not importing\n", filename);
+ meshlink_errno = MESHLINK_EEXIST;
return false;
}
if(errno != ENOENT) {
fprintf(stderr, "Error accessing %s: %s\n", filename, strerror(errno));
+ meshlink_errno = MESHLINK_ESTORAGE;
return false;
}
FILE *f = fopen(filename, "w");
if(!f) {
fprintf(stderr, "Could not create %s: %s\n", filename, strerror(errno));
+ meshlink_errno = MESHLINK_ESTORAGE;
return false;
}
}
void meshlink_blacklist(meshlink_handle_t *mesh, meshlink_node_t *node) {
- if(!mesh || !node)
+ if(!mesh || !node) {
+ meshlink_errno = MESHLINK_EINVAL;
return;
+ }
node_t *n;
n = (node_t*)node;
}
void meshlink_whitelist(meshlink_handle_t *mesh, meshlink_node_t *node) {
- if(!mesh || !node)
+ if(!mesh || !node) {
+ meshlink_errno = MESHLINK_EINVAL;
return;
+ }
node_t *n = (node_t *)node;
n->status.blacklisted = false;
projects / meshlink / commitdiff