Переглянути джерело

netfilter: ebt_ulog: Move away from NLMSG_PUT().

And use nlmsg_data() while we're here too.

Also, free and NULL out skb when nlmsg_put() fails and remove
pointless kernel log message.

Signed-off-by: David S. Miller <davem@davemloft.net>
David S. Miller 13 роки тому
батько
коміт
62566ca55d
1 змінених файлів з 10 додано та 13 видалено
  1. 10 13
      net/bridge/netfilter/ebt_ulog.c

+ 10 - 13
net/bridge/netfilter/ebt_ulog.c

@@ -145,19 +145,24 @@ static void ebt_ulog_packet(unsigned int hooknr, const struct sk_buff *skb,
 
 
 	if (!ub->skb) {
 	if (!ub->skb) {
 		if (!(ub->skb = ulog_alloc_skb(size)))
 		if (!(ub->skb = ulog_alloc_skb(size)))
-			goto alloc_failure;
+			goto unlock;
 	} else if (size > skb_tailroom(ub->skb)) {
 	} else if (size > skb_tailroom(ub->skb)) {
 		ulog_send(group);
 		ulog_send(group);
 
 
 		if (!(ub->skb = ulog_alloc_skb(size)))
 		if (!(ub->skb = ulog_alloc_skb(size)))
-			goto alloc_failure;
+			goto unlock;
 	}
 	}
 
 
-	nlh = NLMSG_PUT(ub->skb, 0, ub->qlen, 0,
-			size - NLMSG_ALIGN(sizeof(*nlh)));
+	nlh = nlmsg_put(ub->skb, 0, ub->qlen, 0,
+			size - NLMSG_ALIGN(sizeof(*nlh)), 0);
+	if (!nlh) {
+		kfree(ub->skb);
+		ub->skb = NULL;
+		goto unlock;
+	}
 	ub->qlen++;
 	ub->qlen++;
 
 
-	pm = NLMSG_DATA(nlh);
+	pm = nlmsg_data(nlh);
 
 
 	/* Fill in the ulog data */
 	/* Fill in the ulog data */
 	pm->version = EBT_ULOG_VERSION;
 	pm->version = EBT_ULOG_VERSION;
@@ -209,14 +214,6 @@ static void ebt_ulog_packet(unsigned int hooknr, const struct sk_buff *skb,
 
 
 unlock:
 unlock:
 	spin_unlock_bh(lock);
 	spin_unlock_bh(lock);
-
-	return;
-
-nlmsg_failure:
-	pr_debug("error during NLMSG_PUT. This should "
-		 "not happen, please report to author.\n");
-alloc_failure:
-	goto unlock;
 }
 }
 
 
 /* this function is registered with the netfilter core */
 /* this function is registered with the netfilter core */