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 | |
| 6 | #include <cstddef> |
| 7 | #include <stdint.h> |
| 8 | |
| 9 | #include "mem_map.h" |
| 10 | #include "object.h" |
| 11 | #include "os.h" |
| 12 | #include "scoped_ptr.h" |
| 13 | |
| 14 | namespace art { |
| 15 | |
Brian Carlstrom | 4e777d4 | 2011-08-15 13:53:52 -0700 | [diff] [blame] | 16 | // Write a Space built during compilation for use during execution. |
Brian Carlstrom | db4d540 | 2011-08-09 12:18:28 -0700 | [diff] [blame] | 17 | class ImageWriter { |
| 18 | |
| 19 | public: |
Brian Carlstrom | 4e777d4 | 2011-08-15 13:53:52 -0700 | [diff] [blame] | 20 | ImageWriter() : image_top_(0), image_base_(NULL) {}; |
Brian Carlstrom | db4d540 | 2011-08-09 12:18:28 -0700 | [diff] [blame] | 21 | bool Write(Space* space, const char* filename, byte* image_base); |
| 22 | ~ImageWriter() {}; |
| 23 | |
| 24 | private: |
| 25 | |
| 26 | bool Init(Space* space); |
| 27 | |
| 28 | // we use the lock word to store the offset of the object in the image |
| 29 | void SetImageOffset(Object* object, size_t offset) { |
| 30 | DCHECK(object != NULL); |
| 31 | DCHECK(object->monitor_ == NULL); // should be no lock |
| 32 | DCHECK_NE(0U, offset); |
| 33 | object->monitor_ = reinterpret_cast<Monitor*>(offset); |
| 34 | } |
| 35 | size_t GetImageOffset(const Object* object) { |
| 36 | DCHECK(object != NULL); |
| 37 | size_t offset = reinterpret_cast<size_t>(object->monitor_); |
| 38 | DCHECK_NE(0U, offset); |
| 39 | return offset; |
| 40 | } |
| 41 | Object* GetImageAddress(const Object* object) { |
| 42 | if (object == NULL) { |
| 43 | return NULL; |
| 44 | } |
| 45 | return reinterpret_cast<Object*>(image_base_ + GetImageOffset(object)); |
| 46 | } |
| 47 | |
| 48 | void CalculateNewObjectOffsets(); |
| 49 | static void CalculateNewObjectOffsetsCallback(Object *obj, void *arg); |
| 50 | |
| 51 | void CopyAndFixupObjects(); |
| 52 | static void CopyAndFixupObjectsCallback(Object *obj, void *arg); |
| 53 | void FixupObject(Object* orig, Object* copy); |
| 54 | void FixupObjectArray(ObjectArray<Object>* orig, ObjectArray<Object>* copy); |
| 55 | void FixupInstanceFields(Object* orig, Object* copy); |
| 56 | |
| 57 | // memory mapped for generating the image |
Brian Carlstrom | 4e777d4 | 2011-08-15 13:53:52 -0700 | [diff] [blame] | 58 | scoped_ptr<MemMap> image_; |
Brian Carlstrom | db4d540 | 2011-08-09 12:18:28 -0700 | [diff] [blame] | 59 | |
Brian Carlstrom | 4e777d4 | 2011-08-15 13:53:52 -0700 | [diff] [blame] | 60 | // Offset to the free space in image_ |
| 61 | size_t image_top_; |
Brian Carlstrom | db4d540 | 2011-08-09 12:18:28 -0700 | [diff] [blame] | 62 | |
| 63 | // Target base address for the output image |
| 64 | byte* image_base_; |
| 65 | }; |
| 66 | |
| 67 | } // namespace art |
| 68 | |
Brian Carlstrom | 4a289ed | 2011-08-16 17:17:49 -0700 | [diff] [blame] | 69 | #endif // ART_SRC_IMAGE_WRITER_H_ |