#include #include #include #include #include #define UNWRAP \ assert(!args.Holder().IsEmpty()); \ assert(args.Holder()->InternalFieldCount() > 0); \ ProcessWrap* wrap = \ static_cast(args.Holder()->GetPointerFromInternalField(0)); \ if (!wrap) { \ SetErrno(UV_EBADF); \ return scope.Close(Integer::New(-1)); \ } namespace node { using v8::Object; using v8::Handle; using v8::Local; using v8::Persistent; using v8::Value; using v8::HandleScope; using v8::FunctionTemplate; using v8::String; using v8::Array; using v8::Function; using v8::TryCatch; using v8::Context; using v8::Arguments; using v8::Integer; class ProcessWrap : public HandleWrap { public: static void Initialize(Handle target) { HandleScope scope; HandleWrap::Initialize(target); Local constructor = FunctionTemplate::New(New); constructor->InstanceTemplate()->SetInternalFieldCount(1); constructor->SetClassName(String::NewSymbol("Process")); NODE_SET_PROTOTYPE_METHOD(constructor, "close", HandleWrap::Close); NODE_SET_PROTOTYPE_METHOD(constructor, "spawn", Spawn); NODE_SET_PROTOTYPE_METHOD(constructor, "kill", Kill); target->Set(String::NewSymbol("Process"), constructor->GetFunction()); } private: static Handle New(const Arguments& args) { // This constructor should not be exposed to public javascript. // Therefore we assert that we are not trying to call this as a // normal function. assert(args.IsConstructCall()); HandleScope scope; ProcessWrap *wrap = new ProcessWrap(args.This()); assert(wrap); return scope.Close(args.This()); } ProcessWrap(Handle object) : HandleWrap(object, NULL) { } ~ProcessWrap() { } static Handle Spawn(const Arguments& args) { HandleScope scope; UNWRAP Local js_options = args[0]->ToObject(); uv_process_options_t options; memset(&options, 0, sizeof(uv_process_options_t)); options.exit_cb = OnExit; // TODO is this possible to do without mallocing ? // options.file Local file_v = js_options->Get(String::New("file")); if (!file_v.IsEmpty() && file_v->IsString()) { String::Utf8Value file(file_v->ToString()); options.file = strdup(*file); } // options.args Local argv_v = js_options->Get(String::New("args")); if (!argv_v.IsEmpty() && argv_v->IsArray()) { Local js_argv = Local::Cast(argv_v); int argc = js_argv->Length(); // Heap allocate to detect errors. +1 is for NULL. options.args = new char*[argc + 1]; for (int i = 0; i < argc; i++) { String::Utf8Value arg(js_argv->Get(i)->ToString()); options.args[i] = strdup(*arg); } options.args[argc] = NULL; } // options.cwd Local cwd_v = js_options->Get(String::New("cwd")); if (!cwd_v.IsEmpty() && cwd_v->IsString()) { String::Utf8Value cwd(cwd_v->ToString()); if (cwd.length() > 0) { options.cwd = strdup(*cwd); } } // options.env Local env_v = js_options->Get(String::New("envPairs")); if (!env_v.IsEmpty() && env_v->IsArray()) { Local env = Local::Cast(env_v); int envc = env->Length(); options.env = new char*[envc + 1]; // Heap allocated to detect errors. for (int i = 0; i < envc; i++) { String::Utf8Value pair(env->Get(i)->ToString()); options.env[i] = strdup(*pair); } options.env[envc] = NULL; } // options.stdin_stream Local stdin_stream_v = js_options->Get(String::New("stdinStream")); if (!stdin_stream_v.IsEmpty() && stdin_stream_v->IsObject()) { PipeWrap* stdin_wrap = PipeWrap::Unwrap(stdin_stream_v->ToObject()); options.stdin_stream = stdin_wrap->UVHandle(); } // options.stdout_stream Local stdout_stream_v = js_options->Get(String::New("stdoutStream")); if (!stdout_stream_v.IsEmpty() && stdout_stream_v->IsObject()) { PipeWrap* stdout_wrap = PipeWrap::Unwrap(stdout_stream_v->ToObject()); options.stdout_stream = stdout_wrap->UVHandle(); } // options.stderr_stream Local stderr_stream_v = js_options->Get(String::New("stderrStream")); if (!stderr_stream_v.IsEmpty() && stderr_stream_v->IsObject()) { PipeWrap* stderr_wrap = PipeWrap::Unwrap(stderr_stream_v->ToObject()); options.stderr_stream = stderr_wrap->UVHandle(); } // options.windows_verbatim_arguments #if defined(_WIN32) && !defined(__CYGWIN__) options.windows_verbatim_arguments = js_options-> Get(String::NewSymbol("windowsVerbatimArguments"))->IsTrue(); #endif int r = uv_spawn(&wrap->process_, options); wrap->SetHandle((uv_handle_t*)&wrap->process_); assert(wrap->process_.data == wrap); wrap->object_->Set(String::New("pid"), Integer::New(wrap->process_.pid)); if (options.args) { for (int i = 0; options.args[i]; i++) free(options.args[i]); delete [] options.args; } free(options.cwd); if (options.env) { for (int i = 0; options.env[i]; i++) free(options.env[i]); delete [] options.env; } if (r) SetErrno(uv_last_error().code); return scope.Close(Integer::New(r)); } static Handle Kill(const Arguments& args) { HandleScope scope; UNWRAP int signal = args[0]->Int32Value(); int r = uv_process_kill(&wrap->process_, signal); if (r) SetErrno(uv_last_error().code); return scope.Close(Integer::New(r)); } static void OnExit(uv_process_t* handle, int exit_status, int term_signal) { HandleScope scope; ProcessWrap* wrap = static_cast(handle->data); assert(wrap); assert(&wrap->process_ == handle); Local argv[2] = { Integer::New(exit_status), String::New(signo_string(term_signal)) }; MakeCallback(wrap->object_, "onexit", 2, argv); } uv_process_t process_; }; } // namespace node NODE_MODULE(node_process_wrap, node::ProcessWrap::Initialize);