ptrace.c 6.3 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12
/*
 * Server-side ptrace support
 *
 * Copyright (C) 1999 Alexandre Julliard
 */

#include "config.h"

#include <assert.h>
#include <errno.h>
#include <stdio.h>
#include <signal.h>
Alexandre Julliard's avatar
Alexandre Julliard committed
13
#include <sys/types.h>
14 15 16
#ifdef HAVE_SYS_PTRACE_H
# include <sys/ptrace.h>
#endif
17
#ifdef HAVE_SYS_WAIT_H
18
# include <sys/wait.h>
19 20 21 22 23 24 25 26 27 28
#endif
#include <unistd.h>

#include "process.h"
#include "thread.h"


#ifndef PTRACE_CONT
#define PTRACE_CONT PT_CONTINUE
#endif
29 30 31
#ifndef PTRACE_SINGLESTEP
#define PTRACE_SINGLESTEP PT_STEP
#endif
32 33 34 35 36 37 38 39 40 41 42 43 44
#ifndef PTRACE_ATTACH
#define PTRACE_ATTACH PT_ATTACH
#endif
#ifndef PTRACE_DETACH
#define PTRACE_DETACH PT_DETACH
#endif
#ifndef PTRACE_PEEKDATA
#define PTRACE_PEEKDATA PT_READ_D
#endif
#ifndef PTRACE_POKEDATA
#define PTRACE_POKEDATA PT_WRITE_D
#endif

45
#ifndef HAVE_SYS_PTRACE_H
46 47 48 49 50
#define PT_CONTINUE 0
#define PT_ATTACH   1
#define PT_DETACH   2
#define PT_READ_D   3
#define PT_WRITE_D  4
51
#define PT_STEP     5
52 53
inline static int ptrace(int req, ...) { errno = EPERM; return -1; /*FAIL*/ }
#endif  /* HAVE_SYS_PTRACE_H */
54

55
static const int use_ptrace = 1;  /* set to 0 to disable ptrace */
56

57 58
/* handle a status returned by wait4 */
static int handle_child_status( struct thread *thread, int pid, int status )
59 60 61 62
{
    if (WIFSTOPPED(status))
    {
        int sig = WSTOPSIG(status);
63 64
        if (debug_level && thread)
            fprintf( stderr, "%08x: *signal* signal=%d\n", (unsigned int)thread, sig );
65 66 67
        switch(sig)
        {
        case SIGSTOP:  /* continue at once if not suspended */
68 69
            if (thread && (thread->process->suspend + thread->suspend)) break;
            /* fall through */
70
        default:  /* ignore other signals for now */
71 72 73 74
            if (thread && get_thread_single_step( thread ))
                ptrace( PTRACE_SINGLESTEP, pid, (caddr_t)1, sig );
            else
                ptrace( PTRACE_CONT, pid, (caddr_t)1, sig );
75 76
            break;
        }
77
        return sig;
78
    }
79
    if (thread && (WIFSIGNALED(status) || WIFEXITED(status)))
80
    {
81 82
        thread->attached = 0;
        thread->unix_pid = 0;
83
        if (debug_level)
84 85 86 87 88 89 90 91
        {
            if (WIFSIGNALED(status))
                fprintf( stderr, "%08x: *exited* signal=%d\n",
                         (unsigned int)thread, WTERMSIG(status) );
            else
                fprintf( stderr, "%08x: *exited* status=%d\n",
                         (unsigned int)thread, WEXITSTATUS(status) );
        }
92
    }
93 94 95 96 97 98 99 100 101
    return 0;
}

/* handle a SIGCHLD signal */
void sigchld_handler()
{
    int pid, status;

    for (;;)
102
    {
103 104 105
        if (!(pid = wait4( -1, &status, WUNTRACED | WNOHANG, NULL ))) break;
        if (pid != -1) handle_child_status( get_thread_from_pid(pid), pid, status );
        else break;
106
    }
107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122
}

/* wait for a ptraced child to get a certain signal */
void wait4_thread( struct thread *thread, int signal )
{
    int res, status;

    do
    {
        if ((res = wait4( thread->unix_pid, &status, WUNTRACED, NULL )) == -1)
        {
            perror( "wait4" );
            return;
        }
        res = handle_child_status( thread, res, status );
    } while (res && res != signal);
123 124 125 126 127 128
}

/* attach to a Unix thread */
static int attach_thread( struct thread *thread )
{
    /* this may fail if the client is already being debugged */
129 130 131 132 133 134
    if (!use_ptrace) return 0;
    if (ptrace( PTRACE_ATTACH, thread->unix_pid, 0, 0 ) == -1)
    {
        if (errno == ESRCH) thread->unix_pid = 0;  /* process got killed */
        return 0;
    }
135
    if (debug_level) fprintf( stderr, "%08x: *attached*\n", (unsigned int)thread );
136 137 138 139 140 141
    thread->attached = 1;
    wait4_thread( thread, SIGSTOP );
    return 1;
}

/* detach from a Unix thread and kill it */
142
void detach_thread( struct thread *thread, int sig )
143 144 145 146
{
    if (!thread->unix_pid) return;
    if (thread->attached)
    {
147
        /* make sure it is stopped */
148
        suspend_thread( thread, 0 );
149
        if (sig) kill( thread->unix_pid, sig );
150
        if (debug_level) fprintf( stderr, "%08x: *detached*\n", (unsigned int)thread );
151
        ptrace( PTRACE_DETACH, thread->unix_pid, (caddr_t)1, sig );
152
        thread->suspend = 0;  /* detach makes it continue */
153 154
        thread->attached = 0;
    }
155 156
    else
    {
157
        if (sig) kill( thread->unix_pid, sig );
158 159
        if (thread->suspend + thread->process->suspend) continue_thread( thread );
    }
160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179
}

/* stop a thread (at the Unix level) */
void stop_thread( struct thread *thread )
{
    /* can't stop a thread while initialisation is in progress */
    if (!thread->unix_pid || thread->process->init_event) return;
    /* first try to attach to it */
    if (!thread->attached)
        if (attach_thread( thread )) return;  /* this will have stopped it */
    /* attached already, or attach failed -> send a signal */
    kill( thread->unix_pid, SIGSTOP );
    if (thread->attached) wait4_thread( thread, SIGSTOP );
}

/* make a thread continue (at the Unix level) */
void continue_thread( struct thread *thread )
{
    if (!thread->unix_pid) return;
    if (!thread->attached) kill( thread->unix_pid, SIGCONT );
180 181
    else ptrace( get_thread_single_step(thread) ? PTRACE_SINGLESTEP : PTRACE_CONT,
                 thread->unix_pid, (caddr_t)1, SIGSTOP );
182 183
}

184 185 186 187 188 189 190 191 192
/* suspend a thread to allow using ptrace on it */
/* you must do a resume_thread when finished with the thread */
int suspend_for_ptrace( struct thread *thread )
{
    if (thread->attached)
    {
        suspend_thread( thread, 0 );
        return 1;
    }
193 194 195 196 197 198
    /* can't stop a thread while initialisation is in progress */
    if (!thread->unix_pid || thread->process->init_event) goto error;
    thread->suspend++;
    if (attach_thread( thread )) return 1;
    thread->suspend--;
 error:
199 200 201 202
    set_error( STATUS_ACCESS_DENIED );
    return 0;
}

203 204 205
/* read an int from a thread address space */
int read_thread_int( struct thread *thread, const int *addr, int *data )
{
206 207
    *data = ptrace( PTRACE_PEEKDATA, thread->unix_pid, (caddr_t)addr, 0 );
    if ( *data == -1 && errno)
208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223
    {
        file_set_error();
        return -1;
    }
    return 0;
}

/* write an int to a thread address space */
int write_thread_int( struct thread *thread, int *addr, int data, unsigned int mask )
{
    int res;
    if (mask != ~0)
    {
        if (read_thread_int( thread, addr, &res ) == -1) return -1;
        data = (data & mask) | (res & ~mask);
    }
224 225
    if ((res = ptrace( PTRACE_POKEDATA, thread->unix_pid, (caddr_t)addr, data )) == -1)
        file_set_error();
226 227
    return res;
}