mirror of
https://github.com/warmcat/libwebsockets.git
synced 2025-03-09 00:00:04 +01:00

There are a few build options that are trying to keep and report various statistics - DETAILED_LATENCY - SERVER_STATUS - WITH_STATS remove all those and establish a generic rplacement, lws_metrics. lws_metrics makes its stats available via an lws_system ops function pointer that the user code can set. Openmetrics export is supported, for, eg, prometheus scraping.
1706 lines
43 KiB
C
1706 lines
43 KiB
C
/*
|
|
* libwebsockets - small server side websockets and web server implementation
|
|
*
|
|
* Copyright (C) 2010 - 2020 Andy Green <andy@warmcat.com>
|
|
*
|
|
* 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"
|
|
|
|
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_http_client_socket_service(struct lws *wsi, struct lws_pollfd *pollfd)
|
|
{
|
|
struct lws_context *context = wsi->a.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: %s: WAITING_DNS\n", __func__, lws_wsi_tag(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)
|
|
if (lws_client_connect_3_connect(wsi, NULL, NULL, 0, NULL) == NULL) {
|
|
lwsl_client("closed\n");
|
|
return -1;
|
|
}
|
|
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 conn %s (fd=%d) dead\n",
|
|
lws_wsi_tag(wsi), pollfd->fd);
|
|
|
|
cce = "proxy conn dead";
|
|
goto bail3;
|
|
}
|
|
|
|
n = (int)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);
|
|
|
|
/* fallthrough */
|
|
|
|
#endif
|
|
|
|
case LRS_H1C_ISSUE_HANDSHAKE:
|
|
|
|
lwsl_debug("%s: LRS_H1C_ISSUE_HANDSHAKE\n", __func__);
|
|
|
|
/*
|
|
* 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_ROLE_H2) || defined(LWS_WITH_TLS)
|
|
if (
|
|
#if defined(LWS_WITH_TLS)
|
|
!(wsi->tls.use_ssl & LCCSCF_USE_SSL)
|
|
#endif
|
|
#if defined(LWS_ROLE_H2) && defined(LWS_WITH_TLS)
|
|
&&
|
|
#endif
|
|
#if defined(LWS_ROLE_H2)
|
|
!(wsi->flags & LCCSCF_H2_PRIOR_KNOWLEDGE)
|
|
#endif
|
|
)
|
|
goto hs2;
|
|
#endif
|
|
|
|
#if defined(LWS_WITH_TLS)
|
|
n = lws_client_create_tls(wsi, &cce, 1);
|
|
if (n == CCTLS_RETURN_ERROR)
|
|
goto bail3;
|
|
if (n == CCTLS_RETURN_RETRY)
|
|
return 0;
|
|
|
|
/*
|
|
* lws_client_create_tls() can already have done the
|
|
* whole tls setup and preface send... if so he set our state
|
|
* to LRS_H1C_ISSUE_HANDSHAKE2... let's proceed but be prepared
|
|
* to notice our state and not resend the preface...
|
|
*/
|
|
|
|
lwsl_debug("%s: LRS_H1C_ISSUE_HANDSHAKE fallthru\n", __func__);
|
|
|
|
/* 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;
|
|
if (wsi->flags & LCCSCF_H2_PRIOR_KNOWLEDGE) {
|
|
lwsl_info("h2 prior knowledge\n");
|
|
lws_role_call_alpn_negotiated(wsi, "h2");
|
|
}
|
|
}
|
|
#endif
|
|
|
|
#if defined (LWS_WITH_HTTP2)
|
|
if (wsi->client_h2_alpn && lwsi_state(wsi) != LRS_H1C_ISSUE_HANDSHAKE2) {
|
|
/*
|
|
* We connected to the server and set up tls and
|
|
* negotiated "h2" or connected as clear text
|
|
* with http/2 prior knowledge.
|
|
*
|
|
* So this is it, we are an h2 nwsi client connection
|
|
* now, not an h1 client connection.
|
|
*/
|
|
|
|
#if defined(LWS_WITH_TLS)
|
|
if (wsi->tls.use_ssl & LCCSCF_USE_SSL)
|
|
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,
|
|
(int)context->timeout_secs);
|
|
|
|
break;
|
|
}
|
|
#endif
|
|
|
|
/* fallthru */
|
|
|
|
case LRS_H1C_ISSUE_HANDSHAKE2:
|
|
|
|
hs2:
|
|
|
|
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: %s: sending headers "
|
|
"(wsistate 0x%lx), w sock %d\n",
|
|
__func__, lws_wsi_tag(wsi),
|
|
(unsigned long)wsi->wsistate, wsi->desc.sockfd);
|
|
|
|
n = lws_ssl_capable_write(wsi, (unsigned char *)sb, lws_ptr_diff_size_t(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,
|
|
(int)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;
|
|
wsi->http.ah->unk_pos = 0;
|
|
/* 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,
|
|
(int)wsi->a.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;
|
|
wsi->http.ah->unk_pos = 0;
|
|
#endif
|
|
lwsi_set_state(wsi, LRS_WAITING_SERVER_REPLY);
|
|
lws_set_timeout(wsi, PENDING_TIMEOUT_AWAITING_SERVER_RESPONSE,
|
|
(int)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) {
|
|
|
|
if (lws_buflist_total_len(&wsi->buflist))
|
|
lws_set_timeout(wsi, PENDING_TIMEOUT_CLOSE_ACK, 3);
|
|
else {
|
|
lwsl_debug("Server conn %s (fd=%d) dead\n",
|
|
lws_wsi_tag(wsi), pollfd->fd);
|
|
cce = "Peer hung up";
|
|
goto bail3;
|
|
}
|
|
}
|
|
|
|
if (pollfd->revents & LWS_POLLOUT)
|
|
if (lws_change_pollfd(wsi, LWS_POLLOUT, 0))
|
|
return -1;
|
|
|
|
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;
|
|
|
|
/*
|
|
* coverity: uncomment if extended
|
|
*
|
|
* 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("%s: closing conn at LWS_CONNMODE...SERVER_REPLY, %s, state 0x%x\n",
|
|
__func__, lws_wsi_tag(wsi), lwsi_state(wsi));
|
|
if (cce)
|
|
lwsl_info("reason: %s\n", cce);
|
|
else
|
|
cce = "unknown";
|
|
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)
|
|
{
|
|
struct lws_context_per_thread *pt = &wsi->a.context->pt[(int)wsi->tsi];
|
|
int n;
|
|
|
|
lwsl_info("%s: %s (%s)\n", __func__, lws_wsi_tag(wsi),
|
|
wsi->a.protocol->name);
|
|
|
|
// if (wsi->http.ah && wsi->http.ah->http_response)
|
|
/* we're only judging if any (200, or 500 etc) http txn completed */
|
|
lws_metrics_caliper_report(wsi->cal_conn, METRES_GO);
|
|
|
|
if (user_callback_handle_rxflow(wsi->a.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.
|
|
*/
|
|
lws_pt_lock(pt, __func__);
|
|
n = _lws_generic_transaction_completed_active_conn(&wsi, 1);
|
|
lws_pt_unlock(pt);
|
|
|
|
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;
|
|
wsi->http.ah->unk_pos = 0;
|
|
|
|
lws_set_timeout(wsi, PENDING_TIMEOUT_AWAITING_SERVER_RESPONSE,
|
|
(int)wsi->a.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: %s: new queued transaction\n", __func__, lws_wsi_tag(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, *simp;
|
|
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: %s: transitioning to h2 client\n",
|
|
__func__, lws_wsi_tag(wsi));
|
|
lws_role_transition(wsi, LWSIFR_CLIENT,
|
|
LRS_ESTABLISHED, &role_ops_h2);
|
|
} else
|
|
#endif
|
|
{
|
|
#if defined(LWS_ROLE_H1)
|
|
{
|
|
lwsl_debug("%s: %s: transitioning to h1 client\n",
|
|
__func__, lws_wsi_tag(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 = (unsigned int)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;
|
|
}
|
|
|
|
/* let's let the user code know, if he cares */
|
|
|
|
if (wsi->a.protocol->callback(wsi,
|
|
LWS_CALLBACK_CLIENT_HTTP_REDIRECT,
|
|
wsi->user_space, p, (unsigned int)n)) {
|
|
cce = "HS: user code rejected redirect";
|
|
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) -
|
|
(unsigned int)(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 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->a.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->a.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
|
|
|
|
/* 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)) {
|
|
simp = lws_hdr_simple_ptr(wsi, WSI_TOKEN_HTTP_TRANSFER_ENCODING);
|
|
|
|
/* cannot be NULL, since it has nonzero length... coverity */
|
|
if (!simp)
|
|
goto bail2;
|
|
wsi->chunked = !strcmp(simp, "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)) {
|
|
simp = lws_hdr_simple_ptr(wsi, WSI_TOKEN_HTTP_CONTENT_LENGTH);
|
|
|
|
/* cannot be NULL, since it has nonzero length... coverity */
|
|
if (!simp)
|
|
goto bail2;
|
|
|
|
wsi->http.rx_content_length = (lws_filepos_t)atoll(simp);
|
|
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__);
|
|
}
|
|
|
|
if (wsi->do_ws) {
|
|
/*
|
|
* Give one last opportunity to ws protocols to inspect server reply
|
|
* before the ws upgrade code discard it. ie: download reply body in case
|
|
* of any other response code than 101.
|
|
*/
|
|
if (wsi->a.protocol->callback(wsi,
|
|
LWS_CALLBACK_ESTABLISHED_CLIENT_HTTP,
|
|
wsi->user_space, NULL, 0)) {
|
|
|
|
cce = "HS: disallowed by client filter";
|
|
goto bail2;
|
|
}
|
|
} else {
|
|
/* 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;
|
|
}
|
|
|
|
|
|
/*
|
|
* 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->a.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->a.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: %s: client conn up\n", __func__, lws_wsi_tag(wsi));
|
|
|
|
/*
|
|
* Did we get a response from the server with an explicit
|
|
* content-length of zero? If so, and it's not H2 which will
|
|
* notice it via END_STREAM, this transaction is already
|
|
* completed at the end of the header processing...
|
|
*/
|
|
if (!wsi->mux_substream && !wsi->client_mux_substream &&
|
|
lws_hdr_total_length(wsi, WSI_TOKEN_HTTP_CONTENT_LENGTH) &&
|
|
!wsi->http.rx_content_length)
|
|
return !!lws_http_transaction_completed_client(wsi);
|
|
|
|
/*
|
|
* We can also get a case where it's http/1 and there's no
|
|
* content-length at all, so anything that comes is the body
|
|
* until it hangs up on us. With that situation, hanging up
|
|
* on us past this point should generate a valid
|
|
* LWS_CALLBACK_COMPLETED_CLIENT_HTTP.
|
|
*
|
|
* In that situation, he can't pipeline because in h1 there's
|
|
* no post-header in-band way to signal the end of the
|
|
* transaction except hangup.
|
|
*
|
|
* lws_http_transaction_completed_client() is the right guy to
|
|
* issue it when we see the peer has hung up on us.
|
|
*/
|
|
|
|
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->a.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->a.protocol ?
|
|
wsi->a.protocol->name : "unknown", cce);
|
|
|
|
/* closing will free up his parsing allocations */
|
|
lws_close_free_wsi(wsi, (enum lws_close_status)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;
|
|
|
|
if (lws_get_random(wsi->a.context, (uint8_t *)buf, sizeof(buf)) !=
|
|
sizeof(buf))
|
|
return NULL;
|
|
|
|
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_size_t(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_size_t(end, *p), "\xd\xa");
|
|
wsi->client_subsequent_mime_part = 1;
|
|
|
|
*p += lws_snprintf((char *)(*p), lws_ptr_diff_size_t(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_size_t(end, *p),
|
|
"; filename=\"%s\"", filename);
|
|
|
|
if (content_type)
|
|
*p += lws_snprintf((char *)(*p), lws_ptr_diff_size_t(end, *p), "\xd\xa"
|
|
"Content-Type: %s", content_type);
|
|
|
|
*p += lws_snprintf((char *)(*p), lws_ptr_diff_size_t(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, *end = p + wsi->a.context->pt_serv_buf_size;
|
|
|
|
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->a.vhost, pp);
|
|
|
|
if (!pr) {
|
|
lwsl_err("protocol %s not enabled on vhost\n",
|
|
pp);
|
|
return NULL;
|
|
}
|
|
|
|
lws_bind_protocol(wsi, pr, __func__);
|
|
}
|
|
|
|
if ((wsi->a.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, lws_ptr_diff_size_t(end, p),
|
|
"%s %s HTTP/1.1\x0d\x0a", meth,
|
|
lws_hdr_simple_ptr(wsi, _WSI_TOKEN_CLIENT_URI));
|
|
|
|
p += lws_snprintf(p, lws_ptr_diff_size_t(end, p),
|
|
"Pragma: no-cache\x0d\x0a"
|
|
"Cache-Control: no-cache\x0d\x0a");
|
|
|
|
p += lws_snprintf(p, lws_ptr_diff_size_t(end, p),
|
|
"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->a.context->options,
|
|
LWS_SERVER_OPTION_JUST_USE_RAW_ORIGIN))
|
|
p += lws_snprintf(p, lws_ptr_diff_size_t(end, p),
|
|
"Origin: %s\x0d\x0a",
|
|
lws_hdr_simple_ptr(wsi,
|
|
_WSI_TOKEN_CLIENT_ORIGIN));
|
|
else
|
|
p += lws_snprintf(p, lws_ptr_diff_size_t(end, p),
|
|
"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, lws_ptr_diff_size_t(end, p),
|
|
"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, lws_ptr_diff_size_t(end, p),
|
|
"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, lws_ptr_diff_size_t(end, p),
|
|
"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->a.protocol->callback(wsi,
|
|
LWS_CALLBACK_CLIENT_APPEND_HANDSHAKE_HEADER,
|
|
wsi->user_space, &p,
|
|
(unsigned int)((pkt + wsi->a.context->pt_serv_buf_size) - p - 12)))
|
|
return NULL;
|
|
|
|
if (wsi->flags & LCCSCF_HTTP_X_WWW_FORM_URLENCODED) {
|
|
p += lws_snprintf(p, lws_ptr_diff_size_t(end, p), "Content-Type: application/x-www-form-urlencoded\x0d\x0a");
|
|
p += lws_snprintf(p, lws_ptr_diff_size_t(end, p), "Content-Length: %lu\x0d\x0a", wsi->http.writeable_len);
|
|
lws_client_http_body_pending(wsi, 1);
|
|
}
|
|
|
|
p += lws_snprintf(p, lws_ptr_diff_size_t(end, p), "\x0d\x0a");
|
|
|
|
if (wsi->client_http_body_pending)
|
|
lws_callback_on_writable(wsi);
|
|
|
|
lws_metrics_caliper_bind(wsi->cal_conn, wsi->a.context->mt_http_txn);
|
|
|
|
// 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 = (unsigned int)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->a.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__);
|
|
|
|
if (wsi->http.ah &&
|
|
wsi->http.ah->parser_state == WSI_PARSING_COMPLETE &&
|
|
!lws_hdr_total_length(wsi, WSI_TOKEN_HTTP_CONTENT_LENGTH))
|
|
/*
|
|
* We had the headers from this stream, but as there
|
|
* was no content-length: we had to wait until the
|
|
* stream ended to inform the user code the transaction
|
|
* has completed to the best of our knowledge
|
|
*/
|
|
if (lws_http_transaction_completed_client(wsi))
|
|
/*
|
|
* We're going to close anyway, but that api has
|
|
* warn_unused_result
|
|
*/
|
|
return -1;
|
|
|
|
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, (unsigned int)*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, (unsigned int)*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, (unsigned int)*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
|
|
) {
|
|
int q;
|
|
|
|
q = user_callback_handle_rxflow(wsi->a.protocol->callback,
|
|
wsi, LWS_CALLBACK_RECEIVE_CLIENT_HTTP_READ,
|
|
wsi->user_space, *buf, (unsigned int)n);
|
|
if (q) {
|
|
lwsl_info("%s: RECEIVE_CLIENT_HTTP_READ returned %d\n",
|
|
__func__, q);
|
|
|
|
return q;
|
|
}
|
|
} 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 -= (unsigned int)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_info("%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
|
|
|
|
static uint8_t hnames2[] = {
|
|
_WSI_TOKEN_CLIENT_ORIGIN,
|
|
_WSI_TOKEN_CLIENT_SENT_PROTOCOLS,
|
|
_WSI_TOKEN_CLIENT_METHOD,
|
|
_WSI_TOKEN_CLIENT_IFACE,
|
|
_WSI_TOKEN_CLIENT_ALPN
|
|
};
|
|
|
|
/**
|
|
* lws_client_reset() - retarget a connected wsi to start over with a new
|
|
* connection (ie, redirect)
|
|
* this only works if still in HTTP, ie, not upgraded yet
|
|
* wsi: connection to reset
|
|
* address: network address of the new server
|
|
* port: port to connect to
|
|
* path: uri path to connect to on the new server
|
|
* host: host header to send to the new server
|
|
*/
|
|
struct lws *
|
|
lws_client_reset(struct lws **pwsi, int ssl, const char *address, int port,
|
|
const char *path, const char *host, char weak)
|
|
{
|
|
#if defined(LWS_ROLE_WS)
|
|
struct _lws_websocket_related *ws;
|
|
#endif
|
|
char *stash, *p;
|
|
struct lws *wsi;
|
|
size_t size = 0;
|
|
int n;
|
|
|
|
if (!pwsi)
|
|
return NULL;
|
|
|
|
wsi = *pwsi;
|
|
|
|
lwsl_debug("%s: %s: redir %d: %s\n", __func__, lws_wsi_tag(wsi),
|
|
wsi->redirects, address);
|
|
|
|
if (wsi->redirects == 3) {
|
|
lwsl_err("%s: Too many redirects\n", __func__);
|
|
return NULL;
|
|
}
|
|
wsi->redirects++;
|
|
|
|
/*
|
|
* goal is to close our role part, close the sockfd, detach the ah
|
|
* but leave our wsi extant and still bound to whatever vhost it was
|
|
*/
|
|
|
|
for (n = 0; n < (int)LWS_ARRAY_SIZE(hnames2); n++)
|
|
size += (unsigned int)lws_hdr_total_length(wsi, hnames2[n]) + 1u;
|
|
|
|
if (size < (size_t)lws_hdr_total_length(wsi, _WSI_TOKEN_CLIENT_URI) + 1)
|
|
size = (unsigned int)lws_hdr_total_length(wsi, _WSI_TOKEN_CLIENT_URI) + 1u;
|
|
|
|
/*
|
|
* The incoming address and host can be from inside the existing ah
|
|
* we are going to detach and reattch
|
|
*/
|
|
|
|
size += strlen(path) + 1 + strlen(address) + 1 + strlen(host) + 1 + 1;
|
|
|
|
p = stash = lws_malloc(size, __func__);
|
|
if (!stash)
|
|
return NULL;
|
|
|
|
/*
|
|
* _WSI_TOKEN_CLIENT_ORIGIN,
|
|
* _WSI_TOKEN_CLIENT_SENT_PROTOCOLS,
|
|
* _WSI_TOKEN_CLIENT_METHOD,
|
|
* _WSI_TOKEN_CLIENT_IFACE,
|
|
* _WSI_TOKEN_CLIENT_ALPN
|
|
* address
|
|
* host
|
|
* path
|
|
*/
|
|
|
|
for (n = 0; n < (int)LWS_ARRAY_SIZE(hnames2); n++)
|
|
if (lws_hdr_total_length(wsi, hnames2[n]) &&
|
|
lws_hdr_simple_ptr(wsi, hnames2[n])) {
|
|
memcpy(p, lws_hdr_simple_ptr(wsi, hnames2[n]), (size_t)(
|
|
lws_hdr_total_length(wsi, hnames2[n]) + 1));
|
|
p += (size_t)(lws_hdr_total_length(wsi, hnames2[n]) + 1);
|
|
} else
|
|
*p++ = '\0';
|
|
|
|
memcpy(p, address, strlen(address) + (size_t)1);
|
|
address = p;
|
|
p += strlen(address) + 1;
|
|
memcpy(p, host, strlen(host) + (size_t)1);
|
|
host = p;
|
|
p += strlen(host) + 1;
|
|
memcpy(p, path, strlen(path) + (size_t)1);
|
|
path = p;
|
|
|
|
if (!port) {
|
|
lwsl_info("%s: forcing port 443\n", __func__);
|
|
|
|
port = 443;
|
|
ssl = 1;
|
|
}
|
|
|
|
lwsl_info("redirect ads='%s', port=%d, path='%s', ssl = %d, pifds %d\n",
|
|
address, port, path, ssl, wsi->position_in_fds_table);
|
|
|
|
__remove_wsi_socket_from_fds(wsi);
|
|
#if defined(LWS_ROLE_WS)
|
|
if (weak) {
|
|
ws = wsi->ws;
|
|
wsi->ws = NULL;
|
|
}
|
|
#endif
|
|
__lws_reset_wsi(wsi); /* detaches ah here */
|
|
#if defined(LWS_ROLE_WS)
|
|
if (weak)
|
|
wsi->ws = ws;
|
|
#endif
|
|
wsi->client_pipeline = 1;
|
|
|
|
/* close the connection by hand */
|
|
|
|
#if defined(LWS_WITH_TLS)
|
|
lws_ssl_close(wsi);
|
|
#endif
|
|
|
|
if (wsi->role_ops &&
|
|
lws_rops_fidx(wsi->role_ops, LWS_ROPS_close_kill_connection))
|
|
lws_rops_func_fidx(wsi->role_ops, LWS_ROPS_close_kill_connection).
|
|
close_kill_connection(wsi, 1);
|
|
|
|
if (wsi->a.context->event_loop_ops->close_handle_manually)
|
|
wsi->a.context->event_loop_ops->close_handle_manually(wsi);
|
|
else
|
|
if (wsi->desc.sockfd != LWS_SOCK_INVALID)
|
|
compatible_close(wsi->desc.sockfd);
|
|
|
|
#if defined(LWS_WITH_TLS)
|
|
if (!ssl)
|
|
wsi->tls.use_ssl &= (unsigned int)~LCCSCF_USE_SSL;
|
|
else
|
|
wsi->tls.use_ssl |= LCCSCF_USE_SSL;
|
|
#else
|
|
if (ssl) {
|
|
lwsl_err("%s: not configured for ssl\n", __func__);
|
|
goto bail;
|
|
}
|
|
#endif
|
|
|
|
if (wsi->a.protocol && wsi->role_ops && wsi->protocol_bind_balance) {
|
|
wsi->a.protocol->callback(wsi,
|
|
wsi->role_ops->protocol_unbind_cb[
|
|
!!lwsi_role_server(wsi)],
|
|
wsi->user_space, (void *)__func__, 0);
|
|
|
|
wsi->protocol_bind_balance = 0;
|
|
}
|
|
|
|
wsi->desc.sockfd = LWS_SOCK_INVALID;
|
|
lws_role_transition(wsi, LWSIFR_CLIENT, LRS_UNCONNECTED, &role_ops_h1);
|
|
// wsi->a.protocol = NULL;
|
|
if (wsi->a.protocol)
|
|
lws_bind_protocol(wsi, wsi->a.protocol, "client_reset");
|
|
wsi->pending_timeout = NO_PENDING_TIMEOUT;
|
|
wsi->c_port = (uint16_t)port;
|
|
wsi->hdr_parsing_completed = 0;
|
|
|
|
if (lws_header_table_attach(wsi, 0)) {
|
|
lwsl_err("%s: failed to get ah\n", __func__);
|
|
goto bail;
|
|
}
|
|
//_lws_header_table_reset(wsi->http.ah);
|
|
|
|
if (lws_hdr_simple_create(wsi, _WSI_TOKEN_CLIENT_PEER_ADDRESS, address))
|
|
goto bail;
|
|
|
|
if (lws_hdr_simple_create(wsi, _WSI_TOKEN_CLIENT_HOST, host))
|
|
goto bail;
|
|
|
|
/*
|
|
* _WSI_TOKEN_CLIENT_ORIGIN,
|
|
* _WSI_TOKEN_CLIENT_SENT_PROTOCOLS,
|
|
* _WSI_TOKEN_CLIENT_METHOD,
|
|
* _WSI_TOKEN_CLIENT_IFACE,
|
|
* _WSI_TOKEN_CLIENT_ALPN
|
|
* address
|
|
* host
|
|
* path
|
|
*/
|
|
|
|
p = stash;
|
|
for (n = 0; n < (int)LWS_ARRAY_SIZE(hnames2); n++) {
|
|
if (lws_hdr_simple_create(wsi, hnames2[n], p))
|
|
goto bail;
|
|
p += lws_hdr_total_length(wsi, hnames2[n]) + 1;
|
|
}
|
|
|
|
stash[0] = '/';
|
|
memmove(&stash[1], path, size - 1 < strlen(path) + 1 ?
|
|
size - 1 : strlen(path) + (size_t)1);
|
|
if (lws_hdr_simple_create(wsi, _WSI_TOKEN_CLIENT_URI, stash))
|
|
goto bail;
|
|
|
|
lws_free_set_NULL(stash);
|
|
|
|
#if defined(LWS_WITH_HTTP2)
|
|
if (wsi->client_mux_substream)
|
|
wsi->h2.END_STREAM = wsi->h2.END_HEADERS = 0;
|
|
#endif
|
|
|
|
*pwsi = lws_client_connect_2_dnsreq(wsi);
|
|
|
|
return *pwsi;
|
|
|
|
bail:
|
|
lws_free_set_NULL(stash);
|
|
|
|
return NULL;
|
|
}
|