1004e86ffSlogin #![allow(dead_code)] 2004e86ffSlogin use core::cmp::min; 3004e86ffSlogin 4004e86ffSlogin use alloc::{ 5004e86ffSlogin string::{String, ToString}, 6004e86ffSlogin sync::Arc, 7004e86ffSlogin vec::Vec, 8004e86ffSlogin }; 9004e86ffSlogin 10004e86ffSlogin use crate::{ 11004e86ffSlogin io::{device::LBA_SIZE, SeekFrom}, 12004e86ffSlogin kwarn, 13004e86ffSlogin libs::vec_cursor::VecCursor, 14676b8ef6SMork syscall::SystemError, 15004e86ffSlogin }; 16004e86ffSlogin 17004e86ffSlogin use super::{ 18004e86ffSlogin fs::{Cluster, FATFileSystem, MAX_FILE_SIZE}, 19004e86ffSlogin utils::decode_u8_ascii, 20004e86ffSlogin }; 21004e86ffSlogin 22004e86ffSlogin #[derive(Debug, Clone, Copy, Default)] 23004e86ffSlogin pub struct FileAttributes { 24004e86ffSlogin value: u8, 25004e86ffSlogin } 26004e86ffSlogin 27004e86ffSlogin /// FAT表中,关于每个簇的信息 28004e86ffSlogin #[derive(Debug, Eq, PartialEq)] 29004e86ffSlogin pub enum FATEntry { 30004e86ffSlogin /// 当前簇未使用 31004e86ffSlogin Unused, 32004e86ffSlogin /// 当前簇是坏簇 33004e86ffSlogin Bad, 34004e86ffSlogin /// 当前簇是整个FAT簇链的最后一个簇 35004e86ffSlogin EndOfChain, 36004e86ffSlogin /// 在整个链中,当前簇的下一个簇的值 37004e86ffSlogin Next(Cluster), 38004e86ffSlogin } 39004e86ffSlogin 40004e86ffSlogin /// FAT目录项的枚举类型 41004e86ffSlogin #[derive(Debug, Clone)] 42004e86ffSlogin pub enum FATDirEntry { 43004e86ffSlogin File(FATFile), 44004e86ffSlogin VolId(FATFile), 45004e86ffSlogin Dir(FATDir), 46004e86ffSlogin UnInit, 47004e86ffSlogin } 48004e86ffSlogin 49004e86ffSlogin /// FAT文件系统中的文件 50004e86ffSlogin #[derive(Debug, Default, Clone)] 51004e86ffSlogin pub struct FATFile { 52004e86ffSlogin /// 文件的第一个簇 53004e86ffSlogin pub first_cluster: Cluster, 54004e86ffSlogin /// 文件名 55004e86ffSlogin pub file_name: String, 56004e86ffSlogin /// 文件对应的短目录项 57004e86ffSlogin pub short_dir_entry: ShortDirEntry, 58004e86ffSlogin /// 文件目录项的起始、终止簇。格式:(簇,簇内偏移量) 59004e86ffSlogin pub loc: ((Cluster, u64), (Cluster, u64)), 60004e86ffSlogin } 61004e86ffSlogin 62004e86ffSlogin impl FATFile { 63004e86ffSlogin /// @brief 获取文件大小 64004e86ffSlogin #[inline] 65004e86ffSlogin pub fn size(&self) -> u64 { 66004e86ffSlogin return self.short_dir_entry.file_size as u64; 67004e86ffSlogin } 68004e86ffSlogin 69004e86ffSlogin /// @brief 设置当前文件大小(仅仅更改short_dir_entry内的值) 70004e86ffSlogin #[inline] 71004e86ffSlogin pub fn set_size(&mut self, size: u32) { 72004e86ffSlogin self.short_dir_entry.file_size = size; 73004e86ffSlogin } 74004e86ffSlogin 75004e86ffSlogin /// @brief 从文件读取数据。读取的字节数与buf长度相等 76004e86ffSlogin /// 77004e86ffSlogin /// @param buf 输出缓冲区 78004e86ffSlogin /// @param offset 起始位置在文件中的偏移量 79004e86ffSlogin /// 80004e86ffSlogin /// @return Ok(usize) 成功读取到的字节数 81676b8ef6SMork /// @return Err(SystemError) 读取时出现错误,返回错误码 82*78bf93f0SYJwu2023 pub fn read( 83*78bf93f0SYJwu2023 &self, 84*78bf93f0SYJwu2023 fs: &Arc<FATFileSystem>, 85*78bf93f0SYJwu2023 buf: &mut [u8], 86*78bf93f0SYJwu2023 offset: u64, 87*78bf93f0SYJwu2023 ) -> Result<usize, SystemError> { 88004e86ffSlogin if offset >= self.size() { 89004e86ffSlogin return Ok(0); 90004e86ffSlogin } 91004e86ffSlogin 92004e86ffSlogin // 文件内的簇偏移量 93004e86ffSlogin let start_cluster_number: u64 = offset / fs.bytes_per_cluster(); 94004e86ffSlogin // 计算对应在分区内的簇号 95004e86ffSlogin let mut current_cluster = if let Some(c) = 96004e86ffSlogin fs.get_cluster_by_relative(self.first_cluster, start_cluster_number as usize) 97004e86ffSlogin { 98004e86ffSlogin c 99004e86ffSlogin } else { 100004e86ffSlogin return Ok(0); 101004e86ffSlogin }; 102004e86ffSlogin 103004e86ffSlogin let bytes_remain: u64 = self.size() - offset; 104004e86ffSlogin 105004e86ffSlogin // 计算簇内偏移量 106004e86ffSlogin let mut in_cluster_offset: u64 = offset % fs.bytes_per_cluster(); 107004e86ffSlogin let to_read_size: usize = min(buf.len(), bytes_remain as usize); 108004e86ffSlogin 109004e86ffSlogin let mut start = 0; 110004e86ffSlogin let mut read_ok = 0; 111004e86ffSlogin 112004e86ffSlogin loop { 113004e86ffSlogin // 当前簇已经读取完,尝试读取下一个簇 114004e86ffSlogin if in_cluster_offset >= fs.bytes_per_cluster() { 115004e86ffSlogin if let Some(FATEntry::Next(c)) = fs.get_fat_entry(current_cluster).ok() { 116004e86ffSlogin current_cluster = c; 117004e86ffSlogin in_cluster_offset %= fs.bytes_per_cluster(); 118004e86ffSlogin } else { 119004e86ffSlogin break; 120004e86ffSlogin } 121004e86ffSlogin } 122004e86ffSlogin 123004e86ffSlogin // 计算下一次读取,能够读多少字节 124004e86ffSlogin let end_len: usize = min( 125004e86ffSlogin to_read_size - read_ok, 126004e86ffSlogin min( 127004e86ffSlogin (fs.bytes_per_cluster() - in_cluster_offset) as usize, 128004e86ffSlogin buf.len() - read_ok, 129004e86ffSlogin ), 130004e86ffSlogin ); 131004e86ffSlogin 132004e86ffSlogin // 从磁盘上读取数据 133004e86ffSlogin let offset = fs.cluster_bytes_offset(current_cluster) + in_cluster_offset; 134004e86ffSlogin let r = fs.partition.disk().device().read_at( 135004e86ffSlogin offset as usize, 136004e86ffSlogin end_len, 137004e86ffSlogin &mut buf[start..start + end_len], 138004e86ffSlogin )?; 139004e86ffSlogin 140004e86ffSlogin // 更新偏移量计数信息 141004e86ffSlogin read_ok += r; 142004e86ffSlogin start += r; 143004e86ffSlogin in_cluster_offset += r as u64; 144004e86ffSlogin if read_ok == to_read_size { 145004e86ffSlogin break; 146004e86ffSlogin } 147004e86ffSlogin } 148004e86ffSlogin // todo: 更新时间信息 149004e86ffSlogin return Ok(read_ok); 150004e86ffSlogin } 151004e86ffSlogin 152004e86ffSlogin /// @brief 向文件写入数据。写入的字节数与buf长度相等 153004e86ffSlogin /// 154004e86ffSlogin /// @param buf 输入缓冲区 155004e86ffSlogin /// @param offset 起始位置在文件中的偏移量 156004e86ffSlogin /// 157004e86ffSlogin /// @return Ok(usize) 成功写入的字节数 158676b8ef6SMork /// @return Err(SystemError) 写入时出现错误,返回错误码 159004e86ffSlogin pub fn write( 160004e86ffSlogin &mut self, 161004e86ffSlogin fs: &Arc<FATFileSystem>, 162004e86ffSlogin buf: &[u8], 163004e86ffSlogin offset: u64, 164676b8ef6SMork ) -> Result<usize, SystemError> { 165004e86ffSlogin self.ensure_len(fs, offset, buf.len() as u64)?; 166004e86ffSlogin 167004e86ffSlogin // 要写入的第一个簇的簇号 168004e86ffSlogin let start_cluster_num = offset / fs.bytes_per_cluster(); 169004e86ffSlogin 170004e86ffSlogin // 获取要写入的第一个簇 171004e86ffSlogin let mut current_cluster: Cluster = if let Some(c) = 172004e86ffSlogin fs.get_cluster_by_relative(self.first_cluster, start_cluster_num as usize) 173004e86ffSlogin { 174004e86ffSlogin c 175004e86ffSlogin } else { 176004e86ffSlogin return Ok(0); 177004e86ffSlogin }; 178004e86ffSlogin 179004e86ffSlogin let mut in_cluster_bytes_offset: u64 = offset % fs.bytes_per_cluster(); 180004e86ffSlogin 181004e86ffSlogin let mut start: usize = 0; 182004e86ffSlogin let mut write_ok: usize = 0; 183004e86ffSlogin 184004e86ffSlogin // 循环写入数据 185004e86ffSlogin loop { 186004e86ffSlogin if in_cluster_bytes_offset >= fs.bytes_per_cluster() { 187004e86ffSlogin if let Some(FATEntry::Next(c)) = fs.get_fat_entry(current_cluster).ok() { 188004e86ffSlogin current_cluster = c; 189004e86ffSlogin in_cluster_bytes_offset = in_cluster_bytes_offset % fs.bytes_per_cluster(); 190004e86ffSlogin } else { 191004e86ffSlogin break; 192004e86ffSlogin } 193004e86ffSlogin } 194004e86ffSlogin 195004e86ffSlogin let end_len = min( 196004e86ffSlogin (fs.bytes_per_cluster() - in_cluster_bytes_offset) as usize, 197004e86ffSlogin buf.len() - write_ok, 198004e86ffSlogin ); 199004e86ffSlogin 200004e86ffSlogin // 计算本次写入位置在磁盘上的偏移量 201004e86ffSlogin let offset = fs.cluster_bytes_offset(current_cluster) + in_cluster_bytes_offset; 202004e86ffSlogin // 写入磁盘 203004e86ffSlogin let w: usize = fs.partition.disk().device().write_at( 204004e86ffSlogin offset as usize, 205004e86ffSlogin end_len, 206004e86ffSlogin &buf[start..start + end_len], 207004e86ffSlogin )?; 208004e86ffSlogin 209004e86ffSlogin // 更新偏移量数据 210004e86ffSlogin write_ok += w; 211004e86ffSlogin start += w; 212004e86ffSlogin in_cluster_bytes_offset += w as u64; 213004e86ffSlogin 214004e86ffSlogin if write_ok == buf.len() { 215004e86ffSlogin break; 216004e86ffSlogin } 217004e86ffSlogin } 218004e86ffSlogin // todo: 更新时间信息 219004e86ffSlogin return Ok(write_ok); 220004e86ffSlogin } 221004e86ffSlogin 222004e86ffSlogin /// @brief 确保文件从指定偏移量开始,仍有长度为len的空间。 223004e86ffSlogin /// 如果文件大小不够,就尝试分配更多的空间给这个文件。 224004e86ffSlogin /// 225004e86ffSlogin /// @param fs 当前文件所属的文件系统 226004e86ffSlogin /// @param offset 起始位置在文件内的字节偏移量 227004e86ffSlogin /// @param len 期待的空闲空间长度 228004e86ffSlogin /// 229004e86ffSlogin /// @return Ok(()) 经过操作后,offset后面具有长度至少为len的空闲空间 230676b8ef6SMork /// @return Err(SystemError) 处理过程中出现了异常。 231*78bf93f0SYJwu2023 fn ensure_len( 232*78bf93f0SYJwu2023 &mut self, 233*78bf93f0SYJwu2023 fs: &Arc<FATFileSystem>, 234*78bf93f0SYJwu2023 offset: u64, 235*78bf93f0SYJwu2023 len: u64, 236*78bf93f0SYJwu2023 ) -> Result<(), SystemError> { 237004e86ffSlogin // 文件内本身就还有空余的空间 238004e86ffSlogin if offset + len <= self.size() { 239004e86ffSlogin return Ok(()); 240004e86ffSlogin } 241004e86ffSlogin 242004e86ffSlogin // 如果文件大小为0,证明它还没有分配簇,因此分配一个簇给它 243004e86ffSlogin if self.size() == 0 { 244004e86ffSlogin // first_cluster应当为0,否则将产生空间泄露的错误 245004e86ffSlogin assert_eq!(self.first_cluster, Cluster::default()); 246004e86ffSlogin self.first_cluster = fs.allocate_cluster(None)?; 247004e86ffSlogin self.short_dir_entry.set_first_cluster(self.first_cluster); 248004e86ffSlogin } 249004e86ffSlogin 250004e86ffSlogin // 计算文件的最后一个簇中有多少空闲空间 251004e86ffSlogin 252004e86ffSlogin let in_cluster_offset = self.size() % fs.bytes_per_cluster(); 253004e86ffSlogin let bytes_remain_in_cluster = fs.bytes_per_cluster() - in_cluster_offset; 254004e86ffSlogin 255004e86ffSlogin // 计算还需要申请多少空间 256004e86ffSlogin let extra_bytes = min((offset + len) - self.size(), MAX_FILE_SIZE - self.size()); 257004e86ffSlogin 258004e86ffSlogin // 如果还需要更多的簇 259004e86ffSlogin if bytes_remain_in_cluster < extra_bytes { 260004e86ffSlogin let clusters_to_allocate = 261004e86ffSlogin (extra_bytes - bytes_remain_in_cluster + fs.bytes_per_cluster() - 1) 262004e86ffSlogin / fs.bytes_per_cluster(); 263004e86ffSlogin let last_cluster = if let Some(c) = fs.get_last_cluster(self.first_cluster) { 264004e86ffSlogin c 265004e86ffSlogin } else { 266004e86ffSlogin kwarn!("FAT: last cluster not found, File = {self:?}"); 267676b8ef6SMork return Err(SystemError::EINVAL); 268004e86ffSlogin }; 269004e86ffSlogin // 申请簇 270004e86ffSlogin let mut current_cluster: Cluster = last_cluster; 271004e86ffSlogin for _ in 0..clusters_to_allocate { 272004e86ffSlogin current_cluster = fs.allocate_cluster(Some(current_cluster))?; 273004e86ffSlogin } 274004e86ffSlogin } 275004e86ffSlogin 276004e86ffSlogin // 如果文件被扩展,则清空刚刚被扩展的部分的数据 277004e86ffSlogin if offset > self.size() { 278004e86ffSlogin // 文件内的簇偏移 279004e86ffSlogin let start_cluster: u64 = self.size() / fs.bytes_per_cluster(); 280004e86ffSlogin let start_cluster: Cluster = fs 281004e86ffSlogin .get_cluster_by_relative(self.first_cluster, start_cluster as usize) 282004e86ffSlogin .unwrap(); 283004e86ffSlogin // 计算当前文件末尾在磁盘上的字节偏移量 284004e86ffSlogin let start_offset: u64 = 285004e86ffSlogin fs.cluster_bytes_offset(start_cluster) + self.size() % fs.bytes_per_cluster(); 286004e86ffSlogin // 扩展之前,最后一个簇内还剩下多少字节的空间 287004e86ffSlogin let bytes_remain: u64 = fs.bytes_per_cluster() - (self.size() % fs.bytes_per_cluster()); 288004e86ffSlogin // 计算在扩展之后的最后一个簇内,文件的终止字节 289004e86ffSlogin let cluster_offset_start = offset / fs.bytes_per_cluster(); 290004e86ffSlogin // 扩展后,文件的最后 291004e86ffSlogin let end_cluster: Cluster = fs 292004e86ffSlogin .get_cluster_by_relative(self.first_cluster, cluster_offset_start as usize) 293004e86ffSlogin .unwrap(); 294004e86ffSlogin 295004e86ffSlogin if start_cluster != end_cluster { 296004e86ffSlogin self.zero_range(fs, start_offset, start_offset + bytes_remain)?; 297004e86ffSlogin } else { 298004e86ffSlogin self.zero_range(fs, start_offset, start_offset + offset - self.size())?; 299004e86ffSlogin } 300004e86ffSlogin } 301004e86ffSlogin // 计算文件的新大小 302004e86ffSlogin let new_size = self.size() + extra_bytes; 303004e86ffSlogin self.set_size(new_size as u32); 304004e86ffSlogin // 计算短目录项所在的位置,更新短目录项 305004e86ffSlogin let short_entry_offset = fs.cluster_bytes_offset(self.loc.1 .0) + self.loc.1 .1; 306004e86ffSlogin // todo: 更新时间信息 307004e86ffSlogin // 把短目录项写入磁盘 308004e86ffSlogin self.short_dir_entry.flush(fs, short_entry_offset)?; 309004e86ffSlogin return Ok(()); 310004e86ffSlogin } 311004e86ffSlogin 312004e86ffSlogin /// @brief 把磁盘上[range_start, range_end)范围的数据清零 313004e86ffSlogin /// 314004e86ffSlogin /// @param range_start 磁盘上起始位置(单位:字节) 315004e86ffSlogin /// @param range_end 磁盘上终止位置(单位:字节) 316004e86ffSlogin fn zero_range( 317004e86ffSlogin &self, 318004e86ffSlogin fs: &Arc<FATFileSystem>, 319004e86ffSlogin range_start: u64, 320004e86ffSlogin range_end: u64, 321676b8ef6SMork ) -> Result<(), SystemError> { 322004e86ffSlogin if range_end <= range_start { 323004e86ffSlogin return Ok(()); 324004e86ffSlogin } 325004e86ffSlogin 326004e86ffSlogin let zeroes: Vec<u8> = vec![0u8; (range_end - range_start) as usize]; 327004e86ffSlogin fs.partition.disk().device().write_at( 328004e86ffSlogin range_start as usize, 329004e86ffSlogin zeroes.len(), 330004e86ffSlogin zeroes.as_slice(), 331004e86ffSlogin )?; 332004e86ffSlogin return Ok(()); 333004e86ffSlogin } 334004e86ffSlogin 335004e86ffSlogin /// @brief 截断文件的内容,并设置新的文件大小。如果new_size大于当前文件大小,则不做操作。 336004e86ffSlogin /// 337004e86ffSlogin /// @param new_size 新的文件大小,如果它大于当前文件大小,则不做操作。 338004e86ffSlogin /// 339004e86ffSlogin /// @return Ok(()) 操作成功 340676b8ef6SMork /// @return Err(SystemError) 在操作时出现错误 341676b8ef6SMork pub fn truncate(&mut self, fs: &Arc<FATFileSystem>, new_size: u64) -> Result<(), SystemError> { 342004e86ffSlogin if new_size >= self.size() { 343004e86ffSlogin return Ok(()); 344004e86ffSlogin } 345004e86ffSlogin 346004e86ffSlogin let new_last_cluster = new_size / fs.bytes_per_cluster(); 347004e86ffSlogin if let Some(begin_delete) = 348004e86ffSlogin fs.get_cluster_by_relative(self.first_cluster, (new_last_cluster + 1) as usize) 349004e86ffSlogin { 350004e86ffSlogin fs.deallocate_cluster(begin_delete)?; 351004e86ffSlogin }; 352004e86ffSlogin 353004e86ffSlogin self.set_size(new_size as u32); 354004e86ffSlogin // 计算短目录项在磁盘内的字节偏移量 355004e86ffSlogin let short_entry_offset = fs.cluster_bytes_offset((self.loc.1).0) + (self.loc.1).1; 356004e86ffSlogin self.short_dir_entry.flush(fs, short_entry_offset)?; 357004e86ffSlogin return Ok(()); 358004e86ffSlogin } 359004e86ffSlogin } 360004e86ffSlogin 361004e86ffSlogin /// FAT文件系统中的文件夹 362004e86ffSlogin #[derive(Debug, Default, Clone)] 363004e86ffSlogin pub struct FATDir { 364004e86ffSlogin /// 目录的第一个簇 365004e86ffSlogin pub first_cluster: Cluster, 366004e86ffSlogin /// 该字段仅对FAT12、FAT16生效 367004e86ffSlogin pub root_offset: Option<u64>, 368004e86ffSlogin /// 文件夹名称 369004e86ffSlogin pub dir_name: String, 370004e86ffSlogin pub short_dir_entry: Option<ShortDirEntry>, 371004e86ffSlogin /// 文件的起始、终止簇。格式:(簇,簇内偏移量) 372004e86ffSlogin pub loc: Option<((Cluster, u64), (Cluster, u64))>, 373004e86ffSlogin } 374004e86ffSlogin 375004e86ffSlogin impl FATDir { 376004e86ffSlogin /// @brief 获得用于遍历当前目录的迭代器 377004e86ffSlogin /// 378004e86ffSlogin /// @param fs 当前目录所在的文件系统 379004e86ffSlogin pub fn to_iter(&self, fs: Arc<FATFileSystem>) -> FATDirIter { 380004e86ffSlogin return FATDirIter { 381004e86ffSlogin current_cluster: self.first_cluster, 382004e86ffSlogin offset: self.root_offset.unwrap_or(0), 383004e86ffSlogin is_root: self.is_root(), 384004e86ffSlogin fs: fs, 385004e86ffSlogin }; 386004e86ffSlogin } 387004e86ffSlogin 388004e86ffSlogin /// @brief 判断当前目录是否为根目录(仅对FAT12和FAT16生效) 389004e86ffSlogin #[inline] 390004e86ffSlogin pub fn is_root(&self) -> bool { 391004e86ffSlogin return self.root_offset.is_some(); 392004e86ffSlogin } 393004e86ffSlogin 394004e86ffSlogin /// @brief 获取当前目录所占用的大小 395004e86ffSlogin pub fn size(&self, fs: &Arc<FATFileSystem>) -> u64 { 396004e86ffSlogin return fs.num_clusters_chain(self.first_cluster) * fs.bytes_per_cluster(); 397004e86ffSlogin } 398004e86ffSlogin 399004e86ffSlogin /// @brief 在目录项中,寻找num_free个连续空闲目录项 400004e86ffSlogin /// 401004e86ffSlogin /// @param num_free 需要的空闲目录项数目. 402004e86ffSlogin /// @param fs 当前文件夹属于的文件系统 403004e86ffSlogin /// 404004e86ffSlogin /// @return Ok(Option<(第一个符合条件的空闲目录项所在的簇,簇内偏移量)) 405004e86ffSlogin /// @return Err(错误码) 406004e86ffSlogin pub fn find_free_entries( 407004e86ffSlogin &self, 408004e86ffSlogin num_free: u64, 409004e86ffSlogin fs: Arc<FATFileSystem>, 410676b8ef6SMork ) -> Result<Option<(Cluster, u64)>, SystemError> { 411004e86ffSlogin let mut free = 0; 412004e86ffSlogin let mut current_cluster: Cluster = self.first_cluster; 413004e86ffSlogin let mut offset = self.root_offset.unwrap_or(0); 414004e86ffSlogin // 第一个符合条件的空闲目录项 415004e86ffSlogin let mut first_free: Option<(Cluster, u64)> = None; 416004e86ffSlogin 417004e86ffSlogin loop { 418004e86ffSlogin // 如果当前簇没有空间了,并且当前不是FAT12和FAT16的根目录,那么就读取下一个簇。 419004e86ffSlogin if offset >= fs.bytes_per_cluster() && !self.is_root() { 420004e86ffSlogin // 成功读取下一个簇 421004e86ffSlogin if let Some(FATEntry::Next(c)) = fs.get_fat_entry(current_cluster).ok() { 422004e86ffSlogin current_cluster = c; 423004e86ffSlogin // 计算簇内偏移量 424004e86ffSlogin offset = offset % fs.bytes_per_cluster(); 425004e86ffSlogin } else { 426004e86ffSlogin // 读取失败,当前已经是最后一个簇,退出循环 427004e86ffSlogin break; 428004e86ffSlogin } 429004e86ffSlogin } 430004e86ffSlogin // 如果当前目录是FAT12和FAT16的根目录,且已经读取完,就直接返回。 431004e86ffSlogin if self.is_root() && offset > fs.root_dir_end_bytes_offset().unwrap() { 432004e86ffSlogin return Ok(None); 433004e86ffSlogin } 434004e86ffSlogin 435004e86ffSlogin let e_offset = fs.cluster_bytes_offset(current_cluster) + offset; 436004e86ffSlogin let entry: FATRawDirEntry = get_raw_dir_entry(&fs, e_offset)?; 437004e86ffSlogin 438004e86ffSlogin match entry { 439004e86ffSlogin FATRawDirEntry::Free | FATRawDirEntry::FreeRest => { 440004e86ffSlogin if free == 0 { 441004e86ffSlogin first_free = Some((current_cluster, offset)); 442004e86ffSlogin } 443004e86ffSlogin 444004e86ffSlogin free += 1; 445004e86ffSlogin if free == num_free { 446004e86ffSlogin // kdebug!("first_free = {first_free:?}, current_free = ({current_cluster:?}, {offset})"); 447004e86ffSlogin return Ok(first_free); 448004e86ffSlogin } 449004e86ffSlogin } 450004e86ffSlogin 451004e86ffSlogin // 遇到一个不空闲的目录项,那么重新开始计算空闲目录项 452004e86ffSlogin _ => { 453004e86ffSlogin free = 0; 454004e86ffSlogin } 455004e86ffSlogin } 456004e86ffSlogin offset += FATRawDirEntry::DIR_ENTRY_LEN; 457004e86ffSlogin } 458004e86ffSlogin 459004e86ffSlogin // 剩余的需要获取的目录项 460004e86ffSlogin let remain_entries = num_free - free; 461004e86ffSlogin 462004e86ffSlogin // 计算需要申请多少个簇 463004e86ffSlogin let clusters_required = 464004e86ffSlogin (remain_entries * FATRawDirEntry::DIR_ENTRY_LEN + fs.bytes_per_cluster() - 1) 465004e86ffSlogin / fs.bytes_per_cluster(); 466004e86ffSlogin let mut first_cluster = Cluster::default(); 467004e86ffSlogin let mut prev_cluster = current_cluster; 468004e86ffSlogin // kdebug!( 469004e86ffSlogin // "clusters_required={clusters_required}, prev_cluster={prev_cluster:?}, free ={free}" 470004e86ffSlogin // ); 471004e86ffSlogin // 申请簇 472004e86ffSlogin for i in 0..clusters_required { 473004e86ffSlogin let c: Cluster = fs.allocate_cluster(Some(prev_cluster))?; 474004e86ffSlogin if i == 0 { 475004e86ffSlogin first_cluster = c; 476004e86ffSlogin } 477004e86ffSlogin 478004e86ffSlogin prev_cluster = c; 479004e86ffSlogin } 480004e86ffSlogin 481004e86ffSlogin if free > 0 { 482004e86ffSlogin // 空闲目录项跨越了簇,返回第一个空闲目录项 483004e86ffSlogin return Ok(first_free); 484004e86ffSlogin } else { 485004e86ffSlogin // 空闲目录项是在全新的簇开始的 486004e86ffSlogin return Ok(Some((first_cluster, 0))); 487004e86ffSlogin } 488004e86ffSlogin } 489004e86ffSlogin 490004e86ffSlogin /// @brief 在当前目录中寻找目录项 491004e86ffSlogin /// 492004e86ffSlogin /// @param name 目录项的名字 493004e86ffSlogin /// @param expect_dir 该值为Some时有效。如果期待目标目录项是文件夹,那么值为Some(true), 否则为Some(false). 494004e86ffSlogin /// @param short_name_gen 短目录项名称生成器 495004e86ffSlogin /// @param fs 当前目录所属的文件系统 496004e86ffSlogin /// 497004e86ffSlogin /// @return Ok(FATDirEntry) 找到期待的目录项 498676b8ef6SMork /// @return Err(SystemError) 错误码 499004e86ffSlogin pub fn find_entry( 500004e86ffSlogin &self, 501004e86ffSlogin name: &str, 502004e86ffSlogin expect_dir: Option<bool>, 503004e86ffSlogin mut short_name_gen: Option<&mut ShortNameGenerator>, 504004e86ffSlogin fs: Arc<FATFileSystem>, 505676b8ef6SMork ) -> Result<FATDirEntry, SystemError> { 506004e86ffSlogin LongDirEntry::validate_long_name(name)?; 507004e86ffSlogin // 迭代当前目录下的文件/文件夹 508004e86ffSlogin for e in self.to_iter(fs) { 509004e86ffSlogin if e.eq_name(name) { 510004e86ffSlogin if expect_dir.is_some() && Some(e.is_dir()) != expect_dir { 511004e86ffSlogin if e.is_dir() { 512004e86ffSlogin // 期望得到文件,但是是文件夹 513676b8ef6SMork return Err(SystemError::EISDIR); 514004e86ffSlogin } else { 515004e86ffSlogin // 期望得到文件夹,但是是文件 516676b8ef6SMork return Err(SystemError::ENOTDIR); 517004e86ffSlogin } 518004e86ffSlogin } 519004e86ffSlogin // 找到期望的目录项 520004e86ffSlogin return Ok(e); 521004e86ffSlogin } 522004e86ffSlogin 523004e86ffSlogin if let Some(ref mut sng) = short_name_gen { 524004e86ffSlogin sng.add_name(&e.short_name_raw()) 525004e86ffSlogin } 526004e86ffSlogin } 527004e86ffSlogin // 找不到文件/文件夹 528676b8ef6SMork return Err(SystemError::ENOENT); 529004e86ffSlogin } 530004e86ffSlogin 531004e86ffSlogin /// @brief 在当前目录下打开文件,获取FATFile结构体 532676b8ef6SMork pub fn open_file(&self, name: &str, fs: Arc<FATFileSystem>) -> Result<FATFile, SystemError> { 533004e86ffSlogin let f: FATFile = self.find_entry(name, Some(false), None, fs)?.to_file()?; 534004e86ffSlogin return Ok(f); 535004e86ffSlogin } 536004e86ffSlogin 537004e86ffSlogin /// @brief 在当前目录下打开文件夹,获取FATDir结构体 538676b8ef6SMork pub fn open_dir(&self, name: &str, fs: Arc<FATFileSystem>) -> Result<FATDir, SystemError> { 539004e86ffSlogin let d: FATDir = self.find_entry(name, Some(true), None, fs)?.to_dir()?; 540004e86ffSlogin return Ok(d); 541004e86ffSlogin } 542004e86ffSlogin 543004e86ffSlogin /// @brief 在当前文件夹下创建文件。 544004e86ffSlogin /// 545004e86ffSlogin /// @param name 文件名 546004e86ffSlogin /// @param fs 当前文件夹所属的文件系统 547676b8ef6SMork pub fn create_file(&self, name: &str, fs: &Arc<FATFileSystem>) -> Result<FATFile, SystemError> { 548676b8ef6SMork let r: Result<FATDirEntryOrShortName, SystemError> = 549004e86ffSlogin self.check_existence(name, Some(false), fs.clone()); 550004e86ffSlogin // 检查错误码,如果能够表明目录项已经存在,则返回-EEXIST 551004e86ffSlogin if r.is_err() { 552004e86ffSlogin let err_val = r.unwrap_err(); 553676b8ef6SMork if err_val == (SystemError::EISDIR) || err_val == (SystemError::ENOTDIR) { 554676b8ef6SMork return Err(SystemError::EEXIST); 555004e86ffSlogin } else { 556004e86ffSlogin return Err(err_val); 557004e86ffSlogin } 558004e86ffSlogin } 559004e86ffSlogin 560004e86ffSlogin match r.unwrap() { 561004e86ffSlogin FATDirEntryOrShortName::ShortName(short_name) => { 562004e86ffSlogin // 确认名称是一个可行的长文件名 563004e86ffSlogin LongDirEntry::validate_long_name(name)?; 564004e86ffSlogin // 创建目录项 565676b8ef6SMork let x: Result<FATFile, SystemError> = self 566004e86ffSlogin .create_dir_entries( 567004e86ffSlogin name.trim(), 568004e86ffSlogin &short_name, 569004e86ffSlogin None, 570004e86ffSlogin FileAttributes { 571004e86ffSlogin value: FileAttributes::ARCHIVE, 572004e86ffSlogin }, 573004e86ffSlogin fs.clone(), 574004e86ffSlogin ) 575004e86ffSlogin .map(|e| e.to_file())?; 576004e86ffSlogin return x; 577004e86ffSlogin } 578004e86ffSlogin 579004e86ffSlogin FATDirEntryOrShortName::DirEntry(_) => { 580004e86ffSlogin // 已经存在这样的一个目录项了 581676b8ef6SMork return Err(SystemError::EEXIST); 582004e86ffSlogin } 583004e86ffSlogin } 584004e86ffSlogin } 585004e86ffSlogin 586676b8ef6SMork pub fn create_dir(&self, name: &str, fs: &Arc<FATFileSystem>) -> Result<FATDir, SystemError> { 587676b8ef6SMork let r: Result<FATDirEntryOrShortName, SystemError> = 588004e86ffSlogin self.check_existence(name, Some(true), fs.clone()); 589004e86ffSlogin // kdebug!("check existence ok"); 590004e86ffSlogin // 检查错误码,如果能够表明目录项已经存在,则返回-EEXIST 591004e86ffSlogin if r.is_err() { 592004e86ffSlogin let err_val = r.unwrap_err(); 593676b8ef6SMork if err_val == (SystemError::EISDIR) || err_val == (SystemError::ENOTDIR) { 594676b8ef6SMork return Err(SystemError::EEXIST); 595004e86ffSlogin } else { 596004e86ffSlogin return Err(err_val); 597004e86ffSlogin } 598004e86ffSlogin } 599004e86ffSlogin 600004e86ffSlogin match r.unwrap() { 601004e86ffSlogin // 文件夹不存在,创建文件夹 602004e86ffSlogin FATDirEntryOrShortName::ShortName(short_name) => { 603004e86ffSlogin LongDirEntry::validate_long_name(name)?; 604004e86ffSlogin // 目标目录项 605004e86ffSlogin let mut short_entry = ShortDirEntry::default(); 606004e86ffSlogin // kdebug!("to allocate cluster"); 607004e86ffSlogin let first_cluster: Cluster = fs.allocate_cluster(None)?; 608004e86ffSlogin short_entry.set_first_cluster(first_cluster); 609004e86ffSlogin 610004e86ffSlogin // kdebug!("to create dot"); 611004e86ffSlogin // === 接下来在子目录中创建'.'目录项和'..'目录项 612004e86ffSlogin let mut offset = 0; 613004e86ffSlogin // '.'目录项 614004e86ffSlogin let mut dot_entry = ShortDirEntry::default(); 615004e86ffSlogin dot_entry.name = ShortNameGenerator::new(".").generate().unwrap(); 616004e86ffSlogin dot_entry.attributes.value = FileAttributes::DIRECTORY; 617004e86ffSlogin dot_entry.set_first_cluster(first_cluster); 618004e86ffSlogin 619004e86ffSlogin // todo: 设置创建、访问时间 620004e86ffSlogin dot_entry.flush(&fs, fs.cluster_bytes_offset(first_cluster) + offset)?; 621004e86ffSlogin 622004e86ffSlogin drop(dot_entry); 623004e86ffSlogin // 偏移量加上一个目录项的长度 624004e86ffSlogin offset += FATRawDirEntry::DIR_ENTRY_LEN; 625004e86ffSlogin 626004e86ffSlogin // kdebug!("to create dot dot"); 627004e86ffSlogin // '..'目录项 628004e86ffSlogin let mut dot_dot_entry = ShortDirEntry::default(); 629004e86ffSlogin dot_dot_entry.name = ShortNameGenerator::new("..").generate().unwrap(); 630004e86ffSlogin dot_dot_entry.attributes.value = FileAttributes::DIRECTORY; 631004e86ffSlogin dot_dot_entry.set_first_cluster(self.first_cluster); 632004e86ffSlogin // todo: 设置创建、访问时间 633004e86ffSlogin 634004e86ffSlogin dot_dot_entry.flush(&fs, fs.cluster_bytes_offset(first_cluster) + offset)?; 635004e86ffSlogin 636004e86ffSlogin // kdebug!("to create dentries"); 637004e86ffSlogin // 在当前目录下创建目标目录项 638004e86ffSlogin let res = self 639004e86ffSlogin .create_dir_entries( 640004e86ffSlogin name.trim(), 641004e86ffSlogin &short_name, 642004e86ffSlogin Some(short_entry), 643004e86ffSlogin FileAttributes { 644004e86ffSlogin value: FileAttributes::DIRECTORY, 645004e86ffSlogin }, 646004e86ffSlogin fs.clone(), 647004e86ffSlogin ) 648004e86ffSlogin .map(|e| e.to_dir())?; 649004e86ffSlogin // kdebug!("create dentries ok"); 650004e86ffSlogin return res; 651004e86ffSlogin } 652004e86ffSlogin FATDirEntryOrShortName::DirEntry(_) => { 653004e86ffSlogin // 已经存在这样的一个目录项了 654676b8ef6SMork return Err(SystemError::EEXIST); 655004e86ffSlogin } 656004e86ffSlogin } 657004e86ffSlogin } 658004e86ffSlogin /// @brief 检查目录项在当前文件夹下是否存在 659004e86ffSlogin /// 660004e86ffSlogin /// @param name 目录项的名字 661004e86ffSlogin /// @param expect_dir 该值为Some时有效。如果期待目标目录项是文件夹,那么值为Some(true), 否则为Some(false). 662004e86ffSlogin /// @param fs 当前目录所属的文件系统 663004e86ffSlogin /// 664004e86ffSlogin /// @return Ok(FATDirEntryOrShortName::DirEntry) 找到期待的目录项 665004e86ffSlogin /// @return Ok(FATDirEntryOrShortName::ShortName) 当前文件夹下不存在指定的目录项,因此返回一个可行的短文件名 666676b8ef6SMork /// @return Err(SystemError) 错误码 667004e86ffSlogin pub fn check_existence( 668004e86ffSlogin &self, 669004e86ffSlogin name: &str, 670004e86ffSlogin expect_dir: Option<bool>, 671004e86ffSlogin fs: Arc<FATFileSystem>, 672676b8ef6SMork ) -> Result<FATDirEntryOrShortName, SystemError> { 673004e86ffSlogin let mut sng = ShortNameGenerator::new(name); 674004e86ffSlogin 675004e86ffSlogin loop { 676676b8ef6SMork let e: Result<FATDirEntry, SystemError> = 677004e86ffSlogin self.find_entry(name, expect_dir, Some(&mut sng), fs.clone()); 678004e86ffSlogin match e { 679004e86ffSlogin Ok(e) => { 680004e86ffSlogin // 找到,返回目录项 681004e86ffSlogin return Ok(FATDirEntryOrShortName::DirEntry(e)); 682004e86ffSlogin } 683004e86ffSlogin Err(e) => { 684004e86ffSlogin // 如果没找到,则不返回错误 685676b8ef6SMork if e == SystemError::ENOENT { 686004e86ffSlogin } else { 687004e86ffSlogin // 其他错误,则返回 688004e86ffSlogin return Err(e); 689004e86ffSlogin } 690004e86ffSlogin } 691004e86ffSlogin } 692004e86ffSlogin 693004e86ffSlogin // 没找到文件,则生成短文件名 694004e86ffSlogin if let Ok(name) = sng.generate() { 695004e86ffSlogin return Ok(FATDirEntryOrShortName::ShortName(name)); 696004e86ffSlogin } 697004e86ffSlogin 698004e86ffSlogin sng.next_iteration(); 699004e86ffSlogin } 700004e86ffSlogin } 701004e86ffSlogin 702004e86ffSlogin /// @brief 创建一系列的目录项 703004e86ffSlogin /// 704004e86ffSlogin /// @param long_name 长文件名 705004e86ffSlogin /// @param short_name 短文件名 706004e86ffSlogin /// @param short_dentry 可选的生成好的短目录项结构体 707004e86ffSlogin /// @param attrs FAT目录项的属性 708004e86ffSlogin /// @param fs 当前文件夹所属的文件系统 709004e86ffSlogin /// 710004e86ffSlogin /// @return Ok(FATDirEntry) FAT目录项的枚举类型(目录项链条的最后一个长目录项) 711004e86ffSlogin fn create_dir_entries( 712004e86ffSlogin &self, 713004e86ffSlogin long_name: &str, 714004e86ffSlogin short_name: &[u8; 11], 715004e86ffSlogin short_dentry: Option<ShortDirEntry>, 716004e86ffSlogin attrs: FileAttributes, 717004e86ffSlogin fs: Arc<FATFileSystem>, 718676b8ef6SMork ) -> Result<FATDirEntry, SystemError> { 719004e86ffSlogin let mut short_dentry: ShortDirEntry = short_dentry.unwrap_or(ShortDirEntry::default()); 720004e86ffSlogin short_dentry.name = short_name.clone(); 721004e86ffSlogin short_dentry.attributes = attrs; 722004e86ffSlogin 723004e86ffSlogin // todo: 设置创建时间、修改时间 724004e86ffSlogin 725004e86ffSlogin let mut long_name_gen: LongNameEntryGenerator = 726004e86ffSlogin LongNameEntryGenerator::new(long_name, short_dentry.checksum()); 727004e86ffSlogin let num_entries = long_name_gen.num_entries() as u64; 728004e86ffSlogin 729004e86ffSlogin // kdebug!("to find free entries"); 730004e86ffSlogin let free_entries: Option<(Cluster, u64)> = 731004e86ffSlogin self.find_free_entries(num_entries, fs.clone())?; 732004e86ffSlogin // 目录项开始位置 733004e86ffSlogin let start_loc: (Cluster, u64) = match free_entries { 734004e86ffSlogin Some(c) => c, 735676b8ef6SMork None => return Err(SystemError::ENOSPC), 736004e86ffSlogin }; 737004e86ffSlogin let offsets: Vec<(Cluster, u64)> = 738004e86ffSlogin FATDirEntryOffsetIter::new(fs.clone(), start_loc, num_entries, None).collect(); 739004e86ffSlogin 740004e86ffSlogin // 迭代长目录项 741004e86ffSlogin for off in &offsets.as_slice()[..offsets.len() - 1] { 742004e86ffSlogin // 获取生成的下一个长目录项 743004e86ffSlogin let long_entry: LongDirEntry = long_name_gen.next().unwrap(); 744004e86ffSlogin // 获取这个长目录项在磁盘内的字节偏移量 745004e86ffSlogin let bytes_offset = fs.cluster_bytes_offset(off.0) + off.1; 746004e86ffSlogin long_entry.flush(fs.clone(), bytes_offset)?; 747004e86ffSlogin } 748004e86ffSlogin 749004e86ffSlogin let start: (Cluster, u64) = offsets[0]; 750004e86ffSlogin let end: (Cluster, u64) = *offsets.last().unwrap(); 751004e86ffSlogin // 短目录项在磁盘上的字节偏移量 752004e86ffSlogin let offset = fs.cluster_bytes_offset(end.0) + end.1; 753004e86ffSlogin short_dentry.flush(&fs, offset)?; 754004e86ffSlogin 755004e86ffSlogin return Ok(short_dentry.to_dir_entry_with_long_name(long_name.to_string(), (start, end))); 756004e86ffSlogin } 757004e86ffSlogin 758004e86ffSlogin /// @brief 判断当前目录是否为空 759004e86ffSlogin /// 760004e86ffSlogin /// @return true 当前目录为空 761004e86ffSlogin /// @return false 当前目录不为空 762004e86ffSlogin pub fn is_empty(&self, fs: Arc<FATFileSystem>) -> bool { 763004e86ffSlogin for e in self.to_iter(fs) { 764004e86ffSlogin let s = e.short_name(); 765004e86ffSlogin if s == "." || s == ".." { 766004e86ffSlogin continue; 767004e86ffSlogin } else { 768004e86ffSlogin return false; 769004e86ffSlogin } 770004e86ffSlogin } 771004e86ffSlogin return true; 772004e86ffSlogin } 773004e86ffSlogin 774004e86ffSlogin /// @brief 从当前文件夹中删除文件或者文件夹。如果目标文件夹不为空,则不能删除,返回-ENOTEMPTY. 775004e86ffSlogin /// 776004e86ffSlogin /// @param fs 当前FATDir所属的文件系统 777004e86ffSlogin /// @param name 目录项的名字 778004e86ffSlogin /// @param remove_clusters 是否删除与指定的目录项相关联的数据簇 779004e86ffSlogin /// 780004e86ffSlogin /// @return Ok() 成功时无返回值 781676b8ef6SMork /// @return Err(SystemError) 如果目标文件夹不为空,则不能删除,返回-ENOTEMPTY. 或者返回底层传上来的错误 782004e86ffSlogin pub fn remove( 783004e86ffSlogin &self, 784004e86ffSlogin fs: Arc<FATFileSystem>, 785004e86ffSlogin name: &str, 786004e86ffSlogin remove_clusters: bool, 787676b8ef6SMork ) -> Result<(), SystemError> { 788004e86ffSlogin let e: FATDirEntry = self.find_entry(name, None, None, fs.clone())?; 789004e86ffSlogin 790004e86ffSlogin // 判断文件夹是否为空,如果空,则不删除,报错。 791004e86ffSlogin if e.is_dir() && !(e.to_dir().unwrap().is_empty(fs.clone())) { 792676b8ef6SMork return Err(SystemError::ENOTEMPTY); 793004e86ffSlogin } 794004e86ffSlogin 795004e86ffSlogin if e.first_cluster().cluster_num >= 2 && remove_clusters { 796004e86ffSlogin // 删除与指定的目录项相关联的数据簇 797004e86ffSlogin fs.deallocate_cluster_chain(e.first_cluster())?; 798004e86ffSlogin } 799004e86ffSlogin 800004e86ffSlogin if e.get_dir_range().is_some() { 801004e86ffSlogin self.remove_dir_entries(fs, e.get_dir_range().unwrap())?; 802004e86ffSlogin } 803004e86ffSlogin 804004e86ffSlogin return Ok(()); 805004e86ffSlogin } 806004e86ffSlogin 807004e86ffSlogin /// @brief 在当前目录中删除多个目录项 808004e86ffSlogin /// 809004e86ffSlogin /// @param fs 当前目录所属的文件系统 810004e86ffSlogin /// @param cluster_range 要删除的目录项的范围(以簇+簇内偏移量的形式表示) 811004e86ffSlogin fn remove_dir_entries( 812004e86ffSlogin &self, 813004e86ffSlogin fs: Arc<FATFileSystem>, 814004e86ffSlogin cluster_range: ((Cluster, u64), (Cluster, u64)), 815676b8ef6SMork ) -> Result<(), SystemError> { 816004e86ffSlogin // 收集所有的要移除的目录项 817004e86ffSlogin let offsets: Vec<(Cluster, u64)> = 818004e86ffSlogin FATDirEntryOffsetIter::new(fs.clone(), cluster_range.0, 15, Some(cluster_range.1)) 819004e86ffSlogin .collect(); 820004e86ffSlogin // 逐个设置这些目录项为“空闲”状态 821004e86ffSlogin for off in offsets { 822004e86ffSlogin let disk_bytes_offset = fs.cluster_bytes_offset(off.0) + off.1; 823004e86ffSlogin let mut short_entry = ShortDirEntry::default(); 824004e86ffSlogin short_entry.name[0] = 0xe5; 825004e86ffSlogin short_entry.flush(&fs, disk_bytes_offset)?; 826004e86ffSlogin } 827004e86ffSlogin return Ok(()); 828004e86ffSlogin } 829004e86ffSlogin 830004e86ffSlogin /// @brief 根据名字在当前文件夹下寻找目录项 831004e86ffSlogin /// 832004e86ffSlogin /// @return Ok(FATDirEntry) 目标目录项 833676b8ef6SMork /// @return Err(SystemError) 底层传上来的错误码 834*78bf93f0SYJwu2023 pub fn get_dir_entry( 835*78bf93f0SYJwu2023 &self, 836*78bf93f0SYJwu2023 fs: Arc<FATFileSystem>, 837*78bf93f0SYJwu2023 name: &str, 838*78bf93f0SYJwu2023 ) -> Result<FATDirEntry, SystemError> { 839004e86ffSlogin if name == "." || name == "/" { 840004e86ffSlogin return Ok(FATDirEntry::Dir(self.clone())); 841004e86ffSlogin } 842004e86ffSlogin 843004e86ffSlogin LongDirEntry::validate_long_name(name)?; 844004e86ffSlogin return self.find_entry(name, None, None, fs); 845004e86ffSlogin } 846004e86ffSlogin 847004e86ffSlogin /// @brief 在当前目录内,重命名一个目录项 848004e86ffSlogin /// 849004e86ffSlogin pub fn rename( 850004e86ffSlogin &self, 851004e86ffSlogin fs: Arc<FATFileSystem>, 852004e86ffSlogin old_name: &str, 853004e86ffSlogin new_name: &str, 854676b8ef6SMork ) -> Result<FATDirEntry, SystemError> { 855004e86ffSlogin // 判断源目录项是否存在 856004e86ffSlogin let old_dentry: FATDirEntry = if let FATDirEntryOrShortName::DirEntry(dentry) = 857004e86ffSlogin self.check_existence(old_name, None, fs.clone())? 858004e86ffSlogin { 859004e86ffSlogin dentry 860004e86ffSlogin } else { 861004e86ffSlogin // 如果目标目录项不存在,则返回错误 862676b8ef6SMork return Err(SystemError::ENOENT); 863004e86ffSlogin }; 864004e86ffSlogin 865004e86ffSlogin let short_name = if let FATDirEntryOrShortName::ShortName(s) = 866004e86ffSlogin self.check_existence(new_name, None, fs.clone())? 867004e86ffSlogin { 868004e86ffSlogin s 869004e86ffSlogin } else { 870004e86ffSlogin // 如果目标目录项存在,那么就返回错误 871676b8ef6SMork return Err(SystemError::EEXIST); 872004e86ffSlogin }; 873004e86ffSlogin 874004e86ffSlogin let old_short_dentry: Option<ShortDirEntry> = old_dentry.short_dir_entry(); 875004e86ffSlogin if let Some(se) = old_short_dentry { 876004e86ffSlogin // 删除原来的目录项 877004e86ffSlogin self.remove(fs.clone(), old_dentry.name().as_str(), false)?; 878004e86ffSlogin 879004e86ffSlogin // 创建新的目录项 880004e86ffSlogin let new_dentry: FATDirEntry = self.create_dir_entries( 881004e86ffSlogin new_name, 882004e86ffSlogin &short_name, 883004e86ffSlogin Some(se), 884004e86ffSlogin se.attributes, 885004e86ffSlogin fs.clone(), 886004e86ffSlogin )?; 887004e86ffSlogin 888004e86ffSlogin return Ok(new_dentry); 889004e86ffSlogin } else { 890004e86ffSlogin // 不允许对根目录项进行重命名 891676b8ef6SMork return Err(SystemError::EPERM); 892004e86ffSlogin } 893004e86ffSlogin } 894004e86ffSlogin } 895004e86ffSlogin 896004e86ffSlogin impl FileAttributes { 897004e86ffSlogin pub const READ_ONLY: u8 = 1 << 0; 898004e86ffSlogin pub const HIDDEN: u8 = 1 << 1; 899004e86ffSlogin pub const SYSTEM: u8 = 1 << 2; 900004e86ffSlogin pub const VOLUME_ID: u8 = 1 << 3; 901004e86ffSlogin pub const DIRECTORY: u8 = 1 << 4; 902004e86ffSlogin pub const ARCHIVE: u8 = 1 << 5; 903004e86ffSlogin pub const LONG_NAME: u8 = FileAttributes::READ_ONLY 904004e86ffSlogin | FileAttributes::HIDDEN 905004e86ffSlogin | FileAttributes::SYSTEM 906004e86ffSlogin | FileAttributes::VOLUME_ID; 907004e86ffSlogin 908004e86ffSlogin /// @brief 判断属性是否存在 909004e86ffSlogin #[inline] 910004e86ffSlogin pub fn contains(&self, attr: u8) -> bool { 911004e86ffSlogin return (self.value & attr) != 0; 912004e86ffSlogin } 913004e86ffSlogin 914004e86ffSlogin pub fn new(attr: u8) -> Self { 915004e86ffSlogin return Self { value: attr }; 916004e86ffSlogin } 917004e86ffSlogin } 918004e86ffSlogin 919004e86ffSlogin /// FAT32的短目录项 920004e86ffSlogin #[derive(Debug, Clone, Copy, Default)] 921004e86ffSlogin pub struct ShortDirEntry { 922004e86ffSlogin /// short name 923004e86ffSlogin name: [u8; 11], 924004e86ffSlogin /// 目录项属性 (见 FileAttributes ) 925004e86ffSlogin attributes: FileAttributes, 926004e86ffSlogin 927004e86ffSlogin /// Windows NT系统的保留字段。用来表示短目录项文件名。 928004e86ffSlogin /// EXT|BASE => 8(BASE).3(EXT) 929004e86ffSlogin /// BASE:LowerCase(8),UpperCase(0) 930004e86ffSlogin /// EXT:LowerCase(16),UpperCase(0) 931004e86ffSlogin nt_res: u8, 932004e86ffSlogin 933004e86ffSlogin /// 文件创建时间的毫秒级时间戳 934004e86ffSlogin crt_time_tenth: u8, 935004e86ffSlogin /// 创建时间 936004e86ffSlogin crt_time: u16, 937004e86ffSlogin /// 创建日期 938004e86ffSlogin crt_date: u16, 939004e86ffSlogin /// 最后一次访问日期 940004e86ffSlogin lst_acc_date: u16, 941004e86ffSlogin /// High word of first cluster(0 for FAT12 and FAT16) 942004e86ffSlogin fst_clus_hi: u16, 943004e86ffSlogin /// 最后写入时间 944004e86ffSlogin wrt_time: u16, 945004e86ffSlogin /// 最后写入日期 946004e86ffSlogin wrt_date: u16, 947004e86ffSlogin /// Low word of first cluster 948004e86ffSlogin fst_clus_lo: u16, 949004e86ffSlogin /// 文件大小 950004e86ffSlogin file_size: u32, 951004e86ffSlogin } 952004e86ffSlogin 953004e86ffSlogin /// FAT32的长目录项 954004e86ffSlogin #[derive(Debug, Clone, Copy, Default)] 955004e86ffSlogin pub struct LongDirEntry { 956004e86ffSlogin /// 长目录项的序号 957004e86ffSlogin ord: u8, 958004e86ffSlogin /// 长文件名的第1-5个字符,每个字符占2bytes 959004e86ffSlogin name1: [u16; 5], 960004e86ffSlogin /// 目录项属性必须为ATTR_LONG_NAME 961004e86ffSlogin file_attrs: FileAttributes, 962004e86ffSlogin /// Entry Type: 如果为0,则说明这是长目录项的子项 963004e86ffSlogin /// 非零值是保留的。 964004e86ffSlogin dirent_type: u8, 965004e86ffSlogin /// 短文件名的校验和 966004e86ffSlogin checksum: u8, 967004e86ffSlogin /// 长文件名的第6-11个字符,每个字符占2bytes 968004e86ffSlogin name2: [u16; 6], 969004e86ffSlogin /// 必须为0 970004e86ffSlogin first_clus_low: u16, 971004e86ffSlogin /// 长文件名的12-13个字符,每个字符占2bytes 972004e86ffSlogin name3: [u16; 2], 973004e86ffSlogin } 974004e86ffSlogin 975004e86ffSlogin impl LongDirEntry { 976004e86ffSlogin /// 长目录项的字符串长度(单位:word) 977004e86ffSlogin pub const LONG_NAME_STR_LEN: usize = 13; 978004e86ffSlogin 979004e86ffSlogin /// @brief 初始化一个新的长目录项 980004e86ffSlogin /// 981004e86ffSlogin /// @param ord 顺序 982004e86ffSlogin /// @param name_part 长目录项名称的数组(长度必须为13) 983004e86ffSlogin /// @param check_sum 短目录项的校验和 984004e86ffSlogin /// 985004e86ffSlogin /// @return Self 初始化好的长目录项对象 986004e86ffSlogin fn new(ord: u8, name_part: &[u16], check_sum: u8) -> Self { 987004e86ffSlogin let mut result = LongDirEntry::default(); 988004e86ffSlogin result.ord = ord; 989004e86ffSlogin result 990004e86ffSlogin .insert_name(name_part) 991004e86ffSlogin .expect("Name part's len should be equal to 13."); 992004e86ffSlogin result.file_attrs.value = FileAttributes::LONG_NAME; 993004e86ffSlogin result.dirent_type = 0; 994004e86ffSlogin result.checksum = check_sum; 995004e86ffSlogin // 该字段需要外层的代码手动赋值 996004e86ffSlogin result.first_clus_low = 0; 997004e86ffSlogin return result; 998004e86ffSlogin } 999004e86ffSlogin 1000004e86ffSlogin /// @brief 填写长目录项的名称字段。 1001004e86ffSlogin /// 1002004e86ffSlogin /// @param name_part 要被填入当前长目录项的名字(数组长度必须为13) 1003004e86ffSlogin /// 1004004e86ffSlogin /// @return Ok(()) 1005676b8ef6SMork /// @return Err(SystemError) 错误码 1006676b8ef6SMork fn insert_name(&mut self, name_part: &[u16]) -> Result<(), SystemError> { 1007004e86ffSlogin if name_part.len() != Self::LONG_NAME_STR_LEN { 1008676b8ef6SMork return Err(SystemError::EINVAL); 1009004e86ffSlogin } 1010004e86ffSlogin self.name1.copy_from_slice(&name_part[0..5]); 1011004e86ffSlogin self.name2.copy_from_slice(&name_part[5..11]); 1012004e86ffSlogin self.name3.copy_from_slice(&name_part[11..13]); 1013004e86ffSlogin return Ok(()); 1014004e86ffSlogin } 1015004e86ffSlogin 1016004e86ffSlogin /// @brief 将当前长目录项的名称字段,原样地拷贝到一个长度为13的u16数组中。 1017004e86ffSlogin /// @param dst 拷贝的目的地,一个[u16]数组,长度必须为13。 1018676b8ef6SMork pub fn copy_name_to_slice(&self, dst: &mut [u16]) -> Result<(), SystemError> { 1019004e86ffSlogin if dst.len() != Self::LONG_NAME_STR_LEN { 1020676b8ef6SMork return Err(SystemError::EINVAL); 1021004e86ffSlogin } 1022004e86ffSlogin dst[0..5].copy_from_slice(&self.name1); 1023004e86ffSlogin dst[5..11].copy_from_slice(&self.name2); 1024004e86ffSlogin dst[11..13].copy_from_slice(&self.name3); 1025004e86ffSlogin return Ok(()); 1026004e86ffSlogin } 1027004e86ffSlogin 1028004e86ffSlogin /// @brief 是否为最后一个长目录项 1029004e86ffSlogin /// 1030004e86ffSlogin /// @return true 是最后一个长目录项 1031004e86ffSlogin /// @return false 不是最后一个长目录项 1032004e86ffSlogin pub fn is_last(&self) -> bool { 1033004e86ffSlogin return self.ord & 0x40 > 0; 1034004e86ffSlogin } 1035004e86ffSlogin 1036004e86ffSlogin /// @brief 校验字符串是否符合长目录项的命名要求 1037004e86ffSlogin /// 1038004e86ffSlogin /// @return Ok(()) 名称合法 1039676b8ef6SMork /// @return Err(SystemError) 名称不合法,返回错误码 1040676b8ef6SMork pub fn validate_long_name(mut name: &str) -> Result<(), SystemError> { 1041004e86ffSlogin // 去除首尾多余的空格 1042004e86ffSlogin name = name.trim(); 1043004e86ffSlogin 1044004e86ffSlogin // 名称不能为0 1045004e86ffSlogin if name.len() == 0 { 1046676b8ef6SMork return Err(SystemError::EINVAL); 1047004e86ffSlogin } 1048004e86ffSlogin 1049004e86ffSlogin // 名称长度不能大于255 1050004e86ffSlogin if name.len() > 255 { 1051676b8ef6SMork return Err(SystemError::ENAMETOOLONG); 1052004e86ffSlogin } 1053004e86ffSlogin 1054004e86ffSlogin // 检查是否符合命名要求 1055004e86ffSlogin for c in name.chars() { 1056004e86ffSlogin match c { 1057004e86ffSlogin 'a'..='z' | 'A'..='Z' | '0'..='9' => {} 1058004e86ffSlogin '\u{80}'..='\u{ffff}' => {} 1059004e86ffSlogin '$' | '%' | '\'' | '-' | '_' | '@' | '~' | '`' | '!' | '(' | ')' | '{' | '}' 1060004e86ffSlogin | '^' | '#' | '&' => {} 1061004e86ffSlogin '+' | ',' | ';' | '=' | '[' | ']' | '.' | ' ' => {} 1062004e86ffSlogin _ => { 1063676b8ef6SMork return Err(SystemError::EILSEQ); 1064004e86ffSlogin } 1065004e86ffSlogin } 1066004e86ffSlogin } 1067004e86ffSlogin return Ok(()); 1068004e86ffSlogin } 1069004e86ffSlogin 1070004e86ffSlogin /// @brief 把当前长目录项写入磁盘 1071004e86ffSlogin /// 1072004e86ffSlogin /// @param fs 对应的文件系统 1073004e86ffSlogin /// @param disk_bytes_offset 长目录项所在位置对应的在磁盘上的字节偏移量 1074004e86ffSlogin /// 1075004e86ffSlogin /// @return Ok(()) 1076676b8ef6SMork /// @return Err(SystemError) 错误码 1077676b8ef6SMork pub fn flush(&self, fs: Arc<FATFileSystem>, disk_bytes_offset: u64) -> Result<(), SystemError> { 1078004e86ffSlogin // 从磁盘读取数据 1079004e86ffSlogin let blk_offset = fs.get_in_block_offset(disk_bytes_offset); 1080004e86ffSlogin let lba = fs.get_lba_from_offset( 1081004e86ffSlogin fs.bytes_to_sector(fs.get_in_partition_bytes_offset(disk_bytes_offset)), 1082004e86ffSlogin ); 1083004e86ffSlogin let mut v: Vec<u8> = Vec::new(); 1084004e86ffSlogin v.resize(1 * fs.lba_per_sector() * LBA_SIZE, 0); 1085004e86ffSlogin fs.partition 1086004e86ffSlogin .disk() 1087004e86ffSlogin .read_at(lba, 1 * fs.lba_per_sector(), &mut v)?; 1088004e86ffSlogin 1089004e86ffSlogin let mut cursor: VecCursor = VecCursor::new(v); 1090004e86ffSlogin // 切换游标到对应位置 1091004e86ffSlogin cursor.seek(SeekFrom::SeekSet(blk_offset as i64))?; 1092004e86ffSlogin 1093004e86ffSlogin // 写入数据 1094004e86ffSlogin cursor.write_u8(self.ord)?; 1095004e86ffSlogin for b in &self.name1 { 1096004e86ffSlogin cursor.write_u16(*b)?; 1097004e86ffSlogin } 1098004e86ffSlogin 1099004e86ffSlogin cursor.write_u8(self.file_attrs.value)?; 1100004e86ffSlogin cursor.write_u8(self.dirent_type)?; 1101004e86ffSlogin cursor.write_u8(self.checksum)?; 1102004e86ffSlogin 1103004e86ffSlogin for b in &self.name2 { 1104004e86ffSlogin cursor.write_u16(*b)?; 1105004e86ffSlogin } 1106004e86ffSlogin 1107004e86ffSlogin cursor.write_u16(self.first_clus_low)?; 1108004e86ffSlogin 1109004e86ffSlogin for b in &self.name3 { 1110004e86ffSlogin cursor.write_u16(*b)?; 1111004e86ffSlogin } 1112004e86ffSlogin 1113004e86ffSlogin // 把修改后的长目录项刷入磁盘 1114004e86ffSlogin fs.partition 1115004e86ffSlogin .disk() 1116004e86ffSlogin .write_at(lba, 1 * fs.lba_per_sector(), cursor.as_slice())?; 1117004e86ffSlogin fs.partition.disk().sync()?; 1118004e86ffSlogin 1119004e86ffSlogin return Ok(()); 1120004e86ffSlogin } 1121004e86ffSlogin } 1122004e86ffSlogin 1123004e86ffSlogin impl ShortDirEntry { 1124004e86ffSlogin const PADDING: u8 = ' ' as u8; 1125004e86ffSlogin 1126004e86ffSlogin /// @brief 判断当前目录项是否为文件夹 1127004e86ffSlogin /// 1128004e86ffSlogin /// @return true 是文件夹 1129004e86ffSlogin /// @return false 不是文件夹 1130004e86ffSlogin pub fn is_dir(&self) -> bool { 1131004e86ffSlogin return (self.attributes.contains(FileAttributes::DIRECTORY)) 1132004e86ffSlogin && (!self.attributes.contains(FileAttributes::VOLUME_ID)); 1133004e86ffSlogin } 1134004e86ffSlogin 1135004e86ffSlogin /// @brief 判断当前目录项是否为文件 1136004e86ffSlogin /// 1137004e86ffSlogin /// @return true 是文件 1138004e86ffSlogin /// @return false 不是文件 1139004e86ffSlogin pub fn is_file(&self) -> bool { 1140004e86ffSlogin return (!self.attributes.contains(FileAttributes::DIRECTORY)) 1141004e86ffSlogin && (!self.attributes.contains(FileAttributes::VOLUME_ID)); 1142004e86ffSlogin } 1143004e86ffSlogin 1144004e86ffSlogin /// @brief 判断当前目录项是否为卷号 1145004e86ffSlogin /// 1146004e86ffSlogin /// @return true 是卷号 1147004e86ffSlogin /// @return false 不是卷号 1148004e86ffSlogin pub fn is_volume_id(&self) -> bool { 1149004e86ffSlogin return (!self.attributes.contains(FileAttributes::DIRECTORY)) 1150004e86ffSlogin && self.attributes.contains(FileAttributes::VOLUME_ID); 1151004e86ffSlogin } 1152004e86ffSlogin 1153004e86ffSlogin /// @brief 将短目录项的名字转换为String 1154004e86ffSlogin fn name_to_string(&self) -> String { 1155004e86ffSlogin // 计算基础名的长度 1156004e86ffSlogin let base_len = self.name[..8] 1157004e86ffSlogin .iter() 1158004e86ffSlogin .rposition(|x| *x != ShortDirEntry::PADDING) 1159004e86ffSlogin .map(|len| len + 1) 1160004e86ffSlogin .unwrap_or(0); 1161004e86ffSlogin // 计算扩展名的长度 1162004e86ffSlogin let ext_len = self.name[8..] 1163004e86ffSlogin .iter() 1164004e86ffSlogin .rposition(|x| *x != ShortDirEntry::PADDING) 1165004e86ffSlogin .map(|len| len + 1) 1166004e86ffSlogin .unwrap_or(0); 1167004e86ffSlogin 1168004e86ffSlogin // 声明存储完整名字的数组(包含“.”) 1169004e86ffSlogin let mut name = [ShortDirEntry::PADDING; 12]; 1170004e86ffSlogin // 拷贝基础名 1171004e86ffSlogin name[..base_len].copy_from_slice(&self.name[..base_len]); 1172004e86ffSlogin 1173004e86ffSlogin // 拷贝扩展名,并计算总的长度 1174004e86ffSlogin let total_len = if ext_len > 0 { 1175004e86ffSlogin name[base_len] = '.' as u8; 1176004e86ffSlogin name[base_len + 1..base_len + 1 + ext_len].copy_from_slice(&self.name[8..8 + ext_len]); 1177004e86ffSlogin // 总长度为基础名长度+点号+扩展名长度 1178004e86ffSlogin base_len + 1 + ext_len 1179004e86ffSlogin } else { 1180004e86ffSlogin base_len 1181004e86ffSlogin }; 1182004e86ffSlogin 1183004e86ffSlogin if name[0] == 0x05 { 1184004e86ffSlogin name[0] = 0xe5; 1185004e86ffSlogin } 1186004e86ffSlogin 1187004e86ffSlogin let iter = name[..total_len].iter().map(|c| decode_u8_ascii(*c)); 1188004e86ffSlogin // 返回最终的字符串 1189004e86ffSlogin return String::from_iter(iter); 1190004e86ffSlogin } 1191004e86ffSlogin 1192004e86ffSlogin /// @brief 将短目录项结构体,转换为FATDirEntry枚举类型 1193004e86ffSlogin /// 1194004e86ffSlogin /// @param loc 当前文件的起始、终止簇。格式:(簇,簇内偏移量) 1195004e86ffSlogin /// @return 生成的FATDirENtry枚举类型 1196004e86ffSlogin pub fn to_dir_entry(&self, loc: (Cluster, u64)) -> FATDirEntry { 1197004e86ffSlogin // 当前文件的第一个簇 1198004e86ffSlogin let first_cluster = 1199004e86ffSlogin Cluster::new(((self.fst_clus_hi as u64) << 16) | (self.fst_clus_lo as u64)); 1200004e86ffSlogin 1201004e86ffSlogin // 当前是文件或卷号 1202004e86ffSlogin if self.is_file() || self.is_volume_id() { 1203004e86ffSlogin let mut file: FATFile = FATFile::default(); 1204004e86ffSlogin 1205004e86ffSlogin file.file_name = self.name_to_string(); 1206004e86ffSlogin file.first_cluster = first_cluster; 1207004e86ffSlogin file.short_dir_entry = self.clone(); 1208004e86ffSlogin file.loc = (loc, loc); 1209004e86ffSlogin 1210004e86ffSlogin // 根据当前短目录项的类型的不同,返回对应的枚举类型。 1211004e86ffSlogin if self.is_file() { 1212004e86ffSlogin return FATDirEntry::File(file); 1213004e86ffSlogin } else { 1214004e86ffSlogin return FATDirEntry::VolId(file); 1215004e86ffSlogin } 1216004e86ffSlogin } else { 1217004e86ffSlogin // 当前是文件夹 1218004e86ffSlogin let mut dir = FATDir::default(); 1219004e86ffSlogin dir.dir_name = self.name_to_string(); 1220004e86ffSlogin dir.first_cluster = first_cluster; 1221004e86ffSlogin dir.root_offset = None; 1222004e86ffSlogin dir.short_dir_entry = Some(self.clone()); 1223004e86ffSlogin dir.loc = Some((loc, loc)); 1224004e86ffSlogin 1225004e86ffSlogin return FATDirEntry::Dir(dir); 1226004e86ffSlogin } 1227004e86ffSlogin } 1228004e86ffSlogin 1229004e86ffSlogin /// @brief 将短目录项结构体,转换为FATDirEntry枚举类型. 并且,该短目录项具有对应的长目录项。 1230004e86ffSlogin /// 因此,需要传入从长目录项获得的完整的文件名 1231004e86ffSlogin /// 1232004e86ffSlogin /// @param name 从长目录项获取的完整文件名 1233004e86ffSlogin /// @param loc 当前文件的起始、终止簇。格式:(簇,簇内偏移量) 1234004e86ffSlogin /// @return 生成的FATDirENtry枚举类型 1235004e86ffSlogin pub fn to_dir_entry_with_long_name( 1236004e86ffSlogin &self, 1237004e86ffSlogin name: String, 1238004e86ffSlogin loc: ((Cluster, u64), (Cluster, u64)), 1239004e86ffSlogin ) -> FATDirEntry { 1240004e86ffSlogin // 当前文件的第一个簇 1241004e86ffSlogin let first_cluster = 1242004e86ffSlogin Cluster::new(((self.fst_clus_hi as u64) << 16) | (self.fst_clus_lo as u64)); 1243004e86ffSlogin 1244004e86ffSlogin if self.is_file() || self.is_volume_id() { 1245004e86ffSlogin let mut file = FATFile::default(); 1246004e86ffSlogin 1247004e86ffSlogin file.first_cluster = first_cluster; 1248004e86ffSlogin file.file_name = name; 1249004e86ffSlogin file.loc = loc; 1250004e86ffSlogin file.short_dir_entry = self.clone(); 1251004e86ffSlogin 1252004e86ffSlogin if self.is_file() { 1253004e86ffSlogin return FATDirEntry::File(file); 1254004e86ffSlogin } else { 1255004e86ffSlogin return FATDirEntry::VolId(file); 1256004e86ffSlogin } 1257004e86ffSlogin } else { 1258004e86ffSlogin let mut dir = FATDir::default(); 1259004e86ffSlogin 1260004e86ffSlogin dir.first_cluster = first_cluster; 1261004e86ffSlogin dir.dir_name = name; 1262004e86ffSlogin dir.loc = Some(loc); 1263004e86ffSlogin dir.short_dir_entry = Some(self.clone()); 1264004e86ffSlogin dir.root_offset = None; 1265004e86ffSlogin 1266004e86ffSlogin return FATDirEntry::Dir(dir); 1267004e86ffSlogin } 1268004e86ffSlogin } 1269004e86ffSlogin 1270004e86ffSlogin /// @brief 计算短目录项的名称的校验和 1271004e86ffSlogin fn checksum(&self) -> u8 { 1272004e86ffSlogin let mut result = 0; 1273004e86ffSlogin 1274004e86ffSlogin for c in &self.name { 1275004e86ffSlogin result = (result << 7) + (result >> 1) + *c; 1276004e86ffSlogin } 1277004e86ffSlogin return result; 1278004e86ffSlogin } 1279004e86ffSlogin 1280004e86ffSlogin /// @brief 把当前短目录项写入磁盘 1281004e86ffSlogin /// 1282004e86ffSlogin /// @param fs 对应的文件系统 1283004e86ffSlogin /// @param disk_bytes_offset 短目录项所在位置对应的在磁盘上的字节偏移量 1284004e86ffSlogin /// 1285004e86ffSlogin /// @return Ok(()) 1286676b8ef6SMork /// @return Err(SystemError) 错误码 1287*78bf93f0SYJwu2023 pub fn flush( 1288*78bf93f0SYJwu2023 &self, 1289*78bf93f0SYJwu2023 fs: &Arc<FATFileSystem>, 1290*78bf93f0SYJwu2023 disk_bytes_offset: u64, 1291*78bf93f0SYJwu2023 ) -> Result<(), SystemError> { 1292004e86ffSlogin // 从磁盘读取数据 1293004e86ffSlogin let blk_offset = fs.get_in_block_offset(disk_bytes_offset); 1294004e86ffSlogin let lba = fs.get_lba_from_offset( 1295004e86ffSlogin fs.bytes_to_sector(fs.get_in_partition_bytes_offset(disk_bytes_offset)), 1296004e86ffSlogin ); 1297004e86ffSlogin let mut v: Vec<u8> = Vec::new(); 1298004e86ffSlogin v.resize(1 * fs.lba_per_sector() * LBA_SIZE, 0); 1299004e86ffSlogin fs.partition 1300004e86ffSlogin .disk() 1301004e86ffSlogin .read_at(lba, 1 * fs.lba_per_sector(), &mut v)?; 1302004e86ffSlogin 1303004e86ffSlogin let mut cursor: VecCursor = VecCursor::new(v); 1304004e86ffSlogin // 切换游标到对应位置 1305004e86ffSlogin cursor.seek(SeekFrom::SeekSet(blk_offset as i64))?; 1306004e86ffSlogin cursor.write_exact(&self.name)?; 1307004e86ffSlogin cursor.write_u8(self.attributes.value)?; 1308004e86ffSlogin cursor.write_u8(self.nt_res)?; 1309004e86ffSlogin cursor.write_u8(self.crt_time_tenth)?; 1310004e86ffSlogin cursor.write_u16(self.crt_time)?; 1311004e86ffSlogin cursor.write_u16(self.crt_date)?; 1312004e86ffSlogin cursor.write_u16(self.lst_acc_date)?; 1313004e86ffSlogin cursor.write_u16(self.fst_clus_hi)?; 1314004e86ffSlogin cursor.write_u16(self.wrt_time)?; 1315004e86ffSlogin cursor.write_u16(self.wrt_date)?; 1316004e86ffSlogin cursor.write_u16(self.fst_clus_lo)?; 1317004e86ffSlogin cursor.write_u32(self.file_size)?; 1318004e86ffSlogin 1319004e86ffSlogin // 把修改后的长目录项刷入磁盘 1320004e86ffSlogin fs.partition 1321004e86ffSlogin .disk() 1322004e86ffSlogin .write_at(lba, 1 * fs.lba_per_sector(), cursor.as_slice())?; 1323004e86ffSlogin fs.partition.disk().sync()?; 1324004e86ffSlogin 1325004e86ffSlogin return Ok(()); 1326004e86ffSlogin } 1327004e86ffSlogin 1328004e86ffSlogin /// @brief 设置短目录项的“第一个簇”字段的值 1329004e86ffSlogin pub fn set_first_cluster(&mut self, cluster: Cluster) { 1330004e86ffSlogin self.fst_clus_lo = (cluster.cluster_num & 0x0000ffff) as u16; 1331004e86ffSlogin self.fst_clus_hi = ((cluster.cluster_num & 0xffff0000) >> 16) as u16; 1332004e86ffSlogin } 1333004e86ffSlogin } 1334004e86ffSlogin 1335004e86ffSlogin /// @brief FAT文件系统标准定义的目录项 1336004e86ffSlogin #[derive(Debug, Clone)] 1337004e86ffSlogin pub enum FATRawDirEntry { 1338004e86ffSlogin /// 短目录项 1339004e86ffSlogin Short(ShortDirEntry), 1340004e86ffSlogin /// 长目录项 1341004e86ffSlogin Long(LongDirEntry), 1342004e86ffSlogin /// 当前目录项的Name[0]==0xe5, 是空闲目录项 1343004e86ffSlogin Free, 1344004e86ffSlogin /// 当前目录项的Name[0]==0xe5, 是空闲目录项,且在这之后没有被分配过的目录项了。 1345004e86ffSlogin FreeRest, 1346004e86ffSlogin } 1347004e86ffSlogin 1348004e86ffSlogin impl FATRawDirEntry { 1349004e86ffSlogin /// 每个目录项的长度(单位:字节) 1350004e86ffSlogin pub const DIR_ENTRY_LEN: u64 = 32; 1351004e86ffSlogin 1352004e86ffSlogin /// @brief 判断当前目录项是否为这个文件的最后一个目录项 1353004e86ffSlogin fn is_last(&self) -> bool { 1354004e86ffSlogin match self { 1355004e86ffSlogin &Self::Short(_) => { 1356004e86ffSlogin return true; 1357004e86ffSlogin } 1358004e86ffSlogin &Self::Long(l) => { 1359004e86ffSlogin return l.is_last(); 1360004e86ffSlogin } 1361004e86ffSlogin _ => { 1362004e86ffSlogin return false; 1363004e86ffSlogin } 1364004e86ffSlogin } 1365004e86ffSlogin } 1366004e86ffSlogin 1367004e86ffSlogin /// @brief 判断当前目录项是否为长目录项 1368004e86ffSlogin fn is_long(&self) -> bool { 1369004e86ffSlogin if let Self::Long(_) = self { 1370004e86ffSlogin return true; 1371004e86ffSlogin } else { 1372004e86ffSlogin return false; 1373004e86ffSlogin } 1374004e86ffSlogin } 1375004e86ffSlogin 1376004e86ffSlogin /// @brief 判断当前目录项是否为短目录项 1377004e86ffSlogin fn is_short(&self) -> bool { 1378004e86ffSlogin if let Self::Short(_) = self { 1379004e86ffSlogin return true; 1380004e86ffSlogin } else { 1381004e86ffSlogin return false; 1382004e86ffSlogin } 1383004e86ffSlogin } 1384004e86ffSlogin } 1385004e86ffSlogin 1386004e86ffSlogin /// @brief FAT文件系统的目录项迭代器 1387004e86ffSlogin #[derive(Debug)] 1388004e86ffSlogin pub struct FATDirIter { 1389004e86ffSlogin /// 当前正在迭代的簇 1390004e86ffSlogin current_cluster: Cluster, 1391004e86ffSlogin /// 当前正在迭代的簇的簇内偏移量 1392004e86ffSlogin offset: u64, 1393004e86ffSlogin /// True for the root directories of FAT12 and FAT16 1394004e86ffSlogin is_root: bool, 1395004e86ffSlogin /// 指向当前文件系统的指针 1396004e86ffSlogin fs: Arc<FATFileSystem>, 1397004e86ffSlogin } 1398004e86ffSlogin 1399004e86ffSlogin impl FATDirIter { 1400004e86ffSlogin /// @brief 迭代当前inode的目录项(获取下一个目录项) 1401004e86ffSlogin /// 1402004e86ffSlogin /// @return Ok(Cluster, u64, Option<FATDirEntry>) 1403004e86ffSlogin /// Cluster: 下一个要读取的簇号 1404004e86ffSlogin /// u64: 下一个要读取的簇内偏移量 1405004e86ffSlogin /// Option<FATDirEntry>: 读取到的目录项(如果没有读取到,就返回失败) 1406004e86ffSlogin /// @return Err(错误码) 可能出现了内部错误,或者是磁盘错误等。具体原因看错误码。 1407676b8ef6SMork fn get_dir_entry(&mut self) -> Result<(Cluster, u64, Option<FATDirEntry>), SystemError> { 1408004e86ffSlogin loop { 1409004e86ffSlogin // 如果当前簇已经被读完,那么尝试获取下一个簇 1410004e86ffSlogin if self.offset >= self.fs.bytes_per_cluster() && !self.is_root { 1411004e86ffSlogin match self.fs.get_fat_entry(self.current_cluster)? { 1412004e86ffSlogin FATEntry::Next(c) => { 1413004e86ffSlogin // 获得下一个簇的信息 1414004e86ffSlogin self.current_cluster = c; 1415004e86ffSlogin self.offset %= self.fs.bytes_per_cluster(); 1416004e86ffSlogin } 1417004e86ffSlogin 1418004e86ffSlogin _ => { 1419004e86ffSlogin // 没有下一个簇了,返回None 1420004e86ffSlogin return Ok((self.current_cluster, self.offset, None)); 1421004e86ffSlogin } 1422004e86ffSlogin } 1423004e86ffSlogin } 1424004e86ffSlogin 1425004e86ffSlogin // 如果当前是FAT12/FAT16文件系统,并且当前inode是根目录项。 1426004e86ffSlogin // 如果offset大于根目录项的最大大小(已经遍历完根目录),那么就返回None 1427004e86ffSlogin if self.is_root && self.offset > self.fs.root_dir_end_bytes_offset().unwrap() { 1428004e86ffSlogin return Ok((self.current_cluster, self.offset, None)); 1429004e86ffSlogin } 1430004e86ffSlogin 1431004e86ffSlogin // 获取簇在磁盘内的字节偏移量 1432004e86ffSlogin let offset: u64 = self.fs.cluster_bytes_offset(self.current_cluster) + self.offset; 1433004e86ffSlogin 1434004e86ffSlogin // 从磁盘读取原始的dentry 1435004e86ffSlogin let raw_dentry: FATRawDirEntry = get_raw_dir_entry(&self.fs, offset)?; 1436004e86ffSlogin 1437004e86ffSlogin // 由于迭代顺序从前往后,因此: 1438004e86ffSlogin // 如果找到1个短目录项,那么证明有一个完整的entry被找到,因此返回。 1439004e86ffSlogin // 如果找到1个长目录项,那么,就依次往下迭代查找,直到找到一个短目录项,然后返回结果。这里找到的所有的目录项,都属于同一个文件/文件夹。 1440004e86ffSlogin match raw_dentry { 1441004e86ffSlogin FATRawDirEntry::Short(s) => { 1442004e86ffSlogin // 当前找到一个短目录项,更新offset之后,直接返回 1443004e86ffSlogin self.offset += FATRawDirEntry::DIR_ENTRY_LEN; 1444004e86ffSlogin return Ok(( 1445004e86ffSlogin self.current_cluster, 1446004e86ffSlogin self.offset, 1447004e86ffSlogin Some(s.to_dir_entry(( 1448004e86ffSlogin self.current_cluster, 1449004e86ffSlogin self.offset - FATRawDirEntry::DIR_ENTRY_LEN, 1450004e86ffSlogin ))), 1451004e86ffSlogin )); 1452004e86ffSlogin } 1453004e86ffSlogin FATRawDirEntry::Long(_) => { 1454004e86ffSlogin // 当前找到一个长目录项 1455004e86ffSlogin 1456004e86ffSlogin // 声明一个数组,来容纳所有的entry。(先把最后一个entry放进去) 1457004e86ffSlogin let mut long_name_entries: Vec<FATRawDirEntry> = vec![raw_dentry]; 1458004e86ffSlogin let start_offset: u64 = self.offset; 1459004e86ffSlogin let start_cluster: Cluster = self.current_cluster; 1460004e86ffSlogin 1461004e86ffSlogin self.offset += FATRawDirEntry::DIR_ENTRY_LEN; 1462004e86ffSlogin 1463004e86ffSlogin // 由于在FAT文件系统中,文件名最长为255字节,因此,最多有20个长目录项以及1个短目录项。 1464004e86ffSlogin // 由于上面已经塞了1个长目录项,因此接下来最多需要迭代20次 1465004e86ffSlogin // 循环查找目录项,直到遇到1个短目录项,或者是空闲目录项 1466004e86ffSlogin for _ in 0..20 { 1467004e86ffSlogin // 如果当前簇已经被读完,那么尝试获取下一个簇 1468004e86ffSlogin if self.offset >= self.fs.bytes_per_cluster() && !self.is_root { 1469004e86ffSlogin match self.fs.get_fat_entry(self.current_cluster)? { 1470004e86ffSlogin FATEntry::Next(c) => { 1471004e86ffSlogin // 获得下一个簇的信息 1472004e86ffSlogin self.current_cluster = c; 1473004e86ffSlogin self.offset %= self.fs.bytes_per_cluster(); 1474004e86ffSlogin } 1475004e86ffSlogin 1476004e86ffSlogin _ => { 1477004e86ffSlogin // 没有下一个簇了,退出迭代 1478004e86ffSlogin break; 1479004e86ffSlogin } 1480004e86ffSlogin } 1481004e86ffSlogin } 1482004e86ffSlogin // 如果当前是FAT12/FAT16文件系统,并且当前inode是根目录项。 1483004e86ffSlogin // 如果offset大于根目录项的最大大小(已经遍历完根目录),那么就退出迭代 1484004e86ffSlogin if self.is_root 1485004e86ffSlogin && self.offset > self.fs.root_dir_end_bytes_offset().unwrap() 1486004e86ffSlogin { 1487004e86ffSlogin break; 1488004e86ffSlogin } 1489004e86ffSlogin 1490004e86ffSlogin // 获取簇在磁盘内的字节偏移量 1491004e86ffSlogin let offset: u64 = 1492004e86ffSlogin self.fs.cluster_bytes_offset(self.current_cluster) + self.offset; 1493004e86ffSlogin // 从磁盘读取原始的dentry 1494004e86ffSlogin let raw_dentry: FATRawDirEntry = get_raw_dir_entry(&self.fs, offset)?; 1495004e86ffSlogin 1496004e86ffSlogin match raw_dentry { 1497004e86ffSlogin FATRawDirEntry::Short(_) => { 1498004e86ffSlogin // 当前遇到1个短目录项,证明当前文件/文件夹的所有dentry都被读取完了,因此在将其加入数组后,退出迭代。 1499004e86ffSlogin long_name_entries.push(raw_dentry); 1500004e86ffSlogin break; 1501004e86ffSlogin } 1502004e86ffSlogin FATRawDirEntry::Long(_) => { 1503004e86ffSlogin // 当前遇到1个长目录项,将其加入数组,然后更新offset,继续迭代。 1504004e86ffSlogin long_name_entries.push(raw_dentry); 1505004e86ffSlogin self.offset += FATRawDirEntry::DIR_ENTRY_LEN; 1506004e86ffSlogin } 1507004e86ffSlogin 1508004e86ffSlogin _ => { 1509004e86ffSlogin // 遇到了空闲簇,但没遇到短目录项,说明文件系统出错了,退出。 1510004e86ffSlogin break; 1511004e86ffSlogin } 1512004e86ffSlogin } 1513004e86ffSlogin } 1514004e86ffSlogin // kdebug!("collect dentries done. long_name_entries={long_name_entries:?}"); 1515676b8ef6SMork let dir_entry: Result<FATDirEntry, SystemError> = FATDirEntry::new( 1516004e86ffSlogin long_name_entries, 1517004e86ffSlogin ( 1518004e86ffSlogin (start_cluster, start_offset), 1519004e86ffSlogin (self.current_cluster, self.offset), 1520004e86ffSlogin ), 1521004e86ffSlogin ); 1522004e86ffSlogin // kdebug!("dir_entry={:?}", dir_entry); 1523004e86ffSlogin match dir_entry { 1524004e86ffSlogin Ok(d) => { 1525004e86ffSlogin // kdebug!("dir_entry ok"); 1526004e86ffSlogin self.offset += FATRawDirEntry::DIR_ENTRY_LEN; 1527004e86ffSlogin return Ok((self.current_cluster, self.offset, Some(d))); 1528004e86ffSlogin } 1529004e86ffSlogin 1530004e86ffSlogin Err(_) => { 1531004e86ffSlogin // kdebug!("dir_entry err, e={}", e); 1532004e86ffSlogin self.offset += FATRawDirEntry::DIR_ENTRY_LEN; 1533004e86ffSlogin } 1534004e86ffSlogin } 1535004e86ffSlogin } 1536004e86ffSlogin FATRawDirEntry::Free => { 1537004e86ffSlogin // 当前目录项是空的 1538004e86ffSlogin self.offset += FATRawDirEntry::DIR_ENTRY_LEN; 1539004e86ffSlogin } 1540004e86ffSlogin FATRawDirEntry::FreeRest => { 1541004e86ffSlogin // 当前目录项是空的,且之后都是空的,因此直接返回 1542004e86ffSlogin return Ok((self.current_cluster, self.offset, None)); 1543004e86ffSlogin } 1544004e86ffSlogin } 1545004e86ffSlogin } 1546004e86ffSlogin } 1547004e86ffSlogin } 1548004e86ffSlogin 1549004e86ffSlogin /// 为DirIter实现迭代器trait 1550004e86ffSlogin impl Iterator for FATDirIter { 1551004e86ffSlogin type Item = FATDirEntry; 1552004e86ffSlogin 1553004e86ffSlogin fn next(&mut self) -> Option<Self::Item> { 1554004e86ffSlogin match self.get_dir_entry() { 1555004e86ffSlogin Ok((cluster, offset, result)) => { 1556004e86ffSlogin self.current_cluster = cluster; 1557004e86ffSlogin self.offset = offset; 1558004e86ffSlogin return result; 1559004e86ffSlogin } 1560004e86ffSlogin Err(_) => { 1561004e86ffSlogin return None; 1562004e86ffSlogin } 1563004e86ffSlogin } 1564004e86ffSlogin } 1565004e86ffSlogin } 1566004e86ffSlogin 1567004e86ffSlogin impl FATDirEntry { 1568004e86ffSlogin /// @brief 构建FATDirEntry枚举类型 1569004e86ffSlogin /// 1570004e86ffSlogin /// @param long_name_entries 长目录项的数组。 1571004e86ffSlogin /// 格式:[第20个(或者是最大ord的那个), 19, 18, ..., 1, 短目录项] 1572004e86ffSlogin /// 1573004e86ffSlogin /// @return Ok(FATDirEntry) 构建好的FATDirEntry类型的对象 1574676b8ef6SMork /// @return Err(SystemError) 错误码 1575004e86ffSlogin pub fn new( 1576004e86ffSlogin mut long_name_entries: Vec<FATRawDirEntry>, 1577004e86ffSlogin loc: ((Cluster, u64), (Cluster, u64)), 1578676b8ef6SMork ) -> Result<Self, SystemError> { 1579004e86ffSlogin if long_name_entries.is_empty() { 1580676b8ef6SMork return Err(SystemError::EINVAL); 1581004e86ffSlogin } 1582004e86ffSlogin 1583004e86ffSlogin if !long_name_entries[0].is_last() || !long_name_entries.last().unwrap().is_short() { 1584004e86ffSlogin // 存在孤立的目录项,文件系统出现异常,因此返回错误,表明其只读。 1585004e86ffSlogin // TODO: 标记整个FAT文件系统为只读的 1586676b8ef6SMork return Err(SystemError::EROFS); 1587004e86ffSlogin } 1588004e86ffSlogin 1589004e86ffSlogin // 取出短目录项(位于vec的末尾) 1590004e86ffSlogin let short_dentry: ShortDirEntry = match long_name_entries.pop().unwrap() { 1591004e86ffSlogin FATRawDirEntry::Short(s) => s, 1592004e86ffSlogin _ => unreachable!(), 1593004e86ffSlogin }; 1594004e86ffSlogin 1595004e86ffSlogin let mut extractor = LongNameExtractor::new(); 1596004e86ffSlogin for entry in &long_name_entries { 1597004e86ffSlogin match entry { 1598004e86ffSlogin &FATRawDirEntry::Long(l) => { 1599004e86ffSlogin extractor.process(l)?; 1600004e86ffSlogin } 1601004e86ffSlogin 1602004e86ffSlogin _ => { 1603676b8ef6SMork return Err(SystemError::EROFS); 1604004e86ffSlogin } 1605004e86ffSlogin } 1606004e86ffSlogin } 1607004e86ffSlogin // 检验校验和是否正确 1608004e86ffSlogin if extractor.validate_checksum(&short_dentry) { 1609004e86ffSlogin // 校验和正确,返回一个长目录项 1610004e86ffSlogin return Ok(short_dentry.to_dir_entry_with_long_name(extractor.to_string(), loc)); 1611004e86ffSlogin } else { 1612004e86ffSlogin // 校验和不相同,认为文件系统出错 1613676b8ef6SMork return Err(SystemError::EROFS); 1614004e86ffSlogin } 1615004e86ffSlogin } 1616004e86ffSlogin 1617004e86ffSlogin /// @brief 获取短目录项的名字 1618004e86ffSlogin pub fn short_name(&self) -> String { 1619004e86ffSlogin match self { 1620004e86ffSlogin FATDirEntry::File(f) | FATDirEntry::VolId(f) => { 1621004e86ffSlogin return f.short_dir_entry.name_to_string(); 1622004e86ffSlogin } 1623004e86ffSlogin FATDirEntry::Dir(d) => match d.short_dir_entry { 1624004e86ffSlogin Some(s) => { 1625004e86ffSlogin return s.name_to_string(); 1626004e86ffSlogin } 1627004e86ffSlogin None => { 1628004e86ffSlogin return String::from("/"); 1629004e86ffSlogin } 1630004e86ffSlogin }, 1631004e86ffSlogin FATDirEntry::UnInit => unreachable!("FATFS: FATDirEntry uninitialized."), 1632004e86ffSlogin } 1633004e86ffSlogin } 1634004e86ffSlogin 1635004e86ffSlogin /// @brief 获取短目录项结构体 1636004e86ffSlogin pub fn short_dir_entry(&self) -> Option<ShortDirEntry> { 1637004e86ffSlogin match &self { 1638004e86ffSlogin FATDirEntry::File(f) => { 1639004e86ffSlogin return Some(f.short_dir_entry); 1640004e86ffSlogin } 1641004e86ffSlogin FATDirEntry::Dir(d) => { 1642004e86ffSlogin return d.short_dir_entry; 1643004e86ffSlogin } 1644004e86ffSlogin FATDirEntry::VolId(s) => { 1645004e86ffSlogin return Some(s.short_dir_entry); 1646004e86ffSlogin } 1647004e86ffSlogin FATDirEntry::UnInit => unreachable!("FATFS: FATDirEntry uninitialized."), 1648004e86ffSlogin } 1649004e86ffSlogin } 1650004e86ffSlogin 1651004e86ffSlogin /// @brief 获取目录项的第一个簇的簇号 1652004e86ffSlogin pub fn first_cluster(&self) -> Cluster { 1653004e86ffSlogin match self { 1654004e86ffSlogin FATDirEntry::File(f) => { 1655004e86ffSlogin return f.first_cluster; 1656004e86ffSlogin } 1657004e86ffSlogin FATDirEntry::Dir(d) => { 1658004e86ffSlogin return d.first_cluster; 1659004e86ffSlogin } 1660004e86ffSlogin FATDirEntry::VolId(s) => { 1661004e86ffSlogin return s.first_cluster; 1662004e86ffSlogin } 1663004e86ffSlogin FATDirEntry::UnInit => unreachable!("FATFS: FATDirEntry uninitialized."), 1664004e86ffSlogin } 1665004e86ffSlogin } 1666004e86ffSlogin 1667004e86ffSlogin /// @brief 获取当前目录项所占用的簇的范围 1668004e86ffSlogin /// 1669004e86ffSlogin /// @return (起始簇,簇内偏移量), (终止簇,簇内偏移量) 1670004e86ffSlogin pub fn get_dir_range(&self) -> Option<((Cluster, u64), (Cluster, u64))> { 1671004e86ffSlogin match self { 1672004e86ffSlogin FATDirEntry::File(f) => Some(f.loc), 1673004e86ffSlogin FATDirEntry::Dir(d) => d.loc, 1674004e86ffSlogin FATDirEntry::VolId(s) => Some(s.loc), 1675004e86ffSlogin FATDirEntry::UnInit => unreachable!("FATFS: FATDirEntry uninitialized."), 1676004e86ffSlogin } 1677004e86ffSlogin } 1678004e86ffSlogin 1679004e86ffSlogin /// @brief 获取原始的短目录项名(FAT标准规定的) 1680004e86ffSlogin pub fn short_name_raw(&self) -> [u8; 11] { 1681004e86ffSlogin match self { 1682004e86ffSlogin FATDirEntry::File(f) => { 1683004e86ffSlogin return f.short_dir_entry.name; 1684004e86ffSlogin } 1685004e86ffSlogin FATDirEntry::Dir(d) => match d.short_dir_entry { 1686004e86ffSlogin // 存在短目录项,直接返回 1687004e86ffSlogin Some(s) => { 1688004e86ffSlogin return s.name; 1689004e86ffSlogin } 1690004e86ffSlogin // 是根目录项 1691004e86ffSlogin None => { 1692004e86ffSlogin let mut s = [0x20u8; 11]; 1693004e86ffSlogin s[0] = '/' as u8; 1694004e86ffSlogin return s; 1695004e86ffSlogin } 1696004e86ffSlogin }, 1697004e86ffSlogin FATDirEntry::VolId(s) => { 1698004e86ffSlogin return s.short_dir_entry.name; 1699004e86ffSlogin } 1700004e86ffSlogin 1701004e86ffSlogin FATDirEntry::UnInit => unreachable!("FATFS: FATDirEntry uninitialized."), 1702004e86ffSlogin } 1703004e86ffSlogin } 1704004e86ffSlogin 1705004e86ffSlogin /// @brief 获取目录项的名字 1706004e86ffSlogin pub fn name(&self) -> String { 1707004e86ffSlogin match self { 1708004e86ffSlogin FATDirEntry::File(f) => { 1709004e86ffSlogin return f.file_name.clone(); 1710004e86ffSlogin } 1711004e86ffSlogin FATDirEntry::VolId(s) => { 1712004e86ffSlogin return s.file_name.clone(); 1713004e86ffSlogin } 1714004e86ffSlogin FATDirEntry::Dir(d) => { 1715004e86ffSlogin return d.dir_name.clone(); 1716004e86ffSlogin } 1717004e86ffSlogin FATDirEntry::UnInit => unreachable!("FATFS: FATDirEntry uninitialized."), 1718004e86ffSlogin } 1719004e86ffSlogin } 1720004e86ffSlogin 1721004e86ffSlogin /// @brief 判断目录项是否为文件 1722004e86ffSlogin pub fn is_file(&self) -> bool { 1723004e86ffSlogin match self { 1724004e86ffSlogin &FATDirEntry::File(_) | &FATDirEntry::VolId(_) => true, 1725004e86ffSlogin _ => false, 1726004e86ffSlogin } 1727004e86ffSlogin } 1728004e86ffSlogin 1729004e86ffSlogin /// @brief 判断目录项是否为文件夹 1730004e86ffSlogin pub fn is_dir(&self) -> bool { 1731004e86ffSlogin match &self { 1732004e86ffSlogin &FATDirEntry::Dir(_) => true, 1733004e86ffSlogin _ => false, 1734004e86ffSlogin } 1735004e86ffSlogin } 1736004e86ffSlogin 1737004e86ffSlogin /// @brief 判断目录项是否为Volume id 1738004e86ffSlogin pub fn is_vol_id(&self) -> bool { 1739004e86ffSlogin match self { 1740004e86ffSlogin &FATDirEntry::VolId(_) => true, 1741004e86ffSlogin _ => false, 1742004e86ffSlogin } 1743004e86ffSlogin } 1744004e86ffSlogin 1745004e86ffSlogin /// @brief 判断FAT目录项的名字与给定的是否相等 1746004e86ffSlogin /// 1747004e86ffSlogin /// 由于FAT32对大小写不敏感,因此将字符都转为大写,然后比较 1748004e86ffSlogin /// 1749004e86ffSlogin /// @return bool 相等 => true 1750004e86ffSlogin /// 不相等 => false 1751004e86ffSlogin pub fn eq_name(&self, name: &str) -> bool { 1752004e86ffSlogin // 由于FAT32对大小写不敏感,因此将字符都转为大写,然后比较。 1753004e86ffSlogin let binding = self.short_name(); 1754004e86ffSlogin let short_name = binding.chars().flat_map(|c| c.to_uppercase()); 1755004e86ffSlogin let binding = self.name(); 1756004e86ffSlogin let long_name = binding.chars().flat_map(|c| c.to_uppercase()); 1757004e86ffSlogin let name = name.chars().flat_map(|c| c.to_uppercase()); 1758004e86ffSlogin 1759004e86ffSlogin let long_name_matches: bool = long_name.eq(name.clone()); 1760004e86ffSlogin let short_name_matches: bool = short_name.eq(name); 1761004e86ffSlogin 1762004e86ffSlogin return long_name_matches || short_name_matches; 1763004e86ffSlogin } 1764004e86ffSlogin 1765004e86ffSlogin /// @brief 将FATDirEntry转换为FATFile对象 1766676b8ef6SMork pub fn to_file(&self) -> Result<FATFile, SystemError> { 1767004e86ffSlogin if self.is_file() == false { 1768676b8ef6SMork return Err(SystemError::EISDIR); 1769004e86ffSlogin } 1770004e86ffSlogin 1771004e86ffSlogin match &self { 1772004e86ffSlogin FATDirEntry::File(f) | FATDirEntry::VolId(f) => { 1773004e86ffSlogin return Ok(f.clone()); 1774004e86ffSlogin } 1775004e86ffSlogin _ => unreachable!(), 1776004e86ffSlogin } 1777004e86ffSlogin } 1778004e86ffSlogin 1779004e86ffSlogin /// @brief 将FATDirEntry转换为FATDir对象 1780676b8ef6SMork pub fn to_dir(&self) -> Result<FATDir, SystemError> { 1781004e86ffSlogin if self.is_dir() == false { 1782676b8ef6SMork return Err(SystemError::ENOTDIR); 1783004e86ffSlogin } 1784004e86ffSlogin match &self { 1785004e86ffSlogin FATDirEntry::Dir(d) => { 1786004e86ffSlogin return Ok(d.clone()); 1787004e86ffSlogin } 1788004e86ffSlogin _ => unreachable!(), 1789004e86ffSlogin } 1790004e86ffSlogin } 1791004e86ffSlogin } 1792004e86ffSlogin 1793004e86ffSlogin /// 用于生成短目录项文件名的生成器。 1794004e86ffSlogin #[derive(Debug, Default)] 1795004e86ffSlogin pub struct ShortNameGenerator { 1796004e86ffSlogin /// 短目录项的名字 1797004e86ffSlogin name: [u8; 11], 1798004e86ffSlogin /// 生成器的标志位(使用impl里面的mask来解析) 1799004e86ffSlogin flags: u8, 1800004e86ffSlogin /// 基础名的长度 1801004e86ffSlogin basename_len: u8, 1802004e86ffSlogin /// 对于文件名形如(TE021F~1.TXT)的,短前缀+校验码的短目录项,该字段表示基础名末尾数字的对应位。 1803004e86ffSlogin checksum_bitmask: u16, 1804004e86ffSlogin /// Fletcher-16 Checksum(与填写到ShortDirEntry里面的不一样) 1805004e86ffSlogin checksum: u16, 1806004e86ffSlogin /// 对于形如(TEXTFI~1.TXT)的短目录项名称,其中的数字的bitmask(第0位置位则表示这个数字是0) 1807004e86ffSlogin suffix_bitmask: u16, 1808004e86ffSlogin } 1809004e86ffSlogin 1810004e86ffSlogin impl ShortNameGenerator { 1811004e86ffSlogin /// 短目录项的名称的长度 1812004e86ffSlogin const SHORT_NAME_LEN: usize = 8; 1813004e86ffSlogin 1814004e86ffSlogin // ===== flags标志位的含义 ===== 1815004e86ffSlogin const IS_LOSSY: u8 = (1 << 0); 1816004e86ffSlogin const IS_EXACT_MATCH: u8 = (1 << 1); 1817004e86ffSlogin const IS_DOT: u8 = (1 << 2); 1818004e86ffSlogin const IS_DOTDOT: u8 = (1 << 3); 1819004e86ffSlogin /// 名称被完全拷贝 1820004e86ffSlogin const NAME_FITS: u8 = (1 << 4); 1821004e86ffSlogin 1822004e86ffSlogin /// @brief 初始化一个短目录项名称生成器 1823004e86ffSlogin pub fn new(mut name: &str) -> Self { 1824004e86ffSlogin name = name.trim(); 1825004e86ffSlogin 1826004e86ffSlogin let mut short_name: [u8; 11] = [0x20u8; 11]; 1827004e86ffSlogin if name == "." { 1828004e86ffSlogin short_name[0] = '.' as u8; 1829004e86ffSlogin } 1830004e86ffSlogin 1831004e86ffSlogin if name == ".." { 1832004e86ffSlogin short_name[0] = '.' as u8; 1833004e86ffSlogin short_name[1] = '.' as u8; 1834004e86ffSlogin } 1835004e86ffSlogin 1836004e86ffSlogin // @name_fits: 名称是否被完全拷贝 1837004e86ffSlogin // @basename_len: 基础名的长度 1838004e86ffSlogin // @is_lossy: 是否存在不合法的字符 1839004e86ffSlogin let (name_fits, basename_len, is_lossy) = match name.rfind('.') { 1840004e86ffSlogin Some(index) => { 1841004e86ffSlogin // 文件名里面有".", 且index为最右边的点号所在的下标(bytes index) 1842004e86ffSlogin // 拷贝基础名 1843004e86ffSlogin let (b_len, fits, b_lossy) = 1844004e86ffSlogin Self::copy_part(&mut short_name[..Self::SHORT_NAME_LEN], &name[..index]); 1845004e86ffSlogin 1846004e86ffSlogin // 拷贝扩展名 1847004e86ffSlogin let (_, ext_fits, ext_lossy) = Self::copy_part( 1848004e86ffSlogin &mut short_name[Self::SHORT_NAME_LEN..Self::SHORT_NAME_LEN + 3], 1849004e86ffSlogin &name[index + 1..], 1850004e86ffSlogin ); 1851004e86ffSlogin 1852004e86ffSlogin (fits && ext_fits, b_len, b_lossy || ext_lossy) 1853004e86ffSlogin } 1854004e86ffSlogin None => { 1855004e86ffSlogin // 文件名中,不存在"." 1856004e86ffSlogin let (b_len, fits, b_lossy) = 1857004e86ffSlogin Self::copy_part(&mut short_name[..Self::SHORT_NAME_LEN], &name); 1858004e86ffSlogin (fits, b_len, b_lossy) 1859004e86ffSlogin } 1860004e86ffSlogin }; 1861004e86ffSlogin 1862004e86ffSlogin let mut flags: u8 = 0; 1863004e86ffSlogin // 设置flags 1864004e86ffSlogin if is_lossy { 1865004e86ffSlogin flags |= Self::IS_LOSSY; 1866004e86ffSlogin } 1867004e86ffSlogin if name == "." { 1868004e86ffSlogin flags |= Self::IS_DOT; 1869004e86ffSlogin } 1870004e86ffSlogin if name == ".." { 1871004e86ffSlogin flags |= Self::IS_DOTDOT; 1872004e86ffSlogin } 1873004e86ffSlogin 1874004e86ffSlogin if name_fits { 1875004e86ffSlogin flags |= Self::NAME_FITS; 1876004e86ffSlogin } 1877004e86ffSlogin 1878004e86ffSlogin return ShortNameGenerator { 1879004e86ffSlogin name: short_name, 1880004e86ffSlogin flags: flags, 1881004e86ffSlogin basename_len: basename_len, 1882004e86ffSlogin checksum: Self::fletcher_16_checksum(name), 1883004e86ffSlogin ..Default::default() 1884004e86ffSlogin }; 1885004e86ffSlogin } 1886004e86ffSlogin 1887004e86ffSlogin /// @brief 拷贝字符串到一个u8数组 1888004e86ffSlogin /// 1889004e86ffSlogin /// @return (u8, bool, bool) 1890004e86ffSlogin /// return.0: 拷贝了的字符串的长度 1891004e86ffSlogin /// return.1: 是否完全拷贝完整个字符串 1892004e86ffSlogin /// return.2: 拷贝过程中,是否出现了不合法字符 1893004e86ffSlogin fn copy_part(dest: &mut [u8], src: &str) -> (u8, bool, bool) { 1894004e86ffSlogin let mut dest_len: usize = 0; 1895004e86ffSlogin let mut lossy_conv = false; 1896004e86ffSlogin 1897004e86ffSlogin for c in src.chars() { 1898004e86ffSlogin // 如果src还有字符,而dest已经满了,那么表示没有完全拷贝完。 1899004e86ffSlogin if dest_len == dest.len() { 1900004e86ffSlogin return (dest_len as u8, false, lossy_conv); 1901004e86ffSlogin } 1902004e86ffSlogin 1903004e86ffSlogin if c == ' ' || c == '.' { 1904004e86ffSlogin lossy_conv = true; 1905004e86ffSlogin continue; 1906004e86ffSlogin } 1907004e86ffSlogin 1908004e86ffSlogin let cp: char = match c { 1909004e86ffSlogin 'a'..='z' | 'A'..='Z' | '0'..='9' => c, 1910004e86ffSlogin '$' | '%' | '\'' | '-' | '_' | '@' | '~' | '`' | '!' | '(' | ')' | '{' | '}' 1911004e86ffSlogin | '^' | '#' | '&' => c, 1912004e86ffSlogin _ => '_', 1913004e86ffSlogin }; 1914004e86ffSlogin 1915004e86ffSlogin // 判断是否存在不符合条件的字符 1916004e86ffSlogin lossy_conv = lossy_conv || c != cp; 1917004e86ffSlogin 1918004e86ffSlogin // 拷贝字符 1919004e86ffSlogin dest[dest_len] = c.to_ascii_uppercase() as u8; 1920004e86ffSlogin dest_len += 1; 1921004e86ffSlogin } 1922004e86ffSlogin 1923004e86ffSlogin // 返回结果 1924004e86ffSlogin return (dest_len as u8, true, lossy_conv); 1925004e86ffSlogin } 1926004e86ffSlogin 1927004e86ffSlogin fn fletcher_16_checksum(name: &str) -> u16 { 1928004e86ffSlogin let mut sum1: u16 = 0; 1929004e86ffSlogin let mut sum2: u16 = 0; 1930004e86ffSlogin for c in name.chars() { 1931004e86ffSlogin sum1 = (sum1 + (c as u16)) % 0xff; 1932004e86ffSlogin sum2 = (sum1 + sum2) & 0xff; 1933004e86ffSlogin } 1934004e86ffSlogin return (sum2 << 8) | sum1; 1935004e86ffSlogin } 1936004e86ffSlogin 1937004e86ffSlogin /// @brief 更新生成器的状态 1938004e86ffSlogin /// 当长目录项不存在的时候,需要调用这个函数来更新生成器的状态 1939004e86ffSlogin pub fn add_name(&mut self, name: &[u8; 11]) { 1940004e86ffSlogin // === 判断名称是否严格的完全匹配 1941004e86ffSlogin if name == &self.name { 1942004e86ffSlogin self.flags |= Self::IS_EXACT_MATCH; 1943004e86ffSlogin } 1944004e86ffSlogin 1945004e86ffSlogin // === 检查是否存在长前缀的格式冲突。对于这样的短目录项名称:(TEXTFI~1.TXT) 1946004e86ffSlogin // 获取名称前缀 1947004e86ffSlogin let prefix_len = min(self.basename_len, 6) as usize; 1948004e86ffSlogin // 获取后缀的那个数字 1949004e86ffSlogin let num_suffix: Option<u32> = if name[prefix_len] as char == '~' { 1950004e86ffSlogin (name[prefix_len + 1] as char).to_digit(10) 1951004e86ffSlogin } else { 1952004e86ffSlogin None 1953004e86ffSlogin }; 1954004e86ffSlogin 1955004e86ffSlogin // 判断扩展名是否匹配 1956004e86ffSlogin let ext_matches: bool = name[8..] == self.name[8..]; 1957004e86ffSlogin 1958004e86ffSlogin if name[..prefix_len] == self.name[..prefix_len] // 基础名前缀相同 1959004e86ffSlogin && num_suffix.is_some() // 基础名具有数字后缀 1960004e86ffSlogin && ext_matches 1961004e86ffSlogin // 扩展名相匹配 1962004e86ffSlogin { 1963004e86ffSlogin let num = num_suffix.unwrap(); 1964004e86ffSlogin self.suffix_bitmask |= 1 << num; 1965004e86ffSlogin } 1966004e86ffSlogin 1967004e86ffSlogin // === 检查是否存在短前缀+校验和的冲突,文件名形如:(TE021F~1.TXT) 1968004e86ffSlogin let prefix_len = min(self.basename_len, 2) as usize; 1969004e86ffSlogin let num_suffix: Option<u32> = if name[prefix_len + 4] as char == '~' { 1970004e86ffSlogin (name[prefix_len + 1] as char).to_digit(10) 1971004e86ffSlogin } else { 1972004e86ffSlogin None 1973004e86ffSlogin }; 1974004e86ffSlogin 1975004e86ffSlogin if name[..prefix_len] == self.name[..prefix_len] && num_suffix.is_some() && ext_matches { 1976004e86ffSlogin // 获取短文件名中的校验码字段 1977004e86ffSlogin let checksum_result: Result< 1978004e86ffSlogin Result<u16, core::num::ParseIntError>, 1979004e86ffSlogin core::str::Utf8Error, 1980004e86ffSlogin > = core::str::from_utf8(&name[prefix_len..prefix_len + 4]) 1981004e86ffSlogin .map(|s| u16::from_str_radix(s, 16)); 1982004e86ffSlogin // 如果校验码相同 1983004e86ffSlogin if checksum_result == Ok(Ok(self.checksum)) { 1984004e86ffSlogin let num = num_suffix.unwrap(); 1985004e86ffSlogin // 置位checksum_bitmask中,基础名末尾数字的对应位 1986004e86ffSlogin self.checksum_bitmask |= 1 << num; 1987004e86ffSlogin } 1988004e86ffSlogin } 1989004e86ffSlogin } 1990004e86ffSlogin 1991676b8ef6SMork pub fn generate(&self) -> Result<[u8; 11], SystemError> { 1992004e86ffSlogin if self.is_dot() || self.is_dotdot() { 1993004e86ffSlogin return Ok(self.name); 1994004e86ffSlogin } 1995004e86ffSlogin 1996004e86ffSlogin // 如果当前名字不存在不合法的字符,且名称被完整拷贝,但是exact match为false,可以认为名称没有冲突,直接返回 1997004e86ffSlogin if !self.is_lossy() && self.name_fits() && !self.is_exact_match() { 1998004e86ffSlogin return Ok(self.name); 1999004e86ffSlogin } 2000004e86ffSlogin 2001004e86ffSlogin // 尝试使用长前缀(6字符) 2002004e86ffSlogin for i in 1..5 { 2003004e86ffSlogin if self.suffix_bitmask & (1 << i) == 0 { 2004004e86ffSlogin return Ok(self.build_prefixed_name(i as u32, false)); 2005004e86ffSlogin } 2006004e86ffSlogin } 2007004e86ffSlogin 2008004e86ffSlogin // 尝试使用短前缀+校验码 2009004e86ffSlogin for i in 1..10 { 2010004e86ffSlogin if self.checksum_bitmask & (1 << i) == 0 { 2011004e86ffSlogin return Ok(self.build_prefixed_name(i as u32, true)); 2012004e86ffSlogin } 2013004e86ffSlogin } 2014004e86ffSlogin // 由于产生太多的冲突,因此返回错误(“短文件名已经存在”) 2015676b8ef6SMork return Err(SystemError::EEXIST); 2016004e86ffSlogin } 2017004e86ffSlogin 2018004e86ffSlogin pub fn next_iteration(&mut self) { 2019004e86ffSlogin // 在下一次迭代中,尝试一个不同的校验和 2020004e86ffSlogin self.checksum = (core::num::Wrapping(self.checksum) + core::num::Wrapping(1)).0; 2021004e86ffSlogin // 清空bitmask 2022004e86ffSlogin self.suffix_bitmask = 0; 2023004e86ffSlogin self.checksum_bitmask = 0; 2024004e86ffSlogin } 2025004e86ffSlogin 2026004e86ffSlogin /// @brief 构造具有前缀的短目录项名称 2027004e86ffSlogin /// 2028004e86ffSlogin /// @param num 这是第几个重名的前缀名 2029004e86ffSlogin /// @param with_checksum 前缀名中是否包含校验码 2030004e86ffSlogin /// 2031004e86ffSlogin /// @return 构造好的短目录项名称数组 2032004e86ffSlogin fn build_prefixed_name(&self, num: u32, with_checksum: bool) -> [u8; 11] { 2033004e86ffSlogin let mut buf: [u8; 11] = [0x20u8; 11]; 2034004e86ffSlogin let prefix_len: usize = if with_checksum { 2035004e86ffSlogin let prefix_len: usize = min(self.basename_len as usize, 2); 2036004e86ffSlogin buf[..prefix_len].copy_from_slice(&self.name[..prefix_len]); 2037004e86ffSlogin buf[prefix_len..prefix_len + 4].copy_from_slice(&Self::u16_to_u8_array(self.checksum)); 2038004e86ffSlogin prefix_len + 4 2039004e86ffSlogin } else { 2040004e86ffSlogin let prefix_len = min(self.basename_len as usize, 6); 2041004e86ffSlogin buf[..prefix_len].copy_from_slice(&self.name[..prefix_len]); 2042004e86ffSlogin prefix_len 2043004e86ffSlogin }; 2044004e86ffSlogin 2045004e86ffSlogin buf[prefix_len] = '~' as u8; 2046004e86ffSlogin buf[prefix_len + 1] = char::from_digit(num, 10).unwrap() as u8; 2047004e86ffSlogin buf[8..].copy_from_slice(&self.name[8..]); 2048004e86ffSlogin return buf; 2049004e86ffSlogin } 2050004e86ffSlogin 2051004e86ffSlogin /// @brief 将一个u16数字转换为十六进制大写字符串对应的ascii数组。 2052004e86ffSlogin /// 举例:将x=12345转换为16进制字符串“3039”对应的ascii码数组:[51,48,51,57] 2053004e86ffSlogin fn u16_to_u8_array(x: u16) -> [u8; 4] { 2054004e86ffSlogin let c1 = char::from_digit((x as u32 >> 12) & 0xf, 16) 2055004e86ffSlogin .unwrap() 2056004e86ffSlogin .to_ascii_uppercase() as u8; 2057004e86ffSlogin let c2 = char::from_digit((x as u32 >> 8) & 0xf, 16) 2058004e86ffSlogin .unwrap() 2059004e86ffSlogin .to_ascii_uppercase() as u8; 2060004e86ffSlogin let c3 = char::from_digit((x as u32 >> 4) & 0xf, 16) 2061004e86ffSlogin .unwrap() 2062004e86ffSlogin .to_ascii_uppercase() as u8; 2063004e86ffSlogin let c4 = char::from_digit((x as u32 >> 0) & 0xf, 16) 2064004e86ffSlogin .unwrap() 2065004e86ffSlogin .to_ascii_uppercase() as u8; 2066004e86ffSlogin return [c1, c2, c3, c4]; 2067004e86ffSlogin } 2068004e86ffSlogin 2069004e86ffSlogin #[inline] 2070004e86ffSlogin fn is_lossy(&self) -> bool { 2071004e86ffSlogin return (self.flags & Self::IS_LOSSY) > 0; 2072004e86ffSlogin } 2073004e86ffSlogin 2074004e86ffSlogin #[inline] 2075004e86ffSlogin fn is_exact_match(&self) -> bool { 2076004e86ffSlogin return (self.flags & Self::IS_EXACT_MATCH) > 0; 2077004e86ffSlogin } 2078004e86ffSlogin 2079004e86ffSlogin #[inline] 2080004e86ffSlogin fn is_dot(&self) -> bool { 2081004e86ffSlogin return (self.flags & Self::IS_DOT) > 0; 2082004e86ffSlogin } 2083004e86ffSlogin 2084004e86ffSlogin #[inline] 2085004e86ffSlogin fn is_dotdot(&self) -> bool { 2086004e86ffSlogin return (self.flags & Self::IS_DOTDOT) > 0; 2087004e86ffSlogin } 2088004e86ffSlogin 2089004e86ffSlogin #[inline] 2090004e86ffSlogin fn name_fits(&self) -> bool { 2091004e86ffSlogin return (self.flags & Self::NAME_FITS) > 0; 2092004e86ffSlogin } 2093004e86ffSlogin } 2094004e86ffSlogin 2095004e86ffSlogin /// 从多个LongName中提取完整文件名字段的提取器 2096004e86ffSlogin struct LongNameExtractor { 2097004e86ffSlogin name: Vec<u16>, 2098004e86ffSlogin checksum: u8, 2099004e86ffSlogin index: u8, 2100004e86ffSlogin } 2101004e86ffSlogin 2102004e86ffSlogin impl LongNameExtractor { 2103004e86ffSlogin fn new() -> Self { 2104004e86ffSlogin return LongNameExtractor { 2105004e86ffSlogin name: Vec::new(), 2106004e86ffSlogin checksum: 0, 2107004e86ffSlogin index: 0, 2108004e86ffSlogin }; 2109004e86ffSlogin } 2110004e86ffSlogin 2111004e86ffSlogin /// @brief 提取长目录项的名称 2112004e86ffSlogin /// @param longname_dentry 长目录项 2113004e86ffSlogin /// 请注意,必须倒序输入长目录项对象 2114676b8ef6SMork fn process(&mut self, longname_dentry: LongDirEntry) -> Result<(), SystemError> { 2115004e86ffSlogin let is_last: bool = longname_dentry.is_last(); 2116004e86ffSlogin let index: u8 = longname_dentry.ord & 0x1f; 2117004e86ffSlogin 2118004e86ffSlogin if index == 0 { 2119004e86ffSlogin self.name.clear(); 2120676b8ef6SMork return Err(SystemError::EROFS); 2121004e86ffSlogin } 2122004e86ffSlogin 2123004e86ffSlogin // 如果是最后一个LongDirEntry,则初始化当前生成器 2124004e86ffSlogin if is_last { 2125004e86ffSlogin self.index = index; 2126004e86ffSlogin self.checksum = longname_dentry.checksum; 2127004e86ffSlogin self.name 2128004e86ffSlogin .resize(index as usize * LongDirEntry::LONG_NAME_STR_LEN, 0); 2129004e86ffSlogin } else if self.index == 0 2130004e86ffSlogin || index != self.index - 1 2131004e86ffSlogin || self.checksum != longname_dentry.checksum 2132004e86ffSlogin { 2133004e86ffSlogin // 如果当前index为0,或者index不连续,或者是校验和不同,那么认为文件系统损坏,清除生成器的名称字段 2134004e86ffSlogin // TODO: 对文件系统的变为只读状态状况的拦截 2135004e86ffSlogin self.name.clear(); 2136676b8ef6SMork return Err(SystemError::EROFS); 2137004e86ffSlogin } else { 2138004e86ffSlogin // 由于dentry倒序输入,因此index是每次减1的 2139004e86ffSlogin self.index -= 1; 2140004e86ffSlogin } 2141004e86ffSlogin 2142004e86ffSlogin let pos: usize = ((index - 1) as usize) * LongDirEntry::LONG_NAME_STR_LEN; 2143004e86ffSlogin // 将当前目录项的值,拷贝到生成器的数组中 2144004e86ffSlogin longname_dentry 2145004e86ffSlogin .copy_name_to_slice(&mut self.name[pos..pos + LongDirEntry::LONG_NAME_STR_LEN])?; 2146004e86ffSlogin return Ok(()); 2147004e86ffSlogin } 2148004e86ffSlogin 2149004e86ffSlogin /// @brief 返回名称的长度 2150004e86ffSlogin #[inline] 2151004e86ffSlogin fn len(&self) -> usize { 2152004e86ffSlogin return self.name.len(); 2153004e86ffSlogin } 2154004e86ffSlogin 2155004e86ffSlogin /// @brief 返回抽取得到的名称字符串 2156004e86ffSlogin fn to_string(&self) -> String { 2157004e86ffSlogin let mut s = String::from_utf16_lossy(self.name.as_slice()); 2158004e86ffSlogin // 计算字符串的长度。如果字符串中有\0,那么就截取字符串的前面部分 2159004e86ffSlogin if let Some(len) = s.find('\u{0}') { 2160004e86ffSlogin s.truncate(len); 2161004e86ffSlogin } 2162004e86ffSlogin return s; 2163004e86ffSlogin } 2164004e86ffSlogin 2165004e86ffSlogin /// @brief 判断校验码是否与指定的短目录项的校验码相同 2166004e86ffSlogin /// 2167004e86ffSlogin /// @return bool 相同 => true 2168004e86ffSlogin /// 不同 => false 2169004e86ffSlogin fn validate_checksum(&self, short_dentry: &ShortDirEntry) -> bool { 2170004e86ffSlogin return self.checksum == short_dentry.checksum(); 2171004e86ffSlogin } 2172004e86ffSlogin } 2173004e86ffSlogin 2174004e86ffSlogin /// @brief 长目录项生成器 2175004e86ffSlogin #[derive(Debug)] 2176004e86ffSlogin struct LongNameEntryGenerator { 2177004e86ffSlogin name: Vec<u16>, 2178004e86ffSlogin // 短目录项的校验和 2179004e86ffSlogin checksum: u8, 2180004e86ffSlogin // 当前迭代器的索引 2181004e86ffSlogin idx: u8, 2182004e86ffSlogin /// 最后一个目录项的索引 2183004e86ffSlogin last_index: u8, 2184004e86ffSlogin } 2185004e86ffSlogin 2186004e86ffSlogin impl LongNameEntryGenerator { 2187004e86ffSlogin /// @brief 初始化长目录项生成器 2188004e86ffSlogin /// 2189004e86ffSlogin /// @param name 长文件名数组 2190004e86ffSlogin /// @param checksum 短目录项的校验和 2191004e86ffSlogin pub fn new(name: &str, checksum: u8) -> Self { 2192004e86ffSlogin let mut name: Vec<u16> = name.chars().map(|c| c as u16).collect(); 2193004e86ffSlogin 2194004e86ffSlogin let padding_bytes: usize = (13 - (name.len() % 13)) % 13; 2195004e86ffSlogin // 填充最后一个长目录项的文件名 2196004e86ffSlogin for i in 0..padding_bytes { 2197004e86ffSlogin if i == 0 { 2198004e86ffSlogin name.push(0); 2199004e86ffSlogin } else { 2200004e86ffSlogin name.push(0xffff); 2201004e86ffSlogin } 2202004e86ffSlogin } 2203004e86ffSlogin 2204004e86ffSlogin // 先从最后一个长目录项开始生成 2205004e86ffSlogin let start_index = (name.len() / 13) as u8; 2206004e86ffSlogin return LongNameEntryGenerator { 2207004e86ffSlogin name: name, 2208004e86ffSlogin checksum: checksum, 2209004e86ffSlogin idx: start_index, 2210004e86ffSlogin last_index: start_index, 2211004e86ffSlogin }; 2212004e86ffSlogin } 2213004e86ffSlogin 2214004e86ffSlogin /// @brief 返回要生成的长目录项的总数 2215004e86ffSlogin pub fn num_entries(&self) -> u8 { 2216004e86ffSlogin return self.last_index + 1; 2217004e86ffSlogin } 2218004e86ffSlogin } 2219004e86ffSlogin 2220004e86ffSlogin impl Iterator for LongNameEntryGenerator { 2221004e86ffSlogin type Item = LongDirEntry; 2222004e86ffSlogin 2223004e86ffSlogin fn next(&mut self) -> Option<Self::Item> { 2224004e86ffSlogin match self.idx { 2225004e86ffSlogin 0 => { 2226004e86ffSlogin return None; 2227004e86ffSlogin } 2228004e86ffSlogin // 最后一个长目录项 2229004e86ffSlogin n if n == self.last_index => { 2230004e86ffSlogin // 最后一个长目录项的ord需要与0x40相或 2231004e86ffSlogin let ord: u8 = n | 0x40; 2232004e86ffSlogin let start_idx = ((n - 1) * 13) as usize; 2233004e86ffSlogin self.idx -= 1; 2234004e86ffSlogin return Some(LongDirEntry::new( 2235004e86ffSlogin ord, 2236004e86ffSlogin &self.name.as_slice()[start_idx..start_idx + 13], 2237004e86ffSlogin self.checksum, 2238004e86ffSlogin )); 2239004e86ffSlogin } 2240004e86ffSlogin n => { 2241004e86ffSlogin // 其它的长目录项 2242004e86ffSlogin let start_idx = ((n - 1) * 13) as usize; 2243004e86ffSlogin self.idx -= 1; 2244004e86ffSlogin return Some(LongDirEntry::new( 2245004e86ffSlogin n, 2246004e86ffSlogin &self.name.as_slice()[start_idx..start_idx + 13], 2247004e86ffSlogin self.checksum, 2248004e86ffSlogin )); 2249004e86ffSlogin } 2250004e86ffSlogin } 2251004e86ffSlogin } 2252004e86ffSlogin } 2253004e86ffSlogin 2254004e86ffSlogin #[derive(Debug)] 2255004e86ffSlogin pub enum FATDirEntryOrShortName { 2256004e86ffSlogin DirEntry(FATDirEntry), 2257004e86ffSlogin ShortName([u8; 11]), 2258004e86ffSlogin } 2259004e86ffSlogin 2260004e86ffSlogin /// @brief 对FAT目录项的迭代器(基于簇和簇内偏移量) 2261004e86ffSlogin #[derive(Debug)] 2262004e86ffSlogin struct FATDirEntryOffsetIter { 2263004e86ffSlogin /// 当前迭代的偏移量(下一次迭代要返回的值) 2264004e86ffSlogin current_offset: (Cluster, u64), 2265004e86ffSlogin /// 截止迭代的位置(end_offset所在的位置也会被迭代器返回) 2266004e86ffSlogin end_offset: Option<(Cluster, u64)>, 2267004e86ffSlogin /// 属于的文件系统 2268004e86ffSlogin fs: Arc<FATFileSystem>, 2269004e86ffSlogin /// 当前已经迭代了多少次 2270004e86ffSlogin index: u64, 2271004e86ffSlogin /// 总共要迭代多少次 2272004e86ffSlogin len: u64, 2273004e86ffSlogin /// 如果end_offset不为None,该字段表示“是否已经到达了迭代终点” 2274004e86ffSlogin fin: bool, 2275004e86ffSlogin } 2276004e86ffSlogin 2277004e86ffSlogin impl FATDirEntryOffsetIter { 2278004e86ffSlogin /// @brief 初始化FAT目录项的迭代器(基于簇和簇内偏移量) 2279004e86ffSlogin /// 2280004e86ffSlogin /// @param fs 属于的文件系统 2281004e86ffSlogin /// @param start 起始偏移量 2282004e86ffSlogin /// @param len 要迭代的次数 2283004e86ffSlogin /// @param end_offset 截止迭代的位置(end_offset所在的位置也会被迭代器返回) 2284004e86ffSlogin /// 2285004e86ffSlogin /// @return 构建好的迭代器对象 2286004e86ffSlogin pub fn new( 2287004e86ffSlogin fs: Arc<FATFileSystem>, 2288004e86ffSlogin start: (Cluster, u64), 2289004e86ffSlogin len: u64, 2290004e86ffSlogin end_offset: Option<(Cluster, u64)>, 2291004e86ffSlogin ) -> Self { 2292004e86ffSlogin return FATDirEntryOffsetIter { 2293004e86ffSlogin current_offset: start, 2294004e86ffSlogin end_offset, 2295004e86ffSlogin fs, 2296004e86ffSlogin index: 0, 2297004e86ffSlogin len, 2298004e86ffSlogin fin: false, 2299004e86ffSlogin }; 2300004e86ffSlogin } 2301004e86ffSlogin } 2302004e86ffSlogin 2303004e86ffSlogin impl Iterator for FATDirEntryOffsetIter { 2304004e86ffSlogin type Item = (Cluster, u64); 2305004e86ffSlogin 2306004e86ffSlogin fn next(&mut self) -> Option<Self::Item> { 2307004e86ffSlogin if self.index == self.len || self.fin { 2308004e86ffSlogin return None; 2309004e86ffSlogin } 2310004e86ffSlogin 2311004e86ffSlogin let r: (Cluster, u64) = self.current_offset; 2312004e86ffSlogin // 计算新的字节偏移量 2313004e86ffSlogin let mut new_offset = r.1 + FATRawDirEntry::DIR_ENTRY_LEN; 2314004e86ffSlogin let mut new_cluster: Cluster = r.0; 2315004e86ffSlogin // 越过了当前簇,则获取下一个簇 2316004e86ffSlogin if new_offset >= self.fs.bytes_per_cluster() { 2317004e86ffSlogin new_offset %= self.fs.bytes_per_cluster(); 2318004e86ffSlogin 2319004e86ffSlogin match self.fs.get_fat_entry(new_cluster) { 2320004e86ffSlogin Ok(FATEntry::Next(c)) => { 2321004e86ffSlogin new_cluster = c; 2322004e86ffSlogin } 2323004e86ffSlogin // 没有下一个簇了 2324004e86ffSlogin _ => { 2325004e86ffSlogin self.fin = true; 2326004e86ffSlogin } 2327004e86ffSlogin } 2328004e86ffSlogin } 2329004e86ffSlogin 2330004e86ffSlogin if let Some(off) = self.end_offset { 2331004e86ffSlogin // 判断当前簇是否是要求停止搜索的最后一个位置 2332004e86ffSlogin self.fin = off == self.current_offset; 2333004e86ffSlogin } 2334004e86ffSlogin // 更新当前迭代的偏移量 2335004e86ffSlogin self.current_offset = (new_cluster, new_offset); 2336004e86ffSlogin self.index += 1; 2337004e86ffSlogin 2338004e86ffSlogin return Some(r); 2339004e86ffSlogin } 2340004e86ffSlogin } 2341004e86ffSlogin 2342004e86ffSlogin /// @brief 根据磁盘内字节偏移量,读取磁盘,并生成一个FATRawDirEntry对象 2343004e86ffSlogin pub fn get_raw_dir_entry( 2344004e86ffSlogin fs: &Arc<FATFileSystem>, 2345004e86ffSlogin in_disk_bytes_offset: u64, 2346676b8ef6SMork ) -> Result<FATRawDirEntry, SystemError> { 2347004e86ffSlogin // 块内偏移量 2348004e86ffSlogin let blk_offset: u64 = fs.get_in_block_offset(in_disk_bytes_offset); 2349004e86ffSlogin let lba = fs.get_lba_from_offset( 2350004e86ffSlogin fs.bytes_to_sector(fs.get_in_partition_bytes_offset(in_disk_bytes_offset)), 2351004e86ffSlogin ); 2352004e86ffSlogin 2353004e86ffSlogin // let step1 = fs.get_in_partition_bytes_offset(in_disk_bytes_offset); 2354004e86ffSlogin // let step2 = fs.bytes_to_sector(step1); 2355004e86ffSlogin // let lba = fs.get_lba_from_offset(step2); 2356004e86ffSlogin // kdebug!("step1={step1}, step2={step2}, lba={lba}"); 2357004e86ffSlogin let mut v: Vec<u8> = Vec::new(); 2358004e86ffSlogin v.resize(1 * LBA_SIZE, 0); 2359004e86ffSlogin 2360004e86ffSlogin fs.partition.disk().read_at(lba, 1, &mut v)?; 2361004e86ffSlogin 2362004e86ffSlogin let mut cursor: VecCursor = VecCursor::new(v); 2363004e86ffSlogin // 切换游标到对应位置 2364004e86ffSlogin cursor.seek(SeekFrom::SeekSet(blk_offset as i64))?; 2365004e86ffSlogin 2366004e86ffSlogin let dir_0 = cursor.read_u8()?; 2367004e86ffSlogin 2368004e86ffSlogin match dir_0 { 2369004e86ffSlogin 0x00 => { 2370004e86ffSlogin return Ok(FATRawDirEntry::FreeRest); 2371004e86ffSlogin } 2372004e86ffSlogin 0xe5 => { 2373004e86ffSlogin return Ok(FATRawDirEntry::Free); 2374004e86ffSlogin } 2375004e86ffSlogin _ => { 2376004e86ffSlogin cursor.seek(SeekFrom::SeekCurrent(10))?; 2377004e86ffSlogin let file_attr: FileAttributes = FileAttributes::new(cursor.read_u8()?); 2378004e86ffSlogin 2379004e86ffSlogin // 指针回到目录项的开始处 2380004e86ffSlogin cursor.seek(SeekFrom::SeekSet(blk_offset as i64))?; 2381004e86ffSlogin 2382004e86ffSlogin if file_attr.contains(FileAttributes::LONG_NAME) { 2383004e86ffSlogin // 当前目录项是一个长目录项 2384004e86ffSlogin let mut long_dentry = LongDirEntry::default(); 2385004e86ffSlogin 2386004e86ffSlogin long_dentry.ord = cursor.read_u8()?; 2387004e86ffSlogin cursor.read_u16_into(&mut long_dentry.name1)?; 2388004e86ffSlogin long_dentry.file_attrs = FileAttributes::new(cursor.read_u8()?); 2389004e86ffSlogin long_dentry.dirent_type = cursor.read_u8()?; 2390004e86ffSlogin long_dentry.checksum = cursor.read_u8()?; 2391004e86ffSlogin 2392004e86ffSlogin cursor.read_u16_into(&mut long_dentry.name2)?; 2393004e86ffSlogin long_dentry.first_clus_low = cursor.read_u16()?; 2394004e86ffSlogin cursor.read_u16_into(&mut long_dentry.name3)?; 2395004e86ffSlogin 2396004e86ffSlogin return Ok(FATRawDirEntry::Long(long_dentry)); 2397004e86ffSlogin } else { 2398004e86ffSlogin // 当前目录项是一个短目录项 2399004e86ffSlogin let mut short_dentry = ShortDirEntry::default(); 2400004e86ffSlogin cursor.read_exact(&mut short_dentry.name)?; 2401004e86ffSlogin 2402004e86ffSlogin short_dentry.attributes = FileAttributes::new(cursor.read_u8()?); 2403004e86ffSlogin 2404004e86ffSlogin short_dentry.nt_res = cursor.read_u8()?; 2405004e86ffSlogin short_dentry.crt_time_tenth = cursor.read_u8()?; 2406004e86ffSlogin short_dentry.crt_time = cursor.read_u16()?; 2407004e86ffSlogin short_dentry.crt_date = cursor.read_u16()?; 2408004e86ffSlogin short_dentry.lst_acc_date = cursor.read_u16()?; 2409004e86ffSlogin short_dentry.fst_clus_hi = cursor.read_u16()?; 2410004e86ffSlogin short_dentry.wrt_time = cursor.read_u16()?; 2411004e86ffSlogin short_dentry.wrt_date = cursor.read_u16()?; 2412004e86ffSlogin short_dentry.fst_clus_lo = cursor.read_u16()?; 2413004e86ffSlogin short_dentry.file_size = cursor.read_u32()?; 2414004e86ffSlogin 2415004e86ffSlogin return Ok(FATRawDirEntry::Short(short_dentry)); 2416004e86ffSlogin } 2417004e86ffSlogin } 2418004e86ffSlogin } 2419004e86ffSlogin } 2420