2010-11-08 20:20:42 +00:00
|
|
|
/*
|
|
|
|
* libwebsockets - small server side websockets and web server implementation
|
2010-11-13 10:03:47 +00:00
|
|
|
*
|
2013-01-18 11:43:21 +08:00
|
|
|
* Copyright (C) 2010-2013 Andy Green <andy@warmcat.com>
|
2010-11-08 20:20:42 +00:00
|
|
|
*
|
|
|
|
* This library is free software; you can redistribute it and/or
|
|
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
|
|
* License as published by the Free Software Foundation:
|
|
|
|
* version 2.1 of the License.
|
|
|
|
*
|
|
|
|
* This library is distributed in the hope that it will be useful,
|
|
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
|
|
* Lesser General Public License for more details.
|
|
|
|
*
|
|
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
|
|
* License along with this library; if not, write to the Free Software
|
|
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
|
|
|
|
* MA 02110-1301 USA
|
|
|
|
*/
|
|
|
|
|
|
|
|
#include "private-libwebsockets.h"
|
|
|
|
|
2011-01-18 15:39:02 +00:00
|
|
|
/*
|
|
|
|
* -04 of the protocol (actually the 80th version) has a radically different
|
|
|
|
* handshake. The 04 spec gives the following idea
|
|
|
|
*
|
|
|
|
* The handshake from the client looks as follows:
|
|
|
|
*
|
|
|
|
* GET /chat HTTP/1.1
|
|
|
|
* Host: server.example.com
|
|
|
|
* Upgrade: websocket
|
|
|
|
* Connection: Upgrade
|
|
|
|
* Sec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==
|
|
|
|
* Sec-WebSocket-Origin: http://example.com
|
|
|
|
* Sec-WebSocket-Protocol: chat, superchat
|
2011-01-18 17:14:03 +00:00
|
|
|
* Sec-WebSocket-Version: 4
|
2011-01-18 15:39:02 +00:00
|
|
|
*
|
|
|
|
* The handshake from the server looks as follows:
|
|
|
|
*
|
|
|
|
* HTTP/1.1 101 Switching Protocols
|
|
|
|
* Upgrade: websocket
|
|
|
|
* Connection: Upgrade
|
|
|
|
* Sec-WebSocket-Accept: me89jWimTRKTWwrS3aRrL53YZSo=
|
|
|
|
* Sec-WebSocket-Nonce: AQIDBAUGBwgJCgsMDQ4PEC==
|
|
|
|
* Sec-WebSocket-Protocol: chat
|
|
|
|
*/
|
|
|
|
|
2010-11-08 20:20:42 +00:00
|
|
|
/*
|
|
|
|
* We have to take care about parsing because the headers may be split
|
|
|
|
* into multiple fragments. They may contain unknown headers with arbitrary
|
|
|
|
* argument lengths. So, we parse using a single-character at a time state
|
|
|
|
* machine that is completely independent of packet size.
|
|
|
|
*/
|
|
|
|
|
2013-03-30 09:52:21 +08:00
|
|
|
LWS_VISIBLE int
|
2012-04-09 15:09:01 +08:00
|
|
|
libwebsocket_read(struct libwebsocket_context *context,
|
2013-02-11 17:13:32 +08:00
|
|
|
struct libwebsocket *wsi, unsigned char *buf, size_t len)
|
2010-11-08 20:20:42 +00:00
|
|
|
{
|
|
|
|
size_t n;
|
2010-11-13 10:03:47 +00:00
|
|
|
|
2010-11-08 20:20:42 +00:00
|
|
|
switch (wsi->state) {
|
2013-11-19 13:38:16 +01:00
|
|
|
|
|
|
|
case WSI_STATE_HTTP_BODY:
|
|
|
|
http_postbody:
|
|
|
|
while (len--) {
|
|
|
|
|
|
|
|
if (wsi->u.http.content_length_seen >= wsi->u.http.content_length)
|
|
|
|
break;
|
|
|
|
|
|
|
|
wsi->u.http.post_buffer[wsi->u.http.body_index++] = *buf++;
|
|
|
|
wsi->u.http.content_length_seen++;
|
2013-12-08 21:26:52 +08:00
|
|
|
n = wsi->protocol->rx_buffer_size;
|
|
|
|
if (!n)
|
|
|
|
n = LWS_MAX_SOCKET_IO_BUF;
|
|
|
|
|
|
|
|
if (wsi->u.http.body_index != n &&
|
2013-11-19 13:38:16 +01:00
|
|
|
wsi->u.http.content_length_seen != wsi->u.http.content_length)
|
|
|
|
continue;
|
|
|
|
|
|
|
|
if (wsi->protocol->callback) {
|
|
|
|
n = wsi->protocol->callback(
|
|
|
|
wsi->protocol->owning_server, wsi,
|
|
|
|
LWS_CALLBACK_HTTP_BODY,
|
|
|
|
wsi->user_space, wsi->u.http.post_buffer,
|
|
|
|
wsi->u.http.body_index);
|
|
|
|
wsi->u.http.body_index = 0;
|
|
|
|
if (n)
|
|
|
|
goto bail;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (wsi->u.http.content_length_seen == wsi->u.http.content_length) {
|
|
|
|
/* he sent the content in time */
|
|
|
|
libwebsocket_set_timeout(wsi, NO_PENDING_TIMEOUT, 0);
|
|
|
|
n = wsi->protocol->callback(
|
|
|
|
wsi->protocol->owning_server, wsi,
|
|
|
|
LWS_CALLBACK_HTTP_BODY_COMPLETION,
|
|
|
|
wsi->user_space, NULL, 0);
|
|
|
|
wsi->u.http.body_index = 0;
|
|
|
|
if (n)
|
|
|
|
goto bail;
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* we need to spill here so everything is seen in the case
|
|
|
|
* there is no content-length
|
|
|
|
*/
|
|
|
|
if (wsi->u.http.body_index && wsi->protocol->callback) {
|
|
|
|
n = wsi->protocol->callback(
|
|
|
|
wsi->protocol->owning_server, wsi,
|
|
|
|
LWS_CALLBACK_HTTP_BODY,
|
|
|
|
wsi->user_space, wsi->u.http.post_buffer,
|
|
|
|
wsi->u.http.body_index);
|
|
|
|
wsi->u.http.body_index = 0;
|
|
|
|
if (n)
|
|
|
|
goto bail;
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
|
2013-01-15 13:40:23 +08:00
|
|
|
case WSI_STATE_HTTP_ISSUING_FILE:
|
2010-11-08 20:20:42 +00:00
|
|
|
case WSI_STATE_HTTP:
|
|
|
|
wsi->state = WSI_STATE_HTTP_HEADERS;
|
2013-01-21 11:04:23 +08:00
|
|
|
wsi->u.hdr.parser_state = WSI_TOKEN_NAME_PART;
|
|
|
|
wsi->u.hdr.lextable_pos = 0;
|
2010-11-08 20:20:42 +00:00
|
|
|
/* fallthru */
|
|
|
|
case WSI_STATE_HTTP_HEADERS:
|
2010-11-13 10:03:47 +00:00
|
|
|
|
2013-01-10 19:50:35 +08:00
|
|
|
lwsl_parser("issuing %d bytes to parser\n", (int)len);
|
2011-01-22 12:51:57 +00:00
|
|
|
|
2014-04-03 09:03:37 +08:00
|
|
|
if (lws_handshake_client(wsi, &buf, len))
|
|
|
|
goto bail;
|
2013-03-23 09:53:17 +08:00
|
|
|
|
2014-04-03 09:03:37 +08:00
|
|
|
switch (lws_handshake_server(context, wsi, &buf, len)) {
|
|
|
|
case 1:
|
|
|
|
goto bail;
|
|
|
|
case 2:
|
|
|
|
goto http_postbody;
|
|
|
|
}
|
2010-11-08 20:20:42 +00:00
|
|
|
break;
|
|
|
|
|
2011-03-07 07:08:12 +00:00
|
|
|
case WSI_STATE_AWAITING_CLOSE_ACK:
|
2010-11-08 20:20:42 +00:00
|
|
|
case WSI_STATE_ESTABLISHED:
|
2014-04-10 14:08:10 +08:00
|
|
|
if (lws_handshake_client(wsi, &buf, len))
|
|
|
|
goto bail;
|
2011-02-09 07:16:34 +00:00
|
|
|
switch (wsi->mode) {
|
2014-04-10 14:08:10 +08:00
|
|
|
case LWS_CONNMODE_WS_SERVING:
|
2011-01-22 12:51:57 +00:00
|
|
|
|
2014-04-10 14:08:10 +08:00
|
|
|
if (libwebsocket_interpret_incoming_packet(wsi, buf, len) < 0) {
|
|
|
|
lwsl_info("interpret_incoming_packet has bailed\n");
|
|
|
|
goto bail;
|
|
|
|
}
|
2011-02-09 07:16:34 +00:00
|
|
|
break;
|
2011-01-22 12:51:57 +00:00
|
|
|
}
|
2010-11-08 20:20:42 +00:00
|
|
|
break;
|
|
|
|
default:
|
2013-01-15 13:40:23 +08:00
|
|
|
lwsl_err("libwebsocket_read: Unhandled state\n");
|
2010-11-08 20:20:42 +00:00
|
|
|
break;
|
|
|
|
}
|
2010-11-13 10:03:47 +00:00
|
|
|
|
2010-11-08 20:20:42 +00:00
|
|
|
return 0;
|
2010-11-13 10:03:47 +00:00
|
|
|
|
2010-11-08 20:20:42 +00:00
|
|
|
bail:
|
2014-03-15 09:32:40 +08:00
|
|
|
lwsl_debug("closing connection at libwebsocket_read bail:\n");
|
2013-02-06 15:15:25 +09:00
|
|
|
|
2011-03-02 22:03:47 +00:00
|
|
|
libwebsocket_close_and_free_session(context, wsi,
|
2011-02-26 11:04:01 +00:00
|
|
|
LWS_CLOSE_STATUS_NOSTATUS);
|
2011-01-23 16:50:33 +00:00
|
|
|
|
2010-11-08 20:20:42 +00:00
|
|
|
return -1;
|
|
|
|
}
|