This is a defense-in-depth measure in response to bugs like
4d6fa57b4dab0d77f4d8e9d9c73d1e63f6fe8fee.

Signed-off-by: Jason A. Donenfeld <ja...@zx2c4.com>
---
 net/core/skbuff.c | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/net/core/skbuff.c b/net/core/skbuff.c
index f86bf69cfb8d..7ed2cdf54c0a 100644
--- a/net/core/skbuff.c
+++ b/net/core/skbuff.c
@@ -3489,7 +3489,9 @@ void __init skb_init(void)
  *     @len: Length of buffer space to be mapped
  *
  *     Fill the specified scatter-gather list with mappings/pointers into a
- *     region of the buffer space attached to a socket buffer.
+ *     region of the buffer space attached to a socket buffer. Returns either
+ *     the number of scatterlist items used, or -EMSGSIZE if the contents
+ *     could not fit.
  */
 static int
 __skb_to_sgvec(struct sk_buff *skb, struct scatterlist *sg, int offset, int 
len)
@@ -3512,6 +3514,9 @@ __skb_to_sgvec(struct sk_buff *skb, struct scatterlist 
*sg, int offset, int len)
        for (i = 0; i < skb_shinfo(skb)->nr_frags; i++) {
                int end;
 
+               if (elt && sg_is_last(&sg[elt - 1]))
+                       return -EMSGSIZE;
+
                WARN_ON(start > offset + len);
 
                end = start + skb_frag_size(&skb_shinfo(skb)->frags[i]);
@@ -3535,6 +3540,9 @@ __skb_to_sgvec(struct sk_buff *skb, struct scatterlist 
*sg, int offset, int len)
 
                WARN_ON(start > offset + len);
 
+               if (elt && sg_is_last(&sg[elt - 1]))
+                       return -EMSGSIZE;
+
                end = start + frag_iter->len;
                if ((copy = end - offset) > 0) {
                        if (copy > len)
@@ -3581,6 +3589,9 @@ int skb_to_sgvec(struct sk_buff *skb, struct scatterlist 
*sg, int offset, int le
 {
        int nsg = __skb_to_sgvec(skb, sg, offset, len);
 
+       if (nsg < 0)
+               return nsg;
+
        sg_mark_end(&sg[nsg - 1]);
 
        return nsg;
-- 
2.12.2

Reply via email to