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