xref: /linux/tools/testing/selftests/drivers/net/hw/userns_devmem.py (revision 78c1930198fc63f2d4761848cbe148c5b2958b01)
1#!/usr/bin/env python3
2# SPDX-License-Identifier: GPL-2.0
3"""
4Devmem tests for non-init userns.
5"""
6
7import os
8
9from devmem_lib import run_rx, run_rx_hds, run_tx, run_tx_chunks, setup_test
10from lib.py import NetDrvContEnv, ksft_disruptive, ksft_exit, ksft_run
11
12
13@ksft_disruptive
14def check_userns_rx(cfg) -> None:
15    """Run the devmem RX test through non-init userns netkit."""
16    run_rx(cfg)
17
18
19@ksft_disruptive
20def check_userns_tx(cfg) -> None:
21    """Run the devmem TX test through non-init userns netkit."""
22    run_tx(cfg)
23
24
25@ksft_disruptive
26def check_userns_tx_chunks(cfg) -> None:
27    """Run the devmem TX chunking test through non-init userns netkit."""
28    run_tx_chunks(cfg)
29
30
31def check_userns_rx_hds(cfg) -> None:
32    """Run the HDS test through non-init userns netkit."""
33    run_rx_hds(cfg)
34
35
36def main() -> None:
37    """Run userns devmem RX selftests against the test environment."""
38    with NetDrvContEnv(__file__, userns=True, rxqueues=2,
39                       primary_rx_redirect=True) as cfg:
40        setup_test(cfg,
41                   os.path.join(os.path.dirname(os.path.abspath(__file__)),
42                                "ncdevmem"))
43        ksft_run([check_userns_rx, check_userns_tx, check_userns_tx_chunks,
44                  check_userns_rx_hds], args=(cfg,))
45    ksft_exit()
46
47
48if __name__ == "__main__":
49    main()
50