| OLD | NEW |
| 1 // Copyright (c) 2012, the Dart project authors. Please see the AUTHORS file | 1 // Copyright (c) 2012, the Dart project authors. Please see the AUTHORS file |
| 2 // for details. All rights reserved. Use of this source code is governed by a | 2 // for details. All rights reserved. Use of this source code is governed by a |
| 3 // BSD-style license that can be found in the LICENSE file. | 3 // BSD-style license that can be found in the LICENSE file. |
| 4 | 4 |
| 5 #include "vm/code_patcher.h" | 5 #include "vm/code_patcher.h" |
| 6 #include "vm/cpu.h" | 6 #include "vm/cpu.h" |
| 7 #include "vm/instructions.h" | 7 #include "vm/instructions.h" |
| 8 #include "vm/object.h" | 8 #include "vm/object.h" |
| 9 #include "vm/virtual_memory.h" | 9 #include "vm/virtual_memory.h" |
| 10 | 10 |
| (...skipping 17 matching lines...) Expand all Loading... |
| 28 WritableInstructionsScope::~WritableInstructionsScope() { | 28 WritableInstructionsScope::~WritableInstructionsScope() { |
| 29 if (FLAG_write_protect_code) { | 29 if (FLAG_write_protect_code) { |
| 30 bool status = VirtualMemory::Protect(reinterpret_cast<void*>(address_), | 30 bool status = VirtualMemory::Protect(reinterpret_cast<void*>(address_), |
| 31 size_, | 31 size_, |
| 32 VirtualMemory::kReadExecute); | 32 VirtualMemory::kReadExecute); |
| 33 ASSERT(status); | 33 ASSERT(status); |
| 34 } | 34 } |
| 35 } | 35 } |
| 36 | 36 |
| 37 | 37 |
| 38 static void SwapCode(intptr_t num_bytes, char* code, char* buffer) { | |
| 39 uword code_address = reinterpret_cast<uword>(code); | |
| 40 for (intptr_t i = 0; i < num_bytes; i++) { | |
| 41 char tmp = *code; | |
| 42 *code = *buffer; | |
| 43 *buffer = tmp; | |
| 44 code++; | |
| 45 buffer++; | |
| 46 } | |
| 47 CPU::FlushICache(code_address, num_bytes); | |
| 48 // The buffer is not executed. No need to flush. | |
| 49 } | |
| 50 | |
| 51 | |
| 52 // The patch code buffer contains the jmp code which will be inserted at | 38 // The patch code buffer contains the jmp code which will be inserted at |
| 53 // entry point. | 39 // entry point. |
| 54 void CodePatcher::PatchEntry(const Code& code) { | 40 void CodePatcher::PatchEntry(const Code& code, const Code& new_code) { |
| 55 ASSERT(!IsEntryPatched(code)); | 41 ASSERT(code.instructions() == code.active_instructions()); |
| 56 const uword patch_addr = code.GetEntryPatchPc(); | 42 code.set_active_instructions(new_code.instructions()); |
| 57 ASSERT(patch_addr != 0); | |
| 58 JumpPattern jmp_entry(patch_addr, code); | |
| 59 ASSERT(!jmp_entry.IsValid()); | |
| 60 const uword patch_buffer = code.GetPatchCodePc(); | |
| 61 ASSERT(patch_buffer != 0); | |
| 62 JumpPattern jmp_patch(patch_buffer, code); | |
| 63 ASSERT(jmp_patch.IsValid()); | |
| 64 const uword jump_target = jmp_patch.TargetAddress(); | |
| 65 intptr_t length = jmp_patch.pattern_length_in_bytes(); | |
| 66 { | |
| 67 WritableInstructionsScope writable_code(patch_addr, length); | |
| 68 WritableInstructionsScope writable_buffer(patch_buffer, length); | |
| 69 SwapCode(jmp_patch.pattern_length_in_bytes(), | |
| 70 reinterpret_cast<char*>(patch_addr), | |
| 71 reinterpret_cast<char*>(patch_buffer)); | |
| 72 jmp_entry.SetTargetAddress(jump_target); | |
| 73 } | |
| 74 } | 43 } |
| 75 | 44 |
| 76 | 45 |
| 77 // The entry point is a jmp instruction, the patch code buffer contains | 46 // The entry point is a jmp instruction, the patch code buffer contains |
| 78 // original code, the entry point contains the jump instruction. | 47 // original code, the entry point contains the jump instruction. |
| 79 void CodePatcher::RestoreEntry(const Code& code) { | 48 void CodePatcher::RestoreEntry(const Code& code) { |
| 80 if (!IsEntryPatched(code)) return; | 49 if (!IsEntryPatched(code)) return; |
| 81 const uword patch_addr = code.GetEntryPatchPc(); | 50 ASSERT(code.instructions() != code.active_instructions()); |
| 82 ASSERT(patch_addr != 0); | 51 code.set_active_instructions(code.instructions()); |
| 83 JumpPattern jmp_entry(patch_addr, code); | |
| 84 ASSERT(jmp_entry.IsValid()); | |
| 85 const uword jump_target = jmp_entry.TargetAddress(); | |
| 86 const uword patch_buffer = code.GetPatchCodePc(); | |
| 87 ASSERT(patch_buffer != 0); | |
| 88 // 'patch_buffer' contains original entry code. | |
| 89 JumpPattern jmp_patch(patch_buffer, code); | |
| 90 ASSERT(!jmp_patch.IsValid()); | |
| 91 intptr_t length = jmp_patch.pattern_length_in_bytes(); | |
| 92 { | |
| 93 WritableInstructionsScope writable_code(patch_addr, length); | |
| 94 WritableInstructionsScope writable_buffer(patch_buffer, length); | |
| 95 SwapCode(jmp_patch.pattern_length_in_bytes(), | |
| 96 reinterpret_cast<char*>(patch_addr), | |
| 97 reinterpret_cast<char*>(patch_buffer)); | |
| 98 ASSERT(jmp_patch.IsValid()); | |
| 99 jmp_patch.SetTargetAddress(jump_target); | |
| 100 } | |
| 101 } | 52 } |
| 102 | 53 |
| 103 | 54 |
| 104 bool CodePatcher::IsEntryPatched(const Code& code) { | 55 bool CodePatcher::IsEntryPatched(const Code& code) { |
| 105 const uword patch_addr = code.GetEntryPatchPc(); | 56 return code.instructions() != code.active_instructions(); |
| 106 if (patch_addr == 0) { | |
| 107 return false; | |
| 108 } | |
| 109 JumpPattern jmp_entry(patch_addr, code); | |
| 110 return jmp_entry.IsValid(); | |
| 111 } | |
| 112 | |
| 113 | |
| 114 bool CodePatcher::CodeIsPatchable(const Code& code) { | |
| 115 const uword patch_addr = code.GetEntryPatchPc(); | |
| 116 // Zero means means that the function is not patchable. | |
| 117 if (patch_addr == 0) { | |
| 118 return false; | |
| 119 } | |
| 120 JumpPattern jmp_entry(patch_addr, code); | |
| 121 if (code.Size() < (jmp_entry.pattern_length_in_bytes() * 2)) { | |
| 122 return false; | |
| 123 } | |
| 124 const uword limit = patch_addr + jmp_entry.pattern_length_in_bytes(); | |
| 125 // Check no object stored between patch_addr .. limit. | |
| 126 for (intptr_t i = 0; i < code.pointer_offsets_length(); i++) { | |
| 127 const uword obj_start = code.GetPointerOffsetAt(i) + code.EntryPoint(); | |
| 128 const uword obj_end = obj_start + kWordSize; | |
| 129 if ((obj_start < limit) && (obj_end > patch_addr)) { | |
| 130 return false; | |
| 131 } | |
| 132 } | |
| 133 return true; | |
| 134 } | 57 } |
| 135 | 58 |
| 136 } // namespace dart | 59 } // namespace dart |
| OLD | NEW |