Refactor to move prov stuff to coreprov.c (3/4)
[akaros.git] / kern / src / coreprov.c
1 /* Copyright (c) 2009, 2012, 2015 The Regents of the University of California
2  * Barret Rhoden <brho@cs.berkeley.edu>
3  * Valmon Leymarie <leymariv@berkeley.edu>
4  * Kevin Klues <klueska@cs.berkeley.edu>
5  * See LICENSE for details.
6  */
7
8 #include <env.h>
9 #include <schedule.h>
10
11 /* Provision a core to proc p. This code assumes that the scheduler that uses
12  * it holds a lock for the duration of the call. */
13 void __provision_core(struct proc *p, struct sched_pcore *spc)
14 {
15         struct sched_pcore_tailq *prov_list;
16         /* If the core is already prov to someone else, take it away.  (last write
17          * wins, some other layer or new func can handle permissions). */
18         if (spc->prov_proc) {
19                 /* the list the spc is on depends on whether it is alloced to the
20                  * prov_proc or not */
21                 prov_list = (spc->alloc_proc == spc->prov_proc ?
22                              &spc->prov_proc->ksched_data.crd.prov_alloc_me :
23                              &spc->prov_proc->ksched_data.crd.prov_not_alloc_me);
24                 TAILQ_REMOVE(prov_list, spc, prov_next);
25         }
26         /* Now prov it to p.  Again, the list it goes on depends on whether it is
27          * alloced to p or not.  Callers can also send in 0 to de-provision. */
28         if (p) {
29                 if (spc->alloc_proc == p) {
30                         TAILQ_INSERT_TAIL(&p->ksched_data.crd.prov_alloc_me, spc,
31                                           prov_next);
32                 } else {
33                         /* this is be the victim list, which can be sorted so that we pick
34                          * the right victim (sort by alloc_proc reverse priority, etc). */
35                         TAILQ_INSERT_TAIL(&p->ksched_data.crd.prov_not_alloc_me, spc,
36                                           prov_next);
37                 }
38         }
39         spc->prov_proc = p;
40 }
41
42 /* Unprovisions any pcores for the given list */
43 static void __unprov_pcore_list(struct sched_pcore_tailq *list_head)
44 {
45         struct sched_pcore *spc_i;
46         /* We can leave them connected within the tailq, since the scps don't have a
47          * default list (if they aren't on a proc's list, then we don't care about
48          * them), and since the INSERTs don't care what list you were on before
49          * (chummy with the implementation).  Pretty sure this is right.  If there's
50          * suspected list corruption, be safer here. */
51         TAILQ_FOREACH(spc_i, list_head, prov_next)
52                 spc_i->prov_proc = 0;
53         TAILQ_INIT(list_head);
54 }
55
56 /* Unprovision all cores from proc p. This code assumes that the scheduler
57  * that uses * it holds a lock for the duration of the call. */
58 void __unprovision_all_cores(struct proc *p)
59 {
60         __unprov_pcore_list(&p->ksched_data.crd.prov_alloc_me);
61         __unprov_pcore_list(&p->ksched_data.crd.prov_not_alloc_me);
62 }
63
64 /* Print a list of the cores currently provisioned to p. */
65 void print_proc_coreprov(struct proc *p)
66 {
67         struct sched_pcore *spc_i;
68
69         if (!p)
70                 return;
71         printk("Prov cores alloced to proc %d (%p)\n----------\n", p->pid, p);
72         TAILQ_FOREACH(spc_i, &p->ksched_data.crd.prov_alloc_me, prov_next)
73                 printk("Pcore %d\n", spc2pcoreid(spc_i));
74         printk("Prov cores not alloced to proc %d (%p)\n----------\n", p->pid, p);
75         TAILQ_FOREACH(spc_i, &p->ksched_data.crd.prov_not_alloc_me, prov_next)
76                 printk("Pcore %d (alloced to %d (%p))\n", spc2pcoreid(spc_i),
77                        spc_i->alloc_proc ? spc_i->alloc_proc->pid : 0,
78                        spc_i->alloc_proc);
79 }
80
81 /* Print the processes attached to each provisioned core. */
82 void print_coreprov_map(void)
83 {
84         struct sched_pcore *spc_i;
85         /* Doing this unlocked, which is dangerous, but won't deadlock */
86         printk("Which cores are provisioned to which procs:\n------------------\n");
87         for (int i = 0; i < num_cores; i++) {
88                 spc_i = pcoreid2spc(i);
89                 printk("Core %02d, prov: %d(%p) alloc: %d(%p)\n", i,
90                        spc_i->prov_proc ? spc_i->prov_proc->pid : 0, spc_i->prov_proc,
91                        spc_i->alloc_proc ? spc_i->alloc_proc->pid : 0,
92                        spc_i->alloc_proc);
93         }
94 }