From 21000505e89af623f261faa7608f1b89a970506e Mon Sep 17 00:00:00 2001
From: "tianfeng.yang" <130436698+tianfeng-yang@users.noreply.github.com>
Date: Tue, 20 Feb 2024 22:41:17 +0800
Subject: [PATCH] Restore state before restart dnssd (#31985)

* Restore state before restart dnssd

* Needs to call Shutdown
---
 src/lib/dnssd/Discovery_ImplPlatform.cpp | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/lib/dnssd/Discovery_ImplPlatform.cpp b/src/lib/dnssd/Discovery_ImplPlatform.cpp
index abbc4647ff3805..3369f8c00b3b3b 100644
--- a/src/lib/dnssd/Discovery_ImplPlatform.cpp
+++ b/src/lib/dnssd/Discovery_ImplPlatform.cpp
@@ -420,8 +420,13 @@ void DiscoveryImplPlatform::HandleDnssdInit(void * context, CHIP_ERROR initError
 
 void DiscoveryImplPlatform::HandleDnssdError(void * context, CHIP_ERROR error)
 {
+    DiscoveryImplPlatform * publisher = static_cast<DiscoveryImplPlatform *>(context);
+
     if (error == CHIP_ERROR_FORCED_RESET)
     {
+        // Restore dnssd state before restart, also needs to call ChipDnssdShutdown()
+        publisher->Shutdown();
+
         DeviceLayer::ChipDeviceEvent event;
         event.Type = DeviceLayer::DeviceEventType::kDnssdRestartNeeded;
         error      = DeviceLayer::PlatformMgr().PostEvent(&event);