wifi: mt76: testmode: add nla_policy for MT76_TM_ATTR_TX_LENGTH

It seems that the nla_policy in mt76_tm_policy is missed for attribute
MT76_TM_ATTR_TX_LENGTH. This patch adds the correct description to make
sure the

  u32 val = nla_get_u32(tb[MT76_TM_ATTR_TX_LENGTH]);

in function mt76_testmode_cmd() is safe and will not result in
out-of-attribute read.

Fixes: f0efa8621550 ("mt76: add API for testmode support")
Signed-off-by: Lin Ma <linma@zju.edu.cn>
Signed-off-by: Felix Fietkau <nbd@nbd.name>

diff --git a/drivers/net/wireless/mediatek/mt76/testmode.c b/drivers/net/wireless/mediatek/mt76/testmode.c
index 0accc71..4644dac 100644 (file)
--- a/drivers/net/wireless/mediatek/mt76/testmode.c
+++ b/drivers/net/wireless/mediatek/mt76/testmode.c
@@ -8,6 +8,7 @@ const struct nla_policy mt76_tm_policy[NUM_MT76_TM_ATTRS] = {
        [MT76_TM_ATTR_RESET] = { .type = NLA_FLAG },
        [MT76_TM_ATTR_STATE] = { .type = NLA_U8 },
        [MT76_TM_ATTR_TX_COUNT] = { .type = NLA_U32 },
+       [MT76_TM_ATTR_TX_LENGTH] = { .type = NLA_U32 },
        [MT76_TM_ATTR_TX_RATE_MODE] = { .type = NLA_U8 },
        [MT76_TM_ATTR_TX_RATE_NSS] = { .type = NLA_U8 },
        [MT76_TM_ATTR_TX_RATE_IDX] = { .type = NLA_U8 },