src/dbus-manager.c

   1 /*-*- Mode: C; c-basic-offset: 8 -*-*/
   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 <errno.h>
  23
  24 #include "dbus.h"
  25 #include "log.h"
  26 #include "dbus-manager.h"
  27 #include "strv.h"
  28 #include "bus-errors.h"
  29
  30 #define BUS_MANAGER_INTERFACE                                           \
  31         " <interface name=\"org.freedesktop.systemd1.Manager\">\n"      \
  32         "  <method name=\"GetUnit\">\n"                                 \
  33         "   <arg name=\"name\" type=\"s\" direction=\"in\"/>\n"         \
  34         "   <arg name=\"unit\" type=\"o\" direction=\"out\"/>\n"        \
  35         "  </method>\n"                                                 \
  36         "  <method name=\"LoadUnit\">\n"                                \
  37         "   <arg name=\"name\" type=\"s\" direction=\"in\"/>\n"         \
  38         "   <arg name=\"unit\" type=\"o\" direction=\"out\"/>\n"        \
  39         "  </method>\n"                                                 \
  40         "  <method name=\"StartUnit\">\n"                               \
  41         "   <arg name=\"name\" type=\"s\" direction=\"in\"/>\n"         \
  42         "   <arg name=\"mode\" type=\"s\" direction=\"in\"/>\n"         \
  43         "   <arg name=\"job\" type=\"o\" direction=\"out\"/>\n"         \
  44         "  </method>\n"                                                 \
  45         "  <method name=\"StopUnit\">\n"                                \
  46         "   <arg name=\"name\" type=\"s\" direction=\"in\"/>\n"         \
  47         "   <arg name=\"mode\" type=\"s\" direction=\"in\"/>\n"         \
  48         "   <arg name=\"job\" type=\"o\" direction=\"out\"/>\n"         \
  49         "  </method>\n"                                                 \
  50         "  <method name=\"ReloadUnit\">\n"                              \
  51         "   <arg name=\"name\" type=\"s\" direction=\"in\"/>\n"         \
  52         "   <arg name=\"mode\" type=\"s\" direction=\"in\"/>\n"         \
  53         "   <arg name=\"job\" type=\"o\" direction=\"out\"/>\n"         \
  54         "  </method>\n"                                                 \
  55         "  <method name=\"RestartUnit\">\n"                             \
  56         "   <arg name=\"name\" type=\"s\" direction=\"in\"/>\n"         \
  57         "   <arg name=\"mode\" type=\"s\" direction=\"in\"/>\n"         \
  58         "   <arg name=\"job\" type=\"o\" direction=\"out\"/>\n"         \
  59         "  </method>\n"                                                 \
  60         "  <method name=\"TryRestartUnit\">\n"                          \
  61         "   <arg name=\"name\" type=\"s\" direction=\"in\"/>\n"         \
  62         "   <arg name=\"mode\" type=\"s\" direction=\"in\"/>\n"         \
  63         "   <arg name=\"job\" type=\"o\" direction=\"out\"/>\n"         \
  64         "  </method>\n"                                                 \
  65         "  <method name=\"ReloadOrRestartUnit\">\n"                     \
  66         "   <arg name=\"name\" type=\"s\" direction=\"in\"/>\n"         \
  67         "   <arg name=\"mode\" type=\"s\" direction=\"in\"/>\n"         \
  68         "   <arg name=\"job\" type=\"o\" direction=\"out\"/>\n"         \
  69         "  </method>\n"                                                 \
  70         "  <method name=\"ReloadOrTryRestartUnit\">\n"                  \
  71         "   <arg name=\"name\" type=\"s\" direction=\"in\"/>\n"         \
  72         "   <arg name=\"mode\" type=\"s\" direction=\"in\"/>\n"         \
  73         "   <arg name=\"job\" type=\"o\" direction=\"out\"/>\n"         \
  74         "  </method>\n"                                                 \
  75         "  <method name=\"ResetMaintenanceUnit\">\n"                    \
  76         "   <arg name=\"name\" type=\"s\" direction=\"in\"/>\n"         \
  77         "  </method>\n"                                                 \
  78         "  <method name=\"GetJob\">\n"                                  \
  79         "   <arg name=\"id\" type=\"u\" direction=\"in\"/>\n"           \
  80         "   <arg name=\"job\" type=\"o\" direction=\"out\"/>\n"         \
  81         "  </method>\n"                                                 \
  82         "  <method name=\"ClearJobs\"/>\n"                              \
  83         "  <method name=\"ResetMaintenance\"/>\n"                       \
  84         "  <method name=\"ListUnits\">\n"                               \
  85         "   <arg name=\"units\" type=\"a(ssssssouso)\" direction=\"out\"/>\n" \
  86         "  </method>\n"                                                 \
  87         "  <method name=\"ListJobs\">\n"                                \
  88         "   <arg name=\"jobs\" type=\"a(usssoo)\" direction=\"out\"/>\n" \
  89         "  </method>\n"                                                 \
  90         "  <method name=\"Subscribe\"/>\n"                              \
  91         "  <method name=\"Unsubscribe\"/>\n"                            \
  92         "  <method name=\"Dump\"/>\n"                                   \
  93         "  <method name=\"CreateSnapshot\">\n"                          \
  94         "   <arg name=\"name\" type=\"s\" direction=\"in\"/>\n"         \
  95         "   <arg nane=\"cleanup\" type=\"b\" direction=\"in\"/>\n"      \
  96         "   <arg name=\"unit\" type=\"o\" direction=\"out\"/>\n"        \
  97         "  </method>\n"                                                 \
  98         "  <method name=\"Reload\"/>\n"                                 \
  99         "  <method name=\"Reexecute\"/>\n"                              \
 100         "  <method name=\"Exit\"/>\n"                                   \
 101         "  <method name=\"SetEnvironment\">\n"                          \
 102         "   <arg name=\"names\" type=\"as\" direction=\"in\"/>\n"       \
 103         "  </method>\n"                                                 \
 104         "  <method name=\"UnsetEnvironment\">\n"                        \
 105         "   <arg name=\"names\" type=\"as\" direction=\"in\"/>\n"       \
 106         "  </method>\n"                                                 \
 107         "  <signal name=\"UnitNew\">\n"                                 \
 108         "   <arg name=\"id\" type=\"s\"/>\n"                            \
 109         "   <arg name=\"unit\" type=\"o\"/>\n"                          \
 110         "  </signal>\n"                                                 \
 111         "  <signal name=\"UnitRemoved\">\n"                             \
 112         "   <arg name=\"id\" type=\"s\"/>\n"                            \
 113         "   <arg name=\"unit\" type=\"o\"/>\n"                          \
 114         "  </signal>\n"                                                 \
 115         "  <signal name=\"JobNew\">\n"                                  \
 116         "   <arg name=\"id\" type=\"u\"/>\n"                            \
 117         "   <arg name=\"job\" type=\"o\"/>\n"                           \
 118         "  </signal>\n"                                                 \
 119         "  <signal name=\"JobRemoved\">\n"                              \
 120         "   <arg name=\"id\" type=\"u\"/>\n"                            \
 121         "   <arg name=\"job\" type=\"o\"/>\n"                           \
 122         "   <arg name=\"success\" type=\"b\"/>\n"                       \
 123         "  </signal>"                                                   \
 124         "  <property name=\"Version\" type=\"s\" access=\"read\"/>\n"   \
 125         "  <property name=\"RunningAs\" type=\"s\" access=\"read\"/>\n" \
 126         "  <property name=\"StartupTimestamp\" type=\"t\" access=\"read\"/>\n" \
 127         "  <property name=\"LogLevel\" type=\"s\" access=\"read\"/>\n"  \
 128         "  <property name=\"LogTarget\" type=\"s\" access=\"read\"/>\n" \
 129         "  <property name=\"NNames\" type=\"u\" access=\"read\"/>\n"    \
 130         "  <property name=\"NJobs\" type=\"u\" access=\"read\"/>\n"     \
 131         "  <property name=\"Environment\" type=\"as\" access=\"read\"/>\n" \
 132         "  <property name=\"ConfirmSpawn\" type=\"b\" access=\"read\"/>\n" \
 133         "  <property name=\"ShowStatus\" type=\"b\" access=\"read\"/>\n" \
 134         "  <property name=\"UnitPath\" type=\"as\" access=\"read\"/>\n" \
 135         "  <property name=\"SysVInitPath\" type=\"as\" access=\"read\"/>\n" \
 136         "  <property name=\"SysVRcndPath\" type=\"as\" access=\"read\"/>\n" \
 137         "  <property name=\"NotifySocket\" type=\"s\" access=\"read\"/>\n" \
 138         "  <property name=\"ControlGroupHierarchy\" type=\"s\" access=\"read\"/>\n" \
 139         " </interface>\n"
 140
 141 #define INTROSPECTION_BEGIN                                             \
 142         DBUS_INTROSPECT_1_0_XML_DOCTYPE_DECL_NODE                       \
 143         "<node>\n"                                                      \
 144         BUS_MANAGER_INTERFACE                                           \
 145         BUS_PROPERTIES_INTERFACE                                        \
 146         BUS_INTROSPECTABLE_INTERFACE
 147
 148 #define INTROSPECTION_END                                               \
 149         "</node>\n"
 150
 151 const char bus_manager_interface[] = BUS_MANAGER_INTERFACE;
 152
 153 static DEFINE_BUS_PROPERTY_APPEND_ENUM(bus_manager_append_running_as, manager_running_as, ManagerRunningAs);
 154
 155 static int bus_manager_append_log_target(Manager *m, DBusMessageIter *i, const char *property, void *data) {
 156         const char *t;
 157
 158         assert(m);
 159         assert(i);
 160         assert(property);
 161
 162         t = log_target_to_string(log_get_target());
 163
 164         if (!dbus_message_iter_append_basic(i, DBUS_TYPE_STRING, &t))
 165                 return -ENOMEM;
 166
 167         return 0;
 168 }
 169
 170 static int bus_manager_append_log_level(Manager *m, DBusMessageIter *i, const char *property, void *data) {
 171         const char *t;
 172
 173         assert(m);
 174         assert(i);
 175         assert(property);
 176
 177         t = log_level_to_string(log_get_max_level());
 178
 179         if (!dbus_message_iter_append_basic(i, DBUS_TYPE_STRING, &t))
 180                 return -ENOMEM;
 181
 182         return 0;
 183 }
 184
 185 static int bus_manager_append_n_names(Manager *m, DBusMessageIter *i, const char *property, void *data) {
 186         uint32_t u;
 187
 188         assert(m);
 189         assert(i);
 190         assert(property);
 191
 192         u = hashmap_size(m->units);
 193
 194         if (!dbus_message_iter_append_basic(i, DBUS_TYPE_UINT32, &u))
 195                 return -ENOMEM;
 196
 197         return 0;
 198 }
 199
 200 static int bus_manager_append_n_jobs(Manager *m, DBusMessageIter *i, const char *property, void *data) {
 201         uint32_t u;
 202
 203         assert(m);
 204         assert(i);
 205         assert(property);
 206
 207         u = hashmap_size(m->jobs);
 208
 209         if (!dbus_message_iter_append_basic(i, DBUS_TYPE_UINT32, &u))
 210                 return -ENOMEM;
 211
 212         return 0;
 213 }
 214
 215 static const char *message_get_sender_with_fallback(DBusMessage *m) {
 216         const char *s;
 217
 218         assert(m);
 219
 220         if ((s = dbus_message_get_sender(m)))
 221                 return s;
 222
 223         /* When the message came in from a direct connection the
 224          * message will have no sender. We fix that here. */
 225
 226         return ":no-sender";
 227 }
 228
 229 static DBusHandlerResult bus_manager_message_handler(DBusConnection *connection, DBusMessage *message, void *data) {
 230         Manager *m = data;
 231
 232         const BusProperty properties[] = {
 233                 { "org.freedesktop.systemd1.Manager", "Version",       bus_property_append_string,    "s",  PACKAGE_STRING     },
 234                 { "org.freedesktop.systemd1.Manager", "RunningAs",     bus_manager_append_running_as, "s",  &m->running_as     },
 235                 { "org.freedesktop.systemd1.Manager", "StartupTimestamp", bus_property_append_uint64, "t",  &m->startup_timestamp.realtime },
 236                 { "org.freedesktop.systemd1.Manager", "LogLevel",      bus_manager_append_log_level,  "s",  NULL               },
 237                 { "org.freedesktop.systemd1.Manager", "LogTarget",     bus_manager_append_log_target, "s",  NULL               },
 238                 { "org.freedesktop.systemd1.Manager", "NNames",        bus_manager_append_n_names,    "u",  NULL               },
 239                 { "org.freedesktop.systemd1.Manager", "NJobs",         bus_manager_append_n_jobs,     "u",  NULL               },
 240                 { "org.freedesktop.systemd1.Manager", "Environment",   bus_property_append_strv,      "as", m->environment     },
 241                 { "org.freedesktop.systemd1.Manager", "ConfirmSpawn",  bus_property_append_bool,      "b",  &m->confirm_spawn  },
 242                 { "org.freedesktop.systemd1.Manager", "ShowStatus",    bus_property_append_bool,      "b",  &m->show_status    },
 243                 { "org.freedesktop.systemd1.Manager", "UnitPath",      bus_property_append_strv,      "as", m->lookup_paths.unit_path },
 244                 { "org.freedesktop.systemd1.Manager", "SysVInitPath",  bus_property_append_strv,      "as", m->lookup_paths.sysvinit_path },
 245                 { "org.freedesktop.systemd1.Manager", "SysVRcndPath",  bus_property_append_strv,      "as", m->lookup_paths.sysvrcnd_path },
 246                 { "org.freedesktop.systemd1.Manager", "NotifySocket",  bus_property_append_string,    "s",  m->notify_socket   },
 247                 { "org.freedesktop.systemd1.Manager", "ControlGroupHierarchy", bus_property_append_string, "s", m->cgroup_hierarchy },
 248                 { NULL, NULL, NULL, NULL, NULL }
 249         };
 250
 251         int r;
 252         DBusError error;
 253         DBusMessage *reply = NULL;
 254         char * path = NULL;
 255         JobType job_type = _JOB_TYPE_INVALID;
 256         bool reload_if_possible = false;
 257
 258         assert(connection);
 259         assert(message);
 260         assert(m);
 261
 262         dbus_error_init(&error);
 263
 264         if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Manager", "GetUnit")) {
 265                 const char *name;
 266                 Unit *u;
 267
 268                 if (!dbus_message_get_args(
 269                                     message,
 270                                     &error,
 271                                     DBUS_TYPE_STRING, &name,
 272                                     DBUS_TYPE_INVALID))
 273                         return bus_send_error_reply(m, connection, message, &error, -EINVAL);
 274
 275                 if (!(u = manager_get_unit(m, name))) {
 276                         dbus_set_error(&error, BUS_ERROR_NO_SUCH_UNIT, "Unit %s is not loaded.", name);
 277                         return bus_send_error_reply(m, connection, message, &error, -ENOENT);
 278                 }
 279
 280                 if (!(reply = dbus_message_new_method_return(message)))
 281                         goto oom;
 282
 283                 if (!(path = unit_dbus_path(u)))
 284                         goto oom;
 285
 286                 if (!dbus_message_append_args(
 287                                     reply,
 288                                     DBUS_TYPE_OBJECT_PATH, &path,
 289                                     DBUS_TYPE_INVALID))
 290                         goto oom;
 291
 292         } else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Manager", "LoadUnit")) {
 293                 const char *name;
 294                 Unit *u;
 295
 296                 if (!dbus_message_get_args(
 297                                     message,
 298                                     &error,
 299                                     DBUS_TYPE_STRING, &name,
 300                                     DBUS_TYPE_INVALID))
 301                         return bus_send_error_reply(m, connection, message, &error, -EINVAL);
 302
 303                 if ((r = manager_load_unit(m, name, NULL, &error, &u)) < 0)
 304                         return bus_send_error_reply(m, connection, message, &error, r);
 305
 306                 if (!(reply = dbus_message_new_method_return(message)))
 307                         goto oom;
 308
 309                 if (!(path = unit_dbus_path(u)))
 310                         goto oom;
 311
 312                 if (!dbus_message_append_args(
 313                                     reply,
 314                                     DBUS_TYPE_OBJECT_PATH, &path,
 315                                     DBUS_TYPE_INVALID))
 316                         goto oom;
 317
 318         } else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Manager", "StartUnit"))
 319                 job_type = JOB_START;
 320         else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Manager", "StopUnit"))
 321                 job_type = JOB_STOP;
 322         else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Manager", "ReloadUnit"))
 323                 job_type = JOB_RELOAD;
 324         else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Manager", "RestartUnit"))
 325                 job_type = JOB_RESTART;
 326         else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Manager", "TryRestartUnit"))
 327                 job_type = JOB_TRY_RESTART;
 328         else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Manager", "ReloadOrRestartUnit")) {
 329                 reload_if_possible = true;
 330                 job_type = JOB_RESTART;
 331         } else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Manager", "ReloadOrTryRestartUnit")) {
 332                 reload_if_possible = true;
 333                 job_type = JOB_TRY_RESTART;
 334         } else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Manager", "GetJob")) {
 335                 uint32_t id;
 336                 Job *j;
 337
 338                 if (!dbus_message_get_args(
 339                                     message,
 340                                     &error,
 341                                     DBUS_TYPE_UINT32, &id,
 342                                     DBUS_TYPE_INVALID))
 343                         return bus_send_error_reply(m, connection, message, &error, -EINVAL);
 344
 345                 if (!(j = manager_get_job(m, id))) {
 346                         dbus_set_error(&error, BUS_ERROR_NO_SUCH_JOB, "Job %u does not exist.", (unsigned) id);
 347                         return bus_send_error_reply(m, connection, message, &error, -ENOENT);
 348                 }
 349
 350                 if (!(reply = dbus_message_new_method_return(message)))
 351                         goto oom;
 352
 353                 if (!(path = job_dbus_path(j)))
 354                         goto oom;
 355
 356                 if (!dbus_message_append_args(
 357                                     reply,
 358                                     DBUS_TYPE_OBJECT_PATH, &path,
 359                                     DBUS_TYPE_INVALID))
 360                         goto oom;
 361
 362         } else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Manager", "ClearJobs")) {
 363
 364                 manager_clear_jobs(m);
 365
 366                 if (!(reply = dbus_message_new_method_return(message)))
 367                         goto oom;
 368
 369         } else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Manager", "ResetMaintenance")) {
 370
 371                 manager_reset_maintenance(m);
 372
 373                 if (!(reply = dbus_message_new_method_return(message)))
 374                         goto oom;
 375
 376         } else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Manager", "ResetMaintenanceUnit")) {
 377                 const char *name;
 378                 Unit *u;
 379
 380                 if (!dbus_message_get_args(
 381                                     message,
 382                                     &error,
 383                                     DBUS_TYPE_STRING, &name,
 384                                     DBUS_TYPE_INVALID))
 385                         return bus_send_error_reply(m, connection, message, &error, -EINVAL);
 386
 387                 if (!(u = manager_get_unit(m, name))) {
 388                         dbus_set_error(&error, BUS_ERROR_NO_SUCH_UNIT, "Unit %s is not loaded.", name);
 389                         return bus_send_error_reply(m, connection, message, &error, -ENOENT);
 390                 }
 391
 392                 unit_reset_maintenance(u);
 393
 394                 if (!(reply = dbus_message_new_method_return(message)))
 395                         goto oom;
 396
 397         } else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Manager", "ListUnits")) {
 398                 DBusMessageIter iter, sub;
 399                 Iterator i;
 400                 Unit *u;
 401                 const char *k;
 402
 403                 if (!(reply = dbus_message_new_method_return(message)))
 404                         goto oom;
 405
 406                 dbus_message_iter_init_append(reply, &iter);
 407
 408                 if (!dbus_message_iter_open_container(&iter, DBUS_TYPE_ARRAY, "(ssssssouso)", &sub))
 409                         goto oom;
 410
 411                 HASHMAP_FOREACH_KEY(u, k, m->units, i) {
 412                         char *u_path, *j_path;
 413                         const char *description, *load_state, *active_state, *sub_state, *sjob_type, *following;
 414                         DBusMessageIter sub2;
 415                         uint32_t job_id;
 416
 417                         if (k != u->meta.id)
 418                                 continue;
 419
 420                         if (!dbus_message_iter_open_container(&sub, DBUS_TYPE_STRUCT, NULL, &sub2))
 421                                 goto oom;
 422
 423                         description = unit_description(u);
 424                         load_state = unit_load_state_to_string(u->meta.load_state);
 425                         active_state = unit_active_state_to_string(unit_active_state(u));
 426                         sub_state = unit_sub_state_to_string(u);
 427                         following = u->meta.following ? u->meta.following->meta.id : "";
 428
 429                         if (!(u_path = unit_dbus_path(u)))
 430                                 goto oom;
 431
 432                         if (u->meta.job) {
 433                                 job_id = (uint32_t) u->meta.job->id;
 434
 435                                 if (!(j_path = job_dbus_path(u->meta.job))) {
 436                                         free(u_path);
 437                                         goto oom;
 438                                 }
 439
 440                                 sjob_type = job_type_to_string(u->meta.job->type);
 441                         } else {
 442                                 job_id = 0;
 443                                 j_path = u_path;
 444                                 sjob_type = "";
 445                         }
 446
 447                         if (!dbus_message_iter_append_basic(&sub2, DBUS_TYPE_STRING, &u->meta.id) ||
 448                             !dbus_message_iter_append_basic(&sub2, DBUS_TYPE_STRING, &description) ||
 449                             !dbus_message_iter_append_basic(&sub2, DBUS_TYPE_STRING, &load_state) ||
 450                             !dbus_message_iter_append_basic(&sub2, DBUS_TYPE_STRING, &active_state) ||
 451                             !dbus_message_iter_append_basic(&sub2, DBUS_TYPE_STRING, &sub_state) ||
 452                             !dbus_message_iter_append_basic(&sub2, DBUS_TYPE_STRING, &following) ||
 453                             !dbus_message_iter_append_basic(&sub2, DBUS_TYPE_OBJECT_PATH, &u_path) ||
 454                             !dbus_message_iter_append_basic(&sub2, DBUS_TYPE_UINT32, &job_id) ||
 455                             !dbus_message_iter_append_basic(&sub2, DBUS_TYPE_STRING, &sjob_type) ||
 456                             !dbus_message_iter_append_basic(&sub2, DBUS_TYPE_OBJECT_PATH, &j_path)) {
 457                                 free(u_path);
 458                                 if (u->meta.job)
 459                                         free(j_path);
 460                                 goto oom;
 461                         }
 462
 463                         free(u_path);
 464                         if (u->meta.job)
 465                                 free(j_path);
 466
 467                         if (!dbus_message_iter_close_container(&sub, &sub2))
 468                                 goto oom;
 469                 }
 470
 471                 if (!dbus_message_iter_close_container(&iter, &sub))
 472                         goto oom;
 473
 474         } else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Manager", "ListJobs")) {
 475                 DBusMessageIter iter, sub;
 476                 Iterator i;
 477                 Job *j;
 478
 479                 if (!(reply = dbus_message_new_method_return(message)))
 480                         goto oom;
 481
 482                 dbus_message_iter_init_append(reply, &iter);
 483
 484                 if (!dbus_message_iter_open_container(&iter, DBUS_TYPE_ARRAY, "(usssoo)", &sub))
 485                         goto oom;
 486
 487                 HASHMAP_FOREACH(j, m->jobs, i) {
 488                         char *u_path, *j_path;
 489                         const char *state, *type;
 490                         uint32_t id;
 491                         DBusMessageIter sub2;
 492
 493                         if (!dbus_message_iter_open_container(&sub, DBUS_TYPE_STRUCT, NULL, &sub2))
 494                                 goto oom;
 495
 496                         id = (uint32_t) j->id;
 497                         state = job_state_to_string(j->state);
 498                         type = job_type_to_string(j->type);
 499
 500                         if (!(j_path = job_dbus_path(j)))
 501                                 goto oom;
 502
 503                         if (!(u_path = unit_dbus_path(j->unit))) {
 504                                 free(j_path);
 505                                 goto oom;
 506                         }
 507
 508                         if (!dbus_message_iter_append_basic(&sub2, DBUS_TYPE_UINT32, &id) ||
 509                             !dbus_message_iter_append_basic(&sub2, DBUS_TYPE_STRING, &j->unit->meta.id) ||
 510                             !dbus_message_iter_append_basic(&sub2, DBUS_TYPE_STRING, &type) ||
 511                             !dbus_message_iter_append_basic(&sub2, DBUS_TYPE_STRING, &state) ||
 512                             !dbus_message_iter_append_basic(&sub2, DBUS_TYPE_OBJECT_PATH, &j_path) ||
 513                             !dbus_message_iter_append_basic(&sub2, DBUS_TYPE_OBJECT_PATH, &u_path)) {
 514                                 free(j_path);
 515                                 free(u_path);
 516                                 goto oom;
 517                         }
 518
 519                         free(j_path);
 520                         free(u_path);
 521
 522                         if (!dbus_message_iter_close_container(&sub, &sub2))
 523                                 goto oom;
 524                 }
 525
 526                 if (!dbus_message_iter_close_container(&iter, &sub))
 527                         goto oom;
 528
 529         } else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Manager", "Subscribe")) {
 530                 char *client;
 531                 Set *s;
 532
 533                 if (!(s = BUS_CONNECTION_SUBSCRIBED(m, connection))) {
 534                         if (!(s = set_new(string_hash_func, string_compare_func)))
 535                                 goto oom;
 536
 537                         if (!(dbus_connection_set_data(connection, m->subscribed_data_slot, s, NULL))) {
 538                                 set_free(s);
 539                                 goto oom;
 540                         }
 541                 }
 542
 543                 if (!(client = strdup(message_get_sender_with_fallback(message))))
 544                         goto oom;
 545
 546                 if ((r = set_put(s, client)) < 0) {
 547                         free(client);
 548                         return bus_send_error_reply(m, connection, message, NULL, r);
 549                 }
 550
 551                 if (!(reply = dbus_message_new_method_return(message)))
 552                         goto oom;
 553
 554         } else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Manager", "Unsubscribe")) {
 555                 char *client;
 556
 557                 if (!(client = set_remove(BUS_CONNECTION_SUBSCRIBED(m, connection), (char*) message_get_sender_with_fallback(message)))) {
 558                         dbus_set_error(&error, BUS_ERROR_NOT_SUBSCRIBED, "Client is not subscribed.");
 559                         return bus_send_error_reply(m, connection, message, &error, -ENOENT);
 560                 }
 561
 562                 free(client);
 563
 564                 if (!(reply = dbus_message_new_method_return(message)))
 565                         goto oom;
 566
 567         } else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Manager", "Dump")) {
 568                 FILE *f;
 569                 char *dump = NULL;
 570                 size_t size;
 571
 572                 if (!(reply = dbus_message_new_method_return(message)))
 573                         goto oom;
 574
 575                 if (!(f = open_memstream(&dump, &size)))
 576                         goto oom;
 577
 578                 manager_dump_units(m, f, NULL);
 579                 manager_dump_jobs(m, f, NULL);
 580
 581                 if (ferror(f)) {
 582                         fclose(f);
 583                         free(dump);
 584                         goto oom;
 585                 }
 586
 587                 fclose(f);
 588
 589                 if (!dbus_message_append_args(reply, DBUS_TYPE_STRING, &dump, DBUS_TYPE_INVALID)) {
 590                         free(dump);
 591                         goto oom;
 592                 }
 593
 594                 free(dump);
 595         } else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Manager", "CreateSnapshot")) {
 596                 const char *name;
 597                 dbus_bool_t cleanup;
 598                 Snapshot *s;
 599
 600                 if (!dbus_message_get_args(
 601                                     message,
 602                                     &error,
 603                                     DBUS_TYPE_STRING, &name,
 604                                     DBUS_TYPE_BOOLEAN, &cleanup,
 605                                     DBUS_TYPE_INVALID))
 606                         return bus_send_error_reply(m, connection, message, &error, -EINVAL);
 607
 608                 if (name && name[0] == 0)
 609                         name = NULL;
 610
 611                 if ((r = snapshot_create(m, name, cleanup, &error, &s)) < 0)
 612                         return bus_send_error_reply(m, connection, message, &error, r);
 613
 614                 if (!(reply = dbus_message_new_method_return(message)))
 615                         goto oom;
 616
 617                 if (!(path = unit_dbus_path(UNIT(s))))
 618                         goto oom;
 619
 620                 if (!dbus_message_append_args(
 621                                     reply,
 622                                     DBUS_TYPE_OBJECT_PATH, &path,
 623                                     DBUS_TYPE_INVALID))
 624                         goto oom;
 625
 626         } else if (dbus_message_is_method_call(message, "org.freedesktop.DBus.Introspectable", "Introspect")) {
 627                 char *introspection = NULL;
 628                 FILE *f;
 629                 Iterator i;
 630                 Unit *u;
 631                 Job *j;
 632                 const char *k;
 633                 size_t size;
 634
 635                 if (!(reply = dbus_message_new_method_return(message)))
 636                         goto oom;
 637
 638                 /* We roll our own introspection code here, instead of
 639                  * relying on bus_default_message_handler() because we
 640                  * need to generate our introspection string
 641                  * dynamically. */
 642
 643                 if (!(f = open_memstream(&introspection, &size)))
 644                         goto oom;
 645
 646                 fputs(INTROSPECTION_BEGIN, f);
 647
 648                 HASHMAP_FOREACH_KEY(u, k, m->units, i) {
 649                         char *p;
 650
 651                         if (k != u->meta.id)
 652                                 continue;
 653
 654                         if (!(p = bus_path_escape(k))) {
 655                                 fclose(f);
 656                                 free(introspection);
 657                                 goto oom;
 658                         }
 659
 660                         fprintf(f, "<node name=\"unit/%s\"/>", p);
 661                         free(p);
 662                 }
 663
 664                 HASHMAP_FOREACH(j, m->jobs, i)
 665                         fprintf(f, "<node name=\"job/%lu\"/>", (unsigned long) j->id);
 666
 667                 fputs(INTROSPECTION_END, f);
 668
 669                 if (ferror(f)) {
 670                         fclose(f);
 671                         free(introspection);
 672                         goto oom;
 673                 }
 674
 675                 fclose(f);
 676
 677                 if (!introspection)
 678                         goto oom;
 679
 680                 if (!dbus_message_append_args(reply, DBUS_TYPE_STRING, &introspection, DBUS_TYPE_INVALID)) {
 681                         free(introspection);
 682                         goto oom;
 683                 }
 684
 685                 free(introspection);
 686
 687         } else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Manager", "Reload")) {
 688
 689                 assert(!m->queued_message);
 690
 691                 /* Instead of sending the reply back right away, we
 692                  * just remember that we need to and then send it
 693                  * after the reload is finished. That way the caller
 694                  * knows when the reload finished. */
 695
 696                 if (!(m->queued_message = dbus_message_new_method_return(message)))
 697                         goto oom;
 698
 699                 m->queued_message_connection = connection;
 700                 m->exit_code = MANAGER_RELOAD;
 701
 702         } else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Manager", "Reexecute")) {
 703
 704                 if (!(reply = dbus_message_new_method_return(message)))
 705                         goto oom;
 706
 707                 m->exit_code = MANAGER_REEXECUTE;
 708
 709         } else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Manager", "Exit")) {
 710
 711                 if (m->running_as == MANAGER_SYSTEM) {
 712                         dbus_set_error(&error, BUS_ERROR_NOT_SUPPORTED, "Exit is only supported for session managers.");
 713                         return bus_send_error_reply(m, connection, message, &error, -ENOTSUP);
 714                 }
 715
 716                 if (!(reply = dbus_message_new_method_return(message)))
 717                         goto oom;
 718
 719                 m->exit_code = MANAGER_EXIT;
 720
 721         } else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Manager", "SetEnvironment")) {
 722                 char **l = NULL, **e = NULL;
 723
 724                 if ((r = bus_parse_strv(message, &l)) < 0) {
 725                         if (r == -ENOMEM)
 726                                 goto oom;
 727
 728                         return bus_send_error_reply(m, connection, message, NULL, r);
 729                 }
 730
 731                 e = strv_env_merge(2, m->environment, l);
 732                 strv_free(l);
 733
 734                 if (!e)
 735                         goto oom;
 736
 737                 if (!(reply = dbus_message_new_method_return(message))) {
 738                         strv_free(e);
 739                         goto oom;
 740                 }
 741
 742                 strv_free(m->environment);
 743                 m->environment = e;
 744
 745         } else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Manager", "UnsetEnvironment")) {
 746                 char **l = NULL, **e = NULL;
 747
 748                 if ((r = bus_parse_strv(message, &l)) < 0) {
 749                         if (r == -ENOMEM)
 750                                 goto oom;
 751
 752                         return bus_send_error_reply(m, connection, message, NULL, r);
 753                 }
 754
 755                 e = strv_env_delete(m->environment, 1, l);
 756                 strv_free(l);
 757
 758                 if (!e)
 759                         goto oom;
 760
 761                 if (!(reply = dbus_message_new_method_return(message)))
 762                         goto oom;
 763
 764                 strv_free(m->environment);
 765                 m->environment = e;
 766
 767         } else
 768                 return bus_default_message_handler(m, connection, message, NULL, properties);
 769
 770         if (job_type != _JOB_TYPE_INVALID) {
 771                 const char *name, *smode;
 772                 JobMode mode;
 773                 Job *j;
 774                 Unit *u;
 775
 776                 if (!dbus_message_get_args(
 777                                     message,
 778                                     &error,
 779                                     DBUS_TYPE_STRING, &name,
 780                                     DBUS_TYPE_STRING, &smode,
 781                                     DBUS_TYPE_INVALID))
 782                         return bus_send_error_reply(m, connection, message, &error, -EINVAL);
 783
 784                 if ((mode = job_mode_from_string(smode)) == _JOB_MODE_INVALID) {
 785                         dbus_set_error(&error, BUS_ERROR_INVALID_JOB_MODE, "Job mode %s is invalid.", smode);
 786                         return bus_send_error_reply(m, connection, message, &error, -EINVAL);
 787                 }
 788
 789                 if ((r = manager_load_unit(m, name, NULL, &error, &u)) < 0)
 790                         return bus_send_error_reply(m, connection, message, &error, r);
 791
 792                 if (reload_if_possible && unit_can_reload(u)) {
 793                         if (job_type == JOB_RESTART)
 794                                 job_type = JOB_RELOAD_OR_START;
 795                         else if (job_type == JOB_TRY_RESTART)
 796                                 job_type = JOB_RELOAD;
 797                 }
 798
 799                 if (job_type == JOB_START && u->meta.only_by_dependency) {
 800                         dbus_set_error(&error, BUS_ERROR_ONLY_BY_DEPENDENCY, "Unit may be activated by dependency only.");
 801                         return bus_send_error_reply(m, connection, message, &error, -EPERM);
 802                 }
 803
 804                 if ((r = manager_add_job(m, job_type, u, mode, true, &error, &j)) < 0)
 805                         return bus_send_error_reply(m, connection, message, &error, r);
 806
 807                 if (!(j->bus_client = strdup(message_get_sender_with_fallback(message))))
 808                         goto oom;
 809
 810                 j->bus = connection;
 811
 812                 if (!(reply = dbus_message_new_method_return(message)))
 813                         goto oom;
 814
 815                 if (!(path = job_dbus_path(j)))
 816                         goto oom;
 817
 818                 if (!dbus_message_append_args(
 819                                     reply,
 820                                     DBUS_TYPE_OBJECT_PATH, &path,
 821                                     DBUS_TYPE_INVALID))
 822                         goto oom;
 823         }
 824
 825         free(path);
 826
 827         if (reply) {
 828                 if (!dbus_connection_send(connection, reply, NULL))
 829                         goto oom;
 830
 831                 dbus_message_unref(reply);
 832         }
 833
 834         return DBUS_HANDLER_RESULT_HANDLED;
 835
 836 oom:
 837         free(path);
 838
 839         if (reply)
 840                 dbus_message_unref(reply);
 841
 842         dbus_error_free(&error);
 843
 844         return DBUS_HANDLER_RESULT_NEED_MEMORY;
 845 }
 846
 847 const DBusObjectPathVTable bus_manager_vtable = {
 848         .message_function = bus_manager_message_handler
 849 };