From 2c6f069c2f5092616a9e60677d6ea056c8b292a6 Mon Sep 17 00:00:00 2001
From: Drew Bonasera <Drewsif@users.noreply.github.com>
Date: Tue, 19 Apr 2016 04:21:22 -0400
Subject: [PATCH] Add safeties

-/etc/rc.local wont be overwritten if the hash matches what the script will write
-If resize2fs fails the script is aborted
---
 pishrink.sh | 21 +++++++++++++++++----
 1 file changed, 17 insertions(+), 4 deletions(-)

diff --git a/pishrink.sh b/pishrink.sh
index 6e30d58..78ccc6b 100755
--- a/pishrink.sh
+++ b/pishrink.sh
@@ -26,20 +26,33 @@ loopback=`losetup -f --show -o $partstart $img`
 #Make pi expand rootfs on next boot
 mountdir=`mktemp -d`
 mount $loopback $mountdir
-mv $mountdir/etc/rc.local $mountdir/etc/rc.expand.tmp
+
+if [ `md5sum $mountdir/etc/rc.local | cut -d ' ' -f 1` != "a27a4d8192ea6ba713d2ddd15a55b1df" ]; then
+echo Creating new /etc/rc.local
+mv $mountdir/etc/rc.local $mountdir/etc/rc.local.bak
 cat <<\EOF > $mountdir/etc/rc.local
-#!/bin/sh
-/usr/bin/raspi-config --expand-rootfs; mv -f /etc/rc.expand.tmp /etc/rc.local; reboot
+#!/bin/bash
+/usr/bin/raspi-config --expand-rootfs
+rm -f /etc/rc.local; cp -f /etc/rc.local.bak /etc/rc.local; reboot
 exit 0
 EOF
 chmod +x $mountdir/etc/rc.local
-umount $loopback
+fi
+umount $mountdir
 
 #Shrink filesystem
 e2fsck -f $loopback
 minsize=`resize2fs -P $loopback | awk -F': ' ' { print $2 } '`
 minsize=`echo $minsize+20000 | bc`
 resize2fs -p $loopback $minsize
+if [[ $? != 0 ]]; then
+  echo ERROR: resize2fs failed...
+  mount $loopback $mountdir
+  mv $mountdir/etc/rc.local.bak $mountdir/etc/rc.local
+  umount $mountdir
+  losetup -d $loopback
+  exit $rc
+fi
 sleep 1
 
 #Shrink partition