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

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