1917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul/* 2917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * Copyright (C) 2007 The Android Open Source Project 3917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * 4917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * Licensed under the Apache License, Version 2.0 (the "License"); 5917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * you may not use this file except in compliance with the License. 6917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * You may obtain a copy of the License at 7917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * 8917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * http://www.apache.org/licenses/LICENSE-2.0 9917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * 10917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * Unless required by applicable law or agreed to in writing, software 11917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * distributed under the License is distributed on an "AS IS" BASIS, 12917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * See the License for the specific language governing permissions and 14917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * limitations under the License. 15917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul */ 16917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 17917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgulpackage com.android.dexgen.dex.file; 18917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 19917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgulimport com.android.dexgen.rop.cst.Constant; 20917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgulimport com.android.dexgen.rop.cst.CstType; 21917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgulimport com.android.dexgen.rop.type.Type; 22917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgulimport com.android.dexgen.util.AnnotatedOutput; 23917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgulimport com.android.dexgen.util.Hex; 24917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 25917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgulimport java.util.Collection; 26917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgulimport java.util.TreeMap; 27917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 28917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul/** 29917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * Type identifiers list section of a {@code .dex} file. 30917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul */ 31917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgulpublic final class TypeIdsSection extends UniformItemSection { 32917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul /** 33917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * {@code non-null;} map from types to {@link TypeIdItem} instances 34917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul */ 35917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul private final TreeMap<Type, TypeIdItem> typeIds; 36917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 37917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul /** 38917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * Constructs an instance. The file offset is initially unknown. 39917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * 40917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * @param file {@code non-null;} file that this instance is part of 41917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul */ 42917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul public TypeIdsSection(DexFile file) { 43917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul super("type_ids", file, 4); 44917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 45917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul typeIds = new TreeMap<Type, TypeIdItem>(); 46917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul } 47917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 48917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul /** {@inheritDoc} */ 49917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul @Override 50917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul public Collection<? extends Item> items() { 51917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul return typeIds.values(); 52917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul } 53917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 54917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul /** {@inheritDoc} */ 55917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul @Override 56917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul public IndexedItem get(Constant cst) { 57917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul if (cst == null) { 58917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul throw new NullPointerException("cst == null"); 59917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul } 60917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 61917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul throwIfNotPrepared(); 62917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 63917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul Type type = ((CstType) cst).getClassType(); 64917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul IndexedItem result = typeIds.get(type); 65917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 66917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul if (result == null) { 67917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul throw new IllegalArgumentException("not found: " + cst); 68917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul } 69917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 70917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul return result; 71917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul } 72917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 73917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul /** 74917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * Writes the portion of the file header that refers to this instance. 75917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * 76917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * @param out {@code non-null;} where to write 77917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul */ 78917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul public void writeHeaderPart(AnnotatedOutput out) { 79917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul throwIfNotPrepared(); 80917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 81917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul int sz = typeIds.size(); 82917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul int offset = (sz == 0) ? 0 : getFileOffset(); 83917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 84917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul if (sz > 65536) { 85917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul throw new UnsupportedOperationException("too many type ids"); 86917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul } 87917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 88917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul if (out.annotates()) { 89917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul out.annotate(4, "type_ids_size: " + Hex.u4(sz)); 90917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul out.annotate(4, "type_ids_off: " + Hex.u4(offset)); 91917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul } 92917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 93917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul out.writeInt(sz); 94917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul out.writeInt(offset); 95917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul } 96917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 97917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul /** 98917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * Interns an element into this instance. 99917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * 100917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * @param type {@code non-null;} the type to intern 101917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * @return {@code non-null;} the interned reference 102917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul */ 103917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul public TypeIdItem intern(Type type) { 104917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul if (type == null) { 105917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul throw new NullPointerException("type == null"); 106917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul } 107917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 108917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul throwIfPrepared(); 109917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 110917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul TypeIdItem result = typeIds.get(type); 111917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 112917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul if (result == null) { 113917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul result = new TypeIdItem(new CstType(type)); 114917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul typeIds.put(type, result); 115917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul } 116917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 117917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul return result; 118917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul } 119917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 120917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul /** 121917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * Interns an element into this instance. 122917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * 123917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * @param type {@code non-null;} the type to intern 124917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * @return {@code non-null;} the interned reference 125917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul */ 126917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul public TypeIdItem intern(CstType type) { 127917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul if (type == null) { 128917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul throw new NullPointerException("type == null"); 129917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul } 130917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 131917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul throwIfPrepared(); 132917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 133917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul Type typePerSe = type.getClassType(); 134917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul TypeIdItem result = typeIds.get(typePerSe); 135917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 136917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul if (result == null) { 137917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul result = new TypeIdItem(type); 138917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul typeIds.put(typePerSe, result); 139917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul } 140917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 141917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul return result; 142917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul } 143917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 144917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul /** 145917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * Gets the index of the given type, which must have 146917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * been added to this instance. 147917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * 148917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * @param type {@code non-null;} the type to look up 149917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * @return {@code >= 0;} the reference's index 150917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul */ 151917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul public int indexOf(Type type) { 152917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul if (type == null) { 153917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul throw new NullPointerException("type == null"); 154917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul } 155917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 156917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul throwIfNotPrepared(); 157917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 158917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul TypeIdItem item = typeIds.get(type); 159917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 160917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul if (item == null) { 161917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul throw new IllegalArgumentException("not found: " + type); 162917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul } 163917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 164917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul return item.getIndex(); 165917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul } 166917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 167917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul /** 168917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * Gets the index of the given type, which must have 169917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * been added to this instance. 170917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * 171917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * @param type {@code non-null;} the type to look up 172917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * @return {@code >= 0;} the reference's index 173917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul */ 174917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul public int indexOf(CstType type) { 175917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul if (type == null) { 176917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul throw new NullPointerException("type == null"); 177917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul } 178917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 179917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul return indexOf(type.getClassType()); 180917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul } 181917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 182917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul /** {@inheritDoc} */ 183917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul @Override 184917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul protected void orderItems() { 185917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul int idx = 0; 186917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 187917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul for (Object i : items()) { 188917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul ((TypeIdItem) i).setIndex(idx); 189917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul idx++; 190917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul } 191917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul } 192917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul} 193