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
99
100
101
102
103
#![cfg_attr(not(feature = "std"), no_std)]
#![feature(alloc_prelude)]
#[cfg(feature = "std")]
#[allow(unused_imports)]
#[macro_use]
extern crate std;
#[cfg(feature = "std")]
#[allow(unused_imports)]
use std::prelude::v1 as alloc_prelude;
#[cfg(not(feature = "std"))]
#[allow(unused_imports)]
#[macro_use]
extern crate alloc;
#[cfg(not(feature = "std"))]
#[allow(unused_imports)]
use alloc::prelude::v1 as alloc_prelude;
use self::alloc_prelude::*;
pub mod header;
#[cfg(test)]
mod tests;
use self::header::{PAK_HEADER_LENGTH, PAK_FILE_LENGTH, PakHeader, PakFileEntry};
pub struct PakArchive<'a> {
pub files: Vec<(String, &'a [u8])>,
}
impl<'a> PakArchive<'a> {
pub fn new() -> Self {
Self {
files: Vec::new(),
}
}
pub fn add_file(&mut self, file_name: String, file_data: &'a [u8]) {
self.files.push((file_name, file_data));
}
pub fn from_bytes(data: &'a [u8]) -> Result<Self, ()> {
let header = PakHeader::from_bytes(data)?;
let mut files: Vec<(String, &'a [u8])> = Vec::new();
let mut offset: usize = header.table_offset as usize;
while offset < data.len() {
let file_entry = PakFileEntry::from_bytes(&data[offset..]);
let file_offset = file_entry.offset as usize;
let file_size = file_entry.size as usize;
let file_data = &data[file_offset..(file_offset + file_size)];
files.push((file_entry.get_name(), file_data));
offset += PAK_FILE_LENGTH as usize;
}
Ok(Self {
files,
})
}
pub fn as_bytes(&self) -> Vec<u8> {
let mut offset: u32 = PAK_HEADER_LENGTH;
let mut output: Vec<u8> = Vec::new();
let mut file_table: Vec<PakFileEntry> = Vec::new();
for (fname, fdata) in self.files.iter() {
let flength: u32 = (fdata.len() & 0xFFFFFFFF) as u32;
file_table.push(PakFileEntry::new(fname.clone(), offset, flength));
output.extend(&fdata[..]);
offset += flength;
}
let ftable_offset = offset;
for ftable_entry in file_table.iter() {
let tdata = ftable_entry.as_bytes();
output.extend(&tdata[..]);
offset += (tdata.len() & 0xFFFFFFFF) as u32
}
let header = PakHeader::new(ftable_offset, offset - ftable_offset);
Vec::from([header.as_bytes(), output].concat())
}
}