| /* |
| * PE (Portable Execute) File Resources |
| * |
| * Copyright 1995 Thomas Sandford |
| * Copyright 1996 Martin von Loewis |
| * |
| * Based on the Win16 resource handling code in loader/resource.c |
| * Copyright 1993 Robert J. Amstadt |
| * Copyright 1995 Alexandre Julliard |
| * Copyright 1997 Marcus Meissner |
| */ |
| |
| #include <stdlib.h> |
| #include <sys/types.h> |
| |
| #include "wine/winestring.h" |
| #include "wine/unicode.h" |
| #include "windef.h" |
| #include "winnls.h" |
| #include "winerror.h" |
| #include "module.h" |
| #include "heap.h" |
| #include "task.h" |
| #include "process.h" |
| #include "stackframe.h" |
| #include "neexe.h" |
| #include "debugtools.h" |
| |
| DEFAULT_DEBUG_CHANNEL(resource); |
| |
| /********************************************************************** |
| * get_resdir |
| * |
| * Get the resource directory of a PE module |
| */ |
| static const IMAGE_RESOURCE_DIRECTORY* get_resdir( HMODULE hmod ) |
| { |
| const IMAGE_DATA_DIRECTORY *dir; |
| const IMAGE_RESOURCE_DIRECTORY *ret = NULL; |
| |
| if (!hmod) hmod = GetModuleHandleA( NULL ); |
| else if (!HIWORD(hmod)) |
| { |
| FIXME("Enumeration of 16-bit resources is not supported\n"); |
| SetLastError(ERROR_INVALID_HANDLE); |
| return NULL; |
| } |
| dir = &PE_HEADER(hmod)->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_RESOURCE]; |
| if (dir->Size && dir->VirtualAddress) |
| ret = (IMAGE_RESOURCE_DIRECTORY *)((char *)hmod + dir->VirtualAddress); |
| return ret; |
| } |
| |
| |
| /********************************************************************** |
| * find_entry_by_name |
| * |
| * Find an entry by name in a resource directory |
| */ |
| static IMAGE_RESOURCE_DIRECTORY *find_entry_by_name( const IMAGE_RESOURCE_DIRECTORY *dir, |
| LPCWSTR name, const char *root ) |
| { |
| |
| const IMAGE_RESOURCE_DIRECTORY_ENTRY *entry; |
| const IMAGE_RESOURCE_DIR_STRING_U *str; |
| int min, max, res, pos, namelen = strlenW(name); |
| |
| entry = (const IMAGE_RESOURCE_DIRECTORY_ENTRY *)(dir + 1); |
| min = 0; |
| max = dir->NumberOfNamedEntries - 1; |
| while (min <= max) |
| { |
| pos = (min + max) / 2; |
| str = (IMAGE_RESOURCE_DIR_STRING_U *)(root + entry[pos].u1.s.NameOffset); |
| res = strncmpiW( name, str->NameString, str->Length ); |
| if (!res && namelen == str->Length) |
| return (IMAGE_RESOURCE_DIRECTORY *)(root + entry[pos].u2.s.OffsetToDirectory); |
| if (res < 0) max = pos - 1; |
| else min = pos + 1; |
| } |
| |
| /* now do a linear search just in case */ |
| for (pos = 0; pos < dir->NumberOfNamedEntries; pos++) |
| { |
| str = (IMAGE_RESOURCE_DIR_STRING_U *)(root + entry[pos].u1.s.NameOffset); |
| if (namelen != str->Length) continue; |
| if (!strncmpiW( name, str->NameString, str->Length )) |
| { |
| ERR( "entry '%s' required linear search, please report\n", debugstr_w(name) ); |
| return (IMAGE_RESOURCE_DIRECTORY *)(root + entry[pos].u2.s.OffsetToDirectory); |
| } |
| } |
| return NULL; |
| } |
| |
| |
| /********************************************************************** |
| * find_entry_by_id |
| * |
| * Find an entry by id in a resource directory |
| */ |
| static IMAGE_RESOURCE_DIRECTORY *find_entry_by_id( const IMAGE_RESOURCE_DIRECTORY *dir, |
| WORD id, const char *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 (IMAGE_RESOURCE_DIRECTORY *)(root + entry[pos].u2.s.OffsetToDirectory); |
| if (entry[pos].u1.Id > id) max = pos - 1; |
| else min = pos + 1; |
| } |
| |
| /* now do a linear search just in case */ |
| min = dir->NumberOfNamedEntries; |
| max = min + dir->NumberOfIdEntries - 1; |
| for (pos = min; pos <= max; pos++) |
| { |
| if (entry[pos].u1.Id == id) |
| { |
| ERR( "entry %04x required linear search, please report\n", id ); |
| return (IMAGE_RESOURCE_DIRECTORY *)(root + entry[pos].u2.s.OffsetToDirectory); |
| } |
| } |
| return NULL; |
| } |
| |
| |
| /********************************************************************** |
| * find_entry_default |
| * |
| * Find a default entry in a resource directory |
| */ |
| static IMAGE_RESOURCE_DIRECTORY *find_entry_default( const IMAGE_RESOURCE_DIRECTORY *dir, |
| const char *root ) |
| { |
| const IMAGE_RESOURCE_DIRECTORY_ENTRY *entry; |
| |
| entry = (const IMAGE_RESOURCE_DIRECTORY_ENTRY *)(dir + 1); |
| if (dir->NumberOfNamedEntries || dir->NumberOfIdEntries) |
| return (IMAGE_RESOURCE_DIRECTORY *)(root + entry->u2.s.OffsetToDirectory); |
| return NULL; |
| } |
| |
| |
| /********************************************************************** |
| * GetResDirEntryW |
| * |
| * Helper function - goes down one level of PE resource tree |
| * |
| */ |
| const IMAGE_RESOURCE_DIRECTORY *GetResDirEntryW(const IMAGE_RESOURCE_DIRECTORY *resdirptr, |
| LPCWSTR name, LPCVOID root, |
| BOOL allowdefault) |
| { |
| if (HIWORD(name)) |
| { |
| if (name[0]=='#') |
| { |
| char buf[10]; |
| if (!WideCharToMultiByte( CP_ACP, 0, name+1, -1, buf, sizeof(buf), NULL, NULL )) |
| return NULL; |
| return find_entry_by_id( resdirptr, atoi(buf), root ); |
| } |
| return find_entry_by_name( resdirptr, name, root ); |
| } |
| else |
| { |
| const IMAGE_RESOURCE_DIRECTORY *ret; |
| ret = find_entry_by_id( resdirptr, LOWORD(name), root ); |
| if (!ret && !name && allowdefault) ret = find_entry_default( resdirptr, root ); |
| return ret; |
| } |
| } |
| |
| /********************************************************************** |
| * GetResDirEntryA |
| */ |
| const IMAGE_RESOURCE_DIRECTORY *GetResDirEntryA( const IMAGE_RESOURCE_DIRECTORY *resdirptr, |
| LPCSTR name, LPCVOID root, |
| BOOL allowdefault ) |
| { |
| const IMAGE_RESOURCE_DIRECTORY *retv; |
| LPWSTR nameW = HIWORD(name)? HEAP_strdupAtoW( GetProcessHeap(), 0, name ) |
| : (LPWSTR)name; |
| |
| retv = GetResDirEntryW( resdirptr, nameW, root, allowdefault ); |
| |
| if ( HIWORD(name) ) HeapFree( GetProcessHeap(), 0, nameW ); |
| |
| return retv; |
| } |
| |
| /********************************************************************** |
| * PE_FindResourceExW |
| * |
| * FindResourceExA/W does search in the following order: |
| * 1. Exact specified language |
| * 2. Language with neutral sublanguage |
| * 3. Neutral language with neutral sublanguage |
| * 4. Neutral language with default sublanguage |
| */ |
| HRSRC PE_FindResourceExW( HMODULE hmod, LPCWSTR name, LPCWSTR type, WORD lang ) |
| { |
| const IMAGE_RESOURCE_DIRECTORY *resdirptr = get_resdir(hmod); |
| const void *root; |
| HRSRC result; |
| |
| if (!resdirptr) return 0; |
| |
| root = resdirptr; |
| if ((resdirptr = GetResDirEntryW(resdirptr, type, root, FALSE)) == NULL) |
| return 0; |
| if ((resdirptr = GetResDirEntryW(resdirptr, name, root, FALSE)) == NULL) |
| return 0; |
| |
| /* 1. Exact specified language */ |
| if ((result = (HRSRC)find_entry_by_id( resdirptr, lang, root ))) goto found; |
| |
| /* 2. Language with neutral sublanguage */ |
| lang = MAKELANGID(PRIMARYLANGID(lang), SUBLANG_NEUTRAL); |
| if ((result = (HRSRC)find_entry_by_id( resdirptr, lang, root ))) goto found; |
| |
| /* 3. Neutral language with neutral sublanguage */ |
| lang = MAKELANGID(LANG_NEUTRAL, SUBLANG_NEUTRAL); |
| if ((result = (HRSRC)find_entry_by_id( resdirptr, lang, root ))) goto found; |
| |
| /* 4. Neutral language with default sublanguage */ |
| lang = MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT); |
| result = (HRSRC)find_entry_by_id( resdirptr, lang, root ); |
| |
| found: |
| return result; |
| } |
| |
| /********************************************************************** |
| * PE_FindResourceW |
| * |
| * Load[String]/[Icon]/[Menu]/[etc.] does use FindResourceA/W. |
| * FindResourceA/W does search in the following order: |
| * 1. Neutral language with neutral sublanguage |
| * 2. Neutral language with default sublanguage |
| * 3. Current locale lang id |
| * 4. Current locale lang id with neutral sublanguage |
| * 5. (!) LANG_ENGLISH, SUBLANG_DEFAULT |
| * 6. Return first in the list |
| */ |
| HRSRC PE_FindResourceW( HMODULE hmod, LPCWSTR name, LPCWSTR type ) |
| { |
| const IMAGE_RESOURCE_DIRECTORY *resdirptr = get_resdir(hmod); |
| const void *root; |
| HRSRC result; |
| WORD lang; |
| |
| if (!resdirptr) return 0; |
| |
| root = resdirptr; |
| if ((resdirptr = GetResDirEntryW(resdirptr, type, root, FALSE)) == NULL) |
| return 0; |
| if ((resdirptr = GetResDirEntryW(resdirptr, name, root, FALSE)) == NULL) |
| return 0; |
| |
| /* 1. Neutral language with neutral sublanguage */ |
| lang = MAKELANGID(LANG_NEUTRAL, SUBLANG_NEUTRAL); |
| if ((result = (HRSRC)find_entry_by_id( resdirptr, lang, root ))) goto found; |
| |
| /* 2. Neutral language with default sublanguage */ |
| lang = MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT); |
| if ((result = (HRSRC)find_entry_by_id( resdirptr, lang, root ))) goto found; |
| |
| /* 3. Current locale lang id */ |
| lang = LANGIDFROMLCID(GetUserDefaultLCID()); |
| if ((result = (HRSRC)find_entry_by_id( resdirptr, lang, root ))) goto found; |
| |
| /* 4. Current locale lang id with neutral sublanguage */ |
| lang = MAKELANGID(PRIMARYLANGID(lang), SUBLANG_NEUTRAL); |
| if ((result = (HRSRC)find_entry_by_id( resdirptr, lang, root ))) goto found; |
| |
| /* 5. (!) LANG_ENGLISH, SUBLANG_DEFAULT */ |
| lang = MAKELANGID(LANG_ENGLISH, SUBLANG_DEFAULT); |
| if ((result = (HRSRC)find_entry_by_id( resdirptr, lang, root ))) goto found; |
| |
| /* 6. Return first in the list */ |
| result = (HRSRC)find_entry_default( resdirptr, root ); |
| |
| found: |
| return result; |
| } |
| |
| |
| /********************************************************************** |
| * PE_LoadResource |
| */ |
| HANDLE PE_LoadResource( HMODULE hmod, HANDLE hRsrc ) |
| { |
| if (!hRsrc) return 0; |
| return (HANDLE)(hmod + ((PIMAGE_RESOURCE_DATA_ENTRY)hRsrc)->OffsetToData); |
| } |
| |
| |
| /********************************************************************** |
| * PE_SizeofResource |
| */ |
| DWORD PE_SizeofResource( HANDLE hRsrc ) |
| { |
| if (!hRsrc) return 0; |
| return ((PIMAGE_RESOURCE_DATA_ENTRY)hRsrc)->Size; |
| } |
| |
| |
| /********************************************************************** |
| * EnumResourceTypesA (KERNEL32.90) |
| */ |
| BOOL WINAPI EnumResourceTypesA( HMODULE hmod, ENUMRESTYPEPROCA lpfun, LONG lparam) |
| { |
| int i; |
| const IMAGE_RESOURCE_DIRECTORY *resdir = get_resdir(hmod); |
| const IMAGE_RESOURCE_DIRECTORY_ENTRY *et; |
| BOOL ret; |
| |
| if (!resdir) return FALSE; |
| |
| et =(PIMAGE_RESOURCE_DIRECTORY_ENTRY)(resdir + 1); |
| ret = FALSE; |
| for (i=0;i<resdir->NumberOfNamedEntries+resdir->NumberOfIdEntries;i++) { |
| LPSTR type; |
| |
| if (et[i].u1.s.NameIsString) |
| { |
| PIMAGE_RESOURCE_DIR_STRING_U pResString = (PIMAGE_RESOURCE_DIR_STRING_U) ((LPBYTE) resdir + et[i].u1.s.NameOffset); |
| DWORD len = WideCharToMultiByte( CP_ACP, 0, pResString->NameString, pResString->Length, |
| NULL, 0, NULL, NULL); |
| if (!(type = HeapAlloc(GetProcessHeap(), 0, len + 1))) |
| return FALSE; |
| WideCharToMultiByte( CP_ACP, 0, pResString->NameString, pResString->Length, |
| type, len, NULL, NULL); |
| type[len] = '\0'; |
| ret = lpfun(hmod,type,lparam); |
| HeapFree(GetProcessHeap(), 0, type); |
| } |
| else |
| { |
| type = (LPSTR)(int)et[i].u1.Id; |
| ret = lpfun(hmod,type,lparam); |
| } |
| if (!ret) |
| break; |
| } |
| return ret; |
| } |
| |
| |
| /********************************************************************** |
| * EnumResourceTypesW (KERNEL32.91) |
| */ |
| BOOL WINAPI EnumResourceTypesW( HMODULE hmod, ENUMRESTYPEPROCW lpfun, LONG lparam) |
| { |
| int i; |
| const IMAGE_RESOURCE_DIRECTORY *resdir = get_resdir(hmod); |
| const IMAGE_RESOURCE_DIRECTORY_ENTRY *et; |
| BOOL ret; |
| |
| if (!resdir) return FALSE; |
| |
| et =(PIMAGE_RESOURCE_DIRECTORY_ENTRY)(resdir + 1); |
| ret = FALSE; |
| for (i=0;i<resdir->NumberOfNamedEntries+resdir->NumberOfIdEntries;i++) { |
| LPWSTR type; |
| |
| if (et[i].u1.s.NameIsString) |
| { |
| PIMAGE_RESOURCE_DIR_STRING_U pResString = (PIMAGE_RESOURCE_DIR_STRING_U) ((LPBYTE) resdir + et[i].u1.s.NameOffset); |
| if (!(type = HeapAlloc(GetProcessHeap(), 0, (pResString->Length+1) * sizeof (WCHAR)))) |
| return FALSE; |
| memcpy(type, pResString->NameString, pResString->Length * sizeof (WCHAR)); |
| type[pResString->Length] = '\0'; |
| ret = lpfun(hmod,type,lparam); |
| HeapFree(GetProcessHeap(), 0, type); |
| } |
| else |
| { |
| type = (LPWSTR)(int)et[i].u1.Id; |
| ret = lpfun(hmod,type,lparam); |
| } |
| if (!ret) |
| break; |
| } |
| return ret; |
| } |
| |
| |
| /********************************************************************** |
| * EnumResourceNamesA (KERNEL32.88) |
| */ |
| BOOL WINAPI EnumResourceNamesA( HMODULE hmod, LPCSTR type, ENUMRESNAMEPROCA lpfun, LONG lparam ) |
| { |
| int i; |
| const IMAGE_RESOURCE_DIRECTORY *basedir = get_resdir(hmod); |
| const IMAGE_RESOURCE_DIRECTORY *resdir; |
| const IMAGE_RESOURCE_DIRECTORY_ENTRY *et; |
| BOOL ret; |
| |
| if (!basedir) return FALSE; |
| |
| if (!(resdir = GetResDirEntryA(basedir,type,basedir,FALSE))) return FALSE; |
| |
| et =(PIMAGE_RESOURCE_DIRECTORY_ENTRY)(resdir + 1); |
| ret = FALSE; |
| for (i=0;i<resdir->NumberOfNamedEntries+resdir->NumberOfIdEntries;i++) { |
| LPSTR name; |
| |
| if (et[i].u1.s.NameIsString) |
| { |
| PIMAGE_RESOURCE_DIR_STRING_U pResString = (PIMAGE_RESOURCE_DIR_STRING_U) ((LPBYTE) basedir + et[i].u1.s.NameOffset); |
| DWORD len = WideCharToMultiByte(CP_ACP, 0, pResString->NameString, pResString->Length, |
| NULL, 0, NULL, NULL); |
| if (!(name = HeapAlloc(GetProcessHeap(), 0, len + 1 ))) |
| return FALSE; |
| WideCharToMultiByte( CP_ACP, 0, pResString->NameString, pResString->Length, |
| name, len, NULL, NULL ); |
| name[len] = '\0'; |
| ret = lpfun(hmod,type,name,lparam); |
| HeapFree( GetProcessHeap(), 0, name ); |
| } |
| else |
| { |
| name = (LPSTR)(int)et[i].u1.Id; |
| ret = lpfun(hmod,type,name,lparam); |
| } |
| if (!ret) |
| break; |
| } |
| return ret; |
| } |
| |
| |
| /********************************************************************** |
| * EnumResourceNamesW (KERNEL32.89) |
| */ |
| BOOL WINAPI EnumResourceNamesW( HMODULE hmod, LPCWSTR type, ENUMRESNAMEPROCW lpfun, LONG lparam ) |
| { |
| int i; |
| const IMAGE_RESOURCE_DIRECTORY *basedir = get_resdir(hmod); |
| const IMAGE_RESOURCE_DIRECTORY *resdir; |
| const IMAGE_RESOURCE_DIRECTORY_ENTRY *et; |
| BOOL ret; |
| |
| if (!basedir) return FALSE; |
| |
| resdir = GetResDirEntryW(basedir,type,basedir,FALSE); |
| if (!resdir) |
| return FALSE; |
| et = (PIMAGE_RESOURCE_DIRECTORY_ENTRY)(resdir + 1); |
| ret = FALSE; |
| for (i=0;i<resdir->NumberOfNamedEntries+resdir->NumberOfIdEntries;i++) { |
| LPWSTR name; |
| |
| if (et[i].u1.s.NameIsString) |
| { |
| PIMAGE_RESOURCE_DIR_STRING_U pResString = (PIMAGE_RESOURCE_DIR_STRING_U) ((LPBYTE) basedir + et[i].u1.s.NameOffset); |
| if (!(name = HeapAlloc(GetProcessHeap(), 0, (pResString->Length + 1) * sizeof (WCHAR)))) |
| return FALSE; |
| memcpy(name, pResString->NameString, pResString->Length * sizeof (WCHAR)); |
| name[pResString->Length] = '\0'; |
| ret = lpfun(hmod,type,name,lparam); |
| HeapFree(GetProcessHeap(), 0, name); |
| } |
| else |
| { |
| name = (LPWSTR)(int)et[i].u1.Id; |
| ret = lpfun(hmod,type,name,lparam); |
| } |
| if (!ret) |
| break; |
| } |
| return ret; |
| } |
| |
| |
| /********************************************************************** |
| * EnumResourceLanguagesA (KERNEL32.86) |
| */ |
| BOOL WINAPI EnumResourceLanguagesA( HMODULE hmod, LPCSTR type, LPCSTR name, |
| ENUMRESLANGPROCA lpfun, LONG lparam ) |
| { |
| int i; |
| const IMAGE_RESOURCE_DIRECTORY *basedir = get_resdir(hmod); |
| const IMAGE_RESOURCE_DIRECTORY *resdir; |
| const IMAGE_RESOURCE_DIRECTORY_ENTRY *et; |
| BOOL ret; |
| |
| if (!basedir) return FALSE; |
| if (!(resdir = GetResDirEntryA(basedir,type,basedir,FALSE))) return FALSE; |
| if (!(resdir = GetResDirEntryA(resdir,name,basedir,FALSE))) return FALSE; |
| |
| et =(PIMAGE_RESOURCE_DIRECTORY_ENTRY)(resdir + 1); |
| ret = FALSE; |
| for (i=0;i<resdir->NumberOfNamedEntries+resdir->NumberOfIdEntries;i++) { |
| /* languages are just ids... I hope */ |
| ret = lpfun(hmod,type,name,et[i].u1.Id,lparam); |
| if (!ret) |
| break; |
| } |
| return ret; |
| } |
| |
| |
| /********************************************************************** |
| * EnumResourceLanguagesW (KERNEL32.87) |
| */ |
| BOOL WINAPI EnumResourceLanguagesW( HMODULE hmod, LPCWSTR type, LPCWSTR name, |
| ENUMRESLANGPROCW lpfun, LONG lparam ) |
| { |
| int i; |
| const IMAGE_RESOURCE_DIRECTORY *basedir = get_resdir(hmod); |
| const IMAGE_RESOURCE_DIRECTORY *resdir; |
| const IMAGE_RESOURCE_DIRECTORY_ENTRY *et; |
| BOOL ret; |
| |
| if (!basedir) return FALSE; |
| |
| resdir = GetResDirEntryW(basedir,type,basedir,FALSE); |
| if (!resdir) |
| return FALSE; |
| resdir = GetResDirEntryW(resdir,name,basedir,FALSE); |
| if (!resdir) |
| return FALSE; |
| et =(PIMAGE_RESOURCE_DIRECTORY_ENTRY)(resdir + 1); |
| ret = FALSE; |
| for (i=0;i<resdir->NumberOfNamedEntries+resdir->NumberOfIdEntries;i++) { |
| ret = lpfun(hmod,type,name,et[i].u1.Id,lparam); |
| if (!ret) |
| break; |
| } |
| return ret; |
| } |
