dlls/mscoree/assembly.c

   1 /*
   2  * assembly parser
   3  *
   4  * Copyright 2008 James Hawkins
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 #include <stdarg.h>
  22 #include <stdio.h>
  23
  24 #include "windef.h"
  25 #include "winbase.h"
  26 #include "winuser.h"
  27 #include "winver.h"
  28 #include "dbghelp.h"
  29 #include "ole2.h"
  30 #include "mscoree.h"
  31 #include "corhdr.h"
  32 #include "metahost.h"
  33 #include "cordebug.h"
  34 #include "wine/list.h"
  35 #include "mscoree_private.h"
  36
  37 #include "wine/debug.h"
  38
  39 typedef struct
  40 {
  41     ULONG Signature;
  42     USHORT MajorVersion;
  43     USHORT MinorVersion;
  44     ULONG Reserved;
  45     ULONG VersionLength;
  46     LPSTR Version;
  47     BYTE Flags;
  48     WORD Streams;
  49 } METADATAHDR;
  50
  51 typedef struct
  52 {
  53     DWORD Offset;
  54     DWORD Size;
  55 } METADATASTREAMHDR;
  56
  57 typedef struct tagCLRTABLE
  58 {
  59     INT rows;
  60     DWORD offset;
  61 } CLRTABLE;
  62
  63 struct tagASSEMBLY
  64 {
  65     BOOL is_mapped_file;
  66
  67     /* mapped files */
  68     LPWSTR path;
  69     HANDLE hfile;
  70     HANDLE hmap;
  71
  72     BYTE *data;
  73
  74     IMAGE_NT_HEADERS *nthdr;
  75     IMAGE_COR20_HEADER *corhdr;
  76
  77     METADATAHDR *metadatahdr;
  78 };
  79
  80 static void* assembly_rva_to_va(ASSEMBLY *assembly, ULONG rva)
  81 {
  82     if (assembly->is_mapped_file)
  83         return ImageRvaToVa(assembly->nthdr, assembly->data, rva, NULL);
  84     else
  85         return assembly->data + rva;
  86 }
  87
  88 static ULONG assembly_datadir_get_data(ASSEMBLY *assembly,
  89     IMAGE_DATA_DIRECTORY *datadir, void **data)
  90 {
  91     if (!datadir->VirtualAddress || !datadir->Size)
  92     {
  93         *data = NULL;
  94         return 0;
  95     }
  96     else
  97     {
  98         *data = assembly_rva_to_va(assembly, datadir->VirtualAddress);
  99         return datadir->Size;
 100     }
 101 }
 102
 103 static HRESULT parse_metadata_header(ASSEMBLY *assembly, DWORD *hdrsz)
 104 {
 105     METADATAHDR *metadatahdr;
 106     BYTE *ptr, *dest;
 107     DWORD size, ofs;
 108     ULONG rva;
 109
 110     rva = assembly->corhdr->MetaData.VirtualAddress;
 111     ptr = assembly_rva_to_va(assembly, rva);
 112     if (!ptr)
 113         return E_FAIL;
 114
 115     metadatahdr = (METADATAHDR *)ptr;
 116
 117     assembly->metadatahdr = malloc(sizeof(METADATAHDR));
 118     if (!assembly->metadatahdr)
 119         return E_OUTOFMEMORY;
 120
 121     size = FIELD_OFFSET(METADATAHDR, Version);
 122     memcpy(assembly->metadatahdr, metadatahdr, size);
 123
 124     assembly->metadatahdr->Version = (LPSTR)&metadatahdr->Version;
 125
 126     ofs = FIELD_OFFSET(METADATAHDR, Flags);
 127     ptr += FIELD_OFFSET(METADATAHDR, Version) + metadatahdr->VersionLength + 1;
 128     dest = (BYTE *)assembly->metadatahdr + ofs;
 129     memcpy(dest, ptr, sizeof(METADATAHDR) - ofs);
 130
 131     *hdrsz = sizeof(METADATAHDR) - sizeof(LPSTR) + metadatahdr->VersionLength + 1;
 132
 133     return S_OK;
 134 }
 135
 136 static HRESULT parse_clr_metadata(ASSEMBLY *assembly)
 137 {
 138     HRESULT hr;
 139     DWORD hdrsz;
 140
 141     hr = parse_metadata_header(assembly, &hdrsz);
 142     if (FAILED(hr))
 143         return hr;
 144
 145     return S_OK;
 146 }
 147
 148 static HRESULT parse_pe_header(ASSEMBLY *assembly)
 149 {
 150     IMAGE_DATA_DIRECTORY *datadirs;
 151
 152     assembly->nthdr = ImageNtHeader(assembly->data);
 153     if (!assembly->nthdr)
 154         return E_FAIL;
 155
 156     if (assembly->nthdr->OptionalHeader.Magic == IMAGE_NT_OPTIONAL_HDR64_MAGIC)
 157     {
 158         IMAGE_OPTIONAL_HEADER64 *opthdr =
 159                 (IMAGE_OPTIONAL_HEADER64 *)&assembly->nthdr->OptionalHeader;
 160         datadirs = opthdr->DataDirectory;
 161     }
 162     else
 163     {
 164         IMAGE_OPTIONAL_HEADER32 *opthdr =
 165                 (IMAGE_OPTIONAL_HEADER32 *)&assembly->nthdr->OptionalHeader;
 166         datadirs = opthdr->DataDirectory;
 167     }
 168
 169     if (!datadirs)
 170         return E_FAIL;
 171
 172     if (!assembly_datadir_get_data(assembly, &datadirs[IMAGE_DIRECTORY_ENTRY_COM_DESCRIPTOR], (void**)&assembly->corhdr))
 173         return E_FAIL;
 174
 175     return S_OK;
 176 }
 177
 178 static HRESULT parse_headers(ASSEMBLY *assembly)
 179 {
 180     HRESULT hr;
 181
 182     hr = parse_pe_header(assembly);
 183
 184     if (SUCCEEDED(hr))
 185         hr = parse_clr_metadata(assembly);
 186
 187     return hr;
 188 }
 189
 190 HRESULT assembly_create(ASSEMBLY **out, LPCWSTR file)
 191 {
 192     ASSEMBLY *assembly;
 193     HRESULT hr;
 194
 195     *out = NULL;
 196
 197     assembly = calloc(1, sizeof(ASSEMBLY));
 198     if (!assembly)
 199         return E_OUTOFMEMORY;
 200
 201     assembly->is_mapped_file = TRUE;
 202
 203     assembly->path = wcsdup(file);
 204     if (!assembly->path)
 205     {
 206         hr = E_OUTOFMEMORY;
 207         goto failed;
 208     }
 209
 210     assembly->hfile = CreateFileW(file, GENERIC_READ, FILE_SHARE_READ,
 211                                   NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
 212     if (assembly->hfile == INVALID_HANDLE_VALUE)
 213     {
 214         hr = HRESULT_FROM_WIN32(GetLastError());
 215         goto failed;
 216     }
 217
 218     assembly->hmap = CreateFileMappingW(assembly->hfile, NULL, PAGE_READONLY,
 219                                         0, 0, NULL);
 220     if (!assembly->hmap)
 221     {
 222         hr = HRESULT_FROM_WIN32(GetLastError());
 223         goto failed;
 224     }
 225
 226     assembly->data = MapViewOfFile(assembly->hmap, FILE_MAP_READ, 0, 0, 0);
 227     if (!assembly->data)
 228     {
 229         hr = HRESULT_FROM_WIN32(GetLastError());
 230         goto failed;
 231     }
 232
 233     hr = parse_headers(assembly);
 234     if (FAILED(hr)) goto failed;
 235
 236     *out = assembly;
 237     return S_OK;
 238
 239 failed:
 240     assembly_release(assembly);
 241     return hr;
 242 }
 243
 244 HRESULT assembly_from_hmodule(ASSEMBLY **out, HMODULE hmodule)
 245 {
 246     ASSEMBLY *assembly;
 247     HRESULT hr;
 248
 249     *out = NULL;
 250
 251     assembly = calloc(1, sizeof(ASSEMBLY));
 252     if (!assembly)
 253         return E_OUTOFMEMORY;
 254
 255     assembly->is_mapped_file = FALSE;
 256
 257     assembly->data = (BYTE*)hmodule;
 258
 259     hr = parse_headers(assembly);
 260     if (SUCCEEDED(hr))
 261         *out = assembly;
 262     else
 263         assembly_release(assembly);
 264
 265     return hr;
 266 }
 267
 268 HRESULT assembly_release(ASSEMBLY *assembly)
 269 {
 270     if (!assembly)
 271         return S_OK;
 272
 273     if (assembly->is_mapped_file)
 274     {
 275         UnmapViewOfFile(assembly->data);
 276         CloseHandle(assembly->hmap);
 277         CloseHandle(assembly->hfile);
 278     }
 279     free(assembly->metadatahdr);
 280     free(assembly->path);
 281     free(assembly);
 282
 283     return S_OK;
 284 }
 285
 286 HRESULT assembly_get_runtime_version(ASSEMBLY *assembly, LPSTR *version)
 287 {
 288     *version = assembly->metadatahdr->Version;
 289
 290     return S_OK;
 291 }
 292
 293 HRESULT assembly_get_vtable_fixups(ASSEMBLY *assembly, VTableFixup **fixups, DWORD *count)
 294 {
 295     ULONG size;
 296
 297     size = assembly_datadir_get_data(assembly, &assembly->corhdr->VTableFixups, (void**)fixups);
 298     *count = size / sizeof(VTableFixup);
 299
 300     return S_OK;
 301 }
 302
 303 HRESULT assembly_get_native_entrypoint(ASSEMBLY *assembly, NativeEntryPointFunc *func)
 304 {
 305     if (assembly->corhdr->Flags & COMIMAGE_FLAGS_NATIVE_ENTRYPOINT)
 306     {
 307         *func = assembly_rva_to_va(assembly, assembly->corhdr->EntryPointRVA);
 308         return S_OK;
 309     }
 310     else
 311     {
 312         *func = NULL;
 313         return S_FALSE;
 314     }
 315 }