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); }