From d7398e1014e90449a02b4ee8b19fe6be16609591 Mon Sep 17 00:00:00 2001
From: Kekoa Kaaikala <kekoa.kaaikala@gmail.com>
Date: Tue, 6 Sep 2022 14:03:17 +0000
Subject: [PATCH] Agent: Move keep_connection into the run() method
---
monkey/infection_monkey/network/relay/sockets_pipe.py | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/monkey/infection_monkey/network/relay/sockets_pipe.py b/monkey/infection_monkey/network/relay/sockets_pipe.py
index 8031c541d..0b33cb533 100644
--- a/monkey/infection_monkey/network/relay/sockets_pipe.py
+++ b/monkey/infection_monkey/network/relay/sockets_pipe.py
@@ -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()