tracechild.cc revision 3115
1/* 2 * Copyright (c) 2006 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#include "tracechild.hh" 32#include <sys/wait.h> 33#include <sys/ptrace.h> 34#include <iostream> 35#include <errno.h> 36 37using namespace std; 38 39bool TraceChild::startTracing(const char * pathToFile, const char * arg) 40{ 41 pid = fork(); 42 if(pid == -1) 43 { 44 cout << "fork failed" << endl; 45 return false; 46 } 47 else if(pid == 0) 48 { 49 //We're the child. Get things ready and then exec the 50 //program to trace. 51 52 //Let our parent trace us 53 ptrace(PTRACE_TRACEME, 0, 0, 0); 54 55 //Start the program to trace 56 execl(pathToFile, arg); 57 58 //We should never get here, so this is an error! 59 return false; 60 } 61 62 //From this point forward, we know we're in the parent process. 63 if(!doWait()) 64 { 65 cout << "Didn't wait successfully" << endl; 66 return false; 67 } 68 tracing = true; 69 if(!update(pid)) 70 { 71 cout << "Didn't update successfully!" << endl; 72 return false; 73 } 74 return true; 75} 76 77bool TraceChild::stopTracing() 78{ 79 if(ptrace(PTRACE_KILL, pid, 0, 0) != 0) 80 return false; 81 tracing = false; 82 return true; 83} 84 85bool TraceChild::step() 86{ 87 ptraceSingleStep(); 88} 89 90bool TraceChild::ptraceSingleStep() 91{ 92 if(!tracing) 93 { 94 cout << "Not tracing!" << endl; 95 return false; 96 } 97 if(ptrace(PTRACE_SINGLESTEP, pid, 0, 0) != 0) 98 { 99 switch(errno) 100 { 101 case EBUSY: cout << "EBUSY" << endl; break; 102 case EFAULT: cout << "EFAULT" << endl; break; 103 case EIO: cout << "EIO" << endl; break; 104 case EPERM: cout << "EPERM" << endl; break; 105 case ESRCH: cout << "ESRCH" << endl; break; 106 default: cout << "Unknown error" << endl; break; 107 } 108 cout << "Not able to single step!" << endl; 109 tracing == false; 110 return false; 111 } 112 doWait(); 113 update(pid); 114} 115 116bool TraceChild::doWait() 117{ 118 int wait_val; 119 wait(&wait_val); 120 if(WIFEXITED(wait_val)) 121 { 122 cerr << "Program exited! Exit status is " 123 << WEXITSTATUS(wait_val) << endl; 124 tracing = false; 125 return false; 126 } 127 if(WIFSIGNALED(wait_val)) 128 { 129 if(WTERMSIG(wait_val)) 130 cerr << "Program terminated by signal " 131 << WTERMSIG(wait_val) << endl; 132 if(WCOREDUMP(wait_val)) 133 cerr << "Program core dumped!" << endl; 134 tracing = false; 135 return false; 136 } 137 if(WIFSTOPPED(wait_val) && WSTOPSIG(wait_val) != SIGTRAP) 138 { 139 cerr << "Program stopped by signal " 140 << WSTOPSIG(wait_val) << endl; 141 tracing = false; 142 return false; 143 } 144 return true; 145} 146