urlmon: Support version option in ObtainUserAgentString.

Signed-off-by: Jacek Caban <jacek@codeweavers.com>
Signed-off-by: Alexandre Julliard <julliard@winehq.org>
This commit is contained in:
Jacek Caban 2021-03-25 19:34:22 +01:00 committed by Alexandre Julliard
parent 8efbff575a
commit 3d95eef66f
1 changed files with 69 additions and 35 deletions

View File

@ -505,61 +505,95 @@ static BOOL get_url_encoding(HKEY root, DWORD *encoding)
static LPWSTR user_agent;
static size_t obtain_user_agent(WCHAR *ret, size_t size)
static size_t obtain_user_agent(unsigned int version, WCHAR *ret, size_t size)
{
OSVERSIONINFOW info = {sizeof(info)};
const WCHAR *os_type, *is_nt;
DWORD res, idx=0;
BOOL is_wow, quirks = FALSE;
DWORD res;
size_t len = 0;
BOOL is_wow;
HKEY key;
EnterCriticalSection(&session_cs);
if(user_agent) {
len = wcslen(user_agent) + 1;
memcpy(ret, user_agent, min(size, len) * sizeof(WCHAR));
if(version & 0x1000) {
version &= ~0x1000;
if(version == 7)
quirks = TRUE;
else
version = 7;
}else if(version < 7) {
version = 7;
}
if(version > 11) {
FIXME("Unsupported version %u\n", version);
version = 11;
}
if(version < 7 || (version == 7 && !quirks)) {
EnterCriticalSection(&session_cs);
if(user_agent) {
len = wcslen(user_agent) + 1;
memcpy(ret, user_agent, min(size, len) * sizeof(WCHAR));
}
LeaveCriticalSection(&session_cs);
if(len) return len;
}
swprintf(ret, size, L"Mozilla/%s (", version < 9 ? L"4.0" : L"5.0");
len = lstrlenW(ret);
if(version < 11) {
swprintf(ret + len, size - len, L"compatible; MSIE %u.0; ", version);
len += wcslen(ret + len);
}
LeaveCriticalSection(&session_cs);
if(len) return len;
GetVersionExW(&info);
is_nt = info.dwPlatformId == VER_PLATFORM_WIN32_NT ? L"NT " : L"";
if(sizeof(void*) == 8)
#ifdef __x86_64__
os_type = L"Win64; x64; ";
#else
os_type = L"Win64; ";
#endif
else if(IsWow64Process(GetCurrentProcess(), &is_wow) && is_wow)
os_type = L"WOW64; ";
else
os_type = L"";
swprintf(ret, size, L"Mozilla/4.0 (compatible; MSIE 7.0; Windows %s%d.%d; %sTrident/7.0",
is_nt, info.dwMajorVersion, info.dwMinorVersion, os_type);
swprintf(ret + len, size - len, L"Windows %s%d.%d; %s", is_nt, info.dwMajorVersion,
info.dwMinorVersion, os_type);
len = lstrlenW(ret);
res = RegOpenKeyW(HKEY_LOCAL_MACHINE, L"SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\"
"Internet Settings\\5.0\\User Agent\\Post Platform", &key);
if(res == ERROR_SUCCESS) {
DWORD value_len;
while(1) {
if(idx) {
ret[len++] = ';';
ret[len++] = ' ';
}
value_len = size - len - 2;
res = RegEnumValueW(key, idx, ret + len, &value_len, NULL, NULL, NULL, NULL);
if(res != ERROR_SUCCESS)
break;
idx++;
len += value_len;
}
RegCloseKey(key);
if(!quirks) {
wcscpy(ret + len, L"Trident/7.0");
len += ARRAY_SIZE(L"Trident/7.0") - 1;
}
ret[len++] = ')';
ret[len++] = 0;
if(version < 9) {
res = RegOpenKeyW(HKEY_LOCAL_MACHINE, L"SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\"
"Internet Settings\\5.0\\User Agent\\Post Platform", &key);
if(res == ERROR_SUCCESS) {
DWORD value_len, idx;
for(idx = 0;; idx++) {
if(idx) {
ret[len++] = ';';
ret[len++] = ' ';
}
value_len = size - len - 2;
res = RegEnumValueW(key, idx, ret + len, &value_len, NULL, NULL, NULL, NULL);
if(res != ERROR_SUCCESS)
break;
len += value_len;
}
RegCloseKey(key);
if(idx) len -= 2;
}
}
wcscpy(ret + len, version >= 11 ? L"; rv:11.0) like Gecko" : L")");
len += wcslen(ret + len) + 1;
TRACE("Using user agent %s\n", debugstr_w(ret));
return len;
}
@ -570,7 +604,7 @@ static void ensure_user_agent(void)
if(!user_agent) {
WCHAR buf[1024];
obtain_user_agent(buf, ARRAY_SIZE(buf));
obtain_user_agent(0, buf, ARRAY_SIZE(buf));
user_agent = heap_strdupW(buf);
}
@ -704,7 +738,7 @@ HRESULT WINAPI ObtainUserAgentString(DWORD option, char *ret, DWORD *ret_size)
if(!ret || !ret_size)
return E_INVALIDARG;
len = obtain_user_agent(buf, ARRAY_SIZE(buf));
len = obtain_user_agent(option, buf, ARRAY_SIZE(buf));
size = WideCharToMultiByte(CP_ACP, 0, buf, len, NULL, 0, NULL, NULL);
if(size <= *ret_size)
WideCharToMultiByte(CP_ACP, 0, buf, len, ret, *ret_size+1, NULL, NULL);