1 /*
2 * Copyright (c) 2023, Oracle and/or its affiliates. All rights reserved.
3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4 *
5 * This code is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License version 2 only, as
7 * published by the Free Software Foundation.
8 *
9 * This code is distributed in the hope that it will be useful, but WITHOUT
10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
12 * version 2 for more details (a copy is included in the LICENSE file that
13 * accompanied this code).
14 *
15 * You should have received a copy of the GNU General Public License version
16 * 2 along with this work; if not, write to the Free Software Foundation,
17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
18 *
19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
20 * or visit www.oracle.com if you need additional information or have any
21 * questions.
22 */
23
24 /**
25 * @test
26 * @summary Test parking when pinned and emitting the JFR VirtualThreadPinnedEvent throws
27 * @modules java.base/jdk.internal.event
28 * @library /test/lib
29 * @compile/module=java.base jdk/internal/event/VirtualThreadPinnedEvent.java
30 * @run junit/othervm --enable-native-access=ALL-UNNAMED VirtualThreadPinnedEventThrows
31 */
32
33 import java.lang.ref.Reference;
34 import java.util.concurrent.atomic.AtomicBoolean;
35 import java.util.concurrent.atomic.AtomicReference;
36 import java.util.concurrent.locks.LockSupport;
37 import jdk.internal.event.VirtualThreadPinnedEvent;
38
39 import jdk.test.lib.thread.VThreadPinner;
40 import org.junit.jupiter.api.Test;
41 import static org.junit.jupiter.api.Assertions.*;
42
43 class VirtualThreadPinnedEventThrows {
44
45 /**
46 * Test parking when pinned and creating the VirtualThreadPinnedEvent fails with OOME.
47 */
48 @Test
49 void testVirtualThreadPinnedEventCreateThrows() throws Exception {
50 VirtualThreadPinnedEvent.setCreateThrows(true);
51 try {
52 testParkWhenPinned();
53 } finally {
54 VirtualThreadPinnedEvent.setCreateThrows(false);
55 }
56 }
57
58 /**
59 * Test parking when pinned and VirtualThreadPinnedEvent.begin fails with OOME.
60 */
61 @Test
62 void testVirtualThreadPinnedEventBeginThrows() throws Exception {
63 VirtualThreadPinnedEvent.setBeginThrows(true);
64 try {
65 testParkWhenPinned();
66 } finally {
67 VirtualThreadPinnedEvent.setBeginThrows(false);
68 }
69 }
70
71 /**
72 * Test parking when pinned and VirtualThreadPinnedEvent.commit fails with OOME.
73 */
74 @Test
75 void testVirtualThreadPinnedEventCommitThrows() throws Exception {
76 VirtualThreadPinnedEvent.setCommitThrows(true);
77 try {
78 testParkWhenPinned();
79 } finally {
80 VirtualThreadPinnedEvent.setCommitThrows(false);
81 }
82 }
83
84 /**
85 * Test parking a virtual thread when pinned.
86 */
87 private void testParkWhenPinned() throws Exception {
88 var exception = new AtomicReference<Throwable>();
89 var done = new AtomicBoolean();
90 Thread thread = Thread.startVirtualThread(() -> {
91 try {
92 VThreadPinner.runPinned(() -> {
93 while (!done.get()) {
94 LockSupport.park();
95 }
96 });
97 } catch (Throwable e) {
98 exception.set(e);
99 }
100 });
101 try {
102 // wait for thread to park
103 Thread.State state;
104 while ((state = thread.getState()) != Thread.State.WAITING) {
105 assertTrue(state != Thread.State.TERMINATED);
106 Thread.sleep(10);
107 }
108 } finally {
109 done.set(true);
110 LockSupport.unpark(thread);
111 thread.join();
112 }
113 assertNull(exception.get());
114 }
115 }