+int InstallVentoy2FileImage(PHY_DRIVE_INFO *pPhyDrive, int PartStyle)\r
+{\r
+ int i;\r
+ int rc = 1;\r
+ int Len = 0;\r
+ int dataLen = 0;\r
+ UINT size = 0;\r
+ UINT segnum = 0;\r
+ UINT32 chksum = 0;\r
+ UINT64 data_offset = 0;\r
+ UINT64 Part2StartSector = 0;\r
+ UINT64 Part1StartSector = 0;\r
+ UINT64 Part1SectorCount = 0;\r
+ UINT8 *pData = NULL; \r
+ UINT8 *pBkGptPartTbl = NULL;\r
+ BYTE *ImgBuf = NULL;\r
+ MBR_HEAD *pMBR = NULL;\r
+ VTSI_FOOTER *pImgFooter = NULL;\r
+ VTSI_SEGMENT *pSegment = NULL;\r
+ VTOY_GPT_INFO *pGptInfo = NULL;\r
+ VTOY_GPT_HDR *pBkGptHdr = NULL;\r
+ FILE *fp = NULL;\r
+\r
+ Log("InstallVentoy2FileImage %s PhyDrive%d <<%s %s %dGB>>",\r
+ PartStyle ? "GPT" : "MBR", pPhyDrive->PhyDrive, pPhyDrive->VendorId, pPhyDrive->ProductId,\r
+ GetHumanReadableGBSize(pPhyDrive->SizeInBytes));\r
+\r
+ PROGRESS_BAR_SET_POS(PT_LOCK_FOR_CLEAN);\r
+\r
+ size = SIZE_1MB + VENTOY_EFI_PART_SIZE + 33 * 512 + VTSI_IMG_MAX_SEG * sizeof(VTSI_SEGMENT) + sizeof(VTSI_FOOTER);\r
+\r
+ pData = (UINT8 *)malloc(size);\r
+ if (!pData)\r
+ {\r
+ Log("malloc image buffer failed %d.", size);\r
+ goto End;\r
+ }\r
+\r
+ pImgFooter = (VTSI_FOOTER *)(pData + size - sizeof(VTSI_FOOTER));\r
+ pSegment = (VTSI_SEGMENT *)((UINT8 *)pImgFooter - VTSI_IMG_MAX_SEG * sizeof(VTSI_SEGMENT));\r
+ memset(pImgFooter, 0, sizeof(VTSI_FOOTER));\r
+ memset(pSegment, 0, VTSI_IMG_MAX_SEG * sizeof(VTSI_SEGMENT));\r
+\r
+ PROGRESS_BAR_SET_POS(PT_WRITE_VENTOY_START);\r
+\r
+ Log("Writing Boot Image ............................. ");\r
+ if (ReadWholeFileToBuf(VENTOY_FILE_STG1_IMG, 0, (void **)&ImgBuf, &Len))\r
+ {\r
+ Log("Failed to read stage1 img");\r
+ goto End;\r
+ }\r
+\r
+ unxz(ImgBuf, Len, NULL, NULL, pData, &dataLen, unxz_error);\r
+ SAFE_FREE(ImgBuf);\r
+\r
+ if (PartStyle)\r
+ {\r
+ pGptInfo = (VTOY_GPT_INFO *)pData;\r
+ memset(pGptInfo, 0, sizeof(VTOY_GPT_INFO));\r
+ VentoyFillGpt(pPhyDrive->SizeInBytes, pGptInfo);\r
+\r
+ pBkGptPartTbl = pData + SIZE_1MB + VENTOY_EFI_PART_SIZE;\r
+ memset(pBkGptPartTbl, 0, 33 * 512);\r
+\r
+ memcpy(pBkGptPartTbl, pGptInfo->PartTbl, 32 * 512);\r
+ pBkGptHdr = (VTOY_GPT_HDR *)(pBkGptPartTbl + 32 * 512);\r
+ VentoyFillBackupGptHead(pGptInfo, pBkGptHdr);\r
+\r
+ Part1StartSector = pGptInfo->PartTbl[0].StartLBA;\r
+ Part1SectorCount = pGptInfo->PartTbl[0].LastLBA - Part1StartSector + 1;\r
+ Part2StartSector = pGptInfo->PartTbl[1].StartLBA;\r
+\r
+ Log("Write GPT Info OK ...");\r
+ }\r
+ else\r
+ {\r
+ pMBR = (MBR_HEAD *)pData;\r
+ VentoyFillMBR(pPhyDrive->SizeInBytes, pMBR, PartStyle);\r
+ Part1StartSector = pMBR->PartTbl[0].StartSectorId;\r
+ Part1SectorCount = pMBR->PartTbl[0].SectorCount;\r
+ Part2StartSector = pMBR->PartTbl[1].StartSectorId;\r
+\r
+ Log("Write MBR OK ...");\r
+ }\r
+\r
+ Log("Writing EFI part Image ............................. ");\r
+ rc = ReadWholeFileToBuf(VENTOY_FILE_DISK_IMG, 0, (void **)&ImgBuf, &Len);\r
+ if (rc)\r
+ {\r
+ Log("Failed to read img file %p %u", ImgBuf, Len);\r
+ goto End;\r
+ }\r
+\r
+ PROGRESS_BAR_SET_POS(PT_WRITE_VENTOY_START + 28);\r
+ memset(g_part_img_buf, 0, sizeof(g_part_img_buf));\r
+ unxz(ImgBuf, Len, NULL, NULL, pData + SIZE_1MB, &dataLen, unxz_error);\r
+ if (dataLen == Len)\r
+ {\r
+ Log("decompress finished success");\r
+ g_part_img_buf[0] = pData + SIZE_1MB;\r
+\r
+ VentoyProcSecureBoot(g_SecureBoot);\r
+ }\r
+ else\r
+ {\r
+ Log("decompress finished failed");\r
+ goto End;\r
+ }\r
+\r
+ fopen_s(&fp, "VentoySparseImg.vtsi", "wb+");\r
+ if (!fp)\r
+ {\r
+ Log("Failed to create Ventoy img file");\r
+ goto End;\r
+ }\r
+\r
+ Log("Writing stage1 data ............................. ");\r
+ fwrite(pData, 1, SIZE_1MB, fp);\r
+ pSegment[0].disk_start_sector = 0;\r
+ pSegment[0].sector_num = SIZE_1MB / 512;\r
+ pSegment[0].data_offset = data_offset;\r
+ data_offset += pSegment[0].sector_num * 512;\r
+\r
+ disk_io_set_param(INVALID_HANDLE_VALUE, Part1StartSector + Part1SectorCount);// include the 2048 sector gap\r
+ disk_io_set_imghook(fp, pSegment + 1, VTSI_IMG_MAX_SEG - 1, data_offset);\r
+\r
+ Log("Formatting part1 exFAT ...");\r
+ if (0 != FormatPart1exFAT(pPhyDrive->SizeInBytes))\r
+ {\r
+ Log("FormatPart1exFAT failed.");\r
+ disk_io_reset_imghook(&segnum, &data_offset);\r
+ goto End;\r
+ }\r
+\r
+ disk_io_reset_imghook(&segnum, &data_offset);\r
+ segnum++;\r
+\r
+ Log("current segment number:%d dataoff:%ld", segnum, (long)data_offset);\r
+\r
+ //write data\r
+ Log("Writing part2 data ............................. ");\r
+ fwrite(pData + SIZE_1MB, 1, VENTOY_EFI_PART_SIZE, fp);\r
+ pSegment[segnum].disk_start_sector = Part2StartSector;\r
+ pSegment[segnum].sector_num = VENTOY_EFI_PART_SIZE / 512;\r
+ pSegment[segnum].data_offset = data_offset;\r
+ data_offset += pSegment[segnum].sector_num * 512;\r
+ segnum++;\r
+\r
+ if (PartStyle)\r
+ {\r
+ Log("Writing backup gpt table ............................. ");\r
+ fwrite(pBkGptPartTbl, 1, 33 * 512, fp);\r
+ pSegment[segnum].disk_start_sector = pPhyDrive->SizeInBytes / 512 - 33;\r
+ pSegment[segnum].sector_num = 33;\r
+ pSegment[segnum].data_offset = data_offset;\r
+ data_offset += pSegment[segnum].sector_num * 512;\r
+ segnum++;\r
+ }\r
+\r
+ Log("Writing segment metadata ............................. ");\r
+\r
+ for (i = 0; i < (int)segnum; i++)\r
+ {\r
+ Log("SEG[%d]: PhySector:%llu SectorNum:%llu DataOffset:%llu(sector:%llu)", i, pSegment[i].disk_start_sector, pSegment[i].sector_num,\r
+ pSegment[i].data_offset, pSegment[i].data_offset / 512);\r
+ }\r
+\r
+ dataLen = segnum * sizeof(VTSI_SEGMENT);\r
+ fwrite(pSegment, 1, dataLen, fp);\r
+\r
+ if (dataLen % 512)\r
+ {\r
+ //pData + SIZE_1MB - 8192 is a temp data buffer with zero\r
+ fwrite(pData + SIZE_1MB - 8192, 1, 512 - (dataLen % 512), fp);\r
+ }\r
+\r
+ //Fill footer\r
+ pImgFooter->magic = VTSI_IMG_MAGIC;\r
+ pImgFooter->version = 1;\r
+ pImgFooter->disk_size = pPhyDrive->SizeInBytes;\r
+ memcpy(&pImgFooter->disk_signature, pPhyDrive->MBR.BootCode + 0x1b8, 4);\r
+ pImgFooter->segment_num = segnum;\r
+ pImgFooter->segment_offset = data_offset;\r
+\r
+ for (i = 0, chksum = 0; i < (int)(segnum * sizeof(VTSI_SEGMENT)); i++)\r
+ {\r
+ chksum += *((UINT8 *)pSegment + i);\r
+ }\r
+ pImgFooter->segment_chksum = ~chksum;\r
+\r
+ for (i = 0, chksum = 0; i < sizeof(VTSI_FOOTER); i++)\r
+ {\r
+ chksum += *((UINT8 *)pImgFooter + i);\r
+ }\r
+ pImgFooter->foot_chksum = ~chksum;\r
+\r
+ Log("Writing footer segnum(%u) segoffset(%llu) ......................", segnum, data_offset);\r
+ Log("disk_size=%llu disk_signature=%lx segment_offset=%llu", pImgFooter->disk_size, pImgFooter->disk_signature, pImgFooter->segment_offset);\r
+\r
+ fwrite(pImgFooter, 1, sizeof(VTSI_FOOTER), fp);\r
+ fclose(fp);\r
+\r
+ Log("Writing Ventoy image file finished, the file size should be %llu .", data_offset + 512 + ((dataLen + 511) / 512 * 512));\r
+\r
+ rc = 0;\r
+\r
+End:\r
+\r
+ PROGRESS_BAR_SET_POS(PT_MOUNT_VOLUME);\r
+\r
+ Log("retcode:%d\n", rc);\r
+\r
+ SAFE_FREE(pData);\r
+ SAFE_FREE(ImgBuf);\r
+ \r
+ return rc;\r
+}\r
+\r
+\r
projects / Ventoy.git / commitdiff