summaryrefslogtreecommitdiffstats
path: root/src/mnl.c
diff options
context:
space:
mode:
authorPablo Neira Ayuso <pablo@netfilter.org>2014-07-23 13:20:50 +0200
committerPablo Neira Ayuso <pablo@netfilter.org>2014-07-25 03:41:17 +0200
commit371fdadfafd64b3e364f91a21dac231a16622736 (patch)
tree9e43215af6d95f16ebed09b120a31fec8ca26d83 /src/mnl.c
parente2b60d54ab2f580cb4aef5a01c7b36a02aee6cc6 (diff)
mnl: fix crashes when using sets with many elements
nft crashes when adding many elements into a set for two reasons: 1) The overflow of the nla_len field for the NFTA_SET_ELEM_LIST_ELEMENTS attribute. 2) Out-of-bound memory writes to the reserved area for the netlink message, which is solved by the patch entitled ("mnl: introduce NFT_NLMSG_MAXSIZE"). This patch adds the corresponding nla_len overflow check for NFTA_SET_ELEM_LIST_ELEMENTS and it splits the elements in several netlink messages. This should be enough when set updates are handled by the transaction infrastructure. With this patch, nft should be now capable of adding an unlimited number of elements to a given set. Fixes: https://bugzilla.netfilter.org/show_bug.cgi?id=898 Signed-off-by: Pablo Neira Ayuso <pablo@netfilter.org>
Diffstat (limited to 'src/mnl.c')
-rw-r--r--src/mnl.c43
1 files changed, 32 insertions, 11 deletions
diff --git a/src/mnl.c b/src/mnl.c
index 667aa3a7..febf7c23 100644
--- a/src/mnl.c
+++ b/src/mnl.c
@@ -828,13 +828,24 @@ int mnl_nft_setelem_add(struct mnl_socket *nf_sock, struct nft_set *nls,
{
char buf[NFT_NLMSG_MAXSIZE];
struct nlmsghdr *nlh;
+ struct nft_set_elems_iter *iter;
+ int ret, err;
- nlh = nft_set_elem_nlmsg_build_hdr(buf, NFT_MSG_NEWSETELEM,
- nft_set_attr_get_u32(nls, NFT_SET_ATTR_FAMILY),
- NLM_F_CREATE | NLM_F_ACK | flags, seq);
- nft_set_elems_nlmsg_build_payload(nlh, nls);
+ iter = nft_set_elems_iter_create(nls);
+ if (iter == NULL)
+ memory_allocation_error();
- return nft_mnl_talk(nf_sock, nlh, nlh->nlmsg_len, NULL, NULL);
+ do {
+ nlh = nft_set_elem_nlmsg_build_hdr(buf, NFT_MSG_NEWSETELEM,
+ nft_set_attr_get_u32(nls, NFT_SET_ATTR_FAMILY),
+ NLM_F_CREATE | NLM_F_ACK | flags, seq);
+ ret = nft_set_elems_nlmsg_build_payload_iter(nlh, iter);
+ err = nft_mnl_talk(nf_sock, nlh, nlh->nlmsg_len, NULL, NULL);
+ } while (ret > 0 && err >= 0);
+
+ nft_set_elems_iter_destroy(iter);
+
+ return err;
}
int mnl_nft_setelem_delete(struct mnl_socket *nf_sock, struct nft_set *nls,
@@ -861,13 +872,23 @@ int mnl_nft_setelem_batch_add(struct mnl_socket *nf_sock, struct nft_set *nls,
unsigned int flags, uint32_t seqnum)
{
struct nlmsghdr *nlh;
+ struct nft_set_elems_iter *iter;
+ int ret;
- nlh = nft_set_elem_nlmsg_build_hdr(nft_nlmsg_batch_current(),
- NFT_MSG_NEWSETELEM,
- nft_set_attr_get_u32(nls, NFT_SET_ATTR_FAMILY),
- NLM_F_CREATE | flags, seqnum);
- nft_set_elems_nlmsg_build_payload(nlh, nls);
- nft_batch_continue();
+ iter = nft_set_elems_iter_create(nls);
+ if (iter == NULL)
+ memory_allocation_error();
+
+ do {
+ nlh = nft_set_elem_nlmsg_build_hdr(nft_nlmsg_batch_current(),
+ NFT_MSG_NEWSETELEM,
+ nft_set_attr_get_u32(nls, NFT_SET_ATTR_FAMILY),
+ NLM_F_CREATE | flags, seqnum);
+ ret = nft_set_elems_nlmsg_build_payload_iter(nlh, iter);
+ nft_batch_continue();
+ } while (ret > 0);
+
+ nft_set_elems_iter_destroy(iter);
return 0;
}