1 /* 2 * Copyright 2013, Google Inc. 3 * All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions are 7 * met: 8 * 9 * * Redistributions of source code must retain the above copyright 10 * notice, this list of conditions and the following disclaimer. 11 * * Redistributions in binary form must reproduce the above 12 * copyright notice, this list of conditions and the following disclaimer 13 * in the documentation and/or other materials provided with the 14 * distribution. 15 * * Neither the name of Google Inc. nor the names of its 16 * contributors may be used to endorse or promote products derived from 17 * this software without specific prior written permission. 18 * 19 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 20 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 21 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 22 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 23 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 24 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 25 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 26 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 27 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 28 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 29 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 30 */ 31 32 package org.jf.dexlib2; 33 34 import com.google.common.collect.Maps; 35 import com.google.common.collect.RangeMap; 36 37 import javax.annotation.Nonnull; 38 import javax.annotation.Nullable; 39 import java.util.EnumMap; 40 import java.util.HashMap; 41 42 import static org.jf.dexlib2.VersionMap.NO_VERSION; 43 import static org.jf.dexlib2.VersionMap.mapApiToArtVersion; 44 import static org.jf.dexlib2.VersionMap.mapArtVersionToApi; 45 46 public class Opcodes { 47 48 /** 49 * Either the api level for dalvik opcodes, or the art version for art opcodes 50 */ 51 public final int api; 52 public final int artVersion; 53 @Nonnull private final Opcode[] opcodesByValue = new Opcode[256]; 54 @Nonnull private final EnumMap<Opcode, Short> opcodeValues; 55 @Nonnull private final HashMap<String, Opcode> opcodesByName; 56 57 @Nonnull forApi(int api)58 public static Opcodes forApi(int api) { 59 return new Opcodes(api, NO_VERSION); 60 } 61 62 @Nonnull forArtVersion(int artVersion)63 public static Opcodes forArtVersion(int artVersion) { 64 return new Opcodes(NO_VERSION, artVersion); 65 } 66 67 @Nonnull forDexVersion(int dexVersion)68 public static Opcodes forDexVersion(int dexVersion) { 69 int api = VersionMap.mapDexVersionToApi(dexVersion); 70 if (api == NO_VERSION) { 71 throw new RuntimeException("Unsupported dex version " + dexVersion); 72 } 73 return new Opcodes(api, NO_VERSION); 74 } 75 76 /** 77 * @return a default Opcodes instance for when the exact Opcodes to use doesn't matter or isn't known 78 */ 79 @Nonnull getDefault()80 public static Opcodes getDefault() { 81 // The last pre-art api 82 return forApi(20); 83 } 84 Opcodes(int api, int artVersion)85 private Opcodes(int api, int artVersion) { 86 if (api >= 21) { 87 this.api = api; 88 this.artVersion = mapApiToArtVersion(api); 89 } else if (artVersion >= 0 && artVersion < 39) { 90 this.api = mapArtVersionToApi(artVersion); 91 this.artVersion = artVersion; 92 } else { 93 this.api = api; 94 this.artVersion = artVersion; 95 } 96 97 opcodeValues = new EnumMap<Opcode, Short>(Opcode.class); 98 opcodesByName = Maps.newHashMap(); 99 100 int version; 101 if (isArt()) { 102 version = this.artVersion; 103 } else { 104 version = this.api; 105 } 106 107 for (Opcode opcode: Opcode.values()) { 108 RangeMap<Integer, Short> versionToValueMap; 109 110 if (isArt()) { 111 versionToValueMap = opcode.artVersionToValueMap; 112 } else { 113 versionToValueMap = opcode.apiToValueMap; 114 } 115 116 Short opcodeValue = versionToValueMap.get(version); 117 if (opcodeValue != null) { 118 if (!opcode.format.isPayloadFormat) { 119 opcodesByValue[opcodeValue] = opcode; 120 } 121 opcodeValues.put(opcode, opcodeValue); 122 opcodesByName.put(opcode.name.toLowerCase(), opcode); 123 } 124 } 125 } 126 127 @Nullable getOpcodeByName(@onnull String opcodeName)128 public Opcode getOpcodeByName(@Nonnull String opcodeName) { 129 return opcodesByName.get(opcodeName.toLowerCase()); 130 } 131 132 @Nullable getOpcodeByValue(int opcodeValue)133 public Opcode getOpcodeByValue(int opcodeValue) { 134 switch (opcodeValue) { 135 case 0x100: 136 return Opcode.PACKED_SWITCH_PAYLOAD; 137 case 0x200: 138 return Opcode.SPARSE_SWITCH_PAYLOAD; 139 case 0x300: 140 return Opcode.ARRAY_PAYLOAD; 141 default: 142 if (opcodeValue >= 0 && opcodeValue < opcodesByValue.length) { 143 return opcodesByValue[opcodeValue]; 144 } 145 return null; 146 } 147 } 148 149 @Nullable getOpcodeValue(@onnull Opcode opcode)150 public Short getOpcodeValue(@Nonnull Opcode opcode) { 151 return opcodeValues.get(opcode); 152 } 153 isArt()154 public boolean isArt() { 155 return artVersion != NO_VERSION; 156 } 157 } 158