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

Last change on this file since 1720 was 1720, checked in by Sam Hocevar, 14 years ago
  • Implemented -f/--fuzzing (fuzzing mode).
  • Property svn:keywords set to Id
File size: 33.1 KB
Line 
1/*
2 *  zzuf - general purpose fuzzer
3 *  Copyright (c) 2002, 2007 Sam Hocevar <sam@zoy.org>
4 *                All Rights Reserved
5 *
6 *  $Id: zzuf.c 1720 2007-01-27 19:05:46Z sam $
7 *
8 *  This program is free software. It comes without any warranty, to
9 *  the extent permitted by applicable law. You can redistribute it
10 *  and/or modify it under the terms of the Do What The Fuck You Want
11 *  To Public License, Version 2, as published by Sam Hocevar. See
12 *  http://sam.zoy.org/wtfpl/COPYING for more details.
13 */
14
15/*
16 *  main.c: main program
17 */
18
19#include "config.h"
20
21#if defined HAVE_STDINT_H
22#   include <stdint.h>
23#elif defined HAVE_INTTYPES_H
24#   include <inttypes.h>
25#endif
26#if defined HAVE_GETOPT_H
27#   include <getopt.h>
28#endif
29#include <stdio.h>
30#include <stdlib.h>
31#include <unistd.h>
32#if defined HAVE_REGEX_H
33#   include <regex.h>
34#endif
35#if defined HAVE_WINDOWS_H
36#   include <windows.h>
37#endif
38#if defined HAVE_WINSOCK2_H
39#   include <winsock2.h>
40#endif
41#if defined HAVE_IO_H
42#   include <io.h>
43#endif
44#include <string.h>
45#include <fcntl.h>
46#include <errno.h>
47#include <signal.h>
48#if defined HAVE_SYS_WAIT_H
49#   include <sys/wait.h>
50#endif
51#include <sys/time.h>
52#if defined HAVE_SYS_RESOURCE_H
53#   include <sys/resource.h>
54#endif
55
56#include "libzzuf.h"
57#include "opts.h"
58#include "random.h"
59#include "fd.h"
60#include "fuzz.h"
61#include "md5.h"
62#include "timer.h"
63
64#if !defined SIGKILL
65#   define SIGKILL 9
66#endif
67
68/* We use file descriptor 17 as the debug channel */
69#define DEBUG_FILENO 17
70#define DEBUG_FILENO_STR "17"
71
72static void loop_stdin(struct opts *);
73static int run_process(char const *, char *[]);
74
75static void spawn_children(struct opts *);
76static void clean_children(struct opts *);
77static void read_children(struct opts *);
78
79#if !defined HAVE_SETENV
80static void setenv(char const *, char const *, int);
81#endif
82#if defined HAVE_WAITPID
83static char const *sig2str(int);
84#endif
85#if defined HAVE_WINDOWS_H
86static int dll_inject(void *, void *);
87static void *get_entry(char const *);
88#endif
89static void finfo(FILE *, struct opts *, uint32_t);
90#if defined HAVE_REGEX_H
91static char *merge_regex(char *, char *);
92static char *merge_file(char *, char *);
93#endif
94static void version(void);
95#if defined HAVE_GETOPT_H
96static void usage(void);
97#endif
98
99#if defined HAVE_WINDOWS_H
100static inline void addcpy(void *buf, void *x)
101{
102    memcpy(buf, &x, 4);
103}
104#endif
105
106#define ZZUF_FD_SET(fd, p_fdset, maxfd) \
107    if(fd >= 0) \
108    { \
109        FD_SET((unsigned int)fd, p_fdset); \
110        if(fd > maxfd) \
111            maxfd = fd; \
112    }
113
114#define ZZUF_FD_ISSET(fd, p_fdset) \
115    ((fd >= 0) && (FD_ISSET(fd, p_fdset)))
116
117int main(int argc, char *argv[])
118{
119    struct opts _opts, *opts = &_opts;
120    char *tmp;
121#if defined HAVE_REGEX_H
122    char *include = NULL, *exclude = NULL;
123    int cmdline = 0;
124#endif
125    int i;
126
127    _zz_opts_init(opts);
128
129#if defined HAVE_GETOPT_H
130    for(;;)
131    {
132#   if defined HAVE_REGEX_H
133#       define OPTSTR "Ab:B:cC:dD:E:f:F:iI:mM:nP:qr:R:s:ST:vxhV"
134#   else
135#       define OPTSTR "Ab:B:C:dD:f:F:imM:nP:qr:R:s:ST:vxhV"
136#   endif
137#   if defined HAVE_GETOPT_LONG
138#       define MOREINFO "Try `%s --help' for more information.\n"
139        int option_index = 0;
140        static struct option long_options[] =
141        {
142            /* Long option, needs arg, flag, short option */
143            { "autoinc",     0, NULL, 'A' },
144            { "bytes",       1, NULL, 'b' },
145            { "max-bytes",   1, NULL, 'B' },
146#if defined HAVE_REGEX_H
147            { "cmdline",     0, NULL, 'c' },
148#endif
149            { "max-crashes", 1, NULL, 'C' },
150            { "debug",       0, NULL, 'd' },
151            { "delay",       1, NULL, 'D' },
152#if defined HAVE_REGEX_H
153            { "exclude",     1, NULL, 'E' },
154#endif
155            { "fuzzing",     1, NULL, 'f' },
156            { "max-forks",   1, NULL, 'F' },
157            { "stdin",       0, NULL, 'i' },
158#if defined HAVE_REGEX_H
159            { "include",     1, NULL, 'I' },
160#endif
161            { "md5",         0, NULL, 'm' },
162            { "max-memory",  1, NULL, 'M' },
163            { "network",     0, NULL, 'n' },
164            { "protect",     1, NULL, 'P' },
165            { "quiet",       0, NULL, 'q' },
166            { "ratio",       1, NULL, 'r' },
167            { "refuse",      1, NULL, 'R' },
168            { "seed",        1, NULL, 's' },
169            { "signal",      0, NULL, 'S' },
170            { "max-time",    1, NULL, 'T' },
171            { "verbose",     0, NULL, 'v' },
172            { "check-exit",  0, NULL, 'x' },
173            { "help",        0, NULL, 'h' },
174            { "version",     0, NULL, 'V' },
175            { NULL,          0, NULL,  0  }
176        };
177        int c = getopt_long(argc, argv, OPTSTR, long_options, &option_index);
178#   else
179#       define MOREINFO "Try `%s -h' for more information.\n"
180        int c = getopt(argc, argv, OPTSTR);
181#   endif
182        if(c == -1)
183            break;
184
185        switch(c)
186        {
187        case 'A': /* --autoinc */
188            setenv("ZZUF_AUTOINC", "1", 1);
189            break;
190        case 'b': /* --bytes */
191            opts->bytes = optarg;
192            break;
193        case 'B': /* --max-bytes */
194            opts->maxbytes = atoi(optarg);
195            break;
196#if defined HAVE_REGEX_H
197        case 'c': /* --cmdline */
198            cmdline = 1;
199            break;
200#endif
201        case 'C': /* --max-crashes */
202            opts->maxcrashes = atoi(optarg);
203            if(opts->maxcrashes <= 0)
204                opts->maxcrashes = 0;
205            break;
206        case 'd': /* --debug */
207            setenv("ZZUF_DEBUG", DEBUG_FILENO_STR, 1);
208            break;
209        case 'D': /* --delay */
210            opts->delay = (int64_t)(atof(optarg) * 1000000.0);
211            break;
212#if defined HAVE_REGEX_H
213        case 'E': /* --exclude */
214            exclude = merge_regex(exclude, optarg);
215            if(!exclude)
216            {
217                printf("%s: invalid regex -- `%s'\n", argv[0], optarg);
218                _zz_opts_fini(opts);
219                return EXIT_FAILURE;
220            }
221            break;
222#endif
223        case 'f': /* --fuzzing */
224            opts->fuzzing = optarg;
225            break;
226        case 'F': /* --max-forks */
227            opts->maxchild = atoi(optarg) > 1 ? atoi(optarg) : 1;
228            break;
229        case 'i': /* --stdin */
230            setenv("ZZUF_STDIN", "1", 1);
231            break;
232#if defined HAVE_REGEX_H
233        case 'I': /* --include */
234            include = merge_regex(include, optarg);
235            if(!include)
236            {
237                printf("%s: invalid regex -- `%s'\n", argv[0], optarg);
238                _zz_opts_fini(opts);
239                return EXIT_FAILURE;
240            }
241            break;
242#endif
243        case 'm': /* --md5 */
244            opts->md5 = 1;
245            break;
246#if defined HAVE_SETRLIMIT
247        case 'M': /* --max-memory */
248            setenv("ZZUF_MEMORY", "1", 1);
249            opts->maxmem = atoi(optarg);
250            break;
251#endif
252        case 'n': /* --network */
253            setenv("ZZUF_NETWORK", "1", 1);
254            break;
255        case 'P': /* --protect */
256            opts->protect = optarg;
257            break;
258        case 'q': /* --quiet */
259            opts->quiet = 1;
260            break;
261        case 'r': /* --ratio */
262            tmp = strchr(optarg, ':');
263            opts->minratio = atof(optarg);
264            opts->maxratio = tmp ? atof(tmp + 1) : opts->minratio;
265            break;
266        case 'R': /* --refuse */
267            opts->refuse = optarg;
268            break;
269        case 's': /* --seed */
270            tmp = strchr(optarg, ':');
271            opts->seed = atol(optarg);
272            opts->endseed = tmp ? (uint32_t)atoi(tmp + 1) : opts->seed + 1;
273            break;
274        case 'S': /* --signal */
275            setenv("ZZUF_SIGNAL", "1", 1);
276            break;
277        case 'T': /* --max-time */
278            opts->maxtime = (int64_t)(atof(optarg) * 1000000.0);
279            break;
280        case 'x': /* --check-exit */
281            opts->checkexit = 1;
282            break;
283        case 'v': /* --verbose */
284            opts->verbose = 1;
285            break;
286        case 'h': /* --help */
287            usage();
288            _zz_opts_fini(opts);
289            return 0;
290        case 'V': /* --version */
291            version();
292            _zz_opts_fini(opts);
293            return 0;
294        default:
295            printf("%s: invalid option -- %c\n", argv[0], c);
296            printf(MOREINFO, argv[0]);
297            _zz_opts_fini(opts);
298            return EXIT_FAILURE;
299        }
300    }
301#else
302#   define MOREINFO "Usage: %s message...\n"
303    int optind = 1;
304#endif
305
306    _zz_setratio(opts->minratio, opts->maxratio);
307    _zz_setseed(opts->seed);
308
309    /* If asked to read from the standard input */
310    if(optind >= argc)
311    {
312        if(opts->endseed != opts->seed + 1)
313        {
314            printf("%s: seed ranges are incompatible with stdin fuzzing\n",
315                   argv[0]);
316            printf(MOREINFO, argv[0]);
317            _zz_opts_fini(opts);
318            return EXIT_FAILURE;
319        }
320
321        loop_stdin(opts);
322
323        _zz_opts_fini(opts);
324        return EXIT_SUCCESS;
325    }
326
327    /* If asked to launch programs */
328#if defined HAVE_REGEX_H
329    if(cmdline)
330    {
331        int dashdash = 0;
332
333        for(i = optind + 1; i < argc; i++)
334        {
335            if(dashdash)
336                include = merge_file(include, argv[i]);
337            else if(!strcmp("--", argv[i]))
338                dashdash = 1;
339            else if(argv[i][0] != '-')
340                include = merge_file(include, argv[i]);
341        }
342    }
343
344    if(include)
345        setenv("ZZUF_INCLUDE", include, 1);
346    if(exclude)
347        setenv("ZZUF_EXCLUDE", exclude, 1);
348#endif
349
350    if(opts->fuzzing)
351        setenv("ZZUF_FUZZING", opts->fuzzing, 1);
352    if(opts->bytes)
353        setenv("ZZUF_BYTES", opts->bytes, 1);
354    if(opts->protect)
355        setenv("ZZUF_PROTECT", opts->protect, 1);
356    if(opts->refuse)
357        setenv("ZZUF_REFUSE", opts->refuse, 1);
358
359    /* Allocate memory for children handling */
360    opts->child = malloc(opts->maxchild * sizeof(struct child));
361    for(i = 0; i < opts->maxchild; i++)
362        opts->child[i].status = STATUS_FREE;
363    opts->nchild = 0;
364
365    /* Create new argv */
366    opts->oldargv = argv;
367    opts->newargv = malloc((argc - optind + 1) * sizeof(char *));
368    memcpy(opts->newargv, argv + optind, (argc - optind) * sizeof(char *));
369    opts->newargv[argc - optind] = (char *)NULL;
370
371    /* Main loop */
372    while(opts->nchild || opts->seed < opts->endseed)
373    {
374        /* Spawn new children, if necessary */
375        spawn_children(opts);
376
377        /* Cleanup dead or dying children */
378        clean_children(opts);
379
380        /* Read data from children */
381        read_children(opts);
382
383        if(opts->maxcrashes && opts->crashes >= opts->maxcrashes
384            && opts->nchild == 0)
385            break;
386    }
387
388    /* Clean up */
389    _zz_opts_fini(opts);
390
391    return opts->crashes ? EXIT_FAILURE : EXIT_SUCCESS;   
392}
393
394static void loop_stdin(struct opts *opts)
395{
396    uint8_t md5sum[16];
397    struct md5 *ctx = NULL;
398
399    if(opts->md5)
400        ctx = _zz_md5_init();
401
402    if(opts->fuzzing)
403        _zz_fuzzing(opts->fuzzing);
404    if(opts->bytes)
405        _zz_bytes(opts->bytes);
406    if(opts->protect)
407        _zz_protect(opts->protect);
408    if(opts->refuse)
409        _zz_refuse(opts->refuse);
410
411    _zz_fd_init();
412    _zz_register(0);
413
414    for(;;)
415    {
416        uint8_t buf[BUFSIZ];
417        int ret, off = 0, nw = 0;
418
419        ret = read(0, buf, BUFSIZ);
420        if(ret <= 0)
421            break;
422
423        _zz_fuzz(0, buf, ret);
424        _zz_addpos(0, ret);
425
426        if(opts->md5)
427            _zz_md5_add(ctx, buf, ret);
428        else while(ret)
429        {
430            if((nw = write(1, buf + off, (size_t)ret)) < 0)
431                break;
432            ret -= nw;
433            off += nw;
434        }
435    }
436
437    if(opts->md5)
438    {
439        _zz_md5_fini(md5sum, ctx);
440        finfo(stdout, opts, opts->seed);
441        fprintf(stdout, "%.02x%.02x%.02x%.02x%.02x%.02x%.02x%.02x%.02x%.02x"
442                "%.02x%.02x%.02x%.02x%.02x%.02x\n", md5sum[0], md5sum[1],
443                md5sum[2], md5sum[3], md5sum[4], md5sum[5], md5sum[6],
444                md5sum[7], md5sum[8], md5sum[9], md5sum[10], md5sum[11],
445                md5sum[12], md5sum[13], md5sum[14], md5sum[15]);
446        fflush(stdout);
447    }
448
449    _zz_unregister(0);
450    _zz_fd_fini();
451}
452
453static void finfo(FILE *fp, struct opts *opts, uint32_t seed)
454{
455    if(opts->minratio == opts->maxratio)
456        fprintf(fp, "zzuf[s=%i,r=%g]: ", seed, opts->minratio);
457    else
458        fprintf(fp, "zzuf[s=%i,r=%g:%g]: ", seed,
459                opts->minratio, opts->maxratio);
460}
461
462#if defined HAVE_REGEX_H
463static char *merge_file(char *regex, char *file)
464{
465    char *newfile = malloc(5 + 2 * strlen(file) + 1 + 1), *tmp = newfile;
466
467    *tmp++ = '(';
468    *tmp++ = '^';
469    *tmp++ = '|';
470    *tmp++ = '/';
471    *tmp++ = ')';
472    while(*file)
473    {
474        if(strchr("^.[$()|*+?{\\", *file))
475            *tmp++ = '\\';
476        *tmp++ = *file++;
477    }
478    *tmp++ = '$';
479    *tmp++ = '\0';
480
481    tmp = merge_regex(regex, newfile);
482    free(newfile);
483    return tmp;
484}
485
486static char *merge_regex(char *regex, char *string)
487{
488    regex_t optre;
489
490    if(regex)
491    {
492        regex = realloc(regex, strlen(regex) + strlen(string) + 1 + 1);
493        sprintf(regex + strlen(regex) - 1, "|%s)", string);
494    }
495    else
496    {
497        regex = malloc(1 + strlen(string) + 1 + 1);
498        sprintf(regex, "(%s)", string);
499    }
500
501    if(regcomp(&optre, regex, REG_EXTENDED) != 0)
502    {
503        free(regex);
504        return NULL;
505    }
506    regfree(&optre);
507
508    return regex;
509}
510#endif
511
512static void spawn_children(struct opts *opts)
513{
514    static int const files[] = { DEBUG_FILENO, STDERR_FILENO, STDOUT_FILENO };
515    char buf[64];
516    int fd[3][2];
517    int64_t now = _zz_time();
518    pid_t pid;
519    int i, j;
520
521    if(opts->nchild == opts->maxchild)
522        return; /* no slot */
523
524    if(opts->seed == opts->endseed)
525        return; /* job finished */
526
527    if(opts->maxcrashes && opts->crashes >= opts->maxcrashes)
528        return; /* all jobs crashed */
529
530    if(opts->delay > 0 && opts->lastlaunch + opts->delay > now)
531        return; /* too early */
532
533    /* Find the empty slot */
534    for(i = 0; i < opts->maxchild; i++)
535        if(opts->child[i].status == STATUS_FREE)
536            break;
537
538    /* Prepare communication pipe */
539    for(j = 0; j < 3; j++)
540    {
541        int ret;
542#if defined HAVE_PIPE
543        ret = pipe(fd[j]);
544#elif defined HAVE__PIPE
545        ret = _pipe(fd[j], 256, _O_BINARY);
546#endif
547        if(ret < 0)
548        {
549            perror("pipe");
550            return;
551        }
552    }
553
554#if defined HAVE_FORK
555    /* Fork and launch child */
556    pid = fork();
557    if(pid < -1)
558    {
559        perror("fork");
560        return;
561    }
562#else
563    pid = 0;
564#endif
565
566    if(pid == 0)
567    {
568#if defined HAVE_SETRLIMIT
569        if(opts->maxmem >= 0)
570        {
571            struct rlimit rlim;
572            rlim.rlim_cur = opts->maxmem * 1000000;
573            rlim.rlim_max = opts->maxmem * 1000000;
574            setrlimit(RLIMIT_AS, &rlim);
575        }
576#endif
577
578#if defined HAVE_FORK
579        /* We loop in reverse order so that files[0] is done last,
580         * just in case one of the other dup2()ed fds had the value */
581        for(j = 3; j--; )
582        {
583            close(fd[j][0]);
584            if(fd[j][1] != files[j])
585            {
586                dup2(fd[j][1], files[j]);
587                close(fd[j][1]);
588            }
589        }
590#endif
591
592        /* Set environment variables */
593        sprintf(buf, "%i", opts->seed);
594        setenv("ZZUF_SEED", buf, 1);
595        sprintf(buf, "%g", opts->minratio);
596        setenv("ZZUF_MINRATIO", buf, 1);
597        sprintf(buf, "%g", opts->maxratio);
598        setenv("ZZUF_MAXRATIO", buf, 1);
599
600#if defined HAVE_FORK
601        if(run_process(opts->oldargv[0], opts->newargv) < 0)
602            exit(EXIT_FAILURE);
603        exit(EXIT_SUCCESS);
604#else
605        if(run_process(opts->oldargv[0], opts->newargv) < 0)
606            return;
607#endif
608    }
609
610    /* We’re the parent, acknowledge spawn */
611    opts->child[i].date = now;
612    opts->child[i].pid = pid;
613    for(j = 0; j < 3; j++)
614    {
615        close(fd[j][1]);
616        opts->child[i].fd[j] = fd[j][0];
617    }
618    opts->child[i].bytes = 0;
619    opts->child[i].seed = opts->seed;
620    opts->child[i].ratio = _zz_getratio();
621    opts->child[i].status = STATUS_RUNNING;
622    if(opts->md5)
623        opts->child[i].ctx = _zz_md5_init();
624
625    if(opts->verbose)
626    {
627        finfo(stderr, opts, opts->child[i].seed);
628        fprintf(stderr, "launched %s\n", opts->newargv[0]);
629    }
630
631    opts->lastlaunch = now;
632    opts->nchild++;
633    opts->seed++;
634
635    _zz_setseed(opts->seed);
636}
637
638static void clean_children(struct opts *opts)
639{
640#if defined HAVE_KILL
641    int64_t now = _zz_time();
642#endif
643    int i, j;
644
645#if defined HAVE_KILL
646    /* Terminate children if necessary */
647    for(i = 0; i < opts->maxchild; i++)
648    {
649        if(opts->child[i].status == STATUS_RUNNING
650            && opts->maxbytes >= 0
651            && opts->child[i].bytes > opts->maxbytes)
652        {
653            if(opts->verbose)
654            {
655                finfo(stderr, opts, opts->child[i].seed);
656                fprintf(stderr, "data output exceeded, sending SIGTERM\n");
657            }
658            kill(opts->child[i].pid, SIGTERM);
659            opts->child[i].date = now;
660            opts->child[i].status = STATUS_SIGTERM;
661        }
662
663        if(opts->child[i].status == STATUS_RUNNING
664            && opts->maxtime >= 0
665            && now > opts->child[i].date + opts->maxtime)
666        {
667            if(opts->verbose)
668            {
669                finfo(stderr, opts, opts->child[i].seed);
670                fprintf(stderr, "running time exceeded, sending SIGTERM\n");
671            }
672            kill(opts->child[i].pid, SIGTERM);
673            opts->child[i].date = now;
674            opts->child[i].status = STATUS_SIGTERM;
675        }
676    }
677
678    /* Kill children if necessary (still there after 2 seconds) */
679    for(i = 0; i < opts->maxchild; i++)
680    {
681        if(opts->child[i].status == STATUS_SIGTERM
682            && now > opts->child[i].date + 2000000)
683        {
684            if(opts->verbose)
685            {
686                finfo(stderr, opts, opts->child[i].seed);
687                fprintf(stderr, "not responding, sending SIGKILL\n");
688            }
689            kill(opts->child[i].pid, SIGKILL);
690            opts->child[i].status = STATUS_SIGKILL;
691        }
692    }
693#endif
694
695    /* Collect dead children */
696    for(i = 0; i < opts->maxchild; i++)
697    {
698        uint8_t md5sum[16];
699#if defined HAVE_WAITPID
700        int status;
701        pid_t pid;
702#endif
703
704        if(opts->child[i].status != STATUS_SIGKILL
705            && opts->child[i].status != STATUS_SIGTERM
706            && opts->child[i].status != STATUS_EOF)
707            continue;
708
709#if defined HAVE_WAITPID
710        pid = waitpid(opts->child[i].pid, &status, WNOHANG);
711        if(pid <= 0)
712            continue;
713
714        if(opts->checkexit && WIFEXITED(status) && WEXITSTATUS(status))
715        {
716            finfo(stderr, opts, opts->child[i].seed);
717            fprintf(stderr, "exit %i\n", WEXITSTATUS(status));
718            opts->crashes++;
719        }
720        else if(WIFSIGNALED(status)
721                 && !(WTERMSIG(status) == SIGTERM
722                       && opts->child[i].status == STATUS_SIGTERM))
723        {
724            finfo(stderr, opts, opts->child[i].seed);
725            fprintf(stderr, "signal %i%s%s\n",
726                    WTERMSIG(status), sig2str(WTERMSIG(status)),
727                      (WTERMSIG(status) == SIGKILL && opts->maxmem >= 0) ?
728                      " (memory exceeded?)" : "");
729            opts->crashes++;
730        }
731#endif
732
733        for(j = 0; j < 3; j++)
734            if(opts->child[i].fd[j] >= 0)
735                close(opts->child[i].fd[j]);
736
737        if(opts->md5)
738        {
739            _zz_md5_fini(md5sum, opts->child[i].ctx);
740            finfo(stdout, opts, opts->child[i].seed);
741            fprintf(stdout, "%.02x%.02x%.02x%.02x%.02x%.02x%.02x%.02x%.02x"
742                    "%.02x%.02x%.02x%.02x%.02x%.02x%.02x\n", md5sum[0],
743                    md5sum[1], md5sum[2], md5sum[3], md5sum[4], md5sum[5],
744                    md5sum[6], md5sum[7], md5sum[8], md5sum[9], md5sum[10],
745                    md5sum[11], md5sum[12], md5sum[13], md5sum[14], md5sum[15]);
746            fflush(stdout);
747        }
748        opts->child[i].status = STATUS_FREE;
749        opts->nchild--;
750    }
751}
752
753static void read_children(struct opts *opts)
754{
755    struct timeval tv;
756    fd_set fdset;
757    int i, j, ret, maxfd = 0;
758
759    /* Read data from all sockets */
760    FD_ZERO(&fdset);
761    for(i = 0; i < opts->maxchild; i++)
762    {
763        if(opts->child[i].status != STATUS_RUNNING)
764            continue;
765
766        for(j = 0; j < 3; j++)
767            ZZUF_FD_SET(opts->child[i].fd[j], &fdset, maxfd);
768    }
769    tv.tv_sec = 0;
770    tv.tv_usec = 1000;
771
772    ret = select(maxfd + 1, &fdset, NULL, NULL, &tv);
773    if(ret < 0 && errno)
774        perror("select");
775    if(ret <= 0)
776        return;
777
778    /* XXX: cute (i, j) iterating hack */
779    for(i = 0, j = 0; i < opts->maxchild; i += (j == 2), j = (j + 1) % 3)
780    {
781        uint8_t buf[BUFSIZ];
782
783        if(opts->child[i].status != STATUS_RUNNING)
784            continue;
785
786        if(!ZZUF_FD_ISSET(opts->child[i].fd[j], &fdset))
787            continue;
788
789        ret = read(opts->child[i].fd[j], buf, BUFSIZ - 1);
790        if(ret > 0)
791        {
792            /* We got data */
793            if(j != 0)
794                opts->child[i].bytes += ret;
795
796            if(opts->md5 && j == 2)
797                _zz_md5_add(opts->child[i].ctx, buf, ret);
798            else if(!opts->quiet || j == 0)
799                write((j < 2) ? STDERR_FILENO : STDOUT_FILENO, buf, ret);
800        }
801        else if(ret == 0)
802        {
803            /* End of file reached */
804            close(opts->child[i].fd[j]);
805            opts->child[i].fd[j] = -1;
806
807            if(opts->child[i].fd[0] == -1
808                && opts->child[i].fd[1] == -1
809                && opts->child[i].fd[2] == -1)
810                opts->child[i].status = STATUS_EOF;
811        }
812    }
813}
814
815#if !defined HAVE_SETENV
816static void setenv(char const *name, char const *value, int overwrite)
817{
818    char *str;
819
820    if(!overwrite && getenv(name))
821        return;
822
823    str = malloc(strlen(name) + 1 + strlen(value) + 1);
824    sprintf(str, "%s=%s", name, value);
825    putenv(str);
826}
827#endif
828
829#if defined HAVE_WAITPID
830static char const *sig2str(int signum)
831{
832    switch(signum)
833    {
834        case SIGABRT:  return " (SIGABRT)";
835        case SIGFPE:   return " (SIGFPE)";
836        case SIGILL:   return " (SIGILL)";
837#ifdef SIGQUIT
838        case SIGQUIT:  return " (SIGQUIT)";
839#endif
840        case SIGSEGV:  return " (SIGSEGV)";
841#ifdef SIGTRAP
842        case SIGTRAP:  return " (SIGTRAP)";
843#endif
844#ifdef SIGSYS
845        case SIGSYS:   return " (SIGSYS)";
846#endif
847#ifdef SIGEMT
848        case SIGEMT:   return " (SIGEMT)";
849#endif
850#ifdef SIGBUS
851        case SIGBUS:   return " (SIGBUS)";
852#endif
853#ifdef SIGXCPU
854        case SIGXCPU:  return " (SIGXCPU)";
855#endif
856#ifdef SIGXFSZ
857        case SIGXFSZ:  return " (SIGXFSZ)";
858#endif
859    }
860
861    return "";
862}
863#endif
864
865static int run_process(char const *zzuf_exe, char *argv[])
866{
867#if defined HAVE_FORK
868    char *libpath, *tmp;
869    int ret, len = strlen(zzuf_exe);
870#   if defined __APPLE__
871#       define FILENAME "libzzuf.dylib"
872#       define EXTRAINFO ""
873#       define PRELOAD "DYLD_INSERT_LIBRARIES"
874    setenv("DYLD_FORCE_FLAT_NAMESPACE", "1", 1);
875#   elif defined __osf__
876#       define FILENAME "libzzuf.so"
877#       define EXTRAINFO ":DEFAULT"
878#       define PRELOAD "_RLD_LIST"
879#   else
880#       define FILENAME "libzzuf.so"
881#       define EXTRAINFO ""
882#       define PRELOAD "LD_PRELOAD"
883#   endif
884
885    libpath = malloc(len + strlen("/.libs/" FILENAME EXTRAINFO) + 1);
886    strcpy(libpath, zzuf_exe);
887
888    tmp = strrchr(libpath, '/');
889    strcpy(tmp ? tmp + 1 : libpath, ".libs/" FILENAME);
890    ret = access(libpath, R_OK);
891
892    strcpy(tmp ? tmp + 1 : libpath, ".libs/" FILENAME EXTRAINFO);
893    if(ret == 0)
894        setenv(PRELOAD, libpath, 1);
895    else
896        setenv(PRELOAD, LIBDIR "/" FILENAME EXTRAINFO, 1);
897    free(libpath);
898
899    if(execvp(argv[0], argv))
900    {
901        perror(argv[0]);
902        return -1;
903    }
904
905    return 0;
906#elif HAVE_WINDOWS_H
907    PROCESS_INFORMATION pinfo;
908    STARTUPINFO sinfo;
909    void *epaddr;
910    int ret;
911
912    /* Get entry point */
913    epaddr = get_entry(argv[0]);
914    if(!epaddr)
915        return -1;
916   
917    memset(&sinfo, 0, sizeof(sinfo));
918    sinfo.cb = sizeof(sinfo);
919    ret = CreateProcess(NULL, argv[0], NULL, NULL, FALSE,
920                        CREATE_SUSPENDED, NULL, NULL, &sinfo, &pinfo);
921    if(!ret)
922        return -1;
923
924    /* Insert the replacement code */
925    ret = dll_inject(pinfo.hProcess, epaddr);
926    if(ret < 0)
927    {
928        TerminateProcess(pinfo.hProcess, -1);
929        return -1;
930    }
931
932    ret = ResumeThread(pinfo.hThread);
933    if(ret < 0)
934    {
935        TerminateProcess(pinfo.hProcess, -1);
936        return -1;
937    }
938
939    return 0;
940#else
941    return -1;
942#endif
943}
944
945#if defined HAVE_WINDOWS_H
946static int dll_inject(void *process, void *epaddr)
947{
948    uint8_t old_ep[7];
949    uint8_t new_ep[] = "\xb8<01>\xff\xe0";
950    uint8_t loader[] = "libzzuf.dll\0<0000c>\xb8<14>\x50\xb8<1a>\xff\xd0"
951                       "\xb8\0\0\0\0\x50\xb8\x07\x00\x00\x00\x50\xb8<2d>"
952                       "\x50\xb8<33>\x50\xb8<39>\xff\xd0\x50\xb8<41>\xff"
953                       "\xd0\xb8<48>\xff\xe0";
954    void *lib;
955    uint8_t *loaderaddr;
956    DWORD tmp;
957
958    /* Save the old entry-point code */
959    ReadProcessMemory(process, epaddr, old_ep, 7, &tmp);
960    if(tmp != 7)
961        return -1;
962
963    loaderaddr = VirtualAllocEx(process, NULL, 78, MEM_COMMIT,
964                                PAGE_EXECUTE_READWRITE);
965    if(!loaderaddr)
966        return -1;
967
968    addcpy(new_ep + 0x01, loaderaddr + 0x0c + 7);
969    WriteProcessMemory(process, epaddr, new_ep, 7, &tmp);
970    if(tmp != 7)
971        return -1;
972
973    lib = LoadLibrary("kernel32.dll");
974    if(!lib)
975        return -1;
976
977    memcpy(loader + 0x0c, old_ep, 7);
978    addcpy(loader + 0x14, loaderaddr + 0x00); /* offset for dll string */
979    addcpy(loader + 0x1a, GetProcAddress(lib, "LoadLibraryA"));
980    addcpy(loader + 0x2d, loaderaddr + 0x0c);
981    addcpy(loader + 0x33, epaddr);
982    addcpy(loader + 0x39, GetProcAddress(lib, "GetCurrentProcess"));
983    addcpy(loader + 0x41, GetProcAddress(lib, "WriteProcessMemory"));
984    addcpy(loader + 0x48, epaddr);
985    FreeLibrary(lib);
986
987    WriteProcessMemory(process, loaderaddr, loader, 78, &tmp);
988    if(tmp != 78)
989        return -1;
990
991    return 0;
992}
993
994static void *get_entry(char const *name)
995{
996    PIMAGE_DOS_HEADER dos;
997    PIMAGE_NT_HEADERS nt;
998    void *file, *map, *base;
999
1000    file = CreateFile(name, GENERIC_READ, FILE_SHARE_READ,
1001                      NULL, OPEN_EXISTING, 0, NULL);
1002    if(file == INVALID_HANDLE_VALUE)
1003        return NULL;
1004
1005    map = CreateFileMapping(file, NULL, PAGE_READONLY, 0, 0, NULL);
1006    if(!map)
1007    {
1008        CloseHandle(file);
1009        return NULL;
1010    }
1011
1012    base = MapViewOfFile(map, FILE_MAP_READ, 0, 0, 0);
1013    if(!base)
1014    {
1015        CloseHandle(map);
1016        CloseHandle(file);
1017        return NULL;
1018    }
1019
1020    /* Sanity checks */
1021    dos = (PIMAGE_DOS_HEADER)base;
1022    nt = (PIMAGE_NT_HEADERS)((char *)base + dos->e_lfanew);
1023    if(dos->e_magic != IMAGE_DOS_SIGNATURE
1024      || nt->Signature != IMAGE_NT_SIGNATURE
1025      || nt->FileHeader.Machine != IMAGE_FILE_MACHINE_I386
1026      || nt->OptionalHeader.Magic != 0x10b /* IMAGE_NT_OPTIONAL_HDR32_MAGIC */)
1027    {
1028        UnmapViewOfFile(base);
1029        CloseHandle(map);
1030        CloseHandle(file);
1031        return NULL;
1032    }
1033
1034    return (char *)nt->OptionalHeader.ImageBase +
1035                           nt->OptionalHeader.AddressOfEntryPoint;
1036}
1037#endif
1038
1039static void version(void)
1040{
1041    printf("zzuf %s\n", VERSION);
1042    printf("Copyright (C) 2002, 2007 Sam Hocevar <sam@zoy.org>\n");
1043    printf("This is free software.  You may redistribute copies of it under the\n");
1044    printf("terms of the Do What The Fuck You Want To Public License, Version 2\n");
1045    printf("<http://sam.zoy.org/wtfpl/>.\n");
1046    printf("There is NO WARRANTY, to the extent permitted by law.\n");
1047    printf("\n");
1048    printf("Written by Sam Hocevar. Report bugs to <sam@zoy.org>.\n");
1049}
1050
1051#if defined HAVE_GETOPT_H
1052static void usage(void)
1053{
1054#if defined HAVE_REGEX_H
1055    printf("Usage: zzuf [-AcdimnqSvx] [-s seed|-s start:stop] [-r ratio|-r min:max]\n");
1056#else
1057    printf("Usage: zzuf [-AdimnqSvx] [-s seed|-s start:stop] [-r ratio|-r min:max]\n");
1058#endif
1059    printf("              [-f fuzzing] [-D delay] [-F forks] [-C crashes] [-B bytes]\n");
1060    printf("              [-T seconds] [-M bytes] [-b ranges] [-P protect] [-R refuse]\n");
1061#if defined HAVE_REGEX_H
1062    printf("              [-I include] [-E exclude] [PROGRAM [--] [ARGS]...]\n");
1063#else
1064    printf("              [PROGRAM [--] [ARGS]...]\n");
1065#endif
1066#   if defined HAVE_GETOPT_LONG
1067    printf("       zzuf -h | --help\n");
1068    printf("       zzuf -V | --version\n");
1069#   else
1070    printf("       zzuf -h\n");
1071    printf("       zzuf -V\n");
1072#   endif
1073    printf("Run PROGRAM with optional arguments ARGS and fuzz its input.\n");
1074    printf("\n");
1075    printf("Mandatory arguments to long options are mandatory for short options too.\n");
1076#   if defined HAVE_GETOPT_LONG
1077    printf("  -A, --autoinc             increment seed each time a new file is opened\n");
1078    printf("  -b, --bytes <ranges>      only fuzz bytes at offsets within <ranges>\n");
1079    printf("  -B, --max-bytes <n>       kill children that output more than <n> bytes\n");
1080#if defined HAVE_REGEX_H
1081    printf("  -c, --cmdline             only fuzz files specified in the command line\n");
1082#endif
1083    printf("  -C, --max-crashes <n>     stop after <n> children have crashed (default 1)\n");
1084    printf("  -d, --debug               print debug messages\n");
1085    printf("  -D, --delay               delay between forks\n");
1086#if defined HAVE_REGEX_H
1087    printf("  -E, --exclude <regex>     do not fuzz files matching <regex>\n");
1088#endif
1089    printf("  -f, --fuzzing <mode>      use fuzzing mode <mode> ([xor] set unset)\n");
1090    printf("  -F, --max-forks <n>       number of concurrent children (default 1)\n");
1091    printf("  -i, --stdin               fuzz standard input\n");
1092#if defined HAVE_REGEX_H
1093    printf("  -I, --include <regex>     only fuzz files matching <regex>\n");
1094#endif
1095    printf("  -m, --md5                 compute the output's MD5 hash\n");
1096#if defined HAVE_SETRLIMIT
1097    printf("  -M, --max-memory <n>      maximum child virtual memory size in MB\n");
1098#endif
1099    printf("  -n, --network             fuzz network input\n");
1100    printf("  -P, --protect <list>      protect bytes and characters in <list>\n");
1101    printf("  -q, --quiet               do not print children's messages\n");
1102    printf("  -r, --ratio <ratio>       bit fuzzing ratio (default %g)\n", DEFAULT_RATIO);
1103    printf("      --ratio <start:stop>  specify a ratio range\n");
1104    printf("  -R, --refuse <list>       refuse bytes and characters in <list>\n");
1105    printf("  -s, --seed <seed>         random seed (default %i)\n", DEFAULT_SEED);
1106    printf("      --seed <start:stop>   specify a seed range\n");
1107    printf("  -S, --signal              prevent children from diverting crashing signals\n");
1108    printf("  -T, --max-time <n>        kill children that run for more than <n> seconds\n");
1109    printf("  -v, --verbose             print information during the run\n");
1110    printf("  -x, --check-exit          report processes that exit with a non-zero status\n");
1111    printf("  -h, --help                display this help and exit\n");
1112    printf("  -V, --version             output version information and exit\n");
1113#   else
1114    printf("  -A               increment seed each time a new file is opened\n");
1115    printf("  -b <ranges>      only fuzz bytes at offsets within <ranges>\n");
1116    printf("  -B <n>           kill children that output more than <n> bytes\n");
1117#if defined HAVE_REGEX_H
1118    printf("  -c               only fuzz files specified in the command line\n");
1119#endif
1120    printf("  -C <n>           stop after <n> children have crashed (default 1)\n");
1121    printf("  -d               print debug messages\n");
1122    printf("  -D               delay between forks\n");
1123#if defined HAVE_REGEX_H
1124    printf("  -E <regex>       do not fuzz files matching <regex>\n");
1125#endif
1126    printf("  -f <mode>        use fuzzing mode <mode>\n");
1127    printf("  -F <n>           number of concurrent forks (default 1)\n");
1128    printf("  -i               fuzz standard input\n");
1129#if defined HAVE_REGEX_H
1130    printf("  -I <regex>       only fuzz files matching <regex>\n");
1131#endif
1132    printf("  -m               compute the output's MD5 hash\n");
1133#if defined HAVE_SETRLIMIT
1134    printf("  -M               maximum child virtual memory size in MB\n");
1135#endif
1136    printf("  -n               fuzz network input\n");
1137    printf("  -P <list>        protect bytes and characters in <list>\n");
1138    printf("  -q               do not print the fuzzed application's messages\n");
1139    printf("  -r <ratio>       bit fuzzing ratio (default %g)\n", DEFAULT_RATIO);
1140    printf("     <start:stop>  specify a ratio range\n");
1141    printf("  -R <list>        refuse bytes and characters in <list>\n");
1142    printf("  -s <seed>        random seed (default %i)\n", DEFAULT_SEED);
1143    printf("     <start:stop>  specify a seed range\n");
1144    printf("  -S               prevent children from diverting crashing signals\n");
1145    printf("  -T <n>           kill children that run for more than <n> seconds\n");
1146    printf("  -v               print information during the run\n");
1147    printf("  -x               report processes that exit with a non-zero status\n");
1148    printf("  -h               display this help and exit\n");
1149    printf("  -V               output version information and exit\n");
1150#   endif
1151    printf("\n");
1152    printf("Written by Sam Hocevar. Report bugs to <sam@zoy.org>.\n");
1153}
1154#endif
1155
Note: See TracBrowser for help on using the repository browser.