suricatasc: implement autoreconnect

Implement a basic autoreconnect support. It tries to reconnect once
when connection has been lost. If it fails, it discards the command
and try again to connect at next command.

diff --git a/scripts/suricatasc/src/suricatasc.py b/scripts/suricatasc/src/suricatasc.py
index a5389479719341d2deb31489b20006345078cef8..03483f37c2ce5d6dc23e067253cdfb42000a1541 100644 (file)
--- a/scripts/suricatasc/src/suricatasc.py
+++ b/scripts/suricatasc/src/suricatasc.py
@@ -375,7 +375,17 @@ class SuricataSC:
                 except SuricataCommandException as err:
                     print(err)
                     continue
-                cmdret = self.send_command(cmd, arguments)
+                try:
+                    cmdret = self.send_command(cmd, arguments)
+                except IOError as err:
+                    # try to reconnect and resend command
+                    print("Connection lost, trying to reconnect")
+                    try:
+                        self.connect()
+                    except SuricataNetException as err:
+                        print("Can't reconnect to suricata socket, discarding command")
+                        continue
+                    cmdret = self.send_command(cmd, arguments)
                 #decode json message
                 if cmdret["return"] == "NOK":
                     print("Error:")