]> git.meshlink.io Git - meshlink/blobdiff - src/conf.c
Fix __warn_unused_result__, add more of it and fix the resulting warnings.
[meshlink] / src / conf.c
index bafa87d49c5bd5a563e9d82d5969ca4a51f7ec72..9a1387126e8263d29b42bb0605be746718133bb4 100644 (file)
 
 /// Generate a path to the main configuration file.
 static void make_main_path(meshlink_handle_t *mesh, const char *conf_subdir, char *path, size_t len) {
+       assert(conf_subdir);
+       assert(path);
+       assert(len);
+
        snprintf(path, len, "%s" SLASH "%s" SLASH "meshlink.conf", mesh->confbase, conf_subdir);
 }
 
 /// Generate a path to a host configuration file.
 static void make_host_path(meshlink_handle_t *mesh, const char *conf_subdir, const char *name, char *path, size_t len) {
+       assert(conf_subdir);
+       assert(name);
+       assert(path);
+       assert(len);
+
        snprintf(path, len, "%s" SLASH "%s" SLASH "hosts" SLASH "%s", mesh->confbase, conf_subdir, name);
 }
 
 /// Generate a path to an unused invitation file.
 static void make_invitation_path(meshlink_handle_t *mesh, const char *conf_subdir, const char *name, char *path, size_t len) {
+       assert(conf_subdir);
+       assert(name);
+       assert(path);
+       assert(len);
+
        snprintf(path, len, "%s" SLASH "%s" SLASH "invitations" SLASH "%s", mesh->confbase, conf_subdir, name);
 }
 
 /// Generate a path to a used invitation file.
 static void make_used_invitation_path(meshlink_handle_t *mesh, const char *conf_subdir, const char *name, char *path, size_t len) {
+       assert(conf_subdir);
+       assert(name);
+       assert(path);
+       assert(len);
+
        snprintf(path, len, "%s" SLASH "%s" SLASH "invitations" SLASH "%s.used", mesh->confbase, conf_subdir, name);
 }
 
 /// Remove a directory recursively
 static void deltree(const char *dirname) {
+       assert(dirname);
+
        DIR *d = opendir(dirname);
 
        if(d) {
@@ -75,23 +96,28 @@ static void deltree(const char *dirname) {
        rmdir(dirname);
 }
 
-static bool sync_path(const char *pathname) {
+bool sync_path(const char *pathname) {
+       assert(pathname);
+
        int fd = open(pathname, O_RDONLY);
 
        if(fd < 0) {
                logger(NULL, MESHLINK_ERROR, "Failed to open %s: %s\n", pathname, strerror(errno));
+               meshlink_errno = MESHLINK_ESTORAGE;
                return false;
        }
 
        if(fsync(fd)) {
                logger(NULL, MESHLINK_ERROR, "Failed to sync %s: %s\n", pathname, strerror(errno));
                close(fd);
+               meshlink_errno = MESHLINK_ESTORAGE;
                return false;
        }
 
        if(close(fd)) {
                logger(NULL, MESHLINK_ERROR, "Failed to close %s: %s\n", pathname, strerror(errno));
                close(fd);
+               meshlink_errno = MESHLINK_ESTORAGE;
                return false;
        }
 
@@ -100,9 +126,9 @@ static bool sync_path(const char *pathname) {
 
 /// Try decrypting the main configuration file from the given sub-directory.
 static bool main_config_decrypt(meshlink_handle_t *mesh, const char *conf_subdir) {
-       if(!mesh->config_key && !mesh->confbase && !conf_subdir) {
-               return false;
-       }
+       assert(mesh->config_key);
+       assert(mesh->confbase);
+       assert(conf_subdir);
 
        config_t config;
 
@@ -121,19 +147,12 @@ static bool main_config_decrypt(meshlink_handle_t *mesh, const char *conf_subdir
 
 /// Create a fresh configuration directory
 bool config_init(meshlink_handle_t *mesh, const char *conf_subdir) {
+       assert(conf_subdir);
+
        if(!mesh->confbase) {
                return true;
        }
 
-       if(!conf_subdir) {
-               return false;
-       }
-
-       if(mkdir(mesh->confbase, 0700) && errno != EEXIST) {
-               logger(mesh, MESHLINK_DEBUG, "Could not create directory %s: %s\n", mesh->confbase, strerror(errno));
-               return false;
-       }
-
        char path[PATH_MAX];
 
        // Create "current" sub-directory in the confbase
@@ -164,8 +183,10 @@ bool config_init(meshlink_handle_t *mesh, const char *conf_subdir) {
 
 /// Wipe an existing configuration directory
 bool config_destroy(const char *confbase, const char *conf_subdir) {
-       if(!confbase && !conf_subdir) {
-               return false;
+       assert(conf_subdir);
+
+       if(!confbase) {
+               return true;
        }
 
        struct stat st;
@@ -202,9 +223,8 @@ bool config_destroy(const char *confbase, const char *conf_subdir) {
 }
 
 static bool copytree(const char *src_dir_name, const void *src_key, const char *dst_dir_name, const void *dst_key) {
-       if(!src_dir_name || !dst_dir_name) {
-               return false;
-       }
+       assert(src_dir_name);
+       assert(dst_dir_name);
 
        char src_filename[PATH_MAX];
        char dst_filename[PATH_MAX];
@@ -309,6 +329,9 @@ static bool copytree(const char *src_dir_name, const void *src_key, const char *
 }
 
 bool config_copy(meshlink_handle_t *mesh, const char *src_dir_name, const void *src_key, const char *dst_dir_name, const void *dst_key) {
+       assert(src_dir_name);
+       assert(dst_dir_name);
+
        char src_filename[PATH_MAX];
        char dst_filename[PATH_MAX];
 
@@ -320,7 +343,9 @@ bool config_copy(meshlink_handle_t *mesh, const char *src_dir_name, const void *
 
 /// Check the presence of the main configuration file.
 bool main_config_exists(meshlink_handle_t *mesh, const char *conf_subdir) {
-       if(!mesh->confbase && !conf_subdir) {
+       assert(conf_subdir);
+
+       if(!mesh->confbase) {
                return false;
        }
 
@@ -330,7 +355,10 @@ bool main_config_exists(meshlink_handle_t *mesh, const char *conf_subdir) {
 }
 
 bool config_rename(meshlink_handle_t *mesh, const char *old_conf_subdir, const char *new_conf_subdir) {
-       if(!mesh->confbase && !old_conf_subdir && !new_conf_subdir) {
+       assert(old_conf_subdir);
+       assert(new_conf_subdir);
+
+       if(!mesh->confbase) {
                return false;
        }
 
@@ -343,6 +371,29 @@ bool config_rename(meshlink_handle_t *mesh, const char *old_conf_subdir, const c
        return rename(old_path, new_path) == 0;
 }
 
+bool config_sync(meshlink_handle_t *mesh, const char *conf_subdir) {
+       assert(conf_subdir);
+
+       if(!mesh->confbase) {
+               return true;
+       }
+
+       char path[PATH_MAX];
+       snprintf(path, sizeof(path), "%s" SLASH "%s" SLASH "hosts", mesh->confbase, conf_subdir);
+
+       if(!sync_path(path)) {
+               return false;
+       }
+
+       snprintf(path, sizeof(path), "%s" SLASH "%s", mesh->confbase, conf_subdir);
+
+       if(!sync_path(path)) {
+               return false;
+       }
+
+       return true;
+}
+
 bool meshlink_confbase_exists(meshlink_handle_t *mesh) {
        if(!mesh->confbase) {
                return false;
@@ -393,42 +444,50 @@ bool meshlink_confbase_exists(meshlink_handle_t *mesh) {
 
        // Cleanup if current is existing with old and new
        if(confbase_exists && confbase_decryptable) {
-               config_destroy(mesh->confbase, "old");
-               config_destroy(mesh->confbase, "new");
+               if(!config_destroy(mesh->confbase, "old") || !config_destroy(mesh->confbase, "new")) {
+                       return false;
+               }
        }
 
        return confbase_exists;
 }
 
-/// Lock the main configuration file.
+/// Lock the main configuration file. Creates confbase if necessary.
 bool main_config_lock(meshlink_handle_t *mesh) {
        if(!mesh->confbase) {
                return true;
        }
 
+       if(mkdir(mesh->confbase, 0700) && errno != EEXIST) {
+               logger(NULL, MESHLINK_ERROR, "Cannot create configuration directory %s: %s", mesh->confbase, strerror(errno));
+               meshlink_close(mesh);
+               meshlink_errno = MESHLINK_ESTORAGE;
+               return NULL;
+       }
+
        char path[PATH_MAX];
-       make_main_path(mesh, "current", path, sizeof(path));
+       snprintf(path, sizeof(path), "%s" SLASH "meshlink.lock", mesh->confbase);
 
-       mesh->conffile = fopen(path, "r");
+       mesh->lockfile = fopen(path, "w+");
 
-       if(!mesh->conffile) {
+       if(!mesh->lockfile) {
                logger(NULL, MESHLINK_ERROR, "Cannot not open %s: %s\n", path, strerror(errno));
                meshlink_errno = MESHLINK_ESTORAGE;
                return false;
        }
 
 #ifdef FD_CLOEXEC
-       fcntl(fileno(mesh->conffile), F_SETFD, FD_CLOEXEC);
+       fcntl(fileno(mesh->lockfile), F_SETFD, FD_CLOEXEC);
 #endif
 
 #ifdef HAVE_MINGW
        // TODO: use _locking()?
 #else
 
-       if(flock(fileno(mesh->conffile), LOCK_EX | LOCK_NB) != 0) {
+       if(flock(fileno(mesh->lockfile), LOCK_EX | LOCK_NB) != 0) {
                logger(NULL, MESHLINK_ERROR, "Cannot lock %s: %s\n", path, strerror(errno));
-               fclose(mesh->conffile);
-               mesh->conffile = NULL;
+               fclose(mesh->lockfile);
+               mesh->lockfile = NULL;
                meshlink_errno = MESHLINK_EBUSY;
                return false;
        }
@@ -440,20 +499,21 @@ bool main_config_lock(meshlink_handle_t *mesh) {
 
 /// Unlock the main configuration file.
 void main_config_unlock(meshlink_handle_t *mesh) {
-       if(mesh->conffile) {
-               fclose(mesh->conffile);
-               mesh->conffile = NULL;
+       if(mesh->lockfile) {
+               fclose(mesh->lockfile);
+               mesh->lockfile = NULL;
        }
 }
 
 /// Read a configuration file from a FILE handle.
 bool config_read_file(meshlink_handle_t *mesh, FILE *f, config_t *config, const void *key) {
+       assert(f);
+
        long len;
 
        if(fseek(f, 0, SEEK_END) || !(len = ftell(f)) || fseek(f, 0, SEEK_SET)) {
                logger(mesh, MESHLINK_ERROR, "Cannot get config file size: %s\n", strerror(errno));
                meshlink_errno = MESHLINK_ESTORAGE;
-               fclose(f);
                return false;
        }
 
@@ -462,7 +522,6 @@ bool config_read_file(meshlink_handle_t *mesh, FILE *f, config_t *config, const
        if(fread(buf, len, 1, f) != 1) {
                logger(mesh, MESHLINK_ERROR, "Cannot read config file: %s\n", strerror(errno));
                meshlink_errno = MESHLINK_ESTORAGE;
-               fclose(f);
                return false;
        }
 
@@ -496,6 +555,8 @@ bool config_read_file(meshlink_handle_t *mesh, FILE *f, config_t *config, const
 
 /// Write a configuration file to a FILE handle.
 bool config_write_file(meshlink_handle_t *mesh, FILE *f, const config_t *config, const void *key) {
+       assert(f);
+
        if(key) {
                uint8_t buf[config->len + 16];
                size_t len = sizeof(buf);
@@ -507,6 +568,12 @@ bool config_write_file(meshlink_handle_t *mesh, FILE *f, const config_t *config,
 
                if(chacha_poly1305_encrypt_iv96(ctx, seqbuf, config->buf, config->len, buf, &len)) {
                        success = fwrite(seqbuf, sizeof(seqbuf), 1, f) == 1 && fwrite(buf, len, 1, f) == 1;
+
+                       if(!success) {
+                               logger(mesh, MESHLINK_ERROR, "Cannot write config file: %s", strerror(errno));
+                       }
+
+                       meshlink_errno = MESHLINK_ESTORAGE;
                } else {
                        logger(mesh, MESHLINK_ERROR, "Cannot encrypt config file\n");
                        meshlink_errno = MESHLINK_ESTORAGE;
@@ -522,8 +589,15 @@ bool config_write_file(meshlink_handle_t *mesh, FILE *f, const config_t *config,
                return false;
        }
 
+       if(fflush(f)) {
+               logger(mesh, MESHLINK_ERROR, "Failed to flush file: %s", strerror(errno));
+               meshlink_errno = MESHLINK_ESTORAGE;
+               return false;
+       }
+
        if(fsync(fileno(f))) {
                logger(mesh, MESHLINK_ERROR, "Failed to sync file: %s\n", strerror(errno));
+               meshlink_errno = MESHLINK_ESTORAGE;
                return false;
        }
 
@@ -532,6 +606,8 @@ bool config_write_file(meshlink_handle_t *mesh, FILE *f, const config_t *config,
 
 /// Free resources of a loaded configuration file.
 void config_free(config_t *config) {
+       assert(!config->len || config->buf);
+
        free((uint8_t *)config->buf);
        config->buf = NULL;
        config->len = 0;
@@ -539,7 +615,9 @@ void config_free(config_t *config) {
 
 /// Check the presence of a host configuration file.
 bool config_exists(meshlink_handle_t *mesh, const char *conf_subdir, const char *name) {
-       if(!mesh->confbase && !conf_subdir) {
+       assert(conf_subdir);
+
+       if(!mesh->confbase) {
                return false;
        }
 
@@ -551,7 +629,9 @@ bool config_exists(meshlink_handle_t *mesh, const char *conf_subdir, const char
 
 /// Read a host configuration file.
 bool config_read(meshlink_handle_t *mesh, const char *conf_subdir, const char *name, config_t *config, void *key) {
-       if(!mesh->confbase && !conf_subdir) {
+       assert(conf_subdir);
+
+       if(!mesh->confbase) {
                return false;
        }
 
@@ -577,8 +657,11 @@ bool config_read(meshlink_handle_t *mesh, const char *conf_subdir, const char *n
 }
 
 bool config_scan_all(meshlink_handle_t *mesh, const char *conf_subdir, const char *conf_type, config_scan_action_t action, void *arg) {
-       if(!mesh->confbase && !conf_subdir && !conf_type) {
-               return false;
+       assert(conf_subdir);
+       assert(conf_type);
+
+       if(!mesh->confbase) {
+               return true;
        }
 
        DIR *dir;
@@ -611,7 +694,11 @@ bool config_scan_all(meshlink_handle_t *mesh, const char *conf_subdir, const cha
 
 /// Write a host configuration file.
 bool config_write(meshlink_handle_t *mesh, const char *conf_subdir, const char *name, const config_t *config, void *key) {
-       if(!mesh->confbase && !conf_subdir && !name) {
+       assert(conf_subdir);
+       assert(name);
+       assert(config);
+
+       if(!mesh->confbase) {
                return true;
        }
 
@@ -624,6 +711,7 @@ bool config_write(meshlink_handle_t *mesh, const char *conf_subdir, const char *
 
        if(!f) {
                logger(mesh, MESHLINK_ERROR, "Failed to open `%s': %s", tmp_path, strerror(errno));
+               meshlink_errno = MESHLINK_ESTORAGE;
                return false;
        }
 
@@ -633,19 +721,15 @@ bool config_write(meshlink_handle_t *mesh, const char *conf_subdir, const char *
                return false;
        }
 
-       if(fsync(fileno(f))) {
-               logger(mesh, MESHLINK_ERROR, "Failed to sync `%s': %s", tmp_path, strerror(errno));
-               fclose(f);
-               return false;
-       }
-
        if(fclose(f)) {
                logger(mesh, MESHLINK_ERROR, "Failed to close `%s': %s", tmp_path, strerror(errno));
+               meshlink_errno = MESHLINK_ESTORAGE;
                return false;
        }
 
        if(rename(tmp_path, path)) {
                logger(mesh, MESHLINK_ERROR, "Failed to rename `%s' to `%s': %s", tmp_path, path, strerror(errno));
+               meshlink_errno = MESHLINK_ESTORAGE;
                return false;
        }
 
@@ -654,7 +738,10 @@ bool config_write(meshlink_handle_t *mesh, const char *conf_subdir, const char *
 
 /// Read the main configuration file.
 bool main_config_read(meshlink_handle_t *mesh, const char *conf_subdir, config_t *config, void *key) {
-       if(!mesh->confbase && !conf_subdir) {
+       assert(conf_subdir);
+       assert(config);
+
+       if(!mesh->confbase) {
                return false;
        }
 
@@ -681,7 +768,10 @@ bool main_config_read(meshlink_handle_t *mesh, const char *conf_subdir, config_t
 
 /// Write the main configuration file.
 bool main_config_write(meshlink_handle_t *mesh, const char *conf_subdir, const config_t *config, void *key) {
-       if(!mesh->confbase && !conf_subdir) {
+       assert(conf_subdir);
+       assert(config);
+
+       if(!mesh->confbase) {
                return true;
        }
 
@@ -694,6 +784,7 @@ bool main_config_write(meshlink_handle_t *mesh, const char *conf_subdir, const c
 
        if(!f) {
                logger(mesh, MESHLINK_ERROR, "Failed to open `%s': %s", tmp_path, strerror(errno));
+               meshlink_errno = MESHLINK_ESTORAGE;
                return false;
        }
 
@@ -703,19 +794,16 @@ bool main_config_write(meshlink_handle_t *mesh, const char *conf_subdir, const c
                return false;
        }
 
-       if(fsync(fileno(f))) {
-               logger(mesh, MESHLINK_ERROR, "Failed to sync `%s': %s", tmp_path, strerror(errno));
-               fclose(f);
-               return false;
-       }
-
        if(rename(tmp_path, path)) {
                logger(mesh, MESHLINK_ERROR, "Failed to rename `%s' to `%s': %s", tmp_path, path, strerror(errno));
+               meshlink_errno = MESHLINK_ESTORAGE;
+               fclose(f);
                return false;
        }
 
        if(fclose(f)) {
                logger(mesh, MESHLINK_ERROR, "Failed to close `%s': %s", tmp_path, strerror(errno));
+               meshlink_errno = MESHLINK_ESTORAGE;
                return false;
        }
 
@@ -724,7 +812,11 @@ bool main_config_write(meshlink_handle_t *mesh, const char *conf_subdir, const c
 
 /// Read an invitation file from the confbase sub-directory, and immediately delete it.
 bool invitation_read(meshlink_handle_t *mesh, const char *conf_subdir, const char *name, config_t *config, void *key) {
-       if(!mesh->confbase && !conf_subdir) {
+       assert(conf_subdir);
+       assert(name);
+       assert(config);
+
+       if(!mesh->confbase) {
                return false;
        }
 
@@ -761,7 +853,7 @@ bool invitation_read(meshlink_handle_t *mesh, const char *conf_subdir, const cha
                return false;
        }
 
-       if(mesh->loop.now.tv_sec > st.st_mtime + mesh->invitation_timeout) {
+       if(mesh->loop.now.tv_sec >= st.st_mtime + mesh->invitation_timeout) {
                logger(mesh, MESHLINK_ERROR, "Peer tried to use an outdated invitation file %s\n", name);
                fclose(f);
                unlink(used_path);
@@ -777,13 +869,29 @@ bool invitation_read(meshlink_handle_t *mesh, const char *conf_subdir, const cha
 
        fclose(f);
 
-       unlink(used_path);
+       if(unlink(used_path)) {
+               logger(mesh, MESHLINK_ERROR, "Failed to unlink `%s': %s", path, strerror(errno));
+               return false;
+       }
+
+       snprintf(path, sizeof(path), "%s" SLASH "%s" SLASH "invitations", mesh->confbase, conf_subdir);
+
+       if(!sync_path(path)) {
+               logger(mesh, MESHLINK_ERROR, "Failed to sync `%s': %s", path, strerror(errno));
+               meshlink_errno = MESHLINK_ESTORAGE;
+               return false;
+       }
+
        return true;
 }
 
 /// Write an invitation file.
 bool invitation_write(meshlink_handle_t *mesh, const char *conf_subdir, const char *name, const config_t *config, void *key) {
-       if(!mesh->confbase && !conf_subdir) {
+       assert(conf_subdir);
+       assert(name);
+       assert(config);
+
+       if(!mesh->confbase) {
                return false;
        }
 
@@ -794,6 +902,7 @@ bool invitation_write(meshlink_handle_t *mesh, const char *conf_subdir, const ch
 
        if(!f) {
                logger(mesh, MESHLINK_ERROR, "Failed to open `%s': %s", path, strerror(errno));
+               meshlink_errno = MESHLINK_ESTORAGE;
                return false;
        }
 
@@ -803,13 +912,17 @@ bool invitation_write(meshlink_handle_t *mesh, const char *conf_subdir, const ch
                return false;
        }
 
-       if(fsync(fileno(f))) {
-               logger(mesh, MESHLINK_ERROR, "Failed to sync `%s': %s", path, strerror(errno));
+       if(fclose(f)) {
+               logger(mesh, MESHLINK_ERROR, "Failed to close `%s': %s", path, strerror(errno));
+               meshlink_errno = MESHLINK_ESTORAGE;
                return false;
        }
 
-       if(fclose(f)) {
-               logger(mesh, MESHLINK_ERROR, "Failed to close `%s': %s", path, strerror(errno));
+       snprintf(path, sizeof(path), "%s" SLASH "%s" SLASH "invitations", mesh->confbase, conf_subdir);
+
+       if(!sync_path(path)) {
+               logger(mesh, MESHLINK_ERROR, "Failed to sync `%s': %s", path, strerror(errno));
+               meshlink_errno = MESHLINK_ESTORAGE;
                return false;
        }