1 /* 2 * Copyright (C) 2011 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_RUNTIME_THREAD_INL_H_ 18 #define ART_RUNTIME_THREAD_INL_H_ 19 20 #include "thread.h" 21 22 #include <pthread.h> 23 24 #include "base/casts.h" 25 #include "base/mutex-inl.h" 26 #include "gc/heap.h" 27 #include "jni_internal.h" 28 29 namespace art { 30 31 // Quickly access the current thread from a JNIEnv. 32 static inline Thread* ThreadForEnv(JNIEnv* env) { 33 JNIEnvExt* full_env(down_cast<JNIEnvExt*>(env)); 34 return full_env->self; 35 } 36 37 inline Thread* Thread::Current() { 38 // We rely on Thread::Current returning NULL for a detached thread, so it's not obvious 39 // that we can replace this with a direct %fs access on x86. 40 if (!is_started_) { 41 return NULL; 42 } else { 43 void* thread = pthread_getspecific(Thread::pthread_key_self_); 44 return reinterpret_cast<Thread*>(thread); 45 } 46 } 47 48 inline ThreadState Thread::SetState(ThreadState new_state) { 49 // Cannot use this code to change into Runnable as changing to Runnable should fail if 50 // old_state_and_flags.suspend_request is true. 51 DCHECK_NE(new_state, kRunnable); 52 DCHECK_EQ(this, Thread::Current()); 53 union StateAndFlags old_state_and_flags; 54 old_state_and_flags.as_int = tls32_.state_and_flags.as_int; 55 tls32_.state_and_flags.as_struct.state = new_state; 56 return static_cast<ThreadState>(old_state_and_flags.as_struct.state); 57 } 58 59 inline void Thread::AssertThreadSuspensionIsAllowable(bool check_locks) const { 60 if (kIsDebugBuild) { 61 CHECK_EQ(0u, tls32_.no_thread_suspension) << tlsPtr_.last_no_thread_suspension_cause; 62 if (check_locks) { 63 bool bad_mutexes_held = false; 64 for (int i = kLockLevelCount - 1; i >= 0; --i) { 65 // We expect no locks except the mutator_lock_ or thread list suspend thread lock. 66 if (i != kMutatorLock && i != kThreadListSuspendThreadLock) { 67 BaseMutex* held_mutex = GetHeldMutex(static_cast<LockLevel>(i)); 68 if (held_mutex != NULL) { 69 LOG(ERROR) << "holding \"" << held_mutex->GetName() 70 << "\" at point where thread suspension is expected"; 71 bad_mutexes_held = true; 72 } 73 } 74 } 75 CHECK(!bad_mutexes_held); 76 } 77 } 78 } 79 80 inline void Thread::TransitionFromRunnableToSuspended(ThreadState new_state) { 81 AssertThreadSuspensionIsAllowable(); 82 DCHECK_NE(new_state, kRunnable); 83 DCHECK_EQ(this, Thread::Current()); 84 // Change to non-runnable state, thereby appearing suspended to the system. 85 DCHECK_EQ(GetState(), kRunnable); 86 union StateAndFlags old_state_and_flags; 87 union StateAndFlags new_state_and_flags; 88 while (true) { 89 old_state_and_flags.as_int = tls32_.state_and_flags.as_int; 90 if (UNLIKELY((old_state_and_flags.as_struct.flags & kCheckpointRequest) != 0)) { 91 RunCheckpointFunction(); 92 continue; 93 } 94 // Change the state but keep the current flags (kCheckpointRequest is clear). 95 DCHECK_EQ((old_state_and_flags.as_struct.flags & kCheckpointRequest), 0); 96 new_state_and_flags.as_struct.flags = old_state_and_flags.as_struct.flags; 97 new_state_and_flags.as_struct.state = new_state; 98 99 // CAS the value without a memory ordering as that is given by the lock release below. 100 bool done = 101 tls32_.state_and_flags.as_atomic_int.CompareExchangeWeakRelaxed(old_state_and_flags.as_int, 102 new_state_and_flags.as_int); 103 if (LIKELY(done)) { 104 break; 105 } 106 } 107 // Release share on mutator_lock_. 108 Locks::mutator_lock_->SharedUnlock(this); 109 } 110 111 inline ThreadState Thread::TransitionFromSuspendedToRunnable() { 112 bool done = false; 113 union StateAndFlags old_state_and_flags; 114 old_state_and_flags.as_int = tls32_.state_and_flags.as_int; 115 int16_t old_state = old_state_and_flags.as_struct.state; 116 DCHECK_NE(static_cast<ThreadState>(old_state), kRunnable); 117 do { 118 Locks::mutator_lock_->AssertNotHeld(this); // Otherwise we starve GC.. 119 old_state_and_flags.as_int = tls32_.state_and_flags.as_int; 120 DCHECK_EQ(old_state_and_flags.as_struct.state, old_state); 121 if (UNLIKELY((old_state_and_flags.as_struct.flags & kSuspendRequest) != 0)) { 122 // Wait while our suspend count is non-zero. 123 MutexLock mu(this, *Locks::thread_suspend_count_lock_); 124 old_state_and_flags.as_int = tls32_.state_and_flags.as_int; 125 DCHECK_EQ(old_state_and_flags.as_struct.state, old_state); 126 while ((old_state_and_flags.as_struct.flags & kSuspendRequest) != 0) { 127 // Re-check when Thread::resume_cond_ is notified. 128 Thread::resume_cond_->Wait(this); 129 old_state_and_flags.as_int = tls32_.state_and_flags.as_int; 130 DCHECK_EQ(old_state_and_flags.as_struct.state, old_state); 131 } 132 DCHECK_EQ(GetSuspendCount(), 0); 133 } 134 // Re-acquire shared mutator_lock_ access. 135 Locks::mutator_lock_->SharedLock(this); 136 // Atomically change from suspended to runnable if no suspend request pending. 137 old_state_and_flags.as_int = tls32_.state_and_flags.as_int; 138 DCHECK_EQ(old_state_and_flags.as_struct.state, old_state); 139 if (LIKELY((old_state_and_flags.as_struct.flags & kSuspendRequest) == 0)) { 140 union StateAndFlags new_state_and_flags; 141 new_state_and_flags.as_int = old_state_and_flags.as_int; 142 new_state_and_flags.as_struct.state = kRunnable; 143 // CAS the value without a memory ordering as that is given by the lock acquisition above. 144 done = 145 tls32_.state_and_flags.as_atomic_int.CompareExchangeWeakRelaxed(old_state_and_flags.as_int, 146 new_state_and_flags.as_int); 147 } 148 if (UNLIKELY(!done)) { 149 // Failed to transition to Runnable. Release shared mutator_lock_ access and try again. 150 Locks::mutator_lock_->SharedUnlock(this); 151 } else { 152 return static_cast<ThreadState>(old_state); 153 } 154 } while (true); 155 } 156 157 inline void Thread::VerifyStack() { 158 if (kVerifyStack) { 159 if (Runtime::Current()->GetHeap()->IsObjectValidationEnabled()) { 160 VerifyStackImpl(); 161 } 162 } 163 } 164 165 inline size_t Thread::TlabSize() const { 166 return tlsPtr_.thread_local_end - tlsPtr_.thread_local_pos; 167 } 168 169 inline mirror::Object* Thread::AllocTlab(size_t bytes) { 170 DCHECK_GE(TlabSize(), bytes); 171 ++tlsPtr_.thread_local_objects; 172 mirror::Object* ret = reinterpret_cast<mirror::Object*>(tlsPtr_.thread_local_pos); 173 tlsPtr_.thread_local_pos += bytes; 174 return ret; 175 } 176 177 inline bool Thread::PushOnThreadLocalAllocationStack(mirror::Object* obj) { 178 DCHECK_LE(tlsPtr_.thread_local_alloc_stack_top, tlsPtr_.thread_local_alloc_stack_end); 179 if (tlsPtr_.thread_local_alloc_stack_top < tlsPtr_.thread_local_alloc_stack_end) { 180 // There's room. 181 DCHECK_LE(reinterpret_cast<byte*>(tlsPtr_.thread_local_alloc_stack_top) + 182 sizeof(mirror::Object*), 183 reinterpret_cast<byte*>(tlsPtr_.thread_local_alloc_stack_end)); 184 DCHECK(*tlsPtr_.thread_local_alloc_stack_top == nullptr); 185 *tlsPtr_.thread_local_alloc_stack_top = obj; 186 ++tlsPtr_.thread_local_alloc_stack_top; 187 return true; 188 } 189 return false; 190 } 191 192 inline void Thread::SetThreadLocalAllocationStack(mirror::Object** start, mirror::Object** end) { 193 DCHECK(Thread::Current() == this) << "Should be called by self"; 194 DCHECK(start != nullptr); 195 DCHECK(end != nullptr); 196 DCHECK_ALIGNED(start, sizeof(mirror::Object*)); 197 DCHECK_ALIGNED(end, sizeof(mirror::Object*)); 198 DCHECK_LT(start, end); 199 tlsPtr_.thread_local_alloc_stack_end = end; 200 tlsPtr_.thread_local_alloc_stack_top = start; 201 } 202 203 inline void Thread::RevokeThreadLocalAllocationStack() { 204 if (kIsDebugBuild) { 205 // Note: self is not necessarily equal to this thread since thread may be suspended. 206 Thread* self = Thread::Current(); 207 DCHECK(this == self || IsSuspended() || GetState() == kWaitingPerformingGc) 208 << GetState() << " thread " << this << " self " << self; 209 } 210 tlsPtr_.thread_local_alloc_stack_end = nullptr; 211 tlsPtr_.thread_local_alloc_stack_top = nullptr; 212 } 213 214 } // namespace art 215 216 #endif // ART_RUNTIME_THREAD_INL_H_ 217