Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2024-12-28 10:04:49 +00:00
Code Issues Projects Releases Packages Wiki Activity

[SHELL32_APITEST] Add test for SHCreateFileExtractIconW

Browse source 
CORE-14082
This commit is contained in:
Mark Jansen 2017-12-09 00:51:56 +01:00
parent 4fcaf92ec3
commit 4ac511405f
5 changed files with 213 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

BIN
modules/rostests/apitests/shell32/1.ico Normal file
View file

Binary file not shown.
Side by side

After

Width:  |  Height:  |  Size: 4.2 KiB

2
modules/rostests/apitests/shell32/CMakeLists.txt
View file

@ -14,10 +14,12 @@ add_executable(shell32_apitest
CShellLink.cpp
menu.cpp
PathResolve.cpp
SHCreateFileExtractIconW.cpp
ShellExecuteEx.cpp
shelltest.cpp
SHParseDisplayName.cpp
testlist.c
resource.rc
${CMAKE_CURRENT_BINARY_DIR}/shell32_apitest.def)
target_link_libraries(shell32_apitest wine uuid ${PSEH_LIB})
set_module_type(shell32_apitest win32cui)

204
modules/rostests/apitests/shell32/SHCreateFileExtractIconW.cpp Normal file
View file

@ -0,0 +1,204 @@
/*
* PROJECT: ReactOS API tests
* LICENSE: GPL-2.0+ (https://spdx.org/licenses/GPL-2.0+)
* PURPOSE: Test for SHCreateFileExtractIconW
* COPYRIGHT: Copyright 2017 Mark Jansen (mark.jansen@reactos.org)
*/
#include "shelltest.h"
#include <atlbase.h>
#include <atlcom.h>
ULONG DbgPrint(PCH Format,...);
#include <shellutils.h>
#include <wincon.h>
#include <wingdi.h>
HRESULT (STDAPICALLTYPE *pSHCreateFileExtractIconW)(LPCWSTR pszFile, DWORD dwFileAttributes, REFIID riid, void **ppv);
struct test_data
{
const WCHAR* Name;
DWORD dwFlags;
};
static test_data Tests[] =
{
{ L"xxx.zip", FILE_ATTRIBUTE_NORMAL },
{ L"xxx.zip", FILE_ATTRIBUTE_DIRECTORY },
{ L"xxx.exe", FILE_ATTRIBUTE_NORMAL },
{ L"xxx.exe", FILE_ATTRIBUTE_DIRECTORY },
{ L"xxx.dll", FILE_ATTRIBUTE_NORMAL },
{ L"xxx.dll", FILE_ATTRIBUTE_DIRECTORY },
{ L"xxx.txt", FILE_ATTRIBUTE_NORMAL },
{ L"xxx.txt", FILE_ATTRIBUTE_DIRECTORY },
{ NULL, FILE_ATTRIBUTE_NORMAL },
{ NULL, FILE_ATTRIBUTE_DIRECTORY },
};
static void ExtractOneBitmap(HBITMAP hbm, CComHeapPtr<BYTE>& data, DWORD& size)
{
HDC hdc = CreateCompatibleDC(NULL);
HGDIOBJ obj = SelectObject(hdc, hbm);
CComHeapPtr<BITMAPINFO> pInfoBM;
pInfoBM.AllocateBytes(sizeof(BITMAPINFOHEADER) + 256 * sizeof(RGBQUAD));
memset(pInfoBM, 0, sizeof(BITMAPINFOHEADER) + 256 * sizeof(RGBQUAD));
pInfoBM->bmiHeader.biSize = sizeof(pInfoBM->bmiHeader);
if (!GetDIBits(hdc, hbm, 0, 0, NULL, pInfoBM, DIB_RGB_COLORS))
return;
size = pInfoBM->bmiHeader.biSizeImage;
data.Allocate(size);
GetDIBits(hdc, hbm, 0, pInfoBM->bmiHeader.biHeight, data, pInfoBM, DIB_RGB_COLORS);
SelectObject(hdc, obj);
DeleteDC(hdc);
}
static bool GetIconData(HICON icon, CComHeapPtr<BYTE>& colorData, DWORD& colorSize, CComHeapPtr<BYTE>& maskData, DWORD& maskSize)
{
ICONINFO iconinfo;
if (!GetIconInfo(icon, &iconinfo))
return false;
ExtractOneBitmap(iconinfo.hbmColor, colorData, colorSize);
ExtractOneBitmap(iconinfo.hbmMask, maskData, maskSize);
DeleteObject(iconinfo.hbmColor);
DeleteObject(iconinfo.hbmMask);
return true;
}
START_TEST(SHCreateFileExtractIconW)
{
WCHAR CurrentModule[MAX_PATH];
HMODULE shell32 = LoadLibraryA("shell32.dll");
HICON myIcon;
pSHCreateFileExtractIconW = (HRESULT (__stdcall *)(LPCWSTR, DWORD, REFIID, void **))GetProcAddress(shell32, "SHCreateFileExtractIconW");
CoInitialize(NULL);
GetModuleFileNameW(NULL, CurrentModule, _countof(CurrentModule));
{
SHFILEINFOW shfi;
ULONG_PTR firet = SHGetFileInfoW(CurrentModule, 0, &shfi, sizeof(shfi), SHGFI_ICON);
myIcon = shfi.hIcon;
if (!firet)
{
skip("Unable to get my own icon\n");
return;
}
}
if (!pSHCreateFileExtractIconW)
{
skip("SHCreateFileExtractIconW not available\n");
return;
}
for (size_t n = 0; n < _countof(Tests); ++n)
{
test_data& cur = Tests[n];
bool useMyIcon = false;
if (cur.Name == NULL)
{
cur.Name = CurrentModule;
useMyIcon = true;
}
CComPtr<IExtractIconW> spExtract;
HRESULT hr = pSHCreateFileExtractIconW(cur.Name, cur.dwFlags, IID_PPV_ARG(IExtractIconW, &spExtract));
ok(hr == S_OK, "Expected hr to be S_OK, was 0x%lx for %S(%lx)\n", hr, cur.Name, cur.dwFlags);
if (!SUCCEEDED(hr))
continue;
int ilIndex = -1;
UINT wFlags = 0xdeaddead;
WCHAR Buffer[MAX_PATH];
hr = spExtract->GetIconLocation(0, Buffer, _countof(Buffer), &ilIndex, &wFlags);
ok(hr == S_OK, "Expected hr to be S_OK, was 0x%lx for %S(%lx)\n", hr, cur.Name, cur.dwFlags);
if (!SUCCEEDED(hr))
continue;
ok(wFlags & (GIL_NOTFILENAME|GIL_PERCLASS), "Expected GIL_NOTFILENAME|GIL_PERCLASS to be set for %S(%lx)\n", cur.Name, cur.dwFlags);
ok(!wcscmp(Buffer, L"*"), "Expected '*', was '%S' for %S(%lx)\n", Buffer, cur.Name, cur.dwFlags);
HICON ico;
hr = spExtract->Extract(Buffer, ilIndex, &ico, NULL, 0);
/* Visualize the icon extracted for whoever is stepping through this code. */
HWND console = GetConsoleWindow();
SendMessage(console, WM_SETICON, ICON_BIG, (LPARAM)ico);
SendMessage(console, WM_SETICON, ICON_SMALL, (LPARAM)ico);
CComHeapPtr<BYTE> colorData, maskData;
DWORD colorSize = 0, maskSize = 0;
GetIconData(ico, colorData, colorSize, maskData, maskSize);
if (!colorSize || !maskSize)
continue;
SHFILEINFOW shfi;
ULONG_PTR firet = SHGetFileInfoW(cur.Name, cur.dwFlags, &shfi, sizeof(shfi), SHGFI_USEFILEATTRIBUTES | SHGFI_ICON);
if (!firet)
continue;
CComHeapPtr<BYTE> colorDataRef, maskDataRef;
DWORD colorSizeRef = 0, maskSizeRef = 0;
GetIconData(shfi.hIcon, colorDataRef, colorSizeRef, maskDataRef, maskSizeRef);
ok(colorSizeRef == colorSize, "Expected %lu, was %lu for %S(%lx)\n", colorSizeRef, colorSize, cur.Name, cur.dwFlags);
ok(maskSizeRef == maskSize, "Expected %lu, was %lu for %S(%lx)\n", maskSizeRef, maskSize, cur.Name, cur.dwFlags);
if (colorSizeRef == colorSize)
{
ok(!memcmp(colorData, colorDataRef, colorSize), "Expected equal colorData for %S(%lx)\n", cur.Name, cur.dwFlags);
}
if (maskSizeRef == maskSize)
{
ok(!memcmp(maskData, maskDataRef, maskSize), "Expected equal maskData for %S(%lx)\n", cur.Name, cur.dwFlags);
}
if (useMyIcon)
{
colorDataRef.Free();
maskDataRef.Free();
colorSizeRef = maskSizeRef = 0;
GetIconData(myIcon, colorDataRef, colorSizeRef, maskDataRef, maskSizeRef);
ok(colorSizeRef == colorSize, "Expected %lu, was %lu for %S(%lx)\n", colorSizeRef, colorSize, cur.Name, cur.dwFlags);
ok(maskSizeRef == maskSize, "Expected %lu, was %lu for %S(%lx)\n", maskSizeRef, maskSize, cur.Name, cur.dwFlags);
if (colorSizeRef == colorSize)
{
/* Incase requested filetype does not match, the exe icon is not used! */
if (cur.dwFlags == FILE_ATTRIBUTE_DIRECTORY)
{
ok(memcmp(colorData, colorDataRef, colorSize), "Expected colorData to be changed for %S(%lx)\n", cur.Name, cur.dwFlags);
}
else
{
ok(!memcmp(colorData, colorDataRef, colorSize), "Expected equal colorData for %S(%lx)\n", cur.Name, cur.dwFlags);
}
}
// Mask is not reliable for some reason
//if (maskSizeRef == maskSize)
//{
// ok(!memcmp(maskData, maskDataRef, maskSize), "Expected equal maskData for %S(%lx)\n", cur.Name, cur.dwFlags);
//}
}
}
}

5
modules/rostests/apitests/shell32/resource.rc Normal file
View file

@ -0,0 +1,5 @@
#include <windef.h>
LANGUAGE LANG_NEUTRAL, SUBLANG_NEUTRAL
1 ICON "1.ico"

2
modules/rostests/apitests/shell32/testlist.c
View file

@ -11,6 +11,7 @@ extern void func_CShellDesktop(void);
extern void func_CShellLink(void);
extern void func_menu(void);
extern void func_PathResolve(void);
extern void func_SHCreateFileExtractIconW(void);
extern void func_ShellExecuteEx(void);
extern void func_SHParseDisplayName(void);
@ -24,6 +25,7 @@ const struct test winetest_testlist[] =
{ "CShellLink", func_CShellLink },
{ "menu", func_menu },
{ "PathResolve", func_PathResolve },
{ "SHCreateFileExtractIconW", func_SHCreateFileExtractIconW },
{ "ShellExecuteEx", func_ShellExecuteEx },
{ "SHParseDisplayName", func_SHParseDisplayName },
{ 0, 0 }