Saverr/pwncat
1
0
Fork 0
mirror of https://github.com/calebstewart/pwncat.git synced 2024-11-24 01:25:37 +01:00
Code Projects Releases Wiki Activity

Added pwncat prompt colors

Browse Source
This commit is contained in:
Caleb Stewart 2020-05-07 17:06:46 -04:00
parent 1c022ce401
commit be2da59710
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
pwncat/pty.py
View File

@ -1,5 +1,5 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
from prompt_toolkit import prompt, PromptSession from prompt_toolkit import prompt, PromptSession, ANSI
from prompt_toolkit.shortcuts import ProgressBar from prompt_toolkit.shortcuts import ProgressBar
import subprocess import subprocess
import logging import logging
@ -62,7 +62,9 @@ class PtyHandler:
self.lhost = None self.lhost = None
self.known_binaries = {} self.known_binaries = {}
self.vars = {"lhost": None} self.vars = {"lhost": None}
self.prompt = PromptSession("localhost$ ") self.prompt = PromptSession(
[("", "(local) "), ("#ff0000", "pwncat"), ("", "$ ")]
)
self.binary_aliases = { self.binary_aliases = {
"python": [ "python": [
"python2", "python2",