1 #include <elf.h>
2 #include <inttypes.h>
3 #include <sys/ttydefaults.h>
4 #include <string.h>
5 #include "../../util/sort.h"
6 #include "../../util/util.h"
7 #include "../../util/hist.h"
8 #include "../../util/debug.h"
9 #include "../../util/symbol.h"
10 #include "../browser.h"
11 #include "../helpline.h"
12 #include "../libslang.h"
13
14 /* 2048 lines should be enough for a script output */
15 #define MAX_LINES 2048
16
17 /* 160 bytes for one output line */
18 #define AVERAGE_LINE_LEN 160
19
20 struct script_line {
21 struct list_head node;
22 char line[AVERAGE_LINE_LEN];
23 };
24
25 struct perf_script_browser {
26 struct ui_browser b;
27 struct list_head entries;
28 const char *script_name;
29 int nr_lines;
30 };
31
32 #define SCRIPT_NAMELEN 128
33 #define SCRIPT_MAX_NO 64
34 /*
35 * Usually the full path for a script is:
36 * /home/username/libexec/perf-core/scripts/python/xxx.py
37 * /home/username/libexec/perf-core/scripts/perl/xxx.pl
38 * So 256 should be long enough to contain the full path.
39 */
40 #define SCRIPT_FULLPATH_LEN 256
41
42 /*
43 * When success, will copy the full path of the selected script
44 * into the buffer pointed by script_name, and return 0.
45 * Return -1 on failure.
46 */
list_scripts(char * script_name)47 static int list_scripts(char *script_name)
48 {
49 char *buf, *names[SCRIPT_MAX_NO], *paths[SCRIPT_MAX_NO];
50 int i, num, choice, ret = -1;
51
52 /* Preset the script name to SCRIPT_NAMELEN */
53 buf = malloc(SCRIPT_MAX_NO * (SCRIPT_NAMELEN + SCRIPT_FULLPATH_LEN));
54 if (!buf)
55 return ret;
56
57 for (i = 0; i < SCRIPT_MAX_NO; i++) {
58 names[i] = buf + i * (SCRIPT_NAMELEN + SCRIPT_FULLPATH_LEN);
59 paths[i] = names[i] + SCRIPT_NAMELEN;
60 }
61
62 num = find_scripts(names, paths);
63 if (num > 0) {
64 choice = ui__popup_menu(num, names);
65 if (choice < num && choice >= 0) {
66 strcpy(script_name, paths[choice]);
67 ret = 0;
68 }
69 }
70
71 free(buf);
72 return ret;
73 }
74
script_browser__write(struct ui_browser * browser,void * entry,int row)75 static void script_browser__write(struct ui_browser *browser,
76 void *entry, int row)
77 {
78 struct script_line *sline = list_entry(entry, struct script_line, node);
79 bool current_entry = ui_browser__is_current_entry(browser, row);
80
81 ui_browser__set_color(browser, current_entry ? HE_COLORSET_SELECTED :
82 HE_COLORSET_NORMAL);
83
84 slsmg_write_nstring(sline->line, browser->width);
85 }
86
script_browser__run(struct perf_script_browser * self)87 static int script_browser__run(struct perf_script_browser *self)
88 {
89 int key;
90
91 if (ui_browser__show(&self->b, self->script_name,
92 "Press <- or ESC to exit") < 0)
93 return -1;
94
95 while (1) {
96 key = ui_browser__run(&self->b, 0);
97
98 /* We can add some special key handling here if needed */
99 break;
100 }
101
102 ui_browser__hide(&self->b);
103 return key;
104 }
105
106
script_browse(const char * script_opt)107 int script_browse(const char *script_opt)
108 {
109 char cmd[SCRIPT_FULLPATH_LEN*2], script_name[SCRIPT_FULLPATH_LEN];
110 char *line = NULL;
111 size_t len = 0;
112 ssize_t retlen;
113 int ret = -1, nr_entries = 0;
114 FILE *fp;
115 void *buf;
116 struct script_line *sline;
117
118 struct perf_script_browser script = {
119 .b = {
120 .refresh = ui_browser__list_head_refresh,
121 .seek = ui_browser__list_head_seek,
122 .write = script_browser__write,
123 },
124 .script_name = script_name,
125 };
126
127 INIT_LIST_HEAD(&script.entries);
128
129 /* Save each line of the output in one struct script_line object. */
130 buf = zalloc((sizeof(*sline)) * MAX_LINES);
131 if (!buf)
132 return -1;
133 sline = buf;
134
135 memset(script_name, 0, SCRIPT_FULLPATH_LEN);
136 if (list_scripts(script_name))
137 goto exit;
138
139 sprintf(cmd, "perf script -s %s ", script_name);
140
141 if (script_opt)
142 strcat(cmd, script_opt);
143
144 if (input_name) {
145 strcat(cmd, " -i ");
146 strcat(cmd, input_name);
147 }
148
149 strcat(cmd, " 2>&1");
150
151 fp = popen(cmd, "r");
152 if (!fp)
153 goto exit;
154
155 while ((retlen = getline(&line, &len, fp)) != -1) {
156 strncpy(sline->line, line, AVERAGE_LINE_LEN);
157
158 /* If one output line is very large, just cut it short */
159 if (retlen >= AVERAGE_LINE_LEN) {
160 sline->line[AVERAGE_LINE_LEN - 1] = '\0';
161 sline->line[AVERAGE_LINE_LEN - 2] = '\n';
162 }
163 list_add_tail(&sline->node, &script.entries);
164
165 if (script.b.width < retlen)
166 script.b.width = retlen;
167
168 if (nr_entries++ >= MAX_LINES - 1)
169 break;
170 sline++;
171 }
172
173 if (script.b.width > AVERAGE_LINE_LEN)
174 script.b.width = AVERAGE_LINE_LEN;
175
176 if (line)
177 free(line);
178 pclose(fp);
179
180 script.nr_lines = nr_entries;
181 script.b.nr_entries = nr_entries;
182 script.b.entries = &script.entries;
183
184 ret = script_browser__run(&script);
185 exit:
186 free(buf);
187 return ret;
188 }
189