rCore/kernel/src/shell.rs

//! Kernel shell

use alloc::string::String;
use crate::fs::{ROOT_INODE, INodeExt};
use crate::process::*;
use crate::thread;

pub fn run_user_shell() {
    if let Ok(inode) = ROOT_INODE.lookup("sh") {
        println!("Going to user mode shell.");
        println!("Use 'ls' to list available programs.");
        let data = inode.read_as_vec().unwrap();
        processor().manager().add(Thread::new_user(data.as_slice(), "sh".split(' ')), 0);
    } else {
        processor().manager().add(Thread::new_kernel(shell, 0), 0);
    }
}

pub extern fn shell(_arg: usize) -> ! {
    let files = ROOT_INODE.list().unwrap();
    println!("Available programs: {:?}", files);

    loop {
        print!(">> ");
        let cmd = get_line();
        if cmd == "" {
            continue;
        }
        let name = cmd.split(' ').next().unwrap();
        if let Ok(file) = ROOT_INODE.lookup(name) {
            let data = file.read_as_vec().unwrap();
            let pid = processor().manager().add(Thread::new_user(data.as_slice(), cmd.split(' ')), thread::current().id());
            unsafe { thread::JoinHandle::<()>::_of(pid) }.join().unwrap();
        } else {
            println!("Program not exist");
        }
    }
}

fn get_line() -> String {
    let mut s = String::new();
    loop {
        let c = get_char();
        match c {
            '\u{8}' | '\u{7f}' /* '\b' */ => {
                if s.pop().is_some() {
                    print!("\u{7f}");
                }
            }
            ' '...'\u{7e}' => {
                s.push(c);
                print!("{}", c);
            }
            '\n' | '\r' => {
                print!("\n");
                return s;
            }
            _ => {}
        }
    }
}

fn get_char() -> char {
    crate::fs::STDIN.pop()
}