Ignore:
Timestamp:
Jul 2, 2008 4:35:26 PM (6 years ago)
Author:
sam
Message:
  • Fork the grabbed process and exit the parent. Unfortunately for some reason it still survives as a zombie, there must be something wrong with the setsid/setpgid mechanism we use (maybe setpgrp is needed?).
File:
1 edited

Legend:

Unmodified
Added
Removed

  • neercs/trunk/src/mytrace.c

    r2514 r2516  
    3939static int memcpy_into_target(struct mytrace *t, 
    4040                              long dest, char const *src, size_t n); 
    41 static long remote_syscall(pid_t pid, long call, 
     41static long remote_syscall(struct mytrace *t, long call, 
    4242                           long arg1, long arg2, long arg3); 
    4343#   if defined DEBUG 
    … …  
    5353#define SYSCALL_X86   0x80cd  /* CD 80 = int $0x80 */ 
    5454#define SYSCALL_AMD64 0x050fL /* 0F 05 = syscall */ 
    55  
    56 #define MYCALL_OPEN     0 
    57 #define MYCALL_CLOSE    1 
    58 #define MYCALL_WRITE    2 
    59 #define MYCALL_DUP2     3 
    60 #define MYCALL_SETPGID  4 
    61 #define MYCALL_SETSID   5 
    62 #define MYCALL_KILL     6 
    6355 
    6456#if defined __x86_64__ 
    … …  
    8476#endif 
    8577 
     78#define MYCALL_OPEN     0 
     79#define MYCALL_CLOSE    1 
     80#define MYCALL_WRITE    2 
     81#define MYCALL_DUP2     3 
     82#define MYCALL_SETPGID  4 
     83#define MYCALL_SETSID   5 
     84#define MYCALL_KILL     6 
     85#define MYCALL_FORK     7 
     86#define MYCALL_EXIT     8 
     87 
    8688#if defined __x86_64__ 
     89/* from unistd_32.h on an amd64 system */ 
     90int syscalls32[] = { 5, 6, 4, 63, 57, 66, 37, 2, 1 }; 
     91int syscalls64[] = 
     92#else 
    8793int syscalls32[] = 
    88     { 5, 6, 4, 63, 57, 66, 37 }; /* from unistd_32.h on an amd64 system */ 
    89 int syscalls64[] = 
    90 #else 
    91 int syscalls32[] = 
    9294#endif 
    9395    { SYS_open, SYS_close, SYS_write, SYS_dup2, SYS_setpgid, SYS_setsid, 
    94       SYS_kill }; 
     96      SYS_kill, SYS_fork, SYS_exit }; 
     97 
     98char const *syscallnames[] = 
     99    { "open", "close", "write", "dup2", "setpgid", "setsid", "kill", "fork", 
     100      "exit" }; 
    95101 
    96102struct mytrace 
    97103{ 
    98     pid_t pid; 
     104    pid_t pid, child; 
    99105}; 
    100106 
    … …  
    103109#if defined USE_GRAB 
    104110    struct mytrace *t; 
     111    int status; 
    105112 
    106113    if(ptrace(PTRACE_ATTACH, pid, 0, 0) < 0) 
    … …  
    109116        return NULL; 
    110117    } 
    111     waitpid(pid, NULL, 0); 
     118    if(waitpid(pid, &status, 0) < 0) 
     119    { 
     120        perror("waitpid"); 
     121        return NULL; 
     122    } 
     123    if(!WIFSTOPPED(status)) 
     124    { 
     125        fprintf(stderr, "traced process was not stopped\n"); 
     126        ptrace(PTRACE_DETACH, pid, 0, 0); 
     127        return NULL; 
     128    } 
    112129 
    113130    t = malloc(sizeof(struct mytrace)); 
    114131    t->pid = pid; 
     132    t->child = 0; 
    115133 
    116134    return t; 
     135#else 
     136    errno = ENOSYS; 
     137    return NULL; 
     138#endif 
     139} 
     140 
     141struct mytrace* mytrace_fork(struct mytrace *t) 
     142{ 
     143#if defined USE_GRAB 
     144    struct mytrace *child; 
     145 
     146    ptrace(PTRACE_SETOPTIONS, t->pid, NULL, PTRACE_O_TRACEFORK); 
     147    remote_syscall(t, MYCALL_FORK, 0, 0, 0); 
     148    waitpid(t->child, NULL, 0); 
     149 
     150    child = malloc(sizeof(struct mytrace)); 
     151    child->pid = t->child; 
     152    child->child = 0; 
     153 
     154    return child; 
    117155#else 
    118156    errno = ENOSYS; 
    … …  
    128166 
    129167    return 0; 
     168#else 
     169    errno = ENOSYS; 
     170    return -1; 
     171#endif 
     172} 
     173 
     174long mytrace_getpid(struct mytrace *t) 
     175{ 
     176#if defined USE_GRAB 
     177    return t->pid; 
    130178#else 
    131179    errno = ENOSYS; 
    … …  
    154202    memcpy_into_target(t, regs.RSP, path, size); 
    155203 
    156     ret = remote_syscall(t->pid, MYCALL_OPEN, regs.RSP, O_RDWR, 0755); 
     204    ret = remote_syscall(t, MYCALL_OPEN, regs.RSP, O_RDWR, 0755); 
    157205 
    158206    /* Restore the data */ 
    … …  
    175223{ 
    176224#if defined USE_GRAB 
    177     return remote_syscall(t->pid, MYCALL_CLOSE, fd, 0, 0); 
     225    return remote_syscall(t, MYCALL_CLOSE, fd, 0, 0); 
    178226#else 
    179227    errno = ENOSYS; 
    … …  
    203251    memcpy_into_target(t, regs.RSP, data, len); 
    204252 
    205     ret = remote_syscall(t->pid, MYCALL_WRITE, fd, regs.RSP, len); 
     253    ret = remote_syscall(t, MYCALL_WRITE, fd, regs.RSP, len); 
    206254 
    207255    /* Restore the data */ 
    … …  
    224272{ 
    225273#if defined USE_GRAB 
    226     return remote_syscall(t->pid, MYCALL_DUP2, oldfd, newfd, 0); 
     274    return remote_syscall(t, MYCALL_DUP2, oldfd, newfd, 0); 
    227275#else 
    228276    errno = ENOSYS; 
    … …  
    234282{ 
    235283#if defined USE_GRAB 
    236     return remote_syscall(t->pid, MYCALL_SETPGID, pid, pgid, 0); 
     284    return remote_syscall(t, MYCALL_SETPGID, pid, pgid, 0); 
    237285#else 
    238286    errno = ENOSYS; 
    … …  
    244292{ 
    245293#if defined USE_GRAB 
    246     return remote_syscall(t->pid, MYCALL_SETSID, 0, 0, 0); 
     294    return remote_syscall(t, MYCALL_SETSID, 0, 0, 0); 
    247295#else 
    248296    errno = ENOSYS; 
    … …  
    254302{ 
    255303#if defined USE_GRAB 
    256     return remote_syscall(t->pid, MYCALL_KILL, pid, sig, 0); 
     304    return remote_syscall(t, MYCALL_KILL, pid, sig, 0); 
     305#else 
     306    errno = ENOSYS; 
     307    return -1; 
     308#endif 
     309} 
     310 
     311int mytrace_exit(struct mytrace *t, int status) 
     312{ 
     313#if defined USE_GRAB 
     314    ptrace(PTRACE_SETOPTIONS, t->pid, NULL, PTRACE_O_TRACEEXIT); 
     315    return remote_syscall(t, MYCALL_EXIT, status, 0, 0); 
    257316#else 
    258317    errno = ENOSYS; 
    … …  
    333392} 
    334393 
    335 static long remote_syscall(pid_t pid, long call, 
    336                             long arg1, long arg2, long arg3) 
     394static long remote_syscall(struct mytrace *t, long call, 
     395                           long arg1, long arg2, long arg3) 
    337396{ 
    338397    /* Method for remote syscall: 
    … …  
    347406    int bits; 
    348407 
    349 print_registers(pid); 
     408    if(call < 0 || call >= (long)(sizeof(syscallnames)/sizeof(*syscallnames))) 
     409    { 
     410        fprintf(stderr, "unknown remote syscall %li\n", call); 
     411        return -1; 
     412    } 
     413 
     414    debug("remote syscall %s(%lu, %lu, %lu)", 
     415          syscallnames[call], arg1, arg2, arg3); 
     416 
     417    print_registers(t->pid); 
     418 
    350419#if defined __x86_64__ 
    351420    bits = 64; 
    … …  
    356425    for(;;) 
    357426    { 
    358         if(ptrace(PTRACE_GETREGS, pid, NULL, &oldregs) < 0) 
     427        if(ptrace(PTRACE_GETREGS, t->pid, NULL, &oldregs) < 0) 
    359428        { 
    360429            fprintf(stderr, "PTRACE_GETREGS failed\n"); 
    … …  
    362431        } 
    363432 
    364         oinst = ptrace(PTRACE_PEEKTEXT, pid, oldregs.RIP - 2, 0) & 0xffff; 
     433        oinst = ptrace(PTRACE_PEEKTEXT, t->pid, oldregs.RIP - 2, 0) & 0xffff; 
    365434 
    366435#if defined __x86_64__ 
    … …  
    377446#endif 
    378447 
    379         if(ptrace(PTRACE_SYSCALL, pid, NULL, 0) < 0) 
     448        if(ptrace(PTRACE_SYSCALL, t->pid, NULL, 0) < 0) 
    380449        { 
    381450            perror("ptrace_syscall (1)"); 
    382451            return -1; 
    383452        } 
    384         waitpid(pid, NULL, 0); 
    385  
    386         if(ptrace(PTRACE_SYSCALL, pid, NULL, 0) < 0) 
     453        waitpid(t->pid, NULL, 0); 
     454 
     455        if(ptrace(PTRACE_SYSCALL, t->pid, NULL, 0) < 0) 
    387456        { 
    388457            perror("ptrace_syscall (2)"); 
    389458            return -1; 
    390459        } 
    391         waitpid(pid, NULL, 0); 
    392     } 
    393  
    394     print_registers(pid); 
     460        waitpid(t->pid, NULL, 0); 
     461    } 
     462 
     463    print_registers(t->pid); 
    395464 
    396465    regs = oldregs; 
    … …  
    413482    } 
    414483 
    415     if(ptrace(PTRACE_SETREGS, pid, NULL, &regs) < 0) 
     484    if(ptrace(PTRACE_SETREGS, t->pid, NULL, &regs) < 0) 
    416485    { 
    417486        fprintf(stderr, "PTRACE_SETREGS failed\n"); 
    … …  
    419488    } 
    420489 
    421     print_registers(pid); 
    422  
    423     if(ptrace(PTRACE_SINGLESTEP, pid, NULL, NULL) < 0) 
    424     { 
    425         fprintf(stderr, "PTRACE_SINGLESTEP failed\n"); 
    426         return -1; 
    427     } 
    428     waitpid(pid, NULL, 0); 
    429  
    430     print_registers(pid); 
    431  
    432     if(ptrace(PTRACE_GETREGS, pid, NULL, &regs) < 0) 
     490    for(;;) 
     491    { 
     492        int status; 
     493 
     494        print_registers(t->pid); 
     495 
     496        if(ptrace(PTRACE_SINGLESTEP, t->pid, NULL, NULL) < 0) 
     497        { 
     498            fprintf(stderr, "PTRACE_SINGLESTEP failed\n"); 
     499            return -1; 
     500        } 
     501        waitpid(t->pid, &status, 0); 
     502 
     503        if(WIFEXITED(status)) 
     504            return 0; 
     505 
     506        if(!WIFSTOPPED(status) || WSTOPSIG(status) != SIGTRAP) 
     507            continue; 
     508 
     509        /* Fuck Linux: there is no macro for this */ 
     510        switch((status >> 16) & 0xffff) 
     511        { 
     512        case PTRACE_EVENT_FORK: 
     513            if(ptrace(PTRACE_GETEVENTMSG, t->pid, 0, &t->child) < 0) 
     514            { 
     515                fprintf(stderr, "PTRACE_GETEVENTMSG failed\n"); 
     516                return -1; 
     517            } 
     518            continue; 
     519        case PTRACE_EVENT_EXIT: 
     520            /* The process is about to exit, don't do anything else */ 
     521            return 0; 
     522        } 
     523 
     524        break; 
     525    } 
     526 
     527    print_registers(t->pid); 
     528 
     529    if(ptrace(PTRACE_GETREGS, t->pid, NULL, &regs) < 0) 
    433530    { 
    434531        fprintf(stderr, "PTRACE_GETREGS failed\n"); 
    … …  
    436533    } 
    437534 
    438     if(ptrace(PTRACE_SETREGS, pid, NULL, &oldregs) < 0) 
     535    if(ptrace(PTRACE_SETREGS, t->pid, NULL, &oldregs) < 0) 
    439536    { 
    440537        fprintf(stderr, "PTRACE_SETREGS failed\n"); 
    441538        return -1; 
    442539    } 
    443     print_registers(pid); 
    444  
    445     debug("syscall %ld returned %ld", call, regs.RAX); 
     540    print_registers(t->pid); 
     541 
     542    debug("syscall %s returned %ld", syscallnames[call], regs.RAX); 
    446543 
    447544    if((long)regs.RAX < 0) 
Note: See TracChangeset for help on using the changeset viewer.