|
@@ -629,6 +629,33 @@ ssize_t tcp_splice_read(struct socket *sock, loff_t *ppos,
|
|
|
return ret;
|
|
|
}
|
|
|
|
|
|
+struct sk_buff *sk_stream_alloc_pskb(struct sock *sk,
|
|
|
+ int size, int mem, gfp_t gfp)
|
|
|
+{
|
|
|
+ struct sk_buff *skb;
|
|
|
+
|
|
|
+ /* The TCP header must be at least 32-bit aligned. */
|
|
|
+ size = ALIGN(size, 4);
|
|
|
+
|
|
|
+ skb = alloc_skb_fclone(size + sk->sk_prot->max_header, gfp);
|
|
|
+ if (skb) {
|
|
|
+ skb->truesize += mem;
|
|
|
+ if (sk_stream_wmem_schedule(sk, skb->truesize)) {
|
|
|
+ /*
|
|
|
+ * Make sure that we have exactly size bytes
|
|
|
+ * available to the caller, no more, no less.
|
|
|
+ */
|
|
|
+ skb_reserve(skb, skb_tailroom(skb) - size);
|
|
|
+ return skb;
|
|
|
+ }
|
|
|
+ __kfree_skb(skb);
|
|
|
+ } else {
|
|
|
+ sk->sk_prot->enter_memory_pressure();
|
|
|
+ sk_stream_moderate_sndbuf(sk);
|
|
|
+ }
|
|
|
+ return NULL;
|
|
|
+}
|
|
|
+
|
|
|
static ssize_t do_tcp_sendpages(struct sock *sk, struct page **pages, int poffset,
|
|
|
size_t psize, int flags)
|
|
|
{
|