From 9c351a36c47ec8d32c8fc108fd7719eda8cdd467 Mon Sep 17 00:00:00 2001
From: Wolfgang Bumiller <w.bumiller@proxmox.com>
Date: Thu, 25 Apr 2019 08:35:01 +0000
Subject: [PATCH] tools/daemon: add sd_notify wrapper

Signed-off-by: Wolfgang Bumiller <w.bumiller@proxmox.com>
---
 src/tools/daemon.rs | 32 ++++++++++++++++++++++++++++++++
 1 file changed, 32 insertions(+)

diff --git a/src/tools/daemon.rs b/src/tools/daemon.rs
index 432756d8..870e5610 100644
--- a/src/tools/daemon.rs
+++ b/src/tools/daemon.rs
@@ -1,6 +1,7 @@
 //! Helpers for daemons/services.
 
 use std::ffi::CString;
+use std::os::raw::{c_char, c_int};
 use std::os::unix::io::{AsRawFd, FromRawFd, IntoRawFd, RawFd};
 use std::os::unix::ffi::OsStrExt;
 use std::panic::UnwindSafe;
@@ -187,3 +188,34 @@ where
        .map_err(|_| ())
     )
 }
+
+#[link(name = "systemd")]
+extern "C" {
+    fn sd_notify(unset_environment: c_int, state: *const c_char) -> c_int;
+}
+
+pub enum SystemdNotify {
+    Ready,
+    Reloading,
+    Stopping,
+    Status(String),
+    MainPid(nix::unistd::Pid),
+}
+
+pub fn systemd_notify(state: SystemdNotify) -> Result<(), Error> {
+    let message = match state {
+        SystemdNotify::Ready => CString::new("READY=1"),
+        SystemdNotify::Reloading => CString::new("RELOADING=1"),
+        SystemdNotify::Stopping => CString::new("STOPPING=1"),
+        SystemdNotify::Status(msg) => CString::new(format!("STATUS={}", msg)),
+        SystemdNotify::MainPid(pid) => CString::new(format!("MAINPID={}", pid)),
+    }?;
+    let rc = unsafe { sd_notify(0, message.as_ptr()) };
+    if rc < 0 {
+        bail!(
+            "systemd_notify failed: {}",
+            std::io::Error::from_raw_os_error(-rc),
+        );
+    }
+    Ok(())
+}