vato007/picar
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix message unpacking for commander

Browse Source
This commit is contained in:
Michael Pivato
2019-01-15 10:39:25 +10:30
parent a16fbfdd54
commit 778893d081
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
DecisionSystem/CentralisedDecision/commander.py
View File

@@ -60,7 +60,12 @@ class Commander:
self.make_decision() self.make_decision()
def on_message(self, client, userdata, message): def on_message(self, client, userdata, message):
messageDict = umsgpack.unpackb() try:
messageDict = umsgpack.unpackb(message.payload)
except:
print("Incorrect Message")
return
if "type" in messageDict.keys: if "type" in messageDict.keys:
# Need to consider that a malicious message may have a type with incorrect subtypes. # Need to consider that a malicious message may have a type with incorrect subtypes.
if messageDict["type"] == "connect": if messageDict["type"] == "connect":