mirror of
https://github.com/reactos/reactos.git
synced 2024-08-18 01:18:56 +00:00
303 lines
6.9 KiB
C
303 lines
6.9 KiB
C
/*
|
|
* assembly parser
|
|
*
|
|
* Copyright 2008 James Hawkins
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
|
|
*/
|
|
|
|
#include "mscoree_private.h"
|
|
|
|
#include <winver.h>
|
|
#include <dbghelp.h>
|
|
|
|
typedef struct
|
|
{
|
|
ULONG Signature;
|
|
USHORT MajorVersion;
|
|
USHORT MinorVersion;
|
|
ULONG Reserved;
|
|
ULONG VersionLength;
|
|
LPSTR Version;
|
|
BYTE Flags;
|
|
WORD Streams;
|
|
} METADATAHDR;
|
|
|
|
typedef struct
|
|
{
|
|
DWORD Offset;
|
|
DWORD Size;
|
|
} METADATASTREAMHDR;
|
|
|
|
typedef struct tagCLRTABLE
|
|
{
|
|
INT rows;
|
|
DWORD offset;
|
|
} CLRTABLE;
|
|
|
|
struct tagASSEMBLY
|
|
{
|
|
int is_mapped_file;
|
|
|
|
/* mapped files */
|
|
LPWSTR path;
|
|
HANDLE hfile;
|
|
HANDLE hmap;
|
|
|
|
BYTE *data;
|
|
|
|
IMAGE_NT_HEADERS *nthdr;
|
|
IMAGE_COR20_HEADER *corhdr;
|
|
|
|
METADATAHDR *metadatahdr;
|
|
};
|
|
|
|
static inline LPWSTR strdupW(LPCWSTR src)
|
|
{
|
|
LPWSTR dest;
|
|
|
|
if (!src)
|
|
return NULL;
|
|
|
|
dest = HeapAlloc(GetProcessHeap(), 0, (lstrlenW(src) + 1) * sizeof(WCHAR));
|
|
if (dest)
|
|
lstrcpyW(dest, src);
|
|
|
|
return dest;
|
|
}
|
|
|
|
static void* assembly_rva_to_va(ASSEMBLY *assembly, ULONG rva)
|
|
{
|
|
if (assembly->is_mapped_file)
|
|
return ImageRvaToVa(assembly->nthdr, assembly->data, rva, NULL);
|
|
else
|
|
return assembly->data + rva;
|
|
}
|
|
|
|
static ULONG assembly_datadir_get_data(ASSEMBLY *assembly,
|
|
IMAGE_DATA_DIRECTORY *datadir, void **data)
|
|
{
|
|
if (!datadir->VirtualAddress || !datadir->Size)
|
|
{
|
|
*data = NULL;
|
|
return 0;
|
|
}
|
|
else
|
|
{
|
|
*data = assembly_rva_to_va(assembly, datadir->VirtualAddress);
|
|
return datadir->Size;
|
|
}
|
|
}
|
|
|
|
static HRESULT parse_metadata_header(ASSEMBLY *assembly, DWORD *hdrsz)
|
|
{
|
|
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;
|
|
}
|
|
|
|
static HRESULT parse_clr_metadata(ASSEMBLY *assembly)
|
|
{
|
|
HRESULT hr;
|
|
DWORD hdrsz;
|
|
|
|
hr = parse_metadata_header(assembly, &hdrsz);
|
|
if (FAILED(hr))
|
|
return hr;
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT parse_pe_header(ASSEMBLY *assembly)
|
|
{
|
|
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;
|
|
}
|
|
|
|
static HRESULT parse_headers(ASSEMBLY *assembly)
|
|
{
|
|
HRESULT hr;
|
|
|
|
hr = parse_pe_header(assembly);
|
|
|
|
if (SUCCEEDED(hr))
|
|
hr = parse_clr_metadata(assembly);
|
|
|
|
return hr;
|
|
}
|
|
|
|
HRESULT assembly_create(ASSEMBLY **out, LPCWSTR file)
|
|
{
|
|
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;
|
|
}
|
|
|
|
HRESULT assembly_from_hmodule(ASSEMBLY **out, HMODULE hmodule)
|
|
{
|
|
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;
|
|
}
|
|
|
|
HRESULT assembly_release(ASSEMBLY *assembly)
|
|
{
|
|
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;
|
|
}
|
|
|
|
HRESULT assembly_get_runtime_version(ASSEMBLY *assembly, LPSTR *version)
|
|
{
|
|
*version = assembly->metadatahdr->Version;
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT assembly_get_vtable_fixups(ASSEMBLY *assembly, VTableFixup **fixups, DWORD *count)
|
|
{
|
|
ULONG size;
|
|
|
|
size = assembly_datadir_get_data(assembly, &assembly->corhdr->VTableFixups, (void**)fixups);
|
|
*count = size / sizeof(VTableFixup);
|
|
|
|
return S_OK;
|
|
}
|