Add API to invalid sessions/exchanges for UpdateNOC command

project-chip · Jun 9, 2022 · 2f2fb5f · 2f2fb5f
1 parent ac8bdbf
commit 2f2fb5f
Show file tree

Hide file tree

Showing 4 changed files with 28 additions and 0 deletions.
diff --git a/src/messaging/ExchangeMgr.cpp b/src/messaging/ExchangeMgr.cpp
@@ -342,5 +342,16 @@ void ExchangeManager::CloseAllContextsForDelegate(const ExchangeDelegate * deleg
     });
 }
 
+void ExchangeManager::AbortExchangeForNodeExceptOne(const ScopedNodeId & node, ExchangeContext * stay)
+{
+    mContextPool.ForEachActiveObject([&](auto * ec) {
+        if (ec != stay && ec->HasSessionHandle() && ec->GetSessionHandle()->GetPeer() == node)
+        {
+            ec->Abort();
+        }
+        return Loop::Continue;
+    });
+}
+
 } // namespace Messaging
 } // namespace chip
diff --git a/src/messaging/ExchangeMgr.h b/src/messaging/ExchangeMgr.h
@@ -183,6 +183,9 @@ class DLL_EXPORT ExchangeManager : public SessionMessageDelegate
      */
     void CloseAllContextsForDelegate(const ExchangeDelegate * delegate);
 
+    // This API is used by UpdateNOC command, to invalid all exchanges except the given one.
+    void AbortExchangeForNodeExceptOne(const ScopedNodeId & node, ExchangeContext * stay);
+
     SessionManager * GetSessionManager() const { return mSessionManager; }
 
     ReliableMessageMgr * GetReliableMessageMgr() { return &mReliableMessageMgr; };

diff --git a/src/transport/SessionManager.cpp b/src/transport/SessionManager.cpp
@@ -387,6 +387,17 @@ void SessionManager::ExpireAllPASEPairings()
     });
 }
 
+void SessionManager::ReleaseSessionForNodeExceptOne(const ScopedNodeId & node, const SessionHandle & stay)
+{
+    mSecureSessions.ForEachSession([&](auto session) {
+        if (session->GetPeer() == node && session != stay->AsSecureSession())
+        {
+            session->MarkForRemoval();
+        }
+        return Loop::Continue;
+    });
+}
+
 Optional<SessionHandle> SessionManager::AllocateSession(SecureSession::Type secureSessionType)
 {
     return mSecureSessions.CreateNewSecureSession(secureSessionType);

diff --git a/src/transport/SessionManager.h b/src/transport/SessionManager.h
@@ -171,6 +171,9 @@ class DLL_EXPORT SessionManager : public TransportMgrDelegate
     void ExpireAllPairingsForFabric(FabricIndex fabric);
     void ExpireAllPASEPairings();
 
+    // This API is used by UpdateNOC command, to invalid all sessions except the given one.
+    void ReleaseSessionForNodeExceptOne(const ScopedNodeId & node, const SessionHandle & stay);
+
     /**
      * @brief
      *   Return the System Layer pointer used by current SessionManager.