summaryrefslogtreecommitdiffstats
path: root/src/bldprogs/VBoxPeSetVersion.cpp
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--src/bldprogs/VBoxPeSetVersion.cpp201
1 files changed, 198 insertions, 3 deletions
diff --git a/src/bldprogs/VBoxPeSetVersion.cpp b/src/bldprogs/VBoxPeSetVersion.cpp
index 0c17c971..2ce8ac5f 100644
--- a/src/bldprogs/VBoxPeSetVersion.cpp
+++ b/src/bldprogs/VBoxPeSetVersion.cpp
@@ -51,6 +51,8 @@
*********************************************************************************************************************************/
static const char *g_pszFilename;
static unsigned g_cVerbosity = 0;
+static enum { kUpdateCheckSum_Never, kUpdateCheckSum_WhenNeeded, kUpdateCheckSum_Always, kUpdateCheckSum_Zero }
+ g_enmUpdateChecksum = kUpdateCheckSum_WhenNeeded;
static int Error(const char *pszFormat, ...)
@@ -79,8 +81,164 @@ static void Info(unsigned iLevel, const char *pszFormat, ...)
}
+/**
+ * File size.
+ *
+ * @returns file size in bytes.
+ * @returns 0 on failure.
+ * @param pFile File to size.
+ */
+static size_t fsize(FILE *pFile)
+{
+ long cbFile;
+ off_t Pos = ftell(pFile);
+ if ( Pos >= 0
+ && !fseek(pFile, 0, SEEK_END))
+ {
+ cbFile = ftell(pFile);
+ if ( cbFile >= 0
+ && !fseek(pFile, 0, SEEK_SET))
+ return cbFile;
+ }
+ return 0;
+}
+
+
+/**
+ * Calculates a raw PE-style checksum on a plain buffer.
+ *
+ * ASSUMES pvBuf is dword aligned.
+ */
+static uint16_t CalcRawPeChecksum(void const *pvBuf, size_t cb, uint32_t uChksum)
+{
+ /*
+ * Work thru the memory in 64 bits at a time.
+ * ASSUMES well aligned input.
+ */
+ uint64_t uBigSum = uChksum;
+ uint64_t const *pu64 = (uint64_t const *)pvBuf;
+ size_t cQWords = cb / 8;
+ while (cQWords-- > 0)
+ {
+ /* We emulate add with carry here. */
+ uint64_t uTmp = uBigSum + *pu64++;
+ uBigSum = uTmp >= uBigSum ? uTmp : uTmp + 1;
+ }
+
+ /*
+ * Zeropadd any remaining bytes before adding them.
+ */
+ if (cb & 7)
+ {
+ uint8_t const *pb = (uint8_t const *)pu64;
+ uint64_t uQWord = 0;
+ switch (cb & 7)
+ {
+ case 7: uQWord |= (uint64_t)pb[6] << 48; RT_FALL_THRU();
+ case 6: uQWord |= (uint64_t)pb[5] << 40; RT_FALL_THRU();
+ case 5: uQWord |= (uint64_t)pb[4] << 32; RT_FALL_THRU();
+ case 4: uQWord |= (uint64_t)pb[3] << 24; RT_FALL_THRU();
+ case 3: uQWord |= (uint64_t)pb[2] << 16; RT_FALL_THRU();
+ case 2: uQWord |= (uint64_t)pb[1] << 8; RT_FALL_THRU();
+ case 1: uQWord |= (uint64_t)pb[0]; break;
+ }
+
+ uint64_t uTmp = uBigSum + uQWord;
+ uBigSum = uTmp >= uBigSum ? uTmp : uTmp + 1;
+ }
+
+ /*
+ * Convert the 64-bit checksum to a 16-bit one.
+ */
+ uChksum = (uBigSum & 0xffffU)
+ + ((uBigSum >> 16) & 0xffffU)
+ + ((uBigSum >> 32) & 0xffffU)
+ + ((uBigSum >> 48) & 0xffffU);
+ uChksum = (uChksum & 0xffffU)
+ + (uChksum >> 16);
+ if (uChksum > 0xffffU) Error("Checksum IPE#1");
+ return uChksum & 0xffffU;
+}
+
+
+static int UpdateChecksum(FILE *pFile)
+{
+ /*
+ * Read the whole file into memory.
+ */
+ size_t const cbFile = fsize(pFile);
+ if (!cbFile)
+ return Error("Failed to determine file size: %s", strerror(errno));
+ uint8_t * const pbFile = (uint8_t *)malloc(cbFile + 4);
+ if (!pbFile)
+ return Error("Failed to allocate %#lx bytes for checksum calculations", (unsigned long)(cbFile + 4U));
+ memset(pbFile, 0, cbFile + 4);
+
+ int rcExit;
+ size_t cItemsRead = fread(pbFile, cbFile, 1, pFile);
+ if (cItemsRead == 1)
+ {
+ /*
+ * Locate the NT headers as we need the CheckSum field in order to update it.
+ * It has the same location in 32-bit and 64-bit images.
+ */
+ IMAGE_DOS_HEADER const * const pMzHdr = (IMAGE_DOS_HEADER const *)pbFile;
+ AssertCompileMembersAtSameOffset(IMAGE_NT_HEADERS32, OptionalHeader.CheckSum, IMAGE_NT_HEADERS64, OptionalHeader.CheckSum);
+ IMAGE_NT_HEADERS32 * const pNtHdrs
+ = (IMAGE_NT_HEADERS32 *)&pbFile[pMzHdr->e_magic == IMAGE_DOS_SIGNATURE ? pMzHdr->e_lfanew : 0];
+ if ((uintptr_t)&pNtHdrs->OptionalHeader.DataDirectory[0] - (uintptr_t)pbFile < cbFile)
+ {
+ if ( pNtHdrs->Signature == IMAGE_NT_SIGNATURE
+ && ( pNtHdrs->OptionalHeader.Magic == IMAGE_NT_OPTIONAL_HDR32_MAGIC
+ || pNtHdrs->OptionalHeader.Magic == IMAGE_NT_OPTIONAL_HDR64_MAGIC))
+ {
+ /*
+ * Set the checksum field to zero to avoid having to do tedious
+ * adjustments of the raw checksum. Then calculate the raw check sum.
+ */
+ pNtHdrs->OptionalHeader.CheckSum = 0;
+ uint32_t uChksum = CalcRawPeChecksum(pbFile, RT_ALIGN_Z(cbFile, 2), 0);
+
+ /* Finalize the checksum by adding the image size to it. */
+ uChksum += (uint32_t)cbFile;
+ pNtHdrs->OptionalHeader.CheckSum = uChksum;
+
+ /*
+ * Write back the checksum to the file.
+ */
+ size_t const offChecksumField = (uintptr_t)&pNtHdrs->OptionalHeader.CheckSum - (uintptr_t)pbFile;
+
+ if (fseek(pFile, (long)offChecksumField, SEEK_SET) == 0)
+ {
+ if (fwrite(&pNtHdrs->OptionalHeader.CheckSum, sizeof(pNtHdrs->OptionalHeader.CheckSum), 1, pFile) == 1)
+ {
+ Info(1, "Checksum: %#x", uChksum);
+ rcExit = RTEXITCODE_SUCCESS;
+ }
+ else
+ rcExit = Error("Checksum write failed");
+ }
+ else
+ rcExit = Error("Failed seeking to %#lx for checksum write", (long)offChecksumField);
+ }
+ else
+ rcExit = Error("PE header not found");
+ }
+ else
+ rcExit = Error("NT headers not within file when checksumming");
+ }
+ else
+ rcExit = Error("Failed read in file (%#lx bytes) for checksum calculations", (unsigned long)(cbFile + 4U));
+
+ free(pbFile);
+ return rcExit;
+}
+
+
static int UpdateFile(FILE *pFile, unsigned uNtVersion, PIMAGE_SECTION_HEADER *ppaShdr)
{
+ unsigned cFileModifications = 0;
+
/*
* Locate and read the PE header.
*
@@ -181,7 +339,10 @@ static int UpdateFile(FILE *pFile, unsigned uNtVersion, PIMAGE_SECTION_HEADER *p
return Error("Failed to seek to PE header at %#lx: %s", offNtHdrs, strerror(errno));
if (fwrite(&NtHdrsNew, cbNewHdrs, 1, pFile) != 1)
return Error("Failed to write PE header at %#lx: %s", offNtHdrs, strerror(errno));
+ cFileModifications++;
}
+ else
+ Info(3, "No header changes");
/*
* Make the IAT writable for NT 3.1 and drop the non-cachable flag from .bss.
@@ -245,6 +406,7 @@ static int UpdateFile(FILE *pFile, unsigned uNtVersion, PIMAGE_SECTION_HEADER *p
if (fwrite(&paShdrs[i], sizeof(IMAGE_SECTION_HEADER), 1, pFile) != 1)
return Error("Failed to write %8.8s section header header at %#lx: %s",
paShdrs[i].Name, offShdr, strerror(errno));
+ cFileModifications++;
if (uRvaIat == UINT32_MAX && fFoundBss)
break;
}
@@ -252,7 +414,27 @@ static int UpdateFile(FILE *pFile, unsigned uNtVersion, PIMAGE_SECTION_HEADER *p
/* Advance */
uRvaEnd = paShdrs[i].VirtualAddress;
}
+ }
+ /*
+ * Recalculate the checksum if we changed anything or if it is zero.
+ */
+ if ( g_enmUpdateChecksum == kUpdateCheckSum_Always
+ || ( g_enmUpdateChecksum == kUpdateCheckSum_WhenNeeded
+ && (cFileModifications || NtHdrsNew.x32.OptionalHeader.CheckSum == 0)))
+ return UpdateChecksum(pFile);
+
+ /* Zero the checksum if explicitly requested. */
+ if ( g_enmUpdateChecksum == kUpdateCheckSum_Zero
+ && NtHdrsNew.x32.OptionalHeader.CheckSum != 0)
+ {
+ unsigned long const offCheckSumField = offNtHdrs + RT_UOFFSETOF(IMAGE_NT_HEADERS32, OptionalHeader.CheckSum);
+ if (fseek(pFile, offCheckSumField, SEEK_SET) != 0)
+ return Error("Failed to seek to the CheckSum field in the PE at %#lx: %s", offCheckSumField, strerror(errno));
+
+ NtHdrsNew.x32.OptionalHeader.CheckSum = 0;
+ if (fwrite(&NtHdrsNew.x32.OptionalHeader.CheckSum, sizeof(NtHdrsNew.x32.OptionalHeader.CheckSum), 1, pFile) != 1)
+ return Error("Failed to write the CheckSum field in the PE header at %#lx: %s", offCheckSumField, strerror(errno));
}
return RTEXITCODE_SUCCESS;
@@ -270,7 +452,10 @@ static int Usage(FILE *pOutput)
" Quiet operation (default).\n"
" --nt31, --nt350, --nt351, --nt4, --w2k, --xp, --w2k3, --vista,\n"
" --w7, --w8, --w81, --w10\n"
- " Which version to set. Default: --nt31\n"
+ " Which version to set. Default: --nt31 (x86), --w2k3 (amd64)\n"
+ " --update-checksum-when-needed, --always-update-checksum,\n"
+ " --never-update-checksum, --zero-checksum:\n"
+ " Checksum updating. Default: --update-checksum-when-needed\n"
);
return RTEXITCODE_SYNTAX;
}
@@ -311,7 +496,9 @@ int main(int argc, char **argv)
ch = 'V';
else
{
- if (strcmp(psz, "nt31") == 0)
+ if (strcmp(psz, "default") == 0)
+ uNtVersion = 0;
+ else if (strcmp(psz, "nt31") == 0)
uNtVersion = MK_VER(3,10);
else if (strcmp(psz, "nt350") == 0)
uNtVersion = MK_VER(3,50);
@@ -323,7 +510,7 @@ int main(int argc, char **argv)
uNtVersion = MK_VER(5,0);
else if (strcmp(psz, "xp") == 0)
uNtVersion = MK_VER(5,1);
- else if (strcmp(psz, "w2k3") == 0)
+ else if (strcmp(psz, "w2k3") == 0 || strcmp(psz, "xp64") == 0)
uNtVersion = MK_VER(5,2);
else if (strcmp(psz, "vista") == 0)
uNtVersion = MK_VER(6,0);
@@ -335,6 +522,14 @@ int main(int argc, char **argv)
uNtVersion = MK_VER(6,3);
else if (strcmp(psz, "w10") == 0)
uNtVersion = MK_VER(10,0);
+ else if (strcmp(psz, "always-update-checksum") == 0)
+ g_enmUpdateChecksum = kUpdateCheckSum_Always;
+ else if (strcmp(psz, "never-update-checksum") == 0)
+ g_enmUpdateChecksum = kUpdateCheckSum_Never;
+ else if (strcmp(psz, "update-checksum-when-needed") == 0)
+ g_enmUpdateChecksum = kUpdateCheckSum_WhenNeeded;
+ else if (strcmp(psz, "zero-checksum") == 0)
+ g_enmUpdateChecksum = kUpdateCheckSum_Zero;
else
{
fprintf(stderr, "VBoxPeSetVersion: syntax error: Unknown option: --%s\n", psz);
generated by cgit v1.2.3 (git 2.39.1) at 2024-11-12 21:59:11 +0000