1 files changed, 58 insertions, 0 deletions
diff --git a/src/helpers/SdDaemon.cpp b/src/helpers/SdDaemon.cpp
new file mode 100644
index 00000000..0835637a
--- /dev/null
+++ b/src/helpers/SdDaemon.cpp
@@ -0,0 +1,58 @@
+#include "SdDaemon.hpp"
+
+#include <memory>
+
+#include <fcntl.h>
+#include <unistd.h>
+#include <sys/socket.h>
+#include <sys/un.h>
+
+namespace Systemd {
+    int SdBooted(void) {
+        if (!faccessat(AT_FDCWD, "/run/systemd/system/", F_OK, AT_SYMLINK_NOFOLLOW))
+            return true;
+
+        if (errno == ENOENT)
+            return false;
+
+        return -errno;
+    }
+
+    int SdNotify(int unsetEnvironment, const char* state) {
+        int fd = socket(AF_UNIX, SOCK_DGRAM, 0);
+        if (fd == -1)
+            return -errno;
+
+        constexpr char envVar[] = "NOTIFY_SOCKET";
+
+        auto           cleanup = [unsetEnvironment, envVar](int* fd) {
+            if (unsetEnvironment)
+                unsetenv(envVar);
+            close(*fd);
+        };
+        std::unique_ptr<int, decltype(cleanup)> fdCleaup(&fd, cleanup);
+
+        const char*                             addr = getenv(envVar);
+        if (!addr)
+            return 0;
+
+        // address length must be at most this; see man 7 unix
+        size_t             addrLen = strnlen(addr, 107);
+
+        struct sockaddr_un unixAddr;
+        unixAddr.sun_family = AF_UNIX;
+        strncpy(unixAddr.sun_path, addr, addrLen);
+        if (unixAddr.sun_path[0] == '@')
+            unixAddr.sun_path[0] = '\0';
+
+        if (!connect(fd, (const sockaddr*)&unixAddr, sizeof(struct sockaddr_un)))
+            return 1;
+
+        // arbitrary value which seems to be enough for s-d messages
+        size_t stateLen = strnlen(state, 128);
+        if (write(fd, state, stateLen) >= 0)
+            return 1;
+
+        return -errno;
+    }
+}