use std::io::Cursor;
use winreg_core::hive::Hive;
use winreg_core::key::filetime_to_datetime;
#[derive(Debug, Clone, serde::Serialize)]
pub struct AmcacheEntry {
pub file_path: String,
pub sha1: String,
pub size: u64,
pub link_date: Option<String>,
pub publisher: String,
pub product_name: String,
pub product_version: String,
pub bin_file_version: String,
pub key_name: String,
pub last_written: Option<String>,
}
const INVENTORY_APP_FILE: &str = "Root\\InventoryApplicationFile";
pub fn parse(hive: &Hive<Cursor<Vec<u8>>>) -> Vec<AmcacheEntry> {
let container = match hive.open_key(INVENTORY_APP_FILE) {
Ok(Some(k)) => k,
_ => return Vec::new(),
};
let subkeys = match container.subkeys() {
Ok(s) => s,
Err(_) => return Vec::new(),
};
let mut entries = Vec::with_capacity(subkeys.len());
for subkey in subkeys {
let key_name = subkey.name();
let last_written = filetime_to_datetime(subkey.last_written_raw())
.map(|dt| dt.format("%Y-%m-%dT%H:%M:%SZ").to_string());
let read_sz = |name: &str| -> String {
subkey
.value(name)
.ok()
.flatten()
.and_then(|v| v.as_string().ok())
.unwrap_or_default()
};
let file_path = read_sz("LowerCaseLongPath");
let file_id_raw = read_sz("FileId");
let sha1 = if file_id_raw.starts_with("0000") {
file_id_raw[4..].to_string()
} else {
file_id_raw
};
let size = subkey
.value("Size")
.ok()
.flatten()
.and_then(|v| v.as_u32().ok())
.unwrap_or(0) as u64;
let link_date_raw = read_sz("LinkDate");
let link_date = if link_date_raw.is_empty() {
None
} else {
Some(link_date_raw)
};
let publisher = read_sz("Publisher");
let product_name = read_sz("ProductName");
let product_version = read_sz("ProductVersion");
let bin_file_version = read_sz("BinFileVersion");
entries.push(AmcacheEntry {
file_path,
sha1,
size,
link_date,
publisher,
product_name,
product_version,
bin_file_version,
key_name,
last_written,
});
}
entries
}