proc_util.c revision 211184
1/*- 2 * Copyright (c) 2010 The FreeBSD Foundation 3 * Copyright (c) 2008 John Birrell (jb@freebsd.org) 4 * All rights reserved. 5 * 6 * Portions of this software were developed by Rui Paulo under sponsorship 7 * from the FreeBSD Foundation. 8 * 9 * Redistribution and use in source and binary forms, with or without 10 * modification, are permitted provided that the following conditions 11 * are met: 12 * 1. Redistributions of source code must retain the above copyright 13 * notice, this list of conditions and the following disclaimer. 14 * 2. Redistributions in binary form must reproduce the above copyright 15 * notice, this list of conditions and the following disclaimer in the 16 * documentation and/or other materials provided with the distribution. 17 * 18 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 19 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 20 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 21 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 22 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 23 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 24 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 25 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 26 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 27 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 28 * SUCH DAMAGE. 29 * 30 * $FreeBSD: head/lib/libproc/proc_util.c 211184 2010-08-11 17:33:26Z rpaulo $ 31 */ 32 33#include <sys/types.h> 34#include <sys/ptrace.h> 35#include <sys/wait.h> 36#include <err.h> 37#include <errno.h> 38#include <unistd.h> 39#include <stdio.h> 40#include <signal.h> 41#include <string.h> 42#include "_libproc.h" 43 44int 45proc_clearflags(struct proc_handle *phdl, int mask) 46{ 47 48 if (phdl == NULL) 49 return (EINVAL); 50 51 phdl->flags &= ~mask; 52 53 return (0); 54} 55 56/* 57 * NB: we return -1 as the Solaris libproc Psetrun() function. 58 */ 59int 60proc_continue(struct proc_handle *phdl) 61{ 62 63 if (phdl == NULL) 64 return (-1); 65 66 if (ptrace(PT_CONTINUE, phdl->pid, (caddr_t)(uintptr_t) 1, 0) != 0) 67 return (-1); 68 69 phdl->status = PS_RUN; 70 71 return (0); 72} 73 74int 75proc_detach(struct proc_handle *phdl, int reason) 76{ 77 int status; 78 79 if (phdl == NULL) 80 return (EINVAL); 81 if (reason == PRELEASE_KILL) { 82 kill(phdl->pid, SIGKILL); 83 return (0); 84 } 85 if (ptrace(PT_DETACH, phdl->pid, 0, 0) != 0 && errno == ESRCH) 86 return (0); 87 if (errno == EBUSY) { 88 kill(phdl->pid, SIGSTOP); 89 waitpid(phdl->pid, &status, WUNTRACED); 90 ptrace(PT_DETACH, phdl->pid, 0, 0); 91 kill(phdl->pid, SIGCONT); 92 return (0); 93 } 94 95 return (0); 96} 97 98int 99proc_getflags(struct proc_handle *phdl) 100{ 101 102 if (phdl == NULL) 103 return (-1); 104 105 return(phdl->flags); 106} 107 108int 109proc_setflags(struct proc_handle *phdl, int mask) 110{ 111 112 if (phdl == NULL) 113 return (EINVAL); 114 115 phdl->flags |= mask; 116 117 return (0); 118} 119 120int 121proc_state(struct proc_handle *phdl) 122{ 123 124 if (phdl == NULL) 125 return (-1); 126 127 return (phdl->status); 128} 129 130pid_t 131proc_getpid(struct proc_handle *phdl) 132{ 133 134 if (phdl == NULL) 135 return (-1); 136 137 return (phdl->pid); 138} 139 140int 141proc_wstatus(struct proc_handle *phdl) 142{ 143 int status; 144 145 if (phdl == NULL) 146 return (-1); 147 if (waitpid(phdl->pid, &status, WUNTRACED) < 0) { 148 warn("waitpid"); 149 return (-1); 150 } 151 if (WIFSTOPPED(status)) 152 phdl->status = PS_STOP; 153 if (WIFEXITED(status) || WIFSIGNALED(status)) 154 phdl->status = PS_UNDEAD; 155 phdl->wstat = status; 156 157 return (phdl->status); 158} 159 160int 161proc_getwstat(struct proc_handle *phdl) 162{ 163 164 if (phdl == NULL) 165 return (-1); 166 167 return (phdl->wstat); 168} 169 170char * 171proc_signame(int sig, char *name, size_t namesz) 172{ 173 174 strlcpy(name, strsignal(sig), namesz); 175 176 return (name); 177} 178 179int 180proc_read(struct proc_handle *phdl, void *buf, size_t size, size_t addr) 181{ 182 struct ptrace_io_desc piod; 183 184 if (phdl == NULL) 185 return (-1); 186 piod.piod_op = PIOD_READ_D; 187 piod.piod_len = size; 188 piod.piod_addr = (void *)buf; 189 piod.piod_offs = (void *)addr; 190 191 if (ptrace(PT_IO, phdl->pid, (caddr_t)&piod, 0) < 0) 192 return (-1); 193 return (piod.piod_len); 194} 195 196const lwpstatus_t * 197proc_getlwpstatus(struct proc_handle *phdl) 198{ 199 struct ptrace_lwpinfo lwpinfo; 200 lwpstatus_t *psp = &phdl->lwps; 201 siginfo_t *siginfo; 202 203 if (phdl == NULL) 204 return (NULL); 205 if (ptrace(PT_LWPINFO, phdl->pid, (caddr_t)&lwpinfo, 206 sizeof(lwpinfo)) < 0) 207 return (NULL); 208 siginfo = &lwpinfo.pl_siginfo; 209 if (lwpinfo.pl_event == PL_EVENT_SIGNAL && 210 (lwpinfo.pl_flags & PL_FLAG_SI) && 211 siginfo->si_signo == SIGTRAP && 212 (siginfo->si_code == TRAP_BRKPT || 213 siginfo->si_code == TRAP_TRACE)) { 214 psp->pr_why = PR_FAULTED; 215 psp->pr_what = FLTBPT; 216 } else if (lwpinfo.pl_flags & PL_FLAG_SCE) { 217 psp->pr_why = PR_SYSENTRY; 218 } else if (lwpinfo.pl_flags & PL_FLAG_SCX) { 219 psp->pr_why = PR_SYSEXIT; 220 } 221 222 return (psp); 223} 224