|
|
#!/usr/bin/env bash |
|
|
set -Eeuo pipefail |
|
|
|
|
|
getBase() { |
|
|
|
|
|
local base="${1%%\?*}" |
|
|
base=$(basename "$base") |
|
|
base="${base//+/ }" |
|
|
printf -v base '%b' "${base//%/\\x}" |
|
|
base="${base//[!A-Za-z0-9._-]/_}" |
|
|
|
|
|
echo "$base" |
|
|
return 0 |
|
|
} |
|
|
|
|
|
getFolder() { |
|
|
|
|
|
local base="" |
|
|
local result="$1" |
|
|
|
|
|
if [[ "$result" != *"."* ]]; then |
|
|
|
|
|
result="${result,,}" |
|
|
|
|
|
else |
|
|
|
|
|
base=$(getBase "$result") |
|
|
result="${base%.*}" |
|
|
|
|
|
case "${base,,}" in |
|
|
|
|
|
*".gz" | *".gzip" | *".xz" | *".7z" | *".zip" | *".rar" | *".lzma" | *".bz" | *".bz2" ) |
|
|
|
|
|
[[ "$result" == *"."* ]] && result="${result%.*}" ;; |
|
|
|
|
|
esac |
|
|
|
|
|
fi |
|
|
|
|
|
[ -z "$result" ] && result="unknown" |
|
|
echo "$result" |
|
|
|
|
|
return 0 |
|
|
} |
|
|
|
|
|
moveFile() { |
|
|
|
|
|
local file="$1" |
|
|
local ext="${file##*.}" |
|
|
local dest="$STORAGE/boot.$ext" |
|
|
|
|
|
if [[ "$file" == "$dest" ]]; then |
|
|
BOOT="$file" |
|
|
return 0 |
|
|
fi |
|
|
|
|
|
if [[ "${file,,}" == "/boot.${ext,,}" || "${file,,}" == "/custom.${ext,,}" ]]; then |
|
|
BOOT="$file" |
|
|
return 0 |
|
|
fi |
|
|
|
|
|
if ! mv -f "$file" "$dest"; then |
|
|
error "Failed to move $file to $dest !" |
|
|
return 1 |
|
|
fi |
|
|
|
|
|
BOOT="$dest" |
|
|
return 0 |
|
|
} |
|
|
|
|
|
detectType() { |
|
|
|
|
|
local file="$1" |
|
|
local result="" |
|
|
|
|
|
[ ! -f "$file" ] && return 1 |
|
|
[ ! -s "$file" ] && return 1 |
|
|
|
|
|
case "${file,,}" in |
|
|
*".iso" | *".img" | *".raw" | *".qcow2" ) ;; |
|
|
* ) return 1 ;; |
|
|
esac |
|
|
|
|
|
if [ -n "$BOOT_MODE" ] || [[ "${file,,}" == *".qcow2" ]]; then |
|
|
moveFile "$file" && return 0 |
|
|
return 1 |
|
|
fi |
|
|
|
|
|
if [[ "${file,,}" == *".iso" ]]; then |
|
|
|
|
|
result=$(head -c 512 "$file" | tail -c 2 | xxd -p) |
|
|
|
|
|
if [[ "$result" != "0000" ]]; then |
|
|
[ -z "${HYBRID:-}" ] && HYBRID="Y" |
|
|
fi |
|
|
|
|
|
if [[ "${HYBRID:-}" != [Yy]* ]]; then |
|
|
|
|
|
result=$(isoinfo -f -i "$file" 2>/dev/null) |
|
|
|
|
|
if [ -z "$result" ]; then |
|
|
error "Failed to read ISO file, invalid format!" |
|
|
return 1 |
|
|
fi |
|
|
|
|
|
result=$(echo "${result^^}" | grep "^/EFI") |
|
|
[ -z "$result" ] && BOOT_MODE="legacy" |
|
|
|
|
|
moveFile "$file" && return 0 |
|
|
return 1 |
|
|
|
|
|
fi |
|
|
fi |
|
|
|
|
|
result=$(fdisk -l "$file" 2>/dev/null) |
|
|
[[ "${result^^}" != *"EFI "* ]] && BOOT_MODE="legacy" |
|
|
|
|
|
moveFile "$file" && return 0 |
|
|
return 1 |
|
|
} |
|
|
|
|
|
delay() { |
|
|
|
|
|
local i |
|
|
local delay="$1" |
|
|
local msg="Retrying failed download in X seconds..." |
|
|
|
|
|
info "${msg/X/$delay}" |
|
|
|
|
|
for i in $(seq "$delay" -1 1); do |
|
|
html "${msg/X/$i}" |
|
|
sleep 1 |
|
|
done |
|
|
|
|
|
return 0 |
|
|
} |
|
|
|
|
|
downloadFile() { |
|
|
|
|
|
local url="$1" |
|
|
local base="$2" |
|
|
local name="$3" |
|
|
local msg rc total size progress |
|
|
|
|
|
local dest="$STORAGE/$base" |
|
|
|
|
|
|
|
|
if [ -t 1 ]; then |
|
|
progress="--progress=bar:noscroll" |
|
|
else |
|
|
progress="--progress=dot:giga" |
|
|
fi |
|
|
|
|
|
if [ -z "$name" ]; then |
|
|
msg="Downloading image" |
|
|
info "Downloading $base..." |
|
|
else |
|
|
msg="Downloading $name" |
|
|
info "Downloading $name..." |
|
|
fi |
|
|
|
|
|
html "$msg..." |
|
|
|
|
|
/run/progress.sh "$dest" "0" "$msg ([P])..." & |
|
|
|
|
|
{ wget "$url" -O "$dest" --continue -q --timeout=30 --no-http-keep-alive --show-progress "$progress"; rc=$?; } || : |
|
|
|
|
|
fKill "progress.sh" |
|
|
|
|
|
if (( rc == 0 )) && [ -f "$dest" ]; then |
|
|
total=$(stat -c%s "$dest") |
|
|
size=$(formatBytes "$total") |
|
|
if [ "$total" -lt 100000 ]; then |
|
|
error "Invalid image file: is only $size ?" && return 1 |
|
|
fi |
|
|
html "Download finished successfully..." |
|
|
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 space_gb |
|
|
local cur_size cur_gb 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 |
|
|
|
|
|
|
|
|
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 |
|
|
space_gb=$(formatBytes "$space") |
|
|
error "Not enough free space to convert image in $dir, it has only $space_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 |
|
|
|
|
|
|
|
|
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 |
|
|
|
|
|
cur_size=$(stat -c%s "$tmp_file") |
|
|
cur_gb=$(formatBytes "$cur_size") |
|
|
if ! fallocate -l "$cur_size" "$tmp_file" &>/dev/null; then |
|
|
if ! fallocate -l -x "$cur_size" "$tmp_file"; then |
|
|
error "Failed to allocate $cur_gb for image!" |
|
|
fi |
|
|
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 dir file |
|
|
local base="$1" |
|
|
local ext="$2" |
|
|
local fname="${base}.${ext}" |
|
|
|
|
|
dir=$(find / -maxdepth 1 -type d -iname "$fname" -print -quit) |
|
|
[ ! -d "$dir" ] && dir=$(find "$STORAGE" -maxdepth 1 -type d -iname "$fname" -print -quit) |
|
|
|
|
|
if [ -d "$dir" ]; then |
|
|
if hasDisk; then |
|
|
BOOT="none" |
|
|
return 0 |
|
|
fi |
|
|
error "The bind $dir maps to a file that does not exist!" && exit 37 |
|
|
fi |
|
|
|
|
|
file=$(find / -maxdepth 1 -type f -iname "$fname" -print -quit) |
|
|
[ ! -s "$file" ] && file=$(find "$STORAGE" -maxdepth 1 -type f -iname "$fname" -print -quit) |
|
|
|
|
|
detectType "$file" && return 0 |
|
|
|
|
|
return 1 |
|
|
} |
|
|
|
|
|
findFile "boot" "img" && return 0 |
|
|
findFile "boot" "raw" && return 0 |
|
|
findFile "boot" "iso" && return 0 |
|
|
findFile "boot" "qcow2" && return 0 |
|
|
findFile "custom" "iso" && return 0 |
|
|
|
|
|
if hasDisk; then |
|
|
BOOT="none" |
|
|
return 0 |
|
|
fi |
|
|
|
|
|
if [[ "${BOOT}" == \"*\" || "${BOOT}" == \'*\' ]]; then |
|
|
VERSION="${BOOT:1:-1}" |
|
|
fi |
|
|
|
|
|
BOOT=$(expr "$BOOT" : "^\ *\(.*[^ ]\)\ *$") |
|
|
|
|
|
if [ -z "$BOOT" ] || [[ "$BOOT" == *"example.com/"* ]]; then |
|
|
|
|
|
BOOT="alpine" |
|
|
warn "no value specified for the BOOT variable, defaulting to \"${BOOT}\"." |
|
|
|
|
|
fi |
|
|
|
|
|
folder=$(getFolder "$BOOT") |
|
|
STORAGE="$STORAGE/$folder" |
|
|
|
|
|
if [ -d "$STORAGE" ]; then |
|
|
|
|
|
findFile "boot" "img" && return 0 |
|
|
findFile "boot" "raw" && return 0 |
|
|
findFile "boot" "iso" && return 0 |
|
|
findFile "boot" "qcow2" && return 0 |
|
|
findFile "custom" "iso" && return 0 |
|
|
|
|
|
if hasDisk; then |
|
|
BOOT="none" |
|
|
return 0 |
|
|
fi |
|
|
|
|
|
fi |
|
|
|
|
|
name=$(getURL "$BOOT" "name") || exit 34 |
|
|
|
|
|
if [ -n "$name" ]; then |
|
|
|
|
|
msg="Retrieving latest $name version..." |
|
|
info "$msg" && html "$msg..." |
|
|
|
|
|
url=$(getURL "$BOOT" "url") || exit 34 |
|
|
|
|
|
[ -n "$url" ] && BOOT="$url" |
|
|
|
|
|
fi |
|
|
|
|
|
if [[ "$BOOT" != *"."* ]]; then |
|
|
if [ -z "$BOOT" ]; then |
|
|
error "No BOOT value specified!" |
|
|
else |
|
|
error "Invalid BOOT value specified, option \"$BOOT\" is not recognized!" |
|
|
fi |
|
|
exit 64 |
|
|
fi |
|
|
|
|
|
if [[ "${BOOT,,}" != "http"* ]]; then |
|
|
error "Invalid BOOT value specified, \"$BOOT\" is not a valid URL!" && exit 64 |
|
|
fi |
|
|
|
|
|
mkdir -p "$STORAGE" |
|
|
|
|
|
find "$STORAGE" -maxdepth 1 -type f \( -iname '*.rom' -or -iname '*.vars' \) -delete |
|
|
find "$STORAGE" -maxdepth 1 -type f \( -iname 'data.*' -or -iname 'qemu.*' \) -delete |
|
|
|
|
|
base=$(getBase "$BOOT") |
|
|
|
|
|
rm -f "$STORAGE/$base" |
|
|
|
|
|
if ! downloadFile "$BOOT" "$base" "$name"; then |
|
|
delay 5 |
|
|
if ! downloadFile "$BOOT" "$base" "$name"; then |
|
|
delay 10 |
|
|
if ! downloadFile "$BOOT" "$base" "$name"; then |
|
|
rm -f "$STORAGE/$base" && exit 60 |
|
|
fi |
|
|
fi |
|
|
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 convert file \"${base}\"" && exit 36 |
|
|
|
