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
 20int have_xstate_support;
 21
 22int 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
 29int 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
 45int 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
 52int 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__
 68int have_fpx_regs = 1;
 69int 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
 76int 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
 83int 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
 91int 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
 99void 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
116int get_fp_registers(int pid, unsigned long *regs)
117{
118	return save_fp_registers(pid, regs);
119}
120
121int put_fp_registers(int pid, unsigned long *regs)
122{
123	return restore_fp_registers(pid, regs);
124}
125
126void 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
146unsigned 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}