fix calling value() on NoneType (#125)

9 years ago · fcce92c294
parent b7c0fe8751
commit fcce92c294
1 changed files with 3 additions and 1 deletions
--- a/pwndbg/commands/heap.py
+++ b/pwndbg/commands/heap.py
@ -26,7 +26,9 @@ def value_from_type(type_name, addr):

 def get_main_arena(addr=None):
    if addr == None:
-        main_arena = gdb.lookup_symbol('main_arena')[0].value()
+        main_arena = gdb.lookup_symbol('main_arena')[0]
+        if main_arena is not None:
+            main_arena = main_arena.value()
    else:
        main_arena = value_from_type('struct malloc_state', addr)