PrivateExtractIcons.c File Reference

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

Include dependency graph for PrivateExtractIcons.c:

Go to the source code of this file.

Functions
BOOL	FileExists (PCWSTR FileName)
BOOL	ResourceToFile (INT i, PCWSTR FileName)
	START_TEST (PrivateExtractIcons)

Variables
struct {
PCWSTR   FilePath
UINT   cIcons
UINT   cTotalIcons
BOOL   bhIconValid
}	IconTests []
struct {
PCWSTR   FileName
INT   ResourceId
}	IconFiles []

Function Documentation

FileExists()

BOOL FileExists

(

PCWSTR

FileName

)

Definition at line 12 of file PrivateExtractIcons.c.

{
    DWORD Attribute = GetFileAttributesW(FileName);
 
    return (Attribute != INVALID_FILE_ATTRIBUTES &&
            !(Attribute & FILE_ATTRIBUTE_DIRECTORY));
}

Referenced by ResourceToFile().

ResourceToFile()

BOOL ResourceToFile	(	INT	i,
		PCWSTR	FileName
	)

Definition at line 20 of file PrivateExtractIcons.c.

{
    FILE *fout;
    HGLOBAL hData;
    HRSRC hRes;
    PVOID pResLock;
    UINT iSize;
 
    if (FileExists(FileName))
    {
        /* We should only be using %temp% paths, so deleting here should be OK */
        printf("Deleting '%S' that already exists.\n", FileName);
        DeleteFileW(FileName);
    }
 
    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.
    pResLock = LockResource(hData);
    if (pResLock == NULL)
    {
        skip("Could not lock resource (%d). Exiting now\n", i);
        return FALSE;
    }
 
    fout = _wfopen(FileName, L"wb");
    fwrite(pResLock, iSize, 1, fout);
    fclose(fout);
    return TRUE;
}

Referenced by START_TEST().

START_TEST()

START_TEST

(

PrivateExtractIcons

)

Definition at line 113 of file PrivateExtractIcons.c.

{
    HICON ahIcon;
    UINT i, aIconId, cIcons, cIcoTotal;
    WCHAR PathBuffer[MAX_PATH];
 
    /* Extract icons */
    for (i = 0; i < _countof(IconFiles); ++i)
    {
        ExpandEnvironmentStringsW(IconFiles[i].FileName, PathBuffer, _countof(PathBuffer));
 
        if (!ResourceToFile(IconFiles[i].ResourceId, PathBuffer))
            goto Cleanup;
    }
 
    for (i = 0; i < _countof(IconTests); ++i)
    {
        /* Get total number of icon groups in file.
         * None of the hard numbers in the function matter since we have
         * two NULLs for the Icon Handle and Count to be set. */
        cIcoTotal = PrivateExtractIconsW(IconTests[i].FilePath, 0, 16, 16, NULL, NULL, 0, 0);
        ok((i == 3 ?
              cIcoTotal > 232 && cIcoTotal < 240 :    /* shell32 case: ROS has 233, W2K2SP2 has 239 icon groups. */
              cIcoTotal == IconTests[i].cTotalIcons),
           "PrivateExtractIconsW(%u): "
           "got %u, expected %u\n", i, cIcoTotal, IconTests[i].cTotalIcons);
 
        /* Always test extraction of the FIRST icon (index 0) */
        ahIcon = (HICON)UlongToHandle(0xdeadbeef);
        aIconId = 0xdeadbeef;
        cIcons = PrivateExtractIconsW(IconTests[i].FilePath, 0, 16, 16, &ahIcon, &aIconId, 1, 0);
        ok(cIcons == IconTests[i].cIcons, "PrivateExtractIconsW(%u): got %u, expected %u\n", i, cIcons, IconTests[i].cIcons);
        ok(ahIcon != (HICON)UlongToHandle(0xdeadbeef), "PrivateExtractIconsW(%u): icon not set\n", i);
        ok((IconTests[i].bhIconValid && ahIcon) || (!IconTests[i].bhIconValid && !ahIcon),
            "PrivateExtractIconsW(%u): icon expected to be %s, but got 0x%p\n",
            i, IconTests[i].bhIconValid ? "valid" : "not valid", ahIcon);
        if (cIcons == 0xFFFFFFFF)
        {
            ok(aIconId == 0xdeadbeef,
               "PrivateExtractIconsW(%u): id should not be set to 0x%x\n",
               i, aIconId);
        }
        else
        {
            ok(aIconId != 0xdeadbeef, "PrivateExtractIconsW(%u): id not set\n", i);
        }
        if (ahIcon && ahIcon != (HICON)UlongToHandle(0xdeadbeef))
            DestroyIcon(ahIcon);
    }
 
Cleanup:
    for (i = 0; i < _countof(IconFiles); ++i)
    {
        ExpandEnvironmentStringsW(IconFiles[i].FileName, PathBuffer, _countof(PathBuffer));
        DeleteFileW(PathBuffer);
    }
}

Variable Documentation

bhIconValid

BOOL bhIconValid

Definition at line 70 of file PrivateExtractIcons.c.

Referenced by START_TEST().

cIcons

UINT cIcons

Definition at line 68 of file PrivateExtractIcons.c.

Referenced by START_TEST(), and test_PrivateExtractIcons().

cTotalIcons

UINT cTotalIcons

Definition at line 69 of file PrivateExtractIcons.c.

Referenced by START_TEST().

FileName

PCWSTR FileName

Definition at line 104 of file PrivateExtractIcons.c.

FilePath

PCWSTR FilePath

Definition at line 67 of file PrivateExtractIcons.c.

Referenced by AddBackslash(), BcdOpenStoreFromFile(), BiAddStoreFromFile(), BiLoadHive(), ChangeACLsOfFiles(), ChangeFileACL(), CmpAddDriverToList(), CmpAddToHiveFileList(), CmpCmdHiveOpen(), DosCreateFile(), DosCreateFileEx(), DosOpenFile(), EfiInitpConvertEfiFilePath(), EfiInitpCreateApplicationEntry(), EtfsOpen(), FileIoCopyFileName(), FileIoCopyParentDirectoryPath(), filesys_GetFile(), FileTableCompareWithSameAttributes(), FileTableCompareWithSubsetAttributes(), GetPathOfFile(), load_class(), load_progid(), LoadAndBootWindows(), LoadReactOSSetup(), PeLdrLoadBootImage(), PeLdrLoadImage(), PeLdrLoadImageEx(), PrintDaclsOfFiles(), PrintFileDacl(), ReadBootCodeFromFile(), RxCreateVNetRoot(), START_TEST(), StoreNameInSft(), StrDupShortenPath(), WinLdrAddDriverToList(), and WinLdrLoadDeviceDriver().

struct { ... } IconFiles[]

Initial value:

=
{
    {L"%temp%\\ROS.ico", IDR_ICONS_PNG},
    {L"%temp%\\sysicon.ico", IDR_ICONS_NORMAL},
    {L"%temp%\\cpimg2e.exe", IDR_EXE_NORMAL}
}

Referenced by START_TEST().

struct { ... } IconTests[]

Initial value:

=
{
    
    {L"notepad.exe", 1, 1, TRUE},
    {L"%SystemRoot%\\System32\\cmd.exe", 1, 1, TRUE},
 
    
    {L"%SystemRoot%\\System32\\autochk.exe", 0, 0, FALSE},
 
    
    {L"%SystemRoot%\\System32\\shell32.dll", 1, 233, TRUE},
 
    
    {L"%SystemRoot%\\non-existent-file.sdf", 0xFFFFFFFF, 0, FALSE},
 
    
    {L"%SystemRoot%\\explorer.exe", 1, 18, TRUE},
 
    
    {L"%temp%\\sysicon.ico", 1, 1, TRUE},
 
    
    {L"%temp%\\ROS.ico", 1, 1, TRUE},
 
    
    {L"%temp%\\cpimg2e.exe", 1, 1, TRUE},
}

Referenced by START_TEST().

ResourceId

INT ResourceId

Definition at line 105 of file PrivateExtractIcons.c.

Referenced by START_TEST().