-/******************************************************************************
- * Utility.c
- *
- * Copyright (c) 2020, longpanda <admin@ventoy.net>
- *
- * This program is free software; you can redistribute it and/or
- * modify it under the terms of the GNU General Public License as
- * published by the Free Software Foundation; either version 3 of the
- * License, or (at your option) any later version.
- *
- * This program is distributed in the hope that it will be useful, but
- * WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- * General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program; if not, see <http://www.gnu.org/licenses/>.
- *
- */
-#include <Windows.h>
-#include "Ventoy2Disk.h"
-
-void Log(const char *Fmt, ...)
-{
- va_list Arg;
- int Len = 0;
- FILE *File = NULL;
- SYSTEMTIME Sys;
- char szBuf[1024];
-
- GetLocalTime(&Sys);
- Len += safe_sprintf(szBuf,
- "[%4d/%02d/%02d %02d:%02d:%02d.%03d] ",
- Sys.wYear, Sys.wMonth, Sys.wDay,
- Sys.wHour, Sys.wMinute, Sys.wSecond,
- Sys.wMilliseconds);
-
- va_start(Arg, Fmt);
- Len += vsnprintf_s(szBuf + Len, sizeof(szBuf)-Len, sizeof(szBuf)-Len, Fmt, Arg);
- va_end(Arg);
-
- //printf("%s\n", szBuf);
-
-#if 1
- fopen_s(&File, VENTOY_FILE_LOG, "a+");
- if (File)
- {
- fwrite(szBuf, 1, Len, File);
- fwrite("\n", 1, 1, File);
- fclose(File);
- }
-#endif
-
-}
-
-BOOL IsPathExist(BOOL Dir, const char *Fmt, ...)
-{
- va_list Arg;
- HANDLE hFile;
- DWORD Attr;
- CHAR FilePath[MAX_PATH];
-
- va_start(Arg, Fmt);
- vsnprintf_s(FilePath, sizeof(FilePath), sizeof(FilePath), Fmt, Arg);
- va_end(Arg);
-
- hFile = CreateFileA(FilePath, FILE_READ_EA, FILE_SHARE_READ, 0, OPEN_EXISTING, 0, 0);
- if (INVALID_HANDLE_VALUE == hFile)
- {
- return FALSE;
- }
-
- CloseHandle(hFile);
-
- Attr = GetFileAttributesA(FilePath);
-
- if (Dir)
- {
- if ((Attr & FILE_ATTRIBUTE_DIRECTORY) == 0)
- {
- return FALSE;
- }
- }
- else
- {
- if (Attr & FILE_ATTRIBUTE_DIRECTORY)
- {
- return FALSE;
- }
- }
-
- return TRUE;
-}
-
-
-int ReadWholeFileToBuf(const CHAR *FileName, int ExtLen, void **Bufer, int *BufLen)
-{
- int FileSize;
- FILE *File = NULL;
- void *Data = NULL;
-
- fopen_s(&File, FileName, "rb");
- if (File == NULL)
- {
- Log("Failed to open file %s", FileName);
- return 1;
- }
-
- fseek(File, 0, SEEK_END);
- FileSize = (int)ftell(File);
-
- Data = malloc(FileSize + ExtLen);
- if (!Data)
- {
- fclose(File);
- return 1;
- }
-
- fseek(File, 0, SEEK_SET);
- fread(Data, 1, FileSize, File);
-
- fclose(File);
-
- *Bufer = Data;
- *BufLen = FileSize;
-
- return 0;
-}
-
-const CHAR* GetLocalVentoyVersion(void)
-{
- int rc;
- int FileSize;
- CHAR *Pos = NULL;
- CHAR *Buf = NULL;
- static CHAR LocalVersion[64] = { 0 };
-
- if (LocalVersion[0] == 0)
- {
- rc = ReadWholeFileToBuf(VENTOY_FILE_VERSION, 1, (void **)&Buf, &FileSize);
- if (rc)
- {
- return "";
- }
- Buf[FileSize] = 0;
-
- for (Pos = Buf; *Pos; Pos++)
- {
- if (*Pos == '\r' || *Pos == '\n')
- {
- *Pos = 0;
- break;
- }
- }
-
- safe_sprintf(LocalVersion, "%s", Buf);
- free(Buf);
- }
-
- return LocalVersion;
-}
-
-const CHAR* ParseVentoyVersionFromString(CHAR *Buf)
-{
- CHAR *Pos = NULL;
- CHAR *End = NULL;
- static CHAR LocalVersion[64] = { 0 };
-
- Pos = strstr(Buf, "VENTOY_VERSION=");
- if (Pos)
- {
- Pos += strlen("VENTOY_VERSION=");
- if (*Pos == '"')
- {
- Pos++;
- }
-
- End = Pos;
- while (*End != 0 && *End != '"' && *End != '\r' && *End != '\n')
- {
- End++;
- }
-
- *End = 0;
-
- safe_sprintf(LocalVersion, "%s", Pos);
- return LocalVersion;
- }
-
- return "";
-}
-
-BOOL IsWow64(void)
-{
- typedef BOOL(WINAPI *LPFN_ISWOW64PROCESS)(HANDLE, PBOOL);
- LPFN_ISWOW64PROCESS fnIsWow64Process;
- BOOL bIsWow64 = FALSE;
-
- fnIsWow64Process = (LPFN_ISWOW64PROCESS)GetProcAddress(GetModuleHandleA("kernel32"), "IsWow64Process");
- if (NULL != fnIsWow64Process)
- {
- fnIsWow64Process(GetCurrentProcess(), &bIsWow64);
- }
-
- return bIsWow64;
-}
-
-void DumpWindowsVersion(void)
-{
- int Bit;
- BOOL WsVer;
- DWORD Major, Minor;
- ULONGLONG MajorEqual, MinorEqual;
- OSVERSIONINFOEXA Ver1, Ver2;
- const CHAR *Ver = NULL;
- CHAR WinVer[256] = { 0 };
-
- memset(&Ver1, 0, sizeof(Ver1));
- memset(&Ver2, 0, sizeof(Ver2));
-
- Ver1.dwOSVersionInfoSize = sizeof(Ver1);
-
- // suppress the C4996 warning for GetVersionExA
- #pragma warning(push)
- #pragma warning(disable:4996)
- if (!GetVersionExA((OSVERSIONINFOA *)&Ver1))
- {
- memset(&Ver1, 0, sizeof(Ver1));
- Ver1.dwOSVersionInfoSize = sizeof(OSVERSIONINFOA);
- if (!GetVersionExA((OSVERSIONINFOA *)&Ver1))
- {
- return;
- }
- }
- #pragma warning(pop)
-
- if (Ver1.dwPlatformId == VER_PLATFORM_WIN32_NT)
- {
- if (Ver1.dwMajorVersion > 6 || (Ver1.dwMajorVersion == 6 && Ver1.dwMinorVersion >= 2))
- {
- // GetVersionEx() has problem on some Windows version
-
- MajorEqual = VerSetConditionMask(0, VER_MAJORVERSION, VER_EQUAL);
- for (Major = Ver1.dwMajorVersion; Major <= 9; Major++)
- {
- memset(&Ver2, 0, sizeof(Ver2));
- Ver2.dwOSVersionInfoSize = sizeof(Ver2);
- Ver2.dwMajorVersion = Major;
-
- if (!VerifyVersionInfoA(&Ver2, VER_MAJORVERSION, MajorEqual))
- {
- continue;
- }
-
- if (Ver1.dwMajorVersion < Major)
- {
- Ver1.dwMajorVersion = Major;
- Ver1.dwMinorVersion = 0;
- }
-
- MinorEqual = VerSetConditionMask(0, VER_MINORVERSION, VER_EQUAL);
- for (Minor = Ver1.dwMinorVersion; Minor <= 9; Minor++)
- {
- memset(&Ver2, 0, sizeof(Ver2));
-
- Ver2.dwOSVersionInfoSize = sizeof(Ver2);
- Ver2.dwMinorVersion = Minor;
-
- if (!VerifyVersionInfoA(&Ver2, VER_MINORVERSION, MinorEqual))
- {
- continue;
- }
-
- Ver1.dwMinorVersion = Minor;
- break;
- }
-
- break;
- }
- }
-
- if (Ver1.dwMajorVersion <= 0xF && Ver1.dwMinorVersion <= 0xF)
- {
- WsVer = (Ver1.wProductType <= VER_NT_WORKSTATION);
- switch ((Ver1.dwMajorVersion << 4) | Ver2.dwMinorVersion)
- {
- case 0x51:
- {
- Ver = "XP";
- break;
- }
- case 0x52:
- {
- Ver = GetSystemMetrics(89) ? "Server 2003 R2" : "Server 2003";
- break;
- }
- case 0x60:
- {
- Ver = WsVer ? "Vista" : "Server 2008";
- break;
- }
- case 0x61:
- {
- Ver = WsVer ? "7" : "Server 2008 R2";
- break;
- }
- case 0x62:
- {
- Ver = WsVer ? "8" : "Server 2012";
- break;
- }
- case 0x63:
- {
- Ver = WsVer ? "8.1" : "Server 2012 R2";
- break;
- }
- case 0x64:
- {
- Ver = WsVer ? "10 (Preview 1)" : "Server 10 (Preview 1)";
- break;
- }
- case 0xA0:
- {
- Ver = WsVer ? "10" : ((Ver1.dwBuildNumber > 15000) ? "Server 2019" : "Server 2016");
- break;
- }
- default:
- {
- Ver = "10 or later";
- break;
- }
- }
- }
- }
-
- Bit = IsWow64() ? 64 : 32;
-
- if (Ver1.wServicePackMinor)
- {
- safe_sprintf(WinVer, "Windows %s SP%u.%u %d-bit", Ver, Ver1.wServicePackMajor, Ver1.wServicePackMinor, Bit);
- }
- else if (Ver1.wServicePackMajor)
- {
- safe_sprintf(WinVer, "Windows %s SP%u %d-bit", Ver, Ver1.wServicePackMajor, Bit);
- }
- else
- {
- safe_sprintf(WinVer, "Windows %s %d-bit", Ver, Bit);
- }
-
- if (((Ver1.dwMajorVersion << 4) | Ver2.dwMinorVersion) >= 0x62)
- {
- Log("Windows Version : %s (Build %u)", WinVer, Ver1.dwBuildNumber);
- }
- else
- {
- Log("Windows Version : %s", WinVer);
- }
-
- return;
-}
-
-BOOL IsVentoyLogicalDrive(CHAR DriveLetter)
-{
- int i;
- CONST CHAR *Files[] =
- {
- "EFI\\BOOT\\BOOTX64.EFI",
- "grub\\themes\\ventoy\\theme.txt",
- "ventoy\\ventoy.cpio",
- };
-
- for (i = 0; i < sizeof(Files) / sizeof(Files[0]); i++)
- {
- if (!IsFileExist("%C:\\%s", DriveLetter, Files[i]))
- {
- return FALSE;
- }
- }
-
- return TRUE;
-}
-
-
-int VentoyFillLocation(UINT64 DiskSizeInBytes, UINT32 StartSectorId, UINT32 SectorCount, PART_TABLE *Table)
-{
- BYTE Head;
- BYTE Sector;
- BYTE nSector = 63;
- BYTE nHead = 8;
- UINT32 Cylinder;
- UINT32 EndSectorId;
-
- while (nHead != 0 && (DiskSizeInBytes / 512 / nSector / nHead) > 1024)
- {
- nHead = (BYTE)nHead * 2;
- }
-
- if (nHead == 0)
- {
- nHead = 255;
- }
-
- Cylinder = StartSectorId / nSector / nHead;
- Head = StartSectorId / nSector % nHead;
- Sector = StartSectorId % nSector + 1;
-
- Table->StartHead = Head;
- Table->StartSector = Sector;
- Table->StartCylinder = Cylinder;
-
- EndSectorId = StartSectorId + SectorCount - 1;
- Cylinder = EndSectorId / nSector / nHead;
- Head = EndSectorId / nSector % nHead;
- Sector = EndSectorId % nSector + 1;
-
- Table->EndHead = Head;
- Table->EndSector = Sector;
- Table->EndCylinder = Cylinder;
-
- Table->StartSectorId = StartSectorId;
- Table->SectorCount = SectorCount;
-
- return 0;
-}
-
-int VentoyFillMBR(UINT64 DiskSizeBytes, MBR_HEAD *pMBR, int PartStyle)
-{
- GUID Guid;
- int ReservedValue;
- UINT32 DiskSignature;
- UINT32 DiskSectorCount;
- UINT32 PartSectorCount;
- UINT32 PartStartSector;
- UINT32 ReservedSector;
-
- VentoyGetLocalBootImg(pMBR);
-
- CoCreateGuid(&Guid);
-
- memcpy(&DiskSignature, &Guid, sizeof(UINT32));
-
- Log("Disk signature: 0x%08x", DiskSignature);
-
- *((UINT32 *)(pMBR->BootCode + 0x1B8)) = DiskSignature;
-
- if (DiskSizeBytes / 512 > 0xFFFFFFFF)
- {
- DiskSectorCount = 0xFFFFFFFF;
- }
- else
- {
- DiskSectorCount = (UINT32)(DiskSizeBytes / 512);
- }
-
- ReservedValue = GetReservedSpaceInMB();
- if (ReservedValue <= 0)
- {
- ReservedSector = 0;
- }
- else
- {
- ReservedSector = (UINT32)(ReservedValue * 2048);
- }
-
- if (PartStyle)
- {
- ReservedSector += 33; // backup GPT part table
- }
-
- Log("ReservedSector: %u", ReservedSector);
-
- //Part1
- PartStartSector = VENTOY_PART1_START_SECTOR;
- PartSectorCount = DiskSectorCount - ReservedSector - VENTOY_EFI_PART_SIZE / 512 - PartStartSector;
- VentoyFillLocation(DiskSizeBytes, PartStartSector, PartSectorCount, pMBR->PartTbl);
-
- pMBR->PartTbl[0].Active = 0x80; // bootable
- pMBR->PartTbl[0].FsFlag = 0x07; // exFAT/NTFS/HPFS
-
- //Part2
- PartStartSector += PartSectorCount;
- PartSectorCount = VENTOY_EFI_PART_SIZE / 512;
- VentoyFillLocation(DiskSizeBytes, PartStartSector, PartSectorCount, pMBR->PartTbl + 1);
-
- pMBR->PartTbl[1].Active = 0x00;
- pMBR->PartTbl[1].FsFlag = 0xEF; // EFI System Partition
-
- pMBR->Byte55 = 0x55;
- pMBR->ByteAA = 0xAA;
-
- return 0;
-}
-
-
-static int VentoyFillProtectMBR(UINT64 DiskSizeBytes, MBR_HEAD *pMBR)
-{
- GUID Guid;
- UINT32 DiskSignature;
- UINT64 DiskSectorCount;
-
- VentoyGetLocalBootImg(pMBR);
-
- CoCreateGuid(&Guid);
-
- memcpy(&DiskSignature, &Guid, sizeof(UINT32));
-
- Log("Disk signature: 0x%08x", DiskSignature);
-
- *((UINT32 *)(pMBR->BootCode + 0x1B8)) = DiskSignature;
-
- DiskSectorCount = DiskSizeBytes / 512 - 1;
- if (DiskSectorCount > 0xFFFFFFFF)
- {
- DiskSectorCount = 0xFFFFFFFF;
- }
-
- memset(pMBR->PartTbl, 0, sizeof(pMBR->PartTbl));
-
- pMBR->PartTbl[0].Active = 0x00;
- pMBR->PartTbl[0].FsFlag = 0xee; // EE
-
- pMBR->PartTbl[0].StartHead = 0;
- pMBR->PartTbl[0].StartSector = 1;
- pMBR->PartTbl[0].StartCylinder = 0;
- pMBR->PartTbl[0].EndHead = 254;
- pMBR->PartTbl[0].EndSector = 63;
- pMBR->PartTbl[0].EndCylinder = 1023;
-
- pMBR->PartTbl[0].StartSectorId = 1;
- pMBR->PartTbl[0].SectorCount = (UINT32)DiskSectorCount;
-
- pMBR->Byte55 = 0x55;
- pMBR->ByteAA = 0xAA;
-
- pMBR->BootCode[92] = 0x22;
-
- return 0;
-}
-
-
-int VentoyFillGpt(UINT64 DiskSizeBytes, VTOY_GPT_INFO *pInfo)
-{
- INT64 ReservedValue = 0;
- UINT64 ReservedSector = 33;
- UINT64 Part1SectorCount = 0;
- UINT64 DiskSectorCount = DiskSizeBytes / 512;
- VTOY_GPT_HDR *Head = &pInfo->Head;
- VTOY_GPT_PART_TBL *Table = pInfo->PartTbl;
- static GUID WindowsDataPartType = { 0xebd0a0a2, 0xb9e5, 0x4433, { 0x87, 0xc0, 0x68, 0xb6, 0xb7, 0x26, 0x99, 0xc7 } };
- static GUID EspPartType = { 0xc12a7328, 0xf81f, 0x11d2, { 0xba, 0x4b, 0x00, 0xa0, 0xc9, 0x3e, 0xc9, 0x3b } };
- //static GUID BiosGrubPartType = { 0x21686148, 0x6449, 0x6e6f, { 0x74, 0x4e, 0x65, 0x65, 0x64, 0x45, 0x46, 0x49 } };
-
- VentoyFillProtectMBR(DiskSizeBytes, &pInfo->MBR);
-
- ReservedValue = GetReservedSpaceInMB();
- if (ReservedValue > 0)
- {
- ReservedSector += ReservedValue * 2048;
- }
-
- Part1SectorCount = DiskSectorCount - ReservedSector - (VENTOY_EFI_PART_SIZE / 512) - 2048;
-
- memcpy(Head->Signature, "EFI PART", 8);
- Head->Version[2] = 0x01;
- Head->Length = 92;
- Head->Crc = 0;
- Head->EfiStartLBA = 1;
- Head->EfiBackupLBA = DiskSectorCount - 1;
- Head->PartAreaStartLBA = 34;
- Head->PartAreaEndLBA = DiskSectorCount - 34;
- CoCreateGuid(&Head->DiskGuid);
- Head->PartTblStartLBA = 2;
- Head->PartTblTotNum = 128;
- Head->PartTblEntryLen = 128;
-
-
- memcpy(&(Table[0].PartType), &WindowsDataPartType, sizeof(GUID));
- CoCreateGuid(&(Table[0].PartGuid));
- Table[0].StartLBA = 2048;
- Table[0].LastLBA = 2048 + Part1SectorCount - 1;
- Table[0].Attr = 0;
- memcpy(Table[0].Name, L"Ventoy", 6 * 2);
-
- memcpy(&(Table[1].PartType), &EspPartType, sizeof(GUID));
- CoCreateGuid(&(Table[1].PartGuid));
- Table[1].StartLBA = Table[0].LastLBA + 1;
- Table[1].LastLBA = Table[1].StartLBA + VENTOY_EFI_PART_SIZE / 512 - 1;
- Table[1].Attr = 1;
- memcpy(Table[1].Name, L"VTOYEFI", 7 * 2);
-
-#if 0
- memcpy(&(Table[2].PartType), &BiosGrubPartType, sizeof(GUID));
- CoCreateGuid(&(Table[2].PartGuid));
- Table[2].StartLBA = 34;
- Table[2].LastLBA = 2047;
- Table[2].Attr = 0;
-#endif
-
- //Update CRC
- Head->PartTblCrc = VentoyCrc32(Table, sizeof(pInfo->PartTbl));
- Head->Crc = VentoyCrc32(Head, Head->Length);
-
- return 0;
-}
-
-int VentoyFillBackupGptHead(VTOY_GPT_INFO *pInfo, VTOY_GPT_HDR *pHead)
-{
- UINT64 LBA;
- UINT64 BackupLBA;
-
- memcpy(pHead, &pInfo->Head, sizeof(VTOY_GPT_HDR));
-
- LBA = pHead->EfiStartLBA;
- BackupLBA = pHead->EfiBackupLBA;
-
- pHead->EfiStartLBA = BackupLBA;
- pHead->EfiBackupLBA = LBA;
- pHead->PartTblStartLBA = BackupLBA + 1 - 33;
-
- pHead->Crc = 0;
- pHead->Crc = VentoyCrc32(pHead, pHead->Length);
-
- return 0;
-}
-
-CHAR GetFirstUnusedDriveLetter(void)
-{
- CHAR Letter = 'D';
- DWORD Drives = GetLogicalDrives();
-
- Drives >>= 3;
- while (Drives & 0x1)
- {
- Letter++;
- Drives >>= 1;
- }
-
- return Letter;
-}
-
-const CHAR * GetBusTypeString(STORAGE_BUS_TYPE Type)
-{
- switch (Type)
- {
- case BusTypeUnknown: return "unknown";
- case BusTypeScsi: return "SCSI";
- case BusTypeAtapi: return "Atapi";
- case BusTypeAta: return "ATA";
- case BusType1394: return "1394";
- case BusTypeSsa: return "SSA";
- case BusTypeFibre: return "Fibre";
- case BusTypeUsb: return "USB";
- case BusTypeRAID: return "RAID";
- case BusTypeiScsi: return "iSCSI";
- case BusTypeSas: return "SAS";
- case BusTypeSata: return "SATA";
- case BusTypeSd: return "SD";
- case BusTypeMmc: return "MMC";
- case BusTypeVirtual: return "Virtual";
- case BusTypeFileBackedVirtual: return "FileBackedVirtual";
- case BusTypeSpaces: return "Spaces";
- case BusTypeNvme: return "Nvme";
- }
- return "unknown";
-}
-
-int VentoyGetLocalBootImg(MBR_HEAD *pMBR)
-{
- int Len = 0;
- BYTE *ImgBuf = NULL;
- static int Loaded = 0;
- static MBR_HEAD MBR;
-
- if (Loaded)
- {
- memcpy(pMBR, &MBR, 512);
- return 0;
- }
-
- if (0 == ReadWholeFileToBuf(VENTOY_FILE_BOOT_IMG, 0, (void **)&ImgBuf, &Len))
- {
- Log("Copy boot img success");
- memcpy(pMBR, ImgBuf, 512);
- free(ImgBuf);
-
- CoCreateGuid((GUID *)(pMBR->BootCode + 0x180));
-
- memcpy(&MBR, pMBR, 512);
- Loaded = 1;
-
- return 0;
- }
- else
- {
- Log("Copy boot img failed");
- return 1;
- }
-}
-
-int GetHumanReadableGBSize(UINT64 SizeBytes)
-{
- int i;
- int Pow2 = 1;
- double Delta;
- double GB = SizeBytes * 1.0 / 1000 / 1000 / 1000;
-
- for (i = 0; i < 12; i++)
- {
- if (Pow2 > GB)
- {
- Delta = (Pow2 - GB) / Pow2;
- }
- else
- {
- Delta = (GB - Pow2) / Pow2;
- }
-
- if (Delta < 0.05)
- {
- return Pow2;
- }
-
- Pow2 <<= 1;
- }
-
- return (int)GB;
-}
-
-void TrimString(CHAR *String)
-{
- CHAR *Pos1 = String;
- CHAR *Pos2 = String;
- size_t Len = strlen(String);
-
- while (Len > 0)
- {
- if (String[Len - 1] != ' ' && String[Len - 1] != '\t')
- {
- break;
- }
- String[Len - 1] = 0;
- Len--;
- }
-
- while (*Pos1 == ' ' || *Pos1 == '\t')
- {
- Pos1++;
- }
-
- while (*Pos1)
- {
- *Pos2++ = *Pos1++;
- }
- *Pos2++ = 0;
-
- return;
-}
-
-int GetRegDwordValue(HKEY Key, LPCSTR SubKey, LPCSTR ValueName, DWORD *pValue)
-{
- HKEY hKey;
- DWORD Type;
- DWORD Size;
- LSTATUS lRet;
- DWORD Value;
-
- lRet = RegOpenKeyExA(Key, SubKey, 0, KEY_QUERY_VALUE, &hKey);
- Log("RegOpenKeyExA <%s> Ret:%ld", SubKey, lRet);
-
- if (ERROR_SUCCESS == lRet)
- {
- Size = sizeof(Value);
- lRet = RegQueryValueExA(hKey, ValueName, NULL, &Type, (LPBYTE)&Value, &Size);
- Log("RegQueryValueExA <%s> ret:%u Size:%u Value:%u", ValueName, lRet, Size, Value);
-
- *pValue = Value;
- RegCloseKey(hKey);
-
- return 0;
- }
- else
- {
- return 1;
- }
-}
-
-int GetPhysicalDriveCount(void)
-{
- DWORD Value;
- int Count = 0;
-
- if (GetRegDwordValue(HKEY_LOCAL_MACHINE, "SYSTEM\\CurrentControlSet\\Services\\disk\\Enum", "Count", &Value) == 0)
- {
- Count = (int)Value;
- }
-
- Log("GetPhysicalDriveCount: %d", Count);
- return Count;
-}
-
-
-
+/******************************************************************************\r
+ * Utility.c\r
+ *\r
+ * Copyright (c) 2020, longpanda <admin@ventoy.net>\r
+ *\r
+ * This program is free software; you can redistribute it and/or\r
+ * modify it under the terms of the GNU General Public License as\r
+ * published by the Free Software Foundation; either version 3 of the\r
+ * License, or (at your option) any later version.\r
+ * \r
+ * This program is distributed in the hope that it will be useful, but\r
+ * WITHOUT ANY WARRANTY; without even the implied warranty of\r
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU\r
+ * General Public License for more details.\r
+ * \r
+ * You should have received a copy of the GNU General Public License\r
+ * along with this program; if not, see <http://www.gnu.org/licenses/>.\r
+ *\r
+ */\r
+#include <Windows.h>\r
+#include "Ventoy2Disk.h"\r
+\r
+void Log(const char *Fmt, ...)\r
+{\r
+ va_list Arg;\r
+ int Len = 0;\r
+ FILE *File = NULL;\r
+ SYSTEMTIME Sys;\r
+ char szBuf[1024];\r
+\r
+ GetLocalTime(&Sys);\r
+ Len += safe_sprintf(szBuf,\r
+ "[%4d/%02d/%02d %02d:%02d:%02d.%03d] ",\r
+ Sys.wYear, Sys.wMonth, Sys.wDay,\r
+ Sys.wHour, Sys.wMinute, Sys.wSecond,\r
+ Sys.wMilliseconds);\r
+\r
+ va_start(Arg, Fmt);\r
+ Len += vsnprintf_s(szBuf + Len, sizeof(szBuf)-Len, sizeof(szBuf)-Len, Fmt, Arg);\r
+ va_end(Arg);\r
+\r
+ //printf("%s\n", szBuf);\r
+\r
+#if 1\r
+ fopen_s(&File, VENTOY_FILE_LOG, "a+");\r
+ if (File)\r
+ {\r
+ fwrite(szBuf, 1, Len, File);\r
+ fwrite("\n", 1, 1, File);\r
+ fclose(File);\r
+ }\r
+#endif\r
+\r
+}\r
+\r
+BOOL IsPathExist(BOOL Dir, const char *Fmt, ...)\r
+{\r
+ va_list Arg;\r
+ HANDLE hFile;\r
+ DWORD Attr;\r
+ CHAR FilePath[MAX_PATH];\r
+\r
+ va_start(Arg, Fmt);\r
+ vsnprintf_s(FilePath, sizeof(FilePath), sizeof(FilePath), Fmt, Arg);\r
+ va_end(Arg);\r
+\r
+ hFile = CreateFileA(FilePath, FILE_READ_EA, FILE_SHARE_READ, 0, OPEN_EXISTING, 0, 0);\r
+ if (INVALID_HANDLE_VALUE == hFile)\r
+ {\r
+ return FALSE;\r
+ }\r
+\r
+ CloseHandle(hFile);\r
+\r
+ Attr = GetFileAttributesA(FilePath);\r
+\r
+ if (Dir)\r
+ {\r
+ if ((Attr & FILE_ATTRIBUTE_DIRECTORY) == 0)\r
+ {\r
+ return FALSE;\r
+ }\r
+ }\r
+ else\r
+ {\r
+ if (Attr & FILE_ATTRIBUTE_DIRECTORY)\r
+ {\r
+ return FALSE;\r
+ }\r
+ }\r
+\r
+ return TRUE;\r
+}\r
+\r
+\r
+int ReadWholeFileToBuf(const CHAR *FileName, int ExtLen, void **Bufer, int *BufLen)\r
+{\r
+ int FileSize;\r
+ FILE *File = NULL;\r
+ void *Data = NULL;\r
+\r
+ fopen_s(&File, FileName, "rb");\r
+ if (File == NULL)\r
+ {\r
+ Log("Failed to open file %s", FileName);\r
+ return 1;\r
+ }\r
+\r
+ fseek(File, 0, SEEK_END);\r
+ FileSize = (int)ftell(File);\r
+\r
+ Data = malloc(FileSize + ExtLen);\r
+ if (!Data)\r
+ {\r
+ fclose(File);\r
+ return 1;\r
+ }\r
+\r
+ fseek(File, 0, SEEK_SET);\r
+ fread(Data, 1, FileSize, File);\r
+\r
+ fclose(File);\r
+\r
+ *Bufer = Data;\r
+ *BufLen = FileSize;\r
+\r
+ return 0;\r
+}\r
+\r
+const CHAR* GetLocalVentoyVersion(void)\r
+{\r
+ int rc;\r
+ int FileSize;\r
+ CHAR *Pos = NULL;\r
+ CHAR *Buf = NULL;\r
+ static CHAR LocalVersion[64] = { 0 };\r
+\r
+ if (LocalVersion[0] == 0)\r
+ {\r
+ rc = ReadWholeFileToBuf(VENTOY_FILE_VERSION, 1, (void **)&Buf, &FileSize);\r
+ if (rc)\r
+ {\r
+ return "";\r
+ }\r
+ Buf[FileSize] = 0;\r
+\r
+ for (Pos = Buf; *Pos; Pos++)\r
+ {\r
+ if (*Pos == '\r' || *Pos == '\n')\r
+ {\r
+ *Pos = 0;\r
+ break;\r
+ }\r
+ }\r
+\r
+ safe_sprintf(LocalVersion, "%s", Buf);\r
+ free(Buf);\r
+ }\r
+ \r
+ return LocalVersion;\r
+}\r
+\r
+const CHAR* ParseVentoyVersionFromString(CHAR *Buf)\r
+{\r
+ CHAR *Pos = NULL;\r
+ CHAR *End = NULL;\r
+ static CHAR LocalVersion[64] = { 0 };\r
+\r
+ Pos = strstr(Buf, "VENTOY_VERSION=");\r
+ if (Pos)\r
+ {\r
+ Pos += strlen("VENTOY_VERSION=");\r
+ if (*Pos == '"')\r
+ {\r
+ Pos++;\r
+ }\r
+\r
+ End = Pos;\r
+ while (*End != 0 && *End != '"' && *End != '\r' && *End != '\n')\r
+ {\r
+ End++;\r
+ }\r
+\r
+ *End = 0;\r
+\r
+ safe_sprintf(LocalVersion, "%s", Pos);\r
+ return LocalVersion;\r
+ }\r
+\r
+ return "";\r
+}\r
+\r
+BOOL IsWow64(void)\r
+{\r
+ typedef BOOL(WINAPI *LPFN_ISWOW64PROCESS)(HANDLE, PBOOL);\r
+ LPFN_ISWOW64PROCESS fnIsWow64Process;\r
+ BOOL bIsWow64 = FALSE;\r
+\r
+ fnIsWow64Process = (LPFN_ISWOW64PROCESS)GetProcAddress(GetModuleHandleA("kernel32"), "IsWow64Process");\r
+ if (NULL != fnIsWow64Process)\r
+ {\r
+ fnIsWow64Process(GetCurrentProcess(), &bIsWow64);\r
+ }\r
+\r
+ return bIsWow64;\r
+}\r
+\r
+void DumpWindowsVersion(void)\r
+{\r
+ int Bit; \r
+ BOOL WsVer; \r
+ DWORD Major, Minor;\r
+ ULONGLONG MajorEqual, MinorEqual;\r
+ OSVERSIONINFOEXA Ver1, Ver2;\r
+ const CHAR *Ver = NULL; \r
+ CHAR WinVer[256] = { 0 };\r
+\r
+ memset(&Ver1, 0, sizeof(Ver1));\r
+ memset(&Ver2, 0, sizeof(Ver2));\r
+\r
+ Ver1.dwOSVersionInfoSize = sizeof(Ver1);\r
+ \r
+ // suppress the C4996 warning for GetVersionExA\r
+ #pragma warning(push)\r
+ #pragma warning(disable:4996)\r
+ if (!GetVersionExA((OSVERSIONINFOA *)&Ver1))\r
+ {\r
+ memset(&Ver1, 0, sizeof(Ver1));\r
+ Ver1.dwOSVersionInfoSize = sizeof(OSVERSIONINFOA);\r
+ if (!GetVersionExA((OSVERSIONINFOA *)&Ver1))\r
+ {\r
+ return;\r
+ }\r
+ }\r
+ #pragma warning(pop)\r
+\r
+ if (Ver1.dwPlatformId == VER_PLATFORM_WIN32_NT)\r
+ {\r
+ if (Ver1.dwMajorVersion > 6 || (Ver1.dwMajorVersion == 6 && Ver1.dwMinorVersion >= 2))\r
+ {\r
+ // GetVersionEx() has problem on some Windows version \r
+\r
+ MajorEqual = VerSetConditionMask(0, VER_MAJORVERSION, VER_EQUAL);\r
+ for (Major = Ver1.dwMajorVersion; Major <= 9; Major++) \r
+ {\r
+ memset(&Ver2, 0, sizeof(Ver2));\r
+ Ver2.dwOSVersionInfoSize = sizeof(Ver2);\r
+ Ver2.dwMajorVersion = Major;\r
+\r
+ if (!VerifyVersionInfoA(&Ver2, VER_MAJORVERSION, MajorEqual))\r
+ {\r
+ continue;\r
+ }\r
+ \r
+ if (Ver1.dwMajorVersion < Major) \r
+ {\r
+ Ver1.dwMajorVersion = Major;\r
+ Ver1.dwMinorVersion = 0;\r
+ }\r
+\r
+ MinorEqual = VerSetConditionMask(0, VER_MINORVERSION, VER_EQUAL);\r
+ for (Minor = Ver1.dwMinorVersion; Minor <= 9; Minor++) \r
+ {\r
+ memset(&Ver2, 0, sizeof(Ver2)); \r
+ \r
+ Ver2.dwOSVersionInfoSize = sizeof(Ver2);\r
+ Ver2.dwMinorVersion = Minor;\r
+\r
+ if (!VerifyVersionInfoA(&Ver2, VER_MINORVERSION, MinorEqual))\r
+ {\r
+ continue;\r
+ }\r
+ \r
+ Ver1.dwMinorVersion = Minor;\r
+ break;\r
+ }\r
+\r
+ break;\r
+ }\r
+ }\r
+\r
+ if (Ver1.dwMajorVersion <= 0xF && Ver1.dwMinorVersion <= 0xF)\r
+ {\r
+ WsVer = (Ver1.wProductType <= VER_NT_WORKSTATION);\r
+ switch ((Ver1.dwMajorVersion << 4) | Ver2.dwMinorVersion)\r
+ {\r
+ case 0x51:\r
+ {\r
+ Ver = "XP";\r
+ break;\r
+ }\r
+ case 0x52:\r
+ {\r
+ Ver = GetSystemMetrics(89) ? "Server 2003 R2" : "Server 2003";\r
+ break;\r
+ }\r
+ case 0x60:\r
+ {\r
+ Ver = WsVer ? "Vista" : "Server 2008";\r
+ break;\r
+ }\r
+ case 0x61:\r
+ {\r
+ Ver = WsVer ? "7" : "Server 2008 R2";\r
+ break;\r
+ }\r
+ case 0x62:\r
+ {\r
+ Ver = WsVer ? "8" : "Server 2012";\r
+ break;\r
+ }\r
+ case 0x63:\r
+ {\r
+ Ver = WsVer ? "8.1" : "Server 2012 R2";\r
+ break;\r
+ }\r
+ case 0x64:\r
+ {\r
+ Ver = WsVer ? "10 (Preview 1)" : "Server 10 (Preview 1)";\r
+ break;\r
+ }\r
+ case 0xA0:\r
+ {\r
+ Ver = WsVer ? "10" : ((Ver1.dwBuildNumber > 15000) ? "Server 2019" : "Server 2016");\r
+ break;\r
+ }\r
+ default:\r
+ {\r
+ Ver = "10 or later";\r
+ break;\r
+ }\r
+ }\r
+ }\r
+ }\r
+\r
+ Bit = IsWow64() ? 64 : 32;\r
+\r
+ if (Ver1.wServicePackMinor)\r
+ {\r
+ safe_sprintf(WinVer, "Windows %s SP%u.%u %d-bit", Ver, Ver1.wServicePackMajor, Ver1.wServicePackMinor, Bit);\r
+ }\r
+ else if (Ver1.wServicePackMajor)\r
+ {\r
+ safe_sprintf(WinVer, "Windows %s SP%u %d-bit", Ver, Ver1.wServicePackMajor, Bit);\r
+ }\r
+ else\r
+ {\r
+ safe_sprintf(WinVer, "Windows %s %d-bit", Ver, Bit);\r
+ }\r
+\r
+ if (((Ver1.dwMajorVersion << 4) | Ver2.dwMinorVersion) >= 0x62)\r
+ {\r
+ Log("Windows Version : %s (Build %u)", WinVer, Ver1.dwBuildNumber);\r
+ }\r
+ else\r
+ {\r
+ Log("Windows Version : %s", WinVer);\r
+ }\r
+\r
+ return;\r
+}\r
+\r
+BOOL IsVentoyLogicalDrive(CHAR DriveLetter)\r
+{\r
+ int i;\r
+ CONST CHAR *Files[] =\r
+ {\r
+ "EFI\\BOOT\\BOOTX64.EFI",\r
+ "grub\\themes\\ventoy\\theme.txt",\r
+ "ventoy\\ventoy.cpio",\r
+ };\r
+\r
+ for (i = 0; i < sizeof(Files) / sizeof(Files[0]); i++)\r
+ {\r
+ if (!IsFileExist("%C:\\%s", DriveLetter, Files[i]))\r
+ {\r
+ return FALSE;\r
+ }\r
+ }\r
+\r
+ return TRUE;\r
+}\r
+\r
+\r
+int VentoyFillLocation(UINT64 DiskSizeInBytes, UINT32 StartSectorId, UINT32 SectorCount, PART_TABLE *Table)\r
+{\r
+ BYTE Head;\r
+ BYTE Sector;\r
+ BYTE nSector = 63;\r
+ BYTE nHead = 8; \r
+ UINT32 Cylinder;\r
+ UINT32 EndSectorId;\r
+\r
+ while (nHead != 0 && (DiskSizeInBytes / 512 / nSector / nHead) > 1024)\r
+ {\r
+ nHead = (BYTE)nHead * 2;\r
+ }\r
+\r
+ if (nHead == 0)\r
+ {\r
+ nHead = 255;\r
+ }\r
+\r
+ Cylinder = StartSectorId / nSector / nHead;\r
+ Head = StartSectorId / nSector % nHead;\r
+ Sector = StartSectorId % nSector + 1;\r
+\r
+ Table->StartHead = Head;\r
+ Table->StartSector = Sector;\r
+ Table->StartCylinder = Cylinder;\r
+\r
+ EndSectorId = StartSectorId + SectorCount - 1;\r
+ Cylinder = EndSectorId / nSector / nHead;\r
+ Head = EndSectorId / nSector % nHead;\r
+ Sector = EndSectorId % nSector + 1;\r
+\r
+ Table->EndHead = Head;\r
+ Table->EndSector = Sector;\r
+ Table->EndCylinder = Cylinder;\r
+\r
+ Table->StartSectorId = StartSectorId;\r
+ Table->SectorCount = SectorCount;\r
+\r
+ return 0;\r
+}\r
+\r
+int VentoyFillMBR(UINT64 DiskSizeBytes, MBR_HEAD *pMBR, int PartStyle)\r
+{\r
+ GUID Guid;\r
+ int ReservedValue;\r
+ UINT32 DiskSignature;\r
+ UINT32 DiskSectorCount;\r
+ UINT32 PartSectorCount;\r
+ UINT32 PartStartSector;\r
+ UINT32 ReservedSector;\r
+\r
+ VentoyGetLocalBootImg(pMBR);\r
+\r
+ CoCreateGuid(&Guid);\r
+\r
+ memcpy(&DiskSignature, &Guid, sizeof(UINT32));\r
+\r
+ Log("Disk signature: 0x%08x", DiskSignature);\r
+\r
+ *((UINT32 *)(pMBR->BootCode + 0x1B8)) = DiskSignature;\r
+\r
+ if (DiskSizeBytes / 512 > 0xFFFFFFFF)\r
+ {\r
+ DiskSectorCount = 0xFFFFFFFF;\r
+ }\r
+ else\r
+ {\r
+ DiskSectorCount = (UINT32)(DiskSizeBytes / 512);\r
+ }\r
+\r
+ ReservedValue = GetReservedSpaceInMB();\r
+ if (ReservedValue <= 0)\r
+ {\r
+ ReservedSector = 0;\r
+ }\r
+ else\r
+ {\r
+ ReservedSector = (UINT32)(ReservedValue * 2048);\r
+ }\r
+\r
+ if (PartStyle)\r
+ {\r
+ ReservedSector += 33; // backup GPT part table\r
+ }\r
+\r
+ Log("ReservedSector: %u", ReservedSector);\r
+\r
+ //Part1\r
+ PartStartSector = VENTOY_PART1_START_SECTOR;\r
+ PartSectorCount = DiskSectorCount - ReservedSector - VENTOY_EFI_PART_SIZE / 512 - PartStartSector;\r
+ VentoyFillLocation(DiskSizeBytes, PartStartSector, PartSectorCount, pMBR->PartTbl);\r
+\r
+ pMBR->PartTbl[0].Active = 0x80; // bootable\r
+ pMBR->PartTbl[0].FsFlag = 0x07; // exFAT/NTFS/HPFS\r
+\r
+ //Part2\r
+ PartStartSector += PartSectorCount;\r
+ PartSectorCount = VENTOY_EFI_PART_SIZE / 512;\r
+ VentoyFillLocation(DiskSizeBytes, PartStartSector, PartSectorCount, pMBR->PartTbl + 1);\r
+\r
+ pMBR->PartTbl[1].Active = 0x00; \r
+ pMBR->PartTbl[1].FsFlag = 0xEF; // EFI System Partition\r
+\r
+ pMBR->Byte55 = 0x55;\r
+ pMBR->ByteAA = 0xAA;\r
+\r
+ return 0;\r
+}\r
+\r
+\r
+static int VentoyFillProtectMBR(UINT64 DiskSizeBytes, MBR_HEAD *pMBR)\r
+{\r
+ GUID Guid;\r
+ UINT32 DiskSignature;\r
+ UINT64 DiskSectorCount;\r
+\r
+ VentoyGetLocalBootImg(pMBR);\r
+\r
+ CoCreateGuid(&Guid);\r
+\r
+ memcpy(&DiskSignature, &Guid, sizeof(UINT32));\r
+\r
+ Log("Disk signature: 0x%08x", DiskSignature);\r
+\r
+ *((UINT32 *)(pMBR->BootCode + 0x1B8)) = DiskSignature;\r
+\r
+ DiskSectorCount = DiskSizeBytes / 512 - 1;\r
+ if (DiskSectorCount > 0xFFFFFFFF)\r
+ {\r
+ DiskSectorCount = 0xFFFFFFFF;\r
+ }\r
+\r
+ memset(pMBR->PartTbl, 0, sizeof(pMBR->PartTbl));\r
+\r
+ pMBR->PartTbl[0].Active = 0x00;\r
+ pMBR->PartTbl[0].FsFlag = 0xee; // EE\r
+\r
+ pMBR->PartTbl[0].StartHead = 0;\r
+ pMBR->PartTbl[0].StartSector = 1;\r
+ pMBR->PartTbl[0].StartCylinder = 0;\r
+ pMBR->PartTbl[0].EndHead = 254;\r
+ pMBR->PartTbl[0].EndSector = 63;\r
+ pMBR->PartTbl[0].EndCylinder = 1023;\r
+\r
+ pMBR->PartTbl[0].StartSectorId = 1;\r
+ pMBR->PartTbl[0].SectorCount = (UINT32)DiskSectorCount;\r
+\r
+ pMBR->Byte55 = 0x55;\r
+ pMBR->ByteAA = 0xAA;\r
+\r
+ pMBR->BootCode[92] = 0x22;\r
+\r
+ return 0;\r
+}\r
+\r
+\r
+int VentoyFillGpt(UINT64 DiskSizeBytes, VTOY_GPT_INFO *pInfo)\r
+{\r
+ INT64 ReservedValue = 0;\r
+ UINT64 ReservedSector = 33;\r
+ UINT64 Part1SectorCount = 0;\r
+ UINT64 DiskSectorCount = DiskSizeBytes / 512;\r
+ VTOY_GPT_HDR *Head = &pInfo->Head;\r
+ VTOY_GPT_PART_TBL *Table = pInfo->PartTbl;\r
+ static GUID WindowsDataPartType = { 0xebd0a0a2, 0xb9e5, 0x4433, { 0x87, 0xc0, 0x68, 0xb6, 0xb7, 0x26, 0x99, 0xc7 } };\r
+ static GUID EspPartType = { 0xc12a7328, 0xf81f, 0x11d2, { 0xba, 0x4b, 0x00, 0xa0, 0xc9, 0x3e, 0xc9, 0x3b } };\r
+ static GUID BiosGrubPartType = { 0x21686148, 0x6449, 0x6e6f, { 0x74, 0x4e, 0x65, 0x65, 0x64, 0x45, 0x46, 0x49 } };\r
+\r
+ VentoyFillProtectMBR(DiskSizeBytes, &pInfo->MBR);\r
+\r
+ ReservedValue = GetReservedSpaceInMB();\r
+ if (ReservedValue > 0)\r
+ {\r
+ ReservedSector += ReservedValue * 2048;\r
+ }\r
+\r
+ Part1SectorCount = DiskSectorCount - ReservedSector - (VENTOY_EFI_PART_SIZE / 512) - 2048;\r
+\r
+ memcpy(Head->Signature, "EFI PART", 8);\r
+ Head->Version[2] = 0x01;\r
+ Head->Length = 92;\r
+ Head->Crc = 0;\r
+ Head->EfiStartLBA = 1;\r
+ Head->EfiBackupLBA = DiskSectorCount - 1;\r
+ Head->PartAreaStartLBA = 34;\r
+ Head->PartAreaEndLBA = DiskSectorCount - 34;\r
+ CoCreateGuid(&Head->DiskGuid);\r
+ Head->PartTblStartLBA = 2;\r
+ Head->PartTblTotNum = 128;\r
+ Head->PartTblEntryLen = 128;\r
+\r
+\r
+ memcpy(&(Table[0].PartType), &WindowsDataPartType, sizeof(GUID));\r
+ CoCreateGuid(&(Table[0].PartGuid));\r
+ Table[0].StartLBA = 2048;\r
+ Table[0].LastLBA = 2048 + Part1SectorCount - 1;\r
+ Table[0].Attr = 0;\r
+ memcpy(Table[0].Name, L"Ventoy", 6 * 2);\r
+\r
+ // to fix windows issue\r
+ //memcpy(&(Table[1].PartType), &EspPartType, sizeof(GUID));\r
+ memcpy(&(Table[1].PartType), &WindowsDataPartType, sizeof(GUID));\r
+ CoCreateGuid(&(Table[1].PartGuid));\r
+ Table[1].StartLBA = Table[0].LastLBA + 1;\r
+ Table[1].LastLBA = Table[1].StartLBA + VENTOY_EFI_PART_SIZE / 512 - 1;\r
+ Table[1].Attr = 0xC000000000000001ULL;\r
+ memcpy(Table[1].Name, L"VTOYEFI", 7 * 2);\r
+\r
+#if 0\r
+ memcpy(&(Table[2].PartType), &BiosGrubPartType, sizeof(GUID));\r
+ CoCreateGuid(&(Table[2].PartGuid));\r
+ Table[2].StartLBA = 34;\r
+ Table[2].LastLBA = 2047;\r
+ Table[2].Attr = 0;\r
+#endif\r
+\r
+ //Update CRC\r
+ Head->PartTblCrc = VentoyCrc32(Table, sizeof(pInfo->PartTbl));\r
+ Head->Crc = VentoyCrc32(Head, Head->Length);\r
+\r
+ return 0;\r
+}\r
+\r
+int VentoyFillBackupGptHead(VTOY_GPT_INFO *pInfo, VTOY_GPT_HDR *pHead)\r
+{\r
+ UINT64 LBA;\r
+ UINT64 BackupLBA;\r
+\r
+ memcpy(pHead, &pInfo->Head, sizeof(VTOY_GPT_HDR));\r
+\r
+ LBA = pHead->EfiStartLBA;\r
+ BackupLBA = pHead->EfiBackupLBA;\r
+ \r
+ pHead->EfiStartLBA = BackupLBA;\r
+ pHead->EfiBackupLBA = LBA;\r
+ pHead->PartTblStartLBA = BackupLBA + 1 - 33;\r
+\r
+ pHead->Crc = 0;\r
+ pHead->Crc = VentoyCrc32(pHead, pHead->Length);\r
+\r
+ return 0;\r
+}\r
+\r
+CHAR GetFirstUnusedDriveLetter(void)\r
+{\r
+ CHAR Letter = 'D';\r
+ DWORD Drives = GetLogicalDrives();\r
+\r
+ Drives >>= 3;\r
+ while (Drives & 0x1)\r
+ {\r
+ Letter++;\r
+ Drives >>= 1;\r
+ }\r
+\r
+ return Letter;\r
+}\r
+\r
+const CHAR * GetBusTypeString(STORAGE_BUS_TYPE Type)\r
+{\r
+ switch (Type)\r
+ {\r
+ case BusTypeUnknown: return "unknown";\r
+ case BusTypeScsi: return "SCSI";\r
+ case BusTypeAtapi: return "Atapi";\r
+ case BusTypeAta: return "ATA";\r
+ case BusType1394: return "1394";\r
+ case BusTypeSsa: return "SSA";\r
+ case BusTypeFibre: return "Fibre";\r
+ case BusTypeUsb: return "USB";\r
+ case BusTypeRAID: return "RAID";\r
+ case BusTypeiScsi: return "iSCSI";\r
+ case BusTypeSas: return "SAS";\r
+ case BusTypeSata: return "SATA";\r
+ case BusTypeSd: return "SD";\r
+ case BusTypeMmc: return "MMC";\r
+ case BusTypeVirtual: return "Virtual";\r
+ case BusTypeFileBackedVirtual: return "FileBackedVirtual";\r
+ case BusTypeSpaces: return "Spaces";\r
+ case BusTypeNvme: return "Nvme";\r
+ }\r
+ return "unknown";\r
+}\r
+\r
+int VentoyGetLocalBootImg(MBR_HEAD *pMBR)\r
+{\r
+ int Len = 0;\r
+ BYTE *ImgBuf = NULL;\r
+ static int Loaded = 0;\r
+ static MBR_HEAD MBR;\r
+\r
+ if (Loaded)\r
+ {\r
+ memcpy(pMBR, &MBR, 512);\r
+ return 0;\r
+ }\r
+\r
+ if (0 == ReadWholeFileToBuf(VENTOY_FILE_BOOT_IMG, 0, (void **)&ImgBuf, &Len))\r
+ {\r
+ Log("Copy boot img success");\r
+ memcpy(pMBR, ImgBuf, 512);\r
+ free(ImgBuf);\r
+ \r
+ CoCreateGuid((GUID *)(pMBR->BootCode + 0x180));\r
+\r
+ memcpy(&MBR, pMBR, 512);\r
+ Loaded = 1;\r
+\r
+ return 0;\r
+ }\r
+ else\r
+ {\r
+ Log("Copy boot img failed");\r
+ return 1;\r
+ }\r
+}\r
+\r
+int GetHumanReadableGBSize(UINT64 SizeBytes)\r
+{\r
+ int i;\r
+ int Pow2 = 1;\r
+ double Delta;\r
+ double GB = SizeBytes * 1.0 / 1000 / 1000 / 1000;\r
+\r
+ for (i = 0; i < 12; i++)\r
+ {\r
+ if (Pow2 > GB)\r
+ {\r
+ Delta = (Pow2 - GB) / Pow2;\r
+ }\r
+ else\r
+ {\r
+ Delta = (GB - Pow2) / Pow2;\r
+ }\r
+\r
+ if (Delta < 0.05)\r
+ {\r
+ return Pow2;\r
+ }\r
+\r
+ Pow2 <<= 1;\r
+ }\r
+\r
+ return (int)GB;\r
+}\r
+\r
+void TrimString(CHAR *String)\r
+{\r
+ CHAR *Pos1 = String;\r
+ CHAR *Pos2 = String;\r
+ size_t Len = strlen(String);\r
+\r
+ while (Len > 0)\r
+ {\r
+ if (String[Len - 1] != ' ' && String[Len - 1] != '\t')\r
+ {\r
+ break;\r
+ }\r
+ String[Len - 1] = 0;\r
+ Len--;\r
+ }\r
+\r
+ while (*Pos1 == ' ' || *Pos1 == '\t')\r
+ {\r
+ Pos1++;\r
+ }\r
+\r
+ while (*Pos1)\r
+ {\r
+ *Pos2++ = *Pos1++;\r
+ }\r
+ *Pos2++ = 0;\r
+\r
+ return;\r
+}\r
+\r
+int GetRegDwordValue(HKEY Key, LPCSTR SubKey, LPCSTR ValueName, DWORD *pValue)\r
+{\r
+ HKEY hKey;\r
+ DWORD Type;\r
+ DWORD Size;\r
+ LSTATUS lRet;\r
+ DWORD Value;\r
+\r
+ lRet = RegOpenKeyExA(Key, SubKey, 0, KEY_QUERY_VALUE, &hKey);\r
+ Log("RegOpenKeyExA <%s> Ret:%ld", SubKey, lRet);\r
+\r
+ if (ERROR_SUCCESS == lRet)\r
+ {\r
+ Size = sizeof(Value);\r
+ lRet = RegQueryValueExA(hKey, ValueName, NULL, &Type, (LPBYTE)&Value, &Size);\r
+ Log("RegQueryValueExA <%s> ret:%u Size:%u Value:%u", ValueName, lRet, Size, Value);\r
+\r
+ *pValue = Value;\r
+ RegCloseKey(hKey);\r
+\r
+ return 0;\r
+ }\r
+ else\r
+ {\r
+ return 1;\r
+ }\r
+}\r
+\r
+int GetPhysicalDriveCount(void)\r
+{\r
+ DWORD Value;\r
+ int Count = 0;\r
+\r
+ if (GetRegDwordValue(HKEY_LOCAL_MACHINE, "SYSTEM\\CurrentControlSet\\Services\\disk\\Enum", "Count", &Value) == 0)\r
+ {\r
+ Count = (int)Value;\r
+ }\r
+\r
+ Log("GetPhysicalDriveCount: %d", Count);\r
+ return Count;\r
+}\r
+\r
+\r
+\r
projects / Ventoy.git / blobdiff