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

Agent: Use Threading.Event instead of bool for MonkeyTunnel._stopped

This commit is contained in:
Mike Salvatore 2022-03-02 09:42:32 -05:00
parent 47062071ac
commit 393043545a
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
monkey/infection_monkey/tunnel.py
View File

@ -123,7 +123,7 @@ class MonkeyTunnel(Thread):
self._keep_tunnel_open_time = keep_tunnel_open_time self._keep_tunnel_open_time = keep_tunnel_open_time
self._broad_sock = None self._broad_sock = None
self._timeout = timeout self._timeout = timeout
self._stopped = False self._stopped = Event()
self._clients = [] self._clients = []
self.local_port = None self.local_port = None
super(MonkeyTunnel, self).__init__() super(MonkeyTunnel, self).__init__()
@ -155,7 +155,7 @@ class MonkeyTunnel(Thread):
) )
proxy.start() proxy.start()
while not self._stopped: while not self._stopped.is_set():
try: try:
search, address = self._broad_sock.recvfrom(BUFFER_READ) search, address = self._broad_sock.recvfrom(BUFFER_READ)
if b"?" == search: if b"?" == search:
@ -209,7 +209,7 @@ class MonkeyTunnel(Thread):
def stop(self): def stop(self):
self._wait_for_exploited_machine_connection() self._wait_for_exploited_machine_connection()
self._stopped = True self._stopped.set()
def _wait_for_exploited_machine_connection(self): def _wait_for_exploited_machine_connection(self):
if self._wait_for_exploited_machines.is_set(): if self._wait_for_exploited_machines.is_set():