ext4fs-core 0.2.2

#![forbid(unsafe_code)]

use crate::error::Result;
use crate::inode::InodeReader;
use crate::ondisk::xattr::{XattrBlockHeader, XattrEntry, XattrNamespace, XATTR_MAGIC};
use std::io::{Read, Seek};

/// A parsed extended attribute with its value.
#[derive(Debug, Clone)]
pub struct Xattr {
    pub namespace: XattrNamespace,
    pub name: Vec<u8>,
    pub value: Vec<u8>,
}

/// Read extended attributes for an inode.
///
/// Reads both inline xattrs stored in the inode body (ibody) and
/// block-stored xattrs from `i_file_acl`.
pub fn read_xattrs<R: Read + Seek>(reader: &mut InodeReader<R>, ino: u64) -> Result<Vec<Xattr>> {
    let inode = reader.read_inode(ino)?;
    let mut xattrs = Vec::new();

    // --- Inline xattrs (ibody) ---
    let inode_size = reader.block_reader().superblock().inode_size as usize;
    let ibody_offset = 0x80 + inode.extra_isize as usize;
    if inode.extra_isize > 0 && inode_size > ibody_offset {
        let raw = reader.read_inode_raw(ino)?;
        let ibody_region = &raw[ibody_offset..];
        if ibody_region.len() >= 4 {
            // Check for and skip the 4-byte inline xattr magic (0xEA020000)
            let magic = u32::from_le_bytes([
                ibody_region[0],
                ibody_region[1],
                ibody_region[2],
                ibody_region[3],
            ]);
            let entry_start = if magic == XATTR_MAGIC { 4 } else { 0 };
            let mut offset = entry_start;
            while offset + 16 <= ibody_region.len() {
                // Zero name_index signals end of entries
                if ibody_region[offset] == 0 {
                    break;
                }
                match XattrEntry::parse(&ibody_region[offset..]) {
                    Ok(entry) => {
                        // For ibody xattrs, value_offset is relative to
                        // the start of the entry area (after the magic),
                        // so add entry_start to get the position in ibody_region.
                        let value_start = entry.value_offset as usize + entry_start;
                        let value_end = value_start + entry.value_size as usize;
                        let value = if value_end <= ibody_region.len() {
                            ibody_region[value_start..value_end].to_vec()
                        } else {
                            Vec::new()
                        };
                        xattrs.push(Xattr {
                            namespace: entry.name_index,
                            name: entry.name.clone(),
                            value,
                        });
                        offset += entry.entry_size;
                    }
                    Err(_) => break,
                }
            }
        }
    }

    // --- Block xattrs (from i_file_acl) ---
    if inode.file_acl != 0 {
        let block_data = reader.block_reader_mut().read_block(inode.file_acl)?;
        if let Ok(_header) = XattrBlockHeader::parse(&block_data) {
            let mut offset = 32; // skip header
            while offset + 16 <= block_data.len() {
                if block_data[offset] == 0 {
                    break;
                }
                match XattrEntry::parse(&block_data[offset..]) {
                    Ok(entry) => {
                        let value_start = entry.value_offset as usize;
                        let value_end = value_start + entry.value_size as usize;
                        let value = if value_end <= block_data.len() {
                            block_data[value_start..value_end].to_vec()
                        } else {
                            Vec::new()
                        };

                        xattrs.push(Xattr {
                            namespace: entry.name_index,
                            name: entry.name.clone(),
                            value,
                        });
                        offset += entry.entry_size;
                    }
                    Err(_) => break,
                }
            }
        }
    }

    Ok(xattrs)
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::block::BlockReader;
    use crate::inode::InodeReader;
    use std::io::Cursor;

    fn open_minimal() -> Option<InodeReader<Cursor<Vec<u8>>>> {
        let path = concat!(env!("CARGO_MANIFEST_DIR"), "/../tests/data/minimal.img");
        let data = std::fs::read(path).ok()?;
        let br = BlockReader::open(Cursor::new(data)).ok()?;
        Some(InodeReader::new(br))
    }

    #[test]
    fn read_xattrs_from_file() {
        let mut reader = if let Some(r) = open_minimal() {
            r
        } else {
            eprintln!("skip: minimal.img not found");
            return;
        };
        let xattrs = read_xattrs(&mut reader, 2).unwrap();
        let _ = xattrs;
    }

    fn open_forensic() -> Option<InodeReader<Cursor<Vec<u8>>>> {
        let path = concat!(env!("CARGO_MANIFEST_DIR"), "/../tests/data/forensic.img");
        let data = std::fs::read(path).ok()?;
        let br = BlockReader::open(Cursor::new(data)).ok()?;
        Some(InodeReader::new(br))
    }

    #[test]
    fn read_xattrs_for_hello_txt() {
        let mut reader = if let Some(r) = open_forensic() {
            r
        } else {
            eprintln!("skip: forensic.img not found");
            return;
        };

        // Read the inode to inspect file_acl
        let inode = reader.read_inode(12).unwrap();
        eprintln!("hello.txt (ino 12) file_acl = {}", inode.file_acl);

        let xattrs = read_xattrs(&mut reader, 12).unwrap();
        eprintln!("xattrs found: {}", xattrs.len());
        for xa in &xattrs {
            eprintln!(
                "  namespace={:?} name={:?} value={:?}",
                xa.namespace,
                String::from_utf8_lossy(&xa.name),
                String::from_utf8_lossy(&xa.value),
            );
        }
        // Don't assert specific xattr names yet — just assert no error occurred.
    }

    #[test]
    fn read_xattrs_for_root_inode() {
        let mut reader = if let Some(r) = open_forensic() {
            r
        } else {
            eprintln!("skip: forensic.img not found");
            return;
        };

        let xattrs = read_xattrs(&mut reader, 2).unwrap();
        let _ = xattrs;
    }

    #[test]
    fn read_inline_xattrs_for_hello_txt() {
        let mut reader = if let Some(r) = open_forensic() {
            r
        } else {
            eprintln!("skip: forensic.img not found");
            return;
        };
        // hello.txt is inode 12 in forensic.img
        let xattrs = read_xattrs(&mut reader, 12).unwrap();
        let names: Vec<String> = xattrs
            .iter()
            .map(|x| String::from_utf8_lossy(&x.name).to_string())
            .collect();
        assert!(
            names.contains(&"forensic".to_string()),
            "expected user.forensic xattr, found: {names:?}"
        );
        assert!(
            names.contains(&"case_id".to_string()),
            "expected user.case_id xattr, found: {names:?}"
        );
    }

    #[test]
    fn inline_xattr_values_are_correct() {
        let mut reader = if let Some(r) = open_forensic() {
            r
        } else {
            eprintln!("skip: forensic.img not found");
            return;
        };
        let xattrs = read_xattrs(&mut reader, 12).unwrap();
        let forensic_xattr = xattrs
            .iter()
            .find(|x| x.name == b"forensic")
            .expect("user.forensic xattr not found");
        assert_eq!(
            String::from_utf8_lossy(&forensic_xattr.value),
            "evidence-tag"
        );

        let case_xattr = xattrs
            .iter()
            .find(|x| x.name == b"case_id")
            .expect("user.case_id xattr not found");
        assert_eq!(String::from_utf8_lossy(&case_xattr.value), "2026-0401");
    }
}