moved files from far-rs

This commit is contained in:
husky 2022-04-10 02:16:19 -07:00
commit 71dff76d4d
No known key found for this signature in database
GPG key ID: E00F7FA3381F0647
4 changed files with 200 additions and 0 deletions

2
.gitignore vendored Normal file
View file

@ -0,0 +1,2 @@
/target
Cargo.lock

9
Cargo.toml Normal file
View file

@ -0,0 +1,9 @@
[package]
name = "libfar"
version = "0.1.0"
edition = "2021"
authors = ["husky (fekhesk)"]
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
[dependencies]

188
src/farlib.rs Normal file
View file

@ -0,0 +1,188 @@
use std::io::{BufReader, Read};
pub struct FarFileInfo {
pub name: String,
pub size: u32,
offset: u32,
}
pub struct FarFile {
pub name: String,
pub size: u32,
pub data: Vec<u8>,
}
pub struct FarArchive {
pub version: u32,
pub file_count: u32,
pub file_list: Vec<FarFileInfo>,
pub file_data: Vec<FarFile>,
}
impl FarFile {
pub fn new_from_archive(name : String, size : u32, offset : u32, original_file : &Vec<u8>) -> FarFile {
let mut reader = BufReader::new(&original_file[offset as usize..(offset + size) as usize]);
let mut data = Vec::new();
reader.read_to_end(&mut data).expect("Failed to read file data");
FarFile {
name,
size,
data,
}
}
pub fn new_from_file(name : String, size : u32, data : Vec<u8>) -> FarFile {
FarFile {
name,
size,
data,
}
}
}
impl FarArchive {
pub fn new_from_files(files : Vec<FarFile>) -> FarArchive {
let mut file_list = Vec::new();
let mut file_data = Vec::new();
let mut offset = 0;
for file in files {
offset += &file.size;
file_list.push(FarFileInfo {
name: file.name.clone(),
size: file.size,
offset,
});
file_data.push(file);
}
FarArchive {
version: 1,
file_count: file_list.len() as u32,
file_list,
file_data,
}
}
pub fn load_file_data(self, original_file : &Vec<u8>) -> FarArchive {
let mut new_file_data = Vec::new();
for i in 0..self.file_list.len() {
new_file_data.push(FarFile::new_from_archive(
self.file_list[i].name.clone(),
self.file_list[i].size,
self.file_list[i].offset,
original_file,
));
}
FarArchive {
version: self.version,
file_count: self.file_count,
file_list: self.file_list,
file_data: new_file_data,
}
}
pub fn to_vec(self) -> Vec<u8> {
// write header
let mut header = Vec::new();
for c in "FAR!byAZ".chars() {
header.push(c as u8);
}
header.extend(&self.version.to_le_bytes());
// wait to write manifest offset until calculated later
// write file data
let mut file_data = Vec::new(); // actual data to be written to file
let mut file_list = Vec::new(); // file list used for making manifest later on
let mut bytes_written = 16; // where we should start putting files
for i in 0..self.file_data.len() {
let mut file_data_bytes = Vec::new();
file_data_bytes.extend_from_slice(&self.file_data[i].data);
file_data.extend_from_slice(&file_data_bytes);
file_list.push(FarFileInfo {
name: self.file_data[i].name.clone(),
size: self.file_data[i].size,
offset: bytes_written,
});
bytes_written += self.file_data[i].size;
}
// write manifest
let mut manifest = Vec::new();
// write file count
manifest.extend_from_slice(&self.file_count.to_le_bytes());
// for each file, write (size, size, offset, name length, name)
for i in 0..self.file_list.len() {
manifest.extend_from_slice(&file_list[i].size.to_le_bytes());
manifest.extend_from_slice(&file_list[i].size.to_le_bytes());
manifest.extend_from_slice(&file_list[i].offset.to_le_bytes());
manifest.extend_from_slice(&(file_list[i].name.len() as u32).to_le_bytes());
manifest.extend_from_slice(&file_list[i].name.as_bytes());
}
// write manifest offset
let manifest_offset = bytes_written;
header.extend_from_slice(&manifest_offset.to_le_bytes());
// join vecs together
let mut output = Vec::new();
output.extend_from_slice(&header);
output.extend_from_slice(&file_data);
output.extend_from_slice(&manifest);
output
}
}
// test for FAR!byAZ
pub fn test(file : &Vec<u8>) -> Result<FarArchive, String> {
let mut reader = BufReader::new(&file[..]);
let mut magic = [0u8; 8];
reader.read_exact(&mut magic).unwrap();
if magic != *b"FAR!byAZ" {
return Err("Not a Far archive".to_string());
}
let mut version = [0; 4];
reader.read_exact(&mut version).unwrap();
let version = u32::from_le_bytes(version);
// get list of files
let files = list_files(file).expect("Failed to list files");
Ok(FarArchive {
version,
file_count: files.len() as u32,
file_list: files,
file_data: vec![],
})
}
pub fn list_files(file : &Vec<u8>) -> Result<Vec<FarFileInfo>, String> {
// manifest offset is at 12 bytes (u32)
let mut reader = BufReader::new(&file[12..]);
let mut offset = [0u8; 4];
reader.read_exact(&mut offset).unwrap();
let offset = u32::from_le_bytes(offset);
// move to manifest
reader = BufReader::new(&file[offset as usize..]);
// read u32 for number of files
let mut num_files = [0u8; 4];
reader.read_exact(&mut num_files).unwrap();
let num_files = u32::from_le_bytes(num_files);
// for each file, read u32 for size, u32 for size again (stored twice for some reason), u32 for offset, u32 for name length, name
let mut files = Vec::new();
for i in 0..num_files {
let mut size = [0u8; 4];
reader.read_exact(&mut size).expect(format!("Failed to read size for file {}", i).as_str());
let size = u32::from_le_bytes(size);
let mut size2 = [0u8; 4];
reader.read_exact(&mut size2).expect(format!("Failed to read size for file {}", i).as_str());
let _size2 = u32::from_le_bytes(size2); // why is this stored twice? f*** you EA
let mut offset = [0u8; 4];
reader.read_exact(&mut offset).expect(format!("Failed to read offset for file {}", i).as_str());
let offset = u32::from_le_bytes(offset);
let mut name_len = [0u8; 4];
reader.read_exact(&mut name_len).expect(format!("Failed to read name length for file {}", i).as_str());
let name_len = u32::from_le_bytes(name_len);
let mut name = vec![0u8; name_len as usize];
reader.read_exact(&mut name).expect(format!("Failed to read name for file {}", i).as_str());
files.push(FarFileInfo {
name: String::from_utf8(name).unwrap(),
size,
offset,
});
}
Ok(files)
}

1
src/lib.rs Normal file
View file

@ -0,0 +1 @@
pub mod farlib;