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 = sizeof(config_dir)/sizeof(*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, sizeof(base_dir) / sizeof(*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 = 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         return -1;
1427
1428     FixupVTable(GetModuleHandleW(NULL));
1429
1430     hr = get_runtime_info(filename, NULL, NULL, 0, 0, FALSE, &info);
1431
1432     if (SUCCEEDED(hr))
1433     {
1434         hr = ICLRRuntimeInfo_GetRuntimeHost(info, &host);
1435
1436         if (SUCCEEDED(hr))
1437         {
1438             WCHAR config_file[MAX_PATH];
1439             static const WCHAR dotconfig[] = {'.','c','o','n','f','i','g',0};
1440
1441             strcpyW(config_file, filename);
1442             strcatW(config_file, dotconfig);
1443
1444             hr = RuntimeHost_GetDefaultDomain(host, config_file, &domain);
1445         }
1446
1447         if (SUCCEEDED(hr))
1448         {
1449             image = mono_image_open_from_module_handle(GetModuleHandleW(NULL),
1450                 filenameA, 1, &status);
1451
1452             if (image)
1453                 assembly = mono_assembly_load_from(image, filenameA, &status);
1454
1455             if (assembly)
1456             {
1457                 mono_trace_set_assembly(assembly);
1458
1459                 exit_code = mono_jit_exec(domain, assembly, argc, argv);
1460             }
1461             else
1462             {
1463                 ERR("couldn't load %s, status=%d\n", debugstr_w(filename), status);
1464                 exit_code = -1;
1465             }
1466
1467             RuntimeHost_DeleteDomain(host, domain);
1468         }
1469         else
1470             exit_code = -1;
1471
1472         ICLRRuntimeInfo_Release(info);
1473     }
1474     else
1475         exit_code = -1;
1476
1477     HeapFree(GetProcessHeap(), 0, argv);
1478
1479     if (domain)
1480     {
1481         mono_thread_manage();
1482         mono_runtime_quit();
1483     }
1484
1485     return exit_code;
1486 }
1487
1488 BOOL WINAPI _CorDllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved)
1489 {
1490     ASSEMBLY *assembly=NULL;
1491     HRESULT hr;
1492
1493     TRACE("(%p, %d, %p)\n", hinstDLL, fdwReason, lpvReserved);
1494
1495     hr = assembly_from_hmodule(&assembly, hinstDLL);
1496     if (SUCCEEDED(hr))
1497     {
1498         NativeEntryPointFunc NativeEntryPoint=NULL;
1499
1500         assembly_get_native_entrypoint(assembly, &NativeEntryPoint);
1501         if (fdwReason == DLL_PROCESS_ATTACH)
1502         {
1503             if (!NativeEntryPoint)
1504                 DisableThreadLibraryCalls(hinstDLL);
1505             FixupVTable_Assembly(hinstDLL,assembly);
1506         }
1507         assembly_release(assembly);
1508         /* FIXME: clean up the vtables on DLL_PROCESS_DETACH */
1509         if (NativeEntryPoint)
1510             return NativeEntryPoint(hinstDLL, fdwReason, lpvReserved);
1511     }
1512     else
1513         ERR("failed to read CLR headers, hr=%x\n", hr);
1514
1515     return TRUE;
1516 }
1517
1518 /* called from DLL_PROCESS_ATTACH */
1519 void runtimehost_init(void)
1520 {
1521     dll_fixup_heap = HeapCreate(HEAP_CREATE_ENABLE_EXECUTE, 0, 0);
1522     list_init(&dll_fixups);
1523 }
1524
1525 /* called from DLL_PROCESS_DETACH */
1526 void runtimehost_uninit(void)
1527 {
1528     struct dll_fixup *fixup, *fixup2;
1529
1530     HeapDestroy(dll_fixup_heap);
1531     LIST_FOR_EACH_ENTRY_SAFE(fixup, fixup2, &dll_fixups, struct dll_fixup, entry)
1532     {
1533         HeapFree(GetProcessHeap(), 0, fixup->tokens);
1534         HeapFree(GetProcessHeap(), 0, fixup);
1535     }
1536 }
1537
1538 HRESULT RuntimeHost_Construct(CLRRuntimeInfo *runtime_version, RuntimeHost** result)
1539 {
1540     RuntimeHost *This;
1541
1542     This = HeapAlloc( GetProcessHeap(), 0, sizeof *This );
1543     if ( !This )
1544         return E_OUTOFMEMORY;
1545
1546     This->ICorRuntimeHost_iface.lpVtbl = &corruntimehost_vtbl;
1547     This->ICLRRuntimeHost_iface.lpVtbl = &CLRHostVtbl;
1548
1549     This->ref = 1;
1550     This->version = runtime_version;
1551     list_init(&This->domains);
1552     This->default_domain = NULL;
1553     InitializeCriticalSection(&This->lock);
1554     This->lock.DebugInfo->Spare[0] = (DWORD_PTR)(__FILE__ ": RuntimeHost.lock");
1555
1556     *result = This;
1557
1558     return S_OK;
1559 }
1560
1561 HRESULT RuntimeHost_GetInterface(RuntimeHost *This, REFCLSID clsid, REFIID riid, void **ppv)
1562 {
1563     IUnknown *unk;
1564     HRESULT hr;
1565
1566     if (IsEqualGUID(clsid, &CLSID_CorRuntimeHost))
1567     {
1568         unk = (IUnknown*)&This->ICorRuntimeHost_iface;
1569         IUnknown_AddRef(unk);
1570     }
1571     else if (IsEqualGUID(clsid, &CLSID_CLRRuntimeHost))
1572     {
1573         unk = (IUnknown*)&This->ICLRRuntimeHost_iface;
1574         IUnknown_AddRef(unk);
1575     }
1576     else if (IsEqualGUID(clsid, &CLSID_CorMetaDataDispenser) ||
1577              IsEqualGUID(clsid, &CLSID_CorMetaDataDispenserRuntime))
1578     {
1579         hr = MetaDataDispenser_CreateInstance(&unk);
1580         if (FAILED(hr))
1581             return hr;
1582     }
1583     else if (IsEqualGUID(clsid, &CLSID_CLRDebuggingLegacy))
1584     {
1585         hr = CorDebug_Create(&This->ICLRRuntimeHost_iface, &unk);
1586         if (FAILED(hr))
1587             return hr;
1588     }
1589     else
1590         unk = NULL;
1591
1592     if (unk)
1593     {
1594         hr = IUnknown_QueryInterface(unk, riid, ppv);
1595
1596         IUnknown_Release(unk);
1597
1598         return hr;
1599     }
1600     else
1601         FIXME("not implemented for class %s\n", debugstr_guid(clsid));
1602
1603     return CLASS_E_CLASSNOTAVAILABLE;
1604 }
1605
1606 #define CHARS_IN_GUID 39
1607 #define ARRAYSIZE(array) (sizeof(array)/sizeof((array)[0]))
1608
1609 HRESULT create_monodata(REFIID riid, LPVOID *ppObj )
1610 {
1611     static const WCHAR wszAssembly[] = {'A','s','s','e','m','b','l','y',0};
1612     static const WCHAR wszCodebase[] = {'C','o','d','e','B','a','s','e',0};
1613     static const WCHAR wszClass[] = {'C','l','a','s','s',0};
1614     static const WCHAR wszFileSlash[] = {'f','i','l','e',':','/','/','/',0};
1615     static const WCHAR wszCLSIDSlash[] = {'C','L','S','I','D','\\',0};
1616     static const WCHAR wszInprocServer32[] = {'\\','I','n','p','r','o','c','S','e','r','v','e','r','3','2',0};
1617     static const WCHAR wszDLL[] = {'.','d','l','l',0};
1618     WCHAR path[CHARS_IN_GUID + ARRAYSIZE(wszCLSIDSlash) + ARRAYSIZE(wszInprocServer32) - 1];
1619     MonoDomain *domain;
1620     MonoAssembly *assembly;
1621     ICLRRuntimeInfo *info = NULL;
1622     RuntimeHost *host;
1623     HRESULT hr;
1624     HKEY key, subkey;
1625     LONG res;
1626     int offset = 0;
1627     DWORD numKeys, keyLength;
1628     WCHAR codebase[MAX_PATH + 8];
1629     WCHAR classname[350], subkeyName[256];
1630     WCHAR filename[MAX_PATH];
1631
1632     DWORD dwBufLen = 350;
1633
1634     lstrcpyW(path, wszCLSIDSlash);
1635     StringFromGUID2(riid, path + lstrlenW(wszCLSIDSlash), CHARS_IN_GUID);
1636     lstrcatW(path, wszInprocServer32);
1637
1638     TRACE("Registry key: %s\n", debugstr_w(path));
1639
1640     res = RegOpenKeyExW(HKEY_CLASSES_ROOT, path, 0, KEY_READ, &key);
1641     if (res == ERROR_FILE_NOT_FOUND)
1642         return CLASS_E_CLASSNOTAVAILABLE;
1643
1644     res = RegGetValueW( key, NULL, wszClass, RRF_RT_REG_SZ, NULL, classname, &dwBufLen);
1645     if(res != ERROR_SUCCESS)
1646     {
1647         WARN("Class value cannot be found.\n");
1648         hr = CLASS_E_CLASSNOTAVAILABLE;
1649         goto cleanup;
1650     }
1651
1652     TRACE("classname (%s)\n", debugstr_w(classname));
1653
1654     dwBufLen = MAX_PATH + 8;
1655     res = RegGetValueW( key, NULL, wszCodebase, RRF_RT_REG_SZ, NULL, codebase, &dwBufLen);
1656     if(res == ERROR_SUCCESS)
1657     {
1658         /* Strip file:/// */
1659         if(strncmpW(codebase, wszFileSlash, strlenW(wszFileSlash)) == 0)
1660             offset = strlenW(wszFileSlash);
1661
1662         strcpyW(filename, codebase + offset);
1663     }
1664     else
1665     {
1666         WCHAR assemblyname[MAX_PATH + 8];
1667
1668         hr = CLASS_E_CLASSNOTAVAILABLE;
1669         WARN("CodeBase value cannot be found, trying Assembly.\n");
1670         /* get the last subkey of InprocServer32 */
1671         res = RegQueryInfoKeyW(key, 0, 0, 0, &numKeys, 0, 0, 0, 0, 0, 0, 0);
1672         if (res != ERROR_SUCCESS || numKeys == 0)
1673             goto cleanup;
1674         numKeys--;
1675         keyLength = sizeof(subkeyName) / sizeof(WCHAR);
1676         res = RegEnumKeyExW(key, numKeys, subkeyName, &keyLength, 0, 0, 0, 0);
1677         if (res != ERROR_SUCCESS)
1678             goto cleanup;
1679         res = RegOpenKeyExW(key, subkeyName, 0, KEY_READ, &subkey);
1680         if (res != ERROR_SUCCESS)
1681             goto cleanup;
1682         dwBufLen = MAX_PATH + 8;
1683         res = RegGetValueW(subkey, NULL, wszAssembly, RRF_RT_REG_SZ, NULL, assemblyname, &dwBufLen);
1684         RegCloseKey(subkey);
1685         if (res != ERROR_SUCCESS)
1686             goto cleanup;
1687
1688         hr = get_file_from_strongname(assemblyname, filename, MAX_PATH);
1689         if (FAILED(hr))
1690         {
1691             /*
1692              * The registry doesn't have a CodeBase entry and it's not in the GAC.
1693              *
1694              * Use the Assembly Key to retrieve the filename.
1695              *    Assembly : REG_SZ : AssemblyName, Version=X.X.X.X, Culture=neutral, PublicKeyToken=null
1696              */
1697             WCHAR *ns;
1698
1699             WARN("Attempt to load from the application directory.\n");
1700             GetModuleFileNameW(NULL, filename, MAX_PATH);
1701             ns = strrchrW(filename, '\\');
1702             *(ns+1) = '\0';
1703
1704             ns = strchrW(assemblyname, ',');
1705             *(ns) = '\0';
1706             strcatW(filename, assemblyname);
1707             *(ns) = '.';
1708             strcatW(filename, wszDLL);
1709         }
1710     }
1711
1712     TRACE("filename (%s)\n", debugstr_w(filename));
1713
1714     *ppObj = NULL;
1715
1716
1717     hr = get_runtime_info(filename, NULL, NULL, 0, 0, FALSE, &info);
1718     if (SUCCEEDED(hr))
1719     {
1720         hr = ICLRRuntimeInfo_GetRuntimeHost(info, &host);
1721
1722         if (SUCCEEDED(hr))
1723             hr = RuntimeHost_GetDefaultDomain(host, NULL, &domain);
1724
1725         if (SUCCEEDED(hr))
1726         {
1727             MonoImage *image;
1728             MonoClass *klass;
1729             MonoObject *result;
1730             MonoDomain *prev_domain;
1731             IUnknown *unk = NULL;
1732             char *filenameA, *ns;
1733             char *classA;
1734
1735             hr = CLASS_E_CLASSNOTAVAILABLE;
1736
1737             prev_domain = domain_attach(domain);
1738
1739             filenameA = WtoA(filename);
1740             assembly = mono_domain_assembly_open(domain, filenameA);
1741             HeapFree(GetProcessHeap(), 0, filenameA);
1742             if (!assembly)
1743             {
1744                 ERR("Cannot open assembly %s\n", filenameA);
1745                 domain_restore(prev_domain);
1746                 goto cleanup;
1747             }
1748
1749             image = mono_assembly_get_image(assembly);
1750             if (!image)
1751             {
1752                 ERR("Couldn't get assembly image\n");
1753                 domain_restore(prev_domain);
1754                 goto cleanup;
1755             }
1756
1757             classA = WtoA(classname);
1758             ns = strrchr(classA, '.');
1759             *ns = '\0';
1760
1761             klass = mono_class_from_name(image, classA, ns+1);
1762             HeapFree(GetProcessHeap(), 0, classA);
1763             if (!klass)
1764             {
1765                 ERR("Couldn't get class from image\n");
1766                 domain_restore(prev_domain);
1767                 goto cleanup;
1768             }
1769
1770             /*
1771              * Use the default constructor for the .NET class.
1772              */
1773             result = mono_object_new(domain, klass);
1774             mono_runtime_object_init(result);
1775
1776             hr = RuntimeHost_GetIUnknownForObject(host, result, &unk);
1777             if (SUCCEEDED(hr))
1778             {
1779                 hr = IUnknown_QueryInterface(unk, &IID_IUnknown, ppObj);
1780
1781                 IUnknown_Release(unk);
1782             }
1783             else
1784                 hr = CLASS_E_CLASSNOTAVAILABLE;
1785
1786             domain_restore(prev_domain);
1787         }
1788         else
1789             hr = CLASS_E_CLASSNOTAVAILABLE;
1790     }
1791     else
1792         hr = CLASS_E_CLASSNOTAVAILABLE;
1793
1794 cleanup:
1795     if(info)
1796         ICLRRuntimeInfo_Release(info);
1797
1798     RegCloseKey(key);
1799
1800     return hr;
1801 }