import unittest
import os
import sys
import subprocess
import socket
import time

from testzcc import ZccBaseTestCase

# Store testdir for safe switch back to directory:
testdir = os.path.dirname(os.path.abspath(__file__))

def tryrm(fn):
    try:
        os.remove(fn)
    except OSError:
        pass

def runQemu(kernel, machine='lm3s811evb'):
    """ Runs qemu on a given kernel file """

    tryrm('qemucontrol.sock')
    tryrm('qemuserial.sock')

    # Listen to the control socket:
    qemu_control_serve = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
    qemu_control_serve.bind('qemucontrol.sock')
    qemu_control_serve.listen(0)

    # Listen to the serial output:
    qemu_serial_serve = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
    qemu_serial_serve.bind('qemuserial.sock')
    qemu_serial_serve.listen(0)

    args = ['qemu-system-arm', '-M', machine, '-m', '16M',
        '-nographic', '-kernel', kernel, '-monitor',
        'unix:qemucontrol.sock',
        '-serial', 'unix:qemuserial.sock', '-S']
    p = subprocess.Popen(args)

    # Give process some time to boot:
    qemu_serial, address_peer = qemu_serial_serve.accept()
    qemu_control, address_peer = qemu_control_serve.accept()

    # Give the go command:
    qemu_control.send('cont\n'.encode('ascii'))

    qemu_serial.settimeout(0.2)

    # Receive all data:
    data = bytearray()
    for i in range(400):
        try:
            data += qemu_serial.recv(1)
        except socket.timeout as e:
            break
    data = data.decode('ascii')
    # print(data)

    # Send quit command:
    qemu_control.send("quit\n".encode('ascii'))
    try:
        p.wait(timeout=3)
    except subprocess.TimeoutExpired:
        p.kill()
    qemu_control.close()
    qemu_serial.close()
    qemu_control_serve.close()
    qemu_serial_serve.close()

    tryrm('qemucontrol.sock')
    tryrm('qemuserial.sock')

    # Check that output was correct:
    return data


class EmulationTestCase(ZccBaseTestCase):
    """ Tests the compiler driver """
    def setUp(self):
        if 'TESTEMU' not in os.environ:
            self.skipTest('Not running emulation tests')

    def testM3Bare(self):
        """ Build bare m3 binary and emulate it """
        recipe = os.path.join(testdir, 'm3_bare', 'recipe.yaml')
        self.buildRecipe(recipe)
        data = runQemu('m3_bare/bare.bin')
        self.assertEqual('Hello worle', data)

    def testA9Bare(self):
        """ Build vexpress cortex-A9 binary and emulate it """
        recipe = os.path.join(testdir, '..', 'examples', 'qemu_a9_hello', 'recipe.yaml')
        self.buildRecipe(recipe)
        data = runQemu('../examples/qemu_a9_hello/hello.bin', machine='vexpress-a9')
        self.assertEqual('Hello worle', data)

    def testKernelVexpressA9(self):
        """ Build vexpress cortex-A9 binary and emulate it """
        recipe = os.path.join(testdir, '..', 'kernel', 'arm.yaml')
        self.buildRecipe(recipe)
        data = runQemu('../kernel/kernel_arm.bin', machine='vexpress-a9')
        self.assertEqual('Welcome to lcfos!', data[:17])


if __name__ == '__main__':
    unittest.main()