Brian Carlstrom | db4d540 | 2011-08-09 12:18:28 -0700 | [diff] [blame] | 1 | // Copyright 2011 Google Inc. All Rights Reserved. |
| 2 | |
Brian Carlstrom | 4a289ed | 2011-08-16 17:17:49 -0700 | [diff] [blame] | 3 | #ifndef ART_SRC_IMAGE_WRITER_H_ |
| 4 | #define ART_SRC_IMAGE_WRITER_H_ |
Brian Carlstrom | db4d540 | 2011-08-09 12:18:28 -0700 | [diff] [blame] | 5 | |
Brian Carlstrom | db4d540 | 2011-08-09 12:18:28 -0700 | [diff] [blame] | 6 | #include <stdint.h> |
| 7 | |
Elliott Hughes | 90a3369 | 2011-08-30 13:27:07 -0700 | [diff] [blame] | 8 | #include <cstddef> |
| 9 | |
| 10 | #include "UniquePtr.h" |
Brian Carlstrom | 69b15fb | 2011-09-03 12:25:21 -0700 | [diff] [blame] | 11 | #include "dex_cache.h" |
Brian Carlstrom | db4d540 | 2011-08-09 12:18:28 -0700 | [diff] [blame] | 12 | #include "mem_map.h" |
| 13 | #include "object.h" |
| 14 | #include "os.h" |
Brian Carlstrom | 1f87008 | 2011-08-23 16:02:11 -0700 | [diff] [blame] | 15 | #include "space.h" |
Brian Carlstrom | db4d540 | 2011-08-09 12:18:28 -0700 | [diff] [blame] | 16 | |
| 17 | namespace art { |
| 18 | |
Brian Carlstrom | 4e777d4 | 2011-08-15 13:53:52 -0700 | [diff] [blame] | 19 | // Write a Space built during compilation for use during execution. |
Brian Carlstrom | db4d540 | 2011-08-09 12:18:28 -0700 | [diff] [blame] | 20 | class ImageWriter { |
| 21 | |
| 22 | public: |
Brian Carlstrom | 69b15fb | 2011-09-03 12:25:21 -0700 | [diff] [blame] | 23 | ImageWriter() : source_space_(NULL), image_top_(0), image_base_(NULL) {}; |
| 24 | bool Write(const char* filename, uintptr_t image_base); |
Brian Carlstrom | db4d540 | 2011-08-09 12:18:28 -0700 | [diff] [blame] | 25 | ~ImageWriter() {}; |
| 26 | |
| 27 | private: |
| 28 | |
Brian Carlstrom | 69b15fb | 2011-09-03 12:25:21 -0700 | [diff] [blame] | 29 | bool Init(); |
Brian Carlstrom | db4d540 | 2011-08-09 12:18:28 -0700 | [diff] [blame] | 30 | |
| 31 | // we use the lock word to store the offset of the object in the image |
Brian Carlstrom | c74255f | 2011-09-11 22:47:39 -0700 | [diff] [blame] | 32 | void AssignImageOffset(Object* object) { |
Brian Carlstrom | db4d540 | 2011-08-09 12:18:28 -0700 | [diff] [blame] | 33 | DCHECK(object != NULL); |
Elliott Hughes | 5f79133 | 2011-09-15 17:45:30 -0700 | [diff] [blame] | 34 | DCHECK(object->monitor_ == 0); // should be no lock |
Brian Carlstrom | c74255f | 2011-09-11 22:47:39 -0700 | [diff] [blame] | 35 | SetImageOffset(object, image_top_); |
| 36 | image_top_ += RoundUp(object->SizeOf(), 8); // 64-bit alignment |
| 37 | DCHECK_LT(image_top_, image_->GetLength()); |
| 38 | } |
| 39 | static void SetImageOffset(Object* object, size_t offset) { |
| 40 | DCHECK(object != NULL); |
| 41 | // should be no lock (but it might be forward referenced interned string) |
Elliott Hughes | 5f79133 | 2011-09-15 17:45:30 -0700 | [diff] [blame] | 42 | DCHECK(object->monitor_ == 0 || object->IsString()); |
Brian Carlstrom | db4d540 | 2011-08-09 12:18:28 -0700 | [diff] [blame] | 43 | DCHECK_NE(0U, offset); |
Elliott Hughes | 5f79133 | 2011-09-15 17:45:30 -0700 | [diff] [blame] | 44 | object->monitor_ = offset; |
Brian Carlstrom | db4d540 | 2011-08-09 12:18:28 -0700 | [diff] [blame] | 45 | } |
Brian Carlstrom | c74255f | 2011-09-11 22:47:39 -0700 | [diff] [blame] | 46 | static size_t IsImageOffsetAssigned(const Object* object) { |
| 47 | DCHECK(object != NULL); |
Elliott Hughes | 5f79133 | 2011-09-15 17:45:30 -0700 | [diff] [blame] | 48 | size_t offset = object->monitor_; |
Brian Carlstrom | c74255f | 2011-09-11 22:47:39 -0700 | [diff] [blame] | 49 | return offset != 0U; |
| 50 | } |
Brian Carlstrom | 69b15fb | 2011-09-03 12:25:21 -0700 | [diff] [blame] | 51 | static size_t GetImageOffset(const Object* object) { |
Brian Carlstrom | db4d540 | 2011-08-09 12:18:28 -0700 | [diff] [blame] | 52 | DCHECK(object != NULL); |
Elliott Hughes | 5f79133 | 2011-09-15 17:45:30 -0700 | [diff] [blame] | 53 | size_t offset = object->monitor_; |
Brian Carlstrom | db4d540 | 2011-08-09 12:18:28 -0700 | [diff] [blame] | 54 | DCHECK_NE(0U, offset); |
| 55 | return offset; |
| 56 | } |
Brian Carlstrom | 69b15fb | 2011-09-03 12:25:21 -0700 | [diff] [blame] | 57 | static void ResetImageOffset(Object* object) { |
| 58 | DCHECK(object != NULL); |
Elliott Hughes | 5f79133 | 2011-09-15 17:45:30 -0700 | [diff] [blame] | 59 | DCHECK(object->monitor_ != 0); // should be an offset |
| 60 | object->monitor_ = 0; |
Brian Carlstrom | 69b15fb | 2011-09-03 12:25:21 -0700 | [diff] [blame] | 61 | } |
| 62 | |
| 63 | bool InSourceSpace(const Object* object) { |
| 64 | DCHECK(source_space_ != NULL); |
| 65 | const byte* o = reinterpret_cast<const byte*>(object); |
| 66 | return (o >= source_space_->GetBase() && o < source_space_->GetLimit()); |
| 67 | } |
Brian Carlstrom | db4d540 | 2011-08-09 12:18:28 -0700 | [diff] [blame] | 68 | Object* GetImageAddress(const Object* object) { |
| 69 | if (object == NULL) { |
| 70 | return NULL; |
| 71 | } |
Brian Carlstrom | 69b15fb | 2011-09-03 12:25:21 -0700 | [diff] [blame] | 72 | // if object outside the relocating source_space_, assume unchanged |
| 73 | if (!InSourceSpace(object)) { |
| 74 | return const_cast<Object*>(object); |
| 75 | } |
Brian Carlstrom | db4d540 | 2011-08-09 12:18:28 -0700 | [diff] [blame] | 76 | return reinterpret_cast<Object*>(image_base_ + GetImageOffset(object)); |
| 77 | } |
Brian Carlstrom | 69b15fb | 2011-09-03 12:25:21 -0700 | [diff] [blame] | 78 | Object* GetLocalAddress(const Object* object) { |
| 79 | size_t offset = GetImageOffset(object); |
| 80 | byte* dst = image_->GetAddress() + offset; |
| 81 | return reinterpret_cast<Object*>(dst); |
| 82 | } |
Brian Carlstrom | db4d540 | 2011-08-09 12:18:28 -0700 | [diff] [blame] | 83 | |
| 84 | void CalculateNewObjectOffsets(); |
Brian Carlstrom | 78128a6 | 2011-09-15 17:21:19 -0700 | [diff] [blame] | 85 | static void CalculateNewObjectOffsetsCallback(Object* obj, void* arg); |
Brian Carlstrom | db4d540 | 2011-08-09 12:18:28 -0700 | [diff] [blame] | 86 | |
| 87 | void CopyAndFixupObjects(); |
Brian Carlstrom | 78128a6 | 2011-09-15 17:21:19 -0700 | [diff] [blame] | 88 | static void CopyAndFixupObjectsCallback(Object* obj, void* arg); |
Brian Carlstrom | 4873d46 | 2011-08-21 15:23:39 -0700 | [diff] [blame] | 89 | void FixupClass(const Class* orig, Class* copy); |
| 90 | void FixupMethod(const Method* orig, Method* copy); |
Brian Carlstrom | 4873d46 | 2011-08-21 15:23:39 -0700 | [diff] [blame] | 91 | void FixupObject(const Object* orig, Object* copy); |
| 92 | void FixupObjectArray(const ObjectArray<Object>* orig, ObjectArray<Object>* copy); |
| 93 | void FixupInstanceFields(const Object* orig, Object* copy); |
| 94 | void FixupStaticFields(const Class* orig, Class* copy); |
| 95 | void FixupFields(const Object* orig, Object* copy, uint32_t ref_offsets, bool is_static); |
Brian Carlstrom | db4d540 | 2011-08-09 12:18:28 -0700 | [diff] [blame] | 96 | |
Brian Carlstrom | 69b15fb | 2011-09-03 12:25:21 -0700 | [diff] [blame] | 97 | void FixupDexCaches(); |
| 98 | void FixupDexCache(const DexCache* orig, DexCache* copy); |
| 99 | |
| 100 | // Space we are writing objects from |
| 101 | const Space* source_space_; |
| 102 | |
Brian Carlstrom | db4d540 | 2011-08-09 12:18:28 -0700 | [diff] [blame] | 103 | // memory mapped for generating the image |
Elliott Hughes | 90a3369 | 2011-08-30 13:27:07 -0700 | [diff] [blame] | 104 | UniquePtr<MemMap> image_; |
Brian Carlstrom | db4d540 | 2011-08-09 12:18:28 -0700 | [diff] [blame] | 105 | |
Brian Carlstrom | 4e777d4 | 2011-08-15 13:53:52 -0700 | [diff] [blame] | 106 | // Offset to the free space in image_ |
| 107 | size_t image_top_; |
Brian Carlstrom | db4d540 | 2011-08-09 12:18:28 -0700 | [diff] [blame] | 108 | |
| 109 | // Target base address for the output image |
| 110 | byte* image_base_; |
Brian Carlstrom | 69b15fb | 2011-09-03 12:25:21 -0700 | [diff] [blame] | 111 | |
| 112 | // DexCaches seen while scanning for fixing up CodeAndDirectMethods |
| 113 | typedef std::tr1::unordered_set<DexCache*, DexCacheHash> Set; |
| 114 | Set dex_caches_; |
Brian Carlstrom | db4d540 | 2011-08-09 12:18:28 -0700 | [diff] [blame] | 115 | }; |
| 116 | |
| 117 | } // namespace art |
| 118 | |
Brian Carlstrom | 4a289ed | 2011-08-16 17:17:49 -0700 | [diff] [blame] | 119 | #endif // ART_SRC_IMAGE_WRITER_H_ |