| 1 | |
|
| 2 | |
|
| 3 | |
|
| 4 | | pub struct Entry<'a> { |
| 5 | | pub name: &'a str, |
| 6 | | pub mode: u32, |
| 7 | | pub data: &'a [u8], |
| 8 | | } |
| 9 | |
|
| 10 | 3× | fn header(out: &mut Vec<u8>, ino: u32, mode: u32, size: usize, name: &str) { |
| 11 | 3× | let fields = [ino, mode, 0, 0, 1, 0, size as u32, 0, 0, 0, 0, (name.len() + 1) as u32, 0]; |
| 12 | 3× | out.extend_from_slice(b"070701"); |
| 13 | 39× | for f in fields { |
| 14 | 39× | out.extend_from_slice(format!("{f:08X}").as_bytes()); |
| 15 | 39× | } |
| 16 | 3× | out.extend_from_slice(name.as_bytes()); |
| 17 | 3× | out.push(0); |
| 18 | 3× | pad(out); |
| 19 | 3× | } |
| 20 | |
|
| 21 | 5× | fn pad(out: &mut Vec<u8>) { |
| 22 | 12× | while !out.len().is_multiple_of(4) { |
| 23 | 7× | out.push(0); |
| 24 | 7× | } |
| 25 | 5× | } |
| 26 | |
|
| 27 | 1× | pub fn newc(entries: &[Entry]) -> Vec<u8> { |
| 28 | 1× | let mut out = Vec::new(); |
| 29 | 2× | for (i, e) in entries.iter().enumerate() { |
| 30 | 2× | header(&mut out, i as u32 + 1, 0o100000 | e.mode, e.data.len(), e.name); |
| 31 | 2× | out.extend_from_slice(e.data); |
| 32 | 2× | pad(&mut out); |
| 33 | 2× | } |
| 34 | 1× | header(&mut out, 0, 0, 0, "TRAILER!!!"); |
| 35 | 1× | out |
| 36 | 1× | } |
| 37 | |
|
| 38 | | #[cfg(test)] |
| 39 | | mod tests { |
| 40 | | use super::*; |
| 41 | |
|
| 42 | | #[test] |
| 43 | 1× | fn system_cpio_reads_it() { |
| 44 | 1× | let dir = tempfile::tempdir().unwrap(); |
| 45 | 1× | let archive = newc(&[Entry { name: "init", mode: 0o755, data: b"#!x\n" }, Entry { name: "job.json", mode: 0o644, data: b"{}" }]); |
| 46 | 1× | std::fs::write(dir.path().join("a.cpio"), &archive).unwrap(); |
| 47 | 1× | let listed = std::process::Command::new("cpio").args(["-it", "--quiet", "-F"]).arg(dir.path().join("a.cpio")).output(); |
| 48 | 1× | if let Ok(o) = listed { |
| 49 | 0× | if o.status.success() { |
| 50 | 0× | assert_eq!(String::from_utf8_lossy(&o.stdout), "init\njob.json\n"); |
| 51 | 0× | } |
| 52 | 1× | } |
| 53 | 1× | assert!(archive.len().is_multiple_of(4)); |
| 54 | 355× | assert!(archive.windows(10).any(|w| w == b"TRAILER!!!")); |
| 55 | 1× | } |
| 56 | | } |