|
|
@@ -3,18 +3,14 @@ extern crate nix;
|
|
3
|
3
|
use std::io::{BufRead, Write};
|
|
4
|
4
|
use std::str::FromStr;
|
|
5
|
5
|
use std::env;
|
|
6
|
|
-use std::str::Split;
|
|
7
|
|
-use std::os::unix::io::RawFd;
|
|
8
|
6
|
use std::ffi::{OsString, CString};
|
|
9
|
7
|
use nix::unistd::ForkResult::{Child, Parent};
|
|
10
|
|
-use nix::sys::wait::{wait, WaitStatus};
|
|
11
|
|
-use nix::unistd::dup2;
|
|
12
|
|
-use nix::unistd::close;
|
|
13
|
|
-use nix::unistd::{execvp, fork, pipe, read, write};
|
|
|
8
|
+use nix::sys::wait::wait;
|
|
|
9
|
+use nix::unistd::{dup2, close, execvp, fork, pipe};
|
|
14
|
10
|
|
|
15
|
|
-mod shellerror;
|
|
|
11
|
+mod errortype;
|
|
16
|
12
|
|
|
17
|
|
-use shellerror::ShellError;
|
|
|
13
|
+use errortype::ErrorType;
|
|
18
|
14
|
|
|
19
|
15
|
#[derive(PartialEq, Debug)]
|
|
20
|
16
|
pub enum Command {
|
|
|
@@ -25,7 +21,7 @@ pub enum Command {
|
|
25
|
21
|
}
|
|
26
|
22
|
|
|
27
|
23
|
impl FromStr for Command {
|
|
28
|
|
- type Err = ShellError;
|
|
|
24
|
+ type Err = ErrorType;
|
|
29
|
25
|
fn from_str(s: &str) -> Result<Self, Self::Err> {
|
|
30
|
26
|
|
|
31
|
27
|
let mut parts = s.split_whitespace();
|
|
|
@@ -93,14 +89,14 @@ impl<R: BufRead, W: Write> Shell<R, W> {
|
|
93
|
89
|
}
|
|
94
|
90
|
/// Initializes the Shell Loop.
|
|
95
|
91
|
/// Starts the shell.
|
|
96
|
|
- pub fn start(&mut self) -> Result<(), &str> {
|
|
|
92
|
+ pub fn start(&mut self) -> Result<(), ErrorType> {
|
|
97
|
93
|
self.shell_loop()
|
|
98
|
94
|
}
|
|
99
|
95
|
|
|
100
|
96
|
/// Loops while exit hasn't been called.
|
|
101
|
97
|
/// When *prompt()* returns (the user hit enter) a Command is created through the FromStr-Trait.
|
|
102
|
98
|
/// If the Command-creation succeeds, run the command.
|
|
103
|
|
- fn shell_loop(&mut self) -> Result<(), &str> {
|
|
|
99
|
+ fn shell_loop(&mut self) -> Result<(), ErrorType> {
|
|
104
|
100
|
while !self.should_exit {
|
|
105
|
101
|
if let Ok(Some(line)) = self.prompt() {
|
|
106
|
102
|
let _ = Command::from_str(&line).and_then(|cmd| self.run(cmd));
|
|
|
@@ -111,7 +107,7 @@ impl<R: BufRead, W: Write> Shell<R, W> {
|
|
111
|
107
|
|
|
112
|
108
|
/// Prints the shell prompt.
|
|
113
|
109
|
/// Waits for user input and returns the read line.
|
|
114
|
|
- pub fn prompt(&mut self) -> Result<Option<String>, &str> {
|
|
|
110
|
+ pub fn prompt(&mut self) -> Result<Option<String>, ErrorType> {
|
|
115
|
111
|
match env::current_dir() {
|
|
116
|
112
|
Ok(pwd) => {
|
|
117
|
113
|
let _ = self.writer.write(
|
|
|
@@ -127,61 +123,61 @@ impl<R: BufRead, W: Write> Shell<R, W> {
|
|
127
|
123
|
Err(_) => Ok(None),
|
|
128
|
124
|
}
|
|
129
|
125
|
}
|
|
130
|
|
- Err(_) => return Err("Path Error"),
|
|
|
126
|
+ Err(_) => return Err(ErrorType::PathError),
|
|
131
|
127
|
}
|
|
132
|
128
|
}
|
|
133
|
129
|
|
|
134
|
130
|
/// Runs a command.
|
|
135
|
131
|
/// Currently only `cd` and `exit` are working.
|
|
136
|
|
- fn run(&mut self, command: Command) -> Result<(), ShellError> {
|
|
|
132
|
+ fn run(&mut self, command: Command) -> Result<(), ErrorType> {
|
|
137
|
133
|
|
|
138
|
134
|
match command {
|
|
139
|
135
|
Command::Empty => {},
|
|
140
|
136
|
Command::Program(cmd) => {
|
|
141
|
137
|
|
|
142
|
|
- let mut commands = cmd.split('|');
|
|
|
138
|
+ let commands = cmd.split('|');
|
|
143
|
139
|
let commands_vec: Vec<&str> = commands.collect();
|
|
144
|
|
-
|
|
|
140
|
+
|
|
145
|
141
|
let needs_pipe = commands_vec.len() == 2;
|
|
146
|
142
|
|
|
147
|
|
- if let Ok((reader, writer)) = pipe() {
|
|
148
|
143
|
|
|
149
|
144
|
let fi = fork();
|
|
150
|
145
|
match fi {
|
|
151
|
|
- Ok(Parent { .. }) => {
|
|
152
|
|
- wait();
|
|
153
|
|
- }
|
|
154
|
|
- Ok(Child) => {
|
|
155
|
|
- if let Some(first) = commands_vec.get(0) {
|
|
156
|
|
- if needs_pipe {
|
|
157
|
|
- close(reader).unwrap();
|
|
158
|
|
- dup2(writer, 1).unwrap();
|
|
159
|
|
- }
|
|
160
|
|
- self.execute(first);
|
|
161
|
|
- }
|
|
162
|
|
- }
|
|
163
|
|
- Err(_) => println!("Fatal error: Fork failed"),
|
|
164
|
|
- }
|
|
165
|
|
-
|
|
|
146
|
+ Ok(Parent { child: _child }) => {
|
|
|
147
|
+ let _ = wait();
|
|
166
|
148
|
|
|
167
|
|
- let sec = fork();
|
|
168
|
|
- match sec {
|
|
169
|
|
- Ok(Parent { .. }) => {
|
|
170
|
|
- wait();
|
|
171
|
149
|
}
|
|
172
|
150
|
Ok(Child) => {
|
|
173
|
|
- if let Some(second) = commands_vec.get(1) {
|
|
174
|
|
- close(1).unwrap();
|
|
175
|
|
- close(writer).unwrap();
|
|
176
|
|
- dup2(reader, 0).unwrap();
|
|
177
|
|
- self.execute(second);
|
|
|
151
|
+ if let Ok((reader, writer)) = pipe() {
|
|
|
152
|
+ let sec = fork();
|
|
|
153
|
+ match sec {
|
|
|
154
|
+ Ok(Parent { child: _child }) => {
|
|
|
155
|
+ let _ = wait();
|
|
|
156
|
+ if let Some(second) = commands_vec.get(1) {
|
|
|
157
|
+ match close(writer) {
|
|
|
158
|
+ Ok(_) => {},
|
|
|
159
|
+ Err(_) => return Err(ErrorType::BrokenPipeError),
|
|
|
160
|
+ }
|
|
|
161
|
+ dup2(reader, 0).unwrap();
|
|
|
162
|
+ self.execute(second);
|
|
|
163
|
+ }
|
|
|
164
|
+ }
|
|
|
165
|
+ Ok(Child) => {
|
|
|
166
|
+ if let Some(first) = commands_vec.get(0) {
|
|
|
167
|
+ if needs_pipe {
|
|
|
168
|
+ close(reader).unwrap();
|
|
|
169
|
+ dup2(writer, 1).unwrap();
|
|
|
170
|
+ }
|
|
|
171
|
+ self.execute(first);
|
|
|
172
|
+ }
|
|
|
173
|
+ }
|
|
|
174
|
+ Err(_) => return Err(ErrorType::ForkError),
|
|
|
175
|
+ }
|
|
178
|
176
|
}
|
|
179
|
177
|
}
|
|
180
|
|
- Err(_) => println!("Fatal error: Fork failed"),
|
|
|
178
|
+ Err(_) => return Err(ErrorType::ForkError),
|
|
181
|
179
|
}
|
|
182
|
180
|
|
|
183
|
|
- }
|
|
184
|
|
-
|
|
185
|
181
|
},
|
|
186
|
182
|
Command::Exit => self.should_exit = true,
|
|
187
|
183
|
Command::Cd(_) => {
|
|
|
@@ -192,11 +188,11 @@ impl<R: BufRead, W: Write> Shell<R, W> {
|
|
192
|
188
|
}
|
|
193
|
189
|
|
|
194
|
190
|
fn execute(&self, cmd: &str){
|
|
195
|
|
- let mut parts:Vec<&str> = cmd.split_whitespace().collect();
|
|
|
191
|
+ let parts:Vec<&str> = cmd.split_whitespace().collect();
|
|
196
|
192
|
let args:Vec<CString> = parts.iter().map(|f| CString::new(*f).unwrap()).collect();
|
|
197
|
193
|
let t = args.into_boxed_slice();
|
|
198
|
194
|
|
|
199
|
|
- execvp(&t[0], &t);
|
|
|
195
|
+ execvp(&t[0], &t).unwrap();
|
|
200
|
196
|
}
|
|
201
|
197
|
}
|
|
202
|
198
|
|