blob: f8f930b1ddc92fb3f145dd4b4eb365e89a647815 [file] [log] [blame]
Elliott Hughes2faa5f12012-01-30 14:42:07 -08001/*
2 * Copyright (C) 2011 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
Brian Carlstrom4a289ed2011-08-16 17:17:49 -070016
17#include "image.h"
18
Vladimir Marko80afd022015-05-19 18:08:00 +010019#include "base/bit_utils.h"
Ian Rogers2dd0e2c2013-01-24 12:42:14 -080020#include "mirror/object_array.h"
21#include "mirror/object_array-inl.h"
Ian Rogers4f6ad8a2013-03-18 15:27:28 -070022#include "mirror/object-inl.h"
Ian Rogers2dd0e2c2013-01-24 12:42:14 -080023
Brian Carlstrom4a289ed2011-08-16 17:17:49 -070024namespace art {
25
Ian Rogers13735952014-10-08 12:43:28 -070026const uint8_t ImageHeader::kImageMagic[] = { 'a', 'r', 't', '\n' };
Mathieu Chartierceb07b32015-12-10 09:33:21 -080027const uint8_t ImageHeader::kImageVersion[] = { '0', '2', '4', '\0' };
Brian Carlstrom4a289ed2011-08-16 17:17:49 -070028
Ian Rogers2dd0e2c2013-01-24 12:42:14 -080029ImageHeader::ImageHeader(uint32_t image_begin,
Mathieu Chartier31e89252013-08-28 11:29:12 -070030 uint32_t image_size,
Mathieu Chartiere401d142015-04-22 13:56:20 -070031 ImageSection* sections,
Ian Rogers2dd0e2c2013-01-24 12:42:14 -080032 uint32_t image_roots,
33 uint32_t oat_checksum,
34 uint32_t oat_file_begin,
35 uint32_t oat_data_begin,
36 uint32_t oat_data_end,
Igor Murashkin46774762014-10-22 11:37:02 -070037 uint32_t oat_file_end,
Mathieu Chartiere401d142015-04-22 13:56:20 -070038 uint32_t pointer_size,
Mathieu Chartierceb07b32015-12-10 09:33:21 -080039 bool compile_pic,
40 StorageMode storage_mode,
41 size_t data_size)
Ian Rogers2dd0e2c2013-01-24 12:42:14 -080042 : image_begin_(image_begin),
Mathieu Chartier31e89252013-08-28 11:29:12 -070043 image_size_(image_size),
Ian Rogers2dd0e2c2013-01-24 12:42:14 -080044 oat_checksum_(oat_checksum),
45 oat_file_begin_(oat_file_begin),
46 oat_data_begin_(oat_data_begin),
47 oat_data_end_(oat_data_end),
48 oat_file_end_(oat_file_end),
Alex Light53cb16b2014-06-12 11:26:29 -070049 patch_delta_(0),
Igor Murashkin46774762014-10-22 11:37:02 -070050 image_roots_(image_roots),
Mathieu Chartiere401d142015-04-22 13:56:20 -070051 pointer_size_(pointer_size),
Mathieu Chartierceb07b32015-12-10 09:33:21 -080052 compile_pic_(compile_pic),
53 storage_mode_(storage_mode),
54 data_size_(data_size) {
Ian Rogers2dd0e2c2013-01-24 12:42:14 -080055 CHECK_EQ(image_begin, RoundUp(image_begin, kPageSize));
56 CHECK_EQ(oat_file_begin, RoundUp(oat_file_begin, kPageSize));
57 CHECK_EQ(oat_data_begin, RoundUp(oat_data_begin, kPageSize));
58 CHECK_LT(image_begin, image_roots);
59 CHECK_LT(image_roots, oat_file_begin);
60 CHECK_LE(oat_file_begin, oat_data_begin);
61 CHECK_LT(oat_data_begin, oat_data_end);
62 CHECK_LE(oat_data_end, oat_file_end);
Mathieu Chartiere401d142015-04-22 13:56:20 -070063 CHECK(ValidPointerSize(pointer_size_)) << pointer_size_;
Ian Rogers2dd0e2c2013-01-24 12:42:14 -080064 memcpy(magic_, kImageMagic, sizeof(kImageMagic));
65 memcpy(version_, kImageVersion, sizeof(kImageVersion));
Mathieu Chartiere401d142015-04-22 13:56:20 -070066 std::copy_n(sections, kSectionCount, sections_);
Ian Rogers2dd0e2c2013-01-24 12:42:14 -080067}
68
Alex Light53cb16b2014-06-12 11:26:29 -070069void ImageHeader::RelocateImage(off_t delta) {
70 CHECK_ALIGNED(delta, kPageSize) << " patch delta must be page aligned";
71 image_begin_ += delta;
72 oat_file_begin_ += delta;
73 oat_data_begin_ += delta;
74 oat_data_end_ += delta;
75 oat_file_end_ += delta;
76 image_roots_ += delta;
77 patch_delta_ += delta;
Mathieu Chartiere401d142015-04-22 13:56:20 -070078 for (size_t i = 0; i < kImageMethodsCount; ++i) {
79 image_methods_[i] += delta;
80 }
Alex Light53cb16b2014-06-12 11:26:29 -070081}
82
Brian Carlstrom68708f52013-09-03 14:15:31 -070083bool ImageHeader::IsValid() const {
84 if (memcmp(magic_, kImageMagic, sizeof(kImageMagic)) != 0) {
85 return false;
86 }
87 if (memcmp(version_, kImageVersion, sizeof(kImageVersion)) != 0) {
88 return false;
89 }
Alex Light53cb16b2014-06-12 11:26:29 -070090 // Unsigned so wraparound is well defined.
91 if (image_begin_ >= image_begin_ + image_size_) {
92 return false;
93 }
94 if (oat_file_begin_ > oat_file_end_) {
95 return false;
96 }
97 if (oat_data_begin_ > oat_data_end_) {
98 return false;
99 }
100 if (oat_file_begin_ >= oat_data_begin_) {
101 return false;
102 }
103 if (image_roots_ <= image_begin_ || oat_file_begin_ <= image_roots_) {
104 return false;
105 }
106 if (!IsAligned<kPageSize>(patch_delta_)) {
107 return false;
108 }
Brian Carlstrom68708f52013-09-03 14:15:31 -0700109 return true;
110}
111
112const char* ImageHeader::GetMagic() const {
113 CHECK(IsValid());
114 return reinterpret_cast<const char*>(magic_);
115}
116
Ian Rogers2dd0e2c2013-01-24 12:42:14 -0800117mirror::Object* ImageHeader::GetImageRoot(ImageRoot image_root) const {
118 return GetImageRoots()->Get(image_root);
119}
120
121mirror::ObjectArray<mirror::Object>* ImageHeader::GetImageRoots() const {
Hiroshi Yamauchi2cd334a2015-01-09 14:03:35 -0800122 // Need a read barrier as it's not visited during root scan.
123 // Pass in the address of the local variable to the read barrier
124 // rather than image_roots_ because it won't move (asserted below)
125 // and it's a const member.
126 mirror::ObjectArray<mirror::Object>* image_roots =
127 reinterpret_cast<mirror::ObjectArray<mirror::Object>*>(image_roots_);
128 mirror::ObjectArray<mirror::Object>* result =
Hiroshi Yamauchicc78f3f2015-12-11 15:51:04 -0800129 ReadBarrier::BarrierForRoot<mirror::ObjectArray<mirror::Object>, kWithReadBarrier>(
Hiroshi Yamauchi2cd334a2015-01-09 14:03:35 -0800130 &image_roots);
131 DCHECK_EQ(image_roots, result);
132 return result;
Ian Rogers2dd0e2c2013-01-24 12:42:14 -0800133}
134
Mathieu Chartiere401d142015-04-22 13:56:20 -0700135ArtMethod* ImageHeader::GetImageMethod(ImageMethod index) const {
136 CHECK_LT(static_cast<size_t>(index), kImageMethodsCount);
137 return reinterpret_cast<ArtMethod*>(image_methods_[index]);
138}
139
140void ImageHeader::SetImageMethod(ImageMethod index, ArtMethod* method) {
141 CHECK_LT(static_cast<size_t>(index), kImageMethodsCount);
142 image_methods_[index] = reinterpret_cast<uint64_t>(method);
143}
144
145const ImageSection& ImageHeader::GetImageSection(ImageSections index) const {
146 CHECK_LT(static_cast<size_t>(index), kSectionCount);
147 return sections_[index];
148}
149
150std::ostream& operator<<(std::ostream& os, const ImageSection& section) {
151 return os << "size=" << section.Size() << " range=" << section.Offset() << "-" << section.End();
152}
153
Mathieu Chartier54d220e2015-07-30 16:20:06 -0700154void ImageSection::VisitPackedArtFields(ArtFieldVisitor* visitor, uint8_t* base) const {
155 for (size_t pos = 0; pos < Size(); ) {
156 auto* array = reinterpret_cast<LengthPrefixedArray<ArtField>*>(base + Offset() + pos);
Vladimir Marko35831e82015-09-11 11:59:18 +0100157 for (size_t i = 0; i < array->size(); ++i) {
Mathieu Chartier54d220e2015-07-30 16:20:06 -0700158 visitor->Visit(&array->At(i, sizeof(ArtField)));
159 }
Vladimir Marko35831e82015-09-11 11:59:18 +0100160 pos += array->ComputeSize(array->size());
Mathieu Chartier54d220e2015-07-30 16:20:06 -0700161 }
162}
163
164void ImageSection::VisitPackedArtMethods(ArtMethodVisitor* visitor,
165 uint8_t* base,
Vladimir Markocf36d492015-08-12 19:27:26 +0100166 size_t pointer_size) const {
Vladimir Marko14632852015-08-17 12:07:23 +0100167 const size_t method_alignment = ArtMethod::Alignment(pointer_size);
168 const size_t method_size = ArtMethod::Size(pointer_size);
Mathieu Chartier54d220e2015-07-30 16:20:06 -0700169 for (size_t pos = 0; pos < Size(); ) {
170 auto* array = reinterpret_cast<LengthPrefixedArray<ArtMethod>*>(base + Offset() + pos);
Vladimir Marko35831e82015-09-11 11:59:18 +0100171 for (size_t i = 0; i < array->size(); ++i) {
Vladimir Markocf36d492015-08-12 19:27:26 +0100172 visitor->Visit(&array->At(i, method_size, method_alignment));
Mathieu Chartier54d220e2015-07-30 16:20:06 -0700173 }
Vladimir Marko35831e82015-09-11 11:59:18 +0100174 pos += array->ComputeSize(array->size(), method_size, method_alignment);
Mathieu Chartier54d220e2015-07-30 16:20:06 -0700175 }
176}
177
Brian Carlstrom4a289ed2011-08-16 17:17:49 -0700178} // namespace art