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 *path, const char *suffix) {
  41         DIR *dir;
  42         int r = 0;
  43
  44         dir = opendir(path);
  45         if (!dir) {
  46                 if (errno == ENOENT)
  47                         return 0;
  48                 return -errno;
  49         }
  50
  51         for (;;) {
  52                 struct dirent *de;
  53                 union dirent_storage buf;
  54                 int k;
  55                 char *p;
  56
  57                 k = readdir_r(dir, &buf.de, &de);
  58                 if (k != 0) {
  59                         r = -k;
  60                         goto finish;
  61                 }
  62
  63                 if (!de)
  64                         break;
  65
  66                 if (!dirent_is_file_with_suffix(de, suffix))
  67                         continue;
  68
  69                 if (asprintf(&p, "%s/%s", path, de->d_name) < 0) {
  70                         r = -ENOMEM;
  71                         goto finish;
  72                 }
  73
  74                 if (hashmap_put(h, path_get_file_name(p), p) <= 0) {
  75                         log_debug("Skip overridden file: %s.", p);
  76                         free(p);
  77                 }
  78         }
  79
  80 finish:
  81         closedir(dir);
  82         return r;
  83 }
  84
  85 static int base_cmp(const void *a, const void *b) {
  86         const char *s1, *s2;
  87
  88         s1 = *(char * const *)a;
  89         s2 = *(char * const *)b;
  90         return strcmp(path_get_file_name(s1), path_get_file_name(s2));
  91 }
  92
  93 int conf_files_list_strv(char ***strv, const char *suffix, const char **dirs) {
  94         Hashmap *fh = NULL;
  95         char **files = NULL;
  96         const char **p;
  97         int r;
  98
  99         assert(dirs);
 100
 101         fh = hashmap_new(string_hash_func, string_compare_func);
 102         if (!fh) {
 103                 r = -ENOMEM;
 104                 goto finish;
 105         }
 106
 107         STRV_FOREACH(p, dirs) {
 108                 r = files_add(fh, *p, suffix);
 109                 if (r < 0)
 110                         log_warning("Failed to search for files in %s: %s",
 111                                     *p, strerror(-r));
 112         }
 113
 114         files = hashmap_get_strv(fh);
 115         if (files == NULL) {
 116                 log_error("Failed to compose list of files.");
 117                 r = -ENOMEM;
 118                 goto finish;
 119         }
 120         qsort(files, hashmap_size(fh), sizeof(char *), base_cmp);
 121         r = 0;
 122
 123 finish:
 124         hashmap_free(fh);
 125         *strv = files;
 126         return r;
 127 }
 128
 129 int conf_files_list(char ***strv, const char *suffix, const char *dir, ...) {
 130         char **dirs = NULL;
 131         va_list ap;
 132         int r;
 133
 134         va_start(ap, dir);
 135         dirs = strv_new_ap(dir, ap);
 136         va_end(ap);
 137         if (!dirs) {
 138                 r = -ENOMEM;
 139                 goto finish;
 140         }
 141
 142         if (!path_strv_canonicalize(dirs)) {
 143                 r = -ENOMEM;
 144                 goto finish;
 145         }
 146         strv_uniq(dirs);
 147
 148         r = conf_files_list_strv(strv, suffix, (const char **)dirs);
 149
 150 finish:
 151         strv_free(dirs);
 152         return r;
 153 }