403ca3ffdab7a5b81d9261b8c3ecd77792644387
[akaros.git] / user / vmm / include / vmm / sched.h
1 /* Copyright (c) 2016 Google Inc.
2  * Barret Rhoden <brho@cs.berkeley.edu>
3  * See LICENSE for details.
4  *
5  * 2LS for virtual machines */
6
7 #pragma once
8
9 #include <parlib/uthread.h>
10 #include <sys/queue.h>
11
12 __BEGIN_DECLS
13
14 /* Three types of threads.  Guests are actual guest VMs.  Controllers are
15  * threads that are paired to guests and handles their exits.  Guests and
16  * controllers are 1:1 (via *buddy).  Task threads are for the VMM itself, such
17  * as a console thread. */
18
19 #define VMM_THREAD_GUEST                1
20 #define VMM_THREAD_CTLR                 2
21 #define VMM_THREAD_TASK                 3
22
23 #define VMM_THR_STACKSIZE               8192
24
25 struct guest_thread;
26 struct ctlr_thread;
27 struct task_thread;
28
29 struct guest_thread {
30         struct uthread                          uthread;
31         struct ctlr_thread                      *buddy;
32         unsigned int                            gpc_id;
33         uth_mutex_t                                     halt_mtx;
34         uth_cond_var_t                          halt_cv;
35 };
36
37 struct ctlr_thread {
38         struct uthread                          uthread;
39         struct guest_thread                     *buddy;
40         size_t                                          stacksize;
41         void                                            *stacktop;
42 };
43
44 struct task_thread {
45         struct uthread                          uthread;
46         void                                            (*func)(void *);
47         void                                            *arg;
48         size_t                                          stacksize;
49         void                                            *stacktop;
50 };
51
52 struct virtual_machine;                 /* in vmm/vmm.h */
53 struct vmm_thread {
54         union {
55                 struct guest_thread             guest;
56                 struct ctlr_thread              ctlr;
57                 struct task_thread              task;
58         };
59         int                                                     type;
60         TAILQ_ENTRY(vmm_thread)         tq_next;
61         struct virtual_machine          *vm;
62 };
63
64 TAILQ_HEAD(vmm_thread_tq, vmm_thread);
65
66 extern int vmm_sched_period_usec;
67
68 /* Initialize a VMM for a virtual machine, which the caller fills out, except
69  * for gths.  This will set **gths in the struct virtual machine.  Do not free()
70  * the array.
71  *
72  * Set the parlib control variables (e.g. parlib_wants_to_be_mcp) before calling
73  * this initializer.
74  *
75  * Returns 0 on success, -1 o/w. */
76 int vmm_init(struct virtual_machine *vm, int flags);
77 /* Starts a guest thread/core. */
78 void start_guest_thread(struct guest_thread *gth);
79 /* Start and run a task thread. */
80 struct task_thread *vmm_run_task(struct virtual_machine *vm,
81                                  void (*func)(void *), void *arg);
82
83 int vthread_attr_init(struct virtual_machine *vm, int vmmflags);
84 int vthread_attr_kernel_init(struct virtual_machine *vm, int vmmflags);
85 int vthread_create(struct virtual_machine *vm, int guest, void *rip, void *arg);
86
87 __END_DECLS