]> glassweightruler.freedombox.rocks Git - Ventoy.git/blobdiff - Vlnk/src/main_windows.c
projects / Ventoy.git / blobdiff
? search:
summary | shortlog | log | commit | commitdiff | tree
raw | inline | side by side
VentoyVlnk add cmdline mode.
[Ventoy.git] / Vlnk / src / main_windows.c
diff --git a/Vlnk/src/main_windows.c b/Vlnk/src/main_windows.c
index 6596c29bec158f9692b0e5d008985265048f77b5..064604058723a847847cb375fc90aaa32382d232 100644 (file)
--- a/Vlnk/src/main_windows.c
+++ b/Vlnk/src/main_windows.c
@@ -13,12 +13,17 @@ static CHAR g_LogFile[MAX_PATH];
 static HWND g_create_button;\r
 static HWND g_parse_button;\r
 \r
+static BOOL g_ShowHelp = FALSE;\r
+static WCHAR g_CmdInFile[MAX_PATH];\r
+static WCHAR g_CmdOutFile[MAX_PATH];\r
+\r
 typedef enum MSGID\r
 {\r
     MSGID_ERROR = 0,\r
        MSGID_INFO,\r
        MSGID_BTN_CREATE,\r
        MSGID_BTN_PARSE,\r
+    MSGID_SRC_NONEXIST,\r
     MSGID_SRC_UNSUPPORTED,\r
     MSGID_FS_UNSUPPORTED,\r
     MSGID_SUFFIX_UNSUPPORTED,\r
@@ -42,6 +47,7 @@ const WCHAR *g_msg_cn[MSGID_BUTT] =
        L"ÌáÐÑ",\r
        L"´´½¨",\r
        L"½âÎö",        \r
+    L"Ö¸¶¨µÄÎļþ²»´æÔÚ", \r
     L"²»Ö§³ÖΪ´ËÎļþ´´½¨vlnk",\r
     L"²»Ö§³ÖµÄÎļþϵͳ",\r
     L"²»Ö§³ÖµÄÎļþºó׺Ãû",\r
@@ -61,6 +67,7 @@ const WCHAR *g_msg_en[MSGID_BUTT] =
        L"Info",   \r
        L"Create",\r
        L"Parse",\r
+    L"The specified file is not exist!",\r
     L"This file is not supported for vlnk",\r
     L"Unsupported file system!", \r
     L"Unsupported file suffix!",\r
@@ -78,6 +85,22 @@ const WCHAR *g_msg_en[MSGID_BUTT] =
 const WCHAR **g_msg_lang = NULL;\r
 \r
 HINSTANCE g_hInst;\r
+\r
+static int VtoyMessageBox\r
+(\r
+    _In_opt_ HWND hWnd,\r
+    _In_opt_ LPCWSTR lpText,\r
+    _In_opt_ LPCWSTR lpCaption,\r
+    _In_ UINT uType\r
+)\r
+{\r
+    if (g_CmdInFile[0] && g_CmdOutFile[0])\r
+    {\r
+        return 0;\r
+    }\r
+\r
+    return MessageBox(hWnd, lpText, lpCaption, uType);\r
+}\r
 
 static void Log2File(const char *log)
 {
@@ -340,7 +363,7 @@ End:
 }\r
 \r
 \r
-static int CreateVlnk(HWND hWnd, WCHAR *Dir)\r
+static int CreateVlnk(HWND hWnd, WCHAR *Dir, WCHAR *InFile, WCHAR *OutFile)\r
 {\r
     int i;\r
     int end;\r
@@ -358,20 +381,27 @@ static int CreateVlnk(HWND hWnd, WCHAR *Dir)
     WCHAR *Pos = NULL;\r
     ventoy_vlnk *vlnk = NULL;\r
 \r
-    ofn.lStructSize = sizeof(ofn);\r
-    ofn.hwndOwner = hWnd;\r
-    ofn.lpstrFile = szFile;\r
-    ofn.nMaxFile = sizeof(szFile);\r
-    ofn.lpstrFilter = L"Vlnk Source File\0*.iso;*.img;*.wim;*.vhd;*.vhdx;*.vtoy;*.efi;*.dat\0";\r
-    ofn.nFilterIndex = 1;\r
-    ofn.lpstrFileTitle = NULL;\r
-    ofn.nMaxFileTitle = 0;\r
-    ofn.lpstrInitialDir = NULL;\r
-    ofn.Flags = OFN_PATHMUSTEXIST | OFN_FILEMUSTEXIST;\r
-\r
-    if (GetOpenFileName(&ofn) != TRUE)\r
+    if (InFile)\r
     {\r
-        return 1;\r
+        wcscpy_s(szFile, MAX_PATH, InFile);\r
+    }\r
+    else\r
+    {\r
+        ofn.lStructSize = sizeof(ofn);\r
+        ofn.hwndOwner = hWnd;\r
+        ofn.lpstrFile = szFile;\r
+        ofn.nMaxFile = sizeof(szFile);\r
+        ofn.lpstrFilter = L"Vlnk Source File\0*.iso;*.img;*.wim;*.vhd;*.vhdx;*.vtoy;*.efi;*.dat\0";\r
+        ofn.nFilterIndex = 1;\r
+        ofn.lpstrFileTitle = NULL;\r
+        ofn.nMaxFileTitle = 0;\r
+        ofn.lpstrInitialDir = NULL;\r
+        ofn.Flags = OFN_PATHMUSTEXIST | OFN_FILEMUSTEXIST;\r
+\r
+        if (GetOpenFileName(&ofn) != TRUE)\r
+        {\r
+            return 1;\r
+        }\r
     }\r
 \r
     LogW(L"Create vlnk for <%ls>\n", szFile);\r
@@ -380,7 +410,7 @@ static int CreateVlnk(HWND hWnd, WCHAR *Dir)
 \r
     if (len < 5 || szFile[0] == '.' || szFile[1] != ':')\r
     {\r
-        MessageBox(hWnd, g_msg_lang[MSGID_SRC_UNSUPPORTED], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR);\r
+        VtoyMessageBox(hWnd, g_msg_lang[MSGID_SRC_UNSUPPORTED], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR);\r
         return 1;\r
     }\r
     \r
@@ -406,7 +436,7 @@ static int CreateVlnk(HWND hWnd, WCHAR *Dir)
     }\r
     else\r
     {\r
-        MessageBox(hWnd, g_msg_lang[MSGID_FS_UNSUPPORTED], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR);\r
+        VtoyMessageBox(hWnd, g_msg_lang[MSGID_FS_UNSUPPORTED], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR);\r
         return 1;\r
     }\r
 \r
@@ -419,13 +449,13 @@ static int CreateVlnk(HWND hWnd, WCHAR *Dir)
 \r
     if (!IsSupportedImgSuffix(suffix))\r
     {\r
-        MessageBox(hWnd, g_msg_lang[MSGID_SUFFIX_UNSUPPORTED], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR);\r
+        VtoyMessageBox(hWnd, g_msg_lang[MSGID_SUFFIX_UNSUPPORTED], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR);\r
         return 1;\r
     }\r
 \r
     if (IsVlnkFile(szFile, NULL))\r
     {\r
-        MessageBox(hWnd, g_msg_lang[MSGID_ALREADY_VLNK], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR);\r
+        VtoyMessageBox(hWnd, g_msg_lang[MSGID_ALREADY_VLNK], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR);\r
         return 1;\r
     }\r
 \r
@@ -459,14 +489,14 @@ static int CreateVlnk(HWND hWnd, WCHAR *Dir)
     if (len >= VLNK_NAME_MAX)\r
     {\r
         LogA("File name length %d overflow\n", len);\r
-        MessageBox(hWnd, g_msg_lang[MSGID_FILE_NAME_TOO_LONG], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR);\r
+        VtoyMessageBox(hWnd, g_msg_lang[MSGID_FILE_NAME_TOO_LONG], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR);\r
         return 1;\r
     }\r
 \r
     DiskExtend.StartingOffset.QuadPart = 0;\r
     if (GetPhyDiskInfo((char)szFile[0], &DiskSig, &DiskExtend))\r
     {\r
-        MessageBox(hWnd, g_msg_lang[MSGID_DISK_INFO_ERR], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR);\r
+        VtoyMessageBox(hWnd, g_msg_lang[MSGID_DISK_INFO_ERR], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR);\r
         return 1;\r
     }\r
 \r
@@ -477,22 +507,38 @@ static int CreateVlnk(HWND hWnd, WCHAR *Dir)
         vlnk = (ventoy_vlnk *)Buf;\r
         ventoy_create_vlnk(DiskSig, (uint64_t)DiskExtend.StartingOffset.QuadPart, UTF8Path, vlnk);\r
 \r
-        DefaultVlnkDstFullPath(Pos + 1, Dir, DstFullPath);\r
+        if (OutFile)\r
+        {\r
+            wcscpy_s(DstFullPath, MAX_PATH, OutFile);\r
+        }\r
+        else\r
+        {\r
+            DefaultVlnkDstFullPath(Pos + 1, Dir, DstFullPath);\r
+        }\r
+\r
         LogW(L"vlnk output file path is <%ls>\n", DstFullPath);\r
 \r
         if (SaveBuffer2File(DstFullPath, Buf, VLNK_FILE_LEN) == 0)\r
         {\r
             WCHAR Msg[1024];\r
 \r
-            swprintf_s(Msg, 1024, L"%ls\r\n\r\n%ls", g_msg_lang[MSGID_VLNK_SUCCESS], DstFullPath + lstrlen(Dir) + 1);\r
-\r
             LogW(L"Vlnk file create success <%ls>\n", DstFullPath);\r
-            MessageBox(hWnd, Msg, g_msg_lang[MSGID_INFO], MB_OK | MB_ICONINFORMATION);\r
+\r
+            if (OutFile)\r
+            {\r
+                swprintf_s(Msg, 1024, L"%ls\r\n\r\n%ls", g_msg_lang[MSGID_VLNK_SUCCESS], DstFullPath);\r
+                VtoyMessageBox(hWnd, Msg, g_msg_lang[MSGID_INFO], MB_OK | MB_ICONINFORMATION);\r
+            }\r
+            else\r
+            {\r
+                swprintf_s(Msg, 1024, L"%ls\r\n\r\n%ls", g_msg_lang[MSGID_VLNK_SUCCESS], DstFullPath + lstrlen(Dir) + 1);\r
+                VtoyMessageBox(hWnd, Msg, g_msg_lang[MSGID_INFO], MB_OK | MB_ICONINFORMATION);\r
+            }\r
         }\r
         else\r
         {\r
             LogA("Vlnk file save failed\n");\r
-            MessageBox(hWnd, g_msg_lang[MSGID_CREATE_FILE_ERR], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR);\r
+            VtoyMessageBox(hWnd, g_msg_lang[MSGID_CREATE_FILE_ERR], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR);\r
         }\r
 \r
         free(Buf);\r
@@ -567,7 +613,7 @@ static int ParseVlnk(HWND hWnd)
 \r
     if (!IsVlnkFile(szFile, &vlnk))\r
     {\r
-        MessageBox(hWnd, g_msg_lang[MSGID_INVALID_VLNK], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR);\r
+        VtoyMessageBox(hWnd, g_msg_lang[MSGID_INVALID_VLNK], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR);\r
         return 1;\r
     }\r
 \r
@@ -581,7 +627,7 @@ static int ParseVlnk(HWND hWnd)
 \r
     if (!IsSupportedImgSuffix(suffix))\r
     {\r
-        MessageBox(hWnd, g_msg_lang[MSGID_SUFFIX_UNSUPPORTED], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR);\r
+        VtoyMessageBox(hWnd, g_msg_lang[MSGID_SUFFIX_UNSUPPORTED], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR);\r
         return 1;\r
     }\r
 \r
@@ -598,7 +644,7 @@ static int ParseVlnk(HWND hWnd)
     Letter = GetDriveLetter(vlnk.disk_signature, vlnk.part_offset);\r
     if (Letter == 0)\r
     {\r
-        MessageBox(hWnd, g_msg_lang[MSGID_VLNK_NO_DST], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR);\r
+        VtoyMessageBox(hWnd, g_msg_lang[MSGID_VLNK_NO_DST], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR);\r
         return 1;\r
     }\r
 \r
@@ -609,13 +655,13 @@ static int ParseVlnk(HWND hWnd)
     hFile = CreateFileW(szDst, FILE_READ_EA, FILE_SHARE_READ, 0, OPEN_EXISTING, 0, 0);\r
     if (INVALID_HANDLE_VALUE == hFile)\r
     {\r
-        MessageBox(hWnd, g_msg_lang[MSGID_VLNK_NO_DST], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR);\r
+        VtoyMessageBox(hWnd, g_msg_lang[MSGID_VLNK_NO_DST], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR);\r
         return 1;\r
     }\r
     CloseHandle(hFile);\r
 \r
     swprintf_s(Msg, 1024, L"%ls %ls", g_msg_lang[MSGID_VLNK_POINT_TO], szDst);\r
-    MessageBox(hWnd, Msg, g_msg_lang[MSGID_INFO], MB_OK | MB_ICONINFORMATION);\r
+    VtoyMessageBox(hWnd, Msg, g_msg_lang[MSGID_INFO], MB_OK | MB_ICONINFORMATION);\r
 \r
     return 0;\r
 }\r
@@ -637,7 +683,7 @@ INT_PTR CALLBACK DialogProc(HWND hWnd, UINT Message, WPARAM wParam, LPARAM lPara
                                if (CtrlID == IDC_BUTTON1)\r
                                {\r
                     EnableWindow(g_create_button, FALSE);\r
-                    CreateVlnk(hWnd, g_CurDirW);\r
+                    CreateVlnk(hWnd, g_CurDirW, NULL, NULL);\r
                     EnableWindow(g_create_button, TRUE);\r
                                }\r
                                else if (CtrlID == IDC_BUTTON2)\r
@@ -664,9 +710,46 @@ INT_PTR CALLBACK DialogProc(HWND hWnd, UINT Message, WPARAM wParam, LPARAM lPara
     return 0;\r
 }\r
 \r
-int APIENTRY WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine, INT nCmdShow)\r
+static int ParseCmdLine(LPSTR lpCmdLine)\r
 {\r
     int i;\r
+    int argc = 0;\r
+    LPWSTR *lpszArgv = NULL;\r
+    \r
+    lpszArgv = CommandLineToArgvW(GetCommandLineW(), &argc);\r
+\r
+    for (i = 0; i < argc; i++)\r
+    {\r
+        if (lstrcmp(lpszArgv[i], L"-q") == 0 || lstrcmp(lpszArgv[i], L"-Q") == 0)\r
+        {\r
+            g_LogFile[0] = 0;\r
+        }\r
+        else if (lstrcmp(lpszArgv[i], L"-h") == 0 || lstrcmp(lpszArgv[i], L"-H") == 0)\r
+        {\r
+            g_ShowHelp = TRUE;\r
+        }\r
+        else if (lstrcmp(lpszArgv[i], L"-i") == 0 || lstrcmp(lpszArgv[i], L"-I") == 0)\r
+        {\r
+            if (i + 1 < argc)\r
+            {\r
+                wcscpy_s(g_CmdInFile, MAX_PATH, lpszArgv[i + 1]);\r
+            }\r
+        }\r
+        else if (lstrcmp(lpszArgv[i], L"-o") == 0 || lstrcmp(lpszArgv[i], L"-O") == 0)\r
+        {\r
+            if (i + 1 < argc)\r
+            {\r
+                wcscpy_s(g_CmdOutFile, MAX_PATH, lpszArgv[i + 1]);\r
+            }\r
+        }\r
+    }\r
+\r
+    return argc;\r
+}\r
+\r
+int APIENTRY WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine, INT nCmdShow)\r
+{\r
+    DWORD dwAttrib;\r
        HANDLE hMutex;\r
 \r
     UNREFERENCED_PARAMETER(hPrevInstance);\r
@@ -680,32 +763,45 @@ int APIENTRY WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLi
         g_msg_lang = g_msg_en;\r
     }\r
 \r
-       hMutex = CreateMutexA(NULL, TRUE, "VtoyVlnkMUTEX");\r
-       if ((hMutex != NULL) && (GetLastError() == ERROR_ALREADY_EXISTS))\r
-       {\r
-               MessageBoxW(NULL, g_msg_lang[MSGID_RUNNING_TIP], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR);\r
-               return 1;\r
-       }\r
+    hMutex = CreateMutexA(NULL, TRUE, "VtoyVlnkMUTEX");\r
+    if ((hMutex != NULL) && (GetLastError() == ERROR_ALREADY_EXISTS))\r
+    {\r
+        MessageBoxW(NULL, g_msg_lang[MSGID_RUNNING_TIP], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR);\r
+        return 1;\r
+    }\r
 \r
     GetCurrentDirectoryA(MAX_PATH, g_CurDirA);\r
     GetCurrentDirectoryW(MAX_PATH, g_CurDirW);\r
     sprintf_s(g_LogFile, sizeof(g_LogFile), "%s\\VentoyVlnk.log", g_CurDirA);\r
 \r
-    for (i = 0; i < __argc; i++)\r
+    ParseCmdLine(lpCmdLine);\r
+    \r
+    g_hInst = hInstance;\r
+\r
+    if (g_ShowHelp)\r
     {\r
-        if (strncmp(__argv[i], "-Q", 2) == 0 ||\r
-            strncmp(__argv[i], "-q", 2) == 0)\r
-        {\r
-            g_LogFile[0] = 0;\r
-            break;\r
-        }\r
+        VtoyMessageBox(NULL, L"VentoyVlnk.exe  CMD\r\n  -i  Input file path\r\n  -o  Output vlnk file path\r\n  -q  Quite mode (no log)", L"Tip", MB_OK);\r
+        return 0;\r
     }\r
-    \r
+    else if (g_CmdInFile[0] && g_CmdOutFile[0])\r
+    {\r
+        LogA("========= VentoyVlnk Cmdline Mode =========\n");\r
 \r
-    LogA("========= VentoyVlnk =========\n");\r
+        dwAttrib = GetFileAttributesW(g_CmdInFile);\r
+        if (dwAttrib == INVALID_FILE_ATTRIBUTES || (dwAttrib & FILE_ATTRIBUTE_DIRECTORY))\r
+        {\r
+            LogW(L"File <<%ls>> does not exist!\n", g_CmdInFile);\r
+            VtoyMessageBox(NULL, g_msg_lang[MSGID_SRC_NONEXIST], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR);\r
+            return 1;\r
+        }\r
 \r
-    g_hInst = hInstance;\r
-    DialogBoxA(hInstance, MAKEINTRESOURCEA(IDD_DIALOG1), NULL, DialogProc);\r
+        return CreateVlnk(NULL, g_CurDirW, g_CmdInFile, g_CmdOutFile);\r
+    }\r
+    else\r
+    {\r
+        LogA("========= VentoyVlnk GUI Mode =========\n");\r
 \r
-    return 0;\r
+        DialogBoxA(hInstance, MAKEINTRESOURCEA(IDD_DIALOG1), NULL, DialogProc);\r
+        return 0;\r
+    }\r
 }\r
https://github.com/ventoy/Ventoy