Home | History | Annotate | Download | only in reference
      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