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

Optimization for Lenovo Recovery Image (USB unplug process)

This commit is contained in:
longpanda 2023-01-05 22:18:39 +08:00
parent 7fe4762faa
commit 686ed11037
3 changed files with 67 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
INSTALL/ventoy/vtoyjump32.exe
View File

Binary file not shown.

BIN
INSTALL/ventoy/vtoyjump64.exe
View File

Binary file not shown.

66
vtoyjump/vtoyjump/vtoyjump.c
View File

@ -2362,6 +2362,9 @@ static BOOL VentoyIsLenovoRecovery(CHAR *IsoPath, CHAR *VTLRIPath)
static int MountVTLRI(CHAR *ImgPath, DWORD PhyDrive)
{
STARTUPINFOA Si;
PROCESS_INFORMATION Pi;
CHAR Cmdline[256];
CHAR ImDiskPath[256];
Log("MountVTLRI <%s> %u", ImgPath, PhyDrive);
@ -2370,6 +2373,18 @@ static int MountVTLRI(CHAR *ImgPath, DWORD PhyDrive)
VentoyRunImdisk("VTLRI", ImgPath, ImDiskPath, "ro,rem");
CopyFileA(g_prog_full_path, "ventoy\\VTLRISRV.exe", FALSE);
sprintf_s(Cmdline, sizeof(Cmdline), "ventoy\\VTLRISRV.exe VTLRI_SRV %C Z", ImgPath[0]);
GetStartupInfoA(&Si);
Si.dwFlags |= STARTF_USESHOWWINDOW;
Si.wShowWindow = SW_HIDE;
CreateProcessA(NULL, Cmdline, NULL, NULL, FALSE, 0, NULL, NULL, &Si, &Pi);
Log("Process cmdline <%s>", Cmdline);
return 0;
}
@ -2939,6 +2954,52 @@ static int vtoy_remove_duplicate_file(char *File)
return 0;
}
static int VTLRI_ServiceMain(int argc, char **argv)
{
int n = 0;
DWORD Err;
BOOL bRet = TRUE;
CHAR Drive[16];
CHAR FsType[64];
CHAR Cmdline[256];
PROCESS_INFORMATION Pi;
STARTUPINFOA Si;
//XXX.exe VTLRI_SRV D Z
Log("VTLRI_ServiceMain start %s %s %s ...", argv[0], argv[1], argv[2]);
sprintf_s(Drive, sizeof(Drive), "%C:\\", argv[2][0]);
while (n < 3)
{
bRet = GetVolumeInformationA(Drive, NULL, 0, NULL, NULL, NULL, FsType, sizeof(FsType));
if (bRet)
{
Sleep(400);
}
else
{
Err = LASTERR;
if (Err == ERROR_PATH_NOT_FOUND)
{
Log("%s not found", Drive);
n++;
}
}
}
sprintf_s(Cmdline, sizeof(Cmdline), "ventoy\\imdisk.exe -d -m %C:", argv[3][0]);
Log("Remove disk by <%s>", Cmdline);
GetStartupInfoA(&Si);
Si.dwFlags |= STARTF_USESHOWWINDOW;
Si.wShowWindow = SW_HIDE;
CreateProcessA(NULL, Cmdline, NULL, NULL, FALSE, 0, NULL, NULL, &Si, &Pi);
WaitForSingleObject(Pi.hProcess, INFINITE);
return 0;
}
int main(int argc, char **argv)
{
int i;
@ -2953,6 +3014,11 @@ int main(int argc, char **argv)
Log("######## VentoyJump %dbit ##########", g_system_bit);
if (argc > 1 && strcmp(argv[1], "VTLRI_SRV") == 0)
{
return VTLRI_ServiceMain(argc, argv);
}
GetCurrentDirectoryA(sizeof(CurDir), CurDir);
Log("Current directory is <%s>", CurDir);