| /* SPDX-License-Identifier: LGPL-2.1+ */ |
| |
| #include <errno.h> |
| #include <fcntl.h> |
| #include <inttypes.h> |
| #include <stdio.h> |
| #include <stdlib.h> |
| #include <string.h> |
| #include <sys/ioctl.h> |
| #include <sys/resource.h> |
| #include <sys/socket.h> |
| #include <unistd.h> |
| |
| #include "sd-bus-protocol.h" |
| #include "sd-bus.h" |
| #include "sd-daemon.h" |
| #include "sd-event.h" |
| #include "sd-id128.h" |
| |
| #include "alloc-util.h" |
| #include "bus-internal.h" |
| #include "bus-label.h" |
| #include "bus-message.h" |
| #include "bus-util.h" |
| #include "cap-list.h" |
| #include "cgroup-util.h" |
| #include "def.h" |
| #include "escape.h" |
| #include "fd-util.h" |
| #include "missing.h" |
| #include "mount-util.h" |
| #include "nsflags.h" |
| #include "parse-util.h" |
| #include "proc-cmdline.h" |
| #include "rlimit-util.h" |
| #include "stdio-util.h" |
| #include "strv.h" |
| #include "user-util.h" |
| |
| static int name_owner_change_callback(sd_bus_message *m, void *userdata, sd_bus_error *ret_error) { |
| sd_event *e = userdata; |
| |
| assert(m); |
| assert(e); |
| |
| sd_bus_close(sd_bus_message_get_bus(m)); |
| sd_event_exit(e, 0); |
| |
| return 1; |
| } |
| |
| int bus_async_unregister_and_exit(sd_event *e, sd_bus *bus, const char *name) { |
| const char *match; |
| const char *unique; |
| int r; |
| |
| assert(e); |
| assert(bus); |
| assert(name); |
| |
| /* We unregister the name here and then wait for the |
| * NameOwnerChanged signal for this event to arrive before we |
| * quit. We do this in order to make sure that any queued |
| * requests are still processed before we really exit. */ |
| |
| r = sd_bus_get_unique_name(bus, &unique); |
| if (r < 0) |
| return r; |
| |
| match = strjoina( |
| "sender='org.freedesktop.DBus'," |
| "type='signal'," |
| "interface='org.freedesktop.DBus'," |
| "member='NameOwnerChanged'," |
| "path='/org/freedesktop/DBus'," |
| "arg0='", name, "',", |
| "arg1='", unique, "',", |
| "arg2=''"); |
| |
| r = sd_bus_add_match_async(bus, NULL, match, name_owner_change_callback, NULL, e); |
| if (r < 0) |
| return r; |
| |
| r = sd_bus_release_name_async(bus, NULL, name, NULL, NULL); |
| if (r < 0) |
| return r; |
| |
| return 0; |
| } |
| |
| int bus_event_loop_with_idle( |
| sd_event *e, |
| sd_bus *bus, |
| const char *name, |
| usec_t timeout, |
| check_idle_t check_idle, |
| void *userdata) { |
| bool exiting = false; |
| int r, code; |
| |
| assert(e); |
| assert(bus); |
| assert(name); |
| |
| for (;;) { |
| bool idle; |
| |
| r = sd_event_get_state(e); |
| if (r < 0) |
| return r; |
| if (r == SD_EVENT_FINISHED) |
| break; |
| |
| if (check_idle) |
| idle = check_idle(userdata); |
| else |
| idle = true; |
| |
| r = sd_event_run(e, exiting || !idle ? (uint64_t) -1 : timeout); |
| if (r < 0) |
| return r; |
| |
| if (r == 0 && !exiting && idle) { |
| |
| r = sd_bus_try_close(bus); |
| if (r == -EBUSY) |
| continue; |
| |
| /* Fallback for dbus1 connections: we |
| * unregister the name and wait for the |
| * response to come through for it */ |
| if (r == -EOPNOTSUPP) { |
| |
| /* Inform the service manager that we |
| * are going down, so that it will |
| * queue all further start requests, |
| * instead of assuming we are already |
| * running. */ |
| sd_notify(false, "STOPPING=1"); |
| |
| r = bus_async_unregister_and_exit(e, bus, name); |
| if (r < 0) |
| return r; |
| |
| exiting = true; |
| continue; |
| } |
| |
| if (r < 0) |
| return r; |
| |
| sd_event_exit(e, 0); |
| break; |
| } |
| } |
| |
| r = sd_event_get_exit_code(e, &code); |
| if (r < 0) |
| return r; |
| |
| return code; |
| } |
| |
| int bus_name_has_owner(sd_bus *c, const char *name, sd_bus_error *error) { |
| _cleanup_(sd_bus_message_unrefp) sd_bus_message *rep = NULL; |
| int r, has_owner = 0; |
| |
| assert(c); |
| assert(name); |
| |
| r = sd_bus_call_method(c, |
| "org.freedesktop.DBus", |
| "/org/freedesktop/dbus", |
| "org.freedesktop.DBus", |
| "NameHasOwner", |
| error, |
| &rep, |
| "s", |
| name); |
| if (r < 0) |
| return r; |
| |
| r = sd_bus_message_read_basic(rep, 'b', &has_owner); |
| if (r < 0) |
| return sd_bus_error_set_errno(error, r); |
| |
| return has_owner; |
| } |
| |
| static int check_good_user(sd_bus_message *m, uid_t good_user) { |
| _cleanup_(sd_bus_creds_unrefp) sd_bus_creds *creds = NULL; |
| uid_t sender_uid; |
| int r; |
| |
| assert(m); |
| |
| if (good_user == UID_INVALID) |
| return 0; |
| |
| r = sd_bus_query_sender_creds(m, SD_BUS_CREDS_EUID, &creds); |
| if (r < 0) |
| return r; |
| |
| /* Don't trust augmented credentials for authorization */ |
| assert_return((sd_bus_creds_get_augmented_mask(creds) & SD_BUS_CREDS_EUID) == 0, -EPERM); |
| |
| r = sd_bus_creds_get_euid(creds, &sender_uid); |
| if (r < 0) |
| return r; |
| |
| return sender_uid == good_user; |
| } |
| |
| int bus_test_polkit( |
| sd_bus_message *call, |
| int capability, |
| const char *action, |
| const char **details, |
| uid_t good_user, |
| bool *_challenge, |
| sd_bus_error *e) { |
| |
| int r; |
| |
| assert(call); |
| assert(action); |
| |
| /* Tests non-interactively! */ |
| |
| r = check_good_user(call, good_user); |
| if (r != 0) |
| return r; |
| |
| r = sd_bus_query_sender_privilege(call, capability); |
| if (r < 0) |
| return r; |
| else if (r > 0) |
| return 1; |
| #if ENABLE_POLKIT |
| else { |
| _cleanup_(sd_bus_message_unrefp) sd_bus_message *request = NULL; |
| _cleanup_(sd_bus_message_unrefp) sd_bus_message *reply = NULL; |
| int authorized = false, challenge = false; |
| const char *sender, **k, **v; |
| |
| sender = sd_bus_message_get_sender(call); |
| if (!sender) |
| return -EBADMSG; |
| |
| r = sd_bus_message_new_method_call( |
| call->bus, |
| &request, |
| "org.freedesktop.PolicyKit1", |
| "/org/freedesktop/PolicyKit1/Authority", |
| "org.freedesktop.PolicyKit1.Authority", |
| "CheckAuthorization"); |
| if (r < 0) |
| return r; |
| |
| r = sd_bus_message_append( |
| request, |
| "(sa{sv})s", |
| "system-bus-name", 1, "name", "s", sender, |
| action); |
| if (r < 0) |
| return r; |
| |
| r = sd_bus_message_open_container(request, 'a', "{ss}"); |
| if (r < 0) |
| return r; |
| |
| STRV_FOREACH_PAIR(k, v, details) { |
| r = sd_bus_message_append(request, "{ss}", *k, *v); |
| if (r < 0) |
| return r; |
| } |
| |
| r = sd_bus_message_close_container(request); |
| if (r < 0) |
| return r; |
| |
| r = sd_bus_message_append(request, "us", 0, NULL); |
| if (r < 0) |
| return r; |
| |
| r = sd_bus_call(call->bus, request, 0, e, &reply); |
| if (r < 0) { |
| /* Treat no PK available as access denied */ |
| if (sd_bus_error_has_name(e, SD_BUS_ERROR_SERVICE_UNKNOWN)) { |
| sd_bus_error_free(e); |
| return -EACCES; |
| } |
| |
| return r; |
| } |
| |
| r = sd_bus_message_enter_container(reply, 'r', "bba{ss}"); |
| if (r < 0) |
| return r; |
| |
| r = sd_bus_message_read(reply, "bb", &authorized, &challenge); |
| if (r < 0) |
| return r; |
| |
| if (authorized) |
| return 1; |
| |
| if (_challenge) { |
| *_challenge = challenge; |
| return 0; |
| } |
| } |
| #endif |
| |
| return -EACCES; |
| } |
| |
| #if ENABLE_POLKIT |
| |
| typedef struct AsyncPolkitQuery { |
| sd_bus_message *request, *reply; |
| sd_bus_message_handler_t callback; |
| void *userdata; |
| sd_bus_slot *slot; |
| Hashmap *registry; |
| } AsyncPolkitQuery; |
| |
| static void async_polkit_query_free(AsyncPolkitQuery *q) { |
| |
| if (!q) |
| return; |
| |
| sd_bus_slot_unref(q->slot); |
| |
| if (q->registry && q->request) |
| hashmap_remove(q->registry, q->request); |
| |
| sd_bus_message_unref(q->request); |
| sd_bus_message_unref(q->reply); |
| |
| free(q); |
| } |
| |
| static int async_polkit_callback(sd_bus_message *reply, void *userdata, sd_bus_error *error) { |
| _cleanup_(sd_bus_error_free) sd_bus_error error_buffer = SD_BUS_ERROR_NULL; |
| AsyncPolkitQuery *q = userdata; |
| int r; |
| |
| assert(reply); |
| assert(q); |
| |
| q->slot = sd_bus_slot_unref(q->slot); |
| q->reply = sd_bus_message_ref(reply); |
| |
| r = sd_bus_message_rewind(q->request, true); |
| if (r < 0) { |
| r = sd_bus_reply_method_errno(q->request, r, NULL); |
| goto finish; |
| } |
| |
| r = q->callback(q->request, q->userdata, &error_buffer); |
| r = bus_maybe_reply_error(q->request, r, &error_buffer); |
| |
| finish: |
| async_polkit_query_free(q); |
| |
| return r; |
| } |
| |
| #endif |
| |
| int bus_verify_polkit_async( |
| sd_bus_message *call, |
| int capability, |
| const char *action, |
| const char **details, |
| bool interactive, |
| uid_t good_user, |
| Hashmap **registry, |
| sd_bus_error *error) { |
| |
| #if ENABLE_POLKIT |
| _cleanup_(sd_bus_message_unrefp) sd_bus_message *pk = NULL; |
| AsyncPolkitQuery *q; |
| const char *sender, **k, **v; |
| sd_bus_message_handler_t callback; |
| void *userdata; |
| int c; |
| #endif |
| int r; |
| |
| assert(call); |
| assert(action); |
| assert(registry); |
| |
| r = check_good_user(call, good_user); |
| if (r != 0) |
| return r; |
| |
| #if ENABLE_POLKIT |
| q = hashmap_get(*registry, call); |
| if (q) { |
| int authorized, challenge; |
| |
| /* This is the second invocation of this function, and |
| * there's already a response from polkit, let's |
| * process it */ |
| assert(q->reply); |
| |
| if (sd_bus_message_is_method_error(q->reply, NULL)) { |
| const sd_bus_error *e; |
| |
| /* Copy error from polkit reply */ |
| e = sd_bus_message_get_error(q->reply); |
| sd_bus_error_copy(error, e); |
| |
| /* Treat no PK available as access denied */ |
| if (sd_bus_error_has_name(e, SD_BUS_ERROR_SERVICE_UNKNOWN)) |
| return -EACCES; |
| |
| return -sd_bus_error_get_errno(e); |
| } |
| |
| r = sd_bus_message_enter_container(q->reply, 'r', "bba{ss}"); |
| if (r >= 0) |
| r = sd_bus_message_read(q->reply, "bb", &authorized, &challenge); |
| |
| if (r < 0) |
| return r; |
| |
| if (authorized) |
| return 1; |
| |
| if (challenge) |
| return sd_bus_error_set(error, SD_BUS_ERROR_INTERACTIVE_AUTHORIZATION_REQUIRED, "Interactive authentication required."); |
| |
| return -EACCES; |
| } |
| #endif |
| |
| r = sd_bus_query_sender_privilege(call, capability); |
| if (r < 0) |
| return r; |
| else if (r > 0) |
| return 1; |
| |
| #if ENABLE_POLKIT |
| if (sd_bus_get_current_message(call->bus) != call) |
| return -EINVAL; |
| |
| callback = sd_bus_get_current_handler(call->bus); |
| if (!callback) |
| return -EINVAL; |
| |
| userdata = sd_bus_get_current_userdata(call->bus); |
| |
| sender = sd_bus_message_get_sender(call); |
| if (!sender) |
| return -EBADMSG; |
| |
| c = sd_bus_message_get_allow_interactive_authorization(call); |
| if (c < 0) |
| return c; |
| if (c > 0) |
| interactive = true; |
| |
| r = hashmap_ensure_allocated(registry, NULL); |
| if (r < 0) |
| return r; |
| |
| r = sd_bus_message_new_method_call( |
| call->bus, |
| &pk, |
| "org.freedesktop.PolicyKit1", |
| "/org/freedesktop/PolicyKit1/Authority", |
| "org.freedesktop.PolicyKit1.Authority", |
| "CheckAuthorization"); |
| if (r < 0) |
| return r; |
| |
| r = sd_bus_message_append( |
| pk, |
| "(sa{sv})s", |
| "system-bus-name", 1, "name", "s", sender, |
| action); |
| if (r < 0) |
| return r; |
| |
| r = sd_bus_message_open_container(pk, 'a', "{ss}"); |
| if (r < 0) |
| return r; |
| |
| STRV_FOREACH_PAIR(k, v, details) { |
| r = sd_bus_message_append(pk, "{ss}", *k, *v); |
| if (r < 0) |
| return r; |
| } |
| |
| r = sd_bus_message_close_container(pk); |
| if (r < 0) |
| return r; |
| |
| r = sd_bus_message_append(pk, "us", !!interactive, NULL); |
| if (r < 0) |
| return r; |
| |
| q = new0(AsyncPolkitQuery, 1); |
| if (!q) |
| return -ENOMEM; |
| |
| q->request = sd_bus_message_ref(call); |
| q->callback = callback; |
| q->userdata = userdata; |
| |
| r = hashmap_put(*registry, call, q); |
| if (r < 0) { |
| async_polkit_query_free(q); |
| return r; |
| } |
| |
| q->registry = *registry; |
| |
| r = sd_bus_call_async(call->bus, &q->slot, pk, async_polkit_callback, q, 0); |
| if (r < 0) { |
| async_polkit_query_free(q); |
| return r; |
| } |
| |
| return 0; |
| #endif |
| |
| return -EACCES; |
| } |
| |
| void bus_verify_polkit_async_registry_free(Hashmap *registry) { |
| #if ENABLE_POLKIT |
| hashmap_free_with_destructor(registry, async_polkit_query_free); |
| #endif |
| } |
| |
| int bus_check_peercred(sd_bus *c) { |
| struct ucred ucred; |
| int fd, r; |
| |
| assert(c); |
| |
| fd = sd_bus_get_fd(c); |
| if (fd < 0) |
| return fd; |
| |
| r = getpeercred(fd, &ucred); |
| if (r < 0) |
| return r; |
| |
| if (ucred.uid != 0 && ucred.uid != geteuid()) |
| return -EPERM; |
| |
| return 1; |
| } |
| |
| int bus_connect_system_systemd(sd_bus **_bus) { |
| _cleanup_(sd_bus_unrefp) sd_bus *bus = NULL; |
| int r; |
| |
| assert(_bus); |
| |
| if (geteuid() != 0) |
| return sd_bus_default_system(_bus); |
| |
| /* If we are root then let's talk directly to the system |
| * instance, instead of going via the bus */ |
| |
| r = sd_bus_new(&bus); |
| if (r < 0) |
| return r; |
| |
| r = sd_bus_set_address(bus, "unix:path=/run/systemd/private"); |
| if (r < 0) |
| return r; |
| |
| r = sd_bus_start(bus); |
| if (r < 0) |
| return sd_bus_default_system(_bus); |
| |
| r = bus_check_peercred(bus); |
| if (r < 0) |
| return r; |
| |
| *_bus = TAKE_PTR(bus); |
| |
| return 0; |
| } |
| |
| int bus_connect_user_systemd(sd_bus **_bus) { |
| _cleanup_(sd_bus_unrefp) sd_bus *bus = NULL; |
| _cleanup_free_ char *ee = NULL; |
| const char *e; |
| int r; |
| |
| assert(_bus); |
| |
| e = secure_getenv("XDG_RUNTIME_DIR"); |
| if (!e) |
| return sd_bus_default_user(_bus); |
| |
| ee = bus_address_escape(e); |
| if (!ee) |
| return -ENOMEM; |
| |
| r = sd_bus_new(&bus); |
| if (r < 0) |
| return r; |
| |
| bus->address = strjoin("unix:path=", ee, "/systemd/private"); |
| if (!bus->address) |
| return -ENOMEM; |
| |
| r = sd_bus_start(bus); |
| if (r < 0) |
| return sd_bus_default_user(_bus); |
| |
| r = bus_check_peercred(bus); |
| if (r < 0) |
| return r; |
| |
| *_bus = TAKE_PTR(bus); |
| |
| return 0; |
| } |
| |
| #define print_property(name, fmt, ...) \ |
| do { \ |
| if (value) \ |
| printf(fmt "\n", __VA_ARGS__); \ |
| else \ |
| printf("%s=" fmt "\n", name, __VA_ARGS__); \ |
| } while (0) |
| |
| int bus_print_property(const char *name, sd_bus_message *m, bool value, bool all) { |
| char type; |
| const char *contents; |
| int r; |
| |
| assert(name); |
| assert(m); |
| |
| r = sd_bus_message_peek_type(m, &type, &contents); |
| if (r < 0) |
| return r; |
| |
| switch (type) { |
| |
| case SD_BUS_TYPE_STRING: { |
| const char *s; |
| |
| r = sd_bus_message_read_basic(m, type, &s); |
| if (r < 0) |
| return r; |
| |
| if (all || !isempty(s)) { |
| bool good; |
| |
| /* This property has a single value, so we need to take |
| * care not to print a new line, everything else is OK. */ |
| good = !strchr(s, '\n'); |
| print_property(name, "%s", good ? s : "[unprintable]"); |
| } |
| |
| return 1; |
| } |
| |
| case SD_BUS_TYPE_BOOLEAN: { |
| int b; |
| |
| r = sd_bus_message_read_basic(m, type, &b); |
| if (r < 0) |
| return r; |
| |
| print_property(name, "%s", yes_no(b)); |
| |
| return 1; |
| } |
| |
| case SD_BUS_TYPE_UINT64: { |
| uint64_t u; |
| |
| r = sd_bus_message_read_basic(m, type, &u); |
| if (r < 0) |
| return r; |
| |
| /* Yes, heuristics! But we can change this check |
| * should it turn out to not be sufficient */ |
| |
| if (endswith(name, "Timestamp") || |
| STR_IN_SET(name, "NextElapseUSecRealtime", "LastTriggerUSec", "TimeUSec", "RTCTimeUSec")) { |
| char timestamp[FORMAT_TIMESTAMP_MAX]; |
| const char *t; |
| |
| t = format_timestamp(timestamp, sizeof(timestamp), u); |
| if (t || all) |
| print_property(name, "%s", strempty(t)); |
| |
| } else if (strstr(name, "USec")) { |
| char timespan[FORMAT_TIMESPAN_MAX]; |
| |
| print_property(name, "%s", format_timespan(timespan, sizeof(timespan), u, 0)); |
| } else if (streq(name, "RestrictNamespaces")) { |
| _cleanup_free_ char *s = NULL; |
| const char *result; |
| |
| if ((u & NAMESPACE_FLAGS_ALL) == 0) |
| result = "yes"; |
| else if ((u & NAMESPACE_FLAGS_ALL) == NAMESPACE_FLAGS_ALL) |
| result = "no"; |
| else { |
| r = namespace_flags_to_string(u, &s); |
| if (r < 0) |
| return r; |
| |
| result = s; |
| } |
| |
| print_property(name, "%s", result); |
| |
| } else if (streq(name, "MountFlags")) { |
| const char *result; |
| |
| result = mount_propagation_flags_to_string(u); |
| if (!result) |
| return -EINVAL; |
| |
| print_property(name, "%s", result); |
| |
| } else if (STR_IN_SET(name, "CapabilityBoundingSet", "AmbientCapabilities")) { |
| _cleanup_free_ char *s = NULL; |
| |
| r = capability_set_to_string_alloc(u, &s); |
| if (r < 0) |
| return r; |
| |
| print_property(name, "%s", s); |
| |
| } else if ((STR_IN_SET(name, "CPUWeight", "StartupCPUWeight", "IOWeight", "StartupIOWeight") && u == CGROUP_WEIGHT_INVALID) || |
| (STR_IN_SET(name, "CPUShares", "StartupCPUShares") && u == CGROUP_CPU_SHARES_INVALID) || |
| (STR_IN_SET(name, "BlockIOWeight", "StartupBlockIOWeight") && u == CGROUP_BLKIO_WEIGHT_INVALID) || |
| (STR_IN_SET(name, "MemoryCurrent", "TasksCurrent") && u == (uint64_t) -1) || |
| (endswith(name, "NSec") && u == (uint64_t) -1)) |
| |
| print_property(name, "%s", "[not set]"); |
| |
| else if ((STR_IN_SET(name, "MemoryLow", "MemoryHigh", "MemoryMax", "MemorySwapMax", "MemoryLimit") && u == CGROUP_LIMIT_MAX) || |
| (STR_IN_SET(name, "TasksMax", "DefaultTasksMax") && u == (uint64_t) -1) || |
| (startswith(name, "Limit") && u == (uint64_t) -1) || |
| (startswith(name, "DefaultLimit") && u == (uint64_t) -1)) |
| |
| print_property(name, "%s", "infinity"); |
| else |
| print_property(name, "%"PRIu64, u); |
| |
| return 1; |
| } |
| |
| case SD_BUS_TYPE_INT64: { |
| int64_t i; |
| |
| r = sd_bus_message_read_basic(m, type, &i); |
| if (r < 0) |
| return r; |
| |
| print_property(name, "%"PRIi64, i); |
| |
| return 1; |
| } |
| |
| case SD_BUS_TYPE_UINT32: { |
| uint32_t u; |
| |
| r = sd_bus_message_read_basic(m, type, &u); |
| if (r < 0) |
| return r; |
| |
| if (strstr(name, "UMask") || strstr(name, "Mode")) |
| print_property(name, "%04o", u); |
| else if (streq(name, "UID")) { |
| if (u == UID_INVALID) |
| print_property(name, "%s", "[not set]"); |
| else |
| print_property(name, "%"PRIu32, u); |
| } else if (streq(name, "GID")) { |
| if (u == GID_INVALID) |
| print_property(name, "%s", "[not set]"); |
| else |
| print_property(name, "%"PRIu32, u); |
| } else |
| print_property(name, "%"PRIu32, u); |
| |
| return 1; |
| } |
| |
| case SD_BUS_TYPE_INT32: { |
| int32_t i; |
| |
| r = sd_bus_message_read_basic(m, type, &i); |
| if (r < 0) |
| return r; |
| |
| print_property(name, "%"PRIi32, i); |
| return 1; |
| } |
| |
| case SD_BUS_TYPE_DOUBLE: { |
| double d; |
| |
| r = sd_bus_message_read_basic(m, type, &d); |
| if (r < 0) |
| return r; |
| |
| print_property(name, "%g", d); |
| return 1; |
| } |
| |
| case SD_BUS_TYPE_ARRAY: |
| if (streq(contents, "s")) { |
| bool first = true; |
| const char *str; |
| |
| r = sd_bus_message_enter_container(m, SD_BUS_TYPE_ARRAY, contents); |
| if (r < 0) |
| return r; |
| |
| while ((r = sd_bus_message_read_basic(m, SD_BUS_TYPE_STRING, &str)) > 0) { |
| bool good; |
| |
| if (first && !value) |
| printf("%s=", name); |
| |
| /* This property has multiple space-separated values, so |
| * neither spaces nor newlines can be allowed in a value. */ |
| good = str[strcspn(str, " \n")] == '\0'; |
| |
| printf("%s%s", first ? "" : " ", good ? str : "[unprintable]"); |
| |
| first = false; |
| } |
| if (r < 0) |
| return r; |
| |
| if (first && all && !value) |
| printf("%s=", name); |
| if (!first || all) |
| puts(""); |
| |
| r = sd_bus_message_exit_container(m); |
| if (r < 0) |
| return r; |
| |
| return 1; |
| |
| } else if (streq(contents, "y")) { |
| const uint8_t *u; |
| size_t n; |
| |
| r = sd_bus_message_read_array(m, SD_BUS_TYPE_BYTE, (const void**) &u, &n); |
| if (r < 0) |
| return r; |
| |
| if (all || n > 0) { |
| unsigned int i; |
| |
| if (!value) |
| printf("%s=", name); |
| |
| for (i = 0; i < n; i++) |
| printf("%02x", u[i]); |
| |
| puts(""); |
| } |
| |
| return 1; |
| |
| } else if (streq(contents, "u")) { |
| uint32_t *u; |
| size_t n; |
| |
| r = sd_bus_message_read_array(m, SD_BUS_TYPE_UINT32, (const void**) &u, &n); |
| if (r < 0) |
| return r; |
| |
| if (all || n > 0) { |
| unsigned int i; |
| |
| if (!value) |
| printf("%s=", name); |
| |
| for (i = 0; i < n; i++) |
| printf("%08x", u[i]); |
| |
| puts(""); |
| } |
| |
| return 1; |
| } |
| |
| break; |
| } |
| |
| return 0; |
| } |
| |
| int bus_message_print_all_properties( |
| sd_bus_message *m, |
| bus_message_print_t func, |
| char **filter, |
| bool value, |
| bool all, |
| Set **found_properties) { |
| |
| int r; |
| |
| assert(m); |
| |
| r = sd_bus_message_enter_container(m, SD_BUS_TYPE_ARRAY, "{sv}"); |
| if (r < 0) |
| return r; |
| |
| while ((r = sd_bus_message_enter_container(m, SD_BUS_TYPE_DICT_ENTRY, "sv")) > 0) { |
| const char *name; |
| const char *contents; |
| |
| r = sd_bus_message_read_basic(m, SD_BUS_TYPE_STRING, &name); |
| if (r < 0) |
| return r; |
| |
| if (found_properties) { |
| r = set_ensure_allocated(found_properties, &string_hash_ops); |
| if (r < 0) |
| return log_oom(); |
| |
| r = set_put(*found_properties, name); |
| if (r < 0 && r != EEXIST) |
| return log_oom(); |
| } |
| |
| if (!filter || strv_find(filter, name)) { |
| r = sd_bus_message_peek_type(m, NULL, &contents); |
| if (r < 0) |
| return r; |
| |
| r = sd_bus_message_enter_container(m, SD_BUS_TYPE_VARIANT, contents); |
| if (r < 0) |
| return r; |
| |
| if (func) |
| r = func(name, m, value, all); |
| if (!func || r == 0) |
| r = bus_print_property(name, m, value, all); |
| if (r < 0) |
| return r; |
| if (r == 0) { |
| if (all) |
| printf("%s=[unprintable]\n", name); |
| /* skip what we didn't read */ |
| r = sd_bus_message_skip(m, contents); |
| if (r < 0) |
| return r; |
| } |
| |
| r = sd_bus_message_exit_container(m); |
| if (r < 0) |
| return r; |
| } else { |
| r = sd_bus_message_skip(m, "v"); |
| if (r < 0) |
| return r; |
| } |
| |
| r = sd_bus_message_exit_container(m); |
| if (r < 0) |
| return r; |
| } |
| if (r < 0) |
| return r; |
| |
| r = sd_bus_message_exit_container(m); |
| if (r < 0) |
| return r; |
| |
| return 0; |
| } |
| |
| int bus_print_all_properties( |
| sd_bus *bus, |
| const char *dest, |
| const char *path, |
| bus_message_print_t func, |
| char **filter, |
| bool value, |
| bool all, |
| Set **found_properties) { |
| |
| _cleanup_(sd_bus_message_unrefp) sd_bus_message *reply = NULL; |
| _cleanup_(sd_bus_error_free) sd_bus_error error = SD_BUS_ERROR_NULL; |
| int r; |
| |
| assert(bus); |
| assert(path); |
| |
| r = sd_bus_call_method(bus, |
| dest, |
| path, |
| "org.freedesktop.DBus.Properties", |
| "GetAll", |
| &error, |
| &reply, |
| "s", ""); |
| if (r < 0) |
| return r; |
| |
| return bus_message_print_all_properties(reply, func, filter, value, all, found_properties); |
| } |
| |
| int bus_map_id128(sd_bus *bus, const char *member, sd_bus_message *m, sd_bus_error *error, void *userdata) { |
| sd_id128_t *p = userdata; |
| const void *v; |
| size_t n; |
| int r; |
| |
| r = sd_bus_message_read_array(m, SD_BUS_TYPE_BYTE, &v, &n); |
| if (r < 0) |
| return r; |
| |
| if (n == 0) |
| *p = SD_ID128_NULL; |
| else if (n == 16) |
| memcpy((*p).bytes, v, n); |
| else |
| return -EINVAL; |
| |
| return 0; |
| } |
| |
| static int map_basic(sd_bus *bus, const char *member, sd_bus_message *m, unsigned flags, sd_bus_error *error, void *userdata) { |
| char type; |
| int r; |
| |
| r = sd_bus_message_peek_type(m, &type, NULL); |
| if (r < 0) |
| return r; |
| |
| switch (type) { |
| |
| case SD_BUS_TYPE_STRING: { |
| const char **p = userdata; |
| const char *s; |
| |
| r = sd_bus_message_read_basic(m, type, &s); |
| if (r < 0) |
| return r; |
| |
| if (isempty(s)) |
| s = NULL; |
| |
| if (flags & BUS_MAP_STRDUP) |
| return free_and_strdup((char **) userdata, s); |
| |
| *p = s; |
| return 0; |
| } |
| |
| case SD_BUS_TYPE_ARRAY: { |
| _cleanup_strv_free_ char **l = NULL; |
| char ***p = userdata; |
| |
| r = bus_message_read_strv_extend(m, &l); |
| if (r < 0) |
| return r; |
| |
| return strv_free_and_replace(*p, l); |
| } |
| |
| case SD_BUS_TYPE_BOOLEAN: { |
| int b; |
| |
| r = sd_bus_message_read_basic(m, type, &b); |
| if (r < 0) |
| return r; |
| |
| if (flags & BUS_MAP_BOOLEAN_AS_BOOL) |
| *(bool*) userdata = b; |
| else |
| *(int*) userdata = b; |
| |
| return 0; |
| } |
| |
| case SD_BUS_TYPE_INT32: |
| case SD_BUS_TYPE_UINT32: { |
| uint32_t u, *p = userdata; |
| |
| r = sd_bus_message_read_basic(m, type, &u); |
| if (r < 0) |
| return r; |
| |
| *p = u; |
| return 0; |
| } |
| |
| case SD_BUS_TYPE_INT64: |
| case SD_BUS_TYPE_UINT64: { |
| uint64_t t, *p = userdata; |
| |
| r = sd_bus_message_read_basic(m, type, &t); |
| if (r < 0) |
| return r; |
| |
| *p = t; |
| return 0; |
| } |
| |
| case SD_BUS_TYPE_DOUBLE: { |
| double d, *p = userdata; |
| |
| r = sd_bus_message_read_basic(m, type, &d); |
| if (r < 0) |
| return r; |
| |
| *p = d; |
| return 0; |
| }} |
| |
| return -EOPNOTSUPP; |
| } |
| |
| int bus_message_map_all_properties( |
| sd_bus_message *m, |
| const struct bus_properties_map *map, |
| unsigned flags, |
| sd_bus_error *error, |
| void *userdata) { |
| |
| int r; |
| |
| assert(m); |
| assert(map); |
| |
| r = sd_bus_message_enter_container(m, SD_BUS_TYPE_ARRAY, "{sv}"); |
| if (r < 0) |
| return r; |
| |
| while ((r = sd_bus_message_enter_container(m, SD_BUS_TYPE_DICT_ENTRY, "sv")) > 0) { |
| const struct bus_properties_map *prop; |
| const char *member; |
| const char *contents; |
| void *v; |
| unsigned i; |
| |
| r = sd_bus_message_read_basic(m, SD_BUS_TYPE_STRING, &member); |
| if (r < 0) |
| return r; |
| |
| for (i = 0, prop = NULL; map[i].member; i++) |
| if (streq(map[i].member, member)) { |
| prop = &map[i]; |
| break; |
| } |
| |
| if (prop) { |
| r = sd_bus_message_peek_type(m, NULL, &contents); |
| if (r < 0) |
| return r; |
| |
| r = sd_bus_message_enter_container(m, SD_BUS_TYPE_VARIANT, contents); |
| if (r < 0) |
| return r; |
| |
| v = (uint8_t *)userdata + prop->offset; |
| if (map[i].set) |
| r = prop->set(sd_bus_message_get_bus(m), member, m, error, v); |
| else |
| r = map_basic(sd_bus_message_get_bus(m), member, m, flags, error, v); |
| if (r < 0) |
| return r; |
| |
| r = sd_bus_message_exit_container(m); |
| if (r < 0) |
| return r; |
| } else { |
| r = sd_bus_message_skip(m, "v"); |
| if (r < 0) |
| return r; |
| } |
| |
| r = sd_bus_message_exit_container(m); |
| if (r < 0) |
| return r; |
| } |
| if (r < 0) |
| return r; |
| |
| return sd_bus_message_exit_container(m); |
| } |
| |
| int bus_message_map_properties_changed( |
| sd_bus_message *m, |
| const struct bus_properties_map *map, |
| unsigned flags, |
| sd_bus_error *error, |
| void *userdata) { |
| |
| const char *member; |
| int r, invalidated, i; |
| |
| assert(m); |
| assert(map); |
| |
| r = bus_message_map_all_properties(m, map, flags, error, userdata); |
| if (r < 0) |
| return r; |
| |
| r = sd_bus_message_enter_container(m, SD_BUS_TYPE_ARRAY, "s"); |
| if (r < 0) |
| return r; |
| |
| invalidated = 0; |
| while ((r = sd_bus_message_read_basic(m, SD_BUS_TYPE_STRING, &member)) > 0) |
| for (i = 0; map[i].member; i++) |
| if (streq(map[i].member, member)) { |
| ++invalidated; |
| break; |
| } |
| if (r < 0) |
| return r; |
| |
| r = sd_bus_message_exit_container(m); |
| if (r < 0) |
| return r; |
| |
| return invalidated; |
| } |
| |
| int bus_map_all_properties( |
| sd_bus *bus, |
| const char *destination, |
| const char *path, |
| const struct bus_properties_map *map, |
| unsigned flags, |
| sd_bus_error *error, |
| sd_bus_message **reply, |
| void *userdata) { |
| |
| _cleanup_(sd_bus_message_unrefp) sd_bus_message *m = NULL; |
| int r; |
| |
| assert(bus); |
| assert(destination); |
| assert(path); |
| assert(map); |
| assert(reply || (flags & BUS_MAP_STRDUP)); |
| |
| r = sd_bus_call_method( |
| bus, |
| destination, |
| path, |
| "org.freedesktop.DBus.Properties", |
| "GetAll", |
| error, |
| &m, |
| "s", ""); |
| if (r < 0) |
| return r; |
| |
| r = bus_message_map_all_properties(m, map, flags, error, userdata); |
| if (r < 0) |
| return r; |
| |
| if (reply) |
| *reply = sd_bus_message_ref(m); |
| |
| return r; |
| } |
| |
| int bus_connect_transport(BusTransport transport, const char *host, bool user, sd_bus **ret) { |
| _cleanup_(sd_bus_unrefp) sd_bus *bus = NULL; |
| int r; |
| |
| assert(transport >= 0); |
| assert(transport < _BUS_TRANSPORT_MAX); |
| assert(ret); |
| |
| assert_return((transport == BUS_TRANSPORT_LOCAL) == !host, -EINVAL); |
| assert_return(transport == BUS_TRANSPORT_LOCAL || !user, -EOPNOTSUPP); |
| |
| switch (transport) { |
| |
| case BUS_TRANSPORT_LOCAL: |
| if (user) |
| r = sd_bus_default_user(&bus); |
| else { |
| if (sd_booted() <= 0) { |
| /* Print a friendly message when the local system is actually not running systemd as PID 1. */ |
| log_error("System has not been booted with systemd as init system (PID 1). Can't operate."); |
| |
| return -EHOSTDOWN; |
| } |
| r = sd_bus_default_system(&bus); |
| } |
| break; |
| |
| case BUS_TRANSPORT_REMOTE: |
| r = sd_bus_open_system_remote(&bus, host); |
| break; |
| |
| case BUS_TRANSPORT_MACHINE: |
| r = sd_bus_open_system_machine(&bus, host); |
| break; |
| |
| default: |
| assert_not_reached("Hmm, unknown transport type."); |
| } |
| if (r < 0) |
| return r; |
| |
| r = sd_bus_set_exit_on_disconnect(bus, true); |
| if (r < 0) |
| return r; |
| |
| *ret = TAKE_PTR(bus); |
| |
| return 0; |
| } |
| |
| int bus_connect_transport_systemd(BusTransport transport, const char *host, bool user, sd_bus **bus) { |
| int r; |
| |
| assert(transport >= 0); |
| assert(transport < _BUS_TRANSPORT_MAX); |
| assert(bus); |
| |
| assert_return((transport == BUS_TRANSPORT_LOCAL) == !host, -EINVAL); |
| assert_return(transport == BUS_TRANSPORT_LOCAL || !user, -EOPNOTSUPP); |
| |
| switch (transport) { |
| |
| case BUS_TRANSPORT_LOCAL: |
| if (user) |
| r = bus_connect_user_systemd(bus); |
| else { |
| if (sd_booted() <= 0) { |
| /* Print a friendly message when the local system is actually not running systemd as PID 1. */ |
| log_error("System has not been booted with systemd as init system (PID 1). Can't operate."); |
| |
| return -EHOSTDOWN; |
| } |
| r = bus_connect_system_systemd(bus); |
| } |
| break; |
| |
| case BUS_TRANSPORT_REMOTE: |
| r = sd_bus_open_system_remote(bus, host); |
| break; |
| |
| case BUS_TRANSPORT_MACHINE: |
| r = sd_bus_open_system_machine(bus, host); |
| break; |
| |
| default: |
| assert_not_reached("Hmm, unknown transport type."); |
| } |
| |
| return r; |
| } |
| |
| int bus_property_get_bool( |
| sd_bus *bus, |
| const char *path, |
| const char *interface, |
| const char *property, |
| sd_bus_message *reply, |
| void *userdata, |
| sd_bus_error *error) { |
| |
| int b = *(bool*) userdata; |
| |
| return sd_bus_message_append_basic(reply, 'b', &b); |
| } |
| |
| int bus_property_set_bool( |
| sd_bus *bus, |
| const char *path, |
| const char *interface, |
| const char *property, |
| sd_bus_message *value, |
| void *userdata, |
| sd_bus_error *error) { |
| |
| int b, r; |
| |
| r = sd_bus_message_read(value, "b", &b); |
| if (r < 0) |
| return r; |
| |
| *(bool*) userdata = b; |
| return 0; |
| } |
| |
| int bus_property_get_id128( |
| sd_bus *bus, |
| const char *path, |
| const char *interface, |
| const char *property, |
| sd_bus_message *reply, |
| void *userdata, |
| sd_bus_error *error) { |
| |
| sd_id128_t *id = userdata; |
| |
| if (sd_id128_is_null(*id)) /* Add an empty array if the ID is zero */ |
| return sd_bus_message_append(reply, "ay", 0); |
| else |
| return sd_bus_message_append_array(reply, 'y', id->bytes, 16); |
| } |
| |
| #if __SIZEOF_SIZE_T__ != 8 |
| int bus_property_get_size( |
| sd_bus *bus, |
| const char *path, |
| const char *interface, |
| const char *property, |
| sd_bus_message *reply, |
| void *userdata, |
| sd_bus_error *error) { |
| |
| uint64_t sz = *(size_t*) userdata; |
| |
| return sd_bus_message_append_basic(reply, 't', &sz); |
| } |
| #endif |
| |
| #if __SIZEOF_LONG__ != 8 |
| int bus_property_get_long( |
| sd_bus *bus, |
| const char *path, |
| const char *interface, |
| const char *property, |
| sd_bus_message *reply, |
| void *userdata, |
| sd_bus_error *error) { |
| |
| int64_t l = *(long*) userdata; |
| |
| return sd_bus_message_append_basic(reply, 'x', &l); |
| } |
| |
| int bus_property_get_ulong( |
| sd_bus *bus, |
| const char *path, |
| const char *interface, |
| const char *property, |
| sd_bus_message *reply, |
| void *userdata, |
| sd_bus_error *error) { |
| |
| uint64_t ul = *(unsigned long*) userdata; |
| |
| return sd_bus_message_append_basic(reply, 't', &ul); |
| } |
| #endif |
| |
| int bus_log_parse_error(int r) { |
| return log_error_errno(r, "Failed to parse bus message: %m"); |
| } |
| |
| int bus_log_create_error(int r) { |
| return log_error_errno(r, "Failed to create bus message: %m"); |
| } |
| |
| /** |
| * bus_path_encode_unique() - encode unique object path |
| * @b: bus connection or NULL |
| * @prefix: object path prefix |
| * @sender_id: unique-name of client, or NULL |
| * @external_id: external ID to be chosen by client, or NULL |
| * @ret_path: storage for encoded object path pointer |
| * |
| * Whenever we provide a bus API that allows clients to create and manage |
| * server-side objects, we need to provide a unique name for these objects. If |
| * we let the server choose the name, we suffer from a race condition: If a |
| * client creates an object asynchronously, it cannot destroy that object until |
| * it received the method reply. It cannot know the name of the new object, |
| * thus, it cannot destroy it. Furthermore, it enforces a round-trip. |
| * |
| * Therefore, many APIs allow the client to choose the unique name for newly |
| * created objects. There're two problems to solve, though: |
| * 1) Object names are usually defined via dbus object paths, which are |
| * usually globally namespaced. Therefore, multiple clients must be able |
| * to choose unique object names without interference. |
| * 2) If multiple libraries share the same bus connection, they must be |
| * able to choose unique object names without interference. |
| * The first problem is solved easily by prefixing a name with the |
| * unique-bus-name of a connection. The server side must enforce this and |
| * reject any other name. The second problem is solved by providing unique |
| * suffixes from within sd-bus. |
| * |
| * This helper allows clients to create unique object-paths. It uses the |
| * template '/prefix/sender_id/external_id' and returns the new path in |
| * @ret_path (must be freed by the caller). |
| * If @sender_id is NULL, the unique-name of @b is used. If @external_id is |
| * NULL, this function allocates a unique suffix via @b (by requesting a new |
| * cookie). If both @sender_id and @external_id are given, @b can be passed as |
| * NULL. |
| * |
| * Returns: 0 on success, negative error code on failure. |
| */ |
| int bus_path_encode_unique(sd_bus *b, const char *prefix, const char *sender_id, const char *external_id, char **ret_path) { |
| _cleanup_free_ char *sender_label = NULL, *external_label = NULL; |
| char external_buf[DECIMAL_STR_MAX(uint64_t)], *p; |
| int r; |
| |
| assert_return(b || (sender_id && external_id), -EINVAL); |
| assert_return(object_path_is_valid(prefix), -EINVAL); |
| assert_return(ret_path, -EINVAL); |
| |
| if (!sender_id) { |
| r = sd_bus_get_unique_name(b, &sender_id); |
| if (r < 0) |
| return r; |
| } |
| |
| if (!external_id) { |
| xsprintf(external_buf, "%"PRIu64, ++b->cookie); |
| external_id = external_buf; |
| } |
| |
| sender_label = bus_label_escape(sender_id); |
| if (!sender_label) |
| return -ENOMEM; |
| |
| external_label = bus_label_escape(external_id); |
| if (!external_label) |
| return -ENOMEM; |
| |
| p = strjoin(prefix, "/", sender_label, "/", external_label); |
| if (!p) |
| return -ENOMEM; |
| |
| *ret_path = p; |
| return 0; |
| } |
| |
| /** |
| * bus_path_decode_unique() - decode unique object path |
| * @path: object path to decode |
| * @prefix: object path prefix |
| * @ret_sender: output parameter for sender-id label |
| * @ret_external: output parameter for external-id label |
| * |
| * This does the reverse of bus_path_encode_unique() (see its description for |
| * details). Both trailing labels, sender-id and external-id, are unescaped and |
| * returned in the given output parameters (the caller must free them). |
| * |
| * Note that this function returns 0 if the path does not match the template |
| * (see bus_path_encode_unique()), 1 if it matched. |
| * |
| * Returns: Negative error code on failure, 0 if the given object path does not |
| * match the template (return parameters are set to NULL), 1 if it was |
| * parsed successfully (return parameters contain allocated labels). |
| */ |
| int bus_path_decode_unique(const char *path, const char *prefix, char **ret_sender, char **ret_external) { |
| const char *p, *q; |
| char *sender, *external; |
| |
| assert(object_path_is_valid(path)); |
| assert(object_path_is_valid(prefix)); |
| assert(ret_sender); |
| assert(ret_external); |
| |
| p = object_path_startswith(path, prefix); |
| if (!p) { |
| *ret_sender = NULL; |
| *ret_external = NULL; |
| return 0; |
| } |
| |
| q = strchr(p, '/'); |
| if (!q) { |
| *ret_sender = NULL; |
| *ret_external = NULL; |
| return 0; |
| } |
| |
| sender = bus_label_unescape_n(p, q - p); |
| external = bus_label_unescape(q + 1); |
| if (!sender || !external) { |
| free(sender); |
| free(external); |
| return -ENOMEM; |
| } |
| |
| *ret_sender = sender; |
| *ret_external = external; |
| return 1; |
| } |
| |
| int bus_property_get_rlimit( |
| sd_bus *bus, |
| const char *path, |
| const char *interface, |
| const char *property, |
| sd_bus_message *reply, |
| void *userdata, |
| sd_bus_error *error) { |
| |
| const char *is_soft; |
| struct rlimit *rl; |
| uint64_t u; |
| rlim_t x; |
| |
| assert(bus); |
| assert(reply); |
| assert(userdata); |
| |
| is_soft = endswith(property, "Soft"); |
| |
| rl = *(struct rlimit**) userdata; |
| if (rl) |
| x = is_soft ? rl->rlim_cur : rl->rlim_max; |
| else { |
| struct rlimit buf = {}; |
| const char *s, *p; |
| int z; |
| |
| /* Chop off "Soft" suffix */ |
| s = is_soft ? strndupa(property, is_soft - property) : property; |
| |
| /* Skip over any prefix, such as "Default" */ |
| assert_se(p = strstr(s, "Limit")); |
| |
| z = rlimit_from_string(p + 5); |
| assert(z >= 0); |
| |
| (void) getrlimit(z, &buf); |
| x = is_soft ? buf.rlim_cur : buf.rlim_max; |
| } |
| |
| /* rlim_t might have different sizes, let's map RLIMIT_INFINITY to (uint64_t) -1, so that it is the same on all |
| * archs */ |
| u = x == RLIM_INFINITY ? (uint64_t) -1 : (uint64_t) x; |
| |
| return sd_bus_message_append(reply, "t", u); |
| } |
| |
| int bus_track_add_name_many(sd_bus_track *t, char **l) { |
| int r = 0; |
| char **i; |
| |
| assert(t); |
| |
| /* Continues adding after failure, and returns the first failure. */ |
| |
| STRV_FOREACH(i, l) { |
| int k; |
| |
| k = sd_bus_track_add_name(t, *i); |
| if (k < 0 && r >= 0) |
| r = k; |
| } |
| |
| return r; |
| } |
| |
| int bus_open_system_watch_bind_with_description(sd_bus **ret, const char *description) { |
| _cleanup_(sd_bus_unrefp) sd_bus *bus = NULL; |
| const char *e; |
| int r; |
| |
| assert(ret); |
| |
| /* Match like sd_bus_open_system(), but with the "watch_bind" feature and the Connected() signal |
| * turned on. */ |
| |
| r = sd_bus_new(&bus); |
| if (r < 0) |
| return r; |
| |
| if (description) { |
| r = sd_bus_set_description(bus, description); |
| if (r < 0) |
| return r; |
| } |
| |
| e = secure_getenv("DBUS_SYSTEM_BUS_ADDRESS"); |
| if (!e) |
| e = DEFAULT_SYSTEM_BUS_ADDRESS; |
| |
| r = sd_bus_set_address(bus, e); |
| if (r < 0) |
| return r; |
| |
| r = sd_bus_set_bus_client(bus, true); |
| if (r < 0) |
| return r; |
| |
| r = sd_bus_negotiate_creds(bus, true, SD_BUS_CREDS_UID|SD_BUS_CREDS_EUID|SD_BUS_CREDS_EFFECTIVE_CAPS); |
| if (r < 0) |
| return r; |
| |
| r = sd_bus_set_watch_bind(bus, true); |
| if (r < 0) |
| return r; |
| |
| r = sd_bus_set_connected_signal(bus, true); |
| if (r < 0) |
| return r; |
| |
| r = sd_bus_start(bus); |
| if (r < 0) |
| return r; |
| |
| *ret = TAKE_PTR(bus); |
| |
| return 0; |
| } |
| |
| struct request_name_data { |
| unsigned n_ref; |
| |
| const char *name; |
| uint64_t flags; |
| void *userdata; |
| }; |
| |
| static void request_name_destroy_callback(void *userdata) { |
| struct request_name_data *data = userdata; |
| |
| assert(data); |
| assert(data->n_ref > 0); |
| |
| log_info("%s n_ref=%u", __func__, data->n_ref); |
| |
| data->n_ref--; |
| if (data->n_ref == 0) |
| free(data); |
| } |
| |
| static int reload_dbus_handler(sd_bus_message *m, void *userdata, sd_bus_error *ret_error) { |
| struct request_name_data *data = userdata; |
| const sd_bus_error *e; |
| int r; |
| |
| assert(data); |
| assert(data->name); |
| assert(data->n_ref > 0); |
| |
| e = sd_bus_message_get_error(m); |
| if (e) { |
| log_error_errno(sd_bus_error_get_errno(e), "Failed to reload DBus configuration: %s", e->message); |
| return 1; |
| } |
| |
| /* Here, use the default request name handler to avoid an infinite loop of reloading and requesting. */ |
| r = sd_bus_request_name_async(sd_bus_message_get_bus(m), NULL, data->name, data->flags, NULL, data->userdata); |
| if (r < 0) |
| log_error_errno(r, "Failed to request name: %m"); |
| |
| return 1; |
| } |
| |
| static int request_name_handler_may_reload_dbus(sd_bus_message *m, void *userdata, sd_bus_error *ret_error) { |
| struct request_name_data *data = userdata; |
| uint32_t ret; |
| int r; |
| |
| assert(m); |
| assert(data); |
| |
| if (sd_bus_message_is_method_error(m, NULL)) { |
| const sd_bus_error *e = sd_bus_message_get_error(m); |
| _cleanup_(sd_bus_slot_unrefp) sd_bus_slot *slot = NULL; |
| |
| if (!sd_bus_error_has_name(e, SD_BUS_ERROR_ACCESS_DENIED)) { |
| log_debug_errno(sd_bus_error_get_errno(e), |
| "Unable to request name, failing connection: %s", |
| e->message); |
| |
| bus_enter_closing(sd_bus_message_get_bus(m)); |
| return 1; |
| } |
| |
| log_debug_errno(sd_bus_error_get_errno(e), |
| "Unable to request name, will retry after reloading DBus configuration: %s", |
| e->message); |
| |
| /* If systemd-timesyncd.service enables DynamicUser= and dbus.service |
| * started before the dynamic user is realized, then the DBus policy |
| * about timesyncd has not been enabled yet. So, let's try to reload |
| * DBus configuration, and after that request the name again. Note that it |
| * seems that no privileges are necessary to call the following method. */ |
| |
| r = sd_bus_call_method_async( |
| sd_bus_message_get_bus(m), |
| &slot, |
| "org.freedesktop.DBus", |
| "/org/freedesktop/DBus", |
| "org.freedesktop.DBus", |
| "ReloadConfig", |
| reload_dbus_handler, |
| data, NULL); |
| if (r < 0) { |
| log_error_errno(r, "Failed to reload DBus configuration: %m"); |
| bus_enter_closing(sd_bus_message_get_bus(m)); |
| return 1; |
| } |
| |
| data->n_ref ++; |
| assert_se(sd_bus_slot_set_destroy_callback(slot, request_name_destroy_callback) >= 0); |
| |
| r = sd_bus_slot_set_floating(slot, true); |
| if (r < 0) |
| return r; |
| |
| return 1; |
| } |
| |
| r = sd_bus_message_read(m, "u", &ret); |
| if (r < 0) |
| return r; |
| |
| switch (ret) { |
| |
| case BUS_NAME_ALREADY_OWNER: |
| log_debug("Already owner of requested service name, ignoring."); |
| return 1; |
| |
| case BUS_NAME_IN_QUEUE: |
| log_debug("In queue for requested service name."); |
| return 1; |
| |
| case BUS_NAME_PRIMARY_OWNER: |
| log_debug("Successfully acquired requested service name."); |
| return 1; |
| |
| case BUS_NAME_EXISTS: |
| log_debug("Requested service name already owned, failing connection."); |
| bus_enter_closing(sd_bus_message_get_bus(m)); |
| return 1; |
| } |
| |
| log_debug("Unexpected response from RequestName(), failing connection."); |
| bus_enter_closing(sd_bus_message_get_bus(m)); |
| return 1; |
| } |
| |
| int bus_request_name_async_may_reload_dbus(sd_bus *bus, sd_bus_slot **ret_slot, const char *name, uint64_t flags, void *userdata) { |
| _cleanup_free_ struct request_name_data *data = NULL; |
| _cleanup_(sd_bus_slot_unrefp) sd_bus_slot *slot = NULL; |
| int r; |
| |
| data = new(struct request_name_data, 1); |
| if (!data) |
| return -ENOMEM; |
| |
| *data = (struct request_name_data) { |
| .n_ref = 1, |
| .name = name, |
| .flags = flags, |
| .userdata = userdata, |
| }; |
| |
| r = sd_bus_request_name_async(bus, &slot, name, flags, request_name_handler_may_reload_dbus, data); |
| if (r < 0) |
| return r; |
| |
| assert_se(sd_bus_slot_set_destroy_callback(slot, request_name_destroy_callback) >= 0); |
| TAKE_PTR(data); |
| |
| if (ret_slot) |
| *ret_slot = TAKE_PTR(slot); |
| else { |
| r = sd_bus_slot_set_floating(slot, true); |
| if (r < 0) |
| return r; |
| } |
| |
| return 0; |
| } |
| |
| int bus_reply_pair_array(sd_bus_message *m, char **l) { |
| _cleanup_(sd_bus_message_unrefp) sd_bus_message *reply = NULL; |
| char **k, **v; |
| int r; |
| |
| assert(m); |
| |
| /* Reply to the specified message with a message containing a dictionary put together from the |
| * specified strv */ |
| |
| r = sd_bus_message_new_method_return(m, &reply); |
| if (r < 0) |
| return r; |
| |
| r = sd_bus_message_open_container(reply, 'a', "{ss}"); |
| if (r < 0) |
| return r; |
| |
| STRV_FOREACH_PAIR(k, v, l) { |
| r = sd_bus_message_append(reply, "{ss}", *k, *v); |
| if (r < 0) |
| return r; |
| } |
| |
| r = sd_bus_message_close_container(reply); |
| if (r < 0) |
| return r; |
| |
| return sd_bus_send(NULL, reply, NULL); |
| } |