]>
Commit | Line | Data |
---|---|---|
e8edbbd4 DM |
1 | use failure::*; |
2 | ||
3 | use std::thread; | |
4 | use std::os::unix::io::FromRawFd; | |
23bb8780 | 5 | use std::path::PathBuf; |
e8edbbd4 DM |
6 | |
7 | use futures::{Async, Poll}; | |
8 | use futures::stream::Stream; | |
9 | ||
10 | use nix::fcntl::OFlag; | |
11 | use nix::sys::stat::Mode; | |
23bb8780 | 12 | use nix::dir::Dir; |
e8edbbd4 DM |
13 | |
14 | use crate::catar::encoder::*; | |
15 | ||
16 | pub struct CaTarBackupStream { | |
17 | pipe: Option<std::fs::File>, | |
18 | buffer: Vec<u8>, | |
19 | child: Option<thread::JoinHandle<()>>, | |
20 | } | |
21 | ||
22 | impl Drop for CaTarBackupStream { | |
23 | ||
24 | fn drop(&mut self) { | |
25 | drop(self.pipe.take()); | |
26 | self.child.take().unwrap().join().unwrap(); | |
27 | } | |
28 | } | |
29 | ||
30 | impl CaTarBackupStream { | |
31 | ||
23bb8780 | 32 | pub fn new(mut dir: Dir, path: PathBuf) -> Result<Self, Error> { |
e8edbbd4 DM |
33 | let mut buffer = Vec::with_capacity(4096); |
34 | unsafe { buffer.set_len(buffer.capacity()); } | |
35 | ||
36 | let (rx, tx) = nix::unistd::pipe()?; | |
37 | ||
e8edbbd4 DM |
38 | let child = thread::spawn(move|| { |
39 | let mut writer = unsafe { std::fs::File::from_raw_fd(tx) }; | |
40 | if let Err(err) = CaTarEncoder::encode(path, &mut dir, None, &mut writer) { | |
41 | eprintln!("catar encode failed - {}", err); | |
42 | } | |
43 | }); | |
44 | ||
45 | let pipe = unsafe { std::fs::File::from_raw_fd(rx) }; | |
46 | ||
47 | Ok(Self { pipe: Some(pipe), buffer, child: Some(child) }) | |
48 | } | |
23bb8780 DM |
49 | |
50 | pub fn open(dirname: &str) -> Result<Self, Error> { | |
51 | ||
728797d0 | 52 | let dir = nix::dir::Dir::open(dirname, OFlag::O_DIRECTORY, Mode::empty())?; |
23bb8780 DM |
53 | let path = std::path::PathBuf::from(dirname); |
54 | ||
55 | Self::new(dir, path) | |
56 | } | |
e8edbbd4 DM |
57 | } |
58 | ||
59 | impl Stream for CaTarBackupStream { | |
60 | ||
61 | type Item = Vec<u8>; | |
62 | type Error = Error; | |
63 | ||
64 | // Note: This is not async!! | |
65 | ||
66 | fn poll(&mut self) -> Poll<Option<Vec<u8>>, Error> { | |
67 | ||
68 | use std::io::Read; | |
69 | ||
23bb8780 | 70 | loop { |
e8edbbd4 DM |
71 | let pipe = match self.pipe { |
72 | Some(ref mut pipe) => pipe, | |
73 | None => unreachable!(), | |
74 | }; | |
75 | match pipe.read(&mut self.buffer) { | |
76 | Ok(n) => { | |
77 | if n == 0 { | |
78 | return Ok(Async::Ready(None)) | |
79 | } else { | |
80 | let data = self.buffer[..n].to_vec(); | |
81 | return Ok(Async::Ready(Some(data))) | |
82 | } | |
83 | } | |
84 | Err(ref e) if e.kind() == std::io::ErrorKind::Interrupted => { | |
85 | // try again | |
86 | } | |
87 | Err(err) => { | |
88 | return Err(err.into()) | |
89 | } | |
90 | }; | |
91 | } | |
92 | } | |
93 | } |