Separate oat from image
Change-Id: If2abdb99826ead14e3465d90ba2acffd89709389
diff --git a/src/oat_file.cc b/src/oat_file.cc
new file mode 100644
index 0000000..6266494
--- /dev/null
+++ b/src/oat_file.cc
@@ -0,0 +1,167 @@
+// Copyright 2011 Google Inc. All Rights Reserved.
+
+#include "oat_file.h"
+
+#include <sys/mman.h>
+
+#include "file.h"
+#include "os.h"
+#include "stl_util.h"
+
+namespace art {
+
+OatFile* OatFile::Open(const std::string& filename,
+ const std::string& strip_location_prefix,
+ byte* requested_base) {
+ StringPiece location = filename;
+ if (!location.starts_with(strip_location_prefix)) {
+ LOG(ERROR) << filename << " does not start with " << strip_location_prefix;
+ return NULL;
+ }
+ location.remove_prefix(strip_location_prefix.size());
+
+ UniquePtr<OatFile> oat_file(new OatFile(location.ToString()));
+ bool success = oat_file->Read(filename, requested_base);
+ if (!success) {
+ return NULL;
+ }
+ return oat_file.release();
+}
+
+OatFile::OatFile(const std::string& filename) : location_(filename) {}
+
+OatFile::~OatFile() {
+ STLDeleteValues(&oat_dex_files_);
+}
+
+bool OatFile::Read(const std::string& filename, byte* requested_base) {
+ UniquePtr<File> file(OS::OpenFile(filename.c_str(), false));
+ if (file.get() == NULL) {
+ return false;
+ }
+
+ OatHeader oat_header;
+ bool success = file->ReadFully(&oat_header, sizeof(oat_header));
+ if (!success || !oat_header.IsValid()) {
+ LOG(WARNING) << "Invalid oat header " << filename;
+ return false;
+ }
+
+ UniquePtr<MemMap> map(MemMap::Map(requested_base,
+ file->Length(),
+ PROT_READ,
+ MAP_PRIVATE | ((requested_base != NULL) ? MAP_FIXED : 0),
+ file->Fd(),
+ 0));
+ if (map.get() == NULL) {
+ LOG(WARNING) << "Failed to map oat file " << filename;
+ return false;
+ }
+ CHECK(requested_base == 0 || requested_base == map->GetAddress()) << map->GetAddress();
+ DCHECK_EQ(0, memcmp(&oat_header, map->GetAddress(), sizeof(OatHeader)));
+
+ off_t code_offset = oat_header.GetExecutableOffset();
+ if (code_offset < file->Length()) {
+ byte* code_address = map->GetAddress() + code_offset;
+ size_t code_length = file->Length() - code_offset;
+ if (mprotect(code_address, code_length, PROT_READ | PROT_EXEC) != 0) {
+ PLOG(ERROR) << "Failed to make oat code executable.";
+ return false;
+ }
+ } else {
+ // its possible to have no code if all the methods were abstract, native, etc
+ DCHECK_EQ(code_offset, RoundUp(file->Length(), kPageSize));
+ }
+
+ const byte* oat = map->GetAddress();
+ oat += sizeof(OatHeader);
+ CHECK_LT(oat, map->GetLimit());
+ for (size_t i = 0; i < oat_header.GetDexFileCount(); i++) {
+ size_t dex_file_location_size = *reinterpret_cast<const uint32_t*>(oat);
+ oat += sizeof(dex_file_location_size);
+ CHECK_LT(oat, map->GetLimit());
+
+ const char* dex_file_location_data = reinterpret_cast<const char*>(oat);
+ oat += dex_file_location_size;
+ CHECK_LT(oat, map->GetLimit());
+
+ std::string dex_file_location(dex_file_location_data, dex_file_location_size);
+
+ uint32_t dex_file_checksum = *reinterpret_cast<const uint32_t*>(oat);
+ oat += sizeof(dex_file_checksum);
+ CHECK_LT(oat, map->GetLimit());
+
+ uint32_t classes_offset = *reinterpret_cast<const uint32_t*>(oat);
+ CHECK_GT(classes_offset, 0U);
+ CHECK_LT(classes_offset, static_cast<uint32_t>(file->Length()));
+ oat += sizeof(classes_offset);
+ CHECK_LT(oat, map->GetLimit());
+
+ uint32_t* classes_pointer = reinterpret_cast<uint32_t*>(map->GetAddress() + classes_offset);
+
+ oat_dex_files_[dex_file_location] = new OatDexFile(this,
+ dex_file_location,
+ dex_file_checksum,
+ classes_pointer);
+ }
+
+ mem_map_.reset(map.release());
+ return true;
+}
+
+const OatHeader& OatFile::GetOatHeader() const {
+ return *reinterpret_cast<const OatHeader*>(GetBase());
+}
+
+const byte* OatFile::GetBase() const {
+ CHECK(mem_map_->GetAddress() != NULL);
+ return mem_map_->GetAddress();
+}
+
+const byte* OatFile::GetLimit() const {
+ CHECK(mem_map_->GetLimit() != NULL);
+ return mem_map_->GetLimit();
+}
+
+const OatFile::OatDexFile& OatFile::GetOatDexFile(const DexFile& dex_file) {
+ Table::const_iterator it = oat_dex_files_.find(dex_file.GetLocation());
+ if (it == oat_dex_files_.end()) {
+ LOG(FATAL) << "Failed to find OatDexFile for DexFile " << dex_file.GetLocation();
+ }
+ return *it->second;
+}
+
+OatFile::OatDexFile::OatDexFile(const OatFile* oat_file,
+ std::string dex_file_location,
+ uint32_t dex_file_checksum,
+ uint32_t* classes_pointer)
+ : oat_file_(oat_file),
+ dex_file_location_(dex_file_location),
+ dex_file_checksum_(dex_file_checksum),
+ classes_pointer_(classes_pointer) {}
+
+OatFile::OatDexFile::~OatDexFile() {}
+
+const OatFile::OatClass OatFile::OatDexFile::GetOatClass(uint32_t class_def_index) const {
+ uint32_t methods_offset = classes_pointer_[class_def_index];
+ const byte* methods_pointer = oat_file_->GetBase() + methods_offset;
+ CHECK_LT(methods_pointer, oat_file_->GetLimit());
+ return OatClass(oat_file_, reinterpret_cast<const uint32_t*>(methods_pointer));
+}
+
+OatFile::OatClass::OatClass(const OatFile* oat_file, const uint32_t* methods_pointer)
+ : oat_file_(oat_file), methods_pointer_(methods_pointer) {}
+
+OatFile::OatClass::~OatClass() {}
+
+const void* OatFile::OatClass::GetMethodCode(uint32_t method_index) const {
+ uint32_t code_offset = methods_pointer_[method_index];
+ if (code_offset == 0) {
+ return NULL;
+ }
+ const void* code_pointer = reinterpret_cast<const void*>(oat_file_->GetBase() + code_offset);
+ CHECK_LT(code_pointer, oat_file_->GetLimit());
+ return code_pointer;
+}
+
+} // namespace art