1004e86ffSlogin use super::{_port, hba::HbaCmdTable, virt_2_phys}; 2b087521eSChiichen use crate::driver::base::block::block_device::{BlockDevice, BlockId}; 3b087521eSChiichen use crate::driver::base::block::disk_info::Partition; 4b087521eSChiichen use crate::driver::base::block::SeekFrom; 508a2ee40SLoGin use crate::driver::base::class::Class; 606d5e247SLoGin use crate::driver::base::device::bus::Bus; 7ad1d649eSGnoCiYeH 8a03c4f9dSLoGin use crate::driver::base::device::driver::Driver; 906d5e247SLoGin use crate::driver::base::device::{Device, DeviceType, IdTable}; 1006d5e247SLoGin use crate::driver::base::kobject::{KObjType, KObject, KObjectState}; 1106d5e247SLoGin use crate::driver::base::kset::KSet; 12004e86ffSlogin use crate::driver::disk::ahci::HBA_PxIS_TFES; 13a03c4f9dSLoGin 1406d5e247SLoGin use crate::filesystem::kernfs::KernFSInode; 15004e86ffSlogin use crate::filesystem::mbr::MbrDiskPartionTable; 16004e86ffSlogin 17b087521eSChiichen use crate::kdebug; 1806d5e247SLoGin use crate::libs::rwlock::{RwLockReadGuard, RwLockWriteGuard}; 19004e86ffSlogin use crate::libs::{spinlock::SpinLock, vec_cursor::VecCursor}; 204fda81ceSLoGin use crate::mm::{phys_2_virt, verify_area, VirtAddr}; 21004e86ffSlogin use crate::{ 22004e86ffSlogin driver::disk::ahci::hba::{ 23004e86ffSlogin FisRegH2D, FisType, HbaCmdHeader, ATA_CMD_READ_DMA_EXT, ATA_CMD_WRITE_DMA_EXT, 24004e86ffSlogin ATA_DEV_BUSY, ATA_DEV_DRQ, 25004e86ffSlogin }, 26004e86ffSlogin kerror, 27004e86ffSlogin }; 2891e9d4abSLoGin use system_error::SystemError; 29004e86ffSlogin 30004e86ffSlogin use alloc::sync::Weak; 31004e86ffSlogin use alloc::{string::String, sync::Arc, vec::Vec}; 32004e86ffSlogin 33004e86ffSlogin use core::fmt::Debug; 347ae679ddSLoGin use core::sync::atomic::{compiler_fence, Ordering}; 35004e86ffSlogin use core::{mem::size_of, ptr::write_bytes}; 36004e86ffSlogin 37004e86ffSlogin /// @brief: 只支持MBR分区格式的磁盘结构体 38004e86ffSlogin pub struct AhciDisk { 39004e86ffSlogin pub name: String, 40004e86ffSlogin pub flags: u16, // 磁盘的状态flags 41004e86ffSlogin pub partitions: Vec<Arc<Partition>>, // 磁盘分区数组 42004e86ffSlogin // port: &'static mut HbaPort, // 控制硬盘的端口 43004e86ffSlogin pub ctrl_num: u8, 44004e86ffSlogin pub port_num: u8, 45004e86ffSlogin /// 指向LockAhciDisk的弱引用 46004e86ffSlogin self_ref: Weak<LockedAhciDisk>, 47004e86ffSlogin } 48004e86ffSlogin 49004e86ffSlogin /// @brief: 带锁的AhciDisk 50004e86ffSlogin #[derive(Debug)] 51004e86ffSlogin pub struct LockedAhciDisk(pub SpinLock<AhciDisk>); 52004e86ffSlogin /// 函数实现 53004e86ffSlogin impl Debug for AhciDisk { 54004e86ffSlogin fn fmt(&self, f: &mut core::fmt::Formatter<'_>) -> core::fmt::Result { 55004e86ffSlogin write!( 56004e86ffSlogin f, 57004e86ffSlogin "{{ name: {}, flags: {}, part_s: {:?} }}", 58004e86ffSlogin self.name, self.flags, self.partitions 59004e86ffSlogin )?; 60004e86ffSlogin return Ok(()); 61004e86ffSlogin } 62004e86ffSlogin } 63004e86ffSlogin 64004e86ffSlogin impl AhciDisk { 65004e86ffSlogin fn read_at( 66004e86ffSlogin &self, 67b087521eSChiichen lba_id_start: BlockId, // 起始lba编号 68004e86ffSlogin count: usize, // 读取lba的数量 69004e86ffSlogin buf: &mut [u8], 70676b8ef6SMork ) -> Result<usize, SystemError> { 717ae679ddSLoGin assert!((buf.len() & 511) == 0); 72004e86ffSlogin compiler_fence(core::sync::atomic::Ordering::SeqCst); 73004e86ffSlogin let check_length = ((count - 1) >> 4) + 1; // prdt length 74b5b571e0SLoGin if count * 512 > buf.len() || check_length > 8_usize { 75004e86ffSlogin kerror!("ahci read: e2big"); 76004e86ffSlogin // 不可能的操作 77676b8ef6SMork return Err(SystemError::E2BIG); 78004e86ffSlogin } else if count == 0 { 79004e86ffSlogin return Ok(0); 80004e86ffSlogin } 81004e86ffSlogin 82004e86ffSlogin let port = _port(self.ctrl_num, self.port_num); 83004e86ffSlogin volatile_write!(port.is, u32::MAX); // Clear pending interrupt bits 84004e86ffSlogin 85004e86ffSlogin let slot = port.find_cmdslot().unwrap_or(u32::MAX); 86004e86ffSlogin 87004e86ffSlogin if slot == u32::MAX { 88676b8ef6SMork return Err(SystemError::EIO); 89004e86ffSlogin } 90004e86ffSlogin 91004e86ffSlogin #[allow(unused_unsafe)] 92004e86ffSlogin let cmdheader: &mut HbaCmdHeader = unsafe { 93004e86ffSlogin (phys_2_virt( 94b5b571e0SLoGin volatile_read!(port.clb) as usize + slot as usize * size_of::<HbaCmdHeader>(), 95004e86ffSlogin ) as *mut HbaCmdHeader) 96004e86ffSlogin .as_mut() 97004e86ffSlogin .unwrap() 98004e86ffSlogin }; 99004e86ffSlogin 1007ae679ddSLoGin cmdheader.cfl = (size_of::<FisRegH2D>() / size_of::<u32>()) as u8; 101004e86ffSlogin 102004e86ffSlogin volatile_set_bit!(cmdheader.cfl, 1 << 6, false); // Read/Write bit : Read from device 103004e86ffSlogin volatile_write!(cmdheader.prdtl, check_length as u16); // PRDT entries count 104004e86ffSlogin 105004e86ffSlogin // 设置数据存放地址 106004e86ffSlogin let mut buf_ptr = buf as *mut [u8] as *mut usize as usize; 107660a04ceSlogin 108660a04ceSlogin // 由于目前的内存管理机制无法把用户空间的内存地址转换为物理地址,所以只能先把数据拷贝到内核空间 109660a04ceSlogin // TODO:在内存管理重构后,可以直接使用用户空间的内存地址 1104fda81ceSLoGin 111b5b571e0SLoGin let user_buf = verify_area(VirtAddr::new(buf_ptr), buf.len()).is_ok(); 112660a04ceSlogin let mut kbuf = if user_buf { 113b5b571e0SLoGin let x: Vec<u8> = vec![0; buf.len()]; 114660a04ceSlogin Some(x) 115660a04ceSlogin } else { 116660a04ceSlogin None 117660a04ceSlogin }; 118660a04ceSlogin 119660a04ceSlogin if kbuf.is_some() { 120660a04ceSlogin buf_ptr = kbuf.as_mut().unwrap().as_mut_ptr() as usize; 121660a04ceSlogin } 122660a04ceSlogin 123004e86ffSlogin #[allow(unused_unsafe)] 124004e86ffSlogin let cmdtbl = unsafe { 125004e86ffSlogin (phys_2_virt(volatile_read!(cmdheader.ctba) as usize) as *mut HbaCmdTable) 126004e86ffSlogin .as_mut() 127004e86ffSlogin .unwrap() // 必须使用 as_mut ,得到的才是原来的变量 128004e86ffSlogin }; 129004e86ffSlogin let mut tmp_count = count; 130004e86ffSlogin 131004e86ffSlogin unsafe { 132004e86ffSlogin // 清空整个table的旧数据 133004e86ffSlogin write_bytes(cmdtbl, 0, 1); 134004e86ffSlogin } 1357ae679ddSLoGin // kdebug!("cmdheader.prdtl={}", volatile_read!(cmdheader.prdtl)); 136004e86ffSlogin 137004e86ffSlogin // 8K bytes (16 sectors) per PRDT 138004e86ffSlogin for i in 0..((volatile_read!(cmdheader.prdtl) - 1) as usize) { 139004e86ffSlogin volatile_write!(cmdtbl.prdt_entry[i].dba, virt_2_phys(buf_ptr) as u64); 1407ae679ddSLoGin cmdtbl.prdt_entry[i].dbc = 8 * 1024 - 1; 141004e86ffSlogin volatile_set_bit!(cmdtbl.prdt_entry[i].dbc, 1 << 31, true); // 允许中断 prdt_entry.i 142004e86ffSlogin buf_ptr += 8 * 1024; 143004e86ffSlogin tmp_count -= 16; 144004e86ffSlogin } 145004e86ffSlogin 146004e86ffSlogin // Last entry 147004e86ffSlogin let las = (volatile_read!(cmdheader.prdtl) - 1) as usize; 148004e86ffSlogin volatile_write!(cmdtbl.prdt_entry[las].dba, virt_2_phys(buf_ptr) as u64); 1497ae679ddSLoGin cmdtbl.prdt_entry[las].dbc = ((tmp_count << 9) - 1) as u32; // 数据长度 1507ae679ddSLoGin 151004e86ffSlogin volatile_set_bit!(cmdtbl.prdt_entry[las].dbc, 1 << 31, true); // 允许中断 152004e86ffSlogin 153004e86ffSlogin // 设置命令 154004e86ffSlogin let cmdfis = unsafe { 155004e86ffSlogin ((&mut cmdtbl.cfis) as *mut [u8] as *mut usize as *mut FisRegH2D) 156004e86ffSlogin .as_mut() 157004e86ffSlogin .unwrap() 158004e86ffSlogin }; 159004e86ffSlogin volatile_write!(cmdfis.fis_type, FisType::RegH2D as u8); 160004e86ffSlogin volatile_set_bit!(cmdfis.pm, 1 << 7, true); // command_bit set 161004e86ffSlogin volatile_write!(cmdfis.command, ATA_CMD_READ_DMA_EXT); 162004e86ffSlogin 163004e86ffSlogin volatile_write!(cmdfis.lba0, (lba_id_start & 0xFF) as u8); 164004e86ffSlogin volatile_write!(cmdfis.lba1, ((lba_id_start >> 8) & 0xFF) as u8); 165004e86ffSlogin volatile_write!(cmdfis.lba2, ((lba_id_start >> 16) & 0xFF) as u8); 166004e86ffSlogin volatile_write!(cmdfis.lba3, ((lba_id_start >> 24) & 0xFF) as u8); 167004e86ffSlogin volatile_write!(cmdfis.lba4, ((lba_id_start >> 32) & 0xFF) as u8); 168004e86ffSlogin volatile_write!(cmdfis.lba5, ((lba_id_start >> 40) & 0xFF) as u8); 169004e86ffSlogin 170004e86ffSlogin volatile_write!(cmdfis.countl, (count & 0xFF) as u8); 171004e86ffSlogin volatile_write!(cmdfis.counth, ((count >> 8) & 0xFF) as u8); 172004e86ffSlogin 173004e86ffSlogin volatile_write!(cmdfis.device, 1 << 6); // LBA Mode 174004e86ffSlogin 175004e86ffSlogin // 等待之前的操作完成 176004e86ffSlogin let mut spin_count = 0; 177004e86ffSlogin const SPIN_LIMIT: u32 = 10000; 178004e86ffSlogin 179004e86ffSlogin while (volatile_read!(port.tfd) as u8 & (ATA_DEV_BUSY | ATA_DEV_DRQ)) > 0 180004e86ffSlogin && spin_count < SPIN_LIMIT 181004e86ffSlogin { 182004e86ffSlogin spin_count += 1; 183004e86ffSlogin } 184004e86ffSlogin 185004e86ffSlogin if spin_count == SPIN_LIMIT { 186004e86ffSlogin kerror!("Port is hung"); 187676b8ef6SMork return Err(SystemError::EIO); 188004e86ffSlogin } 189004e86ffSlogin 190004e86ffSlogin volatile_set_bit!(port.ci, 1 << slot, true); // Issue command 191004e86ffSlogin // kdebug!("To wait ahci read complete."); 192004e86ffSlogin // 等待操作完成 193004e86ffSlogin loop { 194004e86ffSlogin if (volatile_read!(port.ci) & (1 << slot)) == 0 { 195004e86ffSlogin break; 196004e86ffSlogin } 197004e86ffSlogin if (volatile_read!(port.is) & HBA_PxIS_TFES) > 0 { 198004e86ffSlogin kerror!("Read disk error"); 199676b8ef6SMork return Err(SystemError::EIO); 200004e86ffSlogin } 201004e86ffSlogin } 202004e86ffSlogin 203660a04ceSlogin if kbuf.is_some() { 204660a04ceSlogin buf.copy_from_slice(kbuf.as_ref().unwrap()); 205660a04ceSlogin } 206660a04ceSlogin 207004e86ffSlogin compiler_fence(core::sync::atomic::Ordering::SeqCst); 208004e86ffSlogin // successfully read 209004e86ffSlogin return Ok(count * 512); 210004e86ffSlogin } 211004e86ffSlogin 212004e86ffSlogin fn write_at( 213004e86ffSlogin &self, 214b087521eSChiichen lba_id_start: BlockId, 215004e86ffSlogin count: usize, 216004e86ffSlogin buf: &[u8], 217676b8ef6SMork ) -> Result<usize, SystemError> { 2187ae679ddSLoGin assert!((buf.len() & 511) == 0); 219004e86ffSlogin compiler_fence(core::sync::atomic::Ordering::SeqCst); 220004e86ffSlogin let check_length = ((count - 1) >> 4) + 1; // prdt length 221b5b571e0SLoGin if count * 512 > buf.len() || check_length > 8 { 222004e86ffSlogin // 不可能的操作 223676b8ef6SMork return Err(SystemError::E2BIG); 224004e86ffSlogin } else if count == 0 { 225004e86ffSlogin return Ok(0); 226004e86ffSlogin } 227004e86ffSlogin 228004e86ffSlogin let port = _port(self.ctrl_num, self.port_num); 229004e86ffSlogin 230004e86ffSlogin volatile_write!(port.is, u32::MAX); // Clear pending interrupt bits 231004e86ffSlogin 232004e86ffSlogin let slot = port.find_cmdslot().unwrap_or(u32::MAX); 233004e86ffSlogin 234004e86ffSlogin if slot == u32::MAX { 235676b8ef6SMork return Err(SystemError::EIO); 236004e86ffSlogin } 237004e86ffSlogin 238004e86ffSlogin compiler_fence(core::sync::atomic::Ordering::SeqCst); 239004e86ffSlogin #[allow(unused_unsafe)] 240004e86ffSlogin let cmdheader: &mut HbaCmdHeader = unsafe { 241004e86ffSlogin (phys_2_virt( 242b5b571e0SLoGin volatile_read!(port.clb) as usize + slot as usize * size_of::<HbaCmdHeader>(), 243004e86ffSlogin ) as *mut HbaCmdHeader) 244004e86ffSlogin .as_mut() 245004e86ffSlogin .unwrap() 246004e86ffSlogin }; 247004e86ffSlogin compiler_fence(core::sync::atomic::Ordering::SeqCst); 248004e86ffSlogin 249004e86ffSlogin volatile_write_bit!( 250004e86ffSlogin cmdheader.cfl, 251b5b571e0SLoGin (1 << 5) - 1_u8, 252004e86ffSlogin (size_of::<FisRegH2D>() / size_of::<u32>()) as u8 253004e86ffSlogin ); // Command FIS size 254004e86ffSlogin 255004e86ffSlogin volatile_set_bit!(cmdheader.cfl, 7 << 5, true); // (p,c,w)都设置为1, Read/Write bit : Write from device 256004e86ffSlogin volatile_write!(cmdheader.prdtl, check_length as u16); // PRDT entries count 257004e86ffSlogin 258004e86ffSlogin // 设置数据存放地址 259004e86ffSlogin compiler_fence(core::sync::atomic::Ordering::SeqCst); 260004e86ffSlogin let mut buf_ptr = buf as *const [u8] as *mut usize as usize; 261660a04ceSlogin 262660a04ceSlogin // 由于目前的内存管理机制无法把用户空间的内存地址转换为物理地址,所以只能先把数据拷贝到内核空间 263660a04ceSlogin // TODO:在内存管理重构后,可以直接使用用户空间的内存地址 264b5b571e0SLoGin let user_buf = verify_area(VirtAddr::new(buf_ptr), buf.len()).is_ok(); 265660a04ceSlogin let mut kbuf = if user_buf { 266b5b571e0SLoGin let mut x: Vec<u8> = vec![0; buf.len()]; 267660a04ceSlogin x.resize(buf.len(), 0); 268660a04ceSlogin x.copy_from_slice(buf); 269660a04ceSlogin Some(x) 270660a04ceSlogin } else { 271660a04ceSlogin None 272660a04ceSlogin }; 273660a04ceSlogin 274660a04ceSlogin if kbuf.is_some() { 275660a04ceSlogin buf_ptr = kbuf.as_mut().unwrap().as_mut_ptr() as usize; 276660a04ceSlogin } 277660a04ceSlogin 278004e86ffSlogin #[allow(unused_unsafe)] 279004e86ffSlogin let cmdtbl = unsafe { 280004e86ffSlogin (phys_2_virt(volatile_read!(cmdheader.ctba) as usize) as *mut HbaCmdTable) 281004e86ffSlogin .as_mut() 282004e86ffSlogin .unwrap() 283004e86ffSlogin }; 284004e86ffSlogin let mut tmp_count = count; 285004e86ffSlogin compiler_fence(core::sync::atomic::Ordering::SeqCst); 286004e86ffSlogin 287004e86ffSlogin unsafe { 288004e86ffSlogin // 清空整个table的旧数据 289004e86ffSlogin write_bytes(cmdtbl, 0, 1); 290004e86ffSlogin } 291004e86ffSlogin 292004e86ffSlogin // 8K bytes (16 sectors) per PRDT 293004e86ffSlogin for i in 0..((volatile_read!(cmdheader.prdtl) - 1) as usize) { 294004e86ffSlogin volatile_write!(cmdtbl.prdt_entry[i].dba, virt_2_phys(buf_ptr) as u64); 295004e86ffSlogin volatile_write_bit!(cmdtbl.prdt_entry[i].dbc, (1 << 22) - 1, 8 * 1024 - 1); // 数据长度 296004e86ffSlogin volatile_set_bit!(cmdtbl.prdt_entry[i].dbc, 1 << 31, true); // 允许中断 297004e86ffSlogin buf_ptr += 8 * 1024; 298004e86ffSlogin tmp_count -= 16; 299004e86ffSlogin } 300004e86ffSlogin 301004e86ffSlogin // Last entry 302004e86ffSlogin let las = (volatile_read!(cmdheader.prdtl) - 1) as usize; 303004e86ffSlogin volatile_write!(cmdtbl.prdt_entry[las].dba, virt_2_phys(buf_ptr) as u64); 304004e86ffSlogin volatile_set_bit!(cmdtbl.prdt_entry[las].dbc, 1 << 31, true); // 允许中断 305004e86ffSlogin volatile_write_bit!( 306004e86ffSlogin cmdtbl.prdt_entry[las].dbc, 307004e86ffSlogin (1 << 22) - 1, 308004e86ffSlogin ((tmp_count << 9) - 1) as u32 309004e86ffSlogin ); // 数据长度 310004e86ffSlogin 311004e86ffSlogin // 设置命令 312004e86ffSlogin let cmdfis = unsafe { 313004e86ffSlogin ((&mut cmdtbl.cfis) as *mut [u8] as *mut usize as *mut FisRegH2D) 314004e86ffSlogin .as_mut() 315004e86ffSlogin .unwrap() 316004e86ffSlogin }; 317004e86ffSlogin volatile_write!(cmdfis.fis_type, FisType::RegH2D as u8); 318004e86ffSlogin volatile_set_bit!(cmdfis.pm, 1 << 7, true); // command_bit set 319004e86ffSlogin volatile_write!(cmdfis.command, ATA_CMD_WRITE_DMA_EXT); 320004e86ffSlogin 321004e86ffSlogin volatile_write!(cmdfis.lba0, (lba_id_start & 0xFF) as u8); 322004e86ffSlogin volatile_write!(cmdfis.lba1, ((lba_id_start >> 8) & 0xFF) as u8); 323004e86ffSlogin volatile_write!(cmdfis.lba2, ((lba_id_start >> 16) & 0xFF) as u8); 324004e86ffSlogin volatile_write!(cmdfis.lba3, ((lba_id_start >> 24) & 0xFF) as u8); 325004e86ffSlogin volatile_write!(cmdfis.lba4, ((lba_id_start >> 32) & 0xFF) as u8); 326004e86ffSlogin volatile_write!(cmdfis.lba5, ((lba_id_start >> 40) & 0xFF) as u8); 327004e86ffSlogin 328004e86ffSlogin volatile_write!(cmdfis.countl, (count & 0xFF) as u8); 329004e86ffSlogin volatile_write!(cmdfis.counth, ((count >> 8) & 0xFF) as u8); 330004e86ffSlogin 331004e86ffSlogin volatile_write!(cmdfis.device, 1 << 6); // LBA Mode 332004e86ffSlogin 333004e86ffSlogin volatile_set_bit!(port.ci, 1 << slot, true); // Issue command 334004e86ffSlogin 335004e86ffSlogin // 等待操作完成 336004e86ffSlogin loop { 337004e86ffSlogin if (volatile_read!(port.ci) & (1 << slot)) == 0 { 338004e86ffSlogin break; 339004e86ffSlogin } 340004e86ffSlogin if (volatile_read!(port.is) & HBA_PxIS_TFES) > 0 { 341004e86ffSlogin kerror!("Write disk error"); 342676b8ef6SMork return Err(SystemError::EIO); 343004e86ffSlogin } 344004e86ffSlogin } 345004e86ffSlogin 346004e86ffSlogin compiler_fence(core::sync::atomic::Ordering::SeqCst); 347004e86ffSlogin // successfully read 348004e86ffSlogin return Ok(count * 512); 349004e86ffSlogin } 350004e86ffSlogin 351676b8ef6SMork fn sync(&self) -> Result<(), SystemError> { 352004e86ffSlogin // 由于目前没有block cache, 因此sync返回成功即可 353004e86ffSlogin return Ok(()); 354004e86ffSlogin } 355004e86ffSlogin } 356004e86ffSlogin 357004e86ffSlogin impl LockedAhciDisk { 358004e86ffSlogin pub fn new( 359004e86ffSlogin name: String, 360004e86ffSlogin flags: u16, 361004e86ffSlogin ctrl_num: u8, 362004e86ffSlogin port_num: u8, 363676b8ef6SMork ) -> Result<Arc<LockedAhciDisk>, SystemError> { 364004e86ffSlogin // 构建磁盘结构体 365004e86ffSlogin let result: Arc<LockedAhciDisk> = Arc::new(LockedAhciDisk(SpinLock::new(AhciDisk { 366004e86ffSlogin name, 367004e86ffSlogin flags, 368004e86ffSlogin partitions: Default::default(), 369004e86ffSlogin ctrl_num, 370004e86ffSlogin port_num, 371004e86ffSlogin self_ref: Weak::default(), 372004e86ffSlogin }))); 373004e86ffSlogin 374004e86ffSlogin let table: MbrDiskPartionTable = result.read_mbr_table()?; 375004e86ffSlogin 376004e86ffSlogin // 求出有多少可用分区 377004e86ffSlogin for i in 0..4 { 3787ae679ddSLoGin compiler_fence(Ordering::SeqCst); 379004e86ffSlogin if table.dpte[i].part_type != 0 { 3807ae679ddSLoGin let w = Arc::downgrade(&result); 3817ae679ddSLoGin result.0.lock().partitions.push(Partition::new( 382004e86ffSlogin table.dpte[i].starting_sector() as u64, 383004e86ffSlogin table.dpte[i].starting_lba as u64, 384004e86ffSlogin table.dpte[i].total_sectors as u64, 3857ae679ddSLoGin w, 386004e86ffSlogin i as u16, 387004e86ffSlogin )); 388004e86ffSlogin } 389004e86ffSlogin } 3907ae679ddSLoGin 3917ae679ddSLoGin result.0.lock().self_ref = Arc::downgrade(&result); 3927ae679ddSLoGin 393004e86ffSlogin return Ok(result); 394004e86ffSlogin } 395004e86ffSlogin 396004e86ffSlogin /// @brief: 从磁盘中读取 MBR 分区表结构体 TODO: Cursor 397676b8ef6SMork pub fn read_mbr_table(&self) -> Result<MbrDiskPartionTable, SystemError> { 398004e86ffSlogin let mut table: MbrDiskPartionTable = Default::default(); 399004e86ffSlogin 400004e86ffSlogin // 数据缓冲区 401b5b571e0SLoGin let mut buf: Vec<u8> = vec![0; size_of::<MbrDiskPartionTable>()]; 402004e86ffSlogin buf.resize(size_of::<MbrDiskPartionTable>(), 0); 403004e86ffSlogin 404*eb49bb99S曾俊 self.read_at_sync(0, 1, &mut buf)?; 405004e86ffSlogin // 创建 Cursor 用于按字节读取 406004e86ffSlogin let mut cursor = VecCursor::new(buf); 407004e86ffSlogin cursor.seek(SeekFrom::SeekCurrent(446))?; 408004e86ffSlogin 409004e86ffSlogin for i in 0..4 { 410b087521eSChiichen kdebug!("infomation of partition {}:\n", i); 411004e86ffSlogin 412004e86ffSlogin table.dpte[i].flags = cursor.read_u8()?; 413004e86ffSlogin table.dpte[i].starting_head = cursor.read_u8()?; 414004e86ffSlogin table.dpte[i].starting_sector_cylinder = cursor.read_u16()?; 415004e86ffSlogin table.dpte[i].part_type = cursor.read_u8()?; 416004e86ffSlogin table.dpte[i].ending_head = cursor.read_u8()?; 417004e86ffSlogin table.dpte[i].ending_sector_cylingder = cursor.read_u16()?; 418004e86ffSlogin table.dpte[i].starting_lba = cursor.read_u32()?; 419004e86ffSlogin table.dpte[i].total_sectors = cursor.read_u32()?; 420004e86ffSlogin 421b087521eSChiichen kdebug!("dpte[i] = {:?}", table.dpte[i]); 422004e86ffSlogin } 423004e86ffSlogin table.bs_trailsig = cursor.read_u16()?; 424004e86ffSlogin // kdebug!("bs_trailsig = {}", unsafe { 425004e86ffSlogin // read_unaligned(addr_of!(table.bs_trailsig)) 426004e86ffSlogin // }); 427004e86ffSlogin 428004e86ffSlogin return Ok(table); 429004e86ffSlogin } 430004e86ffSlogin } 431004e86ffSlogin 43206d5e247SLoGin impl KObject for LockedAhciDisk { 43306d5e247SLoGin fn as_any_ref(&self) -> &dyn core::any::Any { 43406d5e247SLoGin self 43506d5e247SLoGin } 43606d5e247SLoGin 43706d5e247SLoGin fn inode(&self) -> Option<Arc<KernFSInode>> { 43806d5e247SLoGin todo!() 43906d5e247SLoGin } 44006d5e247SLoGin 44106d5e247SLoGin fn kobj_type(&self) -> Option<&'static dyn KObjType> { 44206d5e247SLoGin todo!() 44306d5e247SLoGin } 44406d5e247SLoGin 44506d5e247SLoGin fn kset(&self) -> Option<Arc<KSet>> { 44606d5e247SLoGin todo!() 44706d5e247SLoGin } 44806d5e247SLoGin 44906d5e247SLoGin fn parent(&self) -> Option<Weak<dyn KObject>> { 45006d5e247SLoGin todo!() 45106d5e247SLoGin } 45206d5e247SLoGin 45306d5e247SLoGin fn set_inode(&self, _inode: Option<Arc<KernFSInode>>) { 45406d5e247SLoGin todo!() 45506d5e247SLoGin } 45606d5e247SLoGin 45706d5e247SLoGin fn kobj_state(&self) -> RwLockReadGuard<KObjectState> { 45806d5e247SLoGin todo!() 45906d5e247SLoGin } 46006d5e247SLoGin 46106d5e247SLoGin fn kobj_state_mut(&self) -> RwLockWriteGuard<KObjectState> { 46206d5e247SLoGin todo!() 46306d5e247SLoGin } 46406d5e247SLoGin 46506d5e247SLoGin fn set_kobj_state(&self, _state: KObjectState) { 46606d5e247SLoGin todo!() 46706d5e247SLoGin } 46806d5e247SLoGin 46906d5e247SLoGin fn name(&self) -> alloc::string::String { 47006d5e247SLoGin todo!() 47106d5e247SLoGin } 47206d5e247SLoGin 47306d5e247SLoGin fn set_name(&self, _name: alloc::string::String) { 47406d5e247SLoGin todo!() 47506d5e247SLoGin } 47606d5e247SLoGin 47706d5e247SLoGin fn set_kset(&self, _kset: Option<Arc<KSet>>) { 47806d5e247SLoGin todo!() 47906d5e247SLoGin } 48006d5e247SLoGin 48106d5e247SLoGin fn set_parent(&self, _parent: Option<Weak<dyn KObject>>) { 48206d5e247SLoGin todo!() 48306d5e247SLoGin } 484a03c4f9dSLoGin 485a03c4f9dSLoGin fn set_kobj_type(&self, _ktype: Option<&'static dyn KObjType>) { 486a03c4f9dSLoGin todo!() 487a03c4f9dSLoGin } 48806d5e247SLoGin } 489b087521eSChiichen 490b087521eSChiichen impl Device for LockedAhciDisk { 491b087521eSChiichen fn dev_type(&self) -> DeviceType { 492b087521eSChiichen return DeviceType::Block; 493b087521eSChiichen } 494b087521eSChiichen 49506d5e247SLoGin fn id_table(&self) -> IdTable { 496b087521eSChiichen todo!() 497b087521eSChiichen } 498b087521eSChiichen 499c566df45SLoGin fn bus(&self) -> Option<Weak<dyn Bus>> { 50006d5e247SLoGin todo!("LockedAhciDisk::bus()") 501b087521eSChiichen } 502b087521eSChiichen 503c566df45SLoGin fn set_bus(&self, _bus: Option<Weak<dyn Bus>>) { 504a03c4f9dSLoGin todo!("LockedAhciDisk::set_bus()") 505a03c4f9dSLoGin } 506a03c4f9dSLoGin 50706d5e247SLoGin fn driver(&self) -> Option<Arc<dyn Driver>> { 50806d5e247SLoGin todo!("LockedAhciDisk::driver()") 50906d5e247SLoGin } 51006d5e247SLoGin 51106d5e247SLoGin fn is_dead(&self) -> bool { 51206d5e247SLoGin false 51306d5e247SLoGin } 51406d5e247SLoGin 515a03c4f9dSLoGin fn set_driver(&self, _driver: Option<Weak<dyn Driver>>) { 51606d5e247SLoGin todo!("LockedAhciDisk::set_driver()") 517b087521eSChiichen } 518a03c4f9dSLoGin 519a03c4f9dSLoGin fn can_match(&self) -> bool { 520a03c4f9dSLoGin todo!() 521a03c4f9dSLoGin } 522a03c4f9dSLoGin 523a03c4f9dSLoGin fn set_can_match(&self, _can_match: bool) { 524a03c4f9dSLoGin todo!() 525a03c4f9dSLoGin } 526a03c4f9dSLoGin 527a03c4f9dSLoGin fn state_synced(&self) -> bool { 528a03c4f9dSLoGin todo!() 529a03c4f9dSLoGin } 53008a2ee40SLoGin 5314256da7fSLoGin fn set_class(&self, _class: Option<Weak<dyn Class>>) { 53208a2ee40SLoGin todo!() 53308a2ee40SLoGin } 534b087521eSChiichen } 535b087521eSChiichen 536004e86ffSlogin impl BlockDevice for LockedAhciDisk { 537004e86ffSlogin #[inline] 538004e86ffSlogin fn as_any_ref(&self) -> &dyn core::any::Any { 539004e86ffSlogin self 540004e86ffSlogin } 541004e86ffSlogin 542004e86ffSlogin #[inline] 543004e86ffSlogin fn blk_size_log2(&self) -> u8 { 544004e86ffSlogin 9 545004e86ffSlogin } 546004e86ffSlogin 547676b8ef6SMork fn sync(&self) -> Result<(), SystemError> { 548004e86ffSlogin return self.0.lock().sync(); 549004e86ffSlogin } 550004e86ffSlogin 551004e86ffSlogin #[inline] 552b087521eSChiichen fn device(&self) -> Arc<dyn Device> { 553004e86ffSlogin return self.0.lock().self_ref.upgrade().unwrap(); 554004e86ffSlogin } 555004e86ffSlogin 556004e86ffSlogin fn block_size(&self) -> usize { 557004e86ffSlogin todo!() 558004e86ffSlogin } 559004e86ffSlogin 560004e86ffSlogin fn partitions(&self) -> Vec<Arc<Partition>> { 561004e86ffSlogin return self.0.lock().partitions.clone(); 562004e86ffSlogin } 563b087521eSChiichen 564b087521eSChiichen #[inline] 565*eb49bb99S曾俊 fn read_at_sync( 566b087521eSChiichen &self, 567b087521eSChiichen lba_id_start: BlockId, // 起始lba编号 568b087521eSChiichen count: usize, // 读取lba的数量 569b087521eSChiichen buf: &mut [u8], 570b087521eSChiichen ) -> Result<usize, SystemError> { 571b087521eSChiichen self.0.lock().read_at(lba_id_start, count, buf) 572b087521eSChiichen } 573b087521eSChiichen 574b087521eSChiichen #[inline] 575*eb49bb99S曾俊 fn write_at_sync( 576b087521eSChiichen &self, 577b087521eSChiichen lba_id_start: BlockId, 578b087521eSChiichen count: usize, 579b087521eSChiichen buf: &[u8], 580b087521eSChiichen ) -> Result<usize, SystemError> { 581b087521eSChiichen self.0.lock().write_at(lba_id_start, count, buf) 582b087521eSChiichen } 583004e86ffSlogin } 584