test/py/u_boot_console_exec_attach.py

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