Skip to content
Snippets Groups Projects
cron-callback.sh 5.9 KiB
Newer Older
Recolic's avatar
.  
Recolic committed
#!/bin/bash
Recolic's avatar
Recolic committed

Recolic's avatar
.  
Recolic committed
workdir=./data
mkdir -p "$workdir"
Recolic's avatar
Recolic committed
cd "$workdir" || exit $?
Recolic's avatar
.  
Recolic committed
mkdir -p base vm tmp
Recolic's avatar
Recolic committed

Recolic's avatar
Recolic committed
function echo2 () {
    echo "$@" 1>&2
}
Recolic's avatar
Recolic committed
function generate_metadata () {
    local name=$1
    echo "local-hostname: $name"
}
function generate_userdata () {
    local username=$1
    local password=$2
    local name=$3
    # TODO: allow public key?
    echo "#cloud-config
system_info:
  default_user:
    name: $username
    home: /home/$username

password: $password
chpasswd: { expire: False }
hostname: $name

Recolic's avatar
Recolic committed
# allow password login
Recolic's avatar
Recolic committed
ssh_pwauth: True
"
}

Recolic's avatar
Recolic committed
function download_cloud_img_if_not_exist () {
    local cloudimg="$1"
Recolic's avatar
Recolic committed
    [[ -f "base/$cloudimg" ]] && return
Recolic's avatar
Recolic committed

    declare -A knowledge
    knowledge["focal-server-cloudimg-amd64.img"]=https://cloud-images.ubuntu.com/focal/current/focal-server-cloudimg-amd64.img
    knowledge["Arch-Linux-x86_64-cloudimg.qcow2"]=https://geo.mirror.pkgbuild.com/images/latest/Arch-Linux-x86_64-cloudimg.qcow2
    knowledge["win10pro-22h2-virtio-uefi.qcow2"]=https://recolic.net/hms.php?/systems/win10pro-22h2-virtio-uefi.qcow2
    knowledge["win10-tiny10-virtio-uefi.qcow2"]=https://recolic.net/hms.php?/systems/win10-tiny10-virtio-uefi.qcow2
Recolic's avatar
Recolic committed
    [ ! "${knowledge[$cloudimg]+abc}" ] && echo2 "Unknown cloudimg $cloudimg. cannot download it." && return 1

Recolic's avatar
Recolic committed
    echo2 "+ Downloading cloudimg $cloudimg..."
Recolic's avatar
Recolic committed
    if which aria2c; then
        aria2c -o "base/$cloudimg" "${knowledge[$cloudimg]}" || ! echo2 "Failed to download ubuntu cloudimg" || return $?
    elif which wget; then
        wget   -O "base/$cloudimg" "${knowledge[$cloudimg]}" || ! echo2 "Failed to download ubuntu cloudimg" || return $?
    elif which curl; then
        curl   -o "base/$cloudimg" "${knowledge[$cloudimg]}" || ! echo2 "Failed to download ubuntu cloudimg" || return $?
    fi
Recolic's avatar
Recolic committed
}

function create_vm_if_not_exist () {
Recolic's avatar
Recolic committed
    # TODO: support create VM from existing qcow2 snapshot
Recolic's avatar
Recolic committed
    local name=$1
Recolic's avatar
Recolic committed
    local cloudimg=$2
    local disk=$3
    local username=$4
    local password="$5"
Recolic's avatar
Recolic committed

Recolic's avatar
Recolic committed
    # Check if disk img already exists.
    [[ -f "vm/$name/disk.img" ]] && return

    rm -rf "vm/$name" ; mkdir -p "vm/$name"
Recolic's avatar
Recolic committed

Recolic's avatar
Recolic committed
    echo2 "+ Creating VM image $name with options $@..."
    if [ "$disk" != "" ]; then
Recolic's avatar
Recolic committed
        # create from cloudimg
        download_cloud_img_if_not_exist "$cloudimg" || return $?
Recolic's avatar
Recolic committed
        generate_metadata "$name" > "vm/$name/meta-data" || return $?
        generate_userdata "$username" "$password" "$name" > "vm/$name/user-data" || return $?
        ( cd "vm/$name" ; genisoimage  -output initimg.iso -volid cidata -joliet -rock user-data meta-data ) || return $?
        qemu-img create -f qcow2 -F qcow2 -b "../../base/$cloudimg" "vm/$name/disk.img" || return $?
        qemu-img resize "vm/$name/disk.img" "$disk" || return $?
    else
        # create from baseimg
        download_cloud_img_if_not_exist "$cloudimg" || return $?
Recolic's avatar
Recolic committed
        qemu-img create -f qcow2 -F qcow2 -b "../../base/$cloudimg" "vm/$name/disk.img" || return $?
    fi
Recolic's avatar
Recolic committed
}

function start_vm_if_not_running () {
    local name=$1
Recolic's avatar
Recolic committed
    local options_txt="$2"
    read -a options <<< "$options_txt"
Recolic's avatar
Recolic committed

Recolic's avatar
Recolic committed
    # For tracking started instance
    local uuid=`uuidgen --namespace @oid --name "qemu.$name" --sha1`

    # Check if qemu already running for this instance.
Recolic's avatar
Recolic committed
    ps aux | grep -F "uuid $uuid" | grep qemu > /dev/null 2>&1 && return 0
Recolic's avatar
Recolic committed

Recolic's avatar
.  
Recolic committed
    # start it
Recolic's avatar
Recolic committed
    [[ ! -f "vm/$name/disk.img" ]] && echo2 "In start_vm, disk image vm/$name/disk.img doesn't exist. Did init_vm fail?" && return 1
Recolic's avatar
Recolic committed
    echo2 "+ Starting VM $name with options_txt '$options_txt' and uuid $uuid..."
Recolic's avatar
Recolic committed
    [[ -f "vm/$name/initimg.iso" ]] && options+=(-cdrom "vm/$name/initimg.iso")
Recolic's avatar
Recolic committed
    nohup qemu-system-x86_64 --uuid "$uuid" -drive file="vm/$name/disk.img",if=virtio -cpu host --enable-kvm -net nic,model=rtl8139 "${options[@]}" >> tmp/qemu.log 2>&1 & disown
Recolic's avatar
Recolic committed
}

Recolic's avatar
Recolic committed
function do_init () {
    while IFS= read -r line; do
Recolic's avatar
Recolic committed
        # Ignore lines starting with #
        if [[ "$line" =~ ^\# ]]; then
            continue
        fi
        # Trim leading and trailing whitespaces
        line=$(echo "$line" | sed -e 's/^[[:space:]]*//' -e 's/[[:space:]]*$//')
Recolic's avatar
Recolic committed
        # Check if the line is non-empty
        if [ -n "$line" ]; then
            # Parse the line as "name;cloudimg;disk;username;password", trim space
            IFS=';' read -r name cloudimg disk username password <<< "$(echo "$line" | tr -s '[:space:]' ';')"

Recolic's avatar
Recolic committed
            # 2 options or 5 options allowed, otherwise bad config line.
Recolic's avatar
Recolic committed
            if [ -n "$name" ] && [ -n "$cloudimg" ] && [ -n "$disk" ] && [ -n "$username" ] && [ -n "$password" ]; then
                create_vm_if_not_exist "$name" "$cloudimg" "$disk" "$username" "$password" || echo2 "Failed to create_vm_if_not_exist. $?"
Recolic's avatar
Recolic committed
            elif [ -n "$name" ] && [ -n "$cloudimg" ] && [ ! -n "$disk" ] && [ ! -n "$username" ] && [ ! -n "$password" ]; then
                create_vm_if_not_exist "$name" "$cloudimg" || echo2 "Failed to create_vm_if_not_exist. $?"
Recolic's avatar
Recolic committed
            else
                echo2 "Error: Bad configuration line: $line"
            fi
        fi
    done < ../init.settings
}
Recolic's avatar
Recolic committed

Recolic's avatar
Recolic committed
function do_start () {
    while IFS= read -r line; do
        # Ignore lines starting with #
        if [[ "$line" =~ ^\# ]]; then
            continue
        fi
        # Trim leading and trailing whitespaces
        line=$(echo "$line" | sed -e 's/^[[:space:]]*//' -e 's/[[:space:]]*$//')
        # Check if the line is non-empty
        if [ -n "$line" ]; then
            # Parse the line as "name;options", only trim space in name, options can contain ;
            name=$(echo "$line" | sed -e 's/[[:space:]]*;.*$//' -e 's/^[[:space:]]*//')
Recolic's avatar
Recolic committed
            options=$(echo "$line" | sed 's/^[^;]*;//')
Recolic's avatar
Recolic committed
    
            # Check if the name is empty
            if [ -n "$name" ]; then
Recolic's avatar
Recolic committed
                start_vm_if_not_running "$name" "$options" || echo2 "Failed to start_vm_if_not_running. $?"
Recolic's avatar
Recolic committed
            else
Recolic's avatar
Recolic committed
                echo2 "Error: Bad configuration line: $line"
Recolic's avatar
Recolic committed
            fi
        fi
    done < ../runtime.settings
}
Recolic's avatar
Recolic committed

Recolic's avatar
Recolic committed
do_init
do_start