189 lines
5.8 KiB
C++
189 lines
5.8 KiB
C++
// Copyright (c) 2013 GitHub, Inc. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style license that can be
|
|
// found in the LICENSE file.
|
|
|
|
#include "common/api/atom_bindings.h"
|
|
|
|
#include "base/logging.h"
|
|
#include "common/atom_version.h"
|
|
#include "common/v8/native_type_conversions.h"
|
|
|
|
#include "common/v8/node_common.h"
|
|
|
|
namespace atom {
|
|
|
|
namespace {
|
|
|
|
static int kMaxCallStackSize = 200; // Same with WebKit.
|
|
|
|
// Async handle to wake up uv loop.
|
|
static uv_async_t g_next_tick_uv_handle;
|
|
|
|
// Async handle to execute the stored v8 callback.
|
|
static uv_async_t g_callback_uv_handle;
|
|
|
|
// Stored v8 callback, to be called by the async handler.
|
|
RefCountedV8Function g_v8_callback;
|
|
|
|
// Dummy class type that used for crashing the program.
|
|
struct DummyClass { bool crash; };
|
|
|
|
// Async handler to call next process.nextTick callbacks.
|
|
void UvCallNextTick(uv_async_t* handle, int status) {
|
|
node::Environment* env = node::Environment::GetCurrent(node_isolate);
|
|
node::Environment::TickInfo* tick_info = env->tick_info();
|
|
|
|
if (tick_info->in_tick())
|
|
return;
|
|
|
|
if (tick_info->length() == 0) {
|
|
tick_info->set_index(0);
|
|
return;
|
|
}
|
|
|
|
tick_info->set_in_tick(true);
|
|
env->tick_callback_function()->Call(env->process_object(), 0, NULL);
|
|
tick_info->set_in_tick(false);
|
|
}
|
|
|
|
// Async handler to execute the stored v8 callback.
|
|
void UvOnCallback(uv_async_t* handle, int status) {
|
|
v8::HandleScope handle_scope(node_isolate);
|
|
v8::Handle<v8::Object> global = v8::Context::GetCurrent()->Global();
|
|
g_v8_callback->NewHandle()->Call(global, 0, NULL);
|
|
}
|
|
|
|
v8::Handle<v8::Object> DumpStackFrame(v8::Handle<v8::StackFrame> stack_frame) {
|
|
v8::Local<v8::Object> result = v8::Object::New();
|
|
result->Set(ToV8Value("line"), ToV8Value(stack_frame->GetLineNumber()));
|
|
result->Set(ToV8Value("column"), ToV8Value(stack_frame->GetColumn()));
|
|
|
|
v8::Handle<v8::String> script = stack_frame->GetScriptName();
|
|
if (!script.IsEmpty())
|
|
result->Set(ToV8Value("script"), script);
|
|
|
|
v8::Handle<v8::String> function = stack_frame->GetScriptNameOrSourceURL();
|
|
if (!function.IsEmpty())
|
|
result->Set(ToV8Value("function"), function);
|
|
return result;
|
|
}
|
|
|
|
} // namespace
|
|
|
|
// Defined in atom_extensions.cc.
|
|
node::node_module_struct* GetBuiltinModule(const char *name, bool is_browser);
|
|
|
|
AtomBindings::AtomBindings() {
|
|
uv_async_init(uv_default_loop(), &g_next_tick_uv_handle, UvCallNextTick);
|
|
uv_async_init(uv_default_loop(), &g_callback_uv_handle, UvOnCallback);
|
|
}
|
|
|
|
AtomBindings::~AtomBindings() {
|
|
}
|
|
|
|
void AtomBindings::BindTo(v8::Handle<v8::Object> process) {
|
|
v8::HandleScope handle_scope(node_isolate);
|
|
|
|
NODE_SET_METHOD(process, "atomBinding", Binding);
|
|
NODE_SET_METHOD(process, "crash", Crash);
|
|
NODE_SET_METHOD(process, "activateUvLoop", ActivateUVLoop);
|
|
NODE_SET_METHOD(process, "log", Log);
|
|
NODE_SET_METHOD(process, "getCurrentStackTrace", GetCurrentStackTrace);
|
|
NODE_SET_METHOD(process, "scheduleCallback", ScheduleCallback);
|
|
|
|
process->Get(v8::String::New("versions"))->ToObject()->
|
|
Set(v8::String::New("atom-shell"), v8::String::New(ATOM_VERSION_STRING));
|
|
}
|
|
|
|
// static
|
|
void AtomBindings::Binding(const v8::FunctionCallbackInfo<v8::Value>& args) {
|
|
v8::Local<v8::String> module = args[0]->ToString();
|
|
v8::String::Utf8Value module_v(module);
|
|
node::node_module_struct* modp;
|
|
|
|
v8::Local<v8::Object> process = v8::Context::GetCurrent()->Global()->
|
|
Get(v8::String::New("process"))->ToObject();
|
|
DCHECK(!process.IsEmpty());
|
|
|
|
// is_browser = process.__atom_type == 'browser'.
|
|
bool is_browser = std::string("browser") == *v8::String::Utf8Value(
|
|
process->Get(v8::String::New("__atom_type")));
|
|
|
|
// Cached in process.__atom_binding_cache.
|
|
v8::Local<v8::Object> binding_cache;
|
|
v8::Local<v8::String> bc_name = v8::String::New("__atomBindingCache");
|
|
if (process->Has(bc_name)) {
|
|
binding_cache = process->Get(bc_name)->ToObject();
|
|
DCHECK(!binding_cache.IsEmpty());
|
|
} else {
|
|
binding_cache = v8::Object::New();
|
|
process->Set(bc_name, binding_cache);
|
|
}
|
|
|
|
v8::Local<v8::Object> exports;
|
|
|
|
if (binding_cache->Has(module)) {
|
|
exports = binding_cache->Get(module)->ToObject();
|
|
return args.GetReturnValue().Set(exports);
|
|
}
|
|
|
|
if ((modp = GetBuiltinModule(*module_v, is_browser)) != NULL) {
|
|
exports = v8::Object::New();
|
|
// Internal bindings don't have a "module" object,
|
|
// only exports.
|
|
modp->register_func(exports, v8::Undefined());
|
|
binding_cache->Set(module, exports);
|
|
return args.GetReturnValue().Set(exports);
|
|
}
|
|
|
|
return node::ThrowError("No such module");
|
|
}
|
|
|
|
// static
|
|
void AtomBindings::Crash(const v8::FunctionCallbackInfo<v8::Value>& args) {
|
|
static_cast<DummyClass*>(NULL)->crash = true;
|
|
}
|
|
|
|
// static
|
|
void AtomBindings::ActivateUVLoop(
|
|
const v8::FunctionCallbackInfo<v8::Value>& args) {
|
|
uv_async_send(&g_next_tick_uv_handle);
|
|
}
|
|
|
|
// static
|
|
void AtomBindings::Log(const v8::FunctionCallbackInfo<v8::Value>& args) {
|
|
std::string message;
|
|
for (int i = 0; i < args.Length(); ++i)
|
|
message += *v8::String::Utf8Value(args[i]);
|
|
|
|
logging::LogMessage("CONSOLE", 0, 0).stream() << message;
|
|
}
|
|
|
|
// static
|
|
void AtomBindings::GetCurrentStackTrace(
|
|
const v8::FunctionCallbackInfo<v8::Value>& args) {
|
|
v8::HandleScope handle_scope(args.GetIsolate());
|
|
|
|
int stack_limit = kMaxCallStackSize;
|
|
FromV8Arguments(args, &stack_limit);
|
|
|
|
v8::Local<v8::StackTrace> stack_trace = v8::StackTrace::CurrentStackTrace(
|
|
stack_limit, v8::StackTrace::kDetailed);
|
|
|
|
int frame_count = stack_trace->GetFrameCount();
|
|
v8::Local<v8::Array> result = v8::Array::New(frame_count);
|
|
for (int i = 0; i < frame_count; ++i)
|
|
result->Set(i, DumpStackFrame(stack_trace->GetFrame(i)));
|
|
|
|
args.GetReturnValue().Set(result);
|
|
}
|
|
|
|
// static
|
|
void AtomBindings::ScheduleCallback(
|
|
const v8::FunctionCallbackInfo<v8::Value>& args) {
|
|
if (!FromV8Arguments(args, &g_v8_callback))
|
|
return node::ThrowTypeError("Bad arguments");
|
|
uv_async_send(&g_callback_uv_handle);
|
|
}
|
|
|
|
} // namespace atom
|