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

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