#include <errno.h>

#include "strv.h"

#include "path-util.h"

#include "dbus-unit.h"

#include "dbus-execute.h"

#include "dbus-kill.h"

#include "dbus-cgroup.h"

#include "dbus-common.h"

#include "selinux-access.h"

#include "dbus-service.h"

#define BUS_SERVICE_INTERFACE \

" <interface name=\"org.freedesktop.systemd1.Service\">\n" \

" <property name=\"Type\" type=\"s\" access=\"read\"/>\n" \

" <property name=\"Restart\" type=\"s\" access=\"read\"/>\n" \

" <property name=\"PIDFile\" type=\"s\" access=\"read\"/>\n" \

" <property name=\"NotifyAccess\" type=\"s\" access=\"read\"/>\n" \

" <property name=\"RestartUSec\" type=\"t\" access=\"read\"/>\n" \

" <property name=\"TimeoutStartUSec\" type=\"t\" access=\"read\"/>\n" \

" <property name=\"TimeoutStopUSec\" type=\"t\" access=\"read\"/>\n" \

" <property name=\"WatchdogUSec\" type=\"t\" access=\"read\"/>\n" \

" <property name=\"WatchdogTimestamp\" type=\"t\" access=\"read\"/>\n" \

" <property name=\"WatchdogTimestampMonotonic\" type=\"t\" access=\"read\"/>\n" \

" <property name=\"StartLimitInterval\" type=\"t\" access=\"read\"/>\n" \

" <property name=\"StartLimitBurst\" type=\"u\" access=\"read\"/>\n" \

" <property name=\"StartLimitAction\" type=\"s\" access=\"readwrite\"/>\n" \

BUS_UNIT_CGROUP_INTERFACE \

BUS_EXEC_COMMAND_INTERFACE("ExecStartPre") \

BUS_EXEC_COMMAND_INTERFACE("ExecStart") \

BUS_EXEC_COMMAND_INTERFACE("ExecStartPost") \

BUS_EXEC_COMMAND_INTERFACE("ExecReload") \

BUS_EXEC_COMMAND_INTERFACE("ExecStop") \

BUS_EXEC_COMMAND_INTERFACE("ExecStopPost") \

BUS_EXEC_CONTEXT_INTERFACE \

BUS_KILL_CONTEXT_INTERFACE \

BUS_CGROUP_CONTEXT_INTERFACE \

" <property name=\"PermissionsStartOnly\" type=\"b\" access=\"read\"/>\n" \

" <property name=\"RootDirectoryStartOnly\" type=\"b\" access=\"read\"/>\n" \

" <property name=\"RemainAfterExit\" type=\"b\" access=\"read\"/>\n" \

BUS_EXEC_STATUS_INTERFACE("ExecMain") \

" <property name=\"MainPID\" type=\"u\" access=\"read\"/>\n" \

" <property name=\"ControlPID\" type=\"u\" access=\"read\"/>\n" \

" <property name=\"BusName\" type=\"s\" access=\"read\"/>\n" \

" <property name=\"StatusText\" type=\"s\" access=\"read\"/>\n" \

" <property name=\"Result\" type=\"s\" access=\"read\"/>\n" \

" </interface>\n"

#define INTROSPECTION \

DBUS_INTROSPECT_1_0_XML_DOCTYPE_DECL_NODE \

"<node>\n" \

BUS_UNIT_INTERFACE \

BUS_SERVICE_INTERFACE \

BUS_PROPERTIES_INTERFACE \

BUS_PEER_INTERFACE \

BUS_INTROSPECTABLE_INTERFACE \

"</node>\n"

#define INTERFACES_LIST \

BUS_UNIT_INTERFACES_LIST \

"org.freedesktop.systemd1.Service\0"

const char bus_service_interface[] _introspect_("Service") = BUS_SERVICE_INTERFACE;

const char bus_service_invalidating_properties[] =

"ExecStartPre\0"

"ExecStart\0"

"ExecStartPost\0"

"ExecReload\0"

"ExecStop\0"

"ExecStopPost\0"

"ExecMain\0"

"WatchdogTimestamp\0"

"WatchdogTimestampMonotonic\0"

"MainPID\0"

"ControlPID\0"

"StatusText\0"

"Result\0";

static DEFINE_BUS_PROPERTY_APPEND_ENUM(bus_service_append_type, service_type, ServiceType);

static DEFINE_BUS_PROPERTY_APPEND_ENUM(bus_service_append_restart, service_restart, ServiceRestart);

static DEFINE_BUS_PROPERTY_APPEND_ENUM(bus_service_append_notify_access, notify_access, NotifyAccess);

static DEFINE_BUS_PROPERTY_APPEND_ENUM(bus_service_append_service_result, service_result, ServiceResult);

static DEFINE_BUS_PROPERTY_APPEND_ENUM(bus_service_append_start_limit_action, start_limit_action, StartLimitAction);

static DEFINE_BUS_PROPERTY_SET_ENUM(bus_service_set_start_limit_action, start_limit_action, StartLimitAction);

static const BusProperty bus_exec_main_status_properties[] = {

{ "ExecMainStartTimestamp", bus_property_append_usec, "t", offsetof(ExecStatus, start_timestamp.realtime) },

{ "ExecMainStartTimestampMonotonic",bus_property_append_usec, "t", offsetof(ExecStatus, start_timestamp.monotonic) },

{ "ExecMainExitTimestamp", bus_property_append_usec, "t", offsetof(ExecStatus, exit_timestamp.realtime) },

{ "ExecMainExitTimestampMonotonic", bus_property_append_usec, "t", offsetof(ExecStatus, exit_timestamp.monotonic) },

{ "ExecMainPID", bus_property_append_pid, "u", offsetof(ExecStatus, pid) },

{ "ExecMainCode", bus_property_append_int, "i", offsetof(ExecStatus, code) },

{ "ExecMainStatus", bus_property_append_int, "i", offsetof(ExecStatus, status) },

{}

};

static const BusProperty bus_service_properties[] = {

{ "Type", bus_service_append_type, "s", offsetof(Service, type) },

{ "Restart", bus_service_append_restart, "s", offsetof(Service, restart) },

{ "PIDFile", bus_property_append_string, "s", offsetof(Service, pid_file), true },

{ "NotifyAccess", bus_service_append_notify_access, "s", offsetof(Service, notify_access) },

{ "RestartUSec", bus_property_append_usec, "t", offsetof(Service, restart_usec) },

{ "TimeoutStartUSec", bus_property_append_usec, "t", offsetof(Service, timeout_start_usec) },

{ "TimeoutStopUSec", bus_property_append_usec, "t", offsetof(Service, timeout_stop_usec) },

{ "WatchdogUSec", bus_property_append_usec, "t", offsetof(Service, watchdog_usec) },

{ "WatchdogTimestamp", bus_property_append_usec, "t", offsetof(Service, watchdog_timestamp.realtime) },

{ "WatchdogTimestampMonotonic",bus_property_append_usec, "t", offsetof(Service, watchdog_timestamp.monotonic) },

{ "StartLimitInterval", bus_property_append_usec, "t", offsetof(Service, start_limit.interval) },

{ "StartLimitBurst", bus_property_append_uint32, "u", offsetof(Service, start_limit.burst) },

{ "StartLimitAction", bus_service_append_start_limit_action,"s", offsetof(Service, start_limit_action), false, bus_service_set_start_limit_action},

BUS_EXEC_COMMAND_PROPERTY("ExecStartPre", offsetof(Service, exec_command[SERVICE_EXEC_START_PRE]), true ),

BUS_EXEC_COMMAND_PROPERTY("ExecStart", offsetof(Service, exec_command[SERVICE_EXEC_START]), true ),

BUS_EXEC_COMMAND_PROPERTY("ExecStartPost", offsetof(Service, exec_command[SERVICE_EXEC_START_POST]), true ),

BUS_EXEC_COMMAND_PROPERTY("ExecReload", offsetof(Service, exec_command[SERVICE_EXEC_RELOAD]), true ),

BUS_EXEC_COMMAND_PROPERTY("ExecStop", offsetof(Service, exec_command[SERVICE_EXEC_STOP]), true ),

BUS_EXEC_COMMAND_PROPERTY("ExecStopPost", offsetof(Service, exec_command[SERVICE_EXEC_STOP_POST]), true ),

{ "PermissionsStartOnly", bus_property_append_bool, "b", offsetof(Service, permissions_start_only) },

{ "RootDirectoryStartOnly", bus_property_append_bool, "b", offsetof(Service, root_directory_start_only) },

{ "RemainAfterExit", bus_property_append_bool, "b", offsetof(Service, remain_after_exit) },

{ "GuessMainPID", bus_property_append_bool, "b", offsetof(Service, guess_main_pid) },

{ "MainPID", bus_property_append_pid, "u", offsetof(Service, main_pid) },

{ "ControlPID", bus_property_append_pid, "u", offsetof(Service, control_pid) },

{ "BusName", bus_property_append_string, "s", offsetof(Service, bus_name), true },

{ "StatusText", bus_property_append_string, "s", offsetof(Service, status_text), true },

{ "Result", bus_service_append_service_result,"s", offsetof(Service, result) },

{}

};

DBusHandlerResult bus_service_message_handler(Unit *u, DBusConnection *connection, DBusMessage *message) {

Service *s = SERVICE(u);

const BusBoundProperties bps[] = {

{ "org.freedesktop.systemd1.Unit", bus_unit_properties, u },

{ "org.freedesktop.systemd1.Service", bus_unit_cgroup_properties, u },

{ "org.freedesktop.systemd1.Service", bus_service_properties, s },

{ "org.freedesktop.systemd1.Service", bus_exec_context_properties, &s->exec_context },

{ "org.freedesktop.systemd1.Service", bus_kill_context_properties, &s->kill_context },

{ "org.freedesktop.systemd1.Service", bus_cgroup_context_properties, &s->cgroup_context },

{ "org.freedesktop.systemd1.Service", bus_exec_main_status_properties, &s->main_exec_status },

{}

};

SELINUX_UNIT_ACCESS_CHECK(u, connection, message, "status");

return bus_default_message_handler(connection, message, INTROSPECTION, INTERFACES_LIST, bps);

}

static int bus_service_set_transient_property(

Service *s,

const char *name,

DBusMessageIter *i,

UnitSetPropertiesMode mode,

DBusError *error) {

int r;

assert(name);

assert(s);

assert(i);

if (streq(name, "RemainAfterExit")) {

if (dbus_message_iter_get_arg_type(i) != DBUS_TYPE_BOOLEAN)

return -EINVAL;

if (mode != UNIT_CHECK) {

dbus_bool_t b;

dbus_message_iter_get_basic(i, &b);

s->remain_after_exit = b;

unit_write_drop_in_private_format(UNIT(s), mode, name, "RemainAfterExit=%s\n", yes_no(b));

}

return 1;

} else if (streq(name, "ExecStart")) {

DBusMessageIter sub;

unsigned n = 0;

if (dbus_message_iter_get_arg_type(i) != DBUS_TYPE_ARRAY ||

dbus_message_iter_get_element_type(i) != DBUS_TYPE_STRUCT)

return -EINVAL;

dbus_message_iter_recurse(i, &sub);

while (dbus_message_iter_get_arg_type(&sub) == DBUS_TYPE_STRUCT) {

_cleanup_strv_free_ char **argv = NULL;

DBusMessageIter sub2;

dbus_bool_t ignore;

const char *path;

dbus_message_iter_recurse(&sub, &sub2);

if (bus_iter_get_basic_and_next(&sub2, DBUS_TYPE_STRING, &path, true) < 0)

return -EINVAL;

if (!path_is_absolute(path)) {

dbus_set_error(error, DBUS_ERROR_INVALID_ARGS, "Path %s is not absolute.", path);

return -EINVAL;

}

r = bus_parse_strv_iter(&sub2, &argv);

if (r < 0)

return r;

dbus_message_iter_next(&sub2);

if (bus_iter_get_basic_and_next(&sub2, DBUS_TYPE_BOOLEAN, &ignore, false) < 0)

return -EINVAL;

if (mode != UNIT_CHECK) {

ExecCommand *c;

c = new0(ExecCommand, 1);

if (!c)

return -ENOMEM;

c->path = strdup(path);

if (!c->path) {

free(c);

return -ENOMEM;

}

c->argv = argv;

argv = NULL;

c->ignore = ignore;

path_kill_slashes(c->path);

exec_command_append_list(&s->exec_command[SERVICE_EXEC_START], c);

}

n++;

dbus_message_iter_next(&sub);

}

if (mode != UNIT_CHECK) {

_cleanup_free_ char *buf = NULL;

_cleanup_fclose_ FILE *f = NULL;

ExecCommand *c;

size_t size = 0;

if (n == 0) {

exec_command_free_list(s->exec_command[SERVICE_EXEC_START]);

s->exec_command[SERVICE_EXEC_START] = NULL;

}

f = open_memstream(&buf, &size);

if (!f)

return -ENOMEM;

fputs("ExecStart=\n", f);

LIST_FOREACH(command, c, s->exec_command[SERVICE_EXEC_START]) {

char **a;

fputs("ExecStart=", f);

if (c->ignore)

fputc('-', f);

fputc('@', f);

fputs(c->path, f);

STRV_FOREACH(a, c->argv) {

fputc(' ', f);

fputs(*a, f);

}

fputc('\n', f);

}

fflush(f);

unit_write_drop_in_private(UNIT(s), mode, name, buf);

}

return 1;

}

return 0;

}

int bus_service_set_property(

Unit *u,

const char *name,

DBusMessageIter *i,

UnitSetPropertiesMode mode,

DBusError *error) {

Service *s = SERVICE(u);

int r;

assert(name);

assert(u);

assert(i);

r = bus_cgroup_set_property(u, &s->cgroup_context, name, i, mode, error);

if (r != 0)

return r;

if (u->transient && u->load_state == UNIT_STUB) {

/* This is a transient unit, let's load a little more */

r = bus_service_set_transient_property(s, name, i, mode, error);

if (r != 0)

return r;

r = bus_kill_context_set_transient_property(u, &s->kill_context, name, i, mode, error);

if (r != 0)

return r;

}

return 0;

}

int bus_service_commit_properties(Unit *u) {

assert(u);

unit_realize_cgroup(u);

return 0;

}