Merge pull request #698 from dadav/feature/usr1_handler
Add signal handler
This commit is contained in:
commit
b79c59c639
@ -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:
|
||||||
|
Loading…
x
Reference in New Issue
Block a user