xref: /openbmc/linux/tools/lib/bpf/btf_dump.c (revision c39028b3)
1351131b5SAndrii Nakryiko // SPDX-License-Identifier: (LGPL-2.1 OR BSD-2-Clause)
2351131b5SAndrii Nakryiko 
3351131b5SAndrii Nakryiko /*
4351131b5SAndrii Nakryiko  * BTF-to-C type converter.
5351131b5SAndrii Nakryiko  *
6351131b5SAndrii Nakryiko  * Copyright (c) 2019 Facebook
7351131b5SAndrii Nakryiko  */
8351131b5SAndrii Nakryiko 
9351131b5SAndrii Nakryiko #include <stdbool.h>
10351131b5SAndrii Nakryiko #include <stddef.h>
11351131b5SAndrii Nakryiko #include <stdlib.h>
12351131b5SAndrii Nakryiko #include <string.h>
13920d16afSAlan Maguire #include <ctype.h>
14920d16afSAlan Maguire #include <endian.h>
15351131b5SAndrii Nakryiko #include <errno.h>
1621a9a1bcSAndrii Nakryiko #include <limits.h>
17351131b5SAndrii Nakryiko #include <linux/err.h>
18351131b5SAndrii Nakryiko #include <linux/btf.h>
19702eddc7SJean-Philippe Brucker #include <linux/kernel.h>
20351131b5SAndrii Nakryiko #include "btf.h"
21351131b5SAndrii Nakryiko #include "hashmap.h"
22351131b5SAndrii Nakryiko #include "libbpf.h"
23351131b5SAndrii Nakryiko #include "libbpf_internal.h"
24351131b5SAndrii Nakryiko 
25351131b5SAndrii Nakryiko static const char PREFIXES[] = "\t\t\t\t\t\t\t\t\t\t\t\t\t";
26351131b5SAndrii Nakryiko static const size_t PREFIX_CNT = sizeof(PREFIXES) - 1;
27351131b5SAndrii Nakryiko 
pfx(int lvl)28351131b5SAndrii Nakryiko static const char *pfx(int lvl)
29351131b5SAndrii Nakryiko {
30351131b5SAndrii Nakryiko 	return lvl >= PREFIX_CNT ? PREFIXES : &PREFIXES[PREFIX_CNT - lvl];
31351131b5SAndrii Nakryiko }
32351131b5SAndrii Nakryiko 
33351131b5SAndrii Nakryiko enum btf_dump_type_order_state {
34351131b5SAndrii Nakryiko 	NOT_ORDERED,
35351131b5SAndrii Nakryiko 	ORDERING,
36351131b5SAndrii Nakryiko 	ORDERED,
37351131b5SAndrii Nakryiko };
38351131b5SAndrii Nakryiko 
39351131b5SAndrii Nakryiko enum btf_dump_type_emit_state {
40351131b5SAndrii Nakryiko 	NOT_EMITTED,
41351131b5SAndrii Nakryiko 	EMITTING,
42351131b5SAndrii Nakryiko 	EMITTED,
43351131b5SAndrii Nakryiko };
44351131b5SAndrii Nakryiko 
45351131b5SAndrii Nakryiko /* per-type auxiliary state */
46351131b5SAndrii Nakryiko struct btf_dump_type_aux_state {
47351131b5SAndrii Nakryiko 	/* topological sorting state */
48351131b5SAndrii Nakryiko 	enum btf_dump_type_order_state order_state: 2;
49351131b5SAndrii Nakryiko 	/* emitting state used to determine the need for forward declaration */
50351131b5SAndrii Nakryiko 	enum btf_dump_type_emit_state emit_state: 2;
51351131b5SAndrii Nakryiko 	/* whether forward declaration was already emitted */
52351131b5SAndrii Nakryiko 	__u8 fwd_emitted: 1;
53351131b5SAndrii Nakryiko 	/* whether unique non-duplicate name was already assigned */
54351131b5SAndrii Nakryiko 	__u8 name_resolved: 1;
5539529a99SAndrii Nakryiko 	/* whether type is referenced from any other type */
5639529a99SAndrii Nakryiko 	__u8 referenced: 1;
57351131b5SAndrii Nakryiko };
58351131b5SAndrii Nakryiko 
59920d16afSAlan Maguire /* indent string length; one indent string is added for each indent level */
60920d16afSAlan Maguire #define BTF_DATA_INDENT_STR_LEN			32
61920d16afSAlan Maguire 
62920d16afSAlan Maguire /*
63920d16afSAlan Maguire  * Common internal data for BTF type data dump operations.
64920d16afSAlan Maguire  */
65920d16afSAlan Maguire struct btf_dump_data {
66920d16afSAlan Maguire 	const void *data_end;		/* end of valid data to show */
67920d16afSAlan Maguire 	bool compact;
68920d16afSAlan Maguire 	bool skip_names;
69920d16afSAlan Maguire 	bool emit_zeroes;
70920d16afSAlan Maguire 	__u8 indent_lvl;	/* base indent level */
71920d16afSAlan Maguire 	char indent_str[BTF_DATA_INDENT_STR_LEN];
72920d16afSAlan Maguire 	/* below are used during iteration */
73920d16afSAlan Maguire 	int depth;
74920d16afSAlan Maguire 	bool is_array_member;
75920d16afSAlan Maguire 	bool is_array_terminated;
76920d16afSAlan Maguire 	bool is_array_char;
77920d16afSAlan Maguire };
78920d16afSAlan Maguire 
79351131b5SAndrii Nakryiko struct btf_dump {
80351131b5SAndrii Nakryiko 	const struct btf *btf;
81351131b5SAndrii Nakryiko 	btf_dump_printf_fn_t printf_fn;
826084f5dcSAndrii Nakryiko 	void *cb_ctx;
8344ad23dfSAndrii Nakryiko 	int ptr_sz;
847c819e70SAndrii Nakryiko 	bool strip_mods;
85920d16afSAlan Maguire 	bool skip_anon_defs;
869c6c5c48SAndrii Nakryiko 	int last_id;
87351131b5SAndrii Nakryiko 
88351131b5SAndrii Nakryiko 	/* per-type auxiliary state */
89351131b5SAndrii Nakryiko 	struct btf_dump_type_aux_state *type_states;
909c6c5c48SAndrii Nakryiko 	size_t type_states_cap;
91351131b5SAndrii Nakryiko 	/* per-type optional cached unique name, must be freed, if present */
92351131b5SAndrii Nakryiko 	const char **cached_names;
939c6c5c48SAndrii Nakryiko 	size_t cached_names_cap;
94351131b5SAndrii Nakryiko 
95351131b5SAndrii Nakryiko 	/* topo-sorted list of dependent type definitions */
96351131b5SAndrii Nakryiko 	__u32 *emit_queue;
97351131b5SAndrii Nakryiko 	int emit_queue_cap;
98351131b5SAndrii Nakryiko 	int emit_queue_cnt;
99351131b5SAndrii Nakryiko 
100351131b5SAndrii Nakryiko 	/*
101351131b5SAndrii Nakryiko 	 * stack of type declarations (e.g., chain of modifiers, arrays,
102351131b5SAndrii Nakryiko 	 * funcs, etc)
103351131b5SAndrii Nakryiko 	 */
104351131b5SAndrii Nakryiko 	__u32 *decl_stack;
105351131b5SAndrii Nakryiko 	int decl_stack_cap;
106351131b5SAndrii Nakryiko 	int decl_stack_cnt;
107351131b5SAndrii Nakryiko 
108351131b5SAndrii Nakryiko 	/* maps struct/union/enum name to a number of name occurrences */
109351131b5SAndrii Nakryiko 	struct hashmap *type_names;
110351131b5SAndrii Nakryiko 	/*
111351131b5SAndrii Nakryiko 	 * maps typedef identifiers and enum value names to a number of such
112351131b5SAndrii Nakryiko 	 * name occurrences
113351131b5SAndrii Nakryiko 	 */
114351131b5SAndrii Nakryiko 	struct hashmap *ident_names;
115920d16afSAlan Maguire 	/*
116920d16afSAlan Maguire 	 * data for typed display; allocated if needed.
117920d16afSAlan Maguire 	 */
118920d16afSAlan Maguire 	struct btf_dump_data *typed_dump;
119351131b5SAndrii Nakryiko };
120351131b5SAndrii Nakryiko 
str_hash_fn(long key,void * ctx)121c302378bSEduard Zingerman static size_t str_hash_fn(long key, void *ctx)
122351131b5SAndrii Nakryiko {
123c302378bSEduard Zingerman 	return str_hash((void *)key);
124351131b5SAndrii Nakryiko }
125351131b5SAndrii Nakryiko 
str_equal_fn(long a,long b,void * ctx)126c302378bSEduard Zingerman static bool str_equal_fn(long a, long b, void *ctx)
127351131b5SAndrii Nakryiko {
128c302378bSEduard Zingerman 	return strcmp((void *)a, (void *)b) == 0;
129351131b5SAndrii Nakryiko }
130351131b5SAndrii Nakryiko 
btf_name_of(const struct btf_dump * d,__u32 name_off)131351131b5SAndrii Nakryiko static const char *btf_name_of(const struct btf_dump *d, __u32 name_off)
132351131b5SAndrii Nakryiko {
133351131b5SAndrii Nakryiko 	return btf__name_by_offset(d->btf, name_off);
134351131b5SAndrii Nakryiko }
135351131b5SAndrii Nakryiko 
btf_dump_printf(const struct btf_dump * d,const char * fmt,...)136351131b5SAndrii Nakryiko static void btf_dump_printf(const struct btf_dump *d, const char *fmt, ...)
137351131b5SAndrii Nakryiko {
138351131b5SAndrii Nakryiko 	va_list args;
139351131b5SAndrii Nakryiko 
140351131b5SAndrii Nakryiko 	va_start(args, fmt);
1416084f5dcSAndrii Nakryiko 	d->printf_fn(d->cb_ctx, fmt, args);
142351131b5SAndrii Nakryiko 	va_end(args);
143351131b5SAndrii Nakryiko }
144351131b5SAndrii Nakryiko 
1459f81654eSAndrii Nakryiko static int btf_dump_mark_referenced(struct btf_dump *d);
1469c6c5c48SAndrii Nakryiko static int btf_dump_resize(struct btf_dump *d);
1479f81654eSAndrii Nakryiko 
btf_dump__new(const struct btf * btf,btf_dump_printf_fn_t printf_fn,void * ctx,const struct btf_dump_opts * opts)148aaf6886dSAndrii Nakryiko struct btf_dump *btf_dump__new(const struct btf *btf,
1496084f5dcSAndrii Nakryiko 			       btf_dump_printf_fn_t printf_fn,
1506084f5dcSAndrii Nakryiko 			       void *ctx,
1516084f5dcSAndrii Nakryiko 			       const struct btf_dump_opts *opts)
152351131b5SAndrii Nakryiko {
153351131b5SAndrii Nakryiko 	struct btf_dump *d;
154351131b5SAndrii Nakryiko 	int err;
155351131b5SAndrii Nakryiko 
156aaf6886dSAndrii Nakryiko 	if (!OPTS_VALID(opts, btf_dump_opts))
157aaf6886dSAndrii Nakryiko 		return libbpf_err_ptr(-EINVAL);
158aaf6886dSAndrii Nakryiko 
1596084f5dcSAndrii Nakryiko 	if (!printf_fn)
1606084f5dcSAndrii Nakryiko 		return libbpf_err_ptr(-EINVAL);
1616084f5dcSAndrii Nakryiko 
162351131b5SAndrii Nakryiko 	d = calloc(1, sizeof(struct btf_dump));
163351131b5SAndrii Nakryiko 	if (!d)
164e9fc3ce9SAndrii Nakryiko 		return libbpf_err_ptr(-ENOMEM);
165351131b5SAndrii Nakryiko 
166351131b5SAndrii Nakryiko 	d->btf = btf;
167351131b5SAndrii Nakryiko 	d->printf_fn = printf_fn;
1686084f5dcSAndrii Nakryiko 	d->cb_ctx = ctx;
16944ad23dfSAndrii Nakryiko 	d->ptr_sz = btf__pointer_size(btf) ? : sizeof(void *);
170351131b5SAndrii Nakryiko 
171351131b5SAndrii Nakryiko 	d->type_names = hashmap__new(str_hash_fn, str_equal_fn, NULL);
172351131b5SAndrii Nakryiko 	if (IS_ERR(d->type_names)) {
173351131b5SAndrii Nakryiko 		err = PTR_ERR(d->type_names);
174351131b5SAndrii Nakryiko 		d->type_names = NULL;
175bc0eb9a3SAndrii Nakryiko 		goto err;
176351131b5SAndrii Nakryiko 	}
177351131b5SAndrii Nakryiko 	d->ident_names = hashmap__new(str_hash_fn, str_equal_fn, NULL);
178351131b5SAndrii Nakryiko 	if (IS_ERR(d->ident_names)) {
179351131b5SAndrii Nakryiko 		err = PTR_ERR(d->ident_names);
180351131b5SAndrii Nakryiko 		d->ident_names = NULL;
1819f81654eSAndrii Nakryiko 		goto err;
1829f81654eSAndrii Nakryiko 	}
183351131b5SAndrii Nakryiko 
1849c6c5c48SAndrii Nakryiko 	err = btf_dump_resize(d);
1859f81654eSAndrii Nakryiko 	if (err)
1869f81654eSAndrii Nakryiko 		goto err;
1879f81654eSAndrii Nakryiko 
188351131b5SAndrii Nakryiko 	return d;
1899f81654eSAndrii Nakryiko err:
1909f81654eSAndrii Nakryiko 	btf_dump__free(d);
191e9fc3ce9SAndrii Nakryiko 	return libbpf_err_ptr(err);
192351131b5SAndrii Nakryiko }
193351131b5SAndrii Nakryiko 
btf_dump_resize(struct btf_dump * d)1949c6c5c48SAndrii Nakryiko static int btf_dump_resize(struct btf_dump *d)
1959c6c5c48SAndrii Nakryiko {
1966a886de0SHengqi Chen 	int err, last_id = btf__type_cnt(d->btf) - 1;
1979c6c5c48SAndrii Nakryiko 
1989c6c5c48SAndrii Nakryiko 	if (last_id <= d->last_id)
1999c6c5c48SAndrii Nakryiko 		return 0;
2009c6c5c48SAndrii Nakryiko 
2013b029e06SAndrii Nakryiko 	if (libbpf_ensure_mem((void **)&d->type_states, &d->type_states_cap,
2029c6c5c48SAndrii Nakryiko 			      sizeof(*d->type_states), last_id + 1))
2039c6c5c48SAndrii Nakryiko 		return -ENOMEM;
2043b029e06SAndrii Nakryiko 	if (libbpf_ensure_mem((void **)&d->cached_names, &d->cached_names_cap,
2059c6c5c48SAndrii Nakryiko 			      sizeof(*d->cached_names), last_id + 1))
2069c6c5c48SAndrii Nakryiko 		return -ENOMEM;
2079c6c5c48SAndrii Nakryiko 
2089c6c5c48SAndrii Nakryiko 	if (d->last_id == 0) {
2099c6c5c48SAndrii Nakryiko 		/* VOID is special */
2109c6c5c48SAndrii Nakryiko 		d->type_states[0].order_state = ORDERED;
2119c6c5c48SAndrii Nakryiko 		d->type_states[0].emit_state = EMITTED;
2129c6c5c48SAndrii Nakryiko 	}
2139c6c5c48SAndrii Nakryiko 
2149c6c5c48SAndrii Nakryiko 	/* eagerly determine referenced types for anon enums */
2159c6c5c48SAndrii Nakryiko 	err = btf_dump_mark_referenced(d);
2169c6c5c48SAndrii Nakryiko 	if (err)
2179c6c5c48SAndrii Nakryiko 		return err;
2189c6c5c48SAndrii Nakryiko 
2199c6c5c48SAndrii Nakryiko 	d->last_id = last_id;
2209c6c5c48SAndrii Nakryiko 	return 0;
2219c6c5c48SAndrii Nakryiko }
2229c6c5c48SAndrii Nakryiko 
btf_dump_free_names(struct hashmap * map)22393c660caSXu Kuohai static void btf_dump_free_names(struct hashmap *map)
22493c660caSXu Kuohai {
22593c660caSXu Kuohai 	size_t bkt;
22693c660caSXu Kuohai 	struct hashmap_entry *cur;
22793c660caSXu Kuohai 
22893c660caSXu Kuohai 	hashmap__for_each_entry(map, cur, bkt)
229c302378bSEduard Zingerman 		free((void *)cur->pkey);
23093c660caSXu Kuohai 
23193c660caSXu Kuohai 	hashmap__free(map);
23293c660caSXu Kuohai }
23393c660caSXu Kuohai 
btf_dump__free(struct btf_dump * d)234351131b5SAndrii Nakryiko void btf_dump__free(struct btf_dump *d)
235351131b5SAndrii Nakryiko {
2369c6c5c48SAndrii Nakryiko 	int i;
237351131b5SAndrii Nakryiko 
23850450fc7SAndrii Nakryiko 	if (IS_ERR_OR_NULL(d))
239351131b5SAndrii Nakryiko 		return;
240351131b5SAndrii Nakryiko 
241351131b5SAndrii Nakryiko 	free(d->type_states);
242351131b5SAndrii Nakryiko 	if (d->cached_names) {
243351131b5SAndrii Nakryiko 		/* any set cached name is owned by us and should be freed */
2449c6c5c48SAndrii Nakryiko 		for (i = 0; i <= d->last_id; i++) {
245351131b5SAndrii Nakryiko 			if (d->cached_names[i])
246351131b5SAndrii Nakryiko 				free((void *)d->cached_names[i]);
247351131b5SAndrii Nakryiko 		}
248351131b5SAndrii Nakryiko 	}
249351131b5SAndrii Nakryiko 	free(d->cached_names);
250351131b5SAndrii Nakryiko 	free(d->emit_queue);
251351131b5SAndrii Nakryiko 	free(d->decl_stack);
25293c660caSXu Kuohai 	btf_dump_free_names(d->type_names);
25393c660caSXu Kuohai 	btf_dump_free_names(d->ident_names);
254351131b5SAndrii Nakryiko 
255351131b5SAndrii Nakryiko 	free(d);
256351131b5SAndrii Nakryiko }
257351131b5SAndrii Nakryiko 
258351131b5SAndrii Nakryiko static int btf_dump_order_type(struct btf_dump *d, __u32 id, bool through_ptr);
259351131b5SAndrii Nakryiko static void btf_dump_emit_type(struct btf_dump *d, __u32 id, __u32 cont_id);
260351131b5SAndrii Nakryiko 
261351131b5SAndrii Nakryiko /*
262351131b5SAndrii Nakryiko  * Dump BTF type in a compilable C syntax, including all the necessary
263351131b5SAndrii Nakryiko  * dependent types, necessary for compilation. If some of the dependent types
264351131b5SAndrii Nakryiko  * were already emitted as part of previous btf_dump__dump_type() invocation
265351131b5SAndrii Nakryiko  * for another type, they won't be emitted again. This API allows callers to
266351131b5SAndrii Nakryiko  * filter out BTF types according to user-defined criterias and emitted only
267351131b5SAndrii Nakryiko  * minimal subset of types, necessary to compile everything. Full struct/union
268351131b5SAndrii Nakryiko  * definitions will still be emitted, even if the only usage is through
269351131b5SAndrii Nakryiko  * pointer and could be satisfied with just a forward declaration.
270351131b5SAndrii Nakryiko  *
271351131b5SAndrii Nakryiko  * Dumping is done in two high-level passes:
272351131b5SAndrii Nakryiko  *   1. Topologically sort type definitions to satisfy C rules of compilation.
273351131b5SAndrii Nakryiko  *   2. Emit type definitions in C syntax.
274351131b5SAndrii Nakryiko  *
275351131b5SAndrii Nakryiko  * Returns 0 on success; <0, otherwise.
276351131b5SAndrii Nakryiko  */
btf_dump__dump_type(struct btf_dump * d,__u32 id)277351131b5SAndrii Nakryiko int btf_dump__dump_type(struct btf_dump *d, __u32 id)
278351131b5SAndrii Nakryiko {
279351131b5SAndrii Nakryiko 	int err, i;
280351131b5SAndrii Nakryiko 
2816a886de0SHengqi Chen 	if (id >= btf__type_cnt(d->btf))
282e9fc3ce9SAndrii Nakryiko 		return libbpf_err(-EINVAL);
283351131b5SAndrii Nakryiko 
2849c6c5c48SAndrii Nakryiko 	err = btf_dump_resize(d);
2859c6c5c48SAndrii Nakryiko 	if (err)
286e9fc3ce9SAndrii Nakryiko 		return libbpf_err(err);
2879c6c5c48SAndrii Nakryiko 
288351131b5SAndrii Nakryiko 	d->emit_queue_cnt = 0;
289351131b5SAndrii Nakryiko 	err = btf_dump_order_type(d, id, false);
290351131b5SAndrii Nakryiko 	if (err < 0)
291e9fc3ce9SAndrii Nakryiko 		return libbpf_err(err);
292351131b5SAndrii Nakryiko 
293351131b5SAndrii Nakryiko 	for (i = 0; i < d->emit_queue_cnt; i++)
294351131b5SAndrii Nakryiko 		btf_dump_emit_type(d, d->emit_queue[i], 0 /*top-level*/);
295351131b5SAndrii Nakryiko 
296351131b5SAndrii Nakryiko 	return 0;
297351131b5SAndrii Nakryiko }
298351131b5SAndrii Nakryiko 
29939529a99SAndrii Nakryiko /*
30039529a99SAndrii Nakryiko  * Mark all types that are referenced from any other type. This is used to
30139529a99SAndrii Nakryiko  * determine top-level anonymous enums that need to be emitted as an
30239529a99SAndrii Nakryiko  * independent type declarations.
30339529a99SAndrii Nakryiko  * Anonymous enums come in two flavors: either embedded in a struct's field
30439529a99SAndrii Nakryiko  * definition, in which case they have to be declared inline as part of field
30539529a99SAndrii Nakryiko  * type declaration; or as a top-level anonymous enum, typically used for
30639529a99SAndrii Nakryiko  * declaring global constants. It's impossible to distinguish between two
30739529a99SAndrii Nakryiko  * without knowning whether given enum type was referenced from other type:
30839529a99SAndrii Nakryiko  * top-level anonymous enum won't be referenced by anything, while embedded
30939529a99SAndrii Nakryiko  * one will.
31039529a99SAndrii Nakryiko  */
btf_dump_mark_referenced(struct btf_dump * d)31139529a99SAndrii Nakryiko static int btf_dump_mark_referenced(struct btf_dump *d)
31239529a99SAndrii Nakryiko {
3136a886de0SHengqi Chen 	int i, j, n = btf__type_cnt(d->btf);
31439529a99SAndrii Nakryiko 	const struct btf_type *t;
31539529a99SAndrii Nakryiko 	__u16 vlen;
31639529a99SAndrii Nakryiko 
3176a886de0SHengqi Chen 	for (i = d->last_id + 1; i < n; i++) {
31839529a99SAndrii Nakryiko 		t = btf__type_by_id(d->btf, i);
31939529a99SAndrii Nakryiko 		vlen = btf_vlen(t);
32039529a99SAndrii Nakryiko 
32139529a99SAndrii Nakryiko 		switch (btf_kind(t)) {
32239529a99SAndrii Nakryiko 		case BTF_KIND_INT:
32339529a99SAndrii Nakryiko 		case BTF_KIND_ENUM:
324d90ec262SYonghong Song 		case BTF_KIND_ENUM64:
32539529a99SAndrii Nakryiko 		case BTF_KIND_FWD:
32622541a9eSIlya Leoshkevich 		case BTF_KIND_FLOAT:
32739529a99SAndrii Nakryiko 			break;
32839529a99SAndrii Nakryiko 
32939529a99SAndrii Nakryiko 		case BTF_KIND_VOLATILE:
33039529a99SAndrii Nakryiko 		case BTF_KIND_CONST:
33139529a99SAndrii Nakryiko 		case BTF_KIND_RESTRICT:
33239529a99SAndrii Nakryiko 		case BTF_KIND_PTR:
33339529a99SAndrii Nakryiko 		case BTF_KIND_TYPEDEF:
33439529a99SAndrii Nakryiko 		case BTF_KIND_FUNC:
33539529a99SAndrii Nakryiko 		case BTF_KIND_VAR:
336223f903eSYonghong Song 		case BTF_KIND_DECL_TAG:
3372dc1e488SYonghong Song 		case BTF_KIND_TYPE_TAG:
33839529a99SAndrii Nakryiko 			d->type_states[t->type].referenced = 1;
33939529a99SAndrii Nakryiko 			break;
34039529a99SAndrii Nakryiko 
34139529a99SAndrii Nakryiko 		case BTF_KIND_ARRAY: {
34239529a99SAndrii Nakryiko 			const struct btf_array *a = btf_array(t);
34339529a99SAndrii Nakryiko 
34439529a99SAndrii Nakryiko 			d->type_states[a->index_type].referenced = 1;
34539529a99SAndrii Nakryiko 			d->type_states[a->type].referenced = 1;
34639529a99SAndrii Nakryiko 			break;
34739529a99SAndrii Nakryiko 		}
34839529a99SAndrii Nakryiko 		case BTF_KIND_STRUCT:
34939529a99SAndrii Nakryiko 		case BTF_KIND_UNION: {
35039529a99SAndrii Nakryiko 			const struct btf_member *m = btf_members(t);
35139529a99SAndrii Nakryiko 
35239529a99SAndrii Nakryiko 			for (j = 0; j < vlen; j++, m++)
35339529a99SAndrii Nakryiko 				d->type_states[m->type].referenced = 1;
35439529a99SAndrii Nakryiko 			break;
35539529a99SAndrii Nakryiko 		}
35639529a99SAndrii Nakryiko 		case BTF_KIND_FUNC_PROTO: {
35739529a99SAndrii Nakryiko 			const struct btf_param *p = btf_params(t);
35839529a99SAndrii Nakryiko 
35939529a99SAndrii Nakryiko 			for (j = 0; j < vlen; j++, p++)
36039529a99SAndrii Nakryiko 				d->type_states[p->type].referenced = 1;
36139529a99SAndrii Nakryiko 			break;
36239529a99SAndrii Nakryiko 		}
36339529a99SAndrii Nakryiko 		case BTF_KIND_DATASEC: {
36439529a99SAndrii Nakryiko 			const struct btf_var_secinfo *v = btf_var_secinfos(t);
36539529a99SAndrii Nakryiko 
36639529a99SAndrii Nakryiko 			for (j = 0; j < vlen; j++, v++)
36739529a99SAndrii Nakryiko 				d->type_states[v->type].referenced = 1;
36839529a99SAndrii Nakryiko 			break;
36939529a99SAndrii Nakryiko 		}
37039529a99SAndrii Nakryiko 		default:
37139529a99SAndrii Nakryiko 			return -EINVAL;
37239529a99SAndrii Nakryiko 		}
37339529a99SAndrii Nakryiko 	}
37439529a99SAndrii Nakryiko 	return 0;
37539529a99SAndrii Nakryiko }
3769c6c5c48SAndrii Nakryiko 
btf_dump_add_emit_queue_id(struct btf_dump * d,__u32 id)377351131b5SAndrii Nakryiko static int btf_dump_add_emit_queue_id(struct btf_dump *d, __u32 id)
378351131b5SAndrii Nakryiko {
379351131b5SAndrii Nakryiko 	__u32 *new_queue;
380351131b5SAndrii Nakryiko 	size_t new_cap;
381351131b5SAndrii Nakryiko 
382351131b5SAndrii Nakryiko 	if (d->emit_queue_cnt >= d->emit_queue_cap) {
383351131b5SAndrii Nakryiko 		new_cap = max(16, d->emit_queue_cap * 3 / 2);
384029258d7SAndrii Nakryiko 		new_queue = libbpf_reallocarray(d->emit_queue, new_cap, sizeof(new_queue[0]));
385351131b5SAndrii Nakryiko 		if (!new_queue)
386351131b5SAndrii Nakryiko 			return -ENOMEM;
387351131b5SAndrii Nakryiko 		d->emit_queue = new_queue;
388351131b5SAndrii Nakryiko 		d->emit_queue_cap = new_cap;
389351131b5SAndrii Nakryiko 	}
390351131b5SAndrii Nakryiko 
391351131b5SAndrii Nakryiko 	d->emit_queue[d->emit_queue_cnt++] = id;
392351131b5SAndrii Nakryiko 	return 0;
393351131b5SAndrii Nakryiko }
394351131b5SAndrii Nakryiko 
395351131b5SAndrii Nakryiko /*
396351131b5SAndrii Nakryiko  * Determine order of emitting dependent types and specified type to satisfy
397351131b5SAndrii Nakryiko  * C compilation rules.  This is done through topological sorting with an
398351131b5SAndrii Nakryiko  * additional complication which comes from C rules. The main idea for C is
399351131b5SAndrii Nakryiko  * that if some type is "embedded" into a struct/union, it's size needs to be
400351131b5SAndrii Nakryiko  * known at the time of definition of containing type. E.g., for:
401351131b5SAndrii Nakryiko  *
402351131b5SAndrii Nakryiko  *	struct A {};
403351131b5SAndrii Nakryiko  *	struct B { struct A x; }
404351131b5SAndrii Nakryiko  *
405351131b5SAndrii Nakryiko  * struct A *HAS* to be defined before struct B, because it's "embedded",
406351131b5SAndrii Nakryiko  * i.e., it is part of struct B layout. But in the following case:
407351131b5SAndrii Nakryiko  *
408351131b5SAndrii Nakryiko  *	struct A;
409351131b5SAndrii Nakryiko  *	struct B { struct A *x; }
410351131b5SAndrii Nakryiko  *	struct A {};
411351131b5SAndrii Nakryiko  *
412351131b5SAndrii Nakryiko  * it's enough to just have a forward declaration of struct A at the time of
413351131b5SAndrii Nakryiko  * struct B definition, as struct B has a pointer to struct A, so the size of
414351131b5SAndrii Nakryiko  * field x is known without knowing struct A size: it's sizeof(void *).
415351131b5SAndrii Nakryiko  *
416351131b5SAndrii Nakryiko  * Unfortunately, there are some trickier cases we need to handle, e.g.:
417351131b5SAndrii Nakryiko  *
418351131b5SAndrii Nakryiko  *	struct A {}; // if this was forward-declaration: compilation error
419351131b5SAndrii Nakryiko  *	struct B {
420351131b5SAndrii Nakryiko  *		struct { // anonymous struct
421351131b5SAndrii Nakryiko  *			struct A y;
422351131b5SAndrii Nakryiko  *		} *x;
423351131b5SAndrii Nakryiko  *	};
424351131b5SAndrii Nakryiko  *
425351131b5SAndrii Nakryiko  * In this case, struct B's field x is a pointer, so it's size is known
426351131b5SAndrii Nakryiko  * regardless of the size of (anonymous) struct it points to. But because this
427351131b5SAndrii Nakryiko  * struct is anonymous and thus defined inline inside struct B, *and* it
428351131b5SAndrii Nakryiko  * embeds struct A, compiler requires full definition of struct A to be known
429351131b5SAndrii Nakryiko  * before struct B can be defined. This creates a transitive dependency
430351131b5SAndrii Nakryiko  * between struct A and struct B. If struct A was forward-declared before
431351131b5SAndrii Nakryiko  * struct B definition and fully defined after struct B definition, that would
432351131b5SAndrii Nakryiko  * trigger compilation error.
433351131b5SAndrii Nakryiko  *
434351131b5SAndrii Nakryiko  * All this means that while we are doing topological sorting on BTF type
435351131b5SAndrii Nakryiko  * graph, we need to determine relationships between different types (graph
436351131b5SAndrii Nakryiko  * nodes):
437351131b5SAndrii Nakryiko  *   - weak link (relationship) between X and Y, if Y *CAN* be
438351131b5SAndrii Nakryiko  *   forward-declared at the point of X definition;
439351131b5SAndrii Nakryiko  *   - strong link, if Y *HAS* to be fully-defined before X can be defined.
440351131b5SAndrii Nakryiko  *
441351131b5SAndrii Nakryiko  * The rule is as follows. Given a chain of BTF types from X to Y, if there is
442351131b5SAndrii Nakryiko  * BTF_KIND_PTR type in the chain and at least one non-anonymous type
443351131b5SAndrii Nakryiko  * Z (excluding X, including Y), then link is weak. Otherwise, it's strong.
444351131b5SAndrii Nakryiko  * Weak/strong relationship is determined recursively during DFS traversal and
445351131b5SAndrii Nakryiko  * is returned as a result from btf_dump_order_type().
446351131b5SAndrii Nakryiko  *
447351131b5SAndrii Nakryiko  * btf_dump_order_type() is trying to avoid unnecessary forward declarations,
448351131b5SAndrii Nakryiko  * but it is not guaranteeing that no extraneous forward declarations will be
449351131b5SAndrii Nakryiko  * emitted.
450351131b5SAndrii Nakryiko  *
451351131b5SAndrii Nakryiko  * To avoid extra work, algorithm marks some of BTF types as ORDERED, when
452351131b5SAndrii Nakryiko  * it's done with them, but not for all (e.g., VOLATILE, CONST, RESTRICT,
453351131b5SAndrii Nakryiko  * ARRAY, FUNC_PROTO), as weak/strong semantics for those depends on the
454351131b5SAndrii Nakryiko  * entire graph path, so depending where from one came to that BTF type, it
455351131b5SAndrii Nakryiko  * might cause weak or strong ordering. For types like STRUCT/UNION/INT/ENUM,
456351131b5SAndrii Nakryiko  * once they are processed, there is no need to do it again, so they are
457351131b5SAndrii Nakryiko  * marked as ORDERED. We can mark PTR as ORDERED as well, as it semi-forces
458351131b5SAndrii Nakryiko  * weak link, unless subsequent referenced STRUCT/UNION/ENUM is anonymous. But
459351131b5SAndrii Nakryiko  * in any case, once those are processed, no need to do it again, as the
460351131b5SAndrii Nakryiko  * result won't change.
461351131b5SAndrii Nakryiko  *
462351131b5SAndrii Nakryiko  * Returns:
463351131b5SAndrii Nakryiko  *   - 1, if type is part of strong link (so there is strong topological
464351131b5SAndrii Nakryiko  *   ordering requirements);
465351131b5SAndrii Nakryiko  *   - 0, if type is part of weak link (so can be satisfied through forward
466351131b5SAndrii Nakryiko  *   declaration);
467351131b5SAndrii Nakryiko  *   - <0, on error (e.g., unsatisfiable type loop detected).
468351131b5SAndrii Nakryiko  */
btf_dump_order_type(struct btf_dump * d,__u32 id,bool through_ptr)469351131b5SAndrii Nakryiko static int btf_dump_order_type(struct btf_dump *d, __u32 id, bool through_ptr)
470351131b5SAndrii Nakryiko {
471351131b5SAndrii Nakryiko 	/*
472351131b5SAndrii Nakryiko 	 * Order state is used to detect strong link cycles, but only for BTF
473351131b5SAndrii Nakryiko 	 * kinds that are or could be an independent definition (i.e.,
474351131b5SAndrii Nakryiko 	 * stand-alone fwd decl, enum, typedef, struct, union). Ptrs, arrays,
475351131b5SAndrii Nakryiko 	 * func_protos, modifiers are just means to get to these definitions.
476351131b5SAndrii Nakryiko 	 * Int/void don't need definitions, they are assumed to be always
477351131b5SAndrii Nakryiko 	 * properly defined.  We also ignore datasec, var, and funcs for now.
478351131b5SAndrii Nakryiko 	 * So for all non-defining kinds, we never even set ordering state,
479351131b5SAndrii Nakryiko 	 * for defining kinds we set ORDERING and subsequently ORDERED if it
480351131b5SAndrii Nakryiko 	 * forms a strong link.
481351131b5SAndrii Nakryiko 	 */
482351131b5SAndrii Nakryiko 	struct btf_dump_type_aux_state *tstate = &d->type_states[id];
483351131b5SAndrii Nakryiko 	const struct btf_type *t;
484b03bc685SAndrii Nakryiko 	__u16 vlen;
485351131b5SAndrii Nakryiko 	int err, i;
486351131b5SAndrii Nakryiko 
487351131b5SAndrii Nakryiko 	/* return true, letting typedefs know that it's ok to be emitted */
488351131b5SAndrii Nakryiko 	if (tstate->order_state == ORDERED)
489351131b5SAndrii Nakryiko 		return 1;
490351131b5SAndrii Nakryiko 
491351131b5SAndrii Nakryiko 	t = btf__type_by_id(d->btf, id);
492351131b5SAndrii Nakryiko 
493351131b5SAndrii Nakryiko 	if (tstate->order_state == ORDERING) {
494351131b5SAndrii Nakryiko 		/* type loop, but resolvable through fwd declaration */
495b03bc685SAndrii Nakryiko 		if (btf_is_composite(t) && through_ptr && t->name_off != 0)
496351131b5SAndrii Nakryiko 			return 0;
497be18010eSKefeng Wang 		pr_warn("unsatisfiable type cycle, id:[%u]\n", id);
498351131b5SAndrii Nakryiko 		return -ELOOP;
499351131b5SAndrii Nakryiko 	}
500351131b5SAndrii Nakryiko 
501b03bc685SAndrii Nakryiko 	switch (btf_kind(t)) {
502351131b5SAndrii Nakryiko 	case BTF_KIND_INT:
50322541a9eSIlya Leoshkevich 	case BTF_KIND_FLOAT:
504351131b5SAndrii Nakryiko 		tstate->order_state = ORDERED;
505351131b5SAndrii Nakryiko 		return 0;
506351131b5SAndrii Nakryiko 
507351131b5SAndrii Nakryiko 	case BTF_KIND_PTR:
508351131b5SAndrii Nakryiko 		err = btf_dump_order_type(d, t->type, true);
509351131b5SAndrii Nakryiko 		tstate->order_state = ORDERED;
510351131b5SAndrii Nakryiko 		return err;
511351131b5SAndrii Nakryiko 
512b03bc685SAndrii Nakryiko 	case BTF_KIND_ARRAY:
513901ee1d7SJean-Philippe Brucker 		return btf_dump_order_type(d, btf_array(t)->type, false);
514351131b5SAndrii Nakryiko 
515351131b5SAndrii Nakryiko 	case BTF_KIND_STRUCT:
516351131b5SAndrii Nakryiko 	case BTF_KIND_UNION: {
517b03bc685SAndrii Nakryiko 		const struct btf_member *m = btf_members(t);
518351131b5SAndrii Nakryiko 		/*
519351131b5SAndrii Nakryiko 		 * struct/union is part of strong link, only if it's embedded
520351131b5SAndrii Nakryiko 		 * (so no ptr in a path) or it's anonymous (so has to be
521351131b5SAndrii Nakryiko 		 * defined inline, even if declared through ptr)
522351131b5SAndrii Nakryiko 		 */
523351131b5SAndrii Nakryiko 		if (through_ptr && t->name_off != 0)
524351131b5SAndrii Nakryiko 			return 0;
525351131b5SAndrii Nakryiko 
526351131b5SAndrii Nakryiko 		tstate->order_state = ORDERING;
527351131b5SAndrii Nakryiko 
528b03bc685SAndrii Nakryiko 		vlen = btf_vlen(t);
529351131b5SAndrii Nakryiko 		for (i = 0; i < vlen; i++, m++) {
530351131b5SAndrii Nakryiko 			err = btf_dump_order_type(d, m->type, false);
531351131b5SAndrii Nakryiko 			if (err < 0)
532351131b5SAndrii Nakryiko 				return err;
533351131b5SAndrii Nakryiko 		}
534351131b5SAndrii Nakryiko 
535351131b5SAndrii Nakryiko 		if (t->name_off != 0) {
536351131b5SAndrii Nakryiko 			err = btf_dump_add_emit_queue_id(d, id);
537351131b5SAndrii Nakryiko 			if (err < 0)
538351131b5SAndrii Nakryiko 				return err;
539351131b5SAndrii Nakryiko 		}
540351131b5SAndrii Nakryiko 
541351131b5SAndrii Nakryiko 		tstate->order_state = ORDERED;
542351131b5SAndrii Nakryiko 		return 1;
543351131b5SAndrii Nakryiko 	}
544351131b5SAndrii Nakryiko 	case BTF_KIND_ENUM:
545d90ec262SYonghong Song 	case BTF_KIND_ENUM64:
546351131b5SAndrii Nakryiko 	case BTF_KIND_FWD:
54739529a99SAndrii Nakryiko 		/*
54839529a99SAndrii Nakryiko 		 * non-anonymous or non-referenced enums are top-level
54939529a99SAndrii Nakryiko 		 * declarations and should be emitted. Same logic can be
55039529a99SAndrii Nakryiko 		 * applied to FWDs, it won't hurt anyways.
55139529a99SAndrii Nakryiko 		 */
55239529a99SAndrii Nakryiko 		if (t->name_off != 0 || !tstate->referenced) {
553351131b5SAndrii Nakryiko 			err = btf_dump_add_emit_queue_id(d, id);
554351131b5SAndrii Nakryiko 			if (err)
555351131b5SAndrii Nakryiko 				return err;
556351131b5SAndrii Nakryiko 		}
557351131b5SAndrii Nakryiko 		tstate->order_state = ORDERED;
558351131b5SAndrii Nakryiko 		return 1;
559351131b5SAndrii Nakryiko 
560351131b5SAndrii Nakryiko 	case BTF_KIND_TYPEDEF: {
561351131b5SAndrii Nakryiko 		int is_strong;
562351131b5SAndrii Nakryiko 
563351131b5SAndrii Nakryiko 		is_strong = btf_dump_order_type(d, t->type, through_ptr);
564351131b5SAndrii Nakryiko 		if (is_strong < 0)
565351131b5SAndrii Nakryiko 			return is_strong;
566351131b5SAndrii Nakryiko 
567351131b5SAndrii Nakryiko 		/* typedef is similar to struct/union w.r.t. fwd-decls */
568351131b5SAndrii Nakryiko 		if (through_ptr && !is_strong)
569351131b5SAndrii Nakryiko 			return 0;
570351131b5SAndrii Nakryiko 
571351131b5SAndrii Nakryiko 		/* typedef is always a named definition */
572351131b5SAndrii Nakryiko 		err = btf_dump_add_emit_queue_id(d, id);
573351131b5SAndrii Nakryiko 		if (err)
574351131b5SAndrii Nakryiko 			return err;
575351131b5SAndrii Nakryiko 
576351131b5SAndrii Nakryiko 		d->type_states[id].order_state = ORDERED;
577351131b5SAndrii Nakryiko 		return 1;
578351131b5SAndrii Nakryiko 	}
579351131b5SAndrii Nakryiko 	case BTF_KIND_VOLATILE:
580351131b5SAndrii Nakryiko 	case BTF_KIND_CONST:
581351131b5SAndrii Nakryiko 	case BTF_KIND_RESTRICT:
5822dc1e488SYonghong Song 	case BTF_KIND_TYPE_TAG:
583351131b5SAndrii Nakryiko 		return btf_dump_order_type(d, t->type, through_ptr);
584351131b5SAndrii Nakryiko 
585351131b5SAndrii Nakryiko 	case BTF_KIND_FUNC_PROTO: {
586b03bc685SAndrii Nakryiko 		const struct btf_param *p = btf_params(t);
587351131b5SAndrii Nakryiko 		bool is_strong;
588351131b5SAndrii Nakryiko 
589351131b5SAndrii Nakryiko 		err = btf_dump_order_type(d, t->type, through_ptr);
590351131b5SAndrii Nakryiko 		if (err < 0)
591351131b5SAndrii Nakryiko 			return err;
592351131b5SAndrii Nakryiko 		is_strong = err > 0;
593351131b5SAndrii Nakryiko 
594b03bc685SAndrii Nakryiko 		vlen = btf_vlen(t);
595351131b5SAndrii Nakryiko 		for (i = 0; i < vlen; i++, p++) {
596351131b5SAndrii Nakryiko 			err = btf_dump_order_type(d, p->type, through_ptr);
597351131b5SAndrii Nakryiko 			if (err < 0)
598351131b5SAndrii Nakryiko 				return err;
599351131b5SAndrii Nakryiko 			if (err > 0)
600351131b5SAndrii Nakryiko 				is_strong = true;
601351131b5SAndrii Nakryiko 		}
602351131b5SAndrii Nakryiko 		return is_strong;
603351131b5SAndrii Nakryiko 	}
604351131b5SAndrii Nakryiko 	case BTF_KIND_FUNC:
605351131b5SAndrii Nakryiko 	case BTF_KIND_VAR:
606351131b5SAndrii Nakryiko 	case BTF_KIND_DATASEC:
607223f903eSYonghong Song 	case BTF_KIND_DECL_TAG:
608351131b5SAndrii Nakryiko 		d->type_states[id].order_state = ORDERED;
609351131b5SAndrii Nakryiko 		return 0;
610351131b5SAndrii Nakryiko 
611351131b5SAndrii Nakryiko 	default:
612351131b5SAndrii Nakryiko 		return -EINVAL;
613351131b5SAndrii Nakryiko 	}
614351131b5SAndrii Nakryiko }
615351131b5SAndrii Nakryiko 
616702eddc7SJean-Philippe Brucker static void btf_dump_emit_missing_aliases(struct btf_dump *d, __u32 id,
617702eddc7SJean-Philippe Brucker 					  const struct btf_type *t);
618702eddc7SJean-Philippe Brucker 
619351131b5SAndrii Nakryiko static void btf_dump_emit_struct_fwd(struct btf_dump *d, __u32 id,
620351131b5SAndrii Nakryiko 				     const struct btf_type *t);
621351131b5SAndrii Nakryiko static void btf_dump_emit_struct_def(struct btf_dump *d, __u32 id,
622351131b5SAndrii Nakryiko 				     const struct btf_type *t, int lvl);
623351131b5SAndrii Nakryiko 
624351131b5SAndrii Nakryiko static void btf_dump_emit_enum_fwd(struct btf_dump *d, __u32 id,
625351131b5SAndrii Nakryiko 				   const struct btf_type *t);
626351131b5SAndrii Nakryiko static void btf_dump_emit_enum_def(struct btf_dump *d, __u32 id,
627351131b5SAndrii Nakryiko 				   const struct btf_type *t, int lvl);
628351131b5SAndrii Nakryiko 
629351131b5SAndrii Nakryiko static void btf_dump_emit_fwd_def(struct btf_dump *d, __u32 id,
630351131b5SAndrii Nakryiko 				  const struct btf_type *t);
631351131b5SAndrii Nakryiko 
632351131b5SAndrii Nakryiko static void btf_dump_emit_typedef_def(struct btf_dump *d, __u32 id,
633351131b5SAndrii Nakryiko 				      const struct btf_type *t, int lvl);
634351131b5SAndrii Nakryiko 
635351131b5SAndrii Nakryiko /* a local view into a shared stack */
636351131b5SAndrii Nakryiko struct id_stack {
637351131b5SAndrii Nakryiko 	const __u32 *ids;
638351131b5SAndrii Nakryiko 	int cnt;
639351131b5SAndrii Nakryiko };
640351131b5SAndrii Nakryiko 
641351131b5SAndrii Nakryiko static void btf_dump_emit_type_decl(struct btf_dump *d, __u32 id,
642351131b5SAndrii Nakryiko 				    const char *fname, int lvl);
643351131b5SAndrii Nakryiko static void btf_dump_emit_type_chain(struct btf_dump *d,
644351131b5SAndrii Nakryiko 				     struct id_stack *decl_stack,
645351131b5SAndrii Nakryiko 				     const char *fname, int lvl);
646351131b5SAndrii Nakryiko 
647351131b5SAndrii Nakryiko static const char *btf_dump_type_name(struct btf_dump *d, __u32 id);
648351131b5SAndrii Nakryiko static const char *btf_dump_ident_name(struct btf_dump *d, __u32 id);
649351131b5SAndrii Nakryiko static size_t btf_dump_name_dups(struct btf_dump *d, struct hashmap *name_map,
650351131b5SAndrii Nakryiko 				 const char *orig_name);
651351131b5SAndrii Nakryiko 
btf_dump_is_blacklisted(struct btf_dump * d,__u32 id)652351131b5SAndrii Nakryiko static bool btf_dump_is_blacklisted(struct btf_dump *d, __u32 id)
653351131b5SAndrii Nakryiko {
654351131b5SAndrii Nakryiko 	const struct btf_type *t = btf__type_by_id(d->btf, id);
655351131b5SAndrii Nakryiko 
656351131b5SAndrii Nakryiko 	/* __builtin_va_list is a compiler built-in, which causes compilation
657351131b5SAndrii Nakryiko 	 * errors, when compiling w/ different compiler, then used to compile
658351131b5SAndrii Nakryiko 	 * original code (e.g., GCC to compile kernel, Clang to use generated
659351131b5SAndrii Nakryiko 	 * C header from BTF). As it is built-in, it should be already defined
660351131b5SAndrii Nakryiko 	 * properly internally in compiler.
661351131b5SAndrii Nakryiko 	 */
662351131b5SAndrii Nakryiko 	if (t->name_off == 0)
663351131b5SAndrii Nakryiko 		return false;
664351131b5SAndrii Nakryiko 	return strcmp(btf_name_of(d, t->name_off), "__builtin_va_list") == 0;
665351131b5SAndrii Nakryiko }
666351131b5SAndrii Nakryiko 
667351131b5SAndrii Nakryiko /*
668351131b5SAndrii Nakryiko  * Emit C-syntax definitions of types from chains of BTF types.
669351131b5SAndrii Nakryiko  *
670351131b5SAndrii Nakryiko  * High-level handling of determining necessary forward declarations are handled
671351131b5SAndrii Nakryiko  * by btf_dump_emit_type() itself, but all nitty-gritty details of emitting type
672351131b5SAndrii Nakryiko  * declarations/definitions in C syntax  are handled by a combo of
673351131b5SAndrii Nakryiko  * btf_dump_emit_type_decl()/btf_dump_emit_type_chain() w/ delegation to
674351131b5SAndrii Nakryiko  * corresponding btf_dump_emit_*_{def,fwd}() functions.
675351131b5SAndrii Nakryiko  *
676351131b5SAndrii Nakryiko  * We also keep track of "containing struct/union type ID" to determine when
677351131b5SAndrii Nakryiko  * we reference it from inside and thus can avoid emitting unnecessary forward
678351131b5SAndrii Nakryiko  * declaration.
679351131b5SAndrii Nakryiko  *
680351131b5SAndrii Nakryiko  * This algorithm is designed in such a way, that even if some error occurs
681351131b5SAndrii Nakryiko  * (either technical, e.g., out of memory, or logical, i.e., malformed BTF
682351131b5SAndrii Nakryiko  * that doesn't comply to C rules completely), algorithm will try to proceed
683351131b5SAndrii Nakryiko  * and produce as much meaningful output as possible.
684351131b5SAndrii Nakryiko  */
btf_dump_emit_type(struct btf_dump * d,__u32 id,__u32 cont_id)685351131b5SAndrii Nakryiko static void btf_dump_emit_type(struct btf_dump *d, __u32 id, __u32 cont_id)
686351131b5SAndrii Nakryiko {
687351131b5SAndrii Nakryiko 	struct btf_dump_type_aux_state *tstate = &d->type_states[id];
688351131b5SAndrii Nakryiko 	bool top_level_def = cont_id == 0;
689351131b5SAndrii Nakryiko 	const struct btf_type *t;
690351131b5SAndrii Nakryiko 	__u16 kind;
691351131b5SAndrii Nakryiko 
692351131b5SAndrii Nakryiko 	if (tstate->emit_state == EMITTED)
693351131b5SAndrii Nakryiko 		return;
694351131b5SAndrii Nakryiko 
695351131b5SAndrii Nakryiko 	t = btf__type_by_id(d->btf, id);
696b03bc685SAndrii Nakryiko 	kind = btf_kind(t);
697351131b5SAndrii Nakryiko 
698351131b5SAndrii Nakryiko 	if (tstate->emit_state == EMITTING) {
699351131b5SAndrii Nakryiko 		if (tstate->fwd_emitted)
700351131b5SAndrii Nakryiko 			return;
701351131b5SAndrii Nakryiko 
702351131b5SAndrii Nakryiko 		switch (kind) {
703351131b5SAndrii Nakryiko 		case BTF_KIND_STRUCT:
704351131b5SAndrii Nakryiko 		case BTF_KIND_UNION:
705351131b5SAndrii Nakryiko 			/*
706351131b5SAndrii Nakryiko 			 * if we are referencing a struct/union that we are
707351131b5SAndrii Nakryiko 			 * part of - then no need for fwd declaration
708351131b5SAndrii Nakryiko 			 */
709351131b5SAndrii Nakryiko 			if (id == cont_id)
710351131b5SAndrii Nakryiko 				return;
711351131b5SAndrii Nakryiko 			if (t->name_off == 0) {
712be18010eSKefeng Wang 				pr_warn("anonymous struct/union loop, id:[%u]\n",
713351131b5SAndrii Nakryiko 					id);
714351131b5SAndrii Nakryiko 				return;
715351131b5SAndrii Nakryiko 			}
716351131b5SAndrii Nakryiko 			btf_dump_emit_struct_fwd(d, id, t);
717351131b5SAndrii Nakryiko 			btf_dump_printf(d, ";\n\n");
718351131b5SAndrii Nakryiko 			tstate->fwd_emitted = 1;
719351131b5SAndrii Nakryiko 			break;
720351131b5SAndrii Nakryiko 		case BTF_KIND_TYPEDEF:
721351131b5SAndrii Nakryiko 			/*
722351131b5SAndrii Nakryiko 			 * for typedef fwd_emitted means typedef definition
723351131b5SAndrii Nakryiko 			 * was emitted, but it can be used only for "weak"
724351131b5SAndrii Nakryiko 			 * references through pointer only, not for embedding
725351131b5SAndrii Nakryiko 			 */
726351131b5SAndrii Nakryiko 			if (!btf_dump_is_blacklisted(d, id)) {
727351131b5SAndrii Nakryiko 				btf_dump_emit_typedef_def(d, id, t, 0);
728351131b5SAndrii Nakryiko 				btf_dump_printf(d, ";\n\n");
729a6bbdf2eSZou Wei 			}
730351131b5SAndrii Nakryiko 			tstate->fwd_emitted = 1;
731351131b5SAndrii Nakryiko 			break;
732351131b5SAndrii Nakryiko 		default:
733351131b5SAndrii Nakryiko 			break;
734351131b5SAndrii Nakryiko 		}
735351131b5SAndrii Nakryiko 
736351131b5SAndrii Nakryiko 		return;
737351131b5SAndrii Nakryiko 	}
738351131b5SAndrii Nakryiko 
739351131b5SAndrii Nakryiko 	switch (kind) {
740351131b5SAndrii Nakryiko 	case BTF_KIND_INT:
741702eddc7SJean-Philippe Brucker 		/* Emit type alias definitions if necessary */
742702eddc7SJean-Philippe Brucker 		btf_dump_emit_missing_aliases(d, id, t);
743702eddc7SJean-Philippe Brucker 
744351131b5SAndrii Nakryiko 		tstate->emit_state = EMITTED;
745351131b5SAndrii Nakryiko 		break;
746351131b5SAndrii Nakryiko 	case BTF_KIND_ENUM:
747d90ec262SYonghong Song 	case BTF_KIND_ENUM64:
748351131b5SAndrii Nakryiko 		if (top_level_def) {
749351131b5SAndrii Nakryiko 			btf_dump_emit_enum_def(d, id, t, 0);
750351131b5SAndrii Nakryiko 			btf_dump_printf(d, ";\n\n");
751351131b5SAndrii Nakryiko 		}
752351131b5SAndrii Nakryiko 		tstate->emit_state = EMITTED;
753351131b5SAndrii Nakryiko 		break;
754351131b5SAndrii Nakryiko 	case BTF_KIND_PTR:
755351131b5SAndrii Nakryiko 	case BTF_KIND_VOLATILE:
756351131b5SAndrii Nakryiko 	case BTF_KIND_CONST:
757351131b5SAndrii Nakryiko 	case BTF_KIND_RESTRICT:
7582dc1e488SYonghong Song 	case BTF_KIND_TYPE_TAG:
759351131b5SAndrii Nakryiko 		btf_dump_emit_type(d, t->type, cont_id);
760351131b5SAndrii Nakryiko 		break;
761b03bc685SAndrii Nakryiko 	case BTF_KIND_ARRAY:
762b03bc685SAndrii Nakryiko 		btf_dump_emit_type(d, btf_array(t)->type, cont_id);
763351131b5SAndrii Nakryiko 		break;
764351131b5SAndrii Nakryiko 	case BTF_KIND_FWD:
765351131b5SAndrii Nakryiko 		btf_dump_emit_fwd_def(d, id, t);
766351131b5SAndrii Nakryiko 		btf_dump_printf(d, ";\n\n");
767351131b5SAndrii Nakryiko 		tstate->emit_state = EMITTED;
768351131b5SAndrii Nakryiko 		break;
769351131b5SAndrii Nakryiko 	case BTF_KIND_TYPEDEF:
770351131b5SAndrii Nakryiko 		tstate->emit_state = EMITTING;
771351131b5SAndrii Nakryiko 		btf_dump_emit_type(d, t->type, id);
772351131b5SAndrii Nakryiko 		/*
773351131b5SAndrii Nakryiko 		 * typedef can server as both definition and forward
774351131b5SAndrii Nakryiko 		 * declaration; at this stage someone depends on
775351131b5SAndrii Nakryiko 		 * typedef as a forward declaration (refers to it
776351131b5SAndrii Nakryiko 		 * through pointer), so unless we already did it,
777351131b5SAndrii Nakryiko 		 * emit typedef as a forward declaration
778351131b5SAndrii Nakryiko 		 */
779351131b5SAndrii Nakryiko 		if (!tstate->fwd_emitted && !btf_dump_is_blacklisted(d, id)) {
780351131b5SAndrii Nakryiko 			btf_dump_emit_typedef_def(d, id, t, 0);
781351131b5SAndrii Nakryiko 			btf_dump_printf(d, ";\n\n");
782351131b5SAndrii Nakryiko 		}
783351131b5SAndrii Nakryiko 		tstate->emit_state = EMITTED;
784351131b5SAndrii Nakryiko 		break;
785351131b5SAndrii Nakryiko 	case BTF_KIND_STRUCT:
786351131b5SAndrii Nakryiko 	case BTF_KIND_UNION:
787351131b5SAndrii Nakryiko 		tstate->emit_state = EMITTING;
788351131b5SAndrii Nakryiko 		/* if it's a top-level struct/union definition or struct/union
789351131b5SAndrii Nakryiko 		 * is anonymous, then in C we'll be emitting all fields and
790351131b5SAndrii Nakryiko 		 * their types (as opposed to just `struct X`), so we need to
791351131b5SAndrii Nakryiko 		 * make sure that all types, referenced from struct/union
792351131b5SAndrii Nakryiko 		 * members have necessary forward-declarations, where
793351131b5SAndrii Nakryiko 		 * applicable
794351131b5SAndrii Nakryiko 		 */
795351131b5SAndrii Nakryiko 		if (top_level_def || t->name_off == 0) {
796b03bc685SAndrii Nakryiko 			const struct btf_member *m = btf_members(t);
797b03bc685SAndrii Nakryiko 			__u16 vlen = btf_vlen(t);
798351131b5SAndrii Nakryiko 			int i, new_cont_id;
799351131b5SAndrii Nakryiko 
800351131b5SAndrii Nakryiko 			new_cont_id = t->name_off == 0 ? cont_id : id;
801351131b5SAndrii Nakryiko 			for (i = 0; i < vlen; i++, m++)
802351131b5SAndrii Nakryiko 				btf_dump_emit_type(d, m->type, new_cont_id);
803351131b5SAndrii Nakryiko 		} else if (!tstate->fwd_emitted && id != cont_id) {
804351131b5SAndrii Nakryiko 			btf_dump_emit_struct_fwd(d, id, t);
805351131b5SAndrii Nakryiko 			btf_dump_printf(d, ";\n\n");
806351131b5SAndrii Nakryiko 			tstate->fwd_emitted = 1;
807351131b5SAndrii Nakryiko 		}
808351131b5SAndrii Nakryiko 
809351131b5SAndrii Nakryiko 		if (top_level_def) {
810351131b5SAndrii Nakryiko 			btf_dump_emit_struct_def(d, id, t, 0);
811351131b5SAndrii Nakryiko 			btf_dump_printf(d, ";\n\n");
812351131b5SAndrii Nakryiko 			tstate->emit_state = EMITTED;
813351131b5SAndrii Nakryiko 		} else {
814351131b5SAndrii Nakryiko 			tstate->emit_state = NOT_EMITTED;
815351131b5SAndrii Nakryiko 		}
816351131b5SAndrii Nakryiko 		break;
817351131b5SAndrii Nakryiko 	case BTF_KIND_FUNC_PROTO: {
818b03bc685SAndrii Nakryiko 		const struct btf_param *p = btf_params(t);
819920d16afSAlan Maguire 		__u16 n = btf_vlen(t);
820351131b5SAndrii Nakryiko 		int i;
821351131b5SAndrii Nakryiko 
822351131b5SAndrii Nakryiko 		btf_dump_emit_type(d, t->type, cont_id);
823920d16afSAlan Maguire 		for (i = 0; i < n; i++, p++)
824351131b5SAndrii Nakryiko 			btf_dump_emit_type(d, p->type, cont_id);
825351131b5SAndrii Nakryiko 
826351131b5SAndrii Nakryiko 		break;
827351131b5SAndrii Nakryiko 	}
828351131b5SAndrii Nakryiko 	default:
829351131b5SAndrii Nakryiko 		break;
830351131b5SAndrii Nakryiko 	}
831351131b5SAndrii Nakryiko }
832351131b5SAndrii Nakryiko 
btf_is_struct_packed(const struct btf * btf,__u32 id,const struct btf_type * t)833351131b5SAndrii Nakryiko static bool btf_is_struct_packed(const struct btf *btf, __u32 id,
834351131b5SAndrii Nakryiko 				 const struct btf_type *t)
835351131b5SAndrii Nakryiko {
836351131b5SAndrii Nakryiko 	const struct btf_member *m;
8374fb877aaSAndrii Nakryiko 	int max_align = 1, align, i, bit_sz;
838351131b5SAndrii Nakryiko 	__u16 vlen;
839351131b5SAndrii Nakryiko 
840b03bc685SAndrii Nakryiko 	m = btf_members(t);
841b03bc685SAndrii Nakryiko 	vlen = btf_vlen(t);
842351131b5SAndrii Nakryiko 	/* all non-bitfield fields have to be naturally aligned */
843351131b5SAndrii Nakryiko 	for (i = 0; i < vlen; i++, m++) {
8444fb877aaSAndrii Nakryiko 		align = btf__align_of(btf, m->type);
845b03bc685SAndrii Nakryiko 		bit_sz = btf_member_bitfield_size(t, i);
8463d208f4cSAndrii Nakryiko 		if (align && bit_sz == 0 && m->offset % (8 * align) != 0)
847351131b5SAndrii Nakryiko 			return true;
8484fb877aaSAndrii Nakryiko 		max_align = max(align, max_align);
849351131b5SAndrii Nakryiko 	}
8504fb877aaSAndrii Nakryiko 	/* size of a non-packed struct has to be a multiple of its alignment */
8514fb877aaSAndrii Nakryiko 	if (t->size % max_align != 0)
8524fb877aaSAndrii Nakryiko 		return true;
853351131b5SAndrii Nakryiko 	/*
854351131b5SAndrii Nakryiko 	 * if original struct was marked as packed, but its layout is
855351131b5SAndrii Nakryiko 	 * naturally aligned, we'll detect that it's not packed
856351131b5SAndrii Nakryiko 	 */
857351131b5SAndrii Nakryiko 	return false;
858351131b5SAndrii Nakryiko }
859351131b5SAndrii Nakryiko 
btf_dump_emit_bit_padding(const struct btf_dump * d,int cur_off,int next_off,int next_align,bool in_bitfield,int lvl)860351131b5SAndrii Nakryiko static void btf_dump_emit_bit_padding(const struct btf_dump *d,
861ea2ce1baSAndrii Nakryiko 				      int cur_off, int next_off, int next_align,
862ea2ce1baSAndrii Nakryiko 				      bool in_bitfield, int lvl)
863351131b5SAndrii Nakryiko {
864ea2ce1baSAndrii Nakryiko 	const struct {
865ea2ce1baSAndrii Nakryiko 		const char *name;
866ea2ce1baSAndrii Nakryiko 		int bits;
867ea2ce1baSAndrii Nakryiko 	} pads[] = {
868ea2ce1baSAndrii Nakryiko 		{"long", d->ptr_sz * 8}, {"int", 32}, {"short", 16}, {"char", 8}
869ea2ce1baSAndrii Nakryiko 	};
870ea2ce1baSAndrii Nakryiko 	int new_off, pad_bits, bits, i;
871351131b5SAndrii Nakryiko 	const char *pad_type;
872351131b5SAndrii Nakryiko 
873ea2ce1baSAndrii Nakryiko 	if (cur_off >= next_off)
874ea2ce1baSAndrii Nakryiko 		return; /* no gap */
875ea2ce1baSAndrii Nakryiko 
876ea2ce1baSAndrii Nakryiko 	/* For filling out padding we want to take advantage of
877ea2ce1baSAndrii Nakryiko 	 * natural alignment rules to minimize unnecessary explicit
878ea2ce1baSAndrii Nakryiko 	 * padding. First, we find the largest type (among long, int,
879ea2ce1baSAndrii Nakryiko 	 * short, or char) that can be used to force naturally aligned
880ea2ce1baSAndrii Nakryiko 	 * boundary. Once determined, we'll use such type to fill in
881ea2ce1baSAndrii Nakryiko 	 * the remaining padding gap. In some cases we can rely on
882ea2ce1baSAndrii Nakryiko 	 * compiler filling some gaps, but sometimes we need to force
883ea2ce1baSAndrii Nakryiko 	 * alignment to close natural alignment with markers like
884ea2ce1baSAndrii Nakryiko 	 * `long: 0` (this is always the case for bitfields).  Note
885ea2ce1baSAndrii Nakryiko 	 * that even if struct itself has, let's say 4-byte alignment
886ea2ce1baSAndrii Nakryiko 	 * (i.e., it only uses up to int-aligned types), using `long:
887ea2ce1baSAndrii Nakryiko 	 * X;` explicit padding doesn't actually change struct's
888ea2ce1baSAndrii Nakryiko 	 * overall alignment requirements, but compiler does take into
889ea2ce1baSAndrii Nakryiko 	 * account that type's (long, in this example) natural
890ea2ce1baSAndrii Nakryiko 	 * alignment requirements when adding implicit padding. We use
891ea2ce1baSAndrii Nakryiko 	 * this fact heavily and don't worry about ruining correct
892ea2ce1baSAndrii Nakryiko 	 * struct alignment requirement.
893ea2ce1baSAndrii Nakryiko 	 */
894ea2ce1baSAndrii Nakryiko 	for (i = 0; i < ARRAY_SIZE(pads); i++) {
895ea2ce1baSAndrii Nakryiko 		pad_bits = pads[i].bits;
896ea2ce1baSAndrii Nakryiko 		pad_type = pads[i].name;
897ea2ce1baSAndrii Nakryiko 
898ea2ce1baSAndrii Nakryiko 		new_off = roundup(cur_off, pad_bits);
899ea2ce1baSAndrii Nakryiko 		if (new_off <= next_off)
900ea2ce1baSAndrii Nakryiko 			break;
901351131b5SAndrii Nakryiko 	}
902ea2ce1baSAndrii Nakryiko 
903ea2ce1baSAndrii Nakryiko 	if (new_off > cur_off && new_off <= next_off) {
904ea2ce1baSAndrii Nakryiko 		/* We need explicit `<type>: 0` aligning mark if next
905ea2ce1baSAndrii Nakryiko 		 * field is right on alignment offset and its
906ea2ce1baSAndrii Nakryiko 		 * alignment requirement is less strict than <type>'s
907ea2ce1baSAndrii Nakryiko 		 * alignment (so compiler won't naturally align to the
908ea2ce1baSAndrii Nakryiko 		 * offset we expect), or if subsequent `<type>: X`,
909ea2ce1baSAndrii Nakryiko 		 * will actually completely fit in the remaining hole,
910ea2ce1baSAndrii Nakryiko 		 * making compiler basically ignore `<type>: X`
911ea2ce1baSAndrii Nakryiko 		 * completely.
912ea2ce1baSAndrii Nakryiko 		 */
913ea2ce1baSAndrii Nakryiko 		if (in_bitfield ||
914ea2ce1baSAndrii Nakryiko 		    (new_off == next_off && roundup(cur_off, next_align * 8) != new_off) ||
915ea2ce1baSAndrii Nakryiko 		    (new_off != next_off && next_off - new_off <= new_off - cur_off))
916ea2ce1baSAndrii Nakryiko 			/* but for bitfields we'll emit explicit bit count */
917ea2ce1baSAndrii Nakryiko 			btf_dump_printf(d, "\n%s%s: %d;", pfx(lvl), pad_type,
918ea2ce1baSAndrii Nakryiko 					in_bitfield ? new_off - cur_off : 0);
919ea2ce1baSAndrii Nakryiko 		cur_off = new_off;
920ea2ce1baSAndrii Nakryiko 	}
921ea2ce1baSAndrii Nakryiko 
922ea2ce1baSAndrii Nakryiko 	/* Now we know we start at naturally aligned offset for a chosen
923ea2ce1baSAndrii Nakryiko 	 * padding type (long, int, short, or char), and so the rest is just
924ea2ce1baSAndrii Nakryiko 	 * a straightforward filling of remaining padding gap with full
925ea2ce1baSAndrii Nakryiko 	 * `<type>: sizeof(<type>);` markers, except for the last one, which
926ea2ce1baSAndrii Nakryiko 	 * might need smaller than sizeof(<type>) padding.
927ea2ce1baSAndrii Nakryiko 	 */
928ea2ce1baSAndrii Nakryiko 	while (cur_off != next_off) {
929ea2ce1baSAndrii Nakryiko 		bits = min(next_off - cur_off, pad_bits);
930ea2ce1baSAndrii Nakryiko 		if (bits == pad_bits) {
931351131b5SAndrii Nakryiko 			btf_dump_printf(d, "\n%s%s: %d;", pfx(lvl), pad_type, pad_bits);
932ea2ce1baSAndrii Nakryiko 			cur_off += bits;
933ea2ce1baSAndrii Nakryiko 			continue;
934ea2ce1baSAndrii Nakryiko 		}
935ea2ce1baSAndrii Nakryiko 		/* For the remainder padding that doesn't cover entire
936ea2ce1baSAndrii Nakryiko 		 * pad_type bit length, we pick the smallest necessary type.
937ea2ce1baSAndrii Nakryiko 		 * This is pure aesthetics, we could have just used `long`,
938ea2ce1baSAndrii Nakryiko 		 * but having smallest necessary one communicates better the
939ea2ce1baSAndrii Nakryiko 		 * scale of the padding gap.
940ea2ce1baSAndrii Nakryiko 		 */
941ea2ce1baSAndrii Nakryiko 		for (i = ARRAY_SIZE(pads) - 1; i >= 0; i--) {
942ea2ce1baSAndrii Nakryiko 			pad_type = pads[i].name;
943ea2ce1baSAndrii Nakryiko 			pad_bits = pads[i].bits;
944ea2ce1baSAndrii Nakryiko 			if (pad_bits < bits)
945ea2ce1baSAndrii Nakryiko 				continue;
946ea2ce1baSAndrii Nakryiko 
947ea2ce1baSAndrii Nakryiko 			btf_dump_printf(d, "\n%s%s: %d;", pfx(lvl), pad_type, bits);
948ea2ce1baSAndrii Nakryiko 			cur_off += bits;
949ea2ce1baSAndrii Nakryiko 			break;
950ea2ce1baSAndrii Nakryiko 		}
951351131b5SAndrii Nakryiko 	}
952351131b5SAndrii Nakryiko }
953351131b5SAndrii Nakryiko 
btf_dump_emit_struct_fwd(struct btf_dump * d,__u32 id,const struct btf_type * t)954351131b5SAndrii Nakryiko static void btf_dump_emit_struct_fwd(struct btf_dump *d, __u32 id,
955351131b5SAndrii Nakryiko 				     const struct btf_type *t)
956351131b5SAndrii Nakryiko {
957920d16afSAlan Maguire 	btf_dump_printf(d, "%s%s%s",
958b03bc685SAndrii Nakryiko 			btf_is_struct(t) ? "struct" : "union",
959920d16afSAlan Maguire 			t->name_off ? " " : "",
960351131b5SAndrii Nakryiko 			btf_dump_type_name(d, id));
961351131b5SAndrii Nakryiko }
962351131b5SAndrii Nakryiko 
btf_dump_emit_struct_def(struct btf_dump * d,__u32 id,const struct btf_type * t,int lvl)963351131b5SAndrii Nakryiko static void btf_dump_emit_struct_def(struct btf_dump *d,
964351131b5SAndrii Nakryiko 				     __u32 id,
965351131b5SAndrii Nakryiko 				     const struct btf_type *t,
966351131b5SAndrii Nakryiko 				     int lvl)
967351131b5SAndrii Nakryiko {
968b03bc685SAndrii Nakryiko 	const struct btf_member *m = btf_members(t);
969b03bc685SAndrii Nakryiko 	bool is_struct = btf_is_struct(t);
970ea2ce1baSAndrii Nakryiko 	bool packed, prev_bitfield = false;
971ea2ce1baSAndrii Nakryiko 	int align, i, off = 0;
972b03bc685SAndrii Nakryiko 	__u16 vlen = btf_vlen(t);
973351131b5SAndrii Nakryiko 
974ea2ce1baSAndrii Nakryiko 	align = btf__align_of(d->btf, id);
975351131b5SAndrii Nakryiko 	packed = is_struct ? btf_is_struct_packed(d->btf, id, t) : 0;
976351131b5SAndrii Nakryiko 
977351131b5SAndrii Nakryiko 	btf_dump_printf(d, "%s%s%s {",
978351131b5SAndrii Nakryiko 			is_struct ? "struct" : "union",
979351131b5SAndrii Nakryiko 			t->name_off ? " " : "",
980351131b5SAndrii Nakryiko 			btf_dump_type_name(d, id));
981351131b5SAndrii Nakryiko 
982351131b5SAndrii Nakryiko 	for (i = 0; i < vlen; i++, m++) {
983351131b5SAndrii Nakryiko 		const char *fname;
984ea2ce1baSAndrii Nakryiko 		int m_off, m_sz, m_align;
985ea2ce1baSAndrii Nakryiko 		bool in_bitfield;
986351131b5SAndrii Nakryiko 
987351131b5SAndrii Nakryiko 		fname = btf_name_of(d, m->name_off);
988b03bc685SAndrii Nakryiko 		m_sz = btf_member_bitfield_size(t, i);
989b03bc685SAndrii Nakryiko 		m_off = btf_member_bit_offset(t, i);
990ea2ce1baSAndrii Nakryiko 		m_align = packed ? 1 : btf__align_of(d->btf, m->type);
991351131b5SAndrii Nakryiko 
992ea2ce1baSAndrii Nakryiko 		in_bitfield = prev_bitfield && m_sz != 0;
993ea2ce1baSAndrii Nakryiko 
994ea2ce1baSAndrii Nakryiko 		btf_dump_emit_bit_padding(d, off, m_off, m_align, in_bitfield, lvl + 1);
995351131b5SAndrii Nakryiko 		btf_dump_printf(d, "\n%s", pfx(lvl + 1));
996351131b5SAndrii Nakryiko 		btf_dump_emit_type_decl(d, m->type, fname, lvl + 1);
997351131b5SAndrii Nakryiko 
998351131b5SAndrii Nakryiko 		if (m_sz) {
999351131b5SAndrii Nakryiko 			btf_dump_printf(d, ": %d", m_sz);
1000351131b5SAndrii Nakryiko 			off = m_off + m_sz;
1001ea2ce1baSAndrii Nakryiko 			prev_bitfield = true;
1002351131b5SAndrii Nakryiko 		} else {
10033fb1a96aSAndrii Nakryiko 			m_sz = max((__s64)0, btf__resolve_size(d->btf, m->type));
1004351131b5SAndrii Nakryiko 			off = m_off + m_sz * 8;
1005ea2ce1baSAndrii Nakryiko 			prev_bitfield = false;
1006351131b5SAndrii Nakryiko 		}
1007ea2ce1baSAndrii Nakryiko 
1008351131b5SAndrii Nakryiko 		btf_dump_printf(d, ";");
1009351131b5SAndrii Nakryiko 	}
1010351131b5SAndrii Nakryiko 
1011b4099769SAndrii Nakryiko 	/* pad at the end, if necessary */
1012ea2ce1baSAndrii Nakryiko 	if (is_struct)
1013ea2ce1baSAndrii Nakryiko 		btf_dump_emit_bit_padding(d, off, t->size * 8, align, false, lvl + 1);
1014b4099769SAndrii Nakryiko 
101544a726c3SEduard Zingerman 	/*
101644a726c3SEduard Zingerman 	 * Keep `struct empty {}` on a single line,
101744a726c3SEduard Zingerman 	 * only print newline when there are regular or padding fields.
101844a726c3SEduard Zingerman 	 */
1019872aec4bSAndrii Nakryiko 	if (vlen || t->size) {
1020351131b5SAndrii Nakryiko 		btf_dump_printf(d, "\n");
1021351131b5SAndrii Nakryiko 		btf_dump_printf(d, "%s}", pfx(lvl));
1022872aec4bSAndrii Nakryiko 	} else {
1023872aec4bSAndrii Nakryiko 		btf_dump_printf(d, "}");
1024872aec4bSAndrii Nakryiko 	}
1025351131b5SAndrii Nakryiko 	if (packed)
1026351131b5SAndrii Nakryiko 		btf_dump_printf(d, " __attribute__((packed))");
1027351131b5SAndrii Nakryiko }
1028351131b5SAndrii Nakryiko 
1029702eddc7SJean-Philippe Brucker static const char *missing_base_types[][2] = {
1030702eddc7SJean-Philippe Brucker 	/*
1031702eddc7SJean-Philippe Brucker 	 * GCC emits typedefs to its internal __PolyX_t types when compiling Arm
1032702eddc7SJean-Philippe Brucker 	 * SIMD intrinsics. Alias them to standard base types.
1033702eddc7SJean-Philippe Brucker 	 */
1034702eddc7SJean-Philippe Brucker 	{ "__Poly8_t",		"unsigned char" },
1035702eddc7SJean-Philippe Brucker 	{ "__Poly16_t",		"unsigned short" },
1036702eddc7SJean-Philippe Brucker 	{ "__Poly64_t",		"unsigned long long" },
1037702eddc7SJean-Philippe Brucker 	{ "__Poly128_t",	"unsigned __int128" },
1038702eddc7SJean-Philippe Brucker };
1039702eddc7SJean-Philippe Brucker 
btf_dump_emit_missing_aliases(struct btf_dump * d,__u32 id,const struct btf_type * t)1040702eddc7SJean-Philippe Brucker static void btf_dump_emit_missing_aliases(struct btf_dump *d, __u32 id,
1041702eddc7SJean-Philippe Brucker 					  const struct btf_type *t)
1042702eddc7SJean-Philippe Brucker {
1043702eddc7SJean-Philippe Brucker 	const char *name = btf_dump_type_name(d, id);
1044702eddc7SJean-Philippe Brucker 	int i;
1045702eddc7SJean-Philippe Brucker 
1046702eddc7SJean-Philippe Brucker 	for (i = 0; i < ARRAY_SIZE(missing_base_types); i++) {
1047702eddc7SJean-Philippe Brucker 		if (strcmp(name, missing_base_types[i][0]) == 0) {
1048702eddc7SJean-Philippe Brucker 			btf_dump_printf(d, "typedef %s %s;\n\n",
1049702eddc7SJean-Philippe Brucker 					missing_base_types[i][1], name);
1050702eddc7SJean-Philippe Brucker 			break;
1051702eddc7SJean-Philippe Brucker 		}
1052702eddc7SJean-Philippe Brucker 	}
1053702eddc7SJean-Philippe Brucker }
1054702eddc7SJean-Philippe Brucker 
btf_dump_emit_enum_fwd(struct btf_dump * d,__u32 id,const struct btf_type * t)1055351131b5SAndrii Nakryiko static void btf_dump_emit_enum_fwd(struct btf_dump *d, __u32 id,
1056351131b5SAndrii Nakryiko 				   const struct btf_type *t)
1057351131b5SAndrii Nakryiko {
1058351131b5SAndrii Nakryiko 	btf_dump_printf(d, "enum %s", btf_dump_type_name(d, id));
1059351131b5SAndrii Nakryiko }
1060351131b5SAndrii Nakryiko 
btf_dump_emit_enum32_val(struct btf_dump * d,const struct btf_type * t,int lvl,__u16 vlen)1061d90ec262SYonghong Song static void btf_dump_emit_enum32_val(struct btf_dump *d,
1062351131b5SAndrii Nakryiko 				     const struct btf_type *t,
1063d90ec262SYonghong Song 				     int lvl, __u16 vlen)
1064351131b5SAndrii Nakryiko {
1065b03bc685SAndrii Nakryiko 	const struct btf_enum *v = btf_enum(t);
1066d90ec262SYonghong Song 	bool is_signed = btf_kflag(t);
1067d90ec262SYonghong Song 	const char *fmt_str;
1068351131b5SAndrii Nakryiko 	const char *name;
1069351131b5SAndrii Nakryiko 	size_t dup_cnt;
1070351131b5SAndrii Nakryiko 	int i;
1071351131b5SAndrii Nakryiko 
1072351131b5SAndrii Nakryiko 	for (i = 0; i < vlen; i++, v++) {
1073351131b5SAndrii Nakryiko 		name = btf_name_of(d, v->name_off);
1074351131b5SAndrii Nakryiko 		/* enumerators share namespace with typedef idents */
1075351131b5SAndrii Nakryiko 		dup_cnt = btf_dump_name_dups(d, d->ident_names, name);
1076351131b5SAndrii Nakryiko 		if (dup_cnt > 1) {
1077d90ec262SYonghong Song 			fmt_str = is_signed ? "\n%s%s___%zd = %d," : "\n%s%s___%zd = %u,";
1078d90ec262SYonghong Song 			btf_dump_printf(d, fmt_str, pfx(lvl + 1), name, dup_cnt, v->val);
1079351131b5SAndrii Nakryiko 		} else {
1080d90ec262SYonghong Song 			fmt_str = is_signed ? "\n%s%s = %d," : "\n%s%s = %u,";
1081d90ec262SYonghong Song 			btf_dump_printf(d, fmt_str, pfx(lvl + 1), name, v->val);
1082d90ec262SYonghong Song 		}
1083d90ec262SYonghong Song 	}
1084d90ec262SYonghong Song }
1085d90ec262SYonghong Song 
btf_dump_emit_enum64_val(struct btf_dump * d,const struct btf_type * t,int lvl,__u16 vlen)1086d90ec262SYonghong Song static void btf_dump_emit_enum64_val(struct btf_dump *d,
1087d90ec262SYonghong Song 				     const struct btf_type *t,
1088d90ec262SYonghong Song 				     int lvl, __u16 vlen)
1089d90ec262SYonghong Song {
1090d90ec262SYonghong Song 	const struct btf_enum64 *v = btf_enum64(t);
1091d90ec262SYonghong Song 	bool is_signed = btf_kflag(t);
1092d90ec262SYonghong Song 	const char *fmt_str;
1093d90ec262SYonghong Song 	const char *name;
1094d90ec262SYonghong Song 	size_t dup_cnt;
1095d90ec262SYonghong Song 	__u64 val;
1096d90ec262SYonghong Song 	int i;
1097d90ec262SYonghong Song 
1098d90ec262SYonghong Song 	for (i = 0; i < vlen; i++, v++) {
1099d90ec262SYonghong Song 		name = btf_name_of(d, v->name_off);
1100d90ec262SYonghong Song 		dup_cnt = btf_dump_name_dups(d, d->ident_names, name);
1101d90ec262SYonghong Song 		val = btf_enum64_value(v);
1102d90ec262SYonghong Song 		if (dup_cnt > 1) {
1103d90ec262SYonghong Song 			fmt_str = is_signed ? "\n%s%s___%zd = %lldLL,"
1104d90ec262SYonghong Song 					    : "\n%s%s___%zd = %lluULL,";
1105d90ec262SYonghong Song 			btf_dump_printf(d, fmt_str,
1106d90ec262SYonghong Song 					pfx(lvl + 1), name, dup_cnt,
1107d90ec262SYonghong Song 					(unsigned long long)val);
1108d90ec262SYonghong Song 		} else {
1109d90ec262SYonghong Song 			fmt_str = is_signed ? "\n%s%s = %lldLL,"
1110d90ec262SYonghong Song 					    : "\n%s%s = %lluULL,";
1111d90ec262SYonghong Song 			btf_dump_printf(d, fmt_str,
1112351131b5SAndrii Nakryiko 					pfx(lvl + 1), name,
1113d90ec262SYonghong Song 					(unsigned long long)val);
1114351131b5SAndrii Nakryiko 		}
1115351131b5SAndrii Nakryiko 	}
1116d90ec262SYonghong Song }
btf_dump_emit_enum_def(struct btf_dump * d,__u32 id,const struct btf_type * t,int lvl)1117d90ec262SYonghong Song static void btf_dump_emit_enum_def(struct btf_dump *d, __u32 id,
1118d90ec262SYonghong Song 				   const struct btf_type *t,
1119d90ec262SYonghong Song 				   int lvl)
1120d90ec262SYonghong Song {
1121d90ec262SYonghong Song 	__u16 vlen = btf_vlen(t);
1122d90ec262SYonghong Song 
1123d90ec262SYonghong Song 	btf_dump_printf(d, "enum%s%s",
1124d90ec262SYonghong Song 			t->name_off ? " " : "",
1125d90ec262SYonghong Song 			btf_dump_type_name(d, id));
1126d90ec262SYonghong Song 
1127d90ec262SYonghong Song 	if (!vlen)
1128d90ec262SYonghong Song 		return;
1129d90ec262SYonghong Song 
1130d90ec262SYonghong Song 	btf_dump_printf(d, " {");
1131d90ec262SYonghong Song 	if (btf_is_enum(t))
1132d90ec262SYonghong Song 		btf_dump_emit_enum32_val(d, t, lvl, vlen);
1133d90ec262SYonghong Song 	else
1134d90ec262SYonghong Song 		btf_dump_emit_enum64_val(d, t, lvl, vlen);
1135351131b5SAndrii Nakryiko 	btf_dump_printf(d, "\n%s}", pfx(lvl));
113621a9a1bcSAndrii Nakryiko 
113721a9a1bcSAndrii Nakryiko 	/* special case enums with special sizes */
113821a9a1bcSAndrii Nakryiko 	if (t->size == 1) {
113921a9a1bcSAndrii Nakryiko 		/* one-byte enums can be forced with mode(byte) attribute */
114021a9a1bcSAndrii Nakryiko 		btf_dump_printf(d, " __attribute__((mode(byte)))");
114121a9a1bcSAndrii Nakryiko 	} else if (t->size == 8 && d->ptr_sz == 8) {
114221a9a1bcSAndrii Nakryiko 		/* enum can be 8-byte sized if one of the enumerator values
114321a9a1bcSAndrii Nakryiko 		 * doesn't fit in 32-bit integer, or by adding mode(word)
114421a9a1bcSAndrii Nakryiko 		 * attribute (but probably only on 64-bit architectures); do
114521a9a1bcSAndrii Nakryiko 		 * our best here to try to satisfy the contract without adding
114621a9a1bcSAndrii Nakryiko 		 * unnecessary attributes
114721a9a1bcSAndrii Nakryiko 		 */
114821a9a1bcSAndrii Nakryiko 		bool needs_word_mode;
114921a9a1bcSAndrii Nakryiko 
115021a9a1bcSAndrii Nakryiko 		if (btf_is_enum(t)) {
115121a9a1bcSAndrii Nakryiko 			/* enum can't represent 64-bit values, so we need word mode */
115221a9a1bcSAndrii Nakryiko 			needs_word_mode = true;
115321a9a1bcSAndrii Nakryiko 		} else {
115421a9a1bcSAndrii Nakryiko 			/* enum64 needs mode(word) if none of its values has
115521a9a1bcSAndrii Nakryiko 			 * non-zero upper 32-bits (which means that all values
115621a9a1bcSAndrii Nakryiko 			 * fit in 32-bit integers and won't cause compiler to
115721a9a1bcSAndrii Nakryiko 			 * bump enum to be 64-bit naturally
115821a9a1bcSAndrii Nakryiko 			 */
115921a9a1bcSAndrii Nakryiko 			int i;
116021a9a1bcSAndrii Nakryiko 
116121a9a1bcSAndrii Nakryiko 			needs_word_mode = true;
116221a9a1bcSAndrii Nakryiko 			for (i = 0; i < vlen; i++) {
116321a9a1bcSAndrii Nakryiko 				if (btf_enum64(t)[i].val_hi32 != 0) {
116421a9a1bcSAndrii Nakryiko 					needs_word_mode = false;
116521a9a1bcSAndrii Nakryiko 					break;
116621a9a1bcSAndrii Nakryiko 				}
116721a9a1bcSAndrii Nakryiko 			}
116821a9a1bcSAndrii Nakryiko 		}
116921a9a1bcSAndrii Nakryiko 		if (needs_word_mode)
117021a9a1bcSAndrii Nakryiko 			btf_dump_printf(d, " __attribute__((mode(word)))");
117121a9a1bcSAndrii Nakryiko 	}
117221a9a1bcSAndrii Nakryiko 
1173351131b5SAndrii Nakryiko }
1174351131b5SAndrii Nakryiko 
btf_dump_emit_fwd_def(struct btf_dump * d,__u32 id,const struct btf_type * t)1175351131b5SAndrii Nakryiko static void btf_dump_emit_fwd_def(struct btf_dump *d, __u32 id,
1176351131b5SAndrii Nakryiko 				  const struct btf_type *t)
1177351131b5SAndrii Nakryiko {
1178351131b5SAndrii Nakryiko 	const char *name = btf_dump_type_name(d, id);
1179351131b5SAndrii Nakryiko 
1180b03bc685SAndrii Nakryiko 	if (btf_kflag(t))
1181351131b5SAndrii Nakryiko 		btf_dump_printf(d, "union %s", name);
1182351131b5SAndrii Nakryiko 	else
1183351131b5SAndrii Nakryiko 		btf_dump_printf(d, "struct %s", name);
1184351131b5SAndrii Nakryiko }
1185351131b5SAndrii Nakryiko 
btf_dump_emit_typedef_def(struct btf_dump * d,__u32 id,const struct btf_type * t,int lvl)1186351131b5SAndrii Nakryiko static void btf_dump_emit_typedef_def(struct btf_dump *d, __u32 id,
1187351131b5SAndrii Nakryiko 				     const struct btf_type *t, int lvl)
1188351131b5SAndrii Nakryiko {
1189351131b5SAndrii Nakryiko 	const char *name = btf_dump_ident_name(d, id);
1190351131b5SAndrii Nakryiko 
1191e78dcbf4SAndrii Nakryiko 	/*
1192e78dcbf4SAndrii Nakryiko 	 * Old GCC versions are emitting invalid typedef for __gnuc_va_list
1193e78dcbf4SAndrii Nakryiko 	 * pointing to VOID. This generates warnings from btf_dump() and
1194e78dcbf4SAndrii Nakryiko 	 * results in uncompilable header file, so we are fixing it up here
1195e78dcbf4SAndrii Nakryiko 	 * with valid typedef into __builtin_va_list.
1196e78dcbf4SAndrii Nakryiko 	 */
1197e78dcbf4SAndrii Nakryiko 	if (t->type == 0 && strcmp(name, "__gnuc_va_list") == 0) {
1198e78dcbf4SAndrii Nakryiko 		btf_dump_printf(d, "typedef __builtin_va_list __gnuc_va_list");
1199e78dcbf4SAndrii Nakryiko 		return;
1200e78dcbf4SAndrii Nakryiko 	}
1201e78dcbf4SAndrii Nakryiko 
1202351131b5SAndrii Nakryiko 	btf_dump_printf(d, "typedef ");
1203351131b5SAndrii Nakryiko 	btf_dump_emit_type_decl(d, t->type, name, lvl);
1204351131b5SAndrii Nakryiko }
1205351131b5SAndrii Nakryiko 
btf_dump_push_decl_stack_id(struct btf_dump * d,__u32 id)1206351131b5SAndrii Nakryiko static int btf_dump_push_decl_stack_id(struct btf_dump *d, __u32 id)
1207351131b5SAndrii Nakryiko {
1208351131b5SAndrii Nakryiko 	__u32 *new_stack;
1209351131b5SAndrii Nakryiko 	size_t new_cap;
1210351131b5SAndrii Nakryiko 
1211351131b5SAndrii Nakryiko 	if (d->decl_stack_cnt >= d->decl_stack_cap) {
1212351131b5SAndrii Nakryiko 		new_cap = max(16, d->decl_stack_cap * 3 / 2);
1213029258d7SAndrii Nakryiko 		new_stack = libbpf_reallocarray(d->decl_stack, new_cap, sizeof(new_stack[0]));
1214351131b5SAndrii Nakryiko 		if (!new_stack)
1215351131b5SAndrii Nakryiko 			return -ENOMEM;
1216351131b5SAndrii Nakryiko 		d->decl_stack = new_stack;
1217351131b5SAndrii Nakryiko 		d->decl_stack_cap = new_cap;
1218351131b5SAndrii Nakryiko 	}
1219351131b5SAndrii Nakryiko 
1220351131b5SAndrii Nakryiko 	d->decl_stack[d->decl_stack_cnt++] = id;
1221351131b5SAndrii Nakryiko 
1222351131b5SAndrii Nakryiko 	return 0;
1223351131b5SAndrii Nakryiko }
1224351131b5SAndrii Nakryiko 
1225351131b5SAndrii Nakryiko /*
1226351131b5SAndrii Nakryiko  * Emit type declaration (e.g., field type declaration in a struct or argument
1227351131b5SAndrii Nakryiko  * declaration in function prototype) in correct C syntax.
1228351131b5SAndrii Nakryiko  *
1229351131b5SAndrii Nakryiko  * For most types it's trivial, but there are few quirky type declaration
1230351131b5SAndrii Nakryiko  * cases worth mentioning:
1231351131b5SAndrii Nakryiko  *   - function prototypes (especially nesting of function prototypes);
1232351131b5SAndrii Nakryiko  *   - arrays;
1233351131b5SAndrii Nakryiko  *   - const/volatile/restrict for pointers vs other types.
1234351131b5SAndrii Nakryiko  *
1235351131b5SAndrii Nakryiko  * For a good discussion of *PARSING* C syntax (as a human), see
1236351131b5SAndrii Nakryiko  * Peter van der Linden's "Expert C Programming: Deep C Secrets",
1237351131b5SAndrii Nakryiko  * Ch.3 "Unscrambling Declarations in C".
1238351131b5SAndrii Nakryiko  *
1239351131b5SAndrii Nakryiko  * It won't help with BTF to C conversion much, though, as it's an opposite
1240351131b5SAndrii Nakryiko  * problem. So we came up with this algorithm in reverse to van der Linden's
1241351131b5SAndrii Nakryiko  * parsing algorithm. It goes from structured BTF representation of type
1242351131b5SAndrii Nakryiko  * declaration to a valid compilable C syntax.
1243351131b5SAndrii Nakryiko  *
1244351131b5SAndrii Nakryiko  * For instance, consider this C typedef:
1245351131b5SAndrii Nakryiko  *	typedef const int * const * arr[10] arr_t;
1246351131b5SAndrii Nakryiko  * It will be represented in BTF with this chain of BTF types:
1247351131b5SAndrii Nakryiko  *	[typedef] -> [array] -> [ptr] -> [const] -> [ptr] -> [const] -> [int]
1248351131b5SAndrii Nakryiko  *
1249351131b5SAndrii Nakryiko  * Notice how [const] modifier always goes before type it modifies in BTF type
1250351131b5SAndrii Nakryiko  * graph, but in C syntax, const/volatile/restrict modifiers are written to
1251351131b5SAndrii Nakryiko  * the right of pointers, but to the left of other types. There are also other
1252351131b5SAndrii Nakryiko  * quirks, like function pointers, arrays of them, functions returning other
1253351131b5SAndrii Nakryiko  * functions, etc.
1254351131b5SAndrii Nakryiko  *
1255351131b5SAndrii Nakryiko  * We handle that by pushing all the types to a stack, until we hit "terminal"
1256351131b5SAndrii Nakryiko  * type (int/enum/struct/union/fwd). Then depending on the kind of a type on
1257351131b5SAndrii Nakryiko  * top of a stack, modifiers are handled differently. Array/function pointers
1258351131b5SAndrii Nakryiko  * have also wildly different syntax and how nesting of them are done. See
1259351131b5SAndrii Nakryiko  * code for authoritative definition.
1260351131b5SAndrii Nakryiko  *
1261351131b5SAndrii Nakryiko  * To avoid allocating new stack for each independent chain of BTF types, we
1262351131b5SAndrii Nakryiko  * share one bigger stack, with each chain working only on its own local view
1263351131b5SAndrii Nakryiko  * of a stack frame. Some care is required to "pop" stack frames after
1264351131b5SAndrii Nakryiko  * processing type declaration chain.
1265351131b5SAndrii Nakryiko  */
btf_dump__emit_type_decl(struct btf_dump * d,__u32 id,const struct btf_dump_emit_type_decl_opts * opts)12669f81654eSAndrii Nakryiko int btf_dump__emit_type_decl(struct btf_dump *d, __u32 id,
12679f81654eSAndrii Nakryiko 			     const struct btf_dump_emit_type_decl_opts *opts)
12689f81654eSAndrii Nakryiko {
12699f81654eSAndrii Nakryiko 	const char *fname;
12709c6c5c48SAndrii Nakryiko 	int lvl, err;
12719f81654eSAndrii Nakryiko 
12729f81654eSAndrii Nakryiko 	if (!OPTS_VALID(opts, btf_dump_emit_type_decl_opts))
1273e9fc3ce9SAndrii Nakryiko 		return libbpf_err(-EINVAL);
12749f81654eSAndrii Nakryiko 
12759c6c5c48SAndrii Nakryiko 	err = btf_dump_resize(d);
12769c6c5c48SAndrii Nakryiko 	if (err)
1277e9fc3ce9SAndrii Nakryiko 		return libbpf_err(err);
12789c6c5c48SAndrii Nakryiko 
1279320a3606SAndrii Nakryiko 	fname = OPTS_GET(opts, field_name, "");
12809f81654eSAndrii Nakryiko 	lvl = OPTS_GET(opts, indent_level, 0);
12817c819e70SAndrii Nakryiko 	d->strip_mods = OPTS_GET(opts, strip_mods, false);
12829f81654eSAndrii Nakryiko 	btf_dump_emit_type_decl(d, id, fname, lvl);
12837c819e70SAndrii Nakryiko 	d->strip_mods = false;
12849f81654eSAndrii Nakryiko 	return 0;
12859f81654eSAndrii Nakryiko }
12869f81654eSAndrii Nakryiko 
btf_dump_emit_type_decl(struct btf_dump * d,__u32 id,const char * fname,int lvl)1287351131b5SAndrii Nakryiko static void btf_dump_emit_type_decl(struct btf_dump *d, __u32 id,
1288351131b5SAndrii Nakryiko 				    const char *fname, int lvl)
1289351131b5SAndrii Nakryiko {
1290351131b5SAndrii Nakryiko 	struct id_stack decl_stack;
1291351131b5SAndrii Nakryiko 	const struct btf_type *t;
1292351131b5SAndrii Nakryiko 	int err, stack_start;
1293351131b5SAndrii Nakryiko 
1294351131b5SAndrii Nakryiko 	stack_start = d->decl_stack_cnt;
1295351131b5SAndrii Nakryiko 	for (;;) {
12967c819e70SAndrii Nakryiko 		t = btf__type_by_id(d->btf, id);
12977c819e70SAndrii Nakryiko 		if (d->strip_mods && btf_is_mod(t))
12987c819e70SAndrii Nakryiko 			goto skip_mod;
12997c819e70SAndrii Nakryiko 
1300351131b5SAndrii Nakryiko 		err = btf_dump_push_decl_stack_id(d, id);
1301351131b5SAndrii Nakryiko 		if (err < 0) {
1302351131b5SAndrii Nakryiko 			/*
1303351131b5SAndrii Nakryiko 			 * if we don't have enough memory for entire type decl
1304351131b5SAndrii Nakryiko 			 * chain, restore stack, emit warning, and try to
1305351131b5SAndrii Nakryiko 			 * proceed nevertheless
1306351131b5SAndrii Nakryiko 			 */
1307be18010eSKefeng Wang 			pr_warn("not enough memory for decl stack:%d", err);
1308351131b5SAndrii Nakryiko 			d->decl_stack_cnt = stack_start;
1309351131b5SAndrii Nakryiko 			return;
1310351131b5SAndrii Nakryiko 		}
13117c819e70SAndrii Nakryiko skip_mod:
1312351131b5SAndrii Nakryiko 		/* VOID */
1313351131b5SAndrii Nakryiko 		if (id == 0)
1314351131b5SAndrii Nakryiko 			break;
1315351131b5SAndrii Nakryiko 
1316b03bc685SAndrii Nakryiko 		switch (btf_kind(t)) {
1317351131b5SAndrii Nakryiko 		case BTF_KIND_PTR:
1318351131b5SAndrii Nakryiko 		case BTF_KIND_VOLATILE:
1319351131b5SAndrii Nakryiko 		case BTF_KIND_CONST:
1320351131b5SAndrii Nakryiko 		case BTF_KIND_RESTRICT:
1321351131b5SAndrii Nakryiko 		case BTF_KIND_FUNC_PROTO:
13222dc1e488SYonghong Song 		case BTF_KIND_TYPE_TAG:
1323351131b5SAndrii Nakryiko 			id = t->type;
1324351131b5SAndrii Nakryiko 			break;
1325b03bc685SAndrii Nakryiko 		case BTF_KIND_ARRAY:
1326b03bc685SAndrii Nakryiko 			id = btf_array(t)->type;
1327351131b5SAndrii Nakryiko 			break;
1328351131b5SAndrii Nakryiko 		case BTF_KIND_INT:
1329351131b5SAndrii Nakryiko 		case BTF_KIND_ENUM:
1330d90ec262SYonghong Song 		case BTF_KIND_ENUM64:
1331351131b5SAndrii Nakryiko 		case BTF_KIND_FWD:
1332351131b5SAndrii Nakryiko 		case BTF_KIND_STRUCT:
1333351131b5SAndrii Nakryiko 		case BTF_KIND_UNION:
1334351131b5SAndrii Nakryiko 		case BTF_KIND_TYPEDEF:
133522541a9eSIlya Leoshkevich 		case BTF_KIND_FLOAT:
1336351131b5SAndrii Nakryiko 			goto done;
1337351131b5SAndrii Nakryiko 		default:
1338be18010eSKefeng Wang 			pr_warn("unexpected type in decl chain, kind:%u, id:[%u]\n",
1339b03bc685SAndrii Nakryiko 				btf_kind(t), id);
1340351131b5SAndrii Nakryiko 			goto done;
1341351131b5SAndrii Nakryiko 		}
1342351131b5SAndrii Nakryiko 	}
1343351131b5SAndrii Nakryiko done:
1344351131b5SAndrii Nakryiko 	/*
1345351131b5SAndrii Nakryiko 	 * We might be inside a chain of declarations (e.g., array of function
1346351131b5SAndrii Nakryiko 	 * pointers returning anonymous (so inlined) structs, having another
1347351131b5SAndrii Nakryiko 	 * array field). Each of those needs its own "stack frame" to handle
1348351131b5SAndrii Nakryiko 	 * emitting of declarations. Those stack frames are non-overlapping
1349351131b5SAndrii Nakryiko 	 * portions of shared btf_dump->decl_stack. To make it a bit nicer to
1350351131b5SAndrii Nakryiko 	 * handle this set of nested stacks, we create a view corresponding to
1351351131b5SAndrii Nakryiko 	 * our own "stack frame" and work with it as an independent stack.
1352351131b5SAndrii Nakryiko 	 * We'll need to clean up after emit_type_chain() returns, though.
1353351131b5SAndrii Nakryiko 	 */
1354351131b5SAndrii Nakryiko 	decl_stack.ids = d->decl_stack + stack_start;
1355351131b5SAndrii Nakryiko 	decl_stack.cnt = d->decl_stack_cnt - stack_start;
1356351131b5SAndrii Nakryiko 	btf_dump_emit_type_chain(d, &decl_stack, fname, lvl);
1357351131b5SAndrii Nakryiko 	/*
1358351131b5SAndrii Nakryiko 	 * emit_type_chain() guarantees that it will pop its entire decl_stack
1359351131b5SAndrii Nakryiko 	 * frame before returning. But it works with a read-only view into
1360351131b5SAndrii Nakryiko 	 * decl_stack, so it doesn't actually pop anything from the
1361351131b5SAndrii Nakryiko 	 * perspective of shared btf_dump->decl_stack, per se. We need to
1362351131b5SAndrii Nakryiko 	 * reset decl_stack state to how it was before us to avoid it growing
1363351131b5SAndrii Nakryiko 	 * all the time.
1364351131b5SAndrii Nakryiko 	 */
1365351131b5SAndrii Nakryiko 	d->decl_stack_cnt = stack_start;
1366351131b5SAndrii Nakryiko }
1367351131b5SAndrii Nakryiko 
btf_dump_emit_mods(struct btf_dump * d,struct id_stack * decl_stack)1368351131b5SAndrii Nakryiko static void btf_dump_emit_mods(struct btf_dump *d, struct id_stack *decl_stack)
1369351131b5SAndrii Nakryiko {
1370351131b5SAndrii Nakryiko 	const struct btf_type *t;
1371351131b5SAndrii Nakryiko 	__u32 id;
1372351131b5SAndrii Nakryiko 
1373351131b5SAndrii Nakryiko 	while (decl_stack->cnt) {
1374351131b5SAndrii Nakryiko 		id = decl_stack->ids[decl_stack->cnt - 1];
1375351131b5SAndrii Nakryiko 		t = btf__type_by_id(d->btf, id);
1376351131b5SAndrii Nakryiko 
1377b03bc685SAndrii Nakryiko 		switch (btf_kind(t)) {
1378351131b5SAndrii Nakryiko 		case BTF_KIND_VOLATILE:
1379351131b5SAndrii Nakryiko 			btf_dump_printf(d, "volatile ");
1380351131b5SAndrii Nakryiko 			break;
1381351131b5SAndrii Nakryiko 		case BTF_KIND_CONST:
1382351131b5SAndrii Nakryiko 			btf_dump_printf(d, "const ");
1383351131b5SAndrii Nakryiko 			break;
1384351131b5SAndrii Nakryiko 		case BTF_KIND_RESTRICT:
1385351131b5SAndrii Nakryiko 			btf_dump_printf(d, "restrict ");
1386351131b5SAndrii Nakryiko 			break;
1387351131b5SAndrii Nakryiko 		default:
1388351131b5SAndrii Nakryiko 			return;
1389351131b5SAndrii Nakryiko 		}
1390351131b5SAndrii Nakryiko 		decl_stack->cnt--;
1391351131b5SAndrii Nakryiko 	}
1392351131b5SAndrii Nakryiko }
1393351131b5SAndrii Nakryiko 
btf_dump_drop_mods(struct btf_dump * d,struct id_stack * decl_stack)139432022fd9SAndrii Nakryiko static void btf_dump_drop_mods(struct btf_dump *d, struct id_stack *decl_stack)
139532022fd9SAndrii Nakryiko {
139632022fd9SAndrii Nakryiko 	const struct btf_type *t;
139732022fd9SAndrii Nakryiko 	__u32 id;
139832022fd9SAndrii Nakryiko 
139932022fd9SAndrii Nakryiko 	while (decl_stack->cnt) {
140032022fd9SAndrii Nakryiko 		id = decl_stack->ids[decl_stack->cnt - 1];
140132022fd9SAndrii Nakryiko 		t = btf__type_by_id(d->btf, id);
140232022fd9SAndrii Nakryiko 		if (!btf_is_mod(t))
140332022fd9SAndrii Nakryiko 			return;
140432022fd9SAndrii Nakryiko 		decl_stack->cnt--;
140532022fd9SAndrii Nakryiko 	}
140632022fd9SAndrii Nakryiko }
140732022fd9SAndrii Nakryiko 
btf_dump_emit_name(const struct btf_dump * d,const char * name,bool last_was_ptr)1408351131b5SAndrii Nakryiko static void btf_dump_emit_name(const struct btf_dump *d,
1409351131b5SAndrii Nakryiko 			       const char *name, bool last_was_ptr)
1410351131b5SAndrii Nakryiko {
1411351131b5SAndrii Nakryiko 	bool separate = name[0] && !last_was_ptr;
1412351131b5SAndrii Nakryiko 
1413351131b5SAndrii Nakryiko 	btf_dump_printf(d, "%s%s", separate ? " " : "", name);
1414351131b5SAndrii Nakryiko }
1415351131b5SAndrii Nakryiko 
btf_dump_emit_type_chain(struct btf_dump * d,struct id_stack * decls,const char * fname,int lvl)1416351131b5SAndrii Nakryiko static void btf_dump_emit_type_chain(struct btf_dump *d,
1417351131b5SAndrii Nakryiko 				     struct id_stack *decls,
1418351131b5SAndrii Nakryiko 				     const char *fname, int lvl)
1419351131b5SAndrii Nakryiko {
1420351131b5SAndrii Nakryiko 	/*
1421351131b5SAndrii Nakryiko 	 * last_was_ptr is used to determine if we need to separate pointer
1422351131b5SAndrii Nakryiko 	 * asterisk (*) from previous part of type signature with space, so
1423351131b5SAndrii Nakryiko 	 * that we get `int ***`, instead of `int * * *`. We default to true
1424351131b5SAndrii Nakryiko 	 * for cases where we have single pointer in a chain. E.g., in ptr ->
1425351131b5SAndrii Nakryiko 	 * func_proto case. func_proto will start a new emit_type_chain call
1426351131b5SAndrii Nakryiko 	 * with just ptr, which should be emitted as (*) or (*<fname>), so we
1427351131b5SAndrii Nakryiko 	 * don't want to prepend space for that last pointer.
1428351131b5SAndrii Nakryiko 	 */
1429351131b5SAndrii Nakryiko 	bool last_was_ptr = true;
1430351131b5SAndrii Nakryiko 	const struct btf_type *t;
1431351131b5SAndrii Nakryiko 	const char *name;
1432351131b5SAndrii Nakryiko 	__u16 kind;
1433351131b5SAndrii Nakryiko 	__u32 id;
1434351131b5SAndrii Nakryiko 
1435351131b5SAndrii Nakryiko 	while (decls->cnt) {
1436351131b5SAndrii Nakryiko 		id = decls->ids[--decls->cnt];
1437351131b5SAndrii Nakryiko 		if (id == 0) {
1438351131b5SAndrii Nakryiko 			/* VOID is a special snowflake */
1439351131b5SAndrii Nakryiko 			btf_dump_emit_mods(d, decls);
1440351131b5SAndrii Nakryiko 			btf_dump_printf(d, "void");
1441351131b5SAndrii Nakryiko 			last_was_ptr = false;
1442351131b5SAndrii Nakryiko 			continue;
1443351131b5SAndrii Nakryiko 		}
1444351131b5SAndrii Nakryiko 
1445351131b5SAndrii Nakryiko 		t = btf__type_by_id(d->btf, id);
1446b03bc685SAndrii Nakryiko 		kind = btf_kind(t);
1447351131b5SAndrii Nakryiko 
1448351131b5SAndrii Nakryiko 		switch (kind) {
1449351131b5SAndrii Nakryiko 		case BTF_KIND_INT:
145022541a9eSIlya Leoshkevich 		case BTF_KIND_FLOAT:
1451351131b5SAndrii Nakryiko 			btf_dump_emit_mods(d, decls);
1452351131b5SAndrii Nakryiko 			name = btf_name_of(d, t->name_off);
1453351131b5SAndrii Nakryiko 			btf_dump_printf(d, "%s", name);
1454351131b5SAndrii Nakryiko 			break;
1455351131b5SAndrii Nakryiko 		case BTF_KIND_STRUCT:
1456351131b5SAndrii Nakryiko 		case BTF_KIND_UNION:
1457351131b5SAndrii Nakryiko 			btf_dump_emit_mods(d, decls);
1458351131b5SAndrii Nakryiko 			/* inline anonymous struct/union */
1459920d16afSAlan Maguire 			if (t->name_off == 0 && !d->skip_anon_defs)
1460351131b5SAndrii Nakryiko 				btf_dump_emit_struct_def(d, id, t, lvl);
1461351131b5SAndrii Nakryiko 			else
1462351131b5SAndrii Nakryiko 				btf_dump_emit_struct_fwd(d, id, t);
1463351131b5SAndrii Nakryiko 			break;
1464351131b5SAndrii Nakryiko 		case BTF_KIND_ENUM:
1465d90ec262SYonghong Song 		case BTF_KIND_ENUM64:
1466351131b5SAndrii Nakryiko 			btf_dump_emit_mods(d, decls);
1467351131b5SAndrii Nakryiko 			/* inline anonymous enum */
1468920d16afSAlan Maguire 			if (t->name_off == 0 && !d->skip_anon_defs)
1469351131b5SAndrii Nakryiko 				btf_dump_emit_enum_def(d, id, t, lvl);
1470351131b5SAndrii Nakryiko 			else
1471351131b5SAndrii Nakryiko 				btf_dump_emit_enum_fwd(d, id, t);
1472351131b5SAndrii Nakryiko 			break;
1473351131b5SAndrii Nakryiko 		case BTF_KIND_FWD:
1474351131b5SAndrii Nakryiko 			btf_dump_emit_mods(d, decls);
1475351131b5SAndrii Nakryiko 			btf_dump_emit_fwd_def(d, id, t);
1476351131b5SAndrii Nakryiko 			break;
1477351131b5SAndrii Nakryiko 		case BTF_KIND_TYPEDEF:
1478351131b5SAndrii Nakryiko 			btf_dump_emit_mods(d, decls);
1479351131b5SAndrii Nakryiko 			btf_dump_printf(d, "%s", btf_dump_ident_name(d, id));
1480351131b5SAndrii Nakryiko 			break;
1481351131b5SAndrii Nakryiko 		case BTF_KIND_PTR:
1482351131b5SAndrii Nakryiko 			btf_dump_printf(d, "%s", last_was_ptr ? "*" : " *");
1483351131b5SAndrii Nakryiko 			break;
1484351131b5SAndrii Nakryiko 		case BTF_KIND_VOLATILE:
1485351131b5SAndrii Nakryiko 			btf_dump_printf(d, " volatile");
1486351131b5SAndrii Nakryiko 			break;
1487351131b5SAndrii Nakryiko 		case BTF_KIND_CONST:
1488351131b5SAndrii Nakryiko 			btf_dump_printf(d, " const");
1489351131b5SAndrii Nakryiko 			break;
1490351131b5SAndrii Nakryiko 		case BTF_KIND_RESTRICT:
1491351131b5SAndrii Nakryiko 			btf_dump_printf(d, " restrict");
1492351131b5SAndrii Nakryiko 			break;
14932dc1e488SYonghong Song 		case BTF_KIND_TYPE_TAG:
14942dc1e488SYonghong Song 			btf_dump_emit_mods(d, decls);
14952dc1e488SYonghong Song 			name = btf_name_of(d, t->name_off);
14962dc1e488SYonghong Song 			btf_dump_printf(d, " __attribute__((btf_type_tag(\"%s\")))", name);
14972dc1e488SYonghong Song 			break;
1498351131b5SAndrii Nakryiko 		case BTF_KIND_ARRAY: {
1499b03bc685SAndrii Nakryiko 			const struct btf_array *a = btf_array(t);
1500351131b5SAndrii Nakryiko 			const struct btf_type *next_t;
1501351131b5SAndrii Nakryiko 			__u32 next_id;
1502351131b5SAndrii Nakryiko 			bool multidim;
1503351131b5SAndrii Nakryiko 			/*
1504351131b5SAndrii Nakryiko 			 * GCC has a bug
1505351131b5SAndrii Nakryiko 			 * (https://gcc.gnu.org/bugzilla/show_bug.cgi?id=8354)
1506351131b5SAndrii Nakryiko 			 * which causes it to emit extra const/volatile
1507351131b5SAndrii Nakryiko 			 * modifiers for an array, if array's element type has
1508351131b5SAndrii Nakryiko 			 * const/volatile modifiers. Clang doesn't do that.
1509351131b5SAndrii Nakryiko 			 * In general, it doesn't seem very meaningful to have
1510351131b5SAndrii Nakryiko 			 * a const/volatile modifier for array, so we are
1511351131b5SAndrii Nakryiko 			 * going to silently skip them here.
1512351131b5SAndrii Nakryiko 			 */
151332022fd9SAndrii Nakryiko 			btf_dump_drop_mods(d, decls);
1514351131b5SAndrii Nakryiko 
1515351131b5SAndrii Nakryiko 			if (decls->cnt == 0) {
1516351131b5SAndrii Nakryiko 				btf_dump_emit_name(d, fname, last_was_ptr);
1517351131b5SAndrii Nakryiko 				btf_dump_printf(d, "[%u]", a->nelems);
1518351131b5SAndrii Nakryiko 				return;
1519351131b5SAndrii Nakryiko 			}
1520351131b5SAndrii Nakryiko 
1521aef70a1fSAndrii Nakryiko 			next_id = decls->ids[decls->cnt - 1];
1522351131b5SAndrii Nakryiko 			next_t = btf__type_by_id(d->btf, next_id);
1523b03bc685SAndrii Nakryiko 			multidim = btf_is_array(next_t);
1524351131b5SAndrii Nakryiko 			/* we need space if we have named non-pointer */
1525351131b5SAndrii Nakryiko 			if (fname[0] && !last_was_ptr)
1526351131b5SAndrii Nakryiko 				btf_dump_printf(d, " ");
1527351131b5SAndrii Nakryiko 			/* no parentheses for multi-dimensional array */
1528351131b5SAndrii Nakryiko 			if (!multidim)
1529351131b5SAndrii Nakryiko 				btf_dump_printf(d, "(");
1530351131b5SAndrii Nakryiko 			btf_dump_emit_type_chain(d, decls, fname, lvl);
1531351131b5SAndrii Nakryiko 			if (!multidim)
1532351131b5SAndrii Nakryiko 				btf_dump_printf(d, ")");
1533351131b5SAndrii Nakryiko 			btf_dump_printf(d, "[%u]", a->nelems);
1534351131b5SAndrii Nakryiko 			return;
1535351131b5SAndrii Nakryiko 		}
1536351131b5SAndrii Nakryiko 		case BTF_KIND_FUNC_PROTO: {
1537b03bc685SAndrii Nakryiko 			const struct btf_param *p = btf_params(t);
1538b03bc685SAndrii Nakryiko 			__u16 vlen = btf_vlen(t);
1539351131b5SAndrii Nakryiko 			int i;
1540351131b5SAndrii Nakryiko 
154132022fd9SAndrii Nakryiko 			/*
154232022fd9SAndrii Nakryiko 			 * GCC emits extra volatile qualifier for
154332022fd9SAndrii Nakryiko 			 * __attribute__((noreturn)) function pointers. Clang
154432022fd9SAndrii Nakryiko 			 * doesn't do it. It's a GCC quirk for backwards
154532022fd9SAndrii Nakryiko 			 * compatibility with code written for GCC <2.5. So,
154632022fd9SAndrii Nakryiko 			 * similarly to extra qualifiers for array, just drop
154732022fd9SAndrii Nakryiko 			 * them, instead of handling them.
154832022fd9SAndrii Nakryiko 			 */
154932022fd9SAndrii Nakryiko 			btf_dump_drop_mods(d, decls);
1550351131b5SAndrii Nakryiko 			if (decls->cnt) {
1551351131b5SAndrii Nakryiko 				btf_dump_printf(d, " (");
1552351131b5SAndrii Nakryiko 				btf_dump_emit_type_chain(d, decls, fname, lvl);
1553351131b5SAndrii Nakryiko 				btf_dump_printf(d, ")");
1554351131b5SAndrii Nakryiko 			} else {
1555351131b5SAndrii Nakryiko 				btf_dump_emit_name(d, fname, last_was_ptr);
1556351131b5SAndrii Nakryiko 			}
1557351131b5SAndrii Nakryiko 			btf_dump_printf(d, "(");
1558351131b5SAndrii Nakryiko 			/*
1559351131b5SAndrii Nakryiko 			 * Clang for BPF target generates func_proto with no
1560351131b5SAndrii Nakryiko 			 * args as a func_proto with a single void arg (e.g.,
1561351131b5SAndrii Nakryiko 			 * `int (*f)(void)` vs just `int (*f)()`). We are
1562351131b5SAndrii Nakryiko 			 * going to pretend there are no args for such case.
1563351131b5SAndrii Nakryiko 			 */
1564351131b5SAndrii Nakryiko 			if (vlen == 1 && p->type == 0) {
1565351131b5SAndrii Nakryiko 				btf_dump_printf(d, ")");
1566351131b5SAndrii Nakryiko 				return;
1567351131b5SAndrii Nakryiko 			}
1568351131b5SAndrii Nakryiko 
1569351131b5SAndrii Nakryiko 			for (i = 0; i < vlen; i++, p++) {
1570351131b5SAndrii Nakryiko 				if (i > 0)
1571351131b5SAndrii Nakryiko 					btf_dump_printf(d, ", ");
1572351131b5SAndrii Nakryiko 
1573351131b5SAndrii Nakryiko 				/* last arg of type void is vararg */
1574351131b5SAndrii Nakryiko 				if (i == vlen - 1 && p->type == 0) {
1575351131b5SAndrii Nakryiko 					btf_dump_printf(d, "...");
1576351131b5SAndrii Nakryiko 					break;
1577351131b5SAndrii Nakryiko 				}
1578351131b5SAndrii Nakryiko 
1579351131b5SAndrii Nakryiko 				name = btf_name_of(d, p->name_off);
1580351131b5SAndrii Nakryiko 				btf_dump_emit_type_decl(d, p->type, name, lvl);
1581351131b5SAndrii Nakryiko 			}
1582351131b5SAndrii Nakryiko 
1583351131b5SAndrii Nakryiko 			btf_dump_printf(d, ")");
1584351131b5SAndrii Nakryiko 			return;
1585351131b5SAndrii Nakryiko 		}
1586351131b5SAndrii Nakryiko 		default:
1587be18010eSKefeng Wang 			pr_warn("unexpected type in decl chain, kind:%u, id:[%u]\n",
1588351131b5SAndrii Nakryiko 				kind, id);
1589351131b5SAndrii Nakryiko 			return;
1590351131b5SAndrii Nakryiko 		}
1591351131b5SAndrii Nakryiko 
1592351131b5SAndrii Nakryiko 		last_was_ptr = kind == BTF_KIND_PTR;
1593351131b5SAndrii Nakryiko 	}
1594351131b5SAndrii Nakryiko 
1595351131b5SAndrii Nakryiko 	btf_dump_emit_name(d, fname, last_was_ptr);
1596351131b5SAndrii Nakryiko }
1597351131b5SAndrii Nakryiko 
1598920d16afSAlan Maguire /* show type name as (type_name) */
btf_dump_emit_type_cast(struct btf_dump * d,__u32 id,bool top_level)1599920d16afSAlan Maguire static void btf_dump_emit_type_cast(struct btf_dump *d, __u32 id,
1600920d16afSAlan Maguire 				    bool top_level)
1601920d16afSAlan Maguire {
1602920d16afSAlan Maguire 	const struct btf_type *t;
1603920d16afSAlan Maguire 
1604920d16afSAlan Maguire 	/* for array members, we don't bother emitting type name for each
1605920d16afSAlan Maguire 	 * member to avoid the redundancy of
1606920d16afSAlan Maguire 	 * .name = (char[4])[(char)'f',(char)'o',(char)'o',]
1607920d16afSAlan Maguire 	 */
1608920d16afSAlan Maguire 	if (d->typed_dump->is_array_member)
1609920d16afSAlan Maguire 		return;
1610920d16afSAlan Maguire 
1611920d16afSAlan Maguire 	/* avoid type name specification for variable/section; it will be done
1612920d16afSAlan Maguire 	 * for the associated variable value(s).
1613920d16afSAlan Maguire 	 */
1614920d16afSAlan Maguire 	t = btf__type_by_id(d->btf, id);
1615920d16afSAlan Maguire 	if (btf_is_var(t) || btf_is_datasec(t))
1616920d16afSAlan Maguire 		return;
1617920d16afSAlan Maguire 
1618920d16afSAlan Maguire 	if (top_level)
1619920d16afSAlan Maguire 		btf_dump_printf(d, "(");
1620920d16afSAlan Maguire 
1621920d16afSAlan Maguire 	d->skip_anon_defs = true;
1622920d16afSAlan Maguire 	d->strip_mods = true;
1623920d16afSAlan Maguire 	btf_dump_emit_type_decl(d, id, "", 0);
1624920d16afSAlan Maguire 	d->strip_mods = false;
1625920d16afSAlan Maguire 	d->skip_anon_defs = false;
1626920d16afSAlan Maguire 
1627920d16afSAlan Maguire 	if (top_level)
1628920d16afSAlan Maguire 		btf_dump_printf(d, ")");
1629920d16afSAlan Maguire }
1630920d16afSAlan Maguire 
1631351131b5SAndrii Nakryiko /* return number of duplicates (occurrences) of a given name */
btf_dump_name_dups(struct btf_dump * d,struct hashmap * name_map,const char * orig_name)1632351131b5SAndrii Nakryiko static size_t btf_dump_name_dups(struct btf_dump *d, struct hashmap *name_map,
1633351131b5SAndrii Nakryiko 				 const char *orig_name)
1634351131b5SAndrii Nakryiko {
163593c660caSXu Kuohai 	char *old_name, *new_name;
1636351131b5SAndrii Nakryiko 	size_t dup_cnt = 0;
163793c660caSXu Kuohai 	int err;
163893c660caSXu Kuohai 
163993c660caSXu Kuohai 	new_name = strdup(orig_name);
164093c660caSXu Kuohai 	if (!new_name)
164193c660caSXu Kuohai 		return 1;
1642351131b5SAndrii Nakryiko 
1643f80e16b6SAndrii Nakryiko 	(void)hashmap__find(name_map, orig_name, &dup_cnt);
1644351131b5SAndrii Nakryiko 	dup_cnt++;
164593c660caSXu Kuohai 
1646c302378bSEduard Zingerman 	err = hashmap__set(name_map, new_name, dup_cnt, &old_name, NULL);
164793c660caSXu Kuohai 	if (err)
164893c660caSXu Kuohai 		free(new_name);
164993c660caSXu Kuohai 
165093c660caSXu Kuohai 	free(old_name);
1651351131b5SAndrii Nakryiko 
1652351131b5SAndrii Nakryiko 	return dup_cnt;
1653351131b5SAndrii Nakryiko }
1654351131b5SAndrii Nakryiko 
btf_dump_resolve_name(struct btf_dump * d,__u32 id,struct hashmap * name_map)1655351131b5SAndrii Nakryiko static const char *btf_dump_resolve_name(struct btf_dump *d, __u32 id,
1656351131b5SAndrii Nakryiko 					 struct hashmap *name_map)
1657351131b5SAndrii Nakryiko {
1658351131b5SAndrii Nakryiko 	struct btf_dump_type_aux_state *s = &d->type_states[id];
1659351131b5SAndrii Nakryiko 	const struct btf_type *t = btf__type_by_id(d->btf, id);
1660351131b5SAndrii Nakryiko 	const char *orig_name = btf_name_of(d, t->name_off);
1661351131b5SAndrii Nakryiko 	const char **cached_name = &d->cached_names[id];
1662351131b5SAndrii Nakryiko 	size_t dup_cnt;
1663351131b5SAndrii Nakryiko 
1664351131b5SAndrii Nakryiko 	if (t->name_off == 0)
1665351131b5SAndrii Nakryiko 		return "";
1666351131b5SAndrii Nakryiko 
1667351131b5SAndrii Nakryiko 	if (s->name_resolved)
1668351131b5SAndrii Nakryiko 		return *cached_name ? *cached_name : orig_name;
1669351131b5SAndrii Nakryiko 
16704226961bSXu Kuohai 	if (btf_is_fwd(t) || (btf_is_enum(t) && btf_vlen(t) == 0)) {
16714226961bSXu Kuohai 		s->name_resolved = 1;
16724226961bSXu Kuohai 		return orig_name;
16734226961bSXu Kuohai 	}
16744226961bSXu Kuohai 
1675351131b5SAndrii Nakryiko 	dup_cnt = btf_dump_name_dups(d, name_map, orig_name);
1676351131b5SAndrii Nakryiko 	if (dup_cnt > 1) {
1677351131b5SAndrii Nakryiko 		const size_t max_len = 256;
1678351131b5SAndrii Nakryiko 		char new_name[max_len];
1679351131b5SAndrii Nakryiko 
1680351131b5SAndrii Nakryiko 		snprintf(new_name, max_len, "%s___%zu", orig_name, dup_cnt);
1681351131b5SAndrii Nakryiko 		*cached_name = strdup(new_name);
1682351131b5SAndrii Nakryiko 	}
1683351131b5SAndrii Nakryiko 
1684351131b5SAndrii Nakryiko 	s->name_resolved = 1;
1685351131b5SAndrii Nakryiko 	return *cached_name ? *cached_name : orig_name;
1686351131b5SAndrii Nakryiko }
1687351131b5SAndrii Nakryiko 
btf_dump_type_name(struct btf_dump * d,__u32 id)1688351131b5SAndrii Nakryiko static const char *btf_dump_type_name(struct btf_dump *d, __u32 id)
1689351131b5SAndrii Nakryiko {
1690351131b5SAndrii Nakryiko 	return btf_dump_resolve_name(d, id, d->type_names);
1691351131b5SAndrii Nakryiko }
1692351131b5SAndrii Nakryiko 
btf_dump_ident_name(struct btf_dump * d,__u32 id)1693351131b5SAndrii Nakryiko static const char *btf_dump_ident_name(struct btf_dump *d, __u32 id)
1694351131b5SAndrii Nakryiko {
1695351131b5SAndrii Nakryiko 	return btf_dump_resolve_name(d, id, d->ident_names);
1696351131b5SAndrii Nakryiko }
1697920d16afSAlan Maguire 
1698920d16afSAlan Maguire static int btf_dump_dump_type_data(struct btf_dump *d,
1699920d16afSAlan Maguire 				   const char *fname,
1700920d16afSAlan Maguire 				   const struct btf_type *t,
1701920d16afSAlan Maguire 				   __u32 id,
1702920d16afSAlan Maguire 				   const void *data,
1703920d16afSAlan Maguire 				   __u8 bits_offset,
1704920d16afSAlan Maguire 				   __u8 bit_sz);
1705920d16afSAlan Maguire 
btf_dump_data_newline(struct btf_dump * d)1706920d16afSAlan Maguire static const char *btf_dump_data_newline(struct btf_dump *d)
1707920d16afSAlan Maguire {
1708920d16afSAlan Maguire 	return d->typed_dump->compact || d->typed_dump->depth == 0 ? "" : "\n";
1709920d16afSAlan Maguire }
1710920d16afSAlan Maguire 
btf_dump_data_delim(struct btf_dump * d)1711920d16afSAlan Maguire static const char *btf_dump_data_delim(struct btf_dump *d)
1712920d16afSAlan Maguire {
1713920d16afSAlan Maguire 	return d->typed_dump->depth == 0 ? "" : ",";
1714920d16afSAlan Maguire }
1715920d16afSAlan Maguire 
btf_dump_data_pfx(struct btf_dump * d)1716920d16afSAlan Maguire static void btf_dump_data_pfx(struct btf_dump *d)
1717920d16afSAlan Maguire {
1718920d16afSAlan Maguire 	int i, lvl = d->typed_dump->indent_lvl + d->typed_dump->depth;
1719920d16afSAlan Maguire 
1720920d16afSAlan Maguire 	if (d->typed_dump->compact)
1721920d16afSAlan Maguire 		return;
1722920d16afSAlan Maguire 
1723920d16afSAlan Maguire 	for (i = 0; i < lvl; i++)
1724920d16afSAlan Maguire 		btf_dump_printf(d, "%s", d->typed_dump->indent_str);
1725920d16afSAlan Maguire }
1726920d16afSAlan Maguire 
1727920d16afSAlan Maguire /* A macro is used here as btf_type_value[s]() appends format specifiers
1728920d16afSAlan Maguire  * to the format specifier passed in; these do the work of appending
1729920d16afSAlan Maguire  * delimiters etc while the caller simply has to specify the type values
1730920d16afSAlan Maguire  * in the format specifier + value(s).
1731920d16afSAlan Maguire  */
1732920d16afSAlan Maguire #define btf_dump_type_values(d, fmt, ...)				\
1733920d16afSAlan Maguire 	btf_dump_printf(d, fmt "%s%s",					\
1734920d16afSAlan Maguire 			##__VA_ARGS__,					\
1735920d16afSAlan Maguire 			btf_dump_data_delim(d),				\
1736920d16afSAlan Maguire 			btf_dump_data_newline(d))
1737920d16afSAlan Maguire 
btf_dump_unsupported_data(struct btf_dump * d,const struct btf_type * t,__u32 id)1738920d16afSAlan Maguire static int btf_dump_unsupported_data(struct btf_dump *d,
1739920d16afSAlan Maguire 				     const struct btf_type *t,
1740920d16afSAlan Maguire 				     __u32 id)
1741920d16afSAlan Maguire {
1742920d16afSAlan Maguire 	btf_dump_printf(d, "<unsupported kind:%u>", btf_kind(t));
1743920d16afSAlan Maguire 	return -ENOTSUP;
1744920d16afSAlan Maguire }
1745920d16afSAlan Maguire 
btf_dump_get_bitfield_value(struct btf_dump * d,const struct btf_type * t,const void * data,__u8 bits_offset,__u8 bit_sz,__u64 * value)1746a1d3cc3cSAlan Maguire static int btf_dump_get_bitfield_value(struct btf_dump *d,
1747920d16afSAlan Maguire 				       const struct btf_type *t,
1748920d16afSAlan Maguire 				       const void *data,
1749920d16afSAlan Maguire 				       __u8 bits_offset,
1750a1d3cc3cSAlan Maguire 				       __u8 bit_sz,
1751a1d3cc3cSAlan Maguire 				       __u64 *value)
1752920d16afSAlan Maguire {
1753920d16afSAlan Maguire 	__u16 left_shift_bits, right_shift_bits;
1754920d16afSAlan Maguire 	const __u8 *bytes = data;
1755c9e982b8SIlya Leoshkevich 	__u8 nr_copy_bits;
1756a1d3cc3cSAlan Maguire 	__u64 num = 0;
1757920d16afSAlan Maguire 	int i;
1758920d16afSAlan Maguire 
1759a1d3cc3cSAlan Maguire 	/* Maximum supported bitfield size is 64 bits */
1760c9e982b8SIlya Leoshkevich 	if (t->size > 8) {
1761c9e982b8SIlya Leoshkevich 		pr_warn("unexpected bitfield size %d\n", t->size);
1762a1d3cc3cSAlan Maguire 		return -EINVAL;
1763a1d3cc3cSAlan Maguire 	}
1764a1d3cc3cSAlan Maguire 
1765920d16afSAlan Maguire 	/* Bitfield value retrieval is done in two steps; first relevant bytes are
1766920d16afSAlan Maguire 	 * stored in num, then we left/right shift num to eliminate irrelevant bits.
1767920d16afSAlan Maguire 	 */
17683930198dSIlya Leoshkevich #if __BYTE_ORDER__ == __ORDER_LITTLE_ENDIAN__
1769c9e982b8SIlya Leoshkevich 	for (i = t->size - 1; i >= 0; i--)
1770920d16afSAlan Maguire 		num = num * 256 + bytes[i];
1771c9e982b8SIlya Leoshkevich 	nr_copy_bits = bit_sz + bits_offset;
17723930198dSIlya Leoshkevich #elif __BYTE_ORDER__ == __ORDER_BIG_ENDIAN__
1773c9e982b8SIlya Leoshkevich 	for (i = 0; i < t->size; i++)
1774920d16afSAlan Maguire 		num = num * 256 + bytes[i];
1775c9e982b8SIlya Leoshkevich 	nr_copy_bits = t->size * 8 - bits_offset;
1776920d16afSAlan Maguire #else
1777920d16afSAlan Maguire # error "Unrecognized __BYTE_ORDER__"
1778920d16afSAlan Maguire #endif
1779a1d3cc3cSAlan Maguire 	left_shift_bits = 64 - nr_copy_bits;
1780a1d3cc3cSAlan Maguire 	right_shift_bits = 64 - bit_sz;
1781920d16afSAlan Maguire 
1782a1d3cc3cSAlan Maguire 	*value = (num << left_shift_bits) >> right_shift_bits;
1783920d16afSAlan Maguire 
1784a1d3cc3cSAlan Maguire 	return 0;
1785920d16afSAlan Maguire }
1786920d16afSAlan Maguire 
btf_dump_bitfield_check_zero(struct btf_dump * d,const struct btf_type * t,const void * data,__u8 bits_offset,__u8 bit_sz)1787920d16afSAlan Maguire static int btf_dump_bitfield_check_zero(struct btf_dump *d,
1788920d16afSAlan Maguire 					const struct btf_type *t,
1789920d16afSAlan Maguire 					const void *data,
1790920d16afSAlan Maguire 					__u8 bits_offset,
1791920d16afSAlan Maguire 					__u8 bit_sz)
1792920d16afSAlan Maguire {
1793a1d3cc3cSAlan Maguire 	__u64 check_num;
1794a1d3cc3cSAlan Maguire 	int err;
1795920d16afSAlan Maguire 
1796a1d3cc3cSAlan Maguire 	err = btf_dump_get_bitfield_value(d, t, data, bits_offset, bit_sz, &check_num);
1797a1d3cc3cSAlan Maguire 	if (err)
1798a1d3cc3cSAlan Maguire 		return err;
1799920d16afSAlan Maguire 	if (check_num == 0)
1800920d16afSAlan Maguire 		return -ENODATA;
1801920d16afSAlan Maguire 	return 0;
1802920d16afSAlan Maguire }
1803920d16afSAlan Maguire 
btf_dump_bitfield_data(struct btf_dump * d,const struct btf_type * t,const void * data,__u8 bits_offset,__u8 bit_sz)1804920d16afSAlan Maguire static int btf_dump_bitfield_data(struct btf_dump *d,
1805920d16afSAlan Maguire 				  const struct btf_type *t,
1806920d16afSAlan Maguire 				  const void *data,
1807920d16afSAlan Maguire 				  __u8 bits_offset,
1808920d16afSAlan Maguire 				  __u8 bit_sz)
1809920d16afSAlan Maguire {
1810a1d3cc3cSAlan Maguire 	__u64 print_num;
1811a1d3cc3cSAlan Maguire 	int err;
1812920d16afSAlan Maguire 
1813a1d3cc3cSAlan Maguire 	err = btf_dump_get_bitfield_value(d, t, data, bits_offset, bit_sz, &print_num);
1814a1d3cc3cSAlan Maguire 	if (err)
1815a1d3cc3cSAlan Maguire 		return err;
1816a1d3cc3cSAlan Maguire 
1817a1d3cc3cSAlan Maguire 	btf_dump_type_values(d, "0x%llx", (unsigned long long)print_num);
1818920d16afSAlan Maguire 
1819920d16afSAlan Maguire 	return 0;
1820920d16afSAlan Maguire }
1821920d16afSAlan Maguire 
1822920d16afSAlan Maguire /* ints, floats and ptrs */
btf_dump_base_type_check_zero(struct btf_dump * d,const struct btf_type * t,__u32 id,const void * data)1823920d16afSAlan Maguire static int btf_dump_base_type_check_zero(struct btf_dump *d,
1824920d16afSAlan Maguire 					 const struct btf_type *t,
1825920d16afSAlan Maguire 					 __u32 id,
1826920d16afSAlan Maguire 					 const void *data)
1827920d16afSAlan Maguire {
1828920d16afSAlan Maguire 	static __u8 bytecmp[16] = {};
1829920d16afSAlan Maguire 	int nr_bytes;
1830920d16afSAlan Maguire 
1831920d16afSAlan Maguire 	/* For pointer types, pointer size is not defined on a per-type basis.
1832920d16afSAlan Maguire 	 * On dump creation however, we store the pointer size.
1833920d16afSAlan Maguire 	 */
1834920d16afSAlan Maguire 	if (btf_kind(t) == BTF_KIND_PTR)
1835920d16afSAlan Maguire 		nr_bytes = d->ptr_sz;
1836920d16afSAlan Maguire 	else
1837920d16afSAlan Maguire 		nr_bytes = t->size;
1838920d16afSAlan Maguire 
1839920d16afSAlan Maguire 	if (nr_bytes < 1 || nr_bytes > 16) {
1840920d16afSAlan Maguire 		pr_warn("unexpected size %d for id [%u]\n", nr_bytes, id);
1841920d16afSAlan Maguire 		return -EINVAL;
1842920d16afSAlan Maguire 	}
1843920d16afSAlan Maguire 
1844920d16afSAlan Maguire 	if (memcmp(data, bytecmp, nr_bytes) == 0)
1845920d16afSAlan Maguire 		return -ENODATA;
1846920d16afSAlan Maguire 	return 0;
1847920d16afSAlan Maguire }
1848920d16afSAlan Maguire 
ptr_is_aligned(const struct btf * btf,__u32 type_id,const void * data)1849632f96d2SIlya Leoshkevich static bool ptr_is_aligned(const struct btf *btf, __u32 type_id,
1850632f96d2SIlya Leoshkevich 			   const void *data)
18518d44c357SAlan Maguire {
1852632f96d2SIlya Leoshkevich 	int alignment = btf__align_of(btf, type_id);
1853632f96d2SIlya Leoshkevich 
1854632f96d2SIlya Leoshkevich 	if (alignment == 0)
1855632f96d2SIlya Leoshkevich 		return false;
1856632f96d2SIlya Leoshkevich 
1857632f96d2SIlya Leoshkevich 	return ((uintptr_t)data) % alignment == 0;
18588d44c357SAlan Maguire }
18598d44c357SAlan Maguire 
btf_dump_int_data(struct btf_dump * d,const struct btf_type * t,__u32 type_id,const void * data,__u8 bits_offset)1860920d16afSAlan Maguire static int btf_dump_int_data(struct btf_dump *d,
1861920d16afSAlan Maguire 			     const struct btf_type *t,
1862920d16afSAlan Maguire 			     __u32 type_id,
1863920d16afSAlan Maguire 			     const void *data,
1864920d16afSAlan Maguire 			     __u8 bits_offset)
1865920d16afSAlan Maguire {
1866920d16afSAlan Maguire 	__u8 encoding = btf_int_encoding(t);
1867920d16afSAlan Maguire 	bool sign = encoding & BTF_INT_SIGNED;
1868fae1b05eSAndrii Nakryiko 	char buf[16] __attribute__((aligned(16)));
1869920d16afSAlan Maguire 	int sz = t->size;
1870920d16afSAlan Maguire 
1871961632d5SIlya Leoshkevich 	if (sz == 0 || sz > sizeof(buf)) {
1872920d16afSAlan Maguire 		pr_warn("unexpected size %d for id [%u]\n", sz, type_id);
1873920d16afSAlan Maguire 		return -EINVAL;
1874920d16afSAlan Maguire 	}
1875920d16afSAlan Maguire 
1876920d16afSAlan Maguire 	/* handle packed int data - accesses of integers not aligned on
1877920d16afSAlan Maguire 	 * int boundaries can cause problems on some platforms.
1878920d16afSAlan Maguire 	 */
1879632f96d2SIlya Leoshkevich 	if (!ptr_is_aligned(d->btf, type_id, data)) {
1880961632d5SIlya Leoshkevich 		memcpy(buf, data, sz);
1881961632d5SIlya Leoshkevich 		data = buf;
1882961632d5SIlya Leoshkevich 	}
1883920d16afSAlan Maguire 
1884920d16afSAlan Maguire 	switch (sz) {
1885a1d3cc3cSAlan Maguire 	case 16: {
1886a1d3cc3cSAlan Maguire 		const __u64 *ints = data;
1887a1d3cc3cSAlan Maguire 		__u64 lsi, msi;
1888a1d3cc3cSAlan Maguire 
1889a1d3cc3cSAlan Maguire 		/* avoid use of __int128 as some 32-bit platforms do not
1890a1d3cc3cSAlan Maguire 		 * support it.
1891a1d3cc3cSAlan Maguire 		 */
18923930198dSIlya Leoshkevich #if __BYTE_ORDER__ == __ORDER_LITTLE_ENDIAN__
1893a1d3cc3cSAlan Maguire 		lsi = ints[0];
1894a1d3cc3cSAlan Maguire 		msi = ints[1];
18953930198dSIlya Leoshkevich #elif __BYTE_ORDER__ == __ORDER_BIG_ENDIAN__
1896a1d3cc3cSAlan Maguire 		lsi = ints[1];
1897a1d3cc3cSAlan Maguire 		msi = ints[0];
1898a1d3cc3cSAlan Maguire #else
1899a1d3cc3cSAlan Maguire # error "Unrecognized __BYTE_ORDER__"
1900a1d3cc3cSAlan Maguire #endif
1901a1d3cc3cSAlan Maguire 		if (msi == 0)
1902a1d3cc3cSAlan Maguire 			btf_dump_type_values(d, "0x%llx", (unsigned long long)lsi);
1903a1d3cc3cSAlan Maguire 		else
1904a1d3cc3cSAlan Maguire 			btf_dump_type_values(d, "0x%llx%016llx", (unsigned long long)msi,
1905a1d3cc3cSAlan Maguire 					     (unsigned long long)lsi);
1906920d16afSAlan Maguire 		break;
1907a1d3cc3cSAlan Maguire 	}
1908920d16afSAlan Maguire 	case 8:
1909920d16afSAlan Maguire 		if (sign)
1910920d16afSAlan Maguire 			btf_dump_type_values(d, "%lld", *(long long *)data);
1911920d16afSAlan Maguire 		else
1912920d16afSAlan Maguire 			btf_dump_type_values(d, "%llu", *(unsigned long long *)data);
1913920d16afSAlan Maguire 		break;
1914920d16afSAlan Maguire 	case 4:
1915920d16afSAlan Maguire 		if (sign)
1916920d16afSAlan Maguire 			btf_dump_type_values(d, "%d", *(__s32 *)data);
1917920d16afSAlan Maguire 		else
1918920d16afSAlan Maguire 			btf_dump_type_values(d, "%u", *(__u32 *)data);
1919920d16afSAlan Maguire 		break;
1920920d16afSAlan Maguire 	case 2:
1921920d16afSAlan Maguire 		if (sign)
1922920d16afSAlan Maguire 			btf_dump_type_values(d, "%d", *(__s16 *)data);
1923920d16afSAlan Maguire 		else
1924920d16afSAlan Maguire 			btf_dump_type_values(d, "%u", *(__u16 *)data);
1925920d16afSAlan Maguire 		break;
1926920d16afSAlan Maguire 	case 1:
1927920d16afSAlan Maguire 		if (d->typed_dump->is_array_char) {
1928920d16afSAlan Maguire 			/* check for null terminator */
1929920d16afSAlan Maguire 			if (d->typed_dump->is_array_terminated)
1930920d16afSAlan Maguire 				break;
1931920d16afSAlan Maguire 			if (*(char *)data == '\0') {
1932920d16afSAlan Maguire 				d->typed_dump->is_array_terminated = true;
1933920d16afSAlan Maguire 				break;
1934920d16afSAlan Maguire 			}
1935920d16afSAlan Maguire 			if (isprint(*(char *)data)) {
1936920d16afSAlan Maguire 				btf_dump_type_values(d, "'%c'", *(char *)data);
1937920d16afSAlan Maguire 				break;
1938920d16afSAlan Maguire 			}
1939920d16afSAlan Maguire 		}
1940920d16afSAlan Maguire 		if (sign)
1941920d16afSAlan Maguire 			btf_dump_type_values(d, "%d", *(__s8 *)data);
1942920d16afSAlan Maguire 		else
1943920d16afSAlan Maguire 			btf_dump_type_values(d, "%u", *(__u8 *)data);
1944920d16afSAlan Maguire 		break;
1945920d16afSAlan Maguire 	default:
1946920d16afSAlan Maguire 		pr_warn("unexpected sz %d for id [%u]\n", sz, type_id);
1947920d16afSAlan Maguire 		return -EINVAL;
1948920d16afSAlan Maguire 	}
1949920d16afSAlan Maguire 	return 0;
1950920d16afSAlan Maguire }
1951920d16afSAlan Maguire 
1952920d16afSAlan Maguire union float_data {
1953920d16afSAlan Maguire 	long double ld;
1954920d16afSAlan Maguire 	double d;
1955920d16afSAlan Maguire 	float f;
1956920d16afSAlan Maguire };
1957920d16afSAlan Maguire 
btf_dump_float_data(struct btf_dump * d,const struct btf_type * t,__u32 type_id,const void * data)1958920d16afSAlan Maguire static int btf_dump_float_data(struct btf_dump *d,
1959920d16afSAlan Maguire 			       const struct btf_type *t,
1960920d16afSAlan Maguire 			       __u32 type_id,
1961920d16afSAlan Maguire 			       const void *data)
1962920d16afSAlan Maguire {
1963920d16afSAlan Maguire 	const union float_data *flp = data;
1964920d16afSAlan Maguire 	union float_data fl;
1965920d16afSAlan Maguire 	int sz = t->size;
1966920d16afSAlan Maguire 
1967920d16afSAlan Maguire 	/* handle unaligned data; copy to local union */
1968632f96d2SIlya Leoshkevich 	if (!ptr_is_aligned(d->btf, type_id, data)) {
1969920d16afSAlan Maguire 		memcpy(&fl, data, sz);
1970920d16afSAlan Maguire 		flp = &fl;
1971920d16afSAlan Maguire 	}
1972920d16afSAlan Maguire 
1973920d16afSAlan Maguire 	switch (sz) {
1974920d16afSAlan Maguire 	case 16:
1975920d16afSAlan Maguire 		btf_dump_type_values(d, "%Lf", flp->ld);
1976920d16afSAlan Maguire 		break;
1977920d16afSAlan Maguire 	case 8:
1978920d16afSAlan Maguire 		btf_dump_type_values(d, "%lf", flp->d);
1979920d16afSAlan Maguire 		break;
1980920d16afSAlan Maguire 	case 4:
1981920d16afSAlan Maguire 		btf_dump_type_values(d, "%f", flp->f);
1982920d16afSAlan Maguire 		break;
1983920d16afSAlan Maguire 	default:
1984920d16afSAlan Maguire 		pr_warn("unexpected size %d for id [%u]\n", sz, type_id);
1985920d16afSAlan Maguire 		return -EINVAL;
1986920d16afSAlan Maguire 	}
1987920d16afSAlan Maguire 	return 0;
1988920d16afSAlan Maguire }
1989920d16afSAlan Maguire 
btf_dump_var_data(struct btf_dump * d,const struct btf_type * v,__u32 id,const void * data)1990920d16afSAlan Maguire static int btf_dump_var_data(struct btf_dump *d,
1991920d16afSAlan Maguire 			     const struct btf_type *v,
1992920d16afSAlan Maguire 			     __u32 id,
1993920d16afSAlan Maguire 			     const void *data)
1994920d16afSAlan Maguire {
1995920d16afSAlan Maguire 	enum btf_func_linkage linkage = btf_var(v)->linkage;
1996920d16afSAlan Maguire 	const struct btf_type *t;
1997920d16afSAlan Maguire 	const char *l;
1998920d16afSAlan Maguire 	__u32 type_id;
1999920d16afSAlan Maguire 
2000920d16afSAlan Maguire 	switch (linkage) {
2001920d16afSAlan Maguire 	case BTF_FUNC_STATIC:
2002920d16afSAlan Maguire 		l = "static ";
2003920d16afSAlan Maguire 		break;
2004920d16afSAlan Maguire 	case BTF_FUNC_EXTERN:
2005920d16afSAlan Maguire 		l = "extern ";
2006920d16afSAlan Maguire 		break;
2007920d16afSAlan Maguire 	case BTF_FUNC_GLOBAL:
2008920d16afSAlan Maguire 	default:
2009920d16afSAlan Maguire 		l = "";
2010920d16afSAlan Maguire 		break;
2011920d16afSAlan Maguire 	}
2012920d16afSAlan Maguire 
2013920d16afSAlan Maguire 	/* format of output here is [linkage] [type] [varname] = (type)value,
2014920d16afSAlan Maguire 	 * for example "static int cpu_profile_flip = (int)1"
2015920d16afSAlan Maguire 	 */
2016920d16afSAlan Maguire 	btf_dump_printf(d, "%s", l);
2017920d16afSAlan Maguire 	type_id = v->type;
2018920d16afSAlan Maguire 	t = btf__type_by_id(d->btf, type_id);
2019920d16afSAlan Maguire 	btf_dump_emit_type_cast(d, type_id, false);
2020920d16afSAlan Maguire 	btf_dump_printf(d, " %s = ", btf_name_of(d, v->name_off));
2021920d16afSAlan Maguire 	return btf_dump_dump_type_data(d, NULL, t, type_id, data, 0, 0);
2022920d16afSAlan Maguire }
2023920d16afSAlan Maguire 
btf_dump_array_data(struct btf_dump * d,const struct btf_type * t,__u32 id,const void * data)2024920d16afSAlan Maguire static int btf_dump_array_data(struct btf_dump *d,
2025920d16afSAlan Maguire 			       const struct btf_type *t,
2026920d16afSAlan Maguire 			       __u32 id,
2027920d16afSAlan Maguire 			       const void *data)
2028920d16afSAlan Maguire {
2029920d16afSAlan Maguire 	const struct btf_array *array = btf_array(t);
2030920d16afSAlan Maguire 	const struct btf_type *elem_type;
20314172843eSDan Carpenter 	__u32 i, elem_type_id;
20324172843eSDan Carpenter 	__s64 elem_size;
2033920d16afSAlan Maguire 	bool is_array_member;
2034920d16afSAlan Maguire 
2035920d16afSAlan Maguire 	elem_type_id = array->type;
2036920d16afSAlan Maguire 	elem_type = skip_mods_and_typedefs(d->btf, elem_type_id, NULL);
2037920d16afSAlan Maguire 	elem_size = btf__resolve_size(d->btf, elem_type_id);
2038920d16afSAlan Maguire 	if (elem_size <= 0) {
2039dc37dc61SAndrii Nakryiko 		pr_warn("unexpected elem size %zd for array type [%u]\n",
2040dc37dc61SAndrii Nakryiko 			(ssize_t)elem_size, id);
2041920d16afSAlan Maguire 		return -EINVAL;
2042920d16afSAlan Maguire 	}
2043920d16afSAlan Maguire 
2044920d16afSAlan Maguire 	if (btf_is_int(elem_type)) {
2045920d16afSAlan Maguire 		/*
2046920d16afSAlan Maguire 		 * BTF_INT_CHAR encoding never seems to be set for
2047920d16afSAlan Maguire 		 * char arrays, so if size is 1 and element is
2048920d16afSAlan Maguire 		 * printable as a char, we'll do that.
2049920d16afSAlan Maguire 		 */
2050920d16afSAlan Maguire 		if (elem_size == 1)
2051920d16afSAlan Maguire 			d->typed_dump->is_array_char = true;
2052920d16afSAlan Maguire 	}
2053920d16afSAlan Maguire 
2054920d16afSAlan Maguire 	/* note that we increment depth before calling btf_dump_print() below;
2055920d16afSAlan Maguire 	 * this is intentional.  btf_dump_data_newline() will not print a
2056920d16afSAlan Maguire 	 * newline for depth 0 (since this leaves us with trailing newlines
2057920d16afSAlan Maguire 	 * at the end of typed display), so depth is incremented first.
2058920d16afSAlan Maguire 	 * For similar reasons, we decrement depth before showing the closing
2059920d16afSAlan Maguire 	 * parenthesis.
2060920d16afSAlan Maguire 	 */
2061920d16afSAlan Maguire 	d->typed_dump->depth++;
2062920d16afSAlan Maguire 	btf_dump_printf(d, "[%s", btf_dump_data_newline(d));
2063920d16afSAlan Maguire 
2064920d16afSAlan Maguire 	/* may be a multidimensional array, so store current "is array member"
2065920d16afSAlan Maguire 	 * status so we can restore it correctly later.
2066920d16afSAlan Maguire 	 */
2067920d16afSAlan Maguire 	is_array_member = d->typed_dump->is_array_member;
2068920d16afSAlan Maguire 	d->typed_dump->is_array_member = true;
2069920d16afSAlan Maguire 	for (i = 0; i < array->nelems; i++, data += elem_size) {
2070920d16afSAlan Maguire 		if (d->typed_dump->is_array_terminated)
2071920d16afSAlan Maguire 			break;
2072920d16afSAlan Maguire 		btf_dump_dump_type_data(d, NULL, elem_type, elem_type_id, data, 0, 0);
2073920d16afSAlan Maguire 	}
2074920d16afSAlan Maguire 	d->typed_dump->is_array_member = is_array_member;
2075920d16afSAlan Maguire 	d->typed_dump->depth--;
2076920d16afSAlan Maguire 	btf_dump_data_pfx(d);
2077920d16afSAlan Maguire 	btf_dump_type_values(d, "]");
2078920d16afSAlan Maguire 
2079920d16afSAlan Maguire 	return 0;
2080920d16afSAlan Maguire }
2081920d16afSAlan Maguire 
btf_dump_struct_data(struct btf_dump * d,const struct btf_type * t,__u32 id,const void * data)2082920d16afSAlan Maguire static int btf_dump_struct_data(struct btf_dump *d,
2083920d16afSAlan Maguire 				const struct btf_type *t,
2084920d16afSAlan Maguire 				__u32 id,
2085920d16afSAlan Maguire 				const void *data)
2086920d16afSAlan Maguire {
2087920d16afSAlan Maguire 	const struct btf_member *m = btf_members(t);
2088920d16afSAlan Maguire 	__u16 n = btf_vlen(t);
2089dfd0afbfSDavid Michael 	int i, err = 0;
2090920d16afSAlan Maguire 
2091920d16afSAlan Maguire 	/* note that we increment depth before calling btf_dump_print() below;
2092920d16afSAlan Maguire 	 * this is intentional.  btf_dump_data_newline() will not print a
2093920d16afSAlan Maguire 	 * newline for depth 0 (since this leaves us with trailing newlines
2094920d16afSAlan Maguire 	 * at the end of typed display), so depth is incremented first.
2095920d16afSAlan Maguire 	 * For similar reasons, we decrement depth before showing the closing
2096920d16afSAlan Maguire 	 * parenthesis.
2097920d16afSAlan Maguire 	 */
2098920d16afSAlan Maguire 	d->typed_dump->depth++;
2099920d16afSAlan Maguire 	btf_dump_printf(d, "{%s", btf_dump_data_newline(d));
2100920d16afSAlan Maguire 
2101920d16afSAlan Maguire 	for (i = 0; i < n; i++, m++) {
2102920d16afSAlan Maguire 		const struct btf_type *mtype;
2103920d16afSAlan Maguire 		const char *mname;
2104920d16afSAlan Maguire 		__u32 moffset;
2105920d16afSAlan Maguire 		__u8 bit_sz;
2106920d16afSAlan Maguire 
2107920d16afSAlan Maguire 		mtype = btf__type_by_id(d->btf, m->type);
2108920d16afSAlan Maguire 		mname = btf_name_of(d, m->name_off);
2109920d16afSAlan Maguire 		moffset = btf_member_bit_offset(t, i);
2110920d16afSAlan Maguire 
2111920d16afSAlan Maguire 		bit_sz = btf_member_bitfield_size(t, i);
2112920d16afSAlan Maguire 		err = btf_dump_dump_type_data(d, mname, mtype, m->type, data + moffset / 8,
2113920d16afSAlan Maguire 					      moffset % 8, bit_sz);
2114920d16afSAlan Maguire 		if (err < 0)
2115920d16afSAlan Maguire 			return err;
2116920d16afSAlan Maguire 	}
2117920d16afSAlan Maguire 	d->typed_dump->depth--;
2118920d16afSAlan Maguire 	btf_dump_data_pfx(d);
2119920d16afSAlan Maguire 	btf_dump_type_values(d, "}");
2120920d16afSAlan Maguire 	return err;
2121920d16afSAlan Maguire }
2122920d16afSAlan Maguire 
21238d44c357SAlan Maguire union ptr_data {
21248d44c357SAlan Maguire 	unsigned int p;
21258d44c357SAlan Maguire 	unsigned long long lp;
21268d44c357SAlan Maguire };
21278d44c357SAlan Maguire 
btf_dump_ptr_data(struct btf_dump * d,const struct btf_type * t,__u32 id,const void * data)2128920d16afSAlan Maguire static int btf_dump_ptr_data(struct btf_dump *d,
2129920d16afSAlan Maguire 			      const struct btf_type *t,
2130920d16afSAlan Maguire 			      __u32 id,
2131920d16afSAlan Maguire 			      const void *data)
2132920d16afSAlan Maguire {
2133632f96d2SIlya Leoshkevich 	if (ptr_is_aligned(d->btf, id, data) && d->ptr_sz == sizeof(void *)) {
2134920d16afSAlan Maguire 		btf_dump_type_values(d, "%p", *(void **)data);
21358d44c357SAlan Maguire 	} else {
21368d44c357SAlan Maguire 		union ptr_data pt;
21378d44c357SAlan Maguire 
21388d44c357SAlan Maguire 		memcpy(&pt, data, d->ptr_sz);
21398d44c357SAlan Maguire 		if (d->ptr_sz == 4)
21408d44c357SAlan Maguire 			btf_dump_type_values(d, "0x%x", pt.p);
21418d44c357SAlan Maguire 		else
21428d44c357SAlan Maguire 			btf_dump_type_values(d, "0x%llx", pt.lp);
21438d44c357SAlan Maguire 	}
2144920d16afSAlan Maguire 	return 0;
2145920d16afSAlan Maguire }
2146920d16afSAlan Maguire 
btf_dump_get_enum_value(struct btf_dump * d,const struct btf_type * t,const void * data,__u32 id,__s64 * value)2147920d16afSAlan Maguire static int btf_dump_get_enum_value(struct btf_dump *d,
2148920d16afSAlan Maguire 				   const struct btf_type *t,
2149920d16afSAlan Maguire 				   const void *data,
2150920d16afSAlan Maguire 				   __u32 id,
2151920d16afSAlan Maguire 				   __s64 *value)
2152920d16afSAlan Maguire {
2153d90ec262SYonghong Song 	bool is_signed = btf_kflag(t);
2154d90ec262SYonghong Song 
2155632f96d2SIlya Leoshkevich 	if (!ptr_is_aligned(d->btf, id, data)) {
2156a1d3cc3cSAlan Maguire 		__u64 val;
2157a1d3cc3cSAlan Maguire 		int err;
2158a1d3cc3cSAlan Maguire 
2159a1d3cc3cSAlan Maguire 		err = btf_dump_get_bitfield_value(d, t, data, 0, 0, &val);
2160a1d3cc3cSAlan Maguire 		if (err)
2161a1d3cc3cSAlan Maguire 			return err;
2162a1d3cc3cSAlan Maguire 		*value = (__s64)val;
2163920d16afSAlan Maguire 		return 0;
2164920d16afSAlan Maguire 	}
2165a1d3cc3cSAlan Maguire 
2166920d16afSAlan Maguire 	switch (t->size) {
2167920d16afSAlan Maguire 	case 8:
2168920d16afSAlan Maguire 		*value = *(__s64 *)data;
2169920d16afSAlan Maguire 		return 0;
2170920d16afSAlan Maguire 	case 4:
2171c6018fc6SDan Carpenter 		*value = is_signed ? (__s64)*(__s32 *)data : *(__u32 *)data;
2172920d16afSAlan Maguire 		return 0;
2173920d16afSAlan Maguire 	case 2:
2174d90ec262SYonghong Song 		*value = is_signed ? *(__s16 *)data : *(__u16 *)data;
2175920d16afSAlan Maguire 		return 0;
2176920d16afSAlan Maguire 	case 1:
2177d90ec262SYonghong Song 		*value = is_signed ? *(__s8 *)data : *(__u8 *)data;
2178920d16afSAlan Maguire 		return 0;
2179920d16afSAlan Maguire 	default:
2180920d16afSAlan Maguire 		pr_warn("unexpected size %d for enum, id:[%u]\n", t->size, id);
2181920d16afSAlan Maguire 		return -EINVAL;
2182920d16afSAlan Maguire 	}
2183920d16afSAlan Maguire }
2184920d16afSAlan Maguire 
btf_dump_enum_data(struct btf_dump * d,const struct btf_type * t,__u32 id,const void * data)2185920d16afSAlan Maguire static int btf_dump_enum_data(struct btf_dump *d,
2186920d16afSAlan Maguire 			      const struct btf_type *t,
2187920d16afSAlan Maguire 			      __u32 id,
2188920d16afSAlan Maguire 			      const void *data)
2189920d16afSAlan Maguire {
2190d90ec262SYonghong Song 	bool is_signed;
2191920d16afSAlan Maguire 	__s64 value;
2192920d16afSAlan Maguire 	int i, err;
2193920d16afSAlan Maguire 
2194920d16afSAlan Maguire 	err = btf_dump_get_enum_value(d, t, data, id, &value);
2195920d16afSAlan Maguire 	if (err)
2196920d16afSAlan Maguire 		return err;
2197920d16afSAlan Maguire 
2198d90ec262SYonghong Song 	is_signed = btf_kflag(t);
2199d90ec262SYonghong Song 	if (btf_is_enum(t)) {
2200d90ec262SYonghong Song 		const struct btf_enum *e;
2201d90ec262SYonghong Song 
2202920d16afSAlan Maguire 		for (i = 0, e = btf_enum(t); i < btf_vlen(t); i++, e++) {
2203920d16afSAlan Maguire 			if (value != e->val)
2204920d16afSAlan Maguire 				continue;
2205920d16afSAlan Maguire 			btf_dump_type_values(d, "%s", btf_name_of(d, e->name_off));
2206920d16afSAlan Maguire 			return 0;
2207920d16afSAlan Maguire 		}
2208920d16afSAlan Maguire 
2209d90ec262SYonghong Song 		btf_dump_type_values(d, is_signed ? "%d" : "%u", value);
2210d90ec262SYonghong Song 	} else {
2211d90ec262SYonghong Song 		const struct btf_enum64 *e;
2212d90ec262SYonghong Song 
2213d90ec262SYonghong Song 		for (i = 0, e = btf_enum64(t); i < btf_vlen(t); i++, e++) {
2214d90ec262SYonghong Song 			if (value != btf_enum64_value(e))
2215d90ec262SYonghong Song 				continue;
2216d90ec262SYonghong Song 			btf_dump_type_values(d, "%s", btf_name_of(d, e->name_off));
2217d90ec262SYonghong Song 			return 0;
2218d90ec262SYonghong Song 		}
2219d90ec262SYonghong Song 
2220d90ec262SYonghong Song 		btf_dump_type_values(d, is_signed ? "%lldLL" : "%lluULL",
2221d90ec262SYonghong Song 				     (unsigned long long)value);
2222d90ec262SYonghong Song 	}
2223920d16afSAlan Maguire 	return 0;
2224920d16afSAlan Maguire }
2225920d16afSAlan Maguire 
btf_dump_datasec_data(struct btf_dump * d,const struct btf_type * t,__u32 id,const void * data)2226920d16afSAlan Maguire static int btf_dump_datasec_data(struct btf_dump *d,
2227920d16afSAlan Maguire 				 const struct btf_type *t,
2228920d16afSAlan Maguire 				 __u32 id,
2229920d16afSAlan Maguire 				 const void *data)
2230920d16afSAlan Maguire {
2231920d16afSAlan Maguire 	const struct btf_var_secinfo *vsi;
2232920d16afSAlan Maguire 	const struct btf_type *var;
2233920d16afSAlan Maguire 	__u32 i;
2234920d16afSAlan Maguire 	int err;
2235920d16afSAlan Maguire 
2236920d16afSAlan Maguire 	btf_dump_type_values(d, "SEC(\"%s\") ", btf_name_of(d, t->name_off));
2237920d16afSAlan Maguire 
2238920d16afSAlan Maguire 	for (i = 0, vsi = btf_var_secinfos(t); i < btf_vlen(t); i++, vsi++) {
2239920d16afSAlan Maguire 		var = btf__type_by_id(d->btf, vsi->type);
2240920d16afSAlan Maguire 		err = btf_dump_dump_type_data(d, NULL, var, vsi->type, data + vsi->offset, 0, 0);
2241920d16afSAlan Maguire 		if (err < 0)
2242920d16afSAlan Maguire 			return err;
2243920d16afSAlan Maguire 		btf_dump_printf(d, ";");
2244920d16afSAlan Maguire 	}
2245920d16afSAlan Maguire 	return 0;
2246920d16afSAlan Maguire }
2247920d16afSAlan Maguire 
2248920d16afSAlan Maguire /* return size of type, or if base type overflows, return -E2BIG. */
btf_dump_type_data_check_overflow(struct btf_dump * d,const struct btf_type * t,__u32 id,const void * data,__u8 bits_offset,__u8 bit_sz)2249920d16afSAlan Maguire static int btf_dump_type_data_check_overflow(struct btf_dump *d,
2250920d16afSAlan Maguire 					     const struct btf_type *t,
2251920d16afSAlan Maguire 					     __u32 id,
2252920d16afSAlan Maguire 					     const void *data,
2253*c39028b3SMartin KaFai Lau 					     __u8 bits_offset,
2254*c39028b3SMartin KaFai Lau 					     __u8 bit_sz)
2255920d16afSAlan Maguire {
2256*c39028b3SMartin KaFai Lau 	__s64 size;
2257*c39028b3SMartin KaFai Lau 
2258*c39028b3SMartin KaFai Lau 	if (bit_sz) {
2259*c39028b3SMartin KaFai Lau 		/* bits_offset is at most 7. bit_sz is at most 128. */
2260*c39028b3SMartin KaFai Lau 		__u8 nr_bytes = (bits_offset + bit_sz + 7) / 8;
2261*c39028b3SMartin KaFai Lau 
2262*c39028b3SMartin KaFai Lau 		/* When bit_sz is non zero, it is called from
2263*c39028b3SMartin KaFai Lau 		 * btf_dump_struct_data() where it only cares about
2264*c39028b3SMartin KaFai Lau 		 * negative error value.
2265*c39028b3SMartin KaFai Lau 		 * Return nr_bytes in success case to make it
2266*c39028b3SMartin KaFai Lau 		 * consistent as the regular integer case below.
2267*c39028b3SMartin KaFai Lau 		 */
2268*c39028b3SMartin KaFai Lau 		return data + nr_bytes > d->typed_dump->data_end ? -E2BIG : nr_bytes;
2269*c39028b3SMartin KaFai Lau 	}
2270*c39028b3SMartin KaFai Lau 
2271*c39028b3SMartin KaFai Lau 	size = btf__resolve_size(d->btf, id);
2272920d16afSAlan Maguire 
2273920d16afSAlan Maguire 	if (size < 0 || size >= INT_MAX) {
227404eb4dffSAlan Maguire 		pr_warn("unexpected size [%zu] for id [%u]\n",
227504eb4dffSAlan Maguire 			(size_t)size, id);
2276920d16afSAlan Maguire 		return -EINVAL;
2277920d16afSAlan Maguire 	}
2278920d16afSAlan Maguire 
2279920d16afSAlan Maguire 	/* Only do overflow checking for base types; we do not want to
2280920d16afSAlan Maguire 	 * avoid showing part of a struct, union or array, even if we
2281920d16afSAlan Maguire 	 * do not have enough data to show the full object.  By
2282920d16afSAlan Maguire 	 * restricting overflow checking to base types we can ensure
2283920d16afSAlan Maguire 	 * that partial display succeeds, while avoiding overflowing
2284920d16afSAlan Maguire 	 * and using bogus data for display.
2285920d16afSAlan Maguire 	 */
2286920d16afSAlan Maguire 	t = skip_mods_and_typedefs(d->btf, id, NULL);
2287920d16afSAlan Maguire 	if (!t) {
2288920d16afSAlan Maguire 		pr_warn("unexpected error skipping mods/typedefs for id [%u]\n",
2289920d16afSAlan Maguire 			id);
2290920d16afSAlan Maguire 		return -EINVAL;
2291920d16afSAlan Maguire 	}
2292920d16afSAlan Maguire 
2293920d16afSAlan Maguire 	switch (btf_kind(t)) {
2294920d16afSAlan Maguire 	case BTF_KIND_INT:
2295920d16afSAlan Maguire 	case BTF_KIND_FLOAT:
2296920d16afSAlan Maguire 	case BTF_KIND_PTR:
2297920d16afSAlan Maguire 	case BTF_KIND_ENUM:
2298d90ec262SYonghong Song 	case BTF_KIND_ENUM64:
2299920d16afSAlan Maguire 		if (data + bits_offset / 8 + size > d->typed_dump->data_end)
2300920d16afSAlan Maguire 			return -E2BIG;
2301920d16afSAlan Maguire 		break;
2302920d16afSAlan Maguire 	default:
2303920d16afSAlan Maguire 		break;
2304920d16afSAlan Maguire 	}
2305920d16afSAlan Maguire 	return (int)size;
2306920d16afSAlan Maguire }
2307920d16afSAlan Maguire 
btf_dump_type_data_check_zero(struct btf_dump * d,const struct btf_type * t,__u32 id,const void * data,__u8 bits_offset,__u8 bit_sz)2308920d16afSAlan Maguire static int btf_dump_type_data_check_zero(struct btf_dump *d,
2309920d16afSAlan Maguire 					 const struct btf_type *t,
2310920d16afSAlan Maguire 					 __u32 id,
2311920d16afSAlan Maguire 					 const void *data,
2312920d16afSAlan Maguire 					 __u8 bits_offset,
2313920d16afSAlan Maguire 					 __u8 bit_sz)
2314920d16afSAlan Maguire {
2315920d16afSAlan Maguire 	__s64 value;
2316920d16afSAlan Maguire 	int i, err;
2317920d16afSAlan Maguire 
2318920d16afSAlan Maguire 	/* toplevel exceptions; we show zero values if
2319920d16afSAlan Maguire 	 * - we ask for them (emit_zeros)
2320920d16afSAlan Maguire 	 * - if we are at top-level so we see "struct empty { }"
2321920d16afSAlan Maguire 	 * - or if we are an array member and the array is non-empty and
2322920d16afSAlan Maguire 	 *   not a char array; we don't want to be in a situation where we
2323920d16afSAlan Maguire 	 *   have an integer array 0, 1, 0, 1 and only show non-zero values.
2324920d16afSAlan Maguire 	 *   If the array contains zeroes only, or is a char array starting
2325920d16afSAlan Maguire 	 *   with a '\0', the array-level check_zero() will prevent showing it;
2326920d16afSAlan Maguire 	 *   we are concerned with determining zero value at the array member
2327920d16afSAlan Maguire 	 *   level here.
2328920d16afSAlan Maguire 	 */
2329920d16afSAlan Maguire 	if (d->typed_dump->emit_zeroes || d->typed_dump->depth == 0 ||
2330920d16afSAlan Maguire 	    (d->typed_dump->is_array_member &&
2331920d16afSAlan Maguire 	     !d->typed_dump->is_array_char))
2332920d16afSAlan Maguire 		return 0;
2333920d16afSAlan Maguire 
2334920d16afSAlan Maguire 	t = skip_mods_and_typedefs(d->btf, id, NULL);
2335920d16afSAlan Maguire 
2336920d16afSAlan Maguire 	switch (btf_kind(t)) {
2337920d16afSAlan Maguire 	case BTF_KIND_INT:
2338920d16afSAlan Maguire 		if (bit_sz)
2339920d16afSAlan Maguire 			return btf_dump_bitfield_check_zero(d, t, data, bits_offset, bit_sz);
2340920d16afSAlan Maguire 		return btf_dump_base_type_check_zero(d, t, id, data);
2341920d16afSAlan Maguire 	case BTF_KIND_FLOAT:
2342920d16afSAlan Maguire 	case BTF_KIND_PTR:
2343920d16afSAlan Maguire 		return btf_dump_base_type_check_zero(d, t, id, data);
2344920d16afSAlan Maguire 	case BTF_KIND_ARRAY: {
2345920d16afSAlan Maguire 		const struct btf_array *array = btf_array(t);
2346920d16afSAlan Maguire 		const struct btf_type *elem_type;
2347920d16afSAlan Maguire 		__u32 elem_type_id, elem_size;
2348920d16afSAlan Maguire 		bool ischar;
2349920d16afSAlan Maguire 
2350920d16afSAlan Maguire 		elem_type_id = array->type;
2351920d16afSAlan Maguire 		elem_size = btf__resolve_size(d->btf, elem_type_id);
2352920d16afSAlan Maguire 		elem_type = skip_mods_and_typedefs(d->btf, elem_type_id, NULL);
2353920d16afSAlan Maguire 
2354920d16afSAlan Maguire 		ischar = btf_is_int(elem_type) && elem_size == 1;
2355920d16afSAlan Maguire 
2356920d16afSAlan Maguire 		/* check all elements; if _any_ element is nonzero, all
2357920d16afSAlan Maguire 		 * of array is displayed.  We make an exception however
2358920d16afSAlan Maguire 		 * for char arrays where the first element is 0; these
2359920d16afSAlan Maguire 		 * are considered zeroed also, even if later elements are
2360920d16afSAlan Maguire 		 * non-zero because the string is terminated.
2361920d16afSAlan Maguire 		 */
2362920d16afSAlan Maguire 		for (i = 0; i < array->nelems; i++) {
2363920d16afSAlan Maguire 			if (i == 0 && ischar && *(char *)data == 0)
2364920d16afSAlan Maguire 				return -ENODATA;
2365920d16afSAlan Maguire 			err = btf_dump_type_data_check_zero(d, elem_type,
2366920d16afSAlan Maguire 							    elem_type_id,
2367920d16afSAlan Maguire 							    data +
2368920d16afSAlan Maguire 							    (i * elem_size),
2369920d16afSAlan Maguire 							    bits_offset, 0);
2370920d16afSAlan Maguire 			if (err != -ENODATA)
2371920d16afSAlan Maguire 				return err;
2372920d16afSAlan Maguire 		}
2373920d16afSAlan Maguire 		return -ENODATA;
2374920d16afSAlan Maguire 	}
2375920d16afSAlan Maguire 	case BTF_KIND_STRUCT:
2376920d16afSAlan Maguire 	case BTF_KIND_UNION: {
2377920d16afSAlan Maguire 		const struct btf_member *m = btf_members(t);
2378920d16afSAlan Maguire 		__u16 n = btf_vlen(t);
2379920d16afSAlan Maguire 
2380920d16afSAlan Maguire 		/* if any struct/union member is non-zero, the struct/union
2381920d16afSAlan Maguire 		 * is considered non-zero and dumped.
2382920d16afSAlan Maguire 		 */
2383920d16afSAlan Maguire 		for (i = 0; i < n; i++, m++) {
2384920d16afSAlan Maguire 			const struct btf_type *mtype;
2385920d16afSAlan Maguire 			__u32 moffset;
2386920d16afSAlan Maguire 
2387920d16afSAlan Maguire 			mtype = btf__type_by_id(d->btf, m->type);
2388920d16afSAlan Maguire 			moffset = btf_member_bit_offset(t, i);
2389920d16afSAlan Maguire 
2390920d16afSAlan Maguire 			/* btf_int_bits() does not store member bitfield size;
2391920d16afSAlan Maguire 			 * bitfield size needs to be stored here so int display
2392920d16afSAlan Maguire 			 * of member can retrieve it.
2393920d16afSAlan Maguire 			 */
2394920d16afSAlan Maguire 			bit_sz = btf_member_bitfield_size(t, i);
2395920d16afSAlan Maguire 			err = btf_dump_type_data_check_zero(d, mtype, m->type, data + moffset / 8,
2396920d16afSAlan Maguire 							    moffset % 8, bit_sz);
2397920d16afSAlan Maguire 			if (err != ENODATA)
2398920d16afSAlan Maguire 				return err;
2399920d16afSAlan Maguire 		}
2400920d16afSAlan Maguire 		return -ENODATA;
2401920d16afSAlan Maguire 	}
2402920d16afSAlan Maguire 	case BTF_KIND_ENUM:
2403d90ec262SYonghong Song 	case BTF_KIND_ENUM64:
2404720c29fcSAlan Maguire 		err = btf_dump_get_enum_value(d, t, data, id, &value);
2405720c29fcSAlan Maguire 		if (err)
2406720c29fcSAlan Maguire 			return err;
2407920d16afSAlan Maguire 		if (value == 0)
2408920d16afSAlan Maguire 			return -ENODATA;
2409920d16afSAlan Maguire 		return 0;
2410920d16afSAlan Maguire 	default:
2411920d16afSAlan Maguire 		return 0;
2412920d16afSAlan Maguire 	}
2413920d16afSAlan Maguire }
2414920d16afSAlan Maguire 
2415920d16afSAlan Maguire /* returns size of data dumped, or error. */
btf_dump_dump_type_data(struct btf_dump * d,const char * fname,const struct btf_type * t,__u32 id,const void * data,__u8 bits_offset,__u8 bit_sz)2416920d16afSAlan Maguire static int btf_dump_dump_type_data(struct btf_dump *d,
2417920d16afSAlan Maguire 				   const char *fname,
2418920d16afSAlan Maguire 				   const struct btf_type *t,
2419920d16afSAlan Maguire 				   __u32 id,
2420920d16afSAlan Maguire 				   const void *data,
2421920d16afSAlan Maguire 				   __u8 bits_offset,
2422920d16afSAlan Maguire 				   __u8 bit_sz)
2423920d16afSAlan Maguire {
242443174f0dSAlan Maguire 	int size, err = 0;
2425920d16afSAlan Maguire 
2426*c39028b3SMartin KaFai Lau 	size = btf_dump_type_data_check_overflow(d, t, id, data, bits_offset, bit_sz);
2427920d16afSAlan Maguire 	if (size < 0)
2428920d16afSAlan Maguire 		return size;
2429920d16afSAlan Maguire 	err = btf_dump_type_data_check_zero(d, t, id, data, bits_offset, bit_sz);
2430920d16afSAlan Maguire 	if (err) {
2431920d16afSAlan Maguire 		/* zeroed data is expected and not an error, so simply skip
2432920d16afSAlan Maguire 		 * dumping such data.  Record other errors however.
2433920d16afSAlan Maguire 		 */
2434920d16afSAlan Maguire 		if (err == -ENODATA)
2435920d16afSAlan Maguire 			return size;
2436920d16afSAlan Maguire 		return err;
2437920d16afSAlan Maguire 	}
2438920d16afSAlan Maguire 	btf_dump_data_pfx(d);
2439920d16afSAlan Maguire 
2440920d16afSAlan Maguire 	if (!d->typed_dump->skip_names) {
2441920d16afSAlan Maguire 		if (fname && strlen(fname) > 0)
2442920d16afSAlan Maguire 			btf_dump_printf(d, ".%s = ", fname);
2443920d16afSAlan Maguire 		btf_dump_emit_type_cast(d, id, true);
2444920d16afSAlan Maguire 	}
2445920d16afSAlan Maguire 
2446920d16afSAlan Maguire 	t = skip_mods_and_typedefs(d->btf, id, NULL);
2447920d16afSAlan Maguire 
2448920d16afSAlan Maguire 	switch (btf_kind(t)) {
2449920d16afSAlan Maguire 	case BTF_KIND_UNKN:
2450920d16afSAlan Maguire 	case BTF_KIND_FWD:
2451920d16afSAlan Maguire 	case BTF_KIND_FUNC:
2452920d16afSAlan Maguire 	case BTF_KIND_FUNC_PROTO:
2453223f903eSYonghong Song 	case BTF_KIND_DECL_TAG:
2454920d16afSAlan Maguire 		err = btf_dump_unsupported_data(d, t, id);
2455920d16afSAlan Maguire 		break;
2456920d16afSAlan Maguire 	case BTF_KIND_INT:
2457920d16afSAlan Maguire 		if (bit_sz)
2458920d16afSAlan Maguire 			err = btf_dump_bitfield_data(d, t, data, bits_offset, bit_sz);
2459920d16afSAlan Maguire 		else
2460920d16afSAlan Maguire 			err = btf_dump_int_data(d, t, id, data, bits_offset);
2461920d16afSAlan Maguire 		break;
2462920d16afSAlan Maguire 	case BTF_KIND_FLOAT:
2463920d16afSAlan Maguire 		err = btf_dump_float_data(d, t, id, data);
2464920d16afSAlan Maguire 		break;
2465920d16afSAlan Maguire 	case BTF_KIND_PTR:
2466920d16afSAlan Maguire 		err = btf_dump_ptr_data(d, t, id, data);
2467920d16afSAlan Maguire 		break;
2468920d16afSAlan Maguire 	case BTF_KIND_ARRAY:
2469920d16afSAlan Maguire 		err = btf_dump_array_data(d, t, id, data);
2470920d16afSAlan Maguire 		break;
2471920d16afSAlan Maguire 	case BTF_KIND_STRUCT:
2472920d16afSAlan Maguire 	case BTF_KIND_UNION:
2473920d16afSAlan Maguire 		err = btf_dump_struct_data(d, t, id, data);
2474920d16afSAlan Maguire 		break;
2475920d16afSAlan Maguire 	case BTF_KIND_ENUM:
2476d90ec262SYonghong Song 	case BTF_KIND_ENUM64:
2477920d16afSAlan Maguire 		/* handle bitfield and int enum values */
2478920d16afSAlan Maguire 		if (bit_sz) {
2479a1d3cc3cSAlan Maguire 			__u64 print_num;
2480920d16afSAlan Maguire 			__s64 enum_val;
2481920d16afSAlan Maguire 
2482a1d3cc3cSAlan Maguire 			err = btf_dump_get_bitfield_value(d, t, data, bits_offset, bit_sz,
2483a1d3cc3cSAlan Maguire 							  &print_num);
2484a1d3cc3cSAlan Maguire 			if (err)
2485a1d3cc3cSAlan Maguire 				break;
2486920d16afSAlan Maguire 			enum_val = (__s64)print_num;
2487920d16afSAlan Maguire 			err = btf_dump_enum_data(d, t, id, &enum_val);
2488920d16afSAlan Maguire 		} else
2489920d16afSAlan Maguire 			err = btf_dump_enum_data(d, t, id, data);
2490920d16afSAlan Maguire 		break;
2491920d16afSAlan Maguire 	case BTF_KIND_VAR:
2492920d16afSAlan Maguire 		err = btf_dump_var_data(d, t, id, data);
2493920d16afSAlan Maguire 		break;
2494920d16afSAlan Maguire 	case BTF_KIND_DATASEC:
2495920d16afSAlan Maguire 		err = btf_dump_datasec_data(d, t, id, data);
2496920d16afSAlan Maguire 		break;
2497920d16afSAlan Maguire 	default:
2498920d16afSAlan Maguire 		pr_warn("unexpected kind [%u] for id [%u]\n",
2499920d16afSAlan Maguire 			BTF_INFO_KIND(t->info), id);
2500920d16afSAlan Maguire 		return -EINVAL;
2501920d16afSAlan Maguire 	}
2502920d16afSAlan Maguire 	if (err < 0)
2503920d16afSAlan Maguire 		return err;
2504920d16afSAlan Maguire 	return size;
2505920d16afSAlan Maguire }
2506920d16afSAlan Maguire 
btf_dump__dump_type_data(struct btf_dump * d,__u32 id,const void * data,size_t data_sz,const struct btf_dump_type_data_opts * opts)2507920d16afSAlan Maguire int btf_dump__dump_type_data(struct btf_dump *d, __u32 id,
2508920d16afSAlan Maguire 			     const void *data, size_t data_sz,
2509920d16afSAlan Maguire 			     const struct btf_dump_type_data_opts *opts)
2510920d16afSAlan Maguire {
2511add192f8SAlan Maguire 	struct btf_dump_data typed_dump = {};
2512920d16afSAlan Maguire 	const struct btf_type *t;
2513920d16afSAlan Maguire 	int ret;
2514920d16afSAlan Maguire 
2515920d16afSAlan Maguire 	if (!OPTS_VALID(opts, btf_dump_type_data_opts))
2516920d16afSAlan Maguire 		return libbpf_err(-EINVAL);
2517920d16afSAlan Maguire 
2518920d16afSAlan Maguire 	t = btf__type_by_id(d->btf, id);
2519920d16afSAlan Maguire 	if (!t)
2520920d16afSAlan Maguire 		return libbpf_err(-ENOENT);
2521920d16afSAlan Maguire 
2522add192f8SAlan Maguire 	d->typed_dump = &typed_dump;
2523920d16afSAlan Maguire 	d->typed_dump->data_end = data + data_sz;
2524920d16afSAlan Maguire 	d->typed_dump->indent_lvl = OPTS_GET(opts, indent_level, 0);
2525add192f8SAlan Maguire 
2526920d16afSAlan Maguire 	/* default indent string is a tab */
25277620bffbSXin Liu 	if (!OPTS_GET(opts, indent_str, NULL))
2528920d16afSAlan Maguire 		d->typed_dump->indent_str[0] = '\t';
2529920d16afSAlan Maguire 	else
25309fc205b4SAndrii Nakryiko 		libbpf_strlcpy(d->typed_dump->indent_str, opts->indent_str,
25319fc205b4SAndrii Nakryiko 			       sizeof(d->typed_dump->indent_str));
2532920d16afSAlan Maguire 
2533920d16afSAlan Maguire 	d->typed_dump->compact = OPTS_GET(opts, compact, false);
2534920d16afSAlan Maguire 	d->typed_dump->skip_names = OPTS_GET(opts, skip_names, false);
2535920d16afSAlan Maguire 	d->typed_dump->emit_zeroes = OPTS_GET(opts, emit_zeroes, false);
2536920d16afSAlan Maguire 
2537920d16afSAlan Maguire 	ret = btf_dump_dump_type_data(d, NULL, t, id, data, 0, 0);
2538920d16afSAlan Maguire 
2539add192f8SAlan Maguire 	d->typed_dump = NULL;
2540920d16afSAlan Maguire 
2541920d16afSAlan Maguire 	return libbpf_err(ret);
2542920d16afSAlan Maguire }
2543