Imported Upstream version 1.6.0+dfsg1

[rustc.git] / src / compiler-rt / test / msan / dso-origin.cc

diff --git a/src/compiler-rt/test/msan/dso-origin.cc b/src/compiler-rt/test/msan/dso-origin.cc
index 9bde029a96707cbea8d43734ad97d80c20e7a915..7f62449842cead897c73999eb9281f1e7bfae4bf 100644 (file)
--- a/src/compiler-rt/test/msan/dso-origin.cc
+++ b/src/compiler-rt/test/msan/dso-origin.cc
@@ -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 -m64 -O0 %s %t-so.so -o %t && not %run %t 2>&1 | FileCheck %s
+// RUN: %clangxx_msan -fsanitize-memory-track-origins -O0 %s -DBUILD_SO -fPIC -shared -o %t-so.so
+// RUN: %clangxx_msan -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