2014-10-31 18:17:05 +00:00
|
|
|
// Copyright (c) 2013 GitHub, Inc.
|
2014-04-25 09:49:37 +00:00
|
|
|
// Use of this source code is governed by the MIT license that can be
|
2013-04-13 10:39:09 +00:00
|
|
|
// found in the LICENSE file.
|
|
|
|
|
2014-03-16 00:30:26 +00:00
|
|
|
#include "atom/common/node_bindings.h"
|
2013-04-13 10:39:09 +00:00
|
|
|
|
2014-03-16 01:13:06 +00:00
|
|
|
#include <string>
|
|
|
|
#include <vector>
|
|
|
|
|
2015-03-17 12:55:53 +00:00
|
|
|
#include "atom/app/atom_main_args.h"
|
2015-02-13 05:32:58 +00:00
|
|
|
#include "atom/common/native_mate_converters/file_path_converter.h"
|
2013-04-13 10:39:09 +00:00
|
|
|
#include "base/command_line.h"
|
2013-12-11 15:59:03 +00:00
|
|
|
#include "base/base_paths.h"
|
2014-04-21 15:49:53 +00:00
|
|
|
#include "base/files/file_path.h"
|
|
|
|
#include "base/message_loop/message_loop.h"
|
2013-12-11 15:59:03 +00:00
|
|
|
#include "base/path_service.h"
|
2013-07-22 06:58:25 +00:00
|
|
|
#include "content/public/browser/browser_thread.h"
|
2014-08-10 11:14:20 +00:00
|
|
|
#include "native_mate/locker.h"
|
2015-02-13 05:32:58 +00:00
|
|
|
#include "native_mate/dictionary.h"
|
2013-04-13 10:39:09 +00:00
|
|
|
|
2014-04-17 05:45:14 +00:00
|
|
|
#include "atom/common/node_includes.h"
|
2013-12-11 07:48:19 +00:00
|
|
|
|
2013-07-22 06:58:25 +00:00
|
|
|
using content::BrowserThread;
|
|
|
|
|
2014-06-29 10:12:46 +00:00
|
|
|
// Force all builtin modules to be referenced so they can actually run their
|
|
|
|
// DSO constructors, see http://git.io/DRIqCg.
|
|
|
|
#define REFERENCE_MODULE(name) \
|
2014-06-30 06:16:16 +00:00
|
|
|
extern "C" void _register_ ## name(void); \
|
2014-06-29 10:12:46 +00:00
|
|
|
void (*fp_register_ ## name)(void) = _register_ ## name
|
2015-04-06 06:21:21 +00:00
|
|
|
#if defined(OS_WIN)
|
|
|
|
// On Windows Node is built as static_library so we need to reference Node's
|
|
|
|
// builtin modules.
|
|
|
|
REFERENCE_MODULE(cares_wrap);
|
|
|
|
REFERENCE_MODULE(fs_event_wrap);
|
|
|
|
REFERENCE_MODULE(buffer);
|
|
|
|
REFERENCE_MODULE(contextify);
|
|
|
|
REFERENCE_MODULE(crypto);
|
|
|
|
REFERENCE_MODULE(fs);
|
|
|
|
REFERENCE_MODULE(http_parser);
|
|
|
|
REFERENCE_MODULE(os);
|
|
|
|
REFERENCE_MODULE(v8);
|
|
|
|
REFERENCE_MODULE(zlib);
|
|
|
|
REFERENCE_MODULE(pipe_wrap);
|
|
|
|
REFERENCE_MODULE(process_wrap);
|
|
|
|
REFERENCE_MODULE(signal_wrap);
|
|
|
|
REFERENCE_MODULE(smalloc);
|
|
|
|
REFERENCE_MODULE(spawn_sync);
|
|
|
|
REFERENCE_MODULE(tcp_wrap);
|
|
|
|
REFERENCE_MODULE(timer_wrap);
|
|
|
|
REFERENCE_MODULE(tls_wrap);
|
|
|
|
REFERENCE_MODULE(tty_wrap);
|
|
|
|
REFERENCE_MODULE(udp_wrap);
|
|
|
|
REFERENCE_MODULE(uv);
|
|
|
|
REFERENCE_MODULE(js_stream);
|
|
|
|
#endif // defined(OS_WIN)
|
2014-06-29 10:12:46 +00:00
|
|
|
// Atom Shell's builtin modules.
|
|
|
|
REFERENCE_MODULE(atom_browser_app);
|
|
|
|
REFERENCE_MODULE(atom_browser_auto_updater);
|
2014-07-31 07:08:54 +00:00
|
|
|
REFERENCE_MODULE(atom_browser_content_tracing);
|
2014-06-29 10:12:46 +00:00
|
|
|
REFERENCE_MODULE(atom_browser_dialog);
|
|
|
|
REFERENCE_MODULE(atom_browser_menu);
|
|
|
|
REFERENCE_MODULE(atom_browser_power_monitor);
|
|
|
|
REFERENCE_MODULE(atom_browser_protocol);
|
2014-07-31 04:22:29 +00:00
|
|
|
REFERENCE_MODULE(atom_browser_global_shortcut);
|
2014-06-29 10:12:46 +00:00
|
|
|
REFERENCE_MODULE(atom_browser_tray);
|
2014-10-23 06:04:13 +00:00
|
|
|
REFERENCE_MODULE(atom_browser_web_contents);
|
2015-02-04 23:39:41 +00:00
|
|
|
REFERENCE_MODULE(atom_browser_web_view_manager);
|
2014-06-29 10:12:46 +00:00
|
|
|
REFERENCE_MODULE(atom_browser_window);
|
2014-09-24 03:10:07 +00:00
|
|
|
REFERENCE_MODULE(atom_common_asar);
|
2014-06-29 10:12:46 +00:00
|
|
|
REFERENCE_MODULE(atom_common_clipboard);
|
|
|
|
REFERENCE_MODULE(atom_common_crash_reporter);
|
|
|
|
REFERENCE_MODULE(atom_common_id_weak_map);
|
2015-02-11 08:03:19 +00:00
|
|
|
REFERENCE_MODULE(atom_common_native_image);
|
2014-06-29 10:12:46 +00:00
|
|
|
REFERENCE_MODULE(atom_common_screen);
|
|
|
|
REFERENCE_MODULE(atom_common_shell);
|
|
|
|
REFERENCE_MODULE(atom_common_v8_util);
|
|
|
|
REFERENCE_MODULE(atom_renderer_ipc);
|
2014-10-24 10:24:12 +00:00
|
|
|
REFERENCE_MODULE(atom_renderer_web_frame);
|
2014-06-29 10:12:46 +00:00
|
|
|
#undef REFERENCE_MODULE
|
|
|
|
|
2013-04-13 10:39:09 +00:00
|
|
|
namespace atom {
|
|
|
|
|
2013-07-22 06:58:25 +00:00
|
|
|
namespace {
|
|
|
|
|
2013-12-15 06:20:28 +00:00
|
|
|
// Empty callback for async handle.
|
2014-06-28 14:33:00 +00:00
|
|
|
void UvNoOp(uv_async_t* handle) {
|
2013-07-22 06:58:25 +00:00
|
|
|
}
|
|
|
|
|
2015-03-29 11:53:47 +00:00
|
|
|
// Convert the given vector to an array of C-strings. The strings in the
|
|
|
|
// returned vector are only guaranteed valid so long as the vector of strings
|
|
|
|
// is not modified.
|
|
|
|
scoped_ptr<const char*[]> StringVectorToArgArray(
|
|
|
|
const std::vector<std::string>& vector) {
|
|
|
|
scoped_ptr<const char*[]> array(new const char*[vector.size()]);
|
|
|
|
for (size_t i = 0; i < vector.size(); ++i) {
|
|
|
|
array[i] = vector[i].c_str();
|
|
|
|
}
|
|
|
|
return array.Pass();
|
|
|
|
}
|
|
|
|
|
2015-03-29 11:59:16 +00:00
|
|
|
base::FilePath GetResourcesPath(bool is_browser) {
|
|
|
|
auto command_line = base::CommandLine::ForCurrentProcess();
|
|
|
|
base::FilePath exec_path(command_line->GetProgram());
|
2015-02-13 05:32:58 +00:00
|
|
|
PathService::Get(base::FILE_EXE, &exec_path);
|
2015-03-29 11:59:16 +00:00
|
|
|
|
2015-02-13 05:32:58 +00:00
|
|
|
base::FilePath resources_path =
|
|
|
|
#if defined(OS_MACOSX)
|
|
|
|
is_browser ? exec_path.DirName().DirName().Append("Resources") :
|
|
|
|
exec_path.DirName().DirName().DirName().DirName().DirName()
|
|
|
|
.Append("Resources");
|
|
|
|
#else
|
|
|
|
exec_path.DirName().Append(FILE_PATH_LITERAL("resources"));
|
|
|
|
#endif
|
|
|
|
return resources_path;
|
|
|
|
}
|
|
|
|
|
2013-07-22 06:58:25 +00:00
|
|
|
} // namespace
|
|
|
|
|
2015-01-24 05:02:31 +00:00
|
|
|
node::Environment* global_env = nullptr;
|
2013-12-11 07:48:19 +00:00
|
|
|
|
2013-04-13 13:10:41 +00:00
|
|
|
NodeBindings::NodeBindings(bool is_browser)
|
2013-07-22 06:58:25 +00:00
|
|
|
: is_browser_(is_browser),
|
2015-01-24 05:02:31 +00:00
|
|
|
message_loop_(nullptr),
|
2013-07-22 06:58:25 +00:00
|
|
|
uv_loop_(uv_default_loop()),
|
2014-01-08 03:55:54 +00:00
|
|
|
embed_closed_(false),
|
2015-01-24 05:02:31 +00:00
|
|
|
uv_env_(nullptr),
|
2014-01-08 03:55:54 +00:00
|
|
|
weak_factory_(this) {
|
2013-04-13 10:39:09 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
NodeBindings::~NodeBindings() {
|
2013-11-11 08:57:40 +00:00
|
|
|
// Quit the embed thread.
|
2013-07-22 06:58:25 +00:00
|
|
|
embed_closed_ = true;
|
2013-11-11 08:57:40 +00:00
|
|
|
uv_sem_post(&embed_sem_);
|
2013-07-23 08:55:09 +00:00
|
|
|
WakeupEmbedThread();
|
2013-11-11 08:57:40 +00:00
|
|
|
|
|
|
|
// Wait for everything to be done.
|
2013-07-22 06:58:25 +00:00
|
|
|
uv_thread_join(&embed_thread_);
|
2013-11-11 08:57:40 +00:00
|
|
|
|
|
|
|
// Clear uv.
|
2013-07-22 06:58:25 +00:00
|
|
|
uv_sem_destroy(&embed_sem_);
|
2013-04-13 10:39:09 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
void NodeBindings::Initialize() {
|
2013-12-15 06:20:28 +00:00
|
|
|
// Open node's error reporting system for browser process.
|
|
|
|
node::g_standalone_mode = is_browser_;
|
|
|
|
node::g_upstream_node_mode = false;
|
2013-07-01 13:44:25 +00:00
|
|
|
|
2013-12-15 06:20:28 +00:00
|
|
|
// Init node.
|
|
|
|
// (we assume it would not node::Init would not modify the parameters under
|
|
|
|
// embedded mode).
|
2015-01-24 05:02:31 +00:00
|
|
|
node::Init(nullptr, nullptr, nullptr, nullptr);
|
2013-12-15 06:20:28 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
node::Environment* NodeBindings::CreateEnvironment(
|
|
|
|
v8::Handle<v8::Context> context) {
|
2015-03-17 12:55:53 +00:00
|
|
|
auto args = AtomCommandLine::argv();
|
2013-04-13 10:39:09 +00:00
|
|
|
|
2013-12-11 15:59:03 +00:00
|
|
|
// Feed node the path to initialization script.
|
2015-02-27 20:21:19 +00:00
|
|
|
base::FilePath::StringType process_type = is_browser_ ?
|
|
|
|
FILE_PATH_LITERAL("browser") : FILE_PATH_LITERAL("renderer");
|
2015-03-29 11:59:16 +00:00
|
|
|
base::FilePath resources_path = GetResourcesPath(is_browser_);
|
2013-12-11 15:59:03 +00:00
|
|
|
base::FilePath script_path =
|
2015-02-13 05:05:51 +00:00
|
|
|
resources_path.Append(FILE_PATH_LITERAL("atom.asar"))
|
2015-02-27 20:21:19 +00:00
|
|
|
.Append(process_type)
|
2015-01-24 05:12:45 +00:00
|
|
|
.Append(FILE_PATH_LITERAL("lib"))
|
|
|
|
.Append(FILE_PATH_LITERAL("init.js"));
|
2013-12-11 15:59:03 +00:00
|
|
|
std::string script_path_str = script_path.AsUTF8Unsafe();
|
|
|
|
args.insert(args.begin() + 1, script_path_str.c_str());
|
|
|
|
|
2015-03-29 11:53:47 +00:00
|
|
|
scoped_ptr<const char*[]> c_argv = StringVectorToArgArray(args);
|
2015-03-29 11:59:16 +00:00
|
|
|
node::Environment* env = node::CreateEnvironment(
|
2015-02-13 05:32:58 +00:00
|
|
|
context->GetIsolate(), uv_default_loop(), context,
|
2015-03-29 11:53:47 +00:00
|
|
|
args.size(), c_argv.get(), 0, nullptr);
|
2015-02-13 05:32:58 +00:00
|
|
|
|
|
|
|
mate::Dictionary process(context->GetIsolate(), env->process_object());
|
|
|
|
process.Set("type", process_type);
|
|
|
|
process.Set("resourcesPath", resources_path);
|
|
|
|
return env;
|
2013-04-20 03:13:06 +00:00
|
|
|
}
|
|
|
|
|
2015-01-21 22:00:19 +00:00
|
|
|
void NodeBindings::LoadEnvironment(node::Environment* env) {
|
|
|
|
node::LoadEnvironment(env);
|
|
|
|
}
|
|
|
|
|
2013-07-22 06:58:25 +00:00
|
|
|
void NodeBindings::PrepareMessageLoop() {
|
|
|
|
DCHECK(!is_browser_ || BrowserThread::CurrentlyOn(BrowserThread::UI));
|
|
|
|
|
|
|
|
// Add dummy handle for libuv, otherwise libuv would quit when there is
|
|
|
|
// nothing to do.
|
|
|
|
uv_async_init(uv_loop_, &dummy_uv_handle_, UvNoOp);
|
|
|
|
|
|
|
|
// Start worker that will interrupt main loop when having uv events.
|
|
|
|
uv_sem_init(&embed_sem_, 0);
|
|
|
|
uv_thread_create(&embed_thread_, EmbedThreadRunner, this);
|
|
|
|
}
|
|
|
|
|
|
|
|
void NodeBindings::RunMessageLoop() {
|
|
|
|
DCHECK(!is_browser_ || BrowserThread::CurrentlyOn(BrowserThread::UI));
|
|
|
|
|
|
|
|
// The MessageLoop should have been created, remember the one in main thread.
|
|
|
|
message_loop_ = base::MessageLoop::current();
|
|
|
|
|
|
|
|
// Run uv loop for once to give the uv__io_poll a chance to add all events.
|
|
|
|
UvRunOnce();
|
|
|
|
}
|
|
|
|
|
|
|
|
void NodeBindings::UvRunOnce() {
|
|
|
|
DCHECK(!is_browser_ || BrowserThread::CurrentlyOn(BrowserThread::UI));
|
|
|
|
|
2014-06-28 14:33:00 +00:00
|
|
|
// By default the global env would be used unless user specified another one
|
|
|
|
// (this happens for renderer process, which wraps the uv loop with web page
|
|
|
|
// context).
|
|
|
|
node::Environment* env = uv_env() ? uv_env() : global_env;
|
2014-01-27 07:14:11 +00:00
|
|
|
|
2014-06-28 14:33:00 +00:00
|
|
|
// Use Locker in browser process.
|
2014-08-10 11:14:20 +00:00
|
|
|
mate::Locker locker(env->isolate());
|
2014-06-28 14:33:00 +00:00
|
|
|
v8::HandleScope handle_scope(env->isolate());
|
2014-01-09 13:35:29 +00:00
|
|
|
|
2014-06-28 14:33:00 +00:00
|
|
|
// Enter node context while dealing with uv events.
|
2014-01-09 13:35:29 +00:00
|
|
|
v8::Context::Scope context_scope(env->context());
|
2013-07-22 06:58:25 +00:00
|
|
|
|
|
|
|
// Deal with uv events.
|
2015-01-13 00:17:44 +00:00
|
|
|
int r = uv_run(uv_loop_, UV_RUN_NOWAIT);
|
2013-07-22 06:58:25 +00:00
|
|
|
if (r == 0 || uv_loop_->stop_flag != 0)
|
|
|
|
message_loop_->QuitWhenIdle(); // Quit from uv.
|
|
|
|
|
|
|
|
// Tell the worker thread to continue polling.
|
|
|
|
uv_sem_post(&embed_sem_);
|
|
|
|
}
|
|
|
|
|
|
|
|
void NodeBindings::WakeupMainThread() {
|
|
|
|
DCHECK(message_loop_);
|
|
|
|
message_loop_->PostTask(FROM_HERE, base::Bind(&NodeBindings::UvRunOnce,
|
2014-01-08 03:55:54 +00:00
|
|
|
weak_factory_.GetWeakPtr()));
|
2013-07-22 06:58:25 +00:00
|
|
|
}
|
|
|
|
|
2013-07-23 05:08:40 +00:00
|
|
|
void NodeBindings::WakeupEmbedThread() {
|
|
|
|
uv_async_send(&dummy_uv_handle_);
|
|
|
|
}
|
|
|
|
|
2013-07-22 06:58:25 +00:00
|
|
|
// static
|
|
|
|
void NodeBindings::EmbedThreadRunner(void *arg) {
|
|
|
|
NodeBindings* self = static_cast<NodeBindings*>(arg);
|
|
|
|
|
2014-01-08 02:30:30 +00:00
|
|
|
while (true) {
|
2013-07-22 06:58:25 +00:00
|
|
|
// Wait for the main loop to deal with events.
|
|
|
|
uv_sem_wait(&self->embed_sem_);
|
2014-01-08 02:30:30 +00:00
|
|
|
if (self->embed_closed_)
|
|
|
|
break;
|
|
|
|
|
|
|
|
// Wait for something to happen in uv loop.
|
|
|
|
// Note that the PollEvents() is implemented by derived classes, so when
|
|
|
|
// this class is being destructed the PollEvents() would not be available
|
|
|
|
// anymore. Because of it we must make sure we only invoke PollEvents()
|
|
|
|
// when this class is alive.
|
2013-07-22 06:58:25 +00:00
|
|
|
self->PollEvents();
|
2014-01-08 02:30:30 +00:00
|
|
|
if (self->embed_closed_)
|
|
|
|
break;
|
2013-07-22 06:58:25 +00:00
|
|
|
|
|
|
|
// Deal with event in main thread.
|
|
|
|
self->WakeupMainThread();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-04-13 13:10:41 +00:00
|
|
|
} // namespace atom
|