1 /*
2  * Copyright (C) 2004 PathScale, Inc
3  * Copyright (C) 2004 - 2007 Jeff Dike (jdike@{addtoit,linux.intel}.com)
4  * Licensed under the GPL
5  */
6 
7 #include <errno.h>
8 #include <stdlib.h>
9 #include <sys/ptrace.h>
10 #ifdef __i386__
11 #include <sys/user.h>
12 #endif
13 #include <longjmp.h>
14 #include <sysdep/ptrace_user.h>
15 #include <sys/uio.h>
16 #include <asm/sigcontext.h>
17 #include <linux/elf.h>
18 #include <registers.h>
19 
20 int have_xstate_support;
21 
save_i387_registers(int pid,unsigned long * fp_regs)22 int save_i387_registers(int pid, unsigned long *fp_regs)
23 {
24 	if (ptrace(PTRACE_GETFPREGS, pid, 0, fp_regs) < 0)
25 		return -errno;
26 	return 0;
27 }
28 
save_fp_registers(int pid,unsigned long * fp_regs)29 int save_fp_registers(int pid, unsigned long *fp_regs)
30 {
31 #ifdef PTRACE_GETREGSET
32 	struct iovec iov;
33 
34 	if (have_xstate_support) {
35 		iov.iov_base = fp_regs;
36 		iov.iov_len = FP_SIZE * sizeof(unsigned long);
37 		if (ptrace(PTRACE_GETREGSET, pid, NT_X86_XSTATE, &iov) < 0)
38 			return -errno;
39 		return 0;
40 	} else
41 #endif
42 		return save_i387_registers(pid, fp_regs);
43 }
44 
restore_i387_registers(int pid,unsigned long * fp_regs)45 int restore_i387_registers(int pid, unsigned long *fp_regs)
46 {
47 	if (ptrace(PTRACE_SETFPREGS, pid, 0, fp_regs) < 0)
48 		return -errno;
49 	return 0;
50 }
51 
restore_fp_registers(int pid,unsigned long * fp_regs)52 int restore_fp_registers(int pid, unsigned long *fp_regs)
53 {
54 #ifdef PTRACE_SETREGSET
55 	struct iovec iov;
56 	if (have_xstate_support) {
57 		iov.iov_base = fp_regs;
58 		iov.iov_len = FP_SIZE * sizeof(unsigned long);
59 		if (ptrace(PTRACE_SETREGSET, pid, NT_X86_XSTATE, &iov) < 0)
60 			return -errno;
61 		return 0;
62 	} else
63 #endif
64 		return restore_i387_registers(pid, fp_regs);
65 }
66 
67 #ifdef __i386__
68 int have_fpx_regs = 1;
save_fpx_registers(int pid,unsigned long * fp_regs)69 int save_fpx_registers(int pid, unsigned long *fp_regs)
70 {
71 	if (ptrace(PTRACE_GETFPXREGS, pid, 0, fp_regs) < 0)
72 		return -errno;
73 	return 0;
74 }
75 
restore_fpx_registers(int pid,unsigned long * fp_regs)76 int restore_fpx_registers(int pid, unsigned long *fp_regs)
77 {
78 	if (ptrace(PTRACE_SETFPXREGS, pid, 0, fp_regs) < 0)
79 		return -errno;
80 	return 0;
81 }
82 
get_fp_registers(int pid,unsigned long * regs)83 int get_fp_registers(int pid, unsigned long *regs)
84 {
85 	if (have_fpx_regs)
86 		return save_fpx_registers(pid, regs);
87 	else
88 		return save_fp_registers(pid, regs);
89 }
90 
put_fp_registers(int pid,unsigned long * regs)91 int put_fp_registers(int pid, unsigned long *regs)
92 {
93 	if (have_fpx_regs)
94 		return restore_fpx_registers(pid, regs);
95 	else
96 		return restore_fp_registers(pid, regs);
97 }
98 
arch_init_registers(int pid)99 void arch_init_registers(int pid)
100 {
101 	struct user_fpxregs_struct fpx_regs;
102 	int err;
103 
104 	err = ptrace(PTRACE_GETFPXREGS, pid, 0, &fpx_regs);
105 	if (!err)
106 		return;
107 
108 	if (errno != EIO)
109 		panic("check_ptrace : PTRACE_GETFPXREGS failed, errno = %d",
110 		      errno);
111 
112 	have_fpx_regs = 0;
113 }
114 #else
115 
get_fp_registers(int pid,unsigned long * regs)116 int get_fp_registers(int pid, unsigned long *regs)
117 {
118 	return save_fp_registers(pid, regs);
119 }
120 
put_fp_registers(int pid,unsigned long * regs)121 int put_fp_registers(int pid, unsigned long *regs)
122 {
123 	return restore_fp_registers(pid, regs);
124 }
125 
arch_init_registers(int pid)126 void arch_init_registers(int pid)
127 {
128 #ifdef PTRACE_GETREGSET
129 	void * fp_regs;
130 	struct iovec iov;
131 
132 	fp_regs = malloc(FP_SIZE * sizeof(unsigned long));
133 	if(fp_regs == NULL)
134 		return;
135 
136 	iov.iov_base = fp_regs;
137 	iov.iov_len = FP_SIZE * sizeof(unsigned long);
138 	if (ptrace(PTRACE_GETREGSET, pid, NT_X86_XSTATE, &iov) == 0)
139 		have_xstate_support = 1;
140 
141 	free(fp_regs);
142 #endif
143 }
144 #endif
145 
get_thread_reg(int reg,jmp_buf * buf)146 unsigned long get_thread_reg(int reg, jmp_buf *buf)
147 {
148 	switch (reg) {
149 #ifdef __i386__
150 	case HOST_IP:
151 		return buf[0]->__eip;
152 	case HOST_SP:
153 		return buf[0]->__esp;
154 	case HOST_BP:
155 		return buf[0]->__ebp;
156 #else
157 	case HOST_IP:
158 		return buf[0]->__rip;
159 	case HOST_SP:
160 		return buf[0]->__rsp;
161 	case HOST_BP:
162 		return buf[0]->__rbp;
163 #endif
164 	default:
165 		printk(UM_KERN_ERR "get_thread_regs - unknown register %d\n",
166 		       reg);
167 		return 0;
168 	}
169 }
170