Add type checking when sendning data

We use this in various ways so add an early check to make things clear
in case anything is called incorrectly.

1 files changed, 9 insertions, 0 deletions

diff --git a/websockify/websocket.py b/websockify/websocket.py
index a9f85a9..d15113b 100644
--- a/websockify/websocket.py
+++ b/websockify/websocket.py
@@ -436,6 +436,9 @@ class WebSocket(object):
         WebSocketWantWriteError can be raised if there is insufficient
         space in the underlying socket.
         """
+        if not isinstance(msg, bytes):
+            raise TypeError
+
         if not self._sent_close:
             # Only called to flush?
             self._sendmsg(0x2, msg)
@@ -445,10 +448,16 @@ class WebSocket(object):
 
     def ping(self, data=''.encode('ascii')):
         """Write a ping message to the WebSocket."""
+        if not isinstance(data, bytes):
+            raise TypeError
+
         self._sendmsg(0x9, data)
 
     def pong(self, data=''.encode('ascii')):
         """Write a pong message to the WebSocket."""
+        if not isinstance(data, bytes):
+            raise TypeError
+
         self._sendmsg(0xA, data)
 
     def shutdown(self, how, code=1000, reason=None):