Exchange handles SecureSession closing

src/messaging/ExchangeMgr.cpp

@@ -295,6 +295,19 @@ void ExchangeManager::OnMessageReceived(const PacketHeader & packetHeader, const
     DispatchMessage(packetHeader, payloadHeader, msgBuf);
 }
 
+void ExchangeManager::OnConnectionExpired(Transport::PeerConnectionState * state, SecureSessionMgrBase * mgr)
+{
+    // Search for an existing exchange that the message applies to. If a match is found...
+    ExchangeContext * ec = ContextPool;
+    for (int i = 0; i < CHIP_CONFIG_MAX_EXCHANGE_CONTEXTS; i++, ec++)
+    {
+        if (ec->GetReferenceCount() > 0 && ec->mPeerNodeId == state->GetPeerNodeId())
+        {
+            ec->Close();
+        }
+    }
+}
+
 void ExchangeManager::IncrementContextsInUse()
 {
     mContextsInUse++;

src/messaging/ExchangeMgr.h

@@ -201,6 +201,8 @@ class DLL_EXPORT ExchangeManager : public SecureSessionMgrDelegate
     void OnMessageReceived(const PacketHeader & packetHeader, const PayloadHeader & payloadHeader,
                            Transport::PeerConnectionState * state, System::PacketBuffer * msgBuf,
                            SecureSessionMgrBase * msgLayer) override;
+
+    void OnConnectionExpired(Transport::PeerConnectionState * state, SecureSessionMgrBase * mgr) override;
 };
 
 } // namespace chip

src/transport/PeerConnections.h

@@ -326,7 +326,7 @@ class PeerConnections
      *
      */
     template <class T>
-    void SetConnectionExpiredHandler(void (*handler)(const PeerConnectionState &, T *), T * param)
+    void SetConnectionExpiredHandler(void (*handler)(PeerConnectionState &, T *), T * param)
     {
         mConnectionExpiredArgument = param;
         OnConnectionExpired        = reinterpret_cast<ConnectionExpiredHandler>(handler);

src/transport/SecureSessionMgr.cpp

@@ -355,13 +355,18 @@ void SecureSessionMgrBase::HandleDataReceived(const PacketHeader & packetHeader,
     }
 }
 
-void SecureSessionMgrBase::HandleConnectionExpired(const Transport::PeerConnectionState & state, SecureSessionMgrBase * mgr)
+void SecureSessionMgrBase::HandleConnectionExpired(Transport::PeerConnectionState & state, SecureSessionMgrBase * mgr)
 {
     char addr[Transport::PeerAddress::kMaxToStringSize];
     state.GetPeerAddress().ToString(addr, sizeof(addr));
 
     ChipLogDetail(Inet, "Connection from '%s' expired", addr);
 
+    if (mgr->mCB != nullptr)
+    {
+        mgr->mCB->OnConnectionExpired(&state, mgr);
+    }
+
     mgr->mTransport->Disconnect(state.GetPeerAddress());
 }
 

src/transport/SecureSessionMgr.h

@@ -88,6 +88,15 @@ class DLL_EXPORT SecureSessionMgrDelegate
      */
     virtual void OnNewConnection(Transport::PeerConnectionState * state, SecureSessionMgrBase * mgr) {}
 
+    /**
+     * @brief
+     *   Called when a new connection is closing
+     *
+     * @param state   connection state
+     * @param mgr     A pointer to the SecureSessionMgr
+     */
+    virtual void OnConnectionExpired(Transport::PeerConnectionState * state, SecureSessionMgrBase * mgr) {}
+
     /**
      * @brief
      *   Called when the peer address is resolved from NodeID.
@@ -184,7 +193,7 @@ class DLL_EXPORT SecureSessionMgrBase : public Mdns::ResolveDelegate
     /**
      * Called when a specific connection expires.
      */
-    static void HandleConnectionExpired(const Transport::PeerConnectionState & state, SecureSessionMgrBase * mgr);
+    static void HandleConnectionExpired(Transport::PeerConnectionState & state, SecureSessionMgrBase * mgr);
 
     /**
      * Callback for timer expiry check

src/transport/tests/TestPeerConnections.cpp

@@ -201,7 +201,7 @@ struct ExpiredCallInfo
     PeerAddress lastCallPeerAddress = PeerAddress::Uninitialized();
 };
 
-void OnConnectionExpired(const PeerConnectionState & state, ExpiredCallInfo * info)
+void OnConnectionExpired(PeerConnectionState & state, ExpiredCallInfo * info)
 {
     info->callCount++;
     info->lastCallNodeId      = state.GetPeerNodeId();