From efd63af453834fd88321b71f7758efb94c1002be Mon Sep 17 00:00:00 2001
From: John Carr <john.carr@unrouted.co.uk>
Date: Mon, 1 Feb 2021 10:58:36 +0000
Subject: [PATCH] Fixes for hpfeeds reconnection

---
 modules/python/dionaea/hpfeeds.py | 19 +++++++++++++++----
 1 file changed, 15 insertions(+), 4 deletions(-)

diff --git a/modules/python/dionaea/hpfeeds.py b/modules/python/dionaea/hpfeeds.py
index 9a6cb19b..782b9cf9 100644
--- a/modules/python/dionaea/hpfeeds.py
+++ b/modules/python/dionaea/hpfeeds.py
@@ -104,6 +104,9 @@ class FeedUnpack(object):
     def __init__(self):
         self.buf = bytearray()
 
+    def reset(self):
+        self.buf = bytearray()
+
     def __iter__(self):
         return self
 
@@ -163,6 +166,7 @@ def handle_io_in(self, indata):
                     logger.debug('hpclient server name {0} rand {1}'.format(name, rand))
                     self.send(msgauth(rand, self.ident, self.secret))
                     self.authenticated = True
+                    self.handle_io_out()
 
                 elif opcode == OP_PUBLISH:
                     ident, data = strunpack8(data)
@@ -180,6 +184,9 @@ def handle_io_in(self, indata):
         return len(indata)
 
     def handle_io_out(self):
+        if not self.authenticated:
+            return
+
         if self.filehandle:
             self.sendfiledata()
         else:
@@ -227,16 +234,20 @@ def sendfiledata(self):
     def handle_timeout_idle(self):
         pass
 
-    def handle_disconnect(self):
-        logger.info('hpclient disconnect')
+    def _reset_connection_state(self):
         self.connected = False
         self.authenticated = False
+        self.filehandle = None
+        self.unpacker.reset()
+
+    def handle_disconnect(self):
+        logger.info('hpclient disconnect')
+        self._reset_connection_state()
         return 1
 
     def handle_error(self, err):
         logger.warn(str(err))
-        self.connected = False
-        self.authenticated = False
+        self._reset_connection_state()
         return 1