common/tcb.c


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

#include <apos/tcb.h>
#include <apos/utils.h>
#include <apos/sp_tree.h>

static struct sp_root t_root = (struct sp_root){0};

#define tcb_container(x) \
	container_of(x, struct tcb, sp_n)

void threads_insert(struct tcb *t)
{
	if(!sp_root(t_root)){
		sp_root(t_root) = &t->sp_n;
		return;
	}

	struct sp_node *n = sp_root(t_root), *p = NULL;
	enum sp_dir d = LEFT;

	while(n){
		struct tcb *tc = tcb_container(n);
		p = n;

		if(t->tid < tc->tid){
			n = sp_left(n);
			d = LEFT;
		}

		else {
			n = sp_right(n);
			d = RIGHT;
		}
	}

	sp_insert(&sp_root(t_root), p, &t->sp_n, d);
}

struct tcb *threads_find(id_t tid)
{
	struct sp_node *n = sp_root(t_root);

	while(n){
		struct tcb *t = tcb_container(n);

		if(t->tid == tid)
			return t;

		if(t->tid < tid)
			n = sp_left(n);
		else
			n = sp_right(n);
	}

	return 0;
}