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 * @enablePreview
27 * @library ../ /test/lib
28 * @requires jdk.foreign.linker != "UNSUPPORTED"
29 * @run testng/othervm --enable-native-access=ALL-UNNAMED TestTrivial
30 */
31
32 import org.testng.annotations.Test;
33
34 import java.lang.foreign.Arena;
35 import java.lang.foreign.FunctionDescriptor;
36 import java.lang.foreign.Linker;
37 import java.lang.foreign.MemoryLayout;
38 import java.lang.foreign.MemorySegment;
39 import java.lang.foreign.SegmentAllocator;
40 import java.lang.foreign.StructLayout;
41 import java.lang.invoke.MethodHandle;
42 import java.lang.invoke.VarHandle;
43
44 import static org.testng.Assert.assertEquals;
45
46 public class TestTrivial extends NativeTestHelper {
47
48 static {
49 System.loadLibrary("Trivial");
50 }
51
52 @Test
53 public void testEmpty() throws Throwable {
54 MethodHandle handle = downcallHandle("empty", FunctionDescriptor.ofVoid(), Linker.Option.isTrivial());
55 handle.invokeExact();
56 }
57
58 @Test
59 public void testIdentity() throws Throwable {
60 MethodHandle handle = downcallHandle("identity", FunctionDescriptor.of(C_INT, C_INT), Linker.Option.isTrivial());
61 int result = (int) handle.invokeExact(42);
62 assertEquals(result, 42);
63 }
64
65 @Test
66 public void testWithReturnBuffer() throws Throwable {
67 StructLayout bigLayout = MemoryLayout.structLayout(
68 C_LONG_LONG.withName("x"),
69 C_LONG_LONG.withName("y"));
70
71 MethodHandle handle = downcallHandle("with_return_buffer", FunctionDescriptor.of(bigLayout), Linker.Option.isTrivial());
72 VarHandle vhX = bigLayout.varHandle(MemoryLayout.PathElement.groupElement("x"));
73 VarHandle vhY = bigLayout.varHandle(MemoryLayout.PathElement.groupElement("y"));
74 try (Arena arena = Arena.ofConfined()) {
75 MemorySegment result = (MemorySegment) handle.invokeExact((SegmentAllocator) arena);
76 long x = (long) vhX.get(result);
77 assertEquals(x, 10);
78 long y = (long) vhY.get(result);
79 assertEquals(y, 11);
80 }
81 }
82
83 @Test
84 public void testCaptureErrno() throws Throwable {
85 Linker.Option ccs = Linker.Option.captureCallState("errno");
86 MethodHandle handle = downcallHandle("capture_errno", FunctionDescriptor.ofVoid(C_INT), Linker.Option.isTrivial(), ccs);
87 StructLayout capturedStateLayout = Linker.Option.captureStateLayout();
88 VarHandle errnoHandle = capturedStateLayout.varHandle(MemoryLayout.PathElement.groupElement("errno"));
89 try (Arena arena = Arena.ofConfined()) {
90 MemorySegment captureSeg = arena.allocate(capturedStateLayout);
91 handle.invokeExact(captureSeg, 42);
92 int capturedErrno = (int) errnoHandle.get(captureSeg);
93 assertEquals(capturedErrno, 42);
94 }
95 }
96
97
98 }