1 /* 2 * Copyright 2016, 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.dexbacked.reference; 33 34 import com.google.common.collect.ImmutableList; 35 import org.jf.dexlib2.base.reference.BaseMethodProtoReference; 36 import org.jf.dexlib2.dexbacked.DexBackedDexFile; 37 import org.jf.dexlib2.dexbacked.raw.ProtoIdItem; 38 import org.jf.dexlib2.dexbacked.raw.TypeListItem; 39 import org.jf.dexlib2.dexbacked.util.FixedSizeList; 40 41 import java.util.List; 42 import javax.annotation.Nonnull; 43 44 public class DexBackedMethodProtoReference extends BaseMethodProtoReference { 45 @Nonnull public final DexBackedDexFile dexFile; 46 private final int protoIdItemOffset; 47 48 public DexBackedMethodProtoReference(@Nonnull DexBackedDexFile dexFile, int protoIndex) { 49 this.dexFile = dexFile; 50 this.protoIdItemOffset = dexFile.getProtoIdItemOffset(protoIndex); 51 } 52 53 @Nonnull 54 @Override 55 public List<String> getParameterTypes() { 56 final int parametersOffset = dexFile.readSmallUint(protoIdItemOffset + ProtoIdItem.PARAMETERS_OFFSET); 57 if (parametersOffset > 0) { 58 final int parameterCount = dexFile.readSmallUint(parametersOffset + TypeListItem.SIZE_OFFSET); 59 final int paramListStart = parametersOffset + TypeListItem.LIST_OFFSET; 60 return new FixedSizeList<String>() { 61 @Nonnull 62 @Override 63 public String readItem(final int index) { 64 return dexFile.getType(dexFile.readUshort(paramListStart + 2*index)); 65 } 66 @Override public int size() { return parameterCount; } 67 }; 68 } 69 return ImmutableList.of(); 70 } 71 72 @Nonnull 73 @Override 74 public String getReturnType() { 75 return dexFile.getType(dexFile.readSmallUint(protoIdItemOffset + ProtoIdItem.RETURN_TYPE_OFFSET)); 76 } 77 78 /** 79 * Calculate and return the private size of a method proto. 80 * 81 * Calculated as: shorty_idx + return_type_idx + parameters_off + type_list size 82 * 83 * @return size in bytes 84 */ 85 public int getSize() { 86 int size = ProtoIdItem.ITEM_SIZE; //3 * uint 87 List<String> parameters = getParameterTypes(); 88 if (!parameters.isEmpty()) { 89 size += 4 + parameters.size() * 2; //uint + size * ushort for type_idxs 90 } 91 return size; 92 } 93 } 94