Instead of carrying the family value, carry the handle (which contains
the family value) and relieve expression parsers from having to call
nft_family_ops_lookup().

Signed-off-by: Phil Sutter <p...@nwl.cc>
---
 iptables/nft-shared.c | 40 ++++++++++++++--------------------------
 iptables/nft-shared.h |  2 +-
 2 files changed, 15 insertions(+), 27 deletions(-)

diff --git a/iptables/nft-shared.c b/iptables/nft-shared.c
index bdbd3238b2890..80d4e1fcdcea1 100644
--- a/iptables/nft-shared.c
+++ b/iptables/nft-shared.c
@@ -310,7 +310,6 @@ static void nft_parse_target(struct nft_xt_ctx *ctx, struct 
nftnl_expr *e)
        struct xtables_target *target;
        struct xt_entry_target *t;
        size_t size;
-       struct nft_family_ops *ops;
        void *data = ctx->cs;
 
        target = xtables_find_target(targname, XTF_TRY_LOAD);
@@ -327,8 +326,7 @@ static void nft_parse_target(struct nft_xt_ctx *ctx, struct 
nftnl_expr *e)
 
        target->t = t;
 
-       ops = nft_family_ops_lookup(ctx->family);
-       ops->parse_target(target, data);
+       ctx->h->ops->parse_target(target, data);
 }
 
 static void nft_parse_match(struct nft_xt_ctx *ctx, struct nftnl_expr *e)
@@ -339,9 +337,8 @@ static void nft_parse_match(struct nft_xt_ctx *ctx, struct 
nftnl_expr *e)
        struct xtables_match *match;
        struct xtables_rule_match **matches;
        struct xt_entry_match *m;
-       struct nft_family_ops *ops;
 
-       switch (ctx->family) {
+       switch (ctx->h->family) {
        case NFPROTO_IPV4:
        case NFPROTO_IPV6:
        case NFPROTO_BRIDGE:
@@ -349,7 +346,7 @@ static void nft_parse_match(struct nft_xt_ctx *ctx, struct 
nftnl_expr *e)
                break;
        default:
                fprintf(stderr, "BUG: nft_parse_match() unknown family %d\n",
-                       ctx->family);
+                       ctx->h->family);
                exit(EXIT_FAILURE);
        }
 
@@ -365,9 +362,8 @@ static void nft_parse_match(struct nft_xt_ctx *ctx, struct 
nftnl_expr *e)
 
        match->m = m;
 
-       ops = nft_family_ops_lookup(ctx->family);
-       if (ops->parse_match != NULL)
-               ops->parse_match(match, ctx->cs);
+       if (ctx->h->ops->parse_match != NULL)
+               ctx->h->ops->parse_match(match, ctx->cs);
 }
 
 void print_proto(uint16_t proto, int invert)
@@ -400,7 +396,6 @@ void get_cmp_data(struct nftnl_expr *e, void *data, size_t 
dlen, bool *inv)
 
 static void nft_meta_set_to_target(struct nft_xt_ctx *ctx)
 {
-       const struct nft_family_ops *ops;
        struct xtables_target *target;
        struct xt_entry_target *t;
        unsigned int size;
@@ -429,8 +424,7 @@ static void nft_meta_set_to_target(struct nft_xt_ctx *ctx)
 
        target->t = t;
 
-       ops = nft_family_ops_lookup(ctx->family);
-       ops->parse_target(target, ctx->cs);
+       ctx->h->ops->parse_target(target, ctx->cs);
 }
 
 static void nft_parse_meta(struct nft_xt_ctx *ctx, struct nftnl_expr *e)
@@ -474,7 +468,6 @@ static void nft_parse_bitwise(struct nft_xt_ctx *ctx, 
struct nftnl_expr *e)
 
 static void nft_parse_cmp(struct nft_xt_ctx *ctx, struct nftnl_expr *e)
 {
-       struct nft_family_ops *ops = nft_family_ops_lookup(ctx->family);
        void *data = ctx->cs;
        uint32_t reg;
 
@@ -483,12 +476,12 @@ static void nft_parse_cmp(struct nft_xt_ctx *ctx, struct 
nftnl_expr *e)
                return;
 
        if (ctx->flags & NFT_XT_CTX_META) {
-               ops->parse_meta(ctx, e, data);
+               ctx->h->ops->parse_meta(ctx, e, data);
                ctx->flags &= ~NFT_XT_CTX_META;
        }
        /* bitwise context is interpreted from payload */
        if (ctx->flags & NFT_XT_CTX_PAYLOAD) {
-               ops->parse_payload(ctx, e, data);
+               ctx->h->ops->parse_payload(ctx, e, data);
                ctx->flags &= ~NFT_XT_CTX_PAYLOAD;
        }
 }
@@ -502,7 +495,6 @@ static void nft_parse_counter(struct nftnl_expr *e, struct 
xt_counters *counters
 static void nft_parse_immediate(struct nft_xt_ctx *ctx, struct nftnl_expr *e)
 {
        const char *chain = nftnl_expr_get_str(e, NFTNL_EXPR_IMM_CHAIN);
-       struct nft_family_ops *ops;
        const char *jumpto = NULL;
        bool nft_goto = false;
        void *data = ctx->cs;
@@ -544,8 +536,7 @@ static void nft_parse_immediate(struct nft_xt_ctx *ctx, 
struct nftnl_expr *e)
                break;
        }
 
-       ops = nft_family_ops_lookup(ctx->family);
-       ops->parse_immediate(jumpto, nft_goto, data);
+       ctx->h->ops->parse_immediate(jumpto, nft_goto, data);
 }
 
 static void nft_parse_limit(struct nft_xt_ctx *ctx, struct nftnl_expr *e)
@@ -555,11 +546,10 @@ static void nft_parse_limit(struct nft_xt_ctx *ctx, 
struct nftnl_expr *e)
        __u64 rate = nftnl_expr_get_u64(e, NFTNL_EXPR_LIMIT_RATE);
        struct xtables_rule_match **matches;
        struct xtables_match *match;
-       struct nft_family_ops *ops;
        struct xt_rateinfo *rinfo;
        size_t size;
 
-       switch (ctx->family) {
+       switch (ctx->h->family) {
        case NFPROTO_IPV4:
        case NFPROTO_IPV6:
        case NFPROTO_BRIDGE:
@@ -567,7 +557,7 @@ static void nft_parse_limit(struct nft_xt_ctx *ctx, struct 
nftnl_expr *e)
                break;
        default:
                fprintf(stderr, "BUG: nft_parse_limit() unknown family %d\n",
-                       ctx->family);
+                       ctx->h->family);
                exit(EXIT_FAILURE);
        }
 
@@ -586,9 +576,8 @@ static void nft_parse_limit(struct nft_xt_ctx *ctx, struct 
nftnl_expr *e)
        rinfo->avg = XT_LIMIT_SCALE * unit / rate;
        rinfo->burst = burst;
 
-       ops = nft_family_ops_lookup(ctx->family);
-       if (ops->parse_match != NULL)
-               ops->parse_match(match, ctx->cs);
+       if (ctx->h->ops->parse_match != NULL)
+               ctx->h->ops->parse_match(match, ctx->cs);
 }
 
 void nft_rule_to_iptables_command_state(struct nft_handle *h,
@@ -597,10 +586,9 @@ void nft_rule_to_iptables_command_state(struct nft_handle 
*h,
 {
        struct nftnl_expr_iter *iter;
        struct nftnl_expr *expr;
-       int family = nftnl_rule_get_u32(r, NFTNL_RULE_FAMILY);
        struct nft_xt_ctx ctx = {
                .cs = cs,
-               .family = family,
+               .h = h,
        };
 
        iter = nftnl_expr_iter_create(r);
diff --git a/iptables/nft-shared.h b/iptables/nft-shared.h
index 947a4eb00c4d4..efc40e7714e0f 100644
--- a/iptables/nft-shared.h
+++ b/iptables/nft-shared.h
@@ -48,7 +48,7 @@ enum {
 struct nft_xt_ctx {
        struct iptables_command_state *cs;
        struct nftnl_expr_iter *iter;
-       int family;
+       struct nft_handle *h;
        uint32_t flags;
 
        uint32_t reg;
-- 
2.23.0

Reply via email to