1/*-
2 * Copyright (c) 2005 Doug Rabson
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 *    notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 *    notice, this list of conditions and the following disclaimer in the
12 *    documentation and/or other materials provided with the distribution.
13 *
14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24 * SUCH DAMAGE.
25 *
26 */
27
28#include <sys/cdefs.h>
29__FBSDID("$FreeBSD: stable/11/sys/i386/i386/ptrace_machdep.c 314210 2017-02-24 16:02:01Z kib $");
30
31#include "opt_cpu.h"
32
33#include <sys/param.h>
34#include <sys/systm.h>
35#include <sys/malloc.h>
36#include <sys/proc.h>
37#include <sys/ptrace.h>
38#include <machine/frame.h>
39#include <machine/md_var.h>
40#include <machine/pcb.h>
41
42static int
43cpu_ptrace_xstate(struct thread *td, int req, void *addr, int data)
44{
45	struct ptrace_xstate_info info;
46	char *savefpu;
47	int error;
48
49	if (!use_xsave)
50		return (EOPNOTSUPP);
51
52	switch (req) {
53	case PT_GETXSTATE_OLD:
54		npxgetregs(td);
55		savefpu = (char *)(get_pcb_user_save_td(td) + 1);
56		error = copyout(savefpu, addr,
57		    cpu_max_ext_state_size - sizeof(union savefpu));
58		break;
59
60	case PT_SETXSTATE_OLD:
61		if (data > cpu_max_ext_state_size - sizeof(union savefpu)) {
62			error = EINVAL;
63			break;
64		}
65		savefpu = malloc(data, M_TEMP, M_WAITOK);
66		error = copyin(addr, savefpu, data);
67		if (error == 0) {
68			npxgetregs(td);
69			error = npxsetxstate(td, savefpu, data);
70		}
71		free(savefpu, M_TEMP);
72		break;
73
74	case PT_GETXSTATE_INFO:
75		if (data != sizeof(info)) {
76			error  = EINVAL;
77			break;
78		}
79		info.xsave_len = cpu_max_ext_state_size;
80		info.xsave_mask = xsave_mask;
81		error = copyout(&info, addr, data);
82		break;
83
84	case PT_GETXSTATE:
85		npxgetregs(td);
86		savefpu = (char *)(get_pcb_user_save_td(td));
87		error = copyout(savefpu, addr, cpu_max_ext_state_size);
88		break;
89
90	case PT_SETXSTATE:
91		if (data < sizeof(union savefpu) ||
92		    data > cpu_max_ext_state_size) {
93			error = EINVAL;
94			break;
95		}
96		savefpu = malloc(data, M_TEMP, M_WAITOK);
97		error = copyin(addr, savefpu, data);
98		if (error == 0)
99			error = npxsetregs(td, (union savefpu *)savefpu,
100			    savefpu + sizeof(union savefpu), data -
101			    sizeof(union savefpu));
102		free(savefpu, M_TEMP);
103		break;
104
105	default:
106		error = EINVAL;
107		break;
108	}
109
110	return (error);
111}
112
113static int
114cpu_ptrace_xmm(struct thread *td, int req, void *addr, int data)
115{
116	struct savexmm *fpstate;
117	int error;
118
119	if (!cpu_fxsr)
120		return (EINVAL);
121
122	fpstate = &get_pcb_user_save_td(td)->sv_xmm;
123	switch (req) {
124	case PT_GETXMMREGS:
125		npxgetregs(td);
126		error = copyout(fpstate, addr, sizeof(*fpstate));
127		break;
128
129	case PT_SETXMMREGS:
130		npxgetregs(td);
131		error = copyin(addr, fpstate, sizeof(*fpstate));
132		fpstate->sv_env.en_mxcsr &= cpu_mxcsr_mask;
133		break;
134
135	case PT_GETXSTATE_OLD:
136	case PT_SETXSTATE_OLD:
137	case PT_GETXSTATE_INFO:
138	case PT_GETXSTATE:
139	case PT_SETXSTATE:
140		error = cpu_ptrace_xstate(td, req, addr, data);
141		break;
142
143	default:
144		return (EINVAL);
145	}
146
147	return (error);
148}
149
150int
151cpu_ptrace(struct thread *td, int req, void *addr, int data)
152{
153	struct segment_descriptor *sdp, sd;
154	register_t r;
155	int error;
156
157	switch (req) {
158	case PT_GETXMMREGS:
159	case PT_SETXMMREGS:
160	case PT_GETXSTATE_OLD:
161	case PT_SETXSTATE_OLD:
162	case PT_GETXSTATE_INFO:
163	case PT_GETXSTATE:
164	case PT_SETXSTATE:
165		error = cpu_ptrace_xmm(td, req, addr, data);
166		break;
167
168	case PT_GETFSBASE:
169	case PT_GETGSBASE:
170		sdp = req == PT_GETFSBASE ? &td->td_pcb->pcb_fsd :
171		    &td->td_pcb->pcb_gsd;
172		r = sdp->sd_hibase << 24 | sdp->sd_lobase;
173		error = copyout(&r, addr, sizeof(r));
174		break;
175
176	case PT_SETFSBASE:
177	case PT_SETGSBASE:
178		error = copyin(addr, &r, sizeof(r));
179		if (error != 0)
180			break;
181		fill_based_sd(&sd, r);
182		if (req == PT_SETFSBASE) {
183			td->td_pcb->pcb_fsd = sd;
184			td->td_frame->tf_fs = GSEL(GUFS_SEL, SEL_UPL);
185		} else {
186			td->td_pcb->pcb_gsd = sd;
187			td->td_pcb->pcb_gs = GSEL(GUGS_SEL, SEL_UPL);
188		}
189		break;
190
191	default:
192		return (EINVAL);
193	}
194
195	return (error);
196}
197