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 31b742c402631ca85ffe72e556ce97f4533cb6083eBen Gruverimport org.jf.baksmali.Adaptors.CommentingIndentingWriter; 32fbea4e751fa6f1748ded4379a4b64601cb53ba7bJesusFreke@JesusFreke.comimport org.jf.baksmali.Adaptors.LabelMethodItem; 33b2e1e2067d5179b0ec38c76a5191909f5a7dc1ccJesusFreke@JesusFreke.comimport org.jf.baksmali.Adaptors.MethodDefinition; 34754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruverimport org.jf.dexlib2.iface.instruction.SwitchElement; 35754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruverimport org.jf.dexlib2.iface.instruction.formats.SparseSwitchPayload; 364b72225e9d81201838f387171a68a832486903f9JesusFreke@JesusFreke.comimport org.jf.util.IndentingWriter; 376eae34831fee1f116f3a453bdc5e143d68e05e03JesusFreke@JesusFreke.comimport org.jf.baksmali.Renderers.IntegerRenderer; 389bc4978139ca38e422ae89065d3062146534132dJesusFreke@JesusFreke.com 396eae34831fee1f116f3a453bdc5e143d68e05e03JesusFreke@JesusFreke.comimport java.io.IOException; 403c23129eecb7127646f2901c1b0ec3b94a83c08fJesusFreke@JesusFreke.comimport java.util.ArrayList; 417ab77bc90be62b0688c97d4476e3bd219eace0daJesusFreke@JesusFreke.comimport java.util.List; 42cf6729c1ee64e732f28eb40386ecc69d4fb6b401JesusFreke@JesusFreke.com 43754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruverpublic class SparseSwitchMethodItem extends InstructionMethodItem<SparseSwitchPayload> { 44db26b663aa3b5bb721185b8798b6767710d3c243JesusFreke@JesusFreke.com private final List<SparseSwitchTarget> targets; 459bc4978139ca38e422ae89065d3062146534132dJesusFreke@JesusFreke.com 46b742c402631ca85ffe72e556ce97f4533cb6083eBen Gruver // Whether this sparse switch instruction should be commented out because it is never referenced 47b742c402631ca85ffe72e556ce97f4533cb6083eBen Gruver private boolean commentedOut; 48b742c402631ca85ffe72e556ce97f4533cb6083eBen Gruver 49754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver public SparseSwitchMethodItem(MethodDefinition methodDef, int codeAddress, SparseSwitchPayload instruction) { 50754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver super(methodDef, codeAddress, instruction); 51cf6729c1ee64e732f28eb40386ecc69d4fb6b401JesusFreke@JesusFreke.com 52754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver int baseCodeAddress = methodDef.getSparseSwitchBaseAddress(codeAddress); 53fda2e631ac0b1ca092973b7fff4b2f38d2c23437JesusFreke@JesusFreke.com 54b2e1e2067d5179b0ec38c76a5191909f5a7dc1ccJesusFreke@JesusFreke.com targets = new ArrayList<SparseSwitchTarget>(); 55e00d96c438364844a5c87797119e8770d8ad1f67Ben Gruver if (baseCodeAddress >= 0) { 56754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver for (SwitchElement switchElement: instruction.getSwitchElements()) { 5709058f9914385025020e01125452a884f1b1fe11Ben Gruver LabelMethodItem label = methodDef.getLabelCache().internLabel( 5893aa50139c4641d931b05608f73af8879c0de1c2Ben Gruver new LabelMethodItem( methodDef.classDef.options, baseCodeAddress + switchElement.getOffset(), 5993aa50139c4641d931b05608f73af8879c0de1c2Ben Gruver "sswitch_")); 60754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver targets.add(new SparseSwitchLabelTarget(switchElement.getKey(), label)); 61cb3e0ea38a669633b8daefd538750c8ded46f524Ben Gruver } 62cb3e0ea38a669633b8daefd538750c8ded46f524Ben Gruver } else { 63b742c402631ca85ffe72e556ce97f4533cb6083eBen Gruver commentedOut = true; 64cb3e0ea38a669633b8daefd538750c8ded46f524Ben Gruver //if we couldn't determine a base address, just use relative offsets rather than labels 65754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver for (SwitchElement switchElement: instruction.getSwitchElements()) { 66754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver targets.add(new SparseSwitchOffsetTarget(switchElement.getKey(), switchElement.getOffset())); 67cb3e0ea38a669633b8daefd538750c8ded46f524Ben Gruver } 687ab77bc90be62b0688c97d4476e3bd219eace0daJesusFreke@JesusFreke.com } 69e2684fa2191e04f27faba763f2bcc19593513b25JesusFreke@JesusFreke.com } 70e2684fa2191e04f27faba763f2bcc19593513b25JesusFreke@JesusFreke.com 716eae34831fee1f116f3a453bdc5e143d68e05e03JesusFreke@JesusFreke.com @Override 720b2f7d6a57e90424b3ee455c041aab3996c05f2cJesusFreke@JesusFreke.com public boolean writeTo(IndentingWriter writer) throws IOException { 73b742c402631ca85ffe72e556ce97f4533cb6083eBen Gruver if (commentedOut) { 74b742c402631ca85ffe72e556ce97f4533cb6083eBen Gruver writer = new CommentingIndentingWriter(writer); 75b742c402631ca85ffe72e556ce97f4533cb6083eBen Gruver } 76b742c402631ca85ffe72e556ce97f4533cb6083eBen Gruver 77e01409c11f10de58a47df9bc02c6c715b75c6289JesusFreke@JesusFreke.com writer.write(".sparse-switch\n"); 78e01409c11f10de58a47df9bc02c6c715b75c6289JesusFreke@JesusFreke.com writer.indent(4); 79e01409c11f10de58a47df9bc02c6c715b75c6289JesusFreke@JesusFreke.com for (SparseSwitchTarget target: targets) { 80754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver IntegerRenderer.writeTo(writer, target.getKey()); 81e01409c11f10de58a47df9bc02c6c715b75c6289JesusFreke@JesusFreke.com writer.write(" -> "); 82cb3e0ea38a669633b8daefd538750c8ded46f524Ben Gruver target.writeTargetTo(writer); 830b836342e21b4de21d1d452d5b43b54a364a35c6Jeff Smith writeCommentIfResourceId(writer, target.getKey()); 84e01409c11f10de58a47df9bc02c6c715b75c6289JesusFreke@JesusFreke.com writer.write('\n'); 856eae34831fee1f116f3a453bdc5e143d68e05e03JesusFreke@JesusFreke.com } 86e01409c11f10de58a47df9bc02c6c715b75c6289JesusFreke@JesusFreke.com writer.deindent(4); 87e01409c11f10de58a47df9bc02c6c715b75c6289JesusFreke@JesusFreke.com writer.write(".end sparse-switch"); 886eae34831fee1f116f3a453bdc5e143d68e05e03JesusFreke@JesusFreke.com return true; 89e2684fa2191e04f27faba763f2bcc19593513b25JesusFreke@JesusFreke.com } 90e2684fa2191e04f27faba763f2bcc19593513b25JesusFreke@JesusFreke.com 91cb3e0ea38a669633b8daefd538750c8ded46f524Ben Gruver private static abstract class SparseSwitchTarget { 92754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver private final int key; 93754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver public SparseSwitchTarget(int key) { 94754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver this.key = key; 95754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver } 96754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver public int getKey() { return key; } 97cb3e0ea38a669633b8daefd538750c8ded46f524Ben Gruver public abstract void writeTargetTo(IndentingWriter writer) throws IOException; 98e2684fa2191e04f27faba763f2bcc19593513b25JesusFreke@JesusFreke.com } 99e2684fa2191e04f27faba763f2bcc19593513b25JesusFreke@JesusFreke.com 100cb3e0ea38a669633b8daefd538750c8ded46f524Ben Gruver private static class SparseSwitchLabelTarget extends SparseSwitchTarget { 101754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver private final LabelMethodItem target; 102754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver public SparseSwitchLabelTarget(int key, LabelMethodItem target) { 103754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver super(key); 104754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver this.target = target; 105754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver } 106754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver 107cb3e0ea38a669633b8daefd538750c8ded46f524Ben Gruver public void writeTargetTo(IndentingWriter writer) throws IOException { 108754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver target.writeTo(writer); 109cb3e0ea38a669633b8daefd538750c8ded46f524Ben Gruver } 110cb3e0ea38a669633b8daefd538750c8ded46f524Ben Gruver } 111cb3e0ea38a669633b8daefd538750c8ded46f524Ben Gruver 112cb3e0ea38a669633b8daefd538750c8ded46f524Ben Gruver private static class SparseSwitchOffsetTarget extends SparseSwitchTarget { 113754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver private final int target; 114754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver public SparseSwitchOffsetTarget(int key, int target) { 115754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver super(key); 116754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver this.target = target; 117754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver } 118754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver 119cb3e0ea38a669633b8daefd538750c8ded46f524Ben Gruver public void writeTargetTo(IndentingWriter writer) throws IOException { 120754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver if (target >= 0) { 121cb3e0ea38a669633b8daefd538750c8ded46f524Ben Gruver writer.write('+'); 122cb3e0ea38a669633b8daefd538750c8ded46f524Ben Gruver } 123754b3c4dc009b7a02e39001560c3f0fd6a7cc2c0Ben Gruver writer.printSignedIntAsDec(target); 124cb3e0ea38a669633b8daefd538750c8ded46f524Ben Gruver } 1259bc4978139ca38e422ae89065d3062146534132dJesusFreke@JesusFreke.com } 1269bc4978139ca38e422ae89065d3062146534132dJesusFreke@JesusFreke.com} 127