From 2af585660df939ea99c7aadd72108fa408147077 Mon Sep 17 00:00:00 2001 From: Andy Green Date: Tue, 30 Sep 2014 08:15:49 +0800 Subject: [PATCH] confirm upgrade actually for websocket Signed-off-by: Andy Green --- lib/server.c | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/lib/server.c b/lib/server.c index da08d07b..f0ebb1fa 100644 --- a/lib/server.c +++ b/lib/server.c @@ -354,6 +354,16 @@ int lws_handshake_server(struct libwebsocket_context *context, return 0; /* don't bail out of libwebsocket_read, just yet */ } + if (!strcasecmp(lws_hdr_simple_ptr(wsi, WSI_TOKEN_UPGRADE), + "websocket")) + goto upgrade_ws; + + + /* dunno what he wanted to upgrade to */ + goto bail_nuke_ah; + + +upgrade_ws: if (!wsi->protocol) lwsl_err("NULL protocol at libwebsocket_read\n");