ceviriler/VentoyWorker.sh

#!/bin/sh

. ./tool/ventoy_lib.sh

print_usage() {
    
    echo 'Kullanımı:  Ventoy2Disk.sh CMD [ OPTION ] /dev/sdX'
    echo '  CMD:'
    echo '   -i  install Ventoy to sdX (fails 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 '   -l  list Ventoy information in sdX'
    echo ''
    echo '  OPTION: (optional)'
    echo '   -r SIZE_MB  preserve some space at the bottom of the disk (only for install)'
    echo '   -s/-S       enable/disable secure boot support (default is disabled)'
    echo '   -g          use GPT partition style, default is MBR (only for install)'
    echo '   -L          Label of the 1st exfat partition (default is Ventoy)'
    echo ''
}


VTNEW_LABEL='Ventoy'
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" = "-l" ]; then
        MODE="list"
    elif [ "$1" = "-s" ]; then
        SECUREBOOT="YES"
    elif [ "$1" = "-S" ]; then
        SECUREBOOT="NO"
    elif [ "$1" = "-g" ]; then
        VTGPT="YES"
    elif [ "$1" = "-L" ]; then
        shift
        VTNEW_LABEL=$1
    elif [ "$1" = "-r" ]; then
        RESERVE_SPACE="YES"
        shift
        RESERVE_SIZE_MB=$1
    elif [ "$1" = "-V" ] || [ "$1" = "--version" ]; then
        exit 0
    elif [ "$1" = "-h" ] || [ "$1" = "--help" ]; then
        print_usage
        exit 0
    else
        if ! [ -b "$1" ]; then
            vterr "$1 geçerli bir cihaz DEĞİLDİR"
            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 diski mevcut değil"
    exit 1
fi

if [ -e /sys/class/block/${DISK#/dev/}/start ]; then
    vterr  "$DISK bir bölüm, lütfen tüm diski kullanın."
    echo   "Örnek:"
    vterr  "    sudo sh Ventoy2Disk.sh -i /dev/sdb1 <=== Bu yanlış"
    vtinfo "    sudo sh Ventoy2Disk.sh -i /dev/sdb  <=== Bu doğru"
    echo ""
    exit 1
fi

if [ -n "$RESERVE_SPACE" -a "$MODE" = "install" ]; then
    if echo $RESERVE_SIZE_MB | grep -q '^[0-9][0-9]*$'; then
        vtdebug "Kullanıcı $RESERVE_SIZE_MB  MB disk alanı ayıracak"
    else
        vterr "$RESERVE_SIZE_MB ayrılmış alan için geçersiz"
        exit 1
    fi
fi

vtdebug "MODE=$MODE FORCE=$FORCE RESERVE_SPACE=$RESERVE_SPACE RESERVE_SIZE_MB=$RESERVE_SIZE_MB"

#check tools
if check_tool_work_ok; then
    vtdebug "aracı kontrol et tamam"
else
    vterr "Bazı araçlar mevcut sistemde çalışamaz. Ayrıntılar için lütfen /var/log/ventoy.log adresini kontrol edin."
    exit 1
fi

if [ "$MODE" = "list" ]; then
    version=$(get_disk_ventoy_version $DISK)
    if [ $? -eq 0 ]; then
        echo "Diskteki Ventoy Sürümü: $version"
        
        vtPart1Type=$(dd if=$DISK bs=1 count=1 skip=450 status=none | hexdump -n1 -e  '1/1 "%02X"')
        if [ "$vtPart1Type" = "EE" ]; then            
            echo "Disk Bölüm Türü  : GPT"
        else
            echo "Disk Bölüm Türü   : MBR"
        fi
        
        if check_disk_secure_boot $DISK; then
            echo "Secure Boot Desteği   : EVET"
        else
            echo "Secure Boot Desteği   : HAYIR"
        fi        
    else
        echo "Ventoy Sürümü: NA"
    fi
    echo ""
    exit 0
fi

#check mountpoint
check_umount_disk "$DISK"

if grep "$DISK" /proc/mounts; then
    vterr "$DISK zaten bağlı, lütfen önce onu ayırın! "
    exit 1
fi

#check swap partition
if swapon --help 2>&1 | grep -q '^ \-s,'; then
    if swapon -s | grep -q "^${DISK}[0-9]"; then
        vterr "$DISK is used as swap, please swapoff it first!"
        exit 1
    fi
fi

#check access 
if dd if="$DISK" of=/dev/null bs=1 count=1 >/dev/null 2>&1; then
    vtdebug "kök izin kontrolü tamam..."
else
    vterr "Failed to access $DISK, maybe root privilege is needed!"
    echo ''
    exit 1
fi


#check tmp_mnt directory
if [ -d ./tmp_mnt ]; then
    vtdebug "Bir tmp_mnt dizini var, şimdi silin."
    umount ./tmp_mnt >/dev/null 2>&1
    rm -rf ./tmp_mnt
    if [ -d ./tmp_mnt ]; then
        vterr "tmp_mnt dizin var, lütfen önce onu silin."
        exit 1
    fi
fi


if [ "$MODE" = "install" ]; then
    vtdebug "Ventoy kur ..."

    if [ -n "$VTGPT" ]; then
        if parted -v > /dev/null 2>&1; then
            PARTTOOL='parted'
        else
            vterr "parted sistemde bulunmazsa, Ventoy onsuz yeni bölümler oluşturamaz."
            vterr "Önce\"GNU parted\" kurmalısınız."
            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 "Hem parted hem de fdisk sistemde bulunmazsa, Ventoy yeni bölümler oluşturamaz."
            exit 1
        fi
    fi
    
    version=$(get_disk_ventoy_version $DISK)
    if [ $? -eq 0 ]; then
        if [ -z "$FORCE" ]; then
            vtwarn "$DISK zaten $version sürümüne sahip bir Ventoy içeriyor "
            vtwarn "Güvenli bir yükseltme işlemi yapmak için -u seçeneğini kullanın."
            vtwarn "VEYA Ventoy'u gerçekten $DISK diskine yeniden yüklemek istiyorsanız, lütfen -I seçeneğini kullanın."
            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 2TB boyutunun üzerinde, MBR çalışmayacaktır.."
        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 "Tür: GPT"
    else
        echo "Tür: MBR"
    fi    
    echo ''

    if [ -n "$RESERVE_SPACE" ]; then
        echo "$RESERVE_SIZE_MB MB disk alanı ayıracaksınız "
    fi
    echo ''

    vtwarn "Dikkat:"
    vtwarn "Ventoy $DISK diskine kurulacak."
    vtwarn "Bu işlem $DISK içindeki tüm verilerin kaybolmasına sebep olacak!!!"
    echo ""

    read -p 'Devam edilsin mi? (e/h) '  Answer
    if [ "$Answer" != "e" ]; then
        if [ "$Answer" != "E" ]; then
            exit 0
        fi
    fi

    echo ""
    vtwarn "$DISK diskindeki tüm veriler silinecek !!!"
    read -p 'İkinci kontrol, emin misiniz? (e/h) '  Answer
    if [ "$Answer" != "e" ]; then
        if [ "$Answer" != "E" ]; then
            exit 0
        fi
    fi

    if [ $disk_sector_num -le $VENTOY_SECTOR_NUM ]; then  
        vterr "$DISK diskinde yeterli alan yok"
        exit 1
    fi

    if ! dd if=/dev/zero of=$DISK bs=1 count=512 status=none conv=fsync; then
        vterr "$DISK diskine veri yazılamadı, lütfen kullanımda olup olmadığını kontrol edin."
        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

    # 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

    PART1=$(get_disk_part_name $DISK 1)  
    PART2=$(get_disk_part_name $DISK 2)  

    mkexfatfs -n "$VTNEW_LABEL" -s $cluster_sectors ${PART1}

    vtinfo "diske veri yazılıyor ..."
    
    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        
        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
        xzcat ./boot/core.img.xz | dd status=none conv=fsync of=$DISK bs=512 count=2047 seek=1
    fi
    
    # check and umount
    check_umount_disk "$DISK"
    
    xzcat ./ventoy/ventoy.disk.img.xz | dd status=none conv=fsync of=$DISK bs=512 count=$VENTOY_SECTOR_NUM seek=$part2_start_sector
    
    #test UUID
    testUUIDStr=$(vtoy_gen_uuid | hexdump -C)
    vtdebug "test uuid: $testUUIDStr"
    
    #disk uuid
    vtoy_gen_uuid | dd status=none conv=fsync of=${DISK} seek=384 bs=1 count=16
    
    #disk signature
    vtoy_gen_uuid | dd status=none conv=fsync of=${DISK} skip=12 seek=440 bs=1 count=4

    vtinfo "sync data ..."
    sync
    
    vtinfo "esp bölüm işleniyor.."
    
    sleep 1
    check_umount_disk "$DISK"    
    
    if [ "$SECUREBOOT" != "YES" ]; then        
        mkdir ./tmp_mnt
        
        vtdebug "part2 bağlanıyor ...."
        for tt in 1 2 3 4 5; do            
            if mount ${PART2} ./tmp_mnt > /dev/null 2>&1; then
                vtdebug "part2 bağlama başarılı"
                break
            fi
            
            check_umount_disk "$DISK"            
            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/BOOTIA32.EFI
        rm -f ./tmp_mnt/EFI/BOOT/grubia32.efi
        rm -f ./tmp_mnt/EFI/BOOT/MokManager.efi
        rm -f ./tmp_mnt/EFI/BOOT/mmia32.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
        mv ./tmp_mnt/EFI/BOOT/grubia32_real.efi  ./tmp_mnt/EFI/BOOT/BOOTIA32.EFI
        
        sync
        
        for tt in 1 2 3; do
            if umount ./tmp_mnt; then
                vtdebug "part2 ayırma başarılı"
                rm -rf ./tmp_mnt
                break
            else
                vtdebug "part2 ayırma hatalı, şimdi yeniden dene.."
                sleep 1
            fi
        done
    fi

    echo ""
    vtinfo "Ventoy'u  $DISK diskine yükleme başarıyla tamamlandı."
    echo ""
    
else
    vtdebug " Ventoy güncelle ..."
    
    oldver=$(get_disk_ventoy_version $DISK)
    if [ $? -ne 0 ]; then
        vtwarn "$DISK Ventoy içermiyor veya veri bozuk"
        echo ""
        vtwarn "Ventoy'i $DISK diskine yüklemek istiyorsanız lütfen -i seçeneğini kullanın "
        echo ""
        exit 1
    fi

    #reserve secure boot option
    if [ -z "$SECUREBOOT" ]; then
        if check_disk_secure_boot $DISK; then
            SECUREBOOT="YES"
        else
            SECUREBOOT="NO"
        fi
    fi

    curver=$(cat ./ventoy/version)

    vtinfo "Yükseltme işlemi güvenlidir, 1. bölümdeki tüm veriler (iso dosyaları ve diğerleri) değişmeyecektir!"
    echo ""

    read -p "Ventoy güncelleme  $oldver ===> $curver   Devam edilsin mi? (e/h)"  Answer
    if [ "$Answer" != "e" ]; then
        if [ "$Answer" != "E" ]; 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 | hexdump -n1 -e  '1/1 "%02X"')
    
    #reserve disk uuid
    rm -f ./diskuuid.bin
    dd status=none conv=fsync if=${DISK} skip=384 bs=1 count=16 of=./diskuuid.bin
    
    dd status=none conv=fsync if=./boot/boot.img of=$DISK bs=1 count=440
    dd status=none conv=fsync if=./diskuuid.bin of=$DISK bs=1 count=16 seek=384
    rm -f ./diskuuid.bin

    #reserve data
    rm -f ./rsvdata.bin
    dd status=none conv=fsync if=${DISK} skip=2040 bs=512 count=8 of=./rsvdata.bin

    if [ "$PART1_TYPE" = "EE" ]; then
        vtdebug "Bu GPT bölüm stili ..."    
        echo -en '\x22' | dd status=none of=$DISK conv=fsync bs=1 count=1 seek=92
        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 "Bu MBR bölüm stili..."
    
        PART1_ACTIVE=$(dd if=$DISK bs=1 count=1 skip=446 status=none | hexdump -n1 -e  '1/1 "%02X"')
        PART2_ACTIVE=$(dd if=$DISK bs=1 count=1 skip=462 status=none | hexdump -n1 -e  '1/1 "%02X"')
        
        vtdebug "PART1_ACTIVE=$PART1_ACTIVE  PART2_ACTIVE=$PART2_ACTIVE"
        if [ "$PART1_ACTIVE" = "00" ] && [ "$PART2_ACTIVE" = "80" ]; then
            vtdebug "1. bölümü değiştir aktif, 2. bölüm aktif değil ..."
            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
        xzcat ./boot/core.img.xz | dd status=none conv=fsync of=$DISK bs=512 count=2047 seek=1
    fi

    dd status=none conv=fsync if=./rsvdata.bin seek=2040 bs=512 count=8 of=${DISK}
    rm -f ./rsvdata.bin

    check_umount_disk "$DISK"
    
    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 "part2 bağlanıyor ...."        
        for tt in 1 2 3 4 5; do
            check_umount_disk "$DISK"

            if mount ${PART2} ./tmp_mnt > /dev/null 2>&1; then
                vtdebug "part2 bağlama başarılı"
                break
            else
                vtdebug "part2 bağlama başarısız oldu, şimdi bekleyin ve yeniden deneyin..."
            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/BOOTIA32.EFI
        rm -f ./tmp_mnt/EFI/BOOT/grubia32.efi
        rm -f ./tmp_mnt/EFI/BOOT/MokManager.efi
        rm -f ./tmp_mnt/EFI/BOOT/mmia32.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
        mv ./tmp_mnt/EFI/BOOT/grubia32_real.efi  ./tmp_mnt/EFI/BOOT/BOOTIA32.EFI
        
        sync
        
        for tt in 1 2 3; do
            if umount ./tmp_mnt > /dev/null 2>&1; then
                vtdebug "part2 ayırma başarılı"
                rm -rf ./tmp_mnt
                break
            else
                vtdebug "umount part2 başarısız oldu, şimdi yeniden deneyin..."
                sleep 1
            fi
        done        
    fi

    echo ""
    vtinfo "$DISK üzerindeki Ventoy güncellemesi başarıyla tamamlandı."
    echo ""
    
fi