common: return nlmsghdr in nftnl_batch_{begin,end}()

Useful to append netlink attributes after the batch headers.

Signed-off-by: Pablo Neira Ayuso <pablo@netfilter.org>

diff --git a/include/libnftnl/common.h b/include/libnftnl/common.h
index f67f1866560fc7168f312d637fd0c67abb428da4..62c8f6b5e0a3009f247851b68a08a6255ff79894 100644 (file)
--- a/include/libnftnl/common.h
+++ b/include/libnftnl/common.h
@@ -49,7 +49,7 @@ void nftnl_parse_err_free(struct nftnl_parse_err *);
 int nftnl_parse_perror(const char *str, struct nftnl_parse_err *err);
 
 int nftnl_batch_is_supported(void);
-void nftnl_batch_begin(char *buf, uint32_t seq);
-void nftnl_batch_end(char *buf, uint32_t seq);
+struct nlmsghdr *nftnl_batch_begin(char *buf, uint32_t seq);
+struct nlmsghdr *nftnl_batch_end(char *buf, uint32_t seq);
 
 #endif

diff --git a/src/common.c b/src/common.c
index 0f23785e05d307d9b3cdfb610e1092f6cac6713b..a95883c19080843333a6c0a4d129d672bc662d97 100644 (file)
--- a/src/common.c
+++ b/src/common.c
@@ -165,17 +165,17 @@ int nftnl_cmd_footer_fprintf(FILE *fp, uint32_t cmd, uint32_t type,
                           nftnl_cmd_footer_fprintf_cb);
 }
 
-void nftnl_batch_begin(char *buf, uint32_t seq)
+struct nlmsghdr *nftnl_batch_begin(char *buf, uint32_t seq)
 {
-       __nftnl_nlmsg_build_hdr(buf, NFNL_MSG_BATCH_BEGIN, AF_UNSPEC, 0, seq,
-                               NFNL_SUBSYS_NFTABLES);
+       return __nftnl_nlmsg_build_hdr(buf, NFNL_MSG_BATCH_BEGIN, AF_UNSPEC,
+                                      0, seq, NFNL_SUBSYS_NFTABLES);
 }
 EXPORT_SYMBOL(nftnl_batch_begin);
 
-void nftnl_batch_end(char *buf, uint32_t seq)
+struct nlmsghdr *nftnl_batch_end(char *buf, uint32_t seq)
 {
-       __nftnl_nlmsg_build_hdr(buf, NFNL_MSG_BATCH_END, AF_UNSPEC, 0, seq,
-                               NFNL_SUBSYS_NFTABLES);
+       return __nftnl_nlmsg_build_hdr(buf, NFNL_MSG_BATCH_END, AF_UNSPEC,
+                                      0, seq, NFNL_SUBSYS_NFTABLES);
 }
 EXPORT_SYMBOL(nftnl_batch_end);