2 * i386 register context support
4 * Copyright (C) 1999 Alexandre Julliard
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
32 #ifdef HAVE_SYS_PTRACE_H
33 # include <sys/ptrace.h>
35 #ifdef HAVE_SYS_PARAM_H
36 # include <sys/param.h>
46 #ifndef PTRACE_PEEKUSER
47 # ifdef PTRACE_PEEKUSR /* libc5 uses USR not USER */
48 # define PTRACE_PEEKUSER PTRACE_PEEKUSR
50 # define PTRACE_PEEKUSER PT_READ_U
54 #ifndef PTRACE_POKEUSER
55 # ifdef PTRACE_POKEUSR /* libc5 uses USR not USER */
56 # define PTRACE_POKEUSER PTRACE_POKEUSR
58 # define PTRACE_POKEUSER PT_WRITE_U
62 #ifndef PTRACE_GETREGS
63 #define PTRACE_GETREGS PT_GETREGS
65 #ifndef PTRACE_GETFPREGS
66 #define PTRACE_GETFPREGS PT_GETFPREGS
68 #ifndef PTRACE_SETREGS
69 #define PTRACE_SETREGS PT_SETREGS
71 #ifndef PTRACE_SETFPREGS
72 #define PTRACE_SETFPREGS PT_SETFPREGS
76 #define PTRACE_GETDBREGS PT_GETDBREGS
80 #define PTRACE_SETDBREGS PT_SETDBREGS
84 #ifdef HAVE_SYS_USER_H
85 # include <sys/user.h>
88 /* user_regs definitions from asm/user.h */
89 struct kernel_user_regs_struct
91 long ebx
, ecx
, edx
, esi
, edi
, ebp
, eax
;
92 unsigned short ds
, __ds
, es
, __es
;
93 unsigned short fs
, __fs
, gs
, __gs
;
95 unsigned short cs
, __cs
;
97 unsigned short ss
, __ss
;
100 /* debug register offset in struct user */
101 #define DR_OFFSET(dr) ((int)((((struct user *)0)->u_debugreg) + (dr)))
103 /* retrieve a debug register */
104 static inline int get_debug_reg( int pid
, int num
, DWORD
*data
)
108 res
= ptrace( PTRACE_PEEKUSER
, pid
, DR_OFFSET(num
), 0 );
109 if ((res
== -1) && errno
)
118 /* retrieve a thread context */
119 static void get_thread_context( struct thread
*thread
, unsigned int flags
, CONTEXT
*context
)
121 int pid
= get_ptrace_pid(thread
);
122 if (flags
& CONTEXT_FULL
)
124 struct kernel_user_regs_struct regs
;
125 if (ptrace( PTRACE_GETREGS
, pid
, 0, ®s
) == -1) goto error
;
126 if (flags
& CONTEXT_INTEGER
)
128 context
->Eax
= regs
.eax
;
129 context
->Ebx
= regs
.ebx
;
130 context
->Ecx
= regs
.ecx
;
131 context
->Edx
= regs
.edx
;
132 context
->Esi
= regs
.esi
;
133 context
->Edi
= regs
.edi
;
135 if (flags
& CONTEXT_CONTROL
)
137 context
->Ebp
= regs
.ebp
;
138 context
->Esp
= regs
.esp
;
139 context
->Eip
= regs
.eip
;
140 context
->SegCs
= regs
.cs
;
141 context
->SegSs
= regs
.ss
;
142 context
->EFlags
= regs
.eflags
;
144 if (flags
& CONTEXT_SEGMENTS
)
146 context
->SegDs
= regs
.ds
;
147 context
->SegEs
= regs
.es
;
148 context
->SegFs
= regs
.fs
;
149 context
->SegGs
= regs
.gs
;
152 if (flags
& CONTEXT_DEBUG_REGISTERS
)
154 if (get_debug_reg( pid
, 0, &context
->Dr0
) == -1) goto error
;
155 if (get_debug_reg( pid
, 1, &context
->Dr1
) == -1) goto error
;
156 if (get_debug_reg( pid
, 2, &context
->Dr2
) == -1) goto error
;
157 if (get_debug_reg( pid
, 3, &context
->Dr3
) == -1) goto error
;
158 if (get_debug_reg( pid
, 6, &context
->Dr6
) == -1) goto error
;
159 if (get_debug_reg( pid
, 7, &context
->Dr7
) == -1) goto error
;
161 if (flags
& CONTEXT_FLOATING_POINT
)
163 /* we can use context->FloatSave directly as it is using the */
164 /* correct structure (the same as fsave/frstor) */
165 if (ptrace( PTRACE_GETFPREGS
, pid
, 0, &context
->FloatSave
) == -1) goto error
;
166 context
->FloatSave
.Cr0NpxState
= 0; /* FIXME */
174 /* set a thread context */
175 static void set_thread_context( struct thread
*thread
, unsigned int flags
, const CONTEXT
*context
)
177 int pid
= get_ptrace_pid(thread
);
178 if (flags
& CONTEXT_FULL
)
180 struct kernel_user_regs_struct regs
;
182 /* need to preserve some registers (at a minimum orig_eax must always be preserved) */
183 if (ptrace( PTRACE_GETREGS
, pid
, 0, ®s
) == -1) goto error
;
185 if (flags
& CONTEXT_INTEGER
)
187 regs
.eax
= context
->Eax
;
188 regs
.ebx
= context
->Ebx
;
189 regs
.ecx
= context
->Ecx
;
190 regs
.edx
= context
->Edx
;
191 regs
.esi
= context
->Esi
;
192 regs
.edi
= context
->Edi
;
194 if (flags
& CONTEXT_CONTROL
)
196 regs
.ebp
= context
->Ebp
;
197 regs
.esp
= context
->Esp
;
198 regs
.eip
= context
->Eip
;
199 regs
.cs
= context
->SegCs
;
200 regs
.ss
= context
->SegSs
;
201 regs
.eflags
= context
->EFlags
;
203 if (flags
& CONTEXT_SEGMENTS
)
205 regs
.ds
= context
->SegDs
;
206 regs
.es
= context
->SegEs
;
207 regs
.fs
= context
->SegFs
;
208 regs
.gs
= context
->SegGs
;
210 if (ptrace( PTRACE_SETREGS
, pid
, 0, ®s
) == -1) goto error
;
212 if (flags
& CONTEXT_DEBUG_REGISTERS
)
214 if (ptrace( PTRACE_POKEUSER
, pid
, DR_OFFSET(0), context
->Dr0
) == -1) goto error
;
215 if (ptrace( PTRACE_POKEUSER
, pid
, DR_OFFSET(1), context
->Dr1
) == -1) goto error
;
216 if (ptrace( PTRACE_POKEUSER
, pid
, DR_OFFSET(2), context
->Dr2
) == -1) goto error
;
217 if (ptrace( PTRACE_POKEUSER
, pid
, DR_OFFSET(3), context
->Dr3
) == -1) goto error
;
218 if (ptrace( PTRACE_POKEUSER
, pid
, DR_OFFSET(6), context
->Dr6
) == -1) goto error
;
219 if (ptrace( PTRACE_POKEUSER
, pid
, DR_OFFSET(7), context
->Dr7
) == -1) goto error
;
221 if (flags
& CONTEXT_FLOATING_POINT
)
223 /* we can use context->FloatSave directly as it is using the */
224 /* correct structure (the same as fsave/frstor) */
225 if (ptrace( PTRACE_SETFPREGS
, pid
, 0, &context
->FloatSave
) == -1) goto error
;
232 #elif defined(__sun) || defined(__sun__)
234 /* retrieve a thread context */
235 static void get_thread_context( struct thread
*thread
, unsigned int flags
, CONTEXT
*context
)
237 int pid
= get_ptrace_pid(thread
);
238 if (flags
& CONTEXT_FULL
)
241 if (ptrace( PTRACE_GETREGS
, pid
, (int) ®s
, 0 ) == -1) goto error
;
242 if (flags
& CONTEXT_INTEGER
)
244 context
->Eax
= regs
.r_eax
;
245 context
->Ebx
= regs
.r_ebx
;
246 context
->Ecx
= regs
.r_ecx
;
247 context
->Edx
= regs
.r_edx
;
248 context
->Esi
= regs
.r_esi
;
249 context
->Edi
= regs
.r_edi
;
251 if (flags
& CONTEXT_CONTROL
)
253 context
->Ebp
= regs
.r_ebp
;
254 context
->Esp
= regs
.r_esp
;
255 context
->Eip
= regs
.r_eip
;
256 context
->SegCs
= regs
.r_cs
& 0xffff;
257 context
->SegSs
= regs
.r_ss
& 0xffff;
258 context
->EFlags
= regs
.r_efl
;
260 if (flags
& CONTEXT_SEGMENTS
)
262 context
->SegDs
= regs
.r_ds
& 0xffff;
263 context
->SegEs
= regs
.r_es
& 0xffff;
264 context
->SegFs
= regs
.r_fs
& 0xffff;
265 context
->SegGs
= regs
.r_gs
& 0xffff;
268 if (flags
& CONTEXT_DEBUG_REGISTERS
)
270 /* FIXME: How is this done on Solaris? */
272 if (flags
& CONTEXT_FLOATING_POINT
)
274 /* we can use context->FloatSave directly as it is using the */
275 /* correct structure (the same as fsave/frstor) */
276 if (ptrace( PTRACE_GETFPREGS
, pid
, (int) &context
->FloatSave
, 0 ) == -1) goto error
;
277 context
->FloatSave
.Cr0NpxState
= 0; /* FIXME */
285 /* set a thread context */
286 static void set_thread_context( struct thread
*thread
, unsigned int flags
, const CONTEXT
*context
)
288 int pid
= get_ptrace_pid(thread
);
289 if (flags
& CONTEXT_FULL
)
292 if (((flags
| CONTEXT_i386
) & CONTEXT_FULL
) != CONTEXT_FULL
)
294 /* need to preserve some registers */
295 if (ptrace( PTRACE_GETREGS
, pid
, (int) ®s
, 0 ) == -1) goto error
;
297 if (flags
& CONTEXT_INTEGER
)
299 regs
.r_eax
= context
->Eax
;
300 regs
.r_ebx
= context
->Ebx
;
301 regs
.r_ecx
= context
->Ecx
;
302 regs
.r_edx
= context
->Edx
;
303 regs
.r_esi
= context
->Esi
;
304 regs
.r_edi
= context
->Edi
;
306 if (flags
& CONTEXT_CONTROL
)
308 regs
.r_ebp
= context
->Ebp
;
309 regs
.r_esp
= context
->Esp
;
310 regs
.r_eip
= context
->Eip
;
311 regs
.r_cs
= context
->SegCs
;
312 regs
.r_ss
= context
->SegSs
;
313 regs
.r_efl
= context
->EFlags
;
315 if (flags
& CONTEXT_SEGMENTS
)
317 regs
.r_ds
= context
->SegDs
;
318 regs
.r_es
= context
->SegEs
;
319 regs
.r_fs
= context
->SegFs
;
320 regs
.r_gs
= context
->SegGs
;
322 if (ptrace( PTRACE_SETREGS
, pid
, (int) ®s
, 0 ) == -1) goto error
;
324 if (flags
& CONTEXT_DEBUG_REGISTERS
)
326 /* FIXME: How is this done on Solaris? */
328 if (flags
& CONTEXT_FLOATING_POINT
)
330 /* we can use context->FloatSave directly as it is using the */
331 /* correct structure (the same as fsave/frstor) */
332 if (ptrace( PTRACE_SETFPREGS
, pid
, (int) &context
->FloatSave
, 0 ) == -1) goto error
;
339 #elif defined(__FreeBSD__) || defined(__OpenBSD__) || defined(__NetBSD__)
340 #include <machine/reg.h>
342 /* retrieve a thread context */
343 static void get_thread_context( struct thread
*thread
, unsigned int flags
, CONTEXT
*context
)
345 int pid
= get_ptrace_pid(thread
);
346 if (flags
& CONTEXT_FULL
)
349 if (ptrace( PTRACE_GETREGS
, pid
, (caddr_t
) ®s
, 0 ) == -1) goto error
;
350 if (flags
& CONTEXT_INTEGER
)
352 context
->Eax
= regs
.r_eax
;
353 context
->Ebx
= regs
.r_ebx
;
354 context
->Ecx
= regs
.r_ecx
;
355 context
->Edx
= regs
.r_edx
;
356 context
->Esi
= regs
.r_esi
;
357 context
->Edi
= regs
.r_edi
;
359 if (flags
& CONTEXT_CONTROL
)
361 context
->Ebp
= regs
.r_ebp
;
362 context
->Esp
= regs
.r_esp
;
363 context
->Eip
= regs
.r_eip
;
364 context
->SegCs
= regs
.r_cs
& 0xffff;
365 context
->SegSs
= regs
.r_ss
& 0xffff;
366 context
->EFlags
= regs
.r_eflags
;
368 if (flags
& CONTEXT_SEGMENTS
)
370 context
->SegDs
= regs
.r_ds
& 0xffff;
371 context
->SegEs
= regs
.r_es
& 0xffff;
372 context
->SegFs
= regs
.r_fs
& 0xffff;
373 context
->SegGs
= regs
.r_gs
& 0xffff;
376 if (flags
& CONTEXT_DEBUG_REGISTERS
)
378 #ifdef PTRACE_GETDBREGS
380 if (ptrace( PTRACE_GETDBREGS
, pid
, (caddr_t
) &dbregs
, 0 ) == -1)
383 /* needed for FreeBSD, the structure fields have changed under 5.x */
384 context
->Dr0
= DBREG_DRX((&dbregs
), 0);
385 context
->Dr1
= DBREG_DRX((&dbregs
), 1);
386 context
->Dr2
= DBREG_DRX((&dbregs
), 2);
387 context
->Dr3
= DBREG_DRX((&dbregs
), 3);
388 context
->Dr6
= DBREG_DRX((&dbregs
), 6);
389 context
->Dr7
= DBREG_DRX((&dbregs
), 7);
391 context
->Dr0
= dbregs
.dr0
;
392 context
->Dr1
= dbregs
.dr1
;
393 context
->Dr2
= dbregs
.dr2
;
394 context
->Dr3
= dbregs
.dr3
;
395 context
->Dr6
= dbregs
.dr6
;
396 context
->Dr7
= dbregs
.dr7
;
401 if (flags
& CONTEXT_FLOATING_POINT
)
403 /* we can use context->FloatSave directly as it is using the */
404 /* correct structure (the same as fsave/frstor) */
405 if (ptrace( PTRACE_GETFPREGS
, pid
, (caddr_t
) &context
->FloatSave
, 0 ) == -1) goto error
;
406 context
->FloatSave
.Cr0NpxState
= 0; /* FIXME */
414 /* set a thread context */
415 static void set_thread_context( struct thread
*thread
, unsigned int flags
, const CONTEXT
*context
)
417 int pid
= get_ptrace_pid(thread
);
418 if (flags
& CONTEXT_FULL
)
421 if (((flags
| CONTEXT_i386
) & CONTEXT_FULL
) != CONTEXT_FULL
)
423 /* need to preserve some registers */
424 if (ptrace( PTRACE_GETREGS
, pid
, (caddr_t
) ®s
, 0 ) == -1) goto error
;
426 if (flags
& CONTEXT_INTEGER
)
428 regs
.r_eax
= context
->Eax
;
429 regs
.r_ebx
= context
->Ebx
;
430 regs
.r_ecx
= context
->Ecx
;
431 regs
.r_edx
= context
->Edx
;
432 regs
.r_esi
= context
->Esi
;
433 regs
.r_edi
= context
->Edi
;
435 if (flags
& CONTEXT_CONTROL
)
437 regs
.r_ebp
= context
->Ebp
;
438 regs
.r_esp
= context
->Esp
;
439 regs
.r_eip
= context
->Eip
;
440 regs
.r_cs
= context
->SegCs
;
441 regs
.r_ss
= context
->SegSs
;
442 regs
.r_eflags
= context
->EFlags
;
444 if (flags
& CONTEXT_SEGMENTS
)
446 regs
.r_ds
= context
->SegDs
;
447 regs
.r_es
= context
->SegEs
;
448 regs
.r_fs
= context
->SegFs
;
449 regs
.r_gs
= context
->SegGs
;
451 if (ptrace( PTRACE_SETREGS
, pid
, (caddr_t
) ®s
, 0 ) == -1) goto error
;
453 if (flags
& CONTEXT_DEBUG_REGISTERS
)
455 #ifdef PTRACE_SETDBREGS
458 /* needed for FreeBSD, the structure fields have changed under 5.x */
459 DBREG_DRX((&dbregs
), 0) = context
->Dr0
;
460 DBREG_DRX((&dbregs
), 1) = context
->Dr1
;
461 DBREG_DRX((&dbregs
), 2) = context
->Dr2
;
462 DBREG_DRX((&dbregs
), 3) = context
->Dr3
;
463 DBREG_DRX((&dbregs
), 4) = 0;
464 DBREG_DRX((&dbregs
), 5) = 0;
465 DBREG_DRX((&dbregs
), 6) = context
->Dr6
;
466 DBREG_DRX((&dbregs
), 7) = context
->Dr7
;
468 dbregs
.dr0
= context
->Dr0
;
469 dbregs
.dr1
= context
->Dr1
;
470 dbregs
.dr2
= context
->Dr2
;
471 dbregs
.dr3
= context
->Dr3
;
474 dbregs
.dr6
= context
->Dr6
;
475 dbregs
.dr7
= context
->Dr7
;
477 if (ptrace( PTRACE_SETDBREGS
, pid
, (caddr_t
) &dbregs
, 0 ) == -1)
481 if (flags
& CONTEXT_FLOATING_POINT
)
483 /* we can use context->FloatSave directly as it is using the */
484 /* correct structure (the same as fsave/frstor) */
485 if (ptrace( PTRACE_SETFPREGS
, pid
, (caddr_t
) &context
->FloatSave
, 0 ) == -1) goto error
;
492 #else /* linux || __sun__ || __FreeBSD__ */
493 #error You must implement get/set_thread_context for your platform
494 #endif /* linux || __sun__ || __FreeBSD__ */
497 /* copy a context structure according to the flags */
498 static void copy_context( CONTEXT
*to
, const CONTEXT
*from
, int flags
)
500 if (flags
& CONTEXT_CONTROL
)
505 to
->SegCs
= from
->SegCs
;
506 to
->SegSs
= from
->SegSs
;
507 to
->EFlags
= from
->EFlags
;
509 if (flags
& CONTEXT_INTEGER
)
518 if (flags
& CONTEXT_SEGMENTS
)
520 to
->SegDs
= from
->SegDs
;
521 to
->SegEs
= from
->SegEs
;
522 to
->SegFs
= from
->SegFs
;
523 to
->SegGs
= from
->SegGs
;
525 if (flags
& CONTEXT_FLOATING_POINT
)
527 to
->FloatSave
= from
->FloatSave
;
529 /* we don't bother copying the debug registers, since they */
530 /* always need to be accessed by ptrace anyway */
533 /* retrieve the current instruction pointer of a thread */
534 void *get_thread_ip( struct thread
*thread
)
538 if (suspend_for_ptrace( thread
))
540 get_thread_context( thread
, CONTEXT_CONTROL
, &context
);
541 resume_after_ptrace( thread
);
543 return (void *)context
.Eip
;
546 /* determine if we should continue the thread in single-step mode */
547 int get_thread_single_step( struct thread
*thread
)
550 if (thread
->context
) return 0; /* don't single-step inside exception event */
551 get_thread_context( thread
, CONTEXT_CONTROL
, &context
);
552 return (context
.EFlags
& 0x100) != 0;
555 /* send a signal to a specific thread */
556 int tkill( int pid
, int sig
)
560 __asm__( "pushl %%ebx\n\t"
565 : "0" (238) /*SYS_tkill*/, "r" (pid
), "c" (sig
) );
566 if (ret
>= 0) return ret
;
575 /* retrieve the current context of a thread */
576 DECL_HANDLER(get_thread_context
)
578 struct thread
*thread
;
580 int flags
= req
->flags
& ~CONTEXT_i386
; /* get rid of CPU id */
582 if (get_reply_max_size() < sizeof(CONTEXT
))
584 set_error( STATUS_INVALID_PARAMETER
);
587 if (!(thread
= get_thread_from_handle( req
->handle
, THREAD_GET_CONTEXT
))) return;
589 if ((data
= set_reply_data_size( sizeof(CONTEXT
) )))
591 /* copy incoming context into reply */
592 memset( data
, 0, sizeof(CONTEXT
) );
593 memcpy( data
, get_req_data(), min( get_req_data_size(), sizeof(CONTEXT
) ));
595 if (thread
->context
) /* thread is inside an exception event */
597 copy_context( data
, thread
->context
, flags
);
598 flags
&= CONTEXT_DEBUG_REGISTERS
;
600 if (flags
&& suspend_for_ptrace( thread
))
602 get_thread_context( thread
, flags
, data
);
603 resume_after_ptrace( thread
);
606 release_object( thread
);
610 /* set the current context of a thread */
611 DECL_HANDLER(set_thread_context
)
613 struct thread
*thread
;
614 int flags
= req
->flags
& ~CONTEXT_i386
; /* get rid of CPU id */
616 if (get_req_data_size() < sizeof(CONTEXT
))
618 set_error( STATUS_INVALID_PARAMETER
);
621 if ((thread
= get_thread_from_handle( req
->handle
, THREAD_SET_CONTEXT
)))
623 if (thread
->context
) /* thread is inside an exception event */
625 copy_context( thread
->context
, get_req_data(), flags
);
626 flags
&= CONTEXT_DEBUG_REGISTERS
;
628 if (flags
&& suspend_for_ptrace( thread
))
630 set_thread_context( thread
, flags
, get_req_data() );
631 resume_after_ptrace( thread
);
633 release_object( thread
);
637 #endif /* __i386__ */