| /* |
| * Copyright (c) 2006-2007 The Regents of The University of Michigan |
| * All rights reserved. |
| * |
| * Redistribution and use in source and binary forms, with or without |
| * modification, are permitted provided that the following conditions are |
| * met: redistributions of source code must retain the above copyright |
| * notice, this list of conditions and the following disclaimer; |
| * redistributions in binary form must reproduce the above copyright |
| * notice, this list of conditions and the following disclaimer in the |
| * documentation and/or other materials provided with the distribution; |
| * neither the name of the copyright holders nor the names of its |
| * contributors may be used to endorse or promote products derived from |
| * this software without specific prior written permission. |
| * |
| * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
| * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
| * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
| * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
| * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
| * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
| * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
| * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
| * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
| * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
| * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| * |
| * Authors: Gabe Black |
| */ |
| |
| #include <sys/ptrace.h> |
| #include <stdint.h> |
| |
| #include <cerrno> |
| #include <iostream> |
| |
| #include "arch/sparc/tracechild.hh" |
| |
| using namespace std; |
| |
| bool |
| SparcTraceChild::sendState(int socket) |
| { |
| uint64_t regVal = 0; |
| for (int x = 0; x <= I7; x++) { |
| regVal = getRegVal(x); |
| if (write(socket, ®Val, sizeof(regVal)) == -1) { |
| cerr << "Write failed! " << strerror(errno) << endl; |
| tracing = false; |
| return false; |
| } |
| } |
| regVal = getRegVal(PC); |
| if (write(socket, ®Val, sizeof(regVal)) == -1) { |
| cerr << "Write failed! " << strerror(errno) << endl; |
| tracing = false; |
| return false; |
| } |
| regVal = getRegVal(NPC); |
| if (write(socket, ®Val, sizeof(regVal)) == -1) { |
| cerr << "Write failed! " << strerror(errno) << endl; |
| tracing = false; |
| return false; |
| } |
| regVal = getRegVal(CCR); |
| if (write(socket, ®Val, sizeof(regVal)) == -1) { |
| cerr << "Write failed! " << strerror(errno) << endl; |
| tracing = false; |
| return false; |
| } |
| return true; |
| } |
| |
| int64_t |
| getRegs(regs & myregs, fpu & myfpu, uint64_t * locals, |
| uint64_t * inputs, int num) |
| { |
| assert(num < SparcTraceChild::numregs && num >= 0); |
| switch (num) { |
| //Global registers |
| case SparcTraceChild::G0: return 0; |
| case SparcTraceChild::G1: return myregs.r_g1; |
| case SparcTraceChild::G2: return myregs.r_g2; |
| case SparcTraceChild::G3: return myregs.r_g3; |
| case SparcTraceChild::G4: return myregs.r_g4; |
| case SparcTraceChild::G5: return myregs.r_g5; |
| case SparcTraceChild::G6: return myregs.r_g6; |
| case SparcTraceChild::G7: return myregs.r_g7; |
| //Output registers |
| case SparcTraceChild::O0: return myregs.r_o0; |
| case SparcTraceChild::O1: return myregs.r_o1; |
| case SparcTraceChild::O2: return myregs.r_o2; |
| case SparcTraceChild::O3: return myregs.r_o3; |
| case SparcTraceChild::O4: return myregs.r_o4; |
| case SparcTraceChild::O5: return myregs.r_o5; |
| case SparcTraceChild::O6: return myregs.r_o6; |
| case SparcTraceChild::O7: return myregs.r_o7; |
| //Local registers |
| case SparcTraceChild::L0: return locals[0]; |
| case SparcTraceChild::L1: return locals[1]; |
| case SparcTraceChild::L2: return locals[2]; |
| case SparcTraceChild::L3: return locals[3]; |
| case SparcTraceChild::L4: return locals[4]; |
| case SparcTraceChild::L5: return locals[5]; |
| case SparcTraceChild::L6: return locals[6]; |
| case SparcTraceChild::L7: return locals[7]; |
| //Input registers |
| case SparcTraceChild::I0: return inputs[0]; |
| case SparcTraceChild::I1: return inputs[1]; |
| case SparcTraceChild::I2: return inputs[2]; |
| case SparcTraceChild::I3: return inputs[3]; |
| case SparcTraceChild::I4: return inputs[4]; |
| case SparcTraceChild::I5: return inputs[5]; |
| case SparcTraceChild::I6: return inputs[6]; |
| case SparcTraceChild::I7: return inputs[7]; |
| //Floating point |
| case SparcTraceChild::F0: return myfpu.f_fpstatus.fpu_fr[0]; |
| case SparcTraceChild::F2: return myfpu.f_fpstatus.fpu_fr[1]; |
| case SparcTraceChild::F4: return myfpu.f_fpstatus.fpu_fr[2]; |
| case SparcTraceChild::F6: return myfpu.f_fpstatus.fpu_fr[3]; |
| case SparcTraceChild::F8: return myfpu.f_fpstatus.fpu_fr[4]; |
| case SparcTraceChild::F10: return myfpu.f_fpstatus.fpu_fr[5]; |
| case SparcTraceChild::F12: return myfpu.f_fpstatus.fpu_fr[6]; |
| case SparcTraceChild::F14: return myfpu.f_fpstatus.fpu_fr[7]; |
| case SparcTraceChild::F16: return myfpu.f_fpstatus.fpu_fr[8]; |
| case SparcTraceChild::F18: return myfpu.f_fpstatus.fpu_fr[9]; |
| case SparcTraceChild::F20: return myfpu.f_fpstatus.fpu_fr[10]; |
| case SparcTraceChild::F22: return myfpu.f_fpstatus.fpu_fr[11]; |
| case SparcTraceChild::F24: return myfpu.f_fpstatus.fpu_fr[12]; |
| case SparcTraceChild::F26: return myfpu.f_fpstatus.fpu_fr[13]; |
| case SparcTraceChild::F28: return myfpu.f_fpstatus.fpu_fr[14]; |
| case SparcTraceChild::F30: return myfpu.f_fpstatus.fpu_fr[15]; |
| case SparcTraceChild::F32: return myfpu.f_fpstatus.fpu_fr[16]; |
| case SparcTraceChild::F34: return myfpu.f_fpstatus.fpu_fr[17]; |
| case SparcTraceChild::F36: return myfpu.f_fpstatus.fpu_fr[18]; |
| case SparcTraceChild::F38: return myfpu.f_fpstatus.fpu_fr[19]; |
| case SparcTraceChild::F40: return myfpu.f_fpstatus.fpu_fr[20]; |
| case SparcTraceChild::F42: return myfpu.f_fpstatus.fpu_fr[21]; |
| case SparcTraceChild::F44: return myfpu.f_fpstatus.fpu_fr[22]; |
| case SparcTraceChild::F46: return myfpu.f_fpstatus.fpu_fr[23]; |
| case SparcTraceChild::F48: return myfpu.f_fpstatus.fpu_fr[24]; |
| case SparcTraceChild::F50: return myfpu.f_fpstatus.fpu_fr[25]; |
| case SparcTraceChild::F52: return myfpu.f_fpstatus.fpu_fr[26]; |
| case SparcTraceChild::F54: return myfpu.f_fpstatus.fpu_fr[27]; |
| case SparcTraceChild::F56: return myfpu.f_fpstatus.fpu_fr[28]; |
| case SparcTraceChild::F58: return myfpu.f_fpstatus.fpu_fr[29]; |
| case SparcTraceChild::F60: return myfpu.f_fpstatus.fpu_fr[30]; |
| case SparcTraceChild::F62: return myfpu.f_fpstatus.fpu_fr[31]; |
| //Miscelaneous |
| case SparcTraceChild::FSR: return myfpu.f_fpstatus.Fpu_fsr; |
| case SparcTraceChild::FPRS: return myregs.r_fprs; |
| case SparcTraceChild::PC: return myregs.r_tpc; |
| case SparcTraceChild::NPC: return myregs.r_tnpc; |
| case SparcTraceChild::Y: return myregs.r_y; |
| case SparcTraceChild::CWP: |
| return (myregs.r_tstate >> 0) & ((1 << 5) - 1); |
| case SparcTraceChild::PSTATE: |
| return (myregs.r_tstate >> 8) & ((1 << 13) - 1); |
| case SparcTraceChild::ASI: |
| return (myregs.r_tstate >> 24) & ((1 << 8) - 1); |
| case SparcTraceChild::CCR: |
| return (myregs.r_tstate >> 32) & ((1 << 8) - 1); |
| default: |
| assert(0); |
| return 0; |
| } |
| } |
| |
| bool |
| SparcTraceChild::update(int pid) |
| { |
| memcpy(&oldregs, &theregs, sizeof(regs)); |
| memcpy(&oldfpregs, &thefpregs, sizeof(fpu)); |
| memcpy(oldLocals, locals, 8 * sizeof(uint64_t)); |
| memcpy(oldInputs, inputs, 8 * sizeof(uint64_t)); |
| if (ptrace(PTRACE_GETREGS, pid, &theregs, 0) != 0) { |
| cerr << "Update failed" << endl; |
| return false; |
| } |
| uint64_t stackPointer = getSP(); |
| uint64_t stackBias = 2047; |
| bool v9 = stackPointer % 2; |
| for (unsigned int x = 0; x < 8; x++) { |
| uint64_t localAddr = stackPointer + |
| (v9 ? (stackBias + x * 8) : (x * 4)); |
| locals[x] = ptrace(PTRACE_PEEKTEXT, pid, localAddr, 0); |
| if (!v9) locals[x] >>= 32; |
| uint64_t inputAddr = stackPointer + |
| (v9 ? (stackBias + x * 8 + (8 * 8)) : (x * 4 + 8 * 4)); |
| inputs[x] = ptrace(PTRACE_PEEKTEXT, pid, inputAddr, 0); |
| if (!v9) inputs[x] >>= 32; |
| } |
| if (ptrace(PTRACE_GETFPREGS, pid, &thefpregs, 0) != 0) |
| return false; |
| for (unsigned int x = 0; x < numregs; x++) |
| regDiffSinceUpdate[x] = (getRegVal(x) != getOldRegVal(x)); |
| return true; |
| } |
| |
| SparcTraceChild::SparcTraceChild() |
| { |
| for (unsigned int x = 0; x < numregs; x++) |
| regDiffSinceUpdate[x] = false; |
| } |
| |
| int |
| SparcTraceChild::getTargets(uint32_t inst, uint64_t pc, uint64_t npc, |
| uint64_t &target1, uint64_t &target2) |
| { |
| //We can identify the instruction categories we care about using the top |
| //10 bits of the instruction, excluding the annul bit in the 3rd most |
| //significant bit position and the condition field. We'll call these |
| //bits the "sig" for signature. |
| uint32_t sig = (inst >> 22) & 0x307; |
| uint32_t cond = (inst >> 25) & 0xf; |
| bool annul = (inst & (1 << 29)); |
| |
| //Check if it's a ba... |
| bool ba = (cond == 0x8) && |
| (sig == 0x1 || sig == 0x2 || sig == 0x5 || sig == 0x6); |
| //or a bn... |
| bool bn = (cond == 0x0) && |
| (sig == 0x1 || sig == 0x2 || sig == 0x5 || sig == 0x6); |
| //or a bcc |
| bool bcc = (cond & 0x7) && |
| (sig == 0x1 || sig == 0x2 || sig == 0x3 || sig == 0x5 || sig == 0x6); |
| |
| if (annul) { |
| if (bcc) { |
| target1 = npc; |
| target2 = npc + 4; |
| return 2; |
| } else if (ba) { |
| //This branches immediately to the effective address of the branch |
| //which we'll have to calculate. |
| uint64_t disp = 0; |
| int64_t extender = 0; |
| //Figure out how big the displacement field is, and grab the bits |
| if (sig == 0x1 || sig == 0x5) { |
| disp = inst & ((1 << 19) - 1); |
| extender = 1 << 18; |
| } else { |
| disp = inst & ((1 << 22) - 1); |
| extender = 1 << 21; |
| } |
| //This does sign extension, believe it or not. |
| disp = (disp ^ extender) - extender; |
| //Multiply the displacement by 4. I'm assuming the compiler is |
| //smart enough to turn this into a shift. |
| disp *= 4; |
| target1 = pc + disp; |
| } else if (bn) |
| target1 = npc + 4; |
| else |
| target1 = npc; |
| return 1; |
| } else { |
| target1 = npc; |
| return 1; |
| } |
| } |
| |
| bool |
| SparcTraceChild::step() |
| { |
| //Increment the count of the number of instructions executed |
| instructions++; |
| //Two important considerations are that the address of the instruction |
| //being breakpointed should be word (64bit) aligned, and that both the |
| //next instruction and the instruction after that need to be breakpointed |
| //so that annulled branches will still stop as well. |
| |
| /* |
| * Useful constants |
| */ |
| const static uint64_t breakInst = 0x91d02001; |
| const static uint64_t lowBreakInst = breakInst; |
| const static uint64_t highBreakInst = breakInst << 32; |
| const static uint64_t breakWord = breakInst | (breakInst << 32); |
| const static uint64_t lowMask = 0xFFFFFFFFULL; |
| const static uint64_t highMask = lowMask << 32; |
| |
| /* |
| * storage for the original contents of the child process's memory |
| */ |
| uint64_t originalInst, originalAnnulInst; |
| |
| /* |
| * Get information about where the process is and is headed next. |
| */ |
| uint64_t currentPC = getRegVal(PC); |
| bool unalignedPC = currentPC & 7; |
| uint64_t alignedPC = currentPC & (~7); |
| uint64_t nextPC = getRegVal(NPC); |
| bool unalignedNPC = nextPC & 7; |
| uint64_t alignedNPC = nextPC & (~7); |
| |
| //Get the current instruction |
| uint64_t curInst = ptrace(PTRACE_PEEKTEXT, pid, alignedPC); |
| curInst = unalignedPC ? (curInst & 0xffffffffULL) : (curInst >> 32); |
| |
| uint64_t bp1, bp2; |
| int numTargets = getTargets(curInst, currentPC, nextPC, bp1, bp2); |
| assert(numTargets == 1 || numTargets == 2); |
| |
| bool unalignedBp1 = bp1 & 7; |
| uint64_t alignedBp1 = bp1 & (~7); |
| bool unalignedBp2 = bp2 & 7; |
| uint64_t alignedBp2 = bp2 & (~7); |
| uint64_t origBp1, origBp2; |
| |
| /* |
| * Set the first breakpoint |
| */ |
| origBp1 = ptrace(PTRACE_PEEKTEXT, pid, alignedBp1, 0); |
| uint64_t newBp1 = origBp1; |
| newBp1 &= unalignedBp1 ? highMask : lowMask; |
| newBp1 |= unalignedBp1 ? lowBreakInst : highBreakInst; |
| if (ptrace(PTRACE_POKETEXT, pid, alignedBp1, newBp1) != 0) |
| cerr << "Poke failed" << endl; |
| /* |
| * Set the second breakpoint if necessary |
| */ |
| if (numTargets == 2) { |
| origBp2 = ptrace(PTRACE_PEEKTEXT, pid, alignedBp2, 0); |
| uint64_t newBp2 = origBp2; |
| newBp2 &= unalignedBp2 ? highMask : lowMask; |
| newBp2 |= unalignedBp2 ? lowBreakInst : highBreakInst; |
| if (ptrace(PTRACE_POKETEXT, pid, alignedBp2, newBp2) != 0) |
| cerr << "Poke failed" << endl; |
| } |
| |
| /* |
| * Restart the child process |
| */ |
| //Note that the "addr" parameter is supposed to be ignored, but in at |
| //least one version of the kernel, it must be 1 or it will set what |
| //pc to continue from |
| if (ptrace(PTRACE_CONT, pid, 1, 0) != 0) |
| cerr << "Cont failed" << endl; |
| doWait(); |
| |
| /* |
| * Update our record of the child's state |
| */ |
| update(pid); |
| |
| /* |
| * Put back the original contents of the childs address space in the |
| * reverse order. |
| */ |
| if (numTargets == 2) { |
| if (ptrace(PTRACE_POKETEXT, pid, alignedBp2, origBp2) != 0) |
| cerr << "Poke failed" << endl; |
| } |
| if (ptrace(PTRACE_POKETEXT, pid, alignedBp1, origBp1) != 0) |
| cerr << "Poke failed" << endl; |
| } |
| |
| int64_t |
| SparcTraceChild::getRegVal(int num) |
| { |
| return getRegs(theregs, thefpregs, locals, inputs, num); |
| } |
| |
| int64_t |
| SparcTraceChild::getOldRegVal(int num) |
| { |
| return getRegs(oldregs, oldfpregs, oldLocals, oldInputs, num); |
| } |
| |
| ostream & |
| SparcTraceChild::outputStartState(ostream & os) |
| { |
| bool v8 = false; |
| uint64_t sp = getSP(); |
| if (sp % 2) { |
| os << "Detected a 64 bit executable.\n"; |
| v8 = false; |
| } else { |
| os << "Detected a 32 bit executable.\n"; |
| v8 = true; |
| } |
| uint64_t pc = getPC(); |
| char obuf[1024]; |
| sprintf(obuf, "Initial stack pointer = 0x%016llx\n", sp); |
| os << obuf; |
| sprintf(obuf, "Initial program counter = 0x%016llx\n", pc); |
| os << obuf; |
| if (!v8) { |
| //Take out the stack bias |
| sp += 2047; |
| } |
| //Output the window save area |
| for (unsigned int x = 0; x < 16; x++) { |
| uint64_t regspot = ptrace(PTRACE_PEEKDATA, pid, sp, 0); |
| if (v8) regspot = regspot >> 32; |
| sprintf(obuf, "0x%016llx: Window save %d = 0x%016llx\n", |
| sp, x+1, regspot); |
| os << obuf; |
| sp += v8 ? 4 : 8; |
| } |
| //Output the argument count |
| uint64_t cargc = ptrace(PTRACE_PEEKDATA, pid, sp, 0); |
| if (v8) cargc = cargc >> 32; |
| sprintf(obuf, "0x%016llx: Argc = 0x%016llx\n", sp, cargc); |
| os << obuf; |
| sp += v8 ? 4 : 8; |
| //Output argv pointers |
| int argCount = 0; |
| uint64_t cargv; |
| do { |
| cargv = ptrace(PTRACE_PEEKDATA, pid, sp, 0); |
| if (v8) cargv = cargv >> 32; |
| sprintf(obuf, "0x%016llx: argv[%d] = 0x%016llx\n", |
| sp, argCount++, cargv); |
| os << obuf; |
| sp += v8 ? 4 : 8; |
| } while (cargv); |
| //Output the envp pointers |
| int envCount = 0; |
| uint64_t cenvp; |
| do { |
| cenvp = ptrace(PTRACE_PEEKDATA, pid, sp, 0); |
| if (v8) cenvp = cenvp >> 32; |
| sprintf(obuf, "0x%016llx: envp[%d] = 0x%016llx\n", |
| sp, envCount++, cenvp); |
| os << obuf; |
| sp += v8 ? 4 : 8; |
| } while (cenvp); |
| uint64_t auxType, auxVal; |
| do { |
| auxType = ptrace(PTRACE_PEEKDATA, pid, sp, 0); |
| if (v8) auxType = auxType >> 32; |
| sp += (v8 ? 4 : 8); |
| auxVal = ptrace(PTRACE_PEEKDATA, pid, sp, 0); |
| if (v8) auxVal = auxVal >> 32; |
| sp += (v8 ? 4 : 8); |
| sprintf(obuf, "0x%016llx: Auxiliary vector = {0x%016llx, 0x%016llx}\n", |
| sp - 8, auxType, auxVal); |
| os << obuf; |
| } while (auxType != 0 || auxVal != 0); |
| //Print out the argument strings, environment strings, and file name. |
| string current; |
| uint64_t buf; |
| uint64_t currentStart = sp; |
| bool clearedInitialPadding = false; |
| do { |
| buf = ptrace(PTRACE_PEEKDATA, pid, sp, 0); |
| char * cbuf = (char *)&buf; |
| for (int x = 0; x < sizeof(uint32_t); x++) { |
| if (cbuf[x]) |
| current += cbuf[x]; |
| else { |
| sprintf(obuf, "0x%016llx: \"%s\"\n", |
| currentStart, current.c_str()); |
| os << obuf; |
| current = ""; |
| currentStart = sp + x + 1; |
| } |
| } |
| sp += (v8 ? 4 : 8); |
| clearedInitialPadding = clearedInitialPadding || buf != 0; |
| } while (!clearedInitialPadding || buf != 0); |
| return os; |
| } |
| |
| TraceChild * |
| genTraceChild() |
| { |
| return new SparcTraceChild; |
| } |
| |
