/* * PE file resources * * Copyright 1995 Thomas Sandford * Copyright 1996 Martin von Loewis * Copyright 2003 Alexandre Julliard * * Based on the Win16 resource handling code in loader/resource.c * Copyright 1993 Robert J. Amstadt * Copyright 1995 Alexandre Julliard * Copyright 1997 Marcus Meissner * * 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 "wine/port.h" #include <stdarg.h> #include <stdlib.h> #include <sys/types.h> #define NONAMELESSUNION #define NONAMELESSSTRUCT #include "ntstatus.h" #define WIN32_NO_STATUS #include "windef.h" #include "winbase.h" #include "winnt.h" #include "winternl.h" #include "wine/exception.h" #include "wine/unicode.h" #include "wine/debug.h" WINE_DEFAULT_DEBUG_CHANNEL(resource); static LCID user_lcid, system_lcid; static LANGID user_ui_language, system_ui_language; #define IS_INTRESOURCE(x) (((ULONG_PTR)(x) >> 16) == 0) /********************************************************************** * is_data_file_module * * Check if a module handle is for a LOAD_LIBRARY_AS_DATAFILE module. */ static inline BOOL is_data_file_module( HMODULE hmod ) { return (ULONG_PTR)hmod & 1; } /********************************************************************** * push_language * * push a language in the list of languages to try */ static inline int push_language( WORD *list, int pos, WORD lang ) { int i; for (i = 0; i < pos; i++) if (list[i] == lang) return pos; list[pos++] = lang; return pos; } /********************************************************************** * find_first_entry * * Find the first suitable entry in a resource directory */ static const IMAGE_RESOURCE_DIRECTORY *find_first_entry( const IMAGE_RESOURCE_DIRECTORY *dir, const void *root, int want_dir ) { const IMAGE_RESOURCE_DIRECTORY_ENTRY *entry = (const IMAGE_RESOURCE_DIRECTORY_ENTRY *)(dir + 1); int pos; for (pos = 0; pos < dir->NumberOfNamedEntries + dir->NumberOfIdEntries; pos++) { if (!entry[pos].u2.s2.DataIsDirectory == !want_dir) return (const IMAGE_RESOURCE_DIRECTORY *)((const char *)root + entry[pos].u2.s2.OffsetToDirectory); } return NULL; } /********************************************************************** * find_entry_by_id * * Find an entry by id in a resource directory */ static const IMAGE_RESOURCE_DIRECTORY *find_entry_by_id( const IMAGE_RESOURCE_DIRECTORY *dir, WORD id, const void *root, int want_dir ) { 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].u.Id == id) { if (!entry[pos].u2.s2.DataIsDirectory == !want_dir) { TRACE("root %p dir %p id %04x ret %p\n", root, dir, id, (const char*)root + entry[pos].u2.s2.OffsetToDirectory); return (const IMAGE_RESOURCE_DIRECTORY *)((const char *)root + entry[pos].u2.s2.OffsetToDirectory); } break; } if (entry[pos].u.Id > id) max = pos - 1; else min = pos + 1; } TRACE("root %p dir %p id %04x not found\n", root, dir, id ); return NULL; } /********************************************************************** * find_entry_by_name * * Find an entry by name in a resource directory */ static const IMAGE_RESOURCE_DIRECTORY *find_entry_by_name( const IMAGE_RESOURCE_DIRECTORY *dir, LPCWSTR name, const void *root, int want_dir ) { const IMAGE_RESOURCE_DIRECTORY_ENTRY *entry; const IMAGE_RESOURCE_DIR_STRING_U *str; int min, max, res, pos, namelen; if (IS_INTRESOURCE(name)) return find_entry_by_id( dir, LOWORD(name), root, want_dir ); entry = (const IMAGE_RESOURCE_DIRECTORY_ENTRY *)(dir + 1); namelen = strlenW(name); min = 0; max = dir->NumberOfNamedEntries - 1; while (min <= max) { pos = (min + max) / 2; str = (const IMAGE_RESOURCE_DIR_STRING_U *)((const char *)root + entry[pos].u.s.NameOffset); res = strncmpW( name, str->NameString, str->Length ); if (!res && namelen == str->Length) { if (!entry[pos].u2.s2.DataIsDirectory == !want_dir) { TRACE("root %p dir %p name %s ret %p\n", root, dir, debugstr_w(name), (const char*)root + entry[pos].u2.s2.OffsetToDirectory); return (const IMAGE_RESOURCE_DIRECTORY *)((const char *)root + entry[pos].u2.s2.OffsetToDirectory); } break; } if (res < 0) max = pos - 1; else min = pos + 1; } TRACE("root %p dir %p name %s not found\n", root, dir, debugstr_w(name) ); return NULL; } /********************************************************************** * find_entry * * Find a resource entry */ static NTSTATUS find_entry( HMODULE hmod, const LDR_RESOURCE_INFO *info, ULONG level, const void **ret, int want_dir ) { ULONG size; const void *root; const IMAGE_RESOURCE_DIRECTORY *resdirptr; WORD list[9]; /* list of languages to try */ int i, pos = 0; root = RtlImageDirectoryEntryToData( hmod, TRUE, IMAGE_DIRECTORY_ENTRY_RESOURCE, &size ); if (!root) return STATUS_RESOURCE_DATA_NOT_FOUND; if (size < sizeof(*resdirptr)) return STATUS_RESOURCE_DATA_NOT_FOUND; resdirptr = root; if (!level--) goto done; if (!(*ret = find_entry_by_name( resdirptr, (LPCWSTR)info->Type, root, want_dir || level ))) return STATUS_RESOURCE_TYPE_NOT_FOUND; if (!level--) return STATUS_SUCCESS; resdirptr = *ret; if (!(*ret = find_entry_by_name( resdirptr, (LPCWSTR)info->Name, root, want_dir || level ))) return STATUS_RESOURCE_NAME_NOT_FOUND; if (!level--) return STATUS_SUCCESS; if (level) return STATUS_INVALID_PARAMETER; /* level > 3 */ /* 1. specified language */ pos = push_language( list, pos, info->Language ); /* 2. specified language with neutral sublanguage */ pos = push_language( list, pos, MAKELANGID( PRIMARYLANGID(info->Language), SUBLANG_NEUTRAL ) ); /* 3. neutral language with neutral sublanguage */ pos = push_language( list, pos, MAKELANGID( LANG_NEUTRAL, SUBLANG_NEUTRAL ) ); /* if no explicitly specified language, try some defaults */ if (PRIMARYLANGID(info->Language) == LANG_NEUTRAL) { /* user defaults, unless SYS_DEFAULT sublanguage specified */ if (SUBLANGID(info->Language) != SUBLANG_SYS_DEFAULT) { /* 4. current thread locale language */ pos = push_language( list, pos, LANGIDFROMLCID(NtCurrentTeb()->CurrentLocale) ); /* 5. user locale language */ pos = push_language( list, pos, LANGIDFROMLCID(user_lcid) ); /* 6. user locale language with neutral sublanguage */ pos = push_language( list, pos, MAKELANGID( PRIMARYLANGID(user_lcid), SUBLANG_NEUTRAL ) ); } /* now system defaults */ /* 7. system locale language */ pos = push_language( list, pos, LANGIDFROMLCID( system_lcid ) ); /* 8. system locale language with neutral sublanguage */ pos = push_language( list, pos, MAKELANGID( PRIMARYLANGID(system_lcid), SUBLANG_NEUTRAL ) ); /* 9. English */ pos = push_language( list, pos, MAKELANGID( LANG_ENGLISH, SUBLANG_DEFAULT ) ); } resdirptr = *ret; for (i = 0; i < pos; i++) if ((*ret = find_entry_by_id( resdirptr, list[i], root, want_dir ))) return STATUS_SUCCESS; /* if no explicitly specified language, return the first entry */ if (PRIMARYLANGID(info->Language) == LANG_NEUTRAL) { if ((*ret = find_first_entry( resdirptr, root, want_dir ))) return STATUS_SUCCESS; } return STATUS_RESOURCE_LANG_NOT_FOUND; done: *ret = resdirptr; return STATUS_SUCCESS; } /********************************************************************** * LdrFindResourceDirectory_U (NTDLL.@) */ NTSTATUS WINAPI LdrFindResourceDirectory_U( HMODULE hmod, const LDR_RESOURCE_INFO *info, ULONG level, const IMAGE_RESOURCE_DIRECTORY **dir ) { const void *res; NTSTATUS status; __TRY { if (info) TRACE( "module %p type %s name %s lang %04x level %d\n", hmod, debugstr_w((LPCWSTR)info->Type), level > 1 ? debugstr_w((LPCWSTR)info->Name) : "", level > 2 ? info->Language : 0, level ); status = find_entry( hmod, info, level, &res, TRUE ); if (status == STATUS_SUCCESS) *dir = res; } __EXCEPT_PAGE_FAULT { return GetExceptionCode(); } __ENDTRY; return status; } /********************************************************************** * LdrFindResource_U (NTDLL.@) */ NTSTATUS WINAPI LdrFindResource_U( HMODULE hmod, const LDR_RESOURCE_INFO *info, ULONG level, const IMAGE_RESOURCE_DATA_ENTRY **entry ) { const void *res; NTSTATUS status; __TRY { if (info) TRACE( "module %p type %s name %s lang %04x level %d\n", hmod, debugstr_w((LPCWSTR)info->Type), level > 1 ? debugstr_w((LPCWSTR)info->Name) : "", level > 2 ? info->Language : 0, level ); status = find_entry( hmod, info, level, &res, FALSE ); if (status == STATUS_SUCCESS) *entry = res; } __EXCEPT_PAGE_FAULT { return GetExceptionCode(); } __ENDTRY; return status; } /* don't penalize other platforms stuff needed on i386 for compatibility */ #ifdef __i386__ NTSTATUS WINAPI access_resource( HMODULE hmod, const IMAGE_RESOURCE_DATA_ENTRY *entry, void **ptr, ULONG *size ) #else static inline NTSTATUS access_resource( HMODULE hmod, const IMAGE_RESOURCE_DATA_ENTRY *entry, void **ptr, ULONG *size ) #endif { NTSTATUS status; __TRY { ULONG dirsize; if (!RtlImageDirectoryEntryToData( hmod, TRUE, IMAGE_DIRECTORY_ENTRY_RESOURCE, &dirsize )) status = STATUS_RESOURCE_DATA_NOT_FOUND; else { if (ptr) { if (is_data_file_module(hmod)) { HMODULE mod = (HMODULE)((ULONG_PTR)hmod & ~1); *ptr = RtlImageRvaToVa( RtlImageNtHeader(mod), mod, entry->OffsetToData, NULL ); } else *ptr = (char *)hmod + entry->OffsetToData; } if (size) *size = entry->Size; status = STATUS_SUCCESS; } } __EXCEPT_PAGE_FAULT { return GetExceptionCode(); } __ENDTRY; return status; } /********************************************************************** * LdrAccessResource (NTDLL.@) * * NOTE * On x86, Shrinker, an executable compressor, depends on the * "call access_resource" instruction being there. */ #ifdef __i386__ __ASM_STDCALL_FUNC( LdrAccessResource, 16, "pushl %ebp\n\t" "movl %esp, %ebp\n\t" "subl $4,%esp\n\t" "pushl 24(%ebp)\n\t" "pushl 20(%ebp)\n\t" "pushl 16(%ebp)\n\t" "pushl 12(%ebp)\n\t" "pushl 8(%ebp)\n\t" "call " __ASM_NAME("access_resource") "\n\t" "leave\n\t" "ret $16" ) #else NTSTATUS WINAPI LdrAccessResource( HMODULE hmod, const IMAGE_RESOURCE_DATA_ENTRY *entry, void **ptr, ULONG *size ) { return access_resource( hmod, entry, ptr, size ); } #endif /********************************************************************** * RtlFindMessage (NTDLL.@) */ NTSTATUS WINAPI RtlFindMessage( HMODULE hmod, ULONG type, ULONG lang, ULONG msg_id, const MESSAGE_RESOURCE_ENTRY **ret ) { const MESSAGE_RESOURCE_DATA *data; const MESSAGE_RESOURCE_BLOCK *block; const IMAGE_RESOURCE_DATA_ENTRY *rsrc; LDR_RESOURCE_INFO info; NTSTATUS status; void *ptr; unsigned int i; info.Type = type; info.Name = 1; info.Language = lang; if ((status = LdrFindResource_U( hmod, &info, 3, &rsrc )) != STATUS_SUCCESS) return status; if ((status = LdrAccessResource( hmod, rsrc, &ptr, NULL )) != STATUS_SUCCESS) return status; data = ptr; block = data->Blocks; for (i = 0; i < data->NumberOfBlocks; i++, block++) { if (msg_id >= block->LowId && msg_id <= block->HighId) { const MESSAGE_RESOURCE_ENTRY *entry; entry = (const MESSAGE_RESOURCE_ENTRY *)((const char *)data + block->OffsetToEntries); for (i = msg_id - block->LowId; i > 0; i--) entry = (const MESSAGE_RESOURCE_ENTRY *)((const char *)entry + entry->Length); *ret = entry; return STATUS_SUCCESS; } } return STATUS_MESSAGE_NOT_FOUND; } /********************************************************************** * RtlFormatMessage (NTDLL.@) * * Formats a message (similar to sprintf). * * PARAMS * Message [I] Message to format. * MaxWidth [I] Maximum width in characters of each output line. * IgnoreInserts [I] Whether to copy the message without processing inserts. * Ansi [I] Whether Arguments may have ANSI strings. * ArgumentsIsArray [I] Whether Arguments is actually an array rather than a va_list *. * Buffer [O] Buffer to store processed message in. * BufferSize [I] Size of Buffer (in bytes?). * * RETURNS * NTSTATUS code. */ NTSTATUS WINAPI RtlFormatMessage( LPWSTR Message, UCHAR MaxWidth, BOOLEAN IgnoreInserts, BOOLEAN Ansi, BOOLEAN ArgumentIsArray, __ms_va_list * Arguments, LPWSTR Buffer, ULONG BufferSize ) { FIXME("(%s, %u, %s, %s, %s, %p, %p, %d)\n", debugstr_w(Message), MaxWidth, IgnoreInserts ? "TRUE" : "FALSE", Ansi ? "TRUE" : "FALSE", ArgumentIsArray ? "TRUE" : "FALSE", Arguments, Buffer, BufferSize); return STATUS_SUCCESS; } /********************************************************************** * NtQueryDefaultLocale (NTDLL.@) */ NTSTATUS WINAPI NtQueryDefaultLocale( BOOLEAN user, LCID *lcid ) { *lcid = user ? user_lcid : system_lcid; return STATUS_SUCCESS; } /********************************************************************** * NtSetDefaultLocale (NTDLL.@) */ NTSTATUS WINAPI NtSetDefaultLocale( BOOLEAN user, LCID lcid ) { if (user) user_lcid = lcid; else { system_lcid = lcid; system_ui_language = LANGIDFROMLCID(lcid); /* there is no separate call to set it */ } return STATUS_SUCCESS; } /********************************************************************** * NtQueryDefaultUILanguage (NTDLL.@) */ NTSTATUS WINAPI NtQueryDefaultUILanguage( LANGID *lang ) { *lang = user_ui_language; return STATUS_SUCCESS; } /********************************************************************** * NtSetDefaultUILanguage (NTDLL.@) */ NTSTATUS WINAPI NtSetDefaultUILanguage( LANGID lang ) { user_ui_language = lang; return STATUS_SUCCESS; } /********************************************************************** * NtQueryInstallUILanguage (NTDLL.@) */ NTSTATUS WINAPI NtQueryInstallUILanguage( LANGID *lang ) { *lang = system_ui_language; return STATUS_SUCCESS; }