implement drop-in directories

1 files changed, 61 insertions, 0 deletions

diff --git a/load-dropin.c b/load-dropin.c
index 3023cdae60..b975d83f04 100644
--- a/load-dropin.c
+++ b/load-dropin.c
@@ -1,11 +1,72 @@
 /*-*- Mode: C; c-basic-offset: 8 -*-*/
 
+#include <dirent.h>
+#include <errno.h>
+
+#include "unit.h"
 #include "load-dropin.h"
 
 int unit_load_dropin(Unit *u) {
+        Iterator i;
+        int r;
+        char *t;
+
         assert(u);
 
         /* Load dependencies from supplementary drop-in directories */
 
+        SET_FOREACH(t, u->meta.names, i) {
+                char *path;
+                DIR *d;
+                struct dirent *de;
+
+                if (asprintf(&path, "%s/%s.wants", unit_path(), t) < 0)
+                        return -ENOMEM;
+
+                if (!(d = opendir(path))) {
+                        r = -errno;
+                        free(path);
+
+                        if (r == -ENOENT)
+                                continue;
+
+                        return r;
+                }
+
+                free(path);
+
+                while ((de = readdir(d))) {
+                        Unit *other;
+
+                        if (de->d_name[0] == '.')
+                                continue;
+
+                        assert(de->d_name[0]);
+
+                        if (de->d_name[strlen(de->d_name)-1] == '~')
+                                continue;
+
+                        if (asprintf(&path, "%s/%s.wants/%s", unit_path(), t, de->d_name) < 0) {
+                                closedir(d);
+                                return -ENOMEM;
+                        }
+
+                        r = manager_load_unit(u->meta.manager, path, &other);
+                        free(path);
+
+                        if (r < 0) {
+                                closedir(d);
+                                return r;
+                        }
+
+                        if ((r = unit_add_dependency(u, UNIT_WANTS, other)) < 0) {
+                                closedir(d);
+                                return r;
+                        }
+                }
+
+                closedir(d);
+        }
+
         return 0;
 }