mrtoth/baseimage-tutorial-framework
1
0
Fork 0
mirror of https://github.com/avatao-content/baseimage-tutorial-framework synced 2024-11-22 23:21:32 +00:00
Code Issues Releases Wiki Activity

Implement a ZMQ prefix-match countermeasure in EHBase

Browse Source
This commit is contained in:
Kristóf Tóth 2018-07-12 15:17:39 +02:00
parent 558e660268
commit 17b24377dd
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
lib/tfw/event_handler_base.py
View File

@ -31,10 +31,25 @@ class EventHandlerBase(ABC):
a response back in case the handler returned something. a response back in case the handler returned something.
This is subscribed in __init__(). This is subscribed in __init__().
""" """
if not self.check_key(message):
return
response = self.dispatch_handling(message) response = self.dispatch_handling(message)
if response: if response:
self.server_connector.send(response) self.server_connector.send(response)
def check_key(self, message):
"""
Checks whether the message is intended for this
EventHandler.
This is necessary because ZMQ handles PUB - SUB
connetions with pattern matching (e.g. someone
subscribed to 'fsm' will receive 'fsm_update'
messages as well.
"""
return self.key == message['key']
def dispatch_handling(self, message): def dispatch_handling(self, message):
""" """
Used to dispatch messages to their specific handlers. Used to dispatch messages to their specific handlers.