X-Git-Url: http://git.meshlink.io/?p=meshlink;a=blobdiff_plain;f=src%2Fconf.c;h=65ff2272ef68e50e293ad23cda5d4045315b3d7e;hp=2195d69c42ff32230e85f54c67bae00aeac9a674;hb=a5a8005b2d89712e124ab7295165a3e229abdad5;hpb=06e5c20b092578b496593c4e2934ee726cb12469 diff --git a/src/conf.c b/src/conf.c index 2195d69c..65ff2272 100644 --- a/src/conf.c +++ b/src/conf.c @@ -1,6 +1,6 @@ /* - conf.c -- configuration code - Copyright (C) 2014 Guus Sliepen + econf.c -- configuration code + Copyright (C) 2018 Guus Sliepen This program is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by @@ -18,469 +18,945 @@ */ #include "system.h" +#include +#include +#include -#include "splay_tree.h" -#include "connection.h" #include "conf.h" -#include "list.h" +#include "crypto.h" #include "logger.h" #include "meshlink_internal.h" -#include "netutl.h" /* for str2address */ -#include "protocol.h" -#include "utils.h" /* for cp */ #include "xalloc.h" +#include "packmsg.h" -static int config_compare(const config_t *a, const config_t *b) { - int result; +/// 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); - result = strcasecmp(a->variable, b->variable); - - if(result) - return result; + snprintf(path, len, "%s" SLASH "%s" SLASH "meshlink.conf", mesh->confbase, conf_subdir); +} - result = a->line - b->line; +/// 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); - if(result) - return result; - else - return a->file ? strcmp(a->file, b->file) : 0; + snprintf(path, len, "%s" SLASH "%s" SLASH "hosts" SLASH "%s", mesh->confbase, conf_subdir, name); } -void init_configuration(splay_tree_t **config_tree) { - *config_tree = splay_alloc_tree((splay_compare_t) config_compare, (splay_action_t) free_config); -} +/// 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); -void exit_configuration(splay_tree_t **config_tree) { - if(*config_tree) - splay_delete_tree(*config_tree); - *config_tree = NULL; + snprintf(path, len, "%s" SLASH "%s" SLASH "invitations" SLASH "%s", mesh->confbase, conf_subdir, name); } -config_t *new_config(void) { - return xzalloc(sizeof(config_t)); +/// 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); } -void free_config(config_t *cfg) { - if(cfg->variable) - free(cfg->variable); +/// Remove a directory recursively +static void deltree(const char *dirname) { + assert(dirname); - if(cfg->value) - free(cfg->value); + DIR *d = opendir(dirname); - if(cfg->file) - free(cfg->file); + if(d) { + struct dirent *ent; - free(cfg); -} + while((ent = readdir(d))) { + if(ent->d_name[0] == '.') { + continue; + } + + char filename[PATH_MAX]; + snprintf(filename, sizeof(filename), "%s" SLASH "%s", dirname, ent->d_name); -void config_add(splay_tree_t *config_tree, config_t *cfg) { - splay_insert(config_tree, cfg); + if(unlink(filename)) { + deltree(filename); + } + } + + closedir(d); + } + + rmdir(dirname); } -config_t *lookup_config(splay_tree_t *config_tree, char *variable) { - config_t cfg, *found; +static bool sync_path(const char *pathname) { + assert(pathname); - cfg.variable = variable; - cfg.file = NULL; - cfg.line = 0; + int fd = open(pathname, O_RDONLY); - found = splay_search_closest_greater(config_tree, &cfg); + if(fd < 0) { + logger(NULL, MESHLINK_ERROR, "Failed to open %s: %s\n", pathname, strerror(errno)); + meshlink_errno = MESHLINK_ESTORAGE; + return false; + } - if(!found) - return NULL; + 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(strcasecmp(found->variable, variable)) - return NULL; + if(close(fd)) { + logger(NULL, MESHLINK_ERROR, "Failed to close %s: %s\n", pathname, strerror(errno)); + close(fd); + meshlink_errno = MESHLINK_ESTORAGE; + return false; + } - return found; + return true; } -config_t *lookup_config_next(splay_tree_t *config_tree, const config_t *cfg) { - splay_node_t *node; - config_t *found; - - node = splay_search_node(config_tree, cfg); +/// Try decrypting the main configuration file from the given sub-directory. +static bool main_config_decrypt(meshlink_handle_t *mesh, const char *conf_subdir) { + assert(mesh->config_key); + assert(mesh->confbase); + assert(conf_subdir); - if(node) { - if(node->next) { - found = node->next->data; + config_t config; - if(!strcasecmp(found->variable, cfg->variable)) - return found; - } + if(!main_config_read(mesh, conf_subdir, &config, mesh->config_key)) { + logger(mesh, MESHLINK_ERROR, "Could not read main configuration file"); + return false; } - return NULL; + packmsg_input_t in = {config.buf, config.len}; + + uint32_t version = packmsg_get_uint32(&in); + config_free(&config); + + return version == MESHLINK_CONFIG_VERSION; } -bool get_config_bool(const config_t *cfg, bool *result) { - if(!cfg) - return false; +/// Create a fresh configuration directory +bool config_init(meshlink_handle_t *mesh, const char *conf_subdir) { + assert(conf_subdir); - if(!strcasecmp(cfg->value, "yes")) { - *result = true; - return true; - } else if(!strcasecmp(cfg->value, "no")) { - *result = false; + if(!mesh->confbase) { return true; } - logger(NULL, MESHLINK_ERROR, "\"yes\" or \"no\" expected for configuration variable %s in %s line %d", - cfg->variable, cfg->file, cfg->line); + if(mkdir(mesh->confbase, 0700) && errno != EEXIST) { + logger(mesh, MESHLINK_DEBUG, "Could not create directory %s: %s\n", mesh->confbase, strerror(errno)); + return false; + } - return false; -} + char path[PATH_MAX]; + + // Create "current" sub-directory in the confbase + snprintf(path, sizeof(path), "%s" SLASH "%s", mesh->confbase, conf_subdir); + deltree(path); -bool get_config_int(const config_t *cfg, int *result) { - if(!cfg) + if(mkdir(path, 0700)) { + logger(mesh, MESHLINK_DEBUG, "Could not create directory %s: %s\n", path, strerror(errno)); return false; + } - if(sscanf(cfg->value, "%d", result) == 1) - return true; + make_host_path(mesh, conf_subdir, "", path, sizeof(path)); - logger(NULL, MESHLINK_ERROR, "Integer expected for configuration variable %s in %s line %d", - cfg->variable, cfg->file, cfg->line); + if(mkdir(path, 0700)) { + logger(mesh, MESHLINK_DEBUG, "Could not create directory %s: %s\n", path, strerror(errno)); + return false; + } + + make_invitation_path(mesh, conf_subdir, "", path, sizeof(path)); + + if(mkdir(path, 0700)) { + logger(mesh, MESHLINK_DEBUG, "Could not create directory %s: %s\n", path, strerror(errno)); + return false; + } - return false; + return true; } -bool set_config_int(config_t *cfg, int val) -{ - if(!cfg) +/// Wipe an existing configuration directory +bool config_destroy(const char *confbase, const char *conf_subdir) { + assert(conf_subdir); + + if(!confbase) { return false; + } + + struct stat st; - char val_str[1024]; - snprintf(val_str, sizeof(val_str), "%d", val); + char path[PATH_MAX]; - if(cfg->value) - free(cfg->value); + // Check the presence of configuration base sub directory. + snprintf(path, sizeof(path), "%s" SLASH "%s", confbase, conf_subdir); + + if(stat(path, &st)) { + if(errno == ENOENT) { + return true; + } else { + logger(NULL, MESHLINK_ERROR, "Cannot stat %s: %s\n", path, strerror(errno)); + meshlink_errno = MESHLINK_ESTORAGE; + return false; + } + } - cfg->value = xstrdup(val_str); + // Remove meshlink.conf + snprintf(path, sizeof(path), "%s" SLASH "%s" SLASH "meshlink.conf", confbase, conf_subdir); + if(unlink(path)) { + if(errno != ENOENT) { + logger(NULL, MESHLINK_ERROR, "Cannot delete %s: %s\n", path, strerror(errno)); + meshlink_errno = MESHLINK_ESTORAGE; + return false; + } + } + + snprintf(path, sizeof(path), "%s" SLASH "%s", confbase, conf_subdir); + deltree(path); return true; } -bool get_config_string(const config_t *cfg, char **result) { - if(!cfg) +static bool copytree(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]; + struct dirent *ent; + + DIR *src_dir = opendir(src_dir_name); + + if(!src_dir) { + logger(NULL, MESHLINK_ERROR, "Could not open directory file %s\n", src_dir_name); + return false; + } + + // Delete if already exists and create a new destination directory + deltree(dst_dir_name); + + if(mkdir(dst_dir_name, 0700)) { + logger(NULL, MESHLINK_ERROR, "Could not create directory %s\n", dst_filename); return false; + } + + while((ent = readdir(src_dir))) { + if(ent->d_name[0] == '.') { + continue; + } + + snprintf(dst_filename, sizeof(dst_filename), "%s" SLASH "%s", dst_dir_name, ent->d_name); + snprintf(src_filename, sizeof(src_filename), "%s" SLASH "%s", src_dir_name, ent->d_name); + + if(ent->d_type == DT_DIR) { + if(!copytree(src_filename, src_key, dst_filename, dst_key)) { + logger(NULL, MESHLINK_ERROR, "Copying %s to %s failed\n", src_filename, dst_filename); + return false; + } + + if(!sync_path(dst_filename)) { + return false; + } + } else if(ent->d_type == DT_REG) { + struct stat st; + config_t config; + + if(stat(src_filename, &st)) { + logger(NULL, MESHLINK_ERROR, "Could not stat file `%s': %s\n", src_filename, strerror(errno)); + return false; + } + + FILE *f = fopen(src_filename, "r"); + + if(!f) { + logger(NULL, MESHLINK_ERROR, "Failed to open `%s': %s\n", src_filename, strerror(errno)); + return false; + } + + if(!config_read_file(NULL, f, &config, src_key)) { + logger(NULL, MESHLINK_ERROR, "Failed to read `%s': %s\n", src_filename, strerror(errno)); + fclose(f); + return false; + } + + if(fclose(f)) { + logger(NULL, MESHLINK_ERROR, "Failed to close `%s': %s\n", src_filename, strerror(errno)); + config_free(&config); + return false; + } + + f = fopen(dst_filename, "w"); + + if(!f) { + logger(NULL, MESHLINK_ERROR, "Failed to open `%s': %s", dst_filename, strerror(errno)); + config_free(&config); + return false; + } + + if(!config_write_file(NULL, f, &config, dst_key)) { + logger(NULL, MESHLINK_ERROR, "Failed to write `%s': %s", dst_filename, strerror(errno)); + config_free(&config); + fclose(f); + return false; + } + + if(fclose(f)) { + logger(NULL, MESHLINK_ERROR, "Failed to close `%s': %s", dst_filename, strerror(errno)); + config_free(&config); + return false; + } - *result = xstrdup(cfg->value); + config_free(&config); + struct utimbuf times; + times.modtime = st.st_mtime; + times.actime = st.st_atime; + + if(utime(dst_filename, ×)) { + logger(NULL, MESHLINK_ERROR, "Failed to utime `%s': %s", dst_filename, strerror(errno)); + return false; + } + } + } + + closedir(src_dir); return true; } -bool set_config_string(config_t *cfg, const char* val) -{ - if(!cfg) - return false; +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); - if(cfg->value) - free(cfg->value); + char src_filename[PATH_MAX]; + char dst_filename[PATH_MAX]; - cfg->value = xstrdup(val); + snprintf(dst_filename, sizeof(dst_filename), "%s" SLASH "%s", mesh->confbase, dst_dir_name); + snprintf(src_filename, sizeof(src_filename), "%s" SLASH "%s", mesh->confbase, src_dir_name); - return true; + return copytree(src_filename, src_key, dst_filename, dst_key); +} + +/// Check the presence of the main configuration file. +bool main_config_exists(meshlink_handle_t *mesh, const char *conf_subdir) { + assert(conf_subdir); + + if(!mesh->confbase) { + return false; + } + + char path[PATH_MAX]; + make_main_path(mesh, conf_subdir, path, sizeof(path)); + return access(path, F_OK) == 0; } -bool get_config_address(const config_t *cfg, struct addrinfo **result) { - struct addrinfo *ai; +bool config_rename(meshlink_handle_t *mesh, const char *old_conf_subdir, const char *new_conf_subdir) { + assert(old_conf_subdir); + assert(new_conf_subdir); - if(!cfg) + if(!mesh->confbase) { return false; + } + + char old_path[PATH_MAX]; + char new_path[PATH_MAX]; + + snprintf(old_path, sizeof(old_path), "%s" SLASH "%s", mesh->confbase, old_conf_subdir); + snprintf(new_path, sizeof(new_path), "%s" SLASH "%s", mesh->confbase, new_conf_subdir); - ai = str2addrinfo(cfg->value, NULL, 0); + return rename(old_path, new_path) == 0; +} + +bool config_sync(meshlink_handle_t *mesh, const char *conf_subdir) { + assert(conf_subdir); - if(ai) { - *result = ai; + if(!mesh->confbase) { return true; } - logger(NULL, MESHLINK_ERROR, "Hostname or IP address expected for configuration variable %s in %s line %d", - cfg->variable, cfg->file, cfg->line); + char path[PATH_MAX]; + snprintf(path, sizeof(path), "%s" SLASH "%s" SLASH "hosts", mesh->confbase, conf_subdir); + + if(!sync_path(path)) { + return false; + } - return false; + snprintf(path, sizeof(path), "%s" SLASH "%s", mesh->confbase, conf_subdir); + + if(!sync_path(path)) { + return false; + } + + return true; } -/* - Read exactly one line and strip the trailing newline if any. -*/ -static char *readline(FILE * fp, char *buf, size_t buflen) { - char *newline = NULL; - char *p; +bool meshlink_confbase_exists(meshlink_handle_t *mesh) { + if(!mesh->confbase) { + return false; + } - if(feof(fp)) - return NULL; + bool confbase_exists = false; + bool confbase_decryptable = false; - p = fgets(buf, buflen, fp); + if(main_config_exists(mesh, "current")) { + confbase_exists = true; - if(!p) - return NULL; + if(mesh->config_key && main_config_decrypt(mesh, "current")) { + confbase_decryptable = true; + } + } + + if(mesh->config_key && !confbase_decryptable && main_config_exists(mesh, "new")) { + confbase_exists = true; + + if(main_config_decrypt(mesh, "new")) { + if(!config_destroy(mesh->confbase, "current")) { + return false; + } - newline = strchr(p, '\n'); + if(!config_rename(mesh, "new", "current")) { + return false; + } - if(!newline) - return buf; + confbase_decryptable = true; + } + } - /* kill newline and carriage return if necessary */ - *newline = '\0'; - if(newline > p && newline[-1] == '\r') - newline[-1] = '\0'; + if(mesh->config_key && !confbase_decryptable && main_config_exists(mesh, "old")) { + confbase_exists = true; + + if(main_config_decrypt(mesh, "old")) { + if(!config_destroy(mesh->confbase, "current")) { + return false; + } - return buf; + if(!config_rename(mesh, "old", "current")) { + return false; + } + + confbase_decryptable = true; + } + } + + // Cleanup if current is existing with old and new + if(confbase_exists && confbase_decryptable) { + config_destroy(mesh->confbase, "old"); + config_destroy(mesh->confbase, "new"); + } + + return confbase_exists; } -config_t *parse_config_line(char *line, const char *fname, int lineno) { - config_t *cfg; - int len; - char *variable, *value, *eol; - variable = value = line; +/// Lock the main configuration file. +bool main_config_lock(meshlink_handle_t *mesh) { + if(!mesh->confbase) { + return true; + } + + char path[PATH_MAX]; + make_main_path(mesh, "current", path, sizeof(path)); - eol = line + strlen(line); - while(strchr("\t ", *--eol)) - *eol = '\0'; + mesh->conffile = fopen(path, "r"); - len = strcspn(value, "\t ="); - value += len; - value += strspn(value, "\t "); - if(*value == '=') { - value++; - value += strspn(value, "\t "); + if(!mesh->conffile) { + logger(NULL, MESHLINK_ERROR, "Cannot not open %s: %s\n", path, strerror(errno)); + meshlink_errno = MESHLINK_ESTORAGE; + return false; } - variable[len] = '\0'; - if(!*value) { - const char err[] = "No value for variable"; - logger(NULL, MESHLINK_ERROR, "%s `%s' on line %d while reading config file %s", - err, variable, lineno, fname); - return NULL; +#ifdef FD_CLOEXEC + fcntl(fileno(mesh->conffile), F_SETFD, FD_CLOEXEC); +#endif + +#ifdef HAVE_MINGW + // TODO: use _locking()? +#else + + if(flock(fileno(mesh->conffile), LOCK_EX | LOCK_NB) != 0) { + logger(NULL, MESHLINK_ERROR, "Cannot lock %s: %s\n", path, strerror(errno)); + fclose(mesh->conffile); + mesh->conffile = NULL; + meshlink_errno = MESHLINK_EBUSY; + return false; } - cfg = new_config(); - cfg->variable = xstrdup(variable); - cfg->value = xstrdup(value); - cfg->file = xstrdup(fname); - cfg->line = lineno; +#endif - return cfg; + return true; } -/* - Parse a configuration file and put the results in the configuration tree - starting at *base. -*/ -bool read_config_file(splay_tree_t *config_tree, const char *fname) { - FILE *fp; - char buffer[MAX_STRING_SIZE]; - char *line; - int lineno = 0; - bool ignore = false; - config_t *cfg; - bool result = false; +/// Unlock the main configuration file. +void main_config_unlock(meshlink_handle_t *mesh) { + if(mesh->conffile) { + fclose(mesh->conffile); + mesh->conffile = 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); - fp = fopen(fname, "r"); + long len; - if(!fp) { - logger(NULL, MESHLINK_ERROR, "Cannot open config file %s: %s", fname, strerror(errno)); + 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; return false; } - for(;;) { - line = readline(fp, buffer, sizeof buffer); + uint8_t *buf = xmalloc(len); - if(!line) { - if(feof(fp)) - result = true; - break; + if(fread(buf, len, 1, f) != 1) { + logger(mesh, MESHLINK_ERROR, "Cannot read config file: %s\n", strerror(errno)); + meshlink_errno = MESHLINK_ESTORAGE; + return false; + } + + if(key) { + uint8_t *decrypted = xmalloc(len); + size_t decrypted_len = len; + chacha_poly1305_ctx_t *ctx = chacha_poly1305_init(); + chacha_poly1305_set_key(ctx, key); + + if(len > 12 && chacha_poly1305_decrypt_iv96(ctx, buf, buf + 12, len - 12, decrypted, &decrypted_len)) { + chacha_poly1305_exit(ctx); + free(buf); + config->buf = decrypted; + config->len = decrypted_len; + return true; + } else { + logger(mesh, MESHLINK_ERROR, "Cannot decrypt config file\n"); + meshlink_errno = MESHLINK_ESTORAGE; + chacha_poly1305_exit(ctx); + free(decrypted); + free(buf); + return false; } + } - lineno++; + config->buf = buf; + config->len = len; - if(!*line || *line == '#') - continue; + return true; +} - if(ignore) { - if(!strncmp(line, "-----END", 8)) - ignore = false; - continue; +/// 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); + uint8_t seqbuf[12]; + randomize(&seqbuf, sizeof(seqbuf)); + chacha_poly1305_ctx_t *ctx = chacha_poly1305_init(); + chacha_poly1305_set_key(ctx, key); + bool success = false; + + 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; + } else { + logger(mesh, MESHLINK_ERROR, "Cannot encrypt config file\n"); + meshlink_errno = MESHLINK_ESTORAGE; } - if(!strncmp(line, "-----BEGIN", 10)) { - ignore = true; - continue; - } + chacha_poly1305_exit(ctx); + return success; + } - cfg = parse_config_line(line, fname, lineno); - if (!cfg) - break; - config_add(config_tree, cfg); + if(fwrite(config->buf, config->len, 1, f) != 1) { + logger(mesh, MESHLINK_ERROR, "Cannot write config file: %s", strerror(errno)); + meshlink_errno = MESHLINK_ESTORAGE; + return false; } - fclose(fp); + if(fsync(fileno(f))) { + logger(mesh, MESHLINK_ERROR, "Failed to sync file: %s\n", strerror(errno)); + return false; + } - return result; + return true; } -bool write_config_file(const struct splay_tree_t *config_tree, const char *fname) -{ - FILE *fp; +/// Free resources of a loaded configuration file. +void config_free(config_t *config) { + assert(!config->len || config->buf); - fp = fopen(fname, "w+"); + free((uint8_t *)config->buf); + config->buf = NULL; + config->len = 0; +} - if(!fp) { - logger(NULL, MESHLINK_ERROR, "Cannot open config file %s: %s", fname, strerror(errno)); +/// Check the presence of a host configuration file. +bool config_exists(meshlink_handle_t *mesh, const char *conf_subdir, const char *name) { + assert(conf_subdir); + + if(!mesh->confbase) { return false; } - for splay_each(config_t, cnf, config_tree) - { - if(fwrite(cnf->variable, sizeof(char), strlen(cnf->variable), fp) < strlen(cnf->variable)) { - logger(NULL, MESHLINK_ERROR, "Cannot write to config file %s: %s", fname, strerror(errno)); - return false; - } + char path[PATH_MAX]; + make_host_path(mesh, conf_subdir, name, path, sizeof(path)); - if(fwrite(" = ", sizeof(char), 3, fp) < 3) { - logger(NULL, MESHLINK_ERROR, "Cannot write to config file %s: %s", fname, strerror(errno)); - return false; - } + return access(path, F_OK) == 0; +} - if(fwrite(cnf->value, sizeof(char), strlen(cnf->value), fp) < strlen(cnf->value)) { - logger(NULL, MESHLINK_ERROR, "Cannot write to config file %s: %s", fname, strerror(errno)); - return false; +/// Read a host configuration file. +bool config_read(meshlink_handle_t *mesh, const char *conf_subdir, const char *name, config_t *config, void *key) { + assert(conf_subdir); + + if(!mesh->confbase) { + return false; + } + + char path[PATH_MAX]; + make_host_path(mesh, conf_subdir, name, path, sizeof(path)); + + FILE *f = fopen(path, "r"); + + if(!f) { + logger(mesh, MESHLINK_ERROR, "Failed to open `%s': %s", path, strerror(errno)); + return false; + } + + if(!config_read_file(mesh, f, config, key)) { + logger(mesh, MESHLINK_ERROR, "Failed to read `%s': %s", path, strerror(errno)); + fclose(f); + return false; + } + + fclose(f); + + return true; +} + +bool config_scan_all(meshlink_handle_t *mesh, const char *conf_subdir, const char *conf_type, config_scan_action_t action, void *arg) { + assert(conf_subdir); + assert(conf_type); + + if(!mesh->confbase) { + return true; + } + + DIR *dir; + struct dirent *ent; + char dname[PATH_MAX]; + snprintf(dname, sizeof(dname), "%s" SLASH "%s" SLASH "%s", mesh->confbase, conf_subdir, conf_type); + + dir = opendir(dname); + + if(!dir) { + logger(mesh, MESHLINK_ERROR, "Could not open %s: %s", dname, strerror(errno)); + meshlink_errno = MESHLINK_ESTORAGE; + return false; + } + + while((ent = readdir(dir))) { + if(ent->d_name[0] == '.') { + continue; } - if(fwrite("\n", sizeof(char), 1, fp) < 1) { - logger(NULL, MESHLINK_ERROR, "Cannot write to config file %s: %s", fname, strerror(errno)); + if(!action(mesh, ent->d_name, arg)) { + closedir(dir); return false; } } - fclose(fp); - + closedir(dir); return true; } -bool read_server_config(meshlink_handle_t *mesh) { - char filename[PATH_MAX]; - bool x; +/// 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) { + assert(conf_subdir); + assert(name); + assert(config); - snprintf(filename, PATH_MAX,"%s" SLASH "meshlink.conf", mesh->confbase); - errno = 0; - x = read_config_file(mesh->config, filename); + if(!mesh->confbase) { + return true; + } - if(!x && errno) - logger(mesh, MESHLINK_ERROR, "Failed to read `%s': %s", filename, strerror(errno)); + char path[PATH_MAX]; + char tmp_path[PATH_MAX + 4]; + make_host_path(mesh, conf_subdir, name, path, sizeof(path)); + snprintf(tmp_path, sizeof(tmp_path), "%s.tmp", path); - return x; -} + FILE *f = fopen(tmp_path, "w"); + + if(!f) { + logger(mesh, MESHLINK_ERROR, "Failed to open `%s': %s", tmp_path, strerror(errno)); + return false; + } + + if(!config_write_file(mesh, f, config, key)) { + logger(mesh, MESHLINK_ERROR, "Failed to write `%s': %s", tmp_path, strerror(errno)); + fclose(f); + return false; + } + + if(fsync(fileno(f))) { + logger(mesh, MESHLINK_ERROR, "Failed to sync `%s': %s", tmp_path, strerror(errno)); + fclose(f); + return false; + } -bool read_host_config(meshlink_handle_t *mesh, splay_tree_t *config_tree, const char *name) { - char filename[PATH_MAX]; - bool x; + if(fclose(f)) { + logger(mesh, MESHLINK_ERROR, "Failed to close `%s': %s", tmp_path, strerror(errno)); + return false; + } - snprintf(filename,PATH_MAX, "%s" SLASH "hosts" SLASH "%s", mesh->confbase, name); - x = read_config_file(config_tree, filename); + if(rename(tmp_path, path)) { + logger(mesh, MESHLINK_ERROR, "Failed to rename `%s' to `%s': %s", tmp_path, path, strerror(errno)); + return false; + } - return x; + return true; } -bool write_host_config(struct meshlink_handle *mesh, const struct splay_tree_t *config_tree, const char *name) -{ - char filename[PATH_MAX]; +/// Read the main configuration file. +bool main_config_read(meshlink_handle_t *mesh, const char *conf_subdir, config_t *config, void *key) { + assert(conf_subdir); + assert(config); + + if(!mesh->confbase) { + return false; + } + + char path[PATH_MAX]; + make_main_path(mesh, conf_subdir, path, sizeof(path)); + + FILE *f = fopen(path, "r"); + + if(!f) { + logger(mesh, MESHLINK_ERROR, "Failed to open `%s': %s", path, strerror(errno)); + return false; + } + + if(!config_read_file(mesh, f, config, key)) { + logger(mesh, MESHLINK_ERROR, "Failed to read `%s': %s", path, strerror(errno)); + fclose(f); + return false; + } + + fclose(f); - snprintf(filename,PATH_MAX, "%s" SLASH "hosts" SLASH "%s", mesh->confbase, name); - return write_config_file(config_tree, filename); + return true; } -bool change_config_file(meshlink_handle_t *mesh, const char *name, const char *key, const char *value) { - char filename[PATH_MAX]; - char tmpname[PATH_MAX]; - char buf[MAX_STRING_SIZE]; - const int keylen = strlen(key); +/// Write the main configuration file. +bool main_config_write(meshlink_handle_t *mesh, const char *conf_subdir, const config_t *config, void *key) { + assert(conf_subdir); + assert(config); + + if(!mesh->confbase) { + return true; + } + + char path[PATH_MAX]; + char tmp_path[PATH_MAX + 4]; + make_main_path(mesh, conf_subdir, path, sizeof(path)); + snprintf(tmp_path, sizeof(tmp_path), "%s.tmp", path); - snprintf(filename, PATH_MAX, "%s" SLASH "hosts" SLASH "%s", mesh->confbase, name); - snprintf(tmpname, PATH_MAX, "%s.tmp", filename); + FILE *f = fopen(tmp_path, "w"); + + if(!f) { + logger(mesh, MESHLINK_ERROR, "Failed to open `%s': %s", tmp_path, strerror(errno)); + return false; + } + + if(!config_write_file(mesh, f, config, key)) { + logger(mesh, MESHLINK_ERROR, "Failed to write `%s': %s", tmp_path, strerror(errno)); + fclose(f); + 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)); + fclose(f); + return false; + } - FILE *in = fopen(filename, "r"); - if(!in) { - // Hm, maybe the file does not exist? Try appending. - return append_config_file(mesh, name, key, value); + if(fclose(f)) { + logger(mesh, MESHLINK_ERROR, "Failed to close `%s': %s", tmp_path, strerror(errno)); + return false; } - FILE *out = fopen(tmpname, "w"); - if(!out) { - logger(mesh, MESHLINK_ERROR, "Failed to write `%s': %s", tmpname, strerror(errno)); - fclose(in); + return true; +} + +/// 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) { + assert(conf_subdir); + assert(name); + assert(config); + + if(!mesh->confbase) { return false; } - bool ignore = false; + char path[PATH_MAX]; + char used_path[PATH_MAX]; + make_invitation_path(mesh, conf_subdir, name, path, sizeof(path)); + make_used_invitation_path(mesh, conf_subdir, name, used_path, sizeof(used_path)); - while(readline(in, buf, sizeof buf)) { - if(ignore) { - if(!strncmp(buf, "-----END", 8)) - ignore = false; + // Atomically rename the invitation file + if(rename(path, used_path)) { + if(errno == ENOENT) { + logger(mesh, MESHLINK_ERROR, "Peer tried to use non-existing invitation %s\n", name); } else { - if(!strncmp(buf, "-----BEGIN", 10)) - ignore = true; + logger(mesh, MESHLINK_ERROR, "Error trying to rename invitation %s\n", name); } - if(!ignore && !strncmp(buf, key, keylen)) { - if(strchr("\t =", buf[keylen])) { - continue; - } - } + return false; + } - fputs(buf, out); - fputc('\n', out); - } + FILE *f = fopen(used_path, "r"); - if(ferror(in)) { - logger(mesh, MESHLINK_ERROR, "Failed to read `%s': %s", filename, strerror(errno)); - fclose(in); - fclose(out); + if(!f) { + logger(mesh, MESHLINK_ERROR, "Failed to open `%s': %s", path, strerror(errno)); return false; } - fclose(in); + // Check the timestamp + struct stat st; - fprintf(out, "%s = %s\n", key, value); + if(fstat(fileno(f), &st)) { + logger(mesh, MESHLINK_ERROR, "Could not stat invitation file %s\n", name); + fclose(f); + unlink(used_path); + return false; + } - if(ferror(out)) { - logger(mesh, MESHLINK_ERROR, "Failed to write `%s': %s", tmpname, strerror(errno)); - fclose(out); + 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); return false; } - fclose(out); + if(!config_read_file(mesh, f, config, key)) { + logger(mesh, MESHLINK_ERROR, "Failed to read `%s': %s", path, strerror(errno)); + fclose(f); + unlink(used_path); + return false; + } -#ifdef HAVE_MINGW - // We cannot atomically replace files on Windows. - char bakname[PATH_MAX]; - snprintf(bakname, PATH_MAX, "%s.bak", filename); - if(rename(tmpname, bakfile) || rename(bakfile, filename)) { - rename(bakfile, filename); -#else - if(rename(tmpname, filename)) { -#endif - logger(mesh, MESHLINK_ERROR, "Failed to update `%s': %s", filename, strerror(errno)); + fclose(f); + + unlink(used_path); + 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) { + assert(conf_subdir); + assert(name); + assert(config); + + if(!mesh->confbase) { + return false; + } + + char path[PATH_MAX]; + make_invitation_path(mesh, conf_subdir, name, path, sizeof(path)); + + FILE *f = fopen(path, "w"); + + if(!f) { + logger(mesh, MESHLINK_ERROR, "Failed to open `%s': %s", path, strerror(errno)); + return false; + } + + if(!config_write_file(mesh, f, config, key)) { + logger(mesh, MESHLINK_ERROR, "Failed to write `%s': %s", path, strerror(errno)); + fclose(f); + return false; + } + + if(fsync(fileno(f))) { + logger(mesh, MESHLINK_ERROR, "Failed to sync `%s': %s", path, strerror(errno)); + fclose(f); + return false; + } + + if(fclose(f)) { + logger(mesh, MESHLINK_ERROR, "Failed to close `%s': %s", path, strerror(errno)); return false; } return true; } -bool append_config_file(meshlink_handle_t *mesh, const char *name, const char *key, const char *value) { - char filename[PATH_MAX]; - snprintf(filename,PATH_MAX, "%s" SLASH "hosts" SLASH "%s", mesh->confbase, name); +/// Purge old invitation files +size_t invitation_purge_old(meshlink_handle_t *mesh, time_t deadline) { + if(!mesh->confbase) { + return true; + } + + char path[PATH_MAX]; + make_invitation_path(mesh, "current", "", path, sizeof(path)); - FILE *fp = fopen(filename, "a"); + DIR *dir = opendir(path); - if(!fp) { - logger(mesh, MESHLINK_ERROR, "Cannot open config file %s: %s", filename, strerror(errno)); - } else { - fprintf(fp, "%s = %s\n", key, value); - fclose(fp); + if(!dir) { + logger(mesh, MESHLINK_DEBUG, "Could not read directory %s: %s\n", path, strerror(errno)); + meshlink_errno = MESHLINK_ESTORAGE; + return 0; } - return fp != NULL; + errno = 0; + size_t count = 0; + struct dirent *ent; + + while((ent = readdir(dir))) { + if(strlen(ent->d_name) != 24) { + continue; + } + + char invname[PATH_MAX]; + struct stat st; + + if(snprintf(invname, sizeof(invname), "%s" SLASH "%s", path, ent->d_name) >= PATH_MAX) { + logger(mesh, MESHLINK_DEBUG, "Filename too long: %s" SLASH "%s", path, ent->d_name); + continue; + } + + if(!stat(invname, &st)) { + if(mesh->invitation_key && deadline < st.st_mtime) { + count++; + } else { + unlink(invname); + } + } else { + logger(mesh, MESHLINK_DEBUG, "Could not stat %s: %s\n", invname, strerror(errno)); + errno = 0; + } + } + + if(errno) { + logger(mesh, MESHLINK_DEBUG, "Error while reading directory %s: %s\n", path, strerror(errno)); + closedir(dir); + meshlink_errno = MESHLINK_ESTORAGE; + return 0; + } + + closedir(dir); + + return count; }