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 -- |
---|
8 | // |
---|
9 | // Author : Thierry Delisle |
---|
10 | // Created On : Tue Jan 17 12:27:26 2017 |
---|
11 | // Last Modified By : Peter A. Buhr |
---|
12 | // Last Modified On : Tue Feb 4 12:29:26 2020 |
---|
13 | // Update Count : 22 |
---|
14 | // |
---|
15 | |
---|
16 | #pragma once |
---|
17 | |
---|
18 | #include <stdbool.h> |
---|
19 | #include <stdint.h> |
---|
20 | |
---|
21 | #include "invoke.h" |
---|
22 | #include "time_t.hfa" |
---|
23 | #include "coroutine.hfa" |
---|
24 | |
---|
25 | extern "C" { |
---|
26 | #include <pthread.h> |
---|
27 | #include <semaphore.h> |
---|
28 | } |
---|
29 | |
---|
30 | //----------------------------------------------------------------------------- |
---|
31 | // Locks |
---|
32 | struct semaphore { |
---|
33 | __spinlock_t lock; |
---|
34 | int count; |
---|
35 | __queue_t($thread) waiting; |
---|
36 | }; |
---|
37 | |
---|
38 | void ?{}(semaphore & this, int count = 1); |
---|
39 | void ^?{}(semaphore & this); |
---|
40 | void P (semaphore & this); |
---|
41 | bool V (semaphore & this); |
---|
42 | bool V (semaphore & this, unsigned count); |
---|
43 | |
---|
44 | |
---|
45 | //----------------------------------------------------------------------------- |
---|
46 | // Processor |
---|
47 | extern struct cluster * mainCluster; |
---|
48 | |
---|
49 | // Processor |
---|
50 | coroutine processorCtx_t { |
---|
51 | struct processor * proc; |
---|
52 | }; |
---|
53 | |
---|
54 | // Wrapper around kernel threads |
---|
55 | struct processor { |
---|
56 | // Main state |
---|
57 | // Coroutine ctx who does keeps the state of the processor |
---|
58 | struct processorCtx_t runner; |
---|
59 | |
---|
60 | // Cluster from which to get threads |
---|
61 | struct cluster * cltr; |
---|
62 | |
---|
63 | // Name of the processor |
---|
64 | const char * name; |
---|
65 | |
---|
66 | // Handle to pthreads |
---|
67 | pthread_t kernel_thread; |
---|
68 | |
---|
69 | // RunThread data |
---|
70 | // Action to do after a thread is ran |
---|
71 | $thread * destroyer; |
---|
72 | |
---|
73 | // Preemption data |
---|
74 | // Node which is added in the discrete event simulaiton |
---|
75 | struct alarm_node_t * preemption_alarm; |
---|
76 | |
---|
77 | // If true, a preemption was triggered in an unsafe region, the processor must preempt as soon as possible |
---|
78 | bool pending_preemption; |
---|
79 | |
---|
80 | // Idle lock (kernel semaphore) |
---|
81 | __bin_sem_t idle; |
---|
82 | |
---|
83 | // Termination |
---|
84 | // Set to true to notify the processor should terminate |
---|
85 | volatile bool do_terminate; |
---|
86 | |
---|
87 | // Termination synchronisation (user semaphore) |
---|
88 | semaphore terminated; |
---|
89 | |
---|
90 | // pthread Stack |
---|
91 | void * stack; |
---|
92 | |
---|
93 | // Link lists fields |
---|
94 | struct __dbg_node_proc { |
---|
95 | struct processor * next; |
---|
96 | struct processor * prev; |
---|
97 | } node; |
---|
98 | |
---|
99 | #ifdef __CFA_DEBUG__ |
---|
100 | // Last function to enable preemption on this processor |
---|
101 | const char * last_enable; |
---|
102 | #endif |
---|
103 | }; |
---|
104 | |
---|
105 | void ?{}(processor & this, const char name[], struct cluster & cltr); |
---|
106 | void ^?{}(processor & this); |
---|
107 | |
---|
108 | static inline void ?{}(processor & this) { this{ "Anonymous Processor", *mainCluster}; } |
---|
109 | static inline void ?{}(processor & this, struct cluster & cltr) { this{ "Anonymous Processor", cltr}; } |
---|
110 | static inline void ?{}(processor & this, const char name[]) { this{name, *mainCluster }; } |
---|
111 | |
---|
112 | static inline [processor *&, processor *& ] __get( processor & this ) __attribute__((const)) { return this.node.[next, prev]; } |
---|
113 | |
---|
114 | //----------------------------------------------------------------------------- |
---|
115 | // I/O |
---|
116 | #if defined(HAVE_LINUX_IO_URING_H) |
---|
117 | struct io_uring_sq { |
---|
118 | // Head and tail of the ring (associated with array) |
---|
119 | volatile uint32_t * head; |
---|
120 | volatile uint32_t * tail; |
---|
121 | |
---|
122 | // The actual kernel ring which uses head/tail |
---|
123 | // indexes into the sqes arrays |
---|
124 | uint32_t * array; |
---|
125 | |
---|
126 | // number of entries and mask to go with it |
---|
127 | const uint32_t * num; |
---|
128 | const uint32_t * mask; |
---|
129 | |
---|
130 | // Submission flags (Not sure what for) |
---|
131 | uint32_t * flags; |
---|
132 | |
---|
133 | // number of sqes not submitted (whatever that means) |
---|
134 | uint32_t * dropped; |
---|
135 | |
---|
136 | // Like head/tail but not seen by the kernel |
---|
137 | volatile uint32_t alloc; |
---|
138 | |
---|
139 | __spinlock_t lock; |
---|
140 | |
---|
141 | // A buffer of sqes (not the actual ring) |
---|
142 | struct io_uring_sqe * sqes; |
---|
143 | |
---|
144 | // The location and size of the mmaped area |
---|
145 | void * ring_ptr; |
---|
146 | size_t ring_sz; |
---|
147 | |
---|
148 | // Statistics |
---|
149 | #if !defined(__CFA_NO_STATISTICS__) |
---|
150 | struct { |
---|
151 | struct { |
---|
152 | unsigned long long int val; |
---|
153 | unsigned long long int cnt; |
---|
154 | } submit_avg; |
---|
155 | } stats; |
---|
156 | #endif |
---|
157 | }; |
---|
158 | |
---|
159 | struct io_uring_cq { |
---|
160 | // Head and tail of the ring |
---|
161 | volatile uint32_t * head; |
---|
162 | volatile uint32_t * tail; |
---|
163 | |
---|
164 | // number of entries and mask to go with it |
---|
165 | const uint32_t * mask; |
---|
166 | const uint32_t * num; |
---|
167 | |
---|
168 | // number of cqes not submitted (whatever that means) |
---|
169 | uint32_t * overflow; |
---|
170 | |
---|
171 | // the kernel ring |
---|
172 | struct io_uring_cqe * cqes; |
---|
173 | |
---|
174 | // The location and size of the mmaped area |
---|
175 | void * ring_ptr; |
---|
176 | size_t ring_sz; |
---|
177 | |
---|
178 | // Statistics |
---|
179 | #if !defined(__CFA_NO_STATISTICS__) |
---|
180 | struct { |
---|
181 | struct { |
---|
182 | unsigned long long int val; |
---|
183 | unsigned long long int cnt; |
---|
184 | } completed_avg; |
---|
185 | } stats; |
---|
186 | #endif |
---|
187 | }; |
---|
188 | |
---|
189 | struct io_ring { |
---|
190 | struct io_uring_sq submit_q; |
---|
191 | struct io_uring_cq completion_q; |
---|
192 | uint32_t flags; |
---|
193 | int fd; |
---|
194 | pthread_t poller; |
---|
195 | void * stack; |
---|
196 | volatile bool done; |
---|
197 | semaphore submit; |
---|
198 | }; |
---|
199 | #endif |
---|
200 | |
---|
201 | //----------------------------------------------------------------------------- |
---|
202 | // Cluster |
---|
203 | struct cluster { |
---|
204 | // Ready queue locks |
---|
205 | __spinlock_t ready_queue_lock; |
---|
206 | |
---|
207 | // Ready queue for threads |
---|
208 | __queue_t($thread) ready_queue; |
---|
209 | |
---|
210 | // Name of the cluster |
---|
211 | const char * name; |
---|
212 | |
---|
213 | // Preemption rate on this cluster |
---|
214 | Duration preemption_rate; |
---|
215 | |
---|
216 | // List of processors |
---|
217 | __spinlock_t idle_lock; |
---|
218 | __dllist_t(struct processor) procs; |
---|
219 | __dllist_t(struct processor) idles; |
---|
220 | unsigned int nprocessors; |
---|
221 | |
---|
222 | // List of threads |
---|
223 | __spinlock_t thread_list_lock; |
---|
224 | __dllist_t(struct $thread) threads; |
---|
225 | unsigned int nthreads; |
---|
226 | |
---|
227 | // Link lists fields |
---|
228 | struct __dbg_node_cltr { |
---|
229 | cluster * next; |
---|
230 | cluster * prev; |
---|
231 | } node; |
---|
232 | |
---|
233 | #if defined(HAVE_LINUX_IO_URING_H) |
---|
234 | struct io_ring io; |
---|
235 | #endif |
---|
236 | |
---|
237 | #if !defined(__CFA_NO_STATISTICS__) |
---|
238 | bool print_stats; |
---|
239 | #endif |
---|
240 | }; |
---|
241 | extern Duration default_preemption(); |
---|
242 | |
---|
243 | void ?{} (cluster & this, const char name[], Duration preemption_rate); |
---|
244 | void ^?{}(cluster & this); |
---|
245 | |
---|
246 | static inline void ?{} (cluster & this) { this{"Anonymous Cluster", default_preemption()}; } |
---|
247 | static inline void ?{} (cluster & this, Duration preemption_rate) { this{"Anonymous Cluster", preemption_rate}; } |
---|
248 | static inline void ?{} (cluster & this, const char name[]) { this{name, default_preemption()}; } |
---|
249 | |
---|
250 | static inline [cluster *&, cluster *& ] __get( cluster & this ) __attribute__((const)) { return this.node.[next, prev]; } |
---|
251 | |
---|
252 | static inline struct processor * active_processor() { return TL_GET( this_processor ); } // UNSAFE |
---|
253 | static inline struct cluster * active_cluster () { return TL_GET( this_processor )->cltr; } |
---|
254 | |
---|
255 | #if !defined(__CFA_NO_STATISTICS__) |
---|
256 | static inline void print_stats_at_exit( cluster & this ) { |
---|
257 | this.print_stats = true; |
---|
258 | } |
---|
259 | #endif |
---|
260 | |
---|
261 | // Local Variables: // |
---|
262 | // mode: c // |
---|
263 | // tab-width: 4 // |
---|
264 | // End: // |
---|