Implement __contains__ for reg module (#1504)

3 years ago · 5767fe30cb
parent 65462f63dd
commit 5767fe30cb
1 changed files with 4 additions and 0 deletions
--- a/pwndbg/gdblib/regs.py
+++ b/pwndbg/gdblib/regs.py
@ -94,6 +94,10 @@ class module(ModuleType):

        return item

+    def __contains__(self, reg):
+        regs = set(reg_sets[pwndbg.gdblib.arch.current]) | {"pc", "sp"}
+        return reg in regs
+
    def __iter__(self):
        regs = set(reg_sets[pwndbg.gdblib.arch.current]) | {"pc", "sp"}
        for item in regs: