Fixes preempt/indir tracing in lock_test
[akaros.git] / user / parlib / include / uthread.h
index c88793e..50faa90 100644 (file)
@@ -7,6 +7,7 @@
 #define UTHREAD_DONT_MIGRATE           0x001 /* don't move to another vcore */
 #define UTHREAD_SAVED                          0x002 /* uthread's state is in utf */
 #define UTHREAD_FPSAVED                                0x004 /* uthread's FP state is in uth->as */
+#define UTHREAD_IS_THREAD0                     0x008 /* thread0: glibc's main() thread */
 
 /* Thread States */
 #define UT_RUNNING             1
@@ -24,15 +25,18 @@ struct uthread {
        struct ancillary_state as;
        void *tls_desc;
        int flags;
-       struct syscall *sysc;   /* syscall we're blocking on, if any */
        int state;
+       struct syscall *sysc;   /* syscall we're blocking on, if any */
+       struct syscall local_sysc;      /* for when we don't want to use the stack */
        void (*yield_func)(struct uthread*, void*);
        void *yield_arg;
+       int err_no;
+       char err_str[MAX_ERRSTR_LEN];
 };
 typedef struct uthread uthread_t;
 extern __thread struct uthread *current_uthread;
 
-/* 2L-Scheduler operations.  Can be 0.  Examples in pthread.c. */
+/* 2L-Scheduler operations.  Examples in pthread.c. */
 struct schedule_ops {
        /* Functions supporting thread ops */
        void (*sched_entry)(void);
@@ -40,6 +44,8 @@ struct schedule_ops {
        void (*thread_paused)(struct uthread *);
        void (*thread_blockon_sysc)(struct uthread *, void *);
        void (*thread_has_blocked)(struct uthread *, int);
+       void (*thread_refl_fault)(struct uthread *, unsigned int, unsigned int,
+                                                         unsigned long);
        /* Functions event handling wants */
        void (*preempt_pending)(void);
        void (*spawn_thread)(uintptr_t pc_start, void *data);   /* don't run yet */
@@ -68,6 +74,7 @@ void uthread_cleanup(struct uthread *uthread);
 void uthread_runnable(struct uthread *uthread);
 void uthread_yield(bool save_state, void (*yield_func)(struct uthread*, void*),
                    void *yield_arg);
+void uthread_sleep(unsigned int seconds);
 
 /* Utility functions */
 bool __check_preempt_pending(uint32_t vcoreid);        /* careful: check the code */
@@ -81,10 +88,7 @@ void deregister_evq(struct syscall *sysc);
 void highjack_current_uthread(struct uthread *uthread);
 void run_current_uthread(void);
 void run_uthread(struct uthread *uthread);
-
-/* Event handlers - exported globally so programs can wrap them */
-void handle_vc_preempt(struct event_msg *ev_msg, unsigned int ev_type);
-void handle_vc_indir(struct event_msg *ev_msg, unsigned int ev_type);
+static inline struct uthread **get_cur_uth_addr(uint32_t vcoreid);
 
 /* Asking for trouble with this API, when we just want stacktop (or whatever
  * the SP will be). */
@@ -111,4 +115,13 @@ static inline void init_uthread_ctx(struct uthread *uth, void (*entry)(void),
        val;                                                                       \
 })
 
+/* This works so long as we don't dlopen parlib (which we never do) */
+static inline struct uthread **get_cur_uth_addr(uint32_t vcoreid)
+{
+       uintptr_t vc_tls_desc = (uintptr_t)get_vcpd_tls_desc(vcoreid);
+       uintptr_t cur_uth_off = (uintptr_t)&current_uthread -
+                               (uintptr_t)get_tls_desc(vcore_id());
+       return (struct uthread**)(vc_tls_desc + cur_uth_off);
+}
+
 #endif /* _UTHREAD_H */