blob: fe7457231a405a7d2b0aa1f03f2ae6ddc4bf1774 [file] [log] [blame]
/*
* Copyright (C) 2016 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
* Header file of an in-memory representation of DEX files.
*/
#ifndef ART_DEXLAYOUT_DEX_IR_H_
#define ART_DEXLAYOUT_DEX_IR_H_
#include <map>
#include <vector>
#include <stdint.h>
#include "base/stl_util.h"
#include "dex_file-inl.h"
#include "leb128.h"
#include "utf.h"
namespace art {
namespace dex_ir {
// Forward declarations for classes used in containers or pointed to.
class AnnotationItem;
class AnnotationsDirectoryItem;
class AnnotationSetItem;
class AnnotationSetRefList;
class CallSiteId;
class ClassData;
class ClassDef;
class CodeItem;
class DebugInfoItem;
class EncodedAnnotation;
class EncodedArrayItem;
class EncodedValue;
class FieldId;
class FieldItem;
class Header;
class MapList;
class MapItem;
class MethodHandleItem;
class MethodId;
class MethodItem;
class ParameterAnnotation;
class ProtoId;
class StringData;
class StringId;
class TryItem;
class TypeId;
class TypeList;
// Item size constants.
static constexpr size_t kHeaderItemSize = 112;
static constexpr size_t kStringIdItemSize = 4;
static constexpr size_t kTypeIdItemSize = 4;
static constexpr size_t kProtoIdItemSize = 12;
static constexpr size_t kFieldIdItemSize = 8;
static constexpr size_t kMethodIdItemSize = 8;
static constexpr size_t kClassDefItemSize = 32;
static constexpr size_t kCallSiteIdItemSize = 4;
static constexpr size_t kMethodHandleItemSize = 8;
// Visitor support
class AbstractDispatcher {
public:
AbstractDispatcher() = default;
virtual ~AbstractDispatcher() { }
virtual void Dispatch(Header* header) = 0;
virtual void Dispatch(const StringData* string_data) = 0;
virtual void Dispatch(const StringId* string_id) = 0;
virtual void Dispatch(const TypeId* type_id) = 0;
virtual void Dispatch(const ProtoId* proto_id) = 0;
virtual void Dispatch(const FieldId* field_id) = 0;
virtual void Dispatch(const MethodId* method_id) = 0;
virtual void Dispatch(const CallSiteId* call_site_id) = 0;
virtual void Dispatch(const MethodHandleItem* method_handle_item) = 0;
virtual void Dispatch(ClassData* class_data) = 0;
virtual void Dispatch(ClassDef* class_def) = 0;
virtual void Dispatch(FieldItem* field_item) = 0;
virtual void Dispatch(MethodItem* method_item) = 0;
virtual void Dispatch(EncodedArrayItem* array_item) = 0;
virtual void Dispatch(CodeItem* code_item) = 0;
virtual void Dispatch(TryItem* try_item) = 0;
virtual void Dispatch(DebugInfoItem* debug_info_item) = 0;
virtual void Dispatch(AnnotationItem* annotation_item) = 0;
virtual void Dispatch(AnnotationSetItem* annotation_set_item) = 0;
virtual void Dispatch(AnnotationSetRefList* annotation_set_ref_list) = 0;
virtual void Dispatch(AnnotationsDirectoryItem* annotations_directory_item) = 0;
virtual void Dispatch(MapList* map_list) = 0;
virtual void Dispatch(MapItem* map_item) = 0;
private:
DISALLOW_COPY_AND_ASSIGN(AbstractDispatcher);
};
// Collections become owners of the objects added by moving them into unique pointers.
template<class T> class CollectionBase {
public:
CollectionBase() = default;
uint32_t GetOffset() const { return offset_; }
void SetOffset(uint32_t new_offset) { offset_ = new_offset; }
private:
uint32_t offset_ = 0;
DISALLOW_COPY_AND_ASSIGN(CollectionBase);
};
template<class T> class CollectionVector : public CollectionBase<T> {
public:
CollectionVector() = default;
void AddIndexedItem(T* object, uint32_t offset, uint32_t index) {
object->SetOffset(offset);
object->SetIndex(index);
collection_.push_back(std::unique_ptr<T>(object));
}
uint32_t Size() const { return collection_.size(); }
std::vector<std::unique_ptr<T>>& Collection() { return collection_; }
private:
std::vector<std::unique_ptr<T>> collection_;
DISALLOW_COPY_AND_ASSIGN(CollectionVector);
};
template<class T> class CollectionMap : public CollectionBase<T> {
public:
CollectionMap() = default;
// Returns the existing item if it is already inserted, null otherwise.
T* GetExistingObject(uint32_t offset) {
auto it = collection_.find(offset);
return it != collection_.end() ? it->second.get() : nullptr;
}
void AddItem(T* object, uint32_t offset) {
object->SetOffset(offset);
auto it = collection_.emplace(offset, std::unique_ptr<T>(object));
CHECK(it.second) << "CollectionMap already has an object with offset " << offset << " "
<< " and address " << it.first->second.get();
}
uint32_t Size() const { return collection_.size(); }
std::map<uint32_t, std::unique_ptr<T>>& Collection() { return collection_; }
private:
std::map<uint32_t, std::unique_ptr<T>> collection_;
DISALLOW_COPY_AND_ASSIGN(CollectionMap);
};
class Collections {
public:
Collections() = default;
std::vector<std::unique_ptr<StringId>>& StringIds() { return string_ids_.Collection(); }
std::vector<std::unique_ptr<TypeId>>& TypeIds() { return type_ids_.Collection(); }
std::vector<std::unique_ptr<ProtoId>>& ProtoIds() { return proto_ids_.Collection(); }
std::vector<std::unique_ptr<FieldId>>& FieldIds() { return field_ids_.Collection(); }
std::vector<std::unique_ptr<MethodId>>& MethodIds() { return method_ids_.Collection(); }
std::vector<std::unique_ptr<ClassDef>>& ClassDefs() { return class_defs_.Collection(); }
std::vector<std::unique_ptr<CallSiteId>>& CallSiteIds() { return call_site_ids_.Collection(); }
std::vector<std::unique_ptr<MethodHandleItem>>& MethodHandleItems()
{ return method_handle_items_.Collection(); }
std::map<uint32_t, std::unique_ptr<StringData>>& StringDatas()
{ return string_datas_.Collection(); }
std::map<uint32_t, std::unique_ptr<TypeList>>& TypeLists() { return type_lists_.Collection(); }
std::map<uint32_t, std::unique_ptr<EncodedArrayItem>>& EncodedArrayItems()
{ return encoded_array_items_.Collection(); }
std::map<uint32_t, std::unique_ptr<AnnotationItem>>& AnnotationItems()
{ return annotation_items_.Collection(); }
std::map<uint32_t, std::unique_ptr<AnnotationSetItem>>& AnnotationSetItems()
{ return annotation_set_items_.Collection(); }
std::map<uint32_t, std::unique_ptr<AnnotationSetRefList>>& AnnotationSetRefLists()
{ return annotation_set_ref_lists_.Collection(); }
std::map<uint32_t, std::unique_ptr<AnnotationsDirectoryItem>>& AnnotationsDirectoryItems()
{ return annotations_directory_items_.Collection(); }
std::map<uint32_t, std::unique_ptr<DebugInfoItem>>& DebugInfoItems()
{ return debug_info_items_.Collection(); }
std::map<uint32_t, std::unique_ptr<CodeItem>>& CodeItems() { return code_items_.Collection(); }
std::map<uint32_t, std::unique_ptr<ClassData>>& ClassDatas() { return class_datas_.Collection(); }
void CreateStringId(const DexFile& dex_file, uint32_t i);
void CreateTypeId(const DexFile& dex_file, uint32_t i);
void CreateProtoId(const DexFile& dex_file, uint32_t i);
void CreateFieldId(const DexFile& dex_file, uint32_t i);
void CreateMethodId(const DexFile& dex_file, uint32_t i);
void CreateClassDef(const DexFile& dex_file, uint32_t i);
void CreateCallSiteId(const DexFile& dex_file, uint32_t i);
void CreateMethodHandleItem(const DexFile& dex_file, uint32_t i);
void CreateCallSitesAndMethodHandles(const DexFile& dex_file);
TypeList* CreateTypeList(const DexFile::TypeList* type_list, uint32_t offset);
EncodedArrayItem* CreateEncodedArrayItem(const uint8_t* static_data, uint32_t offset);
AnnotationItem* CreateAnnotationItem(const DexFile::AnnotationItem* annotation, uint32_t offset);
AnnotationSetItem* CreateAnnotationSetItem(const DexFile& dex_file,
const DexFile::AnnotationSetItem* disk_annotations_item, uint32_t offset);
AnnotationsDirectoryItem* CreateAnnotationsDirectoryItem(const DexFile& dex_file,
const DexFile::AnnotationsDirectoryItem* disk_annotations_item, uint32_t offset);
CodeItem* CreateCodeItem(
const DexFile& dex_file, const DexFile::CodeItem& disk_code_item, uint32_t offset);
ClassData* CreateClassData(const DexFile& dex_file, const uint8_t* encoded_data, uint32_t offset);
StringId* GetStringId(uint32_t index) {
CHECK_LT(index, StringIdsSize());
return StringIds()[index].get();
}
TypeId* GetTypeId(uint32_t index) {
CHECK_LT(index, TypeIdsSize());
return TypeIds()[index].get();
}
ProtoId* GetProtoId(uint32_t index) {
CHECK_LT(index, ProtoIdsSize());
return ProtoIds()[index].get();
}
FieldId* GetFieldId(uint32_t index) {
CHECK_LT(index, FieldIdsSize());
return FieldIds()[index].get();
}
MethodId* GetMethodId(uint32_t index) {
CHECK_LT(index, MethodIdsSize());
return MethodIds()[index].get();
}
ClassDef* GetClassDef(uint32_t index) {
CHECK_LT(index, ClassDefsSize());
return ClassDefs()[index].get();
}
CallSiteId* GetCallSiteId(uint32_t index) {
CHECK_LT(index, CallSiteIdsSize());
return CallSiteIds()[index].get();
}
MethodHandleItem* GetMethodHandle(uint32_t index) {
CHECK_LT(index, MethodHandleItemsSize());
return MethodHandleItems()[index].get();
}
StringId* GetStringIdOrNullPtr(uint32_t index) {
return index == DexFile::kDexNoIndex ? nullptr : GetStringId(index);
}
TypeId* GetTypeIdOrNullPtr(uint16_t index) {
return index == DexFile::kDexNoIndex16 ? nullptr : GetTypeId(index);
}
uint32_t StringIdsOffset() const { return string_ids_.GetOffset(); }
uint32_t TypeIdsOffset() const { return type_ids_.GetOffset(); }
uint32_t ProtoIdsOffset() const { return proto_ids_.GetOffset(); }
uint32_t FieldIdsOffset() const { return field_ids_.GetOffset(); }
uint32_t MethodIdsOffset() const { return method_ids_.GetOffset(); }
uint32_t ClassDefsOffset() const { return class_defs_.GetOffset(); }
uint32_t CallSiteIdsOffset() const { return call_site_ids_.GetOffset(); }
uint32_t MethodHandleItemsOffset() const { return method_handle_items_.GetOffset(); }
uint32_t StringDatasOffset() const { return string_datas_.GetOffset(); }
uint32_t TypeListsOffset() const { return type_lists_.GetOffset(); }
uint32_t EncodedArrayItemsOffset() const { return encoded_array_items_.GetOffset(); }
uint32_t AnnotationItemsOffset() const { return annotation_items_.GetOffset(); }
uint32_t AnnotationSetItemsOffset() const { return annotation_set_items_.GetOffset(); }
uint32_t AnnotationSetRefListsOffset() const { return annotation_set_ref_lists_.GetOffset(); }
uint32_t AnnotationsDirectoryItemsOffset() const
{ return annotations_directory_items_.GetOffset(); }
uint32_t DebugInfoItemsOffset() const { return debug_info_items_.GetOffset(); }
uint32_t CodeItemsOffset() const { return code_items_.GetOffset(); }
uint32_t ClassDatasOffset() const { return class_datas_.GetOffset(); }
uint32_t MapListOffset() const { return map_list_offset_; }
void SetStringIdsOffset(uint32_t new_offset) { string_ids_.SetOffset(new_offset); }
void SetTypeIdsOffset(uint32_t new_offset) { type_ids_.SetOffset(new_offset); }
void SetProtoIdsOffset(uint32_t new_offset) { proto_ids_.SetOffset(new_offset); }
void SetFieldIdsOffset(uint32_t new_offset) { field_ids_.SetOffset(new_offset); }
void SetMethodIdsOffset(uint32_t new_offset) { method_ids_.SetOffset(new_offset); }
void SetClassDefsOffset(uint32_t new_offset) { class_defs_.SetOffset(new_offset); }
void SetCallSiteIdsOffset(uint32_t new_offset) { call_site_ids_.SetOffset(new_offset); }
void SetMethodHandleItemsOffset(uint32_t new_offset)
{ method_handle_items_.SetOffset(new_offset); }
void SetStringDatasOffset(uint32_t new_offset) { string_datas_.SetOffset(new_offset); }
void SetTypeListsOffset(uint32_t new_offset) { type_lists_.SetOffset(new_offset); }
void SetEncodedArrayItemsOffset(uint32_t new_offset)
{ encoded_array_items_.SetOffset(new_offset); }
void SetAnnotationItemsOffset(uint32_t new_offset) { annotation_items_.SetOffset(new_offset); }
void SetAnnotationSetItemsOffset(uint32_t new_offset)
{ annotation_set_items_.SetOffset(new_offset); }
void SetAnnotationSetRefListsOffset(uint32_t new_offset)
{ annotation_set_ref_lists_.SetOffset(new_offset); }
void SetAnnotationsDirectoryItemsOffset(uint32_t new_offset)
{ annotations_directory_items_.SetOffset(new_offset); }
void SetDebugInfoItemsOffset(uint32_t new_offset) { debug_info_items_.SetOffset(new_offset); }
void SetCodeItemsOffset(uint32_t new_offset) { code_items_.SetOffset(new_offset); }
void SetClassDatasOffset(uint32_t new_offset) { class_datas_.SetOffset(new_offset); }
void SetMapListOffset(uint32_t new_offset) { map_list_offset_ = new_offset; }
uint32_t StringIdsSize() const { return string_ids_.Size(); }
uint32_t TypeIdsSize() const { return type_ids_.Size(); }
uint32_t ProtoIdsSize() const { return proto_ids_.Size(); }
uint32_t FieldIdsSize() const { return field_ids_.Size(); }
uint32_t MethodIdsSize() const { return method_ids_.Size(); }
uint32_t ClassDefsSize() const { return class_defs_.Size(); }
uint32_t CallSiteIdsSize() const { return call_site_ids_.Size(); }
uint32_t MethodHandleItemsSize() const { return method_handle_items_.Size(); }
uint32_t StringDatasSize() const { return string_datas_.Size(); }
uint32_t TypeListsSize() const { return type_lists_.Size(); }
uint32_t EncodedArrayItemsSize() const { return encoded_array_items_.Size(); }
uint32_t AnnotationItemsSize() const { return annotation_items_.Size(); }
uint32_t AnnotationSetItemsSize() const { return annotation_set_items_.Size(); }
uint32_t AnnotationSetRefListsSize() const { return annotation_set_ref_lists_.Size(); }
uint32_t AnnotationsDirectoryItemsSize() const { return annotations_directory_items_.Size(); }
uint32_t DebugInfoItemsSize() const { return debug_info_items_.Size(); }
uint32_t CodeItemsSize() const { return code_items_.Size(); }
uint32_t ClassDatasSize() const { return class_datas_.Size(); }
private:
EncodedValue* ReadEncodedValue(const uint8_t** data);
EncodedValue* ReadEncodedValue(const uint8_t** data, uint8_t type, uint8_t length);
void ReadEncodedValue(const uint8_t** data, uint8_t type, uint8_t length, EncodedValue* item);
ParameterAnnotation* GenerateParameterAnnotation(const DexFile& dex_file, MethodId* method_id,
const DexFile::AnnotationSetRefList* annotation_set_ref_list, uint32_t offset);
MethodItem* GenerateMethodItem(const DexFile& dex_file, ClassDataItemIterator& cdii);
CollectionVector<StringId> string_ids_;
CollectionVector<TypeId> type_ids_;
CollectionVector<ProtoId> proto_ids_;
CollectionVector<FieldId> field_ids_;
CollectionVector<MethodId> method_ids_;
CollectionVector<ClassDef> class_defs_;
CollectionVector<CallSiteId> call_site_ids_;
CollectionVector<MethodHandleItem> method_handle_items_;
CollectionMap<StringData> string_datas_;
CollectionMap<TypeList> type_lists_;
CollectionMap<EncodedArrayItem> encoded_array_items_;
CollectionMap<AnnotationItem> annotation_items_;
CollectionMap<AnnotationSetItem> annotation_set_items_;
CollectionMap<AnnotationSetRefList> annotation_set_ref_lists_;
CollectionMap<AnnotationsDirectoryItem> annotations_directory_items_;
CollectionMap<DebugInfoItem> debug_info_items_;
CollectionMap<CodeItem> code_items_;
CollectionMap<ClassData> class_datas_;
uint32_t map_list_offset_ = 0;
DISALLOW_COPY_AND_ASSIGN(Collections);
};
class Item {
public:
Item() { }
virtual ~Item() { }
uint32_t GetOffset() const { return offset_; }
uint32_t GetSize() const { return size_; }
void SetOffset(uint32_t offset) { offset_ = offset; }
void SetSize(uint32_t size) { size_ = size; }
protected:
Item(uint32_t offset, uint32_t size) : offset_(offset), size_(size) { }
uint32_t offset_ = 0;
uint32_t size_ = 0;
};
class IndexedItem : public Item {
public:
IndexedItem() { }
virtual ~IndexedItem() { }
uint32_t GetIndex() const { return index_; }
void SetIndex(uint32_t index) { index_ = index; }
protected:
IndexedItem(uint32_t offset, uint32_t size, uint32_t index)
: Item(offset, size), index_(index) { }
uint32_t index_ = 0;
};
class Header : public Item {
public:
Header(const uint8_t* magic,
uint32_t checksum,
const uint8_t* signature,
uint32_t endian_tag,
uint32_t file_size,
uint32_t header_size,
uint32_t link_size,
uint32_t link_offset,
uint32_t data_size,
uint32_t data_offset)
: Item(0, kHeaderItemSize),
checksum_(checksum),
endian_tag_(endian_tag),
file_size_(file_size),
header_size_(header_size),
link_size_(link_size),
link_offset_(link_offset),
data_size_(data_size),
data_offset_(data_offset) {
memcpy(magic_, magic, sizeof(magic_));
memcpy(signature_, signature, sizeof(signature_));
}
~Header() OVERRIDE { }
static size_t ItemSize() { return kHeaderItemSize; }
const uint8_t* Magic() const { return magic_; }
uint32_t Checksum() const { return checksum_; }
const uint8_t* Signature() const { return signature_; }
uint32_t EndianTag() const { return endian_tag_; }
uint32_t FileSize() const { return file_size_; }
uint32_t HeaderSize() const { return header_size_; }
uint32_t LinkSize() const { return link_size_; }
uint32_t LinkOffset() const { return link_offset_; }
uint32_t DataSize() const { return data_size_; }
uint32_t DataOffset() const { return data_offset_; }
void SetChecksum(uint32_t new_checksum) { checksum_ = new_checksum; }
void SetSignature(const uint8_t* new_signature) {
memcpy(signature_, new_signature, sizeof(signature_));
}
void SetFileSize(uint32_t new_file_size) { file_size_ = new_file_size; }
void SetHeaderSize(uint32_t new_header_size) { header_size_ = new_header_size; }
void SetLinkSize(uint32_t new_link_size) { link_size_ = new_link_size; }
void SetLinkOffset(uint32_t new_link_offset) { link_offset_ = new_link_offset; }
void SetDataSize(uint32_t new_data_size) { data_size_ = new_data_size; }
void SetDataOffset(uint32_t new_data_offset) { data_offset_ = new_data_offset; }
Collections& GetCollections() { return collections_; }
void Accept(AbstractDispatcher* dispatch) { dispatch->Dispatch(this); }
private:
uint8_t magic_[8];
uint32_t checksum_;
uint8_t signature_[DexFile::kSha1DigestSize];
uint32_t endian_tag_;
uint32_t file_size_;
uint32_t header_size_;
uint32_t link_size_;
uint32_t link_offset_;
uint32_t data_size_;
uint32_t data_offset_;
Collections collections_;
DISALLOW_COPY_AND_ASSIGN(Header);
};
class StringData : public Item {
public:
explicit StringData(const char* data) : data_(strdup(data)) {
size_ = UnsignedLeb128Size(CountModifiedUtf8Chars(data)) + strlen(data);
}
const char* Data() const { return data_.get(); }
void Accept(AbstractDispatcher* dispatch) const { dispatch->Dispatch(this); }
private:
UniqueCPtr<const char> data_;
DISALLOW_COPY_AND_ASSIGN(StringData);
};
class StringId : public IndexedItem {
public:
explicit StringId(StringData* string_data) : string_data_(string_data) {
size_ = kStringIdItemSize;
}
~StringId() OVERRIDE { }
static size_t ItemSize() { return kStringIdItemSize; }
const char* Data() const { return string_data_->Data(); }
StringData* DataItem() const { return string_data_; }
void Accept(AbstractDispatcher* dispatch) const { dispatch->Dispatch(this); }
private:
StringData* string_data_;
DISALLOW_COPY_AND_ASSIGN(StringId);
};
class TypeId : public IndexedItem {
public:
explicit TypeId(StringId* string_id) : string_id_(string_id) { size_ = kTypeIdItemSize; }
~TypeId() OVERRIDE { }
static size_t ItemSize() { return kTypeIdItemSize; }
StringId* GetStringId() const { return string_id_; }
void Accept(AbstractDispatcher* dispatch) const { dispatch->Dispatch(this); }
private:
StringId* string_id_;
DISALLOW_COPY_AND_ASSIGN(TypeId);
};
using TypeIdVector = std::vector<const TypeId*>;
class TypeList : public Item {
public:
explicit TypeList(TypeIdVector* type_list) : type_list_(type_list) {
size_ = sizeof(uint32_t) + (type_list->size() * sizeof(uint16_t));
}
~TypeList() OVERRIDE { }
const TypeIdVector* GetTypeList() const { return type_list_.get(); }
private:
std::unique_ptr<TypeIdVector> type_list_;
DISALLOW_COPY_AND_ASSIGN(TypeList);
};
class ProtoId : public IndexedItem {
public:
ProtoId(const StringId* shorty, const TypeId* return_type, TypeList* parameters)
: shorty_(shorty), return_type_(return_type), parameters_(parameters)
{ size_ = kProtoIdItemSize; }
~ProtoId() OVERRIDE { }
static size_t ItemSize() { return kProtoIdItemSize; }
const StringId* Shorty() const { return shorty_; }
const TypeId* ReturnType() const { return return_type_; }
const TypeList* Parameters() const { return parameters_; }
void Accept(AbstractDispatcher* dispatch) const { dispatch->Dispatch(this); }
private:
const StringId* shorty_;
const TypeId* return_type_;
TypeList* parameters_; // This can be nullptr.
DISALLOW_COPY_AND_ASSIGN(ProtoId);
};
class FieldId : public IndexedItem {
public:
FieldId(const TypeId* klass, const TypeId* type, const StringId* name)
: class_(klass), type_(type), name_(name) { size_ = kFieldIdItemSize; }
~FieldId() OVERRIDE { }
static size_t ItemSize() { return kFieldIdItemSize; }
const TypeId* Class() const { return class_; }
const TypeId* Type() const { return type_; }
const StringId* Name() const { return name_; }
void Accept(AbstractDispatcher* dispatch) const { dispatch->Dispatch(this); }
private:
const TypeId* class_;
const TypeId* type_;
const StringId* name_;
DISALLOW_COPY_AND_ASSIGN(FieldId);
};
class MethodId : public IndexedItem {
public:
MethodId(const TypeId* klass, const ProtoId* proto, const StringId* name)
: class_(klass), proto_(proto), name_(name) { size_ = kMethodIdItemSize; }
~MethodId() OVERRIDE { }
static size_t ItemSize() { return kMethodIdItemSize; }
const TypeId* Class() const { return class_; }
const ProtoId* Proto() const { return proto_; }
const StringId* Name() const { return name_; }
void Accept(AbstractDispatcher* dispatch) const { dispatch->Dispatch(this); }
private:
const TypeId* class_;
const ProtoId* proto_;
const StringId* name_;
DISALLOW_COPY_AND_ASSIGN(MethodId);
};
class FieldItem : public Item {
public:
FieldItem(uint32_t access_flags, const FieldId* field_id)
: access_flags_(access_flags), field_id_(field_id) { }
~FieldItem() OVERRIDE { }
uint32_t GetAccessFlags() const { return access_flags_; }
const FieldId* GetFieldId() const { return field_id_; }
void Accept(AbstractDispatcher* dispatch) { dispatch->Dispatch(this); }
private:
uint32_t access_flags_;
const FieldId* field_id_;
DISALLOW_COPY_AND_ASSIGN(FieldItem);
};
using FieldItemVector = std::vector<std::unique_ptr<FieldItem>>;
class MethodItem : public Item {
public:
MethodItem(uint32_t access_flags, const MethodId* method_id, CodeItem* code)
: access_flags_(access_flags), method_id_(method_id), code_(code) { }
~MethodItem() OVERRIDE { }
uint32_t GetAccessFlags() const { return access_flags_; }
const MethodId* GetMethodId() const { return method_id_; }
CodeItem* GetCodeItem() { return code_; }
void Accept(AbstractDispatcher* dispatch) { dispatch->Dispatch(this); }
private:
uint32_t access_flags_;
const MethodId* method_id_;
CodeItem* code_; // This can be nullptr.
DISALLOW_COPY_AND_ASSIGN(MethodItem);
};
using MethodItemVector = std::vector<std::unique_ptr<MethodItem>>;
class EncodedValue {
public:
explicit EncodedValue(uint8_t type) : type_(type) { }
int8_t Type() const { return type_; }
void SetBoolean(bool z) { u_.bool_val_ = z; }
void SetByte(int8_t b) { u_.byte_val_ = b; }
void SetShort(int16_t s) { u_.short_val_ = s; }
void SetChar(uint16_t c) { u_.char_val_ = c; }
void SetInt(int32_t i) { u_.int_val_ = i; }
void SetLong(int64_t l) { u_.long_val_ = l; }
void SetFloat(float f) { u_.float_val_ = f; }
void SetDouble(double d) { u_.double_val_ = d; }
void SetStringId(StringId* string_id) { u_.string_val_ = string_id; }
void SetTypeId(TypeId* type_id) { u_.type_val_ = type_id; }
void SetProtoId(ProtoId* proto_id) { u_.proto_val_ = proto_id; }
void SetFieldId(FieldId* field_id) { u_.field_val_ = field_id; }
void SetMethodId(MethodId* method_id) { u_.method_val_ = method_id; }
void SetMethodHandle(MethodHandleItem* method_handle) { u_.method_handle_val_ = method_handle; }
void SetEncodedArray(EncodedArrayItem* encoded_array) { encoded_array_.reset(encoded_array); }
void SetEncodedAnnotation(EncodedAnnotation* encoded_annotation)
{ encoded_annotation_.reset(encoded_annotation); }
bool GetBoolean() const { return u_.bool_val_; }
int8_t GetByte() const { return u_.byte_val_; }
int16_t GetShort() const { return u_.short_val_; }
uint16_t GetChar() const { return u_.char_val_; }
int32_t GetInt() const { return u_.int_val_; }
int64_t GetLong() const { return u_.long_val_; }
float GetFloat() const { return u_.float_val_; }
double GetDouble() const { return u_.double_val_; }
StringId* GetStringId() const { return u_.string_val_; }
TypeId* GetTypeId() const { return u_.type_val_; }
ProtoId* GetProtoId() const { return u_.proto_val_; }
FieldId* GetFieldId() const { return u_.field_val_; }
MethodId* GetMethodId() const { return u_.method_val_; }
MethodHandleItem* GetMethodHandle() const { return u_.method_handle_val_; }
EncodedArrayItem* GetEncodedArray() const { return encoded_array_.get(); }
EncodedAnnotation* GetEncodedAnnotation() const { return encoded_annotation_.get(); }
EncodedAnnotation* ReleaseEncodedAnnotation() { return encoded_annotation_.release(); }
private:
uint8_t type_;
union {
bool bool_val_;
int8_t byte_val_;
int16_t short_val_;
uint16_t char_val_;
int32_t int_val_;
int64_t long_val_;
float float_val_;
double double_val_;
StringId* string_val_;
TypeId* type_val_;
ProtoId* proto_val_;
FieldId* field_val_;
MethodId* method_val_;
MethodHandleItem* method_handle_val_;
} u_;
std::unique_ptr<EncodedArrayItem> encoded_array_;
std::unique_ptr<EncodedAnnotation> encoded_annotation_;
DISALLOW_COPY_AND_ASSIGN(EncodedValue);
};
using EncodedValueVector = std::vector<std::unique_ptr<EncodedValue>>;
class AnnotationElement {
public:
AnnotationElement(StringId* name, EncodedValue* value) : name_(name), value_(value) { }
StringId* GetName() const { return name_; }
EncodedValue* GetValue() const { return value_.get(); }
private:
StringId* name_;
std::unique_ptr<EncodedValue> value_;
DISALLOW_COPY_AND_ASSIGN(AnnotationElement);
};
using AnnotationElementVector = std::vector<std::unique_ptr<AnnotationElement>>;
class EncodedAnnotation {
public:
EncodedAnnotation(TypeId* type, AnnotationElementVector* elements)
: type_(type), elements_(elements) { }
TypeId* GetType() const { return type_; }
AnnotationElementVector* GetAnnotationElements() const { return elements_.get(); }
private:
TypeId* type_;
std::unique_ptr<AnnotationElementVector> elements_;
DISALLOW_COPY_AND_ASSIGN(EncodedAnnotation);
};
class EncodedArrayItem : public Item {
public:
explicit EncodedArrayItem(EncodedValueVector* encoded_values)
: encoded_values_(encoded_values) { }
EncodedValueVector* GetEncodedValues() const { return encoded_values_.get(); }
private:
std::unique_ptr<EncodedValueVector> encoded_values_;
DISALLOW_COPY_AND_ASSIGN(EncodedArrayItem);
};
class ClassData : public Item {
public:
ClassData(FieldItemVector* static_fields,
FieldItemVector* instance_fields,
MethodItemVector* direct_methods,
MethodItemVector* virtual_methods)
: static_fields_(static_fields),
instance_fields_(instance_fields),
direct_methods_(direct_methods),
virtual_methods_(virtual_methods) { }
~ClassData() OVERRIDE = default;
FieldItemVector* StaticFields() { return static_fields_.get(); }
FieldItemVector* InstanceFields() { return instance_fields_.get(); }
MethodItemVector* DirectMethods() { return direct_methods_.get(); }
MethodItemVector* VirtualMethods() { return virtual_methods_.get(); }
void Accept(AbstractDispatcher* dispatch) { dispatch->Dispatch(this); }
private:
std::unique_ptr<FieldItemVector> static_fields_;
std::unique_ptr<FieldItemVector> instance_fields_;
std::unique_ptr<MethodItemVector> direct_methods_;
std::unique_ptr<MethodItemVector> virtual_methods_;
DISALLOW_COPY_AND_ASSIGN(ClassData);
};
class ClassDef : public IndexedItem {
public:
ClassDef(const TypeId* class_type,
uint32_t access_flags,
const TypeId* superclass,
TypeList* interfaces,
const StringId* source_file,
AnnotationsDirectoryItem* annotations,
EncodedArrayItem* static_values,
ClassData* class_data)
: class_type_(class_type),
access_flags_(access_flags),
superclass_(superclass),
interfaces_(interfaces),
source_file_(source_file),
annotations_(annotations),
class_data_(class_data),
static_values_(static_values) { size_ = kClassDefItemSize; }
~ClassDef() OVERRIDE { }
static size_t ItemSize() { return kClassDefItemSize; }
const TypeId* ClassType() const { return class_type_; }
uint32_t GetAccessFlags() const { return access_flags_; }
const TypeId* Superclass() const { return superclass_; }
const TypeList* Interfaces() { return interfaces_; }
uint32_t InterfacesOffset() { return interfaces_ == nullptr ? 0 : interfaces_->GetOffset(); }
const StringId* SourceFile() const { return source_file_; }
AnnotationsDirectoryItem* Annotations() const { return annotations_; }
ClassData* GetClassData() { return class_data_; }
EncodedArrayItem* StaticValues() { return static_values_; }
MethodItem* GenerateMethodItem(Header& header, ClassDataItemIterator& cdii);
void Accept(AbstractDispatcher* dispatch) { dispatch->Dispatch(this); }
private:
const TypeId* class_type_;
uint32_t access_flags_;
const TypeId* superclass_; // This can be nullptr.
TypeList* interfaces_; // This can be nullptr.
const StringId* source_file_; // This can be nullptr.
AnnotationsDirectoryItem* annotations_; // This can be nullptr.
ClassData* class_data_; // This can be nullptr.
EncodedArrayItem* static_values_; // This can be nullptr.
DISALLOW_COPY_AND_ASSIGN(ClassDef);
};
class TypeAddrPair {
public:
TypeAddrPair(const TypeId* type_id, uint32_t address) : type_id_(type_id), address_(address) { }
const TypeId* GetTypeId() const { return type_id_; }
uint32_t GetAddress() const { return address_; }
private:
const TypeId* type_id_; // This can be nullptr.
uint32_t address_;
DISALLOW_COPY_AND_ASSIGN(TypeAddrPair);
};
using TypeAddrPairVector = std::vector<std::unique_ptr<const TypeAddrPair>>;
class CatchHandler {
public:
explicit CatchHandler(bool catch_all, uint16_t list_offset, TypeAddrPairVector* handlers)
: catch_all_(catch_all), list_offset_(list_offset), handlers_(handlers) { }
bool HasCatchAll() const { return catch_all_; }
uint16_t GetListOffset() const { return list_offset_; }
TypeAddrPairVector* GetHandlers() const { return handlers_.get(); }
private:
bool catch_all_;
uint16_t list_offset_;
std::unique_ptr<TypeAddrPairVector> handlers_;
DISALLOW_COPY_AND_ASSIGN(CatchHandler);
};
using CatchHandlerVector = std::vector<std::unique_ptr<const CatchHandler>>;
class TryItem : public Item {
public:
TryItem(uint32_t start_addr, uint16_t insn_count, const CatchHandler* handlers)
: start_addr_(start_addr), insn_count_(insn_count), handlers_(handlers) { }
~TryItem() OVERRIDE { }
uint32_t StartAddr() const { return start_addr_; }
uint16_t InsnCount() const { return insn_count_; }
const CatchHandler* GetHandlers() const { return handlers_; }
void Accept(AbstractDispatcher* dispatch) { dispatch->Dispatch(this); }
private:
uint32_t start_addr_;
uint16_t insn_count_;
const CatchHandler* handlers_;
DISALLOW_COPY_AND_ASSIGN(TryItem);
};
using TryItemVector = std::vector<std::unique_ptr<const TryItem>>;
class CodeFixups {
public:
CodeFixups(std::vector<TypeId*>* type_ids,
std::vector<StringId*>* string_ids,
std::vector<MethodId*>* method_ids,
std::vector<FieldId*>* field_ids)
: type_ids_(type_ids),
string_ids_(string_ids),
method_ids_(method_ids),
field_ids_(field_ids) { }
std::vector<TypeId*>* TypeIds() const { return type_ids_.get(); }
std::vector<StringId*>* StringIds() const { return string_ids_.get(); }
std::vector<MethodId*>* MethodIds() const { return method_ids_.get(); }
std::vector<FieldId*>* FieldIds() const { return field_ids_.get(); }
private:
std::unique_ptr<std::vector<TypeId*>> type_ids_;
std::unique_ptr<std::vector<StringId*>> string_ids_;
std::unique_ptr<std::vector<MethodId*>> method_ids_;
std::unique_ptr<std::vector<FieldId*>> field_ids_;
DISALLOW_COPY_AND_ASSIGN(CodeFixups);
};
class CodeItem : public Item {
public:
CodeItem(uint16_t registers_size,
uint16_t ins_size,
uint16_t outs_size,
DebugInfoItem* debug_info,
uint32_t insns_size,
uint16_t* insns,
TryItemVector* tries,
CatchHandlerVector* handlers)
: registers_size_(registers_size),
ins_size_(ins_size),
outs_size_(outs_size),
debug_info_(debug_info),
insns_size_(insns_size),
insns_(insns),
tries_(tries),
handlers_(handlers) { }
~CodeItem() OVERRIDE { }
uint16_t RegistersSize() const { return registers_size_; }
uint16_t InsSize() const { return ins_size_; }
uint16_t OutsSize() const { return outs_size_; }
uint16_t TriesSize() const { return tries_ == nullptr ? 0 : tries_->size(); }
DebugInfoItem* DebugInfo() const { return debug_info_; }
uint32_t InsnsSize() const { return insns_size_; }
uint16_t* Insns() const { return insns_.get(); }
TryItemVector* Tries() const { return tries_.get(); }
CatchHandlerVector* Handlers() const { return handlers_.get(); }
void SetCodeFixups(CodeFixups* fixups) { fixups_.reset(fixups); }
CodeFixups* GetCodeFixups() const { return fixups_.get(); }
void Accept(AbstractDispatcher* dispatch) { dispatch->Dispatch(this); }
private:
uint16_t registers_size_;
uint16_t ins_size_;
uint16_t outs_size_;
DebugInfoItem* debug_info_; // This can be nullptr.
uint32_t insns_size_;
std::unique_ptr<uint16_t[]> insns_;
std::unique_ptr<TryItemVector> tries_; // This can be nullptr.
std::unique_ptr<CatchHandlerVector> handlers_; // This can be nullptr.
std::unique_ptr<CodeFixups> fixups_; // This can be nullptr.
DISALLOW_COPY_AND_ASSIGN(CodeItem);
};
struct PositionInfo {
PositionInfo(uint32_t address, uint32_t line) : address_(address), line_(line) { }
uint32_t address_;
uint32_t line_;
};
using PositionInfoVector = std::vector<std::unique_ptr<PositionInfo>>;
struct LocalInfo {
LocalInfo(const char* name,
const char* descriptor,
const char* signature,
uint32_t start_address,
uint32_t end_address,
uint16_t reg)
: name_(name),
descriptor_(descriptor),
signature_(signature),
start_address_(start_address),
end_address_(end_address),
reg_(reg) { }
std::string name_;
std::string descriptor_;
std::string signature_;
uint32_t start_address_;
uint32_t end_address_;
uint16_t reg_;
};
using LocalInfoVector = std::vector<std::unique_ptr<LocalInfo>>;
class DebugInfoItem : public Item {
public:
DebugInfoItem(uint32_t debug_info_size, uint8_t* debug_info)
: debug_info_size_(debug_info_size), debug_info_(debug_info) { }
uint32_t GetDebugInfoSize() const { return debug_info_size_; }
uint8_t* GetDebugInfo() const { return debug_info_.get(); }
PositionInfoVector& GetPositionInfo() { return positions_; }
LocalInfoVector& GetLocalInfo() { return locals_; }
private:
uint32_t debug_info_size_;
std::unique_ptr<uint8_t[]> debug_info_;
PositionInfoVector positions_;
LocalInfoVector locals_;
DISALLOW_COPY_AND_ASSIGN(DebugInfoItem);
};
class AnnotationItem : public Item {
public:
AnnotationItem(uint8_t visibility, EncodedAnnotation* annotation)
: visibility_(visibility), annotation_(annotation) { }
uint8_t GetVisibility() const { return visibility_; }
EncodedAnnotation* GetAnnotation() const { return annotation_.get(); }
void Accept(AbstractDispatcher* dispatch) { dispatch->Dispatch(this); }
private:
uint8_t visibility_;
std::unique_ptr<EncodedAnnotation> annotation_;
DISALLOW_COPY_AND_ASSIGN(AnnotationItem);
};
class AnnotationSetItem : public Item {
public:
explicit AnnotationSetItem(std::vector<AnnotationItem*>* items) : items_(items) {
size_ = sizeof(uint32_t) + items->size() * sizeof(uint32_t);
}
~AnnotationSetItem() OVERRIDE { }
std::vector<AnnotationItem*>* GetItems() { return items_.get(); }
void Accept(AbstractDispatcher* dispatch) { dispatch->Dispatch(this); }
private:
std::unique_ptr<std::vector<AnnotationItem*>> items_;
DISALLOW_COPY_AND_ASSIGN(AnnotationSetItem);
};
class AnnotationSetRefList : public Item {
public:
explicit AnnotationSetRefList(std::vector<AnnotationSetItem*>* items) : items_(items) {
size_ = sizeof(uint32_t) + items->size() * sizeof(uint32_t);
}
~AnnotationSetRefList() OVERRIDE { }
std::vector<AnnotationSetItem*>* GetItems() { return items_.get(); }
void Accept(AbstractDispatcher* dispatch) { dispatch->Dispatch(this); }
private:
std::unique_ptr<std::vector<AnnotationSetItem*>> items_; // Elements of vector can be nullptr.
DISALLOW_COPY_AND_ASSIGN(AnnotationSetRefList);
};
class FieldAnnotation {
public:
FieldAnnotation(FieldId* field_id, AnnotationSetItem* annotation_set_item)
: field_id_(field_id), annotation_set_item_(annotation_set_item) { }
FieldId* GetFieldId() const { return field_id_; }
AnnotationSetItem* GetAnnotationSetItem() const { return annotation_set_item_; }
private:
FieldId* field_id_;
AnnotationSetItem* annotation_set_item_;
DISALLOW_COPY_AND_ASSIGN(FieldAnnotation);
};
using FieldAnnotationVector = std::vector<std::unique_ptr<FieldAnnotation>>;
class MethodAnnotation {
public:
MethodAnnotation(MethodId* method_id, AnnotationSetItem* annotation_set_item)
: method_id_(method_id), annotation_set_item_(annotation_set_item) { }
MethodId* GetMethodId() const { return method_id_; }
AnnotationSetItem* GetAnnotationSetItem() const { return annotation_set_item_; }
private:
MethodId* method_id_;
AnnotationSetItem* annotation_set_item_;
DISALLOW_COPY_AND_ASSIGN(MethodAnnotation);
};
using MethodAnnotationVector = std::vector<std::unique_ptr<MethodAnnotation>>;
class ParameterAnnotation {
public:
ParameterAnnotation(MethodId* method_id, AnnotationSetRefList* annotations)
: method_id_(method_id), annotations_(annotations) { }
MethodId* GetMethodId() const { return method_id_; }
AnnotationSetRefList* GetAnnotations() { return annotations_; }
private:
MethodId* method_id_;
AnnotationSetRefList* annotations_;
DISALLOW_COPY_AND_ASSIGN(ParameterAnnotation);
};
using ParameterAnnotationVector = std::vector<std::unique_ptr<ParameterAnnotation>>;
class AnnotationsDirectoryItem : public Item {
public:
AnnotationsDirectoryItem(AnnotationSetItem* class_annotation,
FieldAnnotationVector* field_annotations,
MethodAnnotationVector* method_annotations,
ParameterAnnotationVector* parameter_annotations)
: class_annotation_(class_annotation),
field_annotations_(field_annotations),
method_annotations_(method_annotations),
parameter_annotations_(parameter_annotations) { }
AnnotationSetItem* GetClassAnnotation() const { return class_annotation_; }
FieldAnnotationVector* GetFieldAnnotations() { return field_annotations_.get(); }
MethodAnnotationVector* GetMethodAnnotations() { return method_annotations_.get(); }
ParameterAnnotationVector* GetParameterAnnotations() { return parameter_annotations_.get(); }
void Accept(AbstractDispatcher* dispatch) { dispatch->Dispatch(this); }
private:
AnnotationSetItem* class_annotation_; // This can be nullptr.
std::unique_ptr<FieldAnnotationVector> field_annotations_; // This can be nullptr.
std::unique_ptr<MethodAnnotationVector> method_annotations_; // This can be nullptr.
std::unique_ptr<ParameterAnnotationVector> parameter_annotations_; // This can be nullptr.
DISALLOW_COPY_AND_ASSIGN(AnnotationsDirectoryItem);
};
class CallSiteId : public IndexedItem {
public:
explicit CallSiteId(EncodedArrayItem* call_site_item) : call_site_item_(call_site_item) {
size_ = kCallSiteIdItemSize;
}
~CallSiteId() OVERRIDE { }
static size_t ItemSize() { return kCallSiteIdItemSize; }
EncodedArrayItem* CallSiteItem() const { return call_site_item_; }
void Accept(AbstractDispatcher* dispatch) const { dispatch->Dispatch(this); }
private:
EncodedArrayItem* call_site_item_;
DISALLOW_COPY_AND_ASSIGN(CallSiteId);
};
class MethodHandleItem : public IndexedItem {
public:
MethodHandleItem(DexFile::MethodHandleType method_handle_type, IndexedItem* field_or_method_id)
: method_handle_type_(method_handle_type),
field_or_method_id_(field_or_method_id) {
size_ = kMethodHandleItemSize;
}
~MethodHandleItem() OVERRIDE { }
static size_t ItemSize() { return kMethodHandleItemSize; }
DexFile::MethodHandleType GetMethodHandleType() const { return method_handle_type_; }
IndexedItem* GetFieldOrMethodId() const { return field_or_method_id_; }
void Accept(AbstractDispatcher* dispatch) const { dispatch->Dispatch(this); }
private:
DexFile::MethodHandleType method_handle_type_;
IndexedItem* field_or_method_id_;
DISALLOW_COPY_AND_ASSIGN(MethodHandleItem);
};
// TODO(sehr): implement MapList.
class MapList : public Item {
public:
void Accept(AbstractDispatcher* dispatch) { dispatch->Dispatch(this); }
private:
DISALLOW_COPY_AND_ASSIGN(MapList);
};
class MapItem : public Item {
public:
void Accept(AbstractDispatcher* dispatch) { dispatch->Dispatch(this); }
private:
DISALLOW_COPY_AND_ASSIGN(MapItem);
};
// Interface for building a vector of file sections for use by other clients.
struct DexFileSection {
public:
DexFileSection(const std::string& name, uint16_t type, uint32_t size, uint32_t offset)
: name(name), type(type), size(size), offset(offset) { }
std::string name;
// The type (DexFile::MapItemType).
uint16_t type;
// The size (in elements, not bytes).
uint32_t size;
// The byte offset from the start of the file.
uint32_t offset;
};
enum class SortDirection {
kSortAscending,
kSortDescending
};
std::vector<DexFileSection> GetSortedDexFileSections(dex_ir::Header* header,
SortDirection direction);
} // namespace dex_ir
} // namespace art
#endif // ART_DEXLAYOUT_DEX_IR_H_