dlls/version/resource.c

   1 /*
   2  * Implementation of VERSION.DLL - Resource Access routines
   3  *
   4  * Copyright 1996,1997 Marcus Meissner
   5  * Copyright 1997 David Cuthbert
   6  * Copyright 1999 Ulrich Weigand
   7  *
   8  * This library is free software; you can redistribute it and/or
   9  * modify it under the terms of the GNU Lesser General Public
  10  * License as published by the Free Software Foundation; either
  11  * version 2.1 of the License, or (at your option) any later version.
  12  *
  13  * This library is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  16  * Lesser General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU Lesser General Public
  19  * License along with this library; if not, write to the Free Software
  20  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  21  */
  22
  23 #include "config.h"
  24
  25 #include <stdarg.h>
  26 #include <stdio.h>
  27 #include <stdlib.h>
  28 #include <string.h>
  29 #include <sys/types.h>
  30 #ifdef HAVE_UNISTD_H
  31 # include <unistd.h>
  32 #endif
  33
  34 #define NONAMELESSUNION
  35 #define NONAMELESSSTRUCT
  36 #include "windef.h"
  37 #include "winbase.h"
  38 #include "lzexpand.h"
  39 #include "winver.h"
  40
  41 #include "wine/unicode.h"
  42 #include "wine/winbase16.h"
  43
  44 #include "wine/debug.h"
  45
  46 WINE_DEFAULT_DEBUG_CHANNEL(ver);
  47
  48
  49 /**********************************************************************
  50  *  find_entry_by_id
  51  *
  52  * Find an entry by id in a resource directory
  53  * Copied from loader/pe_resource.c
  54  */
  55 static const IMAGE_RESOURCE_DIRECTORY *find_entry_by_id( const IMAGE_RESOURCE_DIRECTORY *dir,
  56                                                          WORD id, const void *root )
  57 {
  58     const IMAGE_RESOURCE_DIRECTORY_ENTRY *entry;
  59     int min, max, pos;
  60
  61     entry = (const IMAGE_RESOURCE_DIRECTORY_ENTRY *)(dir + 1);
  62     min = dir->NumberOfNamedEntries;
  63     max = min + dir->NumberOfIdEntries - 1;
  64     while (min <= max)
  65     {
  66         pos = (min + max) / 2;
  67         if (entry[pos].u1.s2.Id == id)
  68             return (const IMAGE_RESOURCE_DIRECTORY *)((const char *)root + entry[pos].u2.s3.OffsetToDirectory);
  69         if (entry[pos].u1.s2.Id > id) max = pos - 1;
  70         else min = pos + 1;
  71     }
  72     return NULL;
  73 }
  74
  75
  76 /**********************************************************************
  77  *  find_entry_default
  78  *
  79  * Find a default entry in a resource directory
  80  * Copied from loader/pe_resource.c
  81  */
  82 static const IMAGE_RESOURCE_DIRECTORY *find_entry_default( const IMAGE_RESOURCE_DIRECTORY *dir,
  83                                                            const void *root )
  84 {
  85     const IMAGE_RESOURCE_DIRECTORY_ENTRY *entry;
  86
  87     entry = (const IMAGE_RESOURCE_DIRECTORY_ENTRY *)(dir + 1);
  88     return (const IMAGE_RESOURCE_DIRECTORY *)((const char *)root + entry->u2.s3.OffsetToDirectory);
  89 }
  90
  91
  92 /***********************************************************************
  93  *           read_xx_header         [internal]
  94  */
  95 static int read_xx_header( HFILE lzfd )
  96 {
  97     IMAGE_DOS_HEADER mzh;
  98     char magic[3];
  99
 100     LZSeek( lzfd, 0, SEEK_SET );
 101     if ( sizeof(mzh) != LZRead( lzfd, (LPSTR)&mzh, sizeof(mzh) ) )
 102         return 0;
 103     if ( mzh.e_magic != IMAGE_DOS_SIGNATURE )
 104     {
 105         if (!memcmp( &mzh, "\177ELF", 4 )) return 1;  /* ELF */
 106         if (*(UINT *)&mzh == 0xfeedface || *(UINT *)&mzh == 0xcefaedfe) return 1;  /* Mach-O */
 107         return 0;
 108     }
 109
 110     LZSeek( lzfd, mzh.e_lfanew, SEEK_SET );
 111     if ( 2 != LZRead( lzfd, magic, 2 ) )
 112         return 0;
 113
 114     LZSeek( lzfd, mzh.e_lfanew, SEEK_SET );
 115
 116     if ( magic[0] == 'N' && magic[1] == 'E' )
 117         return IMAGE_OS2_SIGNATURE;
 118     if ( magic[0] == 'P' && magic[1] == 'E' )
 119         return IMAGE_NT_SIGNATURE;
 120
 121     magic[2] = '\0';
 122     WARN("Can't handle %s files.\n", magic );
 123     return 0;
 124 }
 125
 126 /***********************************************************************
 127  *           find_ne_resource         [internal]
 128  */
 129 static BOOL find_ne_resource( HFILE lzfd, DWORD *resLen, DWORD *resOff )
 130 {
 131     const WORD typeid = VS_FILE_INFO | 0x8000;
 132     const WORD resid = VS_VERSION_INFO | 0x8000;
 133     IMAGE_OS2_HEADER nehd;
 134     NE_TYPEINFO *typeInfo;
 135     NE_NAMEINFO *nameInfo;
 136     DWORD nehdoffset;
 137     LPBYTE resTab;
 138     DWORD resTabSize;
 139     int count;
 140
 141     /* Read in NE header */
 142     nehdoffset = LZSeek( lzfd, 0, SEEK_CUR );
 143     if ( sizeof(nehd) != LZRead( lzfd, (LPSTR)&nehd, sizeof(nehd) ) ) return 0;
 144
 145     resTabSize = nehd.ne_restab - nehd.ne_rsrctab;
 146     if ( !resTabSize )
 147     {
 148         TRACE("No resources in NE dll\n" );
 149         return FALSE;
 150     }
 151
 152     /* Read in resource table */
 153     resTab = HeapAlloc( GetProcessHeap(), 0, resTabSize );
 154     if ( !resTab ) return FALSE;
 155
 156     LZSeek( lzfd, nehd.ne_rsrctab + nehdoffset, SEEK_SET );
 157     if ( resTabSize != LZRead( lzfd, (char*)resTab, resTabSize ) )
 158     {
 159         HeapFree( GetProcessHeap(), 0, resTab );
 160         return FALSE;
 161     }
 162
 163     /* Find resource */
 164     typeInfo = (NE_TYPEINFO *)(resTab + 2);
 165     while (typeInfo->type_id)
 166     {
 167         if (typeInfo->type_id == typeid) goto found_type;
 168         typeInfo = (NE_TYPEINFO *)((char *)(typeInfo + 1) +
 169                                    typeInfo->count * sizeof(NE_NAMEINFO));
 170     }
 171     TRACE("No typeid entry found\n" );
 172     HeapFree( GetProcessHeap(), 0, resTab );
 173     return FALSE;
 174
 175  found_type:
 176     nameInfo = (NE_NAMEINFO *)(typeInfo + 1);
 177
 178     for (count = typeInfo->count; count > 0; count--, nameInfo++)
 179         if (nameInfo->id == resid) goto found_name;
 180
 181     TRACE("No resid entry found\n" );
 182     HeapFree( GetProcessHeap(), 0, resTab );
 183     return FALSE;
 184
 185  found_name:
 186     /* Return resource data */
 187     if ( resLen ) *resLen = nameInfo->length << *(WORD *)resTab;
 188     if ( resOff ) *resOff = nameInfo->offset << *(WORD *)resTab;
 189
 190     HeapFree( GetProcessHeap(), 0, resTab );
 191     return TRUE;
 192 }
 193
 194 /***********************************************************************
 195  *           find_pe_resource         [internal]
 196  */
 197 static BOOL find_pe_resource( HFILE lzfd, DWORD *resLen, DWORD *resOff )
 198 {
 199     union
 200     {
 201         IMAGE_NT_HEADERS32 nt32;
 202         IMAGE_NT_HEADERS64 nt64;
 203     } pehd;
 204     DWORD pehdoffset;
 205     PIMAGE_DATA_DIRECTORY resDataDir;
 206     PIMAGE_SECTION_HEADER sections;
 207     LPBYTE resSection;
 208     DWORD resSectionSize;
 209     const void *resDir;
 210     const IMAGE_RESOURCE_DIRECTORY *resPtr;
 211     const IMAGE_RESOURCE_DATA_ENTRY *resData;
 212     int i, len, nSections;
 213     BOOL ret = FALSE;
 214
 215     /* Read in PE header */
 216     pehdoffset = LZSeek( lzfd, 0, SEEK_CUR );
 217     len = LZRead( lzfd, (LPSTR)&pehd, sizeof(pehd) );
 218     if (len < sizeof(pehd.nt32.FileHeader)) return 0;
 219     if (len < sizeof(pehd)) memset( (char *)&pehd + len, 0, sizeof(pehd) - len );
 220
 221     switch (pehd.nt32.OptionalHeader.Magic)
 222     {
 223     case IMAGE_NT_OPTIONAL_HDR32_MAGIC:
 224         resDataDir = pehd.nt32.OptionalHeader.DataDirectory + IMAGE_DIRECTORY_ENTRY_RESOURCE;
 225         break;
 226     case IMAGE_NT_OPTIONAL_HDR64_MAGIC:
 227         resDataDir = pehd.nt64.OptionalHeader.DataDirectory + IMAGE_DIRECTORY_ENTRY_RESOURCE;
 228         break;
 229     default:
 230         return 0;
 231     }
 232
 233     if ( !resDataDir->Size )
 234     {
 235         TRACE("No resources in PE dll\n" );
 236         return FALSE;
 237     }
 238
 239     /* Read in section table */
 240     nSections = pehd.nt32.FileHeader.NumberOfSections;
 241     sections = HeapAlloc( GetProcessHeap(), 0,
 242                           nSections * sizeof(IMAGE_SECTION_HEADER) );
 243     if ( !sections ) return FALSE;
 244
 245     len = FIELD_OFFSET( IMAGE_NT_HEADERS32, OptionalHeader ) + pehd.nt32.FileHeader.SizeOfOptionalHeader;
 246     LZSeek( lzfd, pehdoffset + len, SEEK_SET );
 247
 248     if ( nSections * sizeof(IMAGE_SECTION_HEADER) !=
 249          LZRead( lzfd, (LPSTR)sections, nSections * sizeof(IMAGE_SECTION_HEADER) ) )
 250     {
 251         HeapFree( GetProcessHeap(), 0, sections );
 252         return FALSE;
 253     }
 254
 255     /* Find resource section */
 256     for ( i = 0; i < nSections; i++ )
 257         if (    resDataDir->VirtualAddress >= sections[i].VirtualAddress
 258              && resDataDir->VirtualAddress <  sections[i].VirtualAddress +
 259                                               sections[i].SizeOfRawData )
 260             break;
 261
 262     if ( i == nSections )
 263     {
 264         HeapFree( GetProcessHeap(), 0, sections );
 265         TRACE("Couldn't find resource section\n" );
 266         return FALSE;
 267     }
 268
 269     /* Read in resource section */
 270     resSectionSize = sections[i].SizeOfRawData;
 271     resSection = HeapAlloc( GetProcessHeap(), 0, resSectionSize );
 272     if ( !resSection )
 273     {
 274         HeapFree( GetProcessHeap(), 0, sections );
 275         return FALSE;
 276     }
 277
 278     LZSeek( lzfd, sections[i].PointerToRawData, SEEK_SET );
 279     if ( resSectionSize != LZRead( lzfd, (char*)resSection, resSectionSize ) ) goto done;
 280
 281     /* Find resource */
 282     resDir = resSection + (resDataDir->VirtualAddress - sections[i].VirtualAddress);
 283
 284     resPtr = resDir;
 285     resPtr = find_entry_by_id( resPtr, VS_FILE_INFO, resDir );
 286     if ( !resPtr )
 287     {
 288         TRACE("No typeid entry found\n" );
 289         goto done;
 290     }
 291     resPtr = find_entry_by_id( resPtr, VS_VERSION_INFO, resDir );
 292     if ( !resPtr )
 293     {
 294         TRACE("No resid entry found\n" );
 295         goto done;
 296     }
 297     resPtr = find_entry_default( resPtr, resDir );
 298     if ( !resPtr )
 299     {
 300         TRACE("No default language entry found\n" );
 301         goto done;
 302     }
 303
 304     /* Find resource data section */
 305     resData = (const IMAGE_RESOURCE_DATA_ENTRY*)resPtr;
 306     for ( i = 0; i < nSections; i++ )
 307         if (    resData->OffsetToData >= sections[i].VirtualAddress
 308              && resData->OffsetToData <  sections[i].VirtualAddress +
 309                                          sections[i].SizeOfRawData )
 310             break;
 311
 312     if ( i == nSections )
 313     {
 314         TRACE("Couldn't find resource data section\n" );
 315         goto done;
 316     }
 317
 318     /* Return resource data */
 319     if ( resLen ) *resLen = resData->Size;
 320     if ( resOff ) *resOff = resData->OffsetToData - sections[i].VirtualAddress
 321                             + sections[i].PointerToRawData;
 322     ret = TRUE;
 323
 324  done:
 325     HeapFree( GetProcessHeap(), 0, resSection );
 326     HeapFree( GetProcessHeap(), 0, sections );
 327     return ret;
 328 }
 329
 330
 331 /***********************************************************************
 332  *           find_version_resource         [internal]
 333  */
 334 DWORD find_version_resource( HFILE lzfd, DWORD *reslen, DWORD *offset )
 335 {
 336     DWORD magic = read_xx_header( lzfd );
 337
 338     switch (magic)
 339     {
 340     case IMAGE_OS2_SIGNATURE:
 341         if (!find_ne_resource( lzfd, reslen, offset )) magic = 0;
 342         break;
 343     case IMAGE_NT_SIGNATURE:
 344         if (!find_pe_resource( lzfd, reslen, offset )) magic = 0;
 345         break;
 346     }
 347     return magic;
 348 }