1 /*
 2  *  Copyright (c) 2021, 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  * @requires os.arch=="amd64" | os.arch=="x86_64" | os.arch=="aarch64"
27  * @run testng/othervm
28  *   --enable-native-access=ALL-UNNAMED
29  *   TestVirtualCalls
30  */
31 
32 import jdk.incubator.foreign.Addressable;
33 import jdk.incubator.foreign.CLinker;
34 import jdk.incubator.foreign.FunctionDescriptor;
35 
36 import java.lang.invoke.MethodHandle;
37 import java.lang.invoke.MethodType;
38 
39 import jdk.incubator.foreign.SymbolLookup;
40 import jdk.incubator.foreign.MemoryAddress;
41 import org.testng.annotations.*;
42 
43 import static jdk.incubator.foreign.CLinker.*;
44 import static org.testng.Assert.assertEquals;
45 
46 public class TestVirtualCalls {
47 
48     static final CLinker abi = CLinker.getInstance();
49 
50     static final MethodHandle func;
51     static final MemoryAddress funcA;
52     static final MemoryAddress funcB;
53     static final MemoryAddress funcC;
54 
55     static {
56         func = abi.downcallHandle(
57             MethodType.methodType(int.class),
58             FunctionDescriptor.of(C_INT));
59 
60         System.loadLibrary("Virtual");
61         SymbolLookup lookup = SymbolLookup.loaderLookup();
62         funcA = lookup.lookup("funcA").get();
63         funcB = lookup.lookup("funcB").get();
64         funcC = lookup.lookup("funcC").get();
65     }
66 
67     @Test
68     public void testVirtualCalls() throws Throwable {
69         assertEquals((int) func.invokeExact((Addressable) funcA), 1);
70         assertEquals((int) func.invokeExact((Addressable) funcB), 2);
71         assertEquals((int) func.invokeExact((Addressable) funcC), 3);
72     }
73 
74     @Test(expectedExceptions = NullPointerException.class)
75     public void testNullTarget() throws Throwable {
76         int x = (int) func.invokeExact((Addressable) null);
77     }
78 
79 }