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