2013-04-12 01:46:58 +00:00
|
|
|
// Copyright (c) 2013 GitHub, Inc. All rights reserved.
|
2014-04-25 09:49:37 +00:00
|
|
|
// Use of this source code is governed by the MIT license that can be
|
2013-04-12 01:46:58 +00:00
|
|
|
// found in the LICENSE file.
|
|
|
|
|
2014-03-16 00:30:26 +00:00
|
|
|
#include "atom/renderer/atom_renderer_client.h"
|
2013-04-12 01:46:58 +00:00
|
|
|
|
2014-01-09 14:13:06 +00:00
|
|
|
#include <algorithm>
|
2014-03-16 01:13:06 +00:00
|
|
|
#include <string>
|
2014-01-09 14:13:06 +00:00
|
|
|
|
2014-03-16 00:30:26 +00:00
|
|
|
#include "atom/common/node_bindings.h"
|
|
|
|
#include "atom/common/options_switches.h"
|
|
|
|
#include "atom/renderer/api/atom_renderer_bindings.h"
|
|
|
|
#include "atom/renderer/atom_render_view_observer.h"
|
2014-08-21 08:14:22 +00:00
|
|
|
#include "chrome/renderer/printing/print_web_view_helper.h"
|
2014-07-18 01:04:41 +00:00
|
|
|
#include "content/public/renderer/render_frame.h"
|
|
|
|
#include "content/public/renderer/render_frame_observer.h"
|
2014-09-01 10:43:43 +00:00
|
|
|
#include "content/public/renderer/render_thread.h"
|
2014-03-16 01:37:04 +00:00
|
|
|
#include "base/command_line.h"
|
2014-06-28 14:33:00 +00:00
|
|
|
#include "native_mate/converter.h"
|
2014-02-21 14:42:33 +00:00
|
|
|
#include "third_party/WebKit/public/web/WebDocument.h"
|
2014-01-30 14:47:21 +00:00
|
|
|
#include "third_party/WebKit/public/web/WebFrame.h"
|
2014-09-01 10:43:43 +00:00
|
|
|
#include "third_party/WebKit/public/web/WebKit.h"
|
2014-09-09 03:08:30 +00:00
|
|
|
#include "third_party/WebKit/public/web/WebRuntimeFeatures.h"
|
2013-12-11 07:48:19 +00:00
|
|
|
|
2014-04-17 05:45:14 +00:00
|
|
|
#include "atom/common/node_includes.h"
|
2013-04-20 03:13:06 +00:00
|
|
|
|
2013-04-12 01:46:58 +00:00
|
|
|
namespace atom {
|
|
|
|
|
2014-01-31 02:41:20 +00:00
|
|
|
namespace {
|
|
|
|
|
2014-02-21 14:42:33 +00:00
|
|
|
// Security tokens.
|
2014-03-04 12:12:10 +00:00
|
|
|
const char* kSecurityAll = "all";
|
|
|
|
const char* kSecurityExceptIframe = "except-iframe";
|
|
|
|
const char* kSecurityManualEnableIframe = "manual-enable-iframe";
|
|
|
|
const char* kSecurityDisable = "disable";
|
|
|
|
const char* kSecurityEnableNodeIntegration = "enable-node-integration";
|
2014-01-31 02:41:20 +00:00
|
|
|
|
2014-09-09 03:08:30 +00:00
|
|
|
bool IsSwitchEnabled(base::CommandLine* command_line,
|
|
|
|
const char* switch_string,
|
|
|
|
bool* enabled) {
|
|
|
|
std::string value = command_line->GetSwitchValueASCII(switch_string);
|
|
|
|
if (value == "true")
|
|
|
|
*enabled = true;
|
|
|
|
else if (value == "false")
|
|
|
|
*enabled = false;
|
|
|
|
else
|
|
|
|
return false;
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2014-07-18 01:04:41 +00:00
|
|
|
// Helper class to forward the WillReleaseScriptContext message to the client.
|
|
|
|
class AtomRenderFrameObserver : public content::RenderFrameObserver {
|
|
|
|
public:
|
|
|
|
AtomRenderFrameObserver(content::RenderFrame* frame,
|
|
|
|
AtomRendererClient* renderer_client)
|
|
|
|
: content::RenderFrameObserver(frame),
|
|
|
|
renderer_client_(renderer_client) {}
|
|
|
|
|
|
|
|
// content::RenderFrameObserver:
|
|
|
|
virtual void WillReleaseScriptContext(v8::Handle<v8::Context> context,
|
|
|
|
int world_id) OVERRIDE {
|
|
|
|
renderer_client_->WillReleaseScriptContext(
|
|
|
|
render_frame()->GetWebFrame(), context, world_id);
|
|
|
|
}
|
|
|
|
|
|
|
|
private:
|
|
|
|
AtomRendererClient* renderer_client_;
|
|
|
|
|
|
|
|
DISALLOW_COPY_AND_ASSIGN(AtomRenderFrameObserver);
|
|
|
|
};
|
|
|
|
|
2014-01-31 02:41:20 +00:00
|
|
|
} // namespace
|
|
|
|
|
2013-04-20 03:13:06 +00:00
|
|
|
AtomRendererClient::AtomRendererClient()
|
2014-03-04 12:12:10 +00:00
|
|
|
: node_integration_(EXCEPT_IFRAME),
|
2014-01-30 15:20:12 +00:00
|
|
|
main_frame_(NULL) {
|
|
|
|
// Translate the token.
|
|
|
|
std::string token = CommandLine::ForCurrentProcess()->
|
|
|
|
GetSwitchValueASCII(switches::kNodeIntegration);
|
2014-03-04 12:12:10 +00:00
|
|
|
if (token == kSecurityExceptIframe)
|
2014-01-30 15:20:12 +00:00
|
|
|
node_integration_ = EXCEPT_IFRAME;
|
2014-03-04 12:12:10 +00:00
|
|
|
else if (token == kSecurityManualEnableIframe)
|
2014-01-30 15:20:12 +00:00
|
|
|
node_integration_ = MANUAL_ENABLE_IFRAME;
|
2014-03-04 12:12:10 +00:00
|
|
|
else if (token == kSecurityDisable)
|
2014-01-30 15:20:12 +00:00
|
|
|
node_integration_ = DISABLE;
|
2014-03-04 12:12:10 +00:00
|
|
|
else if (token == kSecurityAll)
|
|
|
|
node_integration_ = ALL;
|
2014-01-30 14:47:21 +00:00
|
|
|
|
|
|
|
if (IsNodeBindingEnabled()) {
|
|
|
|
node_bindings_.reset(NodeBindings::Create(false));
|
|
|
|
atom_bindings_.reset(new AtomRendererBindings);
|
|
|
|
}
|
2013-04-12 01:46:58 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
AtomRendererClient::~AtomRendererClient() {
|
|
|
|
}
|
|
|
|
|
2014-09-01 10:43:43 +00:00
|
|
|
void AtomRendererClient::WebKitInitialized() {
|
2014-09-09 03:08:30 +00:00
|
|
|
EnableWebRuntimeFeatures();
|
|
|
|
|
2014-01-30 14:47:21 +00:00
|
|
|
if (!IsNodeBindingEnabled())
|
|
|
|
return;
|
|
|
|
|
2013-04-20 03:13:06 +00:00
|
|
|
node_bindings_->Initialize();
|
2013-12-15 06:20:28 +00:00
|
|
|
node_bindings_->PrepareMessageLoop();
|
2013-04-20 03:13:06 +00:00
|
|
|
|
2013-12-15 06:20:28 +00:00
|
|
|
DCHECK(!global_env);
|
2013-04-20 03:13:06 +00:00
|
|
|
|
2013-12-15 06:20:28 +00:00
|
|
|
// Create a default empty environment which would be used when we need to
|
|
|
|
// run V8 code out of a window context (like running a uv callback).
|
2014-09-01 10:43:43 +00:00
|
|
|
v8::Isolate* isolate = blink::mainThreadIsolate();
|
2014-06-28 14:33:00 +00:00
|
|
|
v8::HandleScope handle_scope(isolate);
|
|
|
|
v8::Local<v8::Context> context = v8::Context::New(isolate);
|
2013-12-15 06:20:28 +00:00
|
|
|
global_env = node::Environment::New(context);
|
2013-04-20 03:13:06 +00:00
|
|
|
}
|
|
|
|
|
2014-09-01 10:43:43 +00:00
|
|
|
void AtomRendererClient::RenderThreadStarted() {
|
|
|
|
content::RenderThread::Get()->AddObserver(this);
|
|
|
|
}
|
|
|
|
|
2014-06-28 14:33:00 +00:00
|
|
|
void AtomRendererClient::RenderFrameCreated(
|
|
|
|
content::RenderFrame* render_frame) {
|
2014-07-18 01:04:41 +00:00
|
|
|
new AtomRenderFrameObserver(render_frame, this);
|
2014-06-28 14:33:00 +00:00
|
|
|
}
|
|
|
|
|
2013-04-12 01:46:58 +00:00
|
|
|
void AtomRendererClient::RenderViewCreated(content::RenderView* render_view) {
|
2014-08-21 08:14:22 +00:00
|
|
|
new printing::PrintWebViewHelper(render_view);
|
2013-04-20 03:13:06 +00:00
|
|
|
new AtomRenderViewObserver(render_view, this);
|
2013-04-12 01:46:58 +00:00
|
|
|
}
|
|
|
|
|
2014-10-07 13:18:44 +00:00
|
|
|
blink::WebSpeechSynthesizer* AtomRendererClient::OverrideSpeechSynthesizer(
|
|
|
|
blink::WebSpeechSynthesizerClient* client) {
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
2014-06-28 14:33:00 +00:00
|
|
|
void AtomRendererClient::DidCreateScriptContext(blink::WebFrame* frame,
|
2013-12-23 14:08:45 +00:00
|
|
|
v8::Handle<v8::Context> context,
|
|
|
|
int extension_group,
|
|
|
|
int world_id) {
|
2014-01-30 15:20:12 +00:00
|
|
|
// The first web frame is the main frame.
|
|
|
|
if (main_frame_ == NULL)
|
|
|
|
main_frame_ = frame;
|
|
|
|
|
2014-01-30 14:47:21 +00:00
|
|
|
if (!IsNodeBindingEnabled(frame))
|
|
|
|
return;
|
|
|
|
|
2013-12-23 14:08:45 +00:00
|
|
|
v8::Context::Scope scope(context);
|
|
|
|
|
|
|
|
// Check the existance of process object to prevent duplicate initialization.
|
2014-06-28 14:33:00 +00:00
|
|
|
if (context->Global()->Has(
|
|
|
|
mate::StringToV8(context->GetIsolate(), "process")))
|
2013-12-23 14:08:45 +00:00
|
|
|
return;
|
|
|
|
|
|
|
|
// Give the node loop a run to make sure everything is ready.
|
|
|
|
node_bindings_->RunMessageLoop();
|
|
|
|
|
|
|
|
// Setup node environment for each window.
|
2014-01-09 13:35:29 +00:00
|
|
|
node::Environment* env = node_bindings_->CreateEnvironment(context);
|
2013-12-23 14:08:45 +00:00
|
|
|
|
|
|
|
// Add atom-shell extended APIs.
|
|
|
|
atom_bindings_->BindToFrame(frame);
|
2014-01-09 13:35:29 +00:00
|
|
|
|
2014-01-09 14:13:06 +00:00
|
|
|
// Store the created environment.
|
|
|
|
web_page_envs_.push_back(env);
|
|
|
|
|
2014-01-09 13:35:29 +00:00
|
|
|
// Make uv loop being wrapped by window context.
|
2014-01-10 08:29:38 +00:00
|
|
|
if (node_bindings_->uv_env() == NULL)
|
2014-01-09 14:13:06 +00:00
|
|
|
node_bindings_->set_uv_env(env);
|
2013-12-23 14:08:45 +00:00
|
|
|
}
|
|
|
|
|
2013-12-23 14:19:51 +00:00
|
|
|
void AtomRendererClient::WillReleaseScriptContext(
|
2014-06-28 14:33:00 +00:00
|
|
|
blink::WebFrame* frame,
|
2013-12-23 14:19:51 +00:00
|
|
|
v8::Handle<v8::Context> context,
|
|
|
|
int world_id) {
|
2014-01-30 14:47:21 +00:00
|
|
|
if (!IsNodeBindingEnabled(frame))
|
|
|
|
return;
|
|
|
|
|
2013-12-23 14:19:51 +00:00
|
|
|
node::Environment* env = node::Environment::GetCurrent(context);
|
|
|
|
if (env == NULL) {
|
|
|
|
LOG(ERROR) << "Encounter a non-node context when releasing script context";
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2014-01-09 14:13:06 +00:00
|
|
|
// Clear the environment.
|
|
|
|
web_page_envs_.erase(
|
|
|
|
std::remove(web_page_envs_.begin(), web_page_envs_.end(), env),
|
|
|
|
web_page_envs_.end());
|
|
|
|
|
2014-01-10 03:04:54 +00:00
|
|
|
// Notice that we are not disposing the environment object here, because there
|
|
|
|
// may still be pending uv operations in the uv loop, and when they got done
|
|
|
|
// they would be needing the original environment.
|
|
|
|
// So we are leaking the environment object here, just like Chrome leaking the
|
|
|
|
// memory :) . Since it's only leaked when refreshing or unloading, so as long
|
|
|
|
// as we make sure renderer process is restared then the memory would not be
|
|
|
|
// leaked.
|
|
|
|
// env->Dispose();
|
|
|
|
|
2014-01-09 14:13:06 +00:00
|
|
|
// Wrap the uv loop with another environment.
|
2014-01-10 08:29:38 +00:00
|
|
|
if (env == node_bindings_->uv_env()) {
|
2014-01-09 14:13:06 +00:00
|
|
|
node::Environment* env = web_page_envs_.size() > 0 ? web_page_envs_[0] :
|
|
|
|
NULL;
|
|
|
|
node_bindings_->set_uv_env(env);
|
|
|
|
}
|
2013-12-23 14:08:45 +00:00
|
|
|
}
|
|
|
|
|
2014-06-28 14:33:00 +00:00
|
|
|
bool AtomRendererClient::ShouldFork(blink::WebFrame* frame,
|
2014-01-14 08:03:01 +00:00
|
|
|
const GURL& url,
|
|
|
|
const std::string& http_method,
|
|
|
|
bool is_initial_navigation,
|
|
|
|
bool is_server_redirect,
|
|
|
|
bool* send_referrer) {
|
|
|
|
// Handle all the navigations and reloads in browser.
|
2014-05-13 06:45:30 +00:00
|
|
|
// FIXME We only support GET here because http method will be ignored when
|
|
|
|
// the OpenURLFromTab is triggered, which means form posting would not work,
|
|
|
|
// we should solve this by patching Chromium in future.
|
|
|
|
return http_method == "GET";
|
2014-01-14 08:03:01 +00:00
|
|
|
}
|
|
|
|
|
2014-06-28 14:33:00 +00:00
|
|
|
bool AtomRendererClient::IsNodeBindingEnabled(blink::WebFrame* frame) {
|
2014-01-30 15:20:12 +00:00
|
|
|
if (node_integration_ == DISABLE)
|
2014-01-30 14:47:21 +00:00
|
|
|
return false;
|
2014-01-30 15:20:12 +00:00
|
|
|
// Node integration is enabled in main frame unless explictly disabled.
|
|
|
|
else if (frame == main_frame_)
|
|
|
|
return true;
|
2014-08-28 04:24:52 +00:00
|
|
|
// Enable node integration in chrome extensions.
|
|
|
|
else if (frame != NULL &&
|
|
|
|
GURL(frame->document().url()).SchemeIs("chrome-extension"))
|
|
|
|
return true;
|
2014-01-30 15:20:12 +00:00
|
|
|
else if (node_integration_ == MANUAL_ENABLE_IFRAME &&
|
2014-01-30 14:47:21 +00:00
|
|
|
frame != NULL &&
|
2014-03-04 12:12:10 +00:00
|
|
|
frame->uniqueName().utf8().find(kSecurityEnableNodeIntegration)
|
2014-01-31 02:41:20 +00:00
|
|
|
== std::string::npos)
|
2014-01-30 14:47:21 +00:00
|
|
|
return false;
|
2014-01-30 15:20:12 +00:00
|
|
|
else if (node_integration_ == EXCEPT_IFRAME && frame != NULL)
|
|
|
|
return false;
|
2014-01-30 14:47:21 +00:00
|
|
|
else
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2014-09-09 03:08:30 +00:00
|
|
|
void AtomRendererClient::EnableWebRuntimeFeatures() {
|
|
|
|
base::CommandLine* command_line = base::CommandLine::ForCurrentProcess();
|
|
|
|
bool b;
|
|
|
|
if (IsSwitchEnabled(command_line, switches::kExperimentalFeatures, &b))
|
|
|
|
blink::WebRuntimeFeatures::enableExperimentalFeatures(b);
|
|
|
|
if (IsSwitchEnabled(command_line, switches::kExperimentalCanvasFeatures, &b))
|
|
|
|
blink::WebRuntimeFeatures::enableExperimentalCanvasFeatures(b);
|
|
|
|
if (IsSwitchEnabled(command_line, switches::kSubpixelFontScaling, &b))
|
|
|
|
blink::WebRuntimeFeatures::enableSubpixelFontScaling(b);
|
|
|
|
if (IsSwitchEnabled(command_line, switches::kOverlayScrollbars, &b))
|
|
|
|
blink::WebRuntimeFeatures::enableOverlayScrollbars(b);
|
|
|
|
if (IsSwitchEnabled(command_line, switches::kOverlayFullscreenVideo, &b))
|
|
|
|
blink::WebRuntimeFeatures::enableOverlayFullscreenVideo(b);
|
|
|
|
if (IsSwitchEnabled(command_line, switches::kSharedWorker, &b))
|
|
|
|
blink::WebRuntimeFeatures::enableSharedWorker(b);
|
|
|
|
}
|
|
|
|
|
2013-04-12 01:46:58 +00:00
|
|
|
} // namespace atom
|