[dfsan] Add test for origin tracking stack traces

[stephan.yichao.zhao]

We have test cases about origin tracking stacks. See compiler-rt/test/dfsan/origin_*.c
Does this one check any thing uncovered by existing cases?

[gbalats]

In D100518#2690407, @stephan.yichao.zhao wrote:

We have test cases about origin tracking stacks. See compiler-rt/test/dfsan/origin_*.c
Does this one check any thing uncovered by existing cases?

In most test cases I saw, the the stack trace had just main. This test would focus on more complicated cases. I also intend to expand it in the future to include origin tracing APIs that use vectors instead.

[stephan.yichao.zhao]

compiler-rt/test/dfsan/origin_stack_trace.c

44

This is not used yet.

[gbalats]

compiler-rt/test/dfsan/origin_stack_trace.c

44

This is special FileCheck syntax that you can add to check if a line occurs multiple times.
See https://llvm.org/docs/CommandGuide/FileCheck.html#the-check-count-directive.

[stephan.yichao.zhao]

compiler-rt/test/dfsan/origin_stack_trace.c

44

Sorry. I forgot this feature completely...

compiler-rt/test/dfsan/origin_stack_trace.c

+// RUN: %clang_dfsan -gmlt -mllvm -dfsan-track-origins=1 -mllvm -dfsan-fast-16-labels=true %s -o %t && \
+// RUN:     %run %t >%t.out 2>&1
+// RUN: FileCheck %s --check-prefix=CHECK < %t.out
+//
+// RUN: %clang_dfsan -gmlt -mllvm -dfsan-track-origins=1 -mllvm -dfsan-fast-16-labels=true -mllvm -dfsan-instrument-with-call-threshold=0 %s -o %t && \
+// RUN:     %run %t >%t.out 2>&1
+// RUN: FileCheck %s --check-prefix=CHECK < %t.out
+//
+// REQUIRES: x86_64-target-arch
+
+#include <sanitizer/dfsan_interface.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-6]]
+
+// CHECK: Origin value: {{.*}}, Taint value was created at
+// CHECK: #0 {{.*}} in dfs$bar {{.*}}origin_stack_trace.c:[[@LINE-21]]
+// CHECK-COUNT-8: #{{[0-9]+}} {{.*}} in dfs$bar {{.*}}origin_stack_trace.c:[[@LINE-24]]
+// CHECK: #9 {{.*}} in dfs$baz {{.*}}origin_stack_trace.c:[[@LINE-18]]