diff --git a/pwnagotchi/__init__.py b/pwnagotchi/__init__.py index c75af906..af9d12dd 100644 --- a/pwnagotchi/__init__.py +++ b/pwnagotchi/__init__.py @@ -65,3 +65,9 @@ def shutdown(): time.sleep(5) os.system("sync") os.system("halt") + + +def reboot(): + logging.warning("rebooting ...") + os.system("sync") + os.system("shutdown -r now") diff --git a/pwnagotchi/agent.py b/pwnagotchi/agent.py index 497c91ed..24013fb0 100644 --- a/pwnagotchi/agent.py +++ b/pwnagotchi/agent.py @@ -481,9 +481,7 @@ class Agent(Client, AsyncAdvertiser, AsyncTrainer): def _reboot(self): self.set_rebooting() self._save_recovery_data() - logging.warning("rebooting the system ...") - os.system("/usr/bin/sync") - os.system("/usr/sbin/shutdown -r now") + pwnagotchi.reboot() def next_epoch(self): was_stale = self.is_stale()