1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3 * Copyright (c) 2012, The Chromium Authors
4 */
5
6 #define DEBUG
7
8 #include <common.h>
9 #if CONFIG_IS_ENABLED(EFI_LOADER) && !defined(API_BUILD)
10 #include <efi_api.h>
11 #endif
12 #include <display_options.h>
13 #include <version_string.h>
14
15 #define FAKE_BUILD_TAG "jenkins-u-boot-denx_uboot_dm-master-build-aarch64" \
16 "and a lot more text to come"
17
18 /* Test efi_loader specific printing */
efi_ut_print(void)19 static void efi_ut_print(void)
20 {
21 #if CONFIG_IS_ENABLED(EFI_LOADER) && !defined(API_BUILD)
22 char str[10];
23 u8 buf[sizeof(struct efi_device_path_sd_mmc_path) +
24 sizeof(struct efi_device_path)];
25 u8 *pos = buf;
26 struct efi_device_path *dp_end;
27 struct efi_device_path_sd_mmc_path *dp_sd =
28 (struct efi_device_path_sd_mmc_path *)pos;
29
30 /* Create a device path for an SD card */
31 dp_sd->dp.type = DEVICE_PATH_TYPE_MESSAGING_DEVICE;
32 dp_sd->dp.sub_type = DEVICE_PATH_SUB_TYPE_MSG_SD;
33 dp_sd->dp.length = sizeof(struct efi_device_path_sd_mmc_path);
34 dp_sd->slot_number = 3;
35 pos += sizeof(struct efi_device_path_sd_mmc_path);
36 /* Append end node */
37 dp_end = (struct efi_device_path *)pos;
38 dp_end->type = DEVICE_PATH_TYPE_END;
39 dp_end->sub_type = DEVICE_PATH_SUB_TYPE_END;
40 dp_end->length = sizeof(struct efi_device_path);
41
42 snprintf(str, sizeof(str), "_%pD_", buf);
43 assert(!strcmp("_/SD(3)_", str));
44
45 /* NULL device path */
46 snprintf(str, sizeof(str), "_%pD_", NULL);
47 assert(!strcmp("_<NULL>_", str));
48 #endif
49 }
50
do_ut_print(cmd_tbl_t * cmdtp,int flag,int argc,char * const argv[])51 static int do_ut_print(cmd_tbl_t *cmdtp, int flag, int argc,
52 char *const argv[])
53 {
54 char big_str[400];
55 int big_str_len;
56 char str[10], *s;
57 int len;
58
59 printf("%s: Testing print\n", __func__);
60
61 snprintf(str, sizeof(str), "testing");
62 assert(!strcmp("testing", str));
63
64 snprintf(str, sizeof(str), "testing but too long");
65 assert(!strcmp("testing b", str));
66
67 snprintf(str, 1, "testing none");
68 assert(!strcmp("", str));
69
70 *str = 'x';
71 snprintf(str, 0, "testing none");
72 assert(*str == 'x');
73
74 sprintf(big_str, "_%ls_", L"foo");
75 assert(!strcmp("_foo_", big_str));
76
77 /* Test the banner function */
78 s = display_options_get_banner(true, str, sizeof(str));
79 assert(s == str);
80 assert(!strcmp("\n\nU-Boo\n\n", s));
81
82 s = display_options_get_banner(true, str, 1);
83 assert(s == str);
84 assert(!strcmp("", s));
85
86 s = display_options_get_banner(true, str, 2);
87 assert(s == str);
88 assert(!strcmp("\n", s));
89
90 s = display_options_get_banner(false, str, sizeof(str));
91 assert(s == str);
92 assert(!strcmp("U-Boot \n\n", s));
93
94 /* Give it enough space for some of the version */
95 big_str_len = strlen(version_string) - 5;
96 s = display_options_get_banner_priv(false, FAKE_BUILD_TAG, big_str,
97 big_str_len);
98 assert(s == big_str);
99 assert(!strncmp(version_string, s, big_str_len - 3));
100 assert(!strcmp("\n\n", s + big_str_len - 3));
101
102 /* Give it enough space for the version and some of the build tag */
103 big_str_len = strlen(version_string) + 9 + 20;
104 s = display_options_get_banner_priv(false, FAKE_BUILD_TAG, big_str,
105 big_str_len);
106 assert(s == big_str);
107 len = strlen(version_string);
108 assert(!strncmp(version_string, s, len));
109 assert(!strncmp(", Build: ", s + len, 9));
110 assert(!strncmp(FAKE_BUILD_TAG, s + 9 + len, 12));
111 assert(!strcmp("\n\n", s + big_str_len - 3));
112
113 /* Test efi_loader specific printing */
114 efi_ut_print();
115
116 printf("%s: Everything went swimmingly\n", __func__);
117 return 0;
118 }
119
120 U_BOOT_CMD(
121 ut_print, 1, 1, do_ut_print,
122 "Very basic test of printf(), etc.",
123 ""
124 );
125