| // RUN: %clang_dfsan -gmlt -mllvm -dfsan-track-origins=1 %s -o %t && \ |
| // RUN: %run %t >%t.out 2>&1 |
| // RUN: FileCheck %s < %t.out |
| // |
| // RUN: %clang_dfsan -gmlt -mllvm -dfsan-track-origins=1 -mllvm -dfsan-instrument-with-call-threshold=0 %s -o %t && \ |
| // RUN: %run %t >%t.out 2>&1 |
| // RUN: FileCheck %s < %t.out |
| |
| #include <assert.h> |
| #include <malloc.h> |
| #include <sanitizer/dfsan_interface.h> |
| #include <stdio.h> |
| #include <string.h> |
| |
| #define NOINLINE __attribute__((noinline)) |
| |
| NOINLINE int foo(int a, int b) { return a + b; } |
| |
| NOINLINE void bar(int depth, void *addr, int size) { |
| if (depth) { |
| bar(depth - 1, addr, size); |
| } else { |
| dfsan_set_label(1, addr, size); |
| } |
| } |
| |
| NOINLINE void baz(int depth, void *addr, int size) { bar(depth, addr, size); } |
| |
| int main(int argc, char *argv[]) { |
| int a = 10; |
| int b = 20; |
| baz(8, &a, sizeof(a)); |
| int c = foo(a, b); |
| dfsan_print_origin_trace(&c, NULL); |
| // CHECK: Taint value 0x1 {{.*}} origin tracking () |
| // CHECK: Origin value: {{.*}}, Taint value was stored to memory at |
| // CHECK: #0 {{.*}} in main {{.*}}origin_stack_trace.c:[[@LINE-4]] |
| |
| // CHECK: Origin value: {{.*}}, Taint value was created at |
| // CHECK: #0 {{.*}} in bar.dfsan {{.*}}origin_stack_trace.c:[[@LINE-17]] |
| // CHECK-COUNT-8: #{{[0-9]+}} {{.*}} in bar.dfsan {{.*}}origin_stack_trace.c:[[@LINE-20]] |
| // CHECK: #9 {{.*}} in baz.dfsan {{.*}}origin_stack_trace.c:[[@LINE-15]] |
| |
| // Test logic expects this buffer to be large enough. |
| // String contains current paths, which could vary in length. |
| // Make this buffer much larger than necessary to accomodate variation. |
| const size_t kBufSize = 8192; |
| char *buf = (char *)malloc(kBufSize); |
| size_t length = dfsan_sprint_origin_trace(&c, NULL, buf, kBufSize); |
| assert(kBufSize > length); |
| |
| printf("==OUTPUT==\n\n%s==EOS==\n", buf); |
| // CHECK: ==OUTPUT== |
| // CHECK: Taint value 0x1 {{.*}} origin tracking () |
| // CHECK: Origin value: {{.*}}, Taint value was stored to memory at |
| // CHECK: #0 {{.*}} in main {{.*}}origin_stack_trace.c:[[@LINE-23]] |
| |
| // CHECK: Origin value: {{.*}}, Taint value was created at |
| // CHECK: #0 {{.*}} in bar.dfsan {{.*}}origin_stack_trace.c:[[@LINE-36]] |
| // CHECK-COUNT-8: #{{[0-9]+}} {{.*}} in bar.dfsan {{.*}}origin_stack_trace.c:[[@LINE-39]] |
| // CHECK: #9 {{.*}} in baz.dfsan {{.*}}origin_stack_trace.c:[[@LINE-34]] |
| // CHECK: ==EOS== |
| |
| char tinybuf[18]; |
| size_t same_length = |
| dfsan_sprint_origin_trace(&c, NULL, tinybuf, sizeof(tinybuf)); |
| |
| printf("==TRUNCATED OUTPUT==\n\n%s==EOS==\n", tinybuf); |
| // CHECK: ==TRUNCATED OUTPUT== |
| // CHECK: Taint value 0x1==EOS== |
| |
| printf("Returned length: %zu\n", length); |
| printf("Actual length: %zu\n", strlen(buf)); |
| printf("Returned length with truncation: %zu\n", same_length); |
| |
| // CHECK: Returned length: [[#LEN:]] |
| // CHECK: Actual length: [[#LEN]] |
| // CHECK: Returned length with truncation: [[#LEN]] |
| |
| size_t length_with_desc = |
| dfsan_sprint_origin_trace(&c, "DESCRIPTION", buf, kBufSize); |
| assert(kBufSize > length_with_desc); |
| |
| printf("==OUTPUT==\n\n%s==EOS==\n", buf); |
| // CHECK: ==OUTPUT== |
| // CHECK: Taint value 0x1 {{.*}} origin tracking (DESCRIPTION) |
| // CHECK: Origin value: {{.*}}, Taint value was stored to memory at |
| // CHECK: #0 {{.*}} in main {{.*}}origin_stack_trace.c:[[@LINE-55]] |
| |
| // CHECK: Origin value: {{.*}}, Taint value was created at |
| // CHECK: #0 {{.*}} in bar.dfsan {{.*}}origin_stack_trace.c:[[@LINE-68]] |
| // CHECK-COUNT-8: #{{[0-9]+}} {{.*}} in bar.dfsan {{.*}}origin_stack_trace.c:[[@LINE-71]] |
| // CHECK: #9 {{.*}} in baz.dfsan {{.*}}origin_stack_trace.c:[[@LINE-66]] |
| // CHECK: ==EOS== |
| |
| printf("Returned length: %zu\n", length_with_desc); |
| // COMM: Message length is increased by 11: the length of "DESCRIPTION". |
| // CHECK: Returned length: [[#LEN + 11]] |
| |
| buf[0] = '\0'; |
| length = dfsan_sprint_origin_trace(&c, NULL, buf, 0); |
| printf("Output=\"%s\"\n", buf); |
| printf("Returned length: %zu\n", length); |
| // CHECK: Output="" |
| // CHECK: Returned length: [[#LEN]] |
| |
| free(buf); |
| } |