113 lines
3.6 KiB
Rust
113 lines
3.6 KiB
Rust
use alloc::boxed::Box;
|
|
use alloc::vec::Vec;
|
|
use core::{
|
|
mem::MaybeUninit,
|
|
sync::atomic::{AtomicPtr, AtomicUsize, Ordering},
|
|
};
|
|
use protocol::info::{Info, Severity};
|
|
|
|
pub const LOG_LEVEL: u8 = Severity::Warning as u8;
|
|
// pub const LOG_LEVEL: u8 = Severity::Trace as u8;
|
|
|
|
pub const MAX_LOG_LINE_SIZE: usize = 150;
|
|
const SIZE_OF_LOG_LINE_BUFFER: usize = 1024;
|
|
static mut LOG_LINES: [AtomicPtr<Info>; SIZE_OF_LOG_LINE_BUFFER] =
|
|
unsafe { MaybeUninit::zeroed().assume_init() };
|
|
static START_INDEX: AtomicUsize = unsafe { MaybeUninit::zeroed().assume_init() };
|
|
static END_INDEX: AtomicUsize = unsafe { MaybeUninit::zeroed().assume_init() };
|
|
|
|
pub fn add_line(log_line: Info) {
|
|
let mut index = END_INDEX.fetch_add(1, Ordering::Acquire);
|
|
unsafe {
|
|
index %= SIZE_OF_LOG_LINE_BUFFER;
|
|
let ptr = &mut LOG_LINES[index];
|
|
let line = Box::new(log_line);
|
|
let old = ptr.swap(Box::into_raw(line), Ordering::SeqCst);
|
|
if !old.is_null() {
|
|
_ = Box::from_raw(old);
|
|
}
|
|
}
|
|
}
|
|
|
|
pub fn flush() -> Vec<Info> {
|
|
let mut vec = Vec::new();
|
|
let end_index = END_INDEX.load(Ordering::Acquire);
|
|
let start_index = START_INDEX.load(Ordering::Acquire);
|
|
if end_index <= start_index {
|
|
return vec;
|
|
}
|
|
unsafe {
|
|
let count = end_index - start_index;
|
|
for i in start_index..start_index + count {
|
|
let index = i % SIZE_OF_LOG_LINE_BUFFER;
|
|
let ptr = LOG_LINES[index].swap(core::ptr::null_mut(), Ordering::SeqCst);
|
|
if !ptr.is_null() {
|
|
vec.push(*Box::from_raw(ptr));
|
|
}
|
|
}
|
|
}
|
|
|
|
START_INDEX.store(end_index, Ordering::Release);
|
|
vec
|
|
}
|
|
|
|
#[macro_export]
|
|
macro_rules! log_internal {
|
|
($log_line:expr, $($arg:tt)*) => ({
|
|
use core::fmt::Write;
|
|
_ = write!($log_line, "{}:{} ", file!(), line!());
|
|
_ = write!($log_line, $($arg)*);
|
|
$crate::logger::add_line($log_line);
|
|
});
|
|
}
|
|
|
|
#[macro_export]
|
|
macro_rules! crit {
|
|
($($arg:tt)*) => ({
|
|
if protocol::info::Severity::Critical as u8 >= $crate::logger::LOG_LEVEL {
|
|
let message = alloc::format!($($arg)*);
|
|
$crate::logger::add_line(protocol::info::Severity::Critical, alloc::format!("{}:{} ", file!(), line!()), message)
|
|
}
|
|
});
|
|
}
|
|
|
|
#[macro_export]
|
|
macro_rules! err {
|
|
($($arg:tt)*) => ({
|
|
if protocol::info::Severity::Error as u8 >= $crate::logger::LOG_LEVEL {
|
|
let mut log_line = protocol::info::log_line(protocol::info::Severity::Error, $crate::logger::MAX_LOG_LINE_SIZE);
|
|
$crate::log_internal!(log_line, $($arg)*);
|
|
}
|
|
});
|
|
}
|
|
|
|
#[macro_export]
|
|
macro_rules! warn {
|
|
($($arg:tt)*) => ({
|
|
if protocol::info::Severity::Warning as u8 >= $crate::logger::LOG_LEVEL {
|
|
let mut log_line = protocol::info::log_line(protocol::info::Severity::Warning, $crate::logger::MAX_LOG_LINE_SIZE);
|
|
$crate::log_internal!(log_line, $($arg)*);
|
|
}
|
|
});
|
|
}
|
|
|
|
#[macro_export]
|
|
macro_rules! dbg {
|
|
($($arg:tt)*) => ({
|
|
if protocol::info::Severity::Debug as u8 >= $crate::logger::LOG_LEVEL {
|
|
let mut log_line = protocol::info::log_line(protocol::info::Severity::Debug, $crate::logger::MAX_LOG_LINE_SIZE);
|
|
$crate::log_internal!(log_line, $($arg)*);
|
|
}
|
|
});
|
|
}
|
|
|
|
#[macro_export]
|
|
macro_rules! info {
|
|
($($arg:tt)*) => ({
|
|
if protocol::info::Severity::Info as u8 >= $crate::logger::LOG_LEVEL {
|
|
let mut log_line = protocol::info::log_line(protocol::info::Severity::Info, $crate::logger::MAX_LOG_LINE_SIZE);
|
|
$crate::log_internal!(log_line, $($arg)*);
|
|
}
|
|
});
|
|
}
|