mirror of
https://github.com/flatcar/scripts.git
synced 2025-08-08 13:36:58 +02:00
637b55326b
As a warning, this is a pretty big change. At a high-level, this changes the harness to move the managing of the devserver from image_to_live into the actual test harness. Paths of the cache locations (for archive_url) are taken when pre-generating the updates and stored in a dictionary (maps "path_to_base->path_to_target" (or path for full updates)-> cache paths). This change also has the tests run in parallel. Because we now start X number of VM's at once, each VM needs it's own pid file and ssh_port. This logic was added as well as running the actual tests in parallel. Change-Id: I1275d79740c50c2a8028489b43dcbbcf5bbd56c4 BUG=chromium-os:10723 TEST=Ran it ... a lot with -q but without a test_prefix (so full test suite). Review URL: http://codereview.chromium.org/6277015
109 lines
3.2 KiB
Bash
109 lines
3.2 KiB
Bash
# Copyright (c) 2009 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.
|
|
|
|
# Library for setting up remote access and running remote commands.
|
|
|
|
DEFAULT_PRIVATE_KEY="${GCLIENT_ROOT}/src/scripts/mod_for_test_scripts/\
|
|
ssh_keys/testing_rsa"
|
|
|
|
DEFINE_string remote "" "remote hostname/IP of running Chromium OS instance"
|
|
DEFINE_string private_key "$DEFAULT_PRIVATE_KEY" \
|
|
"Private key of root account on remote host"
|
|
DEFINE_integer ssh_port 22 \
|
|
"SSH port of the remote machine running Chromium OS instance"
|
|
|
|
# Copies $1 to $2 on remote host
|
|
function remote_cp_to() {
|
|
REMOTE_OUT=$(scp -P ${FLAGS_ssh_port} -o StrictHostKeyChecking=no \
|
|
-o ServerAliveInterval=60 -o UserKnownHostsFile=$TMP_KNOWN_HOSTS \
|
|
-i $TMP_PRIVATE_KEY $1 \
|
|
root@$FLAGS_remote:$2)
|
|
return ${PIPESTATUS[0]}
|
|
}
|
|
|
|
# Copies a list of remote files specified in file $1 to local location
|
|
# $2. Directory paths in $1 are collapsed into $2.
|
|
function remote_rsync_from() {
|
|
rsync -e "ssh -p ${FLAGS_ssh_port} -o StrictHostKeyChecking=no \
|
|
-o ServerAliveInterval=60 -o UserKnownHostsFile=$TMP_KNOWN_HOSTS \
|
|
-i $TMP_PRIVATE_KEY" \
|
|
--no-R --files-from=$1 root@${FLAGS_remote}:/ $2
|
|
}
|
|
|
|
function remote_sh() {
|
|
REMOTE_OUT=$(ssh -p ${FLAGS_ssh_port} -o StrictHostKeyChecking=no \
|
|
-o UserKnownHostsFile=$TMP_KNOWN_HOSTS -i $TMP_PRIVATE_KEY \
|
|
-o ServerAliveInterval=60 \
|
|
root@$FLAGS_remote "$@")
|
|
return ${PIPESTATUS[0]}
|
|
}
|
|
|
|
function remote_sh_allow_changed_host_key() {
|
|
rm -f $TMP_KNOWN_HOSTS
|
|
remote_sh "$@"
|
|
}
|
|
|
|
function set_up_remote_access() {
|
|
cp $FLAGS_private_key $TMP_PRIVATE_KEY
|
|
chmod 0400 $TMP_PRIVATE_KEY
|
|
|
|
# Verify the client is reachable before continuing
|
|
echo "Initiating first contact with remote host"
|
|
remote_sh "true"
|
|
echo "Connection OK"
|
|
}
|
|
|
|
# Ask the target what board it is
|
|
function learn_board() {
|
|
[ -n "${FLAGS_board}" ] && return
|
|
remote_sh grep CHROMEOS_RELEASE_BOARD /etc/lsb-release
|
|
FLAGS_board=$(echo "${REMOTE_OUT}" | cut -d '=' -f 2)
|
|
if [ -z "${FLAGS_board}" ]; then
|
|
error "Board required"
|
|
exit 1
|
|
fi
|
|
info "Target reports board is ${FLAGS_board}"
|
|
}
|
|
|
|
function remote_reboot {
|
|
info "Rebooting."
|
|
remote_sh "touch /tmp/awaiting_reboot; reboot"
|
|
local output_file
|
|
output_file="${TMP}/output"
|
|
|
|
while true; do
|
|
REMOTE_OUT=""
|
|
# This may fail while the machine is down so generate output and a
|
|
# boolean result to distinguish between down/timeout and real failure
|
|
! remote_sh_allow_changed_host_key \
|
|
"echo 0; [ -e /tmp/awaiting_reboot ] && echo '1'; true"
|
|
echo "${REMOTE_OUT}" > "${output_file}"
|
|
if grep -q "0" "${output_file}"; then
|
|
if grep -q "1" "${output_file}"; then
|
|
info "Not yet rebooted"
|
|
else
|
|
info "Rebooted and responding"
|
|
break
|
|
fi
|
|
fi
|
|
sleep .5
|
|
done
|
|
}
|
|
|
|
# Called by clients before exiting.
|
|
# Part of the remote_access.sh interface but now empty.
|
|
function cleanup_remote_access() {
|
|
true
|
|
}
|
|
|
|
function remote_access_init() {
|
|
TMP_PRIVATE_KEY=$TMP/private_key
|
|
TMP_KNOWN_HOSTS=$TMP/known_hosts
|
|
if [ -z "$FLAGS_remote" ]; then
|
|
echo "Please specify --remote=<IP-or-hostname> of the Chromium OS instance"
|
|
exit 1
|
|
fi
|
|
set_up_remote_access
|
|
}
|