66 lines · c
1// RUN: %clang_dfsan -fno-sanitize=dataflow -O2 -fPIE -DCALLBACKS -c %s -o %t-callbacks.o2// RUN: %clang_dfsan -gmlt -fsanitize-ignorelist=%S/Inputs/flags_abilist.txt -O2 -mllvm -dfsan-reaches-function-callbacks=1 %s %t-callbacks.o -o %t3// RUN: %run %t 2>&1 | FileCheck %s4 5// RUN: %clang_dfsan -fno-sanitize=dataflow -O2 -fPIE -DCALLBACKS -DORIGIN_TRACKING -c %s -o %t-callbacks.o6// RUN: %clang_dfsan -gmlt -fsanitize-ignorelist=%S/Inputs/flags_abilist.txt -O2 -mllvm -dfsan-reaches-function-callbacks=1 -mllvm -dfsan-track-origins=2 %s %t-callbacks.o -o %t7// RUN: %run %t 2>&1 | FileCheck --check-prefix=CHECK-ORIGIN-TRACKING %s8 9// Tests that callbacks are inserted for reached functions when10// -dfsan-reaches-function-callbacks is specified.11 12#include <assert.h>13#include <stdio.h>14#include <string.h>15#include <sanitizer/dfsan_interface.h>16 17#ifdef CALLBACKS18// Compile this code without DFSan to avoid recursive instrumentation.19 20void my_dfsan_reaches_function_callback(dfsan_label label, dfsan_origin origin,21 const char *file, unsigned int line,22 const char *function) {23#ifdef ORIGIN_TRACKING24 dfsan_print_origin_id_trace(origin);25#else26 printf("%s:%d %s\n", file, line, function);27#endif28}29 30#else31 32__attribute__((noinline)) uint64_t add(uint64_t *a, uint64_t *b) {33 34 return *a + *b;35 // CHECK: reaches_function.c:[[# @LINE - 1]] add.dfsan36 // CHECK-ORIGIN-TRACKING: Origin value: 0x10000002, Taint value was stored to memory at37 // CHECK-ORIGIN-TRACKING: #0 {{.*}} in add.dfsan {{.*}}reaches_function.c:[[# @LINE - 3]]:{{.*}}38 // CHECK-ORIGIN-TRACKING: Origin value: 0x1, Taint value was created at39 // CHECK-ORIGIN-TRACKING: #0 {{.*}} in main {{.*}}reaches_function.c:{{.*}}40}41 42extern void my_dfsan_reaches_function_callback(dfsan_label label,43 dfsan_origin origin,44 const char *file,45 unsigned int line,46 const char *function);47 48int main(int argc, char *argv[]) {49 50 dfsan_set_reaches_function_callback(my_dfsan_reaches_function_callback);51 52 uint64_t a = 0;53 uint64_t b = 0;54 55 dfsan_set_label(8, &a, sizeof(a));56 uint64_t c = add(&a, &b);57 // CHECK: reaches_function.c:[[# @LINE - 1]] main58 // CHECK-ORIGIN-TRACKING: Origin value: 0x10000002, Taint value was stored to memory at59 // CHECK-ORIGIN-TRACKING: #0 {{.*}} in add.dfsan {{.*}}reaches_function.c:{{.*}}60 // CHECK-ORIGIN-TRACKING: Origin value: 0x1, Taint value was created at61 // CHECK-ORIGIN-TRACKING: #0 {{.*}} in main {{.*}}reaches_function.c:[[# @LINE - 6]]:{{.*}}62 return c;63}64 65#endif // #ifdef CALLBACKS66