#!/bin/bash
echo "Disk Restore!";
if [ $# -eq 0 ]; then
    echo "Usage: restore.sh <Backup disk name> <Block Device in /dev>";
    exit 2;
fi
echo "Destination Disk: /dev/$2";
echo "Source Backup Name: $1";
echo "Press Enter to Start...";
read;
uuidTmp="/tmp/a-$(uuidgen)";
mkdir -p $uuidTmp;
gunzip -c "/mnt/parts/$1.part.gz" | dd "of=/dev/$2" status=progress conv=sync,noerror;
echo "Calculating SHA512 sums...";
(sha512sum "/dev/$2" > "$uuidTmp/$2.sum") && dd "if=$uuidTmp/$2.sum" bs=128 count=1 "of=$uuidTmp/$2.hash";
cmp -s "$uuidTmp/$2.hash" "/mnt/hsums/$1.hash";
if [ $? -ne 0 ]; then
    echo "Hashes do not match!";
    exit 1;
fi
(sha512sum "/mnt/tabls/$1.tbl" > "$uuidTmp/$1.tbl.sum") && dd "if=$uuidTmp/$1.tbl.sum" bs=128 count=1 "of=$uuidTmp/$1-tbl.hash";
cmp -s "$uuidTmp/$1-tbl.hash" "/mnt/hsums/$1-tbl.hash";
if [ $? -ne 0 ]; then
    echo "Partition hashes do not match!";
    exit 1;
fi
echo;
echo "Restoring up partition table...";
sfdisk -d "/dev/$2" < "/mnt/tabls/$1.tbl";
rm -rf $uuidTmp;
echo "Finished!";
exit 0;