Skip to content
Snippets Groups Projects
client-handshake.c 27 KiB
Newer Older
  • Learn to ignore specific revisions
  • static int
    lws_getaddrinfo46(struct lws *wsi, const char *ads, struct addrinfo **result)
    
    Andy Green's avatar
    Andy Green committed
    {
    
    	struct addrinfo hints;
    
    	memset(&hints, 0, sizeof(hints));
    	*result = NULL;
    
    
    #ifdef LWS_WITH_IPV6
    
    	if (wsi->ipv6) {
    
    #if !defined(__ANDROID__)
    		hints.ai_family = AF_INET6;
    		hints.ai_flags = AI_V4MAPPED;
    #endif
    	} else
    
    James Devine's avatar
    James Devine committed
    #endif
    
    	{
    		hints.ai_family = PF_UNSPEC;
    		hints.ai_socktype = SOCK_STREAM;
    	}
    
    	return getaddrinfo(ads, NULL, &hints, result);
    }
    
    struct lws *
    lws_client_connect_2(struct lws *wsi)
    {
    
    #if defined(LWS_ROLE_H1) || defined(LWS_ROLE_H2)
    
    	struct lws_context *context = wsi->context;
    
    Andy Green's avatar
    Andy Green committed
    	struct lws_context_per_thread *pt = &context->pt[(int)wsi->tsi];
    
    	const char *adsin;
    
    Andy Green's avatar
    Andy Green committed
    	struct lws *wsi_piggyback = NULL;
    
    Andy Green's avatar
    Andy Green committed
    	struct lws_pollfd pfd;
    
    Andy Green's avatar
    Andy Green committed
    	ssize_t plen = 0;
    
    #endif
    	struct addrinfo *result;
    
    #if defined(LWS_WITH_UNIX_SOCK)
    	struct sockaddr_un sau;
    	char unix_skt = 0;
    #endif
    
    Andy Green's avatar
    Andy Green committed
    	sockaddr46 sa46;
    
    	const struct sockaddr *psa;
    
    	const char *cce = "", *iface;
    	const char *meth = NULL;
    
    #ifdef LWS_WITH_IPV6
    
    	char ipv6only = lws_check_opt(wsi->vhost->options,
    			LWS_SERVER_OPTION_IPV6_V6ONLY_MODIFY |
    			LWS_SERVER_OPTION_IPV6_V6ONLY_VALUE);
    
    #if defined(__ANDROID__)
    	ipv6only = 0;
    #endif
    #endif
    
    Andy Green's avatar
    Andy Green committed
    	lwsl_client("%s: %p\n", __func__, wsi);
    
    #if defined(LWS_ROLE_H1) || defined(LWS_ROLE_H2)
    
    		cce = "ah was NULL at cc2";
    		lwsl_err("%s\n", cce);
    		goto oom4;
    	}
    
    
    	/* we can only piggyback GET or POST */
    
    Andy Green's avatar
    Andy Green committed
    
    	meth = lws_hdr_simple_ptr(wsi, _WSI_TOKEN_CLIENT_METHOD);
    
    	if (meth && strcmp(meth, "GET") && strcmp(meth, "POST"))
    
    Andy Green's avatar
    Andy Green committed
    		goto create_new_conn;
    
    	/* we only pipeline connections that said it was okay */
    
    	if (!wsi->client_pipeline)
    		goto create_new_conn;
    
    	/*
    	 * let's take a look first and see if there are any already-active
    	 * client connections we can piggy-back on.
    	 */
    
    	adsin = lws_hdr_simple_ptr(wsi, _WSI_TOKEN_CLIENT_PEER_ADDRESS);
    
    
    	lws_vhost_lock(wsi->vhost); /* ----------------------------------- { */
    
    
    Andy Green's avatar
    Andy Green committed
    	lws_start_foreach_dll_safe(struct lws_dll_lws *, d, d1,
    				   wsi->vhost->dll_active_client_conns.next) {
    		struct lws *w = lws_container_of(d, struct lws,
    						 dll_active_client_conns);
    
    
    		lwsl_debug("%s: check %s %s %d %d\n", __func__, adsin,
    			   w->client_hostname_copy, wsi->c_port, w->c_port);
    
    Andy Green's avatar
    Andy Green committed
    
    		if (w != wsi && w->client_hostname_copy &&
    
    Andy Green's avatar
    Andy Green committed
    		    !strcmp(adsin, w->client_hostname_copy) &&
    
    Andy Green's avatar
    Andy Green committed
    #if defined(LWS_WITH_TLS)
    
    		    (wsi->tls.use_ssl & LCCSCF_USE_SSL) ==
    		     (w->tls.use_ssl & LCCSCF_USE_SSL) &&
    
    Andy Green's avatar
    Andy Green committed
    #endif
    
    Andy Green's avatar
    Andy Green committed
    		    wsi->c_port == w->c_port) {
    
    Andy Green's avatar
    Andy Green committed
    
    
    Andy Green's avatar
    Andy Green committed
    			/* someone else is already connected to the right guy */
    
    			/* do we know for a fact pipelining won't fly? */
    			if (w->keepalive_rejected) {
    
    Andy Green's avatar
    Andy Green committed
    				lwsl_info("defeating pipelining due to no "
    					    "keepalive on server\n");
    
    				lws_vhost_unlock(wsi->vhost); /* } ---------- */
    
    Andy Green's avatar
    Andy Green committed
    				goto create_new_conn;
    			}
    
    Andy Green's avatar
    Andy Green committed
    #if defined (LWS_WITH_HTTP2)
    			/*
    			 * h2: in usable state already: just use it without
    			 *     going through the queue
    			 */
    			if (w->client_h2_alpn &&
    
    			    (lwsi_state(w) == LRS_H2_WAITING_TO_SEND_HEADERS ||
    			     lwsi_state(w) == LRS_ESTABLISHED)) {
    
    Andy Green's avatar
    Andy Green committed
    
    				lwsl_info("%s: just join h2 directly\n",
    						__func__);
    
    
    				wsi->client_h2_alpn = 1;
    
    Andy Green's avatar
    Andy Green committed
    				lws_wsi_h2_adopt(w, wsi);
    
    				lws_vhost_unlock(wsi->vhost); /* } ---------- */
    
    Andy Green's avatar
    Andy Green committed
    
    				return wsi;
    			}
    #endif
    
    
    			lwsl_info("applying %p to txn queue on %p state 0x%x\n",
    
    Andy Green's avatar
    Andy Green committed
    			/*
    			 * ...let's add ourselves to his transaction queue...
    
    			 * we are adding ourselves at the HEAD
    
    Andy Green's avatar
    Andy Green committed
    			 */
    			lws_dll_lws_add_front(&wsi->dll_client_transaction_queue,
    				&w->dll_client_transaction_queue_head);
    
    Andy Green's avatar
    Andy Green committed
    
    
    Andy Green's avatar
    Andy Green committed
    			 * h1: pipeline our headers out on him,
    			 * and wait for our turn at client transaction_complete
    			 * to take over parsing the rx.
    
    Andy Green's avatar
    Andy Green committed
    			wsi_piggyback = w;
    
    			lws_vhost_unlock(wsi->vhost); /* } ---------- */
    
    Andy Green's avatar
    Andy Green committed
    			goto send_hs;
    		}
    
    	} lws_end_foreach_dll_safe(d, d1);
    
    
    	lws_vhost_unlock(wsi->vhost); /* } ---------------------------------- */
    
    Andy Green's avatar
    Andy Green committed
    
    create_new_conn:
    
    Andy Green's avatar
    Andy Green committed
    	/*
    	 * clients who will create their own fresh connection keep a copy of
    	 * the hostname they originally connected to, in case other connections
    	 * want to use it too
    	 */
    
    	if (!wsi->client_hostname_copy)
    		wsi->client_hostname_copy =
    
    			lws_strdup(lws_hdr_simple_ptr(wsi,
    
    Andy Green's avatar
    Andy Green committed
    					_WSI_TOKEN_CLIENT_PEER_ADDRESS));
    
    
    Andy Green's avatar
    Andy Green committed
    	/*
    	 * If we made our own connection, and we're doing a method that can take
    	 * a pipeline, we are an "active client connection".
    	 *
    	 * Add ourselves to the vhost list of those so that others can
    	 * piggyback on our transaction queue
    	 */
    
    
    	if (meth && (!strcmp(meth, "GET") || !strcmp(meth, "POST")) &&
    
    Andy Green's avatar
    Andy Green committed
    	    lws_dll_is_null(&wsi->dll_client_transaction_queue) &&
    	    lws_dll_is_null(&wsi->dll_active_client_conns)) {
    		lws_vhost_lock(wsi->vhost);
    
    		/* caution... we will have to unpick this on oom4 path */
    
    Andy Green's avatar
    Andy Green committed
    		lws_dll_lws_add_front(&wsi->dll_active_client_conns,
    				      &wsi->vhost->dll_active_client_conns);
    		lws_vhost_unlock(wsi->vhost);
    	}
    
    
    	/*
    	 * unix socket destination?
    	 */
    
    	ads = lws_hdr_simple_ptr(wsi, _WSI_TOKEN_CLIENT_PEER_ADDRESS);
    #if defined(LWS_WITH_UNIX_SOCK)
    	if (*ads == '+') {
    		ads++;
    		memset(&sau, 0, sizeof(sau));
    		sau.sun_family = AF_UNIX;
    		strncpy(sau.sun_path, ads, sizeof(sau.sun_path));
    		sau.sun_path[sizeof(sau.sun_path) - 1] = '\0';
    
    		lwsl_info("%s: Unix skt: %s\n", __func__, ads);
    
    		if (sau.sun_path[0] == '@')
    			sau.sun_path[0] = '\0';
    
    		unix_skt = 1;
    		goto ads_known;
    	}
    #endif
    
    
    	/*
    	 * start off allowing ipv6 on connection if vhost allows it
    	 */
    	wsi->ipv6 = LWS_IPV6_ENABLED(wsi->vhost);
    
    
    #if defined(LWS_ROLE_H1) || defined(LWS_ROLE_H2)
    
    
    	/* Decide what it is we need to connect to:
    	 *
    	 * Priority 1: connect to http proxy */
    
    Andy Green's avatar
    Andy Green committed
    		plen = sprintf((char *)pt->serv_buf,
    
    			"CONNECT %s:%u HTTP/1.0\x0d\x0a"
    
    Andy Green's avatar
    Andy Green committed
    			"User-agent: libwebsockets\x0d\x0a",
    
    			lws_hdr_simple_ptr(wsi, _WSI_TOKEN_CLIENT_PEER_ADDRESS),
    
    Andy Green's avatar
    Andy Green committed
    
    
    Andy Green's avatar
    Andy Green committed
    		if (wsi->vhost->proxy_basic_auth_token[0])
    
    Andy Green's avatar
    Andy Green committed
    			plen += sprintf((char *)pt->serv_buf + plen,
    
    Andy Green's avatar
    Andy Green committed
    					"Proxy-authorization: basic %s\x0d\x0a",
    
    Andy Green's avatar
    Andy Green committed
    					wsi->vhost->proxy_basic_auth_token);
    
    Andy Green's avatar
    Andy Green committed
    
    
    Andy Green's avatar
    Andy Green committed
    		plen += sprintf((char *)pt->serv_buf + plen, "\x0d\x0a");
    
    		ads = wsi->vhost->http.http_proxy_address;
    		port = wsi->vhost->http.http_proxy_port;
    
    James Devine's avatar
    James Devine committed
    
    
    #if defined(LWS_WITH_SOCKS5)
    
    
    	/* Priority 2: Connect to SOCK5 Proxy */
    
    	} else if (wsi->vhost->socks_proxy_port) {
    
    Andy Green's avatar
    Andy Green committed
    		socks_generate_msg(wsi, SOCKS_MSG_GREETING, &plen);
    
    		lwsl_client("Sending SOCKS Greeting\n");
    
    		ads = wsi->vhost->socks_proxy_address;
    
    		port = wsi->vhost->socks_proxy_port;
    
    	} else {
    
    		/* Priority 3: Connect directly */
    
    		ads = lws_hdr_simple_ptr(wsi, _WSI_TOKEN_CLIENT_PEER_ADDRESS);
    
    		port = wsi->c_port;
    
    	 * prepare the actual connection
    	 * to whatever we decided to connect to
    
    	lwsl_info("%s: %p: address %s\n", __func__, wsi, ads);
    
    	n = lws_getaddrinfo46(wsi, ads, &result);
    	memset(&sa46, 0, sizeof(sa46));
    
    #ifdef LWS_WITH_IPV6
    
    	if (wsi->ipv6) {
    
    		struct sockaddr_in6 *sa6;
    
    James Devine's avatar
    James Devine committed
    
    
    		if (n || !result) {
    
    			/* lws_getaddrinfo46 failed, there is no usable result */
    			lwsl_notice("%s: lws_getaddrinfo46 failed %d\n",
    					__func__, n);
    			cce = "ipv6 lws_getaddrinfo46 failed";
    			goto oom4;
    		}
    
    
    		sa6 = ((struct sockaddr_in6 *)result->ai_addr);
    
    
    		sa46.sa6.sin6_family = AF_INET6;
    
    James Devine's avatar
    James Devine committed
    		switch (result->ai_family) {
    
    			if (ipv6only)
    				break;
    
    			bzero((char *)&sa46.sa6.sin6_addr,
    						sizeof(sa46.sa6.sin6_addr));
    			sa46.sa6.sin6_addr.s6_addr[10] = 0xff;
    			sa46.sa6.sin6_addr.s6_addr[11] = 0xff;
    			memcpy(&sa46.sa6.sin6_addr.s6_addr[12],
    
    				&((struct sockaddr_in *)result->ai_addr)->sin_addr,
    							sizeof(struct in_addr));
    
    			lwsl_notice("uplevelling AF_INET to AF_INET6\n");
    
    James Devine's avatar
    James Devine committed
    		case AF_INET6:
    
    Andy Green's avatar
    Andy Green committed
    			memcpy(&sa46.sa6.sin6_addr, &sa6->sin6_addr,
    
    James Devine's avatar
    James Devine committed
    						sizeof(struct in6_addr));
    
    Andy Green's avatar
    Andy Green committed
    			sa46.sa6.sin6_scope_id = sa6->sin6_scope_id;
    			sa46.sa6.sin6_flowinfo = sa6->sin6_flowinfo;
    
    James Devine's avatar
    James Devine committed
    			break;
    		default:
    			lwsl_err("Unknown address family\n");
    			freeaddrinfo(result);
    
    			cce = "unknown address family";
    
    James Devine's avatar
    James Devine committed
    			goto oom4;
    		}
    	} else
    
    #endif /* use ipv6 */
    
    	/* use ipv4 */
    
    James Devine's avatar
    James Devine committed
    	{
    
    		if (!n) {
    			struct addrinfo *res = result;
    
    			/* pick the first AF_INET (IPv4) result */
    
    andSpace's avatar
    andSpace committed
    			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 (n == EAI_SYSTEM) {
    
    andSpace's avatar
    andSpace committed
    			struct hostent *host;
    
    
    			lwsl_info("ipv4 getaddrinfo err, try gethostbyname\n");
    
    andSpace's avatar
    andSpace committed
    			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: %d\n", n);
    
    			cce = "getaddrinfo failed";
    
    James Devine's avatar
    James Devine committed
    			goto oom4;
    
    		if (!p) {
    
    andSpace's avatar
    andSpace committed
    			if (result)
    				freeaddrinfo(result);
    
    			lwsl_err("Couldn't identify address\n");
    
    			cce = "unable to lookup address";
    
    James Devine's avatar
    James Devine committed
    
    
    		sa46.sa4.sin_family = AF_INET;
    		sa46.sa4.sin_addr = *((struct in_addr *)p);
    
    		bzero(&sa46.sa4.sin_zero, sizeof(sa46.sa4.sin_zero));
    
    	if (result)
    		freeaddrinfo(result);
    
    
    #if defined(LWS_WITH_UNIX_SOCK)
    ads_known:
    #endif
    
    
    	/* now we decided on ipv4 or ipv6, set the port */
    
    
    	if (!lws_socket_is_valid(wsi->desc.sockfd)) {
    
    		if (wsi->context->event_loop_ops->check_client_connect_ok &&
    		    wsi->context->event_loop_ops->check_client_connect_ok(wsi)) {
    			cce = "waiting for event loop watcher to close";
    			goto oom4;
    		}
    
    #if defined(LWS_WITH_UNIX_SOCK)
    		if (unix_skt) {
    			wsi->unix_skt = 1;
    			wsi->desc.sockfd = socket(AF_UNIX, SOCK_STREAM, 0);
    		} else
    #endif
    		{
    
    
    #ifdef LWS_WITH_IPV6
    
    			wsi->desc.sockfd = socket(AF_INET6, SOCK_STREAM, 0);
    
    James Devine's avatar
    James Devine committed
    #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";
    
    		if (lws_plat_set_socket_options(wsi->vhost, wsi->desc.sockfd,
    #if defined(LWS_WITH_UNIX_SOCK)
    						unix_skt)) {
    #else
    						0)) {
    #endif
    
    			lwsl_err("Failed to set wsi socket options\n");
    
    			compatible_close(wsi->desc.sockfd);
    
    			cce = "set socket opts failed";
    
    		lwsi_set_state(wsi, LRS_WAITING_CONNECT);
    
    		if (wsi->context->event_loop_ops->accept)
    
    Andy Green's avatar
    Andy Green committed
    			if (wsi->context->event_loop_ops->accept(wsi)) {
    				compatible_close(wsi->desc.sockfd);
    				cce = "event loop accept failed";
    				goto oom4;
    			}
    
    		if (__insert_wsi_socket_into_fds(wsi->context, wsi)) {
    
    			compatible_close(wsi->desc.sockfd);
    
    			cce = "insert wsi failed";
    
    			goto oom4;
    
    Andy Green's avatar
    Andy Green committed
    		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);
    
    		iface = lws_hdr_simple_ptr(wsi, _WSI_TOKEN_CLIENT_IFACE);
    
    		if (iface) {
    
    			n = lws_socket_bind(wsi->vhost, wsi->desc.sockfd, 0,
    					    iface);
    
    			if (n < 0) {
    				cce = "unable to bind socket";
    				goto failed;
    			}
    
    #if defined(LWS_WITH_UNIX_SOCK)
    	if (unix_skt) {
    		psa = (const struct sockaddr *)&sau;
    		n = sizeof(sau);
    
    James Devine's avatar
    James Devine committed
    #endif
    
    #ifdef LWS_WITH_IPV6
    		if (wsi->ipv6) {
    			sa46.sa6.sin6_port = htons(port);
    			n = sizeof(struct sockaddr_in6);
    			psa = (const struct sockaddr *)&sa46;
    		} else
    #endif
    		{
    			sa46.sa4.sin_port = htons(port);
    			n = sizeof(struct sockaddr);
    			psa = (const struct sockaddr *)&sa46;
    		}
    
    	if (connect(wsi->desc.sockfd, (const struct sockaddr *)psa, n) == -1 ||
    
    Andy Green's avatar
    Andy Green committed
    	    LWS_ERRNO == LWS_EISCONN) {
    
    Andy Green's avatar
    Andy Green committed
    		if (LWS_ERRNO == LWS_EALREADY ||
    		    LWS_ERRNO == LWS_EINPROGRESS ||
    		    LWS_ERRNO == LWS_EWOULDBLOCK
    
    #ifdef _WIN32
    			|| LWS_ERRNO == WSAEINVAL
    #endif
    
    Andy Green's avatar
    Andy Green committed
    		) {
    
    			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";
    
    		if (LWS_ERRNO != LWS_EISCONN) {
    
    			lwsl_notice("Connect failed errno=%d\n", LWS_ERRNO);
    			cce = "connect failed";
    
    	lwsl_client("connected\n");
    
    #if defined(LWS_ROLE_H1) || defined(LWS_ROLE_H2)
    
    	/* we are connected to server, or proxy */
    
    
    	/* http proxy */
    
    Andy Green's avatar
    Andy Green committed
    		 * 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,
    
    		n = send(wsi->desc.sockfd, (char *)pt->serv_buf, (int)plen,
    
    Andy Green's avatar
    Andy Green committed
    			 MSG_NOSIGNAL);
    
    		if (n < 0) {
    
    			lwsl_debug("ERROR writing to proxy socket\n");
    
    			cce = "proxy write failed";
    
    Andy Green's avatar
    Andy Green committed
    		lws_set_timeout(wsi, PENDING_TIMEOUT_AWAITING_PROXY_RESPONSE,
    				AWAITING_TIMEOUT);
    
    		lwsi_set_state(wsi, LRS_WAITING_PROXY_REPLY);
    
    #if defined(LWS_WITH_SOCKS5)
    	/* socks proxy */
    	else if (wsi->vhost->socks_proxy_port) {
    		n = send(wsi->desc.sockfd, (char *)pt->serv_buf, plen,
    			 MSG_NOSIGNAL);
    		if (n < 0) {
    
    			lwsl_debug("ERROR writing socks greeting\n");
    
    			cce = "socks write failed";
    			goto failed;
    		}
    
    
    		lws_set_timeout(wsi,
    				PENDING_TIMEOUT_AWAITING_SOCKS_GREETING_REPLY,
    
    				AWAITING_TIMEOUT);
    
    
    		lwsi_set_state(wsi, LRS_WAITING_SOCKS_GREETING_REPLY);
    
    #if defined(LWS_ROLE_H1) || defined(LWS_ROLE_H2)
    
    Andy Green's avatar
    Andy Green committed
    send_hs:
    
    Andy Green's avatar
    Andy Green committed
    	if (wsi_piggyback &&
    
    	    !lws_dll_is_null(&wsi->dll_client_transaction_queue)) {
    
    Andy Green's avatar
    Andy Green committed
    		/*
    		 * We are pipelining on an already-established connection...
    		 * we can skip tls establishment.
    		 */
    
    
    		lwsi_set_state(wsi, LRS_H1C_ISSUE_HANDSHAKE2);
    
    Andy Green's avatar
    Andy Green committed
    
    		/*
    		 * we can't send our headers directly, because they have to
    		 * be sent when the parent is writeable.  The parent will check
    		 * for anybody on his client transaction queue that is in
    
    		 * LRS_H1C_ISSUE_HANDSHAKE2, and let them write.
    
    Andy Green's avatar
    Andy Green committed
    		 *
    		 * If we are trying to do this too early, before the master
    
    		 * connection has written his own headers, then it will just
    		 * wait in the queue until it's possible to send them.
    
    Andy Green's avatar
    Andy Green committed
    		lws_callback_on_writable(wsi_piggyback);
    
    		lwsl_info("%s: wsi %p: waiting to send hdrs (par state 0x%x)\n",
    
    			    __func__, wsi, lwsi_state(wsi_piggyback));
    
    Andy Green's avatar
    Andy Green committed
    	} else {
    
    		lwsl_info("%s: wsi %p: client creating own connection\n",
    			    __func__, wsi);
    
    Andy Green's avatar
    Andy Green committed
    
    
    Andy Green's avatar
    Andy Green committed
    		/* we are making our own connection */
    
    		lwsi_set_state(wsi, LRS_H1C_ISSUE_HANDSHAKE);
    
    Andy Green's avatar
    Andy Green committed
    
    		/*
    		 * 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 many retries from main loop.  To stop that
    		 * becoming endless, cover with a timeout.
    		 */
    
    Andy Green's avatar
    Andy Green committed
    		lws_set_timeout(wsi, PENDING_TIMEOUT_SENT_CLIENT_HANDSHAKE,
    				AWAITING_TIMEOUT);
    
    Andy Green's avatar
    Andy Green committed
    		pfd.fd = wsi->desc.sockfd;
    		pfd.events = LWS_POLLIN;
    		pfd.revents = LWS_POLLIN;
    
    Andy Green's avatar
    Andy Green committed
    		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;
    	}
    
    	if (lwsi_role_client(wsi) /* && lwsi_state_est(wsi) */) {
    
    		wsi->protocol->callback(wsi,
    
    			LWS_CALLBACK_CLIENT_CONNECTION_ERROR,
    
    			wsi->user_space, (void *)cce, strlen(cce));
    		wsi->already_did_cce = 1;
    	}
    
    Andy Green's avatar
    Andy Green committed
    	/* take care that we might be inserted in fds already */
    
    	if (wsi->position_in_fds_table != LWS_NO_FDS_POS)
    
    		goto failed1;
    
    	/*
    	 * We can't be an active client connection any more, if we thought
    	 * that was what we were going to be doing.  It should be if we are
    	 * failing by oom4 path, we are still called by
    	 * lws_client_connect_via_info() and will be returning NULL to that,
    	 * so nobody else should have had a chance to queue on us.
    	 */
    
    	{
    		struct lws_vhost *vhost = wsi->vhost;
    
    		lws_vhost_lock(vhost);
    		__lws_free_wsi(wsi);
    		lws_vhost_unlock(vhost);
    	}
    
    	wsi->protocol->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, "client_connect2");
    
    Andy Green's avatar
    Andy Green committed
    
    
    #if defined(LWS_ROLE_H1) || defined(LWS_ROLE_H2)
    
    
    Andy Green's avatar
    Andy Green committed
    /**
    
     * lws_client_reset() - retarget a connected wsi to start over with a new
     * 			connection (ie, redirect)
    
    Andy Green's avatar
    Andy Green committed
     *			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)
    
    Andy Green's avatar
    Andy Green committed
    {
    
    Andy Green's avatar
    Andy Green committed
    	char origin[300] = "", protocol[300] = "", method[32] = "",
    
    Andy Green's avatar
    Andy Green committed
    	     iface[16] = "", alpn[32] = "", *p;
    
    	if (wsi->redirects == 3) {
    
    Andy Green's avatar
    Andy Green committed
    		lwsl_err("%s: Too many redirects\n", __func__);
    		return NULL;
    	}
    
    	wsi->redirects++;
    
    Andy Green's avatar
    Andy Green committed
    
    
    	p = lws_hdr_simple_ptr(wsi, _WSI_TOKEN_CLIENT_ORIGIN);
    	if (p)
    
    Andy Green's avatar
    Andy Green committed
    		lws_strncpy(origin, p, sizeof(origin));
    
    Andy Green's avatar
    Andy Green committed
    
    
    	p = lws_hdr_simple_ptr(wsi, _WSI_TOKEN_CLIENT_SENT_PROTOCOLS);
    	if (p)
    
    Andy Green's avatar
    Andy Green committed
    		lws_strncpy(protocol, p, sizeof(protocol));
    
    Andy Green's avatar
    Andy Green committed
    
    
    	p = lws_hdr_simple_ptr(wsi, _WSI_TOKEN_CLIENT_METHOD);
    	if (p)
    
    Andy Green's avatar
    Andy Green committed
    		lws_strncpy(method, p, sizeof(method));
    
    Andy Green's avatar
    Andy Green committed
    
    
    	p = lws_hdr_simple_ptr(wsi, _WSI_TOKEN_CLIENT_IFACE);
    	if (p)
    
    		lws_strncpy(iface, p, sizeof(iface));
    
    	p = lws_hdr_simple_ptr(wsi, _WSI_TOKEN_CLIENT_ALPN);
    	if (p)
    		lws_strncpy(alpn, p, sizeof(alpn));
    
    	lwsl_info("redirect ads='%s', port=%d, path='%s', ssl = %d\n",
    
    		   address, port, path, ssl);
    
    
    	/* close the connection by hand */
    
    Andy Green's avatar
    Andy Green committed
    
    
    Andy Green's avatar
    Andy Green committed
    #if defined(LWS_WITH_TLS)
    
    	lws_ssl_close(wsi);
    #endif
    
    
    	__remove_wsi_socket_from_fds(wsi);
    
    
    	if (wsi->context->event_loop_ops->close_handle_manually)
    		wsi->context->event_loop_ops->close_handle_manually(wsi);
    	else
    		compatible_close(wsi->desc.sockfd);
    
    Andy Green's avatar
    Andy Green committed
    #if defined(LWS_WITH_TLS)
    
    #else
    	if (ssl) {
    		lwsl_err("%s: not configured for ssl\n", __func__);
    		return NULL;
    	}
    #endif
    
    
    	wsi->desc.sockfd = LWS_SOCK_INVALID;
    
    	lwsi_set_state(wsi, LRS_UNCONNECTED);
    
    Andy Green's avatar
    Andy Green committed
    	wsi->protocol = NULL;
    	wsi->pending_timeout = NO_PENDING_TIMEOUT;
    
    	wsi->c_port = port;
    
    	wsi->hdr_parsing_completed = 0;
    
    
    	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;
    
    Andy Green's avatar
    Andy Green committed
    
    
    	if (iface[0])
    		if (lws_hdr_simple_create(wsi, _WSI_TOKEN_CLIENT_IFACE,
    					  iface))
    			return NULL;
    
    	if (alpn[0])
    		if (lws_hdr_simple_create(wsi, _WSI_TOKEN_CLIENT_ALPN,
    					  alpn))
    			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;
    
    Andy Green's avatar
    Andy Green committed
    }
    
    
    #if defined(LWS_WITH_HTTP_PROXY) && defined(LWS_WITH_HUBBUB)
    
    Andy Green's avatar
    Andy Green committed
    hubbub_error
    
    Andy Green's avatar
    Andy Green committed
    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:
    
    
    Andy Green's avatar
    Andy Green committed
    		p += lws_snprintf(p, end - p, "<!DOCTYPE %.*s %s ",
    
    Andy Green's avatar
    Andy Green committed
    				(int) token->data.doctype.name.len,
    				token->data.doctype.name.ptr,
    				token->data.doctype.force_quirks ?
    						"(force-quirks) " : "");
    
    		if (token->data.doctype.public_missing)
    
    Andy Green's avatar
    Andy Green committed
    			lwsl_debug("\tpublic: missing\n");
    
    Andy Green's avatar
    Andy Green committed
    		else
    
    Andy Green's avatar
    Andy Green committed
    			p += lws_snprintf(p, end - p, "PUBLIC \"%.*s\"\n",
    
    Andy Green's avatar
    Andy Green committed
    				(int) token->data.doctype.public_id.len,
    				token->data.doctype.public_id.ptr);
    
    		if (token->data.doctype.system_missing)
    
    Andy Green's avatar
    Andy Green committed
    			lwsl_debug("\tsystem: missing\n");
    
    Andy Green's avatar
    Andy Green committed
    		else
    
    Andy Green's avatar
    Andy Green committed
    			p += lws_snprintf(p, end - p, " \"%.*s\">\n",
    
    Andy Green's avatar
    Andy Green committed
    				(int) token->data.doctype.system_id.len,
    				token->data.doctype.system_id.ptr);
    
    		break;
    	case HUBBUB_TOKEN_START_TAG:
    
    Andy Green's avatar
    Andy Green committed
    		p += lws_snprintf(p, end - p, "<%.*s", (int)token->data.tag.name.len,
    
    Andy Green's avatar
    Andy Green committed
    				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;
    
    
    Andy Green's avatar
    Andy Green committed
    				if (strncmp(pp, "http:", 5) && strncmp(pp, "https:", 6)) {
    
    					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);
    					continue;
    
    Andy Green's avatar
    Andy Green committed
    				}
    
    Andy Green's avatar
    Andy Green committed
    			}
    
    			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);
    
    Andy Green's avatar
    Andy Green committed
    		}
    
    Andy Green's avatar
    Andy Green committed
    		p += lws_snprintf(p, end - p, ">");
    
    Andy Green's avatar
    Andy Green committed
    		break;
    	case HUBBUB_TOKEN_END_TAG:
    
    Andy Green's avatar
    Andy Green committed
    		p += lws_snprintf(p, end - p, "</%.*s", (int) token->data.tag.name.len,
    
    Andy Green's avatar
    Andy Green committed
    				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++) {
    
    Andy Green's avatar
    Andy Green committed
    			p += lws_snprintf(p, end - p, " %.*s='%.*s'\n",
    
    Andy Green's avatar
    Andy Green committed
    				(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);
    		}
    
    Andy Green's avatar
    Andy Green committed
    		p += lws_snprintf(p, end - p, ">");
    
    Andy Green's avatar
    Andy Green committed
    		break;
    	case HUBBUB_TOKEN_COMMENT:
    
    Andy Green's avatar
    Andy Green committed
    		p += lws_snprintf(p, end - p, "<!-- %.*s -->\n",
    
    Andy Green's avatar
    Andy Green committed
    				(int) token->data.comment.len,
    				token->data.comment.ptr);
    		break;
    	case HUBBUB_TOKEN_CHARACTER:
    
    Andy Green's avatar
    Andy Green committed
    		if (token->data.character.len == 1) {
    			if (*token->data.character.ptr == '<') {
    				p += lws_snprintf(p, end - p, "&lt;");
    				break;
    			}
    			if (*token->data.character.ptr == '>') {
    				p += lws_snprintf(p, end - p, "&gt;");
    				break;
    			}
    			if (*token->data.character.ptr == '&') {
    				p += lws_snprintf(p, end - p, "&amp;");
    				break;
    			}
    		}
    
    
    Andy Green's avatar
    Andy Green committed
    		p += lws_snprintf(p, end - p, "%.*s", (int) token->data.character.len,
    
    Andy Green's avatar
    Andy Green committed
    				token->data.character.ptr);
    		break;
    	case HUBBUB_TOKEN_EOF:
    
    Andy Green's avatar
    Andy Green committed
    		p += lws_snprintf(p, end - p, "\n");
    
    Andy Green's avatar
    Andy Green committed
    		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;
    }
    
    Andy Green's avatar
    Andy Green committed
    #endif
    
    lws_http_client_connect_via_info2(struct lws *wsi)
    
    Andy Green's avatar
    Andy Green committed
    	struct client_info_stash *stash = wsi->stash;
    
    
    	/*
    	 * we're not necessarily in a position to action these right away,
    
    	 * stash them... we only need during connect phase so into a temp
    	 * allocated stash
    
    	if (lws_hdr_simple_create(wsi, _WSI_TOKEN_CLIENT_PEER_ADDRESS,
    				  stash->address))
    
    	if (lws_hdr_simple_create(wsi, _WSI_TOKEN_CLIENT_URI, stash->path))
    
    	if (lws_hdr_simple_create(wsi, _WSI_TOKEN_CLIENT_HOST, stash->host))
    
    	if (stash->origin)
    
    		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)
    
    Andy Green's avatar
    Andy Green committed
    		if (lws_hdr_simple_create(wsi, _WSI_TOKEN_CLIENT_SENT_PROTOCOLS,
    
    	if (stash->method)
    
    		if (lws_hdr_simple_create(wsi, _WSI_TOKEN_CLIENT_METHOD,
    					  stash->method))
    			goto bail1;
    
    	if (stash->iface)
    
    		if (lws_hdr_simple_create(wsi, _WSI_TOKEN_CLIENT_IFACE,
    					  stash->iface))
    			goto bail1;
    
    	if (stash->alpn)
    		if (lws_hdr_simple_create(wsi, _WSI_TOKEN_CLIENT_ALPN,
    					  stash->alpn))
    			goto bail1;
    
    #if defined(LWS_WITH_SOCKS5)
    	if (!wsi->vhost->socks_proxy_port)
    
    		lws_client_stash_destroy(wsi);
    
    Andy Green's avatar
    Andy Green committed
    	wsi->context->count_wsi_allocated++;
    
    
    	return lws_client_connect_2(wsi);
    
    #if defined(LWS_WITH_SOCKS5)
    	if (!wsi->vhost->socks_proxy_port)
    
    Andy Green's avatar
    Andy Green committed
    		lws_free_set_NULL(wsi->stash);
    
    
    	return NULL;
    }
    
    #if defined(LWS_WITH_SOCKS5)
    void socks_generate_msg(struct lws *wsi, enum socks_msg_type type,
    
    Andy Green's avatar
    Andy Green committed
    			ssize_t *msg_len)
    
    {
    	struct lws_context *context = wsi->context;
    	struct lws_context_per_thread *pt = &context->pt[(int)wsi->tsi];
    
    	ssize_t len = 0, n, passwd_len;
    	short net_num;
    	char *p;
    
    	switch (type) {
    	case SOCKS_MSG_GREETING:
    
    		/* socks version, version 5 only */