1 extern crate klog_types; 2 3 use core::sync::atomic::{compiler_fence, Ordering}; 4 5 use klog_types::{AllocatorLog, AllocatorLogType, LogSource, MMLogChannel}; 6 7 use crate::{arch::CurrentTimeArch, libs::spinlock::SpinLock, process::Pid, time::TimeArch}; 8 9 /// 全局的内存分配器日志通道 10 /// 11 /// 标记为`no_mangle`是为了让调试器能够找到这个变量 12 #[no_mangle] 13 static __MM_ALLOCATOR_LOG_CHANNEL: MMLogChannel<{ MMDebugLogManager::MAX_ALLOC_LOG_NUM }> = 14 MMLogChannel::new(MMDebugLogManager::MAX_ALLOC_LOG_NUM); 15 16 /// 全局的内存分配器日志id分配器 17 /// 18 /// id从1开始, 因为0是无效的id 19 static __MM_DEBUG_LOG_IDA: SpinLock<ida::IdAllocator> = 20 SpinLock::new(ida::IdAllocator::new(1, usize::MAX).unwrap()); 21 22 /// 记录内存分配器的日志 23 /// 24 /// ## 参数 25 /// 26 /// - `log_type`：日志类型 27 /// - `source`：日志来源 28 pub fn mm_debug_log(_log_type: AllocatorLogType, _source: LogSource) { 29 // todo: 由于目前底层的thingbuf存在卡死的问题，因此这里暂时注释掉。 30 // let pid = if unlikely(!ProcessManager::initialized()) { 31 // Some(Pid::new(0)) 32 // } else { 33 // Some(ProcessManager::current_pcb().pid()) 34 // }; 35 // MMDebugLogManager::log(log_type, source, pid); 36 } 37 38 #[derive(Debug)] 39 pub(super) struct MMDebugLogManager; 40 41 impl MMDebugLogManager { 42 /// 最大的内存分配器日志数量 43 pub const MAX_ALLOC_LOG_NUM: usize = 10000; 44 45 /// 记录内存分配器的日志 46 /// 47 /// ## 参数 48 /// 49 /// - `log_type`：日志类型 50 /// - `source`：日志来源 51 /// - `pid`：日志来源的pid 52 #[allow(dead_code)] 53 pub fn log(log_type: AllocatorLogType, source: LogSource, pid: Option<Pid>) { 54 let id = __MM_DEBUG_LOG_IDA.lock_irqsave().alloc().unwrap(); 55 let log = AllocatorLog::new( 56 id as u64, 57 log_type, 58 source, 59 pid.map(|p| p.data()), 60 CurrentTimeArch::get_cycles() as u64, 61 ); 62 63 let mut log = log; 64 loop { 65 compiler_fence(Ordering::SeqCst); 66 let r = __MM_ALLOCATOR_LOG_CHANNEL.buf.push(log); 67 compiler_fence(Ordering::SeqCst); 68 if let Err(r) = r { 69 // 如果日志通道满了，就把最早的日志丢弃 70 if __MM_ALLOCATOR_LOG_CHANNEL.buf.remaining() == 0 { 71 compiler_fence(Ordering::SeqCst); 72 __MM_ALLOCATOR_LOG_CHANNEL.buf.pop(); 73 compiler_fence(Ordering::SeqCst); 74 } 75 log = r.into_inner(); 76 compiler_fence(Ordering::SeqCst); 77 } else { 78 break; 79 } 80 } 81 } 82 } 83