src/portable/portabled.c

   1 /* SPDX-License-Identifier: LGPL-2.1+ */
   2
   3 #include <sys/stat.h>
   4 #include <sys/types.h>
   5
   6 #include "sd-bus.h"
   7 #include "sd-daemon.h"
   8
   9 #include "alloc-util.h"
  10 #include "bus-polkit.h"
  11 #include "def.h"
  12 #include "main-func.h"
  13 #include "portabled-bus.h"
  14 #include "portabled-image-bus.h"
  15 #include "portabled.h"
  16 #include "process-util.h"
  17 #include "signal-util.h"
  18
  19 static Manager* manager_unref(Manager *m);
  20 DEFINE_TRIVIAL_CLEANUP_FUNC(Manager*, manager_unref);
  21
  22 static int manager_new(Manager **ret) {
  23         _cleanup_(manager_unrefp) Manager *m = NULL;
  24         int r;
  25
  26         assert(ret);
  27
  28         m = new0(Manager, 1);
  29         if (!m)
  30                 return -ENOMEM;
  31
  32         r = sd_event_default(&m->event);
  33         if (r < 0)
  34                 return r;
  35
  36         r = sd_event_add_signal(m->event, NULL, SIGINT, NULL, NULL);
  37         if (r < 0)
  38                 return r;
  39
  40         r = sd_event_add_signal(m->event, NULL, SIGTERM, NULL, NULL);
  41         if (r < 0)
  42                 return r;
  43
  44         (void) sd_event_set_watchdog(m->event, true);
  45
  46         *ret = TAKE_PTR(m);
  47         return 0;
  48 }
  49
  50 static Manager* manager_unref(Manager *m) {
  51         assert(m);
  52
  53         hashmap_free(m->image_cache);
  54
  55         sd_event_source_unref(m->image_cache_defer_event);
  56
  57         bus_verify_polkit_async_registry_free(m->polkit_registry);
  58
  59         sd_bus_flush_close_unref(m->bus);
  60         sd_event_unref(m->event);
  61
  62         return mfree(m);
  63 }
  64
  65 static int manager_connect_bus(Manager *m) {
  66         int r;
  67
  68         assert(m);
  69         assert(!m->bus);
  70
  71         r = sd_bus_default_system(&m->bus);
  72         if (r < 0)
  73                 return log_error_errno(r, "Failed to connect to system bus: %m");
  74
  75         r = sd_bus_add_object_vtable(m->bus, NULL, "/org/freedesktop/portable1", "org.freedesktop.portable1.Manager", manager_vtable, m);
  76         if (r < 0)
  77                 return log_error_errno(r, "Failed to add manager object vtable: %m");
  78
  79         r = sd_bus_add_fallback_vtable(m->bus, NULL, "/org/freedesktop/portable1/image", "org.freedesktop.portable1.Image", image_vtable, bus_image_object_find, m);
  80         if (r < 0)
  81                 return log_error_errno(r, "Failed to add image object vtable: %m");
  82
  83         r = sd_bus_add_node_enumerator(m->bus, NULL, "/org/freedesktop/portable1/image", bus_image_node_enumerator, m);
  84         if (r < 0)
  85                 return log_error_errno(r, "Failed to add image enumerator: %m");
  86
  87         r = sd_bus_request_name_async(m->bus, NULL, "org.freedesktop.portable1", 0, NULL, NULL);
  88         if (r < 0)
  89                 return log_error_errno(r, "Failed to request name: %m");
  90
  91         r = sd_bus_attach_event(m->bus, m->event, 0);
  92         if (r < 0)
  93                 return log_error_errno(r, "Failed to attach bus to event loop: %m");
  94
  95         (void) sd_bus_set_exit_on_disconnect(m->bus, true);
  96
  97         return 0;
  98 }
  99
 100 static int manager_startup(Manager *m) {
 101         int r;
 102
 103         assert(m);
 104
 105         r = manager_connect_bus(m);
 106         if (r < 0)
 107                 return r;
 108
 109         return 0;
 110 }
 111
 112 static bool check_idle(void *userdata) {
 113         Manager *m = userdata;
 114
 115         return !m->operations;
 116 }
 117
 118 static int manager_run(Manager *m) {
 119         assert(m);
 120
 121         return bus_event_loop_with_idle(
 122                         m->event,
 123                         m->bus,
 124                         "org.freedesktop.portable1",
 125                         DEFAULT_EXIT_USEC,
 126                         check_idle, m);
 127 }
 128
 129 static int run(int argc, char *argv[]) {
 130         _cleanup_(manager_unrefp) Manager *m = NULL;
 131         int r;
 132
 133         log_setup_service();
 134
 135         umask(0022);
 136
 137         if (argc != 1)
 138                 return log_error_errno(SYNTHETIC_ERRNO(EINVAL), "This program takes no arguments.");
 139
 140         assert_se(sigprocmask_many(SIG_BLOCK, NULL, SIGCHLD, SIGTERM, SIGINT, -1) >= 0);
 141
 142         r = manager_new(&m);
 143         if (r < 0)
 144                 return log_error_errno(r, "Failed to allocate manager object: %m");
 145
 146         r = manager_startup(m);
 147         if (r < 0)
 148                 return log_error_errno(r, "Failed to fully start up daemon: %m");
 149
 150         log_debug("systemd-portabled running as pid " PID_FMT, getpid_cached());
 151         sd_notify(false,
 152                   "READY=1\n"
 153                   "STATUS=Processing requests...");
 154
 155         r = manager_run(m);
 156
 157         (void) sd_notify(false,
 158                          "STOPPING=1\n"
 159                          "STATUS=Shutting down...");
 160         log_debug("systemd-portabled stopped as pid " PID_FMT, getpid_cached());
 161         return r;
 162 }
 163
 164 DEFINE_MAIN_FUNCTION(run);