mirror of
https://github.com/AsahiLinux/u-boot
synced 2024-12-23 03:23:47 +00:00
3b7bf8a9e4
Encapsulate the UEFI EFI_TCG2_PROTOCOL unit test in an Python test. Signed-off-by: Heinrich Schuchardt <heinrich.schuchardt@canonical.com>
232 lines
9.7 KiB
Python
232 lines
9.7 KiB
Python
# SPDX-License-Identifier: GPL-2.0
|
|
# Copyright (c) 2017, Heinrich Schuchardt <xypron.glpk@gmx.de>
|
|
|
|
"""
|
|
Test UEFI API implementation
|
|
"""
|
|
|
|
import pytest
|
|
|
|
@pytest.mark.buildconfigspec('cmd_bootefi_selftest')
|
|
def test_efi_selftest(u_boot_console):
|
|
"""Run UEFI unit tests
|
|
|
|
:param u_boot_console: U-Boot console
|
|
|
|
This function executes all selftests that are not marked as on request.
|
|
"""
|
|
u_boot_console.run_command(cmd='setenv efi_selftest')
|
|
u_boot_console.run_command(cmd='bootefi selftest', wait_for_prompt=False)
|
|
m = u_boot_console.p.expect(['Summary: 0 failures', 'Press any key'])
|
|
if m != 0:
|
|
raise Exception('Failures occurred during the EFI selftest')
|
|
u_boot_console.restart_uboot()
|
|
|
|
@pytest.mark.buildconfigspec('cmd_bootefi_selftest')
|
|
@pytest.mark.buildconfigspec('hush_parser')
|
|
@pytest.mark.buildconfigspec('of_control')
|
|
@pytest.mark.notbuildconfigspec('generate_acpi_table')
|
|
def test_efi_selftest_device_tree(u_boot_console):
|
|
"""Test the device tree support in the UEFI sub-system
|
|
|
|
:param u_boot_console: U-Boot console
|
|
|
|
This test executes the UEFI unit test by calling 'bootefi selftest'.
|
|
"""
|
|
u_boot_console.run_command(cmd='setenv efi_selftest list')
|
|
output = u_boot_console.run_command('bootefi selftest')
|
|
assert '\'device tree\'' in output
|
|
u_boot_console.run_command(cmd='setenv efi_selftest device tree')
|
|
# Set serial# if it is not already set.
|
|
u_boot_console.run_command(cmd='setenv efi_test "${serial#}x"')
|
|
u_boot_console.run_command(cmd='test "${efi_test}" = x && setenv serial# 0')
|
|
u_boot_console.run_command(cmd='bootefi selftest ${fdtcontroladdr}', wait_for_prompt=False)
|
|
m = u_boot_console.p.expect(['serial-number:', 'U-Boot'])
|
|
if m != 0:
|
|
raise Exception('serial-number missing in device tree')
|
|
u_boot_console.restart_uboot()
|
|
|
|
@pytest.mark.buildconfigspec('cmd_bootefi_selftest')
|
|
def test_efi_selftest_watchdog_reboot(u_boot_console):
|
|
"""Test the watchdog timer
|
|
|
|
:param u_boot_console: U-Boot console
|
|
|
|
This function executes the 'watchdog reboot' unit test.
|
|
"""
|
|
u_boot_console.run_command(cmd='setenv efi_selftest list')
|
|
output = u_boot_console.run_command('bootefi selftest')
|
|
assert '\'watchdog reboot\'' in output
|
|
u_boot_console.run_command(cmd='setenv efi_selftest watchdog reboot')
|
|
u_boot_console.run_command(cmd='bootefi selftest', wait_for_prompt=False)
|
|
m = u_boot_console.p.expect(['resetting', 'U-Boot'])
|
|
if m != 0:
|
|
raise Exception('Reset failed in \'watchdog reboot\' test')
|
|
u_boot_console.restart_uboot()
|
|
|
|
@pytest.mark.buildconfigspec('cmd_bootefi_selftest')
|
|
def test_efi_selftest_text_input(u_boot_console):
|
|
"""Test the EFI_SIMPLE_TEXT_INPUT_PROTOCOL
|
|
|
|
:param u_boot_console: U-Boot console
|
|
|
|
This function calls the text input EFI selftest.
|
|
"""
|
|
u_boot_console.run_command(cmd='setenv efi_selftest text input')
|
|
u_boot_console.run_command(cmd='bootefi selftest', wait_for_prompt=False)
|
|
m = u_boot_console.p.expect([r'To terminate type \'x\''])
|
|
if m != 0:
|
|
raise Exception('No prompt for \'text input\' test')
|
|
u_boot_console.drain_console()
|
|
u_boot_console.p.timeout = 500
|
|
# EOT
|
|
u_boot_console.run_command(cmd=chr(4), wait_for_echo=False,
|
|
send_nl=False, wait_for_prompt=False)
|
|
m = u_boot_console.p.expect(
|
|
[r'Unicode char 4 \(unknown\), scan code 0 \(Null\)'])
|
|
if m != 0:
|
|
raise Exception('EOT failed in \'text input\' test')
|
|
u_boot_console.drain_console()
|
|
# BS
|
|
u_boot_console.run_command(cmd=chr(8), wait_for_echo=False,
|
|
send_nl=False, wait_for_prompt=False)
|
|
m = u_boot_console.p.expect(
|
|
[r'Unicode char 8 \(BS\), scan code 0 \(Null\)'])
|
|
if m != 0:
|
|
raise Exception('BS failed in \'text input\' test')
|
|
u_boot_console.drain_console()
|
|
# TAB
|
|
u_boot_console.run_command(cmd=chr(9), wait_for_echo=False,
|
|
send_nl=False, wait_for_prompt=False)
|
|
m = u_boot_console.p.expect(
|
|
[r'Unicode char 9 \(TAB\), scan code 0 \(Null\)'])
|
|
if m != 0:
|
|
raise Exception('BS failed in \'text input\' test')
|
|
u_boot_console.drain_console()
|
|
# a
|
|
u_boot_console.run_command(cmd='a', wait_for_echo=False, send_nl=False,
|
|
wait_for_prompt=False)
|
|
m = u_boot_console.p.expect(
|
|
[r'Unicode char 97 \(\'a\'\), scan code 0 \(Null\)'])
|
|
if m != 0:
|
|
raise Exception('\'a\' failed in \'text input\' test')
|
|
u_boot_console.drain_console()
|
|
# UP escape sequence
|
|
u_boot_console.run_command(cmd=chr(27) + '[A', wait_for_echo=False,
|
|
send_nl=False, wait_for_prompt=False)
|
|
m = u_boot_console.p.expect(
|
|
[r'Unicode char 0 \(Null\), scan code 1 \(Up\)'])
|
|
if m != 0:
|
|
raise Exception('UP failed in \'text input\' test')
|
|
u_boot_console.drain_console()
|
|
# Euro sign
|
|
u_boot_console.run_command(cmd=b'\xe2\x82\xac'.decode(), wait_for_echo=False,
|
|
send_nl=False, wait_for_prompt=False)
|
|
m = u_boot_console.p.expect([r'Unicode char 8364 \(\''])
|
|
if m != 0:
|
|
raise Exception('Euro sign failed in \'text input\' test')
|
|
u_boot_console.drain_console()
|
|
u_boot_console.run_command(cmd='x', wait_for_echo=False, send_nl=False,
|
|
wait_for_prompt=False)
|
|
m = u_boot_console.p.expect(['Summary: 0 failures', 'Press any key'])
|
|
if m != 0:
|
|
raise Exception('Failures occurred during the EFI selftest')
|
|
u_boot_console.restart_uboot()
|
|
|
|
@pytest.mark.buildconfigspec('cmd_bootefi_selftest')
|
|
def test_efi_selftest_text_input_ex(u_boot_console):
|
|
"""Test the EFI_SIMPLE_TEXT_INPUT_EX_PROTOCOL
|
|
|
|
:param u_boot_console: U-Boot console
|
|
|
|
This function calls the extended text input EFI selftest.
|
|
"""
|
|
u_boot_console.run_command(cmd='setenv efi_selftest extended text input')
|
|
u_boot_console.run_command(cmd='bootefi selftest', wait_for_prompt=False)
|
|
m = u_boot_console.p.expect([r'To terminate type \'CTRL\+x\''])
|
|
if m != 0:
|
|
raise Exception('No prompt for \'text input\' test')
|
|
u_boot_console.drain_console()
|
|
u_boot_console.p.timeout = 500
|
|
# EOT
|
|
u_boot_console.run_command(cmd=chr(4), wait_for_echo=False,
|
|
send_nl=False, wait_for_prompt=False)
|
|
m = u_boot_console.p.expect(
|
|
[r'Unicode char 100 \(\'d\'\), scan code 0 \(CTRL\+Null\)'])
|
|
if m != 0:
|
|
raise Exception('EOT failed in \'text input\' test')
|
|
u_boot_console.drain_console()
|
|
# BS
|
|
u_boot_console.run_command(cmd=chr(8), wait_for_echo=False,
|
|
send_nl=False, wait_for_prompt=False)
|
|
m = u_boot_console.p.expect(
|
|
[r'Unicode char 8 \(BS\), scan code 0 \(\+Null\)'])
|
|
if m != 0:
|
|
raise Exception('BS failed in \'text input\' test')
|
|
u_boot_console.drain_console()
|
|
# TAB
|
|
u_boot_console.run_command(cmd=chr(9), wait_for_echo=False,
|
|
send_nl=False, wait_for_prompt=False)
|
|
m = u_boot_console.p.expect(
|
|
[r'Unicode char 9 \(TAB\), scan code 0 \(\+Null\)'])
|
|
if m != 0:
|
|
raise Exception('TAB failed in \'text input\' test')
|
|
u_boot_console.drain_console()
|
|
# a
|
|
u_boot_console.run_command(cmd='a', wait_for_echo=False, send_nl=False,
|
|
wait_for_prompt=False)
|
|
m = u_boot_console.p.expect(
|
|
[r'Unicode char 97 \(\'a\'\), scan code 0 \(Null\)'])
|
|
if m != 0:
|
|
raise Exception('\'a\' failed in \'text input\' test')
|
|
u_boot_console.drain_console()
|
|
# UP escape sequence
|
|
u_boot_console.run_command(cmd=chr(27) + '[A', wait_for_echo=False,
|
|
send_nl=False, wait_for_prompt=False)
|
|
m = u_boot_console.p.expect(
|
|
[r'Unicode char 0 \(Null\), scan code 1 \(\+Up\)'])
|
|
if m != 0:
|
|
raise Exception('UP failed in \'text input\' test')
|
|
u_boot_console.drain_console()
|
|
# Euro sign
|
|
u_boot_console.run_command(cmd=b'\xe2\x82\xac'.decode(), wait_for_echo=False,
|
|
send_nl=False, wait_for_prompt=False)
|
|
m = u_boot_console.p.expect([r'Unicode char 8364 \(\''])
|
|
if m != 0:
|
|
raise Exception('Euro sign failed in \'text input\' test')
|
|
u_boot_console.drain_console()
|
|
# SHIFT+ALT+FN 5
|
|
u_boot_console.run_command(cmd=b'\x1b\x5b\x31\x35\x3b\x34\x7e'.decode(),
|
|
wait_for_echo=False, send_nl=False,
|
|
wait_for_prompt=False)
|
|
m = u_boot_console.p.expect(
|
|
[r'Unicode char 0 \(Null\), scan code 15 \(SHIFT\+ALT\+FN 5\)'])
|
|
if m != 0:
|
|
raise Exception('SHIFT+ALT+FN 5 failed in \'text input\' test')
|
|
u_boot_console.drain_console()
|
|
u_boot_console.run_command(cmd=chr(24), wait_for_echo=False, send_nl=False,
|
|
wait_for_prompt=False)
|
|
m = u_boot_console.p.expect(['Summary: 0 failures', 'Press any key'])
|
|
if m != 0:
|
|
raise Exception('Failures occurred during the EFI selftest')
|
|
u_boot_console.restart_uboot()
|
|
|
|
@pytest.mark.buildconfigspec('cmd_bootefi_selftest')
|
|
@pytest.mark.buildconfigspec('efi_tcg2_protocol')
|
|
def test_efi_selftest_tcg2(u_boot_console):
|
|
"""Test the EFI_TCG2 PROTOCOL
|
|
|
|
:param u_boot_console: U-Boot console
|
|
|
|
This function executes the 'tcg2' unit test.
|
|
"""
|
|
u_boot_console.restart_uboot()
|
|
u_boot_console.run_command(cmd='setenv efi_selftest list')
|
|
output = u_boot_console.run_command('bootefi selftest')
|
|
assert '\'tcg2\'' in output
|
|
u_boot_console.run_command(cmd='setenv efi_selftest tcg2')
|
|
u_boot_console.run_command(cmd='bootefi selftest', wait_for_prompt=False)
|
|
m = u_boot_console.p.expect(['Summary: 0 failures', 'Press any key'])
|
|
if m != 0:
|
|
raise Exception('Failures occurred during the EFI selftest')
|
|
u_boot_console.restart_uboot()
|