1*4882a593Smuzhiyun /*
2*4882a593Smuzhiyun * Copyright (c) 2011 The Chromium OS Authors.
3*4882a593Smuzhiyun *
4*4882a593Smuzhiyun * SPDX-License-Identifier: GPL-2.0+
5*4882a593Smuzhiyun */
6*4882a593Smuzhiyun
7*4882a593Smuzhiyun #include <common.h>
8*4882a593Smuzhiyun #include <command.h>
9*4882a593Smuzhiyun #include <mapmem.h>
10*4882a593Smuzhiyun #include <trace.h>
11*4882a593Smuzhiyun #include <asm/io.h>
12*4882a593Smuzhiyun
get_args(int argc,char * const argv[],char ** buff,size_t * buff_ptr,size_t * buff_size)13*4882a593Smuzhiyun static int get_args(int argc, char * const argv[], char **buff,
14*4882a593Smuzhiyun size_t *buff_ptr, size_t *buff_size)
15*4882a593Smuzhiyun {
16*4882a593Smuzhiyun if (argc < 2)
17*4882a593Smuzhiyun return -1;
18*4882a593Smuzhiyun if (argc < 4) {
19*4882a593Smuzhiyun *buff_size = env_get_ulong("profsize", 16, 0);
20*4882a593Smuzhiyun *buff = map_sysmem(env_get_ulong("profbase", 16, 0),
21*4882a593Smuzhiyun *buff_size);
22*4882a593Smuzhiyun *buff_ptr = env_get_ulong("profoffset", 16, 0);
23*4882a593Smuzhiyun } else {
24*4882a593Smuzhiyun *buff_size = simple_strtoul(argv[3], NULL, 16);
25*4882a593Smuzhiyun *buff = map_sysmem(simple_strtoul(argv[2], NULL, 16),
26*4882a593Smuzhiyun *buff_size);
27*4882a593Smuzhiyun *buff_ptr = 0;
28*4882a593Smuzhiyun };
29*4882a593Smuzhiyun return 0;
30*4882a593Smuzhiyun }
31*4882a593Smuzhiyun
create_func_list(int argc,char * const argv[])32*4882a593Smuzhiyun static int create_func_list(int argc, char * const argv[])
33*4882a593Smuzhiyun {
34*4882a593Smuzhiyun size_t buff_size, avail, buff_ptr, used;
35*4882a593Smuzhiyun unsigned int needed;
36*4882a593Smuzhiyun char *buff;
37*4882a593Smuzhiyun int err;
38*4882a593Smuzhiyun
39*4882a593Smuzhiyun if (get_args(argc, argv, &buff, &buff_ptr, &buff_size))
40*4882a593Smuzhiyun return -1;
41*4882a593Smuzhiyun
42*4882a593Smuzhiyun avail = buff_size - buff_ptr;
43*4882a593Smuzhiyun err = trace_list_functions(buff + buff_ptr, avail, &needed);
44*4882a593Smuzhiyun if (err)
45*4882a593Smuzhiyun printf("Error: truncated (%#x bytes needed)\n", needed);
46*4882a593Smuzhiyun used = min(avail, (size_t)needed);
47*4882a593Smuzhiyun printf("Function trace dumped to %08lx, size %#zx\n",
48*4882a593Smuzhiyun (ulong)map_to_sysmem(buff + buff_ptr), used);
49*4882a593Smuzhiyun env_set_hex("profbase", map_to_sysmem(buff));
50*4882a593Smuzhiyun env_set_hex("profsize", buff_size);
51*4882a593Smuzhiyun env_set_hex("profoffset", buff_ptr + used);
52*4882a593Smuzhiyun
53*4882a593Smuzhiyun return 0;
54*4882a593Smuzhiyun }
55*4882a593Smuzhiyun
create_call_list(int argc,char * const argv[])56*4882a593Smuzhiyun static int create_call_list(int argc, char * const argv[])
57*4882a593Smuzhiyun {
58*4882a593Smuzhiyun size_t buff_size, avail, buff_ptr, used;
59*4882a593Smuzhiyun unsigned int needed;
60*4882a593Smuzhiyun char *buff;
61*4882a593Smuzhiyun int err;
62*4882a593Smuzhiyun
63*4882a593Smuzhiyun if (get_args(argc, argv, &buff, &buff_ptr, &buff_size))
64*4882a593Smuzhiyun return -1;
65*4882a593Smuzhiyun
66*4882a593Smuzhiyun avail = buff_size - buff_ptr;
67*4882a593Smuzhiyun err = trace_list_calls(buff + buff_ptr, avail, &needed);
68*4882a593Smuzhiyun if (err)
69*4882a593Smuzhiyun printf("Error: truncated (%#x bytes needed)\n", needed);
70*4882a593Smuzhiyun used = min(avail, (size_t)needed);
71*4882a593Smuzhiyun printf("Call list dumped to %08lx, size %#zx\n",
72*4882a593Smuzhiyun (ulong)map_to_sysmem(buff + buff_ptr), used);
73*4882a593Smuzhiyun
74*4882a593Smuzhiyun env_set_hex("profbase", map_to_sysmem(buff));
75*4882a593Smuzhiyun env_set_hex("profsize", buff_size);
76*4882a593Smuzhiyun env_set_hex("profoffset", buff_ptr + used);
77*4882a593Smuzhiyun
78*4882a593Smuzhiyun return 0;
79*4882a593Smuzhiyun }
80*4882a593Smuzhiyun
do_trace(cmd_tbl_t * cmdtp,int flag,int argc,char * const argv[])81*4882a593Smuzhiyun int do_trace(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
82*4882a593Smuzhiyun {
83*4882a593Smuzhiyun const char *cmd = argc < 2 ? NULL : argv[1];
84*4882a593Smuzhiyun
85*4882a593Smuzhiyun if (!cmd)
86*4882a593Smuzhiyun return cmd_usage(cmdtp);
87*4882a593Smuzhiyun switch (*cmd) {
88*4882a593Smuzhiyun case 'p':
89*4882a593Smuzhiyun trace_set_enabled(0);
90*4882a593Smuzhiyun break;
91*4882a593Smuzhiyun case 'c':
92*4882a593Smuzhiyun if (create_call_list(argc, argv))
93*4882a593Smuzhiyun return cmd_usage(cmdtp);
94*4882a593Smuzhiyun break;
95*4882a593Smuzhiyun case 'r':
96*4882a593Smuzhiyun trace_set_enabled(1);
97*4882a593Smuzhiyun break;
98*4882a593Smuzhiyun case 'f':
99*4882a593Smuzhiyun if (create_func_list(argc, argv))
100*4882a593Smuzhiyun return cmd_usage(cmdtp);
101*4882a593Smuzhiyun break;
102*4882a593Smuzhiyun case 's':
103*4882a593Smuzhiyun trace_print_stats();
104*4882a593Smuzhiyun break;
105*4882a593Smuzhiyun default:
106*4882a593Smuzhiyun return CMD_RET_USAGE;
107*4882a593Smuzhiyun }
108*4882a593Smuzhiyun
109*4882a593Smuzhiyun return 0;
110*4882a593Smuzhiyun }
111*4882a593Smuzhiyun
112*4882a593Smuzhiyun U_BOOT_CMD(
113*4882a593Smuzhiyun trace, 4, 1, do_trace,
114*4882a593Smuzhiyun "trace utility commands",
115*4882a593Smuzhiyun "stats - display tracing statistics\n"
116*4882a593Smuzhiyun "trace pause - pause tracing\n"
117*4882a593Smuzhiyun "trace resume - resume tracing\n"
118*4882a593Smuzhiyun "trace funclist [<addr> <size>] - dump function list into buffer\n"
119*4882a593Smuzhiyun "trace calls [<addr> <size>] "
120*4882a593Smuzhiyun "- dump function call trace into buffer"
121*4882a593Smuzhiyun );
122