atom/common/node_bindings.cc

   1 // Copyright (c) 2013 GitHub, Inc.
   2 // Use of this source code is governed by the MIT license that can be
   3 // found in the LICENSE file.
   4
   5 #include "atom/common/node_bindings.h"
   6
   7 #include <string>
   8 #include <vector>
   9
  10 #include "atom/common/api/event_emitter_caller.h"
  11 #include "atom/common/api/locker.h"
  12 #include "atom/common/atom_command_line.h"
  13 #include "atom/common/native_mate_converters/file_path_converter.h"
  14 #include "atom/common/node_includes.h"
  15 #include "base/command_line.h"
  16 #include "base/base_paths.h"
  17 #include "base/environment.h"
  18 #include "base/files/file_path.h"
  19 #include "base/message_loop/message_loop.h"
  20 #include "base/path_service.h"
  21 #include "content/public/browser/browser_thread.h"
  22 #include "content/public/common/content_paths.h"
  23 #include "native_mate/dictionary.h"
  24 #include "third_party/WebKit/public/web/WebScopedMicrotaskSuppression.h"
  25
  26 using content::BrowserThread;
  27
  28 // Force all builtin modules to be referenced so they can actually run their
  29 // DSO constructors, see http://git.io/DRIqCg.
  30 #define REFERENCE_MODULE(name) \
  31   extern "C" void _register_ ## name(void); \
  32   void (*fp_register_ ## name)(void) = _register_ ## name
  33 // Electron's builtin modules.
  34 REFERENCE_MODULE(atom_browser_app);
  35 REFERENCE_MODULE(atom_browser_auto_updater);
  36 REFERENCE_MODULE(atom_browser_content_tracing);
  37 REFERENCE_MODULE(atom_browser_dialog);
  38 REFERENCE_MODULE(atom_browser_debugger);
  39 REFERENCE_MODULE(atom_browser_desktop_capturer);
  40 REFERENCE_MODULE(atom_browser_download_item);
  41 REFERENCE_MODULE(atom_browser_menu);
  42 REFERENCE_MODULE(atom_browser_power_monitor);
  43 REFERENCE_MODULE(atom_browser_power_save_blocker);
  44 REFERENCE_MODULE(atom_browser_protocol);
  45 REFERENCE_MODULE(atom_browser_global_shortcut);
  46 REFERENCE_MODULE(atom_browser_session);
  47 REFERENCE_MODULE(atom_browser_tray);
  48 REFERENCE_MODULE(atom_browser_web_contents);
  49 REFERENCE_MODULE(atom_browser_web_view_manager);
  50 REFERENCE_MODULE(atom_browser_window);
  51 REFERENCE_MODULE(atom_common_asar);
  52 REFERENCE_MODULE(atom_common_clipboard);
  53 REFERENCE_MODULE(atom_common_crash_reporter);
  54 REFERENCE_MODULE(atom_common_id_weak_map);
  55 REFERENCE_MODULE(atom_common_native_image);
  56 REFERENCE_MODULE(atom_common_screen);
  57 REFERENCE_MODULE(atom_common_shell);
  58 REFERENCE_MODULE(atom_common_v8_util);
  59 REFERENCE_MODULE(atom_renderer_ipc);
  60 REFERENCE_MODULE(atom_renderer_web_frame);
  61 #undef REFERENCE_MODULE
  62
  63 // The "v8::Function::kLineOffsetNotFound" is exported in node.dll, but the
  64 // linker can not find it, could be a bug of VS.
  65 #if defined(OS_WIN) && !defined(DEBUG)
  66 namespace v8 {
  67 const int Function::kLineOffsetNotFound = -1;
  68 }
  69 #endif
  70
  71 namespace atom {
  72
  73 namespace {
  74
  75 // Empty callback for async handle.
  76 void UvNoOp(uv_async_t* handle) {
  77 }
  78
  79 // Convert the given vector to an array of C-strings. The strings in the
  80 // returned vector are only guaranteed valid so long as the vector of strings
  81 // is not modified.
  82 scoped_ptr<const char*[]> StringVectorToArgArray(
  83     const std::vector<std::string>& vector) {
  84   scoped_ptr<const char*[]> array(new const char*[vector.size()]);
  85   for (size_t i = 0; i < vector.size(); ++i) {
  86     array[i] = vector[i].c_str();
  87   }
  88   return array;
  89 }
  90
  91 base::FilePath GetResourcesPath(bool is_browser) {
  92   auto command_line = base::CommandLine::ForCurrentProcess();
  93   base::FilePath exec_path(command_line->GetProgram());
  94   PathService::Get(base::FILE_EXE, &exec_path);
  95
  96   base::FilePath resources_path =
  97 #if defined(OS_MACOSX)
  98       is_browser ? exec_path.DirName().DirName().Append("Resources") :
  99                    exec_path.DirName().DirName().DirName().DirName().DirName()
 100                             .Append("Resources");
 101 #else
 102       exec_path.DirName().Append(FILE_PATH_LITERAL("resources"));
 103 #endif
 104   return resources_path;
 105 }
 106
 107 }  // namespace
 108
 109 node::Environment* global_env = nullptr;
 110
 111 NodeBindings::NodeBindings(bool is_browser)
 112     : is_browser_(is_browser),
 113       message_loop_(nullptr),
 114       uv_loop_(uv_default_loop()),
 115       embed_closed_(false),
 116       uv_env_(nullptr),
 117       weak_factory_(this) {
 118 }
 119
 120 NodeBindings::~NodeBindings() {
 121   // Quit the embed thread.
 122   embed_closed_ = true;
 123   uv_sem_post(&embed_sem_);
 124   WakeupEmbedThread();
 125
 126   // Wait for everything to be done.
 127   uv_thread_join(&embed_thread_);
 128
 129   // Clear uv.
 130   uv_sem_destroy(&embed_sem_);
 131 }
 132
 133 void NodeBindings::Initialize() {
 134   // Open node's error reporting system for browser process.
 135   node::g_standalone_mode = is_browser_;
 136   node::g_upstream_node_mode = false;
 137
 138 #if defined(OS_LINUX)
 139   // Get real command line in renderer process forked by zygote.
 140   if (!is_browser_)
 141     AtomCommandLine::InitializeFromCommandLine();
 142 #endif
 143
 144   // Init node.
 145   // (we assume node::Init would not modify the parameters under embedded mode).
 146   node::Init(nullptr, nullptr, nullptr, nullptr);
 147
 148 #if defined(OS_WIN)
 149   // uv_init overrides error mode to suppress the default crash dialog, bring
 150   // it back if user wants to show it.
 151   scoped_ptr<base::Environment> env(base::Environment::Create());
 152   if (env->HasVar("ELECTRON_DEFAULT_ERROR_MODE"))
 153     SetErrorMode(0);
 154 #endif
 155 }
 156
 157 node::Environment* NodeBindings::CreateEnvironment(
 158     v8::Handle<v8::Context> context) {
 159   auto args = AtomCommandLine::argv();
 160
 161   // Feed node the path to initialization script.
 162   base::FilePath::StringType process_type = is_browser_ ?
 163       FILE_PATH_LITERAL("browser") : FILE_PATH_LITERAL("renderer");
 164   base::FilePath resources_path = GetResourcesPath(is_browser_);
 165   base::FilePath script_path =
 166       resources_path.Append(FILE_PATH_LITERAL("atom.asar"))
 167                     .Append(process_type)
 168                     .Append(FILE_PATH_LITERAL("init.js"));
 169   std::string script_path_str = script_path.AsUTF8Unsafe();
 170   args.insert(args.begin() + 1, script_path_str.c_str());
 171
 172   scoped_ptr<const char*[]> c_argv = StringVectorToArgArray(args);
 173   node::Environment* env = node::CreateEnvironment(
 174       context->GetIsolate(), uv_default_loop(), context,
 175       args.size(), c_argv.get(), 0, nullptr);
 176
 177   mate::Dictionary process(context->GetIsolate(), env->process_object());
 178   process.Set("type", process_type);
 179   process.Set("resourcesPath", resources_path);
 180   // The path to helper app.
 181   base::FilePath helper_exec_path;
 182   PathService::Get(content::CHILD_PROCESS_EXE, &helper_exec_path);
 183   process.Set("helperExecPath", helper_exec_path);
 184   return env;
 185 }
 186
 187 void NodeBindings::LoadEnvironment(node::Environment* env) {
 188   node::LoadEnvironment(env);
 189   mate::EmitEvent(env->isolate(), env->process_object(), "loaded");
 190 }
 191
 192 void NodeBindings::PrepareMessageLoop() {
 193   DCHECK(!is_browser_ || BrowserThread::CurrentlyOn(BrowserThread::UI));
 194
 195   // Add dummy handle for libuv, otherwise libuv would quit when there is
 196   // nothing to do.
 197   uv_async_init(uv_loop_, &dummy_uv_handle_, UvNoOp);
 198
 199   // Start worker that will interrupt main loop when having uv events.
 200   uv_sem_init(&embed_sem_, 0);
 201   uv_thread_create(&embed_thread_, EmbedThreadRunner, this);
 202 }
 203
 204 void NodeBindings::RunMessageLoop() {
 205   DCHECK(!is_browser_ || BrowserThread::CurrentlyOn(BrowserThread::UI));
 206
 207   // The MessageLoop should have been created, remember the one in main thread.
 208   message_loop_ = base::MessageLoop::current();
 209
 210   // Run uv loop for once to give the uv__io_poll a chance to add all events.
 211   UvRunOnce();
 212 }
 213
 214 void NodeBindings::UvRunOnce() {
 215   DCHECK(!is_browser_ || BrowserThread::CurrentlyOn(BrowserThread::UI));
 216
 217   // By default the global env would be used unless user specified another one
 218   // (this happens for renderer process, which wraps the uv loop with web page
 219   // context).
 220   node::Environment* env = uv_env() ? uv_env() : global_env;
 221
 222   // Use Locker in browser process.
 223   mate::Locker locker(env->isolate());
 224   v8::HandleScope handle_scope(env->isolate());
 225
 226   // Enter node context while dealing with uv events.
 227   v8::Context::Scope context_scope(env->context());
 228
 229   // Perform microtask checkpoint after running JavaScript.
 230   scoped_ptr<blink::WebScopedRunV8Script> script_scope(
 231       is_browser_ ? nullptr : new blink::WebScopedRunV8Script);
 232
 233   // Deal with uv events.
 234   int r = uv_run(uv_loop_, UV_RUN_NOWAIT);
 235   if (r == 0 || uv_loop_->stop_flag != 0)
 236     message_loop_->QuitWhenIdle();  // Quit from uv.
 237
 238   // Tell the worker thread to continue polling.
 239   uv_sem_post(&embed_sem_);
 240 }
 241
 242 void NodeBindings::WakeupMainThread() {
 243   DCHECK(message_loop_);
 244   message_loop_->PostTask(FROM_HERE, base::Bind(&NodeBindings::UvRunOnce,
 245                                                 weak_factory_.GetWeakPtr()));
 246 }
 247
 248 void NodeBindings::WakeupEmbedThread() {
 249   uv_async_send(&dummy_uv_handle_);
 250 }
 251
 252 // static
 253 void NodeBindings::EmbedThreadRunner(void *arg) {
 254   NodeBindings* self = static_cast<NodeBindings*>(arg);
 255
 256   while (true) {
 257     // Wait for the main loop to deal with events.
 258     uv_sem_wait(&self->embed_sem_);
 259     if (self->embed_closed_)
 260       break;
 261
 262     // Wait for something to happen in uv loop.
 263     // Note that the PollEvents() is implemented by derived classes, so when
 264     // this class is being destructed the PollEvents() would not be available
 265     // anymore. Because of it we must make sure we only invoke PollEvents()
 266     // when this class is alive.
 267     self->PollEvents();
 268     if (self->embed_closed_)
 269       break;
 270
 271     // Deal with event in main thread.
 272     self->WakeupMainThread();
 273   }
 274 }
 275
 276 }  // namespace atom