为什么 fmt::Arguments as_str 在传递参数时不起作用？

Question

我正在使用no_std 为我的操作系统制作一个VGA 打印宏，但由于某种原因无法正常工作。我正在使用vga crate，这样我就不必自己完成所有的 VGA 代码。我有一个名为_print的函数：

use core::fmt::{Arguments, Write};
use vga::colors::Color16;
use vga::writers::{Graphics640x480x16, GraphicsWriter};

pub fn _print(args: Arguments) {
    unsafe {
        let mut i: usize = 0;
        for (_offset, character) in args.as_str().unwrap().chars().enumerate() {
            Mode.draw_character(CurrentTextX + i, CurrentTextY, character, CurrentTextColor);
            i += 8;
        }
    }
}

然后我有一个名为vprint!的宏：

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

我没有使用alloc。我已经看到与此工作完全一样的代码，但由于某种原因，我的代码无法正常工作。它显示我输入的文本，但如果我传递任何参数，它就会恐慌。我做错了什么？

Answer 1

您的代码会出现混乱，因为 as_str() 仅在没有参数的情况下返回 Some。因此，当您立即 unwrap() 它会恐慌，例如您有要格式化的参数。

如果没有要格式化的参数，则获取格式化的字符串。

这可以用来避免在最琐碎的情况下进行分配。

——std::fmt::Arguments::as_str() docs

相反，您可以使用args.to_string() 而不是args.as_str().unwrap()，将其格式化为String。所以不管有没有参数，它实际上都是格式化的。

pub fn _print(args: Arguments) {
    unsafe {
        let mut i: usize = 0;
        for (_offset, character) in args.to_string().chars().enumerate() {
            Mode.draw_character(CurrentTextX + i, CurrentTextY, character, CurrentTextColor);
            i += 8;
        }
    }
}

---

如果你想避免不必要的分配，那么你可以使用unwrap_or_else()，并且只有在as_str()返回None时才使用to_string()。在包装Cow 时，避免将&str 转换为String。

use std::borrow::Cow;

let formatted_str = args
    .as_str()
    .map(Cow::Borrowed)
    .unwrap_or_else(|| Cow::Owned(args.to_string()));

---

no_std

因为您使用的是 no_std 而不是 alloc 板条箱。然后你可以定义一个u8 缓冲区来格式化。例如。使用let buf = [u8; N];，其中N 为格式化字符串提供了足够的空间。

之后，您可以围绕 &mut [u8] 缓冲区实现自己的包装器，该缓冲区实现 core::fmt::Write 特征。您可以将该包装器与core::fmt::write() 一起使用。

您可以使用此答案中的struct WriteTo，而不是自己实现：How can I use the format! macro in a no_std environment?

pub fn _print(args: Arguments) {
    let mut buf = [0u8; 64];
    let formatted_str: &str = write_to::show(&mut buf, args).unwrap();

    unsafe {
        let mut i: usize = 0;
        for (_offset, _character) in formatted_str.chars().enumerate() {
            Mode.draw_character(CurrentTextX + i, CurrentTextY, character, CurrentTextColor);

            i += 8;
        }
    }
}