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

Added options to change compression level for fast, default and best compression levels and added the option to compress in extreme mode for xz.

This commit is contained in:
nhkhai 2020-08-09 16:05:02 +08:00
parent 526ca5ae89
commit 4e90956eb1
2 changed files with 61 additions and 41 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
README.md
View File

@ -2,8 +2,7 @@
# 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. 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.
In addition the shrinked image can be compressed with gzip and xz to create an even smaller image. Parallel compression of the image In addition the shrinked image can be compressed with gzip and xz to create an even smaller image. Parallel compression of the image using multiple cores is supported.
using multiple cores is supported.
## Usage ## ## Usage ##
@ -13,8 +12,11 @@ Usage: $0 [-adhrspvzZ] imagefile.img [newimagefile.img]
-s Don't expand filesystem when image is booted the first time -s Don't expand filesystem when image is booted the first time
-v Be verbose -v Be verbose
-r Use advanced filesystem repair option if the normal one fails -r Use advanced filesystem repair option if the normal one fails
-z Compress image after shrinking with gzip -z Compress image after shrinking with gzip (Default is balanced (6) compression)
-Z Compress image after shrinking with xz -Z Compress image after shrinking with xz (Default is balanced (6) compression)
-c Change the compression level from the default, balanced (6) to fast (1 for gzip, 0 or xz)
-C Change the compression level from the default, balanced (6) to best (9)
-e Compress image in extreme mode (Only applicable to xz compression)
-a Compress image in parallel using multiple cores -a Compress image in parallel using multiple cores
-p Remove logs, apt archives, dhcp leases and ssh hostkeys -p Remove logs, apt archives, dhcp leases and ssh hostkeys
-d Write debug messages in a debug log file -d Write debug messages in a debug log file
@ -22,12 +24,15 @@ Usage: $0 [-adhrspvzZ] imagefile.img [newimagefile.img]
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. 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 expantion on the images next boot * `-s` Prevents automatic filesystem expantion on the images next boot.
* `-v` enables more verbose output * `-v` Enables more verbose output.
* `-r` will attempt to repair the filesystem using aditional options if the normal repair fails * `-r` Will attempt to repair the filesystem using aditional 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 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. * `-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. * `-c Will use option -1 for pigz and option -0 for xz.
* `-C Will use option -9.
* `-e Will use option -e for xz.
* `-a` will use option -f for pigz and option -T0 for xz and compress in parallel.
* `-d` will create a logfile `pishrink.log` which may help for problem analysis. * `-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. Default options for compressors can be overwritten by defining PISHRINK_GZIP or PSHRINK_XZ environment variables for gzip and xz.

41
pishrink.sh
View File

@ -8,9 +8,9 @@ MYNAME="${SCRIPTNAME%.*}"
LOGFILE="${CURRENT_DIR}/${SCRIPTNAME%.*}.log" LOGFILE="${CURRENT_DIR}/${SCRIPTNAME%.*}.log"
REQUIRED_TOOLS="parted losetup tune2fs md5sum e2fsck resize2fs" REQUIRED_TOOLS="parted losetup tune2fs md5sum e2fsck resize2fs"
ZIPTOOLS=("gzip xz") ZIPTOOLS=("gzip xz")
declare -A ZIP_PARALLEL_TOOL=( [gzip]="pigz" [xz]="xz" ) # parallel zip tool to use in parallel mode 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 ZIP_PARALLEL_OPTIONS=( [gzip]="-f" [xz]="-T0" ) # Options for zip tools in parallel mode
declare -A ZIPEXTENSIONS=( [gzip]="gz" [xz]="xz" ) # extensions of zipped files declare -A ZIPEXTENSIONS=( [gzip]="gz" [xz]="xz" ) # Extensions of zipped files
function info() { function info() {
echo "$SCRIPTNAME: $1 ..." echo "$SCRIPTNAME: $1 ..."
@ -160,9 +160,12 @@ Usage: $0 [-adhrspvzZ] imagefile.img [newimagefile.img]
-s Don't expand filesystem when image is booted the first time -s Don't expand filesystem when image is booted the first time
-v Be verbose -v Be verbose
-r Use advanced filesystem repair option if the normal one fails -r Use advanced filesystem repair option if the normal one fails
-z Compress image after shrinking with gzip -z Compress image after shrinking with gzip (Default is balanced (6) compression)
-Z Compress image after shrinking with xz -Z Compress image after shrinking with xz (Default is balanced (6) compression)
-a Compress image in parallel using multiple cores -a Compress image in parallel using multiple cores
-c Change the compression level from the default, balanced (6) to fast (1 for gzip, 0 or xz)
-C Change the compression level from the default, balanced (6) to best (9)
-e Compress image in extreme mode (Only applicable to xz compression)
-p Remove logs, apt archives, dhcp leases and ssh hostkeys -p Remove logs, apt archives, dhcp leases and ssh hostkeys
-d Write debug messages in a debug log file -d Write debug messages in a debug log file
EOM EOM
@ -171,17 +174,22 @@ EOM
} }
should_skip_autoexpand=false should_skip_autoexpand=false
compression=6
debug=false debug=false
extreme=false
repair=false repair=false
parallel=false parallel=false
verbose=false verbose=false
prep=false prep=false
ziptool="" ziptool=""
while getopts ":adhprsvzZ" opt; do while getopts ":acCdehprsvzZ" opt; do
case "${opt}" in case "${opt}" in
a) parallel=true;; a) parallel=true;;
c) compression=0;;
C) compression=9;;
d) debug=true;; d) debug=true;;
e) extreme=true;;
h) help;; h) help;;
p) prep=true;; p) prep=true;;
r) repair=true;; r) repair=true;;
@ -194,6 +202,10 @@ while getopts ":adhprsvzZ" opt; do
done done
shift $((OPTIND-1)) shift $((OPTIND-1))
if [ $compression == 0 ] && [[ "$ziptool" == "gzip" ]]; then
compression=1
fi
if [ "$debug" = true ]; then if [ "$debug" = true ]; then
info "Creating log file $LOGFILE" info "Creating log file $LOGFILE"
rm "$LOGFILE" &>/dev/null rm "$LOGFILE" &>/dev/null
@ -221,7 +233,7 @@ if (( EUID != 0 )); then
exit 3 exit 3
fi fi
# check selected compression tool is supported and installed # Check selected compression tool is supported and installed
if [[ -n $ziptool ]]; then if [[ -n $ziptool ]]; then
if [[ ! " ${ZIPTOOLS[@]} " =~ $ziptool ]]; then if [[ ! " ${ZIPTOOLS[@]} " =~ $ziptool ]]; then
error $LINENO "$ziptool is an unsupported ziptool." error $LINENO "$ziptool is an unsupported ziptool."
@ -247,7 +259,7 @@ done
# Copy to new file if requested # Copy to new file if requested
if [ -n "$2" ]; then if [ -n "$2" ]; then
f="$2" f="$2"
if [[ -n $ziptool && "${f##*.}" == "${ZIPEXTENSIONS[$ziptool]}" ]]; then # remove zip extension if zip requested because zip tool will complain about extension if [[ -n $ziptool && "${f##*.}" == "${ZIPEXTENSIONS[$ziptool]}" ]]; then # Remove zip extension if zip requested because zip tool will complain about extension
f="${f%.*}" f="${f%.*}"
fi fi
info "Copying $1 to $f..." info "Copying $1 to $f..."
@ -261,7 +273,7 @@ if [ -n "$2" ]; then
img="$f" img="$f"
fi fi
# cleanup at script exit # Cleanup at script exit
trap cleanup EXIT trap cleanup EXIT
# Gather info # Gather info
@ -389,13 +401,15 @@ if (( $rc )); then
exit 16 exit 16
fi fi
# handle compression # Handle the compression
if [[ -n $ziptool ]]; then if [[ -n $ziptool ]]; then
options="" options=""
envVarname="${MYNAME^^}_${ziptool^^}" # PISHRINK_GZIP or PISHRINK_XZ environment variables allow to override all options for gzip or xz 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]}" [[ $parallel == true ]] && options="${ZIP_PARALLEL_OPTIONS[$ziptool]}"
[[ -v $envVarname ]] && options="${!envVarname}" # if environment variable defined use these options [[ $compression != 6 ]] && options="$options -$compression" # Update the compression value if requested (default value is 6)
[[ $verbose == true ]] && options="$options -v" # add verbose flag if requested [[ "$ziptool" == "xz" ]] && [[ $extreme == true ]] && options="$options -e" # Add extreme flag if requested for xz
[[ -v $envVarname ]] && options="${!envVarname}" # If environment variables defined use these options
[[ $verbose == true ]] && options="$options -v" # Add verbose flag if requested
if [[ $parallel == true ]]; then if [[ $parallel == true ]]; then
parallel_tool="${ZIP_PARALLEL_TOOL[$ziptool]}" parallel_tool="${ZIP_PARALLEL_TOOL[$ziptool]}"
@ -406,7 +420,7 @@ if [[ -n $ziptool ]]; then
exit 18 exit 18
fi fi
else # sequential else # Sequential
info "Using $ziptool on the shrunk image" info "Using $ziptool on the shrunk image"
if ! $ziptool ${options} "$img"; then if ! $ziptool ${options} "$img"; then
rc=$? rc=$?
@ -414,6 +428,7 @@ if [[ -n $ziptool ]]; then
exit 19 exit 19
fi fi
fi fi
img=$img.${ZIPEXTENSIONS[$ziptool]} img=$img.${ZIPEXTENSIONS[$ziptool]}
fi fi