Home | History | Annotate | Download | only in instruction 
      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 
     32 package org.jf.smalidea.dexlib.instruction;
     33 
     34 import com.google.common.base.Function;
     35 import com.google.common.collect.Lists;
     36 import org.jf.dexlib2.Opcode;
     37 import org.jf.dexlib2.iface.instruction.SwitchElement;
     38 import org.jf.dexlib2.iface.instruction.formats.SparseSwitchPayload;
     39 import org.jf.smalidea.psi.impl.*;
     40 import org.jf.smalidea.util.InstructionUtils;
     41 
     42 import javax.annotation.Nonnull;
     43 import java.util.List;
     44 
     45 public 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