深入解析Phil-opp/blog_os项目：实现屏幕打印功能-优快云博客

本文链接：https://blog.youkuaiyun.com/gitblog_00183/article/details/148375832

深入解析Phil-opp/blog_os项目：实现屏幕打印功能

blog_os Writing an OS in Rust 项目地址: https://gitcode.com/gh_mirrors/bl/blog_os

在操作系统开发中，最基本的任务之一就是实现屏幕输出功能。本文将详细讲解如何在Rust编写的操作系统中实现安全的VGA文本缓冲区打印功能。

VGA文本缓冲区基础

VGA文本缓冲区是显示文本的标准方式，它位于物理地址0xb8000处，具有以下特性：

25行×80列的文本显示区域
每个字符占用16位（2字节）：
- 低8位：ASCII字符编码
- 8-11位：前景色
- 12-14位：背景色
- 15位：闪烁控制位

颜色编码

VGA支持16种颜色，分为标准色和亮色：

| 编号 | 标准色 | 编号+亮位 | 亮色 | |------|-----------|-----------|------------| | 0x0 | 黑色 | 0x8 | 深灰 | | 0x1 | 蓝色 | 0x9 | 浅蓝 | | 0x2 | 绿色 | 0xa | 浅绿 | | 0x3 | 青色 | 0xb | 浅青 | | 0x4 | 红色 | 0xc | 浅红 | | 0x5 | 品红 | 0xd | 粉色 | | 0x6 | 棕色 | 0xe | 黄色 | | 0x7 | 浅灰 | 0xf | 白色 |

Rust模块实现

颜色枚举定义

首先定义颜色枚举，使用Rust的C风格枚举确保每个变体有明确的数值：

#[repr(u8)]
#[derive(Debug, Clone, Copy)]
pub enum Color {
    Black = 0,
    Blue = 1,
    // 其他颜色变体...
    White = 15,
}

repr(u8)属性确保枚举以u8类型存储，derive自动实现了调试、克隆和拷贝特性。

颜色代码结构

颜色代码组合了前景色和背景色：

#[derive(Debug, Clone, Copy)]
struct ColorCode(u8);

impl ColorCode {
    const fn new(foreground: Color, background: Color) -> ColorCode {
        ColorCode((background as u8) << 4 | (foreground as u8))
    }
}

屏幕字符与缓冲区

定义屏幕字符和缓冲区结构：

#[repr(C)]
#[derive(Debug, Clone, Copy)]
struct ScreenChar {
    ascii_character: u8,
    color_code: ColorCode,
}

const BUFFER_HEIGHT: usize = 25;
const BUFFER_WIDTH: usize = 80;

struct Buffer {
    chars: [[Volatile<ScreenChar>; BUFFER_WIDTH]; BUFFER_HEIGHT],
}

使用Volatile包装器确保编译器不会优化掉对缓冲区的写入操作。

写入器实现

基本结构

pub struct Writer {
    column_position: usize,
    color_code: ColorCode,
    buffer: Unique<Buffer>,
}

字节写入

实现单字节写入功能：

impl Writer {
    pub fn write_byte(&mut self, byte: u8) {
        match byte {
            b'\n' => self.new_line(),
            byte => {
                if self.column_position >= BUFFER_WIDTH {
                    self.new_line();
                }
                
                let row = BUFFER_HEIGHT - 1;
                let col = self.column_position;
                
                self.buffer().chars[row][col].write(ScreenChar {
                    ascii_character: byte,
                    color_code: self.color_code,
                });
                
                self.column_position += 1;
            }
        }
    }
    // 其他方法...
}

字符串写入

通过实现fmt::Write特性支持字符串写入：

impl fmt::Write for Writer {
    fn write_str(&mut self, s: &str) -> fmt::Result {
        for byte in s.bytes() {
            self.write_byte(byte);
        }
        Ok(())
    }
}

这使得我们可以使用Rust的格式化宏如write!和writeln!。

换行处理

换行时需要滚动屏幕内容：

fn new_line(&mut self) {
    for row in 1..BUFFER_HEIGHT {
        for col in 0..BUFFER_WIDTH {
            let character = self.buffer().chars[row][col].read();
            self.buffer().chars[row-1][col].write(character);
        }
    }
    self.clear_row(BUFFER_HEIGHT-1);
    self.column_position = 0;
}

全局接口

为了在整个系统中方便地使用打印功能，我们实现全局的打印接口：

pub static WRITER: Mutex<Writer> = Mutex::new(Writer {
    column_position: 0,
    color_code: ColorCode::new(Color::LightGreen, Color::Black),
    buffer: unsafe { Unique::new_unchecked(0xb8000 as *mut _) },
});

#[macro_export]
macro_rules! print {
    ($($arg:tt)*) => ($crate::vga_buffer::_print(format_args!($($arg)*)));
}

#[macro_export]
macro_rules! println {
    () => (print!("\n"));
    ($($arg:tt)*) => (print!("{}\n", format_args!($($arg)*)));
}

pub fn _print(args: fmt::Arguments) {
    use core::fmt::Write;
    WRITER.lock().write_fmt(args).unwrap();
}

这样我们就可以像标准Rust程序一样使用print!和println!宏了。