From 0c91e0b7341f28cfcf9240f6069dd68d488ae0ab Mon Sep 17 00:00:00 2001 From: Pablo Neira Ayuso Date: Wed, 1 Feb 2012 12:58:52 +0100 Subject: nfacct: use NFACCT_ATTR_* flags in nfacct_nlmsg_build_payload MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Instead of checking is name, packets and bytes are set. Reported-by: Alejandro CastaƱo del Castillo Signed-off-by: Pablo Neira Ayuso --- src/libnetfilter_acct.c | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/libnetfilter_acct.c b/src/libnetfilter_acct.c index 324a3c8..5dc402e 100644 --- a/src/libnetfilter_acct.c +++ b/src/libnetfilter_acct.c @@ -318,13 +318,13 @@ EXPORT_SYMBOL(nfacct_nlmsg_build_hdr); */ void nfacct_nlmsg_build_payload(struct nlmsghdr *nlh, struct nfacct *nfacct) { - if (nfacct->name) + if (nfacct->bitset & (1 << NFACCT_ATTR_NAME)) mnl_attr_put_strz(nlh, NFACCT_NAME, nfacct->name); - if (nfacct->pkts) + if (nfacct->bitset & (1 << NFACCT_ATTR_PKTS)) mnl_attr_put_u64(nlh, NFACCT_PKTS, htobe64(nfacct->pkts)); - if (nfacct->bytes) + if (nfacct->bitset & (1 << NFACCT_ATTR_BYTES)) mnl_attr_put_u64(nlh, NFACCT_PKTS, htobe64(nfacct->bytes)); } EXPORT_SYMBOL(nfacct_nlmsg_build_payload); -- cgit v1.2.3