scifijunkie/pwnagotchi
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add signal handler

Browse Source
This commit is contained in:
dadav 2019-12-07 09:36:56 +01:00
parent e04e053cee
commit 1a8472268e
1 changed files with 8 additions and 0 deletions

8
bin/pwnagotchi

@ -3,6 +3,7 @@ import logging
import argparse import argparse
import time import time
import yaml import yaml
import signal
import pwnagotchi import pwnagotchi
import pwnagotchi.grid as grid import pwnagotchi.grid as grid
@ -12,6 +13,7 @@ import pwnagotchi.plugins as plugins
from pwnagotchi.identity import KeyPair from pwnagotchi.identity import KeyPair
from pwnagotchi.agent import Agent from pwnagotchi.agent import Agent
from pwnagotchi.ui.display import Display from pwnagotchi.ui.display import Display
from pwnagotch import restart
def do_clear(display): def do_clear(display):
@ -134,6 +136,12 @@ if __name__ == '__main__':
agent = Agent(view=display, config=config, keypair=KeyPair(view=display)) agent = Agent(view=display, config=config, keypair=KeyPair(view=display))
def usr1_handler(*unused):
logging.info('Received USR1 singal. Restart process ...')
restart("MANU" if args.do_manual else "AUTO")
signal.signal(signal.SIGUSR1, usr1_handler)
if args.do_manual: if args.do_manual:
do_manual_mode(agent) do_manual_mode(agent)
else: else: