static void test_getstring_bad(void)
{
HRESULT hr;
DWORD len;
if (!pAssocQueryStringW)
{
win_skip("AssocQueryStringW() is missing\n");
return;
}
hr = pAssocQueryStringW(0, ASSOCSTR_EXECUTABLE, NULL, open, NULL, &len);
expect_hr(E_INVALIDARG, hr);
hr = pAssocQueryStringW(0, ASSOCSTR_EXECUTABLE, badBad, open, NULL, &len);
ok(hr == E_FAIL ||
hr == HRESULT_FROM_WIN32(ERROR_NO_ASSOCIATION), /* Win9x/WinMe/NT4/W2K/Vista/W2K8 */
"Unexpected result : %08x\n", hr);
hr = pAssocQueryStringW(0, ASSOCSTR_EXECUTABLE, dotBad, open, NULL, &len);
ok(hr == E_FAIL ||
hr == HRESULT_FROM_WIN32(ERROR_NO_ASSOCIATION), /* Win9x/WinMe/NT4/W2K/Vista/W2K8 */
"Unexpected result : %08x\n", hr);
hr = pAssocQueryStringW(0, ASSOCSTR_EXECUTABLE, dotHtml, invalid, NULL,
&len);
ok(hr == HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND) ||
hr == HRESULT_FROM_WIN32(ERROR_NO_ASSOCIATION), /* Win9x/WinMe/NT4/W2K/Vista/W2K8 */
"Unexpected result : %08x\n", hr);
hr = pAssocQueryStringW(0, ASSOCSTR_EXECUTABLE, dotHtml, open, NULL, NULL);
ok(hr == E_UNEXPECTED ||
hr == E_INVALIDARG, /* Win9x/WinMe/NT4/W2K/Vista/W2K8 */
"Unexpected result : %08x\n", hr);
hr = pAssocQueryStringW(0, ASSOCSTR_FRIENDLYAPPNAME, NULL, open, NULL, &len);
expect_hr(E_INVALIDARG, hr);
hr = pAssocQueryStringW(0, ASSOCSTR_FRIENDLYAPPNAME, badBad, open, NULL,
&len);
ok(hr == E_FAIL ||
hr == HRESULT_FROM_WIN32(ERROR_NO_ASSOCIATION), /* Win9x/WinMe/NT4/W2K/Vista/W2K8 */
"Unexpected result : %08x\n", hr);
hr = pAssocQueryStringW(0, ASSOCSTR_FRIENDLYAPPNAME, dotBad, open, NULL,
&len);
ok(hr == E_FAIL ||
hr == HRESULT_FROM_WIN32(ERROR_NO_ASSOCIATION) /* Win9x/WinMe/NT4/W2K/Vista/W2K8 */ ||
hr == HRESULT_FROM_WIN32(ERROR_NOT_FOUND), /* Win8 */
"Unexpected result : %08x\n", hr);
hr = pAssocQueryStringW(0, ASSOCSTR_FRIENDLYAPPNAME, dotHtml, invalid, NULL,
&len);
ok(hr == HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND) ||
hr == HRESULT_FROM_WIN32(ERROR_NO_ASSOCIATION) || /* W2K/Vista/W2K8 */
hr == E_FAIL, /* Win9x/WinMe/NT4 */
"Unexpected result : %08x\n", hr);
hr = pAssocQueryStringW(0, ASSOCSTR_FRIENDLYAPPNAME, dotHtml, open, NULL,
NULL);
ok(hr == E_UNEXPECTED ||
hr == E_INVALIDARG, /* Win9x/WinMe/NT4/W2K/Vista/W2K8 */
"Unexpected result : %08x\n", hr);
}
static void test_assoc_one(const struct assoc_test_struct* test)
{
LPWSTR extension, type_expected, type_returned;
PERCEIVED perceived;
HRESULT hr;
INT flags;
/* if SHStrDupA receives a nullptr as input, it will null the output */
SHStrDupA(test->extension, &extension);
SHStrDupA(test->type, &type_expected);
perceived = 0xdeadbeef;
flags = 0xdeadbeef;
hr = pAssocGetPerceivedType(extension, &perceived, &flags, NULL);
expect_hr(type_expected ? S_OK : test->hr, hr);
ok(perceived == test->perceived, "%s: got perceived 0x%x, expected 0x%x\n",
test->extension, perceived, test->perceived);
ok(flags == test->flags, "%s: got flags 0x%x, expected 0x%x\n",
test->extension, flags, test->flags);
type_returned = (void *)0xdeadbeef;
perceived = 0xdeadbeef;
flags = 0xdeadbeef;
hr = pAssocGetPerceivedType(extension, &perceived, &flags, &type_returned);
expect_hr(type_expected ? S_OK : test->hr, hr);
ok(perceived == test->perceived, "%s: got perceived 0x%x, expected 0x%x\n",
test->extension, perceived, test->perceived);
ok(flags == test->flags, "%s: got flags 0x%x, expected 0x%x\n",
test->extension, flags, test->flags);
if (!type_expected)
{
ok(type_returned == (void *)0xdeadbeef || broken(type_returned == NULL) /* Win 8 */,
"%s: got type %p, expected 0xdeadbeef\n", test->extension, type_returned);
}
else if (type_returned == (void *)0xdeadbeef)
{
ok(type_returned != (void *)0xdeadbeef, "%s: got type %p, expected '%s'\n",
test->extension, type_returned, test->type);
}
else
{
ok(StrCmpIW(type_expected, type_returned) == 0, "%s: got type %s, expected '%s'\n",
test->extension, wine_dbgstr_w(type_returned), test->type);
}
CoTaskMemFree(type_returned);
CoTaskMemFree(extension);
CoTaskMemFree(type_expected);
}
static void test_getstring_bad(void)
{
static const WCHAR openwith[] = {'O','p','e','n','W','i','t','h','.','e','x','e',0};
WCHAR buf[MAX_PATH];
HRESULT hr;
DWORD len;
if (!pAssocQueryStringW)
{
win_skip("AssocQueryStringW() is missing\n");
return;
}
len = 0xdeadbeef;
hr = pAssocQueryStringW(0, ASSOCSTR_EXECUTABLE, NULL, open, NULL, &len);
expect_hr(E_INVALIDARG, hr);
ok(len == 0xdeadbeef, "got %u\n", len);
len = 0xdeadbeef;
hr = pAssocQueryStringW(0, ASSOCSTR_EXECUTABLE, badBad, open, NULL, &len);
ok(hr == E_FAIL ||
hr == HRESULT_FROM_WIN32(ERROR_NO_ASSOCIATION), /* Win9x/WinMe/NT4/W2K/Vista/W2K8 */
"Unexpected result : %08x\n", hr);
ok(len == 0xdeadbeef, "got %u\n", len);
len = sizeof(buf)/sizeof(buf[0]);
hr = pAssocQueryStringW(0, ASSOCSTR_EXECUTABLE, dotBad, open, buf, &len);
ok(hr == E_FAIL ||
hr == HRESULT_FROM_WIN32(ERROR_NO_ASSOCIATION) /* Win9x/WinMe/NT4/W2K/Vista/W2K8 */ ||
hr == S_OK /* Win8 */,
"Unexpected result : %08x\n", hr);
if (hr == S_OK)
{
ok(len < sizeof(buf)/sizeof(buf[0]), "got %u\n", len);
ok(!lstrcmpiW(buf + len - sizeof(openwith)/sizeof(openwith[0]), openwith), "wrong data\n");
}
len = 0xdeadbeef;
hr = pAssocQueryStringW(0, ASSOCSTR_EXECUTABLE, dotHtml, invalid, NULL, &len);
ok(hr == HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND) ||
hr == HRESULT_FROM_WIN32(ERROR_NO_ASSOCIATION), /* Win9x/WinMe/NT4/W2K/Vista/W2K8 */
"Unexpected result : %08x\n", hr);
ok(len == 0xdeadbeef, "got %u\n", len);
hr = pAssocQueryStringW(0, ASSOCSTR_EXECUTABLE, dotHtml, open, NULL, NULL);
ok(hr == E_UNEXPECTED ||
hr == E_INVALIDARG, /* Win9x/WinMe/NT4/W2K/Vista/W2K8 */
"Unexpected result : %08x\n", hr);
len = 0xdeadbeef;
hr = pAssocQueryStringW(0, ASSOCSTR_FRIENDLYAPPNAME, NULL, open, NULL, &len);
expect_hr(E_INVALIDARG, hr);
ok(len == 0xdeadbeef, "got %u\n", len);
len = 0xdeadbeef;
hr = pAssocQueryStringW(0, ASSOCSTR_FRIENDLYAPPNAME, badBad, open, NULL, &len);
ok(hr == E_FAIL ||
hr == HRESULT_FROM_WIN32(ERROR_NO_ASSOCIATION), /* Win9x/WinMe/NT4/W2K/Vista/W2K8 */
"Unexpected result : %08x\n", hr);
ok(len == 0xdeadbeef, "got %u\n", len);
len = 0xdeadbeef;
hr = pAssocQueryStringW(0, ASSOCSTR_FRIENDLYAPPNAME, dotBad, open, NULL, &len);
ok(hr == E_FAIL ||
hr == HRESULT_FROM_WIN32(ERROR_NO_ASSOCIATION) /* Win9x/WinMe/NT4/W2K/Vista/W2K8 */ ||
hr == HRESULT_FROM_WIN32(ERROR_NOT_FOUND), /* Win8 */
"Unexpected result : %08x\n", hr);
ok(len == 0xdeadbeef, "got %u\n", len);
len = 0xdeadbeef;
hr = pAssocQueryStringW(0, ASSOCSTR_FRIENDLYAPPNAME, dotHtml, invalid, NULL, &len);
ok(hr == HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND) ||
hr == HRESULT_FROM_WIN32(ERROR_NO_ASSOCIATION) || /* W2K/Vista/W2K8 */
hr == E_FAIL, /* Win9x/WinMe/NT4 */
"Unexpected result : %08x\n", hr);
ok(len == 0xdeadbeef, "got %u\n", len);
hr = pAssocQueryStringW(0, ASSOCSTR_FRIENDLYAPPNAME, dotHtml, open, NULL, NULL);
ok(hr == E_UNEXPECTED ||
hr == E_INVALIDARG, /* Win9x/WinMe/NT4/W2K/Vista/W2K8 */
"Unexpected result : %08x\n", hr);
}
static void test_getstring_no_extra(void)
{
LONG ret;
HKEY hkey;
HRESULT hr;
static const CHAR dotWinetest[] = {
'.','w','i','n','e','t','e','s','t',0
};
static const CHAR winetestfile[] = {
'w','i','n','e','t','e','s','t', 'f','i','l','e',0
};
static const CHAR winetestfileAction[] = {
'w','i','n','e','t','e','s','t','f','i','l','e',
'\\','s','h','e','l','l',
'\\','f','o','o',
'\\','c','o','m','m','a','n','d',0
};
static const CHAR action[] = {
'n','o','t','e','p','a','d','.','e','x','e',0
};
CHAR buf[MAX_PATH];
DWORD len = MAX_PATH;
if (!pAssocQueryStringA)
{
win_skip("AssocQueryStringA() is missing\n");
return;
}
buf[0] = '\0';
ret = RegCreateKeyA(HKEY_CLASSES_ROOT, dotWinetest, &hkey);
if (ret != ERROR_SUCCESS) {
skip("failed to create dotWinetest key\n");
return;
}
ret = RegSetValueA(hkey, NULL, REG_SZ, winetestfile, lstrlenA(winetestfile));
RegCloseKey(hkey);
if (ret != ERROR_SUCCESS)
{
skip("failed to set dotWinetest key\n");
goto cleanup;
}
ret = RegCreateKeyA(HKEY_CLASSES_ROOT, winetestfileAction, &hkey);
if (ret != ERROR_SUCCESS)
{
skip("failed to create winetestfileAction key\n");
goto cleanup;
}
ret = RegSetValueA(hkey, NULL, REG_SZ, action, lstrlenA(action));
RegCloseKey(hkey);
if (ret != ERROR_SUCCESS)
{
skip("failed to set winetestfileAction key\n");
goto cleanup;
}
hr = pAssocQueryStringA(0, ASSOCSTR_EXECUTABLE, dotWinetest, NULL, buf, &len);
ok(hr == S_OK ||
hr == HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND), /* XP and W2K3 */
"Unexpected result : %08x\n", hr);
hr = pAssocQueryStringA(0, ASSOCSTR_EXECUTABLE, dotWinetest, "foo", buf, &len);
expect_hr(S_OK, hr);
ok(strstr(buf, action) != NULL,
"got '%s' (Expected result to include 'notepad.exe')\n", buf);
cleanup:
SHDeleteKeyA(HKEY_CLASSES_ROOT, dotWinetest);
SHDeleteKeyA(HKEY_CLASSES_ROOT, winetestfile);
}
static void test_getstring_basic(void)
{
HRESULT hr;
WCHAR * friendlyName;
WCHAR * executableName;
DWORD len, len2, slen;
if (!pAssocQueryStringW)
{
win_skip("AssocQueryStringW() is missing\n");
return;
}
hr = pAssocQueryStringW(0, ASSOCSTR_EXECUTABLE, dotHtml, open, NULL, &len);
expect_hr(S_FALSE, hr);
if (hr != S_FALSE)
{
skip("failed to get initial len\n");
return;
}
executableName = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY,
len * sizeof(WCHAR));
if (!executableName)
{
skip("failed to allocate memory\n");
return;
}
len2 = len;
hr = pAssocQueryStringW(0, ASSOCSTR_EXECUTABLE, dotHtml, open,
executableName, &len2);
expect_hr(S_OK, hr);
slen = lstrlenW(executableName) + 1;
expect(len, len2);
expect(len, slen);
hr = pAssocQueryStringW(0, ASSOCSTR_FRIENDLYAPPNAME, dotHtml, open, NULL,
&len);
ok(hr == S_FALSE ||
hr == HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND) /* Win9x/NT4 */ ||
hr == HRESULT_FROM_WIN32(ERROR_NOT_FOUND), /* Win8 */
"Unexpected result : %08x\n", hr);
if (hr != S_FALSE)
{
HeapFree(GetProcessHeap(), 0, executableName);
skip("failed to get initial len\n");
return;
}
friendlyName = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY,
len * sizeof(WCHAR));
if (!friendlyName)
{
HeapFree(GetProcessHeap(), 0, executableName);
skip("failed to allocate memory\n");
return;
}
len2 = len;
hr = pAssocQueryStringW(0, ASSOCSTR_FRIENDLYAPPNAME, dotHtml, open,
friendlyName, &len2);
expect_hr(S_OK, hr);
slen = lstrlenW(friendlyName) + 1;
expect(len, len2);
expect(len, slen);
HeapFree(GetProcessHeap(), 0, executableName);
HeapFree(GetProcessHeap(), 0, friendlyName);
}
