lix/nix-rust/src/tarfile.rs

46 lines
1.3 KiB
Rust
Raw Normal View History

2019-09-10 19:55:32 +00:00
use crate::{foreign::Source, Error};
use std::fs;
use std::io;
use std::os::unix::fs::OpenOptionsExt;
use std::path::Path;
use tar::Archive;
pub fn unpack_tarfile(source: Source, dest_dir: &str) -> Result<(), Error> {
let dest_dir = Path::new(dest_dir);
let mut tar = Archive::new(source);
for file in tar.entries()? {
let mut file = file?;
2019-09-10 19:55:32 +00:00
let dest_file = dest_dir.join(file.path()?);
2019-09-10 19:55:32 +00:00
fs::create_dir_all(dest_file.parent().unwrap())?;
2019-09-10 19:55:32 +00:00
match file.header().entry_type() {
tar::EntryType::Directory => {
fs::create_dir(dest_file)?;
2019-09-10 19:55:32 +00:00
}
tar::EntryType::Regular => {
let mode = if file.header().mode()? & libc::S_IXUSR == 0 {
2019-09-10 19:55:32 +00:00
0o666
} else {
0o777
};
let mut f = fs::OpenOptions::new()
.create(true)
.write(true)
.mode(mode)
.open(dest_file)?;
io::copy(&mut file, &mut f)?;
2019-09-10 19:55:32 +00:00
}
tar::EntryType::Symlink => {
std::os::unix::fs::symlink(file.header().link_name()?.unwrap(), dest_file)?;
2019-09-10 19:55:32 +00:00
}
t => return Err(Error::Misc(format!("unsupported tar entry type '{:?}'", t))),
}
}
Ok(())
}