diff --git a/quark.c b/quark.c index 8ef60b0..3e18509 100644 --- a/quark.c +++ b/quark.c @@ -14,6 +14,7 @@ #include #include #include +#include #include #include #include @@ -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;