Saverr/pwncat
1
0
Fork 0
mirror of https://github.com/calebstewart/pwncat.git synced 2024-11-27 19:04:15 +01:00
Code Projects Releases Wiki Activity

Added colors to upload and download feature

Browse Source
This commit is contained in:
John Hammond 2020-05-07 17:35:41 -04:00
parent b0ab851140
commit f4e7e2ddb2
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
pwncat/pty.py
View File

@ -256,6 +256,7 @@ class PtyHandler:
self.enter_raw(save=False) self.enter_raw(save=False)
def do_download(self, argv): def do_download(self, argv):
""" Download a file from the remote host """
parser = argparse.ArgumentParser(prog="download") parser = argparse.ArgumentParser(prog="download")
parser.add_argument( parser.add_argument(
@ -299,8 +300,9 @@ class PtyHandler:
return return
size = int(size) size = int(size)
with ProgressBar(f"downloading with {download.NAME}") as pb: with ProgressBar(
[("#888888", "downloading with "), ("fg:ansiyellow", f"{download.NAME}")]
) as pb:
counter = pb(range(os.path.getsize(path))) counter = pb(range(os.path.getsize(path)))
last_update = time.time() last_update = time.time()
@ -370,7 +372,9 @@ class PtyHandler:
upload = UploaderClass(self, remote_path=outfile, local_path=path) upload = UploaderClass(self, remote_path=outfile, local_path=path)
with ProgressBar(f"uploading via {upload.NAME}") as pb: with ProgressBar(
[("#888888", "uploading via "), ("fg:ansiyellow", f"{upload.NAME}")]
) as pb:
counter = pb(range(os.path.getsize(path))) counter = pb(range(os.path.getsize(path)))
last_update = time.time() last_update = time.time()