2 Copyright (C) 2015-2020 IoT.bzh
4 author: José Bollo <jose.bollo@iot.bzh>
6 Licensed under the Apache License, Version 2.0 (the "License");
7 you may not use this file except in compliance with the License.
8 You may obtain a copy of the License at
10 http://www.apache.org/licenses/LICENSE-2.0
12 Unless required by applicable law or agreed to in writing, software
13 distributed under the License is distributed on an "AS IS" BASIS,
14 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15 See the License for the specific language governing permissions and
16 limitations under the License.
27 #include <sys/types.h>
31 # include <systemd/sd-bus.h>
32 # include <systemd/sd-bus-protocol.h>
35 struct sd_bus_message;
36 typedef struct { const char *name; const char *message; } sd_bus_error;
37 # define sd_bus_unref(...) ((void)0)
38 # define sd_bus_default_user(p) ((*(p)=NULL),(-ENOTSUP))
39 # define sd_bus_default_system(p) ((*(p)=NULL),(-ENOTSUP))
40 # define sd_bus_call_method(...) (-ENOTSUP)
41 # define SD_BUS_ERROR_NULL {NULL,NULL}
42 # define sd_bus_message_read_basic(...) (-ENOTSUP)
43 # define sd_bus_message_unref(...) (NULL)
44 # define sd_bus_get_property_string(...) (-ENOTSUP)
45 # define sd_bus_get_property_trivial(...) (-ENOTSUP)
48 #include "utils-systemd.h"
50 #if !defined(SYSTEMD_UNITS_ROOT)
51 # define SYSTEMD_UNITS_ROOT "/usr/local/lib/systemd"
54 static const char sdb_path[] = "/org/freedesktop/systemd1";
55 static const char sdb_destination[] = "org.freedesktop.systemd1";
56 static const char sdbi_manager[] = "org.freedesktop.systemd1.Manager";
57 static const char sdbi_unit[] = "org.freedesktop.systemd1.Unit";
58 static const char sdbi_service[] = "org.freedesktop.systemd1.Service";
59 static const char sdbm_reload[] = "Reload";
60 static const char sdbm_start_unit[] = "StartUnit";
61 static const char sdbm_restart_unit[] = "RestartUnit";
62 static const char sdbm_stop_unit[] = "StopUnit";
63 static const char sdbm_start[] = "Start";
64 static const char sdbm_restart[] = "Restart";
65 static const char sdbm_stop[] = "Stop";
66 static const char sdbm_get_unit[] = "GetUnit";
67 static const char sdbm_get_unit_by_pid[] = "GetUnitByPID";
68 static const char sdbm_load_unit[] = "LoadUnit";
69 static const char sdbp_active_state[] = "ActiveState";
70 static const char sdbp_exec_main_pid[] = "ExecMainPID";
72 static const char *sds_state_names[] = {
82 static struct sd_bus *sysbus;
83 static struct sd_bus *usrbus;
86 * Translate systemd errors to errno errors
88 static int seterrno(int value)
94 static int sderr2errno(int rc)
96 return rc < 0 ? seterrno(-rc) : rc;
99 static int errno2sderr(int rc)
101 return rc < 0 ? -errno : rc;
105 * Returns in 'ret' either the system bus (if isuser==0)
106 * or the user bus (if isuser!=0).
107 * Returns 0 in case of success or -1 in case of error
109 int systemd_get_bus(int isuser, struct sd_bus **ret)
114 bus = isuser ? usrbus : sysbus;
118 rc = sd_bus_default_user(ret);
123 rc = sd_bus_default_system(ret);
130 return sderr2errno(rc);
133 void systemd_set_bus(int isuser, struct sd_bus *bus)
135 struct sd_bus **target = isuser ? &usrbus : &sysbus;
137 sd_bus_unref(*target);
142 /********************************************************************
143 * routines for escaping unit names to compute dbus path of units
144 *******************************************************************/
146 * Should the char 'c' be escaped?
148 static inline int should_escape_for_path(char c)
151 return c <= (c >= 'a' ? 'z' : 'Z');
153 return c >= '0' && c <= '9';
158 * ascii char for the hexadecimal digit 'x'
160 static inline char d2h(int x)
162 return (char)(x + (x > 9 ? ('a' - 10) : '0'));
166 * escapes in 'path' of 'pathlen' the 'unit'
167 * returns 0 in case of success or -1 in case of error
169 static int unit_escape_for_path(char *path, size_t pathlen, const char *unit)
176 if (should_escape_for_path(c)) {
177 if (w + 2 >= pathlen)
180 path[w++] = d2h((c >> 4) & 15);;
181 path[w++] = d2h(c & 15);
194 return seterrno(ENAMETOOLONG);
198 /********************************************************************
199 * Routines for getting paths
200 *******************************************************************/
202 static char *get_dpath(struct sd_bus_message *msg)
208 rc = sd_bus_message_read_basic(msg, 'o', &reply);
209 rc = sderr2errno(rc);
213 result = strdup(reply);
217 sd_bus_message_unref(msg);
221 static char *get_unit_dpath(struct sd_bus *bus, const char *unit, int load)
224 struct sd_bus_message *ret = NULL;
225 sd_bus_error err = SD_BUS_ERROR_NULL;
227 rc = sd_bus_call_method(bus, sdb_destination, sdb_path, sdbi_manager, load ? sdbm_load_unit : sdbm_get_unit, &err, &ret, "s", unit);
231 return get_dpath(ret);
233 sd_bus_message_unref(ret);
237 static char *get_unit_dpath_by_pid(struct sd_bus *bus, unsigned pid)
240 struct sd_bus_message *ret = NULL;
241 sd_bus_error err = SD_BUS_ERROR_NULL;
243 rc = sd_bus_call_method(bus, sdb_destination, sdb_path, sdbi_manager, sdbm_get_unit_by_pid, &err, &ret, "u", pid);
247 return get_dpath(ret);
249 sd_bus_message_unref(ret);
253 static int unit_pid(struct sd_bus *bus, const char *dpath)
257 sd_bus_error err = SD_BUS_ERROR_NULL;
259 rc = sd_bus_get_property_trivial(bus, sdb_destination, dpath, sdbi_service, sdbp_exec_main_pid, &err, 'u', &u);
260 return rc < 0 ? rc : (int)u;
263 static enum SysD_State unit_state(struct sd_bus *bus, const char *dpath)
267 enum SysD_State resu;
268 sd_bus_error err = SD_BUS_ERROR_NULL;
270 resu = SysD_State_INVALID;
271 rc = sd_bus_get_property_string(bus, sdb_destination, dpath, sdbi_unit, sdbp_active_state, &err, &st);
277 if (!strcmp(st, sds_state_names[SysD_State_Active]))
278 resu = SysD_State_Active;
279 else if (!strcmp(st, sds_state_names[SysD_State_Activating]))
280 resu = SysD_State_Activating;
283 if (!strcmp(st, sds_state_names[SysD_State_Deactivating]))
284 resu = SysD_State_Deactivating;
287 if (!strcmp(st, sds_state_names[SysD_State_Failed]))
288 resu = SysD_State_Failed;
291 if (!strcmp(st, sds_state_names[SysD_State_Inactive]))
292 resu = SysD_State_Inactive;
295 if (!strcmp(st, sds_state_names[SysD_State_Reloading]))
296 resu = SysD_State_Reloading;
308 static int unit_start(struct sd_bus *bus, const char *dpath)
311 struct sd_bus_message *ret = NULL;
312 sd_bus_error err = SD_BUS_ERROR_NULL;
314 rc = sd_bus_call_method(bus, sdb_destination, dpath, sdbi_unit, sdbm_start, &err, &ret, "s", "replace");
315 sd_bus_message_unref(ret);
319 static int unit_restart(struct sd_bus *bus, const char *dpath)
322 struct sd_bus_message *ret = NULL;
323 sd_bus_error err = SD_BUS_ERROR_NULL;
325 rc = sd_bus_call_method(bus, sdb_destination, dpath, sdbi_unit, sdbm_restart, &err, &ret, "s", "replace");
326 sd_bus_message_unref(ret);
330 static int unit_stop(struct sd_bus *bus, const char *dpath)
333 struct sd_bus_message *ret = NULL;
334 sd_bus_error err = SD_BUS_ERROR_NULL;
336 rc = sd_bus_call_method(bus, sdb_destination, dpath, sdbi_unit, sdbm_stop, &err, &ret, "s", "replace");
337 sd_bus_message_unref(ret);
341 static int unit_start_name(struct sd_bus *bus, const char *name)
344 struct sd_bus_message *ret = NULL;
345 sd_bus_error err = SD_BUS_ERROR_NULL;
347 rc = sd_bus_call_method(bus, sdb_destination, sdb_path, sdbi_manager, sdbm_start_unit, &err, &ret, "ss", name, "replace");
348 sd_bus_message_unref(ret);
352 static int unit_restart_name(struct sd_bus *bus, const char *name)
355 struct sd_bus_message *ret = NULL;
356 sd_bus_error err = SD_BUS_ERROR_NULL;
358 rc = sd_bus_call_method(bus, sdb_destination, sdb_path, sdbi_manager, sdbm_restart_unit, &err, &ret, "ss", name, "replace");
359 sd_bus_message_unref(ret);
363 static int unit_stop_name(struct sd_bus *bus, const char *name)
366 struct sd_bus_message *ret = NULL;
367 sd_bus_error err = SD_BUS_ERROR_NULL;
369 rc = sd_bus_call_method(bus, sdb_destination, sdb_path, sdbi_manager, sdbm_stop_unit, &err, &ret, "ss", name, "replace");
370 sd_bus_message_unref(ret);
374 /********************************************************************
376 *******************************************************************/
378 static int check_snprintf_result(int rc, size_t buflen)
380 return (rc >= 0 && (size_t)rc >= buflen) ? seterrno(ENAMETOOLONG) : rc;
383 int systemd_get_units_dir(char *path, size_t pathlen, int isuser)
385 int rc = snprintf(path, pathlen, "%s/%s",
387 isuser ? "user" : "system");
389 return check_snprintf_result(rc, pathlen);
392 int systemd_get_unit_path(char *path, size_t pathlen, int isuser, const char *unit, const char *uext)
394 int rc = snprintf(path, pathlen, "%s/%s/%s.%s",
396 isuser ? "user" : "system",
400 return check_snprintf_result(rc, pathlen);
403 int systemd_get_wants_path(char *path, size_t pathlen, int isuser, const char *wanter, const char *unit, const char *uext)
405 int rc = snprintf(path, pathlen, "%s/%s/%s.wants/%s.%s",
407 isuser ? "user" : "system",
412 return check_snprintf_result(rc, pathlen);
415 int systemd_get_wants_target(char *path, size_t pathlen, const char *unit, const char *uext)
417 int rc = snprintf(path, pathlen, "../%s.%s", unit, uext);
419 return check_snprintf_result(rc, pathlen);
422 int systemd_daemon_reload(int isuser)
426 struct sd_bus_message *ret = NULL;
427 sd_bus_error err = SD_BUS_ERROR_NULL;
429 rc = systemd_get_bus(isuser, &bus);
431 /* TODO: asynchronous bind... */
432 /* TODO: more diagnostic... */
433 rc = sd_bus_call_method(bus, sdb_destination, sdb_path, sdbi_manager, sdbm_reload, &err, &ret, NULL);
434 sd_bus_message_unref(ret);
439 int systemd_unit_list(int isuser, int (*callback)(void *closure, const char *name, const char *path, int isuser), void *closure)
442 char path[PATH_MAX + 1];
449 rc = systemd_get_units_dir(path, sizeof path - 1, isuser);
454 /* open the directory */
460 path[offset++] = '/';
462 /* read the directory */
474 if (dent->d_type == DT_REG)
476 else if (dent->d_type != DT_UNKNOWN)
479 rc = fstatat(dirfd(dir), dent->d_name, &st, AT_SYMLINK_NOFOLLOW|AT_NO_AUTOMOUNT);
482 isfile = S_ISREG(st.st_mode);
485 /* calls the callback if is a file */
487 len = strlen(dent->d_name);
488 if (offset + len >= sizeof path) {
489 rc = seterrno(ENAMETOOLONG);
492 memcpy(&path[offset], dent->d_name, 1 + len);
493 rc = callback(closure, &path[offset], path, isuser);
502 int systemd_unit_list_all(int (*callback)(void *closure, const char *name, const char *path, int isuser), void *closure)
504 return systemd_unit_list(1, callback, closure) ? : systemd_unit_list(0, callback, closure);
507 char *systemd_unit_dpath_by_name(int isuser, const char *name, int load)
511 return systemd_get_bus(isuser, &bus) < 0 ? NULL : get_unit_dpath(bus, name, load);
514 char *systemd_unit_dpath_by_pid(int isuser, unsigned pid)
518 return systemd_get_bus(isuser, &bus) < 0 ? NULL : get_unit_dpath_by_pid(bus, pid);
521 int systemd_unit_start_dpath(int isuser, const char *dpath)
526 rc = systemd_get_bus(isuser, &bus);
527 return rc < 0 ? rc : unit_start(bus, dpath);
530 int systemd_unit_restart_dpath(int isuser, const char *dpath)
535 rc = systemd_get_bus(isuser, &bus);
536 return rc < 0 ? rc : unit_restart(bus, dpath);
539 int systemd_unit_stop_dpath(int isuser, const char *dpath)
544 rc = systemd_get_bus(isuser, &bus);
545 return rc < 0 ? rc : unit_stop(bus, dpath);
548 int systemd_unit_start_name(int isuser, const char *name)
553 rc = systemd_get_bus(isuser, &bus);
555 rc = unit_start_name(bus, name);
559 int systemd_unit_restart_name(int isuser, const char *name)
564 rc = systemd_get_bus(isuser, &bus);
566 rc = unit_restart_name(bus, name);
570 int systemd_unit_stop_name(int isuser, const char *name)
575 rc = systemd_get_bus(isuser, &bus);
577 rc = unit_stop_name(bus, name);
581 int systemd_unit_stop_pid(int isuser, unsigned pid)
587 rc = systemd_get_bus(isuser, &bus);
589 dpath = get_unit_dpath_by_pid(bus, pid);
593 rc = unit_stop(bus, dpath);
600 int systemd_unit_pid_of_dpath(int isuser, const char *dpath)
605 rc = systemd_get_bus(isuser, &bus);
606 return rc < 0 ? rc : unit_pid(bus, dpath);
609 enum SysD_State systemd_unit_state_of_dpath(int isuser, const char *dpath)
614 rc = systemd_get_bus(isuser, &bus);
615 return rc < 0 ? SysD_State_INVALID : unit_state(bus, dpath);
618 const char *systemd_state_name(enum SysD_State state)
620 return sds_state_names[state];