From cd98d48d8f2cb6f822578f90814bedc213026bc8 Mon Sep 17 00:00:00 2001 From: Kroese Date: Wed, 24 Apr 2024 07:11:45 +0200 Subject: [PATCH] fix: Check size of downloaded image --- src/install.sh | 137 ++++++++++++++++++++----------------------------- 1 file changed, 57 insertions(+), 80 deletions(-) diff --git a/src/install.sh b/src/install.sh index 0645877..dc02ac9 100644 --- a/src/install.sh +++ b/src/install.sh @@ -99,15 +99,13 @@ finishInstall() { local aborted="$2" if [ ! -s "$iso" ] || [ ! -f "$iso" ]; then - error "Failed to find ISO: $iso" - return 1 + error "Failed to find ISO: $iso" && return 1 fi if [ -w "$iso" ] && [[ "$aborted" != [Yy1]* ]]; then # Mark ISO as prepared via magic byte if ! printf '\x16' | dd of="$iso" bs=1 seek=0 count=1 conv=notrunc status=none; then - error "Failed to set magic byte!" - return 1 + error "Failed to set magic byte!" && return 1 fi fi @@ -125,17 +123,13 @@ abortInstall() { if [[ "$iso" != "$STORAGE/$BASE" ]]; then if ! mv -f "$iso" "$STORAGE/$BASE"; then - error "Failed to move ISO: $iso" - exit 69 + error "Failed to move ISO: $iso" && return 1 fi fi - if ! finishInstall "$STORAGE/$BASE" "Y"; then - error "Failed to finish installation!" - exit 69 - fi + finishInstall "$STORAGE/$BASE" "Y" && return 0 - return 0 + return 1 } startInstall() { @@ -230,8 +224,7 @@ getESD() { winCatalog="https://go.microsoft.com/fwlink/?LinkId=841361" ;; *) - error "Invalid version specified: $VERSION" - return 1 + error "Invalid version specified: $VERSION" && return 1 ;; esac @@ -276,7 +269,7 @@ getESD() { ESD_URL=$(xmllint --nonet --xpath '//FilePath' "${dir}/esd_edition.xml" | sed -E -e 's/<[\/]?FilePath>//g') if [ -z "$ESD_URL" ]; then - error "Failed to find ESD url!" && return 1 + error "Failed to find ESD URL!" && return 1 fi rm -rf "$dir" @@ -328,12 +321,15 @@ downloadImage() { { wget "$url" -O "$iso" -q --no-check-certificate --show-progress "$progress"; rc=$?; } || : fKill "progress.sh" - (( rc != 0 )) && error "Failed to download $url , reason: $rc" && exit 60 + (( rc != 0 )) && error "Failed to download $url , reason: $rc" && return 1 - [ ! -s "$iso" ] || [ ! -f "$iso" ] && return 1 + if [ -f "$iso" ]; then + if [ $(stat -c%s "$iso") -gt 100000000 ]; then + html "Download finished successfully..." && return 0 + fi + fi - html "Download finished successfully..." - return 0 + error "Failed to download $url" && return 1 } extractESD() { @@ -346,22 +342,22 @@ extractESD() { local msg="Extracting $desc bootdisk..." info "$msg" && html "$msg" - size=16106127360 - size_gb=$(( (size + 1073741823)/1073741824 )) - space=$(df --output=avail -B 1 "$TMP" | tail -n 1) - space_gb=$(( (space + 1073741823)/1073741824 )) - - if ((size<10000000)); then - error "Invalid ESD file: Size is smaller than 10 MB" && exit 62 - fi - - if (( size > space )); then - error "Not enough free space in $STORAGE, have $space_gb GB available but need at least $size_gb GB." && exit 63 + if [ $(stat -c%s "$iso") -lt 100000000 ]; then + error "Invalid ESD file: Size is smaller than 100 MB" && return 1 fi rm -rf "$dir" mkdir -p "$dir" + size=16106127360 + size_gb=$(( (size + 1073741823)/1073741824 )) + space=$(df --output=avail -B 1 "$dir" | tail -n 1) + space_gb=$(( (space + 1073741823)/1073741824 )) + + if (( size > space )); then + error "Not enough free space in $STORAGE, have $space_gb GB available but need at least $size_gb GB." && return 1 + fi + local esdImageCount esdImageCount=$(wimlib-imagex info "${iso}" | awk '/Image Count:/ {print $3}') @@ -402,8 +398,7 @@ extractESD() { edition="10 pro" ;; *) - error "Invalid version specified: $VERSION" - return 1 + error "Invalid version specified: $VERSION" && return 1 ;; esac @@ -417,8 +412,7 @@ extractESD() { return 0 done - error "Failed to find product in install.wim!" - return 1 + error "Failed to find product in install.wim!" && return 1 } extractImage() { @@ -429,12 +423,8 @@ extractImage() { local size size_gb space space_gb if [[ "${iso,,}" == *".esd" ]]; then - if ! extractESD "$iso" "$dir"; then - rm -f "$iso" - error "Failed to extract ESD file!" - exit 67 - fi - return 0 + extractESD "$iso" "$dir" && return 0 + return 1 fi if [[ "$EXTERNAL" != [Yy1]* ]] && [ -z "$CUSTOM" ]; then @@ -446,25 +436,26 @@ extractImage() { [ -n "$CUSTOM" ] && msg="Extracting local ISO image..." info "$msg" && html "$msg" + rm -rf "$dir" + mkdir -p "$dir" + size=$(stat -c%s "$iso") size_gb=$(( (size + 1073741823)/1073741824 )) - space=$(df --output=avail -B 1 "$TMP" | tail -n 1) + space=$(df --output=avail -B 1 "$dir" | tail -n 1) space_gb=$(( (space + 1073741823)/1073741824 )) - if ((size<10000000)); then - error "Invalid ISO file: Size is smaller than 10 MB" && exit 62 + if ((size<100000000)); then + error "Invalid ISO file: Size is smaller than 100 MB" && return 1 fi if (( size > space )); then - error "Not enough free space in $STORAGE, have $space_gb GB available but need at least $size_gb GB." && exit 63 + error "Not enough free space in $STORAGE, have $space_gb GB available but need at least $size_gb GB." && return 1 fi rm -rf "$dir" if ! 7z x "$iso" -o"$dir" > /dev/null; then - rm -f "$iso" - error "Failed to extract ISO file!" - exit 66 + error "Failed to extract ISO file: $iso" && return 1 fi return 0 @@ -508,16 +499,14 @@ detectImage() { src=$(find "$dir" -maxdepth 1 -type d -iname sources | head -n 1) if [ ! -d "$src" ]; then - warn "failed to locate 'sources' folder in ISO image, $FB" - return 0 + warn "failed to locate 'sources' folder in ISO image, $FB" && return 1 fi loc=$(find "$src" -maxdepth 1 -type f -iname install.wim | head -n 1) [ ! -f "$loc" ] && loc=$(find "$src" -maxdepth 1 -type f -iname install.esd | head -n 1) if [ ! -f "$loc" ]; then - warn "failed to locate 'install.wim' or 'install.esd' in ISO image, $FB" - return 0 + warn "failed to locate 'install.wim' or 'install.esd' in ISO image, $FB" && return 1 fi tag="DISPLAYNAME" @@ -535,8 +524,7 @@ detectImage() { fi if [ -z "$DETECTED" ]; then - warn "failed to determine Windows version from string '$name', $FB" - return 0 + warn "failed to determine Windows version from string '$name', $FB" && return 0 fi desc=$(printVersion "$DETECTED") @@ -585,16 +573,14 @@ updateImage() { src=$(find "$dir" -maxdepth 1 -type d -iname sources | head -n 1) if [ ! -d "$src" ]; then - warn "failed to locate 'sources' folder in ISO image, $FB" - return 1 + warn "failed to locate 'sources' folder in ISO image, $FB" && return 1 fi loc=$(find "$src" -maxdepth 1 -type f -iname boot.wim | head -n 1) [ ! -f "$loc" ] && loc=$(find "$src" -maxdepth 1 -type f -iname boot.esd | head -n 1) if [ ! -f "$loc" ]; then - warn "failed to locate 'boot.wim' or 'boot.esd' in ISO image, $FB" - return 1 + warn "failed to locate 'boot.wim' or 'boot.esd' in ISO image, $FB" && return 1 fi info "Adding XML file for automatic installation..." @@ -607,8 +593,7 @@ updateImage() { fi if ! wimlib-imagex update "$loc" "$index" --command "add $asset /autounattend.xml" > /dev/null; then - warn "failed to add XML to ISO image, $FB" - return 1 + warn "failed to add XML to ISO image, $FB" && return 1 fi return 0 @@ -638,14 +623,13 @@ buildImage() { space_gb=$(( (space + 1073741823)/1073741824 )) if (( size > space )); then - error "Not enough free space in $STORAGE, have $space_gb GB available but need at least $size_gb GB." - return 1 + error "Not enough free space in $STORAGE, have $space_gb GB available but need at least $size_gb GB." && return 1 fi if ! genisoimage -o "$out" -b "$ETFS" -no-emul-boot -c "$cat" -iso-level 4 -J -l -D -N -joliet-long -relaxed-filenames -V "$label" \ -udf -boot-info-table -eltorito-alt-boot -eltorito-boot "$EFISYS" -no-emul-boot -allow-limited-size -quiet "$dir" 2> "$log"; then [ -s "$log" ] && echo "$(<"$log")" - return 1 + error "Failed to build image!" && return 1 fi local error="" @@ -655,8 +639,7 @@ buildImage() { [[ "$error" != "$hide" ]] && echo "$error" if [ -f "$STORAGE/$BASE" ]; then - error "File $STORAGE/$BASE does already exist?!" - return 1 + error "File $STORAGE/$BASE does already exist?!" && return 1 fi mv "$out" "$STORAGE/$BASE" @@ -678,39 +661,35 @@ bootWindows() { ###################################### if ! startInstall; then - if ! bootWindows; then - error "Failed to boot Windows!" - exit 68 - fi - return 0 + bootWindows && return 0 + exit 68 fi if [ ! -s "$ISO" ] || [ ! -f "$ISO" ]; then - rm -f "$ISO" if ! downloadImage "$ISO" "$VERSION"; then - error "Failed to download $VERSION" + rm -f "$ISO" exit 61 fi fi if ! extractImage "$ISO" "$DIR"; then - abortInstall "$ISO" - return 0 + rm -f "$ISO" + exit 62 fi if ! detectImage "$DIR"; then - abortInstall "$ISO" - return 0 + abortInstall "$ISO" && return 0 + exit 60 fi if ! prepareImage "$ISO" "$DIR"; then - abortInstall "$ISO" - return 0 + abortInstall "$ISO" && return 0 + exit 60 fi if ! updateImage "$ISO" "$DIR" "$XML"; then - abortInstall "$ISO" - return 0 + abortInstall "$ISO" && return 0 + exit 60 fi if ! rm -f "$ISO" 2> /dev/null; then @@ -720,12 +699,10 @@ if ! rm -f "$ISO" 2> /dev/null; then fi if ! buildImage "$DIR"; then - error "Failed to build image!" exit 65 fi if ! finishInstall "$STORAGE/$BASE" "N"; then - error "Failed to finish installation!" exit 69 fi