server/context_x86_64.c - wine - Git at Google

 /*
  * 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 <assert.h>
 #include <errno.h>
 #include <stdarg.h>
 #include <unistd.h>
 #ifdef HAVE_SYS_PTRACE_H
 # include <sys/ptrace.h>
 #endif
 #ifdef HAVE_SYS_PARAM_H
 # include <sys/param.h>
 #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 <sys/user.h>
 #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, &regs ) == -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, &regs ) == -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, &regs ) == -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) )))
     {
         memset( data, 0, 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__ */
	/*
	* 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 <assert.h>
	#include <errno.h>
	#include <stdarg.h>
	#include <unistd.h>
	#ifdef HAVE_SYS_PTRACE_H
	# include <sys/ptrace.h>
	#endif
	#ifdef HAVE_SYS_PARAM_H
	# include <sys/param.h>
	#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 <sys/user.h>
	#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, &regs ) == -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, &regs ) == -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, &regs ) == -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) )))
	{
	memset( data, 0, 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__ */