mirror of
https://github.com/ventoy/Ventoy.git
synced 2024-12-27 15:45:58 -05:00
217 lines
7.7 KiB
Bash
217 lines
7.7 KiB
Bash
#!/ventoy/busybox/sh
|
|
#************************************************************************************
|
|
# Copyright (c) 2020, longpanda <admin@ventoy.net>
|
|
#
|
|
# This program is free software; you can redistribute it and/or
|
|
# modify it under the terms of the GNU General Public License as
|
|
# published by the Free Software Foundation; either version 3 of the
|
|
# License, or (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful, but
|
|
# WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
# General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program; if not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
#************************************************************************************
|
|
|
|
|
|
###################################################################
|
|
# #
|
|
# Step 1 : parse kernel debug parameter #
|
|
# #
|
|
####################################################################
|
|
[ -d /proc ] || mkdir /proc; mount -t proc proc /proc
|
|
vtcmdline=$(cat /proc/cmdline)
|
|
vtkerver=$(cat /proc/version)
|
|
umount /proc; rm -rf /proc
|
|
|
|
echo "kenel version=$vtkerver" >>$VTLOG
|
|
echo "kenel cmdline=$vtcmdline" >>$VTLOG
|
|
|
|
#break here for debug
|
|
if [ "$VTOY_BREAK_LEVEL" = "01" ] || [ "$VTOY_BREAK_LEVEL" = "11" ]; then
|
|
sleep 5
|
|
echo -e "\n\n\033[32m ################################################# \033[0m"
|
|
echo -e "\033[32m ################ VENTOY DEBUG ################### \033[0m"
|
|
echo -e "\033[32m ################################################# \033[0m \n"
|
|
|
|
if [ "$VTOY_BREAK_LEVEL" = "11" ]; then
|
|
cat $VTLOG
|
|
fi
|
|
exec $BUSYBOX_PATH/sh
|
|
fi
|
|
|
|
|
|
####################################################################
|
|
# #
|
|
# Step 2 : extract real initramfs to / #
|
|
# #
|
|
####################################################################
|
|
cd /
|
|
rm -rf /init /linuxrc /dev/ /root
|
|
|
|
vtSbinFileNum=$(ls -1 /sbin | wc -l)
|
|
if [ $vtSbinFileNum -eq 1 ]; then
|
|
echo "remove whole sbin directory" >> $VTLOG
|
|
rm -rf /sbin
|
|
else
|
|
echo "remove only sbin/init file" >> $VTLOG
|
|
ls -l /sbin >> $VTLOG
|
|
rm -f /sbin/init
|
|
fi
|
|
|
|
ventoy_is_initrd_ramdisk() {
|
|
#As I known, PCLinuxOS use ramdisk
|
|
if echo $vtkerver | grep -i -q 'PCLinuxOS'; then
|
|
true
|
|
else
|
|
false
|
|
fi
|
|
}
|
|
|
|
ventoy_mount_squashfs() {
|
|
mkdir /dev
|
|
mount -t devtmpfs devtmpfs /dev
|
|
dd if=$1 of=/dev/ram0 status=none
|
|
umount /dev && rm -rf /dev
|
|
}
|
|
|
|
# param: file skip magic tmp
|
|
ventoy_unpack_initramfs() {
|
|
vtfile=$1; vtskip=$2; vtmagic=$3; vttmp=$4
|
|
echo "=====ventoy_unpack_initramfs: #$*#" >> $VTLOG
|
|
|
|
#special process
|
|
#if [ "${vtmagic:0:4}" = '5678' ]; then
|
|
# echo -en '\x1F\x8B' | dd status=none of=$vtfile bs=1 count=2 conv=notrunc
|
|
# vtmagic='1F8B'
|
|
#fi
|
|
|
|
if [ "${vtmagic:0:4}" = '6873' ]; then
|
|
ventoy_mount_squashfs $vtfile
|
|
return
|
|
fi
|
|
|
|
for vtx in '1F8B zcat' '1F9E zcat' '425A bzcat' '5D00 lzcat' 'FD37 xzcat' '894C lzopcat' '0221 lz4cat' '28B5 zstdcat' '3037 cat'; do
|
|
if [ "${vtx:0:4}" = "${vtmagic:0:4}" ]; then
|
|
echo "vtx=$vtx" >> $VTLOG
|
|
if [ $vtskip -eq 0 ]; then
|
|
if [ "${vtx:5}" = "xzcat" ]; then
|
|
rm -f $VTOY_PATH/xzlog
|
|
${vtx:5} $vtfile 2> $VTOY_PATH/xzlog | (cpio -idmu 2>>$VTLOG; cat > $vttmp)
|
|
|
|
if grep -q 'corrupted data' $VTOY_PATH/xzlog; then
|
|
echo 'xzcat failed, now try xzminidec...' >> $VTLOG
|
|
cat $vtfile | xzminidec | (cpio -idmu 2>>$VTLOG; cat > $vttmp)
|
|
fi
|
|
|
|
else
|
|
${vtx:5} $vtfile | (cpio -idmu 2>>$VTLOG; cat > $vttmp)
|
|
fi
|
|
else
|
|
dd if=$vtfile skip=$vtskip iflag=skip_bytes status=none | ${vtx:5} | (cpio -idmu 2>>$VTLOG; cat > $vttmp)
|
|
fi
|
|
break
|
|
fi
|
|
done
|
|
}
|
|
|
|
# param: file magic tmp
|
|
ventoy_unpack_initrd() {
|
|
vtfile=$1; vtmagic=$2; vttmp=$3
|
|
echo "=====ventoy_unpack_initrd: #$*#" >> $VTLOG
|
|
|
|
for vtx in '1F8B zcat' '1F9E zcat' '425A bzcat' '5D00 lzcat' 'FD37 xzcat' '894C lzopcat' '0221 lz4cat' '28B5 zstdcat' '3037 cat'; do
|
|
if [ "${vtx:0:4}" = "${vtmagic:0:4}" ]; then
|
|
echo "vtx=$vtx" >> $VTLOG
|
|
${vtx:5} $vtfile > $vttmp
|
|
break
|
|
fi
|
|
done
|
|
}
|
|
|
|
|
|
# This export is for busybox cpio command
|
|
export EXTRACT_UNSAFE_SYMLINKS=1
|
|
|
|
for vtfile in $(ls /initrd*); do
|
|
#decompress first initrd
|
|
vtmagic=$(hexdump -n 2 -e '2/1 "%02X"' $vtfile)
|
|
|
|
if ventoy_is_initrd_ramdisk; then
|
|
ventoy_unpack_initrd $vtfile $vtmagic ${vtfile}_tmp
|
|
mv ${vtfile}_tmp $vtfile
|
|
break
|
|
else
|
|
ventoy_unpack_initramfs $vtfile 0 $vtmagic ${vtfile}_tmp
|
|
fi
|
|
|
|
#only for cpio,cpio,...,initrd sequence, initrd,cpio or initrd,initrd sequence is not supported
|
|
while [ -e ${vtfile}_tmp ] && [ $(stat -c '%s' ${vtfile}_tmp) -gt 512 ]; do
|
|
mv ${vtfile}_tmp $vtfile
|
|
vtdump=$(hexdump -n 512 -e '512/1 "%02X"' $vtfile)
|
|
vtmagic=$(echo $vtdump | sed 's/^\(00\)*//')
|
|
let vtoffset="(${#vtdump}-${#vtmagic})/2"
|
|
|
|
if [ -z "$vtmagic" ]; then
|
|
echo "terminate with all zero data file" >> $VTLOG
|
|
break
|
|
fi
|
|
|
|
ventoy_unpack_initramfs $vtfile $vtoffset ${vtmagic:0:4} ${vtfile}_tmp
|
|
done
|
|
|
|
rm -f $vtfile ${vtfile}_tmp
|
|
done
|
|
|
|
|
|
#break here for debug
|
|
if [ "$VTOY_BREAK_LEVEL" = "02" ] || [ "$VTOY_BREAK_LEVEL" = "12" ]; then
|
|
sleep 5
|
|
echo -e "\n\n\033[32m ################################################# \033[0m"
|
|
echo -e "\033[32m ################ VENTOY DEBUG ################### \033[0m"
|
|
echo -e "\033[32m ################################################# \033[0m \n"
|
|
if [ "$VTOY_BREAK_LEVEL" = "12" ]; then
|
|
cat $VTOY_PATH/log
|
|
fi
|
|
exec $BUSYBOX_PATH/sh
|
|
fi
|
|
|
|
|
|
####################################################################
|
|
# #
|
|
# Step 3 : Extract injection archive #
|
|
# #
|
|
####################################################################
|
|
if [ -e $VTOY_PATH/ventoy_injection ]; then
|
|
echo "decompress injection ..." >>$VTLOG
|
|
|
|
vtmagic=$(hexdump -n 2 -e '2/1 "%02X"' $VTOY_PATH/ventoy_injection)
|
|
echo "vtmagic=$vtmagic ..." >>$VTLOG
|
|
|
|
if [ "1F8B" = "vtmagic" ] || [ "1F9E" = "vtmagic" ]; then
|
|
zcat $VTOY_PATH/ventoy_injection | tar -xf -C /
|
|
elif [ "425A" = "vtmagic" ]; then
|
|
bzcat $VTOY_PATH/ventoy_injection | tar -xf -C /
|
|
elif [ "FD37" = "vtmagic" ]; then
|
|
xzcat $VTOY_PATH/ventoy_injection | tar -xf -C /
|
|
else
|
|
unzip -o -q $VTOY_PATH/ventoy_injection -d /
|
|
fi
|
|
fi
|
|
|
|
|
|
####################################################################
|
|
# #
|
|
# Step 4 : Hand over to ventoy.sh #
|
|
# #
|
|
####################################################################
|
|
echo "Now hand over to ventoy.sh" >>$VTLOG
|
|
. $VTOY_PATH/tool/vtoytool_install.sh
|
|
|
|
export PATH=$VTOY_ORG_PATH
|
|
exec $BUSYBOX_PATH/sh $VTOY_PATH/ventoy.sh
|