ExtractIconEx.cpp File Reference

#include "shelltest.h"
#include <stdio.h>

Include dependency graph for ExtractIconEx.cpp:

Go to the source code of this file.

Classes
struct	EXTRACTICONTESTS

Functions
EXTERN_C BOOL WINAPI	SHAreIconsEqual (HICON hIcon1, HICON hIcon2)
static void	SafeDestroyIcon (HICON hIco)
static UINT	GetIcoSize (HICON hIco)
BOOL	FileExists (LPCSTR FileName)
BOOL	ResourceToFile (INT i, LPCSTR FileName)
	START_TEST (ExtractIconEx)
static HRESULT	SHDEI (LPCWSTR pszIconFile, int Index=0, UINT GIL=0, UINT Size=0)
	START_TEST (SHDefExtractIcon)

Variables
EXTRACTICONTESTS	IconTests []

Function Documentation

FileExists()

BOOL FileExists

(

LPCSTR

FileName

)

Definition at line 40 of file ExtractIconEx.cpp.

{
    FILE *fp = NULL;
    bool exists = FALSE;
 
    fp = fopen(FileName, "r");
    if (fp != NULL)
    {
        exists = TRUE;
        fclose(fp);
    }
    return exists;
}

Referenced by ResourceToFile().

GetIcoSize()

static UINT GetIcoSize ( HICON  hIco )

static

Definition at line 20 of file ExtractIconEx.cpp.

{
    ICONINFO info;
    if (!GetIconInfo(hIco, &info))
        return 0;
 
    BITMAP bm;
    if (!GetObject(info.hbmColor ? info.hbmColor : info.hbmMask, sizeof(bm), &bm))
        bm.bmWidth = 0;
    DeleteObject(info.hbmMask);
    DeleteObject(info.hbmColor);
    return bm.bmWidth;
}

Referenced by SHDEI().

ResourceToFile()

BOOL ResourceToFile	(	INT	i,
		LPCSTR	FileName
	)

Definition at line 54 of file ExtractIconEx.cpp.

{
    FILE *fout;
    HGLOBAL hData;
    HRSRC hRes;
    LPVOID lpResLock;
    UINT iSize;
 
    if (FileExists(FileName))
    {
        skip("'%s' already exists. Exiting now\n", FileName);
        return FALSE;
    }
 
    hRes = FindResourceW(NULL, MAKEINTRESOURCEW(i), MAKEINTRESOURCEW(RT_RCDATA));
    if (hRes == NULL)
    {
        skip("Could not locate resource (%d). Exiting now\n", i);
        return FALSE;
    }
 
    iSize = SizeofResource(NULL, hRes);
 
    hData = LoadResource(NULL, hRes);
    if (hData == NULL)
    {
        skip("Could not load resource (%d). Exiting now\n", i);
        return FALSE;
    }
 
    // Lock the resource into global memory.
    lpResLock = LockResource(hData);
    if (lpResLock == NULL)
    {
        skip("Could not lock resource (%d). Exiting now\n", i);
        return FALSE;
    }
 
    fout = fopen(FileName, "wb");
    fwrite(lpResLock, iSize, 1, fout);
    fclose(fout);
    return TRUE;
}

Referenced by START_TEST().

SafeDestroyIcon()

static void SafeDestroyIcon ( HICON  hIco )

static

Definition at line 14 of file ExtractIconEx.cpp.

{
    if (hIco)
        DestroyIcon(hIco);
}

Referenced by SHDEI(), and START_TEST().

SHAreIconsEqual()

EXTERN_C BOOL WINAPI SHAreIconsEqual	(	HICON	hIcon1,
		HICON	hIcon2
	)

Definition at line 43 of file rosordinal.c.

{
    ICONINFO iconInfo1, iconInfo2;
    BITMAP bm1, bm2;
    BOOL bSame = FALSE;
 
    if (!hIcon1 || !hIcon2)
        return FALSE;
 
    if (!GetIconInfo(hIcon1, &iconInfo1))
        return FALSE;
 
    if (!GetIconInfo(hIcon2, &iconInfo2))
    {
        DeleteObject(iconInfo1.hbmColor);
        DeleteObject(iconInfo1.hbmMask);
        return FALSE;
    }
 
    GetObjectW(iconInfo1.hbmColor, sizeof(bm1), &bm1);
    GetObjectW(iconInfo2.hbmColor, sizeof(bm2), &bm2);
 
    if (bm1.bmWidth == bm2.bmWidth && bm1.bmHeight == bm2.bmHeight)
    {
        BITMAPINFO bmi = { { sizeof(bmi), bm1.bmWidth, bm1.bmHeight, 1, 32 } };
        HDC hdc = GetDC(0);
        SIZE_T size = bm1.bmWidth * bm1.bmHeight * 4;
        BYTE *data1, *data2;
 
        data1 = HeapAlloc(GetProcessHeap(), 0, size);
        data2 = HeapAlloc(GetProcessHeap(), 0, size);
 
        if (data1 && data2)
        {
            if (GetDIBits(hdc, iconInfo1.hbmColor, 0, bm1.bmHeight, data1, &bmi, DIB_RGB_COLORS) &&
                GetDIBits(hdc, iconInfo2.hbmColor, 0, bm2.bmHeight, data2, &bmi, DIB_RGB_COLORS))
            {
                bSame = memcmp(data1, data2, size) == 0;
            }
        }
        HeapFree(GetProcessHeap(), 0, data1);
        HeapFree(GetProcessHeap(), 0, data2);
 
        ReleaseDC(NULL, hdc);
    }
 
    DeleteObject(iconInfo1.hbmColor);
    DeleteObject(iconInfo1.hbmMask);
 
    DeleteObject(iconInfo2.hbmColor);
    DeleteObject(iconInfo2.hbmMask);
 
    return bSame;
}

Referenced by START_TEST().

SHDEI()

static HRESULT SHDEI ( LPCWSTR  pszIconFile, int  Index = 0, UINT  GIL = 0, UINT  Size = 0  )

static

Definition at line 148 of file ExtractIconEx.cpp.

{
    HICON hIco = NULL;
    HRESULT hr = SHDefExtractIcon(pszIconFile, Index, GIL, &hIco, NULL, Size);
    if (hr == S_OK)
    {
        hr = GetIcoSize(hIco);
        SafeDestroyIcon(hIco);
    }
    return hr;
}

Referenced by START_TEST().

START_TEST() [1/2]

START_TEST

(

ExtractIconEx

)

Definition at line 120 of file ExtractIconEx.cpp.

{
    UINT i, nReturnedIcons, nExtractedIcons;
    CHAR FileName[2][13] = { "ROS.ico", "sysicon.ico" };
 
    if (!ResourceToFile(2, FileName[0]))
        return;
    if (!ResourceToFile(3, FileName[1]))
        return;
 
    /* Check count of icons returned */
    for (i = 0; i < _countof(IconTests); ++i)
    {
        nReturnedIcons = ExtractIconExW(IconTests[i].pszFilePath, -1, NULL, NULL, 0);
        ok(nReturnedIcons == IconTests[i].nIcons, "ExtractIconExW(%u): Expects %u icons, got %u\n", i, IconTests[i].nIcons, nReturnedIcons);
    }
 
    /* Check if the 0th icon can be extracted successfully */
    for (i = 0; i < _countof(IconTests); ++i)
    {
        nExtractedIcons = ExtractIconExW(IconTests[i].pszFilePath, 0, NULL, NULL, 1);
        ok(nExtractedIcons == IconTests[i].nIcons, "ExtractIconExW(%u): Expects %u icons, got %u\n", i, IconTests[i].nIcons, nExtractedIcons);
    }
 
    DeleteFileA(FileName[0]);
    DeleteFileA(FileName[1]);
}

START_TEST() [2/2]

START_TEST

(

SHDefExtractIcon

)

Definition at line 160 of file ExtractIconEx.cpp.

{
    HRESULT hr;
    int SysBigIconSize = GetSystemMetrics(SM_CXICON);
 
    // Modern Windows requires the system image list to be initialized for GIL_SIMULATEDOC to work!
    SHFILEINFOW shfi;
    SHGetFileInfoW(L"x", 0, &shfi, sizeof(shfi), SHGFI_SYSICONINDEX | SHGFI_USEFILEATTRIBUTES);
 
    WCHAR path[MAX_PATH];
    GetSystemDirectoryW(path, _countof(path));
    PathAppendW(path, L"user32.dll");
    int index = 1;
 
    ok(SHDEI(path, index, 0, 0) == SysBigIconSize, "0 size must match GetSystemMetrics\n");
    ok(SHDEI(path, index, 0, SysBigIconSize * 2) == SysBigIconSize * 2, "Resize failed\n");
 
    HICON hIcoLarge, hIcoSmall;
    if (SHDefExtractIcon(path, index, 0, &hIcoLarge, &hIcoSmall, 0) != S_OK)
        hIcoLarge = hIcoSmall = NULL;
    ok(hIcoLarge && hIcoSmall && !SHAreIconsEqual(hIcoLarge, hIcoSmall), "Large+Small failed\n");
    SafeDestroyIcon(hIcoLarge);
    SafeDestroyIcon(hIcoSmall);
 
    static const int sizes[] = { 0, SysBigIconSize * 2 };
    for (UINT i = 0; i < _countof(sizes); ++i)
    {
        HICON hIcoNormal, hIcoSimDoc;
        if (FAILED(hr = SHDefExtractIcon(path, index, 0, &hIcoNormal, NULL, sizes[i])))
            hIcoNormal = NULL;
        if (FAILED(hr = SHDefExtractIcon(path, index, GIL_SIMULATEDOC, &hIcoSimDoc, NULL, sizes[i])))
            hIcoSimDoc = NULL;
        ok(hIcoNormal && hIcoSimDoc && !SHAreIconsEqual(hIcoNormal, hIcoSimDoc), "GIL_SIMULATEDOC failed\n");
        SafeDestroyIcon(hIcoNormal);
        SafeDestroyIcon(hIcoSimDoc);
    }
 
    GetTempPathW(_countof(path), path);
    GetTempFileNameW(path, L"TEST", 0, path);
    ok(SHDEI(path) == S_FALSE, "Empty file should return S_FALSE\n");
    HANDLE hFile = CreateFileW(path, GENERIC_WRITE, FILE_SHARE_READ | FILE_SHARE_DELETE, NULL, OPEN_EXISTING, 0, NULL);
    if (hFile != INVALID_HANDLE_VALUE)
    {
        DWORD io;
        WriteFile(hFile, "!", 1, &io, NULL);
        CloseHandle(hFile);
        ok(SHDEI(path) == S_FALSE, "File without icons should return S_FALSE\n");
    }
    DeleteFile(path);
}

Variable Documentation

IconTests

EXTRACTICONTESTS IconTests[]

Initial value:

=
{
    
    {L"%SystemRoot%\\System32\\cmd.exe", 1},
 
    
    {L"%SystemRoot%\\System32\\autochk.exe", 0},
 
    
    {L"%SystemRoot%\\non-existent-file.sdf", 0},
 
    
    {L"%SystemRoot%\\explorer.exe", 18},
 
    
    {L"sysicon.ico", 1},
 
    
    {L"ROS.ico", 0}
}

Definition at line 98 of file ExtractIconEx.cpp.

Referenced by START_TEST().