Fix message unpacking for commander

This commit is contained in:
Michael Pivato
2019-01-15 10:39:25 +10:30
parent a16fbfdd54
commit 778893d081

View File

@@ -60,7 +60,12 @@ class Commander:
self.make_decision()
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:
# Need to consider that a malicious message may have a type with incorrect subtypes.
if messageDict["type"] == "connect":