src/basic/conf-files.c

   1 /* SPDX-License-Identifier: LGPL-2.1+ */
   2
   3 #include <errno.h>
   4 #include <stdarg.h>
   5 #include <stdio.h>
   6 #include <stdlib.h>
   7
   8 #include "conf-files.h"
   9 #include "def.h"
  10 #include "dirent-util.h"
  11 #include "fd-util.h"
  12 #include "hashmap.h"
  13 #include "log.h"
  14 #include "macro.h"
  15 #include "path-util.h"
  16 #include "set.h"
  17 #include "sort-util.h"
  18 #include "stat-util.h"
  19 #include "string-util.h"
  20 #include "strv.h"
  21 #include "terminal-util.h"
  22
  23 static int files_add(
  24                 Hashmap *h,
  25                 Set *masked,
  26                 const char *suffix,
  27                 const char *root,
  28                 unsigned flags,
  29                 const char *path) {
  30
  31         _cleanup_closedir_ DIR *dir = NULL;
  32         const char *dirpath;
  33         struct dirent *de;
  34         int r;
  35
  36         assert(h);
  37         assert((flags & CONF_FILES_FILTER_MASKED) == 0 || masked);
  38         assert(path);
  39
  40         dirpath = prefix_roota(root, path);
  41
  42         dir = opendir(dirpath);
  43         if (!dir) {
  44                 if (errno == ENOENT)
  45                         return 0;
  46
  47                 return log_debug_errno(errno, "Failed to open directory '%s': %m", dirpath);
  48         }
  49
  50         FOREACH_DIRENT(de, dir, return -errno) {
  51                 struct stat st;
  52                 char *p, *key;
  53
  54                 /* Does this match the suffix? */
  55                 if (suffix && !endswith(de->d_name, suffix))
  56                         continue;
  57
  58                 /* Has this file already been found in an earlier directory? */
  59                 if (hashmap_contains(h, de->d_name)) {
  60                         log_debug("Skipping overridden file '%s/%s'.", dirpath, de->d_name);
  61                         continue;
  62                 }
  63
  64                 /* Has this been masked in an earlier directory? */
  65                 if ((flags & CONF_FILES_FILTER_MASKED) && set_contains(masked, de->d_name)) {
  66                         log_debug("File '%s/%s' is masked by previous entry.", dirpath, de->d_name);
  67                         continue;
  68                 }
  69
  70                 /* Read file metadata if we shall validate the check for file masks, for node types or whether the node is marked executable. */
  71                 if (flags & (CONF_FILES_FILTER_MASKED|CONF_FILES_REGULAR|CONF_FILES_DIRECTORY|CONF_FILES_EXECUTABLE))
  72                         if (fstatat(dirfd(dir), de->d_name, &st, 0) < 0) {
  73                                 log_debug_errno(errno, "Failed to stat '%s/%s', ignoring: %m", dirpath, de->d_name);
  74                                 continue;
  75                         }
  76
  77                 /* Is this a masking entry? */
  78                 if ((flags & CONF_FILES_FILTER_MASKED))
  79                         if (null_or_empty(&st)) {
  80                                 assert(masked);
  81
  82                                 /* Mark this one as masked */
  83                                 r = set_put_strdup(&masked, de->d_name);
  84                                 if (r < 0)
  85                                         return r;
  86
  87                                 log_debug("File '%s/%s' is a mask.", dirpath, de->d_name);
  88                                 continue;
  89                         }
  90
  91                 /* Does this node have the right type? */
  92                 if (flags & (CONF_FILES_REGULAR|CONF_FILES_DIRECTORY))
  93                         if (!((flags & CONF_FILES_DIRECTORY) && S_ISDIR(st.st_mode)) &&
  94                             !((flags & CONF_FILES_REGULAR) && S_ISREG(st.st_mode))) {
  95                                 log_debug("Ignoring '%s/%s', as it is not a of the right type.", dirpath, de->d_name);
  96                                 continue;
  97                         }
  98
  99                 /* Does this node have the executable bit set? */
 100                 if (flags & CONF_FILES_EXECUTABLE)
 101                         /* As requested: check if the file is marked executable. Note that we don't check access(X_OK)
 102                          * here, as we care about whether the file is marked executable at all, and not whether it is
 103                          * executable for us, because if so, such errors are stuff we should log about. */
 104
 105                         if ((st.st_mode & 0111) == 0) { /* not executable */
 106                                 log_debug("Ignoring '%s/%s', as it is not marked executable.", dirpath, de->d_name);
 107                                 continue;
 108                         }
 109
 110                 if (flags & CONF_FILES_BASENAME) {
 111                         p = strdup(de->d_name);
 112                         if (!p)
 113                                 return -ENOMEM;
 114
 115                         key = p;
 116                 } else {
 117                         p = path_join(dirpath, de->d_name);
 118                         if (!p)
 119                                 return -ENOMEM;
 120
 121                         key = basename(p);
 122                 }
 123
 124                 r = hashmap_put(h, key, p);
 125                 if (r < 0) {
 126                         free(p);
 127                         return log_debug_errno(r, "Failed to add item to hashmap: %m");
 128                 }
 129
 130                 assert(r > 0);
 131         }
 132
 133         return 0;
 134 }
 135
 136 static int base_cmp(char * const *a, char * const *b) {
 137         return strcmp(basename(*a), basename(*b));
 138 }
 139
 140 static int conf_files_list_strv_internal(char ***strv, const char *suffix, const char *root, unsigned flags, char **dirs) {
 141         _cleanup_hashmap_free_ Hashmap *fh = NULL;
 142         _cleanup_set_free_free_ Set *masked = NULL;
 143         char **files, **p;
 144         int r;
 145
 146         assert(strv);
 147
 148         /* This alters the dirs string array */
 149         if (!path_strv_resolve_uniq(dirs, root))
 150                 return -ENOMEM;
 151
 152         fh = hashmap_new(&path_hash_ops);
 153         if (!fh)
 154                 return -ENOMEM;
 155
 156         if (flags & CONF_FILES_FILTER_MASKED) {
 157                 masked = set_new(&path_hash_ops);
 158                 if (!masked)
 159                         return -ENOMEM;
 160         }
 161
 162         STRV_FOREACH(p, dirs) {
 163                 r = files_add(fh, masked, suffix, root, flags, *p);
 164                 if (r == -ENOMEM)
 165                         return r;
 166                 if (r < 0)
 167                         log_debug_errno(r, "Failed to search for files in %s, ignoring: %m", *p);
 168         }
 169
 170         files = hashmap_get_strv(fh);
 171         if (!files)
 172                 return -ENOMEM;
 173
 174         typesafe_qsort(files, hashmap_size(fh), base_cmp);
 175         *strv = files;
 176
 177         return 0;
 178 }
 179
 180 int conf_files_insert(char ***strv, const char *root, char **dirs, const char *path) {
 181         /* Insert a path into strv, at the place honouring the usual sorting rules:
 182          * - we first compare by the basename
 183          * - and then we compare by dirname, allowing just one file with the given
 184          *   basename.
 185          * This means that we will
 186          * - add a new entry if basename(path) was not on the list,
 187          * - do nothing if an entry with higher priority was already present,
 188          * - do nothing if our new entry matches the existing entry,
 189          * - replace the existing entry if our new entry has higher priority.
 190          */
 191         size_t i, n;
 192         char *t;
 193         int r;
 194
 195         n = strv_length(*strv);
 196         for (i = 0; i < n; i++) {
 197                 int c;
 198
 199                 c = base_cmp((char* const*) *strv + i, (char* const*) &path);
 200                 if (c == 0) {
 201                         char **dir;
 202
 203                         /* Oh, there already is an entry with a matching name (the last component). */
 204
 205                         STRV_FOREACH(dir, dirs) {
 206                                 _cleanup_free_ char *rdir = NULL;
 207                                 char *p1, *p2;
 208
 209                                 rdir = path_join(root, *dir);
 210                                 if (!rdir)
 211                                         return -ENOMEM;
 212
 213                                 p1 = path_startswith((*strv)[i], rdir);
 214                                 if (p1)
 215                                         /* Existing entry with higher priority
 216                                          * or same priority, no need to do anything. */
 217                                         return 0;
 218
 219                                 p2 = path_startswith(path, *dir);
 220                                 if (p2) {
 221                                         /* Our new entry has higher priority */
 222
 223                                         t = path_join(root, path);
 224                                         if (!t)
 225                                                 return log_oom();
 226
 227                                         return free_and_replace((*strv)[i], t);
 228                                 }
 229                         }
 230
 231                 } else if (c > 0)
 232                         /* Following files have lower priority, let's go insert our
 233                          * new entry. */
 234                         break;
 235
 236                 /* … we are not there yet, let's continue */
 237         }
 238
 239         /* The new file has lower priority than all the existing entries */
 240         t = path_join(root, path);
 241         if (!t)
 242                 return -ENOMEM;
 243
 244         r = strv_insert(strv, i, t);
 245         if (r < 0)
 246                 free(t);
 247
 248         return r;
 249 }
 250
 251 int conf_files_list_strv(char ***strv, const char *suffix, const char *root, unsigned flags, const char* const* dirs) {
 252         _cleanup_strv_free_ char **copy = NULL;
 253
 254         assert(strv);
 255
 256         copy = strv_copy((char**) dirs);
 257         if (!copy)
 258                 return -ENOMEM;
 259
 260         return conf_files_list_strv_internal(strv, suffix, root, flags, copy);
 261 }
 262
 263 int conf_files_list(char ***strv, const char *suffix, const char *root, unsigned flags, const char *dir) {
 264         _cleanup_strv_free_ char **dirs = NULL;
 265
 266         assert(strv);
 267
 268         dirs = strv_new(dir);
 269         if (!dirs)
 270                 return -ENOMEM;
 271
 272         return conf_files_list_strv_internal(strv, suffix, root, flags, dirs);
 273 }
 274
 275 int conf_files_list_nulstr(char ***strv, const char *suffix, const char *root, unsigned flags, const char *dirs) {
 276         _cleanup_strv_free_ char **d = NULL;
 277
 278         assert(strv);
 279
 280         d = strv_split_nulstr(dirs);
 281         if (!d)
 282                 return -ENOMEM;
 283
 284         return conf_files_list_strv_internal(strv, suffix, root, flags, d);
 285 }
 286
 287 int conf_files_list_with_replacement(
 288                 const char *root,
 289                 char **config_dirs,
 290                 const char *replacement,
 291                 char ***files,
 292                 char **replace_file) {
 293
 294         _cleanup_strv_free_ char **f = NULL;
 295         _cleanup_free_ char *p = NULL;
 296         int r;
 297
 298         assert(config_dirs);
 299         assert(files);
 300         assert(replace_file || !replacement);
 301
 302         r = conf_files_list_strv(&f, ".conf", root, 0, (const char* const*) config_dirs);
 303         if (r < 0)
 304                 return log_error_errno(r, "Failed to enumerate config files: %m");
 305
 306         if (replacement) {
 307                 r = conf_files_insert(&f, root, config_dirs, replacement);
 308                 if (r < 0)
 309                         return log_error_errno(r, "Failed to extend config file list: %m");
 310
 311                 p = path_join(root, replacement);
 312                 if (!p)
 313                         return log_oom();
 314         }
 315
 316         *files = TAKE_PTR(f);
 317         if (replace_file)
 318                 *replace_file = TAKE_PTR(p);
 319         return 0;
 320 }