]> git.meshlink.io Git - meshlink/blobdiff - src/conf.c
Add a function to change a configuration file on disk.
[meshlink] / src / conf.c
index bf8183f7d9b8e1fe4eb10f8f4d870f0e2bb4b735..2195d69c42ff32230e85f54c67bae00aeac9a674 100644 (file)
@@ -1,10 +1,6 @@
 /*
     conf.c -- configuration code
-    Copyright (C) 1998 Robert van der Meulen
-                  1998-2005 Ivo Timmermans
-                  2000-2010 Guus Sliepen <guus@tinc-vpn.org>
-                  2010-2011 Julien Muchembled <jm@jmuchemb.eu>
-                 2000 Cris van Pelt
+    Copyright (C) 2014 Guus Sliepen <guus@meshlink.io>
 
     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
 #include "splay_tree.h"
 #include "connection.h"
 #include "conf.h"
+#include "list.h"
 #include "logger.h"
-#include "netutl.h"                            /* for str2address */
+#include "meshlink_internal.h"
+#include "netutl.h"             /* for str2address */
 #include "protocol.h"
-#include "utils.h"                             /* for cp */
+#include "utils.h"              /* for cp */
 #include "xalloc.h"
 
-splay_tree_t *config_tree;
-
-int pinginterval = 0;                  /* seconds between pings */
-int pingtimeout = 0;                   /* seconds to wait for response */
-char *confbase = NULL;                 /* directory in which all config files are */
-char *netname = NULL;                  /* name of the vpn network */
-list_t *cmdline_conf = NULL;   /* global/host configuration values given at the command line */
-
-
 static int config_compare(const config_t *a, const config_t *b) {
        int result;
 
@@ -49,11 +38,6 @@ static int config_compare(const config_t *a, const config_t *b) {
        if(result)
                return result;
 
-       /* give priority to command line options */
-       result = !b->file - !a->file;
-       if (result)
-               return result;
-
        result = a->line - b->line;
 
        if(result)
@@ -62,17 +46,18 @@ static int config_compare(const config_t *a, const config_t *b) {
                return a->file ? strcmp(a->file, b->file) : 0;
 }
 
-void init_configuration(splay_tree_t ** config_tree) {
+void init_configuration(splay_tree_t **config_tree) {
        *config_tree = splay_alloc_tree((splay_compare_t) config_compare, (splay_action_t) free_config);
 }
 
-void exit_configuration(splay_tree_t ** config_tree) {
-       splay_delete_tree(*config_tree);
+void exit_configuration(splay_tree_t **config_tree) {
+       if(*config_tree)
+               splay_delete_tree(*config_tree);
        *config_tree = NULL;
 }
 
 config_t *new_config(void) {
-       return xmalloc_and_zero(sizeof(config_t));
+       return xzalloc(sizeof(config_t));
 }
 
 void free_config(config_t *cfg) {
@@ -140,7 +125,7 @@ bool get_config_bool(const config_t *cfg, bool *result) {
                return true;
        }
 
-       logger(LOG_ERR, "\"yes\" or \"no\" expected for configuration variable %s in %s line %d",
+       logger(NULL, MESHLINK_ERROR, "\"yes\" or \"no\" expected for configuration variable %s in %s line %d",
                   cfg->variable, cfg->file, cfg->line);
 
        return false;
@@ -153,12 +138,28 @@ bool get_config_int(const config_t *cfg, int *result) {
        if(sscanf(cfg->value, "%d", result) == 1)
                return true;
 
-       logger(LOG_ERR, "Integer expected for configuration variable %s in %s line %d",
+       logger(NULL, MESHLINK_ERROR, "Integer expected for configuration variable %s in %s line %d",
                   cfg->variable, cfg->file, cfg->line);
 
        return false;
 }
 
+bool set_config_int(config_t *cfg, int val)
+{
+       if(!cfg)
+               return false;
+
+       char val_str[1024];
+       snprintf(val_str, sizeof(val_str), "%d", val);
+
+       if(cfg->value)
+               free(cfg->value);
+
+       cfg->value = xstrdup(val_str);
+
+       return true;
+}
+
 bool get_config_string(const config_t *cfg, char **result) {
        if(!cfg)
                return false;
@@ -168,6 +169,19 @@ bool get_config_string(const config_t *cfg, char **result) {
        return true;
 }
 
+bool set_config_string(config_t *cfg, const char* val)
+{
+       if(!cfg)
+               return false;
+
+       if(cfg->value)
+               free(cfg->value);
+
+       cfg->value = xstrdup(val);
+
+       return true;
+}
+
 bool get_config_address(const config_t *cfg, struct addrinfo **result) {
        struct addrinfo *ai;
 
@@ -181,40 +195,12 @@ bool get_config_address(const config_t *cfg, struct addrinfo **result) {
                return true;
        }
 
-       logger(LOG_ERR, "Hostname or IP address expected for configuration variable %s in %s line %d",
+       logger(NULL, MESHLINK_ERROR, "Hostname or IP address expected for configuration variable %s in %s line %d",
                   cfg->variable, cfg->file, cfg->line);
 
        return false;
 }
 
-bool get_config_subnet(const config_t *cfg, subnet_t ** result) {
-       subnet_t subnet = {0};
-
-       if(!cfg)
-               return false;
-
-       if(!str2net(&subnet, cfg->value)) {
-               logger(LOG_ERR, "Subnet expected for configuration variable %s in %s line %d",
-                          cfg->variable, cfg->file, cfg->line);
-               return false;
-       }
-
-       /* Teach newbies what subnets are... */
-
-       if(((subnet.type == SUBNET_IPV4)
-               && !maskcheck(&subnet.net.ipv4.address, subnet.net.ipv4.prefixlength, sizeof subnet.net.ipv4.address))
-               || ((subnet.type == SUBNET_IPV6)
-               && !maskcheck(&subnet.net.ipv6.address, subnet.net.ipv6.prefixlength, sizeof subnet.net.ipv6.address))) {
-               logger(LOG_ERR, "Network address and prefix length do not match for configuration variable %s in %s line %d",
-                          cfg->variable, cfg->file, cfg->line);
-               return false;
-       }
-
-       *(*result = new_subnet()) = subnet;
-
-       return true;
-}
-
 /*
   Read exactly one line and strip the trailing newline if any.
 */
@@ -235,8 +221,9 @@ static char *readline(FILE * fp, char *buf, size_t buflen) {
        if(!newline)
                return buf;
 
-       *newline = '\0';        /* kill newline */
-       if(newline > p && newline[-1] == '\r')  /* and carriage return if necessary */
+       /* kill newline and carriage return if necessary */
+       *newline = '\0';
+       if(newline > p && newline[-1] == '\r')
                newline[-1] = '\0';
 
        return buf;
@@ -263,19 +250,15 @@ config_t *parse_config_line(char *line, const char *fname, int lineno) {
 
        if(!*value) {
                const char err[] = "No value for variable";
-               if (fname)
-                       logger(LOG_ERR, "%s `%s' on line %d while reading config file %s",
-                               err, variable, lineno, fname);
-               else
-                       logger(LOG_ERR, "%s `%s' in command line option %d",
-                               err, variable, lineno);
+               logger(NULL, MESHLINK_ERROR, "%s `%s' on line %d while reading config file %s",
+                       err, variable, lineno, fname);
                return NULL;
        }
 
        cfg = new_config();
        cfg->variable = xstrdup(variable);
        cfg->value = xstrdup(value);
-       cfg->file = fname ? xstrdup(fname) : NULL;
+       cfg->file = xstrdup(fname);
        cfg->line = lineno;
 
        return cfg;
@@ -297,7 +280,7 @@ bool read_config_file(splay_tree_t *config_tree, const char *fname) {
        fp = fopen(fname, "r");
 
        if(!fp) {
-               logger(LOG_ERR, "Cannot open config file %s: %s", fname, strerror(errno));
+               logger(NULL, MESHLINK_ERROR, "Cannot open config file %s: %s", fname, strerror(errno));
                return false;
        }
 
@@ -320,7 +303,7 @@ bool read_config_file(splay_tree_t *config_tree, const char *fname) {
                                ignore = false;
                        continue;
                }
-               
+
                if(!strncmp(line, "-----BEGIN", 10)) {
                        ignore = true;
                        continue;
@@ -337,104 +320,167 @@ bool read_config_file(splay_tree_t *config_tree, const char *fname) {
        return result;
 }
 
-void read_config_options(splay_tree_t *config_tree, const char *prefix) {
-       list_node_t *node, *next;
-       size_t prefix_len = prefix ? strlen(prefix) : 0;
+bool write_config_file(const struct splay_tree_t *config_tree, const char *fname)
+{
+       FILE *fp;
 
-       for(node = cmdline_conf->tail; node; node = next) {
-               config_t *orig_cfg, *cfg = (config_t *)node->data;
-               next = node->prev;
+       fp = fopen(fname, "w+");
 
-               if(!prefix) {
-                       if(strchr(cfg->variable, '.'))
-                               continue;
-                       node->data = NULL;
-                       list_unlink_node(cmdline_conf, node);
-               } else {
-                       if(strncmp(prefix, cfg->variable, prefix_len) ||
-                          cfg->variable[prefix_len] != '.')
-                               continue;
-                       /* Because host configuration is parsed again when
-                          reconnecting, nodes must not be freed when a prefix
-                          is given. */
-                       orig_cfg = cfg;
-                       cfg = new_config();
-                       cfg->variable = xstrdup(orig_cfg->variable + prefix_len + 1);
-                       cfg->value = xstrdup(orig_cfg->value);
-                       cfg->file = NULL;
-                       cfg->line = orig_cfg->line;
-               }
-               config_add(config_tree, cfg);
+       if(!fp) {
+               logger(NULL, MESHLINK_ERROR, "Cannot open config file %s: %s", fname, strerror(errno));
+               return false;
        }
-}
 
-bool read_server_config() {
-       char *fname;
-       bool x;
+       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;
+               }
 
-       read_config_options(config_tree, NULL);
+               if(fwrite(" = ", sizeof(char), 3, fp) < 3) {
+                       logger(NULL, MESHLINK_ERROR, "Cannot write to config file %s: %s", fname, strerror(errno));
+                       return false;
+               }
 
-       xasprintf(&fname, "%s/tinc.conf", confbase);
-       x = read_config_file(config_tree, fname);
+               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;
+               }
 
-       if(!x) {                                /* System error: complain */
-               logger(LOG_ERR, "Failed to read `%s': %s", fname, strerror(errno));
+               if(fwrite("\n", sizeof(char), 1, fp) < 1) {
+                       logger(NULL, MESHLINK_ERROR, "Cannot write to config file %s: %s", fname, strerror(errno));
+                       return false;
+               }
        }
 
-       free(fname);
+       fclose(fp);
 
-       return x;
+       return true;
 }
 
-bool read_connection_config(connection_t *c) {
-       char *fname;
+bool read_server_config(meshlink_handle_t *mesh) {
+       char filename[PATH_MAX];
        bool x;
 
-       read_config_options(c->config_tree, c->name);
+       snprintf(filename, PATH_MAX,"%s" SLASH "meshlink.conf", mesh->confbase);
+       errno = 0;
+       x = read_config_file(mesh->config, filename);
 
-       xasprintf(&fname, "%s/hosts/%s", confbase, c->name);
-       x = read_config_file(c->config_tree, fname);
-       free(fname);
+       if(!x && errno)
+               logger(mesh, MESHLINK_ERROR, "Failed to read `%s': %s", filename, strerror(errno));
 
        return x;
 }
 
-bool disable_old_keys(FILE *f) {
-       char buf[100];
-       long pos;
-       bool disabled = false;
+bool read_host_config(meshlink_handle_t *mesh, splay_tree_t *config_tree, const char *name) {
+       char filename[PATH_MAX];
+       bool x;
 
-       rewind(f);
-       pos = ftell(f);
+       snprintf(filename,PATH_MAX, "%s" SLASH "hosts" SLASH "%s", mesh->confbase, name);
+       x = read_config_file(config_tree, filename);
 
-       if(pos < 0)
+       return x;
+}
+
+bool write_host_config(struct meshlink_handle *mesh, const struct splay_tree_t *config_tree, const char *name)
+{
+       char filename[PATH_MAX];
+
+       snprintf(filename,PATH_MAX, "%s" SLASH "hosts" SLASH "%s", mesh->confbase, name);
+       return write_config_file(config_tree, filename);
+}
+
+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);
+
+       snprintf(filename, PATH_MAX, "%s" SLASH "hosts" SLASH "%s", mesh->confbase, name);
+       snprintf(tmpname, PATH_MAX, "%s.tmp", filename);
+
+       FILE *in = fopen(filename, "r");
+       if(!in) {
+               // Hm, maybe the file does not exist? Try appending.
+               return append_config_file(mesh, name, key, value);
+       }
+
+       FILE *out = fopen(tmpname, "w");
+       if(!out) {
+               logger(mesh, MESHLINK_ERROR, "Failed to write `%s': %s", tmpname, strerror(errno));
+               fclose(in);
                return false;
+       }
 
-       while(fgets(buf, sizeof buf, f)) {
-               if(!strncmp(buf, "-----BEGIN RSA", 14)) {       
-                       buf[11] = 'O';
-                       buf[12] = 'L';
-                       buf[13] = 'D';
-                       if(fseek(f, pos, SEEK_SET))
-                               break;
-                       if(fputs(buf, f) <= 0)
-                               break;
-                       disabled = true;
+       bool ignore = false;
+
+       while(readline(in, buf, sizeof buf)) {
+               if(ignore) {
+                       if(!strncmp(buf, "-----END", 8))
+                               ignore = false;
+               } else {
+                       if(!strncmp(buf, "-----BEGIN", 10))
+                               ignore = true;
                }
-               else if(!strncmp(buf, "-----END RSA", 12)) {    
-                       buf[ 9] = 'O';
-                       buf[10] = 'L';
-                       buf[11] = 'D';
-                       if(fseek(f, pos, SEEK_SET))
-                               break;
-                       if(fputs(buf, f) <= 0)
-                               break;
-                       disabled = true;
+
+               if(!ignore && !strncmp(buf, key, keylen)) {
+                       if(strchr("\t =", buf[keylen])) {
+                               continue;
+                       }
                }
-               pos = ftell(f);
-               if(pos < 0)
-                       break;
+
+               fputs(buf, out);
+               fputc('\n', out);
+       }       
+
+       if(ferror(in)) {
+               logger(mesh, MESHLINK_ERROR, "Failed to read `%s': %s", filename, strerror(errno));
+               fclose(in);
+               fclose(out);
+               return false;
+       }
+
+       fclose(in);
+
+       fprintf(out, "%s = %s\n", key, value);
+
+       if(ferror(out)) {
+               logger(mesh, MESHLINK_ERROR, "Failed to write `%s': %s", tmpname, strerror(errno));
+               fclose(out);
+               return false;
+       }
+
+       fclose(out);
+
+#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));
+               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);
+
+       FILE *fp = fopen(filename, "a");
+
+       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);
        }
 
-       return disabled;
+       return fp != NULL;
 }