Просмотр исходного кода

modules/websocket: Added explicit parsing of all headers before searching for WebSocket specific headers

Peter Dunkley 13 лет назад
Родитель
Сommit
067051b8c9
1 измененных файлов с 8 добавлено и 0 удалено
  1. 8 0
      modules/websocket/ws_handshake.c

+ 8 - 0
modules/websocket/ws_handshake.c

@@ -149,6 +149,14 @@ int ws_handle_handshake(struct sip_msg *msg)
 		return 0;
 	}
 
+	if (parse_headers(msg, HDR_EOH_F, 0) < 0)
+	{
+		LM_ERR("error parsing headers\n");
+		ws_send_reply(msg, 500, &str_status_internal_server_error,
+				NULL);
+		return 0;
+	}
+
 	/* Process HTTP headers */
 	while (hdr != NULL)
 	{