From 25545a33aab48c1c9780e5a634d5f94d623f8930 Mon Sep 17 00:00:00 2001
From: Iru Cai <mytbk920423@gmail.com>
Date: Sat, 6 Aug 2016 13:14:38 +0800
Subject: download files with wget's -c option

---
 functions.sh | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

(limited to 'functions.sh')

diff --git a/functions.sh b/functions.sh
index 1ee9590..07e3c15 100644
--- a/functions.sh
+++ b/functions.sh
@@ -16,7 +16,7 @@ checksum_verify() {
 	if [[ $_cksum == $_hashsum ]]; then
 		msg "$ISOFILE ok."
 	else
-		fatalerror "$ISOFILE checksum bad!"
+		msg "$ISOFILE checksum bad!" && return 1
 	fi
 }
 
@@ -32,7 +32,17 @@ download_iso() {
 	mkdir -p isofiles
 	for url in ${MIRRORLIST[@]}
 	do
-		wget -O "isofiles/$ISOFILE" "$url/$ISOURL" && return 0
+		wget -c -O "isofiles/$ISOFILE" "$url/$ISOURL"
+		if checksum_verify; then
+			return 0
+		else
+			# checksum bad, may be due to a bad partial download
+			# so remove the file and try again
+			rm -f "isofiles/$ISOFILE"
+			wget -O "isofiles/$ISOFILE" "$url/$ISOURL"
+			checksum_verify && return 0
+			rm -f "isofiles/$ISOFILE" # then try next mirror
+		fi
 	done
 	fatalerror "Fail to download $ISOFILE!"
 }
-- 
cgit v1.2.3