um/os-Linux/registers.c

0001 /*
0002  * Copyright (C) 2004 PathScale, Inc
0003  * Copyright (C) 2004 - 2007 Jeff Dike (jdike@{addtoit,linux.intel}.com)
0004  * Licensed under the GPL
0005  */
0006
0007 #include <errno.h>
0008 #include <stdlib.h>
0009 #include <sys/ptrace.h>
0010 #ifdef __i386__
0011 #include <sys/user.h>
0012 #endif
0013 #include <longjmp.h>
0014 #include <sysdep/ptrace_user.h>
0015 #include <sys/uio.h>
0016 #include <asm/sigcontext.h>
0017 #include <linux/elf.h>
0018 #include <registers.h>
0019
0020 int have_xstate_support;
0021
0022 int save_i387_registers(int pid, unsigned long *fp_regs)
0023 {
0024     if (ptrace(PTRACE_GETFPREGS, pid, 0, fp_regs) < 0)
0025         return -errno;
0026     return 0;
0027 }
0028
0029 int save_fp_registers(int pid, unsigned long *fp_regs)
0030 {
0031 #ifdef PTRACE_GETREGSET
0032     struct iovec iov;
0033
0034     if (have_xstate_support) {
0035         iov.iov_base = fp_regs;
0036         iov.iov_len = FP_SIZE * sizeof(unsigned long);
0037         if (ptrace(PTRACE_GETREGSET, pid, NT_X86_XSTATE, &iov) < 0)
0038             return -errno;
0039         return 0;
0040     } else
0041 #endif
0042         return save_i387_registers(pid, fp_regs);
0043 }
0044
0045 int restore_i387_registers(int pid, unsigned long *fp_regs)
0046 {
0047     if (ptrace(PTRACE_SETFPREGS, pid, 0, fp_regs) < 0)
0048         return -errno;
0049     return 0;
0050 }
0051
0052 int restore_fp_registers(int pid, unsigned long *fp_regs)
0053 {
0054 #ifdef PTRACE_SETREGSET
0055     struct iovec iov;
0056     if (have_xstate_support) {
0057         iov.iov_base = fp_regs;
0058         iov.iov_len = FP_SIZE * sizeof(unsigned long);
0059         if (ptrace(PTRACE_SETREGSET, pid, NT_X86_XSTATE, &iov) < 0)
0060             return -errno;
0061         return 0;
0062     } else
0063 #endif
0064         return restore_i387_registers(pid, fp_regs);
0065 }
0066
0067 #ifdef __i386__
0068 int have_fpx_regs = 1;
0069 int save_fpx_registers(int pid, unsigned long *fp_regs)
0070 {
0071     if (ptrace(PTRACE_GETFPXREGS, pid, 0, fp_regs) < 0)
0072         return -errno;
0073     return 0;
0074 }
0075
0076 int restore_fpx_registers(int pid, unsigned long *fp_regs)
0077 {
0078     if (ptrace(PTRACE_SETFPXREGS, pid, 0, fp_regs) < 0)
0079         return -errno;
0080     return 0;
0081 }
0082
0083 int get_fp_registers(int pid, unsigned long *regs)
0084 {
0085     if (have_fpx_regs)
0086         return save_fpx_registers(pid, regs);
0087     else
0088         return save_fp_registers(pid, regs);
0089 }
0090
0091 int put_fp_registers(int pid, unsigned long *regs)
0092 {
0093     if (have_fpx_regs)
0094         return restore_fpx_registers(pid, regs);
0095     else
0096         return restore_fp_registers(pid, regs);
0097 }
0098
0099 void arch_init_registers(int pid)
0100 {
0101     struct user_fpxregs_struct fpx_regs;
0102     int err;
0103
0104     err = ptrace(PTRACE_GETFPXREGS, pid, 0, &fpx_regs);
0105     if (!err)
0106         return;
0107
0108     if (errno != EIO)
0109         panic("check_ptrace : PTRACE_GETFPXREGS failed, errno = %d",
0110               errno);
0111
0112     have_fpx_regs = 0;
0113 }
0114 #else
0115
0116 int get_fp_registers(int pid, unsigned long *regs)
0117 {
0118     return save_fp_registers(pid, regs);
0119 }
0120
0121 int put_fp_registers(int pid, unsigned long *regs)
0122 {
0123     return restore_fp_registers(pid, regs);
0124 }
0125
0126 void arch_init_registers(int pid)
0127 {
0128 #ifdef PTRACE_GETREGSET
0129     void * fp_regs;
0130     struct iovec iov;
0131
0132     fp_regs = malloc(FP_SIZE * sizeof(unsigned long));
0133     if(fp_regs == NULL)
0134         return;
0135
0136     iov.iov_base = fp_regs;
0137     iov.iov_len = FP_SIZE * sizeof(unsigned long);
0138     if (ptrace(PTRACE_GETREGSET, pid, NT_X86_XSTATE, &iov) == 0)
0139         have_xstate_support = 1;
0140
0141     free(fp_regs);
0142 #endif
0143 }
0144 #endif
0145
0146 unsigned long get_thread_reg(int reg, jmp_buf *buf)
0147 {
0148     switch (reg) {
0149 #ifdef __i386__
0150     case HOST_IP:
0151         return buf[0]->__eip;
0152     case HOST_SP:
0153         return buf[0]->__esp;
0154     case HOST_BP:
0155         return buf[0]->__ebp;
0156 #else
0157     case HOST_IP:
0158         return buf[0]->__rip;
0159     case HOST_SP:
0160         return buf[0]->__rsp;
0161     case HOST_BP:
0162         return buf[0]->__rbp;
0163 #endif
0164     default:
0165         printk(UM_KERN_ERR "get_thread_regs - unknown register %d\n",
0166                reg);
0167         return 0;
0168     }
0169 }