From 7c43dc46179a6f85406eb861e970cbb06bac05d2 Mon Sep 17 00:00:00 2001 From: Phil Sutter Date: Tue, 22 Aug 2017 14:40:04 +0200 Subject: src: Keep cache in struct nft_ctx This is preliminary work for Eric's libnftables patchset. Cc: Eric Leblond Signed-off-by: Phil Sutter --- src/main.c | 24 +++++++++++------------- 1 file changed, 11 insertions(+), 13 deletions(-) (limited to 'src/main.c') diff --git a/src/main.c b/src/main.c index 88839598..b86ae62f 100644 --- a/src/main.c +++ b/src/main.c @@ -187,7 +187,7 @@ static const struct input_descriptor indesc_cmdline = { .name = "", }; -static int nft_netlink(struct nft_ctx *nft, struct nft_cache *cache, +static int nft_netlink(struct nft_ctx *nft, struct parser_state *state, struct list_head *msgs, struct mnl_socket *nf_sock) { @@ -211,7 +211,7 @@ static int nft_netlink(struct nft_ctx *nft, struct nft_cache *cache, ctx.batch_supported = batch_supported; ctx.octx = &nft->output; ctx.nf_sock = nf_sock; - ctx.cache = cache; + ctx.cache = &nft->cache; init_list_head(&ctx.list); ret = do_command(&ctx, cmd); if (ret < 0) @@ -247,7 +247,7 @@ out: } int nft_run(struct nft_ctx *nft, struct mnl_socket *nf_sock, - struct nft_cache *cache, void *scanner, struct parser_state *state, + void *scanner, struct parser_state *state, struct list_head *msgs) { struct cmd *cmd, *next; @@ -262,7 +262,7 @@ int nft_run(struct nft_ctx *nft, struct mnl_socket *nf_sock, list_for_each_entry(cmd, &state->cmds, list) nft_cmd_expand(cmd); - ret = nft_netlink(nft, cache, state, msgs, nf_sock); + ret = nft_netlink(nft, state, msgs, nf_sock); err1: list_for_each_entry_safe(cmd, next, &state->cmds, list) { list_del(&cmd->list); @@ -297,7 +297,6 @@ void nft_exit(void) int main(int argc, char * const *argv) { struct parser_state state; - struct nft_cache cache; void *scanner; LIST_HEAD(msgs); char *buf = NULL, *filename = NULL; @@ -306,8 +305,7 @@ int main(int argc, char * const *argv) int i, val, rc = NFT_EXIT_SUCCESS; struct mnl_socket *nf_sock; - memset(&cache, 0, sizeof(cache)); - init_list_head(&cache.list); + init_list_head(&nft.cache.list); nft_init(); nf_sock = netlink_open_sock(); @@ -407,20 +405,20 @@ int main(int argc, char * const *argv) strcat(buf, " "); } strcat(buf, "\n"); - parser_init(nf_sock, &cache, &state, &msgs); + parser_init(nf_sock, &nft.cache, &state, &msgs); scanner = scanner_init(&state); scanner_push_buffer(scanner, &indesc_cmdline, buf); } else if (filename != NULL) { - rc = cache_update(nf_sock, &cache, CMD_INVALID, &msgs); + rc = cache_update(nf_sock, &nft.cache, CMD_INVALID, &msgs); if (rc < 0) return rc; - parser_init(nf_sock, &cache, &state, &msgs); + parser_init(nf_sock, &nft.cache, &state, &msgs); scanner = scanner_init(&state); if (scanner_read_file(scanner, filename, &internal_location) < 0) goto out; } else if (interactive) { - if (cli_init(&nft, nf_sock, &cache, &state) < 0) { + if (cli_init(&nft, nf_sock, &state) < 0) { fprintf(stderr, "%s: interactive CLI not supported in this build\n", argv[0]); exit(NFT_EXIT_FAILURE); @@ -431,13 +429,13 @@ int main(int argc, char * const *argv) exit(NFT_EXIT_FAILURE); } - if (nft_run(&nft, nf_sock, &cache, scanner, &state, &msgs) != 0) + if (nft_run(&nft, nf_sock, scanner, &state, &msgs) != 0) rc = NFT_EXIT_FAILURE; out: scanner_destroy(scanner); erec_print_list(stderr, &msgs); xfree(buf); - cache_release(&cache); + cache_release(&nft.cache); iface_cache_release(); netlink_close_sock(nf_sock); nft_exit(); -- cgit v1.2.3