1*8eb20f36SRui Paulo /* 2*8eb20f36SRui Paulo * Copyright (c) 2010 The FreeBSD Foundation 3*8eb20f36SRui Paulo * All rights reserved. 4*8eb20f36SRui Paulo * 5*8eb20f36SRui Paulo * This software was developed by Rui Paulo under sponsorship from the 6*8eb20f36SRui Paulo * FreeBSD Foundation. 7*8eb20f36SRui Paulo * 8*8eb20f36SRui Paulo * Redistribution and use in source and binary forms, with or without 9*8eb20f36SRui Paulo * modification, are permitted provided that the following conditions 10*8eb20f36SRui Paulo * are met: 11*8eb20f36SRui Paulo * 1. Redistributions of source code must retain the above copyright 12*8eb20f36SRui Paulo * notice, this list of conditions and the following disclaimer. 13*8eb20f36SRui Paulo * 2. Redistributions in binary form must reproduce the above copyright 14*8eb20f36SRui Paulo * notice, this list of conditions and the following disclaimer in the 15*8eb20f36SRui Paulo * documentation and/or other materials provided with the distribution. 16*8eb20f36SRui Paulo * 17*8eb20f36SRui Paulo * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 18*8eb20f36SRui Paulo * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 19*8eb20f36SRui Paulo * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 20*8eb20f36SRui Paulo * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 21*8eb20f36SRui Paulo * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 22*8eb20f36SRui Paulo * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 23*8eb20f36SRui Paulo * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 24*8eb20f36SRui Paulo * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 25*8eb20f36SRui Paulo * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 26*8eb20f36SRui Paulo * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 27*8eb20f36SRui Paulo * SUCH DAMAGE. 28*8eb20f36SRui Paulo */ 29*8eb20f36SRui Paulo 30*8eb20f36SRui Paulo #include <sys/cdefs.h> 31*8eb20f36SRui Paulo __FBSDID("$FreeBSD$"); 32*8eb20f36SRui Paulo 33*8eb20f36SRui Paulo #include <stdio.h> 34*8eb20f36SRui Paulo #include <string.h> 35*8eb20f36SRui Paulo #include <stdlib.h> 36*8eb20f36SRui Paulo #include <rtld_db.h> 37*8eb20f36SRui Paulo #include "libproc.h" 38*8eb20f36SRui Paulo #include "_libproc.h" 39*8eb20f36SRui Paulo 40*8eb20f36SRui Paulo static int 41*8eb20f36SRui Paulo map_iter(const rd_loadobj_t *lop, void *arg) 42*8eb20f36SRui Paulo { 43*8eb20f36SRui Paulo struct proc_handle *phdl = arg; 44*8eb20f36SRui Paulo 45*8eb20f36SRui Paulo phdl->nobjs++; 46*8eb20f36SRui Paulo if (phdl->nobjs >= phdl->rdobjsz) { 47*8eb20f36SRui Paulo phdl->rdobjsz *= 2; 48*8eb20f36SRui Paulo phdl->rdobjs = realloc(phdl->rdobjs, phdl->rdobjsz); 49*8eb20f36SRui Paulo if (phdl->rdobjs == NULL) 50*8eb20f36SRui Paulo return (-1); 51*8eb20f36SRui Paulo } 52*8eb20f36SRui Paulo memcpy(&phdl->rdobjs[phdl->nobjs++], lop, sizeof(*phdl->rdobjs)); 53*8eb20f36SRui Paulo 54*8eb20f36SRui Paulo return (0); 55*8eb20f36SRui Paulo } 56*8eb20f36SRui Paulo 57*8eb20f36SRui Paulo rd_agent_t * 58*8eb20f36SRui Paulo proc_rdagent(struct proc_handle *phdl) 59*8eb20f36SRui Paulo { 60*8eb20f36SRui Paulo if (phdl->rdap == NULL && phdl->status != PS_UNDEAD && 61*8eb20f36SRui Paulo phdl->status != PS_IDLE) { 62*8eb20f36SRui Paulo if ((phdl->rdap = rd_new(phdl)) != NULL) { 63*8eb20f36SRui Paulo phdl->rdobjs = malloc(sizeof(*phdl->rdobjs) * 64); 64*8eb20f36SRui Paulo if (phdl->rdobjs == NULL) 65*8eb20f36SRui Paulo return (phdl->rdap); 66*8eb20f36SRui Paulo rd_loadobj_iter(phdl->rdap, map_iter, phdl); 67*8eb20f36SRui Paulo } 68*8eb20f36SRui Paulo } 69*8eb20f36SRui Paulo 70*8eb20f36SRui Paulo return (phdl->rdap); 71*8eb20f36SRui Paulo } 72*8eb20f36SRui Paulo 73*8eb20f36SRui Paulo void 74*8eb20f36SRui Paulo proc_updatesyms(struct proc_handle *phdl) 75*8eb20f36SRui Paulo { 76*8eb20f36SRui Paulo memset(&phdl->rdobjs, 0, sizeof(*phdl->rdobjs) * phdl->rdobjsz); 77*8eb20f36SRui Paulo phdl->nobjs = 0; 78*8eb20f36SRui Paulo rd_loadobj_iter(phdl->rdap, map_iter, phdl); 79*8eb20f36SRui Paulo } 80