ReceiveTempSMS
/
Sweden-Number
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

urlmon/tests: Add tests for IInternetZoneManager_SetZoneAttributes.

This commit is contained in:
Detlef Riekenberg 2011-05-04 19:42:21 +02:00 committed by Alexandre Julliard
parent 16e27e4d6c
commit 7aaf9f0324
1 changed files with 148 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

149
dlls/urlmon/tests/sec_mgr.c
View File

@ -1,6 +1,6 @@
/*
* Copyright 2005-2006 Jacek Caban for CodeWeavers
* Copyright 2009-2010 Detlef Riekenberg
* Copyright 2009-2011 Detlef Riekenberg
* Copyright 2011 Thomas Mullaly for CodeWeavers
*
* This library is free software; you can redistribute it and/or
@ -105,6 +105,7 @@ static const WCHAR security_expectedW[] = {'w','i','n','e','t','e','s','t',':','
static const WCHAR winetest_to_httpW[] = {'w','i','n','e','t','e','s','t',':','h',0};
static const char *szZoneMapDomainsKey = "Software\\Microsoft\\Windows\\CurrentVersion\\Internet Settings\\ZoneMap\\Domains";
static const char *szInternetSettingsKey = "Software\\Microsoft\\Windows\\CurrentVersion\\Internet Settings";
static const BYTE secid1[] = {'f','i','l','e',':',0,0,0,0};
static const BYTE secid5[] = {'h','t','t','p',':','w','w','w','.','z','o','n','e','3',
@ -176,6 +177,74 @@ static inline DWORD strcmp_aw(LPCSTR strA, LPCWSTR strB) {
return ret;
}
/* Based on RegDeleteTreeW from dlls/advapi32/registry.c */
static LONG myRegDeleteTreeA(HKEY hKey, LPCSTR lpszSubKey)
{
LONG ret;
DWORD dwMaxSubkeyLen, dwMaxValueLen;
DWORD dwMaxLen, dwSize;
CHAR szNameBuf[MAX_PATH], *lpszName = szNameBuf;
HKEY hSubKey = hKey;
if(lpszSubKey)
{
ret = RegOpenKeyExA(hKey, lpszSubKey, 0, KEY_READ, &hSubKey);
if (ret) return ret;
}
/* Get highest length for keys, values */
ret = RegQueryInfoKeyA(hSubKey, NULL, NULL, NULL, NULL,
&dwMaxSubkeyLen, NULL, NULL, &dwMaxValueLen, NULL, NULL, NULL);
if (ret) goto cleanup;
dwMaxSubkeyLen++;
dwMaxValueLen++;
dwMaxLen = max(dwMaxSubkeyLen, dwMaxValueLen);
if (dwMaxLen > sizeof(szNameBuf)/sizeof(CHAR))
{
/* Name too big: alloc a buffer for it */
if (!(lpszName = HeapAlloc( GetProcessHeap(), 0, dwMaxLen*sizeof(CHAR))))
{
ret = ERROR_NOT_ENOUGH_MEMORY;
goto cleanup;
}
}
/* Recursively delete all the subkeys */
while (TRUE)
{
dwSize = dwMaxLen;
if (RegEnumKeyExA(hSubKey, 0, lpszName, &dwSize, NULL,
NULL, NULL, NULL)) break;
ret = myRegDeleteTreeA(hSubKey, lpszName);
if (ret) goto cleanup;
}
if (lpszSubKey)
ret = RegDeleteKeyA(hKey, lpszSubKey);
else
while (TRUE)
{
dwSize = dwMaxLen;
if (RegEnumValueA(hKey, 0, lpszName, &dwSize,
NULL, NULL, NULL, NULL)) break;
ret = RegDeleteValueA(hKey, lpszName);
if (ret) goto cleanup;
}
cleanup:
/* Free buffer if allocated */
if (lpszName != szNameBuf)
heap_free(lpszName);
if(lpszSubKey)
RegCloseKey(hSubKey);
return ret;
}
static void test_SecurityManager(void)
{
int i;
@ -1011,6 +1080,83 @@ static void test_GetZoneAttributes(void)
ok(hr == S_OK, "got 0x%x (expected S_OK)\n", hr);
}
static void test_SetZoneAttributes(void)
{
IInternetZoneManager *zonemgr = NULL;
CHAR buffer [sizeof(ZONEATTRIBUTES) + 16];
ZONEATTRIBUTES* pZA = (ZONEATTRIBUTES*) buffer;
CHAR regpath[MAX_PATH];
HKEY hkey;
HRESULT hr;
DWORD res;
trace("testing SetZoneAttributes...\n");
hr = pCoInternetCreateZoneManager(NULL, &zonemgr, 0);
ok(hr == S_OK, "CoInternetCreateZoneManager result: 0x%x\n", hr);
if (FAILED(hr))
return;
memset(buffer, -1, sizeof(buffer));
hr = IInternetZoneManager_GetZoneAttributes(zonemgr, URLZONE_LOCAL_MACHINE, pZA);
ok(hr == S_OK, "got 0x%x (expected S_OK)\n", hr);
sprintf(regpath, "%s\\Zones\\%d", szInternetSettingsKey, URLZONE_CUSTOM);
res = RegCreateKeyA(HKEY_CURRENT_USER, regpath, &hkey);
RegCloseKey(hkey);
ok(res == ERROR_SUCCESS, "got %d (expected ERROR_SUCCESS)\n", res);
if (res != ERROR_SUCCESS)
goto cleanup;
pZA->cbSize = sizeof(ZONEATTRIBUTES);
hr = IInternetZoneManager_SetZoneAttributes(zonemgr, URLZONE_CUSTOM, NULL);
ok(hr == E_INVALIDARG, "got 0x%x (expected E_INVALIDARG)\n", hr);
/* normal use */
hr = IInternetZoneManager_SetZoneAttributes(zonemgr, URLZONE_CUSTOM, pZA);
if (hr == E_FAIL) {
win_skip("SetZoneAttributes not supported: IE too old\n");
goto cleanup;
}
ok(hr == S_OK, "got 0x%x (expected S_OK)\n", hr);
/* native urlmon ignores cbSize */
pZA->cbSize = sizeof(ZONEATTRIBUTES) + sizeof(DWORD);
hr = IInternetZoneManager_SetZoneAttributes(zonemgr, URLZONE_CUSTOM, pZA);
ok(hr == S_OK, "got 0x%x for sizeof(ZONEATTRIBUTES) + sizeof(DWORD) (expected S_OK)\n", hr);
pZA->cbSize = sizeof(ZONEATTRIBUTES) - sizeof(DWORD);
hr = IInternetZoneManager_SetZoneAttributes(zonemgr, URLZONE_CUSTOM, pZA);
ok(hr == S_OK, "got 0x%x for sizeof(ZONEATTRIBUTES) - sizeof(DWORD) (expected S_OK)\n", hr);
pZA->cbSize = 0;
hr = IInternetZoneManager_SetZoneAttributes(zonemgr, URLZONE_CUSTOM, pZA);
ok(hr == S_OK, "got 0x%x for size 0 (expected S_OK)\n", hr);
/* The key for the zone must be present, when calling SetZoneAttributes */
myRegDeleteTreeA(HKEY_CURRENT_USER, regpath);
/* E_FAIL is returned from IE6 here, which is resonable.
All newer IE return S_OK without saving the zone attributes to the registry.
This is a Windows bug, but we have to accept that as standard */
hr = IInternetZoneManager_SetZoneAttributes(zonemgr, URLZONE_CUSTOM, pZA);
ok((hr == S_OK) || broken(hr == E_FAIL), "got 0x%x (expected S_OK)\n", hr);
/* SetZoneAttributes did not create the directory */
res = RegOpenKeyA(HKEY_CURRENT_USER, regpath, &hkey);
ok((res == ERROR_FILE_NOT_FOUND) && (hkey == NULL),
"got %u with %p (expected ERROR_FILE_NOT_FOUND with NULL)\n", res, hkey);
if (hkey) RegCloseKey(hkey);
cleanup:
/* delete zone settings in the registry */
myRegDeleteTreeA(HKEY_CURRENT_USER, regpath);
hr = IInternetZoneManager_Release(zonemgr);
ok(hr == S_OK, "got 0x%x (expected S_OK)\n", hr);
}
static void test_InternetSecurityMarshalling(void)
{
IInternetSecurityManager *secmgr = NULL;
@ -1509,6 +1655,7 @@ START_TEST(sec_mgr)
test_GetZoneActionPolicy();
test_GetZoneAt();
test_GetZoneAttributes();
test_SetZoneAttributes();
test_InternetSecurityMarshalling();
unregister_protocols();