123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839 |
- #include "private-libwebsockets.h"
- struct lws *
- lws_client_connect_2(struct lws *wsi)
- {
- #ifdef LWS_USE_IPV6
- struct sockaddr_in6 server_addr6;
- struct addrinfo hints, *result;
- #endif
- struct lws_context *context = wsi->context;
- struct lws_context_per_thread *pt = &context->pt[(int)wsi->tsi];
- struct sockaddr_in server_addr4;
- struct lws_pollfd pfd;
- struct sockaddr *v;
- const char *cce = "";
- int n, plen = 0;
- const char *ads;
- lwsl_client("%s\n", __func__);
- if (!wsi->u.hdr.ah) {
- cce = "ah was NULL at cc2";
- lwsl_err("%s\n", cce);
- goto oom4;
- }
- /* proxy? */
- if (wsi->vhost->http_proxy_port) {
- plen = sprintf((char *)pt->serv_buf,
- "CONNECT %s:%u HTTP/1.0\x0d\x0a"
- "User-agent: libwebsockets\x0d\x0a",
- lws_hdr_simple_ptr(wsi, _WSI_TOKEN_CLIENT_PEER_ADDRESS),
- wsi->c_port);
- if (wsi->vhost->proxy_basic_auth_token[0])
- plen += sprintf((char *)pt->serv_buf + plen,
- "Proxy-authorization: basic %s\x0d\x0a",
- wsi->vhost->proxy_basic_auth_token);
- plen += sprintf((char *)pt->serv_buf + plen, "\x0d\x0a");
- ads = wsi->vhost->http_proxy_address;
- #ifdef LWS_USE_IPV6
- if (LWS_IPV6_ENABLED(wsi->vhost)) {
- memset(&server_addr6, 0, sizeof(struct sockaddr_in6));
- server_addr6.sin6_port = htons(wsi->vhost->http_proxy_port);
- } else
- #endif
- server_addr4.sin_port = htons(wsi->vhost->http_proxy_port);
- } else {
- ads = lws_hdr_simple_ptr(wsi, _WSI_TOKEN_CLIENT_PEER_ADDRESS);
- #ifdef LWS_USE_IPV6
- if (LWS_IPV6_ENABLED(wsi->vhost)) {
- memset(&server_addr6, 0, sizeof(struct sockaddr_in6));
- server_addr6.sin6_port = htons(wsi->c_port);
- } else
- #endif
- server_addr4.sin_port = htons(wsi->c_port);
- }
- /*
- * prepare the actual connection (to the proxy, if any)
- */
- lwsl_notice("%s: address %s\n", __func__, ads);
- #ifdef LWS_USE_IPV6
- if (LWS_IPV6_ENABLED(wsi->vhost)) {
- memset(&hints, 0, sizeof(struct addrinfo));
- #if !defined(__ANDROID__)
- hints.ai_family = AF_INET6;
- hints.ai_flags = AI_V4MAPPED;
- #endif
- n = getaddrinfo(ads, NULL, &hints, &result);
- if (n) {
- #ifdef _WIN32
- lwsl_err("getaddrinfo: %ls\n", gai_strerrorW(n));
- #else
- lwsl_err("getaddrinfo: %s\n", gai_strerror(n));
- #endif
- cce = "getaddrinfo (ipv6) failed";
- goto oom4;
- }
- server_addr6.sin6_family = AF_INET6;
- switch (result->ai_family) {
- #if defined(__ANDROID__)
- case AF_INET:
- /* map IPv4 to IPv6 */
- bzero((char *)&server_addr6.sin6_addr,
- sizeof(struct in6_addr));
- server_addr6.sin6_addr.s6_addr[10] = 0xff;
- server_addr6.sin6_addr.s6_addr[11] = 0xff;
- memcpy(&server_addr6.sin6_addr.s6_addr[12],
- &((struct sockaddr_in *)result->ai_addr)->sin_addr,
- sizeof(struct in_addr));
- break;
- #endif
- case AF_INET6:
- memcpy(&server_addr6.sin6_addr,
- &((struct sockaddr_in6 *)result->ai_addr)->sin6_addr,
- sizeof(struct in6_addr));
- server_addr6.sin6_scope_id = ((struct sockaddr_in6 *)result->ai_addr)->sin6_scope_id;
- server_addr6.sin6_flowinfo = ((struct sockaddr_in6 *)result->ai_addr)->sin6_flowinfo;
- break;
- default:
- lwsl_err("Unknown address family\n");
- freeaddrinfo(result);
- cce = "unknown address family";
- goto oom4;
- }
- freeaddrinfo(result);
- } else
- #endif
- {
- struct addrinfo ai, *res, *result = NULL;
- void *p = NULL;
- int addr_rv;
- memset (&ai, 0, sizeof ai);
- ai.ai_family = PF_UNSPEC;
- ai.ai_socktype = SOCK_STREAM;
- ai.ai_flags = AI_CANONNAME;
- addr_rv = getaddrinfo(ads, NULL, &ai, &result);
- if (!addr_rv) {
- res = result;
- while (!p && res) {
- switch (res->ai_family) {
- case AF_INET:
- p = &((struct sockaddr_in *)res->ai_addr)->sin_addr;
- break;
- }
- res = res->ai_next;
- }
- #if defined(LWS_FALLBACK_GETHOSTBYNAME)
- } else if (addr_rv == EAI_SYSTEM) {
- struct hostent *host;
- lwsl_info("getaddrinfo (ipv4) failed, trying gethostbyname\n");
- host = gethostbyname(ads);
- if (host) {
- p = host->h_addr;
- } else {
- lwsl_err("gethostbyname failed\n");
- cce = "gethostbyname (ipv4) failed";
- goto oom4;
- }
- #endif
- } else {
- lwsl_err("getaddrinfo failed\n");
- cce = "getaddrinfo (ipv4) failed";
- goto oom4;
- }
- if (!p) {
- if (result)
- freeaddrinfo(result);
- lwsl_err("Couldn't identify address\n");
- cce = "unable to lookup address";
- goto oom4;
- }
- server_addr4.sin_family = AF_INET;
- server_addr4.sin_addr = *((struct in_addr *)p);
- bzero(&server_addr4.sin_zero, 8);
- if (result)
- freeaddrinfo(result);
- }
- if (!lws_socket_is_valid(wsi->desc.sockfd)) {
- #ifdef LWS_USE_IPV6
- if (LWS_IPV6_ENABLED(wsi->vhost))
- wsi->desc.sockfd = socket(AF_INET6, SOCK_STREAM, 0);
- else
- #endif
- wsi->desc.sockfd = socket(AF_INET, SOCK_STREAM, 0);
- if (!lws_socket_is_valid(wsi->desc.sockfd)) {
- lwsl_warn("Unable to open socket\n");
- cce = "unable to open socket";
- goto oom4;
- }
- if (lws_plat_set_socket_options(wsi->vhost, wsi->desc.sockfd)) {
- lwsl_err("Failed to set wsi socket options\n");
- compatible_close(wsi->desc.sockfd);
- cce = "set socket opts failed";
- goto oom4;
- }
- wsi->mode = LWSCM_WSCL_WAITING_CONNECT;
- lws_libev_accept(wsi, wsi->desc);
- lws_libuv_accept(wsi, wsi->desc);
- if (insert_wsi_socket_into_fds(context, wsi)) {
- compatible_close(wsi->desc.sockfd);
- cce = "insert wsi failed";
- goto oom4;
- }
- lws_change_pollfd(wsi, 0, LWS_POLLIN);
- /*
- * past here, we can't simply free the structs as error
- * handling as oom4 does. We have to run the whole close flow.
- */
- if (!wsi->protocol)
- wsi->protocol = &wsi->vhost->protocols[0];
- wsi->protocol->callback(wsi, LWS_CALLBACK_WSI_CREATE,
- wsi->user_space, NULL, 0);
- lws_set_timeout(wsi, PENDING_TIMEOUT_AWAITING_CONNECT_RESPONSE,
- AWAITING_TIMEOUT);
- n = lws_socket_bind(wsi->vhost, wsi->desc.sockfd, 0, wsi->vhost->iface);
- if (n < 0) {
- cce = "unable to bind socket";
- goto failed;
- }
- }
- #ifdef LWS_USE_IPV6
- if (LWS_IPV6_ENABLED(wsi->vhost)) {
- v = (struct sockaddr *)&server_addr6;
- n = sizeof(struct sockaddr_in6);
- } else
- #endif
- {
- v = (struct sockaddr *)&server_addr4;
- n = sizeof(struct sockaddr);
- }
- if (connect(wsi->desc.sockfd, v, n) == -1 || LWS_ERRNO == LWS_EISCONN) {
- if (LWS_ERRNO == LWS_EALREADY ||
- LWS_ERRNO == LWS_EINPROGRESS ||
- LWS_ERRNO == LWS_EWOULDBLOCK
- #ifdef _WIN32
- || LWS_ERRNO == WSAEINVAL
- #endif
- ) {
- lwsl_client("nonblocking connect retry (errno = %d)\n",
- LWS_ERRNO);
- if (lws_plat_check_connection_error(wsi)) {
- cce = "socket connect failed";
- goto failed;
- }
- /*
- * must do specifically a POLLOUT poll to hear
- * about the connect completion
- */
- if (lws_change_pollfd(wsi, 0, LWS_POLLOUT)) {
- cce = "POLLOUT set failed";
- goto failed;
- }
- return wsi;
- }
- if (LWS_ERRNO != LWS_EISCONN) {
- lwsl_notice("Connect failed errno=%d\n", LWS_ERRNO);
- cce = "connect failed";
- goto failed;
- }
- }
- lwsl_client("connected\n");
- /* we are connected to server, or proxy */
- if (wsi->vhost->http_proxy_port) {
- /*
- * OK from now on we talk via the proxy, so connect to that
- *
- * (will overwrite existing pointer,
- * leaving old string/frag there but unreferenced)
- */
- if (lws_hdr_simple_create(wsi, _WSI_TOKEN_CLIENT_PEER_ADDRESS,
- wsi->vhost->http_proxy_address))
- goto failed;
- wsi->c_port = wsi->vhost->http_proxy_port;
- n = send(wsi->desc.sockfd, (char *)pt->serv_buf, plen,
- MSG_NOSIGNAL);
- if (n < 0) {
- lwsl_debug("ERROR writing to proxy socket\n");
- cce = "proxy write failed";
- goto failed;
- }
- lws_set_timeout(wsi, PENDING_TIMEOUT_AWAITING_PROXY_RESPONSE,
- AWAITING_TIMEOUT);
- wsi->mode = LWSCM_WSCL_WAITING_PROXY_REPLY;
- return wsi;
- }
- /*
- * provoke service to issue the handshake directly
- * we need to do it this way because in the proxy case, this is the
- * next state and executed only if and when we get a good proxy
- * response inside the state machine... but notice in SSL case this
- * may not have sent anything yet with 0 return, and won't until some
- * many retries from main loop. To stop that becoming endless,
- * cover with a timeout.
- */
- lws_set_timeout(wsi, PENDING_TIMEOUT_SENT_CLIENT_HANDSHAKE,
- AWAITING_TIMEOUT);
- wsi->mode = LWSCM_WSCL_ISSUE_HANDSHAKE;
- pfd.fd = wsi->desc.sockfd;
- pfd.events = LWS_POLLIN;
- pfd.revents = LWS_POLLIN;
- n = lws_service_fd(context, &pfd);
- if (n < 0) {
- cce = "first service failed";
- goto failed;
- }
- if (n) /* returns 1 on failure after closing wsi */
- return NULL;
- return wsi;
- oom4:
- /* we're closing, losing some rx is OK */
- if (wsi->u.hdr.ah)
- wsi->u.hdr.ah->rxpos = wsi->u.hdr.ah->rxlen;
- if (wsi->mode == LWSCM_HTTP_CLIENT ||
- wsi->mode == LWSCM_HTTP_CLIENT_ACCEPTED ||
- wsi->mode == LWSCM_WSCL_WAITING_CONNECT) {
- wsi->vhost->protocols[0].callback(wsi,
- LWS_CALLBACK_CLIENT_CONNECTION_ERROR,
- wsi->user_space, (void *)cce, strlen(cce));
- wsi->already_did_cce = 1;
- }
- /* take care that we might be inserted in fds already */
- if (wsi->position_in_fds_table != -1)
- goto failed1;
- lws_remove_from_timeout_list(wsi);
- lws_header_table_detach(wsi, 0);
- lws_free(wsi);
- return NULL;
- failed:
- wsi->vhost->protocols[0].callback(wsi,
- LWS_CALLBACK_CLIENT_CONNECTION_ERROR,
- wsi->user_space, (void *)cce, strlen(cce));
- wsi->already_did_cce = 1;
- failed1:
- lws_close_free_wsi(wsi, LWS_CLOSE_STATUS_NOSTATUS);
- return NULL;
- }
- /**
- * 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
- */
- LWS_VISIBLE struct lws *
- lws_client_reset(struct lws **pwsi, int ssl, const char *address, int port,
- const char *path, const char *host)
- {
- char origin[300] = "", protocol[300] = "", method[32] = "", *p;
- struct lws *wsi = *pwsi;
- if (wsi->redirects == 3) {
- lwsl_err("%s: Too many redirects\n", __func__);
- return NULL;
- }
- wsi->redirects++;
- #ifdef LWS_OPENSSL_SUPPORT
- wsi->use_ssl = ssl;
- #else
- if (ssl) {
- lwsl_err("%s: not configured for ssl\n", __func__);
- return NULL;
- }
- #endif
- p = lws_hdr_simple_ptr(wsi, _WSI_TOKEN_CLIENT_ORIGIN);
- if (p)
- strncpy(origin, p, sizeof(origin) - 1);
- p = lws_hdr_simple_ptr(wsi, _WSI_TOKEN_CLIENT_SENT_PROTOCOLS);
- if (p)
- strncpy(protocol, p, sizeof(protocol) - 1);
- p = lws_hdr_simple_ptr(wsi, _WSI_TOKEN_CLIENT_METHOD);
- if (p)
- strncpy(method, p, sizeof(method) - 1);
- lwsl_debug("redirect ads='%s', port=%d, path='%s', ssl = %d\n",
- address, port, path, ssl);
- /* close the connection by hand */
- compatible_close(wsi->desc.sockfd);
- remove_wsi_socket_from_fds(wsi);
- wsi->desc.sockfd = LWS_SOCK_INVALID;
- wsi->state = LWSS_CLIENT_UNCONNECTED;
- wsi->protocol = NULL;
- wsi->pending_timeout = NO_PENDING_TIMEOUT;
- wsi->c_port = port;
- wsi->hdr_parsing_completed = 0;
- _lws_header_table_reset(wsi->u.hdr.ah);
- if (lws_hdr_simple_create(wsi, _WSI_TOKEN_CLIENT_PEER_ADDRESS, address))
- return NULL;
- if (lws_hdr_simple_create(wsi, _WSI_TOKEN_CLIENT_HOST, host))
- return NULL;
- if (origin[0])
- if (lws_hdr_simple_create(wsi, _WSI_TOKEN_CLIENT_ORIGIN,
- origin))
- return NULL;
- if (protocol[0])
- if (lws_hdr_simple_create(wsi, _WSI_TOKEN_CLIENT_SENT_PROTOCOLS,
- protocol))
- return NULL;
- if (method[0])
- if (lws_hdr_simple_create(wsi, _WSI_TOKEN_CLIENT_METHOD,
- method))
- return NULL;
- origin[0] = '/';
- strncpy(&origin[1], path, sizeof(origin) - 2);
- if (lws_hdr_simple_create(wsi, _WSI_TOKEN_CLIENT_URI, origin))
- return NULL;
- *pwsi = lws_client_connect_2(wsi);
- return *pwsi;
- }
- #ifdef LWS_WITH_HTTP_PROXY
- static hubbub_error
- html_parser_cb(const hubbub_token *token, void *pw)
- {
- struct lws_rewrite *r = (struct lws_rewrite *)pw;
- char buf[1024], *start = buf + LWS_PRE, *p = start,
- *end = &buf[sizeof(buf) - 1];
- size_t i;
- switch (token->type) {
- case HUBBUB_TOKEN_DOCTYPE:
- p += lws_snprintf(p, end - p, "<!DOCTYPE %.*s %s ",
- (int) token->data.doctype.name.len,
- token->data.doctype.name.ptr,
- token->data.doctype.force_quirks ?
- "(force-quirks) " : "");
- if (token->data.doctype.public_missing)
- printf("\tpublic: missing\n");
- else
- p += lws_snprintf(p, end - p, "PUBLIC \"%.*s\"\n",
- (int) token->data.doctype.public_id.len,
- token->data.doctype.public_id.ptr);
- if (token->data.doctype.system_missing)
- printf("\tsystem: missing\n");
- else
- p += lws_snprintf(p, end - p, " \"%.*s\">\n",
- (int) token->data.doctype.system_id.len,
- token->data.doctype.system_id.ptr);
- break;
- case HUBBUB_TOKEN_START_TAG:
- p += lws_snprintf(p, end - p, "<%.*s", (int)token->data.tag.name.len,
- token->data.tag.name.ptr);
- /* (token->data.tag.self_closing) ?
- "(self-closing) " : "",
- (token->data.tag.n_attributes > 0) ?
- "attributes:" : "");
- */
- for (i = 0; i < token->data.tag.n_attributes; i++) {
- if (!hstrcmp(&token->data.tag.attributes[i].name, "href", 4) ||
- !hstrcmp(&token->data.tag.attributes[i].name, "action", 6) ||
- !hstrcmp(&token->data.tag.attributes[i].name, "src", 3)) {
- const char *pp = (const char *)token->data.tag.attributes[i].value.ptr;
- int plen = (int) token->data.tag.attributes[i].value.len;
- if (!hstrcmp(&token->data.tag.attributes[i].value,
- r->from, r->from_len)) {
- pp += r->from_len;
- plen -= r->from_len;
- }
- p += lws_snprintf(p, end - p, " %.*s=\"%s/%.*s\"",
- (int) token->data.tag.attributes[i].name.len,
- token->data.tag.attributes[i].name.ptr,
- r->to, plen, pp);
- } else
- p += lws_snprintf(p, end - p, " %.*s=\"%.*s\"",
- (int) token->data.tag.attributes[i].name.len,
- token->data.tag.attributes[i].name.ptr,
- (int) token->data.tag.attributes[i].value.len,
- token->data.tag.attributes[i].value.ptr);
- }
- p += lws_snprintf(p, end - p, ">\n");
- break;
- case HUBBUB_TOKEN_END_TAG:
- p += lws_snprintf(p, end - p, "</%.*s", (int) token->data.tag.name.len,
- token->data.tag.name.ptr);
- /*
- (token->data.tag.self_closing) ?
- "(self-closing) " : "",
- (token->data.tag.n_attributes > 0) ?
- "attributes:" : "");
- */
- for (i = 0; i < token->data.tag.n_attributes; i++) {
- p += lws_snprintf(p, end - p, " %.*s='%.*s'\n",
- (int) token->data.tag.attributes[i].name.len,
- token->data.tag.attributes[i].name.ptr,
- (int) token->data.tag.attributes[i].value.len,
- token->data.tag.attributes[i].value.ptr);
- }
- p += lws_snprintf(p, end - p, ">\n");
- break;
- case HUBBUB_TOKEN_COMMENT:
- p += lws_snprintf(p, end - p, "<!-- %.*s -->\n",
- (int) token->data.comment.len,
- token->data.comment.ptr);
- break;
- case HUBBUB_TOKEN_CHARACTER:
- p += lws_snprintf(p, end - p, "%.*s", (int) token->data.character.len,
- token->data.character.ptr);
- break;
- case HUBBUB_TOKEN_EOF:
- p += lws_snprintf(p, end - p, "\n");
- break;
- }
- if (user_callback_handle_rxflow(r->wsi->protocol->callback,
- r->wsi, LWS_CALLBACK_RECEIVE_CLIENT_HTTP_READ,
- r->wsi->user_space, start, p - start))
- return -1;
- return HUBBUB_OK;
- }
- #endif
- LWS_VISIBLE struct lws *
- lws_client_connect_via_info(struct lws_client_connect_info *i)
- {
- struct lws *wsi;
- int v = SPEC_LATEST_SUPPORTED;
- const struct lws_protocols *p;
- if (i->context->requested_kill)
- return NULL;
- if (!i->context->protocol_init_done)
- lws_protocol_init(i->context);
- wsi = lws_zalloc(sizeof(struct lws));
- if (wsi == NULL)
- goto bail;
- wsi->context = i->context;
- /* assert the mode and union status (hdr) clearly */
- lws_union_transition(wsi, LWSCM_HTTP_CLIENT);
- wsi->desc.sockfd = LWS_SOCK_INVALID;
- /* 1) fill up the wsi with stuff from the connect_info as far as it
- * can go. It's because not only is our connection async, we might
- * not even be able to get ahold of an ah at this point.
- */
- /* -1 means just use latest supported */
- if (i->ietf_version_or_minus_one != -1 && i->ietf_version_or_minus_one)
- v = i->ietf_version_or_minus_one;
- wsi->ietf_spec_revision = v;
- wsi->user_space = NULL;
- wsi->state = LWSS_CLIENT_UNCONNECTED;
- wsi->pending_timeout = NO_PENDING_TIMEOUT;
- wsi->position_in_fds_table = -1;
- wsi->c_port = i->port;
- wsi->vhost = i->vhost;
- if (!wsi->vhost)
- wsi->vhost = i->context->vhost_list;
- wsi->protocol = &wsi->vhost->protocols[0];
- /* for http[s] connection, allow protocol selection by name */
- if (i->method && i->vhost && i->protocol) {
- p = lws_vhost_name_to_protocol(i->vhost, i->protocol);
- if (p)
- wsi->protocol = p;
- }
- if (wsi && !wsi->user_space && i->userdata) {
- wsi->user_space_externally_allocated = 1;
- wsi->user_space = i->userdata;
- } else
- /* if we stay in http, we can assign the user space now,
- * otherwise do it after the protocol negotiated
- */
- if (i->method)
- if (lws_ensure_user_space(wsi))
- goto bail;
- #ifdef LWS_OPENSSL_SUPPORT
- wsi->use_ssl = i->ssl_connection;
- #else
- if (i->ssl_connection) {
- lwsl_err("libwebsockets not configured for ssl\n");
- goto bail;
- }
- #endif
- /* 2) stash the things from connect_info that we can't process without
- * an ah. Because if no ah, we will go on the ah waiting list and
- * process those things later (after the connect_info and maybe the
- * things pointed to have gone out of scope.
- */
- wsi->u.hdr.stash = lws_malloc(sizeof(*wsi->u.hdr.stash));
- if (!wsi->u.hdr.stash) {
- lwsl_err("%s: OOM\n", __func__);
- goto bail;
- }
- wsi->u.hdr.stash->origin[0] = '\0';
- wsi->u.hdr.stash->protocol[0] = '\0';
- wsi->u.hdr.stash->method[0] = '\0';
- strncpy(wsi->u.hdr.stash->address, i->address,
- sizeof(wsi->u.hdr.stash->address) - 1);
- strncpy(wsi->u.hdr.stash->path, i->path,
- sizeof(wsi->u.hdr.stash->path) - 1);
- strncpy(wsi->u.hdr.stash->host, i->host,
- sizeof(wsi->u.hdr.stash->host) - 1);
- if (i->origin)
- strncpy(wsi->u.hdr.stash->origin, i->origin,
- sizeof(wsi->u.hdr.stash->origin) - 1);
- if (i->protocol)
- strncpy(wsi->u.hdr.stash->protocol, i->protocol,
- sizeof(wsi->u.hdr.stash->protocol) - 1);
- if (i->method)
- strncpy(wsi->u.hdr.stash->method, i->method,
- sizeof(wsi->u.hdr.stash->method) - 1);
- wsi->u.hdr.stash->address[sizeof(wsi->u.hdr.stash->address) - 1] = '\0';
- wsi->u.hdr.stash->path[sizeof(wsi->u.hdr.stash->path) - 1] = '\0';
- wsi->u.hdr.stash->host[sizeof(wsi->u.hdr.stash->host) - 1] = '\0';
- wsi->u.hdr.stash->origin[sizeof(wsi->u.hdr.stash->origin) - 1] = '\0';
- wsi->u.hdr.stash->protocol[sizeof(wsi->u.hdr.stash->protocol) - 1] = '\0';
- wsi->u.hdr.stash->method[sizeof(wsi->u.hdr.stash->method) - 1] = '\0';
- if (i->pwsi)
- *i->pwsi = wsi;
- /* if we went on the waiting list, no probs just return the wsi
- * when we get the ah, now or later, he will call
- * lws_client_connect_via_info2() below.
- */
- if (lws_header_table_attach(wsi, 0) < 0) {
- /*
- * if we failed here, the connection is already closed
- * and freed.
- */
- goto bail1;
- }
- if (i->parent_wsi) {
- lwsl_info("%s: created child %p of parent %p\n", __func__,
- wsi, i->parent_wsi);
- wsi->parent = i->parent_wsi;
- wsi->sibling_list = i->parent_wsi->child_list;
- i->parent_wsi->child_list = wsi;
- }
- #ifdef LWS_WITH_HTTP_PROXY
- if (i->uri_replace_to)
- wsi->rw = lws_rewrite_create(wsi, html_parser_cb,
- i->uri_replace_from,
- i->uri_replace_to);
- #endif
- return wsi;
- bail:
- lws_free(wsi);
- bail1:
- if (i->pwsi)
- *i->pwsi = NULL;
- return NULL;
- }
- struct lws *
- lws_client_connect_via_info2(struct lws *wsi)
- {
- struct client_info_stash *stash = wsi->u.hdr.stash;
- if (!stash)
- return wsi;
- /*
- * we're not necessarily in a position to action these right away,
- * stash them... we only need during connect phase so u.hdr is fine
- */
- if (lws_hdr_simple_create(wsi, _WSI_TOKEN_CLIENT_PEER_ADDRESS,
- stash->address))
- goto bail1;
- /* these only need u.hdr lifetime as well */
- if (lws_hdr_simple_create(wsi, _WSI_TOKEN_CLIENT_URI, stash->path))
- goto bail1;
- if (lws_hdr_simple_create(wsi, _WSI_TOKEN_CLIENT_HOST, stash->host))
- goto bail1;
- if (stash->origin[0])
- if (lws_hdr_simple_create(wsi, _WSI_TOKEN_CLIENT_ORIGIN,
- stash->origin))
- goto bail1;
- /*
- * this is a list of protocols we tell the server we're okay with
- * stash it for later when we compare server response with it
- */
- if (stash->protocol[0])
- if (lws_hdr_simple_create(wsi, _WSI_TOKEN_CLIENT_SENT_PROTOCOLS,
- stash->protocol))
- goto bail1;
- if (stash->method[0])
- if (lws_hdr_simple_create(wsi, _WSI_TOKEN_CLIENT_METHOD,
- stash->method))
- goto bail1;
- lws_free_set_NULL(wsi->u.hdr.stash);
- /*
- * Check with each extension if it is able to route and proxy this
- * connection for us. For example, an extension like x-google-mux
- * can handle this and then we don't need an actual socket for this
- * connection.
- */
- if (lws_ext_cb_all_exts(wsi->context, wsi,
- LWS_EXT_CB_CAN_PROXY_CLIENT_CONNECTION,
- (void *)stash->address,
- wsi->c_port) > 0) {
- lwsl_client("lws_client_connect: ext handling conn\n");
- lws_set_timeout(wsi,
- PENDING_TIMEOUT_AWAITING_EXTENSION_CONNECT_RESPONSE,
- AWAITING_TIMEOUT);
- wsi->mode = LWSCM_WSCL_WAITING_EXTENSION_CONNECT;
- return wsi;
- }
- lwsl_client("lws_client_connect: direct conn\n");
- wsi->context->count_wsi_allocated++;
- return lws_client_connect_2(wsi);
- bail1:
- lws_free_set_NULL(wsi->u.hdr.stash);
- return NULL;
- }
- LWS_VISIBLE struct lws *
- lws_client_connect_extended(struct lws_context *context, const char *address,
- int port, int ssl_connection, const char *path,
- const char *host, const char *origin,
- const char *protocol, int ietf_version_or_minus_one,
- void *userdata)
- {
- struct lws_client_connect_info i;
- memset(&i, 0, sizeof(i));
- i.context = context;
- i.address = address;
- i.port = port;
- i.ssl_connection = ssl_connection;
- i.path = path;
- i.host = host;
- i.origin = origin;
- i.protocol = protocol;
- i.ietf_version_or_minus_one = ietf_version_or_minus_one;
- i.userdata = userdata;
- return lws_client_connect_via_info(&i);
- }
- LWS_VISIBLE struct lws *
- lws_client_connect(struct lws_context *context, const char *address,
- int port, int ssl_connection, const char *path,
- const char *host, const char *origin,
- const char *protocol, int ietf_version_or_minus_one)
- {
- struct lws_client_connect_info i;
- memset(&i, 0, sizeof(i));
- i.context = context;
- i.address = address;
- i.port = port;
- i.ssl_connection = ssl_connection;
- i.path = path;
- i.host = host;
- i.origin = origin;
- i.protocol = protocol;
- i.ietf_version_or_minus_one = ietf_version_or_minus_one;
- i.userdata = NULL;
- return lws_client_connect_via_info(&i);
- }
|