2 * libwebsockets - small server side websockets and web server implementation
4 * Copyright (C) 2010-2015 Andy Green <andy@warmcat.com>
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation:
9 * version 2.1 of the License.
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
22 #include "private-libwebsockets.h"
25 * -04 of the protocol (actually the 80th version) has a radically different
26 * handshake. The 04 spec gives the following idea
28 * The handshake from the client looks as follows:
31 * Host: server.example.com
34 * Sec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==
35 * Sec-WebSocket-Origin: http://example.com
36 * Sec-WebSocket-Protocol: chat, superchat
37 * Sec-WebSocket-Version: 4
39 * The handshake from the server looks as follows:
41 * HTTP/1.1 101 Switching Protocols
44 * Sec-WebSocket-Accept: me89jWimTRKTWwrS3aRrL53YZSo=
45 * Sec-WebSocket-Nonce: AQIDBAUGBwgJCgsMDQ4PEC==
46 * Sec-WebSocket-Protocol: chat
50 #define min(a, b) ((a) < (b) ? (a) : (b))
54 * We have to take care about parsing because the headers may be split
55 * into multiple fragments. They may contain unknown headers with arbitrary
56 * argument lengths. So, we parse using a single-character at a time state
57 * machine that is completely independent of packet size.
61 lws_read(struct lws_context *context, struct lws *wsi, unsigned char *buf,
64 unsigned char *last_char;
70 case WSI_STATE_HTTP2_AWAIT_CLIENT_PREFACE:
71 case WSI_STATE_HTTP2_ESTABLISHED_PRE_SETTINGS:
72 case WSI_STATE_HTTP2_ESTABLISHED:
76 * we were accepting input but now we stopped doing so
78 if (!(wsi->rxflow_change_to & LWS_RXFLOW_ALLOW)) {
79 lws_rxflow_cache(wsi, buf, n, len);
84 /* account for what we're using in rxflow buffer */
85 if (wsi->rxflow_buffer)
87 if (lws_http2_parser(context, wsi, buf[n++]))
94 wsi->hdr_parsing_completed = 0;
96 case WSI_STATE_HTTP_ISSUING_FILE:
97 wsi->state = WSI_STATE_HTTP_HEADERS;
98 wsi->u.hdr.parser_state = WSI_TOKEN_NAME_PART;
99 wsi->u.hdr.lextable_pos = 0;
101 case WSI_STATE_HTTP_HEADERS:
102 lwsl_parser("issuing %d bytes to parser\n", (int)len);
104 if (lws_handshake_client(wsi, &buf, len))
108 if (lws_handshake_server(context, wsi, &buf, len))
109 /* Handshake indicates this session is done. */
112 /* It's possible that we've exhausted our data already, but
113 * lws_handshake_server doesn't update len for us. Figure out how
114 * much was read, so that we can proceed appropriately: */
115 len -= (buf - last_char);
117 if (!wsi->hdr_parsing_completed)
118 /* More header content on the way */
121 switch (wsi->state) {
123 case WSI_STATE_HTTP_HEADERS:
125 case WSI_STATE_HTTP_ISSUING_FILE:
127 case WSI_STATE_HTTP_BODY:
128 wsi->u.http.content_remain =
129 wsi->u.http.content_length;
130 if (wsi->u.http.content_remain)
133 /* there is no POST content */
134 goto postbody_completion;
140 case WSI_STATE_HTTP_BODY:
142 while (len && wsi->u.http.content_remain) {
143 /* Copy as much as possible, up to the limit of:
144 * what we have in the read buffer (len)
145 * remaining portion of the POST body (content_remain)
147 body_chunk_len = min(wsi->u.http.content_remain,len);
148 wsi->u.http.content_remain -= body_chunk_len;
149 len -= body_chunk_len;
151 n = wsi->protocol->callback(
152 lws_get_ctx(wsi), wsi,
153 LWS_CALLBACK_HTTP_BODY, wsi->user_space,
154 buf, body_chunk_len);
158 buf += body_chunk_len;
160 if (wsi->u.http.content_remain) {
161 lws_set_timeout(wsi, PENDING_TIMEOUT_HTTP_CONTENT,
165 /* he sent all the content in time */
167 lws_set_timeout(wsi, NO_PENDING_TIMEOUT, 0);
168 n = wsi->protocol->callback(
169 lws_get_ctx(wsi), wsi,
170 LWS_CALLBACK_HTTP_BODY_COMPLETION,
171 wsi->user_space, NULL, 0);
179 case WSI_STATE_ESTABLISHED:
180 case WSI_STATE_AWAITING_CLOSE_ACK:
181 if (lws_handshake_client(wsi, &buf, len))
184 case LWS_CONNMODE_WS_SERVING:
186 if (lws_interpret_incoming_packet(wsi, buf, len) < 0) {
187 lwsl_info("interpret_incoming_packet has bailed\n");
194 lwsl_err("lws_read: Unhandled state\n");
199 /* Nothing more to do for now */
200 lwsl_debug("lws_read: read_ok\n");
205 lwsl_debug("lws_read: http_complete\n");
207 #ifndef LWS_NO_SERVER
208 /* Did the client want to keep the HTTP connection going? */
209 if (lws_http_transaction_completed(wsi))
212 /* If we have more data, loop back around: */
219 lwsl_debug("closing connection at lws_read bail:\n");
220 lws_close_and_free_session(context, wsi, LWS_CLOSE_STATUS_NOSTATUS);