1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
#include <libtf/defines.h>
#include <libtf/list.h>
#include <libtf/vmach.h>
#include <libtf/io.h>
#include "uctx.h"
__thread struct tf_fiber *tf_current_fiber;
__thread struct tf_vcpu *tf_current_vcpu;
__thread struct tf_vmach *tf_current_vmach;
extern struct tf_poll_hooks tf_epoll_hooks;
struct tf_vmachine {
struct tf_vmach vmach;
struct tf_uctx startup_uctx;
void *machine_init_fiber;
};
static void tf_vcpu_main(void *fiber_data)
{
struct tf_vcpu *vcpu = fiber_data;
struct tf_vmach *vmach = vcpu->machine;
struct tf_fiber *self = container_of(fiber_data, struct tf_fiber, data);
struct tf_fiber *f;
tf_current_vmach = vmach;
tf_current_vcpu = vcpu;
while (vmach->num_user_fibers != 0) {
if (tf_list_empty(&vmach->run_q)) {
/* sleep */
continue;
}
f = tf_list_entry(tf_list_pop(&vmach->run_q),
struct tf_fiber, queue_node);
f->return_context = self->context;
tf_current_fiber = f;
tf_uctx_transfer(self->context, f->context);
tf_list_splice_tail(&f->wakeup_q, &vmach->run_q);
}
}
static void tf_vmach_main(void *fiber_data)
{
struct tf_fiber *self = container_of(fiber_data, struct tf_fiber, data);
struct tf_vcpu *vcpu = fiber_data;
struct tf_vmach *vmach = vcpu->machine;
tf_current_vmach = vmach;
tf_current_vcpu = vcpu;
tf_current_fiber = self;
/* Initialize IO subsystem */
vmach->poll_ops = &tf_epoll_hooks;
vmach->poll_fiber = vmach->poll_ops->create();
vmach->timeout_fiber = tf_timeout_fiber_create();
vmach->startup_fiber.timeout.manager = vmach->timeout_fiber;
/* Run the initial fiber */
tf_fiber_wakeup(&vmach->startup_fiber);
/* Use main thread as a regular vcpu */
vmach->num_user_fibers = 1;
tf_list_splice_tail(&self->wakeup_q, &vmach->run_q);
tf_vcpu_main(vcpu);
/* Kill all stuff */
/* Return to main fiber */
vmach->startup_fiber.return_context = NULL;
tf_current_fiber = NULL;
tf_current_vcpu = NULL;
tf_current_vmach = NULL;
tf_uctx_transfer(self->context, vmach->startup_fiber.context);
}
void tf_vmach_start(void)
{
struct tf_vmachine *vmach;
struct tf_vcpu *vcpu;
TF_BUG_ON(tf_current_vcpu != NULL);
/* Create a self-fiber so we can surrender control to vcpu */
vmach = calloc(1, sizeof(struct tf_vmachine));
vmach->vmach.startup_fiber = (struct tf_fiber) {
.ref_count = 1,
.queue_node = TF_LIST_INITIALIZER(vmach->vmach.startup_fiber.queue_node),
.wakeup_q = TF_LIST_HEAD_INITIALIZER(vmach->vmach.startup_fiber.wakeup_q),
.context = tf_uctx_create_self(&vmach->startup_uctx),
};
tf_list_init_head(&vmach->vmach.run_q);
vcpu = tf_fiber_create(tf_vmach_main, sizeof(struct tf_vcpu));
vmach->machine_init_fiber = vcpu;
vcpu->machine = &vmach->vmach;
/* Create manager fiber to initialize vcpu */
tf_uctx_transfer(vmach->vmach.startup_fiber.context,
container_of((void *) vcpu, struct tf_fiber, data)->context);
}
void tf_vmach_stop(void)
{
struct tf_fiber *self = tf_vmach_get_current_fiber();
struct tf_vmach *vmach = tf_vmach_get_current();
TF_BUG_ON(self != &vmach->startup_fiber);
/* Wait for the vmachine to stop */
tf_vmach_get_current()->num_user_fibers--;
while (self->return_context != NULL)
tf_fiber_schedule();
/* And clean up */
tf_uctx_destroy(vmach->startup_fiber.context);
free(vmach);
}
|
