summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--include/nftables.h3
-rw-r--r--src/main.c26
2 files changed, 14 insertions, 15 deletions
diff --git a/include/nftables.h b/include/nftables.h
index 256b06ee..5035567a 100644
--- a/include/nftables.h
+++ b/include/nftables.h
@@ -38,7 +38,10 @@ struct nft_cache {
uint32_t seqnum;
};
+struct mnl_socket;
+
struct nft_ctx {
+ struct mnl_socket *nf_sock;
const char *include_paths[INCLUDE_PATHS_MAX];
unsigned int num_include_paths;
unsigned int parser_max_errors;
diff --git a/src/main.c b/src/main.c
index eecd430a..a891832e 100644
--- a/src/main.c
+++ b/src/main.c
@@ -305,7 +305,6 @@ static void nft_ctx_free(const struct nft_ctx *ctx)
}
static int nft_run_cmd_from_buffer(struct nft_ctx *nft,
- struct mnl_socket *nf_sock,
char *buf, size_t buflen)
{
int rc = NFT_EXIT_SUCCESS;
@@ -313,11 +312,11 @@ static int nft_run_cmd_from_buffer(struct nft_ctx *nft,
LIST_HEAD(msgs);
void *scanner;
- parser_init(nf_sock, &nft->cache, &state, &msgs, nft->debug_mask);
+ parser_init(nft->nf_sock, &nft->cache, &state, &msgs, nft->debug_mask);
scanner = scanner_init(&state);
scanner_push_buffer(scanner, &indesc_cmdline, buf);
- if (nft_run(nft, nf_sock, scanner, &state, &msgs) != 0)
+ if (nft_run(nft, nft->nf_sock, scanner, &state, &msgs) != 0)
rc = NFT_EXIT_FAILURE;
erec_print_list(stderr, &msgs, nft->debug_mask);
@@ -326,26 +325,24 @@ static int nft_run_cmd_from_buffer(struct nft_ctx *nft,
return rc;
}
-static int nft_run_cmd_from_filename(struct nft_ctx *nft,
- struct mnl_socket *nf_sock,
- const char *filename)
+static int nft_run_cmd_from_filename(struct nft_ctx *nft, const char *filename)
{
struct parser_state state;
LIST_HEAD(msgs);
void *scanner;
int rc;
- rc = cache_update(nf_sock, &nft->cache, CMD_INVALID, &msgs,
+ rc = cache_update(nft->nf_sock, &nft->cache, CMD_INVALID, &msgs,
nft->debug_mask);
if (rc < 0)
return NFT_EXIT_FAILURE;
- parser_init(nf_sock, &nft->cache, &state, &msgs, nft->debug_mask);
+ parser_init(nft->nf_sock, &nft->cache, &state, &msgs, nft->debug_mask);
scanner = scanner_init(&state);
if (scanner_read_file(scanner, filename, &internal_location) < 0)
goto err;
- if (nft_run(nft, nf_sock, scanner, &state, &msgs) != 0)
+ if (nft_run(nft, nft->nf_sock, scanner, &state, &msgs) != 0)
rc = NFT_EXIT_FAILURE;
err:
erec_print_list(stderr, &msgs, nft->debug_mask);
@@ -359,13 +356,12 @@ int main(int argc, char * const *argv)
char *buf = NULL, *filename = NULL;
unsigned int len;
bool interactive = false;
- struct mnl_socket *nf_sock;
struct parser_state state;
int i, val, rc;
nft = nft_ctx_new();
- nf_sock = netlink_open_sock();
+ nft->nf_sock = netlink_open_sock();
while (1) {
val = getopt_long(argc, argv, OPTSTRING, options, NULL);
if (val == -1)
@@ -460,11 +456,11 @@ int main(int argc, char * const *argv)
strcat(buf, " ");
}
strcat(buf, "\n");
- rc = nft_run_cmd_from_buffer(nft, nf_sock, buf, len + 2);
+ rc = nft_run_cmd_from_buffer(nft, buf, len + 2);
} else if (filename != NULL) {
- rc = nft_run_cmd_from_filename(nft, nf_sock, filename);
+ rc = nft_run_cmd_from_filename(nft, filename);
} else if (interactive) {
- if (cli_init(nft, nf_sock, &state) < 0) {
+ if (cli_init(nft, nft->nf_sock, &state) < 0) {
fprintf(stderr, "%s: interactive CLI not supported in this build\n",
argv[0]);
exit(NFT_EXIT_FAILURE);
@@ -476,7 +472,7 @@ int main(int argc, char * const *argv)
}
xfree(buf);
- netlink_close_sock(nf_sock);
+ netlink_close_sock(nft->nf_sock);
nft_ctx_free(nft);
return rc;