From: Eric Leblond <eric@regit.org>
Date: Mon, 27 May 2019 09:12:10 +0000 (+0200)
Subject: suricatasc: fix reconnect
X-Git-Tag: suricata-5.0.0-rc1~452
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=e9be6126e2683460e510daf9042f6471e6aa43f8;p=thirdparty%2Fsuricata.git

suricatasc: fix reconnect
---

diff --git a/python/suricata/sc/suricatasc.py b/python/suricata/sc/suricatasc.py
index 6b81149ea8..d6ddd3d192 100644
--- a/python/suricata/sc/suricatasc.py
+++ b/python/suricata/sc/suricatasc.py
@@ -156,6 +156,8 @@ class SuricataSC:
 
     def connect(self):
         try:
+            if self.socket == None:
+                self.socket = socket(AF_UNIX)
             self.socket.connect(self.sck_path)
         except error as err:
             raise SuricataNetException(err)
@@ -193,6 +195,7 @@ class SuricataSC:
 
     def close(self):
         self.socket.close()
+        self.socket = None
 
     def execute(self, command):
         full_cmd = command.split()
@@ -252,6 +255,7 @@ class SuricataSC:
                     # try to reconnect and resend command
                     print("Connection lost, trying to reconnect")
                     try:
+                        self.close()
                         self.connect()
                     except SuricataNetException as err:
                         print("Can't reconnect to suricata socket, discarding command")