resource.c File Reference

#include <windows.h>
#include <stdio.h>
#include "wine/test.h"

Include dependency graph for resource.c:

Go to the source code of this file.

Classes
struct	_sec_build
struct	_sec_verify
struct	_sec_variants

Typedefs
typedef struct _sec_build	sec_build
typedef struct _sec_verify	sec_verify

Enumerations
enum	constants { page_size = 0x1000 , rva_rsrc_start = page_size * 3 , max_sections = 3 }

Functions
static int	build_exe (const sec_build *sec_descr)
static void	update_missing_exe (void)
static void	update_empty_exe (void)
static void	update_resources_none (void)
static void	update_resources_delete (void)
static void	update_resources_version (void)
static void	update_resources_bigdata (void)
static void	check_exe (const sec_verify *verify)
static void	test_find_resource (void)
	START_TEST (resource)

Variables
static const char	filename [] = "test_.exe"
static DWORD	GLE
static const IMAGE_SECTION_HEADER	sh_rodata_1
static const IMAGE_SECTION_HEADER	sh_rodata_2
static const IMAGE_SECTION_HEADER	sh_rsrc_1
static const IMAGE_SECTION_HEADER	sh_rsrc_2
static const IMAGE_SECTION_HEADER	sh_rsrc_3
static const IMAGE_SECTION_HEADER	sh_rsrc_4
static const IMAGE_SECTION_HEADER	sh_rsrc_5
static const IMAGE_SECTION_HEADER	sh_rsrc_6
static const IMAGE_SECTION_HEADER	sh_junk
static const IMAGE_SECTION_HEADER	sh_junk_2
static const struct _sec_variants	sec_variants []

Typedef Documentation

sec_build

typedef struct _sec_build sec_build

sec_verify

typedef struct _sec_verify sec_verify

Enumeration Type Documentation

constants

enum constants

Enumerator
page_size
rva_rsrc_start
max_sections

Definition at line 29 of file resource.c.

               {
    page_size = 0x1000,
    rva_rsrc_start = page_size * 3,
    max_sections = 3
};

Function Documentation

build_exe()

static int build_exe ( const sec_build *  sec_descr )

static

Definition at line 161 of file resource.c.

{
    IMAGE_DOS_HEADER *dos;
    IMAGE_NT_HEADERS *nt;
    IMAGE_SECTION_HEADER *sec;
    IMAGE_OPTIONAL_HEADER *opt;
    HANDLE file;
    DWORD written, i, file_size;
    BYTE page[page_size];
 
    memset( page, 0, sizeof page );
 
    dos = (void*) page;
    dos->e_magic = IMAGE_DOS_SIGNATURE;
    dos->e_lfanew = sizeof *dos;
 
    nt = (void*) &dos[1];
 
    nt->Signature = IMAGE_NT_SIGNATURE;
    nt->FileHeader.Machine = IMAGE_FILE_MACHINE_I386;
    nt->FileHeader.NumberOfSections = 0;
    nt->FileHeader.SizeOfOptionalHeader = sizeof nt->OptionalHeader;
    nt->FileHeader.Characteristics = IMAGE_FILE_EXECUTABLE_IMAGE | IMAGE_FILE_DLL;
 
    opt = &nt->OptionalHeader;
 
    opt->Magic = IMAGE_NT_OPTIONAL_HDR_MAGIC;
    opt->MajorLinkerVersion = 1;
    opt->BaseOfCode = 0x10;
    opt->ImageBase = 0x10000000;
    opt->MajorOperatingSystemVersion = 4;
    opt->MajorImageVersion = 1;
    opt->MajorSubsystemVersion = 4;
    opt->SizeOfHeaders = sizeof *dos + sizeof *nt + sizeof *sec * 2;
    opt->SizeOfImage = page_size;
    opt->Subsystem = IMAGE_SUBSYSTEM_WINDOWS_CUI;
 
    /* if SectionAlignment and File alignment are not specified */
    /* UpdateResource fails trying to create a huge temporary file */
    opt->SectionAlignment = page_size;
    opt->FileAlignment = page_size;
 
    opt->DataDirectory[IMAGE_FILE_RESOURCE_DIRECTORY].VirtualAddress = rva_rsrc_start;
    opt->DataDirectory[IMAGE_FILE_RESOURCE_DIRECTORY].Size = page_size;
 
    sec = (void*) &nt[1];
 
    file_size = 0;
    for ( i = 0; i < max_sections; i++ )
        if ( sec_descr->sect_in[i] )
        {
            DWORD virt_end_of_section = sec_descr->sect_in[i]->Misc.VirtualSize +
                sec_descr->sect_in[i]->VirtualAddress;
            DWORD phys_end_of_section = sec_descr->sect_in[i]->SizeOfRawData +
                sec_descr->sect_in[i]->PointerToRawData;
            memcpy( sec + nt->FileHeader.NumberOfSections, sec_descr->sect_in[i],
                    sizeof(sec[0]) );
            nt->FileHeader.NumberOfSections++;
            if ( opt->SizeOfImage < virt_end_of_section )
                opt->SizeOfImage = virt_end_of_section;
            if ( file_size < phys_end_of_section )
                file_size = phys_end_of_section;
        }
 
    file = CreateFileA(filename, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, 0, 0);
    ok (file != INVALID_HANDLE_VALUE, "failed to create file\n");
 
    /* write out the header */
    WriteFile( file, page, sizeof page, &written, NULL );
 
    /* write out zeroed pages for sections */
    memset( page, 0, sizeof page );
    for ( i = page_size; i < file_size; i += page_size )
    {
    DWORD size = min(page_size, file_size - i);
        WriteFile( file, page, size, &written, NULL );
    }
 
    CloseHandle( file );
 
    return 0;
}

Referenced by START_TEST().

check_exe()

static void check_exe ( const sec_verify *  verify )

static

Definition at line 359 of file resource.c.

{
    int i;
    IMAGE_DOS_HEADER *dos;
    IMAGE_NT_HEADERS *nt;
    IMAGE_OPTIONAL_HEADER *opt;
    IMAGE_SECTION_HEADER *sec;
    IMAGE_RESOURCE_DIRECTORY *dir;
    HANDLE file, mapping;
    DWORD length, sec_count = 0;
 
    file = CreateFileA(filename, GENERIC_READ, 0, NULL, OPEN_EXISTING, 0, 0);
    ok (file != INVALID_HANDLE_VALUE, "failed to create file (%d)\n", GetLastError());
 
    length = GetFileSize( file, NULL );
    ok( length >= verify->length, "file size wrong\n");
 
    mapping = CreateFileMappingA( file, NULL, PAGE_READONLY, 0, 0, NULL );
    ok (mapping != NULL, "failed to create file\n");
 
    dos = MapViewOfFile( mapping, FILE_MAP_READ, 0, 0, length );
    ok( dos != NULL, "failed to map file\n");
 
    if (!dos)
        goto end;
 
    nt = (void*) ((BYTE*) dos + dos->e_lfanew);
    opt = &nt->OptionalHeader;
    sec = (void*) &nt[1];
 
    for(i = 0; i < max_sections; i++)
        if (verify->sect_out[i])
        {
            ok( !memcmp(&verify->sect_out[i]->Name, &sec[sec_count].Name, 8), "section %d name wrong\n", sec_count);
            ok( verify->sect_out[i]->VirtualAddress == sec[sec_count].VirtualAddress, "section %d vaddr wrong\n", sec_count);
            ok( verify->sect_out[i]->SizeOfRawData <= sec[sec_count].SizeOfRawData, "section %d SizeOfRawData wrong (%d vs %d)\n", sec_count, verify->sect_out[i]->SizeOfRawData ,sec[sec_count].SizeOfRawData);
            ok( verify->sect_out[i]->PointerToRawData == sec[sec_count].PointerToRawData, "section %d PointerToRawData wrong\n", sec_count);
            ok( verify->sect_out[i]->Characteristics == sec[sec_count].Characteristics , "section %d characteristics wrong\n", sec_count);
            sec_count++;
        }
 
    ok( nt->FileHeader.NumberOfSections == sec_count, "number of sections wrong\n" );
 
    if (verify->rsrc_section >= 0 && verify->rsrc_section < nt->FileHeader.NumberOfSections)
    {
        dir = (void*) ((BYTE*) dos + sec[verify->rsrc_section].VirtualAddress);
 
        ok( dir->Characteristics == 0, "Characteristics wrong\n");
        ok( dir->TimeDateStamp == 0 || abs( dir->TimeDateStamp - GetTickCount() ) < 1000 /* nt4 */,
            "TimeDateStamp wrong %u\n", dir->TimeDateStamp);
        ok( dir->MajorVersion == 4, "MajorVersion wrong\n");
        ok( dir->MinorVersion == 0, "MinorVersion wrong\n");
 
        ok( dir->NumberOfNamedEntries == verify->NumberOfNamedEntries, "NumberOfNamedEntries should be %d instead of %d\n",
                verify->NumberOfNamedEntries, dir->NumberOfNamedEntries);
        ok( dir->NumberOfIdEntries == verify->NumberOfIdEntries, "NumberOfIdEntries should be %d instead of %d\n",
                verify->NumberOfIdEntries, dir->NumberOfIdEntries);
 
        ok(opt->DataDirectory[IMAGE_FILE_RESOURCE_DIRECTORY].VirtualAddress == sec[verify->rsrc_section].VirtualAddress,
                "VirtualAddress in optional header should be %d instead of %d\n",
                sec[verify->rsrc_section].VirtualAddress, opt->DataDirectory[IMAGE_FILE_RESOURCE_DIRECTORY].VirtualAddress);
    }
 
end:
    UnmapViewOfFile( dos );
 
    CloseHandle( mapping );
 
    CloseHandle( file );
}

Referenced by START_TEST().

START_TEST()

START_TEST

(

resource

)

Definition at line 466 of file resource.c.

{
    DWORD i;
 
    DeleteFileA( filename );
    update_missing_exe();
 
    if (GLE == ERROR_CALL_NOT_IMPLEMENTED)
    {
        win_skip("Resource calls are not implemented\n");
        return;
    }
 
    update_empty_exe();
 
    for(i=0; i < sizeof( sec_variants ) / sizeof( sec_variants[0] ); i++)
    {
        const struct _sec_variants *sec = &sec_variants[i];
        build_exe( &sec->build );
        update_resources_none();
        check_exe( &sec->chk_none );
        update_resources_delete();
        check_exe( &sec->chk_delete );
        update_resources_version();
        check_exe( &sec->chk_version );
        update_resources_bigdata();
        check_exe( &sec->chk_bigdata );
        DeleteFileA( filename );
    }
    test_find_resource();
}

test_find_resource()

static void test_find_resource ( void  )

static

Definition at line 430 of file resource.c.

{
    HRSRC rsrc;
 
    rsrc = FindResourceW( GetModuleHandleW(NULL), MAKEINTRESOURCEW(1), (LPCWSTR)RT_MENU );
    ok( rsrc != 0, "resource not found\n" );
    rsrc = FindResourceExW( GetModuleHandleW(NULL), (LPCWSTR)RT_MENU, MAKEINTRESOURCEW(1),
                            MAKELANGID( LANG_NEUTRAL, SUBLANG_NEUTRAL ));
    ok( rsrc != 0, "resource not found\n" );
    rsrc = FindResourceExW( GetModuleHandleW(NULL), (LPCWSTR)RT_MENU, MAKEINTRESOURCEW(1),
                            MAKELANGID( LANG_GERMAN, SUBLANG_DEFAULT ));
    ok( rsrc != 0, "resource not found\n" );
 
    SetLastError( 0xdeadbeef );
    rsrc = FindResourceW( GetModuleHandleW(NULL), MAKEINTRESOURCEW(1), (LPCWSTR)RT_DIALOG );
    ok( !rsrc, "resource found\n" );
    ok( GetLastError() == ERROR_RESOURCE_TYPE_NOT_FOUND, "wrong error %u\n", GetLastError() );
 
    SetLastError( 0xdeadbeef );
    rsrc = FindResourceW( GetModuleHandleW(NULL), MAKEINTRESOURCEW(2), (LPCWSTR)RT_MENU );
    ok( !rsrc, "resource found\n" );
    ok( GetLastError() == ERROR_RESOURCE_NAME_NOT_FOUND, "wrong error %u\n", GetLastError() );
 
    SetLastError( 0xdeadbeef );
    rsrc = FindResourceExW( GetModuleHandleW(NULL), (LPCWSTR)RT_MENU, MAKEINTRESOURCEW(1),
                            MAKELANGID( LANG_ENGLISH, SUBLANG_DEFAULT ) );
    ok( !rsrc, "resource found\n" );
    ok( GetLastError() == ERROR_RESOURCE_LANG_NOT_FOUND, "wrong error %u\n", GetLastError() );
 
    SetLastError( 0xdeadbeef );
    rsrc = FindResourceExW( GetModuleHandleW(NULL), (LPCWSTR)RT_MENU, MAKEINTRESOURCEW(1),
                            MAKELANGID( LANG_FRENCH, SUBLANG_DEFAULT ) );
    ok( !rsrc, "resource found\n" );
    ok( GetLastError() == ERROR_RESOURCE_LANG_NOT_FOUND, "wrong error %u\n", GetLastError() );
}

Referenced by START_TEST().

update_empty_exe()

static void update_empty_exe ( void  )

static

Definition at line 254 of file resource.c.

{
    HANDLE file, res, test;
    BOOL r;
 
    file = CreateFileA(filename, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, 0, 0);
    ok (file != INVALID_HANDLE_VALUE, "failed to create file\n");
 
    CloseHandle( file );
 
    res = BeginUpdateResourceA( filename, TRUE );
    if ( res != NULL || GetLastError() != ERROR_FILE_INVALID )
    {
        ok( res != NULL, "BeginUpdateResource failed\n");
 
        /* check if it's possible to open the file now */
        test = CreateFileA(filename, GENERIC_READ, 0, NULL, OPEN_EXISTING, 0, 0);
        ok (test != INVALID_HANDLE_VALUE, "failed to create file\n");
 
        CloseHandle( test );
 
        r = EndUpdateResourceA( res, FALSE );
        ok( r == FALSE, "EndUpdateResource failed\n");
    }
    else
        skip( "Can't update resource in empty file\n" );
 
    res = BeginUpdateResourceA( filename, FALSE );
    ok( res == NULL, "BeginUpdateResource failed\n");
}

Referenced by START_TEST().

update_missing_exe()

static void update_missing_exe ( void  )

static

Definition at line 244 of file resource.c.

{
    HANDLE res;
 
    SetLastError(0xdeadbeef);
    res = BeginUpdateResourceA( filename, TRUE );
    GLE = GetLastError();
    ok( res == NULL, "BeginUpdateResource should fail\n");
}

Referenced by START_TEST().

update_resources_bigdata()

static void update_resources_bigdata ( void  )

static

Definition at line 339 of file resource.c.

{
    HANDLE res = NULL;
    BOOL r;
    char foo[2*page_size] = "foobar";
 
    res = BeginUpdateResourceA( filename, TRUE );
    ok( res != NULL, "BeginUpdateResource succeeded\n");
 
    r = UpdateResourceA( res,
                        MAKEINTRESOURCEA(0x3012),
                        MAKEINTRESOURCEA(0x5647),
                        0xcdba,
                        foo, sizeof foo );
    ok( r == TRUE, "UpdateResource failed: %d\n", GetLastError());
 
    r = EndUpdateResourceA( res, FALSE );
    ok( r, "EndUpdateResource failed\n");
}

Referenced by START_TEST().

update_resources_delete()

static void update_resources_delete ( void  )

static

Definition at line 297 of file resource.c.

{
    HMODULE res;
    BOOL r;
 
    res = BeginUpdateResourceA( filename, TRUE );
    ok( res != NULL, "BeginUpdateResource failed\n");
 
    r = EndUpdateResourceA( res, FALSE );
    ok( r, "EndUpdateResource failed\n");
}

Referenced by START_TEST().

update_resources_none()

static void update_resources_none ( void  )

static

Definition at line 285 of file resource.c.

{
    HMODULE res;
    BOOL r;
 
    res = BeginUpdateResourceA( filename, FALSE );
    ok( res != NULL, "BeginUpdateResource failed\n");
 
    r = EndUpdateResourceA( res, FALSE );
    ok( r, "EndUpdateResource failed\n");
}

Referenced by START_TEST().

update_resources_version()

static void update_resources_version ( void  )

static

Definition at line 309 of file resource.c.

{
    HANDLE res = NULL;
    BOOL r;
    char foo[] = "red and white";
 
    res = BeginUpdateResourceA( filename, TRUE );
    ok( res != NULL, "BeginUpdateResource failed\n");
 
    if (0)  /* this causes subsequent tests to fail on Vista */
    {
        r = UpdateResourceA( res,
                            MAKEINTRESOURCEA(0x1230),
                            MAKEINTRESOURCEA(0x4567),
                            0xabcd,
                            NULL, 0 );
        ok( r == FALSE, "UpdateResource failed\n");
    }
 
    r = UpdateResourceA( res,
                        MAKEINTRESOURCEA(0x1230),
                        MAKEINTRESOURCEA(0x4567),
                        0xabcd,
                        foo, sizeof foo );
    ok( r == TRUE, "UpdateResource failed: %d\n", GetLastError());
 
    r = EndUpdateResourceA( res, FALSE );
    ok( r, "EndUpdateResource failed: %d\n", GetLastError());
}

Referenced by START_TEST().

Variable Documentation

filename

const char filename[] = "test_.exe"

static

Definition at line 26 of file resource.c.

GLE

DWORD GLE

static

Definition at line 27 of file resource.c.

Referenced by START_TEST(), and update_missing_exe().

sec_variants

const struct _sec_variants sec_variants[]

static

Referenced by START_TEST().

sh_junk

const IMAGE_SECTION_HEADER sh_junk

static

Initial value:

=
{
    ".reloc\0", {page_size}, 4*page_size, page_size, 4*page_size, 0, 0, 0, 0,
    IMAGE_SCN_CNT_INITIALIZED_DATA | IMAGE_SCN_MEM_READ
}

Definition at line 92 of file resource.c.

sh_junk_2

const IMAGE_SECTION_HEADER sh_junk_2

static

Initial value:

=
{
    ".reloc\0", {page_size}, 6*page_size, page_size, 6*page_size, 0, 0, 0, 0,
    IMAGE_SCN_CNT_INITIALIZED_DATA | IMAGE_SCN_MEM_READ
}

Definition at line 99 of file resource.c.

sh_rodata_1

const IMAGE_SECTION_HEADER sh_rodata_1

static

Initial value:

=
{
    ".rodata", {2*page_size}, page_size, 2*page_size, page_size, 0, 0, 0, 0,
    IMAGE_SCN_CNT_INITIALIZED_DATA | IMAGE_SCN_MEM_READ
}

Definition at line 36 of file resource.c.

sh_rodata_2

const IMAGE_SECTION_HEADER sh_rodata_2

static

Initial value:

=
{
    ".rodata", {page_size}, page_size, page_size, page_size, 0, 0, 0, 0,
    IMAGE_SCN_CNT_INITIALIZED_DATA | IMAGE_SCN_MEM_READ
}

Definition at line 43 of file resource.c.

sh_rsrc_1

const IMAGE_SECTION_HEADER sh_rsrc_1

static

Initial value:

=
{
    ".rsrc\0\0", {page_size}, rva_rsrc_start, page_size, rva_rsrc_start, 0, 0, 0, 0,
    IMAGE_SCN_CNT_INITIALIZED_DATA | IMAGE_SCN_MEM_READ
}

Definition at line 50 of file resource.c.

sh_rsrc_2

const IMAGE_SECTION_HEADER sh_rsrc_2

static

Initial value:

=
{
    ".rsrc\0\0", {2*page_size}, rva_rsrc_start-page_size, 2*page_size, rva_rsrc_start-page_size, 0, 0, 0, 0,
    IMAGE_SCN_CNT_INITIALIZED_DATA | IMAGE_SCN_MEM_READ
}

Definition at line 57 of file resource.c.

sh_rsrc_3

const IMAGE_SECTION_HEADER sh_rsrc_3

static

Initial value:

=
{
    ".rsrc\0\0", {page_size}, rva_rsrc_start-page_size, page_size, rva_rsrc_start-page_size, 0, 0, 0, 0,
    IMAGE_SCN_CNT_INITIALIZED_DATA | IMAGE_SCN_MEM_READ
}

Definition at line 64 of file resource.c.

sh_rsrc_4

const IMAGE_SECTION_HEADER sh_rsrc_4

static

Initial value:

=
{
    ".rsrc\0\0", {3*page_size}, rva_rsrc_start, 3*page_size, rva_rsrc_start, 0, 0, 0, 0,
    IMAGE_SCN_CNT_INITIALIZED_DATA | IMAGE_SCN_MEM_READ
}

Definition at line 71 of file resource.c.

sh_rsrc_5

const IMAGE_SECTION_HEADER sh_rsrc_5

static

Initial value:

=
{
    ".rsrc\0\0", {3*page_size}, 2*page_size, 3*page_size, 2*page_size, 0, 0, 0, 0,
    IMAGE_SCN_CNT_INITIALIZED_DATA | IMAGE_SCN_MEM_READ
}

Definition at line 78 of file resource.c.

sh_rsrc_6

const IMAGE_SECTION_HEADER sh_rsrc_6

static

Initial value:

=
{
    ".rsrc\0\0", {page_size}, rva_rsrc_start, 8, rva_rsrc_start, 0, 0, 0, 0,
    IMAGE_SCN_CNT_INITIALIZED_DATA | IMAGE_SCN_MEM_READ
}

Definition at line 85 of file resource.c.