netfilter: nfnetlink_acct: fix nfnl_acct_get operation
The get operation was not sending the message that was built to user-space. This patch also includes the appropriate handling for the return value of netlink_unicast(). Moreover, fix error codes on error (for example, for non-existing entry was uncorrect). Signed-off-by: Pablo Neira Ayuso <pablo@netfilter.org>
This commit is contained in:
parent
b236916a68
commit
3ab0b245aa
|
@ -166,7 +166,7 @@ static int
|
||||||
nfnl_acct_get(struct sock *nfnl, struct sk_buff *skb,
|
nfnl_acct_get(struct sock *nfnl, struct sk_buff *skb,
|
||||||
const struct nlmsghdr *nlh, const struct nlattr * const tb[])
|
const struct nlmsghdr *nlh, const struct nlattr * const tb[])
|
||||||
{
|
{
|
||||||
int ret = 0;
|
int ret = -ENOENT;
|
||||||
struct nf_acct *cur;
|
struct nf_acct *cur;
|
||||||
char *acct_name;
|
char *acct_name;
|
||||||
|
|
||||||
|
@ -186,17 +186,26 @@ nfnl_acct_get(struct sock *nfnl, struct sk_buff *skb,
|
||||||
continue;
|
continue;
|
||||||
|
|
||||||
skb2 = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
|
skb2 = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
|
||||||
if (skb2 == NULL)
|
if (skb2 == NULL) {
|
||||||
|
ret = -ENOMEM;
|
||||||
break;
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
ret = nfnl_acct_fill_info(skb2, NETLINK_CB(skb).pid,
|
ret = nfnl_acct_fill_info(skb2, NETLINK_CB(skb).pid,
|
||||||
nlh->nlmsg_seq,
|
nlh->nlmsg_seq,
|
||||||
NFNL_MSG_TYPE(nlh->nlmsg_type),
|
NFNL_MSG_TYPE(nlh->nlmsg_type),
|
||||||
NFNL_MSG_ACCT_NEW, cur);
|
NFNL_MSG_ACCT_NEW, cur);
|
||||||
if (ret <= 0)
|
if (ret <= 0) {
|
||||||
kfree_skb(skb2);
|
kfree_skb(skb2);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
ret = netlink_unicast(nfnl, skb2, NETLINK_CB(skb).pid,
|
||||||
|
MSG_DONTWAIT);
|
||||||
|
if (ret > 0)
|
||||||
|
ret = 0;
|
||||||
|
|
||||||
break;
|
/* this avoids a loop in nfnetlink. */
|
||||||
|
return ret == -EAGAIN ? -ENOBUFS : ret;
|
||||||
}
|
}
|
||||||
return ret;
|
return ret;
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue