Add a simple test to make sure that these functions obey the buffer size passed into them. Signed-off-by: Simon Glass <sjg@chromium.org> Reviewed-by: Bin Meng <bmeng.cn@gmail.com> Tested-by: Stephen Warren <swarren@nvidia.com>master
parent
b0895384be
commit
e5a9d27fdb
@ -0,0 +1,83 @@ |
||||
/*
|
||||
* Copyright (c) 2012, The Chromium Authors |
||||
* |
||||
* SPDX-License-Identifier: GPL-2.0+ |
||||
*/ |
||||
|
||||
#define DEBUG |
||||
|
||||
#include <common.h> |
||||
#include <display_options.h> |
||||
#include <version.h> |
||||
|
||||
#define FAKE_BUILD_TAG "jenkins-u-boot-denx_uboot_dm-master-build-aarch64" \ |
||||
"and a lot more text to come" |
||||
|
||||
static int do_ut_print(cmd_tbl_t *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'); |
||||
|
||||
/* Test the banner function */ |
||||
s = display_options_get_banner(true, str, sizeof(str)); |
||||
assert(s == str); |
||||
assert(!strcmp("\n\nU-Boo\n\n", s)); |
||||
|
||||
s = display_options_get_banner(true, str, 1); |
||||
assert(s == str); |
||||
assert(!strcmp("", s)); |
||||
|
||||
s = display_options_get_banner(true, str, 2); |
||||
assert(s == str); |
||||
assert(!strcmp("\n", s)); |
||||
|
||||
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)); |
||||
|
||||
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.", |
||||
"" |
||||
); |
Loading…
Reference in new issue