s_stats workaround

[irc/rqf/shadowircd.git] / src / listener.c

/*\r
 *  ircd-ratbox: A slightly useful ircd.\r
 *  listener.c: Listens on a port.\r
 *\r
 *  Copyright (C) 1990 Jarkko Oikarinen and University of Oulu, Co Center\r
 *  Copyright (C) 1996-2002 Hybrid Development Team\r
 *  Copyright (C) 2002-2005 ircd-ratbox development team\r
 *\r
 *  This program is free software; you can redistribute it and/or modify\r
 *  it under the terms of the GNU General Public License as published by\r
 *  the Free Software Foundation; either version 2 of the License, or\r
 *  (at your option) any later version.\r
 *\r
 *  This program is distributed in the hope that it will be useful,\r
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of\r
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the\r
 *  GNU General Public License for more details.\r
 *\r
 *  You should have received a copy of the GNU General Public License\r
 *  along with this program; if not, write to the Free Software\r
 *  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301\r
 *  USA\r
 *\r
 *  $Id: listener.c 25169 2008-03-29 21:41:31Z jilles $\r
 */\r
\r
#include "stdinc.h"\r
#include "ratbox_lib.h"\r
#include "listener.h"\r
#include "client.h"\r
#include "ircd.h"\r
#include "numeric.h"\r
#include "s_conf.h"\r
#include "s_newconf.h"\r
#include "s_stats.h"\r
#include "send.h"\r
#include "s_auth.h"\r
#include "reject.h"\r
#include "s_log.h"\r
#include "hash.h"\r
#include "sslproc.h"\r
\r
static rb_dlink_list listener_list;\r
static int accept_precallback(rb_fde_t *F, struct sockaddr *addr, rb_socklen_t addrlen, void *data);\r
static void accept_callback(rb_fde_t *F, int status, struct sockaddr *addr, rb_socklen_t addrlen, void *data);\r
\r
\r
\r
static struct Listener *\r
make_listener(struct rb_sockaddr_storage *addr)\r
{\r
	struct Listener *listener = rb_malloc(sizeof(struct Listener));\r
	s_assert(0 != listener);\r
	listener->name = ServerInfo.name; /* me.name may not be valid yet -- jilles */\r
	listener->F = NULL;\r
	memcpy(&listener->addr, addr, sizeof(struct rb_sockaddr_storage));\r
	return listener;\r
}\r
\r
void\r
free_listener(struct Listener *listener)\r
{\r
	s_assert(NULL != listener);\r
	if(listener == NULL)\r
		return;\r
	\r
	rb_dlinkDelete(&listener->node, &listener_list);\r
	rb_free(listener);\r
}\r
\r
#define PORTNAMELEN 6		/* ":31337" */\r
\r
/*\r
 * get_listener_name - return displayable listener name and port\r
 * returns "host.foo.org:6667" for a given listener\r
 */\r
const char *\r
get_listener_name(struct Listener *listener)\r
{\r
	static char buf[HOSTLEN + HOSTLEN + PORTNAMELEN + 4];\r
	int port = 0;\r
\r
	s_assert(NULL != listener);\r
	if(listener == NULL)\r
		return NULL;\r
\r
#ifdef IPV6\r
	if(GET_SS_FAMILY(&listener->addr) == AF_INET6)\r
		port = ntohs(((const struct sockaddr_in6 *)&listener->addr)->sin6_port);\r
	else\r
#endif\r
		port = ntohs(((const struct sockaddr_in *)&listener->addr)->sin_port);	\r
\r
	rb_snprintf(buf, sizeof(buf), "%s[%s/%u]", me.name, listener->name, port);\r
	return buf;\r
}\r
\r
/*\r
 * show_ports - send port listing to a client\r
 * inputs       - pointer to client to show ports to\r
 * output       - none\r
 * side effects - show ports\r
 */\r
void\r
show_ports(struct Client *source_p)\r
{\r
	struct Listener *listener;\r
	rb_dlink_node *ptr;\r
	\r
	RB_DLINK_FOREACH(ptr, listener_list.head)\r
	{\r
		listener = ptr->data;\r
		sendto_one_numeric(source_p, RPL_STATSPLINE, \r
				   form_str(RPL_STATSPLINE), 'P',\r
#ifdef IPV6\r
			   ntohs(GET_SS_FAMILY(&listener->addr) == AF_INET ? ((struct sockaddr_in *)&listener->addr)->sin_port :\r
				 ((struct sockaddr_in6 *)&listener->addr)->sin6_port),\r
#else\r
			   ntohs(((struct sockaddr_in *)&listener->addr)->sin_port),\r
#endif\r
			   IsOperAdmin(source_p) ? listener->name : me.name,\r
			   listener->ref_count, (listener->active) ? "active" : "disabled",\r
			   listener->ssl ? " ssl" : "");\r
	}\r
}\r
\r
/*\r
 * inetport - create a listener socket in the AF_INET or AF_INET6 domain,\r
 * bind it to the port given in 'port' and listen to it\r
 * returns true (1) if successful false (0) on error.\r
 *\r
 * If the operating system has a define for SOMAXCONN, use it, otherwise\r
 * use RATBOX_SOMAXCONN\r
 */\r
#ifdef SOMAXCONN\r
#undef RATBOX_SOMAXCONN\r
#define RATBOX_SOMAXCONN SOMAXCONN\r
#endif\r
\r
static int\r
inetport(struct Listener *listener)\r
{\r
	rb_fde_t *F;\r
	int ret;\r
	int opt = 1;\r
\r
	/*\r
	 * At first, open a new socket\r
	 */\r
	\r
	F = rb_socket(GET_SS_FAMILY(&listener->addr), SOCK_STREAM, 0, "Listener socket");\r
\r
#ifdef IPV6\r
	if(GET_SS_FAMILY(&listener->addr) == AF_INET6)\r
	{\r
		struct sockaddr_in6 *in6 = (struct sockaddr_in6 *)&listener->addr;\r
		if(!IN6_ARE_ADDR_EQUAL(&in6->sin6_addr, &in6addr_any))\r
		{\r
			rb_inet_ntop(AF_INET6, &in6->sin6_addr, listener->vhost, sizeof(listener->vhost));\r
			listener->name = listener->vhost;\r
		}\r
	} else\r
#endif\r
	{\r
		struct sockaddr_in *in = (struct sockaddr_in *)&listener->addr;\r
		if(in->sin_addr.s_addr != INADDR_ANY)\r
		{\r
			rb_inet_ntop(AF_INET, &in->sin_addr, listener->vhost, sizeof(listener->vhost));\r
			listener->name = listener->vhost;\r
		}	\r
	}\r
\r
\r
	if(F == NULL)\r
	{\r
		report_error("opening listener socket %s:%s",\r
			     get_listener_name(listener), \r
			     get_listener_name(listener), errno);\r
		return 0;\r
	}\r
	else if((maxconnections - 10) < rb_get_fd(F)) /* XXX this is kinda bogus*/\r
	{\r
		report_error("no more connections left for listener %s:%s",\r
			     get_listener_name(listener), \r
			     get_listener_name(listener), errno);\r
		rb_close(F);\r
		return 0;\r
	}\r
	/*\r
	 * XXX - we don't want to do all this crap for a listener\r
	 * set_sock_opts(listener);\r
	 */\r
	if(setsockopt(rb_get_fd(F), SOL_SOCKET, SO_REUSEADDR, (char *) &opt, sizeof(opt)))\r
	{\r
		report_error("setting SO_REUSEADDR for listener %s:%s",\r
			     get_listener_name(listener), \r
			     get_listener_name(listener), errno);\r
		rb_close(F);\r
		return 0;\r
	}\r
\r
	/*\r
	 * Bind a port to listen for new connections if port is non-null,\r
	 * else assume it is already open and try get something from it.\r
	 */\r
\r
	if(bind(rb_get_fd(F), (struct sockaddr *) &listener->addr, GET_SS_LEN(&listener->addr)))\r
	{\r
		report_error("binding listener socket %s:%s",\r
			     get_listener_name(listener), \r
			     get_listener_name(listener), errno);\r
		rb_close(F);\r
		return 0;\r
	}\r
\r
	if((ret = rb_listen(F, RATBOX_SOMAXCONN)))\r
	{\r
		report_error("listen failed for %s:%s", \r
			     get_listener_name(listener), \r
			     get_listener_name(listener), errno);\r
		rb_close(F);\r
		return 0;\r
	}\r
\r
	listener->F = F;\r
\r
	rb_accept_tcp(listener->F, accept_precallback, accept_callback, listener);\r
	return 1;\r
}\r
\r
static struct Listener *\r
find_listener(struct rb_sockaddr_storage *addr)\r
{\r
	struct Listener *listener = NULL;\r
	struct Listener *last_closed = NULL;\r
	rb_dlink_node *ptr;\r
\r
	RB_DLINK_FOREACH(ptr, listener_list.head)\r
	{\r
		listener = ptr->data;\r
		if(GET_SS_FAMILY(addr) != GET_SS_FAMILY(&listener->addr))\r
			continue;\r
		\r
		switch(GET_SS_FAMILY(addr))\r
		{\r
			case AF_INET:\r
			{\r
				struct sockaddr_in *in4 = (struct sockaddr_in *)addr;\r
				struct sockaddr_in *lin4 = (struct sockaddr_in *)&listener->addr;\r
				if(in4->sin_addr.s_addr == lin4->sin_addr.s_addr && \r
					in4->sin_port == lin4->sin_port )\r
				{\r
					if(listener->F == NULL)\r
						last_closed = listener;\r
					else\r
						return(listener);\r
				}\r
				break;\r
			}\r
#ifdef IPV6\r
			case AF_INET6:\r
			{\r
				struct sockaddr_in6 *in6 = (struct sockaddr_in6 *)addr;\r
				struct sockaddr_in6 *lin6 =(struct sockaddr_in6 *)&listener->addr;\r
				if(IN6_ARE_ADDR_EQUAL(&in6->sin6_addr, &lin6->sin6_addr) &&\r
				  in6->sin6_port == lin6->sin6_port)\r
				{\r
					if(listener->F == NULL)\r
						last_closed = listener;\r
					else\r
						return(listener);\r
				}\r
				break;\r
				\r
			}\r
#endif\r
\r
			default:\r
				break;\r
		}\r
	}\r
	return last_closed;\r
}\r
\r
/*\r
 * add_listener- create a new listener\r
 * port - the port number to listen on\r
 * vhost_ip - if non-null must contain a valid IP address string in\r
 * the format "255.255.255.255"\r
 */\r
void\r
add_listener(int port, const char *vhost_ip, int family, int ssl)\r
{\r
	struct Listener *listener;\r
	struct rb_sockaddr_storage vaddr;\r
\r
	/*\r
	 * if no port in conf line, don't bother\r
	 */\r
	if(port == 0)\r
		return;\r
		\r
	memset(&vaddr, 0, sizeof(vaddr));\r
	GET_SS_FAMILY(&vaddr) = family;\r
\r
	if(vhost_ip != NULL)\r
	{\r
		if(rb_inet_pton_sock(vhost_ip, (struct sockaddr *)&vaddr) <= 0)\r
			return;\r
	} else\r
	{\r
		switch(family)\r
		{\r
			case AF_INET:\r
				((struct sockaddr_in *)&vaddr)->sin_addr.s_addr = INADDR_ANY;\r
				break;\r
#ifdef IPV6\r
			case AF_INET6:\r
				memcpy(&((struct sockaddr_in6 *)&vaddr)->sin6_addr, &in6addr_any, sizeof(struct in6_addr));\r
				break;\r
#endif\r
			default:\r
				return;\r
		} \r
	}\r
	switch(family)\r
	{\r
		case AF_INET:\r
			SET_SS_LEN(&vaddr, sizeof(struct sockaddr_in));\r
			((struct sockaddr_in *)&vaddr)->sin_port = htons(port);\r
			break;\r
#ifdef IPV6\r
		case AF_INET6:\r
			SET_SS_LEN(&vaddr, sizeof(struct sockaddr_in6));\r
			((struct sockaddr_in6 *)&vaddr)->sin6_port = htons(port);\r
			break;\r
#endif\r
		default:\r
			break;\r
	}\r
	if((listener = find_listener(&vaddr)))\r
	{\r
		if(listener->F != NULL)\r
			return;\r
	}\r
	else\r
	{\r
		listener = make_listener(&vaddr);\r
		rb_dlinkAdd(listener, &listener->node, &listener_list);\r
	}\r
\r
	listener->F = NULL;\r
	listener->ssl = ssl;\r
	if(inetport(listener))\r
		listener->active = 1;\r
	else\r
		close_listener(listener);\r
}\r
\r
/*\r
 * close_listener - close a single listener\r
 */\r
void\r
close_listener(struct Listener *listener)\r
{\r
	s_assert(listener != NULL);\r
	if(listener == NULL)\r
		return;\r
	if(listener->F != NULL)\r
	{\r
		rb_close(listener->F);\r
		listener->F = NULL;\r
	}\r
\r
	listener->active = 0;\r
\r
	if(listener->ref_count)\r
		return;\r
\r
	free_listener(listener);\r
}\r
\r
/*\r
 * close_listeners - close and free all listeners that are not being used\r
 */\r
void\r
close_listeners()\r
{\r
	struct Listener *listener;\r
	rb_dlink_node *ptr, *next;\r
\r
	RB_DLINK_FOREACH_SAFE(ptr, next, listener_list.head)\r
	{\r
		listener = ptr->data;\r
		close_listener(listener);\r
	}\r
}\r
\r
/*\r
 * add_connection - creates a client which has just connected to us on \r
 * the given fd. The sockhost field is initialized with the ip# of the host.\r
 * The client is sent to the auth module for verification, and not put in\r
 * any client list yet.\r
 */\r
static void\r
add_connection(struct Listener *listener, rb_fde_t *F, struct sockaddr *sai, struct sockaddr *lai, void *ssl_ctl)\r
{\r
	struct Client *new_client;\r
	s_assert(NULL != listener);\r
	/* \r
	 * get the client socket name from the socket\r
	 * the client has already been checked out in accept_connection\r
	 */\r
	new_client = make_client(NULL);\r
\r
	memcpy(&new_client->localClient->ip, sai, sizeof(struct rb_sockaddr_storage));\r
	new_client->localClient->lip = rb_malloc(sizeof(struct rb_sockaddr_storage));\r
	memcpy(new_client->localClient->lip, lai, sizeof(struct rb_sockaddr_storage));\r
\r
	/* \r
	 * copy address to 'sockhost' as a string, copy it to host too\r
	 * so we have something valid to put into error messages...\r
	 */\r
	rb_inet_ntop_sock((struct sockaddr *)&new_client->localClient->ip, new_client->sockhost, \r
		sizeof(new_client->sockhost));\r
\r
\r
	rb_strlcpy(new_client->host, new_client->sockhost, sizeof(new_client->host));\r
\r
#ifdef IPV6\r
	if(GET_SS_FAMILY(&new_client->localClient->ip) == AF_INET6 && ConfigFileEntry.dot_in_ip6_addr == 1)\r
	{\r
		rb_strlcat(new_client->host, ".", sizeof(new_client->host));\r
	}\r
#endif\r
\r
	new_client->localClient->F = F;\r
	add_to_cli_fd_hash(new_client);\r
	new_client->localClient->listener = listener;\r
	new_client->localClient->ssl_ctl = ssl_ctl;\r
	if(ssl_ctl != NULL || rb_fd_ssl(F))\r
		SetSSL(new_client);\r
\r
	++listener->ref_count;\r
\r
	start_auth(new_client);\r
}\r
\r
static time_t last_oper_notice = 0;\r
\r
static const char *toofast = "ERROR :Reconnecting too fast, throttled.\r\n";\r
\r
static int\r
accept_precallback(rb_fde_t *F, struct sockaddr *addr, rb_socklen_t addrlen, void *data)\r
{\r
	struct Listener *listener = (struct Listener *)data;\r
	char buf[BUFSIZE];\r
	struct ConfItem *aconf;\r
\r
	if(listener->ssl && (!ssl_ok || !get_ssld_count()))\r
	{\r
		fprintf(stderr, "closed socket\n");\r
		rb_close(F);\r
		return 0;\r
	}\r
	\r
	if((maxconnections - 10) < rb_get_fd(F)) /* XXX this is kinda bogus */\r
	{\r
		++ServerStats.is_ref;\r
		/*\r
		 * slow down the whining to opers bit\r
		 */\r
		if((last_oper_notice + 20) <= rb_current_time())\r
		{\r
			sendto_realops_flags(UMODE_ALL, L_ALL,\r
					     "All connections in use. (%s)",\r
					     get_listener_name(listener));\r
			last_oper_notice = rb_current_time();\r
		}\r
			\r
		rb_write(F, "ERROR :All connections in use\r\n", 32);\r
		rb_close(F);\r
		/* Re-register a new IO request for the next accept .. */\r
		return 0;\r
	}\r
\r
	aconf = find_dline(addr);\r
	if(aconf != NULL && (aconf->status & CONF_EXEMPTDLINE))\r
		return 1;\r
	\r
	/* Do an initial check we aren't connecting too fast or with too many\r
	 * from this IP... */\r
	if(aconf != NULL)\r
	{\r
		ServerStats.is_ref++;\r
			\r
		if(ConfigFileEntry.dline_with_reason)\r
		{\r
			if (rb_snprintf(buf, sizeof(buf), "ERROR :*** Banned: %s\r\n", aconf->passwd) >= (int)(sizeof(buf)-1))\r
			{\r
				buf[sizeof(buf) - 3] = '\r';\r
				buf[sizeof(buf) - 2] = '\n';\r
				buf[sizeof(buf) - 1] = '\0';\r
			}\r
		}\r
		else\r
			strcpy(buf, "ERROR :You have been D-lined.\r\n");\r
	\r
		rb_write(F, buf, strlen(buf));\r
		rb_close(F);\r
		return 0;\r
	}\r
\r
	if(check_reject(F, addr))\r
		return 0;\r
		\r
	if(throttle_add(addr))\r
	{\r
		rb_write(F, toofast, strlen(toofast));\r
		rb_close(F);\r
		return 0;\r
	}\r
\r
	return 1;\r
}\r
\r
static void\r
accept_ssld(rb_fde_t *F, struct sockaddr *addr, struct sockaddr *laddr, struct Listener *listener)\r
{\r
	ssl_ctl_t *ctl;\r
	rb_fde_t *xF[2];\r
	rb_socketpair(AF_UNIX, SOCK_STREAM, 0, &xF[0], &xF[1], "Incoming ssld Connection");\r
	ctl = start_ssld_accept(F, xF[1], rb_get_fd(xF[0])); /* this will close F for us */\r
	add_connection(listener, xF[0], addr, laddr, ctl);\r
}\r
\r
static void\r
accept_callback(rb_fde_t *F, int status, struct sockaddr *addr, rb_socklen_t addrlen, void *data)\r
{\r
	struct Listener *listener = data;\r
	struct rb_sockaddr_storage lip;\r
	unsigned int locallen = sizeof(struct rb_sockaddr_storage);\r
	\r
	ServerStats.is_ac++;\r
	if(getsockname(rb_get_fd(F), (struct sockaddr *) &lip, &locallen) < 0)\r
	{\r
		/* this shouldn't fail so... */\r
		/* XXX add logging of this */\r
		rb_close(F);\r
	}\r
	if(listener->ssl)\r
		accept_ssld(F, addr, (struct sockaddr *)&lip, listener);\r
	else\r
		add_connection(listener, F, addr, (struct sockaddr *)&lip, NULL);\r
}\r