blob: 6125a758d338b685e181cd20097fd7f3c7012965 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
import json
import time
from threading import Thread
import dbus
import dbus.service
from session import Session
class RWAService(dbus.service.Object):
def __init__(self):
self.bus = dbus.SessionBus()
name = dbus.service.BusName("de.rwa.rwa", bus=self.bus)
self.update_service_running = False
self.sessions = {}
super().__init__(name, "/RWA")
@dbus.service.method("de.rwa.rwa", out_signature="s")
def start(self):
"""Start a new remote session."""
# Start session
session = Session()
# Add session to sessions list
self.sessions[session.pid] = session
# Start session update service
self._ensure_update_service()
return json.dumps(session.client_meta)
@dbus.service.method("de.rwa.rwa", in_signature="i")
def stop(self, pid: int):
"""Stop a remote session."""
session = self.sessions[pid]
session.stop()
def _ensure_update_service(self):
"""Start session update thread if it isn't already running."""
if not self.update_service_running:
self.update_thread = Thread(target=self._update_sessions)
self.update_thread.start()
def _update_sessions(self):
"""Go through all running sessions and update their status.
Things that this function will do:
- Check if VNC is still running
- Kill websockify if VNC process is dead
"""
while len(self.sessions.values()) > 0:
for session in list(self.sessions.values()):
print(f"Session #{session.pid}")
# Check if VNC process is still running
running = session.vnc_process_running
if running:
print("Session is running")
else:
print("Session is dead.")
session.stop()
del self.sessions[session.pid]
time.sleep(2)
self.update_service_running = False
# TODO Probably kill daemon here (quit main loop)
if __name__ == "__main__":
import dbus.mainloop.glib
from gi.repository import GLib
dbus.mainloop.glib.DBusGMainLoop(set_as_default=True)
loop = GLib.MainLoop()
object = RWAService()
loop.run()
|
