1 |
|
|
#ifndef SRC_STREAM_BASE_INL_H_ |
2 |
|
|
#define SRC_STREAM_BASE_INL_H_ |
3 |
|
|
|
4 |
|
|
#if defined(NODE_WANT_INTERNALS) && NODE_WANT_INTERNALS |
5 |
|
|
|
6 |
|
|
#include "allocated_buffer-inl.h" |
7 |
|
|
#include "async_wrap-inl.h" |
8 |
|
|
#include "base_object-inl.h" |
9 |
|
|
#include "node.h" |
10 |
|
|
#include "stream_base.h" |
11 |
|
|
#include "v8.h" |
12 |
|
|
|
13 |
|
|
namespace node { |
14 |
|
|
|
15 |
|
17083 |
StreamReq::StreamReq( |
16 |
|
|
StreamBase* stream, |
17 |
|
17083 |
v8::Local<v8::Object> req_wrap_obj) : stream_(stream) { |
18 |
|
17083 |
AttachToObject(req_wrap_obj); |
19 |
|
17083 |
} |
20 |
|
|
|
21 |
|
17083 |
void StreamReq::AttachToObject(v8::Local<v8::Object> req_wrap_obj) { |
22 |
✗✓ |
34166 |
CHECK_EQ(req_wrap_obj->GetAlignedPointerFromInternalField( |
23 |
|
|
StreamReq::kStreamReqField), |
24 |
|
|
nullptr); |
25 |
|
17083 |
req_wrap_obj->SetAlignedPointerInInternalField( |
26 |
|
|
StreamReq::kStreamReqField, this); |
27 |
|
17083 |
} |
28 |
|
|
|
29 |
|
7466 |
StreamReq* StreamReq::FromObject(v8::Local<v8::Object> req_wrap_obj) { |
30 |
|
|
return static_cast<StreamReq*>( |
31 |
|
7466 |
req_wrap_obj->GetAlignedPointerFromInternalField( |
32 |
|
7466 |
StreamReq::kStreamReqField)); |
33 |
|
|
} |
34 |
|
|
|
35 |
|
17071 |
void StreamReq::Dispose() { |
36 |
|
34142 |
BaseObjectPtr<AsyncWrap> destroy_me{GetAsyncWrap()}; |
37 |
|
34142 |
object()->SetAlignedPointerInInternalField( |
38 |
|
|
StreamReq::kStreamReqField, nullptr); |
39 |
|
17071 |
destroy_me->Detach(); |
40 |
|
17071 |
} |
41 |
|
|
|
42 |
|
17691 |
v8::Local<v8::Object> StreamReq::object() { |
43 |
|
17691 |
return GetAsyncWrap()->object(); |
44 |
|
|
} |
45 |
|
|
|
46 |
|
9472 |
ShutdownWrap::ShutdownWrap( |
47 |
|
|
StreamBase* stream, |
48 |
|
9472 |
v8::Local<v8::Object> req_wrap_obj) |
49 |
|
9472 |
: StreamReq(stream, req_wrap_obj) { } |
50 |
|
|
|
51 |
|
7611 |
WriteWrap::WriteWrap( |
52 |
|
|
StreamBase* stream, |
53 |
|
7611 |
v8::Local<v8::Object> req_wrap_obj) |
54 |
|
7611 |
: StreamReq(stream, req_wrap_obj) { } |
55 |
|
|
|
56 |
|
12973 |
void StreamListener::PassReadErrorToPreviousListener(ssize_t nread) { |
57 |
✗✓ |
12973 |
CHECK_NOT_NULL(previous_listener_); |
58 |
|
12973 |
previous_listener_->OnStreamRead(nread, uv_buf_init(nullptr, 0)); |
59 |
|
12973 |
} |
60 |
|
|
|
61 |
|
105874 |
void StreamResource::PushStreamListener(StreamListener* listener) { |
62 |
✗✓ |
105874 |
CHECK_NOT_NULL(listener); |
63 |
✗✓ |
105874 |
CHECK_NULL(listener->stream_); |
64 |
|
|
|
65 |
|
105874 |
listener->previous_listener_ = listener_; |
66 |
|
105874 |
listener->stream_ = this; |
67 |
|
|
|
68 |
|
105874 |
listener_ = listener; |
69 |
|
105874 |
} |
70 |
|
|
|
71 |
|
105202 |
void StreamResource::RemoveStreamListener(StreamListener* listener) { |
72 |
✗✓ |
105202 |
CHECK_NOT_NULL(listener); |
73 |
|
|
|
74 |
|
|
StreamListener* previous; |
75 |
|
|
StreamListener* current; |
76 |
|
|
|
77 |
|
|
// Remove from the linked list. |
78 |
|
105202 |
for (current = listener_, previous = nullptr; |
79 |
|
|
/* No loop condition because we want a crash if listener is not found */ |
80 |
|
618 |
; previous = current, current = current->previous_listener_) { |
81 |
✗✓ |
105820 |
CHECK_NOT_NULL(current); |
82 |
✓✓ |
105820 |
if (current == listener) { |
83 |
✓✓ |
105202 |
if (previous != nullptr) |
84 |
|
618 |
previous->previous_listener_ = current->previous_listener_; |
85 |
|
|
else |
86 |
|
104584 |
listener_ = listener->previous_listener_; |
87 |
|
105202 |
break; |
88 |
|
|
} |
89 |
|
618 |
} |
90 |
|
|
|
91 |
|
105202 |
listener->stream_ = nullptr; |
92 |
|
105202 |
listener->previous_listener_ = nullptr; |
93 |
|
105202 |
} |
94 |
|
|
|
95 |
|
91759 |
uv_buf_t StreamResource::EmitAlloc(size_t suggested_size) { |
96 |
|
91759 |
DebugSealHandleScope seal_handle_scope; |
97 |
|
91759 |
return listener_->OnStreamAlloc(suggested_size); |
98 |
|
|
} |
99 |
|
|
|
100 |
|
106017 |
void StreamResource::EmitRead(ssize_t nread, const uv_buf_t& buf) { |
101 |
|
106017 |
DebugSealHandleScope seal_handle_scope; |
102 |
✓✓ |
106017 |
if (nread > 0) |
103 |
|
85740 |
bytes_read_ += static_cast<uint64_t>(nread); |
104 |
|
106017 |
listener_->OnStreamRead(nread, buf); |
105 |
|
105895 |
} |
106 |
|
|
|
107 |
|
7594 |
void StreamResource::EmitAfterWrite(WriteWrap* w, int status) { |
108 |
|
7594 |
DebugSealHandleScope seal_handle_scope; |
109 |
|
7594 |
listener_->OnStreamAfterWrite(w, status); |
110 |
|
7593 |
} |
111 |
|
|
|
112 |
|
9465 |
void StreamResource::EmitAfterShutdown(ShutdownWrap* w, int status) { |
113 |
|
9465 |
DebugSealHandleScope seal_handle_scope; |
114 |
|
9465 |
listener_->OnStreamAfterShutdown(w, status); |
115 |
|
9465 |
} |
116 |
|
|
|
117 |
|
2660 |
void StreamResource::EmitWantsWrite(size_t suggested_size) { |
118 |
|
2660 |
DebugSealHandleScope seal_handle_scope; |
119 |
|
2660 |
listener_->OnStreamWantsWrite(suggested_size); |
120 |
|
2660 |
} |
121 |
|
|
|
122 |
|
65929 |
StreamBase::StreamBase(Environment* env) : env_(env) { |
123 |
|
65929 |
PushStreamListener(&default_listener_); |
124 |
|
65929 |
} |
125 |
|
|
|
126 |
|
33504 |
int StreamBase::Shutdown(v8::Local<v8::Object> req_wrap_obj) { |
127 |
|
33504 |
Environment* env = stream_env(); |
128 |
|
|
|
129 |
|
67008 |
v8::HandleScope handle_scope(env->isolate()); |
130 |
|
|
|
131 |
✓✓ |
33504 |
if (req_wrap_obj.IsEmpty()) { |
132 |
|
599 |
if (!env->shutdown_wrap_template() |
133 |
|
599 |
->NewInstance(env->context()) |
134 |
✗✓ |
599 |
.ToLocal(&req_wrap_obj)) { |
135 |
|
|
return UV_EBUSY; |
136 |
|
|
} |
137 |
|
599 |
StreamReq::ResetObject(req_wrap_obj); |
138 |
|
|
} |
139 |
|
|
|
140 |
|
67008 |
BaseObjectPtr<AsyncWrap> req_wrap_ptr; |
141 |
|
33504 |
AsyncHooks::DefaultTriggerAsyncIdScope trigger_scope(GetAsyncWrap()); |
142 |
|
33504 |
ShutdownWrap* req_wrap = CreateShutdownWrap(req_wrap_obj); |
143 |
✓✓ |
33504 |
if (req_wrap != nullptr) |
144 |
|
9472 |
req_wrap_ptr.reset(req_wrap->GetAsyncWrap()); |
145 |
|
33504 |
int err = DoShutdown(req_wrap); |
146 |
|
|
|
147 |
✓✓✓✓
|
33504 |
if (err != 0 && req_wrap != nullptr) { |
148 |
|
5 |
req_wrap->Dispose(); |
149 |
|
|
} |
150 |
|
|
|
151 |
|
33504 |
const char* msg = Error(); |
152 |
✗✓ |
33504 |
if (msg != nullptr) { |
153 |
|
|
req_wrap_obj->Set( |
154 |
|
|
env->context(), |
155 |
|
|
env->error_string(), OneByteString(env->isolate(), msg)).Check(); |
156 |
|
|
ClearError(); |
157 |
|
|
} |
158 |
|
|
|
159 |
|
33504 |
return err; |
160 |
|
|
} |
161 |
|
|
|
162 |
|
100975 |
StreamWriteResult StreamBase::Write( |
163 |
|
|
uv_buf_t* bufs, |
164 |
|
|
size_t count, |
165 |
|
|
uv_stream_t* send_handle, |
166 |
|
|
v8::Local<v8::Object> req_wrap_obj) { |
167 |
|
100975 |
Environment* env = stream_env(); |
168 |
|
|
int err; |
169 |
|
|
|
170 |
|
100975 |
size_t total_bytes = 0; |
171 |
✓✓ |
349013 |
for (size_t i = 0; i < count; ++i) |
172 |
|
248038 |
total_bytes += bufs[i].len; |
173 |
|
100975 |
bytes_written_ += total_bytes; |
174 |
|
|
|
175 |
✓✓ |
100975 |
if (send_handle == nullptr) { |
176 |
|
100869 |
err = DoTryWrite(&bufs, &count); |
177 |
✓✓✓✓
|
100869 |
if (err != 0 || count == 0) { |
178 |
|
93364 |
return StreamWriteResult { false, err, nullptr, total_bytes, {} }; |
179 |
|
|
} |
180 |
|
|
} |
181 |
|
|
|
182 |
|
15221 |
v8::HandleScope handle_scope(env->isolate()); |
183 |
|
|
|
184 |
✓✓ |
7611 |
if (req_wrap_obj.IsEmpty()) { |
185 |
|
2281 |
if (!env->write_wrap_template() |
186 |
|
2281 |
->NewInstance(env->context()) |
187 |
✗✓ |
2281 |
.ToLocal(&req_wrap_obj)) { |
188 |
|
|
return StreamWriteResult { false, UV_EBUSY, nullptr, 0, {} }; |
189 |
|
|
} |
190 |
|
2281 |
StreamReq::ResetObject(req_wrap_obj); |
191 |
|
|
} |
192 |
|
|
|
193 |
|
15221 |
AsyncHooks::DefaultTriggerAsyncIdScope trigger_scope(GetAsyncWrap()); |
194 |
|
7611 |
WriteWrap* req_wrap = CreateWriteWrap(req_wrap_obj); |
195 |
|
15221 |
BaseObjectPtr<AsyncWrap> req_wrap_ptr(req_wrap->GetAsyncWrap()); |
196 |
|
|
|
197 |
|
7611 |
err = DoWrite(req_wrap, bufs, count, send_handle); |
198 |
|
7610 |
bool async = err == 0; |
199 |
|
|
|
200 |
✓✓ |
7610 |
if (!async) { |
201 |
|
8 |
req_wrap->Dispose(); |
202 |
|
8 |
req_wrap = nullptr; |
203 |
|
|
} |
204 |
|
|
|
205 |
|
7610 |
const char* msg = Error(); |
206 |
✓✓ |
7610 |
if (msg != nullptr) { |
207 |
|
6 |
req_wrap_obj->Set(env->context(), |
208 |
|
|
env->error_string(), |
209 |
|
18 |
OneByteString(env->isolate(), msg)).Check(); |
210 |
|
6 |
ClearError(); |
211 |
|
|
} |
212 |
|
|
|
213 |
|
|
return StreamWriteResult { |
214 |
|
7610 |
async, err, req_wrap, total_bytes, std::move(req_wrap_ptr) }; |
215 |
|
|
} |
216 |
|
|
|
217 |
|
|
template <typename OtherBase> |
218 |
|
9472 |
SimpleShutdownWrap<OtherBase>::SimpleShutdownWrap( |
219 |
|
|
StreamBase* stream, |
220 |
|
|
v8::Local<v8::Object> req_wrap_obj) |
221 |
|
|
: ShutdownWrap(stream, req_wrap_obj), |
222 |
|
|
OtherBase(stream->stream_env(), |
223 |
|
|
req_wrap_obj, |
224 |
|
9472 |
AsyncWrap::PROVIDER_SHUTDOWNWRAP) { |
225 |
|
9472 |
} |
226 |
|
|
|
227 |
|
|
template <typename OtherBase> |
228 |
|
7611 |
SimpleWriteWrap<OtherBase>::SimpleWriteWrap( |
229 |
|
|
StreamBase* stream, |
230 |
|
|
v8::Local<v8::Object> req_wrap_obj) |
231 |
|
|
: WriteWrap(stream, req_wrap_obj), |
232 |
|
|
OtherBase(stream->stream_env(), |
233 |
|
|
req_wrap_obj, |
234 |
|
7611 |
AsyncWrap::PROVIDER_WRITEWRAP) { |
235 |
|
7611 |
} |
236 |
|
|
|
237 |
|
65929 |
void StreamBase::AttachToObject(v8::Local<v8::Object> obj) { |
238 |
|
65929 |
obj->SetAlignedPointerInInternalField( |
239 |
|
|
StreamBase::kStreamBaseField, this); |
240 |
|
65929 |
} |
241 |
|
|
|
242 |
|
295047 |
StreamBase* StreamBase::FromObject(v8::Local<v8::Object> obj) { |
243 |
✓✓ |
590094 |
if (obj->GetAlignedPointerFromInternalField(StreamBase::kSlot) == nullptr) |
244 |
|
1 |
return nullptr; |
245 |
|
|
|
246 |
|
|
return static_cast<StreamBase*>( |
247 |
|
295046 |
obj->GetAlignedPointerFromInternalField( |
248 |
|
295046 |
StreamBase::kStreamBaseField)); |
249 |
|
|
} |
250 |
|
|
|
251 |
|
6857 |
WriteWrap* WriteWrap::FromObject(v8::Local<v8::Object> req_wrap_obj) { |
252 |
|
6857 |
return static_cast<WriteWrap*>(StreamReq::FromObject(req_wrap_obj)); |
253 |
|
|
} |
254 |
|
|
|
255 |
|
|
template <typename T, bool kIsWeak> |
256 |
|
6857 |
WriteWrap* WriteWrap::FromObject( |
257 |
|
|
const BaseObjectPtrImpl<T, kIsWeak>& base_obj) { |
258 |
✗✓ |
6857 |
if (!base_obj) return nullptr; |
259 |
|
6857 |
return FromObject(base_obj->object()); |
260 |
|
|
} |
261 |
|
|
|
262 |
|
|
ShutdownWrap* ShutdownWrap::FromObject(v8::Local<v8::Object> req_wrap_obj) { |
263 |
|
|
return static_cast<ShutdownWrap*>(StreamReq::FromObject(req_wrap_obj)); |
264 |
|
|
} |
265 |
|
|
|
266 |
|
|
template <typename T, bool kIsWeak> |
267 |
|
|
ShutdownWrap* ShutdownWrap::FromObject( |
268 |
|
|
const BaseObjectPtrImpl<T, kIsWeak>& base_obj) { |
269 |
|
|
if (!base_obj) return nullptr; |
270 |
|
|
return FromObject(base_obj->object()); |
271 |
|
|
} |
272 |
|
|
|
273 |
|
3423 |
void WriteWrap::SetAllocatedStorage(AllocatedBuffer&& storage) { |
274 |
✗✓ |
3423 |
CHECK_NULL(storage_.data()); |
275 |
|
3423 |
storage_ = std::move(storage); |
276 |
|
3423 |
} |
277 |
|
|
|
278 |
|
17059 |
void StreamReq::Done(int status, const char* error_str) { |
279 |
|
17059 |
AsyncWrap* async_wrap = GetAsyncWrap(); |
280 |
|
17059 |
Environment* env = async_wrap->env(); |
281 |
✓✓ |
17059 |
if (error_str != nullptr) { |
282 |
|
4 |
v8::HandleScope handle_scope(env->isolate()); |
283 |
|
4 |
async_wrap->object()->Set(env->context(), |
284 |
|
|
env->error_string(), |
285 |
|
16 |
OneByteString(env->isolate(), error_str)) |
286 |
|
|
.Check(); |
287 |
|
|
} |
288 |
|
|
|
289 |
|
17059 |
OnDone(status); |
290 |
|
17058 |
} |
291 |
|
|
|
292 |
|
121444 |
void StreamReq::ResetObject(v8::Local<v8::Object> obj) { |
293 |
|
|
DCHECK_GT(obj->InternalFieldCount(), StreamReq::kStreamReqField); |
294 |
|
|
|
295 |
|
121444 |
obj->SetAlignedPointerInInternalField(StreamReq::kSlot, nullptr); |
296 |
|
121444 |
obj->SetAlignedPointerInInternalField(StreamReq::kStreamReqField, nullptr); |
297 |
|
121444 |
} |
298 |
|
|
|
299 |
|
|
} // namespace node |
300 |
|
|
|
301 |
|
|
#endif // defined(NODE_WANT_INTERNALS) && NODE_WANT_INTERNALS |
302 |
|
|
|
303 |
|
|
#endif // SRC_STREAM_BASE_INL_H_ |