| 1 | #include <assert.h> | 
|---|
| 2 | #include <stddef.h> | 
|---|
| 3 | struct $thread; | 
|---|
| 4 | struct __thread_desc_link { | 
|---|
| 5 | struct $thread * next; | 
|---|
| 6 | struct $thread * prev; | 
|---|
| 7 | volatile unsigned long long ts; | 
|---|
| 8 | int preferred; | 
|---|
| 9 | }; | 
|---|
| 10 | struct $thread { | 
|---|
| 11 | int f[64]; // FIXME: make "is volatile" consistent; given bug #TBD | 
|---|
| 12 | struct __thread_desc_link link; | 
|---|
| 13 | }; | 
|---|
| 14 |  | 
|---|
| 15 | // Intrusives lanes which are used by the relaxed ready queue | 
|---|
| 16 | struct __attribute__((aligned(128))) __intrusive_lane_t { | 
|---|
| 17 |  | 
|---|
| 18 | #if defined(USE_MPSC) | 
|---|
| 19 | mpsc_queue($thread) queue; | 
|---|
| 20 | __attribute__((aligned(128))) | 
|---|
| 21 | #else | 
|---|
| 22 | // anchor for the head and the tail of the queue | 
|---|
| 23 | __attribute__((aligned(128))) struct __sentinel_t { | 
|---|
| 24 | // Link lists fields | 
|---|
| 25 | // instrusive link field for threads | 
|---|
| 26 | // must be exactly as in $thread | 
|---|
| 27 | __thread_desc_link link; | 
|---|
| 28 | } before, after; | 
|---|
| 29 | #endif | 
|---|
| 30 |  | 
|---|
| 31 | // spin lock protecting the queue | 
|---|
| 32 | volatile bool lock; | 
|---|
| 33 |  | 
|---|
| 34 | // Optional statistic counters | 
|---|
| 35 | #if !defined(__CFA_NO_SCHED_STATS__) | 
|---|
| 36 | struct __attribute__((aligned(64))) { | 
|---|
| 37 | // difference between number of push and pops | 
|---|
| 38 | ssize_t diff; | 
|---|
| 39 |  | 
|---|
| 40 | // total number of pushes and pops | 
|---|
| 41 | size_t  push; | 
|---|
| 42 | size_t  pop ; | 
|---|
| 43 | } stat; | 
|---|
| 44 | #endif | 
|---|
| 45 | }; | 
|---|
| 46 |  | 
|---|
| 47 | void  ?{}(__intrusive_lane_t & this); | 
|---|
| 48 | void ^?{}(__intrusive_lane_t & this); | 
|---|
| 49 |  | 
|---|
| 50 | // Get the head pointer (one before the first element) from the anchor | 
|---|
| 51 | static inline $thread * head(const __intrusive_lane_t & this) { | 
|---|
| 52 | #if defined(USE_MPSC) | 
|---|
| 53 | return this.queue.head; | 
|---|
| 54 | #else | 
|---|
| 55 | $thread * rhead = ($thread *)( | 
|---|
| 56 | (uintptr_t)( &this.before ) - offsetof( $thread, link ) | 
|---|
| 57 | ); | 
|---|
| 58 | /* paranoid */ verify(rhead); | 
|---|
| 59 | return rhead; | 
|---|
| 60 | #endif | 
|---|
| 61 | } | 
|---|
| 62 |  | 
|---|
| 63 | // Get the tail pointer (one after the last element) from the anchor | 
|---|
| 64 | static inline $thread * tail(const __intrusive_lane_t & this) { | 
|---|
| 65 | #if defined(USE_MPSC) | 
|---|
| 66 | return this.queue.tail; | 
|---|
| 67 | #else | 
|---|
| 68 | $thread * rtail = ($thread *)( | 
|---|
| 69 | (uintptr_t)( &this.after ) - offsetof( $thread, link ) | 
|---|
| 70 | ); | 
|---|
| 71 | /* paranoid */ verify(rtail); | 
|---|
| 72 | return rtail; | 
|---|
| 73 | #endif | 
|---|
| 74 | } | 
|---|
| 75 |  | 
|---|
| 76 | // Ctor | 
|---|
| 77 | void ?{}( __intrusive_lane_t & this ) { | 
|---|
| 78 | this.lock = false; | 
|---|
| 79 |  | 
|---|
| 80 | #if !defined(USE_MPSC) | 
|---|
| 81 | this.before.link.prev = 0p; | 
|---|
| 82 | this.before.link.next = tail(this); | 
|---|
| 83 | this.before.link.ts   = 0; | 
|---|
| 84 |  | 
|---|
| 85 | this.after .link.prev = head(this); | 
|---|
| 86 | this.after .link.next = 0p; | 
|---|
| 87 | this.after .link.ts   = 0; | 
|---|
| 88 |  | 
|---|
| 89 | #if !defined(__CFA_NO_SCHED_STATS__) | 
|---|
| 90 | this.stat.diff = 0; | 
|---|
| 91 | this.stat.push = 0; | 
|---|
| 92 | this.stat.pop  = 0; | 
|---|
| 93 | #endif | 
|---|
| 94 |  | 
|---|
| 95 | // We add a boat-load of assertions here because the anchor code is very fragile | 
|---|
| 96 | /* paranoid */ verify(((uintptr_t)( head(this) ) + offsetof( $thread, link )) == (uintptr_t)(&this.before)); | 
|---|
| 97 | /* paranoid */ verify(((uintptr_t)( tail(this) ) + offsetof( $thread, link )) == (uintptr_t)(&this.after )); | 
|---|
| 98 | /* paranoid */ verify(head(this)->link.prev == 0p ); | 
|---|
| 99 | /* paranoid */ verify(head(this)->link.next == tail(this) ); | 
|---|
| 100 | /* paranoid */ verify(tail(this)->link.next == 0p ); | 
|---|
| 101 | /* paranoid */ verify(tail(this)->link.prev == head(this) ); | 
|---|
| 102 | /* paranoid */ verify(&head(this)->link.prev == &this.before.link.prev ); | 
|---|
| 103 | /* paranoid */ verify(&head(this)->link.next == &this.before.link.next ); | 
|---|
| 104 | /* paranoid */ verify(&tail(this)->link.prev == &this.after .link.prev ); | 
|---|
| 105 | /* paranoid */ verify(&tail(this)->link.next == &this.after .link.next ); | 
|---|
| 106 | /* paranoid */ verify(__alignof__(__intrusive_lane_t) == 128); | 
|---|
| 107 | /* paranoid */ verify(__alignof__(this) == 128); | 
|---|
| 108 | /* paranoid */ verifyf(((intptr_t)(&this) % 128) == 0, "Expected address to be aligned %p %% 128 == %zd", &this, ((intptr_t)(&this) % 128)); | 
|---|
| 109 | #endif | 
|---|
| 110 | } | 
|---|
| 111 |  | 
|---|
| 112 | // Dtor is trivial | 
|---|
| 113 | void ^?{}( __intrusive_lane_t & this ) { | 
|---|
| 114 | #if !defined(USE_MPSC) | 
|---|
| 115 | // Make sure the list is empty | 
|---|
| 116 | /* paranoid */ verify(head(this)->link.prev == 0p ); | 
|---|
| 117 | /* paranoid */ verify(head(this)->link.next == tail(this) ); | 
|---|
| 118 | /* paranoid */ verify(tail(this)->link.next == 0p ); | 
|---|
| 119 | /* paranoid */ verify(tail(this)->link.prev == head(this) ); | 
|---|
| 120 | #endif | 
|---|
| 121 | } | 
|---|
| 122 |  | 
|---|
| 123 | // Push a thread onto this lane | 
|---|
| 124 | // returns true of lane was empty before push, false otherwise | 
|---|
| 125 | bool push(__intrusive_lane_t & this, $thread * node) { | 
|---|
| 126 | #if defined(USE_MPSC) | 
|---|
| 127 | inline $thread * volatile & ?`next ( $thread * this )  __attribute__((const)) { | 
|---|
| 128 | return this->link.next; | 
|---|
| 129 | } | 
|---|
| 130 | push(this.queue, node); | 
|---|
| 131 | #else | 
|---|
| 132 | #if defined(__CFA_WITH_VERIFY__) | 
|---|
| 133 | /* paranoid */ verify(this.lock); | 
|---|
| 134 | /* paranoid */ verify(node->link.ts != 0); | 
|---|
| 135 | /* paranoid */ verify(node->link.next == 0p); | 
|---|
| 136 | /* paranoid */ verify(node->link.prev == 0p); | 
|---|
| 137 | /* paranoid */ verify(tail(this)->link.next == 0p); | 
|---|
| 138 | /* paranoid */ verify(head(this)->link.prev == 0p); | 
|---|
| 139 |  | 
|---|
| 140 | if(this.before.link.ts == 0l) { | 
|---|
| 141 | /* paranoid */ verify(tail(this)->link.prev == head(this)); | 
|---|
| 142 | /* paranoid */ verify(head(this)->link.next == tail(this)); | 
|---|
| 143 | } else { | 
|---|
| 144 | /* paranoid */ verify(tail(this)->link.prev != head(this)); | 
|---|
| 145 | /* paranoid */ verify(head(this)->link.next != tail(this)); | 
|---|
| 146 | } | 
|---|
| 147 | #endif | 
|---|
| 148 |  | 
|---|
| 149 | // Get the relevant nodes locally | 
|---|
| 150 | $thread * tail = tail(this); | 
|---|
| 151 | $thread * prev = tail->link.prev; | 
|---|
| 152 |  | 
|---|
| 153 | // Do the push | 
|---|
| 154 | node->link.next = tail; | 
|---|
| 155 | node->link.prev = prev; | 
|---|
| 156 | prev->link.next = node; | 
|---|
| 157 | tail->link.prev = node; | 
|---|
| 158 |  | 
|---|
| 159 | // Update stats | 
|---|
| 160 | #if !defined(__CFA_NO_SCHED_STATS__) | 
|---|
| 161 | this.stat.diff++; | 
|---|
| 162 | this.stat.push++; | 
|---|
| 163 | #endif | 
|---|
| 164 |  | 
|---|
| 165 | verify(node->link.next == tail(this)); | 
|---|
| 166 |  | 
|---|
| 167 | // Check if the queue used to be empty | 
|---|
| 168 | if(this.before.link.ts == 0l) { | 
|---|
| 169 | this.before.link.ts = node->link.ts; | 
|---|
| 170 | /* paranoid */ verify(node->link.prev == head(this)); | 
|---|
| 171 | return true; | 
|---|
| 172 | } | 
|---|
| 173 | return false; | 
|---|
| 174 | #endif | 
|---|
| 175 | } | 
|---|
| 176 |  | 
|---|
| 177 | // Pop a thread from this lane (must be non-empty) | 
|---|
| 178 | // returns popped | 
|---|
| 179 | // returns true of lane was empty before push, false otherwise | 
|---|
| 180 | $thread * pop(__intrusive_lane_t & this) { | 
|---|
| 181 | /* paranoid */ verify(this.lock); | 
|---|
| 182 | #if defined(USE_MPSC) | 
|---|
| 183 | inline $thread * volatile & ?`next ( $thread * this )  __attribute__((const)) { | 
|---|
| 184 | return this->link.next; | 
|---|
| 185 | } | 
|---|
| 186 | return pop(this.queue); | 
|---|
| 187 | #else | 
|---|
| 188 | /* paranoid */ verify(this.before.link.ts != 0ul); | 
|---|
| 189 |  | 
|---|
| 190 | // Get anchors locally | 
|---|
| 191 | $thread * head = head(this); | 
|---|
| 192 | $thread * tail = tail(this); | 
|---|
| 193 |  | 
|---|
| 194 | // Get the relevant nodes locally | 
|---|
| 195 | $thread * node = head->link.next; | 
|---|
| 196 | $thread * next = node->link.next; | 
|---|
| 197 |  | 
|---|
| 198 | /* paranoid */ verify(node != tail); | 
|---|
| 199 | /* paranoid */ verify(node); | 
|---|
| 200 |  | 
|---|
| 201 | // Do the pop | 
|---|
| 202 | head->link.next = next; | 
|---|
| 203 | next->link.prev = head; | 
|---|
| 204 | node->link.next = 0p; | 
|---|
| 205 | node->link.prev = 0p; | 
|---|
| 206 |  | 
|---|
| 207 | // Update head time stamp | 
|---|
| 208 | this.before.link.ts = next->link.ts; | 
|---|
| 209 |  | 
|---|
| 210 | // Update stats | 
|---|
| 211 | #ifndef __CFA_NO_SCHED_STATS__ | 
|---|
| 212 | this.stat.diff--; | 
|---|
| 213 | this.stat.pop ++; | 
|---|
| 214 | #endif | 
|---|
| 215 |  | 
|---|
| 216 | // Check if we emptied list and return accordingly | 
|---|
| 217 | /* paranoid */ verify(tail(this)->link.next == 0p); | 
|---|
| 218 | /* paranoid */ verify(head(this)->link.prev == 0p); | 
|---|
| 219 | if(next == tail) { | 
|---|
| 220 | /* paranoid */ verify(this.before.link.ts == 0); | 
|---|
| 221 | /* paranoid */ verify(tail(this)->link.prev == head(this)); | 
|---|
| 222 | /* paranoid */ verify(head(this)->link.next == tail(this)); | 
|---|
| 223 | return node; | 
|---|
| 224 | } | 
|---|
| 225 | else { | 
|---|
| 226 | /* paranoid */ verify(next->link.ts != 0); | 
|---|
| 227 | /* paranoid */ verify(tail(this)->link.prev != head(this)); | 
|---|
| 228 | /* paranoid */ verify(head(this)->link.next != tail(this)); | 
|---|
| 229 | /* paranoid */ verify(this.before.link.ts != 0); | 
|---|
| 230 | return node; | 
|---|
| 231 | } | 
|---|
| 232 | #endif | 
|---|
| 233 | } | 
|---|
| 234 |  | 
|---|
| 235 | // Check whether or not list is empty | 
|---|
| 236 | static inline bool is_empty(__intrusive_lane_t & this) { | 
|---|
| 237 | #if defined(USE_MPSC) | 
|---|
| 238 | return this.queue.head == 0p; | 
|---|
| 239 | #else | 
|---|
| 240 | // Cannot verify here since it may not be locked | 
|---|
| 241 | return this.before.link.ts == 0; | 
|---|
| 242 | #endif | 
|---|
| 243 | } | 
|---|
| 244 |  | 
|---|
| 245 | // Return the timestamp | 
|---|
| 246 | static inline unsigned long long ts(__intrusive_lane_t & this) { | 
|---|
| 247 | #if defined(USE_MPSC) | 
|---|
| 248 | $thread * tl = this.queue.head; | 
|---|
| 249 | if(!tl) return -1ull; | 
|---|
| 250 | return tl->link.ts; | 
|---|
| 251 | #else | 
|---|
| 252 | // Cannot verify here since it may not be locked | 
|---|
| 253 | return this.before.link.ts; | 
|---|
| 254 | #endif | 
|---|
| 255 | } | 
|---|