oat_file_assistant_test.cc revision 94f5bda4828255b290c7b1472f38929f3b901e58
166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler/* 266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler * Copyright (C) 2014 The Android Open Source Project 366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler * 466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler * Licensed under the Apache License, Version 2.0 (the "License"); 566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler * you may not use this file except in compliance with the License. 666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler * You may obtain a copy of the License at 766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler * 866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler * http://www.apache.org/licenses/LICENSE-2.0 966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler * 1066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler * Unless required by applicable law or agreed to in writing, software 1166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler * distributed under the License is distributed on an "AS IS" BASIS, 1266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 1366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler * See the License for the specific language governing permissions and 1466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler * limitations under the License. 1566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler */ 1666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 1766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler#include "oat_file_assistant.h" 1866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 1966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler#include <algorithm> 2066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler#include <fstream> 2166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler#include <string> 2266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler#include <vector> 2366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler#include <sys/param.h> 2466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 2566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler#include <backtrace/BacktraceMap.h> 2666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler#include <gtest/gtest.h> 2766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 28c785344b87221f5e4e6473e5b762e4e61fe65dcfMathieu Chartier#include "art_field-inl.h" 293481ba2c4e4f3aa80d8c6d50a9f85dacb56b508bVladimir Marko#include "class_linker-inl.h" 3066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler#include "common_runtime_test.h" 31bb9c6b1c55e9e2308b4f5892a398a8837231fdbdAndreas Gampe#include "compiler_callbacks.h" 32f16d5727f9062379519043bc063a2c0527c59eb8Richard Uhler#include "gc/space/image_space.h" 3366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler#include "mem_map.h" 3466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler#include "os.h" 3523cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler#include "scoped_thread_state_change.h" 3666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler#include "thread-inl.h" 3766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler#include "utils.h" 3866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 3966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhlernamespace art { 4066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 4194f5bda4828255b290c7b1472f38929f3b901e58Richard Uhler// Some tests very occasionally fail: we expect to have an unrelocated non-pic 4294f5bda4828255b290c7b1472f38929f3b901e58Richard Uhler// odex file that is reported as needing relocation, but it is reported 4394f5bda4828255b290c7b1472f38929f3b901e58Richard Uhler// instead as being up to date (b/22599792). 4494f5bda4828255b290c7b1472f38929f3b901e58Richard Uhler// 4594f5bda4828255b290c7b1472f38929f3b901e58Richard Uhler// This function adds extra checks for diagnosing why the given oat file is 4694f5bda4828255b290c7b1472f38929f3b901e58Richard Uhler// reported up to date, when it should be non-pic needing relocation. 4794f5bda4828255b290c7b1472f38929f3b901e58Richard Uhler// These extra diagnostics checks should be removed once b/22599792 has been 4894f5bda4828255b290c7b1472f38929f3b901e58Richard Uhler// resolved. 4994f5bda4828255b290c7b1472f38929f3b901e58Richard Uhlerstatic void DiagnoseFlakyTestFailure(const OatFile& oat_file) { 5094f5bda4828255b290c7b1472f38929f3b901e58Richard Uhler Runtime* runtime = Runtime::Current(); 5194f5bda4828255b290c7b1472f38929f3b901e58Richard Uhler const gc::space::ImageSpace* image_space = runtime->GetHeap()->GetImageSpace(); 5294f5bda4828255b290c7b1472f38929f3b901e58Richard Uhler ASSERT_TRUE(image_space != nullptr); 5394f5bda4828255b290c7b1472f38929f3b901e58Richard Uhler const ImageHeader& image_header = image_space->GetImageHeader(); 5494f5bda4828255b290c7b1472f38929f3b901e58Richard Uhler const OatHeader& oat_header = oat_file.GetOatHeader(); 5594f5bda4828255b290c7b1472f38929f3b901e58Richard Uhler EXPECT_FALSE(oat_file.IsPic()); 5694f5bda4828255b290c7b1472f38929f3b901e58Richard Uhler EXPECT_EQ(image_header.GetOatChecksum(), oat_header.GetImageFileLocationOatChecksum()); 5794f5bda4828255b290c7b1472f38929f3b901e58Richard Uhler EXPECT_NE(reinterpret_cast<uintptr_t>(image_header.GetOatDataBegin()), 5894f5bda4828255b290c7b1472f38929f3b901e58Richard Uhler oat_header.GetImageFileLocationOatDataBegin()); 5994f5bda4828255b290c7b1472f38929f3b901e58Richard Uhler EXPECT_NE(image_header.GetPatchDelta(), oat_header.GetImagePatchDelta()); 6094f5bda4828255b290c7b1472f38929f3b901e58Richard Uhler} 6194f5bda4828255b290c7b1472f38929f3b901e58Richard Uhler 6294f5bda4828255b290c7b1472f38929f3b901e58Richard Uhler 6366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhlerclass OatFileAssistantTest : public CommonRuntimeTest { 6466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler public: 6566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler virtual void SetUp() { 6666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ReserveImageSpace(); 6766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler CommonRuntimeTest::SetUp(); 6866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 6966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Create a scratch directory to work from. 7066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler scratch_dir_ = android_data_ + "/OatFileAssistantTest"; 7166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_EQ(0, mkdir(scratch_dir_.c_str(), 0700)); 7266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 736343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler // Create a subdirectory in scratch for odex files. 746343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler odex_oat_dir_ = scratch_dir_ + "/oat"; 756343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler ASSERT_EQ(0, mkdir(odex_oat_dir_.c_str(), 0700)); 766343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler 776343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler odex_dir_ = odex_oat_dir_ + "/" + std::string(GetInstructionSetString(kRuntimeISA)); 786343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler ASSERT_EQ(0, mkdir(odex_dir_.c_str(), 0700)); 796343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler 8066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 8166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Verify the environment is as we expect 8266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler uint32_t checksum; 8366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string error_msg; 8466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(OS::FileExists(GetImageFile().c_str())) 8566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler << "Expected pre-compiled boot image to be at: " << GetImageFile(); 8666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(OS::FileExists(GetDexSrc1().c_str())) 8766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler << "Expected dex file to be at: " << GetDexSrc1(); 8866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(OS::FileExists(GetStrippedDexSrc1().c_str())) 8966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler << "Expected stripped dex file to be at: " << GetStrippedDexSrc1(); 9066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_FALSE(DexFile::GetChecksum(GetStrippedDexSrc1().c_str(), &checksum, &error_msg)) 9166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler << "Expected stripped dex file to be stripped: " << GetStrippedDexSrc1(); 9266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(OS::FileExists(GetDexSrc2().c_str())) 9366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler << "Expected dex file to be at: " << GetDexSrc2(); 9467ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler 9567ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler // GetMultiDexSrc2 should have the same primary dex checksum as 9667ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler // GetMultiDexSrc1, but a different secondary dex checksum. 9767ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler std::vector<std::unique_ptr<const DexFile>> multi1; 9867ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler ASSERT_TRUE(DexFile::Open(GetMultiDexSrc1().c_str(), 9967ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler GetMultiDexSrc1().c_str(), &error_msg, &multi1)) << error_msg; 10067ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler ASSERT_GT(multi1.size(), 1u); 10167ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler 10267ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler std::vector<std::unique_ptr<const DexFile>> multi2; 10367ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler ASSERT_TRUE(DexFile::Open(GetMultiDexSrc2().c_str(), 10467ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler GetMultiDexSrc2().c_str(), &error_msg, &multi2)) << error_msg; 10567ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler ASSERT_GT(multi2.size(), 1u); 10667ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler 10767ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler ASSERT_EQ(multi1[0]->GetLocationChecksum(), multi2[0]->GetLocationChecksum()); 10867ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler ASSERT_NE(multi1[1]->GetLocationChecksum(), multi2[1]->GetLocationChecksum()); 10966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 11066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 11166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler virtual void SetUpRuntimeOptions(RuntimeOptions* options) { 112892fc96694b51ac7a896dacd75af5b235f955825Richard Uhler // options->push_back(std::make_pair("-verbose:oat", nullptr)); 11366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 11466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Set up the image location. 11566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler options->push_back(std::make_pair("-Ximage:" + GetImageLocation(), 11666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler nullptr)); 11766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Make sure compilercallbacks are not set so that relocation will be 11866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // enabled. 119bb9c6b1c55e9e2308b4f5892a398a8837231fdbdAndreas Gampe callbacks_.reset(); 12066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 12166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 12266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler virtual void PreRuntimeCreate() { 12366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler UnreserveImageSpace(); 12466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 12566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 12666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler virtual void PostRuntimeCreate() { 12766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ReserveImageSpace(); 12866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 12966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 13066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler virtual void TearDown() { 1316343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler ClearDirectory(odex_dir_.c_str()); 1326343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler ASSERT_EQ(0, rmdir(odex_dir_.c_str())); 1336343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler 1346343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler ClearDirectory(odex_oat_dir_.c_str()); 1356343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler ASSERT_EQ(0, rmdir(odex_oat_dir_.c_str())); 13666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 13766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ClearDirectory(scratch_dir_.c_str()); 13866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_EQ(0, rmdir(scratch_dir_.c_str())); 13966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 14066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler CommonRuntimeTest::TearDown(); 14166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 14266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 14366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler void Copy(std::string src, std::string dst) { 14466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::ifstream src_stream(src, std::ios::binary); 14566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::ofstream dst_stream(dst, std::ios::binary); 14666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 14766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler dst_stream << src_stream.rdbuf(); 14866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 14966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 15066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Returns the directory where the pre-compiled core.art can be found. 15166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // TODO: We should factor out this into common tests somewhere rather than 15266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // re-hardcoding it here (This was copied originally from the elf writer 15366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // test). 15466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string GetImageDirectory() { 15566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler if (IsHost()) { 15666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler const char* host_dir = getenv("ANDROID_HOST_OUT"); 1572cebb24bfc3247d3e9be138a3350106737455918Mathieu Chartier CHECK(host_dir != nullptr); 15866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler return std::string(host_dir) + "/framework"; 15966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } else { 16066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler return std::string("/data/art-test"); 16166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 16266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 16366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 16466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string GetImageLocation() { 16566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler return GetImageDirectory() + "/core.art"; 16666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 16766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 16866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string GetImageFile() { 16966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler return GetImageDirectory() + "/" + GetInstructionSetString(kRuntimeISA) 17066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler + "/core.art"; 17166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 17266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 17366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string GetDexSrc1() { 17466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler return GetTestDexFileName("Main"); 17566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 17666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 17766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Returns the path to a dex file equivalent to GetDexSrc1, but with the dex 17866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // file stripped. 17966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string GetStrippedDexSrc1() { 18066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler return GetTestDexFileName("MainStripped"); 18166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 18266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 18366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string GetMultiDexSrc1() { 18466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler return GetTestDexFileName("MultiDex"); 18566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 18666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 18767ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler // Returns the path to a multidex file equivalent to GetMultiDexSrc2, but 18867ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler // with the contents of the secondary dex file changed. 18967ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler std::string GetMultiDexSrc2() { 19067ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler return GetTestDexFileName("MultiDexModifiedSecondary"); 19167ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler } 19267ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler 19366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string GetDexSrc2() { 19466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler return GetTestDexFileName("Nested"); 19566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 19666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 19766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Scratch directory, for dex and odex files (oat files will go in the 19866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // dalvik cache). 19966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string GetScratchDir() { 20066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler return scratch_dir_; 20166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 20266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 2036343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler // Odex directory is the subdirectory in the scratch directory where odex 20466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // files should be located. 2056343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler std::string GetOdexDir() { 2066343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler return odex_dir_; 20766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 20866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 20966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Generate an odex file for the purposes of test. 21066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // If pic is true, generates a PIC odex. 21194f5bda4828255b290c7b1472f38929f3b901e58Richard Uhler // The generated odex file will be un-relocated. 21266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler void GenerateOdexForTest(const std::string& dex_location, 21366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler const std::string& odex_location, 21466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler bool pic = false) { 21566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // For this operation, we temporarily redirect the dalvik cache so dex2oat 21666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // doesn't find the relocated image file. 21766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string android_data_tmp = GetScratchDir() + "AndroidDataTmp"; 21866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler setenv("ANDROID_DATA", android_data_tmp.c_str(), 1); 21966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::string> args; 22066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler args.push_back("--dex-file=" + dex_location); 22166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler args.push_back("--oat-file=" + odex_location); 22266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler if (pic) { 22366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler args.push_back("--compile-pic"); 22466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } else { 22566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler args.push_back("--include-patch-information"); 22605dd8a63e213d2bf025c97b9cd04eee354d0e5b4Richard Uhler 22705dd8a63e213d2bf025c97b9cd04eee354d0e5b4Richard Uhler // We need to use the quick compiler to generate non-PIC code, because 22805dd8a63e213d2bf025c97b9cd04eee354d0e5b4Richard Uhler // the optimizing compiler always generates PIC. 22905dd8a63e213d2bf025c97b9cd04eee354d0e5b4Richard Uhler args.push_back("--compiler-backend=Quick"); 23066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 23166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler args.push_back("--runtime-arg"); 23266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler args.push_back("-Xnorelocate"); 23366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string error_msg; 23466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(OatFileAssistant::Dex2Oat(args, &error_msg)) << error_msg; 23566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler setenv("ANDROID_DATA", android_data_.c_str(), 1); 23694f5bda4828255b290c7b1472f38929f3b901e58Richard Uhler 23794f5bda4828255b290c7b1472f38929f3b901e58Richard Uhler // Verify the odex file was generated as expected. 23894f5bda4828255b290c7b1472f38929f3b901e58Richard Uhler std::unique_ptr<OatFile> odex_file(OatFile::Open( 23994f5bda4828255b290c7b1472f38929f3b901e58Richard Uhler odex_location.c_str(), odex_location.c_str(), nullptr, nullptr, 24094f5bda4828255b290c7b1472f38929f3b901e58Richard Uhler false, dex_location.c_str(), &error_msg)); 24194f5bda4828255b290c7b1472f38929f3b901e58Richard Uhler ASSERT_TRUE(odex_file.get() != nullptr) << error_msg; 24294f5bda4828255b290c7b1472f38929f3b901e58Richard Uhler 24394f5bda4828255b290c7b1472f38929f3b901e58Richard Uhler if (!pic) { 24494f5bda4828255b290c7b1472f38929f3b901e58Richard Uhler DiagnoseFlakyTestFailure(*odex_file); 24594f5bda4828255b290c7b1472f38929f3b901e58Richard Uhler } 24666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 24766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 24866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler void GeneratePicOdexForTest(const std::string& dex_location, 24966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler const std::string& odex_location) { 25066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler GenerateOdexForTest(dex_location, odex_location, true); 25166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 25266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 25366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler private: 25466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Reserve memory around where the image will be loaded so other memory 25566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // won't conflict when it comes time to load the image. 25666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // This can be called with an already loaded image to reserve the space 25766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // around it. 25866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler void ReserveImageSpace() { 25966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler MemMap::Init(); 26066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 26166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Ensure a chunk of memory is reserved for the image space. 26266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler uintptr_t reservation_start = ART_BASE_ADDRESS + ART_BASE_ADDRESS_MIN_DELTA; 26366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler uintptr_t reservation_end = ART_BASE_ADDRESS + ART_BASE_ADDRESS_MAX_DELTA 2643dbf23412481c4da51f0ebe32bec5d300c36834bHiroshi Yamauchi // Include the main space that has to come right after the 2653dbf23412481c4da51f0ebe32bec5d300c36834bHiroshi Yamauchi // image in case of the GSS collector. 2663dbf23412481c4da51f0ebe32bec5d300c36834bHiroshi Yamauchi + 384 * MB; 26766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 26866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::unique_ptr<BacktraceMap> map(BacktraceMap::Create(getpid(), true)); 26966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(map.get() != nullptr) << "Failed to build process map"; 27066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler for (BacktraceMap::const_iterator it = map->begin(); 27166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler reservation_start < reservation_end && it != map->end(); ++it) { 2723efe979d4292330c8fab1708a4361e58681a88cbRichard Uhler ReserveImageSpaceChunk(reservation_start, std::min(it->start, reservation_end)); 2733efe979d4292330c8fab1708a4361e58681a88cbRichard Uhler reservation_start = std::max(reservation_start, it->end); 2743efe979d4292330c8fab1708a4361e58681a88cbRichard Uhler } 2753efe979d4292330c8fab1708a4361e58681a88cbRichard Uhler ReserveImageSpaceChunk(reservation_start, reservation_end); 2763efe979d4292330c8fab1708a4361e58681a88cbRichard Uhler } 27766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 2783efe979d4292330c8fab1708a4361e58681a88cbRichard Uhler // Reserve a chunk of memory for the image space in the given range. 2793efe979d4292330c8fab1708a4361e58681a88cbRichard Uhler // Only has effect for chunks with a positive number of bytes. 2803efe979d4292330c8fab1708a4361e58681a88cbRichard Uhler void ReserveImageSpaceChunk(uintptr_t start, uintptr_t end) { 2813efe979d4292330c8fab1708a4361e58681a88cbRichard Uhler if (start < end) { 2823efe979d4292330c8fab1708a4361e58681a88cbRichard Uhler std::string error_msg; 28366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler image_reservation_.push_back(std::unique_ptr<MemMap>( 28466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler MemMap::MapAnonymous("image reservation", 2853efe979d4292330c8fab1708a4361e58681a88cbRichard Uhler reinterpret_cast<uint8_t*>(start), end - start, 28666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler PROT_NONE, false, false, &error_msg))); 28766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(image_reservation_.back().get() != nullptr) << error_msg; 28866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler LOG(INFO) << "Reserved space for image " << 28966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler reinterpret_cast<void*>(image_reservation_.back()->Begin()) << "-" << 29066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler reinterpret_cast<void*>(image_reservation_.back()->End()); 29166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 29266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 29366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 29466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 29566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Unreserve any memory reserved by ReserveImageSpace. This should be called 29666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // before the image is loaded. 29766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler void UnreserveImageSpace() { 29866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler image_reservation_.clear(); 29966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 30066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 30166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string scratch_dir_; 3026343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler std::string odex_oat_dir_; 3036343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler std::string odex_dir_; 30466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::unique_ptr<MemMap>> image_reservation_; 30566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler}; 30666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 30766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhlerclass OatFileAssistantNoDex2OatTest : public OatFileAssistantTest { 30866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler public: 30966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler virtual void SetUpRuntimeOptions(RuntimeOptions* options) { 31066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistantTest::SetUpRuntimeOptions(options); 31166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler options->push_back(std::make_pair("-Xnodex2oat", nullptr)); 31266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 31366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler}; 31466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 31566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Generate an oat file for the purposes of test, as opposed to testing 31666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// generation of oat files. 31766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhlerstatic void GenerateOatForTest(const char* dex_location) { 31866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location, kRuntimeISA, false); 31966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 32066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string error_msg; 32166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(oat_file_assistant.GenerateOatFile(&error_msg)) << error_msg; 32266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 32366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 32466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: We have a DEX file, but no OAT file for it. 32595abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler// Expect: The status is kDex2OatNeeded. 32666d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, DexNoOat) { 32766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/DexNoOat.jar"; 32866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc1(), dex_location); 32966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 33066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, false); 33166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 33295abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_EQ(OatFileAssistant::kDex2OatNeeded, oat_file_assistant.GetDexOptNeeded()); 33366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 33466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.IsInBootClassPath()); 33566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileExists()); 33666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileIsOutOfDate()); 33766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileNeedsRelocation()); 33866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsUpToDate()); 33995abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_EQ(OatFileAssistant::kOatOutOfDate, oat_file_assistant.OdexFileStatus()); 34066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileExists()); 34166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileIsOutOfDate()); 34266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileNeedsRelocation()); 34366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsUpToDate()); 34495abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_EQ(OatFileAssistant::kOatOutOfDate, oat_file_assistant.OatFileStatus()); 3459b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_TRUE(oat_file_assistant.HasOriginalDexFiles()); 34666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 34766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 34866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: We have no DEX file and no OAT file. 3499b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler// Expect: Status is kNoDexOptNeeded. Loading should fail, but not crash. 35066d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, NoDexNoOat) { 35166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/NoDexNoOat.jar"; 35266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 35366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, true); 35466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 3559b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_EQ(OatFileAssistant::kNoDexOptNeeded, oat_file_assistant.GetDexOptNeeded()); 3569b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_FALSE(oat_file_assistant.HasOriginalDexFiles()); 3579b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler 3589b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler // Trying to make the oat file up to date should not fail or crash. 3599b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler std::string error_msg; 3609b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_TRUE(oat_file_assistant.MakeUpToDate(&error_msg)); 3619b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler 3629b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler // Trying to get the best oat file should fail, but not crash. 36366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::unique_ptr<OatFile> oat_file = oat_file_assistant.GetBestOatFile(); 36466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(nullptr, oat_file.get()); 36566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 36666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 36766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: We have a DEX file and up-to-date OAT file for it. 36895abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler// Expect: The status is kNoDexOptNeeded. 36966d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, OatUpToDate) { 37066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/OatUpToDate.jar"; 37166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc1(), dex_location); 37266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler GenerateOatForTest(dex_location.c_str()); 37366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 37466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, false); 37566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 37695abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_EQ(OatFileAssistant::kNoDexOptNeeded, oat_file_assistant.GetDexOptNeeded()); 37766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.IsInBootClassPath()); 37866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileExists()); 37966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileIsOutOfDate()); 38066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsUpToDate()); 38166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileExists()); 38266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsOutOfDate()); 38366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileNeedsRelocation()); 38466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileIsUpToDate()); 38595abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_EQ(OatFileAssistant::kOatUpToDate, oat_file_assistant.OatFileStatus()); 3869b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_TRUE(oat_file_assistant.HasOriginalDexFiles()); 38766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 38866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 38966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: We have a MultiDEX file and up-to-date OAT file for it. 39095abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler// Expect: The status is kNoDexOptNeeded and we load all dex files. 39166d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, MultiDexOatUpToDate) { 39266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/MultiDexOatUpToDate.jar"; 39366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetMultiDexSrc1(), dex_location); 39466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler GenerateOatForTest(dex_location.c_str()); 39566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 396e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, true); 39795abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_EQ(OatFileAssistant::kNoDexOptNeeded, oat_file_assistant.GetDexOptNeeded()); 3989b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_TRUE(oat_file_assistant.HasOriginalDexFiles()); 39995abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler 40095abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler // Verify we can load both dex files. 401e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler std::unique_ptr<OatFile> oat_file = oat_file_assistant.GetBestOatFile(); 402e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler ASSERT_TRUE(oat_file.get() != nullptr); 403e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler EXPECT_TRUE(oat_file->IsExecutable()); 404e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler std::vector<std::unique_ptr<const DexFile>> dex_files; 405e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler dex_files = oat_file_assistant.LoadDexFiles(*oat_file, dex_location.c_str()); 406e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler EXPECT_EQ(2u, dex_files.size()); 407e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler} 408e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler 40967ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler// Case: We have a MultiDEX file where the secondary dex file is out of date. 41067ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler// Expect: The status is kDex2OatNeeded. 41167ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard UhlerTEST_F(OatFileAssistantTest, MultiDexSecondaryOutOfDate) { 41267ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler std::string dex_location = GetScratchDir() + "/MultiDexSecondaryOutOfDate.jar"; 41367ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler 41467ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler // Compile code for GetMultiDexSrc1. 41567ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler Copy(GetMultiDexSrc1(), dex_location); 41667ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler GenerateOatForTest(dex_location.c_str()); 41767ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler 41867ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler // Now overwrite the dex file with GetMultiDexSrc2 so the secondary checksum 41967ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler // is out of date. 42067ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler Copy(GetMultiDexSrc2(), dex_location); 42167ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler 42267ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, true); 42367ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler EXPECT_EQ(OatFileAssistant::kDex2OatNeeded, oat_file_assistant.GetDexOptNeeded()); 4249b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_TRUE(oat_file_assistant.HasOriginalDexFiles()); 42567ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler} 42667ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler 427e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler// Case: We have a MultiDEX file and up-to-date OAT file for it with relative 428e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler// encoded dex locations. 42995abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler// Expect: The oat file status is kNoDexOptNeeded. 430e5fed03772144595c0904faf3d6974cc55214c8cRichard UhlerTEST_F(OatFileAssistantTest, RelativeEncodedDexLocation) { 431e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler std::string dex_location = GetScratchDir() + "/RelativeEncodedDexLocation.jar"; 4326343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler std::string oat_location = GetOdexDir() + "/RelativeEncodedDexLocation.oat"; 433e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler 434e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler // Create the dex file 435e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler Copy(GetMultiDexSrc1(), dex_location); 436e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler 437e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler // Create the oat file with relative encoded dex location. 438e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler std::vector<std::string> args; 439e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler args.push_back("--dex-file=" + dex_location); 440e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler args.push_back("--dex-location=" + std::string("RelativeEncodedDexLocation.jar")); 441e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler args.push_back("--oat-file=" + oat_location); 442e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler 443e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler std::string error_msg; 444e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler ASSERT_TRUE(OatFileAssistant::Dex2Oat(args, &error_msg)) << error_msg; 445e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler 446e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler // Verify we can load both dex files. 447e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), 448e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler oat_location.c_str(), 449e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler kRuntimeISA, true); 450e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler std::unique_ptr<OatFile> oat_file = oat_file_assistant.GetBestOatFile(); 45166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(oat_file.get() != nullptr); 45266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file->IsExecutable()); 45366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::unique_ptr<const DexFile>> dex_files; 454e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler dex_files = oat_file_assistant.LoadDexFiles(*oat_file, dex_location.c_str()); 45566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(2u, dex_files.size()); 45666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 45766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 45895abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler// Case: We have a DEX file and out-of-date OAT file. 45995abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler// Expect: The status is kDex2OatNeeded. 46066d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, OatOutOfDate) { 46166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/OatOutOfDate.jar"; 46266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 46366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // We create a dex, generate an oat for it, then overwrite the dex with a 46466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // different dex to make the oat out of date. 46566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc1(), dex_location); 46666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler GenerateOatForTest(dex_location.c_str()); 46766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc2(), dex_location); 46866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 46966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, false); 47095abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_EQ(OatFileAssistant::kDex2OatNeeded, oat_file_assistant.GetDexOptNeeded()); 47166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 47266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.IsInBootClassPath()); 47366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileExists()); 47466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileIsOutOfDate()); 47566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsUpToDate()); 47666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileExists()); 47766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileIsOutOfDate()); 47866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsUpToDate()); 4799b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_TRUE(oat_file_assistant.HasOriginalDexFiles()); 48066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 48166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 48266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: We have a DEX file and an ODEX file, but no OAT file. 48395abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler// Expect: The status is kPatchOatNeeded. 48466d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, DexOdexNoOat) { 48566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/DexOdexNoOat.jar"; 4866343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler std::string odex_location = GetOdexDir() + "/DexOdexNoOat.odex"; 48766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 48866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Create the dex and odex files 48966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc1(), dex_location); 49066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler GenerateOdexForTest(dex_location, odex_location); 49166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 49266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Verify the status. 49366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, false); 49466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 49595abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_EQ(OatFileAssistant::kPatchOatNeeded, oat_file_assistant.GetDexOptNeeded()); 49666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 49766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.IsInBootClassPath()); 49866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileExists()); 49966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsOutOfDate()); 50066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsUpToDate()); 50166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileNeedsRelocation()); 50266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileExists()); 50366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileIsOutOfDate()); 50466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsUpToDate()); 5059b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_TRUE(oat_file_assistant.HasOriginalDexFiles()); 5065f946da9c362216e9144b142ec0e5b90073b836dRichard Uhler 5075f946da9c362216e9144b142ec0e5b90073b836dRichard Uhler // We should still be able to get the non-executable odex file to run from. 5085f946da9c362216e9144b142ec0e5b90073b836dRichard Uhler std::unique_ptr<OatFile> oat_file = oat_file_assistant.GetBestOatFile(); 5095f946da9c362216e9144b142ec0e5b90073b836dRichard Uhler ASSERT_TRUE(oat_file.get() != nullptr); 5102639e8e8c9b3ce33c97ddac1b0e36a185cc02f8fRichard Uhler 5112639e8e8c9b3ce33c97ddac1b0e36a185cc02f8fRichard Uhler DiagnoseFlakyTestFailure(*oat_file); 51266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 51366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 51466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: We have a stripped DEX file and an ODEX file, but no OAT file. 51595abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler// Expect: The status is kPatchOatNeeded 51666d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, StrippedDexOdexNoOat) { 51766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/StrippedDexOdexNoOat.jar"; 5186343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler std::string odex_location = GetOdexDir() + "/StrippedDexOdexNoOat.odex"; 51966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 52066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Create the dex and odex files 52166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc1(), dex_location); 52266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler GenerateOdexForTest(dex_location, odex_location); 52366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 52466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Strip the dex file 52566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetStrippedDexSrc1(), dex_location); 52666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 52766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Verify the status. 52866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, true); 52966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 53095abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_EQ(OatFileAssistant::kPatchOatNeeded, oat_file_assistant.GetDexOptNeeded()); 53166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 53266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.IsInBootClassPath()); 53366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileExists()); 53466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsOutOfDate()); 53566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsUpToDate()); 53666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileExists()); 53766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileIsOutOfDate()); 53866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsUpToDate()); 5399b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_FALSE(oat_file_assistant.HasOriginalDexFiles()); 54066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 54166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Make the oat file up to date. 54266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string error_msg; 54366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(oat_file_assistant.MakeUpToDate(&error_msg)) << error_msg; 54466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 54595abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_EQ(OatFileAssistant::kNoDexOptNeeded, oat_file_assistant.GetDexOptNeeded()); 54666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 54766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.IsInBootClassPath()); 54866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileExists()); 54966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsOutOfDate()); 55066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsUpToDate()); 55166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileExists()); 55266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsOutOfDate()); 55366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileIsUpToDate()); 5549b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_FALSE(oat_file_assistant.HasOriginalDexFiles()); 55566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 55666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Verify we can load the dex files from it. 55766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::unique_ptr<OatFile> oat_file = oat_file_assistant.GetBestOatFile(); 55866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(oat_file.get() != nullptr); 55966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file->IsExecutable()); 56066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::unique_ptr<const DexFile>> dex_files; 56166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler dex_files = oat_file_assistant.LoadDexFiles(*oat_file, dex_location.c_str()); 56266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(1u, dex_files.size()); 56366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 56466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 56595abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler// Case: We have a stripped DEX file, an ODEX file, and an out-of-date OAT file. 56695abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler// Expect: The status is kPatchOatNeeded. 56766d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, StrippedDexOdexOat) { 56866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/StrippedDexOdexOat.jar"; 5696343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler std::string odex_location = GetOdexDir() + "/StrippedDexOdexOat.odex"; 57066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 57166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Create the oat file from a different dex file so it looks out of date. 57266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc2(), dex_location); 57366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler GenerateOatForTest(dex_location.c_str()); 57466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 57566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Create the odex file 57666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc1(), dex_location); 57766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler GenerateOdexForTest(dex_location, odex_location); 57866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 57966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Strip the dex file. 58066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetStrippedDexSrc1(), dex_location); 58166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 58266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Verify the status. 58366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, true); 58466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 58595abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_EQ(OatFileAssistant::kPatchOatNeeded, oat_file_assistant.GetDexOptNeeded()); 58666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 58766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.IsInBootClassPath()); 58866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileExists()); 58966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsOutOfDate()); 59066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileNeedsRelocation()); 59166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsUpToDate()); 59266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileExists()); 59366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileIsOutOfDate()); 59466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsUpToDate()); 5959b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_FALSE(oat_file_assistant.HasOriginalDexFiles()); 59666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 59766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Make the oat file up to date. 59866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string error_msg; 59966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(oat_file_assistant.MakeUpToDate(&error_msg)) << error_msg; 60066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 60195abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_EQ(OatFileAssistant::kNoDexOptNeeded, oat_file_assistant.GetDexOptNeeded()); 60266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 60366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.IsInBootClassPath()); 60466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileExists()); 60566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsOutOfDate()); 60666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileNeedsRelocation()); 60766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsUpToDate()); 60866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileExists()); 60966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsOutOfDate()); 61066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileNeedsRelocation()); 61166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileIsUpToDate()); 6129b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_FALSE(oat_file_assistant.HasOriginalDexFiles()); 61366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 61466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Verify we can load the dex files from it. 61566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::unique_ptr<OatFile> oat_file = oat_file_assistant.GetBestOatFile(); 61666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(oat_file.get() != nullptr); 61766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file->IsExecutable()); 61866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::unique_ptr<const DexFile>> dex_files; 61966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler dex_files = oat_file_assistant.LoadDexFiles(*oat_file, dex_location.c_str()); 62066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(1u, dex_files.size()); 62166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 62266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 6239b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler// Case: We have a stripped (or resource-only) DEX file, no ODEX file and no 6249b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler// OAT file. Expect: The status is kNoDexOptNeeded. 6259b994ea841eaaefbdda652251894a74db9cefcc8Richard UhlerTEST_F(OatFileAssistantTest, ResourceOnlyDex) { 6269b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler std::string dex_location = GetScratchDir() + "/ResourceOnlyDex.jar"; 6279b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler 6289b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler Copy(GetStrippedDexSrc1(), dex_location); 6299b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler 6309b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler // Verify the status. 6319b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, true); 6329b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler 6339b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_EQ(OatFileAssistant::kNoDexOptNeeded, oat_file_assistant.GetDexOptNeeded()); 6349b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler 6359b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_FALSE(oat_file_assistant.IsInBootClassPath()); 6369b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileExists()); 6379b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileIsOutOfDate()); 6389b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileNeedsRelocation()); 6399b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsUpToDate()); 6409b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_FALSE(oat_file_assistant.OatFileExists()); 6419b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_TRUE(oat_file_assistant.OatFileIsOutOfDate()); 6429b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsUpToDate()); 6439b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_FALSE(oat_file_assistant.HasOriginalDexFiles()); 6449b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler 6459b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler // Make the oat file up to date. This should have no effect. 6469b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler std::string error_msg; 6479b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_TRUE(oat_file_assistant.MakeUpToDate(&error_msg)) << error_msg; 6489b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler 6499b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_EQ(OatFileAssistant::kNoDexOptNeeded, oat_file_assistant.GetDexOptNeeded()); 6509b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler 6519b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_FALSE(oat_file_assistant.IsInBootClassPath()); 6529b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileExists()); 6539b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileIsOutOfDate()); 6549b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileNeedsRelocation()); 6559b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsUpToDate()); 6569b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_FALSE(oat_file_assistant.OatFileExists()); 6579b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_TRUE(oat_file_assistant.OatFileIsOutOfDate()); 6589b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsUpToDate()); 6599b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_FALSE(oat_file_assistant.HasOriginalDexFiles()); 6609b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler} 6619b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler 66295abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler// Case: We have a DEX file, no ODEX file and an OAT file that needs 66395abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler// relocation. 66495abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler// Expect: The status is kSelfPatchOatNeeded. 66595abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard UhlerTEST_F(OatFileAssistantTest, SelfRelocation) { 66695abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler std::string dex_location = GetScratchDir() + "/SelfRelocation.jar"; 66795abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler std::string oat_location = GetOdexDir() + "/SelfRelocation.oat"; 66895abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler 66995abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler // Create the dex and odex files 67095abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler Copy(GetDexSrc1(), dex_location); 67195abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler GenerateOdexForTest(dex_location, oat_location); 67295abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler 67395abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), 67495abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler oat_location.c_str(), kRuntimeISA, true); 67595abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler 67695abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_EQ(OatFileAssistant::kSelfPatchOatNeeded, oat_file_assistant.GetDexOptNeeded()); 67795abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler 67895abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_FALSE(oat_file_assistant.IsInBootClassPath()); 67995abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileExists()); 68095abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileIsOutOfDate()); 68195abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileNeedsRelocation()); 68295abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsUpToDate()); 68395abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_TRUE(oat_file_assistant.OatFileExists()); 68495abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_TRUE(oat_file_assistant.OatFileNeedsRelocation()); 68595abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsOutOfDate()); 68695abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsUpToDate()); 6879b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_TRUE(oat_file_assistant.HasOriginalDexFiles()); 68895abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler 68995abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler // Make the oat file up to date. 69095abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler std::string error_msg; 69195abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler ASSERT_TRUE(oat_file_assistant.MakeUpToDate(&error_msg)) << error_msg; 69295abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler 69395abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_EQ(OatFileAssistant::kNoDexOptNeeded, oat_file_assistant.GetDexOptNeeded()); 69495abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler 69595abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_FALSE(oat_file_assistant.IsInBootClassPath()); 69695abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileExists()); 69795abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileIsOutOfDate()); 69895abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileNeedsRelocation()); 69995abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsUpToDate()); 70095abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_TRUE(oat_file_assistant.OatFileExists()); 70195abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsOutOfDate()); 70295abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_FALSE(oat_file_assistant.OatFileNeedsRelocation()); 70395abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_TRUE(oat_file_assistant.OatFileIsUpToDate()); 7049b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_TRUE(oat_file_assistant.HasOriginalDexFiles()); 70595abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler 70695abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler std::unique_ptr<OatFile> oat_file = oat_file_assistant.GetBestOatFile(); 70795abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler ASSERT_TRUE(oat_file.get() != nullptr); 70895abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_TRUE(oat_file->IsExecutable()); 70995abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler std::vector<std::unique_ptr<const DexFile>> dex_files; 71095abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler dex_files = oat_file_assistant.LoadDexFiles(*oat_file, dex_location.c_str()); 71195abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_EQ(1u, dex_files.size()); 71295abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler} 71395abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler 71466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: We have a DEX file, an ODEX file and an OAT file, where the ODEX and 71566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// OAT files both have patch delta of 0. 71695abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler// Expect: It shouldn't crash, and status is kPatchOatNeeded. 71766d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, OdexOatOverlap) { 71866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/OdexOatOverlap.jar"; 7196343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler std::string odex_location = GetOdexDir() + "/OdexOatOverlap.odex"; 7206343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler std::string oat_location = GetOdexDir() + "/OdexOatOverlap.oat"; 72166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 72266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Create the dex and odex files 72366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc1(), dex_location); 72466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler GenerateOdexForTest(dex_location, odex_location); 72566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 72666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Create the oat file by copying the odex so they are located in the same 72766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // place in memory. 72866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(odex_location, oat_location); 72966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 73066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Verify things don't go bad. 73166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), 73266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler oat_location.c_str(), kRuntimeISA, true); 73366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 73495abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_EQ(OatFileAssistant::kPatchOatNeeded, oat_file_assistant.GetDexOptNeeded()); 73566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 73666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.IsInBootClassPath()); 73766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileExists()); 73866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsOutOfDate()); 73966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsUpToDate()); 74066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileExists()); 74166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsOutOfDate()); 74266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsUpToDate()); 7439b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_TRUE(oat_file_assistant.HasOriginalDexFiles()); 74466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 74566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Things aren't relocated, so it should fall back to interpreted. 74666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::unique_ptr<OatFile> oat_file = oat_file_assistant.GetBestOatFile(); 74766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(oat_file.get() != nullptr); 748f16d5727f9062379519043bc063a2c0527c59eb8Richard Uhler 74966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file->IsExecutable()); 75066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::unique_ptr<const DexFile>> dex_files; 75166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler dex_files = oat_file_assistant.LoadDexFiles(*oat_file, dex_location.c_str()); 75266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(1u, dex_files.size()); 753f16d5727f9062379519043bc063a2c0527c59eb8Richard Uhler 7542639e8e8c9b3ce33c97ddac1b0e36a185cc02f8fRichard Uhler DiagnoseFlakyTestFailure(*oat_file); 75566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 75666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 75766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: We have a DEX file and a PIC ODEX file, but no OAT file. 75895abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler// Expect: The status is kNoDexOptNeeded, because PIC needs no relocation. 75966d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, DexPicOdexNoOat) { 76066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/DexPicOdexNoOat.jar"; 7616343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler std::string odex_location = GetOdexDir() + "/DexPicOdexNoOat.odex"; 76266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 76366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Create the dex and odex files 76466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc1(), dex_location); 76566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler GeneratePicOdexForTest(dex_location, odex_location); 76666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 76766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Verify the status. 76866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, false); 76966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 77095abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_EQ(OatFileAssistant::kNoDexOptNeeded, oat_file_assistant.GetDexOptNeeded()); 77166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 77266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.IsInBootClassPath()); 77366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileExists()); 77466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsOutOfDate()); 77566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileIsUpToDate()); 77666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileExists()); 77766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileIsOutOfDate()); 77866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsUpToDate()); 7799b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_TRUE(oat_file_assistant.HasOriginalDexFiles()); 78066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 78166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 78266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: We have a DEX file and up-to-date OAT file for it. 78366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Expect: We should load an executable dex file. 78466d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, LoadOatUpToDate) { 78566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/LoadOatUpToDate.jar"; 78666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 78766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc1(), dex_location); 78866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler GenerateOatForTest(dex_location.c_str()); 78966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 79066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Load the oat using an oat file assistant. 79166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, true); 79266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 79366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::unique_ptr<OatFile> oat_file = oat_file_assistant.GetBestOatFile(); 79466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(oat_file.get() != nullptr); 79566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file->IsExecutable()); 79666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::unique_ptr<const DexFile>> dex_files; 79766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler dex_files = oat_file_assistant.LoadDexFiles(*oat_file, dex_location.c_str()); 79866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(1u, dex_files.size()); 79966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 80066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 80166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: We have a DEX file and up-to-date OAT file for it. 80266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Expect: Loading non-executable should load the oat non-executable. 80366d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, LoadNoExecOatUpToDate) { 80466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/LoadNoExecOatUpToDate.jar"; 80566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 80666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc1(), dex_location); 80766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler GenerateOatForTest(dex_location.c_str()); 80866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 80966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Load the oat using an oat file assistant. 81066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, false); 81166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 81266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::unique_ptr<OatFile> oat_file = oat_file_assistant.GetBestOatFile(); 81366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(oat_file.get() != nullptr); 81466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file->IsExecutable()); 81566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::unique_ptr<const DexFile>> dex_files; 81666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler dex_files = oat_file_assistant.LoadDexFiles(*oat_file, dex_location.c_str()); 81766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(1u, dex_files.size()); 81866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 81966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 82066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: We have a DEX file. 82166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Expect: We should load an executable dex file from an alternative oat 82266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// location. 82366d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, LoadDexNoAlternateOat) { 82466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/LoadDexNoAlternateOat.jar"; 82566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string oat_location = GetScratchDir() + "/LoadDexNoAlternateOat.oat"; 82666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 82766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc1(), dex_location); 82866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 82966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant( 83066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler dex_location.c_str(), oat_location.c_str(), kRuntimeISA, true); 83166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string error_msg; 83266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(oat_file_assistant.MakeUpToDate(&error_msg)) << error_msg; 83366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 83466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::unique_ptr<OatFile> oat_file = oat_file_assistant.GetBestOatFile(); 83566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(oat_file.get() != nullptr); 83666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file->IsExecutable()); 83766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::unique_ptr<const DexFile>> dex_files; 83866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler dex_files = oat_file_assistant.LoadDexFiles(*oat_file, dex_location.c_str()); 83966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(1u, dex_files.size()); 84066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 84166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(OS::FileExists(oat_location.c_str())); 84266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 84366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Verify it didn't create an oat in the default location. 84466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant ofm(dex_location.c_str(), kRuntimeISA, false); 84566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(ofm.OatFileExists()); 84666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 84766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 84866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Turn an absolute path into a path relative to the current working 84966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// directory. 85066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhlerstatic std::string MakePathRelative(std::string target) { 85166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler char buf[MAXPATHLEN]; 85266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string cwd = getcwd(buf, MAXPATHLEN); 85366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 85466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Split the target and cwd paths into components. 85566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::string> target_path; 85666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::string> cwd_path; 85766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Split(target, '/', &target_path); 85866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Split(cwd, '/', &cwd_path); 85966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 86066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Reverse the path components, so we can use pop_back(). 86166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::reverse(target_path.begin(), target_path.end()); 86266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::reverse(cwd_path.begin(), cwd_path.end()); 86366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 86466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Drop the common prefix of the paths. Because we reversed the path 86566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // components, this becomes the common suffix of target_path and cwd_path. 86666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler while (!target_path.empty() && !cwd_path.empty() 86766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler && target_path.back() == cwd_path.back()) { 86866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler target_path.pop_back(); 86966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler cwd_path.pop_back(); 87066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 87166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 87266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // For each element of the remaining cwd_path, add '..' to the beginning 87366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // of the target path. Because we reversed the path components, we add to 87466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // the end of target_path. 87566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler for (unsigned int i = 0; i < cwd_path.size(); i++) { 87666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler target_path.push_back(".."); 87766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 87866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 87966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Reverse again to get the right path order, and join to get the result. 88066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::reverse(target_path.begin(), target_path.end()); 88166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler return Join(target_path, '/'); 88266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 88366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 88466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: Non-absolute path to Dex location. 88566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Expect: Not sure, but it shouldn't crash. 88666d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, NonAbsoluteDexLocation) { 88766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string abs_dex_location = GetScratchDir() + "/NonAbsoluteDexLocation.jar"; 88866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc1(), abs_dex_location); 88966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 89066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = MakePathRelative(abs_dex_location); 89166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, true); 89266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 89366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.IsInBootClassPath()); 89495abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_EQ(OatFileAssistant::kDex2OatNeeded, oat_file_assistant.GetDexOptNeeded()); 89566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileExists()); 89666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileExists()); 89766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileIsOutOfDate()); 89866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsUpToDate()); 89966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileIsOutOfDate()); 90066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsUpToDate()); 90166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 90266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 90366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: Very short, non-existent Dex location. 9049b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler// Expect: kNoDexOptNeeded. 90566d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, ShortDexLocation) { 90666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = "/xx"; 90766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 90866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, true); 90966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 91066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.IsInBootClassPath()); 9119b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_EQ(OatFileAssistant::kNoDexOptNeeded, oat_file_assistant.GetDexOptNeeded()); 91266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileExists()); 91366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileExists()); 91466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileIsOutOfDate()); 91566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsUpToDate()); 91666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileIsOutOfDate()); 91766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsUpToDate()); 9189b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_FALSE(oat_file_assistant.HasOriginalDexFiles()); 91966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 9209b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler // Trying to make it up to date should have no effect. 92166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string error_msg; 9229b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_TRUE(oat_file_assistant.MakeUpToDate(&error_msg)); 9239b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_TRUE(error_msg.empty()); 92466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 92566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 92666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: Non-standard extension for dex file. 92795abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler// Expect: The status is kDex2OatNeeded. 92866d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, LongDexExtension) { 92966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/LongDexExtension.jarx"; 93066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc1(), dex_location); 93166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 93266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, false); 93366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 93495abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_EQ(OatFileAssistant::kDex2OatNeeded, oat_file_assistant.GetDexOptNeeded()); 93566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 93666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.IsInBootClassPath()); 93766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileExists()); 93866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileIsOutOfDate()); 93966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsUpToDate()); 94066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileExists()); 94166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileIsOutOfDate()); 94266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsUpToDate()); 94366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 94466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 94566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// A task to generate a dex location. Used by the RaceToGenerate test. 94666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhlerclass RaceGenerateTask : public Task { 94766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler public: 94866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler explicit RaceGenerateTask(const std::string& dex_location, const std::string& oat_location) 94966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler : dex_location_(dex_location), oat_location_(oat_location), 95066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler loaded_oat_file_(nullptr) 95166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler {} 95266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 95366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler void Run(Thread* self) { 95466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler UNUSED(self); 95566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 95666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Load the dex files, and save a pointer to the loaded oat file, so that 95766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // we can verify only one oat file was loaded for the dex location. 95866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ClassLinker* linker = Runtime::Current()->GetClassLinker(); 95966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::unique_ptr<const DexFile>> dex_files; 96066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::string> error_msgs; 96166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler dex_files = linker->OpenDexFilesFromOat(dex_location_.c_str(), oat_location_.c_str(), &error_msgs); 96266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler CHECK(!dex_files.empty()) << Join(error_msgs, '\n'); 96307b3c2351bb527ea91c084dc19434600af9ae66bRichard Uhler CHECK(dex_files[0]->GetOatDexFile() != nullptr) << dex_files[0]->GetLocation(); 96407b3c2351bb527ea91c084dc19434600af9ae66bRichard Uhler loaded_oat_file_ = dex_files[0]->GetOatDexFile()->GetOatFile(); 96566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 96666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 96766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler const OatFile* GetLoadedOatFile() const { 96866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler return loaded_oat_file_; 96966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 97066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 97166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler private: 97266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location_; 97366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string oat_location_; 97466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler const OatFile* loaded_oat_file_; 97566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler}; 97666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 97766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Test the case where multiple processes race to generate an oat file. 97866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// This simulates multiple processes using multiple threads. 97966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// 98066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// We want only one Oat file to be loaded when there is a race to load, to 98166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// avoid using up the virtual memory address space. 98266d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, RaceToGenerate) { 98366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/RaceToGenerate.jar"; 9846343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler std::string oat_location = GetOdexDir() + "/RaceToGenerate.oat"; 98566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 98666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // We use the lib core dex file, because it's large, and hopefully should 98766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // take a while to generate. 98866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetLibCoreDexFileName(), dex_location); 98966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 99066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler const int kNumThreads = 32; 99166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Thread* self = Thread::Current(); 99266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ThreadPool thread_pool("Oat file assistant test thread pool", kNumThreads); 99366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::unique_ptr<RaceGenerateTask>> tasks; 99466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler for (int i = 0; i < kNumThreads; i++) { 99566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::unique_ptr<RaceGenerateTask> task(new RaceGenerateTask(dex_location, oat_location)); 99666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler thread_pool.AddTask(self, task.get()); 99766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler tasks.push_back(std::move(task)); 99866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 99966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler thread_pool.StartWorkers(self); 100066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler thread_pool.Wait(self, true, false); 100166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 100266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Verify every task got the same pointer. 100366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler const OatFile* expected = tasks[0]->GetLoadedOatFile(); 100466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler for (auto& task : tasks) { 100566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(expected, task->GetLoadedOatFile()); 100666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 100766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 100866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 100966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: We have a DEX file and an ODEX file, no OAT file, and dex2oat is 101066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// disabled. 101166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Expect: We should load the odex file non-executable. 101266d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantNoDex2OatTest, LoadDexOdexNoOat) { 101366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/LoadDexOdexNoOat.jar"; 10146343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler std::string odex_location = GetOdexDir() + "/LoadDexOdexNoOat.odex"; 101566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 101666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Create the dex and odex files 101766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc1(), dex_location); 101866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler GenerateOdexForTest(dex_location, odex_location); 101966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 102066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Load the oat using an executable oat file assistant. 102166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, true); 102266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 102366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::unique_ptr<OatFile> oat_file = oat_file_assistant.GetBestOatFile(); 102466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(oat_file.get() != nullptr); 102566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file->IsExecutable()); 102666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::unique_ptr<const DexFile>> dex_files; 102766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler dex_files = oat_file_assistant.LoadDexFiles(*oat_file, dex_location.c_str()); 102866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(1u, dex_files.size()); 102966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 103066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 103166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: We have a MultiDEX file and an ODEX file, no OAT file, and dex2oat is 103266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// disabled. 103366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Expect: We should load the odex file non-executable. 103466d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantNoDex2OatTest, LoadMultiDexOdexNoOat) { 103566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/LoadMultiDexOdexNoOat.jar"; 10366343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler std::string odex_location = GetOdexDir() + "/LoadMultiDexOdexNoOat.odex"; 103766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 103866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Create the dex and odex files 103966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetMultiDexSrc1(), dex_location); 104066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler GenerateOdexForTest(dex_location, odex_location); 104166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 104266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Load the oat using an executable oat file assistant. 104366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, true); 104466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 104566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::unique_ptr<OatFile> oat_file = oat_file_assistant.GetBestOatFile(); 104666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(oat_file.get() != nullptr); 104766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file->IsExecutable()); 104866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::unique_ptr<const DexFile>> dex_files; 104966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler dex_files = oat_file_assistant.LoadDexFiles(*oat_file, dex_location.c_str()); 105066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(2u, dex_files.size()); 105166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 105266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 105366d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST(OatFileAssistantUtilsTest, DexFilenameToOdexFilename) { 105466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string error_msg; 105566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string odex_file; 105666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 105766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(OatFileAssistant::DexFilenameToOdexFilename( 105866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler "/foo/bar/baz.jar", kArm, &odex_file, &error_msg)) << error_msg; 10596343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler EXPECT_EQ("/foo/bar/oat/arm/baz.odex", odex_file); 106066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 106166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(OatFileAssistant::DexFilenameToOdexFilename( 106266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler "/foo/bar/baz.funnyext", kArm, &odex_file, &error_msg)) << error_msg; 10636343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler EXPECT_EQ("/foo/bar/oat/arm/baz.odex", odex_file); 106466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 106566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(OatFileAssistant::DexFilenameToOdexFilename( 106666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler "nopath.jar", kArm, &odex_file, &error_msg)); 106766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(OatFileAssistant::DexFilenameToOdexFilename( 106866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler "/foo/bar/baz_noext", kArm, &odex_file, &error_msg)); 106966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 107066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 107123cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler// Verify the dexopt status values from dalvik.system.DexFile 107223cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler// match the OatFileAssistant::DexOptStatus values. 107323cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard UhlerTEST_F(OatFileAssistantTest, DexOptStatusValues) { 107423cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler ScopedObjectAccess soa(Thread::Current()); 107523cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler StackHandleScope<1> hs(soa.Self()); 107623cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler ClassLinker* linker = Runtime::Current()->GetClassLinker(); 107723cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler Handle<mirror::Class> dexfile( 107823cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler hs.NewHandle(linker->FindSystemClass(soa.Self(), "Ldalvik/system/DexFile;"))); 107923cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler ASSERT_FALSE(dexfile.Get() == nullptr); 108023cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler linker->EnsureInitialized(soa.Self(), dexfile, true, true); 108123cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler 1082c785344b87221f5e4e6473e5b762e4e61fe65dcfMathieu Chartier ArtField* no_dexopt_needed = mirror::Class::FindStaticField( 108323cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler soa.Self(), dexfile, "NO_DEXOPT_NEEDED", "I"); 108423cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler ASSERT_FALSE(no_dexopt_needed == nullptr); 108523cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler EXPECT_EQ(no_dexopt_needed->GetTypeAsPrimitiveType(), Primitive::kPrimInt); 108623cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler EXPECT_EQ(OatFileAssistant::kNoDexOptNeeded, no_dexopt_needed->GetInt(dexfile.Get())); 108723cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler 1088c785344b87221f5e4e6473e5b762e4e61fe65dcfMathieu Chartier ArtField* dex2oat_needed = mirror::Class::FindStaticField( 108923cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler soa.Self(), dexfile, "DEX2OAT_NEEDED", "I"); 109023cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler ASSERT_FALSE(dex2oat_needed == nullptr); 109123cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler EXPECT_EQ(dex2oat_needed->GetTypeAsPrimitiveType(), Primitive::kPrimInt); 109223cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler EXPECT_EQ(OatFileAssistant::kDex2OatNeeded, dex2oat_needed->GetInt(dexfile.Get())); 109323cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler 1094c785344b87221f5e4e6473e5b762e4e61fe65dcfMathieu Chartier ArtField* patchoat_needed = mirror::Class::FindStaticField( 109523cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler soa.Self(), dexfile, "PATCHOAT_NEEDED", "I"); 109623cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler ASSERT_FALSE(patchoat_needed == nullptr); 109723cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler EXPECT_EQ(patchoat_needed->GetTypeAsPrimitiveType(), Primitive::kPrimInt); 109823cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler EXPECT_EQ(OatFileAssistant::kPatchOatNeeded, patchoat_needed->GetInt(dexfile.Get())); 109923cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler 1100c785344b87221f5e4e6473e5b762e4e61fe65dcfMathieu Chartier ArtField* self_patchoat_needed = mirror::Class::FindStaticField( 110123cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler soa.Self(), dexfile, "SELF_PATCHOAT_NEEDED", "I"); 110223cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler ASSERT_FALSE(self_patchoat_needed == nullptr); 110323cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler EXPECT_EQ(self_patchoat_needed->GetTypeAsPrimitiveType(), Primitive::kPrimInt); 110423cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler EXPECT_EQ(OatFileAssistant::kSelfPatchOatNeeded, self_patchoat_needed->GetInt(dexfile.Get())); 110523cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler} 110666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 110766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// TODO: More Tests: 110866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// * Test class linker falls back to unquickened dex for DexNoOat 110966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// * Test class linker falls back to unquickened dex for MultiDexNoOat 111066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// * Test using secondary isa 111166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// * Test with profiling info? 111266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// * Test for status of oat while oat is being generated (how?) 111366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// * Test case where 32 and 64 bit boot class paths differ, 111466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// and we ask IsInBootClassPath for a class in exactly one of the 32 or 111566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// 64 bit boot class paths. 111666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// * Test unexpected scenarios (?): 111766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// - Dex is stripped, don't have odex. 111866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// - Oat file corrupted after status check, before reload unexecutable 111966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// because it's unrelocated and no dex2oat 112066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 112166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} // namespace art 1122