source: libcfa/src/concurrency/monitor.cfa @ 4f0534f

ADTarm-ehast-experimentalenumforall-pointer-decayjacob/cs343-translationjenkins-sandboxnew-astnew-ast-unique-exprpthread-emulationqualifiedEnum
Last change on this file since 4f0534f was dac55004, checked in by Thierry Delisle <tdelisle@…>, 5 years ago

Moved assert which wasn't really checking for an invariant

  • Property mode set to 100644
File size: 31.7 KB
Line 
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// $monitor.c --
8//
9// Author           : Thierry Delisle
10// Created On       : Thd Feb 23 12:27:26 2017
11// Last Modified By : Peter A. Buhr
12// Last Modified On : Wed Dec  4 07:55:14 2019
13// Update Count     : 10
14//
15
16#define __cforall_thread__
17
18#include "monitor.hfa"
19
20#include <stdlib.hfa>
21#include <inttypes.h>
22
23#include "kernel_private.hfa"
24
25#include "bits/algorithm.hfa"
26
27//-----------------------------------------------------------------------------
28// Forward declarations
29static inline void __set_owner ( $monitor * this, $thread * owner );
30static inline void __set_owner ( $monitor * storage [], __lock_size_t count, $thread * owner );
31static inline void set_mask  ( $monitor * storage [], __lock_size_t count, const __waitfor_mask_t & mask );
32static inline void reset_mask( $monitor * this );
33
34static inline $thread * next_thread( $monitor * this );
35static inline bool is_accepted( $monitor * this, const __monitor_group_t & monitors );
36
37static inline void lock_all  ( __spinlock_t * locks [], __lock_size_t count );
38static inline void lock_all  ( $monitor * source [], __spinlock_t * /*out*/ locks [], __lock_size_t count );
39static inline void unlock_all( __spinlock_t * locks [], __lock_size_t count );
40static inline void unlock_all( $monitor * locks [], __lock_size_t count );
41
42static inline void save   ( $monitor * ctx [], __lock_size_t count, __spinlock_t * locks [], unsigned int /*out*/ recursions [], __waitfor_mask_t /*out*/ masks [] );
43static inline void restore( $monitor * ctx [], __lock_size_t count, __spinlock_t * locks [], unsigned int /*in */ recursions [], __waitfor_mask_t /*in */ masks [] );
44
45static inline void init     ( __lock_size_t count, $monitor * monitors [], __condition_node_t & waiter, __condition_criterion_t criteria [] );
46static inline void init_push( __lock_size_t count, $monitor * monitors [], __condition_node_t & waiter, __condition_criterion_t criteria [] );
47
48static inline $thread *        check_condition   ( __condition_criterion_t * );
49static inline void                 brand_condition   ( condition & );
50static inline [$thread *, int] search_entry_queue( const __waitfor_mask_t &, $monitor * monitors [], __lock_size_t count );
51
52forall(dtype T | sized( T ))
53static inline __lock_size_t insert_unique( T * array [], __lock_size_t & size, T * val );
54static inline __lock_size_t count_max    ( const __waitfor_mask_t & mask );
55static inline __lock_size_t aggregate    ( $monitor * storage [], const __waitfor_mask_t & mask );
56
57//-----------------------------------------------------------------------------
58// Useful defines
59#define wait_ctx(thrd, user_info)                               /* Create the necessary information to use the signaller stack                         */ \
60        __condition_node_t waiter = { thrd, count, user_info };   /* Create the node specific to this wait operation                                     */ \
61        __condition_criterion_t criteria[count];                  /* Create the creteria this wait operation needs to wake up                            */ \
62        init( count, monitors, waiter, criteria );                /* Link everything together                                                            */ \
63
64#define wait_ctx_primed(thrd, user_info)                        /* Create the necessary information to use the signaller stack                         */ \
65        __condition_node_t waiter = { thrd, count, user_info };   /* Create the node specific to this wait operation                                     */ \
66        __condition_criterion_t criteria[count];                  /* Create the creteria this wait operation needs to wake up                            */ \
67        init_push( count, monitors, waiter, criteria );           /* Link everything together and push it to the AS-Stack                                */ \
68
69#define monitor_ctx( mons, cnt )                                /* Define that create the necessary struct for internal/external scheduling operations */ \
70        $monitor ** monitors = mons;                          /* Save the targeted monitors                                                          */ \
71        __lock_size_t count = cnt;                                /* Save the count to a local variable                                                  */ \
72        unsigned int recursions[ count ];                         /* Save the current recursion levels to restore them later                             */ \
73        __waitfor_mask_t masks [ count ];                         /* Save the current waitfor masks to restore them later                                */ \
74        __spinlock_t *   locks [ count ];                         /* We need to pass-in an array of locks to BlockInternal                               */ \
75
76#define monitor_save    save   ( monitors, count, locks, recursions, masks )
77#define monitor_restore restore( monitors, count, locks, recursions, masks )
78
79
80//-----------------------------------------------------------------------------
81// Enter/Leave routines
82// Enter single monitor
83static void __enter( $monitor * this, const __monitor_group_t & group ) {
84        // Lock the monitor spinlock
85        lock( this->lock __cfaabi_dbg_ctx2 );
86        // Interrupts disable inside critical section
87        $thread * thrd = kernelTLS.this_thread;
88
89        __cfaabi_dbg_print_safe( "Kernel : %10p Entering mon %p (%p)\n", thrd, this, this->owner);
90
91        if( !this->owner ) {
92                // No one has the monitor, just take it
93                __set_owner( this, thrd );
94
95                __cfaabi_dbg_print_safe( "Kernel :  mon is free \n" );
96        }
97        else if( this->owner == thrd) {
98                // We already have the monitor, just note how many times we took it
99                this->recursion += 1;
100
101                __cfaabi_dbg_print_safe( "Kernel :  mon already owned \n" );
102        }
103        else if( is_accepted( this, group) ) {
104                // Some one was waiting for us, enter
105                __set_owner( this, thrd );
106
107                // Reset mask
108                reset_mask( this );
109
110                __cfaabi_dbg_print_safe( "Kernel :  mon accepts \n" );
111        }
112        else {
113                __cfaabi_dbg_print_safe( "Kernel :  blocking \n" );
114
115                // Some one else has the monitor, wait in line for it
116                /* paranoid */ verify( thrd->next == 0p );
117                append( this->entry_queue, thrd );
118                /* paranoid */ verify( thrd->next == 1p );
119
120                unlock( this->lock );
121                park();
122
123                __cfaabi_dbg_print_safe( "Kernel : %10p Entered  mon %p\n", thrd, this);
124
125                /* paranoid */ verifyf( kernelTLS.this_thread == this->owner, "Expected owner to be %p, got %p (r: %i, m: %p)", kernelTLS.this_thread, this->owner, this->recursion, this );
126                return;
127        }
128
129        __cfaabi_dbg_print_safe( "Kernel : %10p Entered  mon %p\n", thrd, this);
130
131        /* paranoid */ verifyf( kernelTLS.this_thread == this->owner, "Expected owner to be %p, got %p (r: %i, m: %p)", kernelTLS.this_thread, this->owner, this->recursion, this );
132        /* paranoid */ verify( this->lock.lock );
133
134        // Release the lock and leave
135        unlock( this->lock );
136        return;
137}
138
139static void __dtor_enter( $monitor * this, fptr_t func ) {
140        // Lock the monitor spinlock
141        lock( this->lock __cfaabi_dbg_ctx2 );
142        // Interrupts disable inside critical section
143        $thread * thrd = kernelTLS.this_thread;
144
145        __cfaabi_dbg_print_safe( "Kernel : %10p Entering dtor for mon %p (%p)\n", thrd, this, this->owner);
146
147
148        if( !this->owner ) {
149                __cfaabi_dbg_print_safe( "Kernel : Destroying free mon %p\n", this);
150
151                // No one has the monitor, just take it
152                __set_owner( this, thrd );
153
154                verifyf( kernelTLS.this_thread == this->owner, "Expected owner to be %p, got %p (r: %i, m: %p)", kernelTLS.this_thread, this->owner, this->recursion, this );
155
156                unlock( this->lock );
157                return;
158        }
159        else if( this->owner == thrd) {
160                // We already have the monitor... but where about to destroy it so the nesting will fail
161                // Abort!
162                abort( "Attempt to destroy monitor %p by thread \"%.256s\" (%p) in nested mutex.", this, thrd->self_cor.name, thrd );
163        }
164
165        __lock_size_t count = 1;
166        $monitor ** monitors = &this;
167        __monitor_group_t group = { &this, 1, func };
168        if( is_accepted( this, group) ) {
169                __cfaabi_dbg_print_safe( "Kernel :  mon accepts dtor, block and signal it \n" );
170
171                // Wake the thread that is waiting for this
172                __condition_criterion_t * urgent = pop( this->signal_stack );
173                /* paranoid */ verify( urgent );
174
175                // Reset mask
176                reset_mask( this );
177
178                // Create the node specific to this wait operation
179                wait_ctx_primed( thrd, 0 )
180
181                // Some one else has the monitor, wait for him to finish and then run
182                unlock( this->lock );
183
184                // Release the next thread
185                /* paranoid */ verifyf( urgent->owner->waiting_thread == this->owner, "Expected owner to be %p, got %p (r: %i, m: %p)", kernelTLS.this_thread, this->owner, this->recursion, this );
186                unpark( urgent->owner->waiting_thread );
187
188                // Park current thread waiting
189                park();
190
191                // Some one was waiting for us, enter
192                /* paranoid */ verifyf( kernelTLS.this_thread == this->owner, "Expected owner to be %p, got %p (r: %i, m: %p)", kernelTLS.this_thread, this->owner, this->recursion, this );
193        }
194        else {
195                __cfaabi_dbg_print_safe( "Kernel :  blocking \n" );
196
197                wait_ctx( thrd, 0 )
198                this->dtor_node = &waiter;
199
200                // Some one else has the monitor, wait in line for it
201                /* paranoid */ verify( thrd->next == 0p );
202                append( this->entry_queue, thrd );
203                /* paranoid */ verify( thrd->next == 1p );
204                unlock( this->lock );
205
206                // Park current thread waiting
207                park();
208
209                /* paranoid */ verifyf( kernelTLS.this_thread == this->owner, "Expected owner to be %p, got %p (r: %i, m: %p)", kernelTLS.this_thread, this->owner, this->recursion, this );
210                return;
211        }
212
213        __cfaabi_dbg_print_safe( "Kernel : Destroying %p\n", this);
214
215}
216
217// Leave single monitor
218void __leave( $monitor * this ) {
219        // Lock the monitor spinlock
220        lock( this->lock __cfaabi_dbg_ctx2 );
221
222        __cfaabi_dbg_print_safe( "Kernel : %10p Leaving mon %p (%p)\n", kernelTLS.this_thread, this, this->owner);
223
224        /* paranoid */ verifyf( kernelTLS.this_thread == this->owner, "Expected owner to be %p, got %p (r: %i, m: %p)", kernelTLS.this_thread, this->owner, this->recursion, this );
225
226        // Leaving a recursion level, decrement the counter
227        this->recursion -= 1;
228
229        // If we haven't left the last level of recursion
230        // it means we don't need to do anything
231        if( this->recursion != 0) {
232                __cfaabi_dbg_print_safe( "Kernel :  recursion still %d\n", this->recursion);
233                unlock( this->lock );
234                return;
235        }
236
237        // Get the next thread, will be null on low contention monitor
238        $thread * new_owner = next_thread( this );
239
240        // Check the new owner is consistent with who we wake-up
241        // new_owner might be null even if someone owns the monitor when the owner is still waiting for another monitor
242        /* paranoid */ verifyf( !new_owner || new_owner == this->owner, "Expected owner to be %p, got %p (m: %p)", new_owner, this->owner, this );
243
244        // We can now let other threads in safely
245        unlock( this->lock );
246
247        //We need to wake-up the thread
248        /* paranoid */ verifyf( !new_owner || new_owner == this->owner, "Expected owner to be %p, got %p (m: %p)", new_owner, this->owner, this );
249        unpark( new_owner );
250}
251
252// Leave single monitor for the last time
253void __dtor_leave( $monitor * this ) {
254        __cfaabi_dbg_debug_do(
255                if( TL_GET( this_thread ) != this->owner ) {
256                        abort( "Destroyed monitor %p has inconsistent owner, expected %p got %p.\n", this, TL_GET( this_thread ), this->owner);
257                }
258                if( this->recursion != 1 ) {
259                        abort( "Destroyed monitor %p has %d outstanding nested calls.\n", this, this->recursion - 1);
260                }
261        )
262}
263
264extern "C" {
265        // Leave the thread monitor
266        // last routine called by a thread.
267        // Should never return
268        void __cfactx_thrd_leave() {
269                $thread * thrd = TL_GET( this_thread );
270                $monitor * this = &thrd->self_mon;
271
272                // Lock the monitor now
273                lock( this->lock __cfaabi_dbg_ctx2 );
274
275                disable_interrupts();
276
277                thrd->state = Halted;
278
279                /* paranoid */ verifyf( thrd == this->owner, "Expected owner to be %p, got %p (r: %i, m: %p)", thrd, this->owner, this->recursion, this );
280
281                // Leaving a recursion level, decrement the counter
282                this->recursion -= 1;
283
284                // If we haven't left the last level of recursion
285                // it must mean there is an error
286                if( this->recursion != 0) { abort( "Thread internal monitor has unbalanced recursion" ); }
287
288                // Fetch the next thread, can be null
289                $thread * new_owner = next_thread( this );
290
291                // Release the monitor lock
292                unlock( this->lock );
293
294                // Unpark the next owner if needed
295                /* paranoid */ verifyf( !new_owner || new_owner == this->owner, "Expected owner to be %p, got %p (m: %p)", new_owner, this->owner, this );
296                /* paranoid */ verify( ! kernelTLS.preemption_state.enabled );
297                /* paranoid */ verify( ! kernelTLS.this_processor->destroyer );
298                /* paranoid */ verify( thrd->state == Halted );
299
300                kernelTLS.this_processor->destroyer = new_owner;
301
302                // Leave the thread
303                __leave_thread();
304
305                // Control flow should never reach here!
306        }
307}
308
309// Enter multiple monitor
310// relies on the monitor array being sorted
311static inline void enter( __monitor_group_t monitors ) {
312        for( __lock_size_t i = 0; i < monitors.size; i++) {
313                __enter( monitors[i], monitors );
314        }
315}
316
317// Leave multiple monitor
318// relies on the monitor array being sorted
319static inline void leave($monitor * monitors [], __lock_size_t count) {
320        for( __lock_size_t i = count - 1; i >= 0; i--) {
321                __leave( monitors[i] );
322        }
323}
324
325// Ctor for monitor guard
326// Sorts monitors before entering
327void ?{}( monitor_guard_t & this, $monitor * m [], __lock_size_t count, fptr_t func ) {
328        $thread * thrd = TL_GET( this_thread );
329
330        // Store current array
331        this.m = m;
332        this.count = count;
333
334        // Sort monitors based on address
335        __libcfa_small_sort(this.m, count);
336
337        // Save previous thread context
338        this.prev = thrd->monitors;
339
340        // Update thread context (needed for conditions)
341        (thrd->monitors){m, count, func};
342
343        // __cfaabi_dbg_print_safe( "MGUARD : enter %d\n", count);
344
345        // Enter the monitors in order
346        __monitor_group_t group = {this.m, this.count, func};
347        enter( group );
348
349        // __cfaabi_dbg_print_safe( "MGUARD : entered\n" );
350}
351
352
353// Dtor for monitor guard
354void ^?{}( monitor_guard_t & this ) {
355        // __cfaabi_dbg_print_safe( "MGUARD : leaving %d\n", this.count);
356
357        // Leave the monitors in order
358        leave( this.m, this.count );
359
360        // __cfaabi_dbg_print_safe( "MGUARD : left\n" );
361
362        // Restore thread context
363        TL_GET( this_thread )->monitors = this.prev;
364}
365
366// Ctor for monitor guard
367// Sorts monitors before entering
368void ?{}( monitor_dtor_guard_t & this, $monitor * m [], fptr_t func ) {
369        // optimization
370        $thread * thrd = TL_GET( this_thread );
371
372        // Store current array
373        this.m = *m;
374
375        // Save previous thread context
376        this.prev = thrd->monitors;
377
378        // Update thread context (needed for conditions)
379        (thrd->monitors){m, 1, func};
380
381        __dtor_enter( this.m, func );
382}
383
384// Dtor for monitor guard
385void ^?{}( monitor_dtor_guard_t & this ) {
386        // Leave the monitors in order
387        __dtor_leave( this.m );
388
389        // Restore thread context
390        TL_GET( this_thread )->monitors = this.prev;
391}
392
393//-----------------------------------------------------------------------------
394// Internal scheduling types
395void ?{}(__condition_node_t & this, $thread * waiting_thread, __lock_size_t count, uintptr_t user_info ) {
396        this.waiting_thread = waiting_thread;
397        this.count = count;
398        this.next = 0p;
399        this.user_info = user_info;
400}
401
402void ?{}(__condition_criterion_t & this ) with( this ) {
403        ready  = false;
404        target = 0p;
405        owner  = 0p;
406        next   = 0p;
407}
408
409void ?{}(__condition_criterion_t & this, $monitor * target, __condition_node_t & owner ) {
410        this.ready  = false;
411        this.target = target;
412        this.owner  = &owner;
413        this.next   = 0p;
414}
415
416//-----------------------------------------------------------------------------
417// Internal scheduling
418void wait( condition & this, uintptr_t user_info = 0 ) {
419        brand_condition( this );
420
421        // Check that everything is as expected
422        assertf( this.monitors != 0p, "Waiting with no monitors (%p)", this.monitors );
423        verifyf( this.monitor_count != 0, "Waiting with 0 monitors (%"PRIiFAST16")", this.monitor_count );
424        verifyf( this.monitor_count < 32u, "Excessive monitor count (%"PRIiFAST16")", this.monitor_count );
425
426        // Create storage for monitor context
427        monitor_ctx( this.monitors, this.monitor_count );
428
429        // Create the node specific to this wait operation
430        wait_ctx( TL_GET( this_thread ), user_info );
431
432        // Append the current wait operation to the ones already queued on the condition
433        // We don't need locks for that since conditions must always be waited on inside monitor mutual exclusion
434        /* paranoid */ verify( waiter.next == 0p );
435        append( this.blocked, &waiter );
436        /* paranoid */ verify( waiter.next == 1p );
437
438        // Lock all monitors (aggregates the locks as well)
439        lock_all( monitors, locks, count );
440
441        // Find the next thread(s) to run
442        __lock_size_t thread_count = 0;
443        $thread * threads[ count ];
444        __builtin_memset( threads, 0, sizeof( threads ) );
445
446        // Save monitor states
447        monitor_save;
448
449        // Remove any duplicate threads
450        for( __lock_size_t i = 0; i < count; i++) {
451                $thread * new_owner = next_thread( monitors[i] );
452                insert_unique( threads, thread_count, new_owner );
453        }
454
455        // Unlock the locks, we don't need them anymore
456        for(int i = 0; i < count; i++) {
457                unlock( *locks[i] );
458        }
459
460        // Wake the threads
461        for(int i = 0; i < thread_count; i++) {
462                unpark( threads[i] );
463        }
464
465        // Everything is ready to go to sleep
466        park();
467
468        // We are back, restore the owners and recursions
469        monitor_restore;
470}
471
472bool signal( condition & this ) {
473        if( is_empty( this ) ) { return false; }
474
475        //Check that everything is as expected
476        verify( this.monitors );
477        verify( this.monitor_count != 0 );
478
479        //Some more checking in debug
480        __cfaabi_dbg_debug_do(
481                $thread * this_thrd = TL_GET( this_thread );
482                if ( this.monitor_count != this_thrd->monitors.size ) {
483                        abort( "Signal on condition %p made with different number of monitor(s), expected %zi got %zi", &this, this.monitor_count, this_thrd->monitors.size );
484                }
485
486                for(int i = 0; i < this.monitor_count; i++) {
487                        if ( this.monitors[i] != this_thrd->monitors[i] ) {
488                                abort( "Signal on condition %p made with different monitor, expected %p got %p", &this, this.monitors[i], this_thrd->monitors[i] );
489                        }
490                }
491        );
492
493        __lock_size_t count = this.monitor_count;
494
495        // Lock all monitors
496        lock_all( this.monitors, 0p, count );
497
498        //Pop the head of the waiting queue
499        __condition_node_t * node = pop_head( this.blocked );
500
501        //Add the thread to the proper AS stack
502        for(int i = 0; i < count; i++) {
503                __condition_criterion_t * crit = &node->criteria[i];
504                assert( !crit->ready );
505                push( crit->target->signal_stack, crit );
506        }
507
508        //Release
509        unlock_all( this.monitors, count );
510
511        return true;
512}
513
514bool signal_block( condition & this ) {
515        if( !this.blocked.head ) { return false; }
516
517        //Check that everything is as expected
518        verifyf( this.monitors != 0p, "Waiting with no monitors (%p)", this.monitors );
519        verifyf( this.monitor_count != 0, "Waiting with 0 monitors (%"PRIiFAST16")", this.monitor_count );
520
521        // Create storage for monitor context
522        monitor_ctx( this.monitors, this.monitor_count );
523
524        // Lock all monitors (aggregates the locks them as well)
525        lock_all( monitors, locks, count );
526
527
528        // Create the node specific to this wait operation
529        wait_ctx_primed( kernelTLS.this_thread, 0 )
530
531        //save contexts
532        monitor_save;
533
534        //Find the thread to run
535        $thread * signallee = pop_head( this.blocked )->waiting_thread;
536        __set_owner( monitors, count, signallee );
537
538        __cfaabi_dbg_print_buffer_decl( "Kernel : signal_block condition %p (s: %p)\n", &this, signallee );
539
540        // unlock all the monitors
541        unlock_all( locks, count );
542
543        // unpark the thread we signalled
544        unpark( signallee );
545
546        //Everything is ready to go to sleep
547        park();
548
549
550        // WE WOKE UP
551
552
553        __cfaabi_dbg_print_buffer_local( "Kernel :   signal_block returned\n" );
554
555        //We are back, restore the masks and recursions
556        monitor_restore;
557
558        return true;
559}
560
561// Access the user_info of the thread waiting at the front of the queue
562uintptr_t front( condition & this ) {
563        verifyf( !is_empty(this),
564                "Attempt to access user data on an empty condition.\n"
565                "Possible cause is not checking if the condition is empty before reading stored data."
566        );
567        return ((typeof(this.blocked.head))this.blocked.head)->user_info;
568}
569
570//-----------------------------------------------------------------------------
571// External scheduling
572// cases to handle :
573//      - target already there :
574//              block and wake
575//      - dtor already there
576//              put thread on signaller stack
577//      - non-blocking
578//              return else
579//      - timeout
580//              return timeout
581//      - block
582//              setup mask
583//              block
584void __waitfor_internal( const __waitfor_mask_t & mask, int duration ) {
585        // This statment doesn't have a contiguous list of monitors...
586        // Create one!
587        __lock_size_t max = count_max( mask );
588        $monitor * mon_storage[max];
589        __builtin_memset( mon_storage, 0, sizeof( mon_storage ) );
590        __lock_size_t actual_count = aggregate( mon_storage, mask );
591
592        __cfaabi_dbg_print_buffer_decl( "Kernel : waitfor %"PRIdFAST16" (s: %"PRIdFAST16", m: %"PRIdFAST16")\n", actual_count, mask.size, (__lock_size_t)max);
593
594        if(actual_count == 0) return;
595
596        __cfaabi_dbg_print_buffer_local( "Kernel : waitfor internal proceeding\n" );
597
598        // Create storage for monitor context
599        monitor_ctx( mon_storage, actual_count );
600
601        // Lock all monitors (aggregates the locks as well)
602        lock_all( monitors, locks, count );
603
604        {
605                // Check if the entry queue
606                $thread * next; int index;
607                [next, index] = search_entry_queue( mask, monitors, count );
608
609                if( next ) {
610                        *mask.accepted = index;
611                        __acceptable_t& accepted = mask[index];
612                        if( accepted.is_dtor ) {
613                                __cfaabi_dbg_print_buffer_local( "Kernel : dtor already there\n" );
614                                verifyf( accepted.size == 1,  "ERROR: Accepted dtor has more than 1 mutex parameter." );
615
616                                $monitor * mon2dtor = accepted[0];
617                                verifyf( mon2dtor->dtor_node, "ERROR: Accepted monitor has no dtor_node." );
618
619                                __condition_criterion_t * dtor_crit = mon2dtor->dtor_node->criteria;
620                                push( mon2dtor->signal_stack, dtor_crit );
621
622                                unlock_all( locks, count );
623                        }
624                        else {
625                                __cfaabi_dbg_print_buffer_local( "Kernel : thread present, baton-passing\n" );
626
627                                // Create the node specific to this wait operation
628                                wait_ctx_primed( kernelTLS.this_thread, 0 );
629
630                                // Save monitor states
631                                monitor_save;
632
633                                __cfaabi_dbg_print_buffer_local( "Kernel :  baton of %"PRIdFAST16" monitors : ", count );
634                                #ifdef __CFA_DEBUG_PRINT__
635                                        for( int i = 0; i < count; i++) {
636                                                __cfaabi_dbg_print_buffer_local( "%p %p ", monitors[i], monitors[i]->signal_stack.top );
637                                        }
638                                #endif
639                                __cfaabi_dbg_print_buffer_local( "\n" );
640
641                                // Set the owners to be the next thread
642                                __set_owner( monitors, count, next );
643
644                                // unlock all the monitors
645                                unlock_all( locks, count );
646
647                                // unpark the thread we signalled
648                                unpark( next );
649
650                                //Everything is ready to go to sleep
651                                park();
652
653                                // We are back, restore the owners and recursions
654                                monitor_restore;
655
656                                __cfaabi_dbg_print_buffer_local( "Kernel : thread present, returned\n" );
657                        }
658
659                        __cfaabi_dbg_print_buffer_local( "Kernel : accepted %d\n", *mask.accepted);
660                        return;
661                }
662        }
663
664
665        if( duration == 0 ) {
666                __cfaabi_dbg_print_buffer_local( "Kernel : non-blocking, exiting\n" );
667
668                unlock_all( locks, count );
669
670                __cfaabi_dbg_print_buffer_local( "Kernel : accepted %d\n", *mask.accepted);
671                return;
672        }
673
674
675        verifyf( duration < 0, "Timeout on waitfor statments not supported yet." );
676
677        __cfaabi_dbg_print_buffer_local( "Kernel : blocking waitfor\n" );
678
679        // Create the node specific to this wait operation
680        wait_ctx_primed( kernelTLS.this_thread, 0 );
681
682        monitor_save;
683        set_mask( monitors, count, mask );
684
685        for( __lock_size_t i = 0; i < count; i++) {
686                verify( monitors[i]->owner == kernelTLS.this_thread );
687        }
688
689        // unlock all the monitors
690        unlock_all( locks, count );
691
692        //Everything is ready to go to sleep
693        park();
694
695
696        // WE WOKE UP
697
698
699        //We are back, restore the masks and recursions
700        monitor_restore;
701
702        __cfaabi_dbg_print_buffer_local( "Kernel : exiting\n" );
703
704        __cfaabi_dbg_print_buffer_local( "Kernel : accepted %d\n", *mask.accepted);
705}
706
707//-----------------------------------------------------------------------------
708// Utilities
709
710static inline void __set_owner( $monitor * this, $thread * owner ) {
711        /* paranoid */ verify( this->lock.lock );
712
713        //Pass the monitor appropriately
714        this->owner = owner;
715
716        //We are passing the monitor to someone else, which means recursion level is not 0
717        this->recursion = owner ? 1 : 0;
718}
719
720static inline void __set_owner( $monitor * monitors [], __lock_size_t count, $thread * owner ) {
721        /* paranoid */ verify ( monitors[0]->lock.lock );
722        /* paranoid */ verifyf( monitors[0]->owner == kernelTLS.this_thread, "Expected owner to be %p, got %p (r: %i, m: %p)", kernelTLS.this_thread, monitors[0]->owner, monitors[0]->recursion, monitors[0] );
723        monitors[0]->owner        = owner;
724        monitors[0]->recursion    = 1;
725        for( __lock_size_t i = 1; i < count; i++ ) {
726                /* paranoid */ verify ( monitors[i]->lock.lock );
727                /* paranoid */ verifyf( monitors[i]->owner == kernelTLS.this_thread, "Expected owner to be %p, got %p (r: %i, m: %p)", kernelTLS.this_thread, monitors[i]->owner, monitors[i]->recursion, monitors[i] );
728                monitors[i]->owner        = owner;
729                monitors[i]->recursion    = 0;
730        }
731}
732
733static inline void set_mask( $monitor * storage [], __lock_size_t count, const __waitfor_mask_t & mask ) {
734        for( __lock_size_t i = 0; i < count; i++) {
735                storage[i]->mask = mask;
736        }
737}
738
739static inline void reset_mask( $monitor * this ) {
740        this->mask.accepted = 0p;
741        this->mask.data = 0p;
742        this->mask.size = 0;
743}
744
745static inline $thread * next_thread( $monitor * this ) {
746        //Check the signaller stack
747        __cfaabi_dbg_print_safe( "Kernel :  mon %p AS-stack top %p\n", this, this->signal_stack.top);
748        __condition_criterion_t * urgent = pop( this->signal_stack );
749        if( urgent ) {
750                //The signaller stack is not empty,
751                //regardless of if we are ready to baton pass,
752                //we need to set the monitor as in use
753                /* paranoid */ verifyf( !this->owner || kernelTLS.this_thread == this->owner, "Expected owner to be %p, got %p (r: %i, m: %p)", kernelTLS.this_thread, this->owner, this->recursion, this );
754                __set_owner( this,  urgent->owner->waiting_thread );
755
756                return check_condition( urgent );
757        }
758
759        // No signaller thread
760        // Get the next thread in the entry_queue
761        $thread * new_owner = pop_head( this->entry_queue );
762        /* paranoid */ verifyf( !this->owner || kernelTLS.this_thread == this->owner, "Expected owner to be %p, got %p (r: %i, m: %p)", kernelTLS.this_thread, this->owner, this->recursion, this );
763        /* paranoid */ verify( !new_owner || new_owner->next == 0p );
764        __set_owner( this, new_owner );
765
766        return new_owner;
767}
768
769static inline bool is_accepted( $monitor * this, const __monitor_group_t & group ) {
770        __acceptable_t * it = this->mask.data; // Optim
771        __lock_size_t count = this->mask.size;
772
773        // Check if there are any acceptable functions
774        if( !it ) return false;
775
776        // If this isn't the first monitor to test this, there is no reason to repeat the test.
777        if( this != group[0] ) return group[0]->mask.accepted >= 0;
778
779        // For all acceptable functions check if this is the current function.
780        for( __lock_size_t i = 0; i < count; i++, it++ ) {
781                if( *it == group ) {
782                        *this->mask.accepted = i;
783                        return true;
784                }
785        }
786
787        // No function matched
788        return false;
789}
790
791static inline void init( __lock_size_t count, $monitor * monitors [], __condition_node_t & waiter, __condition_criterion_t criteria [] ) {
792        for( __lock_size_t i = 0; i < count; i++) {
793                (criteria[i]){ monitors[i], waiter };
794        }
795
796        waiter.criteria = criteria;
797}
798
799static inline void init_push( __lock_size_t count, $monitor * monitors [], __condition_node_t & waiter, __condition_criterion_t criteria [] ) {
800        for( __lock_size_t i = 0; i < count; i++) {
801                (criteria[i]){ monitors[i], waiter };
802                __cfaabi_dbg_print_safe( "Kernel :  target %p = %p\n", criteria[i].target, &criteria[i] );
803                push( criteria[i].target->signal_stack, &criteria[i] );
804        }
805
806        waiter.criteria = criteria;
807}
808
809static inline void lock_all( __spinlock_t * locks [], __lock_size_t count ) {
810        for( __lock_size_t i = 0; i < count; i++ ) {
811                lock( *locks[i] __cfaabi_dbg_ctx2 );
812        }
813}
814
815static inline void lock_all( $monitor * source [], __spinlock_t * /*out*/ locks [], __lock_size_t count ) {
816        for( __lock_size_t i = 0; i < count; i++ ) {
817                __spinlock_t * l = &source[i]->lock;
818                lock( *l __cfaabi_dbg_ctx2 );
819                if(locks) locks[i] = l;
820        }
821}
822
823static inline void unlock_all( __spinlock_t * locks [], __lock_size_t count ) {
824        for( __lock_size_t i = 0; i < count; i++ ) {
825                unlock( *locks[i] );
826        }
827}
828
829static inline void unlock_all( $monitor * locks [], __lock_size_t count ) {
830        for( __lock_size_t i = 0; i < count; i++ ) {
831                unlock( locks[i]->lock );
832        }
833}
834
835static inline void save(
836        $monitor * ctx [],
837        __lock_size_t count,
838        __attribute((unused)) __spinlock_t * locks [],
839        unsigned int /*out*/ recursions [],
840        __waitfor_mask_t /*out*/ masks []
841) {
842        for( __lock_size_t i = 0; i < count; i++ ) {
843                recursions[i] = ctx[i]->recursion;
844                masks[i]      = ctx[i]->mask;
845        }
846}
847
848static inline void restore(
849        $monitor * ctx [],
850        __lock_size_t count,
851        __spinlock_t * locks [],
852        unsigned int /*out*/ recursions [],
853        __waitfor_mask_t /*out*/ masks []
854) {
855        lock_all( locks, count );
856        for( __lock_size_t i = 0; i < count; i++ ) {
857                ctx[i]->recursion = recursions[i];
858                ctx[i]->mask      = masks[i];
859        }
860        unlock_all( locks, count );
861}
862
863// Function has 2 different behavior
864// 1 - Marks a monitors as being ready to run
865// 2 - Checks if all the monitors are ready to run
866//     if so return the thread to run
867static inline $thread * check_condition( __condition_criterion_t * target ) {
868        __condition_node_t * node = target->owner;
869        unsigned short count = node->count;
870        __condition_criterion_t * criteria = node->criteria;
871
872        bool ready2run = true;
873
874        for(    int i = 0; i < count; i++ ) {
875
876                // __cfaabi_dbg_print_safe( "Checking %p for %p\n", &criteria[i], target );
877                if( &criteria[i] == target ) {
878                        criteria[i].ready = true;
879                        // __cfaabi_dbg_print_safe( "True\n" );
880                }
881
882                ready2run = criteria[i].ready && ready2run;
883        }
884
885        __cfaabi_dbg_print_safe( "Kernel :  Runing %i (%p)\n", ready2run, ready2run ? node->waiting_thread : 0p );
886        return ready2run ? node->waiting_thread : 0p;
887}
888
889static inline void brand_condition( condition & this ) {
890        $thread * thrd = TL_GET( this_thread );
891        if( !this.monitors ) {
892                // __cfaabi_dbg_print_safe( "Branding\n" );
893                assertf( thrd->monitors.data != 0p, "No current monitor to brand condition %p", thrd->monitors.data );
894                this.monitor_count = thrd->monitors.size;
895
896                this.monitors = ($monitor **)malloc( this.monitor_count * sizeof( *this.monitors ) );
897                for( int i = 0; i < this.monitor_count; i++ ) {
898                        this.monitors[i] = thrd->monitors[i];
899                }
900        }
901}
902
903static inline [$thread *, int] search_entry_queue( const __waitfor_mask_t & mask, $monitor * monitors [], __lock_size_t count ) {
904
905        __queue_t($thread) & entry_queue = monitors[0]->entry_queue;
906
907        // For each thread in the entry-queue
908        for(    $thread ** thrd_it = &entry_queue.head;
909                *thrd_it != 1p;
910                thrd_it = &(*thrd_it)->next
911        ) {
912                // For each acceptable check if it matches
913                int i = 0;
914                __acceptable_t * end   = end  (mask);
915                __acceptable_t * begin = begin(mask);
916                for( __acceptable_t * it = begin; it != end; it++, i++ ) {
917                        // Check if we have a match
918                        if( *it == (*thrd_it)->monitors ) {
919
920                                // If we have a match return it
921                                // after removeing it from the entry queue
922                                return [remove( entry_queue, thrd_it ), i];
923                        }
924                }
925        }
926
927        return [0, -1];
928}
929
930forall(dtype T | sized( T ))
931static inline __lock_size_t insert_unique( T * array [], __lock_size_t & size, T * val ) {
932        if( !val ) return size;
933
934        for( __lock_size_t i = 0; i <= size; i++) {
935                if( array[i] == val ) return size;
936        }
937
938        array[size] = val;
939        size = size + 1;
940        return size;
941}
942
943static inline __lock_size_t count_max( const __waitfor_mask_t & mask ) {
944        __lock_size_t max = 0;
945        for( __lock_size_t i = 0; i < mask.size; i++ ) {
946                __acceptable_t & accepted = mask[i];
947                max += accepted.size;
948        }
949        return max;
950}
951
952static inline __lock_size_t aggregate( $monitor * storage [], const __waitfor_mask_t & mask ) {
953        __lock_size_t size = 0;
954        for( __lock_size_t i = 0; i < mask.size; i++ ) {
955                __acceptable_t & accepted = mask[i];
956                __libcfa_small_sort( accepted.data, accepted.size );
957                for( __lock_size_t j = 0; j < accepted.size; j++) {
958                        insert_unique( storage, size, accepted[j] );
959                }
960        }
961        // TODO insertion sort instead of this
962        __libcfa_small_sort( storage, size );
963        return size;
964}
965
966// Local Variables: //
967// mode: c //
968// tab-width: 4 //
969// End: //
Note: See TracBrowser for help on using the repository browser.