Skip to content
Snippets Groups Projects
context.c 26.1 KiB
Newer Older
  • Learn to ignore specific revisions
  • /*
     * libwebsockets - small server side websockets and web server implementation
     *
    
    Andy Green's avatar
    Andy Green committed
     * Copyright (C) 2010-2015 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"
    
    #ifndef LWS_BUILD_HASH
    #define LWS_BUILD_HASH "unknown-build-hash"
    #endif
    
    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;
    }
    
    
    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 *
    lws_protocol_vh_priv_zalloc(struct lws_vhost *vhost, const struct lws_protocols *prot,
    			    int size)
    {
    	int n = 0;
    
    	/* allocate the vh priv array only on demand */
    	if (!vhost->protocol_vh_privs) {
    		vhost->protocol_vh_privs = (void **)lws_zalloc(
    				vhost->count_protocols * sizeof(void *));
    		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
    
    	vhost->protocol_vh_privs[n] = lws_zalloc(size);
    	return vhost->protocol_vh_privs[n];
    }
    
    LWS_VISIBLE void *
    lws_protocol_vh_priv_get(struct lws_vhost *vhost, const struct lws_protocols *prot)
    {
    	int n = 0;
    
    	if (!vhost->protocol_vh_privs)
    		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) {
    		// lwsl_notice("%s: '%s' '%s'\n", __func__, pvo->name, name);
    		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
    int
    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;
    
    	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)
    			goto next;
    
    
    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];
    
    
    			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_notice("    vh %s prot %s opt %s\n",
    							vh->name,
    							vh->protocols[n].name,
    							pvo->name);
    
    					if (!strcmp(pvo->name, "default")) {
    						lwsl_notice("Setting default "
    						   "protocol for vh %s to %s\n",
    						   vh->name,
    						   vh->protocols[n].name);
    						vh->default_protocol_index = n;
    					}
    					pvo = pvo->next;
    				}
    
    				pvo = pvo1->options;
    			}
    
    Andy Green's avatar
    Andy Green committed
    			/*
    			 * inform all the protocols that they are doing their one-time
    			 * initialization if they want to.
    			 *
    			 * NOTE the wsi is all zeros except for the context, vh and
    			 * protocol ptrs so lws_get_context(wsi) etc can work
    			 */
    
    Andy Green's avatar
    Andy Green committed
    			if (vh->protocols[n].callback(&wsi,
    
    				LWS_CALLBACK_PROTOCOL_INIT, NULL,
    
    Andy Green's avatar
    Andy Green committed
    				(void *)pvo, 0))
    				return 1;
    
    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;
    
    	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)
    
    #ifdef LWS_WITH_CGI
    	struct lws_cgi_args *args;
    	char buf[128];
    	int n;
    #endif
    
    	switch (reason) {
    	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;
    
    	case LWS_CALLBACK_HTTP_WRITEABLE:
    #ifdef LWS_WITH_CGI
    		if (wsi->reason_bf & 1) {
    			if (lws_cgi_write_split_stdout_headers(wsi) < 0)
    				return -1;
    
    			wsi->reason_bf &= ~1;
    			break;
    		}
    #endif
    
    		break;
    
    #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:
    			wsi->reason_bf |= 1;
    			/* when writing to MASTER would not block */
    			lws_callback_on_writable(wsi);
    			break;
    		case LWS_STDERR:
    			n = read(lws_get_socket_fd(args->stdwsi[LWS_STDERR]),
    
    			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:
    		return -1;
    
    	case LWS_CALLBACK_CGI_STDIN_DATA:  /* POST body for stdin */
    		args = (struct lws_cgi_args *)in;
    		args->data[args->len] = '\0';
    		n = write(lws_get_socket_fd(args->stdwsi[LWS_STDIN]),
    			  args->data, args->len);
    		if (n < args->len)
    			lwsl_notice("LWS_CALLBACK_CGI_STDIN_DATA: "
    				    "sent %d only %d went", n, args->len);
    		return n;
    #endif
    	default:
    		break;
    	}
    
    
    	return 0;
    }
    
    /* list of supported protocols and callbacks */
    
    static const struct lws_protocols protocols_dummy[] = {
    	/* first protocol must always be HTTP handler */
    
    	{
    		"http-only",		/* name */
    
    Andy Green's avatar
    Andy Green committed
    		lws_callback_http_dummy,		/* callback */
    
    		0,			/* max frame size / rx buffer */
    
    	},
    	/*
    	 * the other protocols are provided by lws plugins
    	 */
    	{ NULL, NULL, 0, 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,
    
    		 struct lws_context_creation_info *info)
    
    Andy Green's avatar
    Andy Green committed
    {
    	struct lws_vhost *vh = lws_zalloc(sizeof(*vh)),
    			 **vh1 = &context->vhost_list;
    
    	const struct lws_http_mount *mounts;
    
    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;
    	struct lws_protocols *lwsp;
    
    Andy Green's avatar
    Andy Green committed
    	int m, f = !info->pvo;
    
    Andy Green's avatar
    Andy Green committed
    #endif
    
    Andy Green's avatar
    Andy Green committed
    #ifdef 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;
    
    
    	if (!info->protocols)
    		info->protocols = &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;
    
    	vh->iface = info->iface;
    	for (vh->count_protocols = 0;
    	     info->protocols[vh->count_protocols].callback;
    	     vh->count_protocols++)
    		;
    
    	vh->pvo = info->pvo;
    
    	vh->headers = info->headers;
    
    Andy Green's avatar
    Andy Green committed
    	if (info->keepalive_timeout)
    		vh->keepalive_timeout = info->keepalive_timeout;
    	else
    		vh->keepalive_timeout = 5;
    
    Andy Green's avatar
    Andy Green committed
    #ifdef LWS_WITH_PLUGINS
    	if (plugin) {
    		/*
    		 * give the vhost a unified list of protocols including the
    		 * ones that came from plugins
    		 */
    		lwsp = lws_zalloc(sizeof(struct lws_protocols) *
    					   (vh->count_protocols +
    					   context->plugin_protocol_count + 1));
    		if (!lwsp)
    			return NULL;
    
    		m = vh->count_protocols;
    		memcpy(lwsp, info->protocols,
    		       sizeof(struct lws_protocols) * m);
    
    		/* 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 (info->options & LWS_SERVER_OPTION_EXPLICIT_VHOSTS)
    			f = 0;
    
    
    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;
    		}
    		vh->protocols = lwsp;
    	} else
    #endif
    		vh->protocols = info->protocols;
    
    
    	vh->same_vh_protocol_list = (struct lws **)
    			lws_zalloc(sizeof(struct lws *) * vh->count_protocols);
    
    
    	vh->mount_list = info->mounts;
    
    Andy Green's avatar
    Andy Green committed
    
    
    Yeonjun Lim's avatar
    Yeonjun Lim committed
    #ifdef LWS_USE_UNIX_SOCK
    	if (LWS_UNIX_SOCK_ENABLED(context)) {
    		lwsl_notice("Creating Vhost '%s' path \"%s\", %d protocols\n",
    				vh->name, info->iface, vh->count_protocols);
    	} else
    #endif
    
    Denis Osvald's avatar
    Denis Osvald committed
    	lwsl_notice("Creating Vhost '%s' port %d, %d protocols, IPv6 %s\n",
    			vh->name, info->port, vh->count_protocols, LWS_IPV6_ENABLED(vh) ? "on" : "off");
    
    Andy Green's avatar
    Andy Green committed
    
    
    	mounts = info->mounts;
    
    Andy Green's avatar
    Andy Green committed
    	while (mounts) {
    		lwsl_notice("   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) {
    			for (n = 0; n < vh->count_protocols; n++)
    				if (!strcmp(pvo->value, vh->protocols[n].name)) {
    					((struct lws_protocol_vhost_options *)pvo)->value =
    							(const char *)(long)n;
    					break;
    				}
    			if (n == vh->count_protocols)
    				lwsl_err("ignoring unknown interpret protocol %s\n", pvo->value);
    			pvo = pvo->next;
    		}
    
    
    Andy Green's avatar
    Andy Green committed
    		mounts = mounts->mount_next;
    	}
    
    #ifndef LWS_NO_EXTENSIONS
    
    Andy Green's avatar
    Andy Green committed
    #ifdef LWS_WITH_PLUGINS
    	if (context->plugin_extension_count) {
    
    		m = 0;
    		while (info->extensions && info->extensions[m].callback)
    			m++;
    
    		/*
    		 * give the vhost a unified list of extensions including the
    		 * ones that came from plugins
    		 */
    		vh->extensions = lws_zalloc(sizeof(struct lws_extension) *
    					   (m +
    					   context->plugin_extension_count + 1));
    		if (!vh->extensions)
    			return NULL;
    
    		memcpy((struct lws_extension *)vh->extensions, info->extensions,
    		       sizeof(struct lws_extension) * m);
    
    Andy Green's avatar
    Andy Green committed
    		plugin = context->plugin_list;
    
    Andy Green's avatar
    Andy Green committed
    		while (plugin) {
    
    Andy Green's avatar
    Andy Green committed
    			memcpy((struct lws_extension *)&vh->extensions[m],
    				plugin->caps.extensions,
    
    Andy Green's avatar
    Andy Green committed
    			       sizeof(struct lws_extension) *
    			       plugin->caps.count_extensions);
    			m += plugin->caps.count_extensions;
    			plugin = plugin->list;
    		}
    	} else
    #endif
    		vh->extensions = info->extensions;
    
    Andy Green's avatar
    Andy Green committed
    #endif
    
    	vh->listen_port = info->port;
    
    Andy Green's avatar
    Andy Green committed
    #if !defined(LWS_WITH_ESP8266)
    
    Andy Green's avatar
    Andy Green committed
    	vh->http_proxy_port = 0;
    	vh->http_proxy_address[0] = '\0';
    
    	/* either use proxy from info, or try get it from env var */
    
    	if (info->http_proxy_address) {
    		/* override for backwards compatibility */
    		if (info->http_proxy_port)
    			vh->http_proxy_port = info->http_proxy_port;
    		lws_set_proxy(vh, info->http_proxy_address);
    	} else {
    #ifdef LWS_HAVE_GETENV
    		p = getenv("http_proxy");
    		if (p)
    			lws_set_proxy(vh, p);
    #endif
    	}
    
    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 = open(info->log_filepath, 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;
    
    Andy Green's avatar
    Andy Green committed
    	if (lws_context_init_server_ssl(info, vh))
    		goto bail;
    	if (lws_context_init_client_ssl(info, vh))
    		goto bail;
    	if (lws_context_init_server(info, vh))
    		goto bail;
    
    	while (1) {
    		if (!(*vh1)) {
    			*vh1 = vh;
    			break;
    		}
    		vh1 = &(*vh1)->vhost_next;
    	};
    
    	/* for the case we are adding a vhost much later, after server init */
    
    	if (context->protocol_init_done)
    		lws_protocol_init(context);
    
    
    Andy Green's avatar
    Andy Green committed
    	return vh;
    
    bail:
    	lws_free(vh);
    
    	return NULL;
    }
    
    
    LWS_VISIBLE int
    lws_init_vhost_client_ssl(const struct lws_context_creation_info *info,
    			  struct lws_vhost *vhost)
    {
    	struct lws_context_creation_info i;
    
    	memcpy(&i, info, sizeof(i));
    	i.port = CONTEXT_PORT_NO_LISTEN;
    
    	return lws_context_init_client_ssl(&i, vhost);
    }
    
    
    LWS_VISIBLE struct lws_context *
    
    lws_create_context(struct lws_context_creation_info *info)
    
    	struct lws_context *context = NULL;
    
    #ifndef LWS_NO_DAEMONIZE
    
    	int pid_daemon = get_daemonize_pid();
    
    Andy Green's avatar
    Andy Green committed
    #endif
    
    Andy Green's avatar
    Andy Green committed
    	int n, m;
    
    Galen Ma's avatar
    Galen Ma committed
    #if defined(__ANDROID__)
    	struct rlimit rt;
    #endif
    
    
    	lwsl_notice("Initial logging level %d\n", log_level);
    
    Andy Green's avatar
    Andy Green committed
    	lwsl_notice("Libwebsockets version: %s\n", library_version);
    
    #if defined(GCC_VER)
    	lwsl_notice("Compiled with  %s\n", GCC_VER);
    #endif
    
    Andy Green's avatar
    Andy Green committed
    #if LWS_POSIX
    
    #ifdef LWS_USE_IPV6
    
    	if (!lws_check_opt(info->options, LWS_SERVER_OPTION_DISABLE_IPV6))
    
    		lwsl_notice("IPV6 compiled in and enabled\n");
    	else
    		lwsl_notice("IPV6 compiled in but disabled\n");
    #else
    	lwsl_notice("IPV6 not compiled in\n");
    #endif
    
    #ifndef LWS_PLAT_OPTEE
    
    	lws_feature_status_libev(info);
    
    Andy Green's avatar
    Andy Green committed
    	lws_feature_status_libuv(info);
    
    Andy Green's avatar
    Andy Green committed
    	lwsl_info(" LWS_DEF_HEADER_LEN    : %u\n", LWS_DEF_HEADER_LEN);
    
    Andy Green's avatar
    Andy Green committed
    	lwsl_info(" LWS_MAX_PROTOCOLS     : %u\n", LWS_MAX_PROTOCOLS);
    	lwsl_info(" LWS_MAX_SMP           : %u\n", LWS_MAX_SMP);
    	lwsl_info(" SPEC_LATEST_SUPPORTED : %u\n", SPEC_LATEST_SUPPORTED);
    
    	lwsl_info(" sizeof (*info)        : %ld\n", (long)sizeof(*info));
    
    	lwsl_info(" SYSTEM_RANDOM_FILEPATH: '%s'\n", SYSTEM_RANDOM_FILEPATH);
    
    	if (lws_plat_context_early_init())
    		return NULL;
    
    
    	context = lws_zalloc(sizeof(struct lws_context));
    
    	if (!context) {
    		lwsl_err("No memory for websocket context\n");
    		return NULL;
    	}
    
    	if (info->pt_serv_buf_size)
    		context->pt_serv_buf_size = info->pt_serv_buf_size;
    	else
    		context->pt_serv_buf_size = 4096;
    
    
    	context->reject_service_keywords = info->reject_service_keywords;
    
    	if (info->external_baggage_free_on_destroy)
    		context->external_baggage_free_on_destroy =
    			info->external_baggage_free_on_destroy;
    
    Andy Green's avatar
    Andy Green committed
    	context->time_up = time(NULL);
    
    Andy Green's avatar
    Andy Green committed
    
    
    #ifndef LWS_NO_DAEMONIZE
    
    	if (pid_daemon) {
    		context->started_with_parent = pid_daemon;
    		lwsl_notice(" Started with daemon pid %d\n", pid_daemon);
    	}
    
    Andy Green's avatar
    Andy Green committed
    #endif
    
    Galen Ma's avatar
    Galen Ma committed
    #if defined(__ANDROID__)
    		n = getrlimit ( RLIMIT_NOFILE,&rt);
    		if (-1 == n) {
    			lwsl_err("Get RLIMIT_NOFILE failed!\n");
    			return NULL;
    		}
    		context->max_fds = rt.rlim_cur;
    #else
    		context->max_fds = getdtablesize();
    #endif
    
    Andy Green's avatar
    Andy Green committed
    
    	if (info->count_threads)
    		context->count_threads = info->count_threads;
    	else
    		context->count_threads = 1;
    
    	if (context->count_threads > LWS_MAX_SMP)
    		context->count_threads = LWS_MAX_SMP;
    
    
    	context->token_limits = info->token_limits;
    
    Andy Green's avatar
    Andy Green committed
    
    
    	context->options = info->options;
    
    Andy Green's avatar
    Andy Green committed
    
    
    Andy Green's avatar
    Andy Green committed
    	if (info->timeout_secs)
    		context->timeout_secs = info->timeout_secs;
    	else
    		context->timeout_secs = AWAITING_TIMEOUT;
    
    
    	context->ws_ping_pong_interval = info->ws_ping_pong_interval;
    
    
    Andy Green's avatar
    Andy Green committed
    	lwsl_info(" default timeout (secs): %u\n", context->timeout_secs);
    
    Andy Green's avatar
    Andy Green committed
    	if (info->max_http_header_data)
    		context->max_http_header_data = info->max_http_header_data;
    	else
    
    		if (info->max_http_header_data2)
    			context->max_http_header_data =
    					info->max_http_header_data2;
    		else
    			context->max_http_header_data = LWS_DEF_HEADER_LEN;
    
    Andy Green's avatar
    Andy Green committed
    	if (info->max_http_header_pool)
    		context->max_http_header_pool = info->max_http_header_pool;
    	else
    		context->max_http_header_pool = LWS_DEF_HEADER_POOL;
    
    	/*
    	 * Allocate the per-thread storage for scratchpad buffers,
    	 * and header data pool
    
    Andy Green's avatar
    Andy Green committed
    	 */
    	for (n = 0; n < context->count_threads; n++) {
    
    		context->pt[n].serv_buf = lws_zalloc(context->pt_serv_buf_size);
    
    Andy Green's avatar
    Andy Green committed
    		if (!context->pt[n].serv_buf) {
    			lwsl_err("OOM\n");
    			return NULL;
    		}
    
    		context->pt[n].context = context;
    
    		context->pt[n].tid = n;
    
    Andy Green's avatar
    Andy Green committed
    		context->pt[n].http_header_data = lws_malloc(context->max_http_header_data *
    						       context->max_http_header_pool);
    		if (!context->pt[n].http_header_data)
    			goto bail;
    
    		context->pt[n].ah_pool = lws_zalloc(sizeof(struct allocated_headers) *
    					      context->max_http_header_pool);
    		for (m = 0; m < context->max_http_header_pool; m++)
    			context->pt[n].ah_pool[m].data =
    				(char *)context->pt[n].http_header_data +
    				(m * context->max_http_header_data);
    		if (!context->pt[n].ah_pool)
    			goto bail;
    
    		lws_pt_mutex_init(&context->pt[n]);
    
    Andy Green's avatar
    Andy Green committed
    	}
    
    	if (info->fd_limit_per_thread)
    		context->fd_limit_per_thread = info->fd_limit_per_thread;
    	else
    		context->fd_limit_per_thread = context->max_fds /
    					       context->count_threads;
    
    	lwsl_notice(" Threads: %d each %d fds\n", context->count_threads,
    		    context->fd_limit_per_thread);
    
    
    	if (!info->ka_interval && info->ka_time > 0) {
    		lwsl_err("info->ka_interval can't be 0 if ka_time used\n");
    		return NULL;
    	}
    
    #ifdef LWS_USE_LIBEV
    	/* (Issue #264) In order to *avoid breaking backwards compatibility*, we
    	 * enable libev mediated SIGINT handling with a default handler of
    
    	 * lws_sigint_cb. The handler can be overridden or disabled
    	 * by invoking lws_sigint_cfg after creating the context, but
    	 * before invoking lws_initloop:
    
    	 */
    	context->use_ev_sigint = 1;
    
    Andy Green's avatar
    Andy Green committed
    	context->lws_ev_sigint_cb = &lws_ev_sigint_cb;
    
    #endif /* LWS_USE_LIBEV */
    
    Andy Green's avatar
    Andy Green committed
    #ifdef LWS_USE_LIBUV
    	/* (Issue #264) In order to *avoid breaking backwards compatibility*, we
    	 * enable libev mediated SIGINT handling with a default handler of
    	 * lws_sigint_cb. The handler can be overridden or disabled
    	 * by invoking lws_sigint_cfg after creating the context, but
    	 * before invoking lws_initloop:
    	 */
    	context->use_ev_sigint = 1;
    	context->lws_uv_sigint_cb = &lws_uv_sigint_cb;
    #endif
    
    	lwsl_info(" mem: context:         %5lu bytes (%ld ctx + (%ld thr x %d))\n",
    		  (long)sizeof(struct lws_context) +
    
    		  (context->count_threads * context->pt_serv_buf_size),
    
    		  (long)sizeof(struct lws_context),
    		  (long)context->count_threads,
    
    	lwsl_info(" mem: http hdr rsvd:   %5lu bytes (%u thr x (%u + %lu) x %u))\n",
    		    (long)(context->max_http_header_data +
    
    Andy Green's avatar
    Andy Green committed
    		     sizeof(struct allocated_headers)) *
    
    Andy Green's avatar
    Andy Green committed
    		    context->max_http_header_pool * context->count_threads,
    		    context->count_threads,
    
    Andy Green's avatar
    Andy Green committed
    		    context->max_http_header_data,
    
    		    (long)sizeof(struct allocated_headers),
    
    		    context->max_http_header_pool);
    
    Andy Green's avatar
    Andy Green committed
    	n = sizeof(struct lws_pollfd) * context->count_threads *
    	    context->fd_limit_per_thread;
    	context->pt[0].fds = lws_zalloc(n);
    	if (context->pt[0].fds == NULL) {
    
    		lwsl_err("OOM allocating %d fds\n", context->max_fds);
    
    		goto bail;
    
    Andy Green's avatar
    Andy Green committed
    	lwsl_info(" mem: pollfd map:      %5u\n", n);
    
    	if (info->server_string) {
    		context->server_string = info->server_string;
    
    		context->server_string_len = (short)
    				strlen(context->server_string);
    
    	} else {
    		context->server_string = "libwebsockets";
    		context->server_string_len = 13;
    	}
    
    
    Andy Green's avatar
    Andy Green committed
    #if LWS_MAX_SMP > 1
    
    Andy Green's avatar
    Andy Green committed
    	/* each thread serves his own chunk of fds */
    	for (n = 1; n < (int)info->count_threads; n++)
    
    Andy Green's avatar
    Andy Green committed
    		context->pt[n].fds = context->pt[n - 1].fds +
    				     context->fd_limit_per_thread;
    #endif
    
    	if (lws_plat_init(context, info))
    
    		goto bail;
    
    Andy Green's avatar
    Andy Green committed
    	lws_context_init_ssl_library(info);
    
    
    Andy Green's avatar
    Andy Green committed
    	context->user_space = info->user;
    
    
    Andy Green's avatar
    Andy Green committed
    	/*
    	 * if he's not saying he'll make his own vhosts later then act
    	 * compatibly and make a default vhost using the data in the info
    	 */
    	if (!lws_check_opt(info->options, LWS_SERVER_OPTION_EXPLICIT_VHOSTS))
    
    		if (!lws_create_vhost(context, info)) {
    
    Andy Green's avatar
    Andy Green committed
    			lwsl_err("Failed to create default vhost\n");
    			return NULL;
    		}
    
    
    	lws_context_init_extensions(info, context);
    
    
    	lwsl_notice(" mem: per-conn:        %5lu bytes + protocol rx buf\n",
    		    (unsigned long)sizeof(struct lws));
    
    	strcpy(context->canonical_hostname, "unknown");
    
    	lws_server_get_canonical_hostname(context, info);
    
    Andy Green's avatar
    Andy Green committed
    	context->uid = info->uid;
    	context->gid = info->gid;
    
    
    	/*
    	 * drop any root privs for this process
    	 * to listen on port < 1023 we would have needed root, but now we are
    	 * listening, we don't want the power for anything else
    	 */
    
    Andy Green's avatar
    Andy Green committed
    	if (!lws_check_opt(info->options, LWS_SERVER_OPTION_EXPLICIT_VHOSTS))
    		lws_plat_drop_app_privileges(info);
    
    
    	/*
    	 * give all extensions a chance to create any per-context
    	 * allocations they need
    	 */
    	if (info->port != CONTEXT_PORT_NO_LISTEN) {
    
    Andy Green's avatar
    Andy Green committed
    		if (lws_ext_cb_all_exts(context, NULL,
    
    			LWS_EXT_CB_SERVER_CONTEXT_CONSTRUCT, NULL, 0) < 0)
    
    Andy Green's avatar
    Andy Green committed
    		if (lws_ext_cb_all_exts(context, NULL,
    
    			LWS_EXT_CB_CLIENT_CONTEXT_CONSTRUCT, NULL, 0) < 0)
    
    	lws_context_destroy(context);
    
    Andy Green's avatar
    Andy Green committed
    LWS_VISIBLE LWS_EXTERN void
    lws_context_deprecate(struct lws_context *context, lws_reload_func cb)
    {
    	struct lws_vhost *vh = context->vhost_list, *vh1;
    	struct lws *wsi;
    
    	/*
    	 * "deprecation" means disable the context from accepting any new
    	 * connections and free up listen sockets to be used by a replacement
    	 * context.
    	 *
    	 * Otherwise the deprecated context remains operational, until its
    	 * number of connected sockets falls to zero, when it is deleted.
    	 */
    
    	/* for each vhost, close his listen socket */
    
    	while (vh) {
    		wsi = vh->lserv_wsi;
    		if (wsi) {
    			wsi->socket_is_permanently_unusable = 1;
    			lws_close_free_wsi(wsi, LWS_CLOSE_STATUS_NOSTATUS);
    			wsi->context->deprecation_pending_listen_close_count++;
    			/*
    			 * other vhosts can share the listen port, they
    			 * point to the same wsi.  So zap those too.
    			 */
    			vh1 = context->vhost_list;
    			while (vh1) {
    				if (vh1->lserv_wsi == wsi)
    					vh1->lserv_wsi = NULL;
    				vh1 = vh1->vhost_next;
    			}
    		}
    		vh = vh->vhost_next;
    	}
    
    	context->deprecated = 1;
    	context->deprecation_cb = cb;
    }
    
    LWS_VISIBLE LWS_EXTERN int
    lws_context_is_deprecated(struct lws_context *context)
    {
    	return context->deprecated;
    }
    
    LWS_VISIBLE void
    lws_context_destroy2(struct lws_context *context);
    
    
    lws_context_destroy(struct lws_context *context)
    
    	const struct lws_protocols *protocol = NULL;
    
    Andy Green's avatar
    Andy Green committed
    	struct lws_context_per_thread *pt;
    
    Andy Green's avatar
    Andy Green committed
    	struct lws_vhost *vh = NULL;
    
    Andy Green's avatar
    Andy Green committed
    	if (!context) {
    		lwsl_notice("%s: ctx %p\n", __func__, context);
    		return;
    	}
    	if (context->being_destroyed1) {
    		lwsl_notice("%s: ctx %p: already being destroyed\n", __func__, context);
    
    Andy Green's avatar
    Andy Green committed
    	}
    
    	lwsl_notice("%s: ctx %p\n", __func__, context);
    
    	context->being_destroyed = 1;
    
    Andy Green's avatar
    Andy Green committed
    	context->being_destroyed1 = 1;
    
    	memset(&wsi, 0, sizeof(wsi));
    	wsi.context = context;
    
    
    #ifdef LWS_LATENCY
    	if (context->worst_latency_info[0])
    		lwsl_notice("Worst latency: %s\n", context->worst_latency_info);
    #endif
    
    
    Andy Green's avatar
    Andy Green committed
    	while (m--) {
    		pt = &context->pt[m];
    
    
    Andy Green's avatar
    Andy Green committed
    		for (n = 0; (unsigned int)n < context->pt[m].fds_count; n++) {
    
    Andy Green's avatar
    Andy Green committed
    			struct lws *wsi = wsi_from_fd(context, pt->fds[n].fd);
    
    Andy Green's avatar
    Andy Green committed
    			if (!wsi)
    				continue;
    
    Andy Green's avatar
    Andy Green committed
    
    			lws_close_free_wsi(wsi,
    				LWS_CLOSE_STATUS_NOSTATUS_CONTEXT_DESTROY
    
    				/* no protocol close */);
    
    Andy Green's avatar
    Andy Green committed
    			n--;
    		}
    
    		lws_pt_mutex_destroy(pt);
    
    Andy Green's avatar
    Andy Green committed
    	}
    
    Andy Green's avatar
    Andy Green committed
    
    
    	/*
    	 * give all extensions a chance to clean up any per-context
    	 * allocations they might have made
    	 */
    
    Andy Green's avatar
    Andy Green committed
    	n = lws_ext_cb_all_exts(context, NULL,
    
    Andy Green's avatar
    Andy Green committed
    				LWS_EXT_CB_SERVER_CONTEXT_DESTRUCT, NULL, 0);
    
    Andy Green's avatar
    Andy Green committed
    	n = lws_ext_cb_all_exts(context, NULL,
    
    Andy Green's avatar
    Andy Green committed
    				LWS_EXT_CB_CLIENT_CONTEXT_DESTRUCT, NULL, 0);
    
    
    	/*
    	 * inform all the protocols that they are done and will have no more
    
    Andy Green's avatar
    Andy Green committed
    	 * callbacks.
    	 *
    	 * We can't free things until after the event loop shuts down.
    
    	if (context->protocol_init_done)
    		vh = context->vhost_list;
    
    Andy Green's avatar
    Andy Green committed
    	while (vh) {
    
    Andy Green's avatar
    Andy Green committed
    		wsi.vhost = vh;
    
    Andy Green's avatar
    Andy Green committed
    		protocol = vh->protocols;
    
    Andy Green's avatar
    Andy Green committed
    		if (protocol) {
    			n = 0;
    			while (n < vh->count_protocols) {
    				wsi.protocol = protocol;
    
    Andy Green's avatar
    Andy Green committed
    				protocol->callback(&wsi, LWS_CALLBACK_PROTOCOL_DESTROY,
    						   NULL, NULL, 0);
    				protocol++;
    
    Andy Green's avatar
    Andy Green committed
    				n++;
    
    Andy Green's avatar
    Andy Green committed
    			}
    
    Andy Green's avatar
    Andy Green committed
    		}
    
    
    Andy Green's avatar
    Andy Green committed
    		vh = vh->vhost_next;
    	}
    
    Andy Green's avatar
    Andy Green committed
    	for (n = 0; n < context->count_threads; n++) {
    
    Andy Green's avatar
    Andy Green committed
    		pt = &context->pt[n];
    
    		lws_libev_destroyloop(context, n);
    		lws_libuv_destroyloop(context, n);
    
    
    Andy Green's avatar
    Andy Green committed
    		lws_free_set_NULL(context->pt[n].serv_buf);
    
    Andy Green's avatar
    Andy Green committed
    		if (pt->ah_pool)
    			lws_free(pt->ah_pool);
    		if (pt->http_header_data)
    			lws_free(pt->http_header_data);
    
    Andy Green's avatar
    Andy Green committed
    	}
    
    	lws_plat_context_early_destroy(context);
    
    Andy Green's avatar
    Andy Green committed
    
    
    Andy Green's avatar
    Andy Green committed
    	if (context->pt[0].fds)
    
    Andy Green's avatar
    Andy Green committed
    		lws_free_set_NULL(context->pt[0].fds);
    
    Andy Green's avatar
    Andy Green committed
    	if (!LWS_LIBUV_ENABLED(context))
    		lws_context_destroy2(context);
    }
    
    /*
     * call the second one after the event loop has been shut down cleanly
     */
    
    LWS_VISIBLE void
    lws_context_destroy2(struct lws_context *context)
    {
    	const struct lws_protocols *protocol = NULL;
    	struct lws_vhost *vh = NULL, *vh1;
    	int n;
    
    	lwsl_notice("%s: ctx %p\n", __func__, context);