#include "unit.h"

#include "scope.h"

#include "dbus-unit.h"

#include "dbus-cgroup.h"

#include "dbus-kill.h"

#include "dbus-scope.h"

#include "bus-util.h"

#include "bus-internal.h"

static int bus_scope_abandon(sd_bus *bus, sd_bus_message *message, void *userdata, sd_bus_error *error) {

Scope *s = userdata;

assert(bus);

assert(message);

assert(s);

return scope_abandon(s);

}

static BUS_DEFINE_PROPERTY_GET_ENUM(property_get_result, scope_result, ScopeResult);

const sd_bus_vtable bus_scope_vtable[] = {

SD_BUS_VTABLE_START(0),

SD_BUS_PROPERTY("Controller", "s", NULL, offsetof(Scope, controller), SD_BUS_VTABLE_PROPERTY_CONST),

SD_BUS_PROPERTY("TimeoutStopUSec", "t", bus_property_get_usec, offsetof(Scope, timeout_stop_usec), SD_BUS_VTABLE_PROPERTY_CONST),

SD_BUS_PROPERTY("Result", "s", property_get_result, offsetof(Scope, result), SD_BUS_VTABLE_PROPERTY_EMITS_CHANGE),

SD_BUS_SIGNAL("RequestStop", NULL, 0),

SD_BUS_METHOD("Abandon", NULL, NULL, bus_scope_abandon, 0),

SD_BUS_VTABLE_END

};

static int bus_scope_set_transient_property(

Scope *s,

const char *name,

sd_bus_message *message,

UnitSetPropertiesMode mode,

sd_bus_error *error) {

int r;

assert(s);

assert(name);

assert(message);

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

unsigned n = 0;

uint32_t pid;

r = sd_bus_message_enter_container(message, 'a', "u");

if (r < 0)

return r;

while ((r = sd_bus_message_read(message, "u", &pid)) > 0) {

if (pid <= 1)

return -EINVAL;

if (mode != UNIT_CHECK) {

r = unit_watch_pid(UNIT(s), pid);

if (r < 0 && r != -EEXIST)

return r;

}

n++;

}

if (r < 0)

return r;

r = sd_bus_message_exit_container(message);

if (r < 0)

return r;

if (n <= 0)

return -EINVAL;

return 1;

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

const char *controller;

char *c;

r = sd_bus_message_read(message, "s", &controller);

if (r < 0)

return r;

if (!isempty(controller) && !service_name_is_valid(controller))

return sd_bus_error_setf(error, SD_BUS_ERROR_INVALID_ARGS, "Controller '%s' is not a valid bus name.", controller);

if (mode != UNIT_CHECK) {

if (isempty(controller))

c = NULL;

else {

c = strdup(controller);

if (!c)

return -ENOMEM;

}

free(s->controller);

s->controller = c;

}

return 1;

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

if (mode != UNIT_CHECK) {

r = sd_bus_message_read(message, "t", &s->timeout_stop_usec);

if (r < 0)

return r;

unit_write_drop_in_format(UNIT(s), mode, name, "[Scope]\nTimeoutStopSec=%lluus\n", (unsigned long long) s->timeout_stop_usec);

} else {

r = sd_bus_message_skip(message, "t");

if (r < 0)

return r;

}

return 1;

}

return 0;

}

int bus_scope_set_property(

Unit *u,

const char *name,

sd_bus_message *message,

UnitSetPropertiesMode mode,

sd_bus_error *error) {

Scope *s = SCOPE(u);

int r;

assert(s);

assert(name);

assert(message);

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

if (r != 0)

return r;

if (u->load_state == UNIT_STUB) {

/* While we are created we still accept PIDs */

r = bus_scope_set_transient_property(s, name, message, mode, error);

if (r != 0)

return r;

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

if (r != 0)

return r;

}

return 0;

}

int bus_scope_commit_properties(Unit *u) {

assert(u);

unit_realize_cgroup(u);

return 0;

}

int bus_scope_send_request_stop(Scope *s) {

_cleanup_bus_message_unref_ sd_bus_message *m = NULL;

_cleanup_free_ char *p = NULL;

int r;

assert(s);

if (!s->controller)

return 0;

p = unit_dbus_path(UNIT(s));

if (!p)

return -ENOMEM;

r = sd_bus_message_new_signal(

UNIT(s)->manager->api_bus,

p,

"org.freedesktop.systemd1.Scope",

"RequestStop",

&m);

if (r < 0)

return r;

return sd_bus_send_to(UNIT(s)->manager->api_bus, m, /* s->controller */ NULL, NULL);

}