diff --git a/README.md b/README.md index 31cf7e2..b2494c5 100644 --- a/README.md +++ b/README.md @@ -145,32 +145,6 @@ slipped through the cracks and been observed in the wild. When this happens, `pw will do whatever it can to preserve your terminal, but you may be greeted with some peculiar output or command failures. -### Dash Support - -The Debian shell `dash` aims to be a very minimalistic shell. It's focus is not on user -interface, but on running scripts quickly and correctly. As a result, some of the features -we expect from an interactive shell simply don't work in `dash`. `pwncat` tries not to -depend on a specific shell environment, so if you start your reverse or bind shell with -`/bin/sh` or `/bin/dash`, then you may get a weird prompt. `dash` does not obey the -terminal escape sequences which `pwncat` adds, so you may get a very long terminal like this: - -```shell script -\[\033[01;31m\](remote)\[\033[00m\] \[\033[01;33m\]\u@\h\[\033[00m\]:\[\033[01;36m\]\w\[\033[00m\]$ -``` - -Currently, the only workaround is to use the `prompt` command at the local `pwncat` prompt. -The command allows you to modify the prompt which `pwncat` will automatically set whenever -resetting the remote terminal. Two options are provided: "basic" and "fancy". The "fancy" -prompt is the default which causes the above output in Dash. To switch to the basic prompt -you can use the following command at the `pwncat` prompt: - -```shell script -prompt --basic -``` - -While this is inconvenient, it does not affect the behaviour of `pwncat`. All `pwncat` -features will continue to function properly no matter what your prompt looks like. - ### BSD Support While BSD is a Unix-based kernel, in practice it's userland tools are noticeably diff --git a/pwncat/commands/prompt.py b/pwncat/commands/prompt.py index 210d955..8da6d19 100644 --- a/pwncat/commands/prompt.py +++ b/pwncat/commands/prompt.py @@ -20,7 +20,7 @@ class Command(CommandDefinition): Complete.NONE, group="mutex", action="store_true", - help="Set a basic prompt with no color or automatic system information", + help="Set a basic prompt with no color or automatic system information. There _should_ be no reason to use that anymore (unless your local terminal has no ANSI support)", ), "--fancy,-f": Parameter( Complete.NONE, @@ -33,13 +33,8 @@ class Command(CommandDefinition): def run(self, args): if args.fancy: - pwncat.victim.remote_prefix = "\\[\\033[01;31m\\](remote)\\[\\033[00m\\]" - pwncat.victim.remote_prompt = ( - "\\[\\033[01;33m\\]\\u@\\h\\[\\033[00m\\]:\\[" - "\\033[01;36m\\]\\w\\[\\033[00m\\]\\$ " - ) + pwncat.victim.remote_prompt = """$(command printf "\\033[01;31m(remote)\\033[0m \\033[01;33m$(whoami)@$(hostname)\\033[0m:\\033[1;36m$PWD\\033[0m$ ")""" else: - pwncat.victim.remote_prefix = "(remote)" - pwncat.victim.remote_prompt = f"{pwncat.victim.host.ip}:$PWD\\$ " + pwncat.victim.remote_prompt = f"(remote) {pwncat.victim.host.ip}:$PWD\\$ " pwncat.victim.reset(hard=False) diff --git a/pwncat/remote/victim.py b/pwncat/remote/victim.py index 79ebf30..f30650a 100644 --- a/pwncat/remote/victim.py +++ b/pwncat/remote/victim.py @@ -113,12 +113,8 @@ class Victim: # Saved remote terminal state (for transition to/from raw mode) self.saved_term_state = None # util.enter_raw_mode() # util.restore_terminal(self.saved_term_state, new_line=False) - # Prompt and prompt prefix - self.remote_prefix = "\\[\\033[01;31m\\](remote)\\[\\033[00m\\]" - self.remote_prompt = ( - "\\[\\033[01;33m\\]\\u@\\h\\[\\033[00m\\]:\\[" - "\\033[01;36m\\]\\w\\[\\033[00m\\]\\$ " - ) + # Prompt + self.remote_prompt = """$(command printf "\\033[01;31m(remote)\\033[0m \\033[01;33m$(whoami)@$(hostname)\\033[0m:\\033[1;36m$PWD\\033[0m$ ")""" # Aliases for equivalent commands self.binary_aliases = { "python": [ @@ -368,14 +364,7 @@ class Victim: progress.update(task_id, status="prompt") - if self.shell == "/bin/sh": - progress.log( - f"[yellow]warning[/yellow]: /bin/sh does not support colored prompts." - ) - self.remote_prefix = "(remote)" - self.remote_prompt = f"{pwncat.victim.host.ip}:$PWD\\$ " - - self.run(f"export PS1='{self.remote_prefix} {self.remote_prompt}'") + self.run(f"export PS1='{self.remote_prompt}'") # This should be valid in any POSIX compliant shell progress.update(task_id, status="checking for pty") @@ -416,7 +405,7 @@ class Victim: # This stuff won't carry through to the PTY, so we need to reset it again. self.run("unset PROMPT_COMMAND") - self.run(f"export PS1='{self.remote_prefix} {self.remote_prompt}'") + self.run(f"export PS1='{self.remote_prompt}'") # Make sure HISTFILE is unset in this PTY (it resets when a pty is # opened) @@ -1839,7 +1828,7 @@ class Victim: self.has_cr = True self.has_echo = True self.run("echo") - self.run(f"export PS1='{self.remote_prefix} {self.remote_prompt}'") + self.run(f"export PS1='{self.remote_prompt}'") def flush_output(self, some=False): """ @@ -1909,7 +1898,7 @@ class Victim: self.run("unset HISTFILE; export HISTCONTROL=ignorespace") self.run("unset PROMPT_COMMAND") self.run("unalias -a") - self.run(f"export PS1='{self.remote_prefix} {self.remote_prompt}'") + self.run(f"export PS1='{self.remote_prompt}'") self.run(f"tput rmam") def recvuntil(self, needle: bytes, interp=False, timeout=None):