ref: 7a8e875668d92020548d07b5ad30f7b83b61d1fc
dir: /sys/src/9/port/dtracysys.c/
#include "u.h" #include "../port/lib.h" #include "mem.h" #include "dat.h" #include "fns.h" #include "../port/error.h" #include "/sys/src/libc/9syscall/sys.h" #include <dtracy.h> #include <ctype.h> static DTProbe **dtpsysentry, **dtpsysreturn; typedef uintptr Syscall(va_list); extern Syscall *systab[]; #define WRAP0(x,y,z)\ Syscall z; uintptr x(va_list va){\ uintptr rc;\ dtptrigger(dtpsysentry[y], m->machno, 0, 0, 0, 0);\ rc = z(va);\ dtptrigger(dtpsysreturn[y], m->machno, 0, 0, 0, 0);\ return rc;\ } #define WRAP1(x,y,z,type0)\ Syscall z; uintptr x(va_list va){\ uintptr rc;\ va_list vb = va;\ type0 arg0 = va_arg(vb, type0);\ dtptrigger(dtpsysentry[y], m->machno, (uvlong)arg0, 0, 0, 0);\ rc = z(va);\ dtptrigger(dtpsysreturn[y], m->machno, (uvlong)arg0, 0, 0, 0);\ return rc;\ } #define WRAP2(x,y,z,type0,type1)\ Syscall z; uintptr x(va_list va){\ uintptr rc;\ va_list vb = va;\ type0 arg0 = va_arg(vb, type0);\ type1 arg1 = va_arg(vb, type1);\ dtptrigger(dtpsysentry[y], m->machno, (uvlong)arg0, (uvlong)arg1, 0, 0);\ rc = z(va);\ dtptrigger(dtpsysreturn[y], m->machno, (uvlong)arg0, (uvlong)arg1, 0, 0);\ return rc;\ } #define WRAP3(x,y,z,type0,type1,type2)\ Syscall z; uintptr x(va_list va){\ uintptr rc;\ va_list vb = va;\ type0 arg0 = va_arg(vb, type0);\ type1 arg1 = va_arg(vb, type1);\ type2 arg2 = va_arg(vb, type2);\ dtptrigger(dtpsysentry[y], m->machno, (uvlong)arg0, (uvlong)arg1, (uvlong)arg2, 0);\ rc = z(va);\ dtptrigger(dtpsysreturn[y], m->machno, (uvlong)arg0, (uvlong)arg1, (uvlong)arg2, 0);\ return rc;\ } #define WRAP4(x,y,z,type0,type1,type2,type3)\ Syscall z; uintptr x(va_list va){\ uintptr rc;\ va_list vb = va;\ type0 arg0 = va_arg(vb, type0);\ type1 arg1 = va_arg(vb, type1);\ type2 arg2 = va_arg(vb, type2);\ type3 arg3 = va_arg(vb, type3);\ dtptrigger(dtpsysentry[y], m->machno, (uvlong)arg0, (uvlong)arg1, (uvlong)arg2, (uvlong)arg3);\ rc = z(va);\ dtptrigger(dtpsysreturn[y], m->machno, (uvlong)arg0, (uvlong)arg1, (uvlong)arg2, (uvlong)arg3);\ return rc;\ } /*TODO*/ #define WRAP5(x,y,z,type0,type1,type2,type3,type4)\ Syscall z; uintptr x(va_list va){\ uintptr rc;\ va_list vb = va;\ type0 arg0 = va_arg(vb, type0);\ type1 arg1 = va_arg(vb, type1);\ type2 arg2 = va_arg(vb, type2);\ type3 arg3 = va_arg(vb, type3);\ dtptrigger(dtpsysentry[y], m->machno, (uvlong)arg0, (uvlong)arg1, (uvlong)arg2, (uvlong)arg3);\ rc = z(va);\ dtptrigger(dtpsysreturn[y], m->machno, (uvlong)arg0, (uvlong)arg1, (uvlong)arg2, (uvlong)arg3);\ return rc;\ } WRAP0(dtwrap_sysr1, SYSR1, sysr1) WRAP1(dtwrap_sys_errstr, _ERRSTR, sys_errstr, char*) WRAP3(dtwrap_sysbind, BIND, sysbind, char*, char*, int) WRAP1(dtwrap_syschdir, CHDIR, syschdir, char*) WRAP1(dtwrap_sysclose, CLOSE, sysclose, int) WRAP2(dtwrap_sysdup, DUP, sysdup, int, int) WRAP1(dtwrap_sysalarm, ALARM, sysalarm, ulong) WRAP2(dtwrap_sysexec, EXEC, sysexec, char *, char **) WRAP1(dtwrap_sysexits, EXITS, sysexits, char *) WRAP3(dtwrap_sys_fsession, _FSESSION, sys_fsession, int, char *, uint) WRAP2(dtwrap_sysfauth, FAUTH, sysfauth, int, char *) WRAP2(dtwrap_sys_fstat, _FSTAT, sys_fstat, int, uchar *) WRAP1(dtwrap_syssegbrk, SEGBRK, syssegbrk, void *) WRAP4(dtwrap_sys_mount, _MOUNT, sys_mount, int, char *, int, char *) WRAP2(dtwrap_sysopen, OPEN, sysopen, char *, int) WRAP3(dtwrap_sys_read, _READ, sys_read, int, void*, long) WRAP3(dtwrap_sysoseek, OSEEK, sysoseek, int, long, int) WRAP1(dtwrap_syssleep, SLEEP, syssleep, long) WRAP2(dtwrap_sys_stat, _STAT, sys_stat, char *, uchar *) WRAP1(dtwrap_sysrfork, RFORK, sysrfork, int) WRAP3(dtwrap_sys_write, _WRITE, sys_write, int, void *, long) WRAP1(dtwrap_syspipe, PIPE, syspipe, int*) WRAP3(dtwrap_syscreate, CREATE, syscreate, char*, int, int) WRAP3(dtwrap_sysfd2path, FD2PATH, sysfd2path, int, char*, uint) WRAP1(dtwrap_sysbrk_, BRK_, sysbrk_, uintptr) WRAP1(dtwrap_sysremove, REMOVE, sysremove, char *) WRAP0(dtwrap_sys_wstat, _WSTAT, sys_wstat) WRAP0(dtwrap_sys_fwstat, _FWSTAT, sys_fwstat) WRAP2(dtwrap_sysnotify, NOTIFY, sysnotify, char *, void *) WRAP1(dtwrap_sysnoted, NOTED, sysnoted, int) WRAP4(dtwrap_syssegattach, SEGATTACH, syssegattach, int, char *, uintptr, ulong) WRAP1(dtwrap_syssegdetach, SEGDETACH, syssegdetach, uintptr) WRAP2(dtwrap_syssegfree, SEGFREE, syssegfree, uintptr, ulong) WRAP2(dtwrap_syssegflush, SEGFLUSH, syssegflush, void*, ulong) WRAP2(dtwrap_sysrendezvous, RENDEZVOUS, sysrendezvous, uintptr, uintptr) WRAP2(dtwrap_sysunmount, UNMOUNT, sysunmount, char *, char *) WRAP1(dtwrap_sys_wait, _WAIT, sys_wait, void*) WRAP2(dtwrap_syssemacquire, SEMACQUIRE, syssemacquire, long*, int) WRAP2(dtwrap_syssemrelease, SEMRELEASE, syssemrelease, long*, long) WRAP4(dtwrap_sysfversion, FVERSION, sysfversion, int, int, char *, int) WRAP2(dtwrap_syserrstr, ERRSTR, syserrstr, char *, uint) WRAP3(dtwrap_sysstat, STAT, sysstat, char *, uchar *, uint) WRAP3(dtwrap_sysfstat, FSTAT, sysfstat, int, uchar *, uint) WRAP3(dtwrap_syswstat, WSTAT, syswstat, char *, uchar *, uint) WRAP3(dtwrap_sysfwstat, FWSTAT, sysfwstat, int, uchar *, uint) WRAP5(dtwrap_sysmount, MOUNT, sysmount, int, int, char *, int, char *) WRAP2(dtwrap_sysawait, AWAIT, sysawait, char *, uint) WRAP4(dtwrap_syspread, PREAD, syspread, int, void *, long, vlong) WRAP4(dtwrap_syspwrite, PWRITE, syspwrite, int, void *, long, vlong) WRAP2(dtwrap_systsemacquire, TSEMACQUIRE, systsemacquire, long *, ulong) /* TODO: amd64 */ WRAP4(dtwrap_sysseek, SEEK, sysseek, vlong*, int, vlong, int) WRAP1(dtwrap_sys_nsec, _NSEC, sys_nsec, vlong*) static Syscall *wraptab[]={ [SYSR1] dtwrap_sysr1, [_ERRSTR] dtwrap_sys_errstr, [BIND] dtwrap_sysbind, [CHDIR] dtwrap_syschdir, [CLOSE] dtwrap_sysclose, [DUP] dtwrap_sysdup, [ALARM] dtwrap_sysalarm, [EXEC] dtwrap_sysexec, [EXITS] dtwrap_sysexits, [_FSESSION] dtwrap_sys_fsession, [FAUTH] dtwrap_sysfauth, [_FSTAT] dtwrap_sys_fstat, [SEGBRK] dtwrap_syssegbrk, [_MOUNT] dtwrap_sys_mount, [OPEN] dtwrap_sysopen, [_READ] dtwrap_sys_read, [OSEEK] dtwrap_sysoseek, [SLEEP] dtwrap_syssleep, [_STAT] dtwrap_sys_stat, [RFORK] dtwrap_sysrfork, [_WRITE] dtwrap_sys_write, [PIPE] dtwrap_syspipe, [CREATE] dtwrap_syscreate, [FD2PATH] dtwrap_sysfd2path, [BRK_] dtwrap_sysbrk_, [REMOVE] dtwrap_sysremove, [_WSTAT] dtwrap_sys_wstat, [_FWSTAT] dtwrap_sys_fwstat, [NOTIFY] dtwrap_sysnotify, [NOTED] dtwrap_sysnoted, [SEGATTACH] dtwrap_syssegattach, [SEGDETACH] dtwrap_syssegdetach, [SEGFREE] dtwrap_syssegfree, [SEGFLUSH] dtwrap_syssegflush, [RENDEZVOUS] dtwrap_sysrendezvous, [UNMOUNT] dtwrap_sysunmount, [_WAIT] dtwrap_sys_wait, [SEMACQUIRE] dtwrap_syssemacquire, [SEMRELEASE] dtwrap_syssemrelease, [SEEK] dtwrap_sysseek, [FVERSION] dtwrap_sysfversion, [ERRSTR] dtwrap_syserrstr, [STAT] dtwrap_sysstat, [FSTAT] dtwrap_sysfstat, [WSTAT] dtwrap_syswstat, [FWSTAT] dtwrap_sysfwstat, [MOUNT] dtwrap_sysmount, [AWAIT] dtwrap_sysawait, [PREAD] dtwrap_syspread, [PWRITE] dtwrap_syspwrite, [TSEMACQUIRE] dtwrap_systsemacquire, [_NSEC] dtwrap_sys_nsec, }; static void sysprovide(DTProvider *prov, DTName) { static int provided; char buf[32]; int i; if(provided) return; provided = 1; dtpsysentry = smalloc(sizeof(Syscall *) * nsyscall); dtpsysreturn = smalloc(sizeof(Syscall *) * nsyscall); for(i = 0; i < nsyscall; i++){ if(systab[i] == nil || sysctab[i] == nil) continue; strecpy(buf, buf + sizeof(buf), sysctab[i]); if(isupper(buf[0])) buf[0] += 'a' - 'A'; if(i == SYSR1) strcpy(buf, "r1"); dtpsysentry[i] = dtpnew((DTName){"sys", buf, "entry"}, prov, (void *) i); dtpsysreturn[i] = dtpnew((DTName){"sys", buf, "return"}, prov, (void *) i); } } static int sysenable(DTProbe *p) { int i; Syscall *z; i = (int) p->aux; assert(i >= 0 && i < nsyscall); if(dtpsysentry[i]->nenable + dtpsysreturn[i]->nenable == 0) z = systab[i], systab[i] = wraptab[i], wraptab[i] = z; return 0; } static void sysdisable(DTProbe *p) { int i; Syscall *z; i = (int) p->aux; assert(i >= 0 && i < nsyscall); if(dtpsysentry[i]->nenable + dtpsysreturn[i]->nenable == 0) z = systab[i], systab[i] = wraptab[i], wraptab[i] = z; } DTProvider dtracyprov_sys = { .name = "sys", .provide = sysprovide, .enable = sysenable, .disable = sysdisable, };