[J-u-boot.git] / test / print_ut.c

// SPDX-License-Identifier: GPL-2.0+
/*
 * Copyright (c) 2012, The Chromium Authors
 */

#define DEBUG

#include <common.h>
#if CONFIG_IS_ENABLED(EFI_LOADER) && !defined(API_BUILD)
#include <command.h>
#include <efi_api.h>
#endif
#include <display_options.h>
#include <log.h>
#include <version.h>

#define FAKE_BUILD_TAG	"jenkins-u-boot-denx_uboot_dm-master-build-aarch64" \
			"and a lot more text to come"

/* Test printing GUIDs */
static void guid_ut_print(void)
{
#if CONFIG_IS_ENABLED(LIB_UUID)
	unsigned char guid[16] = {
		1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16
	};
	char str[40];

	sprintf(str, "%pUb", guid);
	assert(!strcmp("01020304-0506-0708-090a-0b0c0d0e0f10", str));
	sprintf(str, "%pUB", guid);
	assert(!strcmp("01020304-0506-0708-090A-0B0C0D0E0F10", str));
	sprintf(str, "%pUl", guid);
	assert(!strcmp("04030201-0605-0807-090a-0b0c0d0e0f10", str));
	sprintf(str, "%pUL", guid);
	assert(!strcmp("04030201-0605-0807-090A-0B0C0D0E0F10", str));
#endif
}

/* Test efi_loader specific printing */
static void efi_ut_print(void)
{
#if CONFIG_IS_ENABLED(EFI_LOADER) && !defined(API_BUILD)
	char str[10];
	u8 buf[sizeof(struct efi_device_path_sd_mmc_path) +
	       sizeof(struct efi_device_path)];
	u8 *pos = buf;
	struct efi_device_path *dp_end;
	struct efi_device_path_sd_mmc_path *dp_sd =
			(struct efi_device_path_sd_mmc_path *)pos;

	/* Create a device path for an SD card */
	dp_sd->dp.type = DEVICE_PATH_TYPE_MESSAGING_DEVICE;
	dp_sd->dp.sub_type = DEVICE_PATH_SUB_TYPE_MSG_SD;
	dp_sd->dp.length = sizeof(struct efi_device_path_sd_mmc_path);
	dp_sd->slot_number = 3;
	pos += sizeof(struct efi_device_path_sd_mmc_path);
	/* Append end node */
	dp_end = (struct efi_device_path *)pos;
	dp_end->type = DEVICE_PATH_TYPE_END;
	dp_end->sub_type = DEVICE_PATH_SUB_TYPE_END;
	dp_end->length = sizeof(struct efi_device_path);

	snprintf(str, sizeof(str), "_%pD_", buf);
	assert(!strcmp("_/SD(3)_", str));

	/* NULL device path */
	snprintf(str, sizeof(str), "_%pD_", NULL);
	assert(!strcmp("_<NULL>_", str));
#endif
}

static int do_ut_print(struct cmd_tbl *cmdtp, int flag, int argc,
		       char *const argv[])
{
	char big_str[400];
	int big_str_len;
	char str[10], *s;
	int len;

	printf("%s: Testing print\n", __func__);

	snprintf(str, sizeof(str), "testing");
	assert(!strcmp("testing", str));

	snprintf(str, sizeof(str), "testing but too long");
	assert(!strcmp("testing b", str));

	snprintf(str, 1, "testing none");
	assert(!strcmp("", str));

	*str = 'x';
	snprintf(str, 0, "testing none");
	assert(*str == 'x');

	sprintf(big_str, "_%ls_", L"foo");
	assert(!strcmp("_foo_", big_str));

	/* Test the banner function */
	s = display_options_get_banner(true, str, sizeof(str));
	assert(s == str);
	assert(!strcmp("\n\nU-Boo\n\n", s));

	/* Assert that we do not overwrite memory before the buffer */
	str[0] = '`';
	s = display_options_get_banner(true, str + 1, 1);
	assert(s == str + 1);
	assert(!strcmp("`", str));

	str[0] = '~';
	s = display_options_get_banner(true, str + 1, 2);
	assert(s == str + 1);
	assert(!strcmp("~\n", str));

	/* The last two characters are set to \n\n for all buffer sizes > 2 */
	s = display_options_get_banner(false, str, sizeof(str));
	assert(s == str);
	assert(!strcmp("U-Boot \n\n", s));

	/* Give it enough space for some of the version */
	big_str_len = strlen(version_string) - 5;
	s = display_options_get_banner_priv(false, FAKE_BUILD_TAG, big_str,
					    big_str_len);
	assert(s == big_str);
	assert(!strncmp(version_string, s, big_str_len - 3));
	assert(!strcmp("\n\n", s + big_str_len - 3));

	/* Give it enough space for the version and some of the build tag */
	big_str_len = strlen(version_string) + 9 + 20;
	s = display_options_get_banner_priv(false, FAKE_BUILD_TAG, big_str,
					    big_str_len);
	assert(s == big_str);
	len = strlen(version_string);
	assert(!strncmp(version_string, s, len));
	assert(!strncmp(", Build: ", s + len, 9));
	assert(!strncmp(FAKE_BUILD_TAG, s + 9 + len, 12));
	assert(!strcmp("\n\n", s + big_str_len - 3));

	/* Test efi_loader specific printing */
	efi_ut_print();

	/* Test printing GUIDs */
	guid_ut_print();

	printf("%s: Everything went swimmingly\n", __func__);
	return 0;
}

U_BOOT_CMD(
	ut_print,	1,	1,	do_ut_print,
	"Very basic test of printf(), etc.",
	""
);
Commit	Line	Data
83d290c5	1	// SPDX-License-Identifier: GPL-2.0+
e5a9d27f SG	2	/*
e5a9d27f SG	3	* Copyright (c) 2012, The Chromium Authors
e5a9d27f SG	4	*/
	5
	6	#define DEBUG
	7
	8	#include <common.h>
7a9e6ee6	9	#if CONFIG_IS_ENABLED(EFI_LOADER) && !defined(API_BUILD)
09140113	10	#include <command.h>
256060e4 HS	11	#include <efi_api.h>
256060e4 HS	12	#endif
e5a9d27f	13	#include <display_options.h>
f7ae49fc	14	#include <log.h>
e5a9d27f SG	15	#include <version.h>
	16
	17	#define FAKE_BUILD_TAG "jenkins-u-boot-denx_uboot_dm-master-build-aarch64" \
	18	"and a lot more text to come"
	19
3bad256f HS	20	/* Test printing GUIDs */
	21	static void guid_ut_print(void)
	22	{
	23	#if CONFIG_IS_ENABLED(LIB_UUID)
	24	unsigned char guid[16] = {
	25	1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16
	26	};
	27	char str[40];
	28
	29	sprintf(str, "%pUb", guid);
	30	assert(!strcmp("01020304-0506-0708-090a-0b0c0d0e0f10", str));
	31	sprintf(str, "%pUB", guid);
	32	assert(!strcmp("01020304-0506-0708-090A-0B0C0D0E0F10", str));
	33	sprintf(str, "%pUl", guid);
	34	assert(!strcmp("04030201-0605-0807-090a-0b0c0d0e0f10", str));
	35	sprintf(str, "%pUL", guid);
	36	assert(!strcmp("04030201-0605-0807-090A-0B0C0D0E0F10", str));
	37	#endif
	38	}
	39
256060e4 HS	40	/* Test efi_loader specific printing */
	41	static void efi_ut_print(void)
	42	{
7a9e6ee6	43	#if CONFIG_IS_ENABLED(EFI_LOADER) && !defined(API_BUILD)
256060e4 HS	44	char str[10];
	45	u8 buf[sizeof(struct efi_device_path_sd_mmc_path) +
	46	sizeof(struct efi_device_path)];
	47	u8 *pos = buf;
	48	struct efi_device_path *dp_end;
	49	struct efi_device_path_sd_mmc_path *dp_sd =
	50	(struct efi_device_path_sd_mmc_path *)pos;
	51
	52	/* Create a device path for an SD card */
	53	dp_sd->dp.type = DEVICE_PATH_TYPE_MESSAGING_DEVICE;
	54	dp_sd->dp.sub_type = DEVICE_PATH_SUB_TYPE_MSG_SD;
	55	dp_sd->dp.length = sizeof(struct efi_device_path_sd_mmc_path);
	56	dp_sd->slot_number = 3;
	57	pos += sizeof(struct efi_device_path_sd_mmc_path);
	58	/* Append end node */
	59	dp_end = (struct efi_device_path *)pos;
	60	dp_end->type = DEVICE_PATH_TYPE_END;
	61	dp_end->sub_type = DEVICE_PATH_SUB_TYPE_END;
	62	dp_end->length = sizeof(struct efi_device_path);
	63
	64	snprintf(str, sizeof(str), "_%pD_", buf);
	65	assert(!strcmp("_/SD(3)_", str));
5f1ce1d4 HS	66
	67	/* NULL device path */
	68	snprintf(str, sizeof(str), "_%pD_", NULL);
	69	assert(!strcmp("_<NULL>_", str));
256060e4 HS	70	#endif
	71	}
	72
09140113	73	static int do_ut_print(struct cmd_tbl *cmdtp, int flag, int argc,
e5a9d27f SG	74	char *const argv[])
	75	{
	76	char big_str[400];
	77	int big_str_len;
	78	char str[10], *s;
	79	int len;
	80
	81	printf("%s: Testing print\n", __func__);
	82
	83	snprintf(str, sizeof(str), "testing");
	84	assert(!strcmp("testing", str));
	85
	86	snprintf(str, sizeof(str), "testing but too long");
	87	assert(!strcmp("testing b", str));
	88
	89	snprintf(str, 1, "testing none");
	90	assert(!strcmp("", str));
	91
	92	*str = 'x';
	93	snprintf(str, 0, "testing none");
	94	assert(*str == 'x');
	95
085391b2 RC	96	sprintf(big_str, "_%ls_", L"foo");
	97	assert(!strcmp("_foo_", big_str));
	98
e5a9d27f SG	99	/* Test the banner function */
	100	s = display_options_get_banner(true, str, sizeof(str));
	101	assert(s == str);
	102	assert(!strcmp("\n\nU-Boo\n\n", s));
	103
6c74e94a HS	104	/* Assert that we do not overwrite memory before the buffer */
	105	str[0] = '`';
	106	s = display_options_get_banner(true, str + 1, 1);
	107	assert(s == str + 1);
	108	assert(!strcmp("`", str));
	109
	110	str[0] = '~';
	111	s = display_options_get_banner(true, str + 1, 2);
	112	assert(s == str + 1);
	113	assert(!strcmp("~\n", str));
	114
	115	/* The last two characters are set to \n\n for all buffer sizes > 2 */
e5a9d27f SG	116	s = display_options_get_banner(false, str, sizeof(str));
	117	assert(s == str);
	118	assert(!strcmp("U-Boot \n\n", s));
	119
	120	/* Give it enough space for some of the version */
	121	big_str_len = strlen(version_string) - 5;
	122	s = display_options_get_banner_priv(false, FAKE_BUILD_TAG, big_str,
	123	big_str_len);
	124	assert(s == big_str);
	125	assert(!strncmp(version_string, s, big_str_len - 3));
	126	assert(!strcmp("\n\n", s + big_str_len - 3));
	127
	128	/* Give it enough space for the version and some of the build tag */
	129	big_str_len = strlen(version_string) + 9 + 20;
	130	s = display_options_get_banner_priv(false, FAKE_BUILD_TAG, big_str,
	131	big_str_len);
	132	assert(s == big_str);
	133	len = strlen(version_string);
	134	assert(!strncmp(version_string, s, len));
	135	assert(!strncmp(", Build: ", s + len, 9));
	136	assert(!strncmp(FAKE_BUILD_TAG, s + 9 + len, 12));
	137	assert(!strcmp("\n\n", s + big_str_len - 3));
	138
256060e4 HS	139	/* Test efi_loader specific printing */
	140	efi_ut_print();
	141
3bad256f HS	142	/* Test printing GUIDs */
	143	guid_ut_print();
	144
e5a9d27f SG	145	printf("%s: Everything went swimmingly\n", __func__);
	146	return 0;
	147	}
	148
	149	U_BOOT_CMD(
	150	ut_print, 1, 1, do_ut_print,
	151	"Very basic test of printf(), etc.",
	152	""
	153	);