Skip to content

GitLab

"sgl-kernel/git@developer.sourcefind.cn:change/sglang.git" did not exist on "5ad296bda140117e4eb85ce781955d94fb221973"
Commit 09494a5b authored by longpanda's avatar longpanda
Browse files

Optimization for Ventoy2Disk.exe

parent e15a1c5e
Hide whitespace changes
Inline Side-by-side
Showing with 67 additions and 27 deletions
INSTALL/Ventoy2Disk.exe
View file @ 09494a5b
No preview for this file type
Ventoy2Disk/Ventoy2Disk/PhyDrive.c
View file @ 09494a5b
...@@ -31,6 +31,8 @@ ...@@ -31,6 +31,8 @@
#include "ff.h" #include "ff.h"
#include "DiskService.h" #include "DiskService.h"
static int g_backup_bin_index = 0;
static DWORD GetVentoyVolumeName(int PhyDrive, UINT64 StartSectorId, CHAR *NameBuf, UINT32 BufLen, BOOL DelSlash) static DWORD GetVentoyVolumeName(int PhyDrive, UINT64 StartSectorId, CHAR *NameBuf, UINT32 BufLen, BOOL DelSlash)
{ {
size_t len; size_t len;
...@@ -1841,7 +1843,7 @@ static BOOL BackupDataBeforeCleanDisk(int PhyDrive, UINT64 DiskSize, BYTE **pBac ...@@ -1841,7 +1843,7 @@ static BOOL BackupDataBeforeCleanDisk(int PhyDrive, UINT64 DiskSize, BYTE **pBac
Log("BackupDataBeforeCleanDisk %d", PhyDrive); Log("BackupDataBeforeCleanDisk %d", PhyDrive);
backup = malloc(SIZE_1MB * 3); backup = malloc(SIZE_1MB * 4);
if (!backup) if (!backup)
{ {
goto out; goto out;
...@@ -1853,7 +1855,7 @@ static BOOL BackupDataBeforeCleanDisk(int PhyDrive, UINT64 DiskSize, BYTE **pBac ...@@ -1853,7 +1855,7 @@ static BOOL BackupDataBeforeCleanDisk(int PhyDrive, UINT64 DiskSize, BYTE **pBac
goto out; goto out;
} }
//read first 1MB //read first 2MB
dwStatus = SetFilePointer(hDrive, 0, NULL, FILE_BEGIN); dwStatus = SetFilePointer(hDrive, 0, NULL, FILE_BEGIN);
if (dwStatus != 0) if (dwStatus != 0)
{ {
...@@ -1861,14 +1863,15 @@ static BOOL BackupDataBeforeCleanDisk(int PhyDrive, UINT64 DiskSize, BYTE **pBac ...@@ -1861,14 +1863,15 @@ static BOOL BackupDataBeforeCleanDisk(int PhyDrive, UINT64 DiskSize, BYTE **pBac
} }
dwSize = 0; dwSize = 0;
ret = ReadFile(hDrive, backup, SIZE_1MB, &dwSize, NULL); ret = ReadFile(hDrive, backup, SIZE_2MB, &dwSize, NULL);
if ((!ret) || (dwSize != SIZE_1MB)) if ((!ret) || (dwSize != SIZE_2MB))
{ {
Log("Failed to read %d %u 0x%x", ret, dwSize, LASTERR); Log("Failed to read %d %u 0x%x", ret, dwSize, LASTERR);
goto out; goto out;
} }
liCurPosition.QuadPart = DiskSize - (SIZE_1MB * 2); //read last 2MB
liCurPosition.QuadPart = DiskSize - SIZE_2MB;
liNewPosition.QuadPart = 0; liNewPosition.QuadPart = 0;
if (0 == SetFilePointerEx(hDrive, liCurPosition, &liNewPosition, FILE_BEGIN) || if (0 == SetFilePointerEx(hDrive, liCurPosition, &liNewPosition, FILE_BEGIN) ||
liNewPosition.QuadPart != liCurPosition.QuadPart) liNewPosition.QuadPart != liCurPosition.QuadPart)
...@@ -1877,8 +1880,8 @@ static BOOL BackupDataBeforeCleanDisk(int PhyDrive, UINT64 DiskSize, BYTE **pBac ...@@ -1877,8 +1880,8 @@ static BOOL BackupDataBeforeCleanDisk(int PhyDrive, UINT64 DiskSize, BYTE **pBac
} }
dwSize = 0; dwSize = 0;
ret = ReadFile(hDrive, backup + SIZE_1MB, 2 * SIZE_1MB, &dwSize, NULL); ret = ReadFile(hDrive, backup + SIZE_2MB, SIZE_2MB, &dwSize, NULL);
if ((!ret) || (dwSize != 2 * SIZE_1MB)) if ((!ret) || (dwSize != SIZE_2MB))
{ {
Log("Failed to read %d %u 0x%x", ret, dwSize, LASTERR); Log("Failed to read %d %u 0x%x", ret, dwSize, LASTERR);
goto out; goto out;
...@@ -1930,11 +1933,12 @@ int UpdateVentoy2PhyDrive(PHY_DRIVE_INFO *pPhyDrive, int TryId) ...@@ -1930,11 +1933,12 @@ int UpdateVentoy2PhyDrive(PHY_DRIVE_INFO *pPhyDrive, int TryId)
{ {
int i; int i;
int rc = 0; int rc = 0;
int MaxRetry = 3; int MaxRetry = 4;
BOOL ForceMBR = FALSE; BOOL ForceMBR = FALSE;
BOOL Esp2Basic = FALSE; BOOL Esp2Basic = FALSE;
BOOL ChangeAttr = FALSE; BOOL ChangeAttr = FALSE;
BOOL CleanDisk = FALSE; BOOL CleanDisk = FALSE;
BOOL bWriteBack = TRUE;
HANDLE hVolume; HANDLE hVolume;
HANDLE hDrive; HANDLE hDrive;
DWORD Status; DWORD Status;
...@@ -1942,6 +1946,7 @@ int UpdateVentoy2PhyDrive(PHY_DRIVE_INFO *pPhyDrive, int TryId) ...@@ -1942,6 +1946,7 @@ int UpdateVentoy2PhyDrive(PHY_DRIVE_INFO *pPhyDrive, int TryId)
BOOL bRet; BOOL bRet;
CHAR DriveName[] = "?:\\"; CHAR DriveName[] = "?:\\";
CHAR DriveLetters[MAX_PATH] = { 0 }; CHAR DriveLetters[MAX_PATH] = { 0 };
CHAR BackBinFile[MAX_PATH];
UINT64 StartSector; UINT64 StartSector;
UINT64 ReservedMB = 0; UINT64 ReservedMB = 0;
MBR_HEAD BootImg; MBR_HEAD BootImg;
...@@ -2044,7 +2049,7 @@ int UpdateVentoy2PhyDrive(PHY_DRIVE_INFO *pPhyDrive, int TryId) ...@@ -2044,7 +2049,7 @@ int UpdateVentoy2PhyDrive(PHY_DRIVE_INFO *pPhyDrive, int TryId)
if (VDS_ChangeVtoyEFI2ESP(pPhyDrive->PhyDrive, StartSector * 512)) if (VDS_ChangeVtoyEFI2ESP(pPhyDrive->PhyDrive, StartSector * 512))
{ {
Esp2Basic = TRUE; Esp2Basic = TRUE;
Sleep(1000); Sleep(3000);
} }
} }
else if (TryId == 2) else if (TryId == 2)
...@@ -2053,7 +2058,7 @@ int UpdateVentoy2PhyDrive(PHY_DRIVE_INFO *pPhyDrive, int TryId) ...@@ -2053,7 +2058,7 @@ int UpdateVentoy2PhyDrive(PHY_DRIVE_INFO *pPhyDrive, int TryId)
if (VDS_ChangeVtoyEFIAttr(pPhyDrive->PhyDrive, 0x8000000000000001)) if (VDS_ChangeVtoyEFIAttr(pPhyDrive->PhyDrive, 0x8000000000000001))
{ {
ChangeAttr = TRUE; ChangeAttr = TRUE;
Sleep(1000); Sleep(2000);
} }
} }
else if (TryId == 3) else if (TryId == 3)
...@@ -2061,10 +2066,19 @@ int UpdateVentoy2PhyDrive(PHY_DRIVE_INFO *pPhyDrive, int TryId) ...@@ -2061,10 +2066,19 @@ int UpdateVentoy2PhyDrive(PHY_DRIVE_INFO *pPhyDrive, int TryId)
Log("Clean disk GPT partition table"); Log("Clean disk GPT partition table");
if (BackupDataBeforeCleanDisk(pPhyDrive->PhyDrive, pPhyDrive->SizeInBytes, &pBackup)) if (BackupDataBeforeCleanDisk(pPhyDrive->PhyDrive, pPhyDrive->SizeInBytes, &pBackup))
{ {
sprintf_s(BackBinFile, sizeof(BackBinFile), ".\\ventoy\\phydrive%d_%u_%d.bin",
pPhyDrive->PhyDrive, GetCurrentProcessId(), g_backup_bin_index++);
SaveBufToFile(BackBinFile, pBackup, 4 * SIZE_1MB);
Log("Save backup data to %s", BackBinFile);
Log("Success to backup data before clean"); Log("Success to backup data before clean");
CleanDisk = TRUE; CleanDisk = TRUE;
VDS_CleanDisk(pPhyDrive->PhyDrive); if (!VDS_CleanDisk(pPhyDrive->PhyDrive))
Sleep(1000); {
Sleep(3000);
DSPT_CleanDisk(pPhyDrive->PhyDrive);
}
Sleep(3000);
} }
else else
{ {
...@@ -2092,7 +2106,11 @@ int UpdateVentoy2PhyDrive(PHY_DRIVE_INFO *pPhyDrive, int TryId) ...@@ -2092,7 +2106,11 @@ int UpdateVentoy2PhyDrive(PHY_DRIVE_INFO *pPhyDrive, int TryId)
//If we change VTOYEFI to ESP, it can not have s volume name, so don't try to get it. //If we change VTOYEFI to ESP, it can not have s volume name, so don't try to get it.
if (CleanDisk) if (CleanDisk)
{ {
WriteBackupDataToDisk(hDrive, pPhyDrive->SizeInBytes - (2 * SIZE_1MB), pBackup + SIZE_1MB, 2 * SIZE_1MB); //writeback the last 2MB
if (!WriteBackupDataToDisk(hDrive, pPhyDrive->SizeInBytes - SIZE_2MB, pBackup + SIZE_2MB, SIZE_2MB))
{
bWriteBack = FALSE;
}
Status = ERROR_NOT_FOUND; Status = ERROR_NOT_FOUND;
} }
else if (Esp2Basic) else if (Esp2Basic)
...@@ -2164,7 +2182,15 @@ int UpdateVentoy2PhyDrive(PHY_DRIVE_INFO *pPhyDrive, int TryId) ...@@ -2164,7 +2182,15 @@ int UpdateVentoy2PhyDrive(PHY_DRIVE_INFO *pPhyDrive, int TryId)
goto End; goto End;
} }
if (!TryWritePart2(hDrive, StartSector)) bRet = TryWritePart2(hDrive, StartSector);
if (FALSE == bRet && Esp2Basic)
{
Log("TryWritePart2 agagin ...");
Sleep(3000);
bRet = TryWritePart2(hDrive, StartSector);
}
if (!bRet)
{ {
if (pPhyDrive->PartStyle == 0) if (pPhyDrive->PartStyle == 0)
{ {
...@@ -2262,9 +2288,23 @@ int UpdateVentoy2PhyDrive(PHY_DRIVE_INFO *pPhyDrive, int TryId) ...@@ -2262,9 +2288,23 @@ int UpdateVentoy2PhyDrive(PHY_DRIVE_INFO *pPhyDrive, int TryId)
if (CleanDisk) if (CleanDisk)
{ {
WriteBackupDataToDisk(hDrive, 4 * 512, pBackup + 4 * 512, SIZE_1MB - 4 * 512); if (!WriteBackupDataToDisk(hDrive, 4 * 512, pBackup + 4 * 512, SIZE_2MB - 4 * 512))
WriteBackupDataToDisk(hDrive, 0, pBackup, 4 * 512); {
bWriteBack = FALSE;
}
if (!WriteBackupDataToDisk(hDrive, 0, pBackup, 4 * 512))
{
bWriteBack = FALSE;
}
free(pBackup); free(pBackup);
if (bWriteBack)
{
Log("Write success, now delete %s", BackBinFile);
DeleteFileA(BackBinFile);
}
} }
//Refresh Drive Layout //Refresh Drive Layout
...@@ -2291,20 +2331,20 @@ End: ...@@ -2291,20 +2331,20 @@ End:
if (pPhyDrive->PartStyle == 1) if (pPhyDrive->PartStyle == 1)
{ {
if (ChangeAttr || ((pPhyDrive->Part2GPTAttr >> 56) != 0xC0)) if (ChangeAttr || ((pPhyDrive->Part2GPTAttr >> 56) != 0xC0))
{
Log("Change EFI partition attr %u <0x%llx> to <0x%llx>", ChangeAttr, pPhyDrive->Part2GPTAttr, 0xC000000000000001ULL);
if (VDS_ChangeVtoyEFIAttr(pPhyDrive->PhyDrive, 0xC000000000000001ULL))
{
Log("Change EFI partition attr success");
pPhyDrive->Part2GPTAttr = 0xC000000000000001ULL;
}
else
{ {
Log("Change EFI partition attr failed"); Log("Change EFI partition attr %u <0x%llx> to <0x%llx>", ChangeAttr, pPhyDrive->Part2GPTAttr, 0xC000000000000001ULL);
if (VDS_ChangeVtoyEFIAttr(pPhyDrive->PhyDrive, 0xC000000000000001ULL))
{
Log("Change EFI partition attr success");
pPhyDrive->Part2GPTAttr = 0xC000000000000001ULL;
}
else
{
Log("Change EFI partition attr failed");
}
} }
} }
}
if (pGptInfo) if (pGptInfo)
{ {
......
Ventoy2Disk/Ventoy2Disk/WinDialog.c
View file @ 09494a5b
File suppressed by a .gitattributes entry or the file's encoding is unsupported.
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment