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

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

Fix a regression in myfork.c causing build failures when libtool does not
define LT_OBJDIR.

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