probe-finder.c 37.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
#include <dwarf-regs.h>
35

36
#include <linux/bitops.h>
37 38
#include "event.h"
#include "debug.h"
39
#include "util.h"
40
#include "symbol.h"
41 42
#include "probe-finder.h"

43 44 45
/* Kprobe tracer basic type is up to u64 */
#define MAX_BASIC_TYPE_BITS	64

46 47 48
/* Line number list operations */

/* Add a line to line number list */
49
static int line_list__add_line(struct list_head *head, int line)
50 51 52 53 54 55 56 57 58 59
{
	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 */
60
			return 1;
61 62 63 64 65
	}
	/* List is empty, or the smallest entry */
	p = head;
found:
	pr_debug("line list: add a line %u\n", line);
66 67 68
	ln = zalloc(sizeof(struct line_node));
	if (ln == NULL)
		return -ENOMEM;
69 70 71
	ln->line = line;
	INIT_LIST_HEAD(&ln->list);
	list_add(&ln->list, p);
72
	return 0;
73 74 75
}

/* Check if the line in line number list */
76
static int line_list__has_line(struct list_head *head, int line)
77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104
{
	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);
	}
}

105 106 107 108 109 110 111 112 113 114 115 116 117 118
/* Dwarf FL wrappers */
static char *debuginfo_path;	/* Currently dummy */

static const Dwfl_Callbacks offline_callbacks = {
	.find_debuginfo = dwfl_standard_find_debuginfo,
	.debuginfo_path = &debuginfo_path,

	.section_address = dwfl_offline_section_address,

	/* We use this table for core files too.  */
	.find_elf = dwfl_build_id_find_elf,
};

/* Get a Dwarf from offline image */
119 120
static int debuginfo__init_offline_dwarf(struct debuginfo *self,
					 const char *path)
121 122
{
	Dwfl_Module *mod;
123
	int fd;
124

125 126 127
	fd = open(path, O_RDONLY);
	if (fd < 0)
		return fd;
128

129 130 131
	self->dwfl = dwfl_begin(&offline_callbacks);
	if (!self->dwfl)
		goto error;
132

133
	mod = dwfl_report_offline(self->dwfl, "", "", fd);
134 135 136
	if (!mod)
		goto error;

137 138 139 140 141
	self->dbg = dwfl_module_getdwarf(mod, &self->bias);
	if (!self->dbg)
		goto error;

	return 0;
142
error:
143 144 145 146 147 148 149
	if (self->dwfl)
		dwfl_end(self->dwfl);
	else
		close(fd);
	memset(self, 0, sizeof(*self));

	return -ENOENT;
150 151
}

152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183
#if _ELFUTILS_PREREQ(0, 148)
/* This method is buggy if elfutils is older than 0.148 */
static int __linux_kernel_find_elf(Dwfl_Module *mod,
				   void **userdata,
				   const char *module_name,
				   Dwarf_Addr base,
				   char **file_name, Elf **elfp)
{
	int fd;
	const char *path = kernel_get_module_path(module_name);

	pr_debug2("Use file %s for %s\n", path, module_name);
	if (path) {
		fd = open(path, O_RDONLY);
		if (fd >= 0) {
			*file_name = strdup(path);
			return fd;
		}
	}
	/* If failed, try to call standard method */
	return dwfl_linux_kernel_find_elf(mod, userdata, module_name, base,
					  file_name, elfp);
}

static const Dwfl_Callbacks kernel_callbacks = {
	.find_debuginfo = dwfl_standard_find_debuginfo,
	.debuginfo_path = &debuginfo_path,

	.find_elf = __linux_kernel_find_elf,
	.section_address = dwfl_linux_kernel_module_section_address,
};

184
/* Get a Dwarf from live kernel image */
185 186
static int debuginfo__init_online_kernel_dwarf(struct debuginfo *self,
					       Dwarf_Addr addr)
187
{
188 189 190
	self->dwfl = dwfl_begin(&kernel_callbacks);
	if (!self->dwfl)
		return -EINVAL;
191 192

	/* Load the kernel dwarves: Don't care the result here */
193 194
	dwfl_linux_kernel_report_kernel(self->dwfl);
	dwfl_linux_kernel_report_modules(self->dwfl);
195

196
	self->dbg = dwfl_addrdwarf(self->dwfl, addr, &self->bias);
197
	/* Here, check whether we could get a real dwarf */
198
	if (!self->dbg) {
199 200
		pr_debug("Failed to find kernel dwarf at %lx\n",
			 (unsigned long)addr);
201 202 203
		dwfl_end(self->dwfl);
		memset(self, 0, sizeof(*self));
		return -ENOENT;
204
	}
205 206

	return 0;
207
}
208 209
#else
/* With older elfutils, this just support kernel module... */
210 211
static int debuginfo__init_online_kernel_dwarf(struct debuginfo *self,
					       Dwarf_Addr addr __used)
212 213 214 215 216
{
	const char *path = kernel_get_module_path("kernel");

	if (!path) {
		pr_err("Failed to find vmlinux path\n");
217
		return -ENOENT;
218 219 220
	}

	pr_debug2("Use file %s for debuginfo\n", path);
221 222 223 224 225 226 227 228
	return debuginfo__init_offline_dwarf(self, path);
}
#endif

struct debuginfo *debuginfo__new(const char *path)
{
	struct debuginfo *self = zalloc(sizeof(struct debuginfo));
	if (!self)
229 230
		return NULL;

231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259
	if (debuginfo__init_offline_dwarf(self, path) < 0) {
		free(self);
		self = NULL;
	}

	return self;
}

struct debuginfo *debuginfo__new_online_kernel(unsigned long addr)
{
	struct debuginfo *self = zalloc(sizeof(struct debuginfo));
	if (!self)
		return NULL;

	if (debuginfo__init_online_kernel_dwarf(self, (Dwarf_Addr)addr) < 0) {
		free(self);
		self = NULL;
	}

	return self;
}

void debuginfo__delete(struct debuginfo *self)
{
	if (self) {
		if (self->dwfl)
			dwfl_end(self->dwfl);
		free(self);
	}
260
}
261

262 263 264 265
/*
 * Probe finder related functions
 */

266
static struct probe_trace_arg_ref *alloc_trace_arg_ref(long offs)
267
{
268 269
	struct probe_trace_arg_ref *ref;
	ref = zalloc(sizeof(struct probe_trace_arg_ref));
270 271 272 273 274
	if (ref != NULL)
		ref->offset = offs;
	return ref;
}

275 276 277 278 279 280 281
/*
 * Convert a location into trace_arg.
 * If tvar == NULL, this just checks variable can be converted.
 */
static int convert_variable_location(Dwarf_Die *vr_die, Dwarf_Addr addr,
				     Dwarf_Op *fb_ops,
				     struct probe_trace_arg *tvar)
282
{
283 284 285
	Dwarf_Attribute attr;
	Dwarf_Op *op;
	size_t nops;
286 287
	unsigned int regn;
	Dwarf_Word offs = 0;
288
	bool ref = false;
289
	const char *regs;
290 291
	int ret;

292 293 294
	if (dwarf_attr(vr_die, DW_AT_external, &attr) != NULL)
		goto static_var;

295 296
	/* TODO: handle more than 1 exprs */
	if (dwarf_attr(vr_die, DW_AT_location, &attr) == NULL ||
297
	    dwarf_getlocation_addr(&attr, addr, &op, &nops, 1) <= 0 ||
298 299 300 301 302 303
	    nops == 0) {
		/* TODO: Support const_value */
		return -ENOENT;
	}

	if (op->atom == DW_OP_addr) {
304
static_var:
305 306
		if (!tvar)
			return 0;
307 308 309 310 311 312 313 314 315 316 317
		/* Static variables on memory (not stack), make @varname */
		ret = strlen(dwarf_diename(vr_die));
		tvar->value = zalloc(ret + 2);
		if (tvar->value == NULL)
			return -ENOMEM;
		snprintf(tvar->value, ret + 2, "@%s", dwarf_diename(vr_die));
		tvar->ref = alloc_trace_arg_ref((long)offs);
		if (tvar->ref == NULL)
			return -ENOMEM;
		return 0;
	}
318 319

	/* If this is based on frame buffer, set the offset */
320
	if (op->atom == DW_OP_fbreg) {
321
		if (fb_ops == NULL)
322
			return -ENOTSUP;
323
		ref = true;
324
		offs = op->number;
325
		op = &fb_ops[0];
326
	}
327

328 329 330
	if (op->atom >= DW_OP_breg0 && op->atom <= DW_OP_breg31) {
		regn = op->atom - DW_OP_breg0;
		offs += op->number;
331
		ref = true;
332 333 334 335 336
	} 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;
337
		ref = true;
338 339
	} else if (op->atom == DW_OP_regx) {
		regn = op->number;
340
	} else {
341
		pr_debug("DW_OP %x is not supported.\n", op->atom);
342 343
		return -ENOTSUP;
	}
344

345 346 347
	if (!tvar)
		return 0;

348
	regs = get_arch_regstr(regn);
349
	if (!regs) {
350
		/* This should be a bug in DWARF or this tool */
M
Masami Hiramatsu 已提交
351 352
		pr_warning("Mapping for the register number %u "
			   "missing on this architecture.\n", regn);
353 354
		return -ERANGE;
	}
355

356 357 358 359
	tvar->value = strdup(regs);
	if (tvar->value == NULL)
		return -ENOMEM;

360
	if (ref) {
361
		tvar->ref = alloc_trace_arg_ref((long)offs);
362 363
		if (tvar->ref == NULL)
			return -ENOMEM;
364
	}
365
	return 0;
366 367
}

368 369
#define BYTES_TO_BITS(nb)	((nb) * BITS_PER_LONG / sizeof(long))

370
static int convert_variable_type(Dwarf_Die *vr_die,
371
				 struct probe_trace_arg *tvar,
372
				 const char *cast)
373
{
374
	struct probe_trace_arg_ref **ref_ptr = &tvar->ref;
375 376
	Dwarf_Die type;
	char buf[16];
377
	int bsize, boffs, total;
378 379
	int ret;

380 381 382 383 384 385 386
	/* TODO: check all types */
	if (cast && strcmp(cast, "string") != 0) {
		/* Non string type is OK */
		tvar->type = strdup(cast);
		return (tvar->type == NULL) ? -ENOMEM : 0;
	}

387 388
	bsize = dwarf_bitsize(vr_die);
	if (bsize > 0) {
389
		/* This is a bitfield */
390 391 392 393 394 395
		boffs = dwarf_bitoffset(vr_die);
		total = dwarf_bytesize(vr_die);
		if (boffs < 0 || total < 0)
			return -ENOENT;
		ret = snprintf(buf, 16, "b%d@%d/%zd", bsize, boffs,
				BYTES_TO_BITS(total));
396 397 398
		goto formatted;
	}

399 400 401 402 403
	if (die_get_real_type(vr_die, &type) == NULL) {
		pr_warning("Failed to get a type information of %s.\n",
			   dwarf_diename(vr_die));
		return -ENOENT;
	}
404

405 406 407
	pr_debug("%s type is %s.\n",
		 dwarf_diename(vr_die), dwarf_diename(&type));

408 409 410 411 412
	if (cast && strcmp(cast, "string") == 0) {	/* String type */
		ret = dwarf_tag(&type);
		if (ret != DW_TAG_pointer_type &&
		    ret != DW_TAG_array_type) {
			pr_warning("Failed to cast into string: "
M
Masami Hiramatsu 已提交
413
				   "%s(%s) is not a pointer nor array.\n",
414 415 416 417 418
				   dwarf_diename(vr_die), dwarf_diename(&type));
			return -EINVAL;
		}
		if (ret == DW_TAG_pointer_type) {
			if (die_get_real_type(&type, &type) == NULL) {
M
Masami Hiramatsu 已提交
419 420
				pr_warning("Failed to get a type"
					   " information.\n");
421 422 423 424 425
				return -ENOENT;
			}
			while (*ref_ptr)
				ref_ptr = &(*ref_ptr)->next;
			/* Add new reference with offset +0 */
426
			*ref_ptr = zalloc(sizeof(struct probe_trace_arg_ref));
427 428 429 430 431
			if (*ref_ptr == NULL) {
				pr_warning("Out of memory error\n");
				return -ENOMEM;
			}
		}
432 433
		if (!die_compare_name(&type, "char") &&
		    !die_compare_name(&type, "unsigned char")) {
434
			pr_warning("Failed to cast into string: "
M
Masami Hiramatsu 已提交
435
				   "%s is not (unsigned) char *.\n",
436 437 438 439 440 441 442
				   dwarf_diename(vr_die));
			return -EINVAL;
		}
		tvar->type = strdup(cast);
		return (tvar->type == NULL) ? -ENOMEM : 0;
	}

443 444
	ret = dwarf_bytesize(&type);
	if (ret <= 0)
445 446
		/* No size ... try to use default type */
		return 0;
447
	ret = BYTES_TO_BITS(ret);
448

449 450 451 452 453
	/* Check the bitwidth */
	if (ret > MAX_BASIC_TYPE_BITS) {
		pr_info("%s exceeds max-bitwidth. Cut down to %d bits.\n",
			dwarf_diename(&type), MAX_BASIC_TYPE_BITS);
		ret = MAX_BASIC_TYPE_BITS;
454
	}
455 456 457 458 459 460 461 462 463 464 465 466 467 468
	ret = snprintf(buf, 16, "%c%d",
		       die_is_signed_type(&type) ? 's' : 'u', ret);

formatted:
	if (ret < 0 || ret >= 16) {
		if (ret >= 16)
			ret = -E2BIG;
		pr_warning("Failed to convert variable type: %s\n",
			   strerror(-ret));
		return ret;
	}
	tvar->type = strdup(buf);
	if (tvar->type == NULL)
		return -ENOMEM;
469
	return 0;
470 471
}

472
static int convert_variable_fields(Dwarf_Die *vr_die, const char *varname,
473
				    struct perf_probe_arg_field *field,
474
				    struct probe_trace_arg_ref **ref_ptr,
475
				    Dwarf_Die *die_mem)
476
{
477
	struct probe_trace_arg_ref *ref = *ref_ptr;
478 479
	Dwarf_Die type;
	Dwarf_Word offs;
480
	int ret, tag;
481 482

	pr_debug("converting %s in %s\n", field->name, varname);
483 484 485 486
	if (die_get_real_type(vr_die, &type) == NULL) {
		pr_warning("Failed to get the type of %s.\n", varname);
		return -ENOENT;
	}
487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502
	pr_debug2("Var real type: (%x)\n", (unsigned)dwarf_dieoffset(&type));
	tag = dwarf_tag(&type);

	if (field->name[0] == '[' &&
	    (tag == DW_TAG_array_type || tag == DW_TAG_pointer_type)) {
		if (field->next)
			/* Save original type for next field */
			memcpy(die_mem, &type, sizeof(*die_mem));
		/* Get the type of this array */
		if (die_get_real_type(&type, &type) == NULL) {
			pr_warning("Failed to get the type of %s.\n", varname);
			return -ENOENT;
		}
		pr_debug2("Array real type: (%x)\n",
			 (unsigned)dwarf_dieoffset(&type));
		if (tag == DW_TAG_pointer_type) {
503
			ref = zalloc(sizeof(struct probe_trace_arg_ref));
504 505 506 507 508 509 510
			if (ref == NULL)
				return -ENOMEM;
			if (*ref_ptr)
				(*ref_ptr)->next = ref;
			else
				*ref_ptr = ref;
		}
511
		ref->offset += dwarf_bytesize(&type) * field->index;
512 513 514 515 516 517
		if (!field->next)
			/* Save vr_die for converting types */
			memcpy(die_mem, vr_die, sizeof(*die_mem));
		goto next;
	} else if (tag == DW_TAG_pointer_type) {
		/* Check the pointer and dereference */
518 519 520 521 522
		if (!field->ref) {
			pr_err("Semantic error: %s must be referred by '->'\n",
			       field->name);
			return -EINVAL;
		}
523
		/* Get the type pointed by this pointer */
524 525 526 527
		if (die_get_real_type(&type, &type) == NULL) {
			pr_warning("Failed to get the type of %s.\n", varname);
			return -ENOENT;
		}
528
		/* Verify it is a data structure  */
529 530 531 532
		if (dwarf_tag(&type) != DW_TAG_structure_type) {
			pr_warning("%s is not a data structure.\n", varname);
			return -EINVAL;
		}
533

534
		ref = zalloc(sizeof(struct probe_trace_arg_ref));
535 536
		if (ref == NULL)
			return -ENOMEM;
537 538 539 540 541
		if (*ref_ptr)
			(*ref_ptr)->next = ref;
		else
			*ref_ptr = ref;
	} else {
542
		/* Verify it is a data structure  */
543
		if (tag != DW_TAG_structure_type) {
544 545 546
			pr_warning("%s is not a data structure.\n", varname);
			return -EINVAL;
		}
547
		if (field->name[0] == '[') {
M
Masami Hiramatsu 已提交
548 549
			pr_err("Semantic error: %s is not a pointor"
			       " nor array.\n", varname);
550 551
			return -EINVAL;
		}
552 553 554 555 556 557 558 559 560 561
		if (field->ref) {
			pr_err("Semantic error: %s must be referred by '.'\n",
			       field->name);
			return -EINVAL;
		}
		if (!ref) {
			pr_warning("Structure on a register is not "
				   "supported yet.\n");
			return -ENOTSUP;
		}
562 563
	}

564 565 566 567 568
	if (die_find_member(&type, field->name, die_mem) == NULL) {
		pr_warning("%s(tyep:%s) has no member %s.\n", varname,
			   dwarf_diename(&type), field->name);
		return -EINVAL;
	}
569 570

	/* Get the offset of the field */
571 572
	ret = die_get_data_member_location(die_mem, &offs);
	if (ret < 0) {
573
		pr_warning("Failed to get the offset of %s.\n", field->name);
574
		return ret;
575
	}
576 577
	ref->offset += (long)offs;

578
next:
579 580
	/* Converting next field */
	if (field->next)
581
		return convert_variable_fields(die_mem, field->name,
582
					field->next, &ref, die_mem);
583 584
	else
		return 0;
585 586
}

587
/* Show a variables in kprobe event format */
588
static int convert_variable(Dwarf_Die *vr_die, struct probe_finder *pf)
589
{
590
	Dwarf_Die die_mem;
591 592
	int ret;

593 594
	pr_debug("Converting variable %s into trace event.\n",
		 dwarf_diename(vr_die));
595

596 597 598 599 600 601 602 603
	ret = convert_variable_location(vr_die, pf->addr, pf->fb_ops,
					pf->tvar);
	if (ret == -ENOENT)
		pr_err("Failed to find the location of %s at this address.\n"
		       " Perhaps, it has been optimized out.\n", pf->pvar->var);
	else if (ret == -ENOTSUP)
		pr_err("Sorry, we don't support this variable location yet.\n");
	else if (pf->pvar->field) {
604 605 606
		ret = convert_variable_fields(vr_die, pf->pvar->var,
					      pf->pvar->field, &pf->tvar->ref,
					      &die_mem);
607 608
		vr_die = &die_mem;
	}
609 610
	if (ret == 0)
		ret = convert_variable_type(vr_die, pf->tvar, pf->pvar->type);
611
	/* *expr will be cached in libdw. Don't free it. */
612
	return ret;
613 614 615
}

/* Find a variable in a subprogram die */
616
static int find_variable(Dwarf_Die *sp_die, struct probe_finder *pf)
617
{
618
	Dwarf_Die vr_die, *scopes;
619
	char buf[32], *ptr;
620
	int ret, nscopes;
621

622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640
	if (!is_c_varname(pf->pvar->var)) {
		/* Copy raw parameters */
		pf->tvar->value = strdup(pf->pvar->var);
		if (pf->tvar->value == NULL)
			return -ENOMEM;
		if (pf->pvar->type) {
			pf->tvar->type = strdup(pf->pvar->type);
			if (pf->tvar->type == NULL)
				return -ENOMEM;
		}
		if (pf->pvar->name) {
			pf->tvar->name = strdup(pf->pvar->name);
			if (pf->tvar->name == NULL)
				return -ENOMEM;
		} else
			pf->tvar->name = NULL;
		return 0;
	}

641
	if (pf->pvar->name)
642
		pf->tvar->name = strdup(pf->pvar->name);
643
	else {
644 645 646
		ret = synthesize_perf_probe_arg(pf->pvar, buf, 32);
		if (ret < 0)
			return ret;
647 648 649
		ptr = strchr(buf, ':');	/* Change type separator to _ */
		if (ptr)
			*ptr = '_';
650
		pf->tvar->name = strdup(buf);
651
	}
652 653
	if (pf->tvar->name == NULL)
		return -ENOMEM;
654

655 656 657
	pr_debug("Searching '%s' variable in context.\n",
		 pf->pvar->var);
	/* Search child die for local variables and parameters. */
658
	if (die_find_variable_at(sp_die, pf->pvar->var, pf->addr, &vr_die))
659 660 661 662
		ret = convert_variable(&vr_die, pf);
	else {
		/* Search upper class */
		nscopes = dwarf_getscopes_die(sp_die, &scopes);
663
		ret = -ENOENT;
664 665 666 667 668 669 670
		while (nscopes-- > 1) {
			pr_debug("Searching variables in %s\n",
				 dwarf_diename(&scopes[nscopes]));
			/* We should check this scope, so give dummy address */
			if (die_find_variable_at(&scopes[nscopes],
						 pf->pvar->var, 0,
						 &vr_die)) {
671
				ret = convert_variable(&vr_die, pf);
672
				break;
673 674 675
			}
		}
		if (scopes)
676 677 678
			free(scopes);
	}
	if (ret < 0)
679 680
		pr_warning("Failed to find '%s' in this function.\n",
			   pf->pvar->var);
681
	return ret;
682 683
}

684 685 686
/* Convert subprogram DIE to trace point */
static int convert_to_trace_point(Dwarf_Die *sp_die, Dwarf_Addr paddr,
				  bool retprobe, struct probe_trace_point *tp)
687
{
688
	Dwarf_Addr eaddr;
689
	const char *name;
690

691
	/* Copy the name of probe point */
692 693
	name = dwarf_diename(sp_die);
	if (name) {
694
		if (dwarf_entrypc(sp_die, &eaddr) != 0) {
M
Masami Hiramatsu 已提交
695
			pr_warning("Failed to get entry address of %s\n",
696 697 698
				   dwarf_diename(sp_die));
			return -ENOENT;
		}
699 700
		tp->symbol = strdup(name);
		if (tp->symbol == NULL)
701
			return -ENOMEM;
702
		tp->offset = (unsigned long)(paddr - eaddr);
703
	} else
704
		/* This function has no name. */
705
		tp->offset = (unsigned long)paddr;
706

707
	/* Return probe must be on the head of a subprogram */
708 709
	if (retprobe) {
		if (eaddr != paddr) {
710
			pr_warning("Return probe must be on the head of"
M
Masami Hiramatsu 已提交
711
				   " a real function.\n");
712 713
			return -EINVAL;
		}
714
		tp->retprobe = true;
715 716
	}

717 718 719 720 721 722 723 724 725 726 727 728 729
	return 0;
}

/* Call probe_finder callback with real subprogram DIE */
static int call_probe_finder(Dwarf_Die *sp_die, struct probe_finder *pf)
{
	Dwarf_Die die_mem;
	Dwarf_Attribute fb_attr;
	size_t nops;
	int ret;

	/* If no real subprogram, find a real one */
	if (!sp_die || dwarf_tag(sp_die) != DW_TAG_subprogram) {
730
		sp_die = die_find_realfunc(&pf->cu_die, pf->addr, &die_mem);
731 732 733 734 735 736
		if (!sp_die) {
			pr_warning("Failed to find probe point in any "
				   "functions.\n");
			return -ENOENT;
		}
	}
737

738 739
	/* Get the frame base attribute/ops */
	dwarf_attr(sp_die, DW_AT_frame_base, &fb_attr);
740
	ret = dwarf_getlocation_addr(&fb_attr, pf->addr, &pf->fb_ops, &nops, 1);
741
	if (ret <= 0 || nops == 0) {
742
		pf->fb_ops = NULL;
743
#if _ELFUTILS_PREREQ(0, 142)
744 745 746
	} else if (nops == 1 && pf->fb_ops[0].atom == DW_OP_call_frame_cfa &&
		   pf->cfi != NULL) {
		Dwarf_Frame *frame;
747 748
		if (dwarf_cfi_addrframe(pf->cfi, pf->addr, &frame) != 0 ||
		    dwarf_frame_cfa(frame, &pf->fb_ops, &nops) != 0) {
M
Masami Hiramatsu 已提交
749
			pr_warning("Failed to get call frame on 0x%jx\n",
750 751 752
				   (uintmax_t)pf->addr);
			return -ENOENT;
		}
753
#endif
754
	}
755

756 757
	/* Call finder's callback handler */
	ret = pf->callback(sp_die, pf);
758 759 760

	/* *pf->fb_ops will be cached in libdw. Don't free it. */
	pf->fb_ops = NULL;
761 762

	return ret;
763 764
}

765 766
static int probe_point_line_walker(const char *fname, int lineno,
				   Dwarf_Addr addr, void *data)
767
{
768 769
	struct probe_finder *pf = data;
	int ret;
770

771 772
	if (lineno != pf->lno || strtailcmp(fname, pf->fname) != 0)
		return 0;
773

774 775
	pf->addr = addr;
	ret = call_probe_finder(NULL, pf);
776

777
	/* Continue if no error, because the line will be in inline function */
778
	return ret < 0 ? ret : 0;
779
}
780

781 782 783 784
/* Find probe point from its line number */
static int find_probe_point_by_line(struct probe_finder *pf)
{
	return die_walk_lines(&pf->cu_die, probe_point_line_walker, pf);
785 786
}

787 788 789 790
/* Find lines which match lazy pattern */
static int find_lazy_match_lines(struct list_head *head,
				 const char *fname, const char *pat)
{
791 792 793 794 795 796 797 798 799
	FILE *fp;
	char *line = NULL;
	size_t line_len;
	ssize_t len;
	int count = 0, linenum = 1;

	fp = fopen(fname, "r");
	if (!fp) {
		pr_warning("Failed to open %s: %s\n", fname, strerror(errno));
800
		return -errno;
801 802
	}

803
	while ((len = getline(&line, &line_len, fp)) > 0) {
804

805 806 807 808 809 810
		if (line[len - 1] == '\n')
			line[len - 1] = '\0';

		if (strlazymatch(line, pat)) {
			line_list__add_line(head, linenum);
			count++;
811
		}
812
		linenum++;
813
	}
814 815 816 817 818 819 820 821 822

	if (ferror(fp))
		count = -errno;
	free(line);
	fclose(fp);

	if (count == 0)
		pr_debug("No matched lines found in %s.\n", fname);
	return count;
823 824
}

825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843
static int probe_point_lazy_walker(const char *fname, int lineno,
				   Dwarf_Addr addr, void *data)
{
	struct probe_finder *pf = data;
	int ret;

	if (!line_list__has_line(&pf->lcache, lineno) ||
	    strtailcmp(fname, pf->fname) != 0)
		return 0;

	pr_debug("Probe line found: line:%d addr:0x%llx\n",
		 lineno, (unsigned long long)addr);
	pf->addr = addr;
	ret = call_probe_finder(NULL, pf);

	/*
	 * Continue if no error, because the lazy pattern will match
	 * to other lines
	 */
844
	return ret < 0 ? ret : 0;
845 846
}

847
/* Find probe points from lazy pattern  */
848
static int find_probe_point_lazy(Dwarf_Die *sp_die, struct probe_finder *pf)
849
{
850
	int ret = 0;
851 852 853 854

	if (list_empty(&pf->lcache)) {
		/* Matching lazy line pattern */
		ret = find_lazy_match_lines(&pf->lcache, pf->fname,
855
					    pf->pev->point.lazy_line);
856
		if (ret <= 0)
857
			return ret;
858 859
	}

860
	return die_walk_lines(sp_die, probe_point_lazy_walker, pf);
861 862
}

863 864 865 866 867 868
/* Callback parameter with return value */
struct dwarf_callback_param {
	void *data;
	int retval;
};

869 870
static int probe_point_inline_cb(Dwarf_Die *in_die, void *data)
{
871 872
	struct dwarf_callback_param *param = data;
	struct probe_finder *pf = param->data;
873
	struct perf_probe_point *pp = &pf->pev->point;
874
	Dwarf_Addr addr;
875

876
	if (pp->lazy_line)
877
		param->retval = find_probe_point_lazy(in_die, pf);
878 879
	else {
		/* Get probe address */
880
		if (dwarf_entrypc(in_die, &addr) != 0) {
M
Masami Hiramatsu 已提交
881
			pr_warning("Failed to get entry address of %s.\n",
882 883 884 885 886
				   dwarf_diename(in_die));
			param->retval = -ENOENT;
			return DWARF_CB_ABORT;
		}
		pf->addr = addr;
887 888 889 890
		pf->addr += pp->offset;
		pr_debug("found inline addr: 0x%jx\n",
			 (uintmax_t)pf->addr);

891
		param->retval = call_probe_finder(in_die, pf);
892 893
		if (param->retval < 0)
			return DWARF_CB_ABORT;
894
	}
895 896 897

	return DWARF_CB_OK;
}
898

899
/* Search function from function name */
900
static int probe_point_search_cb(Dwarf_Die *sp_die, void *data)
901
{
902 903
	struct dwarf_callback_param *param = data;
	struct probe_finder *pf = param->data;
904
	struct perf_probe_point *pp = &pf->pev->point;
905

906 907
	/* Check tag and diename */
	if (dwarf_tag(sp_die) != DW_TAG_subprogram ||
908
	    !die_compare_name(sp_die, pp->function))
909
		return DWARF_CB_OK;
910

911 912 913 914
	/* Check declared file */
	if (pp->file && strtailcmp(pp->file, dwarf_decl_file(sp_die)))
		return DWARF_CB_OK;

915
	pf->fname = dwarf_decl_file(sp_die);
916 917 918
	if (pp->line) { /* Function relative line */
		dwarf_decl_line(sp_die, &pf->lno);
		pf->lno += pp->line;
919
		param->retval = find_probe_point_by_line(pf);
920 921
	} else if (!dwarf_func_inline(sp_die)) {
		/* Real function */
922
		if (pp->lazy_line)
923
			param->retval = find_probe_point_lazy(sp_die, pf);
924
		else {
925
			if (dwarf_entrypc(sp_die, &pf->addr) != 0) {
M
Masami Hiramatsu 已提交
926 927
				pr_warning("Failed to get entry address of "
					   "%s.\n", dwarf_diename(sp_die));
928 929 930
				param->retval = -ENOENT;
				return DWARF_CB_ABORT;
			}
931 932
			pf->addr += pp->offset;
			/* TODO: Check the address in this function */
933
			param->retval = call_probe_finder(sp_die, pf);
934
		}
935 936 937
	} else {
		struct dwarf_callback_param _param = {.data = (void *)pf,
						      .retval = 0};
938
		/* Inlined function: search instances */
939 940 941 942
		dwarf_func_inline_instances(sp_die, probe_point_inline_cb,
					    &_param);
		param->retval = _param.retval;
	}
943

944
	return DWARF_CB_ABORT; /* Exit; no same symbol in this CU. */
945 946
}

947
static int find_probe_point_by_func(struct probe_finder *pf)
948
{
949 950 951 952
	struct dwarf_callback_param _param = {.data = (void *)pf,
					      .retval = 0};
	dwarf_getfuncs(&pf->cu_die, probe_point_search_cb, &_param, 0);
	return _param.retval;
953 954
}

955 956 957 958 959 960 961 962 963 964 965 966 967 968 969 970 971 972 973 974 975 976 977 978 979 980 981 982 983 984 985 986
struct pubname_callback_param {
	char *function;
	char *file;
	Dwarf_Die *cu_die;
	Dwarf_Die *sp_die;
	int found;
};

static int pubname_search_cb(Dwarf *dbg, Dwarf_Global *gl, void *data)
{
	struct pubname_callback_param *param = data;

	if (dwarf_offdie(dbg, gl->die_offset, param->sp_die)) {
		if (dwarf_tag(param->sp_die) != DW_TAG_subprogram)
			return DWARF_CB_OK;

		if (die_compare_name(param->sp_die, param->function)) {
			if (!dwarf_offdie(dbg, gl->cu_offset, param->cu_die))
				return DWARF_CB_OK;

			if (param->file &&
			    strtailcmp(param->file, dwarf_decl_file(param->sp_die)))
				return DWARF_CB_OK;

			param->found = 1;
			return DWARF_CB_ABORT;
		}
	}

	return DWARF_CB_OK;
}

987
/* Find probe points from debuginfo */
988 989
static int debuginfo__find_probes(struct debuginfo *self,
				  struct probe_finder *pf)
990
{
991
	struct perf_probe_point *pp = &pf->pev->point;
992 993 994
	Dwarf_Off off, noff;
	size_t cuhl;
	Dwarf_Die *diep;
995
	int ret = 0;
996

997
#if _ELFUTILS_PREREQ(0, 142)
998
	/* Get the call frame information from this dwarf */
999
	pf->cfi = dwarf_getcfi(self->dbg);
1000
#endif
1001

1002
	off = 0;
1003
	line_list__init(&pf->lcache);
1004 1005 1006 1007 1008 1009 1010 1011

	/* Fastpath: lookup by function name from .debug_pubnames section */
	if (pp->function) {
		struct pubname_callback_param pubname_param = {
			.function = pp->function,
			.file	  = pp->file,
			.cu_die	  = &pf->cu_die,
			.sp_die	  = &pf->sp_die,
1012
			.found	  = 0,
1013 1014 1015 1016 1017
		};
		struct dwarf_callback_param probe_param = {
			.data = pf,
		};

1018 1019
		dwarf_getpubnames(self->dbg, pubname_search_cb,
				  &pubname_param, 0);
1020 1021 1022 1023 1024 1025 1026
		if (pubname_param.found) {
			ret = probe_point_search_cb(&pf->sp_die, &probe_param);
			if (ret)
				goto found;
		}
	}

1027
	/* Loop on CUs (Compilation Unit) */
1028
	while (!dwarf_nextcu(self->dbg, off, &noff, &cuhl, NULL, NULL, NULL)) {
1029
		/* Get the DIE(Debugging Information Entry) of this CU */
1030
		diep = dwarf_offdie(self->dbg, off + cuhl, &pf->cu_die);
1031 1032
		if (!diep)
			continue;
1033 1034 1035

		/* Check if target file is included. */
		if (pp->file)
1036
			pf->fname = cu_find_realpath(&pf->cu_die, pp->file);
1037
		else
1038
			pf->fname = NULL;
1039

1040
		if (!pp->file || pf->fname) {
1041
			if (pp->function)
1042
				ret = find_probe_point_by_func(pf);
1043
			else if (pp->lazy_line)
1044
				ret = find_probe_point_lazy(NULL, pf);
1045
			else {
1046 1047
				pf->lno = pp->line;
				ret = find_probe_point_by_line(pf);
1048
			}
1049
			if (ret < 0)
1050
				break;
1051
		}
1052
		off = noff;
1053
	}
1054 1055

found:
1056
	line_list__free(&pf->lcache);
1057

1058 1059 1060 1061 1062 1063 1064 1065 1066 1067 1068 1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101
	return ret;
}

/* Add a found probe point into trace event list */
static int add_probe_trace_event(Dwarf_Die *sp_die, struct probe_finder *pf)
{
	struct trace_event_finder *tf =
			container_of(pf, struct trace_event_finder, pf);
	struct probe_trace_event *tev;
	int ret, i;

	/* Check number of tevs */
	if (tf->ntevs == tf->max_tevs) {
		pr_warning("Too many( > %d) probe point found.\n",
			   tf->max_tevs);
		return -ERANGE;
	}
	tev = &tf->tevs[tf->ntevs++];

	ret = convert_to_trace_point(sp_die, pf->addr, pf->pev->point.retprobe,
				     &tev->point);
	if (ret < 0)
		return ret;

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

	/* Find each argument */
	tev->nargs = pf->pev->nargs;
	tev->args = zalloc(sizeof(struct probe_trace_arg) * tev->nargs);
	if (tev->args == NULL)
		return -ENOMEM;
	for (i = 0; i < pf->pev->nargs; i++) {
		pf->pvar = &pf->pev->args[i];
		pf->tvar = &tev->args[i];
		ret = find_variable(sp_die, pf);
		if (ret != 0)
			return ret;
	}

	return 0;
}

/* Find probe_trace_events specified by perf_probe_event from debuginfo */
1102 1103 1104
int debuginfo__find_trace_events(struct debuginfo *self,
				 struct perf_probe_event *pev,
				 struct probe_trace_event **tevs, int max_tevs)
1105 1106 1107 1108 1109 1110 1111 1112 1113 1114 1115 1116 1117 1118
{
	struct trace_event_finder tf = {
			.pf = {.pev = pev, .callback = add_probe_trace_event},
			.max_tevs = max_tevs};
	int ret;

	/* Allocate result tevs array */
	*tevs = zalloc(sizeof(struct probe_trace_event) * max_tevs);
	if (*tevs == NULL)
		return -ENOMEM;

	tf.tevs = *tevs;
	tf.ntevs = 0;

1119
	ret = debuginfo__find_probes(self, &tf.pf);
1120 1121 1122 1123 1124 1125 1126 1127 1128 1129 1130 1131 1132 1133 1134 1135 1136 1137 1138 1139 1140 1141 1142 1143 1144 1145 1146 1147
	if (ret < 0) {
		free(*tevs);
		*tevs = NULL;
		return ret;
	}

	return (ret < 0) ? ret : tf.ntevs;
}

#define MAX_VAR_LEN 64

/* Collect available variables in this scope */
static int collect_variables_cb(Dwarf_Die *die_mem, void *data)
{
	struct available_var_finder *af = data;
	struct variable_list *vl;
	char buf[MAX_VAR_LEN];
	int tag, ret;

	vl = &af->vls[af->nvls - 1];

	tag = dwarf_tag(die_mem);
	if (tag == DW_TAG_formal_parameter ||
	    tag == DW_TAG_variable) {
		ret = convert_variable_location(die_mem, af->pf.addr,
						af->pf.fb_ops, NULL);
		if (ret == 0) {
			ret = die_get_varname(die_mem, buf, MAX_VAR_LEN);
1148
			pr_debug2("Add new var: %s\n", buf);
1149 1150 1151 1152 1153
			if (ret > 0)
				strlist__add(vl->vars, buf);
		}
	}

1154
	if (af->child && dwarf_haspc(die_mem, af->pf.addr))
1155 1156 1157 1158 1159 1160 1161 1162 1163 1164 1165
		return DIE_FIND_CB_CONTINUE;
	else
		return DIE_FIND_CB_SIBLING;
}

/* Add a found vars into available variables list */
static int add_available_vars(Dwarf_Die *sp_die, struct probe_finder *pf)
{
	struct available_var_finder *af =
			container_of(pf, struct available_var_finder, pf);
	struct variable_list *vl;
1166 1167
	Dwarf_Die die_mem, *scopes = NULL;
	int ret, nscopes;
1168 1169 1170 1171 1172 1173 1174 1175 1176 1177 1178 1179 1180 1181 1182 1183 1184 1185 1186 1187

	/* Check number of tevs */
	if (af->nvls == af->max_vls) {
		pr_warning("Too many( > %d) probe point found.\n", af->max_vls);
		return -ERANGE;
	}
	vl = &af->vls[af->nvls++];

	ret = convert_to_trace_point(sp_die, pf->addr, pf->pev->point.retprobe,
				     &vl->point);
	if (ret < 0)
		return ret;

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

	/* Find local variables */
	vl->vars = strlist__new(true, NULL);
	if (vl->vars == NULL)
		return -ENOMEM;
1188
	af->child = true;
1189 1190
	die_find_child(sp_die, collect_variables_cb, (void *)af, &die_mem);

1191 1192 1193 1194 1195 1196 1197 1198 1199 1200 1201 1202 1203
	/* Find external variables */
	if (!af->externs)
		goto out;
	/* Don't need to search child DIE for externs. */
	af->child = false;
	nscopes = dwarf_getscopes_die(sp_die, &scopes);
	while (nscopes-- > 1)
		die_find_child(&scopes[nscopes], collect_variables_cb,
			       (void *)af, &die_mem);
	if (scopes)
		free(scopes);

out:
1204 1205 1206 1207 1208 1209 1210 1211 1212
	if (strlist__empty(vl->vars)) {
		strlist__delete(vl->vars);
		vl->vars = NULL;
	}

	return ret;
}

/* Find available variables at given probe point */
1213 1214 1215 1216
int debuginfo__find_available_vars_at(struct debuginfo *self,
				      struct perf_probe_event *pev,
				      struct variable_list **vls,
				      int max_vls, bool externs)
1217 1218 1219
{
	struct available_var_finder af = {
			.pf = {.pev = pev, .callback = add_available_vars},
1220
			.max_vls = max_vls, .externs = externs};
1221 1222 1223 1224 1225 1226 1227 1228 1229 1230
	int ret;

	/* Allocate result vls array */
	*vls = zalloc(sizeof(struct variable_list) * max_vls);
	if (*vls == NULL)
		return -ENOMEM;

	af.vls = *vls;
	af.nvls = 0;

1231
	ret = debuginfo__find_probes(self, &af.pf);
1232 1233 1234 1235 1236 1237 1238 1239 1240 1241 1242 1243 1244 1245
	if (ret < 0) {
		/* Free vlist for error */
		while (af.nvls--) {
			if (af.vls[af.nvls].point.symbol)
				free(af.vls[af.nvls].point.symbol);
			if (af.vls[af.nvls].vars)
				strlist__delete(af.vls[af.nvls].vars);
		}
		free(af.vls);
		*vls = NULL;
		return ret;
	}

	return (ret < 0) ? ret : af.nvls;
1246 1247
}

1248
/* Reverse search */
1249 1250
int debuginfo__find_probe_point(struct debuginfo *self, unsigned long addr,
				struct perf_probe_point *ppt)
1251 1252
{
	Dwarf_Die cudie, spdie, indie;
1253
	Dwarf_Addr _addr, baseaddr;
1254 1255
	const char *fname = NULL, *func = NULL, *tmp;
	int baseline = 0, lineno = 0, ret = 0;
1256

1257
	/* Adjust address with bias */
1258 1259
	addr += self->bias;

1260
	/* Find cu die */
1261
	if (!dwarf_addrdie(self->dbg, (Dwarf_Addr)addr - self->bias, &cudie)) {
M
Masami Hiramatsu 已提交
1262 1263
		pr_warning("Failed to find debug information for address %lx\n",
			   addr);
1264 1265 1266
		ret = -EINVAL;
		goto end;
	}
1267

1268 1269 1270
	/* Find a corresponding line (filename and lineno) */
	cu_find_lineinfo(&cudie, addr, &fname, &lineno);
	/* Don't care whether it failed or not */
1271

1272
	/* Find a corresponding function (name, baseline and baseaddr) */
1273
	if (die_find_realfunc(&cudie, (Dwarf_Addr)addr, &spdie)) {
1274
		/* Get function entry information */
1275
		tmp = dwarf_diename(&spdie);
1276 1277 1278 1279 1280 1281 1282 1283 1284 1285 1286 1287 1288 1289 1290 1291 1292 1293 1294 1295 1296 1297 1298 1299 1300 1301
		if (!tmp ||
		    dwarf_entrypc(&spdie, &baseaddr) != 0 ||
		    dwarf_decl_line(&spdie, &baseline) != 0)
			goto post;
		func = tmp;

		if (addr == (unsigned long)baseaddr)
			/* Function entry - Relative line number is 0 */
			lineno = baseline;
		else if (die_find_inlinefunc(&spdie, (Dwarf_Addr)addr,
					     &indie)) {
			if (dwarf_entrypc(&indie, &_addr) == 0 &&
			    _addr == addr)
				/*
				 * addr is at an inline function entry.
				 * In this case, lineno should be the call-site
				 * line number.
				 */
				lineno = die_get_call_lineno(&indie);
			else {
				/*
				 * addr is in an inline function body.
				 * Since lineno points one of the lines
				 * of the inline function, baseline should
				 * be the entry line of the inline function.
				 */
1302
				tmp = dwarf_diename(&indie);
1303 1304 1305
				if (tmp &&
				    dwarf_decl_line(&spdie, &baseline) == 0)
					func = tmp;
1306
			}
1307
		}
1308 1309 1310 1311 1312 1313 1314 1315 1316 1317 1318 1319
	}

post:
	/* Make a relative line number or an offset */
	if (lineno)
		ppt->line = lineno - baseline;
	else if (func)
		ppt->offset = addr - (unsigned long)baseaddr;

	/* Duplicate strings */
	if (func) {
		ppt->function = strdup(func);
1320 1321 1322 1323
		if (ppt->function == NULL) {
			ret = -ENOMEM;
			goto end;
		}
1324
	}
1325 1326 1327 1328 1329 1330 1331 1332 1333 1334 1335
	if (fname) {
		ppt->file = strdup(fname);
		if (ppt->file == NULL) {
			if (ppt->function) {
				free(ppt->function);
				ppt->function = NULL;
			}
			ret = -ENOMEM;
			goto end;
		}
	}
1336
end:
1337 1338
	if (ret == 0 && (fname || func))
		ret = 1;	/* Found a point */
1339 1340 1341
	return ret;
}

1342 1343 1344 1345
/* Add a line and store the src path */
static int line_range_add_line(const char *src, unsigned int lineno,
			       struct line_range *lr)
{
1346
	/* Copy source path */
1347
	if (!lr->path) {
1348 1349 1350
		lr->path = strdup(src);
		if (lr->path == NULL)
			return -ENOMEM;
1351 1352 1353 1354
	}
	return line_list__add_line(&lr->line_list, lineno);
}

1355 1356 1357
static int line_range_walk_cb(const char *fname, int lineno,
			      Dwarf_Addr addr __used,
			      void *data)
1358
{
1359
	struct line_finder *lf = data;
1360

1361
	if ((strtailcmp(fname, lf->fname) != 0) ||
1362
	    (lf->lno_s > lineno || lf->lno_e < lineno))
1363
		return 0;
1364

1365 1366
	if (line_range_add_line(fname, lineno, lf->lr) < 0)
		return -EINVAL;
1367

1368
	return 0;
1369
}
1370

1371
/* Find line range from its line number */
1372
static int find_line_range_by_line(Dwarf_Die *sp_die, struct line_finder *lf)
1373
{
1374
	int ret;
1375

1376
	ret = die_walk_lines(sp_die ?: &lf->cu_die, line_range_walk_cb, lf);
1377

1378
	/* Update status */
1379 1380 1381 1382 1383
	if (ret >= 0)
		if (!list_empty(&lf->lr->line_list))
			ret = lf->found = 1;
		else
			ret = 0;	/* Lines are not found */
1384 1385 1386 1387
	else {
		free(lf->lr->path);
		lf->lr->path = NULL;
	}
1388
	return ret;
1389 1390
}

1391 1392
static int line_range_inline_cb(Dwarf_Die *in_die, void *data)
{
1393 1394 1395
	struct dwarf_callback_param *param = data;

	param->retval = find_line_range_by_line(in_die, param->data);
1396 1397 1398 1399 1400 1401 1402

	/*
	 * We have to check all instances of inlined function, because
	 * some execution paths can be optimized out depends on the
	 * function argument of instances
	 */
	return DWARF_CB_OK;
1403 1404
}

1405
/* Search function from function name */
1406
static int line_range_search_cb(Dwarf_Die *sp_die, void *data)
1407
{
1408 1409
	struct dwarf_callback_param *param = data;
	struct line_finder *lf = param->data;
1410 1411
	struct line_range *lr = lf->lr;

1412 1413 1414 1415
	/* Check declared file */
	if (lr->file && strtailcmp(lr->file, dwarf_decl_file(sp_die)))
		return DWARF_CB_OK;

1416
	if (dwarf_tag(sp_die) == DW_TAG_subprogram &&
1417
	    die_compare_name(sp_die, lr->function)) {
1418 1419
		lf->fname = dwarf_decl_file(sp_die);
		dwarf_decl_line(sp_die, &lr->offset);
1420
		pr_debug("fname: %s, lineno:%d\n", lf->fname, lr->offset);
1421
		lf->lno_s = lr->offset + lr->start;
1422 1423 1424 1425
		if (lf->lno_s < 0)	/* Overflow */
			lf->lno_s = INT_MAX;
		lf->lno_e = lr->offset + lr->end;
		if (lf->lno_e < 0)	/* Overflow */
1426
			lf->lno_e = INT_MAX;
1427
		pr_debug("New line range: %d to %d\n", lf->lno_s, lf->lno_e);
1428 1429
		lr->start = lf->lno_s;
		lr->end = lf->lno_e;
1430 1431 1432 1433
		if (dwarf_func_inline(sp_die)) {
			struct dwarf_callback_param _param;
			_param.data = (void *)lf;
			_param.retval = 0;
1434
			dwarf_func_inline_instances(sp_die,
1435 1436 1437 1438 1439 1440
						    line_range_inline_cb,
						    &_param);
			param->retval = _param.retval;
		} else
			param->retval = find_line_range_by_line(sp_die, lf);
		return DWARF_CB_ABORT;
1441
	}
1442
	return DWARF_CB_OK;
1443 1444
}

1445
static int find_line_range_by_func(struct line_finder *lf)
1446
{
1447 1448 1449
	struct dwarf_callback_param param = {.data = (void *)lf, .retval = 0};
	dwarf_getfuncs(&lf->cu_die, line_range_search_cb, &param, 0);
	return param.retval;
1450 1451
}

1452
int debuginfo__find_line_range(struct debuginfo *self, struct line_range *lr)
1453
{
1454
	struct line_finder lf = {.lr = lr, .found = 0};
1455
	int ret = 0;
1456 1457 1458
	Dwarf_Off off = 0, noff;
	size_t cuhl;
	Dwarf_Die *diep;
1459
	const char *comp_dir;
1460

1461 1462 1463 1464 1465 1466 1467 1468
	/* Fastpath: lookup by function name from .debug_pubnames section */
	if (lr->function) {
		struct pubname_callback_param pubname_param = {
			.function = lr->function, .file = lr->file,
			.cu_die = &lf.cu_die, .sp_die = &lf.sp_die, .found = 0};
		struct dwarf_callback_param line_range_param = {
			.data = (void *)&lf, .retval = 0};

1469 1470
		dwarf_getpubnames(self->dbg, pubname_search_cb,
				  &pubname_param, 0);
1471 1472 1473 1474 1475 1476 1477
		if (pubname_param.found) {
			line_range_search_cb(&lf.sp_die, &line_range_param);
			if (lf.found)
				goto found;
		}
	}

1478
	/* Loop on CUs (Compilation Unit) */
1479
	while (!lf.found && ret >= 0) {
1480 1481
		if (dwarf_nextcu(self->dbg, off, &noff, &cuhl,
				 NULL, NULL, NULL) != 0)
1482 1483 1484
			break;

		/* Get the DIE(Debugging Information Entry) of this CU */
1485
		diep = dwarf_offdie(self->dbg, off + cuhl, &lf.cu_die);
1486 1487
		if (!diep)
			continue;
1488 1489 1490

		/* Check if target file is included. */
		if (lr->file)
1491
			lf.fname = cu_find_realpath(&lf.cu_die, lr->file);
1492
		else
1493
			lf.fname = 0;
1494

1495
		if (!lr->file || lf.fname) {
1496
			if (lr->function)
1497
				ret = find_line_range_by_func(&lf);
1498 1499
			else {
				lf.lno_s = lr->start;
1500
				lf.lno_e = lr->end;
1501
				ret = find_line_range_by_line(NULL, &lf);
1502 1503
			}
		}
1504
		off = noff;
1505
	}
1506

1507
found:
1508 1509 1510 1511 1512 1513 1514 1515 1516 1517
	/* Store comp_dir */
	if (lf.found) {
		comp_dir = cu_get_comp_dir(&lf.cu_die);
		if (comp_dir) {
			lr->comp_dir = strdup(comp_dir);
			if (!lr->comp_dir)
				ret = -ENOMEM;
		}
	}

1518
	pr_debug("path: %s\n", lr->path);
1519
	return (ret < 0) ? ret : lf.found;
1520 1521
}