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

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