From 1a8472268e30f6eace5ea0d9d8a21d97fdc681ed Mon Sep 17 00:00:00 2001
From: dadav <33197631+dadav@users.noreply.github.com>
Date: Sat, 7 Dec 2019 09:36:56 +0100
Subject: [PATCH] Add signal handler

---
 bin/pwnagotchi | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/bin/pwnagotchi b/bin/pwnagotchi
index 9d21d5d..d8962e8 100755
--- a/bin/pwnagotchi
+++ b/bin/pwnagotchi
@@ -3,6 +3,7 @@ import logging
 import argparse
 import time
 import yaml
+import signal
 
 import pwnagotchi
 import pwnagotchi.grid as grid
@@ -12,6 +13,7 @@ import pwnagotchi.plugins as plugins
 from pwnagotchi.identity import KeyPair
 from pwnagotchi.agent import Agent
 from pwnagotchi.ui.display import Display
+from pwnagotch import restart
 
 
 def do_clear(display):
@@ -134,6 +136,12 @@ if __name__ == '__main__':
 
     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:
         do_manual_mode(agent)
     else: