Add .set_label_first() to only set label if none exists

bsdphk · Sep 2, 2022 · 08cebe2 · 08cebe2
1 parent 0450d93
commit 08cebe2
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/pyreveng/mem.py b/pyreveng/mem.py
@@ -159,6 +159,13 @@ def set_label(self, adr, lbl):
         assert isinstance(lbl, str)
         self.lbl_d.setdefault(adr, []).append(lbl)
 
+    def set_first_label(self, adr, lbl):
+        ''' Set label, if none exists already '''
+        assert isinstance(lbl, str)
+        i = self.lbl_d.get(adr)
+        if not i:
+            self.lbl_d.setdefault(adr, []).append(lbl)
+
     def get_labels(self, adr):
         i = self.lbl_d.get(adr)
         if i:
@@ -380,6 +387,9 @@ def get_all_somethings(self, what):
                 if low <= aa < high:
                     yield aa, b
 
+    def set_first_label(self, *args):
+        self.set_something("set_first_label", *args)
+
     def set_label(self, *args):
         self.set_something("set_label", *args)