allow checking sha1 instead of md5

gabriel pettier · gabriel pettier · commit 2f8cf847db42 · 2015-08-09T21:58:38.000+02:00
diff --git a/distribute.sh b/distribute.sh
@@ -69,6 +69,12 @@ if [ "X$MD5SUM" == "X" ]; then
 	fi
 fi
 
+SHA1SUM=$(which sha1sum)
+if [ "X$SHA1SUM" == "X" ]; then
+	echo "Error: you need sha1sum installed."
+	exit 1
+fi
+
 WGET=$(which wget)
 if [ "X$WGET" == "X" ]; then
 	WGET=$(which curl)
@@ -543,9 +549,13 @@ function run_get_packages() {
 
 		url="URL_$module"
 		url=${!url}
+
 		md5="MD5_$module"
 		md5=${!md5}
 
+		sha1="SHA1_$module"
+		sha1=${!sha1}
+
 		if [ ! -d "$BUILD_PATH/$module" ]; then
 			try mkdir -p $BUILD_PATH/$module
 		fi
@@ -581,6 +591,17 @@ function run_get_packages() {
 					error "Module $module have invalid md5, redownload."
 					rm $filename
 				fi
+			elif [ -n "$sha1" ]; then
+				# check if the sha1 is correct
+				current_sha1=$($SHA1SUM $filename | cut -d\  -f1)
+				if [ "X$current_sha1" == "X$sha1" ]; then
+					# correct, no need to download
+					do_download=0
+				else
+					# invalid download, remove the file
+					error "Module $module have invalid sha1, redownload."
+					rm $filename
+				fi
 			else
 				do_download=0
 			fi
