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