Skip to content
Snippets Groups Projects
server.c 64 KiB
Newer Older
  • Learn to ignore specific revisions
  • /*
     * libwebsockets - small server side websockets and web server implementation
     *
    
     * Copyright (C) 2010-2016 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
     */
    
    
    #include "private-libwebsockets.h"
    
    
    Andy Green's avatar
    Andy Green committed
    #if defined (LWS_WITH_ESP8266)
    #undef memcpy
    void *memcpy(void *dest, const void *src, size_t n)
    {
    	return ets_memcpy(dest, src, n);
    }
    #endif
    
    
    int
    lws_context_init_server(struct lws_context_creation_info *info,
    
    Andy Green's avatar
    Andy Green committed
    			struct lws_vhost *vhost)
    
    Andy Green's avatar
    Andy Green committed
    #if LWS_POSIX
    
    Andy Green's avatar
    Andy Green committed
    	int n, opt = 1, limit = 1;
    
    Andy Green's avatar
    Andy Green committed
    #endif
    
    	lws_sockfd_type sockfd;
    
    Andy Green's avatar
    Andy Green committed
    	struct lws_vhost *vh;
    
    Andy Green's avatar
    Andy Green committed
    	int m = 0;
    
    
    	/* set up our external listening socket we serve on */
    
    	if (info->port == CONTEXT_PORT_NO_LISTEN)
    		return 0;
    
    
    Andy Green's avatar
    Andy Green committed
    	vh = vhost->context->vhost_list;
    	while (vh) {
    		if (vh->listen_port == info->port) {
    			if ((!info->iface && !vh->iface) ||
    			    (info->iface && vh->iface &&
    			    !strcmp(info->iface, vh->iface))) {
    				vhost->listen_port = info->port;
    				vhost->iface = info->iface;
    				lwsl_notice(" using listen skt from vhost %s\n",
    					    vh->name);
    				return 0;
    			}
    		}
    		vh = vh->vhost_next;
    	}
    
    
    Andy Green's avatar
    Andy Green committed
    #if LWS_POSIX
    
    Andy Green's avatar
    Andy Green committed
    #if defined(__linux__)
    
    Andy Green's avatar
    Andy Green committed
    	limit = vhost->context->count_threads;
    
    Andy Green's avatar
    Andy Green committed
    #endif
    
    	for (m = 0; m < limit; m++) {
    
    Yeonjun Lim's avatar
    Yeonjun Lim committed
    #ifdef LWS_USE_UNIX_SOCK
    
    	if (LWS_UNIX_SOCK_ENABLED(vhost))
    
    Yeonjun Lim's avatar
    Yeonjun Lim committed
    		sockfd = socket(AF_UNIX, SOCK_STREAM, 0);
    	else
    #endif
    
    	if (LWS_IPV6_ENABLED(vhost))
    
    		sockfd = socket(AF_INET6, SOCK_STREAM, 0);
    	else
    #endif
    		sockfd = socket(AF_INET, SOCK_STREAM, 0);
    
    	if (sockfd == -1) {
    
    Andy Green's avatar
    Andy Green committed
    #else
    #if defined(LWS_WITH_ESP8266)
    	sockfd = esp8266_create_tcp_listen_socket(vhost);
    	if (!lws_sockfd_valid(sockfd)) {
    
    
    Andy Green's avatar
    Andy Green committed
    #else
    	sockfd = mbed3_create_tcp_stream_socket();
    	if (!lws_sockfd_valid(sockfd)) {
    
    Andy Green's avatar
    Andy Green committed
    #endif
    
    Andy Green's avatar
    Andy Green committed
    #endif
    
    Andy Green's avatar
    Andy Green committed
    #if LWS_POSIX
    
    	/*
    	 * allow us to restart even if old sockets in TIME_WAIT
    	 */
    
    	if (setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR,
    
    		       (const void *)&opt, sizeof(opt)) < 0) {
    
    		compatible_close(sockfd);
    
    
    #if defined(LWS_USE_IPV6) && defined(IPV6_V6ONLY)
    	if (LWS_IPV6_ENABLED(vhost)) {
    		if (vhost->options & LWS_SERVER_OPTION_IPV6_V6ONLY_MODIFY) {
    			int value = (vhost->options & LWS_SERVER_OPTION_IPV6_V6ONLY_VALUE) ? 1 : 0;
    			if (setsockopt(sockfd, SOL_IPV6, IPV6_V6ONLY,
    					(const void*)&value, sizeof(value)) < 0) {
    				compatible_close(sockfd);
    				return 1;
    			}
    		}
    	}
    #endif
    
    
    #if defined(__linux__) && defined(SO_REUSEPORT) && LWS_MAX_SMP > 1
    
    Andy Green's avatar
    Andy Green committed
    	if (vhost->context->count_threads > 1)
    
    		if (setsockopt(sockfd, SOL_SOCKET, SO_REUSEPORT,
    				(const void *)&opt, sizeof(opt)) < 0) {
    			compatible_close(sockfd);
    			return 1;
    		}
    
    Andy Green's avatar
    Andy Green committed
    #endif
    
    Andy Green's avatar
    Andy Green committed
    #endif
    
    Andy Green's avatar
    Andy Green committed
    	lws_plat_set_socket_options(vhost, sockfd);
    
    Andy Green's avatar
    Andy Green committed
    #if LWS_POSIX
    
    	n = lws_socket_bind(vhost, sockfd, info->port, info->iface);
    
    Andy Green's avatar
    Andy Green committed
    		goto bail;
    
    Andy Green's avatar
    Andy Green committed
    #endif
    
    Andy Green's avatar
    Andy Green committed
    	vhost->listen_port = info->port;
    	vhost->iface = info->iface;
    
    	wsi = lws_zalloc(sizeof(struct lws));
    
    	if (wsi == NULL) {
    		lwsl_err("Out of mem\n");
    
    Andy Green's avatar
    Andy Green committed
    		goto bail;
    
    Andy Green's avatar
    Andy Green committed
    	wsi->context = vhost->context;
    
    Andy Green's avatar
    Andy Green committed
    	wsi->mode = LWSCM_SERVER_LISTENER;
    
    Andy Green's avatar
    Andy Green committed
    	wsi->protocol = vhost->protocols;
    
    Andy Green's avatar
    Andy Green committed
    	wsi->tsi = m;
    
    Andy Green's avatar
    Andy Green committed
    	wsi->vhost = vhost;
    	wsi->listener = 1;
    
    Andy Green's avatar
    Andy Green committed
    	vhost->context->pt[m].wsi_listening = wsi;
    	if (insert_wsi_socket_into_fds(vhost->context, wsi))
    
    Andy Green's avatar
    Andy Green committed
    		goto bail;
    
    Andy Green's avatar
    Andy Green committed
    	vhost->context->count_wsi_allocated++;
    	vhost->lserv_wsi = wsi;
    
    Andy Green's avatar
    Andy Green committed
    #if LWS_POSIX
    
    Andy Green's avatar
    Andy Green committed
    	listen(wsi->sock, LWS_SOMAXCONN);
    
    Andy Green's avatar
    Andy Green committed
    	} /* for each thread able to independently listen */
    
    Andy Green's avatar
    Andy Green committed
    #else
    #if defined(LWS_WITH_ESP8266)
    	esp8266_tcp_stream_bind(wsi->sock, info->port, wsi);
    
    Andy Green's avatar
    Andy Green committed
    #else
    
    Andy Green's avatar
    Andy Green committed
    	mbed3_tcp_stream_bind(wsi->sock, info->port, wsi);
    
    Andy Green's avatar
    Andy Green committed
    #endif
    
    Andy Green's avatar
    Andy Green committed
    #endif
    
    Yeonjun Lim's avatar
    Yeonjun Lim committed
    	if (!lws_check_opt(info->options, LWS_SERVER_OPTION_EXPLICIT_VHOSTS)) {
    #ifdef LWS_USE_UNIX_SOCK
    
    		if (LWS_UNIX_SOCK_ENABLED(vhost))
    
    Yeonjun Lim's avatar
    Yeonjun Lim committed
    			lwsl_notice(" Listening on \"%s\"\n", info->iface);
    		else
    #endif
    			lwsl_notice(" Listening on port %d\n", info->port);
            }
    
    Andy Green's avatar
    Andy Green committed
    
    bail:
    	compatible_close(sockfd);
    
    	return 1;
    
    Andy Green's avatar
    Andy Green committed
    _lws_server_listen_accept_flow_control(struct lws *twsi, int on)
    
    Andy Green's avatar
    Andy Green committed
    	struct lws_context_per_thread *pt = &twsi->context->pt[(int)twsi->tsi];
    	struct lws *wsi = pt->wsi_listening;
    
    Andy Green's avatar
    Andy Green committed
    	if (!wsi || twsi->context->being_destroyed)
    
    Andy Green's avatar
    Andy Green committed
    	lwsl_debug("%s: Thr %d: LISTEN wsi %p: state %d\n",
    		   __func__, twsi->tsi, (void *)wsi, on);
    
    	if (on)
    		n = lws_change_pollfd(wsi, 0, LWS_POLLIN);
    	else
    		n = lws_change_pollfd(wsi, LWS_POLLIN, 0);
    
    	return n;
    }
    
    
    Andy Green's avatar
    Andy Green committed
    #if defined(LWS_WITH_ESP8266)
    #undef strchr
    #define strchr ets_strchr
    #endif
    
    
    Andy Green's avatar
    Andy Green committed
    struct lws_vhost *
    lws_select_vhost(struct lws_context *context, int port, const char *servername)
    {
    	struct lws_vhost *vhost = context->vhost_list;
    
    	const char *p;
    	int n, m, colon;
    
    	n = strlen(servername);
    	colon = n;
    	p = strchr(servername, ':');
    	if (p)
    		colon = p - servername;
    
    	/* first try exact matches */
    
    Andy Green's avatar
    Andy Green committed
    
    	while (vhost) {
    		if (port == vhost->listen_port &&
    
    		    !strncmp(vhost->name, servername, colon)) {
    
    Andy Green's avatar
    Andy Green committed
    			lwsl_info("SNI: Found: %s\n", servername);
    			return vhost;
    		}
    		vhost = vhost->vhost_next;
    	}
    
    
    	/*
    	 * if no exact matches, try matching *.vhost-name
    	 * unintentional matches are possible but resolve to x.com for *.x.com
    	 * which is reasonable.  If exact match exists we already chose it and
    	 * never reach here.  SSL will still fail it if the cert doesn't allow
    	 * *.x.com.
    	 */
    
    	vhost = context->vhost_list;
    	while (vhost) {
    		m = strlen(vhost->name);
    		if (port == vhost->listen_port &&
    		    m <= (colon - 2) &&
    		    servername[colon - m - 1] == '.' &&
    		    !strncmp(vhost->name, servername + colon - m, m)) {
    			lwsl_info("SNI: Found %s on wildcard: %s\n",
    				    servername, vhost->name);
    			return vhost;
    		}
    		vhost = vhost->vhost_next;
    	}
    
    
    Andy Green's avatar
    Andy Green committed
    	return NULL;
    }
    
    
    Andy Green's avatar
    Andy Green committed
    LWS_VISIBLE LWS_EXTERN const struct lws_protocols *
    lws_vhost_name_to_protocol(struct lws_vhost *vh, const char *name)
    {
    	int n;
    
    	for (n = 0; n < vh->count_protocols; n++)
    		if (!strcmp(name, vh->protocols[n].name))
    			return &vh->protocols[n];
    
    	return NULL;
    }
    
    
    Andy Green's avatar
    Andy Green committed
    LWS_VISIBLE LWS_EXTERN const char *
    lws_get_mimetype(const char *file, const struct lws_http_mount *m)
    
    Andy Green's avatar
    Andy Green committed
    {
    	int n = strlen(file);
    
    	const struct lws_protocol_vhost_options *pvo = NULL;
    
    	if (m)
    		pvo = m->extra_mimetypes;
    
    Andy Green's avatar
    Andy Green committed
    
    	if (n < 5)
    		return NULL;
    
    	if (!strcmp(&file[n - 4], ".ico"))
    		return "image/x-icon";
    
    
    	if (!strcmp(&file[n - 4], ".gif"))
    		return "image/gif";
    
    	if (!strcmp(&file[n - 3], ".js"))
    		return "text/javascript";
    
    
    Andy Green's avatar
    Andy Green committed
    	if (!strcmp(&file[n - 4], ".png"))
    		return "image/png";
    
    
    Andy Green's avatar
    Andy Green committed
    	if (!strcmp(&file[n - 4], ".jpg"))
    		return "image/jpeg";
    
    
    	if (!strcmp(&file[n - 3], ".gz"))
    		return "application/gzip";
    
    	if (!strcmp(&file[n - 4], ".JPG"))
    		return "image/jpeg";
    
    
    Andy Green's avatar
    Andy Green committed
    	if (!strcmp(&file[n - 5], ".html"))
    		return "text/html";
    
    	if (!strcmp(&file[n - 4], ".css"))
    		return "text/css";
    
    
    	if (!strcmp(&file[n - 4], ".txt"))
    		return "text/plain";
    
    
    	if (!strcmp(&file[n - 4], ".ttf"))
    		return "application/x-font-ttf";
    
    
    	if (!strcmp(&file[n - 5], ".woff"))
    		return "application/font-woff";
    
    	if (!strcmp(&file[n - 4], ".xml"))
    		return "application/xml";
    
    
    	while (pvo) {
    		if (!strcmp(&file[n - strlen(pvo->name)], pvo->name))
    			return pvo->value;
    
    		pvo = pvo->next;
    	}
    
    
    Andy Green's avatar
    Andy Green committed
    	return NULL;
    }
    
    
    static int
    lws_http_serve(struct lws *wsi, char *uri, const char *origin,
    	       const struct lws_http_mount *m)
    
    Andy Green's avatar
    Andy Green committed
    {
    
    Andy Green's avatar
    Andy Green committed
    	const struct lws_protocol_vhost_options *pvo = m->interpret;
    	struct lws_process_html_args args;
    
    Andy Green's avatar
    Andy Green committed
    	const char *mimetype;
    
    Andy Green's avatar
    Andy Green committed
    #if !defined(_WIN32_WCE) && !defined(LWS_WITH_ESP8266)
    
    Andy Green's avatar
    Andy Green committed
    	struct stat st;
    
    Andy Green's avatar
    Andy Green committed
    	int spin = 0;
    
    Andy Green's avatar
    Andy Green committed
    	char path[256], sym[512];
    
    Andy Green's avatar
    Andy Green committed
    	unsigned char *p = (unsigned char *)sym + 32 + LWS_PRE, *start = p;
    	unsigned char *end = p + sizeof(sym) - 32 - LWS_PRE;
    
    Andy Green's avatar
    Andy Green committed
    #if !defined(WIN32) && LWS_POSIX
    
    Stephan Eberle's avatar
    Stephan Eberle committed
    	size_t len;
    
    Andy Green's avatar
    Andy Green committed
    	int n;
    
    Andy Green's avatar
    Andy Green committed
    
    	snprintf(path, sizeof(path) - 1, "%s/%s", origin, uri);
    
    
    Andy Green's avatar
    Andy Green committed
    #if !defined(_WIN32_WCE) && !defined(LWS_WITH_ESP8266)
    
    Andy Green's avatar
    Andy Green committed
    	do {
    		spin++;
    
    		if (stat(path, &st)) {
    
    Andy Green's avatar
    Andy Green committed
    			lwsl_info("unable to stat %s\n", path);
    
    Andy Green's avatar
    Andy Green committed
    			goto bail;
    		}
    
    		lwsl_debug(" %s mode %d\n", path, S_IFMT & st.st_mode);
    
    Andy Green's avatar
    Andy Green committed
    #if !defined(WIN32) && LWS_POSIX
    
    Andy Green's avatar
    Andy Green committed
    		if ((S_IFMT & st.st_mode) == S_IFLNK) {
    
    			len = readlink(path, sym, sizeof(sym) - 1);
    			if (len) {
    
    Andy Green's avatar
    Andy Green committed
    				lwsl_err("Failed to read link %s\n", path);
    				goto bail;
    			}
    
    			sym[len] = '\0';
    
    Andy Green's avatar
    Andy Green committed
    			lwsl_debug("symlink %s -> %s\n", path, sym);
    			snprintf(path, sizeof(path) - 1, "%s", sym);
    		}
    
    Andy Green's avatar
    Andy Green committed
    #endif
    
    Andy Green's avatar
    Andy Green committed
    		if ((S_IFMT & st.st_mode) == S_IFDIR) {
    			lwsl_debug("default filename append to dir\n");
    			snprintf(path, sizeof(path) - 1, "%s/%s/index.html",
    				 origin, uri);
    		}
    
    	} while ((S_IFMT & st.st_mode) != S_IFREG && spin < 5);
    
    
    Andy Green's avatar
    Andy Green committed
    	if (spin == 5)
    
    Andy Green's avatar
    Andy Green committed
    		lwsl_err("symlink loop %s \n", path);
    
    Andy Green's avatar
    Andy Green committed
    
    	n = sprintf(sym, "%08lX%08lX", (unsigned long)st.st_size,
    				   (unsigned long)st.st_mtime);
    
    	if (lws_hdr_total_length(wsi, WSI_TOKEN_HTTP_IF_NONE_MATCH)) {
    		/*
    		 * he thinks he has some version of it already,
    		 * check if the tag matches
    		 */
    		if (!strcmp(sym, lws_hdr_simple_ptr(wsi, WSI_TOKEN_HTTP_IF_NONE_MATCH))) {
    
    
    Andy Green's avatar
    Andy Green committed
    			lwsl_debug("%s: ETAG match %s %s\n", __func__,
    				   uri, origin);
    
    Andy Green's avatar
    Andy Green committed
    
    			/* we don't need to send the payload */
    			if (lws_add_http_header_status(wsi, 304, &p, end))
    				return -1;
    			if (lws_add_http_header_by_token(wsi,
    					WSI_TOKEN_HTTP_ETAG,
    					(unsigned char *)sym, n, &p, end))
    				return -1;
    			if (lws_finalize_http_header(wsi, &p, end))
    				return -1;
    
    			n = lws_write(wsi, start, p - start,
    
    Andy Green's avatar
    Andy Green committed
    				      LWS_WRITE_HTTP_HEADERS);
    
    Andy Green's avatar
    Andy Green committed
    			if (n != (p - start)) {
    				lwsl_err("_write returned %d from %d\n", n, p - start);
    				return -1;
    			}
    
    			return lws_http_transaction_completed(wsi);
    		}
    
    Andy Green's avatar
    Andy Green committed
    	if (lws_add_http_header_by_token(wsi, WSI_TOKEN_HTTP_ETAG,
    			(unsigned char *)sym, n, &p, end))
    		return -1;
    
    Andy Green's avatar
    Andy Green committed
    
    
    Andy Green's avatar
    Andy Green committed
    	mimetype = lws_get_mimetype(path, m);
    
    Andy Green's avatar
    Andy Green committed
    	if (!mimetype) {
    
    Andy Green's avatar
    Andy Green committed
    		lwsl_err("unknown mimetype for %s\n", path);
    
    Andy Green's avatar
    Andy Green committed
    		goto bail;
    
    Andy Green's avatar
    Andy Green committed
    	}
    
    
    Andy Green's avatar
    Andy Green committed
    	wsi->sending_chunked = 0;
    
    	/*
    	 * check if this is in the list of file suffixes to be interpreted by
    	 * a protocol
    	 */
    	while (pvo) {
    		n = strlen(path);
    		if (n > (int)strlen(pvo->name) &&
    		    !strcmp(&path[n - strlen(pvo->name)], pvo->name)) {
    			wsi->sending_chunked = 1;
    			wsi->protocol_interpret_idx = (char)(long)pvo->value;
    			lwsl_info("want %s interpreted by %s\n", path,
    				    wsi->vhost->protocols[(int)(long)(pvo->value)].name);
    			wsi->protocol = &wsi->vhost->protocols[(int)(long)(pvo->value)];
    			if (lws_ensure_user_space(wsi))
    				return -1;
    			break;
    		}
    		pvo = pvo->next;
    	}
    
    	if (m->protocol) {
    		const struct lws_protocols *pp = lws_vhost_name_to_protocol(
    							wsi->vhost, m->protocol);
    
    
    Andy Green's avatar
    Andy Green committed
    		if (lws_bind_protocol(wsi, pp))
    			return 1;
    
    Andy Green's avatar
    Andy Green committed
    		args.p = (char *)p;
    		args.max_len = end - p;
    		if (pp->callback(wsi, LWS_CALLBACK_ADD_HEADERS,
    					  wsi->user_space, &args, 0))
    			return -1;
    		p = (unsigned char *)args.p;
    	}
    
    
    Andy Green's avatar
    Andy Green committed
    	n = lws_serve_http_file(wsi, path, mimetype, (char *)start, p - start);
    
    Andy Green's avatar
    Andy Green committed
    
    
    Andy Green's avatar
    Andy Green committed
    	if (n < 0 || ((n > 0) && lws_http_transaction_completed(wsi)))
    		return -1; /* error or can't reuse connection: close the socket */
    
    	return 0;
    
    Andy Green's avatar
    Andy Green committed
    bail:
    
    	return -1;
    
    Andy Green's avatar
    Andy Green committed
    }
    
    
    Andy Green's avatar
    Andy Green committed
    const struct lws_http_mount *
    lws_find_mount(struct lws *wsi, const char *uri_ptr, int uri_len)
    {
    	const struct lws_http_mount *hm, *hit = NULL;
    	int best = 0;
    
    	hm = wsi->vhost->mount_list;
    	while (hm) {
    		if (uri_len >= hm->mountpoint_len &&
    		    !strncmp(uri_ptr, hm->mountpoint, hm->mountpoint_len) &&
    		    (uri_ptr[hm->mountpoint_len] == '\0' ||
    		     uri_ptr[hm->mountpoint_len] == '/' ||
    		     hm->mountpoint_len == 1)
    		    ) {
    			if (hm->origin_protocol == LWSMPRO_CALLBACK ||
    			    ((hm->origin_protocol == LWSMPRO_CGI ||
    			     lws_hdr_total_length(wsi, WSI_TOKEN_GET_URI) ||
    			     hm->protocol) &&
    			    hm->mountpoint_len > best)) {
    				best = hm->mountpoint_len;
    				hit = hm;
    			}
    		}
    		hm = hm->mount_next;
    	}
    
    	return hit;
    }
    
    
    int
    lws_http_action(struct lws *wsi)
    
    	struct lws_context_per_thread *pt = &wsi->context->pt[(int)wsi->tsi];
    
    	enum http_connection_type connection_type;
    
    	enum http_version request_version;
    
    Andy Green's avatar
    Andy Green committed
    	struct lws_process_html_args args;
    
    Andy Green's avatar
    Andy Green committed
    	const struct lws_http_mount *hit = NULL;
    
    	unsigned int n, count = 0;
    
    	char http_version_str[10];
    	char http_conn_str[20];
    
    	int http_version_len;
    
    Andy Green's avatar
    Andy Green committed
    	char *uri_ptr = NULL, *s;
    	int uri_len = 0;
    
    	int meth = -1;
    
    	static const unsigned char methods[] = {
    		WSI_TOKEN_GET_URI,
    		WSI_TOKEN_POST_URI,
    		WSI_TOKEN_OPTIONS_URI,
    		WSI_TOKEN_PUT_URI,
    		WSI_TOKEN_PATCH_URI,
    		WSI_TOKEN_DELETE_URI,
    
    #ifdef LWS_USE_HTTP2
    
    		WSI_TOKEN_HTTP_COLON_PATH,
    
    #if defined(_DEBUG) || defined(LWS_WITH_ACCESS_LOG)
    
    	static const char * const method_names[] = {
    		"GET", "POST", "OPTIONS", "PUT", "PATCH", "DELETE",
    #ifdef LWS_USE_HTTP2
    		":path",
    #endif
    	};
    
    #endif
    
    Andy Green's avatar
    Andy Green committed
    	static const char * const oprot[] = {
    		"http://", "https://"
    	};
    
    	/* it's not websocket.... shall we accept it as http? */
    
    	for (n = 0; n < ARRAY_SIZE(methods); n++)
    		if (lws_hdr_total_length(wsi, methods[n]))
    			count++;
    	if (!count) {
    
    		lwsl_warn("Missing URI in HTTP request\n");
    		goto bail_nuke_ah;
    	}
    
    
    	if (count != 1) {
    		lwsl_warn("multiple methods?\n");
    
    		goto bail_nuke_ah;
    	}
    
    
    	if (lws_ensure_user_space(wsi))
    
    		goto bail_nuke_ah;
    
    
    	for (n = 0; n < ARRAY_SIZE(methods); n++)
    		if (lws_hdr_total_length(wsi, methods[n])) {
    			uri_ptr = lws_hdr_simple_ptr(wsi, methods[n]);
    			uri_len = lws_hdr_total_length(wsi, methods[n]);
    			lwsl_info("Method: %s request for '%s'\n",
    				  	method_names[n], uri_ptr);
    
    	/* we insist on absolute paths */
    
    	if (uri_ptr[0] != '/') {
    		lws_return_http_status(wsi, HTTP_STATUS_FORBIDDEN, NULL);
    
    		goto bail_nuke_ah;
    	}
    
    
    	/* HTTP header had a content length? */
    
    	wsi->u.http.content_length = 0;
    
    	if (lws_hdr_total_length(wsi, WSI_TOKEN_POST_URI) ||
    		lws_hdr_total_length(wsi, WSI_TOKEN_PATCH_URI) ||
    		lws_hdr_total_length(wsi, WSI_TOKEN_PUT_URI))
    
    		wsi->u.http.content_length = 100 * 1024 * 1024;
    
    	if (lws_hdr_total_length(wsi, WSI_TOKEN_HTTP_CONTENT_LENGTH)) {
    		lws_hdr_copy(wsi, content_length_str,
    
    			     sizeof(content_length_str) - 1,
    			     WSI_TOKEN_HTTP_CONTENT_LENGTH);
    
    		wsi->u.http.content_length = atoi(content_length_str);
    	}
    
    
    Andy Green's avatar
    Andy Green committed
    	if (wsi->http2_substream) {
    		wsi->u.http.request_version = HTTP_VERSION_2;
    	} else {
    		/* http_version? Default to 1.0, override with token: */
    		request_version = HTTP_VERSION_1_0;
    
    		/* Works for single digit HTTP versions. : */
    		http_version_len = lws_hdr_total_length(wsi, WSI_TOKEN_HTTP);
    		if (http_version_len > 7) {
    			lws_hdr_copy(wsi, http_version_str,
    					sizeof(http_version_str) - 1, WSI_TOKEN_HTTP);
    			if (http_version_str[5] == '1' && http_version_str[7] == '1')
    				request_version = HTTP_VERSION_1_1;
    		}
    		wsi->u.http.request_version = request_version;
    
    Andy Green's avatar
    Andy Green committed
    		/* HTTP/1.1 defaults to "keep-alive", 1.0 to "close" */
    		if (request_version == HTTP_VERSION_1_1)
    
    			connection_type = HTTP_CONNECTION_KEEP_ALIVE;
    		else
    
    Andy Green's avatar
    Andy Green committed
    			connection_type = HTTP_CONNECTION_CLOSE;
    
    		/* Override default if http "Connection:" header: */
    		if (lws_hdr_total_length(wsi, WSI_TOKEN_CONNECTION)) {
    			lws_hdr_copy(wsi, http_conn_str, sizeof(http_conn_str) - 1,
    				     WSI_TOKEN_CONNECTION);
    			http_conn_str[sizeof(http_conn_str) - 1] = '\0';
    			if (!strcasecmp(http_conn_str, "keep-alive"))
    				connection_type = HTTP_CONNECTION_KEEP_ALIVE;
    			else
    				if (!strcasecmp(http_conn_str, "close"))
    					connection_type = HTTP_CONNECTION_CLOSE;
    		}
    		wsi->u.http.connection_type = connection_type;
    
    	n = wsi->protocol->callback(wsi, LWS_CALLBACK_FILTER_HTTP_CONNECTION,
    
    				    wsi->user_space, uri_ptr, uri_len);
    
    Andy Green's avatar
    Andy Green committed
    	if (n) {
    		lwsl_info("LWS_CALLBACK_HTTP closing\n");
    
    Andy Green's avatar
    Andy Green committed
    		return 1;
    
    Andy Green's avatar
    Andy Green committed
    	 * if there is content supposed to be coming,
    	 * put a timeout on it having arrived
    
    Andy Green's avatar
    Andy Green committed
    	lws_set_timeout(wsi, PENDING_TIMEOUT_HTTP_CONTENT,
    
    Andy Green's avatar
    Andy Green committed
    			wsi->context->timeout_secs);
    
    #ifdef LWS_OPENSSL_SUPPORT
    
    	if (wsi->redirect_to_https) {
    		/*
    
    		 * we accepted http:// only so we could redirect to
    
    		 * https://, so issue the redirect.  Create the redirection
    		 * URI from the host: header and ignore the path part
    		 */
    		unsigned char *start = pt->serv_buf + LWS_PRE, *p = start,
    			      *end = p + 512;
    
    		if (!lws_hdr_total_length(wsi, WSI_TOKEN_HOST))
    			goto bail_nuke_ah;
    
    Andy Green's avatar
    Andy Green committed
    		n = sprintf((char *)end, "https://%s/",
    
    			    lws_hdr_simple_ptr(wsi, WSI_TOKEN_HOST));
    
    Andy Green's avatar
    Andy Green committed
    		n = lws_http_redirect(wsi, HTTP_STATUS_MOVED_PERMANENTLY,
    				      end, n, &p, end);
    
    			goto bail_nuke_ah;
    
    		return lws_http_transaction_completed(wsi);
    	}
    
    Andy Green's avatar
    Andy Green committed
    #endif
    
    #ifdef LWS_WITH_ACCESS_LOG
    	/*
    	 * Produce Apache-compatible log string for wsi, like this:
    	 *
    	 * 2.31.234.19 - - [27/Mar/2016:03:22:44 +0800]
    	 * "GET /aep-screen.png HTTP/1.1"
    	 * 200 152987 "https://libwebsockets.org/index.html"
    	 * "Mozilla/5.0 (Macint... Chrome/49.0.2623.87 Safari/537.36"
    	 *
    	 */
    	{
    		static const char * const hver[] = {
    			"http/1.0", "http/1.1", "http/2"
    		};
    #ifdef LWS_USE_IPV6
    		char ads[INET6_ADDRSTRLEN];
    #else
    		char ads[INET_ADDRSTRLEN];
    #endif
    		char da[64];
    		const char *pa, *me;
    		struct tm *tmp;
    		time_t t = time(NULL);
    		int l = 256;
    
    		if (wsi->access_log_pending)
    			lws_access_log(wsi);
    
    		wsi->access_log.header_log = lws_malloc(l);
    
    Andy Green's avatar
    Andy Green committed
    		if (wsi->access_log.header_log) {
    
    Andy Green's avatar
    Andy Green committed
    			tmp = localtime(&t);
    			if (tmp)
    				strftime(da, sizeof(da), "%d/%b/%Y:%H:%M:%S %z", tmp);
    			else
    				strcpy(da, "01/Jan/1970:00:00:00 +0000");
    
    Andy Green's avatar
    Andy Green committed
    			pa = lws_get_peer_simple(wsi, ads, sizeof(ads));
    			if (!pa)
    				pa = "(unknown)";
    
    Andy Green's avatar
    Andy Green committed
    			if (meth >= 0)
    				me = method_names[meth];
    			else
    				me = "unknown";
    
    			snprintf(wsi->access_log.header_log, l,
    				 "%s - - [%s] \"%s %s %s\"",
    				 pa, da, me, uri_ptr,
    				 hver[wsi->u.http.request_version]);
    
    			l = lws_hdr_total_length(wsi, WSI_TOKEN_HTTP_USER_AGENT);
    			if (l) {
    				wsi->access_log.user_agent = lws_malloc(l + 2);
    				if (wsi->access_log.user_agent)
    					lws_hdr_copy(wsi, wsi->access_log.user_agent,
    							l + 1, WSI_TOKEN_HTTP_USER_AGENT);
    				else
    					lwsl_err("OOM getting user agent\n");
    			}
    			wsi->access_log_pending = 1;
    
    Andy Green's avatar
    Andy Green committed
    	/* can we serve it from the mount list? */
    
    
    Andy Green's avatar
    Andy Green committed
    	hit = lws_find_mount(wsi, uri_ptr, uri_len);
    	if (!hit) {
    		/* deferred cleanup and reset to protocols[0] */
    
    Andy Green's avatar
    Andy Green committed
    		lwsl_info("no hit\n");
    
    Andy Green's avatar
    Andy Green committed
    		if (lws_bind_protocol(wsi, &wsi->vhost->protocols[0]))
    			return 1;
    
    Andy Green's avatar
    Andy Green committed
    		n = wsi->protocol->callback(wsi, LWS_CALLBACK_HTTP,
    				    wsi->user_space, uri_ptr, uri_len);
    
    Andy Green's avatar
    Andy Green committed
    		goto after;
    	}
    
    Andy Green's avatar
    Andy Green committed
    	s = uri_ptr + hit->mountpoint_len;
    
    Andy Green's avatar
    Andy Green committed
    	args.p = uri_ptr;
    	args.len = uri_len;
    	args.max_len = hit->auth_mask;
    	args.final = 0; /* used to signal callback dealt with it */
    
    	n = wsi->protocol->callback(wsi, LWS_CALLBACK_CHECK_ACCESS_RIGHTS,
    				    wsi->user_space, &args, 0);
    	if (n) {
    		lws_return_http_status(wsi, HTTP_STATUS_UNAUTHORIZED,
    				       NULL);
    		goto bail_nuke_ah;
    	}
    	if (args.final) /* callback completely handled it well */
    		return 0;
    
    	/*
    	 * if we have a mountpoint like https://xxx.com/yyy
    	 * there is an implied / at the end for our purposes since
    	 * we can only mount on a "directory".
    	 *
    	 * But if we just go with that, the browser cannot understand
    	 * that he is actually looking down one "directory level", so
    	 * even though we give him /yyy/abc.html he acts like the
    	 * current directory level is /.  So relative urls like "x.png"
    	 * wrongly look outside the mountpoint.
    	 *
    	 * Therefore if we didn't come in on a url with an explicit
    	 * / at the end, we must redirect to add it so the browser
    	 * understands he is one "directory level" down.
    	 */
    	if ((hit->mountpoint_len > 1 ||
    	     (hit->origin_protocol == LWSMPRO_REDIR_HTTP ||
    	      hit->origin_protocol == LWSMPRO_REDIR_HTTPS)) &&
    	    (*s != '/' ||
    	     (hit->origin_protocol == LWSMPRO_REDIR_HTTP ||
    	      hit->origin_protocol == LWSMPRO_REDIR_HTTPS)) &&
    	    (hit->origin_protocol != LWSMPRO_CGI &&
    	     hit->origin_protocol != LWSMPRO_CALLBACK //&&
    	     //hit->protocol == NULL
    	     )) {
    		unsigned char *start = pt->serv_buf + LWS_PRE,
    			      *p = start, *end = p + 512;
    
    		lwsl_debug("Doing 301 '%s' org %s\n", s, hit->origin);
    
    		if (!lws_hdr_total_length(wsi, WSI_TOKEN_HOST))
    
    Andy Green's avatar
    Andy Green committed
    			goto bail_nuke_ah;
    
    
    Andy Green's avatar
    Andy Green committed
    		/* > at start indicates deal with by redirect */
    		if (hit->origin_protocol == LWSMPRO_REDIR_HTTP ||
    		    hit->origin_protocol == LWSMPRO_REDIR_HTTPS)
    			n = snprintf((char *)end, 256, "%s%s",
    				    oprot[hit->origin_protocol & 1],
    				    hit->origin);
    		else
    			n = snprintf((char *)end, 256,
    			    "%s%s%s/", oprot[lws_is_ssl(wsi)],
    			    lws_hdr_simple_ptr(wsi, WSI_TOKEN_HOST),
    			    uri_ptr);
    
    Andy Green's avatar
    Andy Green committed
    		n = lws_http_redirect(wsi, HTTP_STATUS_MOVED_PERMANENTLY,
    				      end, n, &p, end);
    		if ((int)n < 0)
    			goto bail_nuke_ah;
    
    Andy Green's avatar
    Andy Green committed
    		return lws_http_transaction_completed(wsi);
    	}
    
    	/*
    	 * A particular protocol callback is mounted here?
    	 *
    	 * For the duration of this http transaction, bind us to the
    	 * associated protocol
    	 */
    	if (hit->origin_protocol == LWSMPRO_CALLBACK || hit->protocol) {
    		const struct lws_protocols *pp;
    		const char *name = hit->origin;
    		if (hit->protocol)
    			name = hit->protocol;
    
    		pp = lws_vhost_name_to_protocol(wsi->vhost, name);
    		if (!pp) {
    			n = -1;
    			lwsl_err("Unable to find plugin '%s'\n",
    				 hit->origin);
    			return 1;
    
    Andy Green's avatar
    Andy Green committed
    		if (lws_bind_protocol(wsi, pp))
    			return 1;
    
    Andy Green's avatar
    Andy Green committed
    		if (hit->cgienv && wsi->protocol->callback(wsi,
    				LWS_CALLBACK_HTTP_PMO,
    				wsi->user_space, (void *)hit->cgienv, 0))
    			return 1;
    
    Andy Green's avatar
    Andy Green committed
    		if (lws_hdr_total_length(wsi, WSI_TOKEN_POST_URI)) {
    			n = wsi->protocol->callback(wsi, LWS_CALLBACK_HTTP,
    					    wsi->user_space,
    					    uri_ptr + hit->mountpoint_len,
    					    uri_len - hit->mountpoint_len);
    
    			goto after;
    		}
    
    Andy Green's avatar
    Andy Green committed
    	}
    
    Andy Green's avatar
    Andy Green committed
    #ifdef LWS_WITH_CGI
    
    Andy Green's avatar
    Andy Green committed
    	/* did we hit something with a cgi:// origin? */
    	if (hit->origin_protocol == LWSMPRO_CGI) {
    		const char *cmd[] = {
    			NULL, /* replace with cgi path */
    			NULL
    		};
    		unsigned char *p, *end, buffer[256];
    
    Andy Green's avatar
    Andy Green committed
    		lwsl_debug("%s: cgi\n", __func__);
    		cmd[0] = hit->origin;
    
    Andy Green's avatar
    Andy Green committed
    		n = 5;
    		if (hit->cgi_timeout)
    			n = hit->cgi_timeout;
    
    		n = lws_cgi(wsi, cmd, hit->mountpoint_len, n,
    			    hit->cgienv);
    		if (n) {
    			lwsl_err("%s: cgi failed\n");
    			return -1;
    
    Andy Green's avatar
    Andy Green committed
    		}
    
    Andy Green's avatar
    Andy Green committed
    		p = buffer + LWS_PRE;
    		end = p + sizeof(buffer) - LWS_PRE;
    
    Andy Green's avatar
    Andy Green committed
    		if (lws_add_http_header_status(wsi, 200, &p, end))
    			return 1;
    		if (lws_add_http_header_by_token(wsi, WSI_TOKEN_CONNECTION,
    				(unsigned char *)"close", 5, &p, end))
    			return 1;
    		n = lws_write(wsi, buffer + LWS_PRE,
    			      p - (buffer + LWS_PRE),
    			      LWS_WRITE_HTTP_HEADERS);
    
    Andy Green's avatar
    Andy Green committed
    		goto deal_body;
    	}
    #endif
    
    Andy Green's avatar
    Andy Green committed
    	n = strlen(s);
    	if (s[0] == '\0' || (n == 1 && s[n - 1] == '/'))
    		s = (char *)hit->def;
    	if (!s)
    		s = "index.html";
    
    Andy Green's avatar
    Andy Green committed
    	wsi->cache_secs = hit->cache_max_age;
    	wsi->cache_reuse = hit->cache_reusable;
    	wsi->cache_revalidate = hit->cache_revalidate;
    	wsi->cache_intermediaries = hit->cache_intermediaries;
    
    Andy Green's avatar
    Andy Green committed
    	n = lws_http_serve(wsi, s, hit->origin, hit);
    	if (n) {
    		/*
    		 * 	lws_return_http_status(wsi, HTTP_STATUS_NOT_FOUND, NULL);
    		 */
    		if (hit->protocol) {
    			const struct lws_protocols *pp = lws_vhost_name_to_protocol(
    					wsi->vhost, hit->protocol);
    
    Andy Green's avatar
    Andy Green committed
    			if (lws_bind_protocol(wsi, pp))
    				return 1;
    
    Andy Green's avatar
    Andy Green committed
    			n = pp->callback(wsi, LWS_CALLBACK_HTTP,
    					 wsi->user_space,
    					 uri_ptr + hit->mountpoint_len,
    					 uri_len - hit->mountpoint_len);
    		} else
    			n = wsi->protocol->callback(wsi, LWS_CALLBACK_HTTP,
    
    Andy Green's avatar
    Andy Green committed
    				    wsi->user_space, uri_ptr, uri_len);
    
    	if (n) {
    		lwsl_info("LWS_CALLBACK_HTTP closing\n");
    
    		return 1;
    	}
    
    Andy Green's avatar
    Andy Green committed
    #ifdef LWS_WITH_CGI
    deal_body:
    #endif
    
    	 * If we're not issuing a file, check for content_length or
    	 * HTTP keep-alive. No keep-alive header allocation for
    
    	 * ISSUING_FILE, as this uses HTTP/1.0.
    	 *
    
    	 * In any case, return 0 and let lws_read decide how to
    
    	 * proceed based on state
    	 */
    
    Andy Green's avatar
    Andy Green committed
    	if (wsi->state != LWSS_HTTP_ISSUING_FILE)
    
    		/* Prepare to read body if we have a content length: */
    		if (wsi->u.http.content_length > 0)
    
    Andy Green's avatar
    Andy Green committed
    			wsi->state = LWSS_HTTP_BODY;
    
    
    	return 0;
    
    bail_nuke_ah:
    
    Andy Green's avatar
    Andy Green committed
    	/* we're closing, losing some rx is OK */
    	wsi->u.hdr.ah->rxpos = wsi->u.hdr.ah->rxlen;
    
    Andy Green's avatar
    Andy Green committed
    	// lwsl_notice("%s: drop1\n", __func__);
    
    Andy Green's avatar
    Andy Green committed
    	lws_header_table_detach(wsi, 1);
    
    Andy Green's avatar
    Andy Green committed
    int
    lws_bind_protocol(struct lws *wsi, const struct lws_protocols *p)
    {
    //	if (wsi->protocol == p)
    //		return 0;
    
    	if (wsi->protocol)
    		wsi->protocol->callback(wsi, LWS_CALLBACK_HTTP_DROP_PROTOCOL,
    					wsi->user_space, NULL, 0);
    	if (!wsi->user_space_externally_allocated)
    		lws_free_set_NULL(wsi->user_space);
    
    	wsi->protocol = p;
    	if (!p)
    		return 0;
    
    	if (lws_ensure_user_space(wsi))
    		return 1;
    
    	if (wsi->protocol->callback(wsi, LWS_CALLBACK_HTTP_BIND_PROTOCOL,
    				    wsi->user_space, NULL, 0))
    		return 1;
    
    	return 0;
    }
    
    
    int
    lws_handshake_server(struct lws *wsi, unsigned char **buf, size_t len)