test/print_ut.c

   1 // SPDX-License-Identifier: GPL-2.0+
   2 /*
   3  * Copyright (c) 2012, The Chromium Authors
   4  */
   5
   6 #include <common.h>
   7 #include <command.h>
   8 #include <efi_api.h>
   9 #include <display_options.h>
  10 #include <log.h>
  11 #include <mapmem.h>
  12 #include <version.h>
  13 #include <test/suites.h>
  14 #include <test/test.h>
  15 #include <test/ut.h>
  16
  17 #define BUF_SIZE        0x100
  18
  19 #define FAKE_BUILD_TAG  "jenkins-u-boot-denx_uboot_dm-master-build-aarch64" \
  20                         "and a lot more text to come"
  21
  22 /* Declare a new print test */
  23 #define PRINT_TEST(_name, _flags)       UNIT_TEST(_name, _flags, print_test)
  24
  25 #if CONFIG_IS_ENABLED(LIB_UUID)
  26 /* Test printing GUIDs */
  27 static int print_guid(struct unit_test_state *uts)
  28 {
  29         unsigned char guid[16] = {
  30                 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16
  31         };
  32         char str[40];
  33
  34         sprintf(str, "%pUb", guid);
  35         ut_assertok(strcmp("01020304-0506-0708-090a-0b0c0d0e0f10", str));
  36         sprintf(str, "%pUB", guid);
  37         ut_assertok(strcmp("01020304-0506-0708-090A-0B0C0D0E0F10", str));
  38         sprintf(str, "%pUl", guid);
  39         ut_assertok(strcmp("04030201-0605-0807-090a-0b0c0d0e0f10", str));
  40         sprintf(str, "%pUL", guid);
  41         ut_assertok(strcmp("04030201-0605-0807-090A-0B0C0D0E0F10", str));
  42
  43         return 0;
  44 }
  45 PRINT_TEST(print_guid, 0);
  46 #endif
  47
  48 #if CONFIG_IS_ENABLED(EFI_LOADER) && !defined(API_BUILD)
  49 /* Test efi_loader specific printing */
  50 static int print_efi_ut(struct unit_test_state *uts)
  51 {
  52         char str[10];
  53         u8 buf[sizeof(struct efi_device_path_sd_mmc_path) +
  54                sizeof(struct efi_device_path)];
  55         u8 *pos = buf;
  56         struct efi_device_path *dp_end;
  57         struct efi_device_path_sd_mmc_path *dp_sd =
  58                         (struct efi_device_path_sd_mmc_path *)pos;
  59
  60         /* Create a device path for an SD card */
  61         dp_sd->dp.type = DEVICE_PATH_TYPE_MESSAGING_DEVICE;
  62         dp_sd->dp.sub_type = DEVICE_PATH_SUB_TYPE_MSG_SD;
  63         dp_sd->dp.length = sizeof(struct efi_device_path_sd_mmc_path);
  64         dp_sd->slot_number = 3;
  65         pos += sizeof(struct efi_device_path_sd_mmc_path);
  66         /* Append end node */
  67         dp_end = (struct efi_device_path *)pos;
  68         dp_end->type = DEVICE_PATH_TYPE_END;
  69         dp_end->sub_type = DEVICE_PATH_SUB_TYPE_END;
  70         dp_end->length = sizeof(struct efi_device_path);
  71
  72         snprintf(str, sizeof(str), "_%pD_", buf);
  73         ut_assertok(strcmp("_/SD(3)_", str));
  74
  75         /* NULL device path */
  76         snprintf(str, sizeof(str), "_%pD_", NULL);
  77         ut_assertok(strcmp("_<NULL>_", str));
  78
  79         return 0;
  80 }
  81 PRINT_TEST(print_efi_ut, 0);
  82 #endif
  83
  84 static int print_printf(struct unit_test_state *uts)
  85 {
  86         char big_str[400];
  87         int big_str_len;
  88         char str[10], *s;
  89         int len;
  90
  91         snprintf(str, sizeof(str), "testing");
  92         ut_assertok(strcmp("testing", str));
  93
  94         snprintf(str, sizeof(str), "testing but too long");
  95         ut_assertok(strcmp("testing b", str));
  96
  97         snprintf(str, 1, "testing none");
  98         ut_assertok(strcmp("", str));
  99
 100         *str = 'x';
 101         snprintf(str, 0, "testing none");
 102         ut_asserteq('x', *str);
 103
 104         sprintf(big_str, "_%ls_", L"foo");
 105         ut_assertok(strcmp("_foo_", big_str));
 106
 107         /* Test the banner function */
 108         s = display_options_get_banner(true, str, sizeof(str));
 109         ut_asserteq_ptr(str, s);
 110         ut_assertok(strcmp("\n\nU-Boo\n\n", s));
 111
 112         /* Assert that we do not overwrite memory before the buffer */
 113         str[0] = '`';
 114         s = display_options_get_banner(true, str + 1, 1);
 115         ut_asserteq_ptr(str + 1, s);
 116         ut_assertok(strcmp("`", str));
 117
 118         str[0] = '~';
 119         s = display_options_get_banner(true, str + 1, 2);
 120         ut_asserteq_ptr(str + 1, s);
 121         ut_assertok(strcmp("~\n", str));
 122
 123         /* The last two characters are set to \n\n for all buffer sizes > 2 */
 124         s = display_options_get_banner(false, str, sizeof(str));
 125         ut_asserteq_ptr(str, s);
 126         ut_assertok(strcmp("U-Boot \n\n", s));
 127
 128         /* Give it enough space for some of the version */
 129         big_str_len = strlen(version_string) - 5;
 130         s = display_options_get_banner_priv(false, FAKE_BUILD_TAG, big_str,
 131                                             big_str_len);
 132         ut_asserteq_ptr(big_str, s);
 133         ut_assertok(strncmp(version_string, s, big_str_len - 3));
 134         ut_assertok(strcmp("\n\n", s + big_str_len - 3));
 135
 136         /* Give it enough space for the version and some of the build tag */
 137         big_str_len = strlen(version_string) + 9 + 20;
 138         s = display_options_get_banner_priv(false, FAKE_BUILD_TAG, big_str,
 139                                             big_str_len);
 140         ut_asserteq_ptr(big_str, s);
 141         len = strlen(version_string);
 142         ut_assertok(strncmp(version_string, s, len));
 143         ut_assertok(strncmp(", Build: ", s + len, 9));
 144         ut_assertok(strncmp(FAKE_BUILD_TAG, s + 9 + len, 12));
 145         ut_assertok(strcmp("\n\n", s + big_str_len - 3));
 146
 147         return 0;
 148 }
 149 PRINT_TEST(print_printf, 0);
 150
 151 static int print_display_buffer(struct unit_test_state *uts)
 152 {
 153         u8 *buf;
 154         int i;
 155
 156         buf = map_sysmem(0, BUF_SIZE);
 157         memset(buf, '\0', BUF_SIZE);
 158         for (i = 0; i < 0x11; i++)
 159                 buf[i] = i * 0x11;
 160
 161         /* bytes */
 162         console_record_reset();
 163         print_buffer(0, buf, 1, 0x12, 0);
 164         ut_assert_nextline("00000000: 00 11 22 33 44 55 66 77 88 99 aa bb cc dd ee ff  ..\"3DUfw........");
 165         ut_assert_nextline("00000010: 10 00                                            ..");
 166         ut_assert_console_end();
 167
 168         /* line length */
 169         console_record_reset();
 170         print_buffer(0, buf, 1, 0x12, 8);
 171         ut_assert_nextline("00000000: 00 11 22 33 44 55 66 77  ..\"3DUfw");
 172         ut_assert_nextline("00000008: 88 99 aa bb cc dd ee ff  ........");
 173         ut_assert_nextline("00000010: 10 00                    ..");
 174         ut_assert_console_end();
 175
 176         /* long line */
 177         console_record_reset();
 178         buf[0x41] = 0x41;
 179         print_buffer(0, buf, 1, 0x42, 0x40);
 180         ut_assert_nextline("00000000: 00 11 22 33 44 55 66 77 88 99 aa bb cc dd ee ff 10 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00  ..\"3DUfw........................................................");
 181         ut_assert_nextline("00000040: 00 41                                                                                                                                                                                            .A");
 182         ut_assert_console_end();
 183
 184         /* address */
 185         console_record_reset();
 186         print_buffer(0x12345678, buf, 1, 0x12, 0);
 187         ut_assert_nextline("12345678: 00 11 22 33 44 55 66 77 88 99 aa bb cc dd ee ff  ..\"3DUfw........");
 188         ut_assert_nextline("12345688: 10 00                                            ..");
 189         ut_assert_console_end();
 190
 191         /* 16-bit */
 192         console_record_reset();
 193         print_buffer(0, buf, 2, 9, 0);
 194         ut_assert_nextline("00000000: 1100 3322 5544 7766 9988 bbaa ddcc ffee  ..\"3DUfw........");
 195         ut_assert_nextline("00000010: 0010                                     ..");
 196         ut_assert_console_end();
 197
 198         /* 32-bit */
 199         console_record_reset();
 200         print_buffer(0, buf, 4, 5, 0);
 201         ut_assert_nextline("00000000: 33221100 77665544 bbaa9988 ffeeddcc  ..\"3DUfw........");
 202         ut_assert_nextline("00000010: 00000010                             ....");
 203         ut_assert_console_end();
 204
 205         /* 64-bit */
 206         console_record_reset();
 207         print_buffer(0, buf, 8, 3, 0);
 208         ut_assert_nextline("00000000: 7766554433221100 ffeeddccbbaa9988  ..\"3DUfw........");
 209         ut_assert_nextline("00000010: 0000000000000010                   ........");
 210         ut_assert_console_end();
 211
 212         /* ASCII */
 213         console_record_reset();
 214         buf[1] = 31;
 215         buf[2] = 32;
 216         buf[3] = 33;
 217         for (i = 0; i < 4; i++)
 218                 buf[4 + i] = 126 + i;
 219         buf[8] = 255;
 220         print_buffer(0, buf, 1, 10, 0);
 221         ut_assert_nextline("00000000: 00 1f 20 21 7e 7f 80 81 ff 99                    .. !~.....");
 222         ut_assert_console_end();
 223
 224         unmap_sysmem(buf);
 225
 226         return 0;
 227 }
 228 PRINT_TEST(print_display_buffer, UT_TESTF_CONSOLE_REC);
 229
 230 static int print_do_hex_dump(struct unit_test_state *uts)
 231 {
 232         u8 *buf;
 233         int i;
 234
 235         buf = map_sysmem(0, BUF_SIZE);
 236         memset(buf, '\0', BUF_SIZE);
 237         for (i = 0; i < 0x11; i++)
 238                 buf[i] = i * 0x11;
 239
 240         /* bytes */
 241         console_record_reset();
 242         print_hex_dump_bytes("", DUMP_PREFIX_ADDRESS, buf, 0x12);
 243         ut_assert_nextline("00000000: 00 11 22 33 44 55 66 77 88 99 aa bb cc dd ee ff  ..\"3DUfw........");
 244         ut_assert_nextline("00000010: 10 00                                            ..");
 245         ut_assert_console_end();
 246
 247         /* 16-bit */
 248         console_record_reset();
 249         print_hex_dump("", DUMP_PREFIX_ADDRESS, 16, 2, buf, 0x12, true);
 250         ut_assert_nextline("00000000: 1100 3322 5544 7766 9988 bbaa ddcc ffee  ..\"3DUfw........");
 251         ut_assert_nextline("00000010: 0010                                     ..");
 252         ut_assert_console_end();
 253         unmap_sysmem(buf);
 254
 255         /* 32-bit */
 256         console_record_reset();
 257         print_hex_dump("", DUMP_PREFIX_ADDRESS, 16, 4, buf, 0x14, true);
 258         ut_assert_nextline("00000000: 33221100 77665544 bbaa9988 ffeeddcc  ..\"3DUfw........");
 259         ut_assert_nextline("00000010: 00000010                             ....");
 260         ut_assert_console_end();
 261         unmap_sysmem(buf);
 262
 263         /* 64-bit */
 264         console_record_reset();
 265         print_hex_dump("", DUMP_PREFIX_ADDRESS, 16, 8, buf, 0x18, true);
 266         ut_assert_nextline("00000000: 7766554433221100 ffeeddccbbaa9988  ..\"3DUfw........");
 267         ut_assert_nextline("00000010: 0000000000000010                   ........");
 268         ut_assert_console_end();
 269         unmap_sysmem(buf);
 270
 271         /* ASCII */
 272         console_record_reset();
 273         buf[1] = 31;
 274         buf[2] = 32;
 275         buf[3] = 33;
 276         for (i = 0; i < 4; i++)
 277                 buf[4 + i] = 126 + i;
 278         buf[8] = 255;
 279         print_hex_dump("", DUMP_PREFIX_ADDRESS, 16, 1, buf, 10, true);
 280         ut_assert_nextline("00000000: 00 1f 20 21 7e 7f 80 81 ff 99                    .. !~.....");
 281         ut_assert_console_end();
 282         unmap_sysmem(buf);
 283
 284         return 0;
 285 }
 286 PRINT_TEST(print_do_hex_dump, UT_TESTF_CONSOLE_REC);
 287
 288 int do_ut_print(struct cmd_tbl *cmdtp, int flag, int argc, char *const argv[])
 289 {
 290         struct unit_test *tests = UNIT_TEST_SUITE_START(print_test);
 291         const int n_ents = UNIT_TEST_SUITE_COUNT(print_test);
 292
 293         return cmd_ut_category("print", "print_", tests, n_ents, argc, argv);
 294 }