Ticket #489: choose-windows-by-pid.patch

xpra/server/server_base.py

@@ -138 +138 @@
                     "suspend", "resume", "name", "ungrab",
                     "key", "focus", "workspace", "idle-timeout",
                     "client", "start", "start-child",
-                    "send-file", "print"]
+                    "send-file", "print", "restrict"]
 
         self.init_encodings()
         self.init_packet_handlers()
@@ -1367 +1367 @@
                 assert ss.printing
                 ss.send_file(filename, True, True, ss, maxbitrate, printer, title, options)
             return 0, "printing to %s initiated" % client_uuids
+        elif command=="restrict":
+            if len(args)<1:
+                return argn_err(1)
+            client_uuids = args[0]
+            pids = [int(x) for x in args[1:]]
+            sources = get_sources(client_uuids, "file_transfer")
+            if not sources:
+                return arg_err("no clients found matching: %s" % client_uuids)
+            for csource in sources:
+                csource.only_pids = pids
+            return 0, "%s sources restricted to pids %s" % (len(sources), pids)
         else:
             return ServerCore.do_handle_command_request(self, command, args)
 

xpra/server/source.py

@@ -273 +273 @@
 
         self.window_sources = {}                    #WindowSource for each Window ID
         self.suspended = False
+        self.only_pids = None
 
         self.uuid = ""
         self.machine_id = ""
@@ -1549 +1550 @@
                 self.send("window-metadata", wid, metadata)
 
     def can_send_window(self, window):
+        if self.only_pids:
+            try:
+                pid = window.get_property("pid")
+                log.info("can_send_window(%s) pid=%s, only_pids=%s", window, pid, self.only_pids)
+                if pid and pid not in self.only_pids:
+                    return False
+            except:
+                pass
         if self.send_windows and self.system_tray:
             #common case shortcut
             return True