Merge pull request #3028 from lateralusX/jlorenss/threadpool_warning
[mono.git] / mono / metadata / threadpool-ms.c
1 /*
2  * threadpool-ms.c: Microsoft threadpool runtime support
3  *
4  * Author:
5  *      Ludovic Henry (ludovic.henry@xamarin.com)
6  *
7  * Copyright 2015 Xamarin, Inc (http://www.xamarin.com)
8  * Licensed under the MIT license. See LICENSE file in the project root for full license information.
9  */
10
11 //
12 // Copyright (c) Microsoft. All rights reserved.
13 // Licensed under the MIT license. See LICENSE file in the project root for full license information.
14 //
15 // Files:
16 //  - src/vm/comthreadpool.cpp
17 //  - src/vm/win32threadpoolcpp
18 //  - src/vm/threadpoolrequest.cpp
19 //  - src/vm/hillclimbing.cpp
20 //
21 // Ported from C++ to C and adjusted to Mono runtime
22
23 #include <stdlib.h>
24 #define _USE_MATH_DEFINES // needed by MSVC to define math constants
25 #include <math.h>
26 #include <config.h>
27 #include <glib.h>
28
29 #include <mono/metadata/class-internals.h>
30 #include <mono/metadata/exception.h>
31 #include <mono/metadata/gc-internals.h>
32 #include <mono/metadata/object.h>
33 #include <mono/metadata/object-internals.h>
34 #include <mono/metadata/threadpool-ms.h>
35 #include <mono/metadata/threadpool-ms-io.h>
36 #include <mono/utils/atomic.h>
37 #include <mono/utils/mono-compiler.h>
38 #include <mono/utils/mono-complex.h>
39 #include <mono/utils/mono-lazy-init.h>
40 #include <mono/utils/mono-logger.h>
41 #include <mono/utils/mono-logger-internals.h>
42 #include <mono/utils/mono-proclib.h>
43 #include <mono/utils/mono-threads.h>
44 #include <mono/utils/mono-time.h>
45 #include <mono/utils/mono-rand.h>
46
47 #define CPU_USAGE_LOW 80
48 #define CPU_USAGE_HIGH 95
49
50 #define MONITOR_INTERVAL 500 // ms
51 #define MONITOR_MINIMAL_LIFETIME 60 * 1000 // ms
52
53 #define WORKER_CREATION_MAX_PER_SEC 10
54
55 /* The exponent to apply to the gain. 1.0 means to use linear gain,
56  * higher values will enhance large moves and damp small ones.
57  * default: 2.0 */
58 #define HILL_CLIMBING_GAIN_EXPONENT 2.0
59
60 /* The 'cost' of a thread. 0 means drive for increased throughput regardless
61  * of thread count, higher values bias more against higher thread counts.
62  * default: 0.15 */
63 #define HILL_CLIMBING_BIAS 0.15
64
65 #define HILL_CLIMBING_WAVE_PERIOD 4
66 #define HILL_CLIMBING_MAX_WAVE_MAGNITUDE 20
67 #define HILL_CLIMBING_WAVE_MAGNITUDE_MULTIPLIER 1.0
68 #define HILL_CLIMBING_WAVE_HISTORY_SIZE 8
69 #define HILL_CLIMBING_TARGET_SIGNAL_TO_NOISE_RATIO 3.0
70 #define HILL_CLIMBING_MAX_CHANGE_PER_SECOND 4
71 #define HILL_CLIMBING_MAX_CHANGE_PER_SAMPLE 20
72 #define HILL_CLIMBING_SAMPLE_INTERVAL_LOW 10
73 #define HILL_CLIMBING_SAMPLE_INTERVAL_HIGH 200
74 #define HILL_CLIMBING_ERROR_SMOOTHING_FACTOR 0.01
75 #define HILL_CLIMBING_MAX_SAMPLE_ERROR_PERCENT 0.15
76
77 typedef union {
78         struct {
79                 gint16 max_working; /* determined by heuristic */
80                 gint16 active; /* executing worker_thread */
81                 gint16 working; /* actively executing worker_thread, not parked */
82                 gint16 parked; /* parked */
83         } _;
84         gint64 as_gint64;
85 } ThreadPoolCounter;
86
87 typedef struct {
88         MonoDomain *domain;
89         gint32 outstanding_request;
90 } ThreadPoolDomain;
91
92 typedef MonoInternalThread ThreadPoolWorkingThread;
93
94 typedef struct {
95         gint32 wave_period;
96         gint32 samples_to_measure;
97         gdouble target_throughput_ratio;
98         gdouble target_signal_to_noise_ratio;
99         gdouble max_change_per_second;
100         gdouble max_change_per_sample;
101         gint32 max_thread_wave_magnitude;
102         gint32 sample_interval_low;
103         gdouble thread_magnitude_multiplier;
104         gint32 sample_interval_high;
105         gdouble throughput_error_smoothing_factor;
106         gdouble gain_exponent;
107         gdouble max_sample_error;
108
109         gdouble current_control_setting;
110         gint64 total_samples;
111         gint16 last_thread_count;
112         gdouble elapsed_since_last_change;
113         gdouble completions_since_last_change;
114
115         gdouble average_throughput_noise;
116
117         gdouble *samples;
118         gdouble *thread_counts;
119
120         guint32 current_sample_interval;
121         gpointer random_interval_generator;
122
123         gint32 accumulated_completion_count;
124         gdouble accumulated_sample_duration;
125 } ThreadPoolHillClimbing;
126
127 typedef struct {
128         ThreadPoolCounter counters;
129
130         GPtrArray *domains; // ThreadPoolDomain* []
131         MonoCoopMutex domains_lock;
132
133         GPtrArray *working_threads; // ThreadPoolWorkingThread* []
134         gint32 parked_threads_count;
135         MonoCoopCond parked_threads_cond;
136         MonoCoopMutex active_threads_lock; /* protect access to working_threads and parked_threads */
137
138         guint32 worker_creation_current_second;
139         guint32 worker_creation_current_count;
140         MonoCoopMutex worker_creation_lock;
141
142         gint32 heuristic_completions;
143         gint64 heuristic_sample_start;
144         gint64 heuristic_last_dequeue; // ms
145         gint64 heuristic_last_adjustment; // ms
146         gint64 heuristic_adjustment_interval; // ms
147         ThreadPoolHillClimbing heuristic_hill_climbing;
148         MonoCoopMutex heuristic_lock;
149
150         gint32 limit_worker_min;
151         gint32 limit_worker_max;
152         gint32 limit_io_min;
153         gint32 limit_io_max;
154
155         MonoCpuUsageState *cpu_usage_state;
156         gint32 cpu_usage;
157
158         /* suspended by the debugger */
159         gboolean suspended;
160 } ThreadPool;
161
162 typedef enum {
163         TRANSITION_WARMUP,
164         TRANSITION_INITIALIZING,
165         TRANSITION_RANDOM_MOVE,
166         TRANSITION_CLIMBING_MOVE,
167         TRANSITION_CHANGE_POINT,
168         TRANSITION_STABILIZING,
169         TRANSITION_STARVATION,
170         TRANSITION_THREAD_TIMED_OUT,
171         TRANSITION_UNDEFINED,
172 } ThreadPoolHeuristicStateTransition;
173
174 static mono_lazy_init_t status = MONO_LAZY_INIT_STATUS_NOT_INITIALIZED;
175
176 enum {
177         MONITOR_STATUS_REQUESTED,
178         MONITOR_STATUS_WAITING_FOR_REQUEST,
179         MONITOR_STATUS_NOT_RUNNING,
180 };
181
182 static gint32 monitor_status = MONITOR_STATUS_NOT_RUNNING;
183
184 static ThreadPool* threadpool;
185
186 #define COUNTER_CHECK(counter) \
187         do { \
188                 g_assert (counter._.max_working > 0); \
189                 g_assert (counter._.working >= 0); \
190                 g_assert (counter._.active >= 0); \
191         } while (0)
192
193 #define COUNTER_READ() (InterlockedRead64 (&threadpool->counters.as_gint64))
194
195 #define COUNTER_ATOMIC(var,block) \
196         do { \
197                 ThreadPoolCounter __old; \
198                 do { \
199                         g_assert (threadpool); \
200                         __old.as_gint64 = COUNTER_READ (); \
201                         (var) = __old; \
202                         { block; } \
203                         COUNTER_CHECK (var); \
204                 } while (InterlockedCompareExchange64 (&threadpool->counters.as_gint64, (var).as_gint64, __old.as_gint64) != __old.as_gint64); \
205         } while (0)
206
207 #define COUNTER_TRY_ATOMIC(res,var,block) \
208         do { \
209                 ThreadPoolCounter __old; \
210                 do { \
211                         g_assert (threadpool); \
212                         __old.as_gint64 = COUNTER_READ (); \
213                         (var) = __old; \
214                         (res) = FALSE; \
215                         { block; } \
216                         COUNTER_CHECK (var); \
217                         (res) = InterlockedCompareExchange64 (&threadpool->counters.as_gint64, (var).as_gint64, __old.as_gint64) == __old.as_gint64; \
218                 } while (0); \
219         } while (0)
220
221 static gpointer
222 rand_create (void)
223 {
224         mono_rand_open ();
225         return mono_rand_init (NULL, 0);
226 }
227
228 static guint32
229 rand_next (gpointer *handle, guint32 min, guint32 max)
230 {
231         MonoError error;
232         guint32 val;
233         mono_rand_try_get_uint32 (handle, &val, min, max, &error);
234         // FIXME handle error
235         mono_error_assert_ok (&error);
236         return val;
237 }
238
239 static void
240 rand_free (gpointer handle)
241 {
242         mono_rand_close (handle);
243 }
244
245 static void
246 initialize (void)
247 {
248         ThreadPoolHillClimbing *hc;
249         const char *threads_per_cpu_env;
250         gint threads_per_cpu;
251         gint threads_count;
252
253         g_assert (!threadpool);
254         threadpool = g_new0 (ThreadPool, 1);
255         g_assert (threadpool);
256
257         threadpool->domains = g_ptr_array_new ();
258         mono_coop_mutex_init (&threadpool->domains_lock);
259
260         threadpool->parked_threads_count = 0;
261         mono_coop_cond_init (&threadpool->parked_threads_cond);
262         threadpool->working_threads = g_ptr_array_new ();
263         mono_coop_mutex_init (&threadpool->active_threads_lock);
264
265         threadpool->worker_creation_current_second = -1;
266         mono_coop_mutex_init (&threadpool->worker_creation_lock);
267
268         threadpool->heuristic_adjustment_interval = 10;
269         mono_coop_mutex_init (&threadpool->heuristic_lock);
270
271         mono_rand_open ();
272
273         hc = &threadpool->heuristic_hill_climbing;
274
275         hc->wave_period = HILL_CLIMBING_WAVE_PERIOD;
276         hc->max_thread_wave_magnitude = HILL_CLIMBING_MAX_WAVE_MAGNITUDE;
277         hc->thread_magnitude_multiplier = (gdouble) HILL_CLIMBING_WAVE_MAGNITUDE_MULTIPLIER;
278         hc->samples_to_measure = hc->wave_period * HILL_CLIMBING_WAVE_HISTORY_SIZE;
279         hc->target_throughput_ratio = (gdouble) HILL_CLIMBING_BIAS;
280         hc->target_signal_to_noise_ratio = (gdouble) HILL_CLIMBING_TARGET_SIGNAL_TO_NOISE_RATIO;
281         hc->max_change_per_second = (gdouble) HILL_CLIMBING_MAX_CHANGE_PER_SECOND;
282         hc->max_change_per_sample = (gdouble) HILL_CLIMBING_MAX_CHANGE_PER_SAMPLE;
283         hc->sample_interval_low = HILL_CLIMBING_SAMPLE_INTERVAL_LOW;
284         hc->sample_interval_high = HILL_CLIMBING_SAMPLE_INTERVAL_HIGH;
285         hc->throughput_error_smoothing_factor = (gdouble) HILL_CLIMBING_ERROR_SMOOTHING_FACTOR;
286         hc->gain_exponent = (gdouble) HILL_CLIMBING_GAIN_EXPONENT;
287         hc->max_sample_error = (gdouble) HILL_CLIMBING_MAX_SAMPLE_ERROR_PERCENT;
288         hc->current_control_setting = 0;
289         hc->total_samples = 0;
290         hc->last_thread_count = 0;
291         hc->average_throughput_noise = 0;
292         hc->elapsed_since_last_change = 0;
293         hc->accumulated_completion_count = 0;
294         hc->accumulated_sample_duration = 0;
295         hc->samples = g_new0 (gdouble, hc->samples_to_measure);
296         hc->thread_counts = g_new0 (gdouble, hc->samples_to_measure);
297         hc->random_interval_generator = rand_create ();
298         hc->current_sample_interval = rand_next (&hc->random_interval_generator, hc->sample_interval_low, hc->sample_interval_high);
299
300         if (!(threads_per_cpu_env = g_getenv ("MONO_THREADS_PER_CPU")))
301                 threads_per_cpu = 1;
302         else
303                 threads_per_cpu = CLAMP (atoi (threads_per_cpu_env), 1, 50);
304
305         threads_count = mono_cpu_count () * threads_per_cpu;
306
307         threadpool->limit_worker_min = threadpool->limit_io_min = threads_count;
308
309 #if defined (PLATFORM_ANDROID) || defined (HOST_IOS)
310         threadpool->limit_worker_max = threadpool->limit_io_max = CLAMP (threads_count * 100, MIN (threads_count, 200), MAX (threads_count, 200));
311 #else
312         threadpool->limit_worker_max = threadpool->limit_io_max = threads_count * 100;
313 #endif
314
315         threadpool->counters._.max_working = threadpool->limit_worker_min;
316
317         threadpool->cpu_usage_state = g_new0 (MonoCpuUsageState, 1);
318
319         threadpool->suspended = FALSE;
320 }
321
322 static void worker_kill (ThreadPoolWorkingThread *thread);
323
324 static void
325 cleanup (void)
326 {
327         guint i;
328
329         /* we make the assumption along the code that we are
330          * cleaning up only if the runtime is shutting down */
331         g_assert (mono_runtime_is_shutting_down ());
332
333         while (monitor_status != MONITOR_STATUS_NOT_RUNNING)
334                 mono_thread_info_sleep (1, NULL);
335
336         mono_coop_mutex_lock (&threadpool->active_threads_lock);
337
338         /* stop all threadpool->working_threads */
339         for (i = 0; i < threadpool->working_threads->len; ++i)
340                 worker_kill ((ThreadPoolWorkingThread*) g_ptr_array_index (threadpool->working_threads, i));
341
342         /* unpark all threadpool->parked_threads */
343         mono_coop_cond_broadcast (&threadpool->parked_threads_cond);
344
345         mono_coop_mutex_unlock (&threadpool->active_threads_lock);
346 }
347
348 gboolean
349 mono_threadpool_ms_enqueue_work_item (MonoDomain *domain, MonoObject *work_item, MonoError *error)
350 {
351         static MonoClass *threadpool_class = NULL;
352         static MonoMethod *unsafe_queue_custom_work_item_method = NULL;
353         MonoDomain *current_domain;
354         MonoBoolean f;
355         gpointer args [2];
356
357         mono_error_init (error);
358         g_assert (work_item);
359
360         if (!threadpool_class)
361                 threadpool_class = mono_class_load_from_name (mono_defaults.corlib, "System.Threading", "ThreadPool");
362
363         if (!unsafe_queue_custom_work_item_method)
364                 unsafe_queue_custom_work_item_method = mono_class_get_method_from_name (threadpool_class, "UnsafeQueueCustomWorkItem", 2);
365         g_assert (unsafe_queue_custom_work_item_method);
366
367         f = FALSE;
368
369         args [0] = (gpointer) work_item;
370         args [1] = (gpointer) &f;
371
372         current_domain = mono_domain_get ();
373         if (current_domain == domain) {
374                 mono_runtime_invoke_checked (unsafe_queue_custom_work_item_method, NULL, args, error);
375                 return_val_if_nok (error, FALSE);
376         } else {
377                 mono_thread_push_appdomain_ref (domain);
378                 if (mono_domain_set (domain, FALSE)) {
379                         mono_runtime_invoke_checked (unsafe_queue_custom_work_item_method, NULL, args, error);
380                         if (!is_ok (error)) {
381                                 mono_thread_pop_appdomain_ref ();
382                                 return FALSE;
383                         }
384                         mono_domain_set (current_domain, TRUE);
385                 }
386                 mono_thread_pop_appdomain_ref ();
387         }
388         return TRUE;
389 }
390
391 /* LOCKING: threadpool->domains_lock must be held */
392 static void
393 domain_add (ThreadPoolDomain *tpdomain)
394 {
395         guint i, len;
396
397         g_assert (tpdomain);
398
399         len = threadpool->domains->len;
400         for (i = 0; i < len; ++i) {
401                 if (g_ptr_array_index (threadpool->domains, i) == tpdomain)
402                         break;
403         }
404
405         if (i == len)
406                 g_ptr_array_add (threadpool->domains, tpdomain);
407 }
408
409 /* LOCKING: threadpool->domains_lock must be held */
410 static gboolean
411 domain_remove (ThreadPoolDomain *tpdomain)
412 {
413         g_assert (tpdomain);
414         return g_ptr_array_remove (threadpool->domains, tpdomain);
415 }
416
417 /* LOCKING: threadpool->domains_lock must be held */
418 static ThreadPoolDomain *
419 domain_get (MonoDomain *domain, gboolean create)
420 {
421         ThreadPoolDomain *tpdomain = NULL;
422         guint i;
423
424         g_assert (domain);
425
426         for (i = 0; i < threadpool->domains->len; ++i) {
427                 tpdomain = (ThreadPoolDomain *)g_ptr_array_index (threadpool->domains, i);
428                 if (tpdomain->domain == domain)
429                         return tpdomain;
430         }
431
432         if (create) {
433                 tpdomain = g_new0 (ThreadPoolDomain, 1);
434                 tpdomain->domain = domain;
435                 domain_add (tpdomain);
436         }
437
438         return tpdomain;
439 }
440
441 static void
442 domain_free (ThreadPoolDomain *tpdomain)
443 {
444         g_free (tpdomain);
445 }
446
447 /* LOCKING: threadpool->domains_lock must be held */
448 static gboolean
449 domain_any_has_request (void)
450 {
451         guint i;
452
453         for (i = 0; i < threadpool->domains->len; ++i) {
454                 ThreadPoolDomain *tmp = (ThreadPoolDomain *)g_ptr_array_index (threadpool->domains, i);
455                 if (tmp->outstanding_request > 0)
456                         return TRUE;
457         }
458
459         return FALSE;
460 }
461
462 /* LOCKING: threadpool->domains_lock must be held */
463 static ThreadPoolDomain *
464 domain_get_next (ThreadPoolDomain *current)
465 {
466         ThreadPoolDomain *tpdomain = NULL;
467         guint len;
468
469         len = threadpool->domains->len;
470         if (len > 0) {
471                 guint i, current_idx = -1;
472                 if (current) {
473                         for (i = 0; i < len; ++i) {
474                                 if (current == g_ptr_array_index (threadpool->domains, i)) {
475                                         current_idx = i;
476                                         break;
477                                 }
478                         }
479                         g_assert (current_idx >= 0);
480                 }
481                 for (i = current_idx + 1; i < len + current_idx + 1; ++i) {
482                         ThreadPoolDomain *tmp = (ThreadPoolDomain *)g_ptr_array_index (threadpool->domains, i % len);
483                         if (tmp->outstanding_request > 0) {
484                                 tpdomain = tmp;
485                                 break;
486                         }
487                 }
488         }
489
490         return tpdomain;
491 }
492
493 static void
494 worker_wait_interrupt (gpointer data)
495 {
496         mono_coop_mutex_lock (&threadpool->active_threads_lock);
497         mono_coop_cond_signal (&threadpool->parked_threads_cond);
498         mono_coop_mutex_unlock (&threadpool->active_threads_lock);
499 }
500
501 /* return TRUE if timeout, FALSE otherwise (worker unpark or interrupt) */
502 static gboolean
503 worker_park (void)
504 {
505         gboolean timeout = FALSE;
506
507         mono_trace (G_LOG_LEVEL_DEBUG, MONO_TRACE_THREADPOOL, "[%p] current worker parking", mono_native_thread_id_get ());
508
509         mono_gc_set_skip_thread (TRUE);
510
511         mono_coop_mutex_lock (&threadpool->active_threads_lock);
512
513         if (!mono_runtime_is_shutting_down ()) {
514                 static gpointer rand_handle = NULL;
515                 MonoInternalThread *thread_internal;
516                 gboolean interrupted = FALSE;
517
518                 if (!rand_handle)
519                         rand_handle = rand_create ();
520                 g_assert (rand_handle);
521
522                 thread_internal = mono_thread_internal_current ();
523                 g_assert (thread_internal);
524
525                 threadpool->parked_threads_count += 1;
526                 g_ptr_array_remove_fast (threadpool->working_threads, thread_internal);
527
528                 mono_thread_info_install_interrupt (worker_wait_interrupt, NULL, &interrupted);
529                 if (interrupted)
530                         goto done;
531
532                 if (mono_coop_cond_timedwait (&threadpool->parked_threads_cond, &threadpool->active_threads_lock, rand_next (&rand_handle, 5 * 1000, 60 * 1000)) != 0)
533                         timeout = TRUE;
534
535                 mono_thread_info_uninstall_interrupt (&interrupted);
536
537 done:
538                 g_ptr_array_add (threadpool->working_threads, thread_internal);
539                 threadpool->parked_threads_count -= 1;
540         }
541
542         mono_coop_mutex_unlock (&threadpool->active_threads_lock);
543
544         mono_gc_set_skip_thread (FALSE);
545
546         mono_trace (G_LOG_LEVEL_DEBUG, MONO_TRACE_THREADPOOL, "[%p] current worker unparking, timeout? %s", mono_native_thread_id_get (), timeout ? "yes" : "no");
547
548         return timeout;
549 }
550
551 static gboolean
552 worker_try_unpark (void)
553 {
554         gboolean res = FALSE;
555
556         mono_trace (G_LOG_LEVEL_DEBUG, MONO_TRACE_THREADPOOL, "[%p] try unpark worker", mono_native_thread_id_get ());
557
558         mono_coop_mutex_lock (&threadpool->active_threads_lock);
559         if (threadpool->parked_threads_count > 0) {
560                 mono_coop_cond_signal (&threadpool->parked_threads_cond);
561                 res = TRUE;
562         }
563         mono_coop_mutex_unlock (&threadpool->active_threads_lock);
564
565         mono_trace (G_LOG_LEVEL_DEBUG, MONO_TRACE_THREADPOOL, "[%p] try unpark worker, success? %s", mono_native_thread_id_get (), res ? "yes" : "no");
566
567         return res;
568 }
569
570 static void
571 worker_kill (ThreadPoolWorkingThread *thread)
572 {
573         if (thread == mono_thread_internal_current ())
574                 return;
575
576         mono_thread_internal_stop ((MonoInternalThread*) thread);
577 }
578
579 static void
580 worker_thread (gpointer data)
581 {
582         MonoError error;
583         MonoInternalThread *thread;
584         ThreadPoolDomain *tpdomain, *previous_tpdomain;
585         ThreadPoolCounter counter;
586         gboolean retire = FALSE;
587
588         mono_trace (G_LOG_LEVEL_INFO, MONO_TRACE_THREADPOOL, "[%p] worker starting", mono_native_thread_id_get ());
589
590         g_assert (threadpool);
591
592         thread = mono_thread_internal_current ();
593         g_assert (thread);
594
595         mono_thread_set_name_internal (thread, mono_string_new (mono_get_root_domain (), "Threadpool worker"), FALSE, &error);
596         mono_error_assert_ok (&error);
597
598         mono_coop_mutex_lock (&threadpool->active_threads_lock);
599         g_ptr_array_add (threadpool->working_threads, thread);
600         mono_coop_mutex_unlock (&threadpool->active_threads_lock);
601
602         previous_tpdomain = NULL;
603
604         mono_coop_mutex_lock (&threadpool->domains_lock);
605
606         while (!mono_runtime_is_shutting_down ()) {
607                 tpdomain = NULL;
608
609                 if ((thread->state & (ThreadState_StopRequested | ThreadState_SuspendRequested)) != 0) {
610                         mono_coop_mutex_unlock (&threadpool->domains_lock);
611                         mono_thread_interruption_checkpoint ();
612                         mono_coop_mutex_lock (&threadpool->domains_lock);
613                 }
614
615                 if (retire || !(tpdomain = domain_get_next (previous_tpdomain))) {
616                         gboolean timeout;
617
618                         COUNTER_ATOMIC (counter, {
619                                 counter._.working --;
620                                 counter._.parked ++;
621                         });
622
623                         mono_coop_mutex_unlock (&threadpool->domains_lock);
624                         timeout = worker_park ();
625                         mono_coop_mutex_lock (&threadpool->domains_lock);
626
627                         COUNTER_ATOMIC (counter, {
628                                 counter._.working ++;
629                                 counter._.parked --;
630                         });
631
632                         if (timeout)
633                                 break;
634
635                         if (retire)
636                                 retire = FALSE;
637
638                         continue;
639                 }
640
641                 tpdomain->outstanding_request --;
642                 g_assert (tpdomain->outstanding_request >= 0);
643
644                 mono_trace (G_LOG_LEVEL_DEBUG, MONO_TRACE_THREADPOOL, "[%p] worker running in domain %p",
645                         mono_native_thread_id_get (), tpdomain->domain, tpdomain->outstanding_request);
646
647                 g_assert (tpdomain->domain);
648                 g_assert (tpdomain->domain->threadpool_jobs >= 0);
649                 tpdomain->domain->threadpool_jobs ++;
650
651                 mono_coop_mutex_unlock (&threadpool->domains_lock);
652
653                 mono_thread_push_appdomain_ref (tpdomain->domain);
654                 if (mono_domain_set (tpdomain->domain, FALSE)) {
655                         MonoObject *exc = NULL, *res;
656
657                         res = mono_runtime_try_invoke (mono_defaults.threadpool_perform_wait_callback_method, NULL, NULL, &exc, &error);
658                         if (exc || !mono_error_ok(&error)) {
659                                 if (exc == NULL)
660                                         exc = (MonoObject *) mono_error_convert_to_exception (&error);
661                                 else
662                                         mono_error_cleanup (&error);
663                                 mono_thread_internal_unhandled_exception (exc);
664                         } else if (res && *(MonoBoolean*) mono_object_unbox (res) == FALSE)
665                                 retire = TRUE;
666
667                         mono_thread_clr_state (thread, (MonoThreadState)~ThreadState_Background);
668                         if (!mono_thread_test_state (thread , ThreadState_Background))
669                                 ves_icall_System_Threading_Thread_SetState (thread, ThreadState_Background);
670
671                         mono_domain_set (mono_get_root_domain (), TRUE);
672                 }
673                 mono_thread_pop_appdomain_ref ();
674
675                 mono_coop_mutex_lock (&threadpool->domains_lock);
676
677                 tpdomain->domain->threadpool_jobs --;
678                 g_assert (tpdomain->domain->threadpool_jobs >= 0);
679
680                 if (tpdomain->domain->threadpool_jobs == 0 && mono_domain_is_unloading (tpdomain->domain)) {
681                         gboolean removed = domain_remove (tpdomain);
682                         g_assert (removed);
683                         if (tpdomain->domain->cleanup_semaphore)
684                                 ReleaseSemaphore (tpdomain->domain->cleanup_semaphore, 1, NULL);
685                         domain_free (tpdomain);
686                         tpdomain = NULL;
687                 }
688
689                 previous_tpdomain = tpdomain;
690         }
691
692         mono_coop_mutex_unlock (&threadpool->domains_lock);
693
694         mono_coop_mutex_lock (&threadpool->active_threads_lock);
695         g_ptr_array_remove_fast (threadpool->working_threads, thread);
696         mono_coop_mutex_unlock (&threadpool->active_threads_lock);
697
698         COUNTER_ATOMIC (counter, {
699                 counter._.working--;
700                 counter._.active --;
701         });
702
703         mono_trace (G_LOG_LEVEL_INFO, MONO_TRACE_THREADPOOL, "[%p] worker finishing", mono_native_thread_id_get ());
704 }
705
706 static gboolean
707 worker_try_create (void)
708 {
709         ThreadPoolCounter counter;
710         MonoInternalThread *thread;
711         gint64 current_ticks;
712         gint32 now;
713
714         mono_coop_mutex_lock (&threadpool->worker_creation_lock);
715
716         mono_trace (G_LOG_LEVEL_DEBUG, MONO_TRACE_THREADPOOL, "[%p] try create worker", mono_native_thread_id_get ());
717         current_ticks = mono_100ns_ticks ();
718         now = current_ticks / (10 * 1000 * 1000);
719         if (0 == current_ticks) {
720                 g_warning ("failed to get 100ns ticks");
721         } else {
722                 if (threadpool->worker_creation_current_second != now) {
723                         threadpool->worker_creation_current_second = now;
724                         threadpool->worker_creation_current_count = 0;
725                 } else {
726                         g_assert (threadpool->worker_creation_current_count <= WORKER_CREATION_MAX_PER_SEC);
727                         if (threadpool->worker_creation_current_count == WORKER_CREATION_MAX_PER_SEC) {
728                                 mono_trace (G_LOG_LEVEL_DEBUG, MONO_TRACE_THREADPOOL, "[%p] try create worker, failed: maximum number of worker created per second reached, current count = %d",
729                                         mono_native_thread_id_get (), threadpool->worker_creation_current_count);
730                                 mono_coop_mutex_unlock (&threadpool->worker_creation_lock);
731                                 return FALSE;
732                         }
733                 }
734         }
735
736         COUNTER_ATOMIC (counter, {
737                 if (counter._.working >= counter._.max_working) {
738                         mono_trace (G_LOG_LEVEL_DEBUG, MONO_TRACE_THREADPOOL, "[%p] try create worker, failed: maximum number of working threads reached",
739                                 mono_native_thread_id_get ());
740                         mono_coop_mutex_unlock (&threadpool->worker_creation_lock);
741                         return FALSE;
742                 }
743                 counter._.working ++;
744                 counter._.active ++;
745         });
746
747         MonoError error;
748         if ((thread = mono_thread_create_internal (mono_get_root_domain (), worker_thread, NULL, TRUE, 0, &error)) != NULL) {
749                 threadpool->worker_creation_current_count += 1;
750
751                 mono_trace (G_LOG_LEVEL_DEBUG, MONO_TRACE_THREADPOOL, "[%p] try create worker, created %p, now = %d count = %d", mono_native_thread_id_get (), thread->tid, now, threadpool->worker_creation_current_count);
752                 mono_coop_mutex_unlock (&threadpool->worker_creation_lock);
753                 return TRUE;
754         }
755
756         mono_trace (G_LOG_LEVEL_DEBUG, MONO_TRACE_THREADPOOL, "[%p] try create worker, failed: could not create thread due to %s", mono_native_thread_id_get (), mono_error_get_message (&error));
757         mono_error_cleanup (&error);
758
759         COUNTER_ATOMIC (counter, {
760                 counter._.working --;
761                 counter._.active --;
762         });
763
764         mono_coop_mutex_unlock (&threadpool->worker_creation_lock);
765         return FALSE;
766 }
767
768 static void monitor_ensure_running (void);
769
770 static gboolean
771 worker_request (MonoDomain *domain)
772 {
773         ThreadPoolDomain *tpdomain;
774
775         g_assert (domain);
776         g_assert (threadpool);
777
778         if (mono_runtime_is_shutting_down ())
779                 return FALSE;
780
781         mono_coop_mutex_lock (&threadpool->domains_lock);
782
783         /* synchronize check with worker_thread */
784         if (mono_domain_is_unloading (domain)) {
785                 mono_coop_mutex_unlock (&threadpool->domains_lock);
786                 return FALSE;
787         }
788
789         tpdomain = domain_get (domain, TRUE);
790         g_assert (tpdomain);
791         tpdomain->outstanding_request ++;
792
793         mono_trace (G_LOG_LEVEL_DEBUG, MONO_TRACE_THREADPOOL, "[%p] request worker, domain = %p, outstanding_request = %d",
794                 mono_native_thread_id_get (), tpdomain->domain, tpdomain->outstanding_request);
795
796         mono_coop_mutex_unlock (&threadpool->domains_lock);
797
798         if (threadpool->suspended)
799                 return FALSE;
800
801         monitor_ensure_running ();
802
803         if (worker_try_unpark ()) {
804                 mono_trace (G_LOG_LEVEL_DEBUG, MONO_TRACE_THREADPOOL, "[%p] request worker, unparked", mono_native_thread_id_get ());
805                 return TRUE;
806         }
807
808         if (worker_try_create ()) {
809                 mono_trace (G_LOG_LEVEL_DEBUG, MONO_TRACE_THREADPOOL, "[%p] request worker, created", mono_native_thread_id_get ());
810                 return TRUE;
811         }
812
813         mono_trace (G_LOG_LEVEL_DEBUG, MONO_TRACE_THREADPOOL, "[%p] request worker, failed", mono_native_thread_id_get ());
814         return FALSE;
815 }
816
817 static gboolean
818 monitor_should_keep_running (void)
819 {
820         static gint64 last_should_keep_running = -1;
821
822         g_assert (monitor_status == MONITOR_STATUS_WAITING_FOR_REQUEST || monitor_status == MONITOR_STATUS_REQUESTED);
823
824         if (InterlockedExchange (&monitor_status, MONITOR_STATUS_WAITING_FOR_REQUEST) == MONITOR_STATUS_WAITING_FOR_REQUEST) {
825                 gboolean should_keep_running = TRUE, force_should_keep_running = FALSE;
826
827                 if (mono_runtime_is_shutting_down ()) {
828                         should_keep_running = FALSE;
829                 } else {
830                         mono_coop_mutex_lock (&threadpool->domains_lock);
831                         if (!domain_any_has_request ())
832                                 should_keep_running = FALSE;
833                         mono_coop_mutex_unlock (&threadpool->domains_lock);
834
835                         if (!should_keep_running) {
836                                 if (last_should_keep_running == -1 || mono_100ns_ticks () - last_should_keep_running < MONITOR_MINIMAL_LIFETIME * 1000 * 10) {
837                                         should_keep_running = force_should_keep_running = TRUE;
838                                 }
839                         }
840                 }
841
842                 if (should_keep_running) {
843                         if (last_should_keep_running == -1 || !force_should_keep_running)
844                                 last_should_keep_running = mono_100ns_ticks ();
845                 } else {
846                         last_should_keep_running = -1;
847                         if (InterlockedCompareExchange (&monitor_status, MONITOR_STATUS_NOT_RUNNING, MONITOR_STATUS_WAITING_FOR_REQUEST) == MONITOR_STATUS_WAITING_FOR_REQUEST)
848                                 return FALSE;
849                 }
850         }
851
852         g_assert (monitor_status == MONITOR_STATUS_WAITING_FOR_REQUEST || monitor_status == MONITOR_STATUS_REQUESTED);
853
854         return TRUE;
855 }
856
857 static gboolean
858 monitor_sufficient_delay_since_last_dequeue (void)
859 {
860         gint64 threshold;
861
862         g_assert (threadpool);
863
864         if (threadpool->cpu_usage < CPU_USAGE_LOW) {
865                 threshold = MONITOR_INTERVAL;
866         } else {
867                 ThreadPoolCounter counter;
868                 counter.as_gint64 = COUNTER_READ();
869                 threshold = counter._.max_working * MONITOR_INTERVAL * 2;
870         }
871
872         return mono_msec_ticks () >= threadpool->heuristic_last_dequeue + threshold;
873 }
874
875 static void hill_climbing_force_change (gint16 new_thread_count, ThreadPoolHeuristicStateTransition transition);
876
877 static void
878 monitor_thread (void)
879 {
880         MonoInternalThread *current_thread = mono_thread_internal_current ();
881         guint i;
882
883         mono_cpu_usage (threadpool->cpu_usage_state);
884
885         mono_trace (G_LOG_LEVEL_DEBUG, MONO_TRACE_THREADPOOL, "[%p] monitor thread, started", mono_native_thread_id_get ());
886
887         do {
888                 ThreadPoolCounter counter;
889                 gboolean limit_worker_max_reached;
890                 gint32 interval_left = MONITOR_INTERVAL;
891                 gint32 awake = 0; /* number of spurious awakes we tolerate before doing a round of rebalancing */
892
893                 g_assert (monitor_status != MONITOR_STATUS_NOT_RUNNING);
894
895                 mono_gc_set_skip_thread (TRUE);
896
897                 do {
898                         gint64 ts;
899                         gboolean alerted = FALSE;
900
901                         if (mono_runtime_is_shutting_down ())
902                                 break;
903
904                         ts = mono_msec_ticks ();
905                         if (mono_thread_info_sleep (interval_left, &alerted) == 0)
906                                 break;
907                         interval_left -= mono_msec_ticks () - ts;
908
909                         mono_gc_set_skip_thread (FALSE);
910                         if ((current_thread->state & (ThreadState_StopRequested | ThreadState_SuspendRequested)) != 0)
911                                 mono_thread_interruption_checkpoint ();
912                         mono_gc_set_skip_thread (TRUE);
913                 } while (interval_left > 0 && ++awake < 10);
914
915                 mono_gc_set_skip_thread (FALSE);
916
917                 if (threadpool->suspended)
918                         continue;
919
920                 if (mono_runtime_is_shutting_down ())
921                         continue;
922
923                 mono_coop_mutex_lock (&threadpool->domains_lock);
924                 if (!domain_any_has_request ()) {
925                         mono_coop_mutex_unlock (&threadpool->domains_lock);
926                         continue;
927                 }
928                 mono_coop_mutex_unlock (&threadpool->domains_lock);
929
930                 threadpool->cpu_usage = mono_cpu_usage (threadpool->cpu_usage_state);
931
932                 if (!monitor_sufficient_delay_since_last_dequeue ())
933                         continue;
934
935                 limit_worker_max_reached = FALSE;
936
937                 COUNTER_ATOMIC (counter, {
938                         if (counter._.max_working >= threadpool->limit_worker_max) {
939                                 limit_worker_max_reached = TRUE;
940                                 break;
941                         }
942                         counter._.max_working ++;
943                 });
944
945                 if (limit_worker_max_reached)
946                         continue;
947
948                 hill_climbing_force_change (counter._.max_working, TRANSITION_STARVATION);
949
950                 for (i = 0; i < 5; ++i) {
951                         if (mono_runtime_is_shutting_down ())
952                                 break;
953
954                         if (worker_try_unpark ()) {
955                                 mono_trace (G_LOG_LEVEL_DEBUG, MONO_TRACE_THREADPOOL, "[%p] monitor thread, unparked", mono_native_thread_id_get ());
956                                 break;
957                         }
958
959                         if (worker_try_create ()) {
960                                 mono_trace (G_LOG_LEVEL_DEBUG, MONO_TRACE_THREADPOOL, "[%p] monitor thread, created", mono_native_thread_id_get ());
961                                 break;
962                         }
963                 }
964         } while (monitor_should_keep_running ());
965
966         mono_trace (G_LOG_LEVEL_DEBUG, MONO_TRACE_THREADPOOL, "[%p] monitor thread, finished", mono_native_thread_id_get ());
967 }
968
969 static void
970 monitor_ensure_running (void)
971 {
972         MonoError error;
973         for (;;) {
974                 switch (monitor_status) {
975                 case MONITOR_STATUS_REQUESTED:
976                         return;
977                 case MONITOR_STATUS_WAITING_FOR_REQUEST:
978                         InterlockedCompareExchange (&monitor_status, MONITOR_STATUS_REQUESTED, MONITOR_STATUS_WAITING_FOR_REQUEST);
979                         break;
980                 case MONITOR_STATUS_NOT_RUNNING:
981                         if (mono_runtime_is_shutting_down ())
982                                 return;
983                         if (InterlockedCompareExchange (&monitor_status, MONITOR_STATUS_REQUESTED, MONITOR_STATUS_NOT_RUNNING) == MONITOR_STATUS_NOT_RUNNING) {
984                                 if (!mono_thread_create_internal (mono_get_root_domain (), monitor_thread, NULL, TRUE, SMALL_STACK, &error)) {
985                                         monitor_status = MONITOR_STATUS_NOT_RUNNING;
986                                         mono_error_cleanup (&error);
987                                 }
988                                 return;
989                         }
990                         break;
991                 default: g_assert_not_reached ();
992                 }
993         }
994 }
995
996 static void
997 hill_climbing_change_thread_count (gint16 new_thread_count, ThreadPoolHeuristicStateTransition transition)
998 {
999         ThreadPoolHillClimbing *hc;
1000
1001         g_assert (threadpool);
1002
1003         hc = &threadpool->heuristic_hill_climbing;
1004
1005         mono_trace (G_LOG_LEVEL_INFO, MONO_TRACE_THREADPOOL, "[%p] hill climbing, change max number of threads %d", mono_native_thread_id_get (), new_thread_count);
1006
1007         hc->last_thread_count = new_thread_count;
1008         hc->current_sample_interval = rand_next (&hc->random_interval_generator, hc->sample_interval_low, hc->sample_interval_high);
1009         hc->elapsed_since_last_change = 0;
1010         hc->completions_since_last_change = 0;
1011 }
1012
1013 static void
1014 hill_climbing_force_change (gint16 new_thread_count, ThreadPoolHeuristicStateTransition transition)
1015 {
1016         ThreadPoolHillClimbing *hc;
1017
1018         g_assert (threadpool);
1019
1020         hc = &threadpool->heuristic_hill_climbing;
1021
1022         if (new_thread_count != hc->last_thread_count) {
1023                 hc->current_control_setting += new_thread_count - hc->last_thread_count;
1024                 hill_climbing_change_thread_count (new_thread_count, transition);
1025         }
1026 }
1027
1028 static double_complex
1029 hill_climbing_get_wave_component (gdouble *samples, guint sample_count, gdouble period)
1030 {
1031         ThreadPoolHillClimbing *hc;
1032         gdouble w, cosine, sine, coeff, q0, q1, q2;
1033         guint i;
1034
1035         g_assert (threadpool);
1036         g_assert (sample_count >= period);
1037         g_assert (period >= 2);
1038
1039         hc = &threadpool->heuristic_hill_climbing;
1040
1041         w = 2.0 * M_PI / period;
1042         cosine = cos (w);
1043         sine = sin (w);
1044         coeff = 2.0 * cosine;
1045         q0 = q1 = q2 = 0;
1046
1047         for (i = 0; i < sample_count; ++i) {
1048                 q0 = coeff * q1 - q2 + samples [(hc->total_samples - sample_count + i) % hc->samples_to_measure];
1049                 q2 = q1;
1050                 q1 = q0;
1051         }
1052
1053         return mono_double_complex_scalar_div (mono_double_complex_make (q1 - q2 * cosine, (q2 * sine)), ((gdouble)sample_count));
1054 }
1055
1056 static gint16
1057 hill_climbing_update (gint16 current_thread_count, guint32 sample_duration, gint32 completions, gint64 *adjustment_interval)
1058 {
1059         ThreadPoolHillClimbing *hc;
1060         ThreadPoolHeuristicStateTransition transition;
1061         gdouble throughput;
1062         gdouble throughput_error_estimate;
1063         gdouble confidence;
1064         gdouble move;
1065         gdouble gain;
1066         gint sample_index;
1067         gint sample_count;
1068         gint new_thread_wave_magnitude;
1069         gint new_thread_count;
1070         double_complex thread_wave_component;
1071         double_complex throughput_wave_component;
1072         double_complex ratio;
1073
1074         g_assert (threadpool);
1075         g_assert (adjustment_interval);
1076
1077         hc = &threadpool->heuristic_hill_climbing;
1078
1079         /* If someone changed the thread count without telling us, update our records accordingly. */
1080         if (current_thread_count != hc->last_thread_count)
1081                 hill_climbing_force_change (current_thread_count, TRANSITION_INITIALIZING);
1082
1083         /* Update the cumulative stats for this thread count */
1084         hc->elapsed_since_last_change += sample_duration;
1085         hc->completions_since_last_change += completions;
1086
1087         /* Add in any data we've already collected about this sample */
1088         sample_duration += hc->accumulated_sample_duration;
1089         completions += hc->accumulated_completion_count;
1090
1091         /* We need to make sure we're collecting reasonably accurate data. Since we're just counting the end
1092          * of each work item, we are goinng to be missing some data about what really happened during the
1093          * sample interval. The count produced by each thread includes an initial work item that may have
1094          * started well before the start of the interval, and each thread may have been running some new
1095          * work item for some time before the end of the interval, which did not yet get counted. So
1096          * our count is going to be off by +/- threadCount workitems.
1097          *
1098          * The exception is that the thread that reported to us last time definitely wasn't running any work
1099          * at that time, and the thread that's reporting now definitely isn't running a work item now. So
1100          * we really only need to consider threadCount-1 threads.
1101          *
1102          * Thus the percent error in our count is +/- (threadCount-1)/numCompletions.
1103          *
1104          * We cannot rely on the frequency-domain analysis we'll be doing later to filter out this error, because
1105          * of the way it accumulates over time. If this sample is off by, say, 33% in the negative direction,
1106          * then the next one likely will be too. The one after that will include the sum of the completions
1107          * we missed in the previous samples, and so will be 33% positive. So every three samples we'll have
1108          * two "low" samples and one "high" sample. This will appear as periodic variation right in the frequency
1109          * range we're targeting, which will not be filtered by the frequency-domain translation. */
1110         if (hc->total_samples > 0 && ((current_thread_count - 1.0) / completions) >= hc->max_sample_error) {
1111                 /* Not accurate enough yet. Let's accumulate the data so
1112                  * far, and tell the ThreadPool to collect a little more. */
1113                 hc->accumulated_sample_duration = sample_duration;
1114                 hc->accumulated_completion_count = completions;
1115                 *adjustment_interval = 10;
1116                 return current_thread_count;
1117         }
1118
1119         /* We've got enouugh data for our sample; reset our accumulators for next time. */
1120         hc->accumulated_sample_duration = 0;
1121         hc->accumulated_completion_count = 0;
1122
1123         /* Add the current thread count and throughput sample to our history. */
1124         throughput = ((gdouble) completions) / sample_duration;
1125
1126         sample_index = hc->total_samples % hc->samples_to_measure;
1127         hc->samples [sample_index] = throughput;
1128         hc->thread_counts [sample_index] = current_thread_count;
1129         hc->total_samples ++;
1130
1131         /* Set up defaults for our metrics. */
1132         thread_wave_component = mono_double_complex_make(0, 0);
1133         throughput_wave_component = mono_double_complex_make(0, 0);
1134         throughput_error_estimate = 0;
1135         ratio = mono_double_complex_make(0, 0);
1136         confidence = 0;
1137
1138         transition = TRANSITION_WARMUP;
1139
1140         /* How many samples will we use? It must be at least the three wave periods we're looking for, and it must also
1141          * be a whole multiple of the primary wave's period; otherwise the frequency we're looking for will fall between
1142          * two frequency bands in the Fourier analysis, and we won't be able to measure it accurately. */
1143         sample_count = ((gint) MIN (hc->total_samples - 1, hc->samples_to_measure) / hc->wave_period) * hc->wave_period;
1144
1145         if (sample_count > hc->wave_period) {
1146                 guint i;
1147                 gdouble average_throughput;
1148                 gdouble average_thread_count;
1149                 gdouble sample_sum = 0;
1150                 gdouble thread_sum = 0;
1151
1152                 /* Average the throughput and thread count samples, so we can scale the wave magnitudes later. */
1153                 for (i = 0; i < sample_count; ++i) {
1154                         guint j = (hc->total_samples - sample_count + i) % hc->samples_to_measure;
1155                         sample_sum += hc->samples [j];
1156                         thread_sum += hc->thread_counts [j];
1157                 }
1158
1159                 average_throughput = sample_sum / sample_count;
1160                 average_thread_count = thread_sum / sample_count;
1161
1162                 if (average_throughput > 0 && average_thread_count > 0) {
1163                         gdouble noise_for_confidence, adjacent_period_1, adjacent_period_2;
1164
1165                         /* Calculate the periods of the adjacent frequency bands we'll be using to
1166                          * measure noise levels. We want the two adjacent Fourier frequency bands. */
1167                         adjacent_period_1 = sample_count / (((gdouble) sample_count) / ((gdouble) hc->wave_period) + 1);
1168                         adjacent_period_2 = sample_count / (((gdouble) sample_count) / ((gdouble) hc->wave_period) - 1);
1169
1170                         /* Get the the three different frequency components of the throughput (scaled by average
1171                          * throughput). Our "error" estimate (the amount of noise that might be present in the
1172                          * frequency band we're really interested in) is the average of the adjacent bands. */
1173                         throughput_wave_component = mono_double_complex_scalar_div (hill_climbing_get_wave_component (hc->samples, sample_count, hc->wave_period), average_throughput);
1174                         throughput_error_estimate = cabs (mono_double_complex_scalar_div (hill_climbing_get_wave_component (hc->samples, sample_count, adjacent_period_1), average_throughput));
1175
1176                         if (adjacent_period_2 <= sample_count) {
1177                                 throughput_error_estimate = MAX (throughput_error_estimate, cabs (mono_double_complex_scalar_div (hill_climbing_get_wave_component (
1178                                         hc->samples, sample_count, adjacent_period_2), average_throughput)));
1179                         }
1180
1181                         /* Do the same for the thread counts, so we have something to compare to. We don't
1182                          * measure thread count noise, because there is none; these are exact measurements. */
1183                         thread_wave_component = mono_double_complex_scalar_div (hill_climbing_get_wave_component (hc->thread_counts, sample_count, hc->wave_period), average_thread_count);
1184
1185                         /* Update our moving average of the throughput noise. We'll use this
1186                          * later as feedback to determine the new size of the thread wave. */
1187                         if (hc->average_throughput_noise == 0) {
1188                                 hc->average_throughput_noise = throughput_error_estimate;
1189                         } else {
1190                                 hc->average_throughput_noise = (hc->throughput_error_smoothing_factor * throughput_error_estimate)
1191                                         + ((1.0 + hc->throughput_error_smoothing_factor) * hc->average_throughput_noise);
1192                         }
1193
1194                         if (cabs (thread_wave_component) > 0) {
1195                                 /* Adjust the throughput wave so it's centered around the target wave,
1196                                  * and then calculate the adjusted throughput/thread ratio. */
1197                                 ratio = mono_double_complex_div (mono_double_complex_sub (throughput_wave_component, mono_double_complex_scalar_mul(thread_wave_component, hc->target_throughput_ratio)), thread_wave_component);
1198                                 transition = TRANSITION_CLIMBING_MOVE;
1199                         } else {
1200                                 ratio = mono_double_complex_make (0, 0);
1201                                 transition = TRANSITION_STABILIZING;
1202                         }
1203
1204                         noise_for_confidence = MAX (hc->average_throughput_noise, throughput_error_estimate);
1205                         if (noise_for_confidence > 0) {
1206                                 confidence = cabs (thread_wave_component) / noise_for_confidence / hc->target_signal_to_noise_ratio;
1207                         } else {
1208                                 /* there is no noise! */
1209                                 confidence = 1.0;
1210                         }
1211                 }
1212         }
1213
1214         /* We use just the real part of the complex ratio we just calculated. If the throughput signal
1215          * is exactly in phase with the thread signal, this will be the same as taking the magnitude of
1216          * the complex move and moving that far up. If they're 180 degrees out of phase, we'll move
1217          * backward (because this indicates that our changes are having the opposite of the intended effect).
1218          * If they're 90 degrees out of phase, we won't move at all, because we can't tell wether we're
1219          * having a negative or positive effect on throughput. */
1220         move = creal (ratio);
1221         move = CLAMP (move, -1.0, 1.0);
1222
1223         /* Apply our confidence multiplier. */
1224         move *= CLAMP (confidence, -1.0, 1.0);
1225
1226         /* Now apply non-linear gain, such that values around zero are attenuated, while higher values
1227          * are enhanced. This allows us to move quickly if we're far away from the target, but more slowly
1228         * if we're getting close, giving us rapid ramp-up without wild oscillations around the target. */
1229         gain = hc->max_change_per_second * sample_duration;
1230         move = pow (fabs (move), hc->gain_exponent) * (move >= 0.0 ? 1 : -1) * gain;
1231         move = MIN (move, hc->max_change_per_sample);
1232
1233         /* If the result was positive, and CPU is > 95%, refuse the move. */
1234         if (move > 0.0 && threadpool->cpu_usage > CPU_USAGE_HIGH)
1235                 move = 0.0;
1236
1237         /* Apply the move to our control setting. */
1238         hc->current_control_setting += move;
1239
1240         /* Calculate the new thread wave magnitude, which is based on the moving average we've been keeping of the
1241          * throughput error.  This average starts at zero, so we'll start with a nice safe little wave at first. */
1242         new_thread_wave_magnitude = (gint)(0.5 + (hc->current_control_setting * hc->average_throughput_noise
1243                 * hc->target_signal_to_noise_ratio * hc->thread_magnitude_multiplier * 2.0));
1244         new_thread_wave_magnitude = CLAMP (new_thread_wave_magnitude, 1, hc->max_thread_wave_magnitude);
1245
1246         /* Make sure our control setting is within the ThreadPool's limits. */
1247         hc->current_control_setting = CLAMP (hc->current_control_setting, threadpool->limit_worker_min, threadpool->limit_worker_max - new_thread_wave_magnitude);
1248
1249         /* Calculate the new thread count (control setting + square wave). */
1250         new_thread_count = (gint)(hc->current_control_setting + new_thread_wave_magnitude * ((hc->total_samples / (hc->wave_period / 2)) % 2));
1251
1252         /* Make sure the new thread count doesn't exceed the ThreadPool's limits. */
1253         new_thread_count = CLAMP (new_thread_count, threadpool->limit_worker_min, threadpool->limit_worker_max);
1254
1255         if (new_thread_count != current_thread_count)
1256                 hill_climbing_change_thread_count (new_thread_count, transition);
1257
1258         if (creal (ratio) < 0.0 && new_thread_count == threadpool->limit_worker_min)
1259                 *adjustment_interval = (gint)(0.5 + hc->current_sample_interval * (10.0 * MAX (-1.0 * creal (ratio), 1.0)));
1260         else
1261                 *adjustment_interval = hc->current_sample_interval;
1262
1263         return new_thread_count;
1264 }
1265
1266 static void
1267 heuristic_notify_work_completed (void)
1268 {
1269         g_assert (threadpool);
1270
1271         InterlockedIncrement (&threadpool->heuristic_completions);
1272         threadpool->heuristic_last_dequeue = mono_msec_ticks ();
1273 }
1274
1275 static gboolean
1276 heuristic_should_adjust (void)
1277 {
1278         g_assert (threadpool);
1279
1280         if (threadpool->heuristic_last_dequeue > threadpool->heuristic_last_adjustment + threadpool->heuristic_adjustment_interval) {
1281                 ThreadPoolCounter counter;
1282                 counter.as_gint64 = COUNTER_READ();
1283                 if (counter._.working <= counter._.max_working)
1284                         return TRUE;
1285         }
1286
1287         return FALSE;
1288 }
1289
1290 static void
1291 heuristic_adjust (void)
1292 {
1293         g_assert (threadpool);
1294
1295         if (mono_coop_mutex_trylock (&threadpool->heuristic_lock) == 0) {
1296                 gint32 completions = InterlockedExchange (&threadpool->heuristic_completions, 0);
1297                 gint64 sample_end = mono_msec_ticks ();
1298                 gint64 sample_duration = sample_end - threadpool->heuristic_sample_start;
1299
1300                 if (sample_duration >= threadpool->heuristic_adjustment_interval / 2) {
1301                         ThreadPoolCounter counter;
1302                         gint16 new_thread_count;
1303
1304                         counter.as_gint64 = COUNTER_READ ();
1305                         new_thread_count = hill_climbing_update (counter._.max_working, sample_duration, completions, &threadpool->heuristic_adjustment_interval);
1306
1307                         COUNTER_ATOMIC (counter, { counter._.max_working = new_thread_count; });
1308
1309                         if (new_thread_count > counter._.max_working)
1310                                 worker_request (mono_domain_get ());
1311
1312                         threadpool->heuristic_sample_start = sample_end;
1313                         threadpool->heuristic_last_adjustment = mono_msec_ticks ();
1314                 }
1315
1316                 mono_coop_mutex_unlock (&threadpool->heuristic_lock);
1317         }
1318 }
1319
1320 void
1321 mono_threadpool_ms_cleanup (void)
1322 {
1323         #ifndef DISABLE_SOCKETS
1324                 mono_threadpool_ms_io_cleanup ();
1325         #endif
1326         mono_lazy_cleanup (&status, cleanup);
1327 }
1328
1329 MonoAsyncResult *
1330 mono_threadpool_ms_begin_invoke (MonoDomain *domain, MonoObject *target, MonoMethod *method, gpointer *params, MonoError *error)
1331 {
1332         static MonoClass *async_call_klass = NULL;
1333         MonoMethodMessage *message;
1334         MonoAsyncResult *async_result;
1335         MonoAsyncCall *async_call;
1336         MonoDelegate *async_callback = NULL;
1337         MonoObject *state = NULL;
1338
1339         if (!async_call_klass)
1340                 async_call_klass = mono_class_load_from_name (mono_defaults.corlib, "System", "MonoAsyncCall");
1341
1342         mono_lazy_initialize (&status, initialize);
1343
1344         mono_error_init (error);
1345
1346         message = mono_method_call_message_new (method, params, mono_get_delegate_invoke (method->klass), (params != NULL) ? (&async_callback) : NULL, (params != NULL) ? (&state) : NULL, error);
1347         return_val_if_nok (error, NULL);
1348
1349         async_call = (MonoAsyncCall*) mono_object_new_checked (domain, async_call_klass, error);
1350         return_val_if_nok (error, NULL);
1351
1352         MONO_OBJECT_SETREF (async_call, msg, message);
1353         MONO_OBJECT_SETREF (async_call, state, state);
1354
1355         if (async_callback) {
1356                 MONO_OBJECT_SETREF (async_call, cb_method, mono_get_delegate_invoke (((MonoObject*) async_callback)->vtable->klass));
1357                 MONO_OBJECT_SETREF (async_call, cb_target, async_callback);
1358         }
1359
1360         async_result = mono_async_result_new (domain, NULL, async_call->state, NULL, (MonoObject*) async_call, error);
1361         return_val_if_nok (error, NULL);
1362         MONO_OBJECT_SETREF (async_result, async_delegate, target);
1363
1364         mono_threadpool_ms_enqueue_work_item (domain, (MonoObject*) async_result, error);
1365         return_val_if_nok (error, NULL);
1366
1367         return async_result;
1368 }
1369
1370 MonoObject *
1371 mono_threadpool_ms_end_invoke (MonoAsyncResult *ares, MonoArray **out_args, MonoObject **exc, MonoError *error)
1372 {
1373         MonoAsyncCall *ac;
1374
1375         mono_error_init (error);
1376         g_assert (exc);
1377         g_assert (out_args);
1378
1379         *exc = NULL;
1380         *out_args = NULL;
1381
1382         /* check if already finished */
1383         mono_monitor_enter ((MonoObject*) ares);
1384
1385         if (ares->endinvoke_called) {
1386                 mono_error_set_invalid_operation(error, "Delegate EndInvoke method called more than once");
1387                 mono_monitor_exit ((MonoObject*) ares);
1388                 return NULL;
1389         }
1390
1391         ares->endinvoke_called = 1;
1392
1393         /* wait until we are really finished */
1394         if (ares->completed) {
1395                 mono_monitor_exit ((MonoObject *) ares);
1396         } else {
1397                 gpointer wait_event;
1398                 if (ares->handle) {
1399                         wait_event = mono_wait_handle_get_handle ((MonoWaitHandle*) ares->handle);
1400                 } else {
1401                         wait_event = CreateEvent (NULL, TRUE, FALSE, NULL);
1402                         g_assert(wait_event);
1403                         MonoWaitHandle *wait_handle = mono_wait_handle_new (mono_object_domain (ares), wait_event, error);
1404                         if (!is_ok (error)) {
1405                                 CloseHandle (wait_event);
1406                                 return NULL;
1407                         }
1408                         MONO_OBJECT_SETREF (ares, handle, (MonoObject*) wait_handle);
1409                 }
1410                 mono_monitor_exit ((MonoObject*) ares);
1411                 MONO_ENTER_GC_SAFE;
1412                 WaitForSingleObjectEx (wait_event, INFINITE, TRUE);
1413                 MONO_EXIT_GC_SAFE;
1414         }
1415
1416         ac = (MonoAsyncCall*) ares->object_data;
1417         g_assert (ac);
1418
1419         *exc = ac->msg->exc; /* FIXME: GC add write barrier */
1420         *out_args = ac->out_args;
1421         return ac->res;
1422 }
1423
1424 gboolean
1425 mono_threadpool_ms_remove_domain_jobs (MonoDomain *domain, int timeout)
1426 {
1427         gboolean res = TRUE;
1428         gint64 end;
1429         gpointer sem;
1430
1431         g_assert (domain);
1432         g_assert (timeout >= -1);
1433
1434         g_assert (mono_domain_is_unloading (domain));
1435
1436         if (timeout != -1)
1437                 end = mono_msec_ticks () + timeout;
1438
1439 #ifndef DISABLE_SOCKETS
1440         mono_threadpool_ms_io_remove_domain_jobs (domain);
1441         if (timeout != -1) {
1442                 if (mono_msec_ticks () > end)
1443                         return FALSE;
1444         }
1445 #endif
1446
1447         /*
1448          * There might be some threads out that could be about to execute stuff from the given domain.
1449          * We avoid that by setting up a semaphore to be pulsed by the thread that reaches zero.
1450          */
1451         sem = domain->cleanup_semaphore = CreateSemaphore (NULL, 0, 1, NULL);
1452
1453         /*
1454          * The memory barrier here is required to have global ordering between assigning to cleanup_semaphone
1455          * and reading threadpool_jobs. Otherwise this thread could read a stale version of threadpool_jobs
1456          * and wait forever.
1457          */
1458         mono_memory_write_barrier ();
1459
1460         while (domain->threadpool_jobs) {
1461                 gint64 now;
1462
1463                 if (timeout != -1) {
1464                         now = mono_msec_ticks ();
1465                         if (now > end) {
1466                                 res = FALSE;
1467                                 break;
1468                         }
1469                 }
1470
1471                 MONO_ENTER_GC_SAFE;
1472                 WaitForSingleObject (sem, timeout != -1 ? end - now : timeout);
1473                 MONO_EXIT_GC_SAFE;
1474         }
1475
1476         domain->cleanup_semaphore = NULL;
1477         CloseHandle (sem);
1478
1479         return res;
1480 }
1481
1482 void
1483 mono_threadpool_ms_suspend (void)
1484 {
1485         if (threadpool)
1486                 threadpool->suspended = TRUE;
1487 }
1488
1489 void
1490 mono_threadpool_ms_resume (void)
1491 {
1492         if (threadpool)
1493                 threadpool->suspended = FALSE;
1494 }
1495
1496 void
1497 ves_icall_System_Threading_ThreadPool_GetAvailableThreadsNative (gint32 *worker_threads, gint32 *completion_port_threads)
1498 {
1499         ThreadPoolCounter counter;
1500
1501         if (!worker_threads || !completion_port_threads)
1502                 return;
1503
1504         mono_lazy_initialize (&status, initialize);
1505
1506         counter.as_gint64 = COUNTER_READ ();
1507
1508         *worker_threads = MAX (0, threadpool->limit_worker_max - counter._.active);
1509         *completion_port_threads = threadpool->limit_io_max;
1510 }
1511
1512 void
1513 ves_icall_System_Threading_ThreadPool_GetMinThreadsNative (gint32 *worker_threads, gint32 *completion_port_threads)
1514 {
1515         if (!worker_threads || !completion_port_threads)
1516                 return;
1517
1518         mono_lazy_initialize (&status, initialize);
1519
1520         *worker_threads = threadpool->limit_worker_min;
1521         *completion_port_threads = threadpool->limit_io_min;
1522 }
1523
1524 void
1525 ves_icall_System_Threading_ThreadPool_GetMaxThreadsNative (gint32 *worker_threads, gint32 *completion_port_threads)
1526 {
1527         if (!worker_threads || !completion_port_threads)
1528                 return;
1529
1530         mono_lazy_initialize (&status, initialize);
1531
1532         *worker_threads = threadpool->limit_worker_max;
1533         *completion_port_threads = threadpool->limit_io_max;
1534 }
1535
1536 MonoBoolean
1537 ves_icall_System_Threading_ThreadPool_SetMinThreadsNative (gint32 worker_threads, gint32 completion_port_threads)
1538 {
1539         mono_lazy_initialize (&status, initialize);
1540
1541         if (worker_threads <= 0 || worker_threads > threadpool->limit_worker_max)
1542                 return FALSE;
1543         if (completion_port_threads <= 0 || completion_port_threads > threadpool->limit_io_max)
1544                 return FALSE;
1545
1546         threadpool->limit_worker_min = worker_threads;
1547         threadpool->limit_io_min = completion_port_threads;
1548
1549         return TRUE;
1550 }
1551
1552 MonoBoolean
1553 ves_icall_System_Threading_ThreadPool_SetMaxThreadsNative (gint32 worker_threads, gint32 completion_port_threads)
1554 {
1555         gint cpu_count = mono_cpu_count ();
1556
1557         mono_lazy_initialize (&status, initialize);
1558
1559         if (worker_threads < threadpool->limit_worker_min || worker_threads < cpu_count)
1560                 return FALSE;
1561         if (completion_port_threads < threadpool->limit_io_min || completion_port_threads < cpu_count)
1562                 return FALSE;
1563
1564         threadpool->limit_worker_max = worker_threads;
1565         threadpool->limit_io_max = completion_port_threads;
1566
1567         return TRUE;
1568 }
1569
1570 void
1571 ves_icall_System_Threading_ThreadPool_InitializeVMTp (MonoBoolean *enable_worker_tracking)
1572 {
1573         if (enable_worker_tracking) {
1574                 // TODO implement some kind of switch to have the possibily to use it
1575                 *enable_worker_tracking = FALSE;
1576         }
1577
1578         mono_lazy_initialize (&status, initialize);
1579 }
1580
1581 MonoBoolean
1582 ves_icall_System_Threading_ThreadPool_NotifyWorkItemComplete (void)
1583 {
1584         ThreadPoolCounter counter;
1585
1586         if (mono_domain_is_unloading (mono_domain_get ()) || mono_runtime_is_shutting_down ())
1587                 return FALSE;
1588
1589         heuristic_notify_work_completed ();
1590
1591         if (heuristic_should_adjust ())
1592                 heuristic_adjust ();
1593
1594         counter.as_gint64 = COUNTER_READ ();
1595         return counter._.working <= counter._.max_working;
1596 }
1597
1598 void
1599 ves_icall_System_Threading_ThreadPool_NotifyWorkItemProgressNative (void)
1600 {
1601         heuristic_notify_work_completed ();
1602
1603         if (heuristic_should_adjust ())
1604                 heuristic_adjust ();
1605 }
1606
1607 void
1608 ves_icall_System_Threading_ThreadPool_ReportThreadStatus (MonoBoolean is_working)
1609 {
1610         // TODO
1611         MonoError error;
1612         mono_error_set_not_implemented (&error, "");
1613         mono_error_set_pending_exception (&error);
1614 }
1615
1616 MonoBoolean
1617 ves_icall_System_Threading_ThreadPool_RequestWorkerThread (void)
1618 {
1619         return worker_request (mono_domain_get ());
1620 }
1621
1622 MonoBoolean G_GNUC_UNUSED
1623 ves_icall_System_Threading_ThreadPool_PostQueuedCompletionStatus (MonoNativeOverlapped *native_overlapped)
1624 {
1625         /* This copy the behavior of the current Mono implementation */
1626         MonoError error;
1627         mono_error_set_not_implemented (&error, "");
1628         mono_error_set_pending_exception (&error);
1629         return FALSE;
1630 }
1631
1632 MonoBoolean G_GNUC_UNUSED
1633 ves_icall_System_Threading_ThreadPool_BindIOCompletionCallbackNative (gpointer file_handle)
1634 {
1635         /* This copy the behavior of the current Mono implementation */
1636         return TRUE;
1637 }
1638
1639 MonoBoolean G_GNUC_UNUSED
1640 ves_icall_System_Threading_ThreadPool_IsThreadPoolHosted (void)
1641 {
1642         return FALSE;
1643 }