blob: 7942ea5c7e8173d2ee12f512f442de3eaea8cb82 [file] [log] [blame]
Masami Hiramatsu92f6c722015-07-15 18:14:07 +09001/*
2 * probe-file.c : operate ftrace k/uprobe events files
3 *
4 * Written by Masami Hiramatsu <masami.hiramatsu.pt@hitachi.com>
5 *
6 * This program is free software; you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License as published by
8 * the Free Software Foundation; either version 2 of the License, or
9 * (at your option) any later version.
10 *
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
15 *
16 */
Masami Hiramatsudd975492016-06-15 12:28:30 +090017#include <sys/uio.h>
Masami Hiramatsu92f6c722015-07-15 18:14:07 +090018#include "util.h"
19#include "event.h"
20#include "strlist.h"
21#include "debug.h"
22#include "cache.h"
23#include "color.h"
24#include "symbol.h"
25#include "thread.h"
Jiri Olsafbf99622015-09-02 09:56:45 +020026#include <api/fs/tracing_path.h>
Masami Hiramatsu92f6c722015-07-15 18:14:07 +090027#include "probe-event.h"
28#include "probe-file.h"
29#include "session.h"
Alexis Berlemont3b1f8312016-12-14 01:07:32 +010030#include "perf_regs.h"
Arnaldo Carvalho de Meloa0675582017-04-17 16:51:59 -030031#include "string2.h"
Masami Hiramatsu92f6c722015-07-15 18:14:07 +090032
Ravi Bangoria2e1f8f72017-02-07 11:15:47 +053033/* 4096 - 2 ('\n' + '\0') */
34#define MAX_CMDLEN 4094
Masami Hiramatsu92f6c722015-07-15 18:14:07 +090035
36static void print_open_warning(int err, bool uprobe)
37{
38 char sbuf[STRERR_BUFSIZE];
39
40 if (err == -ENOENT) {
41 const char *config;
42
43 if (uprobe)
44 config = "CONFIG_UPROBE_EVENTS";
45 else
46 config = "CONFIG_KPROBE_EVENTS";
47
48 pr_warning("%cprobe_events file does not exist"
49 " - please rebuild kernel with %s.\n",
50 uprobe ? 'u' : 'k', config);
51 } else if (err == -ENOTSUP)
52 pr_warning("Tracefs or debugfs is not mounted.\n");
53 else
54 pr_warning("Failed to open %cprobe_events: %s\n",
55 uprobe ? 'u' : 'k',
Arnaldo Carvalho de Meloc8b5f2c2016-07-06 11:56:20 -030056 str_error_r(-err, sbuf, sizeof(sbuf)));
Masami Hiramatsu92f6c722015-07-15 18:14:07 +090057}
58
59static void print_both_open_warning(int kerr, int uerr)
60{
61 /* Both kprobes and uprobes are disabled, warn it. */
62 if (kerr == -ENOTSUP && uerr == -ENOTSUP)
63 pr_warning("Tracefs or debugfs is not mounted.\n");
64 else if (kerr == -ENOENT && uerr == -ENOENT)
65 pr_warning("Please rebuild kernel with CONFIG_KPROBE_EVENTS "
66 "or/and CONFIG_UPROBE_EVENTS.\n");
67 else {
68 char sbuf[STRERR_BUFSIZE];
69 pr_warning("Failed to open kprobe events: %s.\n",
Arnaldo Carvalho de Meloc8b5f2c2016-07-06 11:56:20 -030070 str_error_r(-kerr, sbuf, sizeof(sbuf)));
Masami Hiramatsu92f6c722015-07-15 18:14:07 +090071 pr_warning("Failed to open uprobe events: %s.\n",
Arnaldo Carvalho de Meloc8b5f2c2016-07-06 11:56:20 -030072 str_error_r(-uerr, sbuf, sizeof(sbuf)));
Masami Hiramatsu92f6c722015-07-15 18:14:07 +090073 }
74}
75
Naveen N. Raoe491bc22017-02-23 17:07:23 +053076int open_trace_file(const char *trace_file, bool readwrite)
Masami Hiramatsu92f6c722015-07-15 18:14:07 +090077{
78 char buf[PATH_MAX];
Masami Hiramatsu92f6c722015-07-15 18:14:07 +090079 int ret;
80
Masami Hiramatsu04e11962016-08-24 14:57:58 +090081 ret = e_snprintf(buf, PATH_MAX, "%s/%s",
82 tracing_path, trace_file);
Masami Hiramatsu92f6c722015-07-15 18:14:07 +090083 if (ret >= 0) {
84 pr_debug("Opening %s write=%d\n", buf, readwrite);
85 if (readwrite && !probe_event_dry_run)
86 ret = open(buf, O_RDWR | O_APPEND, 0);
87 else
88 ret = open(buf, O_RDONLY, 0);
89
90 if (ret < 0)
91 ret = -errno;
92 }
93 return ret;
94}
95
96static int open_kprobe_events(bool readwrite)
97{
Naveen N. Raoe491bc22017-02-23 17:07:23 +053098 return open_trace_file("kprobe_events", readwrite);
Masami Hiramatsu92f6c722015-07-15 18:14:07 +090099}
100
101static int open_uprobe_events(bool readwrite)
102{
Naveen N. Raoe491bc22017-02-23 17:07:23 +0530103 return open_trace_file("uprobe_events", readwrite);
Masami Hiramatsu92f6c722015-07-15 18:14:07 +0900104}
105
106int probe_file__open(int flag)
107{
108 int fd;
109
110 if (flag & PF_FL_UPROBE)
111 fd = open_uprobe_events(flag & PF_FL_RW);
112 else
113 fd = open_kprobe_events(flag & PF_FL_RW);
114 if (fd < 0)
115 print_open_warning(fd, flag & PF_FL_UPROBE);
116
117 return fd;
118}
119
120int probe_file__open_both(int *kfd, int *ufd, int flag)
121{
122 if (!kfd || !ufd)
123 return -EINVAL;
124
125 *kfd = open_kprobe_events(flag & PF_FL_RW);
126 *ufd = open_uprobe_events(flag & PF_FL_RW);
127 if (*kfd < 0 && *ufd < 0) {
128 print_both_open_warning(*kfd, *ufd);
129 return *kfd;
130 }
131
132 return 0;
133}
134
135/* Get raw string list of current kprobe_events or uprobe_events */
136struct strlist *probe_file__get_rawlist(int fd)
137{
Colin Ian King03258622016-08-12 22:44:56 +0100138 int ret, idx, fddup;
Masami Hiramatsu92f6c722015-07-15 18:14:07 +0900139 FILE *fp;
140 char buf[MAX_CMDLEN];
141 char *p;
142 struct strlist *sl;
143
Wang Nan421fd082015-11-06 09:50:15 +0000144 if (fd < 0)
145 return NULL;
146
Masami Hiramatsu92f6c722015-07-15 18:14:07 +0900147 sl = strlist__new(NULL, NULL);
Arnaldo Carvalho de Melo60ebc152016-08-15 17:06:47 -0300148 if (sl == NULL)
149 return NULL;
Masami Hiramatsu92f6c722015-07-15 18:14:07 +0900150
Colin Ian King03258622016-08-12 22:44:56 +0100151 fddup = dup(fd);
152 if (fddup < 0)
153 goto out_free_sl;
154
155 fp = fdopen(fddup, "r");
156 if (!fp)
157 goto out_close_fddup;
158
Masami Hiramatsu92f6c722015-07-15 18:14:07 +0900159 while (!feof(fp)) {
160 p = fgets(buf, MAX_CMDLEN, fp);
161 if (!p)
162 break;
163
164 idx = strlen(p) - 1;
165 if (p[idx] == '\n')
166 p[idx] = '\0';
167 ret = strlist__add(sl, buf);
168 if (ret < 0) {
169 pr_debug("strlist__add failed (%d)\n", ret);
Arnaldo Carvalho de Melo60ebc152016-08-15 17:06:47 -0300170 goto out_close_fp;
Masami Hiramatsu92f6c722015-07-15 18:14:07 +0900171 }
172 }
173 fclose(fp);
174
175 return sl;
Colin Ian King03258622016-08-12 22:44:56 +0100176
Arnaldo Carvalho de Melo60ebc152016-08-15 17:06:47 -0300177out_close_fp:
178 fclose(fp);
179 goto out_free_sl;
Colin Ian King03258622016-08-12 22:44:56 +0100180out_close_fddup:
181 close(fddup);
182out_free_sl:
183 strlist__delete(sl);
184 return NULL;
Masami Hiramatsu92f6c722015-07-15 18:14:07 +0900185}
186
187static struct strlist *__probe_file__get_namelist(int fd, bool include_group)
188{
189 char buf[128];
190 struct strlist *sl, *rawlist;
191 struct str_node *ent;
192 struct probe_trace_event tev;
193 int ret = 0;
194
195 memset(&tev, 0, sizeof(tev));
196 rawlist = probe_file__get_rawlist(fd);
197 if (!rawlist)
198 return NULL;
199 sl = strlist__new(NULL, NULL);
Arnaldo Carvalho de Melo602a1f42016-06-23 11:31:20 -0300200 strlist__for_each_entry(ent, rawlist) {
Masami Hiramatsu92f6c722015-07-15 18:14:07 +0900201 ret = parse_probe_trace_command(ent->s, &tev);
202 if (ret < 0)
203 break;
204 if (include_group) {
205 ret = e_snprintf(buf, 128, "%s:%s", tev.group,
206 tev.event);
207 if (ret >= 0)
208 ret = strlist__add(sl, buf);
209 } else
210 ret = strlist__add(sl, tev.event);
211 clear_probe_trace_event(&tev);
212 if (ret < 0)
213 break;
214 }
215 strlist__delete(rawlist);
216
217 if (ret < 0) {
218 strlist__delete(sl);
219 return NULL;
220 }
221 return sl;
222}
223
224/* Get current perf-probe event names */
225struct strlist *probe_file__get_namelist(int fd)
226{
227 return __probe_file__get_namelist(fd, false);
228}
229
230int probe_file__add_event(int fd, struct probe_trace_event *tev)
231{
232 int ret = 0;
233 char *buf = synthesize_probe_trace_command(tev);
234 char sbuf[STRERR_BUFSIZE];
235
236 if (!buf) {
237 pr_debug("Failed to synthesize probe trace event.\n");
238 return -EINVAL;
239 }
240
241 pr_debug("Writing event: %s\n", buf);
242 if (!probe_event_dry_run) {
Masami Hiramatsu6ed07202016-04-26 18:03:04 +0900243 if (write(fd, buf, strlen(buf)) < (int)strlen(buf)) {
Masami Hiramatsu92f6c722015-07-15 18:14:07 +0900244 ret = -errno;
245 pr_warning("Failed to write event: %s\n",
Arnaldo Carvalho de Meloc8b5f2c2016-07-06 11:56:20 -0300246 str_error_r(errno, sbuf, sizeof(sbuf)));
Masami Hiramatsu92f6c722015-07-15 18:14:07 +0900247 }
248 }
249 free(buf);
250
251 return ret;
252}
253
254static int __del_trace_probe_event(int fd, struct str_node *ent)
255{
256 char *p;
257 char buf[128];
258 int ret;
259
260 /* Convert from perf-probe event to trace-probe event */
261 ret = e_snprintf(buf, 128, "-:%s", ent->s);
262 if (ret < 0)
263 goto error;
264
265 p = strchr(buf + 2, ':');
266 if (!p) {
267 pr_debug("Internal error: %s should have ':' but not.\n",
268 ent->s);
269 ret = -ENOTSUP;
270 goto error;
271 }
272 *p = '/';
273
274 pr_debug("Writing event: %s\n", buf);
275 ret = write(fd, buf, strlen(buf));
276 if (ret < 0) {
277 ret = -errno;
278 goto error;
279 }
280
Masami Hiramatsu92f6c722015-07-15 18:14:07 +0900281 return 0;
282error:
283 pr_warning("Failed to delete event: %s\n",
Arnaldo Carvalho de Meloc8b5f2c2016-07-06 11:56:20 -0300284 str_error_r(-ret, buf, sizeof(buf)));
Masami Hiramatsu92f6c722015-07-15 18:14:07 +0900285 return ret;
286}
287
Namhyung Kime607f142015-09-04 21:16:03 +0900288int probe_file__get_events(int fd, struct strfilter *filter,
289 struct strlist *plist)
Masami Hiramatsu92f6c722015-07-15 18:14:07 +0900290{
291 struct strlist *namelist;
292 struct str_node *ent;
293 const char *p;
294 int ret = -ENOENT;
295
Wang Nan421fd082015-11-06 09:50:15 +0000296 if (!plist)
297 return -EINVAL;
298
Masami Hiramatsu92f6c722015-07-15 18:14:07 +0900299 namelist = __probe_file__get_namelist(fd, true);
300 if (!namelist)
301 return -ENOENT;
302
Arnaldo Carvalho de Melo602a1f42016-06-23 11:31:20 -0300303 strlist__for_each_entry(ent, namelist) {
Masami Hiramatsu92f6c722015-07-15 18:14:07 +0900304 p = strchr(ent->s, ':');
305 if ((p && strfilter__compare(filter, p + 1)) ||
306 strfilter__compare(filter, ent->s)) {
Namhyung Kime7895e42015-09-04 21:16:02 +0900307 strlist__add(plist, ent->s);
308 ret = 0;
Masami Hiramatsu92f6c722015-07-15 18:14:07 +0900309 }
310 }
311 strlist__delete(namelist);
312
313 return ret;
314}
Namhyung Kime7895e42015-09-04 21:16:02 +0900315
Namhyung Kime607f142015-09-04 21:16:03 +0900316int probe_file__del_strlist(int fd, struct strlist *namelist)
Namhyung Kime7895e42015-09-04 21:16:02 +0900317{
318 int ret = 0;
319 struct str_node *ent;
320
Arnaldo Carvalho de Melo602a1f42016-06-23 11:31:20 -0300321 strlist__for_each_entry(ent, namelist) {
Namhyung Kime7895e42015-09-04 21:16:02 +0900322 ret = __del_trace_probe_event(fd, ent);
323 if (ret < 0)
324 break;
325 }
326 return ret;
327}
328
329int probe_file__del_events(int fd, struct strfilter *filter)
330{
331 struct strlist *namelist;
332 int ret;
333
334 namelist = strlist__new(NULL, NULL);
335 if (!namelist)
336 return -ENOMEM;
337
338 ret = probe_file__get_events(fd, filter, namelist);
339 if (ret < 0)
340 return ret;
341
342 ret = probe_file__del_strlist(fd, namelist);
343 strlist__delete(namelist);
344
345 return ret;
346}
Masami Hiramatsudd975492016-06-15 12:28:30 +0900347
348/* Caller must ensure to remove this entry from list */
349static void probe_cache_entry__delete(struct probe_cache_entry *entry)
350{
351 if (entry) {
352 BUG_ON(!list_empty(&entry->node));
353
354 strlist__delete(entry->tevlist);
355 clear_perf_probe_event(&entry->pev);
356 zfree(&entry->spev);
357 free(entry);
358 }
359}
360
361static struct probe_cache_entry *
362probe_cache_entry__new(struct perf_probe_event *pev)
363{
364 struct probe_cache_entry *entry = zalloc(sizeof(*entry));
365
366 if (entry) {
367 INIT_LIST_HEAD(&entry->node);
368 entry->tevlist = strlist__new(NULL, NULL);
369 if (!entry->tevlist)
370 zfree(&entry);
371 else if (pev) {
372 entry->spev = synthesize_perf_probe_command(pev);
373 if (!entry->spev ||
374 perf_probe_event__copy(&entry->pev, pev) < 0) {
375 probe_cache_entry__delete(entry);
376 return NULL;
377 }
378 }
379 }
380
381 return entry;
382}
383
Masami Hiramatsu42bba262016-07-12 19:05:18 +0900384int probe_cache_entry__get_event(struct probe_cache_entry *entry,
385 struct probe_trace_event **tevs)
386{
387 struct probe_trace_event *tev;
388 struct str_node *node;
389 int ret, i;
390
391 ret = strlist__nr_entries(entry->tevlist);
392 if (ret > probe_conf.max_probes)
393 return -E2BIG;
394
395 *tevs = zalloc(ret * sizeof(*tev));
396 if (!*tevs)
397 return -ENOMEM;
398
399 i = 0;
400 strlist__for_each_entry(node, entry->tevlist) {
401 tev = &(*tevs)[i++];
402 ret = parse_probe_trace_command(node->s, tev);
403 if (ret < 0)
404 break;
405 }
406 return i;
407}
408
409/* For the kernel probe caches, pass target = NULL or DSO__NAME_KALLSYMS */
Masami Hiramatsudd975492016-06-15 12:28:30 +0900410static int probe_cache__open(struct probe_cache *pcache, const char *target)
411{
412 char cpath[PATH_MAX];
413 char sbuildid[SBUILD_ID_SIZE];
Masami Hiramatsu1f3736c2016-07-01 17:03:26 +0900414 char *dir_name = NULL;
Masami Hiramatsu42bba262016-07-12 19:05:18 +0900415 bool is_kallsyms = false;
Masami Hiramatsudd975492016-06-15 12:28:30 +0900416 int ret, fd;
417
Masami Hiramatsu1f3736c2016-07-01 17:03:26 +0900418 if (target && build_id_cache__cached(target)) {
419 /* This is a cached buildid */
420 strncpy(sbuildid, target, SBUILD_ID_SIZE);
421 dir_name = build_id_cache__linkname(sbuildid, NULL, 0);
422 goto found;
423 }
424
Masami Hiramatsu42bba262016-07-12 19:05:18 +0900425 if (!target || !strcmp(target, DSO__NAME_KALLSYMS)) {
Masami Hiramatsudd975492016-06-15 12:28:30 +0900426 target = DSO__NAME_KALLSYMS;
Masami Hiramatsu42bba262016-07-12 19:05:18 +0900427 is_kallsyms = true;
Masami Hiramatsudd975492016-06-15 12:28:30 +0900428 ret = sysfs__sprintf_build_id("/", sbuildid);
Masami Hiramatsu42bba262016-07-12 19:05:18 +0900429 } else
430 ret = filename__sprintf_build_id(target, sbuildid);
431
Masami Hiramatsudd975492016-06-15 12:28:30 +0900432 if (ret < 0) {
433 pr_debug("Failed to get build-id from %s.\n", target);
434 return ret;
435 }
436
437 /* If we have no buildid cache, make it */
438 if (!build_id_cache__cached(sbuildid)) {
439 ret = build_id_cache__add_s(sbuildid, target,
440 is_kallsyms, NULL);
441 if (ret < 0) {
442 pr_debug("Failed to add build-id cache: %s\n", target);
443 return ret;
444 }
445 }
446
447 dir_name = build_id_cache__cachedir(sbuildid, target, is_kallsyms,
448 false);
Masami Hiramatsu1f3736c2016-07-01 17:03:26 +0900449found:
450 if (!dir_name) {
451 pr_debug("Failed to get cache from %s\n", target);
Masami Hiramatsudd975492016-06-15 12:28:30 +0900452 return -ENOMEM;
Masami Hiramatsu1f3736c2016-07-01 17:03:26 +0900453 }
Masami Hiramatsudd975492016-06-15 12:28:30 +0900454
455 snprintf(cpath, PATH_MAX, "%s/probes", dir_name);
456 fd = open(cpath, O_CREAT | O_RDWR, 0644);
457 if (fd < 0)
458 pr_debug("Failed to open cache(%d): %s\n", fd, cpath);
459 free(dir_name);
460 pcache->fd = fd;
461
462 return fd;
463}
464
465static int probe_cache__load(struct probe_cache *pcache)
466{
467 struct probe_cache_entry *entry = NULL;
468 char buf[MAX_CMDLEN], *p;
Colin Ian King03258622016-08-12 22:44:56 +0100469 int ret = 0, fddup;
Masami Hiramatsudd975492016-06-15 12:28:30 +0900470 FILE *fp;
471
Colin Ian King03258622016-08-12 22:44:56 +0100472 fddup = dup(pcache->fd);
473 if (fddup < 0)
474 return -errno;
475 fp = fdopen(fddup, "r");
Arnaldo Carvalho de Melo60ebc152016-08-15 17:06:47 -0300476 if (!fp) {
477 close(fddup);
Masami Hiramatsudd975492016-06-15 12:28:30 +0900478 return -EINVAL;
Arnaldo Carvalho de Melo60ebc152016-08-15 17:06:47 -0300479 }
Masami Hiramatsudd975492016-06-15 12:28:30 +0900480
481 while (!feof(fp)) {
482 if (!fgets(buf, MAX_CMDLEN, fp))
483 break;
484 p = strchr(buf, '\n');
485 if (p)
486 *p = '\0';
Masami Hiramatsu6430a94e2016-07-01 17:04:10 +0900487 /* #perf_probe_event or %sdt_event */
488 if (buf[0] == '#' || buf[0] == '%') {
Masami Hiramatsudd975492016-06-15 12:28:30 +0900489 entry = probe_cache_entry__new(NULL);
490 if (!entry) {
491 ret = -ENOMEM;
492 goto out;
493 }
Masami Hiramatsu6430a94e2016-07-01 17:04:10 +0900494 if (buf[0] == '%')
495 entry->sdt = true;
Masami Hiramatsudd975492016-06-15 12:28:30 +0900496 entry->spev = strdup(buf + 1);
497 if (entry->spev)
498 ret = parse_perf_probe_command(buf + 1,
499 &entry->pev);
500 else
501 ret = -ENOMEM;
502 if (ret < 0) {
503 probe_cache_entry__delete(entry);
504 goto out;
505 }
506 list_add_tail(&entry->node, &pcache->entries);
507 } else { /* trace_probe_event */
508 if (!entry) {
509 ret = -EINVAL;
510 goto out;
511 }
512 strlist__add(entry->tevlist, buf);
513 }
514 }
515out:
516 fclose(fp);
517 return ret;
518}
519
520static struct probe_cache *probe_cache__alloc(void)
521{
522 struct probe_cache *pcache = zalloc(sizeof(*pcache));
523
524 if (pcache) {
525 INIT_LIST_HEAD(&pcache->entries);
526 pcache->fd = -EINVAL;
527 }
528 return pcache;
529}
530
531void probe_cache__purge(struct probe_cache *pcache)
532{
533 struct probe_cache_entry *entry, *n;
534
535 list_for_each_entry_safe(entry, n, &pcache->entries, node) {
536 list_del_init(&entry->node);
537 probe_cache_entry__delete(entry);
538 }
539}
540
541void probe_cache__delete(struct probe_cache *pcache)
542{
543 if (!pcache)
544 return;
545
546 probe_cache__purge(pcache);
547 if (pcache->fd > 0)
548 close(pcache->fd);
549 free(pcache);
550}
551
552struct probe_cache *probe_cache__new(const char *target)
553{
554 struct probe_cache *pcache = probe_cache__alloc();
555 int ret;
556
557 if (!pcache)
558 return NULL;
559
560 ret = probe_cache__open(pcache, target);
561 if (ret < 0) {
562 pr_debug("Cache open error: %d\n", ret);
563 goto out_err;
564 }
565
566 ret = probe_cache__load(pcache);
567 if (ret < 0) {
568 pr_debug("Cache read error: %d\n", ret);
569 goto out_err;
570 }
571
572 return pcache;
573
574out_err:
575 probe_cache__delete(pcache);
576 return NULL;
577}
578
579static bool streql(const char *a, const char *b)
580{
581 if (a == b)
582 return true;
583
584 if (!a || !b)
585 return false;
586
587 return !strcmp(a, b);
588}
589
Masami Hiramatsubc062232016-07-01 17:03:12 +0900590struct probe_cache_entry *
Masami Hiramatsudd975492016-06-15 12:28:30 +0900591probe_cache__find(struct probe_cache *pcache, struct perf_probe_event *pev)
592{
593 struct probe_cache_entry *entry = NULL;
594 char *cmd = synthesize_perf_probe_command(pev);
595
596 if (!cmd)
597 return NULL;
598
Masami Hiramatsu05bf2c82016-07-12 19:05:04 +0900599 for_each_probe_cache_entry(entry, pcache) {
Masami Hiramatsu36a009f2016-07-12 19:04:43 +0900600 if (pev->sdt) {
601 if (entry->pev.event &&
602 streql(entry->pev.event, pev->event) &&
603 (!pev->group ||
604 streql(entry->pev.group, pev->group)))
605 goto found;
606
607 continue;
608 }
Masami Hiramatsudd975492016-06-15 12:28:30 +0900609 /* Hit if same event name or same command-string */
610 if ((pev->event &&
611 (streql(entry->pev.group, pev->group) &&
612 streql(entry->pev.event, pev->event))) ||
613 (!strcmp(entry->spev, cmd)))
614 goto found;
615 }
616 entry = NULL;
617
618found:
619 free(cmd);
620 return entry;
621}
622
Masami Hiramatsubc062232016-07-01 17:03:12 +0900623struct probe_cache_entry *
624probe_cache__find_by_name(struct probe_cache *pcache,
625 const char *group, const char *event)
626{
627 struct probe_cache_entry *entry = NULL;
628
Masami Hiramatsu05bf2c82016-07-12 19:05:04 +0900629 for_each_probe_cache_entry(entry, pcache) {
Masami Hiramatsubc062232016-07-01 17:03:12 +0900630 /* Hit if same event name or same command-string */
631 if (streql(entry->pev.group, group) &&
632 streql(entry->pev.event, event))
633 goto found;
634 }
635 entry = NULL;
636
637found:
638 return entry;
639}
640
Masami Hiramatsudd975492016-06-15 12:28:30 +0900641int probe_cache__add_entry(struct probe_cache *pcache,
642 struct perf_probe_event *pev,
643 struct probe_trace_event *tevs, int ntevs)
644{
645 struct probe_cache_entry *entry = NULL;
646 char *command;
647 int i, ret = 0;
648
649 if (!pcache || !pev || !tevs || ntevs <= 0) {
650 ret = -EINVAL;
651 goto out_err;
652 }
653
654 /* Remove old cache entry */
655 entry = probe_cache__find(pcache, pev);
656 if (entry) {
657 list_del_init(&entry->node);
658 probe_cache_entry__delete(entry);
659 }
660
661 ret = -ENOMEM;
662 entry = probe_cache_entry__new(pev);
663 if (!entry)
664 goto out_err;
665
666 for (i = 0; i < ntevs; i++) {
667 if (!tevs[i].point.symbol)
668 continue;
669
670 command = synthesize_probe_trace_command(&tevs[i]);
671 if (!command)
672 goto out_err;
673 strlist__add(entry->tevlist, command);
674 free(command);
675 }
676 list_add_tail(&entry->node, &pcache->entries);
677 pr_debug("Added probe cache: %d\n", ntevs);
678 return 0;
679
680out_err:
681 pr_debug("Failed to add probe caches\n");
682 probe_cache_entry__delete(entry);
683 return ret;
684}
685
Arnaldo Carvalho de Melo1c1a3a42016-07-12 12:19:09 -0300686#ifdef HAVE_GELF_GETNOTE_SUPPORT
Masami Hiramatsu6430a94e2016-07-01 17:04:10 +0900687static unsigned long long sdt_note__get_addr(struct sdt_note *note)
688{
689 return note->bit32 ? (unsigned long long)note->addr.a32[0]
690 : (unsigned long long)note->addr.a64[0];
691}
692
Alexis Berlemont3b1f8312016-12-14 01:07:32 +0100693static const char * const type_to_suffix[] = {
694 ":s64", "", "", "", ":s32", "", ":s16", ":s8",
695 "", ":u8", ":u16", "", ":u32", "", "", "", ":u64"
696};
697
Ravi Bangoriad451a202017-03-28 15:17:53 +0530698/*
699 * Isolate the string number and convert it into a decimal value;
700 * this will be an index to get suffix of the uprobe name (defining
701 * the type)
702 */
703static int sdt_arg_parse_size(char *n_ptr, const char **suffix)
704{
705 long type_idx;
706
707 type_idx = strtol(n_ptr, NULL, 10);
708 if (type_idx < -8 || type_idx > 8) {
709 pr_debug4("Failed to get a valid sdt type\n");
710 return -1;
711 }
712
713 *suffix = type_to_suffix[type_idx + 8];
714 return 0;
715}
716
Alexis Berlemont3b1f8312016-12-14 01:07:32 +0100717static int synthesize_sdt_probe_arg(struct strbuf *buf, int i, const char *arg)
718{
Ravi Bangoriad451a202017-03-28 15:17:53 +0530719 char *op, *desc = strdup(arg), *new_op = NULL;
720 const char *suffix = "";
Alexis Berlemont3b1f8312016-12-14 01:07:32 +0100721 int ret = -1;
722
723 if (desc == NULL) {
724 pr_debug4("Allocation error\n");
725 return ret;
726 }
727
Ravi Bangoriad451a202017-03-28 15:17:53 +0530728 /*
729 * Argument is in N@OP format. N is size of the argument and OP is
730 * the actual assembly operand. N can be omitted; in that case
731 * argument is just OP(without @).
732 */
733 op = strchr(desc, '@');
734 if (op) {
735 op[0] = '\0';
736 op++;
737
738 if (sdt_arg_parse_size(desc, &suffix))
Alexis Berlemont3b1f8312016-12-14 01:07:32 +0100739 goto error;
Ravi Bangoriad451a202017-03-28 15:17:53 +0530740 } else {
741 op = desc;
Alexis Berlemont3b1f8312016-12-14 01:07:32 +0100742 }
743
Ravi Bangoriad451a202017-03-28 15:17:53 +0530744 ret = arch_sdt_arg_parse_op(op, &new_op);
Alexis Berlemont3b1f8312016-12-14 01:07:32 +0100745
Ravi Bangoriad451a202017-03-28 15:17:53 +0530746 if (ret < 0)
Alexis Berlemont3b1f8312016-12-14 01:07:32 +0100747 goto error;
748
Ravi Bangoriad451a202017-03-28 15:17:53 +0530749 if (ret == SDT_ARG_VALID) {
750 ret = strbuf_addf(buf, " arg%d=%s%s", i + 1, new_op, suffix);
751 if (ret < 0)
752 goto error;
753 }
754
Alexis Berlemont3b1f8312016-12-14 01:07:32 +0100755 ret = 0;
756error:
757 free(desc);
Ravi Bangoriad451a202017-03-28 15:17:53 +0530758 free(new_op);
Alexis Berlemont3b1f8312016-12-14 01:07:32 +0100759 return ret;
760}
761
762static char *synthesize_sdt_probe_command(struct sdt_note *note,
763 const char *pathname,
764 const char *sdtgrp)
765{
766 struct strbuf buf;
767 char *ret = NULL, **args;
768 int i, args_count;
769
770 if (strbuf_init(&buf, 32) < 0)
771 return NULL;
772
773 if (strbuf_addf(&buf, "p:%s/%s %s:0x%llx",
774 sdtgrp, note->name, pathname,
775 sdt_note__get_addr(note)) < 0)
776 goto error;
777
778 if (!note->args)
779 goto out;
780
781 if (note->args) {
782 args = argv_split(note->args, &args_count);
783
784 for (i = 0; i < args_count; ++i) {
785 if (synthesize_sdt_probe_arg(&buf, i, args[i]) < 0)
786 goto error;
787 }
788 }
789
790out:
791 ret = strbuf_detach(&buf, NULL);
792error:
793 strbuf_release(&buf);
794 return ret;
795}
796
Masami Hiramatsu6430a94e2016-07-01 17:04:10 +0900797int probe_cache__scan_sdt(struct probe_cache *pcache, const char *pathname)
798{
799 struct probe_cache_entry *entry = NULL;
800 struct list_head sdtlist;
801 struct sdt_note *note;
802 char *buf;
803 char sdtgrp[64];
804 int ret;
805
806 INIT_LIST_HEAD(&sdtlist);
807 ret = get_sdt_note_list(&sdtlist, pathname);
808 if (ret < 0) {
Adrian Hunterf9655202016-09-23 17:38:40 +0300809 pr_debug4("Failed to get sdt note: %d\n", ret);
Masami Hiramatsu6430a94e2016-07-01 17:04:10 +0900810 return ret;
811 }
812 list_for_each_entry(note, &sdtlist, note_list) {
813 ret = snprintf(sdtgrp, 64, "sdt_%s", note->provider);
814 if (ret < 0)
815 break;
816 /* Try to find same-name entry */
817 entry = probe_cache__find_by_name(pcache, sdtgrp, note->name);
818 if (!entry) {
819 entry = probe_cache_entry__new(NULL);
820 if (!entry) {
821 ret = -ENOMEM;
822 break;
823 }
824 entry->sdt = true;
825 ret = asprintf(&entry->spev, "%s:%s=%s", sdtgrp,
826 note->name, note->name);
827 if (ret < 0)
828 break;
829 entry->pev.event = strdup(note->name);
830 entry->pev.group = strdup(sdtgrp);
831 list_add_tail(&entry->node, &pcache->entries);
832 }
Alexis Berlemont3b1f8312016-12-14 01:07:32 +0100833 buf = synthesize_sdt_probe_command(note, pathname, sdtgrp);
834 if (!buf) {
835 ret = -ENOMEM;
Masami Hiramatsu6430a94e2016-07-01 17:04:10 +0900836 break;
Alexis Berlemont3b1f8312016-12-14 01:07:32 +0100837 }
838
Masami Hiramatsu6430a94e2016-07-01 17:04:10 +0900839 strlist__add(entry->tevlist, buf);
840 free(buf);
841 entry = NULL;
842 }
843 if (entry) {
844 list_del_init(&entry->node);
845 probe_cache_entry__delete(entry);
846 }
847 cleanup_sdt_note_list(&sdtlist);
848 return ret;
849}
Arnaldo Carvalho de Melo1c1a3a42016-07-12 12:19:09 -0300850#endif
Masami Hiramatsu6430a94e2016-07-01 17:04:10 +0900851
Masami Hiramatsudd975492016-06-15 12:28:30 +0900852static int probe_cache_entry__write(struct probe_cache_entry *entry, int fd)
853{
854 struct str_node *snode;
855 struct stat st;
856 struct iovec iov[3];
Masami Hiramatsu6430a94e2016-07-01 17:04:10 +0900857 const char *prefix = entry->sdt ? "%" : "#";
Masami Hiramatsudd975492016-06-15 12:28:30 +0900858 int ret;
859 /* Save stat for rollback */
860 ret = fstat(fd, &st);
861 if (ret < 0)
862 return ret;
863
Masami Hiramatsu6430a94e2016-07-01 17:04:10 +0900864 pr_debug("Writing cache: %s%s\n", prefix, entry->spev);
865 iov[0].iov_base = (void *)prefix; iov[0].iov_len = 1;
Masami Hiramatsudd975492016-06-15 12:28:30 +0900866 iov[1].iov_base = entry->spev; iov[1].iov_len = strlen(entry->spev);
867 iov[2].iov_base = (void *)"\n"; iov[2].iov_len = 1;
868 ret = writev(fd, iov, 3);
869 if (ret < (int)iov[1].iov_len + 2)
870 goto rollback;
871
Arnaldo Carvalho de Melo602a1f42016-06-23 11:31:20 -0300872 strlist__for_each_entry(snode, entry->tevlist) {
Masami Hiramatsudd975492016-06-15 12:28:30 +0900873 iov[0].iov_base = (void *)snode->s;
874 iov[0].iov_len = strlen(snode->s);
875 iov[1].iov_base = (void *)"\n"; iov[1].iov_len = 1;
876 ret = writev(fd, iov, 2);
877 if (ret < (int)iov[0].iov_len + 1)
878 goto rollback;
879 }
880 return 0;
881
882rollback:
883 /* Rollback to avoid cache file corruption */
884 if (ret > 0)
885 ret = -1;
886 if (ftruncate(fd, st.st_size) < 0)
887 ret = -2;
888
889 return ret;
890}
891
892int probe_cache__commit(struct probe_cache *pcache)
893{
894 struct probe_cache_entry *entry;
895 int ret = 0;
896
897 /* TBD: if we do not update existing entries, skip it */
898 ret = lseek(pcache->fd, 0, SEEK_SET);
899 if (ret < 0)
900 goto out;
901
902 ret = ftruncate(pcache->fd, 0);
903 if (ret < 0)
904 goto out;
905
Masami Hiramatsu05bf2c82016-07-12 19:05:04 +0900906 for_each_probe_cache_entry(entry, pcache) {
Masami Hiramatsudd975492016-06-15 12:28:30 +0900907 ret = probe_cache_entry__write(entry, pcache->fd);
908 pr_debug("Cache committed: %d\n", ret);
909 if (ret < 0)
910 break;
911 }
912out:
913 return ret;
914}
Masami Hiramatsu1f3736c2016-07-01 17:03:26 +0900915
Masami Hiramatsu4a0f65c2016-07-01 17:03:36 +0900916static bool probe_cache_entry__compare(struct probe_cache_entry *entry,
917 struct strfilter *filter)
918{
919 char buf[128], *ptr = entry->spev;
920
921 if (entry->pev.event) {
922 snprintf(buf, 128, "%s:%s", entry->pev.group, entry->pev.event);
923 ptr = buf;
924 }
925 return strfilter__compare(filter, ptr);
926}
927
928int probe_cache__filter_purge(struct probe_cache *pcache,
929 struct strfilter *filter)
930{
931 struct probe_cache_entry *entry, *tmp;
932
933 list_for_each_entry_safe(entry, tmp, &pcache->entries, node) {
934 if (probe_cache_entry__compare(entry, filter)) {
935 pr_info("Removed cached event: %s\n", entry->spev);
936 list_del_init(&entry->node);
937 probe_cache_entry__delete(entry);
938 }
939 }
940 return 0;
941}
942
Masami Hiramatsu1f3736c2016-07-01 17:03:26 +0900943static int probe_cache__show_entries(struct probe_cache *pcache,
944 struct strfilter *filter)
945{
946 struct probe_cache_entry *entry;
Masami Hiramatsu1f3736c2016-07-01 17:03:26 +0900947
Masami Hiramatsu05bf2c82016-07-12 19:05:04 +0900948 for_each_probe_cache_entry(entry, pcache) {
Masami Hiramatsu4a0f65c2016-07-01 17:03:36 +0900949 if (probe_cache_entry__compare(entry, filter))
Masami Hiramatsu1f3736c2016-07-01 17:03:26 +0900950 printf("%s\n", entry->spev);
951 }
952 return 0;
953}
954
955/* Show all cached probes */
956int probe_cache__show_all_caches(struct strfilter *filter)
957{
958 struct probe_cache *pcache;
959 struct strlist *bidlist;
960 struct str_node *nd;
961 char *buf = strfilter__string(filter);
962
963 pr_debug("list cache with filter: %s\n", buf);
964 free(buf);
965
Masami Hiramatsuc3492a32016-07-12 19:04:54 +0900966 bidlist = build_id_cache__list_all(true);
Masami Hiramatsu1f3736c2016-07-01 17:03:26 +0900967 if (!bidlist) {
968 pr_debug("Failed to get buildids: %d\n", errno);
969 return -EINVAL;
970 }
971 strlist__for_each_entry(nd, bidlist) {
972 pcache = probe_cache__new(nd->s);
973 if (!pcache)
974 continue;
975 if (!list_empty(&pcache->entries)) {
976 buf = build_id_cache__origname(nd->s);
977 printf("%s (%s):\n", buf, nd->s);
978 free(buf);
979 probe_cache__show_entries(pcache, filter);
980 }
981 probe_cache__delete(pcache);
982 }
983 strlist__delete(bidlist);
984
985 return 0;
986}
Masami Hiramatsu180b2062016-08-18 17:58:31 +0900987
Naveen N. Rao3da3ea72017-03-08 13:56:08 +0530988enum ftrace_readme {
989 FTRACE_README_PROBE_TYPE_X = 0,
Naveen N. Rao7ab31d92017-03-08 13:56:09 +0530990 FTRACE_README_KRETPROBE_OFFSET,
Naveen N. Rao3da3ea72017-03-08 13:56:08 +0530991 FTRACE_README_END,
Masami Hiramatsu180b2062016-08-18 17:58:31 +0900992};
993
Naveen N. Rao3da3ea72017-03-08 13:56:08 +0530994static struct {
995 const char *pattern;
996 bool avail;
997} ftrace_readme_table[] = {
998#define DEFINE_TYPE(idx, pat) \
999 [idx] = {.pattern = pat, .avail = false}
1000 DEFINE_TYPE(FTRACE_README_PROBE_TYPE_X, "*type: * x8/16/32/64,*"),
Naveen N. Rao7ab31d92017-03-08 13:56:09 +05301001 DEFINE_TYPE(FTRACE_README_KRETPROBE_OFFSET, "*place (kretprobe): *"),
Naveen N. Rao3da3ea72017-03-08 13:56:08 +05301002};
1003
1004static bool scan_ftrace_readme(enum ftrace_readme type)
Masami Hiramatsu180b2062016-08-18 17:58:31 +09001005{
Naveen N. Rao3da3ea72017-03-08 13:56:08 +05301006 int fd;
Masami Hiramatsu180b2062016-08-18 17:58:31 +09001007 FILE *fp;
1008 char *buf = NULL;
1009 size_t len = 0;
Naveen N. Rao3da3ea72017-03-08 13:56:08 +05301010 bool ret = false;
1011 static bool scanned = false;
Masami Hiramatsu180b2062016-08-18 17:58:31 +09001012
Naveen N. Rao3da3ea72017-03-08 13:56:08 +05301013 if (scanned)
1014 goto result;
Masami Hiramatsu180b2062016-08-18 17:58:31 +09001015
Naveen N. Raoe491bc22017-02-23 17:07:23 +05301016 fd = open_trace_file("README", false);
1017 if (fd < 0)
Masami Hiramatsu180b2062016-08-18 17:58:31 +09001018 return ret;
1019
Naveen N. Raoe491bc22017-02-23 17:07:23 +05301020 fp = fdopen(fd, "r");
1021 if (!fp) {
1022 close(fd);
1023 return ret;
1024 }
Masami Hiramatsu180b2062016-08-18 17:58:31 +09001025
Naveen N. Rao3da3ea72017-03-08 13:56:08 +05301026 while (getline(&buf, &len, fp) > 0)
1027 for (enum ftrace_readme i = 0; i < FTRACE_README_END; i++)
1028 if (!ftrace_readme_table[i].avail)
1029 ftrace_readme_table[i].avail =
1030 strglobmatch(buf, ftrace_readme_table[i].pattern);
1031 scanned = true;
Masami Hiramatsu180b2062016-08-18 17:58:31 +09001032
1033 fclose(fp);
Masami Hiramatsu180b2062016-08-18 17:58:31 +09001034 free(buf);
1035
Naveen N. Rao3da3ea72017-03-08 13:56:08 +05301036result:
1037 if (type >= FTRACE_README_END)
1038 return false;
1039
1040 return ftrace_readme_table[type].avail;
1041}
1042
1043bool probe_type_is_available(enum probe_type type)
1044{
1045 if (type >= PROBE_TYPE_END)
1046 return false;
1047 else if (type == PROBE_TYPE_X)
1048 return scan_ftrace_readme(FTRACE_README_PROBE_TYPE_X);
1049
1050 return true;
Masami Hiramatsu180b2062016-08-18 17:58:31 +09001051}
Naveen N. Rao7ab31d92017-03-08 13:56:09 +05301052
1053bool kretprobe_offset_is_supported(void)
1054{
1055 return scan_ftrace_readme(FTRACE_README_KRETPROBE_OFFSET);
1056}