Removed font loading, need to figure out a better solution

cz/src/compression.rs

@@ -391,7 +391,7 @@ fn compress_lzw2(data: &[u8], last: Vec<u8>) -> (usize, Vec<u8>, Vec<u8>) {
 
         if dictionary_count >= 0x3FFFE {
             count -= 1;
-            break
+            break;
         }
     }
 

cz/src/dynamic.rs

@@ -223,12 +223,7 @@ impl DynamicCz {
 
     /// Create a CZ# image from RGBA bytes. The bytes *must* be RGBA, as that
     /// is the only format that is used internally.
-    pub fn from_raw(
-        version: CzVersion,
-        width: u16,
-        height: u16,
-        bitmap: Vec<u8>,
-    ) -> Self {
+    pub fn from_raw(version: CzVersion, width: u16, height: u16, bitmap: Vec<u8>) -> Self {
         let header_common = CommonHeader::new(version, width, height);
 
         Self {

luca_pak/src/entry.rs

@@ -1,5 +1,8 @@
 use std::{
-    error::Error, fs::File, io::{BufWriter, Write}, path::Path
+    error::Error,
+    fs::File,
+    io::{BufWriter, Write},
+    path::Path,
 };
 
 /// A single file entry in a PAK file
@@ -87,7 +90,6 @@ impl Entry {
     }
 }
 
-
 #[derive(Debug, Clone, Copy, PartialEq, Eq)]
 pub enum EntryType {
     CZ0,

luca_pak/src/header.rs

@@ -1,5 +1,5 @@
-use std::io::{self, Write};
 use byteorder::WriteBytesExt;
+use std::io::{self, Write};
 
 use crate::LE;
 

luca_pak/src/lib.rs

@@ -1,14 +1,16 @@
 pub mod entry;
 pub mod header;
 
-use byteorder::{LittleEndian, ReadBytesExt};
+use byteorder::{LittleEndian, ReadBytesExt, WriteBytesExt};
 use header::Header;
 use log::{debug, info};
 use std::{
-    ffi::CString, fs::File, io::{self, BufRead, BufReader, BufWriter, Read, Seek, SeekFrom, Write}, path::{Path, PathBuf}
+    ffi::CString,
+    fs::File,
+    io::{self, BufRead, BufReader, BufWriter, Read, Seek, SeekFrom, Write},
+    path::{Path, PathBuf},
 };
 use thiserror::Error;
-use byteorder::WriteBytesExt;
 
 type LE = LittleEndian;
 
@@ -20,8 +22,11 @@ pub enum PakError {
     #[error("Could not read/write file")]
     IoError(#[from] io::Error),
 
-    #[error("Expected {} files, got {} in {}", 0, 1, 2)]
-    FileCountMismatch(usize, usize, &'static str),
+    #[error("Expected {0} files, got {1} in {2}")]
+    EntryCountMismatch(usize, usize, &'static str),
+
+    #[error("Number of entries in header ({0}) exceeds limit of {1}")]
+    EntryLimit(u32, usize),
 
     #[error("Malformed header information")]
     HeaderError,
@@ -35,9 +40,11 @@ pub enum PakError {
 pub struct Pak {
     subdirectory: Option<String>,
 
-    /// The path of the PAK file, can serve as an identifier or name as the
+    /// The path to the PAK file, can serve as an identifier or name as the
     /// header has no name for the file.
     path: PathBuf,
+
+    /// Header information
     header: Header,
 
     unknown_pre_data: Vec<u32>,
@@ -46,23 +53,38 @@ pub struct Pak {
     entries: Vec<Entry>,
 }
 
-struct FileLocation {
+struct EntryLocation {
     offset: u32,
     length: u32,
 }
 
+pub struct PakLimits {
+    pub entry_limit: usize,
+    pub size_limit: usize,
+}
+
+impl Default for PakLimits {
+    fn default() -> Self {
+        Self {
+            entry_limit: 10_000,        // 10,000 entries
+            size_limit: 10_000_000_000, // 10 gb
+        }
+    }
+}
+
 impl Pak {
     /// Convenience method to open a PAK file from a path and decode it
     pub fn open<P: ?Sized + AsRef<Path>>(path: &P) -> Result<Self, PakError> {
         let mut file = File::open(path)?;
 
-        Pak::decode(&mut file, path.as_ref().to_path_buf())
+        Pak::decode(&mut file, path.as_ref().to_path_buf(), PakLimits::default())
     }
 
     /// Decode a PAK file from a byte stream.
     pub fn decode<T: Seek + Read>(
         input: &mut T,
         path: PathBuf,
+        limits: PakLimits,
     ) -> Result<Self, PakError> {
         info!("Reading pak from {:?}", path);
         let mut input = BufReader::new(input);
@@ -80,6 +102,10 @@ impl Pak {
             unknown4: input.read_u32::<LE>()?,
             flags: PakFlags(input.read_u32::<LE>()?),
         };
+
+        if header.entry_count >= limits.entry_limit as u32 {
+            return Err(PakError::EntryLimit(header.entry_count, limits.entry_limit));
+        }
         info!("{} entries detected", header.entry_count);
         debug!("Block size is {} bytes", header.block_size);
         debug!("Flag bits {:#032b}", header.flags().0);
@@ -87,6 +113,7 @@ impl Pak {
         let first_offset = header.data_offset() / header.block_size();
 
         // Read some unknown data before the data we want
+        // TODO: This *must* be done differently for real, figure it out!
         let mut unknown_pre_data = Vec::new();
         while input.stream_position()? < header.data_offset() as u64 {
             let unknown = input.read_u32::<LE>()?;
@@ -111,10 +138,7 @@ impl Pak {
         for _ in 0..header.entry_count() {
             let offset = input.read_u32::<LE>().unwrap();
             let length = input.read_u32::<LE>().unwrap();
-            offsets.push(FileLocation {
-                offset,
-                length,
-            });
+            offsets.push(EntryLocation { offset, length });
         }
 
         // Read all unknown_data1
@@ -152,7 +176,11 @@ impl Pak {
         // Read all entry data
         debug!("Creating entry list");
         let mut entries: Vec<Entry> = Vec::new();
-        for (i, offset_info) in offsets.iter().enumerate().take(header.entry_count() as usize) {
+        for (i, offset_info) in offsets
+            .iter()
+            .enumerate()
+            .take(header.entry_count() as usize)
+        {
             debug!("Seeking to block {}", offset_info.offset);
             // Seek to and read the entry data
             input
@@ -209,18 +237,16 @@ impl Pak {
     }
 
     /// Encode a PAK file into a byte stream.
-    pub fn encode<T: Write>(
-        &self,
-        mut output: &mut T
-    ) -> Result<(), PakError> {
+    pub fn encode<T: Write>(&self, mut output: &mut T) -> Result<(), PakError> {
         self.header.write_into(&mut output)?;
 
         // Write unknown data
         output.write_all(
-            &self.unknown_pre_data
+            &self
+                .unknown_pre_data
                 .iter()
                 .flat_map(|dw| dw.to_le_bytes())
-                .collect::<Vec<u8>>()
+                .collect::<Vec<u8>>(),
         )?;
 
         // Write offsets and lengths
@@ -239,15 +265,11 @@ impl Pak {
         // Write names if the flags indicate it should have them
         if self.header.flags().has_names() {
             if let Some(subdir) = &self.subdirectory {
-                output.write_all(
-                    CString::new(subdir.as_bytes()).unwrap().to_bytes_with_nul()
-                )?;
+                output.write_all(CString::new(subdir.as_bytes()).unwrap().to_bytes_with_nul())?;
             }
             for entry in self.entries() {
                 let name = entry.name.as_ref().unwrap();
-                output.write_all(
-                    CString::new(name.as_bytes()).unwrap().to_bytes_with_nul()
-                )?;
+                output.write_all(CString::new(name.as_bytes()).unwrap().to_bytes_with_nul())?;
             }
         }
 
@@ -257,7 +279,11 @@ impl Pak {
 
         for entry in self.entries() {
             //let block_size = entry.data.len().div_ceil(self.header().block_size as usize);
-            let mut remainder = 2048 - entry.data.len().rem_euclid(self.header().block_size as usize);
+            let mut remainder = 2048
+                - entry
+                    .data
+                    .len()
+                    .rem_euclid(self.header().block_size as usize);
             if remainder == 2048 {
                 remainder = 0;
             }
@@ -278,11 +304,7 @@ impl Pak {
     ///
     /// This function updates the offsets of all entries to fit within the
     /// chunk size specified in the header.
-    pub fn replace(
-        &mut self,
-        index: usize,
-        replacement_bytes: &[u8],
-    ) -> Result<(), PakError> {
+    pub fn replace(&mut self, index: usize, replacement_bytes: &[u8]) -> Result<(), PakError> {
         let block_size = self.header().block_size();
 
         let replaced_entry;
@@ -290,7 +312,7 @@ impl Pak {
             replaced_entry = entry
         } else {
             log::error!("Entry {} not found!", index);
-            return Err(PakError::IndexError)
+            return Err(PakError::IndexError);
         };
 
         if let Some(name) = replaced_entry.name() {
@@ -304,8 +326,7 @@ impl Pak {
         replaced_entry.length = replaced_entry.data.len() as u32;
 
         // Get the offset of the next entry based on the current one
-        let mut next_offset =
-            replaced_entry.offset + replaced_entry.length.div_ceil(block_size);
+        let mut next_offset = replaced_entry.offset + replaced_entry.length.div_ceil(block_size);
 
         // Update the position of all subsequent entries
         let mut i = 0;
@@ -333,7 +354,7 @@ impl Pak {
         let index = if let Some(entry) = entry {
             entry.index
         } else {
-            return Err(PakError::IndexError)
+            return Err(PakError::IndexError);
         };
 
         self.replace(index, replacement_bytes)?;
@@ -341,16 +362,12 @@ impl Pak {
         Ok(())
     }
 
-    pub fn replace_by_id(
-        &mut self,
-        id: u32,
-        replacement_bytes: &[u8],
-    ) -> Result<(), PakError> {
+    pub fn replace_by_id(&mut self, id: u32, replacement_bytes: &[u8]) -> Result<(), PakError> {
         let entry = self.get_entry_by_id(id);
         let index = if let Some(entry) = entry {
             entry.index
         } else {
-            return Err(PakError::IndexError)
+            return Err(PakError::IndexError);
         };
 
         self.replace(index, replacement_bytes)?;
@@ -369,16 +386,13 @@ impl Pak {
 
     /// Get an individual entry from the PAK by its ID
     pub fn get_entry_by_id(&mut self, id: u32) -> Option<&mut Entry> {
-        self.entries
-            .get_mut((id - self.header.id_start) as usize)
+        self.entries.get_mut((id - self.header.id_start) as usize)
     }
 
     pub fn get_entry_by_name(&mut self, name: &str) -> Option<&mut Entry> {
         self.entries
             .iter_mut()
-            .find(|e|
-                e.name.as_ref().is_some_and(|n| n == name)
-            )
+            .find(|e| e.name.as_ref().is_some_and(|n| n == name))
     }
 
     /// Get a list of all entries from the PAK
@@ -390,8 +404,7 @@ impl Pak {
     pub fn contains_name(&self, name: &str) -> bool {
         self.entries
             .iter()
-            .any(|e| e.name.as_ref()
-            .is_some_and(|n| n == name))
+            .any(|e| e.name.as_ref().is_some_and(|n| n == name))
     }
 }
 

pak_explorer/Cargo.toml

@@ -10,9 +10,11 @@ authors.workspace = true
 publish = false
 
 [dependencies]
+colog = "1.3.0"
 cz = { path = "../cz/", features = ["png"] }
 eframe = { version = "0.28.1", default-features = false, features = ["wayland", "x11", "accesskit", "default_fonts", "wgpu"] }
 egui_extras = "0.28.1"
+log = "0.4.22"
 luca_pak = { path = "../luca_pak/" }
 rfd = "0.14.1"
 

pak_explorer/src/main.rs

@@ -1,23 +1,26 @@
 #![cfg_attr(not(debug_assertions), windows_subsystem = "windows")] // hide console window on Windows in release
 
+use colog;
+use eframe::egui::{self, ColorImage, Image, TextureFilter, TextureHandle, TextureOptions};
+use log::error;
+use luca_pak::{entry::EntryType, Pak};
 use std::fs;
 
-use eframe::egui::{self, ColorImage, Image, TextureFilter, TextureHandle, TextureOptions};
-use luca_pak::{entry::EntryType, Pak};
-
 fn main() -> eframe::Result {
+    colog::default_builder()
+        .filter(None, log::LevelFilter::Warn)
+        .init();
+
     let options = eframe::NativeOptions {
         viewport: egui::ViewportBuilder::default().with_inner_size([1024.0, 800.0]),
         follow_system_theme: true,
         ..Default::default()
     };
+
     eframe::run_native(
         "LUCA PAK Explorer",
         options,
-        Box::new(|cc| {
-            // This gives us image support:
-            egui_extras::install_image_loaders(&cc.egui_ctx);
-
+        Box::new(|_ctx| {
             Ok(Box::<PakExplorer>::default())
         }),
     )
@@ -48,17 +51,23 @@ impl eframe::App for PakExplorer {
             ui.heading("PAK File Explorer");
 
             ui.horizontal(|ui| {
-                if ui.button("Open file…").clicked() {
+                if ui.button("Open file").clicked() {
                     if let Some(path) = rfd::FileDialog::new().pick_file() {
-                        let pak = Pak::open(&path).unwrap();
-                        self.open_file = Some(pak);
+                        let pak = match Pak::open(&path) {
+                            Ok(pak) => Some(pak),
+                            Err(e) => {
+                                error!("Unable to read selected file as PAK: {}", e);
+                                None
+                            }
+                        };
+                        self.open_file = pak;
                         self.selected_entry = None;
                         self.image_texture = None;
                         self.hex_string = None;
                     }
                 }
                 if let Some(pak) = &self.open_file {
-                    if ui.button("Save PAK…").clicked() {
+                    if ui.button("Save PAK").clicked() {
                         if let Some(path) = rfd::FileDialog::new()
                             .set_file_name(pak.path().file_name().unwrap().to_string_lossy())
                             .save_file()
@@ -72,7 +81,10 @@ impl eframe::App for PakExplorer {
             ui.separator();
 
             if let Some(pak) = &self.open_file {
-                ui.label(format!("Opened {}", pak.path().file_name().unwrap().to_string_lossy()));
+                ui.label(format!(
+                    "Opened {}",
+                    pak.path().file_name().unwrap().to_string_lossy()
+                ));
                 ui.label(format!("Contains {} Entries", pak.entries().len()));
 
                 let selection = if let Some(entry) = &self.selected_entry {
@@ -84,28 +96,28 @@ impl eframe::App for PakExplorer {
                 egui::ComboBox::from_id_source("my-combobox")
                     .selected_text(selection)
                     .truncate()
-                    .show_ui(ui, |ui|
-                {
+                    .show_ui(ui, |ui| {
                         ui.selectable_value(&mut self.selected_entry, None, "");
                         for entry in pak.entries() {
-                        if ui.selectable_value(
+                            if ui
+                                .selectable_value(
                                     &mut self.selected_entry,
                                     Some(entry.clone()),
                                     entry.display_name(),
-                        ).clicked() {
+                                )
+                                .clicked()
+                            {
                                 self.image_texture = None;
                             };
                         }
                     });
             } else {
-                ui.centered_and_justified(|ui|
-                    ui.label("No File Opened")
-                );
+                ui.centered_and_justified(|ui| ui.label("No File Opened"));
             }
 
             if let Some(entry) = &self.selected_entry {
                 ui.horizontal(|ui| {
-                    if ui.button("Save entry…").clicked() {
+                    if ui.button("Save entry").clicked() {
                         if let Some(path) = rfd::FileDialog::new()
                             .set_file_name(entry.display_name())
                             .save_file()
@@ -115,7 +127,7 @@ impl eframe::App for PakExplorer {
                     }
 
                     if let Some(pak) = &mut self.open_file.as_mut() {
-                        if ui.button("Replace entry…").clicked() {
+                        if ui.button("Replace entry").clicked() {
                             if let Some(path) = rfd::FileDialog::new().pick_file() {
                                 let file_bytes = fs::read(path).unwrap();
                                 pak.replace(entry.index(), &file_bytes).unwrap();
@@ -124,18 +136,23 @@ impl eframe::App for PakExplorer {
                     }
                 });
                 match entry.file_type() {
-                    EntryType::CZ0 | EntryType::CZ1
-                        | EntryType::CZ2 | EntryType::CZ3
-                        | EntryType::CZ4 | EntryType::CZ5 =>
-                    {
-                        if ui.button("Save as PNG…").clicked() {
+                    EntryType::CZ0
+                    | EntryType::CZ1
+                    | EntryType::CZ2
+                    | EntryType::CZ3
+                    | EntryType::CZ4
+                    | EntryType::CZ5 => {
+                        if ui.button("Save as PNG").clicked() {
                             let mut display_name = entry.display_name();
                             display_name.push_str(".png");
                             if let Some(path) = rfd::FileDialog::new()
                                 .set_file_name(display_name)
                                 .save_file()
                             {
-                                let cz = cz::DynamicCz::decode(&mut std::io::Cursor::new(entry.as_bytes())).unwrap();
+                                let cz = cz::DynamicCz::decode(&mut std::io::Cursor::new(
+                                    entry.as_bytes(),
+                                ))
+                                .unwrap();
                                 cz.save_as_png(&path).unwrap();
                             }
                         }
@@ -143,37 +160,39 @@ impl eframe::App for PakExplorer {
                         ui.separator();
 
                         let texture: &TextureHandle = self.image_texture.get_or_insert_with(|| {
-                            let cz = cz::DynamicCz::decode(&mut std::io::Cursor::new(entry.as_bytes())).unwrap();
+                            let cz =
+                                cz::DynamicCz::decode(&mut std::io::Cursor::new(entry.as_bytes()))
+                                    .unwrap();
                             let image = ColorImage::from_rgba_unmultiplied(
                                 [cz.header().width() as usize, cz.header().height() as usize],
-                                cz.as_raw()
+                                cz.as_raw(),
                             );
-                            ui.ctx().load_texture("eventframe", image, TextureOptions {
+                            ui.ctx().load_texture(
+                                "eventframe",
+                                image,
+                                TextureOptions {
                                     magnification: TextureFilter::Nearest,
                                     minification: TextureFilter::Linear,
                                     ..Default::default()
-                            })
+                                },
+                            )
                         });
 
-                        ui.centered_and_justified(|ui|
+                        ui.centered_and_justified(|ui| {
                             ui.add(
                                 Image::from_texture(texture)
                                     .show_loading_spinner(true)
                                     .shrink_to_fit()
-                                    .rounding(2.0)
+                                    .rounding(2.0),
                             )
-                        );
+                        });
                     }
                     _ => {
-                        ui.centered_and_justified(|ui|
-                            ui.label("No Preview Available")
-                        );
-                    },
+                        ui.centered_and_justified(|ui| ui.label("No Preview Available"));
+                    }
                 }
             } else if self.open_file.is_some() {
-                ui.centered_and_justified(|ui|
-                    ui.label("Select an Entry")
-                );
+                ui.centered_and_justified(|ui| ui.label("Select an Entry"));
             }
         });
     }

utils/src/bin/czutil.rs

@@ -1,5 +1,8 @@
 use clap::{error::ErrorKind, Error, Parser, Subcommand};
-use std::{fs, path::{Path, PathBuf}};
+use std::{
+    fs,
+    path::{Path, PathBuf},
+};
 
 /// Utility to maniuplate CZ image files from the LUCA System game engine by
 /// Prototype Ltd.

utils/src/bin/pakutil.rs

@@ -1,6 +1,9 @@
-use std::{fs, path::PathBuf};
-use clap::{error::{Error, ErrorKind}, Parser, Subcommand};
+use clap::{
+    error::{Error, ErrorKind},
+    Parser, Subcommand,
+};
 use luca_pak::Pak;
+use std::{fs, path::PathBuf};
 
 /// Utility to maniuplate PAK archive files from the LUCA System game engine by
 /// Prototype Ltd.
@@ -58,7 +61,7 @@ fn main() {
 
     let mut pak = match Pak::open(&cli.input) {
         Ok(pak) => pak,
-        Err(err) => fmt_error(&format!("Could not open PAK file: {}", err)).exit()
+        Err(err) => fmt_error(&format!("Could not open PAK file: {}", err)).exit(),
     };
 
     match cli.command {
@@ -74,8 +77,14 @@ fn main() {
                 outpath.push(entry.display_name());
                 entry.save(&outpath).unwrap();
             }
-        },
-        Commands::Replace { batch, name, id, replacement, output } => {
+        }
+        Commands::Replace {
+            batch,
+            name,
+            id,
+            replacement,
+            output,
+        } => {
             if id.is_some() && name.is_some() {
                 fmt_error("Cannot use ID and name together").exit()
             }
@@ -90,12 +99,8 @@ fn main() {
 
                 for entry in fs::read_dir(replacement).unwrap() {
                     let entry = entry.unwrap();
-                    let search_name: String = entry
-                            .path()
-                            .file_name()
-                            .unwrap()
-                            .to_string_lossy()
-                            .into();
+                    let search_name: String =
+                        entry.path().file_name().unwrap().to_string_lossy().into();
 
                     let parsed_id: Option<u32> = search_name.parse().ok();
 
@@ -104,9 +109,15 @@ fn main() {
 
                     // Try replacing by name, if that fails, replace by parsed ID
                     if pak.replace_by_name(search_name, &rep_data).is_err() {
-                        fmt_error("Could not replace entry in PAK: Could not find name").print().unwrap()
-                    } else if parsed_id.is_some() && pak.replace_by_id(parsed_id.unwrap(), &rep_data).is_err() {
-                        fmt_error("Could not replace entry in PAK: ID is invalid").print().unwrap()
+                        fmt_error("Could not replace entry in PAK: Could not find name")
+                            .print()
+                            .unwrap()
+                    } else if parsed_id.is_some()
+                        && pak.replace_by_id(parsed_id.unwrap(), &rep_data).is_err()
+                    {
+                        fmt_error("Could not replace entry in PAK: ID is invalid")
+                            .print()
+                            .unwrap()
                     }
                 }
             } else {
@@ -117,11 +128,7 @@ fn main() {
                 let search_name = if let Some(name) = name {
                     name
                 } else {
-                    replacement
-                        .file_name()
-                        .unwrap()
-                        .to_string_lossy()
-                        .into()
+                    replacement.file_name().unwrap().to_string_lossy().into()
                 };
 
                 let search_id = if id.is_some() {
@@ -152,8 +159,5 @@ fn main() {
 
 #[inline(always)]
 fn fmt_error(message: &str) -> Error {
-    Error::raw(
-        ErrorKind::ValueValidation,
-        format!("{}\n", message),
-    )
+    Error::raw(ErrorKind::ValueValidation, format!("{}\n", message))
 }