e2417e4143bb892e4599b01de7b031763421bb64Daniel J Walsh

e2417e4143bb892e4599b01de7b031763421bb64Daniel J Walsh

e2417e4143bb892e4599b01de7b031763421bb64Daniel J Walsh#include "unit.h"

e2417e4143bb892e4599b01de7b031763421bb64Daniel J Walsh#include "scope.h"

718db96199eb307751264e4163555662c9a389faLennart Poettering#include "dbus-unit.h"

718db96199eb307751264e4163555662c9a389faLennart Poettering#include "dbus-cgroup.h"

718db96199eb307751264e4163555662c9a389faLennart Poettering#include "dbus-kill.h"

ffc227c9568eb0a12dc750761d1e5d68ea125aadLennart Poettering#include "dbus-scope.h"

ffc227c9568eb0a12dc750761d1e5d68ea125aadLennart Poettering#include "dbus.h"

ffc227c9568eb0a12dc750761d1e5d68ea125aadLennart Poettering#include "bus-util.h"

718db96199eb307751264e4163555662c9a389faLennart Poettering#include "bus-internal.h"

cad45ba11ec3572296361f53f5852ffb97a97fa3Lennart Poettering#include "bus-errors.h"

cad45ba11ec3572296361f53f5852ffb97a97fa3Lennart Poettering

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

718db96199eb307751264e4163555662c9a389faLennart Poettering Scope *s = userdata;

cad45ba11ec3572296361f53f5852ffb97a97fa3Lennart Poettering int r;

718db96199eb307751264e4163555662c9a389faLennart Poettering

718db96199eb307751264e4163555662c9a389faLennart Poettering assert(bus);

718db96199eb307751264e4163555662c9a389faLennart Poettering assert(message);

cad45ba11ec3572296361f53f5852ffb97a97fa3Lennart Poettering assert(s);

718db96199eb307751264e4163555662c9a389faLennart Poettering

cad45ba11ec3572296361f53f5852ffb97a97fa3Lennart Poettering r = bus_verify_manage_unit_async(UNIT(s)->manager, message, error);

718db96199eb307751264e4163555662c9a389faLennart Poettering if (r < 0)

cad45ba11ec3572296361f53f5852ffb97a97fa3Lennart Poettering return r;

cad45ba11ec3572296361f53f5852ffb97a97fa3Lennart Poettering if (r == 0)

718db96199eb307751264e4163555662c9a389faLennart Poettering return 1; /* No authorization for now, but the async polkit stuff will call us again when it has it */

cad45ba11ec3572296361f53f5852ffb97a97fa3Lennart Poettering

718db96199eb307751264e4163555662c9a389faLennart Poettering r = scope_abandon(s);

718db96199eb307751264e4163555662c9a389faLennart Poettering if (sd_bus_error_is_set(error))

718db96199eb307751264e4163555662c9a389faLennart Poettering return r;

ffc227c9568eb0a12dc750761d1e5d68ea125aadLennart Poettering

718db96199eb307751264e4163555662c9a389faLennart Poettering if (r == -ESTALE)

718db96199eb307751264e4163555662c9a389faLennart Poettering return sd_bus_error_setf(error, BUS_ERROR_SCOPE_NOT_RUNNING, "Scope %s is not running, cannot abandon.", UNIT(s)->id);

cad45ba11ec3572296361f53f5852ffb97a97fa3Lennart Poettering

718db96199eb307751264e4163555662c9a389faLennart Poettering return sd_bus_reply_method_return(message, NULL);

cad45ba11ec3572296361f53f5852ffb97a97fa3Lennart Poettering}

cad45ba11ec3572296361f53f5852ffb97a97fa3Lennart Poettering

cad45ba11ec3572296361f53f5852ffb97a97fa3Lennart Poetteringstatic BUS_DEFINE_PROPERTY_GET_ENUM(property_get_result, scope_result, ScopeResult);

cad45ba11ec3572296361f53f5852ffb97a97fa3Lennart Poettering

718db96199eb307751264e4163555662c9a389faLennart Poetteringconst sd_bus_vtable bus_scope_vtable[] = {

718db96199eb307751264e4163555662c9a389faLennart Poettering SD_BUS_VTABLE_START(0),

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

e2417e4143bb892e4599b01de7b031763421bb64Daniel J Walsh 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="USEC_FMT"us\n", 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_update_cgroup_members_masks(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,

&m,

p,

"org.freedesktop.systemd1.Scope",

"RequestStop");

if (r < 0)

return r;

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

}