1# Copyright (C) 2014 Intel Corporation
2#
3# Released under the MIT license (see COPYING.MIT)
4
5# This module adds support to testimage.bbclass to deploy images and run
6# tests on a BeagleBone (original "white" or Black models). The device must
7# be set up as per README.hardware and the master image should be deployed
8# onto the card so that it boots into it by default. For booting into the
9# image under test we interact with u-boot over serial, so for the
10# BeagleBone Black you will need an additional TTL serial cable since a
11# serial interface isn't automatically provided over the USB connection as
12# it is on the original BeagleBone ("white") version. The separate ext3
13# partition that will contain the image to be tested must be labelled
14# "testrootfs" so that the deployment code below can find it.
15#
16# NOTE: for the BeagleBone "white" (original version) you may need to use
17# a script which handles the serial device disappearing on power down, such
18# as scripts/contrib/serdevtry in OE-Core.
19
20import os
21import bb
22import time
23import subprocess
24import sys
25import pexpect
26
27import oeqa.utils.sshcontrol as sshcontrol
28from oeqa.controllers.masterimage import MasterImageHardwareTarget
29
30
31class BeagleBoneTarget(MasterImageHardwareTarget):
32
33    dtbs = {'uImage-am335x-bone.dtb': 'am335x-bone.dtb', 'uImage-am335x-boneblack.dtb': 'am335x-boneblack.dtb'}
34
35    @classmethod
36    def get_extra_files(self):
37        return list(self.dtbs.keys())
38
39    def __init__(self, d):
40        super(BeagleBoneTarget, self).__init__(d)
41
42        self.image_fstype = self.get_image_fstype(d)
43        self.deploy_cmds = [
44                'mkdir -p /mnt/testrootfs',
45                'mount -L testrootfs /mnt/testrootfs',
46                'rm -rf /mnt/testrootfs/*',
47                'tar xvf ~/test-rootfs.%s -C /mnt/testrootfs' % self.image_fstype,
48                '[ -e /mnt/testrootfs/boot/uImage ] || [ -L /mnt/testrootfs/boot/uImage ] || cp ~/test-kernel /mnt/testrootfs/boot/uImage',
49                ]
50
51        for _, dtbfn in self.dtbs.iteritems():
52            # Kernel and dtb files may not be in the image, so copy them if not
53            self.deploy_cmds.append('[ -e /mnt/testrootfs/boot/{0} ] || cp ~/{0} /mnt/testrootfs/boot/'.format(dtbfn))
54
55        if not self.serialcontrol_cmd:
56            bb.fatal("This TEST_TARGET needs a TEST_SERIALCONTROL_CMD defined in local.conf.")
57
58
59    def _deploy(self):
60        self.master.run("umount /boot; umount /mnt/testrootfs;")
61        self.master.ignore_status = False
62        # Kernel and dtb files may not be in the image, so copy them just in case
63        self.master.copy_to(self.kernel, "~/test-kernel")
64        kernelpath = os.path.dirname(self.kernel)
65        for dtborig, dtbfn in self.dtbs.iteritems():
66            dtbfile = os.path.join(kernelpath, dtborig)
67            if os.path.exists(dtbfile):
68                self.master.copy_to(dtbfile, "~/%s" % dtbfn)
69        self.master.copy_to(self.rootfs, "~/test-rootfs.%s" % self.image_fstype)
70        for cmd in self.deploy_cmds:
71            self.master.run(cmd)
72
73    def _start(self, params=None):
74        self.power_cycle(self.master)
75        try:
76            serialconn = pexpect.spawn(self.serialcontrol_cmd, env=self.origenv, logfile=sys.stdout)
77            # We'd wait for "U-Boot" here but sometimes we connect too late on BeagleBone white to see it
78            serialconn.expect("NAND:")
79            serialconn.expect("MMC:")
80            serialconn.sendline("a")
81            serialconn.expect("U-Boot#")
82            serialconn.sendline("setenv bootpart 0:3")
83            serialconn.expect("U-Boot#")
84            serialconn.sendline("setenv mmcroot /dev/mmcblk0p3 ro")
85            serialconn.expect("U-Boot#")
86            serialconn.sendline("boot")
87            serialconn.expect("login:", timeout=120)
88            serialconn.close()
89        except pexpect.ExceptionPexpect as e:
90            bb.fatal('Serial interaction failed: %s' % str(e))
91
92    def _wait_until_booted(self):
93        try:
94            serialconn = pexpect.spawn(self.serialcontrol_cmd, env=self.origenv, logfile=sys.stdout)
95            serialconn.expect("login:", timeout=120)
96            serialconn.close()
97        except pexpect.ExceptionPexpect as e:
98            bb.fatal('Serial interaction failed: %s' % str(e))
99