1# Copyright (c) 2015 Stephen Warren
2# Copyright (c) 2015-2016, NVIDIA CORPORATION. All rights reserved.
3#
4# SPDX-License-Identifier: GPL-2.0
5
6# Logic to interact with U-Boot running on real hardware, typically via a
7# physical serial port.
8
9import sys
10from u_boot_spawn import Spawn
11from u_boot_console_base import ConsoleBase
12
13class ConsoleExecAttach(ConsoleBase):
14    """Represents a physical connection to a U-Boot console, typically via a
15    serial port. This implementation executes a sub-process to attach to the
16    console, expecting that the stdin/out of the sub-process will be forwarded
17    to/from the physical hardware. This approach isolates the test infra-
18    structure from the user-/installation-specific details of how to
19    communicate with, and the identity of, serial ports etc."""
20
21    def __init__(self, log, config):
22        """Initialize a U-Boot console connection.
23
24        Args:
25            log: A multiplexed_log.Logfile instance.
26            config: A "configuration" object as defined in conftest.py.
27
28        Returns:
29            Nothing.
30        """
31
32        # The max_fifo_fill value might need tweaking per-board/-SoC?
33        # 1 would be safe anywhere, but is very slow (a pexpect issue?).
34        # 16 is a common FIFO size.
35        # HW flow control would mean this could be infinite.
36        super(ConsoleExecAttach, self).__init__(log, config, max_fifo_fill=16)
37
38        with self.log.section('flash'):
39            self.log.action('Flashing U-Boot')
40            cmd = ['u-boot-test-flash', config.board_type, config.board_identity]
41            runner = self.log.get_runner(cmd[0], sys.stdout)
42            runner.run(cmd)
43            runner.close()
44            self.log.status_pass('OK')
45
46    def get_spawn(self):
47        """Connect to a fresh U-Boot instance.
48
49        The target board is reset, so that U-Boot begins running from scratch.
50
51        Args:
52            None.
53
54        Returns:
55            A u_boot_spawn.Spawn object that is attached to U-Boot.
56        """
57
58        args = [self.config.board_type, self.config.board_identity]
59        s = Spawn(['u-boot-test-console'] + args)
60
61        try:
62            self.log.action('Resetting board')
63            cmd = ['u-boot-test-reset'] + args
64            runner = self.log.get_runner(cmd[0], sys.stdout)
65            runner.run(cmd)
66            runner.close()
67        except:
68            s.close()
69            raise
70
71        return s
72