chiark / gitweb /
implement drop-in directories
[elogind.git] / load-fragment.c
index cf1e1498615fab6c5d39c92961e316a57eab6a32..d016f01d119b4e15d5d702b3c158426ff4f9339d 100644 (file)
@@ -1,11 +1,13 @@
 /*-*- Mode: C; c-basic-offset: 8 -*-*/
 
+#include <linux/oom.h>
 #include <assert.h>
 #include <errno.h>
 #include <string.h>
-#include <linux/oom.h>
+#include <unistd.h>
+#include <fcntl.h>
 
-#include "name.h"
+#include "unit.h"
 #include "strv.h"
 #include "conf-parser.h"
 #include "load-fragment.h"
@@ -20,8 +22,8 @@ static int config_parse_deps(
                 void *data,
                 void *userdata) {
 
-        NameDependency d = PTR_TO_UINT(data);
-        Name *name = userdata;
+        UnitDependency d = PTR_TO_UINT(data);
+        Unit *u = userdata;
         char *w;
         size_t l;
         char *state;
@@ -33,18 +35,18 @@ static int config_parse_deps(
         FOREACH_WORD(w, &l, rvalue, state) {
                 char *t;
                 int r;
-                Name *other;
+                Unit *other;
 
                 if (!(t = strndup(w, l)))
                         return -ENOMEM;
 
-                r = manager_load_name(name->meta.manager, t, &other);
+                r = manager_load_unit(u->meta.manager, t, &other);
                 free(t);
 
                 if (r < 0)
                         return r;
 
-                if ((r = name_add_dependency(name, d, other)) < 0)
+                if ((r = unit_add_dependency(u, d, other)) < 0)
                         return r;
         }
 
@@ -60,8 +62,7 @@ static int config_parse_names(
                 void *data,
                 void *userdata) {
 
-        Set **set = data;
-        Name *name = userdata;
+        Unit *u = userdata;
         char *w;
         size_t l;
         char *state;
@@ -74,42 +75,33 @@ static int config_parse_names(
         FOREACH_WORD(w, &l, rvalue, state) {
                 char *t;
                 int r;
-                Name *other;
+                Unit *other;
 
                 if (!(t = strndup(w, l)))
                         return -ENOMEM;
 
-                other = manager_get_name(name->meta.manager, t);
+                other = manager_get_unit(u->meta.manager, t);
 
                 if (other) {
 
-                        if (other != name) {
+                        if (other != u) {
 
-                                if (other->meta.load_state != NAME_STUB) {
+                                if (other->meta.load_state != UNIT_STUB) {
                                         free(t);
                                         return -EEXIST;
                                 }
 
-                                if ((r = name_merge(name, other)) < 0) {
+                                if ((r = unit_merge(u, other)) < 0) {
                                         free(t);
                                         return r;
                                 }
                         }
 
                 } else {
-
-                        if (!*set)
-                                if (!(*set = set_new(trivial_hash_func, trivial_compare_func))) {
-                                        free(t);
-                                        return -ENOMEM;
-                                }
-
-                        if ((r = set_put(*set, t)) < 0) {
+                        if ((r = unit_add_name(u, t)) < 0) {
                                 free(t);
                                 return r;
                         }
-
-                        t = NULL;
                 }
 
                 free(t);
@@ -336,7 +328,7 @@ static int config_parse_exec(
 
         n[k] = NULL;
 
-        if (!n[0] || n[0][0] != '/') {
+        if (!n[0] || !path_is_absolute(n[0])) {
                 log_error("[%s:%u] Invalid executable path in command line: %s", filename, line, rvalue);
                 strv_free(n);
                 return -EINVAL;
@@ -458,17 +450,112 @@ static int config_parse_service_restart(
         return 0;
 }
 
-int name_load_fragment(Name *n) {
-
-        static const char* const section_table[_NAME_TYPE_MAX] = {
-                [NAME_SERVICE]   = "Service",
-                [NAME_TIMER]     = "Timer",
-                [NAME_SOCKET]    = "Socket",
-                [NAME_TARGET]    = "Target",
-                [NAME_DEVICE]    = "Device",
-                [NAME_MOUNT]     = "Mount",
-                [NAME_AUTOMOUNT] = "Automount",
-                [NAME_SNAPSHOT]  = "Snapshot"
+#define FOLLOW_MAX 8
+
+static char *build_path(const char *path, const char *filename) {
+        char *e, *r;
+        size_t k;
+
+        assert(path);
+        assert(filename);
+
+        /* This removes the last component of path and appends
+         * filename, unless the latter is absolute anyway or the
+         * former isn't */
+
+        if (path_is_absolute(filename))
+                return strdup(filename);
+
+        if (!(e = strrchr(path, '/')))
+                return strdup(filename);
+
+        k = strlen(filename);
+        if (!(r = new(char, e-path+1+k+1)))
+                return NULL;
+
+        memcpy(r, path, e-path+1);
+        memcpy(r+(e-path)+1, filename, k+1);
+
+        return r;
+}
+
+static int open_follow(char **filename, FILE **_f, Set *names, char **_id) {
+        unsigned c = 0;
+        int fd, r;
+        FILE *f;
+        char *id = NULL;
+
+        assert(filename);
+        assert(*filename);
+        assert(_f);
+        assert(names);
+
+        /* This will update the filename pointer if the loaded file is
+         * reached by a symlink. The old string will be freed. */
+
+        for (;;) {
+                char *target, *k, *name;
+
+                if (c++ >= FOLLOW_MAX)
+                        return -ELOOP;
+
+                /* Add the file name we are currently looking at to
+                 * the names of this unit */
+                name = file_name_from_path(*filename);
+                if (!(id = set_get(names, name))) {
+
+                        if (!(id = strdup(name)))
+                                return -ENOMEM;
+
+                        if ((r = set_put(names, id)) < 0) {
+                                free(id);
+                                return r;
+                        }
+                }
+
+                /* Try to open the file name, but don't if its a symlink */
+                if ((fd = open(*filename, O_RDONLY|O_CLOEXEC|O_NOCTTY|O_NOFOLLOW)) >= 0)
+                        break;
+
+                if (errno != ELOOP)
+                        return -errno;
+
+                /* Hmm, so this is a symlink. Let's read the name, and follow it manually */
+                if ((r = readlink_malloc(*filename, &target)) < 0)
+                        return r;
+
+                k = build_path(*filename, target);
+                free(target);
+
+                if (!k)
+                        return -ENOMEM;
+
+                free(*filename);
+                *filename = k;
+        }
+
+        if (!(f = fdopen(fd, "r"))) {
+                r = -errno;
+                assert(close_nointr(fd) == 0);
+                return r;
+        }
+
+        *_f = f;
+        *_id = id;
+        return 0;
+}
+
+static int load_from_path(Unit *u, const char *path) {
+
+        static const char* const section_table[_UNIT_TYPE_MAX] = {
+                [UNIT_SERVICE]   = "Service",
+                [UNIT_TIMER]     = "Timer",
+                [UNIT_SOCKET]    = "Socket",
+                [UNIT_TARGET]    = "Target",
+                [UNIT_DEVICE]    = "Device",
+                [UNIT_MOUNT]     = "Mount",
+                [UNIT_AUTOMOUNT] = "Automount",
+                [UNIT_SNAPSHOT]  = "Snapshot"
         };
 
 #define EXEC_CONTEXT_CONFIG_ITEMS(context, section) \
@@ -482,73 +569,126 @@ int name_load_fragment(Name *n) {
                 { "Environment",            config_parse_strv,            &(context).environment,                            section   }
 
         const ConfigItem items[] = {
-                { "Names",                  config_parse_names,           &n->meta.names,                                    "Meta"    },
-                { "Description",            config_parse_string,          &n->meta.description,                              "Meta"    },
-                { "Requires",               config_parse_deps,            UINT_TO_PTR(NAME_REQUIRES),                        "Meta"    },
-                { "SoftRequires",           config_parse_deps,            UINT_TO_PTR(NAME_SOFT_REQUIRES),                   "Meta"    },
-                { "Wants",                  config_parse_deps,            UINT_TO_PTR(NAME_WANTS),                           "Meta"    },
-                { "Requisite",              config_parse_deps,            UINT_TO_PTR(NAME_REQUISITE),                       "Meta"    },
-                { "SoftRequisite",          config_parse_deps,            UINT_TO_PTR(NAME_SOFT_REQUISITE),                  "Meta"    },
-                { "Conflicts",              config_parse_deps,            UINT_TO_PTR(NAME_CONFLICTS),                       "Meta"    },
-                { "Before",                 config_parse_deps,            UINT_TO_PTR(NAME_BEFORE),                          "Meta"    },
-                { "After",                  config_parse_deps,            UINT_TO_PTR(NAME_AFTER),                           "Meta"    },
-
-                { "PIDFile",                config_parse_path,            &n->service.pid_file,                              "Service" },
-                { "ExecStartPre",           config_parse_exec,            &n->service.exec_command[SERVICE_EXEC_START_PRE],  "Service" },
-                { "ExecStart",              config_parse_exec,            &n->service.exec_command[SERVICE_EXEC_START],      "Service" },
-                { "ExecStartPost",          config_parse_exec,            &n->service.exec_command[SERVICE_EXEC_START_POST], "Service" },
-                { "ExecReload",             config_parse_exec,            &n->service.exec_command[SERVICE_EXEC_RELOAD],     "Service" },
-                { "ExecStop",               config_parse_exec,            &n->service.exec_command[SERVICE_EXEC_STOP],       "Service" },
-                { "ExecStopPost",           config_parse_exec,            &n->service.exec_command[SERVICE_EXEC_STOP_POST],  "Service" },
-                { "RestartSec",             config_parse_usec,            &n->service.restart_usec,                          "Service" },
-                { "TimeoutSec",             config_parse_usec,            &n->service.timeout_usec,                          "Service" },
-                { "Type",                   config_parse_service_type,    &n->service,                                       "Service" },
-                { "Restart",                config_parse_service_restart, &n->service,                                       "Service" },
-                EXEC_CONTEXT_CONFIG_ITEMS(n->service.exec_context, "Service"),
-
-                { "ListenStream",           config_parse_listen,          &n->socket,                                        "Socket"  },
-                { "ListenDatagram",         config_parse_listen,          &n->socket,                                        "Socket"  },
-                { "ListenSequentialPacket", config_parse_listen,          &n->socket,                                        "Socket"  },
-                { "ListenFIFO",             config_parse_listen,          &n->socket,                                        "Socket"  },
-                { "BindIPv6Only",           config_parse_socket_bind,     &n->socket,                                        "Socket"  },
-                { "Backlog",                config_parse_unsigned,        &n->socket.backlog,                                "Socket"  },
-                { "ExecStartPre",           config_parse_exec,            &n->service.exec_command[SOCKET_EXEC_START_PRE],   "Socket"  },
-                { "ExecStartPost",          config_parse_exec,            &n->service.exec_command[SOCKET_EXEC_START_POST],  "Socket"  },
-                { "ExecStopPre",            config_parse_exec,            &n->service.exec_command[SOCKET_EXEC_STOP_PRE],    "Socket"  },
-                { "ExecStopPost",           config_parse_exec,            &n->service.exec_command[SOCKET_EXEC_STOP_POST],   "Socket"  },
-                EXEC_CONTEXT_CONFIG_ITEMS(n->socket.exec_context, "Socket"),
-
-                EXEC_CONTEXT_CONFIG_ITEMS(n->automount.exec_context, "Automount"),
+                { "Names",                  config_parse_names,           u,                                                 "Meta"    },
+                { "Description",            config_parse_string,          &u->meta.description,                              "Meta"    },
+                { "Requires",               config_parse_deps,            UINT_TO_PTR(UNIT_REQUIRES),                        "Meta"    },
+                { "SoftRequires",           config_parse_deps,            UINT_TO_PTR(UNIT_SOFT_REQUIRES),                   "Meta"    },
+                { "Wants",                  config_parse_deps,            UINT_TO_PTR(UNIT_WANTS),                           "Meta"    },
+                { "Requisite",              config_parse_deps,            UINT_TO_PTR(UNIT_REQUISITE),                       "Meta"    },
+                { "SoftRequisite",          config_parse_deps,            UINT_TO_PTR(UNIT_SOFT_REQUISITE),                  "Meta"    },
+                { "Conflicts",              config_parse_deps,            UINT_TO_PTR(UNIT_CONFLICTS),                       "Meta"    },
+                { "Before",                 config_parse_deps,            UINT_TO_PTR(UNIT_BEFORE),                          "Meta"    },
+                { "After",                  config_parse_deps,            UINT_TO_PTR(UNIT_AFTER),                           "Meta"    },
+
+                { "PIDFile",                config_parse_path,            &u->service.pid_file,                              "Service" },
+                { "ExecStartPre",           config_parse_exec,            &u->service.exec_command[SERVICE_EXEC_START_PRE],  "Service" },
+                { "ExecStart",              config_parse_exec,            &u->service.exec_command[SERVICE_EXEC_START],      "Service" },
+                { "ExecStartPost",          config_parse_exec,            &u->service.exec_command[SERVICE_EXEC_START_POST], "Service" },
+                { "ExecReload",             config_parse_exec,            &u->service.exec_command[SERVICE_EXEC_RELOAD],     "Service" },
+                { "ExecStop",               config_parse_exec,            &u->service.exec_command[SERVICE_EXEC_STOP],       "Service" },
+                { "ExecStopPost",           config_parse_exec,            &u->service.exec_command[SERVICE_EXEC_STOP_POST],  "Service" },
+                { "RestartSec",             config_parse_usec,            &u->service.restart_usec,                          "Service" },
+                { "TimeoutSec",             config_parse_usec,            &u->service.timeout_usec,                          "Service" },
+                { "Type",                   config_parse_service_type,    &u->service,                                       "Service" },
+                { "Restart",                config_parse_service_restart, &u->service,                                       "Service" },
+                EXEC_CONTEXT_CONFIG_ITEMS(u->service.exec_context, "Service"),
+
+                { "ListenStream",           config_parse_listen,          &u->socket,                                        "Socket"  },
+                { "ListenDatagram",         config_parse_listen,          &u->socket,                                        "Socket"  },
+                { "ListenSequentialPacket", config_parse_listen,          &u->socket,                                        "Socket"  },
+                { "ListenFIFO",             config_parse_listen,          &u->socket,                                        "Socket"  },
+                { "BindIPv6Only",           config_parse_socket_bind,     &u->socket,                                        "Socket"  },
+                { "Backlog",                config_parse_unsigned,        &u->socket.backlog,                                "Socket"  },
+                { "ExecStartPre",           config_parse_exec,            &u->service.exec_command[SOCKET_EXEC_START_PRE],   "Socket"  },
+                { "ExecStartPost",          config_parse_exec,            &u->service.exec_command[SOCKET_EXEC_START_POST],  "Socket"  },
+                { "ExecStopPre",            config_parse_exec,            &u->service.exec_command[SOCKET_EXEC_STOP_PRE],    "Socket"  },
+                { "ExecStopPost",           config_parse_exec,            &u->service.exec_command[SOCKET_EXEC_STOP_POST],   "Socket"  },
+                EXEC_CONTEXT_CONFIG_ITEMS(u->socket.exec_context, "Socket"),
+
+                EXEC_CONTEXT_CONFIG_ITEMS(u->automount.exec_context, "Automount"),
 
                 { NULL, NULL, NULL, NULL }
         };
 
 #undef EXEC_CONTEXT_CONFIG_ITEMS
 
-        char *t;
-        int r;
         const char *sections[3];
-        Iterator i;
-
-        assert(n);
-        assert(n->meta.load_state == NAME_STUB);
+        char *k;
+        int r;
+        Set *symlink_names;
+        FILE *f;
+        char *filename, *id;
 
         sections[0] = "Meta";
-        sections[1] = section_table[n->meta.type];
+        sections[1] = section_table[u->meta.type];
         sections[2] = NULL;
 
-        SET_FOREACH(t, n->meta.names, i) {
+        if (!(symlink_names = set_new(string_hash_func, string_compare_func)))
+                return -ENOMEM;
 
-                /* Try to find a name we can load this with */
-                if ((r = config_parse(t, sections, items, n)) == -ENOENT)
-                        continue;
+        /* Instead of opening the path right away, we manually
+         * follow all symlinks and add their name to our unit
+         * name set while doing so */
+        if (!(filename = path_make_absolute(path, unit_path()))) {
+                r = -ENOMEM;
+                goto finish;
+        }
 
-                /* Yay, we succeeded! Now let's call this our identifier */
-                if (r == 0)
-                        n->meta.id = t;
+        if ((r = open_follow(&filename, &f, symlink_names, &id)) < 0) {
+                if (r == -ENOENT)
+                        r = 0; /* returning 0 means: no suitable config file found */
 
-                return r;
+                goto finish;
+        }
+
+        /* Now, parse the file contents */
+        r = config_parse(filename, f, sections, items, u);
+        if (r < 0)
+                goto finish;
+
+        /* Let's try to add in all symlink names we found */
+        while ((k = set_steal_first(symlink_names))) {
+                if ((r = unit_add_name(u, k)) < 0)
+                        goto finish;
+
+                if (id == k)
+                        assert_se(u->meta.id = set_get(u->meta.names, k));
+
+                free(k);
+        }
+
+        free(u->meta.load_path);
+        u->meta.load_path = filename;
+        filename = NULL;
+
+        r = 1; /* returning 1 means: suitable config file found and loaded */
+
+finish:
+        while ((k = set_steal_first(symlink_names)))
+                free(k);
+        set_free(symlink_names);
+        free(filename);
+
+        return r;
+}
+
+int unit_load_fragment(Unit *u) {
+        int r = -ENOENT;
+
+        assert(u);
+        assert(u->meta.load_state == UNIT_STUB);
+
+        if (u->meta.load_path)
+                r = load_from_path(u, u->meta.load_path);
+        else {
+                Iterator i;
+                char *t;
+
+                /* Try to find a name we can load this with */
+                SET_FOREACH(t, u->meta.names, i)
+                        if ((r = load_from_path(u, t)) != 0)
+                                return r;
         }
 
-        return -ENOENT;
+        return r;
 }