log: Add a command to control the log output format
[oweals/u-boot.git] / cmd / log.c
1 /*
2  * Copyright (c) 2017 Google, Inc
3  * Written by Simon Glass <sjg@chromium.org>
4  *
5  * SPDX-License-Identifier:     GPL-2.0+
6  */
7
8 #include <common.h>
9 #include <command.h>
10 #include <dm.h>
11 #include <log.h>
12
13 static char log_fmt_chars[LOGF_COUNT] = "clFLfm";
14
15 static int do_log_level(cmd_tbl_t *cmdtp, int flag, int argc,
16                         char * const argv[])
17 {
18         if (argc > 1)
19                 gd->default_log_level = simple_strtol(argv[1], NULL, 10);
20         else
21                 printf("Default log level: %d\n", gd->default_log_level);
22
23         return 0;
24 }
25
26 static int do_log_format(cmd_tbl_t *cmdtp, int flag, int argc,
27                          char * const argv[])
28 {
29         int i;
30
31         if (argc > 1) {
32                 const char *str = argv[1];
33
34                 if (!strcmp(str, "default")) {
35                         gd->log_fmt = LOGF_DEFAULT;
36                 } else if (!strcmp(str, "all")) {
37                         gd->log_fmt = LOGF_ALL;
38                 } else {
39                         gd->log_fmt = 0;
40                         for (; *str; str++) {
41                                 char *ptr = strchr(log_fmt_chars, *str);
42
43                                 if (!ptr) {
44                                         printf("Invalid log char '%c'\n", *str);
45                                         return CMD_RET_FAILURE;
46                                 }
47                                 gd->log_fmt |= 1 << (ptr - log_fmt_chars);
48                         }
49                 }
50         } else {
51                 printf("Log format: ");
52                 for (i = 0; i < LOGF_COUNT; i++) {
53                         if (gd->log_fmt & (1 << i))
54                                 printf("%c", log_fmt_chars[i]);
55                 }
56                 printf("\n");
57         }
58
59         return 0;
60 }
61
62 static cmd_tbl_t log_sub[] = {
63         U_BOOT_CMD_MKENT(level, CONFIG_SYS_MAXARGS, 1, do_log_level, "", ""),
64 #ifdef CONFIG_LOG_TEST
65         U_BOOT_CMD_MKENT(test, 2, 1, do_log_test, "", ""),
66 #endif
67         U_BOOT_CMD_MKENT(format, CONFIG_SYS_MAXARGS, 1, do_log_format, "", ""),
68 };
69
70 static int do_log(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
71 {
72         cmd_tbl_t *cp;
73
74         if (argc < 2)
75                 return CMD_RET_USAGE;
76
77         /* drop initial "log" arg */
78         argc--;
79         argv++;
80
81         cp = find_cmd_tbl(argv[0], log_sub, ARRAY_SIZE(log_sub));
82         if (cp)
83                 return cp->cmd(cmdtp, flag, argc, argv);
84
85         return CMD_RET_USAGE;
86 }
87
88 #ifdef CONFIG_SYS_LONGHELP
89 static char log_help_text[] =
90         "level - get/set log level\n"
91 #ifdef CONFIG_LOG_TEST
92         "log test - run log tests\n"
93 #endif
94         "log format <fmt> - set log output format. <fmt> is a string where\n"
95         "\teach letter indicates something that should be displayed:\n"
96         "\tc=category, l=level, F=file, L=line number, f=function, m=msg\n"
97         "\tor 'default', equivalent to 'fm', or 'all' for all"
98         ;
99 #endif
100
101 U_BOOT_CMD(
102         log, CONFIG_SYS_MAXARGS, 1, do_log,
103         "log system", log_help_text
104 );