Compare commits
108 Commits
Author | SHA1 | Date |
---|---|---|
Drew Bonasera | aa14313d04 | |
Drew Bonasera | 48d175607e | |
Drew Bonasera | 87e1ff064e | |
Drew Bonasera | 43f1a885cd | |
wh201906 | be00c478e0 | |
Drew Bonasera | 9d9c0dd32c | |
freddii | fa3c7a2740 | |
Drew Bonasera | 2b6afa184e | |
Jonathan Lai | f6be7f3f50 | |
Drew Bonasera | ee4e7ac5c8 | |
Terry Tsang | a7977b9960 | |
Drew Bonasera | dfdf279278 | |
Drew Bonasera | ad4657a0aa | |
Drew Bonasera | a846e0c013 | |
Drew Bonasera | 762acea7fc | |
Drew Bonasera | cc4ca215b4 | |
Drew Bonasera | 2f5b275675 | |
Drew Bonasera | 66355825bd | |
Drew Bonasera | 53a59cf9a0 | |
Drew Bonasera | 574fab3e52 | |
Drew Bonasera | 370e88d581 | |
Drew Bonasera | c80642512e | |
Drew Bonasera | 9f817a3483 | |
Drew Bonasera | 071799b926 | |
framp | 0a45a5fdae | |
framp | 198e6c944f | |
framp | c1c8a0b4a2 | |
Drew Bonasera | 26cd138b9b | |
Drew Bonasera | d38666fefd | |
framp | 3c3f68f59f | |
framp | 9a2c7fe949 | |
Drew Bonasera | 24c6f57efd | |
Drew Bonasera | 18c86f53fc | |
Drew Bonasera | 6969b930eb | |
kmpm | c99b053afb | |
framp | 547f73f71c | |
OmegaSquad82 | f337674f58 | |
framp | d80d48a645 | |
framp | b4db6c1a64 | |
framp | fda7a38ba1 | |
framp | c7d74a73e7 | |
framp | ab32008981 | |
framp | 965fc206d7 | |
framp | 7515ee5631 | |
framp | 3a931ddb73 | |
framp | 4ed3162284 | |
framp | 9de4f53b96 | |
Drew Bonasera | 4f61a3244a | |
daveilers | 74b106e763 | |
Drew Bonasera | 83eb694145 | |
Drew Bonasera | de507888ad | |
wfroemgen | 6de66c2605 | |
wfroemgen | 054f6c7902 | |
Drew Bonasera | 17346f5d02 | |
Drew Bonasera | 20ae97d2bc | |
Drew Bonasera | 97b9a5db39 | |
Drew Bonasera | 27f8780508 | |
Drew Bonasera | 562142b841 | |
Drew Bonasera | 95f4839152 | |
framp | cfe5cc5534 | |
framp | 1dfe90f650 | |
Drew Bonasera | cbd4720ea9 | |
Drew Bonasera | 76d2254d90 | |
clach04 | c51d0697a2 | |
Drew Bonasera | c857e166c9 | |
framp | db62822a42 | |
framp | 700bf3cdcc | |
framp | 15c43f1199 | |
framp | 8a0637b3da | |
framp | c68ff6d502 | |
framp | 0fea4cc2a7 | |
framp | b0ffc3afd9 | |
Drew Bonasera | fd6c0b1ba4 | |
framp | 8aae06a4b2 | |
framp | 3dbbbd6df6 | |
framp | 67add2e613 | |
framp | eda1092cc5 | |
Drew Bonasera | 83e72eeec3 | |
Drew Bonasera | eac933beb8 | |
Drew Bonasera | 54051ea2e3 | |
Drew Bonasera | 551e9950ac | |
Drew Bonasera | 8dccf7d154 | |
Andrew Scheller | 3568798dd4 | |
Andrew Scheller | 201db16c93 | |
Andrew Scheller | 8331faf8f3 | |
Andrew Scheller | e3ff4d20aa | |
Andrew Scheller | ec95ba8097 | |
Andrew Scheller | c69f57affe | |
Andrew Scheller | 8784bda989 | |
Andrew Scheller | 49964b512c | |
Andrew Scheller | 0b19d6c6ea | |
Andrew Scheller | b787dd4e66 | |
Andrew Scheller | b11de9a8ee | |
Andrew Scheller | 9d6392d0c2 | |
framp | 2d2c4d5802 | |
Drew Bonasera | 2aa44f0ecd | |
Drew Bonasera | e7857ff47c | |
Andrew Scheller | 2229ae89a6 | |
Drew Bonasera | c40b443072 | |
Drew Bonasera | 5a89ff4262 | |
framp | f83e3f341d | |
Drew Bonasera | da3bf15dac | |
framp | 4af698457e | |
Drew Bonasera | ff89083249 | |
Drew Bonasera | 63b7509ade | |
Drew Bonasera | e326f426bb | |
Wesley Archer | f5246e8898 | |
Sean Mayo | 0b04de8118 |
81
README.md
81
README.md
|
@ -1,15 +1,57 @@
|
||||||
|
|
||||||
# PiShrink #
|
# PiShrink #
|
||||||
PiShrink is a bash script that automatically shrink a pi image that will then resize to the max size of the SD card on boot. This will make putting the image back onto the SD card faster and the shrunk images will compress better. The script can also work directly from the SD card reader device.
|
|
||||||
|
|
||||||
`Usage: ./pishrink [-s] [-i] imagefile.img [newimagefile.img]`
|
PiShrink is a bash script that automatically shrink a pi image that will then resize to the max size of the SD card on boot. This will make putting the image back onto the SD card faster and the shrunk images will compress better.
|
||||||
./pishrink [-s] [-i] /dev/sd_device [newimagefile.img]`
|
In addition the shrunk image can be compressed with gzip and xz to create an even smaller image. Parallel compression of the image
|
||||||
|
using multiple cores is supported.
|
||||||
|
|
||||||
If the `-s` option is given the script will skip the autoexpanding part of the process. If you specify the `newimagefile.img` parameter, the script will make a copy of `imagefile.img` and work off that. You will need enough space to make a full copy of the image to use that option.
|
## Usage ##
|
||||||
If a device file (e.g. /dev/mmcblk0) is given as parameter, it will shrink the image directly on the SD card. If the `-i` option is given the script will first shrink in place, then prepare a copy. This is especially useful when the first parameter is a device file and space is limited.
|
|
||||||
|
```
|
||||||
|
Usage: $0 [-adhrsvzZ] imagefile.img [newimagefile.img]
|
||||||
|
|
||||||
|
-s Don't expand filesystem when image is booted the first time
|
||||||
|
-v Be verbose
|
||||||
|
-r Use advanced filesystem repair option if the normal one fails
|
||||||
|
-z Compress image after shrinking with gzip
|
||||||
|
-Z Compress image after shrinking with xz
|
||||||
|
-a Compress image in parallel using multiple cores
|
||||||
|
-d Write debug messages in a debug log file
|
||||||
|
```
|
||||||
|
|
||||||
|
If you specify the `newimagefile.img` parameter, the script will make a copy of `imagefile.img` and work off that. You will need enough space to make a full copy of the image to use that option.
|
||||||
|
|
||||||
|
* `-s` prevents automatic filesystem expansion on the images next boot
|
||||||
|
* `-v` enables more verbose output
|
||||||
|
* `-r` will attempt to repair the filesystem using additional options if the normal repair fails
|
||||||
|
* `-z` will compress the image after shrinking using gzip. `.gz` extension will be added to the filename.
|
||||||
|
* `-Z` will compress the image after shrinking using xz. `.xz` extension will be added to the filename.
|
||||||
|
* `-a` will use option -f9 for pigz and option -T0 for xz and compress in parallel.
|
||||||
|
* `-d` will create a logfile `pishrink.log` which may help for problem analysis.
|
||||||
|
|
||||||
|
Default options for compressors can be overwritten by defining PISHRINK_GZIP or PSHRINK_XZ environment variables for gzip and xz.
|
||||||
|
|
||||||
|
## Prerequisites ##
|
||||||
|
|
||||||
|
If you are running PiShrink in VirtualBox you will likely encounter an error if you
|
||||||
|
attempt to use VirtualBox's "Shared Folder" feature. You can copy the image you wish to
|
||||||
|
shrink on to the VM from a Shared Folder, but shrinking directctly from the Shared Folder
|
||||||
|
is know to cause issues.
|
||||||
|
|
||||||
|
If using Ubuntu, you will likely see an error about `e2fsck` being out of date and `metadata_csum`. The simplest fix for this is to use Ubuntu 16.10 and up, as it will save you a lot of hassle in the long run.
|
||||||
|
|
||||||
|
## Installation ##
|
||||||
|
|
||||||
|
```bash
|
||||||
|
wget https://raw.githubusercontent.com/Drewsif/PiShrink/master/pishrink.sh
|
||||||
|
chmod +x pishrink.sh
|
||||||
|
sudo mv pishrink.sh /usr/local/bin
|
||||||
|
```
|
||||||
|
|
||||||
## Example ##
|
## Example ##
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
[user@localhost PiShrink]$ sudo ./pishrink.sh pi.img
|
[user@localhost PiShrink]$ sudo pishrink.sh pi.img
|
||||||
e2fsck 1.42.9 (28-Dec-2013)
|
e2fsck 1.42.9 (28-Dec-2013)
|
||||||
Pass 1: Checking inodes, blocks, and sizes
|
Pass 1: Checking inodes, blocks, and sizes
|
||||||
Pass 2: Checking directory structure
|
Pass 2: Checking directory structure
|
||||||
|
@ -30,34 +72,9 @@ The filesystem on /dev/loop1 is now 773603 blocks long.
|
||||||
|
|
||||||
Shrunk pi.img from 30G to 3.1G
|
Shrunk pi.img from 30G to 3.1G
|
||||||
```
|
```
|
||||||
## Example: copying shrinking directly from SD ##
|
|
||||||
|
|
||||||
[user@localhost PiShrink]$ sudo ./pishrink.sh /dev/mmcblk0 pi.img
|
|
||||||
Copying /dev/mmcblk0 to pi.img...
|
|
||||||
30528+0 records in
|
|
||||||
30528+0 records out
|
|
||||||
32010928128 bytes (32 GB, 30 GiB) copied, 587,625 s, 54,5 MB/s
|
|
||||||
Creating new /etc/rc.local
|
|
||||||
e2fsck 1.42.13 (17-May-2015)
|
|
||||||
Pass 1: Checking inodes, blocks, and sizes
|
|
||||||
Pass 2: Checking directory structure
|
|
||||||
Pass 3: Checking directory connectivity
|
|
||||||
Pass 4: Checking reference counts
|
|
||||||
Pass 5: Checking group summary information
|
|
||||||
/dev/loop0: 127844/1915424 files (0.1% non-contiguous), 1743616/7798016 blocks
|
|
||||||
resize2fs 1.42.13 (17-May-2015)
|
|
||||||
resize2fs 1.42.13 (17-May-2015)
|
|
||||||
Resizing the filesystem on /dev/loop0 to 2187260 (4k) blocks.
|
|
||||||
Begin pass 3 (max = 238)
|
|
||||||
Scanning inode table XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
|
|
||||||
The filesystem on /dev/loop0 is now 2187260 (4k) blocks long.
|
|
||||||
Shrunk pi.img from 30G to 8,5G
|
|
||||||
|
|
||||||
## Example: shrinking directly on SD ad copying ##
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
## Contributing ##
|
## Contributing ##
|
||||||
|
|
||||||
If you find a bug please create an issue for it. If you would like a new feature added, you can create an issue for it but I can't promise that I will get to it.
|
If you find a bug please create an issue for it. If you would like a new feature added, you can create an issue for it but I can't promise that I will get to it.
|
||||||
|
|
||||||
Pull requests for new features and bug fixes are more than welcome!
|
Pull requests for new features and bug fixes are more than welcome!
|
||||||
|
|
487
pishrink.sh
487
pishrink.sh
|
@ -1,89 +1,113 @@
|
||||||
#!/bin/bash
|
#!/bin/bash
|
||||||
|
|
||||||
usage() { echo "Usage: $0 [-s] [-i] imagefile.img [newimagefile.img]"; exit -1; }
|
version="v0.1.4"
|
||||||
|
|
||||||
should_skip_autoexpand=false
|
CURRENT_DIR="$(pwd)"
|
||||||
force_inplace_and_copy_after=false
|
SCRIPTNAME="${0##*/}"
|
||||||
|
MYNAME="${SCRIPTNAME%.*}"
|
||||||
|
LOGFILE="${CURRENT_DIR}/${SCRIPTNAME%.*}.log"
|
||||||
|
REQUIRED_TOOLS="parted losetup tune2fs md5sum e2fsck resize2fs"
|
||||||
|
ZIPTOOLS=("gzip xz")
|
||||||
|
declare -A ZIP_PARALLEL_TOOL=( [gzip]="pigz" [xz]="xz" ) # parallel zip tool to use in parallel mode
|
||||||
|
declare -A ZIP_PARALLEL_OPTIONS=( [gzip]="-f9" [xz]="-T0" ) # options for zip tools in parallel mode
|
||||||
|
declare -A ZIPEXTENSIONS=( [gzip]="gz" [xz]="xz" ) # extensions of zipped files
|
||||||
|
|
||||||
while getopts ":s:i" opt; do
|
function info() {
|
||||||
case "${opt}" in
|
echo "$SCRIPTNAME: $1 ..."
|
||||||
s) should_skip_autoexpand=true ;;
|
}
|
||||||
i) force_inplace_and_copy_after=true ;;
|
|
||||||
*) usage ;;
|
|
||||||
esac
|
|
||||||
done
|
|
||||||
shift $((OPTIND-1))
|
|
||||||
|
|
||||||
#Args
|
function error() {
|
||||||
img=$1
|
echo -n "$SCRIPTNAME: ERROR occurred in line $1: "
|
||||||
|
shift
|
||||||
|
echo "$@"
|
||||||
|
}
|
||||||
|
|
||||||
#Usage checks
|
function cleanup() {
|
||||||
if [[ -z $img ]]; then
|
if losetup "$loopback" &>/dev/null; then
|
||||||
usage
|
losetup -d "$loopback"
|
||||||
fi
|
|
||||||
if [[ ! -e $img ]]; then
|
|
||||||
echo "ERROR: $img is not a file..."
|
|
||||||
exit -2
|
|
||||||
fi
|
|
||||||
if (( EUID != 0 )); then
|
|
||||||
echo "ERROR: You need to be running as root."
|
|
||||||
exit -3
|
|
||||||
fi
|
|
||||||
|
|
||||||
#Check that what we need is installed
|
|
||||||
A=`which parted 2>&1`
|
|
||||||
if (( $? != 0 )); then
|
|
||||||
echo "ERROR: parted is not installed."
|
|
||||||
exit -4
|
|
||||||
fi
|
|
||||||
|
|
||||||
#Copy to new file if requested
|
|
||||||
if [ -n "$2" -a "$force_inplace_and_copy_after" = false ]; then
|
|
||||||
echo "Copying $1 to $2..."
|
|
||||||
if [[ -f $img ]]; then
|
|
||||||
cp --reflink=auto --sparse=always "$1" "$2"
|
|
||||||
else
|
|
||||||
imgsize=`parted -m $img unit B print | tail -1 | cut -d ':' -f 3 | tr -d 'B\n'`
|
|
||||||
imgsize=`expr $imgsize + 1`
|
|
||||||
dd if="$1" of="$2" conv=sparse count=$imgsize iflag=count_bytes bs=1M
|
|
||||||
fi
|
fi
|
||||||
if (( $? != 0 )); then
|
if [ "$debug" = true ]; then
|
||||||
echo "ERROR: Could not copy file..."
|
local old_owner=$(stat -c %u:%g "$src")
|
||||||
exit -5
|
chown "$old_owner" "$LOGFILE"
|
||||||
fi
|
fi
|
||||||
img=$2
|
|
||||||
|
}
|
||||||
|
|
||||||
|
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
|
||||||
|
}
|
||||||
|
|
||||||
|
function checkFilesystem() {
|
||||||
|
info "Checking filesystem"
|
||||||
|
e2fsck -pf "$loopback"
|
||||||
|
(( $? < 4 )) && return
|
||||||
|
|
||||||
|
info "Filesystem error detected!"
|
||||||
|
|
||||||
|
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
|
fi
|
||||||
|
error $LINENO "Filesystem recoveries failed. Giving up..."
|
||||||
|
exit 9
|
||||||
|
|
||||||
#Gather info
|
}
|
||||||
beforesize=`ls -lah $img | cut -d ' ' -f 5`
|
|
||||||
partnum=`parted -m $img unit B print | tail -n 1 | cut -d ':' -f 1 | tr -d '\n'`
|
|
||||||
partstart=`parted -m $img unit B print | tail -n 1 | cut -d ':' -f 2 | tr -d 'B\n'`
|
|
||||||
loopback=`losetup -f --show -o $partstart $img`
|
|
||||||
currentsize=`tune2fs -l $loopback | grep 'Block count' | tr -d ' ' | cut -d ':' -f 2 | tr -d '\n'`
|
|
||||||
blocksize=`tune2fs -l $loopback | grep 'Block size' | tr -d ' ' | cut -d ':' -f 2 | tr -d '\n'`
|
|
||||||
|
|
||||||
#Check if we should make pi expand rootfs on next boot
|
function set_autoexpand() {
|
||||||
if [ "$should_skip_autoexpand" = false ]; then
|
|
||||||
#Make pi expand rootfs on next boot
|
#Make pi expand rootfs on next boot
|
||||||
mountdir=`mktemp -d`
|
mountdir=$(mktemp -d)
|
||||||
mount $loopback $mountdir
|
partprobe "$loopback"
|
||||||
|
sleep 3
|
||||||
|
umount "$loopback" > /dev/null 2>&1
|
||||||
|
mount "$loopback" "$mountdir" -o rw
|
||||||
|
if (( $? != 0 )); then
|
||||||
|
info "Unable to mount loopback, autoexpand will not be enabled"
|
||||||
|
return
|
||||||
|
fi
|
||||||
|
|
||||||
if [ `md5sum $mountdir/etc/rc.local | cut -d ' ' -f 1` != "c4eb22d9aa99915af319287d194d4529" ]; then
|
if [ ! -d "$mountdir/etc" ]; then
|
||||||
echo Creating new /etc/rc.local
|
info "/etc not found, autoexpand will not be enabled"
|
||||||
mv $mountdir/etc/rc.local $mountdir/etc/rc.local.bak
|
umount "$mountdir"
|
||||||
###Do not touch the following 6 lines including EOF###
|
return
|
||||||
cat <<\EOFE > $mountdir/etc/rc.local
|
fi
|
||||||
|
|
||||||
|
if [[ ! -f "$mountdir/etc/rc.local" ]]; then
|
||||||
|
info "An existing /etc/rc.local was not found, autoexpand may fail..."
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [[ -f "$mountdir/etc/rc.local" ]] && [[ "$(md5sum "$mountdir/etc/rc.local" | cut -d ' ' -f 1)" != "5c286b336c0606ed8e6f87708f7802eb" ]]; then
|
||||||
|
echo "Creating new /etc/rc.local"
|
||||||
|
if [ -f "$mountdir/etc/rc.local" ]; then
|
||||||
|
mv "$mountdir/etc/rc.local" "$mountdir/etc/rc.local.bak"
|
||||||
|
fi
|
||||||
|
|
||||||
|
#####Do not touch the following lines#####
|
||||||
|
cat <<\EOF1 > "$mountdir/etc/rc.local"
|
||||||
#!/bin/bash
|
#!/bin/bash
|
||||||
do_expand_rootfs() {
|
do_expand_rootfs() {
|
||||||
ROOT_PART=$(cat /proc/cmdline | tr -s ' ' '\n' | grep root= | cut -d '=' -f 2)
|
ROOT_PART=$(mount | sed -n 's|^/dev/\(.*\) on / .*|\1|p')
|
||||||
if [[ "$ROOT_PART" != /dev/mmcblk0* ]] ; then
|
|
||||||
echo $ROOT_PART is not an SD card...
|
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
|
return 0
|
||||||
fi
|
fi
|
||||||
PART_NUM=$(echo $ROOT_PART | cut -d 'p' -f 2)
|
|
||||||
|
|
||||||
# Get the starting offset of the root partition
|
# Get the starting offset of the root partition
|
||||||
PART_START=$(fdisk -l /dev/mmcblk0 | grep $ROOT_PART | tr -s ' ' | cut -d ' ' -f 2)
|
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
|
[ "$PART_START" ] || return 1
|
||||||
# Return value will likely be error for fdisk as it fails to reload the
|
# Return value will likely be error for fdisk as it fails to reload the
|
||||||
# partition table because the root fs is mounted
|
# partition table because the root fs is mounted
|
||||||
|
@ -96,89 +120,310 @@ p
|
||||||
$PART_NUM
|
$PART_NUM
|
||||||
$PART_START
|
$PART_START
|
||||||
|
|
||||||
|
p
|
||||||
w
|
w
|
||||||
|
|
||||||
EOF
|
EOF
|
||||||
|
|
||||||
cat <<EOF > /etc/rc.local &&
|
cat <<EOF > /etc/rc.local &&
|
||||||
#!/bin/sh
|
#!/bin/sh
|
||||||
resize2fs $ROOT_PART
|
echo "Expanding /dev/$ROOT_PART"
|
||||||
rm -f /etc/rc.local; cp -f /etc/rc.local.bak /etc/rc.local; /etc/rc.local
|
resize2fs /dev/$ROOT_PART
|
||||||
|
rm -f /etc/rc.local; cp -fp /etc/rc.local.bak /etc/rc.local && /etc/rc.local
|
||||||
|
|
||||||
EOF
|
EOF
|
||||||
reboot
|
reboot
|
||||||
exit
|
exit
|
||||||
}
|
}
|
||||||
do_expand_rootfs
|
raspi_config_expand() {
|
||||||
rm -f /etc/rc.local; cp -f /etc/rc.local.bak /etc/rc.local; /etc/rc.local
|
/usr/bin/env raspi-config --expand-rootfs
|
||||||
exit 0
|
if [[ $? != 0 ]]; then
|
||||||
EOFE
|
return -1
|
||||||
###End no touch zone###
|
|
||||||
chmod +x $mountdir/etc/rc.local
|
|
||||||
fi
|
|
||||||
umount $mountdir
|
|
||||||
else
|
else
|
||||||
echo Skipping autoexpanding process...
|
rm -f /etc/rc.local; cp -fp /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
|
||||||
|
if [[ -f /etc/rc.local.bak ]]; then
|
||||||
|
cp -fp /etc/rc.local.bak /etc/rc.local
|
||||||
|
/etc/rc.local
|
||||||
|
fi
|
||||||
|
exit 0
|
||||||
|
EOF1
|
||||||
|
#####End no touch zone#####
|
||||||
|
chmod +x "$mountdir/etc/rc.local"
|
||||||
|
fi
|
||||||
|
umount "$mountdir"
|
||||||
|
}
|
||||||
|
|
||||||
|
help() {
|
||||||
|
local help
|
||||||
|
read -r -d '' help << EOM
|
||||||
|
Usage: $0 [-adhrsvzZ] imagefile.img [newimagefile.img]
|
||||||
|
|
||||||
|
-s Don't expand filesystem when image is booted the first time
|
||||||
|
-v Be verbose
|
||||||
|
-r Use advanced filesystem repair option if the normal one fails
|
||||||
|
-z Compress image after shrinking with gzip
|
||||||
|
-Z Compress image after shrinking with xz
|
||||||
|
-a Compress image in parallel using multiple cores
|
||||||
|
-d Write debug messages in a debug log file
|
||||||
|
EOM
|
||||||
|
echo "$help"
|
||||||
|
exit 1
|
||||||
|
}
|
||||||
|
|
||||||
|
should_skip_autoexpand=false
|
||||||
|
debug=false
|
||||||
|
repair=false
|
||||||
|
parallel=false
|
||||||
|
verbose=false
|
||||||
|
ziptool=""
|
||||||
|
|
||||||
|
while getopts ":adhrsvzZ" opt; do
|
||||||
|
case "${opt}" in
|
||||||
|
a) parallel=true;;
|
||||||
|
d) debug=true;;
|
||||||
|
h) help;;
|
||||||
|
r) repair=true;;
|
||||||
|
s) should_skip_autoexpand=true ;;
|
||||||
|
v) verbose=true;;
|
||||||
|
z) ziptool="gzip";;
|
||||||
|
Z) ziptool="xz";;
|
||||||
|
*) help;;
|
||||||
|
esac
|
||||||
|
done
|
||||||
|
shift $((OPTIND-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
|
||||||
|
help
|
||||||
|
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
|
||||||
|
|
||||||
|
# set locale to POSIX(English) temporarily
|
||||||
|
# these locale settings only affect the script and its sub processes
|
||||||
|
|
||||||
|
export LANGUAGE=POSIX
|
||||||
|
export LC_ALL=POSIX
|
||||||
|
export LANG=POSIX
|
||||||
|
|
||||||
|
|
||||||
|
# check selected compression tool is supported and installed
|
||||||
|
if [[ -n $ziptool ]]; then
|
||||||
|
if [[ ! " ${ZIPTOOLS[@]} " =~ $ziptool ]]; then
|
||||||
|
error $LINENO "$ziptool is an unsupported ziptool."
|
||||||
|
exit 17
|
||||||
|
else
|
||||||
|
if [[ $parallel == true && $ziptool == "gzip" ]]; then
|
||||||
|
REQUIRED_TOOLS="$REQUIRED_TOOLS pigz"
|
||||||
|
else
|
||||||
|
REQUIRED_TOOLS="$REQUIRED_TOOLS $ziptool"
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
|
||||||
|
#Check that what we need is installed
|
||||||
|
for command in $REQUIRED_TOOLS; 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
|
||||||
|
f="$2"
|
||||||
|
if [[ -n $ziptool && "${f##*.}" == "${ZIPEXTENSIONS[$ziptool]}" ]]; then # remove zip extension if zip requested because zip tool will complain about extension
|
||||||
|
f="${f%.*}"
|
||||||
|
fi
|
||||||
|
info "Copying $1 to $f..."
|
||||||
|
cp --reflink=auto --sparse=always "$1" "$f"
|
||||||
|
if (( $? != 0 )); then
|
||||||
|
error $LINENO "Could not copy file..."
|
||||||
|
exit 5
|
||||||
|
fi
|
||||||
|
old_owner=$(stat -c %u:%g "$1")
|
||||||
|
chown "$old_owner" "$f"
|
||||||
|
img="$f"
|
||||||
|
fi
|
||||||
|
|
||||||
|
# cleanup at script exit
|
||||||
|
trap cleanup EXIT
|
||||||
|
|
||||||
|
#Gather info
|
||||||
|
info "Gathering data"
|
||||||
|
beforesize="$(ls -lh "$img" | cut -d ' ' -f 5)"
|
||||||
|
parted_output="$(parted -ms "$img" unit B print)"
|
||||||
|
rc=$?
|
||||||
|
if (( $rc )); then
|
||||||
|
error $LINENO "parted failed with rc $rc"
|
||||||
|
info "Possibly invalid image. Run 'parted $img unit B print' manually to investigate"
|
||||||
|
exit 6
|
||||||
|
fi
|
||||||
|
partnum="$(echo "$parted_output" | tail -n 1 | cut -d ':' -f 1)"
|
||||||
|
partstart="$(echo "$parted_output" | tail -n 1 | cut -d ':' -f 2 | tr -d 'B')"
|
||||||
|
if [ -z "$(parted -s "$img" unit B print | grep "$partstart" | grep logical)" ]; then
|
||||||
|
parttype="primary"
|
||||||
|
else
|
||||||
|
parttype="logical"
|
||||||
|
fi
|
||||||
|
loopback="$(losetup -f --show -o "$partstart" "$img")"
|
||||||
|
tune2fs_output="$(tune2fs -l "$loopback")"
|
||||||
|
rc=$?
|
||||||
|
if (( $rc )); then
|
||||||
|
echo "$tune2fs_output"
|
||||||
|
error $LINENO "tune2fs failed. Unable to shrink this type of image"
|
||||||
|
exit 7
|
||||||
|
fi
|
||||||
|
|
||||||
|
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)"
|
||||||
|
|
||||||
|
logVariables $LINENO beforesize parted_output partnum partstart parttype tune2fs_output currentsize blocksize
|
||||||
|
|
||||||
|
#Check if we should make pi expand rootfs on next boot
|
||||||
|
if [ "$parttype" == "logical" ]; then
|
||||||
|
echo "WARNING: PiShrink does not yet support autoexpanding of this type of image"
|
||||||
|
elif [ "$should_skip_autoexpand" = false ]; then
|
||||||
|
set_autoexpand
|
||||||
|
else
|
||||||
|
echo "Skipping autoexpanding process..."
|
||||||
fi
|
fi
|
||||||
|
|
||||||
#Make sure filesystem is ok
|
#Make sure filesystem is ok
|
||||||
e2fsck -f $loopback
|
checkFilesystem
|
||||||
minsize=`resize2fs -P $loopback | cut -d ':' -f 2 | tr -d ' ' | tr -d '\n'`
|
|
||||||
|
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 currentsize 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
|
||||||
if [[ `expr $currentsize - $minsize - 5000` -gt 0 ]]; then
|
extra_space=$(($currentsize - $minsize))
|
||||||
minsize=`expr $minsize + 5000 | tr -d '\n'`
|
logVariables $LINENO extra_space
|
||||||
elif [[ `expr $currentsize - $minsize - 1000` -gt 0 ]]; then
|
for space in 5000 1000 100; do
|
||||||
minsize=`expr $minsize + 1000 | tr -d '\n'`
|
if [[ $extra_space -gt $space ]]; then
|
||||||
elif [[ `expr $currentsize - $minsize - 100` -gt 0 ]]; then
|
minsize=$(($minsize + $space))
|
||||||
minsize=`expr $minsize + 100 | tr -d '\n'`
|
break
|
||||||
fi
|
fi
|
||||||
|
done
|
||||||
|
logVariables $LINENO minsize
|
||||||
|
|
||||||
#Shrink filesystem
|
#Shrink filesystem
|
||||||
resize2fs -p $loopback $minsize
|
info "Shrinking filesystem"
|
||||||
if [[ $? != 0 ]]; then
|
resize2fs -p "$loopback" $minsize
|
||||||
echo ERROR: resize2fs failed...
|
rc=$?
|
||||||
mount $loopback $mountdir
|
if (( $rc )); then
|
||||||
mv $mountdir/etc/rc.local.bak $mountdir/etc/rc.local
|
error $LINENO "resize2fs failed with rc $rc"
|
||||||
umount $mountdir
|
mount "$loopback" "$mountdir"
|
||||||
losetup -d $loopback
|
mv "$mountdir/etc/rc.local.bak" "$mountdir/etc/rc.local"
|
||||||
exit $rc
|
umount "$mountdir"
|
||||||
|
losetup -d "$loopback"
|
||||||
|
exit 12
|
||||||
fi
|
fi
|
||||||
sleep 1
|
sleep 1
|
||||||
|
|
||||||
#Shrink partition
|
#Shrink partition
|
||||||
losetup -d $loopback
|
partnewsize=$(($minsize * $blocksize))
|
||||||
partnewsize=`expr $minsize \* $blocksize | tr -d '\n'`
|
newpartend=$(($partstart + $partnewsize))
|
||||||
newpartend=`expr $partstart + $partnewsize | tr -d '\n'`
|
logVariables $LINENO partnewsize newpartend
|
||||||
part1=`parted $img rm $partnum`
|
parted -s -a minimal "$img" rm "$partnum"
|
||||||
part2=`parted $img unit B mkpart primary $partstart $newpartend`
|
rc=$?
|
||||||
|
if (( $rc )); then
|
||||||
|
error $LINENO "parted failed with rc $rc"
|
||||||
|
exit 13
|
||||||
|
fi
|
||||||
|
|
||||||
|
parted -s "$img" unit B mkpart "$parttype" "$partstart" "$newpartend"
|
||||||
|
rc=$?
|
||||||
|
if (( $rc )); then
|
||||||
|
error $LINENO "parted failed with rc $rc"
|
||||||
|
exit 14
|
||||||
|
fi
|
||||||
|
|
||||||
#Truncate the file
|
#Truncate the file
|
||||||
if [[ -f $img ]]; then
|
info "Shrinking image"
|
||||||
endresult=`parted -m $img unit B print free | tail -1 | cut -d ':' -f 2 | tr -d 'B\n'`
|
endresult=$(parted -ms "$img" unit B print free)
|
||||||
truncate -s $endresult $img
|
rc=$?
|
||||||
aftersize=`ls -lah $img | cut -d ' ' -f 5`
|
if (( $rc )); then
|
||||||
echo "Shrunk $img from $beforesize to $aftersize"
|
error $LINENO "parted failed with rc $rc"
|
||||||
|
exit 15
|
||||||
fi
|
fi
|
||||||
|
|
||||||
#Copy to new file if requested
|
endresult=$(tail -1 <<< "$endresult" | cut -d ':' -f 2 | tr -d 'B')
|
||||||
if [ -n "$2" -a "$force_inplace_and_copy_after" = true ]; then
|
logVariables $LINENO endresult
|
||||||
echo "Copying $1 to $2..."
|
truncate -s "$endresult" "$img"
|
||||||
if [[ -f $img ]]; then
|
rc=$?
|
||||||
cp --reflink=auto --sparse=always "$1" "$2"
|
if (( $rc )); then
|
||||||
else
|
error $LINENO "trunate failed with rc $rc"
|
||||||
imgsize=`parted -m $img unit B print | tail -1 | cut -d ':' -f 3 | tr -d 'B\n'`
|
exit 16
|
||||||
imgsize=`expr $imgsize + 1`
|
fi
|
||||||
dd if="$1" of="$2" conv=sparse count=$imgsize iflag=count_bytes bs=1M
|
|
||||||
|
# handle compression
|
||||||
|
if [[ -n $ziptool ]]; then
|
||||||
|
options=""
|
||||||
|
envVarname="${MYNAME^^}_${ziptool^^}" # PISHRINK_GZIP or PISHRINK_XZ environment variables allow to override all options for gzip or xz
|
||||||
|
[[ $parallel == true ]] && options="${ZIP_PARALLEL_OPTIONS[$ziptool]}"
|
||||||
|
[[ -v $envVarname ]] && options="${!envVarname}" # if environment variable defined use these options
|
||||||
|
[[ $verbose == true ]] && options="$options -v" # add verbose flag if requested
|
||||||
|
|
||||||
|
if [[ $parallel == true ]]; then
|
||||||
|
parallel_tool="${ZIP_PARALLEL_TOOL[$ziptool]}"
|
||||||
|
info "Using $parallel_tool on the shrunk image"
|
||||||
|
if ! $parallel_tool ${options} "$img"; then
|
||||||
|
rc=$?
|
||||||
|
error $LINENO "$parallel_tool failed with rc $rc"
|
||||||
|
exit 18
|
||||||
fi
|
fi
|
||||||
|
|
||||||
if (( $? != 0 )); then
|
else # sequential
|
||||||
echo "ERROR: Could not copy file..."
|
info "Using $ziptool on the shrunk image"
|
||||||
exit -5
|
if ! $ziptool ${options} "$img"; then
|
||||||
|
rc=$?
|
||||||
|
error $LINENO "$ziptool failed with rc $rc"
|
||||||
|
exit 19
|
||||||
fi
|
fi
|
||||||
img=$2
|
fi
|
||||||
|
img=$img.${ZIPEXTENSIONS[$ziptool]}
|
||||||
fi
|
fi
|
||||||
|
|
||||||
|
aftersize=$(ls -lh "$img" | cut -d ' ' -f 5)
|
||||||
|
logVariables $LINENO aftersize
|
||||||
|
|
||||||
|
info "Shrunk $img from $beforesize to $aftersize"
|
||||||
|
|
Loading…
Reference in New Issue