summaryrefslogtreecommitdiff
path: root/plugins/warp/src/plugin
diff options
context:
space:
mode:
Diffstat (limited to 'plugins/warp/src/plugin')
-rw-r--r--plugins/warp/src/plugin/ffi.rs10
-rw-r--r--plugins/warp/src/plugin/render_layer.rs6
2 files changed, 13 insertions, 3 deletions
diff --git a/plugins/warp/src/plugin/ffi.rs b/plugins/warp/src/plugin/ffi.rs
index c3a8051c..2826ed34 100644
--- a/plugins/warp/src/plugin/ffi.rs
+++ b/plugins/warp/src/plugin/ffi.rs
@@ -142,7 +142,10 @@ pub unsafe extern "C" fn BNWARPIsLiftedInstructionVariant(
unsafe { LowLevelILFunction::from_raw(analysis_function) };
match lifted_il.instruction_from_index(index) {
Some(instr) => {
- let relocatable_regions = relocatable_regions(&lifted_il.function().view());
+ let Some(owner_function) = lifted_il.function() else {
+ return false;
+ };
+ let relocatable_regions = relocatable_regions(&owner_function.view());
is_variant_instruction(&relocatable_regions, &instr)
}
None => false,
@@ -158,7 +161,10 @@ pub unsafe extern "C" fn BNWARPIsLowLevelInstructionComputedVariant(
unsafe { LowLevelILFunction::from_raw(analysis_function) };
match llil.instruction_from_index(index) {
Some(instr) => {
- let relocatable_regions = relocatable_regions(&llil.function().view());
+ let Some(owner_function) = llil.function() else {
+ return false;
+ };
+ let relocatable_regions = relocatable_regions(&owner_function.view());
is_computed_variant_instruction(&relocatable_regions, &instr)
}
None => false,
diff --git a/plugins/warp/src/plugin/render_layer.rs b/plugins/warp/src/plugin/render_layer.rs
index 6c446e6f..3417c9b1 100644
--- a/plugins/warp/src/plugin/render_layer.rs
+++ b/plugins/warp/src/plugin/render_layer.rs
@@ -48,7 +48,11 @@ impl HighlightRenderLayer {
llil: &LowLevelILRegularFunction,
lines: &mut [DisassemblyTextLine],
) {
- let relocatable_regions = relocatable_regions(&lifted_il.function().view());
+ let Some(owner_function) = lifted_il.function() else {
+ return;
+ };
+
+ let relocatable_regions = relocatable_regions(&owner_function.view());
for line in lines {
// We use address here instead of index since it's more reliable for other IL's.
for lifted_il_instr in filtered_instructions_at(lifted_il, line.address) {