| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "chrome/renderer/extensions/user_script_scheduler.h" | 5 #include "chrome/renderer/extensions/user_script_scheduler.h" |
| 6 | 6 |
| 7 #include "base/bind.h" | 7 #include "base/bind.h" |
| 8 #include "base/logging.h" | 8 #include "base/logging.h" |
| 9 #include "base/message_loop/message_loop.h" | 9 #include "base/message_loop/message_loop.h" |
| 10 #include "chrome/common/extensions/extension_manifest_constants.h" | 10 #include "chrome/common/extensions/extension_manifest_constants.h" |
| 11 #include "chrome/common/extensions/extension_messages.h" | 11 #include "chrome/common/extensions/extension_messages.h" |
| 12 #include "chrome/common/extensions/permissions/permissions_data.h" | 12 #include "chrome/common/extensions/permissions/permissions_data.h" |
| 13 #include "chrome/renderer/chrome_render_process_observer.h" | 13 #include "chrome/renderer/chrome_render_process_observer.h" |
| 14 #include "chrome/renderer/extensions/chrome_v8_context.h" |
| 14 #include "chrome/renderer/extensions/dispatcher.h" | 15 #include "chrome/renderer/extensions/dispatcher.h" |
| 15 #include "chrome/renderer/extensions/dom_activity_logger.h" | 16 #include "chrome/renderer/extensions/dom_activity_logger.h" |
| 16 #include "chrome/renderer/extensions/extension_groups.h" | 17 #include "chrome/renderer/extensions/extension_groups.h" |
| 17 #include "chrome/renderer/extensions/extension_helper.h" | 18 #include "chrome/renderer/extensions/extension_helper.h" |
| 18 #include "chrome/renderer/extensions/user_script_slave.h" | 19 #include "chrome/renderer/extensions/user_script_slave.h" |
| 19 #include "content/public/renderer/render_view.h" | 20 #include "content/public/renderer/render_view.h" |
| 20 #include "content/public/renderer/v8_value_converter.h" | 21 #include "content/public/renderer/v8_value_converter.h" |
| 21 #include "extensions/common/error_utils.h" | 22 #include "extensions/common/error_utils.h" |
| 22 #include "third_party/WebKit/public/platform/WebString.h" | 23 #include "third_party/WebKit/public/platform/WebString.h" |
| 23 #include "third_party/WebKit/public/platform/WebVector.h" | 24 #include "third_party/WebKit/public/platform/WebVector.h" |
| (...skipping 168 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 192 continue; | 193 continue; |
| 193 } else { | 194 } else { |
| 194 error = ErrorUtils::FormatErrorMessage( | 195 error = ErrorUtils::FormatErrorMessage( |
| 195 extension_manifest_errors::kCannotAccessPage, | 196 extension_manifest_errors::kCannotAccessPage, |
| 196 child_frame->document().url().spec()); | 197 child_frame->document().url().spec()); |
| 197 break; | 198 break; |
| 198 } | 199 } |
| 199 } | 200 } |
| 200 | 201 |
| 201 WebScriptSource source(WebString::fromUTF8(params.code)); | 202 WebScriptSource source(WebString::fromUTF8(params.code)); |
| 202 v8::Isolate* isolate = v8::Isolate::GetCurrent(); | 203 v8::HandleScope scope; |
| 203 v8::HandleScope scope(isolate); | |
| 204 | 204 |
| 205 scoped_ptr<content::V8ValueConverter> v8_converter( | 205 scoped_ptr<content::V8ValueConverter> v8_converter( |
| 206 content::V8ValueConverter::create()); | 206 content::V8ValueConverter::create()); |
| 207 v8::Handle<v8::Value> script_value; | 207 v8::Local<v8::Value> script_value; |
| 208 | 208 |
| 209 if (params.in_main_world) { | 209 if (params.in_main_world) { |
| 210 DOMActivityLogger::AttachToWorld( | 210 DOMActivityLogger::AttachToWorld( |
| 211 DOMActivityLogger::kMainWorldId, | 211 DOMActivityLogger::kMainWorldId, |
| 212 extension->id(), | 212 extension->id(), |
| 213 UserScriptSlave::GetDataSourceURLForFrame(child_frame), | 213 UserScriptSlave::GetDataSourceURLForFrame(child_frame), |
| 214 child_frame->document().title()); | 214 child_frame->document().title()); |
| 215 script_value = child_frame->executeScriptAndReturnValue(source); | 215 script_value = child_frame->executeScriptAndReturnValue(source); |
| 216 } else { | 216 } else { |
| 217 WebKit::WebVector<v8::Local<v8::Value> > results; | 217 WebKit::WebVector<v8::Local<v8::Value> > results; |
| 218 std::vector<WebScriptSource> sources; | 218 std::vector<WebScriptSource> sources; |
| 219 sources.push_back(source); | 219 sources.push_back(source); |
| 220 int isolated_world_id = | 220 int isolated_world_id = |
| 221 dispatcher_->user_script_slave()->GetIsolatedWorldIdForExtension( | 221 dispatcher_->user_script_slave()->GetIsolatedWorldIdForExtension( |
| 222 extension, child_frame); | 222 extension, child_frame); |
| 223 DOMActivityLogger::AttachToWorld( | 223 DOMActivityLogger::AttachToWorld( |
| 224 isolated_world_id, | 224 isolated_world_id, |
| 225 extension->id(), | 225 extension->id(), |
| 226 UserScriptSlave::GetDataSourceURLForFrame(child_frame), | 226 UserScriptSlave::GetDataSourceURLForFrame(child_frame), |
| 227 child_frame->document().title()); | 227 child_frame->document().title()); |
| 228 child_frame->executeScriptInIsolatedWorld( | 228 child_frame->executeScriptInIsolatedWorld( |
| 229 isolated_world_id, &sources.front(), | 229 isolated_world_id, &sources.front(), |
| 230 sources.size(), EXTENSION_GROUP_CONTENT_SCRIPTS, &results); | 230 sources.size(), EXTENSION_GROUP_CONTENT_SCRIPTS, &results); |
| 231 // We only expect one value back since we only pushed one source | 231 // We only expect one value back since we only pushed one source |
| 232 if (results.size() == 1 && !results[0].IsEmpty()) | 232 if (results.size() == 1 && !results[0].IsEmpty()) |
| 233 script_value = results[0]; | 233 script_value = results[0]; |
| 234 } | 234 } |
| 235 if (!script_value.IsEmpty()) { | 235 |
| 236 v8::Local<v8::Context> context = v8::Context::New(isolate); | 236 if (params.wants_result && !script_value.IsEmpty()) { |
| 237 base::Value* base_val = | 237 // It's safe to always use the main world context when converting here. |
| 238 v8_converter->FromV8Value(script_value, context); | 238 // V8ValueConverterImpl shouldn't actually care about the context scope, |
| 239 // and it switches to v8::Object's creation context when encountered. |
| 240 v8::Local<v8::Context> context = child_frame->mainWorldScriptContext(); |
| 241 base::Value* result = v8_converter->FromV8Value(script_value, context); |
| 239 // Always append an execution result (i.e. no result == null result) so | 242 // Always append an execution result (i.e. no result == null result) so |
| 240 // that |execution_results| lines up with the frames. | 243 // that |execution_results| lines up with the frames. |
| 241 execution_results.Append(base_val ? base_val : | 244 execution_results.Append( |
| 242 base::Value::CreateNullValue()); | 245 result ? result : base::Value::CreateNullValue()); |
| 243 script_value.Clear(); | |
| 244 } | 246 } |
| 245 } else { | 247 } else { |
| 246 child_frame->document().insertUserStyleSheet( | 248 child_frame->document().insertUserStyleSheet( |
| 247 WebString::fromUTF8(params.code), | 249 WebString::fromUTF8(params.code), |
| 248 // Author level is consistent with WebView::addUserStyleSheet. | 250 // Author level is consistent with WebView::addUserStyleSheet. |
| 249 WebDocument::UserStyleAuthorLevel); | 251 WebDocument::UserStyleAuthorLevel); |
| 250 } | 252 } |
| 251 } | 253 } |
| 252 | 254 |
| 253 render_view->Send(new ExtensionHostMsg_ExecuteCodeFinished( | 255 render_view->Send(new ExtensionHostMsg_ExecuteCodeFinished( |
| (...skipping 13 matching lines...) Expand all Loading... |
| 267 | 269 |
| 268 for (WebFrame* child_frame = parent_frame->firstChild(); child_frame; | 270 for (WebFrame* child_frame = parent_frame->firstChild(); child_frame; |
| 269 child_frame = child_frame->nextSibling()) { | 271 child_frame = child_frame->nextSibling()) { |
| 270 frames_vector->push_back(child_frame); | 272 frames_vector->push_back(child_frame); |
| 271 GetAllChildFrames(child_frame, frames_vector); | 273 GetAllChildFrames(child_frame, frames_vector); |
| 272 } | 274 } |
| 273 return true; | 275 return true; |
| 274 } | 276 } |
| 275 | 277 |
| 276 } // namespace extensions | 278 } // namespace extensions |
| OLD | NEW |