probe-finder.c 26.1 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33
/*
 * probe-finder.c : C expression to kprobe event converter
 *
 * Written by Masami Hiramatsu <mhiramat@redhat.com>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
 *
 */

#include <sys/utsname.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <errno.h>
#include <stdio.h>
#include <unistd.h>
#include <getopt.h>
#include <stdlib.h>
#include <string.h>
#include <stdarg.h>
#include <ctype.h>
34

35
#include "string.h"
36 37
#include "event.h"
#include "debug.h"
38
#include "util.h"
39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86
#include "probe-finder.h"


/*
 * Generic dwarf analysis helpers
 */

#define X86_32_MAX_REGS 8
const char *x86_32_regs_table[X86_32_MAX_REGS] = {
	"%ax",
	"%cx",
	"%dx",
	"%bx",
	"$stack",	/* Stack address instead of %sp */
	"%bp",
	"%si",
	"%di",
};

#define X86_64_MAX_REGS 16
const char *x86_64_regs_table[X86_64_MAX_REGS] = {
	"%ax",
	"%dx",
	"%cx",
	"%bx",
	"%si",
	"%di",
	"%bp",
	"%sp",
	"%r8",
	"%r9",
	"%r10",
	"%r11",
	"%r12",
	"%r13",
	"%r14",
	"%r15",
};

/* TODO: switching by dwarf address size */
#ifdef __x86_64__
#define ARCH_MAX_REGS X86_64_MAX_REGS
#define arch_regs_table x86_64_regs_table
#else
#define ARCH_MAX_REGS X86_32_MAX_REGS
#define arch_regs_table x86_32_regs_table
#endif

87 88 89
/* Kprobe tracer basic type is up to u64 */
#define MAX_BASIC_TYPE_BITS	64

90 91 92 93 94 95 96 97 98 99 100 101 102 103
/* Return architecture dependent register string (for kprobe-tracer) */
static const char *get_arch_regstr(unsigned int n)
{
	return (n <= ARCH_MAX_REGS) ? arch_regs_table[n] : NULL;
}

/*
 * Compare the tail of two strings.
 * Return 0 if whole of either string is same as another's tail part.
 */
static int strtailcmp(const char *s1, const char *s2)
{
	int i1 = strlen(s1);
	int i2 = strlen(s2);
104
	while (--i1 >= 0 && --i2 >= 0) {
105 106 107 108 109 110
		if (s1[i1] != s2[i2])
			return s1[i1] - s2[i2];
	}
	return 0;
}

111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130
/* Line number list operations */

/* Add a line to line number list */
static void line_list__add_line(struct list_head *head, unsigned int line)
{
	struct line_node *ln;
	struct list_head *p;

	/* Reverse search, because new line will be the last one */
	list_for_each_entry_reverse(ln, head, list) {
		if (ln->line < line) {
			p = &ln->list;
			goto found;
		} else if (ln->line == line)	/* Already exist */
			return ;
	}
	/* List is empty, or the smallest entry */
	p = head;
found:
	pr_debug("line list: add a line %u\n", line);
131
	ln = xzalloc(sizeof(struct line_node));
132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170
	ln->line = line;
	INIT_LIST_HEAD(&ln->list);
	list_add(&ln->list, p);
}

/* Check if the line in line number list */
static int line_list__has_line(struct list_head *head, unsigned int line)
{
	struct line_node *ln;

	/* Reverse search, because new line will be the last one */
	list_for_each_entry(ln, head, list)
		if (ln->line == line)
			return 1;

	return 0;
}

/* Init line number list */
static void line_list__init(struct list_head *head)
{
	INIT_LIST_HEAD(head);
}

/* Free line number list */
static void line_list__free(struct list_head *head)
{
	struct line_node *ln;
	while (!list_empty(head)) {
		ln = list_first_entry(head, struct line_node, list);
		list_del(&ln->list);
		free(ln);
	}
}

/* Dwarf wrappers */

/* Find the realpath of the target file. */
static const char *cu_find_realpath(Dwarf_Die *cu_die, const char *fname)
171
{
172 173
	Dwarf_Files *files;
	size_t nfiles, i;
174
	const char *src = NULL;
175 176 177
	int ret;

	if (!fname)
178
		return NULL;
179

180
	ret = dwarf_getsrcfiles(cu_die, &files, &nfiles);
181 182 183 184 185 186 187
	if (ret != 0)
		return NULL;

	for (i = 0; i < nfiles; i++) {
		src = dwarf_filesrc(files, i, NULL, NULL);
		if (strtailcmp(src, fname) == 0)
			break;
188
	}
189 190
	if (i == nfiles)
		return NULL;
191
	return src;
192 193
}

194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213
/* Compare diename and tname */
static bool die_compare_name(Dwarf_Die *dw_die, const char *tname)
{
	const char *name;
	name = dwarf_diename(dw_die);
	DIE_IF(name == NULL);
	return strcmp(tname, name);
}

/* Get entry pc(or low pc, 1st entry of ranges)  of the die */
static Dwarf_Addr die_get_entrypc(Dwarf_Die *dw_die)
{
	Dwarf_Addr epc;
	int ret;

	ret = dwarf_entrypc(dw_die, &epc);
	DIE_IF(ret == -1);
	return epc;
}

214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235
/* Get type die, but skip qualifiers and typedef */
static Dwarf_Die *die_get_real_type(Dwarf_Die *vr_die, Dwarf_Die *die_mem)
{
	Dwarf_Attribute attr;
	int tag;

	do {
		if (dwarf_attr(vr_die, DW_AT_type, &attr) == NULL ||
		    dwarf_formref_die(&attr, die_mem) == NULL)
			return NULL;

		tag = dwarf_tag(die_mem);
		vr_die = die_mem;
	} while (tag == DW_TAG_const_type ||
		 tag == DW_TAG_restrict_type ||
		 tag == DW_TAG_volatile_type ||
		 tag == DW_TAG_shared_type ||
		 tag == DW_TAG_typedef);

	return die_mem;
}

236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260
static bool die_is_signed_type(Dwarf_Die *tp_die)
{
	Dwarf_Attribute attr;
	Dwarf_Word ret;

	if (dwarf_attr(tp_die, DW_AT_encoding, &attr) == NULL ||
	    dwarf_formudata(&attr, &ret) != 0)
		return false;

	return (ret == DW_ATE_signed_char || ret == DW_ATE_signed ||
		ret == DW_ATE_signed_fixed);
}

static int die_get_byte_size(Dwarf_Die *tp_die)
{
	Dwarf_Attribute attr;
	Dwarf_Word ret;

	if (dwarf_attr(tp_die, DW_AT_byte_size, &attr) == NULL ||
	    dwarf_formudata(&attr, &ret) != 0)
		return 0;

	return (int)ret;
}

261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296
/* Return values for die_find callbacks */
enum {
	DIE_FIND_CB_FOUND = 0,		/* End of Search */
	DIE_FIND_CB_CHILD = 1,		/* Search only children */
	DIE_FIND_CB_SIBLING = 2,	/* Search only siblings */
	DIE_FIND_CB_CONTINUE = 3,	/* Search children and siblings */
};

/* Search a child die */
static Dwarf_Die *die_find_child(Dwarf_Die *rt_die,
				 int (*callback)(Dwarf_Die *, void *),
				 void *data, Dwarf_Die *die_mem)
{
	Dwarf_Die child_die;
	int ret;

	ret = dwarf_child(rt_die, die_mem);
	if (ret != 0)
		return NULL;

	do {
		ret = callback(die_mem, data);
		if (ret == DIE_FIND_CB_FOUND)
			return die_mem;

		if ((ret & DIE_FIND_CB_CHILD) &&
		    die_find_child(die_mem, callback, data, &child_die)) {
			memcpy(die_mem, &child_die, sizeof(Dwarf_Die));
			return die_mem;
		}
	} while ((ret & DIE_FIND_CB_SIBLING) &&
		 dwarf_siblingof(die_mem, die_mem) == 0);

	return NULL;
}

297 298 299 300 301 302
struct __addr_die_search_param {
	Dwarf_Addr	addr;
	Dwarf_Die	*die_mem;
};

static int __die_search_func_cb(Dwarf_Die *fn_die, void *data)
303
{
304
	struct __addr_die_search_param *ad = data;
305

306 307 308 309 310 311 312
	if (dwarf_tag(fn_die) == DW_TAG_subprogram &&
	    dwarf_haspc(fn_die, ad->addr)) {
		memcpy(ad->die_mem, fn_die, sizeof(Dwarf_Die));
		return DWARF_CB_ABORT;
	}
	return DWARF_CB_OK;
}
313

314
/* Search a real subprogram including this line, */
315 316
static Dwarf_Die *die_find_real_subprogram(Dwarf_Die *cu_die, Dwarf_Addr addr,
					   Dwarf_Die *die_mem)
317 318 319 320 321 322 323 324 325
{
	struct __addr_die_search_param ad;
	ad.addr = addr;
	ad.die_mem = die_mem;
	/* dwarf_getscopes can't find subprogram. */
	if (!dwarf_getfuncs(cu_die, __die_search_func_cb, &ad, 0))
		return NULL;
	else
		return die_mem;
326 327
}

328 329
/* die_find callback for inline function search */
static int __die_find_inline_cb(Dwarf_Die *die_mem, void *data)
330
{
331
	Dwarf_Addr *addr = data;
332

333 334 335
	if (dwarf_tag(die_mem) == DW_TAG_inlined_subroutine &&
	    dwarf_haspc(die_mem, *addr))
		return DIE_FIND_CB_FOUND;
336

337
	return DIE_FIND_CB_CONTINUE;
338 339
}

340 341 342
/* Similar to dwarf_getfuncs, but returns inlined_subroutine if exists. */
static Dwarf_Die *die_find_inlinefunc(Dwarf_Die *sp_die, Dwarf_Addr addr,
				      Dwarf_Die *die_mem)
343
{
344
	return die_find_child(sp_die, __die_find_inline_cb, &addr, die_mem);
345 346
}

347
static int __die_find_variable_cb(Dwarf_Die *die_mem, void *data)
348
{
349 350
	const char *name = data;
	int tag;
351

352 353 354 355 356 357 358
	tag = dwarf_tag(die_mem);
	if ((tag == DW_TAG_formal_parameter ||
	     tag == DW_TAG_variable) &&
	    (die_compare_name(die_mem, name) == 0))
		return DIE_FIND_CB_FOUND;

	return DIE_FIND_CB_CONTINUE;
359 360
}

361
/* Find a variable called 'name' */
362 363
static Dwarf_Die *die_find_variable(Dwarf_Die *sp_die, const char *name,
				    Dwarf_Die *die_mem)
364
{
365 366
	return die_find_child(sp_die, __die_find_variable_cb, (void *)name,
			      die_mem);
367 368
}

369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387
static int __die_find_member_cb(Dwarf_Die *die_mem, void *data)
{
	const char *name = data;

	if ((dwarf_tag(die_mem) == DW_TAG_member) &&
	    (die_compare_name(die_mem, name) == 0))
		return DIE_FIND_CB_FOUND;

	return DIE_FIND_CB_SIBLING;
}

/* Find a member called 'name' */
static Dwarf_Die *die_find_member(Dwarf_Die *st_die, const char *name,
				  Dwarf_Die *die_mem)
{
	return die_find_child(st_die, __die_find_member_cb, (void *)name,
			      die_mem);
}

388 389 390 391 392
/*
 * Probe finder related functions
 */

/* Show a location */
393
static void convert_location(Dwarf_Op *op, struct probe_finder *pf)
394
{
395 396
	unsigned int regn;
	Dwarf_Word offs = 0;
397
	bool ref = false;
398
	const char *regs;
399
	struct kprobe_trace_arg *tvar = pf->tvar;
400

401
	/* TODO: support CFA */
402
	/* If this is based on frame buffer, set the offset */
403 404 405
	if (op->atom == DW_OP_fbreg) {
		if (pf->fb_ops == NULL)
			die("The attribute of frame base is not supported.\n");
406
		ref = true;
407 408 409
		offs = op->number;
		op = &pf->fb_ops[0];
	}
410

411 412 413
	if (op->atom >= DW_OP_breg0 && op->atom <= DW_OP_breg31) {
		regn = op->atom - DW_OP_breg0;
		offs += op->number;
414
		ref = true;
415 416 417 418 419
	} else if (op->atom >= DW_OP_reg0 && op->atom <= DW_OP_reg31) {
		regn = op->atom - DW_OP_reg0;
	} else if (op->atom == DW_OP_bregx) {
		regn = op->number;
		offs += op->number2;
420
		ref = true;
421 422
	} else if (op->atom == DW_OP_regx) {
		regn = op->number;
423
	} else
424
		die("DW_OP %d is not supported.", op->atom);
425 426 427

	regs = get_arch_regstr(regn);
	if (!regs)
428
		die("%u exceeds max register number.", regn);
429

430 431 432 433 434
	tvar->value = xstrdup(regs);
	if (ref) {
		tvar->ref = xzalloc(sizeof(struct kprobe_trace_arg_ref));
		tvar->ref->offset = (long)offs;
	}
435 436
}

437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465
static void convert_variable_type(Dwarf_Die *vr_die,
				  struct kprobe_trace_arg *targ)
{
	Dwarf_Die type;
	char buf[16];
	int ret;

	if (die_get_real_type(vr_die, &type) == NULL)
		die("Failed to get a type information of %s.",
		    dwarf_diename(vr_die));

	ret = die_get_byte_size(&type) * 8;
	if (ret) {
		/* Check the bitwidth */
		if (ret > MAX_BASIC_TYPE_BITS) {
			pr_warning("  Warning: %s exceeds max-bitwidth."
				   " Cut down to %d bits.\n",
				   dwarf_diename(&type), MAX_BASIC_TYPE_BITS);
			ret = MAX_BASIC_TYPE_BITS;
		}

		ret = snprintf(buf, 16, "%c%d",
			       die_is_signed_type(&type) ? 's' : 'u', ret);
		if (ret < 0 || ret >= 16)
			die("Failed to convert variable type.");
		targ->type = xstrdup(buf);
	}
}

466 467
static void convert_variable_fields(Dwarf_Die *vr_die, const char *varname,
				    struct perf_probe_arg_field *field,
468 469
				    struct kprobe_trace_arg_ref **ref_ptr,
				    Dwarf_Die *die_mem)
470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488
{
	struct kprobe_trace_arg_ref *ref = *ref_ptr;
	Dwarf_Attribute attr;
	Dwarf_Die type;
	Dwarf_Word offs;

	pr_debug("converting %s in %s\n", field->name, varname);
	if (die_get_real_type(vr_die, &type) == NULL)
		die("Failed to get a type information of %s.", varname);

	/* Check the pointer and dereference */
	if (dwarf_tag(&type) == DW_TAG_pointer_type) {
		if (!field->ref)
			die("Semantic error: %s must be referred by '->'",
			    field->name);
		/* Get the type pointed by this pointer */
		if (die_get_real_type(&type, &type) == NULL)
			die("Failed to get a type information of %s.", varname);

489 490 491 492
		/* Verify it is a data structure  */
		if (dwarf_tag(&type) != DW_TAG_structure_type)
			die("%s is not a data structure.", varname);

493 494 495 496 497 498
		ref = xzalloc(sizeof(struct kprobe_trace_arg_ref));
		if (*ref_ptr)
			(*ref_ptr)->next = ref;
		else
			*ref_ptr = ref;
	} else {
499 500 501 502
		/* Verify it is a data structure  */
		if (dwarf_tag(&type) != DW_TAG_structure_type)
			die("%s is not a data structure.", varname);

503 504 505 506 507 508 509
		if (field->ref)
			die("Semantic error: %s must be referred by '.'",
			    field->name);
		if (!ref)
			die("Structure on a register is not supported yet.");
	}

510
	if (die_find_member(&type, field->name, die_mem) == NULL)
511 512 513 514
		die("%s(tyep:%s) has no member %s.", varname,
		    dwarf_diename(&type), field->name);

	/* Get the offset of the field */
515
	if (dwarf_attr(die_mem, DW_AT_data_member_location, &attr) == NULL ||
516 517 518 519 520 521
	    dwarf_formudata(&attr, &offs) != 0)
		die("Failed to get the offset of %s.", field->name);
	ref->offset += (long)offs;

	/* Converting next field */
	if (field->next)
522 523
		convert_variable_fields(die_mem, field->name, field->next,
					&ref, die_mem);
524 525
}

526
/* Show a variables in kprobe event format */
527
static void convert_variable(Dwarf_Die *vr_die, struct probe_finder *pf)
528 529
{
	Dwarf_Attribute attr;
530
	Dwarf_Die die_mem;
531 532
	Dwarf_Op *expr;
	size_t nexpr;
533 534
	int ret;

535
	if (dwarf_attr(vr_die, DW_AT_location, &attr) == NULL)
536
		goto error;
537
	/* TODO: handle more than 1 exprs */
538
	ret = dwarf_getlocation_addr(&attr, pf->addr, &expr, &nexpr, 1);
539
	if (ret <= 0 || nexpr == 0)
540
		goto error;
541

542
	convert_location(expr, pf);
543

544
	if (pf->pvar->field) {
545
		convert_variable_fields(vr_die, pf->pvar->var,
546 547 548 549 550
					pf->pvar->field, &pf->tvar->ref,
					&die_mem);
		vr_die = &die_mem;
	}
	convert_variable_type(vr_die, pf->tvar);
551
	/* *expr will be cached in libdw. Don't free it. */
552 553
	return ;
error:
554
	/* TODO: Support const_value */
555
	die("Failed to find the location of %s at this address.\n"
556
	    " Perhaps, it has been optimized out.", pf->pvar->var);
557 558 559
}

/* Find a variable in a subprogram die */
560
static void find_variable(Dwarf_Die *sp_die, struct probe_finder *pf)
561
{
562
	Dwarf_Die vr_die;
563
	char buf[32];
564

565 566 567 568 569 570 571 572 573
	/* TODO: Support arrays */
	if (pf->pvar->name)
		pf->tvar->name = xstrdup(pf->pvar->name);
	else {
		synthesize_perf_probe_arg(pf->pvar, buf, 32);
		pf->tvar->name = xstrdup(buf);
	}

	if (!is_c_varname(pf->pvar->var)) {
574
		/* Copy raw parameters */
575
		pf->tvar->value = xstrdup(pf->pvar->var);
576 577
	} else {
		pr_debug("Searching '%s' variable in context.\n",
578
			 pf->pvar->var);
579
		/* Search child die for local variables and parameters. */
580
		if (!die_find_variable(sp_die, pf->pvar->var, &vr_die))
581
			die("Failed to find '%s' in this function.",
582
			    pf->pvar->var);
583
		convert_variable(&vr_die, pf);
584 585 586 587
	}
}

/* Show a probe point to output buffer */
588
static void convert_probe_point(Dwarf_Die *sp_die, struct probe_finder *pf)
589
{
590
	struct kprobe_trace_event *tev;
591 592
	Dwarf_Addr eaddr;
	Dwarf_Die die_mem;
593
	const char *name;
594
	int ret, i;
595 596
	Dwarf_Attribute fb_attr;
	size_t nops;
597

598 599 600 601
	if (pf->ntevs == MAX_PROBES)
		die("Too many( > %d) probe point found.\n", MAX_PROBES);
	tev = &pf->tevs[pf->ntevs++];

602 603
	/* If no real subprogram, find a real one */
	if (!sp_die || dwarf_tag(sp_die) != DW_TAG_subprogram) {
604
		sp_die = die_find_real_subprogram(&pf->cu_die,
605 606 607 608 609
						 pf->addr, &die_mem);
		if (!sp_die)
			die("Probe point is not found in subprograms.");
	}

610
	/* Copy the name of probe point */
611 612
	name = dwarf_diename(sp_die);
	if (name) {
613
		dwarf_entrypc(sp_die, &eaddr);
614 615 616
		tev->point.symbol = xstrdup(name);
		tev->point.offset = (unsigned long)(pf->addr - eaddr);
	} else
617
		/* This function has no name. */
618 619 620 621
		tev->point.offset = (unsigned long)pf->addr;

	pr_debug("Probe point found: %s+%lu\n", tev->point.symbol,
		 tev->point.offset);
622

623 624
	/* Get the frame base attribute/ops */
	dwarf_attr(sp_die, DW_AT_frame_base, &fb_attr);
625
	ret = dwarf_getlocation_addr(&fb_attr, pf->addr, &pf->fb_ops, &nops, 1);
626 627 628
	if (ret <= 0 || nops == 0)
		pf->fb_ops = NULL;

629
	/* Find each argument */
630
	/* TODO: use dwarf_cfi_addrframe */
631 632 633 634 635
	tev->nargs = pf->pev->nargs;
	tev->args = xzalloc(sizeof(struct kprobe_trace_arg) * tev->nargs);
	for (i = 0; i < pf->pev->nargs; i++) {
		pf->pvar = &pf->pev->args[i];
		pf->tvar = &tev->args[i];
636 637
		find_variable(sp_die, pf);
	}
638 639 640

	/* *pf->fb_ops will be cached in libdw. Don't free it. */
	pf->fb_ops = NULL;
641 642 643
}

/* Find probe point from its line number */
644
static void find_probe_point_by_line(struct probe_finder *pf)
645
{
646 647 648
	Dwarf_Lines *lines;
	Dwarf_Line *line;
	size_t nlines, i;
649
	Dwarf_Addr addr;
650
	int lineno;
651 652
	int ret;

653 654
	ret = dwarf_getsrclines(&pf->cu_die, &lines, &nlines);
	DIE_IF(ret != 0);
655

656 657 658
	for (i = 0; i < nlines; i++) {
		line = dwarf_onesrcline(lines, i);
		dwarf_lineno(line, &lineno);
659
		if (lineno != pf->lno)
660 661
			continue;

662 663 664
		/* TODO: Get fileno from line, but how? */
		if (strtailcmp(dwarf_linesrc(line, NULL, NULL), pf->fname) != 0)
			continue;
665

666 667 668 669
		ret = dwarf_lineaddr(line, &addr);
		DIE_IF(ret != 0);
		pr_debug("Probe line found: line[%d]:%d addr:0x%jx\n",
			 (int)i, lineno, (uintmax_t)addr);
670
		pf->addr = addr;
671

672
		convert_probe_point(NULL, pf);
673 674 675 676
		/* Continuing, because target line might be inlined. */
	}
}

677 678 679 680 681 682 683 684 685 686 687 688
/* Find lines which match lazy pattern */
static int find_lazy_match_lines(struct list_head *head,
				 const char *fname, const char *pat)
{
	char *fbuf, *p1, *p2;
	int fd, line, nlines = 0;
	struct stat st;

	fd = open(fname, O_RDONLY);
	if (fd < 0)
		die("failed to open %s", fname);
	DIE_IF(fstat(fd, &st) < 0);
689
	fbuf = xmalloc(st.st_size + 2);
690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722
	DIE_IF(read(fd, fbuf, st.st_size) < 0);
	close(fd);
	fbuf[st.st_size] = '\n';	/* Dummy line */
	fbuf[st.st_size + 1] = '\0';
	p1 = fbuf;
	line = 1;
	while ((p2 = strchr(p1, '\n')) != NULL) {
		*p2 = '\0';
		if (strlazymatch(p1, pat)) {
			line_list__add_line(head, line);
			nlines++;
		}
		line++;
		p1 = p2 + 1;
	}
	free(fbuf);
	return nlines;
}

/* Find probe points from lazy pattern  */
static void find_probe_point_lazy(Dwarf_Die *sp_die, struct probe_finder *pf)
{
	Dwarf_Lines *lines;
	Dwarf_Line *line;
	size_t nlines, i;
	Dwarf_Addr addr;
	Dwarf_Die die_mem;
	int lineno;
	int ret;

	if (list_empty(&pf->lcache)) {
		/* Matching lazy line pattern */
		ret = find_lazy_match_lines(&pf->lcache, pf->fname,
723
					    pf->pev->point.lazy_line);
724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747
		if (ret <= 0)
			die("No matched lines found in %s.", pf->fname);
	}

	ret = dwarf_getsrclines(&pf->cu_die, &lines, &nlines);
	DIE_IF(ret != 0);
	for (i = 0; i < nlines; i++) {
		line = dwarf_onesrcline(lines, i);

		dwarf_lineno(line, &lineno);
		if (!line_list__has_line(&pf->lcache, lineno))
			continue;

		/* TODO: Get fileno from line, but how? */
		if (strtailcmp(dwarf_linesrc(line, NULL, NULL), pf->fname) != 0)
			continue;

		ret = dwarf_lineaddr(line, &addr);
		DIE_IF(ret != 0);
		if (sp_die) {
			/* Address filtering 1: does sp_die include addr? */
			if (!dwarf_haspc(sp_die, addr))
				continue;
			/* Address filtering 2: No child include addr? */
748
			if (die_find_inlinefunc(sp_die, addr, &die_mem))
749 750 751 752 753 754 755
				continue;
		}

		pr_debug("Probe line found: line[%d]:%d addr:0x%llx\n",
			 (int)i, lineno, (unsigned long long)addr);
		pf->addr = addr;

756
		convert_probe_point(sp_die, pf);
757 758 759 760 761
		/* Continuing, because target line might be inlined. */
	}
	/* TODO: deallocate lines, but how? */
}

762 763 764
static int probe_point_inline_cb(Dwarf_Die *in_die, void *data)
{
	struct probe_finder *pf = (struct probe_finder *)data;
765
	struct perf_probe_point *pp = &pf->pev->point;
766

767 768 769 770 771 772 773 774 775
	if (pp->lazy_line)
		find_probe_point_lazy(in_die, pf);
	else {
		/* Get probe address */
		pf->addr = die_get_entrypc(in_die);
		pf->addr += pp->offset;
		pr_debug("found inline addr: 0x%jx\n",
			 (uintmax_t)pf->addr);

776
		convert_probe_point(in_die, pf);
777
	}
778 779 780

	return DWARF_CB_OK;
}
781

782
/* Search function from function name */
783
static int probe_point_search_cb(Dwarf_Die *sp_die, void *data)
784 785
{
	struct probe_finder *pf = (struct probe_finder *)data;
786
	struct perf_probe_point *pp = &pf->pev->point;
787

788 789 790 791 792
	/* Check tag and diename */
	if (dwarf_tag(sp_die) != DW_TAG_subprogram ||
	    die_compare_name(sp_die, pp->function) != 0)
		return 0;

793
	pf->fname = dwarf_decl_file(sp_die);
794 795 796 797 798 799
	if (pp->line) { /* Function relative line */
		dwarf_decl_line(sp_die, &pf->lno);
		pf->lno += pp->line;
		find_probe_point_by_line(pf);
	} else if (!dwarf_func_inline(sp_die)) {
		/* Real function */
800 801 802 803 804 805
		if (pp->lazy_line)
			find_probe_point_lazy(sp_die, pf);
		else {
			pf->addr = die_get_entrypc(sp_die);
			pf->addr += pp->offset;
			/* TODO: Check the address in this function */
806
			convert_probe_point(sp_die, pf);
807
		}
808 809 810 811 812
	} else
		/* Inlined function: search instances */
		dwarf_func_inline_instances(sp_die, probe_point_inline_cb, pf);

	return 1; /* Exit; no same symbol in this CU. */
813 814
}

815
static void find_probe_point_by_func(struct probe_finder *pf)
816
{
817
	dwarf_getfuncs(&pf->cu_die, probe_point_search_cb, pf, 0);
818 819
}

820 821 822
/* Find kprobe_trace_events specified by perf_probe_event from debuginfo */
int find_kprobe_trace_events(int fd, struct perf_probe_event *pev,
			     struct kprobe_trace_event **tevs)
823
{
824 825
	struct probe_finder pf = {.pev = pev};
	struct perf_probe_point *pp = &pev->point;
826 827 828 829 830
	Dwarf_Off off, noff;
	size_t cuhl;
	Dwarf_Die *diep;
	Dwarf *dbg;

831 832 833 834
	pf.tevs = xzalloc(sizeof(struct kprobe_trace_event) * MAX_PROBES);
	*tevs = pf.tevs;
	pf.ntevs = 0;

835 836
	dbg = dwarf_begin(fd, DWARF_C_READ);
	if (!dbg)
837
		return -ENOENT;
838

839
	off = 0;
840
	line_list__init(&pf.lcache);
841 842
	/* Loop on CUs (Compilation Unit) */
	while (!dwarf_nextcu(dbg, off, &noff, &cuhl, NULL, NULL, NULL)) {
843
		/* Get the DIE(Debugging Information Entry) of this CU */
844 845 846
		diep = dwarf_offdie(dbg, off + cuhl, &pf.cu_die);
		if (!diep)
			continue;
847 848 849

		/* Check if target file is included. */
		if (pp->file)
850
			pf.fname = cu_find_realpath(&pf.cu_die, pp->file);
851
		else
852
			pf.fname = NULL;
853

854
		if (!pp->file || pf.fname) {
855
			if (pp->function)
856
				find_probe_point_by_func(&pf);
857 858
			else if (pp->lazy_line)
				find_probe_point_lazy(NULL, &pf);
859 860
			else {
				pf.lno = pp->line;
861
				find_probe_point_by_line(&pf);
862
			}
863
		}
864
		off = noff;
865
	}
866
	line_list__free(&pf.lcache);
867
	dwarf_end(dbg);
868

869
	return pf.ntevs;
870 871
}

872 873 874 875 876 877 878 879 880 881 882 883 884 885 886 887
/* Reverse search */
int find_perf_probe_point(int fd, unsigned long addr,
			  struct perf_probe_point *ppt)
{
	Dwarf_Die cudie, spdie, indie;
	Dwarf *dbg;
	Dwarf_Line *line;
	Dwarf_Addr laddr, eaddr;
	const char *tmp;
	int lineno, ret = 0;

	dbg = dwarf_begin(fd, DWARF_C_READ);
	if (!dbg)
		return -ENOENT;

	/* Find cu die */
888 889 890 891
	if (!dwarf_addrdie(dbg, (Dwarf_Addr)addr, &cudie)) {
		ret = -EINVAL;
		goto end;
	}
892 893 894 895 896 897 898 899 900 901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925 926 927 928 929 930 931 932 933 934 935 936 937 938 939 940 941 942 943

	/* Find a corresponding line */
	line = dwarf_getsrc_die(&cudie, (Dwarf_Addr)addr);
	if (line) {
		dwarf_lineaddr(line, &laddr);
		if ((Dwarf_Addr)addr == laddr) {
			dwarf_lineno(line, &lineno);
			ppt->line = lineno;

			tmp = dwarf_linesrc(line, NULL, NULL);
			DIE_IF(!tmp);
			ppt->file = xstrdup(tmp);
			ret = 1;
		}
	}

	/* Find a corresponding function */
	if (die_find_real_subprogram(&cudie, (Dwarf_Addr)addr, &spdie)) {
		tmp = dwarf_diename(&spdie);
		if (!tmp)
			goto end;

		dwarf_entrypc(&spdie, &eaddr);
		if (!lineno) {
			/* We don't have a line number, let's use offset */
			ppt->function = xstrdup(tmp);
			ppt->offset = addr - (unsigned long)eaddr;
			ret = 1;
			goto end;
		}
		if (die_find_inlinefunc(&spdie, (Dwarf_Addr)addr, &indie)) {
			/* addr in an inline function */
			tmp = dwarf_diename(&indie);
			if (!tmp)
				goto end;
			dwarf_decl_line(&indie, &lineno);
		} else {
			if (eaddr == addr)	/* No offset: function entry */
				lineno = ppt->line;
			else
				dwarf_decl_line(&spdie, &lineno);
		}
		ppt->function = xstrdup(tmp);
		ppt->line -= lineno;	/* Make a relative line number */
	}

end:
	dwarf_end(dbg);
	return ret;
}


944
/* Find line range from its line number */
945
static void find_line_range_by_line(Dwarf_Die *sp_die, struct line_finder *lf)
946
{
947 948 949
	Dwarf_Lines *lines;
	Dwarf_Line *line;
	size_t nlines, i;
950
	Dwarf_Addr addr;
951
	int lineno;
952
	int ret;
953
	const char *src;
954
	Dwarf_Die die_mem;
955

956
	line_list__init(&lf->lr->line_list);
957 958
	ret = dwarf_getsrclines(&lf->cu_die, &lines, &nlines);
	DIE_IF(ret != 0);
959

960 961
	for (i = 0; i < nlines; i++) {
		line = dwarf_onesrcline(lines, i);
962 963
		ret = dwarf_lineno(line, &lineno);
		DIE_IF(ret != 0);
964
		if (lf->lno_s > lineno || lf->lno_e < lineno)
965 966
			continue;

967 968 969 970 971 972 973 974
		if (sp_die) {
			/* Address filtering 1: does sp_die include addr? */
			ret = dwarf_lineaddr(line, &addr);
			DIE_IF(ret != 0);
			if (!dwarf_haspc(sp_die, addr))
				continue;

			/* Address filtering 2: No child include addr? */
975
			if (die_find_inlinefunc(sp_die, addr, &die_mem))
976 977 978
				continue;
		}

979 980 981
		/* TODO: Get fileno from line, but how? */
		src = dwarf_linesrc(line, NULL, NULL);
		if (strtailcmp(src, lf->fname) != 0)
982 983
			continue;

984 985
		/* Copy real path */
		if (!lf->lr->path)
986
			lf->lr->path = xstrdup(src);
987
		line_list__add_line(&lf->lr->line_list, (unsigned int)lineno);
988
	}
989
	/* Update status */
990 991
	if (!list_empty(&lf->lr->line_list))
		lf->found = 1;
992 993 994 995
	else {
		free(lf->lr->path);
		lf->lr->path = NULL;
	}
996 997
}

998 999 1000 1001 1002 1003
static int line_range_inline_cb(Dwarf_Die *in_die, void *data)
{
	find_line_range_by_line(in_die, (struct line_finder *)data);
	return DWARF_CB_ABORT;	/* No need to find other instances */
}

1004
/* Search function from function name */
1005
static int line_range_search_cb(Dwarf_Die *sp_die, void *data)
1006 1007 1008 1009
{
	struct line_finder *lf = (struct line_finder *)data;
	struct line_range *lr = lf->lr;

1010 1011 1012 1013
	if (dwarf_tag(sp_die) == DW_TAG_subprogram &&
	    die_compare_name(sp_die, lr->function) == 0) {
		lf->fname = dwarf_decl_file(sp_die);
		dwarf_decl_line(sp_die, &lr->offset);
1014
		pr_debug("fname: %s, lineno:%d\n", lf->fname, lr->offset);
1015 1016
		lf->lno_s = lr->offset + lr->start;
		if (!lr->end)
1017
			lf->lno_e = INT_MAX;
1018 1019 1020 1021
		else
			lf->lno_e = lr->offset + lr->end;
		lr->start = lf->lno_s;
		lr->end = lf->lno_e;
1022 1023 1024 1025 1026
		if (dwarf_func_inline(sp_die))
			dwarf_func_inline_instances(sp_die,
						    line_range_inline_cb, lf);
		else
			find_line_range_by_line(sp_die, lf);
1027 1028 1029 1030 1031 1032 1033
		return 1;
	}
	return 0;
}

static void find_line_range_by_func(struct line_finder *lf)
{
1034
	dwarf_getfuncs(&lf->cu_die, line_range_search_cb, lf, 0);
1035 1036 1037 1038
}

int find_line_range(int fd, struct line_range *lr)
{
1039
	struct line_finder lf = {.lr = lr, .found = 0};
1040
	int ret;
1041 1042 1043 1044 1045 1046 1047
	Dwarf_Off off = 0, noff;
	size_t cuhl;
	Dwarf_Die *diep;
	Dwarf *dbg;

	dbg = dwarf_begin(fd, DWARF_C_READ);
	if (!dbg)
1048 1049
		return -ENOENT;

1050
	/* Loop on CUs (Compilation Unit) */
1051
	while (!lf.found) {
1052 1053
		ret = dwarf_nextcu(dbg, off, &noff, &cuhl, NULL, NULL, NULL);
		if (ret != 0)
1054 1055 1056
			break;

		/* Get the DIE(Debugging Information Entry) of this CU */
1057 1058 1059
		diep = dwarf_offdie(dbg, off + cuhl, &lf.cu_die);
		if (!diep)
			continue;
1060 1061 1062

		/* Check if target file is included. */
		if (lr->file)
1063
			lf.fname = cu_find_realpath(&lf.cu_die, lr->file);
1064
		else
1065
			lf.fname = 0;
1066

1067
		if (!lr->file || lf.fname) {
1068 1069 1070 1071 1072
			if (lr->function)
				find_line_range_by_func(&lf);
			else {
				lf.lno_s = lr->start;
				if (!lr->end)
1073
					lf.lno_e = INT_MAX;
1074 1075
				else
					lf.lno_e = lr->end;
1076
				find_line_range_by_line(NULL, &lf);
1077 1078
			}
		}
1079
		off = noff;
1080
	}
1081 1082
	pr_debug("path: %lx\n", (unsigned long)lr->path);
	dwarf_end(dbg);
1083 1084 1085
	return lf.found;
}