src/portable/portabled.c

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