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.CstFieldRef; 21917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgulimport com.android.dexgen.util.AnnotatedOutput; 22917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgulimport com.android.dexgen.util.Hex; 23917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 24917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgulimport java.util.Collection; 25917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgulimport java.util.TreeMap; 26917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 27917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul/** 28917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * Field refs list section of a {@code .dex} file. 29917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul */ 30917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgulpublic final class FieldIdsSection extends MemberIdsSection { 31917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul /** 32917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * {@code non-null;} map from field constants to {@link 33917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * FieldIdItem} instances 34917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul */ 35917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul private final TreeMap<CstFieldRef, FieldIdItem> fieldIds; 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 FieldIdsSection(DexFile file) { 43917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul super("field_ids", file); 44917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 45917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul fieldIds = new TreeMap<CstFieldRef, FieldIdItem>(); 46917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul } 47917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 48917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul /** {@inheritDoc} */ 49917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul @Override 50917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul public Collection<? extends Item> items() { 51917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul return fieldIds.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 IndexedItem result = fieldIds.get((CstFieldRef) cst); 64917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 65917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul if (result == null) { 66917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul throw new IllegalArgumentException("not found"); 67917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul } 68917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 69917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul return result; 70917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul } 71917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 72917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul /** 73917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * Writes the portion of the file header that refers to this instance. 74917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * 75917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * @param out {@code non-null;} where to write 76917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul */ 77917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul public void writeHeaderPart(AnnotatedOutput out) { 78917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul throwIfNotPrepared(); 79917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 80917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul int sz = fieldIds.size(); 81917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul int offset = (sz == 0) ? 0 : getFileOffset(); 82917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 83917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul if (out.annotates()) { 84917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul out.annotate(4, "field_ids_size: " + Hex.u4(sz)); 85917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul out.annotate(4, "field_ids_off: " + Hex.u4(offset)); 86917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul } 87917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 88917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul out.writeInt(sz); 89917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul out.writeInt(offset); 90917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul } 91917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 92917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul /** 93917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * Interns an element into this instance. 94917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * 95917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * @param field {@code non-null;} the reference to intern 96917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * @return {@code non-null;} the interned reference 97917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul */ 98917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul public FieldIdItem intern(CstFieldRef field) { 99917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul if (field == null) { 100917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul throw new NullPointerException("field == null"); 101917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul } 102917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 103917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul throwIfPrepared(); 104917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 105917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul FieldIdItem result = fieldIds.get(field); 106917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 107917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul if (result == null) { 108917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul result = new FieldIdItem(field); 109917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul fieldIds.put(field, result); 110917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul } 111917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 112917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul return result; 113917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul } 114917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 115917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul /** 116917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * Gets the index of the given reference, which must have been added 117917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * to this instance. 118917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * 119917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * @param ref {@code non-null;} the reference to look up 120917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul * @return {@code >= 0;} the reference's index 121917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul */ 122917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul public int indexOf(CstFieldRef ref) { 123917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul if (ref == null) { 124917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul throw new NullPointerException("ref == null"); 125917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul } 126917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 127917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul throwIfNotPrepared(); 128917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 129917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul FieldIdItem item = fieldIds.get(ref); 130917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 131917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul if (item == null) { 132917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul throw new IllegalArgumentException("not found"); 133917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul } 134917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul 135917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul return item.getIndex(); 136917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul } 137917cb222329ee8c035c3ffaf947e4265761b9367Piotr Gurgul} 138