Merge from vendor: libdtrace MD parts needed by fasttrap.

Sponsored by:	The FreeBSD Foundation
This commit is contained in:
rpaulo 2010-07-17 18:31:31 +00:00
commit 583f4f5d6a
5 changed files with 1145 additions and 0 deletions

View File

@ -0,0 +1,488 @@
/*
* CDDL HEADER START
*
* The contents of this file are subject to the terms of the
* Common Development and Distribution License (the "License").
* You may not use this file except in compliance with the License.
*
* You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
* or http://www.opensolaris.org/os/licensing.
* See the License for the specific language governing permissions
* and limitations under the License.
*
* When distributing Covered Code, include this CDDL HEADER in each
* file and include the License file at usr/src/OPENSOLARIS.LICENSE.
* If applicable, add the following below this CDDL HEADER, with the
* fields enclosed by brackets "[]" replaced with your own identifying
* information: Portions Copyright [yyyy] [name of copyright owner]
*
* CDDL HEADER END
*/
/*
* Copyright 2009 Sun Microsystems, Inc. All rights reserved.
* Use is subject to license terms.
*/
#include <stdlib.h>
#include <assert.h>
#include <errno.h>
#include <string.h>
#include <libgen.h>
#include <dt_impl.h>
#include <dt_pid.h>
#include <dis_tables.h>
#define DT_POPL_EBP 0x5d
#define DT_RET 0xc3
#define DT_RET16 0xc2
#define DT_LEAVE 0xc9
#define DT_JMP32 0xe9
#define DT_JMP8 0xeb
#define DT_REP 0xf3
#define DT_MOVL_EBP_ESP 0xe58b
#define DT_ISJ32(op16) (((op16) & 0xfff0) == 0x0f80)
#define DT_ISJ8(op8) (((op8) & 0xf0) == 0x70)
#define DT_MODRM_REG(modrm) (((modrm) >> 3) & 0x7)
static int dt_instr_size(uchar_t *, dtrace_hdl_t *, pid_t, uintptr_t, char);
/*ARGSUSED*/
int
dt_pid_create_entry_probe(struct ps_prochandle *P, dtrace_hdl_t *dtp,
fasttrap_probe_spec_t *ftp, const GElf_Sym *symp)
{
ftp->ftps_type = DTFTP_ENTRY;
ftp->ftps_pc = (uintptr_t)symp->st_value;
ftp->ftps_size = (size_t)symp->st_size;
ftp->ftps_noffs = 1;
ftp->ftps_offs[0] = 0;
if (ioctl(dtp->dt_ftfd, FASTTRAPIOC_MAKEPROBE, ftp) != 0) {
dt_dprintf("fasttrap probe creation ioctl failed: %s\n",
strerror(errno));
return (dt_set_errno(dtp, errno));
}
return (1);
}
static int
dt_pid_has_jump_table(struct ps_prochandle *P, dtrace_hdl_t *dtp,
uint8_t *text, fasttrap_probe_spec_t *ftp, const GElf_Sym *symp)
{
ulong_t i;
int size;
pid_t pid = Pstatus(P)->pr_pid;
char dmodel = Pstatus(P)->pr_dmodel;
/*
* Take a pass through the function looking for a register-dependant
* jmp instruction. This could be a jump table so we have to be
* ultra conservative.
*/
for (i = 0; i < ftp->ftps_size; i += size) {
size = dt_instr_size(&text[i], dtp, pid, symp->st_value + i,
dmodel);
/*
* Assume the worst if we hit an illegal instruction.
*/
if (size <= 0) {
dt_dprintf("error at %#lx (assuming jump table)\n", i);
return (1);
}
/*
* Register-dependant jmp instructions start with a 0xff byte
* and have the modrm.reg field set to 4. They can have an
* optional REX prefix on the 64-bit ISA.
*/
if ((text[i] == 0xff && DT_MODRM_REG(text[i + 1]) == 4) ||
(dmodel == PR_MODEL_LP64 && (text[i] & 0xf0) == 0x40 &&
text[i + 1] == 0xff && DT_MODRM_REG(text[i + 2]) == 4)) {
dt_dprintf("found a suspected jump table at %s:%lx\n",
ftp->ftps_func, i);
return (1);
}
}
return (0);
}
/*ARGSUSED*/
int
dt_pid_create_return_probe(struct ps_prochandle *P, dtrace_hdl_t *dtp,
fasttrap_probe_spec_t *ftp, const GElf_Sym *symp, uint64_t *stret)
{
uint8_t *text;
ulong_t i, end;
int size;
pid_t pid = Pstatus(P)->pr_pid;
char dmodel = Pstatus(P)->pr_dmodel;
/*
* We allocate a few extra bytes at the end so we don't have to check
* for overrunning the buffer.
*/
if ((text = calloc(1, symp->st_size + 4)) == NULL) {
dt_dprintf("mr sparkle: malloc() failed\n");
return (DT_PROC_ERR);
}
if (Pread(P, text, symp->st_size, symp->st_value) != symp->st_size) {
dt_dprintf("mr sparkle: Pread() failed\n");
free(text);
return (DT_PROC_ERR);
}
ftp->ftps_type = DTFTP_RETURN;
ftp->ftps_pc = (uintptr_t)symp->st_value;
ftp->ftps_size = (size_t)symp->st_size;
ftp->ftps_noffs = 0;
/*
* If there's a jump table in the function we're only willing to
* instrument these specific (and equivalent) instruction sequences:
* leave
* [rep] ret
* and
* movl %ebp,%esp
* popl %ebp
* [rep] ret
*
* We do this to avoid accidentally interpreting jump table
* offsets as actual instructions.
*/
if (dt_pid_has_jump_table(P, dtp, text, ftp, symp)) {
for (i = 0, end = ftp->ftps_size; i < end; i += size) {
size = dt_instr_size(&text[i], dtp, pid,
symp->st_value + i, dmodel);
/* bail if we hit an invalid opcode */
if (size <= 0)
break;
if (text[i] == DT_LEAVE && text[i + 1] == DT_RET) {
dt_dprintf("leave/ret at %lx\n", i + 1);
ftp->ftps_offs[ftp->ftps_noffs++] = i + 1;
size = 2;
} else if (text[i] == DT_LEAVE &&
text[i + 1] == DT_REP && text[i + 2] == DT_RET) {
dt_dprintf("leave/rep ret at %lx\n", i + 1);
ftp->ftps_offs[ftp->ftps_noffs++] = i + 1;
size = 3;
} else if (*(uint16_t *)&text[i] == DT_MOVL_EBP_ESP &&
text[i + 2] == DT_POPL_EBP &&
text[i + 3] == DT_RET) {
dt_dprintf("movl/popl/ret at %lx\n", i + 3);
ftp->ftps_offs[ftp->ftps_noffs++] = i + 3;
size = 4;
} else if (*(uint16_t *)&text[i] == DT_MOVL_EBP_ESP &&
text[i + 2] == DT_POPL_EBP &&
text[i + 3] == DT_REP &&
text[i + 4] == DT_RET) {
dt_dprintf("movl/popl/rep ret at %lx\n", i + 3);
ftp->ftps_offs[ftp->ftps_noffs++] = i + 3;
size = 5;
}
}
} else {
for (i = 0, end = ftp->ftps_size; i < end; i += size) {
size = dt_instr_size(&text[i], dtp, pid,
symp->st_value + i, dmodel);
/* bail if we hit an invalid opcode */
if (size <= 0)
break;
/* ordinary ret */
if (size == 1 && text[i] == DT_RET)
goto is_ret;
/* two-byte ret */
if (size == 2 && text[i] == DT_REP &&
text[i + 1] == DT_RET)
goto is_ret;
/* ret <imm16> */
if (size == 3 && text[i] == DT_RET16)
goto is_ret;
/* two-byte ret <imm16> */
if (size == 4 && text[i] == DT_REP &&
text[i + 1] == DT_RET16)
goto is_ret;
/* 32-bit displacement jmp outside of the function */
if (size == 5 && text[i] == DT_JMP32 && symp->st_size <=
(uintptr_t)(i + size + *(int32_t *)&text[i + 1]))
goto is_ret;
/* 8-bit displacement jmp outside of the function */
if (size == 2 && text[i] == DT_JMP8 && symp->st_size <=
(uintptr_t)(i + size + *(int8_t *)&text[i + 1]))
goto is_ret;
/* 32-bit disp. conditional jmp outside of the func. */
if (size == 6 && DT_ISJ32(*(uint16_t *)&text[i]) &&
symp->st_size <=
(uintptr_t)(i + size + *(int32_t *)&text[i + 2]))
goto is_ret;
/* 8-bit disp. conditional jmp outside of the func. */
if (size == 2 && DT_ISJ8(text[i]) && symp->st_size <=
(uintptr_t)(i + size + *(int8_t *)&text[i + 1]))
goto is_ret;
continue;
is_ret:
dt_dprintf("return at offset %lx\n", i);
ftp->ftps_offs[ftp->ftps_noffs++] = i;
}
}
free(text);
if (ftp->ftps_noffs > 0) {
if (ioctl(dtp->dt_ftfd, FASTTRAPIOC_MAKEPROBE, ftp) != 0) {
dt_dprintf("fasttrap probe creation ioctl failed: %s\n",
strerror(errno));
return (dt_set_errno(dtp, errno));
}
}
return (ftp->ftps_noffs);
}
/*ARGSUSED*/
int
dt_pid_create_offset_probe(struct ps_prochandle *P, dtrace_hdl_t *dtp,
fasttrap_probe_spec_t *ftp, const GElf_Sym *symp, ulong_t off)
{
ftp->ftps_type = DTFTP_OFFSETS;
ftp->ftps_pc = (uintptr_t)symp->st_value;
ftp->ftps_size = (size_t)symp->st_size;
ftp->ftps_noffs = 1;
if (strcmp("-", ftp->ftps_func) == 0) {
ftp->ftps_offs[0] = off;
} else {
uint8_t *text;
ulong_t i;
int size;
pid_t pid = Pstatus(P)->pr_pid;
char dmodel = Pstatus(P)->pr_dmodel;
if ((text = malloc(symp->st_size)) == NULL) {
dt_dprintf("mr sparkle: malloc() failed\n");
return (DT_PROC_ERR);
}
if (Pread(P, text, symp->st_size, symp->st_value) !=
symp->st_size) {
dt_dprintf("mr sparkle: Pread() failed\n");
free(text);
return (DT_PROC_ERR);
}
/*
* We can't instrument offsets in functions with jump tables
* as we might interpret a jump table offset as an
* instruction.
*/
if (dt_pid_has_jump_table(P, dtp, text, ftp, symp)) {
free(text);
return (0);
}
for (i = 0; i < symp->st_size; i += size) {
if (i == off) {
ftp->ftps_offs[0] = i;
break;
}
/*
* If we've passed the desired offset without a
* match, then the given offset must not lie on a
* instruction boundary.
*/
if (i > off) {
free(text);
return (DT_PROC_ALIGN);
}
size = dt_instr_size(&text[i], dtp, pid,
symp->st_value + i, dmodel);
/*
* If we hit an invalid instruction, bail as if we
* couldn't find the offset.
*/
if (size <= 0) {
free(text);
return (DT_PROC_ALIGN);
}
}
free(text);
}
if (ioctl(dtp->dt_ftfd, FASTTRAPIOC_MAKEPROBE, ftp) != 0) {
dt_dprintf("fasttrap probe creation ioctl failed: %s\n",
strerror(errno));
return (dt_set_errno(dtp, errno));
}
return (ftp->ftps_noffs);
}
/*ARGSUSED*/
int
dt_pid_create_glob_offset_probes(struct ps_prochandle *P, dtrace_hdl_t *dtp,
fasttrap_probe_spec_t *ftp, const GElf_Sym *symp, const char *pattern)
{
uint8_t *text;
int size;
ulong_t i, end = symp->st_size;
pid_t pid = Pstatus(P)->pr_pid;
char dmodel = Pstatus(P)->pr_dmodel;
ftp->ftps_type = DTFTP_OFFSETS;
ftp->ftps_pc = (uintptr_t)symp->st_value;
ftp->ftps_size = (size_t)symp->st_size;
ftp->ftps_noffs = 0;
if ((text = malloc(symp->st_size)) == NULL) {
dt_dprintf("mr sparkle: malloc() failed\n");
return (DT_PROC_ERR);
}
if (Pread(P, text, symp->st_size, symp->st_value) != symp->st_size) {
dt_dprintf("mr sparkle: Pread() failed\n");
free(text);
return (DT_PROC_ERR);
}
/*
* We can't instrument offsets in functions with jump tables as
* we might interpret a jump table offset as an instruction.
*/
if (dt_pid_has_jump_table(P, dtp, text, ftp, symp)) {
free(text);
return (0);
}
if (strcmp("*", pattern) == 0) {
for (i = 0; i < end; i += size) {
ftp->ftps_offs[ftp->ftps_noffs++] = i;
size = dt_instr_size(&text[i], dtp, pid,
symp->st_value + i, dmodel);
/* bail if we hit an invalid opcode */
if (size <= 0)
break;
}
} else {
char name[sizeof (i) * 2 + 1];
for (i = 0; i < end; i += size) {
(void) snprintf(name, sizeof (name), "%x", i);
if (gmatch(name, pattern))
ftp->ftps_offs[ftp->ftps_noffs++] = i;
size = dt_instr_size(&text[i], dtp, pid,
symp->st_value + i, dmodel);
/* bail if we hit an invalid opcode */
if (size <= 0)
break;
}
}
free(text);
if (ftp->ftps_noffs > 0) {
if (ioctl(dtp->dt_ftfd, FASTTRAPIOC_MAKEPROBE, ftp) != 0) {
dt_dprintf("fasttrap probe creation ioctl failed: %s\n",
strerror(errno));
return (dt_set_errno(dtp, errno));
}
}
return (ftp->ftps_noffs);
}
typedef struct dtrace_dis {
uchar_t *instr;
dtrace_hdl_t *dtp;
pid_t pid;
uintptr_t addr;
} dtrace_dis_t;
static int
dt_getbyte(void *data)
{
dtrace_dis_t *dis = data;
int ret = *dis->instr;
if (ret == FASTTRAP_INSTR) {
fasttrap_instr_query_t instr;
instr.ftiq_pid = dis->pid;
instr.ftiq_pc = dis->addr;
/*
* If we hit a byte that looks like the fasttrap provider's
* trap instruction (which doubles as the breakpoint
* instruction for debuggers) we need to query the kernel
* for the real value. This may just be part of an immediate
* value so there's no need to return an error if the
* kernel doesn't know about this address.
*/
if (ioctl(dis->dtp->dt_ftfd, FASTTRAPIOC_GETINSTR, &instr) == 0)
ret = instr.ftiq_instr;
}
dis->addr++;
dis->instr++;
return (ret);
}
static int
dt_instr_size(uchar_t *instr, dtrace_hdl_t *dtp, pid_t pid, uintptr_t addr,
char dmodel)
{
dtrace_dis_t data;
dis86_t x86dis;
uint_t cpu_mode;
data.instr = instr;
data.dtp = dtp;
data.pid = pid;
data.addr = addr;
x86dis.d86_data = &data;
x86dis.d86_get_byte = dt_getbyte;
x86dis.d86_check_func = NULL;
cpu_mode = (dmodel == PR_MODEL_ILP32) ? SIZE32 : SIZE64;
if (dtrace_disx86(&x86dis, cpu_mode) != 0)
return (-1);
/*
* If the instruction was a single-byte breakpoint, there may be
* another debugger attached to this process. The original instruction
* can't be recovered so this must fail.
*/
if (x86dis.d86_len == 1 && instr[0] == FASTTRAP_INSTR)
return (-1);
return (x86dis.d86_len);
}

View File

@ -0,0 +1,117 @@
/*
* CDDL HEADER START
*
* The contents of this file are subject to the terms of the
* Common Development and Distribution License, Version 1.0 only
* (the "License"). You may not use this file except in compliance
* with the License.
*
* You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
* or http://www.opensolaris.org/os/licensing.
* See the License for the specific language governing permissions
* and limitations under the License.
*
* When distributing Covered Code, include this CDDL HEADER in each
* file and include the License file at usr/src/OPENSOLARIS.LICENSE.
* If applicable, add the following below this CDDL HEADER, with the
* fields enclosed by brackets "[]" replaced with your own identifying
* information: Portions Copyright [yyyy] [name of copyright owner]
*
* CDDL HEADER END
*/
/*
* Copyright 2004 Sun Microsystems, Inc. All rights reserved.
* Use is subject to license terms.
*/
#pragma ident "%Z%%M% %I% %E% SMI"
inline int R_GS = @GS@;
#pragma D binding "1.0" R_GS
inline int R_FS = @FS@;
#pragma D binding "1.0" R_FS
inline int R_ES = @ES@;
#pragma D binding "1.0" R_ES
inline int R_DS = @DS@;
#pragma D binding "1.0" R_DS
inline int R_EDI = @EDI@;
#pragma D binding "1.0" R_EDI
inline int R_ESI = @ESI@;
#pragma D binding "1.0" R_ESI
inline int R_EBP = @EBP@;
#pragma D binding "1.0" R_EBP
inline int R_ESP = @ESP@;
#pragma D binding "1.0" R_ESP
inline int R_EBX = @EBX@;
#pragma D binding "1.0" R_EBX
inline int R_EDX = @EDX@;
#pragma D binding "1.0" R_EDX
inline int R_ECX = @ECX@;
#pragma D binding "1.0" R_ECX
inline int R_EAX = @EAX@;
#pragma D binding "1.0" R_EAX
inline int R_TRAPNO = @TRAPNO@;
#pragma D binding "1.0" R_TRAPNO
inline int R_ERR = @ERR@;
#pragma D binding "1.0" R_ERR
inline int R_EIP = @EIP@;
#pragma D binding "1.0" R_EIP
inline int R_CS = @CS@;
#pragma D binding "1.0" R_CS
inline int R_EFL = @EFL@;
#pragma D binding "1.0" R_EFL
inline int R_UESP = @UESP@;
#pragma D binding "1.0" R_UESP
inline int R_SS = @SS@;
#pragma D binding "1.0" R_SS
inline int R_PC = R_EIP;
#pragma D binding "1.0" R_PC
inline int R_SP = R_UESP;
#pragma D binding "1.0" R_SP
inline int R_PS = R_EFL;
#pragma D binding "1.0" R_PS
inline int R_R0 = R_EAX;
#pragma D binding "1.0" R_R0
inline int R_R1 = R_EBX;
#pragma D binding "1.0" R_R1
inline int R_RSP = @REG_RSP@;
#pragma D binding "1.0" R_RSP
inline int R_RFL = @REG_RFL@;
#pragma D binding "1.0" R_RFL
inline int R_RIP = @REG_RIP@;
#pragma D binding "1.0" R_RIP
inline int R_RAX = @REG_RAX@;
#pragma D binding "1.0" R_RAX
inline int R_RCX = @REG_RCX@;
#pragma D binding "1.0" R_RCX
inline int R_RDX = @REG_RDX@;
#pragma D binding "1.0" R_RDX
inline int R_RBX = @REG_RBX@;
#pragma D binding "1.0" R_RBX
inline int R_RBP = @REG_RBP@;
#pragma D binding "1.0" R_RBP
inline int R_RSI = @REG_RSI@;
#pragma D binding "1.0" R_RSI
inline int R_RDI = @REG_RDI@;
#pragma D binding "1.0" R_RDI
inline int R_R8 = @REG_R8@;
#pragma D binding "1.0" R_R8
inline int R_R9 = @REG_R9@;
#pragma D binding "1.0" R_R9
inline int R_R10 = @REG_R10@;
#pragma D binding "1.0" R_R10
inline int R_R11 = @REG_R11@;
#pragma D binding "1.0" R_R11
inline int R_R12 = @REG_R12@;
#pragma D binding "1.0" R_R12
inline int R_R13 = @REG_R13@;
#pragma D binding "1.0" R_R13
inline int R_R14 = @REG_R14@;
#pragma D binding "1.0" R_R14
inline int R_R15 = @REG_R15@;
#pragma D binding "1.0" R_R15

View File

@ -0,0 +1,82 @@
/*
* CDDL HEADER START
*
* The contents of this file are subject to the terms of the
* Common Development and Distribution License, Version 1.0 only
* (the "License"). You may not use this file except in compliance
* with the License.
*
* You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
* or http://www.opensolaris.org/os/licensing.
* See the License for the specific language governing permissions
* and limitations under the License.
*
* When distributing Covered Code, include this CDDL HEADER in each
* file and include the License file at usr/src/OPENSOLARIS.LICENSE.
* If applicable, add the following below this CDDL HEADER, with the
* fields enclosed by brackets "[]" replaced with your own identifying
* information: Portions Copyright [yyyy] [name of copyright owner]
*
* CDDL HEADER END
*/
/*
* Copyright 2003 Sun Microsystems, Inc. All rights reserved.
* Use is subject to license terms.
*/
#pragma ident "%Z%%M% %I% %E% SMI"
/*
* This file is a sed script which is first preprocessed by cpp or cc -E to
* define a set of sed directives which replace #define tokens with their
* values. After preprocessing, whitespace is eliminated, and any @ symbols
* are translated into single space. The resulting sed script is then run
* over regs.d.in to replace the #define tokens listed below to create the
* finished regs.d. Refer to the rules in libdtrace/i386/Makefile for more
* information.
*/
#include <sys/regset.h>
#define SED_REPLACE(x) s/#x/x/g
#define SED_REPLACE64(x) s/#x/SS @+@1@+@ x/g
SED_REPLACE(GS)
SED_REPLACE(FS)
SED_REPLACE(ES)
SED_REPLACE(DS)
SED_REPLACE(EDI)
SED_REPLACE(ESI)
SED_REPLACE(EBP)
SED_REPLACE(ESP)
SED_REPLACE(EBX)
SED_REPLACE(EDX)
SED_REPLACE(ECX)
SED_REPLACE(EAX)
SED_REPLACE(TRAPNO)
SED_REPLACE(ERR)
SED_REPLACE(EIP)
SED_REPLACE(CS)
SED_REPLACE(EFL)
SED_REPLACE(UESP)
SED_REPLACE(SS)
SED_REPLACE64(REG_RSP)
SED_REPLACE64(REG_RFL)
SED_REPLACE64(REG_RIP)
SED_REPLACE64(REG_RAX)
SED_REPLACE64(REG_RCX)
SED_REPLACE64(REG_RDX)
SED_REPLACE64(REG_RBX)
SED_REPLACE64(REG_RBP)
SED_REPLACE64(REG_RSI)
SED_REPLACE64(REG_RDI)
SED_REPLACE64(REG_R8)
SED_REPLACE64(REG_R9)
SED_REPLACE64(REG_R10)
SED_REPLACE64(REG_R11)
SED_REPLACE64(REG_R12)
SED_REPLACE64(REG_R13)
SED_REPLACE64(REG_R14)
SED_REPLACE64(REG_R15)

View File

@ -0,0 +1,338 @@
/*
* CDDL HEADER START
*
* The contents of this file are subject to the terms of the
* Common Development and Distribution License, Version 1.0 only
* (the "License"). You may not use this file except in compliance
* with the License.
*
* You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
* or http://www.opensolaris.org/os/licensing.
* See the License for the specific language governing permissions
* and limitations under the License.
*
* When distributing Covered Code, include this CDDL HEADER in each
* file and include the License file at usr/src/OPENSOLARIS.LICENSE.
* If applicable, add the following below this CDDL HEADER, with the
* fields enclosed by brackets "[]" replaced with your own identifying
* information: Portions Copyright [yyyy] [name of copyright owner]
*
* CDDL HEADER END
*/
/*
* Copyright 2005 Sun Microsystems, Inc. All rights reserved.
* Use is subject to license terms.
*/
#pragma ident "%Z%%M% %I% %E% SMI"
#include <stdlib.h>
#include <assert.h>
#include <errno.h>
#include <string.h>
#include <libgen.h>
#include <dt_impl.h>
#include <dt_pid.h>
#define OP(x) ((x) >> 30)
#define OP2(x) (((x) >> 22) & 0x07)
#define COND(x) (((x) >> 25) & 0x0f)
#define A(x) (((x) >> 29) & 0x01)
#define OP_BRANCH 0
#define OP2_BPcc 0x1
#define OP2_Bicc 0x2
#define OP2_BPr 0x3
#define OP2_FBPfcc 0x5
#define OP2_FBfcc 0x6
/*ARGSUSED*/
int
dt_pid_create_entry_probe(struct ps_prochandle *P, dtrace_hdl_t *dtp,
fasttrap_probe_spec_t *ftp, const GElf_Sym *symp)
{
ftp->ftps_type = DTFTP_ENTRY;
ftp->ftps_pc = (uintptr_t)symp->st_value;
ftp->ftps_size = (size_t)symp->st_size;
ftp->ftps_noffs = 1;
ftp->ftps_offs[0] = 0;
if (ioctl(dtp->dt_ftfd, FASTTRAPIOC_MAKEPROBE, ftp) != 0) {
dt_dprintf("fasttrap probe creation ioctl failed: %s\n",
strerror(errno));
return (dt_set_errno(dtp, errno));
}
return (1);
}
int
dt_pid_create_return_probe(struct ps_prochandle *P, dtrace_hdl_t *dtp,
fasttrap_probe_spec_t *ftp, const GElf_Sym *symp, uint64_t *stret)
{
uint32_t *text;
int i;
int srdepth = 0;
if ((text = malloc(symp->st_size + 4)) == NULL) {
dt_dprintf("mr sparkle: malloc() failed\n");
return (DT_PROC_ERR);
}
if (Pread(P, text, symp->st_size, symp->st_value) != symp->st_size) {
dt_dprintf("mr sparkle: Pread() failed\n");
free(text);
return (DT_PROC_ERR);
}
/*
* Leave a dummy instruction in the last slot to simplify edge
* conditions.
*/
text[symp->st_size / 4] = 0;
ftp->ftps_type = DTFTP_RETURN;
ftp->ftps_pc = symp->st_value;
ftp->ftps_size = symp->st_size;
ftp->ftps_noffs = 0;
for (i = 0; i < symp->st_size / 4; i++) {
/*
* If we encounter an existing tracepoint, query the
* kernel to find out the instruction that was
* replaced at this spot.
*/
while (text[i] == FASTTRAP_INSTR) {
fasttrap_instr_query_t instr;
instr.ftiq_pid = Pstatus(P)->pr_pid;
instr.ftiq_pc = symp->st_value + i * 4;
if (ioctl(dtp->dt_ftfd, FASTTRAPIOC_GETINSTR,
&instr) != 0) {
if (errno == ESRCH || errno == ENOENT) {
if (Pread(P, &text[i], 4,
instr.ftiq_pc) != 4) {
dt_dprintf("mr sparkle: "
"Pread() failed\n");
free(text);
return (DT_PROC_ERR);
}
continue;
}
free(text);
dt_dprintf("mr sparkle: getinstr query "
"failed: %s\n", strerror(errno));
return (DT_PROC_ERR);
}
text[i] = instr.ftiq_instr;
break;
}
/* save */
if ((text[i] & 0xc1f80000) == 0x81e00000) {
srdepth++;
continue;
}
/* restore */
if ((text[i] & 0xc1f80000) == 0x81e80000) {
srdepth--;
continue;
}
if (srdepth > 0) {
/* ret */
if (text[i] == 0x81c7e008)
goto is_ret;
/* return */
if (text[i] == 0x81cfe008)
goto is_ret;
/* call or jmpl w/ restore in the slot */
if (((text[i] & 0xc0000000) == 0x40000000 ||
(text[i] & 0xc1f80000) == 0x81c00000) &&
(text[i + 1] & 0xc1f80000) == 0x81e80000)
goto is_ret;
/* call to one of the stret routines */
if ((text[i] & 0xc0000000) == 0x40000000) {
int32_t disp = text[i] << 2;
uint64_t dest = ftp->ftps_pc + i * 4 + disp;
dt_dprintf("dest = %llx\n", (u_longlong_t)dest);
if (dest == stret[0] || dest == stret[1] ||
dest == stret[2] || dest == stret[3])
goto is_ret;
}
} else {
/* external call */
if ((text[i] & 0xc0000000) == 0x40000000) {
int32_t dst = text[i] << 2;
dst += i * 4;
if ((uintptr_t)dst >= (uintptr_t)symp->st_size)
goto is_ret;
}
/* jmpl into %g0 -- this includes the retl pseudo op */
if ((text[i] & 0xfff80000) == 0x81c00000)
goto is_ret;
/* external branch -- possible return site */
if (OP(text[i]) == OP_BRANCH) {
int32_t dst;
int baa;
switch (OP2(text[i])) {
case OP2_BPcc:
dst = text[i] & 0x7ffff;
dst <<= 13;
dst >>= 11;
baa = COND(text[i]) == 8 && A(text[i]);
break;
case OP2_Bicc:
dst = text[i] & 0x3fffff;
dst <<= 10;
dst >>= 8;
baa = COND(text[i]) == 8 && A(text[i]);
break;
case OP2_BPr:
dst = (((text[i]) >> 6) & 0xc000) |
((text[i]) & 0x3fff);
dst <<= 16;
dst >>= 14;
baa = 0;
break;
case OP2_FBPfcc:
dst = text[i] & 0x7ffff;
dst <<= 13;
dst >>= 11;
baa = COND(text[i]) == 8 && A(text[i]);
break;
case OP2_FBfcc:
dst = text[i] & 0x3fffff;
dst <<= 10;
dst >>= 8;
baa = COND(text[i]) == 8 && A(text[i]);
break;
default:
continue;
}
dst += i * 4;
/*
* Interpret branches outside of the function's
* bounds as potential return sites. If the
* branch is a ba,a don't skip the instruction
* in the delay slot.
*/
if ((uintptr_t)dst >=
(uintptr_t)symp->st_size) {
if (baa)
goto is_ret_baa;
else
goto is_ret;
}
}
}
continue;
is_ret:
i++;
is_ret_baa:
dt_dprintf("return at offset %x\n", i * 4);
ftp->ftps_offs[ftp->ftps_noffs++] = i * 4;
}
free(text);
if (ftp->ftps_noffs > 0) {
if (ioctl(dtp->dt_ftfd, FASTTRAPIOC_MAKEPROBE, ftp) != 0) {
dt_dprintf("fasttrap probe creation ioctl failed: %s\n",
strerror(errno));
return (dt_set_errno(dtp, errno));
}
}
return (ftp->ftps_noffs);
}
/*ARGSUSED*/
int
dt_pid_create_offset_probe(struct ps_prochandle *P, dtrace_hdl_t *dtp,
fasttrap_probe_spec_t *ftp, const GElf_Sym *symp, ulong_t off)
{
if (off & 0x3)
return (DT_PROC_ALIGN);
ftp->ftps_type = DTFTP_OFFSETS;
ftp->ftps_pc = (uintptr_t)symp->st_value;
ftp->ftps_size = (size_t)symp->st_size;
ftp->ftps_noffs = 1;
ftp->ftps_offs[0] = off;
if (ioctl(dtp->dt_ftfd, FASTTRAPIOC_MAKEPROBE, ftp) != 0) {
dt_dprintf("fasttrap probe creation ioctl failed: %s\n",
strerror(errno));
return (dt_set_errno(dtp, errno));
}
return (1);
}
/*ARGSUSED*/
int
dt_pid_create_glob_offset_probes(struct ps_prochandle *P, dtrace_hdl_t *dtp,
fasttrap_probe_spec_t *ftp, const GElf_Sym *symp, const char *pattern)
{
ulong_t i;
ftp->ftps_type = DTFTP_OFFSETS;
ftp->ftps_pc = (uintptr_t)symp->st_value;
ftp->ftps_size = (size_t)symp->st_size;
ftp->ftps_noffs = 0;
/*
* If we're matching against everything, just iterate through each
* instruction in the function, otherwise look for matching offset
* names by constructing the string and comparing it against the
* pattern.
*/
if (strcmp("*", pattern) == 0) {
for (i = 0; i < symp->st_size; i += 4) {
ftp->ftps_offs[ftp->ftps_noffs++] = i;
}
} else {
char name[sizeof (i) * 2 + 1];
for (i = 0; i < symp->st_size; i += 4) {
(void) sprintf(name, "%lx", i);
if (gmatch(name, pattern))
ftp->ftps_offs[ftp->ftps_noffs++] = i;
}
}
if (ioctl(dtp->dt_ftfd, FASTTRAPIOC_MAKEPROBE, ftp) != 0) {
dt_dprintf("fasttrap probe creation ioctl failed: %s\n",
strerror(errno));
return (dt_set_errno(dtp, errno));
}
return (ftp->ftps_noffs);
}

View File

@ -0,0 +1,120 @@
/*
* CDDL HEADER START
*
* The contents of this file are subject to the terms of the
* Common Development and Distribution License, Version 1.0 only
* (the "License"). You may not use this file except in compliance
* with the License.
*
* You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
* or http://www.opensolaris.org/os/licensing.
* See the License for the specific language governing permissions
* and limitations under the License.
*
* When distributing Covered Code, include this CDDL HEADER in each
* file and include the License file at usr/src/OPENSOLARIS.LICENSE.
* If applicable, add the following below this CDDL HEADER, with the
* fields enclosed by brackets "[]" replaced with your own identifying
* information: Portions Copyright [yyyy] [name of copyright owner]
*
* CDDL HEADER END
*/
/*
* Copyright 2003 Sun Microsystems, Inc. All rights reserved.
* Use is subject to license terms.
*/
#pragma ident "%Z%%M% %I% %E% SMI"
inline int R_G0 = 0;
#pragma D binding "1.0" R_G0
inline int R_G1 = 1;
#pragma D binding "1.0" R_G1
inline int R_G2 = 2;
#pragma D binding "1.0" R_G2
inline int R_G3 = 3;
#pragma D binding "1.0" R_G3
inline int R_G4 = 4;
#pragma D binding "1.0" R_G4
inline int R_G5 = 5;
#pragma D binding "1.0" R_G5
inline int R_G6 = 6;
#pragma D binding "1.0" R_G6
inline int R_G7 = 7;
#pragma D binding "1.0" R_G7
inline int R_O0 = 8;
#pragma D binding "1.0" R_O0
inline int R_O1 = 9;
#pragma D binding "1.0" R_O1
inline int R_O2 = 10;
#pragma D binding "1.0" R_O2
inline int R_O3 = 11;
#pragma D binding "1.0" R_O3
inline int R_O4 = 12;
#pragma D binding "1.0" R_O4
inline int R_O5 = 13;
#pragma D binding "1.0" R_O5
inline int R_O6 = 14;
#pragma D binding "1.0" R_O6
inline int R_O7 = 15;
#pragma D binding "1.0" R_O7
inline int R_L0 = 16;
#pragma D binding "1.0" R_L0
inline int R_L1 = 17;
#pragma D binding "1.0" R_L1
inline int R_L2 = 18;
#pragma D binding "1.0" R_L2
inline int R_L3 = 19;
#pragma D binding "1.0" R_L3
inline int R_L4 = 20;
#pragma D binding "1.0" R_L4
inline int R_L5 = 21;
#pragma D binding "1.0" R_L5
inline int R_L6 = 22;
#pragma D binding "1.0" R_L6
inline int R_L7 = 23;
#pragma D binding "1.0" R_L7
inline int R_I0 = 24;
#pragma D binding "1.0" R_I0
inline int R_I1 = 25;
#pragma D binding "1.0" R_I1
inline int R_I2 = 26;
#pragma D binding "1.0" R_I2
inline int R_I3 = 27;
#pragma D binding "1.0" R_I3
inline int R_I4 = 28;
#pragma D binding "1.0" R_I4
inline int R_I5 = 29;
#pragma D binding "1.0" R_I5
inline int R_I6 = 30;
#pragma D binding "1.0" R_I6
inline int R_I7 = 31;
#pragma D binding "1.0" R_I7
inline int R_CCR = 32;
#pragma D binding "1.0" R_CCR
inline int R_PC = 33;
#pragma D binding "1.0" R_PC
inline int R_nPC = 34;
#pragma D binding "1.0" R_nPC
inline int R_NPC = R_nPC;
#pragma D binding "1.0" R_NPC
inline int R_Y = 35;
#pragma D binding "1.0" R_Y
inline int R_ASI = 36;
#pragma D binding "1.0" R_ASI
inline int R_FPRS = 37;
#pragma D binding "1.0" R_FPRS
inline int R_PS = R_CCR;
#pragma D binding "1.0" R_PS
inline int R_SP = R_O6;
#pragma D binding "1.0" R_SP
inline int R_FP = R_I6;
#pragma D binding "1.0" R_FP
inline int R_R0 = R_O0;
#pragma D binding "1.0" R_R0
inline int R_R1 = R_O1;
#pragma D binding "1.0" R_R1