display_options: Refactor to allow obtaining the banner

Move the display options code into a separate function so that the U-Boot
banner can be obtained from other code. Adjust the 'version' command to
use it.

Signed-off-by: Simon Glass <sjg@chromium.org>
Reviewed-by: Bin Meng <bmeng.cn@gmail.com>
Tested-by: Bin Meng <bmeng.cn@gmail.com>
Tested-by: Stephen Warren <swarren@nvidia.com>
This commit is contained in:
Simon Glass 2017-06-16 12:51:42 -06:00
parent 32e9ec1f88
commit 6c519f2dc4
3 changed files with 53 additions and 6 deletions

View file

@ -17,7 +17,9 @@ const char __weak version_string[] = U_BOOT_VERSION_STRING;
static int do_version(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
{
printf("\n%s\n", version_string);
char buf[DISPLAY_OPTIONS_BANNER_LENGTH];
printf(display_options_get_banner(false, buf, sizeof(buf)));
#ifdef CC_VERSION_STRING
puts(CC_VERSION_STRING "\n");
#endif

View file

@ -56,4 +56,23 @@ int print_buffer(ulong addr, const void *data, uint width, uint count,
*/
int display_options(void);
/* Suggested length of the buffer to pass to display_options_get_banner() */
#define DISPLAY_OPTIONS_BANNER_LENGTH 200
/**
* display_options_get_banner() - Get the U-Boot banner as a string
*
* This returns the U-Boot banner string
*
* @newlines: true to include two newlines at the start
* @buf: place to put string
* @size: Size of buf (string is truncated to fit)
* @return buf
*/
char *display_options_get_banner(bool newlines, char *buf, int size);
/* This function is used for testing only */
char *display_options_get_banner_priv(bool newlines, const char *build_tag,
char *buf, int size);
#endif

View file

@ -13,13 +13,39 @@
#include <linux/ctype.h>
#include <asm/io.h>
int display_options (void)
char *display_options_get_banner_priv(bool newlines, const char *build_tag,
char *buf, int size)
{
#if defined(BUILD_TAG)
printf ("\n\n%s, Build: %s\n\n", version_string, BUILD_TAG);
#else
printf ("\n\n%s\n\n", version_string);
int len;
len = snprintf(buf, size, "%s%s", newlines ? "\n\n" : "",
version_string);
if (build_tag && len < size)
len += snprintf(buf + len, size - len, ", Build: %s",
build_tag);
if (len > size - 3)
len = size - 3;
strcpy(buf + len, "\n\n");
return buf;
}
#ifndef BUILD_TAG
#define BUILD_TAG NULL
#endif
char *display_options_get_banner(bool newlines, char *buf, int size)
{
return display_options_get_banner_priv(newlines, BUILD_TAG, buf, size);
}
int display_options(void)
{
char buf[DISPLAY_OPTIONS_BANNER_LENGTH];
display_options_get_banner(true, buf, sizeof(buf));
printf("%s", buf);
return 0;
}