Updating trunk VERSION from 2139.0 to 2140.0
[chromium-blink-merge.git] / content / renderer / memory_benchmarking_extension.cc
blobf6b9c7df375faeafedd419bb35deb8c3ded3848b
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"
17 #endif
19 namespace content {
21 gin::WrapperInfo MemoryBenchmarkingExtension::kWrapperInfo = {
22 gin::kEmbedderNativeGin};
24 // static
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())
30 return;
32 v8::Context::Scope context_scope(context);
33 gin::Handle<MemoryBenchmarkingExtension> controller =
34 gin::CreateHandle(isolate, new MemoryBenchmarkingExtension());
35 if (controller.IsEmpty())
36 return;
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"),
46 controller.ToV8());
49 MemoryBenchmarkingExtension::MemoryBenchmarkingExtension() {}
51 MemoryBenchmarkingExtension::~MemoryBenchmarkingExtension() {}
53 gin::ObjectTemplateBuilder
54 MemoryBenchmarkingExtension::GetObjectTemplateBuilder(v8::Isolate* isolate) {
55 return gin::Wrappable<MemoryBenchmarkingExtension>::GetObjectTemplateBuilder(
56 isolate)
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();
66 #else
67 return false;
68 #endif
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));
85 } else {
86 ::HeapProfilerDump(reason.c_str());
88 #endif
91 } // namespace content