From b179f602c4b9c2d3ff703802a21548aeb16fc71a Mon Sep 17 00:00:00 2001
From: Kekoa Kaaikala <kekoa.kaaikala@gmail.com>
Date: Fri, 2 Sep 2022 15:05:19 +0000
Subject: [PATCH] Agent: Give TCPConnectionHandler thread a name

---
 .../network/relay/tcp_connection_handler.py              | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/monkey/infection_monkey/network/relay/tcp_connection_handler.py b/monkey/infection_monkey/network/relay/tcp_connection_handler.py
index 4e88edd0b..bcd44b3d5 100644
--- a/monkey/infection_monkey/network/relay/tcp_connection_handler.py
+++ b/monkey/infection_monkey/network/relay/tcp_connection_handler.py
@@ -1,6 +1,6 @@
 import socket
 from ipaddress import IPv4Address
-from threading import Thread
+from threading import Event, Thread
 from typing import Callable
 
 PROXY_TIMEOUT = 2.5
@@ -18,9 +18,8 @@ class TCPConnectionHandler(Thread):
         self.local_port = local_port
         self.local_host = local_host
         self._client_connected = client_connected
-        super().__init__()
-        self.daemon = True
-        self._stopped = False
+        super().__init__(name="TCPConnectionHandler", daemon=True)
+        self._stopped = Event()
 
     def run(self):
         l_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
@@ -40,7 +39,7 @@ class TCPConnectionHandler(Thread):
         l_socket.close()
 
     def stop(self):
-        self._stopped = True
+        self._stopped.set()
 
     def notify_client_connected(self, callback: Callable[[socket.socket, IPv4Address], None]):
         """