net: keep original skb which only needs header checking during software GSO

Commit 9be6b2c047a58fca1cffcb2996d475758e5f15ab ("net: Don't keep
around original SKB when we software segment GSO frames") frees the
original skb after software GSO even for dodgy gso skbs. This breaks
the stream throughput from untrusted sources, since only header
checking was done during software GSO instead of a true
segmentation. This patch fixes this by freeing the original gso skb
only when it was really segmented by software.

Fixes 9be6b2c047a58fca1cffcb2996d475758e5f15ab ("net: Don't keep
around original SKB when we software segment GSO frames.")

Cc: David S. Miller <davem@davemloft.net>
Cc: Eric Dumazet <eric.dumazet@gmail.com>
Signed-off-by: Jason Wang <jasowang@redhat.com>
Acked-by: Eric Dumazet <edumazet@google.com>
Signed-off-by: David S. Miller <davem@davemloft.net>

diff --git a/net/core/dev.c b/net/core/dev.c
index e916ba8caccfc90c8c2128bbcac415ef74d37da3..52cd71a4a3439c131db57f8c301376af399995f1 100644 (file)
--- a/net/core/dev.c
+++ b/net/core/dev.c
@@ -2694,10 +2694,12 @@ struct sk_buff *validate_xmit_skb(struct sk_buff *skb, struct net_device *dev)
                struct sk_buff *segs;
 
                segs = skb_gso_segment(skb, features);
-               kfree_skb(skb);
-               if (IS_ERR(segs))
+               if (IS_ERR(segs)) {
                        segs = NULL;
-               skb = segs;
+               } else if (segs) {
+                       consume_skb(skb);
+                       skb = segs;
+               }
        } else {
                if (skb_needs_linearize(skb, features) &&
                    __skb_linearize(skb))