Skip to content
Snippets Groups Projects
context.c 54.4 KiB
Newer Older
  • Learn to ignore specific revisions
  • /*
     * libwebsockets - small server side websockets and web server implementation
     *
    
     * Copyright (C) 2010-2018 Andy Green <andy@warmcat.com>
    
     *
     *  This library is free software; you can redistribute it and/or
     *  modify it under the terms of the GNU Lesser General Public
     *  License as published by the Free Software Foundation:
     *  version 2.1 of the License.
     *
     *  This library is distributed in the hope that it will be useful,
     *  but WITHOUT ANY WARRANTY; without even the implied warranty of
     *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
     *  Lesser General Public License for more details.
     *
     *  You should have received a copy of the GNU Lesser General Public
     *  License along with this library; if not, write to the Free Software
     *  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
     *  MA  02110-1301  USA
     */
    
    
    
    #ifndef LWS_BUILD_HASH
    #define LWS_BUILD_HASH "unknown-build-hash"
    #endif
    
    
    const struct lws_role_ops *available_roles[] = {
    
    #if defined(LWS_ROLE_H2)
    	&role_ops_h2,
    #endif
    #if defined(LWS_ROLE_H1)
    	&role_ops_h1,
    #endif
    #if defined(LWS_ROLE_WS)
    	&role_ops_ws,
    #endif
    
    const struct lws_event_loop_ops *available_event_libs[] = {
    #if defined(LWS_WITH_POLL)
    	&event_loop_ops_poll,
    #endif
    #if defined(LWS_WITH_LIBUV)
    	&event_loop_ops_uv,
    #endif
    #if defined(LWS_WITH_LIBEVENT)
    	&event_loop_ops_event,
    #endif
    #if defined(LWS_WITH_LIBEV)
    	&event_loop_ops_ev,
    #endif
    	NULL
    };
    
    
    static const char *library_version = LWS_LIBRARY_VERSION " " LWS_BUILD_HASH;
    
    /**
     * lws_get_library_version: get version and git hash library built from
     *
     *	returns a const char * to a string like "1.1 178d78c"
     *	representing the library version followed by the git head hash it
     *	was built from
     */
    LWS_VISIBLE const char *
    lws_get_library_version(void)
    {
    	return library_version;
    }
    
    
    int
    lws_role_call_alpn_negotiated(struct lws *wsi, const char *alpn)
    {
    #if defined(LWS_WITH_TLS)
    	if (!alpn)
    		return 0;
    
    	lwsl_info("%s: '%s'\n", __func__, alpn);
    
    
    	LWS_FOR_EVERY_AVAILABLE_ROLE_START(ar)
    
    		if (ar->alpn && !strcmp(ar->alpn, alpn) && ar->alpn_negotiated)
    
    			return ar->alpn_negotiated(wsi, alpn);
    	LWS_FOR_EVERY_AVAILABLE_ROLE_END;
    
    #if !defined(LWS_WITHOUT_SERVER)
    int
    lws_role_call_adoption_bind(struct lws *wsi, int type, const char *prot)
    {
    	LWS_FOR_EVERY_AVAILABLE_ROLE_START(ar)
    		if (ar->adoption_bind)
    			if (ar->adoption_bind(wsi, type, prot))
    				return 0;
    	LWS_FOR_EVERY_AVAILABLE_ROLE_END;
    
    	/* fall back to raw socket role if, eg, h1 not configured */
    
    	if (role_ops_raw_skt.adoption_bind &&
    	    role_ops_raw_skt.adoption_bind(wsi, type, prot))
    		return 0;
    
    	/* fall back to raw file role if, eg, h1 not configured */
    
    	if (role_ops_raw_file.adoption_bind &&
    	    role_ops_raw_file.adoption_bind(wsi, type, prot))
    		return 0;
    
    	return 1;
    }
    #endif
    
    #if !defined(LWS_WITHOUT_CLIENT)
    int
    lws_role_call_client_bind(struct lws *wsi,
    			  const struct lws_client_connect_info *i)
    {
    	LWS_FOR_EVERY_AVAILABLE_ROLE_START(ar)
    		if (ar->client_bind) {
    			int m = ar->client_bind(wsi, i);
    			if (m < 0)
    				return m;
    			if (m)
    				return 0;
    		}
    	LWS_FOR_EVERY_AVAILABLE_ROLE_END;
    
    	/* fall back to raw socket role if, eg, h1 not configured */
    
    	if (role_ops_raw_skt.client_bind &&
    	    role_ops_raw_skt.client_bind(wsi, i))
    		return 0;
    
    	return 1;
    }
    #endif
    
    
    Andy Green's avatar
    Andy Green committed
    static const char * const mount_protocols[] = {
    	"http://",
    	"https://",
    	"file://",
    
    	"cgi://",
    	">http://",
    	">https://",
    
    	"callback://"
    
    Andy Green's avatar
    Andy Green committed
    };
    
    
    Andy Green's avatar
    Andy Green committed
    LWS_VISIBLE void *
    
    Andy Green's avatar
    Andy Green committed
    lws_protocol_vh_priv_zalloc(struct lws_vhost *vhost,
    			    const struct lws_protocols *prot, int size)
    
    Andy Green's avatar
    Andy Green committed
    {
    	int n = 0;
    
    	/* allocate the vh priv array only on demand */
    	if (!vhost->protocol_vh_privs) {
    		vhost->protocol_vh_privs = (void **)lws_zalloc(
    
    Andy Green's avatar
    Andy Green committed
    				vhost->count_protocols * sizeof(void *),
    				"protocol_vh_privs");
    
    Andy Green's avatar
    Andy Green committed
    		if (!vhost->protocol_vh_privs)
    			return NULL;
    	}
    
    	while (n < vhost->count_protocols && &vhost->protocols[n] != prot)
    		n++;
    
    
    Andy Green's avatar
    Andy Green committed
    	if (n == vhost->count_protocols) {
    		n = 0;
    		while (n < vhost->count_protocols &&
    		       strcmp(vhost->protocols[n].name, prot->name))
    			n++;
    
    		if (n == vhost->count_protocols)
    			return NULL;
    	}
    
    Andy Green's avatar
    Andy Green committed
    
    
    Andy Green's avatar
    Andy Green committed
    	vhost->protocol_vh_privs[n] = lws_zalloc(size, "vh priv");
    
    Andy Green's avatar
    Andy Green committed
    	return vhost->protocol_vh_privs[n];
    }
    
    LWS_VISIBLE void *
    
    Andy Green's avatar
    Andy Green committed
    lws_protocol_vh_priv_get(struct lws_vhost *vhost,
    			 const struct lws_protocols *prot)
    
    Andy Green's avatar
    Andy Green committed
    {
    	int n = 0;
    
    
    	if (!vhost || !vhost->protocol_vh_privs)
    
    Andy Green's avatar
    Andy Green committed
    		return NULL;
    
    	while (n < vhost->count_protocols && &vhost->protocols[n] != prot)
    		n++;
    
    	if (n == vhost->count_protocols) {
    
    Andy Green's avatar
    Andy Green committed
    		n = 0;
    		while (n < vhost->count_protocols &&
    		       strcmp(vhost->protocols[n].name, prot->name))
    			n++;
    
    		if (n == vhost->count_protocols) {
    			lwsl_err("%s: unknown protocol %p\n", __func__, prot);
    			return NULL;
    		}
    
    Andy Green's avatar
    Andy Green committed
    	}
    
    	return vhost->protocol_vh_privs[n];
    }
    
    
    static const struct lws_protocol_vhost_options *
    
    lws_vhost_protocol_options(struct lws_vhost *vh, const char *name)
    {
    
    	const struct lws_protocol_vhost_options *pvo = vh->pvo;
    
    	while (pvo) {
    		if (!strcmp(pvo->name, name))
    			return pvo;
    		pvo = pvo->next;
    	}
    
    	return NULL;
    }
    
    
    /*
     * inform every vhost that hasn't already done it, that
     * his protocols are initializing
     */
    
    Andy Green's avatar
    Andy Green committed
    LWS_VISIBLE int
    
    Andy Green's avatar
    Andy Green committed
    lws_protocol_init(struct lws_context *context)
    {
    	struct lws_vhost *vh = context->vhost_list;
    
    	const struct lws_protocol_vhost_options *pvo, *pvo1;
    
    Andy Green's avatar
    Andy Green committed
    	struct lws wsi;
    
    	int n, any = 0;
    
    Andy Green's avatar
    Andy Green committed
    
    
    	if (context->doing_protocol_init)
    		return 0;
    
    	context->doing_protocol_init = 1;
    
    
    Andy Green's avatar
    Andy Green committed
    	memset(&wsi, 0, sizeof(wsi));
    	wsi.context = context;
    
    
    	lwsl_info("%s\n", __func__);
    
    Andy Green's avatar
    Andy Green committed
    	while (vh) {
    		wsi.vhost = vh;
    
    
    		/* only do the protocol init once for a given vhost */
    
    		if (vh->created_vhost_protocols ||
    		    (vh->options & LWS_SERVER_OPTION_SKIP_PROTOCOL_INIT))
    
    Andy Green's avatar
    Andy Green committed
    		/* initialize supported protocols on this vhost */
    
    		for (n = 0; n < vh->count_protocols; n++) {
    			wsi.protocol = &vh->protocols[n];
    
    Andy Green's avatar
    Andy Green committed
    			if (!vh->protocols[n].name)
    				continue;
    
    			pvo = lws_vhost_protocol_options(vh,
    							 vh->protocols[n].name);
    
    			if (pvo) {
    
    				/*
    				 * linked list of options specific to
    				 * vh + protocol
    				 */
    
    				pvo1 = pvo;
    				pvo = pvo1->options;
    
    				while (pvo) {
    
    					lwsl_debug(
    
    Andy Green's avatar
    Andy Green committed
    						"    vhost \"%s\", "
    						"protocol \"%s\", "
    						"option \"%s\"\n",
    
    							vh->name,
    							vh->protocols[n].name,
    							pvo->name);
    
    					if (!strcmp(pvo->name, "default")) {
    
    						lwsl_info("Setting default "
    
    						   "protocol for vh %s to %s\n",
    						   vh->name,
    						   vh->protocols[n].name);
    						vh->default_protocol_index = n;
    					}
    
    					if (!strcmp(pvo->name, "raw")) {
    
    						lwsl_info("Setting raw "
    
    						   "protocol for vh %s to %s\n",
    						   vh->name,
    						   vh->protocols[n].name);
    						vh->raw_protocol_index = n;
    					}
    
    					pvo = pvo->next;
    				}
    
    				pvo = pvo1->options;
    			}
    
    Andy Green's avatar
    Andy Green committed
    #if defined(LWS_WITH_TLS)
    
    Andy Green's avatar
    Andy Green committed
    			/*
    
    Andy Green's avatar
    Andy Green committed
    			 * inform all the protocols that they are doing their
    			 * one-time initialization if they want to.
    
    Andy Green's avatar
    Andy Green committed
    			 *
    
    Andy Green's avatar
    Andy Green committed
    			 * NOTE the wsi is all zeros except for the context, vh
    			 * + protocol ptrs so lws_get_context(wsi) etc can work
    
    Andy Green's avatar
    Andy Green committed
    			 */
    
    Andy Green's avatar
    Andy Green committed
    			if (vh->protocols[n].callback(&wsi,
    
    Andy Green's avatar
    Andy Green committed
    					LWS_CALLBACK_PROTOCOL_INIT, NULL,
    
    Andy Green's avatar
    Andy Green committed
    				lws_free(vh->protocol_vh_privs[n]);
    				vh->protocol_vh_privs[n] = NULL;
    				lwsl_err("%s: protocol %s failed init\n", __func__,
    
    Andy Green's avatar
    Andy Green committed
    		}
    
    
    		vh->created_vhost_protocols = 1;
    next:
    
    Andy Green's avatar
    Andy Green committed
    		vh = vh->vhost_next;
    	}
    
    
    	if (!context->protocol_init_done)
    		lws_finalize_startup(context);
    
    
    Andy Green's avatar
    Andy Green committed
    	context->protocol_init_done = 1;
    
    
    	if (any)
    		lws_tls_check_all_cert_lifetimes(context);
    
    
    Andy Green's avatar
    Andy Green committed
    	return 0;
    }
    
    
    Andy Green's avatar
    Andy Green committed
    LWS_VISIBLE int
    lws_callback_http_dummy(struct lws *wsi, enum lws_callback_reasons reason,
    
    		    void *user, void *in, size_t len)
    
    Andy Green's avatar
    Andy Green committed
    	struct lws_ssl_info *si;
    
    #ifdef LWS_WITH_CGI
    	struct lws_cgi_args *args;
    
    Andy Green's avatar
    Andy Green committed
    #endif
    #if defined(LWS_WITH_CGI) || defined(LWS_WITH_HTTP_PROXY)
    
    	char buf[128];
    
    	int n;
    #endif
    
    	switch (reason) {
    
    #if defined(LWS_ROLE_H1) || defined(LWS_ROLE_H2)
    
    	case LWS_CALLBACK_HTTP:
    #ifndef LWS_NO_SERVER
    
    		if (lws_return_http_status(wsi, HTTP_STATUS_NOT_FOUND, NULL))
    			return -1;
    
    
    		if (lws_http_transaction_completed(wsi))
    #endif
    			return -1;
    		break;
    
    #if !defined(LWS_NO_SERVER)
    
    	case LWS_CALLBACK_HTTP_BODY_COMPLETION:
    
    	case LWS_CALLBACK_HTTP_FILE_COMPLETION:
    		if (lws_http_transaction_completed(wsi))
    			return -1;
    		break;
    #endif
    
    
    	case LWS_CALLBACK_HTTP_WRITEABLE:
    #ifdef LWS_WITH_CGI
    
    Andy Green's avatar
    Andy Green committed
    		if (wsi->reason_bf & (LWS_CB_REASON_AUX_BF__CGI_HEADERS |
    				      LWS_CB_REASON_AUX_BF__CGI)) {
    
    			n = lws_cgi_write_split_stdout_headers(wsi);
    			if (n < 0) {
    
    Andy Green's avatar
    Andy Green committed
    				lwsl_debug("AUX_BF__CGI forcing close\n");
    
    Andy Green's avatar
    Andy Green committed
    				lws_rx_flow_control(
    
    					wsi->http.cgi->stdwsi[LWS_STDOUT], 1);
    
    			if (wsi->reason_bf & LWS_CB_REASON_AUX_BF__CGI_HEADERS)
    
    Andy Green's avatar
    Andy Green committed
    				wsi->reason_bf &=
    					~LWS_CB_REASON_AUX_BF__CGI_HEADERS;
    
    				wsi->reason_bf &= ~LWS_CB_REASON_AUX_BF__CGI;
    
    
    			if (wsi->http.cgi && wsi->http.cgi->cgi_transaction_over)
    				return -1;
    
    		if (wsi->reason_bf & LWS_CB_REASON_AUX_BF__CGI_CHUNK_END) {
    
    Andy Green's avatar
    Andy Green committed
    			if (!wsi->http2_substream) {
    				memcpy(buf + LWS_PRE, "0\x0d\x0a\x0d\x0a", 5);
    
    Andy Green's avatar
    Andy Green committed
    				lwsl_debug("writing chunk term and exiting\n");
    				n = lws_write(wsi, (unsigned char *)buf +
    						   LWS_PRE, 5, LWS_WRITE_HTTP);
    
    Andy Green's avatar
    Andy Green committed
    			} else
    
    Andy Green's avatar
    Andy Green committed
    				n = lws_write(wsi, (unsigned char *)buf +
    						   LWS_PRE, 0,
    						   LWS_WRITE_HTTP_FINAL);
    
    Andy Green's avatar
    Andy Green committed
    
    
    			/* always close after sending it */
    			return -1;
    
    Andy Green's avatar
    Andy Green committed
    #if defined(LWS_WITH_HTTP_PROXY)
    
    		if (wsi->reason_bf & LWS_CB_REASON_AUX_BF__PROXY) {
    
    Andy Green's avatar
    Andy Green committed
    			char *px = buf + LWS_PRE;
    			int lenx = sizeof(buf) - LWS_PRE;
    
    Andy Green's avatar
    Andy Green committed
    
    
    Andy Green's avatar
    Andy Green committed
    			/*
    			 * our sink is writeable and our source has something
    			 * to read.  So read a lump of source material of
    			 * suitable size to send or what's available, whichever
    			 * is the smaller.
    			 */
    
    			wsi->reason_bf &= ~LWS_CB_REASON_AUX_BF__PROXY;
    
    Andy Green's avatar
    Andy Green committed
    			if (!lws_get_child(wsi))
    				break;
    
    Andy Green's avatar
    Andy Green committed
    			if (lws_http_client_read(lws_get_child(wsi), &px,
    						 &lenx) < 0)
    
    Andy Green's avatar
    Andy Green committed
    				return -1;
    			break;
    		}
    #endif
    		break;
    
    #if defined(LWS_WITH_HTTP_PROXY)
    	case LWS_CALLBACK_RECEIVE_CLIENT_HTTP:
    		assert(lws_get_parent(wsi));
    		if (!lws_get_parent(wsi))
    			break;
    
    		lws_get_parent(wsi)->reason_bf |= LWS_CB_REASON_AUX_BF__PROXY;
    
    Andy Green's avatar
    Andy Green committed
    		lws_callback_on_writable(lws_get_parent(wsi));
    
    Andy Green's avatar
    Andy Green committed
    	case LWS_CALLBACK_RECEIVE_CLIENT_HTTP_READ:
    		assert(lws_get_parent(wsi));
    		n = lws_write(lws_get_parent(wsi), (unsigned char *)in,
    				len, LWS_WRITE_HTTP);
    		if (n < 0)
    			return -1;
    		break;
    
    	case LWS_CALLBACK_ESTABLISHED_CLIENT_HTTP: {
    		unsigned char *p, *end;
    		char ctype[64], ctlen = 0;
    	
    		p = (unsigned char *)buf + LWS_PRE;
    		end = p + sizeof(buf) - LWS_PRE;
    
    
    Andy Green's avatar
    Andy Green committed
    		if (lws_add_http_header_status(lws_get_parent(wsi),
    					       HTTP_STATUS_OK, &p, end))
    
    Andy Green's avatar
    Andy Green committed
    			return 1;
    		if (lws_add_http_header_by_token(lws_get_parent(wsi),
    				WSI_TOKEN_HTTP_SERVER,
    			    	(unsigned char *)"libwebsockets",
    				13, &p, end))
    			return 1;
    
    
    Andy Green's avatar
    Andy Green committed
    		ctlen = lws_hdr_copy(wsi, ctype, sizeof(ctype),
    				     WSI_TOKEN_HTTP_CONTENT_TYPE);
    
    Andy Green's avatar
    Andy Green committed
    		if (ctlen > 0) {
    			if (lws_add_http_header_by_token(lws_get_parent(wsi),
    				WSI_TOKEN_HTTP_CONTENT_TYPE,
    				(unsigned char *)ctype, ctlen, &p, end))
    				return 1;
    		}
    
    Andy Green's avatar
    Andy Green committed
    
    
    Andy Green's avatar
    Andy Green committed
    		if (lws_finalize_http_header(lws_get_parent(wsi), &p, end))
    			return 1;
    
    		*p = '\0';
    
    Andy Green's avatar
    Andy Green committed
    		n = lws_write(lws_get_parent(wsi),
    			      (unsigned char *)buf + LWS_PRE,
    
    Andy Green's avatar
    Andy Green committed
    			      p - ((unsigned char *)buf + LWS_PRE),
    			      LWS_WRITE_HTTP_HEADERS);
    		if (n < 0)
    			return -1;
    
    		break; }
    
    #endif
    
    
    #ifdef LWS_WITH_CGI
    	/* CGI IO events (POLLIN/OUT) appear here, our default policy is:
    	 *
    	 *  - POST data goes on subprocess stdin
    	 *  - subprocess stdout goes on http via writeable callback
    	 *  - subprocess stderr goes to the logs
    	 */
    	case LWS_CALLBACK_CGI:
    		args = (struct lws_cgi_args *)in;
    		switch (args->ch) { /* which of stdin/out/err ? */
    		case LWS_STDIN:
    			/* TBD stdin rx flow control */
    			break;
    		case LWS_STDOUT:
    
    			/* quench POLLIN on STDOUT until MASTER got writeable */
    			lws_rx_flow_control(args->stdwsi[LWS_STDOUT], 0);
    
    			wsi->reason_bf |= LWS_CB_REASON_AUX_BF__CGI;
    
    			/* when writing to MASTER would not block */
    			lws_callback_on_writable(wsi);
    			break;
    		case LWS_STDERR:
    
    Andy Green's avatar
    Andy Green committed
    			n = lws_get_socket_fd(args->stdwsi[LWS_STDERR]);
    			if (n < 0)
    				break;
    			n = read(n, buf, sizeof(buf) - 2);
    
    			if (n > 0) {
    				if (buf[n - 1] != '\n')
    					buf[n++] = '\n';
    				buf[n] = '\0';
    				lwsl_notice("CGI-stderr: %s\n", buf);
    			}
    			break;
    		}
    		break;
    
    	case LWS_CALLBACK_CGI_TERMINATED:
    
    		lwsl_debug("LWS_CALLBACK_CGI_TERMINATED: %d %" PRIu64 "\n",
    
    				wsi->http.cgi->explicitly_chunked,
    				(uint64_t)wsi->http.cgi->content_length);
    		if (!wsi->http.cgi->explicitly_chunked &&
    		    !wsi->http.cgi->content_length) {
    
    			/* send terminating chunk */
    
    Andy Green's avatar
    Andy Green committed
    			lwsl_debug("LWS_CALLBACK_CGI_TERMINATED: ending\n");
    
    			wsi->reason_bf |= LWS_CB_REASON_AUX_BF__CGI_CHUNK_END;
    
    			lws_callback_on_writable(wsi);
    
    			lws_set_timeout(wsi, PENDING_TIMEOUT_CGI, 3);
    			break;
    
    		return -1;
    
    	case LWS_CALLBACK_CGI_STDIN_DATA:  /* POST body for stdin */
    		args = (struct lws_cgi_args *)in;
    		args->data[args->len] = '\0';
    
    		if (!args->stdwsi[LWS_STDIN])
    			return -1;
    
    Andy Green's avatar
    Andy Green committed
    		n = lws_get_socket_fd(args->stdwsi[LWS_STDIN]);
    		if (n < 0)
    			return -1;
    
    
    #if defined(LWS_WITH_ZLIB)
    		if (wsi->http.cgi->gzip_inflate) {
    				/* gzip handling */
    
    				if (!wsi->http.cgi->gzip_init) {
    					lwsl_err("inflating gzip\n");
    
    					memset(&wsi->http.cgi->inflate, 0, sizeof(wsi->http.cgi->inflate));
    
    					if (inflateInit2(&wsi->http.cgi->inflate, 16 + 15) != Z_OK) {
    						lwsl_err("%s: iniflateInit failed\n", __func__);
    						return -1;
    					}
    
    					wsi->http.cgi->gzip_init = 1;
    				}
    
    				wsi->http.cgi->inflate.next_in = args->data;
    				wsi->http.cgi->inflate.avail_in = args->len;
    
    				do {
    
    					wsi->http.cgi->inflate.next_out = wsi->http.cgi->inflate_buf;
    					wsi->http.cgi->inflate.avail_out = sizeof(wsi->http.cgi->inflate_buf);
    
    					n = inflate(&wsi->http.cgi->inflate, Z_SYNC_FLUSH);
    					lwsl_err("inflate: %d\n", n);
    					switch (n) {
    					case Z_NEED_DICT:
    					case Z_STREAM_ERROR:
    					case Z_DATA_ERROR:
    					case Z_MEM_ERROR:
    						inflateEnd(&wsi->http.cgi->inflate);
    						wsi->http.cgi->gzip_init = 0;
    						lwsl_err("zlib error inflate %d\n", n);
    						return -1;
    					}
    
    					if (wsi->http.cgi->inflate.avail_out != sizeof(wsi->http.cgi->inflate_buf)) {
    						int written;
    
    //				                lwsl_hexdump_notice(wsi->http.cgi->inflate_buf,
      //                                                      sizeof(wsi->http.cgi->inflate_buf) - wsi->http.cgi->inflate.avail_out);
    
    				                written = write(args->stdwsi[LWS_STDIN]->desc.filefd, wsi->http.cgi->inflate_buf,
    							sizeof(wsi->http.cgi->inflate_buf) - wsi->http.cgi->inflate.avail_out);
    
    						if (written != (int)(sizeof(wsi->http.cgi->inflate_buf) - wsi->http.cgi->inflate.avail_out)) {
    				                        lwsl_notice("LWS_CALLBACK_CGI_STDIN_DATA: "
                                        				"sent %d only %d went", n, args->len);
    						}
    						lwsl_err("send inflated on fd %d says %d\n", args->stdwsi[LWS_STDIN]->desc.filefd, written);
    
    						if (n == Z_STREAM_END) {
    							lwsl_err("gzip inflate end\n");
    							inflateEnd(&wsi->http.cgi->inflate);
    							wsi->http.cgi->gzip_init = 0;
    
    							//compatible_close(args->stdwsi[LWS_STDIN]->desc.filefd);
    							//args->stdwsi[LWS_STDIN]->desc.filefd = -1;
    							break;
    						}
    
    					} else
    						break;
    
    					if (wsi->http.cgi->inflate.avail_out)
    						break;
    
    				} while (1);
    
    				return args->len;
    		}
    #endif /* WITH_ZLIB */
    
    
    Andy Green's avatar
    Andy Green committed
    		n = write(n, args->data, args->len);
    
    //		lwsl_hexdump_notice(args->data, args->len);
    
    		if (n < args->len)
    			lwsl_notice("LWS_CALLBACK_CGI_STDIN_DATA: "
    				    "sent %d only %d went", n, args->len);
    
    
    		if (wsi->http.cgi->post_in_expected && args->stdwsi[LWS_STDIN] &&
    		    args->stdwsi[LWS_STDIN]->desc.filefd > 0) {
    			wsi->http.cgi->post_in_expected -= n;
    			if (!wsi->http.cgi->post_in_expected) {
    				struct lws *siwsi = args->stdwsi[LWS_STDIN];
    
    				lwsl_debug("%s: expected POST in end: "
    					   "closing stdin wsi %p, fd %d\n",
    					   __func__, siwsi, siwsi->desc.sockfd);
    
    				__remove_wsi_socket_from_fds(siwsi);
    				lwsi_set_state(siwsi, LRS_DEAD_SOCKET);
    				siwsi->socket_is_permanently_unusable = 1;
    				lws_remove_child_from_any_parent(siwsi);
    				if (wsi->context->event_loop_ops->
    							close_handle_manually) {
    					wsi->context->event_loop_ops->
    						close_handle_manually(siwsi);
    					siwsi->told_event_loop_closed = 1;
    				} else {
    					compatible_close(siwsi->desc.sockfd);
    					__lws_free_wsi(siwsi);
    				}
    				wsi->http.cgi->pipe_fds[LWS_STDIN][1] = -1;
    
    				args->stdwsi[LWS_STDIN] = NULL;
    			}
    		}
    
    
    #endif /* WITH_CGI */
    #endif /* ROLE_ H1 / H2 */
    
    	case LWS_CALLBACK_SSL_INFO:
    
    Andy Green's avatar
    Andy Green committed
    		si = in;
    
    Andy Green's avatar
    Andy Green committed
    		(void)si;
    		lwsl_notice("LWS_CALLBACK_SSL_INFO: where: 0x%x, ret: 0x%x\n",
    			    si->where, si->ret);
    
    #if LWS_MAX_SMP > 1
    	case LWS_CALLBACK_GET_THREAD_ID:
    		return (int)(unsigned long long)pthread_self();
    #endif
    
    
    	return 0;
    }
    
    /* list of supported protocols and callbacks */
    
    static const struct lws_protocols protocols_dummy[] = {
    	/* first protocol must always be HTTP handler */
    
    	{
    
    Andy Green's avatar
    Andy Green committed
    		"http-only",			/* name */
    		lws_callback_http_dummy,	/* callback */
    		0,				/* per_session_data_size */
    		0,				/* rx_buffer_size */
    		0,				/* id */
    		NULL,				/* user */
    		0				/* tx_packet_size */
    
    	},
    	/*
    	 * the other protocols are provided by lws plugins
    	 */
    
    	{ NULL, NULL, 0, 0, 0, NULL, 0} /* terminator */
    
    #ifdef LWS_PLAT_OPTEE
    #undef LWS_HAVE_GETENV
    #endif
    
    
    Andy Green's avatar
    Andy Green committed
    LWS_VISIBLE struct lws_vhost *
    lws_create_vhost(struct lws_context *context,
    
    		 const struct lws_context_creation_info *info)
    
    Andy Green's avatar
    Andy Green committed
    {
    
    Andy Green's avatar
    Andy Green committed
    	struct lws_vhost *vh = lws_zalloc(sizeof(*vh), "create vhost"),
    
    Andy Green's avatar
    Andy Green committed
    			 **vh1 = &context->vhost_list;
    
    	const struct lws_http_mount *mounts;
    
    	const struct lws_protocols *pcols = info->protocols;
    
    Andy Green's avatar
    Andy Green committed
    	const struct lws_protocol_vhost_options *pvo;
    
    Andy Green's avatar
    Andy Green committed
    #ifdef LWS_WITH_PLUGINS
    	struct lws_plugin *plugin = context->plugin_list;
    
    Andy Green's avatar
    Andy Green committed
    #endif
    
    Andy Green's avatar
    Andy Green committed
    	struct lws_protocols *lwsp;
    
    Andy Green's avatar
    Andy Green committed
    	int m, f = !info->pvo;
    
    Andy Green's avatar
    Andy Green committed
    	char buf[20];
    
    #if !defined(LWS_WITHOUT_CLIENT) && defined(LWS_HAVE_GETENV)
    
    Andy Green's avatar
    Andy Green committed
    	char *p;
    
    Andy Green's avatar
    Andy Green committed
    #endif
    
    Andy Green's avatar
    Andy Green committed
    	int n;
    
    Andy Green's avatar
    Andy Green committed
    
    	if (!vh)
    		return NULL;
    
    
    Andy Green's avatar
    Andy Green committed
    #if LWS_MAX_SMP > 1
    	pthread_mutex_init(&vh->lock, NULL);
    #endif
    
    
    	if (!pcols)
    		pcols = &protocols_dummy[0];
    
    Andy Green's avatar
    Andy Green committed
    	vh->context = context;
    	if (!info->vhost_name)
    		vh->name = "default";
    	else
    		vh->name = info->vhost_name;
    
    
    #if defined(LWS_ROLE_H1) || defined(LWS_ROLE_H2)
    	vh->http.error_document_404 = info->error_document_404;
    #endif
    
    	if (info->options & LWS_SERVER_OPTION_ONLY_RAW)
    		lwsl_info("%s set to only support RAW\n", vh->name);
    
    
    Andy Green's avatar
    Andy Green committed
    	vh->iface = info->iface;
    
    Andy Green's avatar
    Andy Green committed
    #if !defined(LWS_WITH_ESP32) && \
    
    Andy Green's avatar
    Andy Green committed
        !defined(OPTEE_TA) && !defined(WIN32)
    
    Andy Green's avatar
    Andy Green committed
    	for (vh->count_protocols = 0;
    
    	     pcols[vh->count_protocols].callback;
    
    Andy Green's avatar
    Andy Green committed
    	     vh->count_protocols++)
    		;
    
    	vh->pvo = info->pvo;
    
    	vh->headers = info->headers;
    
    	vh->finalize = info->finalize;
    	vh->finalize_arg = info->finalize_arg;
    
    Andy Green's avatar
    Andy Green committed
    
    
    	LWS_FOR_EVERY_AVAILABLE_ROLE_START(ar)
    		if (ar->init_vhost)
    			if (ar->init_vhost(vh, info))
    				return NULL;
    	LWS_FOR_EVERY_AVAILABLE_ROLE_END;
    
    Andy Green's avatar
    Andy Green committed
    
    
    Andy Green's avatar
    Andy Green committed
    	if (info->keepalive_timeout)
    		vh->keepalive_timeout = info->keepalive_timeout;
    	else
    		vh->keepalive_timeout = 5;
    
    	if (info->timeout_secs_ah_idle)
    		vh->timeout_secs_ah_idle = info->timeout_secs_ah_idle;
    	else
    		vh->timeout_secs_ah_idle = 10;
    
    
    Andy Green's avatar
    Andy Green committed
    #if defined(LWS_WITH_TLS)
    
    
    	vh->tls.alpn = info->alpn;
    	vh->tls.ssl_info_event_mask = info->ssl_info_event_mask;
    
    
    		lws_strncpy(vh->tls.ecdh_curve, info->ecdh_curve,
    			    sizeof(vh->tls.ecdh_curve));
    
    
    	/* carefully allocate and take a copy of cert + key paths if present */
    	n = 0;
    	if (info->ssl_cert_filepath)
    		n += (int)strlen(info->ssl_cert_filepath) + 1;
    	if (info->ssl_private_key_filepath)
    		n += (int)strlen(info->ssl_private_key_filepath) + 1;
    
    	if (n) {
    
    		vh->tls.key_path = vh->tls.alloc_cert_path = lws_malloc(n, "vh paths");
    
    		if (info->ssl_cert_filepath) {
    			n = (int)strlen(info->ssl_cert_filepath) + 1;
    
    			memcpy(vh->tls.alloc_cert_path, info->ssl_cert_filepath, n);
    			vh->tls.key_path += n;
    
    		}
    		if (info->ssl_private_key_filepath)
    
    			memcpy(vh->tls.key_path, info->ssl_private_key_filepath,
    
    			       strlen(info->ssl_private_key_filepath) + 1);
    	}
    
    Andy Green's avatar
    Andy Green committed
    	/*
    	 * give the vhost a unified list of protocols including the
    	 * ones that came from plugins
    	 */
    
    Andy Green's avatar
    Andy Green committed
    	lwsp = lws_zalloc(sizeof(struct lws_protocols) * (vh->count_protocols +
    				   context->plugin_protocol_count + 1),
    				   "vhost-specific plugin table");
    
    Andy Green's avatar
    Andy Green committed
    	if (!lwsp) {
    		lwsl_err("OOM\n");
    		return NULL;
    	}
    
    Andy Green's avatar
    Andy Green committed
    
    
    Andy Green's avatar
    Andy Green committed
    	m = vh->count_protocols;
    
    	memcpy(lwsp, pcols, sizeof(struct lws_protocols) * m);
    
    Andy Green's avatar
    Andy Green committed
    	/* for compatibility, all protocols enabled on vhost if only
    	 * the default vhost exists.  Otherwise only vhosts who ask
    	 * for a protocol get it enabled.
    	 */
    
    	if (context->options & LWS_SERVER_OPTION_EXPLICIT_VHOSTS)
    
    Andy Green's avatar
    Andy Green committed
    		f = 0;
    	(void)f;
    #ifdef LWS_WITH_PLUGINS
    	if (plugin) {
    
    Andy Green's avatar
    Andy Green committed
    		while (plugin) {
    
    			for (n = 0; n < plugin->caps.count_protocols; n++) {
    				/*
    				 * for compatibility's sake, no pvo implies
    				 * allow all protocols
    				 */
    
    				if (f || lws_vhost_protocol_options(vh,
    
    				    plugin->caps.protocols[n].name)) {
    					memcpy(&lwsp[m],
    					       &plugin->caps.protocols[n],
    					       sizeof(struct lws_protocols));
    					m++;
    					vh->count_protocols++;
    				}
    			}
    
    Andy Green's avatar
    Andy Green committed
    			plugin = plugin->list;
    		}
    
    Andy Green's avatar
    Andy Green committed
    	}
    
    Andy Green's avatar
    Andy Green committed
    #endif
    
    Andy Green's avatar
    Andy Green committed
    
    
    Andy Green's avatar
    Andy Green committed
    	if (
    #ifdef LWS_WITH_PLUGINS
    	    (context->plugin_list) ||
    #endif
    
    	    context->options & LWS_SERVER_OPTION_EXPLICIT_VHOSTS)
    
    Andy Green's avatar
    Andy Green committed
    		vh->protocols = lwsp;
    
    		vh->protocols = pcols;
    
    		lws_free(lwsp);
    
    Andy Green's avatar
    Andy Green committed
    
    
    	vh->same_vh_protocol_list = (struct lws **)
    
    Andy Green's avatar
    Andy Green committed
    			lws_zalloc(sizeof(struct lws *) * vh->count_protocols,
    				   "same vh list");
    
    #if defined(LWS_ROLE_H1) || defined(LWS_ROLE_H2)
    	vh->http.mount_list = info->mounts;
    #endif
    
    Andy Green's avatar
    Andy Green committed
    
    
    #ifdef LWS_WITH_UNIX_SOCK
    
    Yeonjun Lim's avatar
    Yeonjun Lim committed
    	if (LWS_UNIX_SOCK_ENABLED(context)) {
    		lwsl_notice("Creating Vhost '%s' path \"%s\", %d protocols\n",
    
    				vh->name, vh->iface, vh->count_protocols);
    
    Yeonjun Lim's avatar
    Yeonjun Lim committed
    	} else
    #endif
    
    Andy Green's avatar
    Andy Green committed
    	{
    		switch(info->port) {
    		case CONTEXT_PORT_NO_LISTEN:
    			strcpy(buf, "(serving disabled)");
    			break;
    		case CONTEXT_PORT_NO_LISTEN_SERVER:
    			strcpy(buf, "(no listener)");
    			break;
    		default:
    			lws_snprintf(buf, sizeof(buf), "port %u", info->port);
    			break;
    		}
    		lwsl_notice("Creating Vhost '%s' %s, %d protocols, IPv6 %s\n",
    				vh->name, buf, vh->count_protocols,
    				LWS_IPV6_ENABLED(vh) ? "on" : "off");
    	}
    
    	mounts = info->mounts;
    
    Andy Green's avatar
    Andy Green committed
    	while (mounts) {
    
    Sergey Kovalevich's avatar
    Sergey Kovalevich committed
    		(void)mount_protocols[0];
    
    		lwsl_info("   mounting %s%s to %s\n",
    			  mount_protocols[mounts->origin_protocol],
    			  mounts->origin, mounts->mountpoint);
    
    Andy Green's avatar
    Andy Green committed
    
    		/* convert interpreter protocol names to pointers */
    		pvo = mounts->interpret;
    		while (pvo) {
    
    Andy Green's avatar
    Andy Green committed
    			for (n = 0; n < vh->count_protocols; n++) {
    				if (strcmp(pvo->value, vh->protocols[n].name))
    					continue;
    				((struct lws_protocol_vhost_options *)pvo)->
    					value = (const char *)(lws_intptr_t)n;
    				break;
    			}
    
    Andy Green's avatar
    Andy Green committed
    			if (n == vh->count_protocols)
    
    Andy Green's avatar
    Andy Green committed
    				lwsl_err("ignoring unknown interp pr %s\n",
    
    Andy Green's avatar
    Andy Green committed
    					 pvo->value);
    
    Andy Green's avatar
    Andy Green committed
    			pvo = pvo->next;
    		}
    
    
    Andy Green's avatar
    Andy Green committed
    		mounts = mounts->mount_next;
    	}
    
    	vh->listen_port = info->port;
    
    #if defined(LWS_ROLE_H1) || defined(LWS_ROLE_H2)
    	vh->http.http_proxy_port = 0;
    	vh->http.http_proxy_address[0] = '\0';
    #endif
    
    #if defined(LWS_WITH_SOCKS5)
    	vh->socks_proxy_port = 0;
    	vh->socks_proxy_address[0] = '\0';
    #endif
    
    Andy Green's avatar
    Andy Green committed
    
    
    #if !defined(LWS_WITHOUT_CLIENT)
    
    Andy Green's avatar
    Andy Green committed
    	/* either use proxy from info, or try get it from env var */
    
    	/* http proxy */
    
    Andy Green's avatar
    Andy Green committed
    	if (info->http_proxy_address) {
    		/* override for backwards compatibility */
    		if (info->http_proxy_port)
    
    Andy Green's avatar
    Andy Green committed
    		lws_set_proxy(vh, info->http_proxy_address);
    
    Andy Green's avatar
    Andy Green committed
    #ifdef LWS_HAVE_GETENV
    		p = getenv("http_proxy");
    		if (p)
    			lws_set_proxy(vh, p);
    #endif
    	}
    
    #if defined(LWS_WITH_SOCKS5)
    	/* socks proxy */
    	if (info->socks_proxy_address) {
    		/* override for backwards compatibility */
    		if (info->socks_proxy_port)
    			vh->socks_proxy_port = info->socks_proxy_port;
    		lws_set_socks(vh, info->socks_proxy_address);
    	} else {
    #ifdef LWS_HAVE_GETENV
    		p = getenv("socks_proxy");
    		if (p)
    			lws_set_socks(vh, p);
    
    Andy Green's avatar
    Andy Green committed
    #endif
    
    Andy Green's avatar
    Andy Green committed
    	vh->ka_time = info->ka_time;
    	vh->ka_interval = info->ka_interval;
    	vh->ka_probes = info->ka_probes;
    
    
    	if (vh->options & LWS_SERVER_OPTION_STS)
    		lwsl_notice("   STS enabled\n");
    
    
    #ifdef LWS_WITH_ACCESS_LOG
    	if (info->log_filepath) {
    
    		vh->log_fd = lws_open(info->log_filepath,
    
    Andy Green's avatar
    Andy Green committed
    				  O_CREAT | O_APPEND | O_RDWR, 0600);
    
    		if (vh->log_fd == (int)LWS_INVALID_FILE) {
    
    			lwsl_err("unable to open log filepath %s\n",
    				 info->log_filepath);
    			goto bail;
    		}
    
    #ifndef WIN32
    
    		if (context->uid != -1)
    			if (chown(info->log_filepath, context->uid,
    				  context->gid) == -1)
    				lwsl_err("unable to chown log file %s\n",
    						info->log_filepath);
    
    		vh->log_fd = (int)LWS_INVALID_FILE;
    
    	if (lws_context_init_server_ssl(info, vh)) {
    		lwsl_err("%s: lws_context_init_server_ssl failed\n", __func__);
    
    	}
    	if (lws_context_init_client_ssl(info, vh)) {