Agent: Use event.wait() when running TCPRelay

monkey/infection_monkey/tcp_relay.py

@@ -46,8 +46,7 @@ class TCPRelay(Thread):
         )
         proxy.start()
 
-        while not self._stopped.is_set():
-            sleep(0.001)
+        self._stopped.wait()
 
         self._wait_for_users_to_disconnect()
 

vulture_allowlist.py

@@ -286,6 +286,9 @@ event
 deserialize
 serialized_event
 
+# TODO: Remove after #2231 is closed
+relay_users
+
 # pydantic base models
 underscore_attrs_are_private
 extra