src/basic/conf-files.c

   1 /***
   2   This file is part of systemd.
   3
   4   Copyright 2010 Lennart Poettering
   5
   6   systemd is free software; you can redistribute it and/or modify it
   7   under the terms of the GNU Lesser General Public License as published by
   8   the Free Software Foundation; either version 2.1 of the License, or
   9   (at your option) any later version.
  10
  11   systemd is distributed in the hope that it will be useful, but
  12   WITHOUT ANY WARRANTY; without even the implied warranty of
  13   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  14   Lesser General Public License for more details.
  15
  16   You should have received a copy of the GNU Lesser General Public License
  17   along with systemd; If not, see <http://www.gnu.org/licenses/>.
  18 ***/
  19
  20 #include <dirent.h>
  21 #include <errno.h>
  22 #include <stdarg.h>
  23 #include <stdio.h>
  24 #include <stdlib.h>
  25 #include <string.h>
  26
  27 #include "conf-files.h"
  28 #include "dirent-util.h"
  29 #include "fd-util.h"
  30 #include "hashmap.h"
  31 #include "log.h"
  32 #include "macro.h"
  33 #include "missing.h"
  34 #include "path-util.h"
  35 #include "string-util.h"
  36 #include "strv.h"
  37 #include "util.h"
  38
  39 static int files_add(Hashmap *h, const char *root, const char *path, const char *suffix) {
  40         _cleanup_closedir_ DIR *dir = NULL;
  41         const char *dirpath;
  42         struct dirent *de;
  43         int r;
  44
  45         assert(path);
  46
  47         dirpath = prefix_roota(root, path);
  48
  49         dir = opendir(dirpath);
  50         if (!dir) {
  51                 if (errno == ENOENT)
  52                         return 0;
  53                 return -errno;
  54         }
  55
  56         FOREACH_DIRENT(de, dir, return -errno) {
  57                 char *p;
  58
  59                 if (!dirent_is_file_with_suffix(de, suffix))
  60                         continue;
  61
  62                 p = strjoin(dirpath, "/", de->d_name);
  63                 if (!p)
  64                         return -ENOMEM;
  65
  66                 r = hashmap_put(h, basename(p), p);
  67                 if (r == -EEXIST) {
  68                         log_debug("Skipping overridden file: %s.", p);
  69                         free(p);
  70                 } else if (r < 0) {
  71                         free(p);
  72                         return r;
  73                 } else if (r == 0) {
  74                         log_debug("Duplicate file %s", p);
  75                         free(p);
  76                 }
  77         }
  78
  79         return 0;
  80 }
  81
  82 static int base_cmp(const void *a, const void *b) {
  83         const char *s1, *s2;
  84
  85         s1 = *(char * const *)a;
  86         s2 = *(char * const *)b;
  87         return strcmp(basename(s1), basename(s2));
  88 }
  89
  90 static int conf_files_list_strv_internal(char ***strv, const char *suffix, const char *root, char **dirs) {
  91         _cleanup_hashmap_free_ Hashmap *fh = NULL;
  92         char **files, **p;
  93         int r;
  94
  95         assert(strv);
  96
  97         /* This alters the dirs string array */
  98         if (!path_strv_resolve_uniq(dirs, root))
  99                 return -ENOMEM;
 100
 101         fh = hashmap_new(&string_hash_ops);
 102         if (!fh)
 103                 return -ENOMEM;
 104
 105         STRV_FOREACH(p, dirs) {
 106                 r = files_add(fh, root, *p, suffix);
 107                 if (r == -ENOMEM)
 108                         return r;
 109                 if (r < 0)
 110                         log_debug_errno(r, "Failed to search for files in %s, ignoring: %m", *p);
 111         }
 112
 113         files = hashmap_get_strv(fh);
 114         if (!files)
 115                 return -ENOMEM;
 116
 117         qsort_safe(files, hashmap_size(fh), sizeof(char *), base_cmp);
 118         *strv = files;
 119
 120         return 0;
 121 }
 122
 123 int conf_files_list_strv(char ***strv, const char *suffix, const char *root, const char* const* dirs) {
 124         _cleanup_strv_free_ char **copy = NULL;
 125
 126         assert(strv);
 127
 128         copy = strv_copy((char**) dirs);
 129         if (!copy)
 130                 return -ENOMEM;
 131
 132         return conf_files_list_strv_internal(strv, suffix, root, copy);
 133 }
 134
 135 int conf_files_list(char ***strv, const char *suffix, const char *root, const char *dir, ...) {
 136         _cleanup_strv_free_ char **dirs = NULL;
 137         va_list ap;
 138
 139         assert(strv);
 140
 141         va_start(ap, dir);
 142         dirs = strv_new_ap(dir, ap);
 143         va_end(ap);
 144
 145         if (!dirs)
 146                 return -ENOMEM;
 147
 148         return conf_files_list_strv_internal(strv, suffix, root, dirs);
 149 }
 150
 151 int conf_files_list_nulstr(char ***strv, const char *suffix, const char *root, const char *d) {
 152         _cleanup_strv_free_ char **dirs = NULL;
 153
 154         assert(strv);
 155
 156         dirs = strv_split_nulstr(d);
 157         if (!dirs)
 158                 return -ENOMEM;
 159
 160         return conf_files_list_strv_internal(strv, suffix, root, dirs);
 161 }