onelf-rt 0.2.3

Runtime stub for onelf packed binaries
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30

//! Anonymous in-memory execution via `memfd_create`.
//!
//! For small, self-contained entrypoints (no shared library dependencies),
//! the payload is decompressed directly into a memfd and executed without
//! touching the filesystem.

use std::io::{self, Write};
use std::os::unix::io::AsRawFd;
use std::os::unix::process::CommandExt;
use std::process::Command;

use rustix::fs::MemfdFlags;

pub fn execute_memfd(data: &[u8], argv0: &str, args: &[String]) -> io::Result<()> {
    let fd = rustix::fs::memfd_create(c"onelf", MemfdFlags::empty())
        .map_err(|e| io::Error::new(io::ErrorKind::Other, format!("memfd_create: {e}")))?;

    let mut file = std::fs::File::from(fd);
    file.write_all(data)?;

    let raw_fd = file.as_raw_fd();
    let fd_path = format!("/proc/self/fd/{raw_fd}");

    let err = Command::new(&fd_path).arg0(argv0).args(args).exec();

    Err(io::Error::new(
        io::ErrorKind::Other,
        format!("exec failed: {err}"),
    ))
}