source: libcfa/src/concurrency/coroutine.cfa @ 5c1a531

arm-ehcleanup-dtorsjacob/cs343-translationjenkins-sandboxnew-astnew-ast-unique-expr
Last change on this file since 5c1a531 was 5c1a531, checked in by tdelisle <tdelisle@…>, 3 years ago

Fixed errno virtualization and enabled preemption during coroutine context switch

  • Property mode set to 100644
File size: 7.3 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// coroutine.c --
8//
9// Author           : Thierry Delisle
10// Created On       : Mon Nov 28 12:27:26 2016
11// Last Modified By : Peter A. Buhr
12// Last Modified On : Fri Mar 30 17:20:57 2018
13// Update Count     : 9
14//
15
16#include "coroutine.hfa"
17
18extern "C" {
19#include <stddef.h>
20#include <malloc.h>
21#include <errno.h>
22#include <string.h>
23#include <unistd.h>
24// use this define to make unwind.h play nice, definetely a hack
25#define HIDE_EXPORTS
26#include <unwind.h>
27#undef HIDE_EXPORTS
28#include <sys/mman.h>
29}
30
31#include "kernel_private.hfa"
32
33#define __CFA_INVOKE_PRIVATE__
34#include "invoke.h"
35
36extern "C" {
37        void _CtxCoroutine_Unwind(struct _Unwind_Exception * storage, struct coroutine_desc *) __attribute__ ((__noreturn__));
38        static void _CtxCoroutine_UnwindCleanup(_Unwind_Reason_Code, struct _Unwind_Exception *) __attribute__ ((__noreturn__));
39        static void _CtxCoroutine_UnwindCleanup(_Unwind_Reason_Code, struct _Unwind_Exception *) {
40                abort();
41        }
42}
43
44//-----------------------------------------------------------------------------
45// Global state variables
46
47// minimum feasible stack size in bytes
48#define MinStackSize 1000
49extern size_t __page_size;                              // architecture pagesize HACK, should go in proper runtime singleton
50
51void __stack_prepare( __stack_info_t * this, size_t create_size );
52
53//-----------------------------------------------------------------------------
54// Coroutine ctors and dtors
55void ?{}( __stack_info_t & this, void * storage, size_t storageSize ) {
56        this.storage   = (__stack_t *)storage;
57
58        // Did we get a piece of storage ?
59        if (this.storage || storageSize != 0) {
60                // We either got a piece of storage or the user asked for a specific size
61                // Immediately create the stack
62                // (This is slightly unintuitive that non-default sized coroutines create are eagerly created
63                // but it avoids that all coroutines carry an unnecessary size)
64                verify( storageSize != 0 );
65                __stack_prepare( &this, storageSize );
66        }
67}
68
69void ^?{}(__stack_info_t & this) {
70        bool userStack = ((intptr_t)this.storage & 0x1) != 0;
71        if ( ! userStack && this.storage ) {
72                *((intptr_t*)&this.storage) &= (intptr_t)-1;
73                void * storage = this.storage->limit;
74                __cfaabi_dbg_debug_do(
75                        storage = (char*)(storage) - __page_size;
76                        if ( mprotect( storage, __page_size, PROT_READ | PROT_WRITE ) == -1 ) {
77                                abort( "(coStack_t *)%p.^?{}() : internal error, mprotect failure, error(%d) %s.", &this, errno, strerror( errno ) );
78                        }
79                );
80                __cfaabi_dbg_print_safe("Kernel : Deleting stack %p\n", storage);
81                free( storage );
82        }
83}
84
85void ?{}( coroutine_desc & this, const char * name, void * storage, size_t storageSize ) with( this ) {
86        (this.context){NULL, NULL};
87        (this.stack){storage, storageSize};
88        this.name = name;
89        state = Start;
90        starter = NULL;
91        last = NULL;
92        cancellation = NULL;
93}
94
95void ^?{}(coroutine_desc& this) {
96        if(this.state != Halted && this.state != Start) {
97                coroutine_desc * src = TL_GET( this_thread )->curr_cor;
98                coroutine_desc * dst = &this;
99
100                struct _Unwind_Exception storage;
101                storage.exception_class = -1;
102                storage.exception_cleanup = _CtxCoroutine_UnwindCleanup;
103                this.cancellation = &storage;
104                this.last = src;
105
106                // not resuming self ?
107                if ( src == dst ) {
108                        abort( "Attempt by coroutine %.256s (%p) to terminate itself.\n", src->name, src );
109                }
110
111                CoroutineCtxSwitch( src, dst );
112        }
113}
114
115// Part of the Public API
116// Not inline since only ever called once per coroutine
117forall(dtype T | is_coroutine(T))
118void prime(T& cor) {
119        coroutine_desc* this = get_coroutine(cor);
120        assert(this->state == Start);
121
122        this->state = Primed;
123        resume(cor);
124}
125
126// Wrapper for co
127void CoroutineCtxSwitch(coroutine_desc* src, coroutine_desc* dst) {
128        // set state of current coroutine to inactive
129        src->state = src->state == Halted ? Halted : Inactive;
130
131        // set new coroutine that task is executing
132        TL_GET( this_thread )->curr_cor = dst;
133
134        // context switch to specified coroutine
135        verify( dst->context.SP );
136        CtxSwitch( &src->context, &dst->context );
137        // when CtxSwitch returns we are back in the src coroutine
138
139        // set state of new coroutine to active
140        src->state = Active;
141
142        if( unlikely(src->cancellation != NULL) ) {
143                _CtxCoroutine_Unwind(src->cancellation, src);
144        }
145}
146
147[void *, size_t] __stack_alloc( size_t storageSize ) {
148        static const size_t stack_data_size = libCeiling( sizeof(__stack_t), 16 ); // minimum alignment
149        assert(__page_size != 0l);
150        size_t size = libCeiling( storageSize, 16 ) + stack_data_size;
151
152        // If we are running debug, we also need to allocate a guardpage to catch stack overflows.
153        void * storage;
154        __cfaabi_dbg_debug_do(
155                storage = memalign( __page_size, size + __page_size );
156        );
157        __cfaabi_dbg_no_debug_do(
158                storage = (void*)malloc(size);
159        );
160
161        __cfaabi_dbg_print_safe("Kernel : Created stack %p of size %zu\n", storage, size);
162        __cfaabi_dbg_debug_do(
163                if ( mprotect( storage, __page_size, PROT_NONE ) == -1 ) {
164                        abort( "__stack_alloc : internal error, mprotect failure, error(%d) %s.", (int)errno, strerror( (int)errno ) );
165                }
166                storage = (void *)(((intptr_t)storage) + __page_size);
167        );
168
169        verify( ((intptr_t)storage & (libAlign() - 1)) == 0ul );
170        return [storage, size];
171}
172
173void __stack_prepare( __stack_info_t * this, size_t create_size ) {
174        static const size_t stack_data_size = libCeiling( sizeof(__stack_t), 16 ); // minimum alignment
175        bool userStack;
176        void * storage;
177        size_t size;
178        if ( !this->storage ) {
179                userStack = false;
180                [storage, size] = __stack_alloc( create_size );
181        } else {
182                userStack = true;
183                __cfaabi_dbg_print_safe("Kernel : stack obj %p using user stack %p(%zd bytes)\n", this, this->storage, (intptr_t)this->storage->limit - (intptr_t)this->storage->base);
184
185                // The stack must be aligned, advance the pointer to the next align data
186                storage = (void*)libCeiling( (intptr_t)this->storage, libAlign());
187
188                // The size needs to be shrinked to fit all the extra data structure and be aligned
189                ptrdiff_t diff = (intptr_t)storage - (intptr_t)this->storage;
190                size = libFloor(create_size - stack_data_size - diff, libAlign());
191        } // if
192        assertf( size >= MinStackSize, "Stack size %zd provides less than minimum of %d bytes for a stack.", size, MinStackSize );
193
194        this->storage = (__stack_t *)((intptr_t)storage + size);
195        this->storage->limit = storage;
196        this->storage->base  = (void*)((intptr_t)storage + size);
197        *((intptr_t*)&this->storage) |= userStack ? 0x1 : 0x0;
198}
199
200// We need to call suspend from invoke.c, so we expose this wrapper that
201// is not inline (We can't inline Cforall in C)
202extern "C" {
203        void __suspend_internal(void) {
204                suspend();
205        }
206
207        void __leave_coroutine( coroutine_desc * src ) {
208                coroutine_desc * starter = src->cancellation != 0 ? src->last : src->starter;
209
210                src->state = Halted;
211
212                assertf( starter != 0,
213                        "Attempt to suspend/leave coroutine \"%.256s\" (%p) that has never been resumed.\n"
214                        "Possible cause is a suspend executed in a member called by a coroutine user rather than by the coroutine main.",
215                        src->name, src );
216                assertf( starter->state != Halted,
217                        "Attempt by coroutine \"%.256s\" (%p) to suspend/leave back to terminated coroutine \"%.256s\" (%p).\n"
218                        "Possible cause is terminated coroutine's main routine has already returned.",
219                        src->name, src, starter->name, starter );
220
221                CoroutineCtxSwitch( src, starter );
222        }
223}
224
225// Local Variables: //
226// mode: c //
227// tab-width: 4 //
228// End: //
Note: See TracBrowser for help on using the repository browser.