chrissewell/pwnagotchi
1
0
Fork 0
mirror of https://github.com/jayofelony/pwnagotchi.git synced 2025-07-01 18:37:27 -04:00
Code Issues Packages Projects Releases Wiki Activity

add mode parameter to agent._restart, and call agent._restart from USR1 handler instead of pwnagotchi.restart()

Browse Source
This commit is contained in:
Sniffleupagus
2024-09-08 19:25:09 +01:00
parent e5812e86c1
commit 32d244ec6f
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
bin/pwnagotchi
View File

@ -323,7 +323,7 @@ def pwnagotchi_cli():
def usr1_handler(*unused): def usr1_handler(*unused):
logging.info('Received USR1 singal. Restart process ...') logging.info('Received USR1 singal. Restart process ...')
restart("MANU" if args.do_manual else "AUTO") agent._restart("MANU" if args.do_manual else "AUTO")
signal.signal(signal.SIGUSR1, usr1_handler) signal.signal(signal.SIGUSR1, usr1_handler)

4
pwnagotchi/agent.py
View File

@ -269,9 +269,9 @@ class Agent(Client, Automata, AsyncAdvertiser, AsyncTrainer):
self._save_recovery_data() self._save_recovery_data()
pwnagotchi.reboot() pwnagotchi.reboot()
def _restart(self): def _restart(self, mode='AUTO'):
self._save_recovery_data() self._save_recovery_data()
pwnagotchi.restart("AUTO") pwnagotchi.restart(mode)
def _save_recovery_data(self): def _save_recovery_data(self):
logging.warning("writing recovery data to %s ...", RECOVERY_DATA_FILE) logging.warning("writing recovery data to %s ...", RECOVERY_DATA_FILE)