util/statetrace/arch/tracechild_amd64.hh

   1 /*
   2  * Copyright (c) 2007 The Regents of The University of Michigan
   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 are
   7  * met: redistributions of source code must retain the above copyright
   8  * notice, this list of conditions and the following disclaimer;
   9  * redistributions in binary form must reproduce the above copyright
  10  * notice, this list of conditions and the following disclaimer in the
  11  * documentation and/or other materials provided with the distribution;
  12  * neither the name of the copyright holders nor the names of its
  13  * contributors may be used to endorse or promote products derived from
  14  * this software without specific prior written permission.
  15  *
  16  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  17  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  18  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  19  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  20  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  21  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  22  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  23  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  24  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  25  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  26  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  27  *
  28  * Authors: Gabe Black
  29  */
  30
  31 #ifndef REGSTATE_AMD64_HH
  32 #define REGSTATE_AMD64_HH
  33
  34 #include <sys/user.h>
  35 #include <sys/types.h>
  36 #include <sys/ptrace.h>
  37 #include <cassert>
  38 #include <string>
  39
  40 #include "tracechild.hh"
  41
  42 class AMD64TraceChild : public TraceChild
  43 {
  44   public:
  45     enum RegNum
  46     {
  47         //GPRs
  48         RAX, RCX, RDX, RBX,
  49         //Base pointer and stack pointer
  50         RSP, RBP,
  51         //Index registers
  52         RSI, RDI,
  53         //New 64 bit mode registers
  54         R8, R9, R10, R11, R12, R13, R14, R15,
  55         //Segmentation registers
  56         CS, DS, ES, FS, GS, SS, FS_BASE, GS_BASE,
  57         //PC
  58         RIP,
  59         //Flags
  60         EFLAGS,
  61         //MMX
  62         MMX0_0, MMX0_1,
  63         MMX1_0, MMX1_1,
  64         MMX2_0, MMX2_1,
  65         MMX3_0, MMX3_1,
  66         MMX4_0, MMX4_1,
  67         MMX5_0, MMX5_1,
  68         MMX6_0, MMX6_1,
  69         MMX7_0, MMX7_1,
  70         //XMM
  71         XMM0_0,  XMM0_1,  XMM0_2,  XMM0_3,
  72         XMM1_0,  XMM1_1,  XMM1_2,  XMM1_3,
  73         XMM2_0,  XMM2_1,  XMM2_2,  XMM2_3,
  74         XMM3_0,  XMM3_1,  XMM3_2,  XMM3_3,
  75         XMM4_0,  XMM4_1,  XMM4_2,  XMM4_3,
  76         XMM5_0,  XMM5_1,  XMM5_2,  XMM5_3,
  77         XMM6_0,  XMM6_1,  XMM6_2,  XMM6_3,
  78         XMM7_0,  XMM7_1,  XMM7_2,  XMM7_3,
  79         XMM8_0,  XMM8_1,  XMM8_2,  XMM8_3,
  80         XMM9_0,  XMM9_1,  XMM9_2,  XMM9_3,
  81         XMM10_0, XMM10_1, XMM10_2, XMM10_3,
  82         XMM11_0, XMM11_1, XMM11_2, XMM11_3,
  83         XMM12_0, XMM12_1, XMM12_2, XMM12_3,
  84         XMM13_0, XMM13_1, XMM13_2, XMM13_3,
  85         XMM14_0, XMM14_1, XMM14_2, XMM14_3,
  86         XMM15_0, XMM15_1, XMM15_2, XMM15_3,
  87         numregs
  88     };
  89   private:
  90     char printBuffer [256];
  91     static const char * regNames[numregs];
  92     int64_t getRegs(user_regs_struct & myregs,
  93             user_fpregs_struct &myfpregs,int num);
  94     user_regs_struct regs;
  95     user_regs_struct oldregs;
  96     user_fpregs_struct fpregs;
  97     user_fpregs_struct oldfpregs;
  98     bool regDiffSinceUpdate[numregs];
  99
 100     uint64_t findSyscall();
 101
 102   protected:
 103     bool update(int pid);
 104
 105   public:
 106
 107     AMD64TraceChild();
 108
 109     bool sendState(int socket);
 110
 111     int getNumRegs()
 112     {
 113         return numregs;
 114     }
 115
 116     bool diffSinceUpdate(int num)
 117     {
 118         assert(num < numregs && num >= 0);
 119         return regDiffSinceUpdate[num];
 120     }
 121
 122     std::string getRegName(int num)
 123     {
 124         assert(num < numregs && num >= 0);
 125         return regNames[num];
 126     }
 127
 128     int64_t getRegVal(int num);
 129     int64_t getOldRegVal(int num);
 130     uint64_t getPC() {return getRegVal(RIP);}
 131     uint64_t getSP() {return getRegVal(RSP);}
 132     std::ostream & outputStartState(std::ostream & output);
 133
 134     char * printReg(int num);
 135
 136     bool step();
 137 };
 138
 139 #endif