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

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

Don't try to use waitpid() information on Win32.

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