1 files changed, 2 insertions, 2 deletions

diff --git a/net/core/skmsg.c b/net/core/skmsg.c
index 93bffaad2135..6832eeb4b785 100644
--- a/net/core/skmsg.c
+++ b/net/core/skmsg.c
@@ -585,12 +585,12 @@ EXPORT_SYMBOL_GPL(sk_psock_destroy);
 
 void sk_psock_drop(struct sock *sk, struct sk_psock *psock)
 {
-        rcu_assign_sk_user_data(sk, NULL);
         sk_psock_cork_free(psock);
         sk_psock_zap_ingress(psock);
-        sk_psock_restore_proto(sk, psock);
 
         write_lock_bh(&sk->sk_callback_lock);
+        sk_psock_restore_proto(sk, psock);
+        rcu_assign_sk_user_data(sk, NULL);
         if (psock->progs.skb_parser)
                 sk_psock_stop_strp(sk, psock);
         write_unlock_bh(&sk->sk_callback_lock);