kernelbase: Cache the sortid for each locale.

Signed-off-by: Alexandre Julliard <julliard@winehq.org>
This commit is contained in:
Alexandre Julliard 2022-05-16 18:23:30 +02:00
parent 093f6cbb8c
commit 610db32153
1 changed files with 65 additions and 42 deletions

View File

@ -324,11 +324,11 @@ static struct norm_table *norm_info;
struct sortguid struct sortguid
{ {
GUID id; /* sort GUID */ GUID id; /* sort GUID */
DWORD flags; /* flags */ UINT flags; /* flags */
DWORD compr; /* offset to compression table */ UINT compr; /* offset to compression table */
DWORD except; /* exception table offset in sortkey table */ UINT except; /* exception table offset in sortkey table */
DWORD ling_except; /* exception table offset for linguistic casing */ UINT ling_except; /* exception table offset for linguistic casing */
DWORD casemap; /* linguistic casemap table offset */ UINT casemap; /* linguistic casemap table offset */
}; };
#define FLAG_HAS_3_BYTE_WEIGHTS 0x01 #define FLAG_HAS_3_BYTE_WEIGHTS 0x01
@ -336,19 +336,19 @@ struct sortguid
#define FLAG_DOUBLECOMPRESSION 0x20 #define FLAG_DOUBLECOMPRESSION 0x20
#define FLAG_INVERSECASING 0x40 #define FLAG_INVERSECASING 0x40
static const struct sortguid *current_locale_sort;
static const GUID default_sort_guid = { 0x00000001, 0x57ee, 0x1e5c, { 0x00, 0xb4, 0xd0, 0x00, 0x0b, 0xb1, 0xe1, 0x1e }}; static const struct sortguid **locale_sorts;
static const struct sortguid *current_locale_sort;
static struct static struct
{ {
DWORD *keys; /* sortkey table, indexed by char */ UINT version; /* NLS version */
USHORT *casemap; /* casemap table, in l_intl.nls format */ UINT guid_count; /* number of sort GUIDs */
WORD *ctypes; /* CT_CTYPE1,2,3 values */ const UINT *keys; /* sortkey table, indexed by char */
BYTE *ctype_idx; /* index to map char to ctypes array entry */ const USHORT *casemap; /* casemap table, in l_intl.nls format */
DWORD version; /* NLS version */ const WORD *ctypes; /* CT_CTYPE1,2,3 values */
DWORD guid_count; /* number of sort GUIDs */ const BYTE *ctype_idx; /* index to map char to ctypes array entry */
struct sortguid *guids; /* table of sort GUIDs */ const struct sortguid *guids; /* table of sort GUIDs */
} sort; } sort;
static CRITICAL_SECTION locale_section; static CRITICAL_SECTION locale_section;
@ -403,22 +403,36 @@ static void load_locale_nls(void)
} }
static void init_sortkeys( DWORD *ptr ) static void load_sortdefault_nls(void)
{ {
WORD *ctype; const struct
DWORD *table; {
UINT sortkeys;
UINT casemaps;
UINT ctypes;
UINT sortids;
} *header;
sort.keys = (DWORD *)((char *)ptr + ptr[0]); const WORD *ctype;
sort.casemap = (USHORT *)((char *)ptr + ptr[1]); const UINT *table;
SIZE_T size;
ctype = (WORD *)((char *)ptr + ptr[2]); NtGetNlsSectionPtr( 9, 0, NULL, (void **)&header, &size );
sort.keys = (UINT *)((char *)header + header->sortkeys);
sort.casemap = (USHORT *)((char *)header + header->casemaps);
ctype = (WORD *)((char *)header + header->ctypes);
sort.ctypes = ctype + 2; sort.ctypes = ctype + 2;
sort.ctype_idx = (BYTE *)ctype + ctype[1] + 2; sort.ctype_idx = (BYTE *)ctype + ctype[1] + 2;
table = (DWORD *)((char *)ptr + ptr[3]); table = (UINT *)((char *)header + header->sortids);
sort.version = table[0]; sort.version = table[0];
sort.guid_count = table[1]; sort.guid_count = table[1];
sort.guids = (struct sortguid *)(table + 2); sort.guids = (struct sortguid *)(table + 2);
locale_sorts = RtlAllocateHeap( GetProcessHeap(), HEAP_ZERO_MEMORY,
locale_table->nb_lcnames * sizeof(*locale_sorts) );
} }
@ -550,28 +564,36 @@ static const NLS_LOCALE_DATA *get_locale_by_name( const WCHAR *name, LCID *lcid
} }
static const struct sortguid *get_language_sort( const WCHAR *locale ) static const struct sortguid *get_language_sort( const WCHAR *name )
{ {
WCHAR *p, *end, buffer[LOCALE_NAME_MAX_LENGTH], guidstr[39]; const NLS_LOCALE_LCNAME_INDEX *entry;
const NLS_LOCALE_DATA *locale;
WCHAR guidstr[39];
const struct sortguid *ret; const struct sortguid *ret;
UNICODE_STRING str; UNICODE_STRING str;
LCID lcid;
GUID guid; GUID guid;
HKEY key = 0; HKEY key = 0;
DWORD size, type; DWORD size, type;
if (locale == LOCALE_NAME_USER_DEFAULT) if (name == LOCALE_NAME_USER_DEFAULT)
{ {
if (current_locale_sort) return current_locale_sort; if (current_locale_sort) return current_locale_sort;
GetUserDefaultLocaleName( buffer, ARRAY_SIZE( buffer )); name = locale_strings + user_locale->sname + 1;
} }
else lstrcpynW( buffer, locale, LOCALE_NAME_MAX_LENGTH );
if (buffer[0] && !RegOpenKeyExW( nls_key, L"Sorting\\Ids", 0, KEY_READ, &key )) if (!(entry = find_lcname_entry( name ))) return NULL;
if ((ret = locale_sorts[entry - lcnames_index])) return ret;
lcid = entry->id;
name = locale_strings + entry->name + 1;
locale = get_locale_data( entry->idx );
if (!RegOpenKeyExW( nls_key, L"Sorting\\Ids", 0, KEY_READ, &key ))
{ {
for (;;) for (;;)
{ {
size = sizeof(guidstr); size = sizeof(guidstr);
if (!RegQueryValueExW( key, buffer, NULL, &type, (BYTE *)guidstr, &size ) && type == REG_SZ) if (!RegQueryValueExW( key, name, NULL, &type, (BYTE *)guidstr, &size ) && type == REG_SZ)
{ {
RtlInitUnicodeString( &str, guidstr ); RtlInitUnicodeString( &str, guidstr );
if (!RtlGUIDFromString( &str, &guid )) if (!RtlGUIDFromString( &str, &guid ))
@ -581,14 +603,15 @@ static const struct sortguid *get_language_sort( const WCHAR *locale )
} }
break; break;
} }
for (p = end = buffer; *p; p++) if (*p == '-' || *p == '_') end = p; if (!name[0]) break;
if (end == buffer) break; name = locale_strings + (SORTIDFROMLCID( lcid ) ? locale->sname : locale->sparent) + 1;
*end = 0; if (!(locale = get_locale_by_name( name, &lcid ))) break;
} }
} }
ret = find_sortguid( &default_sort_guid ); ret = &sort.guids[0];
done: done:
RegCloseKey( key ); RegCloseKey( key );
if (ret) locale_sorts[entry - lcnames_index] = ret;
return ret; return ret;
} }
@ -1790,7 +1813,6 @@ void init_locale( HMODULE module )
{ {
USHORT utf8[2] = { 0, CP_UTF8 }; USHORT utf8[2] = { 0, CP_UTF8 };
USHORT *ansi_ptr, *oem_ptr; USHORT *ansi_ptr, *oem_ptr;
void *sort_ptr;
WCHAR bufferW[LOCALE_NAME_MAX_LENGTH]; WCHAR bufferW[LOCALE_NAME_MAX_LENGTH];
DYNAMIC_TIME_ZONE_INFORMATION timezone; DYNAMIC_TIME_ZONE_INFORMATION timezone;
const WCHAR *user_locale_name; const WCHAR *user_locale_name;
@ -1800,6 +1822,7 @@ void init_locale( HMODULE module )
kernelbase_handle = module; kernelbase_handle = module;
load_locale_nls(); load_locale_nls();
load_sortdefault_nls();
if (system_lcid == LOCALE_CUSTOM_UNSPECIFIED) system_lcid = MAKELANGID( LANG_ENGLISH, SUBLANG_DEFAULT ); if (system_lcid == LOCALE_CUSTOM_UNSPECIFIED) system_lcid = MAKELANGID( LANG_ENGLISH, SUBLANG_DEFAULT );
system_locale = NlsValidateLocale( &system_lcid, 0 ); system_locale = NlsValidateLocale( &system_lcid, 0 );
@ -1817,13 +1840,11 @@ void init_locale( HMODULE module )
if (GetEnvironmentVariableW( L"WINEUNIXCP", bufferW, ARRAY_SIZE(bufferW) )) if (GetEnvironmentVariableW( L"WINEUNIXCP", bufferW, ARRAY_SIZE(bufferW) ))
unix_cp = wcstoul( bufferW, NULL, 10 ); unix_cp = wcstoul( bufferW, NULL, 10 );
NtGetNlsSectionPtr( 9, 0, NULL, &sort_ptr, &size );
NtGetNlsSectionPtr( 12, NormalizationC, NULL, (void **)&norm_info, &size ); NtGetNlsSectionPtr( 12, NormalizationC, NULL, (void **)&norm_info, &size );
init_sortkeys( sort_ptr );
ansi_ptr = NtCurrentTeb()->Peb->AnsiCodePageData ? NtCurrentTeb()->Peb->AnsiCodePageData : utf8; ansi_ptr = NtCurrentTeb()->Peb->AnsiCodePageData ? NtCurrentTeb()->Peb->AnsiCodePageData : utf8;
oem_ptr = NtCurrentTeb()->Peb->OemCodePageData ? NtCurrentTeb()->Peb->OemCodePageData : utf8; oem_ptr = NtCurrentTeb()->Peb->OemCodePageData ? NtCurrentTeb()->Peb->OemCodePageData : utf8;
RtlInitNlsTables( ansi_ptr, oem_ptr, sort.casemap, &nls_info ); RtlInitNlsTables( ansi_ptr, oem_ptr, (USHORT *)sort.casemap, &nls_info );
RegCreateKeyExW( HKEY_LOCAL_MACHINE, L"System\\CurrentControlSet\\Control\\Nls", RegCreateKeyExW( HKEY_LOCAL_MACHINE, L"System\\CurrentControlSet\\Control\\Nls",
0, NULL, REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, NULL, &nls_key, NULL ); 0, NULL, REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, NULL, &nls_key, NULL );
@ -5003,7 +5024,7 @@ BOOL WINAPI DECLSPEC_HOTPATCH GetNLSVersion( NLS_FUNCTION func, LCID lcid, NLSVE
BOOL WINAPI DECLSPEC_HOTPATCH GetNLSVersionEx( NLS_FUNCTION func, const WCHAR *locale, BOOL WINAPI DECLSPEC_HOTPATCH GetNLSVersionEx( NLS_FUNCTION func, const WCHAR *locale,
NLSVERSIONINFOEX *info ) NLSVERSIONINFOEX *info )
{ {
LCID lcid = 0; const struct sortguid *sortid;
if (func != COMPARE_STRING) if (func != COMPARE_STRING)
{ {
@ -5016,15 +5037,17 @@ BOOL WINAPI DECLSPEC_HOTPATCH GetNLSVersionEx( NLS_FUNCTION func, const WCHAR *l
SetLastError( ERROR_INSUFFICIENT_BUFFER ); SetLastError( ERROR_INSUFFICIENT_BUFFER );
return FALSE; return FALSE;
} }
if (!(sortid = get_language_sort( locale )))
if (!(lcid = LocaleNameToLCID( locale, 0 ))) return FALSE; {
SetLastError( ERROR_INVALID_PARAMETER );
return 0;
}
info->dwNLSVersion = info->dwDefinedVersion = sort.version; info->dwNLSVersion = info->dwDefinedVersion = sort.version;
if (info->dwNLSVersionInfoSize >= sizeof(*info)) if (info->dwNLSVersionInfoSize >= sizeof(*info))
{ {
const struct sortguid *sortid = get_language_sort( locale ); info->dwEffectiveId = LocaleNameToLCID( locale, 0 );
info->dwEffectiveId = lcid; info->guidCustomVersion = sortid->id;
info->guidCustomVersion = sortid ? sortid->id : default_sort_guid;
} }
return TRUE; return TRUE;
} }