mirror of https://github.com/aya-rs/aya
Merge pull request #733 from tamird/kernel-test
integration-test: Implement running on VMspull/736/head
Tamir Duberstein
1 year ago
committed by
GitHub
commit
412a0875b4
@ -0,0 +1,10 @@
|
|||||||
|
[package]
|
||||||
|
name = "init"
|
||||||
|
version = "0.1.0"
|
||||||
|
authors = ["Tamir Duberstein <tamird@gmail.com>"]
|
||||||
|
edition = "2021"
|
||||||
|
publish = false
|
||||||
|
|
||||||
|
[dependencies]
|
||||||
|
anyhow = { workspace = true, features = ["std"] }
|
||||||
|
nix = { workspace = true, features = ["fs", "mount", "reboot"] }
|
||||||
@ -0,0 +1,166 @@
|
|||||||
|
//! init is the first process started by the kernel.
|
||||||
|
//!
|
||||||
|
//! This implementation creates the minimal mounts required to run BPF programs, runs all binaries
|
||||||
|
//! in /bin, prints a final message ("init: success|failure"), and powers off the machine.
|
||||||
|
|
||||||
|
use anyhow::Context as _;
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
struct Errors(Vec<anyhow::Error>);
|
||||||
|
|
||||||
|
impl std::fmt::Display for Errors {
|
||||||
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
||||||
|
let Self(errors) = self;
|
||||||
|
for (i, error) in errors.iter().enumerate() {
|
||||||
|
if i != 0 {
|
||||||
|
writeln!(f)?;
|
||||||
|
}
|
||||||
|
write!(f, "{:?}", error)?;
|
||||||
|
}
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl std::error::Error for Errors {}
|
||||||
|
|
||||||
|
fn run() -> anyhow::Result<()> {
|
||||||
|
const RXRXRX: nix::sys::stat::Mode = nix::sys::stat::Mode::empty()
|
||||||
|
.union(nix::sys::stat::Mode::S_IRUSR)
|
||||||
|
.union(nix::sys::stat::Mode::S_IXUSR)
|
||||||
|
.union(nix::sys::stat::Mode::S_IRGRP)
|
||||||
|
.union(nix::sys::stat::Mode::S_IXGRP)
|
||||||
|
.union(nix::sys::stat::Mode::S_IROTH)
|
||||||
|
.union(nix::sys::stat::Mode::S_IXOTH);
|
||||||
|
|
||||||
|
struct Mount {
|
||||||
|
source: &'static str,
|
||||||
|
target: &'static str,
|
||||||
|
fstype: &'static str,
|
||||||
|
flags: nix::mount::MsFlags,
|
||||||
|
data: Option<&'static str>,
|
||||||
|
target_mode: Option<nix::sys::stat::Mode>,
|
||||||
|
}
|
||||||
|
|
||||||
|
for Mount {
|
||||||
|
source,
|
||||||
|
target,
|
||||||
|
fstype,
|
||||||
|
flags,
|
||||||
|
data,
|
||||||
|
target_mode,
|
||||||
|
} in [
|
||||||
|
Mount {
|
||||||
|
source: "proc",
|
||||||
|
target: "/proc",
|
||||||
|
fstype: "proc",
|
||||||
|
flags: nix::mount::MsFlags::empty(),
|
||||||
|
data: None,
|
||||||
|
target_mode: Some(RXRXRX),
|
||||||
|
},
|
||||||
|
Mount {
|
||||||
|
source: "sysfs",
|
||||||
|
target: "/sys",
|
||||||
|
fstype: "sysfs",
|
||||||
|
flags: nix::mount::MsFlags::empty(),
|
||||||
|
data: None,
|
||||||
|
target_mode: Some(RXRXRX),
|
||||||
|
},
|
||||||
|
Mount {
|
||||||
|
source: "debugfs",
|
||||||
|
target: "/sys/kernel/debug",
|
||||||
|
fstype: "debugfs",
|
||||||
|
flags: nix::mount::MsFlags::empty(),
|
||||||
|
data: None,
|
||||||
|
target_mode: None,
|
||||||
|
},
|
||||||
|
Mount {
|
||||||
|
source: "bpffs",
|
||||||
|
target: "/sys/fs/bpf",
|
||||||
|
fstype: "bpf",
|
||||||
|
flags: nix::mount::MsFlags::empty(),
|
||||||
|
data: None,
|
||||||
|
target_mode: None,
|
||||||
|
},
|
||||||
|
] {
|
||||||
|
match target_mode {
|
||||||
|
None => {
|
||||||
|
// Must exist.
|
||||||
|
let nix::sys::stat::FileStat { st_mode, .. } = nix::sys::stat::stat(target)
|
||||||
|
.with_context(|| format!("stat({target}) failed"))?;
|
||||||
|
let s_flag = nix::sys::stat::SFlag::from_bits_truncate(st_mode);
|
||||||
|
|
||||||
|
if !s_flag.contains(nix::sys::stat::SFlag::S_IFDIR) {
|
||||||
|
anyhow::bail!("{target} is not a directory");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Some(target_mode) => {
|
||||||
|
// Must not exist.
|
||||||
|
nix::unistd::mkdir(target, target_mode)
|
||||||
|
.with_context(|| format!("mkdir({target}) failed"))?;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
nix::mount::mount(Some(source), target, Some(fstype), flags, data).with_context(|| {
|
||||||
|
format!("mount({source}, {target}, {fstype}, {flags:?}, {data:?}) failed")
|
||||||
|
})?;
|
||||||
|
}
|
||||||
|
|
||||||
|
// By contract we run everything in /bin and assume they're rust test binaries.
|
||||||
|
//
|
||||||
|
// If the user requested command line arguments, they're named init.arg={}.
|
||||||
|
|
||||||
|
// Read kernel parameters from /proc/cmdline. They're space separated on a single line.
|
||||||
|
let cmdline = std::fs::read_to_string("/proc/cmdline")
|
||||||
|
.with_context(|| "read_to_string(/proc/cmdline) failed")?;
|
||||||
|
let args = cmdline
|
||||||
|
.split_whitespace()
|
||||||
|
.filter_map(|parameter| {
|
||||||
|
parameter
|
||||||
|
.strip_prefix("init.arg=")
|
||||||
|
.map(std::ffi::OsString::from)
|
||||||
|
})
|
||||||
|
.collect::<Vec<_>>();
|
||||||
|
|
||||||
|
// Iterate files in /bin.
|
||||||
|
let read_dir = std::fs::read_dir("/bin").context("read_dir(/bin) failed")?;
|
||||||
|
let errors = read_dir
|
||||||
|
.filter_map(|entry| {
|
||||||
|
match (|| {
|
||||||
|
let entry = entry.context("read_dir(/bin) failed")?;
|
||||||
|
let path = entry.path();
|
||||||
|
let status = std::process::Command::new(&path)
|
||||||
|
.args(&args)
|
||||||
|
.status()
|
||||||
|
.with_context(|| format!("failed to execute {}", path.display()))?;
|
||||||
|
|
||||||
|
if status.code() == Some(0) {
|
||||||
|
Ok(())
|
||||||
|
} else {
|
||||||
|
Err(anyhow::anyhow!("{} failed: {status:?}", path.display()))
|
||||||
|
}
|
||||||
|
})() {
|
||||||
|
Ok(()) => None,
|
||||||
|
Err(err) => Some(err),
|
||||||
|
}
|
||||||
|
})
|
||||||
|
.collect::<Vec<_>>();
|
||||||
|
if errors.is_empty() {
|
||||||
|
Ok(())
|
||||||
|
} else {
|
||||||
|
Err(Errors(errors).into())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn main() {
|
||||||
|
match run() {
|
||||||
|
Ok(()) => {
|
||||||
|
println!("init: success");
|
||||||
|
}
|
||||||
|
Err(err) => {
|
||||||
|
println!("{err:?}");
|
||||||
|
println!("init: failure");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
let how = nix::sys::reboot::RebootMode::RB_POWER_OFF;
|
||||||
|
let _: std::convert::Infallible = nix::sys::reboot::reboot(how)
|
||||||
|
.unwrap_or_else(|err| panic!("reboot({how:?}) failed: {err:?}"));
|
||||||
|
}
|
||||||
@ -1,264 +0,0 @@
|
|||||||
#!/bin/bash
|
|
||||||
|
|
||||||
VERBOSITY=0
|
|
||||||
TEMP_D=""
|
|
||||||
DEF_DISK_FORMAT="raw"
|
|
||||||
DEF_FILESYSTEM="iso9660"
|
|
||||||
CR="
|
|
||||||
"
|
|
||||||
|
|
||||||
error() { echo "$@" 1>&2; }
|
|
||||||
fail() { [ $# -eq 0 ] || error "$@"; exit 1; }
|
|
||||||
|
|
||||||
Usage() {
|
|
||||||
cat <<EOF
|
|
||||||
Usage: ${0##*/} [ options ] output user-data [meta-data]
|
|
||||||
|
|
||||||
Create a disk for cloud-init to utilize nocloud
|
|
||||||
|
|
||||||
options:
|
|
||||||
-h | --help show usage
|
|
||||||
-d | --disk-format D disk format to output. default: raw
|
|
||||||
can be anything supported by qemu-img or
|
|
||||||
tar, tar-seed-local, tar-seed-net
|
|
||||||
-H | --hostname H set hostname in metadata to H
|
|
||||||
-f | --filesystem F filesystem format (vfat or iso), default: iso9660
|
|
||||||
|
|
||||||
-i | --interfaces F write network interfaces file into metadata
|
|
||||||
-N | --network-config F write network config file to local datasource
|
|
||||||
-m | --dsmode M add 'dsmode' ('local' or 'net') to the metadata
|
|
||||||
default in cloud-init is 'net', meaning network is
|
|
||||||
required.
|
|
||||||
-V | --vendor-data F vendor-data file
|
|
||||||
-v | --verbose increase verbosity
|
|
||||||
|
|
||||||
Note, --dsmode, --hostname, and --interfaces are incompatible
|
|
||||||
with metadata.
|
|
||||||
|
|
||||||
Example:
|
|
||||||
* cat my-user-data
|
|
||||||
#cloud-config
|
|
||||||
password: passw0rd
|
|
||||||
chpasswd: { expire: False }
|
|
||||||
ssh_pwauth: True
|
|
||||||
* echo "instance-id: \$(uuidgen || echo i-abcdefg)" > my-meta-data
|
|
||||||
* ${0##*/} my-seed.img my-user-data my-meta-data
|
|
||||||
* kvm -net nic -net user,hostfwd=tcp::2222-:22 \\
|
|
||||||
-drive file=disk1.img,if=virtio -drive file=my-seed.img,if=virtio
|
|
||||||
* ssh -p 2222 ubuntu@localhost
|
|
||||||
EOF
|
|
||||||
}
|
|
||||||
|
|
||||||
bad_Usage() { Usage 1>&2; [ $# -eq 0 ] || error "$@"; exit 1; }
|
|
||||||
cleanup() {
|
|
||||||
[ -z "${TEMP_D}" -o ! -d "${TEMP_D}" ] || rm -Rf "${TEMP_D}"
|
|
||||||
}
|
|
||||||
|
|
||||||
debug() {
|
|
||||||
local level=${1}; shift;
|
|
||||||
[ "${level}" -gt "${VERBOSITY}" ] && return
|
|
||||||
error "${@}"
|
|
||||||
}
|
|
||||||
|
|
||||||
has_cmd() {
|
|
||||||
command -v "$1" >/dev/null 2>&1
|
|
||||||
}
|
|
||||||
|
|
||||||
short_opts="hH:i:d:f:m:N:o:V:v"
|
|
||||||
long_opts="disk-format:,dsmode:,filesystem:,help,hostname:,interfaces:,"
|
|
||||||
long_opts="${long_opts}network-config:,output:,vendor-data:,verbose"
|
|
||||||
getopt_out=$(getopt -n "${0##*/}" \
|
|
||||||
-o "${short_opts}" -l "${long_opts}" -- "$@") &&
|
|
||||||
eval set -- "${getopt_out}" ||
|
|
||||||
bad_Usage
|
|
||||||
|
|
||||||
## <<insert default variables here>>
|
|
||||||
output=""
|
|
||||||
userdata=""
|
|
||||||
metadata=""
|
|
||||||
vendordata=""
|
|
||||||
filesystem=""
|
|
||||||
diskformat=$DEF_DISK_FORMAT
|
|
||||||
interfaces=_unset
|
|
||||||
dsmode=""
|
|
||||||
hostname=""
|
|
||||||
ncname="network-config"
|
|
||||||
|
|
||||||
|
|
||||||
while [ $# -ne 0 ]; do
|
|
||||||
cur=${1}; next=${2};
|
|
||||||
case "$cur" in
|
|
||||||
-h|--help) Usage ; exit 0;;
|
|
||||||
-d|--disk-format) diskformat=$next; shift;;
|
|
||||||
-f|--filesystem) filesystem=$next; shift;;
|
|
||||||
-H|--hostname) hostname=$next; shift;;
|
|
||||||
-i|--interfaces) interfaces=$next; shift;;
|
|
||||||
-N|--network-config) netcfg=$next; shift;;
|
|
||||||
-m|--dsmode) dsmode=$next; shift;;
|
|
||||||
-v|--verbose) VERBOSITY=$((${VERBOSITY}+1));;
|
|
||||||
-V|--vendor-data) vendordata="$next";;
|
|
||||||
--) shift; break;;
|
|
||||||
esac
|
|
||||||
shift;
|
|
||||||
done
|
|
||||||
|
|
||||||
## check arguments here
|
|
||||||
## how many args do you expect?
|
|
||||||
echo $1
|
|
||||||
echo $2
|
|
||||||
echo $3
|
|
||||||
[ $# -ge 2 ] || bad_Usage "must provide output, userdata"
|
|
||||||
[ $# -le 3 ] || bad_Usage "confused by additional args"
|
|
||||||
|
|
||||||
output=$1
|
|
||||||
userdata=$2
|
|
||||||
metadata=$3
|
|
||||||
|
|
||||||
if [ -n "$metadata" ]; then
|
|
||||||
[ "$interfaces" = "_unset" -a -z "$dsmode" -a -z "$hostname" ] ||
|
|
||||||
fail "metadata is incompatible with:" \
|
|
||||||
"--interfaces, --hostname, --dsmode"
|
|
||||||
fi
|
|
||||||
|
|
||||||
case "$diskformat" in
|
|
||||||
tar|tar-seed-local|tar-seed-net)
|
|
||||||
if [ "${filesystem:-tar}" != "tar" ]; then
|
|
||||||
fail "diskformat=tar is incompatible with filesystem"
|
|
||||||
fi
|
|
||||||
filesystem="$diskformat"
|
|
||||||
;;
|
|
||||||
tar*)
|
|
||||||
fail "supported 'tar' formats are tar, tar-seed-local, tar-seed-net"
|
|
||||||
esac
|
|
||||||
|
|
||||||
if [ -z "$filesystem" ]; then
|
|
||||||
filesystem="$DEF_FILESYSTEM"
|
|
||||||
fi
|
|
||||||
if [ "$filesystem" = "iso" ]; then
|
|
||||||
filesystem="iso9660"
|
|
||||||
fi
|
|
||||||
|
|
||||||
case "$filesystem" in
|
|
||||||
tar*)
|
|
||||||
has_cmd tar ||
|
|
||||||
fail "missing 'tar'. Required for --filesystem=$filesystem";;
|
|
||||||
vfat)
|
|
||||||
has_cmd mkfs.vfat ||
|
|
||||||
fail "missing 'mkfs.vfat'. Required for --filesystem=vfat."
|
|
||||||
has_cmd mcopy ||
|
|
||||||
fail "missing 'mcopy'. Required for --filesystem=vfat."
|
|
||||||
;;
|
|
||||||
iso9660)
|
|
||||||
has_cmd mkisofs ||
|
|
||||||
fail "missing 'mkisofs'. Required for --filesystem=iso9660."
|
|
||||||
;;
|
|
||||||
*) fail "unknown filesystem $filesystem";;
|
|
||||||
esac
|
|
||||||
|
|
||||||
case "$diskformat" in
|
|
||||||
tar*|raw) :;;
|
|
||||||
*) has_cmd "qemu-img" ||
|
|
||||||
fail "missing 'qemu-img'. Required for --disk-format=$diskformat."
|
|
||||||
esac
|
|
||||||
|
|
||||||
[ "$interfaces" = "_unset" -o -r "$interfaces" ] ||
|
|
||||||
fail "$interfaces: not a readable file"
|
|
||||||
|
|
||||||
TEMP_D=$(mktemp -d "${TMPDIR:-/tmp}/${0##*/}.XXXXXX") ||
|
|
||||||
fail "failed to make tempdir"
|
|
||||||
trap cleanup EXIT
|
|
||||||
|
|
||||||
files=( "${TEMP_D}/user-data" "${TEMP_D}/meta-data" )
|
|
||||||
if [ -n "$metadata" ]; then
|
|
||||||
cp "$metadata" "$TEMP_D/meta-data" || fail "$metadata: failed to copy"
|
|
||||||
else
|
|
||||||
instance_id="iid-local01"
|
|
||||||
iface_data=""
|
|
||||||
[ "$interfaces" != "_unset" ] &&
|
|
||||||
iface_data=$(sed ':a;N;$!ba;s/\n/\\n/g' "$interfaces")
|
|
||||||
|
|
||||||
# write json formatted user-data (json is a subset of yaml)
|
|
||||||
mdata=""
|
|
||||||
for kv in "instance-id:$instance_id" "local-hostname:$hostname" \
|
|
||||||
"interfaces:${iface_data}" "dsmode:$dsmode"; do
|
|
||||||
key=${kv%%:*}
|
|
||||||
val=${kv#*:}
|
|
||||||
[ -n "$val" ] || continue
|
|
||||||
mdata="${mdata:+${mdata},${CR}}\"$key\": \"$val\""
|
|
||||||
done
|
|
||||||
printf "{\n%s\n}\n" "$mdata" > "${TEMP_D}/meta-data"
|
|
||||||
fi
|
|
||||||
|
|
||||||
if [ -n "$netcfg" ]; then
|
|
||||||
cp "$netcfg" "${TEMP_D}/$ncname" ||
|
|
||||||
fail "failed to copy network config"
|
|
||||||
files[${#files[@]}]="$TEMP_D/$ncname"
|
|
||||||
fi
|
|
||||||
|
|
||||||
if [ -n "$vendordata" ]; then
|
|
||||||
cp "$vendordata" "${TEMP_D}/vendor-data" ||
|
|
||||||
fail "failed to copy vendor data"
|
|
||||||
files[${#files[@]}]="$TEMP_D/vendor-data"
|
|
||||||
fi
|
|
||||||
|
|
||||||
files_rel=( )
|
|
||||||
for f in "${files[@]}"; do
|
|
||||||
files_rel[${#files_rel[@]}]="${f#${TEMP_D}/}"
|
|
||||||
done
|
|
||||||
|
|
||||||
if [ "$userdata" = "-" ]; then
|
|
||||||
cat > "$TEMP_D/user-data" || fail "failed to read from stdin"
|
|
||||||
else
|
|
||||||
cp "$userdata" "$TEMP_D/user-data" || fail "$userdata: failed to copy"
|
|
||||||
fi
|
|
||||||
|
|
||||||
## alternatively, create a vfat filesystem with same files
|
|
||||||
img="$TEMP_D/seed-data"
|
|
||||||
tar_opts=( --owner=root --group=root )
|
|
||||||
|
|
||||||
case "$filesystem" in
|
|
||||||
tar)
|
|
||||||
tar "${tar_opts[@]}" -C "${TEMP_D}" -cf "$img" "${files_rel[@]}" ||
|
|
||||||
fail "failed to create tarball of ${files_rel[*]}"
|
|
||||||
;;
|
|
||||||
tar-seed-local|tar-seed-net)
|
|
||||||
if [ "$filesystem" = "tar-seed-local" ]; then
|
|
||||||
path="var/lib/cloud/seed/nocloud"
|
|
||||||
else
|
|
||||||
path="var/lib/cloud/seed/nocloud-net"
|
|
||||||
fi
|
|
||||||
mkdir -p "${TEMP_D}/${path}" ||
|
|
||||||
fail "failed making path for seed files"
|
|
||||||
mv "${files[@]}" "${TEMP_D}/$path" ||
|
|
||||||
fail "failed moving files"
|
|
||||||
tar "${tar_opts[@]}" -C "${TEMP_D}" -cf "$img" "${path}" ||
|
|
||||||
fail "failed to create tarball with $path"
|
|
||||||
;;
|
|
||||||
iso9660)
|
|
||||||
mkisofs -output "$img" -volid cidata \
|
|
||||||
-joliet -rock "${files[@]}" > "$TEMP_D/err" 2>&1 ||
|
|
||||||
{ cat "$TEMP_D/err" 1>&2; fail "failed to mkisofs"; }
|
|
||||||
;;
|
|
||||||
vfat)
|
|
||||||
truncate -s 128K "$img" || fail "failed truncate image"
|
|
||||||
out=$(mkfs.vfat -n cidata "$img" 2>&1) ||
|
|
||||||
{ error "failed: mkfs.vfat -n cidata $img"; error "$out"; }
|
|
||||||
mcopy -oi "$img" "${files[@]}" :: ||
|
|
||||||
fail "failed to copy user-data, meta-data to img"
|
|
||||||
;;
|
|
||||||
esac
|
|
||||||
|
|
||||||
[ "$output" = "-" ] && output="$TEMP_D/final"
|
|
||||||
if [ "${diskformat#tar}" != "$diskformat" -o "$diskformat" = "raw" ]; then
|
|
||||||
cp "$img" "$output" ||
|
|
||||||
fail "failed to copy image to $output"
|
|
||||||
else
|
|
||||||
qemu-img convert -f raw -O "$diskformat" "$img" "$output" ||
|
|
||||||
fail "failed to convert to disk format $diskformat"
|
|
||||||
fi
|
|
||||||
|
|
||||||
[ "$output" != "$TEMP_D/final" ] || { cat "$output" && output="-"; } ||
|
|
||||||
fail "failed to write to -"
|
|
||||||
|
|
||||||
debug 1 "wrote ${output} with filesystem=$filesystem and diskformat=$diskformat"
|
|
||||||
# vi: ts=4 noexpandtab
|
|
||||||
@ -1,241 +0,0 @@
|
|||||||
#!/usr/bin/env bash
|
|
||||||
|
|
||||||
set -ex
|
|
||||||
|
|
||||||
if [ "$(uname -s)" = "Darwin" ]; then
|
|
||||||
PATH="$(dirname "$(brew list gnu-getopt | grep "bin/getopt$")"):$PATH"
|
|
||||||
export PATH
|
|
||||||
fi
|
|
||||||
|
|
||||||
AYA_SOURCE_DIR="$(realpath "$(dirname "$0")"/..)"
|
|
||||||
|
|
||||||
# Temporary directory for tests to use.
|
|
||||||
AYA_TMPDIR="${AYA_SOURCE_DIR}/.tmp"
|
|
||||||
|
|
||||||
# Directory for VM images
|
|
||||||
AYA_IMGDIR=${AYA_TMPDIR}
|
|
||||||
|
|
||||||
if [ -z "${AYA_BUILD_TARGET}" ]; then
|
|
||||||
AYA_BUILD_TARGET=$(rustc -vV | sed -n 's|host: ||p')
|
|
||||||
fi
|
|
||||||
|
|
||||||
AYA_HOST_ARCH=$(uname -m)
|
|
||||||
if [ "${AYA_HOST_ARCH}" = "arm64" ]; then
|
|
||||||
AYA_HOST_ARCH="aarch64"
|
|
||||||
fi
|
|
||||||
|
|
||||||
if [ -z "${AYA_GUEST_ARCH}" ]; then
|
|
||||||
AYA_GUEST_ARCH="${AYA_HOST_ARCH}"
|
|
||||||
fi
|
|
||||||
|
|
||||||
if [ "${AYA_GUEST_ARCH}" = "aarch64" ]; then
|
|
||||||
if [ -z "${AARCH64_UEFI}" ]; then
|
|
||||||
AARCH64_UEFI="$(brew list qemu -1 -v | grep edk2-aarch64-code.fd)"
|
|
||||||
fi
|
|
||||||
fi
|
|
||||||
|
|
||||||
if [ -z "$AYA_MUSL_TARGET" ]; then
|
|
||||||
AYA_MUSL_TARGET=${AYA_GUEST_ARCH}-unknown-linux-musl
|
|
||||||
fi
|
|
||||||
|
|
||||||
# Test Image
|
|
||||||
if [ -z "${AYA_TEST_IMAGE}" ]; then
|
|
||||||
AYA_TEST_IMAGE="fedora38"
|
|
||||||
fi
|
|
||||||
|
|
||||||
case "${AYA_TEST_IMAGE}" in
|
|
||||||
fedora*) AYA_SSH_USER="fedora";;
|
|
||||||
centos*) AYA_SSH_USER="centos";;
|
|
||||||
esac
|
|
||||||
|
|
||||||
download_images() {
|
|
||||||
mkdir -p "${AYA_IMGDIR}"
|
|
||||||
case $1 in
|
|
||||||
fedora37)
|
|
||||||
if [ ! -f "${AYA_IMGDIR}/fedora37.${AYA_GUEST_ARCH}.qcow2" ]; then
|
|
||||||
IMAGE="Fedora-Cloud-Base-37-1.7.${AYA_GUEST_ARCH}.qcow2"
|
|
||||||
IMAGE_URL="https://download.fedoraproject.org/pub/fedora/linux/releases/37/Cloud/${AYA_GUEST_ARCH}/images"
|
|
||||||
echo "Downloading: ${IMAGE}, this may take a while..."
|
|
||||||
curl -o "${AYA_IMGDIR}/fedora37.${AYA_GUEST_ARCH}.qcow2" -sSL "${IMAGE_URL}/${IMAGE}"
|
|
||||||
fi
|
|
||||||
;;
|
|
||||||
fedora38)
|
|
||||||
if [ ! -f "${AYA_IMGDIR}/fedora38.${AYA_GUEST_ARCH}.qcow2" ]; then
|
|
||||||
IMAGE="Fedora-Cloud-Base-38_Beta-1.3.${AYA_GUEST_ARCH}.qcow2"
|
|
||||||
IMAGE_URL="https://fr2.rpmfind.net/linux/fedora/linux/releases/test/38_Beta/Cloud/${AYA_GUEST_ARCH}/images"
|
|
||||||
echo "Downloading: ${IMAGE}, this may take a while..."
|
|
||||||
curl -o "${AYA_IMGDIR}/fedora38.${AYA_GUEST_ARCH}.qcow2" -sSL "${IMAGE_URL}/${IMAGE}"
|
|
||||||
fi
|
|
||||||
;;
|
|
||||||
centos8)
|
|
||||||
if [ ! -f "${AYA_IMGDIR}/centos8.${AYA_GUEST_ARCH}.qcow2" ]; then
|
|
||||||
IMAGE="CentOS-8-GenericCloud-8.4.2105-20210603.0.${AYA_GUEST_ARCH}.qcow2"
|
|
||||||
IMAGE_URL="https://cloud.centos.org/centos/8/${AYA_GUEST_ARCH}/images"
|
|
||||||
echo "Downloading: ${IMAGE}, this may take a while..."
|
|
||||||
curl -o "${AYA_IMGDIR}/centos8.${AYA_GUEST_ARCH}.qcow2" -sSL "${IMAGE_URL}/${IMAGE}"
|
|
||||||
fi
|
|
||||||
;;
|
|
||||||
*)
|
|
||||||
echo "$1 is not a recognized image name"
|
|
||||||
return 1
|
|
||||||
;;
|
|
||||||
esac
|
|
||||||
}
|
|
||||||
|
|
||||||
start_vm() {
|
|
||||||
download_images "${AYA_TEST_IMAGE}"
|
|
||||||
# prepare config
|
|
||||||
cat > "${AYA_TMPDIR}/metadata.yaml" <<EOF
|
|
||||||
instance-id: iid-local01
|
|
||||||
local-hostname: test
|
|
||||||
EOF
|
|
||||||
|
|
||||||
if [ ! -f "${AYA_TMPDIR}/test_rsa" ]; then
|
|
||||||
ssh-keygen -t rsa -b 4096 -f "${AYA_TMPDIR}/test_rsa" -N "" -C "" -q
|
|
||||||
pub_key=$(cat "${AYA_TMPDIR}/test_rsa.pub")
|
|
||||||
fi
|
|
||||||
|
|
||||||
if [ ! -f "${AYA_TMPDIR}/ssh_config" ]; then
|
|
||||||
cat > "${AYA_TMPDIR}/ssh_config" <<EOF
|
|
||||||
StrictHostKeyChecking=no
|
|
||||||
UserKnownHostsFile=/dev/null
|
|
||||||
GlobalKnownHostsFile=/dev/null
|
|
||||||
EOF
|
|
||||||
fi
|
|
||||||
|
|
||||||
cat > "${AYA_TMPDIR}/user-data.yaml" <<EOF
|
|
||||||
#cloud-config
|
|
||||||
ssh_authorized_keys:
|
|
||||||
- ${pub_key}
|
|
||||||
EOF
|
|
||||||
|
|
||||||
"$AYA_SOURCE_DIR"/test/cloud-localds "${AYA_TMPDIR}/seed.img" "${AYA_TMPDIR}/user-data.yaml" "${AYA_TMPDIR}/metadata.yaml"
|
|
||||||
case "${AYA_GUEST_ARCH}" in
|
|
||||||
x86_64)
|
|
||||||
QEMU=qemu-system-x86_64
|
|
||||||
machine="q35"
|
|
||||||
cpu="qemu64"
|
|
||||||
nr_cpus="$(nproc --all)"
|
|
||||||
if [ "${AYA_HOST_ARCH}" = "${AYA_GUEST_ARCH}" ]; then
|
|
||||||
if [ -c /dev/kvm ]; then
|
|
||||||
machine="${machine},accel=kvm"
|
|
||||||
cpu="host"
|
|
||||||
elif [ "$(uname -s)" = "Darwin" ]; then
|
|
||||||
machine="${machine},accel=hvf"
|
|
||||||
cpu="host"
|
|
||||||
fi
|
|
||||||
fi
|
|
||||||
;;
|
|
||||||
aarch64)
|
|
||||||
QEMU=qemu-system-aarch64
|
|
||||||
machine="virt"
|
|
||||||
cpu="cortex-a57"
|
|
||||||
uefi=("-drive" "file=${AARCH64_UEFI},if=pflash,format=raw,readonly=on")
|
|
||||||
if [ "${AYA_HOST_ARCH}" = "${AYA_GUEST_ARCH}" ]; then
|
|
||||||
if [ -c /dev/kvm ]; then
|
|
||||||
machine="${machine},accel=kvm"
|
|
||||||
cpu="host"
|
|
||||||
nr_cpus="$(nproc --all)"
|
|
||||||
elif [ "$(uname -s)" = "Darwin" ]; then
|
|
||||||
machine="${machine},accel=hvf,highmem=off"
|
|
||||||
cpu="cortex-a72"
|
|
||||||
# nrpoc --all on apple silicon returns the two extra fancy
|
|
||||||
# cores and then qemu complains that nr_cpus > actual_cores
|
|
||||||
nr_cpus=8
|
|
||||||
fi
|
|
||||||
fi
|
|
||||||
;;
|
|
||||||
*)
|
|
||||||
echo "${AYA_GUEST_ARCH} is not supported"
|
|
||||||
return 1
|
|
||||||
;;
|
|
||||||
esac
|
|
||||||
|
|
||||||
if [ ! -f "${AYA_IMGDIR}/vm.qcow2" ]; then
|
|
||||||
echo "Creating VM image"
|
|
||||||
qemu-img create -F qcow2 -f qcow2 -o backing_file="${AYA_IMGDIR}/${AYA_TEST_IMAGE}.${AYA_GUEST_ARCH}.qcow2" "${AYA_IMGDIR}/vm.qcow2" || return 1
|
|
||||||
else
|
|
||||||
echo "Reusing existing VM image"
|
|
||||||
fi
|
|
||||||
$QEMU \
|
|
||||||
-machine "${machine}" \
|
|
||||||
-cpu "${cpu}" \
|
|
||||||
-m 3G \
|
|
||||||
-smp "${nr_cpus}" \
|
|
||||||
-display none \
|
|
||||||
-monitor none \
|
|
||||||
-daemonize \
|
|
||||||
-pidfile "${AYA_TMPDIR}/vm.pid" \
|
|
||||||
-device virtio-net-pci,netdev=net0 \
|
|
||||||
-netdev user,id=net0,hostfwd=tcp::2222-:22 \
|
|
||||||
"${uefi[@]}" \
|
|
||||||
-drive if=virtio,format=qcow2,file="${AYA_IMGDIR}/vm.qcow2" \
|
|
||||||
-drive if=virtio,format=raw,file="${AYA_TMPDIR}/seed.img" || return 1
|
|
||||||
|
|
||||||
trap cleanup_vm EXIT
|
|
||||||
echo "Waiting for SSH on port 2222..."
|
|
||||||
retry=0
|
|
||||||
max_retries=300
|
|
||||||
while ! ssh -q -F "${AYA_TMPDIR}/ssh_config" -o ConnectTimeout=1 -i "${AYA_TMPDIR}/test_rsa" "${AYA_SSH_USER}"@localhost -p 2222 echo "Hello VM"; do
|
|
||||||
retry=$((retry+1))
|
|
||||||
if [ ${retry} -gt ${max_retries} ]; then
|
|
||||||
echo "Unable to connect to VM"
|
|
||||||
return 1
|
|
||||||
fi
|
|
||||||
sleep 1
|
|
||||||
done
|
|
||||||
|
|
||||||
echo "VM launched"
|
|
||||||
exec_vm uname -a
|
|
||||||
echo "Enabling testing repositories"
|
|
||||||
exec_vm sudo dnf config-manager --set-enabled updates-testing
|
|
||||||
exec_vm sudo dnf config-manager --set-enabled updates-testing-modular
|
|
||||||
}
|
|
||||||
|
|
||||||
scp_vm() {
|
|
||||||
local=$1
|
|
||||||
remote=$(basename "$1")
|
|
||||||
scp -q -F "${AYA_TMPDIR}/ssh_config" \
|
|
||||||
-i "${AYA_TMPDIR}/test_rsa" \
|
|
||||||
-P 2222 "${local}" \
|
|
||||||
"${AYA_SSH_USER}@localhost:${remote}"
|
|
||||||
}
|
|
||||||
|
|
||||||
rsync_vm() {
|
|
||||||
rsync -a -e "ssh -p 2222 -F ${AYA_TMPDIR}/ssh_config -i ${AYA_TMPDIR}/test_rsa" "$1" "$AYA_SSH_USER"@localhost:
|
|
||||||
}
|
|
||||||
|
|
||||||
exec_vm() {
|
|
||||||
ssh -q -F "${AYA_TMPDIR}/ssh_config" \
|
|
||||||
-i "${AYA_TMPDIR}/test_rsa" \
|
|
||||||
-p 2222 \
|
|
||||||
"${AYA_SSH_USER}"@localhost \
|
|
||||||
"$@"
|
|
||||||
}
|
|
||||||
|
|
||||||
stop_vm() {
|
|
||||||
if [ -f "${AYA_TMPDIR}/vm.pid" ]; then
|
|
||||||
echo "Stopping VM forcefully"
|
|
||||||
kill -9 "$(cat "${AYA_TMPDIR}/vm.pid")"
|
|
||||||
rm "${AYA_TMPDIR}/vm.pid"
|
|
||||||
fi
|
|
||||||
}
|
|
||||||
|
|
||||||
cleanup_vm() {
|
|
||||||
if ! stop_vm; then
|
|
||||||
rm -f "${AYA_IMGDIR}/vm.qcow2"
|
|
||||||
fi
|
|
||||||
}
|
|
||||||
|
|
||||||
start_vm
|
|
||||||
trap cleanup_vm EXIT
|
|
||||||
|
|
||||||
# make sure we always use fresh sources (also see comment at the end)
|
|
||||||
rsync_vm "$*"
|
|
||||||
|
|
||||||
exec_vm "find $* -type f -executable -print0 | xargs -0 -I {} sudo {} --test-threads=1"
|
|
||||||
|
|
||||||
# we rm and sync but it doesn't seem to work reliably - I guess we could sleep a
|
|
||||||
# few seconds after but ain't nobody got time for that. Instead we also rm
|
|
||||||
# before rsyncing.
|
|
||||||
exec_vm "rm -rf $*; sync"
|
|
||||||
Loading…
Reference in New Issue