assembly.c File Reference

#include "mscoree_private.h"
#include <winver.h>
#include <dbghelp.h>

Include dependency graph for assembly.c:

Go to the source code of this file.

Classes
struct	METADATAHDR
struct	METADATASTREAMHDR
struct	tagCLRTABLE
struct	tagASSEMBLY

Typedefs
typedef struct tagCLRTABLE	CLRTABLE

Functions
static LPWSTR	strdupW (LPCWSTR src)
static void *	assembly_rva_to_va (ASSEMBLY *assembly, ULONG rva)
static ULONG	assembly_datadir_get_data (ASSEMBLY *assembly, IMAGE_DATA_DIRECTORY *datadir, void **data)
static HRESULT	parse_metadata_header (ASSEMBLY *assembly, DWORD *hdrsz)
static HRESULT	parse_clr_metadata (ASSEMBLY *assembly)
static HRESULT	parse_pe_header (ASSEMBLY *assembly)
static HRESULT	parse_headers (ASSEMBLY *assembly)
HRESULT	assembly_create (ASSEMBLY **out, LPCWSTR file)
HRESULT	assembly_from_hmodule (ASSEMBLY **out, HMODULE hmodule)
HRESULT	assembly_release (ASSEMBLY *assembly)
HRESULT	assembly_get_runtime_version (ASSEMBLY *assembly, LPSTR *version)
HRESULT	assembly_get_vtable_fixups (ASSEMBLY *assembly, VTableFixup **fixups, DWORD *count)

Typedef Documentation

CLRTABLE

typedef struct tagCLRTABLE CLRTABLE

Function Documentation

assembly_create()

HRESULT assembly_create	(	ASSEMBLY **	out,
		LPCWSTR	file
	)

Definition at line 191 of file assembly.c.

{
    ASSEMBLY *assembly;
    HRESULT hr;
 
    *out = NULL;
 
    assembly = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(ASSEMBLY));
    if (!assembly)
        return E_OUTOFMEMORY;
 
    assembly->is_mapped_file = 1;
 
    assembly->path = strdupW(file);
    if (!assembly->path)
    {
        hr = E_OUTOFMEMORY;
        goto failed;
    }
 
    assembly->hfile = CreateFileW(file, GENERIC_READ, FILE_SHARE_READ,
                                  NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
    if (assembly->hfile == INVALID_HANDLE_VALUE)
    {
        hr = HRESULT_FROM_WIN32(GetLastError());
        goto failed;
    }
 
    assembly->hmap = CreateFileMappingW(assembly->hfile, NULL, PAGE_READONLY,
                                        0, 0, NULL);
    if (!assembly->hmap)
    {
        hr = HRESULT_FROM_WIN32(GetLastError());
        goto failed;
    }
 
    assembly->data = MapViewOfFile(assembly->hmap, FILE_MAP_READ, 0, 0, 0);
    if (!assembly->data)
    {
        hr = HRESULT_FROM_WIN32(GetLastError());
        goto failed;
    }
 
    hr = parse_headers(assembly);
    if (FAILED(hr)) goto failed;
 
    *out = assembly;
    return S_OK;
 
failed:
    assembly_release(assembly);
    return hr;
}

assembly_datadir_get_data()

static ULONG assembly_datadir_get_data ( ASSEMBLY *  assembly, IMAGE_DATA_DIRECTORY *  datadir, void **  data  )

static

Definition at line 89 of file assembly.c.

{
    if (!datadir->VirtualAddress || !datadir->Size)
    {
        *data = NULL;
        return 0;
    }
    else
    {
        *data = assembly_rva_to_va(assembly, datadir->VirtualAddress);
        return datadir->Size;
    }
}

Referenced by assembly_get_vtable_fixups(), and parse_pe_header().

assembly_from_hmodule()

HRESULT assembly_from_hmodule	(	ASSEMBLY **	out,
		HMODULE	hmodule
	)

Definition at line 245 of file assembly.c.

{
    ASSEMBLY *assembly;
    HRESULT hr;
 
    *out = NULL;
 
    assembly = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(ASSEMBLY));
    if (!assembly)
        return E_OUTOFMEMORY;
 
    assembly->is_mapped_file = 0;
 
    assembly->data = (BYTE*)hmodule;
 
    hr = parse_headers(assembly);
    if (SUCCEEDED(hr))
        *out = assembly;
    else
        assembly_release(assembly);
 
    return hr;
}

Referenced by FixupVTable().

assembly_get_runtime_version()

HRESULT assembly_get_runtime_version	(	ASSEMBLY *	assembly,
		LPSTR *	version
	)

Definition at line 287 of file assembly.c.

{
    *version = assembly->metadatahdr->Version;
 
    return S_OK;
}

assembly_get_vtable_fixups()

HRESULT assembly_get_vtable_fixups	(	ASSEMBLY *	assembly,
		VTableFixup **	fixups,
		DWORD *	count
	)

Definition at line 294 of file assembly.c.

{
    ULONG size;
 
    size = assembly_datadir_get_data(assembly, &assembly->corhdr->VTableFixups, (void**)fixups);
    *count = size / sizeof(VTableFixup);
 
    return S_OK;
}

Referenced by FixupVTable().

assembly_release()

HRESULT assembly_release

(

ASSEMBLY *

assembly

)

Definition at line 269 of file assembly.c.

{
    if (!assembly)
        return S_OK;
 
    if (assembly->is_mapped_file)
    {
        UnmapViewOfFile(assembly->data);
        CloseHandle(assembly->hmap);
        CloseHandle(assembly->hfile);
    }
    HeapFree(GetProcessHeap(), 0, assembly->metadatahdr);
    HeapFree(GetProcessHeap(), 0, assembly->path);
    HeapFree(GetProcessHeap(), 0, assembly);
 
    return S_OK;
}

assembly_rva_to_va()

static void * assembly_rva_to_va ( ASSEMBLY *  assembly, ULONG  rva  )

static

Definition at line 81 of file assembly.c.

{
    if (assembly->is_mapped_file)
        return ImageRvaToVa(assembly->nthdr, assembly->data, rva, NULL);
    else
        return assembly->data + rva;
}

Referenced by assembly_datadir_get_data(), and parse_metadata_header().

parse_clr_metadata()

static HRESULT parse_clr_metadata ( ASSEMBLY *  assembly )

static

Definition at line 137 of file assembly.c.

{
    HRESULT hr;
    DWORD hdrsz;
 
    hr = parse_metadata_header(assembly, &hdrsz);
    if (FAILED(hr))
        return hr;
 
    return S_OK;
}

parse_headers()

static HRESULT parse_headers ( ASSEMBLY *  assembly )

static

Definition at line 179 of file assembly.c.

{
    HRESULT hr;
 
    hr = parse_pe_header(assembly);
 
    if (SUCCEEDED(hr))
        hr = parse_clr_metadata(assembly);
 
    return hr;
}

Referenced by assembly_create(), and assembly_from_hmodule().

parse_metadata_header()

static HRESULT parse_metadata_header ( ASSEMBLY *  assembly, DWORD *  hdrsz  )

static

Definition at line 104 of file assembly.c.

{
    METADATAHDR *metadatahdr;
    BYTE *ptr, *dest;
    DWORD size, ofs;
    ULONG rva;
 
    rva = assembly->corhdr->MetaData.VirtualAddress;
    ptr = assembly_rva_to_va(assembly, rva);
    if (!ptr)
        return E_FAIL;
 
    metadatahdr = (METADATAHDR *)ptr;
 
    assembly->metadatahdr = HeapAlloc(GetProcessHeap(), 0, sizeof(METADATAHDR));
    if (!assembly->metadatahdr)
        return E_OUTOFMEMORY;
 
    size = FIELD_OFFSET(METADATAHDR, Version);
    memcpy(assembly->metadatahdr, metadatahdr, size);
 
    assembly->metadatahdr->Version = (LPSTR)&metadatahdr->Version;
 
    ofs = FIELD_OFFSET(METADATAHDR, Flags);
    ptr += FIELD_OFFSET(METADATAHDR, Version) + metadatahdr->VersionLength + 1;
    dest = (BYTE *)assembly->metadatahdr + ofs;
    memcpy(dest, ptr, sizeof(METADATAHDR) - ofs);
 
    *hdrsz = sizeof(METADATAHDR) - sizeof(LPSTR) + metadatahdr->VersionLength + 1;
 
    return S_OK;
}

parse_pe_header()

static HRESULT parse_pe_header ( ASSEMBLY *  assembly )

static

Definition at line 149 of file assembly.c.

{
    IMAGE_DATA_DIRECTORY *datadirs;
 
    assembly->nthdr = ImageNtHeader(assembly->data);
    if (!assembly->nthdr)
        return E_FAIL;
 
    if (assembly->nthdr->OptionalHeader.Magic == IMAGE_NT_OPTIONAL_HDR64_MAGIC)
    {
        IMAGE_OPTIONAL_HEADER64 *opthdr =
                (IMAGE_OPTIONAL_HEADER64 *)&assembly->nthdr->OptionalHeader;
        datadirs = opthdr->DataDirectory;
    }
    else
    {
        IMAGE_OPTIONAL_HEADER32 *opthdr =
                (IMAGE_OPTIONAL_HEADER32 *)&assembly->nthdr->OptionalHeader;
        datadirs = opthdr->DataDirectory;
    }
 
    if (!datadirs)
        return E_FAIL;
 
    if (!assembly_datadir_get_data(assembly, &datadirs[IMAGE_DIRECTORY_ENTRY_COM_DESCRIPTOR], (void**)&assembly->corhdr))
        return E_FAIL;
 
    return S_OK;
}

strdupW()

static LPWSTR strdupW ( LPCWSTR  src )

inlinestatic

Definition at line 67 of file assembly.c.

{
    LPWSTR dest;
 
    if (!src)
        return NULL;
 
    dest = HeapAlloc(GetProcessHeap(), 0, (lstrlenW(src) + 1) * sizeof(WCHAR));
    if (dest)
        lstrcpyW(dest, src);
 
    return dest;
}