Andy Green | 7c212cc | 2010-11-08 20:20:42 +0000 | [diff] [blame] | 1 | /* |
| 2 | * libwebsockets - small server side websockets and web server implementation |
Andy Green | e77ddd8 | 2010-11-13 10:03:47 +0000 | [diff] [blame] | 3 | * |
Andy Green | a1ce6be | 2013-01-18 11:43:21 +0800 | [diff] [blame] | 4 | * Copyright (C) 2010-2013 Andy Green <andy@warmcat.com> |
Andy Green | 7c212cc | 2010-11-08 20:20:42 +0000 | [diff] [blame] | 5 | * |
| 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. |
| 10 | * |
| 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. |
| 15 | * |
| 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, |
| 19 | * MA 02110-1301 USA |
| 20 | */ |
| 21 | |
| 22 | #include "private-libwebsockets.h" |
| 23 | |
Andy Green | 7c212cc | 2010-11-08 20:20:42 +0000 | [diff] [blame] | 24 | /* |
Andy Green | d1b11e3 | 2011-01-18 15:39:02 +0000 | [diff] [blame] | 25 | * -04 of the protocol (actually the 80th version) has a radically different |
| 26 | * handshake. The 04 spec gives the following idea |
| 27 | * |
| 28 | * The handshake from the client looks as follows: |
| 29 | * |
| 30 | * GET /chat HTTP/1.1 |
| 31 | * Host: server.example.com |
| 32 | * Upgrade: websocket |
| 33 | * Connection: Upgrade |
| 34 | * Sec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ== |
| 35 | * Sec-WebSocket-Origin: http://example.com |
| 36 | * Sec-WebSocket-Protocol: chat, superchat |
Andy Green | e252217 | 2011-01-18 17:14:03 +0000 | [diff] [blame] | 37 | * Sec-WebSocket-Version: 4 |
Andy Green | d1b11e3 | 2011-01-18 15:39:02 +0000 | [diff] [blame] | 38 | * |
| 39 | * The handshake from the server looks as follows: |
| 40 | * |
| 41 | * HTTP/1.1 101 Switching Protocols |
| 42 | * Upgrade: websocket |
| 43 | * Connection: Upgrade |
| 44 | * Sec-WebSocket-Accept: me89jWimTRKTWwrS3aRrL53YZSo= |
| 45 | * Sec-WebSocket-Nonce: AQIDBAUGBwgJCgsMDQ4PEC== |
| 46 | * Sec-WebSocket-Protocol: chat |
| 47 | */ |
| 48 | |
| 49 | /* |
Andy Green | 7c212cc | 2010-11-08 20:20:42 +0000 | [diff] [blame] | 50 | * We have to take care about parsing because the headers may be split |
| 51 | * into multiple fragments. They may contain unknown headers with arbitrary |
| 52 | * argument lengths. So, we parse using a single-character at a time state |
| 53 | * machine that is completely independent of packet size. |
| 54 | */ |
| 55 | |
Peter Pentchev | 9a4fef7 | 2013-03-30 09:52:21 +0800 | [diff] [blame] | 56 | LWS_VISIBLE int |
Andy Green | 6ee372f | 2012-04-09 15:09:01 +0800 | [diff] [blame] | 57 | libwebsocket_read(struct libwebsocket_context *context, |
Andy Green | b5b2319 | 2013-02-11 17:13:32 +0800 | [diff] [blame] | 58 | struct libwebsocket *wsi, unsigned char *buf, size_t len) |
Andy Green | 7c212cc | 2010-11-08 20:20:42 +0000 | [diff] [blame] | 59 | { |
| 60 | size_t n; |
Andy Green | 0caf9c5 | 2013-02-18 09:48:31 +0800 | [diff] [blame] | 61 | struct allocated_headers *ah; |
| 62 | char *uri_ptr; |
| 63 | int uri_len; |
Andy Green | e77ddd8 | 2010-11-13 10:03:47 +0000 | [diff] [blame] | 64 | |
Andy Green | 7c212cc | 2010-11-08 20:20:42 +0000 | [diff] [blame] | 65 | switch (wsi->state) { |
Andy Green | d280b6e | 2013-01-15 13:40:23 +0800 | [diff] [blame] | 66 | case WSI_STATE_HTTP_ISSUING_FILE: |
Andy Green | 7c212cc | 2010-11-08 20:20:42 +0000 | [diff] [blame] | 67 | case WSI_STATE_HTTP: |
| 68 | wsi->state = WSI_STATE_HTTP_HEADERS; |
Andy Green | 623a98d | 2013-01-21 11:04:23 +0800 | [diff] [blame] | 69 | wsi->u.hdr.parser_state = WSI_TOKEN_NAME_PART; |
| 70 | wsi->u.hdr.lextable_pos = 0; |
Andy Green | 7c212cc | 2010-11-08 20:20:42 +0000 | [diff] [blame] | 71 | /* fallthru */ |
| 72 | case WSI_STATE_HTTP_HEADERS: |
Andy Green | e77ddd8 | 2010-11-13 10:03:47 +0000 | [diff] [blame] | 73 | |
Andy Green | 43db045 | 2013-01-10 19:50:35 +0800 | [diff] [blame] | 74 | lwsl_parser("issuing %d bytes to parser\n", (int)len); |
Andy Green | 4739e5c | 2011-01-22 12:51:57 +0000 | [diff] [blame] | 75 | |
Andy Green | 03674a6 | 2013-01-16 11:47:40 +0800 | [diff] [blame] | 76 | #ifndef LWS_NO_CLIENT |
Andy Green | f3d3b40 | 2011-02-09 07:16:34 +0000 | [diff] [blame] | 77 | switch (wsi->mode) { |
Andy Green | a41314f | 2011-05-23 10:00:03 +0100 | [diff] [blame] | 78 | case LWS_CONNMODE_WS_CLIENT_WAITING_PROXY_REPLY: |
| 79 | case LWS_CONNMODE_WS_CLIENT_ISSUE_HANDSHAKE: |
| 80 | case LWS_CONNMODE_WS_CLIENT_WAITING_SERVER_REPLY: |
| 81 | case LWS_CONNMODE_WS_CLIENT_WAITING_EXTENSION_CONNECT: |
Andy Green | f3d3b40 | 2011-02-09 07:16:34 +0000 | [diff] [blame] | 82 | case LWS_CONNMODE_WS_CLIENT: |
Andy Green | 4739e5c | 2011-01-22 12:51:57 +0000 | [diff] [blame] | 83 | for (n = 0; n < len; n++) |
Andy Green | 3455e67 | 2013-02-04 08:55:42 +0800 | [diff] [blame] | 84 | if (libwebsocket_client_rx_sm(wsi, *buf++)) { |
Andy Green | b5b2319 | 2013-02-11 17:13:32 +0800 | [diff] [blame] | 85 | lwsl_info("client_rx_sm failed\n"); |
Andy Green | 3455e67 | 2013-02-04 08:55:42 +0800 | [diff] [blame] | 86 | goto bail; |
| 87 | } |
Andy Green | 4739e5c | 2011-01-22 12:51:57 +0000 | [diff] [blame] | 88 | return 0; |
Andy Green | f3d3b40 | 2011-02-09 07:16:34 +0000 | [diff] [blame] | 89 | default: |
| 90 | break; |
Andy Green | 4739e5c | 2011-01-22 12:51:57 +0000 | [diff] [blame] | 91 | } |
Andy Green | 03674a6 | 2013-01-16 11:47:40 +0800 | [diff] [blame] | 92 | #endif |
Andy Green | a1ce6be | 2013-01-18 11:43:21 +0800 | [diff] [blame] | 93 | #ifndef LWS_NO_SERVER |
Andy Green | f3d3b40 | 2011-02-09 07:16:34 +0000 | [diff] [blame] | 94 | /* LWS_CONNMODE_WS_SERVING */ |
Andy Green | 4739e5c | 2011-01-22 12:51:57 +0000 | [diff] [blame] | 95 | |
Andy Green | e77ddd8 | 2010-11-13 10:03:47 +0000 | [diff] [blame] | 96 | for (n = 0; n < len; n++) |
Andy Green | 3455e67 | 2013-02-04 08:55:42 +0800 | [diff] [blame] | 97 | if (libwebsocket_parse(wsi, *buf++)) { |
| 98 | lwsl_info("libwebsocket_parse failed\n"); |
Andy Green | 38c570c | 2013-03-09 11:52:18 +0800 | [diff] [blame] | 99 | goto bail_nuke_ah; |
Andy Green | 3455e67 | 2013-02-04 08:55:42 +0800 | [diff] [blame] | 100 | } |
Andy Green | e77ddd8 | 2010-11-13 10:03:47 +0000 | [diff] [blame] | 101 | |
Andy Green | 623a98d | 2013-01-21 11:04:23 +0800 | [diff] [blame] | 102 | if (wsi->u.hdr.parser_state != WSI_PARSING_COMPLETE) |
Andy Green | 7c212cc | 2010-11-08 20:20:42 +0000 | [diff] [blame] | 103 | break; |
| 104 | |
Andy Green | 43db045 | 2013-01-10 19:50:35 +0800 | [diff] [blame] | 105 | lwsl_parser("libwebsocket_parse sees parsing complete\n"); |
Andy Green | 4739e5c | 2011-01-22 12:51:57 +0000 | [diff] [blame] | 106 | |
Andy Green | 38c570c | 2013-03-09 11:52:18 +0800 | [diff] [blame] | 107 | wsi->mode = LWS_CONNMODE_PRE_WS_SERVING_ACCEPT; |
| 108 | |
Andy Green | 7c212cc | 2010-11-08 20:20:42 +0000 | [diff] [blame] | 109 | /* is this websocket protocol or normal http 1.0? */ |
Andy Green | e77ddd8 | 2010-11-13 10:03:47 +0000 | [diff] [blame] | 110 | |
Andy Green | 16ab318 | 2013-02-10 18:02:31 +0800 | [diff] [blame] | 111 | if (!lws_hdr_total_length(wsi, WSI_TOKEN_UPGRADE) || |
| 112 | !lws_hdr_total_length(wsi, WSI_TOKEN_CONNECTION)) { |
Andy Green | 3ee9b31 | 2013-02-12 12:52:39 +0800 | [diff] [blame] | 113 | |
| 114 | /* it's not websocket.... shall we accept it as http? */ |
| 115 | |
| 116 | if (!lws_hdr_total_length(wsi, WSI_TOKEN_GET_URI)) { |
| 117 | lwsl_warn("Missing URI in HTTP request\n"); |
Andy Green | 38c570c | 2013-03-09 11:52:18 +0800 | [diff] [blame] | 118 | goto bail_nuke_ah; |
Andy Green | 3ee9b31 | 2013-02-12 12:52:39 +0800 | [diff] [blame] | 119 | } |
| 120 | |
Andy Green | 0caf9c5 | 2013-02-18 09:48:31 +0800 | [diff] [blame] | 121 | lwsl_info("HTTP request for '%s'\n", |
| 122 | lws_hdr_simple_ptr(wsi, WSI_TOKEN_GET_URI)); |
Andy Green | 3ee9b31 | 2013-02-12 12:52:39 +0800 | [diff] [blame] | 123 | |
Andy Green | 38c570c | 2013-03-09 11:52:18 +0800 | [diff] [blame] | 124 | if (libwebsocket_ensure_user_space(wsi)) |
| 125 | goto bail_nuke_ah; |
Andy Green | e803c82 | 2013-02-14 23:18:10 +0800 | [diff] [blame] | 126 | |
Andy Green | 0caf9c5 | 2013-02-18 09:48:31 +0800 | [diff] [blame] | 127 | /* |
| 128 | * Hm we still need the headers so the |
| 129 | * callback can look at leaders like the URI, but we |
| 130 | * need to transition to http union state.... hold a |
| 131 | * copy of u.hdr.ah and deallocate afterwards |
| 132 | */ |
| 133 | |
| 134 | ah = wsi->u.hdr.ah; |
| 135 | uri_ptr = lws_hdr_simple_ptr(wsi, WSI_TOKEN_GET_URI); |
| 136 | uri_len = lws_hdr_total_length(wsi, WSI_TOKEN_GET_URI); |
| 137 | |
| 138 | /* union transition */ |
| 139 | memset(&wsi->u, 0, sizeof(wsi->u)); |
| 140 | |
Andy Green | 7cf6cb0 | 2013-05-19 14:04:10 +0800 | [diff] [blame] | 141 | wsi->mode = LWS_CONNMODE_HTTP_SERVING_ACCEPTED; |
Nick Dowell | f803c0d | 2012-04-05 10:29:29 +0800 | [diff] [blame] | 142 | wsi->state = WSI_STATE_HTTP; |
Andy Green | 4708a02 | 2013-02-11 11:27:44 +0800 | [diff] [blame] | 143 | n = 0; |
Andy Green | 4f3943a | 2010-11-12 10:44:16 +0000 | [diff] [blame] | 144 | if (wsi->protocol->callback) |
Andy Green | 4708a02 | 2013-02-11 11:27:44 +0800 | [diff] [blame] | 145 | n = wsi->protocol->callback(context, wsi, |
Andy Green | b5b2319 | 2013-02-11 17:13:32 +0800 | [diff] [blame] | 146 | LWS_CALLBACK_HTTP, |
Andy Green | 0caf9c5 | 2013-02-18 09:48:31 +0800 | [diff] [blame] | 147 | wsi->user_space, uri_ptr, uri_len); |
Andy Green | 4708a02 | 2013-02-11 11:27:44 +0800 | [diff] [blame] | 148 | |
Andy Green | 0caf9c5 | 2013-02-18 09:48:31 +0800 | [diff] [blame] | 149 | /* now drop the header info we kept a pointer to */ |
| 150 | if (ah) |
| 151 | free(ah); |
Andy Green | 4708a02 | 2013-02-11 11:27:44 +0800 | [diff] [blame] | 152 | |
| 153 | if (n) { |
Andy Green | b5b2319 | 2013-02-11 17:13:32 +0800 | [diff] [blame] | 154 | lwsl_info("LWS_CALLBACK_HTTP closing\n"); |
Andy Green | 96d48fd | 2013-09-18 08:32:55 +0800 | [diff] [blame^] | 155 | goto bail; /* struct ah ptr already nuked */ |
Andy Green | 4708a02 | 2013-02-11 11:27:44 +0800 | [diff] [blame] | 156 | } |
| 157 | |
Andy Green | 7c212cc | 2010-11-08 20:20:42 +0000 | [diff] [blame] | 158 | return 0; |
| 159 | } |
| 160 | |
Andy Green | cc01247 | 2011-11-07 19:53:23 +0800 | [diff] [blame] | 161 | if (!wsi->protocol) |
Andy Green | 43db045 | 2013-01-10 19:50:35 +0800 | [diff] [blame] | 162 | lwsl_err("NULL protocol at libwebsocket_read\n"); |
Andy Green | a41314f | 2011-05-23 10:00:03 +0100 | [diff] [blame] | 163 | |
Andy Green | e252217 | 2011-01-18 17:14:03 +0000 | [diff] [blame] | 164 | /* |
| 165 | * It's websocket |
| 166 | * |
| 167 | * Make sure user side is happy about protocol |
| 168 | */ |
Andy Green | a2b0ab0 | 2010-11-11 12:28:29 +0000 | [diff] [blame] | 169 | |
Andy Green | 4f3943a | 2010-11-12 10:44:16 +0000 | [diff] [blame] | 170 | while (wsi->protocol->callback) { |
| 171 | |
Andy Green | 16ab318 | 2013-02-10 18:02:31 +0800 | [diff] [blame] | 172 | if (!lws_hdr_total_length(wsi, WSI_TOKEN_PROTOCOL)) { |
Andy Green | 4f3943a | 2010-11-12 10:44:16 +0000 | [diff] [blame] | 173 | if (wsi->protocol->name == NULL) |
| 174 | break; |
| 175 | } else |
Nick Dowell | 5519d9f | 2012-04-03 17:22:19 +0100 | [diff] [blame] | 176 | if (wsi->protocol->name && strcmp( |
Andy Green | b5b2319 | 2013-02-11 17:13:32 +0800 | [diff] [blame] | 177 | lws_hdr_simple_ptr(wsi, |
| 178 | WSI_TOKEN_PROTOCOL), |
Andy Green | 4f3943a | 2010-11-12 10:44:16 +0000 | [diff] [blame] | 179 | wsi->protocol->name) == 0) |
| 180 | break; |
Andy Green | e77ddd8 | 2010-11-13 10:03:47 +0000 | [diff] [blame] | 181 | |
Andy Green | 4f3943a | 2010-11-12 10:44:16 +0000 | [diff] [blame] | 182 | wsi->protocol++; |
| 183 | } |
Andy Green | 47943ae | 2010-11-12 11:15:49 +0000 | [diff] [blame] | 184 | |
| 185 | /* we didn't find a protocol he wanted? */ |
Andy Green | e77ddd8 | 2010-11-13 10:03:47 +0000 | [diff] [blame] | 186 | |
Andy Green | 4f3943a | 2010-11-12 10:44:16 +0000 | [diff] [blame] | 187 | if (wsi->protocol->callback == NULL) { |
Andy Green | b5b2319 | 2013-02-11 17:13:32 +0800 | [diff] [blame] | 188 | if (lws_hdr_simple_ptr(wsi, WSI_TOKEN_PROTOCOL) == |
| 189 | NULL) { |
| 190 | lwsl_info("no protocol -> prot 0 handler\n"); |
Andy Green | 5fc75a9 | 2013-01-30 12:26:14 +0800 | [diff] [blame] | 191 | wsi->protocol = &context->protocols[0]; |
| 192 | } else { |
Andy Green | b5b2319 | 2013-02-11 17:13:32 +0800 | [diff] [blame] | 193 | lwsl_err("Req protocol %s not supported\n", |
| 194 | lws_hdr_simple_ptr(wsi, WSI_TOKEN_PROTOCOL)); |
Andy Green | 38c570c | 2013-03-09 11:52:18 +0800 | [diff] [blame] | 195 | goto bail_nuke_ah; |
Andy Green | 5fc75a9 | 2013-01-30 12:26:14 +0800 | [diff] [blame] | 196 | } |
Andy Green | 4f3943a | 2010-11-12 10:44:16 +0000 | [diff] [blame] | 197 | } |
| 198 | |
Andy Green | 96d48fd | 2013-09-18 08:32:55 +0800 | [diff] [blame^] | 199 | /* allocate wsi->user storage */ |
| 200 | if (libwebsocket_ensure_user_space(wsi)) |
| 201 | goto bail_nuke_ah; |
| 202 | |
Andy Green | e77ddd8 | 2010-11-13 10:03:47 +0000 | [diff] [blame] | 203 | /* |
Andy Green | c85619d | 2011-02-13 08:25:26 +0000 | [diff] [blame] | 204 | * Give the user code a chance to study the request and |
| 205 | * have the opportunity to deny it |
| 206 | */ |
| 207 | |
Andy Green | 62c54d2 | 2011-02-14 09:14:25 +0000 | [diff] [blame] | 208 | if ((wsi->protocol->callback)(wsi->protocol->owning_server, wsi, |
Andy Green | c85619d | 2011-02-13 08:25:26 +0000 | [diff] [blame] | 209 | LWS_CALLBACK_FILTER_PROTOCOL_CONNECTION, |
Andy Green | 96d48fd | 2013-09-18 08:32:55 +0800 | [diff] [blame^] | 210 | wsi->user_space, |
| 211 | lws_hdr_simple_ptr(wsi, WSI_TOKEN_PROTOCOL), 0)) { |
Andy Green | 43db045 | 2013-01-10 19:50:35 +0800 | [diff] [blame] | 212 | lwsl_warn("User code denied connection\n"); |
Andy Green | 96d48fd | 2013-09-18 08:32:55 +0800 | [diff] [blame^] | 213 | goto bail_nuke_ah; |
Andy Green | c85619d | 2011-02-13 08:25:26 +0000 | [diff] [blame] | 214 | } |
| 215 | |
| 216 | |
| 217 | /* |
Andy Green | 38e57bb | 2011-01-19 12:20:27 +0000 | [diff] [blame] | 218 | * Perform the handshake according to the protocol version the |
| 219 | * client announced |
Andy Green | e252217 | 2011-01-18 17:14:03 +0000 | [diff] [blame] | 220 | */ |
Andy Green | e77ddd8 | 2010-11-13 10:03:47 +0000 | [diff] [blame] | 221 | |
Andy Green | e252217 | 2011-01-18 17:14:03 +0000 | [diff] [blame] | 222 | switch (wsi->ietf_spec_revision) { |
Andy Green | d85cb20 | 2011-09-25 09:32:54 +0100 | [diff] [blame] | 223 | case 13: |
Andy Green | b5b2319 | 2013-02-11 17:13:32 +0800 | [diff] [blame] | 224 | lwsl_parser("lws_parse calling handshake_04\n"); |
Andy Green | 3182ece | 2013-01-20 17:08:31 +0800 | [diff] [blame] | 225 | if (handshake_0405(context, wsi)) { |
Andy Green | b5b2319 | 2013-02-11 17:13:32 +0800 | [diff] [blame] | 226 | lwsl_info("hs0405 has failed the connection\n"); |
Andy Green | 96d48fd | 2013-09-18 08:32:55 +0800 | [diff] [blame^] | 227 | goto bail_nuke_ah; |
Andy Green | 3182ece | 2013-01-20 17:08:31 +0800 | [diff] [blame] | 228 | } |
Andy Green | bfb051f | 2011-02-09 08:49:14 +0000 | [diff] [blame] | 229 | break; |
| 230 | |
Andy Green | e252217 | 2011-01-18 17:14:03 +0000 | [diff] [blame] | 231 | default: |
Andy Green | 43db045 | 2013-01-10 19:50:35 +0800 | [diff] [blame] | 232 | lwsl_warn("Unknown client spec version %d\n", |
Andy Green | e252217 | 2011-01-18 17:14:03 +0000 | [diff] [blame] | 233 | wsi->ietf_spec_revision); |
Andy Green | 96d48fd | 2013-09-18 08:32:55 +0800 | [diff] [blame^] | 234 | goto bail_nuke_ah; |
Andy Green | 7c212cc | 2010-11-08 20:20:42 +0000 | [diff] [blame] | 235 | } |
Andy Green | e77ddd8 | 2010-11-13 10:03:47 +0000 | [diff] [blame] | 236 | |
Andy Green | 96d48fd | 2013-09-18 08:32:55 +0800 | [diff] [blame^] | 237 | /* drop the header info -- no bail_nuke_ah after this */ |
Andy Green | 2b57a34 | 2013-02-06 15:15:25 +0900 | [diff] [blame] | 238 | |
Andy Green | 16ab318 | 2013-02-10 18:02:31 +0800 | [diff] [blame] | 239 | if (wsi->u.hdr.ah) |
| 240 | free(wsi->u.hdr.ah); |
Andy Green | 2b57a34 | 2013-02-06 15:15:25 +0900 | [diff] [blame] | 241 | |
Andy Green | d280b6e | 2013-01-15 13:40:23 +0800 | [diff] [blame] | 242 | wsi->mode = LWS_CONNMODE_WS_SERVING; |
| 243 | |
Andy Green | 623a98d | 2013-01-21 11:04:23 +0800 | [diff] [blame] | 244 | /* union transition */ |
Andy Green | b5b2319 | 2013-02-11 17:13:32 +0800 | [diff] [blame] | 245 | memset(&wsi->u, 0, sizeof(wsi->u)); |
Andy Green | ca0a129 | 2013-03-16 11:24:23 +0800 | [diff] [blame] | 246 | wsi->u.ws.rxflow_change_to = LWS_RXFLOW_ALLOW; |
Andy Green | 623a98d | 2013-01-21 11:04:23 +0800 | [diff] [blame] | 247 | |
Andy Green | 5449511 | 2013-02-06 21:10:16 +0900 | [diff] [blame] | 248 | /* |
| 249 | * create the frame buffer for this connection according to the |
| 250 | * size mentioned in the protocol definition. If 0 there, use |
| 251 | * a big default for compatibility |
| 252 | */ |
| 253 | |
| 254 | n = wsi->protocol->rx_buffer_size; |
| 255 | if (!n) |
| 256 | n = LWS_MAX_SOCKET_IO_BUF; |
| 257 | n += LWS_SEND_BUFFER_PRE_PADDING + LWS_SEND_BUFFER_POST_PADDING; |
| 258 | wsi->u.ws.rx_user_buffer = malloc(n); |
| 259 | if (!wsi->u.ws.rx_user_buffer) { |
| 260 | lwsl_err("Out of Mem allocating rx buffer %d\n", n); |
Andy Green | 0480f64 | 2013-02-08 20:10:03 +0800 | [diff] [blame] | 261 | goto bail; |
Andy Green | 5449511 | 2013-02-06 21:10:16 +0900 | [diff] [blame] | 262 | } |
Andy Green | 0480f64 | 2013-02-08 20:10:03 +0800 | [diff] [blame] | 263 | lwsl_info("Allocating RX buffer %d\n", n); |
Andy Green | 5449511 | 2013-02-06 21:10:16 +0900 | [diff] [blame] | 264 | |
Andy Green | 01b206e | 2013-03-23 09:53:17 +0800 | [diff] [blame] | 265 | if (setsockopt(wsi->sock, SOL_SOCKET, SO_SNDBUF, &n, sizeof n)) { |
| 266 | lwsl_warn("Failed to set SNDBUF to %d", n); |
| 267 | goto bail; |
| 268 | } |
| 269 | |
Andy Green | 43db045 | 2013-01-10 19:50:35 +0800 | [diff] [blame] | 270 | lwsl_parser("accepted v%02d connection\n", |
Andy Green | bfb051f | 2011-02-09 08:49:14 +0000 | [diff] [blame] | 271 | wsi->ietf_spec_revision); |
Andy Green | a1ce6be | 2013-01-18 11:43:21 +0800 | [diff] [blame] | 272 | #endif |
Andy Green | 7c212cc | 2010-11-08 20:20:42 +0000 | [diff] [blame] | 273 | break; |
| 274 | |
Andy Green | da527df | 2011-03-07 07:08:12 +0000 | [diff] [blame] | 275 | case WSI_STATE_AWAITING_CLOSE_ACK: |
Andy Green | 7c212cc | 2010-11-08 20:20:42 +0000 | [diff] [blame] | 276 | case WSI_STATE_ESTABLISHED: |
Andy Green | 03674a6 | 2013-01-16 11:47:40 +0800 | [diff] [blame] | 277 | #ifndef LWS_NO_CLIENT |
Andy Green | f3d3b40 | 2011-02-09 07:16:34 +0000 | [diff] [blame] | 278 | switch (wsi->mode) { |
| 279 | case LWS_CONNMODE_WS_CLIENT: |
Andy Green | 4739e5c | 2011-01-22 12:51:57 +0000 | [diff] [blame] | 280 | for (n = 0; n < len; n++) |
Andy Green | b5b2319 | 2013-02-11 17:13:32 +0800 | [diff] [blame] | 281 | if (libwebsocket_client_rx_sm( |
| 282 | wsi, *buf++) < 0) { |
Andy Green | 3182ece | 2013-01-20 17:08:31 +0800 | [diff] [blame] | 283 | lwsl_info("client rx has bailed\n"); |
Andy Green | 66a16f3 | 2011-05-24 22:07:45 +0100 | [diff] [blame] | 284 | goto bail; |
Andy Green | 3182ece | 2013-01-20 17:08:31 +0800 | [diff] [blame] | 285 | } |
Andy Green | 4739e5c | 2011-01-22 12:51:57 +0000 | [diff] [blame] | 286 | |
| 287 | return 0; |
Andy Green | f3d3b40 | 2011-02-09 07:16:34 +0000 | [diff] [blame] | 288 | default: |
| 289 | break; |
Andy Green | 4739e5c | 2011-01-22 12:51:57 +0000 | [diff] [blame] | 290 | } |
Andy Green | 03674a6 | 2013-01-16 11:47:40 +0800 | [diff] [blame] | 291 | #endif |
Andy Green | a1ce6be | 2013-01-18 11:43:21 +0800 | [diff] [blame] | 292 | #ifndef LWS_NO_SERVER |
Andy Green | f3d3b40 | 2011-02-09 07:16:34 +0000 | [diff] [blame] | 293 | /* LWS_CONNMODE_WS_SERVING */ |
| 294 | |
Andy Green | 3182ece | 2013-01-20 17:08:31 +0800 | [diff] [blame] | 295 | if (libwebsocket_interpret_incoming_packet(wsi, buf, len) < 0) { |
| 296 | lwsl_info("interpret_incoming_packet has bailed\n"); |
Andy Green | 7c212cc | 2010-11-08 20:20:42 +0000 | [diff] [blame] | 297 | goto bail; |
Andy Green | 3182ece | 2013-01-20 17:08:31 +0800 | [diff] [blame] | 298 | } |
Andy Green | a1ce6be | 2013-01-18 11:43:21 +0800 | [diff] [blame] | 299 | #endif |
Andy Green | 7c212cc | 2010-11-08 20:20:42 +0000 | [diff] [blame] | 300 | break; |
| 301 | default: |
Andy Green | d280b6e | 2013-01-15 13:40:23 +0800 | [diff] [blame] | 302 | lwsl_err("libwebsocket_read: Unhandled state\n"); |
Andy Green | 7c212cc | 2010-11-08 20:20:42 +0000 | [diff] [blame] | 303 | break; |
| 304 | } |
Andy Green | e77ddd8 | 2010-11-13 10:03:47 +0000 | [diff] [blame] | 305 | |
Andy Green | 7c212cc | 2010-11-08 20:20:42 +0000 | [diff] [blame] | 306 | return 0; |
Andy Green | e77ddd8 | 2010-11-13 10:03:47 +0000 | [diff] [blame] | 307 | |
Andy Green | 38c570c | 2013-03-09 11:52:18 +0800 | [diff] [blame] | 308 | bail_nuke_ah: |
| 309 | /* drop the header info */ |
| 310 | if (wsi->u.hdr.ah) |
| 311 | free(wsi->u.hdr.ah); |
| 312 | |
Andy Green | 7c212cc | 2010-11-08 20:20:42 +0000 | [diff] [blame] | 313 | bail: |
Andy Green | 3182ece | 2013-01-20 17:08:31 +0800 | [diff] [blame] | 314 | lwsl_info("closing connection at libwebsocket_read bail:\n"); |
Andy Green | 2b57a34 | 2013-02-06 15:15:25 +0900 | [diff] [blame] | 315 | |
Peter Hinz | 56885f3 | 2011-03-02 22:03:47 +0000 | [diff] [blame] | 316 | libwebsocket_close_and_free_session(context, wsi, |
Andy Green | 687b018 | 2011-02-26 11:04:01 +0000 | [diff] [blame] | 317 | LWS_CLOSE_STATUS_NOSTATUS); |
Andy Green | 6964bb5 | 2011-01-23 16:50:33 +0000 | [diff] [blame] | 318 | |
Andy Green | 7c212cc | 2010-11-08 20:20:42 +0000 | [diff] [blame] | 319 | return -1; |
| 320 | } |