2 #************************************************************************************
3 # Copyright (c) 2020, longpanda <admin@ventoy.net>
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.
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.
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/>.
18 #************************************************************************************
21 BUSYBOX_PATH
=$VTOY_PATH/busybox
23 FIND
=$BUSYBOX_PATH/find
24 GREP
=$BUSYBOX_PATH/grep
25 EGREP
=$BUSYBOX_PATH/egrep
29 SLEEP
=$BUSYBOX_PATH/sleep
30 HEAD
=$BUSYBOX_PATH/head
31 VTOY_DM_PATH
=/dev
/mapper
/ventoy
32 VTOY_DEBUG_LEVEL
=$($BUSYBOX_PATH/hexdump -n 1 -s 450 -e '1/1 "%02x"' $VTOY_PATH/ventoy_os_param)
34 if [ "$VTOY_DEBUG_LEVEL" = "01" ]; then
35 if [ -e /dev
/console
]; then
41 if [ "$VTLOG" = "$VTOY_PATH/log" ]; then
44 echo -e "\033[32m $* \033[0m" > $VTLOG
50 if [ "$VTLOG" = "$VTOY_PATH/log" ]; then
53 echo -e "\n\033[31m $* \033[0m" > $VTLOG
59 is_ventoy_hook_finished
() {
60 [ -e $VTOY_PATH/hook_finish
]
63 set_ventoy_hook_finish
() {
64 echo 'Y' > $VTOY_PATH/hook_finish
67 get_ventoy_disk_name
() {
68 line
=$($VTOY_PATH/tool/vtoydump -f /ventoy/ventoy_os_param)
76 get_ventoy_iso_name
() {
77 line
=$($VTOY_PATH/tool/vtoydump -f /ventoy/ventoy_os_param)
85 wait_for_usb_disk_ready
() {
87 usb_disk
=$(get_ventoy_disk_name)
88 vtlog
"wait_for_usb_disk_ready $usb_disk ..."
90 if echo $usb_disk | $EGREP -q "nvme|mmc|nbd"; then
96 if [ -e "${vtpart2}" ]; then
97 vtlog
"wait_for_usb_disk_ready $usb_disk finish"
105 check_usb_disk_ready
() {
106 if echo $1 | $EGREP -q "nvme|mmc|nbd"; then
116 if $VTOY_PATH/tool
/vtoydump
-f $VTOY_PATH/ventoy_os_param
-c "$1"; then
124 if echo $1 | $EGREP -q "nvme.*p$|mmc.*p$|nbd.*p$"; then
130 if $VTOY_PATH/tool
/vtoydump
-f $VTOY_PATH/ventoy_os_param
-c "$vtDiskName"; then
137 ventoy_get_vblade_bin
() {
138 if $VTOY_PATH/tool
/vblade_64
-t >>$VTLOG 2>&1; then
139 echo $VTOY_PATH/tool
/vblade_64
141 echo $VTOY_PATH/tool
/vblade_32
145 ventoy_find_bin_path
() {
146 if $BUSYBOX_PATH/which "$1" > /dev
/null
; then
147 $BUSYBOX_PATH/which "$1"; return
150 for vt_path
in '/bin' '/sbin' '/usr/bin' '/usr/sbin' '/usr/local/bin' '/usr/local/sbin' '/root/bin'; do
151 if [ -e "$vt_path/$1" ]; then
152 echo "$vt_path/$1"; return
160 ventoy_find_bin_run
() {
162 if [ "$1" = "sudo" ]; then
167 vtbinpath
=$(ventoy_find_bin_path "$1")
168 if [ -n "$vtbinpath" ]; then
171 if [ $vtsudo -eq 0 ]; then
172 vtlog
"$vtbinpath $*"
175 vtlog
"sudo $vtbinpath $*"
181 ventoy_get_module_postfix
() {
182 vtKerVer
=$($BUSYBOX_PATH/uname -r)
183 vtLine
=$($FIND /lib/modules/$vtKerVer/ -name *.ko* | $HEAD -n1)
184 vtComp
=${vtLine##*/*.ko}
188 ventoy_check_dm_module
() {
189 if $GREP -q 'device-mapper' /proc
/devices
; then
190 $BUSYBOX_PATH/true
; return
193 vtlog
"device-mapper NOT found in /proc/devices, try to load kernel module"
194 $BUSYBOX_PATH/modprobe dm_mod
>>$VTLOG 2>&1
195 $BUSYBOX_PATH/modprobe dm
-mod >>$VTLOG 2>&1
197 if ! $GREP -q 'device-mapper' /proc
/devices
; then
198 vtlog
"modprobe failed, now try to insmod ko..."
200 $FIND /lib
/modules
/ -name "dm-mod.ko*" | while read vtline
; do
201 vtlog
"insmode $vtline "
202 $BUSYBOX_PATH/insmod
$vtline >>$VTLOG 2>&1
206 if $GREP -q 'device-mapper' /proc
/devices
; then
207 vtlog
"device-mapper found in /proc/devices after retry"
208 $BUSYBOX_PATH/true
; return
210 vtlog
"device-mapper still NOT found in /proc/devices after retry"
211 $BUSYBOX_PATH/false
; return
215 create_ventoy_device_mapper
() {
216 vtlog
"create_ventoy_device_mapper $*"
218 VT_DM_BIN
=$(ventoy_find_bin_path dmsetup)
219 if [ -z "$VT_DM_BIN" ]; then
220 vtlog
"no dmsetup avaliable, lastly try inbox dmsetup"
221 VT_DM_BIN
=$VTOY_PATH/tool
/dmsetup
224 vtlog
"dmsetup avaliable in system $VT_DM_BIN"
226 if ventoy_check_dm_module
"$1"; then
227 vtlog
"device-mapper module check success"
229 vterr
"Error: no dm module avaliable"
232 $VTOY_PATH/tool
/vtoydm
-p -f $VTOY_PATH/ventoy_image_map
-d $1 > $VTOY_PATH/ventoy_dm_table
234 $VT_DM_BIN create ventoy
$VTOY_PATH/ventoy_dm_table
>>$VTLOG 2>&1
236 $VT_DM_BIN "$2" create ventoy
$VTOY_PATH/ventoy_dm_table
>>$VTLOG 2>&1
240 create_persistent_device_mapper
() {
241 vtlog
"create_persistent_device_mapper $*"
243 VT_DM_BIN
=$(ventoy_find_bin_path dmsetup)
244 if [ -z "$VT_DM_BIN" ]; then
245 vtlog
"no dmsetup avaliable, lastly try inbox dmsetup"
246 VT_DM_BIN
=$VTOY_PATH/tool
/dmsetup
249 vtlog
"dmsetup avaliable in system $VT_DM_BIN"
251 if ventoy_check_dm_module
"$1"; then
252 vtlog
"device-mapper module check success"
254 vterr
"Error: no dm module avaliable"
257 $VTOY_PATH/tool
/vtoydm
-p -f $VTOY_PATH/ventoy_persistent_map
-d $1 > $VTOY_PATH/persistent_dm_table
258 $VT_DM_BIN create vtoy_persistent
$VTOY_PATH/persistent_dm_table
>>$VTLOG 2>&1
263 wait_for_ventoy_dm_disk_label
() {
264 DM
=$($BUSYBOX_PATH/readlink $VTOY_DM_PATH)
265 vtlog
"wait_for_ventoy_dm_disk_label $DM ..."
267 for i
in 0 1 2 3 4 5 6 7 8 9; do
268 vtlog
"i=$i ####### ls /dev/disk/by-label/"
269 ls -l /dev
/disk
/by
-label/ >> $VTLOG
271 if ls -l /dev
/disk
/by
-label/ | $GREP -q "$DM"; then
280 if ! [ -e "$1" ]; then
285 if [ -d /tmp
/vtoy_udeb
]; then
286 $BUSYBOX_PATH/rm -rf /tmp
/vtoy_udeb
289 $BUSYBOX_PATH/mkdir -p /tmp
/vtoy_udeb
290 $BUSYBOX_PATH/cp -a "$1" /tmp
/vtoy_udeb
/
292 CURDIR
=$($BUSYBOX_PATH/pwd)
295 $BUSYBOX_PATH/ar x
"$1"
297 if [ -e 'data.tar.gz' ]; then
298 $BUSYBOX_PATH/tar -xzf data.
tar.gz
-C /
299 elif [ -e 'data.tar.xz' ]; then
300 $BUSYBOX_PATH/tar -xJf data.
tar.xz
-C /
301 elif [ -e 'data.tar.bz2' ]; then
302 $BUSYBOX_PATH/tar -xjf data.
tar.bz2
-C /
303 elif [ -e 'data.tar.lzma' ]; then
304 $BUSYBOX_PATH/tar -xaf data.
tar.lzma
-C /
307 if [ -e 'control.tar.gz' ]; then
308 $BUSYBOX_PATH/tar -xzf control.
tar.gz
-C /
309 elif [ -e 'control.tar.xz' ]; then
310 $BUSYBOX_PATH/tar -xJf control.
tar.xz
-C /
311 elif [ -e 'control.tar.bz2' ]; then
312 $BUSYBOX_PATH/tar -xjf control.
tar.bz2
-C /
313 elif [ -e 'control.tar.lzma' ]; then
314 $BUSYBOX_PATH/tar -xaf control.
tar.lzma
-C /
318 $BUSYBOX_PATH/rm -rf /tmp
/vtoy_udeb
323 install_udeb_from_line
() {
324 vtlog
"install_udeb_from_line $1"
326 if ! [ -b "$2" ]; then
327 vterr
"disk #$2# not exist"
331 sector
=$(echo $1 | $AWK '{print $(NF-1)}')
332 length=$(echo $1 | $AWK '{print $NF}')
333 vtlog "sector=$sector length=$length"
335 $VTOY_PATH/tool/vtoydm -e -f $VTOY_PATH/ventoy_image_map -d ${2} -s $sector -l $length -o /tmp/xxx.udeb
336 if [ -e /tmp/xxx.udeb ]; then
337 vtlog "extract udeb file from iso success"
339 vterr "extract udeb file from iso fail"
343 install_udeb_pkg /tmp/xxx.udeb
344 $BUSYBOX_PATH/rm -f /tmp/xxx.udeb
347 extract_file_from_line() {
348 vtlog "extract_file_from_line $1 disk=#$2#"
349 if ! [ -b "$2" ]; then
350 vterr "disk #$2# not exist"
354 sector=$(echo $1 | $AWK '{print $(NF-1)}')
355 length
=$(echo $1 | $AWK '{print $NF}')
356 vtlog
"sector=$sector length=$length"
358 $VTOY_PATH/tool
/vtoydm
-e -f $VTOY_PATH/ventoy_image_map
-d ${2} -s $sector -l $length -o $3
360 vtlog
"extract file from iso success"
363 vterr
"extract file from iso fail"
368 extract_rpm_from_line
() {
369 vtlog
"extract_rpm_from_line $1 disk=#$2#"
371 if ! [ -b "$2" ]; then
372 vterr
"disk #$2# not exist"
376 sector
=$(echo $1 | $AWK '{print $(NF-1)}')
377 length=$(echo $1 | $AWK '{print $NF}')
378 vtlog "sector=$sector length=$length"
380 $VTOY_PATH/tool/vtoydm -e -f $VTOY_PATH/ventoy_image_map -d ${2} -s $sector -l $length -o /tmp/xxx.rpm
381 if [ -e /tmp/xxx.rpm ]; then
382 vtlog "extract rpm file from iso success"
384 vterr "extract rpm file from iso fail"
388 CURPWD=$($BUSYBOX_PATH/pwd)
390 $BUSYBOX_PATH/mkdir -p $VTOY_PATH/rpm
392 vtlog "extract rpm..."
393 $BUSYBOX_PATH/rpm2cpio /tmp/xxx.rpm | $BUSYBOX_PATH/cpio -idm 2>>$VTLOG
396 $BUSYBOX_PATH/rm -f /tmp/xxx.rpm
399 install_rpm_from_line() {
400 vtlog "install_rpm_from_line $1 disk=#$2#"
402 if ! [ -b "$2" ]; then
403 vterr "disk #$2# not exist"
407 sector=$(echo $1 | $AWK '{print $(NF-1)}')
408 length
=$(echo $1 | $AWK '{print $NF}')
409 vtlog
"sector=$sector length=$length"
411 $VTOY_PATH/tool
/vtoydm
-e -f $VTOY_PATH/ventoy_image_map
-d ${2} -s $sector -l $length -o /tmp
/xxx.rpm
412 if [ -e /tmp
/xxx.rpm
]; then
413 vtlog
"extract rpm file from iso success"
415 vterr
"extract rpm file from iso fail"
419 CURPWD
=$($BUSYBOX_PATH/pwd)
422 vtlog
"install rpm..."
423 $BUSYBOX_PATH/rpm2cpio
/tmp
/xxx.rpm
| $BUSYBOX_PATH/cpio -idm 2>>$VTLOG
426 $BUSYBOX_PATH/rm -f /tmp
/xxx.rpm
429 dump_whole_iso_file
() {
430 $VTOY_PATH/tool
/vtoydm
-p -f $VTOY_PATH/ventoy_image_map
-d $usb_disk | while read vtline
; do
431 vtlog
"dmtable line: $vtline"
432 vtcount
=$(echo $vtline | $AWK '{print $2}')
433 vtoffset
=$(echo $vtline | $AWK '{print $NF}')
434 $BUSYBOX_PATH/dd if=$usb_disk of
="$1" bs
=512 count
=$vtcount skip
=$vtoffset oflag
=append conv
=notrunc
438 ventoy_copy_device_mapper
() {
439 if [ -L $VTOY_DM_PATH ]; then
440 vtlog
"replace block device link $1..."
441 $BUSYBOX_PATH/mv "$1" $VTOY_PATH/dev_backup_
${1#/dev/}
442 VT_MAPPER_LINK
=$($BUSYBOX_PATH/readlink $VTOY_DM_PATH)
443 $BUSYBOX_PATH/cp -a "/dev/mapper/$VT_MAPPER_LINK" "$1"
444 elif [ -b $VTOY_DM_PATH ]; then
445 vtlog
"replace block device $1..."
446 $BUSYBOX_PATH/mv "$1" $VTOY_PATH/dev_backup_
${1#/dev/}
447 $BUSYBOX_PATH/cp -a "$VTOY_DM_PATH" "$1"
450 vtlog
"$VTOY_DM_PATH not exist, now check /dev/dm-X ..."
451 VT_DM_BIN
=$(ventoy_find_bin_path dmsetup)
452 if [ -z "$VT_DM_BIN" ]; then
453 vtlog
"no dmsetup avaliable, lastly try inbox dmsetup"
454 VT_DM_BIN
=$VTOY_PATH/tool
/dmsetup
457 DM_VT_ID
=$($VT_DM_BIN ls | $GREP ventoy | $SED 's/.*(\([0-9][0-9]*\),.
*\
([0-9][0-9]*\
).
*/\
1 \
2/')
458 vtlog "DM_VT_ID=$DM_VT_ID ..."
459 $BUSYBOX_PATH/mv "$1" $VTOY_PATH/dev_backup_${1#/dev/}
460 $BUSYBOX_PATH/mknod -m 0666 "$1" b $DM_VT_ID
464 # create link for device-mapper
465 ventoy_create_persistent_link() {
466 blkdev_num=$($VTOY_PATH/tool/dmsetup ls | grep vtoy_persistent | sed 's/.*(\([0-9][0-9]*\),.*\([0-9][0-9]*\).*/\1:\2/')
467 vtDM
=$(ventoy_find_dm_id ${blkdev_num})
469 if ! [ -d /dev
/disk
/by
-label ]; then
470 mkdir -p /dev
/disk
/by
-label
473 VTLABEL
=$($BUSYBOX_PATH/blkid /dev/$vtDM | $SED 's/.*LABEL="\([^"]*\)".*/\1/')
474 if [ -z "$VTLABEL" ]; then
478 vtlog "Persistent Label
: ##${VTLABEL}##"
480 if ! [ -e /dev
/disk
/by
-label/$VTLABEL ]; then
482 cd /dev
/disk
/by
-label
483 ln -s ..
/..
/$vtDM $VTLABEL
488 ventoy_udev_disk_common_hook
() {
489 if echo $1 | $EGREP -q "nvme.*p[0-9]$|mmc.*p[0-9]$|nbd.*p[0-9]$"; then
495 if [ -e /vtoy
/vtoy
]; then
501 # create device mapper for iso image file
502 if create_ventoy_device_mapper
"/dev/$VTDISK" $VTRWMOD; then
503 vtlog
"==== create ventoy device mapper success ===="
505 vtlog
"==== create ventoy device mapper failed ===="
509 if $GREP -q "/dev/$VTDISK" /proc
/mounts
; then
510 $GREP "/dev/$VTDISK" /proc
/mounts
| while read vtLine
; do
511 vtPart
=$(echo $vtLine | $AWK '{print $1}')
512 vtMnt
=$(echo $vtLine | $AWK '{print $2}')
513 vtlog
"$vtPart is mounted on $vtMnt now umount it ..."
514 $BUSYBOX_PATH/umount
$vtMnt
518 if create_ventoy_device_mapper
"/dev/$VTDISK" $VTRWMOD; then
519 vtlog
"==== create ventoy device mapper success after retry ===="
521 vtlog
"==== create ventoy device mapper failed after retry ===="
526 if [ "$2" = "noreplace" ]; then
527 vtlog
"no need to replace block device"
529 ventoy_copy_device_mapper
"/dev/$1"
532 if [ -f $VTOY_PATH/ventoy_persistent_map
]; then
533 create_persistent_device_mapper
"/dev/$VTDISK"
534 ventoy_create_persistent_link
538 ventoy_create_dev_ventoy_part
() {
539 blkdev_num
=$($VTOY_PATH/tool/dmsetup ls | $GREP ventoy | $SED 's/.*(\([0-9][0-9]*\),.
*\
([0-9][0-9]*\
).
*/\
1 \
2/')
540 $BUSYBOX_PATH/mknod -m 0666 /dev/ventoy b $blkdev_num
542 if [ -e /vtoy_dm_table ]; then
545 $CAT /vtoy_dm_table | while read vtline; do
546 echo $vtline > /ventoy/dm_table_part${vtPartid}
547 $VTOY_PATH/tool/dmsetup create ventoy${vtPartid} /ventoy/dm_table_part${vtPartid}
549 blkdev_num=$($VTOY_PATH/tool/dmsetup ls | $GREP ventoy${vtPartid} | $SED 's/.*(\([0-9][0-9]*\),.*\([0-9][0-9]*\).*/\1 \2/')
550 $BUSYBOX_PATH/mknod
-m 0666 /dev
/ventoy
${vtPartid} b
$blkdev_num
552 vtPartid
=$(expr $vtPartid + 1)
557 is_inotify_ventoy_part
() {
558 if echo $1 | $GREP -q "2$"; then
559 if ! [ -e /sys
/block
/$1 ]; then
560 if [ -e /sys
/class
/block
/$1 ]; then
561 if echo $1 | $EGREP -q "nvme|mmc|nbd"; then
562 vtShortName
=${1:0:-2}
564 vtShortName
=${1:0:-1}
567 if [ -e /dev
/$vtShortName ]; then
568 $VTOY_PATH/tool
/vtoydump
-f $VTOY_PATH/ventoy_os_param
-c $vtShortName
578 ventoy_find_dm_id
() {
579 for vt
in $($BUSYBOX_PATH/ls /sys/block/); do
580 if [ "${vt:0:3}" = "dm-" ]; then
581 vtMajorMinor
=$($CAT /sys/block/$vt/dev)
582 if [ "$vtMajorMinor" = "$1" ]; then
591 ventoy_swap_device
() {
592 mv $1 $VTOY_PATH/swap_tmp_dev
594 mv $VTOY_PATH/swap_tmp_dev
$2
597 ventoy_extract_vtloopex
() {
599 $BUSYBOX_PATH/mkdir -p $VTOY_PATH/partmnt
$VTOY_PATH/vtloopex
600 $BUSYBOX_PATH/mount
-o ro
-t vfat
$1 $VTOY_PATH/partmnt
601 cd $VTOY_PATH/vtloopex
602 $CAT $VTOY_PATH/partmnt
/ventoy
/vtloopex.
cpio | $BUSYBOX_PATH/cpio -idm >> $VTLOG 2>&1
603 $BUSYBOX_PATH/umount
$VTOY_PATH/partmnt
604 $BUSYBOX_PATH/rm -rf $VTOY_PATH/partmnt
607 cd $VTOY_PATH/vtloopex
/$2/
608 $BUSYBOX_PATH/tar -xJf vtloopex.
tar.xz
614 ventoy_check_install_module_xz
() {
615 if [ -f "${1}.xz" ]; then
616 $BUSYBOX_PATH/xz
-d "${1}.xz"
617 $BUSYBOX_PATH/insmod
"$1"