Listen for all BDX protocol messages rather than just the init messages.

project-chip · Nov 5, 2024 · 1807324 · 1807324
1 parent bc870c2
commit 1807324
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/src/controller/python/chip/bdx/bdx-transfer-server.cpp b/src/controller/python/chip/bdx/bdx-transfer-server.cpp
@@ -30,15 +30,13 @@ CHIP_ERROR BdxTransferServer::Init(Messaging::ExchangeManager * exchangeManager)
 {
     VerifyOrReturnError(exchangeManager != nullptr, CHIP_ERROR_INVALID_ARGUMENT);
     mExchangeManager = exchangeManager;
-    ReturnOnFailure(mExchangeManager->RegisterUnsolicitedMessageHandlerForType(MessageType::ReceiveAccept, this));
-    return mExchangeManager->RegisterUnsolicitedMessageHandlerForType(MessageType::SendAccept, this);
+    return mExchangeManager->RegisterUnsolicitedMessageHandlerForProtocol(Protocols::BDX::Id, this);
 }
 
 void BdxTransferServer::Shutdown()
 {
     VerifyOrReturn(mExchangeManager != nullptr);
-    LogErrorOnFailure(mExchangeManager->UnregisterUnsolicitedMessageHandlerForType(MessageType::ReceiveAccept));
-    LogErrorOnFailure(mExchangeManager->UnregisterUnsolicitedMessageHandlerForType(MessageType::SendAccept));
+    LogErrorOnFailure(mExchangeManager->UnregisterUnsolicitedMessageHandlerForProtocol(Protocols::BDX::Id));
     mExchangeManager = nullptr;
 }