wicinst.sh

#!/bin/bash
#
# Copyright (c) Roberto A. Foglietta, 2022
#
# Authors:
#  Roberto A. Foglietta <roberto.foglietta@gmail.com>
#
# SPDX-License-Identifier: MIT
#

function vmdk_get_uuid() {
        dd if="$1" count=16 2>/dev/null | strings | sed -ne "s,ddb.uuid.image=\"\(.*\)\",\\1,p"
}

if [ "$1" == "" ]; then
	echo
	echo "USAGE: $(basename $0) [--nosync] [file:|pigz:|vmdk:]<destination> [size_in_gigabytes]"
	echo
	exit 1
fi

cd $(dirname $0)
topdir=$PWD

declare -i sizeinc=0
while [ "$1" != "" ]; do
	if [ -b "$1" ]; then
		bdev="$1"
	elif echo "$1" | grep -qe "^file:"; then
		file="${1/file:/}"
		file=$(readlink -f "${file/#\~/$HOME}")
	elif echo "$1" | grep -qe "^pigz:"; then
		pigz="${1/pigz:/}"a
		pigz=$(readlink -f "${pigz/#\~/$HOME}")
	elif echo "$1" | grep -qe "^vmdk:"; then
		vmdk="${1/vmdk:/}"
		vmdk=$(readlink -f "${vmdk/#\~/$HOME}")
	elif echo "$1" | grep -qe "^ovaf:"; then
		ovaf="${1/ovaf:/}"
		ovaf=$(readlink -f "${ovaf/#\~/$HOME}")
	elif [ -e "$1" ]; then
		fimg=$(readlink -f "${fimg/#\~/$HOME}")
	elif [ "x$1" == "x--nosync" ]; then
		sync() { echo "OPTION: no sync"; }
		export -f sync
	else 
		sizeinc=$[($1*1024*1024*1024)-1]
	fi
	shift
done

if [ ! -e "$fimg" ]; then
	fimg=$(ls -1 build/tmp/deploy/images/*/*.wic)
	n=( $fimg )
	if [ ${#n[@]} -gt 1 ]; then
		echo
		echo "WARNING: more than one image found, choose one:"
		echo
		echo "$fimg"
		echo
		exit 1
	fi
fi

if [ ! -e "$fimg" ]; then
	echo
	echo "ERROR: no any image found, abort!"
	echo
	exit 1
fi

declare -i size=$(du -b "$fimg" | cut -f1)
declare -i szmb=$(du -m "$fimg" | cut -f1)
echo
echo "Transfering ${szmb}Mb: $fimg => ${bdev}${file}${pigz}${vmdk}${ovaf} ..."

if [ $sizeinc -gt $[szmb*1024] ]; then
	echo -n "Enlarging wic image from $szmb Mb to "
	trap "truncate -s $size $fimg" EXIT
	dd if=/dev/zero bs=1 seek=$sizeinc count=1 oflag=append conv=notrunc,sparse status=none of=$fimg
	csize=$(du -b $fimg | cut -f1)
	csize=$[(csize+512)/1024]
	echo "$[(csize+512)/1024] Mb"
fi

if [ -n "$bdev" ]; then
	if which bmaptool >/dev/null && test -f "$fimg.bmap"; then
		time (sudo bmaptool copy $fimg $bdev; sync $bdev)
	else
		echo
		echo "WARNING: bmaptool allows to write on a block device much faster than dd"
		echo
		echo "         sudo apt install bmap-tools"
		echo
		time (sudo dd if="$fimg" bs=1M of=$bdev status=progress; sync $bdev)
	fi
elif [ -n "$file" ]; then 
	ln -Pf "$fimg" "$file" 2>/dev/null ||\
	    time (cp -paf "$fimg" "$file"; sync "$file")
elif [ -n "$pigz" ]; then
	time (pigz -c "$fimg" | dd bs=1M status=progress >"$pigz"; sync "$pigz")
elif [ -n "$vmdk" ]; then
	if ! which qemu-img >/dev/null; then
		echo
		echo "WARNING: qemu-utils is missing, press a ENTER to install"
		echo
		read key
		sudo apt install -f qemu-utils
	fi
	uuid=$(fdisk -l "$fimg" | sed -ne "s,Disk identifier: \(.*\),\\1,p" | tr '[A-Z]' '[a-z]')
	echo "UUID read: $uuid"
	echo "$uuid" | grep -q "^........-....-....-....-............$" || uuid=""
	if which VBoxManage >/dev/null; then
		uuid=${uuid:-$(cat /proc/sys/kernel/random/uuid)}
		echo "UUID used: $uuid"
		rm -f "$vmdk"
		time (VBoxManage convertfromraw "$fimg" --format VMDK "$vmdk" --uuid "$uuid"; sync "$vmdk")
	elif which qemu-img >/dev/null; then
		time (qemu-img convert -p -f raw "$fimg" -O vmdk "$vmdk"; sync "$vmdk")
		uuid=$(vmdk_get_uuid "$vmdk")
		echo "UUID used: ${uuid:-none}"
	else
		echo
		echo "ERROR: no VBoxManage nor qemu-img are available, abort!"
		echo
		exit 1
	fi
elif [ -n "$ovaf" ]; then
	$topdir/makeova.sh "$ovaf"
fi
echo