]> git.meshlink.io Git - meshlink/blobdiff - src/meshlink.c
Fix __warn_unused_result__, add more of it and fix the resulting warnings.
[meshlink] / src / meshlink.c
index 98e02f7db623d8c8b4e837eb96e77f23580fddd8..75458fe6940ed221059e39b29035a2f84de6ccc8 100644 (file)
@@ -903,23 +903,10 @@ static bool meshlink_setup(meshlink_handle_t *mesh) {
                return false;
        }
 
-       if(!main_config_lock(mesh)) {
-               logger(NULL, MESHLINK_ERROR, "Cannot lock main config file\n");
-               meshlink_errno = MESHLINK_ESTORAGE;
-               return false;
-       }
-
        return true;
 }
 
 static bool meshlink_read_config(meshlink_handle_t *mesh) {
-       // Open the configuration file and lock it
-       if(!main_config_lock(mesh)) {
-               logger(NULL, MESHLINK_ERROR, "Cannot lock main config file\n");
-               meshlink_errno = MESHLINK_ESTORAGE;
-               return false;
-       }
-
        config_t config;
 
        if(!main_config_read(mesh, "current", &config, mesh->config_key)) {
@@ -1104,14 +1091,11 @@ bool meshlink_encrypted_key_rotate(meshlink_handle_t *mesh, const void *new_key,
 
        devtool_keyrotate_probe(1);
 
-       main_config_unlock(mesh);
-
        // Rename confbase/current/ to confbase/old
 
        if(!config_rename(mesh, "current", "old")) {
                logger(mesh, MESHLINK_ERROR, "Cannot rename %s/current to %s/old\n", mesh->confbase, mesh->confbase);
                meshlink_errno = MESHLINK_ESTORAGE;
-               main_config_lock(mesh);
                pthread_mutex_unlock(&mesh->mutex);
                return false;
        }
@@ -1123,18 +1107,12 @@ bool meshlink_encrypted_key_rotate(meshlink_handle_t *mesh, const void *new_key,
        if(!config_rename(mesh, "new", "current")) {
                logger(mesh, MESHLINK_ERROR, "Cannot rename %s/new to %s/current\n", mesh->confbase, mesh->confbase);
                meshlink_errno = MESHLINK_ESTORAGE;
-               main_config_lock(mesh);
                pthread_mutex_unlock(&mesh->mutex);
                return false;
        }
 
        devtool_keyrotate_probe(3);
 
-       if(!main_config_lock(mesh)) {
-               pthread_mutex_unlock(&mesh->mutex);
-               return false;
-       }
-
        // Cleanup the "old" confbase sub-directory
 
        if(!config_destroy(mesh->confbase, "old")) {
@@ -1325,6 +1303,12 @@ meshlink_handle_t *meshlink_open_ex(const meshlink_open_params_t *params) {
 
        meshlink_queue_init(&mesh->outpacketqueue);
 
+       // Atomically lock the configuration directory.
+       if(!main_config_lock(mesh)) {
+               meshlink_close(mesh);
+               return NULL;
+       }
+
        // If no configuration exists yet, create it.
 
        if(!meshlink_confbase_exists(mesh)) {
@@ -1377,7 +1361,11 @@ meshlink_handle_t *meshlink_open_ex(const meshlink_open_params_t *params) {
        }
 
        add_local_addresses(mesh);
-       node_write_config(mesh, mesh->self);
+
+       if(!node_write_config(mesh, mesh->self)) {
+               logger(NULL, MESHLINK_ERROR, "Cannot update configuration\n");
+               return NULL;
+       }
 
        idle_set(&mesh->loop, idle, mesh);
 
@@ -1566,12 +1554,11 @@ void meshlink_stop(meshlink_handle_t *mesh) {
 
        exit_outgoings(mesh);
 
-       // Write out any changed node config files
+       // Try to write out any changed node config files, ignore errors at this point.
        if(mesh->nodes) {
                for splay_each(node_t, n, mesh->nodes) {
                        if(n->status.dirty) {
-                               node_write_config(mesh, n);
-                               n->status.dirty = false;
+                               n->status.dirty = !node_write_config(mesh, n);
                        }
                }
        }
@@ -1641,16 +1628,57 @@ bool meshlink_destroy(const char *confbase) {
                return false;
        }
 
-       if(!config_destroy(confbase, "current")) {
-               logger(NULL, MESHLINK_ERROR, "Cannot remove confbase sub-directories %s: %s\n", confbase, strerror(errno));
+       /* Exit early if the confbase directory itself doesn't exist */
+       if(access(confbase, F_OK) && errno == ENOENT) {
+               return true;
+       }
+
+       /* Take the lock the same way meshlink_open() would. */
+       char lockfilename[PATH_MAX];
+       snprintf(lockfilename, sizeof(lockfilename), "%s" SLASH "meshlink.lock", confbase);
+
+       FILE *lockfile = fopen(lockfilename, "w+");
+
+       if(!lockfile) {
+               logger(NULL, MESHLINK_ERROR, "Could not open lock file %s: %s", lockfilename, strerror(errno));
+               meshlink_errno = MESHLINK_ESTORAGE;
+               return false;
+       }
+
+#ifdef FD_CLOEXEC
+       fcntl(fileno(lockfile), F_SETFD, FD_CLOEXEC);
+#endif
+
+#ifdef HAVE_MINGW
+       // TODO: use _locking()?
+#else
+
+       if(flock(fileno(lockfile), LOCK_EX | LOCK_NB) != 0) {
+               logger(NULL, MESHLINK_ERROR, "Configuration directory %s still in use\n", lockfilename);
+               fclose(lockfile);
+               meshlink_errno = MESHLINK_EBUSY;
+               return false;
+       }
+
+#endif
+
+       if(!config_destroy(confbase, "current") || !config_destroy(confbase, "new") || !config_destroy(confbase, "old")) {
+               logger(NULL, MESHLINK_ERROR, "Cannot remove sub-directories in %s: %s\n", confbase, strerror(errno));
+               return false;
+       }
+
+       if(unlink(lockfilename)) {
+               logger(NULL, MESHLINK_ERROR, "Cannot remove lock file %s: %s\n", lockfilename, strerror(errno));
+               fclose(lockfile);
+               meshlink_errno = MESHLINK_ESTORAGE;
                return false;
        }
 
-       config_destroy(confbase, "new");
-       config_destroy(confbase, "old");
+       fclose(lockfile);
 
-       if(rmdir(confbase) && errno != ENOENT) {
-               logger(NULL, MESHLINK_ERROR, "Cannot remove directory %s: %s\n", confbase, strerror(errno));
+       /* TODO: do we need to remove confbase? Potential race condition? */
+       if(!sync_path(confbase)) {
+               logger(NULL, MESHLINK_ERROR, "Cannot sync directory %s: %s\n", confbase, strerror(errno));
                meshlink_errno = MESHLINK_ESTORAGE;
                return false;
        }
@@ -2144,11 +2172,15 @@ bool meshlink_set_canonical_address(meshlink_handle_t *mesh, meshlink_node_t *no
        node_t *n = (node_t *)node;
        free(n->canonical_address);
        n->canonical_address = canonical_address;
-       node_write_config(mesh, n);
+
+       if(!node_write_config(mesh, n)) {
+               pthread_mutex_unlock(&mesh->mutex);
+               return false;
+       }
 
        pthread_mutex_unlock(&mesh->mutex);
 
-       return true;
+       return config_sync(mesh, "current");
 }
 
 bool meshlink_add_address(meshlink_handle_t *mesh, const char *address) {
@@ -2780,7 +2812,11 @@ bool meshlink_import(meshlink_handle_t *mesh, const char *data) {
                        break;
                }
 
-               config_write(mesh, "current", n->name, &config, mesh->config_key);
+               if(!config_write(mesh, "current", n->name, &config, mesh->config_key)) {
+                       free_node(n);
+                       return false;
+               }
+
                node_add(mesh, n);
        }
 
@@ -2801,10 +2837,10 @@ bool meshlink_import(meshlink_handle_t *mesh, const char *data) {
        return true;
 }
 
-void meshlink_blacklist(meshlink_handle_t *mesh, meshlink_node_t *node) {
+bool meshlink_blacklist(meshlink_handle_t *mesh, meshlink_node_t *node) {
        if(!mesh || !node) {
                meshlink_errno = MESHLINK_EINVAL;
-               return;
+               return false;
        }
 
        pthread_mutex_lock(&mesh->mutex);
@@ -2816,25 +2852,23 @@ void meshlink_blacklist(meshlink_handle_t *mesh, meshlink_node_t *node) {
                logger(mesh, MESHLINK_ERROR, "%s blacklisting itself?\n", node->name);
                meshlink_errno = MESHLINK_EINVAL;
                pthread_mutex_unlock(&mesh->mutex);
-               return;
+               return false;
        }
 
        if(n->status.blacklisted) {
                logger(mesh, MESHLINK_DEBUG, "Node %s already blacklisted\n", node->name);
                pthread_mutex_unlock(&mesh->mutex);
-               return;
+               return true;
        }
 
        n->status.blacklisted = true;
-       node_write_config(mesh, n);
-       config_sync(mesh, "current");
-
-       logger(mesh, MESHLINK_DEBUG, "Blacklisted %s.\n", node->name);
 
-       //Immediately terminate any connections we have with the blacklisted node
+       /* Immediately shut down any connections we have with the blacklisted node.
+        * We can't call terminate_connection(), because we might be called from a callback function.
+        */
        for list_each(connection_t, c, mesh->connections) {
                if(c->node == n) {
-                       terminate_connection(mesh, c, c->status.active);
+                       shutdown(c->socket, SHUT_RDWR);
                }
        }
 
@@ -2846,17 +2880,22 @@ void meshlink_blacklist(meshlink_handle_t *mesh, meshlink_node_t *node) {
        n->mtuprobes = 0;
        n->status.udp_confirmed = false;
 
-       if(n->status.reachable) {
-               update_node_status(mesh, n);
+       if(!node_write_config(mesh, n)) {
+               pthread_mutex_unlock(&mesh->mutex);
+               return false;
        }
 
+       logger(mesh, MESHLINK_DEBUG, "Blacklisted %s.\n", node->name);
+
        pthread_mutex_unlock(&mesh->mutex);
+
+       return config_sync(mesh, "current");
 }
 
-void meshlink_whitelist(meshlink_handle_t *mesh, meshlink_node_t *node) {
+bool meshlink_whitelist(meshlink_handle_t *mesh, meshlink_node_t *node) {
        if(!mesh || !node) {
                meshlink_errno = MESHLINK_EINVAL;
-               return;
+               return false;
        }
 
        pthread_mutex_lock(&mesh->mutex);
@@ -2867,27 +2906,31 @@ void meshlink_whitelist(meshlink_handle_t *mesh, meshlink_node_t *node) {
                logger(mesh, MESHLINK_ERROR, "%s whitelisting itself?\n", node->name);
                meshlink_errno = MESHLINK_EINVAL;
                pthread_mutex_unlock(&mesh->mutex);
-               return;
+               return false;
        }
 
        if(!n->status.blacklisted) {
                logger(mesh, MESHLINK_DEBUG, "Node %s was already whitelisted\n", node->name);
                pthread_mutex_unlock(&mesh->mutex);
-               return;
+               return true;
        }
 
        n->status.blacklisted = false;
-       node_write_config(mesh, n);
-       config_sync(mesh, "current");
 
        if(n->status.reachable) {
                update_node_status(mesh, n);
        }
 
+       if(!node_write_config(mesh, n)) {
+               pthread_mutex_unlock(&mesh->mutex);
+               return false;
+       }
+
        logger(mesh, MESHLINK_DEBUG, "Whitelisted %s.\n", node->name);
 
        pthread_mutex_unlock(&mesh->mutex);
-       return;
+
+       return config_sync(mesh, "current");
 }
 
 void meshlink_set_default_blacklist(meshlink_handle_t *mesh, bool blacklist) {
@@ -2908,7 +2951,10 @@ void meshlink_hint_address(meshlink_handle_t *mesh, meshlink_node_t *node, const
        node_t *n = (node_t *)node;
        memmove(n->recent + 1, n->recent, 4 * sizeof(*n->recent));
        memcpy(n->recent, addr, SALEN(*addr));
-       node_write_config(mesh, n);
+
+       if(!node_write_config(mesh, n)) {
+               logger(mesh, MESHLINK_DEBUG, "Could not update %s\n", n->name);
+       }
 
        pthread_mutex_unlock(&mesh->mutex);
        // @TODO do we want to fire off a connection attempt right away?