thinkpad-ec/scripts/checksum
2019-02-14 20:57:10 +08:00

84 lines
1.7 KiB
Bash
Executable File

#!/bin/sh
#
# Copyright (C) 2017-2019 Hamish Coleman
#
# Look up the given filename in the description file and confirm that
# the checksum matches
#
# FIXME
# - the description file is expected to be in the current dir
# - does not handle description file entries with more than just the sha1 tag
DESCRIPTIONS='Descriptions.txt'
if [ "$1" = "--mv_on_fail" ]; then
MV_ON_FAIL=yes
shift
elif [ "$1" = "--rm_on_fail" ]; then
RM_ON_FAIL=yes
shift
fi
FILE="$1"
if [ -z "$FILE" ]; then
echo "Need filename" 2>&1
exit 1
fi
shift
# The name of the Descriptions.txt entry to use
CHECKNAME="$1"
if [ -z "$CHECKNAME" ]; then
# default to the same as the filename
CHECKNAME="$FILE"
fi
do_exit_fail() {
if [ "$MV_ON_FAIL" ]; then
BADNAME="${FILE}.bad"
echo "ERROR: moved file to $BADNAME" 2>&1
mv -f "$FILE" "$BADNAME"
elif [ "$RM_ON_FAIL" ]; then
echo "ERROR: removed file" 2>&1
rm -f "$FILE"
fi
exit 1
}
if ! LINE=$(grep -E "^$CHECKNAME " "$DESCRIPTIONS"); then
echo "Unknown file $CHECKNAME" 2>&1
do_exit_fail
fi
if [ ! -e "$FILE" ]; then
echo "File $FILE does not exist" 2>&1
do_exit_fail
fi
SUM_FIELD=$(echo "$LINE" | awk -- '{print $2}')
TYPE=$(echo "$SUM_FIELD" | cut -d: -f1)
EXPECT=$(echo "$SUM_FIELD" | cut -d: -f2 |tr '[:upper:]' '[:lower:]')
case "$TYPE" in
sha1)
GOT=$(sha1sum "$FILE" | cut -d" " -f1)
;;
sha256)
GOT=$(sha256sum "$FILE" | cut -d" " -f1)
;;
*)
echo "Unknown checksum type $TYPE for $CHECKNAME" 2>&1
do_exit_fail
;;
esac
if [ "$EXPECT" != "$GOT" ]; then
echo "File $FILE failed checksum check" 2>&1
do_exit_fail
fi
exit 0