probe-finder.c 26.3 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
					pf->pvar->field, &pf->tvar->ref,
					&die_mem);
		vr_die = &die_mem;
	}
550 551 552 553
	if (pf->pvar->type)
		pf->tvar->type = xstrdup(pf->pvar->type);
	else
		convert_variable_type(vr_die, pf->tvar);
554
	/* *expr will be cached in libdw. Don't free it. */
555 556
	return ;
error:
557
	/* TODO: Support const_value */
558
	die("Failed to find the location of %s at this address.\n"
559
	    " Perhaps, it has been optimized out.", pf->pvar->var);
560 561 562
}

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

568 569 570 571 572
	/* TODO: Support arrays */
	if (pf->pvar->name)
		pf->tvar->name = xstrdup(pf->pvar->name);
	else {
		synthesize_perf_probe_arg(pf->pvar, buf, 32);
573 574 575
		ptr = strchr(buf, ':');	/* Change type separator to _ */
		if (ptr)
			*ptr = '_';
576 577 578 579
		pf->tvar->name = xstrdup(buf);
	}

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

/* Show a probe point to output buffer */
594
static void convert_probe_point(Dwarf_Die *sp_die, struct probe_finder *pf)
595
{
596
	struct kprobe_trace_event *tev;
597 598
	Dwarf_Addr eaddr;
	Dwarf_Die die_mem;
599
	const char *name;
600
	int ret, i;
601 602
	Dwarf_Attribute fb_attr;
	size_t nops;
603

604 605 606 607
	if (pf->ntevs == MAX_PROBES)
		die("Too many( > %d) probe point found.\n", MAX_PROBES);
	tev = &pf->tevs[pf->ntevs++];

608 609
	/* If no real subprogram, find a real one */
	if (!sp_die || dwarf_tag(sp_die) != DW_TAG_subprogram) {
610
		sp_die = die_find_real_subprogram(&pf->cu_die,
611 612 613 614 615
						 pf->addr, &die_mem);
		if (!sp_die)
			die("Probe point is not found in subprograms.");
	}

616
	/* Copy the name of probe point */
617 618
	name = dwarf_diename(sp_die);
	if (name) {
619
		dwarf_entrypc(sp_die, &eaddr);
620 621 622
		tev->point.symbol = xstrdup(name);
		tev->point.offset = (unsigned long)(pf->addr - eaddr);
	} else
623
		/* This function has no name. */
624 625 626 627
		tev->point.offset = (unsigned long)pf->addr;

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

629 630
	/* Get the frame base attribute/ops */
	dwarf_attr(sp_die, DW_AT_frame_base, &fb_attr);
631
	ret = dwarf_getlocation_addr(&fb_attr, pf->addr, &pf->fb_ops, &nops, 1);
632 633 634
	if (ret <= 0 || nops == 0)
		pf->fb_ops = NULL;

635
	/* Find each argument */
636
	/* TODO: use dwarf_cfi_addrframe */
637 638 639 640 641
	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];
642 643
		find_variable(sp_die, pf);
	}
644 645 646

	/* *pf->fb_ops will be cached in libdw. Don't free it. */
	pf->fb_ops = NULL;
647 648 649
}

/* Find probe point from its line number */
650
static void find_probe_point_by_line(struct probe_finder *pf)
651
{
652 653 654
	Dwarf_Lines *lines;
	Dwarf_Line *line;
	size_t nlines, i;
655
	Dwarf_Addr addr;
656
	int lineno;
657 658
	int ret;

659 660
	ret = dwarf_getsrclines(&pf->cu_die, &lines, &nlines);
	DIE_IF(ret != 0);
661

662 663 664
	for (i = 0; i < nlines; i++) {
		line = dwarf_onesrcline(lines, i);
		dwarf_lineno(line, &lineno);
665
		if (lineno != pf->lno)
666 667
			continue;

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

672 673 674 675
		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);
676
		pf->addr = addr;
677

678
		convert_probe_point(NULL, pf);
679 680 681 682
		/* Continuing, because target line might be inlined. */
	}
}

683 684 685 686 687 688 689 690 691 692 693 694
/* 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);
695
	fbuf = xmalloc(st.st_size + 2);
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 723 724 725 726 727 728
	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,
729
					    pf->pev->point.lazy_line);
730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753
		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? */
754
			if (die_find_inlinefunc(sp_die, addr, &die_mem))
755 756 757 758 759 760 761
				continue;
		}

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

762
		convert_probe_point(sp_die, pf);
763 764 765 766 767
		/* Continuing, because target line might be inlined. */
	}
	/* TODO: deallocate lines, but how? */
}

768 769 770
static int probe_point_inline_cb(Dwarf_Die *in_die, void *data)
{
	struct probe_finder *pf = (struct probe_finder *)data;
771
	struct perf_probe_point *pp = &pf->pev->point;
772

773 774 775 776 777 778 779 780 781
	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);

782
		convert_probe_point(in_die, pf);
783
	}
784 785 786

	return DWARF_CB_OK;
}
787

788
/* Search function from function name */
789
static int probe_point_search_cb(Dwarf_Die *sp_die, void *data)
790 791
{
	struct probe_finder *pf = (struct probe_finder *)data;
792
	struct perf_probe_point *pp = &pf->pev->point;
793

794 795 796 797 798
	/* Check tag and diename */
	if (dwarf_tag(sp_die) != DW_TAG_subprogram ||
	    die_compare_name(sp_die, pp->function) != 0)
		return 0;

799
	pf->fname = dwarf_decl_file(sp_die);
800 801 802 803 804 805
	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 */
806 807 808 809 810 811
		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 */
812
			convert_probe_point(sp_die, pf);
813
		}
814 815 816 817 818
	} 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. */
819 820
}

821
static void find_probe_point_by_func(struct probe_finder *pf)
822
{
823
	dwarf_getfuncs(&pf->cu_die, probe_point_search_cb, pf, 0);
824 825
}

826 827 828
/* 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)
829
{
830 831
	struct probe_finder pf = {.pev = pev};
	struct perf_probe_point *pp = &pev->point;
832 833 834 835 836
	Dwarf_Off off, noff;
	size_t cuhl;
	Dwarf_Die *diep;
	Dwarf *dbg;

837 838 839 840
	pf.tevs = xzalloc(sizeof(struct kprobe_trace_event) * MAX_PROBES);
	*tevs = pf.tevs;
	pf.ntevs = 0;

841 842
	dbg = dwarf_begin(fd, DWARF_C_READ);
	if (!dbg)
843
		return -ENOENT;
844

845
	off = 0;
846
	line_list__init(&pf.lcache);
847 848
	/* Loop on CUs (Compilation Unit) */
	while (!dwarf_nextcu(dbg, off, &noff, &cuhl, NULL, NULL, NULL)) {
849
		/* Get the DIE(Debugging Information Entry) of this CU */
850 851 852
		diep = dwarf_offdie(dbg, off + cuhl, &pf.cu_die);
		if (!diep)
			continue;
853 854 855

		/* Check if target file is included. */
		if (pp->file)
856
			pf.fname = cu_find_realpath(&pf.cu_die, pp->file);
857
		else
858
			pf.fname = NULL;
859

860
		if (!pp->file || pf.fname) {
861
			if (pp->function)
862
				find_probe_point_by_func(&pf);
863 864
			else if (pp->lazy_line)
				find_probe_point_lazy(NULL, &pf);
865 866
			else {
				pf.lno = pp->line;
867
				find_probe_point_by_line(&pf);
868
			}
869
		}
870
		off = noff;
871
	}
872
	line_list__free(&pf.lcache);
873
	dwarf_end(dbg);
874

875
	return pf.ntevs;
876 877
}

878 879 880 881 882 883 884 885 886 887 888 889 890 891 892 893
/* 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 */
894 895 896 897
	if (!dwarf_addrdie(dbg, (Dwarf_Addr)addr, &cudie)) {
		ret = -EINVAL;
		goto end;
	}
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 944 945 946 947 948 949

	/* 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;
}


950
/* Find line range from its line number */
951
static void find_line_range_by_line(Dwarf_Die *sp_die, struct line_finder *lf)
952
{
953 954 955
	Dwarf_Lines *lines;
	Dwarf_Line *line;
	size_t nlines, i;
956
	Dwarf_Addr addr;
957
	int lineno;
958
	int ret;
959
	const char *src;
960
	Dwarf_Die die_mem;
961

962
	line_list__init(&lf->lr->line_list);
963 964
	ret = dwarf_getsrclines(&lf->cu_die, &lines, &nlines);
	DIE_IF(ret != 0);
965

966 967
	for (i = 0; i < nlines; i++) {
		line = dwarf_onesrcline(lines, i);
968 969
		ret = dwarf_lineno(line, &lineno);
		DIE_IF(ret != 0);
970
		if (lf->lno_s > lineno || lf->lno_e < lineno)
971 972
			continue;

973 974 975 976 977 978 979 980
		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? */
981
			if (die_find_inlinefunc(sp_die, addr, &die_mem))
982 983 984
				continue;
		}

985 986 987
		/* TODO: Get fileno from line, but how? */
		src = dwarf_linesrc(line, NULL, NULL);
		if (strtailcmp(src, lf->fname) != 0)
988 989
			continue;

990 991
		/* Copy real path */
		if (!lf->lr->path)
992
			lf->lr->path = xstrdup(src);
993
		line_list__add_line(&lf->lr->line_list, (unsigned int)lineno);
994
	}
995
	/* Update status */
996 997
	if (!list_empty(&lf->lr->line_list))
		lf->found = 1;
998 999 1000 1001
	else {
		free(lf->lr->path);
		lf->lr->path = NULL;
	}
1002 1003
}

1004 1005 1006 1007 1008 1009
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 */
}

1010
/* Search function from function name */
1011
static int line_range_search_cb(Dwarf_Die *sp_die, void *data)
1012 1013 1014 1015
{
	struct line_finder *lf = (struct line_finder *)data;
	struct line_range *lr = lf->lr;

1016 1017 1018 1019
	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);
1020
		pr_debug("fname: %s, lineno:%d\n", lf->fname, lr->offset);
1021 1022
		lf->lno_s = lr->offset + lr->start;
		if (!lr->end)
1023
			lf->lno_e = INT_MAX;
1024 1025 1026 1027
		else
			lf->lno_e = lr->offset + lr->end;
		lr->start = lf->lno_s;
		lr->end = lf->lno_e;
1028 1029 1030 1031 1032
		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);
1033 1034 1035 1036 1037 1038 1039
		return 1;
	}
	return 0;
}

static void find_line_range_by_func(struct line_finder *lf)
{
1040
	dwarf_getfuncs(&lf->cu_die, line_range_search_cb, lf, 0);
1041 1042 1043 1044
}

int find_line_range(int fd, struct line_range *lr)
{
1045
	struct line_finder lf = {.lr = lr, .found = 0};
1046
	int ret;
1047 1048 1049 1050 1051 1052 1053
	Dwarf_Off off = 0, noff;
	size_t cuhl;
	Dwarf_Die *diep;
	Dwarf *dbg;

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

1056
	/* Loop on CUs (Compilation Unit) */
1057
	while (!lf.found) {
1058 1059
		ret = dwarf_nextcu(dbg, off, &noff, &cuhl, NULL, NULL, NULL);
		if (ret != 0)
1060 1061 1062
			break;

		/* Get the DIE(Debugging Information Entry) of this CU */
1063 1064 1065
		diep = dwarf_offdie(dbg, off + cuhl, &lf.cu_die);
		if (!diep)
			continue;
1066 1067 1068

		/* Check if target file is included. */
		if (lr->file)
1069
			lf.fname = cu_find_realpath(&lf.cu_die, lr->file);
1070
		else
1071
			lf.fname = 0;
1072

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