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