|
|
|
|
@ -3,15 +3,10 @@ use arch::driver::serial::COM1;
|
|
|
|
|
|
|
|
|
|
mod vga_writer;
|
|
|
|
|
|
|
|
|
|
macro_rules! _debug {
|
|
|
|
|
($($arg:tt)*) => ({
|
|
|
|
|
$crate::io::debug(format_args!($($arg)*));
|
|
|
|
|
});
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
macro_rules! debug {
|
|
|
|
|
($fmt:expr) => (_debug!(concat!($fmt, "\n")));
|
|
|
|
|
($fmt:expr, $($arg:tt)*) => (_debug!(concat!($fmt, "\n"), $($arg)*));
|
|
|
|
|
pub fn init() {
|
|
|
|
|
static LOGGER: SimpleLogger = SimpleLogger;
|
|
|
|
|
log::set_logger(&LOGGER).unwrap();
|
|
|
|
|
log::set_max_level(LevelFilter::Debug);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
macro_rules! print {
|
|
|
|
|
@ -25,6 +20,14 @@ macro_rules! println {
|
|
|
|
|
($fmt:expr, $($arg:tt)*) => (print!(concat!($fmt, "\n"), $($arg)*));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// Add escape sequence to print with color in Linux console
|
|
|
|
|
macro_rules! with_color {
|
|
|
|
|
($args: ident, $color: ident) => {{
|
|
|
|
|
let (show, code) = color_to_console_code($color);
|
|
|
|
|
format_args!("{}[{};{}m{}{}[0m", 27 as char, show.clone(), code + 30, $args, 27 as char)
|
|
|
|
|
}};
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
use arch::driver::vga::Color;
|
|
|
|
|
|
|
|
|
|
fn print_in_color(args: fmt::Arguments, color: Color) {
|
|
|
|
|
@ -38,19 +41,17 @@ fn print_in_color(args: fmt::Arguments, color: Color) {
|
|
|
|
|
// TODO: 解决死锁问题
|
|
|
|
|
// 若进程在持有锁时被中断，中断处理程序请求输出，就会死锁
|
|
|
|
|
unsafe{ COM1.force_unlock(); }
|
|
|
|
|
COM1.lock().write_fmt(args).unwrap();
|
|
|
|
|
COM1.lock().write_fmt(with_color!(args, color)).unwrap();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
pub fn print(args: fmt::Arguments) {
|
|
|
|
|
print_in_color(args, Color::LightGray);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
pub fn debug(args: fmt::Arguments) {
|
|
|
|
|
print_in_color(args, Color::LightRed);
|
|
|
|
|
use core::fmt::Write;
|
|
|
|
|
unsafe { COM1.force_unlock(); }
|
|
|
|
|
COM1.lock().write_fmt(args).unwrap();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
pub fn write(fd: usize, base: *const u8, len: usize) -> i32 {
|
|
|
|
|
debug!("write: fd: {}, base: {:?}, len: {:#x}", fd, base, len);
|
|
|
|
|
info!("write: fd: {}, base: {:?}, len: {:#x}", fd, base, len);
|
|
|
|
|
use core::slice;
|
|
|
|
|
use core::str;
|
|
|
|
|
let slice = unsafe { slice::from_raw_parts(base, len) };
|
|
|
|
|
@ -60,7 +61,7 @@ pub fn write(fd: usize, base: *const u8, len: usize) -> i32 {
|
|
|
|
|
|
|
|
|
|
pub fn open(path: *const u8, flags: usize) -> i32 {
|
|
|
|
|
let path = unsafe { from_cstr(path) };
|
|
|
|
|
debug!("open: path: {:?}, flags: {:?}", path, flags);
|
|
|
|
|
info!("open: path: {:?}, flags: {:?}", path, flags);
|
|
|
|
|
match path {
|
|
|
|
|
"stdin:" => 0,
|
|
|
|
|
"stdout:" => 1,
|
|
|
|
|
@ -69,7 +70,7 @@ pub fn open(path: *const u8, flags: usize) -> i32 {
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
pub fn close(fd: usize) -> i32 {
|
|
|
|
|
debug!("close: fd: {:?}", fd);
|
|
|
|
|
info!("close: fd: {:?}", fd);
|
|
|
|
|
0
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@ -77,4 +78,55 @@ pub unsafe fn from_cstr(s: *const u8) -> &'static str {
|
|
|
|
|
use core::{str, slice};
|
|
|
|
|
let len = (0usize..).find(|&i| *s.offset(i as isize) == 0).unwrap();
|
|
|
|
|
str::from_utf8(slice::from_raw_parts(s, len)).unwrap()
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
use log;
|
|
|
|
|
use log::{Record, Level, Metadata, Log, SetLoggerError, LevelFilter};
|
|
|
|
|
|
|
|
|
|
struct SimpleLogger;
|
|
|
|
|
|
|
|
|
|
impl Log for SimpleLogger {
|
|
|
|
|
fn enabled(&self, metadata: &Metadata) -> bool {
|
|
|
|
|
true
|
|
|
|
|
// metadata.level() <= Level::Info
|
|
|
|
|
}
|
|
|
|
|
fn log(&self, record: &Record) {
|
|
|
|
|
if self.enabled(record.metadata()) {
|
|
|
|
|
print_in_color(format_args!("{}\n", record.args()), Color::from(record.level()));
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
fn flush(&self) {}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
impl From<Level> for Color {
|
|
|
|
|
fn from(level: Level) -> Self {
|
|
|
|
|
match level {
|
|
|
|
|
Level::Error => Color::Red,
|
|
|
|
|
Level::Warn => Color::Yellow,
|
|
|
|
|
Level::Info => Color::Blue,
|
|
|
|
|
Level::Debug => Color::LightRed,
|
|
|
|
|
Level::Trace => Color::DarkGray,
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
fn color_to_console_code(color: Color) -> (u8, u8) {
|
|
|
|
|
match color {
|
|
|
|
|
Color::Black => (0, 0),
|
|
|
|
|
Color::Blue => (0, 4),
|
|
|
|
|
Color::Green => (0, 2),
|
|
|
|
|
Color::Cyan => (0, 6),
|
|
|
|
|
Color::Red => (0, 1),
|
|
|
|
|
Color::Magenta => (0, 5),
|
|
|
|
|
Color::Brown => (0, 3),
|
|
|
|
|
Color::LightGray => (1, 7),
|
|
|
|
|
Color::DarkGray => (0, 7),
|
|
|
|
|
Color::LightBlue => (1, 4),
|
|
|
|
|
Color::LightGreen => (1, 2),
|
|
|
|
|
Color::LightCyan => (1, 6),
|
|
|
|
|
Color::LightRed => (1, 1),
|
|
|
|
|
Color::Pink => (1, 5),
|
|
|
|
|
Color::Yellow => (1, 3),
|
|
|
|
|
Color::White => (1, 0),
|
|
|
|
|
}
|
|
|
|
|
}
|
WangRunji
7 years ago