The default RETA size is not the same for all drivers. In some drivers (mlx5), the RETA size may also be dependent on the number of RX queues.
Introduce a new DriverInfo abstraction for known keys. Define a simple API to expose the RSS key and RETA size (based on the number of RX queues). Use that abstraction for all three known keys. Signed-off-by: Robin Jarry <rja...@redhat.com> --- usertools/dpdk-rss-flows.py | 144 ++++++++++++++++++++++-------------- 1 file changed, 89 insertions(+), 55 deletions(-) diff --git a/usertools/dpdk-rss-flows.py b/usertools/dpdk-rss-flows.py index 73821eb47125..dfdad33dde8a 100755 --- a/usertools/dpdk-rss-flows.py +++ b/usertools/dpdk-rss-flows.py @@ -154,60 +154,81 @@ def balanced_traffic( NO_PORT = (0,) -# fmt: off -# rss_intel_key, see drivers/net/ixgbe/ixgbe_rxtx.c -RSS_KEY_INTEL = bytes( - ( - 0x6d, 0x5a, 0x56, 0xda, 0x25, 0x5b, 0x0e, 0xc2, - 0x41, 0x67, 0x25, 0x3d, 0x43, 0xa3, 0x8f, 0xb0, - 0xd0, 0xca, 0x2b, 0xcb, 0xae, 0x7b, 0x30, 0xb4, - 0x77, 0xcb, 0x2d, 0xa3, 0x80, 0x30, 0xf2, 0x0c, - 0x6a, 0x42, 0xb7, 0x3b, 0xbe, 0xac, 0x01, 0xfa, - ) -) -# rss_hash_default_key, see drivers/net/mlx5/mlx5_rxq.c -RSS_KEY_MLX = bytes( - ( - 0x2c, 0xc6, 0x81, 0xd1, 0x5b, 0xdb, 0xf4, 0xf7, - 0xfc, 0xa2, 0x83, 0x19, 0xdb, 0x1a, 0x3e, 0x94, - 0x6b, 0x9e, 0x38, 0xd9, 0x2c, 0x9c, 0x03, 0xd1, - 0xad, 0x99, 0x44, 0xa7, 0xd9, 0x56, 0x3d, 0x59, - 0x06, 0x3c, 0x25, 0xf3, 0xfc, 0x1f, 0xdc, 0x2a, - ) -) -# rss_key_default, see drivers/net/i40e/i40e_ethdev.c -# i40e is the only driver that takes 52 bytes keys -RSS_KEY_I40E = bytes( - ( - 0x44, 0x39, 0x79, 0x6b, 0xb5, 0x4c, 0x50, 0x23, - 0xb6, 0x75, 0xea, 0x5b, 0x12, 0x4f, 0x9f, 0x30, - 0xb8, 0xa2, 0xc0, 0x3d, 0xdf, 0xdc, 0x4d, 0x02, - 0xa0, 0x8c, 0x9b, 0x33, 0x4a, 0xf6, 0x4a, 0x4c, - 0x05, 0xc6, 0xfa, 0x34, 0x39, 0x58, 0xd8, 0x55, - 0x7d, 0x99, 0x58, 0x3a, 0xe1, 0x38, 0xc9, 0x2e, - 0x81, 0x15, 0x03, 0x66, - ) -) -# fmt: on -DEFAULT_DRIVER_KEYS = { - "intel": RSS_KEY_INTEL, - "mlx": RSS_KEY_MLX, - "i40e": RSS_KEY_I40E, + +class DriverInfo: + def __init__(self, key: bytes = None, reta_size: int = None): + self.__key = key + self.__reta_size = reta_size + + def rss_key(self) -> bytes: + return self.__key + + def reta_size(self, num_queues: int) -> int: + return self.__reta_size + + +class MlxDriverInfo(DriverInfo): + def rss_key(self) -> bytes: + return bytes( + ( + # fmt: off + # rss_hash_default_key, see drivers/net/mlx5/mlx5_rxq.c + 0x2c, 0xc6, 0x81, 0xd1, 0x5b, 0xdb, 0xf4, 0xf7, + 0xfc, 0xa2, 0x83, 0x19, 0xdb, 0x1a, 0x3e, 0x94, + 0x6b, 0x9e, 0x38, 0xd9, 0x2c, 0x9c, 0x03, 0xd1, + 0xad, 0x99, 0x44, 0xa7, 0xd9, 0x56, 0x3d, 0x59, + 0x06, 0x3c, 0x25, 0xf3, 0xfc, 0x1f, 0xdc, 0x2a, + # fmt: on + ) + ) + + def reta_size(self, num_queues: int) -> int: + if num_queues & (num_queues - 1) == 0: + # If the requested number of RX queues is power of two, + # use a table of this size. + return num_queues + # otherwise, use the maximum table size + return 512 + + +DEFAULT_DRIVERS = { + "intel": DriverInfo( + key=bytes( + ( + # fmt: off + # rss_intel_key, see drivers/net/ixgbe/ixgbe_rxtx.c + 0x6d, 0x5a, 0x56, 0xda, 0x25, 0x5b, 0x0e, 0xc2, + 0x41, 0x67, 0x25, 0x3d, 0x43, 0xa3, 0x8f, 0xb0, + 0xd0, 0xca, 0x2b, 0xcb, 0xae, 0x7b, 0x30, 0xb4, + 0x77, 0xcb, 0x2d, 0xa3, 0x80, 0x30, 0xf2, 0x0c, + 0x6a, 0x42, 0xb7, 0x3b, 0xbe, 0xac, 0x01, 0xfa, + # fmt: on + ) + ), + reta_size=128, + ), + "mlx": MlxDriverInfo(), + "i40e": DriverInfo( + key=bytes( + ( + # fmt: off + # rss_key_default, see drivers/net/i40e/i40e_ethdev.c + # i40e is the only driver that takes 52 bytes keys + 0x44, 0x39, 0x79, 0x6b, 0xb5, 0x4c, 0x50, 0x23, + 0xb6, 0x75, 0xea, 0x5b, 0x12, 0x4f, 0x9f, 0x30, + 0xb8, 0xa2, 0xc0, 0x3d, 0xdf, 0xdc, 0x4d, 0x02, + 0xa0, 0x8c, 0x9b, 0x33, 0x4a, 0xf6, 0x4a, 0x4c, + 0x05, 0xc6, 0xfa, 0x34, 0x39, 0x58, 0xd8, 0x55, + 0x7d, 0x99, 0x58, 0x3a, 0xe1, 0x38, 0xc9, 0x2e, + 0x81, 0x15, 0x03, 0x66, + # fmt: on + ) + ), + reta_size=512, + ), } -def rss_key(value): - if value in DEFAULT_DRIVER_KEYS: - return DEFAULT_DRIVER_KEYS[value] - try: - key = binascii.unhexlify(value) - if len(key) not in (40, 52): - raise argparse.ArgumentTypeError("The key must be 40 or 52 bytes long") - return key - except (TypeError, ValueError) as e: - raise argparse.ArgumentTypeError(str(e)) from e - - def port_range(value): try: if "-" in value: @@ -296,8 +317,7 @@ def parse_args(): parser.add_argument( "-k", "--rss-key", - default=RSS_KEY_INTEL, - type=rss_key, + default="intel", help=""" The random 40-bytes key used to compute the RSS hash. This option supports either a well-known name or the hex value of the key @@ -307,10 +327,9 @@ def parse_args(): parser.add_argument( "-t", "--reta-size", - default=128, type=power_of_two, help=""" - Size of the redirection table or "RETA" (default: 128). + Size of the redirection table or "RETA" (default: depends on driver). """, ) parser.add_argument( @@ -339,9 +358,24 @@ def parse_args(): parser.error( f"{args.ip_src} and {args.ip_dst} don't have the same protocol version" ) + + if args.rss_key in DEFAULT_DRIVERS: + driver_info = DEFAULT_DRIVERS[args.rss_key] + else: + try: + key = binascii.unhexlify(args.rss_key) + except (TypeError, ValueError) as e: + parser.error(f"RSS_KEY: {e}") + driver_info = DriverInfo(key=key, reta_size=128) + + if args.reta_size is None: + args.reta_size = driver_info.reta_size(args.rx_queues) + if args.reta_size < args.rx_queues: parser.error("RETA_SIZE must be greater than or equal to RX_QUEUES") + args.rss_key = driver_info.rss_key() + return args -- 2.41.0