Sfoglia il codice sorgente

messages.py: Mitigate random strings coming in privmsg.

parazyd 4 anni fa
parent
commit
3a38a016f5
1 ha cambiato i file con 6 aggiunte e 1 eliminazioni
  1. 6 1
      mallumo/messages.py

+ 6 - 1
mallumo/messages.py

@@ -58,7 +58,12 @@ def message_in_cb(data, modifier, modifier_data, string):
     """Incoming messages callback"""
     """Incoming messages callback"""
     debug(("message_in_cb()", data, modifier, modifier_data, string))
     debug(("message_in_cb()", data, modifier, modifier_data, string))
 
 
-    parsed = parse_privmsg(string.encode(), modifier_data)
+    # Weird typing here, lazy to look into.
+    if isinstance(string, bytes):
+        parsed = parse_privmsg(string, modifier_data)
+    else:
+        parsed = parse_privmsg(string.encode(), modifier_data)
+
     debug(json.dumps(parsed))
     debug(json.dumps(parsed))
 
 
     # If we're in a channel, do nothing more
     # If we're in a channel, do nothing more