19bc4978139ca38e422ae89065d3062146534132dJesusFreke@JesusFreke.com/* 29bc4978139ca38e422ae89065d3062146534132dJesusFreke@JesusFreke.com * [The "BSD licence"] 300fc68adf2e39aeb9fed35293f2576bbe729ec4bJesusFreke@JesusFreke.com * Copyright (c) 2010 Ben Gruver (JesusFreke) 49bc4978139ca38e422ae89065d3062146534132dJesusFreke@JesusFreke.com * All rights reserved. 59bc4978139ca38e422ae89065d3062146534132dJesusFreke@JesusFreke.com * 69bc4978139ca38e422ae89065d3062146534132dJesusFreke@JesusFreke.com * Redistribution and use in source and binary forms, with or without 79bc4978139ca38e422ae89065d3062146534132dJesusFreke@JesusFreke.com * modification, are permitted provided that the following conditions 89bc4978139ca38e422ae89065d3062146534132dJesusFreke@JesusFreke.com * are met: 99bc4978139ca38e422ae89065d3062146534132dJesusFreke@JesusFreke.com * 1. Redistributions of source code must retain the above copyright 109bc4978139ca38e422ae89065d3062146534132dJesusFreke@JesusFreke.com * notice, this list of conditions and the following disclaimer. 119bc4978139ca38e422ae89065d3062146534132dJesusFreke@JesusFreke.com * 2. Redistributions in binary form must reproduce the above copyright 129bc4978139ca38e422ae89065d3062146534132dJesusFreke@JesusFreke.com * notice, this list of conditions and the following disclaimer in the 139bc4978139ca38e422ae89065d3062146534132dJesusFreke@JesusFreke.com * documentation and/or other materials provided with the distribution. 149bc4978139ca38e422ae89065d3062146534132dJesusFreke@JesusFreke.com * 3. The name of the author may not be used to endorse or promote products 159bc4978139ca38e422ae89065d3062146534132dJesusFreke@JesusFreke.com * derived from this software without specific prior written permission. 169bc4978139ca38e422ae89065d3062146534132dJesusFreke@JesusFreke.com * 179bc4978139ca38e422ae89065d3062146534132dJesusFreke@JesusFreke.com * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR 189bc4978139ca38e422ae89065d3062146534132dJesusFreke@JesusFreke.com * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 199bc4978139ca38e422ae89065d3062146534132dJesusFreke@JesusFreke.com * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 209bc4978139ca38e422ae89065d3062146534132dJesusFreke@JesusFreke.com * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, 219bc4978139ca38e422ae89065d3062146534132dJesusFreke@JesusFreke.com * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 229bc4978139ca38e422ae89065d3062146534132dJesusFreke@JesusFreke.com * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 239bc4978139ca38e422ae89065d3062146534132dJesusFreke@JesusFreke.com * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 249bc4978139ca38e422ae89065d3062146534132dJesusFreke@JesusFreke.com * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 259bc4978139ca38e422ae89065d3062146534132dJesusFreke@JesusFreke.com * INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 269bc4978139ca38e422ae89065d3062146534132dJesusFreke@JesusFreke.com * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 279bc4978139ca38e422ae89065d3062146534132dJesusFreke@JesusFreke.com */ 289bc4978139ca38e422ae89065d3062146534132dJesusFreke@JesusFreke.com 296ef13753e78bb7abc7e7683d5e533c3395d4a9b6JesusFreke@JesusFreke.compackage org.jf.baksmali.Adaptors.Format; 309bc4978139ca38e422ae89065d3062146534132dJesusFreke@JesusFreke.com 31fbea4e751fa6f1748ded4379a4b64601cb53ba7bJesusFreke@JesusFreke.comimport org.jf.baksmali.Adaptors.LabelMethodItem; 32b2e1e2067d5179b0ec38c76a5191909f5a7dc1ccJesusFreke@JesusFreke.comimport org.jf.baksmali.Adaptors.MethodDefinition; 33754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruverimport org.jf.dexlib2.iface.instruction.SwitchElement; 34754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruverimport org.jf.dexlib2.iface.instruction.formats.SparseSwitchPayload; 354b72225e9d81201838f387171a68a832486903f9JesusFreke@JesusFreke.comimport org.jf.util.IndentingWriter; 366eae34831fee1f116f3a453bdc5e143d68e05e03JesusFreke@JesusFreke.comimport org.jf.baksmali.Renderers.IntegerRenderer; 379bc4978139ca38e422ae89065d3062146534132dJesusFreke@JesusFreke.com 386eae34831fee1f116f3a453bdc5e143d68e05e03JesusFreke@JesusFreke.comimport java.io.IOException; 393c23129eecb7127646f2901c1b0ec3b94a83c08fJesusFreke@JesusFreke.comimport java.util.ArrayList; 407ab77bc90be62b0688c97d4476e3bd219eace0daJesusFreke@JesusFreke.comimport java.util.List; 41cf6729c1ee64e732f28eb40386ecc69d4fb6b401JesusFreke@JesusFreke.com 42754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruverpublic class SparseSwitchMethodItem extends InstructionMethodItem<SparseSwitchPayload> { 43db26b663aa3b5bb721185b8798b6767710d3c243JesusFreke@JesusFreke.com private final List<SparseSwitchTarget> targets; 449bc4978139ca38e422ae89065d3062146534132dJesusFreke@JesusFreke.com 45754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver public SparseSwitchMethodItem(MethodDefinition methodDef, int codeAddress, SparseSwitchPayload instruction) { 46754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver super(methodDef, codeAddress, instruction); 47cf6729c1ee64e732f28eb40386ecc69d4fb6b401JesusFreke@JesusFreke.com 48754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver int baseCodeAddress = methodDef.getSparseSwitchBaseAddress(codeAddress); 49fda2e631ac0b1ca092973b7fff4b2f38d2c23437JesusFreke@JesusFreke.com 50b2e1e2067d5179b0ec38c76a5191909f5a7dc1ccJesusFreke@JesusFreke.com targets = new ArrayList<SparseSwitchTarget>(); 51e00d96c438364844a5c87797119e8770d8ad1f67Ben Gruver if (baseCodeAddress >= 0) { 52754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver for (SwitchElement switchElement: instruction.getSwitchElements()) { 5309058f9914385025020e01125452a884f1b1fe11Ben Gruver LabelMethodItem label = methodDef.getLabelCache().internLabel( 5493aa50139c4641d931b05608f73af8879c0de1c2Ben Gruver new LabelMethodItem( methodDef.classDef.options, baseCodeAddress + switchElement.getOffset(), 5593aa50139c4641d931b05608f73af8879c0de1c2Ben Gruver "sswitch_")); 56754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver targets.add(new SparseSwitchLabelTarget(switchElement.getKey(), label)); 57cb3e0ea38a669633b8daefd538750c8ded46f524Ben Gruver } 58cb3e0ea38a669633b8daefd538750c8ded46f524Ben Gruver } else { 59cb3e0ea38a669633b8daefd538750c8ded46f524Ben Gruver //if we couldn't determine a base address, just use relative offsets rather than labels 60754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver for (SwitchElement switchElement: instruction.getSwitchElements()) { 61754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver targets.add(new SparseSwitchOffsetTarget(switchElement.getKey(), switchElement.getOffset())); 62cb3e0ea38a669633b8daefd538750c8ded46f524Ben Gruver } 637ab77bc90be62b0688c97d4476e3bd219eace0daJesusFreke@JesusFreke.com } 64e2684fa2191e04f27faba763f2bcc19593513b25JesusFreke@JesusFreke.com } 65e2684fa2191e04f27faba763f2bcc19593513b25JesusFreke@JesusFreke.com 666eae34831fee1f116f3a453bdc5e143d68e05e03JesusFreke@JesusFreke.com @Override 670b2f7d6a57e90424b3ee455c041aab3996c05f2cJesusFreke@JesusFreke.com public boolean writeTo(IndentingWriter writer) throws IOException { 68e01409c11f10de58a47df9bc02c6c715b75c6289JesusFreke@JesusFreke.com writer.write(".sparse-switch\n"); 69e01409c11f10de58a47df9bc02c6c715b75c6289JesusFreke@JesusFreke.com writer.indent(4); 70e01409c11f10de58a47df9bc02c6c715b75c6289JesusFreke@JesusFreke.com for (SparseSwitchTarget target: targets) { 71754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver IntegerRenderer.writeTo(writer, target.getKey()); 72e01409c11f10de58a47df9bc02c6c715b75c6289JesusFreke@JesusFreke.com writer.write(" -> "); 73cb3e0ea38a669633b8daefd538750c8ded46f524Ben Gruver target.writeTargetTo(writer); 747e9231a211bf00451229d88edb5c7fbd5085f73eJeff Smith writeResourceId(writer, target.getKey()); 75e01409c11f10de58a47df9bc02c6c715b75c6289JesusFreke@JesusFreke.com writer.write('\n'); 766eae34831fee1f116f3a453bdc5e143d68e05e03JesusFreke@JesusFreke.com } 77e01409c11f10de58a47df9bc02c6c715b75c6289JesusFreke@JesusFreke.com writer.deindent(4); 78e01409c11f10de58a47df9bc02c6c715b75c6289JesusFreke@JesusFreke.com writer.write(".end sparse-switch"); 796eae34831fee1f116f3a453bdc5e143d68e05e03JesusFreke@JesusFreke.com return true; 80e2684fa2191e04f27faba763f2bcc19593513b25JesusFreke@JesusFreke.com } 81e2684fa2191e04f27faba763f2bcc19593513b25JesusFreke@JesusFreke.com 82cb3e0ea38a669633b8daefd538750c8ded46f524Ben Gruver private static abstract class SparseSwitchTarget { 83754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver private final int key; 84754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver public SparseSwitchTarget(int key) { 85754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver this.key = key; 86754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver } 87754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver public int getKey() { return key; } 88cb3e0ea38a669633b8daefd538750c8ded46f524Ben Gruver public abstract void writeTargetTo(IndentingWriter writer) throws IOException; 89e2684fa2191e04f27faba763f2bcc19593513b25JesusFreke@JesusFreke.com } 90e2684fa2191e04f27faba763f2bcc19593513b25JesusFreke@JesusFreke.com 91cb3e0ea38a669633b8daefd538750c8ded46f524Ben Gruver private static class SparseSwitchLabelTarget extends SparseSwitchTarget { 92754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver private final LabelMethodItem target; 93754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver public SparseSwitchLabelTarget(int key, LabelMethodItem target) { 94754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver super(key); 95754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver this.target = target; 96754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver } 97754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver 98cb3e0ea38a669633b8daefd538750c8ded46f524Ben Gruver public void writeTargetTo(IndentingWriter writer) throws IOException { 99754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver target.writeTo(writer); 100cb3e0ea38a669633b8daefd538750c8ded46f524Ben Gruver } 101cb3e0ea38a669633b8daefd538750c8ded46f524Ben Gruver } 102cb3e0ea38a669633b8daefd538750c8ded46f524Ben Gruver 103cb3e0ea38a669633b8daefd538750c8ded46f524Ben Gruver private static class SparseSwitchOffsetTarget extends SparseSwitchTarget { 104754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver private final int target; 105754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver public SparseSwitchOffsetTarget(int key, int target) { 106754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver super(key); 107754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver this.target = target; 108754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver } 109754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver 110cb3e0ea38a669633b8daefd538750c8ded46f524Ben Gruver public void writeTargetTo(IndentingWriter writer) throws IOException { 111754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver if (target >= 0) { 112cb3e0ea38a669633b8daefd538750c8ded46f524Ben Gruver writer.write('+'); 113cb3e0ea38a669633b8daefd538750c8ded46f524Ben Gruver } 114754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver writer.printSignedIntAsDec(target); 115cb3e0ea38a669633b8daefd538750c8ded46f524Ben Gruver } 1169bc4978139ca38e422ae89065d3062146534132dJesusFreke@JesusFreke.com } 1179bc4978139ca38e422ae89065d3062146534132dJesusFreke@JesusFreke.com} 118