Add guarded_halt for completeness

nerves-project · Nov 25, 2022 · cd9536a · cd9536a
1 parent ff0ddbf
commit cd9536a
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 0 deletions.
diff --git a/src/heart.c b/src/heart.c
@@ -495,6 +495,13 @@ static int message_loop()
 
                         print_log("heart: poweroff signaled. No longer petting the WDT");
                         sync();
+                    } else if (mp_len == 13 && memcmp(m.fill, "guarded_halt", 12) == 0) {
+                        pet_watchdog(now);
+                        stop_petting_watchdog();
+                        kill(1, SIGUSR1); // SIGUSR1 signals "halt" to PID 1
+
+                        print_log("heart: halt signaled. No longer petting the WDT");
+                        sync();
                     } else if (mp_len == 15 && memcmp(m.fill, "init_handshake", 14) == 0) {
                         /* Application has said that it's completed initialization */
                         init_handshake_happened = 1;

diff --git a/tests/heart_test/test/guarded_reboot_test.exs b/tests/heart_test/test/guarded_reboot_test.exs
@@ -54,4 +54,28 @@ defmodule GuardedRebootTest do
     Heart.shutdown(heart)
     assert_receive {:exit, 0}
   end
+
+  test "guarded halt", context do
+    heart = start_supervised!({Heart, context.init_args})
+    assert_receive {:heart, :heart_ack}
+    assert_receive {:event, "open(/dev/watchdog0) succeeded"}
+    assert_receive {:event, "pet(1)"}
+
+    {:ok, :heart_ack} = Heart.set_cmd(heart, "guarded_halt")
+
+    # Final WDT pet
+    assert_receive {:event, "pet(1)"}
+
+    # Tell PID 1 to halt
+    assert_receive {:event, "kill(1, SIGUSR1)"}
+
+    # Proactive sync
+    assert_receive {:event, "sync()"}
+
+    Process.sleep(6)
+
+    # Run normal shutdown and check that there aren't any more WDT pets
+    Heart.shutdown(heart)
+    assert_receive {:exit, 0}
+  end
 end