1579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson/* 2579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * Copyright (C) 2007 The Android Open Source Project 3579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * 4579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * Licensed under the Apache License, Version 2.0 (the "License"); 5579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * you may not use this file except in compliance with the License. 6579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * You may obtain a copy of the License at 7579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * 8579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * http://www.apache.org/licenses/LICENSE-2.0 9579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * 10579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * Unless required by applicable law or agreed to in writing, software 11579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * distributed under the License is distributed on an "AS IS" BASIS, 12579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * See the License for the specific language governing permissions and 14579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * limitations under the License. 15579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson */ 16579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 17579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilsonpackage com.android.dx.dex.file; 18579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 19579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilsonimport com.android.dx.rop.cst.Constant; 20579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilsonimport com.android.dx.rop.cst.CstType; 21579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilsonimport com.android.dx.rop.type.Type; 22579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilsonimport com.android.dx.util.AnnotatedOutput; 23579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilsonimport com.android.dx.util.Hex; 24579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 25579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilsonimport java.util.Collection; 26579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilsonimport java.util.TreeMap; 27579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 28579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson/** 29579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * Type identifiers list section of a {@code .dex} file. 30579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson */ 31579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilsonpublic final class TypeIdsSection extends UniformItemSection { 32579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson /** 33579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * {@code non-null;} map from types to {@link TypeIdItem} instances 34579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson */ 35579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson private final TreeMap<Type, TypeIdItem> typeIds; 36579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 37579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson /** 38579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * Constructs an instance. The file offset is initially unknown. 39579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * 40579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * @param file {@code non-null;} file that this instance is part of 41579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson */ 42579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson public TypeIdsSection(DexFile file) { 43579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson super("type_ids", file, 4); 44579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 45579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson typeIds = new TreeMap<Type, TypeIdItem>(); 46579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 47579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 48579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson /** {@inheritDoc} */ 49579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson @Override 50579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson public Collection<? extends Item> items() { 51579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson return typeIds.values(); 52579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 53579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 54579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson /** {@inheritDoc} */ 55579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson @Override 56579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson public IndexedItem get(Constant cst) { 57579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson if (cst == null) { 58579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson throw new NullPointerException("cst == null"); 59579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 60579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 61579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson throwIfNotPrepared(); 62579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 63579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson Type type = ((CstType) cst).getClassType(); 64579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson IndexedItem result = typeIds.get(type); 65579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 66579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson if (result == null) { 67579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson throw new IllegalArgumentException("not found: " + cst); 68579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 69579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 70579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson return result; 71579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 72579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 73579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson /** 74579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * Writes the portion of the file header that refers to this instance. 75579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * 76579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * @param out {@code non-null;} where to write 77579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson */ 78579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson public void writeHeaderPart(AnnotatedOutput out) { 79579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson throwIfNotPrepared(); 80579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 81579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson int sz = typeIds.size(); 82579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson int offset = (sz == 0) ? 0 : getFileOffset(); 83579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 84579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson if (sz > 65536) { 85579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson throw new UnsupportedOperationException("too many type ids"); 86579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 87579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 88579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson if (out.annotates()) { 89579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson out.annotate(4, "type_ids_size: " + Hex.u4(sz)); 90579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson out.annotate(4, "type_ids_off: " + Hex.u4(offset)); 91579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 92579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 93579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson out.writeInt(sz); 94579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson out.writeInt(offset); 95579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 96579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 97579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson /** 98579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * Interns an element into this instance. 99579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * 100579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * @param type {@code non-null;} the type to intern 101579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * @return {@code non-null;} the interned reference 102579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson */ 103579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson public TypeIdItem intern(Type type) { 104579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson if (type == null) { 105579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson throw new NullPointerException("type == null"); 106579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 107579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 108579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson throwIfPrepared(); 109579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 110579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson TypeIdItem result = typeIds.get(type); 111579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 112579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson if (result == null) { 113579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson result = new TypeIdItem(new CstType(type)); 114579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson typeIds.put(type, result); 115579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 116579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 117579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson return result; 118579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 119579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 120579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson /** 121579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * Interns an element into this instance. 122579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * 123579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * @param type {@code non-null;} the type to intern 124579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * @return {@code non-null;} the interned reference 125579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson */ 126579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson public TypeIdItem intern(CstType type) { 127579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson if (type == null) { 128579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson throw new NullPointerException("type == null"); 129579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 130579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 131579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson throwIfPrepared(); 132579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 133579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson Type typePerSe = type.getClassType(); 134579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson TypeIdItem result = typeIds.get(typePerSe); 135579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 136579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson if (result == null) { 137579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson result = new TypeIdItem(type); 138579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson typeIds.put(typePerSe, result); 139579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 140579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 141579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson return result; 142579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 143579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 144579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson /** 145579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * Gets the index of the given type, which must have 146579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * been added to this instance. 147579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * 148579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * @param type {@code non-null;} the type to look up 149579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * @return {@code >= 0;} the reference's index 150579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson */ 151579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson public int indexOf(Type type) { 152579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson if (type == null) { 153579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson throw new NullPointerException("type == null"); 154579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 155579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 156579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson throwIfNotPrepared(); 157579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 158579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson TypeIdItem item = typeIds.get(type); 159579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 160579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson if (item == null) { 161579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson throw new IllegalArgumentException("not found: " + type); 162579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 163579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 164579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson return item.getIndex(); 165579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 166579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 167579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson /** 168579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * Gets the index of the given type, which must have 169579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * been added to this instance. 170579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * 171579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * @param type {@code non-null;} the type to look up 172579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson * @return {@code >= 0;} the reference's index 173579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson */ 174579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson public int indexOf(CstType type) { 175579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson if (type == null) { 176579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson throw new NullPointerException("type == null"); 177579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 178579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 179579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson return indexOf(type.getClassType()); 180579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 181579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 182579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson /** {@inheritDoc} */ 183579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson @Override 184579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson protected void orderItems() { 185579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson int idx = 0; 186579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson 187579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson for (Object i : items()) { 188579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson ((TypeIdItem) i).setIndex(idx); 189579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson idx++; 190579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 191579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson } 192579d7739c53a2707ad711a2d2cae46d7d782f06Jesse Wilson} 193