IMG/cpio/ventoy/ventoy.sh

   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 # Step 1 : Parse kernel parameter                                  #
  23 #                                                                  #
  24 ####################################################################
  25 if ! [ -e /proc ]; then
  26     $BUSYBOX_PATH/mkdir /proc
  27     rmproc='Y'
  28 fi
  29 $BUSYBOX_PATH/mount -t proc proc /proc
  30
  31 # vtinit=xxx to replace rdinit=xxx
  32 vtcmdline=$($CAT /proc/cmdline)
  33 for i in $vtcmdline; do
  34     if echo $i | $GREP -q vtinit; then
  35         user_rdinit=${i#vtinit=}
  36         echo "user set user_rdinit=${user_rdinit}" >>$VTLOG
  37     fi
  38 done
  39
  40
  41 ####################################################################
  42 #                                                                  #
  43 # Step 2 : Do OS specific hook                                     #
  44 #                                                                  #
  45 ####################################################################
  46 ventoy_get_os_type() {
  47     echo "kernel version" >> $VTLOG
  48     $CAT /proc/version >> $VTLOG
  49
  50     # rhel5/CentOS5 and all other distributions based on them
  51     if $GREP -q 'el5' /proc/version; then
  52         echo 'rhel5'; return
  53
  54     # rhel6/CentOS6 and all other distributions based on them
  55     elif $GREP -q 'el6' /proc/version; then
  56         echo 'rhel6'; return
  57
  58     # rhel7/CentOS7/rhel8/CentOS8 and all other distributions based on them
  59     elif $GREP -q 'el[78]' /proc/version; then
  60         echo 'rhel7'; return
  61
  62     # Maybe rhel9 rhel1x use the same way? Who knows!
  63     elif $EGREP -q 'el9|el1[0-9]' /proc/version; then
  64         echo 'rhel7'; return
  65
  66     # Fedora : do the same process with rhel7
  67     elif $GREP -q '\.fc[0-9][0-9]\.' /proc/version; then
  68         echo 'rhel7'; return
  69
  70     # Debian :
  71     elif $GREP -q '[Dd]ebian' /proc/version; then
  72         echo 'debian'; return
  73
  74     # Ubuntu : do the same process with debian
  75     elif $GREP -q '[Uu]buntu' /proc/version; then
  76         echo 'debian'; return
  77
  78     # Deepin : do the same process with debian
  79     elif $GREP -q '[Dd]eepin' /proc/version; then
  80         echo 'debian'; return
  81
  82     # SUSE
  83     elif $GREP -q 'SUSE' /proc/version; then
  84         echo 'suse'; return
  85
  86     # ArchLinux
  87     elif $EGREP -q 'archlinux|ARCH' /proc/version; then
  88         echo 'arch'; return
  89
  90     # gentoo
  91     elif $EGREP -q '[Gg]entoo' /proc/version; then
  92         echo 'gentoo'; return
  93
  94     # TinyCore
  95     elif $EGREP -q 'tinycore' /proc/version; then
  96         echo 'tinycore'; return
  97
  98     # manjaro
  99     elif $EGREP -q 'manjaro|MANJARO' /proc/version; then
 100         echo 'manjaro'; return
 101
 102     # mageia
 103     elif $EGREP -q 'mageia' /proc/version; then
 104         echo 'mageia'; return
 105
 106     # pclinux OS
 107     elif $GREP -i -q 'PCLinuxOS' /proc/version; then
 108         echo 'pclos'; return
 109
 110     # KaOS
 111     elif $GREP -i -q 'kaos' /proc/version; then
 112         echo 'kaos'; return
 113
 114     # Alpine
 115     elif $GREP -q 'Alpine' /proc/version; then
 116         echo 'alpine'; return
 117
 118     # NixOS
 119     elif $GREP -i -q 'NixOS' /proc/version; then
 120         echo 'nixos'; return
 121
 122     fi
 123
 124     if [ -e /lib/debian-installer ]; then
 125         echo 'debian'; return
 126     fi
 127
 128     if [ -e /etc/os-release ]; then
 129         if $GREP -q 'XenServer' /etc/os-release; then
 130             echo 'xen'; return
 131         elif $GREP -q 'SUSE ' /etc/os-release; then
 132             echo 'suse'; return
 133         fi
 134     fi
 135
 136     if $BUSYBOX_PATH/dmesg | $GREP -q -m1 "Xen:"; then
 137         echo 'xen'; return
 138     fi
 139
 140
 141     if [ -e /etc/HOSTNAME ] && $GREP -i -q 'slackware' /etc/HOSTNAME; then
 142         echo 'slackware'; return
 143     fi
 144
 145     if [ -e /init ]; then
 146         if $GREP -i -q zeroshell /init; then
 147             echo 'zeroshell'; return
 148         fi
 149     fi
 150
 151
 152     echo "default"
 153 }
 154
 155 VTOS=$(ventoy_get_os_type)
 156 echo "OS=###${VTOS}###" >>$VTLOG
 157 if [ -e "$VTOY_PATH/hook/$VTOS/ventoy-hook.sh" ]; then
 158     $BUSYBOX_PATH/sh "$VTOY_PATH/hook/$VTOS/ventoy-hook.sh"
 159 fi
 160
 161
 162 ####################################################################
 163 #                                                                  #
 164 # Step 3 : Check for debug break                                   #
 165 #                                                                  #
 166 ####################################################################
 167 if [ "$VTOY_BREAK_LEVEL" = "03" ] || [ "$VTOY_BREAK_LEVEL" = "13" ]; then
 168     $SLEEP 5
 169     echo -e "\n\n\033[32m ################################################# \033[0m"
 170     echo -e "\033[32m ################ VENTOY DEBUG ################### \033[0m"
 171     echo -e "\033[32m ################################################# \033[0m \n"
 172     if [ "$VTOY_BREAK_LEVEL" = "13" ]; then
 173         $CAT $VTOY_PATH/log
 174     fi
 175     exec $BUSYBOX_PATH/sh
 176 fi
 177
 178
 179
 180 ####################################################################
 181 #                                                                  #
 182 # Step 4 : Hand over to real init                                  #
 183 #                                                                  #
 184 ####################################################################
 185 $BUSYBOX_PATH/umount /proc
 186 if [ "$rmproc" = "Y" ]; then
 187     $BUSYBOX_PATH/rm -rf /proc
 188 fi
 189
 190 cd /
 191 unset VTOY_PATH VTLOG FIND GREP EGREP CAT AWK SED SLEEP HEAD
 192
 193 for vtinit in $user_rdinit /init /sbin/init /linuxrc; do
 194     if [ -d /ventoy_rdroot ]; then
 195         if [ -e "/ventoy_rdroot$vtinit" ]; then
 196             # switch_root will check /init file, this is a cheat code
 197             echo 'switch_root' > /init
 198             exec $BUSYBOX_PATH/switch_root /ventoy_rdroot "$vtinit"
 199         fi
 200     else
 201         if [ -e "$vtinit" ];then
 202             exec "$vtinit"
 203         fi
 204     fi
 205 done
 206
 207 # Should never reach here
 208 echo -e "\n\n\033[31m ############ INIT NOT FOUND ############### \033[0m \n"
 209 exec $BUSYBOX_PATH/sh