mirrors
/
hashboot
mirror of https://schlomp.space/tastytea/hashboot
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Merge pull request #2 from teldra/modular 

Modularitaet eingebaut
pull/3/head 0.9.1
Teldra 2016-10-04 23:36:02 +02:00 committed by GitHub
parent 37a4399293 b1a00d8a4e
commit db4669df68
1 changed files with 87 additions and 50 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

137
hashboot
View File

@ -12,7 +12,7 @@
# meet some day, and you think this stuff is worth it, you can give us a hug. # # meet some day, and you think this stuff is worth it, you can give us a hug. #
################################################################################### ###################################################################################
VERSION="0.9.0" VERSION="0.9.1"
PATH="/bin:/usr/bin:/sbin:/usr/sbin:${PATH}" PATH="/bin:/usr/bin:/sbin:/usr/sbin:${PATH}"
DIGEST_FILE="/var/lib/hashboot.digest" DIGEST_FILE="/var/lib/hashboot.digest"
@ -29,6 +29,11 @@ CONFIG_FILE="/etc/hashboot.cfg"
COUNTER=0 COUNTER=0
DD_STATUS="none" DD_STATUS="none"
PROGRAMMER="no" #standard change enables bios mode PROGRAMMER="no" #standard change enables bios mode
#bitmask:
# 001=mbr
# 010=files
# 100=bios
CKMODES=011
#Umount /boot if we mounted it, exit with given exit code #Umount /boot if we mounted it, exit with given exit code
die () die ()
@ -46,21 +51,31 @@ write_hashes ()
{ {
#Write header to ${1} #Write header to ${1}
echo "#hashboot ${VERSION} - Algorithm: $(basename ${HASHER})" > ${1} echo "#hashboot ${VERSION} - Algorithm: $(basename ${HASHER})" > ${1}
#Write MBR of MBR_DEVICE to ${1}
dd if=${MBR_DEVICE} of=${MBR_TMP} bs=${MBR_SIZE}K count=1 status=${DD_STATUS} || die 8 if [ $((${CKMODES} & 001)) != 0 ]; then
#Write bios to ${1} #copy mbr to file
#if we set an programmer chip in config, read bios and write it to digestfile. dd if=${MBR_DEVICE} of=${MBR_TMP} bs=${MBR_SIZE}K count=1 status=${DD_STATUS} || die 8
if [ ! ${PROGRAMMER} == "no" ]; then #Write hash of MBR to ${1}
flashrom --programmer ${PROGRAMMER} -r ${BIOS_TMP} > /dev/null 2>&1 ${HASHER} ${MBR_TMP} >> ${1}
${HASHER} ${BIOS_TMP} >> ${1} fi
if [ $((${CKMODES} & 010)) != 0 ]; then
#Write hashes of all regular files to ${1}
find /boot -type f -exec ${HASHER} --binary {} >> ${1} +
fi
if [ $((${CKMODES} & 100)) != 0 ]; then
#if we set an programmer chip in config
if [ ! ${PROGRAMMER} == "no" ]; then
#read bios to file
flashrom --programmer ${PROGRAMMER} -r ${BIOS_TMP} > /dev/null 2>&1
#and write hashes of bios files to ${1}
${HASHER} ${BIOS_TMP} >> ${1}
fi
fi fi
#Write hashes of all regular files to ${1}
${HASHER} ${MBR_TMP} >> ${1}
find /boot -type f -exec ${HASHER} --binary {} >> ${1} +
} }
#If we're not root: exit #If we're not root: exit
if [ ${UID} -ne 0 ] if [ ${UID} != "0" ]
then then
die 4 "You have to be root" die 4 "You have to be root"
fi fi
@ -99,11 +114,14 @@ else
echo "#Where the Backup files are stored" >> ${CONFIG_FILE} echo "#Where the Backup files are stored" >> ${CONFIG_FILE}
echo "BACKUP_FILE=${BACKUP_FILE}" >> ${CONFIG_FILE} echo "BACKUP_FILE=${BACKUP_FILE}" >> ${CONFIG_FILE}
echo -n "Include BIOS check? (y/n)" echo -n "Include BIOS check? (y/n)"
read b read prompt
if [ ${b} == "y"]; then while [[ ! $prompt == "y" || $prompt == "Y" || $prompt == "n" || $prompt == "N" ]]; do
read prompt
done
if [ "${prompt}" == "y" ]; then
if which flashrom; then if which flashrom; then
flashrom flashrom
echo -n "Which programmer? (eg. internal)" echo -n "Which programmer? (eg. internal) "
read p read p
echo "PROGRAMMER=${p}" >> ${CONFIG_FILE} echo "PROGRAMMER=${p}" >> ${CONFIG_FILE}
else else
@ -113,33 +131,47 @@ else
else else
echo "PROGRAMMER=no" >> ${CONFIG_FILE} echo "PROGRAMMER=no" >> ${CONFIG_FILE}
fi fi
echo "What do we check?"
echo "001=mbr"
echo "010=files"
echo "100=bios"
echo "eg. 101 for mbr and bios: "
read CKMODES
echo "CKMODES=$CKMODES" >> ${CONFIG_FILE}
else else
echo "#Device with the MBR on it" > ${CONFIG_FILE} echo "#Device with the MBR on it" > ${CONFIG_FILE}
echo "MBR_DEVICE=${MBR_DEVICE}" >> ${CONFIG_FILE} echo "MBR_DEVICE=${MBR_DEVICE}" >> ${CONFIG_FILE}
echo "#Where the Backup files are stored" >> ${CONFIG_FILE} echo "#Where the Backup files are stored" >> ${CONFIG_FILE}
echo "BACKUP_FILE=${BACKUP_FILE}" >> ${CONFIG_FILE} echo "BACKUP_FILE=${BACKUP_FILE}" >> ${CONFIG_FILE}
echo "CKMODES=$CKMODES" >> ${CONFIG_FILE}
echo "PROGRAMMER=${PROGRAMMER}" >> ${CONFIG_FILE} echo "PROGRAMMER=${PROGRAMMER}" >> ${CONFIG_FILE}
fi fi
fi fi
# Find out where the first partition starts and set ${MBR_SIZE} in KiB if [ "${2}" > "1" ]; then
sectorsize=$(LC_ALL=C fdisk -l ${MBR_DEVICE} | grep '^Units' | awk '{print $8}' ) CKMODES=${2}
if [ "${sectorsize}" == "=" ] # Older versions of util-linux
then
sectorsize=$(LC_ALL=C fdisk -l ${MBR_DEVICE} | grep '^Units' | awk '{print $9}' )
fi
startsector=$(LC_ALL=C fdisk -l ${MBR_DEVICE} | grep -A1 'Device' | tail -n1 | awk '{print $2}' )
if [ "${startsector}" == "*" ] # If partition is marked as boot, read next field
then
startsector=$(LC_ALL=C fdisk -l ${MBR_DEVICE} | grep -A1 'Device' | tail -n1 | awk '{print $3}' )
fi fi
MBR_SIZE=$(expr ${sectorsize} \* ${startsector} / 1024) if [ $((${CKMODES} & 001)) != 0 ]; then
# Find out where the first partition starts and set ${MBR_SIZE} in KiB
sectorsize=$(LC_ALL=C fdisk -l ${MBR_DEVICE} | grep '^Units' | awk '{print $8}' )
if [ "${sectorsize}" == "=" ] # Older versions of util-linux
then
sectorsize=$(LC_ALL=C fdisk -l ${MBR_DEVICE} | grep '^Units' | awk '{print $9}' )
fi
startsector=$(LC_ALL=C fdisk -l ${MBR_DEVICE} | grep -A1 'Device' | tail -n1 | awk '{print $2}' )
if [ "${startsector}" == "*" ] # If partition is marked as boot, read next field
then
startsector=$(LC_ALL=C fdisk -l ${MBR_DEVICE} | grep -A1 'Device' | tail -n1 | awk '{print $3}' )
fi
if [ ${?} != 0 ] MBR_SIZE=$(expr ${sectorsize} \* ${startsector} / 1024)
then
echo "Something went wrong. Most likely your partition table is corrupt." >&2 if [ ${?} != 0 ]
die 1 "You have to recover the MBR manually by copying the mbr from ${BACKUP_FILE}" then
echo "Something went wrong. Most likely your partition table is corrupt." >&2
die 1 "You have to recover the MBR manually by copying the mbr from ${BACKUP_FILE}"
fi
fi fi
if [ "${1}" == "index" ] if [ "${1}" == "index" ]
@ -189,29 +221,34 @@ elif [ "${1}" == "check" ]
then then
[ -f ${DIGEST_FILE} ] || die 9 "No digestfile" [ -f ${DIGEST_FILE} ] || die 9 "No digestfile"
HASHER=$(head -n1 ${DIGEST_FILE} | awk '{print $5}') HASHER=$(head -n1 ${DIGEST_FILE} | awk '{print $5}')
dd if=${MBR_DEVICE} of=${MBR_TMP} bs=${MBR_SIZE}K count=1 status=${DD_STATUS} || die 8 if [ $((${CKMODES} & 001)) != 0 ]; then
flashrom --programmer ${PROGRAMMER} -r ${BIOS_TMP} > /dev/null 2>&1 dd if=${MBR_DEVICE} of=${MBR_TMP} bs=${MBR_SIZE}K count=1 status=${DD_STATUS} || die 8
if ! grep ${MBR_TMP} ${DIGEST_FILE} | ${HASHER} --check --warn --quiet --strict | tee ${LOG_FILE} if ! grep ${MBR_TMP} ${DIGEST_FILE} | ${HASHER} --check --warn --quiet --strict | tee ${LOG_FILE}
then then
echo " !! TIME TO PANIK: MBR WAS MODIFIED !!" echo " !! TIME TO PANIK: MBR WAS MODIFIED !!"
COUNTER=$((COUNTER + 1)) COUNTER=$((COUNTER + 1))
fi
fi fi
if ! grep -v ${MBR_TMP} ${DIGEST_FILE} | grep -v ${BIOS_TMP} | ${HASHER} --check --warn --quiet --strict | tee -a ${LOG_FILE} if [ $((${CKMODES} & 010)) != 0 ]; then
then if ! grep -v ${MBR_TMP} ${DIGEST_FILE} | grep -v ${BIOS_TMP} | ${HASHER} --check --warn --quiet --strict | tee -a ${LOG_FILE}
echo " !! TIME TO PANIK: AT LEAST 1 FILE WAS MODIFIED !!" then
COUNTER=$((COUNTER + 2)) echo " !! TIME TO PANIK: AT LEAST 1 FILE WAS MODIFIED !!"
die $COUNTER COUNTER=$((COUNTER + 2))
die $COUNTER
fi
fi fi
#if we set an programmer chip in config, find line with hash for bios and compare. if smthg wrong, panic if [ $((${CKMODES} & 100)) != 0 ]; then
if [ ! ${PROGRAMMER} == "no" ]; then flashrom --programmer ${PROGRAMMER} -r ${BIOS_TMP} > /dev/null 2>&1
if grep ${BIOS_TMP} ${DIGEST_FILE} | ${HASHER} --check --warn --quiet --strict | tee -a ${LOG_FILE} #if we set an programmer chip in config, find line with hash for bios and compare. if smthg wrong, panic
then if [ ! ${PROGRAMMER} == "no" ]; then
echo " !! TIME TO PANIK: BIOS WAS MODIFIED !!" if ! grep ${BIOS_TMP} ${DIGEST_FILE} | ${HASHER} --check --warn --quiet --strict | tee -a ${LOG_FILE}
COUNTER=$((COUNTER + 10)) then
die $COUNTER echo " !! TIME TO PANIK: BIOS WAS MODIFIED !!"
fi COUNTER=$((COUNTER + 10))
die $COUNTER
fi
fi
fi fi
elif [ "${1}" == "recover" ] elif [ "${1}" == "recover" ]
then then
echo "Restoring files from backup... (type yes or no for each file)" echo "Restoring files from backup... (type yes or no for each file)"