TheRock/compiler/amd-llvm/lldb/test/API/macosx/rosetta/TestRosetta.py

import re
import lldb
import lldbsuite.test.lldbutil as lldbutil
from lldbsuite.test.lldbtest import *
from lldbsuite.test.decorators import *
from lldbsuite.test import configuration
from os.path import exists


def get_os_version():
    try:
        os_version_str = subprocess.check_output(["sysctl", "kern.osversion"]).decode(
            "utf-8"
        )
    except subprocess.CalledProcessError:
        return None
    m = re.match(r"kern\.osversion: (\w+)", os_version_str)
    if m:
        return m.group(1)
    return None


def rosetta_debugserver_installed():
    return exists("/Library/Apple/usr/libexec/oah/debugserver")


def has_rosetta_shared_cache(os_version):
    if not os_version:
        return False
    macos_device_support = os.path.join(
        os.path.expanduser("~"), "Library", "Developer", "Xcode", "macOS DeviceSupport"
    )
    for _, subdirs, _ in os.walk(macos_device_support):
        for subdir in subdirs:
            if os_version in subdir:
                return True
    return False


class TestRosetta(TestBase):
    NO_DEBUG_INFO_TESTCASE = True
    SHARED_BUILD_TESTCASE = False

    @skipUnlessAppleSilicon
    @skipIfLLVMTargetMissing("X86")
    @skipIfDarwinEmbedded
    def test_rosetta(self):
        """There can be many tests in a test case - describe this test here."""
        self.build(
            dictionary={
                "TRIPLE": configuration.triple.replace(self.getArchitecture(), "x86_64")
            }
        )
        self.main_source_file = lldb.SBFileSpec("main.c")

        if rosetta_debugserver_installed():
            broadcaster = self.dbg.GetBroadcaster()
            listener = lldbutil.start_listening_from(
                broadcaster, lldb.SBDebugger.eBroadcastBitWarning
            )

            target, process, thread, bkpt = lldbutil.run_to_source_breakpoint(
                self, "Set a breakpoint here", self.main_source_file
            )

            event = lldb.SBEvent()
            os_version = get_os_version()
            if self.TraceOn():
                self.runCmd("image list")
                self.runCmd("target list")
                self.runCmd("platform shell ls ~/Library/Developer/Xcode")
                self.runCmd(
                    'platform shell ls ~/Library/Developer/Xcode/"macOS DeviceSupport/"'
                )
                self.runCmd("process plugin packet send jGetSharedCacheInfo:{}")
            if not has_rosetta_shared_cache(os_version):
                self.assertTrue(listener.GetNextEvent(event))
            else:
                self.assertFalse(listener.GetNextEvent(event))