Initial GetSample implementation.

src/sampler.cc

 // Copyright 2013 the V8 project authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
-
 #include "src/sampler.h"
 
 #if V8_OS_POSIX && !V8_OS_CYGWIN
 
 #define USE_SIGNALS
 
 #include <errno.h>
 #include <pthread.h>
 #include <signal.h>
 #include <sys/time.h>
@@ skipping 20 matching lines @@
     !defined(__BIONIC_HAVE_STRUCT_SIGCONTEXT)
 #include <asm/sigcontext.h>  // NOLINT
 #endif
 
 #elif V8_OS_WIN || V8_OS_CYGWIN
 
 #include "src/base/win32-headers.h"
 
 #endif
 
+#include "include/v8-sampler.h"
+
 #include "src/v8.h"
 
 #include "src/base/platform/platform.h"
 #include "src/cpu-profiler-inl.h"
 #include "src/flags.h"
 #include "src/frames-inl.h"
 #include "src/log.h"
 #include "src/simulator.h"
 #include "src/v8threads.h"
 #include "src/vm-state-inl.h"
@@ skipping 207 matching lines @@
  private:
   Simulator* simulator_;
 };
 #endif  // USE_SIMULATOR
 
 
 #if defined(USE_SIGNALS)
 
 class SignalHandler : public AllStatic {
  public:
-  static void SetUp() { if (!mutex_) mutex_ = new base::Mutex(); }
-  static void TearDown() { delete mutex_; }
+  static void SetUp() {
+    if (!mutex_) mutex_ = new base::Mutex();
+    if (!new_sample_semaphore_)
+      new_sample_semaphore_ = new base::Semaphore(0);
+  }
+
+  static void TearDown() {
+    delete mutex_;
+    delete new_sample_semaphore_;
+  }
 
   static void IncreaseSamplerCount() {
     base::LockGuard<base::Mutex> lock_guard(mutex_);
     if (++client_count_ == 1) Install();
   }
 
   static void DecreaseSamplerCount() {
     base::LockGuard<base::Mutex> lock_guard(mutex_);
     if (--client_count_ == 0) Restore();
   }
 
   static bool Installed() {
     return signal_handler_installed_;
   }
 
+

[fmeawad, 2014/07/29 18:30:50]

Delete the added line.

[gholap, 2014/07/29 23:25:20]

Done.

  private:
   static void Install() {
     struct sigaction sa;
     sa.sa_sigaction = &HandleProfilerSignal;
     sigemptyset(&sa.sa_mask);
 #if V8_OS_QNX
     sa.sa_flags = SA_SIGINFO;
 #else
     sa.sa_flags = SA_RESTART | SA_SIGINFO;
 #endif
     signal_handler_installed_ =
         (sigaction(SIGPROF, &sa, &old_signal_handler_) == 0);
   }
 
   static void Restore() {
     if (signal_handler_installed_) {
       sigaction(SIGPROF, &old_signal_handler_, 0);
       signal_handler_installed_ = false;
     }
   }
 
   static void HandleProfilerSignal(int signal, siginfo_t* info, void* context);
+
+  // The SIGPROF could have come from DoSample or from GetSample.
+  // GetSample is the internal endpoint for the public
+  // GetSample API, which aims to bypass all the internal buffers
+  // and return just one sample instead, rightaway.

[fmeawad, 2014/07/29 18:30:50]

Add a TODO that we will eventually have GetSample, once the CPU Profiler will be
external as well.

[gholap, 2014/07/29 23:25:20]

Done.

+
+  // This is the sample which will be filled by the call from GetSample.
+  static TickSample sample_;
+
+  // For the SIGPROF handler to know whether the call was from GetSample.
+  static bool called_from_get_sample_;
+
+  // GetSample waits synchronously till the SIGPROF handler returns.
+  // this semaphore is used to signal GetSample.
+  static base::Semaphore* new_sample_semaphore_;

[fmeawad, 2014/07/29 18:30:50]

I would call it, sampling_semaphore instead.

[gholap, 2014/07/29 23:25:20]

Done.

+
+  // It is not that every time HandleProfilerSignal is invoked,
+  // it succeeds in obtaining a sample.
+  // This provides GetSample with the information whether
+  // the handler finished with or without getting the sample.
+  static bool new_sample_available_;

[fmeawad, 2014/07/29 18:30:50]

just sample_available_

[gholap, 2014/07/29 23:25:20]

Done.

+
   // Protects the process wide state below.
   static base::Mutex* mutex_;
   static int client_count_;
   static bool signal_handler_installed_;
   static struct sigaction old_signal_handler_;
+
+  friend class Sampler;
 };
 
 
+TickSample SignalHandler::sample_;
+bool SignalHandler::called_from_get_sample_ = false;
+base::Semaphore* SignalHandler::new_sample_semaphore_ = NULL;
+bool SignalHandler::new_sample_available_ = false;
+
 base::Mutex* SignalHandler::mutex_ = NULL;
 int SignalHandler::client_count_ = 0;
 struct sigaction SignalHandler::old_signal_handler_;
 bool SignalHandler::signal_handler_installed_ = false;
 
 
 void SignalHandler::HandleProfilerSignal(int signal, siginfo_t* info,
                                          void* context) {
 #if V8_OS_NACL
   // As Native Client does not support signal handling, profiling
   // is disabled.
   return;
 #else
+
+#define SIGNAL_AND_RETURN(GOT_NEW)              \

[fmeawad, 2014/07/29 18:30:50]

Explain why this macro is needed.

[gholap, 2014/07/29 23:25:20]

Done.

+  {                                             \
+    new_sample_available_ = GOT_NEW;            \
+    new_sample_semaphore_->Signal();            \
+    return;                                     \
+  }
+
   USE(info);
-  if (signal != SIGPROF) return;
+  if (signal != SIGPROF) SIGNAL_AND_RETURN(false);
   Isolate* isolate = Isolate::UncheckedReentrantCurrent();
   if (isolate == NULL || !isolate->IsInitialized() || !isolate->IsInUse()) {
     // We require a fully initialized and entered isolate.
-    return;
+    SIGNAL_AND_RETURN(false);
   }
   if (v8::Locker::IsActive() &&
       !isolate->thread_manager()->IsLockedByCurrentThread()) {
-    return;
+    SIGNAL_AND_RETURN(false);
   }
 
   Sampler* sampler = isolate->logger()->sampler();
-  if (sampler == NULL) return;
+  if (sampler == NULL) SIGNAL_AND_RETURN(false);
 
   RegisterState state;
 
 #if defined(USE_SIMULATOR)
   SimulatorHelper helper;
-  if (!helper.Init(sampler, isolate)) return;
+  if (!helper.Init(sampler, isolate)) SIGNAL_AND_RETURN(false);
   helper.FillRegisters(&state);
   // It possible that the simulator is interrupted while it is updating
   // the sp or fp register. ARM64 simulator does this in two steps:
   // first setting it to zero and then setting it to the new value.
   // Bailout if sp/fp doesn't contain the new value.
-  if (state.sp == 0 || state.fp == 0) return;
+  if (state.sp == 0 || state.fp == 0) SIGNAL_AND_RETURN(false);
 #else
   // Extracting the sample from the context is extremely machine dependent.
   ucontext_t* ucontext = reinterpret_cast<ucontext_t*>(context);
 #if !V8_OS_OPENBSD
   mcontext_t& mcontext = ucontext->uc_mcontext;
 #endif
 #if V8_OS_LINUX
 #if V8_HOST_ARCH_IA32
   state.pc = reinterpret_cast<Address>(mcontext.gregs[REG_EIP]);
   state.sp = reinterpret_cast<Address>(mcontext.gregs[REG_ESP]);
@@ skipping 91 matching lines @@
   state.pc = reinterpret_cast<Address>(mcontext.cpu.eip);
   state.sp = reinterpret_cast<Address>(mcontext.cpu.esp);
   state.fp = reinterpret_cast<Address>(mcontext.cpu.ebp);
 #elif V8_HOST_ARCH_ARM
   state.pc = reinterpret_cast<Address>(mcontext.cpu.gpr[ARM_REG_PC]);
   state.sp = reinterpret_cast<Address>(mcontext.cpu.gpr[ARM_REG_SP]);
   state.fp = reinterpret_cast<Address>(mcontext.cpu.gpr[ARM_REG_FP]);
 #endif  // V8_HOST_ARCH_*
 #endif  // V8_OS_QNX
 #endif  // USE_SIMULATOR
-  sampler->SampleStack(state);
+  if (called_from_get_sample_) {
+    sample_.Init(sampler->isolate(), state);
+  } else {
+    sampler->SampleStack(state);
+  }
+  SIGNAL_AND_RETURN(true);
+#undef SIGNAL_AND_RETURN
 #endif  // V8_OS_NACL
 }
 
 #endif
 
 
 class SamplerThread : public base::Thread {
  public:
   static const int kSamplerThreadStackSize = 64 * KB;
 
@@ skipping 182 matching lines @@
 
 
 void Sampler::DecreaseProfilingDepth() {
 #if defined(USE_SIGNALS)
   SignalHandler::DecreaseSamplerCount();
 #endif
   base::NoBarrier_AtomicIncrement(&profiling_, -1);
 }
 
 
-void Sampler::SampleStack(const RegisterState& state) {
+void Sampler::SampleStack(const RegisterState& state)
   TickSample* sample = isolate_->cpu_profiler()->StartTickSample();
   TickSample sample_obj;
   if (sample == NULL) sample = &sample_obj;
   sample->Init(isolate_, state);
   if (is_counting_samples_) {
     if (sample->state == JS || sample->state == EXTERNAL) {
       ++js_and_external_sample_count_;
     }
   }
   Tick(sample);
   if (sample != &sample_obj) {
     isolate_->cpu_profiler()->FinishTickSample();
   }
 }
 
 
 #if defined(USE_SIGNALS)
 
 void Sampler::DoSample() {
   if (!SignalHandler::Installed()) return;
   pthread_kill(platform_data()->vm_tid(), SIGPROF);
 }
 
+
+TickSample* Sampler::GetSample(TickSample* sample) {
+  if (!SignalHandler::Installed()) return NULL;
+  SignalHandler::called_from_get_sample_ = true;
+  pthread_kill(platform_data()->vm_tid(), SIGPROF);
+  SignalHandler::new_sample_semaphore_->Wait();
+  if (SignalHandler::new_sample_available_) {
+    sample->state = SignalHandler::sample_.state;
+    for (int i = 0; i < SignalHandler::sample_.frames_count; i++) {
+      sample->stack[i] = SignalHandler::sample_.stack[i];
+    }
+    sample->frames_count = SignalHandler::sample_.frames_count;
+  } else {
+    sample = NULL;
+  }
+  SignalHandler::called_from_get_sample_ = false;
+  return sample;
+}
+
 #elif V8_OS_WIN || V8_OS_CYGWIN
 
-void Sampler::DoSample() {
+TickSample* Sampler::GetSampleHelper_(TickSample * sample,
+                                      bool called_from_get_sample) {
   HANDLE profiled_thread = platform_data()->profiled_thread();
-  if (profiled_thread == NULL) return;
+  if (profiled_thread == NULL) return NULL;
 
 #if defined(USE_SIMULATOR)
   SimulatorHelper helper;
-  if (!helper.Init(this, isolate())) return;
+  if (!helper.Init(this, isolate())) return NULL;
 #endif
 
   const DWORD kSuspendFailed = static_cast<DWORD>(-1);
-  if (SuspendThread(profiled_thread) == kSuspendFailed) return;
+  if (SuspendThread(profiled_thread) == kSuspendFailed) return NULL;
 
   // Context used for sampling the register state of the profiled thread.
   CONTEXT context;
   memset(&context, 0, sizeof(context));
   context.ContextFlags = CONTEXT_FULL;
   if (GetThreadContext(profiled_thread, &context) != 0) {
     RegisterState state;
 #if defined(USE_SIMULATOR)
     helper.FillRegisters(&state);
 #else
 #if V8_HOST_ARCH_X64
     state.pc = reinterpret_cast<Address>(context.Rip);
     state.sp = reinterpret_cast<Address>(context.Rsp);
     state.fp = reinterpret_cast<Address>(context.Rbp);
 #else
     state.pc = reinterpret_cast<Address>(context.Eip);
     state.sp = reinterpret_cast<Address>(context.Esp);
     state.fp = reinterpret_cast<Address>(context.Ebp);
 #endif
 #endif  // USE_SIMULATOR
-    SampleStack(state);
+    if (called_from_get_sample) {
+      sample->Init(sampler->isolate(), state);
+    } else {
+      SampleStack(state);
+    }
   }
   ResumeThread(profiled_thread);
+  return sample;
+}
+
+
+void Sampler::DoSample() {
+  GetSampleHelper_(NULL, false);
+}
+
+
+TickSample* Sampler::GetSample(TickSample* sample) {
+  return GetSampleHelper_(sample, true);
 }
 
 #endif  // USE_SIGNALS
 
 
 } }  // namespace v8::internal