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> |
23 |
|
37312 |
ConnectionWrap<WrapType, UVType>::ConnectionWrap(Environment* env, |
24 |
|
|
Local<Object> object, |
25 |
|
|
ProviderType provider) |
26 |
|
|
: LibuvStreamWrap(env, |
27 |
|
|
object, |
28 |
|
37312 |
reinterpret_cast<uv_stream_t*>(&handle_), |
29 |
|
37312 |
provider) {} |
30 |
|
|
|
31 |
|
|
|
32 |
|
|
template <typename WrapType, typename UVType> |
33 |
|
10174 |
void ConnectionWrap<WrapType, UVType>::OnConnection(uv_stream_t* handle, |
34 |
|
|
int status) { |
35 |
|
10174 |
WrapType* wrap_data = static_cast<WrapType*>(handle->data); |
36 |
✗✓ |
10174 |
CHECK_NOT_NULL(wrap_data); |
37 |
✗✓ |
10174 |
CHECK_EQ(&wrap_data->handle_, reinterpret_cast<UVType*>(handle)); |
38 |
|
|
|
39 |
|
10174 |
Environment* env = wrap_data->env(); |
40 |
|
10174 |
HandleScope handle_scope(env->isolate()); |
41 |
|
10174 |
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 |
✗✓ |
10174 |
CHECK_EQ(wrap_data->persistent().IsEmpty(), false); |
46 |
|
|
|
47 |
|
|
Local<Value> client_handle; |
48 |
|
|
|
49 |
✓✗ |
10174 |
if (status == 0) { |
50 |
|
|
// Instantiate the client javascript object and handle. |
51 |
|
|
Local<Object> client_obj; |
52 |
✗✓ |
20348 |
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 |
✗✓ |
10174 |
ASSIGN_OR_RETURN_UNWRAP(&wrap, client_obj); |
59 |
|
10174 |
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 |
✗✓ |
10174 |
if (uv_accept(handle, client)) |
64 |
|
|
return; |
65 |
|
|
|
66 |
|
|
// Successful accept. Call the onconnection callback in JavaScript land. |
67 |
|
10174 |
client_handle = client_obj; |
68 |
|
|
} else { |
69 |
|
|
client_handle = Undefined(env->isolate()); |
70 |
|
|
} |
71 |
|
|
|
72 |
|
10174 |
Local<Value> argv[] = { Integer::New(env->isolate(), status), client_handle }; |
73 |
|
10174 |
wrap_data->MakeCallback(env->onconnection_string(), arraysize(argv), argv); |
74 |
|
|
} |
75 |
|
|
|
76 |
|
|
|
77 |
|
|
template <typename WrapType, typename UVType> |
78 |
|
11314 |
void ConnectionWrap<WrapType, UVType>::AfterConnect(uv_connect_t* req, |
79 |
|
|
int status) { |
80 |
|
11314 |
std::unique_ptr<ConnectWrap> req_wrap |
81 |
|
11314 |
(static_cast<ConnectWrap*>(req->data)); |
82 |
✗✓ |
11314 |
CHECK_NOT_NULL(req_wrap); |
83 |
|
11314 |
WrapType* wrap = static_cast<WrapType*>(req->handle->data); |
84 |
✗✓ |
11314 |
CHECK_EQ(req_wrap->env(), wrap->env()); |
85 |
|
11314 |
Environment* env = wrap->env(); |
86 |
|
|
|
87 |
|
22628 |
HandleScope handle_scope(env->isolate()); |
88 |
|
11314 |
Context::Scope context_scope(env->context()); |
89 |
|
|
|
90 |
|
|
// The wrap and request objects should still be there. |
91 |
✗✓ |
11314 |
CHECK_EQ(req_wrap->persistent().IsEmpty(), false); |
92 |
✗✓ |
11314 |
CHECK_EQ(wrap->persistent().IsEmpty(), false); |
93 |
|
|
|
94 |
|
|
bool readable, writable; |
95 |
|
|
|
96 |
✓✓ |
11314 |
if (status) { |
97 |
|
822 |
readable = writable = false; |
98 |
|
|
} else { |
99 |
|
10492 |
readable = uv_is_readable(req->handle) != 0; |
100 |
|
10492 |
writable = uv_is_writable(req->handle) != 0; |
101 |
|
|
} |
102 |
|
|
|
103 |
|
56570 |
Local<Value> argv[5] = { |
104 |
|
|
Integer::New(env->isolate(), status), |
105 |
|
|
wrap->object(), |
106 |
|
11314 |
req_wrap->object(), |
107 |
|
|
Boolean::New(env->isolate(), readable), |
108 |
|
|
Boolean::New(env->isolate(), writable) |
109 |
|
|
}; |
110 |
|
|
|
111 |
|
11314 |
req_wrap->MakeCallback(env->oncomplete_string(), arraysize(argv), argv); |
112 |
|
11314 |
} |
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 |