functionerror() { echo -n "$SCRIPTNAME: ERROR occured in line $1: " shift echo"$@" }
functioncleanup() { if losetup "$loopback" &>/dev/null; then losetup -d "$loopback" fi if [ "$debug" = true ]; then local old_owner=$(stat -c %u:%g "$src") chown "$old_owner""$LOGFILE" fi
}
functionlogVariables() { 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 }
info "Trying to recover corrupted filesystem" e2fsck -y "$loopback" (( $? < 4 )) && return
if [[ $repair == true ]]; then info "Trying to recover corrupted filesystem - Phase 2" e2fsck -fy -b 32768 "$loopback" (( $? < 4 )) && return fi error $LINENO"Filesystem recoveries failed. Giving up..." exit -9
}
help() { localhelp read -r -d ''help << EOM Usage: $0 [-sdrpzh] imagefile.img [newimagefile.img] -s: Don't expand filesystem when image is booted the first time -d: Write debug messages in a debug log file -r: Use advanced filesystem repair option if the normal one fails -p: Remove logs, apt archives, dhcp leases and ssh hostkeys -z: Gzip compress image after shrinking EOM echo"$help" exit -1 }
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 src="$1" img="$1"
#Usage checks if [[ -z "$img" ]]; then usage fi if [[ ! -f "$img" ]]; then error $LINENO"$img is not a file..." exit -2 fi if (( EUID != 0 )); then error $LINENO"You need to be running as root." exit -3 fi
#Check that what we need is installed forcommandin parted losetup tune2fs md5sum e2fsck resize2fs; do command -v $command >/dev/null 2>&1 if (( $? != 0 )); then error $LINENO"$command is not installed." exit -4 fi done
#Copy to new file if requested if [ -n "$2" ]; then info "Copying $1 to $2..." cp --reflink=auto --sparse=always "$1""$2" if (( $? != 0 )); then error $LINENO"Could not copy file..." exit -5 fi old_owner=$(stat -c %u:%g "$1") chown "$old_owner""$2" img="$2" fi
#Check if we should make pi expand rootfs on next boot if [ "$should_skip_autoexpand" = false ]; then #Make pi expand rootfs on next boot mountdir=$(mktemp -d) mount "$loopback""$mountdir"
if [ "$(md5sum "$mountdir/etc/rc.local" | cut -d ' ' -f 1)" != "0542054e9ff2d2e0507ea1ffe7d4fc87" ]; then echo"Creating new /etc/rc.local" mv "$mountdir/etc/rc.local""$mountdir/etc/rc.local.bak" #####Do not touch the following lines##### cat <<\EOF1 > "$mountdir/etc/rc.local" #!/bin/bash do_expand_rootfs() { ROOT_PART=$(mount | sed -n 's|^/dev/\(.*\) on / .*|\1|p') PART_NUM=${ROOT_PART#mmcblk0p} if [ "$PART_NUM" = "$ROOT_PART" ]; then echo"$ROOT_PART is not an SD card. Don't know how to expand" return 0 fi # Get the starting offset of the root partition PART_START=$(parted /dev/mmcblk0 -ms unit s p | grep "^${PART_NUM}" | cut -f 2 -d: | sed 's/[^0-9]//g') [ "$PART_START" ] || return 1 # Return value will likely be error for fdisk as it fails to reload the # partition table because the root fs is mounted fdisk /dev/mmcblk0 <<EOF p d $PART_NUM n p $PART_NUM $PART_START p w EOF cat <<EOF > /etc/rc.local && #!/bin/sh echo "Expanding /dev/$ROOT_PART" resize2fs /dev/$ROOT_PART rm -f /etc/rc.local; cp -f /etc/rc.local.bak /etc/rc.local; /etc/rc.local EOF reboot exit } raspi_config_expand() { /usr/bin/env raspi-config --expand-rootfs if [[ $? != 0 ]]; then return -1 else rm -f /etc/rc.local; cp -f /etc/rc.local.bak /etc/rc.local; /etc/rc.local reboot exit fi } raspi_config_expand echo"WARNING: Using backup expand..." sleep 5 do_expand_rootfs echo"ERROR: Expanding failed..." sleep 5 rm -f /etc/rc.local; cp -f /etc/rc.local.bak /etc/rc.local; /etc/rc.local exit 0 EOF1 #####End no touch zone##### chmod +x "$mountdir/etc/rc.local" fi umount "$mountdir" else echo"Skipping autoexpanding process..." fi
if [[ $prep == true ]]; then info "Syspreping: Removing logs, apt archives, dhcp leases and ssh hostkeys" mountdir=$(mktemp -d) mount "$loopback""$mountdir" rm -rf "$mountdir/var/cache/apt/archives/*""$mountdir/var/lib/dhcpcd5/*""$mountdir/var/log/*""$mountdir/var/tmp/*""$mountdir/tmp/*""$mountdir/etc/ssh/*_host_*" umount "$mountdir" fi
#Make sure filesystem is ok checkFilesystem
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 error $LINENO"Image already shrunk to smallest size" exit -11 fi
#Add some free space to the end of the filesystem extra_space=$(($currentsize - $minsize)) logVariables $LINENO extra_space for space in 5000 1000 100; do if [[ $extra_space -gt $space ]]; then minsize=$(($minsize + $space)) break fi done logVariables $LINENO minsize
#Shrink filesystem info "Shrinking filesystem" resize2fs -p "$loopback"$minsize if [[ $? != 0 ]]; then error $LINENO"resize2fs failed" mount "$loopback""$mountdir" mv "$mountdir/etc/rc.local.bak""$mountdir/etc/rc.local" umount "$mountdir" losetup -d "$loopback" exit -12 fi sleep 1
#Shrink partition partnewsize=$(($minsize * $blocksize)) newpartend=$(($partstart + $partnewsize)) logVariables $LINENO partnewsize newpartend 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 info "Shrinking image" if ! endresult=$(parted -ms "$img" unit B print free); then rc=$? error $LINENO"parted failed with rc $rc" exit -15 fi
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 fi
if [[ $gzip_compress == true ]]; then info "Gzipping the shrunk image" if [[ ! $(gzip -f9 "$img") ]]; then img=$img.gz fi fi