load-dropin.c

   1 /*-*- Mode: C; c-basic-offset: 8 -*-*/
   2
   3 /***
   4   This file is part of systemd.
   5
   6   Copyright 2010 Lennart Poettering
   7
   8   systemd is free software; you can redistribute it and/or modify it
   9   under the terms of the GNU General Public License as published by
  10   the Free Software Foundation; either version 2 of the License, or
  11   (at your option) any later version.
  12
  13   systemd is distributed in the hope that it will be useful, but
  14   WITHOUT ANY WARRANTY; without even the implied warranty of
  15   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  16   General Public License for more details.
  17
  18   You should have received a copy of the GNU General Public License
  19   along with systemd; If not, see <http://www.gnu.org/licenses/>.
  20 ***/
  21
  22 #include <dirent.h>
  23 #include <errno.h>
  24
  25 #include "unit.h"
  26 #include "load-dropin.h"
  27 #include "log.h"
  28 #include "strv.h"
  29 #include "unit-name.h"
  30
  31 static int iterate_dir(Unit *u, const char *path) {
  32         DIR *d;
  33         struct dirent *de;
  34         int r;
  35
  36         if (!(d = opendir(path))) {
  37
  38                 if (errno == ENOENT)
  39                         return 0;
  40
  41                 return -errno;
  42         }
  43
  44         while ((de = readdir(d))) {
  45                 char *f;
  46
  47                 if (ignore_file(de->d_name))
  48                         continue;
  49
  50                 if (asprintf(&f, "%s/%s", path, de->d_name) < 0) {
  51                         r = -ENOMEM;
  52                         goto finish;
  53                 }
  54
  55                 r = unit_add_dependency_by_name(u, UNIT_WANTS, de->d_name, f, true);
  56                 free(f);
  57
  58                 if (r < 0)
  59                         goto finish;
  60         }
  61
  62         r = 0;
  63
  64 finish:
  65         closedir(d);
  66         return r;
  67 }
  68
  69 int unit_load_dropin(Unit *u) {
  70         Iterator i;
  71         int r;
  72         char *t;
  73
  74         assert(u);
  75
  76         /* Load dependencies from supplementary drop-in directories */
  77
  78         SET_FOREACH(t, u->meta.names, i) {
  79                 char *path;
  80                 char **p;
  81
  82                 STRV_FOREACH(p, u->meta.manager->unit_path) {
  83
  84                         if (asprintf(&path, "%s/%s.wants", *p, t) < 0)
  85                                 return -ENOMEM;
  86
  87                         r = iterate_dir(u, path);
  88                         free(path);
  89
  90                         if (r < 0)
  91                                 return r;
  92
  93                         if (u->meta.instance) {
  94                                 char *template;
  95                                 /* Also try the template dir */
  96
  97                                 if (!(template = unit_name_template(t)))
  98                                         return -ENOMEM;
  99
 100                                 r = asprintf(&path, "%s/%s.wants", *p, template);
 101                                 free(template);
 102
 103                                 if (r < 0)
 104                                         return -ENOMEM;
 105
 106                                 r = iterate_dir(u, path);
 107                                 free(path);
 108
 109                                 if (r < 0)
 110                                         return r;
 111                         }
 112
 113                 }
 114         }
 115
 116         return 0;
 117 }