rxrpc: Add a private skb flag to indicate transmission-phase skbs

    
Add a flag in the private data on an skbuff to indicate that this is a
transmission-phase buffer rather than a receive-phase buffer.

Signed-off-by: David Howells <dhowells@redhat.com>

2 files changed, 3 insertions, 1 deletions

diff --git a/net/rxrpc/ar-internal.h b/net/rxrpc/ar-internal.h
index 20d7907a5bc6..63d3a91ce5e9 100644
--- a/net/rxrpc/ar-internal.h
+++ b/net/rxrpc/ar-internal.h
@@ -188,6 +188,7 @@ struct rxrpc_skb_priv {
         u8              nr_subpackets;          /* Number of subpackets */
         u8              rx_flags;               /* Received packet flags */
 #define RXRPC_SKB_INCL_LAST     0x01            /* - Includes last packet */
+#define RXRPC_SKB_TX_BUFFER     0x02            /* - Is transmit buffer */
         union {
                 int             remain;         /* amount of space remaining for next write */
 
diff --git a/net/rxrpc/sendmsg.c b/net/rxrpc/sendmsg.c
index bae14438f869..472dc3b7d91f 100644
--- a/net/rxrpc/sendmsg.c
+++ b/net/rxrpc/sendmsg.c
@@ -336,6 +336,8 @@ static int rxrpc_send_data(struct rxrpc_sock *rx,
                         if (!skb)
                                 goto maybe_error;
 
+                        sp = rxrpc_skb(skb);
+                        sp->rx_flags |= RXRPC_SKB_TX_BUFFER;
                         rxrpc_new_skb(skb, rxrpc_skb_tx_new);
 
                         _debug("ALLOC SEND %p", skb);
@@ -346,7 +348,6 @@ static int rxrpc_send_data(struct rxrpc_sock *rx,
                         skb_reserve(skb, call->conn->security_size);
                         skb->len += call->conn->security_size;
 
-                        sp = rxrpc_skb(skb);
                         sp->remain = chunk;
                         if (sp->remain > skb_tailroom(skb))
                                 sp->remain = skb_tailroom(skb);