script: set a scricter limit on memory

This commit is contained in:
Přemysl Eric Janouch 2014-09-19 01:23:31 +02:00
parent d4c60dc105
commit 125521ddc1
1 changed files with 12 additions and 0 deletions

View File

@ -32,6 +32,9 @@
#include <strings.h>
#include <math.h>
#define ADDRESS_SPACE_LIMIT (100 * 1024 * 1024)
#include <sys/resource.h>
#if defined __GNUC__
#define ATTRIBUTE_PRINTF(x, y) __attribute__ ((format (printf, x, y)))
#else // ! __GNUC__
@ -2164,6 +2167,15 @@ main (int argc, char *argv[])
freopen (NULL, "rb", stdin); setvbuf (stdin, NULL, _IOLBF, BUFSIZ);
freopen (NULL, "wb", stdout); setvbuf (stdout, NULL, _IOLBF, BUFSIZ);
struct rlimit limit =
{
.rlim_cur = ADDRESS_SPACE_LIMIT,
.rlim_max = ADDRESS_SPACE_LIMIT
};
// Lower the memory limits to something sensible to prevent abuse
(void) setrlimit (RLIMIT_AS, &limit);
read_db ();
init_runtime_library ();
register_handler (".", fn_dot);