mirror of https://github.com/ventoy/Ventoy
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
384 lines
11 KiB
Bash
384 lines
11 KiB
Bash
#!/bin/sh
|
|
|
|
. ./tool/ventoy_lib.sh
|
|
|
|
print_usage() {
|
|
echo 'Usage: Ventoy2Disk.sh CMD [ OPTION ] /dev/sdX'
|
|
echo ' CMD:'
|
|
echo ' -i install ventoy to sdX (fail if disk already installed with ventoy)'
|
|
echo ' -I force install ventoy to sdX (no matter installed or not)'
|
|
echo ' -u update ventoy in sdX'
|
|
echo ''
|
|
echo ' OPTION: (optional)'
|
|
echo ' -r SIZE_MB preserve some space at the bottom of the disk (only for install)'
|
|
echo ' -s enable secure boot support (default is disabled)'
|
|
echo ' -g use GPT partition style, default is MBR (only for install)'
|
|
echo ''
|
|
}
|
|
|
|
RESERVE_SIZE_MB=0
|
|
while [ -n "$1" ]; do
|
|
if [ "$1" = "-i" ]; then
|
|
MODE="install"
|
|
elif [ "$1" = "-I" ]; then
|
|
MODE="install"
|
|
FORCE="Y"
|
|
elif [ "$1" = "-u" ]; then
|
|
MODE="update"
|
|
elif [ "$1" = "-s" ]; then
|
|
SECUREBOOT="YES"
|
|
elif [ "$1" = "-g" ]; then
|
|
VTGPT="YES"
|
|
elif [ "$1" = "-r" ]; then
|
|
RESERVE_SPACE="YES"
|
|
shift
|
|
RESERVE_SIZE_MB=$1
|
|
else
|
|
if ! [ -b "$1" ]; then
|
|
vterr "$1 is NOT a valid device"
|
|
print_usage
|
|
exit 1
|
|
fi
|
|
DISK=$1
|
|
fi
|
|
|
|
shift
|
|
done
|
|
|
|
if [ -z "$MODE" ]; then
|
|
print_usage
|
|
exit 1
|
|
fi
|
|
|
|
if ! [ -b "$DISK" ]; then
|
|
vterr "Disk $DISK does not exist"
|
|
exit 1
|
|
fi
|
|
|
|
if [ -e /sys/class/block/${DISK#/dev/}/start ]; then
|
|
vterr "$DISK is a partition, please use the whole disk"
|
|
exit 1
|
|
fi
|
|
|
|
if [ -n "$RESERVE_SPACE" ]; then
|
|
if echo $RESERVE_SIZE_MB | grep -q '^[0-9][0-9]*$'; then
|
|
vtdebug "User will reserve $RESERVE_SIZE_MB MB disk space"
|
|
else
|
|
vterr "$RESERVE_SIZE_MB is invalid for reserved space"
|
|
exit 1
|
|
fi
|
|
fi
|
|
|
|
if dd if="$DISK" of=/dev/null bs=1 count=1 >/dev/null 2>&1; then
|
|
vtdebug "root permission check ok ..."
|
|
else
|
|
vterr "Failed to access $DISK, maybe root privilege is needed!"
|
|
echo ''
|
|
exit 1
|
|
fi
|
|
|
|
vtdebug "MODE=$MODE FORCE=$FORCE RESERVE_SPACE=$RESERVE_SPACE RESERVE_SIZE_MB=$RESERVE_SIZE_MB"
|
|
|
|
if ! check_tool_work_ok; then
|
|
vterr "Some tools can not run in current system. Please check log.txt for detail."
|
|
exit 1
|
|
fi
|
|
|
|
grep "^$DISK" /proc/mounts | while read mtline; do
|
|
mtpnt=$(echo $mtline | awk '{print $2}')
|
|
vtdebug "Trying to umount $mtpnt ..."
|
|
umount $mtpnt >/dev/null 2>&1
|
|
done
|
|
|
|
if swapon -s | grep -q "^${DISK}[0-9]"; then
|
|
swapon -s | grep "^${DISK}[0-9]" | awk '{print $1}' | while read line; do
|
|
vtdebug "Trying to swapoff $line ..."
|
|
swapoff $line
|
|
done
|
|
fi
|
|
|
|
|
|
if grep "$DISK" /proc/mounts; then
|
|
vterr "$DISK is already mounted, please umount it first!"
|
|
exit 1
|
|
fi
|
|
|
|
if swapon -s | grep -q "^${DISK}[0-9]"; then
|
|
vterr "$DISK is used as swap, please swapoff it first!"
|
|
exit 1
|
|
fi
|
|
|
|
|
|
if [ "$MODE" = "install" ]; then
|
|
vtdebug "install ventoy ..."
|
|
|
|
if [ -n "$VTGPT" ]; then
|
|
if parted -v > /dev/null 2>&1; then
|
|
PARTTOOL='parted'
|
|
else
|
|
vterr "parted is not found in the system, Ventoy can't create new partition."
|
|
exit 1
|
|
fi
|
|
else
|
|
if parted -v > /dev/null 2>&1; then
|
|
PARTTOOL='parted'
|
|
elif fdisk -v >/dev/null 2>&1; then
|
|
PARTTOOL='fdisk'
|
|
else
|
|
vterr "Both parted and fdisk are not found in the system, Ventoy can't create new partition."
|
|
exit 1
|
|
fi
|
|
fi
|
|
|
|
version=$(get_disk_ventoy_version $DISK)
|
|
if [ $? -eq 0 ]; then
|
|
if [ -z "$FORCE" ]; then
|
|
vtwarn "$DISK already contains a Ventoy with version $version"
|
|
vtwarn "Use -u option to do a safe upgrade operation."
|
|
vtwarn "OR if you really want to reinstall ventoy to $DISK, please use -I option."
|
|
vtwarn ""
|
|
exit 1
|
|
fi
|
|
fi
|
|
|
|
disk_sector_num=$(cat /sys/block/${DISK#/dev/}/size)
|
|
disk_size_gb=$(expr $disk_sector_num / 2097152)
|
|
|
|
if [ $disk_sector_num -gt 4294967296 ] && [ -z "$VTGPT" ]; then
|
|
vterr "$DISK is over 2TB size, MBR will not work on it."
|
|
exit 1
|
|
fi
|
|
|
|
if [ -n "$RESERVE_SPACE" ]; then
|
|
sum_size_mb=$(expr $RESERVE_SIZE_MB + $VENTOY_PART_SIZE_MB)
|
|
reserve_sector_num=$(expr $sum_size_mb \* 2048)
|
|
|
|
if [ $disk_sector_num -le $reserve_sector_num ]; then
|
|
vterr "Can't reserve $RESERVE_SIZE_MB MB space from $DISK"
|
|
exit 1
|
|
fi
|
|
fi
|
|
|
|
#Print disk info
|
|
echo "Disk : $DISK"
|
|
parted -s $DISK p 2>&1 | grep Model
|
|
echo "Size : $disk_size_gb GB"
|
|
if [ -n "$VTGPT" ]; then
|
|
echo "Style: GPT"
|
|
else
|
|
echo "Style: MBR"
|
|
fi
|
|
echo ''
|
|
|
|
if [ -n "$RESERVE_SPACE" ]; then
|
|
echo "You will reserve $RESERVE_SIZE_MB MB disk space "
|
|
fi
|
|
echo ''
|
|
|
|
vtwarn "Attention:"
|
|
vtwarn "You will install Ventoy to $DISK."
|
|
vtwarn "All the data on the disk $DISK will be lost!!!"
|
|
echo ""
|
|
|
|
read -p 'Continue? (y/n)' Answer
|
|
if [ "$Answer" != "y" ]; then
|
|
if [ "$Answer" != "Y" ]; then
|
|
exit 0
|
|
fi
|
|
fi
|
|
|
|
echo ""
|
|
vtwarn "All the data on the disk $DISK will be lost!!!"
|
|
read -p 'Double-check. Continue? (y/n)' Answer
|
|
if [ "$Answer" != "y" ]; then
|
|
if [ "$Answer" != "Y" ]; then
|
|
exit 0
|
|
fi
|
|
fi
|
|
|
|
|
|
if [ $disk_sector_num -le $VENTOY_SECTOR_NUM ]; then
|
|
vterr "No enough space in disk $DISK"
|
|
exit 1
|
|
fi
|
|
|
|
if ! dd if=/dev/zero of=$DISK bs=1 count=512 status=none conv=fsync; then
|
|
vterr "Write data to $DISK failed, please check whether it's in use."
|
|
exit 1
|
|
fi
|
|
|
|
if [ -n "$VTGPT" ]; then
|
|
vtdebug "format_ventoy_disk_gpt $RESERVE_SIZE_MB $DISK $PARTTOOL ..."
|
|
format_ventoy_disk_gpt $RESERVE_SIZE_MB $DISK $PARTTOOL
|
|
else
|
|
vtdebug "format_ventoy_disk_mbr $RESERVE_SIZE_MB $DISK $PARTTOOL ..."
|
|
format_ventoy_disk_mbr $RESERVE_SIZE_MB $DISK $PARTTOOL
|
|
fi
|
|
|
|
# format part1
|
|
if ventoy_is_linux64; then
|
|
cmd=./tool/mkexfatfs_64
|
|
else
|
|
cmd=./tool/mkexfatfs_32
|
|
fi
|
|
|
|
if [ -d ./tool/ ]; then
|
|
chmod +x -R ./tool/
|
|
fi
|
|
|
|
# DiskSize > 32GB Cluster Size use 128KB
|
|
# DiskSize < 32GB Cluster Size use 32KB
|
|
if [ $disk_size_gb -gt 32 ]; then
|
|
cluster_sectors=256
|
|
else
|
|
cluster_sectors=64
|
|
fi
|
|
|
|
$cmd -n ventoy -s $cluster_sectors ${DISK}1
|
|
|
|
vtinfo "writing data to disk ..."
|
|
|
|
dd status=none conv=fsync if=./boot/boot.img of=$DISK bs=1 count=446
|
|
|
|
if [ -n "$VTGPT" ]; then
|
|
echo -en '\x22' | dd status=none of=$DISK conv=fsync bs=1 count=1 seek=92
|
|
./tool/xzcat ./boot/core.img.xz | dd status=none conv=fsync of=$DISK bs=512 count=2014 seek=34
|
|
echo -en '\x23' | dd of=$DISK conv=fsync bs=1 count=1 seek=17908 status=none
|
|
else
|
|
./tool/xzcat ./boot/core.img.xz | dd status=none conv=fsync of=$DISK bs=512 count=2047 seek=1
|
|
fi
|
|
|
|
./tool/xzcat ./ventoy/ventoy.disk.img.xz | dd status=none conv=fsync of=$DISK bs=512 count=$VENTOY_SECTOR_NUM seek=$part2_start_sector
|
|
|
|
#disk uuid
|
|
./tool/vtoy_gen_uuid | dd status=none conv=fsync of=${DISK} seek=384 bs=1 count=16
|
|
|
|
#disk signature
|
|
./tool/vtoy_gen_uuid | dd status=none conv=fsync of=${DISK} skip=12 seek=440 bs=1 count=4
|
|
|
|
vtinfo "sync data ..."
|
|
sync
|
|
|
|
vtinfo "esp partition processing ..."
|
|
|
|
sleep 1
|
|
mtpnt=$(grep "^${DISK}2" /proc/mounts | awk '{print $2}')
|
|
if [ -n "$mtpnt" ]; then
|
|
umount $mtpnt >/dev/null 2>&1
|
|
fi
|
|
|
|
if [ "$SECUREBOOT" != "YES" ]; then
|
|
mkdir ./tmp_mnt
|
|
|
|
vtdebug "mounting part2 ...."
|
|
for tt in 1 2 3; do
|
|
if mount ${DISK}2 ./tmp_mnt; then
|
|
vtdebug "mounting part2 success"
|
|
break
|
|
fi
|
|
|
|
mtpnt=$(grep "^${DISK}2" /proc/mounts | awk '{print $2}')
|
|
if [ -n "$mtpnt" ]; then
|
|
umount $mtpnt >/dev/null 2>&1
|
|
fi
|
|
sleep 2
|
|
done
|
|
|
|
rm -f ./tmp_mnt/EFI/BOOT/BOOTX64.EFI
|
|
rm -f ./tmp_mnt/EFI/BOOT/grubx64.efi
|
|
rm -f ./tmp_mnt/EFI/BOOT/MokManager.efi
|
|
rm -f ./tmp_mnt/ENROLL_THIS_KEY_IN_MOKMANAGER.cer
|
|
mv ./tmp_mnt/EFI/BOOT/grubx64_real.efi ./tmp_mnt/EFI/BOOT/BOOTX64.EFI
|
|
|
|
umount ./tmp_mnt
|
|
rm -rf ./tmp_mnt
|
|
fi
|
|
|
|
echo ""
|
|
vtinfo "Install Ventoy to $DISK successfully finished."
|
|
echo ""
|
|
|
|
else
|
|
vtdebug "update ventoy ..."
|
|
|
|
oldver=$(get_disk_ventoy_version $DISK)
|
|
if [ $? -ne 0 ]; then
|
|
vtwarn "$DISK does not contain ventoy or data corupted"
|
|
echo ""
|
|
vtwarn "Please use -i option if you want to install ventoy to $DISK"
|
|
echo ""
|
|
exit 1
|
|
fi
|
|
|
|
curver=$(cat ./ventoy/version)
|
|
|
|
vtinfo "Upgrade operation is safe, all the data in the 1st partition (iso files and other) will be unchanged!"
|
|
echo ""
|
|
|
|
read -p "Update Ventoy $oldver ===> $curver Continue? (y/n)" Answer
|
|
if [ "$Answer" != "y" ]; then
|
|
if [ "$Answer" != "Y" ]; then
|
|
exit 0
|
|
fi
|
|
fi
|
|
|
|
PART2=$(get_disk_part_name $DISK 2)
|
|
SHORT_PART2=${PART2#/dev/}
|
|
part2_start=$(cat /sys/class/block/$SHORT_PART2/start)
|
|
|
|
PART1_TYPE=$(dd if=$DISK bs=1 count=1 skip=450 status=none | ./tool/hexdump -n1 -e '1/1 "%02X"')
|
|
|
|
if [ "$PART1_TYPE" = "EE" ]; then
|
|
vtdebug "This is GPT partition style ..."
|
|
./tool/xzcat ./boot/core.img.xz | dd status=none conv=fsync of=$DISK bs=512 count=2014 seek=34
|
|
echo -en '\x23' | dd of=$DISK conv=fsync bs=1 count=1 seek=17908 status=none
|
|
else
|
|
vtdebug "This is MBR partition style ..."
|
|
dd status=none conv=fsync if=./boot/boot.img of=$DISK bs=1 count=440
|
|
|
|
PART1_ACTIVE=$(dd if=$DISK bs=1 count=1 skip=446 status=none | ./tool/hexdump -n1 -e '1/1 "%02X"')
|
|
PART2_ACTIVE=$(dd if=$DISK bs=1 count=1 skip=462 status=none | ./tool/hexdump -n1 -e '1/1 "%02X"')
|
|
|
|
vtdebug "PART1_ACTIVE=$PART1_ACTIVE PART2_ACTIVE=$PART2_ACTIVE"
|
|
if [ "$PART1_ACTIVE" = "00" ] && [ "$PART2_ACTIVE" = "80" ]; then
|
|
vtdebug "change 1st partition active, 2nd partition inactive ..."
|
|
echo -en '\x80' | dd of=$DISK conv=fsync bs=1 count=1 seek=446 status=none
|
|
echo -en '\x00' | dd of=$DISK conv=fsync bs=1 count=1 seek=462 status=none
|
|
fi
|
|
./tool/xzcat ./boot/core.img.xz | dd status=none conv=fsync of=$DISK bs=512 count=2047 seek=1
|
|
fi
|
|
|
|
./tool/xzcat ./ventoy/ventoy.disk.img.xz | dd status=none conv=fsync of=$DISK bs=512 count=$VENTOY_SECTOR_NUM seek=$part2_start
|
|
|
|
sync
|
|
|
|
if [ "$SECUREBOOT" != "YES" ]; then
|
|
mkdir ./tmp_mnt
|
|
|
|
vtdebug "mounting part2 ...."
|
|
for tt in 1 2 3; do
|
|
if mount ${DISK}2 ./tmp_mnt; then
|
|
vtdebug "mounting part2 success"
|
|
break
|
|
fi
|
|
sleep 2
|
|
done
|
|
|
|
rm -f ./tmp_mnt/EFI/BOOT/BOOTX64.EFI
|
|
rm -f ./tmp_mnt/EFI/BOOT/grubx64.efi
|
|
rm -f ./tmp_mnt/EFI/BOOT/MokManager.efi
|
|
rm -f ./tmp_mnt/ENROLL_THIS_KEY_IN_MOKMANAGER.cer
|
|
mv ./tmp_mnt/EFI/BOOT/grubx64_real.efi ./tmp_mnt/EFI/BOOT/BOOTX64.EFI
|
|
|
|
umount ./tmp_mnt
|
|
rm -rf ./tmp_mnt
|
|
fi
|
|
|
|
echo ""
|
|
vtinfo "Update Ventoy to $DISK successfully finished."
|
|
echo ""
|
|
|
|
fi
|
|
|
|
|