src/load-dropin.c

   1 /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
   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->lookup_paths.unit_path) {
  83
  84                         if (asprintf(&path, "%s/%s.wants", *p, t) < 0)
  85                                 return -ENOMEM;
  86
  87                         if (u->meta.manager->unit_path_cache &&
  88                             !set_get(u->meta.manager->unit_path_cache, path))
  89                                 r = 0;
  90                         else
  91                                 r = iterate_dir(u, path);
  92                         free(path);
  93
  94                         if (r < 0)
  95                                 return r;
  96
  97                         if (u->meta.instance) {
  98                                 char *template;
  99                                 /* Also try the template dir */
 100
 101                                 if (!(template = unit_name_template(t)))
 102                                         return -ENOMEM;
 103
 104                                 r = asprintf(&path, "%s/%s.wants", *p, template);
 105                                 free(template);
 106
 107                                 if (r < 0)
 108                                         return -ENOMEM;
 109
 110                                 if (u->meta.manager->unit_path_cache &&
 111                                     !set_get(u->meta.manager->unit_path_cache, path))
 112                                         r = 0;
 113                                 else
 114                                         r = iterate_dir(u, path);
 115                                 free(path);
 116
 117                                 if (r < 0)
 118                                         return r;
 119                         }
 120
 121                 }
 122         }
 123
 124         return 0;
 125 }