import lldb
from lldbsuite.test.lldbtest import *
from lldbsuite.test.decorators import *
import lldbsuite.test.lldbutil as lldbutil
import json
import unittest2


@skipIfDarwin # rdar://problem/64552748
class TestSimulatorPlatformLaunching(TestBase):

    mydir = TestBase.compute_mydir(__file__)
    NO_DEBUG_INFO_TESTCASE = True

    def check_load_commands(self, expected_load_command):
        """sanity check the built binary for the expected number of load commands"""
        load_cmds = subprocess.check_output(
            ['otool', '-l', self.getBuildArtifact()]
        ).decode("utf-8")
        found = 0
        for line in load_cmds.split('\n'):
            if expected_load_command in line:
              found += 1
        self.assertEquals(found, 1, "wrong load command")

    def check_debugserver(self, log, expected_platform, expected_version):
        """scan the debugserver packet log"""
        logfile = open(log, "r")
        dylib_info = None
        process_info_ostype = None
        expect_dylib_info_response = False
        expect_process_info_response = False
        for line in logfile:
            if expect_dylib_info_response:
                while line[0] != '$':
                    line = line[1:]
                line = line[1:]
                # Unescape '}'.
                dylib_info = json.loads(line.replace('}]','}')[:-4])
                expect_dylib_info_response = False
            if 'send packet: $jGetLoadedDynamicLibrariesInfos:{' in line:
                expect_dylib_info_response = True
            if expect_process_info_response:
                for pair in line.split(';'):
                    keyval = pair.split(':')
                    if len(keyval) == 2 and keyval[0] == 'ostype':
                        process_info_ostype = keyval[1]
            if 'send packet: $qProcessInfo#' in line:
                expect_process_info_response = True

        self.assertEquals(process_info_ostype, expected_platform)
        self.assertTrue(dylib_info)
        aout_info = None
        for image in dylib_info['images']:
            if image['pathname'].endswith('a.out'):
                aout_info = image
        self.assertTrue(aout_info)
        self.assertEquals(aout_info['min_version_os_name'], expected_platform)
        if expected_version:
            self.assertEquals(aout_info['min_version_os_sdk'], expected_version)


    def run_with(self, arch, os, vers, env, expected_load_command):
        self.build(dictionary={'TRIPLE': arch+'-apple-'+os+vers+'-'+env})
        self.check_load_commands(expected_load_command)
        log = self.getBuildArtifact('packets.log')
        self.expect("log enable gdb-remote packets -f "+log)
        lldbutil.run_to_source_breakpoint(self, "break here",
                                          lldb.SBFileSpec("hello.c"))
        self.expect('image list -b -t',
                    patterns=['a\.out '+arch+'-apple-'+os+vers+'.*-'+env])
        self.check_debugserver(log, os+env, vers)

    @skipUnlessDarwin
    @skipIfDarwinEmbedded
    @apple_simulator_test('iphone')
    def test_ios(self):
        """Test running an iOS simulator binary"""
        self.run_with(arch=self.getArchitecture(),
                      os='ios', vers='', env='simulator',
                      expected_load_command='LC_BUILD_VERSION')

    @skipUnlessDarwin
    @skipIfDarwinEmbedded
    @apple_simulator_test('appletv')
    def test_tvos(self):
        """Test running an tvOS simulator binary"""
        self.run_with(arch=self.getArchitecture(),
                      os='tvos', vers='', env='simulator',
                      expected_load_command='LC_BUILD_VERSION')

    @skipUnlessDarwin
    @skipIfDarwinEmbedded
    @apple_simulator_test('watch')
    @skipIfDarwin # rdar://problem/64552748
    @skipIf(archs=['arm64','arm64e'])
    def test_watchos_i386(self):
        """Test running a 32-bit watchOS simulator binary"""
        self.run_with(arch='i386',
                      os='watchos', vers='', env='simulator',
                      expected_load_command='LC_BUILD_VERSION')

    @skipUnlessDarwin
    @skipIfDarwinEmbedded
    @apple_simulator_test('watch')
    @skipIfDarwin # rdar://problem/64552748
    @skipIf(archs=['i386','x86_64'])
    def test_watchos_armv7k(self):
        """Test running a 32-bit watchOS simulator binary"""
        self.run_with(arch='armv7k',
                      os='watchos', vers='', env='simulator',
                      expected_load_command='LC_BUILD_VERSION')


    #
    # Back-deployment tests.
    #
    # Older Mach-O versions used less expressive load commands, such
    # as LC_VERSION_MIN_IPHONEOS that wouldn't distinguish between ios
    # and ios-simulator.  When targeting a simulator on Apple Silicon
    # macOS, however, these legacy load commands are never generated.
    #
        
    @skipUnlessDarwin
    @skipIfDarwinEmbedded
    @apple_simulator_test('iphone')
    @skipIf(archs=['arm64','arm64e'])
    def test_lc_version_min_iphoneos(self):
        """Test running a back-deploying iOS simulator binary
           with a legacy iOS load command"""
        self.run_with(arch=self.getArchitecture(),
                      os='ios', vers='11.0', env='simulator',
                      expected_load_command='LC_VERSION_MIN_IPHONEOS')

    @skipUnlessDarwin
    @skipIfDarwinEmbedded
    @apple_simulator_test('iphone')
    @skipIf(archs=['arm64','arm64e'])
    def test_ios_backdeploy_x86(self):
        """Test running a back-deploying iOS simulator binary
           with a legacy iOS load command"""
        self.run_with(arch=self.getArchitecture(),
                      os='ios', vers='13.0', env='simulator',
                      expected_load_command='LC_BUILD_VERSION')

    @skipUnlessDarwin
    @skipIfDarwinEmbedded
    @apple_simulator_test('iphone')
    @skipIf(archs=['i386','x86_64'])
    def test_ios_backdeploy_apple_silicon(self):
        """Test running a back-deploying iOS simulator binary"""
        self.run_with(arch=self.getArchitecture(),
                      os='ios', vers='11.0', env='simulator',
                      expected_load_command='LC_BUILD_VERSION')

    @skipUnlessDarwin
    @skipIfDarwinEmbedded
    @apple_simulator_test('appletv')
    @skipIf(archs=['arm64','arm64e'])
    def test_lc_version_min_tvos(self):
        """Test running a back-deploying tvOS simulator binary
           with a legacy tvOS load command"""
        self.run_with(arch=self.getArchitecture(),
                      os='tvos', vers='11.0', env='simulator',
                      expected_load_command='LC_VERSION_MIN_TVOS')

    @skipUnlessDarwin
    @skipIfDarwinEmbedded
    @apple_simulator_test('appletv')
    @skipIf(archs=['i386','x86_64'])
    def test_tvos_backdeploy_apple_silicon(self):
        """Test running a back-deploying tvOS simulator binary"""
        self.run_with(arch=self.getArchitecture(),
                      os='tvos', vers='11.0', env='simulator',
                      expected_load_command='LC_BUILD_VERSION')

    @skipUnlessDarwin
    @skipIfDarwinEmbedded
    @apple_simulator_test('watch')
    @skipIf(archs=['arm64','arm64e'])
    @skipIfDarwin # rdar://problem/64552748
    def test_lc_version_min_watchos(self):
        """Test running a back-deploying watchOS simulator binary
           with a legacy watchOS load command"""
        self.run_with(arch='i386',
                      os='watchos', vers='4.0', env='simulator',
                      expected_load_command='LC_VERSION_MIN_WATCHOS')

    @skipUnlessDarwin
    @skipIfDarwinEmbedded
    @apple_simulator_test('watch')
    @skipIf(archs=['arm64','arm64e'])
    @skipIfDarwin # rdar://problem/64552748
    def test_watchos_backdeploy_apple_silicon(self):
        """Test running a back-deploying watchOS simulator binary"""
        self.run_with(arch='armv7k',
                      os='watchos', vers='4.0', env='simulator',
                      expected_load_command='LC_BUILD_VERSION')