#include "kdbus.h"
#include "sd-bus.h"
#include "bus-internal.h"
-
#include "sd-daemon.h"
#include "sd-event.h"
#include "event-util.h"
#include "sd-id128.h"
#include "async.h"
#include "hashmap.h"
+#include "def.h"
+#include "unit-name.h"
+#include "bus-control.h"
+#include "cgroup-util.h"
+
+#define CLIENTS_MAX 1024
+#define MATCHES_MAX 1024
+
+typedef struct Match Match;
+typedef struct Client Client;
+typedef struct Context Context;
+
+struct Match {
+ Client *client;
+ char *match;
+ uint64_t cookie;
+ LIST_FIELDS(Match, matches);
+};
+
+struct Client {
+ Context *context;
+ uint64_t id;
+ uint64_t next_cookie;
+ Hashmap *matches;
+ unsigned n_matches;
+ char *watch;
+};
-#define DBUS_PATH "/org/freedesktop/DBus"
-#define DBUS_INTERFACE "org.freedesktop.DBus"
+struct Context {
+ sd_bus *bus;
+ sd_event *event;
+ Hashmap *clients;
+};
-/*
- * TODO:
- *
- * AddMatch / RemoveMatch
- * ListActivatableNames
- * StartServiceByName
- */
+static void match_free(Match *m) {
-static sd_bus *driver_bus;
+ if (!m)
+ return;
-static int help(void) {
+ if (m->client) {
+ Match *first;
- printf("%s [OPTIONS...] <bus-path>\n\n"
- "Driver to provide a org.freedesktop.DBus interface on the given kdbus node.\n\n"
- " -h --help Show this help\n"
- " --version Show package version\n",
- program_invocation_short_name);
+ first = hashmap_get(m->client->matches, m->match);
+ LIST_REMOVE(matches, first, m);
+ if (first)
+ assert_se(hashmap_replace(m->client->matches, first->match, first) >= 0);
+ else
+ hashmap_remove(m->client->matches, m->match);
- return 0;
+ m->client->n_matches--;
+ }
+
+ free(m->match);
+ free(m);
}
-static int parse_argv(int argc, char *argv[]) {
+static int match_new(Client *c, struct bus_match_component *components, unsigned n_components, Match **_m) {
+ Match *m, *first;
+ int r;
- enum {
- ARG_VERSION = 0x100,
- };
+ assert(c);
+ assert(_m);
- static const struct option options[] = {
- { "help", no_argument, NULL, 'h' },
- { "version", no_argument, NULL, ARG_VERSION },
- {}
- };
+ r = hashmap_ensure_allocated(&c->matches, string_hash_func, string_compare_func);
+ if (r < 0)
+ return r;
- int c;
+ m = new0(Match, 1);
+ if (!m)
+ return -ENOMEM;
- assert(argc >= 0);
- assert(argv);
+ m->match = bus_match_to_string(components, n_components);
+ if (!m->match) {
+ r = -ENOMEM;
+ goto fail;
+ }
- while ((c = getopt_long(argc, argv, "h", options, NULL)) >= 0)
- switch (c) {
+ m->cookie = ++c->next_cookie;
- case 'h':
- help();
- return 0;
+ first = hashmap_get(c->matches, m->match);
+ LIST_PREPEND(matches, first, m);
+ r = hashmap_replace(c->matches, m->match, first);
+ if (r < 0) {
+ LIST_REMOVE(matches, first, m);
+ goto fail;
+ }
- case ARG_VERSION:
- puts(PACKAGE_STRING);
- puts(SYSTEMD_FEATURES);
- return 0;
+ m->client = c;
+ c->n_matches++;
- case '?':
- return -EINVAL;
+ *_m = m;
+ m = NULL;
- default:
- assert_not_reached("Unknown option");
- }
+ return 0;
- return 1;
+fail:
+ match_free(m);
+ return r;
}
-static int driver_name_info_error(sd_bus *bus, sd_bus_message *m, const char *name, int error_code) {
+static int on_name_owner_changed(sd_bus *bus, sd_bus_message *m, void *userdata, sd_bus_error *error);
- if (error_code == -ENXIO || error_code == -ENOENT)
- return sd_bus_reply_method_errorf(m, SD_BUS_ERROR_NAME_HAS_NO_OWNER,
- "Could not get owner of name '%s': no such name",
- name);
+static void client_free(Client *c) {
+ Match *m;
- return sd_bus_reply_method_errno(m, error_code, NULL);
+ if (!c)
+ return;
+
+ if (c->context) {
+ if (c->watch)
+ sd_bus_remove_match(c->context->bus, c->watch, on_name_owner_changed, c);
+
+ assert_se(hashmap_remove(c->context->clients, &c->id) == c);
+ }
+
+ while ((m = hashmap_first(c->matches)))
+ match_free(m);
+
+ hashmap_free(c->matches);
+ free(c->watch);
+
+ free(c);
}
-static int driver_add_match(sd_bus *bus, sd_bus_message *m, void *userdata, sd_bus_error *error) {
+static int on_name_owner_changed(sd_bus *bus, sd_bus_message *m, void *userdata, sd_bus_error *error) {
+ Client *c = userdata;
+
+ assert(bus);
+ assert(m);
+
+ client_free(c);
+ return 0;
+}
- char *arg0, *match;
+static int client_acquire(Context *context, uint64_t id, Client **_c) {
+ char *watch = NULL;
+ Client *c;
int r;
- r = sd_bus_message_read(m, "s", &arg0);
+ assert(context);
+ assert(_c);
+
+ c = hashmap_get(context->clients, &id);
+ if (c) {
+ *_c = c;
+ return 0;
+ }
+
+ if (hashmap_size(context->clients) >= CLIENTS_MAX)
+ return -ENOBUFS;
+
+ r = hashmap_ensure_allocated(&context->clients, uint64_hash_func, uint64_compare_func);
if (r < 0)
return r;
- match = strdup(arg0);
- if (!match)
+ c = new0(Client, 1);
+ if (!c)
return -ENOMEM;
- /* FIXME */
+ c->id = id;
- return sd_bus_reply_method_return(m, NULL);
+ r = hashmap_put(context->clients, &c->id, c);
+ if (r < 0)
+ goto fail;
+
+ c->context = context;
+
+ if (asprintf(&watch,
+ "type='signal',"
+ "sender='org.freedesktop.DBus',"
+ "path='/org/freedesktop/DBus',"
+ "interface='org.freedesktop.DBus',"
+ "member='NameOwnerChanged',"
+ "arg0=':1.%llu'", (unsigned long long) id) < 0) {
+ r = -ENOMEM;
+ goto fail;
+ }
+
+ r = sd_bus_add_match(context->bus, watch, on_name_owner_changed, c);
+ if (r < 0) {
+ free(watch);
+ goto fail;
+ }
+
+ c->watch = watch;
+
+ *_c = c;
+ return 0;
+
+fail:
+ client_free(c);
+ return r;
}
-static int driver_remove_match(sd_bus *bus, sd_bus_message *m, void *userdata, sd_bus_error *error) {
+static int driver_add_match(sd_bus *bus, sd_bus_message *message, void *userdata, sd_bus_error *error) {
+ struct bus_match_component *components = NULL;
+ Context *context = userdata;
+ unsigned n_components = 0;
+ Match *m = NULL;
+ Client *c = NULL;
char *arg0;
+ uint64_t id;
int r;
- r = sd_bus_message_read(m, "s", &arg0);
+ assert(bus);
+ assert(message);
+ assert(context);
+
+ r = sd_bus_message_read(message, "s", &arg0);
if (r < 0)
return r;
- /* FIXME */
+ r = bus_kernel_parse_unique_name(message->sender, &id);
+ if (r < 0)
+ return r;
- return sd_bus_reply_method_return(m, NULL);
+ r = client_acquire(context, id, &c);
+ if (r == -ENOBUFS)
+ return sd_bus_error_setf(error, SD_BUS_ERROR_LIMITS_EXCEEDED, "Reached limit of %u clients", CLIENTS_MAX);
+ if (r < 0)
+ return r;
+
+ if (c->n_matches >= MATCHES_MAX) {
+ r = sd_bus_error_setf(error, SD_BUS_ERROR_LIMITS_EXCEEDED, "Reached limit of %u matches per client", MATCHES_MAX);
+ goto fail;
+ }
+
+ r = bus_match_parse(arg0, &components, &n_components);
+ if (r < 0) {
+ r = sd_bus_error_setf(error, SD_BUS_ERROR_MATCH_RULE_INVALID, "Match rule \"%s\" is not valid", arg0);
+ goto fail;
+ }
+
+ r = match_new(c, components, n_components, &m);
+ if (r < 0)
+ goto fail;
+
+ r = bus_add_match_internal_kernel(bus, id, components, n_components, m->cookie);
+ if (r < 0)
+ goto fail;
+
+ bus_match_parse_free(components, n_components);
+
+ return sd_bus_reply_method_return(message, NULL);
+
+fail:
+ bus_match_parse_free(components, n_components);
+
+ match_free(m);
+
+ if (c->n_matches <= 0)
+ client_free(c);
+
+ return r;
}
-static int driver_get_security_ctx(sd_bus *bus, sd_bus_message *m, void *userdata, sd_bus_error *error) {
+static int driver_remove_match(sd_bus *bus, sd_bus_message *message, void *userdata, sd_bus_error *error) {
- _cleanup_bus_creds_unref_ sd_bus_creds *creds = NULL;
- _cleanup_bus_message_unref_ sd_bus_message *reply = NULL;
+ struct bus_match_component *components = NULL;
+ _cleanup_free_ char *normalized = NULL;
+ Context *context = userdata;
+ unsigned n_components = 0;
+ Client *c = NULL;
+ Match *m = NULL;
char *arg0;
+ uint64_t id;
int r;
- r = sd_bus_message_read(m, "s", &arg0);
+ assert(bus);
+ assert(message);
+ assert(context);
+
+ r = sd_bus_message_read(message, "s", &arg0);
if (r < 0)
return r;
- r = sd_bus_get_owner(bus, arg0, SD_BUS_CREDS_SELINUX_CONTEXT, &creds);
+ r = bus_kernel_parse_unique_name(message->sender, &id);
+ if (r < 0)
+ return r;
+
+ c = hashmap_get(context->clients, &id);
+ if (!c)
+ return sd_bus_error_setf(error, SD_BUS_ERROR_MATCH_RULE_NOT_FOUND, "You have not registered any matches.");
+
+ r = bus_match_parse(arg0, &components, &n_components);
if (r < 0) {
- if (r == -ENOENT)
- return driver_name_info_error(bus, m, arg0, r);
- else
- return r;
+ r = sd_bus_error_setf(error, SD_BUS_ERROR_MATCH_RULE_INVALID, "Match rule \"%s\" is not valid", arg0);
+ goto finish;
+ }
+
+ normalized = bus_match_to_string(components, n_components);
+ if (!normalized) {
+ r = -ENOMEM;
+ goto finish;
+ }
+
+ m = hashmap_get(c->matches, normalized);
+ if (!m) {
+ r = sd_bus_error_setf(error, SD_BUS_ERROR_MATCH_RULE_NOT_FOUND, "Match rule \"%s\" not found.", normalized);
+ goto finish;
}
+ bus_remove_match_internal_kernel(bus, id, m->cookie);
+ match_free(m);
+
+ r = sd_bus_reply_method_return(message, NULL);
+
+finish:
+ bus_match_parse_free(components, n_components);
+
+ if (c->n_matches <= 0)
+ client_free(c);
+
+ return r;
+}
+
+static int get_creds(sd_bus *bus, sd_bus_message *m, uint64_t mask, sd_bus_creds **_creds, sd_bus_error *error) {
+ _cleanup_bus_creds_unref_ sd_bus_creds *c = NULL;
+ const char *name;
+ int r;
+
+ assert(bus);
+ assert(m);
+ assert(_creds);
+
+ r = sd_bus_message_read(m, "s", &name);
+ if (r < 0)
+ return r;
+
+ assert_return(service_name_is_valid(name), -EINVAL);
+
+ r = sd_bus_get_owner(bus, name, mask, &c);
+ if (r == -ENOENT || r == -ENXIO)
+ return sd_bus_error_setf(error, SD_BUS_ERROR_NAME_HAS_NO_OWNER, "Name %s is currently not owned by anyone.", name);
+ if (r < 0)
+ return r;
+
+ if ((c->mask & mask) != mask)
+ return -ENOTSUP;
+
+ *_creds = c;
+ c = NULL;
+
+ return 0;
+}
+
+static int driver_get_security_context(sd_bus *bus, sd_bus_message *m, void *userdata, sd_bus_error *error) {
+ _cleanup_bus_message_unref_ sd_bus_message *reply = NULL;
+ _cleanup_bus_creds_unref_ sd_bus_creds *creds = NULL;
+ int r;
+
+ r = get_creds(bus, m, SD_BUS_CREDS_SELINUX_CONTEXT, &creds, error);
+ if (r < 0)
+ return r;
+
r = sd_bus_message_new_method_return(m, &reply);
if (r < 0)
return r;
}
static int driver_get_pid(sd_bus *bus, sd_bus_message *m, void *userdata, sd_bus_error *error) {
-
_cleanup_bus_creds_unref_ sd_bus_creds *creds = NULL;
- char *arg0;
int r;
- r = sd_bus_message_read(m, "s", &arg0);
+ r = get_creds(bus, m, SD_BUS_CREDS_PID, &creds, error);
if (r < 0)
return r;
- r = sd_bus_get_owner(bus, arg0, SD_BUS_CREDS_PID, &creds);
+ return sd_bus_reply_method_return(m, "u", (uint32_t) creds->pid);
+}
+
+static int driver_get_user(sd_bus *bus, sd_bus_message *m, void *userdata, sd_bus_error *error) {
+ _cleanup_bus_creds_unref_ sd_bus_creds *creds = NULL;
+ int r;
+
+ r = get_creds(bus, m, SD_BUS_CREDS_UID, &creds, error);
if (r < 0)
return r;
- return sd_bus_reply_method_return(m, "u", creds->pid);
+ return sd_bus_reply_method_return(m, "u", (uint32_t) creds->uid);
}
-static int driver_get_user(sd_bus *bus, sd_bus_message *m, void *userdata, sd_bus_error *error) {
-
+static int driver_get_name_owner(sd_bus *bus, sd_bus_message *m, void *userdata, sd_bus_error *error) {
_cleanup_bus_creds_unref_ sd_bus_creds *creds = NULL;
- char *arg0;
int r;
- r = sd_bus_message_read(m, "s", &arg0);
+ r = get_creds(bus, m, SD_BUS_CREDS_UNIQUE_NAME, &creds, error);
if (r < 0)
return r;
- r = sd_bus_get_owner(bus, arg0, SD_BUS_CREDS_UID, &creds);
- if (r < 0) {
- if (r == -ENOENT)
- return driver_name_info_error(bus, m, arg0, r);
- else
- return r;
- }
-
- return sd_bus_reply_method_return(m, "u", creds->uid);
+ return sd_bus_reply_method_return(m, "s", creds->unique_name);
}
static int driver_get_id(sd_bus *bus, sd_bus_message *m, void *userdata, sd_bus_error *error) {
-
sd_id128_t server_id;
char buf[SD_ID128_STRING_MAX];
int r;
return sd_bus_reply_method_return(m, "s", sd_id128_to_string(server_id, buf));
}
-static int driver_get_name_owner(sd_bus *bus, sd_bus_message *m, void *userdata, sd_bus_error *error) {
+static int driver_hello(sd_bus *bus, sd_bus_message *m, void *userdata, sd_bus_error *error) {
+ return sd_bus_reply_method_return(m, "s", m->sender);
+}
- _cleanup_bus_creds_unref_ sd_bus_creds *creds = NULL;
- _cleanup_free_ char *owner = NULL;
- char *arg0;
+static int return_strv(sd_bus *bus, sd_bus_message *m, char **l) {
+ _cleanup_bus_message_unref_ sd_bus_message *reply = NULL;
int r;
- r = sd_bus_message_read(m, "s", &arg0);
+ r = sd_bus_message_new_method_return(m, &reply);
if (r < 0)
return r;
- r = sd_bus_get_owner(bus, arg0, 0, &creds);
- if (r < 0) {
- if (r == -ENOENT)
- return driver_name_info_error(bus, m, arg0, r);
- else
- return r;
- }
-
- return sd_bus_reply_method_return(m, "s", creds->unique_name);
-}
-
-static int driver_hello(sd_bus *bus, sd_bus_message *m, void *userdata, sd_bus_error *error) {
+ r = sd_bus_message_append_strv(reply, l);
+ if (r < 0)
+ return r;
- return sd_bus_reply_method_return(m, "s", m->sender);
+ return sd_bus_send(bus, reply, NULL);
}
static int driver_list_names(sd_bus *bus, sd_bus_message *m, void *userdata, sd_bus_error *error) {
-
- _cleanup_bus_message_unref_ sd_bus_message *reply = NULL;
_cleanup_strv_free_ char **names = NULL;
int r;
if (r < 0)
return r;
- r = sd_bus_message_new_method_return(m, &reply);
- if (r < 0)
- return r;
+ /* Let's sort the names list to make it stable */
+ strv_sort(names);
- r = sd_bus_message_append_strv(reply, names);
- if (r < 0)
- return r;
-
- return sd_bus_send(bus, reply, NULL);
+ return return_strv(bus, m, names);
}
static int driver_list_activatable_names(sd_bus *bus, sd_bus_message *m, void *userdata, sd_bus_error *error) {
-
- _cleanup_bus_message_unref_ sd_bus_message *reply = NULL;
_cleanup_strv_free_ char **names = NULL;
int r;
if (r < 0)
return r;
- r = sd_bus_message_new_method_return(m, &reply);
- if (r < 0)
- return r;
+ /* Let's sort the names list to make it stable */
+ strv_sort(names);
- r = sd_bus_message_append_strv(reply, names);
- if (r < 0)
- return r;
-
- return sd_bus_send(bus, reply, NULL);
+ return return_strv(bus, m, names);
}
static int driver_list_queued_owners(sd_bus *bus, sd_bus_message *m, void *userdata, sd_bus_error *error) {
-
- _cleanup_bus_message_unref_ sd_bus_message *reply = NULL;
struct kdbus_cmd_name_list cmd = {};
struct kdbus_name_list *name_list;
struct kdbus_cmd_name *name;
if (r < 0)
return r;
+ assert_return(service_name_is_valid(arg0), -EINVAL);
+
cmd.flags = KDBUS_NAME_LIST_QUEUED;
r = ioctl(bus->input_fd, KDBUS_CMD_NAME_LIST, &cmd);
name_list = (struct kdbus_name_list *) ((uint8_t *) bus->kdbus_buffer + cmd.offset);
KDBUS_ITEM_FOREACH(name, name_list, names) {
- if (name->size > sizeof(*name) && !streq(name->name, arg0)) {
- char *n;
+ char *n;
+
+ if (name->size <= sizeof(*name))
+ continue;
- if (asprintf(&n, ":1.%llu", (unsigned long long) name->id) < 0)
- return -ENOMEM;
+ if (!streq(name->name, arg0))
+ continue;
- r = strv_push(&owners, n);
- if (r < 0) {
- free(n);
- return -ENOMEM;
- }
+ if (asprintf(&n, ":1.%llu", (unsigned long long) name->owner_id) < 0)
+ return -ENOMEM;
+
+ r = strv_push(&owners, n);
+ if (r < 0) {
+ free(n);
+ return -ENOMEM;
}
}
- r = ioctl(sd_bus_get_fd(bus), KDBUS_CMD_FREE, &cmd.offset);
+ r = ioctl(bus->input_fd, KDBUS_CMD_FREE, &cmd.offset);
if (r < 0)
return -errno;
- r = sd_bus_message_new_method_return(m, &reply);
- if (r < 0)
- return r;
-
- r = sd_bus_message_append_strv(reply, owners);
- if (r < 0)
- return r;
-
- return sd_bus_send(bus, reply, NULL);
+ return return_strv(bus, m, owners);
}
static int driver_name_has_owner(sd_bus *bus, sd_bus_message *m, void *userdata, sd_bus_error *error) {
-
- char *arg0;
+ const char *name;
int r;
- r = sd_bus_message_read(m, "s", &arg0);
+ r = sd_bus_message_read(m, "s", &name);
if (r < 0)
return r;
- r = sd_bus_get_owner(bus, arg0, 0, NULL);
- if (r < 0 && r != -ENOENT)
+ assert_return(service_name_is_valid(name), -EINVAL);
+
+ r = sd_bus_get_owner(bus, name, 0, NULL);
+ if (r < 0 && r != -ENOENT && r != -ENXIO)
return r;
- return sd_bus_reply_method_return(m, "b", r == 0);
+ return sd_bus_reply_method_return(m, "b", r >= 0);
}
static int driver_request_name(sd_bus *bus, sd_bus_message *m, void *userdata, sd_bus_error *error) {
-
- struct kdbus_cmd_name *cmd_name;
+ struct kdbus_cmd_name *n;
uint32_t flags;
- size_t size;
+ size_t size, l;
uint64_t id;
- char *name;
+ const char *name;
int r;
r = sd_bus_message_read(m, "su", &name, &flags);
if (r < 0)
return r;
- size = sizeof(*cmd_name) + strlen(name) + 1;
+ assert_return(service_name_is_valid(name), -EINVAL);
+ assert_return((flags & ~(BUS_NAME_ALLOW_REPLACEMENT|BUS_NAME_REPLACE_EXISTING|BUS_NAME_DO_NOT_QUEUE)) == 0, -EINVAL);
- cmd_name = alloca(size);
- memset(cmd_name, 0, size);
- strcpy(cmd_name->name, name);
- cmd_name->size = size;
- kdbus_translate_request_name_flags(flags, (uint64_t *) &cmd_name->conn_flags);
+ l = strlen(name);
+ size = offsetof(struct kdbus_cmd_name, name) + l + 1;
+ n = alloca0(size);
+ n->size = size;
+ memcpy(n->name, name, l+1);
+ kdbus_translate_request_name_flags(flags, (uint64_t *) &n->flags);
/* This function is open-coded because we request the name 'on behalf'
* of the requesting connection */
if (r < 0)
return r;
- cmd_name->id = id;
+ n->owner_id = id;
+
+ r = ioctl(bus->input_fd, KDBUS_CMD_NAME_ACQUIRE, n);
+ if (r < 0) {
+ if (errno == EEXIST)
+ return sd_bus_reply_method_return(m, "u", BUS_NAME_EXISTS);
+ if (errno == EALREADY)
+ return sd_bus_reply_method_return(m, "u", BUS_NAME_ALREADY_OWNER);
+
+ return -errno;
+ }
+
+ if (n->flags & KDBUS_NAME_IN_QUEUE)
+ return sd_bus_reply_method_return(m, "u", BUS_NAME_IN_QUEUE);
+
+ return sd_bus_reply_method_return(m, "u", BUS_NAME_PRIMARY_OWNER);
+}
+
+static int driver_release_name(sd_bus *bus, sd_bus_message *m, void *userdata, sd_bus_error *error) {
+ struct kdbus_cmd_name *n;
+ const char *name;
+ size_t l, size;
+ uint64_t id;
+ int r;
+
+ r = sd_bus_message_read(m, "s", &name);
+ if (r < 0)
+ return r;
+
+ assert_return(service_name_is_valid(name), -EINVAL);
+
+ l = strlen(name);
+ size = offsetof(struct kdbus_cmd_name, name) + l + 1;
+ n = alloca0(size);
+ n->size = size;
+ memcpy(n->name, name, l+1);
- r = ioctl(sd_bus_get_fd(bus), KDBUS_CMD_NAME_ACQUIRE, cmd_name);
+ /* This function is open-coded because we request the name 'on behalf'
+ * of the requesting connection */
+ r = bus_kernel_parse_unique_name(m->sender, &id);
if (r < 0)
return r;
- return sd_bus_reply_method_return(m, "u", 0);
+ n->owner_id = id;
+
+ r = ioctl(bus->input_fd, KDBUS_CMD_NAME_RELEASE, n);
+ if (r < 0) {
+ if (errno == ESRCH)
+ return sd_bus_reply_method_return(m, "u", BUS_NAME_NON_EXISTENT);
+ if (errno == EADDRINUSE)
+ return sd_bus_reply_method_return(m, "u", BUS_NAME_NOT_OWNER);
+ return -errno;
+ }
+
+ return sd_bus_reply_method_return(m, "u", BUS_NAME_RELEASED);
}
static int driver_start_service_by_name(sd_bus *bus, sd_bus_message *m, void *userdata, sd_bus_error *error) {
-
_cleanup_bus_message_unref_ sd_bus_message *reply = NULL;
- char *s;
+ _cleanup_strv_free_ char **t = NULL;
+ _cleanup_free_ char *path = NULL;
+ uint32_t flags;
+ char *name, *u;
int r;
- return sd_bus_reply_method_return(m, "u", 2);
+ r = sd_bus_message_read(m, "su", &name, &flags);
+ if (r < 0)
+ return r;
+
+ assert_return(service_name_is_valid(name), -EINVAL);
+ assert_return(flags == 0, -ENOTSUP);
+
+ r = sd_bus_get_owner(bus, name, 0, NULL);
+ if (r >= 0)
+ return sd_bus_reply_method_return(m, "u", BUS_START_REPLY_ALREADY_RUNNING);
+ if (r != -ENOENT)
+ return r;
+
+ u = strappenda(name, ".busname");
- r = sd_bus_message_read(m, "s", &s);
+ path = unit_dbus_path_from_name(u);
+ if (!path)
+ return -ENOMEM;
+
+ r = sd_bus_get_property_strv(
+ bus,
+ "org.freedesktop.systemd1",
+ path,
+ "org.freedesktop.systemd1.Unit",
+ "Triggers",
+ error,
+ &t);
if (r < 0)
return r;
- /* FIXME */
+ if (!t || !t[0] || t[1])
+ return sd_bus_error_setf(error, SD_BUS_ERROR_SERVICE_UNKNOWN, "Bus name %s not found.", name);
+
+ r = sd_bus_call_method(
+ bus,
+ "org.freedesktop.systemd1",
+ "/org/freedesktop/systemd1",
+ "org.freedesktop.systemd1.Manager",
+ "StartUnit",
+ error,
+ &reply,
+ "ss",
+ t[0],
+ "replace");
+ if (r < 0)
+ return r;
- return sd_bus_send(bus, reply, NULL);
+ return sd_bus_reply_method_return(m, "u", BUS_START_REPLY_SUCCESS);
}
-static int driver_update_env(sd_bus *bus, sd_bus_message *m, void *userdata, sd_bus_error *error) {
+static int driver_update_environment(sd_bus*bus, sd_bus_message *m, void *userdata, sd_bus_error *error) {
+ _cleanup_bus_message_unref_ sd_bus_message *msg = NULL;
+ _cleanup_strv_free_ char **args = NULL;
+ int r;
- return sd_bus_reply_method_errorf(m, SD_BUS_ERROR_NOT_SUPPORTED,
- "UpdateActivationEnvironment is unsupported");
-}
+ r = sd_bus_message_enter_container(m, SD_BUS_TYPE_ARRAY, "{ss}");
+ if (r < 0)
+ return r;
+
+ while ((r = sd_bus_message_enter_container(m, SD_BUS_TYPE_DICT_ENTRY, "ss")) > 0) {
+ _cleanup_free_ char *s = NULL;
+ const char *key;
+ const char *value;
+
+ r = sd_bus_message_read(m, "ss", &key, &value);
+ if (r < 0)
+ return r;
+
+ s = strjoin(key, "=", value, NULL);
+ if (!s)
+ return ENOMEM;
-static int driver_reload_config(sd_bus *bus, sd_bus_message *m, void *userdata, sd_bus_error *error) {
+ r = strv_extend(&args, s);
+ if (r < 0)
+ return r;
+
+ r = sd_bus_message_exit_container(m);
+ if (r < 0)
+ return r;
+ }
- return sd_bus_reply_method_errorf(m, SD_BUS_ERROR_NOT_SUPPORTED,
- "ReloadConfig is unsupported");
+ r = sd_bus_message_exit_container(m);
+ if (r < 0)
+ return r;
+
+ if (!args)
+ return -EINVAL;
+
+ r = sd_bus_message_new_method_call(
+ bus,
+ "org.freedesktop.systemd1",
+ "/org/freedesktop/systemd1",
+ "org.freedesktop.systemd1.Manager",
+ "SetEnvironment",
+ &msg);
+ if (r < 0)
+ return r;
+
+ r = sd_bus_message_append_strv(msg, args);
+ if (r < 0)
+ return r;
+
+ r = sd_bus_call(bus, msg, 0, NULL, NULL);
+ if (r < 0)
+ return r;
+
+ return sd_bus_reply_method_return(m, NULL);
+}
+
+static int driver_unsupported(sd_bus *bus, sd_bus_message *m, void *userdata, sd_bus_error *error) {
+ return sd_bus_error_setf(error, SD_BUS_ERROR_NOT_SUPPORTED, "%s() is not supported", sd_bus_message_get_member(m));
}
-const sd_bus_vtable dbus_vtable[] = {
+static const sd_bus_vtable driver_vtable[] = {
SD_BUS_VTABLE_START(0),
- SD_BUS_METHOD("AddMatch", "s", NULL, driver_add_match, 0),
- SD_BUS_METHOD("GetConnectionSELinuxSecurityContext", "s", "ay", driver_get_security_ctx, 0),
- SD_BUS_METHOD("GetConnectionUnixProcessID", "s", "u", driver_get_pid, 0),
- SD_BUS_METHOD("GetConnectionUnixUser", "s", "u", driver_get_user, 0),
- SD_BUS_METHOD("GetId", NULL, "s", driver_get_id, 0),
- SD_BUS_METHOD("GetNameOwner", "s", "s", driver_get_name_owner, 0),
- SD_BUS_METHOD("Hello", NULL, "s", driver_hello, 0),
- SD_BUS_METHOD("ListActivatableNames", NULL, "as", driver_list_activatable_names, 0),
- SD_BUS_METHOD("ListNames", NULL, "as", driver_list_names, 0),
- SD_BUS_METHOD("ListQueuedOwners", "s", "as", driver_list_queued_owners, 0),
- SD_BUS_METHOD("NameHasOwner", "s", "b", driver_name_has_owner, 0),
- SD_BUS_METHOD("ReloadConfig", NULL, NULL, driver_reload_config, 0),
- SD_BUS_METHOD("RemoveMatch", "s", NULL, driver_remove_match, 0),
- SD_BUS_METHOD("RequestName", "su", "u", driver_request_name, 0),
- SD_BUS_METHOD("StartServiceByName", "su", "u", driver_start_service_by_name, 0),
- SD_BUS_METHOD("UpdateActivationEnvironment", "a{ss}", NULL, driver_update_env, 0),
- SD_BUS_SIGNAL("NameAcquired", "s", 0),
- SD_BUS_SIGNAL("NameLost", "s", 0),
+ SD_BUS_METHOD("AddMatch", "s", NULL, driver_add_match, SD_BUS_VTABLE_UNPRIVILEGED),
+ SD_BUS_METHOD("GetConnectionSELinuxSecurityContext", "s", "ay", driver_get_security_context, SD_BUS_VTABLE_UNPRIVILEGED),
+ SD_BUS_METHOD("GetConnectionUnixProcessID", "s", "u", driver_get_pid, SD_BUS_VTABLE_UNPRIVILEGED),
+ SD_BUS_METHOD("GetConnectionUnixUser", "s", "u", driver_get_user, SD_BUS_VTABLE_UNPRIVILEGED),
+ SD_BUS_METHOD("GetId", NULL, "s", driver_get_id, SD_BUS_VTABLE_UNPRIVILEGED),
+ SD_BUS_METHOD("GetNameOwner", "s", "s", driver_get_name_owner, SD_BUS_VTABLE_UNPRIVILEGED),
+ SD_BUS_METHOD("Hello", NULL, "s", driver_hello, SD_BUS_VTABLE_UNPRIVILEGED),
+ SD_BUS_METHOD("ListActivatableNames", NULL, "as", driver_list_activatable_names, SD_BUS_VTABLE_UNPRIVILEGED),
+ SD_BUS_METHOD("ListNames", NULL, "as", driver_list_names, SD_BUS_VTABLE_UNPRIVILEGED),
+ SD_BUS_METHOD("ListQueuedOwners", "s", "as", driver_list_queued_owners, SD_BUS_VTABLE_UNPRIVILEGED),
+ SD_BUS_METHOD("NameHasOwner", "s", "b", driver_name_has_owner, SD_BUS_VTABLE_UNPRIVILEGED),
+ SD_BUS_METHOD("ReleaseName", "s", "u", driver_release_name, SD_BUS_VTABLE_UNPRIVILEGED),
+ SD_BUS_METHOD("ReloadConfig", NULL, NULL, driver_unsupported, SD_BUS_VTABLE_DEPRECATED),
+ SD_BUS_METHOD("RemoveMatch", "s", NULL, driver_remove_match, SD_BUS_VTABLE_UNPRIVILEGED),
+ SD_BUS_METHOD("RequestName", "su", "u", driver_request_name, SD_BUS_VTABLE_UNPRIVILEGED),
+ SD_BUS_METHOD("StartServiceByName", "su", "u", driver_start_service_by_name, SD_BUS_VTABLE_UNPRIVILEGED),
+ SD_BUS_METHOD("UpdateActivationEnvironment", "a{ss}", NULL, driver_update_environment, 0),
+ SD_BUS_SIGNAL("NameAcquired", "s", SD_BUS_VTABLE_DEPRECATED),
+ SD_BUS_SIGNAL("NameLost", "s", SD_BUS_VTABLE_DEPRECATED),
SD_BUS_SIGNAL("NameOwnerChanged", "sss", 0),
SD_BUS_VTABLE_END
};
-static int driver_main(const char *bus_name) {
+static int find_object(
+ sd_bus *bus,
+ const char *path,
+ const char *interface,
+ void *userdata,
+ void **ret_found,
+ sd_bus_error *ret_error) {
+
+ /* We support the driver interface on exactly two different
+ * paths: the root and the entry point object. This is a bit
+ * different from the original dbus-daemon which supported it
+ * on any path. */
- _cleanup_event_source_unref_ sd_event_source *w_accept = NULL;
- _cleanup_event_source_unref_ sd_event_source *w_root_service = NULL;
- _cleanup_event_unref_ sd_event *e = NULL;
+ if (streq_ptr(path, "/"))
+ return 1;
+
+ if (streq_ptr(path, "/org/freedesktop/DBus"))
+ return 1;
+
+ return 0;
+}
+
+static int node_enumerator(
+ sd_bus *bus,
+ const char *path,
+ void *userdata,
+ char ***ret_nodes,
+ sd_bus_error *ret_error) {
+
+ char **l;
+
+ l = strv_new("/", "/org/freedesktop/DBus", NULL);
+ if (!l)
+ return -ENOMEM;
+
+ *ret_nodes = l;
+ return 0;
+}
+
+static int connect_bus(Context *c) {
int r;
- r = sd_event_new(&e);
- if (r < 0) {
- log_error("Failed to allocate event loop: %s", strerror(-r));
- return r;
- }
+ assert(c);
- /* set up kernel bus connection */
- r = sd_bus_new(&driver_bus);
+ r = sd_bus_default(&c->bus);
if (r < 0) {
log_error("Failed to create bus: %s", strerror(-r));
return r;
}
- r = sd_bus_set_address(driver_bus, bus_name);
- if (r < 0) {
- log_error("Failed to create bus: %s", strerror(-r));
- return r;
+ if (!c->bus->is_kernel) {
+ log_error("Not running on kdbus");
+ return -EPERM;
}
- r = sd_bus_start(driver_bus);
+ r = sd_bus_add_fallback_vtable(c->bus, "/", "org.freedesktop.DBus", driver_vtable, find_object, c);
if (r < 0) {
- log_error("Failed to start kernel bus: %s", strerror(-r));
+ log_error("Failed to add manager object vtable: %s", strerror(-r));
return r;
}
- r = sd_bus_request_name(driver_bus, DBUS_INTERFACE, 0);
+ r = sd_bus_add_node_enumerator(c->bus, "/", node_enumerator, c);
if (r < 0) {
- log_error("Unable to request name '%s': %s\n", DBUS_INTERFACE, strerror(-r));
+ log_error("Failed to add node enumerator: %s", strerror(-r));
return r;
}
- r = sd_bus_add_object_vtable(driver_bus, DBUS_PATH, DBUS_INTERFACE, dbus_vtable, NULL);
+ r = sd_bus_request_name(c->bus, "org.freedesktop.DBus", 0);
if (r < 0) {
- log_error("Failed to add manager object vtable: %s", strerror(-r));
+ log_error("Unable to request name: %s", strerror(-r));
return r;
}
- r = sd_bus_attach_event(driver_bus, e, 0);
+ r = sd_bus_attach_event(c->bus, c->event, 0);
if (r < 0) {
- log_error("Error %d while adding bus to even: %s", r, strerror(-r));
+ log_error("Error while adding bus to event loop: %s", strerror(-r));
return r;
}
- log_debug("Entering main loop.");
+ return 0;
+}
- return sd_event_loop(e);
+static bool check_idle(void *userdata) {
+ Context *c = userdata;
+ assert(c);
+
+ return hashmap_isempty(c->clients);
}
int main(int argc, char *argv[]) {
- char *bus_name;
+ Context context = {};
+ Client *c;
int r;
- setlocale(LC_ALL, "");
+ log_set_target(LOG_TARGET_AUTO);
log_parse_environment();
log_open();
- r = parse_argv(argc, argv);
- if (r <= 0)
- return r;
+ if (argc != 1) {
+ log_error("This program takes no arguments.");
+ r = -EINVAL;
+ goto finish;
+ }
- if (argc <= optind) {
- help();
- return -EINVAL;
+ r = sd_event_default(&context.event);
+ if (r < 0) {
+ log_error("Failed to allocate event loop: %s", strerror(-r));
+ goto finish;
}
- r = asprintf(&bus_name, "kernel:path=%s", argv[optind]);
+ sd_event_set_watchdog(context.event, true);
+
+ r = connect_bus(&context);
if (r < 0)
- return r;
+ goto finish;
+
+ r = bus_event_loop_with_idle(context.event, context.bus, "org.freedesktop.DBus", DEFAULT_EXIT_USEC, check_idle, &context);
+ if (r < 0) {
+ log_error("Failed to run event loop: %s", strerror(-r));
+ goto finish;
+ }
+
+finish:
+ while ((c = hashmap_first(context.clients)))
+ client_free(c);
+
+ sd_bus_unref(context.bus);
+ sd_event_unref(context.event);
- return driver_main(bus_name);
+ return r < 0 ? EXIT_FAILURE : EXIT_SUCCESS;
}