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