Fix a bug when install Ventoy if the USB already mounted at a path that contains...

[Ventoy.git] / LinuxGUI / Ventoy2Disk / main_gtk.c

diff --git a/LinuxGUI/Ventoy2Disk/main_gtk.c b/LinuxGUI/Ventoy2Disk/main_gtk.c
index 4f1d8a0989d1427b955bd54b7a02da650b0713d8..a2b96aa5d055fda0675c1ad33a81d842926a8d9b 100644 (file)
--- a/LinuxGUI/Ventoy2Disk/main_gtk.c
+++ b/LinuxGUI/Ventoy2Disk/main_gtk.c
@@ -15,6 +15,7 @@
 #include <ventoy_util.h>\r
 #include "ventoy_gtk.h"\r
 \r
+static int g_kiosk_mode = 0;\r
 char g_log_file[PATH_MAX];\r
 char g_ini_file[PATH_MAX];\r
 \r
@@ -130,6 +131,15 @@ int main(int argc, char *argv[])
         return 1;\r
     }\r
 \r
+    for (i = 0; i < argc; i++)\r
+    {\r
+        if (argv[i] && strcmp(argv[i], "--kiosk") == 0)\r
+        {\r
+            g_kiosk_mode = 1;\r
+            break;\r
+        }\r
+    }\r
+    \r
     snprintf(g_log_file, sizeof(g_log_file), "log.txt");\r
     snprintf(g_ini_file, sizeof(g_ini_file), "./Ventoy2Disk.ini");\r
     for (i = 0; i < argc; i++)
@@ -179,12 +189,20 @@ int main(int argc, char *argv[])
         return 1;\r
     }\r
 \r
-    pData = get_refresh_icon_raw_data(&len);\r
-    set_image_from_pixbuf(pBuilder, "image_refresh", pData, len);\r
-    \r
-    pData = get_secure_icon_raw_data(&len);\r
-    set_image_from_pixbuf(pBuilder, "image_secure_local", pData, len);\r
-    set_image_from_pixbuf(pBuilder, "image_secure_dev", pData, len);\r
+    if (g_kiosk_mode)\r
+    {\r
+        gtk_image_set_from_file((GtkImage *)gtk_builder_get_object(pBuilder, "image_refresh"), "/ventoy/refresh.png");        \r
+        gtk_image_set_from_file((GtkImage *)gtk_builder_get_object(pBuilder, "image_secure_local"), "/ventoy/secure.png");        \r
+        gtk_image_set_from_file((GtkImage *)gtk_builder_get_object(pBuilder, "image_secure_dev"), "/ventoy/secure.png");        \r
+    }\r
+    else\r
+    {\r
+        pData = get_refresh_icon_raw_data(&len);\r
+        set_image_from_pixbuf(pBuilder, "image_refresh", pData, len);        \r
+        pData = get_secure_icon_raw_data(&len);\r
+        set_image_from_pixbuf(pBuilder, "image_secure_local", pData, len);\r
+        set_image_from_pixbuf(pBuilder, "image_secure_dev", pData, len);\r
+    }\r
 \r
     pWidget = GTK_WIDGET(gtk_builder_get_object(pBuilder, "window"));\r
     gtk_widget_show_all(pWidget);\r