On Thu, 10 Aug 2023 at 14:14, Przemek Kitszel <przemyslaw.kitszel@xxxxxxxxx> wrote: > > On 8/9/23 14:43, Magnus Karlsson wrote: > > From: Magnus Karlsson <magnus.karlsson@xxxxxxxxx> > > > > Add an option -m on the command line that allows the user to run the > > tests in a single mode instead of all of them. Valid modes are skb, > > drv, and zc (zero-copy). An example: > > > > To run test suite in drv mode only: > > > > ./test_xsk.sh -m drv > > > > Signed-off-by: Magnus Karlsson <magnus.karlsson@xxxxxxxxx> > > --- > > tools/testing/selftests/bpf/test_xsk.sh | 10 ++++++- > > tools/testing/selftests/bpf/xskxceiver.c | 34 +++++++++++++++++++++--- > > tools/testing/selftests/bpf/xskxceiver.h | 4 +-- > > 3 files changed, 40 insertions(+), 8 deletions(-) > > > > diff --git a/tools/testing/selftests/bpf/test_xsk.sh b/tools/testing/selftests/bpf/test_xsk.sh > > index 2aa5a3445056..4ec621f4d3db 100755 > > --- a/tools/testing/selftests/bpf/test_xsk.sh > > +++ b/tools/testing/selftests/bpf/test_xsk.sh > > @@ -73,17 +73,21 @@ > > # > > # Run test suite for physical device in loopback mode > > # sudo ./test_xsk.sh -i IFACE > > +# > > +# Run test suite in a specific mode only [skb,drv,zc] > > +# sudo ./test_xsk.sh -m MODE > > > > . xsk_prereqs.sh > > > > ETH="" > > > > -while getopts "vi:d" flag > > +while getopts "vi:dm:" flag > > do > > case "${flag}" in > > v) verbose=1;; > > d) debug=1;; > > i) ETH=${OPTARG};; > > + m) MODE=${OPTARG};; > > esac > > done > > > > @@ -153,6 +157,10 @@ if [[ $verbose -eq 1 ]]; then > > ARGS+="-v " > > fi > > > > +if [ ! -z $MODE ]; then > > better: `if [ -n "$MODE" ]` > > note that quotes are really good invention for such cases, especially > that default value of MODE is "take such named variable from user env". Definitely better. Will fix. Thanks! > > + ARGS+="-m ${MODE} " > > +fi > > + > > retval=$? > > test_status $retval "${TEST_NAME}" > > > > diff --git a/tools/testing/selftests/bpf/xskxceiver.c b/tools/testing/selftests/bpf/xskxceiver.c > > index 514fe994e02b..9f79c2b6aa97 100644 > > --- a/tools/testing/selftests/bpf/xskxceiver.c > > +++ b/tools/testing/selftests/bpf/xskxceiver.c > > @@ -107,6 +107,9 @@ > > static const char *MAC1 = "\x00\x0A\x56\x9E\xEE\x62"; > > static const char *MAC2 = "\x00\x0A\x56\x9E\xEE\x61"; > > > > +static bool opt_verbose; > > +static enum test_mode opt_mode = TEST_MODE_ALL; > > + > > static void __exit_with_error(int error, const char *file, const char *func, int line) > > { > > ksft_test_result_fail("[%s:%s:%i]: ERROR: %d/\"%s\"\n", file, func, line, error, > > @@ -310,17 +313,19 @@ static struct option long_options[] = { > > {"interface", required_argument, 0, 'i'}, > > {"busy-poll", no_argument, 0, 'b'}, > > {"verbose", no_argument, 0, 'v'}, > > + {"mode", required_argument, 0, 'm'}, > > {0, 0, 0, 0} > > }; > > > > static void usage(const char *prog) > > { > > const char *str = > > - " Usage: %s [OPTIONS]\n" > > + " Usage: xskxceiver [OPTIONS]\n" > > " Options:\n" > > " -i, --interface Use interface\n" > > " -v, --verbose Verbose output\n" > > - " -b, --busy-poll Enable busy poll\n"; > > + " -b, --busy-poll Enable busy poll\n" > > + " -m, --mode Run only mode skb, drv, or zc\n"; > > > > ksft_print_msg(str, prog); > > } > > @@ -342,7 +347,7 @@ static void parse_command_line(struct ifobject *ifobj_tx, struct ifobject *ifobj > > opterr = 0; > > > > for (;;) { > > - c = getopt_long(argc, argv, "i:vb", long_options, &option_index); > > + c = getopt_long(argc, argv, "i:vbm:", long_options, &option_index); > > if (c == -1) > > break; > > > > @@ -371,6 +376,21 @@ static void parse_command_line(struct ifobject *ifobj_tx, struct ifobject *ifobj > > ifobj_tx->busy_poll = true; > > ifobj_rx->busy_poll = true; > > break; > > + case 'm': > > + if (!strncmp("skb", optarg, min_t(size_t, strlen(optarg), > > + strlen("skb")))) { > > + opt_mode = TEST_MODE_SKB; > > + } else if (!strncmp("drv", optarg, min_t(size_t, strlen(optarg), > > + strlen("drv")))) { > > + opt_mode = TEST_MODE_DRV; > > + } else if (!strncmp("zc", optarg, min_t(size_t, strlen(optarg), > > + strlen("zc")))) { > > + opt_mode = TEST_MODE_ZC; > > + } else { > > + usage(basename(argv[0])); > > + ksft_exit_xfail(); > > + } > > + break; > > default: > > usage(basename(argv[0])); > > ksft_exit_xfail(); > > @@ -2365,9 +2385,15 @@ int main(int argc, char **argv) > > test.tx_pkt_stream_default = tx_pkt_stream_default; > > test.rx_pkt_stream_default = rx_pkt_stream_default; > > > > - ksft_set_plan(modes * TEST_TYPE_MAX); > > + if (opt_mode == TEST_MODE_ALL) > > + ksft_set_plan(modes * TEST_TYPE_MAX); > > + else > > + ksft_set_plan(TEST_TYPE_MAX); > > > > for (i = 0; i < modes; i++) { > > + if (opt_mode != TEST_MODE_ALL && i != opt_mode) > > + continue; > > + > > for (j = 0; j < TEST_TYPE_MAX; j++) { > > test_spec_init(&test, ifobj_tx, ifobj_rx, i); > > run_pkt_test(&test, i, j); > > diff --git a/tools/testing/selftests/bpf/xskxceiver.h b/tools/testing/selftests/bpf/xskxceiver.h > > index 233b66cef64a..1412492e9618 100644 > > --- a/tools/testing/selftests/bpf/xskxceiver.h > > +++ b/tools/testing/selftests/bpf/xskxceiver.h > > @@ -63,7 +63,7 @@ enum test_mode { > > TEST_MODE_SKB, > > TEST_MODE_DRV, > > TEST_MODE_ZC, > > - TEST_MODE_MAX > > + TEST_MODE_ALL > > }; > > > > enum test_type { > > @@ -98,8 +98,6 @@ enum test_type { > > TEST_TYPE_MAX > > }; > > > > -static bool opt_verbose; > > - > > struct xsk_umem_info { > > struct xsk_ring_prod fq; > > struct xsk_ring_cons cq; >