| /* Unit test suite for SHLWAPI ordinal functions | |
| * | |
| * Copyright 2004 Jon Griffiths | |
| * | |
| * 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 <stdio.h> | |
| #define COBJMACROS | |
| #include "wine/test.h" | |
| #include "winbase.h" | |
| #include "winerror.h" | |
| #include "winuser.h" | |
| #include "ole2.h" | |
| #include "oaidl.h" | |
| #include "ocidl.h" | |
| #include "mlang.h" | |
| /* Function ptrs for ordinal calls */ | |
| static HMODULE hShlwapi; | |
| static int (WINAPI *pSHSearchMapInt)(const int*,const int*,int,int); | |
| static HRESULT (WINAPI *pGetAcceptLanguagesA)(LPSTR,LPDWORD); | |
| static HANDLE (WINAPI *pSHAllocShared)(LPCVOID,DWORD,DWORD); | |
| static LPVOID (WINAPI *pSHLockShared)(HANDLE,DWORD); | |
| static BOOL (WINAPI *pSHUnlockShared)(LPVOID); | |
| static BOOL (WINAPI *pSHFreeShared)(HANDLE,DWORD); | |
| static HRESULT(WINAPIV *pSHPackDispParams)(DISPPARAMS*,VARIANTARG*,UINT,...); | |
| static HRESULT(WINAPI *pIConnectionPoint_SimpleInvoke)(IConnectionPoint*,DISPID,DISPPARAMS*); | |
| static HRESULT(WINAPI *pIConnectionPoint_InvokeWithCancel)(IConnectionPoint*,DISPID,DISPPARAMS*,DWORD,DWORD); | |
| static HRESULT(WINAPI *pConnectToConnectionPoint)(IUnknown*,REFIID,BOOL,IUnknown*, LPDWORD,IConnectionPoint **); | |
| static HRESULT(WINAPI *pSHPropertyBag_ReadLONG)(IPropertyBag *,LPCWSTR,LPLONG); | |
| static HMODULE hmlang; | |
| static HRESULT (WINAPI *pLcidToRfc1766A)(LCID, LPSTR, INT); | |
| static const CHAR ie_international[] = { | |
| 'S','o','f','t','w','a','r','e','\\', | |
| 'M','i','c','r','o','s','o','f','t','\\', | |
| 'I','n','t','e','r','n','e','t',' ','E','x','p','l','o','r','e','r','\\', | |
| 'I','n','t','e','r','n','a','t','i','o','n','a','l',0}; | |
| static const CHAR acceptlanguage[] = { | |
| 'A','c','c','e','p','t','L','a','n','g','u','a','g','e',0}; | |
| static void test_GetAcceptLanguagesA(void) | |
| { | |
| static LPCSTR table[] = {"de,en-gb;q=0.7,en;q=0.3", | |
| "de,en;q=0.3,en-gb;q=0.7", /* sorting is ignored */ | |
| "winetest", /* content is ignored */ | |
| "de-de,de;q=0.5", | |
| "de", | |
| NULL}; | |
| DWORD exactsize; | |
| char original[512]; | |
| char language[32]; | |
| char buffer[64]; | |
| HKEY hroot = NULL; | |
| LONG res_query = ERROR_SUCCESS; | |
| LONG lres; | |
| HRESULT hr; | |
| DWORD maxlen = sizeof(buffer) - 2; | |
| DWORD len; | |
| LCID lcid; | |
| LPCSTR entry; | |
| INT i = 0; | |
| if (!pGetAcceptLanguagesA) { | |
| win_skip("GetAcceptLanguagesA is not available\n"); | |
| return; | |
| } | |
| lcid = GetUserDefaultLCID(); | |
| /* Get the original Value */ | |
| lres = RegOpenKeyA(HKEY_CURRENT_USER, ie_international, &hroot); | |
| if (lres) { | |
| skip("RegOpenKey(%s) failed: %d\n", ie_international, lres); | |
| return; | |
| } | |
| len = sizeof(original); | |
| original[0] = 0; | |
| res_query = RegQueryValueExA(hroot, acceptlanguage, 0, NULL, (PBYTE)original, &len); | |
| RegDeleteValue(hroot, acceptlanguage); | |
| /* Some windows versions use "lang-COUNTRY" as default */ | |
| memset(language, 0, sizeof(language)); | |
| len = GetLocaleInfoA(lcid, LOCALE_SISO639LANGNAME, language, sizeof(language)); | |
| if (len) { | |
| lstrcat(language, "-"); | |
| memset(buffer, 0, sizeof(buffer)); | |
| len = GetLocaleInfoA(lcid, LOCALE_SISO3166CTRYNAME, buffer, sizeof(buffer) - len - 1); | |
| lstrcat(language, buffer); | |
| } | |
| else | |
| { | |
| /* LOCALE_SNAME has additional parts in some languages. Try only as last chance */ | |
| memset(language, 0, sizeof(language)); | |
| len = GetLocaleInfoA(lcid, LOCALE_SNAME, language, sizeof(language)); | |
| } | |
| /* get the default value */ | |
| len = maxlen; | |
| memset(buffer, '#', maxlen); | |
| buffer[maxlen] = 0; | |
| hr = pGetAcceptLanguagesA( buffer, &len); | |
| if (hr != S_OK) { | |
| win_skip("GetAcceptLanguagesA failed with 0x%x\n", hr); | |
| goto restore_original; | |
| } | |
| if (lstrcmpA(buffer, language)) { | |
| /* some windows versions use "lang" or "lang-country" as default */ | |
| language[0] = 0; | |
| if (pLcidToRfc1766A) { | |
| hr = pLcidToRfc1766A(lcid, language, sizeof(language)); | |
| ok(hr == S_OK, "LcidToRfc1766A returned 0x%x and %s\n", hr, language); | |
| } | |
| } | |
| ok(!lstrcmpA(buffer, language), | |
| "have '%s' (searching for '%s')\n", language, buffer); | |
| if (lstrcmpA(buffer, language)) { | |
| win_skip("no more ideas, how to build the default language '%s'\n", buffer); | |
| goto restore_original; | |
| } | |
| trace("detected default: %s\n", language); | |
| while ((entry = table[i])) { | |
| exactsize = lstrlenA(entry); | |
| lres = RegSetValueExA(hroot, acceptlanguage, 0, REG_SZ, (const BYTE *) entry, exactsize + 1); | |
| ok(!lres, "got %d for RegSetValueExA: %s\n", lres, entry); | |
| /* len includes space for the terminating 0 before vista/w2k8 */ | |
| len = exactsize + 2; | |
| memset(buffer, '#', maxlen); | |
| buffer[maxlen] = 0; | |
| hr = pGetAcceptLanguagesA( buffer, &len); | |
| ok(((hr == E_INVALIDARG) && (len == 0)) || | |
| (SUCCEEDED(hr) && | |
| ((len == exactsize) || (len == exactsize+1)) && | |
| !lstrcmpA(buffer, entry)), | |
| "+2_#%d: got 0x%x with %d and %s\n", i, hr, len, buffer); | |
| len = exactsize + 1; | |
| memset(buffer, '#', maxlen); | |
| buffer[maxlen] = 0; | |
| hr = pGetAcceptLanguagesA( buffer, &len); | |
| ok(((hr == E_INVALIDARG) && (len == 0)) || | |
| (SUCCEEDED(hr) && | |
| ((len == exactsize) || (len == exactsize+1)) && | |
| !lstrcmpA(buffer, entry)), | |
| "+1_#%d: got 0x%x with %d and %s\n", i, hr, len, buffer); | |
| len = exactsize; | |
| memset(buffer, '#', maxlen); | |
| buffer[maxlen] = 0; | |
| hr = pGetAcceptLanguagesA( buffer, &len); | |
| /* There is no space for the string in the registry. | |
| When the buffer is large enough, the default language is returned | |
| When the buffer is to small for that fallback, win7_32 and w2k8_64 | |
| and above fail with HRESULT_FROM_WIN32(ERROR_INSUFFICIENT_BUFFER), but | |
| recent os succeed and return a partial result while | |
| older os succeed and overflow the buffer */ | |
| ok(((hr == E_INVALIDARG) && (len == 0)) || | |
| (((hr == S_OK) && !lstrcmpA(buffer, language) && (len == lstrlenA(language))) || | |
| ((hr == S_OK) && !memcmp(buffer, language, len)) || | |
| ((hr == __HRESULT_FROM_WIN32(ERROR_INSUFFICIENT_BUFFER)) && !len)), | |
| "==_#%d: got 0x%x with %d and %s\n", i, hr, len, buffer); | |
| if (exactsize > 1) { | |
| len = exactsize - 1; | |
| memset(buffer, '#', maxlen); | |
| buffer[maxlen] = 0; | |
| hr = pGetAcceptLanguagesA( buffer, &len); | |
| ok(((hr == E_INVALIDARG) && (len == 0)) || | |
| (((hr == S_OK) && !lstrcmpA(buffer, language) && (len == lstrlenA(language))) || | |
| ((hr == S_OK) && !memcmp(buffer, language, len)) || | |
| ((hr == __HRESULT_FROM_WIN32(ERROR_INSUFFICIENT_BUFFER)) && !len)), | |
| "-1_#%d: got 0x%x with %d and %s\n", i, hr, len, buffer); | |
| } | |
| len = 1; | |
| memset(buffer, '#', maxlen); | |
| buffer[maxlen] = 0; | |
| hr = pGetAcceptLanguagesA( buffer, &len); | |
| ok(((hr == E_INVALIDARG) && (len == 0)) || | |
| (((hr == S_OK) && !lstrcmpA(buffer, language) && (len == lstrlenA(language))) || | |
| ((hr == S_OK) && !memcmp(buffer, language, len)) || | |
| ((hr == __HRESULT_FROM_WIN32(ERROR_INSUFFICIENT_BUFFER)) && !len)), | |
| "=1_#%d: got 0x%x with %d and %s\n", i, hr, len, buffer); | |
| len = maxlen; | |
| hr = pGetAcceptLanguagesA( NULL, &len); | |
| /* w2k3 and below: E_FAIL and untouched len, | |
| since w2k8: S_OK and needed size (excluding 0) */ | |
| ok( ((hr == S_OK) && (len == exactsize)) || | |
| ((hr == E_FAIL) && (len == maxlen)), | |
| "NULL,max #%d: got 0x%x with %d and %s\n", i, hr, len, buffer); | |
| i++; | |
| } | |
| /* without a value in the registry, a default language is returned */ | |
| RegDeleteValue(hroot, acceptlanguage); | |
| len = maxlen; | |
| memset(buffer, '#', maxlen); | |
| buffer[maxlen] = 0; | |
| hr = pGetAcceptLanguagesA( buffer, &len); | |
| ok( ((hr == S_OK) && (len == lstrlenA(language))), | |
| "max: got 0x%x with %d and %s (expected S_OK with %d and '%s'\n", | |
| hr, len, buffer, lstrlenA(language), language); | |
| len = 2; | |
| memset(buffer, '#', maxlen); | |
| buffer[maxlen] = 0; | |
| hr = pGetAcceptLanguagesA( buffer, &len); | |
| ok( (((hr == S_OK) || (hr == E_INVALIDARG)) && !memcmp(buffer, language, len)) || | |
| ((hr == __HRESULT_FROM_WIN32(ERROR_INSUFFICIENT_BUFFER)) && !len), | |
| "=2: got 0x%x with %d and %s\n", hr, len, buffer); | |
| len = 1; | |
| memset(buffer, '#', maxlen); | |
| buffer[maxlen] = 0; | |
| hr = pGetAcceptLanguagesA( buffer, &len); | |
| /* When the buffer is to small, win7_32 and w2k8_64 and above fail with | |
| HRESULT_FROM_WIN32(ERROR_INSUFFICIENT_BUFFER), other versions suceed | |
| and return a partial 0 terminated result while other versions | |
| fail with E_INVALIDARG and return a partial unterminated result */ | |
| ok( (((hr == S_OK) || (hr == E_INVALIDARG)) && !memcmp(buffer, language, len)) || | |
| ((hr == __HRESULT_FROM_WIN32(ERROR_INSUFFICIENT_BUFFER)) && !len), | |
| "=1: got 0x%x with %d and %s\n", hr, len, buffer); | |
| len = 0; | |
| memset(buffer, '#', maxlen); | |
| buffer[maxlen] = 0; | |
| hr = pGetAcceptLanguagesA( buffer, &len); | |
| /* w2k3 and below: E_FAIL, since w2k8: E_INVALIDARG */ | |
| ok((hr == E_FAIL) || (hr == E_INVALIDARG), | |
| "got 0x%x (expected E_FAIL or E_INVALIDARG)\n", hr); | |
| memset(buffer, '#', maxlen); | |
| buffer[maxlen] = 0; | |
| hr = pGetAcceptLanguagesA( buffer, NULL); | |
| /* w2k3 and below: E_FAIL, since w2k8: E_INVALIDARG */ | |
| ok((hr == E_FAIL) || (hr == E_INVALIDARG), | |
| "got 0x%x (expected E_FAIL or E_INVALIDARG)\n", hr); | |
| hr = pGetAcceptLanguagesA( NULL, NULL); | |
| /* w2k3 and below: E_FAIL, since w2k8: E_INVALIDARG */ | |
| ok((hr == E_FAIL) || (hr == E_INVALIDARG), | |
| "got 0x%x (expected E_FAIL or E_INVALIDARG)\n", hr); | |
| restore_original: | |
| if (!res_query) { | |
| len = lstrlenA(original); | |
| lres = RegSetValueExA(hroot, acceptlanguage, 0, REG_SZ, (const BYTE *) original, len ? len + 1: 0); | |
| ok(!lres, "RegSetValueEx(%s) failed: %d\n", original, lres); | |
| } | |
| else | |
| { | |
| RegDeleteValue(hroot, acceptlanguage); | |
| } | |
| RegCloseKey(hroot); | |
| } | |
| static void test_SHSearchMapInt(void) | |
| { | |
| int keys[8], values[8]; | |
| int i = 0; | |
| if (!pSHSearchMapInt) | |
| return; | |
| memset(keys, 0, sizeof(keys)); | |
| memset(values, 0, sizeof(values)); | |
| keys[0] = 99; values[0] = 101; | |
| /* NULL key/value lists crash native, so skip testing them */ | |
| /* 1 element */ | |
| i = pSHSearchMapInt(keys, values, 1, keys[0]); | |
| ok(i == values[0], "Len 1, expected %d, got %d\n", values[0], i); | |
| /* Key doesn't exist */ | |
| i = pSHSearchMapInt(keys, values, 1, 100); | |
| ok(i == -1, "Len 1 - bad key, expected -1, got %d\n", i); | |
| /* Len = 0 => not found */ | |
| i = pSHSearchMapInt(keys, values, 0, keys[0]); | |
| ok(i == -1, "Len 1 - passed len 0, expected -1, got %d\n", i); | |
| /* 2 elements, len = 1 */ | |
| keys[1] = 98; values[1] = 102; | |
| i = pSHSearchMapInt(keys, values, 1, keys[1]); | |
| ok(i == -1, "Len 1 - array len 2, expected -1, got %d\n", i); | |
| /* 2 elements, len = 2 */ | |
| i = pSHSearchMapInt(keys, values, 2, keys[1]); | |
| ok(i == values[1], "Len 2, expected %d, got %d\n", values[1], i); | |
| /* Searches forward */ | |
| keys[2] = 99; values[2] = 103; | |
| i = pSHSearchMapInt(keys, values, 3, keys[0]); | |
| ok(i == values[0], "Len 3, expected %d, got %d\n", values[0], i); | |
| } | |
| static void test_alloc_shared(void) | |
| { | |
| DWORD procid; | |
| HANDLE hmem; | |
| int val; | |
| int* p; | |
| BOOL ret; | |
| procid=GetCurrentProcessId(); | |
| hmem=pSHAllocShared(NULL,10,procid); | |
| ok(hmem!=NULL,"SHAllocShared(NULL...) failed: %u\n", GetLastError()); | |
| ret = pSHFreeShared(hmem, procid); | |
| ok( ret, "SHFreeShared failed: %u\n", GetLastError()); | |
| val=0x12345678; | |
| hmem=pSHAllocShared(&val,4,procid); | |
| ok(hmem!=NULL,"SHAllocShared(NULL...) failed: %u\n", GetLastError()); | |
| p=pSHLockShared(hmem,procid); | |
| ok(p!=NULL,"SHLockShared failed: %u\n", GetLastError()); | |
| if (p!=NULL) | |
| ok(*p==val,"Wrong value in shared memory: %d instead of %d\n",*p,val); | |
| ret = pSHUnlockShared(p); | |
| ok( ret, "SHUnlockShared failed: %u\n", GetLastError()); | |
| ret = pSHFreeShared(hmem, procid); | |
| ok( ret, "SHFreeShared failed: %u\n", GetLastError()); | |
| } | |
| static void test_fdsa(void) | |
| { | |
| typedef struct | |
| { | |
| DWORD num_items; /* Number of elements inserted */ | |
| void *mem; /* Ptr to array */ | |
| DWORD blocks_alloced; /* Number of elements allocated */ | |
| BYTE inc; /* Number of elements to grow by when we need to expand */ | |
| BYTE block_size; /* Size in bytes of an element */ | |
| BYTE flags; /* Flags */ | |
| } FDSA_info; | |
| BOOL (WINAPI *pFDSA_Initialize)(DWORD block_size, DWORD inc, FDSA_info *info, void *mem, | |
| DWORD init_blocks); | |
| BOOL (WINAPI *pFDSA_Destroy)(FDSA_info *info); | |
| DWORD (WINAPI *pFDSA_InsertItem)(FDSA_info *info, DWORD where, const void *block); | |
| BOOL (WINAPI *pFDSA_DeleteItem)(FDSA_info *info, DWORD where); | |
| FDSA_info info; | |
| int block_size = 10, init_blocks = 4, inc = 2; | |
| DWORD ret; | |
| char *mem; | |
| pFDSA_Initialize = (void *)GetProcAddress(hShlwapi, (LPSTR)208); | |
| pFDSA_Destroy = (void *)GetProcAddress(hShlwapi, (LPSTR)209); | |
| pFDSA_InsertItem = (void *)GetProcAddress(hShlwapi, (LPSTR)210); | |
| pFDSA_DeleteItem = (void *)GetProcAddress(hShlwapi, (LPSTR)211); | |
| mem = HeapAlloc(GetProcessHeap(), 0, block_size * init_blocks); | |
| memset(&info, 0, sizeof(info)); | |
| ok(pFDSA_Initialize(block_size, inc, &info, mem, init_blocks), "FDSA_Initialize rets FALSE\n"); | |
| ok(info.num_items == 0, "num_items = %d\n", info.num_items); | |
| ok(info.mem == mem, "mem = %p\n", info.mem); | |
| ok(info.blocks_alloced == init_blocks, "blocks_alloced = %d\n", info.blocks_alloced); | |
| ok(info.inc == inc, "inc = %d\n", info.inc); | |
| ok(info.block_size == block_size, "block_size = %d\n", info.block_size); | |
| ok(info.flags == 0, "flags = %d\n", info.flags); | |
| ret = pFDSA_InsertItem(&info, 1234, "1234567890"); | |
| ok(ret == 0, "ret = %d\n", ret); | |
| ok(info.num_items == 1, "num_items = %d\n", info.num_items); | |
| ok(info.mem == mem, "mem = %p\n", info.mem); | |
| ok(info.blocks_alloced == init_blocks, "blocks_alloced = %d\n", info.blocks_alloced); | |
| ok(info.inc == inc, "inc = %d\n", info.inc); | |
| ok(info.block_size == block_size, "block_size = %d\n", info.block_size); | |
| ok(info.flags == 0, "flags = %d\n", info.flags); | |
| ret = pFDSA_InsertItem(&info, 1234, "abcdefghij"); | |
| ok(ret == 1, "ret = %d\n", ret); | |
| ret = pFDSA_InsertItem(&info, 1, "klmnopqrst"); | |
| ok(ret == 1, "ret = %d\n", ret); | |
| ret = pFDSA_InsertItem(&info, 0, "uvwxyzABCD"); | |
| ok(ret == 0, "ret = %d\n", ret); | |
| ok(info.mem == mem, "mem = %p\n", info.mem); | |
| ok(info.flags == 0, "flags = %d\n", info.flags); | |
| /* This next InsertItem will cause shlwapi to allocate its own mem buffer */ | |
| ret = pFDSA_InsertItem(&info, 0, "EFGHIJKLMN"); | |
| ok(ret == 0, "ret = %d\n", ret); | |
| ok(info.mem != mem, "mem = %p\n", info.mem); | |
| ok(info.blocks_alloced == init_blocks + inc, "blocks_alloced = %d\n", info.blocks_alloced); | |
| ok(info.flags == 0x1, "flags = %d\n", info.flags); | |
| ok(!memcmp(info.mem, "EFGHIJKLMNuvwxyzABCD1234567890klmnopqrstabcdefghij", 50), "mem %s\n", (char*)info.mem); | |
| ok(pFDSA_DeleteItem(&info, 2), "rets FALSE\n"); | |
| ok(info.mem != mem, "mem = %p\n", info.mem); | |
| ok(info.blocks_alloced == init_blocks + inc, "blocks_alloced = %d\n", info.blocks_alloced); | |
| ok(info.flags == 0x1, "flags = %d\n", info.flags); | |
| ok(!memcmp(info.mem, "EFGHIJKLMNuvwxyzABCDklmnopqrstabcdefghij", 40), "mem %s\n", (char*)info.mem); | |
| ok(pFDSA_DeleteItem(&info, 3), "rets FALSE\n"); | |
| ok(info.mem != mem, "mem = %p\n", info.mem); | |
| ok(info.blocks_alloced == init_blocks + inc, "blocks_alloced = %d\n", info.blocks_alloced); | |
| ok(info.flags == 0x1, "flags = %d\n", info.flags); | |
| ok(!memcmp(info.mem, "EFGHIJKLMNuvwxyzABCDklmnopqrst", 30), "mem %s\n", (char*)info.mem); | |
| ok(!pFDSA_DeleteItem(&info, 4), "does not ret FALSE\n"); | |
| /* As shlwapi has allocated memory internally, Destroy will ret FALSE */ | |
| ok(!pFDSA_Destroy(&info), "FDSA_Destroy does not ret FALSE\n"); | |
| /* When Initialize is called with inc = 0, set it to 1 */ | |
| ok(pFDSA_Initialize(block_size, 0, &info, mem, init_blocks), "FDSA_Initialize rets FALSE\n"); | |
| ok(info.inc == 1, "inc = %d\n", info.inc); | |
| /* This time, because shlwapi hasn't had to allocate memory | |
| internally, Destroy rets non-zero */ | |
| ok(pFDSA_Destroy(&info), "FDSA_Destroy rets FALSE\n"); | |
| HeapFree(GetProcessHeap(), 0, mem); | |
| } | |
| typedef struct SHELL_USER_SID { | |
| SID_IDENTIFIER_AUTHORITY sidAuthority; | |
| DWORD dwUserGroupID; | |
| DWORD dwUserID; | |
| } SHELL_USER_SID, *PSHELL_USER_SID; | |
| typedef struct SHELL_USER_PERMISSION { | |
| SHELL_USER_SID susID; | |
| DWORD dwAccessType; | |
| BOOL fInherit; | |
| DWORD dwAccessMask; | |
| DWORD dwInheritMask; | |
| DWORD dwInheritAccessMask; | |
| } SHELL_USER_PERMISSION, *PSHELL_USER_PERMISSION; | |
| static void test_GetShellSecurityDescriptor(void) | |
| { | |
| SHELL_USER_PERMISSION supCurrentUserFull = { | |
| { {SECURITY_NULL_SID_AUTHORITY}, 0, 0 }, | |
| ACCESS_ALLOWED_ACE_TYPE, FALSE, | |
| GENERIC_ALL, 0, 0 }; | |
| #define MY_INHERITANCE 0xBE /* invalid value to proof behavior */ | |
| SHELL_USER_PERMISSION supEveryoneDenied = { | |
| { {SECURITY_WORLD_SID_AUTHORITY}, SECURITY_WORLD_RID, 0 }, | |
| ACCESS_DENIED_ACE_TYPE, TRUE, | |
| GENERIC_WRITE, MY_INHERITANCE | 0xDEADBA00, GENERIC_READ }; | |
| PSHELL_USER_PERMISSION rgsup[2] = { | |
| &supCurrentUserFull, &supEveryoneDenied, | |
| }; | |
| SECURITY_DESCRIPTOR* psd; | |
| SECURITY_DESCRIPTOR* (WINAPI*pGetShellSecurityDescriptor)(PSHELL_USER_PERMISSION*,int); | |
| pGetShellSecurityDescriptor=(void*)GetProcAddress(hShlwapi,(char*)475); | |
| if(!pGetShellSecurityDescriptor) | |
| { | |
| win_skip("GetShellSecurityDescriptor not available\n"); | |
| return; | |
| } | |
| psd = pGetShellSecurityDescriptor(NULL, 2); | |
| ok(psd==NULL || | |
| broken(psd==INVALID_HANDLE_VALUE), /* IE5 */ | |
| "GetShellSecurityDescriptor should fail\n"); | |
| psd = pGetShellSecurityDescriptor(rgsup, 0); | |
| ok(psd==NULL, "GetShellSecurityDescriptor should fail\n"); | |
| SetLastError(0xdeadbeef); | |
| psd = pGetShellSecurityDescriptor(rgsup, 2); | |
| if (psd == NULL && GetLastError() == ERROR_CALL_NOT_IMPLEMENTED) | |
| { | |
| /* The previous calls to GetShellSecurityDescriptor don't set the last error */ | |
| win_skip("GetShellSecurityDescriptor is not implemented\n"); | |
| return; | |
| } | |
| if (psd==INVALID_HANDLE_VALUE) | |
| { | |
| win_skip("GetShellSecurityDescriptor is broken on IE5\n"); | |
| return; | |
| } | |
| ok(psd!=NULL, "GetShellSecurityDescriptor failed\n"); | |
| if (psd!=NULL) | |
| { | |
| BOOL bHasDacl = FALSE, bDefaulted; | |
| PACL pAcl; | |
| DWORD dwRev; | |
| SECURITY_DESCRIPTOR_CONTROL control; | |
| ok(IsValidSecurityDescriptor(psd), "returned value is not valid SD\n"); | |
| ok(GetSecurityDescriptorControl(psd, &control, &dwRev), | |
| "GetSecurityDescriptorControl failed with error %u\n", GetLastError()); | |
| ok(0 == (control & SE_SELF_RELATIVE), "SD should be absolute\n"); | |
| ok(GetSecurityDescriptorDacl(psd, &bHasDacl, &pAcl, &bDefaulted), | |
| "GetSecurityDescriptorDacl failed with error %u\n", GetLastError()); | |
| ok(bHasDacl, "SD has no DACL\n"); | |
| if (bHasDacl) | |
| { | |
| ok(!bDefaulted, "DACL should not be defaulted\n"); | |
| ok(pAcl != NULL, "NULL DACL!\n"); | |
| if (pAcl != NULL) | |
| { | |
| ACL_SIZE_INFORMATION asiSize; | |
| ok(IsValidAcl(pAcl), "DACL is not valid\n"); | |
| ok(GetAclInformation(pAcl, &asiSize, sizeof(asiSize), AclSizeInformation), | |
| "GetAclInformation failed with error %u\n", GetLastError()); | |
| ok(asiSize.AceCount == 3, "Incorrect number of ACEs: %d entries\n", asiSize.AceCount); | |
| if (asiSize.AceCount == 3) | |
| { | |
| ACCESS_ALLOWED_ACE *paaa; /* will use for DENIED too */ | |
| ok(GetAce(pAcl, 0, (LPVOID*)&paaa), "GetAce failed with error %u\n", GetLastError()); | |
| ok(paaa->Header.AceType == ACCESS_ALLOWED_ACE_TYPE, | |
| "Invalid ACE type %d\n", paaa->Header.AceType); | |
| ok(paaa->Header.AceFlags == 0, "Invalid ACE flags %x\n", paaa->Header.AceFlags); | |
| ok(paaa->Mask == GENERIC_ALL, "Invalid ACE mask %x\n", paaa->Mask); | |
| ok(GetAce(pAcl, 1, (LPVOID*)&paaa), "GetAce failed with error %u\n", GetLastError()); | |
| ok(paaa->Header.AceType == ACCESS_DENIED_ACE_TYPE, | |
| "Invalid ACE type %d\n", paaa->Header.AceType); | |
| /* first one of two ACEs generated from inheritable entry - without inheritance */ | |
| ok(paaa->Header.AceFlags == 0, "Invalid ACE flags %x\n", paaa->Header.AceFlags); | |
| ok(paaa->Mask == GENERIC_WRITE, "Invalid ACE mask %x\n", paaa->Mask); | |
| ok(GetAce(pAcl, 2, (LPVOID*)&paaa), "GetAce failed with error %u\n", GetLastError()); | |
| ok(paaa->Header.AceType == ACCESS_DENIED_ACE_TYPE, | |
| "Invalid ACE type %d\n", paaa->Header.AceType); | |
| /* second ACE - with inheritance */ | |
| ok(paaa->Header.AceFlags == MY_INHERITANCE, | |
| "Invalid ACE flags %x\n", paaa->Header.AceFlags); | |
| ok(paaa->Mask == GENERIC_READ, "Invalid ACE mask %x\n", paaa->Mask); | |
| } | |
| } | |
| } | |
| LocalFree(psd); | |
| } | |
| } | |
| static void test_SHPackDispParams(void) | |
| { | |
| DISPPARAMS params; | |
| VARIANT vars[10]; | |
| HRESULT hres; | |
| if(!pSHPackDispParams) | |
| win_skip("SHPackSidpParams not available\n"); | |
| memset(¶ms, 0xc0, sizeof(params)); | |
| memset(vars, 0xc0, sizeof(vars)); | |
| hres = pSHPackDispParams(¶ms, vars, 1, VT_I4, 0xdeadbeef); | |
| ok(hres == S_OK, "SHPackDispParams failed: %08x\n", hres); | |
| ok(params.cArgs == 1, "params.cArgs = %d\n", params.cArgs); | |
| ok(params.cNamedArgs == 0, "params.cNamedArgs = %d\n", params.cArgs); | |
| ok(params.rgdispidNamedArgs == NULL, "params.rgdispidNamedArgs = %p\n", params.rgdispidNamedArgs); | |
| ok(params.rgvarg == vars, "params.rgvarg = %p\n", params.rgvarg); | |
| ok(V_VT(vars) == VT_I4, "V_VT(var) = %d\n", V_VT(vars)); | |
| ok(V_I4(vars) == 0xdeadbeef, "failed %x\n", V_I4(vars)); | |
| memset(¶ms, 0xc0, sizeof(params)); | |
| hres = pSHPackDispParams(¶ms, NULL, 0, 0); | |
| ok(hres == S_OK, "SHPackDispParams failed: %08x\n", hres); | |
| ok(params.cArgs == 0, "params.cArgs = %d\n", params.cArgs); | |
| ok(params.cNamedArgs == 0, "params.cNamedArgs = %d\n", params.cArgs); | |
| ok(params.rgdispidNamedArgs == NULL, "params.rgdispidNamedArgs = %p\n", params.rgdispidNamedArgs); | |
| ok(params.rgvarg == NULL, "params.rgvarg = %p\n", params.rgvarg); | |
| memset(vars, 0xc0, sizeof(vars)); | |
| memset(¶ms, 0xc0, sizeof(params)); | |
| hres = pSHPackDispParams(¶ms, vars, 4, VT_BSTR, (void*)0xdeadbeef, VT_EMPTY, 10, | |
| VT_I4, 100, VT_DISPATCH, (void*)0xdeadbeef); | |
| ok(hres == S_OK, "SHPackDispParams failed: %08x\n", hres); | |
| ok(params.cArgs == 4, "params.cArgs = %d\n", params.cArgs); | |
| ok(params.cNamedArgs == 0, "params.cNamedArgs = %d\n", params.cArgs); | |
| ok(params.rgdispidNamedArgs == NULL, "params.rgdispidNamedArgs = %p\n", params.rgdispidNamedArgs); | |
| ok(params.rgvarg == vars, "params.rgvarg = %p\n", params.rgvarg); | |
| ok(V_VT(vars) == VT_DISPATCH, "V_VT(vars[0]) = %x\n", V_VT(vars)); | |
| ok(V_I4(vars) == 0xdeadbeef, "V_I4(vars[0]) = %x\n", V_I4(vars)); | |
| ok(V_VT(vars+1) == VT_I4, "V_VT(vars[1]) = %d\n", V_VT(vars+1)); | |
| ok(V_I4(vars+1) == 100, "V_I4(vars[1]) = %x\n", V_I4(vars+1)); | |
| ok(V_VT(vars+2) == VT_I4, "V_VT(vars[2]) = %d\n", V_VT(vars+2)); | |
| ok(V_I4(vars+2) == 10, "V_I4(vars[2]) = %x\n", V_I4(vars+2)); | |
| ok(V_VT(vars+3) == VT_BSTR, "V_VT(vars[3]) = %d\n", V_VT(vars+3)); | |
| ok(V_BSTR(vars+3) == (void*)0xdeadbeef, "V_BSTR(vars[3]) = %p\n", V_BSTR(vars+3)); | |
| } | |
| typedef struct _disp | |
| { | |
| const IDispatchVtbl *vtbl; | |
| LONG refCount; | |
| } Disp; | |
| typedef struct _contain | |
| { | |
| const IConnectionPointContainerVtbl *vtbl; | |
| LONG refCount; | |
| UINT ptCount; | |
| IConnectionPoint **pt; | |
| } Contain; | |
| typedef struct _cntptn | |
| { | |
| const IConnectionPointVtbl *vtbl; | |
| LONG refCount; | |
| Contain *container; | |
| GUID id; | |
| UINT sinkCount; | |
| IUnknown **sink; | |
| } ConPt; | |
| typedef struct _enum | |
| { | |
| const IEnumConnectionsVtbl *vtbl; | |
| LONG refCount; | |
| UINT idx; | |
| ConPt *pt; | |
| } EnumCon; | |
| typedef struct _enumpt | |
| { | |
| const IEnumConnectionPointsVtbl *vtbl; | |
| LONG refCount; | |
| int idx; | |
| Contain *container; | |
| } EnumPt; | |
| static HRESULT WINAPI Disp_QueryInterface( | |
| IDispatch* This, | |
| REFIID riid, | |
| void **ppvObject) | |
| { | |
| *ppvObject = NULL; | |
| if (IsEqualIID(riid, &IID_IUnknown) || IsEqualIID(riid, &IID_IDispatch)) | |
| { | |
| *ppvObject = This; | |
| } | |
| if (*ppvObject) | |
| { | |
| IUnknown_AddRef(This); | |
| return S_OK; | |
| } | |
| trace("no interface\n"); | |
| return E_NOINTERFACE; | |
| } | |
| static ULONG WINAPI Disp_AddRef(IDispatch* This) | |
| { | |
| Disp *iface = (Disp*)This; | |
| return InterlockedIncrement(&iface->refCount); | |
| } | |
| static ULONG WINAPI Disp_Release(IDispatch* This) | |
| { | |
| Disp *iface = (Disp*)This; | |
| ULONG ret; | |
| ret = InterlockedDecrement(&iface->refCount); | |
| if (ret == 0) | |
| HeapFree(GetProcessHeap(),0,This); | |
| return ret; | |
| } | |
| static HRESULT WINAPI Disp_GetTypeInfoCount( | |
| IDispatch* This, | |
| UINT *pctinfo) | |
| { | |
| return ERROR_SUCCESS; | |
| } | |
| static HRESULT WINAPI Disp_GetTypeInfo( | |
| IDispatch* This, | |
| UINT iTInfo, | |
| LCID lcid, | |
| ITypeInfo **ppTInfo) | |
| { | |
| return ERROR_SUCCESS; | |
| } | |
| static HRESULT WINAPI Disp_GetIDsOfNames( | |
| IDispatch* This, | |
| REFIID riid, | |
| LPOLESTR *rgszNames, | |
| UINT cNames, | |
| LCID lcid, | |
| DISPID *rgDispId) | |
| { | |
| return ERROR_SUCCESS; | |
| } | |
| static HRESULT WINAPI Disp_Invoke( | |
| IDispatch* This, | |
| DISPID dispIdMember, | |
| REFIID riid, | |
| LCID lcid, | |
| WORD wFlags, | |
| DISPPARAMS *pDispParams, | |
| VARIANT *pVarResult, | |
| EXCEPINFO *pExcepInfo, | |
| UINT *puArgErr) | |
| { | |
| trace("%p %x %p %x %x %p %p %p %p\n",This,dispIdMember,riid,lcid,wFlags,pDispParams,pVarResult,pExcepInfo,puArgErr); | |
| ok(dispIdMember == 0xa0 || dispIdMember == 0xa1, "Unknown dispIdMember\n"); | |
| ok(pDispParams != NULL, "Invoked with NULL pDispParams\n"); | |
| ok(wFlags == DISPATCH_METHOD, "Wrong flags %x\n",wFlags); | |
| ok(lcid == 0,"Wrong lcid %x\n",lcid); | |
| if (dispIdMember == 0xa0) | |
| { | |
| ok(pDispParams->cArgs == 0, "params.cArgs = %d\n", pDispParams->cArgs); | |
| ok(pDispParams->cNamedArgs == 0, "params.cNamedArgs = %d\n", pDispParams->cArgs); | |
| ok(pDispParams->rgdispidNamedArgs == NULL, "params.rgdispidNamedArgs = %p\n", pDispParams->rgdispidNamedArgs); | |
| ok(pDispParams->rgvarg == NULL, "params.rgvarg = %p\n", pDispParams->rgvarg); | |
| } | |
| else if (dispIdMember == 0xa1) | |
| { | |
| ok(pDispParams->cArgs == 2, "params.cArgs = %d\n", pDispParams->cArgs); | |
| ok(pDispParams->cNamedArgs == 0, "params.cNamedArgs = %d\n", pDispParams->cArgs); | |
| ok(pDispParams->rgdispidNamedArgs == NULL, "params.rgdispidNamedArgs = %p\n", pDispParams->rgdispidNamedArgs); | |
| ok(V_VT(pDispParams->rgvarg) == VT_BSTR, "V_VT(var) = %d\n", V_VT(pDispParams->rgvarg)); | |
| ok(V_I4(pDispParams->rgvarg) == 0xdeadcafe , "failed %p\n", V_BSTR(pDispParams->rgvarg)); | |
| ok(V_VT(pDispParams->rgvarg+1) == VT_I4, "V_VT(var) = %d\n", V_VT(pDispParams->rgvarg+1)); | |
| ok(V_I4(pDispParams->rgvarg+1) == 0xdeadbeef, "failed %x\n", V_I4(pDispParams->rgvarg+1)); | |
| } | |
| return ERROR_SUCCESS; | |
| } | |
| static const IDispatchVtbl disp_vtbl = { | |
| Disp_QueryInterface, | |
| Disp_AddRef, | |
| Disp_Release, | |
| Disp_GetTypeInfoCount, | |
| Disp_GetTypeInfo, | |
| Disp_GetIDsOfNames, | |
| Disp_Invoke | |
| }; | |
| static HRESULT WINAPI Enum_QueryInterface( | |
| IEnumConnections* This, | |
| REFIID riid, | |
| void **ppvObject) | |
| { | |
| *ppvObject = NULL; | |
| if (IsEqualIID(riid, &IID_IUnknown) || IsEqualIID(riid, &IID_IEnumConnections)) | |
| { | |
| *ppvObject = This; | |
| } | |
| if (*ppvObject) | |
| { | |
| IUnknown_AddRef(This); | |
| return S_OK; | |
| } | |
| trace("no interface\n"); | |
| return E_NOINTERFACE; | |
| } | |
| static ULONG WINAPI Enum_AddRef(IEnumConnections* This) | |
| { | |
| EnumCon *iface = (EnumCon*)This; | |
| return InterlockedIncrement(&iface->refCount); | |
| } | |
| static ULONG WINAPI Enum_Release(IEnumConnections* This) | |
| { | |
| EnumCon *iface = (EnumCon*)This; | |
| ULONG ret; | |
| ret = InterlockedDecrement(&iface->refCount); | |
| if (ret == 0) | |
| HeapFree(GetProcessHeap(),0,This); | |
| return ret; | |
| } | |
| static HRESULT WINAPI Enum_Next( | |
| IEnumConnections* This, | |
| ULONG cConnections, | |
| LPCONNECTDATA rgcd, | |
| ULONG *pcFetched) | |
| { | |
| EnumCon *iface = (EnumCon*)This; | |
| if (cConnections > 0 && iface->idx < iface->pt->sinkCount) | |
| { | |
| rgcd->pUnk = iface->pt->sink[iface->idx]; | |
| IUnknown_AddRef(iface->pt->sink[iface->idx]); | |
| rgcd->dwCookie=0xff; | |
| if (pcFetched) | |
| *pcFetched = 1; | |
| iface->idx++; | |
| return S_OK; | |
| } | |
| return E_FAIL; | |
| } | |
| static HRESULT WINAPI Enum_Skip( | |
| IEnumConnections* This, | |
| ULONG cConnections) | |
| { | |
| return E_FAIL; | |
| } | |
| static HRESULT WINAPI Enum_Reset( | |
| IEnumConnections* This) | |
| { | |
| return E_FAIL; | |
| } | |
| static HRESULT WINAPI Enum_Clone( | |
| IEnumConnections* This, | |
| IEnumConnections **ppEnum) | |
| { | |
| return E_FAIL; | |
| } | |
| static const IEnumConnectionsVtbl enum_vtbl = { | |
| Enum_QueryInterface, | |
| Enum_AddRef, | |
| Enum_Release, | |
| Enum_Next, | |
| Enum_Skip, | |
| Enum_Reset, | |
| Enum_Clone | |
| }; | |
| static HRESULT WINAPI ConPt_QueryInterface( | |
| IConnectionPoint* This, | |
| REFIID riid, | |
| void **ppvObject) | |
| { | |
| *ppvObject = NULL; | |
| if (IsEqualIID(riid, &IID_IUnknown) || IsEqualIID(riid, &IID_IConnectionPoint)) | |
| { | |
| *ppvObject = This; | |
| } | |
| if (*ppvObject) | |
| { | |
| IUnknown_AddRef(This); | |
| return S_OK; | |
| } | |
| trace("no interface\n"); | |
| return E_NOINTERFACE; | |
| } | |
| static ULONG WINAPI ConPt_AddRef( | |
| IConnectionPoint* This) | |
| { | |
| ConPt *iface = (ConPt*)This; | |
| return InterlockedIncrement(&iface->refCount); | |
| } | |
| static ULONG WINAPI ConPt_Release( | |
| IConnectionPoint* This) | |
| { | |
| ConPt *iface = (ConPt*)This; | |
| ULONG ret; | |
| ret = InterlockedDecrement(&iface->refCount); | |
| if (ret == 0) | |
| { | |
| if (iface->sinkCount > 0) | |
| { | |
| int i; | |
| for (i = 0; i < iface->sinkCount; i++) | |
| { | |
| if (iface->sink[i]) | |
| IUnknown_Release(iface->sink[i]); | |
| } | |
| HeapFree(GetProcessHeap(),0,iface->sink); | |
| } | |
| HeapFree(GetProcessHeap(),0,This); | |
| } | |
| return ret; | |
| } | |
| static HRESULT WINAPI ConPt_GetConnectionInterface( | |
| IConnectionPoint* This, | |
| IID *pIID) | |
| { | |
| static int i = 0; | |
| ConPt *iface = (ConPt*)This; | |
| if (i==0) | |
| { | |
| i++; | |
| return E_FAIL; | |
| } | |
| else | |
| memcpy(pIID,&iface->id,sizeof(GUID)); | |
| return S_OK; | |
| } | |
| static HRESULT WINAPI ConPt_GetConnectionPointContainer( | |
| IConnectionPoint* This, | |
| IConnectionPointContainer **ppCPC) | |
| { | |
| ConPt *iface = (ConPt*)This; | |
| *ppCPC = (IConnectionPointContainer*)iface->container; | |
| return S_OK; | |
| } | |
| static HRESULT WINAPI ConPt_Advise( | |
| IConnectionPoint* This, | |
| IUnknown *pUnkSink, | |
| DWORD *pdwCookie) | |
| { | |
| ConPt *iface = (ConPt*)This; | |
| if (iface->sinkCount == 0) | |
| iface->sink = HeapAlloc(GetProcessHeap(),0,sizeof(IUnknown*)); | |
| else | |
| iface->sink = HeapReAlloc(GetProcessHeap(),0,iface->sink,sizeof(IUnknown*)*(iface->sinkCount+1)); | |
| iface->sink[iface->sinkCount] = pUnkSink; | |
| IUnknown_AddRef(pUnkSink); | |
| iface->sinkCount++; | |
| *pdwCookie = iface->sinkCount; | |
| return S_OK; | |
| } | |
| static HRESULT WINAPI ConPt_Unadvise( | |
| IConnectionPoint* This, | |
| DWORD dwCookie) | |
| { | |
| ConPt *iface = (ConPt*)This; | |
| if (dwCookie > iface->sinkCount) | |
| return E_FAIL; | |
| else | |
| { | |
| IUnknown_Release(iface->sink[dwCookie-1]); | |
| iface->sink[dwCookie-1] = NULL; | |
| } | |
| return S_OK; | |
| } | |
| static HRESULT WINAPI ConPt_EnumConnections( | |
| IConnectionPoint* This, | |
| IEnumConnections **ppEnum) | |
| { | |
| EnumCon *ec; | |
| ec = HeapAlloc(GetProcessHeap(),0,sizeof(EnumCon)); | |
| ec->vtbl = &enum_vtbl; | |
| ec->refCount = 1; | |
| ec->pt = (ConPt*)This; | |
| ec->idx = 0; | |
| *ppEnum = (IEnumConnections*)ec; | |
| return S_OK; | |
| } | |
| static const IConnectionPointVtbl point_vtbl = { | |
| ConPt_QueryInterface, | |
| ConPt_AddRef, | |
| ConPt_Release, | |
| ConPt_GetConnectionInterface, | |
| ConPt_GetConnectionPointContainer, | |
| ConPt_Advise, | |
| ConPt_Unadvise, | |
| ConPt_EnumConnections | |
| }; | |
| static HRESULT WINAPI EnumPt_QueryInterface( | |
| IEnumConnectionPoints* This, | |
| REFIID riid, | |
| void **ppvObject) | |
| { | |
| *ppvObject = NULL; | |
| if (IsEqualIID(riid, &IID_IUnknown) || IsEqualIID(riid, &IID_IEnumConnectionPoints)) | |
| { | |
| *ppvObject = This; | |
| } | |
| if (*ppvObject) | |
| { | |
| IUnknown_AddRef(This); | |
| return S_OK; | |
| } | |
| trace("no interface\n"); | |
| return E_NOINTERFACE; | |
| } | |
| static ULONG WINAPI EnumPt_AddRef(IEnumConnectionPoints* This) | |
| { | |
| EnumPt *iface = (EnumPt*)This; | |
| return InterlockedIncrement(&iface->refCount); | |
| } | |
| static ULONG WINAPI EnumPt_Release(IEnumConnectionPoints* This) | |
| { | |
| EnumPt *iface = (EnumPt*)This; | |
| ULONG ret; | |
| ret = InterlockedDecrement(&iface->refCount); | |
| if (ret == 0) | |
| HeapFree(GetProcessHeap(),0,This); | |
| return ret; | |
| } | |
| static HRESULT WINAPI EnumPt_Next( | |
| IEnumConnectionPoints* This, | |
| ULONG cConnections, | |
| IConnectionPoint **rgcd, | |
| ULONG *pcFetched) | |
| { | |
| EnumPt *iface = (EnumPt*)This; | |
| if (cConnections > 0 && iface->idx < iface->container->ptCount) | |
| { | |
| *rgcd = iface->container->pt[iface->idx]; | |
| IUnknown_AddRef(iface->container->pt[iface->idx]); | |
| if (pcFetched) | |
| *pcFetched = 1; | |
| iface->idx++; | |
| return S_OK; | |
| } | |
| return E_FAIL; | |
| } | |
| static HRESULT WINAPI EnumPt_Skip( | |
| IEnumConnectionPoints* This, | |
| ULONG cConnections) | |
| { | |
| return E_FAIL; | |
| } | |
| static HRESULT WINAPI EnumPt_Reset( | |
| IEnumConnectionPoints* This) | |
| { | |
| return E_FAIL; | |
| } | |
| static HRESULT WINAPI EnumPt_Clone( | |
| IEnumConnectionPoints* This, | |
| IEnumConnectionPoints **ppEnumPt) | |
| { | |
| return E_FAIL; | |
| } | |
| static const IEnumConnectionPointsVtbl enumpt_vtbl = { | |
| EnumPt_QueryInterface, | |
| EnumPt_AddRef, | |
| EnumPt_Release, | |
| EnumPt_Next, | |
| EnumPt_Skip, | |
| EnumPt_Reset, | |
| EnumPt_Clone | |
| }; | |
| static HRESULT WINAPI Contain_QueryInterface( | |
| IConnectionPointContainer* This, | |
| REFIID riid, | |
| void **ppvObject) | |
| { | |
| *ppvObject = NULL; | |
| if (IsEqualIID(riid, &IID_IUnknown) || IsEqualIID(riid, &IID_IConnectionPointContainer)) | |
| { | |
| *ppvObject = This; | |
| } | |
| if (*ppvObject) | |
| { | |
| IUnknown_AddRef(This); | |
| return S_OK; | |
| } | |
| trace("no interface\n"); | |
| return E_NOINTERFACE; | |
| } | |
| static ULONG WINAPI Contain_AddRef( | |
| IConnectionPointContainer* This) | |
| { | |
| Contain *iface = (Contain*)This; | |
| return InterlockedIncrement(&iface->refCount); | |
| } | |
| static ULONG WINAPI Contain_Release( | |
| IConnectionPointContainer* This) | |
| { | |
| Contain *iface = (Contain*)This; | |
| ULONG ret; | |
| ret = InterlockedDecrement(&iface->refCount); | |
| if (ret == 0) | |
| { | |
| if (iface->ptCount > 0) | |
| { | |
| int i; | |
| for (i = 0; i < iface->ptCount; i++) | |
| IUnknown_Release(iface->pt[i]); | |
| HeapFree(GetProcessHeap(),0,iface->pt); | |
| } | |
| HeapFree(GetProcessHeap(),0,This); | |
| } | |
| return ret; | |
| } | |
| static HRESULT WINAPI Contain_EnumConnectionPoints( | |
| IConnectionPointContainer* This, | |
| IEnumConnectionPoints **ppEnum) | |
| { | |
| EnumPt *ec; | |
| ec = HeapAlloc(GetProcessHeap(),0,sizeof(EnumPt)); | |
| ec->vtbl = &enumpt_vtbl; | |
| ec->refCount = 1; | |
| ec->idx= 0; | |
| ec->container = (Contain*)This; | |
| *ppEnum = (IEnumConnectionPoints*)ec; | |
| return S_OK; | |
| } | |
| static HRESULT WINAPI Contain_FindConnectionPoint( | |
| IConnectionPointContainer* This, | |
| REFIID riid, | |
| IConnectionPoint **ppCP) | |
| { | |
| Contain *iface = (Contain*)This; | |
| ConPt *pt; | |
| if (!IsEqualIID(riid, &IID_NULL) || iface->ptCount ==0) | |
| { | |
| pt = HeapAlloc(GetProcessHeap(),0,sizeof(ConPt)); | |
| pt->vtbl = &point_vtbl; | |
| pt->refCount = 1; | |
| pt->sinkCount = 0; | |
| pt->sink = NULL; | |
| pt->container = iface; | |
| pt->id = IID_IDispatch; | |
| if (iface->ptCount == 0) | |
| iface->pt =HeapAlloc(GetProcessHeap(),0,sizeof(IUnknown*)); | |
| else | |
| iface->pt = HeapReAlloc(GetProcessHeap(),0,iface->pt,sizeof(IUnknown*)*(iface->ptCount+1)); | |
| iface->pt[iface->ptCount] = (IConnectionPoint*)pt; | |
| iface->ptCount++; | |
| *ppCP = (IConnectionPoint*)pt; | |
| } | |
| else | |
| { | |
| *ppCP = iface->pt[0]; | |
| IUnknown_AddRef((IUnknown*)*ppCP); | |
| } | |
| return S_OK; | |
| } | |
| static const IConnectionPointContainerVtbl contain_vtbl = { | |
| Contain_QueryInterface, | |
| Contain_AddRef, | |
| Contain_Release, | |
| Contain_EnumConnectionPoints, | |
| Contain_FindConnectionPoint | |
| }; | |
| static void test_IConnectionPoint(void) | |
| { | |
| HRESULT rc; | |
| ULONG ref; | |
| IConnectionPoint *point; | |
| Contain *container; | |
| Disp *dispatch; | |
| DWORD cookie = 0xffffffff; | |
| DISPPARAMS params; | |
| VARIANT vars[10]; | |
| if (!pIConnectionPoint_SimpleInvoke || !pConnectToConnectionPoint) | |
| { | |
| win_skip("IConnectionPoint Apis not present\n"); | |
| return; | |
| } | |
| container = HeapAlloc(GetProcessHeap(),0,sizeof(Contain)); | |
| container->vtbl = &contain_vtbl; | |
| container->refCount = 1; | |
| container->ptCount = 0; | |
| container->pt = NULL; | |
| dispatch = HeapAlloc(GetProcessHeap(),0,sizeof(Disp)); | |
| dispatch->vtbl = &disp_vtbl; | |
| dispatch->refCount = 1; | |
| rc = pConnectToConnectionPoint((IUnknown*)dispatch, &IID_NULL, TRUE, (IUnknown*)container, &cookie, &point); | |
| ok(rc == S_OK, "pConnectToConnectionPoint failed with %x\n",rc); | |
| ok(point != NULL, "returned ConnectionPoint is NULL\n"); | |
| ok(cookie != 0xffffffff, "invalid cookie returned\n"); | |
| rc = pIConnectionPoint_SimpleInvoke(point,0xa0,NULL); | |
| ok(rc == S_OK, "pConnectToConnectionPoint failed with %x\n",rc); | |
| if (pSHPackDispParams) | |
| { | |
| memset(¶ms, 0xc0, sizeof(params)); | |
| memset(vars, 0xc0, sizeof(vars)); | |
| rc = pSHPackDispParams(¶ms, vars, 2, VT_I4, 0xdeadbeef, VT_BSTR, 0xdeadcafe); | |
| ok(rc == S_OK, "SHPackDispParams failed: %08x\n", rc); | |
| rc = pIConnectionPoint_SimpleInvoke(point,0xa1,¶ms); | |
| ok(rc == S_OK, "pConnectToConnectionPoint failed with %x\n",rc); | |
| } | |
| else | |
| win_skip("pSHPackDispParams not present\n"); | |
| rc = pConnectToConnectionPoint(NULL, &IID_NULL, FALSE, (IUnknown*)container, &cookie, NULL); | |
| ok(rc == S_OK, "pConnectToConnectionPoint failed with %x\n",rc); | |
| /* MSDN says this should be required but it crashs on XP | |
| IUnknown_Release(point); | |
| */ | |
| ref = IUnknown_Release((IUnknown*)container); | |
| ok(ref == 0, "leftover IConnectionPointContainer reference %i\n",ref); | |
| ref = IUnknown_Release((IUnknown*)dispatch); | |
| ok(ref == 0, "leftover IDispatch reference %i\n",ref); | |
| } | |
| typedef struct _propbag | |
| { | |
| const IPropertyBagVtbl *vtbl; | |
| LONG refCount; | |
| } PropBag; | |
| static HRESULT WINAPI Prop_QueryInterface( | |
| IPropertyBag* This, | |
| REFIID riid, | |
| void **ppvObject) | |
| { | |
| *ppvObject = NULL; | |
| if (IsEqualIID(riid, &IID_IUnknown) || IsEqualIID(riid, &IID_IPropertyBag)) | |
| { | |
| *ppvObject = This; | |
| } | |
| if (*ppvObject) | |
| { | |
| IUnknown_AddRef(This); | |
| return S_OK; | |
| } | |
| trace("no interface\n"); | |
| return E_NOINTERFACE; | |
| } | |
| static ULONG WINAPI Prop_AddRef( | |
| IPropertyBag* This) | |
| { | |
| PropBag *iface = (PropBag*)This; | |
| return InterlockedIncrement(&iface->refCount); | |
| } | |
| static ULONG WINAPI Prop_Release( | |
| IPropertyBag* This) | |
| { | |
| PropBag *iface = (PropBag*)This; | |
| ULONG ret; | |
| ret = InterlockedDecrement(&iface->refCount); | |
| if (ret == 0) | |
| HeapFree(GetProcessHeap(),0,This); | |
| return ret; | |
| } | |
| static HRESULT WINAPI Prop_Read( | |
| IPropertyBag* This, | |
| LPCOLESTR pszPropName, | |
| VARIANT *pVar, | |
| IErrorLog *pErrorLog) | |
| { | |
| V_VT(pVar) = VT_BLOB|VT_BYREF; | |
| V_BYREF(pVar) = (LPVOID)0xdeadcafe; | |
| return S_OK; | |
| } | |
| static HRESULT WINAPI Prop_Write( | |
| IPropertyBag* This, | |
| LPCOLESTR pszPropName, | |
| VARIANT *pVar) | |
| { | |
| return S_OK; | |
| } | |
| static const IPropertyBagVtbl prop_vtbl = { | |
| Prop_QueryInterface, | |
| Prop_AddRef, | |
| Prop_Release, | |
| Prop_Read, | |
| Prop_Write | |
| }; | |
| static void test_SHPropertyBag_ReadLONG(void) | |
| { | |
| PropBag *pb; | |
| HRESULT rc; | |
| LONG out; | |
| static const WCHAR szName1[] = {'n','a','m','e','1',0}; | |
| if (!pSHPropertyBag_ReadLONG) | |
| { | |
| win_skip("SHPropertyBag_ReadLONG not present\n"); | |
| return; | |
| } | |
| pb = HeapAlloc(GetProcessHeap(),0,sizeof(PropBag)); | |
| pb->refCount = 1; | |
| pb->vtbl = &prop_vtbl; | |
| out = 0xfeedface; | |
| rc = pSHPropertyBag_ReadLONG(NULL, szName1, &out); | |
| ok(rc == E_INVALIDARG || broken(rc == 0), "incorrect return %x\n",rc); | |
| ok(out == 0xfeedface, "value should not have changed\n"); | |
| rc = pSHPropertyBag_ReadLONG((IPropertyBag*)pb, NULL, &out); | |
| ok(rc == E_INVALIDARG || broken(rc == 0) || broken(rc == 1), "incorrect return %x\n",rc); | |
| ok(out == 0xfeedface, "value should not have changed\n"); | |
| rc = pSHPropertyBag_ReadLONG((IPropertyBag*)pb, szName1, NULL); | |
| ok(rc == E_INVALIDARG || broken(rc == 0) || broken(rc == 1), "incorrect return %x\n",rc); | |
| ok(out == 0xfeedface, "value should not have changed\n"); | |
| rc = pSHPropertyBag_ReadLONG((IPropertyBag*)pb, szName1, &out); | |
| ok(rc == DISP_E_BADVARTYPE || broken(rc == 0) || broken(rc == 1), "incorrect return %x\n",rc); | |
| ok(out == 0xfeedface || broken(out == 0xfeedfa00), "value should not have changed %x\n",out); | |
| IUnknown_Release((IUnknown*)pb); | |
| } | |
| START_TEST(ordinal) | |
| { | |
| hShlwapi = GetModuleHandleA("shlwapi.dll"); | |
| pGetAcceptLanguagesA = (void*)GetProcAddress(hShlwapi, (LPSTR)14); | |
| pSHSearchMapInt = (void*)GetProcAddress(hShlwapi, (LPSTR)198); | |
| pSHAllocShared=(void*)GetProcAddress(hShlwapi,(char*)7); | |
| pSHLockShared=(void*)GetProcAddress(hShlwapi,(char*)8); | |
| pSHUnlockShared=(void*)GetProcAddress(hShlwapi,(char*)9); | |
| pSHFreeShared=(void*)GetProcAddress(hShlwapi,(char*)10); | |
| pSHPackDispParams=(void*)GetProcAddress(hShlwapi,(char*)282); | |
| pIConnectionPoint_SimpleInvoke=(void*)GetProcAddress(hShlwapi,(char*)284); | |
| pIConnectionPoint_InvokeWithCancel=(void*)GetProcAddress(hShlwapi,(char*)283); | |
| pConnectToConnectionPoint=(void*)GetProcAddress(hShlwapi,(char*)168); | |
| pSHPropertyBag_ReadLONG=(void*)GetProcAddress(hShlwapi,(char*)496); | |
| hmlang = LoadLibraryA("mlang.dll"); | |
| pLcidToRfc1766A = (void *)GetProcAddress(hmlang, "LcidToRfc1766A"); | |
| test_GetAcceptLanguagesA(); | |
| test_SHSearchMapInt(); | |
| test_alloc_shared(); | |
| test_fdsa(); | |
| test_GetShellSecurityDescriptor(); | |
| test_SHPackDispParams(); | |
| test_IConnectionPoint(); | |
| test_SHPropertyBag_ReadLONG(); | |
| } |