diff options
author | Pablo Neira Ayuso <pablo@netfilter.org> | 2018-10-23 18:24:31 +0200 |
---|---|---|
committer | Pablo Neira Ayuso <pablo@netfilter.org> | 2018-10-23 18:26:05 +0200 |
commit | f1c2ace2631fa691b81a181bbbf2a3b29b0e69f6 (patch) | |
tree | 50c444c0643c68233a6bc3732f44d1b98e7d4cdb /src | |
parent | b83dd8ff56ff9f3170ebec52aeebb7a3a62c9dc7 (diff) |
src: move socket open and reopen to mnl.c
These functions are part of the mnl backend, move them there. Remove
netlink_close_sock(), use direct call to mnl_socket_close().
Signed-off-by: Pablo Neira Ayuso <pablo@netfilter.org>
Diffstat (limited to 'src')
-rw-r--r-- | src/libnftables.c | 4 | ||||
-rw-r--r-- | src/mnl.c | 22 | ||||
-rw-r--r-- | src/netlink.c | 27 | ||||
-rw-r--r-- | src/rule.c | 2 |
4 files changed, 25 insertions, 30 deletions
diff --git a/src/libnftables.c b/src/libnftables.c index 44869602..0731c532 100644 --- a/src/libnftables.c +++ b/src/libnftables.c @@ -129,7 +129,7 @@ void nft_ctx_clear_include_paths(struct nft_ctx *ctx) static void nft_ctx_netlink_init(struct nft_ctx *ctx) { - ctx->nf_sock = netlink_open_sock(); + ctx->nf_sock = nft_mnl_socket_open(); } struct nft_ctx *nft_ctx_new(uint32_t flags) @@ -266,7 +266,7 @@ const char *nft_ctx_get_error_buffer(struct nft_ctx *ctx) void nft_ctx_free(struct nft_ctx *ctx) { if (ctx->nf_sock) - netlink_close_sock(ctx->nf_sock); + mnl_socket_close(ctx->nf_sock); exit_cookie(&ctx->output.output_cookie); exit_cookie(&ctx->output.error_cookie); @@ -28,10 +28,32 @@ #include <string.h> #include <sys/socket.h> #include <arpa/inet.h> +#include <fcntl.h> #include <errno.h> #include <utils.h> #include <nftables.h> +struct mnl_socket *nft_mnl_socket_open(void) +{ + struct mnl_socket *nf_sock; + + nf_sock = mnl_socket_open(NETLINK_NETFILTER); + if (!nf_sock) + netlink_init_error(); + + if (fcntl(mnl_socket_get_fd(nf_sock), F_SETFL, O_NONBLOCK)) + netlink_init_error(); + + return nf_sock; +} + +struct mnl_socket *nft_mnl_socket_reopen(struct mnl_socket *nf_sock) +{ + mnl_socket_close(nf_sock); + + return nft_mnl_socket_open(); +} + uint32_t mnl_seqnum_alloc(unsigned int *seqnum) { return (*seqnum)++; diff --git a/src/netlink.c b/src/netlink.c index 403780ff..8eb2ccad 100644 --- a/src/netlink.c +++ b/src/netlink.c @@ -10,7 +10,6 @@ */ #include <string.h> -#include <fcntl.h> #include <errno.h> #include <libmnl/libmnl.h> #include <netinet/in.h> @@ -53,32 +52,6 @@ const struct location netlink_location = { .indesc = &indesc_netlink, }; -struct mnl_socket *netlink_open_sock(void) -{ - struct mnl_socket *nf_sock; - - nf_sock = mnl_socket_open(NETLINK_NETFILTER); - if (nf_sock == NULL) - netlink_init_error(); - - if (fcntl(mnl_socket_get_fd(nf_sock), F_SETFL, O_NONBLOCK)) - netlink_init_error(); - - return nf_sock; -} - -void netlink_close_sock(struct mnl_socket *nf_sock) -{ - if (nf_sock) - mnl_socket_close(nf_sock); -} - -struct mnl_socket *netlink_restart(struct mnl_socket *nf_sock) -{ - netlink_close_sock(nf_sock); - return netlink_open_sock(); -} - void __noreturn __netlink_abi_error(const char *file, int line, const char *reason) { @@ -243,7 +243,7 @@ replay: if (ret < 0) { cache_release(cache); if (errno == EINTR) { - nft->nf_sock = netlink_restart(nft->nf_sock); + nft->nf_sock = nft_mnl_socket_reopen(nft->nf_sock); goto replay; } return -1; |