all lists on lists.proxmox.com
 help / color / mirror / Atom feed
* [pbs-devel] [PATCH proxmox 1/3] add systemd escape_unit and unescape_unit
@ 2021-09-23 10:09 Dietmar Maurer
  2021-09-23 10:09 ` [pbs-devel] [PATCH proxmox 2/3] add UPID api type Dietmar Maurer
  2021-09-23 10:09 ` [pbs-devel] [PATCH proxmox 3/3] bump proxmox version to 0.13.4-1 Dietmar Maurer
  0 siblings, 2 replies; 3+ messages in thread
From: Dietmar Maurer @ 2021-09-23 10:09 UTC (permalink / raw)
  To: pbs-devel

---
 proxmox/src/tools/mod.rs     |  1 +
 proxmox/src/tools/systemd.rs | 83 ++++++++++++++++++++++++++++++++++++
 2 files changed, 84 insertions(+)
 create mode 100644 proxmox/src/tools/systemd.rs

diff --git a/proxmox/src/tools/mod.rs b/proxmox/src/tools/mod.rs
index b172b09..b3a3cb0 100644
--- a/proxmox/src/tools/mod.rs
+++ b/proxmox/src/tools/mod.rs
@@ -20,6 +20,7 @@ pub mod serde;
 pub mod time;
 pub mod uuid;
 pub mod vec;
+pub mod systemd;
 
 #[cfg(feature = "tfa")]
 pub mod tfa;
diff --git a/proxmox/src/tools/systemd.rs b/proxmox/src/tools/systemd.rs
new file mode 100644
index 0000000..68f8631
--- /dev/null
+++ b/proxmox/src/tools/systemd.rs
@@ -0,0 +1,83 @@
+use anyhow::{bail, Error};
+
+fn parse_hex_digit(d: u8) -> Result<u8, Error> {
+    if d >= b'0' && d <= b'9' {
+        return Ok(d - b'0');
+    }
+    if d >= b'A' && d <= b'F' {
+        return Ok(d - b'A' + 10);
+    }
+    if d >= b'a' && d <= b'f' {
+        return Ok(d - b'a' + 10);
+    }
+    bail!("got invalid hex digit");
+}
+
+/// Escape strings for usage in systemd unit names
+pub fn escape_unit(mut unit: &str, is_path: bool) -> String {
+    if is_path {
+        unit = unit.trim_matches('/');
+        if unit.is_empty() {
+            return String::from("-");
+        }
+    }
+
+    let unit = unit.as_bytes();
+
+    let mut escaped = String::new();
+
+    for (i, c) in unit.iter().enumerate() {
+        if *c == b'/' {
+            escaped.push('-');
+            continue;
+        }
+        if (i == 0 && *c == b'.')
+            || !(*c == b'_'
+                || *c == b'.'
+                || (*c >= b'0' && *c <= b'9')
+                || (*c >= b'A' && *c <= b'Z')
+                || (*c >= b'a' && *c <= b'z'))
+        {
+            escaped.push_str(&format!("\\x{:0x}", c));
+        } else {
+            escaped.push(*c as char);
+        }
+    }
+    escaped
+}
+
+/// Unescape strings used in systemd unit names
+pub fn unescape_unit(text: &str) -> Result<String, Error> {
+    let mut i = text.as_bytes();
+
+    let mut data: Vec<u8> = Vec::new();
+
+    loop {
+        if i.is_empty() {
+            break;
+        }
+        let next = i[0];
+        if next == b'\\' {
+            if i.len() < 4 {
+                bail!("short input");
+            }
+            if i[1] != b'x' {
+                bail!("unkwnown escape sequence");
+            }
+            let h1 = parse_hex_digit(i[2])?;
+            let h0 = parse_hex_digit(i[3])?;
+            data.push(h1 << 4 | h0);
+            i = &i[4..]
+        } else if next == b'-' {
+            data.push(b'/');
+            i = &i[1..]
+        } else {
+            data.push(next);
+            i = &i[1..]
+        }
+    }
+
+    let text = String::from_utf8(data)?;
+
+    Ok(text)
+}
-- 
2.30.2





^ permalink raw reply	[flat|nested] 3+ messages in thread
end of thread, other threads:[~2021-09-23 10:10 UTC | newest]

Thread overview: 3+ messages (download: mbox.gz / follow: Atom feed)
-- links below jump to the message on this page --
2021-09-23 10:09 [pbs-devel] [PATCH proxmox 1/3] add systemd escape_unit and unescape_unit Dietmar Maurer
2021-09-23 10:09 ` [pbs-devel] [PATCH proxmox 2/3] add UPID api type Dietmar Maurer
2021-09-23 10:09 ` [pbs-devel] [PATCH proxmox 3/3] bump proxmox version to 0.13.4-1 Dietmar Maurer

This is an external index of several public inboxes,
see mirroring instructions on how to clone and mirror
all data and code used by this external index.
Service provided by Proxmox Server Solutions GmbH | Privacy | Legal