[PATCH net-next 2/3] nexthop: Use a dedicated policy for nh_valid_dump_req()

[Petr Machata]

This function uses the global nexthop policy, but only accepts four
particular attributes. Create a new policy that only includes the four
supported attributes, and use it. Convert the loop to a series of ifs.

Signed-off-by: Petr Machata <pe...@nvidia.com>
Reviewed-by: Ido Schimmel <ido...@nvidia.com>
---
 net/ipv4/nexthop.c | 57 +++++++++++++++++++++-------------------------
 1 file changed, 26 insertions(+), 31 deletions(-)

diff --git a/net/ipv4/nexthop.c b/net/ipv4/nexthop.c
index d5d88f7c5c11..226d73cbc468 100644
--- a/net/ipv4/nexthop.c
+++ b/net/ipv4/nexthop.c
@@ -40,6 +40,13 @@ static const struct nla_policy rtm_nh_policy_get[NHA_MAX + 
1] = {
        [NHA_ID]                = { .type = NLA_U32 },
 };
 
+static const struct nla_policy rtm_nh_policy_dump[NHA_MAX + 1] = {
+       [NHA_OIF]               = { .type = NLA_U32 },
+       [NHA_GROUPS]            = { .type = NLA_FLAG },
+       [NHA_MASTER]            = { .type = NLA_U32 },
+       [NHA_FDB]               = { .type = NLA_FLAG },
+};
+
 static bool nexthop_notifiers_is_empty(struct net *net)
 {
        return !net->nexthop.notifier_chain.head;
@@ -1984,46 +1991,34 @@ static int nh_valid_dump_req(const struct nlmsghdr 
*nlh, int *dev_idx,
        struct netlink_ext_ack *extack = cb->extack;
        struct nlattr *tb[NHA_MAX + 1];
        struct nhmsg *nhm;
-       int err, i;
+       int err;
        u32 idx;
 
-       err = nlmsg_parse(nlh, sizeof(*nhm), tb, NHA_MAX, rtm_nh_policy,
+       err = nlmsg_parse(nlh, sizeof(*nhm), tb, NHA_MAX, rtm_nh_policy_dump,
                          NULL);
        if (err < 0)
                return err;
 
-       for (i = 0; i <= NHA_MAX; ++i) {
-               if (!tb[i])
-                       continue;
-
-               switch (i) {
-               case NHA_OIF:
-                       idx = nla_get_u32(tb[i]);
-                       if (idx > INT_MAX) {
-                               NL_SET_ERR_MSG(extack, "Invalid device index");
-                               return -EINVAL;
-                       }
-                       *dev_idx = idx;
-                       break;
-               case NHA_MASTER:
-                       idx = nla_get_u32(tb[i]);
-                       if (idx > INT_MAX) {
-                               NL_SET_ERR_MSG(extack, "Invalid master device 
index");
-                               return -EINVAL;
-                       }
-                       *master_idx = idx;
-                       break;
-               case NHA_GROUPS:
-                       *group_filter = true;
-                       break;
-               case NHA_FDB:
-                       *fdb_filter = true;
-                       break;
-               default:
-                       NL_SET_ERR_MSG(extack, "Unsupported attribute in dump 
request");
+       if (tb[NHA_OIF]) {
+               idx = nla_get_u32(tb[NHA_OIF]);
+               if (idx > INT_MAX) {
+                       NL_SET_ERR_MSG(extack, "Invalid device index");
+                       return -EINVAL;
+               }
+               *dev_idx = idx;
+       }
+       if (tb[NHA_MASTER]) {
+               idx = nla_get_u32(tb[NHA_MASTER]);
+               if (idx > INT_MAX) {
+                       NL_SET_ERR_MSG(extack, "Invalid master device index");
                        return -EINVAL;
                }
+               *master_idx = idx;
        }
+       if (tb[NHA_GROUPS])
+               *group_filter = true;
+       if (tb[NHA_FDB])
+               *fdb_filter = true;
 
        nhm = nlmsg_data(nlh);
        if (nhm->nh_protocol || nhm->resvd || nhm->nh_scope || nhm->nh_flags) {
-- 
2.26.2