On Thu, Mar 25, 2021 at 10:11:22PM +0100, Toke Høiland-Jørgensen wrote: > This adds a selftest to check that the verifier rejects a TCP CC struct_ops > with a non-GPL license. > > v2: > - Use a minimal struct_ops BPF program instead of rewriting bpf_dctcp's > license in memory. > - Check for the verifier reject message instead of just the return code. > > Signed-off-by: Toke Høiland-Jørgensen <toke@xxxxxxxxxx> > --- > .../selftests/bpf/prog_tests/bpf_tcp_ca.c | 44 +++++++++++++++++++ > .../selftests/bpf/progs/bpf_nogpltcp.c | 19 ++++++++ > 2 files changed, 63 insertions(+) > create mode 100644 tools/testing/selftests/bpf/progs/bpf_nogpltcp.c > > diff --git a/tools/testing/selftests/bpf/prog_tests/bpf_tcp_ca.c b/tools/testing/selftests/bpf/prog_tests/bpf_tcp_ca.c > index 37c5494a0381..a09c716528e1 100644 > --- a/tools/testing/selftests/bpf/prog_tests/bpf_tcp_ca.c > +++ b/tools/testing/selftests/bpf/prog_tests/bpf_tcp_ca.c > @@ -6,6 +6,7 @@ > #include <test_progs.h> > #include "bpf_dctcp.skel.h" > #include "bpf_cubic.skel.h" > +#include "bpf_nogpltcp.skel.h" > > #define min(a, b) ((a) < (b) ? (a) : (b)) > > @@ -227,10 +228,53 @@ static void test_dctcp(void) > bpf_dctcp__destroy(dctcp_skel); > } > > +static char *err_str = NULL; > +static bool found = false; Nit. These two inits are not needed. > + > +static int libbpf_debug_print(enum libbpf_print_level level, > + const char *format, va_list args) > +{ > + char *log_buf; > + > + if (level != LIBBPF_WARN || > + strcmp(format, "libbpf: \n%s\n")) { > + vprintf(format, args); > + return 0; > + } > + > + log_buf = va_arg(args, char *); > + if (!log_buf) > + goto out; > + if (err_str && strstr(log_buf, err_str) != NULL) > + found = true; > +out: > + printf(format, log_buf); > + return 0; > +} > + > +static void test_invalid_license(void) > +{ > + libbpf_print_fn_t old_print_fn = NULL; Nit. Same here. Not need to init NULL. Others lgtm. Acked-by: Martin KaFai Lau <kafai@xxxxxx> > + struct bpf_nogpltcp *skel; > + > + err_str = "struct ops programs must have a GPL compatible license"; > + old_print_fn = libbpf_set_print(libbpf_debug_print); > + > + skel = bpf_nogpltcp__open_and_load(); > + if (CHECK(skel, "bpf_nogplgtcp__open_and_load()", "didn't fail\n")) > + bpf_nogpltcp__destroy(skel); > + > + CHECK(!found, "errmsg check", "expected string '%s'", err_str); > + > + libbpf_set_print(old_print_fn); > +}