src/unit.c

   1 /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
   2
   3 /***
   4   This file is part of systemd.
   5
   6   Copyright 2010 Lennart Poettering
   7
   8   systemd is free software; you can redistribute it and/or modify it
   9   under the terms of the GNU General Public License as published by
  10   the Free Software Foundation; either version 2 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   General Public License for more details.
  17
  18   You should have received a copy of the GNU General Public License
  19   along with systemd; If not, see <http://www.gnu.org/licenses/>.
  20 ***/
  21
  22 #include <assert.h>
  23 #include <errno.h>
  24 #include <string.h>
  25 #include <sys/epoll.h>
  26 #include <sys/timerfd.h>
  27 #include <sys/poll.h>
  28 #include <stdlib.h>
  29 #include <unistd.h>
  30 #include <sys/stat.h>
  31
  32 #include "set.h"
  33 #include "unit.h"
  34 #include "macro.h"
  35 #include "strv.h"
  36 #include "load-fragment.h"
  37 #include "load-dropin.h"
  38 #include "log.h"
  39 #include "unit-name.h"
  40 #include "specifier.h"
  41 #include "dbus-unit.h"
  42 #include "special.h"
  43 #include "cgroup-util.h"
  44 #include "missing.h"
  45
  46 const UnitVTable * const unit_vtable[_UNIT_TYPE_MAX] = {
  47         [UNIT_SERVICE] = &service_vtable,
  48         [UNIT_TIMER] = &timer_vtable,
  49         [UNIT_SOCKET] = &socket_vtable,
  50         [UNIT_TARGET] = &target_vtable,
  51         [UNIT_DEVICE] = &device_vtable,
  52         [UNIT_MOUNT] = &mount_vtable,
  53         [UNIT_AUTOMOUNT] = &automount_vtable,
  54         [UNIT_SNAPSHOT] = &snapshot_vtable,
  55         [UNIT_SWAP] = &swap_vtable,
  56         [UNIT_PATH] = &path_vtable
  57 };
  58
  59 Unit *unit_new(Manager *m) {
  60         Unit *u;
  61
  62         assert(m);
  63
  64         if (!(u = new0(Unit, 1)))
  65                 return NULL;
  66
  67         if (!(u->meta.names = set_new(string_hash_func, string_compare_func))) {
  68                 free(u);
  69                 return NULL;
  70         }
  71
  72         u->meta.manager = m;
  73         u->meta.type = _UNIT_TYPE_INVALID;
  74         u->meta.deserialized_job = _JOB_TYPE_INVALID;
  75         u->meta.default_dependencies = true;
  76
  77         return u;
  78 }
  79
  80 bool unit_has_name(Unit *u, const char *name) {
  81         assert(u);
  82         assert(name);
  83
  84         return !!set_get(u->meta.names, (char*) name);
  85 }
  86
  87 int unit_add_name(Unit *u, const char *text) {
  88         UnitType t;
  89         char *s, *i = NULL;
  90         int r;
  91
  92         assert(u);
  93         assert(text);
  94
  95         if (unit_name_is_template(text)) {
  96                 if (!u->meta.instance)
  97                         return -EINVAL;
  98
  99                 s = unit_name_replace_instance(text, u->meta.instance);
 100         } else
 101                 s = strdup(text);
 102
 103         if (!s)
 104                 return -ENOMEM;
 105
 106         if (!unit_name_is_valid(s, false)) {
 107                 r = -EINVAL;
 108                 goto fail;
 109         }
 110
 111         assert_se((t = unit_name_to_type(s)) >= 0);
 112
 113         if (u->meta.type != _UNIT_TYPE_INVALID && t != u->meta.type) {
 114                 r = -EINVAL;
 115                 goto fail;
 116         }
 117
 118         if ((r = unit_name_to_instance(s, &i)) < 0)
 119                 goto fail;
 120
 121         if (i && unit_vtable[t]->no_instances)
 122                 goto fail;
 123
 124         /* Ensure that this unit is either instanced or not instanced,
 125          * but not both. */
 126         if (u->meta.type != _UNIT_TYPE_INVALID && !u->meta.instance != !i) {
 127                 r = -EINVAL;
 128                 goto fail;
 129         }
 130
 131         if (unit_vtable[t]->no_alias &&
 132             !set_isempty(u->meta.names) &&
 133             !set_get(u->meta.names, s)) {
 134                 r = -EEXIST;
 135                 goto fail;
 136         }
 137
 138         if (hashmap_size(u->meta.manager->units) >= MANAGER_MAX_NAMES) {
 139                 r = -E2BIG;
 140                 goto fail;
 141         }
 142
 143         if ((r = set_put(u->meta.names, s)) < 0) {
 144                 if (r == -EEXIST)
 145                         r = 0;
 146                 goto fail;
 147         }
 148
 149         if ((r = hashmap_put(u->meta.manager->units, s, u)) < 0) {
 150                 set_remove(u->meta.names, s);
 151                 goto fail;
 152         }
 153
 154         if (u->meta.type == _UNIT_TYPE_INVALID) {
 155
 156                 u->meta.type = t;
 157                 u->meta.id = s;
 158                 u->meta.instance = i;
 159
 160                 LIST_PREPEND(Meta, units_per_type, u->meta.manager->units_per_type[t], &u->meta);
 161
 162                 if (UNIT_VTABLE(u)->init)
 163                         UNIT_VTABLE(u)->init(u);
 164         } else
 165                 free(i);
 166
 167         unit_add_to_dbus_queue(u);
 168         return 0;
 169
 170 fail:
 171         free(s);
 172         free(i);
 173
 174         return r;
 175 }
 176
 177 int unit_choose_id(Unit *u, const char *name) {
 178         char *s, *t = NULL, *i;
 179         int r;
 180
 181         assert(u);
 182         assert(name);
 183
 184         if (unit_name_is_template(name)) {
 185
 186                 if (!u->meta.instance)
 187                         return -EINVAL;
 188
 189                 if (!(t = unit_name_replace_instance(name, u->meta.instance)))
 190                         return -ENOMEM;
 191
 192                 name = t;
 193         }
 194
 195         /* Selects one of the names of this unit as the id */
 196         s = set_get(u->meta.names, (char*) name);
 197         free(t);
 198
 199         if (!s)
 200                 return -ENOENT;
 201
 202         if ((r = unit_name_to_instance(s, &i)) < 0)
 203                 return r;
 204
 205         u->meta.id = s;
 206
 207         free(u->meta.instance);
 208         u->meta.instance = i;
 209
 210         unit_add_to_dbus_queue(u);
 211
 212         return 0;
 213 }
 214
 215 int unit_set_description(Unit *u, const char *description) {
 216         char *s;
 217
 218         assert(u);
 219
 220         if (!(s = strdup(description)))
 221                 return -ENOMEM;
 222
 223         free(u->meta.description);
 224         u->meta.description = s;
 225
 226         unit_add_to_dbus_queue(u);
 227         return 0;
 228 }
 229
 230 bool unit_check_gc(Unit *u) {
 231         assert(u);
 232
 233         if (u->meta.load_state == UNIT_STUB)
 234                 return true;
 235
 236         if (UNIT_VTABLE(u)->no_gc)
 237                 return true;
 238
 239         if (u->meta.no_gc)
 240                 return true;
 241
 242         if (u->meta.job)
 243                 return true;
 244
 245         if (unit_active_state(u) != UNIT_INACTIVE)
 246                 return true;
 247
 248         if (UNIT_VTABLE(u)->check_gc)
 249                 if (UNIT_VTABLE(u)->check_gc(u))
 250                         return true;
 251
 252         return false;
 253 }
 254
 255 void unit_add_to_load_queue(Unit *u) {
 256         assert(u);
 257         assert(u->meta.type != _UNIT_TYPE_INVALID);
 258
 259         if (u->meta.load_state != UNIT_STUB || u->meta.in_load_queue)
 260                 return;
 261
 262         LIST_PREPEND(Meta, load_queue, u->meta.manager->load_queue, &u->meta);
 263         u->meta.in_load_queue = true;
 264 }
 265
 266 void unit_add_to_cleanup_queue(Unit *u) {
 267         assert(u);
 268
 269         if (u->meta.in_cleanup_queue)
 270                 return;
 271
 272         LIST_PREPEND(Meta, cleanup_queue, u->meta.manager->cleanup_queue, &u->meta);
 273         u->meta.in_cleanup_queue = true;
 274 }
 275
 276 void unit_add_to_gc_queue(Unit *u) {
 277         assert(u);
 278
 279         if (u->meta.in_gc_queue || u->meta.in_cleanup_queue)
 280                 return;
 281
 282         if (unit_check_gc(u))
 283                 return;
 284
 285         LIST_PREPEND(Meta, gc_queue, u->meta.manager->gc_queue, &u->meta);
 286         u->meta.in_gc_queue = true;
 287
 288         u->meta.manager->n_in_gc_queue ++;
 289
 290         if (u->meta.manager->gc_queue_timestamp <= 0)
 291                 u->meta.manager->gc_queue_timestamp = now(CLOCK_MONOTONIC);
 292 }
 293
 294 void unit_add_to_dbus_queue(Unit *u) {
 295         assert(u);
 296         assert(u->meta.type != _UNIT_TYPE_INVALID);
 297
 298         if (u->meta.load_state == UNIT_STUB || u->meta.in_dbus_queue)
 299                 return;
 300
 301         /* Shortcut things if nobody cares */
 302         if (!bus_has_subscriber(u->meta.manager)) {
 303                 u->meta.sent_dbus_new_signal = true;
 304                 return;
 305         }
 306
 307         LIST_PREPEND(Meta, dbus_queue, u->meta.manager->dbus_unit_queue, &u->meta);
 308         u->meta.in_dbus_queue = true;
 309 }
 310
 311 static void bidi_set_free(Unit *u, Set *s) {
 312         Iterator i;
 313         Unit *other;
 314
 315         assert(u);
 316
 317         /* Frees the set and makes sure we are dropped from the
 318          * inverse pointers */
 319
 320         SET_FOREACH(other, s, i) {
 321                 UnitDependency d;
 322
 323                 for (d = 0; d < _UNIT_DEPENDENCY_MAX; d++)
 324                         set_remove(other->meta.dependencies[d], u);
 325
 326                 unit_add_to_gc_queue(other);
 327         }
 328
 329         set_free(s);
 330 }
 331
 332 void unit_free(Unit *u) {
 333         UnitDependency d;
 334         Iterator i;
 335         char *t;
 336
 337         assert(u);
 338
 339         bus_unit_send_removed_signal(u);
 340
 341         if (u->meta.load_state != UNIT_STUB)
 342                 if (UNIT_VTABLE(u)->done)
 343                         UNIT_VTABLE(u)->done(u);
 344
 345         SET_FOREACH(t, u->meta.names, i)
 346                 hashmap_remove_value(u->meta.manager->units, t, u);
 347
 348         if (u->meta.job)
 349                 job_free(u->meta.job);
 350
 351         for (d = 0; d < _UNIT_DEPENDENCY_MAX; d++)
 352                 bidi_set_free(u, u->meta.dependencies[d]);
 353
 354         if (u->meta.type != _UNIT_TYPE_INVALID)
 355                 LIST_REMOVE(Meta, units_per_type, u->meta.manager->units_per_type[u->meta.type], &u->meta);
 356
 357         if (u->meta.in_load_queue)
 358                 LIST_REMOVE(Meta, load_queue, u->meta.manager->load_queue, &u->meta);
 359
 360         if (u->meta.in_dbus_queue)
 361                 LIST_REMOVE(Meta, dbus_queue, u->meta.manager->dbus_unit_queue, &u->meta);
 362
 363         if (u->meta.in_cleanup_queue)
 364                 LIST_REMOVE(Meta, cleanup_queue, u->meta.manager->cleanup_queue, &u->meta);
 365
 366         if (u->meta.in_gc_queue) {
 367                 LIST_REMOVE(Meta, gc_queue, u->meta.manager->gc_queue, &u->meta);
 368                 u->meta.manager->n_in_gc_queue--;
 369         }
 370
 371         cgroup_bonding_free_list(u->meta.cgroup_bondings);
 372
 373         free(u->meta.description);
 374         free(u->meta.fragment_path);
 375
 376         set_free_free(u->meta.names);
 377
 378         condition_free_list(u->meta.conditions);
 379
 380         free(u->meta.instance);
 381         free(u);
 382 }
 383
 384 UnitActiveState unit_active_state(Unit *u) {
 385         assert(u);
 386
 387         if (u->meta.load_state == UNIT_MERGED)
 388                 return unit_active_state(unit_follow_merge(u));
 389
 390         /* After a reload it might happen that a unit is not correctly
 391          * loaded but still has a process around. That's why we won't
 392          * shortcut failed loading to UNIT_INACTIVE_FAILED. */
 393
 394         return UNIT_VTABLE(u)->active_state(u);
 395 }
 396
 397 const char* unit_sub_state_to_string(Unit *u) {
 398         assert(u);
 399
 400         return UNIT_VTABLE(u)->sub_state_to_string(u);
 401 }
 402
 403 static void complete_move(Set **s, Set **other) {
 404         assert(s);
 405         assert(other);
 406
 407         if (!*other)
 408                 return;
 409
 410         if (*s)
 411                 set_move(*s, *other);
 412         else {
 413                 *s = *other;
 414                 *other = NULL;
 415         }
 416 }
 417
 418 static void merge_names(Unit *u, Unit *other) {
 419         char *t;
 420         Iterator i;
 421
 422         assert(u);
 423         assert(other);
 424
 425         complete_move(&u->meta.names, &other->meta.names);
 426
 427         set_free_free(other->meta.names);
 428         other->meta.names = NULL;
 429         other->meta.id = NULL;
 430
 431         SET_FOREACH(t, u->meta.names, i)
 432                 assert_se(hashmap_replace(u->meta.manager->units, t, u) == 0);
 433 }
 434
 435 static void merge_dependencies(Unit *u, Unit *other, UnitDependency d) {
 436         Iterator i;
 437         Unit *back;
 438         int r;
 439
 440         assert(u);
 441         assert(other);
 442         assert(d < _UNIT_DEPENDENCY_MAX);
 443
 444         /* Fix backwards pointers */
 445         SET_FOREACH(back, other->meta.dependencies[d], i) {
 446                 UnitDependency k;
 447
 448                 for (k = 0; k < _UNIT_DEPENDENCY_MAX; k++)
 449                         if ((r = set_remove_and_put(back->meta.dependencies[k], other, u)) < 0) {
 450
 451                                 if (r == -EEXIST)
 452                                         set_remove(back->meta.dependencies[k], other);
 453                                 else
 454                                         assert(r == -ENOENT);
 455                         }
 456         }
 457
 458         complete_move(&u->meta.dependencies[d], &other->meta.dependencies[d]);
 459
 460         set_free(other->meta.dependencies[d]);
 461         other->meta.dependencies[d] = NULL;
 462 }
 463
 464 int unit_merge(Unit *u, Unit *other) {
 465         UnitDependency d;
 466
 467         assert(u);
 468         assert(other);
 469         assert(u->meta.manager == other->meta.manager);
 470         assert(u->meta.type != _UNIT_TYPE_INVALID);
 471
 472         other = unit_follow_merge(other);
 473
 474         if (other == u)
 475                 return 0;
 476
 477         if (u->meta.type != other->meta.type)
 478                 return -EINVAL;
 479
 480         if (!u->meta.instance != !other->meta.instance)
 481                 return -EINVAL;
 482
 483         if (other->meta.load_state != UNIT_STUB &&
 484             other->meta.load_state != UNIT_ERROR)
 485                 return -EEXIST;
 486
 487         if (other->meta.job)
 488                 return -EEXIST;
 489
 490         if (!UNIT_IS_INACTIVE_OR_FAILED(unit_active_state(other)))
 491                 return -EEXIST;
 492
 493         /* Merge names */
 494         merge_names(u, other);
 495
 496         /* Merge dependencies */
 497         for (d = 0; d < _UNIT_DEPENDENCY_MAX; d++)
 498                 merge_dependencies(u, other, d);
 499
 500         other->meta.load_state = UNIT_MERGED;
 501         other->meta.merged_into = u;
 502
 503         /* If there is still some data attached to the other node, we
 504          * don't need it anymore, and can free it. */
 505         if (other->meta.load_state != UNIT_STUB)
 506                 if (UNIT_VTABLE(other)->done)
 507                         UNIT_VTABLE(other)->done(other);
 508
 509         unit_add_to_dbus_queue(u);
 510         unit_add_to_cleanup_queue(other);
 511
 512         return 0;
 513 }
 514
 515 int unit_merge_by_name(Unit *u, const char *name) {
 516         Unit *other;
 517         int r;
 518         char *s = NULL;
 519
 520         assert(u);
 521         assert(name);
 522
 523         if (unit_name_is_template(name)) {
 524                 if (!u->meta.instance)
 525                         return -EINVAL;
 526
 527                 if (!(s = unit_name_replace_instance(name, u->meta.instance)))
 528                         return -ENOMEM;
 529
 530                 name = s;
 531         }
 532
 533         if (!(other = manager_get_unit(u->meta.manager, name)))
 534                 r = unit_add_name(u, name);
 535         else
 536                 r = unit_merge(u, other);
 537
 538         free(s);
 539         return r;
 540 }
 541
 542 Unit* unit_follow_merge(Unit *u) {
 543         assert(u);
 544
 545         while (u->meta.load_state == UNIT_MERGED)
 546                 assert_se(u = u->meta.merged_into);
 547
 548         return u;
 549 }
 550
 551 int unit_add_exec_dependencies(Unit *u, ExecContext *c) {
 552         int r;
 553
 554         assert(u);
 555         assert(c);
 556
 557         if (c->std_output != EXEC_OUTPUT_KMSG &&
 558             c->std_output != EXEC_OUTPUT_SYSLOG &&
 559             c->std_output != EXEC_OUTPUT_KMSG_AND_CONSOLE &&
 560             c->std_output != EXEC_OUTPUT_SYSLOG_AND_CONSOLE &&
 561             c->std_error != EXEC_OUTPUT_KMSG &&
 562             c->std_error != EXEC_OUTPUT_SYSLOG_AND_CONSOLE &&
 563             c->std_error != EXEC_OUTPUT_KMSG &&
 564             c->std_error != EXEC_OUTPUT_SYSLOG_AND_CONSOLE)
 565                 return 0;
 566
 567         /* If syslog or kernel logging is requested, make sure our own
 568          * logging daemon is run first. */
 569
 570         if ((r = unit_add_dependency_by_name(u, UNIT_AFTER, SPECIAL_LOGGER_SOCKET, NULL, true)) < 0)
 571                 return r;
 572
 573         if (u->meta.manager->running_as == MANAGER_SYSTEM)
 574                 if ((r = unit_add_dependency_by_name(u, UNIT_REQUIRES, SPECIAL_LOGGER_SOCKET, NULL, true)) < 0)
 575                         return r;
 576
 577         return 0;
 578 }
 579
 580 const char *unit_description(Unit *u) {
 581         assert(u);
 582
 583         if (u->meta.description)
 584                 return u->meta.description;
 585
 586         return strna(u->meta.id);
 587 }
 588
 589 void unit_dump(Unit *u, FILE *f, const char *prefix) {
 590         char *t;
 591         UnitDependency d;
 592         Iterator i;
 593         char *p2;
 594         const char *prefix2;
 595         CGroupBonding *b;
 596         char
 597                 timestamp1[FORMAT_TIMESTAMP_MAX],
 598                 timestamp2[FORMAT_TIMESTAMP_MAX],
 599                 timestamp3[FORMAT_TIMESTAMP_MAX],
 600                 timestamp4[FORMAT_TIMESTAMP_MAX],
 601                 timespan[FORMAT_TIMESPAN_MAX];
 602         Unit *following;
 603
 604         assert(u);
 605         assert(u->meta.type >= 0);
 606
 607         if (!prefix)
 608                 prefix = "";
 609         p2 = strappend(prefix, "\t");
 610         prefix2 = p2 ? p2 : prefix;
 611
 612         fprintf(f,
 613                 "%s-> Unit %s:\n"
 614                 "%s\tDescription: %s\n"
 615                 "%s\tInstance: %s\n"
 616                 "%s\tUnit Load State: %s\n"
 617                 "%s\tUnit Active State: %s\n"
 618                 "%s\tInactive Exit Timestamp: %s\n"
 619                 "%s\tActive Enter Timestamp: %s\n"
 620                 "%s\tActive Exit Timestamp: %s\n"
 621                 "%s\tInactive Enter Timestamp: %s\n"
 622                 "%s\tGC Check Good: %s\n"
 623                 "%s\tNeed Daemon Reload: %s\n",
 624                 prefix, u->meta.id,
 625                 prefix, unit_description(u),
 626                 prefix, strna(u->meta.instance),
 627                 prefix, unit_load_state_to_string(u->meta.load_state),
 628                 prefix, unit_active_state_to_string(unit_active_state(u)),
 629                 prefix, strna(format_timestamp(timestamp1, sizeof(timestamp1), u->meta.inactive_exit_timestamp.realtime)),
 630                 prefix, strna(format_timestamp(timestamp2, sizeof(timestamp2), u->meta.active_enter_timestamp.realtime)),
 631                 prefix, strna(format_timestamp(timestamp3, sizeof(timestamp3), u->meta.active_exit_timestamp.realtime)),
 632                 prefix, strna(format_timestamp(timestamp4, sizeof(timestamp4), u->meta.inactive_enter_timestamp.realtime)),
 633                 prefix, yes_no(unit_check_gc(u)),
 634                 prefix, yes_no(unit_need_daemon_reload(u)));
 635
 636         SET_FOREACH(t, u->meta.names, i)
 637                 fprintf(f, "%s\tName: %s\n", prefix, t);
 638
 639         if ((following = unit_following(u)))
 640                 fprintf(f, "%s\tFollowing: %s\n", prefix, following->meta.id);
 641
 642         if (u->meta.fragment_path)
 643                 fprintf(f, "%s\tFragment Path: %s\n", prefix, u->meta.fragment_path);
 644
 645         if (u->meta.job_timeout > 0)
 646                 fprintf(f, "%s\tJob Timeout: %s\n", prefix, format_timespan(timespan, sizeof(timespan), u->meta.job_timeout));
 647
 648         condition_dump_list(u->meta.conditions, f, prefix);
 649
 650         if (dual_timestamp_is_set(&u->meta.condition_timestamp))
 651                 fprintf(f,
 652                         "%s\tCondition Timestamp: %s\n"
 653                         "%s\tCondition Result: %s\n",
 654                         prefix, strna(format_timestamp(timestamp1, sizeof(timestamp1), u->meta.condition_timestamp.realtime)),
 655                         prefix, yes_no(u->meta.condition_result));
 656
 657         for (d = 0; d < _UNIT_DEPENDENCY_MAX; d++) {
 658                 Unit *other;
 659
 660                 SET_FOREACH(other, u->meta.dependencies[d], i)
 661                         fprintf(f, "%s\t%s: %s\n", prefix, unit_dependency_to_string(d), other->meta.id);
 662         }
 663
 664         if (u->meta.load_state == UNIT_LOADED) {
 665                 fprintf(f,
 666                         "%s\tStopWhenUnneeded: %s\n"
 667                         "%s\tRefuseManualStart: %s\n"
 668                         "%s\tRefuseManualStop: %s\n"
 669                         "%s\tDefaultDependencies: %s\n",
 670                         prefix, yes_no(u->meta.stop_when_unneeded),
 671                         prefix, yes_no(u->meta.refuse_manual_start),
 672                         prefix, yes_no(u->meta.refuse_manual_stop),
 673                         prefix, yes_no(u->meta.default_dependencies));
 674
 675                 LIST_FOREACH(by_unit, b, u->meta.cgroup_bondings)
 676                         fprintf(f, "%s\tControlGroup: %s:%s\n",
 677                                 prefix, b->controller, b->path);
 678
 679                 if (UNIT_VTABLE(u)->dump)
 680                         UNIT_VTABLE(u)->dump(u, f, prefix2);
 681
 682         } else if (u->meta.load_state == UNIT_MERGED)
 683                 fprintf(f,
 684                         "%s\tMerged into: %s\n",
 685                         prefix, u->meta.merged_into->meta.id);
 686         else if (u->meta.load_state == UNIT_ERROR)
 687                 fprintf(f, "%s\tLoad Error Code: %s\n", prefix, strerror(-u->meta.load_error));
 688
 689
 690         if (u->meta.job)
 691                 job_dump(u->meta.job, f, prefix2);
 692
 693         free(p2);
 694 }
 695
 696 /* Common implementation for multiple backends */
 697 int unit_load_fragment_and_dropin(Unit *u) {
 698         int r;
 699
 700         assert(u);
 701
 702         /* Load a .service file */
 703         if ((r = unit_load_fragment(u)) < 0)
 704                 return r;
 705
 706         if (u->meta.load_state == UNIT_STUB)
 707                 return -ENOENT;
 708
 709         /* Load drop-in directory data */
 710         if ((r = unit_load_dropin(unit_follow_merge(u))) < 0)
 711                 return r;
 712
 713         return 0;
 714 }
 715
 716 /* Common implementation for multiple backends */
 717 int unit_load_fragment_and_dropin_optional(Unit *u) {
 718         int r;
 719
 720         assert(u);
 721
 722         /* Same as unit_load_fragment_and_dropin(), but whether
 723          * something can be loaded or not doesn't matter. */
 724
 725         /* Load a .service file */
 726         if ((r = unit_load_fragment(u)) < 0)
 727                 return r;
 728
 729         if (u->meta.load_state == UNIT_STUB)
 730                 u->meta.load_state = UNIT_LOADED;
 731
 732         /* Load drop-in directory data */
 733         if ((r = unit_load_dropin(unit_follow_merge(u))) < 0)
 734                 return r;
 735
 736         return 0;
 737 }
 738
 739 int unit_add_default_target_dependency(Unit *u, Unit *target) {
 740         assert(u);
 741         assert(target);
 742
 743         if (target->meta.type != UNIT_TARGET)
 744                 return 0;
 745
 746         /* Only add the dependency if both units are loaded, so that
 747          * that loop check below is reliable */
 748         if (u->meta.load_state != UNIT_LOADED ||
 749             target->meta.load_state != UNIT_LOADED)
 750                 return 0;
 751
 752         /* If either side wants no automatic dependencies, then let's
 753          * skip this */
 754         if (!u->meta.default_dependencies ||
 755             target->meta.default_dependencies)
 756                 return 0;
 757
 758         /* Don't create loops */
 759         if (set_get(target->meta.dependencies[UNIT_BEFORE], u))
 760                 return 0;
 761
 762         return unit_add_dependency(target, UNIT_AFTER, u, true);
 763 }
 764
 765 static int unit_add_default_dependencies(Unit *u) {
 766         static const UnitDependency deps[] = {
 767                 UNIT_REQUIRED_BY,
 768                 UNIT_REQUIRED_BY_OVERRIDABLE,
 769                 UNIT_WANTED_BY,
 770                 UNIT_BOUND_BY
 771         };
 772
 773         Unit *target;
 774         Iterator i;
 775         int r;
 776         unsigned k;
 777
 778         assert(u);
 779
 780         for (k = 0; k < ELEMENTSOF(deps); k++)
 781                 SET_FOREACH(target, u->meta.dependencies[deps[k]], i)
 782                         if ((r = unit_add_default_target_dependency(u, target)) < 0)
 783                                 return r;
 784
 785         return 0;
 786 }
 787
 788 int unit_load(Unit *u) {
 789         int r;
 790
 791         assert(u);
 792
 793         if (u->meta.in_load_queue) {
 794                 LIST_REMOVE(Meta, load_queue, u->meta.manager->load_queue, &u->meta);
 795                 u->meta.in_load_queue = false;
 796         }
 797
 798         if (u->meta.type == _UNIT_TYPE_INVALID)
 799                 return -EINVAL;
 800
 801         if (u->meta.load_state != UNIT_STUB)
 802                 return 0;
 803
 804         if (UNIT_VTABLE(u)->load)
 805                 if ((r = UNIT_VTABLE(u)->load(u)) < 0)
 806                         goto fail;
 807
 808         if (u->meta.load_state == UNIT_STUB) {
 809                 r = -ENOENT;
 810                 goto fail;
 811         }
 812
 813         if (u->meta.load_state == UNIT_LOADED &&
 814             u->meta.default_dependencies)
 815                 if ((r = unit_add_default_dependencies(u)) < 0)
 816                         goto fail;
 817
 818         assert((u->meta.load_state != UNIT_MERGED) == !u->meta.merged_into);
 819
 820         unit_add_to_dbus_queue(unit_follow_merge(u));
 821         unit_add_to_gc_queue(u);
 822
 823         return 0;
 824
 825 fail:
 826         u->meta.load_state = UNIT_ERROR;
 827         u->meta.load_error = r;
 828         unit_add_to_dbus_queue(u);
 829
 830         log_debug("Failed to load configuration for %s: %s", u->meta.id, strerror(-r));
 831
 832         return r;
 833 }
 834
 835 bool unit_condition_test(Unit *u) {
 836         assert(u);
 837
 838         dual_timestamp_get(&u->meta.condition_timestamp);
 839         u->meta.condition_result = condition_test_list(u->meta.conditions);
 840
 841         return u->meta.condition_result;
 842 }
 843
 844 /* Errors:
 845  *         -EBADR:     This unit type does not support starting.
 846  *         -EALREADY:  Unit is already started.
 847  *         -EAGAIN:    An operation is already in progress. Retry later.
 848  *         -ECANCELED: Too many requests for now.
 849  */
 850 int unit_start(Unit *u) {
 851         UnitActiveState state;
 852         Unit *following;
 853
 854         assert(u);
 855
 856         if (u->meta.load_state != UNIT_LOADED)
 857                 return -EINVAL;
 858
 859         /* If this is already (being) started, then this will
 860          * succeed. Note that this will even succeed if this unit is
 861          * not startable by the user. This is relied on to detect when
 862          * we need to wait for units and when waiting is finished. */
 863         state = unit_active_state(u);
 864         if (UNIT_IS_ACTIVE_OR_RELOADING(state))
 865                 return -EALREADY;
 866
 867         /* If the conditions failed, don't do anything at all */
 868         if (!unit_condition_test(u)) {
 869                 log_debug("Starting of %s requested but condition failed. Ignoring.", u->meta.id);
 870                 return -EALREADY;
 871         }
 872
 873         /* Forward to the main object, if we aren't it. */
 874         if ((following = unit_following(u))) {
 875                 log_debug("Redirecting start request from %s to %s.", u->meta.id, following->meta.id);
 876                 return unit_start(following);
 877         }
 878
 879         /* If it is stopped, but we cannot start it, then fail */
 880         if (!UNIT_VTABLE(u)->start)
 881                 return -EBADR;
 882
 883         /* We don't suppress calls to ->start() here when we are
 884          * already starting, to allow this request to be used as a
 885          * "hurry up" call, for example when the unit is in some "auto
 886          * restart" state where it waits for a holdoff timer to elapse
 887          * before it will start again. */
 888
 889         unit_add_to_dbus_queue(u);
 890
 891         unit_status_printf(u, "Starting %s...\n", unit_description(u));
 892         return UNIT_VTABLE(u)->start(u);
 893 }
 894
 895 bool unit_can_start(Unit *u) {
 896         assert(u);
 897
 898         return !!UNIT_VTABLE(u)->start;
 899 }
 900
 901 bool unit_can_isolate(Unit *u) {
 902         assert(u);
 903
 904         return unit_can_start(u) &&
 905                 u->meta.allow_isolate;
 906 }
 907
 908 /* Errors:
 909  *         -EBADR:    This unit type does not support stopping.
 910  *         -EALREADY: Unit is already stopped.
 911  *         -EAGAIN:   An operation is already in progress. Retry later.
 912  */
 913 int unit_stop(Unit *u) {
 914         UnitActiveState state;
 915         Unit *following;
 916
 917         assert(u);
 918
 919         state = unit_active_state(u);
 920         if (UNIT_IS_INACTIVE_OR_FAILED(state))
 921                 return -EALREADY;
 922
 923         if ((following = unit_following(u))) {
 924                 log_debug("Redirecting stop request from %s to %s.", u->meta.id, following->meta.id);
 925                 return unit_stop(following);
 926         }
 927
 928         if (!UNIT_VTABLE(u)->stop)
 929                 return -EBADR;
 930
 931         unit_add_to_dbus_queue(u);
 932
 933         unit_status_printf(u, "Stopping %s...\n", unit_description(u));
 934         return UNIT_VTABLE(u)->stop(u);
 935 }
 936
 937 /* Errors:
 938  *         -EBADR:    This unit type does not support reloading.
 939  *         -ENOEXEC:  Unit is not started.
 940  *         -EAGAIN:   An operation is already in progress. Retry later.
 941  */
 942 int unit_reload(Unit *u) {
 943         UnitActiveState state;
 944         Unit *following;
 945
 946         assert(u);
 947
 948         if (u->meta.load_state != UNIT_LOADED)
 949                 return -EINVAL;
 950
 951         if (!unit_can_reload(u))
 952                 return -EBADR;
 953
 954         state = unit_active_state(u);
 955         if (state == UNIT_RELOADING)
 956                 return -EALREADY;
 957
 958         if (state != UNIT_ACTIVE)
 959                 return -ENOEXEC;
 960
 961         if ((following = unit_following(u))) {
 962                 log_debug("Redirecting reload request from %s to %s.", u->meta.id, following->meta.id);
 963                 return unit_reload(following);
 964         }
 965
 966         unit_add_to_dbus_queue(u);
 967         return UNIT_VTABLE(u)->reload(u);
 968 }
 969
 970 bool unit_can_reload(Unit *u) {
 971         assert(u);
 972
 973         if (!UNIT_VTABLE(u)->reload)
 974                 return false;
 975
 976         if (!UNIT_VTABLE(u)->can_reload)
 977                 return true;
 978
 979         return UNIT_VTABLE(u)->can_reload(u);
 980 }
 981
 982 static void unit_check_unneeded(Unit *u) {
 983         Iterator i;
 984         Unit *other;
 985
 986         assert(u);
 987
 988         /* If this service shall be shut down when unneeded then do
 989          * so. */
 990
 991         if (!u->meta.stop_when_unneeded)
 992                 return;
 993
 994         if (!UNIT_IS_ACTIVE_OR_ACTIVATING(unit_active_state(u)))
 995                 return;
 996
 997         SET_FOREACH(other, u->meta.dependencies[UNIT_REQUIRED_BY], i)
 998                 if (!UNIT_IS_INACTIVE_OR_DEACTIVATING(unit_active_state(other)))
 999                         return;
1000
1001         SET_FOREACH(other, u->meta.dependencies[UNIT_REQUIRED_BY_OVERRIDABLE], i)
1002                 if (!UNIT_IS_INACTIVE_OR_DEACTIVATING(unit_active_state(other)))
1003                         return;
1004
1005         SET_FOREACH(other, u->meta.dependencies[UNIT_WANTED_BY], i)
1006                 if (!UNIT_IS_INACTIVE_OR_DEACTIVATING(unit_active_state(other)))
1007                         return;
1008
1009         SET_FOREACH(other, u->meta.dependencies[UNIT_BOUND_BY], i)
1010                 if (!UNIT_IS_INACTIVE_OR_DEACTIVATING(unit_active_state(other)))
1011                         return;
1012
1013         log_info("Service %s is not needed anymore. Stopping.", u->meta.id);
1014
1015         /* Ok, nobody needs us anymore. Sniff. Then let's commit suicide */
1016         manager_add_job(u->meta.manager, JOB_STOP, u, JOB_FAIL, true, NULL, NULL);
1017 }
1018
1019 static void retroactively_start_dependencies(Unit *u) {
1020         Iterator i;
1021         Unit *other;
1022
1023         assert(u);
1024         assert(UNIT_IS_ACTIVE_OR_ACTIVATING(unit_active_state(u)));
1025
1026         SET_FOREACH(other, u->meta.dependencies[UNIT_REQUIRES], i)
1027                 if (!set_get(u->meta.dependencies[UNIT_AFTER], other) &&
1028                     !UNIT_IS_ACTIVE_OR_ACTIVATING(unit_active_state(other)))
1029                         manager_add_job(u->meta.manager, JOB_START, other, JOB_REPLACE, true, NULL, NULL);
1030
1031         SET_FOREACH(other, u->meta.dependencies[UNIT_BIND_TO], i)
1032                 if (!set_get(u->meta.dependencies[UNIT_AFTER], other) &&
1033                     !UNIT_IS_ACTIVE_OR_ACTIVATING(unit_active_state(other)))
1034                         manager_add_job(u->meta.manager, JOB_START, other, JOB_REPLACE, true, NULL, NULL);
1035
1036         SET_FOREACH(other, u->meta.dependencies[UNIT_REQUIRES_OVERRIDABLE], i)
1037                 if (!set_get(u->meta.dependencies[UNIT_AFTER], other) &&
1038                     !UNIT_IS_ACTIVE_OR_ACTIVATING(unit_active_state(other)))
1039                         manager_add_job(u->meta.manager, JOB_START, other, JOB_FAIL, false, NULL, NULL);
1040
1041         SET_FOREACH(other, u->meta.dependencies[UNIT_REQUISITE], i)
1042                 if (!set_get(u->meta.dependencies[UNIT_AFTER], other) &&
1043                     !UNIT_IS_ACTIVE_OR_ACTIVATING(unit_active_state(other)))
1044                         manager_add_job(u->meta.manager, JOB_START, other, JOB_REPLACE, true, NULL, NULL);
1045
1046         SET_FOREACH(other, u->meta.dependencies[UNIT_WANTS], i)
1047                 if (!set_get(u->meta.dependencies[UNIT_AFTER], other) &&
1048                     !UNIT_IS_ACTIVE_OR_ACTIVATING(unit_active_state(other)))
1049                         manager_add_job(u->meta.manager, JOB_START, other, JOB_FAIL, false, NULL, NULL);
1050
1051         SET_FOREACH(other, u->meta.dependencies[UNIT_CONFLICTS], i)
1052                 if (!UNIT_IS_INACTIVE_OR_DEACTIVATING(unit_active_state(other)))
1053                         manager_add_job(u->meta.manager, JOB_STOP, other, JOB_REPLACE, true, NULL, NULL);
1054
1055         SET_FOREACH(other, u->meta.dependencies[UNIT_CONFLICTED_BY], i)
1056                 if (!UNIT_IS_INACTIVE_OR_DEACTIVATING(unit_active_state(other)))
1057                         manager_add_job(u->meta.manager, JOB_STOP, other, JOB_REPLACE, true, NULL, NULL);
1058 }
1059
1060 static void retroactively_stop_dependencies(Unit *u) {
1061         Iterator i;
1062         Unit *other;
1063
1064         assert(u);
1065         assert(UNIT_IS_INACTIVE_OR_DEACTIVATING(unit_active_state(u)));
1066
1067         /* Pull down units which are bound to us recursively if enabled */
1068         SET_FOREACH(other, u->meta.dependencies[UNIT_BOUND_BY], i)
1069                 if (!UNIT_IS_INACTIVE_OR_DEACTIVATING(unit_active_state(other)))
1070                         manager_add_job(u->meta.manager, JOB_STOP, other, JOB_REPLACE, true, NULL, NULL);
1071
1072         /* Garbage collect services that might not be needed anymore, if enabled */
1073         SET_FOREACH(other, u->meta.dependencies[UNIT_REQUIRES], i)
1074                 if (!UNIT_IS_INACTIVE_OR_DEACTIVATING(unit_active_state(other)))
1075                         unit_check_unneeded(other);
1076         SET_FOREACH(other, u->meta.dependencies[UNIT_REQUIRES_OVERRIDABLE], i)
1077                 if (!UNIT_IS_INACTIVE_OR_DEACTIVATING(unit_active_state(other)))
1078                         unit_check_unneeded(other);
1079         SET_FOREACH(other, u->meta.dependencies[UNIT_WANTS], i)
1080                 if (!UNIT_IS_INACTIVE_OR_DEACTIVATING(unit_active_state(other)))
1081                         unit_check_unneeded(other);
1082         SET_FOREACH(other, u->meta.dependencies[UNIT_REQUISITE], i)
1083                 if (!UNIT_IS_INACTIVE_OR_DEACTIVATING(unit_active_state(other)))
1084                         unit_check_unneeded(other);
1085         SET_FOREACH(other, u->meta.dependencies[UNIT_REQUISITE_OVERRIDABLE], i)
1086                 if (!UNIT_IS_INACTIVE_OR_DEACTIVATING(unit_active_state(other)))
1087                         unit_check_unneeded(other);
1088         SET_FOREACH(other, u->meta.dependencies[UNIT_BIND_TO], i)
1089                 if (!UNIT_IS_INACTIVE_OR_DEACTIVATING(unit_active_state(other)))
1090                         unit_check_unneeded(other);
1091 }
1092
1093 void unit_trigger_on_failure(Unit *u) {
1094         Unit *other;
1095         Iterator i;
1096
1097         assert(u);
1098
1099         SET_FOREACH(other, u->meta.dependencies[UNIT_ON_FAILURE], i)
1100                 manager_add_job(u->meta.manager, JOB_START, other, JOB_REPLACE, true, NULL, NULL);
1101 }
1102
1103 void unit_notify(Unit *u, UnitActiveState os, UnitActiveState ns, bool reload_success) {
1104         bool unexpected;
1105
1106         assert(u);
1107         assert(os < _UNIT_ACTIVE_STATE_MAX);
1108         assert(ns < _UNIT_ACTIVE_STATE_MAX);
1109
1110         /* Note that this is called for all low-level state changes,
1111          * even if they might map to the same high-level
1112          * UnitActiveState! That means that ns == os is OK an expected
1113          * behaviour here. For example: if a mount point is remounted
1114          * this function will be called too! */
1115
1116         if (u->meta.manager->n_deserializing <= 0) {
1117                 dual_timestamp ts;
1118
1119                 dual_timestamp_get(&ts);
1120
1121                 if (UNIT_IS_INACTIVE_OR_FAILED(os) && !UNIT_IS_INACTIVE_OR_FAILED(ns))
1122                         u->meta.inactive_exit_timestamp = ts;
1123                 else if (!UNIT_IS_INACTIVE_OR_FAILED(os) && UNIT_IS_INACTIVE_OR_FAILED(ns))
1124                         u->meta.inactive_enter_timestamp = ts;
1125
1126                 if (!UNIT_IS_ACTIVE_OR_RELOADING(os) && UNIT_IS_ACTIVE_OR_RELOADING(ns))
1127                         u->meta.active_enter_timestamp = ts;
1128                 else if (UNIT_IS_ACTIVE_OR_RELOADING(os) && !UNIT_IS_ACTIVE_OR_RELOADING(ns))
1129                         u->meta.active_exit_timestamp = ts;
1130
1131                 timer_unit_notify(u, ns);
1132                 path_unit_notify(u, ns);
1133         }
1134
1135         if (UNIT_IS_INACTIVE_OR_FAILED(ns))
1136                 cgroup_bonding_trim_list(u->meta.cgroup_bondings, true);
1137
1138         if (u->meta.job) {
1139                 unexpected = false;
1140
1141                 if (u->meta.job->state == JOB_WAITING)
1142
1143                         /* So we reached a different state for this
1144                          * job. Let's see if we can run it now if it
1145                          * failed previously due to EAGAIN. */
1146                         job_add_to_run_queue(u->meta.job);
1147
1148                 /* Let's check whether this state change constitutes a
1149                  * finished job, or maybe contradicts a running job and
1150                  * hence needs to invalidate jobs. */
1151
1152                 switch (u->meta.job->type) {
1153
1154                 case JOB_START:
1155                 case JOB_VERIFY_ACTIVE:
1156
1157                         if (UNIT_IS_ACTIVE_OR_RELOADING(ns))
1158                                 job_finish_and_invalidate(u->meta.job, JOB_DONE);
1159                         else if (u->meta.job->state == JOB_RUNNING && ns != UNIT_ACTIVATING) {
1160                                 unexpected = true;
1161
1162                                 if (UNIT_IS_INACTIVE_OR_FAILED(ns))
1163                                         job_finish_and_invalidate(u->meta.job, ns == UNIT_FAILED ? JOB_FAILED : JOB_DONE);
1164                         }
1165
1166                         break;
1167
1168                 case JOB_RELOAD:
1169                 case JOB_RELOAD_OR_START:
1170
1171                         if (u->meta.job->state == JOB_RUNNING) {
1172                                 if (ns == UNIT_ACTIVE)
1173                                         job_finish_and_invalidate(u->meta.job, reload_success ? JOB_DONE : JOB_FAILED);
1174                                 else if (ns != UNIT_ACTIVATING && ns != UNIT_RELOADING) {
1175                                         unexpected = true;
1176
1177                                         if (UNIT_IS_INACTIVE_OR_FAILED(ns))
1178                                                 job_finish_and_invalidate(u->meta.job, ns == UNIT_FAILED ? JOB_FAILED : JOB_DONE);
1179                                 }
1180                         }
1181
1182                         break;
1183
1184                 case JOB_STOP:
1185                 case JOB_RESTART:
1186                 case JOB_TRY_RESTART:
1187
1188                         if (UNIT_IS_INACTIVE_OR_FAILED(ns))
1189                                 job_finish_and_invalidate(u->meta.job, JOB_DONE);
1190                         else if (u->meta.job->state == JOB_RUNNING && ns != UNIT_DEACTIVATING) {
1191                                 unexpected = true;
1192                                 job_finish_and_invalidate(u->meta.job, JOB_FAILED);
1193                         }
1194
1195                         break;
1196
1197                 default:
1198                         assert_not_reached("Job type unknown");
1199                 }
1200
1201         } else
1202                 unexpected = true;
1203
1204         if (u->meta.manager->n_deserializing <= 0) {
1205
1206                 /* If this state change happened without being
1207                  * requested by a job, then let's retroactively start
1208                  * or stop dependencies. We skip that step when
1209                  * deserializing, since we don't want to create any
1210                  * additional jobs just because something is already
1211                  * activated. */
1212
1213                 if (unexpected) {
1214                         if (UNIT_IS_INACTIVE_OR_FAILED(os) && UNIT_IS_ACTIVE_OR_ACTIVATING(ns))
1215                                 retroactively_start_dependencies(u);
1216                         else if (UNIT_IS_ACTIVE_OR_ACTIVATING(os) && UNIT_IS_INACTIVE_OR_DEACTIVATING(ns))
1217                                 retroactively_stop_dependencies(u);
1218                 }
1219
1220                 if (ns != os && ns == UNIT_FAILED) {
1221                         log_notice("Unit %s entered failed state.", u->meta.id);
1222                         unit_trigger_on_failure(u);
1223                 }
1224         }
1225
1226         /* Some names are special */
1227         if (UNIT_IS_ACTIVE_OR_RELOADING(ns)) {
1228
1229                 if (unit_has_name(u, SPECIAL_DBUS_SERVICE))
1230                         /* The bus just might have become available,
1231                          * hence try to connect to it, if we aren't
1232                          * yet connected. */
1233                         bus_init(u->meta.manager, true);
1234
1235                 if (u->meta.type == UNIT_SERVICE &&
1236                     !UNIT_IS_ACTIVE_OR_RELOADING(os) &&
1237                     u->meta.manager->n_deserializing <= 0) {
1238                         /* Write audit record if we have just finished starting up */
1239                         manager_send_unit_audit(u->meta.manager, u, AUDIT_SERVICE_START, true);
1240                         u->meta.in_audit = true;
1241                 }
1242
1243                 if (!UNIT_IS_ACTIVE_OR_RELOADING(os))
1244                         manager_send_unit_plymouth(u->meta.manager, u);
1245
1246         } else {
1247
1248                 /* We don't care about D-Bus here, since we'll get an
1249                  * asynchronous notification for it anyway. */
1250
1251                 if (u->meta.type == UNIT_SERVICE &&
1252                     UNIT_IS_INACTIVE_OR_FAILED(ns) &&
1253                     !UNIT_IS_INACTIVE_OR_FAILED(os) &&
1254                     u->meta.manager->n_deserializing <= 0) {
1255
1256                         /* Hmm, if there was no start record written
1257                          * write it now, so that we always have a nice
1258                          * pair */
1259                         if (!u->meta.in_audit) {
1260                                 manager_send_unit_audit(u->meta.manager, u, AUDIT_SERVICE_START, ns == UNIT_INACTIVE);
1261
1262                                 if (ns == UNIT_INACTIVE)
1263                                         manager_send_unit_audit(u->meta.manager, u, AUDIT_SERVICE_STOP, true);
1264                         } else
1265                                 /* Write audit record if we have just finished shutting down */
1266                                 manager_send_unit_audit(u->meta.manager, u, AUDIT_SERVICE_STOP, ns == UNIT_INACTIVE);
1267
1268                         u->meta.in_audit = false;
1269                 }
1270         }
1271
1272         manager_recheck_syslog(u->meta.manager);
1273
1274         /* Maybe we finished startup and are now ready for being
1275          * stopped because unneeded? */
1276         unit_check_unneeded(u);
1277
1278         unit_add_to_dbus_queue(u);
1279         unit_add_to_gc_queue(u);
1280 }
1281
1282 int unit_watch_fd(Unit *u, int fd, uint32_t events, Watch *w) {
1283         struct epoll_event ev;
1284
1285         assert(u);
1286         assert(fd >= 0);
1287         assert(w);
1288         assert(w->type == WATCH_INVALID || (w->type == WATCH_FD && w->fd == fd && w->data.unit == u));
1289
1290         zero(ev);
1291         ev.data.ptr = w;
1292         ev.events = events;
1293
1294         if (epoll_ctl(u->meta.manager->epoll_fd,
1295                       w->type == WATCH_INVALID ? EPOLL_CTL_ADD : EPOLL_CTL_MOD,
1296                       fd,
1297                       &ev) < 0)
1298                 return -errno;
1299
1300         w->fd = fd;
1301         w->type = WATCH_FD;
1302         w->data.unit = u;
1303
1304         return 0;
1305 }
1306
1307 void unit_unwatch_fd(Unit *u, Watch *w) {
1308         assert(u);
1309         assert(w);
1310
1311         if (w->type == WATCH_INVALID)
1312                 return;
1313
1314         assert(w->type == WATCH_FD);
1315         assert(w->data.unit == u);
1316         assert_se(epoll_ctl(u->meta.manager->epoll_fd, EPOLL_CTL_DEL, w->fd, NULL) >= 0);
1317
1318         w->fd = -1;
1319         w->type = WATCH_INVALID;
1320         w->data.unit = NULL;
1321 }
1322
1323 int unit_watch_pid(Unit *u, pid_t pid) {
1324         assert(u);
1325         assert(pid >= 1);
1326
1327         /* Watch a specific PID. We only support one unit watching
1328          * each PID for now. */
1329
1330         return hashmap_put(u->meta.manager->watch_pids, LONG_TO_PTR(pid), u);
1331 }
1332
1333 void unit_unwatch_pid(Unit *u, pid_t pid) {
1334         assert(u);
1335         assert(pid >= 1);
1336
1337         hashmap_remove_value(u->meta.manager->watch_pids, LONG_TO_PTR(pid), u);
1338 }
1339
1340 int unit_watch_timer(Unit *u, usec_t delay, Watch *w) {
1341         struct itimerspec its;
1342         int flags, fd;
1343         bool ours;
1344
1345         assert(u);
1346         assert(w);
1347         assert(w->type == WATCH_INVALID || (w->type == WATCH_UNIT_TIMER && w->data.unit == u));
1348
1349         /* This will try to reuse the old timer if there is one */
1350
1351         if (w->type == WATCH_UNIT_TIMER) {
1352                 assert(w->data.unit == u);
1353                 assert(w->fd >= 0);
1354
1355                 ours = false;
1356                 fd = w->fd;
1357         } else if (w->type == WATCH_INVALID) {
1358
1359                 ours = true;
1360                 if ((fd = timerfd_create(CLOCK_MONOTONIC, TFD_NONBLOCK|TFD_CLOEXEC)) < 0)
1361                         return -errno;
1362         } else
1363                 assert_not_reached("Invalid watch type");
1364
1365         zero(its);
1366
1367         if (delay <= 0) {
1368                 /* Set absolute time in the past, but not 0, since we
1369                  * don't want to disarm the timer */
1370                 its.it_value.tv_sec = 0;
1371                 its.it_value.tv_nsec = 1;
1372
1373                 flags = TFD_TIMER_ABSTIME;
1374         } else {
1375                 timespec_store(&its.it_value, delay);
1376                 flags = 0;
1377         }
1378
1379         /* This will also flush the elapse counter */
1380         if (timerfd_settime(fd, flags, &its, NULL) < 0)
1381                 goto fail;
1382
1383         if (w->type == WATCH_INVALID) {
1384                 struct epoll_event ev;
1385
1386                 zero(ev);
1387                 ev.data.ptr = w;
1388                 ev.events = EPOLLIN;
1389
1390                 if (epoll_ctl(u->meta.manager->epoll_fd, EPOLL_CTL_ADD, fd, &ev) < 0)
1391                         goto fail;
1392         }
1393
1394         w->type = WATCH_UNIT_TIMER;
1395         w->fd = fd;
1396         w->data.unit = u;
1397
1398         return 0;
1399
1400 fail:
1401         if (ours)
1402                 close_nointr_nofail(fd);
1403
1404         return -errno;
1405 }
1406
1407 void unit_unwatch_timer(Unit *u, Watch *w) {
1408         assert(u);
1409         assert(w);
1410
1411         if (w->type == WATCH_INVALID)
1412                 return;
1413
1414         assert(w->type == WATCH_UNIT_TIMER);
1415         assert(w->data.unit == u);
1416         assert(w->fd >= 0);
1417
1418         assert_se(epoll_ctl(u->meta.manager->epoll_fd, EPOLL_CTL_DEL, w->fd, NULL) >= 0);
1419         close_nointr_nofail(w->fd);
1420
1421         w->fd = -1;
1422         w->type = WATCH_INVALID;
1423         w->data.unit = NULL;
1424 }
1425
1426 bool unit_job_is_applicable(Unit *u, JobType j) {
1427         assert(u);
1428         assert(j >= 0 && j < _JOB_TYPE_MAX);
1429
1430         switch (j) {
1431
1432         case JOB_VERIFY_ACTIVE:
1433         case JOB_START:
1434         case JOB_STOP:
1435                 return true;
1436
1437         case JOB_RESTART:
1438         case JOB_TRY_RESTART:
1439                 return unit_can_start(u);
1440
1441         case JOB_RELOAD:
1442                 return unit_can_reload(u);
1443
1444         case JOB_RELOAD_OR_START:
1445                 return unit_can_reload(u) && unit_can_start(u);
1446
1447         default:
1448                 assert_not_reached("Invalid job type");
1449         }
1450 }
1451
1452 int unit_add_dependency(Unit *u, UnitDependency d, Unit *other, bool add_reference) {
1453
1454         static const UnitDependency inverse_table[_UNIT_DEPENDENCY_MAX] = {
1455                 [UNIT_REQUIRES] = UNIT_REQUIRED_BY,
1456                 [UNIT_REQUIRES_OVERRIDABLE] = UNIT_REQUIRED_BY_OVERRIDABLE,
1457                 [UNIT_WANTS] = UNIT_WANTED_BY,
1458                 [UNIT_REQUISITE] = UNIT_REQUIRED_BY,
1459                 [UNIT_REQUISITE_OVERRIDABLE] = UNIT_REQUIRED_BY_OVERRIDABLE,
1460                 [UNIT_BIND_TO] = UNIT_BOUND_BY,
1461                 [UNIT_REQUIRED_BY] = _UNIT_DEPENDENCY_INVALID,
1462                 [UNIT_REQUIRED_BY_OVERRIDABLE] = _UNIT_DEPENDENCY_INVALID,
1463                 [UNIT_WANTED_BY] = _UNIT_DEPENDENCY_INVALID,
1464                 [UNIT_BOUND_BY] = UNIT_BIND_TO,
1465                 [UNIT_CONFLICTS] = UNIT_CONFLICTED_BY,
1466                 [UNIT_CONFLICTED_BY] = UNIT_CONFLICTS,
1467                 [UNIT_BEFORE] = UNIT_AFTER,
1468                 [UNIT_AFTER] = UNIT_BEFORE,
1469                 [UNIT_ON_FAILURE] = _UNIT_DEPENDENCY_INVALID,
1470                 [UNIT_REFERENCES] = UNIT_REFERENCED_BY,
1471                 [UNIT_REFERENCED_BY] = UNIT_REFERENCES
1472         };
1473         int r, q = 0, v = 0, w = 0;
1474
1475         assert(u);
1476         assert(d >= 0 && d < _UNIT_DEPENDENCY_MAX);
1477         assert(other);
1478
1479         u = unit_follow_merge(u);
1480         other = unit_follow_merge(other);
1481
1482         /* We won't allow dependencies on ourselves. We will not
1483          * consider them an error however. */
1484         if (u == other)
1485                 return 0;
1486
1487         if ((r = set_ensure_allocated(&u->meta.dependencies[d], trivial_hash_func, trivial_compare_func)) < 0)
1488                 return r;
1489
1490         if (inverse_table[d] != _UNIT_DEPENDENCY_INVALID)
1491                 if ((r = set_ensure_allocated(&other->meta.dependencies[inverse_table[d]], trivial_hash_func, trivial_compare_func)) < 0)
1492                         return r;
1493
1494         if (add_reference)
1495                 if ((r = set_ensure_allocated(&u->meta.dependencies[UNIT_REFERENCES], trivial_hash_func, trivial_compare_func)) < 0 ||
1496                     (r = set_ensure_allocated(&other->meta.dependencies[UNIT_REFERENCED_BY], trivial_hash_func, trivial_compare_func)) < 0)
1497                         return r;
1498
1499         if ((q = set_put(u->meta.dependencies[d], other)) < 0)
1500                 return q;
1501
1502         if (inverse_table[d] != _UNIT_DEPENDENCY_INVALID)
1503                 if ((v = set_put(other->meta.dependencies[inverse_table[d]], u)) < 0) {
1504                         r = v;
1505                         goto fail;
1506                 }
1507
1508         if (add_reference) {
1509                 if ((w = set_put(u->meta.dependencies[UNIT_REFERENCES], other)) < 0) {
1510                         r = w;
1511                         goto fail;
1512                 }
1513
1514                 if ((r = set_put(other->meta.dependencies[UNIT_REFERENCED_BY], u)) < 0)
1515                         goto fail;
1516         }
1517
1518         unit_add_to_dbus_queue(u);
1519         return 0;
1520
1521 fail:
1522         if (q > 0)
1523                 set_remove(u->meta.dependencies[d], other);
1524
1525         if (v > 0)
1526                 set_remove(other->meta.dependencies[inverse_table[d]], u);
1527
1528         if (w > 0)
1529                 set_remove(u->meta.dependencies[UNIT_REFERENCES], other);
1530
1531         return r;
1532 }
1533
1534 int unit_add_two_dependencies(Unit *u, UnitDependency d, UnitDependency e, Unit *other, bool add_reference) {
1535         int r;
1536
1537         assert(u);
1538
1539         if ((r = unit_add_dependency(u, d, other, add_reference)) < 0)
1540                 return r;
1541
1542         if ((r = unit_add_dependency(u, e, other, add_reference)) < 0)
1543                 return r;
1544
1545         return 0;
1546 }
1547
1548 static const char *resolve_template(Unit *u, const char *name, const char*path, char **p) {
1549         char *s;
1550
1551         assert(u);
1552         assert(name || path);
1553
1554         if (!name)
1555                 name = file_name_from_path(path);
1556
1557         if (!unit_name_is_template(name)) {
1558                 *p = NULL;
1559                 return name;
1560         }
1561
1562         if (u->meta.instance)
1563                 s = unit_name_replace_instance(name, u->meta.instance);
1564         else {
1565                 char *i;
1566
1567                 if (!(i = unit_name_to_prefix(u->meta.id)))
1568                         return NULL;
1569
1570                 s = unit_name_replace_instance(name, i);
1571                 free(i);
1572         }
1573
1574         if (!s)
1575                 return NULL;
1576
1577         *p = s;
1578         return s;
1579 }
1580
1581 int unit_add_dependency_by_name(Unit *u, UnitDependency d, const char *name, const char *path, bool add_reference) {
1582         Unit *other;
1583         int r;
1584         char *s;
1585
1586         assert(u);
1587         assert(name || path);
1588
1589         if (!(name = resolve_template(u, name, path, &s)))
1590                 return -ENOMEM;
1591
1592         if ((r = manager_load_unit(u->meta.manager, name, path, NULL, &other)) < 0)
1593                 goto finish;
1594
1595         r = unit_add_dependency(u, d, other, add_reference);
1596
1597 finish:
1598         free(s);
1599         return r;
1600 }
1601
1602 int unit_add_two_dependencies_by_name(Unit *u, UnitDependency d, UnitDependency e, const char *name, const char *path, bool add_reference) {
1603         Unit *other;
1604         int r;
1605         char *s;
1606
1607         assert(u);
1608         assert(name || path);
1609
1610         if (!(name = resolve_template(u, name, path, &s)))
1611                 return -ENOMEM;
1612
1613         if ((r = manager_load_unit(u->meta.manager, name, path, NULL, &other)) < 0)
1614                 goto finish;
1615
1616         r = unit_add_two_dependencies(u, d, e, other, add_reference);
1617
1618 finish:
1619         free(s);
1620         return r;
1621 }
1622
1623 int unit_add_dependency_by_name_inverse(Unit *u, UnitDependency d, const char *name, const char *path, bool add_reference) {
1624         Unit *other;
1625         int r;
1626         char *s;
1627
1628         assert(u);
1629         assert(name || path);
1630
1631         if (!(name = resolve_template(u, name, path, &s)))
1632                 return -ENOMEM;
1633
1634         if ((r = manager_load_unit(u->meta.manager, name, path, NULL, &other)) < 0)
1635                 goto finish;
1636
1637         r = unit_add_dependency(other, d, u, add_reference);
1638
1639 finish:
1640         free(s);
1641         return r;
1642 }
1643
1644 int unit_add_two_dependencies_by_name_inverse(Unit *u, UnitDependency d, UnitDependency e, const char *name, const char *path, bool add_reference) {
1645         Unit *other;
1646         int r;
1647         char *s;
1648
1649         assert(u);
1650         assert(name || path);
1651
1652         if (!(name = resolve_template(u, name, path, &s)))
1653                 return -ENOMEM;
1654
1655         if ((r = manager_load_unit(u->meta.manager, name, path, NULL, &other)) < 0)
1656                 goto finish;
1657
1658         if ((r = unit_add_two_dependencies(other, d, e, u, add_reference)) < 0)
1659                 goto finish;
1660
1661 finish:
1662         free(s);
1663         return r;
1664 }
1665
1666 int set_unit_path(const char *p) {
1667         char *cwd, *c;
1668         int r;
1669
1670         /* This is mostly for debug purposes */
1671
1672         if (path_is_absolute(p)) {
1673                 if (!(c = strdup(p)))
1674                         return -ENOMEM;
1675         } else {
1676                 if (!(cwd = get_current_dir_name()))
1677                         return -errno;
1678
1679                 r = asprintf(&c, "%s/%s", cwd, p);
1680                 free(cwd);
1681
1682                 if (r < 0)
1683                         return -ENOMEM;
1684         }
1685
1686         if (setenv("SYSTEMD_UNIT_PATH", c, 0) < 0) {
1687                 r = -errno;
1688                 free(c);
1689                 return r;
1690         }
1691
1692         return 0;
1693 }
1694
1695 char *unit_dbus_path(Unit *u) {
1696         char *p, *e;
1697
1698         assert(u);
1699
1700         if (!u->meta.id)
1701                 return NULL;
1702
1703         if (!(e = bus_path_escape(u->meta.id)))
1704                 return NULL;
1705
1706         p = strappend("/org/freedesktop/systemd1/unit/", e);
1707         free(e);
1708
1709         return p;
1710 }
1711
1712 int unit_add_cgroup(Unit *u, CGroupBonding *b) {
1713         int r;
1714
1715         assert(u);
1716         assert(b);
1717
1718         assert(b->path);
1719
1720         if (!b->controller)
1721                 if (!(b->controller = strdup(SYSTEMD_CGROUP_CONTROLLER)))
1722                         return -ENOMEM;
1723
1724         /* Ensure this hasn't been added yet */
1725         assert(!b->unit);
1726
1727         if (streq(b->controller, SYSTEMD_CGROUP_CONTROLLER)) {
1728                 CGroupBonding *l;
1729
1730                 l = hashmap_get(u->meta.manager->cgroup_bondings, b->path);
1731                 LIST_PREPEND(CGroupBonding, by_path, l, b);
1732
1733                 if ((r = hashmap_replace(u->meta.manager->cgroup_bondings, b->path, l)) < 0) {
1734                         LIST_REMOVE(CGroupBonding, by_path, l, b);
1735                         return r;
1736                 }
1737         }
1738
1739         LIST_PREPEND(CGroupBonding, by_unit, u->meta.cgroup_bondings, b);
1740         b->unit = u;
1741
1742         return 0;
1743 }
1744
1745 static char *default_cgroup_path(Unit *u) {
1746         char *p;
1747         int r;
1748
1749         assert(u);
1750
1751         if (u->meta.instance) {
1752                 char *t;
1753
1754                 if (!(t = unit_name_template(u->meta.id)))
1755                         return NULL;
1756
1757                 r = asprintf(&p, "%s/%s/%s", u->meta.manager->cgroup_hierarchy, t, u->meta.instance);
1758                 free(t);
1759         } else
1760                 r = asprintf(&p, "%s/%s", u->meta.manager->cgroup_hierarchy, u->meta.id);
1761
1762         return r < 0 ? NULL : p;
1763 }
1764
1765 int unit_add_cgroup_from_text(Unit *u, const char *name) {
1766         char *controller = NULL, *path = NULL;
1767         CGroupBonding *b = NULL;
1768         int r;
1769
1770         assert(u);
1771         assert(name);
1772
1773         if ((r = cg_split_spec(name, &controller, &path)) < 0)
1774                 return r;
1775
1776         if (!path)
1777                 path = default_cgroup_path(u);
1778
1779         if (!controller)
1780                 controller = strdup(SYSTEMD_CGROUP_CONTROLLER);
1781
1782         if (!path || !controller) {
1783                 free(path);
1784                 free(controller);
1785
1786                 return -ENOMEM;
1787         }
1788
1789         if (cgroup_bonding_find_list(u->meta.cgroup_bondings, controller)) {
1790                 r = -EEXIST;
1791                 goto fail;
1792         }
1793
1794         if (!(b = new0(CGroupBonding, 1))) {
1795                 r = -ENOMEM;
1796                 goto fail;
1797         }
1798
1799         b->controller = controller;
1800         b->path = path;
1801         b->ours = false;
1802
1803         if ((r = unit_add_cgroup(u, b)) < 0)
1804                 goto fail;
1805
1806         return 0;
1807
1808 fail:
1809         free(path);
1810         free(controller);
1811         free(b);
1812
1813         return r;
1814 }
1815
1816 static int unit_add_one_default_cgroup(Unit *u, const char *controller) {
1817         CGroupBonding *b = NULL;
1818         int r = -ENOMEM;
1819
1820         assert(u);
1821
1822         if (!controller)
1823                 controller = SYSTEMD_CGROUP_CONTROLLER;
1824
1825         if (cgroup_bonding_find_list(u->meta.cgroup_bondings, controller))
1826                 return 0;
1827
1828         if (!(b = new0(CGroupBonding, 1)))
1829                 return -ENOMEM;
1830
1831         if (!(b->controller = strdup(controller)))
1832                 goto fail;
1833
1834         if (!(b->path = default_cgroup_path(u)))
1835                 goto fail;
1836
1837         b->ours = true;
1838         b->essential = streq(controller, SYSTEMD_CGROUP_CONTROLLER);
1839
1840         if ((r = unit_add_cgroup(u, b)) < 0)
1841                 goto fail;
1842
1843         return 0;
1844
1845 fail:
1846         free(b->path);
1847         free(b->controller);
1848         free(b);
1849
1850         return r;
1851 }
1852
1853 int unit_add_default_cgroups(Unit *u) {
1854         char **c;
1855         int r;
1856         assert(u);
1857
1858         /* Adds in the default cgroups, if they weren't specified
1859          * otherwise. */
1860
1861         if ((r = unit_add_one_default_cgroup(u, NULL)) < 0)
1862                 return r;
1863
1864         STRV_FOREACH(c, u->meta.manager->default_controllers)
1865                 if ((r = unit_add_one_default_cgroup(u, *c)) < 0)
1866                         return r;
1867
1868         return 0;
1869 }
1870
1871 CGroupBonding* unit_get_default_cgroup(Unit *u) {
1872         assert(u);
1873
1874         return cgroup_bonding_find_list(u->meta.cgroup_bondings, SYSTEMD_CGROUP_CONTROLLER);
1875 }
1876
1877 int unit_load_related_unit(Unit *u, const char *type, Unit **_found) {
1878         char *t;
1879         int r;
1880
1881         assert(u);
1882         assert(type);
1883         assert(_found);
1884
1885         if (!(t = unit_name_change_suffix(u->meta.id, type)))
1886                 return -ENOMEM;
1887
1888         assert(!unit_has_name(u, t));
1889
1890         r = manager_load_unit(u->meta.manager, t, NULL, NULL, _found);
1891         free(t);
1892
1893         assert(r < 0 || *_found != u);
1894
1895         return r;
1896 }
1897
1898 int unit_get_related_unit(Unit *u, const char *type, Unit **_found) {
1899         Unit *found;
1900         char *t;
1901
1902         assert(u);
1903         assert(type);
1904         assert(_found);
1905
1906         if (!(t = unit_name_change_suffix(u->meta.id, type)))
1907                 return -ENOMEM;
1908
1909         assert(!unit_has_name(u, t));
1910
1911         found = manager_get_unit(u->meta.manager, t);
1912         free(t);
1913
1914         if (!found)
1915                 return -ENOENT;
1916
1917         *_found = found;
1918         return 0;
1919 }
1920
1921 static char *specifier_prefix_and_instance(char specifier, void *data, void *userdata) {
1922         Unit *u = userdata;
1923         assert(u);
1924
1925         return unit_name_to_prefix_and_instance(u->meta.id);
1926 }
1927
1928 static char *specifier_prefix(char specifier, void *data, void *userdata) {
1929         Unit *u = userdata;
1930         assert(u);
1931
1932         return unit_name_to_prefix(u->meta.id);
1933 }
1934
1935 static char *specifier_prefix_unescaped(char specifier, void *data, void *userdata) {
1936         Unit *u = userdata;
1937         char *p, *r;
1938
1939         assert(u);
1940
1941         if (!(p = unit_name_to_prefix(u->meta.id)))
1942                 return NULL;
1943
1944         r = unit_name_unescape(p);
1945         free(p);
1946
1947         return r;
1948 }
1949
1950 static char *specifier_instance_unescaped(char specifier, void *data, void *userdata) {
1951         Unit *u = userdata;
1952         assert(u);
1953
1954         if (u->meta.instance)
1955                 return unit_name_unescape(u->meta.instance);
1956
1957         return strdup("");
1958 }
1959
1960 static char *specifier_filename(char specifier, void *data, void *userdata) {
1961         Unit *u = userdata;
1962         assert(u);
1963
1964         if (u->meta.instance)
1965                 return unit_name_path_unescape(u->meta.instance);
1966
1967         return unit_name_to_path(u->meta.instance);
1968 }
1969
1970 char *unit_name_printf(Unit *u, const char* format) {
1971
1972         /*
1973          * This will use the passed string as format string and
1974          * replace the following specifiers:
1975          *
1976          * %n: the full id of the unit                 (foo@bar.waldo)
1977          * %N: the id of the unit without the suffix   (foo@bar)
1978          * %p: the prefix                              (foo)
1979          * %i: the instance                            (bar)
1980          */
1981
1982         const Specifier table[] = {
1983                 { 'n', specifier_string,              u->meta.id },
1984                 { 'N', specifier_prefix_and_instance, NULL },
1985                 { 'p', specifier_prefix,              NULL },
1986                 { 'i', specifier_string,              u->meta.instance },
1987                 { 0, NULL, NULL }
1988         };
1989
1990         assert(u);
1991         assert(format);
1992
1993         return specifier_printf(format, table, u);
1994 }
1995
1996 char *unit_full_printf(Unit *u, const char *format) {
1997
1998         /* This is similar to unit_name_printf() but also supports
1999          * unescaping */
2000
2001         const Specifier table[] = {
2002                 { 'n', specifier_string,              u->meta.id },
2003                 { 'N', specifier_prefix_and_instance, NULL },
2004                 { 'p', specifier_prefix,              NULL },
2005                 { 'P', specifier_prefix_unescaped,    NULL },
2006                 { 'i', specifier_string,              u->meta.instance },
2007                 { 'I', specifier_instance_unescaped,  NULL },
2008                 { 'f', specifier_filename,            NULL },
2009                 { 0, NULL, NULL }
2010         };
2011
2012         assert(u);
2013         assert(format);
2014
2015         return specifier_printf(format, table, u);
2016 }
2017
2018 char **unit_full_printf_strv(Unit *u, char **l) {
2019         size_t n;
2020         char **r, **i, **j;
2021
2022         /* Applies unit_full_printf to every entry in l */
2023
2024         assert(u);
2025
2026         n = strv_length(l);
2027         if (!(r = new(char*, n+1)))
2028                 return NULL;
2029
2030         for (i = l, j = r; *i; i++, j++)
2031                 if (!(*j = unit_full_printf(u, *i)))
2032                         goto fail;
2033
2034         *j = NULL;
2035         return r;
2036
2037 fail:
2038         for (j--; j >= r; j--)
2039                 free(*j);
2040
2041         free(r);
2042
2043         return NULL;
2044 }
2045
2046 int unit_watch_bus_name(Unit *u, const char *name) {
2047         assert(u);
2048         assert(name);
2049
2050         /* Watch a specific name on the bus. We only support one unit
2051          * watching each name for now. */
2052
2053         return hashmap_put(u->meta.manager->watch_bus, name, u);
2054 }
2055
2056 void unit_unwatch_bus_name(Unit *u, const char *name) {
2057         assert(u);
2058         assert(name);
2059
2060         hashmap_remove_value(u->meta.manager->watch_bus, name, u);
2061 }
2062
2063 bool unit_can_serialize(Unit *u) {
2064         assert(u);
2065
2066         return UNIT_VTABLE(u)->serialize && UNIT_VTABLE(u)->deserialize_item;
2067 }
2068
2069 int unit_serialize(Unit *u, FILE *f, FDSet *fds) {
2070         int r;
2071
2072         assert(u);
2073         assert(f);
2074         assert(fds);
2075
2076         if (!unit_can_serialize(u))
2077                 return 0;
2078
2079         if ((r = UNIT_VTABLE(u)->serialize(u, f, fds)) < 0)
2080                 return r;
2081
2082         if (u->meta.job)
2083                 unit_serialize_item(u, f, "job", job_type_to_string(u->meta.job->type));
2084
2085         dual_timestamp_serialize(f, "inactive-exit-timestamp", &u->meta.inactive_exit_timestamp);
2086         dual_timestamp_serialize(f, "active-enter-timestamp", &u->meta.active_enter_timestamp);
2087         dual_timestamp_serialize(f, "active-exit-timestamp", &u->meta.active_exit_timestamp);
2088         dual_timestamp_serialize(f, "inactive-enter-timestamp", &u->meta.inactive_enter_timestamp);
2089         dual_timestamp_serialize(f, "condition-timestamp", &u->meta.condition_timestamp);
2090
2091         if (dual_timestamp_is_set(&u->meta.condition_timestamp))
2092                 unit_serialize_item(u, f, "condition-result", yes_no(u->meta.condition_result));
2093
2094         /* End marker */
2095         fputc('\n', f);
2096         return 0;
2097 }
2098
2099 void unit_serialize_item_format(Unit *u, FILE *f, const char *key, const char *format, ...) {
2100         va_list ap;
2101
2102         assert(u);
2103         assert(f);
2104         assert(key);
2105         assert(format);
2106
2107         fputs(key, f);
2108         fputc('=', f);
2109
2110         va_start(ap, format);
2111         vfprintf(f, format, ap);
2112         va_end(ap);
2113
2114         fputc('\n', f);
2115 }
2116
2117 void unit_serialize_item(Unit *u, FILE *f, const char *key, const char *value) {
2118         assert(u);
2119         assert(f);
2120         assert(key);
2121         assert(value);
2122
2123         fprintf(f, "%s=%s\n", key, value);
2124 }
2125
2126 int unit_deserialize(Unit *u, FILE *f, FDSet *fds) {
2127         int r;
2128
2129         assert(u);
2130         assert(f);
2131         assert(fds);
2132
2133         if (!unit_can_serialize(u))
2134                 return 0;
2135
2136         for (;;) {
2137                 char line[1024], *l, *v;
2138                 size_t k;
2139
2140                 if (!fgets(line, sizeof(line), f)) {
2141                         if (feof(f))
2142                                 return 0;
2143                         return -errno;
2144                 }
2145
2146                 char_array_0(line);
2147                 l = strstrip(line);
2148
2149                 /* End marker */
2150                 if (l[0] == 0)
2151                         return 0;
2152
2153                 k = strcspn(l, "=");
2154
2155                 if (l[k] == '=') {
2156                         l[k] = 0;
2157                         v = l+k+1;
2158                 } else
2159                         v = l+k;
2160
2161                 if (streq(l, "job")) {
2162                         JobType type;
2163
2164                         if ((type = job_type_from_string(v)) < 0)
2165                                 log_debug("Failed to parse job type value %s", v);
2166                         else
2167                                 u->meta.deserialized_job = type;
2168
2169                         continue;
2170                 } else if (streq(l, "inactive-exit-timestamp")) {
2171                         dual_timestamp_deserialize(v, &u->meta.inactive_exit_timestamp);
2172                         continue;
2173                 } else if (streq(l, "active-enter-timestamp")) {
2174                         dual_timestamp_deserialize(v, &u->meta.active_enter_timestamp);
2175                         continue;
2176                 } else if (streq(l, "active-exit-timestamp")) {
2177                         dual_timestamp_deserialize(v, &u->meta.active_exit_timestamp);
2178                         continue;
2179                 } else if (streq(l, "inactive-enter-timestamp")) {
2180                         dual_timestamp_deserialize(v, &u->meta.inactive_enter_timestamp);
2181                         continue;
2182                 } else if (streq(l, "condition-timestamp")) {
2183                         dual_timestamp_deserialize(v, &u->meta.condition_timestamp);
2184                         continue;
2185                 } else if (streq(l, "condition-result")) {
2186                         int b;
2187
2188                         if ((b = parse_boolean(v)) < 0)
2189                                 log_debug("Failed to parse condition result value %s", v);
2190                         else
2191                                 u->meta.condition_result = b;
2192
2193                         continue;
2194                 }
2195
2196                 if ((r = UNIT_VTABLE(u)->deserialize_item(u, l, v, fds)) < 0)
2197                         return r;
2198         }
2199 }
2200
2201 int unit_add_node_link(Unit *u, const char *what, bool wants) {
2202         Unit *device;
2203         char *e;
2204         int r;
2205
2206         assert(u);
2207
2208         if (!what)
2209                 return 0;
2210
2211         /* Adds in links to the device node that this unit is based on */
2212
2213         if (!is_device_path(what))
2214                 return 0;
2215
2216         if (!(e = unit_name_build_escape(what+1, NULL, ".device")))
2217                 return -ENOMEM;
2218
2219         r = manager_load_unit(u->meta.manager, e, NULL, NULL, &device);
2220         free(e);
2221
2222         if (r < 0)
2223                 return r;
2224
2225         if ((r = unit_add_two_dependencies(u, UNIT_AFTER, UNIT_BIND_TO, device, true)) < 0)
2226                 return r;
2227
2228         if (wants)
2229                 if ((r = unit_add_dependency(device, UNIT_WANTS, u, false)) < 0)
2230                         return r;
2231
2232         return 0;
2233 }
2234
2235 int unit_coldplug(Unit *u) {
2236         int r;
2237
2238         assert(u);
2239
2240         if (UNIT_VTABLE(u)->coldplug)
2241                 if ((r = UNIT_VTABLE(u)->coldplug(u)) < 0)
2242                         return r;
2243
2244         if (u->meta.deserialized_job >= 0) {
2245                 if ((r = manager_add_job(u->meta.manager, u->meta.deserialized_job, u, JOB_IGNORE_DEPENDENCIES, false, NULL, NULL)) < 0)
2246                         return r;
2247
2248                 u->meta.deserialized_job = _JOB_TYPE_INVALID;
2249         }
2250
2251         return 0;
2252 }
2253
2254 void unit_status_printf(Unit *u, const char *format, ...) {
2255         va_list ap;
2256
2257         assert(u);
2258         assert(format);
2259
2260         if (!UNIT_VTABLE(u)->show_status)
2261                 return;
2262
2263         if (u->meta.manager->running_as != MANAGER_SYSTEM)
2264                 return;
2265
2266         /* If Plymouth is running make sure we show the status, so
2267          * that there's something nice to see when people press Esc */
2268
2269         if (!u->meta.manager->show_status && !plymouth_running())
2270                 return;
2271
2272         if (!manager_is_booting_or_shutting_down(u->meta.manager))
2273                 return;
2274
2275         va_start(ap, format);
2276         status_vprintf(format, ap);
2277         va_end(ap);
2278 }
2279
2280 bool unit_need_daemon_reload(Unit *u) {
2281         struct stat st;
2282
2283         assert(u);
2284
2285         if (!u->meta.fragment_path)
2286                 return false;
2287
2288         zero(st);
2289         if (stat(u->meta.fragment_path, &st) < 0)
2290                 /* What, cannot access this anymore? */
2291                 return true;
2292
2293         return
2294                 u->meta.fragment_mtime &&
2295                 timespec_load(&st.st_mtim) != u->meta.fragment_mtime;
2296 }
2297
2298 void unit_reset_failed(Unit *u) {
2299         assert(u);
2300
2301         if (UNIT_VTABLE(u)->reset_failed)
2302                 UNIT_VTABLE(u)->reset_failed(u);
2303 }
2304
2305 Unit *unit_following(Unit *u) {
2306         assert(u);
2307
2308         if (UNIT_VTABLE(u)->following)
2309                 return UNIT_VTABLE(u)->following(u);
2310
2311         return NULL;
2312 }
2313
2314 bool unit_pending_inactive(Unit *u) {
2315         assert(u);
2316
2317         /* Returns true if the unit is inactive or going down */
2318
2319         if (UNIT_IS_INACTIVE_OR_DEACTIVATING(unit_active_state(u)))
2320                 return true;
2321
2322         if (u->meta.job && u->meta.job->type == JOB_STOP)
2323                 return true;
2324
2325         return false;
2326 }
2327
2328 bool unit_pending_active(Unit *u) {
2329         assert(u);
2330
2331         /* Returns true if the unit is inactive or going down */
2332
2333         if (UNIT_IS_ACTIVE_OR_ACTIVATING(unit_active_state(u)))
2334                 return true;
2335
2336         if (u->meta.job &&
2337             (u->meta.job->type == JOB_START ||
2338              u->meta.job->type == JOB_RELOAD_OR_START ||
2339              u->meta.job->type == JOB_RESTART))
2340                 return true;
2341
2342         return false;
2343 }
2344
2345 UnitType unit_name_to_type(const char *n) {
2346         UnitType t;
2347
2348         assert(n);
2349
2350         for (t = 0; t < _UNIT_TYPE_MAX; t++)
2351                 if (endswith(n, unit_vtable[t]->suffix))
2352                         return t;
2353
2354         return _UNIT_TYPE_INVALID;
2355 }
2356
2357 bool unit_name_is_valid(const char *n, bool template_ok) {
2358         UnitType t;
2359
2360         t = unit_name_to_type(n);
2361         if (t < 0 || t >= _UNIT_TYPE_MAX)
2362                 return false;
2363
2364         return unit_name_is_valid_no_type(n, template_ok);
2365 }
2366
2367 int unit_kill(Unit *u, KillWho w, KillMode m, int signo, DBusError *error) {
2368         assert(u);
2369         assert(w >= 0 && w < _KILL_WHO_MAX);
2370         assert(m >= 0 && m < _KILL_MODE_MAX);
2371         assert(signo > 0);
2372         assert(signo < _NSIG);
2373
2374         if (m == KILL_NONE)
2375                 return 0;
2376
2377         if (!UNIT_VTABLE(u)->kill)
2378                 return -ENOTSUP;
2379
2380         return UNIT_VTABLE(u)->kill(u, w, m, signo, error);
2381 }
2382
2383
2384 int unit_following_set(Unit *u, Set **s) {
2385         assert(u);
2386         assert(s);
2387
2388         if (UNIT_VTABLE(u)->following_set)
2389                 return UNIT_VTABLE(u)->following_set(u, s);
2390
2391         *s = NULL;
2392         return 0;
2393 }
2394
2395 static const char* const unit_load_state_table[_UNIT_LOAD_STATE_MAX] = {
2396         [UNIT_STUB] = "stub",
2397         [UNIT_LOADED] = "loaded",
2398         [UNIT_ERROR] = "error",
2399         [UNIT_MERGED] = "merged",
2400         [UNIT_MASKED] = "masked"
2401 };
2402
2403 DEFINE_STRING_TABLE_LOOKUP(unit_load_state, UnitLoadState);
2404
2405 static const char* const unit_active_state_table[_UNIT_ACTIVE_STATE_MAX] = {
2406         [UNIT_ACTIVE] = "active",
2407         [UNIT_RELOADING] = "reloading",
2408         [UNIT_INACTIVE] = "inactive",
2409         [UNIT_FAILED] = "failed",
2410         [UNIT_ACTIVATING] = "activating",
2411         [UNIT_DEACTIVATING] = "deactivating"
2412 };
2413
2414 DEFINE_STRING_TABLE_LOOKUP(unit_active_state, UnitActiveState);
2415
2416 static const char* const unit_dependency_table[_UNIT_DEPENDENCY_MAX] = {
2417         [UNIT_REQUIRES] = "Requires",
2418         [UNIT_REQUIRES_OVERRIDABLE] = "RequiresOverridable",
2419         [UNIT_WANTS] = "Wants",
2420         [UNIT_REQUISITE] = "Requisite",
2421         [UNIT_REQUISITE_OVERRIDABLE] = "RequisiteOverridable",
2422         [UNIT_REQUIRED_BY] = "RequiredBy",
2423         [UNIT_REQUIRED_BY_OVERRIDABLE] = "RequiredByOverridable",
2424         [UNIT_BIND_TO] = "BindTo",
2425         [UNIT_WANTED_BY] = "WantedBy",
2426         [UNIT_CONFLICTS] = "Conflicts",
2427         [UNIT_CONFLICTED_BY] = "ConflictedBy",
2428         [UNIT_BOUND_BY] = "BoundBy",
2429         [UNIT_BEFORE] = "Before",
2430         [UNIT_AFTER] = "After",
2431         [UNIT_REFERENCES] = "References",
2432         [UNIT_REFERENCED_BY] = "ReferencedBy",
2433         [UNIT_ON_FAILURE] = "OnFailure"
2434 };
2435
2436 DEFINE_STRING_TABLE_LOOKUP(unit_dependency, UnitDependency);