Fix issues in PR #64

tfw/internals/event_handling/__init__.py

@@ -3,3 +3,4 @@ from .event_handler_factory_base import EventHandlerFactoryBase
 from .control_event_handler import ControlEventHandler
 from .fsm_aware_event_handler import FSMAwareEventHandler
 from .signed_event_handler import SignedEventHandler
+from .signed_control_event_handler import SignedControlEventHandler

tfw/internals/event_handling/control_event_handler.py

@@ -4,6 +4,5 @@ from .event_handler import EventHandler
 
 
 class ControlEventHandler(EventHandler): # pylint: disable=abstract-method
-    def _event_callback(self, message):
-        if message.get('intent') != Intent.EVENT.value:
-            self.handle_event(message, self.connector)
+    def _validate_message(self, message):
+        return message.get('intent') != Intent.EVENT.value

tfw/internals/event_handling/event_handler.py

@@ -9,7 +9,11 @@ class EventHandler:
         self.connector.register_callback(self._event_callback)
 
     def _event_callback(self, message):
-        self.handle_event(message, self.connector)
+        if self._validate_message(message):
+            self.handle_event(message, self.connector)
+
+    def _validate_message(self, message):
+        return True
 
     def handle_event(self, message, connector):
         raise NotImplementedError()

tfw/internals/event_handling/signed_control_event_handler.py

@@ -0,0 +1,9 @@
+from .control_event_handler import ControlEventHandler
+from .signed_event_handler import SignedEventHandler
+
+class SignedControlEventHandler(ControlEventHandler, SignedEventHandler):
+    def _validate_message(self, message):
+        return (
+            ControlEventHandler._validate_message(self, message) and
+            SignedEventHandler._validate_message(self, message)
+        )

tfw/internals/event_handling/signed_event_handler.py

@@ -2,19 +2,19 @@ import logging
 
 from tfw.internals.crypto import KeyManager, verify_message
 
-from .control_event_handler import ControlEventHandler
+from .event_handler import EventHandler
 
 LOG = logging.getLogger(__name__)
 
 
 # pylint: disable=abstract-method
-class SignedEventHandler(ControlEventHandler):
+class SignedEventHandler(EventHandler):
     def __init__(self, connector):
         self._auth_key = KeyManager().auth_key
         super().__init__(connector)
 
-    def _event_callback(self, message):
-        if verify_message(self._auth_key, message):
-            self.handle_event(message, self.connector)
-        else:
+    def _validate_message(self, message):
+        is_valid = verify_message(self._auth_key, message)
+        if not is_valid:
             LOG.error('Message does not have valid signature: %s', message)
+        return is_valid