294 lines
7.3 KiB
Bash
294 lines
7.3 KiB
Bash
#!/usr/bin/env bash
|
|
set -Eeuo pipefail
|
|
|
|
detectType() {
|
|
|
|
local dir=""
|
|
local file="$1"
|
|
|
|
[ ! -f "$file" ] && return 1
|
|
[ ! -s "$file" ] && return 1
|
|
|
|
case "${file,,}" in
|
|
*".iso" | *".img" | *".raw" | *".qcow2" )
|
|
BOOT="$file" ;;
|
|
* ) return 1 ;;
|
|
esac
|
|
|
|
[ -n "$BOOT_MODE" ] && return 0
|
|
[[ "${file,,}" != *".iso" ]] && return 0
|
|
|
|
# Automaticly detect UEFI-compatible ISO's
|
|
dir=$(isoinfo -f -i "$file")
|
|
|
|
if [ -z "$dir" ]; then
|
|
BOOT=""
|
|
error "Failed to read ISO file, invalid format!" && return 1
|
|
fi
|
|
|
|
dir=$(echo "${dir^^}" | grep "^/EFI")
|
|
[ -z "$dir" ] && BOOT_MODE="legacy"
|
|
|
|
return 0
|
|
}
|
|
|
|
downloadFile() {
|
|
|
|
local url="$1"
|
|
local base="$2"
|
|
local msg rc total progress
|
|
|
|
local dest="$STORAGE/$base.tmp"
|
|
rm -f "$dest"
|
|
|
|
# Check if running with interactive TTY or redirected to docker log
|
|
if [ -t 1 ]; then
|
|
progress="--progress=bar:noscroll"
|
|
else
|
|
progress="--progress=dot:giga"
|
|
fi
|
|
|
|
msg="Downloading image"
|
|
info "Downloading $base..."
|
|
html "$msg..."
|
|
|
|
/run/progress.sh "$dest" "0" "$msg ([P])..." &
|
|
|
|
{ wget "$url" -O "$dest" -q --timeout=30 --show-progress "$progress"; rc=$?; } || :
|
|
|
|
fKill "progress.sh"
|
|
|
|
if (( rc == 0 )) && [ -f "$dest" ]; then
|
|
total=$(stat -c%s "$dest")
|
|
if [ "$total" -lt 100000 ]; then
|
|
error "Invalid image file: is only $total bytes?" && return 1
|
|
fi
|
|
html "Download finished successfully..."
|
|
mv -f "$dest" "$STORAGE/$base"
|
|
return 0
|
|
fi
|
|
|
|
msg="Failed to download $url"
|
|
(( rc == 3 )) && error "$msg , cannot write file (disk full?)" && return 1
|
|
(( rc == 4 )) && error "$msg , network failure!" && return 1
|
|
(( rc == 8 )) && error "$msg , server issued an error response!" && return 1
|
|
|
|
error "$msg , reason: $rc"
|
|
return 1
|
|
}
|
|
|
|
convertImage() {
|
|
|
|
local source_file=$1
|
|
local source_fmt=$2
|
|
local dst_file=$3
|
|
local dst_fmt=$4
|
|
local dir base fs fa space
|
|
local cur_size src_size disk_param
|
|
|
|
[ -f "$dst_file" ] && error "Conversion failed, destination file $dst_file already exists?" && return 1
|
|
[ ! -f "$source_file" ] && error "Conversion failed, source file $source_file does not exists?" && return 1
|
|
|
|
if [[ "${source_fmt,,}" == "${dst_fmt,,}" ]]; then
|
|
mv -f "$source_file" "$dst_file"
|
|
return 0
|
|
fi
|
|
|
|
local tmp_file="$dst_file.tmp"
|
|
dir=$(dirname "$tmp_file")
|
|
|
|
rm -f "$tmp_file"
|
|
|
|
if [ -n "$ALLOCATE" ] && [[ "$ALLOCATE" != [Nn]* ]]; then
|
|
|
|
# Check free diskspace
|
|
src_size=$(qemu-img info "$source_file" -f "$source_fmt" | grep '^virtual size: ' | sed 's/.*(\(.*\) bytes)/\1/')
|
|
space=$(df --output=avail -B 1 "$dir" | tail -n 1)
|
|
|
|
if (( src_size > space )); then
|
|
local space_gb=$(( (space + 1073741823)/1073741824 ))
|
|
error "Not enough free space to convert image in $dir, it has only $space_gb GB available..." && return 1
|
|
fi
|
|
fi
|
|
|
|
base=$(basename "$source_file")
|
|
info "Converting $base..."
|
|
html "Converting image..."
|
|
|
|
local conv_flags="-p"
|
|
|
|
if [ -z "$ALLOCATE" ] || [[ "$ALLOCATE" == [Nn]* ]]; then
|
|
disk_param="preallocation=off"
|
|
else
|
|
disk_param="preallocation=falloc"
|
|
fi
|
|
|
|
fs=$(stat -f -c %T "$dir")
|
|
[[ "${fs,,}" == "btrfs" ]] && disk_param+=",nocow=on"
|
|
|
|
if [[ "$dst_fmt" != "raw" ]]; then
|
|
if [ -z "$ALLOCATE" ] || [[ "$ALLOCATE" == [Nn]* ]]; then
|
|
conv_flags+=" -c"
|
|
fi
|
|
[ -n "${DISK_FLAGS:-}" ] && disk_param+=",$DISK_FLAGS"
|
|
fi
|
|
|
|
# shellcheck disable=SC2086
|
|
if ! qemu-img convert -f "$source_fmt" $conv_flags -o "$disk_param" -O "$dst_fmt" -- "$source_file" "$tmp_file"; then
|
|
rm -f "$tmp_file"
|
|
error "Failed to convert image in $dir, is there enough space available?" && return 1
|
|
fi
|
|
|
|
if [[ "$dst_fmt" == "raw" ]]; then
|
|
if [ -n "$ALLOCATE" ] && [[ "$ALLOCATE" != [Nn]* ]]; then
|
|
# Work around qemu-img bug
|
|
cur_size=$(stat -c%s "$tmp_file")
|
|
if ! fallocate -l "$cur_size" "$tmp_file"; then
|
|
error "Failed to allocate $cur_size bytes for image!"
|
|
fi
|
|
fi
|
|
fi
|
|
|
|
rm -f "$source_file"
|
|
mv "$tmp_file" "$dst_file"
|
|
|
|
if [[ "${fs,,}" == "btrfs" ]]; then
|
|
fa=$(lsattr "$dst_file")
|
|
if [[ "$fa" != *"C"* ]]; then
|
|
error "Failed to disable COW for image on ${fs^^} filesystem!"
|
|
fi
|
|
fi
|
|
|
|
html "Conversion completed..."
|
|
return 0
|
|
}
|
|
|
|
findFile() {
|
|
|
|
local ext="$1"
|
|
local file
|
|
|
|
file=$(find / -maxdepth 1 -type f -iname "boot.$ext" | head -n 1)
|
|
[ ! -s "$file" ] && file=$(find "$STORAGE" -maxdepth 1 -type f -iname "boot.$ext" | head -n 1)
|
|
detectType "$file" && return 0
|
|
|
|
return 1
|
|
}
|
|
|
|
findFile "iso" && return 0
|
|
findFile "img" && return 0
|
|
findFile "raw" && return 0
|
|
findFile "qcow2" && return 0
|
|
|
|
if [ -z "$BOOT" ] || [[ "$BOOT" == *"example.com/image.iso" ]]; then
|
|
hasDisk && return 0
|
|
error "No boot disk specified, set BOOT= to the URL of a disk image file." && exit 64
|
|
fi
|
|
|
|
base=$(basename "${BOOT%%\?*}")
|
|
: "${base//+/ }"; printf -v base '%b' "${_//%/\\x}"
|
|
base=$(echo "$base" | sed -e 's/[^A-Za-z0-9._-]/_/g')
|
|
|
|
case "${base,,}" in
|
|
|
|
*".iso" | *".img" | *".raw" | *".qcow2" )
|
|
|
|
detectType "$STORAGE/$base" && return 0 ;;
|
|
|
|
*".vdi" | *".vmdk" | *".vhd" | *".vhdx" )
|
|
|
|
detectType "$STORAGE/${base%.*}.img" && return 0
|
|
detectType "$STORAGE/${base%.*}.qcow2" && return 0 ;;
|
|
|
|
*".gz" | *".gzip" | *".xz" | *".7z" | *".zip" | *".rar" | *".lzma" | *".bz" | *".bz2" )
|
|
|
|
case "${base%.*}" in
|
|
|
|
*".iso" | *".img" | *".raw" | *".qcow2" )
|
|
|
|
detectType "$STORAGE/${base%.*}" && return 0 ;;
|
|
|
|
*".vdi" | *".vmdk" | *".vhd" | *".vhdx" )
|
|
|
|
find="${base%.*}"
|
|
|
|
detectType "$STORAGE/${find%.*}.img" && return 0
|
|
detectType "$STORAGE/${find%.*}.qcow2" && return 0 ;;
|
|
|
|
esac ;;
|
|
|
|
* ) error "Unknown file extension, type \".${base/*./}\" is not recognized!" && exit 33 ;;
|
|
esac
|
|
|
|
if ! downloadFile "$BOOT" "$base"; then
|
|
rm -f "$STORAGE/$base.tmp" && exit 60
|
|
fi
|
|
|
|
case "${base,,}" in
|
|
*".gz" | *".gzip" | *".xz" | *".7z" | *".zip" | *".rar" | *".lzma" | *".bz" | *".bz2" )
|
|
info "Extracting $base..."
|
|
html "Extracting image..." ;;
|
|
esac
|
|
|
|
case "${base,,}" in
|
|
*".gz" | *".gzip" )
|
|
|
|
gzip -dc "$STORAGE/$base" > "$STORAGE/${base%.*}"
|
|
rm -f "$STORAGE/$base"
|
|
base="${base%.*}"
|
|
|
|
;;
|
|
*".xz" )
|
|
|
|
xz -dc "$STORAGE/$base" > "$STORAGE/${base%.*}"
|
|
rm -f "$STORAGE/$base"
|
|
base="${base%.*}"
|
|
|
|
;;
|
|
*".7z" | *".zip" | *".rar" | *".lzma" | *".bz" | *".bz2" )
|
|
|
|
tmp="$STORAGE/extract"
|
|
rm -rf "$tmp"
|
|
mkdir -p "$tmp"
|
|
7z x "$STORAGE/$base" -o"$tmp" > /dev/null
|
|
|
|
rm -f "$STORAGE/$base"
|
|
base="${base%.*}"
|
|
|
|
if [ ! -s "$tmp/$base" ]; then
|
|
rm -rf "$tmp"
|
|
error "Cannot find file \"${base}\" in .${BOOT/*./} archive!" && exit 32
|
|
fi
|
|
|
|
mv "$tmp/$base" "$STORAGE/$base"
|
|
rm -rf "$tmp"
|
|
|
|
;;
|
|
esac
|
|
|
|
case "${base,,}" in
|
|
*".iso" | *".img" | *".raw" | *".qcow2" )
|
|
detectType "$STORAGE/$base" && return 0
|
|
error "Cannot read file \"${base}\"" && exit 63 ;;
|
|
esac
|
|
|
|
target_ext="img"
|
|
target_fmt="${DISK_FMT:-}"
|
|
[ -z "$target_fmt" ] && target_fmt="raw"
|
|
[[ "$target_fmt" != "raw" ]] && target_ext="qcow2"
|
|
|
|
case "${base,,}" in
|
|
*".vdi" ) source_fmt="vdi" ;;
|
|
*".vhd" ) source_fmt="vpc" ;;
|
|
*".vhdx" ) source_fmt="vpc" ;;
|
|
*".vmdk" ) source_fmt="vmdk" ;;
|
|
* ) error "Unknown file extension, type \".${base/*./}\" is not recognized!" && exit 33 ;;
|
|
esac
|
|
|
|
dst="$STORAGE/${base%.*}.$target_ext"
|
|
|
|
! convertImage "$STORAGE/$base" "$source_fmt" "$dst" "$target_fmt" && exit 35
|
|
|
|
base=$(basename "$dst")
|
|
detectType "$STORAGE/$base" && return 0
|
|
error "Cannot read file \"${base}\"" && exit 36
|