From 1df979235a5b9c36f0c6bbd32d80c715325a4009 Mon Sep 17 00:00:00 2001 From: Jonas Maier Date: Mon, 9 Mar 2026 19:01:43 +0100 Subject: export fun client --- src/serialization.rs | 69 ++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 69 insertions(+) create mode 100644 src/serialization.rs (limited to 'src/serialization.rs') diff --git a/src/serialization.rs b/src/serialization.rs new file mode 100644 index 0000000..8dc3d99 --- /dev/null +++ b/src/serialization.rs @@ -0,0 +1,69 @@ +use std::{env, os::unix::ffi::OsStrExt}; + +use crate::BString; + +fn w_u32(out: &mut Vec, x: u32) { + for b in x.to_le_bytes() { + out.push(b); + } +} + +fn w_u8s(out: &mut Vec, b: &[u8]) { + w_u32(out, b.len() as u32); + out.extend_from_slice(b); +} + +fn w_u8ss(out: &mut Vec, b: &[&[u8]]) { + w_u32(out, b.len() as u32); + for c in b { + w_u8s(out, c); + } +} + +pub enum Error { + Overrun, +} + +pub type Result = std::result::Result; + +fn r_u32(buf: &mut &[u8]) -> Result { + if buf.len() < 4 { + Err(Error::Overrun)?; + } + let x = u32::from_le_bytes([buf[0], buf[1], buf[2], buf[3]]); + *buf = &buf[4..]; + Ok(x) +} + +fn r_u8s(buf: &mut &[u8]) -> Result> { + let len = r_u32(buf)? as usize; + if len > buf.len() { + Err(Error::Overrun)?; + } + let x = buf[..len].to_vec(); + *buf = &buf[len..]; + Ok(x) +} + +fn r_u8ss(buf: &mut &[u8]) -> Result>> { + let len = r_u32(buf)? as usize; + let mut out = Vec::with_capacity(len); + for _ in 0..len { + out.push(r_u8s(buf)?); + } + Ok(out) +} + +pub fn serialize_cli_args() -> Vec { + let args: Vec> = env::args_os() + .map(|a| a.as_os_str().as_bytes().to_vec()) + .collect(); + let args: Vec<&[u8]> = args.iter().map(|v| v.as_ref()).collect(); + let mut out = Vec::new(); + w_u8ss(&mut out, &args); + out +} + +pub fn deserialize_cli_args(mut buf: &[u8]) -> Result> { + r_u8ss(&mut buf) +} -- cgit v1.2.3