diff options
Diffstat (limited to 'example')
| -rw-r--r-- | example/ida/depgraph.py | 5 | ||||
| -rw-r--r-- | example/symbol_exec/depgraph.py | 2 |
2 files changed, 5 insertions, 2 deletions
diff --git a/example/ida/depgraph.py b/example/ida/depgraph.py index 1784b4e4..eb663986 100644 --- a/example/ida/depgraph.py +++ b/example/ida/depgraph.py @@ -92,7 +92,10 @@ Method to use: @property def elements(self): value = self.cbReg.value - return set([ir_arch.arch.regs.all_regs_ids_byname[value]]) + reg = ir_arch.arch.regs.all_regs_ids_byname.get(value, None) + if reg is None: + raise ValueError("Unknown element '%s'!" % value) + return set([reg]) @property def depgraph(self): diff --git a/example/symbol_exec/depgraph.py b/example/symbol_exec/depgraph.py index 48758ad0..b4f793c0 100644 --- a/example/symbol_exec/depgraph.py +++ b/example/symbol_exec/depgraph.py @@ -42,7 +42,7 @@ elements = set() regs = machine.mn.regs.all_regs_ids_byname for element in args.element: try: - elements.add(regs[element.upper()]) + elements.add(regs[element]) except KeyError: raise ValueError("Unknown element '%s'" % element) |