DexMaker.java revision 579d7739c53a2707ad711a2d2cae46d7d782f06
1579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson/* 2579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * Copyright (C) 2011 The Android Open Source Project 3579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * 4579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * Licensed under the Apache License, Version 2.0 (the "License"); 5579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * you may not use this file except in compliance with the License. 6579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * You may obtain a copy of the License at 7579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * 8579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * http://www.apache.org/licenses/LICENSE-2.0 9579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * 10579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * Unless required by applicable law or agreed to in writing, software 11579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * distributed under the License is distributed on an "AS IS" BASIS, 12579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * See the License for the specific language governing permissions and 14579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * limitations under the License. 15579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson */ 16579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 17579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilsonpackage com.google.dexmaker; 18579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 19579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilsonimport static com.android.dx.rop.code.AccessFlags.ACC_CONSTRUCTOR; 20579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilsonimport static com.android.dx.rop.code.AccessFlags.ACC_PRIVATE; 21579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilsonimport static com.android.dx.rop.code.AccessFlags.ACC_STATIC; 22579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 23579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilsonimport com.android.dx.dex.DexFormat; 24579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilsonimport com.android.dx.dex.DexOptions; 25579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilsonimport com.android.dx.dex.code.DalvCode; 26579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilsonimport com.android.dx.dex.code.PositionList; 27579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilsonimport com.android.dx.dex.code.RopTranslator; 28579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilsonimport com.android.dx.dex.file.ClassDefItem; 29579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilsonimport com.android.dx.dex.file.DexFile; 30579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilsonimport com.android.dx.dex.file.EncodedField; 31579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilsonimport com.android.dx.dex.file.EncodedMethod; 32579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilsonimport com.android.dx.rop.code.AccessFlags; 33579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilsonimport com.android.dx.rop.code.LocalVariableInfo; 34579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilsonimport com.android.dx.rop.code.RopMethod; 35579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilsonimport com.android.dx.rop.cst.CstString; 36579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilsonimport com.android.dx.rop.cst.CstType; 37579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilsonimport com.android.dx.rop.type.StdTypeList; 38579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 39579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilsonimport java.io.File; 40579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilsonimport java.io.FileOutputStream; 41579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilsonimport java.io.IOException; 42579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilsonimport java.lang.reflect.InvocationTargetException; 43579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilsonimport java.util.LinkedHashMap; 44579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilsonimport java.util.Map; 45579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilsonimport java.util.jar.JarEntry; 46579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilsonimport java.util.jar.JarOutputStream; 47579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 48579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson/** 49579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * Define types, fields and methods. 50579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson */ 51579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilsonpublic final class DexGenerator { 52579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson private final Map<Type<?>, TypeDeclaration> types 53579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson = new LinkedHashMap<Type<?>, TypeDeclaration>(); 54579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 55579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson private TypeDeclaration getTypeDeclaration(Type<?> type) { 56579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson TypeDeclaration result = types.get(type); 57579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson if (result == null) { 58579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson result = new TypeDeclaration(type); 59579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson types.put(type, result); 60579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 61579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson return result; 62579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 63579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 64579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson /** 65579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * @param flags any flags masked by {@link com.android.dx.rop.code.AccessFlags#CLASS_FLAGS}. 66579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson */ 67579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson public void declare(Type<?> type, String sourceFile, int flags, 68579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson Type<?> supertype, Type<?>... interfaces) { 69579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson TypeDeclaration declaration = getTypeDeclaration(type); 70579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson if (declaration.declared) { 71579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson throw new IllegalStateException("already declared: " + type); 72579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 73579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson declaration.declared = true; 74579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson declaration.flags = flags; 75579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson declaration.supertype = supertype; 76579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson declaration.sourceFile = sourceFile; 77579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson declaration.interfaces = new TypeList(interfaces); 78579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 79579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 80579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson /** 81579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * @param flags any flags masked by {@link com.android.dx.rop.code.AccessFlags#METHOD_FLAGS}. 82579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson */ 83579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson public Code declare(MethodId<?, ?> method, int flags) { 84579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson TypeDeclaration typeDeclaration = getTypeDeclaration(method.declaringType); 85579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson if (typeDeclaration.methods.containsKey(method)) { 86579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson throw new IllegalStateException("already declared: " + method); 87579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 88579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson MethodDeclaration methodDeclaration = new MethodDeclaration(method, flags); 89579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson typeDeclaration.methods.put(method, methodDeclaration); 90579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson return methodDeclaration.code; 91579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 92579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 93579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson /** 94579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * @param flags any flags masked by {@link AccessFlags#FIELD_FLAGS}. 95579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson */ 96579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson public void declare(FieldId<?, ?> fieldId, int flags, Object staticValue) { 97579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson TypeDeclaration typeDeclaration = getTypeDeclaration(fieldId.declaringType); 98579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson if (typeDeclaration.fields.containsKey(fieldId)) { 99579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson throw new IllegalStateException("already declared: " + fieldId); 100579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 101579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson FieldDeclaration fieldDeclaration = new FieldDeclaration(fieldId, flags, staticValue); 102579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson typeDeclaration.fields.put(fieldId, fieldDeclaration); 103579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 104579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 105579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson /** 106579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * Returns a .dex formatted file. 107579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson */ 108579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson public byte[] generate() { 109579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson DexOptions options = new DexOptions(); 110579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson options.targetApiLevel = DexFormat.API_NO_EXTENDED_OPCODES; 111579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson DexFile outputDex = new DexFile(options); 112579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 113579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson for (TypeDeclaration typeDeclaration : types.values()) { 114579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson outputDex.add(typeDeclaration.toClassDefItem()); 115579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 116579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 117579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson try { 118579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson return outputDex.toDex(null, false); 119579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } catch (IOException e) { 120579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson throw new RuntimeException(e); 121579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 122579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 123579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 124579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson /** 125579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * Loads the generated types into the current process. 126579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * 127579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * <p>All parameters are optional, you may pass {@code null} and suitable 128579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * defaults will be used. 129579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * 130579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * <p>If you opt to provide your own output directories, take care to 131579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * ensure that they are not world-readable, otherwise a malicious app will 132579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * be able to inject code to run. A suitable parameter for these output 133579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * directories would be something like this: 134579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * {@code getApplicationContext().getDir("dx", Context.MODE_PRIVATE); } 135579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * 136579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * @param parent the parent ClassLoader to be used when loading 137579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * our generated types 138579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * @param dexOutputDir the destination directory wherein we will write 139579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * emitted .dex files before they end up in the cache directory 140579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * @param dexOptCacheDir where optimized .dex files are to be written 141579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson */ 142579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson public ClassLoader load(ClassLoader parent, File dexOutputDir, File dexOptCacheDir) 143579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson throws IOException { 144579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson byte[] dex = generate(); 145579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 146579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson /* 147579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * This implementation currently dumps the dex to the filesystem. It 148579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * jars the emitted .dex for the benefit of Gingerbread and earlier 149579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * devices, which can't load .dex files directly. 150579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * 151579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * TODO: load the dex from memory where supported. 152579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson */ 153579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson File result = File.createTempFile("Generated", ".jar", dexOutputDir); 154579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson result.deleteOnExit(); 155579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson JarOutputStream jarOut = new JarOutputStream(new FileOutputStream(result)); 156579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson jarOut.putNextEntry(new JarEntry(DexFormat.DEX_IN_JAR_NAME)); 157579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson jarOut.write(dex); 158579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson jarOut.closeEntry(); 159579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson jarOut.close(); 160579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson try { 161579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson return (ClassLoader) Class.forName("dalvik.system.DexClassLoader") 162579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson .getConstructor(String.class, String.class, String.class, ClassLoader.class) 163579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson .newInstance(result.getPath(), dexOptCacheDir.getAbsolutePath(), null, parent); 164579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } catch (ClassNotFoundException e) { 165579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson throw new UnsupportedOperationException("load() requires a Dalvik VM", e); 166579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } catch (InvocationTargetException e) { 167579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson throw new RuntimeException(e.getCause()); 168579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } catch (InstantiationException e) { 169579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson throw new AssertionError(); 170579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } catch (NoSuchMethodException e) { 171579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson throw new AssertionError(); 172579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } catch (IllegalAccessException e) { 173579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson throw new AssertionError(); 174579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 175579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 176579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 177579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson private static class TypeDeclaration { 178579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson private final Type<?> type; 179579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 180579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson /** declared state */ 181579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson private boolean declared; 182579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson private int flags; 183579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson private Type<?> supertype; 184579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson private String sourceFile; 185579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson private TypeList interfaces; 186579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 187579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson private final Map<FieldId, FieldDeclaration> fields 188579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson = new LinkedHashMap<FieldId, FieldDeclaration>(); 189579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson private final Map<MethodId, MethodDeclaration> methods 190579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson = new LinkedHashMap<MethodId, MethodDeclaration>(); 191579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 192579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson TypeDeclaration(Type<?> type) { 193579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson this.type = type; 194579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 195579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 196579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson ClassDefItem toClassDefItem() { 197579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson if (!declared) { 198579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson throw new IllegalStateException("Undeclared type " + type + " declares members: " 199579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson + fields.keySet() + " " + methods.keySet()); 200579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 201579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 202579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson DexOptions dexOptions = new DexOptions(); 203579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson dexOptions.targetApiLevel = DexFormat.API_NO_EXTENDED_OPCODES; 204579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 205579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson CstType thisType = type.constant; 206579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 207579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson ClassDefItem out = new ClassDefItem(thisType, flags, supertype.constant, 208579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson interfaces.ropTypes, new CstString(sourceFile)); 209579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 210579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson for (MethodDeclaration method : methods.values()) { 211579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson EncodedMethod encoded = method.toEncodedMethod(dexOptions); 212579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson if (method.isDirect()) { 213579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson out.addDirectMethod(encoded); 214579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } else { 215579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson out.addVirtualMethod(encoded); 216579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 217579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 218579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson for (FieldDeclaration field : fields.values()) { 219579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson EncodedField encoded = field.toEncodedField(); 220579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson if (field.isStatic()) { 221579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson out.addStaticField(encoded, Constants.getConstant(field.staticValue)); 222579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } else { 223579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson out.addInstanceField(encoded); 224579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 225579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 226579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 227579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson return out; 228579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 229579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 230579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 231579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson static class FieldDeclaration { 232579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson final FieldId<?, ?> fieldId; 233579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson private final int accessFlags; 234579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson private final Object staticValue; 235579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 236579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson FieldDeclaration(FieldId<?, ?> fieldId, int accessFlags, Object staticValue) { 237579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson if ((accessFlags & (AccessFlags.ACC_STATIC)) == 0 && staticValue != null) { 238579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson throw new IllegalArgumentException("instance fields may not have a value"); 239579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 240579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson this.fieldId = fieldId; 241579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson this.accessFlags = accessFlags; 242579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson this.staticValue = staticValue; 243579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 244579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 245579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson EncodedField toEncodedField() { 246579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson return new EncodedField(fieldId.constant, accessFlags); 247579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 248579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 249579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson public boolean isStatic() { 250579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson return (accessFlags & (AccessFlags.ACC_STATIC)) != 0; 251579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 252579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 253579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 254579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson static class MethodDeclaration { 255579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson final MethodId<?, ?> method; 256579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson private final int flags; 257579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson private final Code code; 258579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 259579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson public MethodDeclaration(MethodId<?, ?> method, int flags) { 260579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson this.method = method; 261579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson this.flags = flags; 262579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson this.code = new Code(this); 263579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 264579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 265579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson boolean isStatic() { 266579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson return (flags & ACC_STATIC) != 0; 267579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 268579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 269579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson boolean isDirect() { 270579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson return (flags & (ACC_STATIC | ACC_PRIVATE | ACC_CONSTRUCTOR)) != 0; 271579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 272579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 273579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson EncodedMethod toEncodedMethod(DexOptions dexOptions) { 274579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson RopMethod ropMethod = new RopMethod(code.toBasicBlocks(), 0); 275579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson LocalVariableInfo locals = null; 276579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson DalvCode dalvCode = RopTranslator.translate( 277579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson ropMethod, PositionList.NONE, locals, code.paramSize(), dexOptions); 278579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson return new EncodedMethod(method.constant, flags, dalvCode, StdTypeList.EMPTY); 279579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 280579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 281579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson} 282