From 53ff4ed57f69c38c5aae7e41e51aa1016cdf2016 Mon Sep 17 00:00:00 2001
From: Quency-D <hj_zzns@163.com>
Date: Mon, 9 Feb 2026 11:43:41 +0800
Subject: [PATCH] Fix watchdog code in repeater

---
 examples/simple_repeater/main.cpp     | 23 ++++++++++++++++++++---
 src/helpers/ExWatchdogManager.h       |  3 ++-
 variants/heltec_mesh_solar/target.cpp |  9 +++++++--
 3 files changed, 29 insertions(+), 6 deletions(-)

diff --git a/examples/simple_repeater/main.cpp b/examples/simple_repeater/main.cpp
index 6cc2ac77..1760abea 100644
--- a/examples/simple_repeater/main.cpp
+++ b/examples/simple_repeater/main.cpp
@@ -29,6 +29,10 @@ void setup() {
 
   board.begin();
 
+#ifdef HAS_EX_WATCHDOG
+  ex_watchdog.begin();
+#endif
+
 #if defined(MESH_DEBUG) && defined(NRF52_PLATFORM)
   // give some extra time for serial to settle so
   // boot debug messages can be seen on terminal
@@ -134,11 +138,24 @@ void loop() {
 #endif
   rtc_clock.tick();
 
+#ifdef HAS_EX_WATCHDOG
+  ex_watchdog.loop();
+#endif
   if (the_mesh.getNodePrefs()->powersaving_enabled && !the_mesh.hasPendingWork()) {
-    #if defined(NRF52_PLATFORM)
+#if defined(NRF52_PLATFORM)
+#ifdef HAS_EX_WATCHDOG
+    uint32_t sleep_interval = ex_watchdog.getIntervalMs()/1000;
+    board.sleep((sleep_interval > 1800) ? 1800 : sleep_interval);             // To sleep. Wake up after 30 minutes or when receiving a LoRa packet
+#else
+    board.sleep(1800);             // To sleep. Wake up after 30 minutes or when receiving a LoRa packet
+#endif
     board.sleep(1800); // nrf ignores seconds param, sleeps whenever possible
-    #else
+#else
     if (the_mesh.millisHasNowPassed(lastActive + nextSleepinSecs * 1000)) { // To check if it is time to sleep
+#ifdef HAS_EX_WATCHDOG
+      uint32_t sleep_interval = ex_watchdog.getIntervalMs()/1000;
+      board.sleep((sleep_interval > 1800) ? 1800 : sleep_interval);             // To sleep. Wake up after 30 minutes or when receiving a LoRa packet
+#else
       board.sleep(1800);             // To sleep. Wake up after 30 minutes or when receiving a LoRa packet
 #endif
       lastActive = millis();
@@ -146,6 +163,6 @@ void loop() {
     } else {
       nextSleepinSecs += 5; // When there is pending work, to work another 5s
     }
-    #endif
+#endif
   }
 }
diff --git a/src/helpers/ExWatchdogManager.h b/src/helpers/ExWatchdogManager.h
index 65f178b0..4df901b7 100644
--- a/src/helpers/ExWatchdogManager.h
+++ b/src/helpers/ExWatchdogManager.h
@@ -1,8 +1,9 @@
 #pragma once
 
 class ExWatchdogManager {
-public:
+protected:
   unsigned long next_feed_watchdog;
+public:
   ExWatchdogManager() { next_feed_watchdog = 0; }
   virtual bool begin() { return false; }
   virtual void loop() { }
diff --git a/variants/heltec_mesh_solar/target.cpp b/variants/heltec_mesh_solar/target.cpp
index c9172aa1..7eb44bfa 100644
--- a/variants/heltec_mesh_solar/target.cpp
+++ b/variants/heltec_mesh_solar/target.cpp
@@ -141,11 +141,16 @@ void SolarExWatchdog::loop() {
 }
 
 unsigned long SolarExWatchdog::getIntervalMs() const {
-    return next_feed_watchdog - millis();
+    unsigned long interval_ms = 0;
+    interval_ms = next_feed_watchdog - millis();
+    if(interval_ms > EX_WATCHDOG_TIMEOUT_MS) {
+      interval_ms = EX_WATCHDOG_TIMEOUT_MS;
+    }
+    return interval_ms;
 }
 
 void SolarExWatchdog::feed() {
     digitalWrite(EX_WATCHDOG_DONE_PIN, HIGH);
     delay(1);
     digitalWrite(EX_WATCHDOG_DONE_PIN, LOW);
-}
\ No newline at end of file
+}