Skip to content

Commit 3a8016d

Browse files
committed
Fix message encoding bug
- prevent loading invalid message types
1 parent 96ea36c commit 3a8016d

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

src/messagetypes/__init__.py

+4-3
Original file line numberDiff line numberDiff line change
@@ -12,9 +12,10 @@ def encode(self):
1212

1313
def constructObject(data):
1414
try:
15-
classBase = eval(data[""] + "." + data[""].title())
16-
except NameError:
17-
logger.error("Don't know how to handle message type: \"%s\"", data[""])
15+
m = import_module("messagetypes." + data[""])
16+
classBase = getattr(m, data[""].title())
17+
except (NameError, ImportError):
18+
logger.error("Don't know how to handle message type: \"%s\"", data[""], exc_info=True)
1819
return None
1920
try:
2021
returnObj = classBase()

0 commit comments

Comments
 (0)