On Fri, Aug 30, 2024 at 12:30 AM Tony Ambardar <tony.ambar...@gmail.com> wrote: > > Track target endianness in 'struct bpf_gen' and process in-memory data in > native byte-order, but on finalization convert the embedded loader BPF > insns to target endianness. > > The light skeleton also includes a target-accessed data blob which is > heterogeneous and thus difficult to convert to target byte-order on > finalization. Add support functions to convert data to target endianness > as it is added to the blob. > > Also add additional debug logging for data blob structure details and > skeleton loading. > > Signed-off-by: Tony Ambardar <tony.ambar...@gmail.com> > --- > tools/lib/bpf/bpf_gen_internal.h | 1 + > tools/lib/bpf/gen_loader.c | 187 +++++++++++++++++++++++-------- > tools/lib/bpf/libbpf.c | 1 + > tools/lib/bpf/skel_internal.h | 3 +- > 4 files changed, 147 insertions(+), 45 deletions(-) > > diff --git a/tools/lib/bpf/bpf_gen_internal.h > b/tools/lib/bpf/bpf_gen_internal.h > index fdf44403ff36..6ff963a491d9 100644 > --- a/tools/lib/bpf/bpf_gen_internal.h > +++ b/tools/lib/bpf/bpf_gen_internal.h > @@ -34,6 +34,7 @@ struct bpf_gen { > void *data_cur; > void *insn_start; > void *insn_cur; > + bool swapped_endian; > ssize_t cleanup_label; > __u32 nr_progs; > __u32 nr_maps; > diff --git a/tools/lib/bpf/gen_loader.c b/tools/lib/bpf/gen_loader.c > index cf3323fd47b8..4374399bc3f8 100644 > --- a/tools/lib/bpf/gen_loader.c > +++ b/tools/lib/bpf/gen_loader.c > @@ -401,6 +401,15 @@ int bpf_gen__finish(struct bpf_gen *gen, int nr_progs, > int nr_maps) > opts->insns_sz = gen->insn_cur - gen->insn_start; > opts->data = gen->data_start; > opts->data_sz = gen->data_cur - gen->data_start; > + > + /* use target endianness for embedded loader */ > + if (gen->swapped_endian) { > + struct bpf_insn *insn = (struct bpf_insn > *)opts->insns; > + int insn_cnt = opts->insns_sz / sizeof(struct > bpf_insn); > + > + for (i = 0; i < insn_cnt; i++) > + bpf_insn_bswap(insn++); > + } > } > return gen->error; > } > @@ -414,6 +423,31 @@ void bpf_gen__free(struct bpf_gen *gen) > free(gen); > } > > +/* > + * Fields of bpf_attr are set to values in native byte-order before being > + * written to the target-bound data blob, and may need endian conversion. > + * This macro allows providing the correct value in situ more simply than > + * writing a separate converter for *all fields* of *all records* included > + * in union bpf_attr. Note that sizeof(rval) should match the assignment > + * target to avoid runtime problems. > + */ > +#define tgt_endian(rval) ({ \ > + typeof(rval) _val; \ > + if (!gen->swapped_endian) \
if/else has to have balanced branches w.r.t. {}. Either both should have it or both shouldn't. In this case both should have it. > + _val = (rval); \ > + else { \ > + switch (sizeof(rval)) { \ > + case 1: _val = (rval); break; \ > + case 2: _val = bswap_16(rval); break; \ > + case 4: _val = bswap_32(rval); break; \ > + case 8: _val = bswap_64(rval); break; \ > + default:_val = (rval); \ > + pr_warn("unsupported bswap size!\n"); \ this is a weird formatting, but you can also just unconditionally assign _val, and only swap it if gen->swapped_endian typeof(rval) _val = (rval); if (gen->swapped_endian) { switch (...) { case 1: ... ... case 8: ... default: pr_warn("..."); } } _val; seems simpler and cleaner, imo > + } \ > + } \ > + _val; \ > +}) > + for the rest, Alexei, can you please review and give your ack?