benchmark.py: Fix type cast

mborgerson · mborgerson · commit 65099eed54d7 · 2025-04-15T18:49:29.000-07:00
diff --git a/scripts/benchmark.py b/scripts/benchmark.py
@@ -108,7 +108,7 @@ def get_blocks(binary_path: str) -> List[Block]:
         cfg = p.analyses[angr.analyses.CFGFast].prep(show_progressbar=True)(
             resolve_indirect_jumps=False, force_smart_scan=False
         )
-        for n in cast(Iterable[angr.analyses.cfg.CFGNode], cfg.model.nodes()):
+        for n in cast(Iterable[angr.knowledge_plugins.cfg.cfg_model.CFGNode], cfg.model.nodes()):
             if n.byte_string:
                 blocks.append(Block(n.addr, n.byte_string))
         log.info("Saving blocks to file '%s' for subsequent benchmarks...", blocks_file_path)

Original file line number	Diff line number	Diff line change
`@@ -108,7 +108,7 @@ def get_blocks(binary_path: str) -> List[Block]:`
`108`	`108`	`cfg = p.analyses[angr.analyses.CFGFast].prep(show_progressbar=True)(`
`109`	`109`	`resolve_indirect_jumps=False, force_smart_scan=False`
`110`	`110`	`)`
`111`		`- for n in cast(Iterable[angr.analyses.cfg.CFGNode], cfg.model.nodes()):`
	`111`	`+ for n in cast(Iterable[angr.knowledge_plugins.cfg.cfg_model.CFGNode], cfg.model.nodes()):`
`112`	`112`	`if n.byte_string:`
`113`	`113`	`blocks.append(Block(n.addr, n.byte_string))`
`114`	`114`	`log.info("Saving blocks to file '%s' for subsequent benchmarks...", blocks_file_path)`