Changeset 1641 for zzuf/trunk/src/zzuf.cTweet

Timestamp:

Jan 10, 2007, 6:17:26 PM (14 years ago)

Author:

Sam Hocevar

Message:

• Implemented memory limits. Probably breaks on other arches because of all the new functions.

File:

• zzuf/trunk/src/zzuf.c (modified) (10 diffs)

zzuf/trunk/src/zzuf.c

@@ -37 +37 @@
 #include <time.h>
 #include <sys/wait.h>
+#include <sys/time.h>
+#include <sys/resource.h>
 
 #include "libzzuf.h"
@@ -81 +83 @@
 static int md5 = 0;
 static int checkexit = 0;
+static int maxmem = -1;
 static double maxtime = -1.0;
 
@@ -120 +123 @@
             { "include",     1, NULL, 'I' },
             { "md5",         0, NULL, 'm' },
+            { "max-memory",  1, NULL, 'M' },
             { "network",     0, NULL, 'n' },
             { "protect",     1, NULL, 'P' },
@@ -132 +136 @@
             { "version",     0, NULL, 'v' },
         };
-        int c = getopt_long(argc, argv, "B:cC:dE:F:iI:mnP:qr:R:s:ST:xhv",
+        int c = getopt_long(argc, argv, "B:cC:dE:F:iI:mM:nP:qr:R:s:ST:xhv",
                             long_options, &option_index);
 #   else
 #       define MOREINFO "Try `%s -h' for more information.\n"
-        int c = getopt(argc, argv, "B:cC:dE:F:iI:mnP:qr:R:s:ST:xhv");
+        int c = getopt(argc, argv, "B:cC:dE:F:iI:mM:nP:qr:R:s:ST:xhv");
 #   endif
         if(c == -1)
@@ -181 +185 @@
         case 'm': /* --md5 */
             md5 = 1;
+            break;
+        case 'M': /* --max-memory */
+            setenv("ZZUF_MEMORY", "1", 1);
+            maxmem = atoi(optarg);
             break;
         case 'n': /* --network */
@@ -425 +433 @@
         case 0:
             /* We’re the child */
+            if(maxmem >= 0)
+            {
+                struct rlimit rlim;
+                rlim.rlim_cur = maxmem * 1000000;
+                rlim.rlim_max = maxmem * 1000000;
+                setrlimit(RLIMIT_AS, &rlim);
+            }
+
             for(j = 0; j < 3; j++)
             {
@@ -529 +545 @@
         else if(WIFSIGNALED(status))
         {
-            fprintf(stdout, "zzuf[seed=%i]: signal %i\n",
-                    child_list[i].seed, WTERMSIG(status));
+            fprintf(stdout, "zzuf[seed=%i]: signal %i%s\n",
+                    child_list[i].seed, WTERMSIG(status),
+                      (WTERMSIG(status) == SIGKILL && maxmem >= 0) ?
+                      " (memory exceeded?)" : "");
             crashes++;
         }
@@ -667 +685 @@
     printf("Usage: zzuf [-cdimnqSx] [-r ratio] [-s seed | -s start:stop]\n");
     printf("                        [-F forks] [-C crashes] [-B bytes] [-T seconds]\n");
-    printf("                        [-P protect] [-R refuse]\n");
+    printf("                        [-M bytes] [-P protect] [-R refuse]\n");
     printf("                        [-I include] [-E exclude] [PROGRAM [ARGS]...]\n");
 #   ifdef HAVE_GETOPT_LONG
@@ -689 +707 @@
     printf("  -I, --include <regex>    only fuzz files matching <regex>\n");
     printf("  -m, --md5                compute the output's MD5 hash\n");
+    printf("  -M, --max-memory <n>     maximum child virtual memory size in MB\n");
     printf("  -n, --network            fuzz network input\n");
     printf("  -P, --protect <list>     protect bytes and characters in <list>\n");
@@ -711 +730 @@
     printf("  -I <regex>       only fuzz files matching <regex>\n");
     printf("  -m               compute the output's MD5 hash\n");
+    printf("  -M               maximum child virtual memory size in MB\n");
     printf("  -n               fuzz network input\n");
     printf("  -P <list>        protect bytes and characters in <list>\n");