source: libcfa/src/concurrency/invoke.c @ 76e069f

aaron-thesisarm-ehcleanup-dtorsdeferred_resnjacob/cs343-translationjenkins-sandboxnew-astnew-ast-unique-exprno_listpersistent-indexer
Last change on this file since 76e069f was 76e069f, checked in by Thierry Delisle <tdelisle@…>, 3 years ago

Implemented stack unwinding on coroutine termination

  • Property mode set to 100644
File size: 6.6 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// invoke.c --
8//
9// Author           : Thierry Delisle
10// Created On       : Tue Jan 17 12:27:26 2016
11// Last Modified By : Peter A. Buhr
12// Last Modified On : Fri Feb  9 16:37:42 2018
13// Update Count     : 5
14//
15
16#include <stdbool.h>
17#include <stdlib.h>
18#include <stdio.h>
19#include <unwind.h>
20
21#include "invoke.h"
22
23#define __CFA_INVOKE_PRIVATE__
24#include "invoke.h"
25
26// magically invoke the "main" of the most derived class
27// Called from the kernel when starting a coroutine or task so must switch back to user mode.
28
29extern void __suspend_internal(void);
30extern void __leave_coroutine(void);
31extern void __finish_creation(void);
32extern void __leave_thread_monitor( struct thread_desc * this );
33extern void disable_interrupts();
34extern void enable_interrupts( __cfaabi_dbg_ctx_param );
35
36void CtxInvokeCoroutine(
37        void (*main)(void *),
38        struct coroutine_desc *(*get_coroutine)(void *),
39        void *this
40) {
41        struct coroutine_desc* cor = get_coroutine( this );
42
43        if(cor->state == Primed) {
44                __suspend_internal();
45        }
46
47        cor->state = Active;
48
49        enable_interrupts( __cfaabi_dbg_ctx );
50
51        main( this );
52
53        //Final suspend, should never return
54        __leave_coroutine();
55        __cabi_abort( "Resumed dead coroutine" );
56}
57
58static _Unwind_Reason_Code _CtxCoroutine_UnwindStop(
59        __attribute((__unused__)) int version,
60        _Unwind_Action actions,
61        __attribute((__unused__)) _Unwind_Exception_Class exceptionClass,
62        __attribute((__unused__)) struct _Unwind_Exception * unwind_exception,
63        __attribute((__unused__)) struct _Unwind_Context * context,
64        __attribute((__unused__)) void * param
65) {
66        if( actions & _UA_END_OF_STACK  ) {
67                // We finished unwinding the coroutine,
68                // leave it
69                __leave_coroutine();
70                __cabi_abort( "Resumed dead coroutine" );
71        }
72        if( actions & _UA_CLEANUP_PHASE ) return _URC_NO_REASON;
73
74        return _URC_FATAL_PHASE2_ERROR;
75}
76
77void _CtxCoroutine_Unwind(struct _Unwind_Exception * storage) __attribute__ ((__noreturn__));
78void _CtxCoroutine_Unwind(struct _Unwind_Exception * storage) {
79        _Unwind_Reason_Code ret = _Unwind_ForcedUnwind( storage, _CtxCoroutine_UnwindStop, NULL );
80        printf("UNWIND ERROR %d after force unwind\n", ret);
81        abort();
82}
83
84void CtxInvokeThread(
85        void (*dtor)(void *),
86        void (*main)(void *),
87        struct thread_desc *(*get_thread)(void *),
88        void *this
89) {
90        // First suspend, once the thread arrives here,
91        // the function pointer to main can be invalidated without risk
92        __finish_creation();
93
94        // Fetch the thread handle from the user defined thread structure
95        struct thread_desc* thrd = get_thread( this );
96        thrd->self_cor.last = NULL;
97
98        // Officially start the thread by enabling preemption
99        enable_interrupts( __cfaabi_dbg_ctx );
100
101        // Call the main of the thread
102        main( this );
103
104        // To exit a thread we must :
105        // 1 - Mark it as halted
106        // 2 - Leave its monitor
107        // 3 - Disable the interupts
108        // 4 - Final suspend
109        // The order of these 4 operations is very important
110        //Final suspend, should never return
111        __leave_thread_monitor( thrd );
112        __cabi_abort( "Resumed dead thread" );
113}
114
115
116void CtxStart(
117        void (*main)(void *),
118        struct coroutine_desc *(*get_coroutine)(void *),
119        void *this,
120        void (*invoke)(void *)
121) {
122        struct coStack_t* stack = &get_coroutine( this )->stack;
123
124#if defined( __i386 )
125
126        struct FakeStack {
127            void *fixedRegisters[3];                    // fixed registers ebx, edi, esi (popped on 1st uSwitch, values unimportant)
128            uint32_t mxcr;                        // SSE Status and Control bits (control bits are preserved across function calls)
129            uint16_t fcw;                         // X97 FPU control word (preserved across function calls)
130            void *rturn;                          // where to go on return from uSwitch
131            void *dummyReturn;                          // fake return compiler would have pushed on call to uInvoke
132            void *argument[3];                          // for 16-byte ABI, 16-byte alignment starts here
133            void *padding;                              // padding to force 16-byte alignment, as "base" is 16-byte aligned
134        };
135
136        ((struct machine_context_t *)stack->context)->SP = (char *)stack->base - sizeof( struct FakeStack );
137        ((struct machine_context_t *)stack->context)->FP = NULL;                // terminate stack with NULL fp
138
139        ((struct FakeStack *)(((struct machine_context_t *)stack->context)->SP))->dummyReturn = NULL;
140        ((struct FakeStack *)(((struct machine_context_t *)stack->context)->SP))->argument[0] = this;     // argument to invoke
141        ((struct FakeStack *)(((struct machine_context_t *)stack->context)->SP))->rturn = invoke;
142        ((struct FakeStack *)(((struct machine_context_t *)stack->context)->SP))->mxcr = 0x1F80; //Vol. 2A 3-520
143        ((struct FakeStack *)(((struct machine_context_t *)stack->context)->SP))->fcw = 0x037F;  //Vol. 1 8-7
144
145#elif defined( __x86_64 )
146
147        struct FakeStack {
148                void *fixedRegisters[5];            // fixed registers rbx, r12, r13, r14, r15
149                uint32_t mxcr;                      // SSE Status and Control bits (control bits are preserved across function calls)
150                uint16_t fcw;                       // X97 FPU control word (preserved across function calls)
151                void *rturn;                        // where to go on return from uSwitch
152                void *dummyReturn;                  // NULL return address to provide proper alignment
153        };
154
155        ((struct machine_context_t *)stack->context)->SP = (char *)stack->base - sizeof( struct FakeStack );
156        ((struct machine_context_t *)stack->context)->FP = NULL;                // terminate stack with NULL fp
157
158        ((struct FakeStack *)(((struct machine_context_t *)stack->context)->SP))->dummyReturn = NULL;
159        ((struct FakeStack *)(((struct machine_context_t *)stack->context)->SP))->rturn = CtxInvokeStub;
160        ((struct FakeStack *)(((struct machine_context_t *)stack->context)->SP))->fixedRegisters[0] = this;
161        ((struct FakeStack *)(((struct machine_context_t *)stack->context)->SP))->fixedRegisters[1] = invoke;
162        ((struct FakeStack *)(((struct machine_context_t *)stack->context)->SP))->mxcr = 0x1F80; //Vol. 2A 3-520
163        ((struct FakeStack *)(((struct machine_context_t *)stack->context)->SP))->fcw = 0x037F;  //Vol. 1 8-7
164
165#elif defined( __ARM_ARCH )
166
167        struct FakeStack {
168                float fpRegs[16];                       // floating point registers
169                void *intRegs[9];                       // integer/pointer registers
170                void *arg[2];                           // placeholder for this pointer
171        };
172
173        ((struct machine_context_t *)stack->context)->SP = (char *)stack->base - sizeof( struct FakeStack );
174        ((struct machine_context_t *)stack->context)->FP = NULL;
175
176        struct FakeStack *fs = (struct FakeStack *)((struct machine_context_t *)stack->context)->SP;
177
178        fs->intRegs[8] = CtxInvokeStub;
179        fs->arg[0] = this;
180        fs->arg[1] = invoke;
181
182#else
183        #error uknown hardware architecture
184#endif
185}
186
187// Local Variables: //
188// mode: c //
189// tab-width: 4 //
190// End: //
Note: See TracBrowser for help on using the repository browser.