From: Adrian Prantl <aprantl@apple.com>
Date: Thu, 23 Aug 2018 17:19:08 +0000 (+0000)
Subject: lldbtest.py: Work around macOS SIP when testing ASANified builds.
X-Git-Tag: llvmorg-8.0.0-rc1~10286
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=748310b193a9dac6bde656e43c94d12fab3ac501;p=platform%2Fupstream%2Fllvm.git

lldbtest.py: Work around macOS SIP when testing ASANified builds.

llvm-svn: 340548
---

diff --git a/lldb/lit/Suite/lldbtest.py b/lldb/lit/Suite/lldbtest.py
index 378cdba..c17214c 100644
--- a/lldb/lit/Suite/lldbtest.py
+++ b/lldb/lit/Suite/lldbtest.py
@@ -9,6 +9,14 @@ import lit.TestRunner
 import lit.util
 from lit.formats.base import TestFormat
 
+def getBuildDir(cmd):
+    found = False
+    for arg in cmd:
+        if found:
+            return arg
+        if arg == '--build-dir':
+            found = True
+    return None
 
 class LLDBTest(TestFormat):
     def __init__(self, dotest_cmd):
@@ -49,6 +57,18 @@ class LLDBTest(TestFormat):
         # python exe as the first parameter of the command.
         cmd = [sys.executable] + self.dotest_cmd + [testPath, '-p', testFile]
 
+        # The macOS system integrity protection (SIP) doesn't allow injecting
+        # libraries into system binaries, but this can be worked around by
+        # copying the binary into a different location.
+        if test.config.environment['DYLD_INSERT_LIBRARIES'] and \
+                sys.executable.startswith('/System/'):
+            builddir = getBuildDir(cmd)
+            assert(builddir)
+            copied_python = os.path.join(builddir, 'copied-system-python')
+            import shutil
+            shutil.copy(sys.executable, os.path.join(builddir, copied_python))
+            cmd[0] = copied_python
+
         try:
             out, err, exitCode = lit.util.executeCommand(
                 cmd,