ivi-compositor: Add systemd notify message

[src/agl-compositor.git] / src / systemd-notify.c

diff --git a/src/systemd-notify.c b/src/systemd-notify.c
new file mode 100644 (file)
index 0000000..1d77936
--- /dev/null
+++ b/src/systemd-notify.c
@@ -0,0 +1,193 @@
+/*
+ * Copyright (c) 2015 General Electric Company. All rights reserved.
+ * Copyright © 2020 Collabora, Ltd.
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining
+ * a copy of this software and associated documentation files (the
+ * "Software"), to deal in the Software without restriction, including
+ * without limitation the rights to use, copy, modify, merge, publish,
+ * distribute, sublicense, and/or sell copies of the Software, and to
+ * permit persons to whom the Software is furnished to do so, subject to
+ * the following conditions:
+ *
+ * The above copyright notice and this permission notice (including the
+ * next paragraph) shall be included in all copies or substantial
+ * portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+ * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+ * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+ * NONINFRINGEMENT.  IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
+ * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
+ * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
+ * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+ * SOFTWARE.
+ */
+
+#include <stdlib.h>
+#include <assert.h>
+#include <errno.h>
+
+#include <systemd/sd-daemon.h>
+#include <sys/socket.h>
+#include <wayland-server.h>
+
+#include "ivi-compositor.h"
+#include "shared/helpers.h"
+
+struct systemd_notifier {
+       int watchdog_time;
+       struct wl_event_source *watchdog_source;
+       struct wl_listener compositor_destroy_listener;
+};
+
+static inline bool
+safe_strtoint(const char *str, int32_t *value)
+{
+       long ret;
+       char *end;
+
+       assert(str != NULL);
+
+       errno = 0;
+       ret = strtol(str, &end, 10);
+       if (errno != 0) {
+               return false;
+       } else if (end == str || *end != '\0') {
+               errno = EINVAL;
+               return false;
+       }
+
+       if ((long)((int32_t)ret) != ret) {
+               errno = ERANGE;
+               return false;
+       }
+
+       *value = (int32_t)ret;
+
+       return true;
+}
+
+static int
+add_systemd_sockets(struct weston_compositor *compositor)
+{
+       int fd;
+       int cnt_systemd_sockets;
+       int current_fd = 0;
+
+       cnt_systemd_sockets = sd_listen_fds(1);
+
+       if (cnt_systemd_sockets < 0) {
+               weston_log("sd_listen_fds failed with: %d\n",
+                               cnt_systemd_sockets);
+               return -1;
+       }
+
+       /* socket-based activation not used, return silently */
+       if (cnt_systemd_sockets == 0)
+               return 0;
+
+       while (current_fd < cnt_systemd_sockets) {
+               fd = SD_LISTEN_FDS_START + current_fd;
+
+               if (sd_is_socket(fd, AF_UNIX, SOCK_STREAM, 1) <= 0) {
+                       weston_log("invalid socket provided from systemd\n");
+                       return -1;
+               }
+
+               if (wl_display_add_socket_fd(compositor->wl_display, fd)) {
+                       weston_log("wl_display_add_socket_fd failed"
+                                       "for systemd provided socket\n");
+                       return -1;
+               }
+               current_fd++;
+       }
+
+       weston_log("info: add %d socket(s) provided by systemd\n",
+                       current_fd);
+
+       return current_fd;
+}
+
+static int
+watchdog_handler(void *data)
+{
+       struct systemd_notifier *notifier = data;
+
+       wl_event_source_timer_update(notifier->watchdog_source,
+                       notifier->watchdog_time);
+
+       sd_notify(0, "WATCHDOG=1");
+
+       return 1;
+}
+
+static void
+weston_compositor_destroy_listener(struct wl_listener *listener, void *data)
+{
+       struct systemd_notifier *notifier;
+
+       sd_notify(0, "STOPPING=1");
+
+       notifier = container_of(listener, struct systemd_notifier,
+                               compositor_destroy_listener);
+
+       if (notifier->watchdog_source)
+               wl_event_source_remove(notifier->watchdog_source);
+
+       wl_list_remove(&notifier->compositor_destroy_listener.link);
+       free(notifier);
+}
+
+int
+ivi_agl_systemd_notify(struct ivi_compositor *ivi)
+{
+       struct weston_compositor *compositor = ivi->compositor;
+       char *watchdog_time_env;
+       struct wl_event_loop *loop;
+       int32_t watchdog_time_conv;
+
+       struct systemd_notifier *notifier;
+
+       notifier = zalloc(sizeof *notifier);
+       if (notifier == NULL)
+               return -1;
+
+       notifier->compositor_destroy_listener.notify =
+               weston_compositor_destroy_listener;
+       wl_signal_add(&compositor->destroy_signal,
+                       &notifier->compositor_destroy_listener);
+
+       if (add_systemd_sockets(compositor) < 0)
+               return -1;
+
+       weston_log("Sending ready to systemd\n");
+
+       sd_notify(0, "READY=1");
+
+       /* 'WATCHDOG_USEC' is environment variable that is set
+        * by systemd to transfer 'WatchdogSec' watchdog timeout
+        * setting from service file.*/
+       watchdog_time_env = getenv("WATCHDOG_USEC");
+       if (!watchdog_time_env)
+               return 0;
+
+       if (!safe_strtoint(watchdog_time_env, &watchdog_time_conv))
+               return 0;
+
+       /* Convert 'WATCHDOG_USEC' to milliseconds and notify
+        * systemd every half of that time.*/
+       watchdog_time_conv /= 1000 * 2;
+       if (watchdog_time_conv <= 0)
+               return 0;
+
+       notifier->watchdog_time = watchdog_time_conv;
+
+       loop = wl_display_get_event_loop(compositor->wl_display);
+       notifier->watchdog_source =
+               wl_event_loop_add_timer(loop, watchdog_handler, notifier);
+       wl_event_source_timer_update(notifier->watchdog_source,
+                       notifier->watchdog_time);
+
+       return 0;
+}