1 /*
 2  *  Copyright (c) 2019, 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 /*
26  * @test
27  * @modules jdk.incubator.foreign/jdk.incubator.foreign.unsafe
28  *          jdk.incubator.foreign/jdk.internal.foreign
29  *          jdk.incubator.foreign/jdk.internal.foreign.abi
30  *          java.base/sun.security.action
31  * @build NativeTestHelper CallGeneratorHelper TestDowncall
32  *
33  * @run testng/othervm -Dforeign.restricted=permit TestDowncall
34  */
35 
36 import jdk.incubator.foreign.CSupport;
37 import jdk.incubator.foreign.FunctionDescriptor;
38 import jdk.incubator.foreign.LibraryLookup;
39 import jdk.incubator.foreign.MemoryAddress;
40 import jdk.incubator.foreign.MemoryLayout;
41 import jdk.incubator.foreign.ForeignLinker;
42 
43 import java.lang.invoke.MethodHandle;
44 import java.lang.invoke.MethodType;
45 import java.util.ArrayList;
46 import java.util.List;
47 import java.util.function.Consumer;
48 
49 import org.testng.annotations.*;
50 
51 public class TestDowncall extends CallGeneratorHelper {
52 
53     static LibraryLookup lib = LibraryLookup.ofLibrary("TestDowncall");
54     static ForeignLinker abi = CSupport.getSystemLinker();
55 
56 
57     @Test(dataProvider="functions", dataProviderClass=CallGeneratorHelper.class)
58     public void testDowncall(String fName, Ret ret, List<ParamType> paramTypes, List<StructFieldType> fields) throws Throwable {
59         List<Consumer<Object>> checks = new ArrayList<>();
60         MemoryAddress addr = lib.lookup(fName);
61         MethodHandle mh = abi.downcallHandle(addr, methodType(ret, paramTypes, fields), function(ret, paramTypes, fields));
62         Object[] args = makeArgs(paramTypes, fields, checks);
63         mh = mh.asSpreader(Object[].class, paramTypes.size());
64         Object res = mh.invoke(args);
65         if (ret == Ret.NON_VOID) {
66             checks.forEach(c -> c.accept(res));
67         }
68         for (Object arg : args) {
69             cleanup(arg);
70         }
71     }
72 
73     static MethodType methodType(Ret ret, List<ParamType> params, List<StructFieldType> fields) {
74         MethodType mt = ret == Ret.VOID ?
75                 MethodType.methodType(void.class) : MethodType.methodType(paramCarrier(params.get(0).layout(fields)));
76         for (ParamType p : params) {
77             mt = mt.appendParameterTypes(paramCarrier(p.layout(fields)));
78         }
79         return mt;
80     }
81 
82     static FunctionDescriptor function(Ret ret, List<ParamType> params, List<StructFieldType> fields) {
83         MemoryLayout[] paramLayouts = params.stream().map(p -> p.layout(fields)).toArray(MemoryLayout[]::new);
84         return ret == Ret.VOID ?
85                 FunctionDescriptor.ofVoid(paramLayouts) :
86                 FunctionDescriptor.of(paramLayouts[0], paramLayouts);
87     }
88 
89     static Object[] makeArgs(List<ParamType> params, List<StructFieldType> fields, List<Consumer<Object>> checks) throws ReflectiveOperationException {
90         Object[] args = new Object[params.size()];
91         for (int i = 0 ; i < params.size() ; i++) {
92             args[i] = makeArg(params.get(i).layout(fields), checks, i == 0);
93         }
94         return args;
95     }
96 }