From 5a12dd17ae2447a8c179e0d37db8a65b1c924155 Mon Sep 17 00:00:00 2001 From: Alexandre Julliard Date: Fri, 16 Sep 2005 18:59:23 +0000 Subject: [PATCH] Preliminary version of the context support for x86-64. --- server/Makefile.in | 1 + server/context_x86_64.c | 347 ++++++++++++++++++++++++++++++++++++++++ 2 files changed, 348 insertions(+) create mode 100644 server/context_x86_64.c diff --git a/server/Makefile.in b/server/Makefile.in index c3fc30ec567..e80e767880e 100644 --- a/server/Makefile.in +++ b/server/Makefile.in @@ -15,6 +15,7 @@ C_SRCS = \ context_i386.c \ context_powerpc.c \ context_sparc.c \ + context_x86_64.c \ debugger.c \ event.c \ fd.c \ diff --git a/server/context_x86_64.c b/server/context_x86_64.c new file mode 100644 index 00000000000..15d41352617 --- /dev/null +++ b/server/context_x86_64.c @@ -0,0 +1,347 @@ +/* + * x86-64 register context support + * + * Copyright (C) 1999, 2005 Alexandre Julliard + * + * This library is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License as published by the Free Software Foundation; either + * version 2.1 of the License, or (at your option) any later version. + * + * This library is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with this library; if not, write to the Free Software + * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA + */ + +#include "config.h" + +#ifdef __x86_64__ + +#include +#include +#include +#include +#ifdef HAVE_SYS_PTRACE_H +# include +#endif +#ifdef HAVE_SYS_PARAM_H +# include +#endif + +#define NONAMELESSUNION +#include "windef.h" +#include "winbase.h" + +#include "file.h" +#include "thread.h" +#include "request.h" + +#ifdef __linux__ + +#ifdef HAVE_SYS_USER_H +# include +#endif + +/* debug register offset in struct user */ +#define DR_OFFSET(dr) ((unsigned long)((((struct user *)0)->u_debugreg) + (dr))) + +/* retrieve a debug register */ +static inline int get_debug_reg( int pid, int num, DWORD64 *data ) +{ + int res; + errno = 0; + res = ptrace( PTRACE_PEEKUSER, pid, DR_OFFSET(num), 0 ); + if ((res == -1) && errno) + { + file_set_error(); + return -1; + } + *data = res; + return 0; +} + +/* retrieve a thread context */ +static void get_thread_context( struct thread *thread, unsigned int flags, CONTEXT *context ) +{ + int pid = get_ptrace_pid(thread); + if (flags & CONTEXT_FULL) + { + struct user_regs_struct regs; + if (ptrace( PTRACE_GETREGS, pid, 0, ®s ) == -1) goto error; + if (flags & CONTEXT_INTEGER) + { + context->Rax = regs.rax; + context->Rbx = regs.rbx; + context->Rcx = regs.rcx; + context->Rdx = regs.rdx; + context->Rsi = regs.rsi; + context->Rdi = regs.rdi; + context->R8 = regs.r8; + context->R9 = regs.r9; + context->R10 = regs.r10; + context->R11 = regs.r11; + context->R12 = regs.r12; + context->R13 = regs.r13; + context->R14 = regs.r14; + context->R15 = regs.r15; + } + if (flags & CONTEXT_CONTROL) + { + context->Rbp = regs.rbp; + context->Rsp = regs.rsp; + context->Rip = regs.rip; + context->SegCs = regs.cs; + context->SegSs = regs.ss; + context->EFlags = regs.eflags; + } + if (flags & CONTEXT_SEGMENTS) + { + context->SegDs = regs.ds; + context->SegEs = regs.es; + context->SegFs = regs.fs; + context->SegGs = regs.gs; + } + } + if (flags & CONTEXT_DEBUG_REGISTERS) + { + if (get_debug_reg( pid, 0, &context->Dr0 ) == -1) goto error; + if (get_debug_reg( pid, 1, &context->Dr1 ) == -1) goto error; + if (get_debug_reg( pid, 2, &context->Dr2 ) == -1) goto error; + if (get_debug_reg( pid, 3, &context->Dr3 ) == -1) goto error; + if (get_debug_reg( pid, 6, &context->Dr6 ) == -1) goto error; + if (get_debug_reg( pid, 7, &context->Dr7 ) == -1) goto error; + } + if (flags & CONTEXT_FLOATING_POINT) + { + /* we can use context->FloatSave directly as it is using the */ + /* correct structure (the same as fsave/frstor) */ + if (ptrace( PTRACE_GETFPREGS, pid, 0, &context->u.FltSave ) == -1) goto error; + } + return; + error: + file_set_error(); +} + + +/* set a thread context */ +static void set_thread_context( struct thread *thread, unsigned int flags, const CONTEXT *context ) +{ + int pid = get_ptrace_pid(thread); + if (flags & CONTEXT_FULL) + { + struct user_regs_struct regs; + + /* need to preserve some registers (at a minimum orig_eax must always be preserved) */ + if (ptrace( PTRACE_GETREGS, pid, 0, ®s ) == -1) goto error; + + if (flags & CONTEXT_INTEGER) + { + regs.rax = context->Rax; + regs.rbx = context->Rbx; + regs.rcx = context->Rcx; + regs.rdx = context->Rdx; + regs.rsi = context->Rsi; + regs.rdi = context->Rdi; + regs.r8 = context->R8; + regs.r9 = context->R9; + regs.r10 = context->R10; + regs.r11 = context->R11; + regs.r12 = context->R12; + regs.r13 = context->R13; + regs.r14 = context->R14; + regs.r15 = context->R15; + } + if (flags & CONTEXT_CONTROL) + { + regs.rbp = context->Rbp; + regs.rip = context->Rip; + regs.rsp = context->Rsp; + regs.cs = context->SegCs; + regs.ss = context->SegSs; + regs.eflags = context->EFlags; + } + if (flags & CONTEXT_SEGMENTS) + { + regs.ds = context->SegDs; + regs.es = context->SegEs; + regs.fs = context->SegFs; + regs.gs = context->SegGs; + } + if (ptrace( PTRACE_SETREGS, pid, 0, ®s ) == -1) goto error; + } + if (flags & CONTEXT_DEBUG_REGISTERS) + { + if (ptrace( PTRACE_POKEUSER, pid, DR_OFFSET(0), context->Dr0 ) == -1) goto error; + if (ptrace( PTRACE_POKEUSER, pid, DR_OFFSET(1), context->Dr1 ) == -1) goto error; + if (ptrace( PTRACE_POKEUSER, pid, DR_OFFSET(2), context->Dr2 ) == -1) goto error; + if (ptrace( PTRACE_POKEUSER, pid, DR_OFFSET(3), context->Dr3 ) == -1) goto error; + if (ptrace( PTRACE_POKEUSER, pid, DR_OFFSET(6), context->Dr6 ) == -1) goto error; + if (ptrace( PTRACE_POKEUSER, pid, DR_OFFSET(7), context->Dr7 ) == -1) goto error; + } + if (flags & CONTEXT_FLOATING_POINT) + { + /* we can use context->FloatSave directly as it is using the */ + /* correct structure (the same as fsave/frstor) */ + if (ptrace( PTRACE_SETFPREGS, pid, 0, &context->u.FltSave ) == -1) goto error; + } + return; + error: + file_set_error(); +} + +#else /* linux */ +#error You must implement get/set_thread_context for your platform +#endif /* linux */ + + +/* copy a context structure according to the flags */ +static void copy_context( CONTEXT *to, const CONTEXT *from, int flags ) +{ + if (flags & CONTEXT_CONTROL) + { + to->Rbp = from->Rbp; + to->Rip = from->Rip; + to->Rsp = from->Rsp; + to->SegCs = from->SegCs; + to->SegSs = from->SegSs; + to->EFlags = from->EFlags; + to->MxCsr = from->MxCsr; + } + if (flags & CONTEXT_INTEGER) + { + to->Rax = from->Rax; + to->Rcx = from->Rcx; + to->Rdx = from->Rdx; + to->Rbx = from->Rbx; + to->Rsi = from->Rsi; + to->Rdi = from->Rdi; + to->R8 = from->R8; + to->R9 = from->R9; + to->R10 = from->R10; + to->R11 = from->R11; + to->R12 = from->R12; + to->R13 = from->R13; + to->R14 = from->R14; + to->R15 = from->R15; + } + if (flags & CONTEXT_SEGMENTS) + { + to->SegDs = from->SegDs; + to->SegEs = from->SegEs; + to->SegFs = from->SegFs; + to->SegGs = from->SegGs; + } + if (flags & CONTEXT_FLOATING_POINT) + { + to->u.FltSave = from->u.FltSave; + } + /* we don't bother copying the debug registers, since they */ + /* always need to be accessed by ptrace anyway */ +} + +/* retrieve the current instruction pointer of a thread */ +void *get_thread_ip( struct thread *thread ) +{ + CONTEXT context; + context.Rip = 0; + if (suspend_for_ptrace( thread )) + { + get_thread_context( thread, CONTEXT_CONTROL, &context ); + resume_after_ptrace( thread ); + } + return (void *)context.Rip; +} + +/* determine if we should continue the thread in single-step mode */ +int get_thread_single_step( struct thread *thread ) +{ + CONTEXT context; + if (thread->context) return 0; /* don't single-step inside exception event */ + get_thread_context( thread, CONTEXT_CONTROL, &context ); + return (context.EFlags & 0x100) != 0; +} + +/* send a signal to a specific thread */ +int tkill( int pid, int sig ) +{ +#ifdef __linux__ + int ret; + __asm__( "syscall" : "=a" (ret) + : "0" (200) /*SYS_tkill*/, "D" (pid), "S" (sig) ); + if (ret >= 0) return ret; + errno = -ret; + return -1; +#else + errno = ENOSYS; + return -1; +#endif +} + +/* retrieve the current context of a thread */ +DECL_HANDLER(get_thread_context) +{ + struct thread *thread; + void *data; + int flags = req->flags & ~CONTEXT_AMD64; /* get rid of CPU id */ + + if (get_reply_max_size() < sizeof(CONTEXT)) + { + set_error( STATUS_INVALID_PARAMETER ); + return; + } + if (!(thread = get_thread_from_handle( req->handle, THREAD_GET_CONTEXT ))) return; + + if ((data = set_reply_data_size( sizeof(CONTEXT) ))) + { + /* copy incoming context into reply */ + memset( data, 0, sizeof(CONTEXT) ); + memcpy( data, get_req_data(), min( get_req_data_size(), sizeof(CONTEXT) )); + + if (thread->context) /* thread is inside an exception event */ + { + copy_context( data, thread->context, flags ); + flags &= CONTEXT_DEBUG_REGISTERS; + } + if (flags && suspend_for_ptrace( thread )) + { + get_thread_context( thread, flags, data ); + resume_after_ptrace( thread ); + } + } + release_object( thread ); +} + + +/* set the current context of a thread */ +DECL_HANDLER(set_thread_context) +{ + struct thread *thread; + int flags = req->flags & ~CONTEXT_AMD64; /* get rid of CPU id */ + + if (get_req_data_size() < sizeof(CONTEXT)) + { + set_error( STATUS_INVALID_PARAMETER ); + return; + } + if ((thread = get_thread_from_handle( req->handle, THREAD_SET_CONTEXT ))) + { + if (thread->context) /* thread is inside an exception event */ + { + copy_context( thread->context, get_req_data(), flags ); + flags &= CONTEXT_DEBUG_REGISTERS; + } + if (flags && suspend_for_ptrace( thread )) + { + set_thread_context( thread, flags, get_req_data() ); + resume_after_ptrace( thread ); + } + release_object( thread ); + } +} + +#endif /* __x86_64__ */