1 /*
  2  * Copyright (c) 2024, 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 import org.testng.Assert;
 25 import org.testng.annotations.Test;
 26 
 27 import java.lang.reflect.code.OpTransformer;
 28 import java.lang.reflect.code.op.CoreOp;
 29 import java.lang.reflect.code.Op;
 30 import java.lang.reflect.code.bytecode.BytecodeGenerator;
 31 import java.lang.invoke.MethodHandle;
 32 import java.lang.invoke.MethodHandles;
 33 import java.lang.reflect.Method;
 34 import java.lang.runtime.CodeReflection;
 35 import java.util.Optional;
 36 import java.util.stream.Stream;
 37 
 38 /*
 39  * @test
 40  * @run testng TestArrayCreation
 41  */
 42 
 43 public class TestArrayCreation {
 44     @CodeReflection
 45     public static String[] f() {
 46         return new String[10];
 47     }
 48 
 49     @Test
 50     public void testf() throws Throwable {
 51         CoreOp.FuncOp f = getFuncOp("f");
 52 
 53         MethodHandle mh = generate(f);
 54 
 55         Assert.assertEquals((String[]) mh.invoke(), f());
 56     }
 57 
 58     @CodeReflection
 59     public static String[][] f2() {
 60         return new String[10][];
 61     }
 62 
 63     @Test
 64     public void testf2() throws Throwable {
 65         CoreOp.FuncOp f = getFuncOp("f2");
 66 
 67         MethodHandle mh = generate(f);
 68 
 69         Assert.assertEquals((String[][]) mh.invoke(), f2());
 70     }
 71 
 72     @CodeReflection
 73     public static String[][] f3() {
 74         return new String[10][10];
 75     }
 76 
 77     @Test
 78     public void testf3() throws Throwable {
 79         CoreOp.FuncOp f = getFuncOp("f3");
 80 
 81         MethodHandle mh = generate(f);
 82 
 83         Assert.assertEquals((String[][]) mh.invoke(), f3());
 84     }
 85 
 86     @CodeReflection
 87     public static String[][] f4() {
 88         return new String[][]{{"one", "two"}, {"three"}};
 89     }
 90 
 91     @Test
 92     public void testf4() throws Throwable {
 93         CoreOp.FuncOp f = getFuncOp("f4");
 94 
 95         MethodHandle mh = generate(f);
 96 
 97         Assert.assertEquals((String[][]) mh.invoke(), f4());
 98     }
 99 
100     static MethodHandle generate(CoreOp.FuncOp f) {
101         f.writeTo(System.out);
102 
103         CoreOp.FuncOp lf = f.transform(OpTransformer.LOWERING_TRANSFORMER);
104         lf.writeTo(System.out);
105 
106         return BytecodeGenerator.generate(MethodHandles.lookup(), lf);
107     }
108 
109     static CoreOp.FuncOp getFuncOp(String name) {
110         Optional<Method> om = Stream.of(TestArrayCreation.class.getDeclaredMethods())
111                 .filter(m -> m.getName().equals(name))
112                 .findFirst();
113 
114         Method m = om.get();
115         return m.getCodeModel().get();
116     }
117 }