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*c9a6ea2eSBryan Cantrill * Copyright (c) 2003, 2010, Oracle and/or its affiliates. All rights reserved. 247c478bd9Sstevel@tonic-gate */ 257c478bd9Sstevel@tonic-gate 267c478bd9Sstevel@tonic-gate #include <unistd.h> 277c478bd9Sstevel@tonic-gate #include <strings.h> 287c478bd9Sstevel@tonic-gate #include <stdlib.h> 297c478bd9Sstevel@tonic-gate #include <errno.h> 307c478bd9Sstevel@tonic-gate #include <assert.h> 31900524f3Sahl #include <ctype.h> 32900524f3Sahl #include <alloca.h> 337c478bd9Sstevel@tonic-gate 347c478bd9Sstevel@tonic-gate #include <dt_impl.h> 351a7c1b72Smws #include <dt_program.h> 367c478bd9Sstevel@tonic-gate #include <dt_printf.h> 37900524f3Sahl #include <dt_provider.h> 387c478bd9Sstevel@tonic-gate 397c478bd9Sstevel@tonic-gate dtrace_prog_t * 401a7c1b72Smws dt_program_create(dtrace_hdl_t *dtp) 417c478bd9Sstevel@tonic-gate { 421a7c1b72Smws dtrace_prog_t *pgp = dt_zalloc(dtp, sizeof (dtrace_prog_t)); 437c478bd9Sstevel@tonic-gate 44*c9a6ea2eSBryan Cantrill if (pgp != NULL) { 457c478bd9Sstevel@tonic-gate dt_list_append(&dtp->dt_programs, pgp); 46*c9a6ea2eSBryan Cantrill } else { 477c478bd9Sstevel@tonic-gate (void) dt_set_errno(dtp, EDT_NOMEM); 48*c9a6ea2eSBryan Cantrill return (NULL); 49*c9a6ea2eSBryan Cantrill } 507c478bd9Sstevel@tonic-gate 51ac448965Sahl /* 52ac448965Sahl * By default, programs start with DOF version 1 so that output files 53ac448965Sahl * containing DOF are backward compatible. If a program requires new 54ac448965Sahl * DOF features, the version is increased as needed. 55ac448965Sahl */ 56ac448965Sahl pgp->dp_dofversion = DOF_VERSION_1; 57ac448965Sahl 587c478bd9Sstevel@tonic-gate return (pgp); 597c478bd9Sstevel@tonic-gate } 607c478bd9Sstevel@tonic-gate 617c478bd9Sstevel@tonic-gate void 621a7c1b72Smws dt_program_destroy(dtrace_hdl_t *dtp, dtrace_prog_t *pgp) 637c478bd9Sstevel@tonic-gate { 647c478bd9Sstevel@tonic-gate dt_stmt_t *stp, *next; 651a7c1b72Smws uint_t i; 667c478bd9Sstevel@tonic-gate 677c478bd9Sstevel@tonic-gate for (stp = dt_list_next(&pgp->dp_stmts); stp != NULL; stp = next) { 687c478bd9Sstevel@tonic-gate next = dt_list_next(stp); 691a7c1b72Smws dtrace_stmt_destroy(dtp, stp->ds_desc); 701a7c1b72Smws dt_free(dtp, stp); 717c478bd9Sstevel@tonic-gate } 727c478bd9Sstevel@tonic-gate 731a7c1b72Smws for (i = 0; i < pgp->dp_xrefslen; i++) 741a7c1b72Smws dt_free(dtp, pgp->dp_xrefs[i]); 751a7c1b72Smws 761a7c1b72Smws dt_free(dtp, pgp->dp_xrefs); 777c478bd9Sstevel@tonic-gate dt_list_delete(&dtp->dt_programs, pgp); 781a7c1b72Smws dt_free(dtp, pgp); 797c478bd9Sstevel@tonic-gate } 807c478bd9Sstevel@tonic-gate 817c478bd9Sstevel@tonic-gate /*ARGSUSED*/ 827c478bd9Sstevel@tonic-gate void 837c478bd9Sstevel@tonic-gate dtrace_program_info(dtrace_hdl_t *dtp, dtrace_prog_t *pgp, 847c478bd9Sstevel@tonic-gate dtrace_proginfo_t *pip) 857c478bd9Sstevel@tonic-gate { 867c478bd9Sstevel@tonic-gate dt_stmt_t *stp; 877c478bd9Sstevel@tonic-gate dtrace_actdesc_t *ap; 887c478bd9Sstevel@tonic-gate dtrace_ecbdesc_t *last = NULL; 897c478bd9Sstevel@tonic-gate 907c478bd9Sstevel@tonic-gate if (pip == NULL) 917c478bd9Sstevel@tonic-gate return; 927c478bd9Sstevel@tonic-gate 937c478bd9Sstevel@tonic-gate bzero(pip, sizeof (dtrace_proginfo_t)); 947c478bd9Sstevel@tonic-gate 957c478bd9Sstevel@tonic-gate if (dt_list_next(&pgp->dp_stmts) != NULL) { 967c478bd9Sstevel@tonic-gate pip->dpi_descattr = _dtrace_maxattr; 977c478bd9Sstevel@tonic-gate pip->dpi_stmtattr = _dtrace_maxattr; 987c478bd9Sstevel@tonic-gate } else { 997c478bd9Sstevel@tonic-gate pip->dpi_descattr = _dtrace_defattr; 1007c478bd9Sstevel@tonic-gate pip->dpi_stmtattr = _dtrace_defattr; 1017c478bd9Sstevel@tonic-gate } 1027c478bd9Sstevel@tonic-gate 1037c478bd9Sstevel@tonic-gate for (stp = dt_list_next(&pgp->dp_stmts); stp; stp = dt_list_next(stp)) { 1047c478bd9Sstevel@tonic-gate dtrace_ecbdesc_t *edp = stp->ds_desc->dtsd_ecbdesc; 1057c478bd9Sstevel@tonic-gate 1067c478bd9Sstevel@tonic-gate if (edp == last) 1077c478bd9Sstevel@tonic-gate continue; 1087c478bd9Sstevel@tonic-gate last = edp; 1097c478bd9Sstevel@tonic-gate 1107c478bd9Sstevel@tonic-gate pip->dpi_descattr = 1117c478bd9Sstevel@tonic-gate dt_attr_min(stp->ds_desc->dtsd_descattr, pip->dpi_descattr); 1127c478bd9Sstevel@tonic-gate 1137c478bd9Sstevel@tonic-gate pip->dpi_stmtattr = 1147c478bd9Sstevel@tonic-gate dt_attr_min(stp->ds_desc->dtsd_stmtattr, pip->dpi_stmtattr); 1157c478bd9Sstevel@tonic-gate 1167c478bd9Sstevel@tonic-gate /* 1177c478bd9Sstevel@tonic-gate * If there aren't any actions, account for the fact that 1187c478bd9Sstevel@tonic-gate * recording the epid will generate a record. 1197c478bd9Sstevel@tonic-gate */ 1207c478bd9Sstevel@tonic-gate if (edp->dted_action == NULL) 1217c478bd9Sstevel@tonic-gate pip->dpi_recgens++; 1227c478bd9Sstevel@tonic-gate 1237c478bd9Sstevel@tonic-gate for (ap = edp->dted_action; ap != NULL; ap = ap->dtad_next) { 1247c478bd9Sstevel@tonic-gate if (ap->dtad_kind == DTRACEACT_SPECULATE) { 1257c478bd9Sstevel@tonic-gate pip->dpi_speculations++; 1267c478bd9Sstevel@tonic-gate continue; 1277c478bd9Sstevel@tonic-gate } 1287c478bd9Sstevel@tonic-gate 1297c478bd9Sstevel@tonic-gate if (DTRACEACT_ISAGG(ap->dtad_kind)) { 1307c478bd9Sstevel@tonic-gate pip->dpi_recgens -= ap->dtad_arg; 1317c478bd9Sstevel@tonic-gate pip->dpi_aggregates++; 1327c478bd9Sstevel@tonic-gate continue; 1337c478bd9Sstevel@tonic-gate } 1347c478bd9Sstevel@tonic-gate 1357c478bd9Sstevel@tonic-gate if (DTRACEACT_ISDESTRUCTIVE(ap->dtad_kind)) 1367c478bd9Sstevel@tonic-gate continue; 1377c478bd9Sstevel@tonic-gate 1387c478bd9Sstevel@tonic-gate if (ap->dtad_kind == DTRACEACT_DIFEXPR && 1397c478bd9Sstevel@tonic-gate ap->dtad_difo->dtdo_rtype.dtdt_kind == 1407c478bd9Sstevel@tonic-gate DIF_TYPE_CTF && 1417c478bd9Sstevel@tonic-gate ap->dtad_difo->dtdo_rtype.dtdt_size == 0) 1427c478bd9Sstevel@tonic-gate continue; 1437c478bd9Sstevel@tonic-gate 1447c478bd9Sstevel@tonic-gate pip->dpi_recgens++; 1457c478bd9Sstevel@tonic-gate } 1467c478bd9Sstevel@tonic-gate } 1477c478bd9Sstevel@tonic-gate } 1487c478bd9Sstevel@tonic-gate 1497c478bd9Sstevel@tonic-gate int 1507c478bd9Sstevel@tonic-gate dtrace_program_exec(dtrace_hdl_t *dtp, dtrace_prog_t *pgp, 1517c478bd9Sstevel@tonic-gate dtrace_proginfo_t *pip) 1527c478bd9Sstevel@tonic-gate { 1537c478bd9Sstevel@tonic-gate void *dof; 1547c478bd9Sstevel@tonic-gate int n, err; 1557c478bd9Sstevel@tonic-gate 1567c478bd9Sstevel@tonic-gate dtrace_program_info(dtp, pgp, pip); 1577c478bd9Sstevel@tonic-gate 1587c478bd9Sstevel@tonic-gate if ((dof = dtrace_dof_create(dtp, pgp, DTRACE_D_STRIP)) == NULL) 1597c478bd9Sstevel@tonic-gate return (-1); 1607c478bd9Sstevel@tonic-gate 1617c478bd9Sstevel@tonic-gate n = dt_ioctl(dtp, DTRACEIOC_ENABLE, dof); 1627c478bd9Sstevel@tonic-gate dtrace_dof_destroy(dtp, dof); 1637c478bd9Sstevel@tonic-gate 1647c478bd9Sstevel@tonic-gate if (n == -1) { 1657c478bd9Sstevel@tonic-gate switch (errno) { 1667c478bd9Sstevel@tonic-gate case EINVAL: 1677c478bd9Sstevel@tonic-gate err = EDT_DIFINVAL; 1687c478bd9Sstevel@tonic-gate break; 1697c478bd9Sstevel@tonic-gate case EFAULT: 1707c478bd9Sstevel@tonic-gate err = EDT_DIFFAULT; 1717c478bd9Sstevel@tonic-gate break; 1727c478bd9Sstevel@tonic-gate case E2BIG: 1737c478bd9Sstevel@tonic-gate err = EDT_DIFSIZE; 1747c478bd9Sstevel@tonic-gate break; 175b9e93c10SJonathan Haslam case EBUSY: 176b9e93c10SJonathan Haslam err = EDT_ENABLING_ERR; 177b9e93c10SJonathan Haslam break; 1787c478bd9Sstevel@tonic-gate default: 1797c478bd9Sstevel@tonic-gate err = errno; 1807c478bd9Sstevel@tonic-gate } 1817c478bd9Sstevel@tonic-gate 1827c478bd9Sstevel@tonic-gate return (dt_set_errno(dtp, err)); 1837c478bd9Sstevel@tonic-gate } 1847c478bd9Sstevel@tonic-gate 1857c478bd9Sstevel@tonic-gate if (pip != NULL) 1867c478bd9Sstevel@tonic-gate pip->dpi_matches += n; 1877c478bd9Sstevel@tonic-gate 1887c478bd9Sstevel@tonic-gate return (0); 1897c478bd9Sstevel@tonic-gate } 1907c478bd9Sstevel@tonic-gate 1911a7c1b72Smws static void 1921a7c1b72Smws dt_ecbdesc_hold(dtrace_ecbdesc_t *edp) 1937c478bd9Sstevel@tonic-gate { 1947c478bd9Sstevel@tonic-gate edp->dted_refcnt++; 1957c478bd9Sstevel@tonic-gate } 1967c478bd9Sstevel@tonic-gate 1977c478bd9Sstevel@tonic-gate void 1981a7c1b72Smws dt_ecbdesc_release(dtrace_hdl_t *dtp, dtrace_ecbdesc_t *edp) 1997c478bd9Sstevel@tonic-gate { 2007c478bd9Sstevel@tonic-gate if (--edp->dted_refcnt > 0) 2017c478bd9Sstevel@tonic-gate return; 2027c478bd9Sstevel@tonic-gate 2031a7c1b72Smws dt_difo_free(dtp, edp->dted_pred.dtpdd_difo); 2047c478bd9Sstevel@tonic-gate assert(edp->dted_action == NULL); 2051a7c1b72Smws dt_free(dtp, edp); 2067c478bd9Sstevel@tonic-gate } 2077c478bd9Sstevel@tonic-gate 2087c478bd9Sstevel@tonic-gate dtrace_ecbdesc_t * 2091a7c1b72Smws dt_ecbdesc_create(dtrace_hdl_t *dtp, const dtrace_probedesc_t *pdp) 2107c478bd9Sstevel@tonic-gate { 2117c478bd9Sstevel@tonic-gate dtrace_ecbdesc_t *edp; 2127c478bd9Sstevel@tonic-gate 2131a7c1b72Smws if ((edp = dt_zalloc(dtp, sizeof (dtrace_ecbdesc_t))) == NULL) { 2147c478bd9Sstevel@tonic-gate (void) dt_set_errno(dtp, EDT_NOMEM); 2157c478bd9Sstevel@tonic-gate return (NULL); 2167c478bd9Sstevel@tonic-gate } 2177c478bd9Sstevel@tonic-gate 2187c478bd9Sstevel@tonic-gate edp->dted_probe = *pdp; 2191a7c1b72Smws dt_ecbdesc_hold(edp); 2207c478bd9Sstevel@tonic-gate return (edp); 2217c478bd9Sstevel@tonic-gate } 2227c478bd9Sstevel@tonic-gate 2237c478bd9Sstevel@tonic-gate dtrace_stmtdesc_t * 2247c478bd9Sstevel@tonic-gate dtrace_stmt_create(dtrace_hdl_t *dtp, dtrace_ecbdesc_t *edp) 2257c478bd9Sstevel@tonic-gate { 2267c478bd9Sstevel@tonic-gate dtrace_stmtdesc_t *sdp; 2277c478bd9Sstevel@tonic-gate 2281a7c1b72Smws if ((sdp = dt_zalloc(dtp, sizeof (dtrace_stmtdesc_t))) == NULL) 2297c478bd9Sstevel@tonic-gate return (NULL); 2307c478bd9Sstevel@tonic-gate 2311a7c1b72Smws dt_ecbdesc_hold(edp); 2327c478bd9Sstevel@tonic-gate sdp->dtsd_ecbdesc = edp; 2337c478bd9Sstevel@tonic-gate sdp->dtsd_descattr = _dtrace_defattr; 2347c478bd9Sstevel@tonic-gate sdp->dtsd_stmtattr = _dtrace_defattr; 2357c478bd9Sstevel@tonic-gate 2367c478bd9Sstevel@tonic-gate return (sdp); 2377c478bd9Sstevel@tonic-gate } 2387c478bd9Sstevel@tonic-gate 2397c478bd9Sstevel@tonic-gate dtrace_actdesc_t * 2407c478bd9Sstevel@tonic-gate dtrace_stmt_action(dtrace_hdl_t *dtp, dtrace_stmtdesc_t *sdp) 2417c478bd9Sstevel@tonic-gate { 2427c478bd9Sstevel@tonic-gate dtrace_actdesc_t *new; 2437c478bd9Sstevel@tonic-gate dtrace_ecbdesc_t *edp = sdp->dtsd_ecbdesc; 2447c478bd9Sstevel@tonic-gate 2451a7c1b72Smws if ((new = dt_alloc(dtp, sizeof (dtrace_actdesc_t))) == NULL) 2467c478bd9Sstevel@tonic-gate return (NULL); 2477c478bd9Sstevel@tonic-gate 2487c478bd9Sstevel@tonic-gate if (sdp->dtsd_action_last != NULL) { 2497c478bd9Sstevel@tonic-gate assert(sdp->dtsd_action != NULL); 2507c478bd9Sstevel@tonic-gate assert(sdp->dtsd_action_last->dtad_next == NULL); 2517c478bd9Sstevel@tonic-gate sdp->dtsd_action_last->dtad_next = new; 2527c478bd9Sstevel@tonic-gate } else { 2537c478bd9Sstevel@tonic-gate dtrace_actdesc_t *ap = edp->dted_action; 2547c478bd9Sstevel@tonic-gate 2557c478bd9Sstevel@tonic-gate assert(sdp->dtsd_action == NULL); 2567c478bd9Sstevel@tonic-gate sdp->dtsd_action = new; 2577c478bd9Sstevel@tonic-gate 2587c478bd9Sstevel@tonic-gate while (ap != NULL && ap->dtad_next != NULL) 2597c478bd9Sstevel@tonic-gate ap = ap->dtad_next; 2607c478bd9Sstevel@tonic-gate 2617c478bd9Sstevel@tonic-gate if (ap == NULL) 2627c478bd9Sstevel@tonic-gate edp->dted_action = new; 2637c478bd9Sstevel@tonic-gate else 2647c478bd9Sstevel@tonic-gate ap->dtad_next = new; 2657c478bd9Sstevel@tonic-gate } 2667c478bd9Sstevel@tonic-gate 2677c478bd9Sstevel@tonic-gate sdp->dtsd_action_last = new; 2687c478bd9Sstevel@tonic-gate bzero(new, sizeof (dtrace_actdesc_t)); 2697c478bd9Sstevel@tonic-gate new->dtad_uarg = (uintptr_t)sdp; 2707c478bd9Sstevel@tonic-gate 2717c478bd9Sstevel@tonic-gate return (new); 2727c478bd9Sstevel@tonic-gate } 2737c478bd9Sstevel@tonic-gate 2747c478bd9Sstevel@tonic-gate int 2757c478bd9Sstevel@tonic-gate dtrace_stmt_add(dtrace_hdl_t *dtp, dtrace_prog_t *pgp, dtrace_stmtdesc_t *sdp) 2767c478bd9Sstevel@tonic-gate { 2771a7c1b72Smws dt_stmt_t *stp = dt_alloc(dtp, sizeof (dt_stmt_t)); 2787c478bd9Sstevel@tonic-gate 2797c478bd9Sstevel@tonic-gate if (stp == NULL) 2801a7c1b72Smws return (-1); /* errno is set for us */ 2817c478bd9Sstevel@tonic-gate 2827c478bd9Sstevel@tonic-gate dt_list_append(&pgp->dp_stmts, stp); 2837c478bd9Sstevel@tonic-gate stp->ds_desc = sdp; 2847c478bd9Sstevel@tonic-gate 2857c478bd9Sstevel@tonic-gate return (0); 2867c478bd9Sstevel@tonic-gate } 2877c478bd9Sstevel@tonic-gate 2887c478bd9Sstevel@tonic-gate int 2897c478bd9Sstevel@tonic-gate dtrace_stmt_iter(dtrace_hdl_t *dtp, dtrace_prog_t *pgp, 2907c478bd9Sstevel@tonic-gate dtrace_stmt_f *func, void *data) 2917c478bd9Sstevel@tonic-gate { 2927c478bd9Sstevel@tonic-gate dt_stmt_t *stp, *next; 2937c478bd9Sstevel@tonic-gate int status = 0; 2947c478bd9Sstevel@tonic-gate 2957c478bd9Sstevel@tonic-gate for (stp = dt_list_next(&pgp->dp_stmts); stp != NULL; stp = next) { 2967c478bd9Sstevel@tonic-gate next = dt_list_next(stp); 2977c478bd9Sstevel@tonic-gate if ((status = func(dtp, pgp, stp->ds_desc, data)) != 0) 2987c478bd9Sstevel@tonic-gate break; 2997c478bd9Sstevel@tonic-gate } 3007c478bd9Sstevel@tonic-gate 3017c478bd9Sstevel@tonic-gate return (status); 3027c478bd9Sstevel@tonic-gate } 3037c478bd9Sstevel@tonic-gate 3047c478bd9Sstevel@tonic-gate void 3051a7c1b72Smws dtrace_stmt_destroy(dtrace_hdl_t *dtp, dtrace_stmtdesc_t *sdp) 3067c478bd9Sstevel@tonic-gate { 3077c478bd9Sstevel@tonic-gate dtrace_ecbdesc_t *edp = sdp->dtsd_ecbdesc; 3087c478bd9Sstevel@tonic-gate 3097c478bd9Sstevel@tonic-gate /* 3107c478bd9Sstevel@tonic-gate * We need to remove any actions that we have on this ECB, and 3117c478bd9Sstevel@tonic-gate * remove our hold on the ECB itself. 3127c478bd9Sstevel@tonic-gate */ 3137c478bd9Sstevel@tonic-gate if (sdp->dtsd_action != NULL) { 3147c478bd9Sstevel@tonic-gate dtrace_actdesc_t *last = sdp->dtsd_action_last; 3157c478bd9Sstevel@tonic-gate dtrace_actdesc_t *ap, *next; 3167c478bd9Sstevel@tonic-gate 3177c478bd9Sstevel@tonic-gate assert(last != NULL); 3187c478bd9Sstevel@tonic-gate 3197c478bd9Sstevel@tonic-gate for (ap = edp->dted_action; ap != NULL; ap = ap->dtad_next) { 3207c478bd9Sstevel@tonic-gate if (ap == sdp->dtsd_action) 3217c478bd9Sstevel@tonic-gate break; 3227c478bd9Sstevel@tonic-gate 3237c478bd9Sstevel@tonic-gate if (ap->dtad_next == sdp->dtsd_action) 3247c478bd9Sstevel@tonic-gate break; 3257c478bd9Sstevel@tonic-gate } 3267c478bd9Sstevel@tonic-gate 3277c478bd9Sstevel@tonic-gate assert(ap != NULL); 3287c478bd9Sstevel@tonic-gate 3291a7c1b72Smws if (ap == edp->dted_action) 3307c478bd9Sstevel@tonic-gate edp->dted_action = last->dtad_next; 3311a7c1b72Smws else 3327c478bd9Sstevel@tonic-gate ap->dtad_next = last->dtad_next; 3337c478bd9Sstevel@tonic-gate 3347c478bd9Sstevel@tonic-gate /* 3357c478bd9Sstevel@tonic-gate * We have now removed our action list from its ECB; we can 3367c478bd9Sstevel@tonic-gate * safely destroy the list. 3377c478bd9Sstevel@tonic-gate */ 3387c478bd9Sstevel@tonic-gate last->dtad_next = NULL; 3397c478bd9Sstevel@tonic-gate 3407c478bd9Sstevel@tonic-gate for (ap = sdp->dtsd_action; ap != NULL; ap = next) { 3417c478bd9Sstevel@tonic-gate assert(ap->dtad_uarg == (uintptr_t)sdp); 3421a7c1b72Smws dt_difo_free(dtp, ap->dtad_difo); 3437c478bd9Sstevel@tonic-gate next = ap->dtad_next; 3441a7c1b72Smws dt_free(dtp, ap); 3457c478bd9Sstevel@tonic-gate } 3467c478bd9Sstevel@tonic-gate } 3477c478bd9Sstevel@tonic-gate 3487c478bd9Sstevel@tonic-gate if (sdp->dtsd_fmtdata != NULL) 3497c478bd9Sstevel@tonic-gate dt_printf_destroy(sdp->dtsd_fmtdata); 3507c478bd9Sstevel@tonic-gate 3511a7c1b72Smws dt_ecbdesc_release(dtp, sdp->dtsd_ecbdesc); 3521a7c1b72Smws dt_free(dtp, sdp); 3537c478bd9Sstevel@tonic-gate } 354900524f3Sahl 355900524f3Sahl typedef struct dt_header_info { 356900524f3Sahl dtrace_hdl_t *dthi_dtp; /* consumer handle */ 357900524f3Sahl FILE *dthi_out; /* output file */ 358900524f3Sahl char *dthi_pmname; /* provider macro name */ 359900524f3Sahl char *dthi_pfname; /* provider function name */ 3602b6e762cSahl int dthi_empty; /* should we generate empty macros */ 361900524f3Sahl } dt_header_info_t; 362900524f3Sahl 363900524f3Sahl static void 364900524f3Sahl dt_header_fmt_macro(char *buf, const char *str) 365900524f3Sahl { 366900524f3Sahl for (;;) { 367900524f3Sahl if (islower(*str)) { 368900524f3Sahl *buf++ = *str++ + 'A' - 'a'; 369900524f3Sahl } else if (*str == '-') { 370900524f3Sahl *buf++ = '_'; 371900524f3Sahl str++; 372900524f3Sahl } else if (*str == '.') { 373900524f3Sahl *buf++ = '_'; 374900524f3Sahl str++; 375900524f3Sahl } else if ((*buf++ = *str++) == '\0') { 376900524f3Sahl break; 377900524f3Sahl } 378900524f3Sahl } 379900524f3Sahl } 380900524f3Sahl 381900524f3Sahl static void 382900524f3Sahl dt_header_fmt_func(char *buf, const char *str) 383900524f3Sahl { 384900524f3Sahl for (;;) { 385900524f3Sahl if (*str == '-') { 386900524f3Sahl *buf++ = '_'; 387900524f3Sahl *buf++ = '_'; 388900524f3Sahl str++; 389900524f3Sahl } else if ((*buf++ = *str++) == '\0') { 390900524f3Sahl break; 391900524f3Sahl } 392900524f3Sahl } 393900524f3Sahl } 394900524f3Sahl 395900524f3Sahl /*ARGSUSED*/ 396900524f3Sahl static int 397900524f3Sahl dt_header_decl(dt_idhash_t *dhp, dt_ident_t *idp, void *data) 398900524f3Sahl { 399900524f3Sahl dt_header_info_t *infop = data; 400900524f3Sahl dtrace_hdl_t *dtp = infop->dthi_dtp; 401900524f3Sahl dt_probe_t *prp = idp->di_data; 402900524f3Sahl dt_node_t *dnp; 403900524f3Sahl char buf[DT_TYPE_NAMELEN]; 404900524f3Sahl char *fname; 405900524f3Sahl const char *p; 406900524f3Sahl int i; 407900524f3Sahl 408900524f3Sahl p = prp->pr_name; 409900524f3Sahl for (i = 0; (p = strchr(p, '-')) != NULL; i++) 410900524f3Sahl p++; 411900524f3Sahl 412900524f3Sahl fname = alloca(strlen(prp->pr_name) + 1 + i); 413900524f3Sahl dt_header_fmt_func(fname, prp->pr_name); 414900524f3Sahl 415900524f3Sahl if (fprintf(infop->dthi_out, "extern void __dtrace_%s___%s(", 416900524f3Sahl infop->dthi_pfname, fname) < 0) 417900524f3Sahl return (dt_set_errno(dtp, errno)); 418900524f3Sahl 419900524f3Sahl for (dnp = prp->pr_nargs, i = 0; dnp != NULL; dnp = dnp->dn_list, i++) { 420900524f3Sahl if (fprintf(infop->dthi_out, "%s", 421900524f3Sahl ctf_type_name(dnp->dn_ctfp, dnp->dn_type, 422900524f3Sahl buf, sizeof (buf))) < 0) 423900524f3Sahl return (dt_set_errno(dtp, errno)); 424900524f3Sahl 425900524f3Sahl if (i + 1 != prp->pr_nargc && 426900524f3Sahl fprintf(infop->dthi_out, ", ") < 0) 427900524f3Sahl return (dt_set_errno(dtp, errno)); 428900524f3Sahl } 429900524f3Sahl 430900524f3Sahl if (i == 0 && fprintf(infop->dthi_out, "void") < 0) 431900524f3Sahl return (dt_set_errno(dtp, errno)); 432900524f3Sahl 433900524f3Sahl if (fprintf(infop->dthi_out, ");\n") < 0) 434900524f3Sahl return (dt_set_errno(dtp, errno)); 435900524f3Sahl 4360bac14eaSahl if (fprintf(infop->dthi_out, 4370bac14eaSahl "#ifndef\t__sparc\n" 4380bac14eaSahl "extern int __dtraceenabled_%s___%s(void);\n" 4390bac14eaSahl "#else\n" 4400bac14eaSahl "extern int __dtraceenabled_%s___%s(long);\n" 4410bac14eaSahl "#endif\n", 4420bac14eaSahl infop->dthi_pfname, fname, infop->dthi_pfname, fname) < 0) 443ac448965Sahl return (dt_set_errno(dtp, errno)); 444ac448965Sahl 445900524f3Sahl return (0); 446900524f3Sahl } 447900524f3Sahl 448900524f3Sahl /*ARGSUSED*/ 449900524f3Sahl static int 450900524f3Sahl dt_header_probe(dt_idhash_t *dhp, dt_ident_t *idp, void *data) 451900524f3Sahl { 452900524f3Sahl dt_header_info_t *infop = data; 453900524f3Sahl dtrace_hdl_t *dtp = infop->dthi_dtp; 454900524f3Sahl dt_probe_t *prp = idp->di_data; 455900524f3Sahl char *mname, *fname; 456900524f3Sahl const char *p; 457900524f3Sahl int i; 458900524f3Sahl 459900524f3Sahl p = prp->pr_name; 460900524f3Sahl for (i = 0; (p = strchr(p, '-')) != NULL; i++) 461900524f3Sahl p++; 462900524f3Sahl 463900524f3Sahl mname = alloca(strlen(prp->pr_name) + 1); 464900524f3Sahl dt_header_fmt_macro(mname, prp->pr_name); 465900524f3Sahl 466900524f3Sahl fname = alloca(strlen(prp->pr_name) + 1 + i); 467900524f3Sahl dt_header_fmt_func(fname, prp->pr_name); 468900524f3Sahl 469900524f3Sahl if (fprintf(infop->dthi_out, "#define\t%s_%s(", 470900524f3Sahl infop->dthi_pmname, mname) < 0) 471900524f3Sahl return (dt_set_errno(dtp, errno)); 472900524f3Sahl 473900524f3Sahl for (i = 0; i < prp->pr_nargc; i++) { 474900524f3Sahl if (fprintf(infop->dthi_out, "arg%d", i) < 0) 475900524f3Sahl return (dt_set_errno(dtp, errno)); 476900524f3Sahl 477900524f3Sahl if (i + 1 != prp->pr_nargc && 478900524f3Sahl fprintf(infop->dthi_out, ", ") < 0) 479900524f3Sahl return (dt_set_errno(dtp, errno)); 480900524f3Sahl } 481900524f3Sahl 4822b6e762cSahl if (!infop->dthi_empty) { 4832b6e762cSahl if (fprintf(infop->dthi_out, ") \\\n\t") < 0) 484900524f3Sahl return (dt_set_errno(dtp, errno)); 485900524f3Sahl 4862b6e762cSahl if (fprintf(infop->dthi_out, "__dtrace_%s___%s(", 4872b6e762cSahl infop->dthi_pfname, fname) < 0) 488900524f3Sahl return (dt_set_errno(dtp, errno)); 4892b6e762cSahl 4902b6e762cSahl for (i = 0; i < prp->pr_nargc; i++) { 4912b6e762cSahl if (fprintf(infop->dthi_out, "arg%d", i) < 0) 4922b6e762cSahl return (dt_set_errno(dtp, errno)); 4932b6e762cSahl 4942b6e762cSahl if (i + 1 != prp->pr_nargc && 4952b6e762cSahl fprintf(infop->dthi_out, ", ") < 0) 4962b6e762cSahl return (dt_set_errno(dtp, errno)); 4972b6e762cSahl } 498900524f3Sahl } 499900524f3Sahl 500900524f3Sahl if (fprintf(infop->dthi_out, ")\n") < 0) 501900524f3Sahl return (dt_set_errno(dtp, errno)); 502900524f3Sahl 5032b6e762cSahl if (!infop->dthi_empty) { 5040bac14eaSahl if (fprintf(infop->dthi_out, 5050bac14eaSahl "#ifndef\t__sparc\n" 5060bac14eaSahl "#define\t%s_%s_ENABLED() \\\n" 5070bac14eaSahl "\t__dtraceenabled_%s___%s()\n" 5080bac14eaSahl "#else\n" 5090bac14eaSahl "#define\t%s_%s_ENABLED() \\\n" 5100bac14eaSahl "\t__dtraceenabled_%s___%s(0)\n" 5110bac14eaSahl "#endif\n", 5120bac14eaSahl infop->dthi_pmname, mname, 5130bac14eaSahl infop->dthi_pfname, fname, 5140bac14eaSahl infop->dthi_pmname, mname, 5152b6e762cSahl infop->dthi_pfname, fname) < 0) 5162b6e762cSahl return (dt_set_errno(dtp, errno)); 5170bac14eaSahl 5182b6e762cSahl } else { 5192b6e762cSahl if (fprintf(infop->dthi_out, "#define\t%s_%s_ENABLED() (0)\n", 5202b6e762cSahl infop->dthi_pmname, mname) < 0) 5212b6e762cSahl return (dt_set_errno(dtp, errno)); 5222b6e762cSahl } 523ac448965Sahl 524900524f3Sahl return (0); 525900524f3Sahl } 526900524f3Sahl 527900524f3Sahl static int 528900524f3Sahl dt_header_provider(dtrace_hdl_t *dtp, dt_provider_t *pvp, FILE *out) 529900524f3Sahl { 530900524f3Sahl dt_header_info_t info; 531900524f3Sahl const char *p; 532900524f3Sahl int i; 533900524f3Sahl 534900524f3Sahl if (pvp->pv_flags & DT_PROVIDER_IMPL) 535900524f3Sahl return (0); 536900524f3Sahl 5372b6e762cSahl /* 5382b6e762cSahl * Count the instances of the '-' character since we'll need to double 5392b6e762cSahl * those up. 5402b6e762cSahl */ 541900524f3Sahl p = pvp->pv_desc.dtvd_name; 542900524f3Sahl for (i = 0; (p = strchr(p, '-')) != NULL; i++) 543900524f3Sahl p++; 544900524f3Sahl 545900524f3Sahl info.dthi_dtp = dtp; 546900524f3Sahl info.dthi_out = out; 5472b6e762cSahl info.dthi_empty = 0; 548900524f3Sahl 549900524f3Sahl info.dthi_pmname = alloca(strlen(pvp->pv_desc.dtvd_name) + 1); 550900524f3Sahl dt_header_fmt_macro(info.dthi_pmname, pvp->pv_desc.dtvd_name); 551900524f3Sahl 552900524f3Sahl info.dthi_pfname = alloca(strlen(pvp->pv_desc.dtvd_name) + 1 + i); 553900524f3Sahl dt_header_fmt_func(info.dthi_pfname, pvp->pv_desc.dtvd_name); 554900524f3Sahl 5552b6e762cSahl if (fprintf(out, "#if _DTRACE_VERSION\n\n") < 0) 5562b6e762cSahl return (dt_set_errno(dtp, errno)); 557900524f3Sahl 558900524f3Sahl if (dt_idhash_iter(pvp->pv_probes, dt_header_probe, &info) != 0) 559900524f3Sahl return (-1); /* dt_errno is set for us */ 560900524f3Sahl if (fprintf(out, "\n\n") < 0) 561900524f3Sahl return (dt_set_errno(dtp, errno)); 562900524f3Sahl if (dt_idhash_iter(pvp->pv_probes, dt_header_decl, &info) != 0) 563900524f3Sahl return (-1); /* dt_errno is set for us */ 564900524f3Sahl 5652b6e762cSahl if (fprintf(out, "\n#else\n\n") < 0) 5662b6e762cSahl return (dt_set_errno(dtp, errno)); 5672b6e762cSahl 5682b6e762cSahl info.dthi_empty = 1; 5692b6e762cSahl 5702b6e762cSahl if (dt_idhash_iter(pvp->pv_probes, dt_header_probe, &info) != 0) 5712b6e762cSahl return (-1); /* dt_errno is set for us */ 5722b6e762cSahl 5732b6e762cSahl if (fprintf(out, "\n#endif\n\n") < 0) 5742b6e762cSahl return (dt_set_errno(dtp, errno)); 5752b6e762cSahl 576900524f3Sahl return (0); 577900524f3Sahl } 578900524f3Sahl 579900524f3Sahl int 580900524f3Sahl dtrace_program_header(dtrace_hdl_t *dtp, FILE *out, const char *fname) 581900524f3Sahl { 582900524f3Sahl dt_provider_t *pvp; 583900524f3Sahl char *mfname, *p; 584900524f3Sahl 585900524f3Sahl if (fname != NULL) { 586900524f3Sahl if ((p = strrchr(fname, '/')) != NULL) 587900524f3Sahl fname = p + 1; 588900524f3Sahl 589900524f3Sahl mfname = alloca(strlen(fname) + 1); 590900524f3Sahl dt_header_fmt_macro(mfname, fname); 591900524f3Sahl if (fprintf(out, "#ifndef\t_%s\n#define\t_%s\n\n", 592900524f3Sahl mfname, mfname) < 0) 593900524f3Sahl return (dt_set_errno(dtp, errno)); 594900524f3Sahl } 595900524f3Sahl 5962b6e762cSahl if (fprintf(out, "#include <unistd.h>\n\n") < 0) 5972b6e762cSahl return (-1); 5982b6e762cSahl 599900524f3Sahl if (fprintf(out, "#ifdef\t__cplusplus\nextern \"C\" {\n#endif\n\n") < 0) 600900524f3Sahl return (-1); 601900524f3Sahl 602900524f3Sahl for (pvp = dt_list_next(&dtp->dt_provlist); 603900524f3Sahl pvp != NULL; pvp = dt_list_next(pvp)) { 604900524f3Sahl if (dt_header_provider(dtp, pvp, out) != 0) 605900524f3Sahl return (-1); /* dt_errno is set for us */ 606900524f3Sahl } 607900524f3Sahl 608900524f3Sahl if (fprintf(out, "\n#ifdef\t__cplusplus\n}\n#endif\n") < 0) 609900524f3Sahl return (dt_set_errno(dtp, errno)); 610900524f3Sahl 611900524f3Sahl if (fname != NULL && fprintf(out, "\n#endif\t/* _%s */\n", mfname) < 0) 612900524f3Sahl return (dt_set_errno(dtp, errno)); 613900524f3Sahl 614900524f3Sahl return (0); 615900524f3Sahl } 616