1 // Copyright (c) 2013 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
5 #include "content/renderer/memory_benchmarking_extension.h"
7 #include "content/common/memory_benchmark_messages.h"
8 #include "content/renderer/render_thread_impl.h"
9 #include "gin/arguments.h"
10 #include "gin/handle.h"
11 #include "gin/object_template_builder.h"
12 #include "third_party/WebKit/public/web/WebFrame.h"
13 #include "third_party/WebKit/public/web/WebKit.h"
15 #if defined(USE_TCMALLOC) && (defined(OS_LINUX) || defined(OS_ANDROID))
16 #include "third_party/tcmalloc/chromium/src/gperftools/heap-profiler.h"
21 gin::WrapperInfo
MemoryBenchmarkingExtension::kWrapperInfo
= {
22 gin::kEmbedderNativeGin
};
25 void MemoryBenchmarkingExtension::Install(blink::WebFrame
* frame
) {
26 v8::Isolate
* isolate
= blink::mainThreadIsolate();
27 v8::HandleScope
handle_scope(isolate
);
28 v8::Handle
<v8::Context
> context
= frame
->mainWorldScriptContext();
29 if (context
.IsEmpty())
32 v8::Context::Scope
context_scope(context
);
33 gin::Handle
<MemoryBenchmarkingExtension
> controller
=
34 gin::CreateHandle(isolate
, new MemoryBenchmarkingExtension());
35 if (controller
.IsEmpty())
38 v8::Handle
<v8::Object
> global
= context
->Global();
39 v8::Handle
<v8::Object
> chrome
=
40 global
->Get(gin::StringToV8(isolate
, "chrome"))->ToObject();
41 if (chrome
.IsEmpty()) {
42 chrome
= v8::Object::New(isolate
);
43 global
->Set(gin::StringToV8(isolate
, "chrome"), chrome
);
45 chrome
->Set(gin::StringToV8(isolate
, "memoryBenchmarking"),
49 MemoryBenchmarkingExtension::MemoryBenchmarkingExtension() {}
51 MemoryBenchmarkingExtension::~MemoryBenchmarkingExtension() {}
53 gin::ObjectTemplateBuilder
54 MemoryBenchmarkingExtension::GetObjectTemplateBuilder(v8::Isolate
* isolate
) {
55 return gin::Wrappable
<MemoryBenchmarkingExtension
>::GetObjectTemplateBuilder(
57 .SetMethod("isHeapProfilerRunning",
58 &MemoryBenchmarkingExtension::IsHeapProfilerRunning
)
59 .SetMethod("heapProfilerDump",
60 &MemoryBenchmarkingExtension::HeapProfilerDump
);
63 bool MemoryBenchmarkingExtension::IsHeapProfilerRunning() {
64 #if defined(USE_TCMALLOC) && (defined(OS_LINUX) || defined(OS_ANDROID))
65 return ::IsHeapProfilerRunning();
71 void MemoryBenchmarkingExtension::HeapProfilerDump(gin::Arguments
* args
) {
72 std::string process_type
;
73 std::string
reason("benchmarking_extension");
75 if (args
->PeekNext()->IsString()) {
76 args
->GetNext(&process_type
);
77 if (args
->PeekNext()->IsString())
78 args
->GetNext(&reason
);
81 #if defined(USE_TCMALLOC) && (defined(OS_LINUX) || defined(OS_ANDROID))
82 if (process_type
== "browser") {
83 content::RenderThreadImpl::current()->Send(
84 new MemoryBenchmarkHostMsg_HeapProfilerDump(reason
));
86 ::HeapProfilerDump(reason
.c_str());
91 } // namespace content