| 1 | // Copyright (c) 2016, the Dart project authors. Please see the AUTHORS file |
|---|---|
| 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. |
| 4 | |
| 5 | #include "vm/lockers.h" |
| 6 | #include "platform/assert.h" |
| 7 | #include "vm/heap/safepoint.h" |
| 8 | #include "vm/isolate.h" |
| 9 | |
| 10 | namespace dart { |
| 11 | |
| 12 | Monitor::WaitResult MonitorLocker::WaitWithSafepointCheck(Thread* thread, |
| 13 | int64_t millis) { |
| 14 | ASSERT(thread == Thread::Current()); |
| 15 | ASSERT(thread->execution_state() == Thread::kThreadInVM); |
| 16 | #if defined(DEBUG) |
| 17 | if (no_safepoint_scope_) { |
| 18 | thread->DecrementNoSafepointScopeDepth(); |
| 19 | } |
| 20 | #endif |
| 21 | thread->set_execution_state(Thread::kThreadInBlockedState); |
| 22 | thread->EnterSafepoint(); |
| 23 | Monitor::WaitResult result = monitor_->Wait(millis); |
| 24 | // First try a fast update of the thread state to indicate it is not at a |
| 25 | // safepoint anymore. |
| 26 | if (!thread->TryExitSafepoint()) { |
| 27 | // Fast update failed which means we could potentially be in the middle |
| 28 | // of a safepoint operation and need to block for it. |
| 29 | monitor_->Exit(); |
| 30 | SafepointHandler* handler = thread->isolate_group()->safepoint_handler(); |
| 31 | handler->ExitSafepointUsingLock(thread); |
| 32 | monitor_->Enter(); |
| 33 | } |
| 34 | thread->set_execution_state(Thread::kThreadInVM); |
| 35 | #if defined(DEBUG) |
| 36 | if (no_safepoint_scope_) { |
| 37 | thread->IncrementNoSafepointScopeDepth(); |
| 38 | } |
| 39 | #endif |
| 40 | return result; |
| 41 | } |
| 42 | |
| 43 | SafepointMutexLocker::SafepointMutexLocker(Mutex* mutex) : mutex_(mutex) { |
| 44 | ASSERT(mutex != NULL); |
| 45 | if (!mutex_->TryLock()) { |
| 46 | // We did not get the lock and could potentially block, so transition |
| 47 | // accordingly. |
| 48 | Thread* thread = Thread::Current(); |
| 49 | if (thread != NULL) { |
| 50 | TransitionVMToBlocked transition(thread); |
| 51 | mutex->Lock(); |
| 52 | } else { |
| 53 | mutex->Lock(); |
| 54 | } |
| 55 | } |
| 56 | } |
| 57 | |
| 58 | SafepointMonitorLocker::SafepointMonitorLocker(Monitor* monitor) |
| 59 | : monitor_(monitor) { |
| 60 | ASSERT(monitor_ != NULL); |
| 61 | if (!monitor_->TryEnter()) { |
| 62 | // We did not get the lock and could potentially block, so transition |
| 63 | // accordingly. |
| 64 | Thread* thread = Thread::Current(); |
| 65 | if (thread != NULL) { |
| 66 | TransitionVMToBlocked transition(thread); |
| 67 | monitor_->Enter(); |
| 68 | } else { |
| 69 | monitor_->Enter(); |
| 70 | } |
| 71 | } |
| 72 | } |
| 73 | |
| 74 | Monitor::WaitResult SafepointMonitorLocker::Wait(int64_t millis) { |
| 75 | Thread* thread = Thread::Current(); |
| 76 | if (thread != NULL) { |
| 77 | Monitor::WaitResult result; |
| 78 | { |
| 79 | TransitionVMToBlocked transition(thread); |
| 80 | result = monitor_->Wait(millis); |
| 81 | } |
| 82 | return result; |
| 83 | } else { |
| 84 | return monitor_->Wait(millis); |
| 85 | } |
| 86 | } |
| 87 | |
| 88 | } // namespace dart |
| 89 |