Open Menu / test / py / tests / test_efi_secboot / test_unsigned.py
Loading...
  1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
# SPDX-License-Identifier:      GPL-2.0+
# Copyright (c) 2019, Linaro Limited
# Author: AKASHI Takahiro <takahiro.akashi@linaro.org>
#
# U-Boot UEFI: Signed Image Authentication Test

"""
This test verifies image authentication for unsigned images.
"""

import pytest
import re
from defs import *

@pytest.mark.boardspec('sandbox')
@pytest.mark.buildconfigspec('efi_secure_boot')
@pytest.mark.buildconfigspec('cmd_efidebug')
@pytest.mark.buildconfigspec('cmd_fat')
@pytest.mark.buildconfigspec('cmd_nvedit_efi')
@pytest.mark.slow
class TestEfiUnsignedImage(object):
    def test_efi_unsigned_image_auth1(self, u_boot_console, efi_boot_env):
        """
        Test Case 1 - rejected when not digest in db or dbx
        """
        u_boot_console.restart_uboot()
        disk_img = efi_boot_env
        with u_boot_console.log.section('Test Case 1'):
            # Test Case 1
            output = u_boot_console.run_command_list([
                'host bind 0 %s' % disk_img,
		'fatload host 0:1 4000000 KEK.auth',
		'setenv -e -nv -bs -rt -at -i 4000000,$filesize KEK; echo',
		'fatload host 0:1 4000000 PK.auth',
		'setenv -e -nv -bs -rt -at -i 4000000,$filesize PK'])
            assert(not re.search('Failed to set EFI variable', ''.join(output)))

            output = u_boot_console.run_command_list([
                'efidebug boot add 1 HELLO host 0:1 /helloworld.efi ""',
                'efidebug boot next 1',
                'bootefi bootmgr'])
            assert(re.search('\'HELLO\' failed', ''.join(output)))
            output = u_boot_console.run_command_list([
                'efidebug boot next 1',
                'efidebug test bootmgr'])
            assert(re.search('efi_start_image[(][)] returned: 26',
                ''.join(output)))
            assert(not re.search('Hello, world!', ''.join(output)))

    def test_efi_unsigned_image_auth2(self, u_boot_console, efi_boot_env):
        """
        Test Case 2 - authenticated by digest in db
        """
        u_boot_console.restart_uboot()
        disk_img = efi_boot_env
        with u_boot_console.log.section('Test Case 2'):
            # Test Case 2
            output = u_boot_console.run_command_list([
                'host bind 0 %s' % disk_img,
		'fatload host 0:1 4000000 db_hello.auth',
		'setenv -e -nv -bs -rt -at -i 4000000,$filesize db; echo',
		'fatload host 0:1 4000000 KEK.auth',
		'setenv -e -nv -bs -rt -at -i 4000000,$filesize KEK',
		'fatload host 0:1 4000000 PK.auth',
		'setenv -e -nv -bs -rt -at -i 4000000,$filesize PK'])
            assert(not re.search('Failed to set EFI variable', ''.join(output)))

            output = u_boot_console.run_command_list([
                'efidebug boot add 1 HELLO host 0:1 /helloworld.efi ""',
                'efidebug boot next 1',
                'bootefi bootmgr'])
            assert(re.search('Hello, world!', ''.join(output)))

    def test_efi_unsigned_image_auth3(self, u_boot_console, efi_boot_env):
        """
        Test Case 3 - rejected by digest in dbx
        """
        u_boot_console.restart_uboot()
        disk_img = efi_boot_env
        with u_boot_console.log.section('Test Case 3a'):
            # Test Case 3a, rejected by dbx
            output = u_boot_console.run_command_list([
                'host bind 0 %s' % disk_img,
		'fatload host 0:1 4000000 db_hello.auth',
		'setenv -e -nv -bs -rt -at -i 4000000,$filesize dbx; echo',
		'fatload host 0:1 4000000 KEK.auth',
		'setenv -e -nv -bs -rt -at -i 4000000,$filesize KEK',
		'fatload host 0:1 4000000 PK.auth',
		'setenv -e -nv -bs -rt -at -i 4000000,$filesize PK'])
            assert(not re.search('Failed to set EFI variable', ''.join(output)))

            output = u_boot_console.run_command_list([
                'efidebug boot add 1 HELLO host 0:1 /helloworld.efi ""',
                'efidebug boot next 1',
                'bootefi bootmgr'])
            assert(re.search('\'HELLO\' failed', ''.join(output)))
            output = u_boot_console.run_command_list([
                'efidebug boot next 1',
                'efidebug test bootmgr'])
            assert(re.search('efi_start_image[(][)] returned: 26',
                ''.join(output)))
            assert(not re.search('Hello, world!', ''.join(output)))

        with u_boot_console.log.section('Test Case 3b'):
            # Test Case 3b, rejected by dbx even if db allows
            output = u_boot_console.run_command_list([
		'fatload host 0:1 4000000 db_hello.auth',
		'setenv -e -nv -bs -rt -at -i 4000000,$filesize db'])
            assert(not re.search('Failed to set EFI variable', ''.join(output)))

            output = u_boot_console.run_command_list([
                'efidebug boot add 1 HELLO host 0:1 /helloworld.efi ""',
                'efidebug boot next 1',
                'bootefi bootmgr'])
            assert(re.search('\'HELLO\' failed', ''.join(output)))
            output = u_boot_console.run_command_list([
                'efidebug boot next 1',
                'efidebug test bootmgr'])
            assert(re.search('efi_start_image[(][)] returned: 26',
                ''.join(output)))
            assert(not re.search('Hello, world!', ''.join(output)))