From ae22a569d4da74d58bbd26bb254086b486a7a69a Mon Sep 17 00:00:00 2001
From: Fabrice Desclaux <fabrice.desclaux@cea.fr>
Date: Tue, 22 Mar 2016 08:21:43 +0100
Subject: Example/Depgraph: allow lower/upper register tracking

---
 example/ida/depgraph.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

(limited to 'example/ida')

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):
-- 
cgit 1.4.1