6e0bee749e9081c846d305caa233c222da771831jhaslam#!/bin/ksh -p
6e0bee749e9081c846d305caa233c222da771831jhaslam#
6e0bee749e9081c846d305caa233c222da771831jhaslam# CDDL HEADER START
6e0bee749e9081c846d305caa233c222da771831jhaslam#
6e0bee749e9081c846d305caa233c222da771831jhaslam# The contents of this file are subject to the terms of the
6e0bee749e9081c846d305caa233c222da771831jhaslam# Common Development and Distribution License (the "License").
6e0bee749e9081c846d305caa233c222da771831jhaslam# You may not use this file except in compliance with the License.
6e0bee749e9081c846d305caa233c222da771831jhaslam#
6e0bee749e9081c846d305caa233c222da771831jhaslam# You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
6e0bee749e9081c846d305caa233c222da771831jhaslam# or http://www.opensolaris.org/os/licensing.
6e0bee749e9081c846d305caa233c222da771831jhaslam# See the License for the specific language governing permissions
6e0bee749e9081c846d305caa233c222da771831jhaslam# and limitations under the License.
6e0bee749e9081c846d305caa233c222da771831jhaslam#
6e0bee749e9081c846d305caa233c222da771831jhaslam# When distributing Covered Code, include this CDDL HEADER in each
6e0bee749e9081c846d305caa233c222da771831jhaslam# file and include the License file at usr/src/OPENSOLARIS.LICENSE.
6e0bee749e9081c846d305caa233c222da771831jhaslam# If applicable, add the following below this CDDL HEADER, with the
6e0bee749e9081c846d305caa233c222da771831jhaslam# fields enclosed by brackets "[]" replaced with your own identifying
6e0bee749e9081c846d305caa233c222da771831jhaslam# information: Portions Copyright [yyyy] [name of copyright owner]
6e0bee749e9081c846d305caa233c222da771831jhaslam#
6e0bee749e9081c846d305caa233c222da771831jhaslam# CDDL HEADER END
6e0bee749e9081c846d305caa233c222da771831jhaslam#
6e0bee749e9081c846d305caa233c222da771831jhaslam
6e0bee749e9081c846d305caa233c222da771831jhaslam#
6e0bee749e9081c846d305caa233c222da771831jhaslam# Copyright 2008 Sun Microsystems, Inc. All rights reserved.
6e0bee749e9081c846d305caa233c222da771831jhaslam# Use is subject to license terms.
6e0bee749e9081c846d305caa233c222da771831jhaslam#
6e0bee749e9081c846d305caa233c222da771831jhaslam
6e0bee749e9081c846d305caa233c222da771831jhaslam#
6e0bee749e9081c846d305caa233c222da771831jhaslam# This test verifies that USDT probes will be picked up after a dlopen(3C)
6e0bee749e9081c846d305caa233c222da771831jhaslam# when a regex in the provider name matches both USDT probes and pid probes
6e0bee749e9081c846d305caa233c222da771831jhaslam# (e.g., p*d$target matches both pid$target and pyramid$target.)
6e0bee749e9081c846d305caa233c222da771831jhaslam#
6e0bee749e9081c846d305caa233c222da771831jhaslam
6e0bee749e9081c846d305caa233c222da771831jhaslamif [ $# != 1 ]; then
6e0bee749e9081c846d305caa233c222da771831jhaslam echo expected one argument: '<'dtrace-path'>'
6e0bee749e9081c846d305caa233c222da771831jhaslam exit 2
6e0bee749e9081c846d305caa233c222da771831jhaslamfi
6e0bee749e9081c846d305caa233c222da771831jhaslam
6e0bee749e9081c846d305caa233c222da771831jhaslamdtrace=$1
6e0bee749e9081c846d305caa233c222da771831jhaslamDIR=${TMPDIR:-/tmp}/dtest.$$
6e0bee749e9081c846d305caa233c222da771831jhaslam
6e0bee749e9081c846d305caa233c222da771831jhaslammkdir $DIR
6e0bee749e9081c846d305caa233c222da771831jhaslamcd $DIR
6e0bee749e9081c846d305caa233c222da771831jhaslam
6e0bee749e9081c846d305caa233c222da771831jhaslamcat > Makefile <<EOF
6e0bee749e9081c846d305caa233c222da771831jhaslamall: main altlib.so
6e0bee749e9081c846d305caa233c222da771831jhaslam
6e0bee749e9081c846d305caa233c222da771831jhaslammain: main.o provmain.o
a386cc11a86ecb60f5a48078d22c1500e2ad003eRobert Mustacchi gcc -m32 -o main main.o provmain.o
6e0bee749e9081c846d305caa233c222da771831jhaslam
6e0bee749e9081c846d305caa233c222da771831jhaslammain.o: main.c prov.h
a386cc11a86ecb60f5a48078d22c1500e2ad003eRobert Mustacchi gcc -m32 -c main.c
6e0bee749e9081c846d305caa233c222da771831jhaslam
6e0bee749e9081c846d305caa233c222da771831jhaslamprov.h: prov.d
6e0bee749e9081c846d305caa233c222da771831jhaslam $dtrace -h -s prov.d
6e0bee749e9081c846d305caa233c222da771831jhaslam
6e0bee749e9081c846d305caa233c222da771831jhaslamprovmain.o: prov.d main.o
6e0bee749e9081c846d305caa233c222da771831jhaslam $dtrace -G -32 -o provmain.o -s prov.d main.o
6e0bee749e9081c846d305caa233c222da771831jhaslam
6e0bee749e9081c846d305caa233c222da771831jhaslamaltlib.so: altlib.o provalt.o
a386cc11a86ecb60f5a48078d22c1500e2ad003eRobert Mustacchi gcc -m32 -shared -o altlib.so altlib.o provalt.o -lc
6e0bee749e9081c846d305caa233c222da771831jhaslam
6e0bee749e9081c846d305caa233c222da771831jhaslamaltlib.o: altlib.c prov.h
a386cc11a86ecb60f5a48078d22c1500e2ad003eRobert Mustacchi gcc -m32 -c altlib.c
6e0bee749e9081c846d305caa233c222da771831jhaslam
6e0bee749e9081c846d305caa233c222da771831jhaslamprovalt.o: prov.d altlib.o
6e0bee749e9081c846d305caa233c222da771831jhaslam $dtrace -G -32 -o provalt.o -s prov.d altlib.o
6e0bee749e9081c846d305caa233c222da771831jhaslamEOF
6e0bee749e9081c846d305caa233c222da771831jhaslam
6e0bee749e9081c846d305caa233c222da771831jhaslamcat > prov.d <<EOF
6e0bee749e9081c846d305caa233c222da771831jhaslamprovider pyramid {
6e0bee749e9081c846d305caa233c222da771831jhaslam probe entry();
6e0bee749e9081c846d305caa233c222da771831jhaslam};
6e0bee749e9081c846d305caa233c222da771831jhaslamEOF
6e0bee749e9081c846d305caa233c222da771831jhaslam
6e0bee749e9081c846d305caa233c222da771831jhaslamcat > altlib.c <<EOF
6e0bee749e9081c846d305caa233c222da771831jhaslam#include <sys/sdt.h>
6e0bee749e9081c846d305caa233c222da771831jhaslam#include "prov.h"
6e0bee749e9081c846d305caa233c222da771831jhaslam
6e0bee749e9081c846d305caa233c222da771831jhaslamvoid
6e0bee749e9081c846d305caa233c222da771831jhaslamgo(void)
6e0bee749e9081c846d305caa233c222da771831jhaslam{
6e0bee749e9081c846d305caa233c222da771831jhaslam PYRAMID_ENTRY();
6e0bee749e9081c846d305caa233c222da771831jhaslam}
6e0bee749e9081c846d305caa233c222da771831jhaslamEOF
6e0bee749e9081c846d305caa233c222da771831jhaslam
6e0bee749e9081c846d305caa233c222da771831jhaslamcat > main.c <<EOF
6e0bee749e9081c846d305caa233c222da771831jhaslam#include <dlfcn.h>
6e0bee749e9081c846d305caa233c222da771831jhaslam#include <unistd.h>
6e0bee749e9081c846d305caa233c222da771831jhaslam#include <stdio.h>
6e0bee749e9081c846d305caa233c222da771831jhaslam#include <sys/sdt.h>
6e0bee749e9081c846d305caa233c222da771831jhaslam#include "prov.h"
6e0bee749e9081c846d305caa233c222da771831jhaslam
6e0bee749e9081c846d305caa233c222da771831jhaslamvoid
6e0bee749e9081c846d305caa233c222da771831jhaslamgo(void)
6e0bee749e9081c846d305caa233c222da771831jhaslam{
6e0bee749e9081c846d305caa233c222da771831jhaslam PYRAMID_ENTRY();
6e0bee749e9081c846d305caa233c222da771831jhaslam}
6e0bee749e9081c846d305caa233c222da771831jhaslam
6e0bee749e9081c846d305caa233c222da771831jhaslamint
6e0bee749e9081c846d305caa233c222da771831jhaslammain(int argc, char **argv)
6e0bee749e9081c846d305caa233c222da771831jhaslam{
6e0bee749e9081c846d305caa233c222da771831jhaslam void *alt;
6e0bee749e9081c846d305caa233c222da771831jhaslam void *alt_go;
6e0bee749e9081c846d305caa233c222da771831jhaslam
6e0bee749e9081c846d305caa233c222da771831jhaslam go();
6e0bee749e9081c846d305caa233c222da771831jhaslam
6e0bee749e9081c846d305caa233c222da771831jhaslam if ((alt = dlopen("./altlib.so", RTLD_LAZY | RTLD_LOCAL))
6e0bee749e9081c846d305caa233c222da771831jhaslam == NULL) {
6e0bee749e9081c846d305caa233c222da771831jhaslam printf("dlopen of altlib.so failed: %s\n", dlerror());
6e0bee749e9081c846d305caa233c222da771831jhaslam return (1);
6e0bee749e9081c846d305caa233c222da771831jhaslam }
6e0bee749e9081c846d305caa233c222da771831jhaslam
6e0bee749e9081c846d305caa233c222da771831jhaslam if ((alt_go = dlsym(alt, "go")) == NULL) {
6e0bee749e9081c846d305caa233c222da771831jhaslam printf("failed to lookup 'go' in altlib.so\n");
6e0bee749e9081c846d305caa233c222da771831jhaslam return (1);
6e0bee749e9081c846d305caa233c222da771831jhaslam }
6e0bee749e9081c846d305caa233c222da771831jhaslam
6e0bee749e9081c846d305caa233c222da771831jhaslam ((void (*)(void))alt_go)();
6e0bee749e9081c846d305caa233c222da771831jhaslam
6e0bee749e9081c846d305caa233c222da771831jhaslam return (0);
6e0bee749e9081c846d305caa233c222da771831jhaslam}
6e0bee749e9081c846d305caa233c222da771831jhaslamEOF
6e0bee749e9081c846d305caa233c222da771831jhaslam
6e0bee749e9081c846d305caa233c222da771831jhaslammake > /dev/null
6e0bee749e9081c846d305caa233c222da771831jhaslamif [ $? -ne 0 ]; then
6e0bee749e9081c846d305caa233c222da771831jhaslam print -u2 "failed to build"
6e0bee749e9081c846d305caa233c222da771831jhaslam exit 1
6e0bee749e9081c846d305caa233c222da771831jhaslamfi
6e0bee749e9081c846d305caa233c222da771831jhaslam
6e0bee749e9081c846d305caa233c222da771831jhaslamcat > main.d <<'EOF'
6e0bee749e9081c846d305caa233c222da771831jhaslamp*d$target::go:entry
6e0bee749e9081c846d305caa233c222da771831jhaslam{
6e0bee749e9081c846d305caa233c222da771831jhaslam @foo[probemod, probefunc, probename] = count();
6e0bee749e9081c846d305caa233c222da771831jhaslam}
6e0bee749e9081c846d305caa233c222da771831jhaslam
6e0bee749e9081c846d305caa233c222da771831jhaslamEND
6e0bee749e9081c846d305caa233c222da771831jhaslam{
6e0bee749e9081c846d305caa233c222da771831jhaslam printa("%s:%s:%s %@u\n",@foo);
6e0bee749e9081c846d305caa233c222da771831jhaslam}
6e0bee749e9081c846d305caa233c222da771831jhaslamEOF
6e0bee749e9081c846d305caa233c222da771831jhaslam
6e0bee749e9081c846d305caa233c222da771831jhaslamscript() {
6e0bee749e9081c846d305caa233c222da771831jhaslam $dtrace -q -s ./main.d -c ./main
6e0bee749e9081c846d305caa233c222da771831jhaslam}
6e0bee749e9081c846d305caa233c222da771831jhaslam
6e0bee749e9081c846d305caa233c222da771831jhaslamscript
6e0bee749e9081c846d305caa233c222da771831jhaslamstatus=$?
6e0bee749e9081c846d305caa233c222da771831jhaslam
6e0bee749e9081c846d305caa233c222da771831jhaslamcd /tmp
6e0bee749e9081c846d305caa233c222da771831jhaslam/usr/bin/rm -rf $DIR
6e0bee749e9081c846d305caa233c222da771831jhaslam
6e0bee749e9081c846d305caa233c222da771831jhaslamexit $status