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

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