INSTALL/tool/ventoy_lib.sh

   1 #!/bin/sh
   2
   3 #Ventoy partition 32MB
   4 VENTOY_PART_SIZE=33554432
   5 VENTOY_PART_SIZE_MB=32
   6 VENTOY_SECTOR_SIZE=512
   7 VENTOY_SECTOR_NUM=65536
   8
   9 ventoy_false() {
  10     [ "1" = "2" ]
  11 }
  12
  13 ventoy_true() {
  14     [ "1" = "1" ]
  15 }
  16
  17 ventoy_is_linux64() {
  18     if uname -a | egrep -q 'x86_64|amd64'; then
  19         ventoy_true
  20         return
  21     fi
  22
  23     ventoy_false
  24 }
  25
  26 vtinfo() {
  27     echo -e "\033[32m$*\033[0m"
  28 }
  29
  30 vtwarn() {
  31     echo -e "\033[33m$*\033[0m"
  32 }
  33
  34
  35 vterr() {
  36     echo -e "\033[31m$*\033[0m"
  37 }
  38
  39 vtdebug() {
  40     echo "$*" >> ./log.txt
  41 }
  42
  43 check_tool_work_ok() {
  44
  45     if ventoy_is_linux64; then
  46         vtdebug "This is linux 64"
  47         mkexfatfs=mkexfatfs_64
  48         vtoyfat=vtoyfat_64
  49     else
  50         vtdebug "This is linux 32"
  51         mkexfatfs=mkexfatfs_32
  52         vtoyfat=vtoyfat_32
  53     fi
  54
  55     if echo 1 | ./tool/hexdump > /dev/null; then
  56         vtdebug "hexdump test ok ..."
  57     else
  58         vtdebug "hexdump test fail ..."
  59         ventoy_false
  60         return
  61     fi
  62
  63     if ./tool/$mkexfatfs -V > /dev/null; then
  64         vtdebug "$mkexfatfs test ok ..."
  65     else
  66         vtdebug "$mkexfatfs test fail ..."
  67         ventoy_false
  68         return
  69     fi
  70
  71     if ./tool/$vtoyfat -T; then
  72         vtdebug "$vtoyfat test ok ..."
  73     else
  74         vtdebug "$vtoyfat test fail ..."
  75         ventoy_false
  76         return
  77     fi
  78
  79     vtdebug "tool check success ..."
  80     ventoy_true
  81 }
  82
  83
  84 get_disk_part_name() {
  85     DISK=$1
  86
  87     if echo $DISK | grep -q "/dev/loop"; then
  88         echo ${DISK}p${2}
  89     elif echo $DISK | grep -q "/dev/nvme[0-9][0-9]*n[0-9]"; then
  90         echo ${DISK}p${2}
  91     else
  92         echo ${DISK}${2}
  93     fi
  94 }
  95
  96
  97 get_ventoy_version_from_cfg() {
  98     if grep -q 'set.*VENTOY_VERSION=' $1; then
  99         grep 'set.*VENTOY_VERSION=' $1 | awk -F'"' '{print $2}'
 100     else
 101         echo 'none'
 102     fi
 103 }
 104
 105 is_disk_contains_ventoy() {
 106     DISK=$1
 107
 108     PART1=$(get_disk_part_name $1 1)
 109     PART2=$(get_disk_part_name $1 2)
 110
 111     if [ -e /sys/class/block/${PART2#/dev/}/size ]; then
 112         SIZE=$(cat /sys/class/block/${PART2#/dev/}/size)
 113     else
 114         SIZE=0
 115     fi
 116
 117     if ! [ -b $PART1 ]; then
 118         vtdebug "$PART1 not exist"
 119         ventoy_false
 120         return
 121     fi
 122
 123     if ! [ -b $PART2 ]; then
 124         vtdebug "$PART2 not exist"
 125         ventoy_false
 126         return
 127     fi
 128
 129     PART2_TYPE=$(dd if=$DISK bs=1 count=1 skip=466 status=none | ./tool/hexdump -n1 -e  '1/1 "%02X"')
 130     if [ "$PART2_TYPE" != "EF" ]; then
 131         vtdebug "part2 type is $PART2_TYPE not EF"
 132         ventoy_false
 133         return
 134     fi
 135
 136     # PART1_TYPE=$(dd if=$DISK bs=1 count=1 skip=450 status=none | ./tool/hexdump -n1 -e  '1/1 "%02X"')
 137     # if [ "$PART1_TYPE" != "07" ]; then
 138         # vtdebug "part1 type is $PART2_TYPE not 07"
 139         # ventoy_false
 140         # return
 141     # fi
 142
 143     if [ -e /sys/class/block/${PART1#/dev/}/start ]; then
 144         PART1_START=$(cat /sys/class/block/${PART1#/dev/}/start)
 145     fi
 146
 147     if [ "$PART1_START" != "2048" ]; then
 148         vtdebug "part1 start is $PART1_START not 2048"
 149         ventoy_false
 150         return
 151     fi
 152
 153     if [ "$VENTOY_SECTOR_NUM" != "$SIZE" ]; then
 154         vtdebug "part2 size is $SIZE not $VENTOY_SECTOR_NUM"
 155         ventoy_false
 156         return
 157     fi
 158
 159     ventoy_true
 160 }
 161
 162 get_disk_ventoy_version() {
 163
 164     if ! is_disk_contains_ventoy $1; then
 165         ventoy_false
 166         return
 167     fi
 168
 169     PART2=$(get_disk_part_name $1 2)
 170
 171     if ventoy_is_linux64; then
 172         cmd=./tool/vtoyfat_64
 173     else
 174         cmd=./tool/vtoyfat_32
 175     fi
 176
 177     ParseVer=$($cmd $PART2)
 178     if [ $? -eq 0 ]; then
 179         vtdebug "Ventoy version in $PART2 is $ParseVer"
 180         echo $ParseVer
 181         ventoy_true
 182         return
 183     fi
 184
 185     ventoy_false
 186 }
 187
 188 format_ventoy_disk() {
 189     reserve_mb=$1
 190     DISK=$2
 191     PARTTOOL=$3
 192
 193     PART1=$(get_disk_part_name $DISK 1)
 194     PART2=$(get_disk_part_name $DISK 2)
 195
 196     sector_num=$(cat /sys/block/${DISK#/dev/}/size)
 197
 198     part1_start_sector=2048
 199
 200     if [ $reserve_mb -gt 0 ]; then
 201         reserve_sector_num=$(expr $reserve_mb \* 2048)
 202         part1_end_sector=$(expr $sector_num - $reserve_sector_num - $VENTOY_SECTOR_NUM - 1)
 203     else
 204         part1_end_sector=$(expr $sector_num - $VENTOY_SECTOR_NUM - 1)
 205     fi
 206
 207     part2_start_sector=$(expr $part1_end_sector + 1)
 208     part2_end_sector=$(expr $part2_start_sector + $VENTOY_SECTOR_NUM - 1)
 209
 210     export part2_start_sector
 211
 212     vtdebug "part1_start_sector=$part1_start_sector  part1_end_sector=$part1_end_sector"
 213     vtdebug "part2_start_sector=$part2_start_sector  part2_end_sector=$part2_end_sector"
 214
 215     if [ -e $PART2 ]; then
 216         echo "delete $PART2"
 217         rm -f $PART2
 218     fi
 219
 220     echo ""
 221     echo "Create partitions on $DISK by $PARTTOOL ..."
 222
 223     if [ "$PARTTOOL" = "parted" ]; then
 224         vtdebug "format disk by parted ..."
 225         parted -a none --script $DISK \
 226             mklabel msdos \
 227             unit s \
 228             mkpart primary ntfs $part1_start_sector $part1_end_sector \
 229             mkpart primary fat16 $part2_start_sector $part2_end_sector \
 230             set 1 boot on \
 231             quit
 232
 233         sync
 234         echo -en '\xEF' | dd of=$DISK conv=fsync bs=1 count=1 seek=466 > /dev/null 2>&1
 235     else
 236     vtdebug "format disk by fdisk ..."
 237
 238 fdisk $DISK >>./log.txt 2>&1 <<EOF
 239 o
 240 n
 241 p
 242 1
 243 $part1_start_sector
 244 $part1_end_sector
 245 n
 246 p
 247 2
 248 $part2_start_sector
 249 $part2_end_sector
 250 t
 251 1
 252 7
 253 t
 254 2
 255 ef
 256 a
 257 1
 258 w
 259 EOF
 260     fi
 261
 262     udevadm trigger >/dev/null 2>&1
 263     partprobe >/dev/null 2>&1
 264     sleep 3
 265     echo "Done"
 266
 267     echo 'mkfs on disk partitions ...'
 268     for i in 1 2 3 4 5 6 7; do
 269         if [ -b $PART2 ]; then
 270             break
 271         else
 272             echo "wait $PART2 ..."
 273             sleep 1
 274         fi
 275     done
 276
 277
 278     if ! [ -b $PART2 ]; then
 279         MajorMinor=$(sed "s/:/ /" /sys/class/block/${PART2#/dev/}/dev)
 280         echo "mknod -m 0660 $PART2 b $MajorMinor ..."
 281         mknod -m 0660 $PART2 b $MajorMinor
 282
 283         if ! [ -b $PART1 ]; then
 284             MajorMinor=$(sed "s/:/ /" /sys/class/block/${PART1#/dev/}/dev)
 285             echo "mknod -m 0660 $PART1 b $MajorMinor ..."
 286             mknod -m 0660 $PART1 b $MajorMinor
 287         fi
 288     fi
 289
 290     echo "create efi fat fs $PART2 ..."
 291     for i in 0 1 2 3 4 5 6 7 8 9; do
 292         if mkfs.vfat -F 16 -n VTOYEFI $PART2; then
 293             echo 'success'
 294             break
 295         else
 296             echo "$? retry ..."
 297             sleep 2
 298         fi
 299     done
 300 }
 301
 302
 303
 304