1/* 2 * Copyright 2014, 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 32package org.jf.smalidea.dexlib.instruction; 33 34import com.google.common.base.Function; 35import com.google.common.collect.Lists; 36import org.jf.dexlib2.Opcode; 37import org.jf.dexlib2.iface.instruction.SwitchElement; 38import org.jf.dexlib2.iface.instruction.formats.SparseSwitchPayload; 39import org.jf.smalidea.psi.impl.*; 40import org.jf.smalidea.util.InstructionUtils; 41 42import javax.annotation.Nonnull; 43import java.util.List; 44 45public class SmalideaSparseSwitchPayload extends SmalideaInstruction implements SparseSwitchPayload { 46 public SmalideaSparseSwitchPayload(@Nonnull SmaliInstruction instruction) { 47 super(instruction); 48 } 49 50 @Nonnull @Override public List<? extends SwitchElement> getSwitchElements() { 51 List<SmaliSparseSwitchElement> elements = psiInstruction.getSparseSwitchElements(); 52 53 SmaliMethod smaliMethod = psiInstruction.getParentMethod(); 54 SmaliInstruction sparseSwitchInstruction = InstructionUtils.findFirstInstructionWithTarget( 55 smaliMethod, Opcode.SPARSE_SWITCH, psiInstruction.getOffset()); 56 final int baseOffset; 57 58 if (sparseSwitchInstruction == null) { 59 baseOffset = 0; 60 } else { 61 baseOffset = sparseSwitchInstruction.getOffset(); 62 } 63 64 return Lists.transform(elements, new Function<SmaliSparseSwitchElement, SwitchElement>() { 65 @Override public SwitchElement apply(final SmaliSparseSwitchElement element) { 66 return new SwitchElement() { 67 @Override public int getKey() { 68 return (int)element.getKey().getIntegralValue(); 69 } 70 71 @Override public int getOffset() { 72 SmaliLabelReference labelReference = element.getTarget(); 73 if (labelReference == null) { 74 return 0; 75 } 76 77 SmaliLabel label = labelReference.resolve(); 78 if (label == null) { 79 return 0; 80 } 81 82 return label.getOffset() - baseOffset; 83 } 84 }; 85 } 86 }); 87 } 88 89 @Override public int getCodeUnits() { 90 return psiInstruction.getInstructionSize()/2; 91 } 92} 93