Abort the main process if child dies.

2 files changed, 12 insertions, 1 deletions

diff --git a/portato.py b/portato.py
index cea4ad7..421901a 100755
--- a/portato.py
+++ b/portato.py
@@ -14,13 +14,20 @@
 
 from __future__ import with_statement, absolute_import
 
+import signal
 import sys, os, subprocess
 import gettext, locale
 from optparse import OptionParser, SUPPRESS_HELP
 
 from portato import get_listener
+from portato.helper import debug, info
 from portato.constants import VERSION, LOCALE_DIR, APP, SU_COMMAND
 
+def sigchld_aborter (signal, frame):
+	"""Aborts the current process when the child dies."""
+	debug("Child process died. Trying to abort.")
+	raise KeyboardInterrupt
+
 def main ():
 	# set gettext stuff
 	locale.setlocale(locale.LC_ALL, '')
@@ -44,7 +51,6 @@ def main ():
 
 	if options.nofork or os.getuid() == 0: # start GUI
 		from portato.gui import run
-		from portato.helper import info
 		info("%s v. %s", _("Starting Portato"), VERSION)
 		
 		if options.shm:
@@ -67,7 +73,11 @@ def main ():
 		env.update(DBUS_SESSION_BUS_ADDRESS="")
 		cmd = SU_COMMAND.split()
 		subprocess.Popen(cmd+["%s --no-fork --shm %ld %ld %ld" % (sys.argv[0], mem.key, sig.key, rw.key)], env = env)
+
+		# set handler to abort the listener if needed
+		signal.signal(signal.SIGCHLD, sigchld_aborter)
 		
+		# stast listener
 		get_listener().set_recv(mem, sig, rw)
 
 if __name__ == "__main__":
diff --git a/portato/plistener.py b/portato/plistener.py
index 8d35ece..a45a17e 100644
--- a/portato/plistener.py
+++ b/portato/plistener.py
@@ -56,6 +56,7 @@ class PListener (object):
 				elif data[0] == "close":
 					break
 			except KeyboardInterrupt:
+				debug("Got KeyboardInterrupt. Aborting.")
 				break
 
 		self._mem.remove()