forked from M-Labs/rust-fatfs
WIP dir_entry module?
This commit is contained in:
parent
ff93bc3b5f
commit
3dc524759f
613
src/dir.rs
613
src/dir.rs
@ -1,18 +1,12 @@
|
|||||||
use std::ascii::AsciiExt;
|
use std::ascii::AsciiExt;
|
||||||
use std::fmt;
|
|
||||||
use std::io::prelude::*;
|
use std::io::prelude::*;
|
||||||
use std::io;
|
use std::io;
|
||||||
use std::io::{Cursor, ErrorKind, SeekFrom};
|
use std::io::{ErrorKind, SeekFrom};
|
||||||
use std::cmp;
|
use std::cmp;
|
||||||
use byteorder::{LittleEndian, ReadBytesExt, WriteBytesExt};
|
|
||||||
|
|
||||||
#[cfg(feature = "chrono")]
|
use fs::{FileSystemRef, DiskSlice};
|
||||||
use chrono::{TimeZone, Local, Datelike, Timelike};
|
|
||||||
#[cfg(feature = "chrono")]
|
|
||||||
use chrono;
|
|
||||||
|
|
||||||
use fs::{FileSystemRef, DiskSlice, FatType};
|
|
||||||
use file::File;
|
use file::File;
|
||||||
|
use dir_entry::*;
|
||||||
|
|
||||||
#[derive(Clone)]
|
#[derive(Clone)]
|
||||||
pub(crate) enum DirRawStream<'a, 'b: 'a> {
|
pub(crate) enum DirRawStream<'a, 'b: 'a> {
|
||||||
@ -69,603 +63,6 @@ impl <'a, 'b> Seek for DirRawStream<'a, 'b> {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
bitflags! {
|
|
||||||
/// FAT file attributes
|
|
||||||
#[derive(Default)]
|
|
||||||
pub struct FileAttributes: u8 {
|
|
||||||
const READ_ONLY = 0x01;
|
|
||||||
const HIDDEN = 0x02;
|
|
||||||
const SYSTEM = 0x04;
|
|
||||||
const VOLUME_ID = 0x08;
|
|
||||||
const DIRECTORY = 0x10;
|
|
||||||
const ARCHIVE = 0x20;
|
|
||||||
const LFN = Self::READ_ONLY.bits | Self::HIDDEN.bits
|
|
||||||
| Self::SYSTEM.bits | Self::VOLUME_ID.bits;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
const LFN_PART_LEN: usize = 13;
|
|
||||||
const DIR_ENTRY_SIZE: u64 = 32;
|
|
||||||
const DIR_ENTRY_FREE_FLAG: u8 = 0xE5;
|
|
||||||
const LFN_ENTRY_LAST_FLAG: u8 = 0x40;
|
|
||||||
|
|
||||||
#[allow(dead_code)]
|
|
||||||
#[derive(Clone, Debug, Default)]
|
|
||||||
pub(crate) struct DirFileEntryData {
|
|
||||||
name: [u8; 11],
|
|
||||||
attrs: FileAttributes,
|
|
||||||
reserved_0: u8,
|
|
||||||
create_time_0: u8,
|
|
||||||
create_time_1: u16,
|
|
||||||
create_date: u16,
|
|
||||||
access_date: u16,
|
|
||||||
first_cluster_hi: u16,
|
|
||||||
modify_time: u16,
|
|
||||||
modify_date: u16,
|
|
||||||
first_cluster_lo: u16,
|
|
||||||
size: u32,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl DirFileEntryData {
|
|
||||||
fn new(name: [u8; 11], attrs: FileAttributes) -> Self {
|
|
||||||
DirFileEntryData {
|
|
||||||
name, attrs,
|
|
||||||
..Default::default()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub(crate) fn first_cluster(&self, fat_type: FatType) -> Option<u32> {
|
|
||||||
let first_cluster_hi = if fat_type == FatType::Fat32 { self.first_cluster_hi } else { 0 };
|
|
||||||
let n = ((first_cluster_hi as u32) << 16) | self.first_cluster_lo as u32;
|
|
||||||
if n == 0 { None } else { Some(n) }
|
|
||||||
}
|
|
||||||
|
|
||||||
fn set_first_cluster(&mut self, cluster: Option<u32>, fat_type: FatType) {
|
|
||||||
let n = cluster.unwrap_or(0);
|
|
||||||
if fat_type == FatType::Fat32 {
|
|
||||||
self.first_cluster_hi = (n >> 16) as u16;
|
|
||||||
}
|
|
||||||
self.first_cluster_lo = (n & 0xFFFF) as u16;
|
|
||||||
}
|
|
||||||
|
|
||||||
pub(crate) fn size(&self) -> Option<u32> {
|
|
||||||
if self.is_file() {
|
|
||||||
Some(self.size)
|
|
||||||
} else {
|
|
||||||
None
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn set_size(&mut self, size: u32) {
|
|
||||||
self.size = size;
|
|
||||||
}
|
|
||||||
|
|
||||||
pub(crate) fn is_dir(&self) -> bool {
|
|
||||||
self.attrs.contains(FileAttributes::DIRECTORY)
|
|
||||||
}
|
|
||||||
|
|
||||||
pub(crate) fn is_file(&self) -> bool {
|
|
||||||
!self.is_dir()
|
|
||||||
}
|
|
||||||
|
|
||||||
fn created(&self) -> DateTime {
|
|
||||||
DateTime::from_u16(self.create_date, self.create_time_1)
|
|
||||||
}
|
|
||||||
|
|
||||||
fn accessed(&self) -> Date {
|
|
||||||
Date::from_u16(self.access_date)
|
|
||||||
}
|
|
||||||
|
|
||||||
fn modified(&self) -> DateTime {
|
|
||||||
DateTime::from_u16(self.modify_date, self.modify_time)
|
|
||||||
}
|
|
||||||
|
|
||||||
fn set_created(&mut self, date_time: DateTime) {
|
|
||||||
self.create_date = date_time.date.to_u16();
|
|
||||||
self.create_time_1 = date_time.time.to_u16();
|
|
||||||
}
|
|
||||||
|
|
||||||
fn set_accessed(&mut self, date: Date) {
|
|
||||||
self.access_date = date.to_u16();
|
|
||||||
}
|
|
||||||
|
|
||||||
fn set_modified(&mut self, date_time: DateTime) {
|
|
||||||
self.modify_date = date_time.date.to_u16();
|
|
||||||
self.modify_time = date_time.time.to_u16();
|
|
||||||
}
|
|
||||||
|
|
||||||
#[cfg(feature = "chrono")]
|
|
||||||
fn reset_created(&mut self) {
|
|
||||||
let now = DateTime::from(chrono::Local::now());
|
|
||||||
self.set_created(now);
|
|
||||||
}
|
|
||||||
|
|
||||||
#[cfg(feature = "chrono")]
|
|
||||||
fn reset_accessed(&mut self) -> bool {
|
|
||||||
let now = Date::from(chrono::Local::now().date());
|
|
||||||
if now == self.accessed() {
|
|
||||||
false
|
|
||||||
} else {
|
|
||||||
self.set_accessed(now);
|
|
||||||
true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[cfg(feature = "chrono")]
|
|
||||||
fn reset_modified(&mut self) {
|
|
||||||
let now = DateTime::from(chrono::Local::now());
|
|
||||||
self.set_modified(now);
|
|
||||||
}
|
|
||||||
|
|
||||||
#[cfg(not(feature = "chrono"))]
|
|
||||||
fn reset_created(&mut self) {
|
|
||||||
// nop - user controls timestamps manually
|
|
||||||
false
|
|
||||||
}
|
|
||||||
|
|
||||||
#[cfg(not(feature = "chrono"))]
|
|
||||||
fn reset_accessed(&mut self) -> bool {
|
|
||||||
// nop - user controls timestamps manually
|
|
||||||
false
|
|
||||||
}
|
|
||||||
|
|
||||||
#[cfg(not(feature = "chrono"))]
|
|
||||||
fn reset_modified(&mut self) {
|
|
||||||
// nop - user controls timestamps manually
|
|
||||||
false
|
|
||||||
}
|
|
||||||
|
|
||||||
fn serialize(&self, wrt: &mut Write) -> io::Result<()> {
|
|
||||||
wrt.write_all(&self.name)?;
|
|
||||||
wrt.write_u8(self.attrs.bits())?;
|
|
||||||
wrt.write_u8(self.reserved_0)?;
|
|
||||||
wrt.write_u8(self.create_time_0)?;
|
|
||||||
wrt.write_u16::<LittleEndian>(self.create_time_1)?;
|
|
||||||
wrt.write_u16::<LittleEndian>(self.create_date)?;
|
|
||||||
wrt.write_u16::<LittleEndian>(self.access_date)?;
|
|
||||||
wrt.write_u16::<LittleEndian>(self.first_cluster_hi)?;
|
|
||||||
wrt.write_u16::<LittleEndian>(self.modify_time)?;
|
|
||||||
wrt.write_u16::<LittleEndian>(self.modify_date)?;
|
|
||||||
wrt.write_u16::<LittleEndian>(self.first_cluster_lo)?;
|
|
||||||
wrt.write_u32::<LittleEndian>(self.size)?;
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
|
|
||||||
fn is_free(&self) -> bool {
|
|
||||||
self.name[0] == DIR_ENTRY_FREE_FLAG
|
|
||||||
}
|
|
||||||
|
|
||||||
fn set_free(&mut self) {
|
|
||||||
self.name[0] = DIR_ENTRY_FREE_FLAG;
|
|
||||||
}
|
|
||||||
|
|
||||||
fn is_end(&self) -> bool {
|
|
||||||
self.name[0] == 0
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[allow(dead_code)]
|
|
||||||
#[derive(Clone, Debug, Default)]
|
|
||||||
struct DirLfnEntryData {
|
|
||||||
order: u8,
|
|
||||||
name_0: [u16; 5],
|
|
||||||
attrs: FileAttributes,
|
|
||||||
entry_type: u8,
|
|
||||||
checksum: u8,
|
|
||||||
name_1: [u16; 6],
|
|
||||||
reserved_0: u16,
|
|
||||||
name_2: [u16; 2],
|
|
||||||
}
|
|
||||||
|
|
||||||
impl DirLfnEntryData {
|
|
||||||
fn new(order: u8, checksum: u8) -> Self {
|
|
||||||
DirLfnEntryData {
|
|
||||||
order, checksum,
|
|
||||||
attrs: FileAttributes::LFN,
|
|
||||||
..Default::default()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn copy_name_from_slice(&mut self, lfn_part: &[u16; LFN_PART_LEN]) {
|
|
||||||
self.name_0.copy_from_slice(&lfn_part[0..5]);
|
|
||||||
self.name_1.copy_from_slice(&lfn_part[5..5+6]);
|
|
||||||
self.name_2.copy_from_slice(&lfn_part[11..11+2]);
|
|
||||||
}
|
|
||||||
|
|
||||||
fn serialize(&self, wrt: &mut Write) -> io::Result<()> {
|
|
||||||
wrt.write_u8(self.order)?;
|
|
||||||
for ch in self.name_0.iter() {
|
|
||||||
wrt.write_u16::<LittleEndian>(*ch)?;
|
|
||||||
}
|
|
||||||
wrt.write_u8(self.attrs.bits())?;
|
|
||||||
wrt.write_u8(self.entry_type)?;
|
|
||||||
wrt.write_u8(self.checksum)?;
|
|
||||||
for ch in self.name_1.iter() {
|
|
||||||
wrt.write_u16::<LittleEndian>(*ch)?;
|
|
||||||
}
|
|
||||||
wrt.write_u16::<LittleEndian>(self.reserved_0)?;
|
|
||||||
for ch in self.name_2.iter() {
|
|
||||||
wrt.write_u16::<LittleEndian>(*ch)?;
|
|
||||||
}
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
|
|
||||||
fn is_free(&self) -> bool {
|
|
||||||
self.order == DIR_ENTRY_FREE_FLAG
|
|
||||||
}
|
|
||||||
|
|
||||||
fn set_free(&mut self) {
|
|
||||||
self.order = DIR_ENTRY_FREE_FLAG;
|
|
||||||
}
|
|
||||||
|
|
||||||
fn is_end(&self) -> bool {
|
|
||||||
self.order == 0
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Clone, Debug)]
|
|
||||||
enum DirEntryData {
|
|
||||||
File(DirFileEntryData),
|
|
||||||
Lfn(DirLfnEntryData),
|
|
||||||
}
|
|
||||||
|
|
||||||
impl DirEntryData {
|
|
||||||
fn serialize(&mut self, wrt: &mut Write) -> io::Result<()> {
|
|
||||||
match self {
|
|
||||||
&mut DirEntryData::File(ref mut file) => file.serialize(wrt),
|
|
||||||
&mut DirEntryData::Lfn(ref mut lfn) => lfn.serialize(wrt),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn deserialize(rdr: &mut Read) -> io::Result<DirEntryData> {
|
|
||||||
let mut name = [0; 11];
|
|
||||||
match rdr.read_exact(&mut name) {
|
|
||||||
Err(ref err) if err.kind() == io::ErrorKind::UnexpectedEof => {
|
|
||||||
return Ok(DirEntryData::File(DirFileEntryData {
|
|
||||||
..Default::default()
|
|
||||||
}));
|
|
||||||
}
|
|
||||||
Err(err) => return Err(err),
|
|
||||||
_ => {},
|
|
||||||
}
|
|
||||||
let attrs = FileAttributes::from_bits_truncate(rdr.read_u8()?);
|
|
||||||
if attrs & FileAttributes::LFN == FileAttributes::LFN {
|
|
||||||
let mut data = DirLfnEntryData {
|
|
||||||
attrs, ..Default::default()
|
|
||||||
};
|
|
||||||
let mut cur = Cursor::new(&name);
|
|
||||||
data.order = cur.read_u8()?;
|
|
||||||
cur.read_u16_into::<LittleEndian>(&mut data.name_0)?;
|
|
||||||
data.entry_type = rdr.read_u8()?;
|
|
||||||
data.checksum = rdr.read_u8()?;
|
|
||||||
rdr.read_u16_into::<LittleEndian>(&mut data.name_1)?;
|
|
||||||
data.reserved_0 = rdr.read_u16::<LittleEndian>()?;
|
|
||||||
rdr.read_u16_into::<LittleEndian>(&mut data.name_2)?;
|
|
||||||
Ok(DirEntryData::Lfn(data))
|
|
||||||
} else {
|
|
||||||
let data = DirFileEntryData {
|
|
||||||
name,
|
|
||||||
attrs,
|
|
||||||
reserved_0: rdr.read_u8()?,
|
|
||||||
create_time_0: rdr.read_u8()?,
|
|
||||||
create_time_1: rdr.read_u16::<LittleEndian>()?,
|
|
||||||
create_date: rdr.read_u16::<LittleEndian>()?,
|
|
||||||
access_date: rdr.read_u16::<LittleEndian>()?,
|
|
||||||
first_cluster_hi: rdr.read_u16::<LittleEndian>()?,
|
|
||||||
modify_time: rdr.read_u16::<LittleEndian>()?,
|
|
||||||
modify_date: rdr.read_u16::<LittleEndian>()?,
|
|
||||||
first_cluster_lo: rdr.read_u16::<LittleEndian>()?,
|
|
||||||
size: rdr.read_u32::<LittleEndian>()?,
|
|
||||||
};
|
|
||||||
Ok(DirEntryData::File(data))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn is_free(&self) -> bool {
|
|
||||||
match self {
|
|
||||||
&DirEntryData::File(ref file) => file.is_free(),
|
|
||||||
&DirEntryData::Lfn(ref lfn) => lfn.is_free(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn set_free(&mut self) {
|
|
||||||
match self {
|
|
||||||
&mut DirEntryData::File(ref mut file) => file.set_free(),
|
|
||||||
&mut DirEntryData::Lfn(ref mut lfn) => lfn.set_free(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn is_end(&self) -> bool {
|
|
||||||
match self {
|
|
||||||
&DirEntryData::File(ref file) => file.is_end(),
|
|
||||||
&DirEntryData::Lfn(ref lfn) => lfn.is_end(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// DOS compatible date
|
|
||||||
#[derive(Clone, Copy, Debug, PartialEq)]
|
|
||||||
pub struct Date {
|
|
||||||
pub year: u16,
|
|
||||||
pub month: u16,
|
|
||||||
pub day: u16,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Date {
|
|
||||||
|
|
||||||
pub(crate) fn from_u16(dos_date: u16) -> Self {
|
|
||||||
let (year, month, day) = ((dos_date >> 9) + 1980, (dos_date >> 5) & 0xF, dos_date & 0x1F);
|
|
||||||
Date { year, month, day }
|
|
||||||
}
|
|
||||||
|
|
||||||
fn to_u16(&self) -> u16 {
|
|
||||||
((self.year - 1980) << 9) | (self.month << 5) | self.day
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// DOS compatible time
|
|
||||||
#[derive(Clone, Copy, Debug, PartialEq)]
|
|
||||||
pub struct Time {
|
|
||||||
pub hour: u16,
|
|
||||||
pub min: u16,
|
|
||||||
pub sec: u16,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Time {
|
|
||||||
pub(crate) fn from_u16(dos_time: u16) -> Self {
|
|
||||||
let (hour, min, sec) = (dos_time >> 11, (dos_time >> 5) & 0x3F, (dos_time & 0x1F) * 2);
|
|
||||||
Time { hour, min, sec }
|
|
||||||
}
|
|
||||||
|
|
||||||
fn to_u16(&self) -> u16 {
|
|
||||||
(self.hour << 11) | (self.min << 5) | (self.sec / 2)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// DOS compatible date and time
|
|
||||||
#[derive(Clone, Copy, Debug, PartialEq)]
|
|
||||||
pub struct DateTime {
|
|
||||||
pub date: Date,
|
|
||||||
pub time: Time,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl DateTime {
|
|
||||||
pub(crate) fn from_u16(dos_date: u16, dos_time: u16) -> Self {
|
|
||||||
DateTime {
|
|
||||||
date: Date::from_u16(dos_date),
|
|
||||||
time: Time::from_u16(dos_time),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[cfg(feature = "chrono")]
|
|
||||||
impl From<Date> for chrono::Date<Local> {
|
|
||||||
fn from(date: Date) -> Self {
|
|
||||||
Local.ymd(date.year as i32, date.month as u32, date.day as u32)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[cfg(feature = "chrono")]
|
|
||||||
impl From<DateTime> for chrono::DateTime<Local> {
|
|
||||||
fn from(date_time: DateTime) -> Self {
|
|
||||||
chrono::Date::<Local>::from(date_time.date)
|
|
||||||
.and_hms(date_time.time.hour as u32, date_time.time.min as u32, date_time.time.sec as u32)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[cfg(feature = "chrono")]
|
|
||||||
impl From<chrono::Date<Local>> for Date {
|
|
||||||
fn from(date: chrono::Date<Local>) -> Self {
|
|
||||||
Date {
|
|
||||||
year: date.year() as u16,
|
|
||||||
month: date.month() as u16,
|
|
||||||
day: date.day() as u16,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[cfg(feature = "chrono")]
|
|
||||||
impl From<chrono::DateTime<Local>> for DateTime {
|
|
||||||
fn from(date_time: chrono::DateTime<Local>) -> Self {
|
|
||||||
DateTime {
|
|
||||||
date: Date::from(date_time.date()),
|
|
||||||
time: Time {
|
|
||||||
hour: date_time.hour() as u16,
|
|
||||||
min: date_time.minute() as u16,
|
|
||||||
sec: date_time.second() as u16,
|
|
||||||
},
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Clone, Debug)]
|
|
||||||
pub(crate) struct DirEntryEditor {
|
|
||||||
data: DirFileEntryData,
|
|
||||||
pos: u64,
|
|
||||||
dirty: bool,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl DirEntryEditor {
|
|
||||||
fn new(data: DirFileEntryData, pos: u64) -> DirEntryEditor {
|
|
||||||
DirEntryEditor {
|
|
||||||
data, pos,
|
|
||||||
dirty: false,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub(crate) fn inner(&self) -> &DirFileEntryData {
|
|
||||||
&self.data
|
|
||||||
}
|
|
||||||
|
|
||||||
pub(crate) fn set_first_cluster(&mut self, first_cluster: Option<u32>, fat_type: FatType) {
|
|
||||||
if first_cluster != self.data.first_cluster(fat_type) {
|
|
||||||
self.data.set_first_cluster(first_cluster, fat_type);
|
|
||||||
self.dirty = true;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub(crate) fn set_size(&mut self, size: u32) {
|
|
||||||
match self.data.size() {
|
|
||||||
Some(n) if size != n => {
|
|
||||||
self.data.set_size(size);
|
|
||||||
self.dirty = true;
|
|
||||||
},
|
|
||||||
_ => {},
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub(crate) fn set_created(&mut self, date_time: DateTime) {
|
|
||||||
if date_time != self.data.created() {
|
|
||||||
self.data.set_created(date_time);
|
|
||||||
self.dirty = true;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub(crate) fn set_accessed(&mut self, date: Date) {
|
|
||||||
if date != self.data.accessed() {
|
|
||||||
self.data.set_accessed(date);
|
|
||||||
self.dirty = true;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub(crate) fn set_modified(&mut self, date_time: DateTime) {
|
|
||||||
if date_time != self.data.modified() {
|
|
||||||
self.data.set_modified(date_time);
|
|
||||||
self.dirty = true;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub(crate) fn reset_accessed(&mut self) {
|
|
||||||
self.dirty |= self.data.reset_accessed();
|
|
||||||
}
|
|
||||||
|
|
||||||
pub(crate) fn reset_modified(&mut self) {
|
|
||||||
self.data.reset_modified();
|
|
||||||
self.dirty = true;
|
|
||||||
}
|
|
||||||
|
|
||||||
pub(crate) fn flush(&mut self, fs: FileSystemRef) -> io::Result<()> {
|
|
||||||
if self.dirty {
|
|
||||||
self.write(fs)?;
|
|
||||||
self.dirty = false;
|
|
||||||
}
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
|
|
||||||
fn write(&self, fs: FileSystemRef) -> io::Result<()> {
|
|
||||||
let mut disk = fs.disk.borrow_mut();
|
|
||||||
disk.seek(io::SeekFrom::Start(self.pos))?;
|
|
||||||
self.data.serialize(&mut *disk)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// FAT directory entry.
|
|
||||||
///
|
|
||||||
/// Returned by DirIter.
|
|
||||||
#[derive(Clone)]
|
|
||||||
pub struct DirEntry<'a, 'b: 'a> {
|
|
||||||
data: DirFileEntryData,
|
|
||||||
lfn: Vec<u16>,
|
|
||||||
entry_pos: u64,
|
|
||||||
offset_range: (u64, u64),
|
|
||||||
fs: FileSystemRef<'a, 'b>,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl <'a, 'b> DirEntry<'a, 'b> {
|
|
||||||
/// Returns short file name
|
|
||||||
pub fn short_file_name(&self) -> String {
|
|
||||||
let name_str = String::from_utf8_lossy(&self.data.name[0..8]);
|
|
||||||
let ext_str = String::from_utf8_lossy(&self.data.name[8..11]);
|
|
||||||
let name_trimmed = name_str.trim_right();
|
|
||||||
let ext_trimmed = ext_str.trim_right();
|
|
||||||
if ext_trimmed.is_empty() {
|
|
||||||
name_trimmed.to_string()
|
|
||||||
} else {
|
|
||||||
format!("{}.{}", name_trimmed, ext_trimmed)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Returns long file name or if it doesn't exist fallbacks to short file name.
|
|
||||||
pub fn file_name(&self) -> String {
|
|
||||||
if self.lfn.len() > 0 {
|
|
||||||
String::from_utf16_lossy(&self.lfn)
|
|
||||||
} else {
|
|
||||||
self.short_file_name()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Returns file attributes
|
|
||||||
pub fn attributes(&self) -> FileAttributes {
|
|
||||||
self.data.attrs
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Checks if entry belongs to directory.
|
|
||||||
pub fn is_dir(&self) -> bool {
|
|
||||||
self.data.is_dir()
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Checks if entry belongs to regular file.
|
|
||||||
pub fn is_file(&self) -> bool {
|
|
||||||
self.data.is_file()
|
|
||||||
}
|
|
||||||
|
|
||||||
pub(crate) fn first_cluster(&self) -> Option<u32> {
|
|
||||||
self.data.first_cluster(self.fs.fat_type)
|
|
||||||
}
|
|
||||||
|
|
||||||
fn entry_info(&self) -> DirEntryEditor {
|
|
||||||
DirEntryEditor::new(self.data.clone(), self.entry_pos)
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Returns File struct for this entry.
|
|
||||||
///
|
|
||||||
/// Panics if this is not a file.
|
|
||||||
pub fn to_file(&self) -> File<'a, 'b> {
|
|
||||||
assert!(!self.is_dir(), "Not a file entry");
|
|
||||||
File::new(self.first_cluster(), Some(self.entry_info()), self.fs)
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Returns Dir struct for this entry.
|
|
||||||
///
|
|
||||||
/// Panics if this is not a directory.
|
|
||||||
pub fn to_dir(&self) -> Dir<'a, 'b> {
|
|
||||||
assert!(self.is_dir(), "Not a directory entry");
|
|
||||||
match self.first_cluster() {
|
|
||||||
Some(n) => {
|
|
||||||
let file = File::new(Some(n), Some(self.entry_info()), self.fs);
|
|
||||||
Dir::new(DirRawStream::File(file), self.fs)
|
|
||||||
},
|
|
||||||
None => self.fs.root_dir(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Returns file size or 0 for directory.
|
|
||||||
pub fn len(&self) -> u64 {
|
|
||||||
self.data.size as u64
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Returns file creation date and time.
|
|
||||||
pub fn created(&self) -> DateTime {
|
|
||||||
self.data.created()
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Returns file last access date.
|
|
||||||
pub fn accessed(&self) -> Date {
|
|
||||||
self.data.accessed()
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Returns file last modification date and time.
|
|
||||||
pub fn modified(&self) -> DateTime {
|
|
||||||
self.data.modified()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl <'a, 'b> fmt::Debug for DirEntry<'a, 'b> {
|
|
||||||
fn fmt(&self, f: &mut fmt::Formatter) -> Result<(), fmt::Error> {
|
|
||||||
self.data.fmt(f)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// FAT directory
|
/// FAT directory
|
||||||
#[derive(Clone)]
|
#[derive(Clone)]
|
||||||
pub struct Dir<'a, 'b: 'a> {
|
pub struct Dir<'a, 'b: 'a> {
|
||||||
@ -1091,9 +488,7 @@ impl LongNameBuilder {
|
|||||||
}
|
}
|
||||||
let pos = LFN_PART_LEN * (index - 1) as usize;
|
let pos = LFN_PART_LEN * (index - 1) as usize;
|
||||||
// copy name parts into LFN buffer
|
// copy name parts into LFN buffer
|
||||||
self.buf[pos+0..pos+5].copy_from_slice(&data.name_0);
|
data.copy_name_to_slice(&mut self.buf[pos..pos+13]);
|
||||||
self.buf[pos+5..pos+11].copy_from_slice(&data.name_1);
|
|
||||||
self.buf[pos+11..pos+13].copy_from_slice(&data.name_2);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
fn validate_chksum(&mut self, short_name: &[u8]) {
|
fn validate_chksum(&mut self, short_name: &[u8]) {
|
||||||
|
621
src/dir_entry.rs
Normal file
621
src/dir_entry.rs
Normal file
@ -0,0 +1,621 @@
|
|||||||
|
use std::fmt;
|
||||||
|
use std::io::prelude::*;
|
||||||
|
use std::io;
|
||||||
|
use std::io::Cursor;
|
||||||
|
use byteorder::{LittleEndian, ReadBytesExt, WriteBytesExt};
|
||||||
|
|
||||||
|
#[cfg(feature = "chrono")]
|
||||||
|
use chrono::{TimeZone, Local, Datelike, Timelike};
|
||||||
|
#[cfg(feature = "chrono")]
|
||||||
|
use chrono;
|
||||||
|
|
||||||
|
use fs::{FileSystemRef, FatType};
|
||||||
|
use file::File;
|
||||||
|
use dir::{Dir, DirRawStream};
|
||||||
|
|
||||||
|
bitflags! {
|
||||||
|
/// FAT file attributes
|
||||||
|
#[derive(Default)]
|
||||||
|
pub struct FileAttributes: u8 {
|
||||||
|
const READ_ONLY = 0x01;
|
||||||
|
const HIDDEN = 0x02;
|
||||||
|
const SYSTEM = 0x04;
|
||||||
|
const VOLUME_ID = 0x08;
|
||||||
|
const DIRECTORY = 0x10;
|
||||||
|
const ARCHIVE = 0x20;
|
||||||
|
const LFN = Self::READ_ONLY.bits | Self::HIDDEN.bits
|
||||||
|
| Self::SYSTEM.bits | Self::VOLUME_ID.bits;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) const LFN_PART_LEN: usize = 13;
|
||||||
|
pub(crate) const DIR_ENTRY_SIZE: u64 = 32;
|
||||||
|
pub(crate) const DIR_ENTRY_FREE_FLAG: u8 = 0xE5;
|
||||||
|
pub(crate) const LFN_ENTRY_LAST_FLAG: u8 = 0x40;
|
||||||
|
|
||||||
|
#[allow(dead_code)]
|
||||||
|
#[derive(Clone, Debug, Default)]
|
||||||
|
pub(crate) struct DirFileEntryData {
|
||||||
|
name: [u8; 11],
|
||||||
|
attrs: FileAttributes,
|
||||||
|
reserved_0: u8,
|
||||||
|
create_time_0: u8,
|
||||||
|
create_time_1: u16,
|
||||||
|
create_date: u16,
|
||||||
|
access_date: u16,
|
||||||
|
first_cluster_hi: u16,
|
||||||
|
modify_time: u16,
|
||||||
|
modify_date: u16,
|
||||||
|
first_cluster_lo: u16,
|
||||||
|
size: u32,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl DirFileEntryData {
|
||||||
|
pub(crate) fn new(name: [u8; 11], attrs: FileAttributes) -> Self {
|
||||||
|
DirFileEntryData {
|
||||||
|
name, attrs,
|
||||||
|
..Default::default()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn attributes(&self) -> FileAttributes {
|
||||||
|
self.attrs
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn first_cluster(&self, fat_type: FatType) -> Option<u32> {
|
||||||
|
let first_cluster_hi = if fat_type == FatType::Fat32 { self.first_cluster_hi } else { 0 };
|
||||||
|
let n = ((first_cluster_hi as u32) << 16) | self.first_cluster_lo as u32;
|
||||||
|
if n == 0 { None } else { Some(n) }
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn set_first_cluster(&mut self, cluster: Option<u32>, fat_type: FatType) {
|
||||||
|
let n = cluster.unwrap_or(0);
|
||||||
|
if fat_type == FatType::Fat32 {
|
||||||
|
self.first_cluster_hi = (n >> 16) as u16;
|
||||||
|
}
|
||||||
|
self.first_cluster_lo = (n & 0xFFFF) as u16;
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn size(&self) -> Option<u32> {
|
||||||
|
if self.is_file() {
|
||||||
|
Some(self.size)
|
||||||
|
} else {
|
||||||
|
None
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn set_size(&mut self, size: u32) {
|
||||||
|
self.size = size;
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn is_dir(&self) -> bool {
|
||||||
|
self.attrs.contains(FileAttributes::DIRECTORY)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn is_file(&self) -> bool {
|
||||||
|
!self.is_dir()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn created(&self) -> DateTime {
|
||||||
|
DateTime::from_u16(self.create_date, self.create_time_1)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn accessed(&self) -> Date {
|
||||||
|
Date::from_u16(self.access_date)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn modified(&self) -> DateTime {
|
||||||
|
DateTime::from_u16(self.modify_date, self.modify_time)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn set_created(&mut self, date_time: DateTime) {
|
||||||
|
self.create_date = date_time.date.to_u16();
|
||||||
|
self.create_time_1 = date_time.time.to_u16();
|
||||||
|
}
|
||||||
|
|
||||||
|
fn set_accessed(&mut self, date: Date) {
|
||||||
|
self.access_date = date.to_u16();
|
||||||
|
}
|
||||||
|
|
||||||
|
fn set_modified(&mut self, date_time: DateTime) {
|
||||||
|
self.modify_date = date_time.date.to_u16();
|
||||||
|
self.modify_time = date_time.time.to_u16();
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(feature = "chrono")]
|
||||||
|
pub(crate) fn reset_created(&mut self) {
|
||||||
|
let now = DateTime::from(chrono::Local::now());
|
||||||
|
self.set_created(now);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(feature = "chrono")]
|
||||||
|
pub(crate) fn reset_accessed(&mut self) -> bool {
|
||||||
|
let now = Date::from(chrono::Local::now().date());
|
||||||
|
if now == self.accessed() {
|
||||||
|
false
|
||||||
|
} else {
|
||||||
|
self.set_accessed(now);
|
||||||
|
true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(feature = "chrono")]
|
||||||
|
pub(crate) fn reset_modified(&mut self) {
|
||||||
|
let now = DateTime::from(chrono::Local::now());
|
||||||
|
self.set_modified(now);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(not(feature = "chrono"))]
|
||||||
|
pub(crate) fn reset_created(&mut self) {
|
||||||
|
// nop - user controls timestamps manually
|
||||||
|
false
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(not(feature = "chrono"))]
|
||||||
|
pub(crate) fn reset_accessed(&mut self) -> bool {
|
||||||
|
// nop - user controls timestamps manually
|
||||||
|
false
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(not(feature = "chrono"))]
|
||||||
|
pub(crate) fn reset_modified(&mut self) {
|
||||||
|
// nop - user controls timestamps manually
|
||||||
|
false
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn serialize(&self, wrt: &mut Write) -> io::Result<()> {
|
||||||
|
wrt.write_all(&self.name)?;
|
||||||
|
wrt.write_u8(self.attrs.bits())?;
|
||||||
|
wrt.write_u8(self.reserved_0)?;
|
||||||
|
wrt.write_u8(self.create_time_0)?;
|
||||||
|
wrt.write_u16::<LittleEndian>(self.create_time_1)?;
|
||||||
|
wrt.write_u16::<LittleEndian>(self.create_date)?;
|
||||||
|
wrt.write_u16::<LittleEndian>(self.access_date)?;
|
||||||
|
wrt.write_u16::<LittleEndian>(self.first_cluster_hi)?;
|
||||||
|
wrt.write_u16::<LittleEndian>(self.modify_time)?;
|
||||||
|
wrt.write_u16::<LittleEndian>(self.modify_date)?;
|
||||||
|
wrt.write_u16::<LittleEndian>(self.first_cluster_lo)?;
|
||||||
|
wrt.write_u32::<LittleEndian>(self.size)?;
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn is_free(&self) -> bool {
|
||||||
|
self.name[0] == DIR_ENTRY_FREE_FLAG
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn set_free(&mut self) {
|
||||||
|
self.name[0] = DIR_ENTRY_FREE_FLAG;
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn is_end(&self) -> bool {
|
||||||
|
self.name[0] == 0
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[allow(dead_code)]
|
||||||
|
#[derive(Clone, Debug, Default)]
|
||||||
|
pub(crate) struct DirLfnEntryData {
|
||||||
|
order: u8,
|
||||||
|
name_0: [u16; 5],
|
||||||
|
attrs: FileAttributes,
|
||||||
|
entry_type: u8,
|
||||||
|
checksum: u8,
|
||||||
|
name_1: [u16; 6],
|
||||||
|
reserved_0: u16,
|
||||||
|
name_2: [u16; 2],
|
||||||
|
}
|
||||||
|
|
||||||
|
impl DirLfnEntryData {
|
||||||
|
pub(crate) fn new(order: u8, checksum: u8) -> Self {
|
||||||
|
DirLfnEntryData {
|
||||||
|
order, checksum,
|
||||||
|
attrs: FileAttributes::LFN,
|
||||||
|
..Default::default()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn copy_name_from_slice(&mut self, lfn_part: &[u16; LFN_PART_LEN]) {
|
||||||
|
self.name_0.copy_from_slice(&lfn_part[0..5]);
|
||||||
|
self.name_1.copy_from_slice(&lfn_part[5..5+6]);
|
||||||
|
self.name_2.copy_from_slice(&lfn_part[11..11+2]);
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn copy_name_to_slice(&self, lfn_part: &mut [u16; LFN_PART_LEN]) {
|
||||||
|
lfn_part[0..5].copy_from_slice(&self.name_0);
|
||||||
|
lfn_part[5..11].copy_from_slice(&self.name_1);
|
||||||
|
lfn_part[11..13].copy_from_slice(&self.name_2);
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn serialize(&self, wrt: &mut Write) -> io::Result<()> {
|
||||||
|
wrt.write_u8(self.order)?;
|
||||||
|
for ch in self.name_0.iter() {
|
||||||
|
wrt.write_u16::<LittleEndian>(*ch)?;
|
||||||
|
}
|
||||||
|
wrt.write_u8(self.attrs.bits())?;
|
||||||
|
wrt.write_u8(self.entry_type)?;
|
||||||
|
wrt.write_u8(self.checksum)?;
|
||||||
|
for ch in self.name_1.iter() {
|
||||||
|
wrt.write_u16::<LittleEndian>(*ch)?;
|
||||||
|
}
|
||||||
|
wrt.write_u16::<LittleEndian>(self.reserved_0)?;
|
||||||
|
for ch in self.name_2.iter() {
|
||||||
|
wrt.write_u16::<LittleEndian>(*ch)?;
|
||||||
|
}
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn is_free(&self) -> bool {
|
||||||
|
self.order == DIR_ENTRY_FREE_FLAG
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn set_free(&mut self) {
|
||||||
|
self.order = DIR_ENTRY_FREE_FLAG;
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn is_end(&self) -> bool {
|
||||||
|
self.order == 0
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Clone, Debug)]
|
||||||
|
pub(crate) enum DirEntryData {
|
||||||
|
File(DirFileEntryData),
|
||||||
|
Lfn(DirLfnEntryData),
|
||||||
|
}
|
||||||
|
|
||||||
|
impl DirEntryData {
|
||||||
|
pub(crate) fn serialize(&mut self, wrt: &mut Write) -> io::Result<()> {
|
||||||
|
match self {
|
||||||
|
&mut DirEntryData::File(ref mut file) => file.serialize(wrt),
|
||||||
|
&mut DirEntryData::Lfn(ref mut lfn) => lfn.serialize(wrt),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn deserialize(rdr: &mut Read) -> io::Result<DirEntryData> {
|
||||||
|
let mut name = [0; 11];
|
||||||
|
match rdr.read_exact(&mut name) {
|
||||||
|
Err(ref err) if err.kind() == io::ErrorKind::UnexpectedEof => {
|
||||||
|
return Ok(DirEntryData::File(DirFileEntryData {
|
||||||
|
..Default::default()
|
||||||
|
}));
|
||||||
|
}
|
||||||
|
Err(err) => return Err(err),
|
||||||
|
_ => {},
|
||||||
|
}
|
||||||
|
let attrs = FileAttributes::from_bits_truncate(rdr.read_u8()?);
|
||||||
|
if attrs & FileAttributes::LFN == FileAttributes::LFN {
|
||||||
|
let mut data = DirLfnEntryData {
|
||||||
|
attrs, ..Default::default()
|
||||||
|
};
|
||||||
|
let mut cur = Cursor::new(&name);
|
||||||
|
data.order = cur.read_u8()?;
|
||||||
|
cur.read_u16_into::<LittleEndian>(&mut data.name_0)?;
|
||||||
|
data.entry_type = rdr.read_u8()?;
|
||||||
|
data.checksum = rdr.read_u8()?;
|
||||||
|
rdr.read_u16_into::<LittleEndian>(&mut data.name_1)?;
|
||||||
|
data.reserved_0 = rdr.read_u16::<LittleEndian>()?;
|
||||||
|
rdr.read_u16_into::<LittleEndian>(&mut data.name_2)?;
|
||||||
|
Ok(DirEntryData::Lfn(data))
|
||||||
|
} else {
|
||||||
|
let data = DirFileEntryData {
|
||||||
|
name,
|
||||||
|
attrs,
|
||||||
|
reserved_0: rdr.read_u8()?,
|
||||||
|
create_time_0: rdr.read_u8()?,
|
||||||
|
create_time_1: rdr.read_u16::<LittleEndian>()?,
|
||||||
|
create_date: rdr.read_u16::<LittleEndian>()?,
|
||||||
|
access_date: rdr.read_u16::<LittleEndian>()?,
|
||||||
|
first_cluster_hi: rdr.read_u16::<LittleEndian>()?,
|
||||||
|
modify_time: rdr.read_u16::<LittleEndian>()?,
|
||||||
|
modify_date: rdr.read_u16::<LittleEndian>()?,
|
||||||
|
first_cluster_lo: rdr.read_u16::<LittleEndian>()?,
|
||||||
|
size: rdr.read_u32::<LittleEndian>()?,
|
||||||
|
};
|
||||||
|
Ok(DirEntryData::File(data))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn is_free(&self) -> bool {
|
||||||
|
match self {
|
||||||
|
&DirEntryData::File(ref file) => file.is_free(),
|
||||||
|
&DirEntryData::Lfn(ref lfn) => lfn.is_free(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn set_free(&mut self) {
|
||||||
|
match self {
|
||||||
|
&mut DirEntryData::File(ref mut file) => file.set_free(),
|
||||||
|
&mut DirEntryData::Lfn(ref mut lfn) => lfn.set_free(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn is_end(&self) -> bool {
|
||||||
|
match self {
|
||||||
|
&DirEntryData::File(ref file) => file.is_end(),
|
||||||
|
&DirEntryData::Lfn(ref lfn) => lfn.is_end(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// DOS compatible date
|
||||||
|
#[derive(Clone, Copy, Debug, PartialEq)]
|
||||||
|
pub struct Date {
|
||||||
|
pub year: u16,
|
||||||
|
pub month: u16,
|
||||||
|
pub day: u16,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Date {
|
||||||
|
|
||||||
|
pub(crate) fn from_u16(dos_date: u16) -> Self {
|
||||||
|
let (year, month, day) = ((dos_date >> 9) + 1980, (dos_date >> 5) & 0xF, dos_date & 0x1F);
|
||||||
|
Date { year, month, day }
|
||||||
|
}
|
||||||
|
|
||||||
|
fn to_u16(&self) -> u16 {
|
||||||
|
((self.year - 1980) << 9) | (self.month << 5) | self.day
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// DOS compatible time
|
||||||
|
#[derive(Clone, Copy, Debug, PartialEq)]
|
||||||
|
pub struct Time {
|
||||||
|
pub hour: u16,
|
||||||
|
pub min: u16,
|
||||||
|
pub sec: u16,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Time {
|
||||||
|
pub(crate) fn from_u16(dos_time: u16) -> Self {
|
||||||
|
let (hour, min, sec) = (dos_time >> 11, (dos_time >> 5) & 0x3F, (dos_time & 0x1F) * 2);
|
||||||
|
Time { hour, min, sec }
|
||||||
|
}
|
||||||
|
|
||||||
|
fn to_u16(&self) -> u16 {
|
||||||
|
(self.hour << 11) | (self.min << 5) | (self.sec / 2)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// DOS compatible date and time
|
||||||
|
#[derive(Clone, Copy, Debug, PartialEq)]
|
||||||
|
pub struct DateTime {
|
||||||
|
pub date: Date,
|
||||||
|
pub time: Time,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl DateTime {
|
||||||
|
pub(crate) fn from_u16(dos_date: u16, dos_time: u16) -> Self {
|
||||||
|
DateTime {
|
||||||
|
date: Date::from_u16(dos_date),
|
||||||
|
time: Time::from_u16(dos_time),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(feature = "chrono")]
|
||||||
|
impl From<Date> for chrono::Date<Local> {
|
||||||
|
fn from(date: Date) -> Self {
|
||||||
|
Local.ymd(date.year as i32, date.month as u32, date.day as u32)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(feature = "chrono")]
|
||||||
|
impl From<DateTime> for chrono::DateTime<Local> {
|
||||||
|
fn from(date_time: DateTime) -> Self {
|
||||||
|
chrono::Date::<Local>::from(date_time.date)
|
||||||
|
.and_hms(date_time.time.hour as u32, date_time.time.min as u32, date_time.time.sec as u32)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(feature = "chrono")]
|
||||||
|
impl From<chrono::Date<Local>> for Date {
|
||||||
|
fn from(date: chrono::Date<Local>) -> Self {
|
||||||
|
Date {
|
||||||
|
year: date.year() as u16,
|
||||||
|
month: date.month() as u16,
|
||||||
|
day: date.day() as u16,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(feature = "chrono")]
|
||||||
|
impl From<chrono::DateTime<Local>> for DateTime {
|
||||||
|
fn from(date_time: chrono::DateTime<Local>) -> Self {
|
||||||
|
DateTime {
|
||||||
|
date: Date::from(date_time.date()),
|
||||||
|
time: Time {
|
||||||
|
hour: date_time.hour() as u16,
|
||||||
|
min: date_time.minute() as u16,
|
||||||
|
sec: date_time.second() as u16,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Clone, Debug)]
|
||||||
|
pub(crate) struct DirEntryEditor {
|
||||||
|
data: DirFileEntryData,
|
||||||
|
pos: u64,
|
||||||
|
dirty: bool,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl DirEntryEditor {
|
||||||
|
fn new(data: DirFileEntryData, pos: u64) -> DirEntryEditor {
|
||||||
|
DirEntryEditor {
|
||||||
|
data, pos,
|
||||||
|
dirty: false,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn inner(&self) -> &DirFileEntryData {
|
||||||
|
&self.data
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn set_first_cluster(&mut self, first_cluster: Option<u32>, fat_type: FatType) {
|
||||||
|
if first_cluster != self.data.first_cluster(fat_type) {
|
||||||
|
self.data.set_first_cluster(first_cluster, fat_type);
|
||||||
|
self.dirty = true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn set_size(&mut self, size: u32) {
|
||||||
|
match self.data.size() {
|
||||||
|
Some(n) if size != n => {
|
||||||
|
self.data.set_size(size);
|
||||||
|
self.dirty = true;
|
||||||
|
},
|
||||||
|
_ => {},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn set_created(&mut self, date_time: DateTime) {
|
||||||
|
if date_time != self.data.created() {
|
||||||
|
self.data.set_created(date_time);
|
||||||
|
self.dirty = true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn set_accessed(&mut self, date: Date) {
|
||||||
|
if date != self.data.accessed() {
|
||||||
|
self.data.set_accessed(date);
|
||||||
|
self.dirty = true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn set_modified(&mut self, date_time: DateTime) {
|
||||||
|
if date_time != self.data.modified() {
|
||||||
|
self.data.set_modified(date_time);
|
||||||
|
self.dirty = true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn reset_accessed(&mut self) {
|
||||||
|
self.dirty |= self.data.reset_accessed();
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn reset_modified(&mut self) {
|
||||||
|
self.data.reset_modified();
|
||||||
|
self.dirty = true;
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn flush(&mut self, fs: FileSystemRef) -> io::Result<()> {
|
||||||
|
if self.dirty {
|
||||||
|
self.write(fs)?;
|
||||||
|
self.dirty = false;
|
||||||
|
}
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
fn write(&self, fs: FileSystemRef) -> io::Result<()> {
|
||||||
|
let mut disk = fs.disk.borrow_mut();
|
||||||
|
disk.seek(io::SeekFrom::Start(self.pos))?;
|
||||||
|
self.data.serialize(&mut *disk)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// FAT directory entry.
|
||||||
|
///
|
||||||
|
/// Returned by DirIter.
|
||||||
|
#[derive(Clone)]
|
||||||
|
pub struct DirEntry<'a, 'b: 'a> {
|
||||||
|
pub(crate) data: DirFileEntryData,
|
||||||
|
pub(crate) lfn: Vec<u16>,
|
||||||
|
pub(crate) entry_pos: u64,
|
||||||
|
pub(crate) offset_range: (u64, u64),
|
||||||
|
pub(crate) fs: FileSystemRef<'a, 'b>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl <'a, 'b> DirEntry<'a, 'b> {
|
||||||
|
/// Returns short file name
|
||||||
|
pub fn short_file_name(&self) -> String {
|
||||||
|
let name_str = String::from_utf8_lossy(&self.data.name[0..8]);
|
||||||
|
let ext_str = String::from_utf8_lossy(&self.data.name[8..11]);
|
||||||
|
let name_trimmed = name_str.trim_right();
|
||||||
|
let ext_trimmed = ext_str.trim_right();
|
||||||
|
if ext_trimmed.is_empty() {
|
||||||
|
name_trimmed.to_string()
|
||||||
|
} else {
|
||||||
|
format!("{}.{}", name_trimmed, ext_trimmed)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Returns long file name or if it doesn't exist fallbacks to short file name.
|
||||||
|
pub fn file_name(&self) -> String {
|
||||||
|
if self.lfn.len() > 0 {
|
||||||
|
String::from_utf16_lossy(&self.lfn)
|
||||||
|
} else {
|
||||||
|
self.short_file_name()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Returns file attributes
|
||||||
|
pub fn attributes(&self) -> FileAttributes {
|
||||||
|
self.data.attrs
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Checks if entry belongs to directory.
|
||||||
|
pub fn is_dir(&self) -> bool {
|
||||||
|
self.data.is_dir()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Checks if entry belongs to regular file.
|
||||||
|
pub fn is_file(&self) -> bool {
|
||||||
|
self.data.is_file()
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn first_cluster(&self) -> Option<u32> {
|
||||||
|
self.data.first_cluster(self.fs.fat_type)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn entry_info(&self) -> DirEntryEditor {
|
||||||
|
DirEntryEditor::new(self.data.clone(), self.entry_pos)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Returns File struct for this entry.
|
||||||
|
///
|
||||||
|
/// Panics if this is not a file.
|
||||||
|
pub fn to_file(&self) -> File<'a, 'b> {
|
||||||
|
assert!(!self.is_dir(), "Not a file entry");
|
||||||
|
File::new(self.first_cluster(), Some(self.entry_info()), self.fs)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Returns Dir struct for this entry.
|
||||||
|
///
|
||||||
|
/// Panics if this is not a directory.
|
||||||
|
pub fn to_dir(&self) -> Dir<'a, 'b> {
|
||||||
|
assert!(self.is_dir(), "Not a directory entry");
|
||||||
|
match self.first_cluster() {
|
||||||
|
Some(n) => {
|
||||||
|
let file = File::new(Some(n), Some(self.entry_info()), self.fs);
|
||||||
|
Dir::new(DirRawStream::File(file), self.fs)
|
||||||
|
},
|
||||||
|
None => self.fs.root_dir(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Returns file size or 0 for directory.
|
||||||
|
pub fn len(&self) -> u64 {
|
||||||
|
self.data.size as u64
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Returns file creation date and time.
|
||||||
|
pub fn created(&self) -> DateTime {
|
||||||
|
self.data.created()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Returns file last access date.
|
||||||
|
pub fn accessed(&self) -> Date {
|
||||||
|
self.data.accessed()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Returns file last modification date and time.
|
||||||
|
pub fn modified(&self) -> DateTime {
|
||||||
|
self.data.modified()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl <'a, 'b> fmt::Debug for DirEntry<'a, 'b> {
|
||||||
|
fn fmt(&self, f: &mut fmt::Formatter) -> Result<(), fmt::Error> {
|
||||||
|
self.data.fmt(f)
|
||||||
|
}
|
||||||
|
}
|
@ -4,7 +4,7 @@ use std::io::{SeekFrom, ErrorKind};
|
|||||||
use std::io;
|
use std::io;
|
||||||
|
|
||||||
use fs::FileSystemRef;
|
use fs::FileSystemRef;
|
||||||
use dir::{DirEntryEditor, DateTime, Date};
|
use dir_entry::{DirEntryEditor, DateTime, Date};
|
||||||
|
|
||||||
/// FAT file used for reading and writing.
|
/// FAT file used for reading and writing.
|
||||||
#[derive(Clone)]
|
#[derive(Clone)]
|
||||||
|
@ -15,6 +15,7 @@ extern crate chrono;
|
|||||||
|
|
||||||
mod fs;
|
mod fs;
|
||||||
mod dir;
|
mod dir;
|
||||||
|
mod dir_entry;
|
||||||
mod file;
|
mod file;
|
||||||
mod table;
|
mod table;
|
||||||
mod utils;
|
mod utils;
|
||||||
|
Loading…
Reference in New Issue
Block a user