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"; then
96 if [ -e "${vtpart2}" ]; then
97 vtlog
"wait_for_usb_disk_ready $usb_disk finish"
106 if $VTOY_PATH/tool
/vtoydump
-f $VTOY_PATH/ventoy_os_param
-c "$1"; then
114 if echo $1 | $EGREP -q "nvme.*p$|mmc.*p$"; then
120 if $VTOY_PATH/tool
/vtoydump
-f $VTOY_PATH/ventoy_os_param
-c "$vtDiskName"; then
127 ventoy_get_vblade_bin
() {
128 if $VTOY_PATH/tool
/vblade_64
-t >>$VTLOG 2>&1; then
129 echo $VTOY_PATH/tool
/vblade_64
131 echo $VTOY_PATH/tool
/vblade_32
135 ventoy_find_bin_path
() {
136 if $BUSYBOX_PATH/which "$1" > /dev
/null
; then
137 $BUSYBOX_PATH/which "$1"; return
140 for vt_path
in '/bin' '/sbin' '/usr/bin' '/usr/sbin' '/usr/local/bin' '/usr/local/sbin' '/root/bin'; do
141 if [ -e "$vt_path/$1" ]; then
142 echo "$vt_path/$1"; return
150 ventoy_find_bin_run
() {
152 if [ "$1" = "sudo" ]; then
157 vtbinpath
=$(ventoy_find_bin_path "$1")
158 if [ -n "$vtbinpath" ]; then
161 if [ $vtsudo -eq 0 ]; then
162 vtlog
"$vtbinpath $*"
165 vtlog
"sudo $vtbinpath $*"
171 ventoy_get_module_postfix
() {
172 vtKerVer
=$($BUSYBOX_PATH/uname -r)
173 vtLine
=$($FIND /lib/modules/$vtKerVer/ -name *.ko* | $HEAD -n1)
174 vtComp
=${vtLine##*/*.ko}
178 ventoy_check_dm_module
() {
179 if $GREP -q 'device-mapper' /proc
/devices
; then
180 $BUSYBOX_PATH/true
; return
183 vtlog
"device-mapper NOT found in /proc/devices, try to load kernel module"
184 $BUSYBOX_PATH/modprobe dm_mod
>>$VTLOG 2>&1
185 $BUSYBOX_PATH/modprobe dm
-mod >>$VTLOG 2>&1
187 if ! $GREP -q 'device-mapper' /proc
/devices
; then
188 vtlog
"modprobe failed, now try to insmod ko..."
190 $FIND /lib
/modules
/ -name "dm-mod.ko*" | while read vtline
; do
191 vtlog
"insmode $vtline "
192 $BUSYBOX_PATH/insmod
$vtline >>$VTLOG 2>&1
196 if $GREP -q 'device-mapper' /proc
/devices
; then
197 vtlog
"device-mapper found in /proc/devices after retry"
198 $BUSYBOX_PATH/true
; return
200 vtlog
"device-mapper still NOT found in /proc/devices after retry"
201 $BUSYBOX_PATH/false
; return
205 create_ventoy_device_mapper
() {
206 vtlog
"create_ventoy_device_mapper $*"
208 VT_DM_BIN
=$(ventoy_find_bin_path dmsetup)
209 if [ -z "$VT_DM_BIN" ]; then
210 vtlog
"no dmsetup avaliable, lastly try inbox dmsetup"
211 VT_DM_BIN
=$VTOY_PATH/tool
/dmsetup
214 vtlog
"dmsetup avaliable in system $VT_DM_BIN"
216 if ventoy_check_dm_module
"$1"; then
217 vtlog
"device-mapper module check success"
219 vterr
"Error: no dm module avaliable"
222 $VTOY_PATH/tool
/vtoydm
-p -f $VTOY_PATH/ventoy_image_map
-d $1 > $VTOY_PATH/ventoy_dm_table
224 $VT_DM_BIN create ventoy
$VTOY_PATH/ventoy_dm_table
>>$VTLOG 2>&1
226 $VT_DM_BIN "$2" create ventoy
$VTOY_PATH/ventoy_dm_table
>>$VTLOG 2>&1
230 create_persistent_device_mapper
() {
231 vtlog
"create_persistent_device_mapper $*"
233 VT_DM_BIN
=$(ventoy_find_bin_path dmsetup)
234 if [ -z "$VT_DM_BIN" ]; then
235 vtlog
"no dmsetup avaliable, lastly try inbox dmsetup"
236 VT_DM_BIN
=$VTOY_PATH/tool
/dmsetup
239 vtlog
"dmsetup avaliable in system $VT_DM_BIN"
241 if ventoy_check_dm_module
"$1"; then
242 vtlog
"device-mapper module check success"
244 vterr
"Error: no dm module avaliable"
247 $VTOY_PATH/tool
/vtoydm
-p -f $VTOY_PATH/ventoy_persistent_map
-d $1 > $VTOY_PATH/persistent_dm_table
248 $VT_DM_BIN create vtoy_persistent
$VTOY_PATH/persistent_dm_table
>>$VTLOG 2>&1
253 wait_for_ventoy_dm_disk_label
() {
254 DM
=$($BUSYBOX_PATH/readlink $VTOY_DM_PATH)
255 vtlog
"wait_for_ventoy_dm_disk_label $DM ..."
257 for i
in 0 1 2 3 4 5 6 7 8 9; do
258 vtlog
"i=$i ####### ls /dev/disk/by-label/"
259 ls -l /dev
/disk
/by
-label/ >> $VTLOG
261 if ls -l /dev
/disk
/by
-label/ | $GREP -q "$DM"; then
270 if ! [ -e "$1" ]; then
275 if [ -d /tmp
/vtoy_udeb
]; then
276 $BUSYBOX_PATH/rm -rf /tmp
/vtoy_udeb
279 $BUSYBOX_PATH/mkdir -p /tmp
/vtoy_udeb
280 $BUSYBOX_PATH/cp -a "$1" /tmp
/vtoy_udeb
/
282 CURDIR
=$($BUSYBOX_PATH/pwd)
285 $BUSYBOX_PATH/ar x
"$1"
287 if [ -e 'data.tar.gz' ]; then
288 $BUSYBOX_PATH/tar -xzf data.
tar.gz
-C /
289 elif [ -e 'data.tar.xz' ]; then
290 $BUSYBOX_PATH/tar -xJf data.
tar.xz
-C /
291 elif [ -e 'data.tar.bz2' ]; then
292 $BUSYBOX_PATH/tar -xjf data.
tar.bz2
-C /
293 elif [ -e 'data.tar.lzma' ]; then
294 $BUSYBOX_PATH/tar -xaf data.
tar.lzma
-C /
297 if [ -e 'control.tar.gz' ]; then
298 $BUSYBOX_PATH/tar -xzf control.
tar.gz
-C /
299 elif [ -e 'control.tar.xz' ]; then
300 $BUSYBOX_PATH/tar -xJf control.
tar.xz
-C /
301 elif [ -e 'control.tar.bz2' ]; then
302 $BUSYBOX_PATH/tar -xjf control.
tar.bz2
-C /
303 elif [ -e 'control.tar.lzma' ]; then
304 $BUSYBOX_PATH/tar -xaf control.
tar.lzma
-C /
308 $BUSYBOX_PATH/rm -rf /tmp
/vtoy_udeb
313 install_udeb_from_line
() {
314 vtlog
"install_udeb_from_line $1"
316 if ! [ -b "$2" ]; then
317 vterr
"disk #$2# not exist"
321 sector
=$(echo $1 | $AWK '{print $(NF-1)}')
322 length=$(echo $1 | $AWK '{print $NF}')
323 vtlog "sector=$sector length=$length"
325 $VTOY_PATH/tool/vtoydm -e -f $VTOY_PATH/ventoy_image_map -d ${2} -s $sector -l $length -o /tmp/xxx.udeb
326 if [ -e /tmp/xxx.udeb ]; then
327 vtlog "extract udeb file from iso success"
329 vterr "extract udeb file from iso fail"
333 install_udeb_pkg /tmp/xxx.udeb
334 $BUSYBOX_PATH/rm -f /tmp/xxx.udeb
337 extract_file_from_line() {
338 vtlog "extract_file_from_line $1 disk=#$2#"
339 if ! [ -b "$2" ]; then
340 vterr "disk #$2# not exist"
344 sector=$(echo $1 | $AWK '{print $(NF-1)}')
345 length
=$(echo $1 | $AWK '{print $NF}')
346 vtlog
"sector=$sector length=$length"
348 $VTOY_PATH/tool
/vtoydm
-e -f $VTOY_PATH/ventoy_image_map
-d ${2} -s $sector -l $length -o $3
350 vtlog
"extract file from iso success"
353 vterr
"extract file from iso fail"
358 extract_rpm_from_line
() {
359 vtlog
"extract_rpm_from_line $1 disk=#$2#"
361 if ! [ -b "$2" ]; then
362 vterr
"disk #$2# not exist"
366 sector
=$(echo $1 | $AWK '{print $(NF-1)}')
367 length=$(echo $1 | $AWK '{print $NF}')
368 vtlog "sector=$sector length=$length"
370 $VTOY_PATH/tool/vtoydm -e -f $VTOY_PATH/ventoy_image_map -d ${2} -s $sector -l $length -o /tmp/xxx.rpm
371 if [ -e /tmp/xxx.rpm ]; then
372 vtlog "extract rpm file from iso success"
374 vterr "extract rpm file from iso fail"
378 CURPWD=$($BUSYBOX_PATH/pwd)
380 $BUSYBOX_PATH/mkdir -p $VTOY_PATH/rpm
382 vtlog "extract rpm..."
383 $BUSYBOX_PATH/rpm2cpio /tmp/xxx.rpm | $BUSYBOX_PATH/cpio -idm 2>>$VTLOG
386 $BUSYBOX_PATH/rm -f /tmp/xxx.rpm
389 install_rpm_from_line() {
390 vtlog "install_rpm_from_line $1 disk=#$2#"
392 if ! [ -b "$2" ]; then
393 vterr "disk #$2# not exist"
397 sector=$(echo $1 | $AWK '{print $(NF-1)}')
398 length
=$(echo $1 | $AWK '{print $NF}')
399 vtlog
"sector=$sector length=$length"
401 $VTOY_PATH/tool
/vtoydm
-e -f $VTOY_PATH/ventoy_image_map
-d ${2} -s $sector -l $length -o /tmp
/xxx.rpm
402 if [ -e /tmp
/xxx.rpm
]; then
403 vtlog
"extract rpm file from iso success"
405 vterr
"extract rpm file from iso fail"
409 CURPWD
=$($BUSYBOX_PATH/pwd)
412 vtlog
"install rpm..."
413 $BUSYBOX_PATH/rpm2cpio
/tmp
/xxx.rpm
| $BUSYBOX_PATH/cpio -idm 2>>$VTLOG
416 $BUSYBOX_PATH/rm -f /tmp
/xxx.rpm
419 dump_whole_iso_file
() {
420 $VTOY_PATH/tool
/vtoydm
-p -f $VTOY_PATH/ventoy_image_map
-d $usb_disk | while read vtline
; do
421 vtlog
"dmtable line: $vtline"
422 vtcount
=$(echo $vtline | $AWK '{print $2}')
423 vtoffset
=$(echo $vtline | $AWK '{print $NF}')
424 $BUSYBOX_PATH/dd if=$usb_disk of
="$1" bs
=512 count
=$vtcount skip
=$vtoffset oflag
=append conv
=notrunc
428 ventoy_copy_device_mapper
() {
429 if [ -L $VTOY_DM_PATH ]; then
430 vtlog
"replace block device link $1..."
431 $BUSYBOX_PATH/mv "$1" $VTOY_PATH/dev_backup_
${1#/dev/}
432 VT_MAPPER_LINK
=$($BUSYBOX_PATH/readlink $VTOY_DM_PATH)
433 $BUSYBOX_PATH/cp -a "/dev/mapper/$VT_MAPPER_LINK" "$1"
434 elif [ -b $VTOY_DM_PATH ]; then
435 vtlog
"replace block device $1..."
436 $BUSYBOX_PATH/mv "$1" $VTOY_PATH/dev_backup_
${1#/dev/}
437 $BUSYBOX_PATH/cp -a "$VTOY_DM_PATH" "$1"
440 vtlog
"$VTOY_DM_PATH not exist, now check /dev/dm-X ..."
441 VT_DM_BIN
=$(ventoy_find_bin_path dmsetup)
442 if [ -z "$VT_DM_BIN" ]; then
443 vtlog
"no dmsetup avaliable, lastly try inbox dmsetup"
444 VT_DM_BIN
=$VTOY_PATH/tool
/dmsetup
447 DM_VT_ID
=$($VT_DM_BIN ls | $GREP ventoy | $SED 's/.*(\([0-9][0-9]*\),.
*\
([0-9][0-9]*\
).
*/\
1 \
2/')
448 vtlog "DM_VT_ID=$DM_VT_ID ..."
449 $BUSYBOX_PATH/mv "$1" $VTOY_PATH/dev_backup_${1#/dev/}
450 $BUSYBOX_PATH/mknod -m 0666 "$1" b $DM_VT_ID
454 # create link for device-mapper
455 ventoy_create_persistent_link() {
456 blkdev_num=$($VTOY_PATH/tool/dmsetup ls | grep vtoy_persistent | sed 's/.*(\([0-9][0-9]*\),.*\([0-9][0-9]*\).*/\1:\2/')
457 vtDM
=$(ventoy_find_dm_id ${blkdev_num})
459 if ! [ -d /dev
/disk
/by
-label ]; then
460 mkdir -p /dev
/disk
/by
-label
463 VTLABEL
=$($BUSYBOX_PATH/blkid /dev/$vtDM | $SED 's/.*LABEL="\([^"]*\)".*/\1/')
464 if [ -z "$VTLABEL" ]; then
468 vtlog "Persistent Label
: ##${VTLABEL}##"
470 if ! [ -e /dev
/disk
/by
-label/$VTLABEL ]; then
472 cd /dev
/disk
/by
-label
473 ln -s ..
/..
/$vtDM $VTLABEL
478 ventoy_udev_disk_common_hook
() {
479 if echo $1 | $EGREP -q "nvme.*p[0-9]$|mmc.*p[0-9]$"; then
485 if [ -e /vtoy
/vtoy
]; then
491 # create device mapper for iso image file
492 if create_ventoy_device_mapper
"/dev/$VTDISK" $VTRWMOD; then
493 vtlog
"==== create ventoy device mapper success ===="
495 vtlog
"==== create ventoy device mapper failed ===="
499 if $GREP -q "/dev/$VTDISK" /proc
/mounts
; then
500 $GREP "/dev/$VTDISK" /proc
/mounts
| while read vtLine
; do
501 vtPart
=$(echo $vtLine | $AWK '{print $1}')
502 vtMnt
=$(echo $vtLine | $AWK '{print $2}')
503 vtlog
"$vtPart is mounted on $vtMnt now umount it ..."
504 $BUSYBOX_PATH/umount
$vtMnt
508 if create_ventoy_device_mapper
"/dev/$VTDISK" $VTRWMOD; then
509 vtlog
"==== create ventoy device mapper success after retry ===="
511 vtlog
"==== create ventoy device mapper failed after retry ===="
516 if [ "$2" = "noreplace" ]; then
517 vtlog
"no need to replace block device"
519 ventoy_copy_device_mapper
"/dev/$1"
522 if [ -f $VTOY_PATH/ventoy_persistent_map
]; then
523 create_persistent_device_mapper
"/dev/$VTDISK"
524 ventoy_create_persistent_link
528 ventoy_create_dev_ventoy_part
() {
529 blkdev_num
=$($VTOY_PATH/tool/dmsetup ls | $GREP ventoy | $SED 's/.*(\([0-9][0-9]*\),.
*\
([0-9][0-9]*\
).
*/\
1 \
2/')
530 $BUSYBOX_PATH/mknod -m 0666 /dev/ventoy b $blkdev_num
532 if [ -e /vtoy_dm_table ]; then
535 $CAT /vtoy_dm_table | while read vtline; do
536 echo $vtline > /ventoy/dm_table_part${vtPartid}
537 $VTOY_PATH/tool/dmsetup create ventoy${vtPartid} /ventoy/dm_table_part${vtPartid}
539 blkdev_num=$($VTOY_PATH/tool/dmsetup ls | $GREP ventoy${vtPartid} | $SED 's/.*(\([0-9][0-9]*\),.*\([0-9][0-9]*\).*/\1 \2/')
540 $BUSYBOX_PATH/mknod
-m 0666 /dev
/ventoy
${vtPartid} b
$blkdev_num
542 vtPartid
=$(expr $vtPartid + 1)
547 is_inotify_ventoy_part
() {
548 if echo $1 | $GREP -q "2$"; then
549 if ! [ -e /sys
/block
/$1 ]; then
550 if [ -e /sys
/class
/block
/$1 ]; then
551 if echo $1 | $EGREP -q "nvme|mmc"; then
552 vtShortName
=${1:0:-2}
554 vtShortName
=${1:0:-1}
557 if [ -e /dev
/$vtShortName ]; then
558 $VTOY_PATH/tool
/vtoydump
-f $VTOY_PATH/ventoy_os_param
-c $vtShortName
568 ventoy_find_dm_id
() {
569 for vt
in $($BUSYBOX_PATH/ls /sys/block/); do
570 if [ "${vt:0:3}" = "dm-" ]; then
571 vtMajorMinor
=$($CAT /sys/block/$vt/dev)
572 if [ "$vtMajorMinor" = "$1" ]; then
581 ventoy_swap_device
() {
582 mv $1 $VTOY_PATH/swap_tmp_dev
584 mv $VTOY_PATH/swap_tmp_dev
$2
587 ventoy_extract_vtloopex
() {
589 $BUSYBOX_PATH/mkdir -p $VTOY_PATH/partmnt
$VTOY_PATH/vtloopex
590 $BUSYBOX_PATH/mount
-o ro
-t vfat
$1 $VTOY_PATH/partmnt
591 cd $VTOY_PATH/vtloopex
592 $CAT $VTOY_PATH/partmnt
/ventoy
/vtloopex.
cpio | $BUSYBOX_PATH/cpio -idm >> $VTLOG 2>&1
593 $BUSYBOX_PATH/umount
$VTOY_PATH/partmnt
594 $BUSYBOX_PATH/rm -rf $VTOY_PATH/partmnt
597 cd $VTOY_PATH/vtloopex
/$2/
598 $BUSYBOX_PATH/tar -xJf vtloopex.
tar.xz