ReceiveTempSMS
/
Sweden-Number
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

kernel32: Fix write_raw_resources to support 32 and 64 PE formats.

This commit is contained in:
Jon Doron 2015-01-01 16:46:58 +02:00 committed by Alexandre Julliard
parent a3f33daa6a
commit 3c28c0a2eb
1 changed files with 51 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

73
dlls/kernel32/resource.c
View File

@ -850,8 +850,10 @@ static IMAGE_SECTION_HEADER *get_section_header( void *base, DWORD mapping_size,
static BOOL check_pe_exe( HANDLE file, QUEUEDUPDATES *updates ) static BOOL check_pe_exe( HANDLE file, QUEUEDUPDATES *updates )
{ {
const IMAGE_NT_HEADERS *nt; const IMAGE_NT_HEADERS32 *nt;
const IMAGE_NT_HEADERS64 *nt64;
const IMAGE_SECTION_HEADER *sec; const IMAGE_SECTION_HEADER *sec;
const IMAGE_DATA_DIRECTORY *dd;
BOOL ret = FALSE; BOOL ret = FALSE;
HANDLE mapping; HANDLE mapping;
DWORD mapping_size, num_sections = 0; DWORD mapping_size, num_sections = 0;
@ -867,13 +869,18 @@ static BOOL check_pe_exe( HANDLE file, QUEUEDUPDATES *updates )
if (!base) if (!base)
goto done; goto done;
nt = get_nt_header( base, mapping_size ); nt = (IMAGE_NT_HEADERS32 *)get_nt_header( base, mapping_size );
if (!nt) if (!nt)
goto done; goto done;
nt64 = (IMAGE_NT_HEADERS64*)nt;
dd = &nt->OptionalHeader.DataDirectory[0];
if (nt->OptionalHeader.Magic == IMAGE_NT_OPTIONAL_HDR64_MAGIC)
dd = &nt64->OptionalHeader.DataDirectory[0];
TRACE("resources: %08x %08x\n", TRACE("resources: %08x %08x\n",
nt->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_RESOURCE].VirtualAddress, dd[IMAGE_DIRECTORY_ENTRY_RESOURCE].VirtualAddress,
nt->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_RESOURCE].Size); dd[IMAGE_DIRECTORY_ENTRY_RESOURCE].Size);
sec = get_section_header( base, mapping_size, &num_sections ); sec = get_section_header( base, mapping_size, &num_sections );
if (!sec) if (!sec)
@ -1393,10 +1400,12 @@ static BOOL write_raw_resources( QUEUEDUPDATES *updates )
DWORD section_size; DWORD section_size;
BOOL ret = FALSE; BOOL ret = FALSE;
IMAGE_SECTION_HEADER *sec; IMAGE_SECTION_HEADER *sec;
IMAGE_NT_HEADERS *nt; IMAGE_NT_HEADERS32 *nt;
IMAGE_NT_HEADERS64 *nt64;
struct resource_size_info res_size; struct resource_size_info res_size;
BYTE *res_base; BYTE *res_base;
struct mapping_info *read_map = NULL, *write_map = NULL; struct mapping_info *read_map = NULL, *write_map = NULL;
DWORD PeSectionAlignment, PeFileAlignment, PeSizeOfImage;
/* copy the exe to a temp file then update the temp file... */ /* copy the exe to a temp file then update the temp file... */
tempdir[0] = 0; tempdir[0] = 0;
@ -1429,19 +1438,30 @@ static BOOL write_raw_resources( QUEUEDUPDATES *updates )
if (!write_map) if (!write_map)
goto done; goto done;
nt = get_nt_header( write_map->base, write_map->size ); nt = (IMAGE_NT_HEADERS32*)get_nt_header( write_map->base, write_map->size );
if (!nt) if (!nt)
goto done; goto done;
if (nt->OptionalHeader.SectionAlignment <= 0) nt64 = (IMAGE_NT_HEADERS64*)nt;
if (nt->OptionalHeader.Magic == IMAGE_NT_OPTIONAL_HDR64_MAGIC) {
PeSectionAlignment = nt64->OptionalHeader.SectionAlignment;
PeFileAlignment = nt64->OptionalHeader.FileAlignment;
PeSizeOfImage = nt64->OptionalHeader.SizeOfImage;
} else {
PeSectionAlignment = nt->OptionalHeader.SectionAlignment;
PeFileAlignment = nt->OptionalHeader.FileAlignment;
PeSizeOfImage = nt->OptionalHeader.SizeOfImage;
}
if ((LONG)PeSectionAlignment <= 0)
{ {
ERR("invalid section alignment %04x\n", nt->OptionalHeader.SectionAlignment); ERR("invalid section alignment %08x\n", PeSectionAlignment);
goto done; goto done;
} }
if (nt->OptionalHeader.FileAlignment <= 0) if ((LONG)PeFileAlignment <= 0)
{ {
ERR("invalid file alignment %04x\n", nt->OptionalHeader.FileAlignment); ERR("invalid file alignment %08x\n", PeFileAlignment);
goto done; goto done;
} }
@ -1460,12 +1480,12 @@ static BOOL write_raw_resources( QUEUEDUPDATES *updates )
memset( sec, 0, sizeof *sec ); memset( sec, 0, sizeof *sec );
memcpy( sec->Name, ".rsrc", 5 ); memcpy( sec->Name, ".rsrc", 5 );
sec->Characteristics = IMAGE_SCN_CNT_INITIALIZED_DATA | IMAGE_SCN_MEM_READ; sec->Characteristics = IMAGE_SCN_CNT_INITIALIZED_DATA | IMAGE_SCN_MEM_READ;
sec->VirtualAddress = nt->OptionalHeader.SizeOfImage; sec->VirtualAddress = PeSizeOfImage;
} }
if (!sec->PointerToRawData) /* empty section */ if (!sec->PointerToRawData) /* empty section */
{ {
sec->PointerToRawData = write_map->size + (-write_map->size) % nt->OptionalHeader.FileAlignment; sec->PointerToRawData = write_map->size + (-write_map->size) % PeFileAlignment;
sec->SizeOfRawData = 0; sec->SizeOfRawData = 0;
} }
@ -1475,7 +1495,7 @@ static BOOL write_raw_resources( QUEUEDUPDATES *updates )
/* round up the section size */ /* round up the section size */
section_size = res_size.total_size; section_size = res_size.total_size;
section_size += (-section_size) % nt->OptionalHeader.FileAlignment; section_size += (-section_size) % PeFileAlignment;
TRACE("requires %08x (%08x) bytes\n", res_size.total_size, section_size ); TRACE("requires %08x (%08x) bytes\n", res_size.total_size, section_size );
@ -1483,10 +1503,10 @@ static BOOL write_raw_resources( QUEUEDUPDATES *updates )
if (section_size != sec->SizeOfRawData) if (section_size != sec->SizeOfRawData)
{ {
DWORD old_size = write_map->size; DWORD old_size = write_map->size;
DWORD virtual_section_size = res_size.total_size + (-res_size.total_size) % nt->OptionalHeader.SectionAlignment; DWORD virtual_section_size = res_size.total_size + (-res_size.total_size) % PeSectionAlignment;
int delta = section_size - (sec->SizeOfRawData + (-sec->SizeOfRawData) % nt->OptionalHeader.FileAlignment); int delta = section_size - (sec->SizeOfRawData + (-sec->SizeOfRawData) % PeFileAlignment);
int rva_delta = virtual_section_size - int rva_delta = virtual_section_size -
(sec->Misc.VirtualSize + (-sec->Misc.VirtualSize) % nt->OptionalHeader.SectionAlignment); (sec->Misc.VirtualSize + (-sec->Misc.VirtualSize) % PeSectionAlignment);
BOOL rsrc_is_last = sec->PointerToRawData + sec->SizeOfRawData == old_size; BOOL rsrc_is_last = sec->PointerToRawData + sec->SizeOfRawData == old_size;
/* align .rsrc size when possible */ /* align .rsrc size when possible */
DWORD mapping_size = rsrc_is_last ? sec->PointerToRawData + section_size : old_size + delta; DWORD mapping_size = rsrc_is_last ? sec->PointerToRawData + section_size : old_size + delta;
@ -1502,12 +1522,13 @@ static BOOL write_raw_resources( QUEUEDUPDATES *updates )
ret = resize_mapping( write_map, mapping_size ); ret = resize_mapping( write_map, mapping_size );
/* get the pointers again - they might be different after remapping */ /* get the pointers again - they might be different after remapping */
nt = get_nt_header( write_map->base, mapping_size ); nt = (IMAGE_NT_HEADERS32*)get_nt_header( write_map->base, mapping_size );
if (!nt) if (!nt)
{ {
ERR("couldn't get NT header\n"); ERR("couldn't get NT header\n");
goto done; goto done;
} }
nt64 = (IMAGE_NT_HEADERS64*)nt;
sec = get_resource_section( write_map->base, mapping_size ); sec = get_resource_section( write_map->base, mapping_size );
if (!sec) if (!sec)
@ -1538,12 +1559,13 @@ static BOOL write_raw_resources( QUEUEDUPDATES *updates )
{ {
ret = resize_mapping( write_map, mapping_size ); ret = resize_mapping( write_map, mapping_size );
nt = get_nt_header( write_map->base, mapping_size ); nt = (IMAGE_NT_HEADERS32*)get_nt_header( write_map->base, mapping_size );
if (!nt) if (!nt)
{ {
ERR("couldn't get NT header\n"); ERR("couldn't get NT header\n");
goto done; goto done;
} }
nt64 = (IMAGE_NT_HEADERS64*)nt;
sec = get_resource_section( write_map->base, mapping_size ); sec = get_resource_section( write_map->base, mapping_size );
if (!sec) if (!sec)
@ -1553,10 +1575,17 @@ static BOOL write_raw_resources( QUEUEDUPDATES *updates )
/* adjust the PE header information */ /* adjust the PE header information */
sec->SizeOfRawData = section_size; sec->SizeOfRawData = section_size;
sec->Misc.VirtualSize = virtual_section_size; sec->Misc.VirtualSize = virtual_section_size;
nt->OptionalHeader.SizeOfImage += rva_delta; if (nt->OptionalHeader.Magic == IMAGE_NT_OPTIONAL_HDR64_MAGIC) {
nt->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_RESOURCE].VirtualAddress = sec->VirtualAddress; nt64->OptionalHeader.SizeOfImage += rva_delta;
nt->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_RESOURCE].Size = res_size.total_size; nt64->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_RESOURCE].VirtualAddress = sec->VirtualAddress;
nt->OptionalHeader.SizeOfInitializedData = get_init_data_size( write_map->base, mapping_size ); nt64->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_RESOURCE].Size = res_size.total_size;
nt64->OptionalHeader.SizeOfInitializedData = get_init_data_size( write_map->base, mapping_size );
} else {
nt->OptionalHeader.SizeOfImage += rva_delta;
nt->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_RESOURCE].VirtualAddress = sec->VirtualAddress;
nt->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_RESOURCE].Size = res_size.total_size;
nt->OptionalHeader.SizeOfInitializedData = get_init_data_size( write_map->base, mapping_size );
}
} }
res_base = (LPBYTE) write_map->base + sec->PointerToRawData; res_base = (LPBYTE) write_map->base + sec->PointerToRawData;