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
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
extern crate libc;
use std::io;
use std::os::unix::io::{AsRawFd, RawFd};
pub fn copy<S: AsRawFd, D: AsRawFd>(count: usize, source: &mut S, dest: &mut D) -> io::Result<()> {
let mut sendfile = SendFile::new(source, count)?;
loop {
match sendfile.send(dest)? {
0 => return Ok(()),
_ => (),
}
}
}
pub fn copy_callback<CB: FnMut(&mut SendFile, u64), S: AsRawFd, D: AsRawFd>(
count: usize,
source: &mut S,
dest: &mut D,
mut cb: CB,
) -> io::Result<()> {
let mut sendfile = SendFile::new(source, count)?;
loop {
match sendfile.send(dest)? {
0 => return Ok(()),
wrote => cb(&mut sendfile, wrote),
}
}
}
pub struct SendFile {
pub file: RawFd,
pub count: usize,
pub offset: libc::off_t,
}
impl SendFile {
pub fn new<F: AsRawFd>(file: &mut F, count: usize) -> io::Result<Self> {
Ok(Self {
file: file.as_raw_fd(),
count,
offset: 0,
})
}
pub fn count(mut self, count: usize) -> Self {
self.count = if count < 1024 { 1024 } else { count };
self
}
pub fn offset(mut self, offset: libc::off_t) -> Self {
self.offset = offset;
self
}
pub fn send<F: AsRawFd>(&mut self, to: &mut F) -> io::Result<u64> {
let result = unsafe {
libc::sendfile(
to.as_raw_fd(),
self.file,
&mut self.offset as *mut libc::off_t,
self.count,
)
};
if result == -1 {
Err(io::Error::last_os_error())
} else {
Ok(result as u64)
}
}
}