1 #include "connection_wrap.h"
2
3 #include "connect_wrap.h"
4 #include "env-inl.h"
5 #include "pipe_wrap.h"
6 #include "stream_base-inl.h"
7 #include "stream_wrap.h"
8 #include "tcp_wrap.h"
9 #include "util-inl.h"
10
11 namespace node {
12
13 using v8::Boolean;
14 using v8::Context;
15 using v8::HandleScope;
16 using v8::Integer;
17 using v8::Local;
18 using v8::Object;
19 using v8::Value;
20
21
22 template <typename WrapType, typename UVType>
ConnectionWrap(Environment * env,Local<Object> object,ProviderType provider)23 ConnectionWrap<WrapType, UVType>::ConnectionWrap(Environment* env,
24 Local<Object> object,
25 ProviderType provider)
26 : LibuvStreamWrap(env,
27 object,
28 reinterpret_cast<uv_stream_t*>(&handle_),
29 provider) {}
30
31
32 template <typename WrapType, typename UVType>
OnConnection(uv_stream_t * handle,int status)33 void ConnectionWrap<WrapType, UVType>::OnConnection(uv_stream_t* handle,
34 int status) {
35 WrapType* wrap_data = static_cast<WrapType*>(handle->data);
36 CHECK_NOT_NULL(wrap_data);
37 CHECK_EQ(&wrap_data->handle_, reinterpret_cast<UVType*>(handle));
38
39 Environment* env = wrap_data->env();
40 HandleScope handle_scope(env->isolate());
41 Context::Scope context_scope(env->context());
42
43 // We should not be getting this callback if someone has already called
44 // uv_close() on the handle.
45 CHECK_EQ(wrap_data->persistent().IsEmpty(), false);
46
47 Local<Value> client_handle;
48
49 if (status == 0) {
50 // Instantiate the client javascript object and handle.
51 Local<Object> client_obj;
52 if (!WrapType::Instantiate(env, wrap_data, WrapType::SOCKET)
53 .ToLocal(&client_obj))
54 return;
55
56 // Unwrap the client javascript object.
57 WrapType* wrap;
58 ASSIGN_OR_RETURN_UNWRAP(&wrap, client_obj);
59 uv_stream_t* client = reinterpret_cast<uv_stream_t*>(&wrap->handle_);
60 // uv_accept can fail if the new connection has already been closed, in
61 // which case an EAGAIN (resource temporarily unavailable) will be
62 // returned.
63 if (uv_accept(handle, client))
64 return;
65
66 // Successful accept. Call the onconnection callback in JavaScript land.
67 client_handle = client_obj;
68 } else {
69 client_handle = Undefined(env->isolate());
70 }
71
72 Local<Value> argv[] = { Integer::New(env->isolate(), status), client_handle };
73 wrap_data->MakeCallback(env->onconnection_string(), arraysize(argv), argv);
74 }
75
76
77 template <typename WrapType, typename UVType>
AfterConnect(uv_connect_t * req,int status)78 void ConnectionWrap<WrapType, UVType>::AfterConnect(uv_connect_t* req,
79 int status) {
80 std::unique_ptr<ConnectWrap> req_wrap
81 (static_cast<ConnectWrap*>(req->data));
82 CHECK_NOT_NULL(req_wrap);
83 WrapType* wrap = static_cast<WrapType*>(req->handle->data);
84 CHECK_EQ(req_wrap->env(), wrap->env());
85 Environment* env = wrap->env();
86
87 HandleScope handle_scope(env->isolate());
88 Context::Scope context_scope(env->context());
89
90 // The wrap and request objects should still be there.
91 CHECK_EQ(req_wrap->persistent().IsEmpty(), false);
92 CHECK_EQ(wrap->persistent().IsEmpty(), false);
93
94 bool readable, writable;
95
96 if (status) {
97 readable = writable = false;
98 } else {
99 readable = uv_is_readable(req->handle) != 0;
100 writable = uv_is_writable(req->handle) != 0;
101 }
102
103 Local<Value> argv[5] = {
104 Integer::New(env->isolate(), status),
105 wrap->object(),
106 req_wrap->object(),
107 Boolean::New(env->isolate(), readable),
108 Boolean::New(env->isolate(), writable)
109 };
110
111 req_wrap->MakeCallback(env->oncomplete_string(), arraysize(argv), argv);
112 }
113
114 template ConnectionWrap<PipeWrap, uv_pipe_t>::ConnectionWrap(
115 Environment* env,
116 Local<Object> object,
117 ProviderType provider);
118
119 template ConnectionWrap<TCPWrap, uv_tcp_t>::ConnectionWrap(
120 Environment* env,
121 Local<Object> object,
122 ProviderType provider);
123
124 template void ConnectionWrap<PipeWrap, uv_pipe_t>::OnConnection(
125 uv_stream_t* handle, int status);
126
127 template void ConnectionWrap<TCPWrap, uv_tcp_t>::OnConnection(
128 uv_stream_t* handle, int status);
129
130 template void ConnectionWrap<PipeWrap, uv_pipe_t>::AfterConnect(
131 uv_connect_t* handle, int status);
132
133 template void ConnectionWrap<TCPWrap, uv_tcp_t>::AfterConnect(
134 uv_connect_t* handle, int status);
135
136
137 } // namespace node
138