diff options
author | Phil Sutter <phil@nwl.cc> | 2020-03-20 10:06:16 +0100 |
---|---|---|
committer | Phil Sutter <phil@nwl.cc> | 2020-05-11 14:28:28 +0200 |
commit | 0e1b72a6d7b6c1c13f5603a916d3847fbe1a3ee3 (patch) | |
tree | e227cd15c7a5654fb831604228dddedfe1c293e6 /iptables/nft-cache.c | |
parent | 13b351c9ba01751c693df68f657d19e8c3d267ce (diff) |
nft-cache: Introduce __fetch_chain_cache()
Extract the inner part of fetch_chain_cache() into a dedicated function,
preparing for individual chain caching.
Signed-off-by: Phil Sutter <phil@nwl.cc>
Diffstat (limited to 'iptables/nft-cache.c')
-rw-r--r-- | iptables/nft-cache.c | 50 |
1 files changed, 30 insertions, 20 deletions
diff --git a/iptables/nft-cache.c b/iptables/nft-cache.c index 61184653..a0c4dfc6 100644 --- a/iptables/nft-cache.c +++ b/iptables/nft-cache.c @@ -322,9 +322,9 @@ static int fetch_set_cache(struct nft_handle *h, return ret; } -static int fetch_chain_cache(struct nft_handle *h, - const struct builtin_table *t, - const char *chain) +static int __fetch_chain_cache(struct nft_handle *h, + const struct builtin_table *t, + const struct nftnl_chain *c) { struct nftnl_chain_list_cb_data d = { .h = h, @@ -334,24 +334,10 @@ static int fetch_chain_cache(struct nft_handle *h, struct nlmsghdr *nlh; int ret; - if (t && chain) { - struct nftnl_chain *c = nftnl_chain_alloc(); - - if (!c) - return -1; - - nlh = nftnl_chain_nlmsg_build_hdr(buf, NFT_MSG_GETCHAIN, - h->family, NLM_F_ACK, - h->seq); - nftnl_chain_set_str(c, NFTNL_CHAIN_TABLE, t->name); - nftnl_chain_set_str(c, NFTNL_CHAIN_NAME, chain); + nlh = nftnl_chain_nlmsg_build_hdr(buf, NFT_MSG_GETCHAIN, h->family, + c ? NLM_F_ACK : NLM_F_DUMP, h->seq); + if (c) nftnl_chain_nlmsg_build_payload(nlh, c); - nftnl_chain_free(c); - } else { - nlh = nftnl_chain_nlmsg_build_hdr(buf, NFT_MSG_GETCHAIN, - h->family, NLM_F_DUMP, - h->seq); - } ret = mnl_talk(h, nlh, nftnl_chain_list_cb, &d); if (ret < 0 && errno == EINTR) @@ -360,6 +346,30 @@ static int fetch_chain_cache(struct nft_handle *h, return ret; } +static int fetch_chain_cache(struct nft_handle *h, + const struct builtin_table *t, + const char *chain) +{ + struct nftnl_chain *c; + int ret; + + if (!chain) + return __fetch_chain_cache(h, t, NULL); + + assert(t); + + c = nftnl_chain_alloc(); + if (!c) + return -1; + + nftnl_chain_set_str(c, NFTNL_CHAIN_TABLE, t->name); + nftnl_chain_set_str(c, NFTNL_CHAIN_NAME, chain); + ret = __fetch_chain_cache(h, t, c); + + nftnl_chain_free(c); + return ret; +} + static int nftnl_rule_list_cb(const struct nlmsghdr *nlh, void *data) { struct nftnl_chain *c = data; |