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

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