diff --git a/src/myfork.c b/src/myfork.c index 64889fa..a84da42 100644 --- a/src/myfork.c +++ b/src/myfork.c @@ -37,6 +37,9 @@ #endif #include #include /* for O_BINARY */ +#if defined HAVE_SYS_RESOURCE_H +# include /* for RLIMIT_AS */ +#endif #include "common.h" #include "opts.h" @@ -47,6 +50,22 @@ #include "md5.h" #include "timer.h" +#if defined RLIMIT_AS +# define ZZUF_RLIMIT_MEM RLIMIT_AS +#elif defined RLIMIT_VMEM +# define ZZUF_RLIMIT_MEM RLIMIT_VMEM +#elif defined RLIMIT_DATA +# define ZZUF_RLIMIT_MEM RLIMIT_DATA +#else +# undef ZZUF_RLIMIT_MEM +#endif + +#if defined RLIMIT_CPU +# define ZZUF_RLIMIT_CPU RLIMIT_CPU +#else +# undef ZZUF_RLIMIT_CPU +#endif + static int run_process(struct opts *, int[][2]); #if defined HAVE_WINDOWS_H diff --git a/src/zzuf.c b/src/zzuf.c index 77c0c06..179fc30 100644 --- a/src/zzuf.c +++ b/src/zzuf.c @@ -54,7 +54,7 @@ # include #endif #if defined HAVE_SYS_RESOURCE_H -# include +# include /* for RLIMIT_AS */ #endif #include "common.h"