134 lines
4.6 KiB
Rust
134 lines
4.6 KiB
Rust
use androscalpel_serializer::Serializable;
|
|
|
|
pub mod apk_signing_block;
|
|
pub mod compression;
|
|
mod cp437;
|
|
pub mod end_of_central_directory;
|
|
pub mod error;
|
|
pub mod extra_fields;
|
|
pub mod file_header;
|
|
pub mod local_file_header;
|
|
pub mod zip_reader;
|
|
pub mod zip_writer;
|
|
|
|
use extra_fields::{ExtraField, Zip64ExtraField};
|
|
use file_header::FileHeader;
|
|
use local_file_header::LocalFileHeader;
|
|
pub use zip_reader::ZipFileReader;
|
|
pub use zip_writer::ZipFileWriter;
|
|
|
|
#[derive(Debug, Clone, PartialEq, Eq, Serializable, Default)]
|
|
pub struct Signature(pub u32);
|
|
|
|
pub enum Encoding {
|
|
CP437,
|
|
UTF8,
|
|
}
|
|
|
|
pub mod general_purpose_flags {
|
|
pub const MASK_ENCRYPTED: u16 = 1 << 0;
|
|
pub const MASK_COMPRESS_OPTION_1: u16 = 1 << 1;
|
|
pub const MASK_COMPRESS_OPTION_2: u16 = 1 << 2;
|
|
pub const MASK_USE_DATA_DESCRIPTOR: u16 = 1 << 3;
|
|
pub const MASK_STRONG_ENCRYPTION: u16 = 1 << 6;
|
|
pub const MASK_UTF8_FILENAME: u16 = 1 << 11;
|
|
pub const MASK_ENCRYPTED_CENTRAL_DIR: u16 = 1 << 13;
|
|
}
|
|
|
|
pub mod external_file_attributes {
|
|
pub const FIFO_FILE: u32 = 0b00010000 << 24;
|
|
// #define S_IFCHR 0020000 /* character special */ ?
|
|
pub const DIR_FILE: u32 = 0b01000000 << 24;
|
|
// #define S_IFBLK 0060000 /* block special */ ?
|
|
pub const SYMBOLIC_LINK_FILE: u32 = 0b10100000 << 24;
|
|
pub const SOCKET_FILE: u32 = 0b11000000 << 24;
|
|
pub const REGULAR_FILE: u32 = 0b10000000 << 24;
|
|
pub const MASK_FILE_TYPE: u32 = 0b11110000 << 24;
|
|
pub const SETUID: u32 = 0b0000100000000000 << 16;
|
|
pub const SETGID: u32 = 0b0000010000000000 << 16;
|
|
pub const STICKY: u32 = 0b0000001000000000 << 16;
|
|
pub const PERM_UR: u32 = 0b0000000100000000 << 16;
|
|
pub const PERM_UW: u32 = 0b0000000010000000 << 16;
|
|
pub const PERM_UX: u32 = 0b0000000001000000 << 16;
|
|
pub const PERM_GR: u32 = 0b0000000000100000 << 16;
|
|
pub const PERM_GW: u32 = 0b0000000000010000 << 16;
|
|
pub const PERM_GX: u32 = 0b0000000000001000 << 16;
|
|
pub const PERM_OR: u32 = 0b0000000000000100 << 16;
|
|
pub const PERM_OW: u32 = 0b0000000000000010 << 16;
|
|
pub const PERM_OX: u32 = 0b0000000000000001 << 16;
|
|
}
|
|
|
|
#[derive(Debug, Clone, PartialEq, Eq)]
|
|
pub struct FileInfo {
|
|
pub local_header: LocalFileHeader,
|
|
pub header: FileHeader,
|
|
}
|
|
// TODO: support data descriptor (MASK_USE_DATA_DESCRIPTOR)
|
|
|
|
impl FileInfo {
|
|
pub fn get_name(&self) -> String {
|
|
self.header.get_name()
|
|
}
|
|
pub fn set_name(&mut self, name: &str) {
|
|
self.header.set_name(name);
|
|
self.local_header.set_name(name);
|
|
}
|
|
pub fn external_file_attributes_set_flag(&mut self, flag: u32) {
|
|
self.header.external_file_attributes_set_flag(flag);
|
|
}
|
|
pub fn external_file_attributes_unset_flag(&mut self, flag: u32) {
|
|
self.header.external_file_attributes_unset_flag(flag);
|
|
}
|
|
pub fn set_file_type(&mut self, file_type: u32) {
|
|
self.header.set_file_type(file_type);
|
|
}
|
|
pub fn get_offset_local_header(&self) -> u64 {
|
|
self.header.get_offset_local_header()
|
|
}
|
|
pub fn get_compressed_size(&self) -> u64 {
|
|
self.header.get_compressed_size()
|
|
}
|
|
pub fn get_file_offset(&self) -> u64 {
|
|
self.get_offset_local_header() + self.local_header.size() as u64
|
|
}
|
|
pub fn set_offset_local_header(&mut self, new_offset: u64) {
|
|
let use_z64 = new_offset > u32::MAX as u64;
|
|
for extra in &mut self.local_header.extra_field {
|
|
if let ExtraField::Zip64(extra) = extra {
|
|
if use_z64 {
|
|
extra.offset_header = Some(new_offset);
|
|
} else {
|
|
extra.offset_header = None;
|
|
}
|
|
}
|
|
}
|
|
let mut z64found = false;
|
|
for extra in &mut self.header.extra_field {
|
|
if let ExtraField::Zip64(extra) = extra {
|
|
if use_z64 {
|
|
extra.offset_header = Some(new_offset);
|
|
} else {
|
|
extra.offset_header = None;
|
|
}
|
|
z64found = true;
|
|
}
|
|
}
|
|
if use_z64 && !z64found {
|
|
self.header
|
|
.extra_field
|
|
.push(ExtraField::Zip64(Zip64ExtraField {
|
|
original_size: Some(self.header.uncompressed_size as u64),
|
|
compressed_size: Some(self.header.compressed_size as u64),
|
|
offset_header: Some(new_offset),
|
|
disk_number: None,
|
|
}));
|
|
self.header.uncompressed_size = u32::MAX;
|
|
self.header.compressed_size = u32::MAX;
|
|
}
|
|
if use_z64 {
|
|
self.header.offset_local_header = u32::MAX;
|
|
} else {
|
|
self.header.offset_local_header = new_offset as u32;
|
|
}
|
|
}
|
|
}
|