DexBackedMethodReference.java revision 12b970ed4dfad768002335503e49c348ea0ed69b 
1/*
2 * Copyright 2012, 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.dexlib2.dexbacked.reference;
33
34import com.google.common.collect.ImmutableList;
35import org.jf.dexlib2.base.reference.BaseMethodReference;
36import org.jf.dexlib2.dexbacked.DexBuffer;
37import org.jf.dexlib2.dexbacked.util.FixedSizeList;
38import org.jf.dexlib2.iface.reference.TypeReference;
39
40import javax.annotation.Nonnull;
41import java.util.List;
42
43public class DexBackedMethodReference extends BaseMethodReference {
44    @Nonnull public final DexBuffer dexBuf;
45    public final int methodIdItemOffset;
46    private int protoIdItemOffset;
47
48    public DexBackedMethodReference(@Nonnull DexBuffer dexBuf, int methodIndex) {
49        this.dexBuf = dexBuf;
50        this.methodIdItemOffset = dexBuf.getMethodIdItemOffset(methodIndex);
51    }
52
53    @Nonnull
54    @Override
55    public String getContainingClass() {
56        return dexBuf.getType(dexBuf.readUshort(methodIdItemOffset + DexBuffer.METHOD_CLASS_IDX_OFFSET));
57    }
58
59    @Nonnull
60    @Override
61    public String getName() {
62        return dexBuf.getString(dexBuf.readSmallUint(methodIdItemOffset + DexBuffer.METHOD_NAME_IDX_OFFSET));
63    }
64
65    @Nonnull
66    @Override
67    public List<? extends TypeReference> getParameters() {
68        int protoIdItemOffset = getProtoIdItemOffset();
69        final int parametersOffset = dexBuf.readSmallUint(protoIdItemOffset + DexBuffer.PROTO_PARAM_LIST_OFF_OFFSET);
70        if (parametersOffset > 0) {
71            final int parameterCount = dexBuf.readSmallUint(parametersOffset + DexBuffer.TYPE_LIST_SIZE_OFFSET);
72            final int paramListStart = parametersOffset + DexBuffer.TYPE_LIST_LIST_OFFSET;
73            return new FixedSizeList<TypeReference>() {
74                @Nonnull
75                @Override
76                public TypeReference readItem(final int index) {
77                    // Can't use DexBackedTypeReference, because we don't want to read in the type index until it
78                    // is asked for
79                    return new TypeReference() {
80                        @Nonnull
81                        @Override
82                        public String getType() {
83                            return dexBuf.getType(dexBuf.readUshort(paramListStart + 2*index));
84                        }
85                    };
86                }
87                @Override public int size() { return parameterCount; }
88            };
89        }
90        return ImmutableList.of();
91    }
92
93    @Nonnull
94    @Override
95    public String getReturnType() {
96        int protoIdItemOffset = getProtoIdItemOffset();
97        return dexBuf.getType(dexBuf.readSmallUint(protoIdItemOffset + DexBuffer.PROTO_RETURN_TYPE_IDX_OFFSET));
98    }
99
100    private int getProtoIdItemOffset() {
101        if (protoIdItemOffset == 0) {
102            protoIdItemOffset = dexBuf.getProtoIdItemOffset(
103                    dexBuf.readUshort(methodIdItemOffset + DexBuffer.METHOD_PROTO_IDX_OFFSET));
104        }
105        return protoIdItemOffset;
106    }
107}
108