170 lines
4.6 KiB
C
170 lines
4.6 KiB
C
/* SPDX-License-Identifier: LGPL-2.1+ */
|
|
|
|
#include <sys/stat.h>
|
|
#include <sys/types.h>
|
|
|
|
#include "sd-bus.h"
|
|
#include "sd-daemon.h"
|
|
|
|
#include "alloc-util.h"
|
|
#include "bus-log-control-api.h"
|
|
#include "bus-polkit.h"
|
|
#include "def.h"
|
|
#include "main-func.h"
|
|
#include "portabled-bus.h"
|
|
#include "portabled-image-bus.h"
|
|
#include "portabled.h"
|
|
#include "process-util.h"
|
|
#include "signal-util.h"
|
|
|
|
static Manager* manager_unref(Manager *m);
|
|
DEFINE_TRIVIAL_CLEANUP_FUNC(Manager*, manager_unref);
|
|
|
|
static int manager_new(Manager **ret) {
|
|
_cleanup_(manager_unrefp) Manager *m = NULL;
|
|
int r;
|
|
|
|
assert(ret);
|
|
|
|
m = new0(Manager, 1);
|
|
if (!m)
|
|
return -ENOMEM;
|
|
|
|
r = sd_event_default(&m->event);
|
|
if (r < 0)
|
|
return r;
|
|
|
|
r = sd_event_add_signal(m->event, NULL, SIGINT, NULL, NULL);
|
|
if (r < 0)
|
|
return r;
|
|
|
|
r = sd_event_add_signal(m->event, NULL, SIGTERM, NULL, NULL);
|
|
if (r < 0)
|
|
return r;
|
|
|
|
(void) sd_event_set_watchdog(m->event, true);
|
|
|
|
*ret = TAKE_PTR(m);
|
|
return 0;
|
|
}
|
|
|
|
static Manager* manager_unref(Manager *m) {
|
|
assert(m);
|
|
|
|
hashmap_free(m->image_cache);
|
|
|
|
sd_event_source_unref(m->image_cache_defer_event);
|
|
|
|
bus_verify_polkit_async_registry_free(m->polkit_registry);
|
|
|
|
sd_bus_flush_close_unref(m->bus);
|
|
sd_event_unref(m->event);
|
|
|
|
return mfree(m);
|
|
}
|
|
|
|
static int manager_connect_bus(Manager *m) {
|
|
int r;
|
|
|
|
assert(m);
|
|
assert(!m->bus);
|
|
|
|
r = sd_bus_default_system(&m->bus);
|
|
if (r < 0)
|
|
return log_error_errno(r, "Failed to connect to system bus: %m");
|
|
|
|
r = sd_bus_add_object_vtable(m->bus, NULL, "/org/freedesktop/portable1", "org.freedesktop.portable1.Manager", manager_vtable, m);
|
|
if (r < 0)
|
|
return log_error_errno(r, "Failed to add manager object vtable: %m");
|
|
|
|
r = sd_bus_add_fallback_vtable(m->bus, NULL, "/org/freedesktop/portable1/image", "org.freedesktop.portable1.Image", image_vtable, bus_image_object_find, m);
|
|
if (r < 0)
|
|
return log_error_errno(r, "Failed to add image object vtable: %m");
|
|
|
|
r = sd_bus_add_node_enumerator(m->bus, NULL, "/org/freedesktop/portable1/image", bus_image_node_enumerator, m);
|
|
if (r < 0)
|
|
return log_error_errno(r, "Failed to add image enumerator: %m");
|
|
|
|
r = bus_log_control_api_register(m->bus);
|
|
if (r < 0)
|
|
return r;
|
|
|
|
r = sd_bus_request_name_async(m->bus, NULL, "org.freedesktop.portable1", 0, NULL, NULL);
|
|
if (r < 0)
|
|
return log_error_errno(r, "Failed to request name: %m");
|
|
|
|
r = sd_bus_attach_event(m->bus, m->event, 0);
|
|
if (r < 0)
|
|
return log_error_errno(r, "Failed to attach bus to event loop: %m");
|
|
|
|
(void) sd_bus_set_exit_on_disconnect(m->bus, true);
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int manager_startup(Manager *m) {
|
|
int r;
|
|
|
|
assert(m);
|
|
|
|
r = manager_connect_bus(m);
|
|
if (r < 0)
|
|
return r;
|
|
|
|
return 0;
|
|
}
|
|
|
|
static bool check_idle(void *userdata) {
|
|
Manager *m = userdata;
|
|
|
|
return !m->operations;
|
|
}
|
|
|
|
static int manager_run(Manager *m) {
|
|
assert(m);
|
|
|
|
return bus_event_loop_with_idle(
|
|
m->event,
|
|
m->bus,
|
|
"org.freedesktop.portable1",
|
|
DEFAULT_EXIT_USEC,
|
|
check_idle, m);
|
|
}
|
|
|
|
static int run(int argc, char *argv[]) {
|
|
_cleanup_(manager_unrefp) Manager *m = NULL;
|
|
int r;
|
|
|
|
log_setup_service();
|
|
|
|
umask(0022);
|
|
|
|
if (argc != 1)
|
|
return log_error_errno(SYNTHETIC_ERRNO(EINVAL), "This program takes no arguments.");
|
|
|
|
assert_se(sigprocmask_many(SIG_BLOCK, NULL, SIGCHLD, SIGTERM, SIGINT, -1) >= 0);
|
|
|
|
r = manager_new(&m);
|
|
if (r < 0)
|
|
return log_error_errno(r, "Failed to allocate manager object: %m");
|
|
|
|
r = manager_startup(m);
|
|
if (r < 0)
|
|
return log_error_errno(r, "Failed to fully start up daemon: %m");
|
|
|
|
log_debug("systemd-portabled running as pid " PID_FMT, getpid_cached());
|
|
sd_notify(false,
|
|
"READY=1\n"
|
|
"STATUS=Processing requests...");
|
|
|
|
r = manager_run(m);
|
|
|
|
(void) sd_notify(false,
|
|
"STOPPING=1\n"
|
|
"STATUS=Shutting down...");
|
|
log_debug("systemd-portabled stopped as pid " PID_FMT, getpid_cached());
|
|
return r;
|
|
}
|
|
|
|
DEFINE_MAIN_FUNCTION(run);
|