]> git.sur5r.net Git - u-boot/blob - test/py/u_boot_console_sandbox.py
test/py: Make print statements python 3.x safe
[u-boot] / test / py / u_boot_console_sandbox.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 the sandbox port of U-Boot, running as a sub-process.
6
7 import time
8 from u_boot_spawn import Spawn
9 from u_boot_console_base import ConsoleBase
10
11 class ConsoleSandbox(ConsoleBase):
12     """Represents a connection to a sandbox U-Boot console, executed as a sub-
13     process."""
14
15     def __init__(self, log, config):
16         """Initialize a U-Boot console connection.
17
18         Args:
19             log: A multiplexed_log.Logfile instance.
20             config: A "configuration" object as defined in conftest.py.
21
22         Returns:
23             Nothing.
24         """
25
26         super(ConsoleSandbox, self).__init__(log, config, max_fifo_fill=1024)
27
28     def get_spawn(self):
29         """Connect to a fresh U-Boot instance.
30
31         A new sandbox process is created, so that U-Boot begins running from
32         scratch.
33
34         Args:
35             None.
36
37         Returns:
38             A u_boot_spawn.Spawn object that is attached to U-Boot.
39         """
40
41         bcfg = self.config.buildconfig
42         config_spl = bcfg.get('config_spl', 'n') == 'y'
43         fname = '/spl/u-boot-spl' if config_spl else '/u-boot'
44         print(fname)
45         cmd = []
46         if self.config.gdbserver:
47             cmd += ['gdbserver', self.config.gdbserver]
48         cmd += [
49             self.config.build_dir + fname,
50             '-v',
51             '-d',
52             self.config.dtb
53         ]
54         return Spawn(cmd, cwd=self.config.source_dir)
55
56     def kill(self, sig):
57         """Send a specific Unix signal to the sandbox process.
58
59         Args:
60             sig: The Unix signal to send to the process.
61
62         Returns:
63             Nothing.
64         """
65
66         self.log.action('kill %d' % sig)
67         self.p.kill(sig)
68
69     def validate_exited(self):
70         """Determine whether the sandbox process has exited.
71
72         If required, this function waits a reasonable time for the process to
73         exit.
74
75         Args:
76             None.
77
78         Returns:
79             Boolean indicating whether the process has exited.
80         """
81
82         p = self.p
83         self.p = None
84         for i in xrange(100):
85             ret = not p.isalive()
86             if ret:
87                 break
88             time.sleep(0.1)
89         p.close()
90         return ret