]>
Commit | Line | Data |
---|---|---|
61d0578b LP |
1 | /* SPDX-License-Identifier: LGPL-2.1+ */ |
2 | ||
ca78ad1d ZJS |
3 | #include <sys/stat.h> |
4 | #include <sys/types.h> | |
5 | ||
61d0578b LP |
6 | #include "sd-bus.h" |
7 | #include "sd-daemon.h" | |
8 | ||
9 | #include "alloc-util.h" | |
269e4d2d | 10 | #include "bus-polkit.h" |
61d0578b | 11 | #include "def.h" |
5e332028 | 12 | #include "main-func.h" |
61d0578b LP |
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 | ||
b07ec5a1 | 53 | hashmap_free(m->image_cache); |
61d0578b LP |
54 | |
55 | sd_event_source_unref(m->image_cache_defer_event); | |
56 | ||
57 | bus_verify_polkit_async_registry_free(m->polkit_registry); | |
58 | ||
92e31da1 | 59 | sd_bus_flush_close_unref(m->bus); |
61d0578b LP |
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 | ||
166a9dc6 | 129 | static int run(int argc, char *argv[]) { |
61d0578b LP |
130 | _cleanup_(manager_unrefp) Manager *m = NULL; |
131 | int r; | |
132 | ||
6bf3c61c | 133 | log_setup_service(); |
61d0578b LP |
134 | |
135 | umask(0022); | |
136 | ||
e4c5d78e LP |
137 | if (argc != 1) |
138 | return log_error_errno(SYNTHETIC_ERRNO(EINVAL), "This program takes no arguments."); | |
61d0578b LP |
139 | |
140 | assert_se(sigprocmask_many(SIG_BLOCK, NULL, SIGCHLD, SIGTERM, SIGINT, -1) >= 0); | |
141 | ||
142 | r = manager_new(&m); | |
166a9dc6 ZJS |
143 | if (r < 0) |
144 | return log_error_errno(r, "Failed to allocate manager object: %m"); | |
61d0578b LP |
145 | |
146 | r = manager_startup(m); | |
166a9dc6 ZJS |
147 | if (r < 0) |
148 | return log_error_errno(r, "Failed to fully start up daemon: %m"); | |
61d0578b LP |
149 | |
150 | log_debug("systemd-portabled running as pid " PID_FMT, getpid_cached()); | |
61d0578b LP |
151 | sd_notify(false, |
152 | "READY=1\n" | |
153 | "STATUS=Processing requests..."); | |
154 | ||
155 | r = manager_run(m); | |
156 | ||
44c5d044 ZJS |
157 | (void) sd_notify(false, |
158 | "STOPPING=1\n" | |
159 | "STATUS=Shutting down..."); | |
61d0578b | 160 | log_debug("systemd-portabled stopped as pid " PID_FMT, getpid_cached()); |
166a9dc6 | 161 | return r; |
61d0578b | 162 | } |
166a9dc6 ZJS |
163 | |
164 | DEFINE_MAIN_FUNCTION(run); |