forked from OSchip/llvm-project
70 lines
1.6 KiB
C
70 lines
1.6 KiB
C
/* Checks that BOLT correctly handles instrumentation shared libraries
|
|
* with further optimization.
|
|
*/
|
|
#include <dlfcn.h>
|
|
#include <stdio.h>
|
|
|
|
#ifdef LIB
|
|
int foo(int x) { return x + 1; }
|
|
|
|
int fib(int x) {
|
|
if (x < 2)
|
|
return x;
|
|
return fib(x - 1) + fib(x - 2);
|
|
}
|
|
|
|
int bar(int x) { return x - 1; }
|
|
#endif
|
|
|
|
#ifndef LIB
|
|
int main(int argc, char **argv) {
|
|
int (*fib_func)(int);
|
|
char *libname;
|
|
void *hndl;
|
|
int val;
|
|
if (argc < 2)
|
|
return -1;
|
|
/*
|
|
* Expected library name as input to switch
|
|
* between original and instrumented file
|
|
*/
|
|
libname = argv[1];
|
|
hndl = dlopen(libname, RTLD_LAZY);
|
|
if (!hndl) {
|
|
printf("library load failed\n");
|
|
return -1;
|
|
}
|
|
fib_func = dlsym(hndl, "fib");
|
|
if (!fib_func) {
|
|
printf("fib_func load failed\n");
|
|
return -1;
|
|
}
|
|
val = fib_func(argc);
|
|
dlclose(hndl);
|
|
printf("fib(%d) = %d\n", argc, val);
|
|
return 0;
|
|
}
|
|
#endif
|
|
|
|
/*
|
|
REQUIRES: system-linux,bolt-runtime
|
|
|
|
RUN: %clang %cflags %s -o %t.so -Wl,-q -fpie -fPIC -shared -DLIB
|
|
RUN: %clang %cflags %s -o %t.exe -Wl,-q -ldl
|
|
|
|
RUN: llvm-bolt %t.so --instrument --instrumentation-file=%t.so.fdata \
|
|
RUN: -o %t.so.instrumented
|
|
|
|
# Program with instrumented shared library needs to finish returning zero
|
|
RUN: %t.exe %t.so.instrumented 1 2 | FileCheck %s -check-prefix=CHECK-OUTPUT
|
|
RUN: test -f %t.so.fdata
|
|
|
|
# Test that the instrumented data makes sense
|
|
RUN: llvm-bolt %t.so -o %t.so.bolted --data %t.so.fdata \
|
|
RUN: --reorder-blocks=ext-tsp --reorder-functions=hfsort+
|
|
|
|
RUN: %t.exe %t.so.bolted 1 2 | FileCheck %s -check-prefix=CHECK-OUTPUT
|
|
|
|
CHECK-OUTPUT: fib(4) = 3
|
|
*/
|