From fc6fdef07006e5382839ef88873e90754b30e6e0 Mon Sep 17 00:00:00 2001 From: Jeff Lucovsky Date: Tue, 11 Jan 2022 14:16:24 -0500 Subject: [PATCH] suricatasc: Handle incomplete/empty recv values Issue: 4947 Improve handling of values returned by recv. Sometimes, recv returns an empty string if suricata terminates asynchronously. --- python/suricata/sc/suricatasc.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/python/suricata/sc/suricatasc.py b/python/suricata/sc/suricatasc.py index e3235ce4aa..fa0638961d 100644 --- a/python/suricata/sc/suricatasc.py +++ b/python/suricata/sc/suricatasc.py @@ -118,9 +118,14 @@ class SuricataSC: data = "" while True: if sys.version < '3': - data += self.socket.recv(INC_SIZE) + received = self.socket.recv(INC_SIZE) else: - data += self.socket.recv(INC_SIZE).decode('iso-8859-1') + received = self.socket.recv(INC_SIZE).decode('iso-8859-1') + + if not received: + break + + data += received if data.endswith('\n'): cmdret = json.loads(data) break