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