1 files changed, 19 insertions, 0 deletions

diff --git a/include/net/tcp.h b/include/net/tcp.h
index 204328b88412..49a178b8d5b2 100644
--- a/include/net/tcp.h
+++ b/include/net/tcp.h
@@ -2232,4 +2232,23 @@ void clean_acked_data_disable(struct inet_connection_sock *icsk);
 void clean_acked_data_flush(void);
 #endif
 
+DECLARE_STATIC_KEY_FALSE(tcp_tx_delay_enabled);
+static inline void tcp_add_tx_delay(struct sk_buff *skb,
+                                    const struct tcp_sock *tp)
+{
+        if (static_branch_unlikely(&tcp_tx_delay_enabled))
+                skb->skb_mstamp_ns += (u64)tp->tcp_tx_delay * NSEC_PER_USEC;
+}
+
+static inline void tcp_set_tx_time(struct sk_buff *skb,
+                                   const struct sock *sk)
+{
+        if (static_branch_unlikely(&tcp_tx_delay_enabled)) {
+                u32 delay = (sk->sk_state == TCP_TIME_WAIT) ?
+                        tcp_twsk(sk)->tw_tx_delay : tcp_sk(sk)->tcp_tx_delay;
+
+                skb->skb_mstamp_ns = tcp_clock_ns() + (u64)delay * NSEC_PER_USEC;
+        }
+}
+
 #endif  /* _TCP_H */