connector: bump skb->users before callback invocation

Dmitry reports memleak with syskaller program.
Problem is that connector bumps skb usecount but might not invoke callback.

So move skb_get to where we invoke the callback.

Reported-by: Dmitry Vyukov <dvyukov@google.com>
Signed-off-by: Florian Westphal <fw@strlen.de>
Signed-off-by: David S. Miller <davem@davemloft.net>
tirimbino
Florian Westphal 9 years ago committed by David S. Miller
parent 3934aa4c1f
commit 55285bf094
  1. 11
      drivers/connector/connector.c

@ -179,26 +179,21 @@ static int cn_call_callback(struct sk_buff *skb)
* *
* It checks skb, netlink header and msg sizes, and calls callback helper. * It checks skb, netlink header and msg sizes, and calls callback helper.
*/ */
static void cn_rx_skb(struct sk_buff *__skb) static void cn_rx_skb(struct sk_buff *skb)
{ {
struct nlmsghdr *nlh; struct nlmsghdr *nlh;
struct sk_buff *skb;
int len, err; int len, err;
skb = skb_get(__skb);
if (skb->len >= NLMSG_HDRLEN) { if (skb->len >= NLMSG_HDRLEN) {
nlh = nlmsg_hdr(skb); nlh = nlmsg_hdr(skb);
len = nlmsg_len(nlh); len = nlmsg_len(nlh);
if (len < (int)sizeof(struct cn_msg) || if (len < (int)sizeof(struct cn_msg) ||
skb->len < nlh->nlmsg_len || skb->len < nlh->nlmsg_len ||
len > CONNECTOR_MAX_MSG_SIZE) { len > CONNECTOR_MAX_MSG_SIZE)
kfree_skb(skb);
return; return;
}
err = cn_call_callback(skb); err = cn_call_callback(skb_get(skb));
if (err < 0) if (err < 0)
kfree_skb(skb); kfree_skb(skb);
} }

Loading…
Cancel
Save