#!/bin/bash -
# supernested Makefile.am
# @configure_input@
# (C) Copyright 2014 Red Hat Inc.
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
#
# Written by Richard W.M. Jones <rjones@redhat.com>

arch="@host_cpu@"
qemu="@QEMU@"
image="supernested-@VERSION@-@DISTRO@.qcow2"
format=qcow2
kernel=kernel
initrd=initrd
append="panic=1 udevtimeout=6000 udev.event-timeout=6000 no_timer_check printk.time=1 root=/dev/sda selinux=0"
smp=
tcg=

case "$arch" in
    aarch64)
	append="$append earlyprintk=pl011,0x9000000 ignore_loglevel efi-rtc=noprobe"
	;;
    *)
	append="$append console=ttyS0"
	;;
esac

TEMP=`getopt \
        -o '' \
        --longoptions 'help,nested,smp:,tcg' \
        -n run-supernested.sh -- "$@"`
if [ $? != 0 ]; then
    echo "$0: problem parsing the command line arguments"
    exit 1
fi
eval set -- "$TEMP"

usage ()
{
    echo "$0: Run the supernested appliance."
    echo "  $0 [--smp=N] [--tcg]"
    echo "Options:"
    echo "  --smp=N    Use N virtual CPUs (at all levels)."
    echo "  --tcg      Use TCG (at all levels) instead of KVM."
}

while true ; do
    case "$1" in
        --nested)
            # When the nested guest calls this script, it uses the
            # --nested flag.  DO NOT ADD THIS FLAG YOURSELF!
            image=/dev/sda
            format=raw
            kernel=/kernel
            initrd=/initrd
	    smp="$(
                grep -Eo 'supernested.smp=[[:digit:]]+' /proc/cmdline |
                grep -Eo '[[:digit:]]+'
                )"
	    tcg=
	    if grep -q supernested.tcg=1 /proc/cmdline; then
		tcg=1
	    fi
            shift
            ;;
	--smp)
	    smp="$2"
	    shift 2
	    ;;
	--tcg)
	    tcg=1
	    shift
	    ;;
        --help)
	    usage
	    exit 0
            ;;
        --)
            shift
	    break
            ;;
        *)
            echo "internal error ($1)"
            exit 1
            ;;
    esac
done

# How much memory to give to the guest?
memory="$(
    grep -Eo '^MemTotal:.*[[:digit:]]+ kB' /proc/meminfo |
    grep -Eo '[[:digit:]]+'
    )"
if [ -z "memory" ]; then
    echo "$0: could not get total memory from /proc/meminfo"
    exit 1
fi
memory=$((memory / 1024))
if [ $memory -lt 2048 ]; then
    echo "$0: not enough memory left: memory=$memory MB"
    exit 1
fi
memory=$((memory - 1500))

# ARM mach_virt (-M virt) model only supports 30GB of RAM.
if [ $memory -gt 30000 ]; then
    case "$arch" in
	arm*|aarch64)
	    echo "$0: limiting memory to 30000 MB for ARM mach_virt"
	    memory=30000
	    ;;
	*)
    esac
fi

# Get current nesting level from /proc/cmdline, increment it, add it
# to the command line.
level="$(
    grep -Eo 'supernested.level=[[:digit:]]+' /proc/cmdline |
    grep -Eo '[[:digit:]]+'
    )"
if [ -z "$level" ]; then level=0; fi
append="$append supernested.level=$((level+1))"

# Append other parameters to nested command line.
append="$append supernested.smp=$smp supernested.tcg=$tcg"

extra_args=
if [ -n "$smp" ]; then extra_args="$extra_args -smp $smp"; fi
if [ -n "$tcg" ]; then
    extra_args="$extra_args -machine accel=tcg"
else
    extra_args="$extra_args -cpu host -machine accel=kvm"
fi

echo "Running"
echo "    qemu = $qemu"
echo "    extra_args = $extra_args"
echo "    memory = $memory"
echo "    kernel = $kernel"
echo "    initrd = $initrd"
echo "    disk = $image"
echo "    append = $append"

# Since this could crash the machine, better to sync ...
sync

exec \
"$qemu" \
    $extra_args \
    -display none \
    -m $memory \
    -no-reboot \
    -rtc driftfix=slew \
    -no-hpet \
    -global kvm-pit.lost_tick_policy=discard \
    -kernel "$kernel" \
    -initrd "$initrd" \
    -device virtio-scsi-pci,id=scsi \
    -drive file="$image",snapshot=on,cache=unsafe,format="$format",id=hd0,if=none \
    -device scsi-hd,drive=hd0 \
    -device virtio-serial-pci \
    -serial stdio \
    -device sga \
    -append "$append"