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

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