INSTALL/tool/VentoyWorker.sh

   1 #!/bin/sh
   2
   3 . ./tool/ventoy_lib.sh
   4
   5 print_usage() {
   6     echo 'Usage:  Ventoy2Disk.sh CMD [ OPTION ] /dev/sdX'
   7     echo '  CMD:'
   8     echo '   -i  install ventoy to sdX (fail if disk already installed with ventoy)'
   9     echo '   -u  update ventoy in sdX'
  10     echo '   -I  force install ventoy to sdX (no matter installed or not)'
  11     echo ''
  12     echo '  OPTION: (optional)'
  13     echo '   -r SIZE_MB  preserve some space at the bottom of the disk (only for install)'
  14     echo '   -s          enable secure boot support (default is disabled)'
  15     echo ''
  16
  17 }
  18
  19 RESERVE_SIZE_MB=0
  20 while [ -n "$1" ]; do
  21     if [ "$1" = "-i" ]; then
  22         MODE="install"
  23     elif [ "$1" = "-I" ]; then
  24         MODE="install"
  25         FORCE="Y"
  26     elif [ "$1" = "-u" ]; then
  27         MODE="update"
  28     elif [ "$1" = "-s" ]; then
  29         SECUREBOOT="YES"
  30     elif [ "$1" = "-r" ]; then
  31         RESERVE_SPACE="YES"
  32         shift
  33         RESERVE_SIZE_MB=$1
  34     else
  35         if ! [ -b "$1" ]; then
  36             vterr "$1 is NOT a valid device"
  37             print_usage
  38             exit 1
  39         fi
  40         DISK=$1
  41     fi
  42
  43     shift
  44 done
  45
  46 if [ -z "$MODE" ]; then
  47     print_usage
  48     exit 1
  49 fi
  50
  51 if ! [ -b "$DISK" ]; then
  52     vterr "Disk $DISK does not exist"
  53     exit 1
  54 fi
  55
  56 if [ -e /sys/class/block/${DISK#/dev/}/start ]; then
  57     vterr "$DISK is a partition, please use the whole disk"
  58     exit 1
  59 fi
  60
  61 if [ -n "$RESERVE_SPACE" ]; then
  62     if echo $RESERVE_SIZE_MB | grep -q '^[0-9][0-9]*$'; then
  63         vtdebug "User will reserve $RESERVE_SIZE_MB MB disk space"
  64     else
  65         vterr "$RESERVE_SIZE_MB is invalid for reserved space"
  66         exit 1
  67     fi
  68 fi
  69
  70 if dd if="$DISK" of=/dev/null bs=1 count=1 >/dev/null 2>&1; then
  71     vtdebug "root permission check ok ..."
  72 else
  73     vterr "Failed to access $DISK, maybe root privilege is needed!"
  74     echo ''
  75     exit 1
  76 fi
  77
  78 vtdebug "MODE=$MODE FORCE=$FORCE RESERVE_SPACE=$RESERVE_SPACE RESERVE_SIZE_MB=$RESERVE_SIZE_MB"
  79
  80 if ! check_tool_work_ok; then
  81     vterr "Some tools can not run in current system. Please check log.txt for detail."
  82     exit 1
  83 fi
  84
  85 grep "^$DISK" /proc/mounts | while read mtline; do
  86     mtpnt=$(echo $mtline | awk '{print $2}')
  87     vtdebug "Trying to umount $mtpnt ..."
  88     umount $mtpnt >/dev/null 2>&1
  89 done
  90
  91 if swapon -s | grep -q "^${DISK}[0-9]"; then
  92     swapon -s | grep "^${DISK}[0-9]" | awk '{print $1}' | while read line; do
  93         vtdebug "Trying to swapoff $line ..."
  94         swapoff $line
  95     done
  96 fi
  97
  98
  99 if grep "$DISK" /proc/mounts; then
 100     vterr "$DISK is already mounted, please umount it first!"
 101     exit 1
 102 fi
 103
 104 if swapon -s | grep -q "^${DISK}[0-9]"; then
 105     vterr "$DISK is used as swap, please swapoff it first!"
 106     exit 1
 107 fi
 108
 109
 110 if [ "$MODE" = "install" ]; then
 111     vtdebug "install ventoy ..."
 112
 113     if parted -v > /dev/null 2>&1; then
 114         PARTTOOL='parted'
 115     elif fdisk -v >/dev/null 2>&1; then
 116         PARTTOOL='fdisk'
 117     else
 118         vterr "Both parted and fdisk are not found in the sysstem, Ventoy can't create new partition."
 119         exit 1
 120     fi
 121
 122     version=$(get_disk_ventoy_version $DISK)
 123     if [ $? -eq 0 ]; then
 124         if [ -z "$FORCE" ]; then
 125             vtwarn "$DISK already contains a Ventoy with version $version"
 126             vtwarn "Use -u option to do a safe upgrade operation."
 127             vtwarn "OR if you really want to reinstall ventoy to $DISK, please use -I option."
 128             vtwarn ""
 129             exit 1
 130         fi
 131     fi
 132
 133     disk_sector_num=$(cat /sys/block/${DISK#/dev/}/size)
 134     disk_size_gb=$(expr $disk_sector_num / 2097152)
 135
 136     if [ $disk_sector_num -gt 4294967296 ]; then
 137         vterr "$DISK is over 2TB size, MBR will not work on it."
 138         exit 1
 139     fi
 140
 141     if [ -n "$RESERVE_SPACE" ]; then
 142         sum_size_mb=$(expr $RESERVE_SIZE_MB + $VENTOY_PART_SIZE_MB)
 143         reserve_sector_num=$(expr $sum_size_mb \* 2048)
 144
 145         if [ $disk_sector_num -le $reserve_sector_num ]; then
 146             vterr "Can't reserve $RESERVE_SIZE_MB MB space from $DISK"
 147             exit 1
 148         fi
 149     fi
 150
 151
 152     #Print disk info
 153     echo "Disk : $DISK"
 154     parted -s $DISK p 2>&1 | grep Model
 155     echo "Size : $disk_size_gb GB"
 156     echo ''
 157
 158     if [ -n "$RESERVE_SPACE" ]; then
 159         echo "You will reserve $RESERVE_SIZE_MB MB disk space "
 160     fi
 161     echo ''
 162
 163     vtwarn "Attention:"
 164     vtwarn "You will install Ventoy to $DISK."
 165     vtwarn "All the data on the disk $DISK will be lost!!!"
 166     echo ""
 167
 168     read -p 'Continue? (y/n)'  Answer
 169     if [ "$Answer" != "y" ]; then
 170         if [ "$Answer" != "Y" ]; then
 171             exit 0
 172         fi
 173     fi
 174
 175     echo ""
 176     vtwarn "All the data on the disk $DISK will be lost!!!"
 177     read -p 'Double-check. Continue? (y/n)'  Answer
 178     if [ "$Answer" != "y" ]; then
 179         if [ "$Answer" != "Y" ]; then
 180             exit 0
 181         fi
 182     fi
 183
 184
 185     if [ $disk_sector_num -le $VENTOY_SECTOR_NUM ]; then
 186         vterr "No enough space in disk $DISK"
 187         exit 1
 188     fi
 189
 190     if ! dd if=/dev/zero of=$DISK bs=1 count=512 status=none conv=fsync; then
 191         vterr "Write data to $DISK failed, please check whether it's in use."
 192         exit 1
 193     fi
 194
 195     format_ventoy_disk $RESERVE_SIZE_MB $DISK $PARTTOOL
 196
 197     # format part1
 198     if ventoy_is_linux64; then
 199         cmd=./tool/mkexfatfs_64
 200     else
 201         cmd=./tool/mkexfatfs_32
 202     fi
 203
 204     chmod +x ./tool/*
 205
 206     # DiskSize > 32GB  Cluster Size use 128KB
 207     # DiskSize < 32GB  Cluster Size use 32KB
 208     if [ $disk_size_gb -gt 32 ]; then
 209         cluster_sectors=256
 210     else
 211         cluster_sectors=64
 212     fi
 213
 214     $cmd -n ventoy -s $cluster_sectors ${DISK}1
 215
 216     chmod +x ./tool/vtoy_gen_uuid
 217
 218     vtinfo "writing data to disk ..."
 219     dd status=none conv=fsync if=./boot/boot.img of=$DISK bs=1 count=446
 220     ./tool/xzcat ./boot/core.img.xz | dd status=none conv=fsync of=$DISK bs=512 count=2047 seek=1
 221     ./tool/xzcat ./ventoy/ventoy.disk.img.xz | dd status=none conv=fsync of=$DISK bs=512 count=$VENTOY_SECTOR_NUM seek=$part2_start_sector
 222
 223     #disk uuid
 224     ./tool/vtoy_gen_uuid | dd status=none conv=fsync of=${DISK} seek=384 bs=1 count=16
 225
 226     #disk signature
 227     ./tool/vtoy_gen_uuid | dd status=none conv=fsync of=${DISK} skip=12 seek=440 bs=1 count=4
 228
 229     vtinfo "sync data ..."
 230     sync
 231
 232     vtinfo "esp partition processing ..."
 233
 234     sleep 1
 235     mtpnt=$(grep "^${DISK}2" /proc/mounts | awk '{print $2}')
 236     if [ -n "$mtpnt" ]; then
 237         umount $mtpnt >/dev/null 2>&1
 238     fi
 239
 240     if [ "$SECUREBOOT" != "YES" ]; then
 241         mkdir ./tmp_mnt
 242
 243         vtdebug "mounting part2 ...."
 244         for tt in 1 2 3; do
 245             if mount ${DISK}2 ./tmp_mnt; then
 246                 vtdebug "mounting part2 success"
 247                 break
 248             fi
 249
 250             mtpnt=$(grep "^${DISK}2" /proc/mounts | awk '{print $2}')
 251             if [ -n "$mtpnt" ]; then
 252                 umount $mtpnt >/dev/null 2>&1
 253             fi
 254             sleep 2
 255         done
 256
 257         rm -f ./tmp_mnt/EFI/BOOT/BOOTX64.EFI
 258         rm -f ./tmp_mnt/EFI/BOOT/grubx64.efi
 259         rm -f ./tmp_mnt/EFI/BOOT/MokManager.efi
 260         rm -f ./tmp_mnt/ENROLL_THIS_KEY_IN_MOKMANAGER.cer
 261         mv ./tmp_mnt/EFI/BOOT/grubx64_real.efi  ./tmp_mnt/EFI/BOOT/BOOTX64.EFI
 262
 263         umount ./tmp_mnt
 264         rm -rf ./tmp_mnt
 265     fi
 266
 267     echo ""
 268     vtinfo "Install Ventoy to $DISK successfully finished."
 269     echo ""
 270
 271 else
 272     vtdebug "update ventoy ..."
 273
 274     oldver=$(get_disk_ventoy_version $DISK)
 275     if [ $? -ne 0 ]; then
 276         vtwarn "$DISK does not contain ventoy or data corupted"
 277         echo ""
 278         vtwarn "Please use -i option if you want to install ventoy to $DISK"
 279         echo ""
 280         exit 1
 281     fi
 282
 283     curver=$(cat ./ventoy/version)
 284
 285     vtinfo "Upgrade operation is safe, all the data in the 1st partition (iso files and other) will be unchanged!"
 286     echo ""
 287
 288     read -p "Update Ventoy  $oldver ===> $curver   Continue? (y/n)"  Answer
 289     if [ "$Answer" != "y" ]; then
 290         if [ "$Answer" != "Y" ]; then
 291             exit 0
 292         fi
 293     fi
 294
 295     PART2=$(get_disk_part_name $DISK 2)
 296     SHORT_PART2=${PART2#/dev/}
 297     part2_start=$(cat /sys/class/block/$SHORT_PART2/start)
 298
 299     dd status=none conv=fsync if=./boot/boot.img of=$DISK bs=1 count=440
 300
 301     PART1_ACTIVE=$(dd if=$DISK bs=1 count=1 skip=446 status=none | ./tool/hexdump -n1 -e  '1/1 "%02X"')
 302     PART2_ACTIVE=$(dd if=$DISK bs=1 count=1 skip=462 status=none | ./tool/hexdump -n1 -e  '1/1 "%02X"')
 303
 304     vtdebug "PART1_ACTIVE=$PART1_ACTIVE  PART2_ACTIVE=$PART2_ACTIVE"
 305     if [ "$PART1_ACTIVE" = "00" ] && [ "$PART2_ACTIVE" = "80" ]; then
 306         vtdebug "change 1st partition active, 2nd partition inactive ..."
 307         echo -en '\x80' | dd of=$DISK conv=fsync bs=1 count=1 seek=446 status=none
 308         echo -en '\x00' | dd of=$DISK conv=fsync bs=1 count=1 seek=462 status=none
 309     fi
 310
 311     ./tool/xzcat ./boot/core.img.xz | dd status=none conv=fsync of=$DISK bs=512 count=2047 seek=1
 312     ./tool/xzcat ./ventoy/ventoy.disk.img.xz | dd status=none conv=fsync of=$DISK bs=512 count=$VENTOY_SECTOR_NUM seek=$part2_start
 313
 314     sync
 315
 316     if [ "$SECUREBOOT" != "YES" ]; then
 317         mkdir ./tmp_mnt
 318
 319         vtdebug "mounting part2 ...."
 320         for tt in 1 2 3; do
 321             if mount ${DISK}2 ./tmp_mnt; then
 322                 vtdebug "mounting part2 success"
 323                 break
 324             fi
 325             sleep 2
 326         done
 327
 328         rm -f ./tmp_mnt/EFI/BOOT/BOOTX64.EFI
 329         rm -f ./tmp_mnt/EFI/BOOT/grubx64.efi
 330         rm -f ./tmp_mnt/EFI/BOOT/MokManager.efi
 331         rm -f ./tmp_mnt/ENROLL_THIS_KEY_IN_MOKMANAGER.cer
 332         mv ./tmp_mnt/EFI/BOOT/grubx64_real.efi  ./tmp_mnt/EFI/BOOT/BOOTX64.EFI
 333
 334         umount ./tmp_mnt
 335         rm -rf ./tmp_mnt
 336     fi
 337
 338     echo ""
 339     vtinfo "Update Ventoy to $DISK successfully finished."
 340     echo ""
 341
 342 fi
 343
 344