/* * libwebsockets - small server side websockets and web server implementation * * Copyright (C) 2010 - 2020 Andy Green * * Permission is hereby granted, free of charge, to any person obtaining a copy * of this software and associated documentation files (the "Software"), to * deal in the Software without restriction, including without limitation the * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or * sell copies of the Software, and to permit persons to whom the Software is * furnished to do so, subject to the following conditions: * * The above copyright notice and this permission notice shall be included in * all copies or substantial portions of the Software. * * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS * IN THE SOFTWARE. */ #include "private-lib-core.h" #if defined(LWS_WITH_TLS) int lws_client_create_tls(struct lws *wsi, const char **pcce, int do_c1) { int n; /* we can retry this... just cook the SSL BIO the first time */ if (wsi->tls.use_ssl & LCCSCF_USE_SSL) { if (!wsi->tls.ssl) { if (lws_ssl_client_bio_create(wsi) < 0) { *pcce = "bio_create failed"; return -1; } if (!wsi->transaction_from_pipeline_queue && lws_tls_restrict_borrow(wsi->context)) { *pcce = "tls restriction limit"; return -1; } } if (!do_c1) return 0; n = lws_ssl_client_connect1(wsi); if (!n) return 1; /* caller should return 0 */ if (n < 0) { *pcce = "lws_ssl_client_connect1 failed"; return -1; } } else wsi->tls.ssl = NULL; #if defined (LWS_WITH_HTTP2) if (wsi->client_h2_alpn) { /* * We connected to the server and set up tls, and * negotiated "h2". * * So this is it, we are an h2 master client connection * now, not an h1 client connection. */ #if defined(LWS_WITH_TLS) lws_tls_server_conn_alpn(wsi); #endif /* send the H2 preface to legitimize the connection */ if (lws_h2_issue_preface(wsi)) { *pcce = "error sending h2 preface"; return -1; } } #endif return 0; /* OK */ } #endif void lws_client_http_body_pending(struct lws *wsi, int something_left_to_send) { wsi->client_http_body_pending = !!something_left_to_send; } int lws_client_socket_service(struct lws *wsi, struct lws_pollfd *pollfd) { struct lws_context *context = wsi->context; struct lws_context_per_thread *pt = &context->pt[(int)wsi->tsi]; char *p = (char *)&pt->serv_buf[0]; #if defined(LWS_WITH_TLS) char ebuf[128]; #endif const char *cce = NULL; char *sb = p; int n = 0; switch (lwsi_state(wsi)) { case LRS_WAITING_DNS: /* * we are under PENDING_TIMEOUT_SENT_CLIENT_HANDSHAKE * timeout protection set in client-handshake.c */ lwsl_err("%s: wsi %p: WAITING_DNS\n", __func__, wsi); if (!lws_client_connect_2_dnsreq(wsi)) { /* closed */ lwsl_client("closed\n"); return -1; } /* either still pending connection, or changed mode */ return 0; case LRS_WAITING_CONNECT: /* * we are under PENDING_TIMEOUT_SENT_CLIENT_HANDSHAKE * timeout protection set in client-handshake.c */ if (pollfd->revents & LWS_POLLOUT) lws_client_connect_3_connect(wsi, NULL, NULL, 0, NULL); break; #if defined(LWS_WITH_SOCKS5) /* SOCKS Greeting Reply */ case LRS_WAITING_SOCKS_GREETING_REPLY: case LRS_WAITING_SOCKS_AUTH_REPLY: case LRS_WAITING_SOCKS_CONNECT_REPLY: switch (lws_socks5c_handle_state(wsi, pollfd, &cce)) { case LW5CHS_RET_RET0: return 0; case LW5CHS_RET_BAIL3: goto bail3; case LW5CHS_RET_STARTHS: goto start_ws_handshake; default: break; } break; #endif #if defined(LWS_CLIENT_HTTP_PROXYING) && (defined(LWS_ROLE_H1) || defined(LWS_ROLE_H2)) case LRS_WAITING_PROXY_REPLY: /* handle proxy hung up on us */ if (pollfd->revents & LWS_POLLHUP) { lwsl_warn("Proxy connection %p (fd=%d) dead\n", (void *)wsi, pollfd->fd); cce = "proxy conn dead"; goto bail3; } n = recv(wsi->desc.sockfd, sb, context->pt_serv_buf_size, 0); if (n < 0) { if (LWS_ERRNO == LWS_EAGAIN) { lwsl_debug("Proxy read EAGAIN... retrying\n"); return 0; } lwsl_err("ERROR reading from proxy socket\n"); cce = "proxy read err"; goto bail3; } pt->serv_buf[13] = '\0'; if (n < 13 || (strncmp(sb, "HTTP/1.0 200 ", 13) && strncmp(sb, "HTTP/1.1 200 ", 13))) { lwsl_err("%s: ERROR proxy did not reply with h1\n", __func__); /* lwsl_hexdump_notice(sb, n); */ cce = "proxy not h1"; goto bail3; } lwsl_info("%s: proxy connection extablished\n", __func__); /* clear his proxy connection timeout */ lws_set_timeout(wsi, NO_PENDING_TIMEOUT, 0); /* fallthru */ #endif case LRS_H1C_ISSUE_HANDSHAKE: /* * we are under PENDING_TIMEOUT_SENT_CLIENT_HANDSHAKE * timeout protection set in client-handshake.c * * take care of our lws_callback_on_writable * happening at a time when there's no real connection yet */ #if defined(LWS_WITH_SOCKS5) start_ws_handshake: #endif if (lws_change_pollfd(wsi, LWS_POLLOUT, 0)) return -1; #if defined(LWS_WITH_TLS) n = lws_client_create_tls(wsi, &cce, 1); if (n < 0) goto bail3; if (n == 1) return 0; /* fallthru */ case LRS_WAITING_SSL: if (wsi->tls.use_ssl & LCCSCF_USE_SSL) { n = lws_ssl_client_connect2(wsi, ebuf, sizeof(ebuf)); if (!n) return 0; if (n < 0) { cce = ebuf; goto bail3; } } else wsi->tls.ssl = NULL; #endif #if defined(LWS_WITH_DETAILED_LATENCY) if (context->detailed_latency_cb) { wsi->detlat.type = LDLT_TLS_NEG_CLIENT; wsi->detlat.latencies[LAT_DUR_PROXY_CLIENT_REQ_TO_WRITE] = lws_now_usecs() - wsi->detlat.earliest_write_req_pre_write; wsi->detlat.latencies[LAT_DUR_USERCB] = 0; lws_det_lat_cb(wsi->context, &wsi->detlat); } #endif #if defined (LWS_WITH_HTTP2) if (wsi->client_h2_alpn) { /* * We connected to the server and set up tls, and * negotiated "h2". * * So this is it, we are an h2 master client connection * now, not an h1 client connection. */ #if defined(LWS_WITH_TLS) lws_tls_server_conn_alpn(wsi); #endif /* send the H2 preface to legitimize the connection */ if (lws_h2_issue_preface(wsi)) { cce = "error sending h2 preface"; goto bail3; } // lwsi_set_state(wsi, LRS_H1C_ISSUE_HANDSHAKE2); lws_set_timeout(wsi, PENDING_TIMEOUT_AWAITING_CLIENT_HS_SEND, context->timeout_secs); break; } #endif /* fallthru */ case LRS_H1C_ISSUE_HANDSHAKE2: p = lws_generate_client_handshake(wsi, p); if (p == NULL) { if (wsi->role_ops == &role_ops_raw_skt #if defined(LWS_ROLE_RAW_FILE) || wsi->role_ops == &role_ops_raw_file #endif ) return 0; lwsl_err("Failed to generate handshake for client\n"); lws_close_free_wsi(wsi, LWS_CLOSE_STATUS_NOSTATUS, "chs"); return 0; } /* send our request to the server */ lwsl_info("%s: HANDSHAKE2: %p: sending headers " "(wsistate 0x%lx), w sock %d\n", __func__, wsi, (unsigned long)wsi->wsistate, wsi->desc.sockfd); #if defined(LWS_WITH_DETAILED_LATENCY) wsi->detlat.earliest_write_req_pre_write = lws_now_usecs(); #endif n = lws_ssl_capable_write(wsi, (unsigned char *)sb, (int)(p - sb)); switch (n) { case LWS_SSL_CAPABLE_ERROR: lwsl_debug("ERROR writing to client socket\n"); lws_close_free_wsi(wsi, LWS_CLOSE_STATUS_NOSTATUS, "cws"); return 0; case LWS_SSL_CAPABLE_MORE_SERVICE: lws_callback_on_writable(wsi); break; } if (wsi->client_http_body_pending) { lwsl_debug("body pending\n"); lwsi_set_state(wsi, LRS_ISSUE_HTTP_BODY); lws_set_timeout(wsi, PENDING_TIMEOUT_CLIENT_ISSUE_PAYLOAD, context->timeout_secs); if (wsi->flags & LCCSCF_HTTP_X_WWW_FORM_URLENCODED) lws_callback_on_writable(wsi); #if defined(LWS_WITH_HTTP_PROXY) if (wsi->http.proxy_clientside) lws_callback_on_writable(wsi); #endif /* user code must ask for writable callback */ break; } lwsi_set_state(wsi, LRS_WAITING_SERVER_REPLY); wsi->hdr_parsing_completed = 0; if (lwsi_state(wsi) == LRS_IDLING) { lwsi_set_state(wsi, LRS_WAITING_SERVER_REPLY); wsi->hdr_parsing_completed = 0; #if defined(LWS_ROLE_H1) || defined(LWS_ROLE_H2) wsi->http.ah->parser_state = WSI_TOKEN_NAME_PART; wsi->http.ah->lextable_pos = 0; #if defined(LWS_WITH_CUSTOM_HEADERS) wsi->http.ah->unk_pos = 0; #endif /* If we're (re)starting on hdr, need other implied init */ wsi->http.ah->ues = URIES_IDLE; #endif } lws_set_timeout(wsi, PENDING_TIMEOUT_AWAITING_SERVER_RESPONSE, wsi->context->timeout_secs); lws_callback_on_writable(wsi); goto client_http_body_sent; case LRS_ISSUE_HTTP_BODY: #if defined(LWS_WITH_HTTP_PROXY) if (wsi->http.proxy_clientside) { lws_callback_on_writable(wsi); break; } #endif if (wsi->client_http_body_pending) { //lws_set_timeout(wsi, // PENDING_TIMEOUT_CLIENT_ISSUE_PAYLOAD, // context->timeout_secs); /* user code must ask for writable callback */ break; } client_http_body_sent: #if defined(LWS_ROLE_H1) || defined(LWS_ROLE_H2) /* prepare ourselves to do the parsing */ wsi->http.ah->parser_state = WSI_TOKEN_NAME_PART; wsi->http.ah->lextable_pos = 0; #if defined(LWS_WITH_CUSTOM_HEADERS) wsi->http.ah->unk_pos = 0; #endif #endif lwsi_set_state(wsi, LRS_WAITING_SERVER_REPLY); lws_set_timeout(wsi, PENDING_TIMEOUT_AWAITING_SERVER_RESPONSE, context->timeout_secs); break; case LRS_WAITING_SERVER_REPLY: /* * handle server hanging up on us... * but if there is POLLIN waiting, handle that first */ if ((pollfd->revents & (LWS_POLLIN | LWS_POLLHUP)) == LWS_POLLHUP) { lwsl_debug("Server connection %p (fd=%d) dead\n", (void *)wsi, pollfd->fd); cce = "Peer hung up"; goto bail3; } if (!(pollfd->revents & LWS_POLLIN)) break; #if defined(LWS_ROLE_H1) || defined(LWS_ROLE_H2) /* interpret the server response * * HTTP/1.1 101 Switching Protocols * Upgrade: websocket * Connection: Upgrade * Sec-WebSocket-Accept: me89jWimTRKTWwrS3aRrL53YZSo= * Sec-WebSocket-Nonce: AQIDBAUGBwgJCgsMDQ4PEC== * Sec-WebSocket-Protocol: chat * * we have to take some care here to only take from the * socket bytewise. The browser may (and has been seen to * in the case that onopen() performs websocket traffic) * coalesce both handshake response and websocket traffic * in one packet, since at that point the connection is * definitively ready from browser pov. */ while (wsi->http.ah->parser_state != WSI_PARSING_COMPLETE) { struct lws_tokens eb; int n, m, buffered; eb.token = NULL; eb.len = 0; buffered = lws_buflist_aware_read(pt, wsi, &eb, 0, __func__); lwsl_debug("%s: buflist-aware-read %d %d\n", __func__, buffered, eb.len); if (eb.len == LWS_SSL_CAPABLE_MORE_SERVICE) return 0; if (buffered < 0 || eb.len < 0) { cce = "read failed"; goto bail3; } if (!eb.len) return 0; n = eb.len; if (lws_parse(wsi, eb.token, &n)) { lwsl_warn("problems parsing header\n"); cce = "problems parsing header"; goto bail3; } m = eb.len - n; if (lws_buflist_aware_finished_consuming(wsi, &eb, m, buffered, __func__)) return -1; eb.token += m; eb.len -= m; if (n) { assert(wsi->http.ah->parser_state == WSI_PARSING_COMPLETE); break; } } /* * hs may also be coming in multiple packets, there is a 5-sec * libwebsocket timeout still active here too, so if parsing did * not complete just wait for next packet coming in this state */ if (wsi->http.ah->parser_state != WSI_PARSING_COMPLETE) break; #endif /* * otherwise deal with the handshake. If there's any * packet traffic already arrived we'll trigger poll() again * right away and deal with it that way */ return lws_client_interpret_server_handshake(wsi); bail3: lwsl_info("closing conn at LWS_CONNMODE...SERVER_REPLY\n"); if (cce) lwsl_info("reason: %s\n", cce); lws_inform_client_conn_fail(wsi, (void *)cce, strlen(cce)); lws_close_free_wsi(wsi, LWS_CLOSE_STATUS_NOSTATUS, "cbail3"); return -1; default: break; } return 0; } #if defined(LWS_ROLE_H1) || defined(LWS_ROLE_H2) int LWS_WARN_UNUSED_RESULT lws_http_transaction_completed_client(struct lws *wsi) { int n; lwsl_info("%s: wsi: %p (%s)\n", __func__, wsi, wsi->protocol->name); if (user_callback_handle_rxflow(wsi->protocol->callback, wsi, LWS_CALLBACK_COMPLETED_CLIENT_HTTP, wsi->user_space, NULL, 0)) { lwsl_debug("%s: Completed call returned nonzero (role 0x%lx)\n", __func__, (unsigned long)lwsi_role(wsi)); return -1; } wsi->http.rx_content_length = 0; /* * For h1, wsi may pass some assets on to a queued child and be * destroyed during this. */ n = _lws_generic_transaction_completed_active_conn(&wsi); if (wsi->http.ah) { if (wsi->client_mux_substream) /* * As an h2 client, once we did our transaction, that is * it for us. Further transactions will happen as new * SIDs on the connection. */ __lws_header_table_detach(wsi, 0); else if (!n) _lws_header_table_reset(wsi->http.ah); } if (!n || !wsi->http.ah) return 0; /* * H1: we can serialize the queued guys into the same ah * H2: everybody needs their own ah until their own STREAM_END */ /* otherwise set ourselves up ready to go again */ lwsi_set_state(wsi, LRS_WAITING_SERVER_REPLY); wsi->http.ah->parser_state = WSI_TOKEN_NAME_PART; wsi->http.ah->lextable_pos = 0; #if defined(LWS_WITH_CUSTOM_HEADERS) wsi->http.ah->unk_pos = 0; #endif lws_set_timeout(wsi, PENDING_TIMEOUT_AWAITING_SERVER_RESPONSE, wsi->context->timeout_secs); /* If we're (re)starting on headers, need other implied init */ wsi->http.ah->ues = URIES_IDLE; lwsi_set_state(wsi, LRS_H1C_ISSUE_HANDSHAKE2); lwsl_info("%s: %p: new queued transaction\n", __func__, wsi); lws_callback_on_writable(wsi); return 0; } unsigned int lws_http_client_http_response(struct lws *wsi) { if (wsi->http.ah && wsi->http.ah->http_response) return wsi->http.ah->http_response; return 0; } #endif #if defined(LWS_ROLE_H1) || defined(LWS_ROLE_H2) int lws_http_is_redirected_to_get(struct lws *wsi) { return wsi->redirected_to_get; } int lws_client_interpret_server_handshake(struct lws *wsi) { int n, port = 0, ssl = 0; int close_reason = LWS_CLOSE_STATUS_PROTOCOL_ERR; const char *prot, *ads = NULL, *path, *cce = NULL; struct allocated_headers *ah, *ah1; struct lws *nwsi = lws_get_network_wsi(wsi); char *p = NULL, *q; char new_path[300]; lws_free_set_NULL(wsi->stash); ah = wsi->http.ah; if (!wsi->do_ws) { /* we are being an http client... */ #if defined(LWS_ROLE_H2) if (wsi->client_h2_alpn || wsi->client_mux_substream) { lwsl_debug("%s: %p: transitioning to h2 client\n", __func__, wsi); lws_role_transition(wsi, LWSIFR_CLIENT, LRS_ESTABLISHED, &role_ops_h2); } else #endif { #if defined(LWS_ROLE_H1) { lwsl_debug("%s: %p: transitioning to h1 client\n", __func__, wsi); lws_role_transition(wsi, LWSIFR_CLIENT, LRS_ESTABLISHED, &role_ops_h1); } #else return -1; #endif } wsi->http.ah = ah; ah->http_response = 0; } /* * well, what the server sent looked reasonable for syntax. * Now let's confirm it sent all the necessary headers * * http (non-ws) client will expect something like this * * HTTP/1.0.200 * server:.libwebsockets * content-type:.text/html * content-length:.17703 * set-cookie:.test=LWS_1456736240_336776_COOKIE;Max-Age=360000 */ wsi->http.conn_type = HTTP_CONNECTION_KEEP_ALIVE; if (!wsi->client_mux_substream) { p = lws_hdr_simple_ptr(wsi, WSI_TOKEN_HTTP); /* if (wsi->do_ws && !p) { lwsl_info("no URI\n"); cce = "HS: URI missing"; goto bail3; } */ if (!p) { p = lws_hdr_simple_ptr(wsi, WSI_TOKEN_HTTP1_0); wsi->http.conn_type = HTTP_CONNECTION_CLOSE; } if (!p) { cce = "HS: URI missing"; lwsl_info("no URI\n"); goto bail3; } #if defined(LWS_ROLE_H2) } else { p = lws_hdr_simple_ptr(wsi, WSI_TOKEN_HTTP_COLON_STATUS); if (!p) { cce = "HS: :status missing"; lwsl_info("no status\n"); goto bail3; } #endif } #if !defined(LWS_ROLE_H2) if (!p) { cce = "HS: :status missing"; lwsl_info("no status\n"); goto bail3; } #endif n = atoi(p); if (ah) ah->http_response = n; if (!wsi->client_no_follow_redirect && #if defined(LWS_WITH_HTTP_PROXY) !wsi->http.proxy_clientside && #endif (n == 301 || n == 302 || n == 303 || n == 307 || n == 308)) { p = lws_hdr_simple_ptr(wsi, WSI_TOKEN_HTTP_LOCATION); if (!p) { cce = "HS: Redirect code but no Location"; goto bail3; } /* * Some redirect codes imply we have to change the method * used for the subsequent transaction, commonly POST -> * 303 -> GET. */ if (n == 303) { char *mp = lws_hdr_simple_ptr(wsi,_WSI_TOKEN_CLIENT_METHOD); int ml = lws_hdr_total_length(wsi, _WSI_TOKEN_CLIENT_METHOD); if (ml >= 3 && mp) { lwsl_info("%s: 303 switching to GET\n", __func__); memcpy(mp, "GET", 4); wsi->redirected_to_get = 1; wsi->http.ah->frags[wsi->http.ah->frag_index[ _WSI_TOKEN_CLIENT_METHOD]].len = 3; } } /* Relative reference absolute path */ if (p[0] == '/' || !strchr(p, ':')) { #if defined(LWS_WITH_TLS) ssl = nwsi->tls.use_ssl & LCCSCF_USE_SSL; #endif ads = lws_hdr_simple_ptr(wsi, _WSI_TOKEN_CLIENT_PEER_ADDRESS); port = nwsi->c_port; path = p; /* lws_client_reset expects leading / omitted */ if (*path == '/') path++; } /* Absolute (Full) URI */ else if (strchr(p, ':')) { if (lws_parse_uri(p, &prot, &ads, &port, &path)) { cce = "HS: URI did not parse"; goto bail3; } if (!strcmp(prot, "wss") || !strcmp(prot, "https")) ssl = LCCSCF_USE_SSL; } /* Relative reference relative path */ else { /* This doesn't try to calculate an absolute path, * that will be left to the server */ #if defined(LWS_WITH_TLS) ssl = nwsi->tls.use_ssl & LCCSCF_USE_SSL; #endif ads = lws_hdr_simple_ptr(wsi, _WSI_TOKEN_CLIENT_PEER_ADDRESS); port = wsi->c_port; /* +1 as lws_client_reset expects leading / omitted */ path = new_path + 1; if (lws_hdr_simple_ptr(wsi, _WSI_TOKEN_CLIENT_URI)) lws_strncpy(new_path, lws_hdr_simple_ptr(wsi, _WSI_TOKEN_CLIENT_URI), sizeof(new_path)); else { new_path[0] = '/'; new_path[1] = '\0'; } q = strrchr(new_path, '/'); if (q) lws_strncpy(q + 1, p, sizeof(new_path) - (q - new_path) - 1); else path = p; } #if defined(LWS_WITH_TLS) if ((wsi->tls.use_ssl & LCCSCF_USE_SSL) && !ssl) { cce = "HS: Redirect attempted SSL downgrade"; goto bail3; } #endif if (!ads) /* make coverity happy */ { cce = "no ads"; goto bail3; } if (!lws_client_reset(&wsi, ssl, ads, port, path, ads, 1)) { /* * There are two ways to fail out with NULL return... * simple, early problem where the wsi is intact, or * we went through with the reconnect attempt and the * wsi is already closed. In the latter case, the wsi * has been set to NULL additionally. */ lwsl_err("Redirect failed\n"); cce = "HS: Redirect failed"; /* coverity[reverse_inull] */ if (wsi) goto bail3; /* wsi has closed */ return 1; } return 0; } if (!wsi->do_ws) { /* if h1 KA is allowed, enable the queued pipeline guys */ if (!wsi->client_h2_alpn && !wsi->client_mux_substream) { /* ie, coming to this for the first time */ if (wsi->http.conn_type == HTTP_CONNECTION_KEEP_ALIVE) wsi->keepalive_active = 1; else { /* * Ugh... now the main http connection has seen * both sides, we learn the server doesn't * support keepalive. * * That means any guys queued on us are going * to have to be restarted from connect2 with * their own connections. */ /* * stick around telling any new guys they can't * pipeline to this server */ wsi->keepalive_rejected = 1; lws_vhost_lock(wsi->vhost); lws_start_foreach_dll_safe(struct lws_dll2 *, d, d1, wsi->dll2_cli_txn_queue_owner.head) { struct lws *ww = lws_container_of(d, struct lws, dll2_cli_txn_queue); /* remove him from our queue */ lws_dll2_remove(&ww->dll2_cli_txn_queue); /* give up on pipelining */ ww->client_pipeline = 0; /* go back to "trying to connect" state */ lws_role_transition(ww, LWSIFR_CLIENT, LRS_UNCONNECTED, #if defined(LWS_ROLE_H1) &role_ops_h1); #else #if defined (LWS_ROLE_H2) &role_ops_h2); #else &role_ops_raw); #endif #endif ww->user_space = NULL; } lws_end_foreach_dll_safe(d, d1); lws_vhost_unlock(wsi->vhost); } } #ifdef LWS_WITH_HTTP_PROXY wsi->http.perform_rewrite = 0; if (lws_hdr_total_length(wsi, WSI_TOKEN_HTTP_CONTENT_TYPE)) { if (!strncmp(lws_hdr_simple_ptr(wsi, WSI_TOKEN_HTTP_CONTENT_TYPE), "text/html", 9)) wsi->http.perform_rewrite = 0; } #endif /* allocate the per-connection user memory (if any) */ if (lws_ensure_user_space(wsi)) { lwsl_err("Problem allocating wsi user mem\n"); cce = "HS: OOM"; goto bail2; } /* he may choose to send us stuff in chunked transfer-coding */ wsi->chunked = 0; wsi->chunk_remaining = 0; /* ie, next thing is chunk size */ if (lws_hdr_total_length(wsi, WSI_TOKEN_HTTP_TRANSFER_ENCODING)) { wsi->chunked = !strcmp(lws_hdr_simple_ptr(wsi, WSI_TOKEN_HTTP_TRANSFER_ENCODING), "chunked"); /* first thing is hex, after payload there is crlf */ wsi->chunk_parser = ELCP_HEX; } wsi->http.content_length_given = 0; if (lws_hdr_total_length(wsi, WSI_TOKEN_HTTP_CONTENT_LENGTH)) { wsi->http.rx_content_length = atoll(lws_hdr_simple_ptr(wsi, WSI_TOKEN_HTTP_CONTENT_LENGTH)); lwsl_info("%s: incoming content length %llu\n", __func__, (unsigned long long) wsi->http.rx_content_length); wsi->http.rx_content_remain = wsi->http.rx_content_length; wsi->http.content_length_given = 1; } else { /* can't do 1.1 without a content length or chunked */ if (!wsi->chunked) wsi->http.conn_type = HTTP_CONNECTION_CLOSE; lwsl_debug("%s: no content length\n", __func__); } /* * we seem to be good to go, give client last chance to check * headers and OK it */ ah1 = wsi->http.ah; wsi->http.ah = ah; if (wsi->protocol->callback(wsi, LWS_CALLBACK_CLIENT_FILTER_PRE_ESTABLISH, wsi->user_space, NULL, 0)) { wsi->http.ah = ah1; cce = "HS: disallowed by client filter"; goto bail2; } /* clear his proxy connection timeout */ lws_set_timeout(wsi, NO_PENDING_TIMEOUT, 0); wsi->rxflow_change_to = LWS_RXFLOW_ALLOW; /* call him back to inform him he is up */ if (wsi->protocol->callback(wsi, LWS_CALLBACK_ESTABLISHED_CLIENT_HTTP, wsi->user_space, NULL, 0)) { wsi->http.ah = ah1; cce = "HS: disallowed at ESTABLISHED"; goto bail3; } wsi->http.ah = ah1; lwsl_info("%s: client connection up\n", __func__); /* * Did we get a response from the server with an explicit * content-length of zero? If so, this transaction is already * completed at the end of the header processing... */ if (lws_hdr_total_length(wsi, WSI_TOKEN_HTTP_CONTENT_LENGTH) && !wsi->http.rx_content_length) return !!lws_http_transaction_completed_client(wsi); return 0; } #if defined(LWS_ROLE_WS) switch (lws_client_ws_upgrade(wsi, &cce)) { case 2: goto bail2; case 3: goto bail3; } return 0; #endif bail3: close_reason = LWS_CLOSE_STATUS_NOSTATUS; bail2: if (wsi->protocol) { n = 0; if (cce) n = (int)strlen(cce); lws_inform_client_conn_fail(wsi, (void *)cce, (unsigned int)n); } lwsl_info("closing connection (prot %s) " "due to bail2 connection error: %s\n", wsi->protocol ? wsi->protocol->name : "unknown", cce); /* closing will free up his parsing allocations */ lws_close_free_wsi(wsi, close_reason, "c hs interp"); return 1; } #endif /* * set the boundary string and the content-type for client multipart mime */ uint8_t * lws_http_multipart_headers(struct lws *wsi, uint8_t *p) { char buf[10], arg[48]; int n; lws_get_random(wsi->context, (uint8_t *)buf, sizeof(buf)); lws_b64_encode_string(buf, sizeof(buf), wsi->http.multipart_boundary, sizeof(wsi->http.multipart_boundary)); n = lws_snprintf(arg, sizeof(arg), "multipart/form-data; boundary=%s", wsi->http.multipart_boundary); if (lws_add_http_header_by_token(wsi, WSI_TOKEN_HTTP_CONTENT_TYPE, (uint8_t *)arg, n, &p, p + 100)) return NULL; wsi->http.multipart = wsi->http.multipart_issue_boundary = 1; lws_client_http_body_pending(wsi, 1); return p; } int lws_client_http_multipart(struct lws *wsi, const char *name, const char *filename, const char *content_type, char **p, char *end) { /* * Client conn must have been created with LCCSCF_HTTP_MULTIPART_MIME * flag to use this api */ assert(wsi->http.multipart); if (!name) { *p += lws_snprintf((char *)(*p), lws_ptr_diff(end, p), "\xd\xa--%s--\xd\xa", wsi->http.multipart_boundary); return 0; } if (wsi->client_subsequent_mime_part) *p += lws_snprintf((char *)(*p), lws_ptr_diff(end, p), "\xd\xa"); wsi->client_subsequent_mime_part = 1; *p += lws_snprintf((char *)(*p), lws_ptr_diff(end, p), "--%s\xd\xa" "Content-Disposition: form-data; " "name=\"%s\"", wsi->http.multipart_boundary, name); if (filename) *p += lws_snprintf((char *)(*p), lws_ptr_diff(end, p), "; filename=\"%s\"", filename); if (content_type) *p += lws_snprintf((char *)(*p), lws_ptr_diff(end, p), "\xd\xa" "Content-Type: %s", content_type); *p += lws_snprintf((char *)(*p), lws_ptr_diff(end, p), "\xd\xa\xd\xa"); return *p == end; } char * lws_generate_client_handshake(struct lws *wsi, char *pkt) { const char *meth, *pp = lws_hdr_simple_ptr(wsi, _WSI_TOKEN_CLIENT_SENT_PROTOCOLS); char *p = pkt, *p1; meth = lws_hdr_simple_ptr(wsi, _WSI_TOKEN_CLIENT_METHOD); if (!meth) { meth = "GET"; wsi->do_ws = 1; } else { wsi->do_ws = 0; } if (!strcmp(meth, "RAW")) { lws_set_timeout(wsi, NO_PENDING_TIMEOUT, 0); lwsl_notice("client transition to raw\n"); if (pp) { const struct lws_protocols *pr; pr = lws_vhost_name_to_protocol(wsi->vhost, pp); if (!pr) { lwsl_err("protocol %s not enabled on vhost\n", pp); return NULL; } lws_bind_protocol(wsi, pr, __func__); } if ((wsi->protocol->callback)(wsi, LWS_CALLBACK_RAW_ADOPT, wsi->user_space, NULL, 0)) return NULL; lws_role_transition(wsi, LWSIFR_CLIENT, LRS_ESTABLISHED, &role_ops_raw_skt); lws_header_table_detach(wsi, 1); return NULL; } /* * 04 example client handshake * * 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 * Sec-WebSocket-Version: 4 */ p += lws_snprintf(p, 2048, "%s %s HTTP/1.1\x0d\x0a", meth, lws_hdr_simple_ptr(wsi, _WSI_TOKEN_CLIENT_URI)); p += lws_snprintf(p, 64, "Pragma: no-cache\x0d\x0a" "Cache-Control: no-cache\x0d\x0a"); p += lws_snprintf(p, 128, "Host: %s\x0d\x0a", lws_hdr_simple_ptr(wsi, _WSI_TOKEN_CLIENT_HOST)); if (lws_hdr_simple_ptr(wsi, _WSI_TOKEN_CLIENT_ORIGIN)) { if (lws_check_opt(wsi->context->options, LWS_SERVER_OPTION_JUST_USE_RAW_ORIGIN)) p += lws_snprintf(p, 128, "Origin: %s\x0d\x0a", lws_hdr_simple_ptr(wsi, _WSI_TOKEN_CLIENT_ORIGIN)); else p += lws_snprintf(p, 128, "Origin: http://%s\x0d\x0a", lws_hdr_simple_ptr(wsi, _WSI_TOKEN_CLIENT_ORIGIN)); } if (wsi->flags & LCCSCF_HTTP_MULTIPART_MIME) { p1 = (char *)lws_http_multipart_headers(wsi, (uint8_t *)p); if (!p1) return NULL; p = p1; } #if defined(LWS_WITH_HTTP_PROXY) if (wsi->parent && lws_hdr_total_length(wsi->parent, WSI_TOKEN_HTTP_CONTENT_LENGTH)) { p += lws_snprintf(p, 128, "Content-Length: %s\x0d\x0a", lws_hdr_simple_ptr(wsi->parent, WSI_TOKEN_HTTP_CONTENT_LENGTH)); if (atoi(lws_hdr_simple_ptr(wsi->parent, WSI_TOKEN_HTTP_CONTENT_LENGTH))) wsi->client_http_body_pending = 1; } if (wsi->parent && lws_hdr_total_length(wsi->parent, WSI_TOKEN_HTTP_AUTHORIZATION)) { p += lws_snprintf(p, 128, "Authorization: %s\x0d\x0a", lws_hdr_simple_ptr(wsi->parent, WSI_TOKEN_HTTP_AUTHORIZATION)); } if (wsi->parent && lws_hdr_total_length(wsi->parent, WSI_TOKEN_HTTP_CONTENT_TYPE)) { p += lws_snprintf(p, 128, "Content-Type: %s\x0d\x0a", lws_hdr_simple_ptr(wsi->parent, WSI_TOKEN_HTTP_CONTENT_TYPE)); } #endif #if defined(LWS_ROLE_WS) if (wsi->do_ws) { const char *conn1 = ""; // if (!wsi->client_pipeline) // conn1 = "close, "; p = lws_generate_client_ws_handshake(wsi, p, conn1); } else #endif { if (!wsi->client_pipeline) p += lws_snprintf(p, 64, "connection: close\x0d\x0a"); } /* give userland a chance to append, eg, cookies */ if (wsi->protocol->callback(wsi, LWS_CALLBACK_CLIENT_APPEND_HANDSHAKE_HEADER, wsi->user_space, &p, (pkt + wsi->context->pt_serv_buf_size) - p - 12)) return NULL; if (wsi->flags & LCCSCF_HTTP_X_WWW_FORM_URLENCODED) { p += lws_snprintf(p, 128, "Content-Type: application/x-www-form-urlencoded\x0d\x0a"); p += lws_snprintf(p, 128, "Content-Length: %lu\x0d\x0a", wsi->http.writeable_len); lws_client_http_body_pending(wsi, 1); } p += lws_snprintf(p, 4, "\x0d\x0a"); if (wsi->client_http_body_pending) lws_callback_on_writable(wsi); // puts(pkt); return p; } #if defined(LWS_ROLE_H1) || defined(LWS_ROLE_H2) #if defined(LWS_WITH_HTTP_BASIC_AUTH) int lws_http_basic_auth_gen(const char *user, const char *pw, char *buf, size_t len) { size_t n = strlen(user), m = strlen(pw); char b[128]; if (len < 6 + ((4 * (n + m + 1)) / 3) + 1) return 1; memcpy(buf, "Basic ", 6); n = lws_snprintf(b, sizeof(b), "%s:%s", user, pw); if (n >= sizeof(b) - 2) return 2; lws_b64_encode_string(b, (int)n, buf + 6, (int)len - 6); buf[len - 1] = '\0'; return 0; } #endif int lws_http_client_read(struct lws *wsi, char **buf, int *len) { struct lws_context_per_thread *pt = &wsi->context->pt[(int)wsi->tsi]; struct lws_tokens eb; int buffered, n, consumed = 0; /* * If the caller provided a non-NULL *buf and nonzero *len, we should * use that as the buffer for the read action, limititing it to *len * (actual payload will be less if chunked headers inside). * * If it's NULL / 0 length, buflist_aware_read will use the pt_serv_buf */ eb.token = (unsigned char *)*buf; eb.len = *len; buffered = lws_buflist_aware_read(pt, wsi, &eb, 0, __func__); *buf = (char *)eb.token; /* may be pointing to buflist or pt_serv_buf */ *len = 0; /* * we're taking on responsibility for handling used / unused eb * when we leave, via lws_buflist_aware_finished_consuming() */ // lwsl_notice("%s: eb.len %d ENTRY chunk remaining %d\n", __func__, eb.len, // wsi->chunk_remaining); /* allow the source to signal he has data again next time */ if (lws_change_pollfd(wsi, 0, LWS_POLLIN)) return -1; if (buffered < 0) { lwsl_debug("%s: SSL capable error\n", __func__); return -1; } if (eb.len <= 0) return 0; *len = eb.len; wsi->client_rx_avail = 0; /* * server may insist on transfer-encoding: chunked, * so http client must deal with it */ spin_chunks: //lwsl_notice("%s: len %d SPIN chunk remaining %d\n", __func__, *len, // wsi->chunk_remaining); while (wsi->chunked && (wsi->chunk_parser != ELCP_CONTENT) && *len) { switch (wsi->chunk_parser) { case ELCP_HEX: if ((*buf)[0] == '\x0d') { wsi->chunk_parser = ELCP_CR; break; } n = char_to_hex((*buf)[0]); if (n < 0) { lwsl_err("%s: chunking failure A\n", __func__); return -1; } wsi->chunk_remaining <<= 4; wsi->chunk_remaining |= n; break; case ELCP_CR: if ((*buf)[0] != '\x0a') { lwsl_err("%s: chunking failure B\n", __func__); return -1; } if (wsi->chunk_remaining) { wsi->chunk_parser = ELCP_CONTENT; //lwsl_notice("starting chunk size %d (block rem %d)\n", // wsi->chunk_remaining, *len); break; } wsi->chunk_parser = ELCP_TRAILER_CR; break; case ELCP_CONTENT: break; case ELCP_POST_CR: if ((*buf)[0] != '\x0d') { lwsl_err("%s: chunking failure C\n", __func__); lwsl_hexdump_err(*buf, *len); return -1; } wsi->chunk_parser = ELCP_POST_LF; break; case ELCP_POST_LF: if ((*buf)[0] != '\x0a') { lwsl_err("%s: chunking failure D\n", __func__); return -1; } wsi->chunk_parser = ELCP_HEX; wsi->chunk_remaining = 0; break; case ELCP_TRAILER_CR: if ((*buf)[0] != '\x0d') { lwsl_err("%s: chunking failure F\n", __func__); lwsl_hexdump_err(*buf, *len); return -1; } wsi->chunk_parser = ELCP_TRAILER_LF; break; case ELCP_TRAILER_LF: if ((*buf)[0] != '\x0a') { lwsl_err("%s: chunking failure F\n", __func__); lwsl_hexdump_err(*buf, *len); return -1; } (*buf)++; (*len)--; consumed++; lwsl_info("final chunk\n"); goto completed; } (*buf)++; (*len)--; consumed++; } if (wsi->chunked && !wsi->chunk_remaining) goto account_and_ret; if (wsi->http.rx_content_remain && wsi->http.rx_content_remain < (unsigned int)*len) n = (int)wsi->http.rx_content_remain; else n = *len; if (wsi->chunked && wsi->chunk_remaining && wsi->chunk_remaining < n) n = wsi->chunk_remaining; #if defined(LWS_WITH_HTTP_PROXY) && defined(LWS_WITH_HUBBUB) /* hubbub */ if (wsi->http.perform_rewrite) lws_rewrite_parse(wsi->http.rw, (unsigned char *)*buf, n); else #endif { if ( #if defined(LWS_WITH_HTTP_PROXY) !wsi->protocol_bind_balance == !!wsi->http.proxy_clientside #else !!wsi->protocol_bind_balance #endif ) { if (user_callback_handle_rxflow(wsi->protocol->callback, wsi, LWS_CALLBACK_RECEIVE_CLIENT_HTTP_READ, wsi->user_space, *buf, n)) { lwsl_info("%s: RECEIVE_CLIENT_HTTP_READ returned -1\n", __func__); return -1; } } else lwsl_notice("%s: swallowed read (%d)\n", __func__, n); } (*buf) += n; *len -= n; if (wsi->chunked && wsi->chunk_remaining) wsi->chunk_remaining -= n; //lwsl_notice("chunk_remaining <- %d, block remaining %d\n", // wsi->chunk_remaining, *len); consumed += n; //eb.token += n; //eb.len -= n; if (wsi->chunked && !wsi->chunk_remaining) wsi->chunk_parser = ELCP_POST_CR; if (wsi->chunked && *len) goto spin_chunks; if (wsi->chunked) goto account_and_ret; /* if we know the content length, decrement the content remaining */ if (wsi->http.rx_content_length > 0) wsi->http.rx_content_remain -= n; // lwsl_notice("rx_content_remain %lld, rx_content_length %lld, giv %d\n", // wsi->http.rx_content_remain, wsi->http.rx_content_length, // wsi->http.content_length_given); if (wsi->http.rx_content_remain || !wsi->http.content_length_given) goto account_and_ret; completed: if (lws_http_transaction_completed_client(wsi)) { lwsl_notice("%s: transaction completed says -1\n", __func__); return -1; } account_and_ret: // lwsl_warn("%s: on way out, consuming %d / %d\n", __func__, consumed, eb.len); if (lws_buflist_aware_finished_consuming(wsi, &eb, consumed, buffered, __func__)) return -1; return 0; } #endif