Index: compiler-rt/trunk/lib/dfsan/dfsan.cc =================================================================== --- compiler-rt/trunk/lib/dfsan/dfsan.cc +++ compiler-rt/trunk/lib/dfsan/dfsan.cc @@ -162,6 +162,8 @@ // this function (the instrumentation pass inlines the equality test). extern "C" SANITIZER_INTERFACE_ATTRIBUTE dfsan_label __dfsan_union(dfsan_label l1, dfsan_label l2) { + if (flags().fast16labels) + return l1 | l2; DCHECK_NE(l1, l2); if (l1 == 0) Index: compiler-rt/trunk/lib/dfsan/dfsan_flags.inc =================================================================== --- compiler-rt/trunk/lib/dfsan/dfsan_flags.inc +++ compiler-rt/trunk/lib/dfsan/dfsan_flags.inc @@ -29,3 +29,7 @@ DFSAN_FLAG(const char *, dump_labels_at_exit, "", "The path of the file where " "to dump the labels when the " "program terminates.") +DFSAN_FLAG(bool, fast16labels, false, + "Enables experimental mode where DFSan supports only 16 power-of-2 labels " + "(1, 2, 4, 8, ... 32768) and the label union is computed as a bit-wise OR." +) Index: compiler-rt/trunk/test/dfsan/fast16labels.c =================================================================== --- compiler-rt/trunk/test/dfsan/fast16labels.c +++ compiler-rt/trunk/test/dfsan/fast16labels.c @@ -0,0 +1,25 @@ +// RUN: %clang_dfsan %s -o %t && DFSAN_OPTIONS=fast16labels=1 %run %t +// +// Tests DFSAN_OPTIONS=fast16labels=1 +// +#include + +#include +#include + +int foo(int a, int b) { + return a + b; +} + +int main() { + int a = 10; + int b = 20; + dfsan_set_label(8, &a, sizeof(a)); + dfsan_set_label(512, &b, sizeof(b)); + int c = foo(a, b); + printf("A: 0x%x\n", dfsan_get_label(a)); + printf("B: 0x%x\n", dfsan_get_label(b)); + dfsan_label l = dfsan_get_label(c); + printf("C: 0x%x\n", l); + assert(l == 520); // OR of the other two labels. +}