src/shared/install.c

   1 /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
   2
   3 /***
   4   This file is part of systemd.
   5
   6   Copyright 2011 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 <errno.h>
  23 #include <fcntl.h>
  24 #include <unistd.h>
  25 #include <string.h>
  26 #include <fnmatch.h>
  27
  28 #include "util.h"
  29 #include "mkdir.h"
  30 #include "hashmap.h"
  31 #include "set.h"
  32 #include "path-util.h"
  33 #include "path-lookup.h"
  34 #include "strv.h"
  35 #include "unit-name.h"
  36 #include "install.h"
  37 #include "conf-parser.h"
  38 #include "conf-files.h"
  39 #include "specifier.h"
  40 #include "install-printf.h"
  41
  42 typedef struct {
  43         Hashmap *will_install;
  44         Hashmap *have_installed;
  45 } InstallContext;
  46
  47 #define _cleanup_lookup_paths_free_ _cleanup_(lookup_paths_free)
  48 #define _cleanup_install_context_done_ _cleanup_(install_context_done)
  49
  50 static int lookup_paths_init_from_scope(LookupPaths *paths, UnitFileScope scope) {
  51         assert(paths);
  52         assert(scope >= 0);
  53         assert(scope < _UNIT_FILE_SCOPE_MAX);
  54
  55         zero(*paths);
  56
  57         return lookup_paths_init(paths,
  58                                  scope == UNIT_FILE_SYSTEM ? SYSTEMD_SYSTEM : SYSTEMD_USER,
  59                                  scope == UNIT_FILE_USER,
  60                                  NULL, NULL, NULL);
  61 }
  62
  63 static int get_config_path(UnitFileScope scope, bool runtime, const char *root_dir, char **ret) {
  64         char *p = NULL;
  65         int r;
  66
  67         assert(scope >= 0);
  68         assert(scope < _UNIT_FILE_SCOPE_MAX);
  69         assert(ret);
  70
  71         switch (scope) {
  72
  73         case UNIT_FILE_SYSTEM:
  74
  75                 if (root_dir && runtime)
  76                         asprintf(&p, "%s/run/systemd/system", root_dir);
  77                 else if (runtime)
  78                         p = strdup("/run/systemd/system");
  79                 else if (root_dir)
  80                         asprintf(&p, "%s/%s", root_dir, SYSTEM_CONFIG_UNIT_PATH);
  81                 else
  82                         p = strdup(SYSTEM_CONFIG_UNIT_PATH);
  83
  84                 break;
  85
  86         case UNIT_FILE_GLOBAL:
  87
  88                 if (root_dir)
  89                         return -EINVAL;
  90
  91                 if (runtime)
  92                         p = strdup("/run/systemd/user");
  93                 else
  94                         p = strdup(USER_CONFIG_UNIT_PATH);
  95                 break;
  96
  97         case UNIT_FILE_USER:
  98
  99                 if (root_dir || runtime)
 100                         return -EINVAL;
 101
 102                 r = user_config_home(&p);
 103                 if (r <= 0)
 104                         return r < 0 ? r : -ENOENT;
 105
 106                 break;
 107
 108         default:
 109                 assert_not_reached("Bad scope");
 110         }
 111
 112         if (!p)
 113                 return -ENOMEM;
 114
 115         *ret = p;
 116         return 0;
 117 }
 118
 119 static int add_file_change(
 120                 UnitFileChange **changes,
 121                 unsigned *n_changes,
 122                 UnitFileChangeType type,
 123                 const char *path,
 124                 const char *source) {
 125
 126         UnitFileChange *c;
 127         unsigned i;
 128
 129         assert(path);
 130         assert(!changes == !n_changes);
 131
 132         if (!changes)
 133                 return 0;
 134
 135         c = realloc(*changes, (*n_changes + 1) * sizeof(UnitFileChange));
 136         if (!c)
 137                 return -ENOMEM;
 138
 139         *changes = c;
 140         i = *n_changes;
 141
 142         c[i].type = type;
 143         c[i].path = strdup(path);
 144         if (!c[i].path)
 145                 return -ENOMEM;
 146
 147         if (source) {
 148                 c[i].source = strdup(source);
 149                 if (!c[i].source) {
 150                         free(c[i].path);
 151                         return -ENOMEM;
 152                 }
 153         } else
 154                 c[i].source = NULL;
 155
 156         *n_changes = i+1;
 157         return 0;
 158 }
 159
 160 static int mark_symlink_for_removal(
 161                 Set **remove_symlinks_to,
 162                 const char *p) {
 163
 164         char *n;
 165         int r;
 166
 167         assert(p);
 168
 169         r = set_ensure_allocated(remove_symlinks_to, string_hash_func, string_compare_func);
 170         if (r < 0)
 171                 return r;
 172
 173         n = strdup(p);
 174         if (!n)
 175                 return -ENOMEM;
 176
 177         path_kill_slashes(n);
 178
 179         r = set_consume(*remove_symlinks_to, n);
 180         if (r < 0)
 181                 return r == -EEXIST ? 0 : r;
 182
 183         return 0;
 184 }
 185
 186 static int remove_marked_symlinks_fd(
 187                 Set *remove_symlinks_to,
 188                 int fd,
 189                 const char *path,
 190                 const char *config_path,
 191                 bool *deleted,
 192                 UnitFileChange **changes,
 193                 unsigned *n_changes,
 194                 char** files) {
 195
 196         int r = 0;
 197         _cleanup_closedir_ DIR *d = NULL;
 198
 199         assert(remove_symlinks_to);
 200         assert(fd >= 0);
 201         assert(path);
 202         assert(config_path);
 203         assert(deleted);
 204
 205         d = fdopendir(fd);
 206         if (!d) {
 207                 close_nointr_nofail(fd);
 208                 return -errno;
 209         }
 210
 211         rewinddir(d);
 212
 213         for (;;) {
 214                 struct dirent *de;
 215                 union dirent_storage buf;
 216                 int k;
 217
 218                 k = readdir_r(d, &buf.de, &de);
 219                 if (k != 0) {
 220                         r = -errno;
 221                         break;
 222                 }
 223
 224                 if (!de)
 225                         break;
 226
 227                 if (ignore_file(de->d_name))
 228                         continue;
 229
 230                 dirent_ensure_type(d, de);
 231
 232                 if (de->d_type == DT_DIR) {
 233                         int nfd, q;
 234                         _cleanup_free_ char *p = NULL;
 235
 236                         nfd = openat(fd, de->d_name, O_RDONLY|O_NONBLOCK|O_DIRECTORY|O_CLOEXEC|O_NOFOLLOW);
 237                         if (nfd < 0) {
 238                                 if (errno == ENOENT)
 239                                         continue;
 240
 241                                 if (r == 0)
 242                                         r = -errno;
 243                                 continue;
 244                         }
 245
 246                         p = path_make_absolute(de->d_name, path);
 247                         if (!p) {
 248                                 close_nointr_nofail(nfd);
 249                                 return -ENOMEM;
 250                         }
 251
 252                         /* This will close nfd, regardless whether it succeeds or not */
 253                         q = remove_marked_symlinks_fd(remove_symlinks_to, nfd, p, config_path, deleted, changes, n_changes, files);
 254
 255                         if (r == 0)
 256                                 r = q;
 257
 258                 } else if (de->d_type == DT_LNK) {
 259                         _cleanup_free_ char *p = NULL, *dest = NULL;
 260                         int q;
 261                         bool found;
 262
 263                         p = path_make_absolute(de->d_name, path);
 264                         if (!p)
 265                                 return -ENOMEM;
 266
 267                         q = readlink_and_canonicalize(p, &dest);
 268                         if (q < 0) {
 269                                 if (q == -ENOENT)
 270                                         continue;
 271
 272                                 if (r == 0)
 273                                         r = q;
 274                                 continue;
 275                         }
 276
 277                         found =
 278                                 set_get(remove_symlinks_to, dest) ||
 279                                 set_get(remove_symlinks_to, path_get_file_name(dest));
 280
 281                         if (unit_name_is_instance(p))
 282                                 found = found && strv_contains(files, path_get_file_name(p));
 283
 284                         if (found) {
 285
 286                                 if (unlink(p) < 0 && errno != ENOENT) {
 287
 288                                         if (r == 0)
 289                                                 r = -errno;
 290                                 } else {
 291                                         rmdir_parents(p, config_path);
 292                                         path_kill_slashes(p);
 293
 294                                         add_file_change(changes, n_changes, UNIT_FILE_UNLINK, p, NULL);
 295
 296                                         if (!set_get(remove_symlinks_to, p)) {
 297
 298                                                 q = mark_symlink_for_removal(&remove_symlinks_to, p);
 299                                                 if (q < 0) {
 300                                                         if (r == 0)
 301                                                                 r = q;
 302                                                 } else
 303                                                         *deleted = true;
 304                                         }
 305                                 }
 306                         }
 307                 }
 308         }
 309
 310         return r;
 311 }
 312
 313 static int remove_marked_symlinks(
 314                 Set *remove_symlinks_to,
 315                 const char *config_path,
 316                 UnitFileChange **changes,
 317                 unsigned *n_changes,
 318                 char** files) {
 319
 320         int fd, r = 0;
 321         bool deleted;
 322
 323         assert(config_path);
 324
 325         if (set_size(remove_symlinks_to) <= 0)
 326                 return 0;
 327
 328         fd = open(config_path, O_RDONLY|O_NONBLOCK|O_DIRECTORY|O_CLOEXEC|O_NOFOLLOW);
 329         if (fd < 0)
 330                 return -errno;
 331
 332         do {
 333                 int q, cfd;
 334                 deleted = false;
 335
 336                 cfd = dup(fd);
 337                 if (cfd < 0) {
 338                         r = -errno;
 339                         break;
 340                 }
 341
 342                 /* This takes possession of cfd and closes it */
 343                 q = remove_marked_symlinks_fd(remove_symlinks_to, cfd, config_path, config_path, &deleted, changes, n_changes, files);
 344                 if (r == 0)
 345                         r = q;
 346         } while (deleted);
 347
 348         close_nointr_nofail(fd);
 349
 350         return r;
 351 }
 352
 353 static int find_symlinks_fd(
 354                 const char *name,
 355                 int fd,
 356                 const char *path,
 357                 const char *config_path,
 358                 bool *same_name_link) {
 359
 360         int r = 0;
 361         _cleanup_closedir_ DIR *d = NULL;
 362
 363         assert(name);
 364         assert(fd >= 0);
 365         assert(path);
 366         assert(config_path);
 367         assert(same_name_link);
 368
 369         d = fdopendir(fd);
 370         if (!d) {
 371                 close_nointr_nofail(fd);
 372                 return -errno;
 373         }
 374
 375         for (;;) {
 376                 int k;
 377                 struct dirent *de;
 378                 union dirent_storage buf;
 379
 380                 k = readdir_r(d, &buf.de, &de);
 381                 if (k != 0)
 382                         return -errno;
 383
 384                 if (!de)
 385                         return r;
 386
 387                 if (ignore_file(de->d_name))
 388                         continue;
 389
 390                 dirent_ensure_type(d, de);
 391
 392                 if (de->d_type == DT_DIR) {
 393                         int nfd, q;
 394                         _cleanup_free_ char *p = NULL;
 395
 396                         nfd = openat(fd, de->d_name, O_RDONLY|O_NONBLOCK|O_DIRECTORY|O_CLOEXEC|O_NOFOLLOW);
 397                         if (nfd < 0) {
 398                                 if (errno == ENOENT)
 399                                         continue;
 400
 401                                 if (r == 0)
 402                                         r = -errno;
 403                                 continue;
 404                         }
 405
 406                         p = path_make_absolute(de->d_name, path);
 407                         if (!p) {
 408                                 close_nointr_nofail(nfd);
 409                                 return -ENOMEM;
 410                         }
 411
 412                         /* This will close nfd, regardless whether it succeeds or not */
 413                         q = find_symlinks_fd(name, nfd, p, config_path, same_name_link);
 414
 415                         if (q > 0)
 416                                 return 1;
 417
 418                         if (r == 0)
 419                                 r = q;
 420
 421                 } else if (de->d_type == DT_LNK) {
 422                         _cleanup_free_ char *p = NULL, *dest = NULL;
 423                         bool found_path, found_dest, b = false;
 424                         int q;
 425
 426                         /* Acquire symlink name */
 427                         p = path_make_absolute(de->d_name, path);
 428                         if (!p)
 429                                 return -ENOMEM;
 430
 431                         /* Acquire symlink destination */
 432                         q = readlink_and_canonicalize(p, &dest);
 433                         if (q < 0) {
 434                                 if (q == -ENOENT)
 435                                         continue;
 436
 437                                 if (r == 0)
 438                                         r = q;
 439                                 continue;
 440                         }
 441
 442                         /* Check if the symlink itself matches what we
 443                          * are looking for */
 444                         if (path_is_absolute(name))
 445                                 found_path = path_equal(p, name);
 446                         else
 447                                 found_path = streq(de->d_name, name);
 448
 449                         /* Check if what the symlink points to
 450                          * matches what we are looking for */
 451                         if (path_is_absolute(name))
 452                                 found_dest = path_equal(dest, name);
 453                         else
 454                                 found_dest = streq(path_get_file_name(dest), name);
 455
 456                         if (found_path && found_dest) {
 457                                 _cleanup_free_ char *t = NULL;
 458
 459                                 /* Filter out same name links in the main
 460                                  * config path */
 461                                 t = path_make_absolute(name, config_path);
 462                                 if (!t)
 463                                         return -ENOMEM;
 464
 465                                 b = path_equal(t, p);
 466                         }
 467
 468                         if (b)
 469                                 *same_name_link = true;
 470                         else if (found_path || found_dest)
 471                                 return 1;
 472                 }
 473         }
 474
 475         return r;
 476 }
 477
 478 static int find_symlinks(
 479                 const char *name,
 480                 const char *config_path,
 481                 bool *same_name_link) {
 482
 483         int fd;
 484
 485         assert(name);
 486         assert(config_path);
 487         assert(same_name_link);
 488
 489         fd = open(config_path, O_RDONLY|O_NONBLOCK|O_DIRECTORY|O_CLOEXEC|O_NOFOLLOW);
 490         if (fd < 0) {
 491                 if (errno == ENOENT)
 492                         return 0;
 493                 return -errno;
 494         }
 495
 496         /* This takes possession of fd and closes it */
 497         return find_symlinks_fd(name, fd, config_path, config_path, same_name_link);
 498 }
 499
 500 static int find_symlinks_in_scope(
 501                 UnitFileScope scope,
 502                 const char *root_dir,
 503                 const char *name,
 504                 UnitFileState *state) {
 505
 506         int r;
 507         _cleanup_free_ char *path2 = NULL;
 508         bool same_name_link_runtime = false, same_name_link = false;
 509
 510         assert(scope >= 0);
 511         assert(scope < _UNIT_FILE_SCOPE_MAX);
 512         assert(name);
 513
 514         if (scope == UNIT_FILE_SYSTEM || scope == UNIT_FILE_GLOBAL) {
 515                 _cleanup_free_ char *path = NULL;
 516
 517                 /* First look in runtime config path */
 518                 r = get_config_path(scope, true, root_dir, &path);
 519                 if (r < 0)
 520                         return r;
 521
 522                 r = find_symlinks(name, path, &same_name_link_runtime);
 523                 if (r < 0)
 524                         return r;
 525                 else if (r > 0) {
 526                         *state = UNIT_FILE_ENABLED_RUNTIME;
 527                         return r;
 528                 }
 529         }
 530
 531         /* Then look in the normal config path */
 532         r = get_config_path(scope, false, root_dir, &path2);
 533         if (r < 0)
 534                 return r;
 535
 536         r = find_symlinks(name, path2, &same_name_link);
 537         if (r < 0)
 538                 return r;
 539         else if (r > 0) {
 540                 *state = UNIT_FILE_ENABLED;
 541                 return r;
 542         }
 543
 544         /* Hmm, we didn't find it, but maybe we found the same name
 545          * link? */
 546         if (same_name_link_runtime) {
 547                 *state = UNIT_FILE_LINKED_RUNTIME;
 548                 return 1;
 549         } else if (same_name_link) {
 550                 *state = UNIT_FILE_LINKED;
 551                 return 1;
 552         }
 553
 554         return 0;
 555 }
 556
 557 int unit_file_mask(
 558                 UnitFileScope scope,
 559                 bool runtime,
 560                 const char *root_dir,
 561                 char *files[],
 562                 bool force,
 563                 UnitFileChange **changes,
 564                 unsigned *n_changes) {
 565
 566         char **i;
 567         _cleanup_free_ char *prefix;
 568         int r;
 569
 570         assert(scope >= 0);
 571         assert(scope < _UNIT_FILE_SCOPE_MAX);
 572
 573         r = get_config_path(scope, runtime, root_dir, &prefix);
 574         if (r < 0)
 575                 return r;
 576
 577         STRV_FOREACH(i, files) {
 578                 _cleanup_free_ char *path = NULL;
 579
 580                 if (!unit_name_is_valid(*i, true)) {
 581                         if (r == 0)
 582                                 r = -EINVAL;
 583                         continue;
 584                 }
 585
 586                 path = path_make_absolute(*i, prefix);
 587                 if (!path) {
 588                         r = -ENOMEM;
 589                         break;
 590                 }
 591
 592                 if (symlink("/dev/null", path) >= 0) {
 593                         add_file_change(changes, n_changes, UNIT_FILE_SYMLINK, path, "/dev/null");
 594
 595                         continue;
 596                 }
 597
 598                 if (errno == EEXIST) {
 599
 600                         if (null_or_empty_path(path) > 0)
 601                                 continue;
 602
 603                         if (force) {
 604                                 unlink(path);
 605
 606                                 if (symlink("/dev/null", path) >= 0) {
 607
 608                                         add_file_change(changes, n_changes, UNIT_FILE_UNLINK, path, NULL);
 609                                         add_file_change(changes, n_changes, UNIT_FILE_SYMLINK, path, "/dev/null");
 610
 611                                         continue;
 612                                 }
 613                         }
 614
 615                         if (r == 0)
 616                                 r = -EEXIST;
 617                 } else {
 618                         if (r == 0)
 619                                 r = -errno;
 620                 }
 621         }
 622
 623         return r;
 624 }
 625
 626 int unit_file_unmask(
 627                 UnitFileScope scope,
 628                 bool runtime,
 629                 const char *root_dir,
 630                 char *files[],
 631                 UnitFileChange **changes,
 632                 unsigned *n_changes) {
 633
 634         char **i, *config_path = NULL;
 635         int r, q;
 636         Set *remove_symlinks_to = NULL;
 637
 638         assert(scope >= 0);
 639         assert(scope < _UNIT_FILE_SCOPE_MAX);
 640
 641         r = get_config_path(scope, runtime, root_dir, &config_path);
 642         if (r < 0)
 643                 goto finish;
 644
 645         STRV_FOREACH(i, files) {
 646                 char *path;
 647
 648                 if (!unit_name_is_valid(*i, true)) {
 649                         if (r == 0)
 650                                 r = -EINVAL;
 651                         continue;
 652                 }
 653
 654                 path = path_make_absolute(*i, config_path);
 655                 if (!path) {
 656                         r = -ENOMEM;
 657                         break;
 658                 }
 659
 660                 q = null_or_empty_path(path);
 661                 if (q > 0) {
 662                         if (unlink(path) >= 0) {
 663                                 mark_symlink_for_removal(&remove_symlinks_to, path);
 664                                 add_file_change(changes, n_changes, UNIT_FILE_UNLINK, path, NULL);
 665
 666                                 free(path);
 667                                 continue;
 668                         }
 669
 670                         q = -errno;
 671                 }
 672
 673                 if (q != -ENOENT && r == 0)
 674                         r = q;
 675
 676                 free(path);
 677         }
 678
 679
 680 finish:
 681         q = remove_marked_symlinks(remove_symlinks_to, config_path, changes, n_changes, files);
 682         if (r == 0)
 683                 r = q;
 684
 685         set_free_free(remove_symlinks_to);
 686         free(config_path);
 687
 688         return r;
 689 }
 690
 691 int unit_file_link(
 692                 UnitFileScope scope,
 693                 bool runtime,
 694                 const char *root_dir,
 695                 char *files[],
 696                 bool force,
 697                 UnitFileChange **changes,
 698                 unsigned *n_changes) {
 699
 700         _cleanup_lookup_paths_free_ LookupPaths paths = {};
 701         char **i;
 702         _cleanup_free_ char *config_path = NULL;
 703         int r, q;
 704
 705         assert(scope >= 0);
 706         assert(scope < _UNIT_FILE_SCOPE_MAX);
 707
 708         r = lookup_paths_init_from_scope(&paths, scope);
 709         if (r < 0)
 710                 return r;
 711
 712         r = get_config_path(scope, runtime, root_dir, &config_path);
 713         if (r < 0)
 714                 return r;
 715
 716         STRV_FOREACH(i, files) {
 717                 _cleanup_free_ char *path = NULL;
 718                 char *fn;
 719                 struct stat st;
 720
 721                 fn = path_get_file_name(*i);
 722
 723                 if (!path_is_absolute(*i) ||
 724                     !unit_name_is_valid(fn, true)) {
 725                         if (r == 0)
 726                                 r = -EINVAL;
 727                         continue;
 728                 }
 729
 730                 if (lstat(*i, &st) < 0) {
 731                         if (r == 0)
 732                                 r = -errno;
 733                         continue;
 734                 }
 735
 736                 if (!S_ISREG(st.st_mode)) {
 737                         r = -ENOENT;
 738                         continue;
 739                 }
 740
 741                 q = in_search_path(*i, paths.unit_path);
 742                 if (q < 0)
 743                         return q;
 744
 745                 if (q > 0)
 746                         continue;
 747
 748                 path = path_make_absolute(fn, config_path);
 749                 if (!path)
 750                         return -ENOMEM;
 751
 752                 if (symlink(*i, path) >= 0) {
 753                         add_file_change(changes, n_changes, UNIT_FILE_SYMLINK, path, *i);
 754                         continue;
 755                 }
 756
 757                 if (errno == EEXIST) {
 758                         _cleanup_free_ char *dest = NULL;
 759
 760                         q = readlink_and_make_absolute(path, &dest);
 761
 762                         if (q < 0 && errno != ENOENT) {
 763                                 if (r == 0)
 764                                         r = q;
 765                                 continue;
 766                         }
 767
 768                         if (q >= 0 && path_equal(dest, *i))
 769                                 continue;
 770
 771                         if (force) {
 772                                 unlink(path);
 773
 774                                 if (symlink(*i, path) >= 0) {
 775
 776                                         add_file_change(changes, n_changes, UNIT_FILE_UNLINK, path, NULL);
 777                                         add_file_change(changes, n_changes, UNIT_FILE_SYMLINK, path, *i);
 778
 779                                         continue;
 780                                 }
 781                         }
 782
 783                         if (r == 0)
 784                                 r = -EEXIST;
 785                 } else {
 786                         if (r == 0)
 787                                 r = -errno;
 788                 }
 789         }
 790
 791         return r;
 792 }
 793
 794 void unit_file_list_free(Hashmap *h) {
 795         UnitFileList *i;
 796
 797         while ((i = hashmap_steal_first(h))) {
 798                 free(i->path);
 799                 free(i);
 800         }
 801
 802         hashmap_free(h);
 803 }
 804
 805 void unit_file_changes_free(UnitFileChange *changes, unsigned n_changes) {
 806         unsigned i;
 807
 808         assert(changes || n_changes == 0);
 809
 810         if (!changes)
 811                 return;
 812
 813         for (i = 0; i < n_changes; i++) {
 814                 free(changes[i].path);
 815                 free(changes[i].source);
 816         }
 817
 818         free(changes);
 819 }
 820
 821 static void install_info_free(InstallInfo *i) {
 822         assert(i);
 823
 824         free(i->name);
 825         free(i->path);
 826         strv_free(i->aliases);
 827         strv_free(i->wanted_by);
 828         strv_free(i->required_by);
 829         free(i);
 830 }
 831
 832 static void install_info_hashmap_free(Hashmap *m) {
 833         InstallInfo *i;
 834
 835         if (!m)
 836                 return;
 837
 838         while ((i = hashmap_steal_first(m)))
 839                 install_info_free(i);
 840
 841         hashmap_free(m);
 842 }
 843
 844 static void install_context_done(InstallContext *c) {
 845         assert(c);
 846
 847         install_info_hashmap_free(c->will_install);
 848         install_info_hashmap_free(c->have_installed);
 849
 850         c->will_install = c->have_installed = NULL;
 851 }
 852
 853 static int install_info_add(
 854                 InstallContext *c,
 855                 const char *name,
 856                 const char *path) {
 857         InstallInfo *i = NULL;
 858         int r;
 859
 860         assert(c);
 861         assert(name || path);
 862
 863         if (!name)
 864                 name = path_get_file_name(path);
 865
 866         if (!unit_name_is_valid(name, true))
 867                 return -EINVAL;
 868
 869         if (hashmap_get(c->have_installed, name) ||
 870             hashmap_get(c->will_install, name))
 871                 return 0;
 872
 873         r = hashmap_ensure_allocated(&c->will_install, string_hash_func, string_compare_func);
 874         if (r < 0)
 875                 return r;
 876
 877         i = new0(InstallInfo, 1);
 878         if (!i)
 879                 return -ENOMEM;
 880
 881         i->name = strdup(name);
 882         if (!i->name) {
 883                 r = -ENOMEM;
 884                 goto fail;
 885         }
 886
 887         if (path) {
 888                 i->path = strdup(path);
 889                 if (!i->path) {
 890                         r = -ENOMEM;
 891                         goto fail;
 892                 }
 893         }
 894
 895         r = hashmap_put(c->will_install, i->name, i);
 896         if (r < 0)
 897                 goto fail;
 898
 899         return 0;
 900
 901 fail:
 902         if (i)
 903                 install_info_free(i);
 904
 905         return r;
 906 }
 907
 908 static int install_info_add_auto(
 909                 InstallContext *c,
 910                 const char *name_or_path) {
 911
 912         assert(c);
 913         assert(name_or_path);
 914
 915         if (path_is_absolute(name_or_path))
 916                 return install_info_add(c, NULL, name_or_path);
 917         else
 918                 return install_info_add(c, name_or_path, NULL);
 919 }
 920
 921 static int config_parse_also(const char *unit,
 922                              const char *filename,
 923                              unsigned line,
 924                              const char *section,
 925                              const char *lvalue,
 926                              int ltype,
 927                              const char *rvalue,
 928                              void *data,
 929                              void *userdata) {
 930
 931         char *w;
 932         size_t l;
 933         char *state;
 934         InstallContext *c = data;
 935
 936         assert(filename);
 937         assert(lvalue);
 938         assert(rvalue);
 939
 940         FOREACH_WORD_QUOTED(w, l, rvalue, state) {
 941                 _cleanup_free_ char *n;
 942                 int r;
 943
 944                 n = strndup(w, l);
 945                 if (!n)
 946                         return -ENOMEM;
 947
 948                 r = install_info_add(c, n, NULL);
 949                 if (r < 0)
 950                         return r;
 951         }
 952
 953         return 0;
 954 }
 955
 956 static int config_parse_user(const char *unit,
 957                              const char *filename,
 958                              unsigned line,
 959                              const char *section,
 960                              const char *lvalue,
 961                              int ltype,
 962                              const char *rvalue,
 963                              void *data,
 964                              void *userdata) {
 965
 966         InstallInfo *i = data;
 967         char* printed;
 968         int r;
 969
 970         assert(filename);
 971         assert(lvalue);
 972         assert(rvalue);
 973
 974         r = install_full_printf(i, rvalue, &printed);
 975         if (r < 0)
 976                 return r;
 977
 978         free(i->user);
 979         i->user = printed;
 980
 981         return 0;
 982 }
 983
 984 static int unit_file_load(
 985                 InstallContext *c,
 986                 InstallInfo *info,
 987                 const char *path,
 988                 bool allow_symlink) {
 989
 990         const ConfigTableItem items[] = {
 991                 { "Install", "Alias",      config_parse_strv, 0, &info->aliases     },
 992                 { "Install", "WantedBy",   config_parse_strv, 0, &info->wanted_by   },
 993                 { "Install", "RequiredBy", config_parse_strv, 0, &info->required_by },
 994                 { "Install", "Also",       config_parse_also, 0, c                  },
 995                 { "Exec",    "User",       config_parse_user, 0, info               },
 996                 { NULL, NULL, NULL, 0, NULL }
 997         };
 998
 999         int fd;
1000         _cleanup_fclose_ FILE *f = NULL;
1001         int r;
1002
1003         assert(c);
1004         assert(info);
1005         assert(path);
1006
1007         fd = open(path, O_RDONLY|O_CLOEXEC|O_NOCTTY|(allow_symlink ? 0 : O_NOFOLLOW));
1008         if (fd < 0)
1009                 return -errno;
1010
1011         f = fdopen(fd, "re");
1012         if (!f) {
1013                 close_nointr_nofail(fd);
1014                 return -ENOMEM;
1015         }
1016
1017         r = config_parse(NULL, path, f, NULL,
1018                          config_item_table_lookup, (void*) items, true, true, info);
1019         if (r < 0)
1020                 return r;
1021
1022         return
1023                 strv_length(info->aliases) +
1024                 strv_length(info->wanted_by) +
1025                 strv_length(info->required_by);
1026 }
1027
1028 static int unit_file_search(
1029                 InstallContext *c,
1030                 InstallInfo *info,
1031                 LookupPaths *paths,
1032                 const char *root_dir,
1033                 bool allow_symlink) {
1034
1035         char **p;
1036         int r;
1037
1038         assert(c);
1039         assert(info);
1040         assert(paths);
1041
1042         if (info->path)
1043                 return unit_file_load(c, info, info->path, allow_symlink);
1044
1045         assert(info->name);
1046
1047         STRV_FOREACH(p, paths->unit_path) {
1048                 char *path = NULL;
1049
1050                 if (isempty(root_dir))
1051                         asprintf(&path, "%s/%s", *p, info->name);
1052                 else
1053                         asprintf(&path, "%s/%s/%s", root_dir, *p, info->name);
1054
1055                 if (!path)
1056                         return -ENOMEM;
1057
1058                 r = unit_file_load(c, info, path, allow_symlink);
1059
1060                 if (r >= 0)
1061                         info->path = path;
1062                 else {
1063                         if (r == -ENOENT && unit_name_is_instance(info->name)) {
1064                                 /* Unit file doesn't exist, however instance enablement was requested.
1065                                  * We will check if it is possible to load template unit file. */
1066                                 char *template = NULL,
1067                                      *template_path = NULL,
1068                                      *template_dir = NULL;
1069
1070                                 template = unit_name_template(info->name);
1071                                 if (!template) {
1072                                         free(path);
1073                                         return -ENOMEM;
1074                                 }
1075
1076                                 /* We will reuse path variable since we don't need it anymore. */
1077                                 template_dir = path;
1078                                 *(strrchr(path, '/') + 1) = '\0';
1079
1080                                 template_path = strjoin(template_dir, template, NULL);
1081                                 if (!template_path) {
1082                                         free(path);
1083                                         free(template);
1084                                         return -ENOMEM;
1085                                 }
1086
1087                                 /* Let's try to load template unit. */
1088                                 r = unit_file_load(c, info, template_path, allow_symlink);
1089                                 if (r >= 0) {
1090                                         info->path = strdup(template_path);
1091                                         if (!info->path) {
1092                                                 free(path);
1093                                                 free(template);
1094                                                 free(template_path);
1095                                                 return -ENOMEM;
1096                                         }
1097                                 }
1098
1099                                 free(template);
1100                                 free(template_path);
1101                         }
1102                         free(path);
1103                 }
1104
1105                 if (r != -ENOENT && r != -ELOOP)
1106                         return r;
1107         }
1108
1109         return -ENOENT;
1110 }
1111
1112 static int unit_file_can_install(
1113                 LookupPaths *paths,
1114                 const char *root_dir,
1115                 const char *name,
1116                 bool allow_symlink) {
1117
1118         _cleanup_install_context_done_ InstallContext c = {};
1119         InstallInfo *i;
1120         int r;
1121
1122         assert(paths);
1123         assert(name);
1124
1125         r = install_info_add_auto(&c, name);
1126         if (r < 0)
1127                 return r;
1128
1129         assert_se(i = hashmap_first(c.will_install));
1130
1131         r = unit_file_search(&c, i, paths, root_dir, allow_symlink);
1132
1133         if (r >= 0)
1134                 r =
1135                         strv_length(i->aliases) +
1136                         strv_length(i->wanted_by) +
1137                         strv_length(i->required_by);
1138
1139         return r;
1140 }
1141
1142 static int create_symlink(
1143                 const char *old_path,
1144                 const char *new_path,
1145                 bool force,
1146                 UnitFileChange **changes,
1147                 unsigned *n_changes) {
1148
1149         _cleanup_free_ char *dest = NULL;
1150         int r;
1151
1152         assert(old_path);
1153         assert(new_path);
1154
1155         mkdir_parents_label(new_path, 0755);
1156
1157         if (symlink(old_path, new_path) >= 0) {
1158                 add_file_change(changes, n_changes, UNIT_FILE_SYMLINK, new_path, old_path);
1159                 return 0;
1160         }
1161
1162         if (errno != EEXIST)
1163                 return -errno;
1164
1165         r = readlink_and_make_absolute(new_path, &dest);
1166         if (r < 0)
1167                 return r;
1168
1169         if (path_equal(dest, old_path))
1170                 return 0;
1171
1172         if (!force)
1173                 return -EEXIST;
1174
1175         unlink(new_path);
1176
1177         if (symlink(old_path, new_path) >= 0) {
1178                 add_file_change(changes, n_changes, UNIT_FILE_UNLINK, new_path, NULL);
1179                 add_file_change(changes, n_changes, UNIT_FILE_SYMLINK, new_path, old_path);
1180                 return 0;
1181         }
1182
1183         return -errno;
1184 }
1185
1186 static int install_info_symlink_alias(
1187                 InstallInfo *i,
1188                 const char *config_path,
1189                 bool force,
1190                 UnitFileChange **changes,
1191                 unsigned *n_changes) {
1192
1193         char **s;
1194         int r = 0, q;
1195
1196         assert(i);
1197         assert(config_path);
1198
1199         STRV_FOREACH(s, i->aliases) {
1200                 _cleanup_free_ char *alias_path = NULL, *dst = NULL;
1201
1202                 q = install_full_printf(i, *s, &dst);
1203                 if (q < 0)
1204                         return q;
1205
1206                 alias_path = path_make_absolute(dst, config_path);
1207                 if (!alias_path)
1208                         return -ENOMEM;
1209
1210                 q = create_symlink(i->path, alias_path, force, changes, n_changes);
1211                 if (r == 0)
1212                         r = q;
1213         }
1214
1215         return r;
1216 }
1217
1218 static int install_info_symlink_wants(
1219                 InstallInfo *i,
1220                 const char *config_path,
1221                 bool force,
1222                 UnitFileChange **changes,
1223                 unsigned *n_changes) {
1224
1225         char **s;
1226         int r = 0, q;
1227
1228         assert(i);
1229         assert(config_path);
1230
1231         STRV_FOREACH(s, i->wanted_by) {
1232                 _cleanup_free_ char *path = NULL, *dst = NULL;
1233
1234                 q = install_full_printf(i, *s, &dst);
1235                 if (q < 0)
1236                         return q;
1237
1238                 if (!unit_name_is_valid(dst, true)) {
1239                         r = -EINVAL;
1240                         continue;
1241                 }
1242
1243                 if (asprintf(&path, "%s/%s.wants/%s", config_path, dst, i->name) < 0)
1244                         return -ENOMEM;
1245
1246                 q = create_symlink(i->path, path, force, changes, n_changes);
1247
1248                 if (r == 0)
1249                         r = q;
1250         }
1251
1252         return r;
1253 }
1254
1255 static int install_info_symlink_requires(
1256                 InstallInfo *i,
1257                 const char *config_path,
1258                 bool force,
1259                 UnitFileChange **changes,
1260                 unsigned *n_changes) {
1261
1262         char **s;
1263         int r = 0, q;
1264
1265         assert(i);
1266         assert(config_path);
1267
1268         STRV_FOREACH(s, i->required_by) {
1269                 _cleanup_free_ char *path = NULL, *dst = NULL;
1270
1271                 q = install_full_printf(i, *s, &dst);
1272                 if (q < 0)
1273                         return q;
1274
1275                 if (!unit_name_is_valid(dst, true)) {
1276                         r = -EINVAL;
1277                         continue;
1278                 }
1279
1280                 if (asprintf(&path, "%s/%s.requires/%s", config_path, dst, i->name) < 0)
1281                         return -ENOMEM;
1282
1283                 q = create_symlink(i->path, path, force, changes, n_changes);
1284
1285                 if (r == 0)
1286                         r = q;
1287         }
1288
1289         return r;
1290 }
1291
1292 static int install_info_symlink_link(
1293                 InstallInfo *i,
1294                 LookupPaths *paths,
1295                 const char *config_path,
1296                 bool force,
1297                 UnitFileChange **changes,
1298                 unsigned *n_changes) {
1299
1300         int r;
1301         _cleanup_free_ char *path = NULL;
1302
1303         assert(i);
1304         assert(paths);
1305         assert(config_path);
1306         assert(i->path);
1307
1308         r = in_search_path(i->path, paths->unit_path);
1309         if (r != 0)
1310                 return r;
1311
1312         if (asprintf(&path, "%s/%s", config_path, i->name) < 0)
1313                 return -ENOMEM;
1314
1315         r = create_symlink(i->path, path, force, changes, n_changes);
1316         return r;
1317 }
1318
1319 static int install_info_apply(
1320                 InstallInfo *i,
1321                 LookupPaths *paths,
1322                 const char *config_path,
1323                 bool force,
1324                 UnitFileChange **changes,
1325                 unsigned *n_changes) {
1326
1327         int r, q;
1328
1329         assert(i);
1330         assert(paths);
1331         assert(config_path);
1332
1333         r = install_info_symlink_alias(i, config_path, force, changes, n_changes);
1334
1335         q = install_info_symlink_wants(i, config_path, force, changes, n_changes);
1336         if (r == 0)
1337                 r = q;
1338
1339         q = install_info_symlink_requires(i, config_path, force, changes, n_changes);
1340         if (r == 0)
1341                 r = q;
1342
1343         q = install_info_symlink_link(i, paths, config_path, force, changes, n_changes);
1344         if (r == 0)
1345                 r = q;
1346
1347         return r;
1348 }
1349
1350 static int install_context_apply(
1351                 InstallContext *c,
1352                 LookupPaths *paths,
1353                 const char *config_path,
1354                 const char *root_dir,
1355                 bool force,
1356                 UnitFileChange **changes,
1357                 unsigned *n_changes) {
1358
1359         InstallInfo *i;
1360         int r = 0, q;
1361
1362         assert(c);
1363         assert(paths);
1364         assert(config_path);
1365
1366         while ((i = hashmap_first(c->will_install))) {
1367
1368                 q = hashmap_ensure_allocated(&c->have_installed, string_hash_func, string_compare_func);
1369                 if (q < 0)
1370                         return q;
1371
1372                 assert_se(hashmap_move_one(c->have_installed, c->will_install, i->name) == 0);
1373
1374                 q = unit_file_search(c, i, paths, root_dir, false);
1375                 if (q < 0) {
1376                         if (r >= 0)
1377                                 r = q;
1378
1379                         return r;
1380                 } else if (r >= 0)
1381                         r += q;
1382
1383                 q = install_info_apply(i, paths, config_path, force, changes, n_changes);
1384                 if (r >= 0 && q < 0)
1385                         r = q;
1386         }
1387
1388         return r;
1389 }
1390
1391 static int install_context_mark_for_removal(
1392                 InstallContext *c,
1393                 LookupPaths *paths,
1394                 Set **remove_symlinks_to,
1395                 const char *config_path,
1396                 const char *root_dir) {
1397
1398         InstallInfo *i;
1399         int r = 0, q;
1400
1401         assert(c);
1402         assert(paths);
1403         assert(config_path);
1404
1405         /* Marks all items for removal */
1406
1407         while ((i = hashmap_first(c->will_install))) {
1408
1409                 q = hashmap_ensure_allocated(&c->have_installed, string_hash_func, string_compare_func);
1410                 if (q < 0)
1411                         return q;
1412
1413                 assert_se(hashmap_move_one(c->have_installed, c->will_install, i->name) == 0);
1414
1415                 q = unit_file_search(c, i, paths, root_dir, false);
1416                 if (q == -ENOENT) {
1417                         /* do nothing */
1418                 } else if (q < 0) {
1419                         if (r >= 0)
1420                                 r = q;
1421
1422                         return r;
1423                 } else if (r >= 0)
1424                         r += q;
1425
1426                 if (unit_name_is_instance(i->name)) {
1427                         char *unit_file;
1428
1429                         if (i->path) {
1430                                 unit_file = path_get_file_name(i->path);
1431
1432                                 if (unit_name_is_instance(unit_file))
1433                                         /* unit file named as instance exists, thus all symlinks
1434                                          * pointing to it will be removed */
1435                                         q = mark_symlink_for_removal(remove_symlinks_to, i->name);
1436                                 else
1437                                         /* does not exist, thus we will mark for removal symlinks
1438                                          * to template unit file */
1439                                         q = mark_symlink_for_removal(remove_symlinks_to, unit_file);
1440                         } else {
1441                                 /* If i->path is not set, it means that we didn't actually find
1442                                  * the unit file. But we can still remove symlinks to the
1443                                  * nonexistent template. */
1444                                 unit_file = unit_name_template(i->name);
1445                                 if (!unit_file)
1446                                         return log_oom();
1447
1448                                 q = mark_symlink_for_removal(remove_symlinks_to, unit_file);
1449                                 free(unit_file);
1450                         }
1451                 } else
1452                         q = mark_symlink_for_removal(remove_symlinks_to, i->name);
1453
1454                 if (r >= 0 && q < 0)
1455                         r = q;
1456         }
1457
1458         return r;
1459 }
1460
1461 int unit_file_enable(
1462                 UnitFileScope scope,
1463                 bool runtime,
1464                 const char *root_dir,
1465                 char *files[],
1466                 bool force,
1467                 UnitFileChange **changes,
1468                 unsigned *n_changes) {
1469
1470         _cleanup_lookup_paths_free_ LookupPaths paths = {};
1471         _cleanup_install_context_done_ InstallContext c = {};
1472         char **i;
1473         _cleanup_free_ char *config_path = NULL;
1474         int r;
1475
1476         assert(scope >= 0);
1477         assert(scope < _UNIT_FILE_SCOPE_MAX);
1478
1479         r = lookup_paths_init_from_scope(&paths, scope);
1480         if (r < 0)
1481                 return r;
1482
1483         r = get_config_path(scope, runtime, root_dir, &config_path);
1484         if (r < 0)
1485                 return r;
1486
1487         STRV_FOREACH(i, files) {
1488                 r = install_info_add_auto(&c, *i);
1489                 if (r < 0)
1490                         return r;
1491         }
1492
1493         /* This will return the number of symlink rules that were
1494         supposed to be created, not the ones actually created. This is
1495         useful to determine whether the passed files had any
1496         installation data at all. */
1497         r = install_context_apply(&c, &paths, config_path, root_dir, force, changes, n_changes);
1498         return r;
1499 }
1500
1501 int unit_file_disable(
1502                 UnitFileScope scope,
1503                 bool runtime,
1504                 const char *root_dir,
1505                 char *files[],
1506                 UnitFileChange **changes,
1507                 unsigned *n_changes) {
1508
1509         _cleanup_lookup_paths_free_ LookupPaths paths = {};
1510         _cleanup_install_context_done_ InstallContext c = {};
1511         char **i;
1512         _cleanup_free_ char *config_path = NULL;
1513         _cleanup_set_free_free_ Set *remove_symlinks_to = NULL;
1514         int r, q;
1515
1516         assert(scope >= 0);
1517         assert(scope < _UNIT_FILE_SCOPE_MAX);
1518
1519         r = lookup_paths_init_from_scope(&paths, scope);
1520         if (r < 0)
1521                 return r;
1522
1523         r = get_config_path(scope, runtime, root_dir, &config_path);
1524         if (r < 0)
1525                 return r;
1526
1527         STRV_FOREACH(i, files) {
1528                 r = install_info_add_auto(&c, *i);
1529                 if (r < 0)
1530                         return r;
1531         }
1532
1533         r = install_context_mark_for_removal(&c, &paths, &remove_symlinks_to, config_path, root_dir);
1534
1535         q = remove_marked_symlinks(remove_symlinks_to, config_path, changes, n_changes, files);
1536         if (r == 0)
1537                 r = q;
1538
1539         return r;
1540 }
1541
1542 int unit_file_reenable(
1543                 UnitFileScope scope,
1544                 bool runtime,
1545                 const char *root_dir,
1546                 char *files[],
1547                 bool force,
1548                 UnitFileChange **changes,
1549                 unsigned *n_changes) {
1550         int r;
1551
1552         r = unit_file_disable(scope, runtime, root_dir, files,
1553                               changes, n_changes);
1554         if (r < 0)
1555                 return r;
1556
1557         return unit_file_enable(scope, runtime, root_dir, files, force,
1558                                 changes, n_changes);
1559 }
1560
1561 int unit_file_set_default(
1562                 UnitFileScope scope,
1563                 const char *root_dir,
1564                 char *file,
1565                 UnitFileChange **changes,
1566                 unsigned *n_changes) {
1567
1568         _cleanup_lookup_paths_free_ LookupPaths paths = {};
1569         _cleanup_install_context_done_ InstallContext c = {};
1570         _cleanup_free_ char *config_path = NULL;
1571         char *path;
1572         int r;
1573         InstallInfo *i = NULL;
1574
1575         assert(scope >= 0);
1576         assert(scope < _UNIT_FILE_SCOPE_MAX);
1577
1578         if (unit_name_to_type(file) != UNIT_TARGET)
1579                 return -EINVAL;
1580
1581         r = lookup_paths_init_from_scope(&paths, scope);
1582         if (r < 0)
1583                 return r;
1584
1585         r = get_config_path(scope, false, root_dir, &config_path);
1586         if (r < 0)
1587                 return r;
1588
1589         r = install_info_add_auto(&c, file);
1590         if (r < 0)
1591                 return r;
1592
1593         i = (InstallInfo*)hashmap_first(c.will_install);
1594
1595         r = unit_file_search(&c, i, &paths, root_dir, false);
1596         if (r < 0)
1597                 return r;
1598
1599         path = strappenda(config_path, "/default.target");
1600         r = create_symlink(i->path, path, true, changes, n_changes);
1601         if (r < 0)
1602                 return r;
1603
1604         return 0;
1605 }
1606
1607 int unit_file_get_default(
1608                 UnitFileScope scope,
1609                 const char *root_dir,
1610                 char **name) {
1611
1612         _cleanup_lookup_paths_free_ LookupPaths paths = {};
1613         char **p;
1614         int r;
1615
1616         r = lookup_paths_init_from_scope(&paths, scope);
1617         if (r < 0)
1618                 return r;
1619
1620         STRV_FOREACH(p, paths.unit_path) {
1621                 _cleanup_free_ char *path = NULL, *tmp = NULL;
1622
1623                 if (isempty(root_dir))
1624                         path = strappend(*p, "/default.target");
1625                 else
1626                         path = strjoin(root_dir, "/", *p, "/default.target", NULL);
1627
1628                 if (!path)
1629                         return -ENOMEM;
1630
1631                 r = readlink_malloc(path, &tmp);
1632                 if (r == -ENOENT)
1633                         continue;
1634                 else if (r < 0)
1635                         return r;
1636
1637                 *name = strdup(path_get_file_name(tmp));
1638                 if (!*name)
1639                         return -ENOMEM;
1640
1641                 return 0;
1642         }
1643
1644         return -ENOENT;
1645 }
1646
1647 UnitFileState unit_file_get_state(
1648                 UnitFileScope scope,
1649                 const char *root_dir,
1650                 const char *name) {
1651
1652         _cleanup_lookup_paths_free_ LookupPaths paths = {};
1653         UnitFileState state = _UNIT_FILE_STATE_INVALID;
1654         char **i;
1655         _cleanup_free_ char *path = NULL;
1656         int r;
1657
1658         assert(scope >= 0);
1659         assert(scope < _UNIT_FILE_SCOPE_MAX);
1660         assert(name);
1661
1662         if (root_dir && scope != UNIT_FILE_SYSTEM)
1663                 return -EINVAL;
1664
1665         if (!unit_name_is_valid(name, true))
1666                 return -EINVAL;
1667
1668         r = lookup_paths_init_from_scope(&paths, scope);
1669         if (r < 0)
1670                 return r;
1671
1672         STRV_FOREACH(i, paths.unit_path) {
1673                 struct stat st;
1674
1675                 free(path);
1676                 path = NULL;
1677
1678                 if (root_dir)
1679                         asprintf(&path, "%s/%s/%s", root_dir, *i, name);
1680                 else
1681                         asprintf(&path, "%s/%s", *i, name);
1682
1683                 if (!path)
1684                         return -ENOMEM;
1685
1686                 /*
1687                  * Search for a unit file in our default paths, to
1688                  * be sure, that there are no broken symlinks.
1689                  */
1690                 if (lstat(path, &st) < 0) {
1691                         r = -errno;
1692                         if (errno != ENOENT)
1693                                 return r;
1694
1695                         if (!unit_name_is_instance(name))
1696                                 continue;
1697                 } else {
1698                         if (!S_ISREG(st.st_mode) && !S_ISLNK(st.st_mode))
1699                                 return -ENOENT;
1700
1701                         r = null_or_empty_path(path);
1702                         if (r < 0 && r != -ENOENT)
1703                                 return r;
1704                         else if (r > 0) {
1705                                 state = path_startswith(*i, "/run") ?
1706                                         UNIT_FILE_MASKED_RUNTIME : UNIT_FILE_MASKED;
1707                                 return state;
1708                         }
1709                 }
1710
1711                 r = find_symlinks_in_scope(scope, root_dir, name, &state);
1712                 if (r < 0)
1713                         return r;
1714                 else if (r > 0)
1715                         return state;
1716
1717                 r = unit_file_can_install(&paths, root_dir, path, true);
1718                 if (r < 0 && errno != ENOENT)
1719                         return r;
1720                 else if (r > 0)
1721                         return UNIT_FILE_DISABLED;
1722                 else if (r == 0)
1723                         return UNIT_FILE_STATIC;
1724         }
1725
1726         return r < 0 ? r : state;
1727 }
1728
1729 int unit_file_query_preset(UnitFileScope scope, const char *name) {
1730         _cleanup_strv_free_ char **files = NULL;
1731         char **i;
1732         int r;
1733
1734         assert(scope >= 0);
1735         assert(scope < _UNIT_FILE_SCOPE_MAX);
1736         assert(name);
1737
1738         if (scope == UNIT_FILE_SYSTEM)
1739                 r = conf_files_list(&files, ".preset", NULL,
1740                                     "/etc/systemd/system-preset",
1741                                     "/usr/local/lib/systemd/system-preset",
1742                                     "/usr/lib/systemd/system-preset",
1743 #ifdef HAVE_SPLIT_USR
1744                                     "/lib/systemd/system-preset",
1745 #endif
1746                                     NULL);
1747         else if (scope == UNIT_FILE_GLOBAL)
1748                 r = conf_files_list(&files, ".preset", NULL,
1749                                     "/etc/systemd/user-preset",
1750                                     "/usr/local/lib/systemd/user-preset",
1751                                     "/usr/lib/systemd/user-preset",
1752                                     NULL);
1753         else
1754                 return 1;
1755
1756         if (r < 0)
1757                 return r;
1758
1759         STRV_FOREACH(i, files) {
1760                 _cleanup_fclose_ FILE *f;
1761
1762                 f = fopen(*i, "re");
1763                 if (!f) {
1764                         if (errno == ENOENT)
1765                                 continue;
1766
1767                         return -errno;
1768                 }
1769
1770                 for (;;) {
1771                         char line[LINE_MAX], *l;
1772
1773                         if (!fgets(line, sizeof(line), f))
1774                                 break;
1775
1776                         l = strstrip(line);
1777                         if (!*l)
1778                                 continue;
1779
1780                         if (strchr(COMMENTS "\n", *l))
1781                                 continue;
1782
1783                         if (first_word(l, "enable")) {
1784                                 l += 6;
1785                                 l += strspn(l, WHITESPACE);
1786
1787                                 if (fnmatch(l, name, FNM_NOESCAPE) == 0)
1788                                         return 1;
1789
1790                         } else if (first_word(l, "disable")) {
1791                                 l += 7;
1792                                 l += strspn(l, WHITESPACE);
1793
1794                                 if (fnmatch(l, name, FNM_NOESCAPE) == 0)
1795                                         return 0;
1796
1797                         } else
1798                                 log_debug("Couldn't parse line '%s'", l);
1799                 }
1800         }
1801
1802         /* Default is "enable" */
1803         return 1;
1804 }
1805
1806 int unit_file_preset(
1807                 UnitFileScope scope,
1808                 bool runtime,
1809                 const char *root_dir,
1810                 char *files[],
1811                 bool force,
1812                 UnitFileChange **changes,
1813                 unsigned *n_changes) {
1814
1815         _cleanup_lookup_paths_free_ LookupPaths paths = {};
1816         _cleanup_install_context_done_ InstallContext plus = {}, minus = {};
1817         char **i;
1818         _cleanup_free_ char *config_path = NULL;
1819         _cleanup_set_free_free_ Set *remove_symlinks_to = NULL;
1820         int r, q;
1821
1822         assert(scope >= 0);
1823         assert(scope < _UNIT_FILE_SCOPE_MAX);
1824
1825         r = lookup_paths_init_from_scope(&paths, scope);
1826         if (r < 0)
1827                 return r;
1828
1829         r = get_config_path(scope, runtime, root_dir, &config_path);
1830         if (r < 0)
1831                 return r;
1832
1833         STRV_FOREACH(i, files) {
1834
1835                 if (!unit_name_is_valid(*i, true))
1836                         return -EINVAL;
1837
1838                 r = unit_file_query_preset(scope, *i);
1839                 if (r < 0)
1840                         return r;
1841
1842                 if (r)
1843                         r = install_info_add_auto(&plus, *i);
1844                 else
1845                         r = install_info_add_auto(&minus, *i);
1846
1847                 if (r < 0)
1848                         return r;
1849         }
1850
1851         r = install_context_mark_for_removal(&minus, &paths, &remove_symlinks_to,
1852                                              config_path, root_dir);
1853
1854         q = remove_marked_symlinks(remove_symlinks_to, config_path,
1855                                    changes, n_changes, files);
1856         if (r == 0)
1857                 r = q;
1858
1859         /* Returns number of symlinks that where supposed to be installed. */
1860         q = install_context_apply(&plus, &paths, config_path, root_dir, force,
1861                                   changes, n_changes);
1862         if (r == 0)
1863                 r = q;
1864
1865         return r;
1866 }
1867
1868 static void unitfilelist_free(UnitFileList **f) {
1869         if (!*f)
1870                 return;
1871
1872         free((*f)->path);
1873         free(*f);
1874 }
1875 #define _cleanup_unitfilelist_free_ _cleanup_(unitfilelist_free)
1876
1877 int unit_file_get_list(
1878                 UnitFileScope scope,
1879                 const char *root_dir,
1880                 Hashmap *h) {
1881
1882         _cleanup_lookup_paths_free_ LookupPaths paths = {};
1883         char **i;
1884         _cleanup_free_ char *buf = NULL;
1885         _cleanup_closedir_ DIR *d = NULL;
1886         int r;
1887
1888         assert(scope >= 0);
1889         assert(scope < _UNIT_FILE_SCOPE_MAX);
1890         assert(h);
1891
1892         if (root_dir && scope != UNIT_FILE_SYSTEM)
1893                 return -EINVAL;
1894
1895         r = lookup_paths_init_from_scope(&paths, scope);
1896         if (r < 0)
1897                 return r;
1898
1899         STRV_FOREACH(i, paths.unit_path) {
1900                 const char *units_dir;
1901
1902                 free(buf);
1903                 buf = NULL;
1904
1905                 if (root_dir) {
1906                         if (asprintf(&buf, "%s/%s", root_dir, *i) < 0)
1907                                 return -ENOMEM;
1908
1909                         units_dir = buf;
1910                 } else
1911                         units_dir = *i;
1912
1913                 if (d)
1914                         closedir(d);
1915
1916                 d = opendir(units_dir);
1917                 if (!d) {
1918                         if (errno == ENOENT)
1919                                 continue;
1920
1921                         return -errno;
1922                 }
1923
1924                 for (;;) {
1925                         struct dirent *de;
1926                         union dirent_storage buffer;
1927                         _cleanup_unitfilelist_free_ UnitFileList *f = NULL;
1928
1929                         r = readdir_r(d, &buffer.de, &de);
1930                         if (r != 0)
1931                                 return -r;
1932
1933                         if (!de)
1934                                 break;
1935
1936                         if (ignore_file(de->d_name))
1937                                 continue;
1938
1939                         if (!unit_name_is_valid(de->d_name, true))
1940                                 continue;
1941
1942                         if (hashmap_get(h, de->d_name))
1943                                 continue;
1944
1945                         r = dirent_ensure_type(d, de);
1946                         if (r < 0) {
1947                                 if (r == -ENOENT)
1948                                         continue;
1949
1950                                 return r;
1951                         }
1952
1953                         if (de->d_type != DT_LNK && de->d_type != DT_REG)
1954                                 continue;
1955
1956                         f = new0(UnitFileList, 1);
1957                         if (!f)
1958                                 return -ENOMEM;
1959
1960                         f->path = path_make_absolute(de->d_name, units_dir);
1961                         if (!f->path)
1962                                 return -ENOMEM;
1963
1964                         r = null_or_empty_path(f->path);
1965                         if (r < 0 && r != -ENOENT)
1966                                 return r;
1967                         else if (r > 0) {
1968                                 f->state =
1969                                         path_startswith(*i, "/run") ?
1970                                         UNIT_FILE_MASKED_RUNTIME : UNIT_FILE_MASKED;
1971                                 goto found;
1972                         }
1973
1974                         r = find_symlinks_in_scope(scope, root_dir, de->d_name, &f->state);
1975                         if (r < 0)
1976                                 return r;
1977                         else if (r > 0) {
1978                                 f->state = UNIT_FILE_ENABLED;
1979                                 goto found;
1980                         }
1981
1982                         r = unit_file_can_install(&paths, root_dir, f->path, true);
1983                         if (r == -EINVAL ||  /* Invalid setting? */
1984                             r == -EBADMSG || /* Invalid format? */
1985                             r == -ENOENT     /* Included file not found? */)
1986                                 f->state = UNIT_FILE_INVALID;
1987                         else if (r < 0)
1988                                 return r;
1989                         else if (r > 0)
1990                                 f->state = UNIT_FILE_DISABLED;
1991                         else
1992                                 f->state = UNIT_FILE_STATIC;
1993
1994                 found:
1995                         r = hashmap_put(h, path_get_file_name(f->path), f);
1996                         if (r < 0)
1997                                 return r;
1998                         f = NULL; /* prevent cleanup */
1999                 }
2000         }
2001
2002         return r;
2003 }
2004
2005 static const char* const unit_file_state_table[_UNIT_FILE_STATE_MAX] = {
2006         [UNIT_FILE_ENABLED] = "enabled",
2007         [UNIT_FILE_ENABLED_RUNTIME] = "enabled-runtime",
2008         [UNIT_FILE_LINKED] = "linked",
2009         [UNIT_FILE_LINKED_RUNTIME] = "linked-runtime",
2010         [UNIT_FILE_MASKED] = "masked",
2011         [UNIT_FILE_MASKED_RUNTIME] = "masked-runtime",
2012         [UNIT_FILE_STATIC] = "static",
2013         [UNIT_FILE_DISABLED] = "disabled",
2014         [UNIT_FILE_INVALID] = "invalid",
2015 };
2016
2017 DEFINE_STRING_TABLE_LOOKUP(unit_file_state, UnitFileState);
2018
2019 static const char* const unit_file_change_type_table[_UNIT_FILE_CHANGE_TYPE_MAX] = {
2020         [UNIT_FILE_SYMLINK] = "symlink",
2021         [UNIT_FILE_UNLINK] = "unlink",
2022 };
2023
2024 DEFINE_STRING_TABLE_LOOKUP(unit_file_change_type, UnitFileChangeType);