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