dlls/mscoree/corruntimehost.c

   1 /*
   2  *
   3  * Copyright 2008 Alistair Leslie-Hughes
   4  *
   5  * This library is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU Lesser General Public
   7  * License as published by the Free Software Foundation; either
   8  * version 2.1 of the License, or (at your option) any later version.
   9  *
  10  * This library is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * Lesser General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU Lesser General Public
  16  * License along with this library; if not, write to the Free Software
  17  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  18  */
  19
  20 #define COBJMACROS
  21
  22 #include <assert.h>
  23 #include <stdarg.h>
  24
  25 #include "windef.h"
  26 #include "winbase.h"
  27 #include "winuser.h"
  28 #include "winnls.h"
  29 #include "winreg.h"
  30 #include "ole2.h"
  31 #include "shellapi.h"
  32
  33 #include "cor.h"
  34 #include "mscoree.h"
  35 #include "metahost.h"
  36 #include "corhdr.h"
  37 #include "cordebug.h"
  38 #include "wine/list.h"
  39 #include "mscoree_private.h"
  40
  41 #include "wine/debug.h"
  42 #include "wine/unicode.h"
  43
  44 WINE_DEFAULT_DEBUG_CHANNEL( mscoree );
  45
  46 #include "initguid.h"
  47
  48 DEFINE_GUID(IID__AppDomain, 0x05f696dc,0x2b29,0x3663,0xad,0x8b,0xc4,0x38,0x9c,0xf2,0xa7,0x13);
  49
  50 struct DomainEntry
  51 {
  52     struct list entry;
  53     MonoDomain *domain;
  54 };
  55
  56 static HANDLE dll_fixup_heap; /* using a separate heap so we can have execute permission */
  57
  58 static struct list dll_fixups;
  59
  60 struct dll_fixup
  61 {
  62     struct list entry;
  63     BOOL done;
  64     HMODULE dll;
  65     void *thunk_code; /* pointer into dll_fixup_heap */
  66     VTableFixup *fixup;
  67     void *vtable;
  68     void *tokens; /* pointer into process heap */
  69 };
  70
  71 static MonoDomain* domain_attach(MonoDomain *domain)
  72 {
  73     MonoDomain *prev_domain = mono_domain_get();
  74
  75     if (prev_domain == domain)
  76         /* Do not set or restore domain. */
  77         return NULL;
  78
  79     mono_thread_attach(domain);
  80
  81     return prev_domain;
  82 }
  83
  84 static void domain_restore(MonoDomain *prev_domain)
  85 {
  86     if (prev_domain != NULL)
  87         mono_domain_set(prev_domain, FALSE);
  88 }
  89
  90 static HRESULT RuntimeHost_AddDefaultDomain(RuntimeHost *This, MonoDomain **result)
  91 {
  92     struct DomainEntry *entry;
  93     HRESULT res=S_OK;
  94
  95     EnterCriticalSection(&This->lock);
  96
  97     entry = HeapAlloc(GetProcessHeap(), 0, sizeof(*entry));
  98     if (!entry)
  99     {
 100         res = E_OUTOFMEMORY;
 101         goto end;
 102     }
 103
 104     /* FIXME: Use exe filename to name the domain? */
 105     entry->domain = mono_jit_init_version("mscorlib.dll", "v4.0.30319");
 106
 107     if (!entry->domain)
 108     {
 109         HeapFree(GetProcessHeap(), 0, entry);
 110         res = E_FAIL;
 111         goto end;
 112     }
 113
 114     is_mono_started = TRUE;
 115
 116     list_add_tail(&This->domains, &entry->entry);
 117
 118     *result = entry->domain;
 119
 120 end:
 121     LeaveCriticalSection(&This->lock);
 122
 123     return res;
 124 }
 125
 126 static HRESULT RuntimeHost_GetDefaultDomain(RuntimeHost *This, const WCHAR *config_path, MonoDomain **result)
 127 {
 128     WCHAR config_dir[MAX_PATH];
 129     WCHAR base_dir[MAX_PATH];
 130     char *base_dirA, *config_pathA, *slash;
 131     HRESULT res=S_OK;
 132
 133     EnterCriticalSection(&This->lock);
 134
 135     if (This->default_domain) goto end;
 136
 137     res = RuntimeHost_AddDefaultDomain(This, &This->default_domain);
 138
 139     if (!config_path)
 140     {
 141         DWORD len = ARRAY_SIZE(config_dir);
 142
 143         static const WCHAR machine_configW[] = {'\\','C','O','N','F','I','G','\\','m','a','c','h','i','n','e','.','c','o','n','f','i','g',0};
 144
 145         res = ICLRRuntimeInfo_GetRuntimeDirectory(&This->version->ICLRRuntimeInfo_iface,
 146                 config_dir, &len);
 147         if (FAILED(res))
 148             goto end;
 149
 150         lstrcatW(config_dir, machine_configW);
 151
 152         config_path = config_dir;
 153     }
 154
 155     config_pathA = WtoA(config_path);
 156     if (!config_pathA)
 157     {
 158         res = E_OUTOFMEMORY;
 159         goto end;
 160     }
 161
 162     GetModuleFileNameW(NULL, base_dir, ARRAY_SIZE(base_dir));
 163     base_dirA = WtoA(base_dir);
 164     if (!base_dirA)
 165     {
 166         HeapFree(GetProcessHeap(), 0, config_pathA);
 167         res = E_OUTOFMEMORY;
 168         goto end;
 169     }
 170
 171     slash = strrchr(base_dirA, '\\');
 172     if (slash)
 173         *(slash + 1) = 0;
 174
 175     TRACE("setting base_dir: %s, config_path: %s\n", base_dirA, config_pathA);
 176     mono_domain_set_config(This->default_domain, base_dirA, config_pathA);
 177
 178     HeapFree(GetProcessHeap(), 0, config_pathA);
 179     HeapFree(GetProcessHeap(), 0, base_dirA);
 180
 181 end:
 182     *result = This->default_domain;
 183
 184     LeaveCriticalSection(&This->lock);
 185
 186     return res;
 187 }
 188
 189 static void RuntimeHost_DeleteDomain(RuntimeHost *This, MonoDomain *domain)
 190 {
 191     struct DomainEntry *entry;
 192
 193     EnterCriticalSection(&This->lock);
 194
 195     LIST_FOR_EACH_ENTRY(entry, &This->domains, struct DomainEntry, entry)
 196     {
 197         if (entry->domain == domain)
 198         {
 199             list_remove(&entry->entry);
 200             if (This->default_domain == domain)
 201                 This->default_domain = NULL;
 202             HeapFree(GetProcessHeap(), 0, entry);
 203             break;
 204         }
 205     }
 206
 207     LeaveCriticalSection(&This->lock);
 208 }
 209
 210 static BOOL RuntimeHost_GetMethod(MonoDomain *domain, const char *assemblyname,
 211     const char *namespace, const char *typename, const char *methodname, int arg_count,
 212     MonoMethod **method)
 213 {
 214     MonoAssembly *assembly;
 215     MonoImage *image;
 216     MonoClass *klass;
 217
 218     assembly = mono_domain_assembly_open(domain, assemblyname);
 219     if (!assembly)
 220     {
 221         ERR("Cannot load assembly %s\n", assemblyname);
 222         return FALSE;
 223     }
 224
 225     image = mono_assembly_get_image(assembly);
 226     if (!image)
 227     {
 228         ERR("Couldn't get assembly image for %s\n", assemblyname);
 229         return FALSE;
 230     }
 231
 232     klass = mono_class_from_name(image, namespace, typename);
 233     if (!klass)
 234     {
 235         ERR("Couldn't get class %s.%s from image\n", namespace, typename);
 236         return FALSE;
 237     }
 238
 239     *method = mono_class_get_method_from_name(klass, methodname, arg_count);
 240     if (!*method)
 241     {
 242         ERR("Couldn't get method %s from class %s.%s\n", methodname, namespace, typename);
 243         return FALSE;
 244     }
 245
 246     return TRUE;
 247 }
 248
 249 static HRESULT RuntimeHost_Invoke(RuntimeHost *This, MonoDomain *domain,
 250     const char *assemblyname, const char *namespace, const char *typename, const char *methodname,
 251     MonoObject *obj, void **args, int arg_count, MonoObject **result);
 252
 253 static HRESULT RuntimeHost_DoInvoke(RuntimeHost *This, MonoDomain *domain,
 254     const char *methodname, MonoMethod *method, MonoObject *obj, void **args, MonoObject **result)
 255 {
 256     MonoObject *exc;
 257     static const char *get_hresult = "get_HResult";
 258
 259     *result = mono_runtime_invoke(method, obj, args, &exc);
 260     if (exc)
 261     {
 262         HRESULT hr;
 263         MonoObject *hr_object;
 264
 265         if (methodname != get_hresult)
 266         {
 267             /* Map the exception to an HRESULT. */
 268             hr = RuntimeHost_Invoke(This, domain, "mscorlib", "System", "Exception", get_hresult,
 269                 exc, NULL, 0, &hr_object);
 270             if (SUCCEEDED(hr))
 271                 hr = *(HRESULT*)mono_object_unbox(hr_object);
 272             if (SUCCEEDED(hr))
 273                 hr = E_FAIL;
 274         }
 275         else
 276             hr = E_FAIL;
 277         *result = NULL;
 278         return hr;
 279     }
 280
 281     return S_OK;
 282 }
 283
 284 static HRESULT RuntimeHost_Invoke(RuntimeHost *This, MonoDomain *domain,
 285     const char *assemblyname, const char *namespace, const char *typename, const char *methodname,
 286     MonoObject *obj, void **args, int arg_count, MonoObject **result)
 287 {
 288     MonoMethod *method;
 289     MonoDomain *prev_domain;
 290     HRESULT hr;
 291
 292     *result = NULL;
 293
 294     prev_domain = domain_attach(domain);
 295
 296     if (!RuntimeHost_GetMethod(domain, assemblyname, namespace, typename, methodname,
 297             arg_count, &method))
 298     {
 299         domain_restore(prev_domain);
 300         return E_FAIL;
 301     }
 302
 303     hr = RuntimeHost_DoInvoke(This, domain, methodname, method, obj, args, result);
 304     if (FAILED(hr))
 305     {
 306         ERR("Method %s.%s:%s raised an exception, hr=%x\n", namespace, typename, methodname, hr);
 307     }
 308
 309     domain_restore(prev_domain);
 310
 311     return hr;
 312 }
 313
 314 static HRESULT RuntimeHost_VirtualInvoke(RuntimeHost *This, MonoDomain *domain,
 315     const char *assemblyname, const char *namespace, const char *typename, const char *methodname,
 316     MonoObject *obj, void **args, int arg_count, MonoObject **result)
 317 {
 318     MonoMethod *method;
 319     MonoDomain *prev_domain;
 320     HRESULT hr;
 321
 322     *result = NULL;
 323
 324     if (!obj)
 325     {
 326         ERR("\"this\" object cannot be null\n");
 327         return E_POINTER;
 328     }
 329
 330     prev_domain = domain_attach(domain);
 331
 332     if (!RuntimeHost_GetMethod(domain, assemblyname, namespace, typename, methodname,
 333             arg_count, &method))
 334     {
 335         domain_restore(prev_domain);
 336         return E_FAIL;
 337     }
 338
 339     method = mono_object_get_virtual_method(obj, method);
 340     if (!method)
 341     {
 342         ERR("Object %p does not support method %s.%s:%s\n", obj, namespace, typename, methodname);
 343         domain_restore(prev_domain);
 344         return E_FAIL;
 345     }
 346
 347     hr = RuntimeHost_DoInvoke(This, domain, methodname, method, obj, args, result);
 348     if (FAILED(hr))
 349     {
 350         ERR("Method %s.%s:%s raised an exception, hr=%x\n", namespace, typename, methodname, hr);
 351     }
 352
 353     domain_restore(prev_domain);
 354
 355     return hr;
 356 }
 357
 358 static HRESULT RuntimeHost_GetObjectForIUnknown(RuntimeHost *This, MonoDomain *domain,
 359     IUnknown *unk, MonoObject **obj)
 360 {
 361     HRESULT hr;
 362     void *args[1];
 363     MonoObject *result;
 364
 365     args[0] = &unk;
 366     hr = RuntimeHost_Invoke(This, domain, "mscorlib", "System.Runtime.InteropServices", "Marshal", "GetObjectForIUnknown",
 367         NULL, args, 1, &result);
 368
 369     if (SUCCEEDED(hr))
 370     {
 371         *obj = result;
 372     }
 373     return hr;
 374 }
 375
 376 static HRESULT RuntimeHost_AddDomain(RuntimeHost *This, const WCHAR *name, IUnknown *setup,
 377     IUnknown *evidence, MonoDomain **result)
 378 {
 379     HRESULT res;
 380     char *nameA;
 381     MonoDomain *domain;
 382     void *args[3];
 383     MonoObject *new_domain, *id;
 384
 385     res = RuntimeHost_GetDefaultDomain(This, NULL, &domain);
 386     if (FAILED(res))
 387     {
 388         return res;
 389     }
 390
 391     nameA = WtoA(name);
 392     if (!nameA)
 393     {
 394         return E_OUTOFMEMORY;
 395     }
 396
 397     args[0] = mono_string_new(domain, nameA);
 398     HeapFree(GetProcessHeap(), 0, nameA);
 399
 400     if (!args[0])
 401     {
 402         return E_OUTOFMEMORY;
 403     }
 404
 405     if (evidence)
 406     {
 407         res = RuntimeHost_GetObjectForIUnknown(This, domain, evidence, (MonoObject **)&args[1]);
 408         if (FAILED(res))
 409         {
 410             return res;
 411         }
 412     }
 413     else
 414     {
 415         args[1] = NULL;
 416     }
 417
 418     if (setup)
 419     {
 420         res = RuntimeHost_GetObjectForIUnknown(This, domain, setup, (MonoObject **)&args[2]);
 421         if (FAILED(res))
 422         {
 423             return res;
 424         }
 425     }
 426     else
 427     {
 428         args[2] = NULL;
 429     }
 430
 431     res = RuntimeHost_Invoke(This, domain, "mscorlib", "System", "AppDomain", "CreateDomain",
 432         NULL, args, 3, &new_domain);
 433
 434     if (FAILED(res))
 435     {
 436         return res;
 437     }
 438
 439     /* new_domain is not the AppDomain itself, but a transparent proxy.
 440      * So, we'll retrieve its ID, and use that to get the real domain object.
 441      * We can't do a regular invoke, because that will bypass the proxy.
 442      * Instead, do a vcall.
 443      */
 444
 445     res = RuntimeHost_VirtualInvoke(This, domain, "mscorlib", "System", "AppDomain", "get_Id",
 446         new_domain, NULL, 0, &id);
 447
 448     if (FAILED(res))
 449     {
 450         return res;
 451     }
 452
 453     TRACE("returning domain id %d\n", *(int *)mono_object_unbox(id));
 454
 455     *result = mono_domain_get_by_id(*(int *)mono_object_unbox(id));
 456
 457     return S_OK;
 458 }
 459
 460 static HRESULT RuntimeHost_GetIUnknownForDomain(RuntimeHost *This, MonoDomain *domain, IUnknown **punk)
 461 {
 462     HRESULT hr;
 463     MonoObject *appdomain_object;
 464     IUnknown *unk;
 465
 466     hr = RuntimeHost_Invoke(This, domain, "mscorlib", "System", "AppDomain", "get_CurrentDomain",
 467         NULL, NULL, 0, &appdomain_object);
 468
 469     if (SUCCEEDED(hr))
 470         hr = RuntimeHost_GetIUnknownForObject(This, appdomain_object, &unk);
 471
 472     if (SUCCEEDED(hr))
 473     {
 474         hr = IUnknown_QueryInterface(unk, &IID__AppDomain, (void**)punk);
 475
 476         IUnknown_Release(unk);
 477     }
 478
 479     return hr;
 480 }
 481
 482 void RuntimeHost_ExitProcess(RuntimeHost *This, INT exitcode)
 483 {
 484     HRESULT hr;
 485     void *args[2];
 486     MonoDomain *domain;
 487     MonoObject *dummy;
 488
 489     hr = RuntimeHost_GetDefaultDomain(This, NULL, &domain);
 490     if (FAILED(hr))
 491     {
 492         ERR("Cannot get domain, hr=%x\n", hr);
 493         return;
 494     }
 495
 496     args[0] = &exitcode;
 497     args[1] = NULL;
 498     RuntimeHost_Invoke(This, domain, "mscorlib", "System", "Environment", "Exit",
 499         NULL, args, 1, &dummy);
 500
 501     ERR("Process should have exited\n");
 502 }
 503
 504 static inline RuntimeHost *impl_from_ICLRRuntimeHost( ICLRRuntimeHost *iface )
 505 {
 506     return CONTAINING_RECORD(iface, RuntimeHost, ICLRRuntimeHost_iface);
 507 }
 508
 509 static inline RuntimeHost *impl_from_ICorRuntimeHost( ICorRuntimeHost *iface )
 510 {
 511     return CONTAINING_RECORD(iface, RuntimeHost, ICorRuntimeHost_iface);
 512 }
 513
 514 /*** IUnknown methods ***/
 515 static HRESULT WINAPI corruntimehost_QueryInterface(ICorRuntimeHost* iface,
 516         REFIID riid,
 517         void **ppvObject)
 518 {
 519     RuntimeHost *This = impl_from_ICorRuntimeHost( iface );
 520     TRACE("%p %s %p\n", This, debugstr_guid(riid), ppvObject);
 521
 522     if ( IsEqualGUID( riid, &IID_ICorRuntimeHost ) ||
 523          IsEqualGUID( riid, &IID_IUnknown ) )
 524     {
 525         *ppvObject = iface;
 526     }
 527     else
 528     {
 529         FIXME("Unsupported interface %s\n", debugstr_guid(riid));
 530         return E_NOINTERFACE;
 531     }
 532
 533     ICorRuntimeHost_AddRef( iface );
 534
 535     return S_OK;
 536 }
 537
 538 static ULONG WINAPI corruntimehost_AddRef(ICorRuntimeHost* iface)
 539 {
 540     RuntimeHost *This = impl_from_ICorRuntimeHost( iface );
 541
 542     return InterlockedIncrement( &This->ref );
 543 }
 544
 545 static ULONG WINAPI corruntimehost_Release(ICorRuntimeHost* iface)
 546 {
 547     RuntimeHost *This = impl_from_ICorRuntimeHost( iface );
 548     ULONG ref;
 549
 550     ref = InterlockedDecrement( &This->ref );
 551
 552     return ref;
 553 }
 554
 555 /*** ICorRuntimeHost methods ***/
 556 static HRESULT WINAPI corruntimehost_CreateLogicalThreadState(
 557                     ICorRuntimeHost* iface)
 558 {
 559     FIXME("stub %p\n", iface);
 560     return E_NOTIMPL;
 561 }
 562
 563 static HRESULT WINAPI corruntimehost_DeleteLogicalThreadState(
 564                     ICorRuntimeHost* iface)
 565 {
 566     FIXME("stub %p\n", iface);
 567     return E_NOTIMPL;
 568 }
 569
 570 static HRESULT WINAPI corruntimehost_SwitchInLogicalThreadState(
 571                     ICorRuntimeHost* iface,
 572                     DWORD *fiberCookie)
 573 {
 574     FIXME("stub %p\n", iface);
 575     return E_NOTIMPL;
 576 }
 577
 578 static HRESULT WINAPI corruntimehost_SwitchOutLogicalThreadState(
 579                     ICorRuntimeHost* iface,
 580                     DWORD **fiberCookie)
 581 {
 582     FIXME("stub %p\n", iface);
 583     return E_NOTIMPL;
 584 }
 585
 586 static HRESULT WINAPI corruntimehost_LocksHeldByLogicalThread(
 587                     ICorRuntimeHost* iface,
 588                     DWORD *pCount)
 589 {
 590     FIXME("stub %p\n", iface);
 591     return E_NOTIMPL;
 592 }
 593
 594 static HRESULT WINAPI corruntimehost_MapFile(
 595     ICorRuntimeHost* iface,
 596     HANDLE hFile,
 597     HMODULE *mapAddress)
 598 {
 599     FIXME("stub %p\n", iface);
 600     return E_NOTIMPL;
 601 }
 602
 603 static HRESULT WINAPI corruntimehost_GetConfiguration(
 604     ICorRuntimeHost* iface,
 605     ICorConfiguration **pConfiguration)
 606 {
 607     FIXME("stub %p\n", iface);
 608     return E_NOTIMPL;
 609 }
 610
 611 static HRESULT WINAPI corruntimehost_Start(
 612     ICorRuntimeHost* iface)
 613 {
 614     RuntimeHost *This = impl_from_ICorRuntimeHost( iface );
 615     MonoDomain *dummy;
 616
 617     TRACE("%p\n", This);
 618
 619     return RuntimeHost_GetDefaultDomain(This, NULL, &dummy);
 620 }
 621
 622 static HRESULT WINAPI corruntimehost_Stop(
 623     ICorRuntimeHost* iface)
 624 {
 625     FIXME("stub %p\n", iface);
 626     return E_NOTIMPL;
 627 }
 628
 629 static HRESULT WINAPI corruntimehost_CreateDomain(
 630     ICorRuntimeHost* iface,
 631     LPCWSTR friendlyName,
 632     IUnknown *identityArray,
 633     IUnknown **appDomain)
 634 {
 635     return ICorRuntimeHost_CreateDomainEx(iface, friendlyName, NULL, NULL, appDomain);
 636 }
 637
 638 static HRESULT WINAPI corruntimehost_GetDefaultDomain(
 639     ICorRuntimeHost* iface,
 640     IUnknown **pAppDomain)
 641 {
 642     RuntimeHost *This = impl_from_ICorRuntimeHost( iface );
 643     HRESULT hr;
 644     MonoDomain *domain;
 645
 646     TRACE("(%p)\n", iface);
 647
 648     hr = RuntimeHost_GetDefaultDomain(This, NULL, &domain);
 649
 650     if (SUCCEEDED(hr))
 651     {
 652         hr = RuntimeHost_GetIUnknownForDomain(This, domain, pAppDomain);
 653     }
 654
 655     return hr;
 656 }
 657
 658 static HRESULT WINAPI corruntimehost_EnumDomains(
 659     ICorRuntimeHost* iface,
 660     HDOMAINENUM *hEnum)
 661 {
 662     FIXME("stub %p\n", iface);
 663     return E_NOTIMPL;
 664 }
 665
 666 static HRESULT WINAPI corruntimehost_NextDomain(
 667     ICorRuntimeHost* iface,
 668     HDOMAINENUM hEnum,
 669     IUnknown **appDomain)
 670 {
 671     FIXME("stub %p\n", iface);
 672     return E_NOTIMPL;
 673 }
 674
 675 static HRESULT WINAPI corruntimehost_CloseEnum(
 676     ICorRuntimeHost* iface,
 677     HDOMAINENUM hEnum)
 678 {
 679     FIXME("stub %p\n", iface);
 680     return E_NOTIMPL;
 681 }
 682
 683 static HRESULT WINAPI corruntimehost_CreateDomainEx(
 684     ICorRuntimeHost* iface,
 685     LPCWSTR friendlyName,
 686     IUnknown *setup,
 687     IUnknown *evidence,
 688     IUnknown **appDomain)
 689 {
 690     RuntimeHost *This = impl_from_ICorRuntimeHost( iface );
 691     HRESULT hr;
 692     MonoDomain *domain;
 693
 694     if (!friendlyName || !appDomain)
 695     {
 696         return E_POINTER;
 697     }
 698     if (!is_mono_started)
 699     {
 700         return E_FAIL;
 701     }
 702
 703     TRACE("(%p)\n", iface);
 704
 705     hr = RuntimeHost_AddDomain(This, friendlyName, setup, evidence, &domain);
 706
 707     if (SUCCEEDED(hr))
 708     {
 709         hr = RuntimeHost_GetIUnknownForDomain(This, domain, appDomain);
 710     }
 711
 712     return hr;
 713 }
 714
 715 static HRESULT WINAPI corruntimehost_CreateDomainSetup(
 716     ICorRuntimeHost* iface,
 717     IUnknown **appDomainSetup)
 718 {
 719     RuntimeHost *This = impl_from_ICorRuntimeHost( iface );
 720     HRESULT hr;
 721     MonoDomain *domain;
 722     MonoObject *obj;
 723     static const WCHAR classnameW[] = {'S','y','s','t','e','m','.','A','p','p','D','o','m','a','i','n','S','e','t','u','p',',','m','s','c','o','r','l','i','b',0};
 724
 725     TRACE("(%p)\n", iface);
 726
 727     hr = RuntimeHost_GetDefaultDomain(This, NULL, &domain);
 728
 729     if (SUCCEEDED(hr))
 730         hr = RuntimeHost_CreateManagedInstance(This, classnameW, domain, &obj);
 731
 732     if (SUCCEEDED(hr))
 733         hr = RuntimeHost_GetIUnknownForObject(This, obj, appDomainSetup);
 734
 735     return hr;
 736 }
 737
 738 static HRESULT WINAPI corruntimehost_CreateEvidence(
 739     ICorRuntimeHost* iface,
 740     IUnknown **evidence)
 741 {
 742     FIXME("stub %p\n", iface);
 743     return E_NOTIMPL;
 744 }
 745
 746 static HRESULT WINAPI corruntimehost_UnloadDomain(
 747     ICorRuntimeHost* iface,
 748     IUnknown *appDomain)
 749 {
 750     FIXME("stub %p\n", iface);
 751     return E_NOTIMPL;
 752 }
 753
 754 static HRESULT WINAPI corruntimehost_CurrentDomain(
 755     ICorRuntimeHost* iface,
 756     IUnknown **appDomain)
 757 {
 758     FIXME("stub %p\n", iface);
 759     return E_NOTIMPL;
 760 }
 761
 762 static const struct ICorRuntimeHostVtbl corruntimehost_vtbl =
 763 {
 764     corruntimehost_QueryInterface,
 765     corruntimehost_AddRef,
 766     corruntimehost_Release,
 767     corruntimehost_CreateLogicalThreadState,
 768     corruntimehost_DeleteLogicalThreadState,
 769     corruntimehost_SwitchInLogicalThreadState,
 770     corruntimehost_SwitchOutLogicalThreadState,
 771     corruntimehost_LocksHeldByLogicalThread,
 772     corruntimehost_MapFile,
 773     corruntimehost_GetConfiguration,
 774     corruntimehost_Start,
 775     corruntimehost_Stop,
 776     corruntimehost_CreateDomain,
 777     corruntimehost_GetDefaultDomain,
 778     corruntimehost_EnumDomains,
 779     corruntimehost_NextDomain,
 780     corruntimehost_CloseEnum,
 781     corruntimehost_CreateDomainEx,
 782     corruntimehost_CreateDomainSetup,
 783     corruntimehost_CreateEvidence,
 784     corruntimehost_UnloadDomain,
 785     corruntimehost_CurrentDomain
 786 };
 787
 788 static HRESULT WINAPI CLRRuntimeHost_QueryInterface(ICLRRuntimeHost* iface,
 789         REFIID riid,
 790         void **ppvObject)
 791 {
 792     RuntimeHost *This = impl_from_ICLRRuntimeHost( iface );
 793     TRACE("%p %s %p\n", This, debugstr_guid(riid), ppvObject);
 794
 795     if ( IsEqualGUID( riid, &IID_ICLRRuntimeHost ) ||
 796          IsEqualGUID( riid, &IID_IUnknown ) )
 797     {
 798         *ppvObject = iface;
 799     }
 800     else
 801     {
 802         FIXME("Unsupported interface %s\n", debugstr_guid(riid));
 803         return E_NOINTERFACE;
 804     }
 805
 806     ICLRRuntimeHost_AddRef( iface );
 807
 808     return S_OK;
 809 }
 810
 811 static ULONG WINAPI CLRRuntimeHost_AddRef(ICLRRuntimeHost* iface)
 812 {
 813     RuntimeHost *This = impl_from_ICLRRuntimeHost( iface );
 814     return ICorRuntimeHost_AddRef(&This->ICorRuntimeHost_iface);
 815 }
 816
 817 static ULONG WINAPI CLRRuntimeHost_Release(ICLRRuntimeHost* iface)
 818 {
 819     RuntimeHost *This = impl_from_ICLRRuntimeHost( iface );
 820     return ICorRuntimeHost_Release(&This->ICorRuntimeHost_iface);
 821 }
 822
 823 static HRESULT WINAPI CLRRuntimeHost_Start(ICLRRuntimeHost* iface)
 824 {
 825     FIXME("(%p)\n", iface);
 826     return E_NOTIMPL;
 827 }
 828
 829 static HRESULT WINAPI CLRRuntimeHost_Stop(ICLRRuntimeHost* iface)
 830 {
 831     FIXME("(%p)\n", iface);
 832     return E_NOTIMPL;
 833 }
 834
 835 static HRESULT WINAPI CLRRuntimeHost_SetHostControl(ICLRRuntimeHost* iface,
 836     IHostControl *pHostControl)
 837 {
 838     FIXME("(%p,%p)\n", iface, pHostControl);
 839     return E_NOTIMPL;
 840 }
 841
 842 static HRESULT WINAPI CLRRuntimeHost_GetCLRControl(ICLRRuntimeHost* iface,
 843     ICLRControl **pCLRControl)
 844 {
 845     FIXME("(%p,%p)\n", iface, pCLRControl);
 846     return E_NOTIMPL;
 847 }
 848
 849 static HRESULT WINAPI CLRRuntimeHost_UnloadAppDomain(ICLRRuntimeHost* iface,
 850     DWORD dwAppDomainId, BOOL fWaitUntilDone)
 851 {
 852     FIXME("(%p,%u,%i)\n", iface, dwAppDomainId, fWaitUntilDone);
 853     return E_NOTIMPL;
 854 }
 855
 856 static HRESULT WINAPI CLRRuntimeHost_ExecuteInAppDomain(ICLRRuntimeHost* iface,
 857     DWORD dwAppDomainId, FExecuteInAppDomainCallback pCallback, void *cookie)
 858 {
 859     FIXME("(%p,%u,%p,%p)\n", iface, dwAppDomainId, pCallback, cookie);
 860     return E_NOTIMPL;
 861 }
 862
 863 static HRESULT WINAPI CLRRuntimeHost_GetCurrentAppDomainId(ICLRRuntimeHost* iface,
 864     DWORD *pdwAppDomainId)
 865 {
 866     FIXME("(%p,%p)\n", iface, pdwAppDomainId);
 867     return E_NOTIMPL;
 868 }
 869
 870 static HRESULT WINAPI CLRRuntimeHost_ExecuteApplication(ICLRRuntimeHost* iface,
 871     LPCWSTR pwzAppFullName, DWORD dwManifestPaths, LPCWSTR *ppwzManifestPaths,
 872     DWORD dwActivationData, LPCWSTR *ppwzActivationData, int *pReturnValue)
 873 {
 874     FIXME("(%p,%s,%u,%u)\n", iface, debugstr_w(pwzAppFullName), dwManifestPaths, dwActivationData);
 875     return E_NOTIMPL;
 876 }
 877
 878 static HRESULT WINAPI CLRRuntimeHost_ExecuteInDefaultAppDomain(ICLRRuntimeHost* iface,
 879     LPCWSTR pwzAssemblyPath, LPCWSTR pwzTypeName, LPCWSTR pwzMethodName,
 880     LPCWSTR pwzArgument, DWORD *pReturnValue)
 881 {
 882     RuntimeHost *This = impl_from_ICLRRuntimeHost( iface );
 883     HRESULT hr;
 884     MonoDomain *domain, *prev_domain;
 885     MonoObject *result;
 886     MonoString *str;
 887     char *filenameA = NULL, *classA = NULL, *methodA = NULL;
 888     char *argsA = NULL, *ns;
 889
 890     TRACE("(%p,%s,%s,%s,%s)\n", iface, debugstr_w(pwzAssemblyPath),
 891         debugstr_w(pwzTypeName), debugstr_w(pwzMethodName), debugstr_w(pwzArgument));
 892
 893     hr = RuntimeHost_GetDefaultDomain(This, NULL, &domain);
 894
 895     if (FAILED(hr))
 896         return hr;
 897
 898     prev_domain = domain_attach(domain);
 899
 900     if (SUCCEEDED(hr))
 901     {
 902         filenameA = WtoA(pwzAssemblyPath);
 903         if (!filenameA) hr = E_OUTOFMEMORY;
 904     }
 905
 906     if (SUCCEEDED(hr))
 907     {
 908         classA = WtoA(pwzTypeName);
 909         if (!classA) hr = E_OUTOFMEMORY;
 910     }
 911
 912     if (SUCCEEDED(hr))
 913     {
 914         ns = strrchr(classA, '.');
 915         if (ns)
 916             *ns = '\0';
 917         else
 918             hr = E_INVALIDARG;
 919     }
 920
 921     if (SUCCEEDED(hr))
 922     {
 923         methodA = WtoA(pwzMethodName);
 924         if (!methodA) hr = E_OUTOFMEMORY;
 925     }
 926
 927     /* The .NET function we are calling has the following declaration
 928      *   public static int functionName(String param)
 929      */
 930     if (SUCCEEDED(hr))
 931     {
 932         argsA = WtoA(pwzArgument);
 933         if (!argsA) hr = E_OUTOFMEMORY;
 934     }
 935
 936     if (SUCCEEDED(hr))
 937     {
 938         str = mono_string_new(domain, argsA);
 939         if (!str) hr = E_OUTOFMEMORY;
 940     }
 941
 942     if (SUCCEEDED(hr))
 943     {
 944         hr = RuntimeHost_Invoke(This, domain, filenameA, classA, ns+1, methodA,
 945             NULL, (void**)&str, 1, &result);
 946     }
 947
 948     if (SUCCEEDED(hr))
 949         *pReturnValue = *(DWORD*)mono_object_unbox(result);
 950
 951     domain_restore(prev_domain);
 952
 953     HeapFree(GetProcessHeap(), 0, filenameA);
 954     HeapFree(GetProcessHeap(), 0, classA);
 955     HeapFree(GetProcessHeap(), 0, argsA);
 956     HeapFree(GetProcessHeap(), 0, methodA);
 957
 958     return hr;
 959 }
 960
 961 static const struct ICLRRuntimeHostVtbl CLRHostVtbl =
 962 {
 963     CLRRuntimeHost_QueryInterface,
 964     CLRRuntimeHost_AddRef,
 965     CLRRuntimeHost_Release,
 966     CLRRuntimeHost_Start,
 967     CLRRuntimeHost_Stop,
 968     CLRRuntimeHost_SetHostControl,
 969     CLRRuntimeHost_GetCLRControl,
 970     CLRRuntimeHost_UnloadAppDomain,
 971     CLRRuntimeHost_ExecuteInAppDomain,
 972     CLRRuntimeHost_GetCurrentAppDomainId,
 973     CLRRuntimeHost_ExecuteApplication,
 974     CLRRuntimeHost_ExecuteInDefaultAppDomain
 975 };
 976
 977 /* Create an instance of a type given its name, by calling its constructor with
 978  * no arguments. Note that result MUST be in the stack, or the garbage
 979  * collector may free it prematurely. */
 980 HRESULT RuntimeHost_CreateManagedInstance(RuntimeHost *This, LPCWSTR name,
 981     MonoDomain *domain, MonoObject **result)
 982 {
 983     HRESULT hr=S_OK;
 984     char *nameA=NULL;
 985     MonoType *type;
 986     MonoClass *klass;
 987     MonoObject *obj;
 988     MonoDomain *prev_domain;
 989
 990     if (!domain)
 991         hr = RuntimeHost_GetDefaultDomain(This, NULL, &domain);
 992
 993     if (FAILED(hr))
 994         return hr;
 995
 996     prev_domain = domain_attach(domain);
 997
 998     if (SUCCEEDED(hr))
 999     {
1000         nameA = WtoA(name);
1001         if (!nameA)
1002             hr = E_OUTOFMEMORY;
1003     }
1004
1005     if (SUCCEEDED(hr))
1006     {
1007         type = mono_reflection_type_from_name(nameA, NULL);
1008         if (!type)
1009         {
1010             ERR("Cannot find type %s\n", debugstr_w(name));
1011             hr = E_FAIL;
1012         }
1013     }
1014
1015     if (SUCCEEDED(hr))
1016     {
1017         klass = mono_class_from_mono_type(type);
1018         if (!klass)
1019         {
1020             ERR("Cannot convert type %s to a class\n", debugstr_w(name));
1021             hr = E_FAIL;
1022         }
1023     }
1024
1025     if (SUCCEEDED(hr))
1026     {
1027         obj = mono_object_new(domain, klass);
1028         if (!obj)
1029         {
1030             ERR("Cannot allocate object of type %s\n", debugstr_w(name));
1031             hr = E_FAIL;
1032         }
1033     }
1034
1035     if (SUCCEEDED(hr))
1036     {
1037         /* FIXME: Detect exceptions from the constructor? */
1038         mono_runtime_object_init(obj);
1039         *result = obj;
1040     }
1041
1042     domain_restore(prev_domain);
1043
1044     HeapFree(GetProcessHeap(), 0, nameA);
1045
1046     return hr;
1047 }
1048
1049 /* Get an IUnknown pointer for a Mono object.
1050  *
1051  * This is just a "light" wrapper around
1052  * System.Runtime.InteropServices.Marshal:GetIUnknownForObject
1053  *
1054  * NOTE: The IUnknown* is created with a reference to the object.
1055  * Until they have a reference, objects must be in the stack to prevent the
1056  * garbage collector from freeing them. */
1057 HRESULT RuntimeHost_GetIUnknownForObject(RuntimeHost *This, MonoObject *obj,
1058     IUnknown **ppUnk)
1059 {
1060     MonoDomain *domain;
1061     MonoObject *result;
1062     HRESULT hr;
1063
1064     domain = mono_object_get_domain(obj);
1065
1066     hr = RuntimeHost_Invoke(This, domain, "mscorlib", "System.Runtime.InteropServices", "Marshal", "GetIUnknownForObject",
1067         NULL, (void**)&obj, 1, &result);
1068
1069     if (SUCCEEDED(hr))
1070         *ppUnk = *(IUnknown**)mono_object_unbox(result);
1071     else
1072         *ppUnk = NULL;
1073
1074     return hr;
1075 }
1076
1077 static void get_utf8_args(int *argc, char ***argv)
1078 {
1079     WCHAR **argvw;
1080     int size=0, i;
1081     char *current_arg;
1082
1083     argvw = CommandLineToArgvW(GetCommandLineW(), argc);
1084
1085     for (i=0; i<*argc; i++)
1086     {
1087         size += sizeof(char*);
1088         size += WideCharToMultiByte(CP_UTF8, 0, argvw[i], -1, NULL, 0, NULL, NULL);
1089     }
1090     size += sizeof(char*);
1091
1092     *argv = HeapAlloc(GetProcessHeap(), 0, size);
1093     current_arg = (char*)(*argv + *argc + 1);
1094
1095     for (i=0; i<*argc; i++)
1096     {
1097         (*argv)[i] = current_arg;
1098         current_arg += WideCharToMultiByte(CP_UTF8, 0, argvw[i], -1, current_arg, size, NULL, NULL);
1099     }
1100
1101     (*argv)[*argc] = NULL;
1102
1103     HeapFree(GetProcessHeap(), 0, argvw);
1104 }
1105
1106 #if __i386__
1107
1108 # define CAN_FIXUP_VTABLE 1
1109
1110 #include "pshpack1.h"
1111
1112 struct vtable_fixup_thunk
1113 {
1114     /* push %ecx */
1115     BYTE i7;
1116     /* sub $0x4,%esp */
1117     BYTE i1[3];
1118     /* mov fixup,(%esp) */
1119     BYTE i2[3];
1120     struct dll_fixup *fixup;
1121     /* mov function,%eax */
1122     BYTE i3;
1123     void (CDECL *function)(struct dll_fixup *);
1124     /* call *%eax */
1125     BYTE i4[2];
1126     /* pop %eax */
1127     BYTE i5;
1128     /* pop %ecx */
1129     BYTE i8;
1130     /* jmp *vtable_entry */
1131     BYTE i6[2];
1132     void *vtable_entry;
1133 };
1134
1135 static const struct vtable_fixup_thunk thunk_template = {
1136     0x51,
1137     {0x83,0xec,0x04},
1138     {0xc7,0x04,0x24},
1139     NULL,
1140     0xb8,
1141     NULL,
1142     {0xff,0xd0},
1143     0x58,
1144     0x59,
1145     {0xff,0x25},
1146     NULL
1147 };
1148
1149 #include "poppack.h"
1150
1151 #elif __x86_64__ /* !__i386__ */
1152
1153 # define CAN_FIXUP_VTABLE 1
1154
1155 #include "pshpack1.h"
1156
1157 struct vtable_fixup_thunk
1158 {
1159     /* push %rbp;
1160        mov %rsp, %rbp
1161        sub $0x80, %rsp ; 0x8*4 + 0x10*4 + 0x20
1162     */
1163     BYTE i1[11];
1164     /*
1165         mov %rcx, 0x60(%rsp); mov %rdx, 0x68(%rsp); mov %r8, 0x70(%rsp); mov %r9, 0x78(%rsp);
1166         movaps %xmm0,0x20(%rsp); ...; movaps %xmm3,0x50(%esp)
1167     */
1168     BYTE i2[40];
1169     /* mov function,%rax */
1170     BYTE i3[2];
1171     void (CDECL *function)(struct dll_fixup *);
1172     /* mov fixup,%rcx */
1173     BYTE i4[2];
1174     struct dll_fixup *fixup;
1175     /* call *%rax */
1176     BYTE i5[2];
1177     /*
1178         mov 0x60(%rsp),%rcx; mov 0x68(%rsp),%rdx; mov 0x70(%rsp),%r8; mov 0x78(%rsp),%r9;
1179         movaps 0x20(%rsp),xmm0; ...; movaps 0x50(%esp),xmm3
1180     */
1181     BYTE i6[40];
1182     /* mov %rbp, %rsp
1183        pop %rbp
1184     */
1185     BYTE i7[4];
1186     /* mov vtable_entry, %rax */
1187     BYTE i8[2];
1188     void *vtable_entry;
1189     /* mov [%rax],%rax
1190        jmp %rax */
1191     BYTE i9[5];
1192 };
1193
1194 static const struct vtable_fixup_thunk thunk_template = {
1195     {0x55,0x48,0x89,0xE5,  0x48,0x81,0xEC,0x80,0x00,0x00,0x00},
1196     {0x48,0x89,0x4C,0x24,0x60, 0x48,0x89,0x54,0x24,0x68,
1197      0x4C,0x89,0x44,0x24,0x70, 0x4C,0x89,0x4C,0x24,0x78,
1198      0x0F,0x29,0x44,0x24,0x20, 0x0F,0x29,0x4C,0x24,0x30,
1199      0x0F,0x29,0x54,0x24,0x40, 0x0F,0x29,0x5C,0x24,0x50,
1200     },
1201     {0x48,0xB8},
1202     NULL,
1203     {0x48,0xB9},
1204     NULL,
1205     {0xFF,0xD0},
1206     {0x48,0x8B,0x4C,0x24,0x60, 0x48,0x8B,0x54,0x24,0x68,
1207      0x4C,0x8B,0x44,0x24,0x70, 0x4C,0x8B,0x4C,0x24,0x78,
1208      0x0F,0x28,0x44,0x24,0x20, 0x0F,0x28,0x4C,0x24,0x30,
1209      0x0F,0x28,0x54,0x24,0x40, 0x0F,0x28,0x5C,0x24,0x50,
1210      },
1211     {0x48,0x89,0xEC, 0x5D},
1212     {0x48,0xB8},
1213     NULL,
1214     {0x48,0x8B,0x00,0xFF,0xE0}
1215 };
1216
1217 #include "poppack.h"
1218
1219 #else /* !__i386__ && !__x86_64__ */
1220
1221 # define CAN_FIXUP_VTABLE 0
1222
1223 struct vtable_fixup_thunk
1224 {
1225     struct dll_fixup *fixup;
1226     void (CDECL *function)(struct dll_fixup *fixup);
1227     void *vtable_entry;
1228 };
1229
1230 static const struct vtable_fixup_thunk thunk_template = {0};
1231
1232 #endif
1233
1234 static void CDECL ReallyFixupVTable(struct dll_fixup *fixup)
1235 {
1236     HRESULT hr=S_OK;
1237     WCHAR filename[MAX_PATH];
1238     ICLRRuntimeInfo *info=NULL;
1239     RuntimeHost *host;
1240     char *filenameA;
1241     MonoImage *image=NULL;
1242     MonoAssembly *assembly=NULL;
1243     MonoImageOpenStatus status=0;
1244     MonoDomain *domain;
1245
1246     if (fixup->done) return;
1247
1248     /* It's possible we'll have two threads doing this at once. This is
1249      * considered preferable to the potential deadlock if we use a mutex. */
1250
1251     GetModuleFileNameW(fixup->dll, filename, MAX_PATH);
1252
1253     TRACE("%p,%p,%s\n", fixup, fixup->dll, debugstr_w(filename));
1254
1255     filenameA = WtoA(filename);
1256     if (!filenameA)
1257         hr = E_OUTOFMEMORY;
1258
1259     if (SUCCEEDED(hr))
1260         hr = get_runtime_info(filename, NULL, NULL, 0, 0, FALSE, &info);
1261
1262     if (SUCCEEDED(hr))
1263         hr = ICLRRuntimeInfo_GetRuntimeHost(info, &host);
1264
1265     if (SUCCEEDED(hr))
1266         hr = RuntimeHost_GetDefaultDomain(host, NULL, &domain);
1267
1268     if (SUCCEEDED(hr))
1269     {
1270         MonoDomain *prev_domain;
1271
1272         prev_domain = domain_attach(domain);
1273
1274         assembly = mono_assembly_open(filenameA, &status);
1275
1276         if (assembly)
1277         {
1278             int i;
1279
1280             /* Mono needs an image that belongs to an assembly. */
1281             image = mono_assembly_get_image(assembly);
1282
1283 #if __x86_64__
1284             if (fixup->fixup->type & COR_VTABLE_64BIT)
1285 #else
1286             if (fixup->fixup->type & COR_VTABLE_32BIT)
1287 #endif
1288             {
1289                 void **vtable = fixup->vtable;
1290                 ULONG_PTR *tokens = fixup->tokens;
1291                 for (i=0; i<fixup->fixup->count; i++)
1292                 {
1293                     TRACE("%#lx\n", tokens[i]);
1294                     vtable[i] = mono_marshal_get_vtfixup_ftnptr(
1295                         image, tokens[i], fixup->fixup->type);
1296                 }
1297             }
1298
1299             fixup->done = TRUE;
1300         }
1301
1302         domain_restore(prev_domain);
1303     }
1304
1305     if (info != NULL)
1306         ICLRRuntimeInfo_Release(info);
1307
1308     HeapFree(GetProcessHeap(), 0, filenameA);
1309
1310     if (!fixup->done)
1311     {
1312         ERR("unable to fixup vtable, hr=%x, status=%d\n", hr, status);
1313         /* If we returned now, we'd get an infinite loop. */
1314         assert(0);
1315     }
1316 }
1317
1318 static void FixupVTableEntry(HMODULE hmodule, VTableFixup *vtable_fixup)
1319 {
1320     /* We can't actually generate code for the functions without loading mono,
1321      * and loading mono inside DllMain is a terrible idea. So we make thunks
1322      * that call ReallyFixupVTable, which will load the runtime and fill in the
1323      * vtable, then do an indirect jump using the (now filled in) vtable. Note
1324      * that we have to keep the thunks around forever, as one of them may get
1325      * called while we're filling in the table, and we can never be sure all
1326      * threads are clear. */
1327     struct dll_fixup *fixup;
1328
1329     fixup = HeapAlloc(GetProcessHeap(), 0, sizeof(*fixup));
1330
1331     fixup->dll = hmodule;
1332     fixup->thunk_code = HeapAlloc(dll_fixup_heap, 0, sizeof(struct vtable_fixup_thunk) * vtable_fixup->count);
1333     fixup->fixup = vtable_fixup;
1334     fixup->vtable = (BYTE*)hmodule + vtable_fixup->rva;
1335     fixup->done = FALSE;
1336
1337     TRACE("vtable_fixup->type=0x%x\n",vtable_fixup->type);
1338 #if __x86_64__
1339     if (vtable_fixup->type & COR_VTABLE_64BIT)
1340 #else
1341     if (vtable_fixup->type & COR_VTABLE_32BIT)
1342 #endif
1343     {
1344         void **vtable = fixup->vtable;
1345         ULONG_PTR *tokens;
1346         int i;
1347         struct vtable_fixup_thunk *thunks = fixup->thunk_code;
1348
1349         tokens = fixup->tokens = HeapAlloc(GetProcessHeap(), 0, sizeof(*tokens) * vtable_fixup->count);
1350         memcpy(tokens, vtable, sizeof(*tokens) * vtable_fixup->count);
1351         for (i=0; i<vtable_fixup->count; i++)
1352         {
1353             thunks[i] = thunk_template;
1354             thunks[i].fixup = fixup;
1355             thunks[i].function = ReallyFixupVTable;
1356             thunks[i].vtable_entry = &vtable[i];
1357             vtable[i] = &thunks[i];
1358         }
1359     }
1360     else
1361     {
1362         ERR("unsupported vtable fixup flags %x\n", vtable_fixup->type);
1363         HeapFree(dll_fixup_heap, 0, fixup->thunk_code);
1364         HeapFree(GetProcessHeap(), 0, fixup);
1365         return;
1366     }
1367
1368     list_add_tail(&dll_fixups, &fixup->entry);
1369 }
1370
1371 static void FixupVTable_Assembly(HMODULE hmodule, ASSEMBLY *assembly)
1372 {
1373     VTableFixup *vtable_fixups;
1374     ULONG vtable_fixup_count, i;
1375
1376     assembly_get_vtable_fixups(assembly, &vtable_fixups, &vtable_fixup_count);
1377     if (CAN_FIXUP_VTABLE)
1378         for (i=0; i<vtable_fixup_count; i++)
1379             FixupVTableEntry(hmodule, &vtable_fixups[i]);
1380     else if (vtable_fixup_count)
1381         FIXME("cannot fixup vtable; expect a crash\n");
1382 }
1383
1384 static void FixupVTable(HMODULE hmodule)
1385 {
1386     ASSEMBLY *assembly;
1387     HRESULT hr;
1388
1389     hr = assembly_from_hmodule(&assembly, hmodule);
1390     if (SUCCEEDED(hr))
1391     {
1392         FixupVTable_Assembly(hmodule, assembly);
1393         assembly_release(assembly);
1394     }
1395     else
1396         ERR("failed to read CLR headers, hr=%x\n", hr);
1397 }
1398
1399 __int32 WINAPI _CorExeMain(void)
1400 {
1401     int exit_code;
1402     int argc;
1403     char **argv;
1404     MonoDomain *domain=NULL;
1405     MonoImage *image;
1406     MonoImageOpenStatus status;
1407     MonoAssembly *assembly=NULL;
1408     WCHAR filename[MAX_PATH];
1409     char *filenameA;
1410     ICLRRuntimeInfo *info;
1411     RuntimeHost *host;
1412     HRESULT hr;
1413     int i;
1414
1415     get_utf8_args(&argc, &argv);
1416
1417     GetModuleFileNameW(NULL, filename, MAX_PATH);
1418
1419     TRACE("%s", debugstr_w(filename));
1420     for (i=0; i<argc; i++)
1421         TRACE(" %s", debugstr_a(argv[i]));
1422     TRACE("\n");
1423
1424     filenameA = WtoA(filename);
1425     if (!filenameA)
1426     {
1427         HeapFree(GetProcessHeap(), 0, argv);
1428         return -1;
1429     }
1430
1431     FixupVTable(GetModuleHandleW(NULL));
1432
1433     hr = get_runtime_info(filename, NULL, NULL, 0, 0, FALSE, &info);
1434
1435     if (SUCCEEDED(hr))
1436     {
1437         hr = ICLRRuntimeInfo_GetRuntimeHost(info, &host);
1438
1439         if (SUCCEEDED(hr))
1440         {
1441             WCHAR config_file[MAX_PATH];
1442             static const WCHAR dotconfig[] = {'.','c','o','n','f','i','g',0};
1443
1444             strcpyW(config_file, filename);
1445             strcatW(config_file, dotconfig);
1446
1447             hr = RuntimeHost_GetDefaultDomain(host, config_file, &domain);
1448         }
1449
1450         if (SUCCEEDED(hr))
1451         {
1452             image = mono_image_open_from_module_handle(GetModuleHandleW(NULL),
1453                 filenameA, 1, &status);
1454
1455             if (image)
1456                 assembly = mono_assembly_load_from(image, filenameA, &status);
1457
1458             if (assembly)
1459             {
1460                 mono_callspec_set_assembly(assembly);
1461
1462                 exit_code = mono_jit_exec(domain, assembly, argc, argv);
1463             }
1464             else
1465             {
1466                 ERR("couldn't load %s, status=%d\n", debugstr_w(filename), status);
1467                 exit_code = -1;
1468             }
1469
1470             RuntimeHost_DeleteDomain(host, domain);
1471         }
1472         else
1473             exit_code = -1;
1474
1475         ICLRRuntimeInfo_Release(info);
1476     }
1477     else
1478         exit_code = -1;
1479
1480     HeapFree(GetProcessHeap(), 0, argv);
1481
1482     if (domain)
1483     {
1484         mono_thread_manage();
1485         mono_runtime_quit();
1486     }
1487
1488     return exit_code;
1489 }
1490
1491 BOOL WINAPI _CorDllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved)
1492 {
1493     ASSEMBLY *assembly=NULL;
1494     HRESULT hr;
1495
1496     TRACE("(%p, %d, %p)\n", hinstDLL, fdwReason, lpvReserved);
1497
1498     hr = assembly_from_hmodule(&assembly, hinstDLL);
1499     if (SUCCEEDED(hr))
1500     {
1501         NativeEntryPointFunc NativeEntryPoint=NULL;
1502
1503         assembly_get_native_entrypoint(assembly, &NativeEntryPoint);
1504         if (fdwReason == DLL_PROCESS_ATTACH)
1505         {
1506             if (!NativeEntryPoint)
1507                 DisableThreadLibraryCalls(hinstDLL);
1508             FixupVTable_Assembly(hinstDLL,assembly);
1509         }
1510         assembly_release(assembly);
1511         /* FIXME: clean up the vtables on DLL_PROCESS_DETACH */
1512         if (NativeEntryPoint)
1513             return NativeEntryPoint(hinstDLL, fdwReason, lpvReserved);
1514     }
1515     else
1516         ERR("failed to read CLR headers, hr=%x\n", hr);
1517
1518     return TRUE;
1519 }
1520
1521 /* called from DLL_PROCESS_ATTACH */
1522 void runtimehost_init(void)
1523 {
1524     dll_fixup_heap = HeapCreate(HEAP_CREATE_ENABLE_EXECUTE, 0, 0);
1525     list_init(&dll_fixups);
1526 }
1527
1528 /* called from DLL_PROCESS_DETACH */
1529 void runtimehost_uninit(void)
1530 {
1531     struct dll_fixup *fixup, *fixup2;
1532
1533     HeapDestroy(dll_fixup_heap);
1534     LIST_FOR_EACH_ENTRY_SAFE(fixup, fixup2, &dll_fixups, struct dll_fixup, entry)
1535     {
1536         HeapFree(GetProcessHeap(), 0, fixup->tokens);
1537         HeapFree(GetProcessHeap(), 0, fixup);
1538     }
1539 }
1540
1541 HRESULT RuntimeHost_Construct(CLRRuntimeInfo *runtime_version, RuntimeHost** result)
1542 {
1543     RuntimeHost *This;
1544
1545     This = HeapAlloc( GetProcessHeap(), 0, sizeof *This );
1546     if ( !This )
1547         return E_OUTOFMEMORY;
1548
1549     This->ICorRuntimeHost_iface.lpVtbl = &corruntimehost_vtbl;
1550     This->ICLRRuntimeHost_iface.lpVtbl = &CLRHostVtbl;
1551
1552     This->ref = 1;
1553     This->version = runtime_version;
1554     list_init(&This->domains);
1555     This->default_domain = NULL;
1556     InitializeCriticalSection(&This->lock);
1557     This->lock.DebugInfo->Spare[0] = (DWORD_PTR)(__FILE__ ": RuntimeHost.lock");
1558
1559     *result = This;
1560
1561     return S_OK;
1562 }
1563
1564 HRESULT RuntimeHost_GetInterface(RuntimeHost *This, REFCLSID clsid, REFIID riid, void **ppv)
1565 {
1566     IUnknown *unk;
1567     HRESULT hr;
1568
1569     if (IsEqualGUID(clsid, &CLSID_CorRuntimeHost))
1570     {
1571         unk = (IUnknown*)&This->ICorRuntimeHost_iface;
1572         IUnknown_AddRef(unk);
1573     }
1574     else if (IsEqualGUID(clsid, &CLSID_CLRRuntimeHost))
1575     {
1576         unk = (IUnknown*)&This->ICLRRuntimeHost_iface;
1577         IUnknown_AddRef(unk);
1578     }
1579     else if (IsEqualGUID(clsid, &CLSID_CorMetaDataDispenser) ||
1580              IsEqualGUID(clsid, &CLSID_CorMetaDataDispenserRuntime))
1581     {
1582         hr = MetaDataDispenser_CreateInstance(&unk);
1583         if (FAILED(hr))
1584             return hr;
1585     }
1586     else if (IsEqualGUID(clsid, &CLSID_CLRDebuggingLegacy))
1587     {
1588         hr = CorDebug_Create(&This->ICLRRuntimeHost_iface, &unk);
1589         if (FAILED(hr))
1590             return hr;
1591     }
1592     else
1593         unk = NULL;
1594
1595     if (unk)
1596     {
1597         hr = IUnknown_QueryInterface(unk, riid, ppv);
1598
1599         IUnknown_Release(unk);
1600
1601         return hr;
1602     }
1603     else
1604         FIXME("not implemented for class %s\n", debugstr_guid(clsid));
1605
1606     return CLASS_E_CLASSNOTAVAILABLE;
1607 }
1608
1609 #define CHARS_IN_GUID 39
1610
1611 HRESULT create_monodata(REFIID riid, LPVOID *ppObj )
1612 {
1613     static const WCHAR wszAssembly[] = {'A','s','s','e','m','b','l','y',0};
1614     static const WCHAR wszCodebase[] = {'C','o','d','e','B','a','s','e',0};
1615     static const WCHAR wszClass[] = {'C','l','a','s','s',0};
1616     static const WCHAR wszFileSlash[] = {'f','i','l','e',':','/','/','/',0};
1617     static const WCHAR wszCLSIDSlash[] = {'C','L','S','I','D','\\',0};
1618     static const WCHAR wszInprocServer32[] = {'\\','I','n','p','r','o','c','S','e','r','v','e','r','3','2',0};
1619     static const WCHAR wszDLL[] = {'.','d','l','l',0};
1620     WCHAR path[CHARS_IN_GUID + ARRAY_SIZE(wszCLSIDSlash) + ARRAY_SIZE(wszInprocServer32) - 1];
1621     MonoDomain *domain;
1622     MonoAssembly *assembly;
1623     ICLRRuntimeInfo *info = NULL;
1624     RuntimeHost *host;
1625     HRESULT hr;
1626     HKEY key, subkey;
1627     LONG res;
1628     int offset = 0;
1629     DWORD numKeys, keyLength;
1630     WCHAR codebase[MAX_PATH + 8];
1631     WCHAR classname[350], subkeyName[256];
1632     WCHAR filename[MAX_PATH];
1633
1634     DWORD dwBufLen = 350;
1635
1636     lstrcpyW(path, wszCLSIDSlash);
1637     StringFromGUID2(riid, path + lstrlenW(wszCLSIDSlash), CHARS_IN_GUID);
1638     lstrcatW(path, wszInprocServer32);
1639
1640     TRACE("Registry key: %s\n", debugstr_w(path));
1641
1642     res = RegOpenKeyExW(HKEY_CLASSES_ROOT, path, 0, KEY_READ, &key);
1643     if (res == ERROR_FILE_NOT_FOUND)
1644         return CLASS_E_CLASSNOTAVAILABLE;
1645
1646     res = RegGetValueW( key, NULL, wszClass, RRF_RT_REG_SZ, NULL, classname, &dwBufLen);
1647     if(res != ERROR_SUCCESS)
1648     {
1649         WARN("Class value cannot be found.\n");
1650         hr = CLASS_E_CLASSNOTAVAILABLE;
1651         goto cleanup;
1652     }
1653
1654     TRACE("classname (%s)\n", debugstr_w(classname));
1655
1656     dwBufLen = MAX_PATH + 8;
1657     res = RegGetValueW( key, NULL, wszCodebase, RRF_RT_REG_SZ, NULL, codebase, &dwBufLen);
1658     if(res == ERROR_SUCCESS)
1659     {
1660         /* Strip file:/// */
1661         if(strncmpW(codebase, wszFileSlash, strlenW(wszFileSlash)) == 0)
1662             offset = strlenW(wszFileSlash);
1663
1664         strcpyW(filename, codebase + offset);
1665     }
1666     else
1667     {
1668         WCHAR assemblyname[MAX_PATH + 8];
1669
1670         hr = CLASS_E_CLASSNOTAVAILABLE;
1671         WARN("CodeBase value cannot be found, trying Assembly.\n");
1672         /* get the last subkey of InprocServer32 */
1673         res = RegQueryInfoKeyW(key, 0, 0, 0, &numKeys, 0, 0, 0, 0, 0, 0, 0);
1674         if (res != ERROR_SUCCESS || numKeys == 0)
1675             goto cleanup;
1676         numKeys--;
1677         keyLength = ARRAY_SIZE(subkeyName);
1678         res = RegEnumKeyExW(key, numKeys, subkeyName, &keyLength, 0, 0, 0, 0);
1679         if (res != ERROR_SUCCESS)
1680             goto cleanup;
1681         res = RegOpenKeyExW(key, subkeyName, 0, KEY_READ, &subkey);
1682         if (res != ERROR_SUCCESS)
1683             goto cleanup;
1684         dwBufLen = MAX_PATH + 8;
1685         res = RegGetValueW(subkey, NULL, wszAssembly, RRF_RT_REG_SZ, NULL, assemblyname, &dwBufLen);
1686         RegCloseKey(subkey);
1687         if (res != ERROR_SUCCESS)
1688             goto cleanup;
1689
1690         hr = get_file_from_strongname(assemblyname, filename, MAX_PATH);
1691         if (FAILED(hr))
1692         {
1693             /*
1694              * The registry doesn't have a CodeBase entry and it's not in the GAC.
1695              *
1696              * Use the Assembly Key to retrieve the filename.
1697              *    Assembly : REG_SZ : AssemblyName, Version=X.X.X.X, Culture=neutral, PublicKeyToken=null
1698              */
1699             WCHAR *ns;
1700
1701             WARN("Attempt to load from the application directory.\n");
1702             GetModuleFileNameW(NULL, filename, MAX_PATH);
1703             ns = strrchrW(filename, '\\');
1704             *(ns+1) = '\0';
1705
1706             ns = strchrW(assemblyname, ',');
1707             *(ns) = '\0';
1708             strcatW(filename, assemblyname);
1709             *(ns) = '.';
1710             strcatW(filename, wszDLL);
1711         }
1712     }
1713
1714     TRACE("filename (%s)\n", debugstr_w(filename));
1715
1716     *ppObj = NULL;
1717
1718
1719     hr = get_runtime_info(filename, NULL, NULL, 0, 0, FALSE, &info);
1720     if (SUCCEEDED(hr))
1721     {
1722         hr = ICLRRuntimeInfo_GetRuntimeHost(info, &host);
1723
1724         if (SUCCEEDED(hr))
1725             hr = RuntimeHost_GetDefaultDomain(host, NULL, &domain);
1726
1727         if (SUCCEEDED(hr))
1728         {
1729             MonoImage *image;
1730             MonoClass *klass;
1731             MonoObject *result;
1732             MonoDomain *prev_domain;
1733             IUnknown *unk = NULL;
1734             char *filenameA, *ns;
1735             char *classA;
1736
1737             hr = CLASS_E_CLASSNOTAVAILABLE;
1738
1739             prev_domain = domain_attach(domain);
1740
1741             filenameA = WtoA(filename);
1742             assembly = mono_domain_assembly_open(domain, filenameA);
1743             HeapFree(GetProcessHeap(), 0, filenameA);
1744             if (!assembly)
1745             {
1746                 ERR("Cannot open assembly %s\n", filenameA);
1747                 domain_restore(prev_domain);
1748                 goto cleanup;
1749             }
1750
1751             image = mono_assembly_get_image(assembly);
1752             if (!image)
1753             {
1754                 ERR("Couldn't get assembly image\n");
1755                 domain_restore(prev_domain);
1756                 goto cleanup;
1757             }
1758
1759             classA = WtoA(classname);
1760             ns = strrchr(classA, '.');
1761             *ns = '\0';
1762
1763             klass = mono_class_from_name(image, classA, ns+1);
1764             HeapFree(GetProcessHeap(), 0, classA);
1765             if (!klass)
1766             {
1767                 ERR("Couldn't get class from image\n");
1768                 domain_restore(prev_domain);
1769                 goto cleanup;
1770             }
1771
1772             /*
1773              * Use the default constructor for the .NET class.
1774              */
1775             result = mono_object_new(domain, klass);
1776             mono_runtime_object_init(result);
1777
1778             hr = RuntimeHost_GetIUnknownForObject(host, result, &unk);
1779             if (SUCCEEDED(hr))
1780             {
1781                 hr = IUnknown_QueryInterface(unk, &IID_IUnknown, ppObj);
1782
1783                 IUnknown_Release(unk);
1784             }
1785             else
1786                 hr = CLASS_E_CLASSNOTAVAILABLE;
1787
1788             domain_restore(prev_domain);
1789         }
1790         else
1791             hr = CLASS_E_CLASSNOTAVAILABLE;
1792     }
1793     else
1794         hr = CLASS_E_CLASSNOTAVAILABLE;
1795
1796 cleanup:
1797     if(info)
1798         ICLRRuntimeInfo_Release(info);
1799
1800     RegCloseKey(key);
1801
1802     return hr;
1803 }