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                                 /* Write audit record if we have just finished starting up */
1238                                 manager_send_unit_audit(u->meta.manager, u, AUDIT_SERVICE_START, true);
1239                                 u->meta.in_audit = true;
1240                         }
1241
1242                         if (!UNIT_IS_ACTIVE_OR_RELOADING(os))
1243                                 manager_send_unit_plymouth(u->meta.manager, u);
1244
1245                 } else {
1246
1247                         /* We don't care about D-Bus here, since we'll get an
1248                          * asynchronous notification for it anyway. */
1249
1250                         if (u->meta.type == UNIT_SERVICE &&
1251                             UNIT_IS_INACTIVE_OR_FAILED(ns) &&
1252                             !UNIT_IS_INACTIVE_OR_FAILED(os)) {
1253
1254                                 /* Hmm, if there was no start record written
1255                                  * write it now, so that we always have a nice
1256                                  * pair */
1257                                 if (!u->meta.in_audit) {
1258                                         manager_send_unit_audit(u->meta.manager, u, AUDIT_SERVICE_START, ns == UNIT_INACTIVE);
1259
1260                                         if (ns == UNIT_INACTIVE)
1261                                                 manager_send_unit_audit(u->meta.manager, u, AUDIT_SERVICE_STOP, true);
1262                                 } else
1263                                         /* Write audit record if we have just finished shutting down */
1264                                         manager_send_unit_audit(u->meta.manager, u, AUDIT_SERVICE_STOP, ns == UNIT_INACTIVE);
1265
1266                                 u->meta.in_audit = false;
1267                         }
1268                 }
1269
1270                 manager_recheck_syslog(u->meta.manager);
1271         }
1272
1273         /* Maybe we finished startup and are now ready for being
1274          * stopped because unneeded? */
1275         unit_check_unneeded(u);
1276
1277         unit_add_to_dbus_queue(u);
1278         unit_add_to_gc_queue(u);
1279 }
1280
1281 int unit_watch_fd(Unit *u, int fd, uint32_t events, Watch *w) {
1282         struct epoll_event ev;
1283
1284         assert(u);
1285         assert(fd >= 0);
1286         assert(w);
1287         assert(w->type == WATCH_INVALID || (w->type == WATCH_FD && w->fd == fd && w->data.unit == u));
1288
1289         zero(ev);
1290         ev.data.ptr = w;
1291         ev.events = events;
1292
1293         if (epoll_ctl(u->meta.manager->epoll_fd,
1294                       w->type == WATCH_INVALID ? EPOLL_CTL_ADD : EPOLL_CTL_MOD,
1295                       fd,
1296                       &ev) < 0)
1297                 return -errno;
1298
1299         w->fd = fd;
1300         w->type = WATCH_FD;
1301         w->data.unit = u;
1302
1303         return 0;
1304 }
1305
1306 void unit_unwatch_fd(Unit *u, Watch *w) {
1307         assert(u);
1308         assert(w);
1309
1310         if (w->type == WATCH_INVALID)
1311                 return;
1312
1313         assert(w->type == WATCH_FD);
1314         assert(w->data.unit == u);
1315         assert_se(epoll_ctl(u->meta.manager->epoll_fd, EPOLL_CTL_DEL, w->fd, NULL) >= 0);
1316
1317         w->fd = -1;
1318         w->type = WATCH_INVALID;
1319         w->data.unit = NULL;
1320 }
1321
1322 int unit_watch_pid(Unit *u, pid_t pid) {
1323         assert(u);
1324         assert(pid >= 1);
1325
1326         /* Watch a specific PID. We only support one unit watching
1327          * each PID for now. */
1328
1329         return hashmap_put(u->meta.manager->watch_pids, LONG_TO_PTR(pid), u);
1330 }
1331
1332 void unit_unwatch_pid(Unit *u, pid_t pid) {
1333         assert(u);
1334         assert(pid >= 1);
1335
1336         hashmap_remove_value(u->meta.manager->watch_pids, LONG_TO_PTR(pid), u);
1337 }
1338
1339 int unit_watch_timer(Unit *u, usec_t delay, Watch *w) {
1340         struct itimerspec its;
1341         int flags, fd;
1342         bool ours;
1343
1344         assert(u);
1345         assert(w);
1346         assert(w->type == WATCH_INVALID || (w->type == WATCH_UNIT_TIMER && w->data.unit == u));
1347
1348         /* This will try to reuse the old timer if there is one */
1349
1350         if (w->type == WATCH_UNIT_TIMER) {
1351                 assert(w->data.unit == u);
1352                 assert(w->fd >= 0);
1353
1354                 ours = false;
1355                 fd = w->fd;
1356         } else if (w->type == WATCH_INVALID) {
1357
1358                 ours = true;
1359                 if ((fd = timerfd_create(CLOCK_MONOTONIC, TFD_NONBLOCK|TFD_CLOEXEC)) < 0)
1360                         return -errno;
1361         } else
1362                 assert_not_reached("Invalid watch type");
1363
1364         zero(its);
1365
1366         if (delay <= 0) {
1367                 /* Set absolute time in the past, but not 0, since we
1368                  * don't want to disarm the timer */
1369                 its.it_value.tv_sec = 0;
1370                 its.it_value.tv_nsec = 1;
1371
1372                 flags = TFD_TIMER_ABSTIME;
1373         } else {
1374                 timespec_store(&its.it_value, delay);
1375                 flags = 0;
1376         }
1377
1378         /* This will also flush the elapse counter */
1379         if (timerfd_settime(fd, flags, &its, NULL) < 0)
1380                 goto fail;
1381
1382         if (w->type == WATCH_INVALID) {
1383                 struct epoll_event ev;
1384
1385                 zero(ev);
1386                 ev.data.ptr = w;
1387                 ev.events = EPOLLIN;
1388
1389                 if (epoll_ctl(u->meta.manager->epoll_fd, EPOLL_CTL_ADD, fd, &ev) < 0)
1390                         goto fail;
1391         }
1392
1393         w->type = WATCH_UNIT_TIMER;
1394         w->fd = fd;
1395         w->data.unit = u;
1396
1397         return 0;
1398
1399 fail:
1400         if (ours)
1401                 close_nointr_nofail(fd);
1402
1403         return -errno;
1404 }
1405
1406 void unit_unwatch_timer(Unit *u, Watch *w) {
1407         assert(u);
1408         assert(w);
1409
1410         if (w->type == WATCH_INVALID)
1411                 return;
1412
1413         assert(w->type == WATCH_UNIT_TIMER);
1414         assert(w->data.unit == u);
1415         assert(w->fd >= 0);
1416
1417         assert_se(epoll_ctl(u->meta.manager->epoll_fd, EPOLL_CTL_DEL, w->fd, NULL) >= 0);
1418         close_nointr_nofail(w->fd);
1419
1420         w->fd = -1;
1421         w->type = WATCH_INVALID;
1422         w->data.unit = NULL;
1423 }
1424
1425 bool unit_job_is_applicable(Unit *u, JobType j) {
1426         assert(u);
1427         assert(j >= 0 && j < _JOB_TYPE_MAX);
1428
1429         switch (j) {
1430
1431         case JOB_VERIFY_ACTIVE:
1432         case JOB_START:
1433         case JOB_STOP:
1434                 return true;
1435
1436         case JOB_RESTART:
1437         case JOB_TRY_RESTART:
1438                 return unit_can_start(u);
1439
1440         case JOB_RELOAD:
1441                 return unit_can_reload(u);
1442
1443         case JOB_RELOAD_OR_START:
1444                 return unit_can_reload(u) && unit_can_start(u);
1445
1446         default:
1447                 assert_not_reached("Invalid job type");
1448         }
1449 }
1450
1451 int unit_add_dependency(Unit *u, UnitDependency d, Unit *other, bool add_reference) {
1452
1453         static const UnitDependency inverse_table[_UNIT_DEPENDENCY_MAX] = {
1454                 [UNIT_REQUIRES] = UNIT_REQUIRED_BY,
1455                 [UNIT_REQUIRES_OVERRIDABLE] = UNIT_REQUIRED_BY_OVERRIDABLE,
1456                 [UNIT_WANTS] = UNIT_WANTED_BY,
1457                 [UNIT_REQUISITE] = UNIT_REQUIRED_BY,
1458                 [UNIT_REQUISITE_OVERRIDABLE] = UNIT_REQUIRED_BY_OVERRIDABLE,
1459                 [UNIT_BIND_TO] = UNIT_BOUND_BY,
1460                 [UNIT_REQUIRED_BY] = _UNIT_DEPENDENCY_INVALID,
1461                 [UNIT_REQUIRED_BY_OVERRIDABLE] = _UNIT_DEPENDENCY_INVALID,
1462                 [UNIT_WANTED_BY] = _UNIT_DEPENDENCY_INVALID,
1463                 [UNIT_BOUND_BY] = UNIT_BIND_TO,
1464                 [UNIT_CONFLICTS] = UNIT_CONFLICTED_BY,
1465                 [UNIT_CONFLICTED_BY] = UNIT_CONFLICTS,
1466                 [UNIT_BEFORE] = UNIT_AFTER,
1467                 [UNIT_AFTER] = UNIT_BEFORE,
1468                 [UNIT_ON_FAILURE] = _UNIT_DEPENDENCY_INVALID,
1469                 [UNIT_REFERENCES] = UNIT_REFERENCED_BY,
1470                 [UNIT_REFERENCED_BY] = UNIT_REFERENCES
1471         };
1472         int r, q = 0, v = 0, w = 0;
1473
1474         assert(u);
1475         assert(d >= 0 && d < _UNIT_DEPENDENCY_MAX);
1476         assert(other);
1477
1478         u = unit_follow_merge(u);
1479         other = unit_follow_merge(other);
1480
1481         /* We won't allow dependencies on ourselves. We will not
1482          * consider them an error however. */
1483         if (u == other)
1484                 return 0;
1485
1486         if ((r = set_ensure_allocated(&u->meta.dependencies[d], trivial_hash_func, trivial_compare_func)) < 0)
1487                 return r;
1488
1489         if (inverse_table[d] != _UNIT_DEPENDENCY_INVALID)
1490                 if ((r = set_ensure_allocated(&other->meta.dependencies[inverse_table[d]], trivial_hash_func, trivial_compare_func)) < 0)
1491                         return r;
1492
1493         if (add_reference)
1494                 if ((r = set_ensure_allocated(&u->meta.dependencies[UNIT_REFERENCES], trivial_hash_func, trivial_compare_func)) < 0 ||
1495                     (r = set_ensure_allocated(&other->meta.dependencies[UNIT_REFERENCED_BY], trivial_hash_func, trivial_compare_func)) < 0)
1496                         return r;
1497
1498         if ((q = set_put(u->meta.dependencies[d], other)) < 0)
1499                 return q;
1500
1501         if (inverse_table[d] != _UNIT_DEPENDENCY_INVALID)
1502                 if ((v = set_put(other->meta.dependencies[inverse_table[d]], u)) < 0) {
1503                         r = v;
1504                         goto fail;
1505                 }
1506
1507         if (add_reference) {
1508                 if ((w = set_put(u->meta.dependencies[UNIT_REFERENCES], other)) < 0) {
1509                         r = w;
1510                         goto fail;
1511                 }
1512
1513                 if ((r = set_put(other->meta.dependencies[UNIT_REFERENCED_BY], u)) < 0)
1514                         goto fail;
1515         }
1516
1517         unit_add_to_dbus_queue(u);
1518         return 0;
1519
1520 fail:
1521         if (q > 0)
1522                 set_remove(u->meta.dependencies[d], other);
1523
1524         if (v > 0)
1525                 set_remove(other->meta.dependencies[inverse_table[d]], u);
1526
1527         if (w > 0)
1528                 set_remove(u->meta.dependencies[UNIT_REFERENCES], other);
1529
1530         return r;
1531 }
1532
1533 int unit_add_two_dependencies(Unit *u, UnitDependency d, UnitDependency e, Unit *other, bool add_reference) {
1534         int r;
1535
1536         assert(u);
1537
1538         if ((r = unit_add_dependency(u, d, other, add_reference)) < 0)
1539                 return r;
1540
1541         if ((r = unit_add_dependency(u, e, other, add_reference)) < 0)
1542                 return r;
1543
1544         return 0;
1545 }
1546
1547 static const char *resolve_template(Unit *u, const char *name, const char*path, char **p) {
1548         char *s;
1549
1550         assert(u);
1551         assert(name || path);
1552
1553         if (!name)
1554                 name = file_name_from_path(path);
1555
1556         if (!unit_name_is_template(name)) {
1557                 *p = NULL;
1558                 return name;
1559         }
1560
1561         if (u->meta.instance)
1562                 s = unit_name_replace_instance(name, u->meta.instance);
1563         else {
1564                 char *i;
1565
1566                 if (!(i = unit_name_to_prefix(u->meta.id)))
1567                         return NULL;
1568
1569                 s = unit_name_replace_instance(name, i);
1570                 free(i);
1571         }
1572
1573         if (!s)
1574                 return NULL;
1575
1576         *p = s;
1577         return s;
1578 }
1579
1580 int unit_add_dependency_by_name(Unit *u, UnitDependency d, const char *name, const char *path, bool add_reference) {
1581         Unit *other;
1582         int r;
1583         char *s;
1584
1585         assert(u);
1586         assert(name || path);
1587
1588         if (!(name = resolve_template(u, name, path, &s)))
1589                 return -ENOMEM;
1590
1591         if ((r = manager_load_unit(u->meta.manager, name, path, NULL, &other)) < 0)
1592                 goto finish;
1593
1594         r = unit_add_dependency(u, d, other, add_reference);
1595
1596 finish:
1597         free(s);
1598         return r;
1599 }
1600
1601 int unit_add_two_dependencies_by_name(Unit *u, UnitDependency d, UnitDependency e, const char *name, const char *path, bool add_reference) {
1602         Unit *other;
1603         int r;
1604         char *s;
1605
1606         assert(u);
1607         assert(name || path);
1608
1609         if (!(name = resolve_template(u, name, path, &s)))
1610                 return -ENOMEM;
1611
1612         if ((r = manager_load_unit(u->meta.manager, name, path, NULL, &other)) < 0)
1613                 goto finish;
1614
1615         r = unit_add_two_dependencies(u, d, e, other, add_reference);
1616
1617 finish:
1618         free(s);
1619         return r;
1620 }
1621
1622 int unit_add_dependency_by_name_inverse(Unit *u, UnitDependency d, const char *name, const char *path, bool add_reference) {
1623         Unit *other;
1624         int r;
1625         char *s;
1626
1627         assert(u);
1628         assert(name || path);
1629
1630         if (!(name = resolve_template(u, name, path, &s)))
1631                 return -ENOMEM;
1632
1633         if ((r = manager_load_unit(u->meta.manager, name, path, NULL, &other)) < 0)
1634                 goto finish;
1635
1636         r = unit_add_dependency(other, d, u, add_reference);
1637
1638 finish:
1639         free(s);
1640         return r;
1641 }
1642
1643 int unit_add_two_dependencies_by_name_inverse(Unit *u, UnitDependency d, UnitDependency e, const char *name, const char *path, bool add_reference) {
1644         Unit *other;
1645         int r;
1646         char *s;
1647
1648         assert(u);
1649         assert(name || path);
1650
1651         if (!(name = resolve_template(u, name, path, &s)))
1652                 return -ENOMEM;
1653
1654         if ((r = manager_load_unit(u->meta.manager, name, path, NULL, &other)) < 0)
1655                 goto finish;
1656
1657         if ((r = unit_add_two_dependencies(other, d, e, u, add_reference)) < 0)
1658                 goto finish;
1659
1660 finish:
1661         free(s);
1662         return r;
1663 }
1664
1665 int set_unit_path(const char *p) {
1666         char *cwd, *c;
1667         int r;
1668
1669         /* This is mostly for debug purposes */
1670
1671         if (path_is_absolute(p)) {
1672                 if (!(c = strdup(p)))
1673                         return -ENOMEM;
1674         } else {
1675                 if (!(cwd = get_current_dir_name()))
1676                         return -errno;
1677
1678                 r = asprintf(&c, "%s/%s", cwd, p);
1679                 free(cwd);
1680
1681                 if (r < 0)
1682                         return -ENOMEM;
1683         }
1684
1685         if (setenv("SYSTEMD_UNIT_PATH", c, 0) < 0) {
1686                 r = -errno;
1687                 free(c);
1688                 return r;
1689         }
1690
1691         return 0;
1692 }
1693
1694 char *unit_dbus_path(Unit *u) {
1695         char *p, *e;
1696
1697         assert(u);
1698
1699         if (!u->meta.id)
1700                 return NULL;
1701
1702         if (!(e = bus_path_escape(u->meta.id)))
1703                 return NULL;
1704
1705         p = strappend("/org/freedesktop/systemd1/unit/", e);
1706         free(e);
1707
1708         return p;
1709 }
1710
1711 int unit_add_cgroup(Unit *u, CGroupBonding *b) {
1712         int r;
1713
1714         assert(u);
1715         assert(b);
1716
1717         assert(b->path);
1718
1719         if (!b->controller)
1720                 if (!(b->controller = strdup(SYSTEMD_CGROUP_CONTROLLER)))
1721                         return -ENOMEM;
1722
1723         /* Ensure this hasn't been added yet */
1724         assert(!b->unit);
1725
1726         if (streq(b->controller, SYSTEMD_CGROUP_CONTROLLER)) {
1727                 CGroupBonding *l;
1728
1729                 l = hashmap_get(u->meta.manager->cgroup_bondings, b->path);
1730                 LIST_PREPEND(CGroupBonding, by_path, l, b);
1731
1732                 if ((r = hashmap_replace(u->meta.manager->cgroup_bondings, b->path, l)) < 0) {
1733                         LIST_REMOVE(CGroupBonding, by_path, l, b);
1734                         return r;
1735                 }
1736         }
1737
1738         LIST_PREPEND(CGroupBonding, by_unit, u->meta.cgroup_bondings, b);
1739         b->unit = u;
1740
1741         return 0;
1742 }
1743
1744 static char *default_cgroup_path(Unit *u) {
1745         char *p;
1746         int r;
1747
1748         assert(u);
1749
1750         if (u->meta.instance) {
1751                 char *t;
1752
1753                 if (!(t = unit_name_template(u->meta.id)))
1754                         return NULL;
1755
1756                 r = asprintf(&p, "%s/%s/%s", u->meta.manager->cgroup_hierarchy, t, u->meta.instance);
1757                 free(t);
1758         } else
1759                 r = asprintf(&p, "%s/%s", u->meta.manager->cgroup_hierarchy, u->meta.id);
1760
1761         return r < 0 ? NULL : p;
1762 }
1763
1764 int unit_add_cgroup_from_text(Unit *u, const char *name) {
1765         char *controller = NULL, *path = NULL;
1766         CGroupBonding *b = NULL;
1767         int r;
1768
1769         assert(u);
1770         assert(name);
1771
1772         if ((r = cg_split_spec(name, &controller, &path)) < 0)
1773                 return r;
1774
1775         if (!path)
1776                 path = default_cgroup_path(u);
1777
1778         if (!controller)
1779                 controller = strdup(SYSTEMD_CGROUP_CONTROLLER);
1780
1781         if (!path || !controller) {
1782                 free(path);
1783                 free(controller);
1784
1785                 return -ENOMEM;
1786         }
1787
1788         if (cgroup_bonding_find_list(u->meta.cgroup_bondings, controller)) {
1789                 r = -EEXIST;
1790                 goto fail;
1791         }
1792
1793         if (!(b = new0(CGroupBonding, 1))) {
1794                 r = -ENOMEM;
1795                 goto fail;
1796         }
1797
1798         b->controller = controller;
1799         b->path = path;
1800         b->ours = false;
1801
1802         if ((r = unit_add_cgroup(u, b)) < 0)
1803                 goto fail;
1804
1805         return 0;
1806
1807 fail:
1808         free(path);
1809         free(controller);
1810         free(b);
1811
1812         return r;
1813 }
1814
1815 static int unit_add_one_default_cgroup(Unit *u, const char *controller) {
1816         CGroupBonding *b = NULL;
1817         int r = -ENOMEM;
1818
1819         assert(u);
1820
1821         if (!controller)
1822                 controller = SYSTEMD_CGROUP_CONTROLLER;
1823
1824         if (cgroup_bonding_find_list(u->meta.cgroup_bondings, controller))
1825                 return 0;
1826
1827         if (!(b = new0(CGroupBonding, 1)))
1828                 return -ENOMEM;
1829
1830         if (!(b->controller = strdup(controller)))
1831                 goto fail;
1832
1833         if (!(b->path = default_cgroup_path(u)))
1834                 goto fail;
1835
1836         b->ours = true;
1837         b->essential = streq(controller, SYSTEMD_CGROUP_CONTROLLER);
1838
1839         if ((r = unit_add_cgroup(u, b)) < 0)
1840                 goto fail;
1841
1842         return 0;
1843
1844 fail:
1845         free(b->path);
1846         free(b->controller);
1847         free(b);
1848
1849         return r;
1850 }
1851
1852 int unit_add_default_cgroups(Unit *u) {
1853         char **c;
1854         int r;
1855         assert(u);
1856
1857         /* Adds in the default cgroups, if they weren't specified
1858          * otherwise. */
1859
1860         if ((r = unit_add_one_default_cgroup(u, NULL)) < 0)
1861                 return r;
1862
1863         STRV_FOREACH(c, u->meta.manager->default_controllers)
1864                 if ((r = unit_add_one_default_cgroup(u, *c)) < 0)
1865                         return r;
1866
1867         return 0;
1868 }
1869
1870 CGroupBonding* unit_get_default_cgroup(Unit *u) {
1871         assert(u);
1872
1873         return cgroup_bonding_find_list(u->meta.cgroup_bondings, SYSTEMD_CGROUP_CONTROLLER);
1874 }
1875
1876 int unit_load_related_unit(Unit *u, const char *type, Unit **_found) {
1877         char *t;
1878         int r;
1879
1880         assert(u);
1881         assert(type);
1882         assert(_found);
1883
1884         if (!(t = unit_name_change_suffix(u->meta.id, type)))
1885                 return -ENOMEM;
1886
1887         assert(!unit_has_name(u, t));
1888
1889         r = manager_load_unit(u->meta.manager, t, NULL, NULL, _found);
1890         free(t);
1891
1892         assert(r < 0 || *_found != u);
1893
1894         return r;
1895 }
1896
1897 int unit_get_related_unit(Unit *u, const char *type, Unit **_found) {
1898         Unit *found;
1899         char *t;
1900
1901         assert(u);
1902         assert(type);
1903         assert(_found);
1904
1905         if (!(t = unit_name_change_suffix(u->meta.id, type)))
1906                 return -ENOMEM;
1907
1908         assert(!unit_has_name(u, t));
1909
1910         found = manager_get_unit(u->meta.manager, t);
1911         free(t);
1912
1913         if (!found)
1914                 return -ENOENT;
1915
1916         *_found = found;
1917         return 0;
1918 }
1919
1920 static char *specifier_prefix_and_instance(char specifier, void *data, void *userdata) {
1921         Unit *u = userdata;
1922         assert(u);
1923
1924         return unit_name_to_prefix_and_instance(u->meta.id);
1925 }
1926
1927 static char *specifier_prefix(char specifier, void *data, void *userdata) {
1928         Unit *u = userdata;
1929         assert(u);
1930
1931         return unit_name_to_prefix(u->meta.id);
1932 }
1933
1934 static char *specifier_prefix_unescaped(char specifier, void *data, void *userdata) {
1935         Unit *u = userdata;
1936         char *p, *r;
1937
1938         assert(u);
1939
1940         if (!(p = unit_name_to_prefix(u->meta.id)))
1941                 return NULL;
1942
1943         r = unit_name_unescape(p);
1944         free(p);
1945
1946         return r;
1947 }
1948
1949 static char *specifier_instance_unescaped(char specifier, void *data, void *userdata) {
1950         Unit *u = userdata;
1951         assert(u);
1952
1953         if (u->meta.instance)
1954                 return unit_name_unescape(u->meta.instance);
1955
1956         return strdup("");
1957 }
1958
1959 static char *specifier_filename(char specifier, void *data, void *userdata) {
1960         Unit *u = userdata;
1961         assert(u);
1962
1963         if (u->meta.instance)
1964                 return unit_name_path_unescape(u->meta.instance);
1965
1966         return unit_name_to_path(u->meta.instance);
1967 }
1968
1969 char *unit_name_printf(Unit *u, const char* format) {
1970
1971         /*
1972          * This will use the passed string as format string and
1973          * replace the following specifiers:
1974          *
1975          * %n: the full id of the unit                 (foo@bar.waldo)
1976          * %N: the id of the unit without the suffix   (foo@bar)
1977          * %p: the prefix                              (foo)
1978          * %i: the instance                            (bar)
1979          */
1980
1981         const Specifier table[] = {
1982                 { 'n', specifier_string,              u->meta.id },
1983                 { 'N', specifier_prefix_and_instance, NULL },
1984                 { 'p', specifier_prefix,              NULL },
1985                 { 'i', specifier_string,              u->meta.instance },
1986                 { 0, NULL, NULL }
1987         };
1988
1989         assert(u);
1990         assert(format);
1991
1992         return specifier_printf(format, table, u);
1993 }
1994
1995 char *unit_full_printf(Unit *u, const char *format) {
1996
1997         /* This is similar to unit_name_printf() but also supports
1998          * unescaping */
1999
2000         const Specifier table[] = {
2001                 { 'n', specifier_string,              u->meta.id },
2002                 { 'N', specifier_prefix_and_instance, NULL },
2003                 { 'p', specifier_prefix,              NULL },
2004                 { 'P', specifier_prefix_unescaped,    NULL },
2005                 { 'i', specifier_string,              u->meta.instance },
2006                 { 'I', specifier_instance_unescaped,  NULL },
2007                 { 'f', specifier_filename,            NULL },
2008                 { 0, NULL, NULL }
2009         };
2010
2011         assert(u);
2012         assert(format);
2013
2014         return specifier_printf(format, table, u);
2015 }
2016
2017 char **unit_full_printf_strv(Unit *u, char **l) {
2018         size_t n;
2019         char **r, **i, **j;
2020
2021         /* Applies unit_full_printf to every entry in l */
2022
2023         assert(u);
2024
2025         n = strv_length(l);
2026         if (!(r = new(char*, n+1)))
2027                 return NULL;
2028
2029         for (i = l, j = r; *i; i++, j++)
2030                 if (!(*j = unit_full_printf(u, *i)))
2031                         goto fail;
2032
2033         *j = NULL;
2034         return r;
2035
2036 fail:
2037         j--;
2038         while (j >= r)
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);