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);
}
|