fix ci

nivi-apple · Oct 2, 2023 · 493d36f · 493d36f
1 parent 0dddcdb
commit 493d36f
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 38 deletions.
diff --git a/src/darwin/Framework/CHIP/MTROTAUnsolicitedBDXMessageHandler.mm b/src/darwin/Framework/CHIP/MTROTAUnsolicitedBDXMessageHandler.mm
@@ -28,10 +28,6 @@
 {
     assertChipStackLockedByCurrentThread();
 
-    if (mExchangeMgr) {
-        mExchangeMgr->UnregisterUnsolicitedMessageHandlerForProtocol(Protocols::BDX::Id);
-        mExchangeMgr = nullptr;
-    }
     MTROTAUnsolicitedBDXMessageHandler::mNumberOfDelegates = 0;
 }
 

diff --git a/src/protocols/bdx/AsyncTransferFacilitator.cpp b/src/protocols/bdx/AsyncTransferFacilitator.cpp
@@ -34,7 +34,6 @@ void AsyncTransferFacilitator::Reset()
 
 CHIP_ERROR AsyncTransferFacilitator::SendMessage(TransferSession::OutputEvent & event)
 {
-    assertChipStackLockedByCurrentThread();
 
     VerifyOrReturnError(mExchange, CHIP_ERROR_INCORRECT_STATE);
 
@@ -68,6 +67,9 @@ CHIP_ERROR AsyncTransferFacilitator::SendMessage(TransferSession::OutputEvent &
 CHIP_ERROR AsyncTransferFacilitator::OnMessageReceived(Messaging::ExchangeContext * ec, const PayloadHeader & payloadHeader,
                                                        System::PacketBufferHandle && payload)
 {
+
+    VerifyOrReturnError(mExchange, CHIP_ERROR_INCORRECT_STATE);
+
     VerifyOrReturnError(ec == mExchange.Get(), CHIP_ERROR_INCORRECT_STATE);
 
     CHIP_ERROR err =
@@ -94,12 +96,7 @@ CHIP_ERROR AsyncTransferFacilitator::OnMessageReceived(Messaging::ExchangeContex
         // If its a message of type kMsgToSend, send the message over the exchange.
         if (outEvent.EventType == TransferSession::OutputEventType::kMsgToSend)
         {
-            err = SendMessage(outEvent);
-            if (err != CHIP_NO_ERROR)
-            {
-                // TODO: Should we abort transfer here or let the other side timeout
-                return err;
-            }
+            SendMessage(outEvent);
         }
         else
         {
@@ -127,7 +124,7 @@ CHIP_ERROR AsyncTransferFacilitator::OnMessageReceived(Messaging::ExchangeContex
 void AsyncTransferFacilitator::OnExchangeClosing(Messaging::ExchangeContext * ec)
 {
     ChipLogDetail(BDX, "OnExchangeClosing, ec: " ChipLogFormatExchange, ChipLogValueExchange(ec));
-    CleanUp();
+    mExchange.Release();
 }
 
 void AsyncTransferFacilitator::OnResponseTimeout(Messaging::ExchangeContext * ec)
@@ -175,36 +172,26 @@ void AsyncResponder::NotifyEventHandled(CHIP_ERROR eventHandlingResult)
     // If there was an error, we abort the transfer.
     if (eventHandlingResult != CHIP_NO_ERROR)
     {
-        mTransfer.AbortTransfer(GetBdxStatusCodeFromChipError(eventHandlingResult));
-    }
+         CleanUp();
+    } else {
 
-    // We need to get the next output event from the state machine which is either a response to the
-    // BDX message handled by the subclass (ReceiveAccept/Block) or a status report or error or timeout
-    // event types.
-    TransferSession::OutputEvent outEvent;
-
-    mTransfer.GetNextAction(outEvent);
-    while (outEvent.EventType != TransferSession::OutputEventType::kNone)
-    {
-        if (outEvent.EventType == TransferSession::OutputEventType::kMsgToSend)
+        // We need to get the next output event from the state machine which is a response to the
+        // BDX message handled by the subclass (ReceiveAccept/Block).
+        TransferSession::OutputEvent outEvent;
+
+        mTransfer.GetNextAction(outEvent);
+        while (outEvent.EventType != TransferSession::OutputEventType::kNone)
         {
-            CHIP_ERROR err = AsyncTransferFacilitator::SendMessage(outEvent);
-            if (err != CHIP_NO_ERROR)
+            if (outEvent.EventType == TransferSession::OutputEventType::kMsgToSend)
             {
-                // TODO: Should we abort transfer here or let the other side timeout
-                return;
+                AsyncTransferFacilitator::SendMessage(outEvent);
+                mTransfer.GetNextAction(outEvent);
+            } else {
+                break;
             }
-        }
-        else if (outEvent.EventType == TransferSession::OutputEventType::kInternalError ||
-                 outEvent.EventType == TransferSession::OutputEventType::kTransferTimeout ||
-                 outEvent.EventType == TransferSession::OutputEventType::kStatusReceived)
-        {
-            HandleTransferSessionOutput(outEvent);
-            CleanUp();
-            return;
-        }
-        mTransfer.GetNextAction(outEvent);
-    };
+
+        };
+    }
 }
 
 } // namespace bdx