| // Copyright 2011 Google Inc. All Rights Reserved. |
| |
| #include "common_test.h" |
| #include "file.h" |
| #include "image.h" |
| #include "image_writer.h" |
| #include "space.h" |
| #include "utils.h" |
| |
| namespace art { |
| |
| class ImageTest : public CommonTest {}; |
| |
| TEST_F(ImageTest, WriteRead) { |
| |
| // TODO: move the touching of classes and GC to the ImageWriter proper |
| for (size_t i = 0; i < java_lang_dex_file_->NumClassDefs(); i++) { |
| const DexFile::ClassDef& class_def = java_lang_dex_file_->GetClassDef(i); |
| const char* descriptor = java_lang_dex_file_->GetClassDescriptor(class_def); |
| Class* klass = class_linker_->FindSystemClass(descriptor); |
| ASSERT_TRUE(klass != NULL) << descriptor; |
| } |
| // TODO: Heap::CollectGarbage before writing |
| |
| const std::vector<Space*>& spaces = Heap::GetSpaces(); |
| // can't currently deal with writing a space that might have pointers between spaces |
| ASSERT_EQ(1U, spaces.size()); |
| Space* space = spaces[0]; |
| |
| ImageWriter writer; |
| ScratchFile tmp; |
| const int image_base = 0x50000000; |
| bool success = writer.Write(space, tmp.GetFilename(), reinterpret_cast<byte*>(image_base)); |
| ASSERT_TRUE(success); |
| |
| { |
| UniquePtr<File> file(OS::OpenFile(tmp.GetFilename(), false)); |
| ASSERT_TRUE(file.get() != NULL); |
| ImageHeader image_header; |
| file->ReadFully(&image_header, sizeof(image_header)); |
| ASSERT_TRUE(image_header.IsValid()); |
| ASSERT_GE(sizeof(image_header) + space->Size(), static_cast<size_t>(file->Length())); |
| } |
| |
| // tear down old runtime before making a new one, clearing out misc state |
| delete runtime_.release(); |
| |
| // don't reuse java_lang_dex_file_ so we make sure we don't get |
| // lucky by pointers that happen to work referencing the earlier |
| // dex. |
| delete java_lang_dex_file_.release(); |
| UniquePtr<const DexFile> dex(GetLibCoreDex()); |
| ASSERT_TRUE(dex.get() != NULL); |
| |
| std::vector<const DexFile*> boot_class_path; |
| boot_class_path.push_back(dex.get()); |
| |
| Runtime::Options options; |
| options.push_back(std::make_pair("bootclasspath", &boot_class_path)); |
| std::string boot_image("-Xbootimage:"); |
| boot_image.append(tmp.GetFilename()); |
| options.push_back(std::make_pair(boot_image.c_str(), reinterpret_cast<void*>(NULL))); |
| |
| runtime_.reset(Runtime::Create(options, false)); |
| ASSERT_TRUE(runtime_.get() != NULL); |
| class_linker_ = runtime_->GetClassLinker(); |
| |
| ASSERT_EQ(2U, Heap::GetSpaces().size()); |
| Space* boot_space = Heap::GetSpaces()[0]; |
| ASSERT_TRUE(boot_space != NULL); |
| |
| // enable to display maps to debug boot_base and boot_limit checking problems below |
| if (false) { |
| const char* maps_file = "/proc/self/maps"; |
| std::string contents = ReadFileToString(maps_file); |
| LG << maps_file << ":\n" << contents; |
| } |
| |
| byte* boot_base = boot_space->GetBase(); |
| byte* boot_limit = boot_space->GetLimit(); |
| for (size_t i = 0; i < dex->NumClassDefs(); i++) { |
| const DexFile::ClassDef& class_def = dex->GetClassDef(i); |
| const char* descriptor = dex->GetClassDescriptor(class_def); |
| Class* klass = class_linker_->FindSystemClass(descriptor); |
| EXPECT_TRUE(klass != NULL) << descriptor; |
| EXPECT_LT(boot_base, reinterpret_cast<byte*>(klass)) << descriptor; |
| EXPECT_LT(reinterpret_cast<byte*>(klass), boot_limit) << descriptor; |
| } |
| } |
| |
| } // namespace art |