isobemak 0.2.4

Create bootable ISO images with FAT32 and UEFI (El Torito) support in Rust.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98

use std::fs::File;
use std::io::{self, Seek, SeekFrom};

pub const ISO_SECTOR_SIZE: usize = 2048;

pub fn seek_to_lba(file: &mut File, lba: u32) -> io::Result<u64> {
    let target_pos = lba as u64 * ISO_SECTOR_SIZE as u64;
    file.seek(SeekFrom::Start(target_pos))
}

/// Helper macro to create consistent IO errors
#[macro_export]
macro_rules! io_error {
    ($kind:expr, $($arg:tt)*) => {
        io::Error::new($kind, format!($($arg)*))
    };
}

/// Helper macro to validate path components
#[macro_export]
macro_rules! ensure_path_component {
    ($component:expr, $path:expr) => {
        $component.as_os_str().to_str().ok_or_else(|| {
            $crate::io_error!(
                io::ErrorKind::InvalidInput,
                "Invalid path component in {}: {:?}",
                $path,
                $component
            )
        })?
    };
}

/// Helper macro to validate file sizes for boot catalog entries
#[macro_export]
macro_rules! ensure_boot_image_size_valid {
    ($size:expr, $max_size:expr, $image_type:expr) => {
        if $size > $max_size {
            return Err($crate::io_error!(
                io::ErrorKind::InvalidInput,
                "{} image is too large for the boot catalog ({} > {})",
                $image_type,
                $size,
                $max_size
            ));
        }
    };
}

/// Helper macro to iterate over sorted children of a directory
#[macro_export]
macro_rules! for_sorted_children {
    ($dir:expr, |$name:ident, $node:ident| $body:block) => {{
        let mut sorted_children: Vec<_> = $dir.children.iter().collect();
        sorted_children.sort_by_key(|(name, _)| *name);
        for ($name, $node) in sorted_children {
            $body
        }
    }};

    ($dir:expr, mut |$name:ident, $node:ident| $body:block) => {{
        let mut sorted_children: Vec<_> = $dir.children.iter_mut().collect();
        sorted_children.sort_by_key(|(name, _)| *name);
        for ($name, $node) in sorted_children {
            $body
        }
    }};
}

#[cfg(test)]
pub mod test_utils {
    use std::fs;
    use std::io::{self, Write};
    use std::path::{Path, PathBuf};

    /// Creates a dummy file with the specified size in a temporary directory.
    pub fn create_dummy_file(temp_dir: &Path, name: &str, size_kb: usize) -> io::Result<PathBuf> {
        let path = temp_dir.join(name);
        let mut file = fs::File::create(&path)?;
        file.write_all(&vec![0u8; size_kb * 1024])?;
        Ok(path)
    }

    /// A macro to simplify the creation of multiple dummy files.
    #[macro_export]
    macro_rules! create_dummy_files {
        ($temp_dir:expr, $($name:expr => $size_kb:expr),*) => {
            {
                let mut paths = std::collections::HashMap::new();
                $(
                    let path = $crate::utils::test_utils::create_dummy_file($temp_dir, $name, $size_kb).unwrap();
                    paths.insert($name.to_string(), path);
                )*
                paths
            }
        };
    }
}