From 386816bfebbbb3435110eeefdae9d678c0b372c0 Mon Sep 17 00:00:00 2001
From: crlonxp <88241281+crlonxp@users.noreply.github.com>
Date: Mon, 7 Oct 2024 12:47:53 +0800
Subject: [PATCH] Enable MRP in WiFi-PAF commissioning (#35500)

* Enable MRP in WiFi-PAF commissioning

Signed-off-by: Lo,Chin-Ran <chin-ran.lo@nxp.com>

* Restyled by clang-format

---------

Signed-off-by: Lo,Chin-Ran <chin-ran.lo@nxp.com>
Co-authored-by: Restyled.io <commits@restyled.io>
---
 src/transport/SecureSession.h               | 6 +++++-
 src/transport/UnauthenticatedSessionTable.h | 6 +++++-
 2 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/src/transport/SecureSession.h b/src/transport/SecureSession.h
index ba586a3095e3dd..574d5f40ccfe5a 100644
--- a/src/transport/SecureSession.h
+++ b/src/transport/SecureSession.h
@@ -158,7 +158,11 @@ class SecureSession : public Session, public ReferenceCounted<SecureSession, Sec
 
     bool IsCommissioningSession() const override;
 
-    bool AllowsMRP() const override { return GetPeerAddress().GetTransportType() == Transport::Type::kUdp; }
+    bool AllowsMRP() const override
+    {
+        return ((GetPeerAddress().GetTransportType() == Transport::Type::kUdp) ||
+                (GetPeerAddress().GetTransportType() == Transport::Type::kWiFiPAF));
+    }
 
     bool AllowsLargePayload() const override { return GetPeerAddress().GetTransportType() == Transport::Type::kTcp; }
 
diff --git a/src/transport/UnauthenticatedSessionTable.h b/src/transport/UnauthenticatedSessionTable.h
index d77bf7080d8fb9..707d4014f01167 100644
--- a/src/transport/UnauthenticatedSessionTable.h
+++ b/src/transport/UnauthenticatedSessionTable.h
@@ -84,7 +84,11 @@ class UnauthenticatedSession : public Session, public ReferenceCounted<Unauthent
         return Access::SubjectDescriptor(); // return an empty ISD for unauthenticated session.
     }
 
-    bool AllowsMRP() const override { return GetPeerAddress().GetTransportType() == Transport::Type::kUdp; }
+    bool AllowsMRP() const override
+    {
+        return ((GetPeerAddress().GetTransportType() == Transport::Type::kUdp) ||
+                (GetPeerAddress().GetTransportType() == Transport::Type::kWiFiPAF));
+    }
 
     bool AllowsLargePayload() const override { return GetPeerAddress().GetTransportType() == Transport::Type::kTcp; }