oat_file_assistant_test.cc revision 9b994ea841eaaefbdda652251894a74db9cefcc8
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 4166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhlerclass OatFileAssistantTest : public CommonRuntimeTest { 4266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler public: 4366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler virtual void SetUp() { 4466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ReserveImageSpace(); 4566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler CommonRuntimeTest::SetUp(); 4666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 4766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Create a scratch directory to work from. 4866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler scratch_dir_ = android_data_ + "/OatFileAssistantTest"; 4966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_EQ(0, mkdir(scratch_dir_.c_str(), 0700)); 5066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 516343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler // Create a subdirectory in scratch for odex files. 526343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler odex_oat_dir_ = scratch_dir_ + "/oat"; 536343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler ASSERT_EQ(0, mkdir(odex_oat_dir_.c_str(), 0700)); 546343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler 556343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler odex_dir_ = odex_oat_dir_ + "/" + std::string(GetInstructionSetString(kRuntimeISA)); 566343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler ASSERT_EQ(0, mkdir(odex_dir_.c_str(), 0700)); 576343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler 5866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 5966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Verify the environment is as we expect 6066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler uint32_t checksum; 6166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string error_msg; 6266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(OS::FileExists(GetImageFile().c_str())) 6366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler << "Expected pre-compiled boot image to be at: " << GetImageFile(); 6466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(OS::FileExists(GetDexSrc1().c_str())) 6566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler << "Expected dex file to be at: " << GetDexSrc1(); 6666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(OS::FileExists(GetStrippedDexSrc1().c_str())) 6766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler << "Expected stripped dex file to be at: " << GetStrippedDexSrc1(); 6866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_FALSE(DexFile::GetChecksum(GetStrippedDexSrc1().c_str(), &checksum, &error_msg)) 6966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler << "Expected stripped dex file to be stripped: " << GetStrippedDexSrc1(); 7066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(OS::FileExists(GetDexSrc2().c_str())) 7166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler << "Expected dex file to be at: " << GetDexSrc2(); 7267ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler 7367ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler // GetMultiDexSrc2 should have the same primary dex checksum as 7467ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler // GetMultiDexSrc1, but a different secondary dex checksum. 7567ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler std::vector<std::unique_ptr<const DexFile>> multi1; 7667ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler ASSERT_TRUE(DexFile::Open(GetMultiDexSrc1().c_str(), 7767ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler GetMultiDexSrc1().c_str(), &error_msg, &multi1)) << error_msg; 7867ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler ASSERT_GT(multi1.size(), 1u); 7967ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler 8067ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler std::vector<std::unique_ptr<const DexFile>> multi2; 8167ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler ASSERT_TRUE(DexFile::Open(GetMultiDexSrc2().c_str(), 8267ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler GetMultiDexSrc2().c_str(), &error_msg, &multi2)) << error_msg; 8367ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler ASSERT_GT(multi2.size(), 1u); 8467ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler 8567ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler ASSERT_EQ(multi1[0]->GetLocationChecksum(), multi2[0]->GetLocationChecksum()); 8667ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler ASSERT_NE(multi1[1]->GetLocationChecksum(), multi2[1]->GetLocationChecksum()); 8766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 8866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 8966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler virtual void SetUpRuntimeOptions(RuntimeOptions* options) { 90892fc96694b51ac7a896dacd75af5b235f955825Richard Uhler // options->push_back(std::make_pair("-verbose:oat", nullptr)); 9166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 9266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Set up the image location. 9366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler options->push_back(std::make_pair("-Ximage:" + GetImageLocation(), 9466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler nullptr)); 9566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Make sure compilercallbacks are not set so that relocation will be 9666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // enabled. 97bb9c6b1c55e9e2308b4f5892a398a8837231fdbdAndreas Gampe callbacks_.reset(); 9866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 9966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 10066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler virtual void PreRuntimeCreate() { 10166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler UnreserveImageSpace(); 10266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 10366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 10466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler virtual void PostRuntimeCreate() { 10566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ReserveImageSpace(); 10666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 10766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 10866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler virtual void TearDown() { 1096343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler ClearDirectory(odex_dir_.c_str()); 1106343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler ASSERT_EQ(0, rmdir(odex_dir_.c_str())); 1116343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler 1126343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler ClearDirectory(odex_oat_dir_.c_str()); 1136343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler ASSERT_EQ(0, rmdir(odex_oat_dir_.c_str())); 11466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 11566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ClearDirectory(scratch_dir_.c_str()); 11666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_EQ(0, rmdir(scratch_dir_.c_str())); 11766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 11866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler CommonRuntimeTest::TearDown(); 11966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 12066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 12166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler void Copy(std::string src, std::string dst) { 12266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::ifstream src_stream(src, std::ios::binary); 12366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::ofstream dst_stream(dst, std::ios::binary); 12466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 12566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler dst_stream << src_stream.rdbuf(); 12666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 12766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 12866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Returns the directory where the pre-compiled core.art can be found. 12966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // TODO: We should factor out this into common tests somewhere rather than 13066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // re-hardcoding it here (This was copied originally from the elf writer 13166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // test). 13266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string GetImageDirectory() { 13366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler if (IsHost()) { 13466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler const char* host_dir = getenv("ANDROID_HOST_OUT"); 1352cebb24bfc3247d3e9be138a3350106737455918Mathieu Chartier CHECK(host_dir != nullptr); 13666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler return std::string(host_dir) + "/framework"; 13766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } else { 13866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler return std::string("/data/art-test"); 13966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 14066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 14166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 14266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string GetImageLocation() { 14366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler return GetImageDirectory() + "/core.art"; 14466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 14566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 14666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string GetImageFile() { 14766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler return GetImageDirectory() + "/" + GetInstructionSetString(kRuntimeISA) 14866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler + "/core.art"; 14966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 15066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 15166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string GetDexSrc1() { 15266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler return GetTestDexFileName("Main"); 15366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 15466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 15566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Returns the path to a dex file equivalent to GetDexSrc1, but with the dex 15666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // file stripped. 15766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string GetStrippedDexSrc1() { 15866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler return GetTestDexFileName("MainStripped"); 15966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 16066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 16166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string GetMultiDexSrc1() { 16266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler return GetTestDexFileName("MultiDex"); 16366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 16466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 16567ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler // Returns the path to a multidex file equivalent to GetMultiDexSrc2, but 16667ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler // with the contents of the secondary dex file changed. 16767ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler std::string GetMultiDexSrc2() { 16867ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler return GetTestDexFileName("MultiDexModifiedSecondary"); 16967ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler } 17067ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler 17166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string GetDexSrc2() { 17266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler return GetTestDexFileName("Nested"); 17366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 17466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 17566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Scratch directory, for dex and odex files (oat files will go in the 17666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // dalvik cache). 17766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string GetScratchDir() { 17866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler return scratch_dir_; 17966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 18066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 1816343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler // Odex directory is the subdirectory in the scratch directory where odex 18266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // files should be located. 1836343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler std::string GetOdexDir() { 1846343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler return odex_dir_; 18566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 18666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 18766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Generate an odex file for the purposes of test. 18866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // If pic is true, generates a PIC odex. 18966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler void GenerateOdexForTest(const std::string& dex_location, 19066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler const std::string& odex_location, 19166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler bool pic = false) { 19266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // For this operation, we temporarily redirect the dalvik cache so dex2oat 19366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // doesn't find the relocated image file. 19466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string android_data_tmp = GetScratchDir() + "AndroidDataTmp"; 19566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler setenv("ANDROID_DATA", android_data_tmp.c_str(), 1); 19666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::string> args; 19766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler args.push_back("--dex-file=" + dex_location); 19866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler args.push_back("--oat-file=" + odex_location); 19966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler if (pic) { 20066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler args.push_back("--compile-pic"); 20166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } else { 20266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler args.push_back("--include-patch-information"); 20305dd8a63e213d2bf025c97b9cd04eee354d0e5b4Richard Uhler 20405dd8a63e213d2bf025c97b9cd04eee354d0e5b4Richard Uhler // We need to use the quick compiler to generate non-PIC code, because 20505dd8a63e213d2bf025c97b9cd04eee354d0e5b4Richard Uhler // the optimizing compiler always generates PIC. 20605dd8a63e213d2bf025c97b9cd04eee354d0e5b4Richard Uhler args.push_back("--compiler-backend=Quick"); 20766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 20866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler args.push_back("--runtime-arg"); 20966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler args.push_back("-Xnorelocate"); 21066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string error_msg; 21166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(OatFileAssistant::Dex2Oat(args, &error_msg)) << error_msg; 21266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler setenv("ANDROID_DATA", android_data_.c_str(), 1); 21366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 21466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 21566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler void GeneratePicOdexForTest(const std::string& dex_location, 21666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler const std::string& odex_location) { 21766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler GenerateOdexForTest(dex_location, odex_location, true); 21866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 21966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 22066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler private: 22166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Reserve memory around where the image will be loaded so other memory 22266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // won't conflict when it comes time to load the image. 22366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // This can be called with an already loaded image to reserve the space 22466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // around it. 22566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler void ReserveImageSpace() { 22666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler MemMap::Init(); 22766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 22866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Ensure a chunk of memory is reserved for the image space. 22966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler uintptr_t reservation_start = ART_BASE_ADDRESS + ART_BASE_ADDRESS_MIN_DELTA; 23066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler uintptr_t reservation_end = ART_BASE_ADDRESS + ART_BASE_ADDRESS_MAX_DELTA 2313dbf23412481c4da51f0ebe32bec5d300c36834bHiroshi Yamauchi // Include the main space that has to come right after the 2323dbf23412481c4da51f0ebe32bec5d300c36834bHiroshi Yamauchi // image in case of the GSS collector. 2333dbf23412481c4da51f0ebe32bec5d300c36834bHiroshi Yamauchi + 384 * MB; 23466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 23566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::unique_ptr<BacktraceMap> map(BacktraceMap::Create(getpid(), true)); 23666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(map.get() != nullptr) << "Failed to build process map"; 23766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler for (BacktraceMap::const_iterator it = map->begin(); 23866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler reservation_start < reservation_end && it != map->end(); ++it) { 2393efe979d4292330c8fab1708a4361e58681a88cbRichard Uhler ReserveImageSpaceChunk(reservation_start, std::min(it->start, reservation_end)); 2403efe979d4292330c8fab1708a4361e58681a88cbRichard Uhler reservation_start = std::max(reservation_start, it->end); 2413efe979d4292330c8fab1708a4361e58681a88cbRichard Uhler } 2423efe979d4292330c8fab1708a4361e58681a88cbRichard Uhler ReserveImageSpaceChunk(reservation_start, reservation_end); 2433efe979d4292330c8fab1708a4361e58681a88cbRichard Uhler } 24466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 2453efe979d4292330c8fab1708a4361e58681a88cbRichard Uhler // Reserve a chunk of memory for the image space in the given range. 2463efe979d4292330c8fab1708a4361e58681a88cbRichard Uhler // Only has effect for chunks with a positive number of bytes. 2473efe979d4292330c8fab1708a4361e58681a88cbRichard Uhler void ReserveImageSpaceChunk(uintptr_t start, uintptr_t end) { 2483efe979d4292330c8fab1708a4361e58681a88cbRichard Uhler if (start < end) { 2493efe979d4292330c8fab1708a4361e58681a88cbRichard Uhler std::string error_msg; 25066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler image_reservation_.push_back(std::unique_ptr<MemMap>( 25166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler MemMap::MapAnonymous("image reservation", 2523efe979d4292330c8fab1708a4361e58681a88cbRichard Uhler reinterpret_cast<uint8_t*>(start), end - start, 25366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler PROT_NONE, false, false, &error_msg))); 25466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(image_reservation_.back().get() != nullptr) << error_msg; 25566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler LOG(INFO) << "Reserved space for image " << 25666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler reinterpret_cast<void*>(image_reservation_.back()->Begin()) << "-" << 25766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler reinterpret_cast<void*>(image_reservation_.back()->End()); 25866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 25966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 26066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 26166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 26266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Unreserve any memory reserved by ReserveImageSpace. This should be called 26366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // before the image is loaded. 26466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler void UnreserveImageSpace() { 26566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler image_reservation_.clear(); 26666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 26766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 26866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string scratch_dir_; 2696343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler std::string odex_oat_dir_; 2706343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler std::string odex_dir_; 27166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::unique_ptr<MemMap>> image_reservation_; 27266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler}; 27366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 27466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhlerclass OatFileAssistantNoDex2OatTest : public OatFileAssistantTest { 27566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler public: 27666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler virtual void SetUpRuntimeOptions(RuntimeOptions* options) { 27766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistantTest::SetUpRuntimeOptions(options); 27866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler options->push_back(std::make_pair("-Xnodex2oat", nullptr)); 27966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 28066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler}; 28166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 28266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Generate an oat file for the purposes of test, as opposed to testing 28366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// generation of oat files. 28466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhlerstatic void GenerateOatForTest(const char* dex_location) { 28566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location, kRuntimeISA, false); 28666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 28766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string error_msg; 28866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(oat_file_assistant.GenerateOatFile(&error_msg)) << error_msg; 28966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 29066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 29166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: We have a DEX file, but no OAT file for it. 29295abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler// Expect: The status is kDex2OatNeeded. 29366d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, DexNoOat) { 29466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/DexNoOat.jar"; 29566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc1(), dex_location); 29666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 29766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, false); 29866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 29995abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_EQ(OatFileAssistant::kDex2OatNeeded, oat_file_assistant.GetDexOptNeeded()); 30066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 30166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.IsInBootClassPath()); 30266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileExists()); 30366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileIsOutOfDate()); 30466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileNeedsRelocation()); 30566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsUpToDate()); 30695abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_EQ(OatFileAssistant::kOatOutOfDate, oat_file_assistant.OdexFileStatus()); 30766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileExists()); 30866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileIsOutOfDate()); 30966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileNeedsRelocation()); 31066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsUpToDate()); 31195abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_EQ(OatFileAssistant::kOatOutOfDate, oat_file_assistant.OatFileStatus()); 3129b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_TRUE(oat_file_assistant.HasOriginalDexFiles()); 31366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 31466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 31566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: We have no DEX file and no OAT file. 3169b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler// Expect: Status is kNoDexOptNeeded. Loading should fail, but not crash. 31766d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, NoDexNoOat) { 31866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/NoDexNoOat.jar"; 31966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 32066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, true); 32166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 3229b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_EQ(OatFileAssistant::kNoDexOptNeeded, oat_file_assistant.GetDexOptNeeded()); 3239b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_FALSE(oat_file_assistant.HasOriginalDexFiles()); 3249b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler 3259b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler // Trying to make the oat file up to date should not fail or crash. 3269b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler std::string error_msg; 3279b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_TRUE(oat_file_assistant.MakeUpToDate(&error_msg)); 3289b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler 3299b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler // Trying to get the best oat file should fail, but not crash. 33066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::unique_ptr<OatFile> oat_file = oat_file_assistant.GetBestOatFile(); 33166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(nullptr, oat_file.get()); 33266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 33366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 33466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: We have a DEX file and up-to-date OAT file for it. 33595abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler// Expect: The status is kNoDexOptNeeded. 33666d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, OatUpToDate) { 33766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/OatUpToDate.jar"; 33866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc1(), dex_location); 33966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler GenerateOatForTest(dex_location.c_str()); 34066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 34166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, false); 34266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 34395abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_EQ(OatFileAssistant::kNoDexOptNeeded, oat_file_assistant.GetDexOptNeeded()); 34466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.IsInBootClassPath()); 34566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileExists()); 34666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileIsOutOfDate()); 34766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsUpToDate()); 34866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileExists()); 34966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsOutOfDate()); 35066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileNeedsRelocation()); 35166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileIsUpToDate()); 35295abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_EQ(OatFileAssistant::kOatUpToDate, oat_file_assistant.OatFileStatus()); 3539b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_TRUE(oat_file_assistant.HasOriginalDexFiles()); 35466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 35566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 35666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: We have a MultiDEX file and up-to-date OAT file for it. 35795abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler// Expect: The status is kNoDexOptNeeded and we load all dex files. 35866d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, MultiDexOatUpToDate) { 35966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/MultiDexOatUpToDate.jar"; 36066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetMultiDexSrc1(), dex_location); 36166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler GenerateOatForTest(dex_location.c_str()); 36266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 363e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, true); 36495abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_EQ(OatFileAssistant::kNoDexOptNeeded, oat_file_assistant.GetDexOptNeeded()); 3659b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_TRUE(oat_file_assistant.HasOriginalDexFiles()); 36695abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler 36795abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler // Verify we can load both dex files. 368e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler std::unique_ptr<OatFile> oat_file = oat_file_assistant.GetBestOatFile(); 369e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler ASSERT_TRUE(oat_file.get() != nullptr); 370e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler EXPECT_TRUE(oat_file->IsExecutable()); 371e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler std::vector<std::unique_ptr<const DexFile>> dex_files; 372e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler dex_files = oat_file_assistant.LoadDexFiles(*oat_file, dex_location.c_str()); 373e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler EXPECT_EQ(2u, dex_files.size()); 374e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler} 375e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler 37667ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler// Case: We have a MultiDEX file where the secondary dex file is out of date. 37767ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler// Expect: The status is kDex2OatNeeded. 37867ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard UhlerTEST_F(OatFileAssistantTest, MultiDexSecondaryOutOfDate) { 37967ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler std::string dex_location = GetScratchDir() + "/MultiDexSecondaryOutOfDate.jar"; 38067ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler 38167ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler // Compile code for GetMultiDexSrc1. 38267ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler Copy(GetMultiDexSrc1(), dex_location); 38367ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler GenerateOatForTest(dex_location.c_str()); 38467ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler 38567ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler // Now overwrite the dex file with GetMultiDexSrc2 so the secondary checksum 38667ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler // is out of date. 38767ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler Copy(GetMultiDexSrc2(), dex_location); 38867ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler 38967ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, true); 39067ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler EXPECT_EQ(OatFileAssistant::kDex2OatNeeded, oat_file_assistant.GetDexOptNeeded()); 3919b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_TRUE(oat_file_assistant.HasOriginalDexFiles()); 39267ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler} 39367ff7d1fd7bcaf4b6b73ecdab6011c8636562b58Richard Uhler 394e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler// Case: We have a MultiDEX file and up-to-date OAT file for it with relative 395e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler// encoded dex locations. 39695abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler// Expect: The oat file status is kNoDexOptNeeded. 397e5fed03772144595c0904faf3d6974cc55214c8cRichard UhlerTEST_F(OatFileAssistantTest, RelativeEncodedDexLocation) { 398e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler std::string dex_location = GetScratchDir() + "/RelativeEncodedDexLocation.jar"; 3996343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler std::string oat_location = GetOdexDir() + "/RelativeEncodedDexLocation.oat"; 400e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler 401e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler // Create the dex file 402e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler Copy(GetMultiDexSrc1(), dex_location); 403e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler 404e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler // Create the oat file with relative encoded dex location. 405e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler std::vector<std::string> args; 406e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler args.push_back("--dex-file=" + dex_location); 407e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler args.push_back("--dex-location=" + std::string("RelativeEncodedDexLocation.jar")); 408e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler args.push_back("--oat-file=" + oat_location); 409e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler 410e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler std::string error_msg; 411e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler ASSERT_TRUE(OatFileAssistant::Dex2Oat(args, &error_msg)) << error_msg; 412e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler 413e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler // Verify we can load both dex files. 414e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), 415e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler oat_location.c_str(), 416e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler kRuntimeISA, true); 417e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler std::unique_ptr<OatFile> oat_file = oat_file_assistant.GetBestOatFile(); 41866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(oat_file.get() != nullptr); 41966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file->IsExecutable()); 42066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::unique_ptr<const DexFile>> dex_files; 421e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler dex_files = oat_file_assistant.LoadDexFiles(*oat_file, dex_location.c_str()); 42266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(2u, dex_files.size()); 42366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 42466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 42595abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler// Case: We have a DEX file and out-of-date OAT file. 42695abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler// Expect: The status is kDex2OatNeeded. 42766d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, OatOutOfDate) { 42866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/OatOutOfDate.jar"; 42966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 43066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // We create a dex, generate an oat for it, then overwrite the dex with a 43166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // different dex to make the oat out of date. 43266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc1(), dex_location); 43366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler GenerateOatForTest(dex_location.c_str()); 43466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc2(), dex_location); 43566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 43666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, false); 43795abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_EQ(OatFileAssistant::kDex2OatNeeded, oat_file_assistant.GetDexOptNeeded()); 43866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 43966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.IsInBootClassPath()); 44066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileExists()); 44166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileIsOutOfDate()); 44266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsUpToDate()); 44366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileExists()); 44466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileIsOutOfDate()); 44566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsUpToDate()); 4469b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_TRUE(oat_file_assistant.HasOriginalDexFiles()); 44766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 44866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 44966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: We have a DEX file and an ODEX file, but no OAT file. 45095abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler// Expect: The status is kPatchOatNeeded. 45166d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, DexOdexNoOat) { 45266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/DexOdexNoOat.jar"; 4536343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler std::string odex_location = GetOdexDir() + "/DexOdexNoOat.odex"; 45466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 45566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Create the dex and odex files 45666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc1(), dex_location); 45766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler GenerateOdexForTest(dex_location, odex_location); 45866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 45966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Verify the status. 46066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, false); 46166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 46295abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_EQ(OatFileAssistant::kPatchOatNeeded, oat_file_assistant.GetDexOptNeeded()); 46366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 46466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.IsInBootClassPath()); 46566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileExists()); 46666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsOutOfDate()); 46766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsUpToDate()); 46866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileNeedsRelocation()); 46966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileExists()); 47066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileIsOutOfDate()); 47166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsUpToDate()); 4729b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_TRUE(oat_file_assistant.HasOriginalDexFiles()); 47366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 47466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 47566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: We have a stripped DEX file and an ODEX file, but no OAT file. 47695abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler// Expect: The status is kPatchOatNeeded 47766d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, StrippedDexOdexNoOat) { 47866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/StrippedDexOdexNoOat.jar"; 4796343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler std::string odex_location = GetOdexDir() + "/StrippedDexOdexNoOat.odex"; 48066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 48166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Create the dex and odex files 48266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc1(), dex_location); 48366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler GenerateOdexForTest(dex_location, odex_location); 48466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 48566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Strip the dex file 48666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetStrippedDexSrc1(), dex_location); 48766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 48866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Verify the status. 48966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, true); 49066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 49195abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_EQ(OatFileAssistant::kPatchOatNeeded, oat_file_assistant.GetDexOptNeeded()); 49266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 49366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.IsInBootClassPath()); 49466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileExists()); 49566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsOutOfDate()); 49666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsUpToDate()); 49766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileExists()); 49866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileIsOutOfDate()); 49966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsUpToDate()); 5009b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_FALSE(oat_file_assistant.HasOriginalDexFiles()); 50166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 50266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Make the oat file up to date. 50366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string error_msg; 50466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(oat_file_assistant.MakeUpToDate(&error_msg)) << error_msg; 50566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 50695abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_EQ(OatFileAssistant::kNoDexOptNeeded, oat_file_assistant.GetDexOptNeeded()); 50766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 50866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.IsInBootClassPath()); 50966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileExists()); 51066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsOutOfDate()); 51166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsUpToDate()); 51266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileExists()); 51366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsOutOfDate()); 51466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileIsUpToDate()); 5159b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_FALSE(oat_file_assistant.HasOriginalDexFiles()); 51666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 51766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Verify we can load the dex files from it. 51866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::unique_ptr<OatFile> oat_file = oat_file_assistant.GetBestOatFile(); 51966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(oat_file.get() != nullptr); 52066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file->IsExecutable()); 52166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::unique_ptr<const DexFile>> dex_files; 52266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler dex_files = oat_file_assistant.LoadDexFiles(*oat_file, dex_location.c_str()); 52366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(1u, dex_files.size()); 52466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 52566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 52695abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler// Case: We have a stripped DEX file, an ODEX file, and an out-of-date OAT file. 52795abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler// Expect: The status is kPatchOatNeeded. 52866d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, StrippedDexOdexOat) { 52966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/StrippedDexOdexOat.jar"; 5306343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler std::string odex_location = GetOdexDir() + "/StrippedDexOdexOat.odex"; 53166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 53266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Create the oat file from a different dex file so it looks out of date. 53366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc2(), dex_location); 53466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler GenerateOatForTest(dex_location.c_str()); 53566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 53666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Create the odex file 53766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc1(), dex_location); 53866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler GenerateOdexForTest(dex_location, odex_location); 53966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 54066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Strip the dex file. 54166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetStrippedDexSrc1(), dex_location); 54266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 54366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Verify the status. 54466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, true); 54566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 54695abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_EQ(OatFileAssistant::kPatchOatNeeded, oat_file_assistant.GetDexOptNeeded()); 54766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 54866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.IsInBootClassPath()); 54966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileExists()); 55066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsOutOfDate()); 55166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileNeedsRelocation()); 55266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsUpToDate()); 55366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileExists()); 55466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileIsOutOfDate()); 55566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsUpToDate()); 5569b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_FALSE(oat_file_assistant.HasOriginalDexFiles()); 55766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 55866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Make the oat file up to date. 55966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string error_msg; 56066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(oat_file_assistant.MakeUpToDate(&error_msg)) << error_msg; 56166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 56295abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_EQ(OatFileAssistant::kNoDexOptNeeded, oat_file_assistant.GetDexOptNeeded()); 56366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 56466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.IsInBootClassPath()); 56566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileExists()); 56666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsOutOfDate()); 56766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileNeedsRelocation()); 56866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsUpToDate()); 56966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileExists()); 57066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsOutOfDate()); 57166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileNeedsRelocation()); 57266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileIsUpToDate()); 5739b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_FALSE(oat_file_assistant.HasOriginalDexFiles()); 57466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 57566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Verify we can load the dex files from it. 57666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::unique_ptr<OatFile> oat_file = oat_file_assistant.GetBestOatFile(); 57766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(oat_file.get() != nullptr); 57866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file->IsExecutable()); 57966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::unique_ptr<const DexFile>> dex_files; 58066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler dex_files = oat_file_assistant.LoadDexFiles(*oat_file, dex_location.c_str()); 58166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(1u, dex_files.size()); 58266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 58366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 5849b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler// Case: We have a stripped (or resource-only) DEX file, no ODEX file and no 5859b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler// OAT file. Expect: The status is kNoDexOptNeeded. 5869b994ea841eaaefbdda652251894a74db9cefcc8Richard UhlerTEST_F(OatFileAssistantTest, ResourceOnlyDex) { 5879b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler std::string dex_location = GetScratchDir() + "/ResourceOnlyDex.jar"; 5889b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler 5899b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler Copy(GetStrippedDexSrc1(), dex_location); 5909b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler 5919b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler // Verify the status. 5929b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, true); 5939b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler 5949b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_EQ(OatFileAssistant::kNoDexOptNeeded, oat_file_assistant.GetDexOptNeeded()); 5959b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler 5969b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_FALSE(oat_file_assistant.IsInBootClassPath()); 5979b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileExists()); 5989b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileIsOutOfDate()); 5999b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileNeedsRelocation()); 6009b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsUpToDate()); 6019b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_FALSE(oat_file_assistant.OatFileExists()); 6029b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_TRUE(oat_file_assistant.OatFileIsOutOfDate()); 6039b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsUpToDate()); 6049b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_FALSE(oat_file_assistant.HasOriginalDexFiles()); 6059b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler 6069b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler // Make the oat file up to date. This should have no effect. 6079b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler std::string error_msg; 6089b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_TRUE(oat_file_assistant.MakeUpToDate(&error_msg)) << error_msg; 6099b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler 6109b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_EQ(OatFileAssistant::kNoDexOptNeeded, oat_file_assistant.GetDexOptNeeded()); 6119b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler 6129b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_FALSE(oat_file_assistant.IsInBootClassPath()); 6139b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileExists()); 6149b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileIsOutOfDate()); 6159b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileNeedsRelocation()); 6169b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsUpToDate()); 6179b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_FALSE(oat_file_assistant.OatFileExists()); 6189b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_TRUE(oat_file_assistant.OatFileIsOutOfDate()); 6199b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsUpToDate()); 6209b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_FALSE(oat_file_assistant.HasOriginalDexFiles()); 6219b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler} 6229b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler 62395abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler// Case: We have a DEX file, no ODEX file and an OAT file that needs 62495abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler// relocation. 62595abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler// Expect: The status is kSelfPatchOatNeeded. 62695abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard UhlerTEST_F(OatFileAssistantTest, SelfRelocation) { 62795abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler std::string dex_location = GetScratchDir() + "/SelfRelocation.jar"; 62895abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler std::string oat_location = GetOdexDir() + "/SelfRelocation.oat"; 62995abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler 63095abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler // Create the dex and odex files 63195abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler Copy(GetDexSrc1(), dex_location); 63295abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler GenerateOdexForTest(dex_location, oat_location); 63395abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler 63495abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), 63595abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler oat_location.c_str(), kRuntimeISA, true); 63695abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler 63795abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_EQ(OatFileAssistant::kSelfPatchOatNeeded, oat_file_assistant.GetDexOptNeeded()); 63895abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler 63995abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_FALSE(oat_file_assistant.IsInBootClassPath()); 64095abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileExists()); 64195abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileIsOutOfDate()); 64295abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileNeedsRelocation()); 64395abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsUpToDate()); 64495abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_TRUE(oat_file_assistant.OatFileExists()); 64595abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_TRUE(oat_file_assistant.OatFileNeedsRelocation()); 64695abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsOutOfDate()); 64795abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsUpToDate()); 6489b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_TRUE(oat_file_assistant.HasOriginalDexFiles()); 64995abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler 65095abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler // Make the oat file up to date. 65195abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler std::string error_msg; 65295abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler ASSERT_TRUE(oat_file_assistant.MakeUpToDate(&error_msg)) << error_msg; 65395abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler 65495abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_EQ(OatFileAssistant::kNoDexOptNeeded, oat_file_assistant.GetDexOptNeeded()); 65595abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler 65695abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_FALSE(oat_file_assistant.IsInBootClassPath()); 65795abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileExists()); 65895abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileIsOutOfDate()); 65995abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileNeedsRelocation()); 66095abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsUpToDate()); 66195abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_TRUE(oat_file_assistant.OatFileExists()); 66295abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsOutOfDate()); 66395abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_FALSE(oat_file_assistant.OatFileNeedsRelocation()); 66495abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_TRUE(oat_file_assistant.OatFileIsUpToDate()); 6659b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_TRUE(oat_file_assistant.HasOriginalDexFiles()); 66695abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler 66795abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler std::unique_ptr<OatFile> oat_file = oat_file_assistant.GetBestOatFile(); 66895abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler ASSERT_TRUE(oat_file.get() != nullptr); 66995abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_TRUE(oat_file->IsExecutable()); 67095abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler std::vector<std::unique_ptr<const DexFile>> dex_files; 67195abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler dex_files = oat_file_assistant.LoadDexFiles(*oat_file, dex_location.c_str()); 67295abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_EQ(1u, dex_files.size()); 67395abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler} 67495abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler 67566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: We have a DEX file, an ODEX file and an OAT file, where the ODEX and 67666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// OAT files both have patch delta of 0. 67795abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler// Expect: It shouldn't crash, and status is kPatchOatNeeded. 67866d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, OdexOatOverlap) { 67966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/OdexOatOverlap.jar"; 6806343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler std::string odex_location = GetOdexDir() + "/OdexOatOverlap.odex"; 6816343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler std::string oat_location = GetOdexDir() + "/OdexOatOverlap.oat"; 68266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 68366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Create the dex and odex files 68466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc1(), dex_location); 68566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler GenerateOdexForTest(dex_location, odex_location); 68666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 68766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Create the oat file by copying the odex so they are located in the same 68866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // place in memory. 68966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(odex_location, oat_location); 69066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 69166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Verify things don't go bad. 69266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), 69366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler oat_location.c_str(), kRuntimeISA, true); 69466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 69595abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_EQ(OatFileAssistant::kPatchOatNeeded, oat_file_assistant.GetDexOptNeeded()); 69666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 69766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.IsInBootClassPath()); 69866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileExists()); 69966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsOutOfDate()); 70066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsUpToDate()); 70166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileExists()); 70266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsOutOfDate()); 70366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsUpToDate()); 7049b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_TRUE(oat_file_assistant.HasOriginalDexFiles()); 70566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 70666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Things aren't relocated, so it should fall back to interpreted. 70766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::unique_ptr<OatFile> oat_file = oat_file_assistant.GetBestOatFile(); 70866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(oat_file.get() != nullptr); 709f16d5727f9062379519043bc063a2c0527c59eb8Richard Uhler 71066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file->IsExecutable()); 71166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::unique_ptr<const DexFile>> dex_files; 71266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler dex_files = oat_file_assistant.LoadDexFiles(*oat_file, dex_location.c_str()); 71366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(1u, dex_files.size()); 714f16d5727f9062379519043bc063a2c0527c59eb8Richard Uhler 715f16d5727f9062379519043bc063a2c0527c59eb8Richard Uhler // Add some extra checks to help diagnose apparently flaky test failures. 716f16d5727f9062379519043bc063a2c0527c59eb8Richard Uhler Runtime* runtime = Runtime::Current(); 717f16d5727f9062379519043bc063a2c0527c59eb8Richard Uhler const gc::space::ImageSpace* image_space = runtime->GetHeap()->GetImageSpace(); 718f16d5727f9062379519043bc063a2c0527c59eb8Richard Uhler ASSERT_TRUE(image_space != nullptr); 719f16d5727f9062379519043bc063a2c0527c59eb8Richard Uhler const ImageHeader& image_header = image_space->GetImageHeader(); 720f16d5727f9062379519043bc063a2c0527c59eb8Richard Uhler const OatHeader& oat_header = oat_file->GetOatHeader(); 721f16d5727f9062379519043bc063a2c0527c59eb8Richard Uhler EXPECT_FALSE(oat_file->IsPic()); 722f16d5727f9062379519043bc063a2c0527c59eb8Richard Uhler EXPECT_EQ(image_header.GetOatChecksum(), oat_header.GetImageFileLocationOatChecksum()); 723f16d5727f9062379519043bc063a2c0527c59eb8Richard Uhler EXPECT_NE(reinterpret_cast<uintptr_t>(image_header.GetOatDataBegin()), 724f16d5727f9062379519043bc063a2c0527c59eb8Richard Uhler oat_header.GetImageFileLocationOatDataBegin()); 725f16d5727f9062379519043bc063a2c0527c59eb8Richard Uhler EXPECT_NE(image_header.GetPatchDelta(), oat_header.GetImagePatchDelta()); 72666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 72766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 72866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: We have a DEX file and a PIC ODEX file, but no OAT file. 72995abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler// Expect: The status is kNoDexOptNeeded, because PIC needs no relocation. 73066d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, DexPicOdexNoOat) { 73166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/DexPicOdexNoOat.jar"; 7326343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler std::string odex_location = GetOdexDir() + "/DexPicOdexNoOat.odex"; 73366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 73466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Create the dex and odex files 73566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc1(), dex_location); 73666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler GeneratePicOdexForTest(dex_location, odex_location); 73766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 73866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Verify the status. 73966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, false); 74066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 74195abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_EQ(OatFileAssistant::kNoDexOptNeeded, oat_file_assistant.GetDexOptNeeded()); 74266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 74366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.IsInBootClassPath()); 74466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileExists()); 74566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsOutOfDate()); 74666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileIsUpToDate()); 74766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileExists()); 74866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileIsOutOfDate()); 74966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsUpToDate()); 7509b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_TRUE(oat_file_assistant.HasOriginalDexFiles()); 75166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 75266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 75366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: We have a DEX file and up-to-date OAT file for it. 75466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Expect: We should load an executable dex file. 75566d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, LoadOatUpToDate) { 75666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/LoadOatUpToDate.jar"; 75766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 75866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc1(), dex_location); 75966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler GenerateOatForTest(dex_location.c_str()); 76066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 76166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Load the oat using an oat file assistant. 76266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, true); 76366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 76466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::unique_ptr<OatFile> oat_file = oat_file_assistant.GetBestOatFile(); 76566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(oat_file.get() != nullptr); 76666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file->IsExecutable()); 76766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::unique_ptr<const DexFile>> dex_files; 76866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler dex_files = oat_file_assistant.LoadDexFiles(*oat_file, dex_location.c_str()); 76966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(1u, dex_files.size()); 77066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 77166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 77266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: We have a DEX file and up-to-date OAT file for it. 77366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Expect: Loading non-executable should load the oat non-executable. 77466d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, LoadNoExecOatUpToDate) { 77566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/LoadNoExecOatUpToDate.jar"; 77666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 77766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc1(), dex_location); 77866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler GenerateOatForTest(dex_location.c_str()); 77966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 78066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Load the oat using an oat file assistant. 78166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, false); 78266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 78366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::unique_ptr<OatFile> oat_file = oat_file_assistant.GetBestOatFile(); 78466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(oat_file.get() != nullptr); 78566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file->IsExecutable()); 78666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::unique_ptr<const DexFile>> dex_files; 78766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler dex_files = oat_file_assistant.LoadDexFiles(*oat_file, dex_location.c_str()); 78866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(1u, dex_files.size()); 78966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 79066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 79166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: We have a DEX file. 79266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Expect: We should load an executable dex file from an alternative oat 79366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// location. 79466d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, LoadDexNoAlternateOat) { 79566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/LoadDexNoAlternateOat.jar"; 79666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string oat_location = GetScratchDir() + "/LoadDexNoAlternateOat.oat"; 79766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 79866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc1(), dex_location); 79966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 80066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant( 80166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler dex_location.c_str(), oat_location.c_str(), kRuntimeISA, true); 80266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string error_msg; 80366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(oat_file_assistant.MakeUpToDate(&error_msg)) << error_msg; 80466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 80566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::unique_ptr<OatFile> oat_file = oat_file_assistant.GetBestOatFile(); 80666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(oat_file.get() != nullptr); 80766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file->IsExecutable()); 80866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::unique_ptr<const DexFile>> dex_files; 80966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler dex_files = oat_file_assistant.LoadDexFiles(*oat_file, dex_location.c_str()); 81066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(1u, dex_files.size()); 81166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 81266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(OS::FileExists(oat_location.c_str())); 81366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 81466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Verify it didn't create an oat in the default location. 81566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant ofm(dex_location.c_str(), kRuntimeISA, false); 81666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(ofm.OatFileExists()); 81766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 81866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 81966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Turn an absolute path into a path relative to the current working 82066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// directory. 82166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhlerstatic std::string MakePathRelative(std::string target) { 82266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler char buf[MAXPATHLEN]; 82366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string cwd = getcwd(buf, MAXPATHLEN); 82466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 82566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Split the target and cwd paths into components. 82666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::string> target_path; 82766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::string> cwd_path; 82866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Split(target, '/', &target_path); 82966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Split(cwd, '/', &cwd_path); 83066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 83166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Reverse the path components, so we can use pop_back(). 83266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::reverse(target_path.begin(), target_path.end()); 83366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::reverse(cwd_path.begin(), cwd_path.end()); 83466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 83566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Drop the common prefix of the paths. Because we reversed the path 83666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // components, this becomes the common suffix of target_path and cwd_path. 83766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler while (!target_path.empty() && !cwd_path.empty() 83866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler && target_path.back() == cwd_path.back()) { 83966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler target_path.pop_back(); 84066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler cwd_path.pop_back(); 84166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 84266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 84366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // For each element of the remaining cwd_path, add '..' to the beginning 84466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // of the target path. Because we reversed the path components, we add to 84566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // the end of target_path. 84666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler for (unsigned int i = 0; i < cwd_path.size(); i++) { 84766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler target_path.push_back(".."); 84866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 84966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 85066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Reverse again to get the right path order, and join to get the result. 85166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::reverse(target_path.begin(), target_path.end()); 85266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler return Join(target_path, '/'); 85366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 85466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 85566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: Non-absolute path to Dex location. 85666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Expect: Not sure, but it shouldn't crash. 85766d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, NonAbsoluteDexLocation) { 85866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string abs_dex_location = GetScratchDir() + "/NonAbsoluteDexLocation.jar"; 85966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc1(), abs_dex_location); 86066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 86166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = MakePathRelative(abs_dex_location); 86266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, true); 86366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 86466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.IsInBootClassPath()); 86595abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_EQ(OatFileAssistant::kDex2OatNeeded, oat_file_assistant.GetDexOptNeeded()); 86666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileExists()); 86766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileExists()); 86866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileIsOutOfDate()); 86966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsUpToDate()); 87066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileIsOutOfDate()); 87166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsUpToDate()); 87266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 87366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 87466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: Very short, non-existent Dex location. 8759b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler// Expect: kNoDexOptNeeded. 87666d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, ShortDexLocation) { 87766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = "/xx"; 87866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 87966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, true); 88066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 88166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.IsInBootClassPath()); 8829b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_EQ(OatFileAssistant::kNoDexOptNeeded, oat_file_assistant.GetDexOptNeeded()); 88366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileExists()); 88466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileExists()); 88566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileIsOutOfDate()); 88666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsUpToDate()); 88766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileIsOutOfDate()); 88866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsUpToDate()); 8899b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_FALSE(oat_file_assistant.HasOriginalDexFiles()); 89066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 8919b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler // Trying to make it up to date should have no effect. 89266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string error_msg; 8939b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_TRUE(oat_file_assistant.MakeUpToDate(&error_msg)); 8949b994ea841eaaefbdda652251894a74db9cefcc8Richard Uhler EXPECT_TRUE(error_msg.empty()); 89566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 89666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 89766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: Non-standard extension for dex file. 89895abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler// Expect: The status is kDex2OatNeeded. 89966d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, LongDexExtension) { 90066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/LongDexExtension.jarx"; 90166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc1(), dex_location); 90266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 90366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, false); 90466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 90595abd04d211470ea4b9b9191b96dd0f32e7ce3a4Richard Uhler EXPECT_EQ(OatFileAssistant::kDex2OatNeeded, oat_file_assistant.GetDexOptNeeded()); 90666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 90766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.IsInBootClassPath()); 90866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileExists()); 90966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileIsOutOfDate()); 91066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsUpToDate()); 91166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileExists()); 91266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileIsOutOfDate()); 91366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsUpToDate()); 91466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 91566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 91666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// A task to generate a dex location. Used by the RaceToGenerate test. 91766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhlerclass RaceGenerateTask : public Task { 91866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler public: 91966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler explicit RaceGenerateTask(const std::string& dex_location, const std::string& oat_location) 92066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler : dex_location_(dex_location), oat_location_(oat_location), 92166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler loaded_oat_file_(nullptr) 92266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler {} 92366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 92466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler void Run(Thread* self) { 92566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler UNUSED(self); 92666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 92766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Load the dex files, and save a pointer to the loaded oat file, so that 92866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // we can verify only one oat file was loaded for the dex location. 92966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ClassLinker* linker = Runtime::Current()->GetClassLinker(); 93066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::unique_ptr<const DexFile>> dex_files; 93166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::string> error_msgs; 93266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler dex_files = linker->OpenDexFilesFromOat(dex_location_.c_str(), oat_location_.c_str(), &error_msgs); 93366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler CHECK(!dex_files.empty()) << Join(error_msgs, '\n'); 93407b3c2351bb527ea91c084dc19434600af9ae66bRichard Uhler CHECK(dex_files[0]->GetOatDexFile() != nullptr) << dex_files[0]->GetLocation(); 93507b3c2351bb527ea91c084dc19434600af9ae66bRichard Uhler loaded_oat_file_ = dex_files[0]->GetOatDexFile()->GetOatFile(); 93666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 93766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 93866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler const OatFile* GetLoadedOatFile() const { 93966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler return loaded_oat_file_; 94066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 94166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 94266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler private: 94366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location_; 94466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string oat_location_; 94566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler const OatFile* loaded_oat_file_; 94666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler}; 94766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 94866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Test the case where multiple processes race to generate an oat file. 94966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// This simulates multiple processes using multiple threads. 95066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// 95166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// We want only one Oat file to be loaded when there is a race to load, to 95266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// avoid using up the virtual memory address space. 95366d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, RaceToGenerate) { 95466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/RaceToGenerate.jar"; 9556343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler std::string oat_location = GetOdexDir() + "/RaceToGenerate.oat"; 95666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 95766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // We use the lib core dex file, because it's large, and hopefully should 95866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // take a while to generate. 95966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetLibCoreDexFileName(), dex_location); 96066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 96166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler const int kNumThreads = 32; 96266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Thread* self = Thread::Current(); 96366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ThreadPool thread_pool("Oat file assistant test thread pool", kNumThreads); 96466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::unique_ptr<RaceGenerateTask>> tasks; 96566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler for (int i = 0; i < kNumThreads; i++) { 96666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::unique_ptr<RaceGenerateTask> task(new RaceGenerateTask(dex_location, oat_location)); 96766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler thread_pool.AddTask(self, task.get()); 96866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler tasks.push_back(std::move(task)); 96966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 97066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler thread_pool.StartWorkers(self); 97166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler thread_pool.Wait(self, true, false); 97266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 97366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Verify every task got the same pointer. 97466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler const OatFile* expected = tasks[0]->GetLoadedOatFile(); 97566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler for (auto& task : tasks) { 97666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(expected, task->GetLoadedOatFile()); 97766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 97866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 97966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 98066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: We have a DEX file and an ODEX file, no OAT file, and dex2oat is 98166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// disabled. 98266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Expect: We should load the odex file non-executable. 98366d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantNoDex2OatTest, LoadDexOdexNoOat) { 98466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/LoadDexOdexNoOat.jar"; 9856343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler std::string odex_location = GetOdexDir() + "/LoadDexOdexNoOat.odex"; 98666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 98766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Create the dex and odex files 98866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc1(), dex_location); 98966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler GenerateOdexForTest(dex_location, odex_location); 99066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 99166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Load the oat using an executable oat file assistant. 99266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, true); 99366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 99466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::unique_ptr<OatFile> oat_file = oat_file_assistant.GetBestOatFile(); 99566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(oat_file.get() != nullptr); 99666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file->IsExecutable()); 99766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::unique_ptr<const DexFile>> dex_files; 99866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler dex_files = oat_file_assistant.LoadDexFiles(*oat_file, dex_location.c_str()); 99966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(1u, dex_files.size()); 100066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 100166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 100266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: We have a MultiDEX file and an ODEX file, no OAT file, and dex2oat is 100366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// disabled. 100466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Expect: We should load the odex file non-executable. 100566d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantNoDex2OatTest, LoadMultiDexOdexNoOat) { 100666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/LoadMultiDexOdexNoOat.jar"; 10076343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler std::string odex_location = GetOdexDir() + "/LoadMultiDexOdexNoOat.odex"; 100866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 100966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Create the dex and odex files 101066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetMultiDexSrc1(), dex_location); 101166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler GenerateOdexForTest(dex_location, odex_location); 101266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 101366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Load the oat using an executable oat file assistant. 101466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, true); 101566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 101666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::unique_ptr<OatFile> oat_file = oat_file_assistant.GetBestOatFile(); 101766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(oat_file.get() != nullptr); 101866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file->IsExecutable()); 101966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::unique_ptr<const DexFile>> dex_files; 102066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler dex_files = oat_file_assistant.LoadDexFiles(*oat_file, dex_location.c_str()); 102166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(2u, dex_files.size()); 102266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 102366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 102466d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST(OatFileAssistantUtilsTest, DexFilenameToOdexFilename) { 102566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string error_msg; 102666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string odex_file; 102766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 102866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(OatFileAssistant::DexFilenameToOdexFilename( 102966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler "/foo/bar/baz.jar", kArm, &odex_file, &error_msg)) << error_msg; 10306343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler EXPECT_EQ("/foo/bar/oat/arm/baz.odex", odex_file); 103166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 103266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(OatFileAssistant::DexFilenameToOdexFilename( 103366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler "/foo/bar/baz.funnyext", kArm, &odex_file, &error_msg)) << error_msg; 10346343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler EXPECT_EQ("/foo/bar/oat/arm/baz.odex", odex_file); 103566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 103666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(OatFileAssistant::DexFilenameToOdexFilename( 103766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler "nopath.jar", kArm, &odex_file, &error_msg)); 103866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(OatFileAssistant::DexFilenameToOdexFilename( 103966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler "/foo/bar/baz_noext", kArm, &odex_file, &error_msg)); 104066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 104166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 104223cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler// Verify the dexopt status values from dalvik.system.DexFile 104323cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler// match the OatFileAssistant::DexOptStatus values. 104423cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard UhlerTEST_F(OatFileAssistantTest, DexOptStatusValues) { 104523cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler ScopedObjectAccess soa(Thread::Current()); 104623cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler StackHandleScope<1> hs(soa.Self()); 104723cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler ClassLinker* linker = Runtime::Current()->GetClassLinker(); 104823cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler Handle<mirror::Class> dexfile( 104923cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler hs.NewHandle(linker->FindSystemClass(soa.Self(), "Ldalvik/system/DexFile;"))); 105023cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler ASSERT_FALSE(dexfile.Get() == nullptr); 105123cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler linker->EnsureInitialized(soa.Self(), dexfile, true, true); 105223cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler 1053c785344b87221f5e4e6473e5b762e4e61fe65dcfMathieu Chartier ArtField* no_dexopt_needed = mirror::Class::FindStaticField( 105423cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler soa.Self(), dexfile, "NO_DEXOPT_NEEDED", "I"); 105523cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler ASSERT_FALSE(no_dexopt_needed == nullptr); 105623cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler EXPECT_EQ(no_dexopt_needed->GetTypeAsPrimitiveType(), Primitive::kPrimInt); 105723cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler EXPECT_EQ(OatFileAssistant::kNoDexOptNeeded, no_dexopt_needed->GetInt(dexfile.Get())); 105823cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler 1059c785344b87221f5e4e6473e5b762e4e61fe65dcfMathieu Chartier ArtField* dex2oat_needed = mirror::Class::FindStaticField( 106023cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler soa.Self(), dexfile, "DEX2OAT_NEEDED", "I"); 106123cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler ASSERT_FALSE(dex2oat_needed == nullptr); 106223cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler EXPECT_EQ(dex2oat_needed->GetTypeAsPrimitiveType(), Primitive::kPrimInt); 106323cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler EXPECT_EQ(OatFileAssistant::kDex2OatNeeded, dex2oat_needed->GetInt(dexfile.Get())); 106423cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler 1065c785344b87221f5e4e6473e5b762e4e61fe65dcfMathieu Chartier ArtField* patchoat_needed = mirror::Class::FindStaticField( 106623cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler soa.Self(), dexfile, "PATCHOAT_NEEDED", "I"); 106723cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler ASSERT_FALSE(patchoat_needed == nullptr); 106823cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler EXPECT_EQ(patchoat_needed->GetTypeAsPrimitiveType(), Primitive::kPrimInt); 106923cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler EXPECT_EQ(OatFileAssistant::kPatchOatNeeded, patchoat_needed->GetInt(dexfile.Get())); 107023cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler 1071c785344b87221f5e4e6473e5b762e4e61fe65dcfMathieu Chartier ArtField* self_patchoat_needed = mirror::Class::FindStaticField( 107223cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler soa.Self(), dexfile, "SELF_PATCHOAT_NEEDED", "I"); 107323cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler ASSERT_FALSE(self_patchoat_needed == nullptr); 107423cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler EXPECT_EQ(self_patchoat_needed->GetTypeAsPrimitiveType(), Primitive::kPrimInt); 107523cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler EXPECT_EQ(OatFileAssistant::kSelfPatchOatNeeded, self_patchoat_needed->GetInt(dexfile.Get())); 107623cedd20e76bdbbdaa3f44a1b83f30698d04fa3fRichard Uhler} 107766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 107866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// TODO: More Tests: 107966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// * Test class linker falls back to unquickened dex for DexNoOat 108066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// * Test class linker falls back to unquickened dex for MultiDexNoOat 108166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// * Test using secondary isa 108266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// * Test with profiling info? 108366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// * Test for status of oat while oat is being generated (how?) 108466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// * Test case where 32 and 64 bit boot class paths differ, 108566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// and we ask IsInBootClassPath for a class in exactly one of the 32 or 108666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// 64 bit boot class paths. 108766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// * Test unexpected scenarios (?): 108866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// - Dex is stripped, don't have odex. 108966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// - Oat file corrupted after status check, before reload unexecutable 109066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// because it's unrelocated and no dex2oat 109166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 109266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} // namespace art 1093