1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
|
#!/usr/bin/python2
#
# Copyright 2010~2015 Google Inc. All Rights Reserved.
"""Script to get past the login screen of ChromeOS.
"""
from __future__ import print_function
import argparse
import os
import sys
import tempfile
from cros_utils import command_executer
LOGIN_PROMPT_VISIBLE_MAGIC_FILE = '/tmp/uptime-login-prompt-visible'
LOGGED_IN_MAGIC_FILE = '/var/run/state/logged-in'
script_header = """
import os
import autox
import time
"""
wait_for_login_screen = """
while True:
print 'Waiting for login screen to appear...'
if os.path.isfile('%s'):
break
time.sleep(1)
print 'Done'
time.sleep(20)
""" % LOGIN_PROMPT_VISIBLE_MAGIC_FILE
do_login = """
xauth_filename = '/home/chronos/.Xauthority'
os.environ.setdefault('XAUTHORITY', xauth_filename)
os.environ.setdefault('DISPLAY', ':0.0')
print 'Now sending the hotkeys for logging in.'
ax = autox.AutoX()
# navigate to login screen
ax.send_hotkey('Ctrl+Shift+q')
ax.send_hotkey('Ctrl+Alt+l')
# escape out of any login screen menus (e.g., the network select menu)
time.sleep(2)
ax.send_hotkey('Escape')
time.sleep(2)
ax.send_hotkey('Tab')
time.sleep(0.5)
ax.send_hotkey('Tab')
time.sleep(0.5)
ax.send_hotkey('Tab')
time.sleep(0.5)
ax.send_hotkey('Tab')
time.sleep(0.5)
ax.send_hotkey('Return')
print 'Waiting for Chrome to appear...'
while True:
if os.path.isfile('%s'):
break
time.sleep(1)
print 'Done'
""" % LOGGED_IN_MAGIC_FILE
def RestartUI(remote, chromeos_root, login=True):
chromeos_root = os.path.expanduser(chromeos_root)
ce = command_executer.GetCommandExecuter()
# First, restart ui.
command = 'rm -rf %s && restart ui' % LOGIN_PROMPT_VISIBLE_MAGIC_FILE
ce.CrosRunCommand(command, machine=remote, chromeos_root=chromeos_root)
host_login_script = tempfile.mktemp()
device_login_script = '/tmp/login.py'
login_script_list = [script_header, wait_for_login_screen]
if login:
login_script_list.append(do_login)
full_login_script_contents = '\n'.join(login_script_list)
with open(host_login_script, 'w') as f:
f.write(full_login_script_contents)
ce.CopyFiles(host_login_script,
device_login_script,
dest_machine=remote,
chromeos_root=chromeos_root,
recursive=False,
dest_cros=True)
ret = ce.CrosRunCommand('python %s' % device_login_script,
chromeos_root=chromeos_root,
machine=remote)
if os.path.exists(host_login_script):
os.remove(host_login_script)
return ret
def Main(argv):
"""The main function."""
parser = argparse.ArgumentParser()
parser.add_argument('-r',
'--remote',
dest='remote',
help='The remote ChromeOS box.')
parser.add_argument('-c',
'--chromeos_root',
dest='chromeos_root',
help='The ChromeOS root.')
options = parser.parse_args(argv)
return RestartUI(options.remote, options.chromeos_root)
if __name__ == '__main__':
retval = Main(sys.argv[1:])
sys.exit(retval)
|
