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
122
123
124
|
/* Copyright (c) 2013 The Chromium OS Authors. All rights reserved.
* Use of this source code is governed by a BSD-style license that can be
* found in the LICENSE file.
*
* Tests for firmware display library.
*/
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "2common.h"
#include "2misc.h"
#include "2nvstorage.h"
#include "2struct.h"
#include "2sysincludes.h"
#include "host_common.h"
#include "test_common.h"
#include "vboot_kernel.h"
#include "vboot_ui_legacy.h"
/* Mock data */
static char debug_info[4096];
static struct vb2_context *ctx;
static struct vb2_shared_data *sd;
static uint8_t workbuf[VB2_KERNEL_WORKBUF_RECOMMENDED_SIZE]
__attribute__((aligned(VB2_WORKBUF_ALIGN)));
static uint32_t mock_localization_count;
static uint32_t mock_altfw_mask;
/* Reset mock data (for use before each test) */
static void ResetMocks(void)
{
mock_localization_count = 3;
mock_altfw_mask = 3 << 1; /* This mask selects 1 and 2 */
TEST_SUCC(vb2api_init(workbuf, sizeof(workbuf), &ctx),
"vb2api_init failed");
vb2_nv_init(ctx);
sd = vb2_get_sd(ctx);
*debug_info = 0;
}
/* Mocks */
uint32_t vb2ex_get_locale_count(void) {
return mock_localization_count;
}
uint32_t VbExGetAltFwIdxMask() {
return mock_altfw_mask;
}
vb2_error_t VbExDisplayDebugInfo(const char *info_str, int full_info)
{
strncpy(debug_info, info_str, sizeof(debug_info));
debug_info[sizeof(debug_info) - 1] = '\0';
return VB2_SUCCESS;
}
vb2_error_t vb2ex_commit_data(struct vb2_context *c)
{
return VB2_SUCCESS;
}
/* Test displaying debug info */
static void DebugInfoTest(void)
{
/* Display debug info */
ResetMocks();
TEST_SUCC(VbDisplayDebugInfo(ctx),
"Display debug info");
TEST_NEQ(*debug_info, '\0', " Some debug info was displayed");
}
/* Test display key checking */
static void DisplayKeyTest(void)
{
ResetMocks();
VbCheckDisplayKey(ctx, 'q', 0, NULL);
TEST_EQ(*debug_info, '\0', "DisplayKey q = does nothing");
ResetMocks();
VbCheckDisplayKey(ctx, '\t', 0, NULL);
TEST_NEQ(*debug_info, '\0', "DisplayKey tab = display");
/* Toggle localization */
ResetMocks();
vb2_nv_set(ctx, VB2_NV_LOCALIZATION_INDEX, 0);
VbCheckDisplayKey(ctx, VB_KEY_DOWN, 0, NULL);
TEST_EQ(vb2_nv_get(ctx, VB2_NV_LOCALIZATION_INDEX), 2,
"DisplayKey up");
VbCheckDisplayKey(ctx, VB_KEY_LEFT, 0, NULL);
vb2_nv_get(ctx, VB2_NV_LOCALIZATION_INDEX);
TEST_EQ(vb2_nv_get(ctx, VB2_NV_LOCALIZATION_INDEX), 1,
"DisplayKey left");
VbCheckDisplayKey(ctx, VB_KEY_RIGHT, 0, NULL);
vb2_nv_get(ctx, VB2_NV_LOCALIZATION_INDEX);
TEST_EQ(vb2_nv_get(ctx, VB2_NV_LOCALIZATION_INDEX), 2,
"DisplayKey right");
VbCheckDisplayKey(ctx, VB_KEY_UP, 0, NULL);
vb2_nv_get(ctx, VB2_NV_LOCALIZATION_INDEX);
TEST_EQ(vb2_nv_get(ctx, VB2_NV_LOCALIZATION_INDEX), 0,
"DisplayKey up");
/* Reset localization if localization count is invalid */
ResetMocks();
vb2_nv_set(ctx, VB2_NV_LOCALIZATION_INDEX, 1);
mock_localization_count = 0;
VbCheckDisplayKey(ctx, VB_KEY_UP, 0, NULL);
TEST_EQ(vb2_nv_get(ctx, VB2_NV_LOCALIZATION_INDEX), 0,
"DisplayKey invalid");
}
int main(void)
{
DebugInfoTest();
DisplayKeyTest();
return gTestSuccess ? 0 : 255;
}
|