add test programs for strace
This commit is contained in:
Tomoki Shirasawa
280
test/strace/issue/960.c
Normal file
280
test/strace/issue/960.c
Normal file
@ -0,0 +1,280 @@
|
||||
#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 <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;
|
||||
int w;
|
||||
syscall_args args;
|
||||
unsigned long sig = 0;
|
||||
char name[64];
|
||||
char ch = ' ';
|
||||
int ok = 0;
|
||||
int ng = 0;
|
||||
int procs = 0;
|
||||
int c = 0;
|
||||
|
||||
init_syscalls();
|
||||
printf("#960 start\n");
|
||||
fflush(stdout);
|
||||
pid = fork();
|
||||
if(pid == 0){
|
||||
_ptrace(PTRACE_TRACEME, 0, NULL, NULL);
|
||||
kill(getpid(), SIGINT);
|
||||
pid = fork();
|
||||
if (pid == 0) {
|
||||
exit(16);
|
||||
}
|
||||
rc = wait(&st);
|
||||
exit(64);
|
||||
}
|
||||
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 (WIFEXITED(st) || WIFSIGNALED(st)) {
|
||||
procs--;
|
||||
if (rc == pid)
|
||||
pid = -1;
|
||||
else if (rc == cpid)
|
||||
cpid = -1;
|
||||
rc = -1;
|
||||
if (!procs) {
|
||||
break;
|
||||
}
|
||||
continue;
|
||||
}
|
||||
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;
|
||||
_ptrace(PTRACE_GETEVENTMSG, pid, NULL, &msg);
|
||||
cpid = msg;
|
||||
crc = waitpid(-1, &st, WUNTRACED);
|
||||
if (crc != cpid || !WIFSTOPPED(st)) {
|
||||
printf("BAD4 rc=%d st=%08x\n", crc, st);
|
||||
exit(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;
|
||||
long ret;
|
||||
get_syscall_args(rc, &args);
|
||||
num = get_syscall_number(&args);
|
||||
ret = get_syscall_return(&args);
|
||||
if (num == SYS_wait4 &&
|
||||
ret == -ENOSYS) {
|
||||
c++;
|
||||
printf("#960-1 wait in OK\n");
|
||||
ok++;
|
||||
}
|
||||
else if (num == SYS_wait4 &&
|
||||
ret != -ENOSYS) {
|
||||
c++;
|
||||
if (c == 2) {
|
||||
printf("#960-2 wait out OK\n");
|
||||
ok++;
|
||||
}
|
||||
else {
|
||||
printf("#960-2 wait out NG\n");
|
||||
ng++;
|
||||
}
|
||||
}
|
||||
}
|
||||
else { // signal
|
||||
sig = WSTOPSIG(st) & 0x7f;
|
||||
if (sig == SIGCHLD) {
|
||||
c++;
|
||||
if (c == 3) {
|
||||
printf("#960-3 SIGCHLD OK\n");
|
||||
ok++;
|
||||
}
|
||||
else {
|
||||
printf("#960-3 SIGCHLD NG\n");
|
||||
ng++;
|
||||
}
|
||||
}
|
||||
else {
|
||||
printf("%d: sig=%d\n", rc, sig);
|
||||
fflush(stdout);
|
||||
ng++;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
printf("#960 terminated ok=%d ng=%d\n", ok, ng);
|
||||
fflush(stdout);
|
||||
exit(0);
|
||||
}
|
||||
Reference in New Issue
Block a user