You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
1425 lines
35 KiB
1425 lines
35 KiB
/*
|
|
* 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"
|
|
|
|
#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
|