#include "connection_wrap.h"

#include "connect_wrap.h"

#include "env-inl.h"

#include "pipe_wrap.h"

#include "stream_base-inl.h"

#include "stream_wrap.h"

#include "tcp_wrap.h"

#include "util-inl.h"

namespace node {

using v8::Boolean;

using v8::Context;

using v8::HandleScope;

using v8::Integer;

using v8::Local;

using v8::Object;

using v8::Value;

template <typename WrapType, typename UVType>

ConnectionWrap<WrapType, UVType>::ConnectionWrap(Environment* env,

Local<Object> object,

ProviderType provider)

: LibuvStreamWrap(env,

object,

reinterpret_cast<uv_stream_t*>(&handle_),

provider) {}

template <typename WrapType, typename UVType>

void ConnectionWrap<WrapType, UVType>::OnConnection(uv_stream_t* handle,

int status) {

WrapType* wrap_data = static_cast<WrapType*>(handle->data);

CHECK_NOT_NULL(wrap_data);

CHECK_EQ(&wrap_data->handle_, reinterpret_cast<UVType*>(handle));

Environment* env = wrap_data->env();

HandleScope handle_scope(env->isolate());

Context::Scope context_scope(env->context());

// We should not be getting this callback if someone has already called

// uv_close() on the handle.

CHECK_EQ(wrap_data->persistent().IsEmpty(), false);

Local<Value> client_handle;

if (status == 0) {

// Instantiate the client javascript object and handle.

Local<Object> client_obj;

if (!WrapType::Instantiate(env, wrap_data, WrapType::SOCKET)

.ToLocal(&client_obj))

return;

// Unwrap the client javascript object.

WrapType* wrap;

ASSIGN_OR_RETURN_UNWRAP(&wrap, client_obj);

uv_stream_t* client = reinterpret_cast<uv_stream_t*>(&wrap->handle_);

// uv_accept can fail if the new connection has already been closed, in

// which case an EAGAIN (resource temporarily unavailable) will be

// returned.

if (uv_accept(handle, client))

return;

// Successful accept. Call the onconnection callback in JavaScript land.

client_handle = client_obj;

} else {

client_handle = Undefined(env->isolate());

}

Local<Value> argv[] = { Integer::New(env->isolate(), status), client_handle };

wrap_data->MakeCallback(env->onconnection_string(), arraysize(argv), argv);

}

template <typename WrapType, typename UVType>

void ConnectionWrap<WrapType, UVType>::AfterConnect(uv_connect_t* req,

int status) {

BaseObjectPtr<ConnectWrap> req_wrap{static_cast<ConnectWrap*>(req->data)};

CHECK(req_wrap);

WrapType* wrap = static_cast<WrapType*>(req->handle->data);

CHECK_EQ(req_wrap->env(), wrap->env());

Environment* env = wrap->env();

HandleScope handle_scope(env->isolate());

Context::Scope context_scope(env->context());

// The wrap and request objects should still be there.

CHECK_EQ(req_wrap->persistent().IsEmpty(), false);

CHECK_EQ(wrap->persistent().IsEmpty(), false);

bool readable, writable;

if (status) {

readable = writable = false;

} else {

readable = uv_is_readable(req->handle) != 0;

writable = uv_is_writable(req->handle) != 0;

}

Local<Value> argv[5] = {

Integer::New(env->isolate(), status),

wrap->object(),

req_wrap->object(),

Boolean::New(env->isolate(), readable),

Boolean::New(env->isolate(), writable)

};

TRACE_EVENT_NESTABLE_ASYNC_END1(TRACING_CATEGORY_NODE2(net, native),

"connect",

req_wrap.get(),

"status",

status);

req_wrap->MakeCallback(env->oncomplete_string(), arraysize(argv), argv);

}

template ConnectionWrap<PipeWrap, uv_pipe_t>::ConnectionWrap(

Environment* env,

Local<Object> object,

ProviderType provider);

template ConnectionWrap<TCPWrap, uv_tcp_t>::ConnectionWrap(

Environment* env,

Local<Object> object,

ProviderType provider);

template void ConnectionWrap<PipeWrap, uv_pipe_t>::OnConnection(

uv_stream_t* handle, int status);

template void ConnectionWrap<TCPWrap, uv_tcp_t>::OnConnection(

uv_stream_t* handle, int status);

template void ConnectionWrap<PipeWrap, uv_pipe_t>::AfterConnect(

uv_connect_t* handle, int status);

template void ConnectionWrap<TCPWrap, uv_tcp_t>::AfterConnect(

uv_connect_t* handle, int status);

} // namespace node