1 files changed, 7 insertions, 1 deletions

diff --git a/include/linux/skmsg.h b/include/linux/skmsg.h
index 50ced8aba9db..e4b3fb4bb77c 100644
--- a/include/linux/skmsg.h
+++ b/include/linux/skmsg.h
@@ -354,7 +354,13 @@ static inline void sk_psock_restore_proto(struct sock *sk,
         sk->sk_write_space = psock->saved_write_space;
 
         if (psock->sk_proto) {
-                sk->sk_prot = psock->sk_proto;
+                struct inet_connection_sock *icsk = inet_csk(sk);
+                bool has_ulp = !!icsk->icsk_ulp_data;
+
+                if (has_ulp)
+                        tcp_update_ulp(sk, psock->sk_proto);
+                else
+                        sk->sk_prot = psock->sk_proto;
                 psock->sk_proto = NULL;
         }
 }