17c478bd9Sstevel@tonic-gate /* 27c478bd9Sstevel@tonic-gate * CDDL HEADER START 37c478bd9Sstevel@tonic-gate * 47c478bd9Sstevel@tonic-gate * The contents of this file are subject to the terms of the 5ac448965Sahl * Common Development and Distribution License (the "License"). 6ac448965Sahl * You may not use this file except in compliance with the License. 77c478bd9Sstevel@tonic-gate * 87c478bd9Sstevel@tonic-gate * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE 97c478bd9Sstevel@tonic-gate * or http://www.opensolaris.org/os/licensing. 107c478bd9Sstevel@tonic-gate * See the License for the specific language governing permissions 117c478bd9Sstevel@tonic-gate * and limitations under the License. 127c478bd9Sstevel@tonic-gate * 137c478bd9Sstevel@tonic-gate * When distributing Covered Code, include this CDDL HEADER in each 147c478bd9Sstevel@tonic-gate * file and include the License file at usr/src/OPENSOLARIS.LICENSE. 157c478bd9Sstevel@tonic-gate * If applicable, add the following below this CDDL HEADER, with the 167c478bd9Sstevel@tonic-gate * fields enclosed by brackets "[]" replaced with your own identifying 177c478bd9Sstevel@tonic-gate * information: Portions Copyright [yyyy] [name of copyright owner] 187c478bd9Sstevel@tonic-gate * 197c478bd9Sstevel@tonic-gate * CDDL HEADER END 207c478bd9Sstevel@tonic-gate */ 21900524f3Sahl 227c478bd9Sstevel@tonic-gate /* 23*b9e93c10SJonathan Haslam * Copyright 2009 Sun Microsystems, Inc. All rights reserved. 247c478bd9Sstevel@tonic-gate * Use is subject to license terms. 257c478bd9Sstevel@tonic-gate */ 267c478bd9Sstevel@tonic-gate 277c478bd9Sstevel@tonic-gate #include <unistd.h> 287c478bd9Sstevel@tonic-gate #include <strings.h> 297c478bd9Sstevel@tonic-gate #include <stdlib.h> 307c478bd9Sstevel@tonic-gate #include <errno.h> 317c478bd9Sstevel@tonic-gate #include <assert.h> 32900524f3Sahl #include <ctype.h> 33900524f3Sahl #include <alloca.h> 347c478bd9Sstevel@tonic-gate 357c478bd9Sstevel@tonic-gate #include <dt_impl.h> 361a7c1b72Smws #include <dt_program.h> 377c478bd9Sstevel@tonic-gate #include <dt_printf.h> 38900524f3Sahl #include <dt_provider.h> 397c478bd9Sstevel@tonic-gate 407c478bd9Sstevel@tonic-gate dtrace_prog_t * 411a7c1b72Smws dt_program_create(dtrace_hdl_t *dtp) 427c478bd9Sstevel@tonic-gate { 431a7c1b72Smws dtrace_prog_t *pgp = dt_zalloc(dtp, sizeof (dtrace_prog_t)); 447c478bd9Sstevel@tonic-gate 457c478bd9Sstevel@tonic-gate if (pgp != NULL) 467c478bd9Sstevel@tonic-gate dt_list_append(&dtp->dt_programs, pgp); 477c478bd9Sstevel@tonic-gate else 487c478bd9Sstevel@tonic-gate (void) dt_set_errno(dtp, EDT_NOMEM); 497c478bd9Sstevel@tonic-gate 50ac448965Sahl /* 51ac448965Sahl * By default, programs start with DOF version 1 so that output files 52ac448965Sahl * containing DOF are backward compatible. If a program requires new 53ac448965Sahl * DOF features, the version is increased as needed. 54ac448965Sahl */ 55ac448965Sahl pgp->dp_dofversion = DOF_VERSION_1; 56ac448965Sahl 577c478bd9Sstevel@tonic-gate return (pgp); 587c478bd9Sstevel@tonic-gate } 597c478bd9Sstevel@tonic-gate 607c478bd9Sstevel@tonic-gate void 611a7c1b72Smws dt_program_destroy(dtrace_hdl_t *dtp, dtrace_prog_t *pgp) 627c478bd9Sstevel@tonic-gate { 637c478bd9Sstevel@tonic-gate dt_stmt_t *stp, *next; 641a7c1b72Smws uint_t i; 657c478bd9Sstevel@tonic-gate 667c478bd9Sstevel@tonic-gate for (stp = dt_list_next(&pgp->dp_stmts); stp != NULL; stp = next) { 677c478bd9Sstevel@tonic-gate next = dt_list_next(stp); 681a7c1b72Smws dtrace_stmt_destroy(dtp, stp->ds_desc); 691a7c1b72Smws dt_free(dtp, stp); 707c478bd9Sstevel@tonic-gate } 717c478bd9Sstevel@tonic-gate 721a7c1b72Smws for (i = 0; i < pgp->dp_xrefslen; i++) 731a7c1b72Smws dt_free(dtp, pgp->dp_xrefs[i]); 741a7c1b72Smws 751a7c1b72Smws dt_free(dtp, pgp->dp_xrefs); 767c478bd9Sstevel@tonic-gate dt_list_delete(&dtp->dt_programs, pgp); 771a7c1b72Smws dt_free(dtp, pgp); 787c478bd9Sstevel@tonic-gate } 797c478bd9Sstevel@tonic-gate 807c478bd9Sstevel@tonic-gate /*ARGSUSED*/ 817c478bd9Sstevel@tonic-gate void 827c478bd9Sstevel@tonic-gate dtrace_program_info(dtrace_hdl_t *dtp, dtrace_prog_t *pgp, 837c478bd9Sstevel@tonic-gate dtrace_proginfo_t *pip) 847c478bd9Sstevel@tonic-gate { 857c478bd9Sstevel@tonic-gate dt_stmt_t *stp; 867c478bd9Sstevel@tonic-gate dtrace_actdesc_t *ap; 877c478bd9Sstevel@tonic-gate dtrace_ecbdesc_t *last = NULL; 887c478bd9Sstevel@tonic-gate 897c478bd9Sstevel@tonic-gate if (pip == NULL) 907c478bd9Sstevel@tonic-gate return; 917c478bd9Sstevel@tonic-gate 927c478bd9Sstevel@tonic-gate bzero(pip, sizeof (dtrace_proginfo_t)); 937c478bd9Sstevel@tonic-gate 947c478bd9Sstevel@tonic-gate if (dt_list_next(&pgp->dp_stmts) != NULL) { 957c478bd9Sstevel@tonic-gate pip->dpi_descattr = _dtrace_maxattr; 967c478bd9Sstevel@tonic-gate pip->dpi_stmtattr = _dtrace_maxattr; 977c478bd9Sstevel@tonic-gate } else { 987c478bd9Sstevel@tonic-gate pip->dpi_descattr = _dtrace_defattr; 997c478bd9Sstevel@tonic-gate pip->dpi_stmtattr = _dtrace_defattr; 1007c478bd9Sstevel@tonic-gate } 1017c478bd9Sstevel@tonic-gate 1027c478bd9Sstevel@tonic-gate for (stp = dt_list_next(&pgp->dp_stmts); stp; stp = dt_list_next(stp)) { 1037c478bd9Sstevel@tonic-gate dtrace_ecbdesc_t *edp = stp->ds_desc->dtsd_ecbdesc; 1047c478bd9Sstevel@tonic-gate 1057c478bd9Sstevel@tonic-gate if (edp == last) 1067c478bd9Sstevel@tonic-gate continue; 1077c478bd9Sstevel@tonic-gate last = edp; 1087c478bd9Sstevel@tonic-gate 1097c478bd9Sstevel@tonic-gate pip->dpi_descattr = 1107c478bd9Sstevel@tonic-gate dt_attr_min(stp->ds_desc->dtsd_descattr, pip->dpi_descattr); 1117c478bd9Sstevel@tonic-gate 1127c478bd9Sstevel@tonic-gate pip->dpi_stmtattr = 1137c478bd9Sstevel@tonic-gate dt_attr_min(stp->ds_desc->dtsd_stmtattr, pip->dpi_stmtattr); 1147c478bd9Sstevel@tonic-gate 1157c478bd9Sstevel@tonic-gate /* 1167c478bd9Sstevel@tonic-gate * If there aren't any actions, account for the fact that 1177c478bd9Sstevel@tonic-gate * recording the epid will generate a record. 1187c478bd9Sstevel@tonic-gate */ 1197c478bd9Sstevel@tonic-gate if (edp->dted_action == NULL) 1207c478bd9Sstevel@tonic-gate pip->dpi_recgens++; 1217c478bd9Sstevel@tonic-gate 1227c478bd9Sstevel@tonic-gate for (ap = edp->dted_action; ap != NULL; ap = ap->dtad_next) { 1237c478bd9Sstevel@tonic-gate if (ap->dtad_kind == DTRACEACT_SPECULATE) { 1247c478bd9Sstevel@tonic-gate pip->dpi_speculations++; 1257c478bd9Sstevel@tonic-gate continue; 1267c478bd9Sstevel@tonic-gate } 1277c478bd9Sstevel@tonic-gate 1287c478bd9Sstevel@tonic-gate if (DTRACEACT_ISAGG(ap->dtad_kind)) { 1297c478bd9Sstevel@tonic-gate pip->dpi_recgens -= ap->dtad_arg; 1307c478bd9Sstevel@tonic-gate pip->dpi_aggregates++; 1317c478bd9Sstevel@tonic-gate continue; 1327c478bd9Sstevel@tonic-gate } 1337c478bd9Sstevel@tonic-gate 1347c478bd9Sstevel@tonic-gate if (DTRACEACT_ISDESTRUCTIVE(ap->dtad_kind)) 1357c478bd9Sstevel@tonic-gate continue; 1367c478bd9Sstevel@tonic-gate 1377c478bd9Sstevel@tonic-gate if (ap->dtad_kind == DTRACEACT_DIFEXPR && 1387c478bd9Sstevel@tonic-gate ap->dtad_difo->dtdo_rtype.dtdt_kind == 1397c478bd9Sstevel@tonic-gate DIF_TYPE_CTF && 1407c478bd9Sstevel@tonic-gate ap->dtad_difo->dtdo_rtype.dtdt_size == 0) 1417c478bd9Sstevel@tonic-gate continue; 1427c478bd9Sstevel@tonic-gate 1437c478bd9Sstevel@tonic-gate pip->dpi_recgens++; 1447c478bd9Sstevel@tonic-gate } 1457c478bd9Sstevel@tonic-gate } 1467c478bd9Sstevel@tonic-gate } 1477c478bd9Sstevel@tonic-gate 1487c478bd9Sstevel@tonic-gate int 1497c478bd9Sstevel@tonic-gate dtrace_program_exec(dtrace_hdl_t *dtp, dtrace_prog_t *pgp, 1507c478bd9Sstevel@tonic-gate dtrace_proginfo_t *pip) 1517c478bd9Sstevel@tonic-gate { 1527c478bd9Sstevel@tonic-gate void *dof; 1537c478bd9Sstevel@tonic-gate int n, err; 1547c478bd9Sstevel@tonic-gate 1557c478bd9Sstevel@tonic-gate dtrace_program_info(dtp, pgp, pip); 1567c478bd9Sstevel@tonic-gate 1577c478bd9Sstevel@tonic-gate if ((dof = dtrace_dof_create(dtp, pgp, DTRACE_D_STRIP)) == NULL) 1587c478bd9Sstevel@tonic-gate return (-1); 1597c478bd9Sstevel@tonic-gate 1607c478bd9Sstevel@tonic-gate n = dt_ioctl(dtp, DTRACEIOC_ENABLE, dof); 1617c478bd9Sstevel@tonic-gate dtrace_dof_destroy(dtp, dof); 1627c478bd9Sstevel@tonic-gate 1637c478bd9Sstevel@tonic-gate if (n == -1) { 1647c478bd9Sstevel@tonic-gate switch (errno) { 1657c478bd9Sstevel@tonic-gate case EINVAL: 1667c478bd9Sstevel@tonic-gate err = EDT_DIFINVAL; 1677c478bd9Sstevel@tonic-gate break; 1687c478bd9Sstevel@tonic-gate case EFAULT: 1697c478bd9Sstevel@tonic-gate err = EDT_DIFFAULT; 1707c478bd9Sstevel@tonic-gate break; 1717c478bd9Sstevel@tonic-gate case E2BIG: 1727c478bd9Sstevel@tonic-gate err = EDT_DIFSIZE; 1737c478bd9Sstevel@tonic-gate break; 174*b9e93c10SJonathan Haslam case EBUSY: 175*b9e93c10SJonathan Haslam err = EDT_ENABLING_ERR; 176*b9e93c10SJonathan Haslam break; 1777c478bd9Sstevel@tonic-gate default: 1787c478bd9Sstevel@tonic-gate err = errno; 1797c478bd9Sstevel@tonic-gate } 1807c478bd9Sstevel@tonic-gate 1817c478bd9Sstevel@tonic-gate return (dt_set_errno(dtp, err)); 1827c478bd9Sstevel@tonic-gate } 1837c478bd9Sstevel@tonic-gate 1847c478bd9Sstevel@tonic-gate if (pip != NULL) 1857c478bd9Sstevel@tonic-gate pip->dpi_matches += n; 1867c478bd9Sstevel@tonic-gate 1877c478bd9Sstevel@tonic-gate return (0); 1887c478bd9Sstevel@tonic-gate } 1897c478bd9Sstevel@tonic-gate 1901a7c1b72Smws static void 1911a7c1b72Smws dt_ecbdesc_hold(dtrace_ecbdesc_t *edp) 1927c478bd9Sstevel@tonic-gate { 1937c478bd9Sstevel@tonic-gate edp->dted_refcnt++; 1947c478bd9Sstevel@tonic-gate } 1957c478bd9Sstevel@tonic-gate 1967c478bd9Sstevel@tonic-gate void 1971a7c1b72Smws dt_ecbdesc_release(dtrace_hdl_t *dtp, dtrace_ecbdesc_t *edp) 1987c478bd9Sstevel@tonic-gate { 1997c478bd9Sstevel@tonic-gate if (--edp->dted_refcnt > 0) 2007c478bd9Sstevel@tonic-gate return; 2017c478bd9Sstevel@tonic-gate 2021a7c1b72Smws dt_difo_free(dtp, edp->dted_pred.dtpdd_difo); 2037c478bd9Sstevel@tonic-gate assert(edp->dted_action == NULL); 2041a7c1b72Smws dt_free(dtp, edp); 2057c478bd9Sstevel@tonic-gate } 2067c478bd9Sstevel@tonic-gate 2077c478bd9Sstevel@tonic-gate dtrace_ecbdesc_t * 2081a7c1b72Smws dt_ecbdesc_create(dtrace_hdl_t *dtp, const dtrace_probedesc_t *pdp) 2097c478bd9Sstevel@tonic-gate { 2107c478bd9Sstevel@tonic-gate dtrace_ecbdesc_t *edp; 2117c478bd9Sstevel@tonic-gate 2121a7c1b72Smws if ((edp = dt_zalloc(dtp, sizeof (dtrace_ecbdesc_t))) == NULL) { 2137c478bd9Sstevel@tonic-gate (void) dt_set_errno(dtp, EDT_NOMEM); 2147c478bd9Sstevel@tonic-gate return (NULL); 2157c478bd9Sstevel@tonic-gate } 2167c478bd9Sstevel@tonic-gate 2177c478bd9Sstevel@tonic-gate edp->dted_probe = *pdp; 2181a7c1b72Smws dt_ecbdesc_hold(edp); 2197c478bd9Sstevel@tonic-gate return (edp); 2207c478bd9Sstevel@tonic-gate } 2217c478bd9Sstevel@tonic-gate 2227c478bd9Sstevel@tonic-gate dtrace_stmtdesc_t * 2237c478bd9Sstevel@tonic-gate dtrace_stmt_create(dtrace_hdl_t *dtp, dtrace_ecbdesc_t *edp) 2247c478bd9Sstevel@tonic-gate { 2257c478bd9Sstevel@tonic-gate dtrace_stmtdesc_t *sdp; 2267c478bd9Sstevel@tonic-gate 2271a7c1b72Smws if ((sdp = dt_zalloc(dtp, sizeof (dtrace_stmtdesc_t))) == NULL) 2287c478bd9Sstevel@tonic-gate return (NULL); 2297c478bd9Sstevel@tonic-gate 2301a7c1b72Smws dt_ecbdesc_hold(edp); 2317c478bd9Sstevel@tonic-gate sdp->dtsd_ecbdesc = edp; 2327c478bd9Sstevel@tonic-gate sdp->dtsd_descattr = _dtrace_defattr; 2337c478bd9Sstevel@tonic-gate sdp->dtsd_stmtattr = _dtrace_defattr; 2347c478bd9Sstevel@tonic-gate 2357c478bd9Sstevel@tonic-gate return (sdp); 2367c478bd9Sstevel@tonic-gate } 2377c478bd9Sstevel@tonic-gate 2387c478bd9Sstevel@tonic-gate dtrace_actdesc_t * 2397c478bd9Sstevel@tonic-gate dtrace_stmt_action(dtrace_hdl_t *dtp, dtrace_stmtdesc_t *sdp) 2407c478bd9Sstevel@tonic-gate { 2417c478bd9Sstevel@tonic-gate dtrace_actdesc_t *new; 2427c478bd9Sstevel@tonic-gate dtrace_ecbdesc_t *edp = sdp->dtsd_ecbdesc; 2437c478bd9Sstevel@tonic-gate 2441a7c1b72Smws if ((new = dt_alloc(dtp, sizeof (dtrace_actdesc_t))) == NULL) 2457c478bd9Sstevel@tonic-gate return (NULL); 2467c478bd9Sstevel@tonic-gate 2477c478bd9Sstevel@tonic-gate if (sdp->dtsd_action_last != NULL) { 2487c478bd9Sstevel@tonic-gate assert(sdp->dtsd_action != NULL); 2497c478bd9Sstevel@tonic-gate assert(sdp->dtsd_action_last->dtad_next == NULL); 2507c478bd9Sstevel@tonic-gate sdp->dtsd_action_last->dtad_next = new; 2517c478bd9Sstevel@tonic-gate } else { 2527c478bd9Sstevel@tonic-gate dtrace_actdesc_t *ap = edp->dted_action; 2537c478bd9Sstevel@tonic-gate 2547c478bd9Sstevel@tonic-gate assert(sdp->dtsd_action == NULL); 2557c478bd9Sstevel@tonic-gate sdp->dtsd_action = new; 2567c478bd9Sstevel@tonic-gate 2577c478bd9Sstevel@tonic-gate while (ap != NULL && ap->dtad_next != NULL) 2587c478bd9Sstevel@tonic-gate ap = ap->dtad_next; 2597c478bd9Sstevel@tonic-gate 2607c478bd9Sstevel@tonic-gate if (ap == NULL) 2617c478bd9Sstevel@tonic-gate edp->dted_action = new; 2627c478bd9Sstevel@tonic-gate else 2637c478bd9Sstevel@tonic-gate ap->dtad_next = new; 2647c478bd9Sstevel@tonic-gate } 2657c478bd9Sstevel@tonic-gate 2667c478bd9Sstevel@tonic-gate sdp->dtsd_action_last = new; 2677c478bd9Sstevel@tonic-gate bzero(new, sizeof (dtrace_actdesc_t)); 2687c478bd9Sstevel@tonic-gate new->dtad_uarg = (uintptr_t)sdp; 2697c478bd9Sstevel@tonic-gate 2707c478bd9Sstevel@tonic-gate return (new); 2717c478bd9Sstevel@tonic-gate } 2727c478bd9Sstevel@tonic-gate 2737c478bd9Sstevel@tonic-gate int 2747c478bd9Sstevel@tonic-gate dtrace_stmt_add(dtrace_hdl_t *dtp, dtrace_prog_t *pgp, dtrace_stmtdesc_t *sdp) 2757c478bd9Sstevel@tonic-gate { 2761a7c1b72Smws dt_stmt_t *stp = dt_alloc(dtp, sizeof (dt_stmt_t)); 2777c478bd9Sstevel@tonic-gate 2787c478bd9Sstevel@tonic-gate if (stp == NULL) 2791a7c1b72Smws return (-1); /* errno is set for us */ 2807c478bd9Sstevel@tonic-gate 2817c478bd9Sstevel@tonic-gate dt_list_append(&pgp->dp_stmts, stp); 2827c478bd9Sstevel@tonic-gate stp->ds_desc = sdp; 2837c478bd9Sstevel@tonic-gate 2847c478bd9Sstevel@tonic-gate return (0); 2857c478bd9Sstevel@tonic-gate } 2867c478bd9Sstevel@tonic-gate 2877c478bd9Sstevel@tonic-gate int 2887c478bd9Sstevel@tonic-gate dtrace_stmt_iter(dtrace_hdl_t *dtp, dtrace_prog_t *pgp, 2897c478bd9Sstevel@tonic-gate dtrace_stmt_f *func, void *data) 2907c478bd9Sstevel@tonic-gate { 2917c478bd9Sstevel@tonic-gate dt_stmt_t *stp, *next; 2927c478bd9Sstevel@tonic-gate int status = 0; 2937c478bd9Sstevel@tonic-gate 2947c478bd9Sstevel@tonic-gate for (stp = dt_list_next(&pgp->dp_stmts); stp != NULL; stp = next) { 2957c478bd9Sstevel@tonic-gate next = dt_list_next(stp); 2967c478bd9Sstevel@tonic-gate if ((status = func(dtp, pgp, stp->ds_desc, data)) != 0) 2977c478bd9Sstevel@tonic-gate break; 2987c478bd9Sstevel@tonic-gate } 2997c478bd9Sstevel@tonic-gate 3007c478bd9Sstevel@tonic-gate return (status); 3017c478bd9Sstevel@tonic-gate } 3027c478bd9Sstevel@tonic-gate 3037c478bd9Sstevel@tonic-gate void 3041a7c1b72Smws dtrace_stmt_destroy(dtrace_hdl_t *dtp, dtrace_stmtdesc_t *sdp) 3057c478bd9Sstevel@tonic-gate { 3067c478bd9Sstevel@tonic-gate dtrace_ecbdesc_t *edp = sdp->dtsd_ecbdesc; 3077c478bd9Sstevel@tonic-gate 3087c478bd9Sstevel@tonic-gate /* 3097c478bd9Sstevel@tonic-gate * We need to remove any actions that we have on this ECB, and 3107c478bd9Sstevel@tonic-gate * remove our hold on the ECB itself. 3117c478bd9Sstevel@tonic-gate */ 3127c478bd9Sstevel@tonic-gate if (sdp->dtsd_action != NULL) { 3137c478bd9Sstevel@tonic-gate dtrace_actdesc_t *last = sdp->dtsd_action_last; 3147c478bd9Sstevel@tonic-gate dtrace_actdesc_t *ap, *next; 3157c478bd9Sstevel@tonic-gate 3167c478bd9Sstevel@tonic-gate assert(last != NULL); 3177c478bd9Sstevel@tonic-gate 3187c478bd9Sstevel@tonic-gate for (ap = edp->dted_action; ap != NULL; ap = ap->dtad_next) { 3197c478bd9Sstevel@tonic-gate if (ap == sdp->dtsd_action) 3207c478bd9Sstevel@tonic-gate break; 3217c478bd9Sstevel@tonic-gate 3227c478bd9Sstevel@tonic-gate if (ap->dtad_next == sdp->dtsd_action) 3237c478bd9Sstevel@tonic-gate break; 3247c478bd9Sstevel@tonic-gate } 3257c478bd9Sstevel@tonic-gate 3267c478bd9Sstevel@tonic-gate assert(ap != NULL); 3277c478bd9Sstevel@tonic-gate 3281a7c1b72Smws if (ap == edp->dted_action) 3297c478bd9Sstevel@tonic-gate edp->dted_action = last->dtad_next; 3301a7c1b72Smws else 3317c478bd9Sstevel@tonic-gate ap->dtad_next = last->dtad_next; 3327c478bd9Sstevel@tonic-gate 3337c478bd9Sstevel@tonic-gate /* 3347c478bd9Sstevel@tonic-gate * We have now removed our action list from its ECB; we can 3357c478bd9Sstevel@tonic-gate * safely destroy the list. 3367c478bd9Sstevel@tonic-gate */ 3377c478bd9Sstevel@tonic-gate last->dtad_next = NULL; 3387c478bd9Sstevel@tonic-gate 3397c478bd9Sstevel@tonic-gate for (ap = sdp->dtsd_action; ap != NULL; ap = next) { 3407c478bd9Sstevel@tonic-gate assert(ap->dtad_uarg == (uintptr_t)sdp); 3411a7c1b72Smws dt_difo_free(dtp, ap->dtad_difo); 3427c478bd9Sstevel@tonic-gate next = ap->dtad_next; 3431a7c1b72Smws dt_free(dtp, ap); 3447c478bd9Sstevel@tonic-gate } 3457c478bd9Sstevel@tonic-gate } 3467c478bd9Sstevel@tonic-gate 3477c478bd9Sstevel@tonic-gate if (sdp->dtsd_fmtdata != NULL) 3487c478bd9Sstevel@tonic-gate dt_printf_destroy(sdp->dtsd_fmtdata); 3497c478bd9Sstevel@tonic-gate 3501a7c1b72Smws dt_ecbdesc_release(dtp, sdp->dtsd_ecbdesc); 3511a7c1b72Smws dt_free(dtp, sdp); 3527c478bd9Sstevel@tonic-gate } 353900524f3Sahl 354900524f3Sahl typedef struct dt_header_info { 355900524f3Sahl dtrace_hdl_t *dthi_dtp; /* consumer handle */ 356900524f3Sahl FILE *dthi_out; /* output file */ 357900524f3Sahl char *dthi_pmname; /* provider macro name */ 358900524f3Sahl char *dthi_pfname; /* provider function name */ 3592b6e762cSahl int dthi_empty; /* should we generate empty macros */ 360900524f3Sahl } dt_header_info_t; 361900524f3Sahl 362900524f3Sahl static void 363900524f3Sahl dt_header_fmt_macro(char *buf, const char *str) 364900524f3Sahl { 365900524f3Sahl for (;;) { 366900524f3Sahl if (islower(*str)) { 367900524f3Sahl *buf++ = *str++ + 'A' - 'a'; 368900524f3Sahl } else if (*str == '-') { 369900524f3Sahl *buf++ = '_'; 370900524f3Sahl str++; 371900524f3Sahl } else if (*str == '.') { 372900524f3Sahl *buf++ = '_'; 373900524f3Sahl str++; 374900524f3Sahl } else if ((*buf++ = *str++) == '\0') { 375900524f3Sahl break; 376900524f3Sahl } 377900524f3Sahl } 378900524f3Sahl } 379900524f3Sahl 380900524f3Sahl static void 381900524f3Sahl dt_header_fmt_func(char *buf, const char *str) 382900524f3Sahl { 383900524f3Sahl for (;;) { 384900524f3Sahl if (*str == '-') { 385900524f3Sahl *buf++ = '_'; 386900524f3Sahl *buf++ = '_'; 387900524f3Sahl str++; 388900524f3Sahl } else if ((*buf++ = *str++) == '\0') { 389900524f3Sahl break; 390900524f3Sahl } 391900524f3Sahl } 392900524f3Sahl } 393900524f3Sahl 394900524f3Sahl /*ARGSUSED*/ 395900524f3Sahl static int 396900524f3Sahl dt_header_decl(dt_idhash_t *dhp, dt_ident_t *idp, void *data) 397900524f3Sahl { 398900524f3Sahl dt_header_info_t *infop = data; 399900524f3Sahl dtrace_hdl_t *dtp = infop->dthi_dtp; 400900524f3Sahl dt_probe_t *prp = idp->di_data; 401900524f3Sahl dt_node_t *dnp; 402900524f3Sahl char buf[DT_TYPE_NAMELEN]; 403900524f3Sahl char *fname; 404900524f3Sahl const char *p; 405900524f3Sahl int i; 406900524f3Sahl 407900524f3Sahl p = prp->pr_name; 408900524f3Sahl for (i = 0; (p = strchr(p, '-')) != NULL; i++) 409900524f3Sahl p++; 410900524f3Sahl 411900524f3Sahl fname = alloca(strlen(prp->pr_name) + 1 + i); 412900524f3Sahl dt_header_fmt_func(fname, prp->pr_name); 413900524f3Sahl 414900524f3Sahl if (fprintf(infop->dthi_out, "extern void __dtrace_%s___%s(", 415900524f3Sahl infop->dthi_pfname, fname) < 0) 416900524f3Sahl return (dt_set_errno(dtp, errno)); 417900524f3Sahl 418900524f3Sahl for (dnp = prp->pr_nargs, i = 0; dnp != NULL; dnp = dnp->dn_list, i++) { 419900524f3Sahl if (fprintf(infop->dthi_out, "%s", 420900524f3Sahl ctf_type_name(dnp->dn_ctfp, dnp->dn_type, 421900524f3Sahl buf, sizeof (buf))) < 0) 422900524f3Sahl return (dt_set_errno(dtp, errno)); 423900524f3Sahl 424900524f3Sahl if (i + 1 != prp->pr_nargc && 425900524f3Sahl fprintf(infop->dthi_out, ", ") < 0) 426900524f3Sahl return (dt_set_errno(dtp, errno)); 427900524f3Sahl } 428900524f3Sahl 429900524f3Sahl if (i == 0 && fprintf(infop->dthi_out, "void") < 0) 430900524f3Sahl return (dt_set_errno(dtp, errno)); 431900524f3Sahl 432900524f3Sahl if (fprintf(infop->dthi_out, ");\n") < 0) 433900524f3Sahl return (dt_set_errno(dtp, errno)); 434900524f3Sahl 4350bac14eaSahl if (fprintf(infop->dthi_out, 4360bac14eaSahl "#ifndef\t__sparc\n" 4370bac14eaSahl "extern int __dtraceenabled_%s___%s(void);\n" 4380bac14eaSahl "#else\n" 4390bac14eaSahl "extern int __dtraceenabled_%s___%s(long);\n" 4400bac14eaSahl "#endif\n", 4410bac14eaSahl infop->dthi_pfname, fname, infop->dthi_pfname, fname) < 0) 442ac448965Sahl return (dt_set_errno(dtp, errno)); 443ac448965Sahl 444900524f3Sahl return (0); 445900524f3Sahl } 446900524f3Sahl 447900524f3Sahl /*ARGSUSED*/ 448900524f3Sahl static int 449900524f3Sahl dt_header_probe(dt_idhash_t *dhp, dt_ident_t *idp, void *data) 450900524f3Sahl { 451900524f3Sahl dt_header_info_t *infop = data; 452900524f3Sahl dtrace_hdl_t *dtp = infop->dthi_dtp; 453900524f3Sahl dt_probe_t *prp = idp->di_data; 454900524f3Sahl char *mname, *fname; 455900524f3Sahl const char *p; 456900524f3Sahl int i; 457900524f3Sahl 458900524f3Sahl p = prp->pr_name; 459900524f3Sahl for (i = 0; (p = strchr(p, '-')) != NULL; i++) 460900524f3Sahl p++; 461900524f3Sahl 462900524f3Sahl mname = alloca(strlen(prp->pr_name) + 1); 463900524f3Sahl dt_header_fmt_macro(mname, prp->pr_name); 464900524f3Sahl 465900524f3Sahl fname = alloca(strlen(prp->pr_name) + 1 + i); 466900524f3Sahl dt_header_fmt_func(fname, prp->pr_name); 467900524f3Sahl 468900524f3Sahl if (fprintf(infop->dthi_out, "#define\t%s_%s(", 469900524f3Sahl infop->dthi_pmname, mname) < 0) 470900524f3Sahl return (dt_set_errno(dtp, errno)); 471900524f3Sahl 472900524f3Sahl for (i = 0; i < prp->pr_nargc; i++) { 473900524f3Sahl if (fprintf(infop->dthi_out, "arg%d", i) < 0) 474900524f3Sahl return (dt_set_errno(dtp, errno)); 475900524f3Sahl 476900524f3Sahl if (i + 1 != prp->pr_nargc && 477900524f3Sahl fprintf(infop->dthi_out, ", ") < 0) 478900524f3Sahl return (dt_set_errno(dtp, errno)); 479900524f3Sahl } 480900524f3Sahl 4812b6e762cSahl if (!infop->dthi_empty) { 4822b6e762cSahl if (fprintf(infop->dthi_out, ") \\\n\t") < 0) 483900524f3Sahl return (dt_set_errno(dtp, errno)); 484900524f3Sahl 4852b6e762cSahl if (fprintf(infop->dthi_out, "__dtrace_%s___%s(", 4862b6e762cSahl infop->dthi_pfname, fname) < 0) 487900524f3Sahl return (dt_set_errno(dtp, errno)); 4882b6e762cSahl 4892b6e762cSahl for (i = 0; i < prp->pr_nargc; i++) { 4902b6e762cSahl if (fprintf(infop->dthi_out, "arg%d", i) < 0) 4912b6e762cSahl return (dt_set_errno(dtp, errno)); 4922b6e762cSahl 4932b6e762cSahl if (i + 1 != prp->pr_nargc && 4942b6e762cSahl fprintf(infop->dthi_out, ", ") < 0) 4952b6e762cSahl return (dt_set_errno(dtp, errno)); 4962b6e762cSahl } 497900524f3Sahl } 498900524f3Sahl 499900524f3Sahl if (fprintf(infop->dthi_out, ")\n") < 0) 500900524f3Sahl return (dt_set_errno(dtp, errno)); 501900524f3Sahl 5022b6e762cSahl if (!infop->dthi_empty) { 5030bac14eaSahl if (fprintf(infop->dthi_out, 5040bac14eaSahl "#ifndef\t__sparc\n" 5050bac14eaSahl "#define\t%s_%s_ENABLED() \\\n" 5060bac14eaSahl "\t__dtraceenabled_%s___%s()\n" 5070bac14eaSahl "#else\n" 5080bac14eaSahl "#define\t%s_%s_ENABLED() \\\n" 5090bac14eaSahl "\t__dtraceenabled_%s___%s(0)\n" 5100bac14eaSahl "#endif\n", 5110bac14eaSahl infop->dthi_pmname, mname, 5120bac14eaSahl infop->dthi_pfname, fname, 5130bac14eaSahl infop->dthi_pmname, mname, 5142b6e762cSahl infop->dthi_pfname, fname) < 0) 5152b6e762cSahl return (dt_set_errno(dtp, errno)); 5160bac14eaSahl 5172b6e762cSahl } else { 5182b6e762cSahl if (fprintf(infop->dthi_out, "#define\t%s_%s_ENABLED() (0)\n", 5192b6e762cSahl infop->dthi_pmname, mname) < 0) 5202b6e762cSahl return (dt_set_errno(dtp, errno)); 5212b6e762cSahl } 522ac448965Sahl 523900524f3Sahl return (0); 524900524f3Sahl } 525900524f3Sahl 526900524f3Sahl static int 527900524f3Sahl dt_header_provider(dtrace_hdl_t *dtp, dt_provider_t *pvp, FILE *out) 528900524f3Sahl { 529900524f3Sahl dt_header_info_t info; 530900524f3Sahl const char *p; 531900524f3Sahl int i; 532900524f3Sahl 533900524f3Sahl if (pvp->pv_flags & DT_PROVIDER_IMPL) 534900524f3Sahl return (0); 535900524f3Sahl 5362b6e762cSahl /* 5372b6e762cSahl * Count the instances of the '-' character since we'll need to double 5382b6e762cSahl * those up. 5392b6e762cSahl */ 540900524f3Sahl p = pvp->pv_desc.dtvd_name; 541900524f3Sahl for (i = 0; (p = strchr(p, '-')) != NULL; i++) 542900524f3Sahl p++; 543900524f3Sahl 544900524f3Sahl info.dthi_dtp = dtp; 545900524f3Sahl info.dthi_out = out; 5462b6e762cSahl info.dthi_empty = 0; 547900524f3Sahl 548900524f3Sahl info.dthi_pmname = alloca(strlen(pvp->pv_desc.dtvd_name) + 1); 549900524f3Sahl dt_header_fmt_macro(info.dthi_pmname, pvp->pv_desc.dtvd_name); 550900524f3Sahl 551900524f3Sahl info.dthi_pfname = alloca(strlen(pvp->pv_desc.dtvd_name) + 1 + i); 552900524f3Sahl dt_header_fmt_func(info.dthi_pfname, pvp->pv_desc.dtvd_name); 553900524f3Sahl 5542b6e762cSahl if (fprintf(out, "#if _DTRACE_VERSION\n\n") < 0) 5552b6e762cSahl return (dt_set_errno(dtp, errno)); 556900524f3Sahl 557900524f3Sahl if (dt_idhash_iter(pvp->pv_probes, dt_header_probe, &info) != 0) 558900524f3Sahl return (-1); /* dt_errno is set for us */ 559900524f3Sahl if (fprintf(out, "\n\n") < 0) 560900524f3Sahl return (dt_set_errno(dtp, errno)); 561900524f3Sahl if (dt_idhash_iter(pvp->pv_probes, dt_header_decl, &info) != 0) 562900524f3Sahl return (-1); /* dt_errno is set for us */ 563900524f3Sahl 5642b6e762cSahl if (fprintf(out, "\n#else\n\n") < 0) 5652b6e762cSahl return (dt_set_errno(dtp, errno)); 5662b6e762cSahl 5672b6e762cSahl info.dthi_empty = 1; 5682b6e762cSahl 5692b6e762cSahl if (dt_idhash_iter(pvp->pv_probes, dt_header_probe, &info) != 0) 5702b6e762cSahl return (-1); /* dt_errno is set for us */ 5712b6e762cSahl 5722b6e762cSahl if (fprintf(out, "\n#endif\n\n") < 0) 5732b6e762cSahl return (dt_set_errno(dtp, errno)); 5742b6e762cSahl 575900524f3Sahl return (0); 576900524f3Sahl } 577900524f3Sahl 578900524f3Sahl int 579900524f3Sahl dtrace_program_header(dtrace_hdl_t *dtp, FILE *out, const char *fname) 580900524f3Sahl { 581900524f3Sahl dt_provider_t *pvp; 582900524f3Sahl char *mfname, *p; 583900524f3Sahl 584900524f3Sahl if (fname != NULL) { 585900524f3Sahl if ((p = strrchr(fname, '/')) != NULL) 586900524f3Sahl fname = p + 1; 587900524f3Sahl 588900524f3Sahl mfname = alloca(strlen(fname) + 1); 589900524f3Sahl dt_header_fmt_macro(mfname, fname); 590900524f3Sahl if (fprintf(out, "#ifndef\t_%s\n#define\t_%s\n\n", 591900524f3Sahl mfname, mfname) < 0) 592900524f3Sahl return (dt_set_errno(dtp, errno)); 593900524f3Sahl } 594900524f3Sahl 5952b6e762cSahl if (fprintf(out, "#include <unistd.h>\n\n") < 0) 5962b6e762cSahl return (-1); 5972b6e762cSahl 598900524f3Sahl if (fprintf(out, "#ifdef\t__cplusplus\nextern \"C\" {\n#endif\n\n") < 0) 599900524f3Sahl return (-1); 600900524f3Sahl 601900524f3Sahl for (pvp = dt_list_next(&dtp->dt_provlist); 602900524f3Sahl pvp != NULL; pvp = dt_list_next(pvp)) { 603900524f3Sahl if (dt_header_provider(dtp, pvp, out) != 0) 604900524f3Sahl return (-1); /* dt_errno is set for us */ 605900524f3Sahl } 606900524f3Sahl 607900524f3Sahl if (fprintf(out, "\n#ifdef\t__cplusplus\n}\n#endif\n") < 0) 608900524f3Sahl return (dt_set_errno(dtp, errno)); 609900524f3Sahl 610900524f3Sahl if (fname != NULL && fprintf(out, "\n#endif\t/* _%s */\n", mfname) < 0) 611900524f3Sahl return (dt_set_errno(dtp, errno)); 612900524f3Sahl 613900524f3Sahl return (0); 614900524f3Sahl } 615