unspeaker/vestal
1
0
Fork 0
mirror of https://codeberg.org/unspeaker/vestal.git synced 2025-12-06 17:16:42 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

count imports

Browse source
This commit is contained in:
🪞👃🪞 2025-02-21 22:32:27 +02:00
parent 1997297d7b
commit 5188bc5581
1 changed files with 14 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

29
crates/vestal/src/main.rs
View file

@ -81,7 +81,7 @@ impl Dll {
&mut calls_by_target,
false
)?;
let imports = Self::imports(&pe);
let (modules_count, methods_count) = Self::imports(&pe)?;
let dll = Arc::new(Self {
name: name.clone(),
path: path.clone(),
@ -92,6 +92,8 @@ impl Dll {
calls_by_source,
calls_by_target,
});
println!(" (deps-modules {modules_count})");
println!(" (deps-methods {methods_count})");
println!(" (call-sites {calls})");
for (call, sites) in dll.calls_by_target.iter() {
println!(" (0x{call:08x}\n {:?})", sites.iter()
@ -103,27 +105,23 @@ impl Dll {
}
fn imports (pe: &VecPE) -> Usually<(usize, usize)> {
let directory = ImportDirectory::parse(pe)?;
let mut total = 0;
for descriptor in directory.descriptors {
let dep = descriptor.get_name(pe)?.as_str()?;
let dep = descriptor.get_name(pe)?.as_str()?.to_lowercase();
let imp = descriptor.get_imports(pe)?;
let iat = descriptor.get_first_thunk(pe)?;
let ilt = descriptor.get_original_first_thunk(pe)?;
let lut = descriptor.get_lookup_thunks(pe)?;
let mut imports = Vec::new();
let unwrap_thunk = |thunk: &Thunk, name|match thunk {
Thunk::Thunk32(t) => panic!("32 bit {name}"),
Thunk::Thunk64(t) => t.0
};
for (index, (import, thunk, orig, lookup)) in izip!(
imp,
iat.iter().map(|thunk|format!("0x{:08x}", match thunk {
Thunk::Thunk32(t) => panic!("32 bit thunk"),
Thunk::Thunk64(t) => t.0
})),
ilt.iter().map(|thunk|format!("0x{:08x}", match thunk {
Thunk::Thunk32(t) => panic!("32 bit original thunk"),
Thunk::Thunk64(t) => t.0
})),
lut.iter().map(|thunk|format!("0x{:08x}", match thunk {
Thunk::Thunk32(t) => panic!("32 bit original thunk"),
Thunk::Thunk64(t) => t.0
})),
iat.iter().map(|thunk|format!("0x{:08x}", unwrap_thunk(thunk, "IAT thunk"))),
ilt.iter().map(|thunk|format!("0x{:08x}", unwrap_thunk(thunk, "ILT (orig) thunk"))),
lut.iter().map(|thunk|format!("0x{:08x}", unwrap_thunk(thunk, "lookup thunk"))),
).enumerate() {
let call_via = descriptor.first_thunk.0 + index as u32 * 8;
let name = match import {
@ -140,6 +138,7 @@ impl Dll {
};
println!(" ({index:5} 0x{call_via:08x} {dep:>20} {name}");
imports.push((thunk, orig, import));
total += 1;
//if let Some(existing) = self.addr_to_import.get(&call_via) {
//panic!("addr space overlap at 0x{call_via:x}: {}::{} vs {}::{}",
//existing.0,
@ -150,7 +149,7 @@ impl Dll {
//self.addr_to_import.insert(call_via, (dep.to_string(), name));
}
}
Ok((0, 0))
Ok((0, total))
}
fn calls (
name: &Arc<str>,