IMG/cpio/ventoy/init

   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 : parse kernel debug parameter                            #
  24 #                                                                  #
  25 ####################################################################
  26 [ -d /proc ] || mkdir /proc; mount -t proc proc /proc
  27 vtcmdline=$(cat /proc/cmdline)
  28 vtkerver=$(cat /proc/version)
  29 umount /proc; rm -rf /proc
  30
  31 echo "kenel version=$vtkerver" >>$VTLOG
  32 echo "kenel cmdline=$vtcmdline" >>$VTLOG
  33
  34 #break here for debug
  35 if [ "$VTOY_BREAK_LEVEL" = "01" ] || [ "$VTOY_BREAK_LEVEL" = "11" ]; then
  36     sleep 5
  37     echo -e "\n\n\033[32m ################################################# \033[0m"
  38     echo -e "\033[32m ################ VENTOY DEBUG ################### \033[0m"
  39     echo -e "\033[32m ################################################# \033[0m \n"
  40
  41     if [ "$VTOY_BREAK_LEVEL" = "11" ]; then
  42         cat $VTLOG
  43     fi
  44     exec $BUSYBOX_PATH/sh
  45 fi
  46
  47
  48 ####################################################################
  49 #                                                                  #
  50 # Step 2 : extract real initramfs to /                             #
  51 #                                                                  #
  52 ####################################################################
  53 cd /
  54 rm -rf /init /linuxrc /dev/  /root
  55
  56 vtSbinFileNum=$(ls -1 /sbin | wc -l)
  57 if [ $vtSbinFileNum -eq 1 ]; then
  58     echo "remove whole sbin directory" >> $VTLOG
  59     rm -rf /sbin
  60 else
  61     echo "remove only sbin/init file" >> $VTLOG
  62     ls -l /sbin >> $VTLOG
  63     rm -f /sbin/init
  64 fi
  65
  66 ventoy_is_initrd_ramdisk() {
  67     #As I known, PCLinuxOS use ramdisk
  68     if echo $vtkerver | grep -i -q 'PCLinuxOS'; then
  69         true
  70     else
  71         false
  72     fi
  73 }
  74
  75 # param: file skip magic tmp
  76 ventoy_unpack_initramfs() {
  77     vtfile=$1; vtskip=$2; vtmagic=$3; vttmp=$4
  78     echo "=====ventoy_unpack_initramfs: #$*#" >> $VTLOG
  79
  80     #special process
  81     #if [ "${vtmagic:0:4}" = '5678' ]; then
  82     #    echo -en '\x1F\x8B' | dd status=none of=$vtfile bs=1 count=2 conv=notrunc
  83     #    vtmagic='1F8B'
  84     #fi
  85
  86     for vtx in '1F8B zcat' '1F9E zcat' '425A bzcat' '5D00 lzcat' 'FD37 xzcat' '894C lzopcat' '0221 lz4cat' '28B5 zstdcat' '3037 cat'; do
  87         if [ "${vtx:0:4}" = "${vtmagic:0:4}" ]; then
  88             echo "vtx=$vtx" >> $VTLOG
  89             if [ $vtskip -eq 0 ]; then
  90                 if [ "${vtx:5}" = "xzcat" ]; then
  91                     rm -f $VTOY_PATH/xzlog
  92                     ${vtx:5} $vtfile 2> $VTOY_PATH/xzlog | (cpio -idmu 2>>$VTLOG; cat > $vttmp)
  93
  94                     if grep -q 'corrupted data' $VTOY_PATH/xzlog; then
  95                         echo 'xzcat failed, now try xzminidec...' >> $VTLOG
  96                         cat $vtfile | xzminidec | (cpio -idmu 2>>$VTLOG; cat > $vttmp)
  97                     fi
  98
  99                 else
 100                     ${vtx:5} $vtfile | (cpio -idmu 2>>$VTLOG; cat > $vttmp)
 101                 fi
 102             else
 103                 dd if=$vtfile skip=$vtskip iflag=skip_bytes status=none | ${vtx:5} | (cpio -idmu 2>>$VTLOG; cat > $vttmp)
 104             fi
 105             break
 106         fi
 107     done
 108 }
 109
 110 # param: file magic tmp
 111 ventoy_unpack_initrd() {
 112     vtfile=$1; vtmagic=$2; vttmp=$3
 113     echo "=====ventoy_unpack_initrd: #$*#" >> $VTLOG
 114
 115     for vtx in '1F8B zcat' '1F9E zcat' '425A bzcat' '5D00 lzcat' 'FD37 xzcat' '894C lzopcat' '0221 lz4cat' '28B5 zstdcat' '3037 cat'; do
 116         if [ "${vtx:0:4}" = "${vtmagic:0:4}" ]; then
 117             echo "vtx=$vtx" >> $VTLOG
 118             ${vtx:5} $vtfile > $vttmp
 119             break
 120         fi
 121     done
 122 }
 123
 124
 125 # This export is for busybox cpio command
 126 export EXTRACT_UNSAFE_SYMLINKS=1
 127
 128 for vtfile in $(ls /initrd*); do
 129     #decompress first initrd
 130     vtmagic=$(hexdump -n 2 -e '2/1 "%02X"' $vtfile)
 131
 132     if ventoy_is_initrd_ramdisk; then
 133         ventoy_unpack_initrd $vtfile $vtmagic ${vtfile}_tmp
 134         mv ${vtfile}_tmp $vtfile
 135         break
 136     else
 137         ventoy_unpack_initramfs $vtfile 0 $vtmagic ${vtfile}_tmp
 138     fi
 139
 140     #only for cpio,cpio,...,initrd sequence, initrd,cpio or initrd,initrd sequence is not supported
 141     while [ -e ${vtfile}_tmp ] && [ $(stat -c '%s' ${vtfile}_tmp) -gt 512 ]; do
 142         mv ${vtfile}_tmp $vtfile
 143         vtdump=$(hexdump -n 512 -e '512/1 "%02X"' $vtfile)
 144         vtmagic=$(echo $vtdump | sed 's/^\(00\)*//')
 145         let vtoffset="(${#vtdump}-${#vtmagic})/2"
 146
 147         if [ -z "$vtmagic" ]; then
 148             echo "terminate with all zero data file" >> $VTLOG
 149             break
 150         fi
 151
 152         ventoy_unpack_initramfs $vtfile $vtoffset ${vtmagic:0:4} ${vtfile}_tmp
 153     done
 154
 155     rm -f $vtfile ${vtfile}_tmp
 156 done
 157
 158
 159 #break here for debug
 160 if [ "$VTOY_BREAK_LEVEL" = "02" ] || [ "$VTOY_BREAK_LEVEL" = "12" ]; then
 161     sleep 5
 162     echo -e "\n\n\033[32m ################################################# \033[0m"
 163     echo -e "\033[32m ################ VENTOY DEBUG ################### \033[0m"
 164     echo -e "\033[32m ################################################# \033[0m \n"
 165     if [ "$VTOY_BREAK_LEVEL" = "12" ]; then
 166         cat $VTOY_PATH/log
 167     fi
 168     exec $BUSYBOX_PATH/sh
 169 fi
 170
 171
 172 ####################################################################
 173 #                                                                  #
 174 # Step 3 : Hand over to ventoy.sh                                  #
 175 #                                                                  #
 176 ####################################################################
 177 echo "Now hand over to ventoy.sh" >>$VTLOG
 178 . $VTOY_PATH/tool/vtoytool_install.sh
 179
 180 export PATH=$VTOY_ORG_PATH
 181 exec $BUSYBOX_PATH/sh $VTOY_PATH/ventoy.sh