1004e86ffSlogin use super::{_port, hba::HbaCmdTable, virt_2_phys}; 2*b087521eSChiichen use crate::driver::base::block::block_device::{BlockDevice, BlockId}; 3*b087521eSChiichen use crate::driver::base::block::disk_info::Partition; 4*b087521eSChiichen use crate::driver::base::block::SeekFrom; 5*b087521eSChiichen use crate::driver::base::device::{Device, DeviceType, KObject}; 6004e86ffSlogin use crate::driver::disk::ahci::HBA_PxIS_TFES; 7004e86ffSlogin use crate::filesystem::mbr::MbrDiskPartionTable; 8660a04ceSlogin use crate::include::bindings::bindings::verify_area; 9004e86ffSlogin 10*b087521eSChiichen use crate::kdebug; 11004e86ffSlogin use crate::libs::{spinlock::SpinLock, vec_cursor::VecCursor}; 12004e86ffSlogin use crate::mm::phys_2_virt; 13676b8ef6SMork use crate::syscall::SystemError; 14004e86ffSlogin use crate::{ 15004e86ffSlogin driver::disk::ahci::hba::{ 16004e86ffSlogin FisRegH2D, FisType, HbaCmdHeader, ATA_CMD_READ_DMA_EXT, ATA_CMD_WRITE_DMA_EXT, 17004e86ffSlogin ATA_DEV_BUSY, ATA_DEV_DRQ, 18004e86ffSlogin }, 19004e86ffSlogin kerror, 20004e86ffSlogin }; 21004e86ffSlogin 22004e86ffSlogin use alloc::sync::Weak; 23004e86ffSlogin use alloc::{string::String, sync::Arc, vec::Vec}; 24004e86ffSlogin 25004e86ffSlogin use core::fmt::Debug; 26004e86ffSlogin use core::sync::atomic::compiler_fence; 27004e86ffSlogin use core::{mem::size_of, ptr::write_bytes}; 28004e86ffSlogin 29004e86ffSlogin /// @brief: 只支持MBR分区格式的磁盘结构体 30004e86ffSlogin pub struct AhciDisk { 31004e86ffSlogin pub name: String, 32004e86ffSlogin pub flags: u16, // 磁盘的状态flags 33004e86ffSlogin pub partitions: Vec<Arc<Partition>>, // 磁盘分区数组 34004e86ffSlogin // port: &'static mut HbaPort, // 控制硬盘的端口 35004e86ffSlogin pub ctrl_num: u8, 36004e86ffSlogin pub port_num: u8, 37004e86ffSlogin /// 指向LockAhciDisk的弱引用 38004e86ffSlogin self_ref: Weak<LockedAhciDisk>, 39004e86ffSlogin } 40004e86ffSlogin 41004e86ffSlogin /// @brief: 带锁的AhciDisk 42004e86ffSlogin #[derive(Debug)] 43004e86ffSlogin pub struct LockedAhciDisk(pub SpinLock<AhciDisk>); 44004e86ffSlogin /// 函数实现 45004e86ffSlogin impl Debug for AhciDisk { 46004e86ffSlogin fn fmt(&self, f: &mut core::fmt::Formatter<'_>) -> core::fmt::Result { 47004e86ffSlogin write!( 48004e86ffSlogin f, 49004e86ffSlogin "{{ name: {}, flags: {}, part_s: {:?} }}", 50004e86ffSlogin self.name, self.flags, self.partitions 51004e86ffSlogin )?; 52004e86ffSlogin return Ok(()); 53004e86ffSlogin } 54004e86ffSlogin } 55004e86ffSlogin 56004e86ffSlogin impl AhciDisk { 57004e86ffSlogin fn read_at( 58004e86ffSlogin &self, 59*b087521eSChiichen lba_id_start: BlockId, // 起始lba编号 60004e86ffSlogin count: usize, // 读取lba的数量 61004e86ffSlogin buf: &mut [u8], 62676b8ef6SMork ) -> Result<usize, SystemError> { 63004e86ffSlogin compiler_fence(core::sync::atomic::Ordering::SeqCst); 64004e86ffSlogin let check_length = ((count - 1) >> 4) + 1; // prdt length 65004e86ffSlogin if count * 512 > buf.len() || check_length > u16::MAX as usize { 66004e86ffSlogin kerror!("ahci read: e2big"); 67004e86ffSlogin // 不可能的操作 68676b8ef6SMork return Err(SystemError::E2BIG); 69004e86ffSlogin } else if count == 0 { 70004e86ffSlogin return Ok(0); 71004e86ffSlogin } 72004e86ffSlogin 73004e86ffSlogin let port = _port(self.ctrl_num, self.port_num); 74004e86ffSlogin volatile_write!(port.is, u32::MAX); // Clear pending interrupt bits 75004e86ffSlogin 76004e86ffSlogin let slot = port.find_cmdslot().unwrap_or(u32::MAX); 77004e86ffSlogin 78004e86ffSlogin if slot == u32::MAX { 79676b8ef6SMork return Err(SystemError::EIO); 80004e86ffSlogin } 81004e86ffSlogin 82004e86ffSlogin #[allow(unused_unsafe)] 83004e86ffSlogin let cmdheader: &mut HbaCmdHeader = unsafe { 84004e86ffSlogin (phys_2_virt( 85004e86ffSlogin volatile_read!(port.clb) as usize 86004e86ffSlogin + slot as usize * size_of::<HbaCmdHeader>() as usize, 87004e86ffSlogin ) as *mut HbaCmdHeader) 88004e86ffSlogin .as_mut() 89004e86ffSlogin .unwrap() 90004e86ffSlogin }; 91004e86ffSlogin 92004e86ffSlogin volatile_write_bit!( 93004e86ffSlogin cmdheader.cfl, 94004e86ffSlogin (1 << 5) - 1 as u8, 95004e86ffSlogin (size_of::<FisRegH2D>() / size_of::<u32>()) as u8 96004e86ffSlogin ); // Command FIS size 97004e86ffSlogin 98004e86ffSlogin volatile_set_bit!(cmdheader.cfl, 1 << 6, false); // Read/Write bit : Read from device 99004e86ffSlogin volatile_write!(cmdheader.prdtl, check_length as u16); // PRDT entries count 100004e86ffSlogin 101004e86ffSlogin // 设置数据存放地址 102004e86ffSlogin let mut buf_ptr = buf as *mut [u8] as *mut usize as usize; 103660a04ceSlogin 104660a04ceSlogin // 由于目前的内存管理机制无法把用户空间的内存地址转换为物理地址,所以只能先把数据拷贝到内核空间 105660a04ceSlogin // TODO:在内存管理重构后,可以直接使用用户空间的内存地址 106660a04ceSlogin let user_buf = if unsafe { verify_area(buf_ptr as u64, buf.len() as u64) } { 107660a04ceSlogin true 108660a04ceSlogin } else { 109660a04ceSlogin false 110660a04ceSlogin }; 111660a04ceSlogin let mut kbuf = if user_buf { 112660a04ceSlogin let mut x: Vec<u8> = Vec::with_capacity(buf.len()); 113660a04ceSlogin unsafe { 114660a04ceSlogin x.set_len(buf.len()); 115660a04ceSlogin } 116660a04ceSlogin Some(x) 117660a04ceSlogin } else { 118660a04ceSlogin None 119660a04ceSlogin }; 120660a04ceSlogin 121660a04ceSlogin if kbuf.is_some() { 122660a04ceSlogin buf_ptr = kbuf.as_mut().unwrap().as_mut_ptr() as usize; 123660a04ceSlogin } 124660a04ceSlogin 125004e86ffSlogin #[allow(unused_unsafe)] 126004e86ffSlogin let cmdtbl = unsafe { 127004e86ffSlogin (phys_2_virt(volatile_read!(cmdheader.ctba) as usize) as *mut HbaCmdTable) 128004e86ffSlogin .as_mut() 129004e86ffSlogin .unwrap() // 必须使用 as_mut ,得到的才是原来的变量 130004e86ffSlogin }; 131004e86ffSlogin let mut tmp_count = count; 132004e86ffSlogin 133004e86ffSlogin unsafe { 134004e86ffSlogin // 清空整个table的旧数据 135004e86ffSlogin write_bytes(cmdtbl, 0, 1); 136004e86ffSlogin } 137004e86ffSlogin 138004e86ffSlogin // 8K bytes (16 sectors) per PRDT 139004e86ffSlogin for i in 0..((volatile_read!(cmdheader.prdtl) - 1) as usize) { 140004e86ffSlogin volatile_write!(cmdtbl.prdt_entry[i].dba, virt_2_phys(buf_ptr) as u64); 141004e86ffSlogin volatile_write_bit!(cmdtbl.prdt_entry[i].dbc, (1 << 22) - 1, 8 * 1024 - 1); // 数据长度 prdt_entry.dbc 142004e86ffSlogin volatile_set_bit!(cmdtbl.prdt_entry[i].dbc, 1 << 31, true); // 允许中断 prdt_entry.i 143004e86ffSlogin buf_ptr += 8 * 1024; 144004e86ffSlogin tmp_count -= 16; 145004e86ffSlogin } 146004e86ffSlogin 147004e86ffSlogin // Last entry 148004e86ffSlogin let las = (volatile_read!(cmdheader.prdtl) - 1) as usize; 149004e86ffSlogin volatile_write!(cmdtbl.prdt_entry[las].dba, virt_2_phys(buf_ptr) as u64); 150004e86ffSlogin volatile_write_bit!( 151004e86ffSlogin cmdtbl.prdt_entry[las].dbc, 152004e86ffSlogin (1 << 22) - 1, 153004e86ffSlogin ((tmp_count << 9) - 1) as u32 154004e86ffSlogin ); // 数据长度 155004e86ffSlogin volatile_set_bit!(cmdtbl.prdt_entry[las].dbc, 1 << 31, true); // 允许中断 156004e86ffSlogin 157004e86ffSlogin // 设置命令 158004e86ffSlogin let cmdfis = unsafe { 159004e86ffSlogin ((&mut cmdtbl.cfis) as *mut [u8] as *mut usize as *mut FisRegH2D) 160004e86ffSlogin .as_mut() 161004e86ffSlogin .unwrap() 162004e86ffSlogin }; 163004e86ffSlogin volatile_write!(cmdfis.fis_type, FisType::RegH2D as u8); 164004e86ffSlogin volatile_set_bit!(cmdfis.pm, 1 << 7, true); // command_bit set 165004e86ffSlogin volatile_write!(cmdfis.command, ATA_CMD_READ_DMA_EXT); 166004e86ffSlogin 167004e86ffSlogin volatile_write!(cmdfis.lba0, (lba_id_start & 0xFF) as u8); 168004e86ffSlogin volatile_write!(cmdfis.lba1, ((lba_id_start >> 8) & 0xFF) as u8); 169004e86ffSlogin volatile_write!(cmdfis.lba2, ((lba_id_start >> 16) & 0xFF) as u8); 170004e86ffSlogin volatile_write!(cmdfis.lba3, ((lba_id_start >> 24) & 0xFF) as u8); 171004e86ffSlogin volatile_write!(cmdfis.lba4, ((lba_id_start >> 32) & 0xFF) as u8); 172004e86ffSlogin volatile_write!(cmdfis.lba5, ((lba_id_start >> 40) & 0xFF) as u8); 173004e86ffSlogin 174004e86ffSlogin volatile_write!(cmdfis.countl, (count & 0xFF) as u8); 175004e86ffSlogin volatile_write!(cmdfis.counth, ((count >> 8) & 0xFF) as u8); 176004e86ffSlogin 177004e86ffSlogin volatile_write!(cmdfis.device, 1 << 6); // LBA Mode 178004e86ffSlogin 179004e86ffSlogin // 等待之前的操作完成 180004e86ffSlogin let mut spin_count = 0; 181004e86ffSlogin const SPIN_LIMIT: u32 = 10000; 182004e86ffSlogin 183004e86ffSlogin while (volatile_read!(port.tfd) as u8 & (ATA_DEV_BUSY | ATA_DEV_DRQ)) > 0 184004e86ffSlogin && spin_count < SPIN_LIMIT 185004e86ffSlogin { 186004e86ffSlogin spin_count += 1; 187004e86ffSlogin } 188004e86ffSlogin 189004e86ffSlogin if spin_count == SPIN_LIMIT { 190004e86ffSlogin kerror!("Port is hung"); 191676b8ef6SMork return Err(SystemError::EIO); 192004e86ffSlogin } 193004e86ffSlogin 194004e86ffSlogin volatile_set_bit!(port.ci, 1 << slot, true); // Issue command 195004e86ffSlogin // kdebug!("To wait ahci read complete."); 196004e86ffSlogin // 等待操作完成 197004e86ffSlogin loop { 198004e86ffSlogin if (volatile_read!(port.ci) & (1 << slot)) == 0 { 199004e86ffSlogin break; 200004e86ffSlogin } 201004e86ffSlogin if (volatile_read!(port.is) & HBA_PxIS_TFES) > 0 { 202004e86ffSlogin kerror!("Read disk error"); 203676b8ef6SMork return Err(SystemError::EIO); 204004e86ffSlogin } 205004e86ffSlogin } 206004e86ffSlogin 207660a04ceSlogin if kbuf.is_some() { 208660a04ceSlogin buf.copy_from_slice(kbuf.as_ref().unwrap()); 209660a04ceSlogin } 210660a04ceSlogin 211004e86ffSlogin compiler_fence(core::sync::atomic::Ordering::SeqCst); 212004e86ffSlogin // successfully read 213004e86ffSlogin return Ok(count * 512); 214004e86ffSlogin } 215004e86ffSlogin 216004e86ffSlogin fn write_at( 217004e86ffSlogin &self, 218*b087521eSChiichen lba_id_start: BlockId, 219004e86ffSlogin count: usize, 220004e86ffSlogin buf: &[u8], 221676b8ef6SMork ) -> Result<usize, SystemError> { 222004e86ffSlogin compiler_fence(core::sync::atomic::Ordering::SeqCst); 223004e86ffSlogin let check_length = ((count - 1) >> 4) + 1; // prdt length 224004e86ffSlogin if count * 512 > buf.len() || check_length > u16::MAX as usize { 225004e86ffSlogin // 不可能的操作 226676b8ef6SMork return Err(SystemError::E2BIG); 227004e86ffSlogin } else if count == 0 { 228004e86ffSlogin return Ok(0); 229004e86ffSlogin } 230004e86ffSlogin 231004e86ffSlogin let port = _port(self.ctrl_num, self.port_num); 232004e86ffSlogin 233004e86ffSlogin volatile_write!(port.is, u32::MAX); // Clear pending interrupt bits 234004e86ffSlogin 235004e86ffSlogin let slot = port.find_cmdslot().unwrap_or(u32::MAX); 236004e86ffSlogin 237004e86ffSlogin if slot == u32::MAX { 238676b8ef6SMork return Err(SystemError::EIO); 239004e86ffSlogin } 240004e86ffSlogin 241004e86ffSlogin compiler_fence(core::sync::atomic::Ordering::SeqCst); 242004e86ffSlogin #[allow(unused_unsafe)] 243004e86ffSlogin let cmdheader: &mut HbaCmdHeader = unsafe { 244004e86ffSlogin (phys_2_virt( 245004e86ffSlogin volatile_read!(port.clb) as usize 246004e86ffSlogin + slot as usize * size_of::<HbaCmdHeader>() as usize, 247004e86ffSlogin ) as *mut HbaCmdHeader) 248004e86ffSlogin .as_mut() 249004e86ffSlogin .unwrap() 250004e86ffSlogin }; 251004e86ffSlogin compiler_fence(core::sync::atomic::Ordering::SeqCst); 252004e86ffSlogin 253004e86ffSlogin volatile_write_bit!( 254004e86ffSlogin cmdheader.cfl, 255004e86ffSlogin (1 << 5) - 1 as u8, 256004e86ffSlogin (size_of::<FisRegH2D>() / size_of::<u32>()) as u8 257004e86ffSlogin ); // Command FIS size 258004e86ffSlogin 259004e86ffSlogin volatile_set_bit!(cmdheader.cfl, 7 << 5, true); // (p,c,w)都设置为1, Read/Write bit : Write from device 260004e86ffSlogin volatile_write!(cmdheader.prdtl, check_length as u16); // PRDT entries count 261004e86ffSlogin 262004e86ffSlogin // 设置数据存放地址 263004e86ffSlogin compiler_fence(core::sync::atomic::Ordering::SeqCst); 264004e86ffSlogin let mut buf_ptr = buf as *const [u8] as *mut usize as usize; 265660a04ceSlogin 266660a04ceSlogin // 由于目前的内存管理机制无法把用户空间的内存地址转换为物理地址,所以只能先把数据拷贝到内核空间 267660a04ceSlogin // TODO:在内存管理重构后,可以直接使用用户空间的内存地址 268660a04ceSlogin let user_buf = if unsafe { verify_area(buf_ptr as u64, buf.len() as u64) } { 269660a04ceSlogin true 270660a04ceSlogin } else { 271660a04ceSlogin false 272660a04ceSlogin }; 273660a04ceSlogin let mut kbuf = if user_buf { 274660a04ceSlogin let mut x: Vec<u8> = Vec::with_capacity(buf.len()); 275660a04ceSlogin x.resize(buf.len(), 0); 276660a04ceSlogin x.copy_from_slice(buf); 277660a04ceSlogin Some(x) 278660a04ceSlogin } else { 279660a04ceSlogin None 280660a04ceSlogin }; 281660a04ceSlogin 282660a04ceSlogin if kbuf.is_some() { 283660a04ceSlogin buf_ptr = kbuf.as_mut().unwrap().as_mut_ptr() as usize; 284660a04ceSlogin } 285660a04ceSlogin 286004e86ffSlogin #[allow(unused_unsafe)] 287004e86ffSlogin let cmdtbl = unsafe { 288004e86ffSlogin (phys_2_virt(volatile_read!(cmdheader.ctba) as usize) as *mut HbaCmdTable) 289004e86ffSlogin .as_mut() 290004e86ffSlogin .unwrap() 291004e86ffSlogin }; 292004e86ffSlogin let mut tmp_count = count; 293004e86ffSlogin compiler_fence(core::sync::atomic::Ordering::SeqCst); 294004e86ffSlogin 295004e86ffSlogin unsafe { 296004e86ffSlogin // 清空整个table的旧数据 297004e86ffSlogin write_bytes(cmdtbl, 0, 1); 298004e86ffSlogin } 299004e86ffSlogin 300004e86ffSlogin // 8K bytes (16 sectors) per PRDT 301004e86ffSlogin for i in 0..((volatile_read!(cmdheader.prdtl) - 1) as usize) { 302004e86ffSlogin volatile_write!(cmdtbl.prdt_entry[i].dba, virt_2_phys(buf_ptr) as u64); 303004e86ffSlogin volatile_write_bit!(cmdtbl.prdt_entry[i].dbc, (1 << 22) - 1, 8 * 1024 - 1); // 数据长度 304004e86ffSlogin volatile_set_bit!(cmdtbl.prdt_entry[i].dbc, 1 << 31, true); // 允许中断 305004e86ffSlogin buf_ptr += 8 * 1024; 306004e86ffSlogin tmp_count -= 16; 307004e86ffSlogin } 308004e86ffSlogin 309004e86ffSlogin // Last entry 310004e86ffSlogin let las = (volatile_read!(cmdheader.prdtl) - 1) as usize; 311004e86ffSlogin volatile_write!(cmdtbl.prdt_entry[las].dba, virt_2_phys(buf_ptr) as u64); 312004e86ffSlogin volatile_set_bit!(cmdtbl.prdt_entry[las].dbc, 1 << 31, true); // 允许中断 313004e86ffSlogin volatile_write_bit!( 314004e86ffSlogin cmdtbl.prdt_entry[las].dbc, 315004e86ffSlogin (1 << 22) - 1, 316004e86ffSlogin ((tmp_count << 9) - 1) as u32 317004e86ffSlogin ); // 数据长度 318004e86ffSlogin 319004e86ffSlogin // 设置命令 320004e86ffSlogin let cmdfis = unsafe { 321004e86ffSlogin ((&mut cmdtbl.cfis) as *mut [u8] as *mut usize as *mut FisRegH2D) 322004e86ffSlogin .as_mut() 323004e86ffSlogin .unwrap() 324004e86ffSlogin }; 325004e86ffSlogin volatile_write!(cmdfis.fis_type, FisType::RegH2D as u8); 326004e86ffSlogin volatile_set_bit!(cmdfis.pm, 1 << 7, true); // command_bit set 327004e86ffSlogin volatile_write!(cmdfis.command, ATA_CMD_WRITE_DMA_EXT); 328004e86ffSlogin 329004e86ffSlogin volatile_write!(cmdfis.lba0, (lba_id_start & 0xFF) as u8); 330004e86ffSlogin volatile_write!(cmdfis.lba1, ((lba_id_start >> 8) & 0xFF) as u8); 331004e86ffSlogin volatile_write!(cmdfis.lba2, ((lba_id_start >> 16) & 0xFF) as u8); 332004e86ffSlogin volatile_write!(cmdfis.lba3, ((lba_id_start >> 24) & 0xFF) as u8); 333004e86ffSlogin volatile_write!(cmdfis.lba4, ((lba_id_start >> 32) & 0xFF) as u8); 334004e86ffSlogin volatile_write!(cmdfis.lba5, ((lba_id_start >> 40) & 0xFF) as u8); 335004e86ffSlogin 336004e86ffSlogin volatile_write!(cmdfis.countl, (count & 0xFF) as u8); 337004e86ffSlogin volatile_write!(cmdfis.counth, ((count >> 8) & 0xFF) as u8); 338004e86ffSlogin 339004e86ffSlogin volatile_write!(cmdfis.device, 1 << 6); // LBA Mode 340004e86ffSlogin 341004e86ffSlogin volatile_set_bit!(port.ci, 1 << slot, true); // Issue command 342004e86ffSlogin 343004e86ffSlogin // 等待操作完成 344004e86ffSlogin loop { 345004e86ffSlogin if (volatile_read!(port.ci) & (1 << slot)) == 0 { 346004e86ffSlogin break; 347004e86ffSlogin } 348004e86ffSlogin if (volatile_read!(port.is) & HBA_PxIS_TFES) > 0 { 349004e86ffSlogin kerror!("Write disk error"); 350676b8ef6SMork return Err(SystemError::EIO); 351004e86ffSlogin } 352004e86ffSlogin } 353004e86ffSlogin 354004e86ffSlogin compiler_fence(core::sync::atomic::Ordering::SeqCst); 355004e86ffSlogin // successfully read 356004e86ffSlogin return Ok(count * 512); 357004e86ffSlogin } 358004e86ffSlogin 359676b8ef6SMork fn sync(&self) -> Result<(), SystemError> { 360004e86ffSlogin // 由于目前没有block cache, 因此sync返回成功即可 361004e86ffSlogin return Ok(()); 362004e86ffSlogin } 363004e86ffSlogin } 364004e86ffSlogin 365004e86ffSlogin impl LockedAhciDisk { 366004e86ffSlogin pub fn new( 367004e86ffSlogin name: String, 368004e86ffSlogin flags: u16, 369004e86ffSlogin ctrl_num: u8, 370004e86ffSlogin port_num: u8, 371676b8ef6SMork ) -> Result<Arc<LockedAhciDisk>, SystemError> { 372004e86ffSlogin let mut part_s: Vec<Arc<Partition>> = Vec::new(); 373004e86ffSlogin 374004e86ffSlogin // 构建磁盘结构体 375004e86ffSlogin let result: Arc<LockedAhciDisk> = Arc::new(LockedAhciDisk(SpinLock::new(AhciDisk { 376004e86ffSlogin name, 377004e86ffSlogin flags, 378004e86ffSlogin partitions: Default::default(), 379004e86ffSlogin ctrl_num, 380004e86ffSlogin port_num, 381004e86ffSlogin self_ref: Weak::default(), 382004e86ffSlogin }))); 383004e86ffSlogin 384004e86ffSlogin let table: MbrDiskPartionTable = result.read_mbr_table()?; 385004e86ffSlogin let weak_this: Weak<LockedAhciDisk> = Arc::downgrade(&result); // 获取this的弱指针 386004e86ffSlogin 387004e86ffSlogin // 求出有多少可用分区 388004e86ffSlogin for i in 0..4 { 389004e86ffSlogin if table.dpte[i].part_type != 0 { 390004e86ffSlogin part_s.push(Partition::new( 391004e86ffSlogin table.dpte[i].starting_sector() as u64, 392004e86ffSlogin table.dpte[i].starting_lba as u64, 393004e86ffSlogin table.dpte[i].total_sectors as u64, 394004e86ffSlogin weak_this.clone(), 395004e86ffSlogin i as u16, 396004e86ffSlogin )); 397004e86ffSlogin } 398004e86ffSlogin } 399004e86ffSlogin result.0.lock().partitions = part_s; 400004e86ffSlogin result.0.lock().self_ref = weak_this; 401004e86ffSlogin return Ok(result); 402004e86ffSlogin } 403004e86ffSlogin 404004e86ffSlogin /// @brief: 从磁盘中读取 MBR 分区表结构体 TODO: Cursor 405676b8ef6SMork pub fn read_mbr_table(&self) -> Result<MbrDiskPartionTable, SystemError> { 406004e86ffSlogin let mut table: MbrDiskPartionTable = Default::default(); 407004e86ffSlogin 408004e86ffSlogin // 数据缓冲区 409004e86ffSlogin let mut buf: Vec<u8> = Vec::new(); 410004e86ffSlogin buf.resize(size_of::<MbrDiskPartionTable>(), 0); 411004e86ffSlogin 412*b087521eSChiichen BlockDevice::read_at(self, 0, 1, &mut buf)?; 413004e86ffSlogin // 创建 Cursor 用于按字节读取 414004e86ffSlogin let mut cursor = VecCursor::new(buf); 415004e86ffSlogin cursor.seek(SeekFrom::SeekCurrent(446))?; 416004e86ffSlogin 417004e86ffSlogin for i in 0..4 { 418*b087521eSChiichen kdebug!("infomation of partition {}:\n", i); 419004e86ffSlogin 420004e86ffSlogin table.dpte[i].flags = cursor.read_u8()?; 421004e86ffSlogin table.dpte[i].starting_head = cursor.read_u8()?; 422004e86ffSlogin table.dpte[i].starting_sector_cylinder = cursor.read_u16()?; 423004e86ffSlogin table.dpte[i].part_type = cursor.read_u8()?; 424004e86ffSlogin table.dpte[i].ending_head = cursor.read_u8()?; 425004e86ffSlogin table.dpte[i].ending_sector_cylingder = cursor.read_u16()?; 426004e86ffSlogin table.dpte[i].starting_lba = cursor.read_u32()?; 427004e86ffSlogin table.dpte[i].total_sectors = cursor.read_u32()?; 428004e86ffSlogin 429*b087521eSChiichen kdebug!("dpte[i] = {:?}", table.dpte[i]); 430004e86ffSlogin } 431004e86ffSlogin table.bs_trailsig = cursor.read_u16()?; 432004e86ffSlogin // kdebug!("bs_trailsig = {}", unsafe { 433004e86ffSlogin // read_unaligned(addr_of!(table.bs_trailsig)) 434004e86ffSlogin // }); 435004e86ffSlogin 436004e86ffSlogin return Ok(table); 437004e86ffSlogin } 438004e86ffSlogin } 439004e86ffSlogin 440*b087521eSChiichen impl KObject for LockedAhciDisk {} 441*b087521eSChiichen 442*b087521eSChiichen impl Device for LockedAhciDisk { 443*b087521eSChiichen fn dev_type(&self) -> DeviceType { 444*b087521eSChiichen return DeviceType::Block; 445*b087521eSChiichen } 446*b087521eSChiichen 447*b087521eSChiichen fn as_any_ref(&self) -> &dyn core::any::Any { 448*b087521eSChiichen return self; 449*b087521eSChiichen } 450*b087521eSChiichen 451*b087521eSChiichen fn id_table(&self) -> crate::driver::base::device::IdTable { 452*b087521eSChiichen todo!() 453*b087521eSChiichen } 454*b087521eSChiichen 455*b087521eSChiichen fn set_sys_info(&self, _sys_info: Option<Arc<dyn crate::filesystem::vfs::IndexNode>>) { 456*b087521eSChiichen todo!() 457*b087521eSChiichen } 458*b087521eSChiichen 459*b087521eSChiichen fn sys_info(&self) -> Option<Arc<dyn crate::filesystem::vfs::IndexNode>> { 460*b087521eSChiichen todo!() 461*b087521eSChiichen } 462*b087521eSChiichen } 463*b087521eSChiichen 464004e86ffSlogin impl BlockDevice for LockedAhciDisk { 465004e86ffSlogin #[inline] 466004e86ffSlogin fn as_any_ref(&self) -> &dyn core::any::Any { 467004e86ffSlogin self 468004e86ffSlogin } 469004e86ffSlogin 470004e86ffSlogin #[inline] 471004e86ffSlogin fn blk_size_log2(&self) -> u8 { 472004e86ffSlogin 9 473004e86ffSlogin } 474004e86ffSlogin 475676b8ef6SMork fn sync(&self) -> Result<(), SystemError> { 476004e86ffSlogin return self.0.lock().sync(); 477004e86ffSlogin } 478004e86ffSlogin 479004e86ffSlogin #[inline] 480*b087521eSChiichen fn device(&self) -> Arc<dyn Device> { 481004e86ffSlogin return self.0.lock().self_ref.upgrade().unwrap(); 482004e86ffSlogin } 483004e86ffSlogin 484004e86ffSlogin fn block_size(&self) -> usize { 485004e86ffSlogin todo!() 486004e86ffSlogin } 487004e86ffSlogin 488004e86ffSlogin fn partitions(&self) -> Vec<Arc<Partition>> { 489004e86ffSlogin return self.0.lock().partitions.clone(); 490004e86ffSlogin } 491*b087521eSChiichen 492*b087521eSChiichen #[inline] 493*b087521eSChiichen fn read_at( 494*b087521eSChiichen &self, 495*b087521eSChiichen lba_id_start: BlockId, // 起始lba编号 496*b087521eSChiichen count: usize, // 读取lba的数量 497*b087521eSChiichen buf: &mut [u8], 498*b087521eSChiichen ) -> Result<usize, SystemError> { 499*b087521eSChiichen self.0.lock().read_at(lba_id_start, count, buf) 500*b087521eSChiichen } 501*b087521eSChiichen 502*b087521eSChiichen #[inline] 503*b087521eSChiichen fn write_at( 504*b087521eSChiichen &self, 505*b087521eSChiichen lba_id_start: BlockId, 506*b087521eSChiichen count: usize, 507*b087521eSChiichen buf: &[u8], 508*b087521eSChiichen ) -> Result<usize, SystemError> { 509*b087521eSChiichen self.0.lock().write_at(lba_id_start, count, buf) 510*b087521eSChiichen } 511004e86ffSlogin } 512