1/*
2 * Copyright 2013, 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.raw;
33
34import org.jf.dexlib2.Opcodes;
35import org.jf.dexlib2.dexbacked.BaseDexBuffer;
36import org.jf.dexlib2.dexbacked.DexBackedDexFile;
37import org.jf.dexlib2.dexbacked.util.FixedSizeList;
38import org.jf.dexlib2.util.AnnotatedBytes;
39
40import javax.annotation.Nonnull;
41import javax.annotation.Nullable;
42import java.io.IOException;
43import java.io.Writer;
44import java.util.Arrays;
45import java.util.List;
46
47public class RawDexFile extends DexBackedDexFile {
48    @Nonnull public final HeaderItem headerItem;
49
50    public RawDexFile(@Nonnull Opcodes opcodes, @Nonnull BaseDexBuffer buf) {
51        super(opcodes, buf);
52        this.headerItem = new HeaderItem(this);
53    }
54
55    public RawDexFile(@Nonnull Opcodes opcodes, @Nonnull byte[] buf) {
56        super(opcodes, buf);
57        this.headerItem = new HeaderItem(this);
58    }
59
60    @Nonnull
61    public byte[] readByteRange(int start, int length) {
62        return Arrays.copyOfRange(getBuf(), start, start+length);
63    }
64
65    public int getMapOffset() {
66        return headerItem.getMapOffset();
67    }
68
69    @Nullable
70    public MapItem getMapItemForSection(int itemType) {
71        for (MapItem mapItem: getMapItems()) {
72            if (mapItem.getType() == itemType) {
73                return mapItem;
74            }
75        }
76        return null;
77    }
78
79    public List<MapItem> getMapItems() {
80        final int mapOffset = getMapOffset();
81        final int mapSize = readSmallUint(mapOffset);
82
83        return new FixedSizeList<MapItem>() {
84            @Override
85            public MapItem readItem(int index) {
86                int mapItemOffset = mapOffset + 4 + index * MapItem.ITEM_SIZE;
87                return new MapItem(RawDexFile.this, mapItemOffset);
88            }
89
90            @Override public int size() {
91                return mapSize;
92            }
93        };
94    }
95
96    public void writeAnnotations(@Nonnull Writer out, @Nonnull AnnotatedBytes annotatedBytes) throws IOException {
97        annotatedBytes.writeAnnotations(out, getBuf());
98    }
99}
100