blob: 28bdc1098f4dac662bd9982b6b085851c3fcbef2 [file] [log] [blame]
#!/usr/bin/env python
# Copyright 2012 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""rsyncs goofy and runs on a remote device."""
import argparse
import glob
import logging
import os
import pipes
import re
import sys
import factory_common # pylint: disable=unused-import
from cros.factory.hwid.v3 import hwid_utils
from cros.factory.test.env import paths
from cros.factory.test.test_lists import manager
from cros.factory.test.test_lists import test_list_common
from cros.factory.utils import cros_board_utils
from cros.factory.utils import file_utils
from cros.factory.utils.process_utils import Spawn
from cros.factory.utils.ssh_utils import SpawnRsyncToDUT
from cros.factory.utils.ssh_utils import SpawnSSHToDUT
from cros.factory.utils.sync_utils import Retry
from cros.factory.utils.sys_utils import InChroot
class GoofyRemoteException(Exception):
"""Goofy remote exception."""
def GetBoard(host):'Checking release board on %s...', host)
release = SpawnSSHToDUT([host, 'cat /etc/lsb-release'],
check_output=True, log=True).stdout_data
match ='^CHROMEOS_RELEASE_BOARD=(.+)', release, re.MULTILINE)
if not match:
logging.warn('Unable to determine release board')
return None
def TweakTestLists(args):
"""Tweaks new-style test lists as required by the arguments.
Note that this runs *on the target DUT*, not locally.
args: The arguments from argparse.
for path in glob.glob(os.path.join(test_list_common.TEST_LISTS_PATH, '*.py')):
with open(path) as f:
data =
def SubLine(variable_re, new_value, string):
"""Replaces certain assignments in the test list with a new value.
We'll replace any line that begins with the given variable name
and an equal sign.
variable_re: A regular expression matching the names of variables whose
value is to be replaced.
new_value: The new value of the variable. We wrap this in repr() before
string: The original contents of the test list.
return re.sub(
r'^(\s*' + # Beginning of line
variable_re + # The name of the variable we're looking for
r'\s*=\s*)' +
r'.+', # The old value
# Keep everything but the value in the original string;
# replace that with new_value.
lambda match_obj: + repr(new_value),
new_data = data
if args.clear_password:
new_data = SubLine('options.engineering_password_sha1', None, new_data)
if args.shopfloor_host:
new_data = SubLine('(?:shop_floor_host|shopfloor_host)',
args.shopfloor_host, new_data)
if args.shopfloor_port:
new_data = SubLine('(?:shop_floor_port|shopfloor_port)',
args.shopfloor_port, new_data)
# Write out the file if anything has changed.
if new_data != data:
with open(path, 'w') as f:'Modified %s', path)
if args.test_list:
def main():
parser = argparse.ArgumentParser(
description='Rsync and run Goofy on a remote device.')
parser.add_argument('host', metavar='HOST',
help='host to run on')
parser.add_argument('-a', dest='clear_state', action='store_true',
help='clear Goofy state and logs on device')
parser.add_argument('-p', dest='clear_password',
help='remove password from test_list')
parser.add_argument('-s', dest='shopfloor_host',
help='set shopfloor host')
parser.add_argument('--shopfloor_port', dest='shopfloor_port', type=int,
default=None, help='set shopfloor port')
parser.add_argument('--board', '-b', dest='board',
help='board to use (default: auto-detect)')
parser.add_argument('--project', '-j', dest='project',
help='project name to use (default: auto-detect)')
parser.add_argument('--norestart', dest='restart', action='store_false',
help="don't restart Goofy")
parser.add_argument('--hwid', action='store_true',
help='update HWID bundle')
parser.add_argument('--run', '-r', dest='run_test',
help='the test to run on device')
help=('test list to activate (defaults to the main test '
parser.add_argument('--local', action='store_true',
help=('Rather than syncing the source tree, only '
'perform test list modifications locally. '
'This must be run only on the target device.'))
args = parser.parse_args()
if args.local:
if InChroot():
sys.exit('--local must be used only on the target device')
if not SRCROOT:
sys.exit('goofy_remote must be run from within the chroot')
Spawn(['make', '--quiet'], cwd=paths.FACTORY_DIR,
check_call=True, log=True)
board = args.board or GetBoard(
# We need to rsync the public factory repo first to set up goofy symlink.
# We need --force to remove the original goofy directory if it's not a
# symlink, -l for re-creating the symlink on DUT, -K for following the symlink
# on DUT.
['-azlKC', '--force', '--exclude', '*.pyc'] +
[os.path.join(paths.FACTORY_DIR, x)
for x in ('bin', 'py', 'py_pkg', 'sh', 'third_party', 'init')]) +
['%s:/usr/local/factory' %],
check_call=True, log=True)
Spawn(['make', 'par-overlay-%s' % board], cwd=paths.FACTORY_DIR,
check_call=True, log=True)
['-az', 'overlay-%s/build/par/factory.par' % board,
'%s:/usr/local/factory/' %],
cwd=paths.FACTORY_DIR, check_call=True, log=True)
private_path = cros_board_utils.GetChromeOSFactoryBoardPath(board)
if private_path:
['-azlKC', '--exclude', 'bundle'] +
[private_path + '/', '%s:/usr/local/factory/' %],
check_call=True, log=True)
# Call goofy_remote on the remote host, allowing it to tweak test lists.
SpawnSSHToDUT([, 'goofy_remote', '--local'] +
[pipes.quote(x) for x in sys.argv[1:]],
check_call=True, log=True)
if args.hwid:
project = args.project
if not project:
# TODO(yhong): Detect the real project name instead of board name.
if board:
project = board.split('_')[-1]
if not project:
sys.exit('Cannot update hwid without project name')
chromeos_hwid_path = os.path.join(
os.path.dirname(paths.FACTORY_DIR), 'chromeos-hwid')
Spawn(['./create_bundle', '--version', '3', project.upper()],
cwd=chromeos_hwid_path, check_call=True, log=True)
SpawnSSHToDUT([, 'bash'],
check_call=True, log=True)
# Make sure all the directories and files have correct permissions. This is
# essential for Chrome to load the factory test extension.
SpawnSSHToDUT([, 'find', '/usr/local/factory', '-type', 'd',
'-exec', 'chmod 755 {} +'], check_call=True, log=True)
SpawnSSHToDUT([, 'find', '/usr/local/factory', '-type', 'f',
'-exec', 'chmod go+r {} +'], check_call=True, log=True)
if args.restart:
SpawnSSHToDUT([, '/usr/local/factory/bin/factory_restart'] +
(['-a'] if args.clear_state else []),
check_call=True, log=True)
if args.run_test:
def GoofyRpcRunTest():
return SpawnSSHToDUT(
[, 'goofy_rpc', r'RunTest\(\"%s\"\)' % args.run_test],
check_call=True, log=True)
Retry(max_retry_times=10, interval=5, callback=None, target=GoofyRpcRunTest)
if __name__ == '__main__':