diff options
Diffstat (limited to 'example/ida/depgraph.py')
| -rw-r--r-- | example/ida/depgraph.py | 5 |
1 files changed, 4 insertions, 1 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): |