12a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#!/usr/bin/python 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)# Copyright (c) 2012 The Chromium Authors. All rights reserved. 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)# Use of this source code is governed by a BSD-style license that can be 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)# found in the LICENSE file. 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)# Virtual Me2Me implementation. This script runs and manages the processes 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)# required for a Virtual Me2Me desktop, which are: X server, X desktop 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)# session, and Host process. 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)# This script is intended to run continuously as a background daemon 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)# process, running under an ordinary (non-root) user account. 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)import atexit 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)import errno 147d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)import fcntl 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)import getpass 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)import hashlib 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)import json 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)import logging 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)import optparse 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)import os 212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)import pipes 222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)import psutil 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)import signal 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)import socket 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)import subprocess 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)import sys 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)import tempfile 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)import time 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)import uuid 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)LOG_FILE_ENV_VAR = "CHROME_REMOTE_DESKTOP_LOG_FILE" 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)# This script has a sensible default for the initial and maximum desktop size, 342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)# which can be overridden either on the command-line, or via a comma-separated 352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)# list of sizes in this environment variable. 362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)DEFAULT_SIZES_ENV_VAR = "CHROME_REMOTE_DESKTOP_DEFAULT_DESKTOP_SIZES" 372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)# By default, provide a relatively small size to handle the case where resize- 392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)# to-client is disabled, and a much larger size to support clients with large 402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)# or mulitple monitors. These defaults can be overridden in ~/.profile. 412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)DEFAULT_SIZES = "1600x1200,3840x1600" 422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)SCRIPT_PATH = sys.path[0] 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)DEFAULT_INSTALL_PATH = "/opt/google/chrome-remote-desktop" 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)if SCRIPT_PATH == DEFAULT_INSTALL_PATH: 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HOST_BINARY_NAME = "chrome-remote-desktop-host" 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)else: 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HOST_BINARY_NAME = "remoting_me2me_host" 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)CHROME_REMOTING_GROUP_NAME = "chrome-remote-desktop" 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)CONFIG_DIR = os.path.expanduser("~/.config/chrome-remote-desktop") 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)HOME_DIR = os.environ["HOME"] 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)X_LOCK_FILE_TEMPLATE = "/tmp/.X%d-lock" 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)FIRST_X_DISPLAY_NUMBER = 20 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)# Amount of time to wait between relaunching processes. 602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)SHORT_BACKOFF_TIME = 5 612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)LONG_BACKOFF_TIME = 60 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)# How long a process must run in order not to be counted against the restart 642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)# thresholds. 652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)MINIMUM_PROCESS_LIFETIME = 60 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)# Thresholds for switching from fast- to slow-restart and for giving up 682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)# trying to restart entirely. 692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)SHORT_BACKOFF_THRESHOLD = 5 702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)MAX_LAUNCH_FAILURES = SHORT_BACKOFF_THRESHOLD + 10 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)# Globals needed by the atexit cleanup() handler. 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)g_desktops = [] 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)g_host_hash = hashlib.md5(socket.gethostname()).hexdigest() 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class Config: 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def __init__(self, path): 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.path = path 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.data = {} 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.changed = False 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def load(self): 832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) """Loads the config from file. 842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) Raises: 862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IOError: Error reading data 872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ValueError: Error parsing JSON 882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) """ 892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) settings_file = open(self.path, 'r') 902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self.data = json.load(settings_file) 912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self.changed = False 922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) settings_file.close() 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def save(self): 952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) """Saves the config to file. 962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) Raises: 982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IOError: Error writing data 992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) TypeError: Error serialising JSON 1002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) """ 1015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if not self.changed: 1022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return 1032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) old_umask = os.umask(0066) 1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) try: 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) settings_file = open(self.path, 'w') 1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) settings_file.write(json.dumps(self.data, indent=2)) 1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) settings_file.close() 1082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self.changed = False 1092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) finally: 1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) os.umask(old_umask) 1112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) def save_and_log_errors(self): 1132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) """Calls self.save(), trapping and logging any errors.""" 1142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) try: 1152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self.save() 1162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) except (IOError, TypeError) as e: 1172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) logging.error("Failed to save config: " + str(e)) 1185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def get(self, key): 1205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return self.data.get(key) 1215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def __getitem__(self, key): 1235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return self.data[key] 1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def __setitem__(self, key, value): 1265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.data[key] = value 1275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.changed = True 1285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 129a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) def clear(self): 130a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) self.data = {} 131868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) self.changed = True 1325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class Authentication: 1355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) """Manage authentication tokens for Chromoting/xmpp""" 1365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def __init__(self): 1385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.login = None 1395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.oauth_refresh_token = None 1405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def copy_from(self, config): 1425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) """Loads the config and returns false if the config is invalid.""" 1435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) try: 1445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.login = config["xmpp_login"] 1455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.oauth_refresh_token = config["oauth_refresh_token"] 1465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) except KeyError: 1475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return False 1485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return True 1495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def copy_to(self, config): 1515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) config["xmpp_login"] = self.login 1525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) config["oauth_refresh_token"] = self.oauth_refresh_token 1535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class Host: 1565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) """This manages the configuration for a host.""" 1575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def __init__(self): 1595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.host_id = str(uuid.uuid1()) 1605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.host_name = socket.gethostname() 1615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.host_secret_hash = None 1625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.private_key = None 1635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def copy_from(self, config): 1655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) try: 1665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.host_id = config["host_id"] 1675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.host_name = config["host_name"] 1685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.host_secret_hash = config.get("host_secret_hash") 1695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.private_key = config["private_key"] 1705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) except KeyError: 1715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return False 1725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return True 1735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def copy_to(self, config): 1755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) config["host_id"] = self.host_id 1765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) config["host_name"] = self.host_name 1775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) config["host_secret_hash"] = self.host_secret_hash 1785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) config["private_key"] = self.private_key 1795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class Desktop: 1825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) """Manage a single virtual desktop""" 1835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def __init__(self, sizes): 1855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.x_proc = None 1865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.session_proc = None 1875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.host_proc = None 1885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.child_env = None 1895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.sizes = sizes 1905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.pulseaudio_pipe = None 1915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.server_supports_exact_resize = False 1927d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) self.host_ready = False 1935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) g_desktops.append(self) 1945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) @staticmethod 1965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def get_unused_display_number(): 1975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) """Return a candidate display number for which there is currently no 1985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) X Server lock file""" 1995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) display = FIRST_X_DISPLAY_NUMBER 2005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) while os.path.exists(X_LOCK_FILE_TEMPLATE % display): 2015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) display += 1 2025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return display 2035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def _init_child_env(self): 2055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Create clean environment for new session, so it is cleanly separated from 2065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # the user's console X session. 2075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.child_env = {} 2082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for key in [ 2105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "HOME", 2115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "LANG", 2125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "LOGNAME", 2135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "PATH", 2145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "SHELL", 2155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "USER", 2165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "USERNAME", 2175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LOG_FILE_ENV_VAR]: 2185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if os.environ.has_key(key): 2195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.child_env[key] = os.environ[key] 2205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # Read from /etc/environment if it exists, as it is a standard place to 2222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # store system-wide environment settings. During a normal login, this would 2232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # typically be done by the pam_env PAM module, depending on the local PAM 2242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # configuration. 2252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) env_filename = "/etc/environment" 2262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) try: 2272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) with open(env_filename, "r") as env_file: 2282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) for line in env_file: 2292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) line = line.rstrip("\n") 2302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # Split at the first "=", leaving any further instances in the value. 2312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) key_value_pair = line.split("=", 1) 2322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if len(key_value_pair) == 2: 2332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) key, value = tuple(key_value_pair) 2342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # The file stores key=value assignments, but the value may be 2352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # quoted, so strip leading & trailing quotes from it. 2362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) value = value.strip("'\"") 2372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self.child_env[key] = value 2382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) except IOError: 2392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) logging.info("Failed to read %s, skipping." % env_filename) 2402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def _setup_pulseaudio(self): 2425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.pulseaudio_pipe = None 2435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # pulseaudio uses UNIX sockets for communication. Length of UNIX socket 2455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # name is limited to 108 characters, so audio will not work properly if 2465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # the path is too long. To workaround this problem we use only first 10 2475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # symbols of the host hash. 2485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) pulse_path = os.path.join(CONFIG_DIR, 2495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "pulseaudio#%s" % g_host_hash[0:10]) 2505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if len(pulse_path) + len("/native") >= 108: 2515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.error("Audio will not be enabled because pulseaudio UNIX " + 2525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "socket path is too long.") 2535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return False 2545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) sink_name = "chrome_remote_desktop_session" 2565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) pipe_name = os.path.join(pulse_path, "fifo_output") 2575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) try: 2595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if not os.path.exists(pulse_path): 2605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) os.mkdir(pulse_path) 2615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if not os.path.exists(pipe_name): 2625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) os.mkfifo(pipe_name) 2635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) except IOError, e: 2645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.error("Failed to create pulseaudio pipe: " + str(e)) 2655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return False 2665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) try: 268c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) pulse_config = open(os.path.join(pulse_path, "daemon.conf"), "w") 269c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) pulse_config.write("default-sample-format = s16le\n") 270c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) pulse_config.write("default-sample-rate = 48000\n") 271c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) pulse_config.write("default-sample-channels = 2\n") 272c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) pulse_config.close() 273c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 274c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) pulse_script = open(os.path.join(pulse_path, "default.pa"), "w") 275c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) pulse_script.write("load-module module-native-protocol-unix\n") 276c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) pulse_script.write( 2775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ("load-module module-pipe-sink sink_name=%s file=\"%s\" " + 2785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "rate=48000 channels=2 format=s16le\n") % 2795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) (sink_name, pipe_name)) 280c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) pulse_script.close() 2815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) except IOError, e: 2825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.error("Failed to write pulseaudio config: " + str(e)) 2835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return False 2845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.child_env["PULSE_CONFIG_PATH"] = pulse_path 2865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.child_env["PULSE_RUNTIME_PATH"] = pulse_path 2875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.child_env["PULSE_STATE_PATH"] = pulse_path 2885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.child_env["PULSE_SINK"] = sink_name 2895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.pulseaudio_pipe = pipe_name 2905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return True 2925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def _launch_x_server(self, extra_x_args): 2942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) x_auth_file = os.path.expanduser("~/.Xauthority") 2952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self.child_env["XAUTHORITY"] = x_auth_file 2965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) devnull = open(os.devnull, "rw") 2975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) display = self.get_unused_display_number() 2982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # Run "xauth add" with |child_env| so that it modifies the same XAUTHORITY 3002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # file which will be used for the X session. 3015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ret_code = subprocess.call("xauth add :%d . `mcookie`" % display, 3022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) env=self.child_env, shell=True) 3035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if ret_code != 0: 3045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) raise Exception("xauth failed with code %d" % ret_code) 3055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) max_width = max([width for width, height in self.sizes]) 3075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) max_height = max([height for width, height in self.sizes]) 3085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) try: 3105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # TODO(jamiewalch): This script expects to be installed alongside 3115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Xvfb-randr, but that's no longer the case. Fix this once we have 3125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # a Xvfb-randr package that installs somewhere sensible. 3135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) xvfb = "/usr/bin/Xvfb-randr" 3145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if not os.path.exists(xvfb): 3155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) xvfb = locate_executable("Xvfb-randr") 3165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.server_supports_exact_resize = True 3175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) except Exception: 3185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) xvfb = "Xvfb" 3195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.server_supports_exact_resize = False 3205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 321c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) # Disable the Composite extension iff the X session is the default 322c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) # Unity-2D, since it uses Metacity which fails to generate DAMAGE 323c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) # notifications correctly. See crbug.com/166468. 3247d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) x_session = choose_x_session() 325c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) if (len(x_session) == 2 and 326c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) x_session[1] == "/usr/bin/gnome-session --session=ubuntu-2d"): 327c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) extra_x_args.extend(["-extension", "Composite"]) 328c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 3295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.info("Starting %s on display :%d" % (xvfb, display)) 3305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) screen_option = "%dx%dx24" % (max_width, max_height) 3312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self.x_proc = subprocess.Popen( 3322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) [xvfb, ":%d" % display, 3332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) "-auth", x_auth_file, 3342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) "-nolisten", "tcp", 3352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) "-noreset", 3362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) "-screen", "0", screen_option 3372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ] + extra_x_args) 3385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if not self.x_proc.pid: 3395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) raise Exception("Could not start Xvfb.") 3405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.child_env["DISPLAY"] = ":%d" % display 3425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.child_env["CHROME_REMOTE_DESKTOP_SESSION"] = "1" 3435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Wait for X to be active. 3455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for _test in range(5): 3465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) proc = subprocess.Popen("xdpyinfo", env=self.child_env, stdout=devnull) 3475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) _pid, retcode = os.waitpid(proc.pid, 0) 3485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if retcode == 0: 3495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) break 3505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) time.sleep(0.5) 3515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if retcode != 0: 3525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) raise Exception("Could not connect to Xvfb.") 3535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) else: 3545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.info("Xvfb is active.") 3555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # The remoting host expects the server to use "evdev" keycodes, but Xvfb 3575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # starts configured to use the "base" ruleset, resulting in XKB configuring 3585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # for "xfree86" keycodes, and screwing up some keys. See crbug.com/119013. 3595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Reconfigure the X server to use "evdev" keymap rules. The X server must 3605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # be started with -noreset otherwise it'll reset as soon as the command 3615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # completes, since there are no other X clients running yet. 3625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) proc = subprocess.Popen("setxkbmap -rules evdev", env=self.child_env, 3635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) shell=True) 3645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) _pid, retcode = os.waitpid(proc.pid, 0) 3655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if retcode != 0: 3665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.error("Failed to set XKB to 'evdev'") 3675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Register the screen sizes if the X server's RANDR extension supports it. 3695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Errors here are non-fatal; the X server will continue to run with the 3705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # dimensions from the "-screen" option. 3715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for width, height in self.sizes: 3725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) label = "%dx%d" % (width, height) 3735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) args = ["xrandr", "--newmode", label, "0", str(width), "0", "0", "0", 3745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) str(height), "0", "0", "0"] 3752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) subprocess.call(args, env=self.child_env, stdout=devnull, stderr=devnull) 3765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) args = ["xrandr", "--addmode", "screen", label] 3772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) subprocess.call(args, env=self.child_env, stdout=devnull, stderr=devnull) 3785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Set the initial mode to the first size specified, otherwise the X server 3805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # would default to (max_width, max_height), which might not even be in the 3815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # list. 3825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) label = "%dx%d" % self.sizes[0] 3835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) args = ["xrandr", "-s", label] 3842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) subprocess.call(args, env=self.child_env, stdout=devnull, stderr=devnull) 3852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # Set the physical size of the display so that the initial mode is running 3872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # at approximately 96 DPI, since some desktops require the DPI to be set to 3882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # something realistic. 3892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) args = ["xrandr", "--dpi", "96"] 3902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) subprocess.call(args, env=self.child_env, stdout=devnull, stderr=devnull) 3915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) devnull.close() 3935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def _launch_x_session(self): 3952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # Start desktop session. 3965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # The /dev/null input redirection is necessary to prevent the X session 3975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # reading from stdin. If this code runs as a shell background job in a 3985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # terminal, any reading from stdin causes the job to be suspended. 3995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Daemonization would solve this problem by separating the process from the 4005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # controlling terminal. 4012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) xsession_command = choose_x_session() 4022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if xsession_command is None: 4032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) raise Exception("Unable to choose suitable X session command.") 4042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) logging.info("Launching X session: %s" % xsession_command) 4062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self.session_proc = subprocess.Popen(xsession_command, 4075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) stdin=open(os.devnull, "r"), 4085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) cwd=HOME_DIR, 4095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) env=self.child_env) 4105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if not self.session_proc.pid: 4115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) raise Exception("Could not start X session") 4125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def launch_session(self, x_args): 4145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self._init_child_env() 4155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self._setup_pulseaudio() 4165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self._launch_x_server(x_args) 4175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self._launch_x_session() 4185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def launch_host(self, host_config): 4205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Start remoting host 421a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) args = [locate_executable(HOST_BINARY_NAME), "--host-config=-"] 4225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if self.pulseaudio_pipe: 4235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) args.append("--audio-pipe-name=%s" % self.pulseaudio_pipe) 4245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if self.server_supports_exact_resize: 4255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) args.append("--server-supports-exact-resize") 4267d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 4277d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) # Have the host process use SIGUSR1 to signal a successful start. 4287d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) def sigusr1_handler(signum, frame): 4297d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) _ = signum, frame 4307d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) logging.info("Host ready to receive connections.") 4317d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) self.host_ready = True 4327d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) if (ParentProcessLogger.instance() and 4337d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) False not in [desktop.host_ready for desktop in g_desktops]): 4347d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) ParentProcessLogger.instance().release_parent() 4357d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 4367d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) signal.signal(signal.SIGUSR1, sigusr1_handler) 4377d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) args.append("--signal-parent") 4387d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 4395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.host_proc = subprocess.Popen(args, env=self.child_env, 4405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) stdin=subprocess.PIPE) 4415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.info(args) 4425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if not self.host_proc.pid: 4435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) raise Exception("Could not start Chrome Remote Desktop host") 4445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.host_proc.stdin.write(json.dumps(host_config.data)) 4455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.host_proc.stdin.close() 4465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)def get_daemon_pid(): 4492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) """Checks if there is already an instance of this script running, and returns 4502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) its PID. 4515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) Returns: 4532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) The process ID of the existing daemon process, or 0 if the daemon is not 4542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) running. 4555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) """ 4562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) uid = os.getuid() 4572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) this_pid = os.getpid() 4585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) for process in psutil.process_iter(): 4602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # Skip any processes that raise an exception, as processes may terminate 4612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # during iteration over the list. 4622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) try: 4632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # Skip other users' processes. 4642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if process.uids.real != uid: 4652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) continue 4665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # Skip the process for this instance. 4682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if process.pid == this_pid: 4692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) continue 4705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # |cmdline| will be [python-interpreter, script-file, other arguments...] 4722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) cmdline = process.cmdline 4732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if len(cmdline) < 2: 4742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) continue 4752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if cmdline[0] == sys.executable and cmdline[1] == sys.argv[0]: 4762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return process.pid 4772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) except psutil.error.Error: 4782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) continue 4795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return 0 4815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)def choose_x_session(): 4845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) """Chooses the most appropriate X session command for this system. 4855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Returns: 4875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) A string containing the command to run, or a list of strings containing 4885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) the executable program and its arguments, which is suitable for passing as 4895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) the first parameter of subprocess.Popen(). If a suitable session cannot 4905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) be found, returns None. 4915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) """ 4925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # If the session wrapper script (see below) is given a specific session as an 4935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # argument (such as ubuntu-2d on Ubuntu 12.04), the wrapper will run that 4945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # session instead of looking for custom .xsession files in the home directory. 4955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # So it's necessary to test for these files here. 4965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) XSESSION_FILES = [ 4975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "~/.chrome-remote-desktop-session", 4985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "~/.xsession", 4995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "~/.Xsession" ] 5005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for startup_file in XSESSION_FILES: 5015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) startup_file = os.path.expanduser(startup_file) 5025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if os.path.exists(startup_file): 5035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Use the same logic that a Debian system typically uses with ~/.xsession 5045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # (see /etc/X11/Xsession.d/50x11-common_determine-startup), to determine 5055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # exactly how to run this file. 5065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if os.access(startup_file, os.X_OK): 5072a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # "/bin/sh -c" is smart about how to execute the session script and 5082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # works in cases where plain exec() fails (for example, if the file is 5092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # marked executable, but is a plain script with no shebang line). 5102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return ["/bin/sh", "-c", pipes.quote(startup_file)] 5115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) else: 5125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) shell = os.environ.get("SHELL", "sh") 5135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return [shell, startup_file] 5145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Choose a session wrapper script to run the session. On some systems, 5165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # /etc/X11/Xsession fails to load the user's .profile, so look for an 5175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # alternative wrapper that is more likely to match the script that the 5185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # system actually uses for console desktop sessions. 5195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SESSION_WRAPPERS = [ 5205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "/usr/sbin/lightdm-session", 5215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "/etc/gdm/Xsession", 5225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "/etc/X11/Xsession" ] 5235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for session_wrapper in SESSION_WRAPPERS: 5245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if os.path.exists(session_wrapper): 5252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if os.path.exists("/usr/bin/unity-2d-panel"): 5262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # On Ubuntu 12.04, the default session relies on 3D-accelerated 5272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # hardware. Trying to run this with a virtual X display produces 5282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # weird results on some systems (for example, upside-down and 5292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # corrupt displays). So if the ubuntu-2d session is available, 5302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # choose it explicitly. 5312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return [session_wrapper, "/usr/bin/gnome-session --session=ubuntu-2d"] 5322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) else: 5332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # Use the session wrapper by itself, and let the system choose a 5342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # session. 535c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) return [session_wrapper] 5362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return None 5375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)def locate_executable(exe_name): 5405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if SCRIPT_PATH == DEFAULT_INSTALL_PATH: 5415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # If we are installed in the default path, then search the host binary 5425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # only in the same directory. 5435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) paths_to_try = [ DEFAULT_INSTALL_PATH ] 5445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) else: 5455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) paths_to_try = map(lambda p: os.path.join(SCRIPT_PATH, p), 5465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) [".", "../../out/Debug", "../../out/Release" ]) 5475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for path in paths_to_try: 5485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) exe_path = os.path.join(path, exe_name) 5495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if os.path.exists(exe_path): 5505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return exe_path 5515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) raise Exception("Could not locate executable '%s'" % exe_name) 5535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5557d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)class ParentProcessLogger(object): 5567d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) """Redirects logs to the parent process, until the host is ready or quits. 5577d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 5587d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) This class creates a pipe to allow logging from the daemon process to be 5597d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) copied to the parent process. The daemon process adds a log-handler that 5607d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) directs logging output to the pipe. The parent process reads from this pipe 5617d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) until and writes the content to stderr. When the pipe is no longer needed 5627d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) (for example, the host signals successful launch or permanent failure), the 5637d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) daemon removes the log-handler and closes the pipe, causing the the parent 5647d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) process to reach end-of-file while reading the pipe and exit. 5657d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 5667d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) The (singleton) logger should be instantiated before forking. The parent 5677d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) process should call wait_for_logs() before exiting. The (grand-)child process 5687d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) should call start_logging() when it starts, and then use logging.* to issue 5697d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) log statements, as usual. When the child has either succesfully started the 5707d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) host or terminated, it must call release_parent() to allow the parent to exit. 5717d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) """ 5727d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 5737d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) __instance = None 5747d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 5757d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) def __init__(self): 5767d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) """Constructor. Must be called before forking.""" 5777d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) read_pipe, write_pipe = os.pipe() 5787d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) # Ensure write_pipe is closed on exec, otherwise it will be kept open by 5797d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) # child processes (X, host), preventing the read pipe from EOF'ing. 5807d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) old_flags = fcntl.fcntl(write_pipe, fcntl.F_GETFD) 5817d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) fcntl.fcntl(write_pipe, fcntl.F_SETFD, old_flags | fcntl.FD_CLOEXEC) 5827d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) self._read_file = os.fdopen(read_pipe, 'r') 5837d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) self._write_file = os.fdopen(write_pipe, 'a') 5847d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) self._logging_handler = None 5857d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) ParentProcessLogger.__instance = self 5867d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 5877d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) def start_logging(self): 5887d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) """Installs a logging handler that sends log entries to a pipe. 5897d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 5907d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) Must be called by the child process. 5917d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) """ 5927d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) self._read_file.close() 5937d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) self._logging_handler = logging.StreamHandler(self._write_file) 5947d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) logging.getLogger().addHandler(self._logging_handler) 5957d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 5967d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) def release_parent(self): 5977d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) """Uninstalls logging handler and closes the pipe, releasing the parent. 5987d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 5997d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) Must be called by the child process. 6007d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) """ 6017d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) if self._logging_handler: 6027d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) logging.getLogger().removeHandler(self._logging_handler) 6037d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) self._logging_handler = None 6047d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) if not self._write_file.closed: 6057d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) self._write_file.close() 6067d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 6077d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) def wait_for_logs(self): 6087d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) """Waits and prints log lines from the daemon until the pipe is closed. 6097d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 6107d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) Must be called by the parent process. 6117d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) """ 6127d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) # If Ctrl-C is pressed, inform the user that the daemon is still running. 6137d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) # This signal will cause the read loop below to stop with an EINTR IOError. 6147d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) def sigint_handler(signum, frame): 6157d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) _ = signum, frame 6167d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) print >> sys.stderr, ("Interrupted. The daemon is still running in the " 6177d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) "background.") 6187d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 6197d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) signal.signal(signal.SIGINT, sigint_handler) 6207d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 6217d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) # Install a fallback timeout to release the parent process, in case the 6227d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) # daemon never responds (e.g. host crash-looping, daemon killed). 6237d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) # This signal will cause the read loop below to stop with an EINTR IOError. 6247d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) def sigalrm_handler(signum, frame): 6257d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) _ = signum, frame 6267d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) print >> sys.stderr, ("No response from daemon. It may have crashed, or " 6277d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) "may still be running in the background.") 6287d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 6297d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) signal.signal(signal.SIGALRM, sigalrm_handler) 6307d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) signal.alarm(30) 6317d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 6327d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) self._write_file.close() 6337d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 6347d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) # Print lines as they're logged to the pipe until EOF is reached or readline 6357d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) # is interrupted by one of the signal handlers above. 6367d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) try: 6377d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) for line in iter(self._read_file.readline, ''): 6387d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) sys.stderr.write(line) 6397d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) except IOError as e: 6407d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) if e.errno != errno.EINTR: 6417d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) raise 6427d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) print >> sys.stderr, "Log file: %s" % os.environ[LOG_FILE_ENV_VAR] 6437d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 6447d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) @staticmethod 6457d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) def instance(): 6467d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) """Returns the singleton instance, if it exists.""" 6477d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) return ParentProcessLogger.__instance 6487d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 6497d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 6507d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)def daemonize(): 6515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) """Background this process and detach from controlling terminal, redirecting 6527d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) stdout/stderr to a log file.""" 6535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # TODO(lambroslambrou): Having stdout/stderr redirected to a log file is not 6555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # ideal - it could create a filesystem DoS if the daemon or a child process 6565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # were to write excessive amounts to stdout/stderr. Ideally, stdout/stderr 6575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # should be redirected to a pipe or socket, and a process at the other end 6585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # should consume the data and write it to a logging facility which can do 6595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # data-capping or log-rotation. The 'logger' command-line utility could be 6605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # used for this, but it might cause too much syslog spam. 6615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Create new (temporary) file-descriptors before forking, so any errors get 6635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # reported to the main process and set the correct exit-code. 6645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # The mode is provided, since Python otherwise sets a default mode of 0777, 6655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # which would result in the new file having permissions of 0777 & ~umask, 6665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # possibly leaving the executable bits set. 6677d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) if not os.environ.has_key(LOG_FILE_ENV_VAR): 6687d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) log_file_prefix = "chrome_remote_desktop_%s_" % time.strftime( 6697d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) '%Y%m%d_%H%M%S', time.localtime(time.time())) 6707d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) log_file = tempfile.NamedTemporaryFile(prefix=log_file_prefix, delete=False) 6717d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) os.environ[LOG_FILE_ENV_VAR] = log_file.name 6727d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) log_fd = log_file.file.fileno() 6737d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) else: 6747d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) log_fd = os.open(os.environ[LOG_FILE_ENV_VAR], 6757d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) os.O_WRONLY | os.O_CREAT | os.O_APPEND, 0600) 6767d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 6775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) devnull_fd = os.open(os.devnull, os.O_RDONLY) 6787d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 6797d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) parent_logger = ParentProcessLogger() 6805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) pid = os.fork() 6825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if pid == 0: 6845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Child process 6855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) os.setsid() 6865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # The second fork ensures that the daemon isn't a session leader, so that 6885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # it doesn't acquire a controlling terminal. 6895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) pid = os.fork() 6905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if pid == 0: 6925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Grandchild process 6935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) pass 6945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) else: 6955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Child process 6962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) os._exit(0) # pylint: disable=W0212 6975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) else: 6985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Parent process 6997d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) parent_logger.wait_for_logs() 7002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) os._exit(0) # pylint: disable=W0212 7015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7027d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) logging.info("Daemon process started in the background, logging to '%s'" % 7037d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) os.environ[LOG_FILE_ENV_VAR]) 7045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) os.chdir(HOME_DIR) 7065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7077d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) parent_logger.start_logging() 7087d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 7095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Copy the file-descriptors to create new stdin, stdout and stderr. Note 7105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # that dup2(oldfd, newfd) closes newfd first, so this will close the current 7115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # stdin, stdout and stderr, detaching from the terminal. 7125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) os.dup2(devnull_fd, sys.stdin.fileno()) 7135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) os.dup2(log_fd, sys.stdout.fileno()) 7145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) os.dup2(log_fd, sys.stderr.fileno()) 7155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Close the temporary file-descriptors. 7175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) os.close(devnull_fd) 7185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) os.close(log_fd) 7195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)def cleanup(): 7225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.info("Cleanup.") 7235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) global g_desktops 7255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for desktop in g_desktops: 7265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if desktop.x_proc: 7275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.info("Terminating Xvfb") 7285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) desktop.x_proc.terminate() 7295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) g_desktops = [] 7307d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) if ParentProcessLogger.instance(): 7317d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) ParentProcessLogger.instance().release_parent() 7325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class SignalHandler: 7345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) """Reload the config file on SIGHUP. Since we pass the configuration to the 7355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) host processes via stdin, they can't reload it, so terminate them. They will 7365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) be relaunched automatically with the new config.""" 7375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def __init__(self, host_config): 7395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.host_config = host_config 7405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def __call__(self, signum, _stackframe): 7425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if signum == signal.SIGHUP: 7435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.info("SIGHUP caught, restarting host.") 7442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) try: 7452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self.host_config.load() 7462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) except (IOError, ValueError) as e: 7472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) logging.error("Failed to load config: " + str(e)) 7485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for desktop in g_desktops: 7495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if desktop.host_proc: 7505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) desktop.host_proc.send_signal(signal.SIGTERM) 7515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) else: 7525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Exit cleanly so the atexit handler, cleanup(), gets called. 7535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) raise SystemExit 7545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class RelaunchInhibitor: 7575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) """Helper class for inhibiting launch of a child process before a timeout has 7585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) elapsed. 7595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) A managed process can be in one of these states: 7615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) running, not inhibited (running == True) 7625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) stopped and inhibited (running == False and is_inhibited() == True) 7635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) stopped but not inhibited (running == False and is_inhibited() == False) 7645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Attributes: 7665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) label: Name of the tracked process. Only used for logging. 7675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) running: Whether the process is currently running. 7685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) earliest_relaunch_time: Time before which the process should not be 7695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) relaunched, or 0 if there is no limit. 7705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) failures: The number of times that the process ran for less than a 7715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) specified timeout, and had to be inhibited. This count is reset to 0 7725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) whenever the process has run for longer than the timeout. 7735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) """ 7745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def __init__(self, label): 7765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.label = label 7775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.running = False 7785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.earliest_relaunch_time = 0 7792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self.earliest_successful_termination = 0 7805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.failures = 0 7815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def is_inhibited(self): 7835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return (not self.running) and (time.time() < self.earliest_relaunch_time) 7845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) def record_started(self, minimum_lifetime, relaunch_delay): 7865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) """Record that the process was launched, and set the inhibit time to 7875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) |timeout| seconds in the future.""" 7882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self.earliest_relaunch_time = time.time() + relaunch_delay 7892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self.earliest_successful_termination = time.time() + minimum_lifetime 7905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.running = True 7915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def record_stopped(self): 7935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) """Record that the process was stopped, and adjust the failure count 7945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) depending on whether the process ran long enough.""" 7955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.running = False 7962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if time.time() < self.earliest_successful_termination: 7975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.failures += 1 7985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) else: 7995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.failures = 0 8005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.info("Failure count for '%s' is now %d", self.label, self.failures) 8015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)def relaunch_self(): 8045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) cleanup() 8055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) os.execvp(sys.argv[0], sys.argv) 8065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)def waitpid_with_timeout(pid, deadline): 8095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) """Wrapper around os.waitpid() which waits until either a child process dies 8105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) or the deadline elapses. 8115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Args: 8135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) pid: Process ID to wait for, or -1 to wait for any child process. 8145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) deadline: Waiting stops when time.time() exceeds this value. 8155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Returns: 8175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) (pid, status): Same as for os.waitpid(), except that |pid| is 0 if no child 8185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) changed state within the timeout. 8195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Raises: 8215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Same as for os.waitpid(). 8225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) """ 8235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) while time.time() < deadline: 8245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) pid, status = os.waitpid(pid, os.WNOHANG) 8255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if pid != 0: 8265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return (pid, status) 8275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) time.sleep(1) 8285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return (0, 0) 8295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)def waitpid_handle_exceptions(pid, deadline): 8325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) """Wrapper around os.waitpid()/waitpid_with_timeout(), which waits until 8335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) either a child process exits or the deadline elapses, and retries if certain 8345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) exceptions occur. 8355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Args: 8375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) pid: Process ID to wait for, or -1 to wait for any child process. 8385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) deadline: If non-zero, waiting stops when time.time() exceeds this value. 8395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) If zero, waiting stops when a child process exits. 8405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Returns: 8425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) (pid, status): Same as for waitpid_with_timeout(). |pid| is non-zero if and 8435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) only if a child exited during the wait. 8445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Raises: 8465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Same as for os.waitpid(), except: 8475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) OSError with errno==EINTR causes the wait to be retried (this can happen, 8485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for example, if this parent process receives SIGHUP). 8495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) OSError with errno==ECHILD means there are no child processes, and so 8505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) this function sleeps until |deadline|. If |deadline| is zero, this is an 8515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) error and the OSError exception is raised in this case. 8525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) """ 8535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) while True: 8545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) try: 8555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if deadline == 0: 8565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) pid_result, status = os.waitpid(pid, 0) 8575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) else: 8585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) pid_result, status = waitpid_with_timeout(pid, deadline) 8595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return (pid_result, status) 8605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) except OSError, e: 8615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if e.errno == errno.EINTR: 8625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) continue 8635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) elif e.errno == errno.ECHILD: 8645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) now = time.time() 8655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if deadline == 0: 8665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # No time-limit and no child processes. This is treated as an error 8675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # (see docstring). 8685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) raise 8695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) elif deadline > now: 8705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) time.sleep(deadline - now) 8715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return (0, 0) 8725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) else: 8735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Anything else is an unexpected error. 8745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) raise 8755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)def main(): 8785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EPILOG = """This script is not intended for use by end-users. To configure 8795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)Chrome Remote Desktop, please install the app from the Chrome 8805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)Web Store: https://chrome.google.com/remotedesktop""" 8815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) parser = optparse.OptionParser( 8825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) usage="Usage: %prog [options] [ -- [ X server options ] ]", 8835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) epilog=EPILOG) 8845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) parser.add_option("-s", "--size", dest="size", action="append", 8852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) help="Dimensions of virtual desktop. This can be specified " 8862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) "multiple times to make multiple screen resolutions " 8872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) "available (if the Xvfb server supports this).") 8885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) parser.add_option("-f", "--foreground", dest="foreground", default=False, 8895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) action="store_true", 8905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) help="Don't run as a background daemon.") 8915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) parser.add_option("", "--start", dest="start", default=False, 8925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) action="store_true", 8935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) help="Start the host.") 8945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) parser.add_option("-k", "--stop", dest="stop", default=False, 8955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) action="store_true", 8965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) help="Stop the daemon currently running.") 8975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) parser.add_option("", "--check-running", dest="check_running", default=False, 8985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) action="store_true", 8995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) help="Return 0 if the daemon is running, or 1 otherwise.") 9002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) parser.add_option("", "--config", dest="config", action="store", 9012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) help="Use the specified configuration file.") 9025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) parser.add_option("", "--reload", dest="reload", default=False, 9035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) action="store_true", 9045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) help="Signal currently running host to reload the config.") 9055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) parser.add_option("", "--add-user", dest="add_user", default=False, 9065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) action="store_true", 9075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) help="Add current user to the chrome-remote-desktop group.") 9085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) parser.add_option("", "--host-version", dest="host_version", default=False, 9095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) action="store_true", 9105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) help="Prints version of the host.") 9115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) (options, args) = parser.parse_args() 9125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # Determine the filename of the host configuration and PID files. 9142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if not options.config: 9152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) options.config = os.path.join(CONFIG_DIR, "host#%s.json" % g_host_hash) 9165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Check for a modal command-line option (start, stop, etc.) 9185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if options.check_running: 9192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) pid = get_daemon_pid() 9202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return 0 if pid != 0 else 1 9215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if options.stop: 9232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) pid = get_daemon_pid() 9242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if pid == 0: 9255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) print "The daemon is not currently running" 9265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) else: 9275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) print "Killing process %s" % pid 9285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) os.kill(pid, signal.SIGTERM) 9295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 0 9305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if options.reload: 9322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) pid = get_daemon_pid() 9332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if pid == 0: 9345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 1 9355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) os.kill(pid, signal.SIGHUP) 9365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 0 9375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if options.add_user: 9392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if os.getenv("DISPLAY"): 9402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) sudo_command = "gksudo --description \"Chrome Remote Desktop\"" 9412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) else: 9422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) sudo_command = "sudo" 9432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) command = ("sudo -k && exec %(sudo)s -- sh -c " 9445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "\"groupadd -f %(group)s && gpasswd --add %(user)s %(group)s\"" % 9455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) { 'group': CHROME_REMOTING_GROUP_NAME, 9465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 'user': getpass.getuser(), 9475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 'sudo': sudo_command }) 9482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) os.execv("/bin/sh", ["/bin/sh", "-c", command]) 9492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return 1 9505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if options.host_version: 9525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # TODO(sergeyu): Also check RPM package version once we add RPM package. 9535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return os.system(locate_executable(HOST_BINARY_NAME) + " --version") >> 8 9545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if not options.start: 9565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # If no modal command-line options specified, print an error and exit. 9575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) print >> sys.stderr, EPILOG 9585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 1 9595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # Collate the list of sizes that XRANDR should support. 9615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if not options.size: 9622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) default_sizes = DEFAULT_SIZES 9632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if os.environ.has_key(DEFAULT_SIZES_ENV_VAR): 9642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) default_sizes = os.environ[DEFAULT_SIZES_ENV_VAR] 9652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) options.size = default_sizes.split(",") 9665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) sizes = [] 9685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for size in options.size: 9695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_components = size.split("x") 9705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if len(size_components) != 2: 9715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) parser.error("Incorrect size format '%s', should be WIDTHxHEIGHT" % size) 9725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) try: 9745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) width = int(size_components[0]) 9755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) height = int(size_components[1]) 9765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Enforce minimum desktop size, as a sanity-check. The limit of 100 will 9785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # detect typos of 2 instead of 3 digits. 9795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if width < 100 or height < 100: 9805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) raise ValueError 9815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) except ValueError: 9825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) parser.error("Width and height should be 100 pixels or greater") 9835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) sizes.append((width, height)) 9855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # Register an exit handler to clean up session process and the PID file. 9875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) atexit.register(cleanup) 9885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # Load the initial host configuration. 9902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) host_config = Config(options.config) 9912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) try: 9922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) host_config.load() 9932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) except (IOError, ValueError) as e: 9942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) print >> sys.stderr, "Failed to load config: " + str(e) 9952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return 1 9965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # Register handler to re-load the configuration in response to signals. 9987d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) for s in [signal.SIGHUP, signal.SIGINT, signal.SIGTERM]: 9995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) signal.signal(s, SignalHandler(host_config)) 10005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 10012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # Verify that the initial host configuration has the necessary fields. 10025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) auth = Authentication() 10035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) auth_config_valid = auth.copy_from(host_config) 10045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) host = Host() 10055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) host_config_valid = host.copy_from(host_config) 10065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if not host_config_valid or not auth_config_valid: 10075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.error("Failed to load host configuration.") 10085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 1 10095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 10102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # Determine whether a desktop is already active for the specified host 10112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # host configuration. 10122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) pid = get_daemon_pid() 10132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if pid != 0: 10145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Debian policy requires that services should "start" cleanly and return 0 10155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # if they are already running. 10165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) print "Service already running." 10175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 0 10185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 10192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # Detach a separate "daemon" process to run the session, unless specifically 10202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # requested to run in the foreground. 10215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if not options.foreground: 10227d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) daemonize() 10235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 10245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.info("Using host_id: " + host.host_id) 10255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 10265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) desktop = Desktop(sizes) 10275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 10285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Keep track of the number of consecutive failures of any child process to 10295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # run for longer than a set period of time. The script will exit after a 10305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # threshold is exceeded. 10315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # There is no point in tracking the X session process separately, since it is 10325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # launched at (roughly) the same time as the X server, and the termination of 10335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # one of these triggers the termination of the other. 10345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) x_server_inhibitor = RelaunchInhibitor("X server") 10355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) host_inhibitor = RelaunchInhibitor("host") 10365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) all_inhibitors = [x_server_inhibitor, host_inhibitor] 10375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 10385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Don't allow relaunching the script on the first loop iteration. 10395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) allow_relaunch_self = False 10405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 10415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) while True: 10422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # Set the backoff interval and exit if a process failed too many times. 10432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) backoff_time = SHORT_BACKOFF_TIME 10445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for inhibitor in all_inhibitors: 10455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if inhibitor.failures >= MAX_LAUNCH_FAILURES: 10465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.error("Too many launch failures of '%s', exiting." 10475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) % inhibitor.label) 10485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 1 10492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) elif inhibitor.failures >= SHORT_BACKOFF_THRESHOLD: 10502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) backoff_time = LONG_BACKOFF_TIME 10515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 10525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) relaunch_times = [] 10535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 10545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # If the session process or X server stops running (e.g. because the user 10555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # logged out), kill the other. This will trigger the next conditional block 10565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # as soon as os.waitpid() reaps its exit-code. 10575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if desktop.session_proc is None and desktop.x_proc is not None: 10585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.info("Terminating X server") 10595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) desktop.x_proc.terminate() 10605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) elif desktop.x_proc is None and desktop.session_proc is not None: 10615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.info("Terminating X session") 10625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) desktop.session_proc.terminate() 10635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) elif desktop.x_proc is None and desktop.session_proc is None: 10645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Both processes have terminated. 10655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (allow_relaunch_self and x_server_inhibitor.failures == 0 and 10665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) host_inhibitor.failures == 0): 10675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Since the user's desktop is already gone at this point, there's no 10685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # state to lose and now is a good time to pick up any updates to this 10695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # script that might have been installed. 10705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.info("Relaunching self") 10715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) relaunch_self() 10725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) else: 10735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # If there is a non-zero |failures| count, restarting the whole script 10745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # would lose this information, so just launch the session as normal. 10755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if x_server_inhibitor.is_inhibited(): 10765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.info("Waiting before launching X server") 10775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) relaunch_times.append(x_server_inhibitor.earliest_relaunch_time) 10785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) else: 10795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.info("Launching X server and X session.") 10805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) desktop.launch_session(args) 10812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) x_server_inhibitor.record_started(MINIMUM_PROCESS_LIFETIME, 10822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) backoff_time) 10835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) allow_relaunch_self = True 10845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 10855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if desktop.host_proc is None: 10865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if host_inhibitor.is_inhibited(): 10875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.info("Waiting before launching host process") 10885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) relaunch_times.append(host_inhibitor.earliest_relaunch_time) 10895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) else: 10905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.info("Launching host process") 10915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) desktop.launch_host(host_config) 10922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) host_inhibitor.record_started(MINIMUM_PROCESS_LIFETIME, 10932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) backoff_time) 10945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 10955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) deadline = min(relaunch_times) if relaunch_times else 0 10965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) pid, status = waitpid_handle_exceptions(-1, deadline) 10975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if pid == 0: 10985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) continue 10995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 11005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.info("wait() returned (%s,%s)" % (pid, status)) 11015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 11025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # When a process has terminated, and we've reaped its exit-code, any Popen 11035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # instance for that process is no longer valid. Reset any affected instance 11045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # to None. 11055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if desktop.x_proc is not None and pid == desktop.x_proc.pid: 11065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.info("X server process terminated") 11075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) desktop.x_proc = None 11085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) x_server_inhibitor.record_stopped() 11095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 11105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if desktop.session_proc is not None and pid == desktop.session_proc.pid: 11115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.info("Session process terminated") 11125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) desktop.session_proc = None 11135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 11145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if desktop.host_proc is not None and pid == desktop.host_proc.pid: 11155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.info("Host process terminated") 11165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) desktop.host_proc = None 11177d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) desktop.host_ready = False 11185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) host_inhibitor.record_stopped() 11195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 11205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # These exit-codes must match the ones used by the host. 11215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # See remoting/host/host_error_codes.h. 11225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Delete the host or auth configuration depending on the returned error 11235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # code, so the next time this script is run, a new configuration 11245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # will be created and registered. 11257d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) if os.WIFEXITED(status): 11267d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) if os.WEXITSTATUS(status) == 100: 11277d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) logging.info("Host configuration is invalid - exiting.") 11287d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) return 0 11297d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) elif os.WEXITSTATUS(status) == 101: 11307d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) logging.info("Host ID has been deleted - exiting.") 1131a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) host_config.clear() 11327d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) host_config.save_and_log_errors() 11337d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) return 0 11347d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) elif os.WEXITSTATUS(status) == 102: 11357d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) logging.info("OAuth credentials are invalid - exiting.") 11367d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) return 0 11377d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) elif os.WEXITSTATUS(status) == 103: 11387d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) logging.info("Host domain is blocked by policy - exiting.") 11397d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) return 0 11407d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) # Nothing to do for Mac-only status 104 (login screen unsupported) 11417d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) elif os.WEXITSTATUS(status) == 105: 11427d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) logging.info("Username is blocked by policy - exiting.") 11437d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) return 0 11447d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) else: 11457d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) logging.info("Host exited with status %s." % os.WEXITSTATUS(status)) 11467d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) elif os.WIFSIGNALED(status): 11477d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) logging.info("Host terminated by signal %s." % os.WTERMSIG(status)) 11485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 11495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 11505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)if __name__ == "__main__": 11515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.basicConfig(level=logging.DEBUG, 11525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) format="%(asctime)s:%(levelname)s:%(message)s") 11535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) sys.exit(main()) 1154