tyler
/
proxmox-backup
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

catar/encoder: detect symlink and regular files

This commit is contained in:
Dietmar Maurer 2018-12-27 14:24:31 +01:00
parent fb8365b79f
commit 45281d4927
1 changed files with 34 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
src/catar/encoder.rs
View File

@ -5,6 +5,10 @@ use super::format_definition::*;
use std::io::Write; use std::io::Write;
use std::os::unix::io::AsRawFd; use std::os::unix::io::AsRawFd;
use std::os::unix::ffi::OsStrExt; use std::os::unix::ffi::OsStrExt;
use std::os::unix::io::RawFd;
// PATH_MAX on Linux is 4096 (NUL byte already included)
const PATH_MAX: usize = 4096;
pub struct CaTarEncoder<W: Write> { pub struct CaTarEncoder<W: Write> {
current_path: std::path::PathBuf, // used for error reporting current_path: std::path::PathBuf, // used for error reporting
@ -46,7 +50,7 @@ impl <W: Write> CaTarEncoder<W> {
Ok(entry) => entry, Ok(entry) => entry,
Err(err) => bail!("readir failed - {}", err), Err(err) => bail!("readir failed - {}", err),
}; };
let mut filename = entry.file_name().to_owned(); let filename = entry.file_name().to_owned();
let name = filename.to_bytes_with_nul(); let name = filename.to_bytes_with_nul();
let name_len = name.len(); let name_len = name.len();
@ -65,18 +69,43 @@ impl <W: Write> CaTarEncoder<W> {
for (filename, stat) in name_list { for (filename, stat) in name_list {
//println!("SORTED {:?}", filename); //println!("SORTED {:?}", filename);
self.current_path.push(std::ffi::OsStr::from_bytes(filename.as_bytes()));
if (stat.st_mode & libc::S_IFDIR) != 0 { if (stat.st_mode & libc::S_IFMT) == libc::S_IFDIR {
let mut dir = nix::dir::Dir::openat( let mut dir = nix::dir::Dir::openat(
rawfd, filename.as_ref(), nix::fcntl::OFlag::O_NOFOLLOW, nix::sys::stat::Mode::empty())?; rawfd, filename.as_ref(), nix::fcntl::OFlag::O_NOFOLLOW, nix::sys::stat::Mode::empty())?;
self.current_path.push(std::ffi::OsStr::from_bytes(filename.as_bytes()));
self.encode_dir(&mut dir)?; self.encode_dir(&mut dir)?;
self.current_path.pop(); } else if (stat.st_mode & libc::S_IFMT) == libc::S_IFREG {
let filefd = nix::fcntl::openat(rawfd, filename.as_ref(), nix::fcntl::OFlag::O_NOFOLLOW, nix::sys::stat::Mode::empty())?;
self.encode_file(filefd);
nix::unistd::close(filefd);
} else if (stat.st_mode & libc::S_IFMT) == libc::S_IFLNK {
let mut buffer = [0u8; PATH_MAX];
let target = nix::fcntl::readlinkat(rawfd, filename.as_ref(), &mut buffer)?;
self.encode_symlink(&target)?;
} else {
bail!("unsupported file type (mode {:o} {:?})", stat.st_mode, self.current_path);
} }
self.current_path.pop();
} }
Ok(()) Ok(())
} }
fn encode_file(&mut self, filefd: RawFd) -> Result<(), Error> {
println!("encode_file: {:?}", self.current_path);
Ok(())
}
fn encode_symlink(&mut self, target: &std::ffi::OsStr) -> Result<(), Error> {
println!("encode_symlink: {:?} -> {:?}", self.current_path, target);
Ok(())
}
} }