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
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
|
#![cfg(target_os = "linux")]
use std::ptr;
use std::c_str::{CString, ToCStr};
use std::path::Path;
use libc::{c_char, c_void, size_t};
use fcntl::{Fd, OFlag};
use syscall::{syscall, SysPivotRoot};
use errno::{SysResult, SysError, from_ffi};
mod ffi {
use libc::{c_char, c_int};
pub use libc::{close, read, write};
extern {
// duplicate a file descriptor
// doc: http://man7.org/linux/man-pages/man2/dup.2.html
pub fn dup(oldfd: c_int) -> c_int;
pub fn dup2(oldfd: c_int, newfd: c_int) -> c_int;
pub fn dup3(oldfd: c_int, newfd: c_int, flags: c_int) -> c_int;
// change working directory
// doc: http://man7.org/linux/man-pages/man2/chdir.2.html
pub fn chdir(path: *const c_char) -> c_int;
// execute program
// doc: http://man7.org/linux/man-pages/man2/execve.2.html
pub fn execve(filename: *const c_char, argv: *const *const c_char, envp: *const *const c_char) -> c_int;
}
}
#[inline]
pub fn dup(oldfd: Fd) -> SysResult<Fd> {
let res = unsafe { ffi::dup(oldfd) };
if res < 0 {
return Err(SysError::last());
}
Ok(res)
}
#[inline]
pub fn dup2(oldfd: Fd, newfd: Fd) -> SysResult<Fd> {
let res = unsafe { ffi::dup2(oldfd, newfd) };
if res < 0 {
return Err(SysError::last());
}
Ok(res)
}
#[inline]
pub fn dup3(oldfd: Fd, newfd: Fd, flags: OFlag) -> SysResult<Fd> {
let res = unsafe { ffi::dup3(oldfd, newfd, flags.bits()) };
if res < 0 {
return Err(SysError::last());
}
Ok(res)
}
#[inline]
pub fn chdir<S: ToCStr>(path: S) -> SysResult<()> {
let path = path.to_c_str();
let res = unsafe { ffi::chdir(path.as_ptr()) };
if res != 0 {
return Err(SysError::last());
}
return Ok(())
}
#[inline]
pub fn execve(filename: CString, args: &[CString], env: &[CString]) -> SysResult<()> {
let mut args_p: Vec<*const c_char> = args.iter().map(|s| s.as_ptr()).collect();
args_p.push(ptr::null());
let mut env_p: Vec<*const c_char> = env.iter().map(|s| s.as_ptr()).collect();
env_p.push(ptr::null());
let res = unsafe {
ffi::execve(filename.as_ptr(), args_p.as_ptr(), env_p.as_ptr())
};
if res != 0 {
return Err(SysError::last());
}
// Should never reach here
Ok(())
}
pub fn pivot_root(new_root: &Path, put_old: &Path) -> SysResult<()> {
let new_root = new_root.to_c_str();
let put_old = put_old.to_c_str();
let res = unsafe {
syscall(SysPivotRoot, new_root.as_ptr(), put_old.as_ptr())
};
if res != 0 {
return Err(SysError::last());
}
Ok(())
}
pub fn close(fd: Fd) -> SysResult<()> {
let res = unsafe { ffi::close(fd) };
from_ffi(res)
}
pub fn read(fd: Fd, buf: &mut [u8]) -> SysResult<uint> {
let res = unsafe { ffi::read(fd, buf.as_mut_ptr() as *mut c_void, buf.len() as size_t) };
if res < 0 {
return Err(SysError::last());
}
return Ok(res as uint)
}
pub fn write(fd: Fd, buf: &[u8]) -> SysResult<uint> {
let res = unsafe { ffi::write(fd, buf.as_ptr() as *const c_void, buf.len() as size_t) };
if res < 0 {
return Err(SysError::last());
}
return Ok(res as uint)
}
|
