| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040 |
- /*
- * libwebsockets - small server side websockets and web server implementation
- *
- * Copyright (C) 2010-2018 Andy Green <[email protected]>
- *
- * 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 <core/private.h>
- #ifndef min
- #define min(a, b) ((a) < (b) ? (a) : (b))
- #endif
- /*
- * 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.
- *
- * Returns <0 for error or length of chars consumed from buf (up to len)
- */
- int
- lws_read_h1(struct lws *wsi, unsigned char *buf, lws_filepos_t len)
- {
- unsigned char *last_char, *oldbuf = buf;
- lws_filepos_t body_chunk_len;
- size_t n;
- lwsl_debug("%s: h1 path: wsi state 0x%x\n", __func__, lwsi_state(wsi));
- switch (lwsi_state(wsi)) {
- case LRS_ISSUING_FILE:
- return 0;
- case LRS_ESTABLISHED:
- if (lwsi_role_ws(wsi))
- goto ws_mode;
- if (lwsi_role_client(wsi))
- break;
- wsi->hdr_parsing_completed = 0;
- /* fallthru */
- case LRS_HEADERS:
- if (!wsi->http.ah) {
- lwsl_err("%s: LRS_HEADERS: NULL ah\n", __func__);
- assert(0);
- }
- lwsl_parser("issuing %d bytes to parser\n", (int)len);
- #if defined(LWS_ROLE_WS) && !defined(LWS_NO_CLIENT)
- if (lws_ws_handshake_client(wsi, &buf, (size_t)len))
- goto bail;
- #endif
- last_char = buf;
- if (lws_handshake_server(wsi, &buf, (size_t)len))
- /* Handshake indicates this session is done. */
- goto bail;
- /* we might have transitioned to RAW */
- if (wsi->role_ops == &role_ops_raw_skt ||
- wsi->role_ops == &role_ops_raw_file)
- /* we gave the read buffer to RAW handler already */
- goto read_ok;
- /*
- * It's possible that we've exhausted our data already, or
- * rx flow control has stopped us dealing with this early,
- * but lws_handshake_server doesn't update len for us.
- * Figure out how much was read, so that we can proceed
- * appropriately:
- */
- len -= (buf - last_char);
- if (!wsi->hdr_parsing_completed)
- /* More header content on the way */
- goto read_ok;
- switch (lwsi_state(wsi)) {
- case LRS_ESTABLISHED:
- case LRS_HEADERS:
- goto read_ok;
- case LRS_ISSUING_FILE:
- goto read_ok;
- case LRS_BODY:
- wsi->http.rx_content_remain =
- wsi->http.rx_content_length;
- if (wsi->http.rx_content_remain)
- goto http_postbody;
- /* there is no POST content */
- goto postbody_completion;
- default:
- break;
- }
- break;
- case LRS_BODY:
- http_postbody:
- lwsl_debug("%s: http post body: remain %d\n", __func__,
- (int)wsi->http.rx_content_remain);
- if (!wsi->http.rx_content_remain)
- goto postbody_completion;
- while (len && wsi->http.rx_content_remain) {
- /* Copy as much as possible, up to the limit of:
- * what we have in the read buffer (len)
- * remaining portion of the POST body (content_remain)
- */
- body_chunk_len = min(wsi->http.rx_content_remain, len);
- wsi->http.rx_content_remain -= body_chunk_len;
- len -= body_chunk_len;
- #ifdef LWS_WITH_CGI
- if (wsi->http.cgi) {
- struct lws_cgi_args args;
- args.ch = LWS_STDIN;
- args.stdwsi = &wsi->http.cgi->stdwsi[0];
- args.data = buf;
- args.len = body_chunk_len;
- /* returns how much used */
- n = user_callback_handle_rxflow(
- wsi->protocol->callback,
- wsi, LWS_CALLBACK_CGI_STDIN_DATA,
- wsi->user_space,
- (void *)&args, 0);
- if ((int)n < 0)
- goto bail;
- } else {
- #endif
- n = wsi->protocol->callback(wsi,
- LWS_CALLBACK_HTTP_BODY, wsi->user_space,
- buf, (size_t)body_chunk_len);
- if (n)
- goto bail;
- n = (size_t)body_chunk_len;
- #ifdef LWS_WITH_CGI
- }
- #endif
- buf += n;
- if (wsi->http.rx_content_remain) {
- lws_set_timeout(wsi,
- PENDING_TIMEOUT_HTTP_CONTENT,
- wsi->context->timeout_secs);
- break;
- }
- /* he sent all the content in time */
- postbody_completion:
- #ifdef LWS_WITH_CGI
- /*
- * If we're running a cgi, we can't let him off the
- * hook just because he sent his POST data
- */
- if (wsi->http.cgi)
- lws_set_timeout(wsi, PENDING_TIMEOUT_CGI,
- wsi->context->timeout_secs);
- else
- #endif
- lws_set_timeout(wsi, NO_PENDING_TIMEOUT, 0);
- #ifdef LWS_WITH_CGI
- if (!wsi->http.cgi)
- #endif
- {
- lwsl_info("HTTP_BODY_COMPLETION: %p (%s)\n",
- wsi, wsi->protocol->name);
- n = wsi->protocol->callback(wsi,
- LWS_CALLBACK_HTTP_BODY_COMPLETION,
- wsi->user_space, NULL, 0);
- if (n)
- goto bail;
- if (wsi->http2_substream)
- lwsi_set_state(wsi, LRS_ESTABLISHED);
- }
- break;
- }
- break;
- case LRS_RETURNED_CLOSE:
- case LRS_AWAITING_CLOSE_ACK:
- case LRS_WAITING_TO_SEND_CLOSE:
- case LRS_SHUTDOWN:
- ws_mode:
- #if !defined(LWS_NO_CLIENT) && defined(LWS_ROLE_WS)
- // lwsl_notice("%s: ws_mode\n", __func__);
- if (lws_ws_handshake_client(wsi, &buf, (size_t)len))
- goto bail;
- #endif
- #if defined(LWS_ROLE_WS)
- if (lwsi_role_ws(wsi) && lwsi_role_server(wsi) &&
- /*
- * for h2 we are on the swsi
- */
- lws_parse_ws(wsi, &buf, (size_t)len) < 0) {
- lwsl_info("%s: lws_parse_ws bailed\n", __func__);
- goto bail;
- }
- #endif
- // lwsl_notice("%s: ws_mode: buf moved on by %d\n", __func__,
- // lws_ptr_diff(buf, oldbuf));
- break;
- case LRS_DEFERRING_ACTION:
- lwsl_notice("%s: LRS_DEFERRING_ACTION\n", __func__);
- break;
- case LRS_SSL_ACK_PENDING:
- break;
- case LRS_DEAD_SOCKET:
- lwsl_err("%s: Unhandled state LRS_DEAD_SOCKET\n", __func__);
- goto bail;
- // assert(0);
- /* fallthru */
- default:
- lwsl_err("%s: Unhandled state %d\n", __func__, lwsi_state(wsi));
- assert(0);
- goto bail;
- }
- read_ok:
- /* Nothing more to do for now */
- // lwsl_info("%s: %p: read_ok, used %ld (len %d, state %d)\n", __func__,
- // wsi, (long)(buf - oldbuf), (int)len, wsi->state);
- return lws_ptr_diff(buf, oldbuf);
- bail:
- /*
- * h2 / h2-ws calls us recursively in
- *
- * lws_read_h1()->
- * lws_h2_parser()->
- * lws_read_h1()
- *
- * pattern, having stripped the h2 framing in the middle.
- *
- * When taking down the whole connection, make sure that only the
- * outer lws_read() does the wsi close.
- */
- if (!wsi->outer_will_close)
- lws_close_free_wsi(wsi, LWS_CLOSE_STATUS_NOSTATUS,
- "lws_read_h1 bail");
- return -1;
- }
- #if !defined(LWS_NO_SERVER)
- static int
- lws_h1_server_socket_service(struct lws *wsi, struct lws_pollfd *pollfd)
- {
- struct lws_context_per_thread *pt = &wsi->context->pt[(int)wsi->tsi];
- struct lws_tokens ebuf;
- int n, buffered;
- if (lwsi_state(wsi) == LRS_DEFERRING_ACTION)
- goto try_pollout;
- /* any incoming data ready? */
- if (!(pollfd->revents & pollfd->events & LWS_POLLIN))
- goto try_pollout;
- /*
- * If we previously just did POLLIN when IN and OUT were signaled
- * (because POLLIN processing may have used up the POLLOUT), don't let
- * that happen twice in a row... next time we see the situation favour
- * POLLOUT
- */
- if (wsi->favoured_pollin &&
- (pollfd->revents & pollfd->events & LWS_POLLOUT)) {
- // lwsl_notice("favouring pollout\n");
- wsi->favoured_pollin = 0;
- goto try_pollout;
- }
- /*
- * We haven't processed that the tunnel is set up yet, so
- * defer reading
- */
- if (lwsi_state(wsi) == LRS_SSL_ACK_PENDING)
- return LWS_HPI_RET_HANDLED;
- /* these states imply we MUST have an ah attached */
- if ((lwsi_state(wsi) == LRS_ESTABLISHED ||
- lwsi_state(wsi) == LRS_ISSUING_FILE ||
- lwsi_state(wsi) == LRS_HEADERS ||
- lwsi_state(wsi) == LRS_BODY)) {
- if (!wsi->http.ah && lws_header_table_attach(wsi, 0)) {
- lwsl_info("%s: wsi %p: ah not available\n", __func__,
- wsi);
- goto try_pollout;
- }
- /*
- * We got here because there was specifically POLLIN...
- * regardless of our buflist state, we need to get it,
- * and either use it, or append to the buflist and use
- * buflist head material.
- *
- * We will not notice a connection close until the buflist is
- * exhausted and we tried to do a read of some kind.
- */
- buffered = lws_buflist_aware_read(pt, wsi, &ebuf);
- switch (ebuf.len) {
- case 0:
- lwsl_info("%s: read 0 len a\n", __func__);
- wsi->seen_zero_length_recv = 1;
- lws_change_pollfd(wsi, LWS_POLLIN, 0);
- #if !defined(LWS_WITHOUT_EXTENSIONS)
- /*
- * autobahn requires us to win the race between close
- * and draining the extensions
- */
- if (wsi->ws &&
- (wsi->ws->rx_draining_ext ||
- wsi->ws->tx_draining_ext))
- goto try_pollout;
- #endif
- /*
- * normally, we respond to close with logically closing
- * our side immediately
- */
- goto fail;
- case LWS_SSL_CAPABLE_ERROR:
- goto fail;
- case LWS_SSL_CAPABLE_MORE_SERVICE:
- goto try_pollout;
- }
- /* just ignore incoming if waiting for close */
- if (lwsi_state(wsi) == LRS_FLUSHING_BEFORE_CLOSE) {
- lwsl_notice("%s: just ignoring\n", __func__);
- goto try_pollout;
- }
- if (lwsi_state(wsi) == LRS_ISSUING_FILE) {
- // lwsl_notice("stashing: wsi %p: bd %d\n", wsi, buffered);
- if (lws_buflist_aware_consume(wsi, &ebuf, 0, buffered))
- return LWS_HPI_RET_PLEASE_CLOSE_ME;
- goto try_pollout;
- }
- /*
- * Otherwise give it to whoever wants it according to the
- * connection state
- */
- #if defined(LWS_ROLE_H2)
- if (lwsi_role_h2(wsi) && lwsi_state(wsi) != LRS_BODY)
- n = lws_read_h2(wsi, (uint8_t *)ebuf.token, ebuf.len);
- else
- #endif
- n = lws_read_h1(wsi, (uint8_t *)ebuf.token, ebuf.len);
- if (n < 0) /* we closed wsi */
- return LWS_HPI_RET_WSI_ALREADY_DIED;
- lwsl_debug("%s: consumed %d\n", __func__, n);
- if (lws_buflist_aware_consume(wsi, &ebuf, n, buffered))
- return LWS_HPI_RET_PLEASE_CLOSE_ME;
- /*
- * during the parsing our role changed to something non-http,
- * so the ah has no further meaning
- */
- if (wsi->http.ah &&
- !lwsi_role_h1(wsi) &&
- !lwsi_role_h2(wsi) &&
- !lwsi_role_cgi(wsi))
- lws_header_table_detach(wsi, 0);
- /*
- * He may have used up the writability above, if we will defer
- * POLLOUT processing in favour of POLLIN, note it
- */
- if (pollfd->revents & LWS_POLLOUT)
- wsi->favoured_pollin = 1;
- return LWS_HPI_RET_HANDLED;
- }
- /*
- * He may have used up the writability above, if we will defer POLLOUT
- * processing in favour of POLLIN, note it
- */
- if (pollfd->revents & LWS_POLLOUT)
- wsi->favoured_pollin = 1;
- try_pollout:
- /* this handles POLLOUT for http serving fragments */
- if (!(pollfd->revents & LWS_POLLOUT))
- return LWS_HPI_RET_HANDLED;
- /* one shot */
- if (lws_change_pollfd(wsi, LWS_POLLOUT, 0)) {
- lwsl_notice("%s a\n", __func__);
- goto fail;
- }
- /* clear back-to-back write detection */
- wsi->could_have_pending = 0;
- if (lwsi_state(wsi) == LRS_DEFERRING_ACTION) {
- lwsl_debug("%s: LRS_DEFERRING_ACTION now writable\n", __func__);
- lwsi_set_state(wsi, LRS_ESTABLISHED);
- if (lws_change_pollfd(wsi, LWS_POLLOUT, 0)) {
- lwsl_info("failed at set pollfd\n");
- goto fail;
- }
- }
- if (!wsi->hdr_parsing_completed)
- return LWS_HPI_RET_HANDLED;
- if (lwsi_state(wsi) != LRS_ISSUING_FILE) {
- if (lws_has_buffered_out(wsi)) {
- //lwsl_notice("%s: completing partial\n", __func__);
- if (lws_issue_raw(wsi, NULL, 0) < 0) {
- lwsl_info("%s signalling to close\n", __func__);
- goto fail;
- }
- return LWS_HPI_RET_HANDLED;
- }
- lws_stats_atomic_bump(wsi->context, pt,
- LWSSTATS_C_WRITEABLE_CB, 1);
- #if defined(LWS_WITH_STATS)
- if (wsi->active_writable_req_us) {
- uint64_t ul = lws_time_in_microseconds() -
- wsi->active_writable_req_us;
- lws_stats_atomic_bump(wsi->context, pt,
- LWSSTATS_MS_WRITABLE_DELAY, ul);
- lws_stats_atomic_max(wsi->context, pt,
- LWSSTATS_MS_WORST_WRITABLE_DELAY, ul);
- wsi->active_writable_req_us = 0;
- }
- #endif
- n = user_callback_handle_rxflow(wsi->protocol->callback, wsi,
- LWS_CALLBACK_HTTP_WRITEABLE,
- wsi->user_space, NULL, 0);
- if (n < 0) {
- lwsl_info("writeable_fail\n");
- goto fail;
- }
- return LWS_HPI_RET_HANDLED;
- }
- /* >0 == completion, <0 == error
- *
- * We'll get a LWS_CALLBACK_HTTP_FILE_COMPLETION callback when
- * it's done. That's the case even if we just completed the
- * send, so wait for that.
- */
- n = lws_serve_http_file_fragment(wsi);
- if (n < 0)
- goto fail;
- return LWS_HPI_RET_HANDLED;
- fail:
- lws_close_free_wsi(wsi, LWS_CLOSE_STATUS_NOSTATUS,
- "server socket svc fail");
- return LWS_HPI_RET_WSI_ALREADY_DIED;
- }
- #endif
- static int
- rops_handle_POLLIN_h1(struct lws_context_per_thread *pt, struct lws *wsi,
- struct lws_pollfd *pollfd)
- {
- // lwsl_notice("%s: %p: wsistate 0x%x %s, revents 0x%x\n", __func__, wsi,
- // wsi->wsistate, wsi->role_ops->name, pollfd->revents);
- #ifdef LWS_WITH_CGI
- if (wsi->http.cgi && (pollfd->revents & LWS_POLLOUT)) {
- if (lws_handle_POLLOUT_event(wsi, pollfd))
- return LWS_HPI_RET_PLEASE_CLOSE_ME;
- return LWS_HPI_RET_HANDLED;
- }
- #endif
- #if 0
- /*
- * !!! lws_serve_http_file_fragment() seems to duplicate most of
- * lws_handle_POLLOUT_event() in its own loop...
- */
- lwsl_debug("%s: %d %d\n", __func__, (pollfd->revents & LWS_POLLOUT),
- lwsi_state_can_handle_POLLOUT(wsi));
- if ((pollfd->revents & LWS_POLLOUT) &&
- lwsi_state_can_handle_POLLOUT(wsi) &&
- lws_handle_POLLOUT_event(wsi, pollfd)) {
- if (lwsi_state(wsi) == LRS_RETURNED_CLOSE)
- lwsi_set_state(wsi, LRS_FLUSHING_BEFORE_CLOSE);
- /* the write failed... it's had it */
- wsi->socket_is_permanently_unusable = 1;
- return LWS_HPI_RET_PLEASE_CLOSE_ME;
- }
- #endif
- /* Priority 2: pre- compression transform */
- #if defined(LWS_WITH_HTTP_STREAM_COMPRESSION)
- if (wsi->http.comp_ctx.buflist_comp ||
- wsi->http.comp_ctx.may_have_more) {
- enum lws_write_protocol wp = LWS_WRITE_HTTP;
- lwsl_info("%s: completing comp partial (buflist_comp %p, may %d)\n",
- __func__, wsi->http.comp_ctx.buflist_comp,
- wsi->http.comp_ctx.may_have_more
- );
- if (wsi->role_ops->write_role_protocol(wsi, NULL, 0, &wp) < 0) {
- lwsl_info("%s signalling to close\n", __func__);
- return LWS_HPI_RET_PLEASE_CLOSE_ME;
- }
- lws_callback_on_writable(wsi);
- if (!wsi->http.comp_ctx.buflist_comp &&
- !wsi->http.comp_ctx.may_have_more &&
- wsi->http.deferred_transaction_completed) {
- wsi->http.deferred_transaction_completed = 0;
- if (lws_http_transaction_completed(wsi))
- return LWS_HPI_RET_PLEASE_CLOSE_ME;
- }
- return LWS_HPI_RET_HANDLED;
- }
- #endif
- if (lws_is_flowcontrolled(wsi))
- /* We cannot deal with any kind of new RX because we are
- * RX-flowcontrolled.
- */
- return LWS_HPI_RET_HANDLED;
- #if !defined(LWS_NO_SERVER)
- if (!lwsi_role_client(wsi)) {
- int n;
- lwsl_debug("%s: %p: wsistate 0x%x\n", __func__, wsi,
- wsi->wsistate);
- n = lws_h1_server_socket_service(wsi, pollfd);
- if (n != LWS_HPI_RET_HANDLED)
- return n;
- if (lwsi_state(wsi) != LRS_SSL_INIT)
- if (lws_server_socket_service_ssl(wsi,
- LWS_SOCK_INVALID))
- return LWS_HPI_RET_PLEASE_CLOSE_ME;
- return LWS_HPI_RET_HANDLED;
- }
- #endif
- #ifndef LWS_NO_CLIENT
- if ((pollfd->revents & LWS_POLLIN) &&
- wsi->hdr_parsing_completed && !wsi->told_user_closed) {
- /*
- * In SSL mode we get POLLIN notification about
- * encrypted data in.
- *
- * But that is not necessarily related to decrypted
- * data out becoming available; in may need to perform
- * other in or out before that happens.
- *
- * simply mark ourselves as having readable data
- * and turn off our POLLIN
- */
- wsi->client_rx_avail = 1;
- lws_change_pollfd(wsi, LWS_POLLIN, 0);
- //lwsl_notice("calling back %s\n", wsi->protocol->name);
- /* let user code know, he'll usually ask for writeable
- * callback and drain / re-enable it there
- */
- if (user_callback_handle_rxflow(wsi->protocol->callback, wsi,
- LWS_CALLBACK_RECEIVE_CLIENT_HTTP,
- wsi->user_space, NULL, 0)) {
- lwsl_info("RECEIVE_CLIENT_HTTP closed it\n");
- return LWS_HPI_RET_PLEASE_CLOSE_ME;
- }
- return LWS_HPI_RET_HANDLED;
- }
- #endif
- // if (lwsi_state(wsi) == LRS_ESTABLISHED)
- // return LWS_HPI_RET_HANDLED;
- #if !defined(LWS_NO_CLIENT)
- if ((pollfd->revents & LWS_POLLOUT) &&
- lws_handle_POLLOUT_event(wsi, pollfd)) {
- lwsl_debug("POLLOUT event closed it\n");
- return LWS_HPI_RET_PLEASE_CLOSE_ME;
- }
- if (lws_client_socket_service(wsi, pollfd, NULL))
- return LWS_HPI_RET_WSI_ALREADY_DIED;
- #endif
- return LWS_HPI_RET_HANDLED;
- }
- int rops_handle_POLLOUT_h1(struct lws *wsi)
- {
- if (lwsi_state(wsi) == LRS_ISSUE_HTTP_BODY)
- return LWS_HP_RET_USER_SERVICE;
- if (lwsi_role_client(wsi))
- return LWS_HP_RET_USER_SERVICE;
- return LWS_HP_RET_BAIL_OK;
- }
- static int
- rops_write_role_protocol_h1(struct lws *wsi, unsigned char *buf, size_t len,
- enum lws_write_protocol *wp)
- {
- size_t olen = len;
- int n;
- #if defined(LWS_WITH_HTTP_STREAM_COMPRESSION)
- if (wsi->http.lcs && (((*wp) & 0x1f) == LWS_WRITE_HTTP_FINAL ||
- ((*wp) & 0x1f) == LWS_WRITE_HTTP)) {
- unsigned char mtubuf[1400 + LWS_PRE +
- LWS_HTTP_CHUNK_HDR_MAX_SIZE +
- LWS_HTTP_CHUNK_TRL_MAX_SIZE],
- *out = mtubuf + LWS_PRE +
- LWS_HTTP_CHUNK_HDR_MAX_SIZE;
- size_t o = sizeof(mtubuf) - LWS_PRE -
- LWS_HTTP_CHUNK_HDR_MAX_SIZE -
- LWS_HTTP_CHUNK_TRL_MAX_SIZE;
- n = lws_http_compression_transform(wsi, buf, len, wp, &out, &o);
- if (n)
- return n;
- lwsl_info("%s: %p: transformed %d bytes to %d "
- "(wp 0x%x, more %d)\n", __func__, wsi, (int)len,
- (int)o, (int)*wp, wsi->http.comp_ctx.may_have_more);
- if (!o)
- return olen;
- if (wsi->http.comp_ctx.chunking) {
- char c[LWS_HTTP_CHUNK_HDR_MAX_SIZE + 2];
- /*
- * this only needs dealing with on http/1.1 to allow
- * pipelining
- */
- n = lws_snprintf(c, sizeof(c), "%X\x0d\x0a", (int)o);
- lwsl_info("%s: chunk (%d) %s", __func__, (int)o, c);
- out -= n;
- o += n;
- memcpy(out, c, n);
- out[o++] = '\x0d';
- out[o++] = '\x0a';
- if (((*wp) & 0x1f) == LWS_WRITE_HTTP_FINAL) {
- lwsl_info("%s: final chunk\n", __func__);
- out[o++] = '0';
- out[o++] = '\x0d';
- out[o++] = '\x0a';
- out[o++] = '\x0d';
- out[o++] = '\x0a';
- }
- }
- buf = out;
- len = o;
- }
- #endif
- n = lws_issue_raw(wsi, (unsigned char *)buf, len);
- if (n < 0)
- return n;
- /* hide there may have been compression */
- return (int)olen;
- }
- static int
- rops_alpn_negotiated_h1(struct lws *wsi, const char *alpn)
- {
- lwsl_debug("%s: client %d\n", __func__, lwsi_role_client(wsi));
- #if !defined(LWS_NO_CLIENT)
- if (lwsi_role_client(wsi)) {
- /*
- * If alpn asserts it is http/1.1, server support for KA is
- * mandatory.
- *
- * Knowing this lets us proceed with sending pipelined headers
- * before we received the first response headers.
- */
- wsi->keepalive_active = 1;
- }
- #endif
- return 0;
- }
- static int
- rops_destroy_role_h1(struct lws *wsi)
- {
- struct lws_context_per_thread *pt = &wsi->context->pt[(int)wsi->tsi];
- struct allocated_headers *ah;
- /* we may not have an ah, but may be on the waiting list... */
- lwsl_info("%s: ah det due to close\n", __func__);
- __lws_header_table_detach(wsi, 0);
- ah = pt->http.ah_list;
- while (ah) {
- if (ah->in_use && ah->wsi == wsi) {
- lwsl_err("%s: ah leak: wsi %p\n", __func__, wsi);
- ah->in_use = 0;
- ah->wsi = NULL;
- pt->http.ah_count_in_use--;
- break;
- }
- ah = ah->next;
- }
- #if defined(LWS_WITH_HTTP_STREAM_COMPRESSION)
- lws_http_compression_destroy(wsi);
- #endif
- #ifdef LWS_ROLE_WS
- lws_free_set_NULL(wsi->ws);
- #endif
- return 0;
- }
- #if !defined(LWS_NO_SERVER)
- static int
- rops_adoption_bind_h1(struct lws *wsi, int type, const char *vh_prot_name)
- {
- if (!(type & LWS_ADOPT_HTTP))
- return 0; /* no match */
- if (type & _LWS_ADOPT_FINISH) {
- if (!lws_header_table_attach(wsi, 0))
- lwsl_debug("Attached ah immediately\n");
- else
- lwsl_info("%s: waiting for ah\n", __func__);
- return 1;
- }
- lws_role_transition(wsi, LWSIFR_SERVER, (type & LWS_ADOPT_ALLOW_SSL) ?
- LRS_SSL_INIT : LRS_HEADERS, &role_ops_h1);
- if (!vh_prot_name)
- wsi->protocol = &wsi->vhost->protocols[
- wsi->vhost->default_protocol_index];
- /* the transport is accepted... give him time to negotiate */
- lws_set_timeout(wsi, PENDING_TIMEOUT_ESTABLISH_WITH_SERVER,
- wsi->context->timeout_secs);
- return 1; /* bound */
- }
- #endif
- #if !defined(LWS_NO_CLIENT)
- static const char * const http_methods[] = {
- "GET", "POST", "OPTIONS", "PUT", "PATCH", "DELETE", "CONNECT"
- };
- static int
- rops_client_bind_h1(struct lws *wsi, const struct lws_client_connect_info *i)
- {
- int n;
- if (!i) {
- /* we are finalizing an already-selected role */
- /*
- * If we stay in http, assuming there wasn't already-set
- * external user_space, since we know our initial protocol
- * we can assign the user space now, otherwise do it after the
- * ws subprotocol negotiated
- */
- if (!wsi->user_space && wsi->stash->method)
- if (lws_ensure_user_space(wsi))
- return 1;
- /*
- * For ws, default to http/1.1 only. If i->alpn had been set
- * though, defer to whatever he has set in there (eg, "h2").
- *
- * The problem is he has to commit to h2 before he can find
- * out if the server has the SETTINGS for ws-over-h2 enabled;
- * if not then ws is not possible on that connection. So we
- * only try h2 if he assertively said to use h2 alpn, otherwise
- * ws implies alpn restriction to h1.
- */
- if (!wsi->stash->method && !wsi->stash->alpn) {
- wsi->stash->alpn = lws_strdup("http/1.1");
- if (!wsi->stash->alpn)
- return 1;
- }
- /* if we went on the ah waiting list, it's ok, we can wait.
- *
- * When we do get the ah, now or later, he will end up at
- * lws_http_client_connect_via_info2().
- */
- if (lws_header_table_attach(wsi, 0) < 0)
- /*
- * if we failed here, the connection is already closed
- * and freed.
- */
- return -1;
- return 0;
- }
- /*
- * Clients that want to be h1, h2, or ws all start out as h1
- * (we don't yet know if the server supports h2 or ws)
- */
- if (!i->method) { /* websockets */
- #if defined(LWS_ROLE_WS)
- if (lws_create_client_ws_object(i, wsi))
- goto fail_wsi;
- #else
- lwsl_err("%s: ws role not configured\n", __func__);
- goto fail_wsi;
- #endif
- goto bind_h1;
- }
- /* if a recognized http method, bind to it */
- for (n = 0; n < (int)LWS_ARRAY_SIZE(http_methods); n++)
- if (!strcmp(i->method, http_methods[n]))
- goto bind_h1;
- /* other roles may bind to it */
- return 0; /* no match */
- bind_h1:
- /* assert the mode and union status (hdr) clearly */
- lws_role_transition(wsi, LWSIFR_CLIENT, LRS_UNCONNECTED, &role_ops_h1);
- return 1; /* matched */
- fail_wsi:
- return -1;
- }
- #endif
- #if 0
- static int
- rops_perform_user_POLLOUT_h1(struct lws *wsi)
- {
- volatile struct lws *vwsi = (volatile struct lws *)wsi;
- int n;
- /* priority 1: post compression-transform buffered output */
- if (lws_has_buffered_out(wsi)) {
- lwsl_debug("%s: completing partial\n", __func__);
- if (lws_issue_raw(wsi, NULL, 0) < 0) {
- lwsl_info("%s signalling to close\n", __func__);
- return -1;
- }
- n = 0;
- vwsi->leave_pollout_active = 1;
- goto cleanup;
- }
- /* priority 2: pre compression-transform buffered output */
- #if defined(LWS_WITH_HTTP_STREAM_COMPRESSION)
- if (wsi->http.comp_ctx.buflist_comp ||
- wsi->http.comp_ctx.may_have_more) {
- enum lws_write_protocol wp = LWS_WRITE_HTTP;
- lwsl_info("%s: completing comp partial"
- "(buflist_comp %p, may %d)\n",
- __func__, wsi->http.comp_ctx.buflist_comp,
- wsi->http.comp_ctx.may_have_more);
- if (rops_write_role_protocol_h1(wsi, NULL, 0, &wp) < 0) {
- lwsl_info("%s signalling to close\n", __func__);
- lws_close_free_wsi(wsi, LWS_CLOSE_STATUS_NOSTATUS,
- "comp write fail");
- }
- n = 0;
- vwsi->leave_pollout_active = 1;
- goto cleanup;
- }
- #endif
- /* priority 3: if no buffered out and waiting for that... */
- if (lwsi_state(wsi) == LRS_FLUSHING_BEFORE_CLOSE) {
- wsi->socket_is_permanently_unusable = 1;
- return -1;
- }
- /* priority 4: user writeable callback */
- vwsi = (volatile struct lws *)wsi;
- vwsi->leave_pollout_active = 0;
- n = lws_callback_as_writeable(wsi);
- cleanup:
- vwsi->handling_pollout = 0;
- if (vwsi->leave_pollout_active)
- lws_change_pollfd(wsi, 0, LWS_POLLOUT);
- return n;
- }
- #endif
- static int
- rops_close_kill_connection_h1(struct lws *wsi, enum lws_close_status reason)
- {
- #if defined(LWS_WITH_HTTP_PROXY)
- struct lws *wsi_eff = lws_client_wsi_effective(wsi);
- if (!wsi_eff->http.proxy_clientside)
- return 0;
- wsi_eff->http.proxy_clientside = 0;
- if (user_callback_handle_rxflow(wsi_eff->protocol->callback, wsi_eff,
- LWS_CALLBACK_COMPLETED_CLIENT_HTTP,
- wsi_eff->user_space, NULL, 0))
- return 0;
- #endif
- return 0;
- }
- struct lws_role_ops role_ops_h1 = {
- /* role name */ "h1",
- /* alpn id */ "http/1.1",
- /* check_upgrades */ NULL,
- /* init_context */ NULL,
- /* init_vhost */ NULL,
- /* destroy_vhost */ NULL,
- /* periodic_checks */ NULL,
- /* service_flag_pending */ NULL,
- /* handle_POLLIN */ rops_handle_POLLIN_h1,
- /* handle_POLLOUT */ rops_handle_POLLOUT_h1,
- /* perform_user_POLLOUT */ NULL,
- /* callback_on_writable */ NULL,
- /* tx_credit */ NULL,
- /* write_role_protocol */ rops_write_role_protocol_h1,
- /* encapsulation_parent */ NULL,
- /* alpn_negotiated */ rops_alpn_negotiated_h1,
- /* close_via_role_protocol */ NULL,
- /* close_role */ NULL,
- /* close_kill_connection */ rops_close_kill_connection_h1,
- /* destroy_role */ rops_destroy_role_h1,
- #if !defined(LWS_NO_SERVER)
- /* adoption_bind */ rops_adoption_bind_h1,
- #else
- NULL,
- #endif
- #if !defined(LWS_NO_CLIENT)
- /* client_bind */ rops_client_bind_h1,
- #else
- NULL,
- #endif
- /* writeable cb clnt, srv */ { LWS_CALLBACK_CLIENT_HTTP_WRITEABLE,
- LWS_CALLBACK_HTTP_WRITEABLE },
- /* close cb clnt, srv */ { LWS_CALLBACK_CLOSED_CLIENT_HTTP,
- LWS_CALLBACK_CLOSED_HTTP },
- /* protocol_bind cb c, srv */ { LWS_CALLBACK_CLIENT_HTTP_BIND_PROTOCOL,
- LWS_CALLBACK_HTTP_BIND_PROTOCOL },
- /* protocol_unbind cb c, srv */ { LWS_CALLBACK_CLIENT_HTTP_DROP_PROTOCOL,
- LWS_CALLBACK_HTTP_DROP_PROTOCOL },
- /* file_handle */ 0,
- };
|