diff options
Diffstat (limited to 'chromium/content/browser/webui/web_ui_impl.cc')
-rw-r--r-- | chromium/content/browser/webui/web_ui_impl.cc | 239 |
1 files changed, 239 insertions, 0 deletions
diff --git a/chromium/content/browser/webui/web_ui_impl.cc b/chromium/content/browser/webui/web_ui_impl.cc new file mode 100644 index 00000000000..e54a6d8c27d --- /dev/null +++ b/chromium/content/browser/webui/web_ui_impl.cc @@ -0,0 +1,239 @@ +// Copyright (c) 2012 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 "content/browser/webui/web_ui_impl.h" + +#include "base/json/json_writer.h" +#include "base/strings/utf_string_conversions.h" +#include "base/values.h" +#include "content/browser/child_process_security_policy_impl.h" +#include "content/browser/renderer_host/dip_util.h" +#include "content/browser/renderer_host/render_process_host_impl.h" +#include "content/browser/renderer_host/render_view_host_impl.h" +#include "content/browser/web_contents/web_contents_impl.h" +#include "content/browser/webui/web_ui_controller_factory_registry.h" +#include "content/common/view_messages.h" +#include "content/public/browser/content_browser_client.h" +#include "content/public/browser/web_contents_delegate.h" +#include "content/public/browser/web_contents_view.h" +#include "content/public/browser/web_ui_controller.h" +#include "content/public/browser/web_ui_message_handler.h" +#include "content/public/common/bindings_policy.h" +#include "content/public/common/content_client.h" + +namespace content { + +const WebUI::TypeID WebUI::kNoWebUI = NULL; + +// static +string16 WebUI::GetJavascriptCall( + const std::string& function_name, + const std::vector<const Value*>& arg_list) { + string16 parameters; + std::string json; + for (size_t i = 0; i < arg_list.size(); ++i) { + if (i > 0) + parameters += char16(','); + + base::JSONWriter::Write(arg_list[i], &json); + parameters += UTF8ToUTF16(json); + } + return ASCIIToUTF16(function_name) + + char16('(') + parameters + char16(')') + char16(';'); +} + +WebUIImpl::WebUIImpl(WebContents* contents) + : link_transition_type_(PAGE_TRANSITION_LINK), + bindings_(BINDINGS_POLICY_WEB_UI), + web_contents_(contents) { + DCHECK(contents); +} + +WebUIImpl::~WebUIImpl() { + // Delete the controller first, since it may also be keeping a pointer to some + // of the handlers and can call them at destruction. + controller_.reset(); +} + +// WebUIImpl, public: ---------------------------------------------------------- + +bool WebUIImpl::OnMessageReceived(const IPC::Message& message) { + bool handled = true; + IPC_BEGIN_MESSAGE_MAP(WebUIImpl, message) + IPC_MESSAGE_HANDLER(ViewHostMsg_WebUISend, OnWebUISend) + IPC_MESSAGE_UNHANDLED(handled = false) + IPC_END_MESSAGE_MAP() + return handled; +} + +void WebUIImpl::OnWebUISend(const GURL& source_url, + const std::string& message, + const ListValue& args) { + WebContentsDelegate* delegate = web_contents_->GetDelegate(); + bool data_urls_allowed = delegate && delegate->CanLoadDataURLsInWebUI(); + if (!ChildProcessSecurityPolicyImpl::GetInstance()-> + HasWebUIBindings(web_contents_->GetRenderProcessHost()->GetID()) || + !WebUIControllerFactoryRegistry::GetInstance()->IsURLAcceptableForWebUI( + web_contents_->GetBrowserContext(), source_url, data_urls_allowed)) { + NOTREACHED() << "Blocked unauthorized use of WebUIBindings."; + return; + } + + ProcessWebUIMessage(source_url, message, args); +} + +void WebUIImpl::RenderViewCreated(RenderViewHost* render_view_host) { + controller_->RenderViewCreated(render_view_host); + + // Do not attempt to set the toolkit property if WebUI is not enabled, e.g., + // the bookmarks manager page. + if (!(bindings_ & BINDINGS_POLICY_WEB_UI)) + return; + +#if defined(TOOLKIT_VIEWS) + render_view_host->SetWebUIProperty("toolkit", "views"); +#elif defined(TOOLKIT_GTK) + render_view_host->SetWebUIProperty("toolkit", "GTK"); +#endif // defined(TOOLKIT_VIEWS) +} + +WebContents* WebUIImpl::GetWebContents() const { + return web_contents_; +} + +ui::ScaleFactor WebUIImpl::GetDeviceScaleFactor() const { + return GetScaleFactorForView(web_contents_->GetRenderWidgetHostView()); +} + +const string16& WebUIImpl::GetOverriddenTitle() const { + return overridden_title_; +} + +void WebUIImpl::OverrideTitle(const string16& title) { + overridden_title_ = title; +} + +PageTransition WebUIImpl::GetLinkTransitionType() const { + return link_transition_type_; +} + +void WebUIImpl::SetLinkTransitionType(PageTransition type) { + link_transition_type_ = type; +} + +int WebUIImpl::GetBindings() const { + return bindings_; +} + +void WebUIImpl::SetBindings(int bindings) { + bindings_ = bindings; +} + +void WebUIImpl::SetFrameXPath(const std::string& xpath) { + frame_xpath_ = xpath; +} + +WebUIController* WebUIImpl::GetController() const { + return controller_.get(); +} + +void WebUIImpl::SetController(WebUIController* controller) { + controller_.reset(controller); +} + +void WebUIImpl::CallJavascriptFunction(const std::string& function_name) { + DCHECK(IsStringASCII(function_name)); + string16 javascript = ASCIIToUTF16(function_name + "();"); + ExecuteJavascript(javascript); +} + +void WebUIImpl::CallJavascriptFunction(const std::string& function_name, + const Value& arg) { + DCHECK(IsStringASCII(function_name)); + std::vector<const Value*> args; + args.push_back(&arg); + ExecuteJavascript(GetJavascriptCall(function_name, args)); +} + +void WebUIImpl::CallJavascriptFunction( + const std::string& function_name, + const Value& arg1, const Value& arg2) { + DCHECK(IsStringASCII(function_name)); + std::vector<const Value*> args; + args.push_back(&arg1); + args.push_back(&arg2); + ExecuteJavascript(GetJavascriptCall(function_name, args)); +} + +void WebUIImpl::CallJavascriptFunction( + const std::string& function_name, + const Value& arg1, const Value& arg2, const Value& arg3) { + DCHECK(IsStringASCII(function_name)); + std::vector<const Value*> args; + args.push_back(&arg1); + args.push_back(&arg2); + args.push_back(&arg3); + ExecuteJavascript(GetJavascriptCall(function_name, args)); +} + +void WebUIImpl::CallJavascriptFunction( + const std::string& function_name, + const Value& arg1, + const Value& arg2, + const Value& arg3, + const Value& arg4) { + DCHECK(IsStringASCII(function_name)); + std::vector<const Value*> args; + args.push_back(&arg1); + args.push_back(&arg2); + args.push_back(&arg3); + args.push_back(&arg4); + ExecuteJavascript(GetJavascriptCall(function_name, args)); +} + +void WebUIImpl::CallJavascriptFunction( + const std::string& function_name, + const std::vector<const Value*>& args) { + DCHECK(IsStringASCII(function_name)); + ExecuteJavascript(GetJavascriptCall(function_name, args)); +} + +void WebUIImpl::RegisterMessageCallback(const std::string &message, + const MessageCallback& callback) { + message_callbacks_.insert(std::make_pair(message, callback)); +} + +void WebUIImpl::ProcessWebUIMessage(const GURL& source_url, + const std::string& message, + const base::ListValue& args) { + if (controller_->OverrideHandleWebUIMessage(source_url, message, args)) + return; + + // Look up the callback for this message. + MessageCallbackMap::const_iterator callback = + message_callbacks_.find(message); + if (callback != message_callbacks_.end()) { + // Forward this message and content on. + callback->second.Run(&args); + } else { + NOTREACHED() << "Unhandled chrome.send(\"" << message << "\");"; + } +} + +// WebUIImpl, protected: ------------------------------------------------------- + +void WebUIImpl::AddMessageHandler(WebUIMessageHandler* handler) { + DCHECK(!handler->web_ui()); + handler->set_web_ui(this); + handler->RegisterMessages(); + handlers_.push_back(handler); +} + +void WebUIImpl::ExecuteJavascript(const string16& javascript) { + static_cast<RenderViewHostImpl*>( + web_contents_->GetRenderViewHost())->ExecuteJavascriptInWebFrame( + ASCIIToUTF16(frame_xpath_), javascript); +} + +} // namespace content |