OLD | NEW |
1 #!/usr/bin/python | 1 #!/usr/bin/python |
2 # Copyright (c) 2012 The Chromium Authors. All rights reserved. | 2 # Copyright (c) 2012 The Chromium Authors. All rights reserved. |
3 # Use of this source code is governed by a BSD-style license that can be | 3 # Use of this source code is governed by a BSD-style license that can be |
4 # found in the LICENSE file. | 4 # found in the LICENSE file. |
5 | 5 |
6 # Virtual Me2Me implementation. This script runs and manages the processes | 6 # Virtual Me2Me implementation. This script runs and manages the processes |
7 # required for a Virtual Me2Me desktop, which are: X server, X desktop | 7 # required for a Virtual Me2Me desktop, which are: X server, X desktop |
8 # session, and Host process. | 8 # session, and Host process. |
9 # This script is intended to run continuously as a background daemon | 9 # This script is intended to run continuously as a background daemon |
10 # process, running under an ordinary (non-root) user account. | 10 # process, running under an ordinary (non-root) user account. |
(...skipping 34 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
45 | 45 |
46 IS_INSTALLED = (os.path.basename(sys.argv[0]) != 'me2me_virtual_host.py') | 46 IS_INSTALLED = (os.path.basename(sys.argv[0]) != 'me2me_virtual_host.py') |
47 | 47 |
48 if IS_INSTALLED: | 48 if IS_INSTALLED: |
49 HOST_BINARY_NAME = "chrome-remote-desktop-host" | 49 HOST_BINARY_NAME = "chrome-remote-desktop-host" |
50 else: | 50 else: |
51 HOST_BINARY_NAME = "remoting_me2me_host" | 51 HOST_BINARY_NAME = "remoting_me2me_host" |
52 | 52 |
53 CHROME_REMOTING_GROUP_NAME = "chrome-remote-desktop" | 53 CHROME_REMOTING_GROUP_NAME = "chrome-remote-desktop" |
54 | 54 |
55 CONFIG_DIR = os.path.expanduser("~/.config/chrome-remote-desktop") | |
56 HOME_DIR = os.environ["HOME"] | 55 HOME_DIR = os.environ["HOME"] |
| 56 CONFIG_DIR = HOME_DIR + "/.config/chrome-remote-desktop" |
| 57 SESSION_FILE_PATH = HOME_DIR + "/.chrome-remote-desktop-session" |
57 | 58 |
58 X_LOCK_FILE_TEMPLATE = "/tmp/.X%d-lock" | 59 X_LOCK_FILE_TEMPLATE = "/tmp/.X%d-lock" |
59 FIRST_X_DISPLAY_NUMBER = 20 | 60 FIRST_X_DISPLAY_NUMBER = 20 |
60 | 61 |
61 # Amount of time to wait between relaunching processes. | 62 # Amount of time to wait between relaunching processes. |
62 SHORT_BACKOFF_TIME = 5 | 63 SHORT_BACKOFF_TIME = 5 |
63 LONG_BACKOFF_TIME = 60 | 64 LONG_BACKOFF_TIME = 60 |
64 | 65 |
65 # How long a process must run in order not to be counted against the restart | 66 # How long a process must run in order not to be counted against the restart |
66 # thresholds. | 67 # thresholds. |
67 MINIMUM_PROCESS_LIFETIME = 60 | 68 MINIMUM_PROCESS_LIFETIME = 60 |
68 | 69 |
69 # Thresholds for switching from fast- to slow-restart and for giving up | 70 # Thresholds for switching from fast- to slow-restart and for giving up |
70 # trying to restart entirely. | 71 # trying to restart entirely. |
71 SHORT_BACKOFF_THRESHOLD = 5 | 72 SHORT_BACKOFF_THRESHOLD = 5 |
72 MAX_LAUNCH_FAILURES = SHORT_BACKOFF_THRESHOLD + 10 | 73 MAX_LAUNCH_FAILURES = SHORT_BACKOFF_THRESHOLD + 10 |
73 | 74 |
74 # Globals needed by the atexit cleanup() handler. | 75 # Globals needed by the atexit cleanup() handler. |
75 g_desktops = [] | 76 g_desktops = [] |
76 g_host_hash = hashlib.md5(socket.gethostname()).hexdigest() | 77 g_host_hash = hashlib.md5(socket.gethostname()).hexdigest() |
77 | 78 |
| 79 def load_lsb_release_file(): |
| 80 result = {} |
| 81 try: |
| 82 for line in open('/etc/lsb-release').readlines(): |
| 83 parts = (line.split('#', 2)[0]).split('=', 2) |
| 84 if len(parts) == 2: |
| 85 result[parts[0].strip()] = parts[1].strip() |
| 86 except IOError as e: |
| 87 logging.error('Failed to load /etc/lsb-release: ' + str(e)) |
| 88 return result |
| 89 |
| 90 def is_supported_platform(): |
| 91 # Always assume that the system is supported if the config directory or |
| 92 # session file exist. |
| 93 if os.path.isdir(CONFIG_DIR) or os.path.isfile(SESSION_FILE_PATH): |
| 94 return True |
| 95 |
| 96 # Currently the host supports only Goobuntu. |
| 97 # TODO(sergeyu): Try other versions of Ubuntu and enable them here. |
| 98 lsb_release = load_lsb_release_file() |
| 99 return (lsb_release.has_key('GOOGLE_CODENAME') and |
| 100 lsb_release['GOOGLE_CODENAME'] == 'precise') |
| 101 |
78 class Config: | 102 class Config: |
79 def __init__(self, path): | 103 def __init__(self, path): |
80 self.path = path | 104 self.path = path |
81 self.data = {} | 105 self.data = {} |
82 self.changed = False | 106 self.changed = False |
83 | 107 |
84 def load(self): | 108 def load(self): |
85 """Loads the config from file. | 109 """Loads the config from file. |
86 | 110 |
87 Raises: | 111 Raises: |
(...skipping 407 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
495 A string containing the command to run, or a list of strings containing | 519 A string containing the command to run, or a list of strings containing |
496 the executable program and its arguments, which is suitable for passing as | 520 the executable program and its arguments, which is suitable for passing as |
497 the first parameter of subprocess.Popen(). If a suitable session cannot | 521 the first parameter of subprocess.Popen(). If a suitable session cannot |
498 be found, returns None. | 522 be found, returns None. |
499 """ | 523 """ |
500 # If the session wrapper script (see below) is given a specific session as an | 524 # If the session wrapper script (see below) is given a specific session as an |
501 # argument (such as ubuntu-2d on Ubuntu 12.04), the wrapper will run that | 525 # argument (such as ubuntu-2d on Ubuntu 12.04), the wrapper will run that |
502 # session instead of looking for custom .xsession files in the home directory. | 526 # session instead of looking for custom .xsession files in the home directory. |
503 # So it's necessary to test for these files here. | 527 # So it's necessary to test for these files here. |
504 XSESSION_FILES = [ | 528 XSESSION_FILES = [ |
505 "~/.chrome-remote-desktop-session", | 529 SESSION_FILE_PATH, |
506 "~/.xsession", | 530 "~/.xsession", |
507 "~/.Xsession" ] | 531 "~/.Xsession" ] |
508 for startup_file in XSESSION_FILES: | 532 for startup_file in XSESSION_FILES: |
509 startup_file = os.path.expanduser(startup_file) | 533 startup_file = os.path.expanduser(startup_file) |
510 if os.path.exists(startup_file): | 534 if os.path.exists(startup_file): |
511 # Use the same logic that a Debian system typically uses with ~/.xsession | 535 # Use the same logic that a Debian system typically uses with ~/.xsession |
512 # (see /etc/X11/Xsession.d/50x11-common_determine-startup), to determine | 536 # (see /etc/X11/Xsession.d/50x11-common_determine-startup), to determine |
513 # exactly how to run this file. | 537 # exactly how to run this file. |
514 if os.access(startup_file, os.X_OK): | 538 if os.access(startup_file, os.X_OK): |
515 # "/bin/sh -c" is smart about how to execute the session script and | 539 # "/bin/sh -c" is smart about how to execute the session script and |
(...skipping 379 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
895 "available (if the Xvfb server supports this).") | 919 "available (if the Xvfb server supports this).") |
896 parser.add_option("-f", "--foreground", dest="foreground", default=False, | 920 parser.add_option("-f", "--foreground", dest="foreground", default=False, |
897 action="store_true", | 921 action="store_true", |
898 help="Don't run as a background daemon.") | 922 help="Don't run as a background daemon.") |
899 parser.add_option("", "--start", dest="start", default=False, | 923 parser.add_option("", "--start", dest="start", default=False, |
900 action="store_true", | 924 action="store_true", |
901 help="Start the host.") | 925 help="Start the host.") |
902 parser.add_option("-k", "--stop", dest="stop", default=False, | 926 parser.add_option("-k", "--stop", dest="stop", default=False, |
903 action="store_true", | 927 action="store_true", |
904 help="Stop the daemon currently running.") | 928 help="Stop the daemon currently running.") |
| 929 parser.add_option("", "--get-status", dest="get_status", default=False, |
| 930 action="store_true", |
| 931 help="Prints host status") |
905 parser.add_option("", "--check-running", dest="check_running", default=False, | 932 parser.add_option("", "--check-running", dest="check_running", default=False, |
906 action="store_true", | 933 action="store_true", |
907 help="Return 0 if the daemon is running, or 1 otherwise.") | 934 help="Return 0 if the daemon is running, or 1 otherwise.") |
908 parser.add_option("", "--config", dest="config", action="store", | 935 parser.add_option("", "--config", dest="config", action="store", |
909 help="Use the specified configuration file.") | 936 help="Use the specified configuration file.") |
910 parser.add_option("", "--reload", dest="reload", default=False, | 937 parser.add_option("", "--reload", dest="reload", default=False, |
911 action="store_true", | 938 action="store_true", |
912 help="Signal currently running host to reload the config.") | 939 help="Signal currently running host to reload the config.") |
913 parser.add_option("", "--add-user", dest="add_user", default=False, | 940 parser.add_option("", "--add-user", dest="add_user", default=False, |
914 action="store_true", | 941 action="store_true", |
915 help="Add current user to the chrome-remote-desktop group.") | 942 help="Add current user to the chrome-remote-desktop group.") |
916 parser.add_option("", "--host-version", dest="host_version", default=False, | 943 parser.add_option("", "--host-version", dest="host_version", default=False, |
917 action="store_true", | 944 action="store_true", |
918 help="Prints version of the host.") | 945 help="Prints version of the host.") |
919 (options, args) = parser.parse_args() | 946 (options, args) = parser.parse_args() |
920 | 947 |
921 # Determine the filename of the host configuration and PID files. | 948 # Determine the filename of the host configuration and PID files. |
922 if not options.config: | 949 if not options.config: |
923 options.config = os.path.join(CONFIG_DIR, "host#%s.json" % g_host_hash) | 950 options.config = os.path.join(CONFIG_DIR, "host#%s.json" % g_host_hash) |
924 | 951 |
925 # Check for a modal command-line option (start, stop, etc.) | 952 # Check for a modal command-line option (start, stop, etc.) |
| 953 |
| 954 if options.get_status: |
| 955 pid = get_daemon_pid() |
| 956 if pid != 0: |
| 957 print "STARTED" |
| 958 elif is_supported_platform(): |
| 959 print "STOPPED" |
| 960 else: |
| 961 print "NOT_IMPLEMENTED" |
| 962 return 0 |
| 963 |
| 964 # TODO(sergeyu): Remove --check-running once NPAPI plugin and NM host are |
| 965 # updated to always use get-status flag instead. |
926 if options.check_running: | 966 if options.check_running: |
927 pid = get_daemon_pid() | 967 pid = get_daemon_pid() |
928 return 0 if pid != 0 else 1 | 968 return 0 if pid != 0 else 1 |
929 | 969 |
930 if options.stop: | 970 if options.stop: |
931 pid = get_daemon_pid() | 971 pid = get_daemon_pid() |
932 if pid == 0: | 972 if pid == 0: |
933 print "The daemon is not currently running" | 973 print "The daemon is not currently running" |
934 else: | 974 else: |
935 print "Killing process %s" % pid | 975 print "Killing process %s" % pid |
(...skipping 225 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
1161 else: | 1201 else: |
1162 logging.info("Host exited with status %s." % os.WEXITSTATUS(status)) | 1202 logging.info("Host exited with status %s." % os.WEXITSTATUS(status)) |
1163 elif os.WIFSIGNALED(status): | 1203 elif os.WIFSIGNALED(status): |
1164 logging.info("Host terminated by signal %s." % os.WTERMSIG(status)) | 1204 logging.info("Host terminated by signal %s." % os.WTERMSIG(status)) |
1165 | 1205 |
1166 | 1206 |
1167 if __name__ == "__main__": | 1207 if __name__ == "__main__": |
1168 logging.basicConfig(level=logging.DEBUG, | 1208 logging.basicConfig(level=logging.DEBUG, |
1169 format="%(asctime)s:%(levelname)s:%(message)s") | 1209 format="%(asctime)s:%(levelname)s:%(message)s") |
1170 sys.exit(main()) | 1210 sys.exit(main()) |
OLD | NEW |