mirror of
https://github.com/reactos/reactos.git
synced 2024-09-06 02:37:06 +00:00
1428 lines
44 KiB
C
Executable file
1428 lines
44 KiB
C
Executable file
/* 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();
|
|
}
|