utils/argv_fuzzing/argv_fuzz_persistent_demo.c


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28

#include <stdio.h>
#include <string.h>
#include "argv-fuzz-inl.h"

__AFL_FUZZ_INIT();

int main(int argc, char **argv) {
#ifdef __AFL_HAVE_MANUAL_CONTROL
  __AFL_INIT();
#endif
  unsigned char *buf = __AFL_FUZZ_TESTCASE_BUF;

  while (__AFL_LOOP(100000)) {
    int len = __AFL_FUZZ_TESTCASE_LEN;

    if (len < 8) continue;

    AFL_INIT_ARGV_PERSISTENT(buf);

    if (argc > 1 && strcmp(argv[1], "XYZ") == 0) {
      if (strcmp(argv[2], "TEST2") == 0) { abort(); }
    } else {
      printf("Bad number of arguments!\n");
    }
  }

  return 0;
}