source: libcfa/src/concurrency/kernel.cfa @ fd90096

ADTast-experimental
Last change on this file since fd90096 was 8bee858, checked in by Thierry Delisle <tdelisle@…>, 2 years ago

Changed io types to have trailing $ instead of leading

  • Property mode set to 100644
File size: 28.8 KB
RevLine 
[8118303]1//
2// Cforall Version 1.0.0 Copyright (C) 2016 University of Waterloo
3//
4// The contents of this file are covered under the licence agreement in the
5// file "LICENCE" distributed with Cforall.
6//
7// kernel.c --
8//
9// Author           : Thierry Delisle
[75f3522]10// Created On       : Tue Jan 17 12:27:26 2017
[6b0b624]11// Last Modified By : Peter A. Buhr
[0190480]12// Last Modified On : Mon Aug 31 07:08:20 2020
13// Update Count     : 71
[8118303]14//
15
[2026bb6]16#define __cforall_thread__
[43784ac]17#define _GNU_SOURCE
18
[4069faad]19// #define __CFA_DEBUG_PRINT_RUNTIME_CORE__
[2026bb6]20
[bfb9bf5]21#pragma GCC diagnostic push
22#pragma GCC diagnostic ignored "-Waddress-of-packed-member"
23
[8118303]24//C Includes
[214e8da]25#include <errno.h>
[9d944b2]26#include <stdio.h>
[445f984]27#include <string.h>
[58b6d1b]28#include <signal.h>
[9d944b2]29#include <unistd.h>
[bfb9bf5]30
[dddb3dd0]31extern "C" {
32        #include <sys/eventfd.h>
[d3605f8]33        #include <sys/uio.h>
[dddb3dd0]34}
[8118303]35
36//CFA Includes
[708ae38]37#include "kernel/private.hfa"
[73abe95]38#include "preemption.hfa"
[445f984]39#include "strstream.hfa"
40#include "device/cpu.hfa"
[7ef162b2]41#include "io/types.hfa"
[8118303]42
43//Private includes
44#define __CFA_INVOKE_PRIVATE__
45#include "invoke.h"
[bfb9bf5]46#pragma GCC diagnostic pop
[8118303]47
[89eff25]48#if !defined(__CFA_NO_STATISTICS__)
[c9c1c1c]49        #define __STATS_DEF( ...) __VA_ARGS__
[89eff25]50#else
[c9c1c1c]51        #define __STATS_DEF( ...)
[89eff25]52#endif
[4069faad]53
[deca0f5]54//-----------------------------------------------------------------------------
55// Some assembly required
[1805b1b]56#if defined( __i386 )
[deca0f5]57        // mxcr : SSE Status and Control bits (control bits are preserved across function calls)
58        // fcw  : X87 FPU control word (preserved across function calls)
59        #define __x87_store         \
60                uint32_t __mxcr;      \
61                uint16_t __fcw;       \
62                __asm__ volatile (    \
63                        "stmxcsr %0\n"  \
64                        "fnstcw  %1\n"  \
65                        : "=m" (__mxcr),\
66                                "=m" (__fcw)  \
67                )
68
69        #define __x87_load         \
70                __asm__ volatile (   \
71                        "fldcw  %1\n"  \
72                        "ldmxcsr %0\n" \
73                        ::"m" (__mxcr),\
74                                "m" (__fcw)  \
75                )
76
77#elif defined( __x86_64 )
78        #define __x87_store         \
79                uint32_t __mxcr;      \
80                uint16_t __fcw;       \
81                __asm__ volatile (    \
82                        "stmxcsr %0\n"  \
83                        "fnstcw  %1\n"  \
84                        : "=m" (__mxcr),\
85                                "=m" (__fcw)  \
86                )
87
88        #define __x87_load          \
89                __asm__ volatile (    \
90                        "fldcw  %1\n"   \
91                        "ldmxcsr %0\n"  \
92                        :: "m" (__mxcr),\
93                                "m" (__fcw)  \
94                )
95
[0190480]96#elif defined( __arm__ )
97        #define __x87_store
98        #define __x87_load
99
100#elif defined( __aarch64__ )
[74f5c83]101        #define __x87_store              \
102                uint32_t __fpcntl[2];    \
103                __asm__ volatile (    \
104                        "mrs x9, FPCR\n" \
105                        "mrs x10, FPSR\n"  \
106                        "stp x9, x10, %0\n"  \
107                        : "=m" (__fpcntl) : : "x9", "x10" \
108                )
109
110        #define __x87_load         \
111                __asm__ volatile (    \
112                        "ldp x9, x10, %0\n"  \
113                        "msr FPSR, x10\n"  \
114                        "msr FPCR, x9\n" \
115                : "=m" (__fpcntl) : : "x9", "x10" \
116                )
117
[deca0f5]118#else
[0190480]119        #error unsupported hardware architecture
[deca0f5]120#endif
121
[e84ab3d]122extern thread$ * mainThread;
[e660761]123extern processor * mainProcessor;
[2ac095d]124
[92e7631]125//-----------------------------------------------------------------------------
126// Kernel Scheduling logic
[e84ab3d]127static thread$ * __next_thread(cluster * this);
128static thread$ * __next_thread_slow(cluster * this);
[c9c1c1c]129static thread$ * __next_thread_search(cluster * this);
[e84ab3d]130static inline bool __must_unpark( thread$ * thrd ) __attribute((nonnull(1)));
131static void __run_thread(processor * this, thread$ * dst);
[e873838]132static void __wake_one(cluster * cltr);
[1eb239e4]133
[18f7858]134static void idle_sleep(processor * proc);
[5f5a729]135static bool mark_idle (__cluster_proc_list & idles, processor & proc);
[6a9b12b]136static void mark_awake(__cluster_proc_list & idles, processor & proc);
[1eb239e4]137
[4479890]138extern bool __cfa_io_drain( processor * proc ) __attribute__((nonnull (1)));
[18f7858]139extern bool __cfa_io_flush( processor * ) __attribute__((nonnull (1)));
140extern void __cfa_io_idle( processor * ) __attribute__((nonnull (1)));
[dddb3dd0]141
[d3605f8]142#if defined(CFA_WITH_IO_URING_IDLE)
143        extern bool __kernel_read(processor * proc, io_future_t & future, iovec &, int fd);
[6ddef36]144#endif
[7ef162b2]145
[dddb3dd0]146extern void __disable_interrupts_hard();
147extern void __enable_interrupts_hard();
[c84e80a]148
[7ef162b2]149
[75f3522]150//=============================================================================================
151// Kernel Scheduling logic
152//=============================================================================================
[8fcbb4c]153//Main of the processor contexts
[83a071f9]154void main(processorCtx_t & runner) {
[21184e3]155        // Because of a bug, we couldn't initialized the seed on construction
156        // Do it here
[8fc652e0]157        __cfaabi_tls.rand_seed ^= rdtscl();
158        __cfaabi_tls.ready_rng.fwd_seed = 25214903917_l64u * (rdtscl() ^ (uintptr_t)&runner);
[f2384c9a]159        __tls_rand_advance_bck();
[21184e3]160
[83a071f9]161        processor * this = runner.proc;
[094476d]162        verify(this);
[c81ebf9]163
[22226e4]164        /* paranoid */ verify( this->idle_wctx.ftr   != 0p );
165        /* paranoid */ verify( this->idle_wctx.rdbuf != 0p );
[7ef162b2]166
[22226e4]167        // used for idle sleep when io_uring is present
168        // mark it as already fulfilled so we know if there is a pending request or not
169        this->idle_wctx.ftr->self.ptr = 1p;
[dddb3dd0]170
[4069faad]171        __cfadbg_print_safe(runtime_core, "Kernel : core %p starting\n", this);
[28d73c1]172        #if !defined(__CFA_NO_STATISTICS__)
173                if( this->print_halts ) {
[c993b15]174                        __cfaabi_bits_print_safe( STDOUT_FILENO, "Processor : %d - %s (%p)\n", this->unique_id, this->name, (void*)this);
[28d73c1]175                }
176        #endif
[b798713]177
[75f3522]178        {
[c81ebf9]179                // Setup preemption data
180                preemption_scope scope = { this };
181
[a5e7233]182                // if we need to run some special setup, now is the time to do it.
183                if(this->init.thrd) {
184                        this->init.thrd->curr_cluster = this->cltr;
185                        __run_thread(this, this->init.thrd);
186                }
[325e6ea]187
[4069faad]188                __cfadbg_print_safe(runtime_core, "Kernel : core %p started\n", this);
[8118303]189
[e84ab3d]190                thread$ * readyThread = 0p;
[1eb239e4]191                MAIN_LOOP:
192                for() {
[dddb3dd0]193                        // Check if there is pending io
[18f7858]194                        __cfa_io_drain( this );
[dddb3dd0]195
[92e7631]196                        // Try to get the next thread
[8c50aed]197                        readyThread = __next_thread( this->cltr );
[75f3522]198
[1eb239e4]199                        if( !readyThread ) {
[18f7858]200                                // there is no point in holding submissions if we are idle
[c9c1c1c]201                                __IO_STATS__(true, io.flush.idle++; )
[18f7858]202                                __cfa_io_flush( this );
203
204                                // drain again in case something showed up
205                                __cfa_io_drain( this );
[c33c2af]206
[c9c1c1c]207                                readyThread = __next_thread( this->cltr );
208                        }
209
210                        if( !readyThread ) for(5) {
[1eb239e4]211                                readyThread = __next_thread_slow( this->cltr );
[c9c1c1c]212
213                                if( readyThread ) break;
214
[18f7858]215                                // It's unlikely we still I/O to submit, but the arbiter could
216                                __IO_STATS__(true, io.flush.idle++; )
217                                __cfa_io_flush( this );
218
219                                // drain again in case something showed up
220                                __cfa_io_drain( this );
[1eb239e4]221                        }
[4e6fb8e]222
[1eb239e4]223                        HALT:
224                        if( !readyThread ) {
225                                // Don't block if we are done
226                                if( __atomic_load_n(&this->do_terminate, __ATOMIC_SEQ_CST) ) break MAIN_LOOP;
[c81ebf9]227
[1eb239e4]228                                // Push self to idle stack
[5f5a729]229                                if(!mark_idle(this->cltr->procs, * this)) continue MAIN_LOOP;
[398e8e9]230
[1eb239e4]231                                // Confirm the ready-queue is empty
[c9c1c1c]232                                readyThread = __next_thread_search( this->cltr );
[1eb239e4]233                                if( readyThread ) {
234                                        // A thread was found, cancel the halt
[6a9b12b]235                                        mark_awake(this->cltr->procs, * this);
[1eb239e4]236
[c9c1c1c]237                                        __STATS__(true, ready.sleep.cancels++; )
[1eb239e4]238
239                                        // continue the mai loop
240                                        break HALT;
241                                }
242
[18f7858]243                                idle_sleep( this );
[1eb239e4]244
245                                // We were woken up, remove self from idle
[6a9b12b]246                                mark_awake(this->cltr->procs, * this);
[1eb239e4]247
248                                // DON'T just proceed, start looking again
249                                continue MAIN_LOOP;
[64a7146]250                        }
[1eb239e4]251
252                        /* paranoid */ verify( readyThread );
253
[dddb3dd0]254                        // Reset io dirty bit
255                        this->io.dirty = false;
256
[1eb239e4]257                        // We found a thread run it
258                        __run_thread(this, readyThread);
259
260                        // Are we done?
261                        if( __atomic_load_n(&this->do_terminate, __ATOMIC_SEQ_CST) ) break MAIN_LOOP;
[325e6ea]262
[d529ad0]263                        if(__atomic_load_n(&this->io.pending, __ATOMIC_RELAXED) && !__atomic_load_n(&this->io.dirty, __ATOMIC_RELAXED)) {
[c9c1c1c]264                                __IO_STATS__(true, io.flush.dirty++; )
[18f7858]265                                __cfa_io_flush( this );
[dddb3dd0]266                        }
[c81ebf9]267                }
268
[4069faad]269                __cfadbg_print_safe(runtime_core, "Kernel : core %p stopping\n", this);
[c84e80a]270        }
[8118303]271
[454f478]272        post( this->terminated );
[bdeba0b]273
[28d73c1]274        if(this == mainProcessor) {
[6a490b2]275                // HACK : the coroutine context switch expects this_thread to be set
276                // and it make sense for it to be set in all other cases except here
277                // fake it
[8fc652e0]278                __cfaabi_tls.this_thread = mainThread;
[6a490b2]279        }
[7768b8d]280
[4069faad]281        __cfadbg_print_safe(runtime_core, "Kernel : core %p terminated\n", this);
[c84e80a]282}
283
[5c1a531]284static int * __volatile_errno() __attribute__((noinline));
285static int * __volatile_errno() { asm(""); return &errno; }
286
[14a61b5]287// KERNEL ONLY
[1c273d0]288// runThread runs a thread by context switching
289// from the processor coroutine to the target thread
[e84ab3d]290static void __run_thread(processor * this, thread$ * thrd_dst) {
[8fc652e0]291        /* paranoid */ verify( ! __preemption_enabled() );
[1eb239e4]292        /* paranoid */ verifyf( thrd_dst->state == Ready || thrd_dst->preempted != __NO_PREEMPTION, "state : %d, preempted %d\n", thrd_dst->state, thrd_dst->preempted);
293        /* paranoid */ verifyf( thrd_dst->link.next == 0p, "Expected null got %p", thrd_dst->link.next );
294        __builtin_prefetch( thrd_dst->context.SP );
295
[dddb3dd0]296        __cfadbg_print_safe(runtime_core, "Kernel : core %p running thread %p (%s)\n", this, thrd_dst, thrd_dst->self_cor.name);
297
[e84ab3d]298        coroutine$ * proc_cor = get_coroutine(this->runner);
[8fcbb4c]299
[9f575ea]300        // set state of processor coroutine to inactive
301        verify(proc_cor->state == Active);
[ae7be7a]302        proc_cor->state = Blocked;
[e8e457e]303
[9f575ea]304        // Actually run the thread
[3381ed7]305        RUNNING:  while(true) {
[ff79d5e]306                thrd_dst->preempted = __NO_PREEMPTION;
[e8e457e]307
[58d64a4]308                // Update global state
[8fc652e0]309                kernelTLS().this_thread = thrd_dst;
[75f3522]310
[8bee858]311                // Update the state after setting this_thread
312                // so that the debugger can find all active threads
313                // in tls storage
314                thrd_dst->state = Active;
315
[8fc652e0]316                /* paranoid */ verify( ! __preemption_enabled() );
317                /* paranoid */ verify( kernelTLS().this_thread == thrd_dst );
[9d6e1b8a]318                /* paranoid */ verify( thrd_dst->curr_cluster == this->cltr );
[b4b63e8]319                /* paranoid */ verify( thrd_dst->context.SP );
[5afb49a]320                /* paranoid */ verify( thrd_dst->state != Halted );
[e84ab3d]321                /* paranoid */ verifyf( ((uintptr_t)thrd_dst->context.SP) < ((uintptr_t)__get_stack(thrd_dst->curr_cor)->base ) || thrd_dst->curr_cor == proc_cor || thrd_dst->corctx_flag, "ERROR : Destination thread$ %p has been corrupted.\n StackPointer too small.\n", thrd_dst ); // add escape condition if we are setting up the processor
322                /* paranoid */ verifyf( ((uintptr_t)thrd_dst->context.SP) > ((uintptr_t)__get_stack(thrd_dst->curr_cor)->limit) || thrd_dst->curr_cor == proc_cor || thrd_dst->corctx_flag, "ERROR : Destination thread$ %p has been corrupted.\n StackPointer too large.\n", thrd_dst ); // add escape condition if we are setting up the processor
[ac12f1f]323                /* paranoid */ verify( 0x0D15EA5E0D15EA5Ep == thrd_dst->canary );
[b4b63e8]324
[3381ed7]325
[58d64a4]326
[9f575ea]327                // set context switch to the thread that the processor is executing
[c7a900a]328                __cfactx_switch( &proc_cor->context, &thrd_dst->context );
329                // when __cfactx_switch returns we are back in the processor coroutine
[9f575ea]330
[50871b4]331
332
[ac12f1f]333                /* paranoid */ verify( 0x0D15EA5E0D15EA5Ep == thrd_dst->canary );
[e84ab3d]334                /* paranoid */ verifyf( ((uintptr_t)thrd_dst->context.SP) > ((uintptr_t)__get_stack(thrd_dst->curr_cor)->limit) || thrd_dst->corctx_flag, "ERROR : Destination thread$ %p has been corrupted.\n StackPointer too large.\n", thrd_dst );
335                /* paranoid */ verifyf( ((uintptr_t)thrd_dst->context.SP) < ((uintptr_t)__get_stack(thrd_dst->curr_cor)->base ) || thrd_dst->corctx_flag, "ERROR : Destination thread$ %p has been corrupted.\n StackPointer too small.\n", thrd_dst );
[b4b63e8]336                /* paranoid */ verify( thrd_dst->context.SP );
[9d6e1b8a]337                /* paranoid */ verify( thrd_dst->curr_cluster == this->cltr );
[8fc652e0]338                /* paranoid */ verify( kernelTLS().this_thread == thrd_dst );
339                /* paranoid */ verify( ! __preemption_enabled() );
[75f3522]340
[3381ed7]341                // We just finished running a thread, there are a few things that could have happened.
342                // 1 - Regular case : the thread has blocked and now one has scheduled it yet.
343                // 2 - Racy case    : the thread has blocked but someone has already tried to schedule it.
344                // 4 - Preempted
345                // In case 1, we may have won a race so we can't write to the state again.
346                // In case 2, we lost the race so we now own the thread.
347
348                if(unlikely(thrd_dst->preempted != __NO_PREEMPTION)) {
[8bee858]349                        // Reset the this_thread now that we know
350                        // the state isn't active anymore
351                        kernelTLS().this_thread = 0p;
352
[3381ed7]353                        // The thread was preempted, reschedule it and reset the flag
[24e321c]354                        schedule_thread$( thrd_dst, UNPARK_LOCAL );
[3381ed7]355                        break RUNNING;
356                }
[75f3522]357
[3ea8ad1]358                if(unlikely(thrd_dst->state == Halting)) {
[8bee858]359                        // Reset the this_thread now that we know
360                        // the state isn't active anymore
361                        kernelTLS().this_thread = 0p;
362
[ff79d5e]363                        // The thread has halted, it should never be scheduled/run again
[5afb49a]364                        // finish the thread
365                        __thread_finish( thrd_dst );
[ff79d5e]366                        break RUNNING;
367                }
368
369                /* paranoid */ verify( thrd_dst->state == Active );
370                thrd_dst->state = Blocked;
371
[8bee858]372                // Reset the this_thread now that we know
373                // the state isn't active anymore
374                kernelTLS().this_thread = 0p;
375
[3381ed7]376                // set state of processor coroutine to active and the thread to inactive
[ff79d5e]377                int old_ticket = __atomic_fetch_sub(&thrd_dst->ticket, 1, __ATOMIC_SEQ_CST);
378                switch(old_ticket) {
[6a77224]379                        case TICKET_RUNNING:
[3381ed7]380                                // This is case 1, the regular case, nothing more is needed
381                                break RUNNING;
[6a77224]382                        case TICKET_UNBLOCK:
[c9c1c1c]383                                __STATS__(true, ready.threads.threads++; )
[3381ed7]384                                // This is case 2, the racy case, someone tried to run this thread before it finished blocking
385                                // In this case, just run it again.
386                                continue RUNNING;
387                        default:
388                                // This makes no sense, something is wrong abort
[ff79d5e]389                                abort();
[3381ed7]390                }
[9f575ea]391        }
[e8e457e]392
[9f575ea]393        // Just before returning to the processor, set the processor coroutine to active
[e8e457e]394        proc_cor->state = Active;
[1eb239e4]395
[dddb3dd0]396        __cfadbg_print_safe(runtime_core, "Kernel : core %p finished running thread %p\n", this, thrd_dst);
397
[c9c1c1c]398        __STATS__(true, ready.threads.threads--; )
[ec43cf9]399
[8fc652e0]400        /* paranoid */ verify( ! __preemption_enabled() );
[82c948c]401}
402
[14a61b5]403// KERNEL_ONLY
[c18bf9e]404static void returnToKernel() {
[8fc652e0]405        /* paranoid */ verify( ! __preemption_enabled() );
[e84ab3d]406        coroutine$ * proc_cor = get_coroutine(kernelTLS().this_processor->runner);
407        thread$ * thrd_src = kernelTLS().this_thread;
[e8e457e]408
[c9c1c1c]409        __STATS_DEF( thrd_src->last_proc = kernelTLS().this_processor; )
[29cb302]410
[9f575ea]411        // Run the thread on this processor
412        {
413                int local_errno = *__volatile_errno();
414                #if defined( __i386 ) || defined( __x86_64 )
415                        __x87_store;
416                #endif
[b4b63e8]417                /* paranoid */ verify( proc_cor->context.SP );
[ac12f1f]418                /* paranoid */ verify( 0x0D15EA5E0D15EA5Ep == thrd_src->canary );
[c7a900a]419                __cfactx_switch( &thrd_src->context, &proc_cor->context );
[ac12f1f]420                /* paranoid */ verify( 0x0D15EA5E0D15EA5Ep == thrd_src->canary );
[9f575ea]421                #if defined( __i386 ) || defined( __x86_64 )
422                        __x87_load;
423                #endif
424                *__volatile_errno() = local_errno;
[8fcbb4c]425        }
[deca0f5]426
[29cb302]427        #if !defined(__CFA_NO_STATISTICS__)
[89eff25]428                /* paranoid */ verify( thrd_src->last_proc != 0p );
429                if(thrd_src->last_proc != kernelTLS().this_processor) {
[29cb302]430                        __tls_stats()->ready.threads.migration++;
431                }
432        #endif
433
[8fc652e0]434        /* paranoid */ verify( ! __preemption_enabled() );
[e84ab3d]435        /* paranoid */ verifyf( ((uintptr_t)thrd_src->context.SP) < ((uintptr_t)__get_stack(thrd_src->curr_cor)->base ) || thrd_src->corctx_flag, "ERROR : Returning thread$ %p has been corrupted.\n StackPointer too small.\n", thrd_src );
436        /* paranoid */ verifyf( ((uintptr_t)thrd_src->context.SP) > ((uintptr_t)__get_stack(thrd_src->curr_cor)->limit) || thrd_src->corctx_flag, "ERROR : Returning thread$ %p has been corrupted.\n StackPointer too large.\n", thrd_src );
[c84e80a]437}
438
[8def349]439//-----------------------------------------------------------------------------
440// Scheduler routines
[14a61b5]441// KERNEL ONLY
[24e321c]442static void __schedule_thread( thread$ * thrd, unpark_hint hint ) {
[8fc652e0]443        /* paranoid */ verify( ! __preemption_enabled() );
[254ad1b]444        /* paranoid */ verify( ready_schedule_islocked());
[6a490b2]445        /* paranoid */ verify( thrd );
446        /* paranoid */ verify( thrd->state != Halted );
[9d6e1b8a]447        /* paranoid */ verify( thrd->curr_cluster );
[3381ed7]448        /* paranoid */ #if defined( __CFA_WITH_VERIFY__ )
[504a7dc]449        /* paranoid */  if( thrd->state == Blocked || thrd->state == Start ) assertf( thrd->preempted == __NO_PREEMPTION,
450                                        "Error inactive thread marked as preempted, state %d, preemption %d\n", thrd->state, thrd->preempted );
[ff79d5e]451        /* paranoid */  if( thrd->preempted != __NO_PREEMPTION ) assertf(thrd->state == Active,
[504a7dc]452                                        "Error preempted thread marked as not currently running, state %d, preemption %d\n", thrd->state, thrd->preempted );
[3381ed7]453        /* paranoid */ #endif
[6a490b2]454        /* paranoid */ verifyf( thrd->link.next == 0p, "Expected null got %p", thrd->link.next );
[ac12f1f]455        /* paranoid */ verify( 0x0D15EA5E0D15EA5Ep == thrd->canary );
[b4b63e8]456
[ae7be7a]457        if (thrd->preempted == __NO_PREEMPTION) thrd->state = Ready;
[6b4cdd3]458
[ec43cf9]459        // Dereference the thread now because once we push it, there is not guaranteed it's still valid.
460        struct cluster * cl = thrd->curr_cluster;
[c9c1c1c]461        __STATS_DEF(bool outside = hint == UNPARK_LOCAL && thrd->last_proc && thrd->last_proc != kernelTLS().this_processor; )
[32a8b61]462
[254ad1b]463        // push the thread to the cluster ready-queue
[24e321c]464        push( cl, thrd, hint );
[32a8b61]465
[254ad1b]466        // variable thrd is no longer safe to use
[734908c]467        thrd = 0xdeaddeaddeaddeadp;
[32a8b61]468
[254ad1b]469        // wake the cluster using the save variable.
470        __wake_one( cl );
[1c273d0]471
[ec43cf9]472        #if !defined(__CFA_NO_STATISTICS__)
473                if( kernelTLS().this_stats ) {
474                        __tls_stats()->ready.threads.threads++;
[89eff25]475                        if(outside) {
476                                __tls_stats()->ready.threads.extunpark++;
477                        }
[ec43cf9]478                }
479                else {
480                        __atomic_fetch_add(&cl->stats->ready.threads.threads, 1, __ATOMIC_RELAXED);
[89eff25]481                        __atomic_fetch_add(&cl->stats->ready.threads.extunpark, 1, __ATOMIC_RELAXED);
[ec43cf9]482                }
483        #endif
484
[254ad1b]485        /* paranoid */ verify( ready_schedule_islocked());
[8fc652e0]486        /* paranoid */ verify( ! __preemption_enabled() );
[db6f06a]487}
488
[24e321c]489void schedule_thread$( thread$ * thrd, unpark_hint hint ) {
[254ad1b]490        ready_schedule_lock();
[24e321c]491                __schedule_thread( thrd, hint );
[254ad1b]492        ready_schedule_unlock();
493}
494
[14a61b5]495// KERNEL ONLY
[e84ab3d]496static inline thread$ * __next_thread(cluster * this) with( *this ) {
[8fc652e0]497        /* paranoid */ verify( ! __preemption_enabled() );
[7768b8d]498
[e873838]499        ready_schedule_lock();
[e84ab3d]500                thread$ * thrd = pop_fast( this );
[e873838]501        ready_schedule_unlock();
[7768b8d]502
[8fc652e0]503        /* paranoid */ verify( ! __preemption_enabled() );
[1eb239e4]504        return thrd;
[eb2e723]505}
506
[64a7146]507// KERNEL ONLY
[e84ab3d]508static inline thread$ * __next_thread_slow(cluster * this) with( *this ) {
[8fc652e0]509        /* paranoid */ verify( ! __preemption_enabled() );
[64a7146]510
[e873838]511        ready_schedule_lock();
[c9c1c1c]512                thread$ * thrd = pop_slow( this );
513        ready_schedule_unlock();
514
515        /* paranoid */ verify( ! __preemption_enabled() );
516        return thrd;
517}
[fc59df78]518
[c9c1c1c]519// KERNEL ONLY
520static inline thread$ * __next_thread_search(cluster * this) with( *this ) {
521        /* paranoid */ verify( ! __preemption_enabled() );
522
523        ready_schedule_lock();
524                thread$ * thrd = pop_search( this );
[e873838]525        ready_schedule_unlock();
[64a7146]526
[8fc652e0]527        /* paranoid */ verify( ! __preemption_enabled() );
[1eb239e4]528        return thrd;
[64a7146]529}
530
[e84ab3d]531static inline bool __must_unpark( thread$ * thrd ) {
[ff79d5e]532        int old_ticket = __atomic_fetch_add(&thrd->ticket, 1, __ATOMIC_SEQ_CST);
533        switch(old_ticket) {
[6a77224]534                case TICKET_RUNNING:
[3381ed7]535                        // Wake won the race, the thread will reschedule/rerun itself
[c6c7e6c]536                        return false;
[6a77224]537                case TICKET_BLOCKED:
[3381ed7]538                        /* paranoid */ verify( ! thrd->preempted != __NO_PREEMPTION );
[ff79d5e]539                        /* paranoid */ verify( thrd->state == Blocked );
[c6c7e6c]540                        return true;
[3381ed7]541                default:
542                        // This makes no sense, something is wrong abort
[7ee8153]543                        abort("Thread %p (%s) has mismatch park/unpark\n", thrd, thrd->self_cor.name);
[de6319f]544        }
[eb2e723]545}
546
[24e321c]547void __kernel_unpark( thread$ * thrd, unpark_hint hint ) {
[e9c0b4c]548        /* paranoid */ verify( ! __preemption_enabled() );
549        /* paranoid */ verify( ready_schedule_islocked());
550
551        if( !thrd ) return;
552
553        if(__must_unpark(thrd)) {
554                // Wake lost the race,
[24e321c]555                __schedule_thread( thrd, hint );
[e9c0b4c]556        }
557
558        /* paranoid */ verify( ready_schedule_islocked());
559        /* paranoid */ verify( ! __preemption_enabled() );
560}
561
[c18bf9e]562void unpark( thread$ * thrd, unpark_hint hint ) libcfa_public {
[c6c7e6c]563        if( !thrd ) return;
[0b33412]564
[c6c7e6c]565        if(__must_unpark(thrd)) {
566                disable_interrupts();
[a3821fa]567                        // Wake lost the race,
[24e321c]568                        schedule_thread$( thrd, hint );
[a3821fa]569                enable_interrupts(false);
[de6319f]570        }
[eb2e723]571}
[0b33412]572
[c18bf9e]573void park( void ) libcfa_public {
[a3821fa]574        __disable_interrupts_checked();
575                /* paranoid */ verify( kernelTLS().this_thread->preempted == __NO_PREEMPTION );
576                returnToKernel();
577        __enable_interrupts_checked();
[0c78741]578
579}
[09800e9]580
[5afb49a]581extern "C" {
582        // Leave the thread monitor
583        // last routine called by a thread.
584        // Should never return
585        void __cfactx_thrd_leave() {
[e84ab3d]586                thread$ * thrd = active_thread();
587                monitor$ * this = &thrd->self_mon;
[5afb49a]588
589                // Lock the monitor now
590                lock( this->lock __cfaabi_dbg_ctx2 );
591
592                disable_interrupts();
593
[9d6e1b8a]594                /* paranoid */ verify( ! __preemption_enabled() );
595                /* paranoid */ verify( thrd->state == Active );
596                /* paranoid */ verify( 0x0D15EA5E0D15EA5Ep == thrd->canary );
597                /* paranoid */ verify( kernelTLS().this_thread == thrd );
598                /* paranoid */ verify( thrd->context.SP );
[e84ab3d]599                /* paranoid */ verifyf( ((uintptr_t)thrd->context.SP) > ((uintptr_t)__get_stack(thrd->curr_cor)->limit), "ERROR : thread$ %p has been corrupted.\n StackPointer too large.\n", thrd );
600                /* paranoid */ verifyf( ((uintptr_t)thrd->context.SP) < ((uintptr_t)__get_stack(thrd->curr_cor)->base ), "ERROR : thread$ %p has been corrupted.\n StackPointer too small.\n", thrd );
[9d6e1b8a]601
[3ea8ad1]602                thrd->state = Halting;
[58688bf]603                if( TICKET_RUNNING != thrd->ticket ) { abort( "Thread terminated with pending unpark" ); }
[9d6e1b8a]604                if( thrd != this->owner ) { abort( "Thread internal monitor has incorrect owner" ); }
605                if( this->recursion != 1) { abort( "Thread internal monitor has unbalanced recursion" ); }
[5afb49a]606
607                // Leave the thread
608                returnToKernel();
609
610                // Control flow should never reach here!
[9d6e1b8a]611                abort();
[5afb49a]612        }
[09800e9]613}
614
[14a61b5]615// KERNEL ONLY
[c18bf9e]616bool force_yield( __Preemption_Reason reason ) libcfa_public {
[a3821fa]617        __disable_interrupts_checked();
[e84ab3d]618                thread$ * thrd = kernelTLS().this_thread;
[a3821fa]619                /* paranoid */ verify(thrd->state == Active);
620
621                // SKULLDUGGERY: It is possible that we are preempting this thread just before
622                // it was going to park itself. If that is the case and it is already using the
623                // intrusive fields then we can't use them to preempt the thread
624                // If that is the case, abandon the preemption.
625                bool preempted = false;
626                if(thrd->link.next == 0p) {
627                        preempted = true;
628                        thrd->preempted = reason;
629                        returnToKernel();
630                }
631        __enable_interrupts_checked( false );
[3381ed7]632        return preempted;
[f2b12406]633}
634
[14a61b5]635//=============================================================================================
[92e7631]636// Kernel Idle Sleep
[14a61b5]637//=============================================================================================
[64a7146]638// Wake a thread from the front if there are any
[e873838]639static void __wake_one(cluster * this) {
[7cf3b1d]640        eventfd_t val;
641
[8fc652e0]642        /* paranoid */ verify( ! __preemption_enabled() );
[e873838]643        /* paranoid */ verify( ready_schedule_islocked() );
[14a61b5]644
[64a7146]645        // Check if there is a sleeping processor
[7cf3b1d]646        struct __fd_waitctx * fdp = __atomic_load_n(&this->procs.fdw, __ATOMIC_SEQ_CST);
[14a61b5]647
[7cf3b1d]648        // If no one is sleeping: we are done
649        if( fdp == 0p ) return;
[14a61b5]650
[7cf3b1d]651        int fd = 1;
[22226e4]652        if( __atomic_load_n(&fdp->sem, __ATOMIC_SEQ_CST) != 1 ) {
653                fd = __atomic_exchange_n(&fdp->sem, 1, __ATOMIC_RELAXED);
[7cf3b1d]654        }
[14a61b5]655
[7cf3b1d]656        switch(fd) {
[4ccc150]657                __attribute__((unused)) int ret;
[7cf3b1d]658        case 0:
659                // If the processor isn't ready to sleep then the exchange will already wake it up
660                #if !defined(__CFA_NO_STATISTICS__)
661                        if( kernelTLS().this_stats ) { __tls_stats()->ready.sleep.early++;
662                        } else { __atomic_fetch_add(&this->stats->ready.sleep.early, 1, __ATOMIC_RELAXED); }
663                #endif
664                break;
665        case 1:
666                // If someone else already said they will wake them: we are done
667                #if !defined(__CFA_NO_STATISTICS__)
668                        if( kernelTLS().this_stats ) { __tls_stats()->ready.sleep.seen++;
669                        } else { __atomic_fetch_add(&this->stats->ready.sleep.seen, 1, __ATOMIC_RELAXED); }
670                #endif
671                break;
672        default:
673                // If the processor was ready to sleep, we need to wake it up with an actual write
674                val = 1;
[4ccc150]675                ret = eventfd_write( fd, val );
[77adaee]676                /* paranoid */ verifyf( ret == 0, "Expected return to be 0, was %d\n", ret );
[7cf3b1d]677
678                #if !defined(__CFA_NO_STATISTICS__)
679                        if( kernelTLS().this_stats ) { __tls_stats()->ready.sleep.wakes++;
680                        } else { __atomic_fetch_add(&this->stats->ready.sleep.wakes, 1, __ATOMIC_RELAXED); }
681                #endif
682                break;
683        }
[1eb239e4]684
[e873838]685        /* paranoid */ verify( ready_schedule_islocked() );
[8fc652e0]686        /* paranoid */ verify( ! __preemption_enabled() );
[1eb239e4]687
688        return;
[64a7146]689}
[14a61b5]690
[64a7146]691// Unconditionnaly wake a thread
[1eb239e4]692void __wake_proc(processor * this) {
[7d0ebd0]693        /* paranoid */ verify( ! __preemption_enabled() );
694
[64a7146]695        __cfadbg_print_safe(runtime_core, "Kernel : waking Processor %p\n", this);
[14a61b5]696
[22226e4]697        this->idle_wctx.sem = 1;
[7cf3b1d]698
[efa28d5]699        this->idle_wctx.wake__time = rdtscl();
[262fafd9]700
[7d0ebd0]701        eventfd_t val;
702        val = 1;
[d080549]703        __attribute__((unused)) int ret = eventfd_write( this->idle_wctx.evfd, val );
[7d0ebd0]704
[d080549]705        /* paranoid */ verifyf( ret == 0, "Expected return to be 0, was %d\n", ret );
[7d0ebd0]706        /* paranoid */ verify( ! __preemption_enabled() );
[92e7631]707}
708
[18f7858]709static void idle_sleep(processor * this) {
[202c80b]710        /* paranoid */ verify( this->idle_wctx.evfd != 1 );
711        /* paranoid */ verify( this->idle_wctx.evfd != 2 );
712
[7cf3b1d]713        // Tell everyone we are ready to go do sleep
714        for() {
[22226e4]715                int expected = this->idle_wctx.sem;
[7cf3b1d]716
717                // Someone already told us to wake-up! No time for a nap.
718                if(expected == 1) { return; }
719
720                // Try to mark that we are going to sleep
[22226e4]721                if(__atomic_compare_exchange_n(&this->idle_wctx.sem, &expected, this->idle_wctx.evfd, false,  __ATOMIC_SEQ_CST, __ATOMIC_SEQ_CST) ) {
[7cf3b1d]722                        // Every one agreed, taking a nap
723                        break;
724                }
725        }
726
727
[d3605f8]728        #if !defined(CFA_WITH_IO_URING_IDLE)
[7ef162b2]729                #if !defined(__CFA_NO_STATISTICS__)
730                        if(this->print_halts) {
731                                __cfaabi_bits_print_safe( STDOUT_FILENO, "PH:%d - %lld 0\n", this->unique_id, rdtscl());
732                        }
733                #endif
[1757f98]734
[7ef162b2]735                __cfadbg_print_safe(runtime_core, "Kernel : core %p waiting on eventfd %d\n", this, this->idle_fd);
[1757f98]736
[7ef162b2]737                {
738                        eventfd_t val;
[22226e4]739                        ssize_t ret = read( this->idle_wctx.evfd, &val, sizeof(val) );
[7ef162b2]740                        if(ret < 0) {
741                                switch((int)errno) {
742                                case EAGAIN:
743                                #if EAGAIN != EWOULDBLOCK
744                                        case EWOULDBLOCK:
745                                #endif
746                                case EINTR:
747                                        // No need to do anything special here, just assume it's a legitimate wake-up
748                                        break;
749                                default:
750                                        abort( "KERNEL : internal error, read failure on idle eventfd, error(%d) %s.", (int)errno, strerror( (int)errno ) );
751                                }
[1757f98]752                        }
753                }
754
[7ef162b2]755                #if !defined(__CFA_NO_STATISTICS__)
756                        if(this->print_halts) {
757                                __cfaabi_bits_print_safe( STDOUT_FILENO, "PH:%d - %lld 1\n", this->unique_id, rdtscl());
758                        }
759                #endif
760        #else
[18f7858]761                __cfa_io_idle( this );
[1757f98]762        #endif
763}
764
[5f5a729]765static bool mark_idle(__cluster_proc_list & this, processor & proc) {
[c9c1c1c]766        __STATS__(true, ready.sleep.halts++; )
[7cf3b1d]767
[22226e4]768        proc.idle_wctx.sem = 0;
[7cf3b1d]769
[8fc652e0]770        /* paranoid */ verify( ! __preemption_enabled() );
[5f5a729]771        if(!try_lock( this )) return false;
[1eb239e4]772                this.idle++;
773                /* paranoid */ verify( this.idle <= this.total );
[fc59b580]774                remove(proc);
[6a9b12b]775                insert_first(this.idles, proc);
[34b8cb7]776
[7cf3b1d]777                __atomic_store_n(&this.fdw, &proc.idle_wctx, __ATOMIC_SEQ_CST);
[1eb239e4]778        unlock( this );
[8fc652e0]779        /* paranoid */ verify( ! __preemption_enabled() );
[5f5a729]780
781        return true;
[1eb239e4]782}
[8e16177]783
[6a9b12b]784static void mark_awake(__cluster_proc_list & this, processor & proc) {
[8fc652e0]785        /* paranoid */ verify( ! __preemption_enabled() );
[1eb239e4]786        lock( this );
787                this.idle--;
788                /* paranoid */ verify( this.idle >= 0 );
789                remove(proc);
[fc59b580]790                insert_last(this.actives, proc);
[6a9b12b]791
[a633f6f]792                {
[7cf3b1d]793                        struct __fd_waitctx * wctx = 0;
794                        if(!this.idles`isEmpty) wctx = &this.idles`first.idle_wctx;
795                        __atomic_store_n(&this.fdw, wctx, __ATOMIC_SEQ_CST);
[a633f6f]796                }
797
[34b8cb7]798        unlock( this );
[6a9b12b]799        /* paranoid */ verify( ! __preemption_enabled() );
[6b4cdd3]800}
801
[dbe9b08]802//=============================================================================================
803// Unexpected Terminating logic
804//=============================================================================================
[92bfda0]805void __kernel_abort_msg( char * abort_text, int abort_text_size ) {
[e84ab3d]806        thread$ * thrd = __cfaabi_tls.this_thread;
[9d944b2]807
[de94a60]808        if(thrd) {
809                int len = snprintf( abort_text, abort_text_size, "Error occurred while executing thread %.256s (%p)", thrd->self_cor.name, thrd );
[1c40091]810                __cfaabi_bits_write( STDERR_FILENO, abort_text, len );
[de94a60]811
[212c2187]812                if ( &thrd->self_cor != thrd->curr_cor ) {
813                        len = snprintf( abort_text, abort_text_size, " in coroutine %.256s (%p).\n", thrd->curr_cor->name, thrd->curr_cor );
[1c40091]814                        __cfaabi_bits_write( STDERR_FILENO, abort_text, len );
[de94a60]815                }
816                else {
[1c40091]817                        __cfaabi_bits_write( STDERR_FILENO, ".\n", 2 );
[de94a60]818                }
[1c273d0]819        }
[9d944b2]820        else {
[de94a60]821                int len = snprintf( abort_text, abort_text_size, "Error occurred outside of any thread.\n" );
[1c40091]822                __cfaabi_bits_write( STDERR_FILENO, abort_text, len );
[9d944b2]823        }
824}
825
[92bfda0]826int __kernel_abort_lastframe( void ) __attribute__ ((__nothrow__)) {
827        return get_coroutine(__cfaabi_tls.this_thread) == get_coroutine(mainThread) ? 4 : 2;
[2b8bc41]828}
829
[de94a60]830static __spinlock_t kernel_debug_lock;
831
[9d944b2]832extern "C" {
[1c40091]833        void __cfaabi_bits_acquire() {
[36982fc]834                lock( kernel_debug_lock __cfaabi_dbg_ctx2 );
[9d944b2]835        }
836
[1c40091]837        void __cfaabi_bits_release() {
[ea7d2b0]838                unlock( kernel_debug_lock );
[9d944b2]839        }
[8118303]840}
841
[fa21ac9]842//=============================================================================================
843// Kernel Utilities
844//=============================================================================================
[dddb3dd0]845#if defined(CFA_HAVE_LINUX_IO_URING_H)
846#include "io/types.hfa"
847#endif
848
[2026bb6]849//-----------------------------------------------------------------------------
850// Debug
[c18bf9e]851bool threading_enabled(void) __attribute__((const)) libcfa_public {
[2026bb6]852        return true;
853}
[c34ebf2]854
855//-----------------------------------------------------------------------------
856// Statistics
857#if !defined(__CFA_NO_STATISTICS__)
[c18bf9e]858        void print_halts( processor & this ) libcfa_public {
[c34ebf2]859                this.print_halts = true;
860        }
[58688bf]861
[69914cbc]862        static void crawl_list( cluster * cltr, dlist(processor) & list, unsigned count ) {
[8464edf]863                /* paranoid */ verify( cltr->stats );
864
865                processor * it = &list`first;
866                for(unsigned i = 0; i < count; i++) {
867                        /* paranoid */ verifyf( it, "Unexpected null iterator, at index %u of %u\n", i, count);
868                        /* paranoid */ verify( it->local_data->this_stats );
[c33c2af]869                        // __print_stats( it->local_data->this_stats, cltr->print_stats, "Processor", it->name, (void*)it );
[8464edf]870                        __tally_stats( cltr->stats, it->local_data->this_stats );
871                        it = &(*it)`next;
872                }
873        }
874
[c18bf9e]875        static void crawl_cluster_stats( cluster & this ) {
[8464edf]876                // Stop the world, otherwise stats could get really messed-up
877                // this doesn't solve all problems but does solve many
878                // so it's probably good enough
[c33c2af]879                disable_interrupts();
[8464edf]880                uint_fast32_t last_size = ready_mutate_lock();
881
882                        crawl_list(&this, this.procs.actives, this.procs.total - this.procs.idle);
883                        crawl_list(&this, this.procs.idles  , this.procs.idle );
884
885                // Unlock the RWlock
886                ready_mutate_unlock( last_size );
[c33c2af]887                enable_interrupts();
[8464edf]888        }
889
890
[c18bf9e]891        void print_stats_now( cluster & this, int flags ) libcfa_public {
[8464edf]892                crawl_cluster_stats( this );
[202c80b]893                __print_stats( this.stats, flags, "Cluster", this.name, (void*)&this );
[1b033b8]894        }
[c34ebf2]895#endif
[8118303]896// Local Variables: //
897// mode: c //
898// tab-width: 4 //
899// End: //
Note: See TracBrowser for help on using the repository browser.