IMG/cpio/ventoy/init_chain

   1 #!/ventoy/busybox/sh
   2 #************************************************************************************
   3 # Copyright (c) 2020, longpanda <admin@ventoy.net>
   4 #
   5 # This program is free software; you can redistribute it and/or
   6 # modify it under the terms of the GNU General Public License as
   7 # published by the Free Software Foundation; either version 3 of the
   8 # License, or (at your option) any later version.
   9 #
  10 # This program is distributed in the hope that it will be useful, but
  11 # WITHOUT ANY WARRANTY; without even the implied warranty of
  12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13 # General Public License for more details.
  14 #
  15 # You should have received a copy of the GNU General Public License
  16 # along with this program; if not, see <http://www.gnu.org/licenses/>.
  17 #
  18 #************************************************************************************
  19
  20
  21 ####################################################################
  22 #                                                                  #
  23 # Step 1 : extract real initramfs to /                             #
  24 #                                                                  #
  25 ####################################################################
  26 cd /
  27 rm -rf /init /linuxrc /dev/  /root
  28
  29 vtSbinFileNum=$(ls -1 /sbin | wc -l)
  30 if [ $vtSbinFileNum -eq 1 ]; then
  31     echo "remove whole sbin directory" >> $VTLOG
  32     rm -rf /sbin
  33 else
  34     echo "remove only sbin/init file" >> $VTLOG
  35     ls -l /sbin >> $VTLOG
  36     rm -f /sbin/init
  37 fi
  38
  39 ventoy_is_initrd_ramdisk() {
  40     #As I known, PCLinuxOS/smgl use ramdisk
  41     if echo $vtkerver | grep -i -q 'PCLinuxOS'; then
  42         true
  43     elif echo $vtkerver | grep -i -q 'SMGL-'; then
  44         true
  45     else
  46         false
  47     fi
  48 }
  49
  50 ventoy_mount_squashfs() {
  51     mkdir /dev
  52     mount -t devtmpfs devtmpfs /dev
  53     dd if=$1 of=/dev/ram0 status=none
  54     umount /dev && rm -rf /dev
  55 }
  56
  57 # param: file skip magic tmp
  58 ventoy_unpack_initramfs() {
  59     vtfile=$1; vtskip=$2; vtmagic=$3; vttmp=$4
  60     echo "=====ventoy_unpack_initramfs: #$*#" >> $VTLOG
  61
  62     #special process
  63     #if [ "${vtmagic:0:4}" = '5678' ]; then
  64     #    echo -en '\x1F\x8B' | dd status=none of=$vtfile bs=1 count=2 conv=notrunc
  65     #    vtmagic='1F8B'
  66     #fi
  67
  68     if [ "${vtmagic:0:4}" = '6873' ]; then
  69         ventoy_mount_squashfs $vtfile
  70         return
  71     fi
  72
  73     for vtx in '1F8B zcat' '1F9E zcat' '425A bzcat' '5D00 lzcat' 'FD37 xzcat' '894C lzopcat' '0221 lz4cat' '28B5 zstdcat' '3037 cat' '4C5A lunzip -c'; do
  74         if [ "${vtx:0:4}" = "${vtmagic:0:4}" ]; then
  75             echo "vtx=$vtx" >> $VTLOG
  76             if [ $vtskip -eq 0 ]; then
  77                 if [ "${vtx:5}" = "xzcat" ]; then
  78                     rm -f $VTOY_PATH/xzlog
  79                     ${vtx:5} $vtfile 2> $VTOY_PATH/xzlog | (cpio -idmu 2>>$VTLOG; cat > $vttmp)
  80                     if grep -q 'corrupted data' $VTOY_PATH/xzlog; then
  81                         echo 'xzcat failed, now try xzminidec...' >> $VTLOG
  82                         rm -f $VTOY_PATH/xzlog
  83                         cat $vtfile | xzminidec 2> $VTOY_PATH/xzlog | (cpio -idmu 2>>$VTLOG; cat > $vttmp)
  84
  85                         if grep -q 'limit' $VTOY_PATH/xzlog; then
  86                             echo 'xzminidec failed, now try xzcat_musl ...' >> $VTLOG
  87                             xzcat_musl $vtfile | (cpio -idmu 2>>$VTLOG; cat > $vttmp)
  88                         fi
  89                     fi
  90                 else
  91                     ${vtx:5} $vtfile | (cpio -idmu 2>>$VTLOG; cat > $vttmp)
  92                 fi
  93             else
  94                 dd if=$vtfile skip=$vtskip iflag=skip_bytes status=none | ${vtx:5} | (cpio -idmu 2>>$VTLOG; cat > $vttmp)
  95             fi
  96             break
  97         fi
  98     done
  99 }
 100
 101 # param: file magic tmp
 102 ventoy_unpack_initrd() {
 103     vtfile=$1; vtmagic=$2; vttmp=$3
 104     echo "=====ventoy_unpack_initrd: #$*#" >> $VTLOG
 105
 106     for vtx in '1F8B zcat' '1F9E zcat' '425A bzcat' '5D00 lzcat' 'FD37 xzcat' '894C lzopcat' '0221 lz4cat' '28B5 zstdcat' '3037 cat'; do
 107         if [ "${vtx:0:4}" = "${vtmagic:0:4}" ]; then
 108             echo "vtx=$vtx" >> $VTLOG
 109             ${vtx:5} $vtfile > $vttmp
 110             break
 111         fi
 112     done
 113 }
 114
 115
 116 # This export is for busybox cpio command
 117 export EXTRACT_UNSAFE_SYMLINKS=1
 118
 119 for vtfile in $(ls /initrd*); do
 120     #decompress first initrd
 121     vtmagic=$(hexdump -n 2 -e '2/1 "%02X"' $vtfile)
 122
 123     if ventoy_is_initrd_ramdisk; then
 124         ventoy_unpack_initrd $vtfile $vtmagic ${vtfile}_tmp
 125         mv ${vtfile}_tmp $vtfile
 126         break
 127     else
 128         ventoy_unpack_initramfs $vtfile 0 $vtmagic ${vtfile}_tmp
 129     fi
 130
 131     #only for cpio,cpio,...,initrd sequence, initrd,cpio or initrd,initrd sequence is not supported
 132     while [ -e ${vtfile}_tmp ] && [ $(stat -c '%s' ${vtfile}_tmp) -gt 512 ]; do
 133         mv ${vtfile}_tmp $vtfile
 134         vtdump=$(hexdump -n 512 -e '512/1 "%02X"' $vtfile)
 135         vtmagic=$(echo $vtdump | sed 's/^\(00\)*//')
 136         let vtoffset="(${#vtdump}-${#vtmagic})/2"
 137
 138         if [ -z "$vtmagic" ]; then
 139             echo "terminate with all zero data file" >> $VTLOG
 140             break
 141         fi
 142
 143         ventoy_unpack_initramfs $vtfile $vtoffset ${vtmagic:0:4} ${vtfile}_tmp
 144     done
 145
 146     rm -f $vtfile ${vtfile}_tmp
 147 done
 148
 149
 150 #break here for debug
 151 if [ "$VTOY_BREAK_LEVEL" = "02" ] || [ "$VTOY_BREAK_LEVEL" = "12" ]; then
 152     sleep 5
 153     echo -e "\n\n\033[32m ################################################# \033[0m"
 154     echo -e "\033[32m ################ VENTOY DEBUG ################### \033[0m"
 155     echo -e "\033[32m ################################################# \033[0m \n"
 156     if [ "$VTOY_BREAK_LEVEL" = "12" ]; then
 157         cat $VTOY_PATH/log
 158     fi
 159     exec $BUSYBOX_PATH/sh
 160 fi
 161
 162
 163 ####################################################################
 164 #                                                                  #
 165 # Step 3 : Extract injection archive                               #
 166 #                                                                  #
 167 ####################################################################
 168 ventoy_unpack_injection() {
 169     vtmagic=$(hexdump -n 2 -e '2/1 "%02X"' $VTOY_PATH/ventoy_injection)
 170     echo "ventoy_unpack_injection  vtmagic=$vtmagic ..."
 171
 172     if [ "1F8B" = "$vtmagic" ] || [ "1F9E" = "$vtmagic" ]; then
 173         echo "tar.gz  tar -xzvf"
 174         tar -xzvf $VTOY_PATH/ventoy_injection -C /
 175     elif [ "425A" = "$vtmagic" ]; then
 176         echo "tar.bz2  tar -xjvf"
 177         tar -xjvf $VTOY_PATH/ventoy_injection -C /
 178     elif [ "FD37" = "$vtmagic" ]; then
 179         echo "tar.xz  tar -xJvf"
 180         tar -xJvf $VTOY_PATH/ventoy_injection -C /
 181     elif [ "5D00" = "$vtmagic" ]; then
 182         echo "tar.lzma  tar -xavf"
 183         tar -xavf $VTOY_PATH/ventoy_injection -C /
 184     else
 185         echo "unzip -o"
 186         unzip -o $VTOY_PATH/ventoy_injection -d /
 187     fi
 188 }
 189
 190 if [ -e $VTOY_PATH/ventoy_injection ]; then
 191     echo "### decompress injection ... ###" >>$VTLOG
 192     ventoy_unpack_injection > $VTOY_PATH/injection.log 2>&1
 193 fi
 194
 195
 196 ####################################################################
 197 #                                                                  #
 198 # Step 4 : Hand over to ventoy_chain.sh                                  #
 199 #                                                                  #
 200 ####################################################################
 201 echo "Now hand over to ventoy.sh" >>$VTLOG
 202 . $VTOY_PATH/tool/vtoytool_install.sh
 203
 204 export PATH=$VTOY_ORG_PATH
 205 exec $BUSYBOX_PATH/sh $VTOY_PATH/ventoy_chain.sh