src/shared/conf-files.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 Lesser General Public License as published by
  10   the Free Software Foundation; either version 2.1 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   Lesser General Public License for more details.
  17
  18   You should have received a copy of the GNU Lesser General Public License
  19   along with systemd; If not, see <http://www.gnu.org/licenses/>.
  20 ***/
  21
  22 #include <assert.h>
  23 #include <string.h>
  24 #include <unistd.h>
  25 #include <errno.h>
  26 #include <stdlib.h>
  27 #include <stdio.h>
  28 #include <sys/stat.h>
  29 #include <dirent.h>
  30
  31 #include "macro.h"
  32 #include "util.h"
  33 #include "missing.h"
  34 #include "log.h"
  35 #include "strv.h"
  36 #include "path-util.h"
  37 #include "hashmap.h"
  38 #include "conf-files.h"
  39
  40 static int files_add(Hashmap *h, const char *root, const char *path, const char *suffix) {
  41         _cleanup_closedir_ DIR *dir = NULL;
  42         _cleanup_free_ char *dirpath = NULL;
  43
  44         if (asprintf(&dirpath, "%s%s", root ? root : "", path) < 0)
  45                 return -ENOMEM;
  46
  47         dir = opendir(dirpath);
  48         if (!dir) {
  49                 if (errno == ENOENT)
  50                         return 0;
  51                 return -errno;
  52         }
  53
  54         for (;;) {
  55                 struct dirent *de;
  56                 union dirent_storage buf;
  57                 char *p;
  58                 int r;
  59
  60                 r = readdir_r(dir, &buf.de, &de);
  61                 if (r != 0)
  62                         return -r;
  63
  64                 if (!de)
  65                         break;
  66
  67                 if (!dirent_is_file_with_suffix(de, suffix))
  68                         continue;
  69
  70                 p = strjoin(dirpath, "/", de->d_name, NULL);
  71                 if (!p)
  72                         return -ENOMEM;
  73
  74                 r = hashmap_put(h, path_get_file_name(p), p);
  75                 if (r == -EEXIST) {
  76                         log_debug("Skipping overridden file: %s.", p);
  77                         free(p);
  78                 } else if (r < 0) {
  79                         free(p);
  80                         return r;
  81                 } else if (r == 0) {
  82                         log_debug("Duplicate file %s", p);
  83                         free(p);
  84                 }
  85         }
  86
  87         return 0;
  88 }
  89
  90 static int base_cmp(const void *a, const void *b) {
  91         const char *s1, *s2;
  92
  93         s1 = *(char * const *)a;
  94         s2 = *(char * const *)b;
  95         return strcmp(path_get_file_name(s1), path_get_file_name(s2));
  96 }
  97
  98 static int conf_files_list_strv_internal(char ***strv, const char *suffix, const char *root, char **dirs) {
  99         Hashmap *fh;
 100         char **files, **p;
 101         int r;
 102
 103         assert(strv);
 104         assert(suffix);
 105
 106         /* This alters the dirs string array */
 107         if (!path_strv_canonicalize_uniq(dirs))
 108                 return -ENOMEM;
 109
 110         fh = hashmap_new(string_hash_func, string_compare_func);
 111         if (!fh)
 112                 return -ENOMEM;
 113
 114         STRV_FOREACH(p, dirs) {
 115                 r = files_add(fh, root, *p, suffix);
 116                 if (r == -ENOMEM) {
 117                         hashmap_free_free(fh);
 118                         return r;
 119                 } else if (r < 0)
 120                         log_debug("Failed to search for files in %s: %s",
 121                                   *p, strerror(-r));
 122         }
 123
 124         files = hashmap_get_strv(fh);
 125         if (files == NULL) {
 126                 hashmap_free_free(fh);
 127                 return -ENOMEM;
 128         }
 129
 130         qsort_safe(files, hashmap_size(fh), sizeof(char *), base_cmp);
 131         *strv = files;
 132
 133         hashmap_free(fh);
 134         return 0;
 135 }
 136
 137 int conf_files_list_strv(char ***strv, const char *suffix, const char *root, const char* const* dirs) {
 138         _cleanup_strv_free_ char **copy = NULL;
 139
 140         assert(strv);
 141         assert(suffix);
 142
 143         copy = strv_copy((char**) dirs);
 144         if (!copy)
 145                 return -ENOMEM;
 146
 147         return conf_files_list_strv_internal(strv, suffix, root, copy);
 148 }
 149
 150 int conf_files_list(char ***strv, const char *suffix, const char *root, const char *dir, ...) {
 151         _cleanup_strv_free_ char **dirs = NULL;
 152         va_list ap;
 153
 154         assert(strv);
 155         assert(suffix);
 156
 157         va_start(ap, dir);
 158         dirs = strv_new_ap(dir, ap);
 159         va_end(ap);
 160
 161         if (!dirs)
 162                 return -ENOMEM;
 163
 164         return conf_files_list_strv_internal(strv, suffix, root, dirs);
 165 }
 166
 167 int conf_files_list_nulstr(char ***strv, const char *suffix, const char *root, const char *d) {
 168         _cleanup_strv_free_ char **dirs = NULL;
 169
 170         assert(strv);
 171         assert(suffix);
 172
 173         dirs = strv_split_nulstr(d);
 174         if (!dirs)
 175                 return -ENOMEM;
 176
 177         return conf_files_list_strv_internal(strv, suffix, root, dirs);
 178 }