Run/local command and shortcuts

Added the "run" and "local" commands for remote and local command
execution respectively and the "shortcut" command to allow for shortcuts
like "!ls" for local commands and "@ls" for remote commands.

data/pwncatrc

@@ -17,6 +17,12 @@ set on_load {
 bind s "sync"
 bind c "set state command"
 
-# Create shortcut aliases for commands
+# Create aliases for commands
 alias up upload
 alias down download
+
+# Shortcuts allow single-character prefix which indicate the entire command
+# string be passed as the arguments to a specific command. For example:
+# "!ls" run "local ls" given the below directives
+shortcut ! local
+shortcut @ run

pwncat/commands/__init__.py

@@ -146,6 +146,7 @@ class CommandParser:
 
         self.loading_complete = False
         self.aliases: Dict[str, CommandDefinition] = {}
+        self.shortcuts: Dict[str, CommandDefinition] = {}
 
     @property
     def loaded(self):
@@ -220,6 +221,11 @@ class CommandParser:
             util.error(e.args[0])
             return
 
+        if argv[0][0] in self.shortcuts:
+            command = self.shortcuts[argv[0][0]]
+            argv[0] = argv[0][1:]
+            args = argv
+        else:
             # Search for a matching command
             for command in self.commands:
                 if command.PROG == argv[0]:
@@ -237,7 +243,9 @@ class CommandParser:
                 )
                 return
 
-        args = [a.encode("utf-8").decode("unicode_escape") for a in argv[1:]]
+            args = argv[1:]
+
+        args = [a.encode("utf-8").decode("unicode_escape") for a in args]
 
         try:
             # Parse the arguments

pwncat/commands/local.py

@@ -0,0 +1,19 @@
+#!/usr/bin/env python3
+import subprocess
+
+from pwncat.commands import CommandDefinition, Complete
+from pwncat.commands.base import parameter
+
+
+class Command(CommandDefinition):
+
+    PROG = "local"
+    ARGS = {
+        "argv": parameter(
+            Complete.NONE, nargs="+", help="the local shell command to run"
+        )
+    }
+    LOCAL = True
+
+    def run(self, args):
+        subprocess.run(args.argv, shell=True)

pwncat/commands/run.py

@@ -0,0 +1,18 @@
+#!/usr/bin/env python3
+import sys
+
+import pwncat
+from pwncat.commands.base import CommandDefinition, Complete, parameter
+
+
+class Command(CommandDefinition):
+
+    PROG = "run"
+    ARGS = {
+        "argv": parameter(
+            Complete.NONE, nargs="+", help="The command to run on the remote host"
+        )
+    }
+
+    def run(self, args):
+        sys.stdout.buffer.write(pwncat.victim.run(args.argv))

pwncat/commands/shortcut.py

@@ -0,0 +1,24 @@
+import pwncat
+from pwncat.commands import CommandDefinition
+from pwncat.commands.base import parameter, Complete
+
+
+class Command(CommandDefinition):
+
+    PROG = "shortcut"
+    ARGS = {
+        "prefix": parameter(
+            Complete.NONE, help="the prefix character used for the shortcut"
+        ),
+        "command": parameter(Complete.NONE, help="the command to execute"),
+    }
+    LOCAL = True
+
+    def run(self, args):
+
+        for command in pwncat.victim.command_parser.commands:
+            if command.PROG == args.command:
+                pwncat.victim.command_parser.shortcuts[args.prefix] = command
+                return
+
+        self.parser.error(f"{args.command}: no such command")