context: bring back args section (default off) (#397)

This allows to use args section via the context-sections config
setting (default off).
Additionally introduce a nearpc-show-args config value making it
possible to disable showing it trice while using the args section.

pwndbg/arguments.py

@@ -15,6 +15,8 @@ from capstone import CS_GRP_INT
 
 import pwndbg.abi
 import pwndbg.arch
+import pwndbg.chain
+import pwndbg.color.nearpc as N
 import pwndbg.constants
 import pwndbg.disasm
 import pwndbg.funcparser
@@ -192,3 +194,12 @@ def arguments(abi=None):
 
     for i in range(len(regs)):
         yield argname(i, abi), argument(i, abi)
+
+
+def format_args(instruction):
+    result = []
+    for arg, value in get(instruction):
+        code   = arg.type != 'char'
+        pretty = pwndbg.chain.format(value, code=code)
+        result.append('%-10s %s' % (N.argument(arg.name) + ':', pretty))
+    return result

pwndbg/commands/context.py

@@ -331,22 +331,17 @@ def context_backtrace(frame_count=10, with_banner=True):
     return result
 
 
-def context_args():
+def context_args(with_banner=True):
-    result = []
+    args = pwndbg.arguments.format_args(pwndbg.disasm.one())
 
-    ##################################################
+    # early exit to skip section if no arg found
-    # DISABLED FOR NOW, I LIKE INLINE DISPLAY BETTER
+    if not args:
-    ##################################################
+        return []
-    # # For call instructions, attempt to resolve the target and
+
-    # # determine the number of arguments.
+    if with_banner:
-    # for arg, value in pwndbg.arguments.arguments(pwndbg.disasm.one()):
+        args.insert(0, pwndbg.ui.banner("arguments"))
-    #     code   = False if arg.type == 'char' else True
+
-    #     pretty = pwndbg.chain.format(value, code=code)
+    return args
-    #     result.append('%-10s %s' % (arg.name+':', pretty))
-    # if not result:
-    #         return []
-    # result.insert(0, pwndbg.ui.banner("arguments"))
-    return result
 
 last_signal = []
 
@@ -386,6 +381,7 @@ def context_signal():
 context_sections = {
     'r': context_regs,
     'd': context_disasm,
+    'a': context_args,
     'c': context_code,
     's': context_stack,
     'b': context_backtrace

pwndbg/commands/nearpc.py

@@ -38,6 +38,7 @@ pwndbg.color.theme.Parameter('highlight-pc', True, 'whether to highlight the cur
 pwndbg.color.theme.Parameter('nearpc-prefix', '►', 'prefix marker for nearpc command')
 pwndbg.config.Parameter('left-pad-disasm', True, 'whether to left-pad disassembly')
 nearpc_lines = pwndbg.config.Parameter('nearpc-lines', 10, 'number of additional lines to print for the nearpc command')
+show_args = pwndbg.config.Parameter('nearpc-show-args', True, 'show call arguments below instruction')
 
 @pwndbg.commands.ParsedCommand
 @pwndbg.commands.OnlyWhenRunning
@@ -152,10 +153,8 @@ def nearpc(pc=None, lines=None, to_string=False, emulate=False):
 
         # For call instructions, attempt to resolve the target and
         # determine the number of arguments.
-        for arg, value in pwndbg.arguments.get(i):
+        if show_args.value:
-            code   = False if arg.type == 'char' else True
+            result.extend(['%8s%s' % ('', arg) for arg in pwndbg.arguments.format_args(instruction=i)])
-            pretty = pwndbg.chain.format(value, code=code)
-            result.append('%8s%-10s %s' % ('', N.argument(arg.name) + ':', pretty))
 
         prev = i