| /* |
| * Implementation of VERSION.DLL - Resource Access routines |
| * |
| * Copyright 1996,1997 Marcus Meissner |
| * Copyright 1997 David Cuthbert |
| * Copyright 1999 Ulrich Weigand |
| */ |
| |
| #include <stdlib.h> |
| #include <string.h> |
| |
| #include "neexe.h" |
| #include "module.h" |
| #include "winver.h" |
| #include "lzexpand.h" |
| #include "debugtools.h" |
| |
| DEFAULT_DEBUG_CHANNEL(ver) |
| |
| |
| /*********************************************************************** |
| * 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, &mzh, sizeof(mzh) ) ) |
| return 0; |
| if ( mzh.e_magic != IMAGE_DOS_SIGNATURE ) |
| 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; |
| } |
| |
| /*********************************************************************** |
| * load_ne_resource [internal] |
| */ |
| static BOOL find_ne_resource( HFILE lzfd, LPCSTR typeid, LPCSTR resid, |
| DWORD *resLen, DWORD *resOff ) |
| { |
| IMAGE_OS2_HEADER nehd; |
| NE_TYPEINFO *typeInfo; |
| NE_NAMEINFO *nameInfo; |
| DWORD nehdoffset; |
| LPBYTE resTab; |
| DWORD resTabSize; |
| |
| /* Read in NE header */ |
| nehdoffset = LZSeek( lzfd, 0, SEEK_CUR ); |
| if ( sizeof(nehd) != LZRead( lzfd, &nehd, sizeof(nehd) ) ) return 0; |
| |
| resTabSize = nehd.rname_tab_offset - nehd.resource_tab_offset; |
| 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.resource_tab_offset + nehdoffset, SEEK_SET ); |
| if ( resTabSize != LZRead( lzfd, resTab, resTabSize ) ) |
| { |
| HeapFree( GetProcessHeap(), 0, resTab ); |
| return FALSE; |
| } |
| |
| /* Find resource */ |
| typeInfo = (NE_TYPEINFO *)(resTab + 2); |
| typeInfo = NE_FindTypeSection( resTab, typeInfo, typeid ); |
| if ( !typeInfo ) |
| { |
| TRACE("No typeid entry found for %p\n", typeid ); |
| HeapFree( GetProcessHeap(), 0, resTab ); |
| return FALSE; |
| } |
| nameInfo = NE_FindResourceFromType( resTab, typeInfo, resid ); |
| if ( !nameInfo ) |
| { |
| TRACE("No resid entry found for %p\n", typeid ); |
| HeapFree( GetProcessHeap(), 0, resTab ); |
| return FALSE; |
| } |
| |
| /* Return resource data */ |
| if ( resLen ) *resLen = nameInfo->length << *(WORD *)resTab; |
| if ( resOff ) *resOff = nameInfo->offset << *(WORD *)resTab; |
| |
| HeapFree( GetProcessHeap(), 0, resTab ); |
| return TRUE; |
| } |
| |
| /*********************************************************************** |
| * load_pe_resource [internal] |
| */ |
| static BOOL find_pe_resource( HFILE lzfd, LPCSTR typeid, LPCSTR resid, |
| DWORD *resLen, DWORD *resOff ) |
| { |
| IMAGE_NT_HEADERS pehd; |
| DWORD pehdoffset; |
| PIMAGE_DATA_DIRECTORY resDataDir; |
| PIMAGE_SECTION_HEADER sections; |
| LPBYTE resSection; |
| DWORD resSectionSize; |
| DWORD resDir; |
| PIMAGE_RESOURCE_DIRECTORY resPtr; |
| PIMAGE_RESOURCE_DATA_ENTRY resData; |
| int i, nSections; |
| |
| |
| /* Read in PE header */ |
| pehdoffset = LZSeek( lzfd, 0, SEEK_CUR ); |
| if ( sizeof(pehd) != LZRead( lzfd, &pehd, sizeof(pehd) ) ) return 0; |
| |
| resDataDir = pehd.OptionalHeader.DataDirectory+IMAGE_FILE_RESOURCE_DIRECTORY; |
| if ( !resDataDir->Size ) |
| { |
| TRACE("No resources in PE dll\n" ); |
| return FALSE; |
| } |
| |
| /* Read in section table */ |
| nSections = pehd.FileHeader.NumberOfSections; |
| sections = HeapAlloc( GetProcessHeap(), 0, |
| nSections * sizeof(IMAGE_SECTION_HEADER) ); |
| if ( !sections ) return FALSE; |
| |
| LZSeek( lzfd, pehdoffset + |
| sizeof(DWORD) + /* Signature */ |
| sizeof(IMAGE_FILE_HEADER) + |
| pehd.FileHeader.SizeOfOptionalHeader, SEEK_SET ); |
| |
| if ( nSections * sizeof(IMAGE_SECTION_HEADER) != |
| LZRead( lzfd, 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, resSection, resSectionSize ) ) |
| { |
| HeapFree( GetProcessHeap(), 0, resSection ); |
| HeapFree( GetProcessHeap(), 0, sections ); |
| return FALSE; |
| } |
| |
| /* Find resource */ |
| resDir = (DWORD)resSection + |
| (resDataDir->VirtualAddress - sections[i].VirtualAddress); |
| |
| resPtr = (PIMAGE_RESOURCE_DIRECTORY)resDir; |
| resPtr = GetResDirEntryA( resPtr, typeid, resDir, FALSE ); |
| if ( !resPtr ) |
| { |
| TRACE("No typeid entry found for %p\n", typeid ); |
| HeapFree( GetProcessHeap(), 0, resSection ); |
| HeapFree( GetProcessHeap(), 0, sections ); |
| return FALSE; |
| } |
| resPtr = GetResDirEntryA( resPtr, resid, resDir, FALSE ); |
| if ( !resPtr ) |
| { |
| TRACE("No resid entry found for %p\n", resid ); |
| HeapFree( GetProcessHeap(), 0, resSection ); |
| HeapFree( GetProcessHeap(), 0, sections ); |
| return FALSE; |
| } |
| resPtr = GetResDirEntryA( resPtr, 0, resDir, TRUE ); |
| if ( !resPtr ) |
| { |
| TRACE("No default language entry found for %p\n", resid ); |
| HeapFree( GetProcessHeap(), 0, resSection ); |
| HeapFree( GetProcessHeap(), 0, sections ); |
| return FALSE; |
| } |
| |
| /* Find resource data section */ |
| resData = (PIMAGE_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" ); |
| HeapFree( GetProcessHeap(), 0, resSection ); |
| HeapFree( GetProcessHeap(), 0, sections ); |
| return FALSE; |
| } |
| |
| /* Return resource data */ |
| if ( resLen ) *resLen = resData->Size; |
| if ( resOff ) *resOff = resData->OffsetToData - sections[i].VirtualAddress |
| + sections[i].PointerToRawData; |
| |
| HeapFree( GetProcessHeap(), 0, resSection ); |
| HeapFree( GetProcessHeap(), 0, sections ); |
| return TRUE; |
| } |
| |
| /*********************************************************************** |
| * GetFileResourceSize32 [internal] |
| */ |
| DWORD WINAPI GetFileResourceSize( LPCSTR lpszFileName, |
| LPCSTR lpszResType, LPCSTR lpszResId, |
| LPDWORD lpdwFileOffset ) |
| { |
| BOOL retv = FALSE; |
| HFILE lzfd; |
| OFSTRUCT ofs; |
| DWORD reslen; |
| |
| TRACE("(%s,type=0x%lx,id=0x%lx,off=%p)\n", |
| debugstr_a(lpszFileName), (LONG)lpszResType, (LONG)lpszResId, |
| lpszResId ); |
| |
| lzfd = LZOpenFileA( lpszFileName, &ofs, OF_READ ); |
| if ( !lzfd ) return 0; |
| |
| switch ( read_xx_header( lzfd ) ) |
| { |
| case IMAGE_OS2_SIGNATURE: |
| retv = find_ne_resource( lzfd, lpszResType, lpszResId, |
| &reslen, lpdwFileOffset ); |
| break; |
| |
| case IMAGE_NT_SIGNATURE: |
| retv = find_pe_resource( lzfd, lpszResType, lpszResId, |
| &reslen, lpdwFileOffset ); |
| break; |
| } |
| |
| LZClose( lzfd ); |
| return retv? reslen : 0; |
| } |
| |
| /*********************************************************************** |
| * GetFileResource32 [internal] |
| */ |
| DWORD WINAPI GetFileResource( LPCSTR lpszFileName, |
| LPCSTR lpszResType, LPCSTR lpszResId, |
| DWORD dwFileOffset, |
| DWORD dwResLen, LPVOID lpvData ) |
| { |
| BOOL retv = FALSE; |
| HFILE lzfd; |
| OFSTRUCT ofs; |
| DWORD reslen = dwResLen; |
| |
| TRACE("(%s,type=0x%lx,id=0x%lx,off=%ld,len=%ld,data=%p)\n", |
| debugstr_a(lpszFileName), (LONG)lpszResType, (LONG)lpszResId, |
| dwFileOffset, dwResLen, lpvData ); |
| |
| lzfd = LZOpenFileA( lpszFileName, &ofs, OF_READ ); |
| if ( lzfd == 0 ) return 0; |
| |
| if ( !dwFileOffset ) |
| { |
| switch ( read_xx_header( lzfd ) ) |
| { |
| case IMAGE_OS2_SIGNATURE: |
| retv = find_ne_resource( lzfd, lpszResType, lpszResId, |
| &reslen, &dwFileOffset ); |
| break; |
| |
| case IMAGE_NT_SIGNATURE: |
| retv = find_pe_resource( lzfd, lpszResType, lpszResId, |
| &reslen, &dwFileOffset ); |
| break; |
| } |
| |
| if ( !retv ) |
| { |
| LZClose( lzfd ); |
| return 0; |
| } |
| } |
| |
| LZSeek( lzfd, dwFileOffset, SEEK_SET ); |
| reslen = LZRead( lzfd, lpvData, min( reslen, dwResLen ) ); |
| LZClose( lzfd ); |
| |
| return reslen; |
| } |
| |
