blob: b81bab136be454d9498b5a795bd538b87c2d78c4 (
plain) (
blame)
| 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
29
30
31
32
33
34
35
36
37
38
39
 | #include <stdio.h>
#include <errno.h>
#include <dlfcn.h>
#include <stdlib.h>
int main(int argc, char **argv) {
  if (!getenv("TEST_DLOPEN_TARGET")) {
    fprintf(stderr, "Error: TEST_DLOPEN_TARGET not set!\n");
    return 1;
  }
  void *lib = dlopen(getenv("TEST_DLOPEN_TARGET"), RTLD_LAZY);
  if (!lib) {
    perror(dlerror());
    return 2;
  }
  int (*func)(int, char **) = dlsym(lib, "main_exported");
  if (!func) {
    fprintf(stderr, "Error: main_exported not found!\n");
    return 3;
  }
  // must use deferred forkserver as otherwise afl++ instrumentation aborts
  // because all dlopen() of instrumented libs must be before the forkserver
  __AFL_INIT();
  fprintf(stderr, "Running main_exported\n");
  return func(argc, argv);
}
 |