libbpf.c 233.6 KB
Newer Older
1
// SPDX-License-Identifier: (LGPL-2.1 OR BSD-2-Clause)
2

3 4 5 6 7 8
/*
 * Common eBPF ELF object loading operations.
 *
 * Copyright (C) 2013-2015 Alexei Starovoitov <ast@kernel.org>
 * Copyright (C) 2015 Wang Nan <wangnan0@huawei.com>
 * Copyright (C) 2015 Huawei Inc.
9
 * Copyright (C) 2017 Nicira, Inc.
10
 * Copyright (C) 2019 Isovalent, Inc.
11 12
 */

13
#ifndef _GNU_SOURCE
14
#define _GNU_SOURCE
15
#endif
16
#include <stdlib.h>
17 18
#include <stdio.h>
#include <stdarg.h>
19
#include <libgen.h>
20
#include <inttypes.h>
21
#include <limits.h>
22
#include <string.h>
23
#include <unistd.h>
24
#include <endian.h>
25 26
#include <fcntl.h>
#include <errno.h>
27
#include <ctype.h>
28
#include <asm/unistd.h>
29
#include <linux/err.h>
30
#include <linux/kernel.h>
31
#include <linux/bpf.h>
32
#include <linux/btf.h>
33
#include <linux/filter.h>
34
#include <linux/list.h>
35
#include <linux/limits.h>
36
#include <linux/perf_event.h>
37
#include <linux/ring_buffer.h>
38
#include <linux/version.h>
A
Andrii Nakryiko 已提交
39
#include <sys/epoll.h>
40
#include <sys/ioctl.h>
A
Andrii Nakryiko 已提交
41
#include <sys/mman.h>
42 43 44
#include <sys/stat.h>
#include <sys/types.h>
#include <sys/vfs.h>
45
#include <sys/utsname.h>
46
#include <sys/resource.h>
47
#include <tools/libc_compat.h>
48 49
#include <libelf.h>
#include <gelf.h>
50
#include <zlib.h>
51 52

#include "libbpf.h"
53
#include "bpf.h"
54
#include "btf.h"
55
#include "str_error.h"
56
#include "libbpf_internal.h"
57
#include "hashmap.h"
58

59 60 61
/* make sure libbpf doesn't use kernel-only integer typedefs */
#pragma GCC poison u8 u16 u32 u64 s8 s16 s32 s64

62 63 64 65
#ifndef EM_BPF
#define EM_BPF 247
#endif

66 67 68 69
#ifndef BPF_FS_MAGIC
#define BPF_FS_MAGIC		0xcafe4a11
#endif

70 71 72 73 74
/* vsprintf() in __base_pr() uses nonliteral format string. It may break
 * compilation if user enables corresponding warning. Disable it explicitly.
 */
#pragma GCC diagnostic ignored "-Wformat-nonliteral"

75 76
#define __printf(a, b)	__attribute__((format(printf, a, b)))

77 78 79 80 81 82
static struct bpf_map *bpf_object__add_map(struct bpf_object *obj);
static struct bpf_program *bpf_object__find_prog_by_idx(struct bpf_object *obj,
							int idx);
static const struct btf_type *
skip_mods_and_typedefs(const struct btf *btf, __u32 id, __u32 *res_id);

S
Stanislav Fomichev 已提交
83 84
static int __base_pr(enum libbpf_print_level level, const char *format,
		     va_list args)
85
{
86 87 88
	if (level == LIBBPF_DEBUG)
		return 0;

S
Stanislav Fomichev 已提交
89
	return vfprintf(stderr, format, args);
90 91
}

S
Stanislav Fomichev 已提交
92
static libbpf_print_fn_t __libbpf_pr = __base_pr;
93

94
libbpf_print_fn_t libbpf_set_print(libbpf_print_fn_t fn)
95
{
96 97
	libbpf_print_fn_t old_print_fn = __libbpf_pr;

98
	__libbpf_pr = fn;
99
	return old_print_fn;
100
}
101

102 103 104 105 106
__printf(2, 3)
void libbpf_print(enum libbpf_print_level level, const char *format, ...)
{
	va_list args;

107 108 109
	if (!__libbpf_pr)
		return;

110
	va_start(args, format);
111
	__libbpf_pr(level, format, args);
112 113 114
	va_end(args);
}

115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130
static void pr_perm_msg(int err)
{
	struct rlimit limit;
	char buf[100];

	if (err != -EPERM || geteuid() != 0)
		return;

	err = getrlimit(RLIMIT_MEMLOCK, &limit);
	if (err)
		return;

	if (limit.rlim_cur == RLIM_INFINITY)
		return;

	if (limit.rlim_cur < 1024)
131
		snprintf(buf, sizeof(buf), "%zu bytes", (size_t)limit.rlim_cur);
132 133 134 135 136 137 138 139 140
	else if (limit.rlim_cur < 1024*1024)
		snprintf(buf, sizeof(buf), "%.1f KiB", (double)limit.rlim_cur / 1024);
	else
		snprintf(buf, sizeof(buf), "%.1f MiB", (double)limit.rlim_cur / (1024*1024));

	pr_warn("permission error while running as root; try raising 'ulimit -l'? current value: %s\n",
		buf);
}

141 142
#define STRERR_BUFSIZE  128

143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162
/* Copied from tools/perf/util/util.h */
#ifndef zfree
# define zfree(ptr) ({ free(*ptr); *ptr = NULL; })
#endif

#ifndef zclose
# define zclose(fd) ({			\
	int ___err = 0;			\
	if ((fd) >= 0)			\
		___err = close((fd));	\
	fd = -1;			\
	___err; })
#endif

#ifdef HAVE_LIBELF_MMAP_SUPPORT
# define LIBBPF_ELF_C_READ_MMAP ELF_C_READ_MMAP
#else
# define LIBBPF_ELF_C_READ_MMAP ELF_C_READ
#endif

163 164 165 166 167
static inline __u64 ptr_to_u64(const void *ptr)
{
	return (__u64) (unsigned long) ptr;
}

168 169 170
struct bpf_capabilities {
	/* v4.14: kernel support for program & map names. */
	__u32 name:1;
171 172
	/* v5.2: kernel support for global data sections. */
	__u32 global_data:1;
173 174 175 176
	/* BTF_KIND_FUNC and BTF_KIND_FUNC_PROTO support */
	__u32 btf_func:1;
	/* BTF_KIND_VAR and BTF_KIND_DATASEC support */
	__u32 btf_datasec:1;
177 178
	/* BPF_F_MMAPABLE is supported for arrays */
	__u32 array_mmap:1;
179 180
	/* BTF_FUNC_GLOBAL is supported */
	__u32 btf_func_global:1;
181 182
	/* kernel support for expected_attach_type in BPF_PROG_LOAD */
	__u32 exp_attach_type:1;
183 184
};

185 186 187 188 189 190 191 192 193 194 195 196 197 198
enum reloc_type {
	RELO_LD64,
	RELO_CALL,
	RELO_DATA,
	RELO_EXTERN,
};

struct reloc_desc {
	enum reloc_type type;
	int insn_idx;
	int map_idx;
	int sym_off;
};

199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214
struct bpf_sec_def;

typedef struct bpf_link *(*attach_fn_t)(const struct bpf_sec_def *sec,
					struct bpf_program *prog);

struct bpf_sec_def {
	const char *sec;
	size_t len;
	enum bpf_prog_type prog_type;
	enum bpf_attach_type expected_attach_type;
	bool is_exp_attach_type_optional;
	bool is_attachable;
	bool is_attach_btf;
	attach_fn_t attach_fn;
};

215 216 217 218 219 220 221
/*
 * bpf_prog should be a better name but it has been used in
 * linux/filter.h.
 */
struct bpf_program {
	/* Index in elf obj file, for relocation use. */
	int idx;
222
	char *name;
223
	int prog_ifindex;
224
	char *section_name;
225
	const struct bpf_sec_def *sec_def;
S
Stanislav Fomichev 已提交
226 227 228 229
	/* section_name with / replaced by _; makes recursive pinning
	 * in bpf_object__pin_programs easier
	 */
	char *pin_name;
230
	struct bpf_insn *insns;
231
	size_t insns_cnt, main_prog_cnt;
232
	enum bpf_prog_type type;
233
	bool load;
234

235
	struct reloc_desc *reloc_desc;
236
	int nr_reloc;
237
	int log_level;
238

239 240 241 242 243
	struct {
		int nr;
		int *fds;
	} instances;
	bpf_program_prep_t preprocessor;
244 245 246 247

	struct bpf_object *obj;
	void *priv;
	bpf_program_clear_priv_t clear_priv;
248 249

	enum bpf_attach_type expected_attach_type;
250
	__u32 attach_btf_id;
251
	__u32 attach_prog_fd;
252 253
	void *func_info;
	__u32 func_info_rec_size;
254
	__u32 func_info_cnt;
255 256

	struct bpf_capabilities *caps;
257 258 259 260

	void *line_info;
	__u32 line_info_rec_size;
	__u32 line_info_cnt;
261
	__u32 prog_flags;
262 263
};

264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284
struct bpf_struct_ops {
	const char *tname;
	const struct btf_type *type;
	struct bpf_program **progs;
	__u32 *kern_func_off;
	/* e.g. struct tcp_congestion_ops in bpf_prog's btf format */
	void *data;
	/* e.g. struct bpf_struct_ops_tcp_congestion_ops in
	 *      btf_vmlinux's format.
	 * struct bpf_struct_ops_tcp_congestion_ops {
	 *	[... some other kernel fields ...]
	 *	struct tcp_congestion_ops data;
	 * }
	 * kern_vdata-size == sizeof(struct bpf_struct_ops_tcp_congestion_ops)
	 * bpf_map__init_kern_struct_ops() will populate the "kern_vdata"
	 * from "data".
	 */
	void *kern_vdata;
	__u32 type_id;
};

285 286 287
#define DATA_SEC ".data"
#define BSS_SEC ".bss"
#define RODATA_SEC ".rodata"
288
#define KCONFIG_SEC ".kconfig"
289
#define KSYMS_SEC ".ksyms"
290
#define STRUCT_OPS_SEC ".struct_ops"
291

292 293 294 295 296
enum libbpf_map_type {
	LIBBPF_MAP_UNSPEC,
	LIBBPF_MAP_DATA,
	LIBBPF_MAP_BSS,
	LIBBPF_MAP_RODATA,
297
	LIBBPF_MAP_KCONFIG,
298 299 300
};

static const char * const libbpf_type_to_btf_name[] = {
301 302 303
	[LIBBPF_MAP_DATA]	= DATA_SEC,
	[LIBBPF_MAP_BSS]	= BSS_SEC,
	[LIBBPF_MAP_RODATA]	= RODATA_SEC,
304
	[LIBBPF_MAP_KCONFIG]	= KCONFIG_SEC,
305 306
};

307
struct bpf_map {
308
	char *name;
309
	int fd;
310 311
	int sec_idx;
	size_t sec_offset;
312
	int map_ifindex;
313
	int inner_map_fd;
314
	struct bpf_map_def def;
315
	__u32 numa_node;
316
	__u32 btf_var_idx;
317 318
	__u32 btf_key_type_id;
	__u32 btf_value_type_id;
319
	__u32 btf_vmlinux_value_type_id;
320 321
	void *priv;
	bpf_map_clear_priv_t clear_priv;
322
	enum libbpf_map_type libbpf_type;
323
	void *mmaped;
324
	struct bpf_struct_ops *st_ops;
325 326 327
	struct bpf_map *inner_map;
	void **init_slots;
	int init_slots_sz;
328 329
	char *pin_path;
	bool pinned;
330
	bool reused;
331 332
};

333 334
enum extern_type {
	EXT_UNKNOWN,
335
	EXT_KCFG,
336
	EXT_KSYM,
337 338 339 340 341 342 343 344 345
};

enum kcfg_type {
	KCFG_UNKNOWN,
	KCFG_CHAR,
	KCFG_BOOL,
	KCFG_INT,
	KCFG_TRISTATE,
	KCFG_CHAR_ARR,
346 347 348
};

struct extern_desc {
349
	enum extern_type type;
350 351
	int sym_idx;
	int btf_id;
352 353
	int sec_btf_id;
	const char *name;
354
	bool is_set;
355 356 357 358 359 360 361 362 363
	bool is_weak;
	union {
		struct {
			enum kcfg_type type;
			int sz;
			int align;
			int data_off;
			bool is_signed;
		} kcfg;
364 365 366
		struct {
			unsigned long long addr;
		} ksym;
367
	};
368 369
};

370 371
static LIST_HEAD(bpf_objects_list);

372
struct bpf_object {
373
	char name[BPF_OBJ_NAME_LEN];
374
	char license[64];
375
	__u32 kern_version;
376

377 378
	struct bpf_program *programs;
	size_t nr_programs;
379 380
	struct bpf_map *maps;
	size_t nr_maps;
381
	size_t maps_cap;
382

383
	char *kconfig;
384 385
	struct extern_desc *externs;
	int nr_extern;
386
	int kconfig_map_idx;
387

388
	bool loaded;
389
	bool has_pseudo_calls;
390

391 392 393 394 395 396
	/*
	 * Information when doing elf related work. Only valid if fd
	 * is valid.
	 */
	struct {
		int fd;
397
		const void *obj_buf;
398
		size_t obj_buf_sz;
399 400
		Elf *elf;
		GElf_Ehdr ehdr;
401
		Elf_Data *symbols;
402 403 404
		Elf_Data *data;
		Elf_Data *rodata;
		Elf_Data *bss;
405
		Elf_Data *st_ops_data;
406
		size_t strtabidx;
407 408 409
		struct {
			GElf_Shdr shdr;
			Elf_Data *data;
410 411
		} *reloc_sects;
		int nr_reloc_sects;
412
		int maps_shndx;
413
		int btf_maps_shndx;
414
		__u32 btf_maps_sec_btf_id;
415
		int text_shndx;
416
		int symbols_shndx;
417 418 419
		int data_shndx;
		int rodata_shndx;
		int bss_shndx;
420
		int st_ops_shndx;
421
	} efile;
422 423 424 425 426 427
	/*
	 * All loaded bpf_object is linked in a list, which is
	 * hidden to caller. bpf_objects__<func> handlers deal with
	 * all objects.
	 */
	struct list_head list;
428

429
	struct btf *btf;
430 431 432 433
	/* Parse and load BTF vmlinux if any of the programs in the object need
	 * it at load time.
	 */
	struct btf *btf_vmlinux;
434
	struct btf_ext *btf_ext;
435

436 437 438
	void *priv;
	bpf_object_clear_priv_t clear_priv;

439 440
	struct bpf_capabilities caps;

441 442 443 444
	char path[];
};
#define obj_elf_valid(o)	((o)->efile.elf)

445
void bpf_program__unload(struct bpf_program *prog)
446
{
447 448
	int i;

449 450 451
	if (!prog)
		return;

452 453 454 455 456 457 458 459
	/*
	 * If the object is opened but the program was never loaded,
	 * it is possible that prog->instances.nr == -1.
	 */
	if (prog->instances.nr > 0) {
		for (i = 0; i < prog->instances.nr; i++)
			zclose(prog->instances.fds[i]);
	} else if (prog->instances.nr != -1) {
460 461
		pr_warn("Internal error: instances.nr is %d\n",
			prog->instances.nr);
462 463 464 465
	}

	prog->instances.nr = -1;
	zfree(&prog->instances.fds);
466 467

	zfree(&prog->func_info);
468
	zfree(&prog->line_info);
469 470
}

471 472 473 474 475
static void bpf_program__exit(struct bpf_program *prog)
{
	if (!prog)
		return;

476 477 478 479 480 481
	if (prog->clear_priv)
		prog->clear_priv(prog, prog->priv);

	prog->priv = NULL;
	prog->clear_priv = NULL;

482
	bpf_program__unload(prog);
483
	zfree(&prog->name);
484
	zfree(&prog->section_name);
S
Stanislav Fomichev 已提交
485
	zfree(&prog->pin_name);
486
	zfree(&prog->insns);
487 488 489
	zfree(&prog->reloc_desc);

	prog->nr_reloc = 0;
490 491 492 493
	prog->insns_cnt = 0;
	prog->idx = -1;
}

S
Stanislav Fomichev 已提交
494 495 496 497 498 499 500 501 502 503 504
static char *__bpf_program__pin_name(struct bpf_program *prog)
{
	char *name, *p;

	name = p = strdup(prog->section_name);
	while ((p = strchr(p, '/')))
		*p = '_';

	return name;
}

505
static int
506 507
bpf_program__init(void *data, size_t size, char *section_name, int idx,
		  struct bpf_program *prog)
508
{
509 510 511
	const size_t bpf_insn_sz = sizeof(struct bpf_insn);

	if (size == 0 || size % bpf_insn_sz) {
512 513
		pr_warn("corrupted section '%s', size: %zu\n",
			section_name, size);
514 515 516
		return -EINVAL;
	}

517
	memset(prog, 0, sizeof(*prog));
518

519
	prog->section_name = strdup(section_name);
520
	if (!prog->section_name) {
521 522
		pr_warn("failed to alloc name for prog under section(%d) %s\n",
			idx, section_name);
523 524 525
		goto errout;
	}

S
Stanislav Fomichev 已提交
526 527
	prog->pin_name = __bpf_program__pin_name(prog);
	if (!prog->pin_name) {
528 529
		pr_warn("failed to alloc pin name for prog under section(%d) %s\n",
			idx, section_name);
S
Stanislav Fomichev 已提交
530 531 532
		goto errout;
	}

533 534
	prog->insns = malloc(size);
	if (!prog->insns) {
535 536
		pr_warn("failed to alloc insns for prog under section %s\n",
			section_name);
537 538
		goto errout;
	}
539 540
	prog->insns_cnt = size / bpf_insn_sz;
	memcpy(prog->insns, data, size);
541
	prog->idx = idx;
542 543
	prog->instances.fds = NULL;
	prog->instances.nr = -1;
544
	prog->type = BPF_PROG_TYPE_UNSPEC;
545
	prog->load = true;
546 547 548 549 550 551 552 553 554

	return 0;
errout:
	bpf_program__exit(prog);
	return -ENOMEM;
}

static int
bpf_object__add_program(struct bpf_object *obj, void *data, size_t size,
555
			char *section_name, int idx)
556 557 558 559
{
	struct bpf_program prog, *progs;
	int nr_progs, err;

560
	err = bpf_program__init(data, size, section_name, idx, &prog);
561 562 563
	if (err)
		return err;

564
	prog.caps = &obj->caps;
565 566 567
	progs = obj->programs;
	nr_progs = obj->nr_programs;

568
	progs = reallocarray(progs, nr_progs + 1, sizeof(progs[0]));
569 570 571 572 573 574
	if (!progs) {
		/*
		 * In this case the original obj->programs
		 * is still valid, so don't need special treat for
		 * bpf_close_object().
		 */
575 576
		pr_warn("failed to alloc a new program under section '%s'\n",
			section_name);
577 578 579 580 581 582 583
		bpf_program__exit(&prog);
		return -ENOMEM;
	}

	pr_debug("found program %s\n", prog.section_name);
	obj->programs = progs;
	obj->nr_programs = nr_progs + 1;
584
	prog.obj = obj;
585 586 587 588
	progs[nr_progs] = prog;
	return 0;
}

589 590 591 592 593 594 595 596
static int
bpf_object__init_prog_names(struct bpf_object *obj)
{
	Elf_Data *symbols = obj->efile.symbols;
	struct bpf_program *prog;
	size_t pi, si;

	for (pi = 0; pi < obj->nr_programs; pi++) {
597
		const char *name = NULL;
598 599 600 601 602 603 604 605 606 607 608

		prog = &obj->programs[pi];

		for (si = 0; si < symbols->d_size / sizeof(GElf_Sym) && !name;
		     si++) {
			GElf_Sym sym;

			if (!gelf_getsym(symbols, si, &sym))
				continue;
			if (sym.st_shndx != prog->idx)
				continue;
609 610
			if (GELF_ST_BIND(sym.st_info) != STB_GLOBAL)
				continue;
611 612 613 614 615

			name = elf_strptr(obj->efile.elf,
					  obj->efile.strtabidx,
					  sym.st_name);
			if (!name) {
616 617
				pr_warn("failed to get sym name string for prog %s\n",
					prog->section_name);
618 619 620 621
				return -LIBBPF_ERRNO__LIBELF;
			}
		}

622 623 624
		if (!name && prog->idx == obj->efile.text_shndx)
			name = ".text";

625
		if (!name) {
626 627
			pr_warn("failed to find sym for prog %s\n",
				prog->section_name);
628 629
			return -EINVAL;
		}
630

631 632
		prog->name = strdup(name);
		if (!prog->name) {
633 634
			pr_warn("failed to allocate memory for prog sym %s\n",
				name);
635 636 637 638 639 640 641
			return -ENOMEM;
		}
	}

	return 0;
}

642 643 644 645 646 647 648 649 650 651 652
static __u32 get_kernel_version(void)
{
	__u32 major, minor, patch;
	struct utsname info;

	uname(&info);
	if (sscanf(info.release, "%u.%u.%u", &major, &minor, &patch) != 3)
		return 0;
	return KERNEL_VERSION(major, minor, patch);
}

653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682
static const struct btf_member *
find_member_by_offset(const struct btf_type *t, __u32 bit_offset)
{
	struct btf_member *m;
	int i;

	for (i = 0, m = btf_members(t); i < btf_vlen(t); i++, m++) {
		if (btf_member_bit_offset(t, i) == bit_offset)
			return m;
	}

	return NULL;
}

static const struct btf_member *
find_member_by_name(const struct btf *btf, const struct btf_type *t,
		    const char *name)
{
	struct btf_member *m;
	int i;

	for (i = 0, m = btf_members(t); i < btf_vlen(t); i++, m++) {
		if (!strcmp(btf__name_by_offset(btf, m->name_off), name))
			return m;
	}

	return NULL;
}

#define STRUCT_OPS_VALUE_PREFIX "bpf_struct_ops_"
683 684
static int find_btf_by_prefix_kind(const struct btf *btf, const char *prefix,
				   const char *name, __u32 kind);
685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709

static int
find_struct_ops_kern_types(const struct btf *btf, const char *tname,
			   const struct btf_type **type, __u32 *type_id,
			   const struct btf_type **vtype, __u32 *vtype_id,
			   const struct btf_member **data_member)
{
	const struct btf_type *kern_type, *kern_vtype;
	const struct btf_member *kern_data_member;
	__s32 kern_vtype_id, kern_type_id;
	__u32 i;

	kern_type_id = btf__find_by_name_kind(btf, tname, BTF_KIND_STRUCT);
	if (kern_type_id < 0) {
		pr_warn("struct_ops init_kern: struct %s is not found in kernel BTF\n",
			tname);
		return kern_type_id;
	}
	kern_type = btf__type_by_id(btf, kern_type_id);

	/* Find the corresponding "map_value" type that will be used
	 * in map_update(BPF_MAP_TYPE_STRUCT_OPS).  For example,
	 * find "struct bpf_struct_ops_tcp_congestion_ops" from the
	 * btf_vmlinux.
	 */
710 711
	kern_vtype_id = find_btf_by_prefix_kind(btf, STRUCT_OPS_VALUE_PREFIX,
						tname, BTF_KIND_STRUCT);
712
	if (kern_vtype_id < 0) {
713 714
		pr_warn("struct_ops init_kern: struct %s%s is not found in kernel BTF\n",
			STRUCT_OPS_VALUE_PREFIX, tname);
715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730
		return kern_vtype_id;
	}
	kern_vtype = btf__type_by_id(btf, kern_vtype_id);

	/* Find "struct tcp_congestion_ops" from
	 * struct bpf_struct_ops_tcp_congestion_ops {
	 *	[ ... ]
	 *	struct tcp_congestion_ops data;
	 * }
	 */
	kern_data_member = btf_members(kern_vtype);
	for (i = 0; i < btf_vlen(kern_vtype); i++, kern_data_member++) {
		if (kern_data_member->type == kern_type_id)
			break;
	}
	if (i == btf_vlen(kern_vtype)) {
731 732
		pr_warn("struct_ops init_kern: struct %s data is not found in struct %s%s\n",
			tname, STRUCT_OPS_VALUE_PREFIX, tname);
733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764 765 766 767 768 769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802 803 804 805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823 824 825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869 870 871 872 873 874 875 876 877 878 879 880 881 882 883 884 885 886 887 888 889 890 891 892
		return -EINVAL;
	}

	*type = kern_type;
	*type_id = kern_type_id;
	*vtype = kern_vtype;
	*vtype_id = kern_vtype_id;
	*data_member = kern_data_member;

	return 0;
}

static bool bpf_map__is_struct_ops(const struct bpf_map *map)
{
	return map->def.type == BPF_MAP_TYPE_STRUCT_OPS;
}

/* Init the map's fields that depend on kern_btf */
static int bpf_map__init_kern_struct_ops(struct bpf_map *map,
					 const struct btf *btf,
					 const struct btf *kern_btf)
{
	const struct btf_member *member, *kern_member, *kern_data_member;
	const struct btf_type *type, *kern_type, *kern_vtype;
	__u32 i, kern_type_id, kern_vtype_id, kern_data_off;
	struct bpf_struct_ops *st_ops;
	void *data, *kern_data;
	const char *tname;
	int err;

	st_ops = map->st_ops;
	type = st_ops->type;
	tname = st_ops->tname;
	err = find_struct_ops_kern_types(kern_btf, tname,
					 &kern_type, &kern_type_id,
					 &kern_vtype, &kern_vtype_id,
					 &kern_data_member);
	if (err)
		return err;

	pr_debug("struct_ops init_kern %s: type_id:%u kern_type_id:%u kern_vtype_id:%u\n",
		 map->name, st_ops->type_id, kern_type_id, kern_vtype_id);

	map->def.value_size = kern_vtype->size;
	map->btf_vmlinux_value_type_id = kern_vtype_id;

	st_ops->kern_vdata = calloc(1, kern_vtype->size);
	if (!st_ops->kern_vdata)
		return -ENOMEM;

	data = st_ops->data;
	kern_data_off = kern_data_member->offset / 8;
	kern_data = st_ops->kern_vdata + kern_data_off;

	member = btf_members(type);
	for (i = 0; i < btf_vlen(type); i++, member++) {
		const struct btf_type *mtype, *kern_mtype;
		__u32 mtype_id, kern_mtype_id;
		void *mdata, *kern_mdata;
		__s64 msize, kern_msize;
		__u32 moff, kern_moff;
		__u32 kern_member_idx;
		const char *mname;

		mname = btf__name_by_offset(btf, member->name_off);
		kern_member = find_member_by_name(kern_btf, kern_type, mname);
		if (!kern_member) {
			pr_warn("struct_ops init_kern %s: Cannot find member %s in kernel BTF\n",
				map->name, mname);
			return -ENOTSUP;
		}

		kern_member_idx = kern_member - btf_members(kern_type);
		if (btf_member_bitfield_size(type, i) ||
		    btf_member_bitfield_size(kern_type, kern_member_idx)) {
			pr_warn("struct_ops init_kern %s: bitfield %s is not supported\n",
				map->name, mname);
			return -ENOTSUP;
		}

		moff = member->offset / 8;
		kern_moff = kern_member->offset / 8;

		mdata = data + moff;
		kern_mdata = kern_data + kern_moff;

		mtype = skip_mods_and_typedefs(btf, member->type, &mtype_id);
		kern_mtype = skip_mods_and_typedefs(kern_btf, kern_member->type,
						    &kern_mtype_id);
		if (BTF_INFO_KIND(mtype->info) !=
		    BTF_INFO_KIND(kern_mtype->info)) {
			pr_warn("struct_ops init_kern %s: Unmatched member type %s %u != %u(kernel)\n",
				map->name, mname, BTF_INFO_KIND(mtype->info),
				BTF_INFO_KIND(kern_mtype->info));
			return -ENOTSUP;
		}

		if (btf_is_ptr(mtype)) {
			struct bpf_program *prog;

			mtype = skip_mods_and_typedefs(btf, mtype->type, &mtype_id);
			kern_mtype = skip_mods_and_typedefs(kern_btf,
							    kern_mtype->type,
							    &kern_mtype_id);
			if (!btf_is_func_proto(mtype) ||
			    !btf_is_func_proto(kern_mtype)) {
				pr_warn("struct_ops init_kern %s: non func ptr %s is not supported\n",
					map->name, mname);
				return -ENOTSUP;
			}

			prog = st_ops->progs[i];
			if (!prog) {
				pr_debug("struct_ops init_kern %s: func ptr %s is not set\n",
					 map->name, mname);
				continue;
			}

			prog->attach_btf_id = kern_type_id;
			prog->expected_attach_type = kern_member_idx;

			st_ops->kern_func_off[i] = kern_data_off + kern_moff;

			pr_debug("struct_ops init_kern %s: func ptr %s is set to prog %s from data(+%u) to kern_data(+%u)\n",
				 map->name, mname, prog->name, moff,
				 kern_moff);

			continue;
		}

		msize = btf__resolve_size(btf, mtype_id);
		kern_msize = btf__resolve_size(kern_btf, kern_mtype_id);
		if (msize < 0 || kern_msize < 0 || msize != kern_msize) {
			pr_warn("struct_ops init_kern %s: Error in size of member %s: %zd != %zd(kernel)\n",
				map->name, mname, (ssize_t)msize,
				(ssize_t)kern_msize);
			return -ENOTSUP;
		}

		pr_debug("struct_ops init_kern %s: copy %s %u bytes from data(+%u) to kern_data(+%u)\n",
			 map->name, mname, (unsigned int)msize,
			 moff, kern_moff);
		memcpy(kern_mdata, mdata, msize);
	}

	return 0;
}

static int bpf_object__init_kern_struct_ops_maps(struct bpf_object *obj)
{
	struct bpf_map *map;
	size_t i;
	int err;

	for (i = 0; i < obj->nr_maps; i++) {
		map = &obj->maps[i];

		if (!bpf_map__is_struct_ops(map))
			continue;

893 894 895
		err = bpf_map__init_kern_struct_ops(map, obj->btf,
						    obj->btf_vmlinux);
		if (err)
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 944 945 946 947 948 949 950 951 952 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 987 988 989 990 991 992 993 994
			return err;
	}

	return 0;
}

static int bpf_object__init_struct_ops_maps(struct bpf_object *obj)
{
	const struct btf_type *type, *datasec;
	const struct btf_var_secinfo *vsi;
	struct bpf_struct_ops *st_ops;
	const char *tname, *var_name;
	__s32 type_id, datasec_id;
	const struct btf *btf;
	struct bpf_map *map;
	__u32 i;

	if (obj->efile.st_ops_shndx == -1)
		return 0;

	btf = obj->btf;
	datasec_id = btf__find_by_name_kind(btf, STRUCT_OPS_SEC,
					    BTF_KIND_DATASEC);
	if (datasec_id < 0) {
		pr_warn("struct_ops init: DATASEC %s not found\n",
			STRUCT_OPS_SEC);
		return -EINVAL;
	}

	datasec = btf__type_by_id(btf, datasec_id);
	vsi = btf_var_secinfos(datasec);
	for (i = 0; i < btf_vlen(datasec); i++, vsi++) {
		type = btf__type_by_id(obj->btf, vsi->type);
		var_name = btf__name_by_offset(obj->btf, type->name_off);

		type_id = btf__resolve_type(obj->btf, vsi->type);
		if (type_id < 0) {
			pr_warn("struct_ops init: Cannot resolve var type_id %u in DATASEC %s\n",
				vsi->type, STRUCT_OPS_SEC);
			return -EINVAL;
		}

		type = btf__type_by_id(obj->btf, type_id);
		tname = btf__name_by_offset(obj->btf, type->name_off);
		if (!tname[0]) {
			pr_warn("struct_ops init: anonymous type is not supported\n");
			return -ENOTSUP;
		}
		if (!btf_is_struct(type)) {
			pr_warn("struct_ops init: %s is not a struct\n", tname);
			return -EINVAL;
		}

		map = bpf_object__add_map(obj);
		if (IS_ERR(map))
			return PTR_ERR(map);

		map->sec_idx = obj->efile.st_ops_shndx;
		map->sec_offset = vsi->offset;
		map->name = strdup(var_name);
		if (!map->name)
			return -ENOMEM;

		map->def.type = BPF_MAP_TYPE_STRUCT_OPS;
		map->def.key_size = sizeof(int);
		map->def.value_size = type->size;
		map->def.max_entries = 1;

		map->st_ops = calloc(1, sizeof(*map->st_ops));
		if (!map->st_ops)
			return -ENOMEM;
		st_ops = map->st_ops;
		st_ops->data = malloc(type->size);
		st_ops->progs = calloc(btf_vlen(type), sizeof(*st_ops->progs));
		st_ops->kern_func_off = malloc(btf_vlen(type) *
					       sizeof(*st_ops->kern_func_off));
		if (!st_ops->data || !st_ops->progs || !st_ops->kern_func_off)
			return -ENOMEM;

		if (vsi->offset + type->size > obj->efile.st_ops_data->d_size) {
			pr_warn("struct_ops init: var %s is beyond the end of DATASEC %s\n",
				var_name, STRUCT_OPS_SEC);
			return -EINVAL;
		}

		memcpy(st_ops->data,
		       obj->efile.st_ops_data->d_buf + vsi->offset,
		       type->size);
		st_ops->tname = tname;
		st_ops->type = type;
		st_ops->type_id = type_id;

		pr_debug("struct_ops init: struct %s(type_id=%u) %s found at offset %u\n",
			 tname, type_id, var_name, vsi->offset);
	}

	return 0;
}

995
static struct bpf_object *bpf_object__new(const char *path,
996
					  const void *obj_buf,
997 998
					  size_t obj_buf_sz,
					  const char *obj_name)
999 1000
{
	struct bpf_object *obj;
1001
	char *end;
1002 1003 1004

	obj = calloc(1, sizeof(struct bpf_object) + strlen(path) + 1);
	if (!obj) {
1005
		pr_warn("alloc memory failed for %s\n", path);
1006
		return ERR_PTR(-ENOMEM);
1007 1008 1009
	}

	strcpy(obj->path, path);
1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020
	if (obj_name) {
		strncpy(obj->name, obj_name, sizeof(obj->name) - 1);
		obj->name[sizeof(obj->name) - 1] = 0;
	} else {
		/* Using basename() GNU version which doesn't modify arg. */
		strncpy(obj->name, basename((void *)path),
			sizeof(obj->name) - 1);
		end = strchr(obj->name, '.');
		if (end)
			*end = 0;
	}
1021

1022
	obj->efile.fd = -1;
1023
	/*
1024
	 * Caller of this function should also call
1025 1026 1027 1028 1029 1030
	 * bpf_object__elf_finish() after data collection to return
	 * obj_buf to user. If not, we should duplicate the buffer to
	 * avoid user freeing them before elf finish.
	 */
	obj->efile.obj_buf = obj_buf;
	obj->efile.obj_buf_sz = obj_buf_sz;
1031
	obj->efile.maps_shndx = -1;
1032
	obj->efile.btf_maps_shndx = -1;
1033 1034 1035
	obj->efile.data_shndx = -1;
	obj->efile.rodata_shndx = -1;
	obj->efile.bss_shndx = -1;
1036
	obj->efile.st_ops_shndx = -1;
1037
	obj->kconfig_map_idx = -1;
1038

1039
	obj->kern_version = get_kernel_version();
1040
	obj->loaded = false;
1041 1042 1043

	INIT_LIST_HEAD(&obj->list);
	list_add(&obj->list, &bpf_objects_list);
1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055
	return obj;
}

static void bpf_object__elf_finish(struct bpf_object *obj)
{
	if (!obj_elf_valid(obj))
		return;

	if (obj->efile.elf) {
		elf_end(obj->efile.elf);
		obj->efile.elf = NULL;
	}
1056
	obj->efile.symbols = NULL;
1057 1058 1059
	obj->efile.data = NULL;
	obj->efile.rodata = NULL;
	obj->efile.bss = NULL;
1060
	obj->efile.st_ops_data = NULL;
1061

1062 1063
	zfree(&obj->efile.reloc_sects);
	obj->efile.nr_reloc_sects = 0;
1064
	zclose(obj->efile.fd);
1065 1066
	obj->efile.obj_buf = NULL;
	obj->efile.obj_buf_sz = 0;
1067 1068 1069 1070 1071 1072 1073 1074
}

static int bpf_object__elf_init(struct bpf_object *obj)
{
	int err = 0;
	GElf_Ehdr *ep;

	if (obj_elf_valid(obj)) {
1075
		pr_warn("elf init: internal error\n");
1076
		return -LIBBPF_ERRNO__LIBELF;
1077 1078
	}

1079 1080 1081 1082 1083
	if (obj->efile.obj_buf_sz > 0) {
		/*
		 * obj_buf should have been validated by
		 * bpf_object__open_buffer().
		 */
1084
		obj->efile.elf = elf_memory((char *)obj->efile.obj_buf,
1085 1086 1087 1088
					    obj->efile.obj_buf_sz);
	} else {
		obj->efile.fd = open(obj->path, O_RDONLY);
		if (obj->efile.fd < 0) {
1089
			char errmsg[STRERR_BUFSIZE], *cp;
1090

1091 1092
			err = -errno;
			cp = libbpf_strerror_r(err, errmsg, sizeof(errmsg));
1093
			pr_warn("failed to open %s: %s\n", obj->path, cp);
1094
			return err;
1095 1096 1097
		}

		obj->efile.elf = elf_begin(obj->efile.fd,
1098
					   LIBBPF_ELF_C_READ_MMAP, NULL);
1099 1100 1101
	}

	if (!obj->efile.elf) {
1102
		pr_warn("failed to open %s as ELF file\n", obj->path);
1103
		err = -LIBBPF_ERRNO__LIBELF;
1104 1105 1106 1107
		goto errout;
	}

	if (!gelf_getehdr(obj->efile.elf, &obj->efile.ehdr)) {
1108
		pr_warn("failed to get EHDR from %s\n", obj->path);
1109
		err = -LIBBPF_ERRNO__FORMAT;
1110 1111 1112 1113
		goto errout;
	}
	ep = &obj->efile.ehdr;

1114
	/* Old LLVM set e_machine to EM_NONE */
1115 1116
	if (ep->e_type != ET_REL ||
	    (ep->e_machine && ep->e_machine != EM_BPF)) {
1117
		pr_warn("%s is not an eBPF object file\n", obj->path);
1118
		err = -LIBBPF_ERRNO__FORMAT;
1119 1120 1121 1122 1123 1124 1125 1126 1127
		goto errout;
	}

	return 0;
errout:
	bpf_object__elf_finish(obj);
	return err;
}

1128
static int bpf_object__check_endianness(struct bpf_object *obj)
1129
{
1130
#if __BYTE_ORDER == __LITTLE_ENDIAN
1131 1132
	if (obj->efile.ehdr.e_ident[EI_DATA] == ELFDATA2LSB)
		return 0;
1133
#elif __BYTE_ORDER == __BIG_ENDIAN
1134 1135 1136 1137 1138
	if (obj->efile.ehdr.e_ident[EI_DATA] == ELFDATA2MSB)
		return 0;
#else
# error "Unrecognized __BYTE_ORDER__"
#endif
1139
	pr_warn("endianness mismatch.\n");
1140
	return -LIBBPF_ERRNO__ENDIAN;
1141 1142
}

1143
static int
1144
bpf_object__init_license(struct bpf_object *obj, void *data, size_t size)
1145
{
1146
	memcpy(obj->license, data, min(size, sizeof(obj->license) - 1));
1147 1148 1149 1150
	pr_debug("license of %s is %s\n", obj->path, obj->license);
	return 0;
}

1151 1152 1153 1154 1155 1156
static int
bpf_object__init_kversion(struct bpf_object *obj, void *data, size_t size)
{
	__u32 kver;

	if (size != sizeof(kver)) {
1157
		pr_warn("invalid kver section in %s\n", obj->path);
1158 1159 1160 1161 1162 1163 1164 1165
		return -LIBBPF_ERRNO__FORMAT;
	}
	memcpy(&kver, data, sizeof(kver));
	obj->kern_version = kver;
	pr_debug("kernel version of %s is %x\n", obj->path, obj->kern_version);
	return 0;
}

1166 1167 1168 1169 1170 1171 1172 1173
static bool bpf_map_type__is_map_in_map(enum bpf_map_type type)
{
	if (type == BPF_MAP_TYPE_ARRAY_OF_MAPS ||
	    type == BPF_MAP_TYPE_HASH_OF_MAPS)
		return true;
	return false;
}

1174 1175 1176 1177 1178 1179 1180 1181 1182 1183 1184 1185 1186 1187 1188
static int bpf_object_search_section_size(const struct bpf_object *obj,
					  const char *name, size_t *d_size)
{
	const GElf_Ehdr *ep = &obj->efile.ehdr;
	Elf *elf = obj->efile.elf;
	Elf_Scn *scn = NULL;
	int idx = 0;

	while ((scn = elf_nextscn(elf, scn)) != NULL) {
		const char *sec_name;
		Elf_Data *data;
		GElf_Shdr sh;

		idx++;
		if (gelf_getshdr(scn, &sh) != &sh) {
1189 1190
			pr_warn("failed to get section(%d) header from %s\n",
				idx, obj->path);
1191 1192 1193 1194 1195
			return -EIO;
		}

		sec_name = elf_strptr(elf, ep->e_shstrndx, sh.sh_name);
		if (!sec_name) {
1196 1197
			pr_warn("failed to get section(%d) name from %s\n",
				idx, obj->path);
1198 1199 1200 1201 1202 1203 1204 1205
			return -EIO;
		}

		if (strcmp(name, sec_name))
			continue;

		data = elf_getdata(scn, 0);
		if (!data) {
1206 1207
			pr_warn("failed to get section(%d) data from %s(%s)\n",
				idx, name, obj->path);
1208 1209 1210 1211 1212 1213 1214 1215 1216 1217 1218 1219 1220 1221 1222 1223 1224 1225 1226
			return -EIO;
		}

		*d_size = data->d_size;
		return 0;
	}

	return -ENOENT;
}

int bpf_object__section_size(const struct bpf_object *obj, const char *name,
			     __u32 *size)
{
	int ret = -ENOENT;
	size_t d_size;

	*size = 0;
	if (!name) {
		return -EINVAL;
1227
	} else if (!strcmp(name, DATA_SEC)) {
1228 1229
		if (obj->efile.data)
			*size = obj->efile.data->d_size;
1230
	} else if (!strcmp(name, BSS_SEC)) {
1231 1232
		if (obj->efile.bss)
			*size = obj->efile.bss->d_size;
1233
	} else if (!strcmp(name, RODATA_SEC)) {
1234 1235
		if (obj->efile.rodata)
			*size = obj->efile.rodata->d_size;
1236 1237 1238
	} else if (!strcmp(name, STRUCT_OPS_SEC)) {
		if (obj->efile.st_ops_data)
			*size = obj->efile.st_ops_data->d_size;
1239 1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260 1261 1262 1263 1264 1265 1266 1267 1268 1269
	} else {
		ret = bpf_object_search_section_size(obj, name, &d_size);
		if (!ret)
			*size = d_size;
	}

	return *size ? 0 : ret;
}

int bpf_object__variable_offset(const struct bpf_object *obj, const char *name,
				__u32 *off)
{
	Elf_Data *symbols = obj->efile.symbols;
	const char *sname;
	size_t si;

	if (!name || !off)
		return -EINVAL;

	for (si = 0; si < symbols->d_size / sizeof(GElf_Sym); si++) {
		GElf_Sym sym;

		if (!gelf_getsym(symbols, si, &sym))
			continue;
		if (GELF_ST_BIND(sym.st_info) != STB_GLOBAL ||
		    GELF_ST_TYPE(sym.st_info) != STT_OBJECT)
			continue;

		sname = elf_strptr(obj->efile.elf, obj->efile.strtabidx,
				   sym.st_name);
		if (!sname) {
1270 1271
			pr_warn("failed to get sym name string for var %s\n",
				name);
1272 1273 1274 1275 1276 1277 1278 1279 1280 1281 1282
			return -EIO;
		}
		if (strcmp(name, sname) == 0) {
			*off = sym.st_value;
			return 0;
		}
	}

	return -ENOENT;
}

1283
static struct bpf_map *bpf_object__add_map(struct bpf_object *obj)
1284
{
1285 1286 1287 1288 1289 1290 1291
	struct bpf_map *new_maps;
	size_t new_cap;
	int i;

	if (obj->nr_maps < obj->maps_cap)
		return &obj->maps[obj->nr_maps++];

1292
	new_cap = max((size_t)4, obj->maps_cap * 3 / 2);
1293 1294
	new_maps = realloc(obj->maps, new_cap * sizeof(*obj->maps));
	if (!new_maps) {
1295
		pr_warn("alloc maps for object failed\n");
1296 1297 1298 1299 1300 1301 1302 1303 1304 1305 1306 1307 1308 1309 1310 1311 1312 1313 1314
		return ERR_PTR(-ENOMEM);
	}

	obj->maps_cap = new_cap;
	obj->maps = new_maps;

	/* zero out new maps */
	memset(obj->maps + obj->nr_maps, 0,
	       (obj->maps_cap - obj->nr_maps) * sizeof(*obj->maps));
	/*
	 * fill all fd with -1 so won't close incorrect fd (fd=0 is stdin)
	 * when failure (zclose won't close negative fd)).
	 */
	for (i = obj->nr_maps; i < obj->maps_cap; i++) {
		obj->maps[i].fd = -1;
		obj->maps[i].inner_map_fd = -1;
	}

	return &obj->maps[obj->nr_maps++];
1315 1316
}

1317 1318 1319 1320 1321
static size_t bpf_map_mmap_sz(const struct bpf_map *map)
{
	long page_sz = sysconf(_SC_PAGE_SIZE);
	size_t map_sz;

1322
	map_sz = (size_t)roundup(map->def.value_size, 8) * map->def.max_entries;
1323 1324 1325 1326
	map_sz = roundup(map_sz, page_sz);
	return map_sz;
}

1327 1328 1329
static char *internal_map_name(struct bpf_object *obj,
			       enum libbpf_map_type type)
{
1330
	char map_name[BPF_OBJ_NAME_LEN], *p;
1331 1332 1333 1334 1335 1336 1337 1338
	const char *sfx = libbpf_type_to_btf_name[type];
	int sfx_len = max((size_t)7, strlen(sfx));
	int pfx_len = min((size_t)BPF_OBJ_NAME_LEN - sfx_len - 1,
			  strlen(obj->name));

	snprintf(map_name, sizeof(map_name), "%.*s%.*s", pfx_len, obj->name,
		 sfx_len, libbpf_type_to_btf_name[type]);

1339 1340 1341 1342 1343
	/* sanitise map name to characters allowed by kernel */
	for (p = map_name; *p && p < map_name + sizeof(map_name); p++)
		if (!isalnum(*p) && *p != '_' && *p != '.')
			*p = '_';

1344 1345 1346
	return strdup(map_name);
}

1347
static int
1348
bpf_object__init_internal_map(struct bpf_object *obj, enum libbpf_map_type type,
1349
			      int sec_idx, void *data, size_t data_sz)
1350
{
1351 1352
	struct bpf_map_def *def;
	struct bpf_map *map;
1353
	int err;
1354 1355 1356 1357

	map = bpf_object__add_map(obj);
	if (IS_ERR(map))
		return PTR_ERR(map);
1358 1359

	map->libbpf_type = type;
1360 1361
	map->sec_idx = sec_idx;
	map->sec_offset = 0;
1362
	map->name = internal_map_name(obj, type);
1363
	if (!map->name) {
1364
		pr_warn("failed to alloc map name\n");
1365 1366 1367
		return -ENOMEM;
	}

1368
	def = &map->def;
1369 1370
	def->type = BPF_MAP_TYPE_ARRAY;
	def->key_size = sizeof(int);
1371
	def->value_size = data_sz;
1372
	def->max_entries = 1;
1373
	def->map_flags = type == LIBBPF_MAP_RODATA || type == LIBBPF_MAP_KCONFIG
1374
			 ? BPF_F_RDONLY_PROG : 0;
1375
	def->map_flags |= BPF_F_MMAPABLE;
1376 1377

	pr_debug("map '%s' (global data): at sec_idx %d, offset %zu, flags %x.\n",
1378
		 map->name, map->sec_idx, map->sec_offset, def->map_flags);
1379

1380 1381 1382 1383 1384 1385 1386 1387 1388
	map->mmaped = mmap(NULL, bpf_map_mmap_sz(map), PROT_READ | PROT_WRITE,
			   MAP_SHARED | MAP_ANONYMOUS, -1, 0);
	if (map->mmaped == MAP_FAILED) {
		err = -errno;
		map->mmaped = NULL;
		pr_warn("failed to alloc map '%s' content buffer: %d\n",
			map->name, err);
		zfree(&map->name);
		return err;
1389 1390
	}

1391
	if (data)
1392 1393
		memcpy(map->mmaped, data, data_sz);

1394
	pr_debug("map %td is \"%s\"\n", map - obj->maps, map->name);
1395 1396 1397
	return 0;
}

1398 1399 1400 1401 1402 1403 1404 1405 1406
static int bpf_object__init_global_data_maps(struct bpf_object *obj)
{
	int err;

	/*
	 * Populate obj->maps with libbpf internal maps.
	 */
	if (obj->efile.data_shndx >= 0) {
		err = bpf_object__init_internal_map(obj, LIBBPF_MAP_DATA,
1407
						    obj->efile.data_shndx,
1408 1409
						    obj->efile.data->d_buf,
						    obj->efile.data->d_size);
1410 1411 1412 1413 1414
		if (err)
			return err;
	}
	if (obj->efile.rodata_shndx >= 0) {
		err = bpf_object__init_internal_map(obj, LIBBPF_MAP_RODATA,
1415
						    obj->efile.rodata_shndx,
1416 1417
						    obj->efile.rodata->d_buf,
						    obj->efile.rodata->d_size);
1418 1419 1420 1421 1422
		if (err)
			return err;
	}
	if (obj->efile.bss_shndx >= 0) {
		err = bpf_object__init_internal_map(obj, LIBBPF_MAP_BSS,
1423
						    obj->efile.bss_shndx,
1424 1425
						    NULL,
						    obj->efile.bss->d_size);
1426 1427 1428 1429 1430 1431
		if (err)
			return err;
	}
	return 0;
}

1432 1433 1434 1435 1436 1437 1438 1439 1440 1441 1442 1443 1444

static struct extern_desc *find_extern_by_name(const struct bpf_object *obj,
					       const void *name)
{
	int i;

	for (i = 0; i < obj->nr_extern; i++) {
		if (strcmp(obj->externs[i].name, name) == 0)
			return &obj->externs[i];
	}
	return NULL;
}

1445 1446
static int set_kcfg_value_tri(struct extern_desc *ext, void *ext_val,
			      char value)
1447
{
1448 1449
	switch (ext->kcfg.type) {
	case KCFG_BOOL:
1450
		if (value == 'm') {
1451
			pr_warn("extern (kcfg) %s=%c should be tristate or char\n",
1452 1453 1454 1455 1456
				ext->name, value);
			return -EINVAL;
		}
		*(bool *)ext_val = value == 'y' ? true : false;
		break;
1457
	case KCFG_TRISTATE:
1458 1459 1460 1461 1462 1463 1464
		if (value == 'y')
			*(enum libbpf_tristate *)ext_val = TRI_YES;
		else if (value == 'm')
			*(enum libbpf_tristate *)ext_val = TRI_MODULE;
		else /* value == 'n' */
			*(enum libbpf_tristate *)ext_val = TRI_NO;
		break;
1465
	case KCFG_CHAR:
1466 1467
		*(char *)ext_val = value;
		break;
1468 1469 1470
	case KCFG_UNKNOWN:
	case KCFG_INT:
	case KCFG_CHAR_ARR:
1471
	default:
1472
		pr_warn("extern (kcfg) %s=%c should be bool, tristate, or char\n",
1473 1474 1475 1476 1477 1478 1479
			ext->name, value);
		return -EINVAL;
	}
	ext->is_set = true;
	return 0;
}

1480 1481
static int set_kcfg_value_str(struct extern_desc *ext, char *ext_val,
			      const char *value)
1482 1483 1484
{
	size_t len;

1485 1486
	if (ext->kcfg.type != KCFG_CHAR_ARR) {
		pr_warn("extern (kcfg) %s=%s should be char array\n", ext->name, value);
1487 1488 1489 1490 1491
		return -EINVAL;
	}

	len = strlen(value);
	if (value[len - 1] != '"') {
1492
		pr_warn("extern (kcfg) '%s': invalid string config '%s'\n",
1493 1494 1495 1496 1497 1498
			ext->name, value);
		return -EINVAL;
	}

	/* strip quotes */
	len -= 2;
1499 1500 1501 1502
	if (len >= ext->kcfg.sz) {
		pr_warn("extern (kcfg) '%s': long string config %s of (%zu bytes) truncated to %d bytes\n",
			ext->name, value, len, ext->kcfg.sz - 1);
		len = ext->kcfg.sz - 1;
1503 1504 1505 1506 1507 1508 1509 1510 1511 1512 1513 1514 1515 1516 1517 1518 1519 1520 1521 1522 1523 1524 1525 1526 1527 1528
	}
	memcpy(ext_val, value + 1, len);
	ext_val[len] = '\0';
	ext->is_set = true;
	return 0;
}

static int parse_u64(const char *value, __u64 *res)
{
	char *value_end;
	int err;

	errno = 0;
	*res = strtoull(value, &value_end, 0);
	if (errno) {
		err = -errno;
		pr_warn("failed to parse '%s' as integer: %d\n", value, err);
		return err;
	}
	if (*value_end) {
		pr_warn("failed to parse '%s' as integer completely\n", value);
		return -EINVAL;
	}
	return 0;
}

1529
static bool is_kcfg_value_in_range(const struct extern_desc *ext, __u64 v)
1530
{
1531
	int bit_sz = ext->kcfg.sz * 8;
1532

1533
	if (ext->kcfg.sz == 8)
1534 1535 1536 1537 1538 1539 1540 1541 1542 1543 1544 1545 1546 1547
		return true;

	/* Validate that value stored in u64 fits in integer of `ext->sz`
	 * bytes size without any loss of information. If the target integer
	 * is signed, we rely on the following limits of integer type of
	 * Y bits and subsequent transformation:
	 *
	 *     -2^(Y-1) <= X           <= 2^(Y-1) - 1
	 *            0 <= X + 2^(Y-1) <= 2^Y - 1
	 *            0 <= X + 2^(Y-1) <  2^Y
	 *
	 *  For unsigned target integer, check that all the (64 - Y) bits are
	 *  zero.
	 */
1548
	if (ext->kcfg.is_signed)
1549 1550 1551 1552 1553
		return v + (1ULL << (bit_sz - 1)) < (1ULL << bit_sz);
	else
		return (v >> bit_sz) == 0;
}

1554 1555
static int set_kcfg_value_num(struct extern_desc *ext, void *ext_val,
			      __u64 value)
1556
{
1557 1558
	if (ext->kcfg.type != KCFG_INT && ext->kcfg.type != KCFG_CHAR) {
		pr_warn("extern (kcfg) %s=%llu should be integer\n",
1559
			ext->name, (unsigned long long)value);
1560 1561
		return -EINVAL;
	}
1562 1563 1564
	if (!is_kcfg_value_in_range(ext, value)) {
		pr_warn("extern (kcfg) %s=%llu value doesn't fit in %d bytes\n",
			ext->name, (unsigned long long)value, ext->kcfg.sz);
1565 1566
		return -ERANGE;
	}
1567
	switch (ext->kcfg.sz) {
1568 1569 1570 1571 1572 1573 1574 1575 1576 1577 1578
		case 1: *(__u8 *)ext_val = value; break;
		case 2: *(__u16 *)ext_val = value; break;
		case 4: *(__u32 *)ext_val = value; break;
		case 8: *(__u64 *)ext_val = value; break;
		default:
			return -EINVAL;
	}
	ext->is_set = true;
	return 0;
}

1579 1580
static int bpf_object__process_kconfig_line(struct bpf_object *obj,
					    char *buf, void *data)
1581 1582
{
	struct extern_desc *ext;
1583
	char *sep, *value;
1584 1585 1586 1587
	int len, err = 0;
	void *ext_val;
	__u64 num;

1588 1589
	if (strncmp(buf, "CONFIG_", 7))
		return 0;
1590

1591 1592 1593 1594 1595 1596 1597 1598 1599 1600 1601 1602 1603 1604 1605 1606 1607 1608 1609 1610 1611 1612
	sep = strchr(buf, '=');
	if (!sep) {
		pr_warn("failed to parse '%s': no separator\n", buf);
		return -EINVAL;
	}

	/* Trim ending '\n' */
	len = strlen(buf);
	if (buf[len - 1] == '\n')
		buf[len - 1] = '\0';
	/* Split on '=' and ensure that a value is present. */
	*sep = '\0';
	if (!sep[1]) {
		*sep = '=';
		pr_warn("failed to parse '%s': no value\n", buf);
		return -EINVAL;
	}

	ext = find_extern_by_name(obj, buf);
	if (!ext || ext->is_set)
		return 0;

1613
	ext_val = data + ext->kcfg.data_off;
1614 1615 1616 1617
	value = sep + 1;

	switch (*value) {
	case 'y': case 'n': case 'm':
1618
		err = set_kcfg_value_tri(ext, ext_val, *value);
1619 1620
		break;
	case '"':
1621
		err = set_kcfg_value_str(ext, ext_val, value);
1622 1623 1624 1625 1626
		break;
	default:
		/* assume integer */
		err = parse_u64(value, &num);
		if (err) {
1627
			pr_warn("extern (kcfg) %s=%s should be integer\n",
1628 1629 1630
				ext->name, value);
			return err;
		}
1631
		err = set_kcfg_value_num(ext, ext_val, num);
1632
		break;
1633
	}
1634 1635
	if (err)
		return err;
1636
	pr_debug("extern (kcfg) %s=%s\n", ext->name, value);
1637 1638 1639 1640 1641 1642 1643 1644 1645 1646 1647 1648 1649 1650 1651 1652 1653 1654 1655 1656 1657 1658
	return 0;
}

static int bpf_object__read_kconfig_file(struct bpf_object *obj, void *data)
{
	char buf[PATH_MAX];
	struct utsname uts;
	int len, err = 0;
	gzFile file;

	uname(&uts);
	len = snprintf(buf, PATH_MAX, "/boot/config-%s", uts.release);
	if (len < 0)
		return -EINVAL;
	else if (len >= PATH_MAX)
		return -ENAMETOOLONG;

	/* gzopen also accepts uncompressed files. */
	file = gzopen(buf, "r");
	if (!file)
		file = gzopen("/proc/config.gz", "r");

1659
	if (!file) {
1660
		pr_warn("failed to open system Kconfig\n");
1661 1662 1663 1664
		return -ENOENT;
	}

	while (gzgets(file, buf, sizeof(buf))) {
1665 1666 1667 1668
		err = bpf_object__process_kconfig_line(obj, buf, data);
		if (err) {
			pr_warn("error parsing system Kconfig line '%s': %d\n",
				buf, err);
1669 1670
			goto out;
		}
1671
	}
1672

1673 1674 1675 1676
out:
	gzclose(file);
	return err;
}
1677

1678 1679 1680 1681 1682 1683
static int bpf_object__read_kconfig_mem(struct bpf_object *obj,
					const char *config, void *data)
{
	char buf[PATH_MAX];
	int err = 0;
	FILE *file;
1684

1685 1686 1687 1688 1689 1690 1691 1692 1693 1694 1695 1696
	file = fmemopen((void *)config, strlen(config), "r");
	if (!file) {
		err = -errno;
		pr_warn("failed to open in-memory Kconfig: %d\n", err);
		return err;
	}

	while (fgets(buf, sizeof(buf), file)) {
		err = bpf_object__process_kconfig_line(obj, buf, data);
		if (err) {
			pr_warn("error parsing in-memory Kconfig line '%s': %d\n",
				buf, err);
1697 1698 1699 1700
			break;
		}
	}

1701
	fclose(file);
1702 1703 1704
	return err;
}

1705
static int bpf_object__init_kconfig_map(struct bpf_object *obj)
1706
{
1707
	struct extern_desc *last_ext = NULL, *ext;
1708
	size_t map_sz;
1709
	int i, err;
1710

1711 1712 1713 1714 1715
	for (i = 0; i < obj->nr_extern; i++) {
		ext = &obj->externs[i];
		if (ext->type == EXT_KCFG)
			last_ext = ext;
	}
1716

1717 1718
	if (!last_ext)
		return 0;
1719

1720
	map_sz = last_ext->kcfg.data_off + last_ext->kcfg.sz;
1721
	err = bpf_object__init_internal_map(obj, LIBBPF_MAP_KCONFIG,
1722 1723 1724 1725 1726
					    obj->efile.symbols_shndx,
					    NULL, map_sz);
	if (err)
		return err;

1727
	obj->kconfig_map_idx = obj->nr_maps - 1;
1728 1729 1730 1731

	return 0;
}

1732
static int bpf_object__init_user_maps(struct bpf_object *obj, bool strict)
1733 1734
{
	Elf_Data *symbols = obj->efile.symbols;
1735
	int i, map_def_sz = 0, nr_maps = 0, nr_syms;
1736
	Elf_Data *data = NULL;
1737 1738 1739 1740
	Elf_Scn *scn;

	if (obj->efile.maps_shndx < 0)
		return 0;
1741

E
Eric Leblond 已提交
1742 1743 1744
	if (!symbols)
		return -EINVAL;

1745 1746 1747 1748
	scn = elf_getscn(obj->efile.elf, obj->efile.maps_shndx);
	if (scn)
		data = elf_getdata(scn, NULL);
	if (!scn || !data) {
1749 1750
		pr_warn("failed to get Elf_Data from map section %d\n",
			obj->efile.maps_shndx);
1751
		return -EINVAL;
E
Eric Leblond 已提交
1752
	}
1753

E
Eric Leblond 已提交
1754 1755 1756 1757 1758 1759 1760
	/*
	 * Count number of maps. Each map has a name.
	 * Array of maps is not supported: only the first element is
	 * considered.
	 *
	 * TODO: Detect array of map and report error.
	 */
1761 1762
	nr_syms = symbols->d_size / sizeof(GElf_Sym);
	for (i = 0; i < nr_syms; i++) {
1763
		GElf_Sym sym;
E
Eric Leblond 已提交
1764 1765 1766 1767 1768 1769 1770

		if (!gelf_getsym(symbols, i, &sym))
			continue;
		if (sym.st_shndx != obj->efile.maps_shndx)
			continue;
		nr_maps++;
	}
1771
	/* Assume equally sized map definitions */
1772 1773 1774
	pr_debug("maps in %s: %d maps in %zd bytes\n",
		 obj->path, nr_maps, data->d_size);

1775
	if (!data->d_size || nr_maps == 0 || (data->d_size % nr_maps) != 0) {
1776
		pr_warn("unable to determine map definition size section %s, %d maps in %zd bytes\n",
1777
			obj->path, nr_maps, data->d_size);
1778
		return -EINVAL;
1779
	}
1780
	map_def_sz = data->d_size / nr_maps;
E
Eric Leblond 已提交
1781

1782 1783
	/* Fill obj->maps using data in "maps" section.  */
	for (i = 0; i < nr_syms; i++) {
E
Eric Leblond 已提交
1784
		GElf_Sym sym;
1785
		const char *map_name;
E
Eric Leblond 已提交
1786
		struct bpf_map_def *def;
1787
		struct bpf_map *map;
1788 1789 1790

		if (!gelf_getsym(symbols, i, &sym))
			continue;
1791
		if (sym.st_shndx != obj->efile.maps_shndx)
1792 1793
			continue;

1794 1795 1796 1797 1798
		map = bpf_object__add_map(obj);
		if (IS_ERR(map))
			return PTR_ERR(map);

		map_name = elf_strptr(obj->efile.elf, obj->efile.strtabidx,
1799
				      sym.st_name);
1800
		if (!map_name) {
1801 1802
			pr_warn("failed to get map #%d name sym string for obj %s\n",
				i, obj->path);
1803 1804
			return -LIBBPF_ERRNO__FORMAT;
		}
1805

1806
		map->libbpf_type = LIBBPF_MAP_UNSPEC;
1807 1808 1809 1810
		map->sec_idx = sym.st_shndx;
		map->sec_offset = sym.st_value;
		pr_debug("map '%s' (legacy): at sec_idx %d, offset %zu.\n",
			 map_name, map->sec_idx, map->sec_offset);
1811
		if (sym.st_value + map_def_sz > data->d_size) {
1812 1813
			pr_warn("corrupted maps section in %s: last map \"%s\" too small\n",
				obj->path, map_name);
E
Eric Leblond 已提交
1814
			return -EINVAL;
1815
		}
E
Eric Leblond 已提交
1816

1817 1818
		map->name = strdup(map_name);
		if (!map->name) {
1819
			pr_warn("failed to alloc map name\n");
1820 1821
			return -ENOMEM;
		}
1822
		pr_debug("map %d is \"%s\"\n", i, map->name);
E
Eric Leblond 已提交
1823
		def = (struct bpf_map_def *)(data->d_buf + sym.st_value);
1824 1825 1826 1827 1828 1829 1830
		/*
		 * If the definition of the map in the object file fits in
		 * bpf_map_def, copy it.  Any extra fields in our version
		 * of bpf_map_def will default to zero as a result of the
		 * calloc above.
		 */
		if (map_def_sz <= sizeof(struct bpf_map_def)) {
1831
			memcpy(&map->def, def, map_def_sz);
1832 1833 1834 1835 1836 1837 1838 1839
		} else {
			/*
			 * Here the map structure being read is bigger than what
			 * we expect, truncate if the excess bits are all zero.
			 * If they are not zero, reject this map as
			 * incompatible.
			 */
			char *b;
1840

1841 1842 1843
			for (b = ((char *)def) + sizeof(struct bpf_map_def);
			     b < ((char *)def) + map_def_sz; b++) {
				if (*b != 0) {
1844
					pr_warn("maps section in %s: \"%s\" has unrecognized, non-zero options\n",
1845
						obj->path, map_name);
1846 1847
					if (strict)
						return -EINVAL;
1848 1849
				}
			}
1850
			memcpy(&map->def, def, sizeof(struct bpf_map_def));
1851
		}
1852
	}
1853 1854
	return 0;
}
E
Eric Leblond 已提交
1855

1856 1857
static const struct btf_type *
skip_mods_and_typedefs(const struct btf *btf, __u32 id, __u32 *res_id)
1858 1859
{
	const struct btf_type *t = btf__type_by_id(btf, id);
1860

1861 1862 1863 1864 1865 1866 1867
	if (res_id)
		*res_id = id;

	while (btf_is_mod(t) || btf_is_typedef(t)) {
		if (res_id)
			*res_id = t->type;
		t = btf__type_by_id(btf, t->type);
1868
	}
1869 1870

	return t;
1871 1872
}

1873 1874 1875 1876 1877 1878 1879 1880 1881 1882 1883 1884 1885 1886
static const struct btf_type *
resolve_func_ptr(const struct btf *btf, __u32 id, __u32 *res_id)
{
	const struct btf_type *t;

	t = skip_mods_and_typedefs(btf, id, NULL);
	if (!btf_is_ptr(t))
		return NULL;

	t = skip_mods_and_typedefs(btf, t->type, res_id);

	return btf_is_func_proto(t) ? t : NULL;
}

1887 1888 1889 1890 1891 1892 1893 1894
/*
 * Fetch integer attribute of BTF map definition. Such attributes are
 * represented using a pointer to an array, in which dimensionality of array
 * encodes specified integer value. E.g., int (*type)[BPF_MAP_TYPE_ARRAY];
 * encodes `type => BPF_MAP_TYPE_ARRAY` key/value pair completely using BTF
 * type definition, while using only sizeof(void *) space in ELF data section.
 */
static bool get_map_field_int(const char *map_name, const struct btf *btf,
1895 1896
			      const struct btf_member *m, __u32 *res)
{
1897
	const struct btf_type *t = skip_mods_and_typedefs(btf, m->type, NULL);
1898
	const char *name = btf__name_by_offset(btf, m->name_off);
1899 1900
	const struct btf_array *arr_info;
	const struct btf_type *arr_t;
1901

1902
	if (!btf_is_ptr(t)) {
1903 1904
		pr_warn("map '%s': attr '%s': expected PTR, got %u.\n",
			map_name, name, btf_kind(t));
1905 1906
		return false;
	}
1907 1908 1909

	arr_t = btf__type_by_id(btf, t->type);
	if (!arr_t) {
1910 1911
		pr_warn("map '%s': attr '%s': type [%u] not found.\n",
			map_name, name, t->type);
1912 1913
		return false;
	}
1914
	if (!btf_is_array(arr_t)) {
1915 1916
		pr_warn("map '%s': attr '%s': expected ARRAY, got %u.\n",
			map_name, name, btf_kind(arr_t));
1917 1918
		return false;
	}
1919
	arr_info = btf_array(arr_t);
1920
	*res = arr_info->nelems;
1921 1922 1923
	return true;
}

1924 1925 1926 1927 1928 1929 1930 1931 1932 1933 1934 1935 1936 1937 1938 1939 1940 1941 1942 1943 1944
static int build_map_pin_path(struct bpf_map *map, const char *path)
{
	char buf[PATH_MAX];
	int err, len;

	if (!path)
		path = "/sys/fs/bpf";

	len = snprintf(buf, PATH_MAX, "%s/%s", path, bpf_map__name(map));
	if (len < 0)
		return -EINVAL;
	else if (len >= PATH_MAX)
		return -ENAMETOOLONG;

	err = bpf_map__set_pin_path(map, buf);
	if (err)
		return err;

	return 0;
}

1945 1946 1947 1948

static int parse_btf_map_def(struct bpf_object *obj,
			     struct bpf_map *map,
			     const struct btf_type *def,
1949
			     bool strict, bool is_inner,
1950
			     const char *pin_root_path)
1951
{
1952
	const struct btf_type *t;
1953 1954 1955
	const struct btf_member *m;
	int vlen, i;

1956 1957
	vlen = btf_vlen(def);
	m = btf_members(def);
1958 1959 1960 1961
	for (i = 0; i < vlen; i++, m++) {
		const char *name = btf__name_by_offset(obj->btf, m->name_off);

		if (!name) {
1962
			pr_warn("map '%s': invalid field #%d.\n", map->name, i);
1963 1964 1965
			return -EINVAL;
		}
		if (strcmp(name, "type") == 0) {
1966
			if (!get_map_field_int(map->name, obj->btf, m,
1967
					       &map->def.type))
1968 1969
				return -EINVAL;
			pr_debug("map '%s': found type = %u.\n",
1970
				 map->name, map->def.type);
1971
		} else if (strcmp(name, "max_entries") == 0) {
1972
			if (!get_map_field_int(map->name, obj->btf, m,
1973
					       &map->def.max_entries))
1974 1975
				return -EINVAL;
			pr_debug("map '%s': found max_entries = %u.\n",
1976
				 map->name, map->def.max_entries);
1977
		} else if (strcmp(name, "map_flags") == 0) {
1978
			if (!get_map_field_int(map->name, obj->btf, m,
1979
					       &map->def.map_flags))
1980 1981
				return -EINVAL;
			pr_debug("map '%s': found map_flags = %u.\n",
1982
				 map->name, map->def.map_flags);
1983 1984 1985 1986
		} else if (strcmp(name, "numa_node") == 0) {
			if (!get_map_field_int(map->name, obj->btf, m, &map->numa_node))
				return -EINVAL;
			pr_debug("map '%s': found numa_node = %u.\n", map->name, map->numa_node);
1987 1988 1989
		} else if (strcmp(name, "key_size") == 0) {
			__u32 sz;

1990
			if (!get_map_field_int(map->name, obj->btf, m, &sz))
1991 1992
				return -EINVAL;
			pr_debug("map '%s': found key_size = %u.\n",
1993
				 map->name, sz);
1994
			if (map->def.key_size && map->def.key_size != sz) {
1995
				pr_warn("map '%s': conflicting key size %u != %u.\n",
1996
					map->name, map->def.key_size, sz);
1997 1998 1999 2000 2001 2002 2003 2004
				return -EINVAL;
			}
			map->def.key_size = sz;
		} else if (strcmp(name, "key") == 0) {
			__s64 sz;

			t = btf__type_by_id(obj->btf, m->type);
			if (!t) {
2005
				pr_warn("map '%s': key type [%d] not found.\n",
2006
					map->name, m->type);
2007 2008
				return -EINVAL;
			}
2009
			if (!btf_is_ptr(t)) {
2010
				pr_warn("map '%s': key spec is not PTR: %u.\n",
2011
					map->name, btf_kind(t));
2012 2013 2014 2015
				return -EINVAL;
			}
			sz = btf__resolve_size(obj->btf, t->type);
			if (sz < 0) {
2016
				pr_warn("map '%s': can't determine key size for type [%u]: %zd.\n",
2017
					map->name, t->type, (ssize_t)sz);
2018 2019
				return sz;
			}
2020
			pr_debug("map '%s': found key [%u], sz = %zd.\n",
2021
				 map->name, t->type, (ssize_t)sz);
2022
			if (map->def.key_size && map->def.key_size != sz) {
2023
				pr_warn("map '%s': conflicting key size %u != %zd.\n",
2024
					map->name, map->def.key_size, (ssize_t)sz);
2025 2026 2027 2028 2029 2030 2031
				return -EINVAL;
			}
			map->def.key_size = sz;
			map->btf_key_type_id = t->type;
		} else if (strcmp(name, "value_size") == 0) {
			__u32 sz;

2032
			if (!get_map_field_int(map->name, obj->btf, m, &sz))
2033 2034
				return -EINVAL;
			pr_debug("map '%s': found value_size = %u.\n",
2035
				 map->name, sz);
2036
			if (map->def.value_size && map->def.value_size != sz) {
2037
				pr_warn("map '%s': conflicting value size %u != %u.\n",
2038
					map->name, map->def.value_size, sz);
2039 2040 2041 2042 2043 2044 2045 2046
				return -EINVAL;
			}
			map->def.value_size = sz;
		} else if (strcmp(name, "value") == 0) {
			__s64 sz;

			t = btf__type_by_id(obj->btf, m->type);
			if (!t) {
2047
				pr_warn("map '%s': value type [%d] not found.\n",
2048
					map->name, m->type);
2049 2050
				return -EINVAL;
			}
2051
			if (!btf_is_ptr(t)) {
2052
				pr_warn("map '%s': value spec is not PTR: %u.\n",
2053
					map->name, btf_kind(t));
2054 2055 2056 2057
				return -EINVAL;
			}
			sz = btf__resolve_size(obj->btf, t->type);
			if (sz < 0) {
2058
				pr_warn("map '%s': can't determine value size for type [%u]: %zd.\n",
2059
					map->name, t->type, (ssize_t)sz);
2060 2061
				return sz;
			}
2062
			pr_debug("map '%s': found value [%u], sz = %zd.\n",
2063
				 map->name, t->type, (ssize_t)sz);
2064
			if (map->def.value_size && map->def.value_size != sz) {
2065
				pr_warn("map '%s': conflicting value size %u != %zd.\n",
2066
					map->name, map->def.value_size, (ssize_t)sz);
2067 2068 2069 2070
				return -EINVAL;
			}
			map->def.value_size = sz;
			map->btf_value_type_id = t->type;
2071 2072 2073 2074 2075 2076 2077 2078 2079 2080 2081 2082 2083 2084 2085 2086 2087 2088 2089 2090 2091 2092 2093 2094 2095 2096 2097 2098 2099 2100 2101 2102 2103 2104 2105 2106 2107 2108 2109 2110 2111 2112 2113 2114 2115 2116 2117 2118 2119 2120 2121 2122 2123 2124 2125 2126 2127 2128 2129 2130 2131 2132 2133 2134
		}
		else if (strcmp(name, "values") == 0) {
			int err;

			if (is_inner) {
				pr_warn("map '%s': multi-level inner maps not supported.\n",
					map->name);
				return -ENOTSUP;
			}
			if (i != vlen - 1) {
				pr_warn("map '%s': '%s' member should be last.\n",
					map->name, name);
				return -EINVAL;
			}
			if (!bpf_map_type__is_map_in_map(map->def.type)) {
				pr_warn("map '%s': should be map-in-map.\n",
					map->name);
				return -ENOTSUP;
			}
			if (map->def.value_size && map->def.value_size != 4) {
				pr_warn("map '%s': conflicting value size %u != 4.\n",
					map->name, map->def.value_size);
				return -EINVAL;
			}
			map->def.value_size = 4;
			t = btf__type_by_id(obj->btf, m->type);
			if (!t) {
				pr_warn("map '%s': map-in-map inner type [%d] not found.\n",
					map->name, m->type);
				return -EINVAL;
			}
			if (!btf_is_array(t) || btf_array(t)->nelems) {
				pr_warn("map '%s': map-in-map inner spec is not a zero-sized array.\n",
					map->name);
				return -EINVAL;
			}
			t = skip_mods_and_typedefs(obj->btf, btf_array(t)->type,
						   NULL);
			if (!btf_is_ptr(t)) {
				pr_warn("map '%s': map-in-map inner def is of unexpected kind %u.\n",
					map->name, btf_kind(t));
				return -EINVAL;
			}
			t = skip_mods_and_typedefs(obj->btf, t->type, NULL);
			if (!btf_is_struct(t)) {
				pr_warn("map '%s': map-in-map inner def is of unexpected kind %u.\n",
					map->name, btf_kind(t));
				return -EINVAL;
			}

			map->inner_map = calloc(1, sizeof(*map->inner_map));
			if (!map->inner_map)
				return -ENOMEM;
			map->inner_map->sec_idx = obj->efile.btf_maps_shndx;
			map->inner_map->name = malloc(strlen(map->name) +
						      sizeof(".inner") + 1);
			if (!map->inner_map->name)
				return -ENOMEM;
			sprintf(map->inner_map->name, "%s.inner", map->name);

			err = parse_btf_map_def(obj, map->inner_map, t, strict,
						true /* is_inner */, NULL);
			if (err)
				return err;
2135 2136 2137 2138
		} else if (strcmp(name, "pinning") == 0) {
			__u32 val;
			int err;

2139 2140 2141 2142 2143
			if (is_inner) {
				pr_debug("map '%s': inner def can't be pinned.\n",
					 map->name);
				return -EINVAL;
			}
2144
			if (!get_map_field_int(map->name, obj->btf, m, &val))
2145 2146
				return -EINVAL;
			pr_debug("map '%s': found pinning = %u.\n",
2147
				 map->name, val);
2148 2149 2150 2151

			if (val != LIBBPF_PIN_NONE &&
			    val != LIBBPF_PIN_BY_NAME) {
				pr_warn("map '%s': invalid pinning value %u.\n",
2152
					map->name, val);
2153 2154 2155 2156 2157 2158
				return -EINVAL;
			}
			if (val == LIBBPF_PIN_BY_NAME) {
				err = build_map_pin_path(map, pin_root_path);
				if (err) {
					pr_warn("map '%s': couldn't build pin path.\n",
2159
						map->name);
2160 2161 2162
					return err;
				}
			}
2163 2164
		} else {
			if (strict) {
2165
				pr_warn("map '%s': unknown field '%s'.\n",
2166
					map->name, name);
2167 2168 2169
				return -ENOTSUP;
			}
			pr_debug("map '%s': ignoring unknown field '%s'.\n",
2170
				 map->name, name);
2171 2172 2173 2174
		}
	}

	if (map->def.type == BPF_MAP_TYPE_UNSPEC) {
2175
		pr_warn("map '%s': map type isn't specified.\n", map->name);
2176 2177 2178 2179 2180 2181
		return -EINVAL;
	}

	return 0;
}

2182 2183 2184 2185 2186 2187 2188 2189 2190 2191 2192 2193 2194 2195 2196 2197 2198 2199 2200 2201 2202 2203 2204 2205 2206 2207 2208 2209 2210 2211 2212 2213 2214 2215 2216 2217 2218 2219 2220 2221 2222 2223 2224 2225 2226 2227 2228 2229 2230 2231 2232 2233 2234 2235 2236 2237 2238 2239 2240 2241
static int bpf_object__init_user_btf_map(struct bpf_object *obj,
					 const struct btf_type *sec,
					 int var_idx, int sec_idx,
					 const Elf_Data *data, bool strict,
					 const char *pin_root_path)
{
	const struct btf_type *var, *def;
	const struct btf_var_secinfo *vi;
	const struct btf_var *var_extra;
	const char *map_name;
	struct bpf_map *map;

	vi = btf_var_secinfos(sec) + var_idx;
	var = btf__type_by_id(obj->btf, vi->type);
	var_extra = btf_var(var);
	map_name = btf__name_by_offset(obj->btf, var->name_off);

	if (map_name == NULL || map_name[0] == '\0') {
		pr_warn("map #%d: empty name.\n", var_idx);
		return -EINVAL;
	}
	if ((__u64)vi->offset + vi->size > data->d_size) {
		pr_warn("map '%s' BTF data is corrupted.\n", map_name);
		return -EINVAL;
	}
	if (!btf_is_var(var)) {
		pr_warn("map '%s': unexpected var kind %u.\n",
			map_name, btf_kind(var));
		return -EINVAL;
	}
	if (var_extra->linkage != BTF_VAR_GLOBAL_ALLOCATED &&
	    var_extra->linkage != BTF_VAR_STATIC) {
		pr_warn("map '%s': unsupported var linkage %u.\n",
			map_name, var_extra->linkage);
		return -EOPNOTSUPP;
	}

	def = skip_mods_and_typedefs(obj->btf, var->type, NULL);
	if (!btf_is_struct(def)) {
		pr_warn("map '%s': unexpected def kind %u.\n",
			map_name, btf_kind(var));
		return -EINVAL;
	}
	if (def->size > vi->size) {
		pr_warn("map '%s': invalid def size.\n", map_name);
		return -EINVAL;
	}

	map = bpf_object__add_map(obj);
	if (IS_ERR(map))
		return PTR_ERR(map);
	map->name = strdup(map_name);
	if (!map->name) {
		pr_warn("map '%s': failed to alloc map name.\n", map_name);
		return -ENOMEM;
	}
	map->libbpf_type = LIBBPF_MAP_UNSPEC;
	map->def.type = BPF_MAP_TYPE_UNSPEC;
	map->sec_idx = sec_idx;
	map->sec_offset = vi->offset;
2242
	map->btf_var_idx = var_idx;
2243 2244 2245
	pr_debug("map '%s': at sec_idx %d, offset %zu.\n",
		 map_name, map->sec_idx, map->sec_offset);

2246
	return parse_btf_map_def(obj, map, def, strict, false, pin_root_path);
2247 2248
}

2249 2250
static int bpf_object__init_user_btf_maps(struct bpf_object *obj, bool strict,
					  const char *pin_root_path)
2251 2252 2253 2254 2255 2256 2257 2258 2259 2260 2261 2262 2263 2264 2265
{
	const struct btf_type *sec = NULL;
	int nr_types, i, vlen, err;
	const struct btf_type *t;
	const char *name;
	Elf_Data *data;
	Elf_Scn *scn;

	if (obj->efile.btf_maps_shndx < 0)
		return 0;

	scn = elf_getscn(obj->efile.elf, obj->efile.btf_maps_shndx);
	if (scn)
		data = elf_getdata(scn, NULL);
	if (!scn || !data) {
2266 2267
		pr_warn("failed to get Elf_Data from map section %d (%s)\n",
			obj->efile.maps_shndx, MAPS_ELF_SEC);
2268 2269 2270 2271 2272 2273
		return -EINVAL;
	}

	nr_types = btf__get_nr_types(obj->btf);
	for (i = 1; i <= nr_types; i++) {
		t = btf__type_by_id(obj->btf, i);
2274
		if (!btf_is_datasec(t))
2275 2276 2277 2278
			continue;
		name = btf__name_by_offset(obj->btf, t->name_off);
		if (strcmp(name, MAPS_ELF_SEC) == 0) {
			sec = t;
2279
			obj->efile.btf_maps_sec_btf_id = i;
2280 2281 2282 2283 2284
			break;
		}
	}

	if (!sec) {
2285
		pr_warn("DATASEC '%s' not found.\n", MAPS_ELF_SEC);
2286 2287 2288
		return -ENOENT;
	}

2289
	vlen = btf_vlen(sec);
2290 2291 2292
	for (i = 0; i < vlen; i++) {
		err = bpf_object__init_user_btf_map(obj, sec, i,
						    obj->efile.btf_maps_shndx,
2293 2294
						    data, strict,
						    pin_root_path);
2295 2296 2297 2298 2299 2300 2301
		if (err)
			return err;
	}

	return 0;
}

2302
static int bpf_object__init_maps(struct bpf_object *obj,
2303
				 const struct bpf_object_open_opts *opts)
2304
{
2305 2306
	const char *pin_root_path;
	bool strict;
2307
	int err;
2308

2309 2310
	strict = !OPTS_GET(opts, relaxed_maps, false);
	pin_root_path = OPTS_GET(opts, pin_root_path, NULL);
2311

2312 2313 2314
	err = bpf_object__init_user_maps(obj, strict);
	err = err ?: bpf_object__init_user_btf_maps(obj, strict, pin_root_path);
	err = err ?: bpf_object__init_global_data_maps(obj);
2315
	err = err ?: bpf_object__init_kconfig_map(obj);
2316
	err = err ?: bpf_object__init_struct_ops_maps(obj);
2317 2318 2319 2320
	if (err)
		return err;

	return 0;
2321 2322
}

2323 2324 2325 2326 2327 2328 2329 2330 2331 2332 2333 2334 2335 2336 2337 2338 2339 2340
static bool section_have_execinstr(struct bpf_object *obj, int idx)
{
	Elf_Scn *scn;
	GElf_Shdr sh;

	scn = elf_getscn(obj->efile.elf, idx);
	if (!scn)
		return false;

	if (gelf_getshdr(scn, &sh) != &sh)
		return false;

	if (sh.sh_flags & SHF_EXECINSTR)
		return true;

	return false;
}

2341 2342 2343 2344 2345 2346 2347 2348 2349 2350
static bool btf_needs_sanitization(struct bpf_object *obj)
{
	bool has_func_global = obj->caps.btf_func_global;
	bool has_datasec = obj->caps.btf_datasec;
	bool has_func = obj->caps.btf_func;

	return !has_func || !has_datasec || !has_func_global;
}

static void bpf_object__sanitize_btf(struct bpf_object *obj, struct btf *btf)
2351
{
2352
	bool has_func_global = obj->caps.btf_func_global;
2353 2354 2355 2356 2357 2358 2359 2360
	bool has_datasec = obj->caps.btf_datasec;
	bool has_func = obj->caps.btf_func;
	struct btf_type *t;
	int i, j, vlen;

	for (i = 1; i <= btf__get_nr_types(btf); i++) {
		t = (struct btf_type *)btf__type_by_id(btf, i);

2361
		if (!has_datasec && btf_is_var(t)) {
2362 2363
			/* replace VAR with INT */
			t->info = BTF_INFO_ENC(BTF_KIND_INT, 0, 0);
2364 2365 2366 2367 2368 2369
			/*
			 * using size = 1 is the safest choice, 4 will be too
			 * big and cause kernel BTF validation failure if
			 * original variable took less than 4 bytes
			 */
			t->size = 1;
2370
			*(int *)(t + 1) = BTF_INT_ENC(0, 0, 8);
2371
		} else if (!has_datasec && btf_is_datasec(t)) {
2372
			/* replace DATASEC with STRUCT */
2373 2374
			const struct btf_var_secinfo *v = btf_var_secinfos(t);
			struct btf_member *m = btf_members(t);
2375 2376 2377 2378 2379 2380 2381 2382 2383 2384
			struct btf_type *vt;
			char *name;

			name = (char *)btf__name_by_offset(btf, t->name_off);
			while (*name) {
				if (*name == '.')
					*name = '_';
				name++;
			}

2385
			vlen = btf_vlen(t);
2386 2387 2388 2389 2390 2391 2392 2393 2394
			t->info = BTF_INFO_ENC(BTF_KIND_STRUCT, 0, vlen);
			for (j = 0; j < vlen; j++, v++, m++) {
				/* order of field assignments is important */
				m->offset = v->offset * 8;
				m->type = v->type;
				/* preserve variable name as member name */
				vt = (void *)btf__type_by_id(btf, v->type);
				m->name_off = vt->name_off;
			}
2395
		} else if (!has_func && btf_is_func_proto(t)) {
2396
			/* replace FUNC_PROTO with ENUM */
2397
			vlen = btf_vlen(t);
2398 2399
			t->info = BTF_INFO_ENC(BTF_KIND_ENUM, 0, vlen);
			t->size = sizeof(__u32); /* kernel enforced */
2400
		} else if (!has_func && btf_is_func(t)) {
2401 2402
			/* replace FUNC with TYPEDEF */
			t->info = BTF_INFO_ENC(BTF_KIND_TYPEDEF, 0, 0);
2403 2404 2405
		} else if (!has_func_global && btf_is_func(t)) {
			/* replace BTF_FUNC_GLOBAL with BTF_FUNC_STATIC */
			t->info = BTF_INFO_ENC(BTF_KIND_FUNC, 0, 0);
2406 2407 2408 2409
		}
	}
}

2410
static bool libbpf_needs_btf(const struct bpf_object *obj)
2411
{
2412 2413 2414 2415 2416 2417 2418 2419
	return obj->efile.btf_maps_shndx >= 0 ||
	       obj->efile.st_ops_shndx >= 0 ||
	       obj->nr_extern > 0;
}

static bool kernel_needs_btf(const struct bpf_object *obj)
{
	return obj->efile.st_ops_shndx >= 0;
2420 2421
}

2422
static int bpf_object__init_btf(struct bpf_object *obj,
2423 2424 2425
				Elf_Data *btf_data,
				Elf_Data *btf_ext_data)
{
2426
	int err = -ENOENT;
2427 2428 2429 2430

	if (btf_data) {
		obj->btf = btf__new(btf_data->d_buf, btf_data->d_size);
		if (IS_ERR(obj->btf)) {
2431 2432
			err = PTR_ERR(obj->btf);
			obj->btf = NULL;
2433 2434
			pr_warn("Error loading ELF section %s: %d.\n",
				BTF_ELF_SEC, err);
2435 2436
			goto out;
		}
2437
		err = 0;
2438 2439 2440 2441 2442 2443 2444 2445 2446 2447
	}
	if (btf_ext_data) {
		if (!obj->btf) {
			pr_debug("Ignore ELF section %s because its depending ELF section %s is not found.\n",
				 BTF_EXT_ELF_SEC, BTF_ELF_SEC);
			goto out;
		}
		obj->btf_ext = btf_ext__new(btf_ext_data->d_buf,
					    btf_ext_data->d_size);
		if (IS_ERR(obj->btf_ext)) {
2448 2449
			pr_warn("Error loading ELF section %s: %ld. Ignored and continue.\n",
				BTF_EXT_ELF_SEC, PTR_ERR(obj->btf_ext));
2450 2451 2452 2453 2454
			obj->btf_ext = NULL;
			goto out;
		}
	}
out:
2455
	if (err && libbpf_needs_btf(obj)) {
2456
		pr_warn("BTF is required, but is missing or corrupted.\n");
2457
		return err;
2458
	}
2459 2460 2461
	return 0;
}

2462 2463 2464 2465 2466 2467 2468 2469
static int bpf_object__finalize_btf(struct bpf_object *obj)
{
	int err;

	if (!obj->btf)
		return 0;

	err = btf__finalize_data(obj, obj->btf);
2470 2471 2472
	if (err) {
		pr_warn("Error finalizing %s: %d.\n", BTF_ELF_SEC, err);
		return err;
2473
	}
2474

2475 2476 2477
	return 0;
}

2478 2479
static inline bool libbpf_prog_needs_vmlinux_btf(struct bpf_program *prog)
{
2480 2481
	if (prog->type == BPF_PROG_TYPE_STRUCT_OPS ||
	    prog->type == BPF_PROG_TYPE_LSM)
2482 2483 2484 2485 2486 2487 2488 2489 2490 2491 2492 2493 2494
		return true;

	/* BPF_PROG_TYPE_TRACING programs which do not attach to other programs
	 * also need vmlinux BTF
	 */
	if (prog->type == BPF_PROG_TYPE_TRACING && !prog->attach_prog_fd)
		return true;

	return false;
}

static int bpf_object__load_vmlinux_btf(struct bpf_object *obj)
{
2495
	bool need_vmlinux_btf = false;
2496 2497 2498
	struct bpf_program *prog;
	int err;

2499 2500 2501 2502
	/* CO-RE relocations need kernel BTF */
	if (obj->btf_ext && obj->btf_ext->field_reloc_info.len)
		need_vmlinux_btf = true;

2503
	bpf_object__for_each_program(prog, obj) {
2504 2505
		if (!prog->load)
			continue;
2506
		if (libbpf_prog_needs_vmlinux_btf(prog)) {
2507 2508
			need_vmlinux_btf = true;
			break;
2509 2510 2511
		}
	}

2512 2513 2514 2515 2516 2517 2518 2519 2520 2521
	if (!need_vmlinux_btf)
		return 0;

	obj->btf_vmlinux = libbpf_find_kernel_btf();
	if (IS_ERR(obj->btf_vmlinux)) {
		err = PTR_ERR(obj->btf_vmlinux);
		pr_warn("Error loading vmlinux BTF: %d\n", err);
		obj->btf_vmlinux = NULL;
		return err;
	}
2522 2523 2524
	return 0;
}

2525 2526
static int bpf_object__sanitize_and_load_btf(struct bpf_object *obj)
{
2527 2528
	struct btf *kern_btf = obj->btf;
	bool btf_mandatory, sanitize;
2529 2530 2531 2532 2533
	int err = 0;

	if (!obj->btf)
		return 0;

2534 2535
	sanitize = btf_needs_sanitization(obj);
	if (sanitize) {
2536
		const void *raw_data;
2537
		__u32 sz;
2538

2539
		/* clone BTF to sanitize a copy and leave the original intact */
2540 2541
		raw_data = btf__get_raw_data(obj->btf, &sz);
		kern_btf = btf__new(raw_data, sz);
2542 2543
		if (IS_ERR(kern_btf))
			return PTR_ERR(kern_btf);
2544

2545
		bpf_object__sanitize_btf(obj, kern_btf);
2546
	}
2547 2548 2549 2550 2551 2552 2553 2554 2555 2556 2557 2558 2559 2560 2561 2562 2563 2564 2565

	err = btf__load(kern_btf);
	if (sanitize) {
		if (!err) {
			/* move fd to libbpf's BTF */
			btf__set_fd(obj->btf, btf__fd(kern_btf));
			btf__set_fd(kern_btf, -1);
		}
		btf__free(kern_btf);
	}
	if (err) {
		btf_mandatory = kernel_needs_btf(obj);
		pr_warn("Error loading .BTF into kernel: %d. %s\n", err,
			btf_mandatory ? "BTF is mandatory, can't proceed."
				      : "BTF is optional, ignoring.");
		if (!btf_mandatory)
			err = 0;
	}
	return err;
2566 2567
}

2568
static int bpf_object__elf_collect(struct bpf_object *obj)
2569 2570 2571
{
	Elf *elf = obj->efile.elf;
	GElf_Ehdr *ep = &obj->efile.ehdr;
2572
	Elf_Data *btf_ext_data = NULL;
2573
	Elf_Data *btf_data = NULL;
2574
	Elf_Scn *scn = NULL;
2575
	int idx = 0, err = 0;
2576 2577 2578

	/* Elf is corrupted/truncated, avoid calling elf_strptr. */
	if (!elf_rawdata(elf_getscn(elf, ep->e_shstrndx), NULL)) {
2579
		pr_warn("failed to get e_shstrndx from %s\n", obj->path);
2580
		return -LIBBPF_ERRNO__FORMAT;
2581 2582 2583 2584 2585 2586 2587 2588 2589
	}

	while ((scn = elf_nextscn(elf, scn)) != NULL) {
		char *name;
		GElf_Shdr sh;
		Elf_Data *data;

		idx++;
		if (gelf_getshdr(scn, &sh) != &sh) {
2590 2591
			pr_warn("failed to get section(%d) header from %s\n",
				idx, obj->path);
2592
			return -LIBBPF_ERRNO__FORMAT;
2593 2594 2595 2596
		}

		name = elf_strptr(elf, ep->e_shstrndx, sh.sh_name);
		if (!name) {
2597 2598
			pr_warn("failed to get section(%d) name from %s\n",
				idx, obj->path);
2599
			return -LIBBPF_ERRNO__FORMAT;
2600 2601 2602 2603
		}

		data = elf_getdata(scn, 0);
		if (!data) {
2604 2605
			pr_warn("failed to get section(%d) data from %s(%s)\n",
				idx, name, obj->path);
2606
			return -LIBBPF_ERRNO__FORMAT;
2607
		}
2608 2609
		pr_debug("section(%d) %s, size %ld, link %d, flags %lx, type=%d\n",
			 idx, name, (unsigned long)data->d_size,
2610 2611
			 (int)sh.sh_link, (unsigned long)sh.sh_flags,
			 (int)sh.sh_type);
2612

2613
		if (strcmp(name, "license") == 0) {
2614 2615 2616
			err = bpf_object__init_license(obj,
						       data->d_buf,
						       data->d_size);
2617 2618
			if (err)
				return err;
2619
		} else if (strcmp(name, "version") == 0) {
2620 2621 2622 2623 2624
			err = bpf_object__init_kversion(obj,
							data->d_buf,
							data->d_size);
			if (err)
				return err;
2625
		} else if (strcmp(name, "maps") == 0) {
2626
			obj->efile.maps_shndx = idx;
2627 2628
		} else if (strcmp(name, MAPS_ELF_SEC) == 0) {
			obj->efile.btf_maps_shndx = idx;
2629 2630
		} else if (strcmp(name, BTF_ELF_SEC) == 0) {
			btf_data = data;
2631
		} else if (strcmp(name, BTF_EXT_ELF_SEC) == 0) {
2632
			btf_ext_data = data;
2633
		} else if (sh.sh_type == SHT_SYMTAB) {
2634
			if (obj->efile.symbols) {
2635 2636
				pr_warn("bpf: multiple SYMTAB in %s\n",
					obj->path);
2637
				return -LIBBPF_ERRNO__FORMAT;
2638
			}
2639
			obj->efile.symbols = data;
2640
			obj->efile.symbols_shndx = idx;
2641
			obj->efile.strtabidx = sh.sh_link;
2642 2643 2644 2645 2646
		} else if (sh.sh_type == SHT_PROGBITS && data->d_size > 0) {
			if (sh.sh_flags & SHF_EXECINSTR) {
				if (strcmp(name, ".text") == 0)
					obj->efile.text_shndx = idx;
				err = bpf_object__add_program(obj, data->d_buf,
2647 2648
							      data->d_size,
							      name, idx);
2649 2650
				if (err) {
					char errmsg[STRERR_BUFSIZE];
2651
					char *cp;
2652

2653 2654
					cp = libbpf_strerror_r(-err, errmsg,
							       sizeof(errmsg));
2655 2656
					pr_warn("failed to alloc program %s (%s): %s",
						name, obj->path, cp);
2657
					return err;
2658
				}
2659
			} else if (strcmp(name, DATA_SEC) == 0) {
2660 2661
				obj->efile.data = data;
				obj->efile.data_shndx = idx;
2662
			} else if (strcmp(name, RODATA_SEC) == 0) {
2663 2664
				obj->efile.rodata = data;
				obj->efile.rodata_shndx = idx;
2665 2666 2667
			} else if (strcmp(name, STRUCT_OPS_SEC) == 0) {
				obj->efile.st_ops_data = data;
				obj->efile.st_ops_shndx = idx;
2668 2669
			} else {
				pr_debug("skip section(%d) %s\n", idx, name);
2670
			}
2671
		} else if (sh.sh_type == SHT_REL) {
2672 2673
			int nr_sects = obj->efile.nr_reloc_sects;
			void *sects = obj->efile.reloc_sects;
2674 2675 2676
			int sec = sh.sh_info; /* points to other section */

			/* Only do relo for section with exec instructions */
2677
			if (!section_have_execinstr(obj, sec) &&
2678 2679
			    strcmp(name, ".rel" STRUCT_OPS_SEC) &&
			    strcmp(name, ".rel" MAPS_ELF_SEC)) {
2680 2681 2682 2683
				pr_debug("skip relo %s(%d) for section(%d)\n",
					 name, idx, sec);
				continue;
			}
2684

2685 2686 2687 2688
			sects = reallocarray(sects, nr_sects + 1,
					     sizeof(*obj->efile.reloc_sects));
			if (!sects) {
				pr_warn("reloc_sects realloc failed\n");
2689 2690
				return -ENOMEM;
			}
2691

2692 2693
			obj->efile.reloc_sects = sects;
			obj->efile.nr_reloc_sects++;
2694

2695 2696
			obj->efile.reloc_sects[nr_sects].shdr = sh;
			obj->efile.reloc_sects[nr_sects].data = data;
2697 2698
		} else if (sh.sh_type == SHT_NOBITS &&
			   strcmp(name, BSS_SEC) == 0) {
2699 2700
			obj->efile.bss = data;
			obj->efile.bss_shndx = idx;
2701 2702
		} else {
			pr_debug("skip section(%d) %s\n", idx, name);
2703
		}
2704
	}
2705

2706
	if (!obj->efile.strtabidx || obj->efile.strtabidx > idx) {
2707
		pr_warn("Corrupted ELF file: index of strtab invalid\n");
2708
		return -LIBBPF_ERRNO__FORMAT;
2709
	}
2710
	return bpf_object__init_btf(obj, btf_data, btf_ext_data);
2711 2712
}

2713 2714 2715 2716 2717 2718 2719 2720 2721 2722 2723 2724 2725 2726 2727 2728 2729 2730 2731 2732 2733 2734 2735 2736 2737 2738 2739 2740 2741 2742 2743 2744 2745 2746 2747 2748 2749 2750
static bool sym_is_extern(const GElf_Sym *sym)
{
	int bind = GELF_ST_BIND(sym->st_info);
	/* externs are symbols w/ type=NOTYPE, bind=GLOBAL|WEAK, section=UND */
	return sym->st_shndx == SHN_UNDEF &&
	       (bind == STB_GLOBAL || bind == STB_WEAK) &&
	       GELF_ST_TYPE(sym->st_info) == STT_NOTYPE;
}

static int find_extern_btf_id(const struct btf *btf, const char *ext_name)
{
	const struct btf_type *t;
	const char *var_name;
	int i, n;

	if (!btf)
		return -ESRCH;

	n = btf__get_nr_types(btf);
	for (i = 1; i <= n; i++) {
		t = btf__type_by_id(btf, i);

		if (!btf_is_var(t))
			continue;

		var_name = btf__name_by_offset(btf, t->name_off);
		if (strcmp(var_name, ext_name))
			continue;

		if (btf_var(t)->linkage != BTF_VAR_GLOBAL_EXTERN)
			return -EINVAL;

		return i;
	}

	return -ENOENT;
}

2751 2752 2753 2754 2755 2756 2757 2758 2759 2760 2761 2762 2763 2764 2765 2766 2767 2768 2769 2770 2771 2772 2773 2774 2775 2776 2777
static int find_extern_sec_btf_id(struct btf *btf, int ext_btf_id) {
	const struct btf_var_secinfo *vs;
	const struct btf_type *t;
	int i, j, n;

	if (!btf)
		return -ESRCH;

	n = btf__get_nr_types(btf);
	for (i = 1; i <= n; i++) {
		t = btf__type_by_id(btf, i);

		if (!btf_is_datasec(t))
			continue;

		vs = btf_var_secinfos(t);
		for (j = 0; j < btf_vlen(t); j++, vs++) {
			if (vs->type == ext_btf_id)
				return i;
		}
	}

	return -ENOENT;
}

static enum kcfg_type find_kcfg_type(const struct btf *btf, int id,
				     bool *is_signed)
2778 2779 2780 2781 2782 2783 2784 2785 2786 2787 2788 2789 2790 2791
{
	const struct btf_type *t;
	const char *name;

	t = skip_mods_and_typedefs(btf, id, NULL);
	name = btf__name_by_offset(btf, t->name_off);

	if (is_signed)
		*is_signed = false;
	switch (btf_kind(t)) {
	case BTF_KIND_INT: {
		int enc = btf_int_encoding(t);

		if (enc & BTF_INT_BOOL)
2792
			return t->size == 1 ? KCFG_BOOL : KCFG_UNKNOWN;
2793 2794 2795
		if (is_signed)
			*is_signed = enc & BTF_INT_SIGNED;
		if (t->size == 1)
2796
			return KCFG_CHAR;
2797
		if (t->size < 1 || t->size > 8 || (t->size & (t->size - 1)))
2798 2799
			return KCFG_UNKNOWN;
		return KCFG_INT;
2800 2801 2802
	}
	case BTF_KIND_ENUM:
		if (t->size != 4)
2803
			return KCFG_UNKNOWN;
2804
		if (strcmp(name, "libbpf_tristate"))
2805 2806
			return KCFG_UNKNOWN;
		return KCFG_TRISTATE;
2807 2808
	case BTF_KIND_ARRAY:
		if (btf_array(t)->nelems == 0)
2809 2810 2811 2812
			return KCFG_UNKNOWN;
		if (find_kcfg_type(btf, btf_array(t)->type, NULL) != KCFG_CHAR)
			return KCFG_UNKNOWN;
		return KCFG_CHAR_ARR;
2813
	default:
2814
		return KCFG_UNKNOWN;
2815 2816 2817 2818 2819 2820 2821 2822
	}
}

static int cmp_externs(const void *_a, const void *_b)
{
	const struct extern_desc *a = _a;
	const struct extern_desc *b = _b;

2823 2824 2825 2826 2827 2828 2829 2830 2831 2832 2833 2834
	if (a->type != b->type)
		return a->type < b->type ? -1 : 1;

	if (a->type == EXT_KCFG) {
		/* descending order by alignment requirements */
		if (a->kcfg.align != b->kcfg.align)
			return a->kcfg.align > b->kcfg.align ? -1 : 1;
		/* ascending order by size, within same alignment class */
		if (a->kcfg.sz != b->kcfg.sz)
			return a->kcfg.sz < b->kcfg.sz ? -1 : 1;
	}

2835 2836 2837 2838
	/* resolve ties by name */
	return strcmp(a->name, b->name);
}

2839 2840 2841 2842 2843 2844 2845 2846 2847 2848 2849 2850 2851 2852 2853 2854
static int find_int_btf_id(const struct btf *btf)
{
	const struct btf_type *t;
	int i, n;

	n = btf__get_nr_types(btf);
	for (i = 1; i <= n; i++) {
		t = btf__type_by_id(btf, i);

		if (btf_is_int(t) && btf_int_bits(t) == 32)
			return i;
	}

	return 0;
}

2855 2856
static int bpf_object__collect_externs(struct bpf_object *obj)
{
2857
	struct btf_type *sec, *kcfg_sec = NULL, *ksym_sec = NULL;
2858 2859
	const struct btf_type *t;
	struct extern_desc *ext;
2860 2861
	int i, n, off;
	const char *ext_name, *sec_name;
2862 2863 2864 2865 2866 2867 2868 2869 2870 2871 2872 2873 2874 2875 2876 2877 2878 2879 2880 2881 2882 2883 2884 2885 2886 2887 2888 2889 2890 2891 2892 2893 2894 2895 2896 2897 2898 2899 2900 2901 2902 2903 2904 2905 2906
	Elf_Scn *scn;
	GElf_Shdr sh;

	if (!obj->efile.symbols)
		return 0;

	scn = elf_getscn(obj->efile.elf, obj->efile.symbols_shndx);
	if (!scn)
		return -LIBBPF_ERRNO__FORMAT;
	if (gelf_getshdr(scn, &sh) != &sh)
		return -LIBBPF_ERRNO__FORMAT;
	n = sh.sh_size / sh.sh_entsize;

	pr_debug("looking for externs among %d symbols...\n", n);
	for (i = 0; i < n; i++) {
		GElf_Sym sym;

		if (!gelf_getsym(obj->efile.symbols, i, &sym))
			return -LIBBPF_ERRNO__FORMAT;
		if (!sym_is_extern(&sym))
			continue;
		ext_name = elf_strptr(obj->efile.elf, obj->efile.strtabidx,
				      sym.st_name);
		if (!ext_name || !ext_name[0])
			continue;

		ext = obj->externs;
		ext = reallocarray(ext, obj->nr_extern + 1, sizeof(*ext));
		if (!ext)
			return -ENOMEM;
		obj->externs = ext;
		ext = &ext[obj->nr_extern];
		memset(ext, 0, sizeof(*ext));
		obj->nr_extern++;

		ext->btf_id = find_extern_btf_id(obj->btf, ext_name);
		if (ext->btf_id <= 0) {
			pr_warn("failed to find BTF for extern '%s': %d\n",
				ext_name, ext->btf_id);
			return ext->btf_id;
		}
		t = btf__type_by_id(obj->btf, ext->btf_id);
		ext->name = btf__name_by_offset(obj->btf, t->name_off);
		ext->sym_idx = i;
		ext->is_weak = GELF_ST_BIND(sym.st_info) == STB_WEAK;
2907 2908 2909 2910 2911 2912

		ext->sec_btf_id = find_extern_sec_btf_id(obj->btf, ext->btf_id);
		if (ext->sec_btf_id <= 0) {
			pr_warn("failed to find BTF for extern '%s' [%d] section: %d\n",
				ext_name, ext->btf_id, ext->sec_btf_id);
			return ext->sec_btf_id;
2913
		}
2914 2915 2916 2917 2918 2919 2920 2921 2922 2923 2924 2925 2926 2927 2928 2929 2930 2931 2932 2933 2934 2935 2936 2937
		sec = (void *)btf__type_by_id(obj->btf, ext->sec_btf_id);
		sec_name = btf__name_by_offset(obj->btf, sec->name_off);

		if (strcmp(sec_name, KCONFIG_SEC) == 0) {
			kcfg_sec = sec;
			ext->type = EXT_KCFG;
			ext->kcfg.sz = btf__resolve_size(obj->btf, t->type);
			if (ext->kcfg.sz <= 0) {
				pr_warn("failed to resolve size of extern (kcfg) '%s': %d\n",
					ext_name, ext->kcfg.sz);
				return ext->kcfg.sz;
			}
			ext->kcfg.align = btf__align_of(obj->btf, t->type);
			if (ext->kcfg.align <= 0) {
				pr_warn("failed to determine alignment of extern (kcfg) '%s': %d\n",
					ext_name, ext->kcfg.align);
				return -EINVAL;
			}
			ext->kcfg.type = find_kcfg_type(obj->btf, t->type,
						        &ext->kcfg.is_signed);
			if (ext->kcfg.type == KCFG_UNKNOWN) {
				pr_warn("extern (kcfg) '%s' type is unsupported\n", ext_name);
				return -ENOTSUP;
			}
2938 2939 2940 2941 2942 2943 2944 2945 2946 2947 2948
		} else if (strcmp(sec_name, KSYMS_SEC) == 0) {
			const struct btf_type *vt;

			ksym_sec = sec;
			ext->type = EXT_KSYM;

			vt = skip_mods_and_typedefs(obj->btf, t->type, NULL);
			if (!btf_is_void(vt)) {
				pr_warn("extern (ksym) '%s' is not typeless (void)\n", ext_name);
				return -ENOTSUP;
			}
2949 2950
		} else {
			pr_warn("unrecognized extern section '%s'\n", sec_name);
2951 2952 2953 2954 2955 2956 2957 2958
			return -ENOTSUP;
		}
	}
	pr_debug("collected %d externs total\n", obj->nr_extern);

	if (!obj->nr_extern)
		return 0;

2959
	/* sort externs by type, for kcfg ones also by (align, size, name) */
2960 2961
	qsort(obj->externs, obj->nr_extern, sizeof(*ext), cmp_externs);

2962 2963 2964 2965 2966 2967 2968 2969 2970 2971 2972 2973 2974 2975 2976 2977 2978 2979 2980 2981 2982 2983 2984 2985 2986 2987 2988 2989 2990 2991 2992 2993 2994 2995 2996 2997 2998 2999 3000 3001
	/* for .ksyms section, we need to turn all externs into allocated
	 * variables in BTF to pass kernel verification; we do this by
	 * pretending that each extern is a 8-byte variable
	 */
	if (ksym_sec) {
		/* find existing 4-byte integer type in BTF to use for fake
		 * extern variables in DATASEC
		 */
		int int_btf_id = find_int_btf_id(obj->btf);

		for (i = 0; i < obj->nr_extern; i++) {
			ext = &obj->externs[i];
			if (ext->type != EXT_KSYM)
				continue;
			pr_debug("extern (ksym) #%d: symbol %d, name %s\n",
				 i, ext->sym_idx, ext->name);
		}

		sec = ksym_sec;
		n = btf_vlen(sec);
		for (i = 0, off = 0; i < n; i++, off += sizeof(int)) {
			struct btf_var_secinfo *vs = btf_var_secinfos(sec) + i;
			struct btf_type *vt;

			vt = (void *)btf__type_by_id(obj->btf, vs->type);
			ext_name = btf__name_by_offset(obj->btf, vt->name_off);
			ext = find_extern_by_name(obj, ext_name);
			if (!ext) {
				pr_warn("failed to find extern definition for BTF var '%s'\n",
					ext_name);
				return -ESRCH;
			}
			btf_var(vt)->linkage = BTF_VAR_GLOBAL_ALLOCATED;
			vt->type = int_btf_id;
			vs->offset = off;
			vs->size = sizeof(int);
		}
		sec->size = off;
	}

3002 3003 3004 3005 3006 3007 3008 3009
	if (kcfg_sec) {
		sec = kcfg_sec;
		/* for kcfg externs calculate their offsets within a .kconfig map */
		off = 0;
		for (i = 0; i < obj->nr_extern; i++) {
			ext = &obj->externs[i];
			if (ext->type != EXT_KCFG)
				continue;
3010

3011 3012
			ext->kcfg.data_off = roundup(off, ext->kcfg.align);
			off = ext->kcfg.data_off + ext->kcfg.sz;
3013
			pr_debug("extern (kcfg) #%d: symbol %d, off %u, name %s\n",
3014 3015 3016 3017 3018 3019 3020 3021 3022 3023 3024 3025 3026 3027 3028 3029 3030
				 i, ext->sym_idx, ext->kcfg.data_off, ext->name);
		}
		sec->size = off;
		n = btf_vlen(sec);
		for (i = 0; i < n; i++) {
			struct btf_var_secinfo *vs = btf_var_secinfos(sec) + i;

			t = btf__type_by_id(obj->btf, vs->type);
			ext_name = btf__name_by_offset(obj->btf, t->name_off);
			ext = find_extern_by_name(obj, ext_name);
			if (!ext) {
				pr_warn("failed to find extern definition for BTF var '%s'\n",
					ext_name);
				return -ESRCH;
			}
			btf_var(t)->linkage = BTF_VAR_GLOBAL_ALLOCATED;
			vs->offset = ext->kcfg.data_off;
3031 3032 3033 3034 3035
		}
	}
	return 0;
}

3036 3037 3038 3039 3040 3041 3042 3043 3044 3045 3046 3047 3048 3049
static struct bpf_program *
bpf_object__find_prog_by_idx(struct bpf_object *obj, int idx)
{
	struct bpf_program *prog;
	size_t i;

	for (i = 0; i < obj->nr_programs; i++) {
		prog = &obj->programs[i];
		if (prog->idx == idx)
			return prog;
	}
	return NULL;
}

3050
struct bpf_program *
A
Andrii Nakryiko 已提交
3051 3052
bpf_object__find_program_by_title(const struct bpf_object *obj,
				  const char *title)
3053 3054 3055 3056 3057 3058 3059 3060 3061 3062
{
	struct bpf_program *pos;

	bpf_object__for_each_program(pos, obj) {
		if (pos->section_name && !strcmp(pos->section_name, title))
			return pos;
	}
	return NULL;
}

3063 3064 3065 3066 3067 3068 3069 3070 3071 3072 3073 3074 3075
struct bpf_program *
bpf_object__find_program_by_name(const struct bpf_object *obj,
				 const char *name)
{
	struct bpf_program *prog;

	bpf_object__for_each_program(prog, obj) {
		if (!strcmp(prog->name, name))
			return prog;
	}
	return NULL;
}

3076 3077 3078 3079 3080 3081 3082 3083 3084 3085 3086
static bool bpf_object__shndx_is_data(const struct bpf_object *obj,
				      int shndx)
{
	return shndx == obj->efile.data_shndx ||
	       shndx == obj->efile.bss_shndx ||
	       shndx == obj->efile.rodata_shndx;
}

static bool bpf_object__shndx_is_maps(const struct bpf_object *obj,
				      int shndx)
{
3087 3088
	return shndx == obj->efile.maps_shndx ||
	       shndx == obj->efile.btf_maps_shndx;
3089 3090 3091 3092 3093 3094 3095 3096 3097 3098 3099
}

static enum libbpf_map_type
bpf_object__section_to_libbpf_map_type(const struct bpf_object *obj, int shndx)
{
	if (shndx == obj->efile.data_shndx)
		return LIBBPF_MAP_DATA;
	else if (shndx == obj->efile.bss_shndx)
		return LIBBPF_MAP_BSS;
	else if (shndx == obj->efile.rodata_shndx)
		return LIBBPF_MAP_RODATA;
3100
	else if (shndx == obj->efile.symbols_shndx)
3101
		return LIBBPF_MAP_KCONFIG;
3102 3103 3104 3105
	else
		return LIBBPF_MAP_UNSPEC;
}

3106 3107 3108 3109 3110 3111 3112 3113 3114 3115 3116 3117 3118 3119 3120 3121 3122 3123 3124 3125 3126 3127 3128 3129
static int bpf_program__record_reloc(struct bpf_program *prog,
				     struct reloc_desc *reloc_desc,
				     __u32 insn_idx, const char *name,
				     const GElf_Sym *sym, const GElf_Rel *rel)
{
	struct bpf_insn *insn = &prog->insns[insn_idx];
	size_t map_idx, nr_maps = prog->obj->nr_maps;
	struct bpf_object *obj = prog->obj;
	__u32 shdr_idx = sym->st_shndx;
	enum libbpf_map_type type;
	struct bpf_map *map;

	/* sub-program call relocation */
	if (insn->code == (BPF_JMP | BPF_CALL)) {
		if (insn->src_reg != BPF_PSEUDO_CALL) {
			pr_warn("incorrect bpf_call opcode\n");
			return -LIBBPF_ERRNO__RELOC;
		}
		/* text_shndx can be 0, if no default "main" program exists */
		if (!shdr_idx || shdr_idx != obj->efile.text_shndx) {
			pr_warn("bad call relo against section %u\n", shdr_idx);
			return -LIBBPF_ERRNO__RELOC;
		}
		if (sym->st_value % 8) {
3130 3131
			pr_warn("bad call relo offset: %zu\n",
				(size_t)sym->st_value);
3132 3133 3134 3135
			return -LIBBPF_ERRNO__RELOC;
		}
		reloc_desc->type = RELO_CALL;
		reloc_desc->insn_idx = insn_idx;
3136
		reloc_desc->sym_off = sym->st_value;
3137 3138 3139 3140 3141
		obj->has_pseudo_calls = true;
		return 0;
	}

	if (insn->code != (BPF_LD | BPF_IMM | BPF_DW)) {
3142
		pr_warn("invalid relo for insns[%d].code 0x%x\n",
3143 3144 3145
			insn_idx, insn->code);
		return -LIBBPF_ERRNO__RELOC;
	}
3146 3147 3148 3149 3150 3151 3152 3153 3154 3155 3156 3157 3158 3159 3160 3161

	if (sym_is_extern(sym)) {
		int sym_idx = GELF_R_SYM(rel->r_info);
		int i, n = obj->nr_extern;
		struct extern_desc *ext;

		for (i = 0; i < n; i++) {
			ext = &obj->externs[i];
			if (ext->sym_idx == sym_idx)
				break;
		}
		if (i >= n) {
			pr_warn("extern relo failed to find extern for sym %d\n",
				sym_idx);
			return -LIBBPF_ERRNO__RELOC;
		}
3162 3163
		pr_debug("found extern #%d '%s' (sym %d) for insn %u\n",
			 i, ext->name, ext->sym_idx, insn_idx);
3164 3165
		reloc_desc->type = RELO_EXTERN;
		reloc_desc->insn_idx = insn_idx;
3166
		reloc_desc->sym_off = i; /* sym_off stores extern index */
3167 3168 3169
		return 0;
	}

3170
	if (!shdr_idx || shdr_idx >= SHN_LORESERVE) {
3171 3172
		pr_warn("invalid relo for \'%s\' in special section 0x%x; forgot to initialize global var?..\n",
			name, shdr_idx);
3173 3174 3175 3176 3177 3178 3179 3180 3181 3182 3183 3184 3185 3186 3187 3188 3189 3190 3191 3192 3193 3194 3195 3196
		return -LIBBPF_ERRNO__RELOC;
	}

	type = bpf_object__section_to_libbpf_map_type(obj, shdr_idx);

	/* generic map reference relocation */
	if (type == LIBBPF_MAP_UNSPEC) {
		if (!bpf_object__shndx_is_maps(obj, shdr_idx)) {
			pr_warn("bad map relo against section %u\n",
				shdr_idx);
			return -LIBBPF_ERRNO__RELOC;
		}
		for (map_idx = 0; map_idx < nr_maps; map_idx++) {
			map = &obj->maps[map_idx];
			if (map->libbpf_type != type ||
			    map->sec_idx != sym->st_shndx ||
			    map->sec_offset != sym->st_value)
				continue;
			pr_debug("found map %zd (%s, sec %d, off %zu) for insn %u\n",
				 map_idx, map->name, map->sec_idx,
				 map->sec_offset, insn_idx);
			break;
		}
		if (map_idx >= nr_maps) {
3197 3198
			pr_warn("map relo failed to find map for sec %u, off %zu\n",
				shdr_idx, (size_t)sym->st_value);
3199 3200 3201 3202 3203
			return -LIBBPF_ERRNO__RELOC;
		}
		reloc_desc->type = RELO_LD64;
		reloc_desc->insn_idx = insn_idx;
		reloc_desc->map_idx = map_idx;
3204
		reloc_desc->sym_off = 0; /* sym->st_value determines map_idx */
3205 3206 3207 3208 3209 3210 3211 3212 3213 3214 3215 3216 3217 3218 3219 3220 3221 3222 3223 3224 3225 3226 3227 3228 3229 3230
		return 0;
	}

	/* global data map relocation */
	if (!bpf_object__shndx_is_data(obj, shdr_idx)) {
		pr_warn("bad data relo against section %u\n", shdr_idx);
		return -LIBBPF_ERRNO__RELOC;
	}
	for (map_idx = 0; map_idx < nr_maps; map_idx++) {
		map = &obj->maps[map_idx];
		if (map->libbpf_type != type)
			continue;
		pr_debug("found data map %zd (%s, sec %d, off %zu) for insn %u\n",
			 map_idx, map->name, map->sec_idx, map->sec_offset,
			 insn_idx);
		break;
	}
	if (map_idx >= nr_maps) {
		pr_warn("data relo failed to find map for sec %u\n",
			shdr_idx);
		return -LIBBPF_ERRNO__RELOC;
	}

	reloc_desc->type = RELO_DATA;
	reloc_desc->insn_idx = insn_idx;
	reloc_desc->map_idx = map_idx;
3231
	reloc_desc->sym_off = sym->st_value;
3232 3233 3234
	return 0;
}

3235
static int
3236 3237
bpf_program__collect_reloc(struct bpf_program *prog, GElf_Shdr *shdr,
			   Elf_Data *data, struct bpf_object *obj)
3238
{
3239
	Elf_Data *symbols = obj->efile.symbols;
3240
	int err, i, nrels;
3241

3242
	pr_debug("collecting relocating info for: '%s'\n", prog->section_name);
3243 3244 3245 3246
	nrels = shdr->sh_size / shdr->sh_entsize;

	prog->reloc_desc = malloc(sizeof(*prog->reloc_desc) * nrels);
	if (!prog->reloc_desc) {
3247
		pr_warn("failed to alloc memory in relocation\n");
3248 3249 3250 3251 3252
		return -ENOMEM;
	}
	prog->nr_reloc = nrels;

	for (i = 0; i < nrels; i++) {
3253
		const char *name;
3254
		__u32 insn_idx;
3255 3256
		GElf_Sym sym;
		GElf_Rel rel;
3257 3258

		if (!gelf_getrel(data, i, &rel)) {
3259
			pr_warn("relocation: failed to get %d reloc\n", i);
3260
			return -LIBBPF_ERRNO__FORMAT;
3261
		}
3262
		if (!gelf_getsym(symbols, GELF_R_SYM(rel.r_info), &sym)) {
3263 3264
			pr_warn("relocation: symbol %"PRIx64" not found\n",
				GELF_R_SYM(rel.r_info));
3265
			return -LIBBPF_ERRNO__FORMAT;
3266
		}
3267 3268
		if (rel.r_offset % sizeof(struct bpf_insn))
			return -LIBBPF_ERRNO__FORMAT;
3269

3270
		insn_idx = rel.r_offset / sizeof(struct bpf_insn);
3271 3272 3273
		name = elf_strptr(obj->efile.elf, obj->efile.strtabidx,
				  sym.st_name) ? : "<?>";

3274 3275 3276
		pr_debug("relo for shdr %u, symb %zu, value %zu, type %d, bind %d, name %d (\'%s\'), insn %u\n",
			 (__u32)sym.st_shndx, (size_t)GELF_R_SYM(rel.r_info),
			 (size_t)sym.st_value, GELF_ST_TYPE(sym.st_info),
3277 3278
			 GELF_ST_BIND(sym.st_info), sym.st_name, name,
			 insn_idx);
3279

3280 3281 3282 3283
		err = bpf_program__record_reloc(prog, &prog->reloc_desc[i],
						insn_idx, name, &sym, &rel);
		if (err)
			return err;
3284 3285 3286 3287
	}
	return 0;
}

3288
static int bpf_map_find_btf_info(struct bpf_object *obj, struct bpf_map *map)
3289 3290
{
	struct bpf_map_def *def = &map->def;
3291
	__u32 key_type_id = 0, value_type_id = 0;
3292
	int ret;
3293

3294 3295 3296 3297 3298 3299
	/* if it's BTF-defined map, we don't need to search for type IDs.
	 * For struct_ops map, it does not need btf_key_type_id and
	 * btf_value_type_id.
	 */
	if (map->sec_idx == obj->efile.btf_maps_shndx ||
	    bpf_map__is_struct_ops(map))
3300 3301
		return 0;

3302
	if (!bpf_map__is_internal(map)) {
3303
		ret = btf__get_map_kv_tids(obj->btf, map->name, def->key_size,
3304 3305 3306 3307 3308 3309 3310
					   def->value_size, &key_type_id,
					   &value_type_id);
	} else {
		/*
		 * LLVM annotates global data differently in BTF, that is,
		 * only as '.data', '.bss' or '.rodata'.
		 */
3311
		ret = btf__find_by_name(obj->btf,
3312 3313 3314
				libbpf_type_to_btf_name[map->libbpf_type]);
	}
	if (ret < 0)
3315
		return ret;
3316

3317
	map->btf_key_type_id = key_type_id;
3318 3319
	map->btf_value_type_id = bpf_map__is_internal(map) ?
				 ret : value_type_id;
3320 3321 3322
	return 0;
}

J
Jakub Kicinski 已提交
3323 3324 3325 3326 3327 3328 3329 3330 3331 3332 3333 3334 3335 3336 3337 3338
int bpf_map__reuse_fd(struct bpf_map *map, int fd)
{
	struct bpf_map_info info = {};
	__u32 len = sizeof(info);
	int new_fd, err;
	char *new_name;

	err = bpf_obj_get_info_by_fd(fd, &info, &len);
	if (err)
		return err;

	new_name = strdup(info.name);
	if (!new_name)
		return -errno;

	new_fd = open("/", O_RDONLY | O_CLOEXEC);
3339 3340
	if (new_fd < 0) {
		err = -errno;
J
Jakub Kicinski 已提交
3341
		goto err_free_new_name;
3342
	}
J
Jakub Kicinski 已提交
3343 3344

	new_fd = dup3(fd, new_fd, O_CLOEXEC);
3345 3346
	if (new_fd < 0) {
		err = -errno;
J
Jakub Kicinski 已提交
3347
		goto err_close_new_fd;
3348
	}
J
Jakub Kicinski 已提交
3349 3350

	err = zclose(map->fd);
3351 3352
	if (err) {
		err = -errno;
J
Jakub Kicinski 已提交
3353
		goto err_close_new_fd;
3354
	}
J
Jakub Kicinski 已提交
3355 3356 3357 3358 3359 3360 3361 3362 3363 3364 3365
	free(map->name);

	map->fd = new_fd;
	map->name = new_name;
	map->def.type = info.type;
	map->def.key_size = info.key_size;
	map->def.value_size = info.value_size;
	map->def.max_entries = info.max_entries;
	map->def.map_flags = info.map_flags;
	map->btf_key_type_id = info.btf_key_type_id;
	map->btf_value_type_id = info.btf_value_type_id;
3366
	map->reused = true;
J
Jakub Kicinski 已提交
3367 3368 3369 3370 3371 3372 3373

	return 0;

err_close_new_fd:
	close(new_fd);
err_free_new_name:
	free(new_name);
3374
	return err;
J
Jakub Kicinski 已提交
3375 3376
}

3377
__u32 bpf_map__max_entries(const struct bpf_map *map)
3378
{
3379 3380
	return map->def.max_entries;
}
3381

3382 3383
int bpf_map__set_max_entries(struct bpf_map *map, __u32 max_entries)
{
3384 3385 3386 3387 3388 3389
	if (map->fd >= 0)
		return -EBUSY;
	map->def.max_entries = max_entries;
	return 0;
}

3390 3391 3392 3393 3394 3395 3396 3397
int bpf_map__resize(struct bpf_map *map, __u32 max_entries)
{
	if (!map || !max_entries)
		return -EINVAL;

	return bpf_map__set_max_entries(map, max_entries);
}

3398
static int
3399
bpf_object__probe_loading(struct bpf_object *obj)
3400 3401 3402 3403 3404 3405 3406 3407 3408 3409 3410 3411 3412 3413 3414 3415 3416 3417 3418
{
	struct bpf_load_program_attr attr;
	char *cp, errmsg[STRERR_BUFSIZE];
	struct bpf_insn insns[] = {
		BPF_MOV64_IMM(BPF_REG_0, 0),
		BPF_EXIT_INSN(),
	};
	int ret;

	/* make sure basic loading works */

	memset(&attr, 0, sizeof(attr));
	attr.prog_type = BPF_PROG_TYPE_SOCKET_FILTER;
	attr.insns = insns;
	attr.insns_cnt = ARRAY_SIZE(insns);
	attr.license = "GPL";

	ret = bpf_load_program_xattr(&attr, NULL, 0);
	if (ret < 0) {
3419 3420 3421 3422 3423 3424 3425
		ret = errno;
		cp = libbpf_strerror_r(ret, errmsg, sizeof(errmsg));
		pr_warn("Error in %s():%s(%d). Couldn't load trivial BPF "
			"program. Make sure your kernel supports BPF "
			"(CONFIG_BPF_SYSCALL=y) and/or that RLIMIT_MEMLOCK is "
			"set to big enough value.\n", __func__, cp, ret);
		return -ret;
3426 3427 3428
	}
	close(ret);

3429 3430 3431 3432 3433 3434 3435 3436 3437 3438 3439 3440 3441 3442
	return 0;
}

static int
bpf_object__probe_name(struct bpf_object *obj)
{
	struct bpf_load_program_attr attr;
	struct bpf_insn insns[] = {
		BPF_MOV64_IMM(BPF_REG_0, 0),
		BPF_EXIT_INSN(),
	};
	int ret;

	/* make sure loading with name works */
3443

3444 3445 3446 3447 3448
	memset(&attr, 0, sizeof(attr));
	attr.prog_type = BPF_PROG_TYPE_SOCKET_FILTER;
	attr.insns = insns;
	attr.insns_cnt = ARRAY_SIZE(insns);
	attr.license = "GPL";
3449 3450 3451 3452 3453 3454 3455 3456 3457 3458
	attr.name = "test";
	ret = bpf_load_program_xattr(&attr, NULL, 0);
	if (ret >= 0) {
		obj->caps.name = 1;
		close(ret);
	}

	return 0;
}

3459 3460 3461 3462 3463 3464 3465 3466 3467 3468 3469 3470 3471 3472 3473 3474 3475 3476 3477 3478 3479 3480
static int
bpf_object__probe_global_data(struct bpf_object *obj)
{
	struct bpf_load_program_attr prg_attr;
	struct bpf_create_map_attr map_attr;
	char *cp, errmsg[STRERR_BUFSIZE];
	struct bpf_insn insns[] = {
		BPF_LD_MAP_VALUE(BPF_REG_1, 0, 16),
		BPF_ST_MEM(BPF_DW, BPF_REG_1, 0, 42),
		BPF_MOV64_IMM(BPF_REG_0, 0),
		BPF_EXIT_INSN(),
	};
	int ret, map;

	memset(&map_attr, 0, sizeof(map_attr));
	map_attr.map_type = BPF_MAP_TYPE_ARRAY;
	map_attr.key_size = sizeof(int);
	map_attr.value_size = 32;
	map_attr.max_entries = 1;

	map = bpf_create_map_xattr(&map_attr);
	if (map < 0) {
3481 3482
		ret = -errno;
		cp = libbpf_strerror_r(ret, errmsg, sizeof(errmsg));
3483
		pr_warn("Error in %s():%s(%d). Couldn't create simple array map.\n",
3484 3485
			__func__, cp, -ret);
		return ret;
3486 3487 3488 3489 3490 3491 3492 3493 3494 3495 3496 3497 3498 3499 3500 3501 3502 3503 3504 3505
	}

	insns[0].imm = map;

	memset(&prg_attr, 0, sizeof(prg_attr));
	prg_attr.prog_type = BPF_PROG_TYPE_SOCKET_FILTER;
	prg_attr.insns = insns;
	prg_attr.insns_cnt = ARRAY_SIZE(insns);
	prg_attr.license = "GPL";

	ret = bpf_load_program_xattr(&prg_attr, NULL, 0);
	if (ret >= 0) {
		obj->caps.global_data = 1;
		close(ret);
	}

	close(map);
	return 0;
}

3506 3507
static int bpf_object__probe_btf_func(struct bpf_object *obj)
{
3508
	static const char strs[] = "\0int\0x\0a";
3509 3510 3511 3512 3513 3514 3515 3516 3517 3518
	/* void x(int a) {} */
	__u32 types[] = {
		/* int */
		BTF_TYPE_INT_ENC(1, BTF_INT_SIGNED, 0, 32, 4),  /* [1] */
		/* FUNC_PROTO */                                /* [2] */
		BTF_TYPE_ENC(0, BTF_INFO_ENC(BTF_KIND_FUNC_PROTO, 0, 1), 0),
		BTF_PARAM_ENC(7, 1),
		/* FUNC x */                                    /* [3] */
		BTF_TYPE_ENC(5, BTF_INFO_ENC(BTF_KIND_FUNC, 0, 0), 2),
	};
3519
	int btf_fd;
3520

3521 3522 3523
	btf_fd = libbpf__load_raw_btf((char *)types, sizeof(types),
				      strs, sizeof(strs));
	if (btf_fd >= 0) {
3524
		obj->caps.btf_func = 1;
3525 3526 3527 3528
		close(btf_fd);
		return 1;
	}

3529 3530 3531
	return 0;
}

3532 3533 3534 3535 3536 3537 3538 3539 3540 3541 3542 3543 3544 3545 3546 3547 3548 3549 3550 3551 3552 3553 3554 3555 3556 3557
static int bpf_object__probe_btf_func_global(struct bpf_object *obj)
{
	static const char strs[] = "\0int\0x\0a";
	/* static void x(int a) {} */
	__u32 types[] = {
		/* int */
		BTF_TYPE_INT_ENC(1, BTF_INT_SIGNED, 0, 32, 4),  /* [1] */
		/* FUNC_PROTO */                                /* [2] */
		BTF_TYPE_ENC(0, BTF_INFO_ENC(BTF_KIND_FUNC_PROTO, 0, 1), 0),
		BTF_PARAM_ENC(7, 1),
		/* FUNC x BTF_FUNC_GLOBAL */                    /* [3] */
		BTF_TYPE_ENC(5, BTF_INFO_ENC(BTF_KIND_FUNC, 0, BTF_FUNC_GLOBAL), 2),
	};
	int btf_fd;

	btf_fd = libbpf__load_raw_btf((char *)types, sizeof(types),
				      strs, sizeof(strs));
	if (btf_fd >= 0) {
		obj->caps.btf_func_global = 1;
		close(btf_fd);
		return 1;
	}

	return 0;
}

3558 3559
static int bpf_object__probe_btf_datasec(struct bpf_object *obj)
{
3560
	static const char strs[] = "\0x\0.data";
3561 3562 3563 3564 3565 3566 3567 3568 3569 3570 3571
	/* static int a; */
	__u32 types[] = {
		/* int */
		BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 32, 4),  /* [1] */
		/* VAR x */                                     /* [2] */
		BTF_TYPE_ENC(1, BTF_INFO_ENC(BTF_KIND_VAR, 0, 0), 1),
		BTF_VAR_STATIC,
		/* DATASEC val */                               /* [3] */
		BTF_TYPE_ENC(3, BTF_INFO_ENC(BTF_KIND_DATASEC, 0, 1), 4),
		BTF_VAR_SECINFO_ENC(2, 0, 4),
	};
3572
	int btf_fd;
3573

3574 3575 3576
	btf_fd = libbpf__load_raw_btf((char *)types, sizeof(types),
				      strs, sizeof(strs));
	if (btf_fd >= 0) {
3577
		obj->caps.btf_datasec = 1;
3578 3579 3580 3581
		close(btf_fd);
		return 1;
	}

3582 3583 3584
	return 0;
}

3585 3586 3587 3588 3589 3590 3591 3592 3593 3594 3595 3596 3597 3598 3599 3600 3601 3602 3603 3604 3605
static int bpf_object__probe_array_mmap(struct bpf_object *obj)
{
	struct bpf_create_map_attr attr = {
		.map_type = BPF_MAP_TYPE_ARRAY,
		.map_flags = BPF_F_MMAPABLE,
		.key_size = sizeof(int),
		.value_size = sizeof(int),
		.max_entries = 1,
	};
	int fd;

	fd = bpf_create_map_xattr(&attr);
	if (fd >= 0) {
		obj->caps.array_mmap = 1;
		close(fd);
		return 1;
	}

	return 0;
}

3606 3607 3608 3609 3610 3611 3612 3613 3614 3615 3616 3617 3618 3619 3620 3621 3622 3623 3624 3625 3626 3627 3628 3629 3630 3631 3632 3633 3634 3635 3636
static int
bpf_object__probe_exp_attach_type(struct bpf_object *obj)
{
	struct bpf_load_program_attr attr;
	struct bpf_insn insns[] = {
		BPF_MOV64_IMM(BPF_REG_0, 0),
		BPF_EXIT_INSN(),
	};
	int fd;

	memset(&attr, 0, sizeof(attr));
	/* use any valid combination of program type and (optional)
	 * non-zero expected attach type (i.e., not a BPF_CGROUP_INET_INGRESS)
	 * to see if kernel supports expected_attach_type field for
	 * BPF_PROG_LOAD command
	 */
	attr.prog_type = BPF_PROG_TYPE_CGROUP_SOCK;
	attr.expected_attach_type = BPF_CGROUP_INET_SOCK_CREATE;
	attr.insns = insns;
	attr.insns_cnt = ARRAY_SIZE(insns);
	attr.license = "GPL";

	fd = bpf_load_program_xattr(&attr, NULL, 0);
	if (fd >= 0) {
		obj->caps.exp_attach_type = 1;
		close(fd);
		return 1;
	}
	return 0;
}

3637 3638 3639
static int
bpf_object__probe_caps(struct bpf_object *obj)
{
3640 3641 3642
	int (*probe_fn[])(struct bpf_object *obj) = {
		bpf_object__probe_name,
		bpf_object__probe_global_data,
3643
		bpf_object__probe_btf_func,
3644
		bpf_object__probe_btf_func_global,
3645
		bpf_object__probe_btf_datasec,
3646
		bpf_object__probe_array_mmap,
3647
		bpf_object__probe_exp_attach_type,
3648 3649 3650 3651 3652 3653
	};
	int i, ret;

	for (i = 0; i < ARRAY_SIZE(probe_fn); i++) {
		ret = probe_fn[i](obj);
		if (ret < 0)
3654
			pr_debug("Probe #%d failed with %d.\n", i, ret);
3655 3656 3657
	}

	return 0;
3658 3659
}

3660 3661 3662 3663 3664 3665 3666 3667 3668 3669 3670 3671 3672 3673 3674 3675 3676 3677 3678 3679 3680 3681 3682 3683 3684 3685 3686 3687 3688 3689 3690 3691 3692 3693 3694 3695 3696 3697 3698 3699 3700 3701 3702 3703 3704 3705 3706 3707 3708 3709 3710 3711 3712 3713 3714 3715 3716 3717 3718 3719
static bool map_is_reuse_compat(const struct bpf_map *map, int map_fd)
{
	struct bpf_map_info map_info = {};
	char msg[STRERR_BUFSIZE];
	__u32 map_info_len;

	map_info_len = sizeof(map_info);

	if (bpf_obj_get_info_by_fd(map_fd, &map_info, &map_info_len)) {
		pr_warn("failed to get map info for map FD %d: %s\n",
			map_fd, libbpf_strerror_r(errno, msg, sizeof(msg)));
		return false;
	}

	return (map_info.type == map->def.type &&
		map_info.key_size == map->def.key_size &&
		map_info.value_size == map->def.value_size &&
		map_info.max_entries == map->def.max_entries &&
		map_info.map_flags == map->def.map_flags);
}

static int
bpf_object__reuse_map(struct bpf_map *map)
{
	char *cp, errmsg[STRERR_BUFSIZE];
	int err, pin_fd;

	pin_fd = bpf_obj_get(map->pin_path);
	if (pin_fd < 0) {
		err = -errno;
		if (err == -ENOENT) {
			pr_debug("found no pinned map to reuse at '%s'\n",
				 map->pin_path);
			return 0;
		}

		cp = libbpf_strerror_r(-err, errmsg, sizeof(errmsg));
		pr_warn("couldn't retrieve pinned map '%s': %s\n",
			map->pin_path, cp);
		return err;
	}

	if (!map_is_reuse_compat(map, pin_fd)) {
		pr_warn("couldn't reuse pinned map at '%s': parameter mismatch\n",
			map->pin_path);
		close(pin_fd);
		return -EINVAL;
	}

	err = bpf_map__reuse_fd(map, pin_fd);
	if (err) {
		close(pin_fd);
		return err;
	}
	map->pinned = true;
	pr_debug("reused pinned map at '%s'\n", map->pin_path);

	return 0;
}

3720 3721 3722
static int
bpf_object__populate_internal_map(struct bpf_object *obj, struct bpf_map *map)
{
3723
	enum libbpf_map_type map_type = map->libbpf_type;
3724 3725 3726
	char *cp, errmsg[STRERR_BUFSIZE];
	int err, zero = 0;

3727 3728 3729 3730 3731 3732 3733 3734
	err = bpf_map_update_elem(map->fd, &zero, map->mmaped, 0);
	if (err) {
		err = -errno;
		cp = libbpf_strerror_r(err, errmsg, sizeof(errmsg));
		pr_warn("Error setting initial map(%s) contents: %s\n",
			map->name, cp);
		return err;
	}
3735

3736 3737
	/* Freeze .rodata and .kconfig map as read-only from syscall side. */
	if (map_type == LIBBPF_MAP_RODATA || map_type == LIBBPF_MAP_KCONFIG) {
3738 3739
		err = bpf_map_freeze(map->fd);
		if (err) {
3740 3741
			err = -errno;
			cp = libbpf_strerror_r(err, errmsg, sizeof(errmsg));
3742 3743
			pr_warn("Error freezing map(%s) as read-only: %s\n",
				map->name, cp);
3744
			return err;
3745 3746
		}
	}
3747
	return 0;
3748 3749
}

3750 3751 3752 3753 3754 3755 3756 3757 3758 3759 3760 3761 3762 3763 3764 3765
static void bpf_map__destroy(struct bpf_map *map);

static int bpf_object__create_map(struct bpf_object *obj, struct bpf_map *map)
{
	struct bpf_create_map_attr create_attr;
	struct bpf_map_def *def = &map->def;

	memset(&create_attr, 0, sizeof(create_attr));

	if (obj->caps.name)
		create_attr.name = map->name;
	create_attr.map_ifindex = map->map_ifindex;
	create_attr.map_type = def->type;
	create_attr.map_flags = def->map_flags;
	create_attr.key_size = def->key_size;
	create_attr.value_size = def->value_size;
3766
	create_attr.numa_node = map->numa_node;
3767 3768 3769 3770 3771 3772 3773 3774 3775 3776 3777 3778 3779 3780 3781 3782 3783 3784 3785 3786 3787 3788 3789

	if (def->type == BPF_MAP_TYPE_PERF_EVENT_ARRAY && !def->max_entries) {
		int nr_cpus;

		nr_cpus = libbpf_num_possible_cpus();
		if (nr_cpus < 0) {
			pr_warn("map '%s': failed to determine number of system CPUs: %d\n",
				map->name, nr_cpus);
			return nr_cpus;
		}
		pr_debug("map '%s': setting size to %d\n", map->name, nr_cpus);
		create_attr.max_entries = nr_cpus;
	} else {
		create_attr.max_entries = def->max_entries;
	}

	if (bpf_map__is_struct_ops(map))
		create_attr.btf_vmlinux_value_type_id =
			map->btf_vmlinux_value_type_id;

	create_attr.btf_fd = 0;
	create_attr.btf_key_type_id = 0;
	create_attr.btf_value_type_id = 0;
3790
	if (obj->btf && btf__fd(obj->btf) >= 0 && !bpf_map_find_btf_info(obj, map)) {
3791 3792 3793 3794 3795
		create_attr.btf_fd = btf__fd(obj->btf);
		create_attr.btf_key_type_id = map->btf_key_type_id;
		create_attr.btf_value_type_id = map->btf_value_type_id;
	}

3796 3797 3798 3799 3800 3801 3802 3803 3804 3805 3806 3807 3808 3809 3810 3811
	if (bpf_map_type__is_map_in_map(def->type)) {
		if (map->inner_map) {
			int err;

			err = bpf_object__create_map(obj, map->inner_map);
			if (err) {
				pr_warn("map '%s': failed to create inner map: %d\n",
					map->name, err);
				return err;
			}
			map->inner_map_fd = bpf_map__fd(map->inner_map);
		}
		if (map->inner_map_fd >= 0)
			create_attr.inner_map_fd = map->inner_map_fd;
	}

3812 3813 3814 3815 3816 3817 3818 3819 3820 3821 3822 3823 3824 3825 3826 3827 3828 3829 3830 3831
	map->fd = bpf_create_map_xattr(&create_attr);
	if (map->fd < 0 && (create_attr.btf_key_type_id ||
			    create_attr.btf_value_type_id)) {
		char *cp, errmsg[STRERR_BUFSIZE];
		int err = -errno;

		cp = libbpf_strerror_r(err, errmsg, sizeof(errmsg));
		pr_warn("Error in bpf_create_map_xattr(%s):%s(%d). Retrying without BTF.\n",
			map->name, cp, err);
		create_attr.btf_fd = 0;
		create_attr.btf_key_type_id = 0;
		create_attr.btf_value_type_id = 0;
		map->btf_key_type_id = 0;
		map->btf_value_type_id = 0;
		map->fd = bpf_create_map_xattr(&create_attr);
	}

	if (map->fd < 0)
		return -errno;

3832 3833 3834 3835 3836
	if (bpf_map_type__is_map_in_map(def->type) && map->inner_map) {
		bpf_map__destroy(map->inner_map);
		zfree(&map->inner_map);
	}

3837 3838 3839
	return 0;
}

3840 3841 3842
static int
bpf_object__create_maps(struct bpf_object *obj)
{
3843 3844 3845
	struct bpf_map *map;
	char *cp, errmsg[STRERR_BUFSIZE];
	unsigned int i, j;
3846
	int err;
3847

3848
	for (i = 0; i < obj->nr_maps; i++) {
3849
		map = &obj->maps[i];
3850

3851 3852 3853
		if (map->pin_path) {
			err = bpf_object__reuse_map(map);
			if (err) {
3854
				pr_warn("map '%s': error reusing pinned map\n",
3855
					map->name);
3856
				goto err_out;
3857 3858 3859
			}
		}

J
Jakub Kicinski 已提交
3860
		if (map->fd >= 0) {
3861
			pr_debug("map '%s': skipping creation (preset fd=%d)\n",
J
Jakub Kicinski 已提交
3862 3863 3864 3865
				 map->name, map->fd);
			continue;
		}

3866 3867 3868
		err = bpf_object__create_map(obj, map);
		if (err)
			goto err_out;
3869

3870 3871
		pr_debug("map '%s': created successfully, fd=%d\n", map->name,
			 map->fd);
3872 3873 3874 3875

		if (bpf_map__is_internal(map)) {
			err = bpf_object__populate_internal_map(obj, map);
			if (err < 0) {
3876
				zclose(map->fd);
3877 3878 3879 3880
				goto err_out;
			}
		}

3881 3882 3883 3884 3885 3886 3887 3888 3889 3890 3891 3892 3893 3894 3895 3896 3897 3898 3899 3900 3901 3902 3903 3904 3905
		if (map->init_slots_sz) {
			for (j = 0; j < map->init_slots_sz; j++) {
				const struct bpf_map *targ_map;
				int fd;

				if (!map->init_slots[j])
					continue;

				targ_map = map->init_slots[j];
				fd = bpf_map__fd(targ_map);
				err = bpf_map_update_elem(map->fd, &j, &fd, 0);
				if (err) {
					err = -errno;
					pr_warn("map '%s': failed to initialize slot [%d] to map '%s' fd=%d: %d\n",
						map->name, j, targ_map->name,
						fd, err);
					goto err_out;
				}
				pr_debug("map '%s': slot [%d] set to map '%s' fd=%d\n",
					 map->name, j, targ_map->name, fd);
			}
			zfree(&map->init_slots);
			map->init_slots_sz = 0;
		}

3906 3907 3908
		if (map->pin_path && !map->pinned) {
			err = bpf_map__pin(map, NULL);
			if (err) {
3909 3910 3911 3912
				pr_warn("map '%s': failed to auto-pin at '%s': %d\n",
					map->name, map->pin_path, err);
				zclose(map->fd);
				goto err_out;
3913 3914
			}
		}
3915 3916 3917
	}

	return 0;
3918 3919 3920 3921 3922 3923 3924 3925

err_out:
	cp = libbpf_strerror_r(err, errmsg, sizeof(errmsg));
	pr_warn("map '%s': failed to create: %s(%d)\n", map->name, cp, err);
	pr_perm_msg(err);
	for (j = 0; j < i; j++)
		zclose(obj->maps[j].fd);
	return err;
3926 3927
}

3928 3929 3930 3931 3932
static int
check_btf_ext_reloc_err(struct bpf_program *prog, int err,
			void *btf_prog_info, const char *info_name)
{
	if (err != -ENOENT) {
3933 3934
		pr_warn("Error in loading %s for sec %s.\n",
			info_name, prog->section_name);
3935 3936 3937 3938 3939 3940 3941 3942
		return err;
	}

	/* err == -ENOENT (i.e. prog->section_name not found in btf_ext) */

	if (btf_prog_info) {
		/*
		 * Some info has already been found but has problem
3943
		 * in the last btf_ext reloc. Must have to error out.
3944
		 */
3945 3946
		pr_warn("Error in relocating %s for sec %s.\n",
			info_name, prog->section_name);
3947 3948 3949
		return err;
	}

3950
	/* Have problem loading the very first info. Ignore the rest. */
3951 3952
	pr_warn("Cannot find %s for main program sec %s. Ignore all %s.\n",
		info_name, prog->section_name, info_name);
3953 3954 3955 3956 3957 3958 3959 3960 3961 3962 3963 3964 3965 3966 3967 3968 3969 3970 3971 3972 3973 3974 3975 3976 3977 3978 3979 3980
	return 0;
}

static int
bpf_program_reloc_btf_ext(struct bpf_program *prog, struct bpf_object *obj,
			  const char *section_name,  __u32 insn_offset)
{
	int err;

	if (!insn_offset || prog->func_info) {
		/*
		 * !insn_offset => main program
		 *
		 * For sub prog, the main program's func_info has to
		 * be loaded first (i.e. prog->func_info != NULL)
		 */
		err = btf_ext__reloc_func_info(obj->btf, obj->btf_ext,
					       section_name, insn_offset,
					       &prog->func_info,
					       &prog->func_info_cnt);
		if (err)
			return check_btf_ext_reloc_err(prog, err,
						       prog->func_info,
						       "bpf_func_info");

		prog->func_info_rec_size = btf_ext__func_info_rec_size(obj->btf_ext);
	}

3981 3982 3983 3984 3985 3986 3987 3988 3989 3990 3991 3992 3993
	if (!insn_offset || prog->line_info) {
		err = btf_ext__reloc_line_info(obj->btf, obj->btf_ext,
					       section_name, insn_offset,
					       &prog->line_info,
					       &prog->line_info_cnt);
		if (err)
			return check_btf_ext_reloc_err(prog, err,
						       prog->line_info,
						       "bpf_line_info");

		prog->line_info_rec_size = btf_ext__line_info_rec_size(obj->btf_ext);
	}

3994 3995 3996
	return 0;
}

3997 3998 3999 4000 4001 4002 4003 4004 4005 4006 4007 4008 4009 4010 4011 4012 4013 4014 4015
#define BPF_CORE_SPEC_MAX_LEN 64

/* represents BPF CO-RE field or array element accessor */
struct bpf_core_accessor {
	__u32 type_id;		/* struct/union type or array element type */
	__u32 idx;		/* field index or array index */
	const char *name;	/* field name or NULL for array accessor */
};

struct bpf_core_spec {
	const struct btf *btf;
	/* high-level spec: named fields and array indices only */
	struct bpf_core_accessor spec[BPF_CORE_SPEC_MAX_LEN];
	/* high-level spec length */
	int len;
	/* raw, low-level spec: 1-to-1 with accessor spec string */
	int raw_spec[BPF_CORE_SPEC_MAX_LEN];
	/* raw spec length */
	int raw_len;
4016 4017
	/* field bit offset represented by spec */
	__u32 bit_offset;
4018 4019 4020 4021 4022 4023 4024
};

static bool str_is_empty(const char *s)
{
	return !s || !s[0];
}

4025 4026 4027 4028 4029 4030 4031 4032 4033 4034 4035 4036 4037 4038 4039
static bool is_flex_arr(const struct btf *btf,
			const struct bpf_core_accessor *acc,
			const struct btf_array *arr)
{
	const struct btf_type *t;

	/* not a flexible array, if not inside a struct or has non-zero size */
	if (!acc->name || arr->nelems > 0)
		return false;

	/* has to be the last member of enclosing struct */
	t = btf__type_by_id(btf, acc->type_id);
	return acc->idx == btf_vlen(t) - 1;
}

4040
/*
4041
 * Turn bpf_field_reloc into a low- and high-level spec representation,
4042
 * validating correctness along the way, as well as calculating resulting
4043 4044
 * field bit offset, specified by accessor string. Low-level spec captures
 * every single level of nestedness, including traversing anonymous
4045 4046 4047 4048 4049 4050 4051 4052 4053 4054 4055 4056 4057 4058 4059 4060 4061 4062 4063 4064 4065 4066 4067 4068 4069 4070 4071 4072 4073 4074 4075 4076
 * struct/union members. High-level one only captures semantically meaningful
 * "turning points": named fields and array indicies.
 * E.g., for this case:
 *
 *   struct sample {
 *       int __unimportant;
 *       struct {
 *           int __1;
 *           int __2;
 *           int a[7];
 *       };
 *   };
 *
 *   struct sample *s = ...;
 *
 *   int x = &s->a[3]; // access string = '0:1:2:3'
 *
 * Low-level spec has 1:1 mapping with each element of access string (it's
 * just a parsed access string representation): [0, 1, 2, 3].
 *
 * High-level spec will capture only 3 points:
 *   - intial zero-index access by pointer (&s->... is the same as &s[0]...);
 *   - field 'a' access (corresponds to '2' in low-level spec);
 *   - array element #3 access (corresponds to '3' in low-level spec).
 *
 */
static int bpf_core_spec_parse(const struct btf *btf,
			       __u32 type_id,
			       const char *spec_str,
			       struct bpf_core_spec *spec)
{
	int access_idx, parsed_len, i;
4077
	struct bpf_core_accessor *acc;
4078 4079 4080 4081 4082 4083 4084 4085 4086 4087 4088 4089 4090 4091 4092 4093 4094 4095 4096 4097 4098 4099 4100 4101 4102 4103 4104 4105 4106 4107 4108 4109 4110 4111 4112 4113 4114 4115 4116
	const struct btf_type *t;
	const char *name;
	__u32 id;
	__s64 sz;

	if (str_is_empty(spec_str) || *spec_str == ':')
		return -EINVAL;

	memset(spec, 0, sizeof(*spec));
	spec->btf = btf;

	/* parse spec_str="0:1:2:3:4" into array raw_spec=[0, 1, 2, 3, 4] */
	while (*spec_str) {
		if (*spec_str == ':')
			++spec_str;
		if (sscanf(spec_str, "%d%n", &access_idx, &parsed_len) != 1)
			return -EINVAL;
		if (spec->raw_len == BPF_CORE_SPEC_MAX_LEN)
			return -E2BIG;
		spec_str += parsed_len;
		spec->raw_spec[spec->raw_len++] = access_idx;
	}

	if (spec->raw_len == 0)
		return -EINVAL;

	/* first spec value is always reloc type array index */
	t = skip_mods_and_typedefs(btf, type_id, &id);
	if (!t)
		return -EINVAL;

	access_idx = spec->raw_spec[0];
	spec->spec[0].type_id = id;
	spec->spec[0].idx = access_idx;
	spec->len++;

	sz = btf__resolve_size(btf, id);
	if (sz < 0)
		return sz;
4117
	spec->bit_offset = access_idx * sz * 8;
4118 4119 4120 4121 4122 4123 4124

	for (i = 1; i < spec->raw_len; i++) {
		t = skip_mods_and_typedefs(btf, id, &id);
		if (!t)
			return -EINVAL;

		access_idx = spec->raw_spec[i];
4125
		acc = &spec->spec[spec->len];
4126 4127 4128

		if (btf_is_composite(t)) {
			const struct btf_member *m;
4129
			__u32 bit_offset;
4130 4131 4132 4133

			if (access_idx >= btf_vlen(t))
				return -EINVAL;

4134 4135
			bit_offset = btf_member_bit_offset(t, access_idx);
			spec->bit_offset += bit_offset;
4136 4137 4138 4139 4140 4141 4142

			m = btf_members(t) + access_idx;
			if (m->name_off) {
				name = btf__name_by_offset(btf, m->name_off);
				if (str_is_empty(name))
					return -EINVAL;

4143 4144 4145
				acc->type_id = id;
				acc->idx = access_idx;
				acc->name = name;
4146 4147 4148 4149 4150 4151
				spec->len++;
			}

			id = m->type;
		} else if (btf_is_array(t)) {
			const struct btf_array *a = btf_array(t);
4152
			bool flex;
4153 4154

			t = skip_mods_and_typedefs(btf, a->type, &id);
4155 4156 4157 4158 4159
			if (!t)
				return -EINVAL;

			flex = is_flex_arr(btf, acc - 1, a);
			if (!flex && access_idx >= a->nelems)
4160 4161 4162 4163 4164 4165 4166 4167 4168
				return -EINVAL;

			spec->spec[spec->len].type_id = id;
			spec->spec[spec->len].idx = access_idx;
			spec->len++;

			sz = btf__resolve_size(btf, id);
			if (sz < 0)
				return sz;
4169
			spec->bit_offset += access_idx * sz * 8;
4170
		} else {
4171 4172
			pr_warn("relo for [%u] %s (at idx %d) captures type [%d] of unexpected kind %d\n",
				type_id, spec_str, i, id, btf_kind(t));
4173 4174 4175 4176 4177 4178 4179 4180 4181 4182 4183 4184 4185 4186 4187 4188 4189 4190 4191 4192 4193 4194 4195 4196 4197 4198 4199 4200 4201 4202 4203 4204 4205 4206 4207 4208 4209 4210 4211 4212 4213 4214 4215 4216 4217 4218 4219 4220 4221 4222 4223 4224 4225 4226 4227 4228 4229 4230 4231 4232 4233 4234 4235 4236 4237 4238 4239 4240 4241 4242 4243 4244 4245 4246
			return -EINVAL;
		}
	}

	return 0;
}

static bool bpf_core_is_flavor_sep(const char *s)
{
	/* check X___Y name pattern, where X and Y are not underscores */
	return s[0] != '_' &&				      /* X */
	       s[1] == '_' && s[2] == '_' && s[3] == '_' &&   /* ___ */
	       s[4] != '_';				      /* Y */
}

/* Given 'some_struct_name___with_flavor' return the length of a name prefix
 * before last triple underscore. Struct name part after last triple
 * underscore is ignored by BPF CO-RE relocation during relocation matching.
 */
static size_t bpf_core_essential_name_len(const char *name)
{
	size_t n = strlen(name);
	int i;

	for (i = n - 5; i >= 0; i--) {
		if (bpf_core_is_flavor_sep(name + i))
			return i + 1;
	}
	return n;
}

/* dynamically sized list of type IDs */
struct ids_vec {
	__u32 *data;
	int len;
};

static void bpf_core_free_cands(struct ids_vec *cand_ids)
{
	free(cand_ids->data);
	free(cand_ids);
}

static struct ids_vec *bpf_core_find_cands(const struct btf *local_btf,
					   __u32 local_type_id,
					   const struct btf *targ_btf)
{
	size_t local_essent_len, targ_essent_len;
	const char *local_name, *targ_name;
	const struct btf_type *t;
	struct ids_vec *cand_ids;
	__u32 *new_ids;
	int i, err, n;

	t = btf__type_by_id(local_btf, local_type_id);
	if (!t)
		return ERR_PTR(-EINVAL);

	local_name = btf__name_by_offset(local_btf, t->name_off);
	if (str_is_empty(local_name))
		return ERR_PTR(-EINVAL);
	local_essent_len = bpf_core_essential_name_len(local_name);

	cand_ids = calloc(1, sizeof(*cand_ids));
	if (!cand_ids)
		return ERR_PTR(-ENOMEM);

	n = btf__get_nr_types(targ_btf);
	for (i = 1; i <= n; i++) {
		t = btf__type_by_id(targ_btf, i);
		targ_name = btf__name_by_offset(targ_btf, t->name_off);
		if (str_is_empty(targ_name))
			continue;

4247 4248 4249 4250
		t = skip_mods_and_typedefs(targ_btf, i, NULL);
		if (!btf_is_composite(t) && !btf_is_array(t))
			continue;

4251 4252 4253 4254 4255 4256 4257
		targ_essent_len = bpf_core_essential_name_len(targ_name);
		if (targ_essent_len != local_essent_len)
			continue;

		if (strncmp(local_name, targ_name, local_essent_len) == 0) {
			pr_debug("[%d] %s: found candidate [%d] %s\n",
				 local_type_id, local_name, i, targ_name);
4258 4259 4260
			new_ids = reallocarray(cand_ids->data,
					       cand_ids->len + 1,
					       sizeof(*cand_ids->data));
4261 4262 4263 4264 4265 4266 4267 4268 4269 4270 4271 4272 4273 4274 4275
			if (!new_ids) {
				err = -ENOMEM;
				goto err_out;
			}
			cand_ids->data = new_ids;
			cand_ids->data[cand_ids->len++] = i;
		}
	}
	return cand_ids;
err_out:
	bpf_core_free_cands(cand_ids);
	return ERR_PTR(err);
}

/* Check two types for compatibility, skipping const/volatile/restrict and
4276
 * typedefs, to ensure we are relocating compatible entities:
4277
 *   - any two STRUCTs/UNIONs are compatible and can be mixed;
4278
 *   - any two FWDs are compatible, if their names match (modulo flavor suffix);
4279
 *   - any two PTRs are always compatible;
4280 4281
 *   - for ENUMs, names should be the same (ignoring flavor suffix) or at
 *     least one of enums should be anonymous;
4282
 *   - for ENUMs, check sizes, names are ignored;
4283
 *   - for INT, size and signedness are ignored;
4284 4285 4286 4287 4288 4289 4290 4291 4292 4293 4294 4295 4296 4297 4298 4299 4300 4301 4302 4303 4304 4305 4306 4307 4308 4309 4310
 *   - for ARRAY, dimensionality is ignored, element types are checked for
 *     compatibility recursively;
 *   - everything else shouldn't be ever a target of relocation.
 * These rules are not set in stone and probably will be adjusted as we get
 * more experience with using BPF CO-RE relocations.
 */
static int bpf_core_fields_are_compat(const struct btf *local_btf,
				      __u32 local_id,
				      const struct btf *targ_btf,
				      __u32 targ_id)
{
	const struct btf_type *local_type, *targ_type;

recur:
	local_type = skip_mods_and_typedefs(local_btf, local_id, &local_id);
	targ_type = skip_mods_and_typedefs(targ_btf, targ_id, &targ_id);
	if (!local_type || !targ_type)
		return -EINVAL;

	if (btf_is_composite(local_type) && btf_is_composite(targ_type))
		return 1;
	if (btf_kind(local_type) != btf_kind(targ_type))
		return 0;

	switch (btf_kind(local_type)) {
	case BTF_KIND_PTR:
		return 1;
4311 4312 4313 4314 4315 4316 4317 4318 4319 4320 4321 4322 4323 4324 4325
	case BTF_KIND_FWD:
	case BTF_KIND_ENUM: {
		const char *local_name, *targ_name;
		size_t local_len, targ_len;

		local_name = btf__name_by_offset(local_btf,
						 local_type->name_off);
		targ_name = btf__name_by_offset(targ_btf, targ_type->name_off);
		local_len = bpf_core_essential_name_len(local_name);
		targ_len = bpf_core_essential_name_len(targ_name);
		/* one of them is anonymous or both w/ same flavor-less names */
		return local_len == 0 || targ_len == 0 ||
		       (local_len == targ_len &&
			strncmp(local_name, targ_name, local_len) == 0);
	}
4326
	case BTF_KIND_INT:
4327 4328 4329
		/* just reject deprecated bitfield-like integers; all other
		 * integers are by default compatible between each other
		 */
4330
		return btf_int_offset(local_type) == 0 &&
4331
		       btf_int_offset(targ_type) == 0;
4332 4333 4334 4335 4336
	case BTF_KIND_ARRAY:
		local_id = btf_array(local_type)->type;
		targ_id = btf_array(targ_type)->type;
		goto recur;
	default:
4337 4338
		pr_warn("unexpected kind %d relocated, local [%d], target [%d]\n",
			btf_kind(local_type), local_id, targ_id);
4339 4340 4341 4342 4343 4344 4345 4346
		return 0;
	}
}

/*
 * Given single high-level named field accessor in local type, find
 * corresponding high-level accessor for a target type. Along the way,
 * maintain low-level spec for target as well. Also keep updating target
4347
 * bit offset.
4348 4349 4350 4351 4352 4353 4354 4355 4356 4357 4358 4359 4360 4361 4362 4363 4364 4365 4366 4367 4368 4369 4370 4371 4372 4373 4374 4375 4376 4377 4378 4379 4380 4381 4382 4383 4384 4385
 *
 * Searching is performed through recursive exhaustive enumeration of all
 * fields of a struct/union. If there are any anonymous (embedded)
 * structs/unions, they are recursively searched as well. If field with
 * desired name is found, check compatibility between local and target types,
 * before returning result.
 *
 * 1 is returned, if field is found.
 * 0 is returned if no compatible field is found.
 * <0 is returned on error.
 */
static int bpf_core_match_member(const struct btf *local_btf,
				 const struct bpf_core_accessor *local_acc,
				 const struct btf *targ_btf,
				 __u32 targ_id,
				 struct bpf_core_spec *spec,
				 __u32 *next_targ_id)
{
	const struct btf_type *local_type, *targ_type;
	const struct btf_member *local_member, *m;
	const char *local_name, *targ_name;
	__u32 local_id;
	int i, n, found;

	targ_type = skip_mods_and_typedefs(targ_btf, targ_id, &targ_id);
	if (!targ_type)
		return -EINVAL;
	if (!btf_is_composite(targ_type))
		return 0;

	local_id = local_acc->type_id;
	local_type = btf__type_by_id(local_btf, local_id);
	local_member = btf_members(local_type) + local_acc->idx;
	local_name = btf__name_by_offset(local_btf, local_member->name_off);

	n = btf_vlen(targ_type);
	m = btf_members(targ_type);
	for (i = 0; i < n; i++, m++) {
4386
		__u32 bit_offset;
4387

4388
		bit_offset = btf_member_bit_offset(targ_type, i);
4389 4390 4391 4392 4393 4394

		/* too deep struct/union/array nesting */
		if (spec->raw_len == BPF_CORE_SPEC_MAX_LEN)
			return -E2BIG;

		/* speculate this member will be the good one */
4395
		spec->bit_offset += bit_offset;
4396 4397 4398 4399 4400 4401 4402 4403 4404 4405 4406 4407 4408 4409 4410 4411 4412 4413 4414 4415 4416 4417 4418 4419 4420 4421 4422 4423
		spec->raw_spec[spec->raw_len++] = i;

		targ_name = btf__name_by_offset(targ_btf, m->name_off);
		if (str_is_empty(targ_name)) {
			/* embedded struct/union, we need to go deeper */
			found = bpf_core_match_member(local_btf, local_acc,
						      targ_btf, m->type,
						      spec, next_targ_id);
			if (found) /* either found or error */
				return found;
		} else if (strcmp(local_name, targ_name) == 0) {
			/* matching named field */
			struct bpf_core_accessor *targ_acc;

			targ_acc = &spec->spec[spec->len++];
			targ_acc->type_id = targ_id;
			targ_acc->idx = i;
			targ_acc->name = targ_name;

			*next_targ_id = m->type;
			found = bpf_core_fields_are_compat(local_btf,
							   local_member->type,
							   targ_btf, m->type);
			if (!found)
				spec->len--; /* pop accessor */
			return found;
		}
		/* member turned out not to be what we looked for */
4424
		spec->bit_offset -= bit_offset;
4425 4426 4427 4428 4429 4430 4431 4432
		spec->raw_len--;
	}

	return 0;
}

/*
 * Try to match local spec to a target type and, if successful, produce full
4433
 * target spec (high-level, low-level + bit offset).
4434 4435 4436 4437 4438 4439 4440 4441 4442 4443 4444 4445 4446 4447 4448 4449 4450 4451 4452 4453 4454 4455 4456 4457 4458 4459 4460 4461 4462 4463 4464 4465 4466 4467 4468 4469
 */
static int bpf_core_spec_match(struct bpf_core_spec *local_spec,
			       const struct btf *targ_btf, __u32 targ_id,
			       struct bpf_core_spec *targ_spec)
{
	const struct btf_type *targ_type;
	const struct bpf_core_accessor *local_acc;
	struct bpf_core_accessor *targ_acc;
	int i, sz, matched;

	memset(targ_spec, 0, sizeof(*targ_spec));
	targ_spec->btf = targ_btf;

	local_acc = &local_spec->spec[0];
	targ_acc = &targ_spec->spec[0];

	for (i = 0; i < local_spec->len; i++, local_acc++, targ_acc++) {
		targ_type = skip_mods_and_typedefs(targ_spec->btf, targ_id,
						   &targ_id);
		if (!targ_type)
			return -EINVAL;

		if (local_acc->name) {
			matched = bpf_core_match_member(local_spec->btf,
							local_acc,
							targ_btf, targ_id,
							targ_spec, &targ_id);
			if (matched <= 0)
				return matched;
		} else {
			/* for i=0, targ_id is already treated as array element
			 * type (because it's the original struct), for others
			 * we should find array element type first
			 */
			if (i > 0) {
				const struct btf_array *a;
4470
				bool flex;
4471 4472 4473 4474 4475

				if (!btf_is_array(targ_type))
					return 0;

				a = btf_array(targ_type);
4476 4477
				flex = is_flex_arr(targ_btf, targ_acc - 1, a);
				if (!flex && local_acc->idx >= a->nelems)
4478 4479 4480 4481 4482 4483 4484 4485 4486 4487 4488 4489 4490 4491 4492 4493 4494 4495 4496 4497
					return 0;
				if (!skip_mods_and_typedefs(targ_btf, a->type,
							    &targ_id))
					return -EINVAL;
			}

			/* too deep struct/union/array nesting */
			if (targ_spec->raw_len == BPF_CORE_SPEC_MAX_LEN)
				return -E2BIG;

			targ_acc->type_id = targ_id;
			targ_acc->idx = local_acc->idx;
			targ_acc->name = NULL;
			targ_spec->len++;
			targ_spec->raw_spec[targ_spec->raw_len] = targ_acc->idx;
			targ_spec->raw_len++;

			sz = btf__resolve_size(targ_btf, targ_id);
			if (sz < 0)
				return sz;
4498
			targ_spec->bit_offset += local_acc->idx * sz * 8;
4499 4500 4501 4502 4503 4504
		}
	}

	return 1;
}

4505 4506 4507 4508 4509 4510 4511 4512 4513 4514 4515
static int bpf_core_calc_field_relo(const struct bpf_program *prog,
				    const struct bpf_field_reloc *relo,
				    const struct bpf_core_spec *spec,
				    __u32 *val, bool *validate)
{
	const struct bpf_core_accessor *acc = &spec->spec[spec->len - 1];
	const struct btf_type *t = btf__type_by_id(spec->btf, acc->type_id);
	__u32 byte_off, byte_sz, bit_off, bit_sz;
	const struct btf_member *m;
	const struct btf_type *mt;
	bool bitfield;
4516
	__s64 sz;
4517 4518 4519

	/* a[n] accessor needs special handling */
	if (!acc->name) {
4520 4521 4522 4523 4524 4525 4526 4527 4528
		if (relo->kind == BPF_FIELD_BYTE_OFFSET) {
			*val = spec->bit_offset / 8;
		} else if (relo->kind == BPF_FIELD_BYTE_SIZE) {
			sz = btf__resolve_size(spec->btf, acc->type_id);
			if (sz < 0)
				return -EINVAL;
			*val = sz;
		} else {
			pr_warn("prog '%s': relo %d at insn #%d can't be applied to array access\n",
4529 4530 4531 4532 4533 4534 4535 4536 4537 4538 4539 4540 4541 4542 4543 4544 4545 4546 4547 4548 4549 4550 4551 4552 4553 4554 4555 4556 4557 4558 4559
				bpf_program__title(prog, false),
				relo->kind, relo->insn_off / 8);
			return -EINVAL;
		}
		if (validate)
			*validate = true;
		return 0;
	}

	m = btf_members(t) + acc->idx;
	mt = skip_mods_and_typedefs(spec->btf, m->type, NULL);
	bit_off = spec->bit_offset;
	bit_sz = btf_member_bitfield_size(t, acc->idx);

	bitfield = bit_sz > 0;
	if (bitfield) {
		byte_sz = mt->size;
		byte_off = bit_off / 8 / byte_sz * byte_sz;
		/* figure out smallest int size necessary for bitfield load */
		while (bit_off + bit_sz - byte_off * 8 > byte_sz * 8) {
			if (byte_sz >= 8) {
				/* bitfield can't be read with 64-bit read */
				pr_warn("prog '%s': relo %d at insn #%d can't be satisfied for bitfield\n",
					bpf_program__title(prog, false),
					relo->kind, relo->insn_off / 8);
				return -E2BIG;
			}
			byte_sz *= 2;
			byte_off = bit_off / 8 / byte_sz * byte_sz;
		}
	} else {
4560 4561 4562 4563
		sz = btf__resolve_size(spec->btf, m->type);
		if (sz < 0)
			return -EINVAL;
		byte_sz = sz;
4564 4565 4566 4567 4568 4569 4570 4571 4572 4573 4574 4575 4576 4577 4578 4579 4580 4581 4582 4583 4584 4585 4586 4587 4588 4589 4590 4591 4592 4593 4594 4595 4596 4597 4598 4599 4600 4601 4602 4603 4604 4605 4606 4607 4608 4609 4610 4611
		byte_off = spec->bit_offset / 8;
		bit_sz = byte_sz * 8;
	}

	/* for bitfields, all the relocatable aspects are ambiguous and we
	 * might disagree with compiler, so turn off validation of expected
	 * value, except for signedness
	 */
	if (validate)
		*validate = !bitfield;

	switch (relo->kind) {
	case BPF_FIELD_BYTE_OFFSET:
		*val = byte_off;
		break;
	case BPF_FIELD_BYTE_SIZE:
		*val = byte_sz;
		break;
	case BPF_FIELD_SIGNED:
		/* enums will be assumed unsigned */
		*val = btf_is_enum(mt) ||
		       (btf_int_encoding(mt) & BTF_INT_SIGNED);
		if (validate)
			*validate = true; /* signedness is never ambiguous */
		break;
	case BPF_FIELD_LSHIFT_U64:
#if __BYTE_ORDER == __LITTLE_ENDIAN
		*val = 64 - (bit_off + bit_sz - byte_off  * 8);
#else
		*val = (8 - byte_sz) * 8 + (bit_off - byte_off * 8);
#endif
		break;
	case BPF_FIELD_RSHIFT_U64:
		*val = 64 - bit_sz;
		if (validate)
			*validate = true; /* right shift is never ambiguous */
		break;
	case BPF_FIELD_EXISTS:
	default:
		pr_warn("prog '%s': unknown relo %d at insn #%d\n",
			bpf_program__title(prog, false),
			relo->kind, relo->insn_off / 8);
		return -EINVAL;
	}

	return 0;
}

4612 4613
/*
 * Patch relocatable BPF instruction.
4614 4615 4616 4617 4618 4619 4620
 *
 * Patched value is determined by relocation kind and target specification.
 * For field existence relocation target spec will be NULL if field is not
 * found.
 * Expected insn->imm value is determined using relocation kind and local
 * spec, and is checked before patching instruction. If actual insn->imm value
 * is wrong, bail out with error.
4621 4622 4623 4624 4625
 *
 * Currently three kinds of BPF instructions are supported:
 * 1. rX = <imm> (assignment with immediate operand);
 * 2. rX += <imm> (arithmetic operations with immediate operand);
 */
4626 4627
static int bpf_core_reloc_insn(struct bpf_program *prog,
			       const struct bpf_field_reloc *relo,
4628
			       int relo_idx,
4629 4630
			       const struct bpf_core_spec *local_spec,
			       const struct bpf_core_spec *targ_spec)
4631
{
4632
	__u32 orig_val, new_val;
4633
	struct bpf_insn *insn;
4634
	bool validate = true;
4635
	int insn_idx, err;
4636 4637
	__u8 class;

4638 4639 4640
	if (relo->insn_off % sizeof(struct bpf_insn))
		return -EINVAL;
	insn_idx = relo->insn_off / sizeof(struct bpf_insn);
4641 4642
	insn = &prog->insns[insn_idx];
	class = BPF_CLASS(insn->code);
4643

4644
	if (relo->kind == BPF_FIELD_EXISTS) {
4645 4646
		orig_val = 1; /* can't generate EXISTS relo w/o local field */
		new_val = targ_spec ? 1 : 0;
4647
	} else if (!targ_spec) {
4648 4649 4650 4651 4652 4653 4654 4655 4656 4657 4658 4659
		pr_debug("prog '%s': relo #%d: substituting insn #%d w/ invalid insn\n",
			 bpf_program__title(prog, false), relo_idx, insn_idx);
		insn->code = BPF_JMP | BPF_CALL;
		insn->dst_reg = 0;
		insn->src_reg = 0;
		insn->off = 0;
		/* if this instruction is reachable (not a dead code),
		 * verifier will complain with the following message:
		 * invalid func unknown#195896080
		 */
		insn->imm = 195896080; /* => 0xbad2310 => "bad relo" */
		return 0;
4660 4661 4662 4663 4664 4665 4666 4667 4668
	} else {
		err = bpf_core_calc_field_relo(prog, relo, local_spec,
					       &orig_val, &validate);
		if (err)
			return err;
		err = bpf_core_calc_field_relo(prog, relo, targ_spec,
					       &new_val, NULL);
		if (err)
			return err;
4669
	}
4670

4671 4672 4673
	switch (class) {
	case BPF_ALU:
	case BPF_ALU64:
4674 4675
		if (BPF_SRC(insn->code) != BPF_K)
			return -EINVAL;
4676 4677 4678 4679
		if (validate && insn->imm != orig_val) {
			pr_warn("prog '%s': relo #%d: unexpected insn #%d (ALU/ALU64) value: got %u, exp %u -> %u\n",
				bpf_program__title(prog, false), relo_idx,
				insn_idx, insn->imm, orig_val, new_val);
4680
			return -EINVAL;
4681 4682
		}
		orig_val = insn->imm;
4683
		insn->imm = new_val;
4684 4685 4686
		pr_debug("prog '%s': relo #%d: patched insn #%d (ALU/ALU64) imm %u -> %u\n",
			 bpf_program__title(prog, false), relo_idx, insn_idx,
			 orig_val, new_val);
4687 4688 4689 4690
		break;
	case BPF_LDX:
	case BPF_ST:
	case BPF_STX:
4691 4692 4693 4694
		if (validate && insn->off != orig_val) {
			pr_warn("prog '%s': relo #%d: unexpected insn #%d (LD/LDX/ST/STX) value: got %u, exp %u -> %u\n",
				bpf_program__title(prog, false), relo_idx,
				insn_idx, insn->off, orig_val, new_val);
4695 4696 4697
			return -EINVAL;
		}
		if (new_val > SHRT_MAX) {
4698 4699 4700
			pr_warn("prog '%s': relo #%d: insn #%d (LDX/ST/STX) value too big: %u\n",
				bpf_program__title(prog, false), relo_idx,
				insn_idx, new_val);
4701 4702 4703 4704
			return -ERANGE;
		}
		orig_val = insn->off;
		insn->off = new_val;
4705 4706 4707
		pr_debug("prog '%s': relo #%d: patched insn #%d (LDX/ST/STX) off %u -> %u\n",
			 bpf_program__title(prog, false), relo_idx, insn_idx,
			 orig_val, new_val);
4708 4709
		break;
	default:
4710 4711
		pr_warn("prog '%s': relo #%d: trying to relocate unrecognized insn #%d, code:%x, src:%x, dst:%x, off:%x, imm:%x\n",
			bpf_program__title(prog, false), relo_idx,
4712 4713
			insn_idx, insn->code, insn->src_reg, insn->dst_reg,
			insn->off, insn->imm);
4714 4715
		return -EINVAL;
	}
4716

4717 4718 4719 4720 4721 4722 4723 4724 4725 4726 4727 4728 4729 4730 4731 4732 4733 4734 4735 4736 4737 4738 4739
	return 0;
}

/* Output spec definition in the format:
 * [<type-id>] (<type-name>) + <raw-spec> => <offset>@<spec>,
 * where <spec> is a C-syntax view of recorded field access, e.g.: x.a[3].b
 */
static void bpf_core_dump_spec(int level, const struct bpf_core_spec *spec)
{
	const struct btf_type *t;
	const char *s;
	__u32 type_id;
	int i;

	type_id = spec->spec[0].type_id;
	t = btf__type_by_id(spec->btf, type_id);
	s = btf__name_by_offset(spec->btf, t->name_off);
	libbpf_print(level, "[%u] %s + ", type_id, s);

	for (i = 0; i < spec->raw_len; i++)
		libbpf_print(level, "%d%s", spec->raw_spec[i],
			     i == spec->raw_len - 1 ? " => " : ":");

4740 4741
	libbpf_print(level, "%u.%u @ &x",
		     spec->bit_offset / 8, spec->bit_offset % 8);
4742 4743 4744 4745 4746 4747 4748 4749 4750 4751 4752 4753 4754 4755 4756 4757 4758 4759 4760 4761 4762 4763 4764 4765 4766 4767 4768 4769 4770 4771 4772 4773 4774 4775 4776 4777 4778 4779 4780 4781 4782 4783 4784 4785 4786 4787 4788 4789 4790 4791 4792 4793 4794 4795 4796 4797

	for (i = 0; i < spec->len; i++) {
		if (spec->spec[i].name)
			libbpf_print(level, ".%s", spec->spec[i].name);
		else
			libbpf_print(level, "[%u]", spec->spec[i].idx);
	}

}

static size_t bpf_core_hash_fn(const void *key, void *ctx)
{
	return (size_t)key;
}

static bool bpf_core_equal_fn(const void *k1, const void *k2, void *ctx)
{
	return k1 == k2;
}

static void *u32_as_hash_key(__u32 x)
{
	return (void *)(uintptr_t)x;
}

/*
 * CO-RE relocate single instruction.
 *
 * The outline and important points of the algorithm:
 * 1. For given local type, find corresponding candidate target types.
 *    Candidate type is a type with the same "essential" name, ignoring
 *    everything after last triple underscore (___). E.g., `sample`,
 *    `sample___flavor_one`, `sample___flavor_another_one`, are all candidates
 *    for each other. Names with triple underscore are referred to as
 *    "flavors" and are useful, among other things, to allow to
 *    specify/support incompatible variations of the same kernel struct, which
 *    might differ between different kernel versions and/or build
 *    configurations.
 *
 *    N.B. Struct "flavors" could be generated by bpftool's BTF-to-C
 *    converter, when deduplicated BTF of a kernel still contains more than
 *    one different types with the same name. In that case, ___2, ___3, etc
 *    are appended starting from second name conflict. But start flavors are
 *    also useful to be defined "locally", in BPF program, to extract same
 *    data from incompatible changes between different kernel
 *    versions/configurations. For instance, to handle field renames between
 *    kernel versions, one can use two flavors of the struct name with the
 *    same common name and use conditional relocations to extract that field,
 *    depending on target kernel version.
 * 2. For each candidate type, try to match local specification to this
 *    candidate target type. Matching involves finding corresponding
 *    high-level spec accessors, meaning that all named fields should match,
 *    as well as all array accesses should be within the actual bounds. Also,
 *    types should be compatible (see bpf_core_fields_are_compat for details).
 * 3. It is supported and expected that there might be multiple flavors
 *    matching the spec. As long as all the specs resolve to the same set of
4798
 *    offsets across all candidates, there is no error. If there is any
4799 4800 4801 4802 4803 4804 4805 4806 4807 4808 4809 4810 4811 4812
 *    ambiguity, CO-RE relocation will fail. This is necessary to accomodate
 *    imprefection of BTF deduplication, which can cause slight duplication of
 *    the same BTF type, if some directly or indirectly referenced (by
 *    pointer) type gets resolved to different actual types in different
 *    object files. If such situation occurs, deduplicated BTF will end up
 *    with two (or more) structurally identical types, which differ only in
 *    types they refer to through pointer. This should be OK in most cases and
 *    is not an error.
 * 4. Candidate types search is performed by linearly scanning through all
 *    types in target BTF. It is anticipated that this is overall more
 *    efficient memory-wise and not significantly worse (if not better)
 *    CPU-wise compared to prebuilding a map from all local type names to
 *    a list of candidate type names. It's also sped up by caching resolved
 *    list of matching candidates per each local "root" type ID, that has at
4813
 *    least one bpf_field_reloc associated with it. This list is shared
4814 4815 4816
 *    between multiple relocations for the same type ID and is updated as some
 *    of the candidates are pruned due to structural incompatibility.
 */
4817 4818
static int bpf_core_reloc_field(struct bpf_program *prog,
				 const struct bpf_field_reloc *relo,
4819 4820 4821 4822 4823 4824 4825 4826 4827 4828 4829 4830 4831 4832 4833 4834 4835 4836 4837 4838 4839 4840 4841 4842 4843 4844 4845 4846 4847 4848
				 int relo_idx,
				 const struct btf *local_btf,
				 const struct btf *targ_btf,
				 struct hashmap *cand_cache)
{
	const char *prog_name = bpf_program__title(prog, false);
	struct bpf_core_spec local_spec, cand_spec, targ_spec;
	const void *type_key = u32_as_hash_key(relo->type_id);
	const struct btf_type *local_type, *cand_type;
	const char *local_name, *cand_name;
	struct ids_vec *cand_ids;
	__u32 local_id, cand_id;
	const char *spec_str;
	int i, j, err;

	local_id = relo->type_id;
	local_type = btf__type_by_id(local_btf, local_id);
	if (!local_type)
		return -EINVAL;

	local_name = btf__name_by_offset(local_btf, local_type->name_off);
	if (str_is_empty(local_name))
		return -EINVAL;

	spec_str = btf__name_by_offset(local_btf, relo->access_str_off);
	if (str_is_empty(spec_str))
		return -EINVAL;

	err = bpf_core_spec_parse(local_btf, local_id, spec_str, &local_spec);
	if (err) {
4849 4850 4851
		pr_warn("prog '%s': relo #%d: parsing [%d] %s + %s failed: %d\n",
			prog_name, relo_idx, local_id, local_name, spec_str,
			err);
4852 4853 4854
		return -EINVAL;
	}

4855 4856
	pr_debug("prog '%s': relo #%d: kind %d, spec is ", prog_name, relo_idx,
		 relo->kind);
4857 4858 4859 4860 4861 4862
	bpf_core_dump_spec(LIBBPF_DEBUG, &local_spec);
	libbpf_print(LIBBPF_DEBUG, "\n");

	if (!hashmap__find(cand_cache, type_key, (void **)&cand_ids)) {
		cand_ids = bpf_core_find_cands(local_btf, local_id, targ_btf);
		if (IS_ERR(cand_ids)) {
4863 4864 4865
			pr_warn("prog '%s': relo #%d: target candidate search failed for [%d] %s: %ld",
				prog_name, relo_idx, local_id, local_name,
				PTR_ERR(cand_ids));
4866 4867 4868 4869 4870 4871 4872 4873 4874 4875 4876 4877 4878 4879 4880 4881 4882 4883 4884 4885 4886
			return PTR_ERR(cand_ids);
		}
		err = hashmap__set(cand_cache, type_key, cand_ids, NULL, NULL);
		if (err) {
			bpf_core_free_cands(cand_ids);
			return err;
		}
	}

	for (i = 0, j = 0; i < cand_ids->len; i++) {
		cand_id = cand_ids->data[i];
		cand_type = btf__type_by_id(targ_btf, cand_id);
		cand_name = btf__name_by_offset(targ_btf, cand_type->name_off);

		err = bpf_core_spec_match(&local_spec, targ_btf,
					  cand_id, &cand_spec);
		pr_debug("prog '%s': relo #%d: matching candidate #%d %s against spec ",
			 prog_name, relo_idx, i, cand_name);
		bpf_core_dump_spec(LIBBPF_DEBUG, &cand_spec);
		libbpf_print(LIBBPF_DEBUG, ": %d\n", err);
		if (err < 0) {
4887 4888
			pr_warn("prog '%s': relo #%d: matching error: %d\n",
				prog_name, relo_idx, err);
4889 4890 4891 4892 4893 4894 4895
			return err;
		}
		if (err == 0)
			continue;

		if (j == 0) {
			targ_spec = cand_spec;
4896
		} else if (cand_spec.bit_offset != targ_spec.bit_offset) {
4897
			/* if there are many candidates, they should all
4898
			 * resolve to the same bit offset
4899
			 */
4900
			pr_warn("prog '%s': relo #%d: offset ambiguity: %u != %u\n",
4901 4902
				prog_name, relo_idx, cand_spec.bit_offset,
				targ_spec.bit_offset);
4903 4904 4905 4906 4907 4908
			return -EINVAL;
		}

		cand_ids->data[j++] = cand_spec.spec[0].type_id;
	}

4909
	/*
4910 4911 4912 4913 4914 4915
	 * For BPF_FIELD_EXISTS relo or when used BPF program has field
	 * existence checks or kernel version/config checks, it's expected
	 * that we might not find any candidates. In this case, if field
	 * wasn't found in any candidate, the list of candidates shouldn't
	 * change at all, we'll just handle relocating appropriately,
	 * depending on relo's kind.
4916 4917 4918 4919
	 */
	if (j > 0)
		cand_ids->len = j;

4920 4921 4922 4923 4924 4925 4926 4927 4928 4929 4930 4931 4932 4933
	/*
	 * If no candidates were found, it might be both a programmer error,
	 * as well as expected case, depending whether instruction w/
	 * relocation is guarded in some way that makes it unreachable (dead
	 * code) if relocation can't be resolved. This is handled in
	 * bpf_core_reloc_insn() uniformly by replacing that instruction with
	 * BPF helper call insn (using invalid helper ID). If that instruction
	 * is indeed unreachable, then it will be ignored and eliminated by
	 * verifier. If it was an error, then verifier will complain and point
	 * to a specific instruction number in its log.
	 */
	if (j == 0)
		pr_debug("prog '%s': relo #%d: no matching targets found for [%d] %s + %s\n",
			 prog_name, relo_idx, local_id, local_name, spec_str);
4934

4935
	/* bpf_core_reloc_insn should know how to handle missing targ_spec */
4936
	err = bpf_core_reloc_insn(prog, relo, relo_idx, &local_spec,
4937
				  j ? &targ_spec : NULL);
4938
	if (err) {
4939 4940
		pr_warn("prog '%s': relo #%d: failed to patch insn at offset %d: %d\n",
			prog_name, relo_idx, relo->insn_off, err);
4941 4942 4943 4944 4945 4946 4947
		return -EINVAL;
	}

	return 0;
}

static int
4948
bpf_core_reloc_fields(struct bpf_object *obj, const char *targ_btf_path)
4949 4950
{
	const struct btf_ext_info_sec *sec;
4951
	const struct bpf_field_reloc *rec;
4952 4953 4954 4955 4956 4957 4958 4959 4960 4961 4962
	const struct btf_ext_info *seg;
	struct hashmap_entry *entry;
	struct hashmap *cand_cache = NULL;
	struct bpf_program *prog;
	struct btf *targ_btf;
	const char *sec_name;
	int i, err = 0;

	if (targ_btf_path)
		targ_btf = btf__parse_elf(targ_btf_path, NULL);
	else
4963 4964
		targ_btf = obj->btf_vmlinux;
	if (IS_ERR_OR_NULL(targ_btf)) {
4965
		pr_warn("failed to get target BTF: %ld\n", PTR_ERR(targ_btf));
4966 4967 4968 4969 4970 4971 4972 4973 4974
		return PTR_ERR(targ_btf);
	}

	cand_cache = hashmap__new(bpf_core_hash_fn, bpf_core_equal_fn, NULL);
	if (IS_ERR(cand_cache)) {
		err = PTR_ERR(cand_cache);
		goto out;
	}

4975
	seg = &obj->btf_ext->field_reloc_info;
4976 4977 4978 4979 4980 4981
	for_each_btf_ext_sec(seg, sec) {
		sec_name = btf__name_by_offset(obj->btf, sec->sec_name_off);
		if (str_is_empty(sec_name)) {
			err = -EINVAL;
			goto out;
		}
4982 4983 4984 4985 4986 4987 4988
		prog = NULL;
		for (i = 0; i < obj->nr_programs; i++) {
			if (!strcmp(obj->programs[i].section_name, sec_name)) {
				prog = &obj->programs[i];
				break;
			}
		}
4989
		if (!prog) {
4990 4991
			pr_warn("failed to find program '%s' for CO-RE offset relocation\n",
				sec_name);
4992 4993 4994 4995 4996 4997 4998 4999
			err = -EINVAL;
			goto out;
		}

		pr_debug("prog '%s': performing %d CO-RE offset relocs\n",
			 sec_name, sec->num_info);

		for_each_btf_ext_rec(seg, sec, i, rec) {
5000 5001
			err = bpf_core_reloc_field(prog, rec, i, obj->btf,
						   targ_btf, cand_cache);
5002
			if (err) {
5003 5004
				pr_warn("prog '%s': relo #%d: failed to relocate: %d\n",
					sec_name, i, err);
5005 5006 5007 5008 5009 5010
				goto out;
			}
		}
	}

out:
5011 5012 5013
	/* obj->btf_vmlinux is freed at the end of object load phase */
	if (targ_btf != obj->btf_vmlinux)
		btf__free(targ_btf);
5014 5015 5016 5017 5018 5019 5020 5021 5022 5023 5024 5025 5026 5027
	if (!IS_ERR_OR_NULL(cand_cache)) {
		hashmap__for_each_entry(cand_cache, entry, i) {
			bpf_core_free_cands(entry->value);
		}
		hashmap__free(cand_cache);
	}
	return err;
}

static int
bpf_object__relocate_core(struct bpf_object *obj, const char *targ_btf_path)
{
	int err = 0;

5028 5029
	if (obj->btf_ext->field_reloc_info.len)
		err = bpf_core_reloc_fields(obj, targ_btf_path);
5030 5031 5032 5033

	return err;
}

5034 5035 5036 5037 5038 5039 5040
static int
bpf_program__reloc_text(struct bpf_program *prog, struct bpf_object *obj,
			struct reloc_desc *relo)
{
	struct bpf_insn *insn, *new_insn;
	struct bpf_program *text;
	size_t new_cnt;
5041
	int err;
5042

5043
	if (prog->idx != obj->efile.text_shndx && prog->main_prog_cnt == 0) {
5044 5045
		text = bpf_object__find_prog_by_idx(obj, obj->efile.text_shndx);
		if (!text) {
5046
			pr_warn("no .text section found yet relo into text exist\n");
5047 5048 5049
			return -LIBBPF_ERRNO__RELOC;
		}
		new_cnt = prog->insns_cnt + text->insns_cnt;
5050
		new_insn = reallocarray(prog->insns, new_cnt, sizeof(*insn));
5051
		if (!new_insn) {
5052
			pr_warn("oom in prog realloc\n");
5053 5054
			return -ENOMEM;
		}
5055
		prog->insns = new_insn;
5056

5057 5058 5059 5060 5061
		if (obj->btf_ext) {
			err = bpf_program_reloc_btf_ext(prog, obj,
							text->section_name,
							prog->insns_cnt);
			if (err)
5062 5063 5064
				return err;
		}

5065 5066 5067 5068
		memcpy(new_insn + prog->insns_cnt, text->insns,
		       text->insns_cnt * sizeof(*insn));
		prog->main_prog_cnt = prog->insns_cnt;
		prog->insns_cnt = new_cnt;
5069 5070 5071
		pr_debug("added %zd insn from %s to prog %s\n",
			 text->insns_cnt, text->section_name,
			 prog->section_name);
5072
	}
5073

5074
	insn = &prog->insns[relo->insn_idx];
5075
	insn->imm += relo->sym_off / 8 + prog->main_prog_cnt - relo->insn_idx;
5076 5077 5078
	return 0;
}

W
Wang Nan 已提交
5079
static int
5080
bpf_program__relocate(struct bpf_program *prog, struct bpf_object *obj)
W
Wang Nan 已提交
5081
{
5082
	int i, err;
W
Wang Nan 已提交
5083

5084 5085 5086
	if (!prog)
		return 0;

5087 5088 5089 5090
	if (obj->btf_ext) {
		err = bpf_program_reloc_btf_ext(prog, obj,
						prog->section_name, 0);
		if (err)
5091 5092 5093 5094
			return err;
	}

	if (!prog->reloc_desc)
W
Wang Nan 已提交
5095 5096 5097
		return 0;

	for (i = 0; i < prog->nr_reloc; i++) {
5098
		struct reloc_desc *relo = &prog->reloc_desc[i];
5099
		struct bpf_insn *insn = &prog->insns[relo->insn_idx];
5100
		struct extern_desc *ext;
W
Wang Nan 已提交
5101

5102 5103 5104 5105 5106
		if (relo->insn_idx + 1 >= (int)prog->insns_cnt) {
			pr_warn("relocation out of range: '%s'\n",
				prog->section_name);
			return -LIBBPF_ERRNO__RELOC;
		}
5107

5108 5109 5110 5111 5112 5113 5114 5115
		switch (relo->type) {
		case RELO_LD64:
			insn[0].src_reg = BPF_PSEUDO_MAP_FD;
			insn[0].imm = obj->maps[relo->map_idx].fd;
			break;
		case RELO_DATA:
			insn[0].src_reg = BPF_PSEUDO_MAP_VALUE;
			insn[1].imm = insn[0].imm + relo->sym_off;
5116
			insn[0].imm = obj->maps[relo->map_idx].fd;
5117 5118
			break;
		case RELO_EXTERN:
5119
			ext = &obj->externs[relo->sym_off];
5120 5121 5122 5123 5124 5125 5126 5127
			if (ext->type == EXT_KCFG) {
				insn[0].src_reg = BPF_PSEUDO_MAP_VALUE;
				insn[0].imm = obj->maps[obj->kconfig_map_idx].fd;
				insn[1].imm = ext->kcfg.data_off;
			} else /* EXT_KSYM */ {
				insn[0].imm = (__u32)ext->ksym.addr;
				insn[1].imm = ext->ksym.addr >> 32;
			}
5128 5129
			break;
		case RELO_CALL:
5130
			err = bpf_program__reloc_text(prog, obj, relo);
5131 5132
			if (err)
				return err;
5133 5134 5135 5136
			break;
		default:
			pr_warn("relo #%d: bad relo type %d\n", i, relo->type);
			return -EINVAL;
W
Wang Nan 已提交
5137 5138 5139 5140 5141 5142 5143 5144 5145
		}
	}

	zfree(&prog->reloc_desc);
	prog->nr_reloc = 0;
	return 0;
}

static int
5146
bpf_object__relocate(struct bpf_object *obj, const char *targ_btf_path)
W
Wang Nan 已提交
5147 5148 5149 5150 5151
{
	struct bpf_program *prog;
	size_t i;
	int err;

5152 5153 5154
	if (obj->btf_ext) {
		err = bpf_object__relocate_core(obj, targ_btf_path);
		if (err) {
5155 5156
			pr_warn("failed to perform CO-RE relocations: %d\n",
				err);
5157 5158 5159
			return err;
		}
	}
5160 5161 5162 5163 5164 5165 5166 5167 5168 5169 5170 5171 5172 5173 5174 5175 5176 5177
	/* ensure .text is relocated first, as it's going to be copied as-is
	 * later for sub-program calls
	 */
	for (i = 0; i < obj->nr_programs; i++) {
		prog = &obj->programs[i];
		if (prog->idx != obj->efile.text_shndx)
			continue;

		err = bpf_program__relocate(prog, obj);
		if (err) {
			pr_warn("failed to relocate '%s'\n", prog->section_name);
			return err;
		}
		break;
	}
	/* now relocate everything but .text, which by now is relocated
	 * properly, so we can copy raw sub-program instructions as is safely
	 */
W
Wang Nan 已提交
5178 5179
	for (i = 0; i < obj->nr_programs; i++) {
		prog = &obj->programs[i];
5180 5181
		if (prog->idx == obj->efile.text_shndx)
			continue;
W
Wang Nan 已提交
5182

5183
		err = bpf_program__relocate(prog, obj);
W
Wang Nan 已提交
5184
		if (err) {
5185
			pr_warn("failed to relocate '%s'\n", prog->section_name);
W
Wang Nan 已提交
5186 5187 5188 5189 5190 5191
			return err;
		}
	}
	return 0;
}

5192 5193 5194 5195 5196 5197 5198
static int bpf_object__collect_st_ops_relos(struct bpf_object *obj,
					    GElf_Shdr *shdr, Elf_Data *data);

static int bpf_object__collect_map_relos(struct bpf_object *obj,
					 GElf_Shdr *shdr, Elf_Data *data)
{
	int i, j, nrels, new_sz, ptr_sz = sizeof(void *);
5199
	const struct btf_var_secinfo *vi = NULL;
5200 5201 5202 5203 5204 5205 5206 5207 5208 5209 5210 5211 5212 5213 5214 5215 5216 5217 5218 5219 5220 5221 5222 5223 5224 5225 5226 5227 5228 5229 5230 5231 5232 5233 5234 5235 5236 5237 5238 5239 5240 5241 5242 5243 5244 5245 5246 5247 5248 5249 5250 5251 5252 5253 5254 5255 5256 5257 5258 5259 5260 5261 5262 5263 5264 5265 5266 5267 5268 5269 5270 5271 5272 5273 5274 5275 5276 5277 5278 5279 5280 5281 5282 5283 5284 5285 5286 5287 5288 5289 5290 5291 5292 5293 5294 5295 5296 5297 5298 5299 5300 5301 5302 5303
	const struct btf_type *sec, *var, *def;
	const struct btf_member *member;
	struct bpf_map *map, *targ_map;
	const char *name, *mname;
	Elf_Data *symbols;
	unsigned int moff;
	GElf_Sym sym;
	GElf_Rel rel;
	void *tmp;

	if (!obj->efile.btf_maps_sec_btf_id || !obj->btf)
		return -EINVAL;
	sec = btf__type_by_id(obj->btf, obj->efile.btf_maps_sec_btf_id);
	if (!sec)
		return -EINVAL;

	symbols = obj->efile.symbols;
	nrels = shdr->sh_size / shdr->sh_entsize;
	for (i = 0; i < nrels; i++) {
		if (!gelf_getrel(data, i, &rel)) {
			pr_warn(".maps relo #%d: failed to get ELF relo\n", i);
			return -LIBBPF_ERRNO__FORMAT;
		}
		if (!gelf_getsym(symbols, GELF_R_SYM(rel.r_info), &sym)) {
			pr_warn(".maps relo #%d: symbol %zx not found\n",
				i, (size_t)GELF_R_SYM(rel.r_info));
			return -LIBBPF_ERRNO__FORMAT;
		}
		name = elf_strptr(obj->efile.elf, obj->efile.strtabidx,
				  sym.st_name) ? : "<?>";
		if (sym.st_shndx != obj->efile.btf_maps_shndx) {
			pr_warn(".maps relo #%d: '%s' isn't a BTF-defined map\n",
				i, name);
			return -LIBBPF_ERRNO__RELOC;
		}

		pr_debug(".maps relo #%d: for %zd value %zd rel.r_offset %zu name %d ('%s')\n",
			 i, (ssize_t)(rel.r_info >> 32), (size_t)sym.st_value,
			 (size_t)rel.r_offset, sym.st_name, name);

		for (j = 0; j < obj->nr_maps; j++) {
			map = &obj->maps[j];
			if (map->sec_idx != obj->efile.btf_maps_shndx)
				continue;

			vi = btf_var_secinfos(sec) + map->btf_var_idx;
			if (vi->offset <= rel.r_offset &&
			    rel.r_offset + sizeof(void *) <= vi->offset + vi->size)
				break;
		}
		if (j == obj->nr_maps) {
			pr_warn(".maps relo #%d: cannot find map '%s' at rel.r_offset %zu\n",
				i, name, (size_t)rel.r_offset);
			return -EINVAL;
		}

		if (!bpf_map_type__is_map_in_map(map->def.type))
			return -EINVAL;
		if (map->def.type == BPF_MAP_TYPE_HASH_OF_MAPS &&
		    map->def.key_size != sizeof(int)) {
			pr_warn(".maps relo #%d: hash-of-maps '%s' should have key size %zu.\n",
				i, map->name, sizeof(int));
			return -EINVAL;
		}

		targ_map = bpf_object__find_map_by_name(obj, name);
		if (!targ_map)
			return -ESRCH;

		var = btf__type_by_id(obj->btf, vi->type);
		def = skip_mods_and_typedefs(obj->btf, var->type, NULL);
		if (btf_vlen(def) == 0)
			return -EINVAL;
		member = btf_members(def) + btf_vlen(def) - 1;
		mname = btf__name_by_offset(obj->btf, member->name_off);
		if (strcmp(mname, "values"))
			return -EINVAL;

		moff = btf_member_bit_offset(def, btf_vlen(def) - 1) / 8;
		if (rel.r_offset - vi->offset < moff)
			return -EINVAL;

		moff = rel.r_offset - vi->offset - moff;
		if (moff % ptr_sz)
			return -EINVAL;
		moff /= ptr_sz;
		if (moff >= map->init_slots_sz) {
			new_sz = moff + 1;
			tmp = realloc(map->init_slots, new_sz * ptr_sz);
			if (!tmp)
				return -ENOMEM;
			map->init_slots = tmp;
			memset(map->init_slots + map->init_slots_sz, 0,
			       (new_sz - map->init_slots_sz) * ptr_sz);
			map->init_slots_sz = new_sz;
		}
		map->init_slots[moff] = targ_map;

		pr_debug(".maps relo #%d: map '%s' slot [%d] points to map '%s'\n",
			 i, map->name, moff, name);
	}

	return 0;
}
5304

5305 5306 5307 5308 5309
static int bpf_object__collect_reloc(struct bpf_object *obj)
{
	int i, err;

	if (!obj_elf_valid(obj)) {
5310
		pr_warn("Internal error: elf object is closed\n");
5311
		return -LIBBPF_ERRNO__INTERNAL;
5312 5313
	}

5314 5315 5316
	for (i = 0; i < obj->efile.nr_reloc_sects; i++) {
		GElf_Shdr *shdr = &obj->efile.reloc_sects[i].shdr;
		Elf_Data *data = obj->efile.reloc_sects[i].data;
5317 5318 5319 5320
		int idx = shdr->sh_info;
		struct bpf_program *prog;

		if (shdr->sh_type != SHT_REL) {
5321
			pr_warn("internal error at %d\n", __LINE__);
5322
			return -LIBBPF_ERRNO__INTERNAL;
5323 5324
		}

5325
		if (idx == obj->efile.st_ops_shndx) {
5326 5327 5328 5329 5330 5331 5332 5333 5334 5335
			err = bpf_object__collect_st_ops_relos(obj, shdr, data);
		} else if (idx == obj->efile.btf_maps_shndx) {
			err = bpf_object__collect_map_relos(obj, shdr, data);
		} else {
			prog = bpf_object__find_prog_by_idx(obj, idx);
			if (!prog) {
				pr_warn("relocation failed: no prog in section(%d)\n", idx);
				return -LIBBPF_ERRNO__RELOC;
			}
			err = bpf_program__collect_reloc(prog, shdr, data, obj);
5336 5337
		}
		if (err)
5338
			return err;
5339 5340 5341 5342
	}
	return 0;
}

5343
static int
5344
load_program(struct bpf_program *prog, struct bpf_insn *insns, int insns_cnt,
5345
	     char *license, __u32 kern_version, int *pfd)
5346
{
5347
	struct bpf_load_program_attr load_attr;
5348
	char *cp, errmsg[STRERR_BUFSIZE];
5349 5350
	size_t log_buf_size = 0;
	char *log_buf = NULL;
5351
	int btf_fd, ret;
5352

5353 5354 5355
	if (!insns || !insns_cnt)
		return -EINVAL;

5356
	memset(&load_attr, 0, sizeof(struct bpf_load_program_attr));
5357
	load_attr.prog_type = prog->type;
5358 5359 5360 5361 5362 5363
	/* old kernels might not support specifying expected_attach_type */
	if (!prog->caps->exp_attach_type && prog->sec_def &&
	    prog->sec_def->is_exp_attach_type_optional)
		load_attr.expected_attach_type = 0;
	else
		load_attr.expected_attach_type = prog->expected_attach_type;
5364 5365
	if (prog->caps->name)
		load_attr.name = prog->name;
5366 5367 5368
	load_attr.insns = insns;
	load_attr.insns_cnt = insns_cnt;
	load_attr.license = license;
5369 5370
	if (prog->type == BPF_PROG_TYPE_STRUCT_OPS ||
	    prog->type == BPF_PROG_TYPE_LSM) {
5371
		load_attr.attach_btf_id = prog->attach_btf_id;
5372 5373
	} else if (prog->type == BPF_PROG_TYPE_TRACING ||
		   prog->type == BPF_PROG_TYPE_EXT) {
5374 5375 5376 5377 5378 5379
		load_attr.attach_prog_fd = prog->attach_prog_fd;
		load_attr.attach_btf_id = prog->attach_btf_id;
	} else {
		load_attr.kern_version = kern_version;
		load_attr.prog_ifindex = prog->prog_ifindex;
	}
5380 5381 5382 5383 5384 5385 5386 5387 5388 5389 5390
	/* specify func_info/line_info only if kernel supports them */
	btf_fd = bpf_object__btf_fd(prog->obj);
	if (btf_fd >= 0 && prog->obj->caps.btf_func) {
		load_attr.prog_btf_fd = btf_fd;
		load_attr.func_info = prog->func_info;
		load_attr.func_info_rec_size = prog->func_info_rec_size;
		load_attr.func_info_cnt = prog->func_info_cnt;
		load_attr.line_info = prog->line_info;
		load_attr.line_info_rec_size = prog->line_info_rec_size;
		load_attr.line_info_cnt = prog->line_info_cnt;
	}
5391
	load_attr.log_level = prog->log_level;
5392
	load_attr.prog_flags = prog->prog_flags;
5393

5394
retry_load:
5395 5396 5397 5398 5399 5400 5401
	if (log_buf_size) {
		log_buf = malloc(log_buf_size);
		if (!log_buf)
			return -ENOMEM;

		*log_buf = 0;
	}
5402

5403
	ret = bpf_load_program_xattr(&load_attr, log_buf, log_buf_size);
5404 5405

	if (ret >= 0) {
5406
		if (log_buf && load_attr.log_level)
5407
			pr_debug("verifier log:\n%s", log_buf);
5408 5409 5410 5411 5412
		*pfd = ret;
		ret = 0;
		goto out;
	}

5413 5414 5415 5416
	if (!log_buf || errno == ENOSPC) {
		log_buf_size = max((size_t)BPF_LOG_BUF_SIZE,
				   log_buf_size << 1);

5417 5418 5419
		free(log_buf);
		goto retry_load;
	}
5420
	ret = -errno;
5421
	cp = libbpf_strerror_r(errno, errmsg, sizeof(errmsg));
5422
	pr_warn("load bpf program failed: %s\n", cp);
5423
	pr_perm_msg(ret);
5424

5425 5426
	if (log_buf && log_buf[0] != '\0') {
		ret = -LIBBPF_ERRNO__VERIFY;
5427 5428 5429
		pr_warn("-- BEGIN DUMP LOG ---\n");
		pr_warn("\n%s\n", log_buf);
		pr_warn("-- END LOG --\n");
5430
	} else if (load_attr.insns_cnt >= BPF_MAXINSNS) {
5431 5432
		pr_warn("Program too large (%zu insns), at most %d insns\n",
			load_attr.insns_cnt, BPF_MAXINSNS);
5433
		ret = -LIBBPF_ERRNO__PROG2BIG;
5434
	} else if (load_attr.prog_type != BPF_PROG_TYPE_KPROBE) {
5435
		/* Wrong program type? */
5436
		int fd;
5437

5438 5439 5440 5441 5442 5443 5444 5445
		load_attr.prog_type = BPF_PROG_TYPE_KPROBE;
		load_attr.expected_attach_type = 0;
		fd = bpf_load_program_xattr(&load_attr, NULL, 0);
		if (fd >= 0) {
			close(fd);
			ret = -LIBBPF_ERRNO__PROGTYPE;
			goto out;
		}
5446 5447 5448 5449 5450 5451 5452
	}

out:
	free(log_buf);
	return ret;
}

5453
static int libbpf_find_attach_btf_id(struct bpf_program *prog);
5454 5455

int bpf_program__load(struct bpf_program *prog, char *license, __u32 kern_ver)
5456
{
5457 5458
	int err = 0, fd, i, btf_id;

5459 5460 5461 5462 5463 5464
	if (prog->obj->loaded) {
		pr_warn("prog '%s'('%s'): can't load after object was loaded\n",
			prog->name, prog->section_name);
		return -EINVAL;
	}

5465
	if ((prog->type == BPF_PROG_TYPE_TRACING ||
5466
	     prog->type == BPF_PROG_TYPE_LSM ||
5467
	     prog->type == BPF_PROG_TYPE_EXT) && !prog->attach_btf_id) {
5468
		btf_id = libbpf_find_attach_btf_id(prog);
5469 5470 5471 5472
		if (btf_id <= 0)
			return btf_id;
		prog->attach_btf_id = btf_id;
	}
5473

5474 5475
	if (prog->instances.nr < 0 || !prog->instances.fds) {
		if (prog->preprocessor) {
5476 5477
			pr_warn("Internal error: can't load program '%s'\n",
				prog->section_name);
5478 5479
			return -LIBBPF_ERRNO__INTERNAL;
		}
5480

5481 5482
		prog->instances.fds = malloc(sizeof(int));
		if (!prog->instances.fds) {
5483
			pr_warn("Not enough memory for BPF fds\n");
5484 5485 5486 5487 5488 5489 5490 5491
			return -ENOMEM;
		}
		prog->instances.nr = 1;
		prog->instances.fds[0] = -1;
	}

	if (!prog->preprocessor) {
		if (prog->instances.nr != 1) {
5492 5493
			pr_warn("Program '%s' is inconsistent: nr(%d) != 1\n",
				prog->section_name, prog->instances.nr);
5494
		}
5495
		err = load_program(prog, prog->insns, prog->insns_cnt,
5496
				   license, kern_ver, &fd);
5497 5498 5499 5500 5501 5502 5503 5504 5505
		if (!err)
			prog->instances.fds[0] = fd;
		goto out;
	}

	for (i = 0; i < prog->instances.nr; i++) {
		struct bpf_prog_prep_result result;
		bpf_program_prep_t preprocessor = prog->preprocessor;

5506
		memset(&result, 0, sizeof(result));
5507 5508 5509
		err = preprocessor(prog, i, prog->insns,
				   prog->insns_cnt, &result);
		if (err) {
5510 5511
			pr_warn("Preprocessing the %dth instance of program '%s' failed\n",
				i, prog->section_name);
5512 5513 5514 5515 5516 5517 5518 5519 5520 5521 5522 5523
			goto out;
		}

		if (!result.new_insn_ptr || !result.new_insn_cnt) {
			pr_debug("Skip loading the %dth instance of program '%s'\n",
				 i, prog->section_name);
			prog->instances.fds[i] = -1;
			if (result.pfd)
				*result.pfd = -1;
			continue;
		}

5524
		err = load_program(prog, result.new_insn_ptr,
5525
				   result.new_insn_cnt, license, kern_ver, &fd);
5526
		if (err) {
5527 5528
			pr_warn("Loading the %dth instance of program '%s' failed\n",
				i, prog->section_name);
5529 5530 5531 5532 5533 5534 5535 5536
			goto out;
		}

		if (result.pfd)
			*result.pfd = fd;
		prog->instances.fds[i] = fd;
	}
out:
5537
	if (err)
5538
		pr_warn("failed to load program '%s'\n", prog->section_name);
5539 5540 5541 5542 5543
	zfree(&prog->insns);
	prog->insns_cnt = 0;
	return err;
}

A
Andrii Nakryiko 已提交
5544 5545
static bool bpf_program__is_function_storage(const struct bpf_program *prog,
					     const struct bpf_object *obj)
5546 5547 5548 5549
{
	return prog->idx == obj->efile.text_shndx && obj->has_pseudo_calls;
}

5550
static int
5551
bpf_object__load_progs(struct bpf_object *obj, int log_level)
5552
{
5553
	struct bpf_program *prog;
5554 5555 5556 5557
	size_t i;
	int err;

	for (i = 0; i < obj->nr_programs; i++) {
5558 5559
		prog = &obj->programs[i];
		if (bpf_program__is_function_storage(prog, obj))
5560
			continue;
5561 5562 5563 5564 5565 5566 5567
		if (!prog->load) {
			pr_debug("prog '%s'('%s'): skipped loading\n",
				 prog->name, prog->section_name);
			continue;
		}
		prog->log_level |= log_level;
		err = bpf_program__load(prog, obj->license, obj->kern_version);
5568 5569 5570 5571 5572 5573
		if (err)
			return err;
	}
	return 0;
}

5574 5575
static const struct bpf_sec_def *find_sec_def(const char *sec_name);

5576
static struct bpf_object *
5577
__bpf_object__open(const char *path, const void *obj_buf, size_t obj_buf_sz,
5578
		   const struct bpf_object_open_opts *opts)
5579
{
5580
	const char *obj_name, *kconfig;
5581
	struct bpf_program *prog;
5582
	struct bpf_object *obj;
5583
	char tmp_name[64];
5584
	int err;
5585 5586

	if (elf_version(EV_CURRENT) == EV_NONE) {
5587 5588
		pr_warn("failed to init libelf for %s\n",
			path ? : "(mem buf)");
5589
		return ERR_PTR(-LIBBPF_ERRNO__LIBELF);
5590 5591
	}

5592 5593 5594
	if (!OPTS_VALID(opts, bpf_object_open_opts))
		return ERR_PTR(-EINVAL);

5595
	obj_name = OPTS_GET(opts, object_name, NULL);
5596 5597 5598 5599 5600 5601 5602 5603 5604 5605 5606
	if (obj_buf) {
		if (!obj_name) {
			snprintf(tmp_name, sizeof(tmp_name), "%lx-%lx",
				 (unsigned long)obj_buf,
				 (unsigned long)obj_buf_sz);
			obj_name = tmp_name;
		}
		path = obj_name;
		pr_debug("loading object '%s' from buffer\n", obj_name);
	}

5607
	obj = bpf_object__new(path, obj_buf, obj_buf_sz, obj_name);
5608 5609
	if (IS_ERR(obj))
		return obj;
5610

5611 5612 5613 5614
	kconfig = OPTS_GET(opts, kconfig, NULL);
	if (kconfig) {
		obj->kconfig = strdup(kconfig);
		if (!obj->kconfig)
5615 5616
			return ERR_PTR(-ENOMEM);
	}
5617

5618 5619 5620
	err = bpf_object__elf_init(obj);
	err = err ? : bpf_object__check_endianness(obj);
	err = err ? : bpf_object__elf_collect(obj);
5621 5622
	err = err ? : bpf_object__collect_externs(obj);
	err = err ? : bpf_object__finalize_btf(obj);
5623 5624 5625 5626 5627
	err = err ? : bpf_object__init_maps(obj, opts);
	err = err ? : bpf_object__init_prog_names(obj);
	err = err ? : bpf_object__collect_reloc(obj);
	if (err)
		goto out;
5628
	bpf_object__elf_finish(obj);
5629 5630

	bpf_object__for_each_program(prog, obj) {
5631 5632
		prog->sec_def = find_sec_def(prog->section_name);
		if (!prog->sec_def)
5633 5634 5635
			/* couldn't guess, but user might manually specify */
			continue;

5636 5637 5638 5639 5640 5641
		bpf_program__set_type(prog, prog->sec_def->prog_type);
		bpf_program__set_expected_attach_type(prog,
				prog->sec_def->expected_attach_type);

		if (prog->sec_def->prog_type == BPF_PROG_TYPE_TRACING ||
		    prog->sec_def->prog_type == BPF_PROG_TYPE_EXT)
5642
			prog->attach_prog_fd = OPTS_GET(opts, attach_prog_fd, 0);
5643 5644
	}

5645 5646 5647
	return obj;
out:
	bpf_object__close(obj);
5648
	return ERR_PTR(err);
5649 5650
}

5651 5652
static struct bpf_object *
__bpf_object__open_xattr(struct bpf_object_open_attr *attr, int flags)
5653
{
5654
	DECLARE_LIBBPF_OPTS(bpf_object_open_opts, opts,
5655 5656 5657
		.relaxed_maps = flags & MAPS_RELAX_COMPAT,
	);

5658
	/* param validation */
5659
	if (!attr->file)
5660 5661
		return NULL;

5662
	pr_debug("loading %s\n", attr->file);
5663
	return __bpf_object__open(attr->file, NULL, 0, &opts);
5664 5665 5666 5667 5668
}

struct bpf_object *bpf_object__open_xattr(struct bpf_object_open_attr *attr)
{
	return __bpf_object__open_xattr(attr, 0);
5669 5670 5671 5672 5673 5674 5675 5676
}

struct bpf_object *bpf_object__open(const char *path)
{
	struct bpf_object_open_attr attr = {
		.file		= path,
		.prog_type	= BPF_PROG_TYPE_UNSPEC,
	};
5677

5678
	return bpf_object__open_xattr(&attr);
5679 5680
}

5681
struct bpf_object *
5682
bpf_object__open_file(const char *path, const struct bpf_object_open_opts *opts)
5683 5684 5685 5686 5687 5688
{
	if (!path)
		return ERR_PTR(-EINVAL);

	pr_debug("loading %s\n", path);

5689
	return __bpf_object__open(path, NULL, 0, opts);
5690 5691 5692 5693
}

struct bpf_object *
bpf_object__open_mem(const void *obj_buf, size_t obj_buf_sz,
5694
		     const struct bpf_object_open_opts *opts)
5695
{
5696 5697
	if (!obj_buf || obj_buf_sz == 0)
		return ERR_PTR(-EINVAL);
5698

5699
	return __bpf_object__open(NULL, obj_buf, obj_buf_sz, opts);
5700 5701 5702 5703 5704 5705
}

struct bpf_object *
bpf_object__open_buffer(const void *obj_buf, size_t obj_buf_sz,
			const char *name)
{
5706
	DECLARE_LIBBPF_OPTS(bpf_object_open_opts, opts,
5707 5708 5709 5710 5711 5712 5713 5714
		.object_name = name,
		/* wrong default, but backwards-compatible */
		.relaxed_maps = true,
	);

	/* returning NULL is wrong, but backwards-compatible */
	if (!obj_buf || obj_buf_sz == 0)
		return NULL;
5715

5716
	return bpf_object__open_mem(obj_buf, obj_buf_sz, &opts);
5717 5718
}

5719 5720 5721 5722 5723 5724 5725
int bpf_object__unload(struct bpf_object *obj)
{
	size_t i;

	if (!obj)
		return -EINVAL;

5726
	for (i = 0; i < obj->nr_maps; i++) {
5727
		zclose(obj->maps[i].fd);
5728 5729 5730
		if (obj->maps[i].st_ops)
			zfree(&obj->maps[i].st_ops->kern_vdata);
	}
5731

5732 5733 5734
	for (i = 0; i < obj->nr_programs; i++)
		bpf_program__unload(&obj->programs[i]);

5735 5736 5737
	return 0;
}

5738 5739 5740 5741 5742 5743 5744 5745 5746 5747 5748 5749 5750 5751 5752 5753 5754 5755
static int bpf_object__sanitize_maps(struct bpf_object *obj)
{
	struct bpf_map *m;

	bpf_object__for_each_map(m, obj) {
		if (!bpf_map__is_internal(m))
			continue;
		if (!obj->caps.global_data) {
			pr_warn("kernel doesn't support global data\n");
			return -ENOTSUP;
		}
		if (!obj->caps.array_mmap)
			m->def.map_flags ^= BPF_F_MMAPABLE;
	}

	return 0;
}

5756 5757 5758 5759 5760 5761 5762 5763 5764 5765 5766 5767 5768 5769 5770 5771 5772 5773 5774 5775 5776
static int bpf_object__read_kallsyms_file(struct bpf_object *obj)
{
	char sym_type, sym_name[500];
	unsigned long long sym_addr;
	struct extern_desc *ext;
	int ret, err = 0;
	FILE *f;

	f = fopen("/proc/kallsyms", "r");
	if (!f) {
		err = -errno;
		pr_warn("failed to open /proc/kallsyms: %d\n", err);
		return err;
	}

	while (true) {
		ret = fscanf(f, "%llx %c %499s%*[^\n]\n",
			     &sym_addr, &sym_type, sym_name);
		if (ret == EOF && feof(f))
			break;
		if (ret != 3) {
5777
			pr_warn("failed to read kallsyms entry: %d\n", ret);
5778 5779 5780 5781 5782 5783 5784 5785 5786 5787 5788 5789 5790 5791 5792 5793 5794 5795 5796 5797 5798 5799 5800 5801 5802 5803
			err = -EINVAL;
			goto out;
		}

		ext = find_extern_by_name(obj, sym_name);
		if (!ext || ext->type != EXT_KSYM)
			continue;

		if (ext->is_set && ext->ksym.addr != sym_addr) {
			pr_warn("extern (ksym) '%s' resolution is ambiguous: 0x%llx or 0x%llx\n",
				sym_name, ext->ksym.addr, sym_addr);
			err = -EINVAL;
			goto out;
		}
		if (!ext->is_set) {
			ext->is_set = true;
			ext->ksym.addr = sym_addr;
			pr_debug("extern (ksym) %s=0x%llx\n", sym_name, sym_addr);
		}
	}

out:
	fclose(f);
	return err;
}

5804
static int bpf_object__resolve_externs(struct bpf_object *obj,
5805
				       const char *extra_kconfig)
5806
{
5807
	bool need_config = false, need_kallsyms = false;
5808
	struct extern_desc *ext;
5809
	void *kcfg_data = NULL;
5810 5811 5812 5813 5814
	int err, i;

	if (obj->nr_extern == 0)
		return 0;

5815 5816
	if (obj->kconfig_map_idx >= 0)
		kcfg_data = obj->maps[obj->kconfig_map_idx].mmaped;
5817 5818 5819 5820

	for (i = 0; i < obj->nr_extern; i++) {
		ext = &obj->externs[i];

5821 5822 5823
		if (ext->type == EXT_KCFG &&
		    strcmp(ext->name, "LINUX_KERNEL_VERSION") == 0) {
			void *ext_val = kcfg_data + ext->kcfg.data_off;
5824 5825 5826 5827 5828 5829
			__u32 kver = get_kernel_version();

			if (!kver) {
				pr_warn("failed to get kernel version\n");
				return -EINVAL;
			}
5830
			err = set_kcfg_value_num(ext, ext_val, kver);
5831 5832
			if (err)
				return err;
5833 5834 5835
			pr_debug("extern (kcfg) %s=0x%x\n", ext->name, kver);
		} else if (ext->type == EXT_KCFG &&
			   strncmp(ext->name, "CONFIG_", 7) == 0) {
5836
			need_config = true;
5837 5838
		} else if (ext->type == EXT_KSYM) {
			need_kallsyms = true;
5839 5840 5841 5842 5843
		} else {
			pr_warn("unrecognized extern '%s'\n", ext->name);
			return -EINVAL;
		}
	}
5844
	if (need_config && extra_kconfig) {
5845
		err = bpf_object__read_kconfig_mem(obj, extra_kconfig, kcfg_data);
5846 5847 5848 5849 5850
		if (err)
			return -EINVAL;
		need_config = false;
		for (i = 0; i < obj->nr_extern; i++) {
			ext = &obj->externs[i];
5851
			if (ext->type == EXT_KCFG && !ext->is_set) {
5852 5853 5854 5855 5856
				need_config = true;
				break;
			}
		}
	}
5857
	if (need_config) {
5858
		err = bpf_object__read_kconfig_file(obj, kcfg_data);
5859 5860 5861
		if (err)
			return -EINVAL;
	}
5862 5863 5864 5865 5866
	if (need_kallsyms) {
		err = bpf_object__read_kallsyms_file(obj);
		if (err)
			return -EINVAL;
	}
5867 5868 5869 5870 5871 5872 5873 5874 5875 5876 5877 5878 5879 5880 5881
	for (i = 0; i < obj->nr_extern; i++) {
		ext = &obj->externs[i];

		if (!ext->is_set && !ext->is_weak) {
			pr_warn("extern %s (strong) not resolved\n", ext->name);
			return -ESRCH;
		} else if (!ext->is_set) {
			pr_debug("extern %s (weak) not resolved, defaulting to zero\n",
				 ext->name);
		}
	}

	return 0;
}

5882
int bpf_object__load_xattr(struct bpf_object_load_attr *attr)
5883
{
5884
	struct bpf_object *obj;
5885
	int err, i;
5886

5887 5888 5889
	if (!attr)
		return -EINVAL;
	obj = attr->obj;
5890 5891 5892 5893
	if (!obj)
		return -EINVAL;

	if (obj->loaded) {
5894
		pr_warn("object '%s': load can't be attempted twice\n", obj->name);
5895 5896 5897
		return -EINVAL;
	}

5898 5899
	err = bpf_object__probe_loading(obj);
	err = err ? : bpf_object__probe_caps(obj);
5900
	err = err ? : bpf_object__resolve_externs(obj, obj->kconfig);
5901 5902
	err = err ? : bpf_object__sanitize_and_load_btf(obj);
	err = err ? : bpf_object__sanitize_maps(obj);
5903
	err = err ? : bpf_object__load_vmlinux_btf(obj);
5904
	err = err ? : bpf_object__init_kern_struct_ops_maps(obj);
5905 5906 5907
	err = err ? : bpf_object__create_maps(obj);
	err = err ? : bpf_object__relocate(obj, attr->target_btf_path);
	err = err ? : bpf_object__load_progs(obj, attr->log_level);
5908 5909 5910 5911

	btf__free(obj->btf_vmlinux);
	obj->btf_vmlinux = NULL;

5912 5913
	obj->loaded = true; /* doesn't matter if successfully or not */

5914 5915
	if (err)
		goto out;
5916 5917 5918

	return 0;
out:
5919 5920 5921 5922 5923
	/* unpin any maps that were auto-pinned during load */
	for (i = 0; i < obj->nr_maps; i++)
		if (obj->maps[i].pinned && !obj->maps[i].reused)
			bpf_map__unpin(&obj->maps[i], NULL);

5924
	bpf_object__unload(obj);
5925
	pr_warn("failed to load object '%s'\n", obj->path);
5926
	return err;
5927 5928
}

5929 5930 5931 5932 5933 5934 5935 5936 5937
int bpf_object__load(struct bpf_object *obj)
{
	struct bpf_object_load_attr attr = {
		.obj = obj,
	};

	return bpf_object__load_xattr(&attr);
}

5938 5939 5940 5941 5942 5943 5944 5945 5946 5947 5948 5949 5950 5951 5952 5953 5954 5955 5956 5957 5958 5959
static int make_parent_dir(const char *path)
{
	char *cp, errmsg[STRERR_BUFSIZE];
	char *dname, *dir;
	int err = 0;

	dname = strdup(path);
	if (dname == NULL)
		return -ENOMEM;

	dir = dirname(dname);
	if (mkdir(dir, 0700) && errno != EEXIST)
		err = -errno;

	free(dname);
	if (err) {
		cp = libbpf_strerror_r(-err, errmsg, sizeof(errmsg));
		pr_warn("failed to mkdir %s: %s\n", path, cp);
	}
	return err;
}

5960 5961
static int check_path(const char *path)
{
5962
	char *cp, errmsg[STRERR_BUFSIZE];
5963 5964 5965 5966 5967 5968 5969 5970 5971 5972 5973 5974 5975
	struct statfs st_fs;
	char *dname, *dir;
	int err = 0;

	if (path == NULL)
		return -EINVAL;

	dname = strdup(path);
	if (dname == NULL)
		return -ENOMEM;

	dir = dirname(dname);
	if (statfs(dir, &st_fs)) {
5976
		cp = libbpf_strerror_r(errno, errmsg, sizeof(errmsg));
5977
		pr_warn("failed to statfs %s: %s\n", dir, cp);
5978 5979 5980 5981 5982
		err = -errno;
	}
	free(dname);

	if (!err && st_fs.f_type != BPF_FS_MAGIC) {
5983
		pr_warn("specified path %s is not on BPF FS\n", path);
5984 5985 5986 5987 5988 5989 5990 5991 5992
		err = -EINVAL;
	}

	return err;
}

int bpf_program__pin_instance(struct bpf_program *prog, const char *path,
			      int instance)
{
5993
	char *cp, errmsg[STRERR_BUFSIZE];
5994 5995
	int err;

5996 5997 5998 5999
	err = make_parent_dir(path);
	if (err)
		return err;

6000 6001 6002 6003 6004
	err = check_path(path);
	if (err)
		return err;

	if (prog == NULL) {
6005
		pr_warn("invalid program pointer\n");
6006 6007 6008 6009
		return -EINVAL;
	}

	if (instance < 0 || instance >= prog->instances.nr) {
6010 6011
		pr_warn("invalid prog instance %d of prog %s (max %d)\n",
			instance, prog->section_name, prog->instances.nr);
6012 6013 6014 6015
		return -EINVAL;
	}

	if (bpf_obj_pin(prog->instances.fds[instance], path)) {
6016 6017
		err = -errno;
		cp = libbpf_strerror_r(err, errmsg, sizeof(errmsg));
6018
		pr_warn("failed to pin program: %s\n", cp);
6019
		return err;
6020 6021 6022 6023 6024 6025
	}
	pr_debug("pinned program '%s'\n", path);

	return 0;
}

6026 6027 6028 6029 6030 6031 6032 6033 6034 6035
int bpf_program__unpin_instance(struct bpf_program *prog, const char *path,
				int instance)
{
	int err;

	err = check_path(path);
	if (err)
		return err;

	if (prog == NULL) {
6036
		pr_warn("invalid program pointer\n");
6037 6038 6039 6040
		return -EINVAL;
	}

	if (instance < 0 || instance >= prog->instances.nr) {
6041 6042
		pr_warn("invalid prog instance %d of prog %s (max %d)\n",
			instance, prog->section_name, prog->instances.nr);
6043 6044 6045 6046 6047 6048 6049 6050 6051 6052 6053
		return -EINVAL;
	}

	err = unlink(path);
	if (err != 0)
		return -errno;
	pr_debug("unpinned program '%s'\n", path);

	return 0;
}

6054 6055 6056 6057
int bpf_program__pin(struct bpf_program *prog, const char *path)
{
	int i, err;

6058 6059 6060 6061
	err = make_parent_dir(path);
	if (err)
		return err;

6062 6063 6064 6065 6066
	err = check_path(path);
	if (err)
		return err;

	if (prog == NULL) {
6067
		pr_warn("invalid program pointer\n");
6068 6069 6070 6071
		return -EINVAL;
	}

	if (prog->instances.nr <= 0) {
6072
		pr_warn("no instances of prog %s to pin\n",
6073 6074 6075 6076
			   prog->section_name);
		return -EINVAL;
	}

6077 6078 6079 6080 6081
	if (prog->instances.nr == 1) {
		/* don't create subdirs when pinning single instance */
		return bpf_program__pin_instance(prog, path, 0);
	}

6082 6083 6084 6085 6086 6087 6088 6089 6090 6091 6092 6093 6094 6095 6096 6097 6098 6099 6100 6101 6102 6103 6104 6105 6106 6107 6108 6109 6110 6111 6112 6113 6114 6115 6116 6117 6118 6119 6120 6121 6122 6123 6124 6125 6126 6127 6128 6129
	for (i = 0; i < prog->instances.nr; i++) {
		char buf[PATH_MAX];
		int len;

		len = snprintf(buf, PATH_MAX, "%s/%d", path, i);
		if (len < 0) {
			err = -EINVAL;
			goto err_unpin;
		} else if (len >= PATH_MAX) {
			err = -ENAMETOOLONG;
			goto err_unpin;
		}

		err = bpf_program__pin_instance(prog, buf, i);
		if (err)
			goto err_unpin;
	}

	return 0;

err_unpin:
	for (i = i - 1; i >= 0; i--) {
		char buf[PATH_MAX];
		int len;

		len = snprintf(buf, PATH_MAX, "%s/%d", path, i);
		if (len < 0)
			continue;
		else if (len >= PATH_MAX)
			continue;

		bpf_program__unpin_instance(prog, buf, i);
	}

	rmdir(path);

	return err;
}

int bpf_program__unpin(struct bpf_program *prog, const char *path)
{
	int i, err;

	err = check_path(path);
	if (err)
		return err;

	if (prog == NULL) {
6130
		pr_warn("invalid program pointer\n");
6131 6132 6133 6134
		return -EINVAL;
	}

	if (prog->instances.nr <= 0) {
6135
		pr_warn("no instances of prog %s to pin\n",
6136 6137
			   prog->section_name);
		return -EINVAL;
6138 6139 6140 6141 6142
	}

	if (prog->instances.nr == 1) {
		/* don't create subdirs when pinning single instance */
		return bpf_program__unpin_instance(prog, path, 0);
6143 6144
	}

6145 6146 6147 6148 6149 6150 6151 6152 6153 6154
	for (i = 0; i < prog->instances.nr; i++) {
		char buf[PATH_MAX];
		int len;

		len = snprintf(buf, PATH_MAX, "%s/%d", path, i);
		if (len < 0)
			return -EINVAL;
		else if (len >= PATH_MAX)
			return -ENAMETOOLONG;

6155
		err = bpf_program__unpin_instance(prog, buf, i);
6156 6157 6158 6159
		if (err)
			return err;
	}

6160 6161 6162 6163
	err = rmdir(path);
	if (err)
		return -errno;

6164 6165 6166
	return 0;
}

J
Joe Stringer 已提交
6167 6168
int bpf_map__pin(struct bpf_map *map, const char *path)
{
6169
	char *cp, errmsg[STRERR_BUFSIZE];
J
Joe Stringer 已提交
6170 6171 6172
	int err;

	if (map == NULL) {
6173
		pr_warn("invalid map pointer\n");
J
Joe Stringer 已提交
6174 6175 6176
		return -EINVAL;
	}

6177 6178 6179 6180 6181 6182 6183 6184 6185 6186 6187 6188 6189 6190 6191 6192 6193 6194 6195 6196 6197 6198 6199 6200 6201
	if (map->pin_path) {
		if (path && strcmp(path, map->pin_path)) {
			pr_warn("map '%s' already has pin path '%s' different from '%s'\n",
				bpf_map__name(map), map->pin_path, path);
			return -EINVAL;
		} else if (map->pinned) {
			pr_debug("map '%s' already pinned at '%s'; not re-pinning\n",
				 bpf_map__name(map), map->pin_path);
			return 0;
		}
	} else {
		if (!path) {
			pr_warn("missing a path to pin map '%s' at\n",
				bpf_map__name(map));
			return -EINVAL;
		} else if (map->pinned) {
			pr_warn("map '%s' already pinned\n", bpf_map__name(map));
			return -EEXIST;
		}

		map->pin_path = strdup(path);
		if (!map->pin_path) {
			err = -errno;
			goto out_err;
		}
J
Joe Stringer 已提交
6202 6203
	}

6204 6205 6206 6207
	err = make_parent_dir(map->pin_path);
	if (err)
		return err;

6208 6209 6210 6211 6212 6213 6214 6215 6216 6217 6218
	err = check_path(map->pin_path);
	if (err)
		return err;

	if (bpf_obj_pin(map->fd, map->pin_path)) {
		err = -errno;
		goto out_err;
	}

	map->pinned = true;
	pr_debug("pinned map '%s'\n", map->pin_path);
6219

J
Joe Stringer 已提交
6220
	return 0;
6221 6222 6223 6224 6225

out_err:
	cp = libbpf_strerror_r(-err, errmsg, sizeof(errmsg));
	pr_warn("failed to pin map: %s\n", cp);
	return err;
J
Joe Stringer 已提交
6226 6227
}

6228 6229 6230 6231 6232
int bpf_map__unpin(struct bpf_map *map, const char *path)
{
	int err;

	if (map == NULL) {
6233
		pr_warn("invalid map pointer\n");
6234 6235 6236
		return -EINVAL;
	}

6237 6238 6239 6240 6241 6242 6243 6244 6245 6246 6247 6248 6249 6250 6251 6252 6253
	if (map->pin_path) {
		if (path && strcmp(path, map->pin_path)) {
			pr_warn("map '%s' already has pin path '%s' different from '%s'\n",
				bpf_map__name(map), map->pin_path, path);
			return -EINVAL;
		}
		path = map->pin_path;
	} else if (!path) {
		pr_warn("no path to unpin map '%s' from\n",
			bpf_map__name(map));
		return -EINVAL;
	}

	err = check_path(path);
	if (err)
		return err;

6254 6255 6256
	err = unlink(path);
	if (err != 0)
		return -errno;
6257 6258 6259

	map->pinned = false;
	pr_debug("unpinned map '%s' from '%s'\n", bpf_map__name(map), path);
6260 6261 6262 6263

	return 0;
}

6264 6265 6266 6267 6268 6269 6270 6271 6272 6273 6274 6275 6276 6277 6278 6279 6280 6281 6282 6283 6284 6285 6286 6287 6288
int bpf_map__set_pin_path(struct bpf_map *map, const char *path)
{
	char *new = NULL;

	if (path) {
		new = strdup(path);
		if (!new)
			return -errno;
	}

	free(map->pin_path);
	map->pin_path = new;
	return 0;
}

const char *bpf_map__get_pin_path(const struct bpf_map *map)
{
	return map->pin_path;
}

bool bpf_map__is_pinned(const struct bpf_map *map)
{
	return map->pinned;
}

6289
int bpf_object__pin_maps(struct bpf_object *obj, const char *path)
6290 6291 6292 6293 6294 6295 6296 6297
{
	struct bpf_map *map;
	int err;

	if (!obj)
		return -ENOENT;

	if (!obj->loaded) {
6298
		pr_warn("object not yet loaded; load it first\n");
6299 6300 6301
		return -ENOENT;
	}

6302
	bpf_object__for_each_map(map, obj) {
6303
		char *pin_path = NULL;
6304 6305
		char buf[PATH_MAX];

6306 6307 6308 6309 6310 6311 6312 6313 6314 6315 6316 6317 6318 6319 6320
		if (path) {
			int len;

			len = snprintf(buf, PATH_MAX, "%s/%s", path,
				       bpf_map__name(map));
			if (len < 0) {
				err = -EINVAL;
				goto err_unpin_maps;
			} else if (len >= PATH_MAX) {
				err = -ENAMETOOLONG;
				goto err_unpin_maps;
			}
			pin_path = buf;
		} else if (!map->pin_path) {
			continue;
6321 6322
		}

6323
		err = bpf_map__pin(map, pin_path);
6324 6325 6326 6327 6328 6329 6330 6331
		if (err)
			goto err_unpin_maps;
	}

	return 0;

err_unpin_maps:
	while ((map = bpf_map__prev(map, obj))) {
6332
		if (!map->pin_path)
6333 6334
			continue;

6335
		bpf_map__unpin(map, NULL);
6336 6337 6338 6339 6340 6341 6342 6343 6344 6345 6346 6347 6348
	}

	return err;
}

int bpf_object__unpin_maps(struct bpf_object *obj, const char *path)
{
	struct bpf_map *map;
	int err;

	if (!obj)
		return -ENOENT;

6349
	bpf_object__for_each_map(map, obj) {
6350
		char *pin_path = NULL;
6351 6352
		char buf[PATH_MAX];

6353 6354 6355 6356 6357 6358 6359 6360 6361 6362 6363 6364 6365
		if (path) {
			int len;

			len = snprintf(buf, PATH_MAX, "%s/%s", path,
				       bpf_map__name(map));
			if (len < 0)
				return -EINVAL;
			else if (len >= PATH_MAX)
				return -ENAMETOOLONG;
			pin_path = buf;
		} else if (!map->pin_path) {
			continue;
		}
6366

6367
		err = bpf_map__unpin(map, pin_path);
6368 6369 6370 6371
		if (err)
			return err;
	}

6372 6373 6374 6375 6376 6377 6378 6379 6380 6381 6382 6383
	return 0;
}

int bpf_object__pin_programs(struct bpf_object *obj, const char *path)
{
	struct bpf_program *prog;
	int err;

	if (!obj)
		return -ENOENT;

	if (!obj->loaded) {
6384
		pr_warn("object not yet loaded; load it first\n");
6385 6386 6387 6388 6389 6390 6391 6392
		return -ENOENT;
	}

	bpf_object__for_each_program(prog, obj) {
		char buf[PATH_MAX];
		int len;

		len = snprintf(buf, PATH_MAX, "%s/%s", path,
S
Stanislav Fomichev 已提交
6393
			       prog->pin_name);
6394 6395 6396 6397 6398 6399 6400 6401 6402 6403 6404 6405 6406 6407 6408 6409 6410 6411 6412 6413 6414
		if (len < 0) {
			err = -EINVAL;
			goto err_unpin_programs;
		} else if (len >= PATH_MAX) {
			err = -ENAMETOOLONG;
			goto err_unpin_programs;
		}

		err = bpf_program__pin(prog, buf);
		if (err)
			goto err_unpin_programs;
	}

	return 0;

err_unpin_programs:
	while ((prog = bpf_program__prev(prog, obj))) {
		char buf[PATH_MAX];
		int len;

		len = snprintf(buf, PATH_MAX, "%s/%s", path,
S
Stanislav Fomichev 已提交
6415
			       prog->pin_name);
6416 6417 6418 6419 6420 6421 6422 6423 6424 6425 6426 6427 6428 6429 6430 6431 6432 6433 6434
		if (len < 0)
			continue;
		else if (len >= PATH_MAX)
			continue;

		bpf_program__unpin(prog, buf);
	}

	return err;
}

int bpf_object__unpin_programs(struct bpf_object *obj, const char *path)
{
	struct bpf_program *prog;
	int err;

	if (!obj)
		return -ENOENT;

6435 6436 6437 6438 6439
	bpf_object__for_each_program(prog, obj) {
		char buf[PATH_MAX];
		int len;

		len = snprintf(buf, PATH_MAX, "%s/%s", path,
S
Stanislav Fomichev 已提交
6440
			       prog->pin_name);
6441 6442 6443 6444 6445
		if (len < 0)
			return -EINVAL;
		else if (len >= PATH_MAX)
			return -ENAMETOOLONG;

6446
		err = bpf_program__unpin(prog, buf);
6447 6448 6449 6450 6451 6452 6453
		if (err)
			return err;
	}

	return 0;
}

6454 6455 6456 6457 6458 6459 6460 6461 6462 6463 6464 6465 6466 6467 6468 6469 6470
int bpf_object__pin(struct bpf_object *obj, const char *path)
{
	int err;

	err = bpf_object__pin_maps(obj, path);
	if (err)
		return err;

	err = bpf_object__pin_programs(obj, path);
	if (err) {
		bpf_object__unpin_maps(obj, path);
		return err;
	}

	return 0;
}

6471 6472 6473 6474 6475 6476 6477
static void bpf_map__destroy(struct bpf_map *map)
{
	if (map->clear_priv)
		map->clear_priv(map, map->priv);
	map->priv = NULL;
	map->clear_priv = NULL;

6478 6479 6480 6481 6482 6483 6484 6485
	if (map->inner_map) {
		bpf_map__destroy(map->inner_map);
		zfree(&map->inner_map);
	}

	zfree(&map->init_slots);
	map->init_slots_sz = 0;

6486 6487 6488 6489 6490 6491 6492 6493 6494 6495 6496 6497 6498 6499 6500 6501 6502 6503 6504
	if (map->mmaped) {
		munmap(map->mmaped, bpf_map_mmap_sz(map));
		map->mmaped = NULL;
	}

	if (map->st_ops) {
		zfree(&map->st_ops->data);
		zfree(&map->st_ops->progs);
		zfree(&map->st_ops->kern_func_off);
		zfree(&map->st_ops);
	}

	zfree(&map->name);
	zfree(&map->pin_path);

	if (map->fd >= 0)
		zclose(map->fd);
}

6505 6506
void bpf_object__close(struct bpf_object *obj)
{
6507 6508
	size_t i;

6509
	if (IS_ERR_OR_NULL(obj))
6510 6511
		return;

6512 6513 6514
	if (obj->clear_priv)
		obj->clear_priv(obj, obj->priv);

6515
	bpf_object__elf_finish(obj);
6516
	bpf_object__unload(obj);
6517
	btf__free(obj->btf);
6518
	btf_ext__free(obj->btf_ext);
6519

6520 6521
	for (i = 0; i < obj->nr_maps; i++)
		bpf_map__destroy(&obj->maps[i]);
6522

6523
	zfree(&obj->kconfig);
6524 6525 6526
	zfree(&obj->externs);
	obj->nr_extern = 0;

6527 6528
	zfree(&obj->maps);
	obj->nr_maps = 0;
6529 6530 6531 6532 6533 6534 6535

	if (obj->programs && obj->nr_programs) {
		for (i = 0; i < obj->nr_programs; i++)
			bpf_program__exit(&obj->programs[i]);
	}
	zfree(&obj->programs);

6536
	list_del(&obj->list);
6537 6538
	free(obj);
}
6539

6540 6541 6542 6543 6544 6545 6546 6547 6548 6549 6550 6551 6552 6553 6554 6555 6556 6557 6558
struct bpf_object *
bpf_object__next(struct bpf_object *prev)
{
	struct bpf_object *next;

	if (!prev)
		next = list_first_entry(&bpf_objects_list,
					struct bpf_object,
					list);
	else
		next = list_next_entry(prev, list);

	/* Empty list is noticed here so don't need checking on entry. */
	if (&next->list == &bpf_objects_list)
		return NULL;

	return next;
}

A
Andrii Nakryiko 已提交
6559
const char *bpf_object__name(const struct bpf_object *obj)
6560
{
6561
	return obj ? obj->name : ERR_PTR(-EINVAL);
6562 6563
}

A
Andrii Nakryiko 已提交
6564
unsigned int bpf_object__kversion(const struct bpf_object *obj)
6565
{
6566
	return obj ? obj->kern_version : 0;
6567 6568
}

A
Andrii Nakryiko 已提交
6569
struct btf *bpf_object__btf(const struct bpf_object *obj)
6570 6571 6572 6573
{
	return obj ? obj->btf : NULL;
}

6574 6575 6576 6577 6578
int bpf_object__btf_fd(const struct bpf_object *obj)
{
	return obj->btf ? btf__fd(obj->btf) : -1;
}

6579 6580 6581 6582 6583 6584 6585 6586 6587 6588 6589
int bpf_object__set_priv(struct bpf_object *obj, void *priv,
			 bpf_object_clear_priv_t clear_priv)
{
	if (obj->priv && obj->clear_priv)
		obj->clear_priv(obj, obj->priv);

	obj->priv = priv;
	obj->clear_priv = clear_priv;
	return 0;
}

A
Andrii Nakryiko 已提交
6590
void *bpf_object__priv(const struct bpf_object *obj)
6591 6592 6593 6594
{
	return obj ? obj->priv : ERR_PTR(-EINVAL);
}

6595
static struct bpf_program *
A
Andrii Nakryiko 已提交
6596 6597
__bpf_program__iter(const struct bpf_program *p, const struct bpf_object *obj,
		    bool forward)
6598
{
6599
	size_t nr_programs = obj->nr_programs;
6600
	ssize_t idx;
6601

6602
	if (!nr_programs)
6603 6604
		return NULL;

6605 6606 6607 6608 6609
	if (!p)
		/* Iter from the beginning */
		return forward ? &obj->programs[0] :
			&obj->programs[nr_programs - 1];

6610
	if (p->obj != obj) {
6611
		pr_warn("error: program handler doesn't match object\n");
6612 6613 6614
		return NULL;
	}

6615
	idx = (p - obj->programs) + (forward ? 1 : -1);
6616
	if (idx >= obj->nr_programs || idx < 0)
6617 6618 6619 6620
		return NULL;
	return &obj->programs[idx];
}

6621
struct bpf_program *
A
Andrii Nakryiko 已提交
6622
bpf_program__next(struct bpf_program *prev, const struct bpf_object *obj)
6623 6624 6625 6626
{
	struct bpf_program *prog = prev;

	do {
6627
		prog = __bpf_program__iter(prog, obj, true);
6628 6629 6630 6631 6632 6633
	} while (prog && bpf_program__is_function_storage(prog, obj));

	return prog;
}

struct bpf_program *
A
Andrii Nakryiko 已提交
6634
bpf_program__prev(struct bpf_program *next, const struct bpf_object *obj)
6635 6636 6637 6638
{
	struct bpf_program *prog = next;

	do {
6639
		prog = __bpf_program__iter(prog, obj, false);
6640 6641 6642 6643 6644
	} while (prog && bpf_program__is_function_storage(prog, obj));

	return prog;
}

6645 6646
int bpf_program__set_priv(struct bpf_program *prog, void *priv,
			  bpf_program_clear_priv_t clear_priv)
6647 6648 6649 6650 6651 6652 6653 6654 6655
{
	if (prog->priv && prog->clear_priv)
		prog->clear_priv(prog, prog->priv);

	prog->priv = priv;
	prog->clear_priv = clear_priv;
	return 0;
}

A
Andrii Nakryiko 已提交
6656
void *bpf_program__priv(const struct bpf_program *prog)
6657
{
6658
	return prog ? prog->priv : ERR_PTR(-EINVAL);
6659 6660
}

6661 6662 6663 6664 6665
void bpf_program__set_ifindex(struct bpf_program *prog, __u32 ifindex)
{
	prog->prog_ifindex = ifindex;
}

6666 6667 6668 6669 6670
const char *bpf_program__name(const struct bpf_program *prog)
{
	return prog->name;
}

A
Andrii Nakryiko 已提交
6671
const char *bpf_program__title(const struct bpf_program *prog, bool needs_copy)
6672 6673 6674 6675
{
	const char *title;

	title = prog->section_name;
6676
	if (needs_copy) {
6677 6678
		title = strdup(title);
		if (!title) {
6679
			pr_warn("failed to strdup program title\n");
6680
			return ERR_PTR(-ENOMEM);
6681 6682 6683 6684 6685 6686
		}
	}

	return title;
}

6687 6688 6689 6690 6691 6692 6693 6694 6695 6696 6697 6698 6699 6700
bool bpf_program__autoload(const struct bpf_program *prog)
{
	return prog->load;
}

int bpf_program__set_autoload(struct bpf_program *prog, bool autoload)
{
	if (prog->obj->loaded)
		return -EINVAL;

	prog->load = autoload;
	return 0;
}

A
Andrii Nakryiko 已提交
6701
int bpf_program__fd(const struct bpf_program *prog)
6702
{
6703 6704 6705
	return bpf_program__nth_fd(prog, 0);
}

6706 6707 6708 6709 6710
size_t bpf_program__size(const struct bpf_program *prog)
{
	return prog->insns_cnt * sizeof(struct bpf_insn);
}

6711 6712 6713 6714 6715 6716 6717 6718 6719
int bpf_program__set_prep(struct bpf_program *prog, int nr_instances,
			  bpf_program_prep_t prep)
{
	int *instances_fds;

	if (nr_instances <= 0 || !prep)
		return -EINVAL;

	if (prog->instances.nr > 0 || prog->instances.fds) {
6720
		pr_warn("Can't set pre-processor after loading\n");
6721 6722 6723 6724 6725
		return -EINVAL;
	}

	instances_fds = malloc(sizeof(int) * nr_instances);
	if (!instances_fds) {
6726
		pr_warn("alloc memory failed for fds\n");
6727 6728 6729 6730 6731 6732 6733 6734 6735 6736 6737 6738
		return -ENOMEM;
	}

	/* fill all fd with -1 */
	memset(instances_fds, -1, sizeof(int) * nr_instances);

	prog->instances.nr = nr_instances;
	prog->instances.fds = instances_fds;
	prog->preprocessor = prep;
	return 0;
}

A
Andrii Nakryiko 已提交
6739
int bpf_program__nth_fd(const struct bpf_program *prog, int n)
6740 6741 6742
{
	int fd;

6743 6744 6745
	if (!prog)
		return -EINVAL;

6746
	if (n >= prog->instances.nr || n < 0) {
6747 6748
		pr_warn("Can't get the %dth fd from program %s: only %d instances\n",
			n, prog->section_name, prog->instances.nr);
6749 6750 6751 6752 6753
		return -EINVAL;
	}

	fd = prog->instances.fds[n];
	if (fd < 0) {
6754 6755
		pr_warn("%dth instance of program '%s' is invalid\n",
			n, prog->section_name);
6756 6757 6758 6759
		return -ENOENT;
	}

	return fd;
6760
}
6761

6762 6763 6764 6765 6766
enum bpf_prog_type bpf_program__get_type(struct bpf_program *prog)
{
	return prog->type;
}

6767
void bpf_program__set_type(struct bpf_program *prog, enum bpf_prog_type type)
6768 6769 6770 6771
{
	prog->type = type;
}

A
Andrii Nakryiko 已提交
6772
static bool bpf_program__is_type(const struct bpf_program *prog,
6773 6774 6775 6776 6777
				 enum bpf_prog_type type)
{
	return prog ? (prog->type == type) : false;
}

A
Andrii Nakryiko 已提交
6778 6779 6780 6781 6782 6783 6784 6785 6786 6787 6788 6789 6790
#define BPF_PROG_TYPE_FNS(NAME, TYPE)				\
int bpf_program__set_##NAME(struct bpf_program *prog)		\
{								\
	if (!prog)						\
		return -EINVAL;					\
	bpf_program__set_type(prog, TYPE);			\
	return 0;						\
}								\
								\
bool bpf_program__is_##NAME(const struct bpf_program *prog)	\
{								\
	return bpf_program__is_type(prog, TYPE);		\
}								\
6791

6792
BPF_PROG_TYPE_FNS(socket_filter, BPF_PROG_TYPE_SOCKET_FILTER);
6793
BPF_PROG_TYPE_FNS(lsm, BPF_PROG_TYPE_LSM);
6794
BPF_PROG_TYPE_FNS(kprobe, BPF_PROG_TYPE_KPROBE);
6795 6796
BPF_PROG_TYPE_FNS(sched_cls, BPF_PROG_TYPE_SCHED_CLS);
BPF_PROG_TYPE_FNS(sched_act, BPF_PROG_TYPE_SCHED_ACT);
6797
BPF_PROG_TYPE_FNS(tracepoint, BPF_PROG_TYPE_TRACEPOINT);
6798
BPF_PROG_TYPE_FNS(raw_tracepoint, BPF_PROG_TYPE_RAW_TRACEPOINT);
6799 6800
BPF_PROG_TYPE_FNS(xdp, BPF_PROG_TYPE_XDP);
BPF_PROG_TYPE_FNS(perf_event, BPF_PROG_TYPE_PERF_EVENT);
6801
BPF_PROG_TYPE_FNS(tracing, BPF_PROG_TYPE_TRACING);
6802
BPF_PROG_TYPE_FNS(struct_ops, BPF_PROG_TYPE_STRUCT_OPS);
6803
BPF_PROG_TYPE_FNS(extension, BPF_PROG_TYPE_EXT);
6804
BPF_PROG_TYPE_FNS(sk_lookup, BPF_PROG_TYPE_SK_LOOKUP);
6805

6806 6807 6808 6809 6810 6811
enum bpf_attach_type
bpf_program__get_expected_attach_type(struct bpf_program *prog)
{
	return prog->expected_attach_type;
}

J
John Fastabend 已提交
6812 6813
void bpf_program__set_expected_attach_type(struct bpf_program *prog,
					   enum bpf_attach_type type)
6814 6815 6816 6817
{
	prog->expected_attach_type = type;
}

6818 6819 6820 6821 6822 6823 6824 6825 6826 6827 6828
#define BPF_PROG_SEC_IMPL(string, ptype, eatype, eatype_optional,	    \
			  attachable, attach_btf)			    \
	{								    \
		.sec = string,						    \
		.len = sizeof(string) - 1,				    \
		.prog_type = ptype,					    \
		.expected_attach_type = eatype,				    \
		.is_exp_attach_type_optional = eatype_optional,		    \
		.is_attachable = attachable,				    \
		.is_attach_btf = attach_btf,				    \
	}
6829

6830
/* Programs that can NOT be attached. */
6831
#define BPF_PROG_SEC(string, ptype) BPF_PROG_SEC_IMPL(string, ptype, 0, 0, 0, 0)
6832

6833 6834
/* Programs that can be attached. */
#define BPF_APROG_SEC(string, ptype, atype) \
6835
	BPF_PROG_SEC_IMPL(string, ptype, atype, true, 1, 0)
6836

6837 6838
/* Programs that must specify expected attach type at load time. */
#define BPF_EAPROG_SEC(string, ptype, eatype) \
6839
	BPF_PROG_SEC_IMPL(string, ptype, eatype, false, 1, 0)
6840 6841

/* Programs that use BTF to identify attach point */
6842
#define BPF_PROG_BTF(string, ptype, eatype) \
6843
	BPF_PROG_SEC_IMPL(string, ptype, eatype, false, 0, 1)
6844 6845 6846 6847 6848

/* Programs that can be attached but attach type can't be identified by section
 * name. Kept for backward compatibility.
 */
#define BPF_APROG_COMPAT(string, ptype) BPF_PROG_SEC(string, ptype)
6849

6850 6851 6852 6853 6854 6855 6856 6857 6858 6859 6860 6861 6862 6863 6864
#define SEC_DEF(sec_pfx, ptype, ...) {					    \
	.sec = sec_pfx,							    \
	.len = sizeof(sec_pfx) - 1,					    \
	.prog_type = BPF_PROG_TYPE_##ptype,				    \
	__VA_ARGS__							    \
}

static struct bpf_link *attach_kprobe(const struct bpf_sec_def *sec,
				      struct bpf_program *prog);
static struct bpf_link *attach_tp(const struct bpf_sec_def *sec,
				  struct bpf_program *prog);
static struct bpf_link *attach_raw_tp(const struct bpf_sec_def *sec,
				      struct bpf_program *prog);
static struct bpf_link *attach_trace(const struct bpf_sec_def *sec,
				     struct bpf_program *prog);
6865 6866
static struct bpf_link *attach_lsm(const struct bpf_sec_def *sec,
				   struct bpf_program *prog);
6867 6868
static struct bpf_link *attach_iter(const struct bpf_sec_def *sec,
				    struct bpf_program *prog);
6869 6870

static const struct bpf_sec_def section_defs[] = {
6871
	BPF_PROG_SEC("socket",			BPF_PROG_TYPE_SOCKET_FILTER),
6872
	BPF_PROG_SEC("sk_reuseport",		BPF_PROG_TYPE_SK_REUSEPORT),
6873 6874
	SEC_DEF("kprobe/", KPROBE,
		.attach_fn = attach_kprobe),
6875
	BPF_PROG_SEC("uprobe/",			BPF_PROG_TYPE_KPROBE),
6876 6877
	SEC_DEF("kretprobe/", KPROBE,
		.attach_fn = attach_kprobe),
6878
	BPF_PROG_SEC("uretprobe/",		BPF_PROG_TYPE_KPROBE),
6879 6880
	BPF_PROG_SEC("classifier",		BPF_PROG_TYPE_SCHED_CLS),
	BPF_PROG_SEC("action",			BPF_PROG_TYPE_SCHED_ACT),
6881 6882 6883 6884 6885 6886 6887 6888 6889 6890 6891 6892 6893 6894 6895 6896
	SEC_DEF("tracepoint/", TRACEPOINT,
		.attach_fn = attach_tp),
	SEC_DEF("tp/", TRACEPOINT,
		.attach_fn = attach_tp),
	SEC_DEF("raw_tracepoint/", RAW_TRACEPOINT,
		.attach_fn = attach_raw_tp),
	SEC_DEF("raw_tp/", RAW_TRACEPOINT,
		.attach_fn = attach_raw_tp),
	SEC_DEF("tp_btf/", TRACING,
		.expected_attach_type = BPF_TRACE_RAW_TP,
		.is_attach_btf = true,
		.attach_fn = attach_trace),
	SEC_DEF("fentry/", TRACING,
		.expected_attach_type = BPF_TRACE_FENTRY,
		.is_attach_btf = true,
		.attach_fn = attach_trace),
6897 6898 6899 6900
	SEC_DEF("fmod_ret/", TRACING,
		.expected_attach_type = BPF_MODIFY_RETURN,
		.is_attach_btf = true,
		.attach_fn = attach_trace),
6901 6902 6903 6904
	SEC_DEF("fexit/", TRACING,
		.expected_attach_type = BPF_TRACE_FEXIT,
		.is_attach_btf = true,
		.attach_fn = attach_trace),
6905 6906 6907
	SEC_DEF("freplace/", EXT,
		.is_attach_btf = true,
		.attach_fn = attach_trace),
6908 6909 6910 6911
	SEC_DEF("lsm/", LSM,
		.is_attach_btf = true,
		.expected_attach_type = BPF_LSM_MAC,
		.attach_fn = attach_lsm),
6912 6913 6914 6915
	SEC_DEF("iter/", TRACING,
		.expected_attach_type = BPF_TRACE_ITER,
		.is_attach_btf = true,
		.attach_fn = attach_iter),
6916
	BPF_EAPROG_SEC("xdp_devmap/",		BPF_PROG_TYPE_XDP,
6917
						BPF_XDP_DEVMAP),
6918 6919
	BPF_EAPROG_SEC("xdp_cpumap/",		BPF_PROG_TYPE_XDP,
						BPF_XDP_CPUMAP),
6920 6921
	BPF_EAPROG_SEC("xdp",			BPF_PROG_TYPE_XDP,
						BPF_XDP),
6922 6923 6924 6925 6926
	BPF_PROG_SEC("perf_event",		BPF_PROG_TYPE_PERF_EVENT),
	BPF_PROG_SEC("lwt_in",			BPF_PROG_TYPE_LWT_IN),
	BPF_PROG_SEC("lwt_out",			BPF_PROG_TYPE_LWT_OUT),
	BPF_PROG_SEC("lwt_xmit",		BPF_PROG_TYPE_LWT_XMIT),
	BPF_PROG_SEC("lwt_seg6local",		BPF_PROG_TYPE_LWT_SEG6LOCAL),
6927 6928 6929 6930
	BPF_APROG_SEC("cgroup_skb/ingress",	BPF_PROG_TYPE_CGROUP_SKB,
						BPF_CGROUP_INET_INGRESS),
	BPF_APROG_SEC("cgroup_skb/egress",	BPF_PROG_TYPE_CGROUP_SKB,
						BPF_CGROUP_INET_EGRESS),
6931
	BPF_APROG_COMPAT("cgroup/skb",		BPF_PROG_TYPE_CGROUP_SKB),
6932 6933 6934 6935
	BPF_EAPROG_SEC("cgroup/sock_create",	BPF_PROG_TYPE_CGROUP_SOCK,
						BPF_CGROUP_INET_SOCK_CREATE),
	BPF_EAPROG_SEC("cgroup/sock_release",	BPF_PROG_TYPE_CGROUP_SOCK,
						BPF_CGROUP_INET_SOCK_RELEASE),
6936 6937 6938 6939 6940 6941 6942 6943 6944 6945
	BPF_APROG_SEC("cgroup/sock",		BPF_PROG_TYPE_CGROUP_SOCK,
						BPF_CGROUP_INET_SOCK_CREATE),
	BPF_EAPROG_SEC("cgroup/post_bind4",	BPF_PROG_TYPE_CGROUP_SOCK,
						BPF_CGROUP_INET4_POST_BIND),
	BPF_EAPROG_SEC("cgroup/post_bind6",	BPF_PROG_TYPE_CGROUP_SOCK,
						BPF_CGROUP_INET6_POST_BIND),
	BPF_APROG_SEC("cgroup/dev",		BPF_PROG_TYPE_CGROUP_DEVICE,
						BPF_CGROUP_DEVICE),
	BPF_APROG_SEC("sockops",		BPF_PROG_TYPE_SOCK_OPS,
						BPF_CGROUP_SOCK_OPS),
6946 6947 6948 6949
	BPF_APROG_SEC("sk_skb/stream_parser",	BPF_PROG_TYPE_SK_SKB,
						BPF_SK_SKB_STREAM_PARSER),
	BPF_APROG_SEC("sk_skb/stream_verdict",	BPF_PROG_TYPE_SK_SKB,
						BPF_SK_SKB_STREAM_VERDICT),
6950 6951 6952 6953 6954 6955 6956 6957 6958 6959 6960 6961 6962 6963 6964 6965 6966 6967 6968
	BPF_APROG_COMPAT("sk_skb",		BPF_PROG_TYPE_SK_SKB),
	BPF_APROG_SEC("sk_msg",			BPF_PROG_TYPE_SK_MSG,
						BPF_SK_MSG_VERDICT),
	BPF_APROG_SEC("lirc_mode2",		BPF_PROG_TYPE_LIRC_MODE2,
						BPF_LIRC_MODE2),
	BPF_APROG_SEC("flow_dissector",		BPF_PROG_TYPE_FLOW_DISSECTOR,
						BPF_FLOW_DISSECTOR),
	BPF_EAPROG_SEC("cgroup/bind4",		BPF_PROG_TYPE_CGROUP_SOCK_ADDR,
						BPF_CGROUP_INET4_BIND),
	BPF_EAPROG_SEC("cgroup/bind6",		BPF_PROG_TYPE_CGROUP_SOCK_ADDR,
						BPF_CGROUP_INET6_BIND),
	BPF_EAPROG_SEC("cgroup/connect4",	BPF_PROG_TYPE_CGROUP_SOCK_ADDR,
						BPF_CGROUP_INET4_CONNECT),
	BPF_EAPROG_SEC("cgroup/connect6",	BPF_PROG_TYPE_CGROUP_SOCK_ADDR,
						BPF_CGROUP_INET6_CONNECT),
	BPF_EAPROG_SEC("cgroup/sendmsg4",	BPF_PROG_TYPE_CGROUP_SOCK_ADDR,
						BPF_CGROUP_UDP4_SENDMSG),
	BPF_EAPROG_SEC("cgroup/sendmsg6",	BPF_PROG_TYPE_CGROUP_SOCK_ADDR,
						BPF_CGROUP_UDP6_SENDMSG),
6969 6970 6971 6972
	BPF_EAPROG_SEC("cgroup/recvmsg4",	BPF_PROG_TYPE_CGROUP_SOCK_ADDR,
						BPF_CGROUP_UDP4_RECVMSG),
	BPF_EAPROG_SEC("cgroup/recvmsg6",	BPF_PROG_TYPE_CGROUP_SOCK_ADDR,
						BPF_CGROUP_UDP6_RECVMSG),
6973 6974 6975 6976 6977 6978 6979 6980
	BPF_EAPROG_SEC("cgroup/getpeername4",	BPF_PROG_TYPE_CGROUP_SOCK_ADDR,
						BPF_CGROUP_INET4_GETPEERNAME),
	BPF_EAPROG_SEC("cgroup/getpeername6",	BPF_PROG_TYPE_CGROUP_SOCK_ADDR,
						BPF_CGROUP_INET6_GETPEERNAME),
	BPF_EAPROG_SEC("cgroup/getsockname4",	BPF_PROG_TYPE_CGROUP_SOCK_ADDR,
						BPF_CGROUP_INET4_GETSOCKNAME),
	BPF_EAPROG_SEC("cgroup/getsockname6",	BPF_PROG_TYPE_CGROUP_SOCK_ADDR,
						BPF_CGROUP_INET6_GETSOCKNAME),
A
Andrey Ignatov 已提交
6981 6982
	BPF_EAPROG_SEC("cgroup/sysctl",		BPF_PROG_TYPE_CGROUP_SYSCTL,
						BPF_CGROUP_SYSCTL),
6983 6984 6985 6986
	BPF_EAPROG_SEC("cgroup/getsockopt",	BPF_PROG_TYPE_CGROUP_SOCKOPT,
						BPF_CGROUP_GETSOCKOPT),
	BPF_EAPROG_SEC("cgroup/setsockopt",	BPF_PROG_TYPE_CGROUP_SOCKOPT,
						BPF_CGROUP_SETSOCKOPT),
6987
	BPF_PROG_SEC("struct_ops",		BPF_PROG_TYPE_STRUCT_OPS),
6988 6989
	BPF_EAPROG_SEC("sk_lookup/",		BPF_PROG_TYPE_SK_LOOKUP,
						BPF_SK_LOOKUP),
6990
};
6991

6992
#undef BPF_PROG_SEC_IMPL
6993
#undef BPF_PROG_SEC
6994 6995 6996
#undef BPF_APROG_SEC
#undef BPF_EAPROG_SEC
#undef BPF_APROG_COMPAT
6997
#undef SEC_DEF
6998

6999 7000
#define MAX_TYPE_NAME_SIZE 32

7001 7002 7003 7004 7005 7006 7007 7008 7009 7010 7011 7012 7013
static const struct bpf_sec_def *find_sec_def(const char *sec_name)
{
	int i, n = ARRAY_SIZE(section_defs);

	for (i = 0; i < n; i++) {
		if (strncmp(sec_name,
			    section_defs[i].sec, section_defs[i].len))
			continue;
		return &section_defs[i];
	}
	return NULL;
}

7014 7015
static char *libbpf_get_type_names(bool attach_type)
{
7016
	int i, len = ARRAY_SIZE(section_defs) * MAX_TYPE_NAME_SIZE;
7017 7018 7019 7020 7021 7022 7023 7024
	char *buf;

	buf = malloc(len);
	if (!buf)
		return NULL;

	buf[0] = '\0';
	/* Forge string buf with all available names */
7025 7026
	for (i = 0; i < ARRAY_SIZE(section_defs); i++) {
		if (attach_type && !section_defs[i].is_attachable)
7027 7028
			continue;

7029
		if (strlen(buf) + strlen(section_defs[i].sec) + 2 > len) {
7030 7031 7032 7033
			free(buf);
			return NULL;
		}
		strcat(buf, " ");
7034
		strcat(buf, section_defs[i].sec);
7035 7036 7037 7038 7039
	}

	return buf;
}

7040 7041
int libbpf_prog_type_by_name(const char *name, enum bpf_prog_type *prog_type,
			     enum bpf_attach_type *expected_attach_type)
7042
{
7043
	const struct bpf_sec_def *sec_def;
7044
	char *type_names;
7045

7046 7047
	if (!name)
		return -EINVAL;
7048

7049 7050 7051 7052
	sec_def = find_sec_def(name);
	if (sec_def) {
		*prog_type = sec_def->prog_type;
		*expected_attach_type = sec_def->expected_attach_type;
7053 7054
		return 0;
	}
7055

7056
	pr_debug("failed to guess program type from ELF section '%s'\n", name);
7057 7058
	type_names = libbpf_get_type_names(false);
	if (type_names != NULL) {
7059
		pr_debug("supported section(type) names are:%s\n", type_names);
7060 7061 7062
		free(type_names);
	}

7063
	return -ESRCH;
7064
}
7065

7066 7067 7068 7069 7070 7071 7072 7073 7074 7075 7076 7077 7078 7079 7080 7081 7082 7083 7084
static struct bpf_map *find_struct_ops_map_by_offset(struct bpf_object *obj,
						     size_t offset)
{
	struct bpf_map *map;
	size_t i;

	for (i = 0; i < obj->nr_maps; i++) {
		map = &obj->maps[i];
		if (!bpf_map__is_struct_ops(map))
			continue;
		if (map->sec_offset <= offset &&
		    offset - map->sec_offset < map->def.value_size)
			return map;
	}

	return NULL;
}

/* Collect the reloc from ELF and populate the st_ops->progs[] */
7085 7086
static int bpf_object__collect_st_ops_relos(struct bpf_object *obj,
					    GElf_Shdr *shdr, Elf_Data *data)
7087 7088 7089 7090 7091 7092 7093 7094 7095 7096
{
	const struct btf_member *member;
	struct bpf_struct_ops *st_ops;
	struct bpf_program *prog;
	unsigned int shdr_idx;
	const struct btf *btf;
	struct bpf_map *map;
	Elf_Data *symbols;
	unsigned int moff;
	const char *name;
7097
	__u32 member_idx;
7098 7099 7100 7101 7102 7103 7104 7105 7106 7107 7108 7109 7110 7111 7112 7113 7114 7115 7116 7117 7118 7119 7120 7121 7122 7123 7124 7125 7126 7127 7128 7129 7130 7131 7132 7133 7134 7135 7136 7137 7138 7139 7140 7141 7142 7143 7144 7145 7146 7147 7148 7149 7150 7151 7152 7153 7154 7155 7156 7157 7158 7159 7160 7161 7162 7163 7164 7165 7166 7167 7168 7169 7170 7171 7172 7173 7174 7175 7176 7177 7178 7179 7180 7181 7182 7183 7184 7185 7186 7187 7188 7189 7190 7191 7192 7193 7194
	GElf_Sym sym;
	GElf_Rel rel;
	int i, nrels;

	symbols = obj->efile.symbols;
	btf = obj->btf;
	nrels = shdr->sh_size / shdr->sh_entsize;
	for (i = 0; i < nrels; i++) {
		if (!gelf_getrel(data, i, &rel)) {
			pr_warn("struct_ops reloc: failed to get %d reloc\n", i);
			return -LIBBPF_ERRNO__FORMAT;
		}

		if (!gelf_getsym(symbols, GELF_R_SYM(rel.r_info), &sym)) {
			pr_warn("struct_ops reloc: symbol %zx not found\n",
				(size_t)GELF_R_SYM(rel.r_info));
			return -LIBBPF_ERRNO__FORMAT;
		}

		name = elf_strptr(obj->efile.elf, obj->efile.strtabidx,
				  sym.st_name) ? : "<?>";
		map = find_struct_ops_map_by_offset(obj, rel.r_offset);
		if (!map) {
			pr_warn("struct_ops reloc: cannot find map at rel.r_offset %zu\n",
				(size_t)rel.r_offset);
			return -EINVAL;
		}

		moff = rel.r_offset - map->sec_offset;
		shdr_idx = sym.st_shndx;
		st_ops = map->st_ops;
		pr_debug("struct_ops reloc %s: for %lld value %lld shdr_idx %u rel.r_offset %zu map->sec_offset %zu name %d (\'%s\')\n",
			 map->name,
			 (long long)(rel.r_info >> 32),
			 (long long)sym.st_value,
			 shdr_idx, (size_t)rel.r_offset,
			 map->sec_offset, sym.st_name, name);

		if (shdr_idx >= SHN_LORESERVE) {
			pr_warn("struct_ops reloc %s: rel.r_offset %zu shdr_idx %u unsupported non-static function\n",
				map->name, (size_t)rel.r_offset, shdr_idx);
			return -LIBBPF_ERRNO__RELOC;
		}

		member = find_member_by_offset(st_ops->type, moff * 8);
		if (!member) {
			pr_warn("struct_ops reloc %s: cannot find member at moff %u\n",
				map->name, moff);
			return -EINVAL;
		}
		member_idx = member - btf_members(st_ops->type);
		name = btf__name_by_offset(btf, member->name_off);

		if (!resolve_func_ptr(btf, member->type, NULL)) {
			pr_warn("struct_ops reloc %s: cannot relocate non func ptr %s\n",
				map->name, name);
			return -EINVAL;
		}

		prog = bpf_object__find_prog_by_idx(obj, shdr_idx);
		if (!prog) {
			pr_warn("struct_ops reloc %s: cannot find prog at shdr_idx %u to relocate func ptr %s\n",
				map->name, shdr_idx, name);
			return -EINVAL;
		}

		if (prog->type == BPF_PROG_TYPE_UNSPEC) {
			const struct bpf_sec_def *sec_def;

			sec_def = find_sec_def(prog->section_name);
			if (sec_def &&
			    sec_def->prog_type != BPF_PROG_TYPE_STRUCT_OPS) {
				/* for pr_warn */
				prog->type = sec_def->prog_type;
				goto invalid_prog;
			}

			prog->type = BPF_PROG_TYPE_STRUCT_OPS;
			prog->attach_btf_id = st_ops->type_id;
			prog->expected_attach_type = member_idx;
		} else if (prog->type != BPF_PROG_TYPE_STRUCT_OPS ||
			   prog->attach_btf_id != st_ops->type_id ||
			   prog->expected_attach_type != member_idx) {
			goto invalid_prog;
		}
		st_ops->progs[member_idx] = prog;
	}

	return 0;

invalid_prog:
	pr_warn("struct_ops reloc %s: cannot use prog %s in sec %s with type %u attach_btf_id %u expected_attach_type %u for func ptr %s\n",
		map->name, prog->name, prog->section_name, prog->type,
		prog->attach_btf_id, prog->expected_attach_type, name);
	return -EINVAL;
}

7195
#define BTF_TRACE_PREFIX "btf_trace_"
7196
#define BTF_LSM_PREFIX "bpf_lsm_"
7197
#define BTF_ITER_PREFIX "bpf_iter_"
7198 7199 7200 7201 7202 7203 7204 7205 7206 7207 7208 7209 7210 7211 7212 7213 7214 7215 7216 7217 7218 7219 7220 7221 7222 7223 7224
#define BTF_MAX_NAME_SIZE 128

static int find_btf_by_prefix_kind(const struct btf *btf, const char *prefix,
				   const char *name, __u32 kind)
{
	char btf_type_name[BTF_MAX_NAME_SIZE];
	int ret;

	ret = snprintf(btf_type_name, sizeof(btf_type_name),
		       "%s%s", prefix, name);
	/* snprintf returns the number of characters written excluding the
	 * the terminating null. So, if >= BTF_MAX_NAME_SIZE are written, it
	 * indicates truncation.
	 */
	if (ret < 0 || ret >= sizeof(btf_type_name))
		return -ENAMETOOLONG;
	return btf__find_by_name_kind(btf, btf_type_name, kind);
}

static inline int __find_vmlinux_btf_id(struct btf *btf, const char *name,
					enum bpf_attach_type attach_type)
{
	int err;

	if (attach_type == BPF_TRACE_RAW_TP)
		err = find_btf_by_prefix_kind(btf, BTF_TRACE_PREFIX, name,
					      BTF_KIND_TYPEDEF);
7225 7226 7227
	else if (attach_type == BPF_LSM_MAC)
		err = find_btf_by_prefix_kind(btf, BTF_LSM_PREFIX, name,
					      BTF_KIND_FUNC);
7228 7229 7230
	else if (attach_type == BPF_TRACE_ITER)
		err = find_btf_by_prefix_kind(btf, BTF_ITER_PREFIX, name,
					      BTF_KIND_FUNC);
7231 7232 7233
	else
		err = btf__find_by_name_kind(btf, name, BTF_KIND_FUNC);

7234 7235 7236
	if (err <= 0)
		pr_warn("%s is not found in vmlinux BTF\n", name);

7237 7238 7239
	return err;
}

7240 7241
int libbpf_find_vmlinux_btf_id(const char *name,
			       enum bpf_attach_type attach_type)
7242
{
7243
	struct btf *btf;
7244
	int err;
7245

7246
	btf = libbpf_find_kernel_btf();
7247 7248 7249 7250 7251
	if (IS_ERR(btf)) {
		pr_warn("vmlinux BTF is not found\n");
		return -EINVAL;
	}

7252 7253 7254
	err = __find_vmlinux_btf_id(btf, name, attach_type);
	btf__free(btf);
	return err;
7255 7256
}

7257 7258 7259 7260 7261 7262 7263 7264 7265 7266 7267 7268 7269 7270 7271 7272 7273 7274 7275 7276 7277 7278 7279 7280 7281 7282 7283 7284 7285 7286 7287 7288 7289
static int libbpf_find_prog_btf_id(const char *name, __u32 attach_prog_fd)
{
	struct bpf_prog_info_linear *info_linear;
	struct bpf_prog_info *info;
	struct btf *btf = NULL;
	int err = -EINVAL;

	info_linear = bpf_program__get_prog_info_linear(attach_prog_fd, 0);
	if (IS_ERR_OR_NULL(info_linear)) {
		pr_warn("failed get_prog_info_linear for FD %d\n",
			attach_prog_fd);
		return -EINVAL;
	}
	info = &info_linear->info;
	if (!info->btf_id) {
		pr_warn("The target program doesn't have BTF\n");
		goto out;
	}
	if (btf__get_from_id(info->btf_id, &btf)) {
		pr_warn("Failed to get BTF of the program\n");
		goto out;
	}
	err = btf__find_by_name_kind(btf, name, BTF_KIND_FUNC);
	btf__free(btf);
	if (err <= 0) {
		pr_warn("%s is not found in prog's BTF\n", name);
		goto out;
	}
out:
	free(info_linear);
	return err;
}

7290
static int libbpf_find_attach_btf_id(struct bpf_program *prog)
7291
{
7292 7293 7294
	enum bpf_attach_type attach_type = prog->expected_attach_type;
	__u32 attach_prog_fd = prog->attach_prog_fd;
	const char *name = prog->section_name;
7295 7296
	int i, err;

7297
	if (!name)
7298
		return -EINVAL;
7299

7300 7301
	for (i = 0; i < ARRAY_SIZE(section_defs); i++) {
		if (!section_defs[i].is_attach_btf)
7302
			continue;
7303
		if (strncmp(name, section_defs[i].sec, section_defs[i].len))
7304
			continue;
7305
		if (attach_prog_fd)
7306
			err = libbpf_find_prog_btf_id(name + section_defs[i].len,
7307 7308
						      attach_prog_fd);
		else
7309 7310 7311
			err = __find_vmlinux_btf_id(prog->obj->btf_vmlinux,
						    name + section_defs[i].len,
						    attach_type);
7312
		return err;
7313 7314
	}
	pr_warn("failed to identify btf_id based on ELF section name '%s'\n", name);
7315
	return -ESRCH;
7316 7317
}

7318 7319 7320
int libbpf_attach_type_by_name(const char *name,
			       enum bpf_attach_type *attach_type)
{
7321
	char *type_names;
7322 7323 7324 7325 7326
	int i;

	if (!name)
		return -EINVAL;

7327 7328
	for (i = 0; i < ARRAY_SIZE(section_defs); i++) {
		if (strncmp(name, section_defs[i].sec, section_defs[i].len))
7329
			continue;
7330
		if (!section_defs[i].is_attachable)
7331
			return -EINVAL;
7332
		*attach_type = section_defs[i].expected_attach_type;
7333 7334
		return 0;
	}
7335
	pr_debug("failed to guess attach type based on ELF section name '%s'\n", name);
7336 7337
	type_names = libbpf_get_type_names(true);
	if (type_names != NULL) {
7338
		pr_debug("attachable section(type) names are:%s\n", type_names);
7339 7340 7341
		free(type_names);
	}

7342 7343 7344
	return -EINVAL;
}

A
Andrii Nakryiko 已提交
7345
int bpf_map__fd(const struct bpf_map *map)
7346
{
7347
	return map ? map->fd : -EINVAL;
7348 7349
}

A
Andrii Nakryiko 已提交
7350
const struct bpf_map_def *bpf_map__def(const struct bpf_map *map)
7351
{
7352
	return map ? &map->def : ERR_PTR(-EINVAL);
7353 7354
}

A
Andrii Nakryiko 已提交
7355
const char *bpf_map__name(const struct bpf_map *map)
7356
{
7357
	return map ? map->name : NULL;
7358 7359
}

7360 7361 7362 7363 7364 7365 7366 7367 7368 7369 7370 7371 7372 7373 7374 7375 7376 7377 7378 7379 7380 7381 7382 7383 7384 7385 7386 7387 7388 7389 7390 7391 7392 7393 7394 7395 7396 7397 7398 7399 7400 7401 7402 7403 7404 7405 7406 7407 7408 7409 7410 7411 7412 7413 7414 7415 7416 7417 7418 7419 7420 7421 7422 7423 7424
enum bpf_map_type bpf_map__type(const struct bpf_map *map)
{
	return map->def.type;
}

int bpf_map__set_type(struct bpf_map *map, enum bpf_map_type type)
{
	if (map->fd >= 0)
		return -EBUSY;
	map->def.type = type;
	return 0;
}

__u32 bpf_map__map_flags(const struct bpf_map *map)
{
	return map->def.map_flags;
}

int bpf_map__set_map_flags(struct bpf_map *map, __u32 flags)
{
	if (map->fd >= 0)
		return -EBUSY;
	map->def.map_flags = flags;
	return 0;
}

__u32 bpf_map__numa_node(const struct bpf_map *map)
{
	return map->numa_node;
}

int bpf_map__set_numa_node(struct bpf_map *map, __u32 numa_node)
{
	if (map->fd >= 0)
		return -EBUSY;
	map->numa_node = numa_node;
	return 0;
}

__u32 bpf_map__key_size(const struct bpf_map *map)
{
	return map->def.key_size;
}

int bpf_map__set_key_size(struct bpf_map *map, __u32 size)
{
	if (map->fd >= 0)
		return -EBUSY;
	map->def.key_size = size;
	return 0;
}

__u32 bpf_map__value_size(const struct bpf_map *map)
{
	return map->def.value_size;
}

int bpf_map__set_value_size(struct bpf_map *map, __u32 size)
{
	if (map->fd >= 0)
		return -EBUSY;
	map->def.value_size = size;
	return 0;
}

7425
__u32 bpf_map__btf_key_type_id(const struct bpf_map *map)
7426
{
7427
	return map ? map->btf_key_type_id : 0;
7428 7429
}

7430
__u32 bpf_map__btf_value_type_id(const struct bpf_map *map)
7431
{
7432
	return map ? map->btf_value_type_id : 0;
7433 7434
}

7435 7436
int bpf_map__set_priv(struct bpf_map *map, void *priv,
		     bpf_map_clear_priv_t clear_priv)
7437 7438 7439 7440 7441 7442 7443 7444 7445 7446 7447 7448 7449 7450
{
	if (!map)
		return -EINVAL;

	if (map->priv) {
		if (map->clear_priv)
			map->clear_priv(map, map->priv);
	}

	map->priv = priv;
	map->clear_priv = clear_priv;
	return 0;
}

A
Andrii Nakryiko 已提交
7451
void *bpf_map__priv(const struct bpf_map *map)
7452
{
7453
	return map ? map->priv : ERR_PTR(-EINVAL);
7454 7455
}

7456 7457 7458 7459 7460 7461 7462 7463 7464 7465 7466
int bpf_map__set_initial_value(struct bpf_map *map,
			       const void *data, size_t size)
{
	if (!map->mmaped || map->libbpf_type == LIBBPF_MAP_KCONFIG ||
	    size != map->def.value_size || map->fd >= 0)
		return -EINVAL;

	memcpy(map->mmaped, data, size);
	return 0;
}

A
Andrii Nakryiko 已提交
7467
bool bpf_map__is_offload_neutral(const struct bpf_map *map)
7468 7469 7470 7471
{
	return map->def.type == BPF_MAP_TYPE_PERF_EVENT_ARRAY;
}

A
Andrii Nakryiko 已提交
7472
bool bpf_map__is_internal(const struct bpf_map *map)
7473 7474 7475 7476
{
	return map->libbpf_type != LIBBPF_MAP_UNSPEC;
}

7477 7478 7479 7480 7481 7482
__u32 bpf_map__ifindex(const struct bpf_map *map)
{
	return map->map_ifindex;
}

int bpf_map__set_ifindex(struct bpf_map *map, __u32 ifindex)
7483
{
7484 7485
	if (map->fd >= 0)
		return -EBUSY;
7486
	map->map_ifindex = ifindex;
7487
	return 0;
7488 7489
}

7490 7491 7492
int bpf_map__set_inner_map_fd(struct bpf_map *map, int fd)
{
	if (!bpf_map_type__is_map_in_map(map->def.type)) {
7493
		pr_warn("error: unsupported map type\n");
7494 7495 7496
		return -EINVAL;
	}
	if (map->inner_map_fd != -1) {
7497
		pr_warn("error: inner_map_fd already specified\n");
7498 7499 7500 7501 7502 7503
		return -EINVAL;
	}
	map->inner_map_fd = fd;
	return 0;
}

7504
static struct bpf_map *
A
Andrii Nakryiko 已提交
7505
__bpf_map__iter(const struct bpf_map *m, const struct bpf_object *obj, int i)
7506
{
7507
	ssize_t idx;
7508 7509 7510 7511 7512 7513 7514 7515
	struct bpf_map *s, *e;

	if (!obj || !obj->maps)
		return NULL;

	s = obj->maps;
	e = obj->maps + obj->nr_maps;

7516
	if ((m < s) || (m >= e)) {
7517 7518
		pr_warn("error in %s: map handler doesn't belong to object\n",
			 __func__);
7519 7520 7521
		return NULL;
	}

7522 7523
	idx = (m - obj->maps) + i;
	if (idx >= obj->nr_maps || idx < 0)
7524 7525 7526
		return NULL;
	return &obj->maps[idx];
}
7527

7528
struct bpf_map *
A
Andrii Nakryiko 已提交
7529
bpf_map__next(const struct bpf_map *prev, const struct bpf_object *obj)
7530 7531 7532 7533 7534 7535 7536 7537
{
	if (prev == NULL)
		return obj->maps;

	return __bpf_map__iter(prev, obj, 1);
}

struct bpf_map *
A
Andrii Nakryiko 已提交
7538
bpf_map__prev(const struct bpf_map *next, const struct bpf_object *obj)
7539 7540 7541 7542 7543 7544 7545 7546 7547 7548
{
	if (next == NULL) {
		if (!obj->nr_maps)
			return NULL;
		return obj->maps + obj->nr_maps - 1;
	}

	return __bpf_map__iter(next, obj, -1);
}

7549
struct bpf_map *
A
Andrii Nakryiko 已提交
7550
bpf_object__find_map_by_name(const struct bpf_object *obj, const char *name)
7551 7552 7553
{
	struct bpf_map *pos;

7554
	bpf_object__for_each_map(pos, obj) {
7555
		if (pos->name && !strcmp(pos->name, name))
7556 7557 7558 7559
			return pos;
	}
	return NULL;
}
7560

7561
int
A
Andrii Nakryiko 已提交
7562
bpf_object__find_map_fd_by_name(const struct bpf_object *obj, const char *name)
7563 7564 7565 7566
{
	return bpf_map__fd(bpf_object__find_map_by_name(obj, name));
}

7567 7568 7569
struct bpf_map *
bpf_object__find_map_by_offset(struct bpf_object *obj, size_t offset)
{
7570
	return ERR_PTR(-ENOTSUP);
7571
}
7572 7573 7574

long libbpf_get_error(const void *ptr)
{
7575
	return PTR_ERR_OR_ZERO(ptr);
7576
}
7577 7578 7579

int bpf_prog_load(const char *file, enum bpf_prog_type type,
		  struct bpf_object **pobj, int *prog_fd)
7580 7581 7582 7583 7584 7585 7586 7587 7588 7589 7590 7591 7592
{
	struct bpf_prog_load_attr attr;

	memset(&attr, 0, sizeof(struct bpf_prog_load_attr));
	attr.file = file;
	attr.prog_type = type;
	attr.expected_attach_type = 0;

	return bpf_prog_load_xattr(&attr, pobj, prog_fd);
}

int bpf_prog_load_xattr(const struct bpf_prog_load_attr *attr,
			struct bpf_object **pobj, int *prog_fd)
7593
{
7594
	struct bpf_object_open_attr open_attr = {};
7595
	struct bpf_program *prog, *first_prog = NULL;
7596
	struct bpf_object *obj;
7597
	struct bpf_map *map;
7598 7599
	int err;

7600 7601
	if (!attr)
		return -EINVAL;
7602 7603
	if (!attr->file)
		return -EINVAL;
7604

7605 7606 7607
	open_attr.file = attr->file;
	open_attr.prog_type = attr->prog_type;

7608
	obj = bpf_object__open_xattr(&open_attr);
7609
	if (IS_ERR_OR_NULL(obj))
7610 7611
		return -ENOENT;

7612
	bpf_object__for_each_program(prog, obj) {
7613
		enum bpf_attach_type attach_type = attr->expected_attach_type;
7614
		/*
7615 7616 7617
		 * to preserve backwards compatibility, bpf_prog_load treats
		 * attr->prog_type, if specified, as an override to whatever
		 * bpf_object__open guessed
7618
		 */
7619 7620 7621 7622 7623 7624 7625 7626 7627 7628 7629 7630
		if (attr->prog_type != BPF_PROG_TYPE_UNSPEC) {
			bpf_program__set_type(prog, attr->prog_type);
			bpf_program__set_expected_attach_type(prog,
							      attach_type);
		}
		if (bpf_program__get_type(prog) == BPF_PROG_TYPE_UNSPEC) {
			/*
			 * we haven't guessed from section name and user
			 * didn't provide a fallback type, too bad...
			 */
			bpf_object__close(obj);
			return -EINVAL;
7631
		}
7632

7633
		prog->prog_ifindex = attr->ifindex;
7634
		prog->log_level = attr->log_level;
7635
		prog->prog_flags = attr->prog_flags;
7636
		if (!first_prog)
7637 7638 7639
			first_prog = prog;
	}

7640
	bpf_object__for_each_map(map, obj) {
7641 7642
		if (!bpf_map__is_offload_neutral(map))
			map->map_ifindex = attr->ifindex;
7643 7644
	}

7645
	if (!first_prog) {
7646
		pr_warn("object file doesn't contain bpf program\n");
7647 7648
		bpf_object__close(obj);
		return -ENOENT;
7649 7650
	}

7651 7652 7653
	err = bpf_object__load(obj);
	if (err) {
		bpf_object__close(obj);
7654
		return err;
7655 7656 7657
	}

	*pobj = obj;
7658
	*prog_fd = bpf_program__fd(first_prog);
7659 7660
	return 0;
}
7661

7662
struct bpf_link {
7663
	int (*detach)(struct bpf_link *link);
7664
	int (*destroy)(struct bpf_link *link);
7665 7666
	char *pin_path;		/* NULL, if not pinned */
	int fd;			/* hook FD, -1 if not applicable */
7667
	bool disconnected;
7668 7669
};

7670 7671 7672 7673 7674 7675
/* Replace link's underlying BPF program with the new one */
int bpf_link__update_program(struct bpf_link *link, struct bpf_program *prog)
{
	return bpf_link_update(bpf_link__fd(link), bpf_program__fd(prog), NULL);
}

7676 7677 7678 7679 7680 7681 7682 7683 7684 7685 7686 7687 7688 7689 7690
/* Release "ownership" of underlying BPF resource (typically, BPF program
 * attached to some BPF hook, e.g., tracepoint, kprobe, etc). Disconnected
 * link, when destructed through bpf_link__destroy() call won't attempt to
 * detach/unregisted that BPF resource. This is useful in situations where,
 * say, attached BPF program has to outlive userspace program that attached it
 * in the system. Depending on type of BPF program, though, there might be
 * additional steps (like pinning BPF program in BPF FS) necessary to ensure
 * exit of userspace program doesn't trigger automatic detachment and clean up
 * inside the kernel.
 */
void bpf_link__disconnect(struct bpf_link *link)
{
	link->disconnected = true;
}

7691 7692
int bpf_link__destroy(struct bpf_link *link)
{
7693
	int err = 0;
7694

7695
	if (IS_ERR_OR_NULL(link))
7696 7697
		return 0;

7698 7699 7700 7701
	if (!link->disconnected && link->detach)
		err = link->detach(link);
	if (link->destroy)
		link->destroy(link);
7702 7703
	if (link->pin_path)
		free(link->pin_path);
7704 7705 7706 7707 7708
	free(link);

	return err;
}

7709 7710 7711 7712 7713 7714 7715 7716 7717 7718 7719 7720 7721 7722 7723 7724 7725 7726 7727 7728 7729 7730 7731 7732 7733 7734 7735 7736 7737 7738 7739 7740 7741 7742 7743 7744 7745 7746 7747 7748 7749 7750 7751 7752
int bpf_link__fd(const struct bpf_link *link)
{
	return link->fd;
}

const char *bpf_link__pin_path(const struct bpf_link *link)
{
	return link->pin_path;
}

static int bpf_link__detach_fd(struct bpf_link *link)
{
	return close(link->fd);
}

struct bpf_link *bpf_link__open(const char *path)
{
	struct bpf_link *link;
	int fd;

	fd = bpf_obj_get(path);
	if (fd < 0) {
		fd = -errno;
		pr_warn("failed to open link at %s: %d\n", path, fd);
		return ERR_PTR(fd);
	}

	link = calloc(1, sizeof(*link));
	if (!link) {
		close(fd);
		return ERR_PTR(-ENOMEM);
	}
	link->detach = &bpf_link__detach_fd;
	link->fd = fd;

	link->pin_path = strdup(path);
	if (!link->pin_path) {
		bpf_link__destroy(link);
		return ERR_PTR(-ENOMEM);
	}

	return link;
}

7753 7754 7755 7756 7757
int bpf_link__detach(struct bpf_link *link)
{
	return bpf_link_detach(link->fd) ? -errno : 0;
}

7758 7759 7760 7761 7762 7763 7764 7765 7766 7767 7768 7769 7770 7771 7772 7773 7774 7775 7776 7777 7778 7779 7780 7781 7782 7783 7784 7785 7786 7787 7788 7789 7790 7791 7792 7793 7794 7795 7796 7797 7798 7799
int bpf_link__pin(struct bpf_link *link, const char *path)
{
	int err;

	if (link->pin_path)
		return -EBUSY;
	err = make_parent_dir(path);
	if (err)
		return err;
	err = check_path(path);
	if (err)
		return err;

	link->pin_path = strdup(path);
	if (!link->pin_path)
		return -ENOMEM;

	if (bpf_obj_pin(link->fd, link->pin_path)) {
		err = -errno;
		zfree(&link->pin_path);
		return err;
	}

	pr_debug("link fd=%d: pinned at %s\n", link->fd, link->pin_path);
	return 0;
}

int bpf_link__unpin(struct bpf_link *link)
{
	int err;

	if (!link->pin_path)
		return -EINVAL;

	err = unlink(link->pin_path);
	if (err != 0)
		return -errno;

	pr_debug("link fd=%d: unpinned from %s\n", link->fd, link->pin_path);
	zfree(&link->pin_path);
	return 0;
}
7800

7801
static int bpf_link__detach_perf_event(struct bpf_link *link)
7802 7803 7804
{
	int err;

7805
	err = ioctl(link->fd, PERF_EVENT_IOC_DISABLE, 0);
7806 7807 7808
	if (err)
		err = -errno;

7809
	close(link->fd);
7810 7811 7812 7813 7814 7815 7816
	return err;
}

struct bpf_link *bpf_program__attach_perf_event(struct bpf_program *prog,
						int pfd)
{
	char errmsg[STRERR_BUFSIZE];
7817
	struct bpf_link *link;
7818 7819 7820
	int prog_fd, err;

	if (pfd < 0) {
7821 7822
		pr_warn("program '%s': invalid perf event FD %d\n",
			bpf_program__title(prog, false), pfd);
7823 7824 7825 7826
		return ERR_PTR(-EINVAL);
	}
	prog_fd = bpf_program__fd(prog);
	if (prog_fd < 0) {
7827 7828
		pr_warn("program '%s': can't attach BPF program w/o FD (did you load it?)\n",
			bpf_program__title(prog, false));
7829 7830 7831
		return ERR_PTR(-EINVAL);
	}

7832
	link = calloc(1, sizeof(*link));
7833 7834
	if (!link)
		return ERR_PTR(-ENOMEM);
7835
	link->detach = &bpf_link__detach_perf_event;
7836 7837 7838 7839 7840
	link->fd = pfd;

	if (ioctl(pfd, PERF_EVENT_IOC_SET_BPF, prog_fd) < 0) {
		err = -errno;
		free(link);
7841 7842
		pr_warn("program '%s': failed to attach to pfd %d: %s\n",
			bpf_program__title(prog, false), pfd,
7843
			   libbpf_strerror_r(err, errmsg, sizeof(errmsg)));
7844 7845 7846
		if (err == -EPROTO)
			pr_warn("program '%s': try add PERF_SAMPLE_CALLCHAIN to or remove exclude_callchain_[kernel|user] from pfd %d\n",
				bpf_program__title(prog, false), pfd);
7847 7848 7849 7850 7851
		return ERR_PTR(err);
	}
	if (ioctl(pfd, PERF_EVENT_IOC_ENABLE, 0) < 0) {
		err = -errno;
		free(link);
7852 7853
		pr_warn("program '%s': failed to enable pfd %d: %s\n",
			bpf_program__title(prog, false), pfd,
7854 7855 7856
			   libbpf_strerror_r(err, errmsg, sizeof(errmsg)));
		return ERR_PTR(err);
	}
7857
	return link;
7858 7859
}

7860 7861 7862 7863 7864 7865 7866 7867 7868 7869 7870 7871 7872 7873 7874 7875 7876 7877 7878 7879 7880 7881 7882 7883 7884 7885 7886 7887 7888 7889 7890 7891 7892 7893 7894 7895 7896 7897 7898 7899 7900 7901 7902 7903 7904 7905 7906 7907 7908 7909 7910 7911 7912 7913 7914 7915 7916 7917 7918 7919 7920 7921 7922 7923 7924 7925 7926 7927
/*
 * this function is expected to parse integer in the range of [0, 2^31-1] from
 * given file using scanf format string fmt. If actual parsed value is
 * negative, the result might be indistinguishable from error
 */
static int parse_uint_from_file(const char *file, const char *fmt)
{
	char buf[STRERR_BUFSIZE];
	int err, ret;
	FILE *f;

	f = fopen(file, "r");
	if (!f) {
		err = -errno;
		pr_debug("failed to open '%s': %s\n", file,
			 libbpf_strerror_r(err, buf, sizeof(buf)));
		return err;
	}
	err = fscanf(f, fmt, &ret);
	if (err != 1) {
		err = err == EOF ? -EIO : -errno;
		pr_debug("failed to parse '%s': %s\n", file,
			libbpf_strerror_r(err, buf, sizeof(buf)));
		fclose(f);
		return err;
	}
	fclose(f);
	return ret;
}

static int determine_kprobe_perf_type(void)
{
	const char *file = "/sys/bus/event_source/devices/kprobe/type";

	return parse_uint_from_file(file, "%d\n");
}

static int determine_uprobe_perf_type(void)
{
	const char *file = "/sys/bus/event_source/devices/uprobe/type";

	return parse_uint_from_file(file, "%d\n");
}

static int determine_kprobe_retprobe_bit(void)
{
	const char *file = "/sys/bus/event_source/devices/kprobe/format/retprobe";

	return parse_uint_from_file(file, "config:%d\n");
}

static int determine_uprobe_retprobe_bit(void)
{
	const char *file = "/sys/bus/event_source/devices/uprobe/format/retprobe";

	return parse_uint_from_file(file, "config:%d\n");
}

static int perf_event_open_probe(bool uprobe, bool retprobe, const char *name,
				 uint64_t offset, int pid)
{
	struct perf_event_attr attr = {};
	char errmsg[STRERR_BUFSIZE];
	int type, pfd, err;

	type = uprobe ? determine_uprobe_perf_type()
		      : determine_kprobe_perf_type();
	if (type < 0) {
7928 7929 7930
		pr_warn("failed to determine %s perf type: %s\n",
			uprobe ? "uprobe" : "kprobe",
			libbpf_strerror_r(type, errmsg, sizeof(errmsg)));
7931 7932 7933 7934 7935 7936 7937
		return type;
	}
	if (retprobe) {
		int bit = uprobe ? determine_uprobe_retprobe_bit()
				 : determine_kprobe_retprobe_bit();

		if (bit < 0) {
7938 7939 7940
			pr_warn("failed to determine %s retprobe bit: %s\n",
				uprobe ? "uprobe" : "kprobe",
				libbpf_strerror_r(bit, errmsg, sizeof(errmsg)));
7941 7942 7943 7944 7945 7946
			return bit;
		}
		attr.config |= 1 << bit;
	}
	attr.size = sizeof(attr);
	attr.type = type;
7947 7948
	attr.config1 = ptr_to_u64(name); /* kprobe_func or uprobe_path */
	attr.config2 = offset;		 /* kprobe_addr or probe_offset */
7949 7950 7951 7952 7953 7954 7955 7956

	/* pid filter is meaningful only for uprobes */
	pfd = syscall(__NR_perf_event_open, &attr,
		      pid < 0 ? -1 : pid /* pid */,
		      pid == -1 ? 0 : -1 /* cpu */,
		      -1 /* group_fd */, PERF_FLAG_FD_CLOEXEC);
	if (pfd < 0) {
		err = -errno;
7957 7958 7959
		pr_warn("%s perf_event_open() failed: %s\n",
			uprobe ? "uprobe" : "kprobe",
			libbpf_strerror_r(err, errmsg, sizeof(errmsg)));
7960 7961 7962 7963 7964 7965 7966 7967 7968 7969 7970 7971 7972 7973 7974 7975
		return err;
	}
	return pfd;
}

struct bpf_link *bpf_program__attach_kprobe(struct bpf_program *prog,
					    bool retprobe,
					    const char *func_name)
{
	char errmsg[STRERR_BUFSIZE];
	struct bpf_link *link;
	int pfd, err;

	pfd = perf_event_open_probe(false /* uprobe */, retprobe, func_name,
				    0 /* offset */, -1 /* pid */);
	if (pfd < 0) {
7976 7977 7978 7979
		pr_warn("program '%s': failed to create %s '%s' perf event: %s\n",
			bpf_program__title(prog, false),
			retprobe ? "kretprobe" : "kprobe", func_name,
			libbpf_strerror_r(pfd, errmsg, sizeof(errmsg)));
7980 7981 7982 7983 7984 7985
		return ERR_PTR(pfd);
	}
	link = bpf_program__attach_perf_event(prog, pfd);
	if (IS_ERR(link)) {
		close(pfd);
		err = PTR_ERR(link);
7986 7987 7988 7989
		pr_warn("program '%s': failed to attach to %s '%s': %s\n",
			bpf_program__title(prog, false),
			retprobe ? "kretprobe" : "kprobe", func_name,
			libbpf_strerror_r(err, errmsg, sizeof(errmsg)));
7990 7991 7992 7993 7994
		return link;
	}
	return link;
}

7995 7996 7997 7998 7999 8000 8001 8002 8003 8004 8005 8006
static struct bpf_link *attach_kprobe(const struct bpf_sec_def *sec,
				      struct bpf_program *prog)
{
	const char *func_name;
	bool retprobe;

	func_name = bpf_program__title(prog, false) + sec->len;
	retprobe = strcmp(sec->sec, "kretprobe/") == 0;

	return bpf_program__attach_kprobe(prog, retprobe, func_name);
}

8007 8008 8009 8010 8011 8012 8013 8014 8015 8016 8017 8018
struct bpf_link *bpf_program__attach_uprobe(struct bpf_program *prog,
					    bool retprobe, pid_t pid,
					    const char *binary_path,
					    size_t func_offset)
{
	char errmsg[STRERR_BUFSIZE];
	struct bpf_link *link;
	int pfd, err;

	pfd = perf_event_open_probe(true /* uprobe */, retprobe,
				    binary_path, func_offset, pid);
	if (pfd < 0) {
8019 8020 8021 8022 8023
		pr_warn("program '%s': failed to create %s '%s:0x%zx' perf event: %s\n",
			bpf_program__title(prog, false),
			retprobe ? "uretprobe" : "uprobe",
			binary_path, func_offset,
			libbpf_strerror_r(pfd, errmsg, sizeof(errmsg)));
8024 8025 8026 8027 8028 8029
		return ERR_PTR(pfd);
	}
	link = bpf_program__attach_perf_event(prog, pfd);
	if (IS_ERR(link)) {
		close(pfd);
		err = PTR_ERR(link);
8030 8031 8032 8033 8034
		pr_warn("program '%s': failed to attach to %s '%s:0x%zx': %s\n",
			bpf_program__title(prog, false),
			retprobe ? "uretprobe" : "uprobe",
			binary_path, func_offset,
			libbpf_strerror_r(err, errmsg, sizeof(errmsg)));
8035 8036 8037 8038 8039
		return link;
	}
	return link;
}

8040 8041 8042 8043 8044 8045 8046 8047 8048 8049 8050 8051 8052 8053 8054 8055 8056 8057 8058 8059 8060 8061 8062 8063 8064 8065 8066 8067
static int determine_tracepoint_id(const char *tp_category,
				   const char *tp_name)
{
	char file[PATH_MAX];
	int ret;

	ret = snprintf(file, sizeof(file),
		       "/sys/kernel/debug/tracing/events/%s/%s/id",
		       tp_category, tp_name);
	if (ret < 0)
		return -errno;
	if (ret >= sizeof(file)) {
		pr_debug("tracepoint %s/%s path is too long\n",
			 tp_category, tp_name);
		return -E2BIG;
	}
	return parse_uint_from_file(file, "%d\n");
}

static int perf_event_open_tracepoint(const char *tp_category,
				      const char *tp_name)
{
	struct perf_event_attr attr = {};
	char errmsg[STRERR_BUFSIZE];
	int tp_id, pfd, err;

	tp_id = determine_tracepoint_id(tp_category, tp_name);
	if (tp_id < 0) {
8068 8069 8070
		pr_warn("failed to determine tracepoint '%s/%s' perf event ID: %s\n",
			tp_category, tp_name,
			libbpf_strerror_r(tp_id, errmsg, sizeof(errmsg)));
8071 8072 8073 8074 8075 8076 8077 8078 8079 8080 8081
		return tp_id;
	}

	attr.type = PERF_TYPE_TRACEPOINT;
	attr.size = sizeof(attr);
	attr.config = tp_id;

	pfd = syscall(__NR_perf_event_open, &attr, -1 /* pid */, 0 /* cpu */,
		      -1 /* group_fd */, PERF_FLAG_FD_CLOEXEC);
	if (pfd < 0) {
		err = -errno;
8082 8083 8084
		pr_warn("tracepoint '%s/%s' perf_event_open() failed: %s\n",
			tp_category, tp_name,
			libbpf_strerror_r(err, errmsg, sizeof(errmsg)));
8085 8086 8087 8088 8089 8090 8091 8092 8093 8094 8095 8096 8097 8098 8099
		return err;
	}
	return pfd;
}

struct bpf_link *bpf_program__attach_tracepoint(struct bpf_program *prog,
						const char *tp_category,
						const char *tp_name)
{
	char errmsg[STRERR_BUFSIZE];
	struct bpf_link *link;
	int pfd, err;

	pfd = perf_event_open_tracepoint(tp_category, tp_name);
	if (pfd < 0) {
8100 8101 8102 8103
		pr_warn("program '%s': failed to create tracepoint '%s/%s' perf event: %s\n",
			bpf_program__title(prog, false),
			tp_category, tp_name,
			libbpf_strerror_r(pfd, errmsg, sizeof(errmsg)));
8104 8105 8106 8107 8108 8109
		return ERR_PTR(pfd);
	}
	link = bpf_program__attach_perf_event(prog, pfd);
	if (IS_ERR(link)) {
		close(pfd);
		err = PTR_ERR(link);
8110 8111 8112 8113
		pr_warn("program '%s': failed to attach to tracepoint '%s/%s': %s\n",
			bpf_program__title(prog, false),
			tp_category, tp_name,
			libbpf_strerror_r(err, errmsg, sizeof(errmsg)));
8114 8115 8116 8117 8118
		return link;
	}
	return link;
}

8119 8120 8121 8122 8123 8124 8125 8126 8127 8128 8129 8130 8131 8132 8133 8134 8135 8136 8137 8138 8139 8140 8141 8142 8143 8144
static struct bpf_link *attach_tp(const struct bpf_sec_def *sec,
				  struct bpf_program *prog)
{
	char *sec_name, *tp_cat, *tp_name;
	struct bpf_link *link;

	sec_name = strdup(bpf_program__title(prog, false));
	if (!sec_name)
		return ERR_PTR(-ENOMEM);

	/* extract "tp/<category>/<name>" */
	tp_cat = sec_name + sec->len;
	tp_name = strchr(tp_cat, '/');
	if (!tp_name) {
		link = ERR_PTR(-EINVAL);
		goto out;
	}
	*tp_name = '\0';
	tp_name++;

	link = bpf_program__attach_tracepoint(prog, tp_cat, tp_name);
out:
	free(sec_name);
	return link;
}

8145 8146 8147 8148
struct bpf_link *bpf_program__attach_raw_tracepoint(struct bpf_program *prog,
						    const char *tp_name)
{
	char errmsg[STRERR_BUFSIZE];
8149
	struct bpf_link *link;
8150 8151 8152 8153
	int prog_fd, pfd;

	prog_fd = bpf_program__fd(prog);
	if (prog_fd < 0) {
8154 8155
		pr_warn("program '%s': can't attach before loaded\n",
			bpf_program__title(prog, false));
8156 8157 8158
		return ERR_PTR(-EINVAL);
	}

8159
	link = calloc(1, sizeof(*link));
8160 8161
	if (!link)
		return ERR_PTR(-ENOMEM);
8162
	link->detach = &bpf_link__detach_fd;
8163 8164 8165 8166 8167

	pfd = bpf_raw_tracepoint_open(tp_name, prog_fd);
	if (pfd < 0) {
		pfd = -errno;
		free(link);
8168 8169 8170
		pr_warn("program '%s': failed to attach to raw tracepoint '%s': %s\n",
			bpf_program__title(prog, false), tp_name,
			libbpf_strerror_r(pfd, errmsg, sizeof(errmsg)));
8171 8172 8173
		return ERR_PTR(pfd);
	}
	link->fd = pfd;
8174
	return link;
8175 8176
}

8177 8178 8179 8180 8181 8182 8183 8184
static struct bpf_link *attach_raw_tp(const struct bpf_sec_def *sec,
				      struct bpf_program *prog)
{
	const char *tp_name = bpf_program__title(prog, false) + sec->len;

	return bpf_program__attach_raw_tracepoint(prog, tp_name);
}

8185 8186
/* Common logic for all BPF program types that attach to a btf_id */
static struct bpf_link *bpf_program__attach_btf_id(struct bpf_program *prog)
8187 8188
{
	char errmsg[STRERR_BUFSIZE];
8189
	struct bpf_link *link;
8190 8191 8192 8193 8194 8195 8196 8197 8198
	int prog_fd, pfd;

	prog_fd = bpf_program__fd(prog);
	if (prog_fd < 0) {
		pr_warn("program '%s': can't attach before loaded\n",
			bpf_program__title(prog, false));
		return ERR_PTR(-EINVAL);
	}

8199
	link = calloc(1, sizeof(*link));
8200 8201
	if (!link)
		return ERR_PTR(-ENOMEM);
8202
	link->detach = &bpf_link__detach_fd;
8203 8204 8205 8206 8207

	pfd = bpf_raw_tracepoint_open(NULL, prog_fd);
	if (pfd < 0) {
		pfd = -errno;
		free(link);
8208
		pr_warn("program '%s': failed to attach: %s\n",
8209 8210 8211 8212 8213 8214 8215 8216
			bpf_program__title(prog, false),
			libbpf_strerror_r(pfd, errmsg, sizeof(errmsg)));
		return ERR_PTR(pfd);
	}
	link->fd = pfd;
	return (struct bpf_link *)link;
}

8217 8218 8219 8220 8221 8222 8223 8224 8225 8226
struct bpf_link *bpf_program__attach_trace(struct bpf_program *prog)
{
	return bpf_program__attach_btf_id(prog);
}

struct bpf_link *bpf_program__attach_lsm(struct bpf_program *prog)
{
	return bpf_program__attach_btf_id(prog);
}

8227 8228 8229 8230 8231 8232
static struct bpf_link *attach_trace(const struct bpf_sec_def *sec,
				     struct bpf_program *prog)
{
	return bpf_program__attach_trace(prog);
}

8233 8234 8235 8236 8237 8238
static struct bpf_link *attach_lsm(const struct bpf_sec_def *sec,
				   struct bpf_program *prog)
{
	return bpf_program__attach_lsm(prog);
}

8239 8240 8241 8242 8243 8244
static struct bpf_link *attach_iter(const struct bpf_sec_def *sec,
				    struct bpf_program *prog)
{
	return bpf_program__attach_iter(prog, NULL);
}

8245 8246 8247
static struct bpf_link *
bpf_program__attach_fd(struct bpf_program *prog, int target_fd,
		       const char *target_name)
8248 8249 8250 8251 8252 8253 8254 8255 8256 8257 8258 8259 8260 8261 8262 8263 8264 8265 8266
{
	enum bpf_attach_type attach_type;
	char errmsg[STRERR_BUFSIZE];
	struct bpf_link *link;
	int prog_fd, link_fd;

	prog_fd = bpf_program__fd(prog);
	if (prog_fd < 0) {
		pr_warn("program '%s': can't attach before loaded\n",
			bpf_program__title(prog, false));
		return ERR_PTR(-EINVAL);
	}

	link = calloc(1, sizeof(*link));
	if (!link)
		return ERR_PTR(-ENOMEM);
	link->detach = &bpf_link__detach_fd;

	attach_type = bpf_program__get_expected_attach_type(prog);
8267
	link_fd = bpf_link_create(prog_fd, target_fd, attach_type, NULL);
8268 8269 8270
	if (link_fd < 0) {
		link_fd = -errno;
		free(link);
8271 8272
		pr_warn("program '%s': failed to attach to %s: %s\n",
			bpf_program__title(prog, false), target_name,
8273 8274 8275 8276 8277 8278 8279
			libbpf_strerror_r(link_fd, errmsg, sizeof(errmsg)));
		return ERR_PTR(link_fd);
	}
	link->fd = link_fd;
	return link;
}

8280 8281 8282 8283 8284 8285 8286 8287 8288 8289 8290 8291
struct bpf_link *
bpf_program__attach_cgroup(struct bpf_program *prog, int cgroup_fd)
{
	return bpf_program__attach_fd(prog, cgroup_fd, "cgroup");
}

struct bpf_link *
bpf_program__attach_netns(struct bpf_program *prog, int netns_fd)
{
	return bpf_program__attach_fd(prog, netns_fd, "netns");
}

8292 8293 8294 8295 8296 8297
struct bpf_link *bpf_program__attach_xdp(struct bpf_program *prog, int ifindex)
{
	/* target_fd/target_ifindex use the same field in LINK_CREATE */
	return bpf_program__attach_fd(prog, ifindex, "xdp");
}

8298 8299 8300 8301
struct bpf_link *
bpf_program__attach_iter(struct bpf_program *prog,
			 const struct bpf_iter_attach_opts *opts)
{
8302
	DECLARE_LIBBPF_OPTS(bpf_link_create_opts, link_create_opts);
8303 8304 8305
	char errmsg[STRERR_BUFSIZE];
	struct bpf_link *link;
	int prog_fd, link_fd;
8306
	__u32 target_fd = 0;
8307 8308 8309 8310

	if (!OPTS_VALID(opts, bpf_iter_attach_opts))
		return ERR_PTR(-EINVAL);

8311 8312
	link_create_opts.iter_info = OPTS_GET(opts, link_info, (void *)0);
	link_create_opts.iter_info_len = OPTS_GET(opts, link_info_len, 0);
8313

8314 8315 8316 8317 8318 8319 8320 8321 8322 8323 8324 8325
	prog_fd = bpf_program__fd(prog);
	if (prog_fd < 0) {
		pr_warn("program '%s': can't attach before loaded\n",
			bpf_program__title(prog, false));
		return ERR_PTR(-EINVAL);
	}

	link = calloc(1, sizeof(*link));
	if (!link)
		return ERR_PTR(-ENOMEM);
	link->detach = &bpf_link__detach_fd;

8326 8327
	link_fd = bpf_link_create(prog_fd, target_fd, BPF_TRACE_ITER,
				  &link_create_opts);
8328 8329 8330 8331 8332 8333 8334 8335 8336 8337 8338 8339
	if (link_fd < 0) {
		link_fd = -errno;
		free(link);
		pr_warn("program '%s': failed to attach to iterator: %s\n",
			bpf_program__title(prog, false),
			libbpf_strerror_r(link_fd, errmsg, sizeof(errmsg)));
		return ERR_PTR(link_fd);
	}
	link->fd = link_fd;
	return link;
}

8340 8341 8342 8343 8344 8345 8346 8347 8348 8349 8350
struct bpf_link *bpf_program__attach(struct bpf_program *prog)
{
	const struct bpf_sec_def *sec_def;

	sec_def = find_sec_def(bpf_program__title(prog, false));
	if (!sec_def || !sec_def->attach_fn)
		return ERR_PTR(-ESRCH);

	return sec_def->attach_fn(sec_def, prog);
}

8351 8352 8353 8354
static int bpf_link__detach_struct_ops(struct bpf_link *link)
{
	__u32 zero = 0;

8355
	if (bpf_map_delete_elem(link->fd, &zero))
8356 8357 8358 8359 8360 8361 8362 8363
		return -errno;

	return 0;
}

struct bpf_link *bpf_map__attach_struct_ops(struct bpf_map *map)
{
	struct bpf_struct_ops *st_ops;
8364
	struct bpf_link *link;
8365 8366 8367 8368 8369 8370 8371 8372 8373 8374 8375 8376 8377 8378 8379 8380 8381 8382 8383 8384 8385 8386 8387 8388 8389 8390 8391 8392 8393 8394 8395
	__u32 i, zero = 0;
	int err;

	if (!bpf_map__is_struct_ops(map) || map->fd == -1)
		return ERR_PTR(-EINVAL);

	link = calloc(1, sizeof(*link));
	if (!link)
		return ERR_PTR(-EINVAL);

	st_ops = map->st_ops;
	for (i = 0; i < btf_vlen(st_ops->type); i++) {
		struct bpf_program *prog = st_ops->progs[i];
		void *kern_data;
		int prog_fd;

		if (!prog)
			continue;

		prog_fd = bpf_program__fd(prog);
		kern_data = st_ops->kern_vdata + st_ops->kern_func_off[i];
		*(unsigned long *)kern_data = prog_fd;
	}

	err = bpf_map_update_elem(map->fd, &zero, st_ops->kern_vdata, 0);
	if (err) {
		err = -errno;
		free(link);
		return ERR_PTR(err);
	}

8396
	link->detach = bpf_link__detach_struct_ops;
8397 8398
	link->fd = map->fd;

8399
	return link;
8400 8401
}

8402
enum bpf_perf_event_ret
8403 8404 8405
bpf_perf_event_read_simple(void *mmap_mem, size_t mmap_size, size_t page_size,
			   void **copy_mem, size_t *copy_size,
			   bpf_perf_event_print_t fn, void *private_data)
8406
{
8407
	struct perf_event_mmap_page *header = mmap_mem;
8408
	__u64 data_head = ring_buffer_read_head(header);
8409
	__u64 data_tail = header->data_tail;
8410 8411 8412 8413 8414 8415 8416 8417 8418 8419 8420 8421 8422 8423 8424 8425 8426 8427 8428
	void *base = ((__u8 *)header) + page_size;
	int ret = LIBBPF_PERF_EVENT_CONT;
	struct perf_event_header *ehdr;
	size_t ehdr_size;

	while (data_head != data_tail) {
		ehdr = base + (data_tail & (mmap_size - 1));
		ehdr_size = ehdr->size;

		if (((void *)ehdr) + ehdr_size > base + mmap_size) {
			void *copy_start = ehdr;
			size_t len_first = base + mmap_size - copy_start;
			size_t len_secnd = ehdr_size - len_first;

			if (*copy_size < ehdr_size) {
				free(*copy_mem);
				*copy_mem = malloc(ehdr_size);
				if (!*copy_mem) {
					*copy_size = 0;
8429 8430 8431
					ret = LIBBPF_PERF_EVENT_ERROR;
					break;
				}
8432
				*copy_size = ehdr_size;
8433 8434
			}

8435 8436 8437
			memcpy(*copy_mem, copy_start, len_first);
			memcpy(*copy_mem + len_first, base, len_secnd);
			ehdr = *copy_mem;
8438 8439
		}

8440 8441
		ret = fn(ehdr, private_data);
		data_tail += ehdr_size;
8442 8443 8444 8445
		if (ret != LIBBPF_PERF_EVENT_CONT)
			break;
	}

8446
	ring_buffer_write_tail(header, data_tail);
8447 8448
	return ret;
}
8449

A
Andrii Nakryiko 已提交
8450 8451 8452 8453 8454 8455 8456 8457 8458 8459 8460 8461 8462 8463 8464 8465 8466 8467 8468 8469 8470 8471 8472 8473 8474 8475 8476 8477 8478 8479 8480 8481 8482 8483 8484
struct perf_buffer;

struct perf_buffer_params {
	struct perf_event_attr *attr;
	/* if event_cb is specified, it takes precendence */
	perf_buffer_event_fn event_cb;
	/* sample_cb and lost_cb are higher-level common-case callbacks */
	perf_buffer_sample_fn sample_cb;
	perf_buffer_lost_fn lost_cb;
	void *ctx;
	int cpu_cnt;
	int *cpus;
	int *map_keys;
};

struct perf_cpu_buf {
	struct perf_buffer *pb;
	void *base; /* mmap()'ed memory */
	void *buf; /* for reconstructing segmented data */
	size_t buf_size;
	int fd;
	int cpu;
	int map_key;
};

struct perf_buffer {
	perf_buffer_event_fn event_cb;
	perf_buffer_sample_fn sample_cb;
	perf_buffer_lost_fn lost_cb;
	void *ctx; /* passed into callbacks */

	size_t page_size;
	size_t mmap_size;
	struct perf_cpu_buf **cpu_bufs;
	struct epoll_event *events;
8485
	int cpu_cnt; /* number of allocated CPU buffers */
A
Andrii Nakryiko 已提交
8486 8487 8488 8489 8490 8491 8492 8493 8494 8495 8496
	int epoll_fd; /* perf event FD */
	int map_fd; /* BPF_MAP_TYPE_PERF_EVENT_ARRAY BPF map FD */
};

static void perf_buffer__free_cpu_buf(struct perf_buffer *pb,
				      struct perf_cpu_buf *cpu_buf)
{
	if (!cpu_buf)
		return;
	if (cpu_buf->base &&
	    munmap(cpu_buf->base, pb->mmap_size + pb->page_size))
8497
		pr_warn("failed to munmap cpu_buf #%d\n", cpu_buf->cpu);
A
Andrii Nakryiko 已提交
8498 8499 8500 8501 8502 8503 8504 8505 8506 8507 8508 8509
	if (cpu_buf->fd >= 0) {
		ioctl(cpu_buf->fd, PERF_EVENT_IOC_DISABLE, 0);
		close(cpu_buf->fd);
	}
	free(cpu_buf->buf);
	free(cpu_buf);
}

void perf_buffer__free(struct perf_buffer *pb)
{
	int i;

8510
	if (IS_ERR_OR_NULL(pb))
A
Andrii Nakryiko 已提交
8511 8512
		return;
	if (pb->cpu_bufs) {
8513
		for (i = 0; i < pb->cpu_cnt; i++) {
A
Andrii Nakryiko 已提交
8514 8515
			struct perf_cpu_buf *cpu_buf = pb->cpu_bufs[i];

8516 8517 8518
			if (!cpu_buf)
				continue;

A
Andrii Nakryiko 已提交
8519 8520 8521 8522 8523 8524 8525 8526 8527 8528 8529 8530 8531 8532 8533 8534 8535 8536 8537 8538 8539 8540 8541 8542 8543 8544 8545 8546 8547 8548 8549
			bpf_map_delete_elem(pb->map_fd, &cpu_buf->map_key);
			perf_buffer__free_cpu_buf(pb, cpu_buf);
		}
		free(pb->cpu_bufs);
	}
	if (pb->epoll_fd >= 0)
		close(pb->epoll_fd);
	free(pb->events);
	free(pb);
}

static struct perf_cpu_buf *
perf_buffer__open_cpu_buf(struct perf_buffer *pb, struct perf_event_attr *attr,
			  int cpu, int map_key)
{
	struct perf_cpu_buf *cpu_buf;
	char msg[STRERR_BUFSIZE];
	int err;

	cpu_buf = calloc(1, sizeof(*cpu_buf));
	if (!cpu_buf)
		return ERR_PTR(-ENOMEM);

	cpu_buf->pb = pb;
	cpu_buf->cpu = cpu;
	cpu_buf->map_key = map_key;

	cpu_buf->fd = syscall(__NR_perf_event_open, attr, -1 /* pid */, cpu,
			      -1, PERF_FLAG_FD_CLOEXEC);
	if (cpu_buf->fd < 0) {
		err = -errno;
8550 8551
		pr_warn("failed to open perf buffer event on cpu #%d: %s\n",
			cpu, libbpf_strerror_r(err, msg, sizeof(msg)));
A
Andrii Nakryiko 已提交
8552 8553 8554 8555 8556 8557 8558 8559 8560
		goto error;
	}

	cpu_buf->base = mmap(NULL, pb->mmap_size + pb->page_size,
			     PROT_READ | PROT_WRITE, MAP_SHARED,
			     cpu_buf->fd, 0);
	if (cpu_buf->base == MAP_FAILED) {
		cpu_buf->base = NULL;
		err = -errno;
8561 8562
		pr_warn("failed to mmap perf buffer on cpu #%d: %s\n",
			cpu, libbpf_strerror_r(err, msg, sizeof(msg)));
A
Andrii Nakryiko 已提交
8563 8564 8565 8566 8567
		goto error;
	}

	if (ioctl(cpu_buf->fd, PERF_EVENT_IOC_ENABLE, 0) < 0) {
		err = -errno;
8568 8569
		pr_warn("failed to enable perf buffer event on cpu #%d: %s\n",
			cpu, libbpf_strerror_r(err, msg, sizeof(msg)));
A
Andrii Nakryiko 已提交
8570 8571 8572 8573 8574 8575 8576 8577 8578 8579 8580 8581 8582 8583 8584 8585 8586
		goto error;
	}

	return cpu_buf;

error:
	perf_buffer__free_cpu_buf(pb, cpu_buf);
	return (struct perf_cpu_buf *)ERR_PTR(err);
}

static struct perf_buffer *__perf_buffer__new(int map_fd, size_t page_cnt,
					      struct perf_buffer_params *p);

struct perf_buffer *perf_buffer__new(int map_fd, size_t page_cnt,
				     const struct perf_buffer_opts *opts)
{
	struct perf_buffer_params p = {};
8587 8588 8589 8590 8591 8592 8593
	struct perf_event_attr attr = { 0, };

	attr.config = PERF_COUNT_SW_BPF_OUTPUT,
	attr.type = PERF_TYPE_SOFTWARE;
	attr.sample_type = PERF_SAMPLE_RAW;
	attr.sample_period = 1;
	attr.wakeup_events = 1;
A
Andrii Nakryiko 已提交
8594 8595 8596 8597 8598 8599 8600 8601 8602 8603 8604 8605 8606 8607 8608 8609 8610 8611 8612 8613 8614 8615 8616 8617 8618 8619 8620 8621

	p.attr = &attr;
	p.sample_cb = opts ? opts->sample_cb : NULL;
	p.lost_cb = opts ? opts->lost_cb : NULL;
	p.ctx = opts ? opts->ctx : NULL;

	return __perf_buffer__new(map_fd, page_cnt, &p);
}

struct perf_buffer *
perf_buffer__new_raw(int map_fd, size_t page_cnt,
		     const struct perf_buffer_raw_opts *opts)
{
	struct perf_buffer_params p = {};

	p.attr = opts->attr;
	p.event_cb = opts->event_cb;
	p.ctx = opts->ctx;
	p.cpu_cnt = opts->cpu_cnt;
	p.cpus = opts->cpus;
	p.map_keys = opts->map_keys;

	return __perf_buffer__new(map_fd, page_cnt, &p);
}

static struct perf_buffer *__perf_buffer__new(int map_fd, size_t page_cnt,
					      struct perf_buffer_params *p)
{
8622
	const char *online_cpus_file = "/sys/devices/system/cpu/online";
8623
	struct bpf_map_info map;
A
Andrii Nakryiko 已提交
8624 8625
	char msg[STRERR_BUFSIZE];
	struct perf_buffer *pb;
8626
	bool *online = NULL;
A
Andrii Nakryiko 已提交
8627
	__u32 map_info_len;
8628
	int err, i, j, n;
A
Andrii Nakryiko 已提交
8629 8630

	if (page_cnt & (page_cnt - 1)) {
8631 8632
		pr_warn("page count should be power of two, but is %zu\n",
			page_cnt);
A
Andrii Nakryiko 已提交
8633 8634 8635
		return ERR_PTR(-EINVAL);
	}

8636 8637
	/* best-effort sanity checks */
	memset(&map, 0, sizeof(map));
A
Andrii Nakryiko 已提交
8638 8639 8640 8641
	map_info_len = sizeof(map);
	err = bpf_obj_get_info_by_fd(map_fd, &map, &map_info_len);
	if (err) {
		err = -errno;
8642 8643 8644 8645 8646 8647 8648 8649 8650 8651 8652 8653 8654 8655 8656 8657
		/* if BPF_OBJ_GET_INFO_BY_FD is supported, will return
		 * -EBADFD, -EFAULT, or -E2BIG on real error
		 */
		if (err != -EINVAL) {
			pr_warn("failed to get map info for map FD %d: %s\n",
				map_fd, libbpf_strerror_r(err, msg, sizeof(msg)));
			return ERR_PTR(err);
		}
		pr_debug("failed to get map info for FD %d; API not supported? Ignoring...\n",
			 map_fd);
	} else {
		if (map.type != BPF_MAP_TYPE_PERF_EVENT_ARRAY) {
			pr_warn("map '%s' should be BPF_MAP_TYPE_PERF_EVENT_ARRAY\n",
				map.name);
			return ERR_PTR(-EINVAL);
		}
A
Andrii Nakryiko 已提交
8658 8659 8660 8661 8662 8663 8664 8665 8666 8667 8668 8669 8670 8671 8672 8673 8674 8675
	}

	pb = calloc(1, sizeof(*pb));
	if (!pb)
		return ERR_PTR(-ENOMEM);

	pb->event_cb = p->event_cb;
	pb->sample_cb = p->sample_cb;
	pb->lost_cb = p->lost_cb;
	pb->ctx = p->ctx;

	pb->page_size = getpagesize();
	pb->mmap_size = pb->page_size * page_cnt;
	pb->map_fd = map_fd;

	pb->epoll_fd = epoll_create1(EPOLL_CLOEXEC);
	if (pb->epoll_fd < 0) {
		err = -errno;
8676 8677
		pr_warn("failed to create epoll instance: %s\n",
			libbpf_strerror_r(err, msg, sizeof(msg)));
A
Andrii Nakryiko 已提交
8678 8679 8680 8681 8682 8683 8684 8685 8686 8687 8688
		goto error;
	}

	if (p->cpu_cnt > 0) {
		pb->cpu_cnt = p->cpu_cnt;
	} else {
		pb->cpu_cnt = libbpf_num_possible_cpus();
		if (pb->cpu_cnt < 0) {
			err = pb->cpu_cnt;
			goto error;
		}
8689
		if (map.max_entries && map.max_entries < pb->cpu_cnt)
A
Andrii Nakryiko 已提交
8690 8691 8692 8693 8694 8695
			pb->cpu_cnt = map.max_entries;
	}

	pb->events = calloc(pb->cpu_cnt, sizeof(*pb->events));
	if (!pb->events) {
		err = -ENOMEM;
8696
		pr_warn("failed to allocate events: out of memory\n");
A
Andrii Nakryiko 已提交
8697 8698 8699 8700 8701
		goto error;
	}
	pb->cpu_bufs = calloc(pb->cpu_cnt, sizeof(*pb->cpu_bufs));
	if (!pb->cpu_bufs) {
		err = -ENOMEM;
8702
		pr_warn("failed to allocate buffers: out of memory\n");
A
Andrii Nakryiko 已提交
8703 8704 8705
		goto error;
	}

8706 8707 8708 8709 8710 8711 8712
	err = parse_cpu_mask_file(online_cpus_file, &online, &n);
	if (err) {
		pr_warn("failed to get online CPU mask: %d\n", err);
		goto error;
	}

	for (i = 0, j = 0; i < pb->cpu_cnt; i++) {
A
Andrii Nakryiko 已提交
8713 8714 8715 8716 8717 8718
		struct perf_cpu_buf *cpu_buf;
		int cpu, map_key;

		cpu = p->cpu_cnt > 0 ? p->cpus[i] : i;
		map_key = p->cpu_cnt > 0 ? p->map_keys[i] : i;

8719 8720 8721 8722 8723 8724
		/* in case user didn't explicitly requested particular CPUs to
		 * be attached to, skip offline/not present CPUs
		 */
		if (p->cpu_cnt <= 0 && (cpu >= n || !online[cpu]))
			continue;

A
Andrii Nakryiko 已提交
8725 8726 8727 8728 8729 8730
		cpu_buf = perf_buffer__open_cpu_buf(pb, p->attr, cpu, map_key);
		if (IS_ERR(cpu_buf)) {
			err = PTR_ERR(cpu_buf);
			goto error;
		}

8731
		pb->cpu_bufs[j] = cpu_buf;
A
Andrii Nakryiko 已提交
8732 8733 8734 8735 8736

		err = bpf_map_update_elem(pb->map_fd, &map_key,
					  &cpu_buf->fd, 0);
		if (err) {
			err = -errno;
8737 8738 8739
			pr_warn("failed to set cpu #%d, key %d -> perf FD %d: %s\n",
				cpu, map_key, cpu_buf->fd,
				libbpf_strerror_r(err, msg, sizeof(msg)));
A
Andrii Nakryiko 已提交
8740 8741 8742
			goto error;
		}

8743 8744
		pb->events[j].events = EPOLLIN;
		pb->events[j].data.ptr = cpu_buf;
A
Andrii Nakryiko 已提交
8745
		if (epoll_ctl(pb->epoll_fd, EPOLL_CTL_ADD, cpu_buf->fd,
8746
			      &pb->events[j]) < 0) {
A
Andrii Nakryiko 已提交
8747
			err = -errno;
8748 8749 8750
			pr_warn("failed to epoll_ctl cpu #%d perf FD %d: %s\n",
				cpu, cpu_buf->fd,
				libbpf_strerror_r(err, msg, sizeof(msg)));
A
Andrii Nakryiko 已提交
8751 8752
			goto error;
		}
8753
		j++;
A
Andrii Nakryiko 已提交
8754
	}
8755 8756
	pb->cpu_cnt = j;
	free(online);
A
Andrii Nakryiko 已提交
8757 8758 8759 8760

	return pb;

error:
8761
	free(online);
A
Andrii Nakryiko 已提交
8762 8763 8764 8765 8766 8767 8768 8769
	if (pb)
		perf_buffer__free(pb);
	return ERR_PTR(err);
}

struct perf_sample_raw {
	struct perf_event_header header;
	uint32_t size;
8770
	char data[];
A
Andrii Nakryiko 已提交
8771 8772 8773 8774 8775 8776 8777 8778 8779 8780 8781 8782 8783 8784 8785 8786 8787 8788 8789 8790 8791 8792 8793 8794 8795 8796 8797 8798 8799 8800 8801 8802 8803 8804 8805 8806
};

struct perf_sample_lost {
	struct perf_event_header header;
	uint64_t id;
	uint64_t lost;
	uint64_t sample_id;
};

static enum bpf_perf_event_ret
perf_buffer__process_record(struct perf_event_header *e, void *ctx)
{
	struct perf_cpu_buf *cpu_buf = ctx;
	struct perf_buffer *pb = cpu_buf->pb;
	void *data = e;

	/* user wants full control over parsing perf event */
	if (pb->event_cb)
		return pb->event_cb(pb->ctx, cpu_buf->cpu, e);

	switch (e->type) {
	case PERF_RECORD_SAMPLE: {
		struct perf_sample_raw *s = data;

		if (pb->sample_cb)
			pb->sample_cb(pb->ctx, cpu_buf->cpu, s->data, s->size);
		break;
	}
	case PERF_RECORD_LOST: {
		struct perf_sample_lost *s = data;

		if (pb->lost_cb)
			pb->lost_cb(pb->ctx, cpu_buf->cpu, s->lost);
		break;
	}
	default:
8807
		pr_warn("unknown perf sample type %d\n", e->type);
A
Andrii Nakryiko 已提交
8808 8809 8810 8811 8812 8813 8814 8815 8816 8817 8818 8819 8820 8821 8822 8823 8824 8825 8826 8827 8828 8829 8830 8831 8832 8833 8834 8835 8836
		return LIBBPF_PERF_EVENT_ERROR;
	}
	return LIBBPF_PERF_EVENT_CONT;
}

static int perf_buffer__process_records(struct perf_buffer *pb,
					struct perf_cpu_buf *cpu_buf)
{
	enum bpf_perf_event_ret ret;

	ret = bpf_perf_event_read_simple(cpu_buf->base, pb->mmap_size,
					 pb->page_size, &cpu_buf->buf,
					 &cpu_buf->buf_size,
					 perf_buffer__process_record, cpu_buf);
	if (ret != LIBBPF_PERF_EVENT_CONT)
		return ret;
	return 0;
}

int perf_buffer__poll(struct perf_buffer *pb, int timeout_ms)
{
	int i, cnt, err;

	cnt = epoll_wait(pb->epoll_fd, pb->events, pb->cpu_cnt, timeout_ms);
	for (i = 0; i < cnt; i++) {
		struct perf_cpu_buf *cpu_buf = pb->events[i].data.ptr;

		err = perf_buffer__process_records(pb, cpu_buf);
		if (err) {
8837
			pr_warn("error while processing records: %d\n", err);
A
Andrii Nakryiko 已提交
8838 8839 8840 8841 8842 8843
			return err;
		}
	}
	return cnt < 0 ? -errno : cnt;
}

8844 8845 8846 8847 8848 8849 8850 8851 8852 8853 8854 8855 8856 8857 8858 8859 8860 8861 8862
int perf_buffer__consume(struct perf_buffer *pb)
{
	int i, err;

	for (i = 0; i < pb->cpu_cnt; i++) {
		struct perf_cpu_buf *cpu_buf = pb->cpu_bufs[i];

		if (!cpu_buf)
			continue;

		err = perf_buffer__process_records(pb, cpu_buf);
		if (err) {
			pr_warn("error while processing records: %d\n", err);
			return err;
		}
	}
	return 0;
}

8863 8864 8865 8866 8867 8868 8869 8870 8871 8872 8873 8874 8875 8876 8877 8878 8879 8880 8881 8882 8883 8884 8885 8886 8887 8888 8889 8890 8891 8892 8893 8894 8895 8896 8897 8898 8899 8900 8901 8902 8903 8904 8905 8906 8907 8908 8909 8910 8911 8912 8913 8914 8915 8916 8917 8918 8919
struct bpf_prog_info_array_desc {
	int	array_offset;	/* e.g. offset of jited_prog_insns */
	int	count_offset;	/* e.g. offset of jited_prog_len */
	int	size_offset;	/* > 0: offset of rec size,
				 * < 0: fix size of -size_offset
				 */
};

static struct bpf_prog_info_array_desc bpf_prog_info_array_desc[] = {
	[BPF_PROG_INFO_JITED_INSNS] = {
		offsetof(struct bpf_prog_info, jited_prog_insns),
		offsetof(struct bpf_prog_info, jited_prog_len),
		-1,
	},
	[BPF_PROG_INFO_XLATED_INSNS] = {
		offsetof(struct bpf_prog_info, xlated_prog_insns),
		offsetof(struct bpf_prog_info, xlated_prog_len),
		-1,
	},
	[BPF_PROG_INFO_MAP_IDS] = {
		offsetof(struct bpf_prog_info, map_ids),
		offsetof(struct bpf_prog_info, nr_map_ids),
		-(int)sizeof(__u32),
	},
	[BPF_PROG_INFO_JITED_KSYMS] = {
		offsetof(struct bpf_prog_info, jited_ksyms),
		offsetof(struct bpf_prog_info, nr_jited_ksyms),
		-(int)sizeof(__u64),
	},
	[BPF_PROG_INFO_JITED_FUNC_LENS] = {
		offsetof(struct bpf_prog_info, jited_func_lens),
		offsetof(struct bpf_prog_info, nr_jited_func_lens),
		-(int)sizeof(__u32),
	},
	[BPF_PROG_INFO_FUNC_INFO] = {
		offsetof(struct bpf_prog_info, func_info),
		offsetof(struct bpf_prog_info, nr_func_info),
		offsetof(struct bpf_prog_info, func_info_rec_size),
	},
	[BPF_PROG_INFO_LINE_INFO] = {
		offsetof(struct bpf_prog_info, line_info),
		offsetof(struct bpf_prog_info, nr_line_info),
		offsetof(struct bpf_prog_info, line_info_rec_size),
	},
	[BPF_PROG_INFO_JITED_LINE_INFO] = {
		offsetof(struct bpf_prog_info, jited_line_info),
		offsetof(struct bpf_prog_info, nr_jited_line_info),
		offsetof(struct bpf_prog_info, jited_line_info_rec_size),
	},
	[BPF_PROG_INFO_PROG_TAGS] = {
		offsetof(struct bpf_prog_info, prog_tags),
		offsetof(struct bpf_prog_info, nr_prog_tags),
		-(int)sizeof(__u8) * BPF_TAG_SIZE,
	},

};

8920 8921
static __u32 bpf_prog_info_read_offset_u32(struct bpf_prog_info *info,
					   int offset)
8922 8923 8924 8925 8926 8927 8928 8929
{
	__u32 *array = (__u32 *)info;

	if (offset >= 0)
		return array[offset / sizeof(__u32)];
	return -(int)offset;
}

8930 8931
static __u64 bpf_prog_info_read_offset_u64(struct bpf_prog_info *info,
					   int offset)
8932 8933 8934 8935 8936 8937 8938 8939 8940 8941 8942 8943 8944 8945 8946 8947 8948 8949 8950 8951 8952 8953 8954 8955 8956 8957 8958 8959 8960 8961 8962 8963 8964 8965 8966 8967 8968 8969 8970 8971 8972 8973 8974 8975 8976 8977 8978 8979 8980 8981 8982 8983 8984 8985 8986 8987 8988 8989 8990 8991 8992 8993 8994 8995 8996 8997 8998 8999 9000 9001 9002 9003 9004 9005 9006 9007 9008 9009 9010 9011 9012 9013 9014 9015 9016 9017 9018 9019 9020 9021 9022 9023 9024 9025 9026 9027 9028 9029 9030 9031 9032 9033 9034 9035 9036 9037 9038 9039 9040 9041 9042 9043 9044 9045 9046 9047 9048 9049 9050 9051 9052 9053 9054
{
	__u64 *array = (__u64 *)info;

	if (offset >= 0)
		return array[offset / sizeof(__u64)];
	return -(int)offset;
}

static void bpf_prog_info_set_offset_u32(struct bpf_prog_info *info, int offset,
					 __u32 val)
{
	__u32 *array = (__u32 *)info;

	if (offset >= 0)
		array[offset / sizeof(__u32)] = val;
}

static void bpf_prog_info_set_offset_u64(struct bpf_prog_info *info, int offset,
					 __u64 val)
{
	__u64 *array = (__u64 *)info;

	if (offset >= 0)
		array[offset / sizeof(__u64)] = val;
}

struct bpf_prog_info_linear *
bpf_program__get_prog_info_linear(int fd, __u64 arrays)
{
	struct bpf_prog_info_linear *info_linear;
	struct bpf_prog_info info = {};
	__u32 info_len = sizeof(info);
	__u32 data_len = 0;
	int i, err;
	void *ptr;

	if (arrays >> BPF_PROG_INFO_LAST_ARRAY)
		return ERR_PTR(-EINVAL);

	/* step 1: get array dimensions */
	err = bpf_obj_get_info_by_fd(fd, &info, &info_len);
	if (err) {
		pr_debug("can't get prog info: %s", strerror(errno));
		return ERR_PTR(-EFAULT);
	}

	/* step 2: calculate total size of all arrays */
	for (i = BPF_PROG_INFO_FIRST_ARRAY; i < BPF_PROG_INFO_LAST_ARRAY; ++i) {
		bool include_array = (arrays & (1UL << i)) > 0;
		struct bpf_prog_info_array_desc *desc;
		__u32 count, size;

		desc = bpf_prog_info_array_desc + i;

		/* kernel is too old to support this field */
		if (info_len < desc->array_offset + sizeof(__u32) ||
		    info_len < desc->count_offset + sizeof(__u32) ||
		    (desc->size_offset > 0 && info_len < desc->size_offset))
			include_array = false;

		if (!include_array) {
			arrays &= ~(1UL << i);	/* clear the bit */
			continue;
		}

		count = bpf_prog_info_read_offset_u32(&info, desc->count_offset);
		size  = bpf_prog_info_read_offset_u32(&info, desc->size_offset);

		data_len += count * size;
	}

	/* step 3: allocate continuous memory */
	data_len = roundup(data_len, sizeof(__u64));
	info_linear = malloc(sizeof(struct bpf_prog_info_linear) + data_len);
	if (!info_linear)
		return ERR_PTR(-ENOMEM);

	/* step 4: fill data to info_linear->info */
	info_linear->arrays = arrays;
	memset(&info_linear->info, 0, sizeof(info));
	ptr = info_linear->data;

	for (i = BPF_PROG_INFO_FIRST_ARRAY; i < BPF_PROG_INFO_LAST_ARRAY; ++i) {
		struct bpf_prog_info_array_desc *desc;
		__u32 count, size;

		if ((arrays & (1UL << i)) == 0)
			continue;

		desc  = bpf_prog_info_array_desc + i;
		count = bpf_prog_info_read_offset_u32(&info, desc->count_offset);
		size  = bpf_prog_info_read_offset_u32(&info, desc->size_offset);
		bpf_prog_info_set_offset_u32(&info_linear->info,
					     desc->count_offset, count);
		bpf_prog_info_set_offset_u32(&info_linear->info,
					     desc->size_offset, size);
		bpf_prog_info_set_offset_u64(&info_linear->info,
					     desc->array_offset,
					     ptr_to_u64(ptr));
		ptr += count * size;
	}

	/* step 5: call syscall again to get required arrays */
	err = bpf_obj_get_info_by_fd(fd, &info_linear->info, &info_len);
	if (err) {
		pr_debug("can't get prog info: %s", strerror(errno));
		free(info_linear);
		return ERR_PTR(-EFAULT);
	}

	/* step 6: verify the data */
	for (i = BPF_PROG_INFO_FIRST_ARRAY; i < BPF_PROG_INFO_LAST_ARRAY; ++i) {
		struct bpf_prog_info_array_desc *desc;
		__u32 v1, v2;

		if ((arrays & (1UL << i)) == 0)
			continue;

		desc = bpf_prog_info_array_desc + i;
		v1 = bpf_prog_info_read_offset_u32(&info, desc->count_offset);
		v2 = bpf_prog_info_read_offset_u32(&info_linear->info,
						   desc->count_offset);
		if (v1 != v2)
9055
			pr_warn("%s: mismatch in element count\n", __func__);
9056 9057 9058 9059 9060

		v1 = bpf_prog_info_read_offset_u32(&info, desc->size_offset);
		v2 = bpf_prog_info_read_offset_u32(&info_linear->info,
						   desc->size_offset);
		if (v1 != v2)
9061
			pr_warn("%s: mismatch in rec size\n", __func__);
9062 9063 9064 9065 9066 9067 9068 9069 9070 9071 9072 9073 9074 9075 9076 9077 9078 9079 9080 9081 9082 9083 9084 9085 9086 9087 9088 9089 9090 9091 9092 9093 9094 9095 9096 9097 9098 9099 9100 9101 9102 9103 9104 9105 9106 9107 9108 9109
	}

	/* step 7: update info_len and data_len */
	info_linear->info_len = sizeof(struct bpf_prog_info);
	info_linear->data_len = data_len;

	return info_linear;
}

void bpf_program__bpil_addr_to_offs(struct bpf_prog_info_linear *info_linear)
{
	int i;

	for (i = BPF_PROG_INFO_FIRST_ARRAY; i < BPF_PROG_INFO_LAST_ARRAY; ++i) {
		struct bpf_prog_info_array_desc *desc;
		__u64 addr, offs;

		if ((info_linear->arrays & (1UL << i)) == 0)
			continue;

		desc = bpf_prog_info_array_desc + i;
		addr = bpf_prog_info_read_offset_u64(&info_linear->info,
						     desc->array_offset);
		offs = addr - ptr_to_u64(info_linear->data);
		bpf_prog_info_set_offset_u64(&info_linear->info,
					     desc->array_offset, offs);
	}
}

void bpf_program__bpil_offs_to_addr(struct bpf_prog_info_linear *info_linear)
{
	int i;

	for (i = BPF_PROG_INFO_FIRST_ARRAY; i < BPF_PROG_INFO_LAST_ARRAY; ++i) {
		struct bpf_prog_info_array_desc *desc;
		__u64 addr, offs;

		if ((info_linear->arrays & (1UL << i)) == 0)
			continue;

		desc = bpf_prog_info_array_desc + i;
		offs = bpf_prog_info_read_offset_u64(&info_linear->info,
						     desc->array_offset);
		addr = offs + ptr_to_u64(info_linear->data);
		bpf_prog_info_set_offset_u64(&info_linear->info,
					     desc->array_offset, addr);
	}
}
9110

9111 9112 9113 9114 9115 9116 9117 9118 9119 9120 9121 9122 9123 9124 9125 9126 9127 9128 9129 9130 9131 9132 9133 9134 9135
int bpf_program__set_attach_target(struct bpf_program *prog,
				   int attach_prog_fd,
				   const char *attach_func_name)
{
	int btf_id;

	if (!prog || attach_prog_fd < 0 || !attach_func_name)
		return -EINVAL;

	if (attach_prog_fd)
		btf_id = libbpf_find_prog_btf_id(attach_func_name,
						 attach_prog_fd);
	else
		btf_id = __find_vmlinux_btf_id(prog->obj->btf_vmlinux,
					       attach_func_name,
					       prog->expected_attach_type);

	if (btf_id < 0)
		return btf_id;

	prog->attach_btf_id = btf_id;
	prog->attach_prog_fd = attach_prog_fd;
	return 0;
}

9136
int parse_cpu_mask_str(const char *s, bool **mask, int *mask_sz)
9137
{
9138 9139
	int err = 0, n, len, start, end = -1;
	bool *tmp;
9140

9141 9142 9143 9144 9145 9146 9147 9148 9149 9150 9151 9152 9153 9154 9155 9156 9157 9158 9159 9160 9161 9162 9163 9164 9165 9166 9167 9168 9169 9170 9171 9172 9173 9174 9175 9176 9177 9178 9179 9180 9181 9182 9183 9184 9185 9186 9187 9188 9189
	*mask = NULL;
	*mask_sz = 0;

	/* Each sub string separated by ',' has format \d+-\d+ or \d+ */
	while (*s) {
		if (*s == ',' || *s == '\n') {
			s++;
			continue;
		}
		n = sscanf(s, "%d%n-%d%n", &start, &len, &end, &len);
		if (n <= 0 || n > 2) {
			pr_warn("Failed to get CPU range %s: %d\n", s, n);
			err = -EINVAL;
			goto cleanup;
		} else if (n == 1) {
			end = start;
		}
		if (start < 0 || start > end) {
			pr_warn("Invalid CPU range [%d,%d] in %s\n",
				start, end, s);
			err = -EINVAL;
			goto cleanup;
		}
		tmp = realloc(*mask, end + 1);
		if (!tmp) {
			err = -ENOMEM;
			goto cleanup;
		}
		*mask = tmp;
		memset(tmp + *mask_sz, 0, start - *mask_sz);
		memset(tmp + start, 1, end - start + 1);
		*mask_sz = end + 1;
		s += len;
	}
	if (!*mask_sz) {
		pr_warn("Empty CPU range\n");
		return -EINVAL;
	}
	return 0;
cleanup:
	free(*mask);
	*mask = NULL;
	return err;
}

int parse_cpu_mask_file(const char *fcpu, bool **mask, int *mask_sz)
{
	int fd, err = 0, len;
	char buf[128];
9190 9191 9192

	fd = open(fcpu, O_RDONLY);
	if (fd < 0) {
9193 9194 9195
		err = -errno;
		pr_warn("Failed to open cpu mask file %s: %d\n", fcpu, err);
		return err;
9196 9197 9198 9199
	}
	len = read(fd, buf, sizeof(buf));
	close(fd);
	if (len <= 0) {
9200 9201 9202
		err = len ? -errno : -EINVAL;
		pr_warn("Failed to read cpu mask from %s: %d\n", fcpu, err);
		return err;
9203
	}
9204 9205 9206
	if (len >= sizeof(buf)) {
		pr_warn("CPU mask is too big in file %s\n", fcpu);
		return -E2BIG;
9207 9208 9209
	}
	buf[len] = '\0';

9210 9211 9212 9213 9214 9215 9216 9217 9218 9219 9220 9221 9222 9223 9224 9225 9226 9227 9228 9229 9230 9231
	return parse_cpu_mask_str(buf, mask, mask_sz);
}

int libbpf_num_possible_cpus(void)
{
	static const char *fcpu = "/sys/devices/system/cpu/possible";
	static int cpus;
	int err, n, i, tmp_cpus;
	bool *mask;

	tmp_cpus = READ_ONCE(cpus);
	if (tmp_cpus > 0)
		return tmp_cpus;

	err = parse_cpu_mask_file(fcpu, &mask, &n);
	if (err)
		return err;

	tmp_cpus = 0;
	for (i = 0; i < n; i++) {
		if (mask[i])
			tmp_cpus++;
9232
	}
9233
	free(mask);
9234 9235 9236

	WRITE_ONCE(cpus, tmp_cpus);
	return tmp_cpus;
9237
}
9238 9239 9240 9241 9242 9243 9244 9245 9246 9247 9248 9249 9250 9251 9252 9253 9254 9255 9256 9257 9258 9259 9260 9261 9262 9263 9264 9265 9266 9267 9268 9269 9270 9271 9272 9273 9274 9275 9276 9277 9278 9279

int bpf_object__open_skeleton(struct bpf_object_skeleton *s,
			      const struct bpf_object_open_opts *opts)
{
	DECLARE_LIBBPF_OPTS(bpf_object_open_opts, skel_opts,
		.object_name = s->name,
	);
	struct bpf_object *obj;
	int i;

	/* Attempt to preserve opts->object_name, unless overriden by user
	 * explicitly. Overwriting object name for skeletons is discouraged,
	 * as it breaks global data maps, because they contain object name
	 * prefix as their own map name prefix. When skeleton is generated,
	 * bpftool is making an assumption that this name will stay the same.
	 */
	if (opts) {
		memcpy(&skel_opts, opts, sizeof(*opts));
		if (!opts->object_name)
			skel_opts.object_name = s->name;
	}

	obj = bpf_object__open_mem(s->data, s->data_sz, &skel_opts);
	if (IS_ERR(obj)) {
		pr_warn("failed to initialize skeleton BPF object '%s': %ld\n",
			s->name, PTR_ERR(obj));
		return PTR_ERR(obj);
	}

	*s->obj = obj;

	for (i = 0; i < s->map_cnt; i++) {
		struct bpf_map **map = s->maps[i].map;
		const char *name = s->maps[i].name;
		void **mmaped = s->maps[i].mmaped;

		*map = bpf_object__find_map_by_name(obj, name);
		if (!*map) {
			pr_warn("failed to find skeleton map '%s'\n", name);
			return -ESRCH;
		}

9280
		/* externs shouldn't be pre-setup from user code */
9281
		if (mmaped && (*map)->libbpf_type != LIBBPF_MAP_KCONFIG)
9282 9283 9284 9285 9286 9287 9288 9289 9290 9291 9292 9293 9294 9295 9296 9297 9298 9299 9300 9301 9302 9303 9304 9305 9306 9307 9308 9309 9310 9311 9312 9313 9314 9315 9316 9317 9318 9319 9320 9321 9322 9323 9324 9325 9326 9327 9328 9329 9330 9331 9332 9333 9334 9335 9336 9337
			*mmaped = (*map)->mmaped;
	}

	for (i = 0; i < s->prog_cnt; i++) {
		struct bpf_program **prog = s->progs[i].prog;
		const char *name = s->progs[i].name;

		*prog = bpf_object__find_program_by_name(obj, name);
		if (!*prog) {
			pr_warn("failed to find skeleton program '%s'\n", name);
			return -ESRCH;
		}
	}

	return 0;
}

int bpf_object__load_skeleton(struct bpf_object_skeleton *s)
{
	int i, err;

	err = bpf_object__load(*s->obj);
	if (err) {
		pr_warn("failed to load BPF skeleton '%s': %d\n", s->name, err);
		return err;
	}

	for (i = 0; i < s->map_cnt; i++) {
		struct bpf_map *map = *s->maps[i].map;
		size_t mmap_sz = bpf_map_mmap_sz(map);
		int prot, map_fd = bpf_map__fd(map);
		void **mmaped = s->maps[i].mmaped;

		if (!mmaped)
			continue;

		if (!(map->def.map_flags & BPF_F_MMAPABLE)) {
			*mmaped = NULL;
			continue;
		}

		if (map->def.map_flags & BPF_F_RDONLY_PROG)
			prot = PROT_READ;
		else
			prot = PROT_READ | PROT_WRITE;

		/* Remap anonymous mmap()-ed "map initialization image" as
		 * a BPF map-backed mmap()-ed memory, but preserving the same
		 * memory address. This will cause kernel to change process'
		 * page table to point to a different piece of kernel memory,
		 * but from userspace point of view memory address (and its
		 * contents, being identical at this point) will stay the
		 * same. This mapping will be released by bpf_object__close()
		 * as per normal clean up procedure, so we don't need to worry
		 * about it from skeleton's clean up perspective.
		 */
9338 9339 9340
		*mmaped = mmap(map->mmaped, mmap_sz, prot,
				MAP_SHARED | MAP_FIXED, map_fd, 0);
		if (*mmaped == MAP_FAILED) {
9341 9342 9343 9344 9345 9346 9347 9348 9349 9350 9351 9352 9353 9354 9355 9356 9357 9358 9359 9360 9361
			err = -errno;
			*mmaped = NULL;
			pr_warn("failed to re-mmap() map '%s': %d\n",
				 bpf_map__name(map), err);
			return err;
		}
	}

	return 0;
}

int bpf_object__attach_skeleton(struct bpf_object_skeleton *s)
{
	int i;

	for (i = 0; i < s->prog_cnt; i++) {
		struct bpf_program *prog = *s->progs[i].prog;
		struct bpf_link **link = s->progs[i].link;
		const struct bpf_sec_def *sec_def;
		const char *sec_name = bpf_program__title(prog, false);

9362 9363 9364
		if (!prog->load)
			continue;

9365 9366 9367 9368 9369 9370 9371 9372 9373 9374 9375 9376 9377 9378 9379 9380 9381 9382 9383 9384 9385 9386
		sec_def = find_sec_def(sec_name);
		if (!sec_def || !sec_def->attach_fn)
			continue;

		*link = sec_def->attach_fn(sec_def, prog);
		if (IS_ERR(*link)) {
			pr_warn("failed to auto-attach program '%s': %ld\n",
				bpf_program__name(prog), PTR_ERR(*link));
			return PTR_ERR(*link);
		}
	}

	return 0;
}

void bpf_object__detach_skeleton(struct bpf_object_skeleton *s)
{
	int i;

	for (i = 0; i < s->prog_cnt; i++) {
		struct bpf_link **link = s->progs[i].link;

9387
		bpf_link__destroy(*link);
9388 9389 9390 9391 9392 9393 9394 9395 9396 9397 9398 9399 9400 9401
		*link = NULL;
	}
}

void bpf_object__destroy_skeleton(struct bpf_object_skeleton *s)
{
	if (s->progs)
		bpf_object__detach_skeleton(s);
	if (s->obj)
		bpf_object__close(*s->obj);
	free(s->maps);
	free(s->progs);
	free(s);
}