[compiler-rt] r212484 - [MSan] Use a single file for dso-origin test

[Alexey Samsonov]

Author: samsonov
Date: Mon Jul  7 15:37:12 2014
New Revision: 212484

URL: http://llvm.org/viewvc/llvm-project?rev=212484&view=rev
Log:
[MSan] Use a single file for dso-origin test

Removed:
    compiler-rt/trunk/test/msan/SharedLibs/
Modified:
    compiler-rt/trunk/test/msan/dso-origin.cc

Modified: compiler-rt/trunk/test/msan/dso-origin.cc
URL: http://llvm.org/viewvc/llvm-project/compiler-rt/trunk/test/msan/dso-origin.cc?rev=212484&r1=212483&r2=212484&view=diff
==============================================================================
--- compiler-rt/trunk/test/msan/dso-origin.cc (original)
+++ compiler-rt/trunk/test/msan/dso-origin.cc Mon Jul  7 15:37:12 2014
@@ -1,12 +1,33 @@
 // Build a library with origin tracking and an executable w/o origin tracking.
 // Test that origin tracking is enabled at runtime.
-// RUN: %clangxx_msan -fsanitize-memory-track-origins -m64 -O0 %p/SharedLibs/dso-origin-so.cc \
-// RUN:     -fPIC -shared -o %t-so.so
+// RUN: %clangxx_msan -fsanitize-memory-track-origins -m64 -O0 %s -DBUILD_SO -fPIC -shared -o %t-so.so
 // RUN: %clangxx_msan -m64 -O0 %s %t-so.so -o %t && not %run %t 2>&1 | FileCheck %s
 
+#ifdef BUILD_SO
+
+#include <stdlib.h>
+
+extern "C" {
+void my_access(int *p) {
+  volatile int tmp;
+  // Force initialize-ness check.
+  if (*p)
+    tmp = 1;
+}
+
+void *my_alloc(unsigned sz) {
+  return malloc(sz);
+}
+}  // extern "C"
+
+#else  // BUILD_SO
+
 #include <stdlib.h>
 
-#include "SharedLibs/dso-origin.h"
+extern "C" {
+void my_access(int *p);
+void *my_alloc(unsigned sz);
+}
 
 int main(int argc, char **argv) {
   int *x = (int *)my_alloc(sizeof(int));
@@ -14,12 +35,14 @@ int main(int argc, char **argv) {
   delete x;
 
   // CHECK: WARNING: MemorySanitizer: use-of-uninitialized-value
-  // CHECK: {{#0 0x.* in my_access .*dso-origin-so.cc:}}
+  // CHECK: {{#0 0x.* in my_access .*dso-origin.cc:}}
   // CHECK: {{#1 0x.* in main .*dso-origin.cc:}}[[@LINE-5]]
   // CHECK: Uninitialized value was created by a heap allocation
   // CHECK: {{#0 0x.* in .*malloc}}
-  // CHECK: {{#1 0x.* in my_alloc .*dso-origin-so.cc:}}
+  // CHECK: {{#1 0x.* in my_alloc .*dso-origin.cc:}}
   // CHECK: {{#2 0x.* in main .*dso-origin.cc:}}[[@LINE-10]]
-  // CHECK: SUMMARY: MemorySanitizer: use-of-uninitialized-value {{.*dso-origin-so.cc:.* my_access}}
+  // CHECK: SUMMARY: MemorySanitizer: use-of-uninitialized-value {{.*dso-origin.cc:.* my_access}}
   return 0;
 }
+
+#endif  // BUILD_SO