me2me_virtual_host.py revision 2a99a7e74a7f215066514fe81d2bfa6639d9eddd
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 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)import getpass 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)import hashlib 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)import json 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)import logging 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)import optparse 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)import os 202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)import pipes 212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)import psutil 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)import signal 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)import socket 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)import subprocess 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)import sys 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)import tempfile 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)import time 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)import uuid 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)LOG_FILE_ENV_VAR = "CHROME_REMOTE_DESKTOP_LOG_FILE" 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)# This script has a sensible default for the initial and maximum desktop size, 332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)# which can be overridden either on the command-line, or via a comma-separated 342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)# list of sizes in this environment variable. 352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)DEFAULT_SIZES_ENV_VAR = "CHROME_REMOTE_DESKTOP_DEFAULT_DESKTOP_SIZES" 362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)# By default, provide a relatively small size to handle the case where resize- 382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)# to-client is disabled, and a much larger size to support clients with large 392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)# or mulitple monitors. These defaults can be overridden in ~/.profile. 402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)DEFAULT_SIZES = "1600x1200,3840x1600" 412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)SCRIPT_PATH = sys.path[0] 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)DEFAULT_INSTALL_PATH = "/opt/google/chrome-remote-desktop" 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)if SCRIPT_PATH == DEFAULT_INSTALL_PATH: 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HOST_BINARY_NAME = "chrome-remote-desktop-host" 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)else: 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HOST_BINARY_NAME = "remoting_me2me_host" 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)CHROME_REMOTING_GROUP_NAME = "chrome-remote-desktop" 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)CONFIG_DIR = os.path.expanduser("~/.config/chrome-remote-desktop") 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)HOME_DIR = os.environ["HOME"] 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)X_LOCK_FILE_TEMPLATE = "/tmp/.X%d-lock" 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)FIRST_X_DISPLAY_NUMBER = 20 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)# Amount of time to wait between relaunching processes. 592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)SHORT_BACKOFF_TIME = 5 602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)LONG_BACKOFF_TIME = 60 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)# How long a process must run in order not to be counted against the restart 632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)# thresholds. 642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)MINIMUM_PROCESS_LIFETIME = 60 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)# Thresholds for switching from fast- to slow-restart and for giving up 672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)# trying to restart entirely. 682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)SHORT_BACKOFF_THRESHOLD = 5 692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)MAX_LAUNCH_FAILURES = SHORT_BACKOFF_THRESHOLD + 10 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)# Globals needed by the atexit cleanup() handler. 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)g_desktops = [] 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)g_host_hash = hashlib.md5(socket.gethostname()).hexdigest() 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class Config: 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def __init__(self, path): 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.path = path 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.data = {} 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.changed = False 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def load(self): 822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) """Loads the config from file. 832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) Raises: 852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IOError: Error reading data 862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ValueError: Error parsing JSON 872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) """ 882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) settings_file = open(self.path, 'r') 892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self.data = json.load(settings_file) 902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self.changed = False 912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) settings_file.close() 925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def save(self): 942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) """Saves the config to file. 952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) Raises: 972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IOError: Error writing data 982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) TypeError: Error serialising JSON 992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) """ 1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if not self.changed: 1012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return 1022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) old_umask = os.umask(0066) 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) try: 1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) settings_file = open(self.path, 'w') 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) settings_file.write(json.dumps(self.data, indent=2)) 1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) settings_file.close() 1072a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self.changed = False 1082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) finally: 1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) os.umask(old_umask) 1102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) def save_and_log_errors(self): 1122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) """Calls self.save(), trapping and logging any errors.""" 1132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) try: 1142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self.save() 1152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) except (IOError, TypeError) as e: 1162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) logging.error("Failed to save config: " + str(e)) 1175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def get(self, key): 1195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return self.data.get(key) 1205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def __getitem__(self, key): 1225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return self.data[key] 1235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def __setitem__(self, key, value): 1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.data[key] = value 1265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.changed = True 1275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def clear_auth(self): 1295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) del self.data["xmpp_login"] 1305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) del self.data["oauth_refresh_token"] 1315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def clear_host_info(self): 1335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) del self.data["host_id"] 1345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) del self.data["host_name"] 1355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) del self.data["host_secret_hash"] 1365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) del self.data["private_key"] 1375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class Authentication: 1405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) """Manage authentication tokens for Chromoting/xmpp""" 1415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def __init__(self): 1435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.login = None 1445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.oauth_refresh_token = None 1455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def copy_from(self, config): 1475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) """Loads the config and returns false if the config is invalid.""" 1485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) try: 1495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.login = config["xmpp_login"] 1505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.oauth_refresh_token = config["oauth_refresh_token"] 1515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) except KeyError: 1525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return False 1535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return True 1545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def copy_to(self, config): 1565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) config["xmpp_login"] = self.login 1575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) config["oauth_refresh_token"] = self.oauth_refresh_token 1585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class Host: 1615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) """This manages the configuration for a host.""" 1625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def __init__(self): 1645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.host_id = str(uuid.uuid1()) 1655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.host_name = socket.gethostname() 1665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.host_secret_hash = None 1675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.private_key = None 1685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def copy_from(self, config): 1705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) try: 1715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.host_id = config["host_id"] 1725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.host_name = config["host_name"] 1735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.host_secret_hash = config.get("host_secret_hash") 1745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.private_key = config["private_key"] 1755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) except KeyError: 1765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return False 1775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return True 1785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def copy_to(self, config): 1805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) config["host_id"] = self.host_id 1815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) config["host_name"] = self.host_name 1825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) config["host_secret_hash"] = self.host_secret_hash 1835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) config["private_key"] = self.private_key 1845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class Desktop: 1875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) """Manage a single virtual desktop""" 1885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def __init__(self, sizes): 1905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.x_proc = None 1915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.session_proc = None 1925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.host_proc = None 1935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.child_env = None 1945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.sizes = sizes 1955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.pulseaudio_pipe = None 1965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.server_supports_exact_resize = False 1975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) g_desktops.append(self) 1985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) @staticmethod 2005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def get_unused_display_number(): 2015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) """Return a candidate display number for which there is currently no 2025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) X Server lock file""" 2035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) display = FIRST_X_DISPLAY_NUMBER 2045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) while os.path.exists(X_LOCK_FILE_TEMPLATE % display): 2055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) display += 1 2065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return display 2075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def _init_child_env(self): 2095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Create clean environment for new session, so it is cleanly separated from 2105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # the user's console X session. 2115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.child_env = {} 2122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for key in [ 2145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "HOME", 2155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "LANG", 2165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "LOGNAME", 2175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "PATH", 2185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "SHELL", 2195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "USER", 2205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "USERNAME", 2215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LOG_FILE_ENV_VAR]: 2225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if os.environ.has_key(key): 2235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.child_env[key] = os.environ[key] 2245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # Read from /etc/environment if it exists, as it is a standard place to 2262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # store system-wide environment settings. During a normal login, this would 2272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # typically be done by the pam_env PAM module, depending on the local PAM 2282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # configuration. 2292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) env_filename = "/etc/environment" 2302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) try: 2312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) with open(env_filename, "r") as env_file: 2322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) for line in env_file: 2332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) line = line.rstrip("\n") 2342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # Split at the first "=", leaving any further instances in the value. 2352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) key_value_pair = line.split("=", 1) 2362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if len(key_value_pair) == 2: 2372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) key, value = tuple(key_value_pair) 2382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # The file stores key=value assignments, but the value may be 2392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # quoted, so strip leading & trailing quotes from it. 2402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) value = value.strip("'\"") 2412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self.child_env[key] = value 2422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) except IOError: 2432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) logging.info("Failed to read %s, skipping." % env_filename) 2442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def _setup_pulseaudio(self): 2465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.pulseaudio_pipe = None 2475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # pulseaudio uses UNIX sockets for communication. Length of UNIX socket 2495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # name is limited to 108 characters, so audio will not work properly if 2505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # the path is too long. To workaround this problem we use only first 10 2515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # symbols of the host hash. 2525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) pulse_path = os.path.join(CONFIG_DIR, 2535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "pulseaudio#%s" % g_host_hash[0:10]) 2545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if len(pulse_path) + len("/native") >= 108: 2555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.error("Audio will not be enabled because pulseaudio UNIX " + 2565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "socket path is too long.") 2575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return False 2585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) sink_name = "chrome_remote_desktop_session" 2605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) pipe_name = os.path.join(pulse_path, "fifo_output") 2615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) try: 2635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if not os.path.exists(pulse_path): 2645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) os.mkdir(pulse_path) 2655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if not os.path.exists(pipe_name): 2665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) os.mkfifo(pipe_name) 2675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) except IOError, e: 2685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.error("Failed to create pulseaudio pipe: " + str(e)) 2695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return False 2705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) try: 2725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) pulse_config = open(os.path.join(pulse_path, "default.pa"), "w") 2735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) pulse_config.write("load-module module-native-protocol-unix\n") 2745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) pulse_config.write( 2755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ("load-module module-pipe-sink sink_name=%s file=\"%s\" " + 2765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "rate=48000 channels=2 format=s16le\n") % 2775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) (sink_name, pipe_name)) 2785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) pulse_config.close() 2795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) except IOError, e: 2805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.error("Failed to write pulseaudio config: " + str(e)) 2815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return False 2825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.child_env["PULSE_CONFIG_PATH"] = pulse_path 2845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.child_env["PULSE_RUNTIME_PATH"] = pulse_path 2855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.child_env["PULSE_STATE_PATH"] = pulse_path 2865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.child_env["PULSE_SINK"] = sink_name 2875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.pulseaudio_pipe = pipe_name 2885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return True 2905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def _launch_x_server(self, extra_x_args): 2922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) x_auth_file = os.path.expanduser("~/.Xauthority") 2932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self.child_env["XAUTHORITY"] = x_auth_file 2945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) devnull = open(os.devnull, "rw") 2955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) display = self.get_unused_display_number() 2962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # Run "xauth add" with |child_env| so that it modifies the same XAUTHORITY 2982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # file which will be used for the X session. 2995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ret_code = subprocess.call("xauth add :%d . `mcookie`" % display, 3002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) env=self.child_env, shell=True) 3015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if ret_code != 0: 3025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) raise Exception("xauth failed with code %d" % ret_code) 3035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) max_width = max([width for width, height in self.sizes]) 3055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) max_height = max([height for width, height in self.sizes]) 3065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) try: 3085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # TODO(jamiewalch): This script expects to be installed alongside 3095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Xvfb-randr, but that's no longer the case. Fix this once we have 3105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # a Xvfb-randr package that installs somewhere sensible. 3115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) xvfb = "/usr/bin/Xvfb-randr" 3125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if not os.path.exists(xvfb): 3135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) xvfb = locate_executable("Xvfb-randr") 3145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.server_supports_exact_resize = True 3155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) except Exception: 3165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) xvfb = "Xvfb" 3175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.server_supports_exact_resize = False 3185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.info("Starting %s on display :%d" % (xvfb, display)) 3205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) screen_option = "%dx%dx24" % (max_width, max_height) 3212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self.x_proc = subprocess.Popen( 3222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) [xvfb, ":%d" % display, 3232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) "-auth", x_auth_file, 3242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) "-nolisten", "tcp", 3252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) "-noreset", 3262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) "-screen", "0", screen_option 3272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ] + extra_x_args) 3285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if not self.x_proc.pid: 3295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) raise Exception("Could not start Xvfb.") 3305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.child_env["DISPLAY"] = ":%d" % display 3325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.child_env["CHROME_REMOTE_DESKTOP_SESSION"] = "1" 3335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Wait for X to be active. 3355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for _test in range(5): 3365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) proc = subprocess.Popen("xdpyinfo", env=self.child_env, stdout=devnull) 3375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) _pid, retcode = os.waitpid(proc.pid, 0) 3385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if retcode == 0: 3395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) break 3405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) time.sleep(0.5) 3415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if retcode != 0: 3425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) raise Exception("Could not connect to Xvfb.") 3435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) else: 3445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.info("Xvfb is active.") 3455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # The remoting host expects the server to use "evdev" keycodes, but Xvfb 3475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # starts configured to use the "base" ruleset, resulting in XKB configuring 3485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # for "xfree86" keycodes, and screwing up some keys. See crbug.com/119013. 3495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Reconfigure the X server to use "evdev" keymap rules. The X server must 3505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # be started with -noreset otherwise it'll reset as soon as the command 3515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # completes, since there are no other X clients running yet. 3525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) proc = subprocess.Popen("setxkbmap -rules evdev", env=self.child_env, 3535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) shell=True) 3545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) _pid, retcode = os.waitpid(proc.pid, 0) 3555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if retcode != 0: 3565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.error("Failed to set XKB to 'evdev'") 3575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Register the screen sizes if the X server's RANDR extension supports it. 3595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Errors here are non-fatal; the X server will continue to run with the 3605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # dimensions from the "-screen" option. 3615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for width, height in self.sizes: 3625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) label = "%dx%d" % (width, height) 3635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) args = ["xrandr", "--newmode", label, "0", str(width), "0", "0", "0", 3645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) str(height), "0", "0", "0"] 3652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) subprocess.call(args, env=self.child_env, stdout=devnull, stderr=devnull) 3665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) args = ["xrandr", "--addmode", "screen", label] 3672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) subprocess.call(args, env=self.child_env, stdout=devnull, stderr=devnull) 3685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Set the initial mode to the first size specified, otherwise the X server 3705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # would default to (max_width, max_height), which might not even be in the 3715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # list. 3725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) label = "%dx%d" % self.sizes[0] 3735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) args = ["xrandr", "-s", label] 3742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) subprocess.call(args, env=self.child_env, stdout=devnull, stderr=devnull) 3752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # Set the physical size of the display so that the initial mode is running 3772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # at approximately 96 DPI, since some desktops require the DPI to be set to 3782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # something realistic. 3792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) args = ["xrandr", "--dpi", "96"] 3802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) subprocess.call(args, env=self.child_env, stdout=devnull, stderr=devnull) 3815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) devnull.close() 3835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def _launch_x_session(self): 3852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # Start desktop session. 3865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # The /dev/null input redirection is necessary to prevent the X session 3875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # reading from stdin. If this code runs as a shell background job in a 3885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # terminal, any reading from stdin causes the job to be suspended. 3895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Daemonization would solve this problem by separating the process from the 3905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # controlling terminal. 3912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) xsession_command = choose_x_session() 3922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if xsession_command is None: 3932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) raise Exception("Unable to choose suitable X session command.") 3942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) logging.info("Launching X session: %s" % xsession_command) 3962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self.session_proc = subprocess.Popen(xsession_command, 3975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) stdin=open(os.devnull, "r"), 3985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) cwd=HOME_DIR, 3995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) env=self.child_env) 4005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if not self.session_proc.pid: 4015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) raise Exception("Could not start X session") 4025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def launch_session(self, x_args): 4045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self._init_child_env() 4055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self._setup_pulseaudio() 4065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self._launch_x_server(x_args) 4075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self._launch_x_session() 4085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def launch_host(self, host_config): 4105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Start remoting host 4115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) args = [locate_executable(HOST_BINARY_NAME), "--host-config=/dev/stdin"] 4125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if self.pulseaudio_pipe: 4135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) args.append("--audio-pipe-name=%s" % self.pulseaudio_pipe) 4145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if self.server_supports_exact_resize: 4155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) args.append("--server-supports-exact-resize") 4165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.host_proc = subprocess.Popen(args, env=self.child_env, 4175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) stdin=subprocess.PIPE) 4185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.info(args) 4195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if not self.host_proc.pid: 4205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) raise Exception("Could not start Chrome Remote Desktop host") 4215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.host_proc.stdin.write(json.dumps(host_config.data)) 4225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.host_proc.stdin.close() 4235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)def get_daemon_pid(): 4262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) """Checks if there is already an instance of this script running, and returns 4272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) its PID. 4285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) Returns: 4302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) The process ID of the existing daemon process, or 0 if the daemon is not 4312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) running. 4325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) """ 4332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) uid = os.getuid() 4342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) this_pid = os.getpid() 4355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) for process in psutil.process_iter(): 4372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # Skip any processes that raise an exception, as processes may terminate 4382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # during iteration over the list. 4392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) try: 4402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # Skip other users' processes. 4412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if process.uids.real != uid: 4422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) continue 4435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # Skip the process for this instance. 4452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if process.pid == this_pid: 4462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) continue 4475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # |cmdline| will be [python-interpreter, script-file, other arguments...] 4492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) cmdline = process.cmdline 4502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if len(cmdline) < 2: 4512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) continue 4522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if cmdline[0] == sys.executable and cmdline[1] == sys.argv[0]: 4532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return process.pid 4542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) except psutil.error.Error: 4552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) continue 4565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return 0 4585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)def choose_x_session(): 4615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) """Chooses the most appropriate X session command for this system. 4625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Returns: 4645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) A string containing the command to run, or a list of strings containing 4655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) the executable program and its arguments, which is suitable for passing as 4665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) the first parameter of subprocess.Popen(). If a suitable session cannot 4675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) be found, returns None. 4685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) """ 4695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # If the session wrapper script (see below) is given a specific session as an 4705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # argument (such as ubuntu-2d on Ubuntu 12.04), the wrapper will run that 4715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # session instead of looking for custom .xsession files in the home directory. 4725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # So it's necessary to test for these files here. 4735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) XSESSION_FILES = [ 4745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "~/.chrome-remote-desktop-session", 4755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "~/.xsession", 4765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "~/.Xsession" ] 4775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for startup_file in XSESSION_FILES: 4785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) startup_file = os.path.expanduser(startup_file) 4795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if os.path.exists(startup_file): 4805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Use the same logic that a Debian system typically uses with ~/.xsession 4815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # (see /etc/X11/Xsession.d/50x11-common_determine-startup), to determine 4825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # exactly how to run this file. 4835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if os.access(startup_file, os.X_OK): 4842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # "/bin/sh -c" is smart about how to execute the session script and 4852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # works in cases where plain exec() fails (for example, if the file is 4862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # marked executable, but is a plain script with no shebang line). 4872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return ["/bin/sh", "-c", pipes.quote(startup_file)] 4885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) else: 4895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) shell = os.environ.get("SHELL", "sh") 4905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return [shell, startup_file] 4915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Choose a session wrapper script to run the session. On some systems, 4935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # /etc/X11/Xsession fails to load the user's .profile, so look for an 4945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # alternative wrapper that is more likely to match the script that the 4955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # system actually uses for console desktop sessions. 4965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SESSION_WRAPPERS = [ 4975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "/usr/sbin/lightdm-session", 4985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "/etc/gdm/Xsession", 4995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "/etc/X11/Xsession" ] 5005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for session_wrapper in SESSION_WRAPPERS: 5015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if os.path.exists(session_wrapper): 5022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if os.path.exists("/usr/bin/unity-2d-panel"): 5032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # On Ubuntu 12.04, the default session relies on 3D-accelerated 5042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # hardware. Trying to run this with a virtual X display produces 5052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # weird results on some systems (for example, upside-down and 5062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # corrupt displays). So if the ubuntu-2d session is available, 5072a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # choose it explicitly. 5082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return [session_wrapper, "/usr/bin/gnome-session --session=ubuntu-2d"] 5092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) else: 5102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # Use the session wrapper by itself, and let the system choose a 5112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # session. 5122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return session_wrapper 5132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return None 5145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)def locate_executable(exe_name): 5175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if SCRIPT_PATH == DEFAULT_INSTALL_PATH: 5185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # If we are installed in the default path, then search the host binary 5195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # only in the same directory. 5205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) paths_to_try = [ DEFAULT_INSTALL_PATH ] 5215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) else: 5225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) paths_to_try = map(lambda p: os.path.join(SCRIPT_PATH, p), 5235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) [".", "../../out/Debug", "../../out/Release" ]) 5245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for path in paths_to_try: 5255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) exe_path = os.path.join(path, exe_name) 5265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if os.path.exists(exe_path): 5275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return exe_path 5285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) raise Exception("Could not locate executable '%s'" % exe_name) 5305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)def daemonize(log_filename): 5335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) """Background this process and detach from controlling terminal, redirecting 5345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) stdout/stderr to |log_filename|.""" 5355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # TODO(lambroslambrou): Having stdout/stderr redirected to a log file is not 5375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # ideal - it could create a filesystem DoS if the daemon or a child process 5385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # were to write excessive amounts to stdout/stderr. Ideally, stdout/stderr 5395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # should be redirected to a pipe or socket, and a process at the other end 5405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # should consume the data and write it to a logging facility which can do 5415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # data-capping or log-rotation. The 'logger' command-line utility could be 5425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # used for this, but it might cause too much syslog spam. 5435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Create new (temporary) file-descriptors before forking, so any errors get 5455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # reported to the main process and set the correct exit-code. 5465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # The mode is provided, since Python otherwise sets a default mode of 0777, 5475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # which would result in the new file having permissions of 0777 & ~umask, 5485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # possibly leaving the executable bits set. 5495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) devnull_fd = os.open(os.devnull, os.O_RDONLY) 5505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) log_fd = os.open(log_filename, os.O_WRONLY | os.O_CREAT | os.O_APPEND, 0600) 5515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) pid = os.fork() 5535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if pid == 0: 5555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Child process 5565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) os.setsid() 5575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # The second fork ensures that the daemon isn't a session leader, so that 5595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # it doesn't acquire a controlling terminal. 5605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) pid = os.fork() 5615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if pid == 0: 5635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Grandchild process 5645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) pass 5655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) else: 5665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Child process 5672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) os._exit(0) # pylint: disable=W0212 5685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) else: 5695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Parent process 5702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) os._exit(0) # pylint: disable=W0212 5715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.info("Daemon process running, logging to '%s'" % log_filename) 5735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) os.chdir(HOME_DIR) 5755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Copy the file-descriptors to create new stdin, stdout and stderr. Note 5775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # that dup2(oldfd, newfd) closes newfd first, so this will close the current 5785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # stdin, stdout and stderr, detaching from the terminal. 5795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) os.dup2(devnull_fd, sys.stdin.fileno()) 5805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) os.dup2(log_fd, sys.stdout.fileno()) 5815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) os.dup2(log_fd, sys.stderr.fileno()) 5825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Close the temporary file-descriptors. 5845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) os.close(devnull_fd) 5855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) os.close(log_fd) 5865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)def cleanup(): 5895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.info("Cleanup.") 5905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) global g_desktops 5925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for desktop in g_desktops: 5935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if desktop.x_proc: 5945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.info("Terminating Xvfb") 5955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) desktop.x_proc.terminate() 5965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) g_desktops = [] 5975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class SignalHandler: 6005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) """Reload the config file on SIGHUP. Since we pass the configuration to the 6015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) host processes via stdin, they can't reload it, so terminate them. They will 6025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) be relaunched automatically with the new config.""" 6035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def __init__(self, host_config): 6055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.host_config = host_config 6065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def __call__(self, signum, _stackframe): 6085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if signum == signal.SIGHUP: 6095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.info("SIGHUP caught, restarting host.") 6102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) try: 6112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self.host_config.load() 6122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) except (IOError, ValueError) as e: 6132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) logging.error("Failed to load config: " + str(e)) 6145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for desktop in g_desktops: 6155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if desktop.host_proc: 6165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) desktop.host_proc.send_signal(signal.SIGTERM) 6175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) else: 6185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Exit cleanly so the atexit handler, cleanup(), gets called. 6195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) raise SystemExit 6205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class RelaunchInhibitor: 6235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) """Helper class for inhibiting launch of a child process before a timeout has 6245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) elapsed. 6255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) A managed process can be in one of these states: 6275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) running, not inhibited (running == True) 6285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) stopped and inhibited (running == False and is_inhibited() == True) 6295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) stopped but not inhibited (running == False and is_inhibited() == False) 6305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Attributes: 6325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) label: Name of the tracked process. Only used for logging. 6335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) running: Whether the process is currently running. 6345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) earliest_relaunch_time: Time before which the process should not be 6355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) relaunched, or 0 if there is no limit. 6365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) failures: The number of times that the process ran for less than a 6375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) specified timeout, and had to be inhibited. This count is reset to 0 6385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) whenever the process has run for longer than the timeout. 6395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) """ 6405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def __init__(self, label): 6425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.label = label 6435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.running = False 6445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.earliest_relaunch_time = 0 6452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self.earliest_successful_termination = 0 6465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.failures = 0 6475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def is_inhibited(self): 6495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return (not self.running) and (time.time() < self.earliest_relaunch_time) 6505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) def record_started(self, minimum_lifetime, relaunch_delay): 6525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) """Record that the process was launched, and set the inhibit time to 6535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) |timeout| seconds in the future.""" 6542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self.earliest_relaunch_time = time.time() + relaunch_delay 6552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self.earliest_successful_termination = time.time() + minimum_lifetime 6565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.running = True 6575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) def record_stopped(self): 6595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) """Record that the process was stopped, and adjust the failure count 6605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) depending on whether the process ran long enough.""" 6615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.running = False 6622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if time.time() < self.earliest_successful_termination: 6635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.failures += 1 6645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) else: 6655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) self.failures = 0 6665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.info("Failure count for '%s' is now %d", self.label, self.failures) 6675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)def relaunch_self(): 6705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) cleanup() 6715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) os.execvp(sys.argv[0], sys.argv) 6725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)def waitpid_with_timeout(pid, deadline): 6755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) """Wrapper around os.waitpid() which waits until either a child process dies 6765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) or the deadline elapses. 6775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Args: 6795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) pid: Process ID to wait for, or -1 to wait for any child process. 6805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) deadline: Waiting stops when time.time() exceeds this value. 6815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Returns: 6835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) (pid, status): Same as for os.waitpid(), except that |pid| is 0 if no child 6845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) changed state within the timeout. 6855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Raises: 6875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Same as for os.waitpid(). 6885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) """ 6895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) while time.time() < deadline: 6905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) pid, status = os.waitpid(pid, os.WNOHANG) 6915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if pid != 0: 6925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return (pid, status) 6935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) time.sleep(1) 6945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return (0, 0) 6955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)def waitpid_handle_exceptions(pid, deadline): 6985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) """Wrapper around os.waitpid()/waitpid_with_timeout(), which waits until 6995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) either a child process exits or the deadline elapses, and retries if certain 7005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) exceptions occur. 7015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Args: 7035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) pid: Process ID to wait for, or -1 to wait for any child process. 7045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) deadline: If non-zero, waiting stops when time.time() exceeds this value. 7055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) If zero, waiting stops when a child process exits. 7065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Returns: 7085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) (pid, status): Same as for waitpid_with_timeout(). |pid| is non-zero if and 7095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) only if a child exited during the wait. 7105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Raises: 7125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Same as for os.waitpid(), except: 7135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) OSError with errno==EINTR causes the wait to be retried (this can happen, 7145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for example, if this parent process receives SIGHUP). 7155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) OSError with errno==ECHILD means there are no child processes, and so 7165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) this function sleeps until |deadline|. If |deadline| is zero, this is an 7175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) error and the OSError exception is raised in this case. 7185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) """ 7195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) while True: 7205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) try: 7215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if deadline == 0: 7225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) pid_result, status = os.waitpid(pid, 0) 7235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) else: 7245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) pid_result, status = waitpid_with_timeout(pid, deadline) 7255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return (pid_result, status) 7265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) except OSError, e: 7275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if e.errno == errno.EINTR: 7285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) continue 7295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) elif e.errno == errno.ECHILD: 7305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) now = time.time() 7315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if deadline == 0: 7325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # No time-limit and no child processes. This is treated as an error 7335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # (see docstring). 7345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) raise 7355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) elif deadline > now: 7365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) time.sleep(deadline - now) 7375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return (0, 0) 7385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) else: 7395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Anything else is an unexpected error. 7405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) raise 7415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)def main(): 7445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EPILOG = """This script is not intended for use by end-users. To configure 7455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)Chrome Remote Desktop, please install the app from the Chrome 7465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)Web Store: https://chrome.google.com/remotedesktop""" 7475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) parser = optparse.OptionParser( 7485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) usage="Usage: %prog [options] [ -- [ X server options ] ]", 7495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) epilog=EPILOG) 7505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) parser.add_option("-s", "--size", dest="size", action="append", 7512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) help="Dimensions of virtual desktop. This can be specified " 7522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) "multiple times to make multiple screen resolutions " 7532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) "available (if the Xvfb server supports this).") 7545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) parser.add_option("-f", "--foreground", dest="foreground", default=False, 7555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) action="store_true", 7565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) help="Don't run as a background daemon.") 7575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) parser.add_option("", "--start", dest="start", default=False, 7585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) action="store_true", 7595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) help="Start the host.") 7605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) parser.add_option("-k", "--stop", dest="stop", default=False, 7615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) action="store_true", 7625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) help="Stop the daemon currently running.") 7635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) parser.add_option("", "--check-running", dest="check_running", default=False, 7645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) action="store_true", 7655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) help="Return 0 if the daemon is running, or 1 otherwise.") 7662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) parser.add_option("", "--config", dest="config", action="store", 7672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) help="Use the specified configuration file.") 7685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) parser.add_option("", "--reload", dest="reload", default=False, 7695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) action="store_true", 7705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) help="Signal currently running host to reload the config.") 7715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) parser.add_option("", "--add-user", dest="add_user", default=False, 7725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) action="store_true", 7735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) help="Add current user to the chrome-remote-desktop group.") 7745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) parser.add_option("", "--host-version", dest="host_version", default=False, 7755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) action="store_true", 7765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) help="Prints version of the host.") 7775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) (options, args) = parser.parse_args() 7785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # Determine the filename of the host configuration and PID files. 7802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if not options.config: 7812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) options.config = os.path.join(CONFIG_DIR, "host#%s.json" % g_host_hash) 7825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Check for a modal command-line option (start, stop, etc.) 7845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if options.check_running: 7852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) pid = get_daemon_pid() 7862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return 0 if pid != 0 else 1 7875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if options.stop: 7892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) pid = get_daemon_pid() 7902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if pid == 0: 7915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) print "The daemon is not currently running" 7925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) else: 7935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) print "Killing process %s" % pid 7945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) os.kill(pid, signal.SIGTERM) 7955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 0 7965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if options.reload: 7982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) pid = get_daemon_pid() 7992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if pid == 0: 8005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 1 8015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) os.kill(pid, signal.SIGHUP) 8025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 0 8035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if options.add_user: 8052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if os.getenv("DISPLAY"): 8062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) sudo_command = "gksudo --description \"Chrome Remote Desktop\"" 8072a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) else: 8082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) sudo_command = "sudo" 8092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) command = ("sudo -k && exec %(sudo)s -- sh -c " 8105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "\"groupadd -f %(group)s && gpasswd --add %(user)s %(group)s\"" % 8115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) { 'group': CHROME_REMOTING_GROUP_NAME, 8125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 'user': getpass.getuser(), 8135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 'sudo': sudo_command }) 8142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) os.execv("/bin/sh", ["/bin/sh", "-c", command]) 8152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return 1 8165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if options.host_version: 8185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # TODO(sergeyu): Also check RPM package version once we add RPM package. 8195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return os.system(locate_executable(HOST_BINARY_NAME) + " --version") >> 8 8205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if not options.start: 8225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # If no modal command-line options specified, print an error and exit. 8235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) print >> sys.stderr, EPILOG 8245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 1 8255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # Collate the list of sizes that XRANDR should support. 8275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if not options.size: 8282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) default_sizes = DEFAULT_SIZES 8292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if os.environ.has_key(DEFAULT_SIZES_ENV_VAR): 8302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) default_sizes = os.environ[DEFAULT_SIZES_ENV_VAR] 8312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) options.size = default_sizes.split(",") 8325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) sizes = [] 8345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for size in options.size: 8355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_components = size.split("x") 8365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if len(size_components) != 2: 8375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) parser.error("Incorrect size format '%s', should be WIDTHxHEIGHT" % size) 8385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) try: 8405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) width = int(size_components[0]) 8415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) height = int(size_components[1]) 8425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Enforce minimum desktop size, as a sanity-check. The limit of 100 will 8445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # detect typos of 2 instead of 3 digits. 8455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if width < 100 or height < 100: 8465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) raise ValueError 8475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) except ValueError: 8485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) parser.error("Width and height should be 100 pixels or greater") 8495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) sizes.append((width, height)) 8515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # Register an exit handler to clean up session process and the PID file. 8535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) atexit.register(cleanup) 8545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # Load the initial host configuration. 8562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) host_config = Config(options.config) 8572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) try: 8582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) host_config.load() 8592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) except (IOError, ValueError) as e: 8602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) print >> sys.stderr, "Failed to load config: " + str(e) 8612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return 1 8625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # Register handler to re-load the configuration in response to signals. 8645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for s in [signal.SIGHUP, signal.SIGINT, signal.SIGTERM, signal.SIGUSR1]: 8655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) signal.signal(s, SignalHandler(host_config)) 8665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # Verify that the initial host configuration has the necessary fields. 8685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) auth = Authentication() 8695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) auth_config_valid = auth.copy_from(host_config) 8705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) host = Host() 8715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) host_config_valid = host.copy_from(host_config) 8725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if not host_config_valid or not auth_config_valid: 8735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.error("Failed to load host configuration.") 8745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 1 8755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # Determine whether a desktop is already active for the specified host 8772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # host configuration. 8782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) pid = get_daemon_pid() 8792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if pid != 0: 8805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Debian policy requires that services should "start" cleanly and return 0 8815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # if they are already running. 8825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) print "Service already running." 8835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 0 8845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # Detach a separate "daemon" process to run the session, unless specifically 8862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # requested to run in the foreground. 8875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if not options.foreground: 8885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if not os.environ.has_key(LOG_FILE_ENV_VAR): 8895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) log_file = tempfile.NamedTemporaryFile( 8905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) prefix="chrome_remote_desktop_", delete=False) 8915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) os.environ[LOG_FILE_ENV_VAR] = log_file.name 8925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) daemonize(os.environ[LOG_FILE_ENV_VAR]) 8935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.info("Using host_id: " + host.host_id) 8955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) desktop = Desktop(sizes) 8975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Keep track of the number of consecutive failures of any child process to 8995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # run for longer than a set period of time. The script will exit after a 9005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # threshold is exceeded. 9015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # There is no point in tracking the X session process separately, since it is 9025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # launched at (roughly) the same time as the X server, and the termination of 9035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # one of these triggers the termination of the other. 9045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) x_server_inhibitor = RelaunchInhibitor("X server") 9055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) host_inhibitor = RelaunchInhibitor("host") 9065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) all_inhibitors = [x_server_inhibitor, host_inhibitor] 9075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Don't allow relaunching the script on the first loop iteration. 9095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) allow_relaunch_self = False 9105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) while True: 9122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # Set the backoff interval and exit if a process failed too many times. 9132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) backoff_time = SHORT_BACKOFF_TIME 9145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for inhibitor in all_inhibitors: 9155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if inhibitor.failures >= MAX_LAUNCH_FAILURES: 9165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.error("Too many launch failures of '%s', exiting." 9175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) % inhibitor.label) 9185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 1 9192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) elif inhibitor.failures >= SHORT_BACKOFF_THRESHOLD: 9202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) backoff_time = LONG_BACKOFF_TIME 9215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) relaunch_times = [] 9235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # If the session process or X server stops running (e.g. because the user 9255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # logged out), kill the other. This will trigger the next conditional block 9265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # as soon as os.waitpid() reaps its exit-code. 9275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if desktop.session_proc is None and desktop.x_proc is not None: 9285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.info("Terminating X server") 9295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) desktop.x_proc.terminate() 9305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) elif desktop.x_proc is None and desktop.session_proc is not None: 9315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.info("Terminating X session") 9325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) desktop.session_proc.terminate() 9335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) elif desktop.x_proc is None and desktop.session_proc is None: 9345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Both processes have terminated. 9355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (allow_relaunch_self and x_server_inhibitor.failures == 0 and 9365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) host_inhibitor.failures == 0): 9375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Since the user's desktop is already gone at this point, there's no 9385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # state to lose and now is a good time to pick up any updates to this 9395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # script that might have been installed. 9405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.info("Relaunching self") 9415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) relaunch_self() 9425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) else: 9435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # If there is a non-zero |failures| count, restarting the whole script 9445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # would lose this information, so just launch the session as normal. 9455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if x_server_inhibitor.is_inhibited(): 9465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.info("Waiting before launching X server") 9475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) relaunch_times.append(x_server_inhibitor.earliest_relaunch_time) 9485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) else: 9495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.info("Launching X server and X session.") 9505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) desktop.launch_session(args) 9512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) x_server_inhibitor.record_started(MINIMUM_PROCESS_LIFETIME, 9522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) backoff_time) 9535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) allow_relaunch_self = True 9545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if desktop.host_proc is None: 9565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if host_inhibitor.is_inhibited(): 9575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.info("Waiting before launching host process") 9585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) relaunch_times.append(host_inhibitor.earliest_relaunch_time) 9595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) else: 9605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.info("Launching host process") 9615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) desktop.launch_host(host_config) 9622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) host_inhibitor.record_started(MINIMUM_PROCESS_LIFETIME, 9632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) backoff_time) 9645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) deadline = min(relaunch_times) if relaunch_times else 0 9665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) pid, status = waitpid_handle_exceptions(-1, deadline) 9675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if pid == 0: 9685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) continue 9695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.info("wait() returned (%s,%s)" % (pid, status)) 9715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # When a process has terminated, and we've reaped its exit-code, any Popen 9735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # instance for that process is no longer valid. Reset any affected instance 9745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # to None. 9755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if desktop.x_proc is not None and pid == desktop.x_proc.pid: 9765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.info("X server process terminated") 9775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) desktop.x_proc = None 9785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) x_server_inhibitor.record_stopped() 9795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if desktop.session_proc is not None and pid == desktop.session_proc.pid: 9815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.info("Session process terminated") 9825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) desktop.session_proc = None 9835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if desktop.host_proc is not None and pid == desktop.host_proc.pid: 9855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.info("Host process terminated") 9865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) desktop.host_proc = None 9875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) host_inhibitor.record_stopped() 9885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # These exit-codes must match the ones used by the host. 9905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # See remoting/host/host_error_codes.h. 9915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Delete the host or auth configuration depending on the returned error 9925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # code, so the next time this script is run, a new configuration 9935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # will be created and registered. 9945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if os.WEXITSTATUS(status) == 100: 9955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.info("Host configuration is invalid - exiting.") 9965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) host_config.clear_auth() 9975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) host_config.clear_host_info() 9982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) host_config.save_and_log_errors() 9995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 0 10005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) elif os.WEXITSTATUS(status) == 101: 10015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.info("Host ID has been deleted - exiting.") 10025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) host_config.clear_host_info() 10032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) host_config.save_and_log_errors() 10045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 0 10055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) elif os.WEXITSTATUS(status) == 102: 10065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.info("OAuth credentials are invalid - exiting.") 10075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) host_config.clear_auth() 10082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) host_config.save_and_log_errors() 10095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 0 10105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) elif os.WEXITSTATUS(status) == 103: 10115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.info("Host domain is blocked by policy - exiting.") 10122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) host_config.clear_auth() 10132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) host_config.clear_host_info() 10142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) host_config.save_and_log_errors() 10155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 0 10165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) # Nothing to do for Mac-only status 104 (login screen unsupported) 10175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) elif os.WEXITSTATUS(status) == 105: 10185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.info("Username is blocked by policy - exiting.") 10192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) host_config.clear_auth() 10202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) host_config.clear_host_info() 10212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) host_config.save_and_log_errors() 10225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 0 10235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 10245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 10255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)if __name__ == "__main__": 10265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) logging.basicConfig(level=logging.DEBUG, 10275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) format="%(asctime)s:%(levelname)s:%(message)s") 10285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) sys.exit(main()) 1029