X-Git-Url: https://glassweightruler.freedombox.rocks/gitweb/waydroid.git/blobdiff_plain/204403056e6c8adc14406be5e64c4f1996e185bc..HEAD:/tools/actions/session_manager.py?ds=sidebyside

diff --git a/tools/actions/session_manager.py b/tools/actions/session_manager.py
index d2daf7a..e02665e 100644
--- a/tools/actions/session_manager.py
+++ b/tools/actions/session_manager.py
@@ -2,10 +2,8 @@
 # SPDX-License-Identifier: GPL-3.0-or-later
 import logging
 import os
-import time
 import signal
 import sys
-import shutil
 import tools.config
 import tools.helpers.ipc
 from tools import services
@@ -26,7 +24,16 @@ class DbusSessionManager(dbus.service.Object):
         do_stop(self.args, self.looper)
         stop_container(quit_session=False)
 
+def handle_disconnect(args, looper):
+    do_stop(args, looper)
+    stop_container(quit_session=False)
+
 def service(args, looper):
+    bus = dbus.SessionBus()
+    bus.set_exit_on_disconnect(False)
+    bus.add_signal_receiver(lambda: handle_disconnect(args, looper),
+                            signal_name='Disconnected',
+                            dbus_interface='org.freedesktop.DBus.Local')
     dbus_obj = DbusSessionManager(looper, dbus.SessionBus(), '/SessionManager', args)
     looper.run()
 
@@ -83,6 +90,7 @@ def start(args, unlocked_cb=None, background=True):
     def sigusr_handler(data):
         do_stop(args, mainloop)
 
+    GLib.unix_signal_add(GLib.PRIORITY_HIGH, signal.SIGHUP, sigint_handler, None)
     GLib.unix_signal_add(GLib.PRIORITY_HIGH, signal.SIGINT, sigint_handler, None)
     GLib.unix_signal_add(GLib.PRIORITY_HIGH, signal.SIGTERM, sigint_handler, None)
     GLib.unix_signal_add(GLib.PRIORITY_HIGH, signal.SIGUSR1, sigusr_handler, None)