src/delta/delta.c

   1 /* SPDX-License-Identifier: LGPL-2.1+ */
   2
   3 #include <errno.h>
   4 #include <getopt.h>
   5 #include <string.h>
   6 #include <sys/prctl.h>
   7 #include <unistd.h>
   8
   9 #include "alloc-util.h"
  10 #include "dirent-util.h"
  11 #include "fd-util.h"
  12 #include "fs-util.h"
  13 #include "hashmap.h"
  14 #include "locale-util.h"
  15 #include "log.h"
  16 #include "main-func.h"
  17 #include "pager.h"
  18 #include "parse-util.h"
  19 #include "path-util.h"
  20 #include "pretty-print.h"
  21 #include "process-util.h"
  22 #include "signal-util.h"
  23 #include "stat-util.h"
  24 #include "string-util.h"
  25 #include "strv.h"
  26 #include "terminal-util.h"
  27 #include "util.h"
  28
  29 static const char prefixes[] =
  30         "/etc\0"
  31         "/run\0"
  32         "/usr/local/lib\0"
  33         "/usr/local/share\0"
  34         "/usr/lib\0"
  35         "/usr/share\0"
  36 #if HAVE_SPLIT_USR
  37         "/lib\0"
  38 #endif
  39         ;
  40
  41 static const char suffixes[] =
  42         "sysctl.d\0"
  43         "tmpfiles.d\0"
  44         "modules-load.d\0"
  45         "binfmt.d\0"
  46         "systemd/system\0"
  47         "systemd/user\0"
  48         "systemd/system-preset\0"
  49         "systemd/user-preset\0"
  50         "udev/rules.d\0"
  51         "modprobe.d\0";
  52
  53 static const char have_dropins[] =
  54         "systemd/system\0"
  55         "systemd/user\0";
  56
  57 static PagerFlags arg_pager_flags = 0;
  58 static int arg_diff = -1;
  59
  60 static enum {
  61         SHOW_MASKED     = 1 << 0,
  62         SHOW_EQUIVALENT = 1 << 1,
  63         SHOW_REDIRECTED = 1 << 2,
  64         SHOW_OVERRIDDEN = 1 << 3,
  65         SHOW_UNCHANGED  = 1 << 4,
  66         SHOW_EXTENDED   = 1 << 5,
  67
  68         SHOW_DEFAULTS =
  69         (SHOW_MASKED | SHOW_EQUIVALENT | SHOW_REDIRECTED | SHOW_OVERRIDDEN | SHOW_EXTENDED)
  70 } arg_flags = 0;
  71
  72 static int equivalent(const char *a, const char *b) {
  73         _cleanup_free_ char *x = NULL, *y = NULL;
  74         int r;
  75
  76         r = chase_symlinks(a, NULL, CHASE_TRAIL_SLASH, &x);
  77         if (r < 0)
  78                 return r;
  79
  80         r = chase_symlinks(b, NULL, CHASE_TRAIL_SLASH, &y);
  81         if (r < 0)
  82                 return r;
  83
  84         return path_equal(x, y);
  85 }
  86
  87 static int notify_override_masked(const char *top, const char *bottom) {
  88         if (!(arg_flags & SHOW_MASKED))
  89                 return 0;
  90
  91         printf("%s%s%s     %s %s %s\n",
  92                ansi_highlight_red(), "[MASKED]", ansi_normal(),
  93                top, special_glyph(SPECIAL_GLYPH_ARROW), bottom);
  94         return 1;
  95 }
  96
  97 static int notify_override_equivalent(const char *top, const char *bottom) {
  98         if (!(arg_flags & SHOW_EQUIVALENT))
  99                 return 0;
 100
 101         printf("%s%s%s %s %s %s\n",
 102                ansi_highlight_green(), "[EQUIVALENT]", ansi_normal(),
 103                top, special_glyph(SPECIAL_GLYPH_ARROW), bottom);
 104         return 1;
 105 }
 106
 107 static int notify_override_redirected(const char *top, const char *bottom) {
 108         if (!(arg_flags & SHOW_REDIRECTED))
 109                 return 0;
 110
 111         printf("%s%s%s %s %s %s\n",
 112                ansi_highlight(), "[REDIRECTED]", ansi_normal(),
 113                top, special_glyph(SPECIAL_GLYPH_ARROW), bottom);
 114         return 1;
 115 }
 116
 117 static int notify_override_overridden(const char *top, const char *bottom) {
 118         if (!(arg_flags & SHOW_OVERRIDDEN))
 119                 return 0;
 120
 121         printf("%s%s%s %s %s %s\n",
 122                ansi_highlight(), "[OVERRIDDEN]", ansi_normal(),
 123                top, special_glyph(SPECIAL_GLYPH_ARROW), bottom);
 124         return 1;
 125 }
 126
 127 static int notify_override_extended(const char *top, const char *bottom) {
 128         if (!(arg_flags & SHOW_EXTENDED))
 129                return 0;
 130
 131         printf("%s%s%s   %s %s %s\n",
 132                ansi_highlight(), "[EXTENDED]", ansi_normal(),
 133                top, special_glyph(SPECIAL_GLYPH_ARROW), bottom);
 134         return 1;
 135 }
 136
 137 static int notify_override_unchanged(const char *f) {
 138         if (!(arg_flags & SHOW_UNCHANGED))
 139                 return 0;
 140
 141         printf("[UNCHANGED]  %s\n", f);
 142         return 1;
 143 }
 144
 145 static int found_override(const char *top, const char *bottom) {
 146         _cleanup_free_ char *dest = NULL;
 147         pid_t pid;
 148         int r;
 149
 150         assert(top);
 151         assert(bottom);
 152
 153         if (null_or_empty_path(top) > 0)
 154                 return notify_override_masked(top, bottom);
 155
 156         r = readlink_malloc(top, &dest);
 157         if (r >= 0) {
 158                 if (equivalent(dest, bottom) > 0)
 159                         return notify_override_equivalent(top, bottom);
 160                 else
 161                         return notify_override_redirected(top, bottom);
 162         }
 163
 164         r = notify_override_overridden(top, bottom);
 165         if (!arg_diff)
 166                 return r;
 167
 168         putchar('\n');
 169
 170         fflush(stdout);
 171
 172         r = safe_fork("(diff)", FORK_RESET_SIGNALS|FORK_DEATHSIG|FORK_CLOSE_ALL_FDS|FORK_RLIMIT_NOFILE_SAFE|FORK_LOG, &pid);
 173         if (r < 0)
 174                 return r;
 175         if (r == 0) {
 176                 execlp("diff", "diff", "-us", "--", bottom, top, NULL);
 177                 log_open();
 178                 log_error_errno(errno, "Failed to execute diff: %m");
 179                 _exit(EXIT_FAILURE);
 180         }
 181
 182         (void) wait_for_terminate_and_check("diff", pid, WAIT_LOG_ABNORMAL);
 183         putchar('\n');
 184
 185         return r;
 186 }
 187
 188 static int enumerate_dir_d(
 189                 OrderedHashmap *top,
 190                 OrderedHashmap *bottom,
 191                 OrderedHashmap *drops,
 192                 const char *toppath, const char *drop) {
 193
 194         _cleanup_free_ char *unit = NULL;
 195         _cleanup_free_ char *path = NULL;
 196         _cleanup_strv_free_ char **list = NULL;
 197         char **file;
 198         char *c;
 199         int r;
 200
 201         assert(!endswith(drop, "/"));
 202
 203         path = strjoin(toppath, "/", drop);
 204         if (!path)
 205                 return -ENOMEM;
 206
 207         log_debug("Looking at %s", path);
 208
 209         unit = strdup(drop);
 210         if (!unit)
 211                 return -ENOMEM;
 212
 213         c = strrchr(unit, '.');
 214         if (!c)
 215                 return -EINVAL;
 216         *c = 0;
 217
 218         r = get_files_in_directory(path, &list);
 219         if (r < 0)
 220                 return log_error_errno(r, "Failed to enumerate %s: %m", path);
 221
 222         strv_sort(list);
 223
 224         STRV_FOREACH(file, list) {
 225                 OrderedHashmap *h;
 226                 int k;
 227                 char *p;
 228                 char *d;
 229
 230                 if (!endswith(*file, ".conf"))
 231                         continue;
 232
 233                 p = strjoin(path, "/", *file);
 234                 if (!p)
 235                         return -ENOMEM;
 236                 d = p + strlen(toppath) + 1;
 237
 238                 log_debug("Adding at top: %s %s %s", d, special_glyph(SPECIAL_GLYPH_ARROW), p);
 239                 k = ordered_hashmap_put(top, d, p);
 240                 if (k >= 0) {
 241                         p = strdup(p);
 242                         if (!p)
 243                                 return -ENOMEM;
 244                         d = p + strlen(toppath) + 1;
 245                 } else if (k != -EEXIST) {
 246                         free(p);
 247                         return k;
 248                 }
 249
 250                 log_debug("Adding at bottom: %s %s %s", d, special_glyph(SPECIAL_GLYPH_ARROW), p);
 251                 free(ordered_hashmap_remove(bottom, d));
 252                 k = ordered_hashmap_put(bottom, d, p);
 253                 if (k < 0) {
 254                         free(p);
 255                         return k;
 256                 }
 257
 258                 h = ordered_hashmap_get(drops, unit);
 259                 if (!h) {
 260                         h = ordered_hashmap_new(&string_hash_ops);
 261                         if (!h)
 262                                 return -ENOMEM;
 263                         ordered_hashmap_put(drops, unit, h);
 264                         unit = strdup(unit);
 265                         if (!unit)
 266                                 return -ENOMEM;
 267                 }
 268
 269                 p = strdup(p);
 270                 if (!p)
 271                         return -ENOMEM;
 272
 273                 log_debug("Adding to drops: %s %s %s %s %s",
 274                           unit, special_glyph(SPECIAL_GLYPH_ARROW), basename(p), special_glyph(SPECIAL_GLYPH_ARROW), p);
 275                 k = ordered_hashmap_put(h, basename(p), p);
 276                 if (k < 0) {
 277                         free(p);
 278                         if (k != -EEXIST)
 279                                 return k;
 280                 }
 281         }
 282         return 0;
 283 }
 284
 285 static int enumerate_dir(
 286                 OrderedHashmap *top,
 287                 OrderedHashmap *bottom,
 288                 OrderedHashmap *drops,
 289                 const char *path, bool dropins) {
 290
 291         _cleanup_closedir_ DIR *d = NULL;
 292         struct dirent *de;
 293         _cleanup_strv_free_ char **files = NULL, **dirs = NULL;
 294         size_t n_files = 0, allocated_files = 0, n_dirs = 0, allocated_dirs = 0;
 295         char **t;
 296         int r;
 297
 298         assert(top);
 299         assert(bottom);
 300         assert(drops);
 301         assert(path);
 302
 303         log_debug("Looking at %s", path);
 304
 305         d = opendir(path);
 306         if (!d) {
 307                 if (errno == ENOENT)
 308                         return 0;
 309
 310                 return log_error_errno(errno, "Failed to open %s: %m", path);
 311         }
 312
 313         FOREACH_DIRENT_ALL(de, d, return -errno) {
 314                 dirent_ensure_type(d, de);
 315
 316                 if (dropins && de->d_type == DT_DIR && endswith(de->d_name, ".d")) {
 317                         if (!GREEDY_REALLOC0(dirs, allocated_dirs, n_dirs + 2))
 318                                 return -ENOMEM;
 319
 320                         dirs[n_dirs] = strdup(de->d_name);
 321                         if (!dirs[n_dirs])
 322                                 return -ENOMEM;
 323                         n_dirs ++;
 324                 }
 325
 326                 if (!dirent_is_file(de))
 327                         continue;
 328
 329                 if (!GREEDY_REALLOC0(files, allocated_files, n_files + 2))
 330                         return -ENOMEM;
 331
 332                 files[n_files] = strdup(de->d_name);
 333                 if (!files[n_files])
 334                         return -ENOMEM;
 335                 n_files ++;
 336         }
 337
 338         strv_sort(dirs);
 339         strv_sort(files);
 340
 341         STRV_FOREACH(t, dirs) {
 342                 r = enumerate_dir_d(top, bottom, drops, path, *t);
 343                 if (r < 0)
 344                         return r;
 345         }
 346
 347         STRV_FOREACH(t, files) {
 348                 _cleanup_free_ char *p = NULL;
 349
 350                 p = strjoin(path, "/", *t);
 351                 if (!p)
 352                         return -ENOMEM;
 353
 354                 log_debug("Adding at top: %s %s %s", basename(p), special_glyph(SPECIAL_GLYPH_ARROW), p);
 355                 r = ordered_hashmap_put(top, basename(p), p);
 356                 if (r >= 0) {
 357                         p = strdup(p);
 358                         if (!p)
 359                                 return -ENOMEM;
 360                 } else if (r != -EEXIST)
 361                         return r;
 362
 363                 log_debug("Adding at bottom: %s %s %s", basename(p), special_glyph(SPECIAL_GLYPH_ARROW), p);
 364                 free(ordered_hashmap_remove(bottom, basename(p)));
 365                 r = ordered_hashmap_put(bottom, basename(p), p);
 366                 if (r < 0)
 367                         return r;
 368                 p = NULL;
 369         }
 370
 371         return 0;
 372 }
 373
 374 static bool should_skip_path(const char *prefix, const char *suffix) {
 375 #if HAVE_SPLIT_USR
 376         _cleanup_free_ char *target = NULL;
 377         const char *p;
 378         char *dirname;
 379
 380         dirname = strjoina(prefix, "/", suffix);
 381
 382         if (chase_symlinks(dirname, NULL, 0, &target) < 0)
 383                 return false;
 384
 385         NULSTR_FOREACH(p, prefixes) {
 386                 if (path_startswith(dirname, p))
 387                         continue;
 388
 389                 if (path_equal(target, strjoina(p, "/", suffix))) {
 390                         log_debug("%s redirects to %s, skipping.", dirname, target);
 391                         return true;
 392                 }
 393         }
 394 #endif
 395         return false;
 396 }
 397
 398 static int process_suffix(const char *suffix, const char *onlyprefix) {
 399         const char *p;
 400         char *f;
 401         OrderedHashmap *top, *bottom, *drops;
 402         OrderedHashmap *h;
 403         char *key;
 404         int r = 0, k;
 405         Iterator i, j;
 406         int n_found = 0;
 407         bool dropins;
 408
 409         assert(suffix);
 410         assert(!startswith(suffix, "/"));
 411         assert(!strstr(suffix, "//"));
 412
 413         dropins = nulstr_contains(have_dropins, suffix);
 414
 415         top = ordered_hashmap_new(&string_hash_ops);
 416         bottom = ordered_hashmap_new(&string_hash_ops);
 417         drops = ordered_hashmap_new(&string_hash_ops);
 418         if (!top || !bottom || !drops) {
 419                 r = -ENOMEM;
 420                 goto finish;
 421         }
 422
 423         NULSTR_FOREACH(p, prefixes) {
 424                 _cleanup_free_ char *t = NULL;
 425
 426                 if (should_skip_path(p, suffix))
 427                         continue;
 428
 429                 t = strjoin(p, "/", suffix);
 430                 if (!t) {
 431                         r = -ENOMEM;
 432                         goto finish;
 433                 }
 434
 435                 k = enumerate_dir(top, bottom, drops, t, dropins);
 436                 if (r == 0)
 437                         r = k;
 438         }
 439
 440         ORDERED_HASHMAP_FOREACH_KEY(f, key, top, i) {
 441                 char *o;
 442
 443                 o = ordered_hashmap_get(bottom, key);
 444                 assert(o);
 445
 446                 if (!onlyprefix || startswith(o, onlyprefix)) {
 447                         if (path_equal(o, f)) {
 448                                 notify_override_unchanged(f);
 449                         } else {
 450                                 k = found_override(f, o);
 451                                 if (k < 0)
 452                                         r = k;
 453                                 else
 454                                         n_found += k;
 455                         }
 456                 }
 457
 458                 h = ordered_hashmap_get(drops, key);
 459                 if (h)
 460                         ORDERED_HASHMAP_FOREACH(o, h, j)
 461                                 if (!onlyprefix || startswith(o, onlyprefix))
 462                                         n_found += notify_override_extended(f, o);
 463         }
 464
 465 finish:
 466         ordered_hashmap_free_free(top);
 467         ordered_hashmap_free_free(bottom);
 468
 469         ORDERED_HASHMAP_FOREACH_KEY(h, key, drops, i) {
 470                 ordered_hashmap_free_free(ordered_hashmap_remove(drops, key));
 471                 ordered_hashmap_remove(drops, key);
 472                 free(key);
 473         }
 474         ordered_hashmap_free(drops);
 475
 476         return r < 0 ? r : n_found;
 477 }
 478
 479 static int process_suffixes(const char *onlyprefix) {
 480         const char *n;
 481         int n_found = 0, r;
 482
 483         NULSTR_FOREACH(n, suffixes) {
 484                 r = process_suffix(n, onlyprefix);
 485                 if (r < 0)
 486                         return r;
 487
 488                 n_found += r;
 489         }
 490
 491         return n_found;
 492 }
 493
 494 static int process_suffix_chop(const char *arg) {
 495         const char *p;
 496
 497         assert(arg);
 498
 499         if (!path_is_absolute(arg))
 500                 return process_suffix(arg, NULL);
 501
 502         /* Strip prefix from the suffix */
 503         NULSTR_FOREACH(p, prefixes) {
 504                 const char *suffix;
 505
 506                 suffix = startswith(arg, p);
 507                 if (suffix) {
 508                         suffix += strspn(suffix, "/");
 509                         if (*suffix)
 510                                 return process_suffix(suffix, p);
 511                         else
 512                                 return process_suffixes(arg);
 513                 }
 514         }
 515
 516         return log_error_errno(SYNTHETIC_ERRNO(EINVAL),
 517                                "Invalid suffix specification %s.", arg);
 518 }
 519
 520 static int help(void) {
 521         _cleanup_free_ char *link = NULL;
 522         int r;
 523
 524         r = terminal_urlify_man("systemd-delta", "1", &link);
 525         if (r < 0)
 526                 return log_oom();
 527
 528         printf("%s [OPTIONS...] [SUFFIX...]\n\n"
 529                "Find overridden configuration files.\n\n"
 530                "  -h --help           Show this help\n"
 531                "     --version        Show package version\n"
 532                "     --no-pager       Do not pipe output into a pager\n"
 533                "     --diff[=1|0]     Show a diff when overridden files differ\n"
 534                "  -t --type=LIST...   Only display a selected set of override types\n"
 535                "\nSee the %s for details.\n"
 536                , program_invocation_short_name
 537                , link
 538         );
 539
 540         return 0;
 541 }
 542
 543 static int parse_flags(const char *flag_str, int flags) {
 544         const char *word, *state;
 545         size_t l;
 546
 547         FOREACH_WORD_SEPARATOR(word, l, flag_str, ",", state) {
 548                 if (strneq("masked", word, l))
 549                         flags |= SHOW_MASKED;
 550                 else if (strneq ("equivalent", word, l))
 551                         flags |= SHOW_EQUIVALENT;
 552                 else if (strneq("redirected", word, l))
 553                         flags |= SHOW_REDIRECTED;
 554                 else if (strneq("overridden", word, l))
 555                         flags |= SHOW_OVERRIDDEN;
 556                 else if (strneq("unchanged", word, l))
 557                         flags |= SHOW_UNCHANGED;
 558                 else if (strneq("extended", word, l))
 559                         flags |= SHOW_EXTENDED;
 560                 else if (strneq("default", word, l))
 561                         flags |= SHOW_DEFAULTS;
 562                 else
 563                         return -EINVAL;
 564         }
 565         return flags;
 566 }
 567
 568 static int parse_argv(int argc, char *argv[]) {
 569
 570         enum {
 571                 ARG_NO_PAGER = 0x100,
 572                 ARG_DIFF,
 573                 ARG_VERSION
 574         };
 575
 576         static const struct option options[] = {
 577                 { "help",      no_argument,       NULL, 'h'          },
 578                 { "version",   no_argument,       NULL, ARG_VERSION  },
 579                 { "no-pager",  no_argument,       NULL, ARG_NO_PAGER },
 580                 { "diff",      optional_argument, NULL, ARG_DIFF     },
 581                 { "type",      required_argument, NULL, 't'          },
 582                 {}
 583         };
 584
 585         int c;
 586
 587         assert(argc >= 1);
 588         assert(argv);
 589
 590         while ((c = getopt_long(argc, argv, "ht:", options, NULL)) >= 0)
 591
 592                 switch (c) {
 593
 594                 case 'h':
 595                         return help();
 596
 597                 case ARG_VERSION:
 598                         return version();
 599
 600                 case ARG_NO_PAGER:
 601                         arg_pager_flags |= PAGER_DISABLE;
 602                         break;
 603
 604                 case 't': {
 605                         int f;
 606                         f = parse_flags(optarg, arg_flags);
 607                         if (f < 0)
 608                                 return log_error_errno(SYNTHETIC_ERRNO(EINVAL),
 609                                                        "Failed to parse flags field.");
 610                         arg_flags = f;
 611                         break;
 612                 }
 613
 614                 case ARG_DIFF:
 615                         if (!optarg)
 616                                 arg_diff = 1;
 617                         else {
 618                                 int b;
 619
 620                                 b = parse_boolean(optarg);
 621                                 if (b < 0)
 622                                         return log_error_errno(SYNTHETIC_ERRNO(EINVAL),
 623                                                                "Failed to parse diff boolean.");
 624
 625                                 arg_diff = b;
 626                         }
 627                         break;
 628
 629                 case '?':
 630                         return -EINVAL;
 631
 632                 default:
 633                         assert_not_reached("Unhandled option");
 634                 }
 635
 636         return 1;
 637 }
 638
 639 static int run(int argc, char *argv[]) {
 640         int r, k, n_found = 0;
 641
 642         log_parse_environment();
 643         log_open();
 644
 645         r = parse_argv(argc, argv);
 646         if (r <= 0)
 647                 return r;
 648
 649         if (arg_flags == 0)
 650                 arg_flags = SHOW_DEFAULTS;
 651
 652         if (arg_diff < 0)
 653                 arg_diff = !!(arg_flags & SHOW_OVERRIDDEN);
 654         else if (arg_diff)
 655                 arg_flags |= SHOW_OVERRIDDEN;
 656
 657         (void) pager_open(arg_pager_flags);
 658
 659         if (optind < argc) {
 660                 int i;
 661
 662                 for (i = optind; i < argc; i++) {
 663                         path_simplify(argv[i], false);
 664
 665                         k = process_suffix_chop(argv[i]);
 666                         if (k < 0)
 667                                 r = k;
 668                         else
 669                                 n_found += k;
 670                 }
 671
 672         } else {
 673                 k = process_suffixes(NULL);
 674                 if (k < 0)
 675                         r = k;
 676                 else
 677                         n_found += k;
 678         }
 679
 680         if (r >= 0)
 681                 printf("%s%i overridden configuration files found.\n", n_found ? "\n" : "", n_found);
 682         return r;
 683 }
 684
 685 DEFINE_MAIN_FUNCTION(run);