From bd91e2da1626ef11725f88a734c964c1aab61a59 Mon Sep 17 00:00:00 2001
From: Alexey Shchepin <alexey@process-one.net>
Date: Thu, 1 Dec 2011 15:43:56 +0200
Subject: [PATCH] Handle invalid input in ejabberd_websocket

---
 src/web/ejabberd_websocket.erl | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/web/ejabberd_websocket.erl b/src/web/ejabberd_websocket.erl
index adb6b571c..df8acfaf9 100644
--- a/src/web/ejabberd_websocket.erl
+++ b/src/web/ejabberd_websocket.erl
@@ -274,7 +274,11 @@ handle_data(L, <<>>, Socket, WsHandleLoopPid, SocketMode, WsAutoExit) ->
 	ws_loop(Socket, L, WsHandleLoopPid, SocketMode, WsAutoExit);
 
 handle_data(<<>>, L, Socket, WsHandleLoopPid, SocketMode, WsAutoExit) ->
-	ws_loop(Socket, L, WsHandleLoopPid, SocketMode, WsAutoExit).
+	ws_loop(Socket, L, WsHandleLoopPid, SocketMode, WsAutoExit);
+
+%% Invalid input
+handle_data(_, _, Socket, WsHandleLoopPid, SocketMode, WsAutoExit) ->
+    websocket_close(Socket, WsHandleLoopPid, SocketMode, WsAutoExit).
 
 % Close socket and custom handling loop dependency
 websocket_close(Socket, WsHandleLoopPid, SocketMode, WsAutoExit) ->