Brian Carlstrom | db4d540 | 2011-08-09 12:18:28 -0700 | [diff] [blame] | 1 | // Copyright 2011 Google Inc. All Rights Reserved. |
| 2 | |
| 3 | #include "common_test.h" |
Brian Carlstrom | 4a289ed | 2011-08-16 17:17:49 -0700 | [diff] [blame] | 4 | #include "file.h" |
| 5 | #include "image.h" |
Brian Carlstrom | db4d540 | 2011-08-09 12:18:28 -0700 | [diff] [blame] | 6 | #include "image_writer.h" |
Brian Carlstrom | 4a289ed | 2011-08-16 17:17:49 -0700 | [diff] [blame] | 7 | #include "space.h" |
buzbee | c143c55 | 2011-08-20 17:38:58 -0700 | [diff] [blame] | 8 | #include "utils.h" |
Brian Carlstrom | db4d540 | 2011-08-09 12:18:28 -0700 | [diff] [blame] | 9 | |
| 10 | #include "gtest/gtest.h" |
| 11 | |
| 12 | namespace art { |
| 13 | |
Brian Carlstrom | f734cf5 | 2011-08-17 16:28:14 -0700 | [diff] [blame] | 14 | class ImageTest : public CommonTest {}; |
Brian Carlstrom | db4d540 | 2011-08-09 12:18:28 -0700 | [diff] [blame] | 15 | |
Brian Carlstrom | 9cff8e1 | 2011-08-18 16:47:29 -0700 | [diff] [blame] | 16 | TEST_F(ImageTest, WriteRead) { |
Brian Carlstrom | a663ea5 | 2011-08-19 23:33:41 -0700 | [diff] [blame] | 17 | |
| 18 | // TODO: move the touching of classes and GC to the ImageWriter proper |
| 19 | for (size_t i = 0; i < java_lang_dex_file_->NumClassDefs(); i++) { |
| 20 | const DexFile::ClassDef class_def = java_lang_dex_file_->GetClassDef(i); |
| 21 | const char* descriptor = java_lang_dex_file_->GetClassDescriptor(class_def); |
| 22 | Class* klass = class_linker_->FindSystemClass(descriptor); |
| 23 | ASSERT_TRUE(klass != NULL) << descriptor; |
| 24 | } |
Brian Carlstrom | 9cff8e1 | 2011-08-18 16:47:29 -0700 | [diff] [blame] | 25 | // TODO: Heap::CollectGarbage before writing |
Brian Carlstrom | a663ea5 | 2011-08-19 23:33:41 -0700 | [diff] [blame] | 26 | |
Brian Carlstrom | db4d540 | 2011-08-09 12:18:28 -0700 | [diff] [blame] | 27 | const std::vector<Space*>& spaces = Heap::GetSpaces(); |
| 28 | // can't currently deal with writing a space that might have pointers between spaces |
Brian Carlstrom | 4a289ed | 2011-08-16 17:17:49 -0700 | [diff] [blame] | 29 | ASSERT_EQ(1U, spaces.size()); |
| 30 | Space* space = spaces[0]; |
Brian Carlstrom | db4d540 | 2011-08-09 12:18:28 -0700 | [diff] [blame] | 31 | |
| 32 | ImageWriter writer; |
| 33 | ScratchFile tmp; |
Brian Carlstrom | 9cff8e1 | 2011-08-18 16:47:29 -0700 | [diff] [blame] | 34 | const int image_base = 0x50000000; |
Brian Carlstrom | 4a289ed | 2011-08-16 17:17:49 -0700 | [diff] [blame] | 35 | bool success = writer.Write(space, tmp.GetFilename(), reinterpret_cast<byte*>(image_base)); |
| 36 | ASSERT_TRUE(success); |
| 37 | |
| 38 | { |
| 39 | scoped_ptr<File> file(OS::OpenFile(tmp.GetFilename(), false)); |
| 40 | ASSERT_TRUE(file != NULL); |
| 41 | ImageHeader image_header; |
| 42 | file->ReadFully(&image_header, sizeof(image_header)); |
| 43 | ASSERT_TRUE(image_header.IsValid()); |
| 44 | ASSERT_GE(sizeof(image_header) + space->Size(), static_cast<size_t>(file->Length())); |
| 45 | } |
Brian Carlstrom | 8a43659 | 2011-08-15 21:27:23 -0700 | [diff] [blame] | 46 | |
Brian Carlstrom | a663ea5 | 2011-08-19 23:33:41 -0700 | [diff] [blame] | 47 | // tear down old runtime before making a new one, clearing out misc state |
Brian Carlstrom | 8a43659 | 2011-08-15 21:27:23 -0700 | [diff] [blame] | 48 | delete runtime_.release(); |
Brian Carlstrom | 9cff8e1 | 2011-08-18 16:47:29 -0700 | [diff] [blame] | 49 | |
| 50 | // don't reuse java_lang_dex_file_ so we make sure we don't get |
| 51 | // lucky by pointers that happen to work referencing the earlier |
| 52 | // dex. |
| 53 | delete java_lang_dex_file_.release(); |
| 54 | scoped_ptr<DexFile> dex(GetLibCoreDex()); |
| 55 | ASSERT_TRUE(dex != NULL); |
Brian Carlstrom | 8a43659 | 2011-08-15 21:27:23 -0700 | [diff] [blame] | 56 | |
| 57 | std::vector<const DexFile*> boot_class_path; |
Brian Carlstrom | 9cff8e1 | 2011-08-18 16:47:29 -0700 | [diff] [blame] | 58 | boot_class_path.push_back(dex.get()); |
Brian Carlstrom | 8a43659 | 2011-08-15 21:27:23 -0700 | [diff] [blame] | 59 | |
| 60 | Runtime::Options options; |
| 61 | options.push_back(std::make_pair("bootclasspath", &boot_class_path)); |
| 62 | std::string boot_image("-Xbootimage:"); |
| 63 | boot_image.append(tmp.GetFilename()); |
| 64 | options.push_back(std::make_pair(boot_image.c_str(), reinterpret_cast<void*>(NULL))); |
| 65 | |
| 66 | runtime_.reset(Runtime::Create(options, false)); |
| 67 | ASSERT_TRUE(runtime_ != NULL); |
| 68 | class_linker_ = runtime_->GetClassLinker(); |
Brian Carlstrom | 9cff8e1 | 2011-08-18 16:47:29 -0700 | [diff] [blame] | 69 | |
| 70 | ASSERT_EQ(2U, Heap::GetSpaces().size()); |
| 71 | Space* boot_space = Heap::GetSpaces()[0]; |
| 72 | ASSERT_TRUE(boot_space != NULL); |
| 73 | |
Brian Carlstrom | a663ea5 | 2011-08-19 23:33:41 -0700 | [diff] [blame] | 74 | // enable to display maps to debug boot_base and boot_limit checking problems below |
| 75 | if (false) { |
| 76 | const char* maps_file = "/proc/self/maps"; |
| 77 | std::string contents = ReadFileToString(maps_file); |
| 78 | LG << maps_file << ":\n" << contents; |
| 79 | } |
| 80 | |
| 81 | byte* boot_base = boot_space->GetBase(); |
| 82 | byte* boot_limit = boot_space->GetLimit(); |
Brian Carlstrom | 9cff8e1 | 2011-08-18 16:47:29 -0700 | [diff] [blame] | 83 | for (size_t i = 0; i < dex->NumClassDefs(); i++) { |
| 84 | const DexFile::ClassDef class_def = dex->GetClassDef(i); |
| 85 | const char* descriptor = dex->GetClassDescriptor(class_def); |
| 86 | Class* klass = class_linker_->FindSystemClass(descriptor); |
Brian Carlstrom | a663ea5 | 2011-08-19 23:33:41 -0700 | [diff] [blame] | 87 | EXPECT_TRUE(klass != NULL) << descriptor; |
| 88 | EXPECT_LT(boot_base, reinterpret_cast<byte*>(klass)) << descriptor; |
| 89 | EXPECT_LT(reinterpret_cast<byte*>(klass), boot_limit) << descriptor; |
Brian Carlstrom | 9cff8e1 | 2011-08-18 16:47:29 -0700 | [diff] [blame] | 90 | } |
Brian Carlstrom | db4d540 | 2011-08-09 12:18:28 -0700 | [diff] [blame] | 91 | } |
| 92 | |
| 93 | } // namespace art |