2016-08-19 11:47:32 +00:00
|
|
|
// Copyright (c) 2016 GitHub, Inc.
|
|
|
|
// Use of this source code is governed by the MIT license that can be
|
|
|
|
// found in the LICENSE file.
|
|
|
|
|
|
|
|
#include "atom/renderer/atom_sandboxed_renderer_client.h"
|
|
|
|
|
2016-08-21 10:30:21 +00:00
|
|
|
#include "atom/common/api/api_messages.h"
|
2017-03-21 13:41:23 +00:00
|
|
|
#include "atom/common/api/atom_bindings.h"
|
2018-10-24 10:49:10 +00:00
|
|
|
#include "atom/common/application_info.h"
|
2016-08-21 10:30:21 +00:00
|
|
|
#include "atom/common/native_mate_converters/string16_converter.h"
|
|
|
|
#include "atom/common/native_mate_converters/value_converter.h"
|
2018-01-06 15:58:24 +00:00
|
|
|
#include "atom/common/node_bindings.h"
|
2016-08-21 10:30:21 +00:00
|
|
|
#include "atom/common/options_switches.h"
|
|
|
|
#include "atom/renderer/api/atom_api_renderer_ipc.h"
|
2018-03-09 09:31:09 +00:00
|
|
|
#include "atom/renderer/atom_render_frame_observer.h"
|
2018-07-30 01:13:42 +00:00
|
|
|
#include "base/base_paths.h"
|
2016-08-21 10:30:21 +00:00
|
|
|
#include "base/command_line.h"
|
2018-03-07 16:40:00 +00:00
|
|
|
#include "base/files/file_path.h"
|
2018-07-30 01:13:42 +00:00
|
|
|
#include "base/path_service.h"
|
2018-08-21 18:05:45 +00:00
|
|
|
#include "base/process/process_handle.h"
|
2016-08-21 10:30:21 +00:00
|
|
|
#include "content/public/renderer/render_frame.h"
|
2018-11-29 01:55:03 +00:00
|
|
|
#include "gin/converter.h"
|
2017-02-20 13:59:39 +00:00
|
|
|
#include "native_mate/dictionary.h"
|
2018-07-20 16:08:18 +00:00
|
|
|
#include "third_party/blink/public/web/blink.h"
|
2018-09-19 11:10:26 +00:00
|
|
|
#include "third_party/blink/public/web/web_document.h"
|
2016-08-21 10:30:21 +00:00
|
|
|
|
2017-04-27 07:56:47 +00:00
|
|
|
#include "atom/common/node_includes.h"
|
2019-01-12 01:00:43 +00:00
|
|
|
#include "third_party/electron_node/src/node_binding.h"
|
|
|
|
#include "third_party/electron_node/src/node_native_module.h"
|
2017-04-27 07:56:47 +00:00
|
|
|
|
2016-08-19 11:47:32 +00:00
|
|
|
namespace atom {
|
|
|
|
|
2016-08-21 10:30:21 +00:00
|
|
|
namespace {
|
|
|
|
|
2018-09-13 00:25:56 +00:00
|
|
|
const char kIpcKey[] = "ipcNative";
|
|
|
|
const char kModuleCacheKey[] = "native-module-cache";
|
2017-02-20 13:59:39 +00:00
|
|
|
|
2018-03-22 07:15:57 +00:00
|
|
|
bool IsDevTools(content::RenderFrame* render_frame) {
|
2018-04-18 01:55:30 +00:00
|
|
|
return render_frame->GetWebFrame()->GetDocument().Url().ProtocolIs(
|
|
|
|
"chrome-devtools");
|
2018-03-22 07:15:57 +00:00
|
|
|
}
|
|
|
|
|
2018-11-21 17:56:58 +00:00
|
|
|
bool IsDevToolsExtension(content::RenderFrame* render_frame) {
|
|
|
|
return render_frame->GetWebFrame()->GetDocument().Url().ProtocolIs(
|
|
|
|
"chrome-extension");
|
|
|
|
}
|
|
|
|
|
2017-02-20 13:59:39 +00:00
|
|
|
v8::Local<v8::Object> GetModuleCache(v8::Isolate* isolate) {
|
|
|
|
mate::Dictionary global(isolate, isolate->GetCurrentContext()->Global());
|
|
|
|
v8::Local<v8::Value> cache;
|
|
|
|
|
|
|
|
if (!global.GetHidden(kModuleCacheKey, &cache)) {
|
|
|
|
cache = v8::Object::New(isolate);
|
|
|
|
global.SetHidden(kModuleCacheKey, cache);
|
|
|
|
}
|
|
|
|
|
2018-11-29 01:55:03 +00:00
|
|
|
return cache->ToObject(isolate);
|
2017-02-20 13:59:39 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// adapted from node.cc
|
2018-04-18 01:55:30 +00:00
|
|
|
v8::Local<v8::Value> GetBinding(v8::Isolate* isolate,
|
|
|
|
v8::Local<v8::String> key,
|
|
|
|
mate::Arguments* margs) {
|
2017-02-20 13:59:39 +00:00
|
|
|
v8::Local<v8::Object> exports;
|
2018-11-29 01:55:03 +00:00
|
|
|
std::string module_key = gin::V8ToString(isolate, key);
|
2017-02-20 13:59:39 +00:00
|
|
|
mate::Dictionary cache(isolate, GetModuleCache(isolate));
|
|
|
|
|
|
|
|
if (cache.Get(module_key.c_str(), &exports)) {
|
|
|
|
return exports;
|
|
|
|
}
|
|
|
|
|
2019-01-12 01:00:43 +00:00
|
|
|
auto* mod = node::binding::get_builtin_module(module_key.c_str());
|
2017-02-20 13:59:39 +00:00
|
|
|
|
|
|
|
if (!mod) {
|
|
|
|
char errmsg[1024];
|
|
|
|
snprintf(errmsg, sizeof(errmsg), "No such module: %s", module_key.c_str());
|
|
|
|
margs->ThrowError(errmsg);
|
|
|
|
return exports;
|
|
|
|
}
|
|
|
|
|
|
|
|
exports = v8::Object::New(isolate);
|
|
|
|
DCHECK_EQ(mod->nm_register_func, nullptr);
|
|
|
|
DCHECK_NE(mod->nm_context_register_func, nullptr);
|
|
|
|
mod->nm_context_register_func(exports, v8::Null(isolate),
|
2018-04-18 01:55:30 +00:00
|
|
|
isolate->GetCurrentContext(), mod->nm_priv);
|
2017-02-20 13:59:39 +00:00
|
|
|
cache.Set(module_key.c_str(), exports);
|
|
|
|
return exports;
|
|
|
|
}
|
|
|
|
|
2018-11-29 01:55:03 +00:00
|
|
|
v8::Local<v8::Value> CreatePreloadScript(v8::Isolate* isolate,
|
|
|
|
v8::Local<v8::String> preloadSrc) {
|
|
|
|
return RendererClientBase::RunScript(isolate->GetCurrentContext(),
|
|
|
|
preloadSrc);
|
|
|
|
}
|
|
|
|
|
2018-03-09 09:31:09 +00:00
|
|
|
class AtomSandboxedRenderFrameObserver : public AtomRenderFrameObserver {
|
2016-08-21 10:30:21 +00:00
|
|
|
public:
|
2018-03-09 09:31:09 +00:00
|
|
|
AtomSandboxedRenderFrameObserver(content::RenderFrame* render_frame,
|
|
|
|
AtomSandboxedRendererClient* renderer_client)
|
|
|
|
: AtomRenderFrameObserver(render_frame, renderer_client),
|
2018-06-13 07:38:31 +00:00
|
|
|
renderer_client_(renderer_client) {}
|
2016-08-21 10:30:21 +00:00
|
|
|
|
|
|
|
protected:
|
2017-08-21 23:09:28 +00:00
|
|
|
void EmitIPCEvent(blink::WebLocalFrame* frame,
|
2018-10-06 11:48:00 +00:00
|
|
|
bool internal,
|
2018-08-24 15:30:37 +00:00
|
|
|
const std::string& channel,
|
2018-09-01 01:13:51 +00:00
|
|
|
const base::ListValue& args,
|
|
|
|
int32_t sender_id) override {
|
2017-08-21 23:09:28 +00:00
|
|
|
if (!frame)
|
2016-08-21 10:30:21 +00:00
|
|
|
return;
|
|
|
|
|
2018-04-17 22:41:47 +00:00
|
|
|
auto* isolate = blink::MainThreadIsolate();
|
2016-08-21 10:30:21 +00:00
|
|
|
v8::HandleScope handle_scope(isolate);
|
2017-06-16 20:42:33 +00:00
|
|
|
auto context = frame->MainWorldScriptContext();
|
2016-08-21 10:30:21 +00:00
|
|
|
v8::Context::Scope context_scope(context);
|
2018-04-18 01:55:30 +00:00
|
|
|
v8::Local<v8::Value> argv[] = {mate::ConvertToV8(isolate, channel),
|
2018-09-01 01:13:51 +00:00
|
|
|
mate::ConvertToV8(isolate, args),
|
|
|
|
mate::ConvertToV8(isolate, sender_id)};
|
2017-02-20 13:59:39 +00:00
|
|
|
renderer_client_->InvokeIpcCallback(
|
2018-10-06 11:48:00 +00:00
|
|
|
context, internal ? "onInternalMessage" : "onMessage",
|
2018-06-13 07:38:31 +00:00
|
|
|
std::vector<v8::Local<v8::Value>>(argv, argv + node::arraysize(argv)));
|
2016-08-21 10:30:21 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
private:
|
|
|
|
AtomSandboxedRendererClient* renderer_client_;
|
2018-03-09 09:31:09 +00:00
|
|
|
DISALLOW_COPY_AND_ASSIGN(AtomSandboxedRenderFrameObserver);
|
2016-08-21 10:30:21 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
} // namespace
|
|
|
|
|
2016-08-19 11:47:32 +00:00
|
|
|
AtomSandboxedRendererClient::AtomSandboxedRendererClient() {
|
2018-01-06 15:58:24 +00:00
|
|
|
// Explicitly register electron's builtin modules.
|
|
|
|
NodeBindings::RegisterBuiltinModules();
|
2018-08-21 18:05:45 +00:00
|
|
|
metrics_ = base::ProcessMetrics::CreateCurrentProcessMetrics();
|
2016-08-19 11:47:32 +00:00
|
|
|
}
|
|
|
|
|
2018-04-18 01:55:30 +00:00
|
|
|
AtomSandboxedRendererClient::~AtomSandboxedRendererClient() {}
|
2016-08-19 11:47:32 +00:00
|
|
|
|
2018-08-21 18:05:45 +00:00
|
|
|
void AtomSandboxedRendererClient::InitializeBindings(
|
|
|
|
v8::Local<v8::Object> binding,
|
|
|
|
v8::Local<v8::Context> context) {
|
|
|
|
auto* isolate = context->GetIsolate();
|
|
|
|
mate::Dictionary b(isolate, binding);
|
|
|
|
b.SetMethod("get", GetBinding);
|
2018-11-29 01:55:03 +00:00
|
|
|
b.SetMethod("createPreloadScript", CreatePreloadScript);
|
2018-09-30 21:24:00 +00:00
|
|
|
|
|
|
|
mate::Dictionary process = mate::Dictionary::CreateEmpty(isolate);
|
|
|
|
b.Set("process", process);
|
|
|
|
|
2018-11-28 16:08:17 +00:00
|
|
|
AtomBindings::BindProcess(isolate, &process, metrics_.get());
|
2018-09-30 21:24:00 +00:00
|
|
|
|
|
|
|
process.Set("argv", base::CommandLine::ForCurrentProcess()->argv());
|
2018-11-08 14:28:06 +00:00
|
|
|
process.SetReadOnly("pid", base::GetCurrentProcId());
|
|
|
|
process.SetReadOnly("sandboxed", true);
|
|
|
|
process.SetReadOnly("type", "renderer");
|
2018-09-30 21:24:00 +00:00
|
|
|
|
2018-08-21 18:05:45 +00:00
|
|
|
// Pass in CLI flags needed to setup the renderer
|
|
|
|
base::CommandLine* command_line = base::CommandLine::ForCurrentProcess();
|
|
|
|
if (command_line->HasSwitch(switches::kGuestInstanceID))
|
|
|
|
b.Set(options::kGuestInstanceID,
|
|
|
|
command_line->GetSwitchValueASCII(switches::kGuestInstanceID));
|
|
|
|
}
|
|
|
|
|
2016-08-21 10:30:21 +00:00
|
|
|
void AtomSandboxedRendererClient::RenderFrameCreated(
|
|
|
|
content::RenderFrame* render_frame) {
|
2018-03-09 09:31:09 +00:00
|
|
|
new AtomSandboxedRenderFrameObserver(render_frame, this);
|
2017-03-27 21:31:18 +00:00
|
|
|
RendererClientBase::RenderFrameCreated(render_frame);
|
2016-08-21 10:30:21 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
void AtomSandboxedRendererClient::RenderViewCreated(
|
|
|
|
content::RenderView* render_view) {
|
2017-03-27 21:31:18 +00:00
|
|
|
RendererClientBase::RenderViewCreated(render_view);
|
2016-08-21 10:30:21 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
void AtomSandboxedRendererClient::DidCreateScriptContext(
|
2018-04-18 01:55:30 +00:00
|
|
|
v8::Handle<v8::Context> context,
|
|
|
|
content::RenderFrame* render_frame) {
|
2018-07-19 04:29:47 +00:00
|
|
|
RendererClientBase::DidCreateScriptContext(context, render_frame);
|
|
|
|
|
2018-03-22 07:15:57 +00:00
|
|
|
// Only allow preload for the main frame or
|
|
|
|
// For devtools we still want to run the preload_bundle script
|
2018-11-21 17:56:58 +00:00
|
|
|
if (!render_frame->IsMainFrame() && !IsDevTools(render_frame) &&
|
|
|
|
!IsDevToolsExtension(render_frame))
|
2017-11-23 06:25:16 +00:00
|
|
|
return;
|
|
|
|
|
2019-01-12 01:00:43 +00:00
|
|
|
// Wrap the bundle into a function that receives the binding object as
|
|
|
|
// argument.
|
2018-04-17 22:41:47 +00:00
|
|
|
auto* isolate = context->GetIsolate();
|
2016-08-21 10:30:21 +00:00
|
|
|
auto binding = v8::Object::New(isolate);
|
2017-02-20 13:59:39 +00:00
|
|
|
InitializeBindings(binding, context);
|
2017-03-27 21:31:18 +00:00
|
|
|
AddRenderBindings(isolate, binding);
|
2019-01-12 01:00:43 +00:00
|
|
|
|
|
|
|
std::vector<v8::Local<v8::String>> preload_bundle_params = {
|
|
|
|
node::FIXED_ONE_BYTE_STRING(isolate, "binding")};
|
|
|
|
|
|
|
|
std::vector<v8::Local<v8::Value>> preload_bundle_args = {binding};
|
|
|
|
|
|
|
|
node::per_process::native_module_loader.CompileAndCall(
|
|
|
|
isolate->GetCurrentContext(), "electron/js2c/preload_bundle",
|
|
|
|
&preload_bundle_params, &preload_bundle_args, nullptr);
|
2016-08-21 10:30:21 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
void AtomSandboxedRendererClient::WillReleaseScriptContext(
|
2018-04-18 01:55:30 +00:00
|
|
|
v8::Handle<v8::Context> context,
|
|
|
|
content::RenderFrame* render_frame) {
|
2017-11-23 06:25:16 +00:00
|
|
|
// Only allow preload for the main frame
|
|
|
|
if (!render_frame->IsMainFrame())
|
|
|
|
return;
|
|
|
|
|
2018-04-17 22:41:47 +00:00
|
|
|
auto* isolate = context->GetIsolate();
|
2016-08-21 10:30:21 +00:00
|
|
|
v8::HandleScope handle_scope(isolate);
|
|
|
|
v8::Context::Scope context_scope(context);
|
2017-02-20 13:59:39 +00:00
|
|
|
InvokeIpcCallback(context, "onExit", std::vector<v8::Local<v8::Value>>());
|
2016-08-21 10:30:21 +00:00
|
|
|
}
|
|
|
|
|
2017-02-20 13:59:39 +00:00
|
|
|
void AtomSandboxedRendererClient::InvokeIpcCallback(
|
2016-08-21 10:30:21 +00:00
|
|
|
v8::Handle<v8::Context> context,
|
2017-03-29 19:29:52 +00:00
|
|
|
const std::string& callback_name,
|
2016-08-21 10:30:21 +00:00
|
|
|
std::vector<v8::Handle<v8::Value>> args) {
|
2018-04-17 22:41:47 +00:00
|
|
|
auto* isolate = context->GetIsolate();
|
2018-11-29 01:55:03 +00:00
|
|
|
auto binding_key = mate::ConvertToV8(isolate, kIpcKey)->ToString(isolate);
|
2016-08-21 10:30:21 +00:00
|
|
|
auto private_binding_key = v8::Private::ForApi(isolate, binding_key);
|
|
|
|
auto global_object = context->Global();
|
|
|
|
v8::Local<v8::Value> value;
|
|
|
|
if (!global_object->GetPrivate(context, private_binding_key).ToLocal(&value))
|
|
|
|
return;
|
|
|
|
if (value.IsEmpty() || !value->IsObject())
|
|
|
|
return;
|
2018-11-29 01:55:03 +00:00
|
|
|
auto binding = value->ToObject(isolate);
|
|
|
|
auto callback_key =
|
|
|
|
mate::ConvertToV8(isolate, callback_name)->ToString(isolate);
|
2016-08-21 10:30:21 +00:00
|
|
|
auto callback_value = binding->Get(callback_key);
|
|
|
|
DCHECK(callback_value->IsFunction()); // set by sandboxed_renderer/init.js
|
|
|
|
auto callback = v8::Handle<v8::Function>::Cast(callback_value);
|
2017-08-16 07:49:22 +00:00
|
|
|
ignore_result(callback->Call(context, binding, args.size(), args.data()));
|
2016-08-21 10:30:21 +00:00
|
|
|
}
|
|
|
|
|
2016-08-19 11:47:32 +00:00
|
|
|
} // namespace atom
|