X-Git-Url: https://glassweightruler.freedombox.rocks/gitweb/Ventoy.git/blobdiff_plain/5291d66736e2fbbea04a5496b960fa4cb3599902..HEAD:/Vlnk/src/main_windows.c?ds=sidebyside diff --git a/Vlnk/src/main_windows.c b/Vlnk/src/main_windows.c index 6596c29..bf9f8af 100644 --- a/Vlnk/src/main_windows.c +++ b/Vlnk/src/main_windows.c @@ -1,4 +1,4 @@ -#include +#include #include #include #include @@ -13,12 +13,18 @@ static CHAR g_LogFile[MAX_PATH]; static HWND g_create_button; static HWND g_parse_button; +static BOOL g_ShowHelp = FALSE; +static BOOL g_SaveAs = FALSE; +static WCHAR g_CmdInFile[MAX_PATH]; +static WCHAR g_CmdOutFile[MAX_PATH]; + typedef enum MSGID { MSGID_ERROR = 0, MSGID_INFO, MSGID_BTN_CREATE, MSGID_BTN_PARSE, + MSGID_SRC_NONEXIST, MSGID_SRC_UNSUPPORTED, MSGID_FS_UNSUPPORTED, MSGID_SUFFIX_UNSUPPORTED, @@ -31,6 +37,7 @@ typedef enum MSGID MSGID_VLNK_POINT_TO, MSGID_VLNK_NO_DST, MSGID_FILE_NAME_TOO_LONG, + MSGID_INVALID_SUFFIX, MSGID_BUTT }MSGID; @@ -38,22 +45,24 @@ typedef enum MSGID const WCHAR *g_msg_cn[MSGID_BUTT] = { - L"´íÎó", - L"ÌáÐÑ", - L"´´½¨", - L"½âÎö", - L"²»Ö§³ÖΪ´ËÎļþ´´½¨vlnk", - L"²»Ö§³ÖµÄÎļþϵͳ", - L"²»Ö§³ÖµÄÎļþºó׺Ãû", - L"»ñÈ¡´ÅÅÌÐÅϢʱ·¢Éú´íÎó", - L"Vlnk Îļþ´´½¨³É¹¦¡£", - L"ÇëÏȹرÕÕýÔÚÔËÐÐµÄ VentoyVlnk ³ÌÐò£¡", - L"´´½¨Îļþʧ°Ü", - L"´ËÎļþÒѾ­ÊÇÒ»¸övlnkÎļþÁË£¡", - L"·Ç·¨µÄvlnkÎļþ!", - L"´Ë vlnk ÎļþÖ¸Ïò ", - L"´Ë vlnk Ö¸ÏòµÄÎļþ²»´æÔÚ£¡", - L"Îļþ·¾¶Ì«³¤£¡", + L"错误", + L"提醒", + L"创建", + L"解析", + L"指定的文件不存在", + L"不支持为此文件创建vlnk", + L"不支持的文件系统", + L"不支持的文件后缀名", + L"获取磁盘信息时发生错误", + L"Vlnk 文件创建成功。", + L"请先关闭正在运行的 VentoyVlnk 程序！", + L"创建文件失败", + L"此文件已经是一个vlnk文件了！", + L"非法的vlnk文件!", + L"æ­¤ vlnk 文件指向 ", + L"æ­¤ vlnk 指向的文件不存在！", + L"文件路径太长！", + L"非法的vlnk文件后缀名!", }; const WCHAR *g_msg_en[MSGID_BUTT] = { @@ -61,6 +70,7 @@ const WCHAR *g_msg_en[MSGID_BUTT] = L"Info", L"Create", L"Parse", + L"The specified file is not exist!", L"This file is not supported for vlnk", L"Unsupported file system!", L"Unsupported file suffix!", @@ -73,67 +83,84 @@ const WCHAR *g_msg_en[MSGID_BUTT] = L"The vlnk file point to ", L"The file pointed by the vlnk does NOT exist!", L"The file full path is too long!", + L"Invalid vlnk file suffix!", }; const WCHAR **g_msg_lang = NULL; HINSTANCE g_hInst; - -static void Log2File(const char *log) -{ - time_t stamp; - struct tm ttm; - FILE *fp; - - time(&stamp); - localtime_s(&ttm, &stamp); - - fopen_s(&fp, g_LogFile, "a+"); - if (fp) - { - fprintf_s(fp, "[%04u/%02u/%02u %02u:%02u:%02u] %s", - ttm.tm_year + 1900, ttm.tm_mon + 1, ttm.tm_mday, - ttm.tm_hour, ttm.tm_min, ttm.tm_sec, log); - fclose(fp); - } -} - -void LogW(const WCHAR *Fmt, ...) -{ - WCHAR log[512]; - CHAR alog[2048]; - va_list arg; - - if (g_LogFile[0] == 0) - { - return; - } - - va_start(arg, Fmt); - vswprintf_s(log, 512, Fmt, arg); - va_end(arg); - - WideCharToMultiByte(CP_UTF8, 0, log, -1, alog, 2048, 0, 0); - - Log2File(alog); + +static int VtoyMessageBox +( + _In_opt_ HWND hWnd, + _In_opt_ LPCWSTR lpText, + _In_opt_ LPCWSTR lpCaption, + _In_ UINT uType +) +{ + if (g_CmdInFile[0] && g_CmdOutFile[0]) + { + return 0; + } + + return MessageBox(hWnd, lpText, lpCaption, uType); +} + +static void Log2File(const char *log) +{ + time_t stamp; + struct tm ttm; + FILE *fp; + + time(&stamp); + localtime_s(&ttm, &stamp); + + fopen_s(&fp, g_LogFile, "a+"); + if (fp) + { + fprintf_s(fp, "[%04u/%02u/%02u %02u:%02u:%02u] %s", + ttm.tm_year + 1900, ttm.tm_mon + 1, ttm.tm_mday, + ttm.tm_hour, ttm.tm_min, ttm.tm_sec, log); + fclose(fp); + } } - -void LogA(const CHAR *Fmt, ...) -{ - CHAR log[512]; - va_list arg; - - if (g_LogFile[0] == 0) - { - return; - } - - va_start(arg, Fmt); - vsprintf_s(log, 512, Fmt, arg); - va_end(arg); - - Log2File(log); +void LogW(const WCHAR *Fmt, ...) +{ + WCHAR log[512]; + CHAR alog[2048]; + va_list arg; + + if (g_LogFile[0] == 0) + { + return; + } + + va_start(arg, Fmt); + vswprintf_s(log, 512, Fmt, arg); + va_end(arg); + + WideCharToMultiByte(CP_UTF8, 0, log, -1, alog, 2048, 0, 0); + + Log2File(alog); +} + + +void LogA(const CHAR *Fmt, ...) +{ + CHAR log[512]; + va_list arg; + + if (g_LogFile[0] == 0) + { + return; + } + + va_start(arg, Fmt); + vsprintf_s(log, 512, Fmt, arg); + va_end(arg); + + Log2File(log); } static int Utf8ToUtf16(const char* src, WCHAR * dst) @@ -340,11 +367,62 @@ End: } -static int CreateVlnk(HWND hWnd, WCHAR *Dir) +static BOOL VentoyGetSaveFileName(HWND hWnd, WCHAR *szFile) +{ + OPENFILENAME ofn = { 0 }; + + ofn.lStructSize = sizeof(ofn); + ofn.hwndOwner = hWnd; + ofn.lpstrFilter = L"Vlnk File\0*.vlnk.iso;*.vlnk.img;*.vlnk.wim;*.vlnk.efi;*.vlnk.vhd;*.vlnk.vhdx;*.vlnk.vtoy;*.vlnk.dat\0"; + ofn.nFilterIndex = 1; + ofn.lpstrFile = szFile; + ofn.nMaxFile = MAX_PATH; + ofn.Flags = OFN_EXPLORER | OFN_PATHMUSTEXIST | OFN_HIDEREADONLY | OFN_OVERWRITEPROMPT; + ofn.lpstrFileTitle = NULL; + ofn.nMaxFileTitle = 0; + ofn.lpstrInitialDir = NULL; + + return GetSaveFileName(&ofn); +} + +static BOOL IsSupportedVlnkSuffix(WCHAR *FileName) +{ + int len; + + len = lstrlen(FileName); + + if (len > 9) + { + if (lstrcmp(FileName + len - 9, L".vlnk.iso") == 0 || + lstrcmp(FileName + len - 9, L".vlnk.img") == 0 || + lstrcmp(FileName + len - 9, L".vlnk.wim") == 0 || + lstrcmp(FileName + len - 9, L".vlnk.vhd") == 0 || + lstrcmp(FileName + len - 9, L".vlnk.efi") == 0 || + lstrcmp(FileName + len - 9, L".vlnk.dat") == 0) + { + return TRUE; + } + } + + + if (len > 10) + { + if (lstrcmp(FileName + len - 10, L".vlnk.vhdx") == 0 || + lstrcmp(FileName + len - 10, L".vlnk.vtoy") == 0) + { + return TRUE; + } + } + + return FALSE; +} + +static int CreateVlnk(HWND hWnd, WCHAR *Dir, WCHAR *InFile, WCHAR *OutFile) { int i; int end; int len; + BOOL SetOutFile = FALSE; UINT32 DiskSig; DISK_EXTENT DiskExtend; OPENFILENAME ofn = { 0 }; @@ -358,20 +436,27 @@ static int CreateVlnk(HWND hWnd, WCHAR *Dir) WCHAR *Pos = NULL; ventoy_vlnk *vlnk = NULL; - ofn.lStructSize = sizeof(ofn); - ofn.hwndOwner = hWnd; - ofn.lpstrFile = szFile; - ofn.nMaxFile = sizeof(szFile); - ofn.lpstrFilter = L"Vlnk Source File\0*.iso;*.img;*.wim;*.vhd;*.vhdx;*.vtoy;*.efi;*.dat\0"; - ofn.nFilterIndex = 1; - ofn.lpstrFileTitle = NULL; - ofn.nMaxFileTitle = 0; - ofn.lpstrInitialDir = NULL; - ofn.Flags = OFN_PATHMUSTEXIST | OFN_FILEMUSTEXIST; - - if (GetOpenFileName(&ofn) != TRUE) + if (InFile) { - return 1; + wcscpy_s(szFile, MAX_PATH, InFile); + } + else + { + ofn.lStructSize = sizeof(ofn); + ofn.hwndOwner = hWnd; + ofn.lpstrFile = szFile; + ofn.nMaxFile = sizeof(szFile); + ofn.lpstrFilter = L"Vlnk Source File\0*.iso;*.img;*.wim;*.vhd;*.vhdx;*.vtoy;*.efi;*.dat\0"; + ofn.nFilterIndex = 1; + ofn.lpstrFileTitle = NULL; + ofn.nMaxFileTitle = 0; + ofn.lpstrInitialDir = NULL; + ofn.Flags = OFN_PATHMUSTEXIST | OFN_FILEMUSTEXIST; + + if (GetOpenFileName(&ofn) != TRUE) + { + return 1; + } } LogW(L"Create vlnk for <%ls>\n", szFile); @@ -380,7 +465,7 @@ static int CreateVlnk(HWND hWnd, WCHAR *Dir) if (len < 5 || szFile[0] == '.' || szFile[1] != ':') { - MessageBox(hWnd, g_msg_lang[MSGID_SRC_UNSUPPORTED], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR); + VtoyMessageBox(hWnd, g_msg_lang[MSGID_SRC_UNSUPPORTED], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR); return 1; } @@ -406,7 +491,7 @@ static int CreateVlnk(HWND hWnd, WCHAR *Dir) } else { - MessageBox(hWnd, g_msg_lang[MSGID_FS_UNSUPPORTED], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR); + VtoyMessageBox(hWnd, g_msg_lang[MSGID_FS_UNSUPPORTED], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR); return 1; } @@ -419,13 +504,13 @@ static int CreateVlnk(HWND hWnd, WCHAR *Dir) if (!IsSupportedImgSuffix(suffix)) { - MessageBox(hWnd, g_msg_lang[MSGID_SUFFIX_UNSUPPORTED], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR); + VtoyMessageBox(hWnd, g_msg_lang[MSGID_SUFFIX_UNSUPPORTED], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR); return 1; } if (IsVlnkFile(szFile, NULL)) { - MessageBox(hWnd, g_msg_lang[MSGID_ALREADY_VLNK], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR); + VtoyMessageBox(hWnd, g_msg_lang[MSGID_ALREADY_VLNK], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR); return 1; } @@ -459,17 +544,18 @@ static int CreateVlnk(HWND hWnd, WCHAR *Dir) if (len >= VLNK_NAME_MAX) { LogA("File name length %d overflow\n", len); - MessageBox(hWnd, g_msg_lang[MSGID_FILE_NAME_TOO_LONG], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR); + VtoyMessageBox(hWnd, g_msg_lang[MSGID_FILE_NAME_TOO_LONG], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR); return 1; } DiskExtend.StartingOffset.QuadPart = 0; if (GetPhyDiskInfo((char)szFile[0], &DiskSig, &DiskExtend)) { - MessageBox(hWnd, g_msg_lang[MSGID_DISK_INFO_ERR], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR); + VtoyMessageBox(hWnd, g_msg_lang[MSGID_DISK_INFO_ERR], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR); return 1; } + Buf = malloc(VLNK_FILE_LEN); if (Buf) { @@ -477,24 +563,65 @@ static int CreateVlnk(HWND hWnd, WCHAR *Dir) vlnk = (ventoy_vlnk *)Buf; ventoy_create_vlnk(DiskSig, (uint64_t)DiskExtend.StartingOffset.QuadPart, UTF8Path, vlnk); - DefaultVlnkDstFullPath(Pos + 1, Dir, DstFullPath); + if (OutFile) + { + wcscpy_s(DstFullPath, MAX_PATH, OutFile); + SetOutFile = TRUE; + } + else + { + DefaultVlnkDstFullPath(Pos + 1, Dir, DstFullPath); + + if (g_SaveAs) + { + wcscpy_s(szFile, MAX_PATH, DstFullPath); + if (VentoyGetSaveFileName(hWnd, szFile)) + { + if (IsSupportedVlnkSuffix(szFile)) + { + wcscpy_s(DstFullPath, MAX_PATH, szFile); + SetOutFile = TRUE; + } + else + { + VtoyMessageBox(hWnd, g_msg_lang[MSGID_INVALID_SUFFIX], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR); + LogA("Invalid vlnk suffix\n"); + goto end; + } + } + else + { + LogA("User cancel the save as diaglog, use default name\n"); + } + } + } + LogW(L"vlnk output file path is <%ls>\n", DstFullPath); if (SaveBuffer2File(DstFullPath, Buf, VLNK_FILE_LEN) == 0) { WCHAR Msg[1024]; - swprintf_s(Msg, 1024, L"%ls\r\n\r\n%ls", g_msg_lang[MSGID_VLNK_SUCCESS], DstFullPath + lstrlen(Dir) + 1); - LogW(L"Vlnk file create success <%ls>\n", DstFullPath); - MessageBox(hWnd, Msg, g_msg_lang[MSGID_INFO], MB_OK | MB_ICONINFORMATION); + + if (SetOutFile) + { + swprintf_s(Msg, 1024, L"%ls\r\n\r\n%ls", g_msg_lang[MSGID_VLNK_SUCCESS], DstFullPath); + VtoyMessageBox(hWnd, Msg, g_msg_lang[MSGID_INFO], MB_OK | MB_ICONINFORMATION); + } + else + { + swprintf_s(Msg, 1024, L"%ls\r\n\r\n%ls", g_msg_lang[MSGID_VLNK_SUCCESS], DstFullPath + lstrlen(Dir) + 1); + VtoyMessageBox(hWnd, Msg, g_msg_lang[MSGID_INFO], MB_OK | MB_ICONINFORMATION); + } } else { LogA("Vlnk file save failed\n"); - MessageBox(hWnd, g_msg_lang[MSGID_CREATE_FILE_ERR], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR); + VtoyMessageBox(hWnd, g_msg_lang[MSGID_CREATE_FILE_ERR], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR); } + end: free(Buf); } @@ -534,7 +661,6 @@ static CHAR GetDriveLetter(UINT32 disksig, UINT64 PartOffset) return 0; } - static int ParseVlnk(HWND hWnd) { int i; @@ -567,7 +693,7 @@ static int ParseVlnk(HWND hWnd) if (!IsVlnkFile(szFile, &vlnk)) { - MessageBox(hWnd, g_msg_lang[MSGID_INVALID_VLNK], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR); + VtoyMessageBox(hWnd, g_msg_lang[MSGID_INVALID_VLNK], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR); return 1; } @@ -581,7 +707,7 @@ static int ParseVlnk(HWND hWnd) if (!IsSupportedImgSuffix(suffix)) { - MessageBox(hWnd, g_msg_lang[MSGID_SUFFIX_UNSUPPORTED], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR); + VtoyMessageBox(hWnd, g_msg_lang[MSGID_SUFFIX_UNSUPPORTED], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR); return 1; } @@ -598,7 +724,7 @@ static int ParseVlnk(HWND hWnd) Letter = GetDriveLetter(vlnk.disk_signature, vlnk.part_offset); if (Letter == 0) { - MessageBox(hWnd, g_msg_lang[MSGID_VLNK_NO_DST], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR); + VtoyMessageBox(hWnd, g_msg_lang[MSGID_VLNK_NO_DST], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR); return 1; } @@ -609,13 +735,13 @@ static int ParseVlnk(HWND hWnd) hFile = CreateFileW(szDst, FILE_READ_EA, FILE_SHARE_READ, 0, OPEN_EXISTING, 0, 0); if (INVALID_HANDLE_VALUE == hFile) { - MessageBox(hWnd, g_msg_lang[MSGID_VLNK_NO_DST], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR); + VtoyMessageBox(hWnd, g_msg_lang[MSGID_VLNK_NO_DST], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR); return 1; } CloseHandle(hFile); swprintf_s(Msg, 1024, L"%ls %ls", g_msg_lang[MSGID_VLNK_POINT_TO], szDst); - MessageBox(hWnd, Msg, g_msg_lang[MSGID_INFO], MB_OK | MB_ICONINFORMATION); + VtoyMessageBox(hWnd, Msg, g_msg_lang[MSGID_INFO], MB_OK | MB_ICONINFORMATION); return 0; } @@ -637,7 +763,7 @@ INT_PTR CALLBACK DialogProc(HWND hWnd, UINT Message, WPARAM wParam, LPARAM lPara if (CtrlID == IDC_BUTTON1) { EnableWindow(g_create_button, FALSE); - CreateVlnk(hWnd, g_CurDirW); + CreateVlnk(hWnd, g_CurDirW, NULL, NULL); EnableWindow(g_create_button, TRUE); } else if (CtrlID == IDC_BUTTON2) @@ -664,13 +790,109 @@ INT_PTR CALLBACK DialogProc(HWND hWnd, UINT Message, WPARAM wParam, LPARAM lPara return 0; } -int APIENTRY WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine, INT nCmdShow) +static int ParseCmdLine(LPSTR lpCmdLine) { int i; + int argc = 0; + LPWSTR *lpszArgv = NULL; + + lpszArgv = CommandLineToArgvW(GetCommandLineW(), &argc); + + for (i = 0; i < argc; i++) + { + if (lstrcmp(lpszArgv[i], L"-q") == 0 || lstrcmp(lpszArgv[i], L"-Q") == 0) + { + g_LogFile[0] = 0; + } + else if (lstrcmp(lpszArgv[i], L"-h") == 0 || lstrcmp(lpszArgv[i], L"-H") == 0) + { + g_ShowHelp = TRUE; + } + else if (lstrcmp(lpszArgv[i], L"-s") == 0 || lstrcmp(lpszArgv[i], L"-S") == 0) + { + g_SaveAs = TRUE; + } + else if (lstrcmp(lpszArgv[i], L"-i") == 0 || lstrcmp(lpszArgv[i], L"-I") == 0) + { + if (i + 1 < argc) + { + wcscpy_s(g_CmdInFile, MAX_PATH, lpszArgv[i + 1]); + } + } + else if (lstrcmp(lpszArgv[i], L"-o") == 0 || lstrcmp(lpszArgv[i], L"-O") == 0) + { + if (i + 1 < argc) + { + wcscpy_s(g_CmdOutFile, MAX_PATH, lpszArgv[i + 1]); + } + } + } + + return argc; +} + + +// +//copy from Rufus +//Copyright © 2011-2021 Pete Batard +// +#include +// For delay-loaded DLLs, use LOAD_LIBRARY_SEARCH_SYSTEM32 to avoid DLL search order hijacking. +FARPROC WINAPI dllDelayLoadHook(unsigned dliNotify, PDelayLoadInfo pdli) +{ + if (dliNotify == dliNotePreLoadLibrary) { + // Windows 7 without KB2533623 does not support the LOAD_LIBRARY_SEARCH_SYSTEM32 flag. + // That is is OK, because the delay load handler will interrupt the NULL return value + // to mean that it should perform a normal LoadLibrary. + return (FARPROC)LoadLibraryExA(pdli->szDll, NULL, LOAD_LIBRARY_SEARCH_SYSTEM32); + } + return NULL; +} + +#if defined(_MSC_VER) +// By default the Windows SDK headers have a `const` while MinGW does not. +const +#endif +PfnDliHook __pfnDliNotifyHook2 = dllDelayLoadHook; + +typedef BOOL(WINAPI *SetDefaultDllDirectories_t)(DWORD); +static void DllProtect(void) +{ + SetDefaultDllDirectories_t pfSetDefaultDllDirectories = NULL; + + // Disable loading system DLLs from the current directory (sideloading mitigation) + // PS: You know that official MSDN documentation for SetDllDirectory() that explicitly + // indicates that "If the parameter is an empty string (""), the call removes the current + // directory from the default DLL search order"? Yeah, that doesn't work. At all. + // Still, we invoke it, for platforms where the following call might actually work... + SetDllDirectoryA(""); + + // For libraries on the KnownDLLs list, the system will always load them from System32. + // For other DLLs we link directly to, we can delay load the DLL and use a delay load + // hook to load them from System32. Note that, for this to work, something like: + // 'somelib.dll;%(DelayLoadDLLs)' must be added to the 'Delay Loaded Dlls' option of + // the linker properties in Visual Studio (which means this won't work with MinGW). + // For all other DLLs, use SetDefaultDllDirectories(LOAD_LIBRARY_SEARCH_SYSTEM32). + // Finally, we need to perform the whole gymkhana below, where we can't call on + // SetDefaultDllDirectories() directly, because Windows 7 doesn't have the API exposed. + // Also, no, Coverity, we never need to care about freeing kernel32 as a library. + // coverity[leaked_storage] + + pfSetDefaultDllDirectories = (SetDefaultDllDirectories_t) + GetProcAddress(LoadLibraryW(L"kernel32.dll"), "SetDefaultDllDirectories"); + if (pfSetDefaultDllDirectories != NULL) + pfSetDefaultDllDirectories(LOAD_LIBRARY_SEARCH_SYSTEM32); +} + +int APIENTRY WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine, INT nCmdShow) +{ + DWORD dwAttrib; HANDLE hMutex; UNREFERENCED_PARAMETER(hPrevInstance); + DllProtect(); + if (GetUserDefaultUILanguage() == 0x0804) { g_msg_lang = g_msg_cn; @@ -680,32 +902,52 @@ int APIENTRY WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLi g_msg_lang = g_msg_en; } - hMutex = CreateMutexA(NULL, TRUE, "VtoyVlnkMUTEX"); - if ((hMutex != NULL) && (GetLastError() == ERROR_ALREADY_EXISTS)) - { - MessageBoxW(NULL, g_msg_lang[MSGID_RUNNING_TIP], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR); - return 1; - } + hMutex = CreateMutexA(NULL, TRUE, "VtoyVlnkMUTEX"); + if ((hMutex != NULL) && (GetLastError() == ERROR_ALREADY_EXISTS)) + { + MessageBoxW(NULL, g_msg_lang[MSGID_RUNNING_TIP], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR); + return 1; + } GetCurrentDirectoryA(MAX_PATH, g_CurDirA); GetCurrentDirectoryW(MAX_PATH, g_CurDirW); sprintf_s(g_LogFile, sizeof(g_LogFile), "%s\\VentoyVlnk.log", g_CurDirA); - for (i = 0; i < __argc; i++) + ParseCmdLine(lpCmdLine); + + g_hInst = hInstance; + + if (g_ShowHelp) + { + 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); + return 0; + } + else if (g_CmdInFile[0] && g_CmdOutFile[0]) { - if (strncmp(__argv[i], "-Q", 2) == 0 || - strncmp(__argv[i], "-q", 2) == 0) + LogA("========= VentoyVlnk Cmdline Mode =========\n"); + + dwAttrib = GetFileAttributesW(g_CmdInFile); + if (dwAttrib == INVALID_FILE_ATTRIBUTES || (dwAttrib & FILE_ATTRIBUTE_DIRECTORY)) { - g_LogFile[0] = 0; - break; + LogW(L"File <<%ls>> does not exist!\n", g_CmdInFile); + VtoyMessageBox(NULL, g_msg_lang[MSGID_SRC_NONEXIST], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR); + return 1; } - } - - LogA("========= VentoyVlnk =========\n"); + if (!IsSupportedVlnkSuffix(g_CmdOutFile)) + { + LogW(L"File <<%ls>> contains invalid vlnk suffix!\n", g_CmdOutFile); + VtoyMessageBox(NULL, g_msg_lang[MSGID_INVALID_SUFFIX], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR); + return 1; + } - g_hInst = hInstance; - DialogBoxA(hInstance, MAKEINTRESOURCEA(IDD_DIALOG1), NULL, DialogProc); + return CreateVlnk(NULL, g_CurDirW, g_CmdInFile, g_CmdOutFile); + } + else + { + LogA("========= VentoyVlnk GUI Mode =========\n"); - return 0; + DialogBoxA(hInstance, MAKEINTRESOURCEA(IDD_DIALOG1), NULL, DialogProc); + return 0; + } }