xref: /linux/tools/lib/bpf/libbpf_internal.h (revision a7d22ca2a483d6c69c0791954447464297315ffa)
1 /* SPDX-License-Identifier: (LGPL-2.1 OR BSD-2-Clause) */
2 
3 /*
4  * Internal libbpf helpers.
5  *
6  * Copyright (c) 2019 Facebook
7  */
8 
9 #ifndef __LIBBPF_LIBBPF_INTERNAL_H
10 #define __LIBBPF_LIBBPF_INTERNAL_H
11 
12 #include "libbpf.h"
13 
14 #define BTF_INFO_ENC(kind, kind_flag, vlen) \
15 	((!!(kind_flag) << 31) | ((kind) << 24) | ((vlen) & BTF_MAX_VLEN))
16 #define BTF_TYPE_ENC(name, info, size_or_type) (name), (info), (size_or_type)
17 #define BTF_INT_ENC(encoding, bits_offset, nr_bits) \
18 	((encoding) << 24 | (bits_offset) << 16 | (nr_bits))
19 #define BTF_TYPE_INT_ENC(name, encoding, bits_offset, bits, sz) \
20 	BTF_TYPE_ENC(name, BTF_INFO_ENC(BTF_KIND_INT, 0, 0), sz), \
21 	BTF_INT_ENC(encoding, bits_offset, bits)
22 #define BTF_MEMBER_ENC(name, type, bits_offset) (name), (type), (bits_offset)
23 #define BTF_PARAM_ENC(name, type) (name), (type)
24 #define BTF_VAR_SECINFO_ENC(type, offset, size) (type), (offset), (size)
25 
26 #ifndef min
27 # define min(x, y) ((x) < (y) ? (x) : (y))
28 #endif
29 #ifndef max
30 # define max(x, y) ((x) < (y) ? (y) : (x))
31 #endif
32 #ifndef offsetofend
33 # define offsetofend(TYPE, FIELD) \
34 	(offsetof(TYPE, FIELD) + sizeof(((TYPE *)0)->FIELD))
35 #endif
36 
37 /* Symbol versioning is different between static and shared library.
38  * Properly versioned symbols are needed for shared library, but
39  * only the symbol of the new version is needed for static library.
40  */
41 #ifdef SHARED
42 # define COMPAT_VERSION(internal_name, api_name, version) \
43 	asm(".symver " #internal_name "," #api_name "@" #version);
44 # define DEFAULT_VERSION(internal_name, api_name, version) \
45 	asm(".symver " #internal_name "," #api_name "@@" #version);
46 #else
47 # define COMPAT_VERSION(internal_name, api_name, version)
48 # define DEFAULT_VERSION(internal_name, api_name, version) \
49 	extern typeof(internal_name) api_name \
50 	__attribute__((alias(#internal_name)));
51 #endif
52 
53 extern void libbpf_print(enum libbpf_print_level level,
54 			 const char *format, ...)
55 	__attribute__((format(printf, 2, 3)));
56 
57 #define __pr(level, fmt, ...)	\
58 do {				\
59 	libbpf_print(level, "libbpf: " fmt, ##__VA_ARGS__);	\
60 } while (0)
61 
62 #define pr_warn(fmt, ...)	__pr(LIBBPF_WARN, fmt, ##__VA_ARGS__)
63 #define pr_info(fmt, ...)	__pr(LIBBPF_INFO, fmt, ##__VA_ARGS__)
64 #define pr_debug(fmt, ...)	__pr(LIBBPF_DEBUG, fmt, ##__VA_ARGS__)
65 
66 static inline bool libbpf_validate_opts(const char *opts,
67 					size_t opts_sz, size_t user_sz,
68 					const char *type_name)
69 {
70 	if (user_sz < sizeof(size_t)) {
71 		pr_warn("%s size (%zu) is too small\n", type_name, user_sz);
72 		return false;
73 	}
74 	if (user_sz > opts_sz) {
75 		size_t i;
76 
77 		for (i = opts_sz; i < user_sz; i++) {
78 			if (opts[i]) {
79 				pr_warn("%s has non-zero extra bytes",
80 					type_name);
81 				return false;
82 			}
83 		}
84 	}
85 	return true;
86 }
87 
88 #define OPTS_VALID(opts, type)						      \
89 	(!(opts) || libbpf_validate_opts((const char *)opts,		      \
90 					 offsetofend(struct type,	      \
91 						     type##__last_field),     \
92 					 (opts)->sz, #type))
93 #define OPTS_HAS(opts, field) \
94 	((opts) && opts->sz >= offsetofend(typeof(*(opts)), field))
95 #define OPTS_GET(opts, field, fallback_value) \
96 	(OPTS_HAS(opts, field) ? (opts)->field : fallback_value)
97 
98 int parse_cpu_mask_str(const char *s, bool **mask, int *mask_sz);
99 int parse_cpu_mask_file(const char *fcpu, bool **mask, int *mask_sz);
100 int libbpf__load_raw_btf(const char *raw_types, size_t types_len,
101 			 const char *str_sec, size_t str_len);
102 
103 struct btf_ext_info {
104 	/*
105 	 * info points to the individual info section (e.g. func_info and
106 	 * line_info) from the .BTF.ext. It does not include the __u32 rec_size.
107 	 */
108 	void *info;
109 	__u32 rec_size;
110 	__u32 len;
111 };
112 
113 #define for_each_btf_ext_sec(seg, sec)					\
114 	for (sec = (seg)->info;						\
115 	     (void *)sec < (seg)->info + (seg)->len;			\
116 	     sec = (void *)sec + sizeof(struct btf_ext_info_sec) +	\
117 		   (seg)->rec_size * sec->num_info)
118 
119 #define for_each_btf_ext_rec(seg, sec, i, rec)				\
120 	for (i = 0, rec = (void *)&(sec)->data;				\
121 	     i < (sec)->num_info;					\
122 	     i++, rec = (void *)rec + (seg)->rec_size)
123 
124 struct btf_ext {
125 	union {
126 		struct btf_ext_header *hdr;
127 		void *data;
128 	};
129 	struct btf_ext_info func_info;
130 	struct btf_ext_info line_info;
131 	struct btf_ext_info field_reloc_info;
132 	__u32 data_size;
133 };
134 
135 struct btf_ext_info_sec {
136 	__u32	sec_name_off;
137 	__u32	num_info;
138 	/* Followed by num_info * record_size number of bytes */
139 	__u8	data[0];
140 };
141 
142 /* The minimum bpf_func_info checked by the loader */
143 struct bpf_func_info_min {
144 	__u32   insn_off;
145 	__u32   type_id;
146 };
147 
148 /* The minimum bpf_line_info checked by the loader */
149 struct bpf_line_info_min {
150 	__u32	insn_off;
151 	__u32	file_name_off;
152 	__u32	line_off;
153 	__u32	line_col;
154 };
155 
156 /* bpf_field_info_kind encodes which aspect of captured field has to be
157  * adjusted by relocations. Currently supported values are:
158  *   - BPF_FIELD_BYTE_OFFSET: field offset (in bytes);
159  *   - BPF_FIELD_EXISTS: field existence (1, if field exists; 0, otherwise);
160  */
161 enum bpf_field_info_kind {
162 	BPF_FIELD_BYTE_OFFSET = 0,	/* field byte offset */
163 	BPF_FIELD_BYTE_SIZE = 1,
164 	BPF_FIELD_EXISTS = 2,		/* field existence in target kernel */
165 	BPF_FIELD_SIGNED = 3,
166 	BPF_FIELD_LSHIFT_U64 = 4,
167 	BPF_FIELD_RSHIFT_U64 = 5,
168 };
169 
170 /* The minimum bpf_field_reloc checked by the loader
171  *
172  * Field relocation captures the following data:
173  * - insn_off - instruction offset (in bytes) within a BPF program that needs
174  *   its insn->imm field to be relocated with actual field info;
175  * - type_id - BTF type ID of the "root" (containing) entity of a relocatable
176  *   field;
177  * - access_str_off - offset into corresponding .BTF string section. String
178  *   itself encodes an accessed field using a sequence of field and array
179  *   indicies, separated by colon (:). It's conceptually very close to LLVM's
180  *   getelementptr ([0]) instruction's arguments for identifying offset to
181  *   a field.
182  *
183  * Example to provide a better feel.
184  *
185  *   struct sample {
186  *       int a;
187  *       struct {
188  *           int b[10];
189  *       };
190  *   };
191  *
192  *   struct sample *s = ...;
193  *   int x = &s->a;     // encoded as "0:0" (a is field #0)
194  *   int y = &s->b[5];  // encoded as "0:1:0:5" (anon struct is field #1,
195  *                      // b is field #0 inside anon struct, accessing elem #5)
196  *   int z = &s[10]->b; // encoded as "10:1" (ptr is used as an array)
197  *
198  * type_id for all relocs in this example  will capture BTF type id of
199  * `struct sample`.
200  *
201  * Such relocation is emitted when using __builtin_preserve_access_index()
202  * Clang built-in, passing expression that captures field address, e.g.:
203  *
204  * bpf_probe_read(&dst, sizeof(dst),
205  *		  __builtin_preserve_access_index(&src->a.b.c));
206  *
207  * In this case Clang will emit field relocation recording necessary data to
208  * be able to find offset of embedded `a.b.c` field within `src` struct.
209  *
210  *   [0] https://llvm.org/docs/LangRef.html#getelementptr-instruction
211  */
212 struct bpf_field_reloc {
213 	__u32   insn_off;
214 	__u32   type_id;
215 	__u32   access_str_off;
216 	enum bpf_field_info_kind kind;
217 };
218 
219 #endif /* __LIBBPF_LIBBPF_INTERNAL_H */
220