1 /*
2 * Copyright (c) 2018, 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 * @summary test value bootstrap methods
28 * @modules java.base/jdk.internal.org.objectweb.asm
29 * @compile -XDallowWithFieldOperator ValueBootstrapMethods.java
30 * @run main/othervm -XX:+EnableValhalla -Dvalue.bsm.salt=1 ValueBootstrapMethods
31 */
32
33 import java.io.IOException;
34 import java.lang.invoke.CallSite;
35 import java.lang.invoke.MethodHandles;
36 import java.lang.invoke.MethodType;
37 import java.lang.reflect.*;
38 import java.nio.file.Files;
39 import java.nio.file.Path;
40 import java.nio.file.Paths;
41 import java.util.HashSet;
42 import java.util.List;
43 import java.util.Objects;
44 import java.util.Set;
45
46 import jdk.internal.org.objectweb.asm.Attribute;
47 import jdk.internal.org.objectweb.asm.ByteVector;
48 import jdk.internal.org.objectweb.asm.ClassReader;
49 import jdk.internal.org.objectweb.asm.ClassWriter;
50 import jdk.internal.org.objectweb.asm.Handle;
51 import jdk.internal.org.objectweb.asm.Label;
52 import jdk.internal.org.objectweb.asm.MethodVisitor;
53 import jdk.internal.org.objectweb.asm.Type;
54 import static jdk.internal.org.objectweb.asm.Opcodes.*;
55
56 public class ValueBootstrapMethods {
57 private static final String TEST_CLASSES = System.getProperty("test.classes", ".");
58
59 public static void main(String... args) throws Throwable {
60 Class<?> test = valueTestClass();
61 Value value = Value.make(10, 5.03, "foo", "bar", "goo");
62
63 Class<?> valueClass = Value.class.asValueType();
64 Method hashCode = test.getMethod("hashCode", valueClass);
65 int hash = (int)hashCode.invoke(null, value);
66 assertEquals(hash, value.hashCode());
67
68 Method toString = test.getMethod("toString", valueClass);
69 String s = (String)toString.invoke(null, value);
70 assertEquals(s, value.toString());
71
72 Method equals = test.getMethod("equals", valueClass, Object.class);
73 boolean rc = (boolean)equals.invoke(null, value, value);
74 if (!rc) {
75 throw new RuntimeException("expected equals");
76 }
77 }
78
79 public static final inline class Value {
80 private final int i;
81 private final double d;
82 private final String s;
83 private final List<String> l;
84 Value() {
85 this.i = 0;
86 this.d = 0;
87 this.s = "default";
88 this.l = List.of();
89 }
90 public static Value make(int i, double d, String s, String... items) {
91 Value v = Value.default;
92 v = __WithField(v.i, i);
93 v = __WithField(v.d, d);
94 v = __WithField(v.s, s);
95 v = __WithField(v.l, List.of(items));
96 return v;
97 }
98
99 List<Object> values() {
100 return List.of(Value.class.asValueType(), i, d, s, l);
101 }
102
103 public int hashCode() {
104 return values().hashCode();
105 }
106
107 public String toString() {
108 System.out.println(l);
109 return String.format("[%s i=%s d=%s s=%s l=%s]", Value.class.getName(),
110 i, String.valueOf(d), s, l.toString());
111 }
112 }
113
114 /*
115 * Generate ValueTest class
116 */
117 private static Class<?> valueTestClass() throws Exception {
118 Path path = Paths.get(TEST_CLASSES, "ValueTest.class");
119 generate(Value.class.asValueType(), "ValueTest", path);
120 return Class.forName("ValueTest");
121 }
122
123 private static void assertEquals(Object o1, Object expected) {
124 if (!Objects.equals(o1, expected)) {
125 throw new RuntimeException(o1 + " expected: " + expected);
126 }
127 }
128
129 static final int CLASSFILE_VERSION = 56;
130 static void generate(Class<?> c, String className, Path path) throws IOException {
131 ClassWriter cw = new ClassWriter(ClassWriter.COMPUTE_MAXS + ClassWriter.COMPUTE_FRAMES);
132
133 cw.visit(CLASSFILE_VERSION,
134 ACC_SUPER + ACC_PUBLIC + ACC_FINAL + ACC_SYNTHETIC,
135 className,
136 null,
137 "java/lang/Object",
138 null
139 );
140
141
142 MethodType mt = MethodType.methodType(CallSite.class,
143 MethodHandles.Lookup.class, String.class, MethodType.class, Class.class);
144 Handle bootstrap = new Handle(H_INVOKESTATIC, Type.getInternalName(java.lang.invoke.ValueBootstrapMethods.class),
145 "makeBootstrapMethod", mt.toMethodDescriptorString(), false);
146
147 Type type = Type.getType(c);
148 MethodVisitor mv = cw.visitMethod(
149 ACC_PUBLIC + ACC_STATIC + ACC_FINAL,
150 "hashCode",
151 Type.getMethodDescriptor(Type.INT_TYPE, type),
152 null,
153 null);
154
155 mv.visitVarInsn(ALOAD, 0);
156 mv.visitInvokeDynamicInsn("hashCode",
157 Type.getMethodDescriptor(Type.INT_TYPE, type),
158 bootstrap, type);
159 mv.visitInsn(IRETURN);
160
161 mv.visitMaxs(-1, -1);
162 mv.visitEnd();
163
164 mv = cw.visitMethod(
165 ACC_PUBLIC + ACC_STATIC + ACC_FINAL,
166 "equals",
167 Type.getMethodDescriptor(Type.BOOLEAN_TYPE, type, Type.getType(Object.class)),
168 null,
169 null);
170
171 mv.visitVarInsn(ALOAD, 0);
172 mv.visitVarInsn(ALOAD, 1);
173 mv.visitInvokeDynamicInsn("equals",
174 Type.getMethodDescriptor(Type.BOOLEAN_TYPE, type, Type.getType(Object.class)),
175 bootstrap, type);
176 mv.visitInsn(IRETURN);
177 mv.visitMaxs(-1, -1);
178 mv.visitEnd();
179
180 mv = cw.visitMethod(
181 ACC_PUBLIC + ACC_STATIC + ACC_FINAL,
182 "toString",
183 Type.getMethodDescriptor(Type.getType(String.class), type),
184 null,
185 null);
186
187 mv.visitVarInsn(ALOAD, 0);
188 mv.visitInvokeDynamicInsn("toString",
189 Type.getMethodDescriptor(Type.getType(String.class), type),
190 bootstrap, type);
191 mv.visitInsn(ARETURN);
192 mv.visitMaxs(-1, -1);
193 mv.visitEnd();
194
195 cw.visitEnd();
196
197 byte[] classBytes = cw.toByteArray();
198 System.out.println("writing " + path);
199 Files.write(path, classBytes);
200 }
201 }