1 /*
2 * Copyright (c) 2023, 2024, 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 jdk.management
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.VThreadRunner; // ensureParallelism requires jdk.management
40 import jdk.test.lib.thread.VThreadPinner;
41 import org.junit.jupiter.api.Test;
42 import org.junit.jupiter.api.BeforeAll;
43 import static org.junit.jupiter.api.Assertions.*;
44
45 class VirtualThreadPinnedEventThrows {
46
47 @BeforeAll
48 static void setup() {
49 // need >=2 carriers for testing pinning when main thread is a virtual thread
50 if (Thread.currentThread().isVirtual()) {
51 VThreadRunner.ensureParallelism(2);
52 }
53 }
54
55 /**
56 * Test parking when pinned and creating the VirtualThreadPinnedEvent fails with OOME.
57 */
58 @Test
59 void testVirtualThreadPinnedEventCreateThrows() throws Exception {
60 VirtualThreadPinnedEvent.setCreateThrows(true);
61 try {
62 testParkWhenPinned();
63 } finally {
64 VirtualThreadPinnedEvent.setCreateThrows(false);
65 }
66 }
67
68 /**
69 * Test parking when pinned and VirtualThreadPinnedEvent.begin fails with OOME.
70 */
71 @Test
72 void testVirtualThreadPinnedEventBeginThrows() throws Exception {
73 VirtualThreadPinnedEvent.setBeginThrows(true);
74 try {
75 testParkWhenPinned();
76 } finally {
77 VirtualThreadPinnedEvent.setBeginThrows(false);
78 }
79 }
80
81 /**
82 * Test parking when pinned and VirtualThreadPinnedEvent.commit fails with OOME.
83 */
84 @Test
85 void testVirtualThreadPinnedEventCommitThrows() throws Exception {
86 VirtualThreadPinnedEvent.setCommitThrows(true);
87 try {
88 testParkWhenPinned();
89 } finally {
90 VirtualThreadPinnedEvent.setCommitThrows(false);
91 }
92 }
93
94 /**
95 * Test parking a virtual thread when pinned.
96 */
97 private void testParkWhenPinned() throws Exception {
98 var exception = new AtomicReference<Throwable>();
99 var done = new AtomicBoolean();
100 Thread thread = Thread.startVirtualThread(() -> {
101 try {
102 VThreadPinner.runPinned(() -> {
103 while (!done.get()) {
104 LockSupport.park();
105 }
106 });
107 } catch (Throwable e) {
108 exception.set(e);
109 }
110 });
111 try {
112 // wait for thread to park
113 Thread.State state;
114 while ((state = thread.getState()) != Thread.State.WAITING) {
115 assertTrue(state != Thread.State.TERMINATED);
116 Thread.sleep(10);
117 }
118 } finally {
119 done.set(true);
120 LockSupport.unpark(thread);
121 thread.join();
122 }
123 assertNull(exception.get());
124 }
125 }