1/*-
2 * Copyright (c) 2011 Konstantin Belousov <kib@FreeBSD.org>
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$");
30
31#include "opt_compat.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 <sys/sysent.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	char *savefpu;
46	int error;
47
48	if (!use_xsave)
49		return (EOPNOTSUPP);
50
51	switch (req) {
52	case PT_GETXSTATE:
53		fpugetregs(td);
54		savefpu = (char *)(get_pcb_user_save_td(td) + 1);
55		error = copyout(savefpu, addr,
56		    cpu_max_ext_state_size - sizeof(struct savefpu));
57		break;
58
59	case PT_SETXSTATE:
60		if (data > cpu_max_ext_state_size - sizeof(struct savefpu)) {
61			error = EINVAL;
62			break;
63		}
64		savefpu = malloc(data, M_TEMP, M_WAITOK);
65		error = copyin(addr, savefpu, data);
66		if (error == 0) {
67			fpugetregs(td);
68			error = fpusetxstate(td, savefpu, data);
69		}
70		free(savefpu, M_TEMP);
71		break;
72
73	default:
74		error = EINVAL;
75		break;
76	}
77
78	return (error);
79}
80
81#ifdef COMPAT_FREEBSD32
82#define PT_I386_GETXMMREGS	(PT_FIRSTMACH + 0)
83#define PT_I386_SETXMMREGS	(PT_FIRSTMACH + 1)
84#define PT_I386_GETXSTATE	(PT_FIRSTMACH + 2)
85#define PT_I386_SETXSTATE	(PT_FIRSTMACH + 3)
86
87static int
88cpu32_ptrace(struct thread *td, int req, void *addr, int data)
89{
90	struct savefpu *fpstate;
91	int error;
92
93	switch (req) {
94	case PT_I386_GETXMMREGS:
95		fpugetregs(td);
96		error = copyout(get_pcb_user_save_td(td), addr,
97		    sizeof(*fpstate));
98		break;
99
100	case PT_I386_SETXMMREGS:
101		fpugetregs(td);
102		fpstate = get_pcb_user_save_td(td);
103		error = copyin(addr, fpstate, sizeof(*fpstate));
104		fpstate->sv_env.en_mxcsr &= cpu_mxcsr_mask;
105		break;
106
107	case PT_I386_GETXSTATE:
108		error = cpu_ptrace_xstate(td, PT_GETXSTATE, addr, data);
109		break;
110
111	case PT_I386_SETXSTATE:
112		error = cpu_ptrace_xstate(td, PT_SETXSTATE, addr, data);
113		break;
114
115	default:
116		error = EINVAL;
117		break;
118	}
119
120	return (error);
121}
122#endif
123
124int
125cpu_ptrace(struct thread *td, int req, void *addr, int data)
126{
127	int error;
128
129#ifdef COMPAT_FREEBSD32
130	if (SV_CURPROC_FLAG(SV_ILP32))
131		return (cpu32_ptrace(td, req, addr, data));
132#endif
133
134	/* Support old values of PT_GETXSTATE and PT_SETXSTATE. */
135	if (req == PT_FIRSTMACH + 0)
136		req = PT_GETXSTATE;
137	if (req == PT_FIRSTMACH + 1)
138		req = PT_SETXSTATE;
139
140	switch (req) {
141	case PT_GETXSTATE:
142	case PT_SETXSTATE:
143		error = cpu_ptrace_xstate(td, req, addr, data);
144		break;
145
146	default:
147		error = EINVAL;
148		break;
149	}
150
151	return (error);
152}
153