use crate::signatures::common::{SignatureError, SignatureResult, CONFIDENCE_MEDIUM};
use crate::structures::pe::parse_pe_header;
pub const DESCRIPTION: &str = "Windows PE binary";
pub fn pe_magic() -> Vec<Vec<u8>> {
vec![
b"\x4d\x5a\x90\x00\x03\x00\x00\x00\x04\x00\x00\x00\xff\xff\x00\x00".to_vec(),
b"\x4d\x5a\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00".to_vec(),
]
}
pub fn pe_parser(file_data: &[u8], offset: usize) -> Result<SignatureResult, SignatureError> {
let mut result = SignatureResult {
offset,
description: DESCRIPTION.to_string(),
confidence: CONFIDENCE_MEDIUM,
..Default::default()
};
if let Ok(pe_header) = parse_pe_header(&file_data[offset..]) {
result.description = format!(
"{}, machine type: {}",
result.description, pe_header.machine
);
return Ok(result);
}
Err(SignatureError)
}