patchoat.h revision d4c4d953035d4418126d36517e402f411d6a87f3
1/*
2 * Copyright (C) 2014 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 */
16
17#ifndef ART_PATCHOAT_PATCHOAT_H_
18#define ART_PATCHOAT_PATCHOAT_H_
19
20#include "base/macros.h"
21#include "base/mutex.h"
22#include "instruction_set.h"
23#include "os.h"
24#include "elf_file.h"
25#include "elf_utils.h"
26#include "gc/accounting/space_bitmap.h"
27#include "gc/heap.h"
28#include "utils.h"
29
30namespace art {
31
32class ImageHeader;
33
34namespace mirror {
35class Object;
36class Reference;
37class Class;
38class ArtMethod;
39}  // namespace mirror
40
41class PatchOat {
42 public:
43  static bool Patch(File* oat_in, off_t delta, File* oat_out, TimingLogger* timings);
44
45  static bool Patch(const std::string& art_location, off_t delta, File* art_out, InstructionSet isa,
46                    TimingLogger* timings);
47
48  static bool Patch(const File* oat_in, const std::string& art_location,
49                    off_t delta, File* oat_out, File* art_out, InstructionSet isa,
50                    TimingLogger* timings);
51
52 private:
53  // Takes ownership only of the ElfFile. All other pointers are only borrowed.
54  PatchOat(ElfFile* oat_file, off_t delta, TimingLogger* timings)
55      : oat_file_(oat_file), image_(nullptr), bitmap_(nullptr), heap_(nullptr), delta_(delta),
56        timings_(timings) {}
57  PatchOat(MemMap* image, gc::accounting::ContinuousSpaceBitmap* bitmap,
58           MemMap* heap, off_t delta, TimingLogger* timings)
59      : image_(image), bitmap_(bitmap), heap_(heap),
60        delta_(delta), timings_(timings) {}
61  PatchOat(ElfFile* oat_file, MemMap* image, gc::accounting::ContinuousSpaceBitmap* bitmap,
62           MemMap* heap, off_t delta, TimingLogger* timings)
63      : oat_file_(oat_file), image_(image), bitmap_(bitmap), heap_(heap),
64        delta_(delta), timings_(timings) {}
65  ~PatchOat() {}
66
67  static void BitmapCallback(mirror::Object* obj, void* arg)
68      SHARED_LOCKS_REQUIRED(Locks::mutator_lock_) {
69    reinterpret_cast<PatchOat*>(arg)->VisitObject(obj);
70  }
71
72  void VisitObject(mirror::Object* obj)
73      SHARED_LOCKS_REQUIRED(Locks::mutator_lock_);
74  void FixupMethod(mirror::ArtMethod* object, mirror::ArtMethod* copy)
75      SHARED_LOCKS_REQUIRED(Locks::mutator_lock_);
76  bool InHeap(mirror::Object*);
77
78  // Patches oat in place, modifying the oat_file given to the constructor.
79  bool PatchElf();
80  template <typename ElfFileImpl>
81  bool PatchElf(ElfFileImpl* oat_file);
82  template <typename ElfFileImpl>
83  bool PatchTextSection(ElfFileImpl* oat_file);
84  // Templatized version to actually do the patching with the right sized offsets.
85  template <typename ElfFileImpl, typename patch_loc_t> bool PatchTextSection(ElfFileImpl* oat_file);
86  template <typename ElfFileImpl, typename patch_loc_t> bool CheckOatFile(ElfFileImpl* oat_filec);
87  template <typename ElfFileImpl>
88  bool PatchOatHeader(ElfFileImpl* oat_file);
89
90  bool PatchImage() SHARED_LOCKS_REQUIRED(Locks::mutator_lock_);
91
92  bool WriteElf(File* out);
93  bool WriteImage(File* out);
94
95  mirror::Object* RelocatedCopyOf(mirror::Object*);
96  mirror::Object* RelocatedAddressOf(mirror::Object* obj);
97
98  // Walks through the old image and patches the mmap'd copy of it to the new offset. It does not
99  // change the heap.
100  class PatchVisitor {
101  public:
102    PatchVisitor(PatchOat* patcher, mirror::Object* copy) : patcher_(patcher), copy_(copy) {}
103    ~PatchVisitor() {}
104    void operator() (mirror::Object* obj, MemberOffset off, bool b) const
105      EXCLUSIVE_LOCKS_REQUIRED(Locks::mutator_lock_, Locks::heap_bitmap_lock_);
106    // For reference classes.
107    void operator() (mirror::Class* cls, mirror::Reference* ref) const
108      EXCLUSIVE_LOCKS_REQUIRED(Locks::mutator_lock_, Locks::heap_bitmap_lock_);
109  private:
110    PatchOat* const patcher_;
111    mirror::Object* const copy_;
112  };
113
114  // The elf file we are patching.
115  std::unique_ptr<ElfFile> oat_file_;
116  // A mmap of the image we are patching. This is modified.
117  const MemMap* const image_;
118  // The bitmap over the image within the heap we are patching. This is not modified.
119  gc::accounting::ContinuousSpaceBitmap* const bitmap_;
120  // The heap we are patching. This is not modified.
121  const MemMap* const heap_;
122  // The amount we are changing the offset by.
123  const off_t delta_;
124  // Timing splits.
125  TimingLogger* const timings_;
126
127  DISALLOW_IMPLICIT_CONSTRUCTORS(PatchOat);
128};
129
130}  // namespace art
131#endif  // ART_PATCHOAT_PATCHOAT_H_
132