Python3 fixes

pwndbg/commands/context.py

@@ -38,7 +38,7 @@ def context(*args):
     result.extend(context_signal())
 
     for line in result:
-        print(line.encode('utf-8'))
+        print(line)
 
 def context_regs():
     result = []

pwndbg/commands/vmmap.py

@@ -6,6 +6,7 @@ Command to print the vitual memory map a la /proc/self/maps.
 import gdb
 import pwndbg.color
 import pwndbg.commands
+import pwndbg.compat
 import pwndbg.vmmap
 
 
@@ -17,7 +18,7 @@ def vmmap(map=None):
     """
     int_map = None
     str_map = None
-    if isinstance(map, str):
+    if isinstance(map, pwndbg.compat.basestring):
         str_map = map
     elif isinstance(map, (long, int, gdb.Value)):
         int_map = int(map)

pwndbg/compat.py

@@ -12,4 +12,6 @@ python2 = sys.version_info.major == 2
 python3 = sys.version_info.major == 3
 
 if python3:
-    globals()['basestring'] = str
+    basestring = str
+else:
+    basestring = basestring

pwndbg/funcparser.py

@@ -40,8 +40,8 @@ def ExtractFuncDecl(node, verbose=False):
     ftype, fderef, fname = extractTypeAndName(node)
 
     if not fname:
-        print "Skipping function without a name!"
-        print node.show()
+        print("Skipping function without a name!")
+        print(node.show())
         return
 
     fargs = []
@@ -55,7 +55,7 @@ def ExtractFuncDecl(node, verbose=False):
     Func = Function(ftype, fderef, fname, fargs)
 
     if verbose:
-        print Stringify(Func) + '(' + ','.join(Stringify(a) for a in Func.args) + ');'
+        print(Stringify(Func) + '(' + ','.join(Stringify(a) for a in Func.args) + ');')
 
     return Func
 
@@ -81,4 +81,4 @@ def ExtractFuncDeclFromSource(source):
     except Exception as e:
         import traceback
         traceback.print_exc()
-        # eat it
+        # eat it

pwndbg/regs.py

@@ -10,6 +10,7 @@ from types import ModuleType
 
 import gdb
 import pwndbg.arch
+import pwndbg.compat
 import pwndbg.events
 import pwndbg.memoize
 
@@ -204,8 +205,8 @@ class module(ModuleType):
         if isinstance(item, int):
             return arch_to_regs[pwndbg.arch.current][item]
 
-        if not isinstance(item, basestring):
-            print "Unknown register type: %r" % (item)
+        if not isinstance(item, pwndbg.compat.basestring):
+            print("Unknown register type: %r" % (item))
             import pdb, traceback
             traceback.print_stack()
             pdb.set_trace()