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

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