1 /*
2 * Copyright (c) 2010, 2013, 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. Oracle designates this
8 * particular file as subject to the "Classpath" exception as provided
9 * by Oracle in the LICENSE file that accompanied this code.
10 *
11 * This code is distributed in the hope that it will be useful, but WITHOUT
12 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
13 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
14 * version 2 for more details (a copy is included in the LICENSE file that
15 * accompanied this code).
16 *
17 * You should have received a copy of the GNU General Public License version
18 * 2 along with this work; if not, write to the Free Software Foundation,
19 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
20 *
21 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
22 * or visit www.oracle.com if you need additional information or have any
23 * questions.
24 */
25
26 package jdk.nashorn.internal.tools.nasgen;
27
28 import static jdk.internal.org.objectweb.asm.Opcodes.ACC_FINAL;
29 import static jdk.internal.org.objectweb.asm.Opcodes.ACC_PUBLIC;
30 import static jdk.internal.org.objectweb.asm.Opcodes.H_INVOKESTATIC;
31 import static jdk.internal.org.objectweb.asm.Opcodes.V1_7;
32 import static jdk.nashorn.internal.tools.nasgen.StringConstants.CONSTRUCTOR_SUFFIX;
33 import static jdk.nashorn.internal.tools.nasgen.StringConstants.DEFAULT_INIT_DESC;
34 import static jdk.nashorn.internal.tools.nasgen.StringConstants.INIT;
35 import static jdk.nashorn.internal.tools.nasgen.StringConstants.OBJECT_DESC;
36 import static jdk.nashorn.internal.tools.nasgen.StringConstants.PROPERTYMAP_DESC;
37 import static jdk.nashorn.internal.tools.nasgen.StringConstants.PROPERTYMAP_FIELD_NAME;
38 import static jdk.nashorn.internal.tools.nasgen.StringConstants.PROTOTYPEOBJECT_SETCONSTRUCTOR;
39 import static jdk.nashorn.internal.tools.nasgen.StringConstants.PROTOTYPEOBJECT_SETCONSTRUCTOR_DESC;
40 import static jdk.nashorn.internal.tools.nasgen.StringConstants.PROTOTYPEOBJECT_TYPE;
41 import static jdk.nashorn.internal.tools.nasgen.StringConstants.SCRIPTFUNCTIONIMPL_INIT_DESC3;
42 import static jdk.nashorn.internal.tools.nasgen.StringConstants.SCRIPTFUNCTIONIMPL_INIT_DESC4;
43 import static jdk.nashorn.internal.tools.nasgen.StringConstants.SCRIPTFUNCTIONIMPL_TYPE;
44 import static jdk.nashorn.internal.tools.nasgen.StringConstants.SCRIPTFUNCTION_SETARITY;
45 import static jdk.nashorn.internal.tools.nasgen.StringConstants.SCRIPTFUNCTION_SETARITY_DESC;
46 import static jdk.nashorn.internal.tools.nasgen.StringConstants.SCRIPTFUNCTION_SETPROTOTYPE;
47 import static jdk.nashorn.internal.tools.nasgen.StringConstants.SCRIPTFUNCTION_SETPROTOTYPE_DESC;
48 import static jdk.nashorn.internal.tools.nasgen.StringConstants.SCRIPTFUNCTION_TYPE;
49 import static jdk.nashorn.internal.tools.nasgen.StringConstants.SCRIPTOBJECT_INIT_DESC;
50 import static jdk.nashorn.internal.tools.nasgen.StringConstants.SCRIPTOBJECT_TYPE;
51
52 import java.io.FileOutputStream;
53 import java.io.IOException;
54 import java.util.List;
55 import jdk.internal.org.objectweb.asm.Handle;
56
57 /**
58 * This class generates constructor class for a @ClassInfo annotated class.
59 *
60 */
61 public class ConstructorGenerator extends ClassGenerator {
62 private final ScriptClassInfo scriptClassInfo;
63 private final String className;
64 private final MemberInfo constructor;
65 private final int memberCount;
66 private final List<MemberInfo> specs;
67
68 ConstructorGenerator(final ScriptClassInfo sci) {
69 this.scriptClassInfo = sci;
70
71 this.className = scriptClassInfo.getConstructorClassName();
72 this.constructor = scriptClassInfo.getConstructor();
73 this.memberCount = scriptClassInfo.getConstructorMemberCount();
74 this.specs = scriptClassInfo.getSpecializedConstructors();
75 }
76
77 byte[] getClassBytes() {
78 // new class extensing from ScriptObject
79 final String superClass = (constructor != null)? SCRIPTFUNCTIONIMPL_TYPE : SCRIPTOBJECT_TYPE;
80 cw.visit(V1_7, ACC_FINAL, className, null, superClass, null);
81 if (memberCount > 0) {
82 // add fields
83 emitFields();
84 // add <clinit>
85 emitStaticInitializer();
86 }
87 // add <init>
88 emitConstructor();
89
90 if (constructor == null) {
91 emitGetClassName(scriptClassInfo.getName());
92 }
93
94 cw.visitEnd();
95 return cw.toByteArray();
96 }
97
98 // --Internals only below this point
99 private void emitFields() {
100 // Introduce "Function" type instance fields for each
101 // constructor @Function in script class and introduce instance
102 // fields for each constructor @Property in the script class.
103 for (MemberInfo memInfo : scriptClassInfo.getMembers()) {
104 if (memInfo.isConstructorFunction()) {
105 addFunctionField(memInfo.getJavaName());
106 memInfo = (MemberInfo)memInfo.clone();
107 memInfo.setJavaDesc(OBJECT_DESC);
108 memInfo.setJavaAccess(ACC_PUBLIC);
109 addGetter(className, memInfo);
110 addSetter(className, memInfo);
111 } else if (memInfo.isConstructorProperty()) {
112 if (memInfo.isStaticFinal()) {
113 addGetter(scriptClassInfo.getJavaName(), memInfo);
114 } else {
115 addField(memInfo.getJavaName(), memInfo.getJavaDesc());
116 memInfo = (MemberInfo)memInfo.clone();
117 memInfo.setJavaAccess(ACC_PUBLIC);
118 addGetter(className, memInfo);
119 addSetter(className, memInfo);
120 }
121 }
122 }
123
124 addMapField();
125 }
126
127 private void emitStaticInitializer() {
128 final MethodGenerator mi = makeStaticInitializer();
129 emitStaticInitPrefix(mi, className, memberCount);
130
131 for (final MemberInfo memInfo : scriptClassInfo.getMembers()) {
132 if (memInfo.isConstructorFunction() || memInfo.isConstructorProperty()) {
133 linkerAddGetterSetter(mi, className, memInfo);
134 } else if (memInfo.isConstructorGetter()) {
135 final MemberInfo setter = scriptClassInfo.findSetter(memInfo);
136 linkerAddGetterSetter(mi, scriptClassInfo.getJavaName(), memInfo, setter);
137 }
138 }
139 emitStaticInitSuffix(mi, className);
140 }
141
142 private void emitConstructor() {
143 final MethodGenerator mi = makeConstructor();
144 mi.visitCode();
145 callSuper(mi);
146
147 if (memberCount > 0) {
148 // initialize Function type fields
149 initFunctionFields(mi);
150 // initialize data fields
151 initDataFields(mi);
152 }
153
154 if (constructor != null) {
155 final int arity = constructor.getArity();
156 if (arity != MemberInfo.DEFAULT_ARITY) {
157 mi.loadThis();
158 mi.push(arity);
159 mi.invokeVirtual(SCRIPTFUNCTION_TYPE, SCRIPTFUNCTION_SETARITY,
160 SCRIPTFUNCTION_SETARITY_DESC);
161 }
162 }
163 mi.returnVoid();
164 mi.computeMaxs();
165 mi.visitEnd();
166 }
167
168 private void loadMap(final MethodGenerator mi) {
169 if (memberCount > 0) {
170 mi.getStatic(className, PROPERTYMAP_FIELD_NAME, PROPERTYMAP_DESC);
171 }
172 }
173
174 private void callSuper(final MethodGenerator mi) {
175 String superClass, superDesc;
176 mi.loadThis();
177 if (constructor == null) {
178 // call ScriptObject.<init>
179 superClass = SCRIPTOBJECT_TYPE;
180 superDesc = (memberCount > 0) ? SCRIPTOBJECT_INIT_DESC : DEFAULT_INIT_DESC;
181 loadMap(mi);
182 } else {
183 // call Function.<init>
184 superClass = SCRIPTFUNCTIONIMPL_TYPE;
185 superDesc = (memberCount > 0) ? SCRIPTFUNCTIONIMPL_INIT_DESC4 : SCRIPTFUNCTIONIMPL_INIT_DESC3;
186 mi.loadLiteral(constructor.getName());
187 mi.visitLdcInsn(new Handle(H_INVOKESTATIC, scriptClassInfo.getJavaName(), constructor.getJavaName(), constructor.getJavaDesc()));
188 loadMap(mi);
189 mi.memberInfoArray(scriptClassInfo.getJavaName(), specs); //pushes null if specs empty
190 }
191
192 mi.invokeSpecial(superClass, INIT, superDesc);
193 }
194
195 private void initFunctionFields(final MethodGenerator mi) {
196 for (final MemberInfo memInfo : scriptClassInfo.getMembers()) {
197 if (!memInfo.isConstructorFunction()) {
198 continue;
199 }
200 mi.loadThis();
201 newFunction(mi, scriptClassInfo.getJavaName(), memInfo, scriptClassInfo.findSpecializations(memInfo.getJavaName()));
202 mi.putField(className, memInfo.getJavaName(), OBJECT_DESC);
203 }
204 }
205
206 private void initDataFields(final MethodGenerator mi) {
207 for (final MemberInfo memInfo : scriptClassInfo.getMembers()) {
208 if (!memInfo.isConstructorProperty() || memInfo.isFinal()) {
209 continue;
210 }
211 final Object value = memInfo.getValue();
212 if (value != null) {
213 mi.loadThis();
214 mi.loadLiteral(value);
215 mi.putField(className, memInfo.getJavaName(), memInfo.getJavaDesc());
216 } else if (!memInfo.getInitClass().isEmpty()) {
217 final String clazz = memInfo.getInitClass();
218 mi.loadThis();
219 mi.newObject(clazz);
220 mi.dup();
221 mi.invokeSpecial(clazz, INIT, DEFAULT_INIT_DESC);
222 mi.putField(className, memInfo.getJavaName(), memInfo.getJavaDesc());
223 }
224 }
225
226 if (constructor != null) {
227 mi.loadThis();
228 final String protoName = scriptClassInfo.getPrototypeClassName();
229 mi.newObject(protoName);
230 mi.dup();
231 mi.invokeSpecial(protoName, INIT, DEFAULT_INIT_DESC);
232 mi.dup();
233 mi.loadThis();
234 mi.invokeStatic(PROTOTYPEOBJECT_TYPE, PROTOTYPEOBJECT_SETCONSTRUCTOR,
235 PROTOTYPEOBJECT_SETCONSTRUCTOR_DESC);
236 mi.invokeVirtual(SCRIPTFUNCTION_TYPE, SCRIPTFUNCTION_SETPROTOTYPE, SCRIPTFUNCTION_SETPROTOTYPE_DESC);
237 }
238 }
239
240 /**
241 * Entry point for ConstructorGenerator run separately as an application. Will display
242 * usage. Takes one argument, a class name.
243 * @param args args vector
244 * @throws IOException if class can't be read
245 */
246 public static void main(final String[] args) throws IOException {
247 if (args.length != 1) {
248 System.err.println("Usage: " + ConstructorGenerator.class.getName() + " <class>");
249 System.exit(1);
250 }
251
252 final String className = args[0].replace('.', '/');
253 final ScriptClassInfo sci = getScriptClassInfo(className + ".class");
254 if (sci == null) {
255 System.err.println("No @ScriptClass in " + className);
256 System.exit(2);
257 throw new IOException(); // get rid of warning for sci.verify() below - may be null
258 }
259
260 try {
261 sci.verify();
262 } catch (final Exception e) {
263 System.err.println(e.getMessage());
264 System.exit(3);
265 }
266 final ConstructorGenerator gen = new ConstructorGenerator(sci);
267 try (FileOutputStream fos = new FileOutputStream(className + CONSTRUCTOR_SUFFIX + ".class")) {
268 fos.write(gen.getClassBytes());
269 }
270 }
271 }