1 files changed, 5 insertions, 0 deletions

diff --git a/net/wireless/mlme.c b/net/wireless/mlme.c
index 390198bf4b36..0c7b7dd855f6 100644
--- a/net/wireless/mlme.c
+++ b/net/wireless/mlme.c
@@ -648,6 +648,11 @@ void cfg80211_mlme_unregister_socket(struct wireless_dev *wdev, u32 nlportid)
 
         spin_unlock_bh(&wdev->mgmt_registrations_lock);
 
+        if (nlportid && rdev->crit_proto_nlportid == nlportid) {
+                rdev->crit_proto_nlportid = 0;
+                rdev_crit_proto_stop(rdev, wdev);
+        }
+
         if (nlportid == wdev->ap_unexpected_nlportid)
                 wdev->ap_unexpected_nlportid = 0;
 }