11771f92aaf6e2d95c4504174803117712a9800bbBen Gruver/*
21771f92aaf6e2d95c4504174803117712a9800bbBen Gruver * Copyright 2013, Google Inc.
31771f92aaf6e2d95c4504174803117712a9800bbBen Gruver * All rights reserved.
41771f92aaf6e2d95c4504174803117712a9800bbBen Gruver *
51771f92aaf6e2d95c4504174803117712a9800bbBen Gruver * Redistribution and use in source and binary forms, with or without
61771f92aaf6e2d95c4504174803117712a9800bbBen Gruver * modification, are permitted provided that the following conditions are
71771f92aaf6e2d95c4504174803117712a9800bbBen Gruver * met:
81771f92aaf6e2d95c4504174803117712a9800bbBen Gruver *
91771f92aaf6e2d95c4504174803117712a9800bbBen Gruver *     * Redistributions of source code must retain the above copyright
101771f92aaf6e2d95c4504174803117712a9800bbBen Gruver * notice, this list of conditions and the following disclaimer.
111771f92aaf6e2d95c4504174803117712a9800bbBen Gruver *     * Redistributions in binary form must reproduce the above
121771f92aaf6e2d95c4504174803117712a9800bbBen Gruver * copyright notice, this list of conditions and the following disclaimer
131771f92aaf6e2d95c4504174803117712a9800bbBen Gruver * in the documentation and/or other materials provided with the
141771f92aaf6e2d95c4504174803117712a9800bbBen Gruver * distribution.
151771f92aaf6e2d95c4504174803117712a9800bbBen Gruver *     * Neither the name of Google Inc. nor the names of its
161771f92aaf6e2d95c4504174803117712a9800bbBen Gruver * contributors may be used to endorse or promote products derived from
171771f92aaf6e2d95c4504174803117712a9800bbBen Gruver * this software without specific prior written permission.
181771f92aaf6e2d95c4504174803117712a9800bbBen Gruver *
191771f92aaf6e2d95c4504174803117712a9800bbBen Gruver * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
201771f92aaf6e2d95c4504174803117712a9800bbBen Gruver * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
211771f92aaf6e2d95c4504174803117712a9800bbBen Gruver * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
221771f92aaf6e2d95c4504174803117712a9800bbBen Gruver * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
231771f92aaf6e2d95c4504174803117712a9800bbBen Gruver * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
241771f92aaf6e2d95c4504174803117712a9800bbBen Gruver * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
251771f92aaf6e2d95c4504174803117712a9800bbBen Gruver * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
261771f92aaf6e2d95c4504174803117712a9800bbBen Gruver * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
271771f92aaf6e2d95c4504174803117712a9800bbBen Gruver * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
281771f92aaf6e2d95c4504174803117712a9800bbBen Gruver * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
291771f92aaf6e2d95c4504174803117712a9800bbBen Gruver * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
301771f92aaf6e2d95c4504174803117712a9800bbBen Gruver */
311771f92aaf6e2d95c4504174803117712a9800bbBen Gruver
321771f92aaf6e2d95c4504174803117712a9800bbBen Gruverpackage org.jf.dexlib2.dexbacked.raw;
331771f92aaf6e2d95c4504174803117712a9800bbBen Gruver
341771f92aaf6e2d95c4504174803117712a9800bbBen Gruverimport org.jf.dexlib2.dexbacked.DexBackedDexFile;
358f8b67f65ec3390e92cce7d710e5b1eaabd4e248Ben Gruverimport org.jf.dexlib2.dexbacked.raw.util.DexAnnotator;
361771f92aaf6e2d95c4504174803117712a9800bbBen Gruverimport org.jf.dexlib2.util.AnnotatedBytes;
371771f92aaf6e2d95c4504174803117712a9800bbBen Gruver
381771f92aaf6e2d95c4504174803117712a9800bbBen Gruverimport javax.annotation.Nonnull;
398f8b67f65ec3390e92cce7d710e5b1eaabd4e248Ben Gruverimport javax.annotation.Nullable;
401771f92aaf6e2d95c4504174803117712a9800bbBen Gruver
411771f92aaf6e2d95c4504174803117712a9800bbBen Gruverpublic class FieldIdItem {
421771f92aaf6e2d95c4504174803117712a9800bbBen Gruver    public static final int ITEM_SIZE = 8;
431771f92aaf6e2d95c4504174803117712a9800bbBen Gruver
441771f92aaf6e2d95c4504174803117712a9800bbBen Gruver    public static final int CLASS_OFFSET = 0;
451771f92aaf6e2d95c4504174803117712a9800bbBen Gruver    public static final int TYPE_OFFSET = 2;
461771f92aaf6e2d95c4504174803117712a9800bbBen Gruver    public static final int NAME_OFFSET = 4;
471771f92aaf6e2d95c4504174803117712a9800bbBen Gruver
4818b069d17ee8f0fb589c31de0afbbb8240975d14Ben Gruver    @Nonnull
498f8b67f65ec3390e92cce7d710e5b1eaabd4e248Ben Gruver    public static SectionAnnotator makeAnnotator(@Nonnull DexAnnotator annotator, @Nonnull MapItem mapItem) {
508f8b67f65ec3390e92cce7d710e5b1eaabd4e248Ben Gruver        return new SectionAnnotator(annotator, mapItem) {
5178612c755447a47945d7179a688120856db0a76dBen Gruver            @Nonnull @Override public String getItemName() {
5278612c755447a47945d7179a688120856db0a76dBen Gruver                return "field_id_item";
531771f92aaf6e2d95c4504174803117712a9800bbBen Gruver            }
541771f92aaf6e2d95c4504174803117712a9800bbBen Gruver
5578612c755447a47945d7179a688120856db0a76dBen Gruver            @Override
568f8b67f65ec3390e92cce7d710e5b1eaabd4e248Ben Gruver            protected void annotateItem(@Nonnull AnnotatedBytes out, int itemIndex, @Nullable String itemIdentity) {
5778612c755447a47945d7179a688120856db0a76dBen Gruver                int classIndex = dexFile.readUshort(out.getCursor());
5878612c755447a47945d7179a688120856db0a76dBen Gruver                out.annotate(2, "class_idx = %s", TypeIdItem.getReferenceAnnotation(dexFile, classIndex));
591771f92aaf6e2d95c4504174803117712a9800bbBen Gruver
6078612c755447a47945d7179a688120856db0a76dBen Gruver                int typeIndex = dexFile.readUshort(out.getCursor());
6178612c755447a47945d7179a688120856db0a76dBen Gruver                out.annotate(2, "return_type_idx = %s", TypeIdItem.getReferenceAnnotation(dexFile, typeIndex));
621771f92aaf6e2d95c4504174803117712a9800bbBen Gruver
6378612c755447a47945d7179a688120856db0a76dBen Gruver                int nameIndex = dexFile.readSmallUint(out.getCursor());
6478612c755447a47945d7179a688120856db0a76dBen Gruver                out.annotate(4, "name_idx = %s", StringIdItem.getReferenceAnnotation(dexFile, nameIndex));
6578612c755447a47945d7179a688120856db0a76dBen Gruver            }
6678612c755447a47945d7179a688120856db0a76dBen Gruver        };
671771f92aaf6e2d95c4504174803117712a9800bbBen Gruver    }
68d5f47bae4902560746db40e65874db68d95ee742Ben Gruver
69d5f47bae4902560746db40e65874db68d95ee742Ben Gruver    @Nonnull
70d5f47bae4902560746db40e65874db68d95ee742Ben Gruver    public static String asString(@Nonnull DexBackedDexFile dexFile, int fieldIndex) {
71d5f47bae4902560746db40e65874db68d95ee742Ben Gruver        int fieldOffset = dexFile.getFieldIdItemOffset(fieldIndex);
72d5f47bae4902560746db40e65874db68d95ee742Ben Gruver        int classIndex = dexFile.readUshort(fieldOffset + CLASS_OFFSET);
73d5f47bae4902560746db40e65874db68d95ee742Ben Gruver        String classType = dexFile.getType(classIndex);
74d5f47bae4902560746db40e65874db68d95ee742Ben Gruver
75d5f47bae4902560746db40e65874db68d95ee742Ben Gruver        int typeIndex = dexFile.readUshort(fieldOffset + TYPE_OFFSET);
76d5f47bae4902560746db40e65874db68d95ee742Ben Gruver        String fieldType = dexFile.getType(typeIndex);
77d5f47bae4902560746db40e65874db68d95ee742Ben Gruver
78d5f47bae4902560746db40e65874db68d95ee742Ben Gruver        int nameIndex = dexFile.readSmallUint(fieldOffset + NAME_OFFSET);
79d5f47bae4902560746db40e65874db68d95ee742Ben Gruver        String fieldName = dexFile.getString(nameIndex);
80d5f47bae4902560746db40e65874db68d95ee742Ben Gruver
81d5f47bae4902560746db40e65874db68d95ee742Ben Gruver        return String.format("%s->%s:%s", classType, fieldName, fieldType);
82d5f47bae4902560746db40e65874db68d95ee742Ben Gruver    }
83d5f47bae4902560746db40e65874db68d95ee742Ben Gruver
84d5f47bae4902560746db40e65874db68d95ee742Ben Gruver    @Nonnull
85d5f47bae4902560746db40e65874db68d95ee742Ben Gruver    public static String getReferenceAnnotation(@Nonnull DexBackedDexFile dexFile, int fieldIndex) {
86d5f47bae4902560746db40e65874db68d95ee742Ben Gruver        try {
87d5f47bae4902560746db40e65874db68d95ee742Ben Gruver            String fieldString = asString(dexFile, fieldIndex);
88d5f47bae4902560746db40e65874db68d95ee742Ben Gruver            return String.format("field_id_item[%d]: %s", fieldIndex, fieldString);
89d5f47bae4902560746db40e65874db68d95ee742Ben Gruver        } catch (Exception ex) {
90d5f47bae4902560746db40e65874db68d95ee742Ben Gruver            ex.printStackTrace(System.err);
91d5f47bae4902560746db40e65874db68d95ee742Ben Gruver        }
92d5f47bae4902560746db40e65874db68d95ee742Ben Gruver        return String.format("field_id_item[%d]", fieldIndex);
93d5f47bae4902560746db40e65874db68d95ee742Ben Gruver    }
94132eeaedf7d0980397fbdac41e1a7aebff01adbfBen Gruver
95132eeaedf7d0980397fbdac41e1a7aebff01adbfBen Gruver    public static String[] getFields(@Nonnull RawDexFile dexFile) {
96132eeaedf7d0980397fbdac41e1a7aebff01adbfBen Gruver        MapItem mapItem = dexFile.getMapItemForSection(ItemType.FIELD_ID_ITEM);
97132eeaedf7d0980397fbdac41e1a7aebff01adbfBen Gruver        if (mapItem == null) {
98132eeaedf7d0980397fbdac41e1a7aebff01adbfBen Gruver            return new String[0];
99132eeaedf7d0980397fbdac41e1a7aebff01adbfBen Gruver        }
100132eeaedf7d0980397fbdac41e1a7aebff01adbfBen Gruver
101132eeaedf7d0980397fbdac41e1a7aebff01adbfBen Gruver        int fieldCount = mapItem.getItemCount();
102132eeaedf7d0980397fbdac41e1a7aebff01adbfBen Gruver        String[] ret = new String[fieldCount];
103132eeaedf7d0980397fbdac41e1a7aebff01adbfBen Gruver        for (int i=0; i<fieldCount; i++) {
104132eeaedf7d0980397fbdac41e1a7aebff01adbfBen Gruver            ret[i] = asString(dexFile, i);
105132eeaedf7d0980397fbdac41e1a7aebff01adbfBen Gruver        }
106132eeaedf7d0980397fbdac41e1a7aebff01adbfBen Gruver        return ret;
107132eeaedf7d0980397fbdac41e1a7aebff01adbfBen Gruver    }
1081771f92aaf6e2d95c4504174803117712a9800bbBen Gruver}
109