maxjhandsome
/
llvm-project
forked from OSchip/llvm-project
1
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

[sanitizer] Implement wait4 and waitpid syscall hooks. 

llvm-svn: 179592
This commit is contained in:
Evgeniy Stepanov 2013-04-16 13:06:20 +00:00
parent d8bb4b7e06
commit 9b59a39f42
2 changed files with 43 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
compiler-rt/include/sanitizer/linux_syscall_hooks.h
View File

@ -29,11 +29,15 @@ void __sanitizer_syscall_pre_rt_sigpending(void *p, size_t s);
void __sanitizer_syscall_pre_getdents(int fd, void *dirp, int count);
void __sanitizer_syscall_pre_getdents64(int fd, void *dirp, int count);
void __sanitizer_syscall_pre_recvmsg(int sockfd, void *msg, int flags);
void __sanitizer_syscall_pre_wait4(int pid, int* status, int options, void* r);
void __sanitizer_syscall_pre_waitpid(int pid, int *status, int options);
void __sanitizer_syscall_post_rt_sigpending(long res, void *p, size_t s);
void __sanitizer_syscall_post_getdents(long res, int fd, void *dirp, int count);
void __sanitizer_syscall_post_getdents64(long res, int fd, void *dirp, int count);
void __sanitizer_syscall_post_recvmsg(long res, int sockfd, void *msg, int flags);
void __sanitizer_syscall_post_wait4(long res, int pid, int* status, int options, void* r);
void __sanitizer_syscall_post_waitpid(long res, int pid, int *status, int options);
// And now a few syscalls we don't handle yet.
@ -408,11 +412,10 @@ void __sanitizer_syscall_post_recvmsg(long res, int sockfd, void *msg, int flags
#define __sanitizer_syscall_pre_vm86old(...)
#define __sanitizer_syscall_pre_vmsplice(...)
#define __sanitizer_syscall_pre_vserver(...)
#define __sanitizer_syscall_pre_wait4(...)
#define __sanitizer_syscall_pre_waitid(...)
#define __sanitizer_syscall_pre_waitpid(...)
#define __sanitizer_syscall_pre_write(...)
#define __sanitizer_syscall_pre_writev(...)
#define __sanitizer_syscall_post_accept4(res, ...)
#define __sanitizer_syscall_post_accept(res, ...)
#define __sanitizer_syscall_post_access(res, ...)
@ -784,9 +787,7 @@ void __sanitizer_syscall_post_recvmsg(long res, int sockfd, void *msg, int flags
#define __sanitizer_syscall_post_vm86(res, ...)
#define __sanitizer_syscall_post_vmsplice(res, ...)
#define __sanitizer_syscall_post_vserver(res, ...)
#define __sanitizer_syscall_post_wait4(res, ...)
#define __sanitizer_syscall_post_waitid(res, ...)
#define __sanitizer_syscall_post_waitpid(res, ...)
#define __sanitizer_syscall_post_write(res, ...)
#define __sanitizer_syscall_post_writev(res, ...)

38
compiler-rt/lib/sanitizer_common/sanitizer_common_syscalls.inc
View File

@ -56,6 +56,16 @@ struct sanitizer_kernel_msghdr {
unsigned msg_flags;
};
struct sanitizer_kernel_timeval {
long tv_sec;
long tv_usec;
};
struct sanitizer_kernel_rusage {
struct sanitizer_kernel_timeval ru_timeval[2];
long ru_long[14];
};
PRE_SYSCALL(recvmsg)(int sockfd, struct sanitizer_kernel_msghdr *msg,
int flags) {
PRE_READ(msg, sizeof(*msg));
@ -92,6 +102,34 @@ POST_SYSCALL(getdents64)(long res, int fd, void *dirp, int count) {
POST_WRITE(dirp, res);
}
PRE_SYSCALL(wait4)(int pid, int *status, int options,
struct sanitizer_kernel_rusage *r) {
if (status)
PRE_WRITE(status, sizeof(*status));
if (r)
PRE_WRITE(r, sizeof(*r));
}
POST_SYSCALL(wait4)(long res, int pid, int *status, int options,
struct sanitizer_kernel_rusage *r) {
if (res > 0) {
if (status)
POST_WRITE(status, sizeof(*status));
if (r)
POST_WRITE(r, sizeof(*r));
}
}
PRE_SYSCALL(waitpid)(int pid, int *status, int options) {
if (status)
PRE_WRITE(status, sizeof(*status));
}
POST_SYSCALL(waitpid)(long res, int pid, int *status, int options) {
if (res > 0 && status)
POST_WRITE(status, sizeof(*status));
}
} // extern "C"
#undef PRE_SYSCALL