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 message for queuing pending channels and confirmation on exit

Browse Source
This commit is contained in:
Caleb Stewart 2021-06-21 21:52:23 -04:00
parent 157cea9c00
commit 21ac079312
1 changed files with 23 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
pwncat/manager.py
View File

@ -214,7 +214,10 @@ class Listener(threading.Thread):
return
def bootstrap_session(
self, channel: pwncat.channel.Channel, platform: str
self,
channel: pwncat.channel.Channel,
platform: str,
_queue_message: bool = False,
) -> "pwncat.manager.Session":
"""
Establish a session from an existing channel using the specified platform.
@ -225,6 +228,8 @@ class Listener(threading.Thread):
:type channel: pwncat.channel.Channel
:param platform: name of the platform to initialize
:type platform: Optional[str]
:param _queue_message: only used internally to show unitialized channel message (default: False)
:type _queue_message: bool
:rtype: pwncat.manager.Session
:raises:
ListenerError: incorrect platform or channel disconnected
@ -238,6 +243,9 @@ class Listener(threading.Thread):
if platform is None:
# We can't initialize this channel, so we just throw it on the queue
self._channel_queue.put_nowait(channel)
self.manager.log(
f"[magenta]listener[/magenta]: {str(self)}: queuing pending channel: {channel} ({self._channel_queue.qsize()} pending)"
)
return None
try:
@ -1002,6 +1010,20 @@ class Manager:
):
continue
cancel = False
for listener in self.listeners:
if listener.pending and not confirm(
"There are pending channels. Are you sure?"
):
cancel = True
break
elif listener.pending:
break
if cancel:
continue
self.log("closing interactive prompt")
break