Adding test case for multproducer async calls.
[akaros.git] / kern / src / arsc.c
1 /* See COPYRIGHT for copyright information. */
2
3 #ifdef __SHARC__
4 #pragma nosharc
5 #endif
6
7
8 #include <ros/common.h>
9 #include <ros/ring_syscall.h>
10 #include <arch/types.h>
11 #include <arch/arch.h>
12 #include <arch/mmu.h>
13 #include <error.h>
14
15 #include <syscall.h>
16 #include <kmalloc.h>
17 #include <pmap.h>
18 #include <stdio.h>
19 #include <hashtable.h>
20 #include <smp.h>
21 #include <arsc_server.h>
22 #include <kref.h>
23
24
25
26 struct proc_list arsc_proc_list = TAILQ_HEAD_INITIALIZER(arsc_proc_list);
27 spinlock_t arsc_proc_lock = SPINLOCK_INITIALIZER;
28
29 intreg_t inline syscall_async(struct proc *p, syscall_req_t *call)
30 {
31         return syscall(p, call->num, call->args[0], call->args[1],
32                        call->args[2], call->args[3], call->args[4], call->args[5]);
33 }
34
35 syscall_sring_t* sys_init_arsc(struct proc *p)
36 {
37         kref_get(&p->kref, 1);          /* we're storing an external ref here */
38         syscall_sring_t* sring;
39         void * va;
40         // TODO: need to pin this page in the future when swapping happens
41         va = do_mmap(p,MMAP_LOWEST_VA, SYSCALLRINGSIZE, PROT_READ|PROT_WRITE, MAP_ANON|MAP_POPULATE, NULL, 0);
42         pte_t *pte = pgdir_walk(p->env_pgdir, (void*)va, 0);
43         assert(pte);
44         sring = (syscall_sring_t*) (ppn2kva(PTE2PPN(*pte)));
45         /*make sure we are able to allocate the shared ring */
46         assert(sring != NULL);
47         p->procdata->syscallring = sring;
48         /* Initialize the generic syscall ring buffer */
49         SHARED_RING_INIT(sring);
50
51         BACK_RING_INIT(&p->syscallbackring,
52                        sring,
53                        SYSCALLRINGSIZE);
54
55         spin_lock_irqsave(&arsc_proc_lock);
56         TAILQ_INSERT_TAIL(&arsc_proc_list, p, proc_arsc_link);
57         spin_unlock_irqsave(&arsc_proc_lock);
58         return (syscall_sring_t*)va;
59 }
60
61 void arsc_server(struct trapframe *tf)
62 {
63         struct proc *p = NULL;
64         TAILQ_INIT(&arsc_proc_list);
65         while (1) {
66                 while (TAILQ_EMPTY(&arsc_proc_list))
67                         cpu_relax();
68                 
69                 TAILQ_FOREACH(p, &arsc_proc_list, proc_link) {
70                         /* Probably want to try to process a dying process's syscalls.  If
71                          * not, just move it to an else case */
72                         process_generic_syscalls (p, MAX_ASRC_BATCH); 
73                         if (p->state == PROC_DYING) {
74                                 TAILQ_REMOVE(&arsc_proc_list, p, proc_arsc_link);
75                                 proc_decref(p);
76                                 /* Need to break out, so the TAILQ_FOREACH doesn't flip out.
77                                  * It's not fair, but we're not dealing with that yet anyway */
78                                 break;
79                         }
80                 }
81         }
82 }
83
84 static intreg_t process_generic_syscalls(struct proc *p, size_t max)
85 {
86         size_t count = 0;
87         syscall_back_ring_t* sysbr = &p->syscallbackring;
88         struct per_cpu_info* coreinfo = &per_cpu_info[core_id()];
89         // looking at a process not initialized to perform arsc. 
90         if (sysbr == NULL) 
91                 return count;
92         
93         // max is the most we'll process.  max = 0 means do as many as possible
94         // TODO: check for initialization of the ring. 
95         while (RING_HAS_UNCONSUMED_REQUESTS(sysbr) && ((!max)||(count < max)) ) {
96                 if (!count) {
97                         // ASSUME: one queue per process
98                         // only switch cr3 for the very first request for this queue
99                         // need to switch to the right context, so we can handle the user pointer
100                         // that points to a data payload of the syscall
101                         lcr3(p->env_cr3);
102                 }
103                 count++;
104                 //printk("DEBUG PRE: sring->req_prod: %d, sring->rsp_prod: %d\n",
105                 //         sysbr->sring->req_prod, sysbr->sring->rsp_prod);
106                 // might want to think about 0-ing this out, if we aren't
107                 // going to explicitly fill in all fields
108                 syscall_rsp_t rsp;
109                 // this assumes we get our answer immediately for the syscall.
110                 syscall_req_t* req = RING_GET_REQUEST(sysbr, ++sysbr->req_cons);
111                 // print req
112                 printd("req no %d, req arg %c\n", req->num, *((char*)req->args[0]));
113                 
114                 /* TODO: when the remote syscall stuff can handle the new async
115                  * syscalls, they need to use a real sysc.  This might at least stop it
116                  * from crashing. */
117                 struct syscall sysc = {0};
118                 coreinfo->cur_sysc = &sysc;
119                 
120                 rsp.retval = syscall_async(p, req);
121                 rsp.syserr = sysc.err;
122                 rsp.cleanup = req->cleanup;
123                 rsp.data = req->data;
124                 // write response into the slot it came from
125                 memcpy(req, &rsp, sizeof(syscall_rsp_t));
126                 (sysbr->rsp_prod_pvt)++;
127                 req->status = RES_ready;
128                 RING_PUSH_RESPONSES(sysbr);
129
130                 //printk("DEBUG POST: sring->req_prod: %d, sring->rsp_prod: %d\n",
131                 //         sysbr->sring->req_prod, sysbr->sring->rsp_prod);
132         }
133         // load sane page tables (and don't rely on decref to do it for you).
134         lcr3(boot_cr3);
135         return (intreg_t)count;
136 }
137