summaryrefslogtreecommitdiffstats
path: root/webkit/extensions/v8/profiler_extension.cc
blob: 18ff2801e2e2f6e50cdf6fe7970a6d44219cf3c4 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
// Copyright (c) 2011 The Chromium 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 "webkit/extensions/v8/profiler_extension.h"

#include "build/build_config.h"
#include "v8/include/v8.h"

#if defined(USE_TCMALLOC) && defined(OS_POSIX) && !defined(OS_MACOSX)
#include "third_party/tcmalloc/chromium/src/google/profiler.h"
#endif

const char kProfilerExtensionName[] = "v8/Profiler";

namespace extensions_v8 {

class ProfilerWrapper : public v8::Extension {
 public:
  ProfilerWrapper() :
      v8::Extension(kProfilerExtensionName,
        "if (typeof(chromium) == 'undefined') {"
        "  chromium = {};"
        "}"
        "chromium.Profiler = function() {"
        "  native function ProfilerStart();"
        "  native function ProfilerStop();"
        "  native function ProfilerFlush();"
        "  native function ProfilerClearData();"
        "  native function ProfilerSetThreadName();"
        "  this.start = function() {"
        "    ProfilerStart();"
        "  };"
        "  this.stop = function() {"
        "    ProfilerStop();"
        "  };"
        "  this.clear = function() {"
        "    ProfilerClearData();"
        "  };"
        "  this.flush = function() {"
        "    ProfilerFlush();"
        "  };"
        "  this.setThreadName = function(name) {"
        "    ProfilerSetThreadName(name);"
        "  };"
        "};") {}

  virtual v8::Handle<v8::FunctionTemplate> GetNativeFunction(
      v8::Handle<v8::String> name) {
    if (name->Equals(v8::String::New("ProfilerStart")))
      return v8::FunctionTemplate::New(ProfilerStart);
    else if (name->Equals(v8::String::New("ProfilerStop")))
      return v8::FunctionTemplate::New(ProfilerStop);
    else if (name->Equals(v8::String::New("ProfilerClearData")))
      return v8::FunctionTemplate::New(ProfilerClearData);
    else if (name->Equals(v8::String::New("ProfilerFlush")))
      return v8::FunctionTemplate::New(ProfilerFlush);
    else if (name->Equals(v8::String::New("ProfilerSetThreadName")))
      return v8::FunctionTemplate::New(ProfilerSetThreadName);

    return v8::Handle<v8::FunctionTemplate>();
  }

  static v8::Handle<v8::Value> ProfilerStart(const v8::Arguments& args) {
#if defined(USE_TCMALLOC) && defined(OS_POSIX) && !defined(OS_MACOSX)
    ::ProfilerStart("chrome-profile");
#endif
    return v8::Undefined();
  }

  static v8::Handle<v8::Value> ProfilerStop(const v8::Arguments& args) {
#if defined(USE_TCMALLOC) && defined(OS_POSIX) && !defined(OS_MACOSX)
    ::ProfilerStop();
#endif
    return v8::Undefined();
  }

  static v8::Handle<v8::Value> ProfilerClearData(const v8::Arguments& args) {
    return v8::Undefined();
  }

  static v8::Handle<v8::Value> ProfilerFlush(const v8::Arguments& args) {
#if defined(USE_TCMALLOC) && defined(OS_POSIX) && !defined(OS_MACOSX)
    ::ProfilerFlush();
#endif
    return v8::Undefined();
  }

  static v8::Handle<v8::Value> ProfilerSetThreadName(
      const v8::Arguments& args) {
    if (args.Length() >= 1 && args[0]->IsString()) {
      v8::Local<v8::String> inputString = args[0]->ToString();
      char nameBuffer[256];
      inputString->WriteAscii(nameBuffer, 0, sizeof(nameBuffer)-1);
    }
    return v8::Undefined();
  }
};

v8::Extension* ProfilerExtension::Get() {
  return new ProfilerWrapper();
}

}  // namespace extensions_v8