oat_file_manager.cc revision ca620d7bc03b23a0bcf0ef58df58603ee000dca0
1f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier/*
2f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier * Copyright (C) 2015 The Android Open Source Project
3f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier *
4f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier * Licensed under the Apache License, Version 2.0 (the "License");
5f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier * you may not use this file except in compliance with the License.
6f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier * You may obtain a copy of the License at
7f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier *
8f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier *      http://www.apache.org/licenses/LICENSE-2.0
9f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier *
10f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier * Unless required by applicable law or agreed to in writing, software
11f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier * distributed under the License is distributed on an "AS IS" BASIS,
12f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier * See the License for the specific language governing permissions and
14f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier * limitations under the License.
15f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier */
16f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier
17f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier#include "oat_file_manager.h"
18f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier
19f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier#include <memory>
20f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier#include <queue>
21f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier#include <vector>
22f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier
23f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier#include "base/logging.h"
24f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier#include "base/stl_util.h"
2532ce2adefb8a3d0eda59a29f5e87c1eb43eef796Mathieu Chartier#include "base/systrace.h"
26fbc31087932a65e036a153afab3049dc5298656aMathieu Chartier#include "class_linker.h"
2780b37b7e679a530738c9bcbd39873b6dacf177e5Mathieu Chartier#include "dex_file-inl.h"
2861d2b2d353ba4ab952247d2bff2c905598118bb4Mathieu Chartier#include "gc/scoped_gc_critical_section.h"
29f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier#include "gc/space/image_space.h"
30fbc31087932a65e036a153afab3049dc5298656aMathieu Chartier#include "handle_scope-inl.h"
31fbc31087932a65e036a153afab3049dc5298656aMathieu Chartier#include "mirror/class_loader.h"
32f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier#include "oat_file_assistant.h"
333398c7874e002beaa6c2b2fadf183e7d1ddad23aMathieu Chartier#include "obj_ptr-inl.h"
340795f23920ee9aabf28e45c63cd592dcccf00216Mathieu Chartier#include "scoped_thread_state_change-inl.h"
35f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier#include "thread-inl.h"
36a9d82fe8bc6960b565245b920e99107a824ca515Mathieu Chartier#include "thread_list.h"
37f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier
38f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartiernamespace art {
39f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier
40fbc31087932a65e036a153afab3049dc5298656aMathieu Chartier// If true, then we attempt to load the application image if it exists.
41fbc31087932a65e036a153afab3049dc5298656aMathieu Chartierstatic constexpr bool kEnableAppImage = true;
42fbc31087932a65e036a153afab3049dc5298656aMathieu Chartier
43f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartierconst OatFile* OatFileManager::RegisterOatFile(std::unique_ptr<const OatFile> oat_file) {
44e58991b3b2282b5761f1a6023a16c803e1c4eb45Mathieu Chartier  WriterMutexLock mu(Thread::Current(), *Locks::oat_file_manager_lock_);
45f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  DCHECK(oat_file != nullptr);
46f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  if (kIsDebugBuild) {
47e58991b3b2282b5761f1a6023a16c803e1c4eb45Mathieu Chartier    CHECK(oat_files_.find(oat_file) == oat_files_.end());
48f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    for (const std::unique_ptr<const OatFile>& existing : oat_files_) {
49f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier      CHECK_NE(oat_file.get(), existing.get()) << oat_file->GetLocation();
50f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier      // Check that we don't have an oat file with the same address. Copies of the same oat file
51f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier      // should be loaded at different addresses.
52f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier      CHECK_NE(oat_file->Begin(), existing->Begin()) << "Oat file already mapped at that location";
53f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    }
54f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  }
55f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  have_non_pic_oat_file_ = have_non_pic_oat_file_ || !oat_file->IsPic();
56e58991b3b2282b5761f1a6023a16c803e1c4eb45Mathieu Chartier  const OatFile* ret = oat_file.get();
57e58991b3b2282b5761f1a6023a16c803e1c4eb45Mathieu Chartier  oat_files_.insert(std::move(oat_file));
58e58991b3b2282b5761f1a6023a16c803e1c4eb45Mathieu Chartier  return ret;
59e58991b3b2282b5761f1a6023a16c803e1c4eb45Mathieu Chartier}
60e58991b3b2282b5761f1a6023a16c803e1c4eb45Mathieu Chartier
61e58991b3b2282b5761f1a6023a16c803e1c4eb45Mathieu Chartiervoid OatFileManager::UnRegisterAndDeleteOatFile(const OatFile* oat_file) {
62e58991b3b2282b5761f1a6023a16c803e1c4eb45Mathieu Chartier  WriterMutexLock mu(Thread::Current(), *Locks::oat_file_manager_lock_);
63e58991b3b2282b5761f1a6023a16c803e1c4eb45Mathieu Chartier  DCHECK(oat_file != nullptr);
64e58991b3b2282b5761f1a6023a16c803e1c4eb45Mathieu Chartier  std::unique_ptr<const OatFile> compare(oat_file);
65e58991b3b2282b5761f1a6023a16c803e1c4eb45Mathieu Chartier  auto it = oat_files_.find(compare);
66e58991b3b2282b5761f1a6023a16c803e1c4eb45Mathieu Chartier  CHECK(it != oat_files_.end());
67e58991b3b2282b5761f1a6023a16c803e1c4eb45Mathieu Chartier  oat_files_.erase(it);
68e58991b3b2282b5761f1a6023a16c803e1c4eb45Mathieu Chartier  compare.release();
69f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier}
70f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier
710b791279f67b0f4b72db991707b16b737871bcdeCalin Juravleconst OatFile* OatFileManager::FindOpenedOatFileFromDexLocation(
720b791279f67b0f4b72db991707b16b737871bcdeCalin Juravle    const std::string& dex_base_location) const {
730b791279f67b0f4b72db991707b16b737871bcdeCalin Juravle  ReaderMutexLock mu(Thread::Current(), *Locks::oat_file_manager_lock_);
740b791279f67b0f4b72db991707b16b737871bcdeCalin Juravle  for (const std::unique_ptr<const OatFile>& oat_file : oat_files_) {
750b791279f67b0f4b72db991707b16b737871bcdeCalin Juravle    const std::vector<const OatDexFile*>& oat_dex_files = oat_file->GetOatDexFiles();
760b791279f67b0f4b72db991707b16b737871bcdeCalin Juravle    for (const OatDexFile* oat_dex_file : oat_dex_files) {
770b791279f67b0f4b72db991707b16b737871bcdeCalin Juravle      if (DexFile::GetBaseLocation(oat_dex_file->GetDexFileLocation()) == dex_base_location) {
780b791279f67b0f4b72db991707b16b737871bcdeCalin Juravle        return oat_file.get();
790b791279f67b0f4b72db991707b16b737871bcdeCalin Juravle      }
800b791279f67b0f4b72db991707b16b737871bcdeCalin Juravle    }
810b791279f67b0f4b72db991707b16b737871bcdeCalin Juravle  }
820b791279f67b0f4b72db991707b16b737871bcdeCalin Juravle  return nullptr;
830b791279f67b0f4b72db991707b16b737871bcdeCalin Juravle}
840b791279f67b0f4b72db991707b16b737871bcdeCalin Juravle
85f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartierconst OatFile* OatFileManager::FindOpenedOatFileFromOatLocation(const std::string& oat_location)
86f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    const {
87f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  ReaderMutexLock mu(Thread::Current(), *Locks::oat_file_manager_lock_);
88e58991b3b2282b5761f1a6023a16c803e1c4eb45Mathieu Chartier  return FindOpenedOatFileFromOatLocationLocked(oat_location);
89e58991b3b2282b5761f1a6023a16c803e1c4eb45Mathieu Chartier}
90e58991b3b2282b5761f1a6023a16c803e1c4eb45Mathieu Chartier
91e58991b3b2282b5761f1a6023a16c803e1c4eb45Mathieu Chartierconst OatFile* OatFileManager::FindOpenedOatFileFromOatLocationLocked(
92e58991b3b2282b5761f1a6023a16c803e1c4eb45Mathieu Chartier    const std::string& oat_location) const {
93f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  for (const std::unique_ptr<const OatFile>& oat_file : oat_files_) {
94f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    if (oat_file->GetLocation() == oat_location) {
95f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier      return oat_file.get();
96f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    }
97f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  }
98f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  return nullptr;
99f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier}
100f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier
101dcdc85bbd569f0ee66c331b4219c19304a616214Jeff Haostd::vector<const OatFile*> OatFileManager::GetBootOatFiles() const {
102dcdc85bbd569f0ee66c331b4219c19304a616214Jeff Hao  std::vector<const OatFile*> oat_files;
103dcdc85bbd569f0ee66c331b4219c19304a616214Jeff Hao  std::vector<gc::space::ImageSpace*> image_spaces =
104dcdc85bbd569f0ee66c331b4219c19304a616214Jeff Hao      Runtime::Current()->GetHeap()->GetBootImageSpaces();
105dcdc85bbd569f0ee66c331b4219c19304a616214Jeff Hao  for (gc::space::ImageSpace* image_space : image_spaces) {
106dcdc85bbd569f0ee66c331b4219c19304a616214Jeff Hao    oat_files.push_back(image_space->GetOatFile());
107dcdc85bbd569f0ee66c331b4219c19304a616214Jeff Hao  }
108dcdc85bbd569f0ee66c331b4219c19304a616214Jeff Hao  return oat_files;
109f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier}
110f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier
111f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartierconst OatFile* OatFileManager::GetPrimaryOatFile() const {
112f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  ReaderMutexLock mu(Thread::Current(), *Locks::oat_file_manager_lock_);
113dcdc85bbd569f0ee66c331b4219c19304a616214Jeff Hao  std::vector<const OatFile*> boot_oat_files = GetBootOatFiles();
114dcdc85bbd569f0ee66c331b4219c19304a616214Jeff Hao  if (!boot_oat_files.empty()) {
115f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    for (const std::unique_ptr<const OatFile>& oat_file : oat_files_) {
116dcdc85bbd569f0ee66c331b4219c19304a616214Jeff Hao      if (std::find(boot_oat_files.begin(), boot_oat_files.end(), oat_file.get()) ==
117dcdc85bbd569f0ee66c331b4219c19304a616214Jeff Hao          boot_oat_files.end()) {
118f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier        return oat_file.get();
119f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier      }
120f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    }
121f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  }
122f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  return nullptr;
123f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier}
124f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier
125f9c6fc610b27887f832e453a0da1789187293408Mathieu ChartierOatFileManager::~OatFileManager() {
126e58991b3b2282b5761f1a6023a16c803e1c4eb45Mathieu Chartier  // Explicitly clear oat_files_ since the OatFile destructor calls back into OatFileManager for
127e58991b3b2282b5761f1a6023a16c803e1c4eb45Mathieu Chartier  // UnRegisterOatFileLocation.
128e58991b3b2282b5761f1a6023a16c803e1c4eb45Mathieu Chartier  oat_files_.clear();
129f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier}
130f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier
131dcdc85bbd569f0ee66c331b4219c19304a616214Jeff Haostd::vector<const OatFile*> OatFileManager::RegisterImageOatFiles(
132dcdc85bbd569f0ee66c331b4219c19304a616214Jeff Hao    std::vector<gc::space::ImageSpace*> spaces) {
133dcdc85bbd569f0ee66c331b4219c19304a616214Jeff Hao  std::vector<const OatFile*> oat_files;
134dcdc85bbd569f0ee66c331b4219c19304a616214Jeff Hao  for (gc::space::ImageSpace* space : spaces) {
135dcdc85bbd569f0ee66c331b4219c19304a616214Jeff Hao    oat_files.push_back(RegisterOatFile(space->ReleaseOatFile()));
136dcdc85bbd569f0ee66c331b4219c19304a616214Jeff Hao  }
137dcdc85bbd569f0ee66c331b4219c19304a616214Jeff Hao  return oat_files;
138f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier}
139f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier
140f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartierclass DexFileAndClassPair : ValueObject {
141f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier public:
142f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  DexFileAndClassPair(const DexFile* dex_file, size_t current_class_index, bool from_loaded_oat)
143f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier     : cached_descriptor_(GetClassDescriptor(dex_file, current_class_index)),
144f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier       dex_file_(dex_file),
145f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier       current_class_index_(current_class_index),
146f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier       from_loaded_oat_(from_loaded_oat) {}
147f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier
14880b37b7e679a530738c9bcbd39873b6dacf177e5Mathieu Chartier  DexFileAndClassPair(const DexFileAndClassPair& rhs) = default;
149f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier
15080b37b7e679a530738c9bcbd39873b6dacf177e5Mathieu Chartier  DexFileAndClassPair& operator=(const DexFileAndClassPair& rhs) = default;
151f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier
152f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  const char* GetCachedDescriptor() const {
153f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    return cached_descriptor_;
154f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  }
155f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier
156f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  bool operator<(const DexFileAndClassPair& rhs) const {
157f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    const int cmp = strcmp(cached_descriptor_, rhs.cached_descriptor_);
158f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    if (cmp != 0) {
159f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier      // Note that the order must be reversed. We want to iterate over the classes in dex files.
160f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier      // They are sorted lexicographically. Thus, the priority-queue must be a min-queue.
161f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier      return cmp > 0;
162f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    }
163f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    return dex_file_ < rhs.dex_file_;
164f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  }
165f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier
166f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  bool DexFileHasMoreClasses() const {
167f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    return current_class_index_ + 1 < dex_file_->NumClassDefs();
168f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  }
169f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier
170f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  void Next() {
171f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    ++current_class_index_;
172f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    cached_descriptor_ = GetClassDescriptor(dex_file_, current_class_index_);
173f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  }
174f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier
175f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  size_t GetCurrentClassIndex() const {
176f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    return current_class_index_;
177f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  }
178f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier
179f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  bool FromLoadedOat() const {
180f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    return from_loaded_oat_;
181f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  }
182f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier
183f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  const DexFile* GetDexFile() const {
184f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    return dex_file_;
185f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  }
186f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao
187f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier private:
188f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  static const char* GetClassDescriptor(const DexFile* dex_file, size_t index) {
189f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    DCHECK(IsUint<16>(index));
190f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    const DexFile::ClassDef& class_def = dex_file->GetClassDef(static_cast<uint16_t>(index));
191f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    return dex_file->StringByTypeIdx(class_def.class_idx_);
192f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  }
193f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier
194f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  const char* cached_descriptor_;
195f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  const DexFile* dex_file_;
196f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  size_t current_class_index_;
197f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  bool from_loaded_oat_;  // We only need to compare mismatches between what we load now
198f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier                          // and what was loaded before. Any old duplicates must have been
199f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier                          // OK, and any new "internal" duplicates are as well (they must
200f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier                          // be from multidex, which resolves correctly).
201f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier};
202f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier
203f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartierstatic void AddDexFilesFromOat(const OatFile* oat_file,
204f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier                               bool already_loaded,
20514d7b3e37b6129d845d6c8da8ee8d612937c63d4Mathieu Chartier                               /*out*/std::priority_queue<DexFileAndClassPair>* heap,
20614d7b3e37b6129d845d6c8da8ee8d612937c63d4Mathieu Chartier                               std::vector<std::unique_ptr<const DexFile>>* opened_dex_files) {
207f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  for (const OatDexFile* oat_dex_file : oat_file->GetOatDexFiles()) {
208f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    std::string error;
209f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    std::unique_ptr<const DexFile> dex_file = oat_dex_file->OpenDexFile(&error);
210f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    if (dex_file == nullptr) {
211f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier      LOG(WARNING) << "Could not create dex file from oat file: " << error;
212f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    } else if (dex_file->NumClassDefs() > 0U) {
21314d7b3e37b6129d845d6c8da8ee8d612937c63d4Mathieu Chartier      heap->emplace(dex_file.get(), /*current_class_index*/0U, already_loaded);
21414d7b3e37b6129d845d6c8da8ee8d612937c63d4Mathieu Chartier      opened_dex_files->push_back(std::move(dex_file));
215f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    }
216f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  }
217f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier}
218f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier
219f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartierstatic void AddNext(/*inout*/DexFileAndClassPair* original,
22014d7b3e37b6129d845d6c8da8ee8d612937c63d4Mathieu Chartier                    /*inout*/std::priority_queue<DexFileAndClassPair>* heap) {
221f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  if (original->DexFileHasMoreClasses()) {
222f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    original->Next();
223f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    heap->push(std::move(*original));
224f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  }
225f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao}
226f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao
227ca620d7bc03b23a0bcf0ef58df58603ee000dca0Andreas Gampetemplate <typename T>
2283398c7874e002beaa6c2b2fadf183e7d1ddad23aMathieu Chartierstatic void IterateOverJavaDexFile(ObjPtr<mirror::Object> dex_file,
229f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao                                   ArtField* const cookie_field,
230ca620d7bc03b23a0bcf0ef58df58603ee000dca0Andreas Gampe                                   const T& fn)
231bdf7f1c3ab65ccb70f62db5ab31dba060632d458Andreas Gampe    REQUIRES_SHARED(Locks::mutator_lock_) {
232f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  if (dex_file != nullptr) {
233f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    mirror::LongArray* long_array = cookie_field->GetObject(dex_file)->AsLongArray();
234f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    if (long_array == nullptr) {
235f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao      // This should never happen so log a warning.
236f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao      LOG(WARNING) << "Null DexFile::mCookie";
237f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao      return;
238f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    }
239f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    int32_t long_array_size = long_array->GetLength();
240f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    // Start from 1 to skip the oat file.
241f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    for (int32_t j = 1; j < long_array_size; ++j) {
242f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao      const DexFile* cp_dex_file = reinterpret_cast<const DexFile*>(static_cast<uintptr_t>(
243f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao          long_array->GetWithoutChecks(j)));
244f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao      if (!fn(cp_dex_file)) {
245f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao        return;
246f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao      }
247f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    }
248f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  }
249f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao}
250f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao
251ca620d7bc03b23a0bcf0ef58df58603ee000dca0Andreas Gampetemplate <typename T>
252f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Haostatic void IterateOverPathClassLoader(
253f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    ScopedObjectAccessAlreadyRunnable& soa,
254f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    Handle<mirror::ClassLoader> class_loader,
255f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    MutableHandle<mirror::ObjectArray<mirror::Object>> dex_elements,
256ca620d7bc03b23a0bcf0ef58df58603ee000dca0Andreas Gampe    const T& fn) REQUIRES_SHARED(Locks::mutator_lock_) {
257f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  // Handle this step.
258f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  // Handle as if this is the child PathClassLoader.
259f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  // The class loader is a PathClassLoader which inherits from BaseDexClassLoader.
260f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  // We need to get the DexPathList and loop through it.
261f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  ArtField* const cookie_field = soa.DecodeField(WellKnownClasses::dalvik_system_DexFile_cookie);
262f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  ArtField* const dex_file_field =
263f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao      soa.DecodeField(WellKnownClasses::dalvik_system_DexPathList__Element_dexFile);
2643398c7874e002beaa6c2b2fadf183e7d1ddad23aMathieu Chartier  ObjPtr<mirror::Object> dex_path_list =
26546aba36a1c982567d7e80a62b0cff9c0935e7455Andreas Gampe      soa.DecodeField(WellKnownClasses::dalvik_system_BaseDexClassLoader_pathList)->
266f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao      GetObject(class_loader.Get());
267f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  if (dex_path_list != nullptr && dex_file_field != nullptr && cookie_field != nullptr) {
268f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    // DexPathList has an array dexElements of Elements[] which each contain a dex file.
2693398c7874e002beaa6c2b2fadf183e7d1ddad23aMathieu Chartier    ObjPtr<mirror::Object> dex_elements_obj =
270f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao        soa.DecodeField(WellKnownClasses::dalvik_system_DexPathList_dexElements)->
271f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao        GetObject(dex_path_list);
272f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    // Loop through each dalvik.system.DexPathList$Element's dalvik.system.DexFile and look
273f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    // at the mCookie which is a DexFile vector.
274f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    if (dex_elements_obj != nullptr) {
275f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao      dex_elements.Assign(dex_elements_obj->AsObjectArray<mirror::Object>());
276f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao      for (int32_t i = 0; i < dex_elements->GetLength(); ++i) {
277f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao        mirror::Object* element = dex_elements->GetWithoutChecks(i);
278f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao        if (element == nullptr) {
279f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao          // Should never happen, fall back to java code to throw a NPE.
280f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao          break;
281f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao        }
2823398c7874e002beaa6c2b2fadf183e7d1ddad23aMathieu Chartier        ObjPtr<mirror::Object> dex_file = dex_file_field->GetObject(element);
283f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao        IterateOverJavaDexFile(dex_file, cookie_field, fn);
284f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao      }
285f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    }
286f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  }
287f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier}
288f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier
289f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Haostatic bool GetDexFilesFromClassLoader(
290f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    ScopedObjectAccessAlreadyRunnable& soa,
291f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    mirror::ClassLoader* class_loader,
292bdf7f1c3ab65ccb70f62db5ab31dba060632d458Andreas Gampe    std::priority_queue<DexFileAndClassPair>* queue) REQUIRES_SHARED(Locks::mutator_lock_) {
293f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  if (ClassLinker::IsBootClassLoader(soa, class_loader)) {
294f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    // The boot class loader. We don't load any of these files, as we know we compiled against
295f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    // them correctly.
296f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    return true;
297f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  }
298f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao
299f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  // Unsupported class-loader?
3000795f23920ee9aabf28e45c63cd592dcccf00216Mathieu Chartier  if (soa.Decode<mirror::Class>(WellKnownClasses::dalvik_system_PathClassLoader) !=
3010795f23920ee9aabf28e45c63cd592dcccf00216Mathieu Chartier      class_loader->GetClass()) {
302709b070044354d9f47641f273edacaeeb0240ab7David Sehr    VLOG(class_linker) << "Unsupported class-loader "
303709b070044354d9f47641f273edacaeeb0240ab7David Sehr                       << mirror::Class::PrettyClass(class_loader->GetClass());
304f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    return false;
305f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  }
306f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao
307f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  bool recursive_result = GetDexFilesFromClassLoader(soa, class_loader->GetParent(), queue);
308f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  if (!recursive_result) {
309f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    // Something wrong up the chain.
310f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    return false;
311f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  }
312f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao
313f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  // Collect all the dex files.
314f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  auto GetDexFilesFn = [&] (const DexFile* cp_dex_file)
315bdf7f1c3ab65ccb70f62db5ab31dba060632d458Andreas Gampe            REQUIRES_SHARED(Locks::mutator_lock_) {
316f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    if (cp_dex_file->NumClassDefs() > 0) {
317f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao      queue->emplace(cp_dex_file, 0U, true);
318f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    }
319f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    return true;  // Continue looking.
320f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  };
321f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao
322f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  // Handle for dex-cache-element.
323f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  StackHandleScope<3> hs(soa.Self());
324f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  MutableHandle<mirror::ObjectArray<mirror::Object>> dex_elements(
325f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao      hs.NewHandle<mirror::ObjectArray<mirror::Object>>(nullptr));
326f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  Handle<mirror::ClassLoader> h_class_loader(hs.NewHandle(class_loader));
327f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao
328f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  IterateOverPathClassLoader(soa, h_class_loader, dex_elements, GetDexFilesFn);
329f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao
330f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  return true;
331f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao}
332f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao
333f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Haostatic void GetDexFilesFromDexElementsArray(
334f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    ScopedObjectAccessAlreadyRunnable& soa,
335f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    Handle<mirror::ObjectArray<mirror::Object>> dex_elements,
336bdf7f1c3ab65ccb70f62db5ab31dba060632d458Andreas Gampe    std::priority_queue<DexFileAndClassPair>* queue) REQUIRES_SHARED(Locks::mutator_lock_) {
337f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  if (dex_elements.Get() == nullptr) {
338f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    // Nothing to do.
339f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    return;
340f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  }
341f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao
342f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  ArtField* const cookie_field = soa.DecodeField(WellKnownClasses::dalvik_system_DexFile_cookie);
343f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  ArtField* const dex_file_field =
344f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao      soa.DecodeField(WellKnownClasses::dalvik_system_DexPathList__Element_dexFile);
3450795f23920ee9aabf28e45c63cd592dcccf00216Mathieu Chartier  ObjPtr<mirror::Class> const element_class = soa.Decode<mirror::Class>(
346f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao      WellKnownClasses::dalvik_system_DexPathList__Element);
3470795f23920ee9aabf28e45c63cd592dcccf00216Mathieu Chartier  ObjPtr<mirror::Class> const dexfile_class = soa.Decode<mirror::Class>(
3480795f23920ee9aabf28e45c63cd592dcccf00216Mathieu Chartier      WellKnownClasses::dalvik_system_DexFile);
349f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao
350f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  // Collect all the dex files.
351f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  auto GetDexFilesFn = [&] (const DexFile* cp_dex_file)
352bdf7f1c3ab65ccb70f62db5ab31dba060632d458Andreas Gampe      REQUIRES_SHARED(Locks::mutator_lock_) {
353f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    if (cp_dex_file != nullptr && cp_dex_file->NumClassDefs() > 0) {
354f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao      queue->emplace(cp_dex_file, 0U, true);
355f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    }
356f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    return true;  // Continue looking.
357f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  };
358f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao
359f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  for (int32_t i = 0; i < dex_elements->GetLength(); ++i) {
360f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    mirror::Object* element = dex_elements->GetWithoutChecks(i);
361f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    if (element == nullptr) {
362f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao      continue;
363f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    }
364f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao
365f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    // We support this being dalvik.system.DexPathList$Element and dalvik.system.DexFile.
366f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao
3673398c7874e002beaa6c2b2fadf183e7d1ddad23aMathieu Chartier    ObjPtr<mirror::Object> dex_file;
3680795f23920ee9aabf28e45c63cd592dcccf00216Mathieu Chartier    if (element_class == element->GetClass()) {
369f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao      dex_file = dex_file_field->GetObject(element);
3700795f23920ee9aabf28e45c63cd592dcccf00216Mathieu Chartier    } else if (dexfile_class == element->GetClass()) {
371f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao      dex_file = element;
372f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    } else {
373709b070044354d9f47641f273edacaeeb0240ab7David Sehr      LOG(WARNING) << "Unsupported element in dex_elements: "
374709b070044354d9f47641f273edacaeeb0240ab7David Sehr                   << mirror::Class::PrettyClass(element->GetClass());
375f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao      continue;
376f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    }
377f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao
378f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    IterateOverJavaDexFile(dex_file, cookie_field, GetDexFilesFn);
379f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  }
380f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao}
381f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao
3825c657fe8e3afdce77dc610446f37a43865cff702Vladimir Markostatic bool AreSharedLibrariesOk(const std::string& shared_libraries,
383f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao                                 std::priority_queue<DexFileAndClassPair>& queue) {
384f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  if (shared_libraries.empty()) {
385f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    if (queue.empty()) {
386f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao      // No shared libraries or oat files, as expected.
387f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao      return true;
388f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    }
389f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  } else {
390f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    if (shared_libraries.compare(OatFile::kSpecialSharedLibrary) == 0) {
391f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao      // If we find the special shared library, skip the shared libraries check.
392f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao      return true;
393f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    }
394f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    // Shared libraries is a series of dex file paths and their checksums, each separated by '*'.
395f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    std::vector<std::string> shared_libraries_split;
396f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    Split(shared_libraries, '*', &shared_libraries_split);
397f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao
398f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    size_t index = 0;
399f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    std::priority_queue<DexFileAndClassPair> temp = queue;
400f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    while (!temp.empty() && index < shared_libraries_split.size() - 1) {
401f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao      DexFileAndClassPair pair(temp.top());
402f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao      const DexFile* dex_file = pair.GetDexFile();
4034c481a4245aebe6821914237249f11d03e687bf9Andreas Gampe      const std::string& dex_filename = dex_file->GetLocation();
404f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao      uint32_t dex_checksum = dex_file->GetLocationChecksum();
405f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao      if (dex_filename != shared_libraries_split[index] ||
406f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao          dex_checksum != std::stoul(shared_libraries_split[index + 1])) {
407f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao        break;
408f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao      }
409f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao      temp.pop();
410f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao      index += 2;
411f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    }
412f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao
413f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    // Check is successful if it made it through the queue and all the shared libraries.
414f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    return temp.empty() && index == shared_libraries_split.size();
415f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  }
416f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  return false;
417f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao}
418f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao
419f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier// Check for class-def collisions in dex files.
420f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier//
421f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao// This first walks the class loader chain, getting all the dex files from the class loader. If
422f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao// the class loader is null or one of the class loaders in the chain is unsupported, we collect
423f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao// dex files from all open non-boot oat files to be safe.
424f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao//
425f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao// This first checks whether the shared libraries are in the expected order and the oat files
426f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao// have the expected checksums. If so, we exit early. Otherwise, we do the collision check.
427f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao//
428f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao// The collision check works by maintaining a heap with one class from each dex file, sorted by the
429f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao// class descriptor. Then a dex-file/class pair is continually removed from the heap and compared
430f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier// against the following top element. If the descriptor is the same, it is now checked whether
431f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier// the two elements agree on whether their dex file was from an already-loaded oat-file or the
432f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier// new oat file. Any disagreement indicates a collision.
433f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartierbool OatFileManager::HasCollisions(const OatFile* oat_file,
434f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao                                   jobject class_loader,
435f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao                                   jobjectArray dex_elements,
436f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier                                   std::string* error_msg /*out*/) const {
437f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  DCHECK(oat_file != nullptr);
438f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  DCHECK(error_msg != nullptr);
439f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao
440f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  std::priority_queue<DexFileAndClassPair> queue;
441f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao
442f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  // Try to get dex files from the given class loader. If the class loader is null, or we do
443f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  // not support one of the class loaders in the chain, conservatively compare against all
444f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  // (non-boot) oat files.
445f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  bool class_loader_ok = false;
446f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  {
447f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    ScopedObjectAccess soa(Thread::Current());
448f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    StackHandleScope<2> hs(Thread::Current());
449f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    Handle<mirror::ClassLoader> h_class_loader =
4500795f23920ee9aabf28e45c63cd592dcccf00216Mathieu Chartier        hs.NewHandle(soa.Decode<mirror::ClassLoader>(class_loader));
451f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    Handle<mirror::ObjectArray<mirror::Object>> h_dex_elements =
4520795f23920ee9aabf28e45c63cd592dcccf00216Mathieu Chartier        hs.NewHandle(soa.Decode<mirror::ObjectArray<mirror::Object>>(dex_elements));
453f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    if (h_class_loader.Get() != nullptr &&
454f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao        GetDexFilesFromClassLoader(soa, h_class_loader.Get(), &queue)) {
455f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao      class_loader_ok = true;
456f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao
457f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao      // In this case, also take into account the dex_elements array, if given. We don't need to
458f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao      // read it otherwise, as we'll compare against all open oat files anyways.
459f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao      GetDexFilesFromDexElementsArray(soa, h_dex_elements, &queue);
460f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    } else if (h_class_loader.Get() != nullptr) {
461f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao      VLOG(class_linker) << "Something unsupported with "
462709b070044354d9f47641f273edacaeeb0240ab7David Sehr                         << mirror::Class::PrettyClass(h_class_loader->GetClass());
463f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    }
464f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  }
465f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier
466f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  // Dex files are registered late - once a class is actually being loaded. We have to compare
467f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  // against the open oat files. Take the oat_file_manager_lock_ that protects oat_files_ accesses.
468f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  ReaderMutexLock mu(Thread::Current(), *Locks::oat_file_manager_lock_);
469f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier
47014d7b3e37b6129d845d6c8da8ee8d612937c63d4Mathieu Chartier  // Vector that holds the newly opened dex files live, this is done to prevent leaks.
47114d7b3e37b6129d845d6c8da8ee8d612937c63d4Mathieu Chartier  std::vector<std::unique_ptr<const DexFile>> opened_dex_files;
47214d7b3e37b6129d845d6c8da8ee8d612937c63d4Mathieu Chartier
473f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  if (!class_loader_ok) {
474f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    // Add dex files from already loaded oat files, but skip boot.
475f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier
476f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    // Clean up the queue.
477f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    while (!queue.empty()) {
478f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao      queue.pop();
479f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    }
480f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao
481f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    std::vector<const OatFile*> boot_oat_files = GetBootOatFiles();
482f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    // The same OatFile can be loaded multiple times at different addresses. In this case, we don't
483f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    // need to check both against each other since they would have resolved the same way at compile
484f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    // time.
485f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    std::unordered_set<std::string> unique_locations;
486f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    for (const std::unique_ptr<const OatFile>& loaded_oat_file : oat_files_) {
487f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao      DCHECK_NE(loaded_oat_file.get(), oat_file);
488f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao      const std::string& location = loaded_oat_file->GetLocation();
489f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao      if (std::find(boot_oat_files.begin(), boot_oat_files.end(), loaded_oat_file.get()) ==
490f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao          boot_oat_files.end() && location != oat_file->GetLocation() &&
491f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao          unique_locations.find(location) == unique_locations.end()) {
492f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao        unique_locations.insert(location);
49314d7b3e37b6129d845d6c8da8ee8d612937c63d4Mathieu Chartier        AddDexFilesFromOat(loaded_oat_file.get(),
49414d7b3e37b6129d845d6c8da8ee8d612937c63d4Mathieu Chartier                           /*already_loaded*/true,
49514d7b3e37b6129d845d6c8da8ee8d612937c63d4Mathieu Chartier                           &queue,
49614d7b3e37b6129d845d6c8da8ee8d612937c63d4Mathieu Chartier                           /*out*/&opened_dex_files);
497f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao      }
498f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    }
499f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  }
500f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier
501f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  // Exit if shared libraries are ok. Do a full duplicate classes check otherwise.
502f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  const std::string
503f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao      shared_libraries(oat_file->GetOatHeader().GetStoreValueByKey(OatHeader::kClassPathKey));
504f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao  if (AreSharedLibrariesOk(shared_libraries, queue)) {
505f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    return false;
506f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  }
507f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier
5081fdbe1b034d97b852551d24e70245df64f03ebebAndreas Gampe  ScopedTrace st("Collision check");
5091fdbe1b034d97b852551d24e70245df64f03ebebAndreas Gampe
510f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  // Add dex files from the oat file to check.
51114d7b3e37b6129d845d6c8da8ee8d612937c63d4Mathieu Chartier  AddDexFilesFromOat(oat_file, /*already_loaded*/false, &queue, &opened_dex_files);
512f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier
513f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  // Now drain the queue.
514f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  while (!queue.empty()) {
515f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    // Modifying the top element is only safe if we pop right after.
51680b37b7e679a530738c9bcbd39873b6dacf177e5Mathieu Chartier    DexFileAndClassPair compare_pop(queue.top());
517f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    queue.pop();
518f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier
519f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    // Compare against the following elements.
520f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    while (!queue.empty()) {
52180b37b7e679a530738c9bcbd39873b6dacf177e5Mathieu Chartier      DexFileAndClassPair top(queue.top());
522f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier
523f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier      if (strcmp(compare_pop.GetCachedDescriptor(), top.GetCachedDescriptor()) == 0) {
524f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier        // Same descriptor. Check whether it's crossing old-oat-files to new-oat-files.
525f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier        if (compare_pop.FromLoadedOat() != top.FromLoadedOat()) {
526f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier          *error_msg =
527f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier              StringPrintf("Found duplicated class when checking oat files: '%s' in %s and %s",
528f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier                           compare_pop.GetCachedDescriptor(),
529f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier                           compare_pop.GetDexFile()->GetLocation().c_str(),
530f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier                           top.GetDexFile()->GetLocation().c_str());
531f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier          return true;
532f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier        }
533f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier        queue.pop();
53414d7b3e37b6129d845d6c8da8ee8d612937c63d4Mathieu Chartier        AddNext(&top, &queue);
535f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier      } else {
536f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier        // Something else. Done here.
537f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier        break;
538f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier      }
539f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    }
54014d7b3e37b6129d845d6c8da8ee8d612937c63d4Mathieu Chartier    AddNext(&compare_pop, &queue);
541f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  }
542f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier
543f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  return false;
544f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier}
545f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier
546f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartierstd::vector<std::unique_ptr<const DexFile>> OatFileManager::OpenDexFilesFromOat(
547f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    const char* dex_location,
548f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    const char* oat_location,
549fbc31087932a65e036a153afab3049dc5298656aMathieu Chartier    jobject class_loader,
550fbc31087932a65e036a153afab3049dc5298656aMathieu Chartier    jobjectArray dex_elements,
551e58991b3b2282b5761f1a6023a16c803e1c4eb45Mathieu Chartier    const OatFile** out_oat_file,
552f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    std::vector<std::string>* error_msgs) {
55332ce2adefb8a3d0eda59a29f5e87c1eb43eef796Mathieu Chartier  ScopedTrace trace(__FUNCTION__);
554f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  CHECK(dex_location != nullptr);
555f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  CHECK(error_msgs != nullptr);
556f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier
557f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  // Verify we aren't holding the mutator lock, which could starve GC if we
558f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  // have to generate or relocate an oat file.
559fbc31087932a65e036a153afab3049dc5298656aMathieu Chartier  Thread* const self = Thread::Current();
560fbc31087932a65e036a153afab3049dc5298656aMathieu Chartier  Locks::mutator_lock_->AssertNotHeld(self);
561fbc31087932a65e036a153afab3049dc5298656aMathieu Chartier  Runtime* const runtime = Runtime::Current();
562b077e15d2d11b7c81aacbcd4a46c2b1e9c9ba20dCalin Juravle
563f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  OatFileAssistant oat_file_assistant(dex_location,
564f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier                                      oat_location,
565f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier                                      kRuntimeISA,
566fbc31087932a65e036a153afab3049dc5298656aMathieu Chartier                                      !runtime->IsAotCompiler());
567f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier
568f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  // Lock the target oat location to avoid races generating and loading the
569f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  // oat file.
570f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  std::string error_msg;
571f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  if (!oat_file_assistant.Lock(/*out*/&error_msg)) {
572f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    // Don't worry too much if this fails. If it does fail, it's unlikely we
573f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    // can generate an oat file anyway.
574f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    VLOG(class_linker) << "OatFileAssistant::Lock: " << error_msg;
575f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  }
576f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier
577f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  const OatFile* source_oat_file = nullptr;
578f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier
57901be68100e96a44b4ff8cb4cb3c9baa346ff048dRichard Uhler  if (!oat_file_assistant.IsUpToDate()) {
58001be68100e96a44b4ff8cb4cb3c9baa346ff048dRichard Uhler    // Update the oat file on disk if we can, based on the --compiler-filter
58101be68100e96a44b4ff8cb4cb3c9baa346ff048dRichard Uhler    // option derived from the current runtime options.
58201be68100e96a44b4ff8cb4cb3c9baa346ff048dRichard Uhler    // This may fail, but that's okay. Best effort is all that matters here.
583d1472a27e856ec68ab5a27abaaebe9c61d91d343Richard Uhler    switch (oat_file_assistant.MakeUpToDate(/*profile_changed*/false, /*out*/ &error_msg)) {
58401be68100e96a44b4ff8cb4cb3c9baa346ff048dRichard Uhler      case OatFileAssistant::kUpdateFailed:
58501be68100e96a44b4ff8cb4cb3c9baa346ff048dRichard Uhler        LOG(WARNING) << error_msg;
58601be68100e96a44b4ff8cb4cb3c9baa346ff048dRichard Uhler        break;
58701be68100e96a44b4ff8cb4cb3c9baa346ff048dRichard Uhler
58801be68100e96a44b4ff8cb4cb3c9baa346ff048dRichard Uhler      case OatFileAssistant::kUpdateNotAttempted:
58901be68100e96a44b4ff8cb4cb3c9baa346ff048dRichard Uhler        // Avoid spamming the logs if we decided not to attempt making the oat
59001be68100e96a44b4ff8cb4cb3c9baa346ff048dRichard Uhler        // file up to date.
59101be68100e96a44b4ff8cb4cb3c9baa346ff048dRichard Uhler        VLOG(oat) << error_msg;
59201be68100e96a44b4ff8cb4cb3c9baa346ff048dRichard Uhler        break;
59301be68100e96a44b4ff8cb4cb3c9baa346ff048dRichard Uhler
59401be68100e96a44b4ff8cb4cb3c9baa346ff048dRichard Uhler      case OatFileAssistant::kUpdateSucceeded:
59501be68100e96a44b4ff8cb4cb3c9baa346ff048dRichard Uhler        // Nothing to do.
59601be68100e96a44b4ff8cb4cb3c9baa346ff048dRichard Uhler        break;
59701be68100e96a44b4ff8cb4cb3c9baa346ff048dRichard Uhler    }
598f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  }
599f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier
600f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  // Get the oat file on disk.
601f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  std::unique_ptr<const OatFile> oat_file(oat_file_assistant.GetBestOatFile().release());
602fbc31087932a65e036a153afab3049dc5298656aMathieu Chartier
603f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  if (oat_file != nullptr) {
604f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    // Take the file only if it has no collisions, or we must take it because of preopting.
605f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao    bool accept_oat_file =
606f0192c86a58b2f43378c9a2113007538dd38ddbfJeff Hao        !HasCollisions(oat_file.get(), class_loader, dex_elements, /*out*/ &error_msg);
607f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    if (!accept_oat_file) {
608f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier      // Failed the collision check. Print warning.
609f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier      if (Runtime::Current()->IsDexFileFallbackEnabled()) {
610f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier        LOG(WARNING) << "Found duplicate classes, falling back to interpreter mode for "
611f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier                     << dex_location;
612f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier      } else {
613f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier        LOG(WARNING) << "Found duplicate classes, dex-file-fallback disabled, will be failing to "
614f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier                        " load classes for " << dex_location;
615f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier      }
616f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier      LOG(WARNING) << error_msg;
617f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier
618f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier      // However, if the app was part of /system and preopted, there is no original dex file
619f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier      // available. In that case grudgingly accept the oat file.
62076f5cb6026b9c83bf00fd1007703503a327f704dRichard Uhler      if (!oat_file_assistant.HasOriginalDexFiles()) {
621f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier        accept_oat_file = true;
622f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier        LOG(WARNING) << "Dex location " << dex_location << " does not seem to include dex file. "
623f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier                     << "Allow oat file use. This is potentially dangerous.";
624f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier      }
625f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    }
626f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier
627f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    if (accept_oat_file) {
628f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier      VLOG(class_linker) << "Registering " << oat_file->GetLocation();
629f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier      source_oat_file = RegisterOatFile(std::move(oat_file));
630e58991b3b2282b5761f1a6023a16c803e1c4eb45Mathieu Chartier      *out_oat_file = source_oat_file;
631f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    }
632f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  }
633f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier
634f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  std::vector<std::unique_ptr<const DexFile>> dex_files;
635f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier
636f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  // Load the dex files from the oat file.
637f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  if (source_oat_file != nullptr) {
638fbc31087932a65e036a153afab3049dc5298656aMathieu Chartier    bool added_image_space = false;
639fbc31087932a65e036a153afab3049dc5298656aMathieu Chartier    if (source_oat_file->IsExecutable()) {
640a463b6a920a2a0bf14f9cca20a561b412b9349d1Andreas Gampe      std::unique_ptr<gc::space::ImageSpace> image_space =
641a463b6a920a2a0bf14f9cca20a561b412b9349d1Andreas Gampe          kEnableAppImage ? oat_file_assistant.OpenImageSpace(source_oat_file) : nullptr;
642fbc31087932a65e036a153afab3049dc5298656aMathieu Chartier      if (image_space != nullptr) {
643fbc31087932a65e036a153afab3049dc5298656aMathieu Chartier        ScopedObjectAccess soa(self);
644fbc31087932a65e036a153afab3049dc5298656aMathieu Chartier        StackHandleScope<1> hs(self);
645fbc31087932a65e036a153afab3049dc5298656aMathieu Chartier        Handle<mirror::ClassLoader> h_loader(
6460795f23920ee9aabf28e45c63cd592dcccf00216Mathieu Chartier            hs.NewHandle(soa.Decode<mirror::ClassLoader>(class_loader)));
647fbc31087932a65e036a153afab3049dc5298656aMathieu Chartier        // Can not load app image without class loader.
648fbc31087932a65e036a153afab3049dc5298656aMathieu Chartier        if (h_loader.Get() != nullptr) {
649fbc31087932a65e036a153afab3049dc5298656aMathieu Chartier          std::string temp_error_msg;
650fbc31087932a65e036a153afab3049dc5298656aMathieu Chartier          // Add image space has a race condition since other threads could be reading from the
651fbc31087932a65e036a153afab3049dc5298656aMathieu Chartier          // spaces array.
652a9d82fe8bc6960b565245b920e99107a824ca515Mathieu Chartier          {
653a9d82fe8bc6960b565245b920e99107a824ca515Mathieu Chartier            ScopedThreadSuspension sts(self, kSuspended);
65461d2b2d353ba4ab952247d2bff2c905598118bb4Mathieu Chartier            gc::ScopedGCCriticalSection gcs(self,
65561d2b2d353ba4ab952247d2bff2c905598118bb4Mathieu Chartier                                            gc::kGcCauseAddRemoveAppImageSpace,
65661d2b2d353ba4ab952247d2bff2c905598118bb4Mathieu Chartier                                            gc::kCollectorTypeAddRemoveAppImageSpace);
657a9d82fe8bc6960b565245b920e99107a824ca515Mathieu Chartier            ScopedSuspendAll ssa("Add image space");
658a9d82fe8bc6960b565245b920e99107a824ca515Mathieu Chartier            runtime->GetHeap()->AddSpace(image_space.get());
659a9d82fe8bc6960b565245b920e99107a824ca515Mathieu Chartier          }
66032ce2adefb8a3d0eda59a29f5e87c1eb43eef796Mathieu Chartier          {
66132ce2adefb8a3d0eda59a29f5e87c1eb43eef796Mathieu Chartier            ScopedTrace trace2(StringPrintf("Adding image space for location %s", dex_location));
66232ce2adefb8a3d0eda59a29f5e87c1eb43eef796Mathieu Chartier            added_image_space = runtime->GetClassLinker()->AddImageSpace(image_space.get(),
66332ce2adefb8a3d0eda59a29f5e87c1eb43eef796Mathieu Chartier                                                                         h_loader,
66432ce2adefb8a3d0eda59a29f5e87c1eb43eef796Mathieu Chartier                                                                         dex_elements,
66532ce2adefb8a3d0eda59a29f5e87c1eb43eef796Mathieu Chartier                                                                         dex_location,
66632ce2adefb8a3d0eda59a29f5e87c1eb43eef796Mathieu Chartier                                                                         /*out*/&dex_files,
66732ce2adefb8a3d0eda59a29f5e87c1eb43eef796Mathieu Chartier                                                                         /*out*/&temp_error_msg);
66832ce2adefb8a3d0eda59a29f5e87c1eb43eef796Mathieu Chartier          }
669a6e81ed4c185b7362cd5199ebe5507d00883a9b0Mathieu Chartier          if (added_image_space) {
670bd064ea2269b23360e32e8139c22d5993ddc385bMathieu Chartier            // Successfully added image space to heap, release the map so that it does not get
671bd064ea2269b23360e32e8139c22d5993ddc385bMathieu Chartier            // freed.
672bd064ea2269b23360e32e8139c22d5993ddc385bMathieu Chartier            image_space.release();
673bd064ea2269b23360e32e8139c22d5993ddc385bMathieu Chartier          } else {
674fbc31087932a65e036a153afab3049dc5298656aMathieu Chartier            LOG(INFO) << "Failed to add image file " << temp_error_msg;
675fbc31087932a65e036a153afab3049dc5298656aMathieu Chartier            dex_files.clear();
676a9d82fe8bc6960b565245b920e99107a824ca515Mathieu Chartier            {
677a9d82fe8bc6960b565245b920e99107a824ca515Mathieu Chartier              ScopedThreadSuspension sts(self, kSuspended);
67861d2b2d353ba4ab952247d2bff2c905598118bb4Mathieu Chartier              gc::ScopedGCCriticalSection gcs(self,
67961d2b2d353ba4ab952247d2bff2c905598118bb4Mathieu Chartier                                              gc::kGcCauseAddRemoveAppImageSpace,
68061d2b2d353ba4ab952247d2bff2c905598118bb4Mathieu Chartier                                              gc::kCollectorTypeAddRemoveAppImageSpace);
681a9d82fe8bc6960b565245b920e99107a824ca515Mathieu Chartier              ScopedSuspendAll ssa("Remove image space");
682a9d82fe8bc6960b565245b920e99107a824ca515Mathieu Chartier              runtime->GetHeap()->RemoveSpace(image_space.get());
683a9d82fe8bc6960b565245b920e99107a824ca515Mathieu Chartier            }
684fbc31087932a65e036a153afab3049dc5298656aMathieu Chartier            // Non-fatal, don't update error_msg.
685fbc31087932a65e036a153afab3049dc5298656aMathieu Chartier          }
686fbc31087932a65e036a153afab3049dc5298656aMathieu Chartier        }
687fbc31087932a65e036a153afab3049dc5298656aMathieu Chartier      }
688fbc31087932a65e036a153afab3049dc5298656aMathieu Chartier    }
689fbc31087932a65e036a153afab3049dc5298656aMathieu Chartier    if (!added_image_space) {
690fbc31087932a65e036a153afab3049dc5298656aMathieu Chartier      DCHECK(dex_files.empty());
691fbc31087932a65e036a153afab3049dc5298656aMathieu Chartier      dex_files = oat_file_assistant.LoadDexFiles(*source_oat_file, dex_location);
692fbc31087932a65e036a153afab3049dc5298656aMathieu Chartier    }
693f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    if (dex_files.empty()) {
694f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier      error_msgs->push_back("Failed to open dex files from " + source_oat_file->GetLocation());
695f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    }
696f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  }
697f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier
698f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  // Fall back to running out of the original dex file if we couldn't load any
699f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  // dex_files from the oat file.
700f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  if (dex_files.empty()) {
701f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    if (oat_file_assistant.HasOriginalDexFiles()) {
702f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier      if (Runtime::Current()->IsDexFileFallbackEnabled()) {
70337d6a3bbf5826abb73f872ed0bc4c6ae9553e274Aart Bik        static constexpr bool kVerifyChecksum = true;
70437d6a3bbf5826abb73f872ed0bc4c6ae9553e274Aart Bik        if (!DexFile::Open(
70537d6a3bbf5826abb73f872ed0bc4c6ae9553e274Aart Bik            dex_location, dex_location, kVerifyChecksum, /*out*/ &error_msg, &dex_files)) {
706f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier          LOG(WARNING) << error_msg;
7073045b66613404fa973aafc8c2aae3728e9c12d9aAlex Light          error_msgs->push_back("Failed to open dex files from " + std::string(dex_location)
7083045b66613404fa973aafc8c2aae3728e9c12d9aAlex Light                                + " because: " + error_msg);
709f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier        }
710f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier      } else {
711f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier        error_msgs->push_back("Fallback mode disabled, skipping dex files.");
712f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier      }
713f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    } else {
714f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier      error_msgs->push_back("No original dex files found for dex location "
715f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier          + std::string(dex_location));
716f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier    }
717f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  }
718c90bc92bc577020ff4d3caced4cee1cdf41fa5deCalin Juravle
719c90bc92bc577020ff4d3caced4cee1cdf41fa5deCalin Juravle  // TODO(calin): Consider optimizing this knowing that is useless to record the
720c90bc92bc577020ff4d3caced4cee1cdf41fa5deCalin Juravle  // use of fully compiled apks.
721c90bc92bc577020ff4d3caced4cee1cdf41fa5deCalin Juravle  Runtime::Current()->NotifyDexLoaded(dex_location);
722f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier  return dex_files;
723f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier}
724f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier
72504680f3d7b021a0afb460266d442f564186a3b6fNicolas Geoffrayvoid OatFileManager::DumpForSigQuit(std::ostream& os) {
72604680f3d7b021a0afb460266d442f564186a3b6fNicolas Geoffray  ReaderMutexLock mu(Thread::Current(), *Locks::oat_file_manager_lock_);
72704680f3d7b021a0afb460266d442f564186a3b6fNicolas Geoffray  std::vector<const OatFile*> boot_oat_files = GetBootOatFiles();
72804680f3d7b021a0afb460266d442f564186a3b6fNicolas Geoffray  for (const std::unique_ptr<const OatFile>& oat_file : oat_files_) {
72904680f3d7b021a0afb460266d442f564186a3b6fNicolas Geoffray    if (ContainsElement(boot_oat_files, oat_file.get())) {
73004680f3d7b021a0afb460266d442f564186a3b6fNicolas Geoffray      continue;
73104680f3d7b021a0afb460266d442f564186a3b6fNicolas Geoffray    }
73229d38e77c553c6cf71fc4dafe2d22b4e3f814872Andreas Gampe    os << oat_file->GetLocation() << ": " << oat_file->GetCompilerFilter() << "\n";
73304680f3d7b021a0afb460266d442f564186a3b6fNicolas Geoffray  }
73404680f3d7b021a0afb460266d442f564186a3b6fNicolas Geoffray}
73504680f3d7b021a0afb460266d442f564186a3b6fNicolas Geoffray
736f9c6fc610b27887f832e453a0da1789187293408Mathieu Chartier}  // namespace art
737