source: zzuf/trunk/src/zzuf.c @ 1681

Last change on this file since 1681 was 1681, checked in by Sam Hocevar, 14 years ago
  • Fix a longopt parsing crash, thanks to Matti Hämäläinen.
  • Property svn:keywords set to Id
File size: 26.1 KB
Line 
1/*
2 *  zzuf - general purpose fuzzer
3 *  Copyright (c) 2002, 2007 Sam Hocevar <sam@zoy.org>
4 *                All Rights Reserved
5 *
6 *  $Id: zzuf.c 1681 2007-01-17 07:22:10Z sam $
7 *
8 *  This program is free software. It comes without any warranty, to
9 *  the extent permitted by applicable law. You can redistribute it
10 *  and/or modify it under the terms of the Do What The Fuck You Want
11 *  To Public License, Version 2, as published by Sam Hocevar. See
12 *  http://sam.zoy.org/wtfpl/COPYING for more details.
13 */
14
15/*
16 *  main.c: main program
17 */
18
19#include "config.h"
20
21#if defined HAVE_STDINT_H
22#   include <stdint.h>
23#elif defined HAVE_INTTYPES_H
24#   include <inttypes.h>
25#endif
26#if defined(HAVE_GETOPT_H)
27#   include <getopt.h>
28#endif
29#include <stdio.h>
30#include <stdlib.h>
31#include <unistd.h>
32#include <regex.h>
33#include <string.h>
34#include <errno.h>
35#include <signal.h>
36#include <sys/wait.h>
37#include <sys/time.h>
38#include <sys/resource.h>
39
40#include "libzzuf.h"
41#include "random.h"
42#include "fd.h"
43#include "fuzz.h"
44#include "md5.h"
45#include "timer.h"
46
47static void loop_stdin(void);
48
49static void spawn_children(void);
50static void clean_children(void);
51static void read_children(void);
52
53static char const *sig2str(int);
54static char *merge_regex(char *, char *);
55static char *merge_file(char *, char *);
56static void set_environment(char const *);
57static void version(void);
58#if defined(HAVE_GETOPT_H)
59static void usage(void);
60#endif
61
62static struct child_list
63{
64    enum status
65    {
66        STATUS_FREE,
67        STATUS_RUNNING,
68        STATUS_SIGTERM,
69        STATUS_SIGKILL,
70        STATUS_EOF,
71    } status;
72
73    pid_t pid;
74    int fd[3]; /* 0 is debug, 1 is stderr, 2 is stdout */
75    int bytes, seed;
76    double ratio;
77    int64_t date;
78    struct md5 *ctx;
79} *child_list;
80static int maxforks = 1, child_count = 0, maxcrashes = 1, crashes = 0;
81
82static char **newargv;
83static char *protect = NULL, *refuse = NULL;
84static uint32_t seed = DEFAULT_SEED;
85static uint32_t endseed = DEFAULT_SEED + 1;
86static double minratio = DEFAULT_RATIO;
87static double maxratio = DEFAULT_RATIO;
88static int quiet = 0;
89static int maxbytes = -1;
90static int md5 = 0;
91static int checkexit = 0;
92static int verbose = 0;
93static int maxmem = -1;
94static int64_t maxtime = -1;
95static int64_t delay = 0;
96static int64_t lastlaunch = 0;
97
98#define ZZUF_FD_SET(fd, p_fdset, maxfd) \
99    if(fd >= 0) \
100    { \
101        FD_SET(fd, p_fdset); \
102        if(fd > maxfd) \
103            maxfd = fd; \
104    }
105
106#define ZZUF_FD_ISSET(fd, p_fdset) \
107    ((fd >= 0) && (FD_ISSET(fd, p_fdset)))
108
109int main(int argc, char *argv[])
110{
111    char *parser, *include, *exclude;
112    int i, cmdline = 0;
113
114    include = exclude = NULL;
115
116#if defined(HAVE_GETOPT_H)
117    for(;;)
118    {
119#   define OPTSTR "AB:cC:dD:E:F:iI:mM:nP:qr:R:s:ST:vxhV"
120#   ifdef HAVE_GETOPT_LONG
121#       define MOREINFO "Try `%s --help' for more information.\n"
122        int option_index = 0;
123        static struct option long_options[] =
124        {
125            /* Long option, needs arg, flag, short option */
126            { "autoinc",     0, NULL, 'A' },
127            { "max-bytes",   1, NULL, 'B' },
128            { "cmdline",     0, NULL, 'c' },
129            { "max-crashes", 1, NULL, 'C' },
130            { "debug",       0, NULL, 'd' },
131            { "delay",       1, NULL, 'D' },
132            { "exclude",     1, NULL, 'E' },
133            { "max-forks",   1, NULL, 'F' },
134            { "stdin",       0, NULL, 'i' },
135            { "include",     1, NULL, 'I' },
136            { "md5",         0, NULL, 'm' },
137            { "max-memory",  1, NULL, 'M' },
138            { "network",     0, NULL, 'n' },
139            { "protect",     1, NULL, 'P' },
140            { "quiet",       0, NULL, 'q' },
141            { "ratio",       1, NULL, 'r' },
142            { "refuse",      1, NULL, 'R' },
143            { "seed",        1, NULL, 's' },
144            { "signal",      0, NULL, 'S' },
145            { "max-time",    1, NULL, 'T' },
146            { "verbose",     0, NULL, 'v' },
147            { "check-exit",  0, NULL, 'x' },
148            { "help",        0, NULL, 'h' },
149            { "version",     0, NULL, 'V' },
150            { NULL,          0, NULL,  0  }
151        };
152        int c = getopt_long(argc, argv, OPTSTR, long_options, &option_index);
153#   else
154#       define MOREINFO "Try `%s -h' for more information.\n"
155        int c = getopt(argc, argv, OPTSTR);
156#   endif
157        if(c == -1)
158            break;
159
160        switch(c)
161        {
162        case 'A': /* --autoinc */
163            setenv("ZZUF_AUTOINC", "1", 1);
164            break;
165        case 'B': /* --max-bytes */
166            maxbytes = atoi(optarg);
167            break;
168        case 'c': /* --cmdline */
169            cmdline = 1;
170            break;
171        case 'C': /* --max-crashes */
172            maxcrashes = atoi(optarg);
173            if(maxcrashes <= 0)
174                maxcrashes = 0;
175            break;
176        case 'd': /* --debug */
177            setenv("ZZUF_DEBUG", "1", 1);
178            break;
179        case 'D': /* --delay */
180            delay = (int64_t)(atof(optarg) * 1000000.0);
181            break;
182        case 'E': /* --exclude */
183            exclude = merge_regex(exclude, optarg);
184            if(!exclude)
185            {
186                printf("%s: invalid regex -- `%s'\n", argv[0], optarg);
187                return EXIT_FAILURE;
188            }
189            break;
190        case 'F': /* --max-forks */
191            maxforks = atoi(optarg) > 1 ? atoi(optarg) : 1;
192            break;
193        case 'i': /* --stdin */
194            setenv("ZZUF_STDIN", "1", 1);
195            break;
196        case 'I': /* --include */
197            include = merge_regex(include, optarg);
198            if(!include)
199            {
200                printf("%s: invalid regex -- `%s'\n", argv[0], optarg);
201                return EXIT_FAILURE;
202            }
203            break;
204        case 'm': /* --md5 */
205            md5 = 1;
206            break;
207        case 'M': /* --max-memory */
208            setenv("ZZUF_MEMORY", "1", 1);
209            maxmem = atoi(optarg);
210            break;
211        case 'n': /* --network */
212            setenv("ZZUF_NETWORK", "1", 1);
213            break;
214        case 'P': /* --protect */
215            protect = optarg;
216            break;
217        case 'q': /* --quiet */
218            quiet = 1;
219            break;
220        case 'r': /* --ratio */
221            parser = strchr(optarg, ':');
222            minratio = atof(optarg);
223            maxratio = parser ? atof(parser + 1) : minratio;
224            break;
225        case 'R': /* --refuse */
226            refuse = optarg;
227            break;
228        case 's': /* --seed */
229            parser = strchr(optarg, ':');
230            seed = atol(optarg);
231            endseed = parser ? (uint32_t)atoi(parser + 1) : seed + 1;
232            break;
233        case 'S': /* --signal */
234            setenv("ZZUF_SIGNAL", "1", 1);
235            break;
236        case 'T': /* --max-time */
237            maxtime = (int64_t)(atof(optarg) * 1000000.0);
238            break;
239        case 'x': /* --check-exit */
240            checkexit = 1;
241            break;
242        case 'v': /* --verbose */
243            verbose = 1;
244            break;
245        case 'h': /* --help */
246            usage();
247            return 0;
248        case 'V': /* --version */
249            version();
250            return 0;
251        default:
252            printf("%s: invalid option -- %c\n", argv[0], c);
253            printf(MOREINFO, argv[0]);
254            return EXIT_FAILURE;
255        }
256    }
257#else
258#   define MOREINFO "Usage: %s message...\n"
259    int optind = 1;
260#endif
261
262    _zz_setratio(minratio, maxratio);
263    _zz_setseed(seed);
264
265    /* If asked to read from the standard input */
266    if(optind >= argc)
267    {
268        if(endseed != seed + 1)
269        {
270            printf("%s: seed ranges are incompatible with stdin fuzzing\n",
271                   argv[0]);
272            printf(MOREINFO, argv[0]);
273            return EXIT_FAILURE;
274        }
275
276        loop_stdin();
277
278        return EXIT_SUCCESS;
279    }
280
281    /* If asked to launch programs */
282    if(cmdline)
283    {
284        int dashdash = 0;
285
286        for(i = optind + 1; i < argc; i++)
287        {
288            if(dashdash)
289                include = merge_file(include, argv[i]);
290            else if(!strcmp("--", argv[i]))
291                dashdash = 1;
292            else if(argv[i][0] != '-')
293                include = merge_file(include, argv[i]);
294        }
295    }
296
297    if(include)
298        setenv("ZZUF_INCLUDE", include, 1);
299    if(exclude)
300        setenv("ZZUF_EXCLUDE", exclude, 1);
301    if(protect)
302        setenv("ZZUF_PROTECT", protect, 1);
303    if(refuse)
304        setenv("ZZUF_REFUSE", refuse, 1);
305
306    /* Allocate memory for children handling */
307    child_list = malloc(maxforks * sizeof(struct child_list));
308    for(i = 0; i < maxforks; i++)
309        child_list[i].status = STATUS_FREE;
310    child_count = 0;
311
312    /* Preload libzzuf.so */
313    set_environment(argv[0]);
314
315    /* Create new argv */
316    newargv = malloc((argc - optind + 1) * sizeof(char *));
317    memcpy(newargv, argv + optind, (argc - optind) * sizeof(char *));
318    newargv[argc - optind] = (char *)NULL;
319
320    /* Main loop */
321    while(child_count || seed < endseed)
322    {
323        /* Spawn new children, if necessary */
324        spawn_children();
325
326        /* Cleanup dead or dying children */
327        clean_children();
328
329        /* Read data from children */
330        read_children();
331
332        if(maxcrashes && crashes >= maxcrashes && child_count == 0)
333            break;
334    }
335
336    /* Clean up */
337    free(newargv);
338    free(child_list);
339
340    return crashes ? EXIT_FAILURE : EXIT_SUCCESS;   
341}
342
343static void loop_stdin(void)
344{
345    uint8_t md5sum[16];
346    struct md5 *ctx = NULL;
347
348    if(md5)
349        ctx = _zz_md5_init();
350
351    if(protect)
352        _zz_protect(protect);
353    if(refuse)
354        _zz_refuse(refuse);
355
356    _zz_fd_init();
357    _zz_register(0);
358
359    for(;;)
360    {
361        uint8_t buf[BUFSIZ];
362        int ret, off = 0, nw = 0;
363
364        ret = read(0, buf, BUFSIZ);
365        if(ret <= 0)
366            break;
367
368        _zz_fuzz(0, buf, ret);
369        _zz_addpos(0, ret);
370
371        if(md5)
372            _zz_md5_add(ctx, buf, ret);
373        else while(ret)
374        {
375            if((nw = write(1, buf + off, (size_t)ret)) < 0)
376                break;
377            ret -= nw;
378            off += nw;
379        }
380    }
381
382    if(md5)
383    {
384        _zz_md5_fini(md5sum, ctx);
385        fprintf(stdout, "zzuf[s=%i,r=%g]: %.02x%.02x%.02x%.02x%.02x"
386                "%.02x%.02x%.02x%.02x%.02x%.02x%.02x%.02x%.02x%.02x%.02x\n",
387                seed, minratio, md5sum[0], md5sum[1], md5sum[2], md5sum[3],
388                md5sum[4], md5sum[5], md5sum[6], md5sum[7],
389                md5sum[8], md5sum[9], md5sum[10], md5sum[11],
390                md5sum[12], md5sum[13], md5sum[14], md5sum[15]);
391        fflush(stdout);
392    }
393
394    _zz_unregister(0);
395    _zz_fd_fini();
396}
397
398static char *merge_file(char *regex, char *file)
399{
400    char *newfile = malloc(5 + 2 * strlen(file) + 1 + 1), *tmp = newfile;
401
402    *tmp++ = '(';
403    *tmp++ = '^';
404    *tmp++ = '|';
405    *tmp++ = '/';
406    *tmp++ = ')';
407    while(*file)
408    {
409        if(strchr("^.[$()|*+?{\\", *file))
410            *tmp++ = '\\';
411        *tmp++ = *file++;
412    }
413    *tmp++ = '$';
414    *tmp++ = '\0';
415
416    tmp = merge_regex(regex, newfile);
417    free(newfile);
418    return tmp;
419}
420
421static char *merge_regex(char *regex, char *string)
422{
423    regex_t optre;
424
425    if(regex)
426    {
427        regex = realloc(regex, strlen(regex) + strlen(string) + 1 + 1);
428        sprintf(regex + strlen(regex) - 1, "|%s)", string);
429    }
430    else
431    {
432        regex = malloc(1 + strlen(string) + 1 + 1);
433        sprintf(regex, "(%s)", string);
434    }
435
436    if(regcomp(&optre, regex, REG_EXTENDED) != 0)
437    {
438        free(regex);
439        return NULL;
440    }
441    regfree(&optre);
442
443    return regex;
444}
445
446static void spawn_children(void)
447{
448    static int const files[] = { DEBUG_FILENO, STDERR_FILENO, STDOUT_FILENO };
449    char buf[64];
450    int fd[3][2];
451    int64_t now = _zz_time();
452    pid_t pid;
453    int i, j;
454
455    if(child_count == maxforks)
456        return; /* no slot */
457
458    if(seed == endseed)
459        return; /* job finished */
460
461    if(maxcrashes && crashes >= maxcrashes)
462        return; /* all jobs crashed */
463
464    if(delay > 0 && lastlaunch + delay > now)
465        return; /* too early */
466
467    /* Find the empty slot */
468    for(i = 0; i < maxforks; i++)
469        if(child_list[i].status == STATUS_FREE)
470            break;
471
472    /* Prepare communication pipe */
473    for(j = 0; j < 3; j++)
474        if(pipe(fd[j]) == -1)
475        {
476            perror("pipe");
477            return;
478        }
479
480    /* Fork and launch child */
481    pid = fork();
482    switch(pid)
483    {
484        case -1:
485            perror("fork");
486            return;
487        case 0:
488            /* We’re the child */
489            if(maxmem >= 0)
490            {
491                struct rlimit rlim;
492                rlim.rlim_cur = maxmem * 1000000;
493                rlim.rlim_max = maxmem * 1000000;
494                setrlimit(RLIMIT_AS, &rlim);
495            }
496
497            /* We loop in reverse order so that files[0] is done last,
498             * just in case one of the other dup2()ed fds had the value */
499            for(j = 3; j--; )
500            {
501                close(fd[j][0]);
502                if(fd[j][1] != files[j])
503                {
504                    dup2(fd[j][1], files[j]);
505                    close(fd[j][1]);
506                }
507            }
508
509            /* Set environment variables */
510            sprintf(buf, "%i", seed);
511            setenv("ZZUF_SEED", buf, 1);
512            sprintf(buf, "%g", minratio);
513            setenv("ZZUF_MINRATIO", buf, 1);
514            sprintf(buf, "%g", maxratio);
515            setenv("ZZUF_MAXRATIO", buf, 1);
516
517            /* Run our process */
518            if(execvp(newargv[0], newargv))
519            {
520                perror(newargv[0]);
521                exit(EXIT_FAILURE);
522            }
523            return;
524    }
525
526    /* We’re the parent, acknowledge spawn */
527    child_list[i].date = now;
528    child_list[i].pid = pid;
529    for(j = 0; j < 3; j++)
530    {
531        close(fd[j][1]);
532        child_list[i].fd[j] = fd[j][0];
533    }
534    child_list[i].bytes = 0;
535    child_list[i].seed = seed;
536    child_list[i].ratio = _zz_getratio();
537    child_list[i].status = STATUS_RUNNING;
538    if(md5)
539        child_list[i].ctx = _zz_md5_init();
540
541    if(verbose)
542        fprintf(stderr, "zzuf[s=%i,r=%g]: launched %s\n",
543                child_list[i].seed, child_list[i].ratio, newargv[0]);
544
545    lastlaunch = now;
546    child_count++;
547    seed++;
548
549    _zz_setseed(seed);
550}
551
552static void clean_children(void)
553{
554    int64_t now = _zz_time();
555    int i, j;
556
557    /* Terminate children if necessary */
558    for(i = 0; i < maxforks; i++)
559    {
560        if(child_list[i].status == STATUS_RUNNING
561            && maxbytes >= 0 && child_list[i].bytes > maxbytes)
562        {
563            if(verbose)
564                fprintf(stderr, "zzuf[s=%i,r=%g]: "
565                        "data output exceeded, sending SIGTERM\n",
566                        child_list[i].seed, child_list[i].ratio);
567            kill(child_list[i].pid, SIGTERM);
568            child_list[i].date = now;
569            child_list[i].status = STATUS_SIGTERM;
570        }
571
572        if(child_list[i].status == STATUS_RUNNING
573            && maxtime >= 0
574            && now > child_list[i].date + maxtime)
575        {
576            if(verbose)
577                fprintf(stderr, "zzuf[s=%i,r=%g]: "
578                        "running time exceeded, sending SIGTERM\n",
579                        child_list[i].seed, child_list[i].ratio);
580            kill(child_list[i].pid, SIGTERM);
581            child_list[i].date = now;
582            child_list[i].status = STATUS_SIGTERM;
583        }
584    }
585
586    /* Kill children if necessary (still there after 2 seconds) */
587    for(i = 0; i < maxforks; i++)
588    {
589        if(child_list[i].status == STATUS_SIGTERM
590            && now > child_list[i].date + 2000000)
591        {
592            if(verbose)
593                fprintf(stderr, "zzuf[s=%i,r=%g]: "
594                        "not responding, sending SIGKILL\n",
595                        child_list[i].seed, child_list[i].ratio);
596            kill(child_list[i].pid, SIGKILL);
597            child_list[i].status = STATUS_SIGKILL;
598        }
599    }
600
601    /* Collect dead children */
602    for(i = 0; i < maxforks; i++)
603    {
604        uint8_t md5sum[16];
605        int status;
606        pid_t pid;
607
608        if(child_list[i].status != STATUS_SIGKILL
609            && child_list[i].status != STATUS_SIGTERM
610            && child_list[i].status != STATUS_EOF)
611            continue;
612
613        pid = waitpid(child_list[i].pid, &status, WNOHANG);
614        if(pid <= 0)
615            continue;
616
617        if(checkexit && WIFEXITED(status) && WEXITSTATUS(status))
618        {
619            fprintf(stderr, "zzuf[s=%i,r=%g]: exit %i\n",
620                    child_list[i].seed, child_list[i].ratio,
621                    WEXITSTATUS(status));
622            crashes++;
623        }
624        else if(WIFSIGNALED(status)
625                 && !(WTERMSIG(status) == SIGTERM
626                       && child_list[i].status == STATUS_SIGTERM))
627        {
628            fprintf(stderr, "zzuf[s=%i,r=%g]: signal %i%s%s\n",
629                    child_list[i].seed, child_list[i].ratio,
630                    WTERMSIG(status), sig2str(WTERMSIG(status)),
631                      (WTERMSIG(status) == SIGKILL && maxmem >= 0) ?
632                      " (memory exceeded?)" : "");
633            crashes++;
634        }
635
636        for(j = 0; j < 3; j++)
637            if(child_list[i].fd[j] >= 0)
638                close(child_list[i].fd[j]);
639
640        if(md5)
641        {
642            _zz_md5_fini(md5sum, child_list[i].ctx);
643            fprintf(stdout, "zzuf[s=%i,r=%g]: %.02x%.02x%.02x%.02x%.02x"
644                    "%.02x%.02x%.02x%.02x%.02x%.02x%.02x%.02x%.02x%.02x%.02x\n",
645                    child_list[i].seed, child_list[i].ratio, md5sum[0],
646                    md5sum[1], md5sum[2], md5sum[3], md5sum[4], md5sum[5],
647                    md5sum[6], md5sum[7], md5sum[8], md5sum[9], md5sum[10],
648                    md5sum[11], md5sum[12], md5sum[13], md5sum[14], md5sum[15]);
649            fflush(stdout);
650        }
651        child_list[i].status = STATUS_FREE;
652        child_count--;
653    }
654}
655
656static void read_children(void)
657{
658    struct timeval tv;
659    fd_set fdset;
660    int i, j, ret, maxfd = 0;
661
662    /* Read data from all sockets */
663    FD_ZERO(&fdset);
664    for(i = 0; i < maxforks; i++)
665    {
666        if(child_list[i].status != STATUS_RUNNING)
667            continue;
668
669        for(j = 0; j < 3; j++)
670            ZZUF_FD_SET(child_list[i].fd[j], &fdset, maxfd);
671    }
672    tv.tv_sec = 0;
673    tv.tv_usec = 1000;
674
675    ret = select(maxfd + 1, &fdset, NULL, NULL, &tv);
676    if(ret < 0)
677        perror("select");
678    if(ret <= 0)
679        return;
680
681    /* XXX: cute (i, j) iterating hack */
682    for(i = 0, j = 0; i < maxforks; i += (j == 2), j = (j + 1) % 3)
683    {
684        uint8_t buf[BUFSIZ];
685
686        if(child_list[i].status != STATUS_RUNNING)
687            continue;
688
689        if(!ZZUF_FD_ISSET(child_list[i].fd[j], &fdset))
690            continue;
691
692        ret = read(child_list[i].fd[j], buf, BUFSIZ - 1);
693        if(ret > 0)
694        {
695            /* We got data */
696            if(j != 0)
697                child_list[i].bytes += ret;
698
699            if(md5 && j == 2)
700                _zz_md5_add(child_list[i].ctx, buf, ret);
701            else if(!quiet || j == 0)
702                write((j < 2) ? STDERR_FILENO : STDOUT_FILENO, buf, ret);
703        }
704        else if(ret == 0)
705        {
706            /* End of file reached */
707            close(child_list[i].fd[j]);
708            child_list[i].fd[j] = -1;
709
710            if(child_list[i].fd[0] == -1 && child_list[i].fd[1] == -1
711               && child_list[i].fd[2] == -1)
712                child_list[i].status = STATUS_EOF;
713        }
714    }
715}
716
717static char const *sig2str(int signum)
718{
719    switch(signum)
720    {
721        case SIGABRT:  return " (SIGABRT)";
722        case SIGFPE:   return " (SIGFPE)";
723        case SIGILL:   return " (SIGILL)";
724        case SIGQUIT:  return " (SIGQUIT)";
725        case SIGSEGV:  return " (SIGSEGV)";
726        case SIGTRAP:  return " (SIGTRAP)";
727#ifdef SIGSYS
728        case SIGSYS:   return " (SIGSYS)";
729#endif
730#ifdef SIGEMT
731        case SIGEMT:   return " (SIGEMT)";
732#endif
733#ifdef SIGBUS
734        case SIGBUS:   return " (SIGBUS)";
735#endif
736#ifdef SIGXCPU
737        case SIGXCPU:  return " (SIGXCPU)";
738#endif
739#ifdef SIGXFSZ
740        case SIGXFSZ:  return " (SIGXFSZ)";
741#endif
742    }
743
744    return "";
745}
746
747static void set_environment(char const *progpath)
748{
749    char *libpath, *tmp;
750    int ret, len = strlen(progpath);
751#if defined __APPLE__
752#   define FILENAME "libzzuf.dylib"
753#   define EXTRAINFO ""
754#   define PRELOAD "DYLD_INSERT_LIBRARIES"
755    setenv("DYLD_FORCE_FLAT_NAMESPACE", "1", 1);
756#elif defined __osf__
757#   define FILENAME "libzzuf.so"
758#   define EXTRAINFO ":DEFAULT"
759#   define PRELOAD "_RLD_LIST"
760#else
761#   define FILENAME "libzzuf.so"
762#   define EXTRAINFO ""
763#   define PRELOAD "LD_PRELOAD"
764#endif
765
766    libpath = malloc(len + strlen("/.libs/" FILENAME EXTRAINFO) + 1);
767    strcpy(libpath, progpath);
768
769    tmp = strrchr(libpath, '/');
770    strcpy(tmp ? tmp + 1 : libpath, ".libs/" FILENAME);
771    ret = access(libpath, R_OK);
772
773    strcpy(tmp ? tmp + 1 : libpath, ".libs/" FILENAME EXTRAINFO);
774    if(ret == 0)
775        setenv(PRELOAD, libpath, 1);
776    else
777        setenv(PRELOAD, LIBDIR "/" FILENAME EXTRAINFO, 1);
778    free(libpath);
779}
780
781static void version(void)
782{
783    printf("zzuf %s\n", VERSION);
784    printf("Copyright (C) 2002, 2007 Sam Hocevar <sam@zoy.org>\n");
785    printf("This is free software.  You may redistribute copies of it under the\n");
786    printf("terms of the Do What The Fuck You Want To Public License, Version 2\n");
787    printf("<http://sam.zoy.org/wtfpl/>.\n");
788    printf("There is NO WARRANTY, to the extent permitted by law.\n");
789    printf("\n");
790    printf("Written by Sam Hocevar. Report bugs to <sam@zoy.org>.\n");
791}
792
793#if defined(HAVE_GETOPT_H)
794static void usage(void)
795{
796    printf("Usage: zzuf [-AcdimnqSvx] [-r ratio] [-s seed | -s start:stop]\n");
797    printf("                          [-D delay] [-F forks] [-C crashes] [-B bytes]\n");
798    printf("                          [-T seconds] [-M bytes] [-P protect] [-R refuse]\n");
799    printf("                          [-I include] [-E exclude] [PROGRAM [--] [ARGS]...]\n");
800#   ifdef HAVE_GETOPT_LONG
801    printf("       zzuf -h | --help\n");
802    printf("       zzuf -V | --version\n");
803#   else
804    printf("       zzuf -h\n");
805    printf("       zzuf -V\n");
806#   endif
807    printf("Run PROGRAM with optional arguments ARGS and fuzz its input.\n");
808    printf("\n");
809    printf("Mandatory arguments to long options are mandatory for short options too.\n");
810#   ifdef HAVE_GETOPT_LONG
811    printf("  -A, --autoinc             increment seed each time a new file is opened\n");
812    printf("  -B, --max-bytes <n>       kill children that output more than <n> bytes\n");
813    printf("  -c, --cmdline             only fuzz files specified in the command line\n");
814    printf("  -C, --max-crashes <n>     stop after <n> children have crashed (default 1)\n");
815    printf("  -d, --debug               print debug messages\n");
816    printf("  -D, --delay               delay between forks\n");
817    printf("  -E, --exclude <regex>     do not fuzz files matching <regex>\n");
818    printf("  -F, --max-forks <n>       number of concurrent children (default 1)\n");
819    printf("  -i, --stdin               fuzz standard input\n");
820    printf("  -I, --include <regex>     only fuzz files matching <regex>\n");
821    printf("  -m, --md5                 compute the output's MD5 hash\n");
822    printf("  -M, --max-memory <n>      maximum child virtual memory size in MB\n");
823    printf("  -n, --network             fuzz network input\n");
824    printf("  -P, --protect <list>      protect bytes and characters in <list>\n");
825    printf("  -q, --quiet               do not print children's messages\n");
826    printf("  -r, --ratio <ratio>       bit fuzzing ratio (default %g)\n", DEFAULT_RATIO);
827    printf("      --ratio <start:stop>  specify a ratio range\n");
828    printf("  -R, --refuse <list>       refuse bytes and characters in <list>\n");
829    printf("  -s, --seed <seed>         random seed (default %i)\n", DEFAULT_SEED);
830    printf("      --seed <start:stop>   specify a seed range\n");
831    printf("  -S, --signal              prevent children from diverting crashing signals\n");
832    printf("  -T, --max-time <n>        kill children that run for more than <n> seconds\n");
833    printf("  -v, --verbose             print information during the run\n");
834    printf("  -x, --check-exit          report processes that exit with a non-zero status\n");
835    printf("  -h, --help                display this help and exit\n");
836    printf("  -V, --version             output version information and exit\n");
837#   else
838    printf("  -A               increment seed each time a new file is opened\n");
839    printf("  -B <n>           kill children that output more than <n> bytes\n");
840    printf("  -c               only fuzz files specified in the command line\n");
841    printf("  -C <n>           stop after <n> children have crashed (default 1)\n");
842    printf("  -d               print debug messages\n");
843    printf("  -D               delay between forks\n");
844    printf("  -E <regex>       do not fuzz files matching <regex>\n");
845    printf("  -F <n>           number of concurrent forks (default 1)\n");
846    printf("  -i               fuzz standard input\n");
847    printf("  -I <regex>       only fuzz files matching <regex>\n");
848    printf("  -m               compute the output's MD5 hash\n");
849    printf("  -M               maximum child virtual memory size in MB\n");
850    printf("  -n               fuzz network input\n");
851    printf("  -P <list>        protect bytes and characters in <list>\n");
852    printf("  -q               do not print the fuzzed application's messages\n");
853    printf("  -r <ratio>       bit fuzzing ratio (default %g)\n", DEFAULT_RATIO);
854    printf("     <start:stop>  specify a ratio range\n");
855    printf("  -R <list>        refuse bytes and characters in <list>\n");
856    printf("  -s <seed>        random seed (default %i)\n", DEFAULT_SEED);
857    printf("     <start:stop>  specify a seed range\n");
858    printf("  -S               prevent children from diverting crashing signals\n");
859    printf("  -T <n>           kill children that run for more than <n> seconds\n");
860    printf("  -v               print information during the run\n");
861    printf("  -x               report processes that exit with a non-zero status\n");
862    printf("  -h               display this help and exit\n");
863    printf("  -V               output version information and exit\n");
864#   endif
865    printf("\n");
866    printf("Written by Sam Hocevar. Report bugs to <sam@zoy.org>.\n");
867}
868#endif
869
Note: See TracBrowser for help on using the repository browser.