support for 360Udisk

[Ventoy.git] / LiveCD / VTOY / ventoy / ventoy.sh

diff --git a/LiveCD/VTOY/ventoy/ventoy.sh b/LiveCD/VTOY/ventoy/ventoy.sh
index 63233797e52077ebf81a2f7d55c47f5a7233f80f..baeb1a9b638550bee12d465302144d52cdd1aa37 100644 (file)
--- a/LiveCD/VTOY/ventoy/ventoy.sh
+++ b/LiveCD/VTOY/ventoy/ventoy.sh
@@ -20,7 +20,7 @@ get_disk_size() {
 enum_disk() {
     id=1
     rm -f /device.list
-    ls /sys/block/ | grep 'd[a-z]' | while read dev; do
+    ls /sys/block/ | egrep 'd[a-z]|nvme|mmc' | while read dev; do
         if ! [ -b /dev/$dev ]; then
             continue
         fi
@@ -145,7 +145,7 @@ ventoy_configuration() {
         elif [ "$Answer" = "1" ]; then            
             while true; do
                 read -p "Please input the preserve space in MB: "  Answer
-                if echo $Answer | grep -q "^[1-9][0-9]*$"; then
+                if echo $Answer | grep -q "^[0-9][0-9]*$"; then
                     echo $Answer > /preserve.txt
                     break
                 fi