mirror of https://github.com/tongzx/nt5src
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
2485 lines
81 KiB
2485 lines
81 KiB
/******************************************************************************
|
|
|
|
PROGRAM: updres.c
|
|
|
|
PURPOSE: Contains API Entry points and routines for updating resource
|
|
sections in exe/dll
|
|
|
|
FUNCTIONS:
|
|
|
|
EndUpdateResource(HANDLE, BOOL) - end update, write changes
|
|
UpdateResource(HANDLE, LPSTR, LPSTR, WORD, PVOID)
|
|
- update individual resource
|
|
BeginUpdateResource(LPSTR) - begin update
|
|
|
|
*******************************************************************************/
|
|
|
|
|
|
#include "precomp.h"
|
|
#include "updres.h"
|
|
|
|
#define cbPadMax 16L
|
|
char *pchPad = "PADDINGXXPADDING";
|
|
char *pchZero = "\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0";
|
|
|
|
|
|
/****************************************************************************
|
|
**
|
|
** API entry points
|
|
**
|
|
****************************************************************************/
|
|
|
|
|
|
HANDLE BeginUpdateResourceInternalW(LPCWSTR pwch, BOOL bDeleteExistingResources)
|
|
|
|
/*++
|
|
Routine Description
|
|
Begins an update of resources. Save away the name
|
|
and current resources in a list, using EnumResourceXxx
|
|
api set.
|
|
|
|
Parameters:
|
|
|
|
lpFileName - Supplies the name of the executable file that the
|
|
resource specified by lpType/lpName/language will be updated
|
|
in. This file must be able to be opened for writing (ie, not
|
|
currently executing, etc.) The file may be fully qualified,
|
|
or if not, the current directory is assumed. It must be a
|
|
valid Windows executable file.
|
|
|
|
bDeleteExistingResources - if TRUE, existing resources are
|
|
deleted, and only new resources will appear in the result.
|
|
Otherwise, all resources in the input file will be in the
|
|
output file unless specifically deleted or replaced.
|
|
|
|
Return Value:
|
|
|
|
NULL - The file specified was not able to be opened for writing.
|
|
Either it was not an executable image, the executable image is
|
|
already loaded, or the filename did not exist. More information may
|
|
be available via GetLastError api.
|
|
|
|
HANDLE - A handle to be passed to the UpdateResource and
|
|
EndUpdateResources function.
|
|
--*/
|
|
|
|
{
|
|
HMODULE hModule;
|
|
PUPDATEDATA pUpdate;
|
|
HANDLE hUpdate;
|
|
LPWSTR pFileName;
|
|
DWORD attr;
|
|
|
|
SetLastError(NO_ERROR);
|
|
if (pwch == NULL)
|
|
{
|
|
SetLastError(ERROR_INVALID_PARAMETER);
|
|
return NULL;
|
|
}
|
|
|
|
hUpdate = GlobalAlloc(GHND, sizeof(UPDATEDATA));
|
|
if (hUpdate == NULL)
|
|
{
|
|
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
|
|
return NULL;
|
|
}
|
|
pUpdate = (PUPDATEDATA)GlobalLock(hUpdate);
|
|
if (pUpdate == NULL)
|
|
{
|
|
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
|
|
return NULL;
|
|
}
|
|
|
|
pUpdate->Status = NO_ERROR;
|
|
pUpdate->hFileName = GlobalAlloc(GHND, (gwcslen(pwch)+1)*sizeof(WCHAR));
|
|
if (pUpdate->hFileName == NULL)
|
|
{
|
|
GlobalUnlock(hUpdate);
|
|
GlobalFree(hUpdate);
|
|
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
|
|
return NULL;
|
|
}
|
|
pFileName = (LPWSTR)GlobalLock(pUpdate->hFileName);
|
|
if (pFileName == NULL)
|
|
{
|
|
GlobalUnlock(hUpdate);
|
|
GlobalFree(hUpdate);
|
|
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
|
|
return NULL;
|
|
}
|
|
gwcscpy(pFileName, pwch);
|
|
GlobalUnlock(pUpdate->hFileName);
|
|
|
|
attr = GetFileAttributesW(pFileName);
|
|
if (attr == 0xffffffff)
|
|
{
|
|
GlobalUnlock(hUpdate);
|
|
GlobalFree(hUpdate);
|
|
return NULL;
|
|
}
|
|
else if (attr & (FILE_ATTRIBUTE_READONLY |
|
|
FILE_ATTRIBUTE_SYSTEM |
|
|
FILE_ATTRIBUTE_HIDDEN |
|
|
FILE_ATTRIBUTE_DIRECTORY))
|
|
{
|
|
GlobalUnlock(hUpdate);
|
|
GlobalFree(hUpdate);
|
|
SetLastError(ERROR_WRITE_PROTECT);
|
|
return NULL;
|
|
}
|
|
|
|
if (!bDeleteExistingResources)
|
|
{
|
|
hModule = LoadLibraryExW(pwch, NULL,LOAD_LIBRARY_AS_DATAFILE| DONT_RESOLVE_DLL_REFERENCES);
|
|
if (hModule == NULL)
|
|
{
|
|
GlobalUnlock(hUpdate);
|
|
GlobalFree(hUpdate);
|
|
if (GetLastError() == NO_ERROR)
|
|
SetLastError(ERROR_BAD_EXE_FORMAT);
|
|
return NULL;
|
|
}
|
|
else
|
|
EnumResourceTypesW(hModule, (ENUMRESTYPEPROCW)EnumTypesFunc, (LONG)pUpdate);
|
|
FreeLibrary(hModule);
|
|
}
|
|
|
|
if (pUpdate->Status != NO_ERROR)
|
|
{
|
|
GlobalUnlock(hUpdate);
|
|
GlobalFree(hUpdate);
|
|
// return code set by enum functions
|
|
return NULL;
|
|
}
|
|
GlobalUnlock(hUpdate);
|
|
return hUpdate;
|
|
}
|
|
|
|
BOOL UpdateResourceInternalW(HANDLE hUpdate, LPCWSTR lpType, LPCWSTR lpName,
|
|
WORD language, LPVOID lpData, ULONG cb)
|
|
|
|
/*++
|
|
Routine Description
|
|
This routine adds, deletes or modifies the input resource
|
|
in the list initialized by BeginUpdateResource. The modify
|
|
case is simple, the add is easy, the delete is hard.
|
|
The ASCII entry point converts inputs to UNICODE.
|
|
|
|
Parameters:
|
|
|
|
hUpdateFile - The handle returned by the BeginUpdateResources
|
|
function.
|
|
|
|
lpType - Points to a null-terminated character string that
|
|
represents the type name of the resource to be updated or
|
|
added. May be an integer value passed to MAKEINTRESOURCE
|
|
macro. For predefined resource types, the lpType parameter
|
|
should be one of the following values:
|
|
|
|
RT_ACCELERATOR - Accelerator table
|
|
RT_BITMAP - Bitmap resource
|
|
RT_DIALOG - Dialog box
|
|
RT_FONT - Font resource
|
|
RT_FONTDIR - Font directory resource
|
|
RT_MENU - Menu resource
|
|
RT_RCDATA - User-defined resource (raw data)
|
|
RT_VERSION - Version resource
|
|
RT_ICON - Icon resource
|
|
RT_CURSOR - Cursor resource
|
|
|
|
|
|
|
|
lpName - Points to a null-terminated character string that
|
|
represents the name of the resource to be updated or added.
|
|
May be an integer value passed to MAKEINTRESOURCE macro.
|
|
|
|
language - Is the word value that specifies the language of the
|
|
resource to be updated. A complete list of values is
|
|
available in winnls.h.
|
|
|
|
lpData - A pointer to the raw data to be inserted into the
|
|
executable image's resource table and data. If the data is
|
|
one of the predefined types, it must be valid and properly
|
|
aligned. If lpData is NULL, the specified resource is to be
|
|
deleted from the executable image.
|
|
|
|
cb - count of bytes in the data.
|
|
|
|
Return Value:
|
|
|
|
TRUE - The resource specified was successfully replaced in, or added
|
|
to, the specified executable image.
|
|
|
|
FALSE/NULL - The resource specified was not successfully added to or
|
|
updated in the executable image. More information may be available
|
|
via GetLastError api.
|
|
--*/
|
|
{
|
|
PUPDATEDATA pUpdate;
|
|
PSDATA Type;
|
|
PSDATA Name;
|
|
PVOID lpCopy;
|
|
LONG fRet;
|
|
|
|
SetLastError(0);
|
|
pUpdate = (PUPDATEDATA)GlobalLock(hUpdate);
|
|
Name = AddStringOrID(lpName, pUpdate);
|
|
if (Name == NULL)
|
|
{
|
|
pUpdate->Status = ERROR_NOT_ENOUGH_MEMORY;
|
|
GlobalUnlock(hUpdate);
|
|
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
|
|
return FALSE;
|
|
}
|
|
Type = AddStringOrID(lpType, pUpdate);
|
|
if (Type == NULL)
|
|
{
|
|
pUpdate->Status = ERROR_NOT_ENOUGH_MEMORY;
|
|
GlobalUnlock(hUpdate);
|
|
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
|
|
return FALSE;
|
|
}
|
|
if (cb == 0)
|
|
{
|
|
lpCopy = NULL;
|
|
}
|
|
else
|
|
{
|
|
lpCopy = GodotHeapAlloc(cb);
|
|
if (lpCopy == NULL)
|
|
{
|
|
pUpdate->Status = ERROR_NOT_ENOUGH_MEMORY;
|
|
GlobalUnlock(hUpdate);
|
|
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
|
|
return FALSE;
|
|
}
|
|
RtlCopyMemory(lpCopy, lpData, cb);
|
|
}
|
|
fRet = AddResource(Type, Name, language, pUpdate, lpCopy, cb);
|
|
GlobalUnlock(hUpdate);
|
|
if (fRet == NO_ERROR)
|
|
return TRUE;
|
|
else
|
|
{
|
|
SetLastError(fRet);
|
|
if (lpData != NULL)
|
|
GodotHeapFree(lpData);
|
|
return FALSE;
|
|
}
|
|
}
|
|
|
|
/*++
|
|
Routine Description
|
|
Finishes the UpdateResource action. Copies the
|
|
input file to a temporary, adds the resources left
|
|
in the list (hUpdate) to the exe.
|
|
|
|
Parameters:
|
|
|
|
hUpdateFile - The handle returned by the BeginUpdateResources
|
|
function.
|
|
|
|
fDiscard - If TRUE, discards all the updates, frees all memory.
|
|
|
|
Return Value:
|
|
|
|
FALSE - The file specified was not able to be written. More
|
|
information may be available via GetLastError api.
|
|
|
|
TRUE - The accumulated resources specified by UpdateResource calls
|
|
were written to the executable file specified by the hUpdateFile
|
|
handle.
|
|
--*/
|
|
BOOL EndUpdateResourceInternalW(HANDLE hUpdate, BOOL fDiscard)
|
|
{
|
|
LPWSTR pFileName;
|
|
PUPDATEDATA pUpdate;
|
|
WCHAR pTempFileName[MAX_PATH];
|
|
INT cch;
|
|
LPWSTR p;
|
|
LONG rc;
|
|
DWORD LastError = 0;
|
|
|
|
SetLastError(0);
|
|
pUpdate = (PUPDATEDATA)GlobalLock(hUpdate);
|
|
if (fDiscard)
|
|
{
|
|
rc = NO_ERROR;
|
|
}
|
|
else
|
|
{
|
|
pFileName = (LPWSTR)GlobalLock(pUpdate->hFileName);
|
|
gwcscpy(pTempFileName, pFileName);
|
|
cch = gwcslen(pTempFileName);
|
|
p = pTempFileName + cch;
|
|
while (*p != L'\\' && p >= pTempFileName)
|
|
p--;
|
|
*(p+1) = 0;
|
|
rc = GetTempFileNameW(pTempFileName, L"RCX", 0, pTempFileName);
|
|
if (rc == 0)
|
|
{
|
|
rc = GetTempPathW(MAX_PATH, pTempFileName);
|
|
if (rc == 0)
|
|
{
|
|
pTempFileName[0] = L'.';
|
|
pTempFileName[1] = L'\\';
|
|
pTempFileName[2] = 0;
|
|
}
|
|
rc = GetTempFileNameW(pTempFileName, L"RCX", 0, pTempFileName);
|
|
if (rc == 0)
|
|
{
|
|
rc = GetLastError();
|
|
}
|
|
else
|
|
{
|
|
rc = WriteResFile(hUpdate, pTempFileName);
|
|
if (rc == NO_ERROR)
|
|
{
|
|
DeleteFileW(pFileName);
|
|
MoveFileW(pTempFileName, pFileName);
|
|
}
|
|
else
|
|
{
|
|
LastError = rc;
|
|
DeleteFileW(pTempFileName);
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
rc = WriteResFile(hUpdate, pTempFileName);
|
|
if (rc == NO_ERROR)
|
|
{
|
|
DeleteFileW(pFileName);
|
|
MoveFileW(pTempFileName, pFileName);
|
|
}
|
|
else
|
|
{
|
|
LastError = rc;
|
|
DeleteFileW(pTempFileName);
|
|
}
|
|
}
|
|
GlobalUnlock(pUpdate->hFileName);
|
|
GlobalFree(pUpdate->hFileName);
|
|
}
|
|
|
|
FreeData(pUpdate);
|
|
GlobalUnlock(hUpdate);
|
|
GlobalFree(hUpdate);
|
|
SetLastError(LastError);
|
|
return(rc ? FALSE : TRUE);
|
|
}
|
|
|
|
|
|
/**********************************************************************
|
|
**
|
|
** End of API entry points.
|
|
**
|
|
** Beginning of private entry points for worker routines to do the
|
|
** real work.
|
|
**
|
|
***********************************************************************/
|
|
|
|
|
|
BOOL EnumTypesFunc(HANDLE hModule, LPWSTR lpType, LONG lParam)
|
|
{
|
|
EnumResourceNamesW(hModule, lpType, (ENUMRESNAMEPROCW)EnumNamesFunc, lParam);
|
|
return TRUE;
|
|
}
|
|
|
|
BOOL EnumNamesFunc(HANDLE hModule, LPWSTR lpType, LPWSTR lpName, LONG lParam)
|
|
{
|
|
EnumResourceLanguagesW(hModule, lpType, lpName, (ENUMRESLANGPROCW)EnumLangsFunc, lParam);
|
|
return TRUE;
|
|
}
|
|
|
|
BOOL EnumLangsFunc(HANDLE hModule, LPWSTR lpType, LPWSTR lpName, WORD language, LONG lParam)
|
|
{
|
|
HANDLE hResInfo;
|
|
LONG fError;
|
|
PSDATA Type;
|
|
PSDATA Name;
|
|
ULONG cb;
|
|
PVOID lpData;
|
|
HANDLE hResource;
|
|
PVOID lpResource;
|
|
|
|
hResInfo = FindResourceExW(hModule, lpType, lpName, language);
|
|
if (hResInfo == NULL)
|
|
{
|
|
return FALSE;
|
|
}
|
|
else
|
|
{
|
|
Type = AddStringOrID(lpType, (PUPDATEDATA)lParam);
|
|
if (Type == NULL)
|
|
{
|
|
((PUPDATEDATA)lParam)->Status = ERROR_NOT_ENOUGH_MEMORY;
|
|
return FALSE;
|
|
}
|
|
Name = AddStringOrID(lpName, (PUPDATEDATA)lParam);
|
|
if (Name == NULL)
|
|
{
|
|
((PUPDATEDATA)lParam)->Status = ERROR_NOT_ENOUGH_MEMORY;
|
|
return FALSE;
|
|
}
|
|
|
|
cb = SizeofResource(hModule, hResInfo);
|
|
if (cb == 0)
|
|
{
|
|
return FALSE;
|
|
}
|
|
lpData = GodotHeapAlloc(cb);
|
|
if (lpData == NULL)
|
|
{
|
|
return FALSE;
|
|
}
|
|
RtlZeroMemory(lpData, cb);
|
|
|
|
hResource = LoadResource(hModule, hResInfo);
|
|
if (hResource == NULL)
|
|
{
|
|
GodotHeapFree(lpData);
|
|
return FALSE;
|
|
}
|
|
|
|
lpResource = (PVOID)LockResource(hResource);
|
|
if (lpResource == NULL)
|
|
{
|
|
GodotHeapFree(lpData);
|
|
return FALSE;
|
|
}
|
|
|
|
RtlCopyMemory(lpData, lpResource, cb);
|
|
(VOID)UnlockResource(hResource);
|
|
(VOID)FreeResource(hResource);
|
|
|
|
fError = AddResource(Type, Name, language, (PUPDATEDATA)lParam, lpData, cb);
|
|
if (fError != NO_ERROR)
|
|
{
|
|
((PUPDATEDATA)lParam)->Status = ERROR_NOT_ENOUGH_MEMORY;
|
|
return FALSE;
|
|
}
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
VOID FreeOne(PRESNAME pRes)
|
|
{
|
|
GodotHeapFree((PVOID)pRes->OffsetToDataEntry);
|
|
if (IS_ID == pRes->Name->discriminant)
|
|
{
|
|
GodotHeapFree((PVOID)pRes->Name);
|
|
}
|
|
if (IS_ID == pRes->Type->discriminant)
|
|
{
|
|
GodotHeapFree((PVOID)pRes->Type);
|
|
}
|
|
GodotHeapFree((PVOID)pRes);
|
|
}
|
|
|
|
|
|
VOID FreeData(PUPDATEDATA pUpd)
|
|
{
|
|
PRESTYPE pType;
|
|
PRESNAME pRes;
|
|
PSDATA pstring, pStringTmp;
|
|
|
|
for (pType=pUpd->ResTypeHeadID ; pUpd->ResTypeHeadID ; pType=pUpd->ResTypeHeadID)
|
|
{
|
|
pUpd->ResTypeHeadID = pUpd->ResTypeHeadID->pnext;
|
|
|
|
for (pRes=pType->NameHeadID ; pType->NameHeadID ; pRes=pType->NameHeadID )
|
|
{
|
|
pType->NameHeadID = pType->NameHeadID->pnext;
|
|
FreeOne(pRes);
|
|
}
|
|
|
|
for (pRes=pType->NameHeadName ; pType->NameHeadName ; pRes=pType->NameHeadName )
|
|
{
|
|
pType->NameHeadName = pType->NameHeadName->pnext;
|
|
FreeOne(pRes);
|
|
}
|
|
|
|
GodotHeapFree((PVOID)pType);
|
|
}
|
|
|
|
for (pType=pUpd->ResTypeHeadName ; pUpd->ResTypeHeadName ; pType=pUpd->ResTypeHeadName)
|
|
{
|
|
pUpd->ResTypeHeadName = pUpd->ResTypeHeadName->pnext;
|
|
|
|
for (pRes=pType->NameHeadID ; pType->NameHeadID ; pRes=pType->NameHeadID )
|
|
{
|
|
pType->NameHeadID = pType->NameHeadID->pnext;
|
|
FreeOne(pRes);
|
|
}
|
|
|
|
for (pRes=pType->NameHeadName ; pType->NameHeadName ; pRes=pType->NameHeadName )
|
|
{
|
|
pType->NameHeadName = pType->NameHeadName->pnext;
|
|
FreeOne(pRes);
|
|
}
|
|
|
|
GodotHeapFree((PVOID)pType);
|
|
}
|
|
|
|
pstring = pUpd->StringHead;
|
|
while (pstring != NULL)
|
|
{
|
|
pStringTmp = pstring->uu.ss.pnext;
|
|
if (pstring->discriminant == IS_STRING)
|
|
GodotHeapFree((PVOID)pstring->szStr);
|
|
GodotHeapFree((PVOID)pstring);
|
|
pstring = pStringTmp;
|
|
}
|
|
|
|
return;
|
|
}
|
|
|
|
|
|
/*+++
|
|
|
|
Routines to register strings
|
|
|
|
---*/
|
|
|
|
//
|
|
// Resources are DWORD aligned and may be in any order.
|
|
//
|
|
|
|
#define TABLE_ALIGN 4
|
|
#define DATA_ALIGN 4L
|
|
|
|
|
|
|
|
PSDATA AddStringOrID(LPCWSTR lp, PUPDATEDATA pupd)
|
|
{
|
|
USHORT cb;
|
|
PSDATA pstring;
|
|
PPSDATA ppstring;
|
|
|
|
if (((ULONG)lp & 0xFFFF0000) == 0)
|
|
{
|
|
//
|
|
// an ID
|
|
//
|
|
pstring = (PSDATA)GodotHeapAlloc(sizeof(SDATA));
|
|
if (pstring == NULL)
|
|
return NULL;
|
|
RtlZeroMemory((PVOID)pstring, sizeof(SDATA));
|
|
pstring->discriminant = IS_ID;
|
|
|
|
pstring->uu.Ordinal = (WORD)((ULONG)lp & 0x0000ffff);
|
|
}
|
|
else
|
|
{
|
|
//
|
|
// a string
|
|
//
|
|
cb = gwcslen(lp) + 1;
|
|
ppstring = &pupd->StringHead;
|
|
|
|
while ((pstring = *ppstring) != NULL)
|
|
{
|
|
if (!gwcsncmp(pstring->szStr, lp, cb))
|
|
break;
|
|
ppstring = &(pstring->uu.ss.pnext);
|
|
}
|
|
|
|
if (!pstring)
|
|
{
|
|
|
|
//
|
|
// allocate a new one
|
|
//
|
|
pstring = (PSDATA)GodotHeapAlloc(sizeof(SDATA));
|
|
if (pstring == NULL)
|
|
return NULL;
|
|
RtlZeroMemory((PVOID)pstring, sizeof(SDATA));
|
|
|
|
pstring->szStr = (WCHAR*)GodotHeapAlloc(cb*sizeof(WCHAR));
|
|
if (pstring->szStr == NULL)
|
|
{
|
|
GodotHeapFree((PVOID)pstring);
|
|
return NULL;
|
|
}
|
|
pstring->discriminant = IS_STRING;
|
|
pstring->OffsetToString = pupd->cbStringTable;
|
|
|
|
pstring->cbData = sizeof(pstring->cbsz) + cb * sizeof(WCHAR);
|
|
pstring->cbsz = cb - 1; /* don't include zero terminator */
|
|
RtlCopyMemory(pstring->szStr, lp, cb*sizeof(WCHAR));
|
|
|
|
pupd->cbStringTable += pstring->cbData;
|
|
|
|
pstring->uu.ss.pnext=NULL;
|
|
*ppstring=pstring;
|
|
}
|
|
}
|
|
|
|
return(pstring);
|
|
}
|
|
|
|
LONG AddResource(IN PSDATA Type, IN PSDATA Name, IN WORD Language,
|
|
IN PUPDATEDATA pupd, IN PVOID lpData, IN ULONG cb)
|
|
//
|
|
// add a resource into the resource directory hiearchy
|
|
//
|
|
{
|
|
PRESTYPE pType;
|
|
PPRESTYPE ppType;
|
|
PRESNAME pName;
|
|
PRESNAME pNameM;
|
|
PPRESNAME ppName = NULL;
|
|
BOOL fTypeID=(Type->discriminant == IS_ID);
|
|
BOOL fNameID=(Name->discriminant == IS_ID);
|
|
BOOL fSame=FALSE;
|
|
|
|
//
|
|
// figure out which list to store it in
|
|
//
|
|
|
|
ppType = fTypeID ? &pupd->ResTypeHeadID : &pupd->ResTypeHeadName;
|
|
|
|
//
|
|
// Try to find the Type in the list
|
|
//
|
|
|
|
while ((pType=*ppType) != NULL)
|
|
{
|
|
if (pType->Type->uu.Ordinal == Type->uu.Ordinal)
|
|
{
|
|
ppName = fNameID ? &pType->NameHeadID : &pType->NameHeadName;
|
|
break;
|
|
}
|
|
if (fTypeID)
|
|
{
|
|
if (Type->uu.Ordinal < pType->Type->uu.Ordinal)
|
|
break;
|
|
}
|
|
else
|
|
{
|
|
if (gwcsncmp(Type->szStr, pType->Type->szStr, Type->cbsz) < 0)
|
|
break;
|
|
}
|
|
ppType = &(pType->pnext);
|
|
}
|
|
|
|
//
|
|
// Create a new type if needed
|
|
//
|
|
|
|
if (ppName == NULL)
|
|
{
|
|
pType = (PRESTYPE)GodotHeapAlloc(sizeof(RESTYPE));
|
|
if (pType == NULL)
|
|
return ERROR_NOT_ENOUGH_MEMORY;
|
|
RtlZeroMemory((PVOID)pType, sizeof(RESTYPE));
|
|
pType->pnext = *ppType;
|
|
*ppType = pType;
|
|
pType->Type = Type;
|
|
ppName = fNameID ? &pType->NameHeadID : &pType->NameHeadName;
|
|
}
|
|
|
|
//
|
|
// Find proper place for name
|
|
//
|
|
|
|
while ( (pName = *ppName) != NULL)
|
|
{
|
|
if (fNameID)
|
|
{
|
|
if (Name->uu.Ordinal == pName->Name->uu.Ordinal)
|
|
{
|
|
fSame = TRUE;
|
|
break;
|
|
}
|
|
if (Name->uu.Ordinal < pName->Name->uu.Ordinal)
|
|
break;
|
|
}
|
|
else
|
|
{
|
|
if (gwcsncmp(Name->szStr, pName->Name->szStr, Name->cbsz) == 0)
|
|
{
|
|
fSame = TRUE;
|
|
break;
|
|
}
|
|
if (gwcsncmp(Name->szStr, pName->Name->szStr, Name->cbsz) < 0)
|
|
break;
|
|
}
|
|
ppName = &(pName->pnext);
|
|
}
|
|
|
|
//
|
|
// check for delete/modify
|
|
//
|
|
|
|
if (fSame) /* same name, new language */
|
|
{
|
|
if (pName->NumberOfLanguages == 1) /* one language currently ? */
|
|
{
|
|
if (Language == pName->LanguageId) /* REPLACE || DELETE */
|
|
{
|
|
pName->DataSize = cb;
|
|
if (lpData == NULL) /* DELETE */
|
|
{
|
|
return DeleteResourceFromList(pupd, pType, pName, Language, fTypeID, fNameID);
|
|
}
|
|
GodotHeapFree((PVOID)pName->OffsetToDataEntry);
|
|
if (IS_ID == Type->discriminant)
|
|
{
|
|
GodotHeapFree((PVOID)Type);
|
|
}
|
|
if (IS_ID == Name->discriminant)
|
|
{
|
|
GodotHeapFree((PVOID)Name);
|
|
}
|
|
(PVOID)pName->OffsetToDataEntry = lpData;
|
|
return NO_ERROR;
|
|
}
|
|
else
|
|
{
|
|
if (lpData == NULL) /* no data but new? */
|
|
{
|
|
return ERROR_INVALID_PARAMETER; /* badness */
|
|
}
|
|
return InsertResourceIntoLangList(pupd, Type, Name, pType, pName,
|
|
Language, fNameID, cb, lpData);
|
|
}
|
|
}
|
|
else /* many languages currently */
|
|
{
|
|
pNameM = pName; /* save head of lang list */
|
|
while ( (pName = *ppName) != NULL) /* find insertion point */
|
|
{
|
|
if (!(fNameID ? pName->Name->uu.Ordinal == (*ppName)->Name->uu.Ordinal :
|
|
!gwcscmp(pName->Name->uu.ss.sz, (*ppName)->Name->uu.ss.sz)) ||
|
|
Language <= pName->LanguageId) /* here? */
|
|
break; /* yes */
|
|
ppName = &(pName->pnext); /* traverse language list */
|
|
}
|
|
|
|
if (pName && Language == pName->LanguageId) /* language found? */
|
|
{
|
|
if (lpData == NULL) /* DELETE */
|
|
{
|
|
return DeleteResourceFromList(pupd, pType, pName, Language, fTypeID, fNameID);
|
|
}
|
|
|
|
pName->DataSize = cb; /* REPLACE */
|
|
GodotHeapFree((PVOID)pName->OffsetToDataEntry);
|
|
if (IS_ID == Type->discriminant)
|
|
{
|
|
GodotHeapFree((PVOID)Type);
|
|
}
|
|
if (IS_ID == Name->discriminant)
|
|
{
|
|
GodotHeapFree((PVOID)Name);
|
|
}
|
|
(PVOID)pName->OffsetToDataEntry = lpData;
|
|
return NO_ERROR;
|
|
}
|
|
else /* add new language */
|
|
{
|
|
return InsertResourceIntoLangList(pupd, Type, Name, pType, pNameM, Language, fNameID, cb, lpData);
|
|
}
|
|
}
|
|
}
|
|
else /* unique name */
|
|
{
|
|
if (lpData == NULL) /* can't delete new name */
|
|
{
|
|
return ERROR_INVALID_PARAMETER;
|
|
}
|
|
}
|
|
|
|
//
|
|
// add new name/language
|
|
//
|
|
if (!fSame)
|
|
{
|
|
if (fNameID)
|
|
pType->NumberOfNamesID++;
|
|
else
|
|
pType->NumberOfNamesName++;
|
|
}
|
|
|
|
pName = (PRESNAME)GodotHeapAlloc(sizeof(RESNAME));
|
|
if (pName == NULL)
|
|
return ERROR_NOT_ENOUGH_MEMORY;
|
|
|
|
RtlZeroMemory((PVOID)pName, sizeof(RESNAME));
|
|
pName->pnext = *ppName;
|
|
*ppName = pName;
|
|
pName->Name = Name;
|
|
pName->Type = Type;
|
|
pName->NumberOfLanguages = 1;
|
|
pName->LanguageId = Language;
|
|
pName->DataSize = cb;
|
|
(PVOID)pName->OffsetToDataEntry = lpData;
|
|
|
|
return NO_ERROR;
|
|
}
|
|
|
|
|
|
BOOL DeleteResourceFromList(PUPDATEDATA pUpd, PRESTYPE pType, PRESNAME pName,
|
|
INT Language, INT fType, INT fName)
|
|
{
|
|
PPRESTYPE ppType;
|
|
PPRESNAME ppName;
|
|
PRESNAME pNameT;
|
|
|
|
/* find previous type node */
|
|
ppType = fType ? &pUpd->ResTypeHeadID : &pUpd->ResTypeHeadName;
|
|
while (*ppType != pType)
|
|
{
|
|
ppType = &((*ppType)->pnext);
|
|
}
|
|
|
|
/* find previous type node */
|
|
ppName = fName ? &pType->NameHeadID : &pType->NameHeadName;
|
|
pNameT = NULL;
|
|
while (*ppName != pName)
|
|
{
|
|
if (pNameT == NULL) /* find first Name in lang list */
|
|
{
|
|
if (fName)
|
|
{
|
|
if ((*ppName)->Name->uu.Ordinal == pName->Name->uu.Ordinal)
|
|
{
|
|
pNameT = *ppName;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (gwcsncmp((*ppName)->Name->szStr, pName->Name->szStr, (*ppName)->Name->cbsz) == 0)
|
|
{
|
|
pNameT = *ppName;
|
|
}
|
|
}
|
|
}
|
|
ppName = &((*ppName)->pnext);
|
|
}
|
|
|
|
if (pNameT == NULL) /* first of this name? */
|
|
{
|
|
pNameT = pName->pnext; /* then (possibly) make next head of lang */
|
|
if (pNameT != NULL)
|
|
{
|
|
if (fName)
|
|
{
|
|
if (pNameT->Name->uu.Ordinal == pName->Name->uu.Ordinal)
|
|
{
|
|
pNameT->NumberOfLanguages = pName->NumberOfLanguages - 1;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (gwcsncmp(pNameT->Name->szStr, pName->Name->szStr, pNameT->Name->cbsz) == 0)
|
|
{
|
|
pNameT->NumberOfLanguages = pName->NumberOfLanguages - 1;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
else
|
|
pNameT->NumberOfLanguages--;
|
|
|
|
if (pNameT)
|
|
{
|
|
if (pNameT->NumberOfLanguages == 0)
|
|
{
|
|
if (fName)
|
|
pType->NumberOfNamesID -= 1;
|
|
else
|
|
pType->NumberOfNamesName -= 1;
|
|
}
|
|
}
|
|
|
|
*ppName = pName->pnext; /* link to next */
|
|
GodotHeapFree( (PVOID)pName->OffsetToDataEntry);
|
|
GodotHeapFree( pName); /* and free */
|
|
|
|
if (*ppName == NULL) /* type list completely empty? */
|
|
{
|
|
*ppType = pType->pnext; /* link to next */
|
|
GodotHeapFree( pType); /* and free */
|
|
}
|
|
|
|
return NO_ERROR;
|
|
}
|
|
|
|
BOOL InsertResourceIntoLangList(PUPDATEDATA pUpd, PSDATA Type, PSDATA Name, PRESTYPE pType,
|
|
PRESNAME pName, INT Language, INT fName, INT cb, PVOID lpData)
|
|
{
|
|
PRESNAME pNameM;
|
|
PRESNAME pNameNew;
|
|
PPRESNAME ppName;
|
|
|
|
pNameNew = (PRESNAME)GodotHeapAlloc(sizeof(RESNAME));
|
|
if (pNameNew == NULL)
|
|
return ERROR_NOT_ENOUGH_MEMORY;
|
|
RtlZeroMemory((PVOID)pNameNew, sizeof(RESNAME));
|
|
pNameNew->Name = Name;
|
|
pNameNew->Type = Type;
|
|
pNameNew->LanguageId = (WORD)Language;
|
|
pNameNew->DataSize = cb;
|
|
(PVOID)pNameNew->OffsetToDataEntry = lpData;
|
|
|
|
if (Language < pName->LanguageId) /* have to add to the front */
|
|
{
|
|
pNameNew->NumberOfLanguages = pName->NumberOfLanguages + 1;
|
|
pName->NumberOfLanguages = 1;
|
|
|
|
ppName = fName ? &pType->NameHeadID : &pType->NameHeadName;
|
|
|
|
/* don't have to look for NULL at end of list !!! */
|
|
while (pName != *ppName) /* find insertion point */
|
|
{
|
|
ppName = &((*ppName)->pnext); /* traverse language list */
|
|
}
|
|
pNameNew->pnext = *ppName; /* insert */
|
|
*ppName = pNameNew;
|
|
}
|
|
else
|
|
{
|
|
pNameM = pName;
|
|
pName->NumberOfLanguages += 1;
|
|
while((pName != NULL) &&
|
|
(fName ? Name->uu.Ordinal == pName->Name->uu.Ordinal :
|
|
!gwcscmp(Name->uu.ss.sz, pName->Name->uu.ss.sz))) /* find insertion point */
|
|
{
|
|
if (Language <= pName->LanguageId) /* here? */
|
|
break; /* yes */
|
|
pNameM = pName;
|
|
pName = pName->pnext; /* traverse language list */
|
|
}
|
|
pName = pNameM->pnext;
|
|
pNameM->pnext = pNameNew;
|
|
pNameNew->pnext = pName;
|
|
}
|
|
return NO_ERROR;
|
|
}
|
|
|
|
|
|
/*
|
|
* Utility routines
|
|
*/
|
|
|
|
ULONG FilePos(int fh)
|
|
{
|
|
return _llseek(fh, 0L, SEEK_CUR);
|
|
}
|
|
|
|
|
|
ULONG MuMoveFilePos(INT fh, ULONG pos)
|
|
{
|
|
return _llseek( fh, pos, SEEK_SET );
|
|
}
|
|
|
|
ULONG MuWrite(INT fh, UCHAR*p, ULONG n)
|
|
{
|
|
ULONG n1;
|
|
|
|
if ((n1 = _lwrite(fh, p, n)) != n)
|
|
|
|
{
|
|
return n1;
|
|
}
|
|
else
|
|
return 0;
|
|
}
|
|
|
|
ULONG MuRead(INT fh, UCHAR*p, ULONG n)
|
|
{
|
|
ULONG n1;
|
|
|
|
if ((n1 = _lread( fh, p, n )) != n)
|
|
{
|
|
return n1;
|
|
}
|
|
else
|
|
return 0;
|
|
}
|
|
|
|
BOOL MuCopy(INT srcfh, INT dstfh, ULONG nbytes)
|
|
{
|
|
ULONG n;
|
|
ULONG cb=0L;
|
|
PUCHAR pb;
|
|
|
|
pb = (PUCHAR)GodotHeapAlloc(BUFSIZE);
|
|
if (pb == NULL)
|
|
return 0;
|
|
RtlZeroMemory((PVOID)pb, BUFSIZE);
|
|
|
|
while (nbytes)
|
|
{
|
|
if (nbytes <= BUFSIZE)
|
|
n = nbytes;
|
|
else
|
|
n = BUFSIZE;
|
|
nbytes -= n;
|
|
|
|
if (!MuRead( srcfh, pb, n ))
|
|
{
|
|
cb += n;
|
|
MuWrite( dstfh, pb, n );
|
|
}
|
|
else
|
|
{
|
|
GodotHeapFree( pb);
|
|
return cb;
|
|
}
|
|
}
|
|
GodotHeapFree( pb);
|
|
return cb;
|
|
}
|
|
|
|
VOID SetResdata(PIMAGE_RESOURCE_DATA_ENTRY pResData, ULONG offset, ULONG size)
|
|
{
|
|
pResData->OffsetToData = offset;
|
|
pResData->Size = size;
|
|
pResData->CodePage = DEFAULT_CODEPAGE;
|
|
pResData->Reserved = 0L;
|
|
}
|
|
|
|
|
|
VOID SetRestab(PIMAGE_RESOURCE_DIRECTORY pRestab, LONG time, WORD cNamed, WORD cId)
|
|
{
|
|
pRestab->Characteristics = 0L;
|
|
pRestab->TimeDateStamp = time;
|
|
pRestab->MajorVersion = MAJOR_RESOURCE_VERSION;
|
|
pRestab->MinorVersion = MINOR_RESOURCE_VERSION;
|
|
pRestab->NumberOfNamedEntries = cNamed;
|
|
pRestab->NumberOfIdEntries = cId;
|
|
}
|
|
|
|
|
|
PIMAGE_SECTION_HEADER FindSection(PIMAGE_SECTION_HEADER pObjBottom,
|
|
PIMAGE_SECTION_HEADER pObjTop, LPSTR pName)
|
|
{
|
|
while (pObjBottom < pObjTop)
|
|
{
|
|
if (strcmp(pObjBottom->Name, pName) == 0)
|
|
return pObjBottom;
|
|
pObjBottom++;
|
|
}
|
|
|
|
return NULL;
|
|
}
|
|
|
|
ULONG AssignResourceToSection(PRESNAME *ppRes, ULONG ExtraSectionOffset,
|
|
ULONG Offset, LONG Size, PLONG pSizeRsrc1)
|
|
{
|
|
ULONG cb;
|
|
|
|
/* Assign this res to this section */
|
|
cb = ROUNDUP((*ppRes)->DataSize, CBLONG);
|
|
if (Offset < ExtraSectionOffset && Offset + cb > (ULONG)Size)
|
|
{
|
|
*pSizeRsrc1 = Offset;
|
|
Offset = ExtraSectionOffset;
|
|
}
|
|
(*ppRes)->OffsetToData = Offset;
|
|
*ppRes = (*ppRes)->pnext;
|
|
return Offset + cb;
|
|
}
|
|
|
|
|
|
/***************************** Main Worker Function ***************************
|
|
* LONG PEWriteResFile
|
|
*
|
|
* This function writes the resources to the named executable file.
|
|
* It assumes that resources have no fixups (even any existing resources
|
|
* that it removes from the executable.) It places all the resources into
|
|
* one or two sections. The resources are packed tightly into the section,
|
|
* being aligned on dword boundaries. Each section is padded to a file
|
|
* sector size (no invalid or zero-filled pages), and each
|
|
* resource is padded to the afore-mentioned dword boundary. This
|
|
* function uses the capabilities of the NT system to enable it to easily
|
|
* manipulate the data: to wit, it assumes that the system can allocate
|
|
* any sized piece of data, in particular the section and resource tables.
|
|
* If it did not, it might have to deal with temporary files (the system
|
|
* may have to grow the swap file, but that's what the system is for.)
|
|
*
|
|
* Return values are:
|
|
* TRUE - file was written succesfully.
|
|
* FALSE - file was not written succesfully.
|
|
*
|
|
* Effects:
|
|
*
|
|
* History:
|
|
* Thur Apr 27, 1989 by Floyd Rogers [floydr]
|
|
* Created.
|
|
* 12/8/89 sanfords Added multiple section support.
|
|
* 12/11/90 floydr Modified for new (NT) Linear Exe format
|
|
* 1/18/92 vich Modified for new (NT) Portable Exe format
|
|
* 5/8/92 bryant General cleanup so resonexe can work with unicode
|
|
* 6/9/92 floydr incorporate bryan's changes
|
|
* 6/15/92 floydr debug section separate from debug table
|
|
* 9/25/92 floydr account for .rsrc not being last-1
|
|
* 9/28/92 floydr account for adding lots of resources by adding
|
|
* a second .rsrc section.
|
|
\****************************************************************************/
|
|
|
|
/* */
|
|
LONG PEWriteResFile(INT inpfh, INT outfh, ULONG cbOldexe, PUPDATEDATA pUpdate)
|
|
{
|
|
IMAGE_NT_HEADERS Old; /* original header */
|
|
IMAGE_NT_HEADERS New; /* working header */
|
|
PRESNAME pRes;
|
|
PRESNAME pResSave;
|
|
PRESTYPE pType;
|
|
ULONG clock = GetTickCount(); /* current time */
|
|
ULONG cbName=0; /* count of bytes in name strings */
|
|
ULONG cbType=0; /* count of bytes in type strings */
|
|
ULONG cTypeStr=0; /* count of strings */
|
|
ULONG cNameStr=0; /* count of strings */
|
|
LONG cb; /* temp byte count and file index */
|
|
ULONG cTypes = 0L; /* count of resource types */
|
|
ULONG cNames = 0L; /* Count of names for multiple languages/name */
|
|
ULONG cRes = 0L; /* count of resources */
|
|
ULONG cbRestab; /* count of resources */
|
|
LONG cbNew = 0L; /* general count */
|
|
ULONG ibObjTab;
|
|
ULONG ibObjTabEnd;
|
|
ULONG ibNewObjTabEnd;
|
|
ULONG ibSave;
|
|
ULONG adjust=0;
|
|
PIMAGE_SECTION_HEADER pObjtblOld,
|
|
pObjtblNew,
|
|
pObjDebug,
|
|
pObjResourceOld,
|
|
pObjResourceNew,
|
|
pObjResourceOldX,
|
|
pObjDebugDirOld,
|
|
pObjDebugDirNew,
|
|
pObjNew,
|
|
pObjOld,
|
|
pObjLast;
|
|
PUCHAR p;
|
|
PIMAGE_RESOURCE_DIRECTORY pResTab;
|
|
PIMAGE_RESOURCE_DIRECTORY pResTabN;
|
|
PIMAGE_RESOURCE_DIRECTORY pResTabL;
|
|
PIMAGE_RESOURCE_DIRECTORY_ENTRY pResDirL;
|
|
PIMAGE_RESOURCE_DIRECTORY_ENTRY pResDirN;
|
|
PIMAGE_RESOURCE_DIRECTORY_ENTRY pResDirT;
|
|
PIMAGE_RESOURCE_DATA_ENTRY pResData;
|
|
PUSHORT pResStr;
|
|
PUSHORT pResStrEnd;
|
|
PSDATA pPreviousName;
|
|
LONG nObjResource=-1;
|
|
LONG nObjResourceX=-1;
|
|
ULONG cbResource;
|
|
ULONG cbMustPad = 0;
|
|
ULONG ibMaxDbgOffsetOld;
|
|
|
|
MuMoveFilePos(inpfh, cbOldexe);
|
|
MuRead(inpfh, (PUCHAR)&Old, sizeof(IMAGE_NT_HEADERS));
|
|
ibObjTab = cbOldexe + sizeof(ULONG) + sizeof(IMAGE_FILE_HEADER) +
|
|
Old.FileHeader.SizeOfOptionalHeader;
|
|
ibObjTabEnd = ibObjTab + Old.FileHeader.NumberOfSections * sizeof(IMAGE_SECTION_HEADER);
|
|
ibNewObjTabEnd = ibObjTabEnd;
|
|
|
|
if (*(PUSHORT)&Old.Signature != IMAGE_NT_SIGNATURE)
|
|
return ERROR_INVALID_EXE_SIGNATURE;
|
|
|
|
if((Old.FileHeader.Characteristics & IMAGE_FILE_EXECUTABLE_IMAGE) == 0 &&
|
|
(Old.FileHeader.Characteristics & IMAGE_FILE_DLL) == 0)
|
|
{
|
|
return ERROR_EXE_MARKED_INVALID;
|
|
}
|
|
|
|
/* New header is like old one. */
|
|
RtlCopyMemory(&New, &Old, sizeof(IMAGE_NT_HEADERS));
|
|
|
|
/* Read section table */
|
|
pObjtblOld = (PIMAGE_SECTION_HEADER)GodotHeapAlloc(
|
|
Old.FileHeader.NumberOfSections * sizeof(IMAGE_SECTION_HEADER));
|
|
if (pObjtblOld == NULL)
|
|
{
|
|
cb = ERROR_NOT_ENOUGH_MEMORY;
|
|
goto AbortExit;
|
|
}
|
|
RtlZeroMemory((PVOID)pObjtblOld, Old.FileHeader.NumberOfSections*sizeof(IMAGE_SECTION_HEADER));
|
|
MuMoveFilePos(inpfh, ibObjTab);
|
|
MuRead(inpfh, (PUCHAR)pObjtblOld,
|
|
Old.FileHeader.NumberOfSections * sizeof(IMAGE_SECTION_HEADER));
|
|
pObjLast = pObjtblOld + Old.FileHeader.NumberOfSections;
|
|
ibMaxDbgOffsetOld = 0;
|
|
for (pObjOld=pObjtblOld ; pObjOld<pObjLast ; pObjOld++)
|
|
{
|
|
if (pObjOld->PointerToRawData > ibMaxDbgOffsetOld)
|
|
{
|
|
ibMaxDbgOffsetOld = pObjOld->PointerToRawData + pObjOld->SizeOfRawData;
|
|
}
|
|
}
|
|
|
|
/*
|
|
* First, count up the resources. We need this information
|
|
* to discover how much room for header information to allocate
|
|
* in the resource section. cRes tells us how
|
|
* many language directory entries/tables. cNames and cTypes
|
|
* is used for the respective tables and/or entries. cbName totals
|
|
* the bytes required to store the alpha names (including the leading
|
|
* length word). cNameStr counts these strings.
|
|
*/
|
|
|
|
/* first, count those in the named type list */
|
|
cbResource = 0;
|
|
pType = pUpdate->ResTypeHeadName;
|
|
while (pType != NULL)
|
|
{
|
|
if (pType->NameHeadName != NULL || pType->NameHeadID != NULL)
|
|
{
|
|
cTypes++;
|
|
cTypeStr++;
|
|
cbType += (pType->Type->cbsz + 1) * sizeof(WORD);
|
|
|
|
pPreviousName = NULL;
|
|
pRes = pType->NameHeadName;
|
|
while (pRes)
|
|
{
|
|
cRes++;
|
|
if (pPreviousName == NULL ||
|
|
gwcsncmp(pPreviousName->szStr, pRes->Name->szStr, pRes->Name->cbsz) != 0)
|
|
{
|
|
cbName += (pRes->Name->cbsz + 1) * sizeof(WORD);
|
|
cNameStr++;
|
|
cNames++;
|
|
}
|
|
cbResource += ROUNDUP(pRes->DataSize, CBLONG);
|
|
pPreviousName = pRes->Name;
|
|
pRes = pRes->pnext;
|
|
}
|
|
|
|
pPreviousName = NULL;
|
|
pRes = pType->NameHeadID;
|
|
while (pRes)
|
|
{
|
|
cRes++;
|
|
if (pPreviousName == NULL || pPreviousName->uu.Ordinal != pRes->Name->uu.Ordinal)
|
|
{
|
|
cNames++;
|
|
}
|
|
cbResource += ROUNDUP(pRes->DataSize, CBLONG);
|
|
pPreviousName = pRes->Name;
|
|
pRes = pRes->pnext;
|
|
}
|
|
}
|
|
pType = pType->pnext;
|
|
}
|
|
|
|
/* second, count those in the ID type list */
|
|
pType = pUpdate->ResTypeHeadID;
|
|
while (pType != NULL)
|
|
{
|
|
if (pType->NameHeadName != NULL || pType->NameHeadID != NULL)
|
|
{
|
|
cTypes++;
|
|
pPreviousName = NULL;
|
|
pRes = pType->NameHeadName;
|
|
while (pRes)
|
|
|
|
{
|
|
cRes++;
|
|
if (pPreviousName == NULL ||
|
|
gwcsncmp(pPreviousName->szStr, pRes->Name->szStr, pRes->Name->cbsz) != 0)
|
|
{
|
|
cNames++;
|
|
cbName += (pRes->Name->cbsz + 1) * sizeof(WORD);
|
|
cNameStr++;
|
|
}
|
|
cbResource += ROUNDUP(pRes->DataSize, CBLONG);
|
|
pPreviousName = pRes->Name;
|
|
pRes = pRes->pnext;
|
|
}
|
|
|
|
pPreviousName = NULL;
|
|
pRes = pType->NameHeadID;
|
|
while (pRes)
|
|
{
|
|
cRes++;
|
|
if (pPreviousName == NULL ||
|
|
pPreviousName->uu.Ordinal != pRes->Name->uu.Ordinal)
|
|
{
|
|
cNames++;
|
|
}
|
|
cbResource += ROUNDUP(pRes->DataSize, CBLONG);
|
|
pPreviousName = pRes->Name;
|
|
pRes = pRes->pnext;
|
|
}
|
|
}
|
|
pType = pType->pnext;
|
|
}
|
|
cb = REMAINDER(cbName + cbType, CBLONG);
|
|
|
|
/* Add up the number of bytes needed to store the directory. There is
|
|
* one type table with cTypes entries. They point to cTypes name tables
|
|
* that have a total of cNames entries. Each of them points to a language
|
|
* table and there are a total of cRes entries in all the language tables.
|
|
* Finally, we have the space needed for the Directory string entries,
|
|
* some extra padding to attain the desired alignment, and the space for
|
|
* cRes data entry headers.
|
|
*/
|
|
cbRestab = sizeof(IMAGE_RESOURCE_DIRECTORY) + /* root dir (types) */
|
|
cTypes * sizeof(IMAGE_RESOURCE_DIRECTORY_ENTRY) +
|
|
cTypes * sizeof(IMAGE_RESOURCE_DIRECTORY) + /* subdir2 (names) */
|
|
cNames * sizeof(IMAGE_RESOURCE_DIRECTORY_ENTRY) +
|
|
cNames * sizeof(IMAGE_RESOURCE_DIRECTORY) + /* subdir3 (langs) */
|
|
cRes * sizeof(IMAGE_RESOURCE_DIRECTORY_ENTRY) +
|
|
(cbName + cbType) + /* name/type strings */
|
|
cb + /* padding */
|
|
cRes * sizeof(IMAGE_RESOURCE_DATA_ENTRY); /* data entries */
|
|
|
|
cbResource += cbRestab; /* add in the resource table */
|
|
|
|
// Find any current resource sections
|
|
|
|
pObjResourceOld = FindSection(pObjtblOld, pObjLast, ".rsrc");
|
|
pObjResourceOldX = FindSection(pObjtblOld, pObjLast, ".rsrc1");
|
|
if (pObjResourceOld == NULL)
|
|
{
|
|
cb = 0x7fffffff; /* can fill forever */
|
|
}
|
|
else if (pObjResourceOld + 1 == pObjResourceOldX)
|
|
{
|
|
nObjResource = pObjResourceOld - pObjtblOld;
|
|
cb = 0x7fffffff; /* merge resource sections */
|
|
}
|
|
else
|
|
{
|
|
nObjResource = pObjResourceOld - pObjtblOld;
|
|
cb = (pObjResourceOld+1)->VirtualAddress - pObjResourceOld->VirtualAddress;
|
|
if (cbRestab > (ULONG)cb)
|
|
{
|
|
return ERROR_INVALID_DATA;
|
|
}
|
|
}
|
|
|
|
/*
|
|
* Discover where the first discardable section is. This is where
|
|
* we will stick any new resource section.
|
|
*
|
|
* Note that we are ignoring discardable sections such as .CRT -
|
|
* this is so that we don't cause any relocation problems.
|
|
* Let's hope that .reloc is the one we want!!!
|
|
*/
|
|
pObjOld = FindSection(pObjtblOld, pObjLast, ".reloc");
|
|
|
|
if (pObjResourceOld != NULL && cbResource > (ULONG)cb)
|
|
{
|
|
if (pObjResourceOld != NULL && pObjOld == pObjResourceOld + 1)
|
|
{
|
|
cb = 0x7fffffff; /* can fill forever */
|
|
}
|
|
else if (pObjResourceOldX == NULL)
|
|
{
|
|
nObjResourceX = pObjOld - pObjtblOld;
|
|
adjust = pObjOld->VirtualAddress - pObjResourceOld->VirtualAddress;
|
|
}
|
|
else /* have already merged .rsrc & .rsrc1, if possible */
|
|
{
|
|
nObjResourceX = pObjResourceOldX - pObjtblOld;
|
|
adjust = pObjResourceOldX->VirtualAddress - pObjResourceOld->VirtualAddress;
|
|
}
|
|
}
|
|
|
|
/*
|
|
* Walk the type lists and figure out where the Data entry header will
|
|
* go. Keep a running total of the size for each data element so we
|
|
* can store this in the section header.
|
|
*/
|
|
|
|
/* first, those in the named type list */
|
|
|
|
cbResource = cbRestab; /* assign resource table to 1st rsrc section */
|
|
/* adjust == offset to .rsrc1 */
|
|
/* cb == size availble in .rsrc */
|
|
cbNew = 0; /* count of bytes in second .rsrc */
|
|
pType = pUpdate->ResTypeHeadName;
|
|
while (pType != NULL)
|
|
{
|
|
if (pType->NameHeadName != NULL || pType->NameHeadID != NULL)
|
|
{
|
|
pRes = pType->NameHeadName;
|
|
while (pRes)
|
|
{
|
|
cbResource = AssignResourceToSection(&pRes, adjust, cbResource, cb, &cbNew);
|
|
}
|
|
pRes = pType->NameHeadID;
|
|
while (pRes)
|
|
{
|
|
cbResource = AssignResourceToSection(&pRes, adjust, cbResource, cb, &cbNew);
|
|
}
|
|
}
|
|
pType = pType->pnext;
|
|
}
|
|
|
|
/* then, count those in the ID type list */
|
|
|
|
pType = pUpdate->ResTypeHeadID;
|
|
while (pType != NULL)
|
|
{
|
|
if (pType->NameHeadName != NULL || pType->NameHeadID != NULL)
|
|
{
|
|
pRes = pType->NameHeadName;
|
|
while (pRes)
|
|
{
|
|
cbResource = AssignResourceToSection(&pRes, adjust, cbResource, cb, &cbNew);
|
|
}
|
|
pRes = pType->NameHeadID;
|
|
while (pRes)
|
|
{
|
|
cbResource = AssignResourceToSection(&pRes, adjust, cbResource, cb, &cbNew);
|
|
}
|
|
}
|
|
pType = pType->pnext;
|
|
}
|
|
/*
|
|
* At this point:
|
|
* cbResource has offset of first byte past the last resource.
|
|
* cbNew has the count of bytes in the first resource section,
|
|
* if there are two sections.
|
|
*/
|
|
if (cbNew == 0)
|
|
cbNew = cbResource;
|
|
|
|
/*
|
|
* Discover where the Debug info is (if any)?
|
|
*/
|
|
pObjDebug = FindSection(pObjtblOld, pObjLast, ".debug");
|
|
if (pObjDebug != NULL)
|
|
{
|
|
if (Old.OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_DEBUG].VirtualAddress == 0)
|
|
{
|
|
return ERROR_INVALID_DATA;
|
|
}
|
|
if (pObjDebug != pObjLast-1)
|
|
{
|
|
return ERROR_INVALID_DATA;
|
|
}
|
|
}
|
|
pObjDebugDirOld = NULL;
|
|
for (pObjOld=pObjtblOld ; pObjOld<pObjLast ; pObjOld++)
|
|
{
|
|
if(Old.OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_DEBUG].VirtualAddress >=
|
|
pObjOld->VirtualAddress &&
|
|
Old.OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_DEBUG].VirtualAddress <
|
|
pObjOld->VirtualAddress+pObjOld->SizeOfRawData)
|
|
{
|
|
pObjDebugDirOld = pObjOld;
|
|
break;
|
|
}
|
|
}
|
|
|
|
/*
|
|
* Discover where the first discardable section is. This is where
|
|
* we will stick any new resource section.
|
|
*
|
|
* Note that we are ignoring discardable sections such as .CRT -
|
|
* this is so that we don't cause any relocation problems.
|
|
* Let's hope that .reloc is the one we want!!!
|
|
*/
|
|
pObjOld = FindSection(pObjtblOld, pObjLast, ".reloc");
|
|
|
|
if (nObjResource == -1) /* no old resource section */
|
|
{
|
|
if (pObjOld != NULL)
|
|
nObjResource = pObjOld - pObjtblOld;
|
|
else if (pObjDebug != NULL)
|
|
nObjResource = pObjDebug - pObjtblOld;
|
|
else
|
|
nObjResource = New.FileHeader.NumberOfSections;
|
|
New.FileHeader.NumberOfSections++;
|
|
}
|
|
|
|
if (nObjResourceX != -1)
|
|
{
|
|
if (pObjResourceOldX != NULL)
|
|
{
|
|
nObjResourceX = pObjResourceOldX - pObjtblOld;
|
|
New.FileHeader.NumberOfSections--;
|
|
}
|
|
else if (pObjOld != NULL)
|
|
nObjResourceX = pObjOld - pObjtblOld;
|
|
else if (pObjDebug != NULL)
|
|
nObjResourceX = pObjDebug - pObjtblOld;
|
|
else
|
|
nObjResourceX = New.FileHeader.NumberOfSections;
|
|
New.FileHeader.NumberOfSections++;
|
|
}
|
|
else if (pObjResourceOldX != NULL) /* Was old .rsrc1 section? */
|
|
{
|
|
New.FileHeader.NumberOfSections--; /* yes, delete it */
|
|
}
|
|
|
|
/*
|
|
* If we had to add anything to the header (section table),
|
|
* then we have to update the header size and rva's in the header.
|
|
*/
|
|
adjust = (New.FileHeader.NumberOfSections -
|
|
Old.FileHeader.NumberOfSections) * sizeof(IMAGE_SECTION_HEADER);
|
|
cb = Old.OptionalHeader.SizeOfHeaders -
|
|
(Old.FileHeader.NumberOfSections*sizeof(IMAGE_SECTION_HEADER) +
|
|
sizeof(IMAGE_NT_HEADERS) + cbOldexe );
|
|
if (adjust > (ULONG)cb)
|
|
{
|
|
int i;
|
|
|
|
adjust -= cb;
|
|
for (i = 0; i < IMAGE_NUMBEROF_DIRECTORY_ENTRIES ; i++)
|
|
{
|
|
if (New.OptionalHeader.DataDirectory[i].VirtualAddress &&
|
|
New.OptionalHeader.DataDirectory[i].VirtualAddress < New.OptionalHeader.SizeOfHeaders)
|
|
{
|
|
New.OptionalHeader.DataDirectory[i].VirtualAddress += adjust;
|
|
}
|
|
}
|
|
New.OptionalHeader.SizeOfHeaders += adjust;
|
|
}
|
|
else if (adjust > 0)
|
|
{
|
|
int i;
|
|
|
|
//
|
|
// Loop over DataDirectory entries and look for any entries that point to
|
|
// information stored in the 'dead' space after the section table but before
|
|
// the SizeOfHeaders length.
|
|
//
|
|
for (i = 0; i < IMAGE_NUMBEROF_DIRECTORY_ENTRIES ; i++)
|
|
{
|
|
if (New.OptionalHeader.DataDirectory[i].VirtualAddress &&
|
|
New.OptionalHeader.DataDirectory[i].VirtualAddress < Old.OptionalHeader.SizeOfHeaders)
|
|
{
|
|
New.OptionalHeader.DataDirectory[i].VirtualAddress += adjust;
|
|
}
|
|
}
|
|
}
|
|
ibNewObjTabEnd += adjust;
|
|
|
|
/* Allocate storage for new section table */
|
|
cb = New.FileHeader.NumberOfSections * sizeof(IMAGE_SECTION_HEADER);
|
|
pObjtblNew = (PIMAGE_SECTION_HEADER)GodotHeapAlloc((short)cb);
|
|
if (pObjtblNew == NULL)
|
|
{
|
|
cb = ERROR_NOT_ENOUGH_MEMORY;
|
|
goto AbortExit;
|
|
}
|
|
RtlZeroMemory((PVOID)pObjtblNew, cb);
|
|
pObjResourceNew = pObjtblNew + nObjResource;
|
|
|
|
/*
|
|
* copy old section table to new
|
|
*/
|
|
adjust = 0; /* adjustment to virtual address */
|
|
for (pObjOld=pObjtblOld,pObjNew=pObjtblNew ; pObjOld<pObjLast ; pObjOld++)
|
|
{
|
|
if (pObjOld == pObjResourceOldX)
|
|
{
|
|
if (nObjResourceX == -1)
|
|
{
|
|
// we have to move back all the other section.
|
|
// the .rsrc1 is bigger than what we need
|
|
// adjust must be a negative number
|
|
adjust -= (pObjOld+1)->VirtualAddress - pObjOld->VirtualAddress;
|
|
}
|
|
continue;
|
|
}
|
|
else if (pObjNew == pObjResourceNew)
|
|
{
|
|
cb = ROUNDUP(cbNew, New.OptionalHeader.FileAlignment);
|
|
if (pObjResourceOld == NULL)
|
|
{
|
|
adjust = ROUNDUP(cbNew, New.OptionalHeader.SectionAlignment);
|
|
RtlZeroMemory(pObjNew, sizeof(IMAGE_SECTION_HEADER));
|
|
strcpy(pObjNew->Name, ".rsrc");
|
|
pObjNew->VirtualAddress = pObjOld->VirtualAddress;
|
|
pObjNew->PointerToRawData = pObjOld->PointerToRawData;
|
|
pObjNew->Characteristics = IMAGE_SCN_MEM_READ |
|
|
/*IMAGE_SCN_MEM_WRITE | */
|
|
IMAGE_SCN_CNT_INITIALIZED_DATA;
|
|
pObjNew->SizeOfRawData = cb;
|
|
pObjNew->Misc.VirtualSize = cbNew; //ROUNDUP(cb, New.OptionalHeader.SectionAlignment);
|
|
}
|
|
else
|
|
{
|
|
*pObjNew = *pObjOld; /* copy obj table entry */
|
|
pObjNew->SizeOfRawData = cb;
|
|
pObjNew->Misc.VirtualSize = cbNew; // ROUNDUP(cb, New.OptionalHeader.SectionAlignment);
|
|
if (pObjNew->SizeOfRawData == pObjOld->SizeOfRawData)
|
|
{
|
|
adjust = 0;
|
|
}
|
|
else if (pObjNew->SizeOfRawData > pObjOld->SizeOfRawData)
|
|
{
|
|
adjust += ROUNDUP(cbNew, New.OptionalHeader.SectionAlignment) -
|
|
((pObjOld+1)->VirtualAddress-pObjOld->VirtualAddress);
|
|
}
|
|
else /* is smaller, but pad so will be valid */
|
|
{
|
|
adjust = 0;
|
|
pObjNew->SizeOfRawData = pObjResourceOld->SizeOfRawData;
|
|
/* if legoized, the VS could be > RawSize !!! */
|
|
pObjNew->Misc.VirtualSize = pObjResourceOld->Misc.VirtualSize;
|
|
cbMustPad = pObjResourceOld->SizeOfRawData;
|
|
}
|
|
}
|
|
pObjNew++;
|
|
if (pObjResourceOld == NULL)
|
|
goto rest_of_table;
|
|
}
|
|
else if (nObjResourceX != -1 && pObjNew == pObjtblNew + nObjResourceX)
|
|
{
|
|
RtlZeroMemory(pObjNew, sizeof(IMAGE_SECTION_HEADER));
|
|
strcpy(pObjNew->Name, ".rsrc1");
|
|
/*
|
|
* Before we copy the virtual address we have to move back the
|
|
* .reloc * virtual address. Otherwise we will keep moving the
|
|
* reloc VirtualAddress forward.
|
|
* We will have to move back the address of .rsrc1
|
|
*/
|
|
if (pObjResourceOldX == NULL)
|
|
{
|
|
// This is the first time we have a .rsrc1
|
|
pObjNew->VirtualAddress = pObjOld->VirtualAddress;
|
|
pObjNew->Characteristics = IMAGE_SCN_MEM_READ |
|
|
IMAGE_SCN_MEM_WRITE |
|
|
IMAGE_SCN_CNT_INITIALIZED_DATA;
|
|
adjust = ROUNDUP(cbResource, New.OptionalHeader.SectionAlignment) +
|
|
pObjResourceNew->VirtualAddress - pObjNew->VirtualAddress;
|
|
}
|
|
else
|
|
{
|
|
// we already have an .rsrc1 use the position of that and
|
|
// calculate the new adjust
|
|
pObjNew->VirtualAddress = pObjResourceOldX->VirtualAddress;
|
|
pObjNew->Characteristics = IMAGE_SCN_MEM_READ |
|
|
IMAGE_SCN_MEM_WRITE |
|
|
IMAGE_SCN_CNT_INITIALIZED_DATA;
|
|
// Check if we have enough room in the old .rsrc1
|
|
// Include the full size of the section, data + roundup
|
|
if (cbResource - (pObjResourceOldX->VirtualAddress - pObjResourceOld->VirtualAddress) <=
|
|
pObjOld->VirtualAddress - pObjNew->VirtualAddress )
|
|
{
|
|
// we have to move back all the other section.
|
|
// the .rsrc1 is bigger than what we need
|
|
// adjust must be a negative number
|
|
// calc new adjust size
|
|
adjust = ROUNDUP(cbResource, New.OptionalHeader.SectionAlignment) +
|
|
pObjResourceNew->VirtualAddress - pObjOld->VirtualAddress;
|
|
}
|
|
else
|
|
{
|
|
// we have to move the section again.
|
|
// The .rsrc1 is too small
|
|
|
|
adjust = ROUNDUP(cbResource, New.OptionalHeader.SectionAlignment) +
|
|
pObjResourceNew->VirtualAddress -
|
|
pObjOld->VirtualAddress;
|
|
}
|
|
}
|
|
pObjNew++;
|
|
goto rest_of_table;
|
|
}
|
|
else if (pObjNew < pObjResourceNew)
|
|
{
|
|
*pObjNew++ = *pObjOld; /* copy obj table entry */
|
|
}
|
|
else
|
|
{
|
|
rest_of_table:
|
|
*pObjNew++ = *pObjOld;
|
|
(pObjNew-1)->VirtualAddress += adjust;
|
|
}
|
|
}
|
|
|
|
|
|
pObjNew = pObjtblNew + New.FileHeader.NumberOfSections - 1;
|
|
New.OptionalHeader.SizeOfImage = ROUNDUP(pObjNew->VirtualAddress +
|
|
pObjNew->SizeOfRawData, New.OptionalHeader.SectionAlignment);
|
|
|
|
/* allocate room to build the resource directory/tables in */
|
|
pResTab = (PIMAGE_RESOURCE_DIRECTORY)GodotHeapAlloc(cbRestab);
|
|
if (pResTab == NULL)
|
|
{
|
|
cb = ERROR_NOT_ENOUGH_MEMORY;
|
|
goto AbortExit;
|
|
}
|
|
|
|
/* First, setup the "root" type directory table. It will be followed by */
|
|
/* Types directory entries. */
|
|
|
|
RtlZeroMemory((PVOID)pResTab, cbRestab);
|
|
p = (PUCHAR)pResTab;
|
|
pResTab->Characteristics = 0L;
|
|
pResTab->TimeDateStamp = clock;
|
|
pResTab->MajorVersion = MAJOR_RESOURCE_VERSION;
|
|
pResTab->MinorVersion = MINOR_RESOURCE_VERSION;
|
|
pResTab->NumberOfNamedEntries = (USHORT)cTypeStr;
|
|
pResTab->NumberOfIdEntries = (USHORT)(cTypes - cTypeStr);
|
|
|
|
/* Calculate the start of the various parts of the resource table. */
|
|
/* We need the start of the Type/Name/Language directories as well */
|
|
/* as the start of the UNICODE strings and the actual data nodes. */
|
|
|
|
pResDirT = (PIMAGE_RESOURCE_DIRECTORY_ENTRY)(pResTab + 1);
|
|
|
|
pResDirN = (PIMAGE_RESOURCE_DIRECTORY_ENTRY)(((PUCHAR)pResDirT) +
|
|
cTypes * sizeof(IMAGE_RESOURCE_DIRECTORY_ENTRY));
|
|
|
|
pResDirL = (PIMAGE_RESOURCE_DIRECTORY_ENTRY)(((PUCHAR)pResDirN) +
|
|
cTypes * sizeof(IMAGE_RESOURCE_DIRECTORY) +
|
|
cNames * sizeof(IMAGE_RESOURCE_DIRECTORY_ENTRY));
|
|
|
|
pResData = (PIMAGE_RESOURCE_DATA_ENTRY)(((PUCHAR)pResDirL) +
|
|
cNames * sizeof(IMAGE_RESOURCE_DIRECTORY) +
|
|
cRes * sizeof(IMAGE_RESOURCE_DIRECTORY_ENTRY));
|
|
|
|
pResStr = (PUSHORT)(((PUCHAR)pResData) +
|
|
cRes * sizeof(IMAGE_RESOURCE_DATA_ENTRY));
|
|
|
|
pResStrEnd = (PUSHORT)(((PUCHAR)pResStr) + cbName + cbType);
|
|
|
|
/*
|
|
* Loop over type table, building the PE resource table.
|
|
*/
|
|
|
|
/*
|
|
* *****************************************************************
|
|
* This code doesn't sort the table - the TYPEINFO and RESINFO **
|
|
* insertion code in rcp.c (AddResType and SaveResFile) do the **
|
|
* insertion by ordinal type and name, so we don't have to sort **
|
|
* it at this point. **
|
|
* *****************************************************************
|
|
*/
|
|
|
|
// First, add all the entries in the Types: Alpha list.
|
|
|
|
pType = pUpdate->ResTypeHeadName;
|
|
while (pType)
|
|
{
|
|
pResDirT->Name = (ULONG)((((PUCHAR)pResStr) - p) | IMAGE_RESOURCE_NAME_IS_STRING);
|
|
pResDirT->OffsetToData = (ULONG)((((PUCHAR)pResDirN) - p) | IMAGE_RESOURCE_DATA_IS_DIRECTORY);
|
|
pResDirT++;
|
|
|
|
*pResStr = pType->Type->cbsz;
|
|
gwcsncpy((WCHAR*)(pResStr+1), pType->Type->szStr, pType->Type->cbsz);
|
|
pResStr += pType->Type->cbsz + 1;
|
|
|
|
pResTabN = (PIMAGE_RESOURCE_DIRECTORY)pResDirN;
|
|
SetRestab(pResTabN, clock, (USHORT)pType->NumberOfNamesName, (USHORT)pType->NumberOfNamesID);
|
|
pResDirN = (PIMAGE_RESOURCE_DIRECTORY_ENTRY)(pResTabN + 1);
|
|
|
|
pPreviousName = NULL;
|
|
|
|
pRes = pType->NameHeadName;
|
|
while (pRes)
|
|
{
|
|
if (pPreviousName == NULL ||
|
|
gwcsncmp(pPreviousName->szStr, pRes->Name->szStr, pRes->Name->cbsz) != 0)
|
|
{
|
|
// Setup a new name directory
|
|
|
|
pResDirN->Name = (ULONG)((((PUCHAR)pResStr)-p) | IMAGE_RESOURCE_NAME_IS_STRING);
|
|
pResDirN->OffsetToData = (ULONG)((((PUCHAR)pResDirL)-p) | IMAGE_RESOURCE_DATA_IS_DIRECTORY);
|
|
pResDirN++;
|
|
|
|
// Copy the alpha name to a string entry
|
|
|
|
*pResStr = pRes->Name->cbsz;
|
|
gwcsncpy((WCHAR*)(pResStr+1),pRes->Name->szStr,pRes->Name->cbsz);
|
|
pResStr += pRes->Name->cbsz + 1;
|
|
|
|
pPreviousName = pRes->Name;
|
|
|
|
// Setup the Language table
|
|
|
|
pResTabL = (PIMAGE_RESOURCE_DIRECTORY)pResDirL;
|
|
SetRestab(pResTabL, clock, (USHORT)0, (USHORT)pRes->NumberOfLanguages);
|
|
pResDirL = (PIMAGE_RESOURCE_DIRECTORY_ENTRY)(pResTabL + 1);
|
|
}
|
|
|
|
// Setup a new Language directory
|
|
|
|
pResDirL->Name = pRes->LanguageId;
|
|
pResDirL->OffsetToData = (ULONG)(((PUCHAR)pResData) - p);
|
|
pResDirL++;
|
|
|
|
// Setup a new resource data entry
|
|
|
|
SetResdata(pResData,
|
|
pRes->OffsetToData+pObjtblNew[nObjResource].VirtualAddress,
|
|
pRes->DataSize);
|
|
pResData++;
|
|
|
|
pRes = pRes->pnext;
|
|
}
|
|
|
|
pPreviousName = NULL;
|
|
|
|
pRes = pType->NameHeadID;
|
|
while (pRes)
|
|
{
|
|
if (pPreviousName == NULL || pPreviousName->uu.Ordinal != pRes->Name->uu.Ordinal)
|
|
{
|
|
// Setup the name directory to point to the next language
|
|
// table
|
|
|
|
pResDirN->Name = pRes->Name->uu.Ordinal;
|
|
pResDirN->OffsetToData = (ULONG)((((PUCHAR)pResDirL)-p) | IMAGE_RESOURCE_DATA_IS_DIRECTORY);
|
|
pResDirN++;
|
|
|
|
pPreviousName = pRes->Name;
|
|
|
|
// Init a new Language table
|
|
|
|
pResTabL = (PIMAGE_RESOURCE_DIRECTORY)pResDirL;
|
|
SetRestab(pResTabL, clock, (USHORT)0, (USHORT)pRes->NumberOfLanguages);
|
|
pResDirL = (PIMAGE_RESOURCE_DIRECTORY_ENTRY)(pResTabL + 1);
|
|
}
|
|
|
|
// Setup a new language directory entry to point to the next
|
|
// resource
|
|
|
|
pResDirL->Name = pRes->LanguageId;
|
|
pResDirL->OffsetToData = (ULONG)(((PUCHAR)pResData) - p);
|
|
pResDirL++;
|
|
|
|
// Setup a new resource data entry
|
|
|
|
SetResdata(pResData,
|
|
pRes->OffsetToData+pObjtblNew[nObjResource].VirtualAddress,
|
|
pRes->DataSize);
|
|
pResData++;
|
|
|
|
pRes = pRes->pnext;
|
|
}
|
|
|
|
pType = pType->pnext;
|
|
}
|
|
|
|
// Do the same thing, but this time, use the Types: ID list.
|
|
pType = pUpdate->ResTypeHeadID;
|
|
while (pType)
|
|
{
|
|
pResDirT->Name = (ULONG)pType->Type->uu.Ordinal;
|
|
pResDirT->OffsetToData = (ULONG)((((PUCHAR)pResDirN) - p) | IMAGE_RESOURCE_DATA_IS_DIRECTORY);
|
|
pResDirT++;
|
|
|
|
pResTabN = (PIMAGE_RESOURCE_DIRECTORY)pResDirN;
|
|
SetRestab(pResTabN, clock, (USHORT)pType->NumberOfNamesName, (USHORT)pType->NumberOfNamesID);
|
|
pResDirN = (PIMAGE_RESOURCE_DIRECTORY_ENTRY)(pResTabN + 1);
|
|
|
|
pPreviousName = NULL;
|
|
|
|
pRes = pType->NameHeadName;
|
|
while (pRes)
|
|
{
|
|
if (pPreviousName == NULL ||
|
|
gwcsncmp(pPreviousName->szStr, pRes->Name->szStr, pRes->Name->cbsz) != 0)
|
|
{
|
|
// Setup a new name directory
|
|
|
|
pResDirN->Name = (ULONG)((((PUCHAR)pResStr)-p) | IMAGE_RESOURCE_NAME_IS_STRING);
|
|
pResDirN->OffsetToData = (ULONG)((((PUCHAR)pResDirL)-p) | IMAGE_RESOURCE_DATA_IS_DIRECTORY);
|
|
pResDirN++;
|
|
|
|
// Copy the alpha name to a string entry.
|
|
|
|
*pResStr = pRes->Name->cbsz;
|
|
gwcsncpy((WCHAR*)(pResStr+1),pRes->Name->szStr,pRes->Name->cbsz);
|
|
pResStr += pRes->Name->cbsz + 1;
|
|
|
|
pPreviousName = pRes->Name;
|
|
|
|
// Setup the Language table
|
|
|
|
pResTabL = (PIMAGE_RESOURCE_DIRECTORY)pResDirL;
|
|
SetRestab(pResTabL, clock, (USHORT)0, (USHORT)pRes->NumberOfLanguages);
|
|
pResDirL = (PIMAGE_RESOURCE_DIRECTORY_ENTRY)(pResTabL + 1);
|
|
}
|
|
|
|
// Setup a new Language directory
|
|
|
|
pResDirL->Name = pRes->LanguageId;
|
|
pResDirL->OffsetToData = (ULONG)(((PUCHAR)pResData) - p);
|
|
pResDirL++;
|
|
|
|
// Setup a new resource data entry
|
|
|
|
SetResdata(pResData,
|
|
pRes->OffsetToData+pObjtblNew[nObjResource].VirtualAddress,
|
|
pRes->DataSize);
|
|
pResData++;
|
|
|
|
pRes = pRes->pnext;
|
|
}
|
|
|
|
pPreviousName = NULL;
|
|
|
|
pRes = pType->NameHeadID;
|
|
while (pRes)
|
|
{
|
|
if (pPreviousName == NULL || pPreviousName->uu.Ordinal != pRes->Name->uu.Ordinal)
|
|
{
|
|
// Setup the name directory to point to the next language
|
|
// table
|
|
|
|
pResDirN->Name = pRes->Name->uu.Ordinal;
|
|
pResDirN->OffsetToData = (ULONG)((((PUCHAR)pResDirL)-p) | IMAGE_RESOURCE_DATA_IS_DIRECTORY);
|
|
pResDirN++;
|
|
|
|
pPreviousName = pRes->Name;
|
|
|
|
// Init a new Language table
|
|
|
|
pResTabL = (PIMAGE_RESOURCE_DIRECTORY)pResDirL;
|
|
SetRestab(pResTabL, clock, (USHORT)0, (USHORT)pRes->NumberOfLanguages);
|
|
pResDirL = (PIMAGE_RESOURCE_DIRECTORY_ENTRY)(pResTabL + 1);
|
|
}
|
|
|
|
// Setup a new language directory entry to point to the next
|
|
// resource
|
|
|
|
pResDirL->Name = pRes->LanguageId;
|
|
pResDirL->OffsetToData = (ULONG)(((PUCHAR)pResData) - p);
|
|
pResDirL++;
|
|
|
|
// Setup a new resource data entry
|
|
|
|
SetResdata(pResData, pRes->OffsetToData+pObjtblNew[nObjResource].VirtualAddress, pRes->DataSize);
|
|
pResData++;
|
|
|
|
pRes = pRes->pnext;
|
|
}
|
|
|
|
pType = pType->pnext;
|
|
}
|
|
while (pResStr < pResStrEnd)
|
|
{
|
|
*pResStr++ = 0;
|
|
}
|
|
|
|
/*
|
|
* copy the Old exe header and stub, and allocate room for the PE header.
|
|
*/
|
|
MuMoveFilePos(inpfh, 0L);
|
|
MuCopy(inpfh, outfh, cbOldexe + sizeof(IMAGE_NT_HEADERS));
|
|
|
|
/*
|
|
* Copy rest of file header
|
|
*/
|
|
MuMoveFilePos(outfh, ibNewObjTabEnd + New.OptionalHeader.SizeOfHeaders -
|
|
Old.OptionalHeader.SizeOfHeaders);
|
|
MuMoveFilePos(inpfh, ibObjTabEnd);
|
|
MuCopy(inpfh, outfh, Old.OptionalHeader.SizeOfHeaders - ibNewObjTabEnd);
|
|
|
|
/*
|
|
* copy existing image sections
|
|
*/
|
|
|
|
/* Align data sections on sector boundary */
|
|
cb = REMAINDER(New.OptionalHeader.SizeOfHeaders, New.OptionalHeader.FileAlignment);
|
|
New.OptionalHeader.SizeOfHeaders += cb;
|
|
while (cb >= cbPadMax)
|
|
{
|
|
MuWrite(outfh, pchZero, cbPadMax);
|
|
cb -= cbPadMax;
|
|
}
|
|
MuWrite(outfh, pchZero, cb);
|
|
|
|
cb = ROUNDUP(Old.OptionalHeader.SizeOfHeaders, Old.OptionalHeader.FileAlignment);
|
|
MuMoveFilePos(inpfh, cb);
|
|
|
|
/* copy one section at a time */
|
|
New.OptionalHeader.SizeOfInitializedData = 0;
|
|
for (pObjOld = pObjtblOld , pObjNew = pObjtblNew ; pObjOld < pObjLast ; pObjNew++)
|
|
{
|
|
if (pObjOld == pObjResourceOldX)
|
|
pObjOld++;
|
|
if (pObjNew == pObjResourceNew)
|
|
{
|
|
|
|
/* Write new resource section */
|
|
pObjNew->PointerToRawData = FilePos(outfh);
|
|
New.OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_RESOURCE].VirtualAddress = pObjResourceNew->VirtualAddress;
|
|
New.OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_RESOURCE].Size = cbResource;
|
|
ibSave = FilePos(outfh);
|
|
MuWrite(outfh, (PUCHAR)pResTab, cbRestab);
|
|
|
|
pResSave = WriteResSection(pUpdate, outfh, New.OptionalHeader.FileAlignment,
|
|
pObjResourceNew->SizeOfRawData-cbRestab, NULL);
|
|
cb = FilePos(outfh);
|
|
if (cbMustPad != 0)
|
|
{
|
|
cbMustPad -= cb - ibSave;
|
|
/* assumes that cbMustPad % cbpadMax == 0 */
|
|
while (cbMustPad > 0)
|
|
{
|
|
MuWrite(outfh, pchZero, cbPadMax);
|
|
cbMustPad -= cbPadMax;
|
|
}
|
|
cb = FilePos(outfh);
|
|
}
|
|
if (nObjResourceX == -1)
|
|
{
|
|
MuMoveFilePos(outfh, ibSave);
|
|
MuWrite(outfh, (PUCHAR)pResTab, cbRestab);
|
|
MuMoveFilePos(outfh, cb);
|
|
cb = FilePos(inpfh);
|
|
MuMoveFilePos(inpfh, cb+pObjOld->SizeOfRawData);
|
|
}
|
|
New.OptionalHeader.SizeOfInitializedData += pObjNew->SizeOfRawData;
|
|
if (pObjResourceOld == NULL)
|
|
{
|
|
pObjNew++;
|
|
goto next_section;
|
|
}
|
|
else
|
|
pObjOld++;
|
|
}
|
|
else if (nObjResourceX != -1 && pObjNew == pObjtblNew + nObjResourceX)
|
|
{
|
|
|
|
/* Write new resource section */
|
|
pObjNew->PointerToRawData = FilePos(outfh);
|
|
(void)WriteResSection(pUpdate, outfh, New.OptionalHeader.FileAlignment, 0xffffffff, pResSave);
|
|
cb = FilePos(outfh);
|
|
pObjNew->SizeOfRawData = cb - pObjNew->PointerToRawData;
|
|
pObjNew->Misc.VirtualSize = ROUNDUP(pObjNew->SizeOfRawData,
|
|
New.OptionalHeader.SectionAlignment);
|
|
MuMoveFilePos(outfh, ibSave);
|
|
MuWrite(outfh, (PUCHAR)pResTab, cbRestab);
|
|
MuMoveFilePos(outfh, cb);
|
|
New.OptionalHeader.SizeOfInitializedData += pObjNew->SizeOfRawData;
|
|
pObjNew++;
|
|
goto next_section;
|
|
}
|
|
else
|
|
{
|
|
if (pObjNew < pObjResourceNew &&
|
|
pObjOld->PointerToRawData != 0 &&
|
|
pObjOld->PointerToRawData != FilePos(outfh))
|
|
{
|
|
MuMoveFilePos(outfh, pObjOld->PointerToRawData);
|
|
}
|
|
next_section:
|
|
if (pObjOld->PointerToRawData != 0)
|
|
{
|
|
pObjNew->PointerToRawData = FilePos(outfh);
|
|
MuMoveFilePos(inpfh, pObjOld->PointerToRawData);
|
|
MuCopy(inpfh, outfh, pObjOld->SizeOfRawData);
|
|
}
|
|
if (pObjOld == pObjDebugDirOld)
|
|
{
|
|
pObjDebugDirNew = pObjNew;
|
|
}
|
|
if ((pObjNew->Characteristics&IMAGE_SCN_CNT_INITIALIZED_DATA) != 0)
|
|
New.OptionalHeader.SizeOfInitializedData += pObjNew->SizeOfRawData;
|
|
pObjOld++;
|
|
}
|
|
}
|
|
if (pObjResourceOldX != NULL)
|
|
New.OptionalHeader.SizeOfInitializedData -= pObjResourceOldX->SizeOfRawData;
|
|
|
|
/* Update the address of the relocation table */
|
|
pObjNew = FindSection(pObjtblNew, pObjtblNew+New.FileHeader.NumberOfSections, ".reloc");
|
|
if (pObjNew != NULL)
|
|
{
|
|
New.OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_BASERELOC].VirtualAddress = pObjNew->VirtualAddress;
|
|
}
|
|
|
|
/*
|
|
* Write new section table out.
|
|
*/
|
|
MuMoveFilePos(outfh, ibObjTab);
|
|
MuWrite(outfh, (PUCHAR)pObjtblNew, New.FileHeader.NumberOfSections * sizeof(IMAGE_SECTION_HEADER));
|
|
|
|
/* Seek to end of output file and issue truncating write */
|
|
adjust = _llseek(outfh, 0L, SEEK_END);
|
|
MuWrite(outfh, NULL, 0);
|
|
|
|
/* If a debug section, fix up the debug table */
|
|
pObjNew = FindSection(pObjtblNew, pObjtblNew+New.FileHeader.NumberOfSections, ".debug");
|
|
cb = PatchDebug(inpfh, outfh, pObjDebug, pObjNew, pObjDebugDirOld,
|
|
pObjDebugDirNew, &Old, &New, ibMaxDbgOffsetOld, &adjust);
|
|
|
|
if (cb == NO_ERROR)
|
|
{
|
|
if (pObjResourceOld == NULL)
|
|
{
|
|
cb = (LONG)pObjResourceNew->SizeOfRawData;
|
|
}
|
|
else
|
|
{
|
|
cb = (LONG)pObjResourceOld->SizeOfRawData - (LONG)pObjResourceNew->SizeOfRawData;
|
|
}
|
|
cb = PatchRVAs(inpfh, outfh, pObjtblNew, cb, &New, Old.OptionalHeader.SizeOfHeaders);
|
|
}
|
|
|
|
/* copy NOTMAPPED debug info */
|
|
if (pObjDebugDirOld != NULL && pObjDebug == NULL &&
|
|
New.OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_DEBUG].Size != 0)
|
|
{
|
|
ULONG ibt;
|
|
|
|
ibSave = _llseek(inpfh, 0L, SEEK_END); /* copy debug data */
|
|
ibt = _llseek(outfh, 0L, SEEK_END); /* to EOF */
|
|
if (New.FileHeader.PointerToSymbolTable != 0)
|
|
New.FileHeader.PointerToSymbolTable += ibt - adjust;
|
|
MuMoveFilePos(inpfh, adjust); /* returned by PatchDebug */
|
|
MuCopy(inpfh, outfh, ibSave-adjust);
|
|
}
|
|
|
|
/*
|
|
* Write updated PE header
|
|
*/
|
|
MuMoveFilePos(outfh, (long)cbOldexe);
|
|
MuWrite(outfh, (char*)&New, sizeof(IMAGE_NT_HEADERS));
|
|
|
|
/* free up allocated memory */
|
|
GodotHeapFree( pObjtblOld);
|
|
GodotHeapFree( pResTab);
|
|
|
|
AbortExit:
|
|
GodotHeapFree( pObjtblNew);
|
|
return cb;
|
|
}
|
|
|
|
|
|
/***************************************************************************
|
|
* WriteResSection
|
|
*
|
|
* This routine writes out the resources asked for into the current section.
|
|
* It pads resources to dword (4-byte) boundaries.
|
|
**************************************************************************/
|
|
|
|
PRESNAME WriteResSection(PUPDATEDATA pUpdate, INT outfh, ULONG align, ULONG cbLeft, PRESNAME pResSave)
|
|
{
|
|
ULONG cbB=0; /* bytes in current section */
|
|
ULONG cbT; /* bytes in current section */
|
|
ULONG size;
|
|
PRESNAME pRes;
|
|
PRESTYPE pType;
|
|
BOOL fName;
|
|
PVOID lpData;
|
|
|
|
/* Output contents associated with each resource */
|
|
pType = pUpdate->ResTypeHeadName;
|
|
while (pType) {
|
|
pRes = pType->NameHeadName;
|
|
fName = TRUE;
|
|
loop1:
|
|
for ( ; pRes ; pRes = pRes->pnext)
|
|
{
|
|
if (pResSave != NULL && pRes != pResSave)
|
|
continue;
|
|
pResSave = NULL;
|
|
lpData = (PVOID)pRes->OffsetToDataEntry;
|
|
|
|
/* if there is room in the current section, write it there */
|
|
size = pRes->DataSize;
|
|
if (cbLeft != 0 && cbLeft >= size) /* resource fits? */
|
|
{
|
|
MuWrite(outfh, lpData, size);
|
|
/* pad resource */
|
|
cbT = REMAINDER(size, CBLONG);
|
|
MuWrite(outfh, pchPad, cbT); /* dword */
|
|
cbB += size + cbT;
|
|
cbLeft -= size + cbT; /* less left */
|
|
continue; /* next resource */
|
|
}
|
|
else /* will fill up section */
|
|
{
|
|
goto write_pad;
|
|
}
|
|
}
|
|
if (fName)
|
|
{
|
|
fName = FALSE;
|
|
pRes = pType->NameHeadID;
|
|
goto loop1;
|
|
}
|
|
pType = pType->pnext;
|
|
}
|
|
|
|
pType = pUpdate->ResTypeHeadID;
|
|
while (pType)
|
|
{
|
|
pRes = pType->NameHeadName;
|
|
fName = TRUE;
|
|
loop2:
|
|
for ( ; pRes ; pRes = pRes->pnext)
|
|
{
|
|
if (pResSave != NULL && pRes != pResSave)
|
|
continue;
|
|
pResSave = NULL;
|
|
lpData = (PVOID)pRes->OffsetToDataEntry;
|
|
|
|
/* if there is room in the current section, write it there */
|
|
size = pRes->DataSize;
|
|
if (cbLeft != 0 && cbLeft >= size) /* resource fits? */
|
|
{
|
|
MuWrite(outfh, lpData, size);
|
|
/* pad resource */
|
|
cbT = REMAINDER(size, CBLONG);
|
|
MuWrite(outfh, pchPad, cbT); /* dword */
|
|
cbB += size + cbT;
|
|
cbLeft -= size + cbT; /* less left */
|
|
continue; /* next resource */
|
|
}
|
|
else /* will fill up section */
|
|
{
|
|
goto write_pad;
|
|
}
|
|
}
|
|
if (fName)
|
|
{
|
|
fName = FALSE;
|
|
pRes = pType->NameHeadID;
|
|
goto loop2;
|
|
}
|
|
pType = pType->pnext;
|
|
}
|
|
pRes = NULL;
|
|
|
|
write_pad:
|
|
/* pad to alignment boundary */
|
|
cbB = FilePos(outfh);
|
|
cbT = ROUNDUP(cbB, align);
|
|
cbLeft = cbT - cbB;
|
|
if (cbLeft != 0)
|
|
{
|
|
while (cbLeft >= cbPadMax)
|
|
{
|
|
MuWrite(outfh, pchPad, cbPadMax);
|
|
cbLeft -= cbPadMax;
|
|
}
|
|
MuWrite(outfh, pchPad, cbLeft);
|
|
}
|
|
return pRes;
|
|
}
|
|
|
|
LONG PatchDebug(int inpfh, int outfh, PIMAGE_SECTION_HEADER pDebugOld, PIMAGE_SECTION_HEADER pDebugNew,
|
|
PIMAGE_SECTION_HEADER pDebugDirOld, PIMAGE_SECTION_HEADER pDebugDirNew,
|
|
PIMAGE_NT_HEADERS pOld, PIMAGE_NT_HEADERS pNew, ULONG ibMaxDbgOffsetOld,
|
|
PULONG pPointerToRawData)
|
|
//
|
|
// adjust debug directory table
|
|
//
|
|
{
|
|
PIMAGE_DEBUG_DIRECTORY pDbgLast;
|
|
PIMAGE_DEBUG_DIRECTORY pDbgSave;
|
|
PIMAGE_DEBUG_DIRECTORY pDbg;
|
|
ULONG ib;
|
|
ULONG adjust;
|
|
ULONG ibNew;
|
|
|
|
if (pDebugDirOld == NULL || pNew->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_DEBUG].Size==0)
|
|
return NO_ERROR;
|
|
|
|
pDbg = (PIMAGE_DEBUG_DIRECTORY)GodotHeapAlloc(pNew->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_DEBUG].Size);
|
|
pDbgSave = pDbg;
|
|
if (pDbg == NULL)
|
|
return ERROR_NOT_ENOUGH_MEMORY;
|
|
|
|
if (pDebugOld)
|
|
{
|
|
}
|
|
else
|
|
adjust = *pPointerToRawData; /* passed in EOF of new file */
|
|
|
|
ib = pOld->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_DEBUG].VirtualAddress - pDebugDirOld->VirtualAddress;
|
|
MuMoveFilePos(inpfh, pDebugDirOld->PointerToRawData+ib);
|
|
pDbgLast = pDbg + (pNew->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_DEBUG].Size)/sizeof(IMAGE_DEBUG_DIRECTORY);
|
|
MuRead(inpfh, (PUCHAR)pDbg, pNew->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_DEBUG].Size);
|
|
|
|
if (pDebugOld == NULL)
|
|
{
|
|
/* find 1st entry - use for offset */
|
|
for (ibNew=0xffffffff ; pDbg<pDbgLast ; pDbg++)
|
|
if (pDbg->PointerToRawData >= ibMaxDbgOffsetOld && pDbg->PointerToRawData < ibNew)
|
|
ibNew = pDbg->PointerToRawData;
|
|
|
|
if (ibNew != 0xffffffff)
|
|
*pPointerToRawData = ibNew;
|
|
else
|
|
*pPointerToRawData = _llseek(inpfh, 0L, SEEK_END);
|
|
|
|
for (pDbg=pDbgSave ; pDbg<pDbgLast ; pDbg++)
|
|
{
|
|
if (pDbg->PointerToRawData >= ibMaxDbgOffsetOld)
|
|
pDbg->PointerToRawData += adjust - ibNew;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
for ( ; pDbg<pDbgLast ; pDbg++)
|
|
{
|
|
pDbg->AddressOfRawData += pDebugNew->VirtualAddress - pDebugOld->VirtualAddress;
|
|
pDbg->PointerToRawData += pDebugNew->PointerToRawData - pDebugOld->PointerToRawData;
|
|
}
|
|
}
|
|
|
|
MuMoveFilePos(outfh, pDebugDirNew->PointerToRawData+ib);
|
|
MuWrite(outfh, (PUCHAR)pDbgSave, pNew->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_DEBUG].Size);
|
|
GodotHeapFree(pDbgSave);
|
|
|
|
return NO_ERROR;
|
|
}
|
|
|
|
|
|
//
|
|
// This routine patches various RVAs in the file to compensate
|
|
// for extra section table entries.
|
|
//
|
|
LONG PatchRVAs(int inpfh, int outfh, PIMAGE_SECTION_HEADER po32,
|
|
ULONG pagedelta, PIMAGE_NT_HEADERS pNew, ULONG OldSize)
|
|
{
|
|
ULONG hdrdelta;
|
|
ULONG offset, rvaiat, offiat, iat;
|
|
IMAGE_EXPORT_DIRECTORY Exp;
|
|
IMAGE_IMPORT_DESCRIPTOR Imp;
|
|
ULONG i, cmod, cimp;
|
|
|
|
hdrdelta = pNew->OptionalHeader.SizeOfHeaders - OldSize;
|
|
if (hdrdelta == 0)
|
|
{
|
|
return NO_ERROR;
|
|
}
|
|
|
|
//
|
|
// Patch export section RVAs
|
|
//
|
|
offset = ((pNew->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_EXPORT].VirtualAddress) == 0);
|
|
if (offset < pNew->OptionalHeader.SizeOfHeaders)
|
|
{
|
|
MuMoveFilePos(inpfh, offset - hdrdelta);
|
|
MuRead(inpfh, (PUCHAR) &Exp, sizeof(Exp));
|
|
Exp.Name += hdrdelta;
|
|
(ULONG)Exp.AddressOfFunctions += hdrdelta;
|
|
(ULONG)Exp.AddressOfNames += hdrdelta;
|
|
(ULONG)Exp.AddressOfNameOrdinals += hdrdelta;
|
|
MuMoveFilePos(outfh, offset);
|
|
MuWrite(outfh, (PUCHAR) &Exp, sizeof(Exp));
|
|
}
|
|
|
|
//
|
|
// Patch import section RVAs
|
|
//
|
|
offset = ((pNew->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_IMPORT].VirtualAddress) == 0);
|
|
if (offset < pNew->OptionalHeader.SizeOfHeaders)
|
|
{
|
|
for (cimp = cmod = 0; ; cmod++)
|
|
{
|
|
MuMoveFilePos(inpfh, offset + cmod * sizeof(Imp) - hdrdelta);
|
|
MuRead(inpfh, (PUCHAR) &Imp, sizeof(Imp));
|
|
if (Imp.FirstThunk == 0)
|
|
{
|
|
break;
|
|
}
|
|
Imp.Name += hdrdelta;
|
|
MuMoveFilePos(outfh, offset + cmod * sizeof(Imp));
|
|
MuWrite(outfh, (PUCHAR) &Imp, sizeof(Imp));
|
|
|
|
rvaiat = (ULONG)Imp.FirstThunk;
|
|
for (i = 0; i < pNew->FileHeader.NumberOfSections; i++)
|
|
{
|
|
if (rvaiat >= po32[i].VirtualAddress &&
|
|
rvaiat < po32[i].VirtualAddress + po32[i].SizeOfRawData)
|
|
{
|
|
offiat = rvaiat - po32[i].VirtualAddress + po32[i].PointerToRawData;
|
|
goto found;
|
|
}
|
|
}
|
|
return ERROR_INVALID_DATA;
|
|
found:
|
|
MuMoveFilePos(inpfh, offiat - pagedelta);
|
|
MuMoveFilePos(outfh, offiat);
|
|
for (;;)
|
|
{
|
|
MuRead(inpfh, (PUCHAR) &iat, sizeof(iat));
|
|
if (iat == 0)
|
|
{
|
|
break;
|
|
}
|
|
if ((iat & IMAGE_ORDINAL_FLAG) == 0) // if import by name
|
|
{
|
|
iat += hdrdelta;
|
|
cimp++;
|
|
}
|
|
MuWrite(outfh, (PUCHAR) &iat, sizeof(iat)); // Avoids seeking
|
|
}
|
|
}
|
|
}
|
|
|
|
return NO_ERROR;
|
|
|
|
}
|
|
|
|
|
|
/*---------------------------------------------------------------------------*/
|
|
/* */
|
|
/* WriteResFile() - */
|
|
/* */
|
|
/*---------------------------------------------------------------------------*/
|
|
LONG WriteResFile(HANDLE hUpdate, WCHAR *pDstname)
|
|
{
|
|
INT inpfh;
|
|
INT outfh;
|
|
ULONG onewexe;
|
|
IMAGE_DOS_HEADER oldexe;
|
|
PUPDATEDATA pUpdate;
|
|
INT rc;
|
|
WCHAR *pFilename;
|
|
|
|
pUpdate = (PUPDATEDATA)GlobalLock(hUpdate);
|
|
pFilename = (WCHAR*)GlobalLock(pUpdate->hFileName);
|
|
|
|
/* open the original exe file */
|
|
inpfh = (INT)CreateFileW(pFilename, GENERIC_READ, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
|
|
GlobalUnlock(pUpdate->hFileName);
|
|
if (inpfh == -1)
|
|
{
|
|
GlobalUnlock(hUpdate);
|
|
return ERROR_OPEN_FAILED;
|
|
}
|
|
|
|
/* read the old format EXE header */
|
|
rc = _lread(inpfh, (char*)&oldexe, sizeof(oldexe));
|
|
if (rc != sizeof(oldexe))
|
|
{
|
|
_lclose(inpfh);
|
|
GlobalUnlock(hUpdate);
|
|
return ERROR_READ_FAULT;
|
|
}
|
|
|
|
/* make sure its really an EXE file */
|
|
if (oldexe.e_magic != IMAGE_DOS_SIGNATURE)
|
|
{
|
|
_lclose(inpfh);
|
|
GlobalUnlock(hUpdate);
|
|
return ERROR_INVALID_EXE_SIGNATURE;
|
|
}
|
|
|
|
/* make sure theres a new EXE header floating around somewhere */
|
|
if (!(onewexe = oldexe.e_lfanew))
|
|
{
|
|
_lclose(inpfh);
|
|
GlobalUnlock(hUpdate);
|
|
return ERROR_BAD_EXE_FORMAT;
|
|
}
|
|
|
|
outfh = (INT)CreateFileW(pDstname, GENERIC_READ|GENERIC_WRITE, 0, NULL,
|
|
CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
|
|
|
|
if (outfh != -1)
|
|
{
|
|
rc = PEWriteResFile(inpfh, outfh, onewexe, pUpdate);
|
|
_lclose(outfh);
|
|
}
|
|
_lclose(inpfh);
|
|
GlobalUnlock(hUpdate);
return rc;
}
|