change fillmeup syscall to use an int32 field to indicate the position within the...
[akaros.git] / kern / src / syscall.c
1 /* See COPYRIGHT for copyright information. */
2
3 #ifdef __SHARC__
4 #pragma nosharc
5 #endif
6
7 #include <ros/common.h>
8 #include <ros/notification.h>
9 #include <arch/types.h>
10 #include <arch/arch.h>
11 #include <arch/mmu.h>
12 #include <arch/console.h>
13 #include <ros/timer.h>
14 #include <error.h>
15
16 #include <elf.h>
17 #include <string.h>
18 #include <assert.h>
19 #include <process.h>
20 #include <schedule.h>
21 #include <pmap.h>
22 #include <mm.h>
23 #include <trap.h>
24 #include <syscall.h>
25 #include <kmalloc.h>
26 #include <stdio.h>
27 #include <resource.h>
28 #include <frontend.h>
29 #include <colored_caches.h>
30 #include <arch/bitmask.h>
31 #include <kfs.h> // eventually replace this with vfs.h
32
33
34 #ifdef __CONFIG_NETWORKING__
35 #include <arch/nic_common.h>
36 extern int (*send_frame)(const char *CT(len) data, size_t len);
37 extern unsigned char device_mac[6];
38 #endif
39
40 /* Tracing Globals */
41 int systrace_flags = 0;
42 struct systrace_record *systrace_buffer = 0;
43 unsigned int systrace_bufidx = 0;
44 size_t systrace_bufsize = 0;
45 struct proc *systrace_procs[MAX_NUM_TRACED] = {0};
46 spinlock_t systrace_lock = SPINLOCK_INITIALIZER;
47
48 /* Not enforcing the packing of systrace_procs yet, but don't rely on that */
49 static bool proc_is_traced(struct proc *p)
50 {
51         for (int i = 0; i < MAX_NUM_TRACED; i++)
52                 if (systrace_procs[i] == p)
53                         return true;
54         return false;
55 }
56
57 /************** Utility Syscalls **************/
58
59 static int sys_null(void)
60 {
61         return 0;
62 }
63
64 // Writes 'val' to 'num_writes' entries of the well-known array in the kernel
65 // address space.  It's just #defined to be some random 4MB chunk (which ought
66 // to be boot_alloced or something).  Meant to grab exclusive access to cache
67 // lines, to simulate doing something useful.
68 static int sys_cache_buster(struct proc *p, uint32_t num_writes,
69                              uint32_t num_pages, uint32_t flags)
70 { TRUSTEDBLOCK /* zra: this is not really part of the kernel */
71         #define BUSTER_ADDR             0xd0000000  // around 512 MB deep
72         #define MAX_WRITES              1048576*8
73         #define MAX_PAGES               32
74         #define INSERT_ADDR     (UINFO + 2*PGSIZE) // should be free for these tests
75         uint32_t* buster = (uint32_t*)BUSTER_ADDR;
76         static spinlock_t buster_lock = SPINLOCK_INITIALIZER;
77         uint64_t ticks = -1;
78         page_t* a_page[MAX_PAGES];
79
80         /* Strided Accesses or Not (adjust to step by cachelines) */
81         uint32_t stride = 1;
82         if (flags & BUSTER_STRIDED) {
83                 stride = 16;
84                 num_writes *= 16;
85         }
86
87         /* Shared Accesses or Not (adjust to use per-core regions)
88          * Careful, since this gives 8MB to each core, starting around 512MB.
89          * Also, doesn't separate memory for core 0 if it's an async call.
90          */
91         if (!(flags & BUSTER_SHARED))
92                 buster = (uint32_t*)(BUSTER_ADDR + core_id() * 0x00800000);
93
94         /* Start the timer, if we're asked to print this info*/
95         if (flags & BUSTER_PRINT_TICKS)
96                 ticks = start_timing();
97
98         /* Allocate num_pages (up to MAX_PAGES), to simulate doing some more
99          * realistic work.  Note we don't write to these pages, even if we pick
100          * unshared.  Mostly due to the inconvenience of having to match up the
101          * number of pages with the number of writes.  And it's unnecessary.
102          */
103         if (num_pages) {
104                 spin_lock(&buster_lock);
105                 for (int i = 0; i < MIN(num_pages, MAX_PAGES); i++) {
106                         upage_alloc(p, &a_page[i],1);
107                         page_insert(p->env_pgdir, a_page[i], (void*)INSERT_ADDR + PGSIZE*i,
108                                     PTE_USER_RW);
109                 }
110                 spin_unlock(&buster_lock);
111         }
112
113         if (flags & BUSTER_LOCKED)
114                 spin_lock(&buster_lock);
115         for (int i = 0; i < MIN(num_writes, MAX_WRITES); i=i+stride)
116                 buster[i] = 0xdeadbeef;
117         if (flags & BUSTER_LOCKED)
118                 spin_unlock(&buster_lock);
119
120         if (num_pages) {
121                 spin_lock(&buster_lock);
122                 for (int i = 0; i < MIN(num_pages, MAX_PAGES); i++) {
123                         page_remove(p->env_pgdir, (void*)(INSERT_ADDR + PGSIZE * i));
124                         page_decref(a_page[i]);
125                 }
126                 spin_unlock(&buster_lock);
127         }
128
129         /* Print info */
130         if (flags & BUSTER_PRINT_TICKS) {
131                 ticks = stop_timing(ticks);
132                 printk("%llu,", ticks);
133         }
134         return 0;
135 }
136
137 static int sys_cache_invalidate(void)
138 {
139         #ifdef __i386__
140                 wbinvd();
141         #endif
142         return 0;
143 }
144
145 /* sys_reboot(): called directly from dispatch table. */
146
147 // Print a string to the system console.
148 // The string is exactly 'len' characters long.
149 // Destroys the environment on memory errors.
150 static ssize_t sys_cputs(env_t* e, const char *DANGEROUS s, size_t len)
151 {
152         // Check that the user has permission to read memory [s, s+len).
153         // Destroy the environment if not.
154         char *COUNT(len) _s = user_mem_assert(e, s, len, PTE_USER_RO);
155
156         // Print the string supplied by the user.
157         printk("%.*s", len, _s);
158         return (ssize_t)len;
159 }
160
161 // Read a character from the system console.
162 // Returns the character.
163 static uint16_t sys_cgetc(env_t* e)
164 {
165         uint16_t c;
166
167         // The cons_getc() primitive doesn't wait for a character,
168         // but the sys_cgetc() system call does.
169         while ((c = cons_getc()) == 0)
170                 cpu_relax();
171
172         return c;
173 }
174
175 /* Returns the id of the cpu this syscall is executed on. */
176 static uint32_t sys_getcpuid(void)
177 {
178         return core_id();
179 }
180
181 // TODO: Temporary hack until thread-local storage is implemented on i386 and
182 // this is removed from the user interface
183 static size_t sys_getvcoreid(struct proc *p)
184 {
185         return proc_get_vcoreid(p, core_id());
186 }
187
188 /************** Process management syscalls **************/
189
190 /* Returns the calling process's pid */
191 static pid_t sys_getpid(struct proc *p)
192 {
193         return p->pid;
194 }
195
196 /*
197  * Creates a process found at the user string 'path'.  Currently uses KFS.
198  * Not runnable by default, so it needs it's status to be changed so that the
199  * next call to schedule() will try to run it.
200  * TODO: once we have a decent VFS, consider splitting this up
201  * and once there's an mmap, can have most of this in process.c
202  */
203 static int sys_proc_create(struct proc *p, const char *DANGEROUS path)
204 {
205         int pid = 0;
206         char tpath[MAX_PATH_LEN];
207         /*
208          * There's a bunch of issues with reading in the path, which we'll
209          * need to sort properly in the VFS.  Main concerns are TOCTOU (copy-in),
210          * whether or not it's a big deal that the pointer could be into kernel
211          * space, and resolving both of these without knowing the length of the
212          * string. (TODO)
213          * Change this so that all syscalls with a pointer take a length.
214          *
215          * zra: I've added this user_mem_strlcpy, which I think eliminates the
216      * the TOCTOU issue. Adding a length arg to this call would allow a more
217          * efficient implementation, though, since only one call to user_mem_check
218          * would be required.
219          */
220         int ret = user_mem_strlcpy(p,tpath, path, MAX_PATH_LEN, PTE_USER_RO);
221         int kfs_inode = kfs_lookup_path(tpath);
222         if (kfs_inode < 0)
223                 return -EINVAL;
224         struct proc *new_p = kfs_proc_create(kfs_inode);
225         pid = new_p->pid;
226         proc_decref(new_p, 1); // let go of the reference created in proc_create()
227         return pid;
228 }
229
230 /* Makes process PID runnable.  Consider moving the functionality to process.c */
231 static error_t sys_proc_run(struct proc *p, unsigned pid)
232 {
233         struct proc *target = pid2proc(pid);
234         error_t retval = 0;
235
236         if (!target)
237                 return -EBADPROC;
238         // note we can get interrupted here. it's not bad.
239         spin_lock(&p->proc_lock);
240         // make sure we have access and it's in the right state to be activated
241         if (!proc_controls(p, target)) {
242                 proc_decref(target, 1);
243                 retval = -EPERM;
244         } else if (target->state != PROC_CREATED) {
245                 proc_decref(target, 1);
246                 retval = -EINVAL;
247         } else {
248                 __proc_set_state(target, PROC_RUNNABLE_S);
249                 schedule_proc(target);
250         }
251         spin_unlock(&p->proc_lock);
252         proc_decref(target, 1);
253         return retval;
254 }
255
256 /* Destroy proc pid.  If this is called by the dying process, it will never
257  * return.  o/w it will return 0 on success, or an error.  Errors include:
258  * - EBADPROC: if there is no such process with pid
259  * - EPERM: if caller does not control pid */
260 static error_t sys_proc_destroy(struct proc *p, pid_t pid, int exitcode)
261 {
262         error_t r;
263         struct proc *p_to_die = pid2proc(pid);
264
265         if (!p_to_die) {
266                 set_errno(current_tf, ESRCH);
267                 return -1;
268         }
269 #ifndef __CONFIG_EXPER_TRADPROC__
270         if (!proc_controls(p, p_to_die)) {
271                 proc_decref(p_to_die, 1);
272                 set_errno(current_tf, EPERM);
273                 return -1;
274         }
275         if (p_to_die == p) {
276 #else
277         if ((p_to_die == p) || (p_to_die == p->true_proc)) {
278 #endif /* __CONFIG_EXPER_TRADPROC__ */
279                 // syscall code and pid2proc both have edible references, only need 1.
280                 p->exitcode = exitcode;
281                 proc_decref(p, 1);
282                 printd("[PID %d] proc exiting gracefully (code %d)\n", p->pid,exitcode);
283         } else {
284                 printd("[%d] destroying proc %d\n", p->pid, p_to_die->pid);
285         }
286         proc_destroy(p_to_die);
287         proc_decref(p_to_die, 1);
288         return ESUCCESS;
289 }
290
291 static int sys_proc_yield(struct proc *p, bool being_nice)
292 {
293         proc_yield(p, being_nice);
294         return 0;
295 }
296
297 static ssize_t sys_run_binary(env_t* e, void *DANGEROUS binary_buf, size_t len,
298                               procinfo_t*DANGEROUS procinfo, size_t num_colors)
299 {
300         env_t* env = proc_create(NULL,0);
301         assert(env != NULL);
302
303         // let me know if you use this.  we need to sort process creation better.
304         printk("sys_run_binary() is deprecated.  Use at your own risk.");
305         if(memcpy_from_user(e,e->procinfo,procinfo,sizeof(*procinfo)))
306                 return -1;
307         proc_init_procinfo(e);
308
309         env_load_icode(env,e,binary_buf,len);
310         __proc_set_state(env, PROC_RUNNABLE_S);
311         schedule_proc(env);
312         if(num_colors > 0) {
313                 env->cache_colors_map = cache_colors_map_alloc();
314                 for(int i=0; i<num_colors; i++)
315                         cache_color_alloc(llc_cache, env->cache_colors_map);
316         }
317         proc_decref(env, 1);
318         proc_yield(e, 0);
319         return 0;
320 }
321
322 static ssize_t sys_fork(env_t* e)
323 {
324         // TODO: right now we only support fork for single-core processes
325         if(e->state != PROC_RUNNING_S)
326         {
327                 set_errno(current_tf,EINVAL);
328                 return -1;
329         }
330
331         env_t* env = proc_create(NULL,0);
332         assert(env != NULL);
333
334         env->heap_top = e->heap_top;
335         env->ppid = e->pid;
336         env->env_tf = *current_tf;
337
338         env->cache_colors_map = cache_colors_map_alloc();
339         for(int i=0; i < llc_cache->num_colors; i++)
340                 if(GET_BITMASK_BIT(e->cache_colors_map,i))
341                         cache_color_alloc(llc_cache, env->cache_colors_map);
342
343         int copy_page(env_t* e, pte_t* pte, void* va, void* arg)
344         {
345                 env_t* env = (env_t*)arg;
346
347                 if(PAGE_PRESENT(*pte))
348                 {
349                         page_t* pp;
350                         if(upage_alloc(env,&pp,0))
351                                 return -1;
352                         if(page_insert(env->env_pgdir,pp,va,*pte & PTE_PERM))
353                         {
354                                 page_decref(pp);
355                                 return -1;
356                         }
357
358                         pagecopy(page2kva(pp),ppn2kva(PTE2PPN(*pte)));
359                 }
360                 else // PAGE_PAGED_OUT(*pte)
361                 {
362                         pte_t* newpte = pgdir_walk(env->env_pgdir,va,1);
363                         if(!newpte)
364                                 return -1;
365
366                         struct file* file = PTE2PFAULT_INFO(*pte)->file;
367                         pfault_info_t* newpfi = pfault_info_alloc(file);
368                         if(!newpfi)
369                                 return -1;
370
371                         *newpfi = *PTE2PFAULT_INFO(*pte);
372                         *newpte = PFAULT_INFO2PTE(newpfi);
373                 }
374
375                 return 0;
376         }
377
378         // TODO: (PC) this won't work.  Needs revisiting.
379         // copy procdata and procinfo
380         memcpy(env->procdata,e->procdata,sizeof(struct procdata));
381         memcpy(env->procinfo,e->procinfo,sizeof(struct procinfo));
382         env->procinfo->pid = env->pid;
383         env->procinfo->ppid = env->ppid;
384
385         // copy all memory below procdata
386         if(env_user_mem_walk(e,0,UDATA,&copy_page,env))
387         {
388                 proc_decref(env,2);
389                 set_errno(current_tf,ENOMEM);
390                 return -1;
391         }
392
393         __proc_set_state(env, PROC_RUNNABLE_S);
394         schedule_proc(env);
395
396         // don't decref the new process.
397         // that will happen when the parent waits for it.
398
399         printd("[PID %d] fork PID %d\n",e->pid,env->pid);
400
401         return env->pid;
402 }
403
404 intreg_t sys_exec(struct proc* p, int fd, procinfo_t* pi)
405 {
406         if(p->state != PROC_RUNNING_S)
407                 return -1;
408
409         int ret = -1;
410         struct file* f = file_open_from_fd(p,fd);
411         if(f == NULL) {
412                 set_errno(current_tf, EBADF);
413                 goto out;
414         }
415
416         // Set the argument stuff needed by glibc
417         if(memcpy_from_user(p,p->procinfo->argp,pi->argp,sizeof(pi->argp))) {
418                 proc_destroy(p);
419                 goto out;
420         }
421         if(memcpy_from_user(p,p->procinfo->argbuf,pi->argbuf,sizeof(pi->argbuf))) {
422                 proc_destroy(p);
423                 goto out;
424         }
425
426         // TODO: don't do this either (PC)
427         memset(p->procdata, 0, sizeof(procdata_t));
428
429         env_user_mem_free(p,0,USTACKTOP);
430
431         if(load_elf(p,f))
432         {
433                 proc_destroy(p);
434                 goto out;
435         }
436         file_decref(f);
437         *current_tf = p->env_tf;
438         ret = 0;
439
440         printd("[PID %d] exec fd %d\n",p->pid,fd);
441
442 out:
443         return ret;
444 }
445
446 static ssize_t sys_trywait(env_t* e, pid_t pid, int* status)
447 {
448         struct proc* p = pid2proc(pid);
449
450         // TODO: this syscall is racy, so we only support for single-core procs
451         if(e->state != PROC_RUNNING_S)
452                 return -1;
453
454         // TODO: need to use errno properly.  sadly, ROS error codes conflict..
455
456         if(p)
457         {
458                 ssize_t ret;
459
460                 if(current->pid == p->ppid)
461                 {
462                         if(p->state == PROC_DYING)
463                         {
464                                 memcpy_to_user(e,status,&p->exitcode,sizeof(int));
465                                 printd("[PID %d] waited for PID %d (code %d)\n",
466                                        e->pid,p->pid,p->exitcode);
467                                 ret = 0;
468                         }
469                         else // not dead yet
470                         {
471                                 set_errno(current_tf,0);
472                                 ret = -1;
473                         }
474                 }
475                 else // not a child of the calling process
476                 {
477                         set_errno(current_tf,1);
478                         ret = -1;
479                 }
480
481                 // if the wait succeeded, decref twice
482                 proc_decref(p,1 + (ret == 0));
483                 return ret;
484         }
485
486         set_errno(current_tf,1);
487         return -1;
488 }
489
490 /************** Memory Management Syscalls **************/
491
492 static void *sys_mmap(struct proc* p, uintreg_t a1, uintreg_t a2, uintreg_t a3,
493                       uintreg_t* a456)
494 {
495         uintreg_t _a456[3];
496         if(memcpy_from_user(p,_a456,a456,3*sizeof(uintreg_t)))
497                 sys_proc_destroy(p,p->pid,-1);
498         return mmap(p,a1,a2,a3,_a456[0],_a456[1],_a456[2]);
499 }
500
501 static intreg_t sys_mprotect(struct proc* p, void* addr, size_t len, int prot)
502 {
503         return mprotect(p, addr, len, prot);
504 }
505
506 static intreg_t sys_munmap(struct proc* p, void* addr, size_t len)
507 {
508         return munmap(p, addr, len);
509 }
510
511 static void* sys_brk(struct proc *p, void* addr) {
512         ssize_t range;
513
514 #ifdef __CONFIG_EXPER_TRADPROC__
515         printk("[kernel] don't use brk, unsupported.\n");
516         return (void*)-1;
517 #endif /* __CONFIG_EXPER_TRADPROC__ */
518
519         spin_lock(&p->proc_lock);
520
521         if((addr < p->procinfo->heap_bottom) || (addr >= (void*)BRK_END))
522                 goto out;
523
524         uintptr_t real_heap_top = ROUNDUP((uintptr_t)p->heap_top,PGSIZE);
525         uintptr_t real_new_heap_top = ROUNDUP((uintptr_t)addr,PGSIZE);
526         range = real_new_heap_top - real_heap_top;
527
528         if (range > 0) {
529                 if(__do_mmap(p, real_heap_top, range, PROT_READ | PROT_WRITE,
530                              MAP_FIXED | MAP_ANONYMOUS, NULL, 0) == MAP_FAILED)
531                         goto out;
532         }
533         else if (range < 0) {
534                 if(__munmap(p, (void*)real_new_heap_top, -range))
535                         goto out;
536         }
537         p->heap_top = addr;
538
539 out:
540         spin_unlock(&p->proc_lock);
541         return p->heap_top;
542 }
543
544 static ssize_t sys_shared_page_alloc(env_t* p1,
545                                      void**DANGEROUS _addr, pid_t p2_id,
546                                      int p1_flags, int p2_flags
547                                     )
548 {
549         //if (!VALID_USER_PERMS(p1_flags)) return -EPERM;
550         //if (!VALID_USER_PERMS(p2_flags)) return -EPERM;
551
552         void * COUNT(1) * COUNT(1) addr = user_mem_assert(p1, _addr, sizeof(void *),
553                                                       PTE_USER_RW);
554         struct proc *p2 = pid2proc(p2_id);
555         if (!p2)
556                 return -EBADPROC;
557
558         page_t* page;
559         error_t e = upage_alloc(p1, &page,1);
560         if (e < 0) {
561                 proc_decref(p2, 1);
562                 return e;
563         }
564
565         void* p2_addr = page_insert_in_range(p2->env_pgdir, page,
566                         (void*SNT)UTEXT, (void*SNT)UTOP, p2_flags);
567         if (p2_addr == NULL) {
568                 page_free(page);
569                 proc_decref(p2, 1);
570                 return -EFAIL;
571         }
572
573         void* p1_addr = page_insert_in_range(p1->env_pgdir, page,
574                         (void*SNT)UTEXT, (void*SNT)UTOP, p1_flags);
575         if(p1_addr == NULL) {
576                 page_remove(p2->env_pgdir, p2_addr);
577                 page_free(page);
578                 proc_decref(p2, 1);
579                 return -EFAIL;
580         }
581         *addr = p1_addr;
582         proc_decref(p2, 1);
583         return ESUCCESS;
584 }
585
586 static int sys_shared_page_free(env_t* p1, void*DANGEROUS addr, pid_t p2)
587 {
588         return -1;
589 }
590
591
592 /* sys_resource_req(): called directly from dispatch table. */
593
594 /* Will notify the target on the given vcore, if the caller controls the target.
595  * Will honor the target's wanted/vcoreid.  u_ne can be NULL. */
596 static int sys_notify(struct proc *p, int target_pid, unsigned int notif,
597                       struct notif_event *u_ne)
598 {
599         struct notif_event local_ne;
600         struct proc *target = pid2proc(target_pid);
601
602         if (!target) {
603                 set_errno(current_tf, EBADPROC);
604                 return -1;
605         }
606         if (!proc_controls(p, target)) {
607                 proc_decref(target, 1);
608                 set_errno(current_tf, EPERM);
609                 return -1;
610         }
611         /* if the user provided a notif_event, copy it in and use that */
612         if (u_ne) {
613                 if (memcpy_from_user(p, &local_ne, u_ne, sizeof(struct notif_event))) {
614                         proc_decref(target, 1);
615                         set_errno(current_tf, EINVAL);
616                         return -1;
617                 }
618                 proc_notify(target, local_ne.ne_type, &local_ne);
619         } else {
620                 proc_notify(target, notif, 0);
621         }
622         proc_decref(target, 1);
623         return 0;
624 }
625
626 /* Will notify the calling process on the given vcore, independently of WANTED
627  * or advertised vcoreid.  If you change the parameters, change pop_ros_tf() */
628 static int sys_self_notify(struct proc *p, uint32_t vcoreid, unsigned int notif,
629                            struct notif_event *u_ne)
630 {
631         struct notif_event local_ne;
632
633         printd("[kernel] received self notify for vcoreid %d, notif %d, ne %08p\n",
634                vcoreid, notif, u_ne);
635         /* if the user provided a notif_event, copy it in and use that */
636         if (u_ne) {
637                 if (memcpy_from_user(p, &local_ne, u_ne, sizeof(struct notif_event))) {
638                         set_errno(current_tf, EINVAL);
639                         return -1;
640                 }
641                 do_notify(p, vcoreid, local_ne.ne_type, &local_ne);
642         } else {
643                 do_notify(p, vcoreid, notif, 0);
644         }
645         return 0;
646 }
647
648 /* This will set a local timer for usec, then shut down the core */
649 static int sys_halt_core(struct proc *p, unsigned int usec)
650 {
651         /* TODO: ought to check and see if a timer was already active, etc, esp so
652          * userspace can't turn off timers.  also note we will also call whatever
653          * timer_interrupt() will do, though all we care about is just
654          * self_ipi/interrupting. */
655         set_core_timer(usec);
656         cpu_halt();
657
658         return 0;
659 }
660
661 /************** Platform Specific Syscalls **************/
662
663 //Read a buffer over the serial port
664 static ssize_t sys_serial_read(env_t* e, char *DANGEROUS _buf, size_t len)
665 {
666         if (len == 0)
667                 return 0;
668
669         #ifdef __CONFIG_SERIAL_IO__
670             char *COUNT(len) buf = user_mem_assert(e, _buf, len, PTE_USER_RO);
671                 size_t bytes_read = 0;
672                 int c;
673                 while((c = serial_read_byte()) != -1) {
674                         buf[bytes_read++] = (uint8_t)c;
675                         if(bytes_read == len) break;
676                 }
677                 return (ssize_t)bytes_read;
678         #else
679                 return -EINVAL;
680         #endif
681 }
682
683 //Write a buffer over the serial port
684 static ssize_t sys_serial_write(env_t* e, const char *DANGEROUS buf, size_t len)
685 {
686         if (len == 0)
687                 return 0;
688         #ifdef __CONFIG_SERIAL_IO__
689                 char *COUNT(len) _buf = user_mem_assert(e, buf, len, PTE_USER_RO);
690                 for(int i =0; i<len; i++)
691                         serial_send_byte(buf[i]);
692                 return (ssize_t)len;
693         #else
694                 return -EINVAL;
695         #endif
696 }
697
698 #ifdef __CONFIG_NETWORKING__
699 // This is not a syscall we want. Its hacky. Here just for syscall stuff until get a stack.
700 static ssize_t sys_eth_read(env_t* e, char *DANGEROUS buf)
701 {
702         if (eth_up) {
703
704                 uint32_t len;
705                 char *ptr;
706
707                 spin_lock(&packet_buffers_lock);
708
709                 if (num_packet_buffers == 0) {
710                         spin_unlock(&packet_buffers_lock);
711                         return 0;
712                 }
713
714                 ptr = packet_buffers[packet_buffers_head];
715                 len = packet_buffers_sizes[packet_buffers_head];
716
717                 num_packet_buffers--;
718                 packet_buffers_head = (packet_buffers_head + 1) % MAX_PACKET_BUFFERS;
719
720                 spin_unlock(&packet_buffers_lock);
721
722                 char* _buf = user_mem_assert(e, buf, len, PTE_U);
723
724                 memcpy(_buf, ptr, len);
725
726                 kfree(ptr);
727
728                 return len;
729         }
730         else
731                 return -EINVAL;
732 }
733
734 // This is not a syscall we want. Its hacky. Here just for syscall stuff until get a stack.
735 static ssize_t sys_eth_write(env_t* e, const char *DANGEROUS buf, size_t len)
736 {
737         if (eth_up) {
738
739                 if (len == 0)
740                         return 0;
741
742                 // HACK TO BYPASS HACK
743                 int just_sent = send_frame(buf, len);
744
745                 if (just_sent < 0) {
746                         printk("Packet send fail\n");
747                         return 0;
748                 }
749
750                 return just_sent;
751
752                 // END OF RECURSIVE HACK
753 /*
754                 char *COUNT(len) _buf = user_mem_assert(e, buf, len, PTE_U);
755                 int total_sent = 0;
756                 int just_sent = 0;
757                 int cur_packet_len = 0;
758                 while (total_sent != len) {
759                         cur_packet_len = ((len - total_sent) > MTU) ? MTU : (len - total_sent);
760                         char dest_mac[6] = APPSERVER_MAC_ADDRESS;
761                         char* wrap_buffer = eth_wrap(_buf + total_sent, cur_packet_len, device_mac, dest_mac, APPSERVER_PORT);
762                         just_sent = send_frame(wrap_buffer, cur_packet_len + sizeof(struct ETH_Header));
763
764                         if (just_sent < 0)
765                                 return 0; // This should be an error code of its own
766
767                         if (wrap_buffer)
768                                 kfree(wrap_buffer);
769
770                         total_sent += cur_packet_len;
771                 }
772
773                 return (ssize_t)len;
774 */
775         }
776         else
777                 return -EINVAL;
778 }
779
780 static ssize_t sys_eth_get_mac_addr(env_t* e, char *DANGEROUS buf) 
781 {
782         if (eth_up) {
783                 for (int i = 0; i < 6; i++)
784                         buf[i] = device_mac[i];
785                 return 0;
786         }
787         else
788                 return -EINVAL;
789 }
790
791 static int sys_eth_recv_check(env_t* e) 
792 {
793         if (num_packet_buffers != 0) 
794                 return 1;
795         else
796                 return 0;
797 }
798
799 #endif // Network
800
801 // Syscalls below here are serviced by the appserver for now.
802 #define ufe(which,a0,a1,a2,a3) \
803         frontend_syscall_errno(p,APPSERVER_SYSCALL_##which,\
804                            (int)(a0),(int)(a1),(int)(a2),(int)(a3))
805
806 intreg_t sys_write(struct proc* p, int fd, const void* buf, int len)
807 {
808         int ret = 0;
809         void* kbuf = user_memdup_errno(p,buf,len);
810         if(kbuf == NULL)
811                 return -1;
812 #ifndef __CONFIG_APPSERVER__
813         /* Catch a common usage of stderr */
814         if (fd == 2) {
815                 ((char*)kbuf)[len-1] = 0;
816                 printk("[stderr]: %s\n", kbuf);
817                 ret = len;
818         } else { // but warn/panic otherwise in ufe()
819                 ret = ufe(write, fd, PADDR(kbuf), len, 0);
820         }
821 #else
822         ret = ufe(write, fd, PADDR(kbuf), len, 0);
823 #endif
824         user_memdup_free(p,kbuf);
825         return ret;
826 }
827
828 intreg_t sys_read(struct proc* p, int fd, void* buf, int len)
829 {
830         void* kbuf = kmalloc_errno(len);
831         if(kbuf == NULL)
832                 return -1;
833         int ret = ufe(read,fd,PADDR(kbuf),len,0);
834         if(ret != -1 && memcpy_to_user_errno(p,buf,kbuf,len))
835                 ret = -1;
836         user_memdup_free(p,kbuf);
837         return ret;
838 }
839
840 intreg_t sys_pwrite(struct proc* p, int fd, const void* buf, int len, int offset)
841 {
842         void* kbuf = user_memdup_errno(p,buf,len);
843         if(kbuf == NULL)
844                 return -1;
845         int ret = ufe(pwrite,fd,PADDR(kbuf),len,offset);
846         user_memdup_free(p,kbuf);
847         return ret;
848 }
849
850 intreg_t sys_pread(struct proc* p, int fd, void* buf, int len, int offset)
851 {
852         void* kbuf = kmalloc_errno(len);
853         if(kbuf == NULL)
854                 return -1;
855         int ret = ufe(pread,fd,PADDR(kbuf),len,offset);
856         if(ret != -1 && memcpy_to_user_errno(p,buf,kbuf,len))
857                 ret = -1;
858         user_memdup_free(p,kbuf);
859         return ret;
860 }
861
862 intreg_t sys_open(struct proc* p, const char* path, int oflag, int mode)
863 {
864         printd("File Open, p: %p, path: %s, oflag: %d, mode: 0x%x\n", p, path, oflag, mode);
865         char* fn = user_strdup_errno(p,path,PGSIZE);
866         if(fn == NULL) {
867                 printd("File Open, user_strdup_errno failed\n");
868                 return -1;
869         }
870         printd("File Open, About to open\n");
871         int ret = ufe(open,PADDR(fn),oflag,mode,0);
872         printd("File Open, res=%d\n", ret);
873         user_memdup_free(p,fn);
874         return ret;
875 }
876 intreg_t sys_close(struct proc* p, int fd)
877 {
878         return ufe(close,fd,0,0,0);
879 }
880
881 #define NEWLIB_STAT_SIZE 64
882 intreg_t sys_fstat(struct proc* p, int fd, void* buf)
883 {
884         int *kbuf = kmalloc(NEWLIB_STAT_SIZE, 0);
885         int ret = ufe(fstat,fd,PADDR(kbuf),0,0);
886         if(ret != -1 && memcpy_to_user_errno(p,buf,kbuf,NEWLIB_STAT_SIZE))
887                 ret = -1;
888         kfree(kbuf);
889         return ret;
890 }
891
892 intreg_t sys_stat(struct proc* p, const char* path, void* buf)
893 {
894         char* fn = user_strdup_errno(p,path,PGSIZE);
895         if(fn == NULL)
896                 return -1;
897
898         int *kbuf = kmalloc(NEWLIB_STAT_SIZE, 0);
899         int ret = ufe(stat,PADDR(fn),PADDR(kbuf),0,0);
900         if(ret != -1 && memcpy_to_user_errno(p,buf,kbuf,NEWLIB_STAT_SIZE))
901                 ret = -1;
902
903         user_memdup_free(p,fn);
904         kfree(kbuf);
905         return ret;
906 }
907
908 intreg_t sys_lstat(struct proc* p, const char* path, void* buf)
909 {
910         char* fn = user_strdup_errno(p,path,PGSIZE);
911         if(fn == NULL)
912                 return -1;
913
914         int *kbuf = kmalloc(NEWLIB_STAT_SIZE, 0);
915         int ret = ufe(lstat,PADDR(fn),PADDR(kbuf),0,0);
916         if(ret != -1 && memcpy_to_user_errno(p,buf,kbuf,NEWLIB_STAT_SIZE))
917                 ret = -1;
918
919         user_memdup_free(p,fn);
920         kfree(kbuf);
921         return ret;
922 }
923
924 intreg_t sys_fcntl(struct proc* p, int fd, int cmd, int arg)
925 {
926         return ufe(fcntl,fd,cmd,arg,0);
927 }
928
929 intreg_t sys_access(struct proc* p, const char* path, int type)
930 {
931         char* fn = user_strdup_errno(p,path,PGSIZE);
932         if(fn == NULL)
933                 return -1;
934         int ret = ufe(access,PADDR(fn),type,0,0);
935         user_memdup_free(p,fn);
936         return ret;
937 }
938
939 intreg_t sys_umask(struct proc* p, int mask)
940 {
941         return ufe(umask,mask,0,0,0);
942 }
943
944 intreg_t sys_chmod(struct proc* p, const char* path, int mode)
945 {
946         char* fn = user_strdup_errno(p,path,PGSIZE);
947         if(fn == NULL)
948                 return -1;
949         int ret = ufe(chmod,PADDR(fn),mode,0,0);
950         user_memdup_free(p,fn);
951         return ret;
952 }
953
954 intreg_t sys_lseek(struct proc* p, int fd, int offset, int whence)
955 {
956         return ufe(lseek,fd,offset,whence,0);
957 }
958
959 intreg_t sys_link(struct proc* p, const char* _old, const char* _new)
960 {
961         char* oldpath = user_strdup_errno(p,_old,PGSIZE);
962         if(oldpath == NULL)
963                 return -1;
964
965         char* newpath = user_strdup_errno(p,_new,PGSIZE);
966         if(newpath == NULL)
967         {
968                 user_memdup_free(p,oldpath);
969                 return -1;
970         }
971
972         int ret = ufe(link,PADDR(oldpath),PADDR(newpath),0,0);
973         user_memdup_free(p,oldpath);
974         user_memdup_free(p,newpath);
975         return ret;
976 }
977
978 intreg_t sys_unlink(struct proc* p, const char* path)
979 {
980         char* fn = user_strdup_errno(p,path,PGSIZE);
981         if(fn == NULL)
982                 return -1;
983         int ret = ufe(unlink,PADDR(fn),0,0,0);
984         user_memdup_free(p,fn);
985         return ret;
986 }
987
988 intreg_t sys_chdir(struct proc* p, const char* path)
989 {
990         char* fn = user_strdup_errno(p,path,PGSIZE);
991         if(fn == NULL)
992                 return -1;
993         int ret = ufe(chdir,PADDR(fn),0,0,0);
994         user_memdup_free(p,fn);
995         return ret;
996 }
997
998 intreg_t sys_getcwd(struct proc* p, char* pwd, int size)
999 {
1000         void* kbuf = kmalloc_errno(size);
1001         if(kbuf == NULL)
1002                 return -1;
1003         int ret = ufe(read,PADDR(kbuf),size,0,0);
1004         if(ret != -1 && memcpy_to_user_errno(p,pwd,kbuf,strnlen(kbuf,size)))
1005                 ret = -1;
1006         user_memdup_free(p,kbuf);
1007         return ret;
1008 }
1009
1010 intreg_t sys_gettimeofday(struct proc* p, int* buf)
1011 {
1012         static spinlock_t gtod_lock = SPINLOCK_INITIALIZER;
1013         static int t0 = 0;
1014
1015         spin_lock(&gtod_lock);
1016         if(t0 == 0)
1017
1018 #if (defined __CONFIG_APPSERVER__) && (!defined __CONFIG_OSDI__)
1019         // For OSDI, do not get time from appserver because it would lead to inaccurate measurements.
1020         t0 = ufe(time,0,0,0,0);
1021 #else
1022         // Nanwan's birthday, bitches!!
1023         t0 = 1242129600;
1024 #endif 
1025         spin_unlock(&gtod_lock);
1026
1027         long long dt = read_tsc();
1028         int kbuf[2] = {t0+dt/system_timing.tsc_freq,
1029             (dt%system_timing.tsc_freq)*1000000/system_timing.tsc_freq};
1030
1031         return memcpy_to_user_errno(p,buf,kbuf,sizeof(kbuf));
1032 }
1033
1034 #define SIZEOF_STRUCT_TERMIOS 60
1035 intreg_t sys_tcgetattr(struct proc* p, int fd, void* termios_p)
1036 {
1037         int* kbuf = kmalloc(SIZEOF_STRUCT_TERMIOS,0);
1038         int ret = ufe(tcgetattr,fd,PADDR(kbuf),0,0);
1039         if(ret != -1 && memcpy_to_user_errno(p,termios_p,kbuf,SIZEOF_STRUCT_TERMIOS))
1040                 ret = -1;
1041         kfree(kbuf);
1042         return ret;
1043 }
1044
1045 intreg_t sys_tcsetattr(struct proc* p, int fd, int optional_actions, const void* termios_p)
1046 {
1047         void* kbuf = user_memdup_errno(p,termios_p,SIZEOF_STRUCT_TERMIOS);
1048         if(kbuf == NULL)
1049                 return -1;
1050         int ret = ufe(tcsetattr,fd,optional_actions,PADDR(kbuf),0);
1051         user_memdup_free(p,kbuf);
1052         return ret;
1053 }
1054
1055 intreg_t sys_fillmeup(struct proc *p, uint8_t *bufs, 
1056                       uint16_t num_bufs, int32_t *last_written)
1057 {
1058 #if defined(__CONFIG_OSDI__) && defined(__CONFIG_NETWORKING__)
1059         extern struct fillmeup fillmeup_data;
1060         fillmeup_data.proc = p;
1061         fillmeup_data.bufs = bufs;
1062         fillmeup_data.num_bufs = num_bufs;
1063         fillmeup_data.last_written = last_written;
1064         *last_written = -1;
1065         return 0;
1066 #else
1067         return -1;
1068 #endif
1069 }
1070
1071 /************** Syscall Invokation **************/
1072
1073 /* Executes the given syscall.
1074  *
1075  * Note tf is passed in, which points to the tf of the context on the kernel
1076  * stack.  If any syscall needs to block, it needs to save this info, as well as
1077  * any silly state.
1078  *
1079  * TODO: Build a dispatch table instead of switching on the syscallno
1080  * Dispatches to the correct kernel function, passing the arguments.
1081  */
1082 intreg_t syscall(struct proc *p, uintreg_t syscallno, uintreg_t a1,
1083                  uintreg_t a2, uintreg_t a3, uintreg_t a4, uintreg_t a5)
1084 {
1085         // Initialize the return value and error code returned to 0
1086         proc_set_syscall_retval(current_tf, 0);
1087         set_errno(current_tf,0);
1088
1089         typedef intreg_t (*syscall_t)(struct proc*,uintreg_t,uintreg_t,
1090                                       uintreg_t,uintreg_t,uintreg_t);
1091
1092         const static syscall_t syscall_table[] = {
1093                 [SYS_null] = (syscall_t)sys_null,
1094                 [SYS_cache_buster] = (syscall_t)sys_cache_buster,
1095                 [SYS_cache_invalidate] = (syscall_t)sys_cache_invalidate,
1096                 [SYS_reboot] = (syscall_t)reboot,
1097                 [SYS_cputs] = (syscall_t)sys_cputs,
1098                 [SYS_cgetc] = (syscall_t)sys_cgetc,
1099                 [SYS_getcpuid] = (syscall_t)sys_getcpuid,
1100                 [SYS_getvcoreid] = (syscall_t)sys_getvcoreid,
1101                 [SYS_getpid] = (syscall_t)sys_getpid,
1102                 [SYS_proc_create] = (syscall_t)sys_proc_create,
1103                 [SYS_proc_run] = (syscall_t)sys_proc_run,
1104                 [SYS_proc_destroy] = (syscall_t)sys_proc_destroy,
1105                 [SYS_yield] = (syscall_t)sys_proc_yield,
1106                 [SYS_run_binary] = (syscall_t)sys_run_binary,
1107                 [SYS_fork] = (syscall_t)sys_fork,
1108                 [SYS_exec] = (syscall_t)sys_exec,
1109                 [SYS_trywait] = (syscall_t)sys_trywait,
1110                 [SYS_mmap] = (syscall_t)sys_mmap,
1111                 [SYS_munmap] = (syscall_t)sys_munmap,
1112                 [SYS_mprotect] = (syscall_t)sys_mprotect,
1113                 [SYS_brk] = (syscall_t)sys_brk,
1114                 [SYS_shared_page_alloc] = (syscall_t)sys_shared_page_alloc,
1115                 [SYS_shared_page_free] = (syscall_t)sys_shared_page_free,
1116                 [SYS_resource_req] = (syscall_t)resource_req,
1117                 [SYS_notify] = (syscall_t)sys_notify,
1118                 [SYS_self_notify] = (syscall_t)sys_self_notify,
1119                 [SYS_halt_core] = (syscall_t)sys_halt_core,
1120         #ifdef __CONFIG_SERIAL_IO__
1121                 [SYS_serial_read] = (syscall_t)sys_serial_read,
1122                 [SYS_serial_write] = (syscall_t)sys_serial_write,
1123         #endif
1124         #ifdef __CONFIG_NETWORKING__
1125                 [SYS_eth_read] = (syscall_t)sys_eth_read,
1126                 [SYS_eth_write] = (syscall_t)sys_eth_write,
1127                 [SYS_eth_get_mac_addr] = (syscall_t)sys_eth_get_mac_addr,
1128                 [SYS_eth_recv_check] = (syscall_t)sys_eth_recv_check,
1129         #endif
1130         #ifdef __CONFIG_OSDI__
1131                 [SYS_fillmeup] = (syscall_t)sys_fillmeup,
1132         #endif
1133                 // Syscalls serviced by the appserver for now.
1134                 [SYS_read] = (syscall_t)sys_read,
1135                 [SYS_write] = (syscall_t)sys_write,
1136                 [SYS_open] = (syscall_t)sys_open,
1137                 [SYS_close] = (syscall_t)sys_close,
1138                 [SYS_fstat] = (syscall_t)sys_fstat,
1139                 [SYS_stat] = (syscall_t)sys_stat,
1140                 [SYS_lstat] = (syscall_t)sys_lstat,
1141                 [SYS_fcntl] = (syscall_t)sys_fcntl,
1142                 [SYS_access] = (syscall_t)sys_access,
1143                 [SYS_umask] = (syscall_t)sys_umask,
1144                 [SYS_chmod] = (syscall_t)sys_chmod,
1145                 [SYS_lseek] = (syscall_t)sys_lseek,
1146                 [SYS_link] = (syscall_t)sys_link,
1147                 [SYS_unlink] = (syscall_t)sys_unlink,
1148                 [SYS_chdir] = (syscall_t)sys_chdir,
1149                 [SYS_getcwd] = (syscall_t)sys_getcwd,
1150                 [SYS_gettimeofday] = (syscall_t)sys_gettimeofday,
1151                 [SYS_tcgetattr] = (syscall_t)sys_tcgetattr,
1152                 [SYS_tcsetattr] = (syscall_t)sys_tcsetattr
1153         };
1154
1155         const int max_syscall = sizeof(syscall_table)/sizeof(syscall_table[0]);
1156
1157         uint32_t coreid, vcoreid;
1158         if (systrace_flags & SYSTRACE_ON) {
1159                 if ((systrace_flags & SYSTRACE_ALLPROC) || (proc_is_traced(p))) {
1160                         coreid = core_id();
1161                         vcoreid = proc_get_vcoreid(p, core_id());
1162                         if (systrace_flags & SYSTRACE_LOUD) {
1163                                 printk("[%16llu] Syscall %d for proc %d on core %d, vcore %d\n",
1164                                        read_tsc(), syscallno, p->pid, coreid, vcoreid);
1165                         } else {
1166                                 struct systrace_record *trace;
1167                                 unsigned int idx, new_idx;
1168                                 do {
1169                                         idx = systrace_bufidx;
1170                                         new_idx = (idx + 1) % systrace_bufsize;
1171                                 } while (!atomic_comp_swap(&systrace_bufidx, idx, new_idx));
1172                                 trace = &systrace_buffer[idx];
1173                                 trace->timestamp = read_tsc();
1174                                 trace->syscallno = syscallno;
1175                                 trace->pid = p->pid;
1176                                 trace->coreid = coreid;
1177                                 trace->vcoreid = vcoreid;
1178                         }
1179                 }
1180         }
1181         //printk("Incoming syscall on core: %d number: %d\n    a1: %x\n   "
1182         //       " a2: %x\n    a3: %x\n    a4: %x\n    a5: %x\n", core_id(),
1183         //       syscallno, a1, a2, a3, a4, a5);
1184
1185         if(syscallno > max_syscall || syscall_table[syscallno] == NULL)
1186                 panic("Invalid syscall number %d for proc %x!", syscallno, *p);
1187
1188         return syscall_table[syscallno](p,a1,a2,a3,a4,a5);
1189 }
1190
1191 intreg_t syscall_async(struct proc *p, syscall_req_t *call)
1192 {
1193         return syscall(p, call->num, call->args[0], call->args[1],
1194                        call->args[2], call->args[3], call->args[4]);
1195 }
1196
1197 /* You should already have a refcnt'd ref to p before calling this */
1198 intreg_t process_generic_syscalls(struct proc *p, size_t max)
1199 {
1200         size_t count = 0;
1201         syscall_back_ring_t* sysbr = &p->syscallbackring;
1202
1203         /* make sure the proc is still alive, and keep it from dying from under us
1204          * incref will return ESUCCESS on success.  This might need some thought
1205          * regarding when the incref should have happened (like by whoever passed us
1206          * the *p). */
1207         // TODO: ought to be unnecessary, if you called this right, kept here for
1208         // now in case anyone actually uses the ARSCs.
1209         proc_incref(p, 1);
1210
1211         // max is the most we'll process.  max = 0 means do as many as possible
1212         while (RING_HAS_UNCONSUMED_REQUESTS(sysbr) && ((!max)||(count < max)) ) {
1213                 if (!count) {
1214                         // ASSUME: one queue per process
1215                         // only switch cr3 for the very first request for this queue
1216                         // need to switch to the right context, so we can handle the user pointer
1217                         // that points to a data payload of the syscall
1218                         lcr3(p->env_cr3);
1219                 }
1220                 count++;
1221                 //printk("DEBUG PRE: sring->req_prod: %d, sring->rsp_prod: %d\n",
1222                 //         sysbr->sring->req_prod, sysbr->sring->rsp_prod);
1223                 // might want to think about 0-ing this out, if we aren't
1224                 // going to explicitly fill in all fields
1225                 syscall_rsp_t rsp;
1226                 // this assumes we get our answer immediately for the syscall.
1227                 syscall_req_t* req = RING_GET_REQUEST(sysbr, ++(sysbr->req_cons));
1228                 rsp.retval = syscall_async(p, req);
1229                 // write response into the slot it came from
1230                 memcpy(req, &rsp, sizeof(syscall_rsp_t));
1231                 // update our counter for what we've produced (assumes we went in order!)
1232                 (sysbr->rsp_prod_pvt)++;
1233                 RING_PUSH_RESPONSES(sysbr);
1234                 //printk("DEBUG POST: sring->req_prod: %d, sring->rsp_prod: %d\n",
1235                 //         sysbr->sring->req_prod, sysbr->sring->rsp_prod);
1236         }
1237         // load sane page tables (and don't rely on decref to do it for you).
1238         lcr3(boot_cr3);
1239         proc_decref(p, 1);
1240         return (intreg_t)count;
1241 }
1242
1243 /* Syscall tracing */
1244 static void __init_systrace(void)
1245 {
1246         systrace_buffer = kmalloc(MAX_SYSTRACES*sizeof(struct systrace_record), 0);
1247         if (!systrace_buffer)
1248                 panic("Unable to alloc a trace buffer\n");
1249         systrace_bufidx = 0;
1250         systrace_bufsize = MAX_SYSTRACES;
1251         /* Note we never free the buffer - it's around forever.  Feel free to change
1252          * this if you want to change the size or something dynamically. */
1253 }
1254
1255 /* If you call this while it is running, it will change the mode */
1256 void systrace_start(bool silent)
1257 {
1258         static bool init = FALSE;
1259         spin_lock_irqsave(&systrace_lock);
1260         if (!init) {
1261                 __init_systrace();
1262                 init = TRUE;
1263         }
1264         systrace_flags = silent ? SYSTRACE_ON : SYSTRACE_ON | SYSTRACE_LOUD; 
1265         spin_unlock_irqsave(&systrace_lock);
1266 }
1267
1268 int systrace_reg(bool all, struct proc *p)
1269 {
1270         int retval = 0;
1271         spin_lock_irqsave(&systrace_lock);
1272         if (all) {
1273                 printk("Tracing syscalls for all processes\n");
1274                 systrace_flags |= SYSTRACE_ALLPROC;
1275                 retval = 0;
1276         } else {
1277                 for (int i = 0; i < MAX_NUM_TRACED; i++) {
1278                         if (!systrace_procs[i]) {
1279                                 printk("Tracing syscalls for process %d\n", p->pid);
1280                                 systrace_procs[i] = p;
1281                                 retval = 0;
1282                                 break;
1283                         }
1284                 }
1285         }
1286         spin_unlock_irqsave(&systrace_lock);
1287         return retval;
1288 }
1289
1290 void systrace_stop(void)
1291 {
1292         spin_lock_irqsave(&systrace_lock);
1293         systrace_flags = 0;
1294         for (int i = 0; i < MAX_NUM_TRACED; i++)
1295                 systrace_procs[i] = 0;
1296         spin_unlock_irqsave(&systrace_lock);
1297 }
1298
1299 /* If you registered a process specifically, then you need to dereg it
1300  * specifically.  Or just fully stop, which will do it for all. */
1301 int systrace_dereg(bool all, struct proc *p)
1302 {
1303         spin_lock_irqsave(&systrace_lock);
1304         if (all) {
1305                 printk("No longer tracing syscalls for all processes.\n");
1306                 systrace_flags &= ~SYSTRACE_ALLPROC;
1307         } else {
1308                 for (int i = 0; i < MAX_NUM_TRACED; i++) {
1309                         if (systrace_procs[i] == p) {
1310                                 systrace_procs[i] = 0;
1311                                 printk("No longer tracing syscalls for process %d\n", p->pid);
1312                         }
1313                 }
1314         }
1315         spin_unlock_irqsave(&systrace_lock);
1316         return 0;
1317 }
1318
1319 /* Regardless of locking, someone could be writing into the buffer */
1320 void systrace_print(bool all, struct proc *p)
1321 {
1322         spin_lock_irqsave(&systrace_lock);
1323         /* if you want to be clever, you could make this start from the earliest
1324          * timestamp and loop around.  Careful of concurrent writes. */
1325         for (int i = 0; i < systrace_bufsize; i++)
1326                 if (systrace_buffer[i].timestamp)
1327                         printk("[%16llu] Syscall %d for proc %d on core %d, vcore %d\n",
1328                                systrace_buffer[i].timestamp,
1329                                systrace_buffer[i].syscallno,
1330                                systrace_buffer[i].pid,
1331                                systrace_buffer[i].coreid,
1332                                systrace_buffer[i].vcoreid);
1333         spin_unlock_irqsave(&systrace_lock);
1334 }
1335
1336 void systrace_clear_buffer(void)
1337 {
1338         spin_lock_irqsave(&systrace_lock);
1339         memset(systrace_buffer, 0, sizeof(struct systrace_record)*MAX_NUM_TRACED);
1340         spin_unlock_irqsave(&systrace_lock);
1341 }