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

Last change on this file since 4239 was 4239, checked in by Sam Hocevar, 11 years ago

Deprecate --deny <list> in favour of --allow !<list>.

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