451 lines
9.3 KiB
C
451 lines
9.3 KiB
C
#define __BSD_SOURCE
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <unistd.h>
|
|
#include <sys/types.h>
|
|
#include <sys/wait.h>
|
|
#include <sys/ptrace.h>
|
|
#include <sys/user.h>
|
|
#include <sys/syscall.h>
|
|
#include <fcntl.h>
|
|
#include <errno.h>
|
|
|
|
long
|
|
_ptrace(enum __ptrace_request request, pid_t pid, void *addr, void *data)
|
|
{
|
|
long rc;
|
|
|
|
rc = ptrace(request, pid, addr, data);
|
|
if (rc == -1) {
|
|
printf("ptrace(%d, %d, %016x, %016x): %d\n", request, pid,
|
|
(long)addr, (long)data, errno);
|
|
exit(1);
|
|
}
|
|
return rc;
|
|
}
|
|
|
|
typedef struct user_regs_struct syscall_args;
|
|
|
|
static inline int
|
|
get_syscall_args(int pid, syscall_args *args)
|
|
{
|
|
return _ptrace(PTRACE_GETREGS, pid, NULL, args);
|
|
}
|
|
|
|
static inline unsigned long
|
|
get_syscall_number(syscall_args *args)
|
|
{
|
|
return args->orig_rax;
|
|
}
|
|
|
|
static inline unsigned long
|
|
get_syscall_return(syscall_args *args)
|
|
{
|
|
return args->rax;
|
|
}
|
|
|
|
static char *syscalls[512];
|
|
|
|
char *
|
|
trim(char *buf)
|
|
{
|
|
char *p;
|
|
char *q;
|
|
|
|
for (p = buf; *p && (isspace(*p)); p++);
|
|
if (!*p)
|
|
return p;
|
|
for (q = strchr(p, '\0') - 1; isspace(*q); q--)
|
|
*q = '\0';
|
|
return p;
|
|
}
|
|
|
|
char **
|
|
split(char *buf, char dlm)
|
|
{
|
|
int n;
|
|
char *t;
|
|
char **r;
|
|
char **p;
|
|
|
|
for (n = 0, t = buf; *t; t++)
|
|
if (*t == dlm)
|
|
n++;
|
|
p = r = malloc(sizeof(char *) * (n + 2) + strlen(buf) + 1);
|
|
t = (char *)(r + n + 2);
|
|
strcpy(t, buf);
|
|
t = trim(t);
|
|
if (*t) {
|
|
*(p++) = t;
|
|
for (; *t; t++)
|
|
if (*t == dlm) {
|
|
*(t++) = '\0';
|
|
t = trim(t);
|
|
trim(p[-1]);
|
|
if (!*t)
|
|
break;
|
|
*(p++) = t;
|
|
}
|
|
}
|
|
*p = NULL;
|
|
return r;
|
|
}
|
|
|
|
void
|
|
init_syscalls()
|
|
{
|
|
char buf[1024];
|
|
FILE *f;
|
|
|
|
f = fopen("/usr/include/asm/unistd_64.h", "r");
|
|
if (!f) {
|
|
perror("open(unistd_64.h)");
|
|
return;
|
|
}
|
|
while (fgets(buf, 1024, f)) {
|
|
char *t;
|
|
char **a;
|
|
|
|
if (strncmp(buf, "#define", 7))
|
|
continue;
|
|
for (t = buf; *t; t++)
|
|
if (isspace(*t))
|
|
*t = ' ';
|
|
a = split(buf, ' ');
|
|
if (a[0] && a[1] && !strncmp(a[1], "__NR_", 5) &&
|
|
a[2] && *(a[2]) >= '0' && *(a[2]) <= '9') {
|
|
int num = atoi(a[2]);
|
|
syscalls[num] = strdup(a[1] + 5);
|
|
}
|
|
free(a);
|
|
}
|
|
fclose(f);
|
|
}
|
|
|
|
const char *
|
|
get_syscall(int n, char *buf)
|
|
{
|
|
if (n < 0 || n >= 512 || !syscalls[n]) {
|
|
sprintf(buf, "unknown(%d)", n);
|
|
return NULL;
|
|
}
|
|
return strcpy(buf, syscalls[n]);
|
|
}
|
|
|
|
int
|
|
main(int argc, char **argv)
|
|
{
|
|
pid_t pid;
|
|
pid_t cpid;
|
|
unsigned long msg;
|
|
int st;
|
|
int rc;
|
|
long ret;
|
|
int w;
|
|
syscall_args args;
|
|
unsigned long sig = 0;
|
|
char name[64];
|
|
int c = 0;
|
|
int ok = 0;
|
|
int ng = 0;
|
|
int procs = 0;
|
|
int forkev = 0;
|
|
int childfork = 0;
|
|
|
|
printf("#961 test start\n");
|
|
init_syscalls();
|
|
|
|
// printf("tracer pid=%d\n", getpid());
|
|
// fflush(stdout);
|
|
pid = fork();
|
|
if(pid == 0){
|
|
if (_ptrace(PTRACE_TRACEME, 0, NULL, NULL) == -1) {
|
|
printf("ptrace error %d\n", errno);
|
|
}
|
|
kill(getpid(), SIGINT);
|
|
|
|
cpid = fork();
|
|
if (cpid == 0) {
|
|
exit(1);
|
|
}
|
|
wait(&st);
|
|
exit(15);
|
|
}
|
|
rc = waitpid(-1, &st, WUNTRACED);
|
|
if (rc != pid || !WIFSTOPPED(st)) {
|
|
printf("BAD1 rc=%d st=%08x\n", rc, st);
|
|
exit(1);
|
|
}
|
|
_ptrace(PTRACE_SETOPTIONS, pid, NULL, (void *)(PTRACE_O_TRACESYSGOOD |
|
|
PTRACE_O_TRACEFORK |
|
|
PTRACE_O_TRACEVFORK |
|
|
PTRACE_O_TRACECLONE |
|
|
PTRACE_O_TRACEEXEC));
|
|
|
|
rc = pid;
|
|
procs = 1;
|
|
for (;;) {
|
|
if (rc != -1)
|
|
_ptrace(PTRACE_SYSCALL, rc, NULL, (void *)sig);
|
|
sig = 0;
|
|
rc = waitpid(-1, &st, WUNTRACED);
|
|
if (rc == -1) {
|
|
printf("wait error %d\n", errno);
|
|
exit(1);
|
|
}
|
|
if (WIFEXITED(st)) {
|
|
// printf("terminate %d st=%08x\n", rc, st);
|
|
procs--;
|
|
if (procs == 0)
|
|
break;
|
|
rc = -1;
|
|
continue;
|
|
}
|
|
if (WIFSIGNALED(st)) {
|
|
printf("tracee BAD signaled %08x\n", st);
|
|
ng++;
|
|
break;
|
|
}
|
|
if (!WIFSTOPPED(st)) {
|
|
printf("wait(1) rc=%d st=%08x\n", rc, st);
|
|
printf("unrecognized status rc=%d st=%08x\n", rc, st);
|
|
fflush(stdout);
|
|
continue;
|
|
}
|
|
if ((w = (st >> 16) & 255) == PTRACE_EVENT_FORK ||
|
|
w == PTRACE_EVENT_VFORK ||
|
|
w == PTRACE_EVENT_CLONE ||
|
|
w == PTRACE_EVENT_VFORK_DONE) {
|
|
int crc;
|
|
// printf("wait(1) rc=%d st=%08x\n", rc, st);
|
|
// printf("%d: fork exent ev=%d\n", rc, w);
|
|
// fflush(stdout);
|
|
_ptrace(PTRACE_GETEVENTMSG, pid, NULL, &msg);
|
|
cpid = msg;
|
|
// printf("child pid=%d\n", cpid);
|
|
// fflush(stdout);
|
|
crc = waitpid(-1, &st, WUNTRACED);
|
|
if (crc != cpid || !WIFSTOPPED(st)) {
|
|
printf("BAD4 rc=%d st=%08x\n", crc, st);
|
|
exit(1);
|
|
}
|
|
// printf("wait(2) rc=%d st=%08x\n", crc, st);
|
|
// fflush(stdout);
|
|
printf("#961-2 PTRACE_EVENT_FORK OK\n");
|
|
ok++;
|
|
forkev = 1;
|
|
procs++;
|
|
_ptrace(PTRACE_SYSCALL, cpid, NULL, (void *)0);
|
|
continue;
|
|
}
|
|
if (w == PTRACE_EVENT_EXEC) {
|
|
printf("wait(1) rc=%d st=%08x\n", rc, st);
|
|
printf("%d: exec event\n", rc);
|
|
fflush(stdout);
|
|
continue;
|
|
}
|
|
if (w == PTRACE_EVENT_EXIT) {
|
|
printf("wait(1) rc=%d st=%08x\n", rc, st);
|
|
printf("%d: exit event\n", rc);
|
|
fflush(stdout);
|
|
continue;
|
|
}
|
|
if (WSTOPSIG(st) & 0x80) { // syscall
|
|
int num;
|
|
get_syscall_args(rc, &args);
|
|
num = get_syscall_number(&args);
|
|
ret = get_syscall_return(&args);
|
|
if (num == SYS_clone) {
|
|
c++;
|
|
if (ret == -ENOSYS) {
|
|
ok++;
|
|
printf("#961-1 fork in OK\n");
|
|
}
|
|
else if (ret == 0) {
|
|
childfork = 1;
|
|
ng++;
|
|
printf("#961-4 fork child side called NG\n");
|
|
}
|
|
else {
|
|
printf("#961-3 fork out OK\n");
|
|
ok++;
|
|
}
|
|
}
|
|
}
|
|
else { // signal
|
|
// printf("wait(1) rc=%d st=%08x\n", rc, st);
|
|
sig = WSTOPSIG(st) & 0x7f;
|
|
if (sig == SIGCHLD) {
|
|
printf("#961-5 tracee receive SIGCHLD OK\n");
|
|
fflush(stdout);
|
|
ok++;
|
|
}
|
|
else if (sig == SIGSTOP) {
|
|
// printf("#961-6 tracee receive SIGSTOP OK %d\n", rc);
|
|
// fflush(stdout);
|
|
// ok++;
|
|
}
|
|
else {
|
|
ng++;
|
|
printf("tracee receive signal %d\n", sig);
|
|
fflush(stdout);
|
|
}
|
|
}
|
|
}
|
|
if (!childfork) {
|
|
ok++;
|
|
printf("#961-4 fork child side didnot be called OK\n");
|
|
}
|
|
if (!forkev) {
|
|
ng++;
|
|
printf("#961-2 didnot receive PTRACE_EVENT_FORK NG\n");
|
|
}
|
|
|
|
pid = fork();
|
|
if(pid == 0){
|
|
if (_ptrace(PTRACE_TRACEME, 0, NULL, NULL) == -1) {
|
|
printf("ptrace error %d\n", errno);
|
|
}
|
|
kill(getpid(), SIGINT);
|
|
|
|
cpid = fork();
|
|
if (cpid == 0) {
|
|
exit(1);
|
|
}
|
|
wait(&st);
|
|
exit(15);
|
|
}
|
|
rc = waitpid(-1, &st, WUNTRACED);
|
|
if (rc != pid || !WIFSTOPPED(st)) {
|
|
printf("BAD1 rc=%d st=%08x\n", rc, st);
|
|
exit(1);
|
|
}
|
|
_ptrace(PTRACE_SETOPTIONS, pid, NULL, (void *)(PTRACE_O_TRACESYSGOOD |
|
|
PTRACE_O_TRACEEXEC));
|
|
|
|
rc = pid;
|
|
forkev = 0;
|
|
childfork = 0;
|
|
procs = 1;
|
|
for (;;) {
|
|
if (rc != -1)
|
|
_ptrace(PTRACE_SYSCALL, rc, NULL, (void *)sig);
|
|
sig = 0;
|
|
rc = waitpid(-1, &st, WUNTRACED);
|
|
if (rc == -1) {
|
|
printf("wait error %d\n", errno);
|
|
exit(1);
|
|
}
|
|
if (WIFEXITED(st)) {
|
|
// printf("terminate %d st=%08x\n", rc, st);
|
|
procs--;
|
|
if (procs == 0)
|
|
break;
|
|
rc = -1;
|
|
continue;
|
|
}
|
|
if (WIFSIGNALED(st)) {
|
|
printf("tracee BAD signaled %08x\n", st);
|
|
ng++;
|
|
break;
|
|
}
|
|
if (!WIFSTOPPED(st)) {
|
|
printf("wait(1) rc=%d st=%08x\n", rc, st);
|
|
printf("unrecognized status rc=%d st=%08x\n", rc, st);
|
|
fflush(stdout);
|
|
continue;
|
|
}
|
|
if ((w = (st >> 16) & 255) == PTRACE_EVENT_FORK ||
|
|
w == PTRACE_EVENT_VFORK ||
|
|
w == PTRACE_EVENT_CLONE ||
|
|
w == PTRACE_EVENT_VFORK_DONE) {
|
|
int crc;
|
|
// printf("wait(1) rc=%d st=%08x\n", rc, st);
|
|
// printf("%d: fork exent ev=%d\n", rc, w);
|
|
// fflush(stdout);
|
|
_ptrace(PTRACE_GETEVENTMSG, pid, NULL, &msg);
|
|
cpid = msg;
|
|
// printf("child pid=%d\n", cpid);
|
|
// fflush(stdout);
|
|
crc = waitpid(-1, &st, WUNTRACED);
|
|
if (crc != cpid || !WIFSTOPPED(st)) {
|
|
printf("BAD4 rc=%d st=%08x\n", crc, st);
|
|
exit(1);
|
|
}
|
|
// printf("wait(2) rc=%d st=%08x\n", crc, st);
|
|
// fflush(stdout);
|
|
printf("#961-7 PTRACE_EVENT_FORK NG\n");
|
|
ng++;
|
|
forkev = 1;
|
|
procs++;
|
|
_ptrace(PTRACE_SYSCALL, cpid, NULL, (void *)0);
|
|
continue;
|
|
}
|
|
if (w == PTRACE_EVENT_EXEC) {
|
|
printf("wait(1) rc=%d st=%08x\n", rc, st);
|
|
printf("%d: exec event\n", rc);
|
|
fflush(stdout);
|
|
continue;
|
|
}
|
|
if (w == PTRACE_EVENT_EXIT) {
|
|
printf("wait(1) rc=%d st=%08x\n", rc, st);
|
|
printf("%d: exit event\n", rc);
|
|
fflush(stdout);
|
|
continue;
|
|
}
|
|
if (WSTOPSIG(st) & 0x80) { // syscall
|
|
int num;
|
|
get_syscall_args(rc, &args);
|
|
num = get_syscall_number(&args);
|
|
ret = get_syscall_return(&args);
|
|
if (num == SYS_clone) {
|
|
c++;
|
|
if (ret == -ENOSYS) {
|
|
ok++;
|
|
printf("#961-6 fork in OK\n");
|
|
}
|
|
else if (ret == 0) {
|
|
childfork = 1;
|
|
ng++;
|
|
printf("#961-9 fork child side called NG\n");
|
|
}
|
|
else {
|
|
printf("#961-8 fork out OK\n");
|
|
ok++;
|
|
}
|
|
}
|
|
}
|
|
else { // signal
|
|
// printf("wait(1) rc=%d st=%08x\n", rc, st);
|
|
sig = WSTOPSIG(st) & 0x7f;
|
|
if (sig == SIGCHLD) {
|
|
printf("#961-10 tracee receive SIGCHLD OK %d\n", rc);
|
|
fflush(stdout);
|
|
ok++;
|
|
}
|
|
else if (sig == SIGSTOP) {
|
|
// printf("#961-6 tracee receive SIGSTOP OK %d\n", rc);
|
|
// fflush(stdout);
|
|
// ok++;
|
|
}
|
|
else {
|
|
ng++;
|
|
printf("$961-10 tracee receive signal %d\n", sig);
|
|
fflush(stdout);
|
|
}
|
|
}
|
|
}
|
|
if (!childfork) {
|
|
ok++;
|
|
printf("#961-9 fork child side didnot be called OK\n");
|
|
}
|
|
if (!forkev) {
|
|
ok++;
|
|
printf("#961-7 didnot receive PTRACE_EVENT_FORK OK\n");
|
|
}
|
|
printf("#961 test terminated ok=%d ng=%d\n", ok, ng);
|
|
fflush(stdout);
|
|
exit(0);
|
|
}
|