rustubs/kthread/
kshell.rs

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
//! a simple shell...
use crate::io::{back_space, read_key};
use crate::kthread::KThread;
use crate::proc::exec::exec;
use crate::proc::task::Task;
use crate::{fs::*, io};
use alloc::vec::Vec;
use core::str;
pub struct Kshell {}

impl KThread for Kshell {
	fn entry() -> ! {
		let fsarchive = get_archive();
		let files = ustar::iter(fsarchive).collect();
		event_loop(&files);
	}
}

fn event_loop(files: &Vec<File>) -> ! {
	let mut input_buffer = Vec::<u8>::new();
	print!("$ ");
	loop {
		let k = read_key().asc;
		match k {
			0x8 => {
				// backspace
				if input_buffer.pop().is_some() {
					back_space();
				}
			}
			0xa => {
				// enter
				println!();
				let cmd: &str = str::from_utf8(&input_buffer).unwrap();
				handle(cmd, files);
				input_buffer.clear();
				print!("$ ");
			}
			_ => {
				print!("{}", k as char);
				input_buffer.push(k)
			}
		}
	}
}

// a known issue of this lint
#[allow(clippy::ptr_arg)]
fn handle(cmd: &str, files: &Vec<File>) {
	let tokens: Vec<&str> = cmd.split(' ').collect();
	match tokens[0] {
		"ls" => files.iter().for_each(|f| println!("{}", f.hdr.name())),
		"cat" | "readelf" => {
			if tokens.len() < 2 {
				println!("need file name");
				return;
			}
			let file_name = tokens[1];
			if let Some(file) = files.iter().find(|f| f.hdr.name() == file_name)
			{
				cat(file)
			} else {
				println!("{}: no such file or directory", file_name,);
			}
		}
		"" => {
			return;
		}
		"clear" => {
			io::reset_screen();
		}
		"mem" => {
			let mm = &Task::current().unwrap().mm;
			for vma in &mm.vmas {
				println!("{:#?}", vma);
			}
		}
		whatever => {
			exec(whatever);
			let mm = &Task::current().unwrap().mm;
			for vma in &mm.vmas {
				println!("{:#?}", vma);
			}
		}
	}
}