Sampling profiler

runtime/platform/thread_win.cc

 // Copyright (c) 2012, the Dart project authors.  Please see the AUTHORS file
 // for details. All rights reserved. Use of this source code is governed by a
 // BSD-style license that can be found in the LICENSE file.
 
 #include "platform/globals.h"
 #if defined(TARGET_OS_WINDOWS)
 
 #include "platform/thread.h"
 
 #include <process.h>  // NOLINT
@@ skipping 21 matching lines @@
 // Dispatch to the thread start function provided by the caller. This trampoline
 // is used to ensure that the thread is properly destroyed if the thread just
 // exits.
 static unsigned int __stdcall ThreadEntry(void* data_ptr) {
   ThreadStartData* data =  reinterpret_cast<ThreadStartData*>(data_ptr);
 
   Thread::ThreadStartFunction function = data->function();
   uword parameter = data->parameter();
   delete data;
 
+  ASSERT(ThreadInlineImpl::thread_id_key != Thread::kUnsetThreadLocalKey);
+
+  ThreadId thread_id = ThreadInlineImpl::CreateThreadId();
+  // Set thread ID in TLS.
+  Thread::SetThreadLocal(ThreadInlineImpl::thread_id_key,
+                         reinterpret_cast<DWORD>(thread_id));
+  MonitorData::GetMonitorWaitDataForThread();
+
   // Call the supplied thread start function handing it its parameters.
   function(parameter);
 
   // Clean up the monitor wait data for this thread.
   MonitorWaitData::ThreadExit();
 
+  // Clear thread ID in TLS.
+  Thread::SetThreadLocal(ThreadInlineImpl::thread_id_key, NULL);
+  ThreadInlineImpl::DestroyThreadId(thread_id);
+
   return 0;
 }
 
 
 int Thread::Start(ThreadStartFunction function, uword parameter) {
   ThreadStartData* start_data = new ThreadStartData(function, parameter);
   uint32_t tid;
   uintptr_t thread = _beginthreadex(NULL, Thread::GetMaxStackSize(),
                                     ThreadEntry, start_data, 0, &tid);
   if (thread == -1L || thread == 0) {
 #ifdef DEBUG
     fprintf(stderr, "_beginthreadex error: %d (%s)\n", errno, strerror(errno));
 #endif
     return errno;
   }
 
   return 0;
 }
 
 
+ThreadId ThreadInlineImpl::CreateThreadId() {
+  // Create an ID for this thread that can be shared with other threads.
+  HANDLE thread_id = OpenThread(THREAD_GET_CONTEXT |
+                                THREAD_SUSPEND_RESUME |
+                                THREAD_QUERY_INFORMATION,
+                                false,
+                                GetCurrentThreadId());
+  ASSERT(thread_id != NULL);
+  return thread_id;
+}
+
+
+void ThreadInlineImpl::DestroyThreadId(ThreadId thread_id) {
+  ASSERT(thread_id != NULL);
+  // Destroy thread ID.
+  CloseHandle(thread_id);
+}
+
+
+ThreadLocalKey ThreadInlineImpl::thread_id_key = Thread::kUnsetThreadLocalKey;
+
 ThreadLocalKey Thread::kUnsetThreadLocalKey = TLS_OUT_OF_INDEXES;
 
 
 ThreadLocalKey Thread::CreateThreadLocal() {
   ThreadLocalKey key = TlsAlloc();
   if (key == kUnsetThreadLocalKey) {
     FATAL1("TlsAlloc failed %d", GetLastError());
   }
   return key;
 }
 
 
 void Thread::DeleteThreadLocal(ThreadLocalKey key) {
   ASSERT(key != kUnsetThreadLocalKey);
   BOOL result = TlsFree(key);
   if (!result) {
     FATAL1("TlsFree failed %d", GetLastError());
   }
 }
 
 
 intptr_t Thread::GetMaxStackSize() {
   const int kStackSize = (128 * kWordSize * KB);
   return kStackSize;
 }
 
 
+ThreadId Thread::GetCurrentThreadId() {
+  ThreadId id = reinterpret_cast<ThreadId>(
+      Thread::GetThreadLocal(ThreadInlineImpl::thread_id_key));
+  ASSERT(id != NULL);
+  return id;
+}
+
+
+void Thread::GetThreadCpuUsage(ThreadId thread_id, int64_t* cpu_usage) {
+  static const int64_t kTimeEpoc = 116444736000000000LL;
+  static const int64_t kTimeScaler = 10;  // 100 ns to us.
+  // Although win32 uses 64-bit integers for representing timestamps,
+  // these are packed into a FILETIME structure. The FILETIME
+  // structure is just a struct representing a 64-bit integer. The
+  // TimeStamp union allows access to both a FILETIME and an integer
+  // representation of the timestamp. The Windows timestamp is in
+  // 100-nanosecond intervals since January 1, 1601.
+  union TimeStamp {
+    FILETIME ft_;
+    int64_t t_;
+  };
+  ASSERT(cpu_usage != NULL);
+  TimeStamp created;
+  TimeStamp exited;
+  TimeStamp kernel;
+  TimeStamp user;
+  BOOL result = GetThreadTimes(thread_id,
+                               &created.ft_,
+                               &exited.ft_,
+                               &kernel.ft_,
+                               &user.ft_);
+  if (!result) {
+    FATAL1("GetThreadCpuUsage failed %d\n", GetLastError());
+  }
+  *cpu_usage = (user.t_ - kTimeEpoc) / kTimeScaler;
+}
+
+
 void Thread::SetThreadLocal(ThreadLocalKey key, uword value) {
   ASSERT(key != kUnsetThreadLocalKey);
   BOOL result = TlsSetValue(key, reinterpret_cast<void*>(value));
   if (!result) {
     FATAL1("TlsSetValue failed %d", GetLastError());
   }
 }
 
 
 Mutex::Mutex() {
@@ skipping 129 matching lines @@
 void MonitorData::SignalAndRemoveFirstWaiter() {
   EnterCriticalSection(&waiters_cs_);
   MonitorWaitData* first = waiters_head_;
   if (first != NULL) {
     // Remove from list.
     if (waiters_head_ == waiters_tail_) {
       waiters_tail_ = waiters_head_ = NULL;
     } else {
       waiters_head_ = waiters_head_->next_;
     }
+    // Clear next.
+    first->next_ = NULL;
     // Signal event.
     BOOL result = SetEvent(first->event_);
     if (result == 0) {
       FATAL1("Monitor::Notify failed to signal event %d", GetLastError());
     }
   }
   LeaveCriticalSection(&waiters_cs_);
 }
 
 
 void MonitorData::SignalAndRemoveAllWaiters() {
   EnterCriticalSection(&waiters_cs_);
   // Extract list to signal.
   MonitorWaitData* current = waiters_head_;
   // Clear list.
   waiters_head_ = waiters_tail_ = NULL;
   // Iterate and signal all events.
   while (current != NULL) {
+    // Copy next.
+    MonitorWaitData* next = current->next_;
+    // Clear next.
+    current->next_ = NULL;
+    // Signal event.
     BOOL result = SetEvent(current->event_);
     if (result == 0) {
       FATAL1("Failed to set event for NotifyAll %d", GetLastError());
     }
-    current = current->next_;
+    current = next;
   }
   LeaveCriticalSection(&waiters_cs_);
 }
 
 
 MonitorWaitData* MonitorData::GetMonitorWaitDataForThread() {
   // Ensure that the thread local key for monitor wait data objects is
   // initialized.
-  EnterCriticalSection(&waiters_cs_);
-  if (MonitorWaitData::monitor_wait_data_key_ == Thread::kUnsetThreadLocalKey) {
-    MonitorWaitData::monitor_wait_data_key_ = Thread::CreateThreadLocal();
-  }
-  LeaveCriticalSection(&waiters_cs_);
+  ASSERT(MonitorWaitData::monitor_wait_data_key_ !=
+         Thread::kUnsetThreadLocalKey);
 
   // Get the MonitorWaitData object containing the event for this
   // thread from thread local storage. Create it if it does not exist.
   uword raw_wait_data =
     Thread::GetThreadLocal(MonitorWaitData::monitor_wait_data_key_);
   MonitorWaitData* wait_data = NULL;
   if (raw_wait_data == 0) {
     HANDLE event = CreateEvent(NULL, FALSE, FALSE, NULL);
     wait_data = new MonitorWaitData(event);
     Thread::SetThreadLocal(MonitorWaitData::monitor_wait_data_key_,
                            reinterpret_cast<uword>(wait_data));
   } else {
     wait_data = reinterpret_cast<MonitorWaitData*>(raw_wait_data);
     wait_data->next_ = NULL;
   }
   return wait_data;
 }
 
 
 Monitor::WaitResult Monitor::Wait(int64_t millis) {
   Monitor::WaitResult retval = kNotified;
 
   // Get the wait data object containing the event to wait for.
-  MonitorWaitData* wait_data = data_.GetMonitorWaitDataForThread();
+  MonitorWaitData* wait_data = MonitorData::GetMonitorWaitDataForThread();
 
   // Start waiting by adding the MonitorWaitData to the list of
   // waiters.
   data_.AddWaiter(wait_data);
 
   // Leave the monitor critical section while waiting.
   LeaveCriticalSection(&data_.cs_);
 
   // Perform the actual wait on the event.
   DWORD result = WAIT_FAILED;
@@ skipping 17 matching lines @@
     }
   }
 
   // Reacquire the monitor critical section before continuing.
   EnterCriticalSection(&data_.cs_);
 
   return retval;
 }
 
 
+Monitor::WaitResult Monitor::WaitMicros(int64_t micros) {
+  // TODO(johnmccutchan): Investigate sub-millisecond sleep times on Windows.
+  int64_t millis = micros / kMicrosecondsPerMillisecond;
+  if ((millis * kMicrosecondsPerMillisecond) < micros) {
+    // We've been asked to sleep for a fraction of a millisecond,
+    // this isn't supported on Windows. Bumps milliseconds up by one
+    // so that we never return too early. We likely return late though.
+    millis += 1;
+  }
+  return Wait(millis);
+}
+
+
 void Monitor::Notify() {
   data_.SignalAndRemoveFirstWaiter();
 }
 
 
 void Monitor::NotifyAll() {
   // If one of the objects in the list of waiters wakes because of a
   // timeout before we signal it, that object will get an extra
   // signal. This will be treated as a spurious wake-up and is OK
   // since all uses of monitors should recheck the condition after a
   // Wait.
   data_.SignalAndRemoveAllWaiters();
 }
 
 }  // namespace dart
 
 #endif  // defined(TARGET_OS_WINDOWS)