Imported Upstream version 1.0.0upstream/1.0.0

1 files changed, 225 insertions, 0 deletions

diff --git a/src/web_server.c b/src/web_server.c
new file mode 100755
index 00000000..cae94aca
--- /dev/null
+++ b/src/web_server.c
@@ -0,0 +1,225 @@
+#ifdef HAVE_CONFIG_H
+#include <config.h>
+#endif
+#include <unistd.h>
+#include <stdlib.h>
+#include <sys/types.h>
+#include <sys/socket.h>
+#include <netinet/in.h>
+#include <arpa/inet.h>
+#include <errno.h>
+#include <pthread.h>
+#include <sys/stat.h>
+#include <fcntl.h>
+#include <netinet/tcp.h>
+#include <malloc.h>
+
+#include "common.h"
+#include "log.h"
+#include "appconfig.h"
+#include "url.h"
+#include "web_buffer.h"
+#include "web_client.h"
+#include "web_server.h"
+#include "global_statistics.h"
+#include "rrd.h"
+#include "rrd2json.h"
+
+int listen_backlog = LISTEN_BACKLOG;
+
+int listen_fd = -1;
+int listen_port = LISTEN_PORT;
+
+static void log_allocations(void)
+{
+	static int mem = 0;
+
+	struct mallinfo mi;
+
+	mi = mallinfo();
+	if(mi.uordblks > mem) {
+		int clients = 0;
+		struct web_client *w;
+		for(w = web_clients; w ; w = w->next) clients++;
+
+		info("Allocated memory increased from %d to %d (increased by %d bytes). There are %d web clients connected.", mem, mi.uordblks, mi.uordblks - mem, clients);
+		mem = mi.uordblks;
+	}
+}
+
+int create_listen_socket4(int port, int listen_backlog)
+{
+		int sock;
+		int sockopt = 1;
+		struct sockaddr_in name;
+
+		debug(D_LISTENER, "IPv4 creating new listening socket on port %d", port);
+
+		sock = socket(AF_INET, SOCK_STREAM, 0);
+		if(sock < 0) {
+			error("IPv4 socket() failed.");
+			return -1;
+		}
+
+		/* avoid "address already in use" */
+		setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, (void*)&sockopt, sizeof(sockopt));
+
+		memset(&name, 0, sizeof(struct sockaddr_in));
+		name.sin_family = AF_INET;
+		name.sin_port = htons (port);
+		name.sin_addr.s_addr = htonl (INADDR_ANY);
+
+		if(bind (sock, (struct sockaddr *) &name, sizeof (name)) < 0) {
+			close(sock);
+			error("IPv4 bind() failed.");
+			return -1;
+		}
+
+		if(listen(sock, listen_backlog) < 0) {
+			close(sock);
+			fatal("IPv4 listen() failed.");
+			return -1;
+		}
+
+		debug(D_LISTENER, "IPv4 listening port %d created", port);
+		return sock;
+}
+
+int create_listen_socket6(int port, int listen_backlog)
+{
+		int sock = -1;
+		int sockopt = 1;
+		struct sockaddr_in6 name;
+
+		debug(D_LISTENER, "IPv6 creating new listening socket on port %d", port);
+
+		sock = socket(AF_INET6, SOCK_STREAM, 0);
+		if (sock < 0) {
+			error("IPv6 socket() failed.");
+			return -1;
+		}
+
+		/* avoid "address already in use" */
+		setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, (void*)&sockopt, sizeof(sockopt));
+
+		memset(&name, 0, sizeof(struct sockaddr_in6));
+		name.sin6_family = AF_INET6;
+		name.sin6_port = htons ((uint16_t) port);
+		name.sin6_addr = in6addr_any;
+		name.sin6_scope_id = 0;
+
+		if (bind (sock, (struct sockaddr *) &name, sizeof (name)) < 0) {
+			close(sock);
+			error("IPv6 bind() failed.");
+			return -1;
+		}
+
+		if (listen(sock, listen_backlog) < 0) {
+			close(sock);
+			fatal("IPv6 listen() failed.");
+			return -1;
+		}
+
+		debug(D_LISTENER, "IPv6 listening port %d created", port);
+		return sock;
+}
+
+
+// --------------------------------------------------------------------------------------
+// the main socket listener
+
+// 1. it accepts new incoming requests on our port
+// 2. creates a new web_client for each connection received
+// 3. spawns a new pthread to serve the client (this is optimal for keep-alive clients)
+// 4. cleans up old web_clients that their pthreads have been exited
+
+void *socket_listen_main(void *ptr)
+{
+	if(ptr) { ; }
+
+	info("WEB SERVER thread created with task id %d", gettid());
+
+	struct web_client *w;
+	struct timeval tv;
+	int retval;
+
+	if(ptr) { ; }
+
+	if(pthread_setcanceltype(PTHREAD_CANCEL_DEFERRED, NULL) != 0)
+		error("Cannot set pthread cancel type to DEFERRED.");
+
+	if(pthread_setcancelstate(PTHREAD_CANCEL_ENABLE, NULL) != 0)
+		error("Cannot set pthread cancel state to ENABLE.");
+
+	web_client_timeout = (int) config_get_number("global", "disconnect idle web clients after seconds", DEFAULT_DISCONNECT_IDLE_WEB_CLIENTS_AFTER_SECONDS);
+	web_enable_gzip = config_get_boolean("global", "enable web responses gzip compression", web_enable_gzip);
+
+	if(listen_fd < 0) fatal("LISTENER: Listen socket is not ready.");
+
+	fd_set ifds, ofds, efds;
+	int fdmax = listen_fd;
+
+	FD_ZERO (&ifds);
+	FD_ZERO (&ofds);
+	FD_ZERO (&efds);
+
+	for(;;) {
+		tv.tv_sec = 0;
+		tv.tv_usec = 200000;
+
+		if(listen_fd >= 0) {
+			FD_SET(listen_fd, &ifds);
+			FD_SET(listen_fd, &efds);
+		}
+
+		// debug(D_WEB_CLIENT, "LISTENER: Waiting...");
+		retval = select(fdmax+1, &ifds, &ofds, &efds, &tv);
+
+		if(retval == -1) {
+			error("LISTENER: select() failed.");
+			continue;
+		}
+		else if(retval) {
+			// check for new incoming connections
+			if(FD_ISSET(listen_fd, &ifds)) {
+				w = web_client_create(listen_fd);
+				if(unlikely(!w)) {
+					// no need for error log - web_client_create already logged the error
+					continue;
+				}
+
+				if(pthread_create(&w->thread, NULL, web_client_main, w) != 0) {
+					error("%llu: failed to create new thread for web client.");
+					w->obsolete = 1;
+				}
+				else if(pthread_detach(w->thread) != 0) {
+					error("%llu: Cannot request detach of newly created web client thread.", w->id);
+					w->obsolete = 1;
+				}
+			}
+			else debug(D_WEB_CLIENT, "LISTENER: select() didn't do anything.");
+
+		}
+		//else {
+		//	debug(D_WEB_CLIENT, "LISTENER: select() timeout.");
+		//}
+
+		// cleanup unused clients
+		for(w = web_clients; w ; w = w?w->next:NULL) {
+			if(w->obsolete) {
+				debug(D_WEB_CLIENT, "%llu: Removing client.", w->id);
+				// pthread_join(w->thread,  NULL);
+				w = web_client_free(w);
+				log_allocations();
+			}
+		}
+	}
+
+	error("LISTENER: exit!");
+
+	if(listen_fd >= 0) close(listen_fd);
+	exit(2);
+
+	return NULL;
+}
+