xref: /linux/tools/lib/bpf/libbpf.c (revision 94fb1afb14c4f0ceb8c5508ddddac6819f662e95)
1 // SPDX-License-Identifier: (LGPL-2.1 OR BSD-2-Clause)
2 
3 /*
4  * Common eBPF ELF object loading operations.
5  *
6  * Copyright (C) 2013-2015 Alexei Starovoitov <ast@kernel.org>
7  * Copyright (C) 2015 Wang Nan <wangnan0@huawei.com>
8  * 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 #include <stdio.h>
18 #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 #include <fcntl.h>
26 #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>
39 #include <sys/epoll.h>
40 #include <sys/ioctl.h>
41 #include <sys/mman.h>
42 #include <sys/stat.h>
43 #include <sys/types.h>
44 #include <sys/vfs.h>
45 #include <sys/utsname.h>
46 #include <sys/resource.h>
47 #include <tools/libc_compat.h>
48 #include <libelf.h>
49 #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 /* make sure libbpf doesn't use kernel-only integer typedefs */
60 #pragma GCC poison u8 u16 u32 u64 s8 s16 s32 s64
61 
62 #ifndef EM_BPF
63 #define EM_BPF 247
64 #endif
65 
66 #ifndef BPF_FS_MAGIC
67 #define BPF_FS_MAGIC		0xcafe4a11
68 #endif
69 
70 /* vsprintf() in __base_pr() uses nonliteral format string. It may break
71  * compilation if user enables corresponding warning. Disable it explicitly.
72  */
73 #pragma GCC diagnostic ignored "-Wformat-nonliteral"
74 
75 #define __printf(a, b)	__attribute__((format(printf, a, b)))
76 
77 static struct bpf_map *bpf_object__add_map(struct bpf_object *obj);
78 static struct bpf_program *bpf_object__find_prog_by_idx(struct bpf_object *obj,
79 							int idx);
80 static const struct btf_type *
81 skip_mods_and_typedefs(const struct btf *btf, __u32 id, __u32 *res_id);
82 
83 static int __base_pr(enum libbpf_print_level level, const char *format,
84 		     va_list args)
85 {
86 	if (level == LIBBPF_DEBUG)
87 		return 0;
88 
89 	return vfprintf(stderr, format, args);
90 }
91 
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 	libbpf_print_fn_t old_print_fn = __libbpf_pr;
97 
98 	__libbpf_pr = fn;
99 	return old_print_fn;
100 }
101 
102 __printf(2, 3)
103 void libbpf_print(enum libbpf_print_level level, const char *format, ...)
104 {
105 	va_list args;
106 
107 	if (!__libbpf_pr)
108 		return;
109 
110 	va_start(args, format);
111 	__libbpf_pr(level, format, args);
112 	va_end(args);
113 }
114 
115 static void pr_perm_msg(int err)
116 {
117 	struct rlimit limit;
118 	char buf[100];
119 
120 	if (err != -EPERM || geteuid() != 0)
121 		return;
122 
123 	err = getrlimit(RLIMIT_MEMLOCK, &limit);
124 	if (err)
125 		return;
126 
127 	if (limit.rlim_cur == RLIM_INFINITY)
128 		return;
129 
130 	if (limit.rlim_cur < 1024)
131 		snprintf(buf, sizeof(buf), "%zu bytes", (size_t)limit.rlim_cur);
132 	else if (limit.rlim_cur < 1024*1024)
133 		snprintf(buf, sizeof(buf), "%.1f KiB", (double)limit.rlim_cur / 1024);
134 	else
135 		snprintf(buf, sizeof(buf), "%.1f MiB", (double)limit.rlim_cur / (1024*1024));
136 
137 	pr_warn("permission error while running as root; try raising 'ulimit -l'? current value: %s\n",
138 		buf);
139 }
140 
141 #define STRERR_BUFSIZE  128
142 
143 /* Copied from tools/perf/util/util.h */
144 #ifndef zfree
145 # define zfree(ptr) ({ free(*ptr); *ptr = NULL; })
146 #endif
147 
148 #ifndef zclose
149 # define zclose(fd) ({			\
150 	int ___err = 0;			\
151 	if ((fd) >= 0)			\
152 		___err = close((fd));	\
153 	fd = -1;			\
154 	___err; })
155 #endif
156 
157 #ifdef HAVE_LIBELF_MMAP_SUPPORT
158 # define LIBBPF_ELF_C_READ_MMAP ELF_C_READ_MMAP
159 #else
160 # define LIBBPF_ELF_C_READ_MMAP ELF_C_READ
161 #endif
162 
163 static inline __u64 ptr_to_u64(const void *ptr)
164 {
165 	return (__u64) (unsigned long) ptr;
166 }
167 
168 struct bpf_capabilities {
169 	/* v4.14: kernel support for program & map names. */
170 	__u32 name:1;
171 	/* v5.2: kernel support for global data sections. */
172 	__u32 global_data:1;
173 	/* BTF_KIND_FUNC and BTF_KIND_FUNC_PROTO support */
174 	__u32 btf_func:1;
175 	/* BTF_KIND_VAR and BTF_KIND_DATASEC support */
176 	__u32 btf_datasec:1;
177 	/* BPF_F_MMAPABLE is supported for arrays */
178 	__u32 array_mmap:1;
179 	/* BTF_FUNC_GLOBAL is supported */
180 	__u32 btf_func_global:1;
181 	/* kernel support for expected_attach_type in BPF_PROG_LOAD */
182 	__u32 exp_attach_type:1;
183 };
184 
185 enum reloc_type {
186 	RELO_LD64,
187 	RELO_CALL,
188 	RELO_DATA,
189 	RELO_EXTERN,
190 };
191 
192 struct reloc_desc {
193 	enum reloc_type type;
194 	int insn_idx;
195 	int map_idx;
196 	int sym_off;
197 };
198 
199 struct bpf_sec_def;
200 
201 typedef struct bpf_link *(*attach_fn_t)(const struct bpf_sec_def *sec,
202 					struct bpf_program *prog);
203 
204 struct bpf_sec_def {
205 	const char *sec;
206 	size_t len;
207 	enum bpf_prog_type prog_type;
208 	enum bpf_attach_type expected_attach_type;
209 	bool is_exp_attach_type_optional;
210 	bool is_attachable;
211 	bool is_attach_btf;
212 	attach_fn_t attach_fn;
213 };
214 
215 /*
216  * bpf_prog should be a better name but it has been used in
217  * linux/filter.h.
218  */
219 struct bpf_program {
220 	/* Index in elf obj file, for relocation use. */
221 	int idx;
222 	char *name;
223 	int prog_ifindex;
224 	char *section_name;
225 	const struct bpf_sec_def *sec_def;
226 	/* section_name with / replaced by _; makes recursive pinning
227 	 * in bpf_object__pin_programs easier
228 	 */
229 	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 	struct {
240 		int nr;
241 		int *fds;
242 	} instances;
243 	bpf_program_prep_t preprocessor;
244 
245 	struct bpf_object *obj;
246 	void *priv;
247 	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 	void *func_info;
253 	__u32 func_info_rec_size;
254 	__u32 func_info_cnt;
255 
256 	struct bpf_capabilities *caps;
257 
258 	void *line_info;
259 	__u32 line_info_rec_size;
260 	__u32 line_info_cnt;
261 	__u32 prog_flags;
262 };
263 
264 struct bpf_struct_ops {
265 	const char *tname;
266 	const struct btf_type *type;
267 	struct bpf_program **progs;
268 	__u32 *kern_func_off;
269 	/* e.g. struct tcp_congestion_ops in bpf_prog's btf format */
270 	void *data;
271 	/* e.g. struct bpf_struct_ops_tcp_congestion_ops in
272 	 *      btf_vmlinux's format.
273 	 * struct bpf_struct_ops_tcp_congestion_ops {
274 	 *	[... some other kernel fields ...]
275 	 *	struct tcp_congestion_ops data;
276 	 * }
277 	 * kern_vdata-size == sizeof(struct bpf_struct_ops_tcp_congestion_ops)
278 	 * bpf_map__init_kern_struct_ops() will populate the "kern_vdata"
279 	 * from "data".
280 	 */
281 	void *kern_vdata;
282 	__u32 type_id;
283 };
284 
285 #define DATA_SEC ".data"
286 #define BSS_SEC ".bss"
287 #define RODATA_SEC ".rodata"
288 #define KCONFIG_SEC ".kconfig"
289 #define KSYMS_SEC ".ksyms"
290 #define STRUCT_OPS_SEC ".struct_ops"
291 
292 enum libbpf_map_type {
293 	LIBBPF_MAP_UNSPEC,
294 	LIBBPF_MAP_DATA,
295 	LIBBPF_MAP_BSS,
296 	LIBBPF_MAP_RODATA,
297 	LIBBPF_MAP_KCONFIG,
298 };
299 
300 static const char * const libbpf_type_to_btf_name[] = {
301 	[LIBBPF_MAP_DATA]	= DATA_SEC,
302 	[LIBBPF_MAP_BSS]	= BSS_SEC,
303 	[LIBBPF_MAP_RODATA]	= RODATA_SEC,
304 	[LIBBPF_MAP_KCONFIG]	= KCONFIG_SEC,
305 };
306 
307 struct bpf_map {
308 	char *name;
309 	int fd;
310 	int sec_idx;
311 	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 	__u32 btf_key_type_id;
318 	__u32 btf_value_type_id;
319 	__u32 btf_vmlinux_value_type_id;
320 	void *priv;
321 	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 	struct bpf_map *inner_map;
326 	void **init_slots;
327 	int init_slots_sz;
328 	char *pin_path;
329 	bool pinned;
330 	bool reused;
331 };
332 
333 enum extern_type {
334 	EXT_UNKNOWN,
335 	EXT_KCFG,
336 	EXT_KSYM,
337 };
338 
339 enum kcfg_type {
340 	KCFG_UNKNOWN,
341 	KCFG_CHAR,
342 	KCFG_BOOL,
343 	KCFG_INT,
344 	KCFG_TRISTATE,
345 	KCFG_CHAR_ARR,
346 };
347 
348 struct extern_desc {
349 	enum extern_type type;
350 	int sym_idx;
351 	int btf_id;
352 	int sec_btf_id;
353 	const char *name;
354 	bool is_set;
355 	bool is_weak;
356 	union {
357 		struct {
358 			enum kcfg_type type;
359 			int sz;
360 			int align;
361 			int data_off;
362 			bool is_signed;
363 		} kcfg;
364 		struct {
365 			unsigned long long addr;
366 		} ksym;
367 	};
368 };
369 
370 static LIST_HEAD(bpf_objects_list);
371 
372 struct bpf_object {
373 	char name[BPF_OBJ_NAME_LEN];
374 	char license[64];
375 	__u32 kern_version;
376 
377 	struct bpf_program *programs;
378 	size_t nr_programs;
379 	struct bpf_map *maps;
380 	size_t nr_maps;
381 	size_t maps_cap;
382 
383 	char *kconfig;
384 	struct extern_desc *externs;
385 	int nr_extern;
386 	int kconfig_map_idx;
387 
388 	bool loaded;
389 	bool has_pseudo_calls;
390 
391 	/*
392 	 * Information when doing elf related work. Only valid if fd
393 	 * is valid.
394 	 */
395 	struct {
396 		int fd;
397 		const void *obj_buf;
398 		size_t obj_buf_sz;
399 		Elf *elf;
400 		GElf_Ehdr ehdr;
401 		Elf_Data *symbols;
402 		Elf_Data *data;
403 		Elf_Data *rodata;
404 		Elf_Data *bss;
405 		Elf_Data *st_ops_data;
406 		size_t strtabidx;
407 		struct {
408 			GElf_Shdr shdr;
409 			Elf_Data *data;
410 		} *reloc_sects;
411 		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 		int data_shndx;
418 		int rodata_shndx;
419 		int bss_shndx;
420 		int st_ops_shndx;
421 	} efile;
422 	/*
423 	 * All loaded bpf_object is linked in a list, which is
424 	 * hidden to caller. bpf_objects__<func> handlers deal with
425 	 * all objects.
426 	 */
427 	struct list_head list;
428 
429 	struct btf *btf;
430 	/* Parse and load BTF vmlinux if any of the programs in the object need
431 	 * it at load time.
432 	 */
433 	struct btf *btf_vmlinux;
434 	struct btf_ext *btf_ext;
435 
436 	void *priv;
437 	bpf_object_clear_priv_t clear_priv;
438 
439 	struct bpf_capabilities caps;
440 
441 	char path[];
442 };
443 #define obj_elf_valid(o)	((o)->efile.elf)
444 
445 void bpf_program__unload(struct bpf_program *prog)
446 {
447 	int i;
448 
449 	if (!prog)
450 		return;
451 
452 	/*
453 	 * If the object is opened but the program was never loaded,
454 	 * it is possible that prog->instances.nr == -1.
455 	 */
456 	if (prog->instances.nr > 0) {
457 		for (i = 0; i < prog->instances.nr; i++)
458 			zclose(prog->instances.fds[i]);
459 	} else if (prog->instances.nr != -1) {
460 		pr_warn("Internal error: instances.nr is %d\n",
461 			prog->instances.nr);
462 	}
463 
464 	prog->instances.nr = -1;
465 	zfree(&prog->instances.fds);
466 
467 	zfree(&prog->func_info);
468 	zfree(&prog->line_info);
469 }
470 
471 static void bpf_program__exit(struct bpf_program *prog)
472 {
473 	if (!prog)
474 		return;
475 
476 	if (prog->clear_priv)
477 		prog->clear_priv(prog, prog->priv);
478 
479 	prog->priv = NULL;
480 	prog->clear_priv = NULL;
481 
482 	bpf_program__unload(prog);
483 	zfree(&prog->name);
484 	zfree(&prog->section_name);
485 	zfree(&prog->pin_name);
486 	zfree(&prog->insns);
487 	zfree(&prog->reloc_desc);
488 
489 	prog->nr_reloc = 0;
490 	prog->insns_cnt = 0;
491 	prog->idx = -1;
492 }
493 
494 static char *__bpf_program__pin_name(struct bpf_program *prog)
495 {
496 	char *name, *p;
497 
498 	name = p = strdup(prog->section_name);
499 	while ((p = strchr(p, '/')))
500 		*p = '_';
501 
502 	return name;
503 }
504 
505 static int
506 bpf_program__init(void *data, size_t size, char *section_name, int idx,
507 		  struct bpf_program *prog)
508 {
509 	const size_t bpf_insn_sz = sizeof(struct bpf_insn);
510 
511 	if (size == 0 || size % bpf_insn_sz) {
512 		pr_warn("corrupted section '%s', size: %zu\n",
513 			section_name, size);
514 		return -EINVAL;
515 	}
516 
517 	memset(prog, 0, sizeof(*prog));
518 
519 	prog->section_name = strdup(section_name);
520 	if (!prog->section_name) {
521 		pr_warn("failed to alloc name for prog under section(%d) %s\n",
522 			idx, section_name);
523 		goto errout;
524 	}
525 
526 	prog->pin_name = __bpf_program__pin_name(prog);
527 	if (!prog->pin_name) {
528 		pr_warn("failed to alloc pin name for prog under section(%d) %s\n",
529 			idx, section_name);
530 		goto errout;
531 	}
532 
533 	prog->insns = malloc(size);
534 	if (!prog->insns) {
535 		pr_warn("failed to alloc insns for prog under section %s\n",
536 			section_name);
537 		goto errout;
538 	}
539 	prog->insns_cnt = size / bpf_insn_sz;
540 	memcpy(prog->insns, data, size);
541 	prog->idx = idx;
542 	prog->instances.fds = NULL;
543 	prog->instances.nr = -1;
544 	prog->type = BPF_PROG_TYPE_UNSPEC;
545 	prog->load = true;
546 
547 	return 0;
548 errout:
549 	bpf_program__exit(prog);
550 	return -ENOMEM;
551 }
552 
553 static int
554 bpf_object__add_program(struct bpf_object *obj, void *data, size_t size,
555 			char *section_name, int idx)
556 {
557 	struct bpf_program prog, *progs;
558 	int nr_progs, err;
559 
560 	err = bpf_program__init(data, size, section_name, idx, &prog);
561 	if (err)
562 		return err;
563 
564 	prog.caps = &obj->caps;
565 	progs = obj->programs;
566 	nr_progs = obj->nr_programs;
567 
568 	progs = reallocarray(progs, nr_progs + 1, sizeof(progs[0]));
569 	if (!progs) {
570 		/*
571 		 * In this case the original obj->programs
572 		 * is still valid, so don't need special treat for
573 		 * bpf_close_object().
574 		 */
575 		pr_warn("failed to alloc a new program under section '%s'\n",
576 			section_name);
577 		bpf_program__exit(&prog);
578 		return -ENOMEM;
579 	}
580 
581 	pr_debug("found program %s\n", prog.section_name);
582 	obj->programs = progs;
583 	obj->nr_programs = nr_progs + 1;
584 	prog.obj = obj;
585 	progs[nr_progs] = prog;
586 	return 0;
587 }
588 
589 static int
590 bpf_object__init_prog_names(struct bpf_object *obj)
591 {
592 	Elf_Data *symbols = obj->efile.symbols;
593 	struct bpf_program *prog;
594 	size_t pi, si;
595 
596 	for (pi = 0; pi < obj->nr_programs; pi++) {
597 		const char *name = NULL;
598 
599 		prog = &obj->programs[pi];
600 
601 		for (si = 0; si < symbols->d_size / sizeof(GElf_Sym) && !name;
602 		     si++) {
603 			GElf_Sym sym;
604 
605 			if (!gelf_getsym(symbols, si, &sym))
606 				continue;
607 			if (sym.st_shndx != prog->idx)
608 				continue;
609 			if (GELF_ST_BIND(sym.st_info) != STB_GLOBAL)
610 				continue;
611 
612 			name = elf_strptr(obj->efile.elf,
613 					  obj->efile.strtabidx,
614 					  sym.st_name);
615 			if (!name) {
616 				pr_warn("failed to get sym name string for prog %s\n",
617 					prog->section_name);
618 				return -LIBBPF_ERRNO__LIBELF;
619 			}
620 		}
621 
622 		if (!name && prog->idx == obj->efile.text_shndx)
623 			name = ".text";
624 
625 		if (!name) {
626 			pr_warn("failed to find sym for prog %s\n",
627 				prog->section_name);
628 			return -EINVAL;
629 		}
630 
631 		prog->name = strdup(name);
632 		if (!prog->name) {
633 			pr_warn("failed to allocate memory for prog sym %s\n",
634 				name);
635 			return -ENOMEM;
636 		}
637 	}
638 
639 	return 0;
640 }
641 
642 static __u32 get_kernel_version(void)
643 {
644 	__u32 major, minor, patch;
645 	struct utsname info;
646 
647 	uname(&info);
648 	if (sscanf(info.release, "%u.%u.%u", &major, &minor, &patch) != 3)
649 		return 0;
650 	return KERNEL_VERSION(major, minor, patch);
651 }
652 
653 static const struct btf_member *
654 find_member_by_offset(const struct btf_type *t, __u32 bit_offset)
655 {
656 	struct btf_member *m;
657 	int i;
658 
659 	for (i = 0, m = btf_members(t); i < btf_vlen(t); i++, m++) {
660 		if (btf_member_bit_offset(t, i) == bit_offset)
661 			return m;
662 	}
663 
664 	return NULL;
665 }
666 
667 static const struct btf_member *
668 find_member_by_name(const struct btf *btf, const struct btf_type *t,
669 		    const char *name)
670 {
671 	struct btf_member *m;
672 	int i;
673 
674 	for (i = 0, m = btf_members(t); i < btf_vlen(t); i++, m++) {
675 		if (!strcmp(btf__name_by_offset(btf, m->name_off), name))
676 			return m;
677 	}
678 
679 	return NULL;
680 }
681 
682 #define STRUCT_OPS_VALUE_PREFIX "bpf_struct_ops_"
683 static int find_btf_by_prefix_kind(const struct btf *btf, const char *prefix,
684 				   const char *name, __u32 kind);
685 
686 static int
687 find_struct_ops_kern_types(const struct btf *btf, const char *tname,
688 			   const struct btf_type **type, __u32 *type_id,
689 			   const struct btf_type **vtype, __u32 *vtype_id,
690 			   const struct btf_member **data_member)
691 {
692 	const struct btf_type *kern_type, *kern_vtype;
693 	const struct btf_member *kern_data_member;
694 	__s32 kern_vtype_id, kern_type_id;
695 	__u32 i;
696 
697 	kern_type_id = btf__find_by_name_kind(btf, tname, BTF_KIND_STRUCT);
698 	if (kern_type_id < 0) {
699 		pr_warn("struct_ops init_kern: struct %s is not found in kernel BTF\n",
700 			tname);
701 		return kern_type_id;
702 	}
703 	kern_type = btf__type_by_id(btf, kern_type_id);
704 
705 	/* Find the corresponding "map_value" type that will be used
706 	 * in map_update(BPF_MAP_TYPE_STRUCT_OPS).  For example,
707 	 * find "struct bpf_struct_ops_tcp_congestion_ops" from the
708 	 * btf_vmlinux.
709 	 */
710 	kern_vtype_id = find_btf_by_prefix_kind(btf, STRUCT_OPS_VALUE_PREFIX,
711 						tname, BTF_KIND_STRUCT);
712 	if (kern_vtype_id < 0) {
713 		pr_warn("struct_ops init_kern: struct %s%s is not found in kernel BTF\n",
714 			STRUCT_OPS_VALUE_PREFIX, tname);
715 		return kern_vtype_id;
716 	}
717 	kern_vtype = btf__type_by_id(btf, kern_vtype_id);
718 
719 	/* Find "struct tcp_congestion_ops" from
720 	 * struct bpf_struct_ops_tcp_congestion_ops {
721 	 *	[ ... ]
722 	 *	struct tcp_congestion_ops data;
723 	 * }
724 	 */
725 	kern_data_member = btf_members(kern_vtype);
726 	for (i = 0; i < btf_vlen(kern_vtype); i++, kern_data_member++) {
727 		if (kern_data_member->type == kern_type_id)
728 			break;
729 	}
730 	if (i == btf_vlen(kern_vtype)) {
731 		pr_warn("struct_ops init_kern: struct %s data is not found in struct %s%s\n",
732 			tname, STRUCT_OPS_VALUE_PREFIX, tname);
733 		return -EINVAL;
734 	}
735 
736 	*type = kern_type;
737 	*type_id = kern_type_id;
738 	*vtype = kern_vtype;
739 	*vtype_id = kern_vtype_id;
740 	*data_member = kern_data_member;
741 
742 	return 0;
743 }
744 
745 static bool bpf_map__is_struct_ops(const struct bpf_map *map)
746 {
747 	return map->def.type == BPF_MAP_TYPE_STRUCT_OPS;
748 }
749 
750 /* Init the map's fields that depend on kern_btf */
751 static int bpf_map__init_kern_struct_ops(struct bpf_map *map,
752 					 const struct btf *btf,
753 					 const struct btf *kern_btf)
754 {
755 	const struct btf_member *member, *kern_member, *kern_data_member;
756 	const struct btf_type *type, *kern_type, *kern_vtype;
757 	__u32 i, kern_type_id, kern_vtype_id, kern_data_off;
758 	struct bpf_struct_ops *st_ops;
759 	void *data, *kern_data;
760 	const char *tname;
761 	int err;
762 
763 	st_ops = map->st_ops;
764 	type = st_ops->type;
765 	tname = st_ops->tname;
766 	err = find_struct_ops_kern_types(kern_btf, tname,
767 					 &kern_type, &kern_type_id,
768 					 &kern_vtype, &kern_vtype_id,
769 					 &kern_data_member);
770 	if (err)
771 		return err;
772 
773 	pr_debug("struct_ops init_kern %s: type_id:%u kern_type_id:%u kern_vtype_id:%u\n",
774 		 map->name, st_ops->type_id, kern_type_id, kern_vtype_id);
775 
776 	map->def.value_size = kern_vtype->size;
777 	map->btf_vmlinux_value_type_id = kern_vtype_id;
778 
779 	st_ops->kern_vdata = calloc(1, kern_vtype->size);
780 	if (!st_ops->kern_vdata)
781 		return -ENOMEM;
782 
783 	data = st_ops->data;
784 	kern_data_off = kern_data_member->offset / 8;
785 	kern_data = st_ops->kern_vdata + kern_data_off;
786 
787 	member = btf_members(type);
788 	for (i = 0; i < btf_vlen(type); i++, member++) {
789 		const struct btf_type *mtype, *kern_mtype;
790 		__u32 mtype_id, kern_mtype_id;
791 		void *mdata, *kern_mdata;
792 		__s64 msize, kern_msize;
793 		__u32 moff, kern_moff;
794 		__u32 kern_member_idx;
795 		const char *mname;
796 
797 		mname = btf__name_by_offset(btf, member->name_off);
798 		kern_member = find_member_by_name(kern_btf, kern_type, mname);
799 		if (!kern_member) {
800 			pr_warn("struct_ops init_kern %s: Cannot find member %s in kernel BTF\n",
801 				map->name, mname);
802 			return -ENOTSUP;
803 		}
804 
805 		kern_member_idx = kern_member - btf_members(kern_type);
806 		if (btf_member_bitfield_size(type, i) ||
807 		    btf_member_bitfield_size(kern_type, kern_member_idx)) {
808 			pr_warn("struct_ops init_kern %s: bitfield %s is not supported\n",
809 				map->name, mname);
810 			return -ENOTSUP;
811 		}
812 
813 		moff = member->offset / 8;
814 		kern_moff = kern_member->offset / 8;
815 
816 		mdata = data + moff;
817 		kern_mdata = kern_data + kern_moff;
818 
819 		mtype = skip_mods_and_typedefs(btf, member->type, &mtype_id);
820 		kern_mtype = skip_mods_and_typedefs(kern_btf, kern_member->type,
821 						    &kern_mtype_id);
822 		if (BTF_INFO_KIND(mtype->info) !=
823 		    BTF_INFO_KIND(kern_mtype->info)) {
824 			pr_warn("struct_ops init_kern %s: Unmatched member type %s %u != %u(kernel)\n",
825 				map->name, mname, BTF_INFO_KIND(mtype->info),
826 				BTF_INFO_KIND(kern_mtype->info));
827 			return -ENOTSUP;
828 		}
829 
830 		if (btf_is_ptr(mtype)) {
831 			struct bpf_program *prog;
832 
833 			mtype = skip_mods_and_typedefs(btf, mtype->type, &mtype_id);
834 			kern_mtype = skip_mods_and_typedefs(kern_btf,
835 							    kern_mtype->type,
836 							    &kern_mtype_id);
837 			if (!btf_is_func_proto(mtype) ||
838 			    !btf_is_func_proto(kern_mtype)) {
839 				pr_warn("struct_ops init_kern %s: non func ptr %s is not supported\n",
840 					map->name, mname);
841 				return -ENOTSUP;
842 			}
843 
844 			prog = st_ops->progs[i];
845 			if (!prog) {
846 				pr_debug("struct_ops init_kern %s: func ptr %s is not set\n",
847 					 map->name, mname);
848 				continue;
849 			}
850 
851 			prog->attach_btf_id = kern_type_id;
852 			prog->expected_attach_type = kern_member_idx;
853 
854 			st_ops->kern_func_off[i] = kern_data_off + kern_moff;
855 
856 			pr_debug("struct_ops init_kern %s: func ptr %s is set to prog %s from data(+%u) to kern_data(+%u)\n",
857 				 map->name, mname, prog->name, moff,
858 				 kern_moff);
859 
860 			continue;
861 		}
862 
863 		msize = btf__resolve_size(btf, mtype_id);
864 		kern_msize = btf__resolve_size(kern_btf, kern_mtype_id);
865 		if (msize < 0 || kern_msize < 0 || msize != kern_msize) {
866 			pr_warn("struct_ops init_kern %s: Error in size of member %s: %zd != %zd(kernel)\n",
867 				map->name, mname, (ssize_t)msize,
868 				(ssize_t)kern_msize);
869 			return -ENOTSUP;
870 		}
871 
872 		pr_debug("struct_ops init_kern %s: copy %s %u bytes from data(+%u) to kern_data(+%u)\n",
873 			 map->name, mname, (unsigned int)msize,
874 			 moff, kern_moff);
875 		memcpy(kern_mdata, mdata, msize);
876 	}
877 
878 	return 0;
879 }
880 
881 static int bpf_object__init_kern_struct_ops_maps(struct bpf_object *obj)
882 {
883 	struct bpf_map *map;
884 	size_t i;
885 	int err;
886 
887 	for (i = 0; i < obj->nr_maps; i++) {
888 		map = &obj->maps[i];
889 
890 		if (!bpf_map__is_struct_ops(map))
891 			continue;
892 
893 		err = bpf_map__init_kern_struct_ops(map, obj->btf,
894 						    obj->btf_vmlinux);
895 		if (err)
896 			return err;
897 	}
898 
899 	return 0;
900 }
901 
902 static int bpf_object__init_struct_ops_maps(struct bpf_object *obj)
903 {
904 	const struct btf_type *type, *datasec;
905 	const struct btf_var_secinfo *vsi;
906 	struct bpf_struct_ops *st_ops;
907 	const char *tname, *var_name;
908 	__s32 type_id, datasec_id;
909 	const struct btf *btf;
910 	struct bpf_map *map;
911 	__u32 i;
912 
913 	if (obj->efile.st_ops_shndx == -1)
914 		return 0;
915 
916 	btf = obj->btf;
917 	datasec_id = btf__find_by_name_kind(btf, STRUCT_OPS_SEC,
918 					    BTF_KIND_DATASEC);
919 	if (datasec_id < 0) {
920 		pr_warn("struct_ops init: DATASEC %s not found\n",
921 			STRUCT_OPS_SEC);
922 		return -EINVAL;
923 	}
924 
925 	datasec = btf__type_by_id(btf, datasec_id);
926 	vsi = btf_var_secinfos(datasec);
927 	for (i = 0; i < btf_vlen(datasec); i++, vsi++) {
928 		type = btf__type_by_id(obj->btf, vsi->type);
929 		var_name = btf__name_by_offset(obj->btf, type->name_off);
930 
931 		type_id = btf__resolve_type(obj->btf, vsi->type);
932 		if (type_id < 0) {
933 			pr_warn("struct_ops init: Cannot resolve var type_id %u in DATASEC %s\n",
934 				vsi->type, STRUCT_OPS_SEC);
935 			return -EINVAL;
936 		}
937 
938 		type = btf__type_by_id(obj->btf, type_id);
939 		tname = btf__name_by_offset(obj->btf, type->name_off);
940 		if (!tname[0]) {
941 			pr_warn("struct_ops init: anonymous type is not supported\n");
942 			return -ENOTSUP;
943 		}
944 		if (!btf_is_struct(type)) {
945 			pr_warn("struct_ops init: %s is not a struct\n", tname);
946 			return -EINVAL;
947 		}
948 
949 		map = bpf_object__add_map(obj);
950 		if (IS_ERR(map))
951 			return PTR_ERR(map);
952 
953 		map->sec_idx = obj->efile.st_ops_shndx;
954 		map->sec_offset = vsi->offset;
955 		map->name = strdup(var_name);
956 		if (!map->name)
957 			return -ENOMEM;
958 
959 		map->def.type = BPF_MAP_TYPE_STRUCT_OPS;
960 		map->def.key_size = sizeof(int);
961 		map->def.value_size = type->size;
962 		map->def.max_entries = 1;
963 
964 		map->st_ops = calloc(1, sizeof(*map->st_ops));
965 		if (!map->st_ops)
966 			return -ENOMEM;
967 		st_ops = map->st_ops;
968 		st_ops->data = malloc(type->size);
969 		st_ops->progs = calloc(btf_vlen(type), sizeof(*st_ops->progs));
970 		st_ops->kern_func_off = malloc(btf_vlen(type) *
971 					       sizeof(*st_ops->kern_func_off));
972 		if (!st_ops->data || !st_ops->progs || !st_ops->kern_func_off)
973 			return -ENOMEM;
974 
975 		if (vsi->offset + type->size > obj->efile.st_ops_data->d_size) {
976 			pr_warn("struct_ops init: var %s is beyond the end of DATASEC %s\n",
977 				var_name, STRUCT_OPS_SEC);
978 			return -EINVAL;
979 		}
980 
981 		memcpy(st_ops->data,
982 		       obj->efile.st_ops_data->d_buf + vsi->offset,
983 		       type->size);
984 		st_ops->tname = tname;
985 		st_ops->type = type;
986 		st_ops->type_id = type_id;
987 
988 		pr_debug("struct_ops init: struct %s(type_id=%u) %s found at offset %u\n",
989 			 tname, type_id, var_name, vsi->offset);
990 	}
991 
992 	return 0;
993 }
994 
995 static struct bpf_object *bpf_object__new(const char *path,
996 					  const void *obj_buf,
997 					  size_t obj_buf_sz,
998 					  const char *obj_name)
999 {
1000 	struct bpf_object *obj;
1001 	char *end;
1002 
1003 	obj = calloc(1, sizeof(struct bpf_object) + strlen(path) + 1);
1004 	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 	if (obj_name) {
1011 		strncpy(obj->name, obj_name, sizeof(obj->name) - 1);
1012 		obj->name[sizeof(obj->name) - 1] = 0;
1013 	} else {
1014 		/* Using basename() GNU version which doesn't modify arg. */
1015 		strncpy(obj->name, basename((void *)path),
1016 			sizeof(obj->name) - 1);
1017 		end = strchr(obj->name, '.');
1018 		if (end)
1019 			*end = 0;
1020 	}
1021 
1022 	obj->efile.fd = -1;
1023 	/*
1024 	 * Caller of this function should also call
1025 	 * bpf_object__elf_finish() after data collection to return
1026 	 * obj_buf to user. If not, we should duplicate the buffer to
1027 	 * avoid user freeing them before elf finish.
1028 	 */
1029 	obj->efile.obj_buf = obj_buf;
1030 	obj->efile.obj_buf_sz = obj_buf_sz;
1031 	obj->efile.maps_shndx = -1;
1032 	obj->efile.btf_maps_shndx = -1;
1033 	obj->efile.data_shndx = -1;
1034 	obj->efile.rodata_shndx = -1;
1035 	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 	INIT_LIST_HEAD(&obj->list);
1043 	list_add(&obj->list, &bpf_objects_list);
1044 	return obj;
1045 }
1046 
1047 static void bpf_object__elf_finish(struct bpf_object *obj)
1048 {
1049 	if (!obj_elf_valid(obj))
1050 		return;
1051 
1052 	if (obj->efile.elf) {
1053 		elf_end(obj->efile.elf);
1054 		obj->efile.elf = NULL;
1055 	}
1056 	obj->efile.symbols = NULL;
1057 	obj->efile.data = NULL;
1058 	obj->efile.rodata = NULL;
1059 	obj->efile.bss = NULL;
1060 	obj->efile.st_ops_data = NULL;
1061 
1062 	zfree(&obj->efile.reloc_sects);
1063 	obj->efile.nr_reloc_sects = 0;
1064 	zclose(obj->efile.fd);
1065 	obj->efile.obj_buf = NULL;
1066 	obj->efile.obj_buf_sz = 0;
1067 }
1068 
1069 static int bpf_object__elf_init(struct bpf_object *obj)
1070 {
1071 	int err = 0;
1072 	GElf_Ehdr *ep;
1073 
1074 	if (obj_elf_valid(obj)) {
1075 		pr_warn("elf init: internal error\n");
1076 		return -LIBBPF_ERRNO__LIBELF;
1077 	}
1078 
1079 	if (obj->efile.obj_buf_sz > 0) {
1080 		/*
1081 		 * obj_buf should have been validated by
1082 		 * bpf_object__open_buffer().
1083 		 */
1084 		obj->efile.elf = elf_memory((char *)obj->efile.obj_buf,
1085 					    obj->efile.obj_buf_sz);
1086 	} else {
1087 		obj->efile.fd = open(obj->path, O_RDONLY);
1088 		if (obj->efile.fd < 0) {
1089 			char errmsg[STRERR_BUFSIZE], *cp;
1090 
1091 			err = -errno;
1092 			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 		goto errout;
1105 	}
1106 
1107 	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 		goto errout;
1111 	}
1112 	ep = &obj->efile.ehdr;
1113 
1114 	/* Old LLVM set e_machine to EM_NONE */
1115 	if (ep->e_type != ET_REL ||
1116 	    (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 		goto errout;
1120 	}
1121 
1122 	return 0;
1123 errout:
1124 	bpf_object__elf_finish(obj);
1125 	return err;
1126 }
1127 
1128 static int bpf_object__check_endianness(struct bpf_object *obj)
1129 {
1130 #if __BYTE_ORDER == __LITTLE_ENDIAN
1131 	if (obj->efile.ehdr.e_ident[EI_DATA] == ELFDATA2LSB)
1132 		return 0;
1133 #elif __BYTE_ORDER == __BIG_ENDIAN
1134 	if (obj->efile.ehdr.e_ident[EI_DATA] == ELFDATA2MSB)
1135 		return 0;
1136 #else
1137 # error "Unrecognized __BYTE_ORDER__"
1138 #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 	pr_debug("license of %s is %s\n", obj->path, obj->license);
1148 	return 0;
1149 }
1150 
1151 static int
1152 bpf_object__init_kversion(struct bpf_object *obj, void *data, size_t size)
1153 {
1154 	__u32 kver;
1155 
1156 	if (size != sizeof(kver)) {
1157 		pr_warn("invalid kver section in %s\n", obj->path);
1158 		return -LIBBPF_ERRNO__FORMAT;
1159 	}
1160 	memcpy(&kver, data, sizeof(kver));
1161 	obj->kern_version = kver;
1162 	pr_debug("kernel version of %s is %x\n", obj->path, obj->kern_version);
1163 	return 0;
1164 }
1165 
1166 static bool bpf_map_type__is_map_in_map(enum bpf_map_type type)
1167 {
1168 	if (type == BPF_MAP_TYPE_ARRAY_OF_MAPS ||
1169 	    type == BPF_MAP_TYPE_HASH_OF_MAPS)
1170 		return true;
1171 	return false;
1172 }
1173 
1174 static int bpf_object_search_section_size(const struct bpf_object *obj,
1175 					  const char *name, size_t *d_size)
1176 {
1177 	const GElf_Ehdr *ep = &obj->efile.ehdr;
1178 	Elf *elf = obj->efile.elf;
1179 	Elf_Scn *scn = NULL;
1180 	int idx = 0;
1181 
1182 	while ((scn = elf_nextscn(elf, scn)) != NULL) {
1183 		const char *sec_name;
1184 		Elf_Data *data;
1185 		GElf_Shdr sh;
1186 
1187 		idx++;
1188 		if (gelf_getshdr(scn, &sh) != &sh) {
1189 			pr_warn("failed to get section(%d) header from %s\n",
1190 				idx, obj->path);
1191 			return -EIO;
1192 		}
1193 
1194 		sec_name = elf_strptr(elf, ep->e_shstrndx, sh.sh_name);
1195 		if (!sec_name) {
1196 			pr_warn("failed to get section(%d) name from %s\n",
1197 				idx, obj->path);
1198 			return -EIO;
1199 		}
1200 
1201 		if (strcmp(name, sec_name))
1202 			continue;
1203 
1204 		data = elf_getdata(scn, 0);
1205 		if (!data) {
1206 			pr_warn("failed to get section(%d) data from %s(%s)\n",
1207 				idx, name, obj->path);
1208 			return -EIO;
1209 		}
1210 
1211 		*d_size = data->d_size;
1212 		return 0;
1213 	}
1214 
1215 	return -ENOENT;
1216 }
1217 
1218 int bpf_object__section_size(const struct bpf_object *obj, const char *name,
1219 			     __u32 *size)
1220 {
1221 	int ret = -ENOENT;
1222 	size_t d_size;
1223 
1224 	*size = 0;
1225 	if (!name) {
1226 		return -EINVAL;
1227 	} else if (!strcmp(name, DATA_SEC)) {
1228 		if (obj->efile.data)
1229 			*size = obj->efile.data->d_size;
1230 	} else if (!strcmp(name, BSS_SEC)) {
1231 		if (obj->efile.bss)
1232 			*size = obj->efile.bss->d_size;
1233 	} else if (!strcmp(name, RODATA_SEC)) {
1234 		if (obj->efile.rodata)
1235 			*size = obj->efile.rodata->d_size;
1236 	} else if (!strcmp(name, STRUCT_OPS_SEC)) {
1237 		if (obj->efile.st_ops_data)
1238 			*size = obj->efile.st_ops_data->d_size;
1239 	} else {
1240 		ret = bpf_object_search_section_size(obj, name, &d_size);
1241 		if (!ret)
1242 			*size = d_size;
1243 	}
1244 
1245 	return *size ? 0 : ret;
1246 }
1247 
1248 int bpf_object__variable_offset(const struct bpf_object *obj, const char *name,
1249 				__u32 *off)
1250 {
1251 	Elf_Data *symbols = obj->efile.symbols;
1252 	const char *sname;
1253 	size_t si;
1254 
1255 	if (!name || !off)
1256 		return -EINVAL;
1257 
1258 	for (si = 0; si < symbols->d_size / sizeof(GElf_Sym); si++) {
1259 		GElf_Sym sym;
1260 
1261 		if (!gelf_getsym(symbols, si, &sym))
1262 			continue;
1263 		if (GELF_ST_BIND(sym.st_info) != STB_GLOBAL ||
1264 		    GELF_ST_TYPE(sym.st_info) != STT_OBJECT)
1265 			continue;
1266 
1267 		sname = elf_strptr(obj->efile.elf, obj->efile.strtabidx,
1268 				   sym.st_name);
1269 		if (!sname) {
1270 			pr_warn("failed to get sym name string for var %s\n",
1271 				name);
1272 			return -EIO;
1273 		}
1274 		if (strcmp(name, sname) == 0) {
1275 			*off = sym.st_value;
1276 			return 0;
1277 		}
1278 	}
1279 
1280 	return -ENOENT;
1281 }
1282 
1283 static struct bpf_map *bpf_object__add_map(struct bpf_object *obj)
1284 {
1285 	struct bpf_map *new_maps;
1286 	size_t new_cap;
1287 	int i;
1288 
1289 	if (obj->nr_maps < obj->maps_cap)
1290 		return &obj->maps[obj->nr_maps++];
1291 
1292 	new_cap = max((size_t)4, obj->maps_cap * 3 / 2);
1293 	new_maps = realloc(obj->maps, new_cap * sizeof(*obj->maps));
1294 	if (!new_maps) {
1295 		pr_warn("alloc maps for object failed\n");
1296 		return ERR_PTR(-ENOMEM);
1297 	}
1298 
1299 	obj->maps_cap = new_cap;
1300 	obj->maps = new_maps;
1301 
1302 	/* zero out new maps */
1303 	memset(obj->maps + obj->nr_maps, 0,
1304 	       (obj->maps_cap - obj->nr_maps) * sizeof(*obj->maps));
1305 	/*
1306 	 * fill all fd with -1 so won't close incorrect fd (fd=0 is stdin)
1307 	 * when failure (zclose won't close negative fd)).
1308 	 */
1309 	for (i = obj->nr_maps; i < obj->maps_cap; i++) {
1310 		obj->maps[i].fd = -1;
1311 		obj->maps[i].inner_map_fd = -1;
1312 	}
1313 
1314 	return &obj->maps[obj->nr_maps++];
1315 }
1316 
1317 static size_t bpf_map_mmap_sz(const struct bpf_map *map)
1318 {
1319 	long page_sz = sysconf(_SC_PAGE_SIZE);
1320 	size_t map_sz;
1321 
1322 	map_sz = (size_t)roundup(map->def.value_size, 8) * map->def.max_entries;
1323 	map_sz = roundup(map_sz, page_sz);
1324 	return map_sz;
1325 }
1326 
1327 static char *internal_map_name(struct bpf_object *obj,
1328 			       enum libbpf_map_type type)
1329 {
1330 	char map_name[BPF_OBJ_NAME_LEN], *p;
1331 	const char *sfx = libbpf_type_to_btf_name[type];
1332 	int sfx_len = max((size_t)7, strlen(sfx));
1333 	int pfx_len = min((size_t)BPF_OBJ_NAME_LEN - sfx_len - 1,
1334 			  strlen(obj->name));
1335 
1336 	snprintf(map_name, sizeof(map_name), "%.*s%.*s", pfx_len, obj->name,
1337 		 sfx_len, libbpf_type_to_btf_name[type]);
1338 
1339 	/* sanitise map name to characters allowed by kernel */
1340 	for (p = map_name; *p && p < map_name + sizeof(map_name); p++)
1341 		if (!isalnum(*p) && *p != '_' && *p != '.')
1342 			*p = '_';
1343 
1344 	return strdup(map_name);
1345 }
1346 
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 	struct bpf_map_def *def;
1352 	struct bpf_map *map;
1353 	int err;
1354 
1355 	map = bpf_object__add_map(obj);
1356 	if (IS_ERR(map))
1357 		return PTR_ERR(map);
1358 
1359 	map->libbpf_type = type;
1360 	map->sec_idx = sec_idx;
1361 	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 		return -ENOMEM;
1366 	}
1367 
1368 	def = &map->def;
1369 	def->type = BPF_MAP_TYPE_ARRAY;
1370 	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 	map->mmaped = mmap(NULL, bpf_map_mmap_sz(map), PROT_READ | PROT_WRITE,
1381 			   MAP_SHARED | MAP_ANONYMOUS, -1, 0);
1382 	if (map->mmaped == MAP_FAILED) {
1383 		err = -errno;
1384 		map->mmaped = NULL;
1385 		pr_warn("failed to alloc map '%s' content buffer: %d\n",
1386 			map->name, err);
1387 		zfree(&map->name);
1388 		return err;
1389 	}
1390 
1391 	if (data)
1392 		memcpy(map->mmaped, data, data_sz);
1393 
1394 	pr_debug("map %td is \"%s\"\n", map - obj->maps, map->name);
1395 	return 0;
1396 }
1397 
1398 static int bpf_object__init_global_data_maps(struct bpf_object *obj)
1399 {
1400 	int err;
1401 
1402 	/*
1403 	 * Populate obj->maps with libbpf internal maps.
1404 	 */
1405 	if (obj->efile.data_shndx >= 0) {
1406 		err = bpf_object__init_internal_map(obj, LIBBPF_MAP_DATA,
1407 						    obj->efile.data_shndx,
1408 						    obj->efile.data->d_buf,
1409 						    obj->efile.data->d_size);
1410 		if (err)
1411 			return err;
1412 	}
1413 	if (obj->efile.rodata_shndx >= 0) {
1414 		err = bpf_object__init_internal_map(obj, LIBBPF_MAP_RODATA,
1415 						    obj->efile.rodata_shndx,
1416 						    obj->efile.rodata->d_buf,
1417 						    obj->efile.rodata->d_size);
1418 		if (err)
1419 			return err;
1420 	}
1421 	if (obj->efile.bss_shndx >= 0) {
1422 		err = bpf_object__init_internal_map(obj, LIBBPF_MAP_BSS,
1423 						    obj->efile.bss_shndx,
1424 						    NULL,
1425 						    obj->efile.bss->d_size);
1426 		if (err)
1427 			return err;
1428 	}
1429 	return 0;
1430 }
1431 
1432 
1433 static struct extern_desc *find_extern_by_name(const struct bpf_object *obj,
1434 					       const void *name)
1435 {
1436 	int i;
1437 
1438 	for (i = 0; i < obj->nr_extern; i++) {
1439 		if (strcmp(obj->externs[i].name, name) == 0)
1440 			return &obj->externs[i];
1441 	}
1442 	return NULL;
1443 }
1444 
1445 static int set_kcfg_value_tri(struct extern_desc *ext, void *ext_val,
1446 			      char value)
1447 {
1448 	switch (ext->kcfg.type) {
1449 	case KCFG_BOOL:
1450 		if (value == 'm') {
1451 			pr_warn("extern (kcfg) %s=%c should be tristate or char\n",
1452 				ext->name, value);
1453 			return -EINVAL;
1454 		}
1455 		*(bool *)ext_val = value == 'y' ? true : false;
1456 		break;
1457 	case KCFG_TRISTATE:
1458 		if (value == 'y')
1459 			*(enum libbpf_tristate *)ext_val = TRI_YES;
1460 		else if (value == 'm')
1461 			*(enum libbpf_tristate *)ext_val = TRI_MODULE;
1462 		else /* value == 'n' */
1463 			*(enum libbpf_tristate *)ext_val = TRI_NO;
1464 		break;
1465 	case KCFG_CHAR:
1466 		*(char *)ext_val = value;
1467 		break;
1468 	case KCFG_UNKNOWN:
1469 	case KCFG_INT:
1470 	case KCFG_CHAR_ARR:
1471 	default:
1472 		pr_warn("extern (kcfg) %s=%c should be bool, tristate, or char\n",
1473 			ext->name, value);
1474 		return -EINVAL;
1475 	}
1476 	ext->is_set = true;
1477 	return 0;
1478 }
1479 
1480 static int set_kcfg_value_str(struct extern_desc *ext, char *ext_val,
1481 			      const char *value)
1482 {
1483 	size_t len;
1484 
1485 	if (ext->kcfg.type != KCFG_CHAR_ARR) {
1486 		pr_warn("extern (kcfg) %s=%s should be char array\n", ext->name, value);
1487 		return -EINVAL;
1488 	}
1489 
1490 	len = strlen(value);
1491 	if (value[len - 1] != '"') {
1492 		pr_warn("extern (kcfg) '%s': invalid string config '%s'\n",
1493 			ext->name, value);
1494 		return -EINVAL;
1495 	}
1496 
1497 	/* strip quotes */
1498 	len -= 2;
1499 	if (len >= ext->kcfg.sz) {
1500 		pr_warn("extern (kcfg) '%s': long string config %s of (%zu bytes) truncated to %d bytes\n",
1501 			ext->name, value, len, ext->kcfg.sz - 1);
1502 		len = ext->kcfg.sz - 1;
1503 	}
1504 	memcpy(ext_val, value + 1, len);
1505 	ext_val[len] = '\0';
1506 	ext->is_set = true;
1507 	return 0;
1508 }
1509 
1510 static int parse_u64(const char *value, __u64 *res)
1511 {
1512 	char *value_end;
1513 	int err;
1514 
1515 	errno = 0;
1516 	*res = strtoull(value, &value_end, 0);
1517 	if (errno) {
1518 		err = -errno;
1519 		pr_warn("failed to parse '%s' as integer: %d\n", value, err);
1520 		return err;
1521 	}
1522 	if (*value_end) {
1523 		pr_warn("failed to parse '%s' as integer completely\n", value);
1524 		return -EINVAL;
1525 	}
1526 	return 0;
1527 }
1528 
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 		return true;
1535 
1536 	/* Validate that value stored in u64 fits in integer of `ext->sz`
1537 	 * bytes size without any loss of information. If the target integer
1538 	 * is signed, we rely on the following limits of integer type of
1539 	 * Y bits and subsequent transformation:
1540 	 *
1541 	 *     -2^(Y-1) <= X           <= 2^(Y-1) - 1
1542 	 *            0 <= X + 2^(Y-1) <= 2^Y - 1
1543 	 *            0 <= X + 2^(Y-1) <  2^Y
1544 	 *
1545 	 *  For unsigned target integer, check that all the (64 - Y) bits are
1546 	 *  zero.
1547 	 */
1548 	if (ext->kcfg.is_signed)
1549 		return v + (1ULL << (bit_sz - 1)) < (1ULL << bit_sz);
1550 	else
1551 		return (v >> bit_sz) == 0;
1552 }
1553 
1554 static int set_kcfg_value_num(struct extern_desc *ext, void *ext_val,
1555 			      __u64 value)
1556 {
1557 	if (ext->kcfg.type != KCFG_INT && ext->kcfg.type != KCFG_CHAR) {
1558 		pr_warn("extern (kcfg) %s=%llu should be integer\n",
1559 			ext->name, (unsigned long long)value);
1560 		return -EINVAL;
1561 	}
1562 	if (!is_kcfg_value_in_range(ext, value)) {
1563 		pr_warn("extern (kcfg) %s=%llu value doesn't fit in %d bytes\n",
1564 			ext->name, (unsigned long long)value, ext->kcfg.sz);
1565 		return -ERANGE;
1566 	}
1567 	switch (ext->kcfg.sz) {
1568 		case 1: *(__u8 *)ext_val = value; break;
1569 		case 2: *(__u16 *)ext_val = value; break;
1570 		case 4: *(__u32 *)ext_val = value; break;
1571 		case 8: *(__u64 *)ext_val = value; break;
1572 		default:
1573 			return -EINVAL;
1574 	}
1575 	ext->is_set = true;
1576 	return 0;
1577 }
1578 
1579 static int bpf_object__process_kconfig_line(struct bpf_object *obj,
1580 					    char *buf, void *data)
1581 {
1582 	struct extern_desc *ext;
1583 	char *sep, *value;
1584 	int len, err = 0;
1585 	void *ext_val;
1586 	__u64 num;
1587 
1588 	if (strncmp(buf, "CONFIG_", 7))
1589 		return 0;
1590 
1591 	sep = strchr(buf, '=');
1592 	if (!sep) {
1593 		pr_warn("failed to parse '%s': no separator\n", buf);
1594 		return -EINVAL;
1595 	}
1596 
1597 	/* Trim ending '\n' */
1598 	len = strlen(buf);
1599 	if (buf[len - 1] == '\n')
1600 		buf[len - 1] = '\0';
1601 	/* Split on '=' and ensure that a value is present. */
1602 	*sep = '\0';
1603 	if (!sep[1]) {
1604 		*sep = '=';
1605 		pr_warn("failed to parse '%s': no value\n", buf);
1606 		return -EINVAL;
1607 	}
1608 
1609 	ext = find_extern_by_name(obj, buf);
1610 	if (!ext || ext->is_set)
1611 		return 0;
1612 
1613 	ext_val = data + ext->kcfg.data_off;
1614 	value = sep + 1;
1615 
1616 	switch (*value) {
1617 	case 'y': case 'n': case 'm':
1618 		err = set_kcfg_value_tri(ext, ext_val, *value);
1619 		break;
1620 	case '"':
1621 		err = set_kcfg_value_str(ext, ext_val, value);
1622 		break;
1623 	default:
1624 		/* assume integer */
1625 		err = parse_u64(value, &num);
1626 		if (err) {
1627 			pr_warn("extern (kcfg) %s=%s should be integer\n",
1628 				ext->name, value);
1629 			return err;
1630 		}
1631 		err = set_kcfg_value_num(ext, ext_val, num);
1632 		break;
1633 	}
1634 	if (err)
1635 		return err;
1636 	pr_debug("extern (kcfg) %s=%s\n", ext->name, value);
1637 	return 0;
1638 }
1639 
1640 static int bpf_object__read_kconfig_file(struct bpf_object *obj, void *data)
1641 {
1642 	char buf[PATH_MAX];
1643 	struct utsname uts;
1644 	int len, err = 0;
1645 	gzFile file;
1646 
1647 	uname(&uts);
1648 	len = snprintf(buf, PATH_MAX, "/boot/config-%s", uts.release);
1649 	if (len < 0)
1650 		return -EINVAL;
1651 	else if (len >= PATH_MAX)
1652 		return -ENAMETOOLONG;
1653 
1654 	/* gzopen also accepts uncompressed files. */
1655 	file = gzopen(buf, "r");
1656 	if (!file)
1657 		file = gzopen("/proc/config.gz", "r");
1658 
1659 	if (!file) {
1660 		pr_warn("failed to open system Kconfig\n");
1661 		return -ENOENT;
1662 	}
1663 
1664 	while (gzgets(file, buf, sizeof(buf))) {
1665 		err = bpf_object__process_kconfig_line(obj, buf, data);
1666 		if (err) {
1667 			pr_warn("error parsing system Kconfig line '%s': %d\n",
1668 				buf, err);
1669 			goto out;
1670 		}
1671 	}
1672 
1673 out:
1674 	gzclose(file);
1675 	return err;
1676 }
1677 
1678 static int bpf_object__read_kconfig_mem(struct bpf_object *obj,
1679 					const char *config, void *data)
1680 {
1681 	char buf[PATH_MAX];
1682 	int err = 0;
1683 	FILE *file;
1684 
1685 	file = fmemopen((void *)config, strlen(config), "r");
1686 	if (!file) {
1687 		err = -errno;
1688 		pr_warn("failed to open in-memory Kconfig: %d\n", err);
1689 		return err;
1690 	}
1691 
1692 	while (fgets(buf, sizeof(buf), file)) {
1693 		err = bpf_object__process_kconfig_line(obj, buf, data);
1694 		if (err) {
1695 			pr_warn("error parsing in-memory Kconfig line '%s': %d\n",
1696 				buf, err);
1697 			break;
1698 		}
1699 	}
1700 
1701 	fclose(file);
1702 	return err;
1703 }
1704 
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 	for (i = 0; i < obj->nr_extern; i++) {
1712 		ext = &obj->externs[i];
1713 		if (ext->type == EXT_KCFG)
1714 			last_ext = ext;
1715 	}
1716 
1717 	if (!last_ext)
1718 		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 					    obj->efile.symbols_shndx,
1723 					    NULL, map_sz);
1724 	if (err)
1725 		return err;
1726 
1727 	obj->kconfig_map_idx = obj->nr_maps - 1;
1728 
1729 	return 0;
1730 }
1731 
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 	Elf_Scn *scn;
1738 
1739 	if (obj->efile.maps_shndx < 0)
1740 		return 0;
1741 
1742 	if (!symbols)
1743 		return -EINVAL;
1744 
1745 	scn = elf_getscn(obj->efile.elf, obj->efile.maps_shndx);
1746 	if (scn)
1747 		data = elf_getdata(scn, NULL);
1748 	if (!scn || !data) {
1749 		pr_warn("failed to get Elf_Data from map section %d\n",
1750 			obj->efile.maps_shndx);
1751 		return -EINVAL;
1752 	}
1753 
1754 	/*
1755 	 * Count number of maps. Each map has a name.
1756 	 * Array of maps is not supported: only the first element is
1757 	 * considered.
1758 	 *
1759 	 * TODO: Detect array of map and report error.
1760 	 */
1761 	nr_syms = symbols->d_size / sizeof(GElf_Sym);
1762 	for (i = 0; i < nr_syms; i++) {
1763 		GElf_Sym sym;
1764 
1765 		if (!gelf_getsym(symbols, i, &sym))
1766 			continue;
1767 		if (sym.st_shndx != obj->efile.maps_shndx)
1768 			continue;
1769 		nr_maps++;
1770 	}
1771 	/* Assume equally sized map definitions */
1772 	pr_debug("maps in %s: %d maps in %zd bytes\n",
1773 		 obj->path, nr_maps, data->d_size);
1774 
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;
1781 
1782 	/* Fill obj->maps using data in "maps" section.  */
1783 	for (i = 0; i < nr_syms; i++) {
1784 		GElf_Sym sym;
1785 		const char *map_name;
1786 		struct bpf_map_def *def;
1787 		struct bpf_map *map;
1788 
1789 		if (!gelf_getsym(symbols, i, &sym))
1790 			continue;
1791 		if (sym.st_shndx != obj->efile.maps_shndx)
1792 			continue;
1793 
1794 		map = bpf_object__add_map(obj);
1795 		if (IS_ERR(map))
1796 			return PTR_ERR(map);
1797 
1798 		map_name = elf_strptr(obj->efile.elf, obj->efile.strtabidx,
1799 				      sym.st_name);
1800 		if (!map_name) {
1801 			pr_warn("failed to get map #%d name sym string for obj %s\n",
1802 				i, obj->path);
1803 			return -LIBBPF_ERRNO__FORMAT;
1804 		}
1805 
1806 		map->libbpf_type = LIBBPF_MAP_UNSPEC;
1807 		map->sec_idx = sym.st_shndx;
1808 		map->sec_offset = sym.st_value;
1809 		pr_debug("map '%s' (legacy): at sec_idx %d, offset %zu.\n",
1810 			 map_name, map->sec_idx, map->sec_offset);
1811 		if (sym.st_value + map_def_sz > data->d_size) {
1812 			pr_warn("corrupted maps section in %s: last map \"%s\" too small\n",
1813 				obj->path, map_name);
1814 			return -EINVAL;
1815 		}
1816 
1817 		map->name = strdup(map_name);
1818 		if (!map->name) {
1819 			pr_warn("failed to alloc map name\n");
1820 			return -ENOMEM;
1821 		}
1822 		pr_debug("map %d is \"%s\"\n", i, map->name);
1823 		def = (struct bpf_map_def *)(data->d_buf + sym.st_value);
1824 		/*
1825 		 * If the definition of the map in the object file fits in
1826 		 * bpf_map_def, copy it.  Any extra fields in our version
1827 		 * of bpf_map_def will default to zero as a result of the
1828 		 * calloc above.
1829 		 */
1830 		if (map_def_sz <= sizeof(struct bpf_map_def)) {
1831 			memcpy(&map->def, def, map_def_sz);
1832 		} else {
1833 			/*
1834 			 * Here the map structure being read is bigger than what
1835 			 * we expect, truncate if the excess bits are all zero.
1836 			 * If they are not zero, reject this map as
1837 			 * incompatible.
1838 			 */
1839 			char *b;
1840 
1841 			for (b = ((char *)def) + sizeof(struct bpf_map_def);
1842 			     b < ((char *)def) + map_def_sz; b++) {
1843 				if (*b != 0) {
1844 					pr_warn("maps section in %s: \"%s\" has unrecognized, non-zero options\n",
1845 						obj->path, map_name);
1846 					if (strict)
1847 						return -EINVAL;
1848 				}
1849 			}
1850 			memcpy(&map->def, def, sizeof(struct bpf_map_def));
1851 		}
1852 	}
1853 	return 0;
1854 }
1855 
1856 static const struct btf_type *
1857 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 	if (res_id)
1862 		*res_id = id;
1863 
1864 	while (btf_is_mod(t) || btf_is_typedef(t)) {
1865 		if (res_id)
1866 			*res_id = t->type;
1867 		t = btf__type_by_id(btf, t->type);
1868 	}
1869 
1870 	return t;
1871 }
1872 
1873 static const struct btf_type *
1874 resolve_func_ptr(const struct btf *btf, __u32 id, __u32 *res_id)
1875 {
1876 	const struct btf_type *t;
1877 
1878 	t = skip_mods_and_typedefs(btf, id, NULL);
1879 	if (!btf_is_ptr(t))
1880 		return NULL;
1881 
1882 	t = skip_mods_and_typedefs(btf, t->type, res_id);
1883 
1884 	return btf_is_func_proto(t) ? t : NULL;
1885 }
1886 
1887 /*
1888  * Fetch integer attribute of BTF map definition. Such attributes are
1889  * represented using a pointer to an array, in which dimensionality of array
1890  * encodes specified integer value. E.g., int (*type)[BPF_MAP_TYPE_ARRAY];
1891  * encodes `type => BPF_MAP_TYPE_ARRAY` key/value pair completely using BTF
1892  * type definition, while using only sizeof(void *) space in ELF data section.
1893  */
1894 static bool get_map_field_int(const char *map_name, const struct btf *btf,
1895 			      const struct btf_member *m, __u32 *res)
1896 {
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 	const struct btf_array *arr_info;
1900 	const struct btf_type *arr_t;
1901 
1902 	if (!btf_is_ptr(t)) {
1903 		pr_warn("map '%s': attr '%s': expected PTR, got %u.\n",
1904 			map_name, name, btf_kind(t));
1905 		return false;
1906 	}
1907 
1908 	arr_t = btf__type_by_id(btf, t->type);
1909 	if (!arr_t) {
1910 		pr_warn("map '%s': attr '%s': type [%u] not found.\n",
1911 			map_name, name, t->type);
1912 		return false;
1913 	}
1914 	if (!btf_is_array(arr_t)) {
1915 		pr_warn("map '%s': attr '%s': expected ARRAY, got %u.\n",
1916 			map_name, name, btf_kind(arr_t));
1917 		return false;
1918 	}
1919 	arr_info = btf_array(arr_t);
1920 	*res = arr_info->nelems;
1921 	return true;
1922 }
1923 
1924 static int build_map_pin_path(struct bpf_map *map, const char *path)
1925 {
1926 	char buf[PATH_MAX];
1927 	int err, len;
1928 
1929 	if (!path)
1930 		path = "/sys/fs/bpf";
1931 
1932 	len = snprintf(buf, PATH_MAX, "%s/%s", path, bpf_map__name(map));
1933 	if (len < 0)
1934 		return -EINVAL;
1935 	else if (len >= PATH_MAX)
1936 		return -ENAMETOOLONG;
1937 
1938 	err = bpf_map__set_pin_path(map, buf);
1939 	if (err)
1940 		return err;
1941 
1942 	return 0;
1943 }
1944 
1945 
1946 static int parse_btf_map_def(struct bpf_object *obj,
1947 			     struct bpf_map *map,
1948 			     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 	const struct btf_member *m;
1954 	int vlen, i;
1955 
1956 	vlen = btf_vlen(def);
1957 	m = btf_members(def);
1958 	for (i = 0; i < vlen; i++, m++) {
1959 		const char *name = btf__name_by_offset(obj->btf, m->name_off);
1960 
1961 		if (!name) {
1962 			pr_warn("map '%s': invalid field #%d.\n", map->name, i);
1963 			return -EINVAL;
1964 		}
1965 		if (strcmp(name, "type") == 0) {
1966 			if (!get_map_field_int(map->name, obj->btf, m,
1967 					       &map->def.type))
1968 				return -EINVAL;
1969 			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 				return -EINVAL;
1975 			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 				return -EINVAL;
1981 			pr_debug("map '%s': found map_flags = %u.\n",
1982 				 map->name, map->def.map_flags);
1983 		} else if (strcmp(name, "numa_node") == 0) {
1984 			if (!get_map_field_int(map->name, obj->btf, m, &map->numa_node))
1985 				return -EINVAL;
1986 			pr_debug("map '%s': found numa_node = %u.\n", map->name, map->numa_node);
1987 		} else if (strcmp(name, "key_size") == 0) {
1988 			__u32 sz;
1989 
1990 			if (!get_map_field_int(map->name, obj->btf, m, &sz))
1991 				return -EINVAL;
1992 			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 				return -EINVAL;
1998 			}
1999 			map->def.key_size = sz;
2000 		} else if (strcmp(name, "key") == 0) {
2001 			__s64 sz;
2002 
2003 			t = btf__type_by_id(obj->btf, m->type);
2004 			if (!t) {
2005 				pr_warn("map '%s': key type [%d] not found.\n",
2006 					map->name, m->type);
2007 				return -EINVAL;
2008 			}
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 				return -EINVAL;
2013 			}
2014 			sz = btf__resolve_size(obj->btf, t->type);
2015 			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 				return sz;
2019 			}
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 				return -EINVAL;
2026 			}
2027 			map->def.key_size = sz;
2028 			map->btf_key_type_id = t->type;
2029 		} else if (strcmp(name, "value_size") == 0) {
2030 			__u32 sz;
2031 
2032 			if (!get_map_field_int(map->name, obj->btf, m, &sz))
2033 				return -EINVAL;
2034 			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 				return -EINVAL;
2040 			}
2041 			map->def.value_size = sz;
2042 		} else if (strcmp(name, "value") == 0) {
2043 			__s64 sz;
2044 
2045 			t = btf__type_by_id(obj->btf, m->type);
2046 			if (!t) {
2047 				pr_warn("map '%s': value type [%d] not found.\n",
2048 					map->name, m->type);
2049 				return -EINVAL;
2050 			}
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 				return -EINVAL;
2055 			}
2056 			sz = btf__resolve_size(obj->btf, t->type);
2057 			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 				return sz;
2061 			}
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 				return -EINVAL;
2068 			}
2069 			map->def.value_size = sz;
2070 			map->btf_value_type_id = t->type;
2071 		}
2072 		else if (strcmp(name, "values") == 0) {
2073 			int err;
2074 
2075 			if (is_inner) {
2076 				pr_warn("map '%s': multi-level inner maps not supported.\n",
2077 					map->name);
2078 				return -ENOTSUP;
2079 			}
2080 			if (i != vlen - 1) {
2081 				pr_warn("map '%s': '%s' member should be last.\n",
2082 					map->name, name);
2083 				return -EINVAL;
2084 			}
2085 			if (!bpf_map_type__is_map_in_map(map->def.type)) {
2086 				pr_warn("map '%s': should be map-in-map.\n",
2087 					map->name);
2088 				return -ENOTSUP;
2089 			}
2090 			if (map->def.value_size && map->def.value_size != 4) {
2091 				pr_warn("map '%s': conflicting value size %u != 4.\n",
2092 					map->name, map->def.value_size);
2093 				return -EINVAL;
2094 			}
2095 			map->def.value_size = 4;
2096 			t = btf__type_by_id(obj->btf, m->type);
2097 			if (!t) {
2098 				pr_warn("map '%s': map-in-map inner type [%d] not found.\n",
2099 					map->name, m->type);
2100 				return -EINVAL;
2101 			}
2102 			if (!btf_is_array(t) || btf_array(t)->nelems) {
2103 				pr_warn("map '%s': map-in-map inner spec is not a zero-sized array.\n",
2104 					map->name);
2105 				return -EINVAL;
2106 			}
2107 			t = skip_mods_and_typedefs(obj->btf, btf_array(t)->type,
2108 						   NULL);
2109 			if (!btf_is_ptr(t)) {
2110 				pr_warn("map '%s': map-in-map inner def is of unexpected kind %u.\n",
2111 					map->name, btf_kind(t));
2112 				return -EINVAL;
2113 			}
2114 			t = skip_mods_and_typedefs(obj->btf, t->type, NULL);
2115 			if (!btf_is_struct(t)) {
2116 				pr_warn("map '%s': map-in-map inner def is of unexpected kind %u.\n",
2117 					map->name, btf_kind(t));
2118 				return -EINVAL;
2119 			}
2120 
2121 			map->inner_map = calloc(1, sizeof(*map->inner_map));
2122 			if (!map->inner_map)
2123 				return -ENOMEM;
2124 			map->inner_map->sec_idx = obj->efile.btf_maps_shndx;
2125 			map->inner_map->name = malloc(strlen(map->name) +
2126 						      sizeof(".inner") + 1);
2127 			if (!map->inner_map->name)
2128 				return -ENOMEM;
2129 			sprintf(map->inner_map->name, "%s.inner", map->name);
2130 
2131 			err = parse_btf_map_def(obj, map->inner_map, t, strict,
2132 						true /* is_inner */, NULL);
2133 			if (err)
2134 				return err;
2135 		} else if (strcmp(name, "pinning") == 0) {
2136 			__u32 val;
2137 			int err;
2138 
2139 			if (is_inner) {
2140 				pr_debug("map '%s': inner def can't be pinned.\n",
2141 					 map->name);
2142 				return -EINVAL;
2143 			}
2144 			if (!get_map_field_int(map->name, obj->btf, m, &val))
2145 				return -EINVAL;
2146 			pr_debug("map '%s': found pinning = %u.\n",
2147 				 map->name, val);
2148 
2149 			if (val != LIBBPF_PIN_NONE &&
2150 			    val != LIBBPF_PIN_BY_NAME) {
2151 				pr_warn("map '%s': invalid pinning value %u.\n",
2152 					map->name, val);
2153 				return -EINVAL;
2154 			}
2155 			if (val == LIBBPF_PIN_BY_NAME) {
2156 				err = build_map_pin_path(map, pin_root_path);
2157 				if (err) {
2158 					pr_warn("map '%s': couldn't build pin path.\n",
2159 						map->name);
2160 					return err;
2161 				}
2162 			}
2163 		} else {
2164 			if (strict) {
2165 				pr_warn("map '%s': unknown field '%s'.\n",
2166 					map->name, name);
2167 				return -ENOTSUP;
2168 			}
2169 			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 		return -EINVAL;
2177 	}
2178 
2179 	return 0;
2180 }
2181 
2182 static int bpf_object__init_user_btf_map(struct bpf_object *obj,
2183 					 const struct btf_type *sec,
2184 					 int var_idx, int sec_idx,
2185 					 const Elf_Data *data, bool strict,
2186 					 const char *pin_root_path)
2187 {
2188 	const struct btf_type *var, *def;
2189 	const struct btf_var_secinfo *vi;
2190 	const struct btf_var *var_extra;
2191 	const char *map_name;
2192 	struct bpf_map *map;
2193 
2194 	vi = btf_var_secinfos(sec) + var_idx;
2195 	var = btf__type_by_id(obj->btf, vi->type);
2196 	var_extra = btf_var(var);
2197 	map_name = btf__name_by_offset(obj->btf, var->name_off);
2198 
2199 	if (map_name == NULL || map_name[0] == '\0') {
2200 		pr_warn("map #%d: empty name.\n", var_idx);
2201 		return -EINVAL;
2202 	}
2203 	if ((__u64)vi->offset + vi->size > data->d_size) {
2204 		pr_warn("map '%s' BTF data is corrupted.\n", map_name);
2205 		return -EINVAL;
2206 	}
2207 	if (!btf_is_var(var)) {
2208 		pr_warn("map '%s': unexpected var kind %u.\n",
2209 			map_name, btf_kind(var));
2210 		return -EINVAL;
2211 	}
2212 	if (var_extra->linkage != BTF_VAR_GLOBAL_ALLOCATED &&
2213 	    var_extra->linkage != BTF_VAR_STATIC) {
2214 		pr_warn("map '%s': unsupported var linkage %u.\n",
2215 			map_name, var_extra->linkage);
2216 		return -EOPNOTSUPP;
2217 	}
2218 
2219 	def = skip_mods_and_typedefs(obj->btf, var->type, NULL);
2220 	if (!btf_is_struct(def)) {
2221 		pr_warn("map '%s': unexpected def kind %u.\n",
2222 			map_name, btf_kind(var));
2223 		return -EINVAL;
2224 	}
2225 	if (def->size > vi->size) {
2226 		pr_warn("map '%s': invalid def size.\n", map_name);
2227 		return -EINVAL;
2228 	}
2229 
2230 	map = bpf_object__add_map(obj);
2231 	if (IS_ERR(map))
2232 		return PTR_ERR(map);
2233 	map->name = strdup(map_name);
2234 	if (!map->name) {
2235 		pr_warn("map '%s': failed to alloc map name.\n", map_name);
2236 		return -ENOMEM;
2237 	}
2238 	map->libbpf_type = LIBBPF_MAP_UNSPEC;
2239 	map->def.type = BPF_MAP_TYPE_UNSPEC;
2240 	map->sec_idx = sec_idx;
2241 	map->sec_offset = vi->offset;
2242 	map->btf_var_idx = var_idx;
2243 	pr_debug("map '%s': at sec_idx %d, offset %zu.\n",
2244 		 map_name, map->sec_idx, map->sec_offset);
2245 
2246 	return parse_btf_map_def(obj, map, def, strict, false, pin_root_path);
2247 }
2248 
2249 static int bpf_object__init_user_btf_maps(struct bpf_object *obj, bool strict,
2250 					  const char *pin_root_path)
2251 {
2252 	const struct btf_type *sec = NULL;
2253 	int nr_types, i, vlen, err;
2254 	const struct btf_type *t;
2255 	const char *name;
2256 	Elf_Data *data;
2257 	Elf_Scn *scn;
2258 
2259 	if (obj->efile.btf_maps_shndx < 0)
2260 		return 0;
2261 
2262 	scn = elf_getscn(obj->efile.elf, obj->efile.btf_maps_shndx);
2263 	if (scn)
2264 		data = elf_getdata(scn, NULL);
2265 	if (!scn || !data) {
2266 		pr_warn("failed to get Elf_Data from map section %d (%s)\n",
2267 			obj->efile.maps_shndx, MAPS_ELF_SEC);
2268 		return -EINVAL;
2269 	}
2270 
2271 	nr_types = btf__get_nr_types(obj->btf);
2272 	for (i = 1; i <= nr_types; i++) {
2273 		t = btf__type_by_id(obj->btf, i);
2274 		if (!btf_is_datasec(t))
2275 			continue;
2276 		name = btf__name_by_offset(obj->btf, t->name_off);
2277 		if (strcmp(name, MAPS_ELF_SEC) == 0) {
2278 			sec = t;
2279 			obj->efile.btf_maps_sec_btf_id = i;
2280 			break;
2281 		}
2282 	}
2283 
2284 	if (!sec) {
2285 		pr_warn("DATASEC '%s' not found.\n", MAPS_ELF_SEC);
2286 		return -ENOENT;
2287 	}
2288 
2289 	vlen = btf_vlen(sec);
2290 	for (i = 0; i < vlen; i++) {
2291 		err = bpf_object__init_user_btf_map(obj, sec, i,
2292 						    obj->efile.btf_maps_shndx,
2293 						    data, strict,
2294 						    pin_root_path);
2295 		if (err)
2296 			return err;
2297 	}
2298 
2299 	return 0;
2300 }
2301 
2302 static int bpf_object__init_maps(struct bpf_object *obj,
2303 				 const struct bpf_object_open_opts *opts)
2304 {
2305 	const char *pin_root_path;
2306 	bool strict;
2307 	int err;
2308 
2309 	strict = !OPTS_GET(opts, relaxed_maps, false);
2310 	pin_root_path = OPTS_GET(opts, pin_root_path, NULL);
2311 
2312 	err = bpf_object__init_user_maps(obj, strict);
2313 	err = err ?: bpf_object__init_user_btf_maps(obj, strict, pin_root_path);
2314 	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 	if (err)
2318 		return err;
2319 
2320 	return 0;
2321 }
2322 
2323 static bool section_have_execinstr(struct bpf_object *obj, int idx)
2324 {
2325 	Elf_Scn *scn;
2326 	GElf_Shdr sh;
2327 
2328 	scn = elf_getscn(obj->efile.elf, idx);
2329 	if (!scn)
2330 		return false;
2331 
2332 	if (gelf_getshdr(scn, &sh) != &sh)
2333 		return false;
2334 
2335 	if (sh.sh_flags & SHF_EXECINSTR)
2336 		return true;
2337 
2338 	return false;
2339 }
2340 
2341 static bool btf_needs_sanitization(struct bpf_object *obj)
2342 {
2343 	bool has_func_global = obj->caps.btf_func_global;
2344 	bool has_datasec = obj->caps.btf_datasec;
2345 	bool has_func = obj->caps.btf_func;
2346 
2347 	return !has_func || !has_datasec || !has_func_global;
2348 }
2349 
2350 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 	bool has_datasec = obj->caps.btf_datasec;
2354 	bool has_func = obj->caps.btf_func;
2355 	struct btf_type *t;
2356 	int i, j, vlen;
2357 
2358 	for (i = 1; i <= btf__get_nr_types(btf); i++) {
2359 		t = (struct btf_type *)btf__type_by_id(btf, i);
2360 
2361 		if (!has_datasec && btf_is_var(t)) {
2362 			/* replace VAR with INT */
2363 			t->info = BTF_INFO_ENC(BTF_KIND_INT, 0, 0);
2364 			/*
2365 			 * using size = 1 is the safest choice, 4 will be too
2366 			 * big and cause kernel BTF validation failure if
2367 			 * original variable took less than 4 bytes
2368 			 */
2369 			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 			const struct btf_var_secinfo *v = btf_var_secinfos(t);
2374 			struct btf_member *m = btf_members(t);
2375 			struct btf_type *vt;
2376 			char *name;
2377 
2378 			name = (char *)btf__name_by_offset(btf, t->name_off);
2379 			while (*name) {
2380 				if (*name == '.')
2381 					*name = '_';
2382 				name++;
2383 			}
2384 
2385 			vlen = btf_vlen(t);
2386 			t->info = BTF_INFO_ENC(BTF_KIND_STRUCT, 0, vlen);
2387 			for (j = 0; j < vlen; j++, v++, m++) {
2388 				/* order of field assignments is important */
2389 				m->offset = v->offset * 8;
2390 				m->type = v->type;
2391 				/* preserve variable name as member name */
2392 				vt = (void *)btf__type_by_id(btf, v->type);
2393 				m->name_off = vt->name_off;
2394 			}
2395 		} else if (!has_func && btf_is_func_proto(t)) {
2396 			/* replace FUNC_PROTO with ENUM */
2397 			vlen = btf_vlen(t);
2398 			t->info = BTF_INFO_ENC(BTF_KIND_ENUM, 0, vlen);
2399 			t->size = sizeof(__u32); /* kernel enforced */
2400 		} else if (!has_func && btf_is_func(t)) {
2401 			/* replace FUNC with TYPEDEF */
2402 			t->info = BTF_INFO_ENC(BTF_KIND_TYPEDEF, 0, 0);
2403 		} else if (!has_func_global && btf_is_func(t)) {
2404 			/* replace BTF_FUNC_GLOBAL with BTF_FUNC_STATIC */
2405 			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 	return obj->efile.btf_maps_shndx >= 0 ||
2413 	       obj->efile.st_ops_shndx >= 0 ||
2414 	       obj->nr_extern > 0;
2415 }
2416 
2417 static bool kernel_needs_btf(const struct bpf_object *obj)
2418 {
2419 	return obj->efile.st_ops_shndx >= 0;
2420 }
2421 
2422 static int bpf_object__init_btf(struct bpf_object *obj,
2423 				Elf_Data *btf_data,
2424 				Elf_Data *btf_ext_data)
2425 {
2426 	int err = -ENOENT;
2427 
2428 	if (btf_data) {
2429 		obj->btf = btf__new(btf_data->d_buf, btf_data->d_size);
2430 		if (IS_ERR(obj->btf)) {
2431 			err = PTR_ERR(obj->btf);
2432 			obj->btf = NULL;
2433 			pr_warn("Error loading ELF section %s: %d.\n",
2434 				BTF_ELF_SEC, err);
2435 			goto out;
2436 		}
2437 		err = 0;
2438 	}
2439 	if (btf_ext_data) {
2440 		if (!obj->btf) {
2441 			pr_debug("Ignore ELF section %s because its depending ELF section %s is not found.\n",
2442 				 BTF_EXT_ELF_SEC, BTF_ELF_SEC);
2443 			goto out;
2444 		}
2445 		obj->btf_ext = btf_ext__new(btf_ext_data->d_buf,
2446 					    btf_ext_data->d_size);
2447 		if (IS_ERR(obj->btf_ext)) {
2448 			pr_warn("Error loading ELF section %s: %ld. Ignored and continue.\n",
2449 				BTF_EXT_ELF_SEC, PTR_ERR(obj->btf_ext));
2450 			obj->btf_ext = NULL;
2451 			goto out;
2452 		}
2453 	}
2454 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 	return 0;
2460 }
2461 
2462 static int bpf_object__finalize_btf(struct bpf_object *obj)
2463 {
2464 	int err;
2465 
2466 	if (!obj->btf)
2467 		return 0;
2468 
2469 	err = btf__finalize_data(obj, obj->btf);
2470 	if (err) {
2471 		pr_warn("Error finalizing %s: %d.\n", BTF_ELF_SEC, err);
2472 		return err;
2473 	}
2474 
2475 	return 0;
2476 }
2477 
2478 static inline bool libbpf_prog_needs_vmlinux_btf(struct bpf_program *prog)
2479 {
2480 	if (prog->type == BPF_PROG_TYPE_STRUCT_OPS ||
2481 	    prog->type == BPF_PROG_TYPE_LSM)
2482 		return true;
2483 
2484 	/* BPF_PROG_TYPE_TRACING programs which do not attach to other programs
2485 	 * also need vmlinux BTF
2486 	 */
2487 	if (prog->type == BPF_PROG_TYPE_TRACING && !prog->attach_prog_fd)
2488 		return true;
2489 
2490 	return false;
2491 }
2492 
2493 static int bpf_object__load_vmlinux_btf(struct bpf_object *obj)
2494 {
2495 	bool need_vmlinux_btf = false;
2496 	struct bpf_program *prog;
2497 	int err;
2498 
2499 	/* CO-RE relocations need kernel BTF */
2500 	if (obj->btf_ext && obj->btf_ext->field_reloc_info.len)
2501 		need_vmlinux_btf = true;
2502 
2503 	bpf_object__for_each_program(prog, obj) {
2504 		if (!prog->load)
2505 			continue;
2506 		if (libbpf_prog_needs_vmlinux_btf(prog)) {
2507 			need_vmlinux_btf = true;
2508 			break;
2509 		}
2510 	}
2511 
2512 	if (!need_vmlinux_btf)
2513 		return 0;
2514 
2515 	obj->btf_vmlinux = libbpf_find_kernel_btf();
2516 	if (IS_ERR(obj->btf_vmlinux)) {
2517 		err = PTR_ERR(obj->btf_vmlinux);
2518 		pr_warn("Error loading vmlinux BTF: %d\n", err);
2519 		obj->btf_vmlinux = NULL;
2520 		return err;
2521 	}
2522 	return 0;
2523 }
2524 
2525 static int bpf_object__sanitize_and_load_btf(struct bpf_object *obj)
2526 {
2527 	struct btf *kern_btf = obj->btf;
2528 	bool btf_mandatory, sanitize;
2529 	int err = 0;
2530 
2531 	if (!obj->btf)
2532 		return 0;
2533 
2534 	sanitize = btf_needs_sanitization(obj);
2535 	if (sanitize) {
2536 		const void *raw_data;
2537 		__u32 sz;
2538 
2539 		/* clone BTF to sanitize a copy and leave the original intact */
2540 		raw_data = btf__get_raw_data(obj->btf, &sz);
2541 		kern_btf = btf__new(raw_data, sz);
2542 		if (IS_ERR(kern_btf))
2543 			return PTR_ERR(kern_btf);
2544 
2545 		bpf_object__sanitize_btf(obj, kern_btf);
2546 	}
2547 
2548 	err = btf__load(kern_btf);
2549 	if (sanitize) {
2550 		if (!err) {
2551 			/* move fd to libbpf's BTF */
2552 			btf__set_fd(obj->btf, btf__fd(kern_btf));
2553 			btf__set_fd(kern_btf, -1);
2554 		}
2555 		btf__free(kern_btf);
2556 	}
2557 	if (err) {
2558 		btf_mandatory = kernel_needs_btf(obj);
2559 		pr_warn("Error loading .BTF into kernel: %d. %s\n", err,
2560 			btf_mandatory ? "BTF is mandatory, can't proceed."
2561 				      : "BTF is optional, ignoring.");
2562 		if (!btf_mandatory)
2563 			err = 0;
2564 	}
2565 	return err;
2566 }
2567 
2568 static int bpf_object__elf_collect(struct bpf_object *obj)
2569 {
2570 	Elf *elf = obj->efile.elf;
2571 	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 	/* Elf is corrupted/truncated, avoid calling elf_strptr. */
2578 	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 	while ((scn = elf_nextscn(elf, scn)) != NULL) {
2584 		char *name;
2585 		GElf_Shdr sh;
2586 		Elf_Data *data;
2587 
2588 		idx++;
2589 		if (gelf_getshdr(scn, &sh) != &sh) {
2590 			pr_warn("failed to get section(%d) header from %s\n",
2591 				idx, obj->path);
2592 			return -LIBBPF_ERRNO__FORMAT;
2593 		}
2594 
2595 		name = elf_strptr(elf, ep->e_shstrndx, sh.sh_name);
2596 		if (!name) {
2597 			pr_warn("failed to get section(%d) name from %s\n",
2598 				idx, obj->path);
2599 			return -LIBBPF_ERRNO__FORMAT;
2600 		}
2601 
2602 		data = elf_getdata(scn, 0);
2603 		if (!data) {
2604 			pr_warn("failed to get section(%d) data from %s(%s)\n",
2605 				idx, name, obj->path);
2606 			return -LIBBPF_ERRNO__FORMAT;
2607 		}
2608 		pr_debug("section(%d) %s, size %ld, link %d, flags %lx, type=%d\n",
2609 			 idx, name, (unsigned long)data->d_size,
2610 			 (int)sh.sh_link, (unsigned long)sh.sh_flags,
2611 			 (int)sh.sh_type);
2612 
2613 		if (strcmp(name, "license") == 0) {
2614 			err = bpf_object__init_license(obj,
2615 						       data->d_buf,
2616 						       data->d_size);
2617 			if (err)
2618 				return err;
2619 		} else if (strcmp(name, "version") == 0) {
2620 			err = bpf_object__init_kversion(obj,
2621 							data->d_buf,
2622 							data->d_size);
2623 			if (err)
2624 				return err;
2625 		} else if (strcmp(name, "maps") == 0) {
2626 			obj->efile.maps_shndx = idx;
2627 		} else if (strcmp(name, MAPS_ELF_SEC) == 0) {
2628 			obj->efile.btf_maps_shndx = idx;
2629 		} else if (strcmp(name, BTF_ELF_SEC) == 0) {
2630 			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 				pr_warn("bpf: multiple SYMTAB in %s\n",
2636 					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 		} else if (sh.sh_type == SHT_PROGBITS && data->d_size > 0) {
2643 			if (sh.sh_flags & SHF_EXECINSTR) {
2644 				if (strcmp(name, ".text") == 0)
2645 					obj->efile.text_shndx = idx;
2646 				err = bpf_object__add_program(obj, data->d_buf,
2647 							      data->d_size,
2648 							      name, idx);
2649 				if (err) {
2650 					char errmsg[STRERR_BUFSIZE];
2651 					char *cp;
2652 
2653 					cp = libbpf_strerror_r(-err, errmsg,
2654 							       sizeof(errmsg));
2655 					pr_warn("failed to alloc program %s (%s): %s",
2656 						name, obj->path, cp);
2657 					return err;
2658 				}
2659 			} else if (strcmp(name, DATA_SEC) == 0) {
2660 				obj->efile.data = data;
2661 				obj->efile.data_shndx = idx;
2662 			} else if (strcmp(name, RODATA_SEC) == 0) {
2663 				obj->efile.rodata = data;
2664 				obj->efile.rodata_shndx = idx;
2665 			} else if (strcmp(name, STRUCT_OPS_SEC) == 0) {
2666 				obj->efile.st_ops_data = data;
2667 				obj->efile.st_ops_shndx = idx;
2668 			} else {
2669 				pr_debug("skip section(%d) %s\n", idx, name);
2670 			}
2671 		} else if (sh.sh_type == SHT_REL) {
2672 			int nr_sects = obj->efile.nr_reloc_sects;
2673 			void *sects = obj->efile.reloc_sects;
2674 			int sec = sh.sh_info; /* points to other section */
2675 
2676 			/* Only do relo for section with exec instructions */
2677 			if (!section_have_execinstr(obj, sec) &&
2678 			    strcmp(name, ".rel" STRUCT_OPS_SEC) &&
2679 			    strcmp(name, ".rel" MAPS_ELF_SEC)) {
2680 				pr_debug("skip relo %s(%d) for section(%d)\n",
2681 					 name, idx, sec);
2682 				continue;
2683 			}
2684 
2685 			sects = reallocarray(sects, nr_sects + 1,
2686 					     sizeof(*obj->efile.reloc_sects));
2687 			if (!sects) {
2688 				pr_warn("reloc_sects realloc failed\n");
2689 				return -ENOMEM;
2690 			}
2691 
2692 			obj->efile.reloc_sects = sects;
2693 			obj->efile.nr_reloc_sects++;
2694 
2695 			obj->efile.reloc_sects[nr_sects].shdr = sh;
2696 			obj->efile.reloc_sects[nr_sects].data = data;
2697 		} else if (sh.sh_type == SHT_NOBITS &&
2698 			   strcmp(name, BSS_SEC) == 0) {
2699 			obj->efile.bss = data;
2700 			obj->efile.bss_shndx = idx;
2701 		} else {
2702 			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 static bool sym_is_extern(const GElf_Sym *sym)
2714 {
2715 	int bind = GELF_ST_BIND(sym->st_info);
2716 	/* externs are symbols w/ type=NOTYPE, bind=GLOBAL|WEAK, section=UND */
2717 	return sym->st_shndx == SHN_UNDEF &&
2718 	       (bind == STB_GLOBAL || bind == STB_WEAK) &&
2719 	       GELF_ST_TYPE(sym->st_info) == STT_NOTYPE;
2720 }
2721 
2722 static int find_extern_btf_id(const struct btf *btf, const char *ext_name)
2723 {
2724 	const struct btf_type *t;
2725 	const char *var_name;
2726 	int i, n;
2727 
2728 	if (!btf)
2729 		return -ESRCH;
2730 
2731 	n = btf__get_nr_types(btf);
2732 	for (i = 1; i <= n; i++) {
2733 		t = btf__type_by_id(btf, i);
2734 
2735 		if (!btf_is_var(t))
2736 			continue;
2737 
2738 		var_name = btf__name_by_offset(btf, t->name_off);
2739 		if (strcmp(var_name, ext_name))
2740 			continue;
2741 
2742 		if (btf_var(t)->linkage != BTF_VAR_GLOBAL_EXTERN)
2743 			return -EINVAL;
2744 
2745 		return i;
2746 	}
2747 
2748 	return -ENOENT;
2749 }
2750 
2751 static int find_extern_sec_btf_id(struct btf *btf, int ext_btf_id) {
2752 	const struct btf_var_secinfo *vs;
2753 	const struct btf_type *t;
2754 	int i, j, n;
2755 
2756 	if (!btf)
2757 		return -ESRCH;
2758 
2759 	n = btf__get_nr_types(btf);
2760 	for (i = 1; i <= n; i++) {
2761 		t = btf__type_by_id(btf, i);
2762 
2763 		if (!btf_is_datasec(t))
2764 			continue;
2765 
2766 		vs = btf_var_secinfos(t);
2767 		for (j = 0; j < btf_vlen(t); j++, vs++) {
2768 			if (vs->type == ext_btf_id)
2769 				return i;
2770 		}
2771 	}
2772 
2773 	return -ENOENT;
2774 }
2775 
2776 static enum kcfg_type find_kcfg_type(const struct btf *btf, int id,
2777 				     bool *is_signed)
2778 {
2779 	const struct btf_type *t;
2780 	const char *name;
2781 
2782 	t = skip_mods_and_typedefs(btf, id, NULL);
2783 	name = btf__name_by_offset(btf, t->name_off);
2784 
2785 	if (is_signed)
2786 		*is_signed = false;
2787 	switch (btf_kind(t)) {
2788 	case BTF_KIND_INT: {
2789 		int enc = btf_int_encoding(t);
2790 
2791 		if (enc & BTF_INT_BOOL)
2792 			return t->size == 1 ? KCFG_BOOL : KCFG_UNKNOWN;
2793 		if (is_signed)
2794 			*is_signed = enc & BTF_INT_SIGNED;
2795 		if (t->size == 1)
2796 			return KCFG_CHAR;
2797 		if (t->size < 1 || t->size > 8 || (t->size & (t->size - 1)))
2798 			return KCFG_UNKNOWN;
2799 		return KCFG_INT;
2800 	}
2801 	case BTF_KIND_ENUM:
2802 		if (t->size != 4)
2803 			return KCFG_UNKNOWN;
2804 		if (strcmp(name, "libbpf_tristate"))
2805 			return KCFG_UNKNOWN;
2806 		return KCFG_TRISTATE;
2807 	case BTF_KIND_ARRAY:
2808 		if (btf_array(t)->nelems == 0)
2809 			return KCFG_UNKNOWN;
2810 		if (find_kcfg_type(btf, btf_array(t)->type, NULL) != KCFG_CHAR)
2811 			return KCFG_UNKNOWN;
2812 		return KCFG_CHAR_ARR;
2813 	default:
2814 		return KCFG_UNKNOWN;
2815 	}
2816 }
2817 
2818 static int cmp_externs(const void *_a, const void *_b)
2819 {
2820 	const struct extern_desc *a = _a;
2821 	const struct extern_desc *b = _b;
2822 
2823 	if (a->type != b->type)
2824 		return a->type < b->type ? -1 : 1;
2825 
2826 	if (a->type == EXT_KCFG) {
2827 		/* descending order by alignment requirements */
2828 		if (a->kcfg.align != b->kcfg.align)
2829 			return a->kcfg.align > b->kcfg.align ? -1 : 1;
2830 		/* ascending order by size, within same alignment class */
2831 		if (a->kcfg.sz != b->kcfg.sz)
2832 			return a->kcfg.sz < b->kcfg.sz ? -1 : 1;
2833 	}
2834 
2835 	/* resolve ties by name */
2836 	return strcmp(a->name, b->name);
2837 }
2838 
2839 static int find_int_btf_id(const struct btf *btf)
2840 {
2841 	const struct btf_type *t;
2842 	int i, n;
2843 
2844 	n = btf__get_nr_types(btf);
2845 	for (i = 1; i <= n; i++) {
2846 		t = btf__type_by_id(btf, i);
2847 
2848 		if (btf_is_int(t) && btf_int_bits(t) == 32)
2849 			return i;
2850 	}
2851 
2852 	return 0;
2853 }
2854 
2855 static int bpf_object__collect_externs(struct bpf_object *obj)
2856 {
2857 	struct btf_type *sec, *kcfg_sec = NULL, *ksym_sec = NULL;
2858 	const struct btf_type *t;
2859 	struct extern_desc *ext;
2860 	int i, n, off;
2861 	const char *ext_name, *sec_name;
2862 	Elf_Scn *scn;
2863 	GElf_Shdr sh;
2864 
2865 	if (!obj->efile.symbols)
2866 		return 0;
2867 
2868 	scn = elf_getscn(obj->efile.elf, obj->efile.symbols_shndx);
2869 	if (!scn)
2870 		return -LIBBPF_ERRNO__FORMAT;
2871 	if (gelf_getshdr(scn, &sh) != &sh)
2872 		return -LIBBPF_ERRNO__FORMAT;
2873 	n = sh.sh_size / sh.sh_entsize;
2874 
2875 	pr_debug("looking for externs among %d symbols...\n", n);
2876 	for (i = 0; i < n; i++) {
2877 		GElf_Sym sym;
2878 
2879 		if (!gelf_getsym(obj->efile.symbols, i, &sym))
2880 			return -LIBBPF_ERRNO__FORMAT;
2881 		if (!sym_is_extern(&sym))
2882 			continue;
2883 		ext_name = elf_strptr(obj->efile.elf, obj->efile.strtabidx,
2884 				      sym.st_name);
2885 		if (!ext_name || !ext_name[0])
2886 			continue;
2887 
2888 		ext = obj->externs;
2889 		ext = reallocarray(ext, obj->nr_extern + 1, sizeof(*ext));
2890 		if (!ext)
2891 			return -ENOMEM;
2892 		obj->externs = ext;
2893 		ext = &ext[obj->nr_extern];
2894 		memset(ext, 0, sizeof(*ext));
2895 		obj->nr_extern++;
2896 
2897 		ext->btf_id = find_extern_btf_id(obj->btf, ext_name);
2898 		if (ext->btf_id <= 0) {
2899 			pr_warn("failed to find BTF for extern '%s': %d\n",
2900 				ext_name, ext->btf_id);
2901 			return ext->btf_id;
2902 		}
2903 		t = btf__type_by_id(obj->btf, ext->btf_id);
2904 		ext->name = btf__name_by_offset(obj->btf, t->name_off);
2905 		ext->sym_idx = i;
2906 		ext->is_weak = GELF_ST_BIND(sym.st_info) == STB_WEAK;
2907 
2908 		ext->sec_btf_id = find_extern_sec_btf_id(obj->btf, ext->btf_id);
2909 		if (ext->sec_btf_id <= 0) {
2910 			pr_warn("failed to find BTF for extern '%s' [%d] section: %d\n",
2911 				ext_name, ext->btf_id, ext->sec_btf_id);
2912 			return ext->sec_btf_id;
2913 		}
2914 		sec = (void *)btf__type_by_id(obj->btf, ext->sec_btf_id);
2915 		sec_name = btf__name_by_offset(obj->btf, sec->name_off);
2916 
2917 		if (strcmp(sec_name, KCONFIG_SEC) == 0) {
2918 			kcfg_sec = sec;
2919 			ext->type = EXT_KCFG;
2920 			ext->kcfg.sz = btf__resolve_size(obj->btf, t->type);
2921 			if (ext->kcfg.sz <= 0) {
2922 				pr_warn("failed to resolve size of extern (kcfg) '%s': %d\n",
2923 					ext_name, ext->kcfg.sz);
2924 				return ext->kcfg.sz;
2925 			}
2926 			ext->kcfg.align = btf__align_of(obj->btf, t->type);
2927 			if (ext->kcfg.align <= 0) {
2928 				pr_warn("failed to determine alignment of extern (kcfg) '%s': %d\n",
2929 					ext_name, ext->kcfg.align);
2930 				return -EINVAL;
2931 			}
2932 			ext->kcfg.type = find_kcfg_type(obj->btf, t->type,
2933 						        &ext->kcfg.is_signed);
2934 			if (ext->kcfg.type == KCFG_UNKNOWN) {
2935 				pr_warn("extern (kcfg) '%s' type is unsupported\n", ext_name);
2936 				return -ENOTSUP;
2937 			}
2938 		} else if (strcmp(sec_name, KSYMS_SEC) == 0) {
2939 			const struct btf_type *vt;
2940 
2941 			ksym_sec = sec;
2942 			ext->type = EXT_KSYM;
2943 
2944 			vt = skip_mods_and_typedefs(obj->btf, t->type, NULL);
2945 			if (!btf_is_void(vt)) {
2946 				pr_warn("extern (ksym) '%s' is not typeless (void)\n", ext_name);
2947 				return -ENOTSUP;
2948 			}
2949 		} else {
2950 			pr_warn("unrecognized extern section '%s'\n", sec_name);
2951 			return -ENOTSUP;
2952 		}
2953 	}
2954 	pr_debug("collected %d externs total\n", obj->nr_extern);
2955 
2956 	if (!obj->nr_extern)
2957 		return 0;
2958 
2959 	/* sort externs by type, for kcfg ones also by (align, size, name) */
2960 	qsort(obj->externs, obj->nr_extern, sizeof(*ext), cmp_externs);
2961 
2962 	/* for .ksyms section, we need to turn all externs into allocated
2963 	 * variables in BTF to pass kernel verification; we do this by
2964 	 * pretending that each extern is a 8-byte variable
2965 	 */
2966 	if (ksym_sec) {
2967 		/* find existing 4-byte integer type in BTF to use for fake
2968 		 * extern variables in DATASEC
2969 		 */
2970 		int int_btf_id = find_int_btf_id(obj->btf);
2971 
2972 		for (i = 0; i < obj->nr_extern; i++) {
2973 			ext = &obj->externs[i];
2974 			if (ext->type != EXT_KSYM)
2975 				continue;
2976 			pr_debug("extern (ksym) #%d: symbol %d, name %s\n",
2977 				 i, ext->sym_idx, ext->name);
2978 		}
2979 
2980 		sec = ksym_sec;
2981 		n = btf_vlen(sec);
2982 		for (i = 0, off = 0; i < n; i++, off += sizeof(int)) {
2983 			struct btf_var_secinfo *vs = btf_var_secinfos(sec) + i;
2984 			struct btf_type *vt;
2985 
2986 			vt = (void *)btf__type_by_id(obj->btf, vs->type);
2987 			ext_name = btf__name_by_offset(obj->btf, vt->name_off);
2988 			ext = find_extern_by_name(obj, ext_name);
2989 			if (!ext) {
2990 				pr_warn("failed to find extern definition for BTF var '%s'\n",
2991 					ext_name);
2992 				return -ESRCH;
2993 			}
2994 			btf_var(vt)->linkage = BTF_VAR_GLOBAL_ALLOCATED;
2995 			vt->type = int_btf_id;
2996 			vs->offset = off;
2997 			vs->size = sizeof(int);
2998 		}
2999 		sec->size = off;
3000 	}
3001 
3002 	if (kcfg_sec) {
3003 		sec = kcfg_sec;
3004 		/* for kcfg externs calculate their offsets within a .kconfig map */
3005 		off = 0;
3006 		for (i = 0; i < obj->nr_extern; i++) {
3007 			ext = &obj->externs[i];
3008 			if (ext->type != EXT_KCFG)
3009 				continue;
3010 
3011 			ext->kcfg.data_off = roundup(off, ext->kcfg.align);
3012 			off = ext->kcfg.data_off + ext->kcfg.sz;
3013 			pr_debug("extern (kcfg) #%d: symbol %d, off %u, name %s\n",
3014 				 i, ext->sym_idx, ext->kcfg.data_off, ext->name);
3015 		}
3016 		sec->size = off;
3017 		n = btf_vlen(sec);
3018 		for (i = 0; i < n; i++) {
3019 			struct btf_var_secinfo *vs = btf_var_secinfos(sec) + i;
3020 
3021 			t = btf__type_by_id(obj->btf, vs->type);
3022 			ext_name = btf__name_by_offset(obj->btf, t->name_off);
3023 			ext = find_extern_by_name(obj, ext_name);
3024 			if (!ext) {
3025 				pr_warn("failed to find extern definition for BTF var '%s'\n",
3026 					ext_name);
3027 				return -ESRCH;
3028 			}
3029 			btf_var(t)->linkage = BTF_VAR_GLOBAL_ALLOCATED;
3030 			vs->offset = ext->kcfg.data_off;
3031 		}
3032 	}
3033 	return 0;
3034 }
3035 
3036 static struct bpf_program *
3037 bpf_object__find_prog_by_idx(struct bpf_object *obj, int idx)
3038 {
3039 	struct bpf_program *prog;
3040 	size_t i;
3041 
3042 	for (i = 0; i < obj->nr_programs; i++) {
3043 		prog = &obj->programs[i];
3044 		if (prog->idx == idx)
3045 			return prog;
3046 	}
3047 	return NULL;
3048 }
3049 
3050 struct bpf_program *
3051 bpf_object__find_program_by_title(const struct bpf_object *obj,
3052 				  const char *title)
3053 {
3054 	struct bpf_program *pos;
3055 
3056 	bpf_object__for_each_program(pos, obj) {
3057 		if (pos->section_name && !strcmp(pos->section_name, title))
3058 			return pos;
3059 	}
3060 	return NULL;
3061 }
3062 
3063 struct bpf_program *
3064 bpf_object__find_program_by_name(const struct bpf_object *obj,
3065 				 const char *name)
3066 {
3067 	struct bpf_program *prog;
3068 
3069 	bpf_object__for_each_program(prog, obj) {
3070 		if (!strcmp(prog->name, name))
3071 			return prog;
3072 	}
3073 	return NULL;
3074 }
3075 
3076 static bool bpf_object__shndx_is_data(const struct bpf_object *obj,
3077 				      int shndx)
3078 {
3079 	return shndx == obj->efile.data_shndx ||
3080 	       shndx == obj->efile.bss_shndx ||
3081 	       shndx == obj->efile.rodata_shndx;
3082 }
3083 
3084 static bool bpf_object__shndx_is_maps(const struct bpf_object *obj,
3085 				      int shndx)
3086 {
3087 	return shndx == obj->efile.maps_shndx ||
3088 	       shndx == obj->efile.btf_maps_shndx;
3089 }
3090 
3091 static enum libbpf_map_type
3092 bpf_object__section_to_libbpf_map_type(const struct bpf_object *obj, int shndx)
3093 {
3094 	if (shndx == obj->efile.data_shndx)
3095 		return LIBBPF_MAP_DATA;
3096 	else if (shndx == obj->efile.bss_shndx)
3097 		return LIBBPF_MAP_BSS;
3098 	else if (shndx == obj->efile.rodata_shndx)
3099 		return LIBBPF_MAP_RODATA;
3100 	else if (shndx == obj->efile.symbols_shndx)
3101 		return LIBBPF_MAP_KCONFIG;
3102 	else
3103 		return LIBBPF_MAP_UNSPEC;
3104 }
3105 
3106 static int bpf_program__record_reloc(struct bpf_program *prog,
3107 				     struct reloc_desc *reloc_desc,
3108 				     __u32 insn_idx, const char *name,
3109 				     const GElf_Sym *sym, const GElf_Rel *rel)
3110 {
3111 	struct bpf_insn *insn = &prog->insns[insn_idx];
3112 	size_t map_idx, nr_maps = prog->obj->nr_maps;
3113 	struct bpf_object *obj = prog->obj;
3114 	__u32 shdr_idx = sym->st_shndx;
3115 	enum libbpf_map_type type;
3116 	struct bpf_map *map;
3117 
3118 	/* sub-program call relocation */
3119 	if (insn->code == (BPF_JMP | BPF_CALL)) {
3120 		if (insn->src_reg != BPF_PSEUDO_CALL) {
3121 			pr_warn("incorrect bpf_call opcode\n");
3122 			return -LIBBPF_ERRNO__RELOC;
3123 		}
3124 		/* text_shndx can be 0, if no default "main" program exists */
3125 		if (!shdr_idx || shdr_idx != obj->efile.text_shndx) {
3126 			pr_warn("bad call relo against section %u\n", shdr_idx);
3127 			return -LIBBPF_ERRNO__RELOC;
3128 		}
3129 		if (sym->st_value % 8) {
3130 			pr_warn("bad call relo offset: %zu\n",
3131 				(size_t)sym->st_value);
3132 			return -LIBBPF_ERRNO__RELOC;
3133 		}
3134 		reloc_desc->type = RELO_CALL;
3135 		reloc_desc->insn_idx = insn_idx;
3136 		reloc_desc->sym_off = sym->st_value;
3137 		obj->has_pseudo_calls = true;
3138 		return 0;
3139 	}
3140 
3141 	if (insn->code != (BPF_LD | BPF_IMM | BPF_DW)) {
3142 		pr_warn("invalid relo for insns[%d].code 0x%x\n",
3143 			insn_idx, insn->code);
3144 		return -LIBBPF_ERRNO__RELOC;
3145 	}
3146 
3147 	if (sym_is_extern(sym)) {
3148 		int sym_idx = GELF_R_SYM(rel->r_info);
3149 		int i, n = obj->nr_extern;
3150 		struct extern_desc *ext;
3151 
3152 		for (i = 0; i < n; i++) {
3153 			ext = &obj->externs[i];
3154 			if (ext->sym_idx == sym_idx)
3155 				break;
3156 		}
3157 		if (i >= n) {
3158 			pr_warn("extern relo failed to find extern for sym %d\n",
3159 				sym_idx);
3160 			return -LIBBPF_ERRNO__RELOC;
3161 		}
3162 		pr_debug("found extern #%d '%s' (sym %d) for insn %u\n",
3163 			 i, ext->name, ext->sym_idx, insn_idx);
3164 		reloc_desc->type = RELO_EXTERN;
3165 		reloc_desc->insn_idx = insn_idx;
3166 		reloc_desc->sym_off = i; /* sym_off stores extern index */
3167 		return 0;
3168 	}
3169 
3170 	if (!shdr_idx || shdr_idx >= SHN_LORESERVE) {
3171 		pr_warn("invalid relo for \'%s\' in special section 0x%x; forgot to initialize global var?..\n",
3172 			name, shdr_idx);
3173 		return -LIBBPF_ERRNO__RELOC;
3174 	}
3175 
3176 	type = bpf_object__section_to_libbpf_map_type(obj, shdr_idx);
3177 
3178 	/* generic map reference relocation */
3179 	if (type == LIBBPF_MAP_UNSPEC) {
3180 		if (!bpf_object__shndx_is_maps(obj, shdr_idx)) {
3181 			pr_warn("bad map relo against section %u\n",
3182 				shdr_idx);
3183 			return -LIBBPF_ERRNO__RELOC;
3184 		}
3185 		for (map_idx = 0; map_idx < nr_maps; map_idx++) {
3186 			map = &obj->maps[map_idx];
3187 			if (map->libbpf_type != type ||
3188 			    map->sec_idx != sym->st_shndx ||
3189 			    map->sec_offset != sym->st_value)
3190 				continue;
3191 			pr_debug("found map %zd (%s, sec %d, off %zu) for insn %u\n",
3192 				 map_idx, map->name, map->sec_idx,
3193 				 map->sec_offset, insn_idx);
3194 			break;
3195 		}
3196 		if (map_idx >= nr_maps) {
3197 			pr_warn("map relo failed to find map for sec %u, off %zu\n",
3198 				shdr_idx, (size_t)sym->st_value);
3199 			return -LIBBPF_ERRNO__RELOC;
3200 		}
3201 		reloc_desc->type = RELO_LD64;
3202 		reloc_desc->insn_idx = insn_idx;
3203 		reloc_desc->map_idx = map_idx;
3204 		reloc_desc->sym_off = 0; /* sym->st_value determines map_idx */
3205 		return 0;
3206 	}
3207 
3208 	/* global data map relocation */
3209 	if (!bpf_object__shndx_is_data(obj, shdr_idx)) {
3210 		pr_warn("bad data relo against section %u\n", shdr_idx);
3211 		return -LIBBPF_ERRNO__RELOC;
3212 	}
3213 	for (map_idx = 0; map_idx < nr_maps; map_idx++) {
3214 		map = &obj->maps[map_idx];
3215 		if (map->libbpf_type != type)
3216 			continue;
3217 		pr_debug("found data map %zd (%s, sec %d, off %zu) for insn %u\n",
3218 			 map_idx, map->name, map->sec_idx, map->sec_offset,
3219 			 insn_idx);
3220 		break;
3221 	}
3222 	if (map_idx >= nr_maps) {
3223 		pr_warn("data relo failed to find map for sec %u\n",
3224 			shdr_idx);
3225 		return -LIBBPF_ERRNO__RELOC;
3226 	}
3227 
3228 	reloc_desc->type = RELO_DATA;
3229 	reloc_desc->insn_idx = insn_idx;
3230 	reloc_desc->map_idx = map_idx;
3231 	reloc_desc->sym_off = sym->st_value;
3232 	return 0;
3233 }
3234 
3235 static int
3236 bpf_program__collect_reloc(struct bpf_program *prog, GElf_Shdr *shdr,
3237 			   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 	nrels = shdr->sh_size / shdr->sh_entsize;
3244 
3245 	prog->reloc_desc = malloc(sizeof(*prog->reloc_desc) * nrels);
3246 	if (!prog->reloc_desc) {
3247 		pr_warn("failed to alloc memory in relocation\n");
3248 		return -ENOMEM;
3249 	}
3250 	prog->nr_reloc = nrels;
3251 
3252 	for (i = 0; i < nrels; i++) {
3253 		const char *name;
3254 		__u32 insn_idx;
3255 		GElf_Sym sym;
3256 		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 			pr_warn("relocation: symbol %"PRIx64" not found\n",
3264 				GELF_R_SYM(rel.r_info));
3265 			return -LIBBPF_ERRNO__FORMAT;
3266 		}
3267 		if (rel.r_offset % sizeof(struct bpf_insn))
3268 			return -LIBBPF_ERRNO__FORMAT;
3269 
3270 		insn_idx = rel.r_offset / sizeof(struct bpf_insn);
3271 		name = elf_strptr(obj->efile.elf, obj->efile.strtabidx,
3272 				  sym.st_name) ? : "<?>";
3273 
3274 		pr_debug("relo for shdr %u, symb %zu, value %zu, type %d, bind %d, name %d (\'%s\'), insn %u\n",
3275 			 (__u32)sym.st_shndx, (size_t)GELF_R_SYM(rel.r_info),
3276 			 (size_t)sym.st_value, GELF_ST_TYPE(sym.st_info),
3277 			 GELF_ST_BIND(sym.st_info), sym.st_name, name,
3278 			 insn_idx);
3279 
3280 		err = bpf_program__record_reloc(prog, &prog->reloc_desc[i],
3281 						insn_idx, name, &sym, &rel);
3282 		if (err)
3283 			return err;
3284 	}
3285 	return 0;
3286 }
3287 
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 	/* if it's BTF-defined map, we don't need to search for type IDs.
3295 	 * For struct_ops map, it does not need btf_key_type_id and
3296 	 * btf_value_type_id.
3297 	 */
3298 	if (map->sec_idx == obj->efile.btf_maps_shndx ||
3299 	    bpf_map__is_struct_ops(map))
3300 		return 0;
3301 
3302 	if (!bpf_map__is_internal(map)) {
3303 		ret = btf__get_map_kv_tids(obj->btf, map->name, def->key_size,
3304 					   def->value_size, &key_type_id,
3305 					   &value_type_id);
3306 	} else {
3307 		/*
3308 		 * LLVM annotates global data differently in BTF, that is,
3309 		 * only as '.data', '.bss' or '.rodata'.
3310 		 */
3311 		ret = btf__find_by_name(obj->btf,
3312 				libbpf_type_to_btf_name[map->libbpf_type]);
3313 	}
3314 	if (ret < 0)
3315 		return ret;
3316 
3317 	map->btf_key_type_id = key_type_id;
3318 	map->btf_value_type_id = bpf_map__is_internal(map) ?
3319 				 ret : value_type_id;
3320 	return 0;
3321 }
3322 
3323 int bpf_map__reuse_fd(struct bpf_map *map, int fd)
3324 {
3325 	struct bpf_map_info info = {};
3326 	__u32 len = sizeof(info);
3327 	int new_fd, err;
3328 	char *new_name;
3329 
3330 	err = bpf_obj_get_info_by_fd(fd, &info, &len);
3331 	if (err)
3332 		return err;
3333 
3334 	new_name = strdup(info.name);
3335 	if (!new_name)
3336 		return -errno;
3337 
3338 	new_fd = open("/", O_RDONLY | O_CLOEXEC);
3339 	if (new_fd < 0) {
3340 		err = -errno;
3341 		goto err_free_new_name;
3342 	}
3343 
3344 	new_fd = dup3(fd, new_fd, O_CLOEXEC);
3345 	if (new_fd < 0) {
3346 		err = -errno;
3347 		goto err_close_new_fd;
3348 	}
3349 
3350 	err = zclose(map->fd);
3351 	if (err) {
3352 		err = -errno;
3353 		goto err_close_new_fd;
3354 	}
3355 	free(map->name);
3356 
3357 	map->fd = new_fd;
3358 	map->name = new_name;
3359 	map->def.type = info.type;
3360 	map->def.key_size = info.key_size;
3361 	map->def.value_size = info.value_size;
3362 	map->def.max_entries = info.max_entries;
3363 	map->def.map_flags = info.map_flags;
3364 	map->btf_key_type_id = info.btf_key_type_id;
3365 	map->btf_value_type_id = info.btf_value_type_id;
3366 	map->reused = true;
3367 
3368 	return 0;
3369 
3370 err_close_new_fd:
3371 	close(new_fd);
3372 err_free_new_name:
3373 	free(new_name);
3374 	return err;
3375 }
3376 
3377 __u32 bpf_map__max_entries(const struct bpf_map *map)
3378 {
3379 	return map->def.max_entries;
3380 }
3381 
3382 int bpf_map__set_max_entries(struct bpf_map *map, __u32 max_entries)
3383 {
3384 	if (map->fd >= 0)
3385 		return -EBUSY;
3386 	map->def.max_entries = max_entries;
3387 	return 0;
3388 }
3389 
3390 int bpf_map__resize(struct bpf_map *map, __u32 max_entries)
3391 {
3392 	if (!map || !max_entries)
3393 		return -EINVAL;
3394 
3395 	return bpf_map__set_max_entries(map, max_entries);
3396 }
3397 
3398 static int
3399 bpf_object__probe_loading(struct bpf_object *obj)
3400 {
3401 	struct bpf_load_program_attr attr;
3402 	char *cp, errmsg[STRERR_BUFSIZE];
3403 	struct bpf_insn insns[] = {
3404 		BPF_MOV64_IMM(BPF_REG_0, 0),
3405 		BPF_EXIT_INSN(),
3406 	};
3407 	int ret;
3408 
3409 	/* make sure basic loading works */
3410 
3411 	memset(&attr, 0, sizeof(attr));
3412 	attr.prog_type = BPF_PROG_TYPE_SOCKET_FILTER;
3413 	attr.insns = insns;
3414 	attr.insns_cnt = ARRAY_SIZE(insns);
3415 	attr.license = "GPL";
3416 
3417 	ret = bpf_load_program_xattr(&attr, NULL, 0);
3418 	if (ret < 0) {
3419 		ret = errno;
3420 		cp = libbpf_strerror_r(ret, errmsg, sizeof(errmsg));
3421 		pr_warn("Error in %s():%s(%d). Couldn't load trivial BPF "
3422 			"program. Make sure your kernel supports BPF "
3423 			"(CONFIG_BPF_SYSCALL=y) and/or that RLIMIT_MEMLOCK is "
3424 			"set to big enough value.\n", __func__, cp, ret);
3425 		return -ret;
3426 	}
3427 	close(ret);
3428 
3429 	return 0;
3430 }
3431 
3432 static int
3433 bpf_object__probe_name(struct bpf_object *obj)
3434 {
3435 	struct bpf_load_program_attr attr;
3436 	struct bpf_insn insns[] = {
3437 		BPF_MOV64_IMM(BPF_REG_0, 0),
3438 		BPF_EXIT_INSN(),
3439 	};
3440 	int ret;
3441 
3442 	/* make sure loading with name works */
3443 
3444 	memset(&attr, 0, sizeof(attr));
3445 	attr.prog_type = BPF_PROG_TYPE_SOCKET_FILTER;
3446 	attr.insns = insns;
3447 	attr.insns_cnt = ARRAY_SIZE(insns);
3448 	attr.license = "GPL";
3449 	attr.name = "test";
3450 	ret = bpf_load_program_xattr(&attr, NULL, 0);
3451 	if (ret >= 0) {
3452 		obj->caps.name = 1;
3453 		close(ret);
3454 	}
3455 
3456 	return 0;
3457 }
3458 
3459 static int
3460 bpf_object__probe_global_data(struct bpf_object *obj)
3461 {
3462 	struct bpf_load_program_attr prg_attr;
3463 	struct bpf_create_map_attr map_attr;
3464 	char *cp, errmsg[STRERR_BUFSIZE];
3465 	struct bpf_insn insns[] = {
3466 		BPF_LD_MAP_VALUE(BPF_REG_1, 0, 16),
3467 		BPF_ST_MEM(BPF_DW, BPF_REG_1, 0, 42),
3468 		BPF_MOV64_IMM(BPF_REG_0, 0),
3469 		BPF_EXIT_INSN(),
3470 	};
3471 	int ret, map;
3472 
3473 	memset(&map_attr, 0, sizeof(map_attr));
3474 	map_attr.map_type = BPF_MAP_TYPE_ARRAY;
3475 	map_attr.key_size = sizeof(int);
3476 	map_attr.value_size = 32;
3477 	map_attr.max_entries = 1;
3478 
3479 	map = bpf_create_map_xattr(&map_attr);
3480 	if (map < 0) {
3481 		cp = libbpf_strerror_r(errno, errmsg, sizeof(errmsg));
3482 		pr_warn("Error in %s():%s(%d). Couldn't create simple array map.\n",
3483 			__func__, cp, errno);
3484 		return -errno;
3485 	}
3486 
3487 	insns[0].imm = map;
3488 
3489 	memset(&prg_attr, 0, sizeof(prg_attr));
3490 	prg_attr.prog_type = BPF_PROG_TYPE_SOCKET_FILTER;
3491 	prg_attr.insns = insns;
3492 	prg_attr.insns_cnt = ARRAY_SIZE(insns);
3493 	prg_attr.license = "GPL";
3494 
3495 	ret = bpf_load_program_xattr(&prg_attr, NULL, 0);
3496 	if (ret >= 0) {
3497 		obj->caps.global_data = 1;
3498 		close(ret);
3499 	}
3500 
3501 	close(map);
3502 	return 0;
3503 }
3504 
3505 static int bpf_object__probe_btf_func(struct bpf_object *obj)
3506 {
3507 	static const char strs[] = "\0int\0x\0a";
3508 	/* void x(int a) {} */
3509 	__u32 types[] = {
3510 		/* int */
3511 		BTF_TYPE_INT_ENC(1, BTF_INT_SIGNED, 0, 32, 4),  /* [1] */
3512 		/* FUNC_PROTO */                                /* [2] */
3513 		BTF_TYPE_ENC(0, BTF_INFO_ENC(BTF_KIND_FUNC_PROTO, 0, 1), 0),
3514 		BTF_PARAM_ENC(7, 1),
3515 		/* FUNC x */                                    /* [3] */
3516 		BTF_TYPE_ENC(5, BTF_INFO_ENC(BTF_KIND_FUNC, 0, 0), 2),
3517 	};
3518 	int btf_fd;
3519 
3520 	btf_fd = libbpf__load_raw_btf((char *)types, sizeof(types),
3521 				      strs, sizeof(strs));
3522 	if (btf_fd >= 0) {
3523 		obj->caps.btf_func = 1;
3524 		close(btf_fd);
3525 		return 1;
3526 	}
3527 
3528 	return 0;
3529 }
3530 
3531 static int bpf_object__probe_btf_func_global(struct bpf_object *obj)
3532 {
3533 	static const char strs[] = "\0int\0x\0a";
3534 	/* static void x(int a) {} */
3535 	__u32 types[] = {
3536 		/* int */
3537 		BTF_TYPE_INT_ENC(1, BTF_INT_SIGNED, 0, 32, 4),  /* [1] */
3538 		/* FUNC_PROTO */                                /* [2] */
3539 		BTF_TYPE_ENC(0, BTF_INFO_ENC(BTF_KIND_FUNC_PROTO, 0, 1), 0),
3540 		BTF_PARAM_ENC(7, 1),
3541 		/* FUNC x BTF_FUNC_GLOBAL */                    /* [3] */
3542 		BTF_TYPE_ENC(5, BTF_INFO_ENC(BTF_KIND_FUNC, 0, BTF_FUNC_GLOBAL), 2),
3543 	};
3544 	int btf_fd;
3545 
3546 	btf_fd = libbpf__load_raw_btf((char *)types, sizeof(types),
3547 				      strs, sizeof(strs));
3548 	if (btf_fd >= 0) {
3549 		obj->caps.btf_func_global = 1;
3550 		close(btf_fd);
3551 		return 1;
3552 	}
3553 
3554 	return 0;
3555 }
3556 
3557 static int bpf_object__probe_btf_datasec(struct bpf_object *obj)
3558 {
3559 	static const char strs[] = "\0x\0.data";
3560 	/* static int a; */
3561 	__u32 types[] = {
3562 		/* int */
3563 		BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 32, 4),  /* [1] */
3564 		/* VAR x */                                     /* [2] */
3565 		BTF_TYPE_ENC(1, BTF_INFO_ENC(BTF_KIND_VAR, 0, 0), 1),
3566 		BTF_VAR_STATIC,
3567 		/* DATASEC val */                               /* [3] */
3568 		BTF_TYPE_ENC(3, BTF_INFO_ENC(BTF_KIND_DATASEC, 0, 1), 4),
3569 		BTF_VAR_SECINFO_ENC(2, 0, 4),
3570 	};
3571 	int btf_fd;
3572 
3573 	btf_fd = libbpf__load_raw_btf((char *)types, sizeof(types),
3574 				      strs, sizeof(strs));
3575 	if (btf_fd >= 0) {
3576 		obj->caps.btf_datasec = 1;
3577 		close(btf_fd);
3578 		return 1;
3579 	}
3580 
3581 	return 0;
3582 }
3583 
3584 static int bpf_object__probe_array_mmap(struct bpf_object *obj)
3585 {
3586 	struct bpf_create_map_attr attr = {
3587 		.map_type = BPF_MAP_TYPE_ARRAY,
3588 		.map_flags = BPF_F_MMAPABLE,
3589 		.key_size = sizeof(int),
3590 		.value_size = sizeof(int),
3591 		.max_entries = 1,
3592 	};
3593 	int fd;
3594 
3595 	fd = bpf_create_map_xattr(&attr);
3596 	if (fd >= 0) {
3597 		obj->caps.array_mmap = 1;
3598 		close(fd);
3599 		return 1;
3600 	}
3601 
3602 	return 0;
3603 }
3604 
3605 static int
3606 bpf_object__probe_exp_attach_type(struct bpf_object *obj)
3607 {
3608 	struct bpf_load_program_attr attr;
3609 	struct bpf_insn insns[] = {
3610 		BPF_MOV64_IMM(BPF_REG_0, 0),
3611 		BPF_EXIT_INSN(),
3612 	};
3613 	int fd;
3614 
3615 	memset(&attr, 0, sizeof(attr));
3616 	/* use any valid combination of program type and (optional)
3617 	 * non-zero expected attach type (i.e., not a BPF_CGROUP_INET_INGRESS)
3618 	 * to see if kernel supports expected_attach_type field for
3619 	 * BPF_PROG_LOAD command
3620 	 */
3621 	attr.prog_type = BPF_PROG_TYPE_CGROUP_SOCK;
3622 	attr.expected_attach_type = BPF_CGROUP_INET_SOCK_CREATE;
3623 	attr.insns = insns;
3624 	attr.insns_cnt = ARRAY_SIZE(insns);
3625 	attr.license = "GPL";
3626 
3627 	fd = bpf_load_program_xattr(&attr, NULL, 0);
3628 	if (fd >= 0) {
3629 		obj->caps.exp_attach_type = 1;
3630 		close(fd);
3631 		return 1;
3632 	}
3633 	return 0;
3634 }
3635 
3636 static int
3637 bpf_object__probe_caps(struct bpf_object *obj)
3638 {
3639 	int (*probe_fn[])(struct bpf_object *obj) = {
3640 		bpf_object__probe_name,
3641 		bpf_object__probe_global_data,
3642 		bpf_object__probe_btf_func,
3643 		bpf_object__probe_btf_func_global,
3644 		bpf_object__probe_btf_datasec,
3645 		bpf_object__probe_array_mmap,
3646 		bpf_object__probe_exp_attach_type,
3647 	};
3648 	int i, ret;
3649 
3650 	for (i = 0; i < ARRAY_SIZE(probe_fn); i++) {
3651 		ret = probe_fn[i](obj);
3652 		if (ret < 0)
3653 			pr_debug("Probe #%d failed with %d.\n", i, ret);
3654 	}
3655 
3656 	return 0;
3657 }
3658 
3659 static bool map_is_reuse_compat(const struct bpf_map *map, int map_fd)
3660 {
3661 	struct bpf_map_info map_info = {};
3662 	char msg[STRERR_BUFSIZE];
3663 	__u32 map_info_len;
3664 
3665 	map_info_len = sizeof(map_info);
3666 
3667 	if (bpf_obj_get_info_by_fd(map_fd, &map_info, &map_info_len)) {
3668 		pr_warn("failed to get map info for map FD %d: %s\n",
3669 			map_fd, libbpf_strerror_r(errno, msg, sizeof(msg)));
3670 		return false;
3671 	}
3672 
3673 	return (map_info.type == map->def.type &&
3674 		map_info.key_size == map->def.key_size &&
3675 		map_info.value_size == map->def.value_size &&
3676 		map_info.max_entries == map->def.max_entries &&
3677 		map_info.map_flags == map->def.map_flags);
3678 }
3679 
3680 static int
3681 bpf_object__reuse_map(struct bpf_map *map)
3682 {
3683 	char *cp, errmsg[STRERR_BUFSIZE];
3684 	int err, pin_fd;
3685 
3686 	pin_fd = bpf_obj_get(map->pin_path);
3687 	if (pin_fd < 0) {
3688 		err = -errno;
3689 		if (err == -ENOENT) {
3690 			pr_debug("found no pinned map to reuse at '%s'\n",
3691 				 map->pin_path);
3692 			return 0;
3693 		}
3694 
3695 		cp = libbpf_strerror_r(-err, errmsg, sizeof(errmsg));
3696 		pr_warn("couldn't retrieve pinned map '%s': %s\n",
3697 			map->pin_path, cp);
3698 		return err;
3699 	}
3700 
3701 	if (!map_is_reuse_compat(map, pin_fd)) {
3702 		pr_warn("couldn't reuse pinned map at '%s': parameter mismatch\n",
3703 			map->pin_path);
3704 		close(pin_fd);
3705 		return -EINVAL;
3706 	}
3707 
3708 	err = bpf_map__reuse_fd(map, pin_fd);
3709 	if (err) {
3710 		close(pin_fd);
3711 		return err;
3712 	}
3713 	map->pinned = true;
3714 	pr_debug("reused pinned map at '%s'\n", map->pin_path);
3715 
3716 	return 0;
3717 }
3718 
3719 static int
3720 bpf_object__populate_internal_map(struct bpf_object *obj, struct bpf_map *map)
3721 {
3722 	enum libbpf_map_type map_type = map->libbpf_type;
3723 	char *cp, errmsg[STRERR_BUFSIZE];
3724 	int err, zero = 0;
3725 
3726 	err = bpf_map_update_elem(map->fd, &zero, map->mmaped, 0);
3727 	if (err) {
3728 		err = -errno;
3729 		cp = libbpf_strerror_r(err, errmsg, sizeof(errmsg));
3730 		pr_warn("Error setting initial map(%s) contents: %s\n",
3731 			map->name, cp);
3732 		return err;
3733 	}
3734 
3735 	/* Freeze .rodata and .kconfig map as read-only from syscall side. */
3736 	if (map_type == LIBBPF_MAP_RODATA || map_type == LIBBPF_MAP_KCONFIG) {
3737 		err = bpf_map_freeze(map->fd);
3738 		if (err) {
3739 			err = -errno;
3740 			cp = libbpf_strerror_r(err, errmsg, sizeof(errmsg));
3741 			pr_warn("Error freezing map(%s) as read-only: %s\n",
3742 				map->name, cp);
3743 			return err;
3744 		}
3745 	}
3746 	return 0;
3747 }
3748 
3749 static void bpf_map__destroy(struct bpf_map *map);
3750 
3751 static int bpf_object__create_map(struct bpf_object *obj, struct bpf_map *map)
3752 {
3753 	struct bpf_create_map_attr create_attr;
3754 	struct bpf_map_def *def = &map->def;
3755 
3756 	memset(&create_attr, 0, sizeof(create_attr));
3757 
3758 	if (obj->caps.name)
3759 		create_attr.name = map->name;
3760 	create_attr.map_ifindex = map->map_ifindex;
3761 	create_attr.map_type = def->type;
3762 	create_attr.map_flags = def->map_flags;
3763 	create_attr.key_size = def->key_size;
3764 	create_attr.value_size = def->value_size;
3765 	create_attr.numa_node = map->numa_node;
3766 
3767 	if (def->type == BPF_MAP_TYPE_PERF_EVENT_ARRAY && !def->max_entries) {
3768 		int nr_cpus;
3769 
3770 		nr_cpus = libbpf_num_possible_cpus();
3771 		if (nr_cpus < 0) {
3772 			pr_warn("map '%s': failed to determine number of system CPUs: %d\n",
3773 				map->name, nr_cpus);
3774 			return nr_cpus;
3775 		}
3776 		pr_debug("map '%s': setting size to %d\n", map->name, nr_cpus);
3777 		create_attr.max_entries = nr_cpus;
3778 	} else {
3779 		create_attr.max_entries = def->max_entries;
3780 	}
3781 
3782 	if (bpf_map__is_struct_ops(map))
3783 		create_attr.btf_vmlinux_value_type_id =
3784 			map->btf_vmlinux_value_type_id;
3785 
3786 	create_attr.btf_fd = 0;
3787 	create_attr.btf_key_type_id = 0;
3788 	create_attr.btf_value_type_id = 0;
3789 	if (obj->btf && btf__fd(obj->btf) >= 0 && !bpf_map_find_btf_info(obj, map)) {
3790 		create_attr.btf_fd = btf__fd(obj->btf);
3791 		create_attr.btf_key_type_id = map->btf_key_type_id;
3792 		create_attr.btf_value_type_id = map->btf_value_type_id;
3793 	}
3794 
3795 	if (bpf_map_type__is_map_in_map(def->type)) {
3796 		if (map->inner_map) {
3797 			int err;
3798 
3799 			err = bpf_object__create_map(obj, map->inner_map);
3800 			if (err) {
3801 				pr_warn("map '%s': failed to create inner map: %d\n",
3802 					map->name, err);
3803 				return err;
3804 			}
3805 			map->inner_map_fd = bpf_map__fd(map->inner_map);
3806 		}
3807 		if (map->inner_map_fd >= 0)
3808 			create_attr.inner_map_fd = map->inner_map_fd;
3809 	}
3810 
3811 	map->fd = bpf_create_map_xattr(&create_attr);
3812 	if (map->fd < 0 && (create_attr.btf_key_type_id ||
3813 			    create_attr.btf_value_type_id)) {
3814 		char *cp, errmsg[STRERR_BUFSIZE];
3815 		int err = -errno;
3816 
3817 		cp = libbpf_strerror_r(err, errmsg, sizeof(errmsg));
3818 		pr_warn("Error in bpf_create_map_xattr(%s):%s(%d). Retrying without BTF.\n",
3819 			map->name, cp, err);
3820 		create_attr.btf_fd = 0;
3821 		create_attr.btf_key_type_id = 0;
3822 		create_attr.btf_value_type_id = 0;
3823 		map->btf_key_type_id = 0;
3824 		map->btf_value_type_id = 0;
3825 		map->fd = bpf_create_map_xattr(&create_attr);
3826 	}
3827 
3828 	if (map->fd < 0)
3829 		return -errno;
3830 
3831 	if (bpf_map_type__is_map_in_map(def->type) && map->inner_map) {
3832 		bpf_map__destroy(map->inner_map);
3833 		zfree(&map->inner_map);
3834 	}
3835 
3836 	return 0;
3837 }
3838 
3839 static int
3840 bpf_object__create_maps(struct bpf_object *obj)
3841 {
3842 	struct bpf_map *map;
3843 	char *cp, errmsg[STRERR_BUFSIZE];
3844 	unsigned int i, j;
3845 	int err;
3846 
3847 	for (i = 0; i < obj->nr_maps; i++) {
3848 		map = &obj->maps[i];
3849 
3850 		if (map->pin_path) {
3851 			err = bpf_object__reuse_map(map);
3852 			if (err) {
3853 				pr_warn("map '%s': error reusing pinned map\n",
3854 					map->name);
3855 				goto err_out;
3856 			}
3857 		}
3858 
3859 		if (map->fd >= 0) {
3860 			pr_debug("map '%s': skipping creation (preset fd=%d)\n",
3861 				 map->name, map->fd);
3862 			continue;
3863 		}
3864 
3865 		err = bpf_object__create_map(obj, map);
3866 		if (err)
3867 			goto err_out;
3868 
3869 		pr_debug("map '%s': created successfully, fd=%d\n", map->name,
3870 			 map->fd);
3871 
3872 		if (bpf_map__is_internal(map)) {
3873 			err = bpf_object__populate_internal_map(obj, map);
3874 			if (err < 0) {
3875 				zclose(map->fd);
3876 				goto err_out;
3877 			}
3878 		}
3879 
3880 		if (map->init_slots_sz) {
3881 			for (j = 0; j < map->init_slots_sz; j++) {
3882 				const struct bpf_map *targ_map;
3883 				int fd;
3884 
3885 				if (!map->init_slots[j])
3886 					continue;
3887 
3888 				targ_map = map->init_slots[j];
3889 				fd = bpf_map__fd(targ_map);
3890 				err = bpf_map_update_elem(map->fd, &j, &fd, 0);
3891 				if (err) {
3892 					err = -errno;
3893 					pr_warn("map '%s': failed to initialize slot [%d] to map '%s' fd=%d: %d\n",
3894 						map->name, j, targ_map->name,
3895 						fd, err);
3896 					goto err_out;
3897 				}
3898 				pr_debug("map '%s': slot [%d] set to map '%s' fd=%d\n",
3899 					 map->name, j, targ_map->name, fd);
3900 			}
3901 			zfree(&map->init_slots);
3902 			map->init_slots_sz = 0;
3903 		}
3904 
3905 		if (map->pin_path && !map->pinned) {
3906 			err = bpf_map__pin(map, NULL);
3907 			if (err) {
3908 				pr_warn("map '%s': failed to auto-pin at '%s': %d\n",
3909 					map->name, map->pin_path, err);
3910 				zclose(map->fd);
3911 				goto err_out;
3912 			}
3913 		}
3914 	}
3915 
3916 	return 0;
3917 
3918 err_out:
3919 	cp = libbpf_strerror_r(err, errmsg, sizeof(errmsg));
3920 	pr_warn("map '%s': failed to create: %s(%d)\n", map->name, cp, err);
3921 	pr_perm_msg(err);
3922 	for (j = 0; j < i; j++)
3923 		zclose(obj->maps[j].fd);
3924 	return err;
3925 }
3926 
3927 static int
3928 check_btf_ext_reloc_err(struct bpf_program *prog, int err,
3929 			void *btf_prog_info, const char *info_name)
3930 {
3931 	if (err != -ENOENT) {
3932 		pr_warn("Error in loading %s for sec %s.\n",
3933 			info_name, prog->section_name);
3934 		return err;
3935 	}
3936 
3937 	/* err == -ENOENT (i.e. prog->section_name not found in btf_ext) */
3938 
3939 	if (btf_prog_info) {
3940 		/*
3941 		 * Some info has already been found but has problem
3942 		 * in the last btf_ext reloc. Must have to error out.
3943 		 */
3944 		pr_warn("Error in relocating %s for sec %s.\n",
3945 			info_name, prog->section_name);
3946 		return err;
3947 	}
3948 
3949 	/* Have problem loading the very first info. Ignore the rest. */
3950 	pr_warn("Cannot find %s for main program sec %s. Ignore all %s.\n",
3951 		info_name, prog->section_name, info_name);
3952 	return 0;
3953 }
3954 
3955 static int
3956 bpf_program_reloc_btf_ext(struct bpf_program *prog, struct bpf_object *obj,
3957 			  const char *section_name,  __u32 insn_offset)
3958 {
3959 	int err;
3960 
3961 	if (!insn_offset || prog->func_info) {
3962 		/*
3963 		 * !insn_offset => main program
3964 		 *
3965 		 * For sub prog, the main program's func_info has to
3966 		 * be loaded first (i.e. prog->func_info != NULL)
3967 		 */
3968 		err = btf_ext__reloc_func_info(obj->btf, obj->btf_ext,
3969 					       section_name, insn_offset,
3970 					       &prog->func_info,
3971 					       &prog->func_info_cnt);
3972 		if (err)
3973 			return check_btf_ext_reloc_err(prog, err,
3974 						       prog->func_info,
3975 						       "bpf_func_info");
3976 
3977 		prog->func_info_rec_size = btf_ext__func_info_rec_size(obj->btf_ext);
3978 	}
3979 
3980 	if (!insn_offset || prog->line_info) {
3981 		err = btf_ext__reloc_line_info(obj->btf, obj->btf_ext,
3982 					       section_name, insn_offset,
3983 					       &prog->line_info,
3984 					       &prog->line_info_cnt);
3985 		if (err)
3986 			return check_btf_ext_reloc_err(prog, err,
3987 						       prog->line_info,
3988 						       "bpf_line_info");
3989 
3990 		prog->line_info_rec_size = btf_ext__line_info_rec_size(obj->btf_ext);
3991 	}
3992 
3993 	return 0;
3994 }
3995 
3996 #define BPF_CORE_SPEC_MAX_LEN 64
3997 
3998 /* represents BPF CO-RE field or array element accessor */
3999 struct bpf_core_accessor {
4000 	__u32 type_id;		/* struct/union type or array element type */
4001 	__u32 idx;		/* field index or array index */
4002 	const char *name;	/* field name or NULL for array accessor */
4003 };
4004 
4005 struct bpf_core_spec {
4006 	const struct btf *btf;
4007 	/* high-level spec: named fields and array indices only */
4008 	struct bpf_core_accessor spec[BPF_CORE_SPEC_MAX_LEN];
4009 	/* high-level spec length */
4010 	int len;
4011 	/* raw, low-level spec: 1-to-1 with accessor spec string */
4012 	int raw_spec[BPF_CORE_SPEC_MAX_LEN];
4013 	/* raw spec length */
4014 	int raw_len;
4015 	/* field bit offset represented by spec */
4016 	__u32 bit_offset;
4017 };
4018 
4019 static bool str_is_empty(const char *s)
4020 {
4021 	return !s || !s[0];
4022 }
4023 
4024 static bool is_flex_arr(const struct btf *btf,
4025 			const struct bpf_core_accessor *acc,
4026 			const struct btf_array *arr)
4027 {
4028 	const struct btf_type *t;
4029 
4030 	/* not a flexible array, if not inside a struct or has non-zero size */
4031 	if (!acc->name || arr->nelems > 0)
4032 		return false;
4033 
4034 	/* has to be the last member of enclosing struct */
4035 	t = btf__type_by_id(btf, acc->type_id);
4036 	return acc->idx == btf_vlen(t) - 1;
4037 }
4038 
4039 /*
4040  * Turn bpf_field_reloc into a low- and high-level spec representation,
4041  * validating correctness along the way, as well as calculating resulting
4042  * field bit offset, specified by accessor string. Low-level spec captures
4043  * every single level of nestedness, including traversing anonymous
4044  * struct/union members. High-level one only captures semantically meaningful
4045  * "turning points": named fields and array indicies.
4046  * E.g., for this case:
4047  *
4048  *   struct sample {
4049  *       int __unimportant;
4050  *       struct {
4051  *           int __1;
4052  *           int __2;
4053  *           int a[7];
4054  *       };
4055  *   };
4056  *
4057  *   struct sample *s = ...;
4058  *
4059  *   int x = &s->a[3]; // access string = '0:1:2:3'
4060  *
4061  * Low-level spec has 1:1 mapping with each element of access string (it's
4062  * just a parsed access string representation): [0, 1, 2, 3].
4063  *
4064  * High-level spec will capture only 3 points:
4065  *   - intial zero-index access by pointer (&s->... is the same as &s[0]...);
4066  *   - field 'a' access (corresponds to '2' in low-level spec);
4067  *   - array element #3 access (corresponds to '3' in low-level spec).
4068  *
4069  */
4070 static int bpf_core_spec_parse(const struct btf *btf,
4071 			       __u32 type_id,
4072 			       const char *spec_str,
4073 			       struct bpf_core_spec *spec)
4074 {
4075 	int access_idx, parsed_len, i;
4076 	struct bpf_core_accessor *acc;
4077 	const struct btf_type *t;
4078 	const char *name;
4079 	__u32 id;
4080 	__s64 sz;
4081 
4082 	if (str_is_empty(spec_str) || *spec_str == ':')
4083 		return -EINVAL;
4084 
4085 	memset(spec, 0, sizeof(*spec));
4086 	spec->btf = btf;
4087 
4088 	/* parse spec_str="0:1:2:3:4" into array raw_spec=[0, 1, 2, 3, 4] */
4089 	while (*spec_str) {
4090 		if (*spec_str == ':')
4091 			++spec_str;
4092 		if (sscanf(spec_str, "%d%n", &access_idx, &parsed_len) != 1)
4093 			return -EINVAL;
4094 		if (spec->raw_len == BPF_CORE_SPEC_MAX_LEN)
4095 			return -E2BIG;
4096 		spec_str += parsed_len;
4097 		spec->raw_spec[spec->raw_len++] = access_idx;
4098 	}
4099 
4100 	if (spec->raw_len == 0)
4101 		return -EINVAL;
4102 
4103 	/* first spec value is always reloc type array index */
4104 	t = skip_mods_and_typedefs(btf, type_id, &id);
4105 	if (!t)
4106 		return -EINVAL;
4107 
4108 	access_idx = spec->raw_spec[0];
4109 	spec->spec[0].type_id = id;
4110 	spec->spec[0].idx = access_idx;
4111 	spec->len++;
4112 
4113 	sz = btf__resolve_size(btf, id);
4114 	if (sz < 0)
4115 		return sz;
4116 	spec->bit_offset = access_idx * sz * 8;
4117 
4118 	for (i = 1; i < spec->raw_len; i++) {
4119 		t = skip_mods_and_typedefs(btf, id, &id);
4120 		if (!t)
4121 			return -EINVAL;
4122 
4123 		access_idx = spec->raw_spec[i];
4124 		acc = &spec->spec[spec->len];
4125 
4126 		if (btf_is_composite(t)) {
4127 			const struct btf_member *m;
4128 			__u32 bit_offset;
4129 
4130 			if (access_idx >= btf_vlen(t))
4131 				return -EINVAL;
4132 
4133 			bit_offset = btf_member_bit_offset(t, access_idx);
4134 			spec->bit_offset += bit_offset;
4135 
4136 			m = btf_members(t) + access_idx;
4137 			if (m->name_off) {
4138 				name = btf__name_by_offset(btf, m->name_off);
4139 				if (str_is_empty(name))
4140 					return -EINVAL;
4141 
4142 				acc->type_id = id;
4143 				acc->idx = access_idx;
4144 				acc->name = name;
4145 				spec->len++;
4146 			}
4147 
4148 			id = m->type;
4149 		} else if (btf_is_array(t)) {
4150 			const struct btf_array *a = btf_array(t);
4151 			bool flex;
4152 
4153 			t = skip_mods_and_typedefs(btf, a->type, &id);
4154 			if (!t)
4155 				return -EINVAL;
4156 
4157 			flex = is_flex_arr(btf, acc - 1, a);
4158 			if (!flex && access_idx >= a->nelems)
4159 				return -EINVAL;
4160 
4161 			spec->spec[spec->len].type_id = id;
4162 			spec->spec[spec->len].idx = access_idx;
4163 			spec->len++;
4164 
4165 			sz = btf__resolve_size(btf, id);
4166 			if (sz < 0)
4167 				return sz;
4168 			spec->bit_offset += access_idx * sz * 8;
4169 		} else {
4170 			pr_warn("relo for [%u] %s (at idx %d) captures type [%d] of unexpected kind %d\n",
4171 				type_id, spec_str, i, id, btf_kind(t));
4172 			return -EINVAL;
4173 		}
4174 	}
4175 
4176 	return 0;
4177 }
4178 
4179 static bool bpf_core_is_flavor_sep(const char *s)
4180 {
4181 	/* check X___Y name pattern, where X and Y are not underscores */
4182 	return s[0] != '_' &&				      /* X */
4183 	       s[1] == '_' && s[2] == '_' && s[3] == '_' &&   /* ___ */
4184 	       s[4] != '_';				      /* Y */
4185 }
4186 
4187 /* Given 'some_struct_name___with_flavor' return the length of a name prefix
4188  * before last triple underscore. Struct name part after last triple
4189  * underscore is ignored by BPF CO-RE relocation during relocation matching.
4190  */
4191 static size_t bpf_core_essential_name_len(const char *name)
4192 {
4193 	size_t n = strlen(name);
4194 	int i;
4195 
4196 	for (i = n - 5; i >= 0; i--) {
4197 		if (bpf_core_is_flavor_sep(name + i))
4198 			return i + 1;
4199 	}
4200 	return n;
4201 }
4202 
4203 /* dynamically sized list of type IDs */
4204 struct ids_vec {
4205 	__u32 *data;
4206 	int len;
4207 };
4208 
4209 static void bpf_core_free_cands(struct ids_vec *cand_ids)
4210 {
4211 	free(cand_ids->data);
4212 	free(cand_ids);
4213 }
4214 
4215 static struct ids_vec *bpf_core_find_cands(const struct btf *local_btf,
4216 					   __u32 local_type_id,
4217 					   const struct btf *targ_btf)
4218 {
4219 	size_t local_essent_len, targ_essent_len;
4220 	const char *local_name, *targ_name;
4221 	const struct btf_type *t;
4222 	struct ids_vec *cand_ids;
4223 	__u32 *new_ids;
4224 	int i, err, n;
4225 
4226 	t = btf__type_by_id(local_btf, local_type_id);
4227 	if (!t)
4228 		return ERR_PTR(-EINVAL);
4229 
4230 	local_name = btf__name_by_offset(local_btf, t->name_off);
4231 	if (str_is_empty(local_name))
4232 		return ERR_PTR(-EINVAL);
4233 	local_essent_len = bpf_core_essential_name_len(local_name);
4234 
4235 	cand_ids = calloc(1, sizeof(*cand_ids));
4236 	if (!cand_ids)
4237 		return ERR_PTR(-ENOMEM);
4238 
4239 	n = btf__get_nr_types(targ_btf);
4240 	for (i = 1; i <= n; i++) {
4241 		t = btf__type_by_id(targ_btf, i);
4242 		targ_name = btf__name_by_offset(targ_btf, t->name_off);
4243 		if (str_is_empty(targ_name))
4244 			continue;
4245 
4246 		t = skip_mods_and_typedefs(targ_btf, i, NULL);
4247 		if (!btf_is_composite(t) && !btf_is_array(t))
4248 			continue;
4249 
4250 		targ_essent_len = bpf_core_essential_name_len(targ_name);
4251 		if (targ_essent_len != local_essent_len)
4252 			continue;
4253 
4254 		if (strncmp(local_name, targ_name, local_essent_len) == 0) {
4255 			pr_debug("[%d] %s: found candidate [%d] %s\n",
4256 				 local_type_id, local_name, i, targ_name);
4257 			new_ids = reallocarray(cand_ids->data,
4258 					       cand_ids->len + 1,
4259 					       sizeof(*cand_ids->data));
4260 			if (!new_ids) {
4261 				err = -ENOMEM;
4262 				goto err_out;
4263 			}
4264 			cand_ids->data = new_ids;
4265 			cand_ids->data[cand_ids->len++] = i;
4266 		}
4267 	}
4268 	return cand_ids;
4269 err_out:
4270 	bpf_core_free_cands(cand_ids);
4271 	return ERR_PTR(err);
4272 }
4273 
4274 /* Check two types for compatibility, skipping const/volatile/restrict and
4275  * typedefs, to ensure we are relocating compatible entities:
4276  *   - any two STRUCTs/UNIONs are compatible and can be mixed;
4277  *   - any two FWDs are compatible, if their names match (modulo flavor suffix);
4278  *   - any two PTRs are always compatible;
4279  *   - for ENUMs, names should be the same (ignoring flavor suffix) or at
4280  *     least one of enums should be anonymous;
4281  *   - for ENUMs, check sizes, names are ignored;
4282  *   - for INT, size and signedness are ignored;
4283  *   - for ARRAY, dimensionality is ignored, element types are checked for
4284  *     compatibility recursively;
4285  *   - everything else shouldn't be ever a target of relocation.
4286  * These rules are not set in stone and probably will be adjusted as we get
4287  * more experience with using BPF CO-RE relocations.
4288  */
4289 static int bpf_core_fields_are_compat(const struct btf *local_btf,
4290 				      __u32 local_id,
4291 				      const struct btf *targ_btf,
4292 				      __u32 targ_id)
4293 {
4294 	const struct btf_type *local_type, *targ_type;
4295 
4296 recur:
4297 	local_type = skip_mods_and_typedefs(local_btf, local_id, &local_id);
4298 	targ_type = skip_mods_and_typedefs(targ_btf, targ_id, &targ_id);
4299 	if (!local_type || !targ_type)
4300 		return -EINVAL;
4301 
4302 	if (btf_is_composite(local_type) && btf_is_composite(targ_type))
4303 		return 1;
4304 	if (btf_kind(local_type) != btf_kind(targ_type))
4305 		return 0;
4306 
4307 	switch (btf_kind(local_type)) {
4308 	case BTF_KIND_PTR:
4309 		return 1;
4310 	case BTF_KIND_FWD:
4311 	case BTF_KIND_ENUM: {
4312 		const char *local_name, *targ_name;
4313 		size_t local_len, targ_len;
4314 
4315 		local_name = btf__name_by_offset(local_btf,
4316 						 local_type->name_off);
4317 		targ_name = btf__name_by_offset(targ_btf, targ_type->name_off);
4318 		local_len = bpf_core_essential_name_len(local_name);
4319 		targ_len = bpf_core_essential_name_len(targ_name);
4320 		/* one of them is anonymous or both w/ same flavor-less names */
4321 		return local_len == 0 || targ_len == 0 ||
4322 		       (local_len == targ_len &&
4323 			strncmp(local_name, targ_name, local_len) == 0);
4324 	}
4325 	case BTF_KIND_INT:
4326 		/* just reject deprecated bitfield-like integers; all other
4327 		 * integers are by default compatible between each other
4328 		 */
4329 		return btf_int_offset(local_type) == 0 &&
4330 		       btf_int_offset(targ_type) == 0;
4331 	case BTF_KIND_ARRAY:
4332 		local_id = btf_array(local_type)->type;
4333 		targ_id = btf_array(targ_type)->type;
4334 		goto recur;
4335 	default:
4336 		pr_warn("unexpected kind %d relocated, local [%d], target [%d]\n",
4337 			btf_kind(local_type), local_id, targ_id);
4338 		return 0;
4339 	}
4340 }
4341 
4342 /*
4343  * Given single high-level named field accessor in local type, find
4344  * corresponding high-level accessor for a target type. Along the way,
4345  * maintain low-level spec for target as well. Also keep updating target
4346  * bit offset.
4347  *
4348  * Searching is performed through recursive exhaustive enumeration of all
4349  * fields of a struct/union. If there are any anonymous (embedded)
4350  * structs/unions, they are recursively searched as well. If field with
4351  * desired name is found, check compatibility between local and target types,
4352  * before returning result.
4353  *
4354  * 1 is returned, if field is found.
4355  * 0 is returned if no compatible field is found.
4356  * <0 is returned on error.
4357  */
4358 static int bpf_core_match_member(const struct btf *local_btf,
4359 				 const struct bpf_core_accessor *local_acc,
4360 				 const struct btf *targ_btf,
4361 				 __u32 targ_id,
4362 				 struct bpf_core_spec *spec,
4363 				 __u32 *next_targ_id)
4364 {
4365 	const struct btf_type *local_type, *targ_type;
4366 	const struct btf_member *local_member, *m;
4367 	const char *local_name, *targ_name;
4368 	__u32 local_id;
4369 	int i, n, found;
4370 
4371 	targ_type = skip_mods_and_typedefs(targ_btf, targ_id, &targ_id);
4372 	if (!targ_type)
4373 		return -EINVAL;
4374 	if (!btf_is_composite(targ_type))
4375 		return 0;
4376 
4377 	local_id = local_acc->type_id;
4378 	local_type = btf__type_by_id(local_btf, local_id);
4379 	local_member = btf_members(local_type) + local_acc->idx;
4380 	local_name = btf__name_by_offset(local_btf, local_member->name_off);
4381 
4382 	n = btf_vlen(targ_type);
4383 	m = btf_members(targ_type);
4384 	for (i = 0; i < n; i++, m++) {
4385 		__u32 bit_offset;
4386 
4387 		bit_offset = btf_member_bit_offset(targ_type, i);
4388 
4389 		/* too deep struct/union/array nesting */
4390 		if (spec->raw_len == BPF_CORE_SPEC_MAX_LEN)
4391 			return -E2BIG;
4392 
4393 		/* speculate this member will be the good one */
4394 		spec->bit_offset += bit_offset;
4395 		spec->raw_spec[spec->raw_len++] = i;
4396 
4397 		targ_name = btf__name_by_offset(targ_btf, m->name_off);
4398 		if (str_is_empty(targ_name)) {
4399 			/* embedded struct/union, we need to go deeper */
4400 			found = bpf_core_match_member(local_btf, local_acc,
4401 						      targ_btf, m->type,
4402 						      spec, next_targ_id);
4403 			if (found) /* either found or error */
4404 				return found;
4405 		} else if (strcmp(local_name, targ_name) == 0) {
4406 			/* matching named field */
4407 			struct bpf_core_accessor *targ_acc;
4408 
4409 			targ_acc = &spec->spec[spec->len++];
4410 			targ_acc->type_id = targ_id;
4411 			targ_acc->idx = i;
4412 			targ_acc->name = targ_name;
4413 
4414 			*next_targ_id = m->type;
4415 			found = bpf_core_fields_are_compat(local_btf,
4416 							   local_member->type,
4417 							   targ_btf, m->type);
4418 			if (!found)
4419 				spec->len--; /* pop accessor */
4420 			return found;
4421 		}
4422 		/* member turned out not to be what we looked for */
4423 		spec->bit_offset -= bit_offset;
4424 		spec->raw_len--;
4425 	}
4426 
4427 	return 0;
4428 }
4429 
4430 /*
4431  * Try to match local spec to a target type and, if successful, produce full
4432  * target spec (high-level, low-level + bit offset).
4433  */
4434 static int bpf_core_spec_match(struct bpf_core_spec *local_spec,
4435 			       const struct btf *targ_btf, __u32 targ_id,
4436 			       struct bpf_core_spec *targ_spec)
4437 {
4438 	const struct btf_type *targ_type;
4439 	const struct bpf_core_accessor *local_acc;
4440 	struct bpf_core_accessor *targ_acc;
4441 	int i, sz, matched;
4442 
4443 	memset(targ_spec, 0, sizeof(*targ_spec));
4444 	targ_spec->btf = targ_btf;
4445 
4446 	local_acc = &local_spec->spec[0];
4447 	targ_acc = &targ_spec->spec[0];
4448 
4449 	for (i = 0; i < local_spec->len; i++, local_acc++, targ_acc++) {
4450 		targ_type = skip_mods_and_typedefs(targ_spec->btf, targ_id,
4451 						   &targ_id);
4452 		if (!targ_type)
4453 			return -EINVAL;
4454 
4455 		if (local_acc->name) {
4456 			matched = bpf_core_match_member(local_spec->btf,
4457 							local_acc,
4458 							targ_btf, targ_id,
4459 							targ_spec, &targ_id);
4460 			if (matched <= 0)
4461 				return matched;
4462 		} else {
4463 			/* for i=0, targ_id is already treated as array element
4464 			 * type (because it's the original struct), for others
4465 			 * we should find array element type first
4466 			 */
4467 			if (i > 0) {
4468 				const struct btf_array *a;
4469 				bool flex;
4470 
4471 				if (!btf_is_array(targ_type))
4472 					return 0;
4473 
4474 				a = btf_array(targ_type);
4475 				flex = is_flex_arr(targ_btf, targ_acc - 1, a);
4476 				if (!flex && local_acc->idx >= a->nelems)
4477 					return 0;
4478 				if (!skip_mods_and_typedefs(targ_btf, a->type,
4479 							    &targ_id))
4480 					return -EINVAL;
4481 			}
4482 
4483 			/* too deep struct/union/array nesting */
4484 			if (targ_spec->raw_len == BPF_CORE_SPEC_MAX_LEN)
4485 				return -E2BIG;
4486 
4487 			targ_acc->type_id = targ_id;
4488 			targ_acc->idx = local_acc->idx;
4489 			targ_acc->name = NULL;
4490 			targ_spec->len++;
4491 			targ_spec->raw_spec[targ_spec->raw_len] = targ_acc->idx;
4492 			targ_spec->raw_len++;
4493 
4494 			sz = btf__resolve_size(targ_btf, targ_id);
4495 			if (sz < 0)
4496 				return sz;
4497 			targ_spec->bit_offset += local_acc->idx * sz * 8;
4498 		}
4499 	}
4500 
4501 	return 1;
4502 }
4503 
4504 static int bpf_core_calc_field_relo(const struct bpf_program *prog,
4505 				    const struct bpf_field_reloc *relo,
4506 				    const struct bpf_core_spec *spec,
4507 				    __u32 *val, bool *validate)
4508 {
4509 	const struct bpf_core_accessor *acc = &spec->spec[spec->len - 1];
4510 	const struct btf_type *t = btf__type_by_id(spec->btf, acc->type_id);
4511 	__u32 byte_off, byte_sz, bit_off, bit_sz;
4512 	const struct btf_member *m;
4513 	const struct btf_type *mt;
4514 	bool bitfield;
4515 	__s64 sz;
4516 
4517 	/* a[n] accessor needs special handling */
4518 	if (!acc->name) {
4519 		if (relo->kind == BPF_FIELD_BYTE_OFFSET) {
4520 			*val = spec->bit_offset / 8;
4521 		} else if (relo->kind == BPF_FIELD_BYTE_SIZE) {
4522 			sz = btf__resolve_size(spec->btf, acc->type_id);
4523 			if (sz < 0)
4524 				return -EINVAL;
4525 			*val = sz;
4526 		} else {
4527 			pr_warn("prog '%s': relo %d at insn #%d can't be applied to array access\n",
4528 				bpf_program__title(prog, false),
4529 				relo->kind, relo->insn_off / 8);
4530 			return -EINVAL;
4531 		}
4532 		if (validate)
4533 			*validate = true;
4534 		return 0;
4535 	}
4536 
4537 	m = btf_members(t) + acc->idx;
4538 	mt = skip_mods_and_typedefs(spec->btf, m->type, NULL);
4539 	bit_off = spec->bit_offset;
4540 	bit_sz = btf_member_bitfield_size(t, acc->idx);
4541 
4542 	bitfield = bit_sz > 0;
4543 	if (bitfield) {
4544 		byte_sz = mt->size;
4545 		byte_off = bit_off / 8 / byte_sz * byte_sz;
4546 		/* figure out smallest int size necessary for bitfield load */
4547 		while (bit_off + bit_sz - byte_off * 8 > byte_sz * 8) {
4548 			if (byte_sz >= 8) {
4549 				/* bitfield can't be read with 64-bit read */
4550 				pr_warn("prog '%s': relo %d at insn #%d can't be satisfied for bitfield\n",
4551 					bpf_program__title(prog, false),
4552 					relo->kind, relo->insn_off / 8);
4553 				return -E2BIG;
4554 			}
4555 			byte_sz *= 2;
4556 			byte_off = bit_off / 8 / byte_sz * byte_sz;
4557 		}
4558 	} else {
4559 		sz = btf__resolve_size(spec->btf, m->type);
4560 		if (sz < 0)
4561 			return -EINVAL;
4562 		byte_sz = sz;
4563 		byte_off = spec->bit_offset / 8;
4564 		bit_sz = byte_sz * 8;
4565 	}
4566 
4567 	/* for bitfields, all the relocatable aspects are ambiguous and we
4568 	 * might disagree with compiler, so turn off validation of expected
4569 	 * value, except for signedness
4570 	 */
4571 	if (validate)
4572 		*validate = !bitfield;
4573 
4574 	switch (relo->kind) {
4575 	case BPF_FIELD_BYTE_OFFSET:
4576 		*val = byte_off;
4577 		break;
4578 	case BPF_FIELD_BYTE_SIZE:
4579 		*val = byte_sz;
4580 		break;
4581 	case BPF_FIELD_SIGNED:
4582 		/* enums will be assumed unsigned */
4583 		*val = btf_is_enum(mt) ||
4584 		       (btf_int_encoding(mt) & BTF_INT_SIGNED);
4585 		if (validate)
4586 			*validate = true; /* signedness is never ambiguous */
4587 		break;
4588 	case BPF_FIELD_LSHIFT_U64:
4589 #if __BYTE_ORDER == __LITTLE_ENDIAN
4590 		*val = 64 - (bit_off + bit_sz - byte_off  * 8);
4591 #else
4592 		*val = (8 - byte_sz) * 8 + (bit_off - byte_off * 8);
4593 #endif
4594 		break;
4595 	case BPF_FIELD_RSHIFT_U64:
4596 		*val = 64 - bit_sz;
4597 		if (validate)
4598 			*validate = true; /* right shift is never ambiguous */
4599 		break;
4600 	case BPF_FIELD_EXISTS:
4601 	default:
4602 		pr_warn("prog '%s': unknown relo %d at insn #%d\n",
4603 			bpf_program__title(prog, false),
4604 			relo->kind, relo->insn_off / 8);
4605 		return -EINVAL;
4606 	}
4607 
4608 	return 0;
4609 }
4610 
4611 /*
4612  * Patch relocatable BPF instruction.
4613  *
4614  * Patched value is determined by relocation kind and target specification.
4615  * For field existence relocation target spec will be NULL if field is not
4616  * found.
4617  * Expected insn->imm value is determined using relocation kind and local
4618  * spec, and is checked before patching instruction. If actual insn->imm value
4619  * is wrong, bail out with error.
4620  *
4621  * Currently three kinds of BPF instructions are supported:
4622  * 1. rX = <imm> (assignment with immediate operand);
4623  * 2. rX += <imm> (arithmetic operations with immediate operand);
4624  */
4625 static int bpf_core_reloc_insn(struct bpf_program *prog,
4626 			       const struct bpf_field_reloc *relo,
4627 			       int relo_idx,
4628 			       const struct bpf_core_spec *local_spec,
4629 			       const struct bpf_core_spec *targ_spec)
4630 {
4631 	__u32 orig_val, new_val;
4632 	struct bpf_insn *insn;
4633 	bool validate = true;
4634 	int insn_idx, err;
4635 	__u8 class;
4636 
4637 	if (relo->insn_off % sizeof(struct bpf_insn))
4638 		return -EINVAL;
4639 	insn_idx = relo->insn_off / sizeof(struct bpf_insn);
4640 	insn = &prog->insns[insn_idx];
4641 	class = BPF_CLASS(insn->code);
4642 
4643 	if (relo->kind == BPF_FIELD_EXISTS) {
4644 		orig_val = 1; /* can't generate EXISTS relo w/o local field */
4645 		new_val = targ_spec ? 1 : 0;
4646 	} else if (!targ_spec) {
4647 		pr_debug("prog '%s': relo #%d: substituting insn #%d w/ invalid insn\n",
4648 			 bpf_program__title(prog, false), relo_idx, insn_idx);
4649 		insn->code = BPF_JMP | BPF_CALL;
4650 		insn->dst_reg = 0;
4651 		insn->src_reg = 0;
4652 		insn->off = 0;
4653 		/* if this instruction is reachable (not a dead code),
4654 		 * verifier will complain with the following message:
4655 		 * invalid func unknown#195896080
4656 		 */
4657 		insn->imm = 195896080; /* => 0xbad2310 => "bad relo" */
4658 		return 0;
4659 	} else {
4660 		err = bpf_core_calc_field_relo(prog, relo, local_spec,
4661 					       &orig_val, &validate);
4662 		if (err)
4663 			return err;
4664 		err = bpf_core_calc_field_relo(prog, relo, targ_spec,
4665 					       &new_val, NULL);
4666 		if (err)
4667 			return err;
4668 	}
4669 
4670 	switch (class) {
4671 	case BPF_ALU:
4672 	case BPF_ALU64:
4673 		if (BPF_SRC(insn->code) != BPF_K)
4674 			return -EINVAL;
4675 		if (validate && insn->imm != orig_val) {
4676 			pr_warn("prog '%s': relo #%d: unexpected insn #%d (ALU/ALU64) value: got %u, exp %u -> %u\n",
4677 				bpf_program__title(prog, false), relo_idx,
4678 				insn_idx, insn->imm, orig_val, new_val);
4679 			return -EINVAL;
4680 		}
4681 		orig_val = insn->imm;
4682 		insn->imm = new_val;
4683 		pr_debug("prog '%s': relo #%d: patched insn #%d (ALU/ALU64) imm %u -> %u\n",
4684 			 bpf_program__title(prog, false), relo_idx, insn_idx,
4685 			 orig_val, new_val);
4686 		break;
4687 	case BPF_LDX:
4688 	case BPF_ST:
4689 	case BPF_STX:
4690 		if (validate && insn->off != orig_val) {
4691 			pr_warn("prog '%s': relo #%d: unexpected insn #%d (LD/LDX/ST/STX) value: got %u, exp %u -> %u\n",
4692 				bpf_program__title(prog, false), relo_idx,
4693 				insn_idx, insn->off, orig_val, new_val);
4694 			return -EINVAL;
4695 		}
4696 		if (new_val > SHRT_MAX) {
4697 			pr_warn("prog '%s': relo #%d: insn #%d (LDX/ST/STX) value too big: %u\n",
4698 				bpf_program__title(prog, false), relo_idx,
4699 				insn_idx, new_val);
4700 			return -ERANGE;
4701 		}
4702 		orig_val = insn->off;
4703 		insn->off = new_val;
4704 		pr_debug("prog '%s': relo #%d: patched insn #%d (LDX/ST/STX) off %u -> %u\n",
4705 			 bpf_program__title(prog, false), relo_idx, insn_idx,
4706 			 orig_val, new_val);
4707 		break;
4708 	default:
4709 		pr_warn("prog '%s': relo #%d: trying to relocate unrecognized insn #%d, code:%x, src:%x, dst:%x, off:%x, imm:%x\n",
4710 			bpf_program__title(prog, false), relo_idx,
4711 			insn_idx, insn->code, insn->src_reg, insn->dst_reg,
4712 			insn->off, insn->imm);
4713 		return -EINVAL;
4714 	}
4715 
4716 	return 0;
4717 }
4718 
4719 /* Output spec definition in the format:
4720  * [<type-id>] (<type-name>) + <raw-spec> => <offset>@<spec>,
4721  * where <spec> is a C-syntax view of recorded field access, e.g.: x.a[3].b
4722  */
4723 static void bpf_core_dump_spec(int level, const struct bpf_core_spec *spec)
4724 {
4725 	const struct btf_type *t;
4726 	const char *s;
4727 	__u32 type_id;
4728 	int i;
4729 
4730 	type_id = spec->spec[0].type_id;
4731 	t = btf__type_by_id(spec->btf, type_id);
4732 	s = btf__name_by_offset(spec->btf, t->name_off);
4733 	libbpf_print(level, "[%u] %s + ", type_id, s);
4734 
4735 	for (i = 0; i < spec->raw_len; i++)
4736 		libbpf_print(level, "%d%s", spec->raw_spec[i],
4737 			     i == spec->raw_len - 1 ? " => " : ":");
4738 
4739 	libbpf_print(level, "%u.%u @ &x",
4740 		     spec->bit_offset / 8, spec->bit_offset % 8);
4741 
4742 	for (i = 0; i < spec->len; i++) {
4743 		if (spec->spec[i].name)
4744 			libbpf_print(level, ".%s", spec->spec[i].name);
4745 		else
4746 			libbpf_print(level, "[%u]", spec->spec[i].idx);
4747 	}
4748 
4749 }
4750 
4751 static size_t bpf_core_hash_fn(const void *key, void *ctx)
4752 {
4753 	return (size_t)key;
4754 }
4755 
4756 static bool bpf_core_equal_fn(const void *k1, const void *k2, void *ctx)
4757 {
4758 	return k1 == k2;
4759 }
4760 
4761 static void *u32_as_hash_key(__u32 x)
4762 {
4763 	return (void *)(uintptr_t)x;
4764 }
4765 
4766 /*
4767  * CO-RE relocate single instruction.
4768  *
4769  * The outline and important points of the algorithm:
4770  * 1. For given local type, find corresponding candidate target types.
4771  *    Candidate type is a type with the same "essential" name, ignoring
4772  *    everything after last triple underscore (___). E.g., `sample`,
4773  *    `sample___flavor_one`, `sample___flavor_another_one`, are all candidates
4774  *    for each other. Names with triple underscore are referred to as
4775  *    "flavors" and are useful, among other things, to allow to
4776  *    specify/support incompatible variations of the same kernel struct, which
4777  *    might differ between different kernel versions and/or build
4778  *    configurations.
4779  *
4780  *    N.B. Struct "flavors" could be generated by bpftool's BTF-to-C
4781  *    converter, when deduplicated BTF of a kernel still contains more than
4782  *    one different types with the same name. In that case, ___2, ___3, etc
4783  *    are appended starting from second name conflict. But start flavors are
4784  *    also useful to be defined "locally", in BPF program, to extract same
4785  *    data from incompatible changes between different kernel
4786  *    versions/configurations. For instance, to handle field renames between
4787  *    kernel versions, one can use two flavors of the struct name with the
4788  *    same common name and use conditional relocations to extract that field,
4789  *    depending on target kernel version.
4790  * 2. For each candidate type, try to match local specification to this
4791  *    candidate target type. Matching involves finding corresponding
4792  *    high-level spec accessors, meaning that all named fields should match,
4793  *    as well as all array accesses should be within the actual bounds. Also,
4794  *    types should be compatible (see bpf_core_fields_are_compat for details).
4795  * 3. It is supported and expected that there might be multiple flavors
4796  *    matching the spec. As long as all the specs resolve to the same set of
4797  *    offsets across all candidates, there is no error. If there is any
4798  *    ambiguity, CO-RE relocation will fail. This is necessary to accomodate
4799  *    imprefection of BTF deduplication, which can cause slight duplication of
4800  *    the same BTF type, if some directly or indirectly referenced (by
4801  *    pointer) type gets resolved to different actual types in different
4802  *    object files. If such situation occurs, deduplicated BTF will end up
4803  *    with two (or more) structurally identical types, which differ only in
4804  *    types they refer to through pointer. This should be OK in most cases and
4805  *    is not an error.
4806  * 4. Candidate types search is performed by linearly scanning through all
4807  *    types in target BTF. It is anticipated that this is overall more
4808  *    efficient memory-wise and not significantly worse (if not better)
4809  *    CPU-wise compared to prebuilding a map from all local type names to
4810  *    a list of candidate type names. It's also sped up by caching resolved
4811  *    list of matching candidates per each local "root" type ID, that has at
4812  *    least one bpf_field_reloc associated with it. This list is shared
4813  *    between multiple relocations for the same type ID and is updated as some
4814  *    of the candidates are pruned due to structural incompatibility.
4815  */
4816 static int bpf_core_reloc_field(struct bpf_program *prog,
4817 				 const struct bpf_field_reloc *relo,
4818 				 int relo_idx,
4819 				 const struct btf *local_btf,
4820 				 const struct btf *targ_btf,
4821 				 struct hashmap *cand_cache)
4822 {
4823 	const char *prog_name = bpf_program__title(prog, false);
4824 	struct bpf_core_spec local_spec, cand_spec, targ_spec;
4825 	const void *type_key = u32_as_hash_key(relo->type_id);
4826 	const struct btf_type *local_type, *cand_type;
4827 	const char *local_name, *cand_name;
4828 	struct ids_vec *cand_ids;
4829 	__u32 local_id, cand_id;
4830 	const char *spec_str;
4831 	int i, j, err;
4832 
4833 	local_id = relo->type_id;
4834 	local_type = btf__type_by_id(local_btf, local_id);
4835 	if (!local_type)
4836 		return -EINVAL;
4837 
4838 	local_name = btf__name_by_offset(local_btf, local_type->name_off);
4839 	if (str_is_empty(local_name))
4840 		return -EINVAL;
4841 
4842 	spec_str = btf__name_by_offset(local_btf, relo->access_str_off);
4843 	if (str_is_empty(spec_str))
4844 		return -EINVAL;
4845 
4846 	err = bpf_core_spec_parse(local_btf, local_id, spec_str, &local_spec);
4847 	if (err) {
4848 		pr_warn("prog '%s': relo #%d: parsing [%d] %s + %s failed: %d\n",
4849 			prog_name, relo_idx, local_id, local_name, spec_str,
4850 			err);
4851 		return -EINVAL;
4852 	}
4853 
4854 	pr_debug("prog '%s': relo #%d: kind %d, spec is ", prog_name, relo_idx,
4855 		 relo->kind);
4856 	bpf_core_dump_spec(LIBBPF_DEBUG, &local_spec);
4857 	libbpf_print(LIBBPF_DEBUG, "\n");
4858 
4859 	if (!hashmap__find(cand_cache, type_key, (void **)&cand_ids)) {
4860 		cand_ids = bpf_core_find_cands(local_btf, local_id, targ_btf);
4861 		if (IS_ERR(cand_ids)) {
4862 			pr_warn("prog '%s': relo #%d: target candidate search failed for [%d] %s: %ld",
4863 				prog_name, relo_idx, local_id, local_name,
4864 				PTR_ERR(cand_ids));
4865 			return PTR_ERR(cand_ids);
4866 		}
4867 		err = hashmap__set(cand_cache, type_key, cand_ids, NULL, NULL);
4868 		if (err) {
4869 			bpf_core_free_cands(cand_ids);
4870 			return err;
4871 		}
4872 	}
4873 
4874 	for (i = 0, j = 0; i < cand_ids->len; i++) {
4875 		cand_id = cand_ids->data[i];
4876 		cand_type = btf__type_by_id(targ_btf, cand_id);
4877 		cand_name = btf__name_by_offset(targ_btf, cand_type->name_off);
4878 
4879 		err = bpf_core_spec_match(&local_spec, targ_btf,
4880 					  cand_id, &cand_spec);
4881 		pr_debug("prog '%s': relo #%d: matching candidate #%d %s against spec ",
4882 			 prog_name, relo_idx, i, cand_name);
4883 		bpf_core_dump_spec(LIBBPF_DEBUG, &cand_spec);
4884 		libbpf_print(LIBBPF_DEBUG, ": %d\n", err);
4885 		if (err < 0) {
4886 			pr_warn("prog '%s': relo #%d: matching error: %d\n",
4887 				prog_name, relo_idx, err);
4888 			return err;
4889 		}
4890 		if (err == 0)
4891 			continue;
4892 
4893 		if (j == 0) {
4894 			targ_spec = cand_spec;
4895 		} else if (cand_spec.bit_offset != targ_spec.bit_offset) {
4896 			/* if there are many candidates, they should all
4897 			 * resolve to the same bit offset
4898 			 */
4899 			pr_warn("prog '%s': relo #%d: offset ambiguity: %u != %u\n",
4900 				prog_name, relo_idx, cand_spec.bit_offset,
4901 				targ_spec.bit_offset);
4902 			return -EINVAL;
4903 		}
4904 
4905 		cand_ids->data[j++] = cand_spec.spec[0].type_id;
4906 	}
4907 
4908 	/*
4909 	 * For BPF_FIELD_EXISTS relo or when used BPF program has field
4910 	 * existence checks or kernel version/config checks, it's expected
4911 	 * that we might not find any candidates. In this case, if field
4912 	 * wasn't found in any candidate, the list of candidates shouldn't
4913 	 * change at all, we'll just handle relocating appropriately,
4914 	 * depending on relo's kind.
4915 	 */
4916 	if (j > 0)
4917 		cand_ids->len = j;
4918 
4919 	/*
4920 	 * If no candidates were found, it might be both a programmer error,
4921 	 * as well as expected case, depending whether instruction w/
4922 	 * relocation is guarded in some way that makes it unreachable (dead
4923 	 * code) if relocation can't be resolved. This is handled in
4924 	 * bpf_core_reloc_insn() uniformly by replacing that instruction with
4925 	 * BPF helper call insn (using invalid helper ID). If that instruction
4926 	 * is indeed unreachable, then it will be ignored and eliminated by
4927 	 * verifier. If it was an error, then verifier will complain and point
4928 	 * to a specific instruction number in its log.
4929 	 */
4930 	if (j == 0)
4931 		pr_debug("prog '%s': relo #%d: no matching targets found for [%d] %s + %s\n",
4932 			 prog_name, relo_idx, local_id, local_name, spec_str);
4933 
4934 	/* bpf_core_reloc_insn should know how to handle missing targ_spec */
4935 	err = bpf_core_reloc_insn(prog, relo, relo_idx, &local_spec,
4936 				  j ? &targ_spec : NULL);
4937 	if (err) {
4938 		pr_warn("prog '%s': relo #%d: failed to patch insn at offset %d: %d\n",
4939 			prog_name, relo_idx, relo->insn_off, err);
4940 		return -EINVAL;
4941 	}
4942 
4943 	return 0;
4944 }
4945 
4946 static int
4947 bpf_core_reloc_fields(struct bpf_object *obj, const char *targ_btf_path)
4948 {
4949 	const struct btf_ext_info_sec *sec;
4950 	const struct bpf_field_reloc *rec;
4951 	const struct btf_ext_info *seg;
4952 	struct hashmap_entry *entry;
4953 	struct hashmap *cand_cache = NULL;
4954 	struct bpf_program *prog;
4955 	struct btf *targ_btf;
4956 	const char *sec_name;
4957 	int i, err = 0;
4958 
4959 	if (targ_btf_path)
4960 		targ_btf = btf__parse_elf(targ_btf_path, NULL);
4961 	else
4962 		targ_btf = obj->btf_vmlinux;
4963 	if (IS_ERR_OR_NULL(targ_btf)) {
4964 		pr_warn("failed to get target BTF: %ld\n", PTR_ERR(targ_btf));
4965 		return PTR_ERR(targ_btf);
4966 	}
4967 
4968 	cand_cache = hashmap__new(bpf_core_hash_fn, bpf_core_equal_fn, NULL);
4969 	if (IS_ERR(cand_cache)) {
4970 		err = PTR_ERR(cand_cache);
4971 		goto out;
4972 	}
4973 
4974 	seg = &obj->btf_ext->field_reloc_info;
4975 	for_each_btf_ext_sec(seg, sec) {
4976 		sec_name = btf__name_by_offset(obj->btf, sec->sec_name_off);
4977 		if (str_is_empty(sec_name)) {
4978 			err = -EINVAL;
4979 			goto out;
4980 		}
4981 		prog = NULL;
4982 		for (i = 0; i < obj->nr_programs; i++) {
4983 			if (!strcmp(obj->programs[i].section_name, sec_name)) {
4984 				prog = &obj->programs[i];
4985 				break;
4986 			}
4987 		}
4988 		if (!prog) {
4989 			pr_warn("failed to find program '%s' for CO-RE offset relocation\n",
4990 				sec_name);
4991 			err = -EINVAL;
4992 			goto out;
4993 		}
4994 
4995 		pr_debug("prog '%s': performing %d CO-RE offset relocs\n",
4996 			 sec_name, sec->num_info);
4997 
4998 		for_each_btf_ext_rec(seg, sec, i, rec) {
4999 			err = bpf_core_reloc_field(prog, rec, i, obj->btf,
5000 						   targ_btf, cand_cache);
5001 			if (err) {
5002 				pr_warn("prog '%s': relo #%d: failed to relocate: %d\n",
5003 					sec_name, i, err);
5004 				goto out;
5005 			}
5006 		}
5007 	}
5008 
5009 out:
5010 	/* obj->btf_vmlinux is freed at the end of object load phase */
5011 	if (targ_btf != obj->btf_vmlinux)
5012 		btf__free(targ_btf);
5013 	if (!IS_ERR_OR_NULL(cand_cache)) {
5014 		hashmap__for_each_entry(cand_cache, entry, i) {
5015 			bpf_core_free_cands(entry->value);
5016 		}
5017 		hashmap__free(cand_cache);
5018 	}
5019 	return err;
5020 }
5021 
5022 static int
5023 bpf_object__relocate_core(struct bpf_object *obj, const char *targ_btf_path)
5024 {
5025 	int err = 0;
5026 
5027 	if (obj->btf_ext->field_reloc_info.len)
5028 		err = bpf_core_reloc_fields(obj, targ_btf_path);
5029 
5030 	return err;
5031 }
5032 
5033 static int
5034 bpf_program__reloc_text(struct bpf_program *prog, struct bpf_object *obj,
5035 			struct reloc_desc *relo)
5036 {
5037 	struct bpf_insn *insn, *new_insn;
5038 	struct bpf_program *text;
5039 	size_t new_cnt;
5040 	int err;
5041 
5042 	if (prog->idx != obj->efile.text_shndx && prog->main_prog_cnt == 0) {
5043 		text = bpf_object__find_prog_by_idx(obj, obj->efile.text_shndx);
5044 		if (!text) {
5045 			pr_warn("no .text section found yet relo into text exist\n");
5046 			return -LIBBPF_ERRNO__RELOC;
5047 		}
5048 		new_cnt = prog->insns_cnt + text->insns_cnt;
5049 		new_insn = reallocarray(prog->insns, new_cnt, sizeof(*insn));
5050 		if (!new_insn) {
5051 			pr_warn("oom in prog realloc\n");
5052 			return -ENOMEM;
5053 		}
5054 		prog->insns = new_insn;
5055 
5056 		if (obj->btf_ext) {
5057 			err = bpf_program_reloc_btf_ext(prog, obj,
5058 							text->section_name,
5059 							prog->insns_cnt);
5060 			if (err)
5061 				return err;
5062 		}
5063 
5064 		memcpy(new_insn + prog->insns_cnt, text->insns,
5065 		       text->insns_cnt * sizeof(*insn));
5066 		prog->main_prog_cnt = prog->insns_cnt;
5067 		prog->insns_cnt = new_cnt;
5068 		pr_debug("added %zd insn from %s to prog %s\n",
5069 			 text->insns_cnt, text->section_name,
5070 			 prog->section_name);
5071 	}
5072 
5073 	insn = &prog->insns[relo->insn_idx];
5074 	insn->imm += relo->sym_off / 8 + prog->main_prog_cnt - relo->insn_idx;
5075 	return 0;
5076 }
5077 
5078 static int
5079 bpf_program__relocate(struct bpf_program *prog, struct bpf_object *obj)
5080 {
5081 	int i, err;
5082 
5083 	if (!prog)
5084 		return 0;
5085 
5086 	if (obj->btf_ext) {
5087 		err = bpf_program_reloc_btf_ext(prog, obj,
5088 						prog->section_name, 0);
5089 		if (err)
5090 			return err;
5091 	}
5092 
5093 	if (!prog->reloc_desc)
5094 		return 0;
5095 
5096 	for (i = 0; i < prog->nr_reloc; i++) {
5097 		struct reloc_desc *relo = &prog->reloc_desc[i];
5098 		struct bpf_insn *insn = &prog->insns[relo->insn_idx];
5099 		struct extern_desc *ext;
5100 
5101 		if (relo->insn_idx + 1 >= (int)prog->insns_cnt) {
5102 			pr_warn("relocation out of range: '%s'\n",
5103 				prog->section_name);
5104 			return -LIBBPF_ERRNO__RELOC;
5105 		}
5106 
5107 		switch (relo->type) {
5108 		case RELO_LD64:
5109 			insn[0].src_reg = BPF_PSEUDO_MAP_FD;
5110 			insn[0].imm = obj->maps[relo->map_idx].fd;
5111 			break;
5112 		case RELO_DATA:
5113 			insn[0].src_reg = BPF_PSEUDO_MAP_VALUE;
5114 			insn[1].imm = insn[0].imm + relo->sym_off;
5115 			insn[0].imm = obj->maps[relo->map_idx].fd;
5116 			break;
5117 		case RELO_EXTERN:
5118 			ext = &obj->externs[relo->sym_off];
5119 			if (ext->type == EXT_KCFG) {
5120 				insn[0].src_reg = BPF_PSEUDO_MAP_VALUE;
5121 				insn[0].imm = obj->maps[obj->kconfig_map_idx].fd;
5122 				insn[1].imm = ext->kcfg.data_off;
5123 			} else /* EXT_KSYM */ {
5124 				insn[0].imm = (__u32)ext->ksym.addr;
5125 				insn[1].imm = ext->ksym.addr >> 32;
5126 			}
5127 			break;
5128 		case RELO_CALL:
5129 			err = bpf_program__reloc_text(prog, obj, relo);
5130 			if (err)
5131 				return err;
5132 			break;
5133 		default:
5134 			pr_warn("relo #%d: bad relo type %d\n", i, relo->type);
5135 			return -EINVAL;
5136 		}
5137 	}
5138 
5139 	zfree(&prog->reloc_desc);
5140 	prog->nr_reloc = 0;
5141 	return 0;
5142 }
5143 
5144 static int
5145 bpf_object__relocate(struct bpf_object *obj, const char *targ_btf_path)
5146 {
5147 	struct bpf_program *prog;
5148 	size_t i;
5149 	int err;
5150 
5151 	if (obj->btf_ext) {
5152 		err = bpf_object__relocate_core(obj, targ_btf_path);
5153 		if (err) {
5154 			pr_warn("failed to perform CO-RE relocations: %d\n",
5155 				err);
5156 			return err;
5157 		}
5158 	}
5159 	/* ensure .text is relocated first, as it's going to be copied as-is
5160 	 * later for sub-program calls
5161 	 */
5162 	for (i = 0; i < obj->nr_programs; i++) {
5163 		prog = &obj->programs[i];
5164 		if (prog->idx != obj->efile.text_shndx)
5165 			continue;
5166 
5167 		err = bpf_program__relocate(prog, obj);
5168 		if (err) {
5169 			pr_warn("failed to relocate '%s'\n", prog->section_name);
5170 			return err;
5171 		}
5172 		break;
5173 	}
5174 	/* now relocate everything but .text, which by now is relocated
5175 	 * properly, so we can copy raw sub-program instructions as is safely
5176 	 */
5177 	for (i = 0; i < obj->nr_programs; i++) {
5178 		prog = &obj->programs[i];
5179 		if (prog->idx == obj->efile.text_shndx)
5180 			continue;
5181 
5182 		err = bpf_program__relocate(prog, obj);
5183 		if (err) {
5184 			pr_warn("failed to relocate '%s'\n", prog->section_name);
5185 			return err;
5186 		}
5187 	}
5188 	return 0;
5189 }
5190 
5191 static int bpf_object__collect_st_ops_relos(struct bpf_object *obj,
5192 					    GElf_Shdr *shdr, Elf_Data *data);
5193 
5194 static int bpf_object__collect_map_relos(struct bpf_object *obj,
5195 					 GElf_Shdr *shdr, Elf_Data *data)
5196 {
5197 	int i, j, nrels, new_sz, ptr_sz = sizeof(void *);
5198 	const struct btf_var_secinfo *vi = NULL;
5199 	const struct btf_type *sec, *var, *def;
5200 	const struct btf_member *member;
5201 	struct bpf_map *map, *targ_map;
5202 	const char *name, *mname;
5203 	Elf_Data *symbols;
5204 	unsigned int moff;
5205 	GElf_Sym sym;
5206 	GElf_Rel rel;
5207 	void *tmp;
5208 
5209 	if (!obj->efile.btf_maps_sec_btf_id || !obj->btf)
5210 		return -EINVAL;
5211 	sec = btf__type_by_id(obj->btf, obj->efile.btf_maps_sec_btf_id);
5212 	if (!sec)
5213 		return -EINVAL;
5214 
5215 	symbols = obj->efile.symbols;
5216 	nrels = shdr->sh_size / shdr->sh_entsize;
5217 	for (i = 0; i < nrels; i++) {
5218 		if (!gelf_getrel(data, i, &rel)) {
5219 			pr_warn(".maps relo #%d: failed to get ELF relo\n", i);
5220 			return -LIBBPF_ERRNO__FORMAT;
5221 		}
5222 		if (!gelf_getsym(symbols, GELF_R_SYM(rel.r_info), &sym)) {
5223 			pr_warn(".maps relo #%d: symbol %zx not found\n",
5224 				i, (size_t)GELF_R_SYM(rel.r_info));
5225 			return -LIBBPF_ERRNO__FORMAT;
5226 		}
5227 		name = elf_strptr(obj->efile.elf, obj->efile.strtabidx,
5228 				  sym.st_name) ? : "<?>";
5229 		if (sym.st_shndx != obj->efile.btf_maps_shndx) {
5230 			pr_warn(".maps relo #%d: '%s' isn't a BTF-defined map\n",
5231 				i, name);
5232 			return -LIBBPF_ERRNO__RELOC;
5233 		}
5234 
5235 		pr_debug(".maps relo #%d: for %zd value %zd rel.r_offset %zu name %d ('%s')\n",
5236 			 i, (ssize_t)(rel.r_info >> 32), (size_t)sym.st_value,
5237 			 (size_t)rel.r_offset, sym.st_name, name);
5238 
5239 		for (j = 0; j < obj->nr_maps; j++) {
5240 			map = &obj->maps[j];
5241 			if (map->sec_idx != obj->efile.btf_maps_shndx)
5242 				continue;
5243 
5244 			vi = btf_var_secinfos(sec) + map->btf_var_idx;
5245 			if (vi->offset <= rel.r_offset &&
5246 			    rel.r_offset + sizeof(void *) <= vi->offset + vi->size)
5247 				break;
5248 		}
5249 		if (j == obj->nr_maps) {
5250 			pr_warn(".maps relo #%d: cannot find map '%s' at rel.r_offset %zu\n",
5251 				i, name, (size_t)rel.r_offset);
5252 			return -EINVAL;
5253 		}
5254 
5255 		if (!bpf_map_type__is_map_in_map(map->def.type))
5256 			return -EINVAL;
5257 		if (map->def.type == BPF_MAP_TYPE_HASH_OF_MAPS &&
5258 		    map->def.key_size != sizeof(int)) {
5259 			pr_warn(".maps relo #%d: hash-of-maps '%s' should have key size %zu.\n",
5260 				i, map->name, sizeof(int));
5261 			return -EINVAL;
5262 		}
5263 
5264 		targ_map = bpf_object__find_map_by_name(obj, name);
5265 		if (!targ_map)
5266 			return -ESRCH;
5267 
5268 		var = btf__type_by_id(obj->btf, vi->type);
5269 		def = skip_mods_and_typedefs(obj->btf, var->type, NULL);
5270 		if (btf_vlen(def) == 0)
5271 			return -EINVAL;
5272 		member = btf_members(def) + btf_vlen(def) - 1;
5273 		mname = btf__name_by_offset(obj->btf, member->name_off);
5274 		if (strcmp(mname, "values"))
5275 			return -EINVAL;
5276 
5277 		moff = btf_member_bit_offset(def, btf_vlen(def) - 1) / 8;
5278 		if (rel.r_offset - vi->offset < moff)
5279 			return -EINVAL;
5280 
5281 		moff = rel.r_offset - vi->offset - moff;
5282 		if (moff % ptr_sz)
5283 			return -EINVAL;
5284 		moff /= ptr_sz;
5285 		if (moff >= map->init_slots_sz) {
5286 			new_sz = moff + 1;
5287 			tmp = realloc(map->init_slots, new_sz * ptr_sz);
5288 			if (!tmp)
5289 				return -ENOMEM;
5290 			map->init_slots = tmp;
5291 			memset(map->init_slots + map->init_slots_sz, 0,
5292 			       (new_sz - map->init_slots_sz) * ptr_sz);
5293 			map->init_slots_sz = new_sz;
5294 		}
5295 		map->init_slots[moff] = targ_map;
5296 
5297 		pr_debug(".maps relo #%d: map '%s' slot [%d] points to map '%s'\n",
5298 			 i, map->name, moff, name);
5299 	}
5300 
5301 	return 0;
5302 }
5303 
5304 static int bpf_object__collect_reloc(struct bpf_object *obj)
5305 {
5306 	int i, err;
5307 
5308 	if (!obj_elf_valid(obj)) {
5309 		pr_warn("Internal error: elf object is closed\n");
5310 		return -LIBBPF_ERRNO__INTERNAL;
5311 	}
5312 
5313 	for (i = 0; i < obj->efile.nr_reloc_sects; i++) {
5314 		GElf_Shdr *shdr = &obj->efile.reloc_sects[i].shdr;
5315 		Elf_Data *data = obj->efile.reloc_sects[i].data;
5316 		int idx = shdr->sh_info;
5317 		struct bpf_program *prog;
5318 
5319 		if (shdr->sh_type != SHT_REL) {
5320 			pr_warn("internal error at %d\n", __LINE__);
5321 			return -LIBBPF_ERRNO__INTERNAL;
5322 		}
5323 
5324 		if (idx == obj->efile.st_ops_shndx) {
5325 			err = bpf_object__collect_st_ops_relos(obj, shdr, data);
5326 		} else if (idx == obj->efile.btf_maps_shndx) {
5327 			err = bpf_object__collect_map_relos(obj, shdr, data);
5328 		} else {
5329 			prog = bpf_object__find_prog_by_idx(obj, idx);
5330 			if (!prog) {
5331 				pr_warn("relocation failed: no prog in section(%d)\n", idx);
5332 				return -LIBBPF_ERRNO__RELOC;
5333 			}
5334 			err = bpf_program__collect_reloc(prog, shdr, data, obj);
5335 		}
5336 		if (err)
5337 			return err;
5338 	}
5339 	return 0;
5340 }
5341 
5342 static int
5343 load_program(struct bpf_program *prog, struct bpf_insn *insns, int insns_cnt,
5344 	     char *license, __u32 kern_version, int *pfd)
5345 {
5346 	struct bpf_load_program_attr load_attr;
5347 	char *cp, errmsg[STRERR_BUFSIZE];
5348 	size_t log_buf_size = 0;
5349 	char *log_buf = NULL;
5350 	int btf_fd, ret;
5351 
5352 	if (!insns || !insns_cnt)
5353 		return -EINVAL;
5354 
5355 	memset(&load_attr, 0, sizeof(struct bpf_load_program_attr));
5356 	load_attr.prog_type = prog->type;
5357 	/* old kernels might not support specifying expected_attach_type */
5358 	if (!prog->caps->exp_attach_type && prog->sec_def &&
5359 	    prog->sec_def->is_exp_attach_type_optional)
5360 		load_attr.expected_attach_type = 0;
5361 	else
5362 		load_attr.expected_attach_type = prog->expected_attach_type;
5363 	if (prog->caps->name)
5364 		load_attr.name = prog->name;
5365 	load_attr.insns = insns;
5366 	load_attr.insns_cnt = insns_cnt;
5367 	load_attr.license = license;
5368 	if (prog->type == BPF_PROG_TYPE_STRUCT_OPS ||
5369 	    prog->type == BPF_PROG_TYPE_LSM) {
5370 		load_attr.attach_btf_id = prog->attach_btf_id;
5371 	} else if (prog->type == BPF_PROG_TYPE_TRACING ||
5372 		   prog->type == BPF_PROG_TYPE_EXT) {
5373 		load_attr.attach_prog_fd = prog->attach_prog_fd;
5374 		load_attr.attach_btf_id = prog->attach_btf_id;
5375 	} else {
5376 		load_attr.kern_version = kern_version;
5377 		load_attr.prog_ifindex = prog->prog_ifindex;
5378 	}
5379 	/* specify func_info/line_info only if kernel supports them */
5380 	btf_fd = bpf_object__btf_fd(prog->obj);
5381 	if (btf_fd >= 0 && prog->obj->caps.btf_func) {
5382 		load_attr.prog_btf_fd = btf_fd;
5383 		load_attr.func_info = prog->func_info;
5384 		load_attr.func_info_rec_size = prog->func_info_rec_size;
5385 		load_attr.func_info_cnt = prog->func_info_cnt;
5386 		load_attr.line_info = prog->line_info;
5387 		load_attr.line_info_rec_size = prog->line_info_rec_size;
5388 		load_attr.line_info_cnt = prog->line_info_cnt;
5389 	}
5390 	load_attr.log_level = prog->log_level;
5391 	load_attr.prog_flags = prog->prog_flags;
5392 
5393 retry_load:
5394 	if (log_buf_size) {
5395 		log_buf = malloc(log_buf_size);
5396 		if (!log_buf)
5397 			return -ENOMEM;
5398 
5399 		*log_buf = 0;
5400 	}
5401 
5402 	ret = bpf_load_program_xattr(&load_attr, log_buf, log_buf_size);
5403 
5404 	if (ret >= 0) {
5405 		if (log_buf && load_attr.log_level)
5406 			pr_debug("verifier log:\n%s", log_buf);
5407 		*pfd = ret;
5408 		ret = 0;
5409 		goto out;
5410 	}
5411 
5412 	if (!log_buf || errno == ENOSPC) {
5413 		log_buf_size = max((size_t)BPF_LOG_BUF_SIZE,
5414 				   log_buf_size << 1);
5415 
5416 		free(log_buf);
5417 		goto retry_load;
5418 	}
5419 	ret = -errno;
5420 	cp = libbpf_strerror_r(errno, errmsg, sizeof(errmsg));
5421 	pr_warn("load bpf program failed: %s\n", cp);
5422 	pr_perm_msg(ret);
5423 
5424 	if (log_buf && log_buf[0] != '\0') {
5425 		ret = -LIBBPF_ERRNO__VERIFY;
5426 		pr_warn("-- BEGIN DUMP LOG ---\n");
5427 		pr_warn("\n%s\n", log_buf);
5428 		pr_warn("-- END LOG --\n");
5429 	} else if (load_attr.insns_cnt >= BPF_MAXINSNS) {
5430 		pr_warn("Program too large (%zu insns), at most %d insns\n",
5431 			load_attr.insns_cnt, BPF_MAXINSNS);
5432 		ret = -LIBBPF_ERRNO__PROG2BIG;
5433 	} else if (load_attr.prog_type != BPF_PROG_TYPE_KPROBE) {
5434 		/* Wrong program type? */
5435 		int fd;
5436 
5437 		load_attr.prog_type = BPF_PROG_TYPE_KPROBE;
5438 		load_attr.expected_attach_type = 0;
5439 		fd = bpf_load_program_xattr(&load_attr, NULL, 0);
5440 		if (fd >= 0) {
5441 			close(fd);
5442 			ret = -LIBBPF_ERRNO__PROGTYPE;
5443 			goto out;
5444 		}
5445 	}
5446 
5447 out:
5448 	free(log_buf);
5449 	return ret;
5450 }
5451 
5452 static int libbpf_find_attach_btf_id(struct bpf_program *prog);
5453 
5454 int bpf_program__load(struct bpf_program *prog, char *license, __u32 kern_ver)
5455 {
5456 	int err = 0, fd, i, btf_id;
5457 
5458 	if (prog->obj->loaded) {
5459 		pr_warn("prog '%s'('%s'): can't load after object was loaded\n",
5460 			prog->name, prog->section_name);
5461 		return -EINVAL;
5462 	}
5463 
5464 	if ((prog->type == BPF_PROG_TYPE_TRACING ||
5465 	     prog->type == BPF_PROG_TYPE_LSM ||
5466 	     prog->type == BPF_PROG_TYPE_EXT) && !prog->attach_btf_id) {
5467 		btf_id = libbpf_find_attach_btf_id(prog);
5468 		if (btf_id <= 0)
5469 			return btf_id;
5470 		prog->attach_btf_id = btf_id;
5471 	}
5472 
5473 	if (prog->instances.nr < 0 || !prog->instances.fds) {
5474 		if (prog->preprocessor) {
5475 			pr_warn("Internal error: can't load program '%s'\n",
5476 				prog->section_name);
5477 			return -LIBBPF_ERRNO__INTERNAL;
5478 		}
5479 
5480 		prog->instances.fds = malloc(sizeof(int));
5481 		if (!prog->instances.fds) {
5482 			pr_warn("Not enough memory for BPF fds\n");
5483 			return -ENOMEM;
5484 		}
5485 		prog->instances.nr = 1;
5486 		prog->instances.fds[0] = -1;
5487 	}
5488 
5489 	if (!prog->preprocessor) {
5490 		if (prog->instances.nr != 1) {
5491 			pr_warn("Program '%s' is inconsistent: nr(%d) != 1\n",
5492 				prog->section_name, prog->instances.nr);
5493 		}
5494 		err = load_program(prog, prog->insns, prog->insns_cnt,
5495 				   license, kern_ver, &fd);
5496 		if (!err)
5497 			prog->instances.fds[0] = fd;
5498 		goto out;
5499 	}
5500 
5501 	for (i = 0; i < prog->instances.nr; i++) {
5502 		struct bpf_prog_prep_result result;
5503 		bpf_program_prep_t preprocessor = prog->preprocessor;
5504 
5505 		memset(&result, 0, sizeof(result));
5506 		err = preprocessor(prog, i, prog->insns,
5507 				   prog->insns_cnt, &result);
5508 		if (err) {
5509 			pr_warn("Preprocessing the %dth instance of program '%s' failed\n",
5510 				i, prog->section_name);
5511 			goto out;
5512 		}
5513 
5514 		if (!result.new_insn_ptr || !result.new_insn_cnt) {
5515 			pr_debug("Skip loading the %dth instance of program '%s'\n",
5516 				 i, prog->section_name);
5517 			prog->instances.fds[i] = -1;
5518 			if (result.pfd)
5519 				*result.pfd = -1;
5520 			continue;
5521 		}
5522 
5523 		err = load_program(prog, result.new_insn_ptr,
5524 				   result.new_insn_cnt, license, kern_ver, &fd);
5525 		if (err) {
5526 			pr_warn("Loading the %dth instance of program '%s' failed\n",
5527 				i, prog->section_name);
5528 			goto out;
5529 		}
5530 
5531 		if (result.pfd)
5532 			*result.pfd = fd;
5533 		prog->instances.fds[i] = fd;
5534 	}
5535 out:
5536 	if (err)
5537 		pr_warn("failed to load program '%s'\n", prog->section_name);
5538 	zfree(&prog->insns);
5539 	prog->insns_cnt = 0;
5540 	return err;
5541 }
5542 
5543 static bool bpf_program__is_function_storage(const struct bpf_program *prog,
5544 					     const struct bpf_object *obj)
5545 {
5546 	return prog->idx == obj->efile.text_shndx && obj->has_pseudo_calls;
5547 }
5548 
5549 static int
5550 bpf_object__load_progs(struct bpf_object *obj, int log_level)
5551 {
5552 	struct bpf_program *prog;
5553 	size_t i;
5554 	int err;
5555 
5556 	for (i = 0; i < obj->nr_programs; i++) {
5557 		prog = &obj->programs[i];
5558 		if (bpf_program__is_function_storage(prog, obj))
5559 			continue;
5560 		if (!prog->load) {
5561 			pr_debug("prog '%s'('%s'): skipped loading\n",
5562 				 prog->name, prog->section_name);
5563 			continue;
5564 		}
5565 		prog->log_level |= log_level;
5566 		err = bpf_program__load(prog, obj->license, obj->kern_version);
5567 		if (err)
5568 			return err;
5569 	}
5570 	return 0;
5571 }
5572 
5573 static const struct bpf_sec_def *find_sec_def(const char *sec_name);
5574 
5575 static struct bpf_object *
5576 __bpf_object__open(const char *path, const void *obj_buf, size_t obj_buf_sz,
5577 		   const struct bpf_object_open_opts *opts)
5578 {
5579 	const char *obj_name, *kconfig;
5580 	struct bpf_program *prog;
5581 	struct bpf_object *obj;
5582 	char tmp_name[64];
5583 	int err;
5584 
5585 	if (elf_version(EV_CURRENT) == EV_NONE) {
5586 		pr_warn("failed to init libelf for %s\n",
5587 			path ? : "(mem buf)");
5588 		return ERR_PTR(-LIBBPF_ERRNO__LIBELF);
5589 	}
5590 
5591 	if (!OPTS_VALID(opts, bpf_object_open_opts))
5592 		return ERR_PTR(-EINVAL);
5593 
5594 	obj_name = OPTS_GET(opts, object_name, NULL);
5595 	if (obj_buf) {
5596 		if (!obj_name) {
5597 			snprintf(tmp_name, sizeof(tmp_name), "%lx-%lx",
5598 				 (unsigned long)obj_buf,
5599 				 (unsigned long)obj_buf_sz);
5600 			obj_name = tmp_name;
5601 		}
5602 		path = obj_name;
5603 		pr_debug("loading object '%s' from buffer\n", obj_name);
5604 	}
5605 
5606 	obj = bpf_object__new(path, obj_buf, obj_buf_sz, obj_name);
5607 	if (IS_ERR(obj))
5608 		return obj;
5609 
5610 	kconfig = OPTS_GET(opts, kconfig, NULL);
5611 	if (kconfig) {
5612 		obj->kconfig = strdup(kconfig);
5613 		if (!obj->kconfig)
5614 			return ERR_PTR(-ENOMEM);
5615 	}
5616 
5617 	err = bpf_object__elf_init(obj);
5618 	err = err ? : bpf_object__check_endianness(obj);
5619 	err = err ? : bpf_object__elf_collect(obj);
5620 	err = err ? : bpf_object__collect_externs(obj);
5621 	err = err ? : bpf_object__finalize_btf(obj);
5622 	err = err ? : bpf_object__init_maps(obj, opts);
5623 	err = err ? : bpf_object__init_prog_names(obj);
5624 	err = err ? : bpf_object__collect_reloc(obj);
5625 	if (err)
5626 		goto out;
5627 	bpf_object__elf_finish(obj);
5628 
5629 	bpf_object__for_each_program(prog, obj) {
5630 		prog->sec_def = find_sec_def(prog->section_name);
5631 		if (!prog->sec_def)
5632 			/* couldn't guess, but user might manually specify */
5633 			continue;
5634 
5635 		bpf_program__set_type(prog, prog->sec_def->prog_type);
5636 		bpf_program__set_expected_attach_type(prog,
5637 				prog->sec_def->expected_attach_type);
5638 
5639 		if (prog->sec_def->prog_type == BPF_PROG_TYPE_TRACING ||
5640 		    prog->sec_def->prog_type == BPF_PROG_TYPE_EXT)
5641 			prog->attach_prog_fd = OPTS_GET(opts, attach_prog_fd, 0);
5642 	}
5643 
5644 	return obj;
5645 out:
5646 	bpf_object__close(obj);
5647 	return ERR_PTR(err);
5648 }
5649 
5650 static struct bpf_object *
5651 __bpf_object__open_xattr(struct bpf_object_open_attr *attr, int flags)
5652 {
5653 	DECLARE_LIBBPF_OPTS(bpf_object_open_opts, opts,
5654 		.relaxed_maps = flags & MAPS_RELAX_COMPAT,
5655 	);
5656 
5657 	/* param validation */
5658 	if (!attr->file)
5659 		return NULL;
5660 
5661 	pr_debug("loading %s\n", attr->file);
5662 	return __bpf_object__open(attr->file, NULL, 0, &opts);
5663 }
5664 
5665 struct bpf_object *bpf_object__open_xattr(struct bpf_object_open_attr *attr)
5666 {
5667 	return __bpf_object__open_xattr(attr, 0);
5668 }
5669 
5670 struct bpf_object *bpf_object__open(const char *path)
5671 {
5672 	struct bpf_object_open_attr attr = {
5673 		.file		= path,
5674 		.prog_type	= BPF_PROG_TYPE_UNSPEC,
5675 	};
5676 
5677 	return bpf_object__open_xattr(&attr);
5678 }
5679 
5680 struct bpf_object *
5681 bpf_object__open_file(const char *path, const struct bpf_object_open_opts *opts)
5682 {
5683 	if (!path)
5684 		return ERR_PTR(-EINVAL);
5685 
5686 	pr_debug("loading %s\n", path);
5687 
5688 	return __bpf_object__open(path, NULL, 0, opts);
5689 }
5690 
5691 struct bpf_object *
5692 bpf_object__open_mem(const void *obj_buf, size_t obj_buf_sz,
5693 		     const struct bpf_object_open_opts *opts)
5694 {
5695 	if (!obj_buf || obj_buf_sz == 0)
5696 		return ERR_PTR(-EINVAL);
5697 
5698 	return __bpf_object__open(NULL, obj_buf, obj_buf_sz, opts);
5699 }
5700 
5701 struct bpf_object *
5702 bpf_object__open_buffer(const void *obj_buf, size_t obj_buf_sz,
5703 			const char *name)
5704 {
5705 	DECLARE_LIBBPF_OPTS(bpf_object_open_opts, opts,
5706 		.object_name = name,
5707 		/* wrong default, but backwards-compatible */
5708 		.relaxed_maps = true,
5709 	);
5710 
5711 	/* returning NULL is wrong, but backwards-compatible */
5712 	if (!obj_buf || obj_buf_sz == 0)
5713 		return NULL;
5714 
5715 	return bpf_object__open_mem(obj_buf, obj_buf_sz, &opts);
5716 }
5717 
5718 int bpf_object__unload(struct bpf_object *obj)
5719 {
5720 	size_t i;
5721 
5722 	if (!obj)
5723 		return -EINVAL;
5724 
5725 	for (i = 0; i < obj->nr_maps; i++) {
5726 		zclose(obj->maps[i].fd);
5727 		if (obj->maps[i].st_ops)
5728 			zfree(&obj->maps[i].st_ops->kern_vdata);
5729 	}
5730 
5731 	for (i = 0; i < obj->nr_programs; i++)
5732 		bpf_program__unload(&obj->programs[i]);
5733 
5734 	return 0;
5735 }
5736 
5737 static int bpf_object__sanitize_maps(struct bpf_object *obj)
5738 {
5739 	struct bpf_map *m;
5740 
5741 	bpf_object__for_each_map(m, obj) {
5742 		if (!bpf_map__is_internal(m))
5743 			continue;
5744 		if (!obj->caps.global_data) {
5745 			pr_warn("kernel doesn't support global data\n");
5746 			return -ENOTSUP;
5747 		}
5748 		if (!obj->caps.array_mmap)
5749 			m->def.map_flags ^= BPF_F_MMAPABLE;
5750 	}
5751 
5752 	return 0;
5753 }
5754 
5755 static int bpf_object__read_kallsyms_file(struct bpf_object *obj)
5756 {
5757 	char sym_type, sym_name[500];
5758 	unsigned long long sym_addr;
5759 	struct extern_desc *ext;
5760 	int ret, err = 0;
5761 	FILE *f;
5762 
5763 	f = fopen("/proc/kallsyms", "r");
5764 	if (!f) {
5765 		err = -errno;
5766 		pr_warn("failed to open /proc/kallsyms: %d\n", err);
5767 		return err;
5768 	}
5769 
5770 	while (true) {
5771 		ret = fscanf(f, "%llx %c %499s%*[^\n]\n",
5772 			     &sym_addr, &sym_type, sym_name);
5773 		if (ret == EOF && feof(f))
5774 			break;
5775 		if (ret != 3) {
5776 			pr_warn("failed to read kallsyms entry: %d\n", ret);
5777 			err = -EINVAL;
5778 			goto out;
5779 		}
5780 
5781 		ext = find_extern_by_name(obj, sym_name);
5782 		if (!ext || ext->type != EXT_KSYM)
5783 			continue;
5784 
5785 		if (ext->is_set && ext->ksym.addr != sym_addr) {
5786 			pr_warn("extern (ksym) '%s' resolution is ambiguous: 0x%llx or 0x%llx\n",
5787 				sym_name, ext->ksym.addr, sym_addr);
5788 			err = -EINVAL;
5789 			goto out;
5790 		}
5791 		if (!ext->is_set) {
5792 			ext->is_set = true;
5793 			ext->ksym.addr = sym_addr;
5794 			pr_debug("extern (ksym) %s=0x%llx\n", sym_name, sym_addr);
5795 		}
5796 	}
5797 
5798 out:
5799 	fclose(f);
5800 	return err;
5801 }
5802 
5803 static int bpf_object__resolve_externs(struct bpf_object *obj,
5804 				       const char *extra_kconfig)
5805 {
5806 	bool need_config = false, need_kallsyms = false;
5807 	struct extern_desc *ext;
5808 	void *kcfg_data = NULL;
5809 	int err, i;
5810 
5811 	if (obj->nr_extern == 0)
5812 		return 0;
5813 
5814 	if (obj->kconfig_map_idx >= 0)
5815 		kcfg_data = obj->maps[obj->kconfig_map_idx].mmaped;
5816 
5817 	for (i = 0; i < obj->nr_extern; i++) {
5818 		ext = &obj->externs[i];
5819 
5820 		if (ext->type == EXT_KCFG &&
5821 		    strcmp(ext->name, "LINUX_KERNEL_VERSION") == 0) {
5822 			void *ext_val = kcfg_data + ext->kcfg.data_off;
5823 			__u32 kver = get_kernel_version();
5824 
5825 			if (!kver) {
5826 				pr_warn("failed to get kernel version\n");
5827 				return -EINVAL;
5828 			}
5829 			err = set_kcfg_value_num(ext, ext_val, kver);
5830 			if (err)
5831 				return err;
5832 			pr_debug("extern (kcfg) %s=0x%x\n", ext->name, kver);
5833 		} else if (ext->type == EXT_KCFG &&
5834 			   strncmp(ext->name, "CONFIG_", 7) == 0) {
5835 			need_config = true;
5836 		} else if (ext->type == EXT_KSYM) {
5837 			need_kallsyms = true;
5838 		} else {
5839 			pr_warn("unrecognized extern '%s'\n", ext->name);
5840 			return -EINVAL;
5841 		}
5842 	}
5843 	if (need_config && extra_kconfig) {
5844 		err = bpf_object__read_kconfig_mem(obj, extra_kconfig, kcfg_data);
5845 		if (err)
5846 			return -EINVAL;
5847 		need_config = false;
5848 		for (i = 0; i < obj->nr_extern; i++) {
5849 			ext = &obj->externs[i];
5850 			if (ext->type == EXT_KCFG && !ext->is_set) {
5851 				need_config = true;
5852 				break;
5853 			}
5854 		}
5855 	}
5856 	if (need_config) {
5857 		err = bpf_object__read_kconfig_file(obj, kcfg_data);
5858 		if (err)
5859 			return -EINVAL;
5860 	}
5861 	if (need_kallsyms) {
5862 		err = bpf_object__read_kallsyms_file(obj);
5863 		if (err)
5864 			return -EINVAL;
5865 	}
5866 	for (i = 0; i < obj->nr_extern; i++) {
5867 		ext = &obj->externs[i];
5868 
5869 		if (!ext->is_set && !ext->is_weak) {
5870 			pr_warn("extern %s (strong) not resolved\n", ext->name);
5871 			return -ESRCH;
5872 		} else if (!ext->is_set) {
5873 			pr_debug("extern %s (weak) not resolved, defaulting to zero\n",
5874 				 ext->name);
5875 		}
5876 	}
5877 
5878 	return 0;
5879 }
5880 
5881 int bpf_object__load_xattr(struct bpf_object_load_attr *attr)
5882 {
5883 	struct bpf_object *obj;
5884 	int err, i;
5885 
5886 	if (!attr)
5887 		return -EINVAL;
5888 	obj = attr->obj;
5889 	if (!obj)
5890 		return -EINVAL;
5891 
5892 	if (obj->loaded) {
5893 		pr_warn("object '%s': load can't be attempted twice\n", obj->name);
5894 		return -EINVAL;
5895 	}
5896 
5897 	err = bpf_object__probe_loading(obj);
5898 	err = err ? : bpf_object__probe_caps(obj);
5899 	err = err ? : bpf_object__resolve_externs(obj, obj->kconfig);
5900 	err = err ? : bpf_object__sanitize_and_load_btf(obj);
5901 	err = err ? : bpf_object__sanitize_maps(obj);
5902 	err = err ? : bpf_object__load_vmlinux_btf(obj);
5903 	err = err ? : bpf_object__init_kern_struct_ops_maps(obj);
5904 	err = err ? : bpf_object__create_maps(obj);
5905 	err = err ? : bpf_object__relocate(obj, attr->target_btf_path);
5906 	err = err ? : bpf_object__load_progs(obj, attr->log_level);
5907 
5908 	btf__free(obj->btf_vmlinux);
5909 	obj->btf_vmlinux = NULL;
5910 
5911 	obj->loaded = true; /* doesn't matter if successfully or not */
5912 
5913 	if (err)
5914 		goto out;
5915 
5916 	return 0;
5917 out:
5918 	/* unpin any maps that were auto-pinned during load */
5919 	for (i = 0; i < obj->nr_maps; i++)
5920 		if (obj->maps[i].pinned && !obj->maps[i].reused)
5921 			bpf_map__unpin(&obj->maps[i], NULL);
5922 
5923 	bpf_object__unload(obj);
5924 	pr_warn("failed to load object '%s'\n", obj->path);
5925 	return err;
5926 }
5927 
5928 int bpf_object__load(struct bpf_object *obj)
5929 {
5930 	struct bpf_object_load_attr attr = {
5931 		.obj = obj,
5932 	};
5933 
5934 	return bpf_object__load_xattr(&attr);
5935 }
5936 
5937 static int make_parent_dir(const char *path)
5938 {
5939 	char *cp, errmsg[STRERR_BUFSIZE];
5940 	char *dname, *dir;
5941 	int err = 0;
5942 
5943 	dname = strdup(path);
5944 	if (dname == NULL)
5945 		return -ENOMEM;
5946 
5947 	dir = dirname(dname);
5948 	if (mkdir(dir, 0700) && errno != EEXIST)
5949 		err = -errno;
5950 
5951 	free(dname);
5952 	if (err) {
5953 		cp = libbpf_strerror_r(-err, errmsg, sizeof(errmsg));
5954 		pr_warn("failed to mkdir %s: %s\n", path, cp);
5955 	}
5956 	return err;
5957 }
5958 
5959 static int check_path(const char *path)
5960 {
5961 	char *cp, errmsg[STRERR_BUFSIZE];
5962 	struct statfs st_fs;
5963 	char *dname, *dir;
5964 	int err = 0;
5965 
5966 	if (path == NULL)
5967 		return -EINVAL;
5968 
5969 	dname = strdup(path);
5970 	if (dname == NULL)
5971 		return -ENOMEM;
5972 
5973 	dir = dirname(dname);
5974 	if (statfs(dir, &st_fs)) {
5975 		cp = libbpf_strerror_r(errno, errmsg, sizeof(errmsg));
5976 		pr_warn("failed to statfs %s: %s\n", dir, cp);
5977 		err = -errno;
5978 	}
5979 	free(dname);
5980 
5981 	if (!err && st_fs.f_type != BPF_FS_MAGIC) {
5982 		pr_warn("specified path %s is not on BPF FS\n", path);
5983 		err = -EINVAL;
5984 	}
5985 
5986 	return err;
5987 }
5988 
5989 int bpf_program__pin_instance(struct bpf_program *prog, const char *path,
5990 			      int instance)
5991 {
5992 	char *cp, errmsg[STRERR_BUFSIZE];
5993 	int err;
5994 
5995 	err = make_parent_dir(path);
5996 	if (err)
5997 		return err;
5998 
5999 	err = check_path(path);
6000 	if (err)
6001 		return err;
6002 
6003 	if (prog == NULL) {
6004 		pr_warn("invalid program pointer\n");
6005 		return -EINVAL;
6006 	}
6007 
6008 	if (instance < 0 || instance >= prog->instances.nr) {
6009 		pr_warn("invalid prog instance %d of prog %s (max %d)\n",
6010 			instance, prog->section_name, prog->instances.nr);
6011 		return -EINVAL;
6012 	}
6013 
6014 	if (bpf_obj_pin(prog->instances.fds[instance], path)) {
6015 		cp = libbpf_strerror_r(errno, errmsg, sizeof(errmsg));
6016 		pr_warn("failed to pin program: %s\n", cp);
6017 		return -errno;
6018 	}
6019 	pr_debug("pinned program '%s'\n", path);
6020 
6021 	return 0;
6022 }
6023 
6024 int bpf_program__unpin_instance(struct bpf_program *prog, const char *path,
6025 				int instance)
6026 {
6027 	int err;
6028 
6029 	err = check_path(path);
6030 	if (err)
6031 		return err;
6032 
6033 	if (prog == NULL) {
6034 		pr_warn("invalid program pointer\n");
6035 		return -EINVAL;
6036 	}
6037 
6038 	if (instance < 0 || instance >= prog->instances.nr) {
6039 		pr_warn("invalid prog instance %d of prog %s (max %d)\n",
6040 			instance, prog->section_name, prog->instances.nr);
6041 		return -EINVAL;
6042 	}
6043 
6044 	err = unlink(path);
6045 	if (err != 0)
6046 		return -errno;
6047 	pr_debug("unpinned program '%s'\n", path);
6048 
6049 	return 0;
6050 }
6051 
6052 int bpf_program__pin(struct bpf_program *prog, const char *path)
6053 {
6054 	int i, err;
6055 
6056 	err = make_parent_dir(path);
6057 	if (err)
6058 		return err;
6059 
6060 	err = check_path(path);
6061 	if (err)
6062 		return err;
6063 
6064 	if (prog == NULL) {
6065 		pr_warn("invalid program pointer\n");
6066 		return -EINVAL;
6067 	}
6068 
6069 	if (prog->instances.nr <= 0) {
6070 		pr_warn("no instances of prog %s to pin\n",
6071 			   prog->section_name);
6072 		return -EINVAL;
6073 	}
6074 
6075 	if (prog->instances.nr == 1) {
6076 		/* don't create subdirs when pinning single instance */
6077 		return bpf_program__pin_instance(prog, path, 0);
6078 	}
6079 
6080 	for (i = 0; i < prog->instances.nr; i++) {
6081 		char buf[PATH_MAX];
6082 		int len;
6083 
6084 		len = snprintf(buf, PATH_MAX, "%s/%d", path, i);
6085 		if (len < 0) {
6086 			err = -EINVAL;
6087 			goto err_unpin;
6088 		} else if (len >= PATH_MAX) {
6089 			err = -ENAMETOOLONG;
6090 			goto err_unpin;
6091 		}
6092 
6093 		err = bpf_program__pin_instance(prog, buf, i);
6094 		if (err)
6095 			goto err_unpin;
6096 	}
6097 
6098 	return 0;
6099 
6100 err_unpin:
6101 	for (i = i - 1; i >= 0; i--) {
6102 		char buf[PATH_MAX];
6103 		int len;
6104 
6105 		len = snprintf(buf, PATH_MAX, "%s/%d", path, i);
6106 		if (len < 0)
6107 			continue;
6108 		else if (len >= PATH_MAX)
6109 			continue;
6110 
6111 		bpf_program__unpin_instance(prog, buf, i);
6112 	}
6113 
6114 	rmdir(path);
6115 
6116 	return err;
6117 }
6118 
6119 int bpf_program__unpin(struct bpf_program *prog, const char *path)
6120 {
6121 	int i, err;
6122 
6123 	err = check_path(path);
6124 	if (err)
6125 		return err;
6126 
6127 	if (prog == NULL) {
6128 		pr_warn("invalid program pointer\n");
6129 		return -EINVAL;
6130 	}
6131 
6132 	if (prog->instances.nr <= 0) {
6133 		pr_warn("no instances of prog %s to pin\n",
6134 			   prog->section_name);
6135 		return -EINVAL;
6136 	}
6137 
6138 	if (prog->instances.nr == 1) {
6139 		/* don't create subdirs when pinning single instance */
6140 		return bpf_program__unpin_instance(prog, path, 0);
6141 	}
6142 
6143 	for (i = 0; i < prog->instances.nr; i++) {
6144 		char buf[PATH_MAX];
6145 		int len;
6146 
6147 		len = snprintf(buf, PATH_MAX, "%s/%d", path, i);
6148 		if (len < 0)
6149 			return -EINVAL;
6150 		else if (len >= PATH_MAX)
6151 			return -ENAMETOOLONG;
6152 
6153 		err = bpf_program__unpin_instance(prog, buf, i);
6154 		if (err)
6155 			return err;
6156 	}
6157 
6158 	err = rmdir(path);
6159 	if (err)
6160 		return -errno;
6161 
6162 	return 0;
6163 }
6164 
6165 int bpf_map__pin(struct bpf_map *map, const char *path)
6166 {
6167 	char *cp, errmsg[STRERR_BUFSIZE];
6168 	int err;
6169 
6170 	if (map == NULL) {
6171 		pr_warn("invalid map pointer\n");
6172 		return -EINVAL;
6173 	}
6174 
6175 	if (map->pin_path) {
6176 		if (path && strcmp(path, map->pin_path)) {
6177 			pr_warn("map '%s' already has pin path '%s' different from '%s'\n",
6178 				bpf_map__name(map), map->pin_path, path);
6179 			return -EINVAL;
6180 		} else if (map->pinned) {
6181 			pr_debug("map '%s' already pinned at '%s'; not re-pinning\n",
6182 				 bpf_map__name(map), map->pin_path);
6183 			return 0;
6184 		}
6185 	} else {
6186 		if (!path) {
6187 			pr_warn("missing a path to pin map '%s' at\n",
6188 				bpf_map__name(map));
6189 			return -EINVAL;
6190 		} else if (map->pinned) {
6191 			pr_warn("map '%s' already pinned\n", bpf_map__name(map));
6192 			return -EEXIST;
6193 		}
6194 
6195 		map->pin_path = strdup(path);
6196 		if (!map->pin_path) {
6197 			err = -errno;
6198 			goto out_err;
6199 		}
6200 	}
6201 
6202 	err = make_parent_dir(map->pin_path);
6203 	if (err)
6204 		return err;
6205 
6206 	err = check_path(map->pin_path);
6207 	if (err)
6208 		return err;
6209 
6210 	if (bpf_obj_pin(map->fd, map->pin_path)) {
6211 		err = -errno;
6212 		goto out_err;
6213 	}
6214 
6215 	map->pinned = true;
6216 	pr_debug("pinned map '%s'\n", map->pin_path);
6217 
6218 	return 0;
6219 
6220 out_err:
6221 	cp = libbpf_strerror_r(-err, errmsg, sizeof(errmsg));
6222 	pr_warn("failed to pin map: %s\n", cp);
6223 	return err;
6224 }
6225 
6226 int bpf_map__unpin(struct bpf_map *map, const char *path)
6227 {
6228 	int err;
6229 
6230 	if (map == NULL) {
6231 		pr_warn("invalid map pointer\n");
6232 		return -EINVAL;
6233 	}
6234 
6235 	if (map->pin_path) {
6236 		if (path && strcmp(path, map->pin_path)) {
6237 			pr_warn("map '%s' already has pin path '%s' different from '%s'\n",
6238 				bpf_map__name(map), map->pin_path, path);
6239 			return -EINVAL;
6240 		}
6241 		path = map->pin_path;
6242 	} else if (!path) {
6243 		pr_warn("no path to unpin map '%s' from\n",
6244 			bpf_map__name(map));
6245 		return -EINVAL;
6246 	}
6247 
6248 	err = check_path(path);
6249 	if (err)
6250 		return err;
6251 
6252 	err = unlink(path);
6253 	if (err != 0)
6254 		return -errno;
6255 
6256 	map->pinned = false;
6257 	pr_debug("unpinned map '%s' from '%s'\n", bpf_map__name(map), path);
6258 
6259 	return 0;
6260 }
6261 
6262 int bpf_map__set_pin_path(struct bpf_map *map, const char *path)
6263 {
6264 	char *new = NULL;
6265 
6266 	if (path) {
6267 		new = strdup(path);
6268 		if (!new)
6269 			return -errno;
6270 	}
6271 
6272 	free(map->pin_path);
6273 	map->pin_path = new;
6274 	return 0;
6275 }
6276 
6277 const char *bpf_map__get_pin_path(const struct bpf_map *map)
6278 {
6279 	return map->pin_path;
6280 }
6281 
6282 bool bpf_map__is_pinned(const struct bpf_map *map)
6283 {
6284 	return map->pinned;
6285 }
6286 
6287 int bpf_object__pin_maps(struct bpf_object *obj, const char *path)
6288 {
6289 	struct bpf_map *map;
6290 	int err;
6291 
6292 	if (!obj)
6293 		return -ENOENT;
6294 
6295 	if (!obj->loaded) {
6296 		pr_warn("object not yet loaded; load it first\n");
6297 		return -ENOENT;
6298 	}
6299 
6300 	bpf_object__for_each_map(map, obj) {
6301 		char *pin_path = NULL;
6302 		char buf[PATH_MAX];
6303 
6304 		if (path) {
6305 			int len;
6306 
6307 			len = snprintf(buf, PATH_MAX, "%s/%s", path,
6308 				       bpf_map__name(map));
6309 			if (len < 0) {
6310 				err = -EINVAL;
6311 				goto err_unpin_maps;
6312 			} else if (len >= PATH_MAX) {
6313 				err = -ENAMETOOLONG;
6314 				goto err_unpin_maps;
6315 			}
6316 			pin_path = buf;
6317 		} else if (!map->pin_path) {
6318 			continue;
6319 		}
6320 
6321 		err = bpf_map__pin(map, pin_path);
6322 		if (err)
6323 			goto err_unpin_maps;
6324 	}
6325 
6326 	return 0;
6327 
6328 err_unpin_maps:
6329 	while ((map = bpf_map__prev(map, obj))) {
6330 		if (!map->pin_path)
6331 			continue;
6332 
6333 		bpf_map__unpin(map, NULL);
6334 	}
6335 
6336 	return err;
6337 }
6338 
6339 int bpf_object__unpin_maps(struct bpf_object *obj, const char *path)
6340 {
6341 	struct bpf_map *map;
6342 	int err;
6343 
6344 	if (!obj)
6345 		return -ENOENT;
6346 
6347 	bpf_object__for_each_map(map, obj) {
6348 		char *pin_path = NULL;
6349 		char buf[PATH_MAX];
6350 
6351 		if (path) {
6352 			int len;
6353 
6354 			len = snprintf(buf, PATH_MAX, "%s/%s", path,
6355 				       bpf_map__name(map));
6356 			if (len < 0)
6357 				return -EINVAL;
6358 			else if (len >= PATH_MAX)
6359 				return -ENAMETOOLONG;
6360 			pin_path = buf;
6361 		} else if (!map->pin_path) {
6362 			continue;
6363 		}
6364 
6365 		err = bpf_map__unpin(map, pin_path);
6366 		if (err)
6367 			return err;
6368 	}
6369 
6370 	return 0;
6371 }
6372 
6373 int bpf_object__pin_programs(struct bpf_object *obj, const char *path)
6374 {
6375 	struct bpf_program *prog;
6376 	int err;
6377 
6378 	if (!obj)
6379 		return -ENOENT;
6380 
6381 	if (!obj->loaded) {
6382 		pr_warn("object not yet loaded; load it first\n");
6383 		return -ENOENT;
6384 	}
6385 
6386 	bpf_object__for_each_program(prog, obj) {
6387 		char buf[PATH_MAX];
6388 		int len;
6389 
6390 		len = snprintf(buf, PATH_MAX, "%s/%s", path,
6391 			       prog->pin_name);
6392 		if (len < 0) {
6393 			err = -EINVAL;
6394 			goto err_unpin_programs;
6395 		} else if (len >= PATH_MAX) {
6396 			err = -ENAMETOOLONG;
6397 			goto err_unpin_programs;
6398 		}
6399 
6400 		err = bpf_program__pin(prog, buf);
6401 		if (err)
6402 			goto err_unpin_programs;
6403 	}
6404 
6405 	return 0;
6406 
6407 err_unpin_programs:
6408 	while ((prog = bpf_program__prev(prog, obj))) {
6409 		char buf[PATH_MAX];
6410 		int len;
6411 
6412 		len = snprintf(buf, PATH_MAX, "%s/%s", path,
6413 			       prog->pin_name);
6414 		if (len < 0)
6415 			continue;
6416 		else if (len >= PATH_MAX)
6417 			continue;
6418 
6419 		bpf_program__unpin(prog, buf);
6420 	}
6421 
6422 	return err;
6423 }
6424 
6425 int bpf_object__unpin_programs(struct bpf_object *obj, const char *path)
6426 {
6427 	struct bpf_program *prog;
6428 	int err;
6429 
6430 	if (!obj)
6431 		return -ENOENT;
6432 
6433 	bpf_object__for_each_program(prog, obj) {
6434 		char buf[PATH_MAX];
6435 		int len;
6436 
6437 		len = snprintf(buf, PATH_MAX, "%s/%s", path,
6438 			       prog->pin_name);
6439 		if (len < 0)
6440 			return -EINVAL;
6441 		else if (len >= PATH_MAX)
6442 			return -ENAMETOOLONG;
6443 
6444 		err = bpf_program__unpin(prog, buf);
6445 		if (err)
6446 			return err;
6447 	}
6448 
6449 	return 0;
6450 }
6451 
6452 int bpf_object__pin(struct bpf_object *obj, const char *path)
6453 {
6454 	int err;
6455 
6456 	err = bpf_object__pin_maps(obj, path);
6457 	if (err)
6458 		return err;
6459 
6460 	err = bpf_object__pin_programs(obj, path);
6461 	if (err) {
6462 		bpf_object__unpin_maps(obj, path);
6463 		return err;
6464 	}
6465 
6466 	return 0;
6467 }
6468 
6469 static void bpf_map__destroy(struct bpf_map *map)
6470 {
6471 	if (map->clear_priv)
6472 		map->clear_priv(map, map->priv);
6473 	map->priv = NULL;
6474 	map->clear_priv = NULL;
6475 
6476 	if (map->inner_map) {
6477 		bpf_map__destroy(map->inner_map);
6478 		zfree(&map->inner_map);
6479 	}
6480 
6481 	zfree(&map->init_slots);
6482 	map->init_slots_sz = 0;
6483 
6484 	if (map->mmaped) {
6485 		munmap(map->mmaped, bpf_map_mmap_sz(map));
6486 		map->mmaped = NULL;
6487 	}
6488 
6489 	if (map->st_ops) {
6490 		zfree(&map->st_ops->data);
6491 		zfree(&map->st_ops->progs);
6492 		zfree(&map->st_ops->kern_func_off);
6493 		zfree(&map->st_ops);
6494 	}
6495 
6496 	zfree(&map->name);
6497 	zfree(&map->pin_path);
6498 
6499 	if (map->fd >= 0)
6500 		zclose(map->fd);
6501 }
6502 
6503 void bpf_object__close(struct bpf_object *obj)
6504 {
6505 	size_t i;
6506 
6507 	if (IS_ERR_OR_NULL(obj))
6508 		return;
6509 
6510 	if (obj->clear_priv)
6511 		obj->clear_priv(obj, obj->priv);
6512 
6513 	bpf_object__elf_finish(obj);
6514 	bpf_object__unload(obj);
6515 	btf__free(obj->btf);
6516 	btf_ext__free(obj->btf_ext);
6517 
6518 	for (i = 0; i < obj->nr_maps; i++)
6519 		bpf_map__destroy(&obj->maps[i]);
6520 
6521 	zfree(&obj->kconfig);
6522 	zfree(&obj->externs);
6523 	obj->nr_extern = 0;
6524 
6525 	zfree(&obj->maps);
6526 	obj->nr_maps = 0;
6527 
6528 	if (obj->programs && obj->nr_programs) {
6529 		for (i = 0; i < obj->nr_programs; i++)
6530 			bpf_program__exit(&obj->programs[i]);
6531 	}
6532 	zfree(&obj->programs);
6533 
6534 	list_del(&obj->list);
6535 	free(obj);
6536 }
6537 
6538 struct bpf_object *
6539 bpf_object__next(struct bpf_object *prev)
6540 {
6541 	struct bpf_object *next;
6542 
6543 	if (!prev)
6544 		next = list_first_entry(&bpf_objects_list,
6545 					struct bpf_object,
6546 					list);
6547 	else
6548 		next = list_next_entry(prev, list);
6549 
6550 	/* Empty list is noticed here so don't need checking on entry. */
6551 	if (&next->list == &bpf_objects_list)
6552 		return NULL;
6553 
6554 	return next;
6555 }
6556 
6557 const char *bpf_object__name(const struct bpf_object *obj)
6558 {
6559 	return obj ? obj->name : ERR_PTR(-EINVAL);
6560 }
6561 
6562 unsigned int bpf_object__kversion(const struct bpf_object *obj)
6563 {
6564 	return obj ? obj->kern_version : 0;
6565 }
6566 
6567 struct btf *bpf_object__btf(const struct bpf_object *obj)
6568 {
6569 	return obj ? obj->btf : NULL;
6570 }
6571 
6572 int bpf_object__btf_fd(const struct bpf_object *obj)
6573 {
6574 	return obj->btf ? btf__fd(obj->btf) : -1;
6575 }
6576 
6577 int bpf_object__set_priv(struct bpf_object *obj, void *priv,
6578 			 bpf_object_clear_priv_t clear_priv)
6579 {
6580 	if (obj->priv && obj->clear_priv)
6581 		obj->clear_priv(obj, obj->priv);
6582 
6583 	obj->priv = priv;
6584 	obj->clear_priv = clear_priv;
6585 	return 0;
6586 }
6587 
6588 void *bpf_object__priv(const struct bpf_object *obj)
6589 {
6590 	return obj ? obj->priv : ERR_PTR(-EINVAL);
6591 }
6592 
6593 static struct bpf_program *
6594 __bpf_program__iter(const struct bpf_program *p, const struct bpf_object *obj,
6595 		    bool forward)
6596 {
6597 	size_t nr_programs = obj->nr_programs;
6598 	ssize_t idx;
6599 
6600 	if (!nr_programs)
6601 		return NULL;
6602 
6603 	if (!p)
6604 		/* Iter from the beginning */
6605 		return forward ? &obj->programs[0] :
6606 			&obj->programs[nr_programs - 1];
6607 
6608 	if (p->obj != obj) {
6609 		pr_warn("error: program handler doesn't match object\n");
6610 		return NULL;
6611 	}
6612 
6613 	idx = (p - obj->programs) + (forward ? 1 : -1);
6614 	if (idx >= obj->nr_programs || idx < 0)
6615 		return NULL;
6616 	return &obj->programs[idx];
6617 }
6618 
6619 struct bpf_program *
6620 bpf_program__next(struct bpf_program *prev, const struct bpf_object *obj)
6621 {
6622 	struct bpf_program *prog = prev;
6623 
6624 	do {
6625 		prog = __bpf_program__iter(prog, obj, true);
6626 	} while (prog && bpf_program__is_function_storage(prog, obj));
6627 
6628 	return prog;
6629 }
6630 
6631 struct bpf_program *
6632 bpf_program__prev(struct bpf_program *next, const struct bpf_object *obj)
6633 {
6634 	struct bpf_program *prog = next;
6635 
6636 	do {
6637 		prog = __bpf_program__iter(prog, obj, false);
6638 	} while (prog && bpf_program__is_function_storage(prog, obj));
6639 
6640 	return prog;
6641 }
6642 
6643 int bpf_program__set_priv(struct bpf_program *prog, void *priv,
6644 			  bpf_program_clear_priv_t clear_priv)
6645 {
6646 	if (prog->priv && prog->clear_priv)
6647 		prog->clear_priv(prog, prog->priv);
6648 
6649 	prog->priv = priv;
6650 	prog->clear_priv = clear_priv;
6651 	return 0;
6652 }
6653 
6654 void *bpf_program__priv(const struct bpf_program *prog)
6655 {
6656 	return prog ? prog->priv : ERR_PTR(-EINVAL);
6657 }
6658 
6659 void bpf_program__set_ifindex(struct bpf_program *prog, __u32 ifindex)
6660 {
6661 	prog->prog_ifindex = ifindex;
6662 }
6663 
6664 const char *bpf_program__name(const struct bpf_program *prog)
6665 {
6666 	return prog->name;
6667 }
6668 
6669 const char *bpf_program__title(const struct bpf_program *prog, bool needs_copy)
6670 {
6671 	const char *title;
6672 
6673 	title = prog->section_name;
6674 	if (needs_copy) {
6675 		title = strdup(title);
6676 		if (!title) {
6677 			pr_warn("failed to strdup program title\n");
6678 			return ERR_PTR(-ENOMEM);
6679 		}
6680 	}
6681 
6682 	return title;
6683 }
6684 
6685 bool bpf_program__autoload(const struct bpf_program *prog)
6686 {
6687 	return prog->load;
6688 }
6689 
6690 int bpf_program__set_autoload(struct bpf_program *prog, bool autoload)
6691 {
6692 	if (prog->obj->loaded)
6693 		return -EINVAL;
6694 
6695 	prog->load = autoload;
6696 	return 0;
6697 }
6698 
6699 int bpf_program__fd(const struct bpf_program *prog)
6700 {
6701 	return bpf_program__nth_fd(prog, 0);
6702 }
6703 
6704 size_t bpf_program__size(const struct bpf_program *prog)
6705 {
6706 	return prog->insns_cnt * sizeof(struct bpf_insn);
6707 }
6708 
6709 int bpf_program__set_prep(struct bpf_program *prog, int nr_instances,
6710 			  bpf_program_prep_t prep)
6711 {
6712 	int *instances_fds;
6713 
6714 	if (nr_instances <= 0 || !prep)
6715 		return -EINVAL;
6716 
6717 	if (prog->instances.nr > 0 || prog->instances.fds) {
6718 		pr_warn("Can't set pre-processor after loading\n");
6719 		return -EINVAL;
6720 	}
6721 
6722 	instances_fds = malloc(sizeof(int) * nr_instances);
6723 	if (!instances_fds) {
6724 		pr_warn("alloc memory failed for fds\n");
6725 		return -ENOMEM;
6726 	}
6727 
6728 	/* fill all fd with -1 */
6729 	memset(instances_fds, -1, sizeof(int) * nr_instances);
6730 
6731 	prog->instances.nr = nr_instances;
6732 	prog->instances.fds = instances_fds;
6733 	prog->preprocessor = prep;
6734 	return 0;
6735 }
6736 
6737 int bpf_program__nth_fd(const struct bpf_program *prog, int n)
6738 {
6739 	int fd;
6740 
6741 	if (!prog)
6742 		return -EINVAL;
6743 
6744 	if (n >= prog->instances.nr || n < 0) {
6745 		pr_warn("Can't get the %dth fd from program %s: only %d instances\n",
6746 			n, prog->section_name, prog->instances.nr);
6747 		return -EINVAL;
6748 	}
6749 
6750 	fd = prog->instances.fds[n];
6751 	if (fd < 0) {
6752 		pr_warn("%dth instance of program '%s' is invalid\n",
6753 			n, prog->section_name);
6754 		return -ENOENT;
6755 	}
6756 
6757 	return fd;
6758 }
6759 
6760 enum bpf_prog_type bpf_program__get_type(struct bpf_program *prog)
6761 {
6762 	return prog->type;
6763 }
6764 
6765 void bpf_program__set_type(struct bpf_program *prog, enum bpf_prog_type type)
6766 {
6767 	prog->type = type;
6768 }
6769 
6770 static bool bpf_program__is_type(const struct bpf_program *prog,
6771 				 enum bpf_prog_type type)
6772 {
6773 	return prog ? (prog->type == type) : false;
6774 }
6775 
6776 #define BPF_PROG_TYPE_FNS(NAME, TYPE)				\
6777 int bpf_program__set_##NAME(struct bpf_program *prog)		\
6778 {								\
6779 	if (!prog)						\
6780 		return -EINVAL;					\
6781 	bpf_program__set_type(prog, TYPE);			\
6782 	return 0;						\
6783 }								\
6784 								\
6785 bool bpf_program__is_##NAME(const struct bpf_program *prog)	\
6786 {								\
6787 	return bpf_program__is_type(prog, TYPE);		\
6788 }								\
6789 
6790 BPF_PROG_TYPE_FNS(socket_filter, BPF_PROG_TYPE_SOCKET_FILTER);
6791 BPF_PROG_TYPE_FNS(lsm, BPF_PROG_TYPE_LSM);
6792 BPF_PROG_TYPE_FNS(kprobe, BPF_PROG_TYPE_KPROBE);
6793 BPF_PROG_TYPE_FNS(sched_cls, BPF_PROG_TYPE_SCHED_CLS);
6794 BPF_PROG_TYPE_FNS(sched_act, BPF_PROG_TYPE_SCHED_ACT);
6795 BPF_PROG_TYPE_FNS(tracepoint, BPF_PROG_TYPE_TRACEPOINT);
6796 BPF_PROG_TYPE_FNS(raw_tracepoint, BPF_PROG_TYPE_RAW_TRACEPOINT);
6797 BPF_PROG_TYPE_FNS(xdp, BPF_PROG_TYPE_XDP);
6798 BPF_PROG_TYPE_FNS(perf_event, BPF_PROG_TYPE_PERF_EVENT);
6799 BPF_PROG_TYPE_FNS(tracing, BPF_PROG_TYPE_TRACING);
6800 BPF_PROG_TYPE_FNS(struct_ops, BPF_PROG_TYPE_STRUCT_OPS);
6801 BPF_PROG_TYPE_FNS(extension, BPF_PROG_TYPE_EXT);
6802 BPF_PROG_TYPE_FNS(sk_lookup, BPF_PROG_TYPE_SK_LOOKUP);
6803 
6804 enum bpf_attach_type
6805 bpf_program__get_expected_attach_type(struct bpf_program *prog)
6806 {
6807 	return prog->expected_attach_type;
6808 }
6809 
6810 void bpf_program__set_expected_attach_type(struct bpf_program *prog,
6811 					   enum bpf_attach_type type)
6812 {
6813 	prog->expected_attach_type = type;
6814 }
6815 
6816 #define BPF_PROG_SEC_IMPL(string, ptype, eatype, eatype_optional,	    \
6817 			  attachable, attach_btf)			    \
6818 	{								    \
6819 		.sec = string,						    \
6820 		.len = sizeof(string) - 1,				    \
6821 		.prog_type = ptype,					    \
6822 		.expected_attach_type = eatype,				    \
6823 		.is_exp_attach_type_optional = eatype_optional,		    \
6824 		.is_attachable = attachable,				    \
6825 		.is_attach_btf = attach_btf,				    \
6826 	}
6827 
6828 /* Programs that can NOT be attached. */
6829 #define BPF_PROG_SEC(string, ptype) BPF_PROG_SEC_IMPL(string, ptype, 0, 0, 0, 0)
6830 
6831 /* Programs that can be attached. */
6832 #define BPF_APROG_SEC(string, ptype, atype) \
6833 	BPF_PROG_SEC_IMPL(string, ptype, atype, true, 1, 0)
6834 
6835 /* Programs that must specify expected attach type at load time. */
6836 #define BPF_EAPROG_SEC(string, ptype, eatype) \
6837 	BPF_PROG_SEC_IMPL(string, ptype, eatype, false, 1, 0)
6838 
6839 /* Programs that use BTF to identify attach point */
6840 #define BPF_PROG_BTF(string, ptype, eatype) \
6841 	BPF_PROG_SEC_IMPL(string, ptype, eatype, false, 0, 1)
6842 
6843 /* Programs that can be attached but attach type can't be identified by section
6844  * name. Kept for backward compatibility.
6845  */
6846 #define BPF_APROG_COMPAT(string, ptype) BPF_PROG_SEC(string, ptype)
6847 
6848 #define SEC_DEF(sec_pfx, ptype, ...) {					    \
6849 	.sec = sec_pfx,							    \
6850 	.len = sizeof(sec_pfx) - 1,					    \
6851 	.prog_type = BPF_PROG_TYPE_##ptype,				    \
6852 	__VA_ARGS__							    \
6853 }
6854 
6855 static struct bpf_link *attach_kprobe(const struct bpf_sec_def *sec,
6856 				      struct bpf_program *prog);
6857 static struct bpf_link *attach_tp(const struct bpf_sec_def *sec,
6858 				  struct bpf_program *prog);
6859 static struct bpf_link *attach_raw_tp(const struct bpf_sec_def *sec,
6860 				      struct bpf_program *prog);
6861 static struct bpf_link *attach_trace(const struct bpf_sec_def *sec,
6862 				     struct bpf_program *prog);
6863 static struct bpf_link *attach_lsm(const struct bpf_sec_def *sec,
6864 				   struct bpf_program *prog);
6865 static struct bpf_link *attach_iter(const struct bpf_sec_def *sec,
6866 				    struct bpf_program *prog);
6867 
6868 static const struct bpf_sec_def section_defs[] = {
6869 	BPF_PROG_SEC("socket",			BPF_PROG_TYPE_SOCKET_FILTER),
6870 	BPF_PROG_SEC("sk_reuseport",		BPF_PROG_TYPE_SK_REUSEPORT),
6871 	SEC_DEF("kprobe/", KPROBE,
6872 		.attach_fn = attach_kprobe),
6873 	BPF_PROG_SEC("uprobe/",			BPF_PROG_TYPE_KPROBE),
6874 	SEC_DEF("kretprobe/", KPROBE,
6875 		.attach_fn = attach_kprobe),
6876 	BPF_PROG_SEC("uretprobe/",		BPF_PROG_TYPE_KPROBE),
6877 	BPF_PROG_SEC("classifier",		BPF_PROG_TYPE_SCHED_CLS),
6878 	BPF_PROG_SEC("action",			BPF_PROG_TYPE_SCHED_ACT),
6879 	SEC_DEF("tracepoint/", TRACEPOINT,
6880 		.attach_fn = attach_tp),
6881 	SEC_DEF("tp/", TRACEPOINT,
6882 		.attach_fn = attach_tp),
6883 	SEC_DEF("raw_tracepoint/", RAW_TRACEPOINT,
6884 		.attach_fn = attach_raw_tp),
6885 	SEC_DEF("raw_tp/", RAW_TRACEPOINT,
6886 		.attach_fn = attach_raw_tp),
6887 	SEC_DEF("tp_btf/", TRACING,
6888 		.expected_attach_type = BPF_TRACE_RAW_TP,
6889 		.is_attach_btf = true,
6890 		.attach_fn = attach_trace),
6891 	SEC_DEF("fentry/", TRACING,
6892 		.expected_attach_type = BPF_TRACE_FENTRY,
6893 		.is_attach_btf = true,
6894 		.attach_fn = attach_trace),
6895 	SEC_DEF("fmod_ret/", TRACING,
6896 		.expected_attach_type = BPF_MODIFY_RETURN,
6897 		.is_attach_btf = true,
6898 		.attach_fn = attach_trace),
6899 	SEC_DEF("fexit/", TRACING,
6900 		.expected_attach_type = BPF_TRACE_FEXIT,
6901 		.is_attach_btf = true,
6902 		.attach_fn = attach_trace),
6903 	SEC_DEF("freplace/", EXT,
6904 		.is_attach_btf = true,
6905 		.attach_fn = attach_trace),
6906 	SEC_DEF("lsm/", LSM,
6907 		.is_attach_btf = true,
6908 		.expected_attach_type = BPF_LSM_MAC,
6909 		.attach_fn = attach_lsm),
6910 	SEC_DEF("iter/", TRACING,
6911 		.expected_attach_type = BPF_TRACE_ITER,
6912 		.is_attach_btf = true,
6913 		.attach_fn = attach_iter),
6914 	BPF_EAPROG_SEC("xdp_devmap/",		BPF_PROG_TYPE_XDP,
6915 						BPF_XDP_DEVMAP),
6916 	BPF_EAPROG_SEC("xdp_cpumap/",		BPF_PROG_TYPE_XDP,
6917 						BPF_XDP_CPUMAP),
6918 	BPF_EAPROG_SEC("xdp",			BPF_PROG_TYPE_XDP,
6919 						BPF_XDP),
6920 	BPF_PROG_SEC("perf_event",		BPF_PROG_TYPE_PERF_EVENT),
6921 	BPF_PROG_SEC("lwt_in",			BPF_PROG_TYPE_LWT_IN),
6922 	BPF_PROG_SEC("lwt_out",			BPF_PROG_TYPE_LWT_OUT),
6923 	BPF_PROG_SEC("lwt_xmit",		BPF_PROG_TYPE_LWT_XMIT),
6924 	BPF_PROG_SEC("lwt_seg6local",		BPF_PROG_TYPE_LWT_SEG6LOCAL),
6925 	BPF_APROG_SEC("cgroup_skb/ingress",	BPF_PROG_TYPE_CGROUP_SKB,
6926 						BPF_CGROUP_INET_INGRESS),
6927 	BPF_APROG_SEC("cgroup_skb/egress",	BPF_PROG_TYPE_CGROUP_SKB,
6928 						BPF_CGROUP_INET_EGRESS),
6929 	BPF_APROG_COMPAT("cgroup/skb",		BPF_PROG_TYPE_CGROUP_SKB),
6930 	BPF_EAPROG_SEC("cgroup/sock_create",	BPF_PROG_TYPE_CGROUP_SOCK,
6931 						BPF_CGROUP_INET_SOCK_CREATE),
6932 	BPF_EAPROG_SEC("cgroup/sock_release",	BPF_PROG_TYPE_CGROUP_SOCK,
6933 						BPF_CGROUP_INET_SOCK_RELEASE),
6934 	BPF_APROG_SEC("cgroup/sock",		BPF_PROG_TYPE_CGROUP_SOCK,
6935 						BPF_CGROUP_INET_SOCK_CREATE),
6936 	BPF_EAPROG_SEC("cgroup/post_bind4",	BPF_PROG_TYPE_CGROUP_SOCK,
6937 						BPF_CGROUP_INET4_POST_BIND),
6938 	BPF_EAPROG_SEC("cgroup/post_bind6",	BPF_PROG_TYPE_CGROUP_SOCK,
6939 						BPF_CGROUP_INET6_POST_BIND),
6940 	BPF_APROG_SEC("cgroup/dev",		BPF_PROG_TYPE_CGROUP_DEVICE,
6941 						BPF_CGROUP_DEVICE),
6942 	BPF_APROG_SEC("sockops",		BPF_PROG_TYPE_SOCK_OPS,
6943 						BPF_CGROUP_SOCK_OPS),
6944 	BPF_APROG_SEC("sk_skb/stream_parser",	BPF_PROG_TYPE_SK_SKB,
6945 						BPF_SK_SKB_STREAM_PARSER),
6946 	BPF_APROG_SEC("sk_skb/stream_verdict",	BPF_PROG_TYPE_SK_SKB,
6947 						BPF_SK_SKB_STREAM_VERDICT),
6948 	BPF_APROG_COMPAT("sk_skb",		BPF_PROG_TYPE_SK_SKB),
6949 	BPF_APROG_SEC("sk_msg",			BPF_PROG_TYPE_SK_MSG,
6950 						BPF_SK_MSG_VERDICT),
6951 	BPF_APROG_SEC("lirc_mode2",		BPF_PROG_TYPE_LIRC_MODE2,
6952 						BPF_LIRC_MODE2),
6953 	BPF_APROG_SEC("flow_dissector",		BPF_PROG_TYPE_FLOW_DISSECTOR,
6954 						BPF_FLOW_DISSECTOR),
6955 	BPF_EAPROG_SEC("cgroup/bind4",		BPF_PROG_TYPE_CGROUP_SOCK_ADDR,
6956 						BPF_CGROUP_INET4_BIND),
6957 	BPF_EAPROG_SEC("cgroup/bind6",		BPF_PROG_TYPE_CGROUP_SOCK_ADDR,
6958 						BPF_CGROUP_INET6_BIND),
6959 	BPF_EAPROG_SEC("cgroup/connect4",	BPF_PROG_TYPE_CGROUP_SOCK_ADDR,
6960 						BPF_CGROUP_INET4_CONNECT),
6961 	BPF_EAPROG_SEC("cgroup/connect6",	BPF_PROG_TYPE_CGROUP_SOCK_ADDR,
6962 						BPF_CGROUP_INET6_CONNECT),
6963 	BPF_EAPROG_SEC("cgroup/sendmsg4",	BPF_PROG_TYPE_CGROUP_SOCK_ADDR,
6964 						BPF_CGROUP_UDP4_SENDMSG),
6965 	BPF_EAPROG_SEC("cgroup/sendmsg6",	BPF_PROG_TYPE_CGROUP_SOCK_ADDR,
6966 						BPF_CGROUP_UDP6_SENDMSG),
6967 	BPF_EAPROG_SEC("cgroup/recvmsg4",	BPF_PROG_TYPE_CGROUP_SOCK_ADDR,
6968 						BPF_CGROUP_UDP4_RECVMSG),
6969 	BPF_EAPROG_SEC("cgroup/recvmsg6",	BPF_PROG_TYPE_CGROUP_SOCK_ADDR,
6970 						BPF_CGROUP_UDP6_RECVMSG),
6971 	BPF_EAPROG_SEC("cgroup/getpeername4",	BPF_PROG_TYPE_CGROUP_SOCK_ADDR,
6972 						BPF_CGROUP_INET4_GETPEERNAME),
6973 	BPF_EAPROG_SEC("cgroup/getpeername6",	BPF_PROG_TYPE_CGROUP_SOCK_ADDR,
6974 						BPF_CGROUP_INET6_GETPEERNAME),
6975 	BPF_EAPROG_SEC("cgroup/getsockname4",	BPF_PROG_TYPE_CGROUP_SOCK_ADDR,
6976 						BPF_CGROUP_INET4_GETSOCKNAME),
6977 	BPF_EAPROG_SEC("cgroup/getsockname6",	BPF_PROG_TYPE_CGROUP_SOCK_ADDR,
6978 						BPF_CGROUP_INET6_GETSOCKNAME),
6979 	BPF_EAPROG_SEC("cgroup/sysctl",		BPF_PROG_TYPE_CGROUP_SYSCTL,
6980 						BPF_CGROUP_SYSCTL),
6981 	BPF_EAPROG_SEC("cgroup/getsockopt",	BPF_PROG_TYPE_CGROUP_SOCKOPT,
6982 						BPF_CGROUP_GETSOCKOPT),
6983 	BPF_EAPROG_SEC("cgroup/setsockopt",	BPF_PROG_TYPE_CGROUP_SOCKOPT,
6984 						BPF_CGROUP_SETSOCKOPT),
6985 	BPF_PROG_SEC("struct_ops",		BPF_PROG_TYPE_STRUCT_OPS),
6986 	BPF_EAPROG_SEC("sk_lookup/",		BPF_PROG_TYPE_SK_LOOKUP,
6987 						BPF_SK_LOOKUP),
6988 };
6989 
6990 #undef BPF_PROG_SEC_IMPL
6991 #undef BPF_PROG_SEC
6992 #undef BPF_APROG_SEC
6993 #undef BPF_EAPROG_SEC
6994 #undef BPF_APROG_COMPAT
6995 #undef SEC_DEF
6996 
6997 #define MAX_TYPE_NAME_SIZE 32
6998 
6999 static const struct bpf_sec_def *find_sec_def(const char *sec_name)
7000 {
7001 	int i, n = ARRAY_SIZE(section_defs);
7002 
7003 	for (i = 0; i < n; i++) {
7004 		if (strncmp(sec_name,
7005 			    section_defs[i].sec, section_defs[i].len))
7006 			continue;
7007 		return &section_defs[i];
7008 	}
7009 	return NULL;
7010 }
7011 
7012 static char *libbpf_get_type_names(bool attach_type)
7013 {
7014 	int i, len = ARRAY_SIZE(section_defs) * MAX_TYPE_NAME_SIZE;
7015 	char *buf;
7016 
7017 	buf = malloc(len);
7018 	if (!buf)
7019 		return NULL;
7020 
7021 	buf[0] = '\0';
7022 	/* Forge string buf with all available names */
7023 	for (i = 0; i < ARRAY_SIZE(section_defs); i++) {
7024 		if (attach_type && !section_defs[i].is_attachable)
7025 			continue;
7026 
7027 		if (strlen(buf) + strlen(section_defs[i].sec) + 2 > len) {
7028 			free(buf);
7029 			return NULL;
7030 		}
7031 		strcat(buf, " ");
7032 		strcat(buf, section_defs[i].sec);
7033 	}
7034 
7035 	return buf;
7036 }
7037 
7038 int libbpf_prog_type_by_name(const char *name, enum bpf_prog_type *prog_type,
7039 			     enum bpf_attach_type *expected_attach_type)
7040 {
7041 	const struct bpf_sec_def *sec_def;
7042 	char *type_names;
7043 
7044 	if (!name)
7045 		return -EINVAL;
7046 
7047 	sec_def = find_sec_def(name);
7048 	if (sec_def) {
7049 		*prog_type = sec_def->prog_type;
7050 		*expected_attach_type = sec_def->expected_attach_type;
7051 		return 0;
7052 	}
7053 
7054 	pr_debug("failed to guess program type from ELF section '%s'\n", name);
7055 	type_names = libbpf_get_type_names(false);
7056 	if (type_names != NULL) {
7057 		pr_debug("supported section(type) names are:%s\n", type_names);
7058 		free(type_names);
7059 	}
7060 
7061 	return -ESRCH;
7062 }
7063 
7064 static struct bpf_map *find_struct_ops_map_by_offset(struct bpf_object *obj,
7065 						     size_t offset)
7066 {
7067 	struct bpf_map *map;
7068 	size_t i;
7069 
7070 	for (i = 0; i < obj->nr_maps; i++) {
7071 		map = &obj->maps[i];
7072 		if (!bpf_map__is_struct_ops(map))
7073 			continue;
7074 		if (map->sec_offset <= offset &&
7075 		    offset - map->sec_offset < map->def.value_size)
7076 			return map;
7077 	}
7078 
7079 	return NULL;
7080 }
7081 
7082 /* Collect the reloc from ELF and populate the st_ops->progs[] */
7083 static int bpf_object__collect_st_ops_relos(struct bpf_object *obj,
7084 					    GElf_Shdr *shdr, Elf_Data *data)
7085 {
7086 	const struct btf_member *member;
7087 	struct bpf_struct_ops *st_ops;
7088 	struct bpf_program *prog;
7089 	unsigned int shdr_idx;
7090 	const struct btf *btf;
7091 	struct bpf_map *map;
7092 	Elf_Data *symbols;
7093 	unsigned int moff;
7094 	const char *name;
7095 	__u32 member_idx;
7096 	GElf_Sym sym;
7097 	GElf_Rel rel;
7098 	int i, nrels;
7099 
7100 	symbols = obj->efile.symbols;
7101 	btf = obj->btf;
7102 	nrels = shdr->sh_size / shdr->sh_entsize;
7103 	for (i = 0; i < nrels; i++) {
7104 		if (!gelf_getrel(data, i, &rel)) {
7105 			pr_warn("struct_ops reloc: failed to get %d reloc\n", i);
7106 			return -LIBBPF_ERRNO__FORMAT;
7107 		}
7108 
7109 		if (!gelf_getsym(symbols, GELF_R_SYM(rel.r_info), &sym)) {
7110 			pr_warn("struct_ops reloc: symbol %zx not found\n",
7111 				(size_t)GELF_R_SYM(rel.r_info));
7112 			return -LIBBPF_ERRNO__FORMAT;
7113 		}
7114 
7115 		name = elf_strptr(obj->efile.elf, obj->efile.strtabidx,
7116 				  sym.st_name) ? : "<?>";
7117 		map = find_struct_ops_map_by_offset(obj, rel.r_offset);
7118 		if (!map) {
7119 			pr_warn("struct_ops reloc: cannot find map at rel.r_offset %zu\n",
7120 				(size_t)rel.r_offset);
7121 			return -EINVAL;
7122 		}
7123 
7124 		moff = rel.r_offset - map->sec_offset;
7125 		shdr_idx = sym.st_shndx;
7126 		st_ops = map->st_ops;
7127 		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",
7128 			 map->name,
7129 			 (long long)(rel.r_info >> 32),
7130 			 (long long)sym.st_value,
7131 			 shdr_idx, (size_t)rel.r_offset,
7132 			 map->sec_offset, sym.st_name, name);
7133 
7134 		if (shdr_idx >= SHN_LORESERVE) {
7135 			pr_warn("struct_ops reloc %s: rel.r_offset %zu shdr_idx %u unsupported non-static function\n",
7136 				map->name, (size_t)rel.r_offset, shdr_idx);
7137 			return -LIBBPF_ERRNO__RELOC;
7138 		}
7139 
7140 		member = find_member_by_offset(st_ops->type, moff * 8);
7141 		if (!member) {
7142 			pr_warn("struct_ops reloc %s: cannot find member at moff %u\n",
7143 				map->name, moff);
7144 			return -EINVAL;
7145 		}
7146 		member_idx = member - btf_members(st_ops->type);
7147 		name = btf__name_by_offset(btf, member->name_off);
7148 
7149 		if (!resolve_func_ptr(btf, member->type, NULL)) {
7150 			pr_warn("struct_ops reloc %s: cannot relocate non func ptr %s\n",
7151 				map->name, name);
7152 			return -EINVAL;
7153 		}
7154 
7155 		prog = bpf_object__find_prog_by_idx(obj, shdr_idx);
7156 		if (!prog) {
7157 			pr_warn("struct_ops reloc %s: cannot find prog at shdr_idx %u to relocate func ptr %s\n",
7158 				map->name, shdr_idx, name);
7159 			return -EINVAL;
7160 		}
7161 
7162 		if (prog->type == BPF_PROG_TYPE_UNSPEC) {
7163 			const struct bpf_sec_def *sec_def;
7164 
7165 			sec_def = find_sec_def(prog->section_name);
7166 			if (sec_def &&
7167 			    sec_def->prog_type != BPF_PROG_TYPE_STRUCT_OPS) {
7168 				/* for pr_warn */
7169 				prog->type = sec_def->prog_type;
7170 				goto invalid_prog;
7171 			}
7172 
7173 			prog->type = BPF_PROG_TYPE_STRUCT_OPS;
7174 			prog->attach_btf_id = st_ops->type_id;
7175 			prog->expected_attach_type = member_idx;
7176 		} else if (prog->type != BPF_PROG_TYPE_STRUCT_OPS ||
7177 			   prog->attach_btf_id != st_ops->type_id ||
7178 			   prog->expected_attach_type != member_idx) {
7179 			goto invalid_prog;
7180 		}
7181 		st_ops->progs[member_idx] = prog;
7182 	}
7183 
7184 	return 0;
7185 
7186 invalid_prog:
7187 	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",
7188 		map->name, prog->name, prog->section_name, prog->type,
7189 		prog->attach_btf_id, prog->expected_attach_type, name);
7190 	return -EINVAL;
7191 }
7192 
7193 #define BTF_TRACE_PREFIX "btf_trace_"
7194 #define BTF_LSM_PREFIX "bpf_lsm_"
7195 #define BTF_ITER_PREFIX "bpf_iter_"
7196 #define BTF_MAX_NAME_SIZE 128
7197 
7198 static int find_btf_by_prefix_kind(const struct btf *btf, const char *prefix,
7199 				   const char *name, __u32 kind)
7200 {
7201 	char btf_type_name[BTF_MAX_NAME_SIZE];
7202 	int ret;
7203 
7204 	ret = snprintf(btf_type_name, sizeof(btf_type_name),
7205 		       "%s%s", prefix, name);
7206 	/* snprintf returns the number of characters written excluding the
7207 	 * the terminating null. So, if >= BTF_MAX_NAME_SIZE are written, it
7208 	 * indicates truncation.
7209 	 */
7210 	if (ret < 0 || ret >= sizeof(btf_type_name))
7211 		return -ENAMETOOLONG;
7212 	return btf__find_by_name_kind(btf, btf_type_name, kind);
7213 }
7214 
7215 static inline int __find_vmlinux_btf_id(struct btf *btf, const char *name,
7216 					enum bpf_attach_type attach_type)
7217 {
7218 	int err;
7219 
7220 	if (attach_type == BPF_TRACE_RAW_TP)
7221 		err = find_btf_by_prefix_kind(btf, BTF_TRACE_PREFIX, name,
7222 					      BTF_KIND_TYPEDEF);
7223 	else if (attach_type == BPF_LSM_MAC)
7224 		err = find_btf_by_prefix_kind(btf, BTF_LSM_PREFIX, name,
7225 					      BTF_KIND_FUNC);
7226 	else if (attach_type == BPF_TRACE_ITER)
7227 		err = find_btf_by_prefix_kind(btf, BTF_ITER_PREFIX, name,
7228 					      BTF_KIND_FUNC);
7229 	else
7230 		err = btf__find_by_name_kind(btf, name, BTF_KIND_FUNC);
7231 
7232 	if (err <= 0)
7233 		pr_warn("%s is not found in vmlinux BTF\n", name);
7234 
7235 	return err;
7236 }
7237 
7238 int libbpf_find_vmlinux_btf_id(const char *name,
7239 			       enum bpf_attach_type attach_type)
7240 {
7241 	struct btf *btf;
7242 	int err;
7243 
7244 	btf = libbpf_find_kernel_btf();
7245 	if (IS_ERR(btf)) {
7246 		pr_warn("vmlinux BTF is not found\n");
7247 		return -EINVAL;
7248 	}
7249 
7250 	err = __find_vmlinux_btf_id(btf, name, attach_type);
7251 	btf__free(btf);
7252 	return err;
7253 }
7254 
7255 static int libbpf_find_prog_btf_id(const char *name, __u32 attach_prog_fd)
7256 {
7257 	struct bpf_prog_info_linear *info_linear;
7258 	struct bpf_prog_info *info;
7259 	struct btf *btf = NULL;
7260 	int err = -EINVAL;
7261 
7262 	info_linear = bpf_program__get_prog_info_linear(attach_prog_fd, 0);
7263 	if (IS_ERR_OR_NULL(info_linear)) {
7264 		pr_warn("failed get_prog_info_linear for FD %d\n",
7265 			attach_prog_fd);
7266 		return -EINVAL;
7267 	}
7268 	info = &info_linear->info;
7269 	if (!info->btf_id) {
7270 		pr_warn("The target program doesn't have BTF\n");
7271 		goto out;
7272 	}
7273 	if (btf__get_from_id(info->btf_id, &btf)) {
7274 		pr_warn("Failed to get BTF of the program\n");
7275 		goto out;
7276 	}
7277 	err = btf__find_by_name_kind(btf, name, BTF_KIND_FUNC);
7278 	btf__free(btf);
7279 	if (err <= 0) {
7280 		pr_warn("%s is not found in prog's BTF\n", name);
7281 		goto out;
7282 	}
7283 out:
7284 	free(info_linear);
7285 	return err;
7286 }
7287 
7288 static int libbpf_find_attach_btf_id(struct bpf_program *prog)
7289 {
7290 	enum bpf_attach_type attach_type = prog->expected_attach_type;
7291 	__u32 attach_prog_fd = prog->attach_prog_fd;
7292 	const char *name = prog->section_name;
7293 	int i, err;
7294 
7295 	if (!name)
7296 		return -EINVAL;
7297 
7298 	for (i = 0; i < ARRAY_SIZE(section_defs); i++) {
7299 		if (!section_defs[i].is_attach_btf)
7300 			continue;
7301 		if (strncmp(name, section_defs[i].sec, section_defs[i].len))
7302 			continue;
7303 		if (attach_prog_fd)
7304 			err = libbpf_find_prog_btf_id(name + section_defs[i].len,
7305 						      attach_prog_fd);
7306 		else
7307 			err = __find_vmlinux_btf_id(prog->obj->btf_vmlinux,
7308 						    name + section_defs[i].len,
7309 						    attach_type);
7310 		return err;
7311 	}
7312 	pr_warn("failed to identify btf_id based on ELF section name '%s'\n", name);
7313 	return -ESRCH;
7314 }
7315 
7316 int libbpf_attach_type_by_name(const char *name,
7317 			       enum bpf_attach_type *attach_type)
7318 {
7319 	char *type_names;
7320 	int i;
7321 
7322 	if (!name)
7323 		return -EINVAL;
7324 
7325 	for (i = 0; i < ARRAY_SIZE(section_defs); i++) {
7326 		if (strncmp(name, section_defs[i].sec, section_defs[i].len))
7327 			continue;
7328 		if (!section_defs[i].is_attachable)
7329 			return -EINVAL;
7330 		*attach_type = section_defs[i].expected_attach_type;
7331 		return 0;
7332 	}
7333 	pr_debug("failed to guess attach type based on ELF section name '%s'\n", name);
7334 	type_names = libbpf_get_type_names(true);
7335 	if (type_names != NULL) {
7336 		pr_debug("attachable section(type) names are:%s\n", type_names);
7337 		free(type_names);
7338 	}
7339 
7340 	return -EINVAL;
7341 }
7342 
7343 int bpf_map__fd(const struct bpf_map *map)
7344 {
7345 	return map ? map->fd : -EINVAL;
7346 }
7347 
7348 const struct bpf_map_def *bpf_map__def(const struct bpf_map *map)
7349 {
7350 	return map ? &map->def : ERR_PTR(-EINVAL);
7351 }
7352 
7353 const char *bpf_map__name(const struct bpf_map *map)
7354 {
7355 	return map ? map->name : NULL;
7356 }
7357 
7358 enum bpf_map_type bpf_map__type(const struct bpf_map *map)
7359 {
7360 	return map->def.type;
7361 }
7362 
7363 int bpf_map__set_type(struct bpf_map *map, enum bpf_map_type type)
7364 {
7365 	if (map->fd >= 0)
7366 		return -EBUSY;
7367 	map->def.type = type;
7368 	return 0;
7369 }
7370 
7371 __u32 bpf_map__map_flags(const struct bpf_map *map)
7372 {
7373 	return map->def.map_flags;
7374 }
7375 
7376 int bpf_map__set_map_flags(struct bpf_map *map, __u32 flags)
7377 {
7378 	if (map->fd >= 0)
7379 		return -EBUSY;
7380 	map->def.map_flags = flags;
7381 	return 0;
7382 }
7383 
7384 __u32 bpf_map__numa_node(const struct bpf_map *map)
7385 {
7386 	return map->numa_node;
7387 }
7388 
7389 int bpf_map__set_numa_node(struct bpf_map *map, __u32 numa_node)
7390 {
7391 	if (map->fd >= 0)
7392 		return -EBUSY;
7393 	map->numa_node = numa_node;
7394 	return 0;
7395 }
7396 
7397 __u32 bpf_map__key_size(const struct bpf_map *map)
7398 {
7399 	return map->def.key_size;
7400 }
7401 
7402 int bpf_map__set_key_size(struct bpf_map *map, __u32 size)
7403 {
7404 	if (map->fd >= 0)
7405 		return -EBUSY;
7406 	map->def.key_size = size;
7407 	return 0;
7408 }
7409 
7410 __u32 bpf_map__value_size(const struct bpf_map *map)
7411 {
7412 	return map->def.value_size;
7413 }
7414 
7415 int bpf_map__set_value_size(struct bpf_map *map, __u32 size)
7416 {
7417 	if (map->fd >= 0)
7418 		return -EBUSY;
7419 	map->def.value_size = size;
7420 	return 0;
7421 }
7422 
7423 __u32 bpf_map__btf_key_type_id(const struct bpf_map *map)
7424 {
7425 	return map ? map->btf_key_type_id : 0;
7426 }
7427 
7428 __u32 bpf_map__btf_value_type_id(const struct bpf_map *map)
7429 {
7430 	return map ? map->btf_value_type_id : 0;
7431 }
7432 
7433 int bpf_map__set_priv(struct bpf_map *map, void *priv,
7434 		     bpf_map_clear_priv_t clear_priv)
7435 {
7436 	if (!map)
7437 		return -EINVAL;
7438 
7439 	if (map->priv) {
7440 		if (map->clear_priv)
7441 			map->clear_priv(map, map->priv);
7442 	}
7443 
7444 	map->priv = priv;
7445 	map->clear_priv = clear_priv;
7446 	return 0;
7447 }
7448 
7449 void *bpf_map__priv(const struct bpf_map *map)
7450 {
7451 	return map ? map->priv : ERR_PTR(-EINVAL);
7452 }
7453 
7454 int bpf_map__set_initial_value(struct bpf_map *map,
7455 			       const void *data, size_t size)
7456 {
7457 	if (!map->mmaped || map->libbpf_type == LIBBPF_MAP_KCONFIG ||
7458 	    size != map->def.value_size || map->fd >= 0)
7459 		return -EINVAL;
7460 
7461 	memcpy(map->mmaped, data, size);
7462 	return 0;
7463 }
7464 
7465 bool bpf_map__is_offload_neutral(const struct bpf_map *map)
7466 {
7467 	return map->def.type == BPF_MAP_TYPE_PERF_EVENT_ARRAY;
7468 }
7469 
7470 bool bpf_map__is_internal(const struct bpf_map *map)
7471 {
7472 	return map->libbpf_type != LIBBPF_MAP_UNSPEC;
7473 }
7474 
7475 __u32 bpf_map__ifindex(const struct bpf_map *map)
7476 {
7477 	return map->map_ifindex;
7478 }
7479 
7480 int bpf_map__set_ifindex(struct bpf_map *map, __u32 ifindex)
7481 {
7482 	if (map->fd >= 0)
7483 		return -EBUSY;
7484 	map->map_ifindex = ifindex;
7485 	return 0;
7486 }
7487 
7488 int bpf_map__set_inner_map_fd(struct bpf_map *map, int fd)
7489 {
7490 	if (!bpf_map_type__is_map_in_map(map->def.type)) {
7491 		pr_warn("error: unsupported map type\n");
7492 		return -EINVAL;
7493 	}
7494 	if (map->inner_map_fd != -1) {
7495 		pr_warn("error: inner_map_fd already specified\n");
7496 		return -EINVAL;
7497 	}
7498 	map->inner_map_fd = fd;
7499 	return 0;
7500 }
7501 
7502 static struct bpf_map *
7503 __bpf_map__iter(const struct bpf_map *m, const struct bpf_object *obj, int i)
7504 {
7505 	ssize_t idx;
7506 	struct bpf_map *s, *e;
7507 
7508 	if (!obj || !obj->maps)
7509 		return NULL;
7510 
7511 	s = obj->maps;
7512 	e = obj->maps + obj->nr_maps;
7513 
7514 	if ((m < s) || (m >= e)) {
7515 		pr_warn("error in %s: map handler doesn't belong to object\n",
7516 			 __func__);
7517 		return NULL;
7518 	}
7519 
7520 	idx = (m - obj->maps) + i;
7521 	if (idx >= obj->nr_maps || idx < 0)
7522 		return NULL;
7523 	return &obj->maps[idx];
7524 }
7525 
7526 struct bpf_map *
7527 bpf_map__next(const struct bpf_map *prev, const struct bpf_object *obj)
7528 {
7529 	if (prev == NULL)
7530 		return obj->maps;
7531 
7532 	return __bpf_map__iter(prev, obj, 1);
7533 }
7534 
7535 struct bpf_map *
7536 bpf_map__prev(const struct bpf_map *next, const struct bpf_object *obj)
7537 {
7538 	if (next == NULL) {
7539 		if (!obj->nr_maps)
7540 			return NULL;
7541 		return obj->maps + obj->nr_maps - 1;
7542 	}
7543 
7544 	return __bpf_map__iter(next, obj, -1);
7545 }
7546 
7547 struct bpf_map *
7548 bpf_object__find_map_by_name(const struct bpf_object *obj, const char *name)
7549 {
7550 	struct bpf_map *pos;
7551 
7552 	bpf_object__for_each_map(pos, obj) {
7553 		if (pos->name && !strcmp(pos->name, name))
7554 			return pos;
7555 	}
7556 	return NULL;
7557 }
7558 
7559 int
7560 bpf_object__find_map_fd_by_name(const struct bpf_object *obj, const char *name)
7561 {
7562 	return bpf_map__fd(bpf_object__find_map_by_name(obj, name));
7563 }
7564 
7565 struct bpf_map *
7566 bpf_object__find_map_by_offset(struct bpf_object *obj, size_t offset)
7567 {
7568 	return ERR_PTR(-ENOTSUP);
7569 }
7570 
7571 long libbpf_get_error(const void *ptr)
7572 {
7573 	return PTR_ERR_OR_ZERO(ptr);
7574 }
7575 
7576 int bpf_prog_load(const char *file, enum bpf_prog_type type,
7577 		  struct bpf_object **pobj, int *prog_fd)
7578 {
7579 	struct bpf_prog_load_attr attr;
7580 
7581 	memset(&attr, 0, sizeof(struct bpf_prog_load_attr));
7582 	attr.file = file;
7583 	attr.prog_type = type;
7584 	attr.expected_attach_type = 0;
7585 
7586 	return bpf_prog_load_xattr(&attr, pobj, prog_fd);
7587 }
7588 
7589 int bpf_prog_load_xattr(const struct bpf_prog_load_attr *attr,
7590 			struct bpf_object **pobj, int *prog_fd)
7591 {
7592 	struct bpf_object_open_attr open_attr = {};
7593 	struct bpf_program *prog, *first_prog = NULL;
7594 	struct bpf_object *obj;
7595 	struct bpf_map *map;
7596 	int err;
7597 
7598 	if (!attr)
7599 		return -EINVAL;
7600 	if (!attr->file)
7601 		return -EINVAL;
7602 
7603 	open_attr.file = attr->file;
7604 	open_attr.prog_type = attr->prog_type;
7605 
7606 	obj = bpf_object__open_xattr(&open_attr);
7607 	if (IS_ERR_OR_NULL(obj))
7608 		return -ENOENT;
7609 
7610 	bpf_object__for_each_program(prog, obj) {
7611 		enum bpf_attach_type attach_type = attr->expected_attach_type;
7612 		/*
7613 		 * to preserve backwards compatibility, bpf_prog_load treats
7614 		 * attr->prog_type, if specified, as an override to whatever
7615 		 * bpf_object__open guessed
7616 		 */
7617 		if (attr->prog_type != BPF_PROG_TYPE_UNSPEC) {
7618 			bpf_program__set_type(prog, attr->prog_type);
7619 			bpf_program__set_expected_attach_type(prog,
7620 							      attach_type);
7621 		}
7622 		if (bpf_program__get_type(prog) == BPF_PROG_TYPE_UNSPEC) {
7623 			/*
7624 			 * we haven't guessed from section name and user
7625 			 * didn't provide a fallback type, too bad...
7626 			 */
7627 			bpf_object__close(obj);
7628 			return -EINVAL;
7629 		}
7630 
7631 		prog->prog_ifindex = attr->ifindex;
7632 		prog->log_level = attr->log_level;
7633 		prog->prog_flags = attr->prog_flags;
7634 		if (!first_prog)
7635 			first_prog = prog;
7636 	}
7637 
7638 	bpf_object__for_each_map(map, obj) {
7639 		if (!bpf_map__is_offload_neutral(map))
7640 			map->map_ifindex = attr->ifindex;
7641 	}
7642 
7643 	if (!first_prog) {
7644 		pr_warn("object file doesn't contain bpf program\n");
7645 		bpf_object__close(obj);
7646 		return -ENOENT;
7647 	}
7648 
7649 	err = bpf_object__load(obj);
7650 	if (err) {
7651 		bpf_object__close(obj);
7652 		return err;
7653 	}
7654 
7655 	*pobj = obj;
7656 	*prog_fd = bpf_program__fd(first_prog);
7657 	return 0;
7658 }
7659 
7660 struct bpf_link {
7661 	int (*detach)(struct bpf_link *link);
7662 	int (*destroy)(struct bpf_link *link);
7663 	char *pin_path;		/* NULL, if not pinned */
7664 	int fd;			/* hook FD, -1 if not applicable */
7665 	bool disconnected;
7666 };
7667 
7668 /* Replace link's underlying BPF program with the new one */
7669 int bpf_link__update_program(struct bpf_link *link, struct bpf_program *prog)
7670 {
7671 	return bpf_link_update(bpf_link__fd(link), bpf_program__fd(prog), NULL);
7672 }
7673 
7674 /* Release "ownership" of underlying BPF resource (typically, BPF program
7675  * attached to some BPF hook, e.g., tracepoint, kprobe, etc). Disconnected
7676  * link, when destructed through bpf_link__destroy() call won't attempt to
7677  * detach/unregisted that BPF resource. This is useful in situations where,
7678  * say, attached BPF program has to outlive userspace program that attached it
7679  * in the system. Depending on type of BPF program, though, there might be
7680  * additional steps (like pinning BPF program in BPF FS) necessary to ensure
7681  * exit of userspace program doesn't trigger automatic detachment and clean up
7682  * inside the kernel.
7683  */
7684 void bpf_link__disconnect(struct bpf_link *link)
7685 {
7686 	link->disconnected = true;
7687 }
7688 
7689 int bpf_link__destroy(struct bpf_link *link)
7690 {
7691 	int err = 0;
7692 
7693 	if (IS_ERR_OR_NULL(link))
7694 		return 0;
7695 
7696 	if (!link->disconnected && link->detach)
7697 		err = link->detach(link);
7698 	if (link->destroy)
7699 		link->destroy(link);
7700 	if (link->pin_path)
7701 		free(link->pin_path);
7702 	free(link);
7703 
7704 	return err;
7705 }
7706 
7707 int bpf_link__fd(const struct bpf_link *link)
7708 {
7709 	return link->fd;
7710 }
7711 
7712 const char *bpf_link__pin_path(const struct bpf_link *link)
7713 {
7714 	return link->pin_path;
7715 }
7716 
7717 static int bpf_link__detach_fd(struct bpf_link *link)
7718 {
7719 	return close(link->fd);
7720 }
7721 
7722 struct bpf_link *bpf_link__open(const char *path)
7723 {
7724 	struct bpf_link *link;
7725 	int fd;
7726 
7727 	fd = bpf_obj_get(path);
7728 	if (fd < 0) {
7729 		fd = -errno;
7730 		pr_warn("failed to open link at %s: %d\n", path, fd);
7731 		return ERR_PTR(fd);
7732 	}
7733 
7734 	link = calloc(1, sizeof(*link));
7735 	if (!link) {
7736 		close(fd);
7737 		return ERR_PTR(-ENOMEM);
7738 	}
7739 	link->detach = &bpf_link__detach_fd;
7740 	link->fd = fd;
7741 
7742 	link->pin_path = strdup(path);
7743 	if (!link->pin_path) {
7744 		bpf_link__destroy(link);
7745 		return ERR_PTR(-ENOMEM);
7746 	}
7747 
7748 	return link;
7749 }
7750 
7751 int bpf_link__detach(struct bpf_link *link)
7752 {
7753 	return bpf_link_detach(link->fd) ? -errno : 0;
7754 }
7755 
7756 int bpf_link__pin(struct bpf_link *link, const char *path)
7757 {
7758 	int err;
7759 
7760 	if (link->pin_path)
7761 		return -EBUSY;
7762 	err = make_parent_dir(path);
7763 	if (err)
7764 		return err;
7765 	err = check_path(path);
7766 	if (err)
7767 		return err;
7768 
7769 	link->pin_path = strdup(path);
7770 	if (!link->pin_path)
7771 		return -ENOMEM;
7772 
7773 	if (bpf_obj_pin(link->fd, link->pin_path)) {
7774 		err = -errno;
7775 		zfree(&link->pin_path);
7776 		return err;
7777 	}
7778 
7779 	pr_debug("link fd=%d: pinned at %s\n", link->fd, link->pin_path);
7780 	return 0;
7781 }
7782 
7783 int bpf_link__unpin(struct bpf_link *link)
7784 {
7785 	int err;
7786 
7787 	if (!link->pin_path)
7788 		return -EINVAL;
7789 
7790 	err = unlink(link->pin_path);
7791 	if (err != 0)
7792 		return -errno;
7793 
7794 	pr_debug("link fd=%d: unpinned from %s\n", link->fd, link->pin_path);
7795 	zfree(&link->pin_path);
7796 	return 0;
7797 }
7798 
7799 static int bpf_link__detach_perf_event(struct bpf_link *link)
7800 {
7801 	int err;
7802 
7803 	err = ioctl(link->fd, PERF_EVENT_IOC_DISABLE, 0);
7804 	if (err)
7805 		err = -errno;
7806 
7807 	close(link->fd);
7808 	return err;
7809 }
7810 
7811 struct bpf_link *bpf_program__attach_perf_event(struct bpf_program *prog,
7812 						int pfd)
7813 {
7814 	char errmsg[STRERR_BUFSIZE];
7815 	struct bpf_link *link;
7816 	int prog_fd, err;
7817 
7818 	if (pfd < 0) {
7819 		pr_warn("program '%s': invalid perf event FD %d\n",
7820 			bpf_program__title(prog, false), pfd);
7821 		return ERR_PTR(-EINVAL);
7822 	}
7823 	prog_fd = bpf_program__fd(prog);
7824 	if (prog_fd < 0) {
7825 		pr_warn("program '%s': can't attach BPF program w/o FD (did you load it?)\n",
7826 			bpf_program__title(prog, false));
7827 		return ERR_PTR(-EINVAL);
7828 	}
7829 
7830 	link = calloc(1, sizeof(*link));
7831 	if (!link)
7832 		return ERR_PTR(-ENOMEM);
7833 	link->detach = &bpf_link__detach_perf_event;
7834 	link->fd = pfd;
7835 
7836 	if (ioctl(pfd, PERF_EVENT_IOC_SET_BPF, prog_fd) < 0) {
7837 		err = -errno;
7838 		free(link);
7839 		pr_warn("program '%s': failed to attach to pfd %d: %s\n",
7840 			bpf_program__title(prog, false), pfd,
7841 			   libbpf_strerror_r(err, errmsg, sizeof(errmsg)));
7842 		if (err == -EPROTO)
7843 			pr_warn("program '%s': try add PERF_SAMPLE_CALLCHAIN to or remove exclude_callchain_[kernel|user] from pfd %d\n",
7844 				bpf_program__title(prog, false), pfd);
7845 		return ERR_PTR(err);
7846 	}
7847 	if (ioctl(pfd, PERF_EVENT_IOC_ENABLE, 0) < 0) {
7848 		err = -errno;
7849 		free(link);
7850 		pr_warn("program '%s': failed to enable pfd %d: %s\n",
7851 			bpf_program__title(prog, false), pfd,
7852 			   libbpf_strerror_r(err, errmsg, sizeof(errmsg)));
7853 		return ERR_PTR(err);
7854 	}
7855 	return link;
7856 }
7857 
7858 /*
7859  * this function is expected to parse integer in the range of [0, 2^31-1] from
7860  * given file using scanf format string fmt. If actual parsed value is
7861  * negative, the result might be indistinguishable from error
7862  */
7863 static int parse_uint_from_file(const char *file, const char *fmt)
7864 {
7865 	char buf[STRERR_BUFSIZE];
7866 	int err, ret;
7867 	FILE *f;
7868 
7869 	f = fopen(file, "r");
7870 	if (!f) {
7871 		err = -errno;
7872 		pr_debug("failed to open '%s': %s\n", file,
7873 			 libbpf_strerror_r(err, buf, sizeof(buf)));
7874 		return err;
7875 	}
7876 	err = fscanf(f, fmt, &ret);
7877 	if (err != 1) {
7878 		err = err == EOF ? -EIO : -errno;
7879 		pr_debug("failed to parse '%s': %s\n", file,
7880 			libbpf_strerror_r(err, buf, sizeof(buf)));
7881 		fclose(f);
7882 		return err;
7883 	}
7884 	fclose(f);
7885 	return ret;
7886 }
7887 
7888 static int determine_kprobe_perf_type(void)
7889 {
7890 	const char *file = "/sys/bus/event_source/devices/kprobe/type";
7891 
7892 	return parse_uint_from_file(file, "%d\n");
7893 }
7894 
7895 static int determine_uprobe_perf_type(void)
7896 {
7897 	const char *file = "/sys/bus/event_source/devices/uprobe/type";
7898 
7899 	return parse_uint_from_file(file, "%d\n");
7900 }
7901 
7902 static int determine_kprobe_retprobe_bit(void)
7903 {
7904 	const char *file = "/sys/bus/event_source/devices/kprobe/format/retprobe";
7905 
7906 	return parse_uint_from_file(file, "config:%d\n");
7907 }
7908 
7909 static int determine_uprobe_retprobe_bit(void)
7910 {
7911 	const char *file = "/sys/bus/event_source/devices/uprobe/format/retprobe";
7912 
7913 	return parse_uint_from_file(file, "config:%d\n");
7914 }
7915 
7916 static int perf_event_open_probe(bool uprobe, bool retprobe, const char *name,
7917 				 uint64_t offset, int pid)
7918 {
7919 	struct perf_event_attr attr = {};
7920 	char errmsg[STRERR_BUFSIZE];
7921 	int type, pfd, err;
7922 
7923 	type = uprobe ? determine_uprobe_perf_type()
7924 		      : determine_kprobe_perf_type();
7925 	if (type < 0) {
7926 		pr_warn("failed to determine %s perf type: %s\n",
7927 			uprobe ? "uprobe" : "kprobe",
7928 			libbpf_strerror_r(type, errmsg, sizeof(errmsg)));
7929 		return type;
7930 	}
7931 	if (retprobe) {
7932 		int bit = uprobe ? determine_uprobe_retprobe_bit()
7933 				 : determine_kprobe_retprobe_bit();
7934 
7935 		if (bit < 0) {
7936 			pr_warn("failed to determine %s retprobe bit: %s\n",
7937 				uprobe ? "uprobe" : "kprobe",
7938 				libbpf_strerror_r(bit, errmsg, sizeof(errmsg)));
7939 			return bit;
7940 		}
7941 		attr.config |= 1 << bit;
7942 	}
7943 	attr.size = sizeof(attr);
7944 	attr.type = type;
7945 	attr.config1 = ptr_to_u64(name); /* kprobe_func or uprobe_path */
7946 	attr.config2 = offset;		 /* kprobe_addr or probe_offset */
7947 
7948 	/* pid filter is meaningful only for uprobes */
7949 	pfd = syscall(__NR_perf_event_open, &attr,
7950 		      pid < 0 ? -1 : pid /* pid */,
7951 		      pid == -1 ? 0 : -1 /* cpu */,
7952 		      -1 /* group_fd */, PERF_FLAG_FD_CLOEXEC);
7953 	if (pfd < 0) {
7954 		err = -errno;
7955 		pr_warn("%s perf_event_open() failed: %s\n",
7956 			uprobe ? "uprobe" : "kprobe",
7957 			libbpf_strerror_r(err, errmsg, sizeof(errmsg)));
7958 		return err;
7959 	}
7960 	return pfd;
7961 }
7962 
7963 struct bpf_link *bpf_program__attach_kprobe(struct bpf_program *prog,
7964 					    bool retprobe,
7965 					    const char *func_name)
7966 {
7967 	char errmsg[STRERR_BUFSIZE];
7968 	struct bpf_link *link;
7969 	int pfd, err;
7970 
7971 	pfd = perf_event_open_probe(false /* uprobe */, retprobe, func_name,
7972 				    0 /* offset */, -1 /* pid */);
7973 	if (pfd < 0) {
7974 		pr_warn("program '%s': failed to create %s '%s' perf event: %s\n",
7975 			bpf_program__title(prog, false),
7976 			retprobe ? "kretprobe" : "kprobe", func_name,
7977 			libbpf_strerror_r(pfd, errmsg, sizeof(errmsg)));
7978 		return ERR_PTR(pfd);
7979 	}
7980 	link = bpf_program__attach_perf_event(prog, pfd);
7981 	if (IS_ERR(link)) {
7982 		close(pfd);
7983 		err = PTR_ERR(link);
7984 		pr_warn("program '%s': failed to attach to %s '%s': %s\n",
7985 			bpf_program__title(prog, false),
7986 			retprobe ? "kretprobe" : "kprobe", func_name,
7987 			libbpf_strerror_r(err, errmsg, sizeof(errmsg)));
7988 		return link;
7989 	}
7990 	return link;
7991 }
7992 
7993 static struct bpf_link *attach_kprobe(const struct bpf_sec_def *sec,
7994 				      struct bpf_program *prog)
7995 {
7996 	const char *func_name;
7997 	bool retprobe;
7998 
7999 	func_name = bpf_program__title(prog, false) + sec->len;
8000 	retprobe = strcmp(sec->sec, "kretprobe/") == 0;
8001 
8002 	return bpf_program__attach_kprobe(prog, retprobe, func_name);
8003 }
8004 
8005 struct bpf_link *bpf_program__attach_uprobe(struct bpf_program *prog,
8006 					    bool retprobe, pid_t pid,
8007 					    const char *binary_path,
8008 					    size_t func_offset)
8009 {
8010 	char errmsg[STRERR_BUFSIZE];
8011 	struct bpf_link *link;
8012 	int pfd, err;
8013 
8014 	pfd = perf_event_open_probe(true /* uprobe */, retprobe,
8015 				    binary_path, func_offset, pid);
8016 	if (pfd < 0) {
8017 		pr_warn("program '%s': failed to create %s '%s:0x%zx' perf event: %s\n",
8018 			bpf_program__title(prog, false),
8019 			retprobe ? "uretprobe" : "uprobe",
8020 			binary_path, func_offset,
8021 			libbpf_strerror_r(pfd, errmsg, sizeof(errmsg)));
8022 		return ERR_PTR(pfd);
8023 	}
8024 	link = bpf_program__attach_perf_event(prog, pfd);
8025 	if (IS_ERR(link)) {
8026 		close(pfd);
8027 		err = PTR_ERR(link);
8028 		pr_warn("program '%s': failed to attach to %s '%s:0x%zx': %s\n",
8029 			bpf_program__title(prog, false),
8030 			retprobe ? "uretprobe" : "uprobe",
8031 			binary_path, func_offset,
8032 			libbpf_strerror_r(err, errmsg, sizeof(errmsg)));
8033 		return link;
8034 	}
8035 	return link;
8036 }
8037 
8038 static int determine_tracepoint_id(const char *tp_category,
8039 				   const char *tp_name)
8040 {
8041 	char file[PATH_MAX];
8042 	int ret;
8043 
8044 	ret = snprintf(file, sizeof(file),
8045 		       "/sys/kernel/debug/tracing/events/%s/%s/id",
8046 		       tp_category, tp_name);
8047 	if (ret < 0)
8048 		return -errno;
8049 	if (ret >= sizeof(file)) {
8050 		pr_debug("tracepoint %s/%s path is too long\n",
8051 			 tp_category, tp_name);
8052 		return -E2BIG;
8053 	}
8054 	return parse_uint_from_file(file, "%d\n");
8055 }
8056 
8057 static int perf_event_open_tracepoint(const char *tp_category,
8058 				      const char *tp_name)
8059 {
8060 	struct perf_event_attr attr = {};
8061 	char errmsg[STRERR_BUFSIZE];
8062 	int tp_id, pfd, err;
8063 
8064 	tp_id = determine_tracepoint_id(tp_category, tp_name);
8065 	if (tp_id < 0) {
8066 		pr_warn("failed to determine tracepoint '%s/%s' perf event ID: %s\n",
8067 			tp_category, tp_name,
8068 			libbpf_strerror_r(tp_id, errmsg, sizeof(errmsg)));
8069 		return tp_id;
8070 	}
8071 
8072 	attr.type = PERF_TYPE_TRACEPOINT;
8073 	attr.size = sizeof(attr);
8074 	attr.config = tp_id;
8075 
8076 	pfd = syscall(__NR_perf_event_open, &attr, -1 /* pid */, 0 /* cpu */,
8077 		      -1 /* group_fd */, PERF_FLAG_FD_CLOEXEC);
8078 	if (pfd < 0) {
8079 		err = -errno;
8080 		pr_warn("tracepoint '%s/%s' perf_event_open() failed: %s\n",
8081 			tp_category, tp_name,
8082 			libbpf_strerror_r(err, errmsg, sizeof(errmsg)));
8083 		return err;
8084 	}
8085 	return pfd;
8086 }
8087 
8088 struct bpf_link *bpf_program__attach_tracepoint(struct bpf_program *prog,
8089 						const char *tp_category,
8090 						const char *tp_name)
8091 {
8092 	char errmsg[STRERR_BUFSIZE];
8093 	struct bpf_link *link;
8094 	int pfd, err;
8095 
8096 	pfd = perf_event_open_tracepoint(tp_category, tp_name);
8097 	if (pfd < 0) {
8098 		pr_warn("program '%s': failed to create tracepoint '%s/%s' perf event: %s\n",
8099 			bpf_program__title(prog, false),
8100 			tp_category, tp_name,
8101 			libbpf_strerror_r(pfd, errmsg, sizeof(errmsg)));
8102 		return ERR_PTR(pfd);
8103 	}
8104 	link = bpf_program__attach_perf_event(prog, pfd);
8105 	if (IS_ERR(link)) {
8106 		close(pfd);
8107 		err = PTR_ERR(link);
8108 		pr_warn("program '%s': failed to attach to tracepoint '%s/%s': %s\n",
8109 			bpf_program__title(prog, false),
8110 			tp_category, tp_name,
8111 			libbpf_strerror_r(err, errmsg, sizeof(errmsg)));
8112 		return link;
8113 	}
8114 	return link;
8115 }
8116 
8117 static struct bpf_link *attach_tp(const struct bpf_sec_def *sec,
8118 				  struct bpf_program *prog)
8119 {
8120 	char *sec_name, *tp_cat, *tp_name;
8121 	struct bpf_link *link;
8122 
8123 	sec_name = strdup(bpf_program__title(prog, false));
8124 	if (!sec_name)
8125 		return ERR_PTR(-ENOMEM);
8126 
8127 	/* extract "tp/<category>/<name>" */
8128 	tp_cat = sec_name + sec->len;
8129 	tp_name = strchr(tp_cat, '/');
8130 	if (!tp_name) {
8131 		link = ERR_PTR(-EINVAL);
8132 		goto out;
8133 	}
8134 	*tp_name = '\0';
8135 	tp_name++;
8136 
8137 	link = bpf_program__attach_tracepoint(prog, tp_cat, tp_name);
8138 out:
8139 	free(sec_name);
8140 	return link;
8141 }
8142 
8143 struct bpf_link *bpf_program__attach_raw_tracepoint(struct bpf_program *prog,
8144 						    const char *tp_name)
8145 {
8146 	char errmsg[STRERR_BUFSIZE];
8147 	struct bpf_link *link;
8148 	int prog_fd, pfd;
8149 
8150 	prog_fd = bpf_program__fd(prog);
8151 	if (prog_fd < 0) {
8152 		pr_warn("program '%s': can't attach before loaded\n",
8153 			bpf_program__title(prog, false));
8154 		return ERR_PTR(-EINVAL);
8155 	}
8156 
8157 	link = calloc(1, sizeof(*link));
8158 	if (!link)
8159 		return ERR_PTR(-ENOMEM);
8160 	link->detach = &bpf_link__detach_fd;
8161 
8162 	pfd = bpf_raw_tracepoint_open(tp_name, prog_fd);
8163 	if (pfd < 0) {
8164 		pfd = -errno;
8165 		free(link);
8166 		pr_warn("program '%s': failed to attach to raw tracepoint '%s': %s\n",
8167 			bpf_program__title(prog, false), tp_name,
8168 			libbpf_strerror_r(pfd, errmsg, sizeof(errmsg)));
8169 		return ERR_PTR(pfd);
8170 	}
8171 	link->fd = pfd;
8172 	return link;
8173 }
8174 
8175 static struct bpf_link *attach_raw_tp(const struct bpf_sec_def *sec,
8176 				      struct bpf_program *prog)
8177 {
8178 	const char *tp_name = bpf_program__title(prog, false) + sec->len;
8179 
8180 	return bpf_program__attach_raw_tracepoint(prog, tp_name);
8181 }
8182 
8183 /* Common logic for all BPF program types that attach to a btf_id */
8184 static struct bpf_link *bpf_program__attach_btf_id(struct bpf_program *prog)
8185 {
8186 	char errmsg[STRERR_BUFSIZE];
8187 	struct bpf_link *link;
8188 	int prog_fd, pfd;
8189 
8190 	prog_fd = bpf_program__fd(prog);
8191 	if (prog_fd < 0) {
8192 		pr_warn("program '%s': can't attach before loaded\n",
8193 			bpf_program__title(prog, false));
8194 		return ERR_PTR(-EINVAL);
8195 	}
8196 
8197 	link = calloc(1, sizeof(*link));
8198 	if (!link)
8199 		return ERR_PTR(-ENOMEM);
8200 	link->detach = &bpf_link__detach_fd;
8201 
8202 	pfd = bpf_raw_tracepoint_open(NULL, prog_fd);
8203 	if (pfd < 0) {
8204 		pfd = -errno;
8205 		free(link);
8206 		pr_warn("program '%s': failed to attach: %s\n",
8207 			bpf_program__title(prog, false),
8208 			libbpf_strerror_r(pfd, errmsg, sizeof(errmsg)));
8209 		return ERR_PTR(pfd);
8210 	}
8211 	link->fd = pfd;
8212 	return (struct bpf_link *)link;
8213 }
8214 
8215 struct bpf_link *bpf_program__attach_trace(struct bpf_program *prog)
8216 {
8217 	return bpf_program__attach_btf_id(prog);
8218 }
8219 
8220 struct bpf_link *bpf_program__attach_lsm(struct bpf_program *prog)
8221 {
8222 	return bpf_program__attach_btf_id(prog);
8223 }
8224 
8225 static struct bpf_link *attach_trace(const struct bpf_sec_def *sec,
8226 				     struct bpf_program *prog)
8227 {
8228 	return bpf_program__attach_trace(prog);
8229 }
8230 
8231 static struct bpf_link *attach_lsm(const struct bpf_sec_def *sec,
8232 				   struct bpf_program *prog)
8233 {
8234 	return bpf_program__attach_lsm(prog);
8235 }
8236 
8237 static struct bpf_link *attach_iter(const struct bpf_sec_def *sec,
8238 				    struct bpf_program *prog)
8239 {
8240 	return bpf_program__attach_iter(prog, NULL);
8241 }
8242 
8243 static struct bpf_link *
8244 bpf_program__attach_fd(struct bpf_program *prog, int target_fd,
8245 		       const char *target_name)
8246 {
8247 	enum bpf_attach_type attach_type;
8248 	char errmsg[STRERR_BUFSIZE];
8249 	struct bpf_link *link;
8250 	int prog_fd, link_fd;
8251 
8252 	prog_fd = bpf_program__fd(prog);
8253 	if (prog_fd < 0) {
8254 		pr_warn("program '%s': can't attach before loaded\n",
8255 			bpf_program__title(prog, false));
8256 		return ERR_PTR(-EINVAL);
8257 	}
8258 
8259 	link = calloc(1, sizeof(*link));
8260 	if (!link)
8261 		return ERR_PTR(-ENOMEM);
8262 	link->detach = &bpf_link__detach_fd;
8263 
8264 	attach_type = bpf_program__get_expected_attach_type(prog);
8265 	link_fd = bpf_link_create(prog_fd, target_fd, attach_type, NULL);
8266 	if (link_fd < 0) {
8267 		link_fd = -errno;
8268 		free(link);
8269 		pr_warn("program '%s': failed to attach to %s: %s\n",
8270 			bpf_program__title(prog, false), target_name,
8271 			libbpf_strerror_r(link_fd, errmsg, sizeof(errmsg)));
8272 		return ERR_PTR(link_fd);
8273 	}
8274 	link->fd = link_fd;
8275 	return link;
8276 }
8277 
8278 struct bpf_link *
8279 bpf_program__attach_cgroup(struct bpf_program *prog, int cgroup_fd)
8280 {
8281 	return bpf_program__attach_fd(prog, cgroup_fd, "cgroup");
8282 }
8283 
8284 struct bpf_link *
8285 bpf_program__attach_netns(struct bpf_program *prog, int netns_fd)
8286 {
8287 	return bpf_program__attach_fd(prog, netns_fd, "netns");
8288 }
8289 
8290 struct bpf_link *bpf_program__attach_xdp(struct bpf_program *prog, int ifindex)
8291 {
8292 	/* target_fd/target_ifindex use the same field in LINK_CREATE */
8293 	return bpf_program__attach_fd(prog, ifindex, "xdp");
8294 }
8295 
8296 struct bpf_link *
8297 bpf_program__attach_iter(struct bpf_program *prog,
8298 			 const struct bpf_iter_attach_opts *opts)
8299 {
8300 	DECLARE_LIBBPF_OPTS(bpf_link_create_opts, link_create_opts);
8301 	char errmsg[STRERR_BUFSIZE];
8302 	struct bpf_link *link;
8303 	int prog_fd, link_fd;
8304 	__u32 target_fd = 0;
8305 
8306 	if (!OPTS_VALID(opts, bpf_iter_attach_opts))
8307 		return ERR_PTR(-EINVAL);
8308 
8309 	if (OPTS_HAS(opts, map_fd)) {
8310 		target_fd = opts->map_fd;
8311 		link_create_opts.flags = BPF_ITER_LINK_MAP_FD;
8312 	}
8313 
8314 	prog_fd = bpf_program__fd(prog);
8315 	if (prog_fd < 0) {
8316 		pr_warn("program '%s': can't attach before loaded\n",
8317 			bpf_program__title(prog, false));
8318 		return ERR_PTR(-EINVAL);
8319 	}
8320 
8321 	link = calloc(1, sizeof(*link));
8322 	if (!link)
8323 		return ERR_PTR(-ENOMEM);
8324 	link->detach = &bpf_link__detach_fd;
8325 
8326 	link_fd = bpf_link_create(prog_fd, target_fd, BPF_TRACE_ITER,
8327 				  &link_create_opts);
8328 	if (link_fd < 0) {
8329 		link_fd = -errno;
8330 		free(link);
8331 		pr_warn("program '%s': failed to attach to iterator: %s\n",
8332 			bpf_program__title(prog, false),
8333 			libbpf_strerror_r(link_fd, errmsg, sizeof(errmsg)));
8334 		return ERR_PTR(link_fd);
8335 	}
8336 	link->fd = link_fd;
8337 	return link;
8338 }
8339 
8340 struct bpf_link *bpf_program__attach(struct bpf_program *prog)
8341 {
8342 	const struct bpf_sec_def *sec_def;
8343 
8344 	sec_def = find_sec_def(bpf_program__title(prog, false));
8345 	if (!sec_def || !sec_def->attach_fn)
8346 		return ERR_PTR(-ESRCH);
8347 
8348 	return sec_def->attach_fn(sec_def, prog);
8349 }
8350 
8351 static int bpf_link__detach_struct_ops(struct bpf_link *link)
8352 {
8353 	__u32 zero = 0;
8354 
8355 	if (bpf_map_delete_elem(link->fd, &zero))
8356 		return -errno;
8357 
8358 	return 0;
8359 }
8360 
8361 struct bpf_link *bpf_map__attach_struct_ops(struct bpf_map *map)
8362 {
8363 	struct bpf_struct_ops *st_ops;
8364 	struct bpf_link *link;
8365 	__u32 i, zero = 0;
8366 	int err;
8367 
8368 	if (!bpf_map__is_struct_ops(map) || map->fd == -1)
8369 		return ERR_PTR(-EINVAL);
8370 
8371 	link = calloc(1, sizeof(*link));
8372 	if (!link)
8373 		return ERR_PTR(-EINVAL);
8374 
8375 	st_ops = map->st_ops;
8376 	for (i = 0; i < btf_vlen(st_ops->type); i++) {
8377 		struct bpf_program *prog = st_ops->progs[i];
8378 		void *kern_data;
8379 		int prog_fd;
8380 
8381 		if (!prog)
8382 			continue;
8383 
8384 		prog_fd = bpf_program__fd(prog);
8385 		kern_data = st_ops->kern_vdata + st_ops->kern_func_off[i];
8386 		*(unsigned long *)kern_data = prog_fd;
8387 	}
8388 
8389 	err = bpf_map_update_elem(map->fd, &zero, st_ops->kern_vdata, 0);
8390 	if (err) {
8391 		err = -errno;
8392 		free(link);
8393 		return ERR_PTR(err);
8394 	}
8395 
8396 	link->detach = bpf_link__detach_struct_ops;
8397 	link->fd = map->fd;
8398 
8399 	return link;
8400 }
8401 
8402 enum bpf_perf_event_ret
8403 bpf_perf_event_read_simple(void *mmap_mem, size_t mmap_size, size_t page_size,
8404 			   void **copy_mem, size_t *copy_size,
8405 			   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 	void *base = ((__u8 *)header) + page_size;
8411 	int ret = LIBBPF_PERF_EVENT_CONT;
8412 	struct perf_event_header *ehdr;
8413 	size_t ehdr_size;
8414 
8415 	while (data_head != data_tail) {
8416 		ehdr = base + (data_tail & (mmap_size - 1));
8417 		ehdr_size = ehdr->size;
8418 
8419 		if (((void *)ehdr) + ehdr_size > base + mmap_size) {
8420 			void *copy_start = ehdr;
8421 			size_t len_first = base + mmap_size - copy_start;
8422 			size_t len_secnd = ehdr_size - len_first;
8423 
8424 			if (*copy_size < ehdr_size) {
8425 				free(*copy_mem);
8426 				*copy_mem = malloc(ehdr_size);
8427 				if (!*copy_mem) {
8428 					*copy_size = 0;
8429 					ret = LIBBPF_PERF_EVENT_ERROR;
8430 					break;
8431 				}
8432 				*copy_size = ehdr_size;
8433 			}
8434 
8435 			memcpy(*copy_mem, copy_start, len_first);
8436 			memcpy(*copy_mem + len_first, base, len_secnd);
8437 			ehdr = *copy_mem;
8438 		}
8439 
8440 		ret = fn(ehdr, private_data);
8441 		data_tail += ehdr_size;
8442 		if (ret != LIBBPF_PERF_EVENT_CONT)
8443 			break;
8444 	}
8445 
8446 	ring_buffer_write_tail(header, data_tail);
8447 	return ret;
8448 }
8449 
8450 struct perf_buffer;
8451 
8452 struct perf_buffer_params {
8453 	struct perf_event_attr *attr;
8454 	/* if event_cb is specified, it takes precendence */
8455 	perf_buffer_event_fn event_cb;
8456 	/* sample_cb and lost_cb are higher-level common-case callbacks */
8457 	perf_buffer_sample_fn sample_cb;
8458 	perf_buffer_lost_fn lost_cb;
8459 	void *ctx;
8460 	int cpu_cnt;
8461 	int *cpus;
8462 	int *map_keys;
8463 };
8464 
8465 struct perf_cpu_buf {
8466 	struct perf_buffer *pb;
8467 	void *base; /* mmap()'ed memory */
8468 	void *buf; /* for reconstructing segmented data */
8469 	size_t buf_size;
8470 	int fd;
8471 	int cpu;
8472 	int map_key;
8473 };
8474 
8475 struct perf_buffer {
8476 	perf_buffer_event_fn event_cb;
8477 	perf_buffer_sample_fn sample_cb;
8478 	perf_buffer_lost_fn lost_cb;
8479 	void *ctx; /* passed into callbacks */
8480 
8481 	size_t page_size;
8482 	size_t mmap_size;
8483 	struct perf_cpu_buf **cpu_bufs;
8484 	struct epoll_event *events;
8485 	int cpu_cnt; /* number of allocated CPU buffers */
8486 	int epoll_fd; /* perf event FD */
8487 	int map_fd; /* BPF_MAP_TYPE_PERF_EVENT_ARRAY BPF map FD */
8488 };
8489 
8490 static void perf_buffer__free_cpu_buf(struct perf_buffer *pb,
8491 				      struct perf_cpu_buf *cpu_buf)
8492 {
8493 	if (!cpu_buf)
8494 		return;
8495 	if (cpu_buf->base &&
8496 	    munmap(cpu_buf->base, pb->mmap_size + pb->page_size))
8497 		pr_warn("failed to munmap cpu_buf #%d\n", cpu_buf->cpu);
8498 	if (cpu_buf->fd >= 0) {
8499 		ioctl(cpu_buf->fd, PERF_EVENT_IOC_DISABLE, 0);
8500 		close(cpu_buf->fd);
8501 	}
8502 	free(cpu_buf->buf);
8503 	free(cpu_buf);
8504 }
8505 
8506 void perf_buffer__free(struct perf_buffer *pb)
8507 {
8508 	int i;
8509 
8510 	if (IS_ERR_OR_NULL(pb))
8511 		return;
8512 	if (pb->cpu_bufs) {
8513 		for (i = 0; i < pb->cpu_cnt; i++) {
8514 			struct perf_cpu_buf *cpu_buf = pb->cpu_bufs[i];
8515 
8516 			if (!cpu_buf)
8517 				continue;
8518 
8519 			bpf_map_delete_elem(pb->map_fd, &cpu_buf->map_key);
8520 			perf_buffer__free_cpu_buf(pb, cpu_buf);
8521 		}
8522 		free(pb->cpu_bufs);
8523 	}
8524 	if (pb->epoll_fd >= 0)
8525 		close(pb->epoll_fd);
8526 	free(pb->events);
8527 	free(pb);
8528 }
8529 
8530 static struct perf_cpu_buf *
8531 perf_buffer__open_cpu_buf(struct perf_buffer *pb, struct perf_event_attr *attr,
8532 			  int cpu, int map_key)
8533 {
8534 	struct perf_cpu_buf *cpu_buf;
8535 	char msg[STRERR_BUFSIZE];
8536 	int err;
8537 
8538 	cpu_buf = calloc(1, sizeof(*cpu_buf));
8539 	if (!cpu_buf)
8540 		return ERR_PTR(-ENOMEM);
8541 
8542 	cpu_buf->pb = pb;
8543 	cpu_buf->cpu = cpu;
8544 	cpu_buf->map_key = map_key;
8545 
8546 	cpu_buf->fd = syscall(__NR_perf_event_open, attr, -1 /* pid */, cpu,
8547 			      -1, PERF_FLAG_FD_CLOEXEC);
8548 	if (cpu_buf->fd < 0) {
8549 		err = -errno;
8550 		pr_warn("failed to open perf buffer event on cpu #%d: %s\n",
8551 			cpu, libbpf_strerror_r(err, msg, sizeof(msg)));
8552 		goto error;
8553 	}
8554 
8555 	cpu_buf->base = mmap(NULL, pb->mmap_size + pb->page_size,
8556 			     PROT_READ | PROT_WRITE, MAP_SHARED,
8557 			     cpu_buf->fd, 0);
8558 	if (cpu_buf->base == MAP_FAILED) {
8559 		cpu_buf->base = NULL;
8560 		err = -errno;
8561 		pr_warn("failed to mmap perf buffer on cpu #%d: %s\n",
8562 			cpu, libbpf_strerror_r(err, msg, sizeof(msg)));
8563 		goto error;
8564 	}
8565 
8566 	if (ioctl(cpu_buf->fd, PERF_EVENT_IOC_ENABLE, 0) < 0) {
8567 		err = -errno;
8568 		pr_warn("failed to enable perf buffer event on cpu #%d: %s\n",
8569 			cpu, libbpf_strerror_r(err, msg, sizeof(msg)));
8570 		goto error;
8571 	}
8572 
8573 	return cpu_buf;
8574 
8575 error:
8576 	perf_buffer__free_cpu_buf(pb, cpu_buf);
8577 	return (struct perf_cpu_buf *)ERR_PTR(err);
8578 }
8579 
8580 static struct perf_buffer *__perf_buffer__new(int map_fd, size_t page_cnt,
8581 					      struct perf_buffer_params *p);
8582 
8583 struct perf_buffer *perf_buffer__new(int map_fd, size_t page_cnt,
8584 				     const struct perf_buffer_opts *opts)
8585 {
8586 	struct perf_buffer_params p = {};
8587 	struct perf_event_attr attr = { 0, };
8588 
8589 	attr.config = PERF_COUNT_SW_BPF_OUTPUT,
8590 	attr.type = PERF_TYPE_SOFTWARE;
8591 	attr.sample_type = PERF_SAMPLE_RAW;
8592 	attr.sample_period = 1;
8593 	attr.wakeup_events = 1;
8594 
8595 	p.attr = &attr;
8596 	p.sample_cb = opts ? opts->sample_cb : NULL;
8597 	p.lost_cb = opts ? opts->lost_cb : NULL;
8598 	p.ctx = opts ? opts->ctx : NULL;
8599 
8600 	return __perf_buffer__new(map_fd, page_cnt, &p);
8601 }
8602 
8603 struct perf_buffer *
8604 perf_buffer__new_raw(int map_fd, size_t page_cnt,
8605 		     const struct perf_buffer_raw_opts *opts)
8606 {
8607 	struct perf_buffer_params p = {};
8608 
8609 	p.attr = opts->attr;
8610 	p.event_cb = opts->event_cb;
8611 	p.ctx = opts->ctx;
8612 	p.cpu_cnt = opts->cpu_cnt;
8613 	p.cpus = opts->cpus;
8614 	p.map_keys = opts->map_keys;
8615 
8616 	return __perf_buffer__new(map_fd, page_cnt, &p);
8617 }
8618 
8619 static struct perf_buffer *__perf_buffer__new(int map_fd, size_t page_cnt,
8620 					      struct perf_buffer_params *p)
8621 {
8622 	const char *online_cpus_file = "/sys/devices/system/cpu/online";
8623 	struct bpf_map_info map;
8624 	char msg[STRERR_BUFSIZE];
8625 	struct perf_buffer *pb;
8626 	bool *online = NULL;
8627 	__u32 map_info_len;
8628 	int err, i, j, n;
8629 
8630 	if (page_cnt & (page_cnt - 1)) {
8631 		pr_warn("page count should be power of two, but is %zu\n",
8632 			page_cnt);
8633 		return ERR_PTR(-EINVAL);
8634 	}
8635 
8636 	/* best-effort sanity checks */
8637 	memset(&map, 0, sizeof(map));
8638 	map_info_len = sizeof(map);
8639 	err = bpf_obj_get_info_by_fd(map_fd, &map, &map_info_len);
8640 	if (err) {
8641 		err = -errno;
8642 		/* if BPF_OBJ_GET_INFO_BY_FD is supported, will return
8643 		 * -EBADFD, -EFAULT, or -E2BIG on real error
8644 		 */
8645 		if (err != -EINVAL) {
8646 			pr_warn("failed to get map info for map FD %d: %s\n",
8647 				map_fd, libbpf_strerror_r(err, msg, sizeof(msg)));
8648 			return ERR_PTR(err);
8649 		}
8650 		pr_debug("failed to get map info for FD %d; API not supported? Ignoring...\n",
8651 			 map_fd);
8652 	} else {
8653 		if (map.type != BPF_MAP_TYPE_PERF_EVENT_ARRAY) {
8654 			pr_warn("map '%s' should be BPF_MAP_TYPE_PERF_EVENT_ARRAY\n",
8655 				map.name);
8656 			return ERR_PTR(-EINVAL);
8657 		}
8658 	}
8659 
8660 	pb = calloc(1, sizeof(*pb));
8661 	if (!pb)
8662 		return ERR_PTR(-ENOMEM);
8663 
8664 	pb->event_cb = p->event_cb;
8665 	pb->sample_cb = p->sample_cb;
8666 	pb->lost_cb = p->lost_cb;
8667 	pb->ctx = p->ctx;
8668 
8669 	pb->page_size = getpagesize();
8670 	pb->mmap_size = pb->page_size * page_cnt;
8671 	pb->map_fd = map_fd;
8672 
8673 	pb->epoll_fd = epoll_create1(EPOLL_CLOEXEC);
8674 	if (pb->epoll_fd < 0) {
8675 		err = -errno;
8676 		pr_warn("failed to create epoll instance: %s\n",
8677 			libbpf_strerror_r(err, msg, sizeof(msg)));
8678 		goto error;
8679 	}
8680 
8681 	if (p->cpu_cnt > 0) {
8682 		pb->cpu_cnt = p->cpu_cnt;
8683 	} else {
8684 		pb->cpu_cnt = libbpf_num_possible_cpus();
8685 		if (pb->cpu_cnt < 0) {
8686 			err = pb->cpu_cnt;
8687 			goto error;
8688 		}
8689 		if (map.max_entries && map.max_entries < pb->cpu_cnt)
8690 			pb->cpu_cnt = map.max_entries;
8691 	}
8692 
8693 	pb->events = calloc(pb->cpu_cnt, sizeof(*pb->events));
8694 	if (!pb->events) {
8695 		err = -ENOMEM;
8696 		pr_warn("failed to allocate events: out of memory\n");
8697 		goto error;
8698 	}
8699 	pb->cpu_bufs = calloc(pb->cpu_cnt, sizeof(*pb->cpu_bufs));
8700 	if (!pb->cpu_bufs) {
8701 		err = -ENOMEM;
8702 		pr_warn("failed to allocate buffers: out of memory\n");
8703 		goto error;
8704 	}
8705 
8706 	err = parse_cpu_mask_file(online_cpus_file, &online, &n);
8707 	if (err) {
8708 		pr_warn("failed to get online CPU mask: %d\n", err);
8709 		goto error;
8710 	}
8711 
8712 	for (i = 0, j = 0; i < pb->cpu_cnt; i++) {
8713 		struct perf_cpu_buf *cpu_buf;
8714 		int cpu, map_key;
8715 
8716 		cpu = p->cpu_cnt > 0 ? p->cpus[i] : i;
8717 		map_key = p->cpu_cnt > 0 ? p->map_keys[i] : i;
8718 
8719 		/* in case user didn't explicitly requested particular CPUs to
8720 		 * be attached to, skip offline/not present CPUs
8721 		 */
8722 		if (p->cpu_cnt <= 0 && (cpu >= n || !online[cpu]))
8723 			continue;
8724 
8725 		cpu_buf = perf_buffer__open_cpu_buf(pb, p->attr, cpu, map_key);
8726 		if (IS_ERR(cpu_buf)) {
8727 			err = PTR_ERR(cpu_buf);
8728 			goto error;
8729 		}
8730 
8731 		pb->cpu_bufs[j] = cpu_buf;
8732 
8733 		err = bpf_map_update_elem(pb->map_fd, &map_key,
8734 					  &cpu_buf->fd, 0);
8735 		if (err) {
8736 			err = -errno;
8737 			pr_warn("failed to set cpu #%d, key %d -> perf FD %d: %s\n",
8738 				cpu, map_key, cpu_buf->fd,
8739 				libbpf_strerror_r(err, msg, sizeof(msg)));
8740 			goto error;
8741 		}
8742 
8743 		pb->events[j].events = EPOLLIN;
8744 		pb->events[j].data.ptr = cpu_buf;
8745 		if (epoll_ctl(pb->epoll_fd, EPOLL_CTL_ADD, cpu_buf->fd,
8746 			      &pb->events[j]) < 0) {
8747 			err = -errno;
8748 			pr_warn("failed to epoll_ctl cpu #%d perf FD %d: %s\n",
8749 				cpu, cpu_buf->fd,
8750 				libbpf_strerror_r(err, msg, sizeof(msg)));
8751 			goto error;
8752 		}
8753 		j++;
8754 	}
8755 	pb->cpu_cnt = j;
8756 	free(online);
8757 
8758 	return pb;
8759 
8760 error:
8761 	free(online);
8762 	if (pb)
8763 		perf_buffer__free(pb);
8764 	return ERR_PTR(err);
8765 }
8766 
8767 struct perf_sample_raw {
8768 	struct perf_event_header header;
8769 	uint32_t size;
8770 	char data[];
8771 };
8772 
8773 struct perf_sample_lost {
8774 	struct perf_event_header header;
8775 	uint64_t id;
8776 	uint64_t lost;
8777 	uint64_t sample_id;
8778 };
8779 
8780 static enum bpf_perf_event_ret
8781 perf_buffer__process_record(struct perf_event_header *e, void *ctx)
8782 {
8783 	struct perf_cpu_buf *cpu_buf = ctx;
8784 	struct perf_buffer *pb = cpu_buf->pb;
8785 	void *data = e;
8786 
8787 	/* user wants full control over parsing perf event */
8788 	if (pb->event_cb)
8789 		return pb->event_cb(pb->ctx, cpu_buf->cpu, e);
8790 
8791 	switch (e->type) {
8792 	case PERF_RECORD_SAMPLE: {
8793 		struct perf_sample_raw *s = data;
8794 
8795 		if (pb->sample_cb)
8796 			pb->sample_cb(pb->ctx, cpu_buf->cpu, s->data, s->size);
8797 		break;
8798 	}
8799 	case PERF_RECORD_LOST: {
8800 		struct perf_sample_lost *s = data;
8801 
8802 		if (pb->lost_cb)
8803 			pb->lost_cb(pb->ctx, cpu_buf->cpu, s->lost);
8804 		break;
8805 	}
8806 	default:
8807 		pr_warn("unknown perf sample type %d\n", e->type);
8808 		return LIBBPF_PERF_EVENT_ERROR;
8809 	}
8810 	return LIBBPF_PERF_EVENT_CONT;
8811 }
8812 
8813 static int perf_buffer__process_records(struct perf_buffer *pb,
8814 					struct perf_cpu_buf *cpu_buf)
8815 {
8816 	enum bpf_perf_event_ret ret;
8817 
8818 	ret = bpf_perf_event_read_simple(cpu_buf->base, pb->mmap_size,
8819 					 pb->page_size, &cpu_buf->buf,
8820 					 &cpu_buf->buf_size,
8821 					 perf_buffer__process_record, cpu_buf);
8822 	if (ret != LIBBPF_PERF_EVENT_CONT)
8823 		return ret;
8824 	return 0;
8825 }
8826 
8827 int perf_buffer__poll(struct perf_buffer *pb, int timeout_ms)
8828 {
8829 	int i, cnt, err;
8830 
8831 	cnt = epoll_wait(pb->epoll_fd, pb->events, pb->cpu_cnt, timeout_ms);
8832 	for (i = 0; i < cnt; i++) {
8833 		struct perf_cpu_buf *cpu_buf = pb->events[i].data.ptr;
8834 
8835 		err = perf_buffer__process_records(pb, cpu_buf);
8836 		if (err) {
8837 			pr_warn("error while processing records: %d\n", err);
8838 			return err;
8839 		}
8840 	}
8841 	return cnt < 0 ? -errno : cnt;
8842 }
8843 
8844 int perf_buffer__consume(struct perf_buffer *pb)
8845 {
8846 	int i, err;
8847 
8848 	for (i = 0; i < pb->cpu_cnt; i++) {
8849 		struct perf_cpu_buf *cpu_buf = pb->cpu_bufs[i];
8850 
8851 		if (!cpu_buf)
8852 			continue;
8853 
8854 		err = perf_buffer__process_records(pb, cpu_buf);
8855 		if (err) {
8856 			pr_warn("error while processing records: %d\n", err);
8857 			return err;
8858 		}
8859 	}
8860 	return 0;
8861 }
8862 
8863 struct bpf_prog_info_array_desc {
8864 	int	array_offset;	/* e.g. offset of jited_prog_insns */
8865 	int	count_offset;	/* e.g. offset of jited_prog_len */
8866 	int	size_offset;	/* > 0: offset of rec size,
8867 				 * < 0: fix size of -size_offset
8868 				 */
8869 };
8870 
8871 static struct bpf_prog_info_array_desc bpf_prog_info_array_desc[] = {
8872 	[BPF_PROG_INFO_JITED_INSNS] = {
8873 		offsetof(struct bpf_prog_info, jited_prog_insns),
8874 		offsetof(struct bpf_prog_info, jited_prog_len),
8875 		-1,
8876 	},
8877 	[BPF_PROG_INFO_XLATED_INSNS] = {
8878 		offsetof(struct bpf_prog_info, xlated_prog_insns),
8879 		offsetof(struct bpf_prog_info, xlated_prog_len),
8880 		-1,
8881 	},
8882 	[BPF_PROG_INFO_MAP_IDS] = {
8883 		offsetof(struct bpf_prog_info, map_ids),
8884 		offsetof(struct bpf_prog_info, nr_map_ids),
8885 		-(int)sizeof(__u32),
8886 	},
8887 	[BPF_PROG_INFO_JITED_KSYMS] = {
8888 		offsetof(struct bpf_prog_info, jited_ksyms),
8889 		offsetof(struct bpf_prog_info, nr_jited_ksyms),
8890 		-(int)sizeof(__u64),
8891 	},
8892 	[BPF_PROG_INFO_JITED_FUNC_LENS] = {
8893 		offsetof(struct bpf_prog_info, jited_func_lens),
8894 		offsetof(struct bpf_prog_info, nr_jited_func_lens),
8895 		-(int)sizeof(__u32),
8896 	},
8897 	[BPF_PROG_INFO_FUNC_INFO] = {
8898 		offsetof(struct bpf_prog_info, func_info),
8899 		offsetof(struct bpf_prog_info, nr_func_info),
8900 		offsetof(struct bpf_prog_info, func_info_rec_size),
8901 	},
8902 	[BPF_PROG_INFO_LINE_INFO] = {
8903 		offsetof(struct bpf_prog_info, line_info),
8904 		offsetof(struct bpf_prog_info, nr_line_info),
8905 		offsetof(struct bpf_prog_info, line_info_rec_size),
8906 	},
8907 	[BPF_PROG_INFO_JITED_LINE_INFO] = {
8908 		offsetof(struct bpf_prog_info, jited_line_info),
8909 		offsetof(struct bpf_prog_info, nr_jited_line_info),
8910 		offsetof(struct bpf_prog_info, jited_line_info_rec_size),
8911 	},
8912 	[BPF_PROG_INFO_PROG_TAGS] = {
8913 		offsetof(struct bpf_prog_info, prog_tags),
8914 		offsetof(struct bpf_prog_info, nr_prog_tags),
8915 		-(int)sizeof(__u8) * BPF_TAG_SIZE,
8916 	},
8917 
8918 };
8919 
8920 static __u32 bpf_prog_info_read_offset_u32(struct bpf_prog_info *info,
8921 					   int offset)
8922 {
8923 	__u32 *array = (__u32 *)info;
8924 
8925 	if (offset >= 0)
8926 		return array[offset / sizeof(__u32)];
8927 	return -(int)offset;
8928 }
8929 
8930 static __u64 bpf_prog_info_read_offset_u64(struct bpf_prog_info *info,
8931 					   int offset)
8932 {
8933 	__u64 *array = (__u64 *)info;
8934 
8935 	if (offset >= 0)
8936 		return array[offset / sizeof(__u64)];
8937 	return -(int)offset;
8938 }
8939 
8940 static void bpf_prog_info_set_offset_u32(struct bpf_prog_info *info, int offset,
8941 					 __u32 val)
8942 {
8943 	__u32 *array = (__u32 *)info;
8944 
8945 	if (offset >= 0)
8946 		array[offset / sizeof(__u32)] = val;
8947 }
8948 
8949 static void bpf_prog_info_set_offset_u64(struct bpf_prog_info *info, int offset,
8950 					 __u64 val)
8951 {
8952 	__u64 *array = (__u64 *)info;
8953 
8954 	if (offset >= 0)
8955 		array[offset / sizeof(__u64)] = val;
8956 }
8957 
8958 struct bpf_prog_info_linear *
8959 bpf_program__get_prog_info_linear(int fd, __u64 arrays)
8960 {
8961 	struct bpf_prog_info_linear *info_linear;
8962 	struct bpf_prog_info info = {};
8963 	__u32 info_len = sizeof(info);
8964 	__u32 data_len = 0;
8965 	int i, err;
8966 	void *ptr;
8967 
8968 	if (arrays >> BPF_PROG_INFO_LAST_ARRAY)
8969 		return ERR_PTR(-EINVAL);
8970 
8971 	/* step 1: get array dimensions */
8972 	err = bpf_obj_get_info_by_fd(fd, &info, &info_len);
8973 	if (err) {
8974 		pr_debug("can't get prog info: %s", strerror(errno));
8975 		return ERR_PTR(-EFAULT);
8976 	}
8977 
8978 	/* step 2: calculate total size of all arrays */
8979 	for (i = BPF_PROG_INFO_FIRST_ARRAY; i < BPF_PROG_INFO_LAST_ARRAY; ++i) {
8980 		bool include_array = (arrays & (1UL << i)) > 0;
8981 		struct bpf_prog_info_array_desc *desc;
8982 		__u32 count, size;
8983 
8984 		desc = bpf_prog_info_array_desc + i;
8985 
8986 		/* kernel is too old to support this field */
8987 		if (info_len < desc->array_offset + sizeof(__u32) ||
8988 		    info_len < desc->count_offset + sizeof(__u32) ||
8989 		    (desc->size_offset > 0 && info_len < desc->size_offset))
8990 			include_array = false;
8991 
8992 		if (!include_array) {
8993 			arrays &= ~(1UL << i);	/* clear the bit */
8994 			continue;
8995 		}
8996 
8997 		count = bpf_prog_info_read_offset_u32(&info, desc->count_offset);
8998 		size  = bpf_prog_info_read_offset_u32(&info, desc->size_offset);
8999 
9000 		data_len += count * size;
9001 	}
9002 
9003 	/* step 3: allocate continuous memory */
9004 	data_len = roundup(data_len, sizeof(__u64));
9005 	info_linear = malloc(sizeof(struct bpf_prog_info_linear) + data_len);
9006 	if (!info_linear)
9007 		return ERR_PTR(-ENOMEM);
9008 
9009 	/* step 4: fill data to info_linear->info */
9010 	info_linear->arrays = arrays;
9011 	memset(&info_linear->info, 0, sizeof(info));
9012 	ptr = info_linear->data;
9013 
9014 	for (i = BPF_PROG_INFO_FIRST_ARRAY; i < BPF_PROG_INFO_LAST_ARRAY; ++i) {
9015 		struct bpf_prog_info_array_desc *desc;
9016 		__u32 count, size;
9017 
9018 		if ((arrays & (1UL << i)) == 0)
9019 			continue;
9020 
9021 		desc  = bpf_prog_info_array_desc + i;
9022 		count = bpf_prog_info_read_offset_u32(&info, desc->count_offset);
9023 		size  = bpf_prog_info_read_offset_u32(&info, desc->size_offset);
9024 		bpf_prog_info_set_offset_u32(&info_linear->info,
9025 					     desc->count_offset, count);
9026 		bpf_prog_info_set_offset_u32(&info_linear->info,
9027 					     desc->size_offset, size);
9028 		bpf_prog_info_set_offset_u64(&info_linear->info,
9029 					     desc->array_offset,
9030 					     ptr_to_u64(ptr));
9031 		ptr += count * size;
9032 	}
9033 
9034 	/* step 5: call syscall again to get required arrays */
9035 	err = bpf_obj_get_info_by_fd(fd, &info_linear->info, &info_len);
9036 	if (err) {
9037 		pr_debug("can't get prog info: %s", strerror(errno));
9038 		free(info_linear);
9039 		return ERR_PTR(-EFAULT);
9040 	}
9041 
9042 	/* step 6: verify the data */
9043 	for (i = BPF_PROG_INFO_FIRST_ARRAY; i < BPF_PROG_INFO_LAST_ARRAY; ++i) {
9044 		struct bpf_prog_info_array_desc *desc;
9045 		__u32 v1, v2;
9046 
9047 		if ((arrays & (1UL << i)) == 0)
9048 			continue;
9049 
9050 		desc = bpf_prog_info_array_desc + i;
9051 		v1 = bpf_prog_info_read_offset_u32(&info, desc->count_offset);
9052 		v2 = bpf_prog_info_read_offset_u32(&info_linear->info,
9053 						   desc->count_offset);
9054 		if (v1 != v2)
9055 			pr_warn("%s: mismatch in element count\n", __func__);
9056 
9057 		v1 = bpf_prog_info_read_offset_u32(&info, desc->size_offset);
9058 		v2 = bpf_prog_info_read_offset_u32(&info_linear->info,
9059 						   desc->size_offset);
9060 		if (v1 != v2)
9061 			pr_warn("%s: mismatch in rec size\n", __func__);
9062 	}
9063 
9064 	/* step 7: update info_len and data_len */
9065 	info_linear->info_len = sizeof(struct bpf_prog_info);
9066 	info_linear->data_len = data_len;
9067 
9068 	return info_linear;
9069 }
9070 
9071 void bpf_program__bpil_addr_to_offs(struct bpf_prog_info_linear *info_linear)
9072 {
9073 	int i;
9074 
9075 	for (i = BPF_PROG_INFO_FIRST_ARRAY; i < BPF_PROG_INFO_LAST_ARRAY; ++i) {
9076 		struct bpf_prog_info_array_desc *desc;
9077 		__u64 addr, offs;
9078 
9079 		if ((info_linear->arrays & (1UL << i)) == 0)
9080 			continue;
9081 
9082 		desc = bpf_prog_info_array_desc + i;
9083 		addr = bpf_prog_info_read_offset_u64(&info_linear->info,
9084 						     desc->array_offset);
9085 		offs = addr - ptr_to_u64(info_linear->data);
9086 		bpf_prog_info_set_offset_u64(&info_linear->info,
9087 					     desc->array_offset, offs);
9088 	}
9089 }
9090 
9091 void bpf_program__bpil_offs_to_addr(struct bpf_prog_info_linear *info_linear)
9092 {
9093 	int i;
9094 
9095 	for (i = BPF_PROG_INFO_FIRST_ARRAY; i < BPF_PROG_INFO_LAST_ARRAY; ++i) {
9096 		struct bpf_prog_info_array_desc *desc;
9097 		__u64 addr, offs;
9098 
9099 		if ((info_linear->arrays & (1UL << i)) == 0)
9100 			continue;
9101 
9102 		desc = bpf_prog_info_array_desc + i;
9103 		offs = bpf_prog_info_read_offset_u64(&info_linear->info,
9104 						     desc->array_offset);
9105 		addr = offs + ptr_to_u64(info_linear->data);
9106 		bpf_prog_info_set_offset_u64(&info_linear->info,
9107 					     desc->array_offset, addr);
9108 	}
9109 }
9110 
9111 int bpf_program__set_attach_target(struct bpf_program *prog,
9112 				   int attach_prog_fd,
9113 				   const char *attach_func_name)
9114 {
9115 	int btf_id;
9116 
9117 	if (!prog || attach_prog_fd < 0 || !attach_func_name)
9118 		return -EINVAL;
9119 
9120 	if (attach_prog_fd)
9121 		btf_id = libbpf_find_prog_btf_id(attach_func_name,
9122 						 attach_prog_fd);
9123 	else
9124 		btf_id = __find_vmlinux_btf_id(prog->obj->btf_vmlinux,
9125 					       attach_func_name,
9126 					       prog->expected_attach_type);
9127 
9128 	if (btf_id < 0)
9129 		return btf_id;
9130 
9131 	prog->attach_btf_id = btf_id;
9132 	prog->attach_prog_fd = attach_prog_fd;
9133 	return 0;
9134 }
9135 
9136 int parse_cpu_mask_str(const char *s, bool **mask, int *mask_sz)
9137 {
9138 	int err = 0, n, len, start, end = -1;
9139 	bool *tmp;
9140 
9141 	*mask = NULL;
9142 	*mask_sz = 0;
9143 
9144 	/* Each sub string separated by ',' has format \d+-\d+ or \d+ */
9145 	while (*s) {
9146 		if (*s == ',' || *s == '\n') {
9147 			s++;
9148 			continue;
9149 		}
9150 		n = sscanf(s, "%d%n-%d%n", &start, &len, &end, &len);
9151 		if (n <= 0 || n > 2) {
9152 			pr_warn("Failed to get CPU range %s: %d\n", s, n);
9153 			err = -EINVAL;
9154 			goto cleanup;
9155 		} else if (n == 1) {
9156 			end = start;
9157 		}
9158 		if (start < 0 || start > end) {
9159 			pr_warn("Invalid CPU range [%d,%d] in %s\n",
9160 				start, end, s);
9161 			err = -EINVAL;
9162 			goto cleanup;
9163 		}
9164 		tmp = realloc(*mask, end + 1);
9165 		if (!tmp) {
9166 			err = -ENOMEM;
9167 			goto cleanup;
9168 		}
9169 		*mask = tmp;
9170 		memset(tmp + *mask_sz, 0, start - *mask_sz);
9171 		memset(tmp + start, 1, end - start + 1);
9172 		*mask_sz = end + 1;
9173 		s += len;
9174 	}
9175 	if (!*mask_sz) {
9176 		pr_warn("Empty CPU range\n");
9177 		return -EINVAL;
9178 	}
9179 	return 0;
9180 cleanup:
9181 	free(*mask);
9182 	*mask = NULL;
9183 	return err;
9184 }
9185 
9186 int parse_cpu_mask_file(const char *fcpu, bool **mask, int *mask_sz)
9187 {
9188 	int fd, err = 0, len;
9189 	char buf[128];
9190 
9191 	fd = open(fcpu, O_RDONLY);
9192 	if (fd < 0) {
9193 		err = -errno;
9194 		pr_warn("Failed to open cpu mask file %s: %d\n", fcpu, err);
9195 		return err;
9196 	}
9197 	len = read(fd, buf, sizeof(buf));
9198 	close(fd);
9199 	if (len <= 0) {
9200 		err = len ? -errno : -EINVAL;
9201 		pr_warn("Failed to read cpu mask from %s: %d\n", fcpu, err);
9202 		return err;
9203 	}
9204 	if (len >= sizeof(buf)) {
9205 		pr_warn("CPU mask is too big in file %s\n", fcpu);
9206 		return -E2BIG;
9207 	}
9208 	buf[len] = '\0';
9209 
9210 	return parse_cpu_mask_str(buf, mask, mask_sz);
9211 }
9212 
9213 int libbpf_num_possible_cpus(void)
9214 {
9215 	static const char *fcpu = "/sys/devices/system/cpu/possible";
9216 	static int cpus;
9217 	int err, n, i, tmp_cpus;
9218 	bool *mask;
9219 
9220 	tmp_cpus = READ_ONCE(cpus);
9221 	if (tmp_cpus > 0)
9222 		return tmp_cpus;
9223 
9224 	err = parse_cpu_mask_file(fcpu, &mask, &n);
9225 	if (err)
9226 		return err;
9227 
9228 	tmp_cpus = 0;
9229 	for (i = 0; i < n; i++) {
9230 		if (mask[i])
9231 			tmp_cpus++;
9232 	}
9233 	free(mask);
9234 
9235 	WRITE_ONCE(cpus, tmp_cpus);
9236 	return tmp_cpus;
9237 }
9238 
9239 int bpf_object__open_skeleton(struct bpf_object_skeleton *s,
9240 			      const struct bpf_object_open_opts *opts)
9241 {
9242 	DECLARE_LIBBPF_OPTS(bpf_object_open_opts, skel_opts,
9243 		.object_name = s->name,
9244 	);
9245 	struct bpf_object *obj;
9246 	int i;
9247 
9248 	/* Attempt to preserve opts->object_name, unless overriden by user
9249 	 * explicitly. Overwriting object name for skeletons is discouraged,
9250 	 * as it breaks global data maps, because they contain object name
9251 	 * prefix as their own map name prefix. When skeleton is generated,
9252 	 * bpftool is making an assumption that this name will stay the same.
9253 	 */
9254 	if (opts) {
9255 		memcpy(&skel_opts, opts, sizeof(*opts));
9256 		if (!opts->object_name)
9257 			skel_opts.object_name = s->name;
9258 	}
9259 
9260 	obj = bpf_object__open_mem(s->data, s->data_sz, &skel_opts);
9261 	if (IS_ERR(obj)) {
9262 		pr_warn("failed to initialize skeleton BPF object '%s': %ld\n",
9263 			s->name, PTR_ERR(obj));
9264 		return PTR_ERR(obj);
9265 	}
9266 
9267 	*s->obj = obj;
9268 
9269 	for (i = 0; i < s->map_cnt; i++) {
9270 		struct bpf_map **map = s->maps[i].map;
9271 		const char *name = s->maps[i].name;
9272 		void **mmaped = s->maps[i].mmaped;
9273 
9274 		*map = bpf_object__find_map_by_name(obj, name);
9275 		if (!*map) {
9276 			pr_warn("failed to find skeleton map '%s'\n", name);
9277 			return -ESRCH;
9278 		}
9279 
9280 		/* externs shouldn't be pre-setup from user code */
9281 		if (mmaped && (*map)->libbpf_type != LIBBPF_MAP_KCONFIG)
9282 			*mmaped = (*map)->mmaped;
9283 	}
9284 
9285 	for (i = 0; i < s->prog_cnt; i++) {
9286 		struct bpf_program **prog = s->progs[i].prog;
9287 		const char *name = s->progs[i].name;
9288 
9289 		*prog = bpf_object__find_program_by_name(obj, name);
9290 		if (!*prog) {
9291 			pr_warn("failed to find skeleton program '%s'\n", name);
9292 			return -ESRCH;
9293 		}
9294 	}
9295 
9296 	return 0;
9297 }
9298 
9299 int bpf_object__load_skeleton(struct bpf_object_skeleton *s)
9300 {
9301 	int i, err;
9302 
9303 	err = bpf_object__load(*s->obj);
9304 	if (err) {
9305 		pr_warn("failed to load BPF skeleton '%s': %d\n", s->name, err);
9306 		return err;
9307 	}
9308 
9309 	for (i = 0; i < s->map_cnt; i++) {
9310 		struct bpf_map *map = *s->maps[i].map;
9311 		size_t mmap_sz = bpf_map_mmap_sz(map);
9312 		int prot, map_fd = bpf_map__fd(map);
9313 		void **mmaped = s->maps[i].mmaped;
9314 
9315 		if (!mmaped)
9316 			continue;
9317 
9318 		if (!(map->def.map_flags & BPF_F_MMAPABLE)) {
9319 			*mmaped = NULL;
9320 			continue;
9321 		}
9322 
9323 		if (map->def.map_flags & BPF_F_RDONLY_PROG)
9324 			prot = PROT_READ;
9325 		else
9326 			prot = PROT_READ | PROT_WRITE;
9327 
9328 		/* Remap anonymous mmap()-ed "map initialization image" as
9329 		 * a BPF map-backed mmap()-ed memory, but preserving the same
9330 		 * memory address. This will cause kernel to change process'
9331 		 * page table to point to a different piece of kernel memory,
9332 		 * but from userspace point of view memory address (and its
9333 		 * contents, being identical at this point) will stay the
9334 		 * same. This mapping will be released by bpf_object__close()
9335 		 * as per normal clean up procedure, so we don't need to worry
9336 		 * about it from skeleton's clean up perspective.
9337 		 */
9338 		*mmaped = mmap(map->mmaped, mmap_sz, prot,
9339 				MAP_SHARED | MAP_FIXED, map_fd, 0);
9340 		if (*mmaped == MAP_FAILED) {
9341 			err = -errno;
9342 			*mmaped = NULL;
9343 			pr_warn("failed to re-mmap() map '%s': %d\n",
9344 				 bpf_map__name(map), err);
9345 			return err;
9346 		}
9347 	}
9348 
9349 	return 0;
9350 }
9351 
9352 int bpf_object__attach_skeleton(struct bpf_object_skeleton *s)
9353 {
9354 	int i;
9355 
9356 	for (i = 0; i < s->prog_cnt; i++) {
9357 		struct bpf_program *prog = *s->progs[i].prog;
9358 		struct bpf_link **link = s->progs[i].link;
9359 		const struct bpf_sec_def *sec_def;
9360 		const char *sec_name = bpf_program__title(prog, false);
9361 
9362 		if (!prog->load)
9363 			continue;
9364 
9365 		sec_def = find_sec_def(sec_name);
9366 		if (!sec_def || !sec_def->attach_fn)
9367 			continue;
9368 
9369 		*link = sec_def->attach_fn(sec_def, prog);
9370 		if (IS_ERR(*link)) {
9371 			pr_warn("failed to auto-attach program '%s': %ld\n",
9372 				bpf_program__name(prog), PTR_ERR(*link));
9373 			return PTR_ERR(*link);
9374 		}
9375 	}
9376 
9377 	return 0;
9378 }
9379 
9380 void bpf_object__detach_skeleton(struct bpf_object_skeleton *s)
9381 {
9382 	int i;
9383 
9384 	for (i = 0; i < s->prog_cnt; i++) {
9385 		struct bpf_link **link = s->progs[i].link;
9386 
9387 		bpf_link__destroy(*link);
9388 		*link = NULL;
9389 	}
9390 }
9391 
9392 void bpf_object__destroy_skeleton(struct bpf_object_skeleton *s)
9393 {
9394 	if (s->progs)
9395 		bpf_object__detach_skeleton(s);
9396 	if (s->obj)
9397 		bpf_object__close(*s->obj);
9398 	free(s->maps);
9399 	free(s->progs);
9400 	free(s);
9401 }
9402