1#!/usr/bin/env python3
2#
3# Functional test that boots the ASPEED SoCs with firmware
4#
5# Copyright (C) 2022 ASPEED Technology Inc
6#
7# SPDX-License-Identifier: GPL-2.0-or-later
8
9import os
10import time
11import subprocess
12import tempfile
13
14from qemu_test import LinuxKernelTest, Asset
15from qemu_test import exec_command_and_wait_for_pattern
16from qemu_test import interrupt_interactive_console_until_pattern
17from qemu_test import has_cmd
18from qemu_test.utils import archive_extract
19from zipfile import ZipFile
20from unittest import skipUnless
21
22class AST1030Machine(LinuxKernelTest):
23
24    ASSET_ZEPHYR_1_04 = Asset(
25        ('https://github.com/AspeedTech-BMC'
26         '/zephyr/releases/download/v00.01.04/ast1030-evb-demo.zip'),
27        '4ac6210adcbc61294927918707c6762483fd844dde5e07f3ba834ad1f91434d3')
28
29    def test_ast1030_zephyros_1_04(self):
30        self.set_machine('ast1030-evb')
31
32        zip_file = self.ASSET_ZEPHYR_1_04.fetch()
33
34        kernel_name = "ast1030-evb-demo/zephyr.elf"
35        with ZipFile(zip_file, 'r') as zf:
36                     zf.extract(kernel_name, path=self.workdir)
37        kernel_file = os.path.join(self.workdir, kernel_name)
38
39        self.vm.set_console()
40        self.vm.add_args('-kernel', kernel_file, '-nographic')
41        self.vm.launch()
42        self.wait_for_console_pattern("Booting Zephyr OS")
43        exec_command_and_wait_for_pattern(self, "help",
44                                          "Available commands")
45
46    ASSET_ZEPHYR_1_07 = Asset(
47        ('https://github.com/AspeedTech-BMC'
48         '/zephyr/releases/download/v00.01.07/ast1030-evb-demo.zip'),
49        'ad52e27959746988afaed8429bf4e12ab988c05c4d07c9d90e13ec6f7be4574c')
50
51    def test_ast1030_zephyros_1_07(self):
52        self.set_machine('ast1030-evb')
53
54        zip_file = self.ASSET_ZEPHYR_1_07.fetch()
55
56        kernel_name = "ast1030-evb-demo/zephyr.bin"
57        with ZipFile(zip_file, 'r') as zf:
58                     zf.extract(kernel_name, path=self.workdir)
59        kernel_file = os.path.join(self.workdir, kernel_name)
60
61        self.vm.set_console()
62        self.vm.add_args('-kernel', kernel_file, '-nographic')
63        self.vm.launch()
64        self.wait_for_console_pattern("Booting Zephyr OS")
65        for shell_cmd in [
66                'kernel stacks',
67                'otp info conf',
68                'otp info scu',
69                'hwinfo devid',
70                'crypto aes256_cbc_vault',
71                'random get',
72                'jtag JTAG1 sw_xfer high TMS',
73                'adc ADC0 resolution 12',
74                'adc ADC0 read 42',
75                'adc ADC1 read 69',
76                'i2c scan I2C_0',
77                'i3c attach I3C_0',
78                'hash test',
79                'kernel uptime',
80                'kernel reboot warm',
81                'kernel uptime',
82                'kernel reboot cold',
83                'kernel uptime',
84        ]: exec_command_and_wait_for_pattern(self, shell_cmd, "uart:~$")
85
86class AST2x00Machine(LinuxKernelTest):
87
88    def do_test_arm_aspeed(self, machine, image):
89        self.set_machine(machine)
90        self.vm.set_console()
91        self.vm.add_args('-drive', 'file=' + image + ',if=mtd,format=raw',
92                         '-net', 'nic', '-snapshot')
93        self.vm.launch()
94
95        self.wait_for_console_pattern("U-Boot 2016.07")
96        self.wait_for_console_pattern("## Loading kernel from FIT Image at 20080000")
97        self.wait_for_console_pattern("Starting kernel ...")
98        self.wait_for_console_pattern("Booting Linux on physical CPU 0x0")
99        self.wait_for_console_pattern(
100                "aspeed-smc 1e620000.spi: read control register: 203b0641")
101        self.wait_for_console_pattern("ftgmac100 1e660000.ethernet eth0: irq ")
102        self.wait_for_console_pattern("systemd[1]: Set hostname to")
103
104    ASSET_PALMETTO_FLASH = Asset(
105        ('https://github.com/openbmc/openbmc/releases/download/2.9.0/'
106         'obmc-phosphor-image-palmetto.static.mtd'),
107        '3e13bbbc28e424865dc42f35ad672b10f2e82cdb11846bb28fa625b48beafd0d');
108
109    def test_arm_ast2400_palmetto_openbmc_v2_9_0(self):
110        image_path = self.ASSET_PALMETTO_FLASH.fetch()
111
112        self.do_test_arm_aspeed('palmetto-bmc', image_path)
113
114    ASSET_ROMULUS_FLASH = Asset(
115        ('https://github.com/openbmc/openbmc/releases/download/2.9.0/'
116         'obmc-phosphor-image-romulus.static.mtd'),
117        '820341076803f1955bc31e647a512c79f9add4f5233d0697678bab4604c7bb25')
118
119    def test_arm_ast2500_romulus_openbmc_v2_9_0(self):
120        image_path = self.ASSET_ROMULUS_FLASH.fetch()
121
122        self.do_test_arm_aspeed('romulus-bmc', image_path)
123
124    def do_test_arm_aspeed_buildroot_start(self, image, cpu_id, pattern='Aspeed EVB'):
125        self.require_netdev('user')
126        self.vm.set_console()
127        self.vm.add_args('-drive', 'file=' + image + ',if=mtd,format=raw,read-only=true',
128                         '-net', 'nic', '-net', 'user')
129        self.vm.launch()
130
131        self.wait_for_console_pattern('U-Boot 2019.04')
132        self.wait_for_console_pattern('## Loading kernel from FIT Image')
133        self.wait_for_console_pattern('Starting kernel ...')
134        self.wait_for_console_pattern('Booting Linux on physical CPU ' + cpu_id)
135        self.wait_for_console_pattern('lease of 10.0.2.15')
136        # the line before login:
137        self.wait_for_console_pattern(pattern)
138        exec_command_and_wait_for_pattern(self, 'root', 'Password:')
139        exec_command_and_wait_for_pattern(self, 'passw0rd', '#')
140
141    def do_test_arm_aspeed_buildroot_poweroff(self):
142        exec_command_and_wait_for_pattern(self, 'poweroff',
143                                          'reboot: System halted');
144
145    ASSET_BR2_202311_AST2500_FLASH = Asset(
146        ('https://github.com/legoater/qemu-aspeed-boot/raw/master/'
147         'images/ast2500-evb/buildroot-2023.11/flash.img'),
148        'c23db6160cf77d0258397eb2051162c8473a56c441417c52a91ba217186e715f')
149
150    def test_arm_ast2500_evb_buildroot(self):
151        self.set_machine('ast2500-evb')
152
153        image_path = self.ASSET_BR2_202311_AST2500_FLASH.fetch()
154
155        self.vm.add_args('-device',
156                         'tmp105,bus=aspeed.i2c.bus.3,address=0x4d,id=tmp-test');
157        self.do_test_arm_aspeed_buildroot_start(image_path, '0x0',
158                                                'ast2500-evb login:')
159
160        exec_command_and_wait_for_pattern(self,
161             'echo lm75 0x4d > /sys/class/i2c-dev/i2c-3/device/new_device',
162             'i2c i2c-3: new_device: Instantiated device lm75 at 0x4d');
163        exec_command_and_wait_for_pattern(self,
164                             'cat /sys/class/hwmon/hwmon1/temp1_input', '0')
165        self.vm.cmd('qom-set', path='/machine/peripheral/tmp-test',
166                    property='temperature', value=18000);
167        exec_command_and_wait_for_pattern(self,
168                             'cat /sys/class/hwmon/hwmon1/temp1_input', '18000')
169
170        self.do_test_arm_aspeed_buildroot_poweroff()
171
172    ASSET_BR2_202311_AST2600_FLASH = Asset(
173        ('https://github.com/legoater/qemu-aspeed-boot/raw/master/'
174         'images/ast2600-evb/buildroot-2023.11/flash.img'),
175        'b62808daef48b438d0728ee07662290490ecfa65987bb91294cafb1bb7ad1a68')
176
177    def test_arm_ast2600_evb_buildroot(self):
178        self.set_machine('ast2600-evb')
179
180        image_path = self.ASSET_BR2_202311_AST2600_FLASH.fetch()
181
182        self.vm.add_args('-device',
183                         'tmp105,bus=aspeed.i2c.bus.3,address=0x4d,id=tmp-test');
184        self.vm.add_args('-device',
185                         'ds1338,bus=aspeed.i2c.bus.3,address=0x32');
186        self.vm.add_args('-device',
187                         'i2c-echo,bus=aspeed.i2c.bus.3,address=0x42');
188        self.do_test_arm_aspeed_buildroot_start(image_path, '0xf00',
189                                                'ast2600-evb login:')
190
191        exec_command_and_wait_for_pattern(self,
192             'echo lm75 0x4d > /sys/class/i2c-dev/i2c-3/device/new_device',
193             'i2c i2c-3: new_device: Instantiated device lm75 at 0x4d');
194        exec_command_and_wait_for_pattern(self,
195                             'cat /sys/class/hwmon/hwmon1/temp1_input', '0')
196        self.vm.cmd('qom-set', path='/machine/peripheral/tmp-test',
197                    property='temperature', value=18000);
198        exec_command_and_wait_for_pattern(self,
199                             'cat /sys/class/hwmon/hwmon1/temp1_input', '18000')
200
201        exec_command_and_wait_for_pattern(self,
202             'echo ds1307 0x32 > /sys/class/i2c-dev/i2c-3/device/new_device',
203             'i2c i2c-3: new_device: Instantiated device ds1307 at 0x32');
204        year = time.strftime("%Y")
205        exec_command_and_wait_for_pattern(self, 'hwclock -f /dev/rtc1', year);
206
207        exec_command_and_wait_for_pattern(self,
208             'echo slave-24c02 0x1064 > /sys/bus/i2c/devices/i2c-3/new_device',
209             'i2c i2c-3: new_device: Instantiated device slave-24c02 at 0x64');
210        exec_command_and_wait_for_pattern(self,
211             'i2cset -y 3 0x42 0x64 0x00 0xaa i', '#');
212        exec_command_and_wait_for_pattern(self,
213             'hexdump /sys/bus/i2c/devices/3-1064/slave-eeprom',
214             '0000000 ffaa ffff ffff ffff ffff ffff ffff ffff');
215        self.do_test_arm_aspeed_buildroot_poweroff()
216
217    ASSET_BR2_202302_AST2600_TPM_FLASH = Asset(
218        ('https://github.com/legoater/qemu-aspeed-boot/raw/master/'
219         'images/ast2600-evb/buildroot-2023.02-tpm/flash.img'),
220        'a46009ae8a5403a0826d607215e731a8c68d27c14c41e55331706b8f9c7bd997')
221
222    @skipUnless(*has_cmd('swtpm'))
223    def test_arm_ast2600_evb_buildroot_tpm(self):
224        self.set_machine('ast2600-evb')
225
226        image_path = self.ASSET_BR2_202302_AST2600_TPM_FLASH.fetch()
227
228        tpmstate_dir = tempfile.TemporaryDirectory(prefix="qemu_")
229        socket = os.path.join(tpmstate_dir.name, 'swtpm-socket')
230
231        # We must put the TPM state dir in /tmp/, not the build dir,
232        # because some distros use AppArmor to lock down swtpm and
233        # restrict the set of locations it can access files in.
234        subprocess.run(['swtpm', 'socket', '-d', '--tpm2',
235                        '--tpmstate', f'dir={tpmstate_dir.name}',
236                        '--ctrl', f'type=unixio,path={socket}'])
237
238        self.vm.add_args('-chardev', f'socket,id=chrtpm,path={socket}')
239        self.vm.add_args('-tpmdev', 'emulator,id=tpm0,chardev=chrtpm')
240        self.vm.add_args('-device',
241                         'tpm-tis-i2c,tpmdev=tpm0,bus=aspeed.i2c.bus.12,address=0x2e')
242        self.do_test_arm_aspeed_buildroot_start(image_path, '0xf00', 'Aspeed AST2600 EVB')
243
244        exec_command_and_wait_for_pattern(self,
245            'echo tpm_tis_i2c 0x2e > /sys/bus/i2c/devices/i2c-12/new_device',
246            'tpm_tis_i2c 12-002e: 2.0 TPM (device-id 0x1, rev-id 1)');
247        exec_command_and_wait_for_pattern(self,
248            'cat /sys/class/tpm/tpm0/pcr-sha256/0',
249            'B804724EA13F52A9072BA87FE8FDCC497DFC9DF9AA15B9088694639C431688E0');
250
251        self.do_test_arm_aspeed_buildroot_poweroff()
252
253    def do_test_arm_aspeed_sdk_start(self, image):
254        self.require_netdev('user')
255        self.vm.set_console()
256        self.vm.add_args('-drive', 'file=' + image + ',if=mtd,format=raw',
257                         '-net', 'nic', '-net', 'user', '-snapshot')
258        self.vm.launch()
259
260        self.wait_for_console_pattern('U-Boot 2019.04')
261        self.wait_for_console_pattern('## Loading kernel from FIT Image')
262        self.wait_for_console_pattern('Starting kernel ...')
263
264    ASSET_SDK_V806_AST2500 = Asset(
265        'https://github.com/AspeedTech-BMC/openbmc/releases/download/v08.06/ast2500-default-obmc.tar.gz',
266        'e1755f3cadff69190438c688d52dd0f0d399b70a1e14b1d3d5540fc4851d38ca')
267
268    def test_arm_ast2500_evb_sdk(self):
269        self.set_machine('ast2500-evb')
270
271        image_path = self.ASSET_SDK_V806_AST2500.fetch()
272
273        archive_extract(image_path, self.workdir)
274
275        self.do_test_arm_aspeed_sdk_start(
276            self.workdir + '/ast2500-default/image-bmc')
277
278        self.wait_for_console_pattern('ast2500-default login:')
279
280    ASSET_SDK_V806_AST2600_A2 = Asset(
281        'https://github.com/AspeedTech-BMC/openbmc/releases/download/v08.06/ast2600-a2-obmc.tar.gz',
282        '9083506135f622d5e7351fcf7d4e1c7125cee5ba16141220c0ba88931f3681a4')
283
284    def test_arm_ast2600_evb_sdk(self):
285        self.set_machine('ast2600-evb')
286
287        image_path = self.ASSET_SDK_V806_AST2600_A2.fetch()
288
289        archive_extract(image_path, self.workdir)
290
291        self.vm.add_args('-device',
292            'tmp105,bus=aspeed.i2c.bus.5,address=0x4d,id=tmp-test');
293        self.vm.add_args('-device',
294            'ds1338,bus=aspeed.i2c.bus.5,address=0x32');
295        self.do_test_arm_aspeed_sdk_start(
296            self.workdir + '/ast2600-a2/image-bmc')
297
298        self.wait_for_console_pattern('ast2600-a2 login:')
299
300        exec_command_and_wait_for_pattern(self, 'root', 'Password:')
301        exec_command_and_wait_for_pattern(self, '0penBmc', 'root@ast2600-a2:~#')
302
303        exec_command_and_wait_for_pattern(self,
304            'echo lm75 0x4d > /sys/class/i2c-dev/i2c-5/device/new_device',
305            'i2c i2c-5: new_device: Instantiated device lm75 at 0x4d');
306        exec_command_and_wait_for_pattern(self,
307             'cat /sys/class/hwmon/hwmon19/temp1_input', '0')
308        self.vm.cmd('qom-set', path='/machine/peripheral/tmp-test',
309                    property='temperature', value=18000);
310        exec_command_and_wait_for_pattern(self,
311             'cat /sys/class/hwmon/hwmon19/temp1_input', '18000')
312
313        exec_command_and_wait_for_pattern(self,
314             'echo ds1307 0x32 > /sys/class/i2c-dev/i2c-5/device/new_device',
315             'i2c i2c-5: new_device: Instantiated device ds1307 at 0x32');
316        year = time.strftime("%Y")
317        exec_command_and_wait_for_pattern(self,
318             '/sbin/hwclock -f /dev/rtc1', year);
319
320
321class AST2x00MachineMMC(LinuxKernelTest):
322
323    ASSET_RAINIER_EMMC = Asset(
324        ('https://fileserver.linaro.org/s/B6pJTwWEkzSDi36/download/'
325         'mmc-p10bmc-20240617.qcow2'),
326        'd523fb478d2b84d5adc5658d08502bc64b1486955683814f89c6137518acd90b')
327
328    def test_arm_aspeed_emmc_boot(self):
329        self.set_machine('rainier-bmc')
330        self.require_netdev('user')
331
332        image_path = self.ASSET_RAINIER_EMMC.fetch()
333
334        self.vm.set_console()
335        self.vm.add_args('-drive',
336                         'file=' + image_path + ',if=sd,id=sd2,index=2',
337                         '-net', 'nic', '-net', 'user', '-snapshot')
338        self.vm.launch()
339
340        self.wait_for_console_pattern('U-Boot SPL 2019.04')
341        self.wait_for_console_pattern('Trying to boot from MMC1')
342        self.wait_for_console_pattern('U-Boot 2019.04')
343        self.wait_for_console_pattern('eMMC 2nd Boot')
344        self.wait_for_console_pattern('## Loading kernel from FIT Image')
345        self.wait_for_console_pattern('Starting kernel ...')
346        self.wait_for_console_pattern('Booting Linux on physical CPU 0xf00')
347        self.wait_for_console_pattern('mmcblk0: p1 p2 p3 p4 p5 p6 p7')
348        self.wait_for_console_pattern('IBM eBMC (OpenBMC for IBM Enterprise')
349
350if __name__ == '__main__':
351    LinuxKernelTest.main()
352