oat_file_assistant_test.cc revision 07b3c2351bb527ea91c084dc19434600af9ae66b
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 2866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler#include "class_linker.h" 2966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler#include "common_runtime_test.h" 30bb9c6b1c55e9e2308b4f5892a398a8837231fdbdAndreas Gampe#include "compiler_callbacks.h" 3166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler#include "mem_map.h" 3266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler#include "os.h" 3366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler#include "thread-inl.h" 3466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler#include "utils.h" 3566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 3666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhlernamespace art { 3766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 3866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhlerclass OatFileAssistantTest : public CommonRuntimeTest { 3966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler public: 4066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler virtual void SetUp() { 4166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ReserveImageSpace(); 4266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler CommonRuntimeTest::SetUp(); 4366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 4466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Create a scratch directory to work from. 4566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler scratch_dir_ = android_data_ + "/OatFileAssistantTest"; 4666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_EQ(0, mkdir(scratch_dir_.c_str(), 0700)); 4766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 486343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler // Create a subdirectory in scratch for odex files. 496343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler odex_oat_dir_ = scratch_dir_ + "/oat"; 506343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler ASSERT_EQ(0, mkdir(odex_oat_dir_.c_str(), 0700)); 516343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler 526343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler odex_dir_ = odex_oat_dir_ + "/" + std::string(GetInstructionSetString(kRuntimeISA)); 536343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler ASSERT_EQ(0, mkdir(odex_dir_.c_str(), 0700)); 546343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler 5566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 5666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Verify the environment is as we expect 5766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler uint32_t checksum; 5866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string error_msg; 5966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(OS::FileExists(GetImageFile().c_str())) 6066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler << "Expected pre-compiled boot image to be at: " << GetImageFile(); 6166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(OS::FileExists(GetDexSrc1().c_str())) 6266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler << "Expected dex file to be at: " << GetDexSrc1(); 6366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(OS::FileExists(GetStrippedDexSrc1().c_str())) 6466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler << "Expected stripped dex file to be at: " << GetStrippedDexSrc1(); 6566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_FALSE(DexFile::GetChecksum(GetStrippedDexSrc1().c_str(), &checksum, &error_msg)) 6666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler << "Expected stripped dex file to be stripped: " << GetStrippedDexSrc1(); 6766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(OS::FileExists(GetMultiDexSrc1().c_str())) 6866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler << "Expected multidex file to be at: " << GetMultiDexSrc1(); 6966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(OS::FileExists(GetDexSrc2().c_str())) 7066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler << "Expected dex file to be at: " << GetDexSrc2(); 7166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 7266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 7366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler virtual void SetUpRuntimeOptions(RuntimeOptions* options) { 74892fc96694b51ac7a896dacd75af5b235f955825Richard Uhler // options->push_back(std::make_pair("-verbose:oat", nullptr)); 7566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 7666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Set up the image location. 7766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler options->push_back(std::make_pair("-Ximage:" + GetImageLocation(), 7866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler nullptr)); 7966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Make sure compilercallbacks are not set so that relocation will be 8066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // enabled. 81bb9c6b1c55e9e2308b4f5892a398a8837231fdbdAndreas Gampe callbacks_.reset(); 8266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 8366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 8466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler virtual void PreRuntimeCreate() { 8566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler UnreserveImageSpace(); 8666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 8766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 8866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler virtual void PostRuntimeCreate() { 8966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ReserveImageSpace(); 9066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 9166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 9266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler virtual void TearDown() { 936343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler ClearDirectory(odex_dir_.c_str()); 946343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler ASSERT_EQ(0, rmdir(odex_dir_.c_str())); 956343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler 966343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler ClearDirectory(odex_oat_dir_.c_str()); 976343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler ASSERT_EQ(0, rmdir(odex_oat_dir_.c_str())); 9866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 9966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ClearDirectory(scratch_dir_.c_str()); 10066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_EQ(0, rmdir(scratch_dir_.c_str())); 10166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 10266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler CommonRuntimeTest::TearDown(); 10366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 10466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 10566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler void Copy(std::string src, std::string dst) { 10666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::ifstream src_stream(src, std::ios::binary); 10766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::ofstream dst_stream(dst, std::ios::binary); 10866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 10966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler dst_stream << src_stream.rdbuf(); 11066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 11166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 11266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Returns the directory where the pre-compiled core.art can be found. 11366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // TODO: We should factor out this into common tests somewhere rather than 11466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // re-hardcoding it here (This was copied originally from the elf writer 11566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // test). 11666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string GetImageDirectory() { 11766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler if (IsHost()) { 11866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler const char* host_dir = getenv("ANDROID_HOST_OUT"); 11966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler CHECK(host_dir != NULL); 12066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler return std::string(host_dir) + "/framework"; 12166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } else { 12266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler return std::string("/data/art-test"); 12366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 12466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 12566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 12666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string GetImageLocation() { 12766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler return GetImageDirectory() + "/core.art"; 12866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 12966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 13066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string GetImageFile() { 13166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler return GetImageDirectory() + "/" + GetInstructionSetString(kRuntimeISA) 13266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler + "/core.art"; 13366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 13466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 13566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string GetDexSrc1() { 13666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler return GetTestDexFileName("Main"); 13766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 13866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 13966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Returns the path to a dex file equivalent to GetDexSrc1, but with the dex 14066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // file stripped. 14166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string GetStrippedDexSrc1() { 14266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler return GetTestDexFileName("MainStripped"); 14366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 14466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 14566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string GetMultiDexSrc1() { 14666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler return GetTestDexFileName("MultiDex"); 14766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 14866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 14966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string GetDexSrc2() { 15066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler return GetTestDexFileName("Nested"); 15166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 15266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 15366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Scratch directory, for dex and odex files (oat files will go in the 15466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // dalvik cache). 15566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string GetScratchDir() { 15666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler return scratch_dir_; 15766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 15866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 1596343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler // Odex directory is the subdirectory in the scratch directory where odex 16066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // files should be located. 1616343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler std::string GetOdexDir() { 1626343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler return odex_dir_; 16366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 16466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 16566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Generate an odex file for the purposes of test. 16666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // If pic is true, generates a PIC odex. 16766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler void GenerateOdexForTest(const std::string& dex_location, 16866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler const std::string& odex_location, 16966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler bool pic = false) { 17066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // For this operation, we temporarily redirect the dalvik cache so dex2oat 17166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // doesn't find the relocated image file. 17266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string android_data_tmp = GetScratchDir() + "AndroidDataTmp"; 17366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler setenv("ANDROID_DATA", android_data_tmp.c_str(), 1); 17466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::string> args; 17566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler args.push_back("--dex-file=" + dex_location); 17666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler args.push_back("--oat-file=" + odex_location); 17766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler if (pic) { 17866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler args.push_back("--compile-pic"); 17966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } else { 18066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler args.push_back("--include-patch-information"); 18105dd8a63e213d2bf025c97b9cd04eee354d0e5b4Richard Uhler 18205dd8a63e213d2bf025c97b9cd04eee354d0e5b4Richard Uhler // We need to use the quick compiler to generate non-PIC code, because 18305dd8a63e213d2bf025c97b9cd04eee354d0e5b4Richard Uhler // the optimizing compiler always generates PIC. 18405dd8a63e213d2bf025c97b9cd04eee354d0e5b4Richard Uhler args.push_back("--compiler-backend=Quick"); 18566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 18666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler args.push_back("--runtime-arg"); 18766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler args.push_back("-Xnorelocate"); 18866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string error_msg; 18966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(OatFileAssistant::Dex2Oat(args, &error_msg)) << error_msg; 19066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler setenv("ANDROID_DATA", android_data_.c_str(), 1); 19166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 19266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 19366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler void GeneratePicOdexForTest(const std::string& dex_location, 19466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler const std::string& odex_location) { 19566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler GenerateOdexForTest(dex_location, odex_location, true); 19666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 19766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 19866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler private: 19966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Reserve memory around where the image will be loaded so other memory 20066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // won't conflict when it comes time to load the image. 20166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // This can be called with an already loaded image to reserve the space 20266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // around it. 20366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler void ReserveImageSpace() { 20466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler MemMap::Init(); 20566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 20666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Ensure a chunk of memory is reserved for the image space. 20766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler uintptr_t reservation_start = ART_BASE_ADDRESS + ART_BASE_ADDRESS_MIN_DELTA; 20866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler uintptr_t reservation_end = ART_BASE_ADDRESS + ART_BASE_ADDRESS_MAX_DELTA 2093dbf23412481c4da51f0ebe32bec5d300c36834bHiroshi Yamauchi // Include the main space that has to come right after the 2103dbf23412481c4da51f0ebe32bec5d300c36834bHiroshi Yamauchi // image in case of the GSS collector. 2113dbf23412481c4da51f0ebe32bec5d300c36834bHiroshi Yamauchi + 384 * MB; 21266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 21366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::unique_ptr<BacktraceMap> map(BacktraceMap::Create(getpid(), true)); 21466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(map.get() != nullptr) << "Failed to build process map"; 21566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler for (BacktraceMap::const_iterator it = map->begin(); 21666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler reservation_start < reservation_end && it != map->end(); ++it) { 2173efe979d4292330c8fab1708a4361e58681a88cbRichard Uhler ReserveImageSpaceChunk(reservation_start, std::min(it->start, reservation_end)); 2183efe979d4292330c8fab1708a4361e58681a88cbRichard Uhler reservation_start = std::max(reservation_start, it->end); 2193efe979d4292330c8fab1708a4361e58681a88cbRichard Uhler } 2203efe979d4292330c8fab1708a4361e58681a88cbRichard Uhler ReserveImageSpaceChunk(reservation_start, reservation_end); 2213efe979d4292330c8fab1708a4361e58681a88cbRichard Uhler } 22266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 2233efe979d4292330c8fab1708a4361e58681a88cbRichard Uhler // Reserve a chunk of memory for the image space in the given range. 2243efe979d4292330c8fab1708a4361e58681a88cbRichard Uhler // Only has effect for chunks with a positive number of bytes. 2253efe979d4292330c8fab1708a4361e58681a88cbRichard Uhler void ReserveImageSpaceChunk(uintptr_t start, uintptr_t end) { 2263efe979d4292330c8fab1708a4361e58681a88cbRichard Uhler if (start < end) { 2273efe979d4292330c8fab1708a4361e58681a88cbRichard Uhler std::string error_msg; 22866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler image_reservation_.push_back(std::unique_ptr<MemMap>( 22966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler MemMap::MapAnonymous("image reservation", 2303efe979d4292330c8fab1708a4361e58681a88cbRichard Uhler reinterpret_cast<uint8_t*>(start), end - start, 23166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler PROT_NONE, false, false, &error_msg))); 23266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(image_reservation_.back().get() != nullptr) << error_msg; 23366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler LOG(INFO) << "Reserved space for image " << 23466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler reinterpret_cast<void*>(image_reservation_.back()->Begin()) << "-" << 23566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler reinterpret_cast<void*>(image_reservation_.back()->End()); 23666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 23766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 23866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 23966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 24066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Unreserve any memory reserved by ReserveImageSpace. This should be called 24166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // before the image is loaded. 24266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler void UnreserveImageSpace() { 24366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler image_reservation_.clear(); 24466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 24566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 24666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string scratch_dir_; 2476343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler std::string odex_oat_dir_; 2486343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler std::string odex_dir_; 24966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::unique_ptr<MemMap>> image_reservation_; 25066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler}; 25166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 25266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhlerclass OatFileAssistantNoDex2OatTest : public OatFileAssistantTest { 25366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler public: 25466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler virtual void SetUpRuntimeOptions(RuntimeOptions* options) { 25566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistantTest::SetUpRuntimeOptions(options); 25666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler options->push_back(std::make_pair("-Xnodex2oat", nullptr)); 25766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 25866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler}; 25966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 26066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Generate an oat file for the purposes of test, as opposed to testing 26166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// generation of oat files. 26266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhlerstatic void GenerateOatForTest(const char* dex_location) { 26366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location, kRuntimeISA, false); 26466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 26566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string error_msg; 26666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(oat_file_assistant.GenerateOatFile(&error_msg)) << error_msg; 26766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 26866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 26966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: We have a DEX file, but no OAT file for it. 27066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Expect: The oat file status is kOutOfDate. 27166d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, DexNoOat) { 27266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/DexNoOat.jar"; 27366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc1(), dex_location); 27466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 27566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, false); 27666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 27766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(OatFileAssistant::kOutOfDate, oat_file_assistant.GetStatus()); 27866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 27966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.IsInBootClassPath()); 28066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileExists()); 28166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileIsOutOfDate()); 28266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileNeedsRelocation()); 28366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsUpToDate()); 28466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(OatFileAssistant::kOutOfDate, oat_file_assistant.OdexFileStatus()); 28566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileExists()); 28666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileIsOutOfDate()); 28766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileNeedsRelocation()); 28866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsUpToDate()); 28966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(OatFileAssistant::kOutOfDate, oat_file_assistant.OatFileStatus()); 29066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 29166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 29266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: We have no DEX file and no OAT file. 29366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Expect: Status is out of date. Loading should fail, but not crash. 29466d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, NoDexNoOat) { 29566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/NoDexNoOat.jar"; 29666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 29766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, true); 29866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 29966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(OatFileAssistant::kOutOfDate, oat_file_assistant.GetStatus()); 30066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::unique_ptr<OatFile> oat_file = oat_file_assistant.GetBestOatFile(); 30166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(nullptr, oat_file.get()); 30266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 30366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 30466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: We have a DEX file and up-to-date OAT file for it. 30566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Expect: The oat file status is kUpToDate. 30666d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, OatUpToDate) { 30766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/OatUpToDate.jar"; 30866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc1(), dex_location); 30966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler GenerateOatForTest(dex_location.c_str()); 31066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 31166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, false); 31266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 31366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(OatFileAssistant::kUpToDate, oat_file_assistant.GetStatus()); 31466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.IsInBootClassPath()); 31566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileExists()); 31666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileIsOutOfDate()); 31766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsUpToDate()); 31866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileExists()); 31966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsOutOfDate()); 32066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileNeedsRelocation()); 32166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileIsUpToDate()); 32266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(OatFileAssistant::kUpToDate, oat_file_assistant.OatFileStatus()); 32366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 32466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 32566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: We have a MultiDEX file and up-to-date OAT file for it. 32666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Expect: The oat file status is kUpToDate. 32766d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, MultiDexOatUpToDate) { 32866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/MultiDexOatUpToDate.jar"; 32966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetMultiDexSrc1(), dex_location); 33066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler GenerateOatForTest(dex_location.c_str()); 33166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 33266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Verify we can load both dex files. 333e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, true); 334e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler std::unique_ptr<OatFile> oat_file = oat_file_assistant.GetBestOatFile(); 335e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler ASSERT_TRUE(oat_file.get() != nullptr); 336e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler EXPECT_TRUE(oat_file->IsExecutable()); 337e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler std::vector<std::unique_ptr<const DexFile>> dex_files; 338e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler dex_files = oat_file_assistant.LoadDexFiles(*oat_file, dex_location.c_str()); 339e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler EXPECT_EQ(2u, dex_files.size()); 340e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler} 341e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler 342e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler// Case: We have a MultiDEX file and up-to-date OAT file for it with relative 343e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler// encoded dex locations. 344e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler// Expect: The oat file status is kUpToDate. 345e5fed03772144595c0904faf3d6974cc55214c8cRichard UhlerTEST_F(OatFileAssistantTest, RelativeEncodedDexLocation) { 346e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler std::string dex_location = GetScratchDir() + "/RelativeEncodedDexLocation.jar"; 3476343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler std::string oat_location = GetOdexDir() + "/RelativeEncodedDexLocation.oat"; 348e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler 349e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler // Create the dex file 350e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler Copy(GetMultiDexSrc1(), dex_location); 351e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler 352e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler // Create the oat file with relative encoded dex location. 353e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler std::vector<std::string> args; 354e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler args.push_back("--dex-file=" + dex_location); 355e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler args.push_back("--dex-location=" + std::string("RelativeEncodedDexLocation.jar")); 356e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler args.push_back("--oat-file=" + oat_location); 357e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler 358e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler std::string error_msg; 359e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler ASSERT_TRUE(OatFileAssistant::Dex2Oat(args, &error_msg)) << error_msg; 360e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler 361e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler // Verify we can load both dex files. 362e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), 363e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler oat_location.c_str(), 364e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler kRuntimeISA, true); 365e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler std::unique_ptr<OatFile> oat_file = oat_file_assistant.GetBestOatFile(); 36666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(oat_file.get() != nullptr); 36766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file->IsExecutable()); 36866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::unique_ptr<const DexFile>> dex_files; 369e5fed03772144595c0904faf3d6974cc55214c8cRichard Uhler dex_files = oat_file_assistant.LoadDexFiles(*oat_file, dex_location.c_str()); 37066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(2u, dex_files.size()); 37166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 37266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 37366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: We have a DEX file and out of date OAT file. 37466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Expect: The oat file status is kOutOfDate. 37566d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, OatOutOfDate) { 37666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/OatOutOfDate.jar"; 37766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 37866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // We create a dex, generate an oat for it, then overwrite the dex with a 37966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // different dex to make the oat out of date. 38066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc1(), dex_location); 38166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler GenerateOatForTest(dex_location.c_str()); 38266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc2(), dex_location); 38366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 38466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, false); 38566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(OatFileAssistant::kOutOfDate, oat_file_assistant.GetStatus()); 38666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 38766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.IsInBootClassPath()); 38866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileExists()); 38966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileIsOutOfDate()); 39066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsUpToDate()); 39166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileExists()); 39266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileIsOutOfDate()); 39366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsUpToDate()); 39466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 39566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 39666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: We have a DEX file and an ODEX file, but no OAT file. 39766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Expect: The oat file status is kNeedsRelocation. 39866d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, DexOdexNoOat) { 39966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/DexOdexNoOat.jar"; 4006343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler std::string odex_location = GetOdexDir() + "/DexOdexNoOat.odex"; 40166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 40266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Create the dex and odex files 40366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc1(), dex_location); 40466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler GenerateOdexForTest(dex_location, odex_location); 40566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 40666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Verify the status. 40766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, false); 40866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 40966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(OatFileAssistant::kNeedsRelocation, oat_file_assistant.GetStatus()); 41066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 41166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.IsInBootClassPath()); 41266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileExists()); 41366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsOutOfDate()); 41466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsUpToDate()); 41566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileNeedsRelocation()); 41666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(OatFileAssistant::kNeedsRelocation, oat_file_assistant.OdexFileNeedsRelocation()); 41766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileExists()); 41866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileIsOutOfDate()); 41966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsUpToDate()); 42066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 42166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 42266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: We have a stripped DEX file and an ODEX file, but no OAT file. 42366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Expect: The oat file status is kNeedsRelocation. 42466d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, StrippedDexOdexNoOat) { 42566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/StrippedDexOdexNoOat.jar"; 4266343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler std::string odex_location = GetOdexDir() + "/StrippedDexOdexNoOat.odex"; 42766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 42866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Create the dex and odex files 42966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc1(), dex_location); 43066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler GenerateOdexForTest(dex_location, odex_location); 43166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 43266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Strip the dex file 43366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetStrippedDexSrc1(), dex_location); 43466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 43566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Verify the status. 43666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, true); 43766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 43866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(OatFileAssistant::kNeedsRelocation, oat_file_assistant.GetStatus()); 43966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 44066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.IsInBootClassPath()); 44166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileExists()); 44266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsOutOfDate()); 44366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsUpToDate()); 44466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileExists()); 44566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileIsOutOfDate()); 44666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsUpToDate()); 44766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 44866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Make the oat file up to date. 44966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string error_msg; 45066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(oat_file_assistant.MakeUpToDate(&error_msg)) << error_msg; 45166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 45266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(OatFileAssistant::kUpToDate, oat_file_assistant.GetStatus()); 45366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 45466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.IsInBootClassPath()); 45566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileExists()); 45666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsOutOfDate()); 45766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsUpToDate()); 45866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileExists()); 45966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsOutOfDate()); 46066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileIsUpToDate()); 46166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 46266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Verify we can load the dex files from it. 46366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::unique_ptr<OatFile> oat_file = oat_file_assistant.GetBestOatFile(); 46466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(oat_file.get() != nullptr); 46566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file->IsExecutable()); 46666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::unique_ptr<const DexFile>> dex_files; 46766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler dex_files = oat_file_assistant.LoadDexFiles(*oat_file, dex_location.c_str()); 46866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(1u, dex_files.size()); 46966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 47066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 47166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: We have a stripped DEX file, an ODEX file, and an out of date OAT file. 47266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Expect: The oat file status is kNeedsRelocation. 47366d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, StrippedDexOdexOat) { 47466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/StrippedDexOdexOat.jar"; 4756343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler std::string odex_location = GetOdexDir() + "/StrippedDexOdexOat.odex"; 47666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 47766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Create the oat file from a different dex file so it looks out of date. 47866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc2(), dex_location); 47966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler GenerateOatForTest(dex_location.c_str()); 48066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 48166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Create the odex file 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 49166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(OatFileAssistant::kNeedsRelocation, oat_file_assistant.GetStatus()); 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_TRUE(oat_file_assistant.OdexFileNeedsRelocation()); 49766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsUpToDate()); 49866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileExists()); 49966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileIsOutOfDate()); 50066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsUpToDate()); 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 50666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(OatFileAssistant::kUpToDate, oat_file_assistant.GetStatus()); 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_TRUE(oat_file_assistant.OdexFileNeedsRelocation()); 51266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsUpToDate()); 51366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileExists()); 51466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsOutOfDate()); 51566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileNeedsRelocation()); 51666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileIsUpToDate()); 51766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 51866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Verify we can load the dex files from it. 51966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::unique_ptr<OatFile> oat_file = oat_file_assistant.GetBestOatFile(); 52066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(oat_file.get() != nullptr); 52166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file->IsExecutable()); 52266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::unique_ptr<const DexFile>> dex_files; 52366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler dex_files = oat_file_assistant.LoadDexFiles(*oat_file, dex_location.c_str()); 52466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(1u, dex_files.size()); 52566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 52666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 52766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: We have a DEX file, an ODEX file and an OAT file, where the ODEX and 52866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// OAT files both have patch delta of 0. 52966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Expect: It shouldn't crash. 53066d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, OdexOatOverlap) { 53166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/OdexOatOverlap.jar"; 5326343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler std::string odex_location = GetOdexDir() + "/OdexOatOverlap.odex"; 5336343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler std::string oat_location = GetOdexDir() + "/OdexOatOverlap.oat"; 53466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 53566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Create the dex and odex files 53666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc1(), dex_location); 53766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler GenerateOdexForTest(dex_location, odex_location); 53866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 53966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Create the oat file by copying the odex so they are located in the same 54066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // place in memory. 54166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(odex_location, oat_location); 54266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 54366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Verify things don't go bad. 54466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), 54566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler oat_location.c_str(), kRuntimeISA, true); 54666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 54766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(OatFileAssistant::kNeedsRelocation, oat_file_assistant.GetStatus()); 54866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 54966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.IsInBootClassPath()); 55066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileExists()); 55166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsOutOfDate()); 55266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsUpToDate()); 55366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileExists()); 55466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsOutOfDate()); 55566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsUpToDate()); 55666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 55766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Things aren't relocated, so it should fall back to interpreted. 55866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::unique_ptr<OatFile> oat_file = oat_file_assistant.GetBestOatFile(); 55966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(oat_file.get() != nullptr); 56066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file->IsExecutable()); 56166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::unique_ptr<const DexFile>> dex_files; 56266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler dex_files = oat_file_assistant.LoadDexFiles(*oat_file, dex_location.c_str()); 56366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(1u, dex_files.size()); 56466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 56566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 56666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: We have a DEX file and a PIC ODEX file, but no OAT file. 56766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Expect: The oat file status is kUpToDate, because PIC needs no relocation. 56866d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, DexPicOdexNoOat) { 56966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/DexPicOdexNoOat.jar"; 5706343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler std::string odex_location = GetOdexDir() + "/DexPicOdexNoOat.odex"; 57166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 57266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Create the dex and odex files 57366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc1(), dex_location); 57466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler GeneratePicOdexForTest(dex_location, odex_location); 57566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 57666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Verify the status. 57766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, false); 57866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 57966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(OatFileAssistant::kUpToDate, oat_file_assistant.GetStatus()); 58066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 58166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.IsInBootClassPath()); 58266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileExists()); 58366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsOutOfDate()); 58466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileIsUpToDate()); 58566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileExists()); 58666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileIsOutOfDate()); 58766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsUpToDate()); 58866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 58966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 59066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: We have a DEX file and up-to-date OAT file for it. 59166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Expect: We should load an executable dex file. 59266d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, LoadOatUpToDate) { 59366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/LoadOatUpToDate.jar"; 59466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 59566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc1(), dex_location); 59666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler GenerateOatForTest(dex_location.c_str()); 59766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 59866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Load the oat using an oat file assistant. 59966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, true); 60066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 60166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::unique_ptr<OatFile> oat_file = oat_file_assistant.GetBestOatFile(); 60266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(oat_file.get() != nullptr); 60366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file->IsExecutable()); 60466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::unique_ptr<const DexFile>> dex_files; 60566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler dex_files = oat_file_assistant.LoadDexFiles(*oat_file, dex_location.c_str()); 60666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(1u, dex_files.size()); 60766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 60866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 60966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: We have a DEX file and up-to-date OAT file for it. 61066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Expect: Loading non-executable should load the oat non-executable. 61166d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, LoadNoExecOatUpToDate) { 61266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/LoadNoExecOatUpToDate.jar"; 61366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 61466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc1(), dex_location); 61566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler GenerateOatForTest(dex_location.c_str()); 61666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 61766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Load the oat using an oat file assistant. 61866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, false); 61966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 62066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::unique_ptr<OatFile> oat_file = oat_file_assistant.GetBestOatFile(); 62166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(oat_file.get() != nullptr); 62266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file->IsExecutable()); 62366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::unique_ptr<const DexFile>> dex_files; 62466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler dex_files = oat_file_assistant.LoadDexFiles(*oat_file, dex_location.c_str()); 62566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(1u, dex_files.size()); 62666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 62766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 62866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: We have a DEX file. 62966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Expect: We should load an executable dex file from an alternative oat 63066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// location. 63166d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, LoadDexNoAlternateOat) { 63266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/LoadDexNoAlternateOat.jar"; 63366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string oat_location = GetScratchDir() + "/LoadDexNoAlternateOat.oat"; 63466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 63566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc1(), dex_location); 63666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 63766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant( 63866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler dex_location.c_str(), oat_location.c_str(), kRuntimeISA, true); 63966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string error_msg; 64066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(oat_file_assistant.MakeUpToDate(&error_msg)) << error_msg; 64166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 64266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::unique_ptr<OatFile> oat_file = oat_file_assistant.GetBestOatFile(); 64366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(oat_file.get() != nullptr); 64466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file->IsExecutable()); 64566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::unique_ptr<const DexFile>> dex_files; 64666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler dex_files = oat_file_assistant.LoadDexFiles(*oat_file, dex_location.c_str()); 64766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(1u, dex_files.size()); 64866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 64966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(OS::FileExists(oat_location.c_str())); 65066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 65166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Verify it didn't create an oat in the default location. 65266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant ofm(dex_location.c_str(), kRuntimeISA, false); 65366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(ofm.OatFileExists()); 65466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 65566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 65666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: Non-existent Dex location. 65766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Expect: The dex code is out of date, and trying to update it fails. 65866d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, NonExsistentDexLocation) { 65966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/BadDexLocation.jar"; 66066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 66166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, true); 66266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 66366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.IsInBootClassPath()); 66466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(OatFileAssistant::kOutOfDate, oat_file_assistant.GetStatus()); 66566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileExists()); 66666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileExists()); 66766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileIsOutOfDate()); 66866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsUpToDate()); 66966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileIsOutOfDate()); 67066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsUpToDate()); 67166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 67266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string error_msg; 67366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.MakeUpToDate(&error_msg)); 67466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(error_msg.empty()); 67566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 67666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 67766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Turn an absolute path into a path relative to the current working 67866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// directory. 67966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhlerstatic std::string MakePathRelative(std::string target) { 68066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler char buf[MAXPATHLEN]; 68166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string cwd = getcwd(buf, MAXPATHLEN); 68266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 68366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Split the target and cwd paths into components. 68466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::string> target_path; 68566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::string> cwd_path; 68666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Split(target, '/', &target_path); 68766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Split(cwd, '/', &cwd_path); 68866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 68966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Reverse the path components, so we can use pop_back(). 69066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::reverse(target_path.begin(), target_path.end()); 69166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::reverse(cwd_path.begin(), cwd_path.end()); 69266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 69366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Drop the common prefix of the paths. Because we reversed the path 69466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // components, this becomes the common suffix of target_path and cwd_path. 69566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler while (!target_path.empty() && !cwd_path.empty() 69666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler && target_path.back() == cwd_path.back()) { 69766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler target_path.pop_back(); 69866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler cwd_path.pop_back(); 69966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 70066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 70166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // For each element of the remaining cwd_path, add '..' to the beginning 70266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // of the target path. Because we reversed the path components, we add to 70366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // the end of target_path. 70466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler for (unsigned int i = 0; i < cwd_path.size(); i++) { 70566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler target_path.push_back(".."); 70666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 70766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 70866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Reverse again to get the right path order, and join to get the result. 70966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::reverse(target_path.begin(), target_path.end()); 71066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler return Join(target_path, '/'); 71166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 71266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 71366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: Non-absolute path to Dex location. 71466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Expect: Not sure, but it shouldn't crash. 71566d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, NonAbsoluteDexLocation) { 71666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string abs_dex_location = GetScratchDir() + "/NonAbsoluteDexLocation.jar"; 71766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc1(), abs_dex_location); 71866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 71966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = MakePathRelative(abs_dex_location); 72066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, true); 72166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 72266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.IsInBootClassPath()); 72366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(OatFileAssistant::kOutOfDate, oat_file_assistant.GetStatus()); 72466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileExists()); 72566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileExists()); 72666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileIsOutOfDate()); 72766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsUpToDate()); 72866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileIsOutOfDate()); 72966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsUpToDate()); 73066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 73166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 73266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: Very short, non-existent Dex location. 73366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Expect: Dex code is out of date, and trying to update it fails. 73466d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, ShortDexLocation) { 73566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = "/xx"; 73666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 73766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, true); 73866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 73966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.IsInBootClassPath()); 74066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(OatFileAssistant::kOutOfDate, oat_file_assistant.GetStatus()); 74166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileExists()); 74266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileExists()); 74366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileIsOutOfDate()); 74466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsUpToDate()); 74566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileIsOutOfDate()); 74666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsUpToDate()); 74766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 74866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string error_msg; 74966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.MakeUpToDate(&error_msg)); 75066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(error_msg.empty()); 75166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 75266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 75366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: Non-standard extension for dex file. 75466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Expect: The oat file status is kOutOfDate. 75566d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, LongDexExtension) { 75666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/LongDexExtension.jarx"; 75766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc1(), dex_location); 75866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 75966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, false); 76066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 76166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(OatFileAssistant::kOutOfDate, oat_file_assistant.GetStatus()); 76266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 76366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.IsInBootClassPath()); 76466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileExists()); 76566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OdexFileIsOutOfDate()); 76666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OdexFileIsUpToDate()); 76766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileExists()); 76866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(oat_file_assistant.OatFileIsOutOfDate()); 76966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file_assistant.OatFileIsUpToDate()); 77066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 77166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 77266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// A task to generate a dex location. Used by the RaceToGenerate test. 77366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhlerclass RaceGenerateTask : public Task { 77466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler public: 77566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler explicit RaceGenerateTask(const std::string& dex_location, const std::string& oat_location) 77666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler : dex_location_(dex_location), oat_location_(oat_location), 77766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler loaded_oat_file_(nullptr) 77866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler {} 77966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 78066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler void Run(Thread* self) { 78166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler UNUSED(self); 78266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 78366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Load the dex files, and save a pointer to the loaded oat file, so that 78466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // we can verify only one oat file was loaded for the dex location. 78566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ClassLinker* linker = Runtime::Current()->GetClassLinker(); 78666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::unique_ptr<const DexFile>> dex_files; 78766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::string> error_msgs; 78866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler dex_files = linker->OpenDexFilesFromOat(dex_location_.c_str(), oat_location_.c_str(), &error_msgs); 78966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler CHECK(!dex_files.empty()) << Join(error_msgs, '\n'); 79007b3c2351bb527ea91c084dc19434600af9ae66bRichard Uhler CHECK(dex_files[0]->GetOatDexFile() != nullptr) << dex_files[0]->GetLocation(); 79107b3c2351bb527ea91c084dc19434600af9ae66bRichard Uhler loaded_oat_file_ = dex_files[0]->GetOatDexFile()->GetOatFile(); 79266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 79366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 79466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler const OatFile* GetLoadedOatFile() const { 79566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler return loaded_oat_file_; 79666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 79766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 79866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler private: 79966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location_; 80066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string oat_location_; 80166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler const OatFile* loaded_oat_file_; 80266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler}; 80366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 80466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Test the case where multiple processes race to generate an oat file. 80566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// This simulates multiple processes using multiple threads. 80666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// 80766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// We want only one Oat file to be loaded when there is a race to load, to 80866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// avoid using up the virtual memory address space. 80966d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantTest, RaceToGenerate) { 81066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/RaceToGenerate.jar"; 8116343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler std::string oat_location = GetOdexDir() + "/RaceToGenerate.oat"; 81266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 81366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // We use the lib core dex file, because it's large, and hopefully should 81466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // take a while to generate. 81566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetLibCoreDexFileName(), dex_location); 81666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 81766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler const int kNumThreads = 32; 81866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Thread* self = Thread::Current(); 81966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ThreadPool thread_pool("Oat file assistant test thread pool", kNumThreads); 82066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::unique_ptr<RaceGenerateTask>> tasks; 82166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler for (int i = 0; i < kNumThreads; i++) { 82266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::unique_ptr<RaceGenerateTask> task(new RaceGenerateTask(dex_location, oat_location)); 82366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler thread_pool.AddTask(self, task.get()); 82466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler tasks.push_back(std::move(task)); 82566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 82666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler thread_pool.StartWorkers(self); 82766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler thread_pool.Wait(self, true, false); 82866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 82966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Verify every task got the same pointer. 83066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler const OatFile* expected = tasks[0]->GetLoadedOatFile(); 83166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler for (auto& task : tasks) { 83266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(expected, task->GetLoadedOatFile()); 83366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler } 83466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 83566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 83666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: We have a DEX file and an ODEX file, no OAT file, and dex2oat is 83766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// disabled. 83866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Expect: We should load the odex file non-executable. 83966d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantNoDex2OatTest, LoadDexOdexNoOat) { 84066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/LoadDexOdexNoOat.jar"; 8416343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler std::string odex_location = GetOdexDir() + "/LoadDexOdexNoOat.odex"; 84266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 84366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Create the dex and odex files 84466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetDexSrc1(), dex_location); 84566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler GenerateOdexForTest(dex_location, odex_location); 84666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 84766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Load the oat using an executable oat file assistant. 84866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, true); 84966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 85066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::unique_ptr<OatFile> oat_file = oat_file_assistant.GetBestOatFile(); 85166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(oat_file.get() != nullptr); 85266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file->IsExecutable()); 85366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::unique_ptr<const DexFile>> dex_files; 85466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler dex_files = oat_file_assistant.LoadDexFiles(*oat_file, dex_location.c_str()); 85566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(1u, dex_files.size()); 85666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 85766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 85866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Case: We have a MultiDEX file and an ODEX file, no OAT file, and dex2oat is 85966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// disabled. 86066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// Expect: We should load the odex file non-executable. 86166d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST_F(OatFileAssistantNoDex2OatTest, LoadMultiDexOdexNoOat) { 86266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string dex_location = GetScratchDir() + "/LoadMultiDexOdexNoOat.jar"; 8636343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler std::string odex_location = GetOdexDir() + "/LoadMultiDexOdexNoOat.odex"; 86466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 86566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Create the dex and odex files 86666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler Copy(GetMultiDexSrc1(), dex_location); 86766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler GenerateOdexForTest(dex_location, odex_location); 86866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 86966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler // Load the oat using an executable oat file assistant. 87066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler OatFileAssistant oat_file_assistant(dex_location.c_str(), kRuntimeISA, true); 87166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 87266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::unique_ptr<OatFile> oat_file = oat_file_assistant.GetBestOatFile(); 87366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler ASSERT_TRUE(oat_file.get() != nullptr); 87466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(oat_file->IsExecutable()); 87566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::vector<std::unique_ptr<const DexFile>> dex_files; 87666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler dex_files = oat_file_assistant.LoadDexFiles(*oat_file, dex_location.c_str()); 87766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_EQ(2u, dex_files.size()); 87866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 87966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 88066d874d96d5699bb090c59f47a5a528956ca053eRichard UhlerTEST(OatFileAssistantUtilsTest, DexFilenameToOdexFilename) { 88166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string error_msg; 88266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler std::string odex_file; 88366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 88466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(OatFileAssistant::DexFilenameToOdexFilename( 88566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler "/foo/bar/baz.jar", kArm, &odex_file, &error_msg)) << error_msg; 8866343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler EXPECT_EQ("/foo/bar/oat/arm/baz.odex", odex_file); 88766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 88866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_TRUE(OatFileAssistant::DexFilenameToOdexFilename( 88966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler "/foo/bar/baz.funnyext", kArm, &odex_file, &error_msg)) << error_msg; 8906343411e52d2a59cd716cb47a85ebd0d68e61a3cRichard Uhler EXPECT_EQ("/foo/bar/oat/arm/baz.odex", odex_file); 89166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 89266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(OatFileAssistant::DexFilenameToOdexFilename( 89366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler "nopath.jar", kArm, &odex_file, &error_msg)); 89466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler EXPECT_FALSE(OatFileAssistant::DexFilenameToOdexFilename( 89566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler "/foo/bar/baz_noext", kArm, &odex_file, &error_msg)); 89666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} 89766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 89866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 89966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// TODO: More Tests: 90066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// * Test class linker falls back to unquickened dex for DexNoOat 90166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// * Test class linker falls back to unquickened dex for MultiDexNoOat 90266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// * Test multidex files: 90366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// - Multidex with only classes2.dex out of date should have status 90466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// kOutOfDate 90566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// * Test using secondary isa 90666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// * Test with profiling info? 90766d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// * Test for status of oat while oat is being generated (how?) 90866d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// * Test case where 32 and 64 bit boot class paths differ, 90966d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// and we ask IsInBootClassPath for a class in exactly one of the 32 or 91066d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// 64 bit boot class paths. 91166d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// * Test unexpected scenarios (?): 91266d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// - Dex is stripped, don't have odex. 91366d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// - Oat file corrupted after status check, before reload unexecutable 91466d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler// because it's unrelocated and no dex2oat 91566d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler 91666d874d96d5699bb090c59f47a5a528956ca053eRichard Uhler} // namespace art 917