mirror of
https://github.com/mozilla/gecko-dev.git
synced 2024-10-31 22:25:30 +00:00
305 lines
8.9 KiB
C++
305 lines
8.9 KiB
C++
// Copyright (c) 2006-2011 The Chromium Authors. All rights reserved.
|
|
//
|
|
// Redistribution and use in source and binary forms, with or without
|
|
// modification, are permitted provided that the following conditions
|
|
// are met:
|
|
// * Redistributions of source code must retain the above copyright
|
|
// notice, this list of conditions and the following disclaimer.
|
|
// * Redistributions in binary form must reproduce the above copyright
|
|
// notice, this list of conditions and the following disclaimer in
|
|
// the documentation and/or other materials provided with the
|
|
// distribution.
|
|
// * Neither the name of Google, Inc. nor the names of its contributors
|
|
// may be used to endorse or promote products derived from this
|
|
// software without specific prior written permission.
|
|
//
|
|
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
|
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
|
|
// FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
|
|
// COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
|
|
// INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
|
|
// BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS
|
|
// OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
|
|
// AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
|
|
// OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
|
|
// OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
|
|
// SUCH DAMAGE.
|
|
|
|
#include <windows.h>
|
|
#include <mmsystem.h>
|
|
#include <process.h>
|
|
#include "platform.h"
|
|
#include "TableTicker.h"
|
|
#include "ProfileEntry.h"
|
|
#include "UnwinderThread2.h"
|
|
|
|
class PlatformData : public Malloced {
|
|
public:
|
|
// Get a handle to the calling thread. This is the thread that we are
|
|
// going to profile. We need to make a copy of the handle because we are
|
|
// going to use it in the sampler thread. Using GetThreadHandle() will
|
|
// not work in this case. We're using OpenThread because DuplicateHandle
|
|
// for some reason doesn't work in Chrome's sandbox.
|
|
PlatformData(int aThreadId) : profiled_thread_(OpenThread(THREAD_GET_CONTEXT |
|
|
THREAD_SUSPEND_RESUME |
|
|
THREAD_QUERY_INFORMATION,
|
|
false,
|
|
aThreadId)) {}
|
|
|
|
~PlatformData() {
|
|
if (profiled_thread_ != NULL) {
|
|
CloseHandle(profiled_thread_);
|
|
profiled_thread_ = NULL;
|
|
}
|
|
}
|
|
|
|
HANDLE profiled_thread() { return profiled_thread_; }
|
|
|
|
private:
|
|
HANDLE profiled_thread_;
|
|
};
|
|
|
|
/* static */ PlatformData*
|
|
Sampler::AllocPlatformData(int aThreadId)
|
|
{
|
|
return new PlatformData(aThreadId);
|
|
}
|
|
|
|
/* static */ void
|
|
Sampler::FreePlatformData(PlatformData* aData)
|
|
{
|
|
delete aData;
|
|
}
|
|
|
|
uintptr_t
|
|
Sampler::GetThreadHandle(PlatformData* aData)
|
|
{
|
|
return (uintptr_t) aData->profiled_thread();
|
|
}
|
|
|
|
class SamplerThread : public Thread {
|
|
public:
|
|
SamplerThread(int interval, Sampler* sampler)
|
|
: Thread("SamplerThread")
|
|
, interval_(interval)
|
|
, sampler_(sampler) {}
|
|
|
|
static void StartSampler(Sampler* sampler) {
|
|
if (instance_ == NULL) {
|
|
instance_ = new SamplerThread(sampler->interval(), sampler);
|
|
instance_->Start();
|
|
} else {
|
|
ASSERT(instance_->interval_ == sampler->interval());
|
|
}
|
|
}
|
|
|
|
static void StopSampler() {
|
|
instance_->Join();
|
|
delete instance_;
|
|
instance_ = NULL;
|
|
}
|
|
|
|
// Implement Thread::Run().
|
|
virtual void Run() {
|
|
|
|
// By default we'll not adjust the timer resolution which tends to be around
|
|
// 16ms. However, if the requested interval is sufficiently low we'll try to
|
|
// adjust the resolution to match.
|
|
if (interval_ < 10)
|
|
::timeBeginPeriod(interval_);
|
|
|
|
while (sampler_->IsActive()) {
|
|
{
|
|
mozilla::MutexAutoLock lock(*Sampler::sRegisteredThreadsMutex);
|
|
std::vector<ThreadInfo*> threads =
|
|
sampler_->GetRegisteredThreads();
|
|
for (uint32_t i = 0; i < threads.size(); i++) {
|
|
ThreadInfo* info = threads[i];
|
|
|
|
// This will be null if we're not interested in profiling this thread.
|
|
if (!info->Profile())
|
|
continue;
|
|
|
|
ThreadProfile* thread_profile = info->Profile();
|
|
|
|
if (!sampler_->IsPaused()) {
|
|
SampleContext(sampler_, thread_profile);
|
|
}
|
|
}
|
|
}
|
|
OS::Sleep(interval_);
|
|
}
|
|
|
|
// disable any timer resolution changes we've made
|
|
if (interval_ < 10)
|
|
::timeEndPeriod(interval_);
|
|
}
|
|
|
|
void SampleContext(Sampler* sampler, ThreadProfile* thread_profile) {
|
|
uintptr_t thread = Sampler::GetThreadHandle(
|
|
thread_profile->GetPlatformData());
|
|
HANDLE profiled_thread = reinterpret_cast<HANDLE>(thread);
|
|
if (profiled_thread == NULL)
|
|
return;
|
|
|
|
// Context used for sampling the register state of the profiled thread.
|
|
CONTEXT context;
|
|
memset(&context, 0, sizeof(context));
|
|
|
|
TickSample sample_obj;
|
|
TickSample* sample = &sample_obj;
|
|
|
|
// Grab the timestamp before pausing the thread, to avoid deadlocks.
|
|
sample->timestamp = mozilla::TimeStamp::Now();
|
|
sample->threadProfile = thread_profile;
|
|
|
|
static const DWORD kSuspendFailed = static_cast<DWORD>(-1);
|
|
if (SuspendThread(profiled_thread) == kSuspendFailed)
|
|
return;
|
|
|
|
context.ContextFlags = CONTEXT_CONTROL;
|
|
if (GetThreadContext(profiled_thread, &context) != 0) {
|
|
#if V8_HOST_ARCH_X64
|
|
sample->pc = reinterpret_cast<Address>(context.Rip);
|
|
sample->sp = reinterpret_cast<Address>(context.Rsp);
|
|
sample->fp = reinterpret_cast<Address>(context.Rbp);
|
|
#else
|
|
sample->pc = reinterpret_cast<Address>(context.Eip);
|
|
sample->sp = reinterpret_cast<Address>(context.Esp);
|
|
sample->fp = reinterpret_cast<Address>(context.Ebp);
|
|
#endif
|
|
sample->context = &context;
|
|
sampler->Tick(sample);
|
|
}
|
|
ResumeThread(profiled_thread);
|
|
}
|
|
|
|
Sampler* sampler_;
|
|
const int interval_;
|
|
|
|
// Protects the process wide state below.
|
|
static SamplerThread* instance_;
|
|
|
|
DISALLOW_COPY_AND_ASSIGN(SamplerThread);
|
|
};
|
|
|
|
SamplerThread* SamplerThread::instance_ = NULL;
|
|
|
|
|
|
Sampler::Sampler(int interval, bool profiling, int entrySize)
|
|
: interval_(interval),
|
|
profiling_(profiling),
|
|
paused_(false),
|
|
active_(false),
|
|
entrySize_(entrySize) {
|
|
}
|
|
|
|
Sampler::~Sampler() {
|
|
ASSERT(!IsActive());
|
|
}
|
|
|
|
void Sampler::Start() {
|
|
ASSERT(!IsActive());
|
|
SetActive(true);
|
|
SamplerThread::StartSampler(this);
|
|
}
|
|
|
|
void Sampler::Stop() {
|
|
ASSERT(IsActive());
|
|
SetActive(false);
|
|
SamplerThread::StopSampler();
|
|
}
|
|
|
|
|
|
static const HANDLE kNoThread = INVALID_HANDLE_VALUE;
|
|
|
|
static unsigned int __stdcall ThreadEntry(void* arg) {
|
|
Thread* thread = reinterpret_cast<Thread*>(arg);
|
|
thread->Run();
|
|
return 0;
|
|
}
|
|
|
|
// Initialize a Win32 thread object. The thread has an invalid thread
|
|
// handle until it is started.
|
|
Thread::Thread(const char* name)
|
|
: stack_size_(0) {
|
|
thread_ = kNoThread;
|
|
set_name(name);
|
|
}
|
|
|
|
void Thread::set_name(const char* name) {
|
|
strncpy(name_, name, sizeof(name_));
|
|
name_[sizeof(name_) - 1] = '\0';
|
|
}
|
|
|
|
// Close our own handle for the thread.
|
|
Thread::~Thread() {
|
|
if (thread_ != kNoThread) CloseHandle(thread_);
|
|
}
|
|
|
|
// Create a new thread. It is important to use _beginthreadex() instead of
|
|
// the Win32 function CreateThread(), because the CreateThread() does not
|
|
// initialize thread specific structures in the C runtime library.
|
|
void Thread::Start() {
|
|
thread_ = reinterpret_cast<HANDLE>(
|
|
_beginthreadex(NULL,
|
|
static_cast<unsigned>(stack_size_),
|
|
ThreadEntry,
|
|
this,
|
|
0,
|
|
&thread_id_));
|
|
}
|
|
|
|
// Wait for thread to terminate.
|
|
void Thread::Join() {
|
|
if (thread_id_ != GetCurrentThreadId()) {
|
|
WaitForSingleObject(thread_, INFINITE);
|
|
}
|
|
}
|
|
|
|
void OS::Sleep(int milliseconds) {
|
|
::Sleep(milliseconds);
|
|
}
|
|
|
|
bool Sampler::RegisterCurrentThread(const char* aName,
|
|
PseudoStack* aPseudoStack,
|
|
bool aIsMainThread, void* stackTop)
|
|
{
|
|
if (!Sampler::sRegisteredThreadsMutex)
|
|
return false;
|
|
|
|
mozilla::MutexAutoLock lock(*Sampler::sRegisteredThreadsMutex);
|
|
|
|
ThreadInfo* info = new ThreadInfo(aName, GetCurrentThreadId(),
|
|
aIsMainThread, aPseudoStack);
|
|
|
|
if (sActiveSampler) {
|
|
sActiveSampler->RegisterThread(info);
|
|
}
|
|
|
|
sRegisteredThreads->push_back(info);
|
|
|
|
uwt__register_thread_for_profiling(stackTop);
|
|
return true;
|
|
}
|
|
|
|
void Sampler::UnregisterCurrentThread()
|
|
{
|
|
if (!Sampler::sRegisteredThreadsMutex)
|
|
return;
|
|
|
|
mozilla::MutexAutoLock lock(*Sampler::sRegisteredThreadsMutex);
|
|
|
|
int id = GetCurrentThreadId();
|
|
|
|
for (uint32_t i = 0; i < sRegisteredThreads->size(); i++) {
|
|
ThreadInfo* info = sRegisteredThreads->at(i);
|
|
if (info->ThreadId() == id) {
|
|
delete info;
|
|
sRegisteredThreads->erase(sRegisteredThreads->begin() + i);
|
|
break;
|
|
}
|
|
}
|
|
}
|