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
7import time
8from u_boot_spawn import Spawn
9from u_boot_console_base import ConsoleBase
10
11class 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
91