From de74d9b8fece294d4bfa02360296c7c490d55d9c Mon Sep 17 00:00:00 2001
From: sin <sin@2f30.org>
Date: Thu, 21 Aug 2014 23:09:03 +0100
Subject: [PATCH] Set up process limits and socket timeouts

(Until we have non-blocking sockets + poll())
---
 quark.c | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/quark.c b/quark.c
index 8ef60b0..3e18509 100644
--- a/quark.c
+++ b/quark.c
@@ -14,6 +14,7 @@
 #include <unistd.h>
 #include <arpa/inet.h>
 #include <netinet/in.h>
+#include <sys/resource.h>
 #include <sys/socket.h>
 #include <sys/stat.h>
 #include <sys/wait.h>
@@ -23,6 +24,7 @@ char *argv0;
 
 #define LENGTH(x)  (sizeof x / sizeof x[0])
 #define MAXBUFLEN  1024
+#define NPROCS     512
 #define MIN(x,y)   ((x) < (y) ? (x) : (y))
 
 #define HttpOk          "200 OK"
@@ -495,6 +497,7 @@ invalid_request:
 void
 serve(int fd) {
 	int result;
+	struct timeval tv;
 	socklen_t salen;
 	struct sockaddr sa;
 
@@ -521,6 +524,14 @@ serve(int fd) {
 				break;
 			}
 
+			/* If we haven't received any data within this period, close the
+			 * socket to avoid spamming the process table */
+			tv.tv_sec = 30;
+			tv.tv_usec = 0;
+			if (setsockopt(req.fd, SOL_SOCKET, SO_RCVTIMEO, &tv, sizeof(tv)) < 0)
+				logerrmsg("error\tsetsockopt SO_RCVTIMEO failed: %s\n",
+					  strerror(errno));
+
 			result = request();
 			shutdown(req.fd, SHUT_RD);
 			status = -1;
@@ -562,6 +573,7 @@ main(int argc, char *argv[]) {
 	struct addrinfo hints, *ai = NULL;
 	struct passwd *upwd = NULL;
 	struct group *gpwd = NULL;
+	struct rlimit rlim;
 	int i;
 
 	ARGBEGIN {
@@ -647,6 +659,13 @@ main(int argc, char *argv[]) {
 		goto err;
 	}
 
+	rlim.rlim_cur = NPROCS;
+	rlim.rlim_max = NPROCS;
+	if (setrlimit(RLIMIT_NPROC, &rlim) == -1) {
+		logerrmsg("error\tsetrlimit RLIMIT_NPROC: %s\n", strerror(errno));
+		goto err;
+	}
+
 	if (chdir(docroot) == -1) {
 		logerrmsg("error\tchdir %s: %s\n", docroot, strerror(errno));
 		goto err;