Analyzer fix for fallthrough switch cases

Fixes #23
encounter · Jan 7, 2024 · 59fe806 · 59fe806
1 parent ee3b297
commit 59fe806
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/analysis/slices.rs b/src/analysis/slices.rs
@@ -248,7 +248,9 @@ impl FunctionSlices {
             StepResult::Continue | StepResult::LoadStore { .. } => {
                 let next_address = ins_addr + 4;
                 // If we already visited the next address, connect the blocks and end
-                if executor.visited(section.address as u32, next_address) {
+                if executor.visited(section.address as u32, next_address)
+                    || self.blocks.contains_key(&next_address)
+                {
                     self.blocks.insert(block_start, Some(next_address));
                     self.branches.insert(ins_addr, vec![next_address]);
                     Ok(ExecCbResult::EndBlock)