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 "os.h" |
| 8 | #include "space.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 | std::string ReadFileToString(const char* file_name) { |
| 17 | scoped_ptr<File> file(OS::OpenFile(file_name, false)); |
| 18 | CHECK(file != NULL); |
Brian Carlstrom | db4d540 | 2011-08-09 12:18:28 -0700 | [diff] [blame] | 19 | |
Brian Carlstrom | 9cff8e1 | 2011-08-18 16:47:29 -0700 | [diff] [blame^] | 20 | std::string contents; |
| 21 | char buf[8 * KB]; |
| 22 | while (true) { |
| 23 | int64_t n = file->Read(buf, sizeof(buf)); |
| 24 | CHECK_NE(-1, n); |
| 25 | if (n == 0) { |
| 26 | break; |
| 27 | } |
| 28 | contents.append(buf, n); |
| 29 | } |
| 30 | return contents; |
| 31 | } |
| 32 | |
| 33 | TEST_F(ImageTest, WriteRead) { |
| 34 | // TODO: Heap::CollectGarbage before writing |
Brian Carlstrom | db4d540 | 2011-08-09 12:18:28 -0700 | [diff] [blame] | 35 | const std::vector<Space*>& spaces = Heap::GetSpaces(); |
| 36 | // 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] | 37 | ASSERT_EQ(1U, spaces.size()); |
| 38 | Space* space = spaces[0]; |
Brian Carlstrom | db4d540 | 2011-08-09 12:18:28 -0700 | [diff] [blame] | 39 | |
| 40 | ImageWriter writer; |
| 41 | ScratchFile tmp; |
Brian Carlstrom | 9cff8e1 | 2011-08-18 16:47:29 -0700 | [diff] [blame^] | 42 | const int image_base = 0x50000000; |
Brian Carlstrom | 4a289ed | 2011-08-16 17:17:49 -0700 | [diff] [blame] | 43 | bool success = writer.Write(space, tmp.GetFilename(), reinterpret_cast<byte*>(image_base)); |
| 44 | ASSERT_TRUE(success); |
| 45 | |
| 46 | { |
| 47 | scoped_ptr<File> file(OS::OpenFile(tmp.GetFilename(), false)); |
| 48 | ASSERT_TRUE(file != NULL); |
| 49 | ImageHeader image_header; |
| 50 | file->ReadFully(&image_header, sizeof(image_header)); |
| 51 | ASSERT_TRUE(image_header.IsValid()); |
| 52 | ASSERT_GE(sizeof(image_header) + space->Size(), static_cast<size_t>(file->Length())); |
| 53 | } |
Brian Carlstrom | 8a43659 | 2011-08-15 21:27:23 -0700 | [diff] [blame] | 54 | |
| 55 | // tear down old runtime and make a new one |
| 56 | delete runtime_.release(); |
Brian Carlstrom | 9cff8e1 | 2011-08-18 16:47:29 -0700 | [diff] [blame^] | 57 | |
| 58 | // don't reuse java_lang_dex_file_ so we make sure we don't get |
| 59 | // lucky by pointers that happen to work referencing the earlier |
| 60 | // dex. |
| 61 | delete java_lang_dex_file_.release(); |
| 62 | scoped_ptr<DexFile> dex(GetLibCoreDex()); |
| 63 | ASSERT_TRUE(dex != NULL); |
Brian Carlstrom | 8a43659 | 2011-08-15 21:27:23 -0700 | [diff] [blame] | 64 | |
| 65 | std::vector<const DexFile*> boot_class_path; |
Brian Carlstrom | 9cff8e1 | 2011-08-18 16:47:29 -0700 | [diff] [blame^] | 66 | boot_class_path.push_back(dex.get()); |
Brian Carlstrom | 8a43659 | 2011-08-15 21:27:23 -0700 | [diff] [blame] | 67 | |
| 68 | Runtime::Options options; |
| 69 | options.push_back(std::make_pair("bootclasspath", &boot_class_path)); |
| 70 | std::string boot_image("-Xbootimage:"); |
| 71 | boot_image.append(tmp.GetFilename()); |
| 72 | options.push_back(std::make_pair(boot_image.c_str(), reinterpret_cast<void*>(NULL))); |
| 73 | |
| 74 | runtime_.reset(Runtime::Create(options, false)); |
| 75 | ASSERT_TRUE(runtime_ != NULL); |
| 76 | class_linker_ = runtime_->GetClassLinker(); |
Brian Carlstrom | 9cff8e1 | 2011-08-18 16:47:29 -0700 | [diff] [blame^] | 77 | |
| 78 | if (true) { |
| 79 | const char* maps_file = "/proc/self/maps"; |
| 80 | std::string contents = ReadFileToString(maps_file); |
| 81 | LG << maps_file << ":\n" << contents; |
| 82 | } |
| 83 | |
| 84 | ASSERT_EQ(2U, Heap::GetSpaces().size()); |
| 85 | Space* boot_space = Heap::GetSpaces()[0]; |
| 86 | ASSERT_TRUE(boot_space != NULL); |
| 87 | |
| 88 | // TODO: need to rebuild ClassLinker::classes_ and ::intern_table_ |
| 89 | // byte* boot_base = boot_space->GetBase(); |
| 90 | // byte* boot_limit = boot_space->GetLimit(); |
| 91 | for (size_t i = 0; i < dex->NumClassDefs(); i++) { |
| 92 | const DexFile::ClassDef class_def = dex->GetClassDef(i); |
| 93 | const char* descriptor = dex->GetClassDescriptor(class_def); |
| 94 | Class* klass = class_linker_->FindSystemClass(descriptor); |
| 95 | EXPECT_TRUE(klass != NULL); |
| 96 | // EXPECT_LT(boot_base, reinterpret_cast<byte*>(klass)); |
| 97 | // EXPECT_LT(reinterpret_cast<byte*>(klass), boot_limit); |
| 98 | } |
Brian Carlstrom | db4d540 | 2011-08-09 12:18:28 -0700 | [diff] [blame] | 99 | } |
| 100 | |
| 101 | } // namespace art |