1 // Copyright 2014 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 "content/renderer/web_ui_mojo.h" 6 7 #include "content/common/view_messages.h" 8 #include "content/public/renderer/render_frame.h" 9 #include "content/public/renderer/render_view.h" 10 #include "content/renderer/web_ui_mojo_context_state.h" 11 #include "gin/per_context_data.h" 12 #include "third_party/WebKit/public/web/WebKit.h" 13 #include "third_party/WebKit/public/web/WebLocalFrame.h" 14 #include "third_party/WebKit/public/web/WebView.h" 15 #include "v8/include/v8.h" 16 17 namespace content { 18 19 namespace { 20 21 const char kWebUIMojoContextStateKey[] = "WebUIMojoContextState"; 22 23 struct WebUIMojoContextStateData : public base::SupportsUserData::Data { 24 scoped_ptr<WebUIMojoContextState> state; 25 }; 26 27 } // namespace 28 MainFrameObserver(WebUIMojo * web_ui_mojo)29WebUIMojo::MainFrameObserver::MainFrameObserver(WebUIMojo* web_ui_mojo) 30 : RenderFrameObserver(RenderFrame::FromWebFrame( 31 web_ui_mojo->render_view()->GetWebView()->mainFrame())), 32 web_ui_mojo_(web_ui_mojo) { 33 } 34 ~MainFrameObserver()35WebUIMojo::MainFrameObserver::~MainFrameObserver() { 36 } 37 WillReleaseScriptContext(v8::Handle<v8::Context> context,int world_id)38void WebUIMojo::MainFrameObserver::WillReleaseScriptContext( 39 v8::Handle<v8::Context> context, 40 int world_id) { 41 web_ui_mojo_->DestroyContextState(context); 42 } 43 DidFinishDocumentLoad()44void WebUIMojo::MainFrameObserver::DidFinishDocumentLoad() { 45 web_ui_mojo_->OnDidFinishDocumentLoad(); 46 } 47 WebUIMojo(RenderView * render_view)48WebUIMojo::WebUIMojo(RenderView* render_view) 49 : RenderViewObserver(render_view), 50 RenderViewObserverTracker<WebUIMojo>(render_view), 51 main_frame_observer_(this), 52 did_finish_document_load_(false) { 53 CreateContextState(); 54 } 55 SetBrowserHandle(mojo::ScopedMessagePipeHandle handle)56void WebUIMojo::SetBrowserHandle(mojo::ScopedMessagePipeHandle handle) { 57 if (did_finish_document_load_) 58 SetHandleOnContextState(handle.Pass()); 59 else 60 pending_handle_ = handle.Pass(); 61 } 62 ~WebUIMojo()63WebUIMojo::~WebUIMojo() { 64 } 65 CreateContextState()66void WebUIMojo::CreateContextState() { 67 v8::HandleScope handle_scope(blink::mainThreadIsolate()); 68 blink::WebLocalFrame* frame = 69 render_view()->GetWebView()->mainFrame()->toWebLocalFrame(); 70 v8::Handle<v8::Context> context = frame->mainWorldScriptContext(); 71 gin::PerContextData* context_data = gin::PerContextData::From(context); 72 WebUIMojoContextStateData* data = new WebUIMojoContextStateData; 73 data->state.reset(new WebUIMojoContextState( 74 render_view()->GetWebView()->mainFrame(), context)); 75 context_data->SetUserData(kWebUIMojoContextStateKey, data); 76 } 77 DestroyContextState(v8::Handle<v8::Context> context)78void WebUIMojo::DestroyContextState(v8::Handle<v8::Context> context) { 79 gin::PerContextData* context_data = gin::PerContextData::From(context); 80 if (!context_data) 81 return; 82 context_data->RemoveUserData(kWebUIMojoContextStateKey); 83 } 84 OnDidFinishDocumentLoad()85void WebUIMojo::OnDidFinishDocumentLoad() { 86 did_finish_document_load_ = true; 87 if (pending_handle_.is_valid()) 88 SetHandleOnContextState(pending_handle_.Pass()); 89 } 90 SetHandleOnContextState(mojo::ScopedMessagePipeHandle handle)91void WebUIMojo::SetHandleOnContextState(mojo::ScopedMessagePipeHandle handle) { 92 DCHECK(did_finish_document_load_); 93 v8::HandleScope handle_scope(blink::mainThreadIsolate()); 94 WebUIMojoContextState* state = GetContextState(); 95 if (state) 96 state->SetHandle(handle.Pass()); 97 } 98 GetContextState()99WebUIMojoContextState* WebUIMojo::GetContextState() { 100 blink::WebLocalFrame* frame = 101 render_view()->GetWebView()->mainFrame()->toWebLocalFrame(); 102 v8::HandleScope handle_scope(blink::mainThreadIsolate()); 103 v8::Handle<v8::Context> context = frame->mainWorldScriptContext(); 104 gin::PerContextData* context_data = gin::PerContextData::From(context); 105 if (!context_data) 106 return NULL; 107 WebUIMojoContextStateData* context_state = 108 static_cast<WebUIMojoContextStateData*>( 109 context_data->GetUserData(kWebUIMojoContextStateKey)); 110 return context_state ? context_state->state.get() : NULL; 111 } 112 DidClearWindowObject(blink::WebLocalFrame * frame)113void WebUIMojo::DidClearWindowObject(blink::WebLocalFrame* frame) { 114 if (frame != render_view()->GetWebView()->mainFrame()) 115 return; 116 117 // NOTE: this function may be called early on twice. From the constructor 118 // mainWorldScriptContext() may trigger this to be called. If we are created 119 // before the page is loaded (which is very likely), then on first load this 120 // is called. In the case of the latter we may have already supplied the 121 // handle to the context state so that if we destroy now the handle is 122 // lost. If this is the result of the first load then the contextstate should 123 // be empty and we don't need to destroy it. 124 WebUIMojoContextState* state = GetContextState(); 125 if (state && !state->module_added()) 126 return; 127 128 v8::HandleScope handle_scope(blink::mainThreadIsolate()); 129 DestroyContextState(frame->mainWorldScriptContext()); 130 CreateContextState(); 131 } 132 133 } // namespace content 134