zabpehely
/
PiShrink
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Initial debug and error checking version

This commit is contained in:
framp 2018-08-17 21:48:21 +02:00
parent fd6c0b1ba4
commit 8a0637b3da
1 changed files with 130 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

148
pishrink.sh
View File

@ -1,24 +1,69 @@
#!/bin/bash #!/bin/bash
version="v0.1"
function info() {
echo "$1..."
}
function error() {
echo -n "ERROR occured in line $1: "
shift
echo "$@"
}
function cleanup() { function cleanup() {
if losetup $loopback &>/dev/null; then if losetup $loopback &>/dev/null; then
losetup -d "$loopback" losetup -d "$loopback"
fi fi
if [ "$debug" = true ]; then
local old_owner=$(stat -c %u:%g "$src")
chown $old_owner "$LOGFILE"
fi
} }
usage() { echo "Usage: $0 [-s] imagefile.img [newimagefile.img]"; exit -1; } function logVariables() {
if [ "$debug" = true ]; then
echo "Line $1" >> "$LOGFILE"
shift
local v var
for var in "$@"; do
eval "v=\$$var"
echo "$var: $v" >> $LOGFILE
done
fi
}
usage() { echo "Usage: $0 [-sd] imagefile.img [newimagefile.img]"; exit -1; }
should_skip_autoexpand=false should_skip_autoexpand=false
debug=false
while getopts ":s" opt; do while getopts ":sd" opt; do
case "${opt}" in case "${opt}" in
s) should_skip_autoexpand=true ;; s) should_skip_autoexpand=true ;;
d) debug=true;;
*) usage ;; *) usage ;;
esac esac
done done
shift $((OPTIND-1)) shift $((OPTIND-1))
CURRENT_DIR=$(pwd)
SCRIPTNAME="${0##*/}"
LOGFILE=${CURRENT_DIR}/${SCRIPTNAME%.*}.log
if [ "$debug" = true ]; then
info "Creating log file $LOGFILE"
rm $LOGFILE &>/dev/null
exec 1> >(stdbuf -i0 -o0 -e0 tee -a "$LOGFILE" >&1)
exec 2> >(stdbuf -i0 -o0 -e0 tee -a "$LOGFILE" >&2)
fi
echo "${0##*/} $version"
#Args #Args
src="$1"
img="$1" img="$1"
#Usage checks #Usage checks
@ -26,11 +71,11 @@ if [[ -z "$img" ]]; then
usage usage
fi fi
if [[ ! -f "$img" ]]; then if [[ ! -f "$img" ]]; then
echo "ERROR: $img is not a file..." error $LINENO "$img is not a file..."
exit -2 exit -2
fi fi
if (( EUID != 0 )); then if (( EUID != 0 )); then
echo "ERROR: You need to be running as root." error $LINENO "You need to be running as root."
exit -3 exit -3
fi fi
@ -38,17 +83,17 @@ fi
for command in parted losetup tune2fs md5sum e2fsck resize2fs; do for command in parted losetup tune2fs md5sum e2fsck resize2fs; do
which $command 2>&1 >/dev/null which $command 2>&1 >/dev/null
if (( $? != 0 )); then if (( $? != 0 )); then
echo "ERROR: $command is not installed." error $LINENO "$command is not installed."
exit -4 exit -4
fi fi
done done
#Copy to new file if requested #Copy to new file if requested
if [ -n "$2" ]; then if [ -n "$2" ]; then
echo "Copying $1 to $2..." info "Copying $1 to $2..."
cp --reflink=auto --sparse=always "$1" "$2" cp --reflink=auto --sparse=always "$1" "$2"
if (( $? != 0 )); then if (( $? != 0 )); then
echo "ERROR: Could not copy file..." error $LINENO "Could not copy file..."
exit -5 exit -5
fi fi
old_owner=$(stat -c %u:%g "$1") old_owner=$(stat -c %u:%g "$1")
@ -60,15 +105,36 @@ fi
trap cleanup ERR EXIT trap cleanup ERR EXIT
#Gather info #Gather info
info "Gatherin data"
beforesize=$(ls -lh "$img" | cut -d ' ' -f 5) beforesize=$(ls -lh "$img" | cut -d ' ' -f 5)
parted_output=$(parted -ms "$img" unit B print | tail -n 1) logVariables $LINENO beforesize
if ! parted_output=$(parted -ms "$img" unit B print); then
rc=$?
error $LINENO "parted failed with rc $rc"
exit -6
fi
parted_output=$(tail -n 1 <<< $parted_output)
logVariables $LINENO parted_output
partnum=$(echo "$parted_output" | cut -d ':' -f 1) partnum=$(echo "$parted_output" | cut -d ':' -f 1)
partstart=$(echo "$parted_output" | cut -d ':' -f 2 | tr -d 'B') partstart=$(echo "$parted_output" | cut -d ':' -f 2 | tr -d 'B')
loopback=$(losetup -f --show -o $partstart "$img") logVariables $LINENO partnum partstart
tune2fs_output=$(tune2fs -l "$loopback")
info "Mounting image"
if ! loopback=$(losetup -f --show -o $partstart "$img"); then
rc=$?
error $LINENO "losetup failed with rc $rc"
exit -7
fi
if ! tune2fs_output=$(tune2fs -l "$loopback"); then
error $LINENO "tunefs failed"
exit -8
fi
currentsize=$(echo "$tune2fs_output" | grep '^Block count:' | tr -d ' ' | cut -d ':' -f 2) currentsize=$(echo "$tune2fs_output" | grep '^Block count:' | tr -d ' ' | cut -d ':' -f 2)
blocksize=$(echo "$tune2fs_output" | grep '^Block size:' | tr -d ' ' | cut -d ':' -f 2) blocksize=$(echo "$tune2fs_output" | grep '^Block size:' | tr -d ' ' | cut -d ':' -f 2)
logVariables $LINENO tune2fs_output currentsize blocksize
#Check if we should make pi expand rootfs on next boot #Check if we should make pi expand rootfs on next boot
if [ "$should_skip_autoexpand" = false ]; then if [ "$should_skip_autoexpand" = false ]; then
#Make pi expand rootfs on next boot #Make pi expand rootfs on next boot
@ -146,43 +212,81 @@ else
fi fi
#Make sure filesystem is ok #Make sure filesystem is ok
e2fsck -p -f "$loopback" info "Checking filesystem"
minsize=$(resize2fs -P "$loopback" | cut -d ':' -f 2 | tr -d ' ') if ! e2fsck -p -f "$loopback"; then
rc=$?
error $LINENO "fsck failed with rc $rc"
exit -9
fi
if ! minsize=$(resize2fs -P "$loopback"); then
rc=$?
error $LINENO "resize2fs failed with rc $rc"
exit -10
fi
minsize=$(cut -d ':' -f 2 <<< $minsize | tr -d ' ')
logVariables $LINENO minsize
if [[ $currentsize -eq $minsize ]]; then if [[ $currentsize -eq $minsize ]]; then
echo "ERROR: Image already shrunk to smallest size" error $LINENO "Image already shrunk to smallest size"
exit -6 exit -11
fi fi
#Add some free space to the end of the filesystem #Add some free space to the end of the filesystem
extra_space=$(($currentsize - $minsize)) extra_space=$(($currentsize - $minsize))
logVariables $LINENO extra_space
for space in 5000 1000 100; do for space in 5000 1000 100; do
if [[ $extra_space -gt $space ]]; then if [[ $extra_space -gt $space ]]; then
minsize=$(($minsize + $space)) minsize=$(($minsize + $space))
break break
fi fi
done done
logVariables $LINENO minsize
#Shrink filesystem #Shrink filesystem
info "Shrinking filesystem"
resize2fs -p "$loopback" $minsize resize2fs -p "$loopback" $minsize
if [[ $? != 0 ]]; then if [[ $? != 0 ]]; then
echo "ERROR: resize2fs failed..." error $LINENO "resize2fs failed"
mount "$loopback" "$mountdir" mount "$loopback" "$mountdir"
mv "$mountdir/etc/rc.local.bak" "$mountdir/etc/rc.local" mv "$mountdir/etc/rc.local.bak" "$mountdir/etc/rc.local"
umount "$mountdir" umount "$mountdir"
losetup -d "$loopback" losetup -d "$loopback"
exit -7 exit -12
fi fi
sleep 1 sleep 1
#Shrink partition #Shrink partition
partnewsize=$(($minsize * $blocksize)) partnewsize=$(($minsize * $blocksize))
newpartend=$(($partstart + $partnewsize)) newpartend=$(($partstart + $partnewsize))
parted -s -a minimal "$img" rm $partnum >/dev/null logVariables $LINENO partnewsize newpartend
parted -s "$img" unit B mkpart primary $partstart $newpartend >/dev/null if ! parted -s -a minimal "$img" rm $partnum; then
rc=$?
error $LINENO "parted failed with rc $rc"
exit -13
fi
if ! parted -s "$img" unit B mkpart primary $partstart $newpartend; then
rc=$?
error $LINENO "parted failed with rc $rc"
exit -14
fi
#Truncate the file #Truncate the file
endresult=$(parted -ms "$img" unit B print free | tail -1 | cut -d ':' -f 2 | tr -d 'B') info "Shrinking image"
truncate -s $endresult "$img" if ! endresult=$(parted -ms "$img" unit B print free); then
aftersize=$(ls -lh "$img" | cut -d ' ' -f 5) rc=$?
error $LINENO "parted failed with rc $rc"
exit -15
fi
echo "Shrunk $img from $beforesize to $aftersize" endresult=$(tail -1 <<< $endresult | cut -d ':' -f 2 | tr -d 'B')
logVariables $LINENO endresult
if ! truncate -s $endresult "$img"; then
rc=$?
error $LINENO "trunate failed with rc $rc"
exit -16
aftersize=$(ls -lh "$img" | cut -d ' ' -f 5)
logVariables $LINENO aftersize
info "Shrunk $img from $beforesize to $aftersize"