| 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 167 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 191 if (child_frame->parent()) { | 192 if (child_frame->parent()) { |
| 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 |
| 202 ChromeV8Context* context = dispatcher_->v8_context_set().GetByWebFrame( |
| 203 child_frame); |
| 204 CHECK(context) << "No v8 context found for frame at " |
| 205 << std::string(child_frame->document().url().spec()); |
| 206 v8::Context::Scope context_scope(context->v8_context()); |
| 207 v8::HandleScope handle_scope(context->v8_context()->GetIsolate()); |
| 208 |
| 201 WebScriptSource source(WebString::fromUTF8(params.code)); | 209 WebScriptSource source(WebString::fromUTF8(params.code)); |
| 202 v8::Isolate* isolate = v8::Isolate::GetCurrent(); | |
| 203 v8::HandleScope scope(isolate); | |
| 204 | 210 |
| 205 scoped_ptr<content::V8ValueConverter> v8_converter( | 211 scoped_ptr<content::V8ValueConverter> v8_converter( |
| 206 content::V8ValueConverter::create()); | 212 content::V8ValueConverter::create()); |
| 207 v8::Handle<v8::Value> script_value; | 213 v8::Local<v8::Value> script_value; |
| 208 | 214 |
| 209 if (params.in_main_world) { | 215 if (params.in_main_world) { |
| 210 DOMActivityLogger::AttachToWorld( | 216 DOMActivityLogger::AttachToWorld( |
| 211 DOMActivityLogger::kMainWorldId, | 217 DOMActivityLogger::kMainWorldId, |
| 212 extension->id(), | 218 extension->id(), |
| 213 UserScriptSlave::GetDataSourceURLForFrame(child_frame), | 219 UserScriptSlave::GetDataSourceURLForFrame(child_frame), |
| 214 child_frame->document().title()); | 220 child_frame->document().title()); |
| 215 script_value = child_frame->executeScriptAndReturnValue(source); | 221 script_value = child_frame->executeScriptAndReturnValue(source); |
| 216 } else { | 222 } else { |
| 217 WebKit::WebVector<v8::Local<v8::Value> > results; | 223 WebKit::WebVector<v8::Local<v8::Value> > results; |
| 218 std::vector<WebScriptSource> sources; | 224 std::vector<WebScriptSource> sources; |
| 219 sources.push_back(source); | 225 sources.push_back(source); |
| 220 int isolated_world_id = | 226 int isolated_world_id = |
| 221 dispatcher_->user_script_slave()->GetIsolatedWorldIdForExtension( | 227 dispatcher_->user_script_slave()->GetIsolatedWorldIdForExtension( |
| 222 extension, child_frame); | 228 extension, child_frame); |
| 223 DOMActivityLogger::AttachToWorld( | 229 DOMActivityLogger::AttachToWorld( |
| 224 isolated_world_id, | 230 isolated_world_id, |
| 225 extension->id(), | 231 extension->id(), |
| 226 UserScriptSlave::GetDataSourceURLForFrame(child_frame), | 232 UserScriptSlave::GetDataSourceURLForFrame(child_frame), |
| 227 child_frame->document().title()); | 233 child_frame->document().title()); |
| 228 child_frame->executeScriptInIsolatedWorld( | 234 child_frame->executeScriptInIsolatedWorld( |
| 229 isolated_world_id, &sources.front(), | 235 isolated_world_id, &sources.front(), |
| 230 sources.size(), EXTENSION_GROUP_CONTENT_SCRIPTS, &results); | 236 sources.size(), EXTENSION_GROUP_CONTENT_SCRIPTS, &results); |
| 231 // We only expect one value back since we only pushed one source | 237 // We only expect one value back since we only pushed one source |
| 232 if (results.size() == 1 && !results[0].IsEmpty()) | 238 if (results.size() == 1 && !results[0].IsEmpty()) |
| 233 script_value = results[0]; | 239 script_value = results[0]; |
| 234 } | 240 } |
| 235 if (!script_value.IsEmpty()) { | 241 |
| 236 v8::Local<v8::Context> context = v8::Context::New(isolate); | 242 if (params.wants_result && !script_value.IsEmpty()) { |
| 237 base::Value* base_val = | 243 base::Value* base_val = |
| 238 v8_converter->FromV8Value(script_value, context); | 244 v8_converter->FromV8Value(script_value, context->v8_context()); |
| 239 // Always append an execution result (i.e. no result == null result) so | 245 // Always append an execution result (i.e. no result == null result) so |
| 240 // that |execution_results| lines up with the frames. | 246 // that |execution_results| lines up with the frames. |
| 241 execution_results.Append(base_val ? base_val : | 247 execution_results.Append( |
| 242 base::Value::CreateNullValue()); | 248 base_val ? base_val : base::Value::CreateNullValue()); |
| 243 script_value.Clear(); | |
| 244 } | 249 } |
| 245 } else { | 250 } else { |
| 246 child_frame->document().insertUserStyleSheet( | 251 child_frame->document().insertUserStyleSheet( |
| 247 WebString::fromUTF8(params.code), | 252 WebString::fromUTF8(params.code), |
| 248 // Author level is consistent with WebView::addUserStyleSheet. | 253 // Author level is consistent with WebView::addUserStyleSheet. |
| 249 WebDocument::UserStyleAuthorLevel); | 254 WebDocument::UserStyleAuthorLevel); |
| 250 } | 255 } |
| 251 } | 256 } |
| 252 | 257 |
| 253 render_view->Send(new ExtensionHostMsg_ExecuteCodeFinished( | 258 render_view->Send(new ExtensionHostMsg_ExecuteCodeFinished( |
| (...skipping 13 matching lines...) Expand all Loading... |
| 267 | 272 |
| 268 for (WebFrame* child_frame = parent_frame->firstChild(); child_frame; | 273 for (WebFrame* child_frame = parent_frame->firstChild(); child_frame; |
| 269 child_frame = child_frame->nextSibling()) { | 274 child_frame = child_frame->nextSibling()) { |
| 270 frames_vector->push_back(child_frame); | 275 frames_vector->push_back(child_frame); |
| 271 GetAllChildFrames(child_frame, frames_vector); | 276 GetAllChildFrames(child_frame, frames_vector); |
| 272 } | 277 } |
| 273 return true; | 278 return true; |
| 274 } | 279 } |
| 275 | 280 |
| 276 } // namespace extensions | 281 } // namespace extensions |
| OLD | NEW |