1 /*
2 * Copyright (c) 2022, 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 * @enablePreview
27 * @library ../ /test/lib
28 * @requires jdk.foreign.linker != "UNSUPPORTED"
29 * @run testng/othervm --enable-native-access=ALL-UNNAMED TestPassHeapSegment
30 */
31
32 import org.testng.annotations.DataProvider;
33 import org.testng.annotations.Test;
34
35 import java.io.IOException;
36 import java.lang.foreign.FunctionDescriptor;
37 import java.lang.foreign.MemorySegment;
38 import java.lang.invoke.MethodHandle;
39
40 import static java.lang.foreign.ValueLayout.ADDRESS;
41
42 public class TestPassHeapSegment extends UpcallTestHelper {
43
44 static {
45 System.loadLibrary("PassHeapSegment");
46 }
47
48 @Test(expectedExceptions = IllegalArgumentException.class,
49 expectedExceptionsMessageRegExp = ".*Heap segment not allowed.*")
50 public void testNoHeapArgs() throws Throwable {
51 MethodHandle handle = downcallHandle("test_args", FunctionDescriptor.ofVoid(ADDRESS));
52 MemorySegment segment = MemorySegment.ofArray(new byte[]{ 0, 1, 2 });
53 handle.invoke(segment); // should throw
54 }
55
56 @Test(dataProvider = "specs")
57 public void testNoHeapReturns(boolean spec) throws IOException, InterruptedException {
58 runInNewProcess(Runner.class, spec).assertStdErrContains("Heap segment not allowed");
59 }
60
61 public static class Runner {
62
63 static {
64 System.loadLibrary("PassHeapSegment");
65 }
66
67 public static void main(String[] args) throws Throwable {
68 MethodHandle handle = downcallHandle("test_return", FunctionDescriptor.ofVoid(ADDRESS));
69 MemorySegment upcallStub = upcallStub(Runner.class, "target", FunctionDescriptor.of(ADDRESS));
70 handle.invoke(upcallStub);
71 }
72
73 public static MemorySegment target() {
74 return MemorySegment.ofArray(new byte[]{ 0, 1, 2 }); // should throw
75 }
76 }
77
78 @DataProvider
79 public static Object[][] specs() {
80 return new Object[][]{
81 { true },
82 { false }
83 };
84 }
85 }