[PATCH] D54113: [sanitizer] Use AT_EXECFN in ReExec() if available

[Benjamin Kramer via Phabricator via llvm-commits]

bkramer updated this revision to Diff 172649.
bkramer added a comment.

Add test


Repository:
  rCRT Compiler Runtime

https://reviews.llvm.org/D54113

Files:
  lib/sanitizer_common/sanitizer_linux.cc
  test/msan/Linux/reexec_unlimited_stack.cc


Index: test/msan/Linux/reexec_unlimited_stack.cc
===================================================================
--- /dev/null
+++ test/msan/Linux/reexec_unlimited_stack.cc
@@ -0,0 +1,20 @@
+// MSAN re-execs on unlimited stacks. We use that to verify ReExec() uses the
+// right path.
+// RUN: %clangxx_msan -O0 %s -o %t && ulimit -s unlimited && %run %t | FileCheck %s
+
+#include <stdio.h>
+
+#if !defined(__GLIBC_PREREQ)
+#define __GLIBC_PREREQ(a, b) 0
+#endif
+
+#if __GLIBC_PREREQ(2, 16)
+#include <sys/auxv.h>
+#endif
+
+int main() {
+#if __GLIBC_PREREQ(2, 16)
+  puts(reinterpret_cast<const char *>(getauxval(AT_EXECFN)));
+#endif
+  // CHECK-NOT: /proc/self/exe
+}
Index: lib/sanitizer_common/sanitizer_linux.cc
===================================================================
--- lib/sanitizer_common/sanitizer_linux.cc
+++ lib/sanitizer_common/sanitizer_linux.cc
@@ -640,6 +640,10 @@
 #elif SANITIZER_SOLARIS
   pathname = getexecname();
   CHECK_NE(pathname, NULL);
+#elif SANITIZER_USE_GETAUXVAL
+  // Calling execve with /proc/self/exe sets that as $EXEC_ORIGIN. Binaries that
+  // rely on that will fail to load shared libraries. Query AT_EXECFN instead.
+  pathname = reinterpret_cast<const char *>(getauxval(AT_EXECFN));
 #endif
 
   GetArgsAndEnv(&argv, &envp);


-------------- next part --------------
A non-text attachment was scrubbed...
Name: D54113.172649.patch
Type: text/x-patch
Size: 1287 bytes
Desc: not available
URL: <http://lists.llvm.org/pipermail/llvm-commits/attachments/20181105/e0ac695c/attachment.bin>