1 // Copyright 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. 4 5 #include "chrome/renderer/extensions/console.h" 6 7 #include "base/compiler_specific.h" 8 #include "base/debug/alias.h" 9 #include "base/lazy_instance.h" 10 #include "base/strings/string_util.h" 11 #include "base/strings/stringprintf.h" 12 #include "base/strings/utf_string_conversions.h" 13 #include "chrome/renderer/extensions/dispatcher.h" 14 #include "chrome/renderer/extensions/extension_helper.h" 15 #include "content/public/renderer/render_view.h" 16 #include "content/public/renderer/render_view_visitor.h" 17 #include "third_party/WebKit/public/web/WebConsoleMessage.h" 18 #include "third_party/WebKit/public/web/WebFrame.h" 19 #include "third_party/WebKit/public/web/WebView.h" 20 21 namespace extensions { 22 namespace console { 23 24 namespace { 25 26 // Finds the RenderView associated with a context. Note: there will be multiple 27 // contexts in each RenderView. 28 class ByContextFinder : public content::RenderViewVisitor { 29 public: 30 static content::RenderView* Find(v8::Handle<v8::Context> context) { 31 ByContextFinder finder(context); 32 content::RenderView::ForEach(&finder); 33 return finder.found_; 34 } 35 36 private: 37 explicit ByContextFinder(v8::Handle<v8::Context> context) 38 : context_(context), found_(NULL) { 39 } 40 41 virtual bool Visit(content::RenderView* render_view) OVERRIDE { 42 ExtensionHelper* helper = ExtensionHelper::Get(render_view); 43 if (helper && 44 helper->dispatcher()->v8_context_set().GetByV8Context(context_)) { 45 found_ = render_view; 46 } 47 return !found_; 48 } 49 50 v8::Handle<v8::Context> context_; 51 content::RenderView* found_; 52 53 DISALLOW_COPY_AND_ASSIGN(ByContextFinder); 54 }; 55 56 // Writes |message| to stack to show up in minidump, then crashes. 57 void CheckWithMinidump(const std::string& message) { 58 char minidump[1024]; 59 base::debug::Alias(&minidump); 60 base::snprintf(minidump, arraysize(minidump), 61 "e::console: %s", message.c_str()); 62 CHECK(false) << message; 63 } 64 65 typedef void (*LogMethod)(v8::Handle<v8::Context> context, 66 const std::string& message); 67 68 void BoundLogMethodCallback(const v8::FunctionCallbackInfo<v8::Value>& info) { 69 LogMethod log_method = reinterpret_cast<LogMethod>( 70 info.Data().As<v8::External>()->Value()); 71 std::string message; 72 for (int i = 0; i < info.Length(); ++i) { 73 if (i > 0) 74 message += " "; 75 message += *v8::String::Utf8Value(info[i]); 76 } 77 (*log_method)(info.GetIsolate()->GetCallingContext(), message); 78 } 79 80 void BindLogMethod(v8::Isolate* isolate, 81 v8::Local<v8::Object> target, 82 const std::string& name, 83 LogMethod log_method) { 84 v8::Local<v8::FunctionTemplate> tmpl = v8::FunctionTemplate::New( 85 isolate, 86 &BoundLogMethodCallback, 87 v8::External::New(isolate, reinterpret_cast<void*>(log_method))); 88 target->Set(v8::String::NewFromUtf8(isolate, name.c_str()), 89 tmpl->GetFunction()); 90 } 91 92 } // namespace 93 94 void Debug(content::RenderView* render_view, const std::string& message) { 95 AddMessage(render_view, content::CONSOLE_MESSAGE_LEVEL_DEBUG, message); 96 } 97 98 void Log(content::RenderView* render_view, const std::string& message) { 99 AddMessage(render_view, content::CONSOLE_MESSAGE_LEVEL_LOG, message); 100 } 101 102 void Warn(content::RenderView* render_view, const std::string& message) { 103 AddMessage(render_view, content::CONSOLE_MESSAGE_LEVEL_WARNING, message); 104 } 105 106 void Error(content::RenderView* render_view, const std::string& message) { 107 AddMessage(render_view, content::CONSOLE_MESSAGE_LEVEL_ERROR, message); 108 } 109 110 void Fatal(content::RenderView* render_view, const std::string& message) { 111 Error(render_view, message); 112 CheckWithMinidump(message); 113 } 114 115 void AddMessage(content::RenderView* render_view, 116 content::ConsoleMessageLevel level, 117 const std::string& message) { 118 blink::WebView* web_view = render_view->GetWebView(); 119 if (!web_view || !web_view->mainFrame()) 120 return; 121 blink::WebConsoleMessage::Level target_level = 122 blink::WebConsoleMessage::LevelLog; 123 switch (level) { 124 case content::CONSOLE_MESSAGE_LEVEL_DEBUG: 125 target_level = blink::WebConsoleMessage::LevelDebug; 126 break; 127 case content::CONSOLE_MESSAGE_LEVEL_LOG: 128 target_level = blink::WebConsoleMessage::LevelLog; 129 break; 130 case content::CONSOLE_MESSAGE_LEVEL_WARNING: 131 target_level = blink::WebConsoleMessage::LevelWarning; 132 break; 133 case content::CONSOLE_MESSAGE_LEVEL_ERROR: 134 target_level = blink::WebConsoleMessage::LevelError; 135 break; 136 } 137 web_view->mainFrame()->addMessageToConsole( 138 blink::WebConsoleMessage(target_level, ASCIIToUTF16(message))); 139 } 140 141 void Debug(v8::Handle<v8::Context> context, const std::string& message) { 142 AddMessage(context, content::CONSOLE_MESSAGE_LEVEL_DEBUG, message); 143 } 144 145 void Log(v8::Handle<v8::Context> context, const std::string& message) { 146 AddMessage(context, content::CONSOLE_MESSAGE_LEVEL_LOG, message); 147 } 148 149 void Warn(v8::Handle<v8::Context> context, const std::string& message) { 150 AddMessage(context, content::CONSOLE_MESSAGE_LEVEL_WARNING, message); 151 } 152 153 void Error(v8::Handle<v8::Context> context, const std::string& message) { 154 AddMessage(context, content::CONSOLE_MESSAGE_LEVEL_ERROR, message); 155 } 156 157 void Fatal(v8::Handle<v8::Context> context, const std::string& message) { 158 Error(context, message); 159 CheckWithMinidump(message); 160 } 161 162 void AddMessage(v8::Handle<v8::Context> context, 163 content::ConsoleMessageLevel level, 164 const std::string& message) { 165 if (context.IsEmpty()) { 166 LOG(WARNING) << "Could not log \"" << message << "\": no context given"; 167 return; 168 } 169 content::RenderView* render_view = ByContextFinder::Find(context); 170 if (!render_view) { 171 LOG(WARNING) << "Could not log \"" << message << "\": no render view found"; 172 return; 173 } 174 AddMessage(render_view, level, message); 175 } 176 177 v8::Local<v8::Object> AsV8Object() { 178 v8::Isolate* isolate = v8::Isolate::GetCurrent(); 179 v8::EscapableHandleScope handle_scope(isolate); 180 v8::Local<v8::Object> console_object = v8::Object::New(); 181 BindLogMethod(isolate, console_object, "debug", &Debug); 182 BindLogMethod(isolate, console_object, "log", &Log); 183 BindLogMethod(isolate, console_object, "warn", &Warn); 184 BindLogMethod(isolate, console_object, "error", &Error); 185 return handle_scope.Escape(console_object); 186 } 187 188 } // namespace console 189 } // namespace extensions 190
