1#!/usr/bin/env python3 2# SPDX-License-Identifier: GPL-2.0 3 4""" 5Test napi threaded states. 6""" 7 8from lib.py import ksft_run, ksft_exit 9from lib.py import ksft_eq, ksft_ne, ksft_ge 10from lib.py import NetDrvEnv, NetdevFamily 11from lib.py import cmd, defer, ethtool 12 13 14def _assert_napi_threaded_enabled(nl, napi_id) -> None: 15 napi = nl.napi_get({'id': napi_id}) 16 ksft_eq(napi['threaded'], 'enabled') 17 ksft_ne(napi.get('pid'), None) 18 19 20def _assert_napi_threaded_disabled(nl, napi_id) -> None: 21 napi = nl.napi_get({'id': napi_id}) 22 ksft_eq(napi['threaded'], 'disabled') 23 ksft_eq(napi.get('pid'), None) 24 25 26def _set_threaded_state(cfg, threaded) -> None: 27 cmd(f"echo {threaded} > /sys/class/net/{cfg.ifname}/threaded") 28 29 30def _setup_deferred_cleanup(cfg) -> None: 31 combined = ethtool(f"-l {cfg.ifname}", json=True)[0].get("combined", 0) 32 ksft_ge(combined, 2) 33 defer(ethtool, f"-L {cfg.ifname} combined {combined}") 34 35 threaded = cmd(f"cat /sys/class/net/{cfg.ifname}/threaded").stdout 36 defer(_set_threaded_state, cfg, threaded) 37 38 return combined 39 40 41def napi_init(cfg, nl) -> None: 42 """ 43 Test that threaded state (in the persistent NAPI config) gets updated 44 even when NAPI with given ID is not allocated at the time. 45 """ 46 47 qcnt = _setup_deferred_cleanup(cfg) 48 49 _set_threaded_state(cfg, 1) 50 cmd(f"ethtool -L {cfg.ifname} combined 1") 51 _set_threaded_state(cfg, 0) 52 cmd(f"ethtool -L {cfg.ifname} combined {qcnt}") 53 54 napis = nl.napi_get({'ifindex': cfg.ifindex}, dump=True) 55 for napi in napis: 56 ksft_eq(napi['threaded'], 'disabled') 57 ksft_eq(napi.get('pid'), None) 58 59 cmd(f"ethtool -L {cfg.ifname} combined 1") 60 _set_threaded_state(cfg, 1) 61 cmd(f"ethtool -L {cfg.ifname} combined {qcnt}") 62 63 napis = nl.napi_get({'ifindex': cfg.ifindex}, dump=True) 64 for napi in napis: 65 ksft_eq(napi['threaded'], 'enabled') 66 ksft_ne(napi.get('pid'), None) 67 68 69def enable_dev_threaded_disable_napi_threaded(cfg, nl) -> None: 70 """ 71 Test that when napi threaded is enabled at device level and 72 then disabled at napi level for one napi, the threaded state 73 of all napis is preserved after a change in number of queues. 74 """ 75 76 napis = nl.napi_get({'ifindex': cfg.ifindex}, dump=True) 77 ksft_ge(len(napis), 2) 78 79 napi0_id = napis[0]['id'] 80 napi1_id = napis[1]['id'] 81 82 qcnt = _setup_deferred_cleanup(cfg) 83 84 # set threaded 85 _set_threaded_state(cfg, 1) 86 87 # check napi threaded is set for both napis 88 _assert_napi_threaded_enabled(nl, napi0_id) 89 _assert_napi_threaded_enabled(nl, napi1_id) 90 91 # disable threaded for napi1 92 nl.napi_set({'id': napi1_id, 'threaded': 'disabled'}) 93 94 cmd(f"ethtool -L {cfg.ifname} combined 1") 95 cmd(f"ethtool -L {cfg.ifname} combined {qcnt}") 96 _assert_napi_threaded_enabled(nl, napi0_id) 97 _assert_napi_threaded_disabled(nl, napi1_id) 98 99 100def change_num_queues(cfg, nl) -> None: 101 """ 102 Test that when napi threaded is enabled at device level, 103 the napi threaded state is preserved after a change in 104 number of queues. 105 """ 106 107 napis = nl.napi_get({'ifindex': cfg.ifindex}, dump=True) 108 ksft_ge(len(napis), 2) 109 110 napi0_id = napis[0]['id'] 111 napi1_id = napis[1]['id'] 112 113 qcnt = _setup_deferred_cleanup(cfg) 114 115 # set threaded 116 _set_threaded_state(cfg, 1) 117 118 # check napi threaded is set for both napis 119 _assert_napi_threaded_enabled(nl, napi0_id) 120 _assert_napi_threaded_enabled(nl, napi1_id) 121 122 cmd(f"ethtool -L {cfg.ifname} combined 1") 123 cmd(f"ethtool -L {cfg.ifname} combined {qcnt}") 124 125 # check napi threaded is set for both napis 126 _assert_napi_threaded_enabled(nl, napi0_id) 127 _assert_napi_threaded_enabled(nl, napi1_id) 128 129 130def main() -> None: 131 """ Ksft boiler plate main """ 132 133 with NetDrvEnv(__file__, queue_count=2) as cfg: 134 ksft_run([napi_init, 135 change_num_queues, 136 enable_dev_threaded_disable_napi_threaded], 137 args=(cfg, NetdevFamily())) 138 ksft_exit() 139 140 141if __name__ == "__main__": 142 main() 143