1/*-
2 * Copyright (c) 2014 Justin Hibbits
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/param.h>
29#include <sys/systm.h>
30#include <sys/malloc.h>
31#include <sys/proc.h>
32#include <sys/ptrace.h>
33#include <sys/sysent.h>
34#include <machine/altivec.h>
35#include <machine/fpu.h>
36#include <machine/cpu.h>
37#include <machine/md_var.h>
38#include <machine/pcb.h>
39
40#ifdef __SPE__
41#define	PPC_FEATURE_VECTOR	PPC_FEATURE_HAS_SPE
42#else
43#define	PPC_FEATURE_VECTOR	PPC_FEATURE_HAS_ALTIVEC
44#endif
45
46int
47cpu_ptrace(struct thread *td, int req, void *addr, int data)
48{
49	int error;
50	struct pcb *pcb;
51	struct vec vec;
52	uint64_t vsr[32];
53	uint64_t *vsr_dw1;
54	int vsr_idx;
55
56	pcb = td->td_pcb;
57
58	bzero(&vec, sizeof(vec));
59	bzero(vsr, sizeof(vsr));
60
61	error = EINVAL;
62	switch (req) {
63	case PT_GETVRREGS:
64		if (!(cpu_features & PPC_FEATURE_VECTOR))
65			break;
66
67		if (pcb->pcb_flags & PCB_VEC) {
68			save_vec_nodrop(td);
69			memcpy(&vec, &pcb->pcb_vec, sizeof(vec));
70		}
71		error = copyout(&vec, addr, sizeof(vec));
72		break;
73	case PT_SETVRREGS:
74		if (!(cpu_features & PPC_FEATURE_VECTOR))
75			break;
76		error = copyin(addr, &vec, sizeof(vec));
77		if (error == 0) {
78			pcb->pcb_flags |= PCB_VEC;
79			memcpy(&pcb->pcb_vec, &vec, sizeof(vec));
80		}
81		break;
82	case PT_GETVSRREGS:
83		if (!(cpu_features & PPC_FEATURE_HAS_VSX))
84			break;
85
86		if (pcb->pcb_flags & PCB_VSX) {
87			save_fpu_nodrop(td);
88
89			/*
90			 * Doubleword 0 of VSR0-VSR31 overlap with FPR0-FPR31 and
91			 * VSR32-VSR63 overlap with VR0-VR31, so we only copy
92			 * the non-overlapping data, which is doubleword 1 of VSR0-VSR31.
93			 */
94			for (vsr_idx = 0; vsr_idx < nitems(vsr); vsr_idx++) {
95				vsr_dw1 = (uint64_t *)&pcb->pcb_fpu.fpr[vsr_idx].vsr[2];
96				vsr[vsr_idx] = *vsr_dw1;
97			}
98		}
99		error = copyout(&vsr, addr, sizeof(vsr));
100		break;
101	case PT_SETVSRREGS:
102		if (!(cpu_features & PPC_FEATURE_HAS_VSX))
103			break;
104		error = copyin(addr, &vsr, sizeof(vsr));
105		if (error == 0) {
106			pcb->pcb_flags |= PCB_VSX;
107
108			/*
109			 * Doubleword 0 of VSR0-VSR31 overlap with FPR0-FPR31 and
110			 * VSR32-VSR63 overlap with VR0-VR31, so we only copy
111			 * the non-overlapping data, which is doubleword 1 of VSR0-VSR31.
112			 */
113			for (vsr_idx = 0; vsr_idx < nitems(vsr); vsr_idx++) {
114				vsr_dw1 = (uint64_t *)&pcb->pcb_fpu.fpr[vsr_idx].vsr[2];
115				*vsr_dw1 = vsr[vsr_idx];
116			}
117		}
118		break;
119
120	default:
121		break;
122	}
123
124	return (error);
125}
126