xref: /illumos-gate/usr/src/cmd/dtrace/test/tst/common/java_api/src/TestAbort.java (revision 581cede61ac9c14d8d4ea452562a567189eead78)
1 /*
2  * CDDL HEADER START
3  *
4  * The contents of this file are subject to the terms of the
5  * Common Development and Distribution License (the "License").
6  * You may not use this file except in compliance with the License.
7  *
8  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9  * or http://www.opensolaris.org/os/licensing.
10  * See the License for the specific language governing permissions
11  * and limitations under the License.
12  *
13  * When distributing Covered Code, include this CDDL HEADER in each
14  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15  * If applicable, add the following below this CDDL HEADER, with the
16  * fields enclosed by brackets "[]" replaced with your own identifying
17  * information: Portions Copyright [yyyy] [name of copyright owner]
18  *
19  * CDDL HEADER END
20  */
21 
22 /*
23  * Copyright 2006 Sun Microsystems, Inc.  All rights reserved.
24  * Use is subject to license terms.
25  *
26  * ident	"%Z%%M%	%I%	%E% SMI"
27  */
28 
29 import org.opensolaris.os.dtrace.*;
30 import java.util.NoSuchElementException;
31 
32 /**
33  * Regression for 6426129 abort() after close() throws
34  * NoSuchElementException.
35  */
36 public class TestAbort {
37     static boolean aborted = false;
38 
39     public static void
40     main(String[] args)
41     {
42 	Consumer consumer = new LocalConsumer();
43 
44 	// Test for deadlock (bug 6419880)
45 	try {
46 	    consumer.open();
47 	    consumer.compile("syscall:::entry { @[execname] = count(); } " +
48 		    "tick-101ms { printa(@); }");
49 	    consumer.enable();
50 	    consumer.go();
51 	    try {
52 		Thread.currentThread().sleep(1000);
53 	    } catch (InterruptedException e) {
54 		e.printStackTrace();
55 		System.exit(1);
56 	    }
57 	    consumer.close();
58 	} catch (DTraceException e) {
59 	    e.printStackTrace();
60 	    System.exit(1);
61 	}
62 
63 	consumer = new LocalConsumer();
64 
65 	// Should be able to abort an unopened consumer
66 	try {
67 	    aborted = false;
68 	    consumer.addConsumerListener(new ConsumerAdapter() {
69 		public void consumerStopped(ConsumerEvent e) {
70 		    aborted = true;
71 		}
72 	    });
73 	    consumer.abort();
74 	    consumer.open();
75 	    consumer.compile("syscall:::entry { @[execname] = count(); } " +
76 		    "tick-101ms { printa(@); }");
77 	    consumer.enable();
78 	    consumer.go();
79 	    try {
80 		Thread.currentThread().sleep(1000);
81 	    } catch (InterruptedException e) {
82 		e.printStackTrace();
83 		System.exit(1);
84 	    }
85 	    if (!aborted) {
86 		throw new IllegalStateException("consumer not aborted");
87 	    }
88 	    consumer.close();
89 	} catch (Exception e) {
90 	    e.printStackTrace();
91 	    System.exit(1);
92 	}
93 
94 	consumer = new LocalConsumer();
95 
96 	// Should be safe to call abort() in any state
97 	try {
98 	    consumer.abort();
99 	    consumer.open();
100 	    consumer.abort();
101 	    consumer.compile("syscall:::entry { @[execname] = count(); } " +
102 		    "tick-101ms { printa(@); }");
103 	    consumer.abort();
104 	    consumer.enable();
105 	    consumer.abort();
106 	    consumer.go();
107 	    consumer.abort();
108 	    consumer.close();
109 	    // Should be safe to call after close()
110 	    try {
111 		consumer.abort();
112 	    } catch (NoSuchElementException e) {
113 		e.printStackTrace();
114 		System.exit(1);
115 	    }
116 	} catch (Exception e) {
117 	    e.printStackTrace();
118 	    System.exit(1);
119 	}
120 
121 	consumer = new LocalConsumer();
122 
123 	// Tests that close() throws expected exception when called on
124 	// synchronized consumer.
125 	try {
126 	    consumer.open();
127 	    consumer.compile("syscall:::entry { @[execname] = count(); } " +
128 		    "tick-101ms { printa(@); }");
129 	    consumer.enable();
130 	    consumer.go();
131 	    try {
132 		Thread.currentThread().sleep(1000);
133 	    } catch (InterruptedException e) {
134 		e.printStackTrace();
135 		System.exit(1);
136 	    }
137 	    try {
138 		synchronized (consumer) {
139 		    consumer.close();
140 		}
141 	    } catch (IllegalThreadStateException e) {
142 		try {
143 		    consumer.close();
144 		    System.out.println("Successful");
145 		    System.exit(0);
146 		} catch (NoSuchElementException x) {
147 		    x.printStackTrace();
148 		    System.exit(1);
149 		}
150 	    }
151 	} catch (DTraceException e) {
152 	    e.printStackTrace();
153 	    System.exit(1);
154 	}
155 	System.err.println("Failed");
156 	System.exit(1);
157     }
158 }
159