mirror of
https://github.com/reactos/reactos.git
synced 2024-08-19 18:02:55 +00:00
368 lines
11 KiB
C
368 lines
11 KiB
C
/*
|
|
* Implementation of VERSION.DLL - Resource Access routines
|
|
*
|
|
* Copyright 1996,1997 Marcus Meissner
|
|
* Copyright 1997 David Cuthbert
|
|
* Copyright 1999 Ulrich Weigand
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
|
|
*/
|
|
|
|
#include "config.h"
|
|
|
|
#include <stdarg.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <sys/types.h>
|
|
#ifdef HAVE_UNISTD_H
|
|
# include <unistd.h>
|
|
#endif
|
|
|
|
#define NONAMELESSUNION
|
|
#define NONAMELESSSTRUCT
|
|
#include "windef.h"
|
|
#include "winbase.h"
|
|
#include "lzexpand.h"
|
|
#include "winuser.h"
|
|
#include "winver.h"
|
|
#undef VS_FILE_INFO
|
|
#define VS_FILE_INFO 16
|
|
|
|
#include "wine/unicode.h"
|
|
|
|
#include "wine/debug.h"
|
|
|
|
typedef struct
|
|
{
|
|
WORD offset;
|
|
WORD length;
|
|
WORD flags;
|
|
WORD id;
|
|
WORD handle;
|
|
WORD usage;
|
|
} NE_NAMEINFO;
|
|
|
|
typedef struct
|
|
{
|
|
WORD type_id;
|
|
WORD count;
|
|
DWORD resloader;
|
|
} NE_TYPEINFO;
|
|
|
|
WINE_DEFAULT_DEBUG_CHANNEL(ver);
|
|
|
|
|
|
/**********************************************************************
|
|
* find_entry_by_id
|
|
*
|
|
* Find an entry by id in a resource directory
|
|
* Copied from loader/pe_resource.c
|
|
*/
|
|
static const IMAGE_RESOURCE_DIRECTORY *find_entry_by_id( const IMAGE_RESOURCE_DIRECTORY *dir,
|
|
WORD id, const void *root )
|
|
{
|
|
const IMAGE_RESOURCE_DIRECTORY_ENTRY *entry;
|
|
int min, max, pos;
|
|
|
|
entry = (const IMAGE_RESOURCE_DIRECTORY_ENTRY *)(dir + 1);
|
|
min = dir->NumberOfNamedEntries;
|
|
max = min + dir->NumberOfIdEntries - 1;
|
|
while (min <= max)
|
|
{
|
|
pos = (min + max) / 2;
|
|
if (entry[pos].u1.Id == id)
|
|
return (const IMAGE_RESOURCE_DIRECTORY *)((const char *)root + entry[pos].u2.s3.OffsetToDirectory);
|
|
if (entry[pos].u1.Id > id) max = pos - 1;
|
|
else min = pos + 1;
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
|
|
/**********************************************************************
|
|
* find_entry_default
|
|
*
|
|
* Find a default entry in a resource directory
|
|
* Copied from loader/pe_resource.c
|
|
*/
|
|
static const IMAGE_RESOURCE_DIRECTORY *find_entry_default( const IMAGE_RESOURCE_DIRECTORY *dir,
|
|
const void *root )
|
|
{
|
|
const IMAGE_RESOURCE_DIRECTORY_ENTRY *entry;
|
|
|
|
entry = (const IMAGE_RESOURCE_DIRECTORY_ENTRY *)(dir + 1);
|
|
return (const IMAGE_RESOURCE_DIRECTORY *)((const char *)root + entry->u2.s3.OffsetToDirectory);
|
|
}
|
|
|
|
|
|
/***********************************************************************
|
|
* read_xx_header [internal]
|
|
*/
|
|
static int read_xx_header( HFILE lzfd )
|
|
{
|
|
IMAGE_DOS_HEADER mzh;
|
|
char magic[3];
|
|
|
|
LZSeek( lzfd, 0, SEEK_SET );
|
|
if ( sizeof(mzh) != LZRead( lzfd, (LPSTR)&mzh, sizeof(mzh) ) )
|
|
return 0;
|
|
if ( mzh.e_magic != IMAGE_DOS_SIGNATURE )
|
|
{
|
|
if (!memcmp( &mzh, "\177ELF", 4 )) return 1; /* ELF */
|
|
if (*(UINT *)&mzh == 0xfeedface || *(UINT *)&mzh == 0xcefaedfe) return 1; /* Mach-O */
|
|
return 0;
|
|
}
|
|
|
|
LZSeek( lzfd, mzh.e_lfanew, SEEK_SET );
|
|
if ( 2 != LZRead( lzfd, magic, 2 ) )
|
|
return 0;
|
|
|
|
LZSeek( lzfd, mzh.e_lfanew, SEEK_SET );
|
|
|
|
if ( magic[0] == 'N' && magic[1] == 'E' )
|
|
return IMAGE_OS2_SIGNATURE;
|
|
if ( magic[0] == 'P' && magic[1] == 'E' )
|
|
return IMAGE_NT_SIGNATURE;
|
|
|
|
magic[2] = '\0';
|
|
WARN("Can't handle %s files.\n", magic );
|
|
return 0;
|
|
}
|
|
|
|
/***********************************************************************
|
|
* find_ne_resource [internal]
|
|
*/
|
|
static BOOL find_ne_resource( HFILE lzfd, DWORD *resLen, DWORD *resOff )
|
|
{
|
|
const WORD typeid = VS_FILE_INFO | 0x8000;
|
|
const WORD resid = VS_VERSION_INFO | 0x8000;
|
|
IMAGE_OS2_HEADER nehd;
|
|
NE_TYPEINFO *typeInfo;
|
|
NE_NAMEINFO *nameInfo;
|
|
DWORD nehdoffset;
|
|
LPBYTE resTab;
|
|
DWORD resTabSize;
|
|
int count;
|
|
|
|
/* Read in NE header */
|
|
nehdoffset = LZSeek( lzfd, 0, SEEK_CUR );
|
|
if ( sizeof(nehd) != LZRead( lzfd, (LPSTR)&nehd, sizeof(nehd) ) ) return 0;
|
|
|
|
resTabSize = nehd.ne_restab - nehd.ne_rsrctab;
|
|
if ( !resTabSize )
|
|
{
|
|
TRACE("No resources in NE dll\n" );
|
|
return FALSE;
|
|
}
|
|
|
|
/* Read in resource table */
|
|
resTab = HeapAlloc( GetProcessHeap(), 0, resTabSize );
|
|
if ( !resTab ) return FALSE;
|
|
|
|
LZSeek( lzfd, nehd.ne_rsrctab + nehdoffset, SEEK_SET );
|
|
if ( resTabSize != LZRead( lzfd, (char*)resTab, resTabSize ) )
|
|
{
|
|
HeapFree( GetProcessHeap(), 0, resTab );
|
|
return FALSE;
|
|
}
|
|
|
|
/* Find resource */
|
|
typeInfo = (NE_TYPEINFO *)(resTab + 2);
|
|
while (typeInfo->type_id)
|
|
{
|
|
if (typeInfo->type_id == typeid) goto found_type;
|
|
typeInfo = (NE_TYPEINFO *)((char *)(typeInfo + 1) +
|
|
typeInfo->count * sizeof(NE_NAMEINFO));
|
|
}
|
|
TRACE("No typeid entry found\n" );
|
|
HeapFree( GetProcessHeap(), 0, resTab );
|
|
return FALSE;
|
|
|
|
found_type:
|
|
nameInfo = (NE_NAMEINFO *)(typeInfo + 1);
|
|
|
|
for (count = typeInfo->count; count > 0; count--, nameInfo++)
|
|
if (nameInfo->id == resid) goto found_name;
|
|
|
|
TRACE("No resid entry found\n" );
|
|
HeapFree( GetProcessHeap(), 0, resTab );
|
|
return FALSE;
|
|
|
|
found_name:
|
|
/* Return resource data */
|
|
if ( resLen ) *resLen = nameInfo->length << *(WORD *)resTab;
|
|
if ( resOff ) *resOff = nameInfo->offset << *(WORD *)resTab;
|
|
|
|
HeapFree( GetProcessHeap(), 0, resTab );
|
|
return TRUE;
|
|
}
|
|
|
|
/***********************************************************************
|
|
* find_pe_resource [internal]
|
|
*/
|
|
static BOOL find_pe_resource( HFILE lzfd, DWORD *resLen, DWORD *resOff )
|
|
{
|
|
union
|
|
{
|
|
IMAGE_NT_HEADERS32 nt32;
|
|
IMAGE_NT_HEADERS64 nt64;
|
|
} pehd;
|
|
DWORD pehdoffset;
|
|
PIMAGE_DATA_DIRECTORY resDataDir;
|
|
PIMAGE_SECTION_HEADER sections;
|
|
LPBYTE resSection;
|
|
DWORD resSectionSize;
|
|
const void *resDir;
|
|
const IMAGE_RESOURCE_DIRECTORY *resPtr;
|
|
const IMAGE_RESOURCE_DATA_ENTRY *resData;
|
|
int i, len, nSections;
|
|
BOOL ret = FALSE;
|
|
|
|
/* Read in PE header */
|
|
pehdoffset = LZSeek( lzfd, 0, SEEK_CUR );
|
|
len = LZRead( lzfd, (LPSTR)&pehd, sizeof(pehd) );
|
|
if (len < sizeof(pehd.nt32.FileHeader)) return 0;
|
|
if (len < sizeof(pehd)) memset( (char *)&pehd + len, 0, sizeof(pehd) - len );
|
|
|
|
switch (pehd.nt32.OptionalHeader.Magic)
|
|
{
|
|
case IMAGE_NT_OPTIONAL_HDR32_MAGIC:
|
|
resDataDir = pehd.nt32.OptionalHeader.DataDirectory + IMAGE_DIRECTORY_ENTRY_RESOURCE;
|
|
break;
|
|
case IMAGE_NT_OPTIONAL_HDR64_MAGIC:
|
|
resDataDir = pehd.nt64.OptionalHeader.DataDirectory + IMAGE_DIRECTORY_ENTRY_RESOURCE;
|
|
break;
|
|
default:
|
|
return 0;
|
|
}
|
|
|
|
if ( !resDataDir->Size )
|
|
{
|
|
TRACE("No resources in PE dll\n" );
|
|
return FALSE;
|
|
}
|
|
|
|
/* Read in section table */
|
|
nSections = pehd.nt32.FileHeader.NumberOfSections;
|
|
sections = HeapAlloc( GetProcessHeap(), 0,
|
|
nSections * sizeof(IMAGE_SECTION_HEADER) );
|
|
if ( !sections ) return FALSE;
|
|
|
|
len = FIELD_OFFSET( IMAGE_NT_HEADERS32, OptionalHeader ) + pehd.nt32.FileHeader.SizeOfOptionalHeader;
|
|
LZSeek( lzfd, pehdoffset + len, SEEK_SET );
|
|
|
|
if ( nSections * sizeof(IMAGE_SECTION_HEADER) !=
|
|
LZRead( lzfd, (LPSTR)sections, nSections * sizeof(IMAGE_SECTION_HEADER) ) )
|
|
{
|
|
HeapFree( GetProcessHeap(), 0, sections );
|
|
return FALSE;
|
|
}
|
|
|
|
/* Find resource section */
|
|
for ( i = 0; i < nSections; i++ )
|
|
if ( resDataDir->VirtualAddress >= sections[i].VirtualAddress
|
|
&& resDataDir->VirtualAddress < sections[i].VirtualAddress +
|
|
sections[i].SizeOfRawData )
|
|
break;
|
|
|
|
if ( i == nSections )
|
|
{
|
|
HeapFree( GetProcessHeap(), 0, sections );
|
|
TRACE("Couldn't find resource section\n" );
|
|
return FALSE;
|
|
}
|
|
|
|
/* Read in resource section */
|
|
resSectionSize = sections[i].SizeOfRawData;
|
|
resSection = HeapAlloc( GetProcessHeap(), 0, resSectionSize );
|
|
if ( !resSection )
|
|
{
|
|
HeapFree( GetProcessHeap(), 0, sections );
|
|
return FALSE;
|
|
}
|
|
|
|
LZSeek( lzfd, sections[i].PointerToRawData, SEEK_SET );
|
|
if ( resSectionSize != LZRead( lzfd, (char*)resSection, resSectionSize ) ) goto done;
|
|
|
|
/* Find resource */
|
|
resDir = resSection + (resDataDir->VirtualAddress - sections[i].VirtualAddress);
|
|
|
|
resPtr = resDir;
|
|
resPtr = find_entry_by_id( resPtr, VS_FILE_INFO, resDir );
|
|
if ( !resPtr )
|
|
{
|
|
TRACE("No typeid entry found\n" );
|
|
goto done;
|
|
}
|
|
resPtr = find_entry_by_id( resPtr, VS_VERSION_INFO, resDir );
|
|
if ( !resPtr )
|
|
{
|
|
TRACE("No resid entry found\n" );
|
|
goto done;
|
|
}
|
|
resPtr = find_entry_default( resPtr, resDir );
|
|
if ( !resPtr )
|
|
{
|
|
TRACE("No default language entry found\n" );
|
|
goto done;
|
|
}
|
|
|
|
/* Find resource data section */
|
|
resData = (const IMAGE_RESOURCE_DATA_ENTRY*)resPtr;
|
|
for ( i = 0; i < nSections; i++ )
|
|
if ( resData->OffsetToData >= sections[i].VirtualAddress
|
|
&& resData->OffsetToData < sections[i].VirtualAddress +
|
|
sections[i].SizeOfRawData )
|
|
break;
|
|
|
|
if ( i == nSections )
|
|
{
|
|
TRACE("Couldn't find resource data section\n" );
|
|
goto done;
|
|
}
|
|
|
|
/* Return resource data */
|
|
if ( resLen ) *resLen = resData->Size;
|
|
if ( resOff ) *resOff = resData->OffsetToData - sections[i].VirtualAddress
|
|
+ sections[i].PointerToRawData;
|
|
ret = TRUE;
|
|
|
|
done:
|
|
HeapFree( GetProcessHeap(), 0, resSection );
|
|
HeapFree( GetProcessHeap(), 0, sections );
|
|
return ret;
|
|
}
|
|
|
|
|
|
/***********************************************************************
|
|
* find_version_resource [internal]
|
|
*/
|
|
DWORD find_version_resource( HFILE lzfd, DWORD *reslen, DWORD *offset )
|
|
{
|
|
DWORD magic = read_xx_header( lzfd );
|
|
|
|
switch (magic)
|
|
{
|
|
case IMAGE_OS2_SIGNATURE:
|
|
if (!find_ne_resource( lzfd, reslen, offset )) magic = 0;
|
|
break;
|
|
case IMAGE_NT_SIGNATURE:
|
|
if (!find_pe_resource( lzfd, reslen, offset )) magic = 0;
|
|
break;
|
|
}
|
|
return magic;
|
|
}
|