java_api/src/TestDrop.java

/*
 * 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 2007 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 */
import java.util.*;
import java.util.concurrent.atomic.*;
import org.opensolaris.os.dtrace.*;

/**
 * Regression test for 6521523 aggregation drops can hang the Java
 * DTrace API.
 */
public class TestDrop {
    static final String PROGRAM =
        "fbt:genunix::entry { @[execname, pid] = count(); }";

    static AtomicLong consumerThreadID = new AtomicLong();
    static AtomicLong getAggregateThreadID = new AtomicLong();
    static AtomicBoolean done = new AtomicBoolean();
    static int seconds;

    private static void
    startTimer()
    {
    if (seconds <= 0) {
        return;
    }

    final Timer timer = new Timer();
    timer.schedule(new TimerTask() {
        public void run() {
        done.set(true);
        timer.cancel();
        }
    }, seconds * 1000L);
    }

    private static void
    sampleAggregate(Consumer consumer) throws DTraceException
    {
    while (consumer.isRunning() && !done.get()) {
        try {
        Thread.sleep(50);
        } catch (InterruptedException e) {
        }

        consumer.getAggregate(Collections. <String> emptySet());
    }
    }

    private static void
    startAggregateThread(final Consumer consumer)
    {
    Runnable aggregateSampler = new Runnable() {
        public void run() {
        Thread t = Thread.currentThread();
        getAggregateThreadID.set(t.getId());
        Throwable x = null;
        try {
            sampleAggregate(consumer);
        } catch (Throwable e) {
            x = e;
        }

        if (Thread.holdsLock(LocalConsumer.class)) {
            if (x != null) {
            x.printStackTrace();
            }
            System.out.println("Lock held");
            System.exit(1);
        } else {
            System.out.println("Lock released");
            consumer.close(); // blocks if lock held
        }
        }
    };

    Thread t = new Thread(aggregateSampler, "Aggregate Sampler");
    t.start();
    }

    static void
    usage()
    {
    System.err.println("usage: java TestDrop [ seconds ]");
    System.exit(2);
    }

    public static void
    main(String[] args)
    {
    if (args.length == 1) {
        try {
        seconds = Integer.parseInt(args[0]);
        } catch (NumberFormatException e) {
        usage();
        }
    } else if (args.length > 1) {
        usage();
    }

    final Consumer consumer = new LocalConsumer() {
        protected Thread createThread() {
        Runnable worker = new Runnable() {
            public void run() {
            Thread t = Thread.currentThread();
            consumerThreadID.set(t.getId());
            work();
            }
        };
        Thread t = new Thread(worker);
        return t;
        }
    };

    consumer.addConsumerListener(new ConsumerAdapter() {
        public void consumerStarted(ConsumerEvent e) {
        startAggregateThread(consumer);
        startTimer();
        }
        public void dataDropped(DropEvent e) throws ConsumerException {
        Thread t = Thread.currentThread();
        if (t.getId() == getAggregateThreadID.get()) {
            Drop drop = e.getDrop();
            throw new ConsumerException(drop.getDefaultMessage(),
                drop);
        }
        }
    });

    try {
        consumer.open();
        consumer.setOption(Option.aggsize, Option.kb(1));
        consumer.setOption(Option.aggrate, Option.millis(101));
        consumer.compile(PROGRAM);
        consumer.enable();
        consumer.go(new ExceptionHandler() {
        public void handleException(Throwable e) {
            e.printStackTrace();
        }
        });
    } catch (DTraceException e) {
        e.printStackTrace();
    }
    }
}