add rip-image script

build/rip-image.sh

@@ -0,0 +1,95 @@
+#!/bin/bash
+
+set -e
+
+function mktmpfifo () {
+	TMP_PATH=$(mktemp)
+	rm $TMP_PATH
+	mkfifo $TMP_PATH
+	echo $TMP_PATH
+}
+
+if ! which pv > /dev/null; then
+	>&2 echo 'This script would like to use `pv` to show a progress indicator, but it is not installed.'
+	if which apt-get > /dev/null; then
+		read -p "Install? [y/N]" -n 1 -r
+		echo
+		if [[ "$REPLY" =~ ^[Yy]$ ]]; then
+			sudo apt-get install pv
+		fi
+	elif which pacman > /dev/null; then
+		read -p "Install? [y/N]" -n 1 -r
+		echo
+		if [[ "$REPLY" =~ ^[Yy]$ ]]; then
+			sudo pacman -S pv
+		fi
+	elif which brew > /dev/null; then
+		read -p "Install? [y/N]" -n 1 -r
+		echo
+		if [[ "$REPLY" =~ ^[Yy]$ ]]; then
+			brew install pv
+		fi
+	else
+		>&2 echo 'This script does not recognize what package manager you have available on your system.'
+		>&2 echo 'Please go install the utility manually if you want progress reporting.'
+	fi
+fi
+
+if [[ "$(uname)" == "Darwin" ]]; then
+	>&2 echo 'OSX not supported'
+	exit 1
+fi
+
+if ! test -e /dev/disk/by-label/green; then
+	>&2 echo '`green` partition not found'
+	exit 1
+fi
+export SOURCE_PARTITION=$(readlink -f /dev/disk/by-label/green)
+
+if [[ "$SOURCE_PARTITION" =~ ^/dev/loop ]]; then
+	>&2 echo 'You are currently ripping from a loop device.'
+	>&2 echo 'This is probably a mistake, and usually means you failed to detach a .img file.'
+	read -p "Continue anyway? [y/N]" -n 1 -r
+	echo
+	if ! [[ "$REPLY" =~ ^[Yy]$ ]]; then
+        	exit 1
+	fi
+fi
+
+sudo e2fsck -f ${SOURCE_PARTITION}
+sudo resize2fs -M ${SOURCE_PARTITION}
+export BLOCK_INFO=$(sudo dumpe2fs ${SOURCE_PARTITION})
+export BLOCK_COUNT=$(echo "$BLOCK_INFO" | grep "Block count:" | sed 's/Block count:\s\+//g')
+export BLOCK_SIZE=$(echo "$BLOCK_INFO" | grep "Block size:" | sed 's/Block size:\s\+//g')
+export FS_SIZE=$[$BLOCK_COUNT*$BLOCK_SIZE]
+
+echo "Ripping $FS_SIZE bytes from $SOURCE_PARTITION"
+if which pv > /dev/null; then
+	sudo cat ${SOURCE_PARTITION} | head -c $FS_SIZE | pv -s $FS_SIZE | sudo dd of=update.img bs=1M iflag=fullblock oflag=direct conv=fsync 2>/dev/null 
+else
+	sudo cat ${SOURCE_PARTITION} | head -c $FS_SIZE | sudo dd of=update.img bs=1M iflag=fullblock oflag=direct conv=fsync
+fi
+echo Verifying...
+export INPUT_HASH=$(mktemp)
+export OUTPUT_HASH=$(mktemp)
+if which pv > /dev/null; then
+	export PV_IN=$(mktmpfifo)
+fi
+sudo cat ${SOURCE_PARTITION} | head -c $FS_SIZE | tee -a $PV_IN | sha256sum > $INPUT_HASH &
+export INPUT_CHILD=$!
+sudo cat update.img | head -c $FS_SIZE | tee -a $PV_IN | sha256sum > $OUTPUT_HASH &
+export OUTPUT_CHILD=$!
+if which pv > /dev/null; then
+	pv -s $[$FS_SIZE*2] < $PV_IN > /dev/null &
+fi
+wait $INPUT_CHILD $OUTPUT_CHILD
+if which pv > /dev/null; then
+	rm $PV_IN
+fi
+if ! [[ "$(cat $INPUT_HASH)" == "$(cat $OUTPUT_HASH)" ]]; then
+	rm $INPUT_HASH $OUTPUT_HASH
+	>&2 echo Verification Failed
+	exit 1
+fi
+rm $INPUT_HASH $OUTPUT_HASH
+echo "Verification Succeeded"