skbuff: introduce skb_expand_head()

Like skb_realloc_headroom(), new helper increases headroom of specified skb.
Unlike skb_realloc_headroom(), it does not allocate a new skb if possible;
copies skb->sk on new skb when as needed and frees original skb in case
of failures.

This helps to simplify ip[6]_finish_output2() and a few other similar cases.

Signed-off-by: Vasily Averin <vvs@virtuozzo.com>
Signed-off-by: David S. Miller <davem@davemloft.net>

diff --git a/include/linux/skbuff.h b/include/linux/skbuff.h
index 2bcdc8cd38be866087604abd7d2aa4fd92b72af6..783cc2368bb119da0ad6b803a9ecb8d3b3314d12 100644 (file)
--- a/include/linux/skbuff.h
+++ b/include/linux/skbuff.h
@@ -1183,6 +1183,7 @@ static inline struct sk_buff *__pskb_copy(struct sk_buff *skb, int headroom,
 int pskb_expand_head(struct sk_buff *skb, int nhead, int ntail, gfp_t gfp_mask);
 struct sk_buff *skb_realloc_headroom(struct sk_buff *skb,
                                     unsigned int headroom);
+struct sk_buff *skb_expand_head(struct sk_buff *skb, unsigned int headroom);
 struct sk_buff *skb_copy_expand(const struct sk_buff *skb, int newheadroom,
                                int newtailroom, gfp_t priority);
 int __must_check skb_to_sgvec_nomark(struct sk_buff *skb, struct scatterlist *sg,

diff --git a/net/core/skbuff.c b/net/core/skbuff.c
index fcbd977186b04b9637ee13bfbc91fd7420702756..8bac7a1a81ba9e45ca13224ff7ba90e0a77fcbba 100644 (file)
--- a/net/core/skbuff.c
+++ b/net/core/skbuff.c
@@ -1789,6 +1789,48 @@ struct sk_buff *skb_realloc_headroom(struct sk_buff *skb, unsigned int headroom)
 }
 EXPORT_SYMBOL(skb_realloc_headroom);
 
+/**
+ *     skb_expand_head - reallocate header of &sk_buff
+ *     @skb: buffer to reallocate
+ *     @headroom: needed headroom
+ *
+ *     Unlike skb_realloc_headroom, this one does not allocate a new skb
+ *     if possible; copies skb->sk to new skb as needed
+ *     and frees original skb in case of failures.
+ *
+ *     It expect increased headroom and generates warning otherwise.
+ */
+
+struct sk_buff *skb_expand_head(struct sk_buff *skb, unsigned int headroom)
+{
+       int delta = headroom - skb_headroom(skb);
+
+       if (WARN_ONCE(delta <= 0,
+                     "%s is expecting an increase in the headroom", __func__))
+               return skb;
+
+       /* pskb_expand_head() might crash, if skb is shared */
+       if (skb_shared(skb)) {
+               struct sk_buff *nskb = skb_clone(skb, GFP_ATOMIC);
+
+               if (likely(nskb)) {
+                       if (skb->sk)
+                               skb_set_owner_w(nskb, skb->sk);
+                       consume_skb(skb);
+               } else {
+                       kfree_skb(skb);
+               }
+               skb = nskb;
+       }
+       if (skb &&
+           pskb_expand_head(skb, SKB_DATA_ALIGN(delta), 0, GFP_ATOMIC)) {
+               kfree_skb(skb);
+               skb = NULL;
+       }
+       return skb;
+}
+EXPORT_SYMBOL(skb_expand_head);
+
 /**
  *     skb_copy_expand -       copy and expand sk_buff
  *     @skb: buffer to copy