p15693087
/
monkey
forked from p15670423/monkey
1
0
Fork 0
Code Issues Pull Requests 1 Projects Releases Wiki Activity

Agent: Move keep_connection into the run() method

This commit is contained in:
Kekoa Kaaikala 2022-09-06 14:03:17 +00:00
parent 333e587aab
commit d7398e1014
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
monkey/infection_monkey/network/relay/sockets_pipe.py
View File

@ -20,14 +20,14 @@ class SocketsPipe(Thread):
self.source = source
self.dest = dest
self.timeout = timeout
self._keep_connection = True
super().__init__(name=f"SocketsPipeThread-{self.ident}", daemon=True)
self._client_disconnected = client_disconnected
def run(self):
sockets = [self.source, self.dest]
while self._keep_connection:
self._keep_connection = False
keep_connection = True
while keep_connection:
keep_connection = False
rlist, _, xlist = select.select(sockets, [], sockets, self.timeout)
if xlist:
break
@ -42,7 +42,7 @@ class SocketsPipe(Thread):
other.sendall(data)
except Exception:
break
self._keep_connection = True
keep_connection = True
self.source.close()
self.dest.close()