MyFavMirrors
/
Ventoy
mirror of https://github.com/ventoy/Ventoy.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add cmdline check

This commit is contained in:
longpanda 2022-02-17 11:09:31 +08:00
parent 9118d5fe45
commit e3506c0f10
2 changed files with 15 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
Vlnk/src/main_windows.c
View File

@ -393,12 +393,12 @@ static BOOL IsSupportedVlnkSuffix(WCHAR *FileName)
if (len > 9) if (len > 9)
{ {
if (lstrcmp(FileName - 9, L".vlnk.iso") == 0 || if (lstrcmp(FileName + len - 9, L".vlnk.iso") == 0 ||
lstrcmp(FileName - 9, L".vlnk.img") == 0 || lstrcmp(FileName + len - 9, L".vlnk.img") == 0 ||
lstrcmp(FileName - 9, L".vlnk.wim") == 0 || lstrcmp(FileName + len - 9, L".vlnk.wim") == 0 ||
lstrcmp(FileName - 9, L".vlnk.vhd") == 0 || lstrcmp(FileName + len - 9, L".vlnk.vhd") == 0 ||
lstrcmp(FileName - 9, L".vlnk.efi") == 0 || lstrcmp(FileName + len - 9, L".vlnk.efi") == 0 ||
lstrcmp(FileName - 9, L".vlnk.dat") == 0) lstrcmp(FileName + len - 9, L".vlnk.dat") == 0)
{ {
return TRUE; return TRUE;
} }
@ -407,8 +407,8 @@ static BOOL IsSupportedVlnkSuffix(WCHAR *FileName)
if (len > 10) if (len > 10)
{ {
if (lstrcmp(FileName - 10, L".vlnk.vhdx") == 0 || if (lstrcmp(FileName + len - 10, L".vlnk.vhdx") == 0 ||
lstrcmp(FileName - 9, L".vlnk.vtoy") == 0) lstrcmp(FileName + len - 10, L".vlnk.vtoy") == 0)
{ {
return TRUE; return TRUE;
} }
@ -879,6 +879,13 @@ int APIENTRY WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLi
return 1; return 1;
} }
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;
}
return CreateVlnk(NULL, g_CurDirW, g_CmdInFile, g_CmdOutFile); return CreateVlnk(NULL, g_CurDirW, g_CmdInFile, g_CmdOutFile);
} }
else else

BIN
Vlnk/vs/VentoyVlnk/Release/VentoyVlnk.exe
View File

Binary file not shown.