mirror of https://github.com/lukechilds/node.git
Browse Source
This implements a nested event loop that makes it possible to control a child process, while blocking the main loop until the process exits.v0.11.12-release
Bert Belder
11 years ago
4 changed files with 1311 additions and 0 deletions
File diff suppressed because it is too large
@ -0,0 +1,247 @@ |
|||||
|
// Copyright Joyent, Inc. and other Node contributors.
|
||||
|
//
|
||||
|
// Permission is hereby granted, free of charge, to any person obtaining a
|
||||
|
// copy of this software and associated documentation files (the
|
||||
|
// "Software"), to deal in the Software without restriction, including
|
||||
|
// without limitation the rights to use, copy, modify, merge, publish,
|
||||
|
// distribute, sublicense, and/or sell copies of the Software, and to permit
|
||||
|
// persons to whom the Software is furnished to do so, subject to the
|
||||
|
// following conditions:
|
||||
|
//
|
||||
|
// The above copyright notice and this permission notice shall be included
|
||||
|
// in all copies or substantial portions of the Software.
|
||||
|
//
|
||||
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
|
||||
|
// OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
||||
|
// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN
|
||||
|
// NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
|
||||
|
// DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
|
||||
|
// OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
|
||||
|
// USE OR OTHER DEALINGS IN THE SOFTWARE.
|
||||
|
|
||||
|
#ifndef SRC_SPAWN_SYNC_H_ |
||||
|
#define SRC_SPAWN_SYNC_H_ |
||||
|
|
||||
|
#include "node.h" |
||||
|
#include "node_buffer.h" |
||||
|
|
||||
|
namespace node { |
||||
|
|
||||
|
using v8::Array; |
||||
|
using v8::Context; |
||||
|
using v8::FunctionCallbackInfo; |
||||
|
using v8::Handle; |
||||
|
using v8::HandleScope; |
||||
|
using v8::Integer; |
||||
|
using v8::Isolate; |
||||
|
using v8::Local; |
||||
|
using v8::Null; |
||||
|
using v8::Number; |
||||
|
using v8::Object; |
||||
|
using v8::String; |
||||
|
using v8::Value; |
||||
|
|
||||
|
|
||||
|
class SyncProcessOutputBuffer; |
||||
|
class SyncProcessStdioPipe; |
||||
|
class SyncProcessRunner; |
||||
|
|
||||
|
|
||||
|
class SyncProcessOutputBuffer { |
||||
|
static const unsigned int kBufferSize = 65536; |
||||
|
|
||||
|
public: |
||||
|
inline SyncProcessOutputBuffer(); |
||||
|
|
||||
|
inline void OnAlloc(size_t suggested_size, uv_buf_t* buf) const; |
||||
|
inline void OnRead(const uv_buf_t* buf, size_t nread); |
||||
|
|
||||
|
inline size_t Copy(char* dest) const; |
||||
|
|
||||
|
inline unsigned int available() const; |
||||
|
inline unsigned int used() const; |
||||
|
|
||||
|
inline SyncProcessOutputBuffer* next() const; |
||||
|
inline void set_next(SyncProcessOutputBuffer* next); |
||||
|
|
||||
|
private: |
||||
|
// Use unsigned int because that's what `uv_buf_init` takes.
|
||||
|
mutable char data_[kBufferSize]; |
||||
|
unsigned int used_; |
||||
|
|
||||
|
SyncProcessOutputBuffer* next_; |
||||
|
}; |
||||
|
|
||||
|
|
||||
|
class SyncProcessStdioPipe { |
||||
|
enum Lifecycle { |
||||
|
kUninitialized = 0, |
||||
|
kInitialized, |
||||
|
kStarted, |
||||
|
kClosing, |
||||
|
kClosed |
||||
|
}; |
||||
|
|
||||
|
public: |
||||
|
SyncProcessStdioPipe(SyncProcessRunner* process_handler, |
||||
|
bool readable, |
||||
|
bool writable, |
||||
|
uv_buf_t input_buffer); |
||||
|
~SyncProcessStdioPipe(); |
||||
|
|
||||
|
int Initialize(uv_loop_t* loop); |
||||
|
int Start(); |
||||
|
void Close(); |
||||
|
|
||||
|
Local<Object> GetOutputAsBuffer() const; |
||||
|
|
||||
|
inline bool readable() const; |
||||
|
inline bool writable() const; |
||||
|
inline uv_stdio_flags uv_flags() const; |
||||
|
|
||||
|
inline uv_pipe_t* uv_pipe() const; |
||||
|
inline uv_stream_t* uv_stream() const; |
||||
|
inline uv_handle_t* uv_handle() const; |
||||
|
|
||||
|
private: |
||||
|
inline size_t OutputLength() const; |
||||
|
inline void CopyOutput(char* dest) const; |
||||
|
|
||||
|
inline void OnAlloc(size_t suggested_size, uv_buf_t* buf); |
||||
|
inline void OnRead(const uv_buf_t* buf, ssize_t nread); |
||||
|
inline void OnWriteDone(int result); |
||||
|
inline void OnShutdownDone(int result); |
||||
|
inline void OnClose(); |
||||
|
|
||||
|
inline void SetError(int error); |
||||
|
|
||||
|
static void AllocCallback(uv_handle_t* handle, |
||||
|
size_t suggested_size, |
||||
|
uv_buf_t* buf); |
||||
|
static void ReadCallback(uv_stream_t* stream, |
||||
|
ssize_t nread, |
||||
|
const uv_buf_t* buf); |
||||
|
static void WriteCallback(uv_write_t* req, int result); |
||||
|
static void ShutdownCallback(uv_shutdown_t* req, int result); |
||||
|
static void CloseCallback(uv_handle_t* handle); |
||||
|
|
||||
|
SyncProcessRunner* process_handler_; |
||||
|
|
||||
|
bool readable_; |
||||
|
bool writable_; |
||||
|
uv_buf_t input_buffer_; |
||||
|
|
||||
|
SyncProcessOutputBuffer* first_output_buffer_; |
||||
|
SyncProcessOutputBuffer* last_output_buffer_; |
||||
|
|
||||
|
mutable uv_pipe_t uv_pipe_; |
||||
|
uv_write_t write_req_; |
||||
|
uv_shutdown_t shutdown_req_; |
||||
|
|
||||
|
Lifecycle lifecycle_; |
||||
|
}; |
||||
|
|
||||
|
|
||||
|
class SyncProcessRunner { |
||||
|
enum Lifecycle { |
||||
|
kUninitialized = 0, |
||||
|
kInitialized, |
||||
|
kHandlesClosed |
||||
|
}; |
||||
|
|
||||
|
public: |
||||
|
static void Initialize(Handle<Object> target, |
||||
|
Handle<Value> unused, |
||||
|
Handle<Context> context); |
||||
|
static void Spawn(const FunctionCallbackInfo<Value>& args); |
||||
|
|
||||
|
private: |
||||
|
friend class SyncProcessStdioPipe; |
||||
|
|
||||
|
explicit SyncProcessRunner(Environment* env_); |
||||
|
~SyncProcessRunner(); |
||||
|
|
||||
|
inline Environment* env() const; |
||||
|
|
||||
|
Local<Object> Run(Local<Value> options); |
||||
|
void TryInitializeAndRunLoop(Local<Value> options); |
||||
|
void CloseHandlesAndDeleteLoop(); |
||||
|
|
||||
|
void CloseStdioPipes(); |
||||
|
void CloseKillTimer(); |
||||
|
|
||||
|
void Kill(); |
||||
|
void IncrementBufferSizeAndCheckOverflow(ssize_t length); |
||||
|
|
||||
|
void OnExit(int64_t exit_status, int term_signal); |
||||
|
void OnKillTimerTimeout(int status); |
||||
|
|
||||
|
int GetError(); |
||||
|
void SetError(int error); |
||||
|
void SetPipeError(int pipe_error); |
||||
|
|
||||
|
Local<Object> BuildResultObject(); |
||||
|
Local<Array> BuildOutputArray(); |
||||
|
|
||||
|
int ParseOptions(Local<Value> js_value); |
||||
|
int ParseStdioOptions(Local<Value> js_value); |
||||
|
int ParseStdioOption(int child_fd, Local<Object> js_stdio_option); |
||||
|
|
||||
|
inline int AddStdioIgnore(uint32_t child_fd); |
||||
|
inline int AddStdioPipe(uint32_t child_fd, |
||||
|
bool readable, |
||||
|
bool writable, |
||||
|
uv_buf_t input_buffer); |
||||
|
inline int AddStdioInheritFD(uint32_t child_fd, int inherit_fd); |
||||
|
|
||||
|
static bool IsSet(Local<Value> value); |
||||
|
template <typename t> static bool CheckRange(Local<Value> js_value); |
||||
|
static int CopyJsString(Local<Value> js_value, const char** target); |
||||
|
static int CopyJsStringArray(Local<Value> js_value, char** target); |
||||
|
|
||||
|
static void ExitCallback(uv_process_t* handle, |
||||
|
int64_t exit_status, |
||||
|
int term_signal); |
||||
|
static void KillTimerCallback(uv_timer_t* handle, int status); |
||||
|
static void KillTimerCloseCallback(uv_handle_t* handle); |
||||
|
|
||||
|
size_t max_buffer_; |
||||
|
uint64_t timeout_; |
||||
|
int kill_signal_; |
||||
|
|
||||
|
uv_loop_t* uv_loop_; |
||||
|
|
||||
|
uint32_t stdio_count_; |
||||
|
uv_stdio_container_t* uv_stdio_containers_; |
||||
|
SyncProcessStdioPipe** stdio_pipes_; |
||||
|
bool stdio_pipes_initialized_; |
||||
|
|
||||
|
uv_process_options_t uv_process_options_; |
||||
|
const char* file_buffer_; |
||||
|
char* args_buffer_; |
||||
|
char* env_buffer_; |
||||
|
const char* cwd_buffer_; |
||||
|
|
||||
|
uv_process_t uv_process_; |
||||
|
bool killed_; |
||||
|
|
||||
|
size_t buffered_output_size_; |
||||
|
int64_t exit_status_; |
||||
|
int term_signal_; |
||||
|
|
||||
|
uv_timer_t uv_timer_; |
||||
|
bool kill_timer_initialized_; |
||||
|
|
||||
|
// Errors that happen in one of the pipe handlers are stored in the
|
||||
|
// `pipe_error` field. They are treated as "low-priority", only to be
|
||||
|
// reported if no more serious errors happened.
|
||||
|
int error_; |
||||
|
int pipe_error_; |
||||
|
|
||||
|
Lifecycle lifecycle_; |
||||
|
|
||||
|
Environment* env_; |
||||
|
}; |
||||
|
} |
||||
|
|
||||
|
#endif // SRC_SPAWN_SYNC_H_
|
||||
Loading…
Reference in new issue