mono/metadata/threadpool.c

   1 /**
   2  * \file
   3  * Microsoft threadpool runtime support
   4  *
   5  * Author:
   6  *      Ludovic Henry (ludovic.henry@xamarin.com)
   7  *
   8  * Copyright 2015 Xamarin, Inc (http://www.xamarin.com)
   9  * Licensed under the MIT license. See LICENSE file in the project root for full license information.
  10  */
  11
  12 //
  13 // Copyright (c) Microsoft. All rights reserved.
  14 // Licensed under the MIT license. See LICENSE file in the project root for full license information.
  15 //
  16 // Files:
  17 //  - src/vm/comthreadpool.cpp
  18 //  - src/vm/win32threadpoolcpp
  19 //  - src/vm/threadpoolrequest.cpp
  20 //  - src/vm/hillclimbing.cpp
  21 //
  22 // Ported from C++ to C and adjusted to Mono runtime
  23
  24 #include <stdlib.h>
  25 #define _USE_MATH_DEFINES // needed by MSVC to define math constants
  26 #include <math.h>
  27 #include <config.h>
  28 #include <glib.h>
  29
  30 #include <mono/metadata/class-internals.h>
  31 #include <mono/metadata/exception.h>
  32 #include <mono/metadata/gc-internals.h>
  33 #include <mono/metadata/object.h>
  34 #include <mono/metadata/object-internals.h>
  35 #include <mono/metadata/threadpool.h>
  36 #include <mono/metadata/threadpool-worker.h>
  37 #include <mono/metadata/threadpool-io.h>
  38 #include <mono/metadata/w32event.h>
  39 #include <mono/utils/atomic.h>
  40 #include <mono/utils/mono-compiler.h>
  41 #include <mono/utils/mono-complex.h>
  42 #include <mono/utils/mono-lazy-init.h>
  43 #include <mono/utils/mono-logger.h>
  44 #include <mono/utils/mono-logger-internals.h>
  45 #include <mono/utils/mono-proclib.h>
  46 #include <mono/utils/mono-threads.h>
  47 #include <mono/utils/mono-time.h>
  48 #include <mono/utils/refcount.h>
  49
  50 typedef struct {
  51         MonoDomain *domain;
  52         /* Number of outstanding jobs */
  53         gint32 outstanding_request;
  54         /* Number of currently executing jobs */
  55         gint32 threadpool_jobs;
  56         /* Signalled when threadpool_jobs + outstanding_request is 0 */
  57         /* Protected by threadpool.domains_lock */
  58         MonoCoopCond cleanup_cond;
  59 } ThreadPoolDomain;
  60
  61 typedef union {
  62         struct {
  63                 gint16 starting; /* starting, but not yet in worker_callback */
  64                 gint16 working; /* executing worker_callback */
  65         } _;
  66         gint32 as_gint32;
  67 } ThreadPoolCounter;
  68
  69 typedef struct {
  70         MonoRefCount ref;
  71
  72         GPtrArray *domains; // ThreadPoolDomain* []
  73         MonoCoopMutex domains_lock;
  74
  75         ThreadPoolCounter counters;
  76
  77         gint32 limit_io_min;
  78         gint32 limit_io_max;
  79 } ThreadPool;
  80
  81 static mono_lazy_init_t status = MONO_LAZY_INIT_STATUS_NOT_INITIALIZED;
  82
  83 static ThreadPool threadpool;
  84
  85 #define COUNTER_ATOMIC(var,block) \
  86         do { \
  87                 ThreadPoolCounter __old; \
  88                 do { \
  89                         (var) = __old = COUNTER_READ (); \
  90                         { block; } \
  91                         if (!(counter._.starting >= 0)) \
  92                                 g_error ("%s: counter._.starting = %d, but should be >= 0", __func__, counter._.starting); \
  93                         if (!(counter._.working >= 0)) \
  94                                 g_error ("%s: counter._.working = %d, but should be >= 0", __func__, counter._.working); \
  95                 } while (InterlockedCompareExchange (&threadpool.counters.as_gint32, (var).as_gint32, __old.as_gint32) != __old.as_gint32); \
  96         } while (0)
  97
  98 static inline ThreadPoolCounter
  99 COUNTER_READ (void)
 100 {
 101         ThreadPoolCounter counter;
 102         counter.as_gint32 = InterlockedRead (&threadpool.counters.as_gint32);
 103         return counter;
 104 }
 105
 106 static inline void
 107 domains_lock (void)
 108 {
 109         mono_coop_mutex_lock (&threadpool.domains_lock);
 110 }
 111
 112 static inline void
 113 domains_unlock (void)
 114 {
 115         mono_coop_mutex_unlock (&threadpool.domains_lock);
 116 }
 117
 118 static void
 119 destroy (gpointer unused)
 120 {
 121         g_ptr_array_free (threadpool.domains, TRUE);
 122         mono_coop_mutex_destroy (&threadpool.domains_lock);
 123 }
 124
 125 static void
 126 worker_callback (void);
 127
 128 static void
 129 initialize (void)
 130 {
 131         g_assert (sizeof (ThreadPoolCounter) == sizeof (gint32));
 132
 133         mono_refcount_init (&threadpool, destroy);
 134
 135         threadpool.domains = g_ptr_array_new ();
 136         mono_coop_mutex_init (&threadpool.domains_lock);
 137
 138         threadpool.limit_io_min = mono_cpu_count ();
 139         threadpool.limit_io_max = CLAMP (threadpool.limit_io_min * 100, MIN (threadpool.limit_io_min, 200), MAX (threadpool.limit_io_min, 200));
 140
 141         mono_threadpool_worker_init (worker_callback);
 142 }
 143
 144 static void
 145 cleanup (void)
 146 {
 147         mono_threadpool_worker_cleanup ();
 148
 149         mono_refcount_dec (&threadpool);
 150 }
 151
 152 gboolean
 153 mono_threadpool_enqueue_work_item (MonoDomain *domain, MonoObject *work_item, MonoError *error)
 154 {
 155         static MonoClass *threadpool_class = NULL;
 156         static MonoMethod *unsafe_queue_custom_work_item_method = NULL;
 157         MonoDomain *current_domain;
 158         MonoBoolean f;
 159         gpointer args [2];
 160
 161         error_init (error);
 162         g_assert (work_item);
 163
 164         if (!threadpool_class)
 165                 threadpool_class = mono_class_load_from_name (mono_defaults.corlib, "System.Threading", "ThreadPool");
 166
 167         if (!unsafe_queue_custom_work_item_method)
 168                 unsafe_queue_custom_work_item_method = mono_class_get_method_from_name (threadpool_class, "UnsafeQueueCustomWorkItem", 2);
 169         g_assert (unsafe_queue_custom_work_item_method);
 170
 171         f = FALSE;
 172
 173         args [0] = (gpointer) work_item;
 174         args [1] = (gpointer) &f;
 175
 176         current_domain = mono_domain_get ();
 177         if (current_domain == domain) {
 178                 mono_runtime_invoke_checked (unsafe_queue_custom_work_item_method, NULL, args, error);
 179                 return_val_if_nok (error, FALSE);
 180         } else {
 181                 mono_thread_push_appdomain_ref (domain);
 182                 if (mono_domain_set (domain, FALSE)) {
 183                         mono_runtime_invoke_checked (unsafe_queue_custom_work_item_method, NULL, args, error);
 184                         if (!is_ok (error)) {
 185                                 mono_thread_pop_appdomain_ref ();
 186                                 return FALSE;
 187                         }
 188                         mono_domain_set (current_domain, TRUE);
 189                 }
 190                 mono_thread_pop_appdomain_ref ();
 191         }
 192         return TRUE;
 193 }
 194
 195 /* LOCKING: domains_lock must be held. */
 196 static ThreadPoolDomain *
 197 tpdomain_create (MonoDomain *domain)
 198 {
 199         ThreadPoolDomain *tpdomain;
 200
 201         tpdomain = g_new0 (ThreadPoolDomain, 1);
 202         tpdomain->domain = domain;
 203         mono_coop_cond_init (&tpdomain->cleanup_cond);
 204
 205         g_ptr_array_add (threadpool.domains, tpdomain);
 206
 207         return tpdomain;
 208 }
 209
 210 /* LOCKING: domains_lock must be held. */
 211 static gboolean
 212 tpdomain_remove (ThreadPoolDomain *tpdomain)
 213 {
 214         g_assert (tpdomain);
 215         return g_ptr_array_remove (threadpool.domains, tpdomain);
 216 }
 217
 218 /* LOCKING: domains_lock must be held */
 219 static ThreadPoolDomain *
 220 tpdomain_get (MonoDomain *domain)
 221 {
 222         gint i;
 223
 224         g_assert (domain);
 225
 226         for (i = 0; i < threadpool.domains->len; ++i) {
 227                 ThreadPoolDomain *tpdomain;
 228
 229                 tpdomain = (ThreadPoolDomain *)g_ptr_array_index (threadpool.domains, i);
 230                 if (tpdomain->domain == domain)
 231                         return tpdomain;
 232         }
 233
 234         return NULL;
 235 }
 236
 237 static void
 238 tpdomain_free (ThreadPoolDomain *tpdomain)
 239 {
 240         g_free (tpdomain);
 241 }
 242
 243 /* LOCKING: domains_lock must be held */
 244 static ThreadPoolDomain *
 245 tpdomain_get_next (ThreadPoolDomain *current)
 246 {
 247         ThreadPoolDomain *tpdomain = NULL;
 248         gint len;
 249
 250         len = threadpool.domains->len;
 251         if (len > 0) {
 252                 gint i, current_idx = -1;
 253                 if (current) {
 254                         for (i = 0; i < len; ++i) {
 255                                 if (current == g_ptr_array_index (threadpool.domains, i)) {
 256                                         current_idx = i;
 257                                         break;
 258                                 }
 259                         }
 260                 }
 261                 for (i = current_idx + 1; i < len + current_idx + 1; ++i) {
 262                         ThreadPoolDomain *tmp = (ThreadPoolDomain *)g_ptr_array_index (threadpool.domains, i % len);
 263                         if (tmp->outstanding_request > 0) {
 264                                 tpdomain = tmp;
 265                                 break;
 266                         }
 267                 }
 268         }
 269
 270         return tpdomain;
 271 }
 272
 273 static MonoObject*
 274 try_invoke_perform_wait_callback (MonoObject** exc, MonoError *error)
 275 {
 276         HANDLE_FUNCTION_ENTER ();
 277         error_init (error);
 278         MonoObject *res = mono_runtime_try_invoke (mono_defaults.threadpool_perform_wait_callback_method, NULL, NULL, exc, error);
 279         HANDLE_FUNCTION_RETURN_VAL (res);
 280 }
 281
 282 static void
 283 worker_callback (void)
 284 {
 285         MonoError error;
 286         ThreadPoolDomain *tpdomain, *previous_tpdomain;
 287         ThreadPoolCounter counter;
 288         MonoInternalThread *thread;
 289
 290         if (!mono_refcount_tryinc (&threadpool))
 291                 return;
 292
 293         thread = mono_thread_internal_current ();
 294
 295         COUNTER_ATOMIC (counter, {
 296                 if (!(counter._.working < 32767 /* G_MAXINT16 */))
 297                         g_error ("%s: counter._.working = %d, but should be < 32767", __func__, counter._.working);
 298
 299                 counter._.starting --;
 300                 counter._.working ++;
 301         });
 302
 303         if (mono_runtime_is_shutting_down ()) {
 304                 COUNTER_ATOMIC (counter, {
 305                         counter._.working --;
 306                 });
 307
 308                 mono_refcount_dec (&threadpool);
 309                 return;
 310         }
 311
 312         /*
 313          * This is needed so there is always an lmf frame in the runtime invoke call below,
 314          * so ThreadAbortExceptions are caught even if the thread is in native code.
 315          */
 316         mono_defaults.threadpool_perform_wait_callback_method->save_lmf = TRUE;
 317
 318         domains_lock ();
 319
 320         previous_tpdomain = NULL;
 321
 322         while (!mono_runtime_is_shutting_down ()) {
 323                 gboolean retire = FALSE;
 324
 325                 if (thread->state & (ThreadState_AbortRequested | ThreadState_SuspendRequested)) {
 326                         domains_unlock ();
 327                         if (mono_thread_interruption_checkpoint ()) {
 328                                 domains_lock ();
 329                                 continue;
 330                         }
 331                         domains_lock ();
 332                 }
 333
 334                 tpdomain = tpdomain_get_next (previous_tpdomain);
 335                 if (!tpdomain)
 336                         break;
 337
 338                 tpdomain->outstanding_request --;
 339                 g_assert (tpdomain->outstanding_request >= 0);
 340
 341                 mono_trace (G_LOG_LEVEL_DEBUG, MONO_TRACE_THREADPOOL, "[%p] worker running in domain %p (outstanding requests %d)",
 342                         mono_native_thread_id_get (), tpdomain->domain, tpdomain->outstanding_request);
 343
 344                 g_assert (tpdomain->threadpool_jobs >= 0);
 345                 tpdomain->threadpool_jobs ++;
 346
 347                 domains_unlock ();
 348
 349                 mono_thread_set_name_internal (thread, mono_string_new (mono_get_root_domain (), "Threadpool worker"), FALSE, TRUE, &error);
 350                 mono_error_assert_ok (&error);
 351
 352                 mono_thread_clr_state (thread, (MonoThreadState)~ThreadState_Background);
 353                 if (!mono_thread_test_state (thread , ThreadState_Background))
 354                         ves_icall_System_Threading_Thread_SetState (thread, ThreadState_Background);
 355
 356                 mono_thread_push_appdomain_ref (tpdomain->domain);
 357                 if (mono_domain_set (tpdomain->domain, FALSE)) {
 358                         MonoObject *exc = NULL, *res;
 359
 360                         res = try_invoke_perform_wait_callback (&exc, &error);
 361                         if (exc || !mono_error_ok(&error)) {
 362                                 if (exc == NULL)
 363                                         exc = (MonoObject *) mono_error_convert_to_exception (&error);
 364                                 else
 365                                         mono_error_cleanup (&error);
 366                                 mono_thread_internal_unhandled_exception (exc);
 367                         } else if (res && *(MonoBoolean*) mono_object_unbox (res) == FALSE) {
 368                                 retire = TRUE;
 369                         }
 370
 371                         mono_domain_set (mono_get_root_domain (), TRUE);
 372                 }
 373                 mono_thread_pop_appdomain_ref ();
 374
 375                 domains_lock ();
 376
 377                 tpdomain->threadpool_jobs --;
 378                 g_assert (tpdomain->threadpool_jobs >= 0);
 379
 380                 if (tpdomain->outstanding_request + tpdomain->threadpool_jobs == 0 && mono_domain_is_unloading (tpdomain->domain)) {
 381                         gboolean removed;
 382
 383                         removed = tpdomain_remove (tpdomain);
 384                         g_assert (removed);
 385
 386                         mono_coop_cond_signal (&tpdomain->cleanup_cond);
 387                         tpdomain = NULL;
 388                 }
 389
 390                 if (retire)
 391                         break;
 392
 393                 previous_tpdomain = tpdomain;
 394         }
 395
 396         domains_unlock ();
 397
 398         COUNTER_ATOMIC (counter, {
 399                 counter._.working --;
 400         });
 401
 402         mono_refcount_dec (&threadpool);
 403 }
 404
 405 void
 406 mono_threadpool_cleanup (void)
 407 {
 408 #ifndef DISABLE_SOCKETS
 409         mono_threadpool_io_cleanup ();
 410 #endif
 411         mono_lazy_cleanup (&status, cleanup);
 412 }
 413
 414 MonoAsyncResult *
 415 mono_threadpool_begin_invoke (MonoDomain *domain, MonoObject *target, MonoMethod *method, gpointer *params, MonoError *error)
 416 {
 417         static MonoClass *async_call_klass = NULL;
 418         MonoMethodMessage *message;
 419         MonoAsyncResult *async_result;
 420         MonoAsyncCall *async_call;
 421         MonoDelegate *async_callback = NULL;
 422         MonoObject *state = NULL;
 423
 424         if (!async_call_klass)
 425                 async_call_klass = mono_class_load_from_name (mono_defaults.corlib, "System", "MonoAsyncCall");
 426
 427         error_init (error);
 428
 429         message = mono_method_call_message_new (method, params, mono_get_delegate_invoke (method->klass), (params != NULL) ? (&async_callback) : NULL, (params != NULL) ? (&state) : NULL, error);
 430         return_val_if_nok (error, NULL);
 431
 432         async_call = (MonoAsyncCall*) mono_object_new_checked (domain, async_call_klass, error);
 433         return_val_if_nok (error, NULL);
 434
 435         MONO_OBJECT_SETREF (async_call, msg, message);
 436         MONO_OBJECT_SETREF (async_call, state, state);
 437
 438         if (async_callback) {
 439                 MONO_OBJECT_SETREF (async_call, cb_method, mono_get_delegate_invoke (((MonoObject*) async_callback)->vtable->klass));
 440                 MONO_OBJECT_SETREF (async_call, cb_target, async_callback);
 441         }
 442
 443         async_result = mono_async_result_new (domain, NULL, async_call->state, NULL, (MonoObject*) async_call, error);
 444         return_val_if_nok (error, NULL);
 445         MONO_OBJECT_SETREF (async_result, async_delegate, target);
 446
 447         mono_threadpool_enqueue_work_item (domain, (MonoObject*) async_result, error);
 448         return_val_if_nok (error, NULL);
 449
 450         return async_result;
 451 }
 452
 453 MonoObject *
 454 mono_threadpool_end_invoke (MonoAsyncResult *ares, MonoArray **out_args, MonoObject **exc, MonoError *error)
 455 {
 456         MonoAsyncCall *ac;
 457
 458         error_init (error);
 459         g_assert (exc);
 460         g_assert (out_args);
 461
 462         *exc = NULL;
 463         *out_args = NULL;
 464
 465         /* check if already finished */
 466         mono_monitor_enter ((MonoObject*) ares);
 467
 468         if (ares->endinvoke_called) {
 469                 mono_error_set_invalid_operation(error, "Delegate EndInvoke method called more than once");
 470                 mono_monitor_exit ((MonoObject*) ares);
 471                 return NULL;
 472         }
 473
 474         ares->endinvoke_called = 1;
 475
 476         /* wait until we are really finished */
 477         if (ares->completed) {
 478                 mono_monitor_exit ((MonoObject *) ares);
 479         } else {
 480                 gpointer wait_event;
 481                 if (ares->handle) {
 482                         wait_event = mono_wait_handle_get_handle ((MonoWaitHandle*) ares->handle);
 483                 } else {
 484                         wait_event = mono_w32event_create (TRUE, FALSE);
 485                         g_assert(wait_event);
 486                         MonoWaitHandle *wait_handle = mono_wait_handle_new (mono_object_domain (ares), wait_event, error);
 487                         if (!is_ok (error)) {
 488                                 mono_w32event_close (wait_event);
 489                                 return NULL;
 490                         }
 491                         MONO_OBJECT_SETREF (ares, handle, (MonoObject*) wait_handle);
 492                 }
 493                 mono_monitor_exit ((MonoObject*) ares);
 494                 MONO_ENTER_GC_SAFE;
 495 #ifdef HOST_WIN32
 496                 WaitForSingleObjectEx (wait_event, INFINITE, TRUE);
 497 #else
 498                 mono_w32handle_wait_one (wait_event, MONO_INFINITE_WAIT, TRUE);
 499 #endif
 500                 MONO_EXIT_GC_SAFE;
 501         }
 502
 503         ac = (MonoAsyncCall*) ares->object_data;
 504         g_assert (ac);
 505
 506         *exc = ac->msg->exc; /* FIXME: GC add write barrier */
 507         *out_args = ac->out_args;
 508         return ac->res;
 509 }
 510
 511 gboolean
 512 mono_threadpool_remove_domain_jobs (MonoDomain *domain, int timeout)
 513 {
 514         gint64 end;
 515         ThreadPoolDomain *tpdomain;
 516         gboolean ret;
 517
 518         g_assert (domain);
 519         g_assert (timeout >= -1);
 520
 521         g_assert (mono_domain_is_unloading (domain));
 522
 523         if (timeout != -1)
 524                 end = mono_msec_ticks () + timeout;
 525
 526 #ifndef DISABLE_SOCKETS
 527         mono_threadpool_io_remove_domain_jobs (domain);
 528         if (timeout != -1) {
 529                 if (mono_msec_ticks () > end)
 530                         return FALSE;
 531         }
 532 #endif
 533
 534         /*
 535          * Wait for all threads which execute jobs in the domain to exit.
 536          * The is_unloading () check in worker_request () ensures that
 537          * no new jobs are added after we enter the lock below.
 538          */
 539
 540         if (!mono_lazy_is_initialized (&status))
 541                 return TRUE;
 542
 543         mono_refcount_inc (&threadpool);
 544
 545         domains_lock ();
 546
 547         tpdomain = tpdomain_get (domain);
 548         if (!tpdomain) {
 549                 domains_unlock ();
 550                 mono_refcount_dec (&threadpool);
 551                 return TRUE;
 552         }
 553
 554         ret = TRUE;
 555
 556         while (tpdomain->outstanding_request + tpdomain->threadpool_jobs > 0) {
 557                 if (timeout == -1) {
 558                         mono_coop_cond_wait (&tpdomain->cleanup_cond, &threadpool.domains_lock);
 559                 } else {
 560                         gint64 now;
 561                         gint res;
 562
 563                         now = mono_msec_ticks();
 564                         if (now > end) {
 565                                 ret = FALSE;
 566                                 break;
 567                         }
 568
 569                         res = mono_coop_cond_timedwait (&tpdomain->cleanup_cond, &threadpool.domains_lock, end - now);
 570                         if (res != 0) {
 571                                 ret = FALSE;
 572                                 break;
 573                         }
 574                 }
 575         }
 576
 577         /* Remove from the list the worker threads look at */
 578         tpdomain_remove (tpdomain);
 579
 580         domains_unlock ();
 581
 582         mono_coop_cond_destroy (&tpdomain->cleanup_cond);
 583         tpdomain_free (tpdomain);
 584
 585         mono_refcount_dec (&threadpool);
 586
 587         return ret;
 588 }
 589
 590 void
 591 mono_threadpool_suspend (void)
 592 {
 593         if (mono_lazy_is_initialized (&status))
 594                 mono_threadpool_worker_set_suspended (TRUE);
 595 }
 596
 597 void
 598 mono_threadpool_resume (void)
 599 {
 600         if (mono_lazy_is_initialized (&status))
 601                 mono_threadpool_worker_set_suspended (FALSE);
 602 }
 603
 604 void
 605 ves_icall_System_Threading_ThreadPool_GetAvailableThreadsNative (gint32 *worker_threads, gint32 *completion_port_threads)
 606 {
 607         ThreadPoolCounter counter;
 608
 609         if (!worker_threads || !completion_port_threads)
 610                 return;
 611
 612         if (!mono_lazy_initialize (&status, initialize) || !mono_refcount_tryinc (&threadpool)) {
 613                 *worker_threads = 0;
 614                 *completion_port_threads = 0;
 615                 return;
 616         }
 617
 618         counter = COUNTER_READ ();
 619
 620         *worker_threads = MAX (0, mono_threadpool_worker_get_max () - counter._.working);
 621         *completion_port_threads = threadpool.limit_io_max;
 622
 623         mono_refcount_dec (&threadpool);
 624 }
 625
 626 void
 627 ves_icall_System_Threading_ThreadPool_GetMinThreadsNative (gint32 *worker_threads, gint32 *completion_port_threads)
 628 {
 629         if (!worker_threads || !completion_port_threads)
 630                 return;
 631
 632         if (!mono_lazy_initialize (&status, initialize) || !mono_refcount_tryinc (&threadpool)) {
 633                 *worker_threads = 0;
 634                 *completion_port_threads = 0;
 635                 return;
 636         }
 637
 638         *worker_threads = mono_threadpool_worker_get_min ();
 639         *completion_port_threads = threadpool.limit_io_min;
 640
 641         mono_refcount_dec (&threadpool);
 642 }
 643
 644 void
 645 ves_icall_System_Threading_ThreadPool_GetMaxThreadsNative (gint32 *worker_threads, gint32 *completion_port_threads)
 646 {
 647         if (!worker_threads || !completion_port_threads)
 648                 return;
 649
 650         if (!mono_lazy_initialize (&status, initialize) || !mono_refcount_tryinc (&threadpool)) {
 651                 *worker_threads = 0;
 652                 *completion_port_threads = 0;
 653                 return;
 654         }
 655
 656         *worker_threads = mono_threadpool_worker_get_max ();
 657         *completion_port_threads = threadpool.limit_io_max;
 658
 659         mono_refcount_dec (&threadpool);
 660 }
 661
 662 MonoBoolean
 663 ves_icall_System_Threading_ThreadPool_SetMinThreadsNative (gint32 worker_threads, gint32 completion_port_threads)
 664 {
 665         if (completion_port_threads <= 0 || completion_port_threads > threadpool.limit_io_max)
 666                 return FALSE;
 667
 668         if (!mono_lazy_initialize (&status, initialize) || !mono_refcount_tryinc (&threadpool))
 669                 return FALSE;
 670
 671         if (!mono_threadpool_worker_set_min (worker_threads)) {
 672                 mono_refcount_dec (&threadpool);
 673                 return FALSE;
 674         }
 675
 676         threadpool.limit_io_min = completion_port_threads;
 677
 678         mono_refcount_dec (&threadpool);
 679         return TRUE;
 680 }
 681
 682 MonoBoolean
 683 ves_icall_System_Threading_ThreadPool_SetMaxThreadsNative (gint32 worker_threads, gint32 completion_port_threads)
 684 {
 685         gint cpu_count = mono_cpu_count ();
 686
 687         if (completion_port_threads < threadpool.limit_io_min || completion_port_threads < cpu_count)
 688                 return FALSE;
 689
 690         if (!mono_lazy_initialize (&status, initialize) || !mono_refcount_tryinc (&threadpool))
 691                 return FALSE;
 692
 693         if (!mono_threadpool_worker_set_max (worker_threads)) {
 694                 mono_refcount_dec (&threadpool);
 695                 return FALSE;
 696         }
 697
 698         threadpool.limit_io_max = completion_port_threads;
 699
 700         mono_refcount_dec (&threadpool);
 701         return TRUE;
 702 }
 703
 704 void
 705 ves_icall_System_Threading_ThreadPool_InitializeVMTp (MonoBoolean *enable_worker_tracking)
 706 {
 707         if (enable_worker_tracking) {
 708                 // TODO implement some kind of switch to have the possibily to use it
 709                 *enable_worker_tracking = FALSE;
 710         }
 711
 712         mono_lazy_initialize (&status, initialize);
 713 }
 714
 715 MonoBoolean
 716 ves_icall_System_Threading_ThreadPool_NotifyWorkItemComplete (void)
 717 {
 718         if (mono_domain_is_unloading (mono_domain_get ()) || mono_runtime_is_shutting_down ())
 719                 return FALSE;
 720
 721         return mono_threadpool_worker_notify_completed ();
 722 }
 723
 724 void
 725 ves_icall_System_Threading_ThreadPool_NotifyWorkItemProgressNative (void)
 726 {
 727         mono_threadpool_worker_notify_completed ();
 728 }
 729
 730 void
 731 ves_icall_System_Threading_ThreadPool_ReportThreadStatus (MonoBoolean is_working)
 732 {
 733         // TODO
 734         MonoError error;
 735         mono_error_set_not_implemented (&error, "");
 736         mono_error_set_pending_exception (&error);
 737 }
 738
 739 MonoBoolean
 740 ves_icall_System_Threading_ThreadPool_RequestWorkerThread (void)
 741 {
 742         MonoDomain *domain;
 743         ThreadPoolDomain *tpdomain;
 744         ThreadPoolCounter counter;
 745
 746         domain = mono_domain_get ();
 747         if (mono_domain_is_unloading (domain))
 748                 return FALSE;
 749
 750         if (!mono_lazy_initialize (&status, initialize) || !mono_refcount_tryinc (&threadpool)) {
 751                 /* threadpool has been destroyed, we are shutting down */
 752                 return FALSE;
 753         }
 754
 755         domains_lock ();
 756
 757         tpdomain = tpdomain_get (domain);
 758         if (!tpdomain) {
 759                 /* synchronize with mono_threadpool_remove_domain_jobs */
 760                 if (mono_domain_is_unloading (domain)) {
 761                         domains_unlock ();
 762                         mono_refcount_dec (&threadpool);
 763                         return FALSE;
 764                 }
 765
 766                 tpdomain = tpdomain_create (domain);
 767         }
 768
 769         g_assert (tpdomain);
 770
 771         tpdomain->outstanding_request ++;
 772         g_assert (tpdomain->outstanding_request >= 1);
 773
 774         domains_unlock ();
 775
 776         COUNTER_ATOMIC (counter, {
 777                 if (counter._.starting == 16) {
 778                         mono_refcount_dec (&threadpool);
 779                         return TRUE;
 780                 }
 781
 782                 counter._.starting ++;
 783         });
 784
 785         mono_threadpool_worker_request ();
 786
 787         mono_refcount_dec (&threadpool);
 788         return TRUE;
 789 }
 790
 791 MonoBoolean G_GNUC_UNUSED
 792 ves_icall_System_Threading_ThreadPool_PostQueuedCompletionStatus (MonoNativeOverlapped *native_overlapped)
 793 {
 794         /* This copy the behavior of the current Mono implementation */
 795         MonoError error;
 796         mono_error_set_not_implemented (&error, "");
 797         mono_error_set_pending_exception (&error);
 798         return FALSE;
 799 }
 800
 801 MonoBoolean G_GNUC_UNUSED
 802 ves_icall_System_Threading_ThreadPool_BindIOCompletionCallbackNative (gpointer file_handle)
 803 {
 804         /* This copy the behavior of the current Mono implementation */
 805         return TRUE;
 806 }
 807
 808 MonoBoolean G_GNUC_UNUSED
 809 ves_icall_System_Threading_ThreadPool_IsThreadPoolHosted (void)
 810 {
 811         return FALSE;
 812 }