Removed event overflow handling
[akaros.git] / user / pthread / pthread.c
1 #include <ros/arch/trapframe.h>
2 #include <pthread.h>
3 #include <vcore.h>
4 #include <mcs.h>
5 #include <stdlib.h>
6 #include <string.h>
7 #include <assert.h>
8 #include <stdio.h>
9 #include <errno.h>
10 #include <parlib.h>
11 #include <ros/event.h>
12 #include <arch/atomic.h>
13 #include <arch/arch.h>
14 #include <sys/queue.h>
15 #include <sys/mman.h>
16 #include <assert.h>
17 #include <event.h>
18
19 struct pthread_queue ready_queue = TAILQ_HEAD_INITIALIZER(ready_queue);
20 struct pthread_queue active_queue = TAILQ_HEAD_INITIALIZER(active_queue);
21 mcs_lock_t queue_lock = MCS_LOCK_INIT;
22 pthread_once_t init_once = PTHREAD_ONCE_INIT;
23 int threads_ready = 0;
24 int threads_active = 0;
25
26 /* Array of per-vcore structs to manage waiting on syscalls and handling
27  * overflow.  Init'd in pth_init(). */
28 struct sysc_mgmt *sysc_mgmt = 0;
29
30 /* Helper / local functions */
31 static int get_next_pid(void);
32 static inline void spin_to_sleep(unsigned int spins, unsigned int *spun);
33
34 /* Pthread 2LS operations */
35 struct uthread *pth_init(void);
36 void pth_sched_entry(void);
37 struct uthread *pth_thread_create(void (*func)(void), void *udata);
38 void pth_thread_runnable(struct uthread *uthread);
39 void pth_thread_yield(struct uthread *uthread);
40 void pth_thread_exit(struct uthread *uthread);
41 void pth_preempt_pending(void);
42 void pth_spawn_thread(uintptr_t pc_start, void *data);
43 void pth_blockon_sysc(struct syscall *sysc);
44
45 /* Event Handlers */
46 static void pth_handle_syscall(struct event_msg *ev_msg, unsigned int ev_type);
47
48 struct schedule_ops pthread_sched_ops = {
49         pth_init,
50         pth_sched_entry,
51         pth_thread_create,
52         pth_thread_runnable,
53         pth_thread_yield,
54         pth_thread_exit,
55         pth_blockon_sysc,
56         0, /* pth_preempt_pending, */
57         0, /* pth_spawn_thread, */
58 };
59
60 /* Publish our sched_ops, overriding the weak defaults */
61 struct schedule_ops *sched_ops = &pthread_sched_ops;
62
63 /* Static helpers */
64 static void __pthread_free_stack(struct pthread_tcb *pt);
65 static int __pthread_allocate_stack(struct pthread_tcb *pt);
66
67 /* Do whatever init you want.  Return a uthread representing thread0 (int
68  * main()) */
69 struct uthread *pth_init(void)
70 {
71         struct mcs_lock_qnode local_qn = {0};
72         /* Tell the kernel where and how we want to receive events.  This is just an
73          * example of what to do to have a notification turned on.  We're turning on
74          * USER_IPIs, posting events to vcore 0's vcpd, and telling the kernel to
75          * send to vcore 0.  Note sys_self_notify will ignore the vcoreid pref.
76          * Also note that enable_kevent() is just an example, and you probably want
77          * to use parts of event.c to do what you want. */
78         enable_kevent(EV_USER_IPI, 0, EVENT_IPI);
79
80         /* Handle syscall events.  Using small ev_qs, with no internal ev_mbox. */
81         ev_handlers[EV_SYSCALL] = pth_handle_syscall;
82         /* Set up the per-vcore structs to track outstanding syscalls */
83         sysc_mgmt = malloc(sizeof(struct sysc_mgmt) * max_vcores());
84         assert(sysc_mgmt);
85         for (int i = 0; i < max_vcores(); i++) {
86                 /* Set up each of the per-vcore syscall event queues so that they point
87                  * to the VCPD/default vcore mailbox (for now)  Note you'll need the
88                  * vcore to be online to get the events (for now). */
89                 sysc_mgmt[i].ev_q.ev_mbox =  &__procdata.vcore_preempt_data[i].ev_mbox;
90                 sysc_mgmt[i].ev_q.ev_flags = EVENT_IPI;         /* totally up to you */
91                 sysc_mgmt[i].ev_q.ev_vcore = i;
92         }
93         /* Create a pthread_tcb for the main thread */
94         pthread_t t = (pthread_t)calloc(1, sizeof(struct pthread_tcb));
95         assert(t);
96         t->id = get_next_pid();
97         assert(t->id == 0);
98         /* Put the new pthread on the active queue */
99         mcs_lock_notifsafe(&queue_lock, &local_qn);
100         threads_active++;
101         TAILQ_INSERT_TAIL(&active_queue, t, next);
102         mcs_unlock_notifsafe(&queue_lock, &local_qn);
103         return (struct uthread*)t;
104 }
105
106 /* Called from vcore entry.  Options usually include restarting whoever was
107  * running there before or running a new thread.  Events are handled out of
108  * event.c (table of function pointers, stuff like that). */
109 void __attribute__((noreturn)) pth_sched_entry(void)
110 {
111         uint32_t vcoreid = vcore_id();
112         if (current_uthread) {
113                 run_current_uthread();
114                 assert(0);
115         }
116         /* no one currently running, so lets get someone from the ready queue */
117         struct pthread_tcb *new_thread = NULL;
118         struct mcs_lock_qnode local_qn = {0};
119         /* For now, let's spin and handle events til we get a thread to run.  This
120          * will help catch races, instead of only having one core ever run a thread
121          * (if there is just one, etc).  Also, we don't need the EVENT_IPIs for this
122          * to work (since we poll handle_events() */
123         while (!new_thread) {
124                 handle_events(vcoreid);
125                 mcs_lock_notifsafe(&queue_lock, &local_qn);
126                 new_thread = TAILQ_FIRST(&ready_queue);
127                 if (new_thread) {
128                         TAILQ_REMOVE(&ready_queue, new_thread, next);
129                         TAILQ_INSERT_TAIL(&active_queue, new_thread, next);
130                         threads_active++;
131                         threads_ready--;
132                 }
133                 mcs_unlock_notifsafe(&queue_lock, &local_qn);
134         }
135         /* Instead of yielding, you could spin, turn off the core, set an alarm,
136          * whatever.  You want some logic to decide this.  Uthread code wil have
137          * helpers for this (like how we provide run_uthread()) */
138         if (!new_thread) {
139                 /* Note, we currently don't get here (due to the while loop) */
140                 printd("[P] No threads, vcore %d is yielding\n", vcore_id());
141                 /* Not actually yielding - just spin for now, so we can get syscall
142                  * unblocking events */
143                 vcore_idle();
144                 //sys_yield(0);
145                 assert(0);
146         }
147         assert(((struct uthread*)new_thread)->state != UT_RUNNING);
148         run_uthread((struct uthread*)new_thread);
149         assert(0);
150 }
151
152 /* Could move this, along with start_routine and arg, into the 2LSs */
153 static void __pthread_run(void)
154 {
155         struct pthread_tcb *me = pthread_self();
156         pthread_exit(me->start_routine(me->arg));
157 }
158
159 /* Responible for creating the uthread and initializing its user trap frame */
160 struct uthread *pth_thread_create(void (*func)(void), void *udata)
161 {
162         struct pthread_tcb *pthread;
163         pthread_attr_t *attr = (pthread_attr_t*)udata;
164         pthread = (pthread_t)calloc(1, sizeof(struct pthread_tcb));
165         assert(pthread);
166         pthread->stacksize = PTHREAD_STACK_SIZE;        /* default */
167         pthread->id = get_next_pid();
168         pthread->detached = FALSE;                              /* default */
169         /* Respect the attributes */
170         if (attr) {
171                 if (attr->stacksize)                                    /* don't set a 0 stacksize */
172                         pthread->stacksize = attr->stacksize;
173                 if (attr->detachstate == PTHREAD_CREATE_DETACHED)
174                         pthread->detached = TRUE;
175         }
176         /* allocate a stack */
177         if (__pthread_allocate_stack(pthread))
178                 printf("We're fucked\n");
179         /* Set the u_tf to start up in __pthread_run, which will call the real
180          * start_routine and pass it the arg.  Note those aren't set until later in
181          * pthread_create(). */
182         init_user_tf(&pthread->uthread.utf, (uint32_t)__pthread_run, 
183                  (uint32_t)(pthread->stacktop));
184         return (struct uthread*)pthread;
185 }
186
187 void pth_thread_runnable(struct uthread *uthread)
188 {
189         struct pthread_tcb *pthread = (struct pthread_tcb*)uthread;
190         struct mcs_lock_qnode local_qn = {0};
191         /* Insert the newly created thread into the ready queue of threads.
192          * It will be removed from this queue later when vcore_entry() comes up */
193         mcs_lock_notifsafe(&queue_lock, &local_qn);
194         TAILQ_INSERT_TAIL(&ready_queue, pthread, next);
195         threads_ready++;
196         mcs_unlock_notifsafe(&queue_lock, &local_qn);
197         /* Smarter schedulers should look at the num_vcores() and how much work is
198          * going on to make a decision about how many vcores to request. */
199         vcore_request(threads_ready);
200 }
201
202 /* The calling thread is yielding.  Do what you need to do to restart (like put
203  * yourself on a runqueue), or do some accounting.  Eventually, this might be a
204  * little more generic than just yield. */
205 void pth_thread_yield(struct uthread *uthread)
206 {
207         struct pthread_tcb *pthread = (struct pthread_tcb*)uthread;
208         struct mcs_lock_qnode local_qn = {0};
209         /* Take from the active list, and put on the ready list (tail).  Don't do
210          * this until we are done completely with the thread, since it can be
211          * restarted somewhere else. */
212         mcs_lock_notifsafe(&queue_lock, &local_qn);
213         threads_active--;
214         TAILQ_REMOVE(&active_queue, pthread, next);
215         threads_ready++;
216         TAILQ_INSERT_TAIL(&ready_queue, pthread, next);
217         mcs_unlock_notifsafe(&queue_lock, &local_qn);
218 }
219
220 /* Thread is exiting, do your 2LS specific stuff.  You're in vcore context.
221  * Don't use the thread's TLS or stack or anything. */
222 void pth_thread_exit(struct uthread *uthread)
223 {
224         struct pthread_tcb *pthread = (struct pthread_tcb*)uthread;
225         struct mcs_lock_qnode local_qn = {0};
226         /* Remove from the active runqueue */
227         mcs_lock_notifsafe(&queue_lock, &local_qn);
228         threads_active--;
229         TAILQ_REMOVE(&active_queue, pthread, next);
230         mcs_unlock_notifsafe(&queue_lock, &local_qn);
231         /* Cleanup, mirroring pth_thread_create() */
232         __pthread_free_stack(pthread);
233         /* TODO: race on detach state */
234         if (pthread->detached)
235                 free(pthread);
236         else
237                 pthread->finished = 1;
238 }
239
240 void pth_preempt_pending(void)
241 {
242 }
243
244 void pth_spawn_thread(uintptr_t pc_start, void *data)
245 {
246 }
247
248 /* Restarts a uthread hanging off a syscall.  For the simple pthread case, we
249  * just make it runnable and let the main scheduler code handle it. */
250 static void restart_thread(struct syscall *sysc)
251 {
252         struct uthread *ut_restartee = (struct uthread*)sysc->u_data;
253         /* uthread stuff here: */
254         assert(ut_restartee);
255         assert(ut_restartee->state == UT_BLOCKED);
256         assert(ut_restartee->sysc == sysc);
257         ut_restartee->sysc = 0; /* so we don't 'reblock' on this later */
258         uthread_runnable(ut_restartee);
259 }
260
261 /* This handler is usually run in vcore context, though I can imagine it being
262  * called by a uthread in some other threading library. */
263 static void pth_handle_syscall(struct event_msg *ev_msg, unsigned int ev_type)
264 {
265         struct syscall *sysc;
266         assert(in_vcore_context());
267         /* It's a bug if we don't have a msg (we're handling a syscall bit-event) */
268         assert(ev_msg);
269         /* Get the sysc from the message and just restart it */
270         sysc = ev_msg->ev_arg3;
271         assert(sysc);
272         restart_thread(sysc);
273 }
274
275 /* This will be called from vcore context, after the current thread has yielded
276  * and is trying to block on sysc.  Need to put it somewhere were we can wake it
277  * up when the sysc is done.  For now, we'll have the kernel send us an event
278  * when the syscall is done. */
279 void pth_blockon_sysc(struct syscall *sysc)
280 {
281         int old_flags;
282         bool need_to_restart = FALSE;
283         uint32_t vcoreid = vcore_id();
284
285         assert(current_uthread->state == UT_BLOCKED);
286         /* rip from the active queue */
287         struct mcs_lock_qnode local_qn = {0};
288         struct pthread_tcb *pthread = (struct pthread_tcb*)current_uthread;
289         mcs_lock_notifsafe(&queue_lock, &local_qn);
290         threads_active--;
291         TAILQ_REMOVE(&active_queue, pthread, next);
292         mcs_unlock_notifsafe(&queue_lock, &local_qn);
293
294         /* Set things up so we can wake this thread up later */
295         sysc->u_data = current_uthread;
296         /* Register our vcore's syscall ev_q to hear about this syscall. */
297         if (!register_evq(sysc, &sysc_mgmt[vcoreid].ev_q)) {
298                 /* Lost the race with the call being done.  The kernel won't send the
299                  * event.  Just restart him. */
300                 restart_thread(sysc);
301         }
302         /* GIANT WARNING: do not touch the thread after this point. */
303 }
304
305 /* Pthread interface stuff and helpers */
306
307 int pthread_attr_init(pthread_attr_t *a)
308 {
309         a->stacksize = PTHREAD_STACK_SIZE;
310         a->detachstate = PTHREAD_CREATE_JOINABLE;
311         return 0;
312 }
313
314 int pthread_attr_destroy(pthread_attr_t *a)
315 {
316         return 0;
317 }
318
319 static void __pthread_free_stack(struct pthread_tcb *pt)
320 {
321         assert(!munmap(pt->stacktop - pt->stacksize, pt->stacksize));
322 }
323
324 static int __pthread_allocate_stack(struct pthread_tcb *pt)
325 {
326         assert(pt->stacksize);
327         void* stackbot = mmap(0, pt->stacksize,
328                               PROT_READ|PROT_WRITE|PROT_EXEC,
329                               MAP_POPULATE|MAP_ANONYMOUS, -1, 0);
330         if (stackbot == MAP_FAILED)
331                 return -1; // errno set by mmap
332         pt->stacktop = stackbot + pt->stacksize;
333         return 0;
334 }
335
336 // Warning, this will reuse numbers eventually
337 static int get_next_pid(void)
338 {
339         static uint32_t next_pid = 0;
340         return next_pid++;
341 }
342
343 int pthread_attr_setstacksize(pthread_attr_t *attr, size_t stacksize)
344 {
345         attr->stacksize = stacksize;
346         return 0;
347 }
348 int pthread_attr_getstacksize(const pthread_attr_t *attr, size_t *stacksize)
349 {
350         *stacksize = attr->stacksize;
351         return 0;
352 }
353
354 int pthread_create(pthread_t* thread, const pthread_attr_t* attr,
355                    void *(*start_routine)(void *), void* arg)
356 {
357         struct pthread_tcb *pthread =
358                (struct pthread_tcb*)uthread_create(__pthread_run, (void*)attr);
359         if (!pthread)
360                 return -1;
361         pthread->start_routine = start_routine;
362         pthread->arg = arg;
363         uthread_runnable((struct uthread*)pthread);
364         *thread = pthread;
365         return 0;
366 }
367
368 int pthread_join(pthread_t thread, void** retval)
369 {
370         /* Not sure if this is the right semantics.  There is a race if we deref
371          * thread and he is already freed (which would have happened if he was
372          * detached. */
373         if (thread->detached) {
374                 printf("[pthread] trying to join on a detached pthread");
375                 return -1;
376         }
377         while (!thread->finished)
378                 pthread_yield();
379         if (retval)
380                 *retval = thread->retval;
381         free(thread);
382         return 0;
383 }
384
385 int pthread_yield(void)
386 {
387         uthread_yield();
388         return 0;
389 }
390
391 int pthread_mutexattr_init(pthread_mutexattr_t* attr)
392 {
393   attr->type = PTHREAD_MUTEX_DEFAULT;
394   return 0;
395 }
396
397 int pthread_mutexattr_destroy(pthread_mutexattr_t* attr)
398 {
399   return 0;
400 }
401
402 int pthread_attr_setdetachstate(pthread_attr_t *__attr, int __detachstate)
403 {
404         __attr->detachstate = __detachstate;
405         return 0;
406 }
407
408 int pthread_mutexattr_gettype(const pthread_mutexattr_t* attr, int* type)
409 {
410   *type = attr ? attr->type : PTHREAD_MUTEX_DEFAULT;
411   return 0;
412 }
413
414 int pthread_mutexattr_settype(pthread_mutexattr_t* attr, int type)
415 {
416   if(type != PTHREAD_MUTEX_NORMAL)
417     return EINVAL;
418   attr->type = type;
419   return 0;
420 }
421
422 int pthread_mutex_init(pthread_mutex_t* m, const pthread_mutexattr_t* attr)
423 {
424   m->attr = attr;
425   atomic_init(&m->lock, 0);
426   return 0;
427 }
428
429 /* Set *spun to 0 when calling this the first time.  It will yield after 'spins'
430  * calls.  Use this for adaptive mutexes and such. */
431 static inline void spin_to_sleep(unsigned int spins, unsigned int *spun)
432 {
433         if ((*spun)++ == spins) {
434                 pthread_yield();
435                 *spun = 0;
436         }
437 }
438
439 int pthread_mutex_lock(pthread_mutex_t* m)
440 {
441         unsigned int spinner = 0;
442         while(pthread_mutex_trylock(m))
443                 while(*(volatile size_t*)&m->lock) {
444                         cpu_relax();
445                         spin_to_sleep(PTHREAD_MUTEX_SPINS, &spinner);
446                 }
447         return 0;
448 }
449
450 int pthread_mutex_trylock(pthread_mutex_t* m)
451 {
452   return atomic_swap(&m->lock, 1) == 0 ? 0 : EBUSY;
453 }
454
455 int pthread_mutex_unlock(pthread_mutex_t* m)
456 {
457   /* Need to prevent the compiler (and some arches) from reordering older
458    * stores */
459   wmb();
460   atomic_set(&m->lock, 0);
461   return 0;
462 }
463
464 int pthread_mutex_destroy(pthread_mutex_t* m)
465 {
466   return 0;
467 }
468
469 int pthread_cond_init(pthread_cond_t *c, const pthread_condattr_t *a)
470 {
471   c->attr = a;
472   memset(c->waiters,0,sizeof(c->waiters));
473   memset(c->in_use,0,sizeof(c->in_use));
474   c->next_waiter = 0;
475   return 0;
476 }
477
478 int pthread_cond_destroy(pthread_cond_t *c)
479 {
480   return 0;
481 }
482
483 int pthread_cond_broadcast(pthread_cond_t *c)
484 {
485   memset(c->waiters,0,sizeof(c->waiters));
486   return 0;
487 }
488
489 int pthread_cond_signal(pthread_cond_t *c)
490 {
491   int i;
492   for(i = 0; i < MAX_PTHREADS; i++)
493   {
494     if(c->waiters[i])
495     {
496       c->waiters[i] = 0;
497       break;
498     }
499   }
500   return 0;
501 }
502
503 int pthread_cond_wait(pthread_cond_t *c, pthread_mutex_t *m)
504 {
505   uint32_t old_waiter = c->next_waiter;
506   uint32_t my_waiter = c->next_waiter;
507   
508   //allocate a slot
509   while (atomic_swap_u32(& (c->in_use[my_waiter]), SLOT_IN_USE) == SLOT_IN_USE)
510   {
511     my_waiter = (my_waiter + 1) % MAX_PTHREADS;
512     assert (old_waiter != my_waiter);  // do not want to wrap around
513   }
514   c->waiters[my_waiter] = WAITER_WAITING;
515   c->next_waiter = (my_waiter+1) % MAX_PTHREADS;  // race on next_waiter but ok, because it is advisary
516
517   pthread_mutex_unlock(m);
518
519   volatile int* poll = &c->waiters[my_waiter];
520   while(*poll);
521   c->in_use[my_waiter] = SLOT_FREE;
522   pthread_mutex_lock(m);
523
524   return 0;
525 }
526
527 int pthread_condattr_init(pthread_condattr_t *a)
528 {
529   a = PTHREAD_PROCESS_PRIVATE;
530   return 0;
531 }
532
533 int pthread_condattr_destroy(pthread_condattr_t *a)
534 {
535   return 0;
536 }
537
538 int pthread_condattr_setpshared(pthread_condattr_t *a, int s)
539 {
540   a->pshared = s;
541   return 0;
542 }
543
544 int pthread_condattr_getpshared(pthread_condattr_t *a, int *s)
545 {
546   *s = a->pshared;
547   return 0;
548 }
549
550 pthread_t pthread_self()
551 {
552   return (struct pthread_tcb*)current_uthread;
553 }
554
555 int pthread_equal(pthread_t t1, pthread_t t2)
556 {
557   return t1 == t2;
558 }
559
560 /* This function cannot be migrated to a different vcore by the userspace
561  * scheduler.  Will need to sort that shit out. */
562 void pthread_exit(void *ret)
563 {
564         struct pthread_tcb *pthread = pthread_self();
565         pthread->retval = ret;
566         uthread_exit();
567 }
568
569 int pthread_once(pthread_once_t* once_control, void (*init_routine)(void))
570 {
571   if (atomic_swap_u32(once_control, 1) == 0)
572     init_routine();
573   return 0;
574 }
575
576 int pthread_barrier_init(pthread_barrier_t* b, const pthread_barrierattr_t* a, int count)
577 {
578   b->nprocs = b->count = count;
579   b->sense = 0;
580   pthread_mutex_init(&b->pmutex, 0);
581   return 0;
582 }
583
584 int pthread_barrier_wait(pthread_barrier_t* b)
585 {
586   unsigned int spinner = 0;
587   int ls = !b->sense;
588
589   pthread_mutex_lock(&b->pmutex);
590   int count = --b->count;
591   pthread_mutex_unlock(&b->pmutex);
592
593   if(count == 0)
594   {
595     printd("Thread %d is last to hit the barrier, resetting...\n", pthread_self()->id);
596     b->count = b->nprocs;
597         wmb();
598     b->sense = ls;
599     return PTHREAD_BARRIER_SERIAL_THREAD;
600   }
601   else
602   {
603     while(b->sense != ls) {
604       cpu_relax();
605       spin_to_sleep(PTHREAD_BARRIER_SPINS, &spinner);
606     }
607     return 0;
608   }
609 }
610
611 int pthread_barrier_destroy(pthread_barrier_t* b)
612 {
613   pthread_mutex_destroy(&b->pmutex);
614   return 0;
615 }
616
617 int pthread_detach(pthread_t thread)
618 {
619         /* TODO: race on this state.  Someone could be trying to join now */
620         thread->detached = TRUE;
621         return 0;
622 }