# Copyright (C) 2014 Intel Corporation
#
# Released under the MIT license (see COPYING.MIT)

# This module adds support to testimage.bbclass to deploy images and run
# tests on a Ubiquiti Networks EdgeRouter Lite. The device must be set up
# to boot into the master image already - the easiest way to do that is as
# follows:
#
# 1. Take out the internal USB drive and plug it into your PC
# 2. Repartition the USB drive so that you have three partitions in this
#    order:
#      1: vfat, labelled "boot" (it will need to be formatted with mkfs.vfat
#         for this to be possible, since FAT partitions formatted under
#         DOS/Windows will only support uppercase labels)
#      2: ext3 (for master image) labelled "testmaster"
#      3: ext3 (for image under test) labelled "testrootfs"
# 3. Copy the kernel to be used by the master image to the FAT partition
#    (it should be named "vmlinux.64" with the factory u-boot configuration)
# 4. Install the master image onto the "testmaster" ext3 partition. If
#    you do this by just extracting the contents of an image onto the
#    partition, you will also likely need to create the master image marker
#    file /etc/masterimage within this partition so that we can tell when
#    we're booted into it that it is the master image.
# 5. Put the USB drive back into the device, and ensure the console port
#    and first ethernet port are connected before powering on
#
# TEST_SERIALCONTROL_CMD will need to be set in local.conf so that we can
# interact with u-boot over the serial console port.

import os
import bb
import time
import subprocess
import sys
import pexpect

from oeqa.controllers.controllerimage import ControllerImageHardwareTarget


class EdgeRouterTarget(ControllerImageHardwareTarget):

    def __init__(self, d):
        super(EdgeRouterTarget, self).__init__(d)

        self.image_fstype = self.get_image_fstype(d)
        self.deploy_cmds = [
                'mount -L boot /boot',
                'mkdir -p /mnt/testrootfs',
                'mount -L testrootfs /mnt/testrootfs',
                'cp ~/test-kernel /boot',
                'rm -rf /mnt/testrootfs/*',
                'tar xvf ~/test-rootfs.%s -C /mnt/testrootfs' % self.image_fstype
                ]
        if not self.serialcontrol_cmd:
            bb.fatal("This TEST_TARGET needs a TEST_SERIALCONTROL_CMD defined in local.conf.")


    def _deploy(self):
        self.controller.run("umount /mnt/testrootfs;")
        self.controller.ignore_status = False
        self.controller.copy_to(self.kernel, "~/test-kernel")
        self.controller.copy_to(self.rootfs, "~/test-rootfs.%s" % self.image_fstype)
        for cmd in self.deploy_cmds:
            self.controller.run(cmd)

    def _start(self, params=None):
        self.power_cycle(self.controller)
        try:
            serialconn = pexpect.spawn(self.serialcontrol_cmd, env=self.origenv, logfile=sys.stdout)
            serialconn.expect("U-Boot")
            serialconn.sendline("a")
            serialconn.expect("Octeon ubnt_e100#")
            serialconn.sendline("fatload usb 0:1 $loadaddr test-kernel")
            serialconn.expect(" bytes read")
            serialconn.expect("Octeon ubnt_e100#")
            serialconn.sendline("bootoctlinux $loadaddr coremask=0x3 root=/dev/sda3 rw rootwait mtdparts=phys_mapped_flash:512k(boot0),512k(boot1),64k@3072k(eeprom)")
            serialconn.expect("login:", timeout=120)
            serialconn.close()
        except pexpect.ExceptionPexpect as e:
            bb.fatal('Serial interaction failed: %s' % str(e))

    def _wait_until_booted(self):
        try:
            serialconn = pexpect.spawn(self.serialcontrol_cmd, env=self.origenv, logfile=sys.stdout)
            serialconn.expect("login:", timeout=120)
            serialconn.close()
        except pexpect.ExceptionPexpect as e:
            bb.fatal('Serial interaction failed: %s' % str(e))