Head

GCC Code Coverage Report

Directory:	./		Exec	Total	Coverage
File:	stream_wrap.cc	Lines:	180	186	96.8 %
Date:	2022-08-16 04:20:39	Branches:	51	76	67.1 %


// 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.

#include "stream_wrap.h"
#include "stream_base-inl.h"

#include "env-inl.h"
#include "handle_wrap.h"
#include "node_buffer.h"
#include "node_errors.h"
#include "node_external_reference.h"
#include "pipe_wrap.h"
#include "req_wrap-inl.h"
#include "tcp_wrap.h"
#include "udp_wrap.h"
#include "util-inl.h"

#include <cstring>  // memcpy()
#include <climits>  // INT_MAX


namespace node {

using errors::TryCatchScope;
using v8::Context;
using v8::DontDelete;
using v8::EscapableHandleScope;
using v8::FunctionCallbackInfo;
using v8::FunctionTemplate;
using v8::HandleScope;
using v8::Isolate;
using v8::JustVoid;
using v8::Local;
using v8::Maybe;
using v8::MaybeLocal;
using v8::Nothing;
using v8::Object;
using v8::PropertyAttribute;
using v8::ReadOnly;
using v8::Signature;
using v8::Value;

void IsConstructCallCallback(const FunctionCallbackInfo<Value>& args) {
  CHECK(args.IsConstructCall());
  StreamReq::ResetObject(args.This());
}

void LibuvStreamWrap::Initialize(Local<Object> target,
                                 Local<Value> unused,
                                 Local<Context> context,
                                 void* priv) {
  Environment* env = Environment::GetCurrent(context);
  Isolate* isolate = env->isolate();

  Local<FunctionTemplate> sw =
      NewFunctionTemplate(isolate, IsConstructCallCallback);
  sw->InstanceTemplate()->SetInternalFieldCount(StreamReq::kInternalFieldCount);

  // we need to set handle and callback to null,
  // so that those fields are created and functions
  // do not become megamorphic
  // Fields:
  // - oncomplete
  // - callback
  // - handle
  sw->InstanceTemplate()->Set(env->oncomplete_string(), v8::Null(isolate));
  sw->InstanceTemplate()->Set(FIXED_ONE_BYTE_STRING(isolate, "callback"),
                              v8::Null(isolate));
  sw->InstanceTemplate()->Set(FIXED_ONE_BYTE_STRING(isolate, "handle"),
                              v8::Null(isolate));

  sw->Inherit(AsyncWrap::GetConstructorTemplate(env));

  SetConstructorFunction(context, target, "ShutdownWrap", sw);
  env->set_shutdown_wrap_template(sw->InstanceTemplate());

  Local<FunctionTemplate> ww =
      FunctionTemplate::New(isolate, IsConstructCallCallback);
  ww->InstanceTemplate()->SetInternalFieldCount(
      StreamReq::kInternalFieldCount);
  ww->Inherit(AsyncWrap::GetConstructorTemplate(env));
  SetConstructorFunction(context, target, "WriteWrap", ww);
  env->set_write_wrap_template(ww->InstanceTemplate());

  NODE_DEFINE_CONSTANT(target, kReadBytesOrError);
  NODE_DEFINE_CONSTANT(target, kArrayBufferOffset);
  NODE_DEFINE_CONSTANT(target, kBytesWritten);
  NODE_DEFINE_CONSTANT(target, kLastWriteWasAsync);
  target
      ->Set(context,
            FIXED_ONE_BYTE_STRING(isolate, "streamBaseState"),
            env->stream_base_state().GetJSArray())
      .Check();
}

void LibuvStreamWrap::RegisterExternalReferences(
    ExternalReferenceRegistry* registry) {
  registry->Register(IsConstructCallCallback);
  registry->Register(GetWriteQueueSize);
  registry->Register(SetBlocking);
  StreamBase::RegisterExternalReferences(registry);
}

LibuvStreamWrap::LibuvStreamWrap(Environment* env,
                                 Local<Object> object,
                                 uv_stream_t* stream,
                                 AsyncWrap::ProviderType provider)
    : HandleWrap(env,
                 object,
                 reinterpret_cast<uv_handle_t*>(stream),
                 provider),
      StreamBase(env),
      stream_(stream) {
  StreamBase::AttachToObject(object);
}


Local<FunctionTemplate> LibuvStreamWrap::GetConstructorTemplate(
    Environment* env) {
  Local<FunctionTemplate> tmpl = env->libuv_stream_wrap_ctor_template();
  if (tmpl.IsEmpty()) {
    Isolate* isolate = env->isolate();
    tmpl = NewFunctionTemplate(isolate, nullptr);
    tmpl->SetClassName(FIXED_ONE_BYTE_STRING(isolate, "LibuvStreamWrap"));
    tmpl->Inherit(HandleWrap::GetConstructorTemplate(env));
    tmpl->InstanceTemplate()->SetInternalFieldCount(
        StreamBase::kInternalFieldCount);
    Local<FunctionTemplate> get_write_queue_size =
        FunctionTemplate::New(isolate,
                              GetWriteQueueSize,
                              Local<Value>(),
                              Signature::New(isolate, tmpl));
    tmpl->PrototypeTemplate()->SetAccessorProperty(
        env->write_queue_size_string(),
        get_write_queue_size,
        Local<FunctionTemplate>(),
        static_cast<PropertyAttribute>(ReadOnly | DontDelete));
    SetProtoMethod(isolate, tmpl, "setBlocking", SetBlocking);
    StreamBase::AddMethods(env, tmpl);
    env->set_libuv_stream_wrap_ctor_template(tmpl);
  }
  return tmpl;
}


LibuvStreamWrap* LibuvStreamWrap::From(Environment* env, Local<Object> object) {
  Local<FunctionTemplate> sw = env->libuv_stream_wrap_ctor_template();
  CHECK(!sw.IsEmpty() && sw->HasInstance(object));
  return Unwrap<LibuvStreamWrap>(object);
}


int LibuvStreamWrap::GetFD() {
#ifdef _WIN32
  return fd_;
#else
  int fd = -1;
  if (stream() != nullptr)
    uv_fileno(reinterpret_cast<uv_handle_t*>(stream()), &fd);
  return fd;
#endif
}


bool LibuvStreamWrap::IsAlive() {
  return HandleWrap::IsAlive(this);
}


bool LibuvStreamWrap::IsClosing() {
  return uv_is_closing(reinterpret_cast<uv_handle_t*>(stream()));
}


AsyncWrap* LibuvStreamWrap::GetAsyncWrap() {
  return static_cast<AsyncWrap*>(this);
}


bool LibuvStreamWrap::IsIPCPipe() {
  return is_named_pipe_ipc();
}

int LibuvStreamWrap::ReadStart() {
  return uv_read_start(
      stream(),
      [](uv_handle_t* handle, size_t suggested_size, uv_buf_t* buf) {
        static_cast<LibuvStreamWrap*>(handle->data)
            ->OnUvAlloc(suggested_size, buf);
      },
      [](uv_stream_t* stream, ssize_t nread, const uv_buf_t* buf) {
        LibuvStreamWrap* wrap = static_cast<LibuvStreamWrap*>(stream->data);
        TryCatchScope try_catch(wrap->env());
        try_catch.SetVerbose(true);
        wrap->OnUvRead(nread, buf);
      });
}


int LibuvStreamWrap::ReadStop() {
  return uv_read_stop(stream());
}


void LibuvStreamWrap::OnUvAlloc(size_t suggested_size, uv_buf_t* buf) {
  HandleScope scope(env()->isolate());
  Context::Scope context_scope(env()->context());

  *buf = EmitAlloc(suggested_size);
}

template <class WrapType>
static MaybeLocal<Object> AcceptHandle(Environment* env,
                                       LibuvStreamWrap* parent) {
  static_assert(std::is_base_of<LibuvStreamWrap, WrapType>::value ||
                std::is_base_of<UDPWrap, WrapType>::value,
                "Can only accept stream handles");

  EscapableHandleScope scope(env->isolate());
  Local<Object> wrap_obj;

  if (!WrapType::Instantiate(env, parent, WrapType::SOCKET).ToLocal(&wrap_obj))
    return Local<Object>();

  HandleWrap* wrap = Unwrap<HandleWrap>(wrap_obj);
  CHECK_NOT_NULL(wrap);
  uv_stream_t* stream = reinterpret_cast<uv_stream_t*>(wrap->GetHandle());
  CHECK_NOT_NULL(stream);

  if (uv_accept(parent->stream(), stream))
    ABORT();

  return scope.Escape(wrap_obj);
}

Maybe<void> LibuvStreamWrap::OnUvRead(ssize_t nread, const uv_buf_t* buf) {
  HandleScope scope(env()->isolate());
  Context::Scope context_scope(env()->context());
  uv_handle_type type = UV_UNKNOWN_HANDLE;

  if (is_named_pipe_ipc() &&
      uv_pipe_pending_count(reinterpret_cast<uv_pipe_t*>(stream())) > 0) {
    type = uv_pipe_pending_type(reinterpret_cast<uv_pipe_t*>(stream()));
  }

  // We should not be getting this callback if someone has already called
  // uv_close() on the handle.
  CHECK_EQ(persistent().IsEmpty(), false);

  if (nread > 0) {
    MaybeLocal<Object> pending_obj;

    if (type == UV_TCP) {
      pending_obj = AcceptHandle<TCPWrap>(env(), this);
    } else if (type == UV_NAMED_PIPE) {
      pending_obj = AcceptHandle<PipeWrap>(env(), this);
    } else if (type == UV_UDP) {
      pending_obj = AcceptHandle<UDPWrap>(env(), this);
    } else {
      CHECK_EQ(type, UV_UNKNOWN_HANDLE);
    }

    Local<Object> local_pending_obj;
    if (type != UV_UNKNOWN_HANDLE &&
        (!pending_obj.ToLocal(&local_pending_obj) ||
         object()
             ->Set(env()->context(),
                   env()->pending_handle_string(),
                   local_pending_obj)
             .IsNothing())) {
      return Nothing<void>();
    }
  }

  EmitRead(nread, *buf);
  return JustVoid();
}

void LibuvStreamWrap::GetWriteQueueSize(
    const FunctionCallbackInfo<Value>& info) {
  LibuvStreamWrap* wrap;
  ASSIGN_OR_RETURN_UNWRAP(&wrap, info.This());

  if (wrap->stream() == nullptr) {
    info.GetReturnValue().Set(0);
    return;
  }

  uint32_t write_queue_size = wrap->stream()->write_queue_size;
  info.GetReturnValue().Set(write_queue_size);
}


void LibuvStreamWrap::SetBlocking(const FunctionCallbackInfo<Value>& args) {
  LibuvStreamWrap* wrap;
  ASSIGN_OR_RETURN_UNWRAP(&wrap, args.Holder());

  CHECK_GT(args.Length(), 0);
  if (!wrap->IsAlive())
    return args.GetReturnValue().Set(UV_EINVAL);

  bool enable = args[0]->IsTrue();
  args.GetReturnValue().Set(uv_stream_set_blocking(wrap->stream(), enable));
}

typedef SimpleShutdownWrap<ReqWrap<uv_shutdown_t>> LibuvShutdownWrap;
typedef SimpleWriteWrap<ReqWrap<uv_write_t>> LibuvWriteWrap;

ShutdownWrap* LibuvStreamWrap::CreateShutdownWrap(Local<Object> object) {
  return new LibuvShutdownWrap(this, object);
}

WriteWrap* LibuvStreamWrap::CreateWriteWrap(Local<Object> object) {
  return new LibuvWriteWrap(this, object);
}


int LibuvStreamWrap::DoShutdown(ShutdownWrap* req_wrap_) {
  LibuvShutdownWrap* req_wrap = static_cast<LibuvShutdownWrap*>(req_wrap_);
  return req_wrap->Dispatch(uv_shutdown, stream(), AfterUvShutdown);
}


void LibuvStreamWrap::AfterUvShutdown(uv_shutdown_t* req, int status) {
  LibuvShutdownWrap* req_wrap = static_cast<LibuvShutdownWrap*>(
      LibuvShutdownWrap::from_req(req));
  CHECK_NOT_NULL(req_wrap);
  HandleScope scope(req_wrap->env()->isolate());
  Context::Scope context_scope(req_wrap->env()->context());
  req_wrap->Done(status);
}


// NOTE: Call to this function could change both `buf`'s and `count`'s
// values, shifting their base and decrementing their length. This is
// required in order to skip the data that was successfully written via
// uv_try_write().
int LibuvStreamWrap::DoTryWrite(uv_buf_t** bufs, size_t* count) {
  int err;
  size_t written;
  uv_buf_t* vbufs = *bufs;
  size_t vcount = *count;

  err = uv_try_write(stream(), vbufs, vcount);
  if (err == UV_ENOSYS || err == UV_EAGAIN)
    return 0;
  if (err < 0)
    return err;

  // Slice off the buffers: skip all written buffers and slice the one that
  // was partially written.
  written = err;
  for (; vcount > 0; vbufs++, vcount--) {
    // Slice
    if (vbufs[0].len > written) {
      vbufs[0].base += written;
      vbufs[0].len -= written;
      written = 0;
      break;

    // Discard
    } else {
      written -= vbufs[0].len;
    }
  }

  *bufs = vbufs;
  *count = vcount;

  return 0;
}


int LibuvStreamWrap::DoWrite(WriteWrap* req_wrap,
                             uv_buf_t* bufs,
                             size_t count,
                             uv_stream_t* send_handle) {
  LibuvWriteWrap* w = static_cast<LibuvWriteWrap*>(req_wrap);
  return w->Dispatch(uv_write2,
                     stream(),
                     bufs,
                     count,
                     send_handle,
                     AfterUvWrite);
}



void LibuvStreamWrap::AfterUvWrite(uv_write_t* req, int status) {
  LibuvWriteWrap* req_wrap = static_cast<LibuvWriteWrap*>(
      LibuvWriteWrap::from_req(req));
  CHECK_NOT_NULL(req_wrap);
  HandleScope scope(req_wrap->env()->isolate());
  Context::Scope context_scope(req_wrap->env()->context());
  req_wrap->Done(status);
}

}  // namespace node

NODE_MODULE_CONTEXT_AWARE_INTERNAL(stream_wrap,
                                   node::LibuvStreamWrap::Initialize)
NODE_MODULE_EXTERNAL_REFERENCE(
    stream_wrap, node::LibuvStreamWrap::RegisterExternalReferences)


Generated by: GCOVR (Version 4.2)

Line	Branch	Exec	Source
1			// Copyright Joyent, Inc. and other Node contributors.
2			//
3			// Permission is hereby granted, free of charge, to any person obtaining a
4			// copy of this software and associated documentation files (the
5			// "Software"), to deal in the Software without restriction, including
6			// without limitation the rights to use, copy, modify, merge, publish,
7			// distribute, sublicense, and/or sell copies of the Software, and to permit
8			// persons to whom the Software is furnished to do so, subject to the
9			// following conditions:
10			//
11			// The above copyright notice and this permission notice shall be included
12			// in all copies or substantial portions of the Software.
13			//
14			// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
15			// OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
16			// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN
17			// NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
18			// DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
19			// OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
20			// USE OR OTHER DEALINGS IN THE SOFTWARE.
21
22			#include "stream_wrap.h"
23			#include "stream_base-inl.h"
24
25			#include "env-inl.h"
26			#include "handle_wrap.h"
27			#include "node_buffer.h"
28			#include "node_errors.h"
29			#include "node_external_reference.h"
30			#include "pipe_wrap.h"
31			#include "req_wrap-inl.h"
32			#include "tcp_wrap.h"
33			#include "udp_wrap.h"
34			#include "util-inl.h"
35
36			#include <cstring> // memcpy()
37			#include <climits> // INT_MAX
38
39
40			namespace node {
41
42			using errors::TryCatchScope;
43			using v8::Context;
44			using v8::DontDelete;
45			using v8::EscapableHandleScope;
46			using v8::FunctionCallbackInfo;
47			using v8::FunctionTemplate;
48			using v8::HandleScope;
49			using v8::Isolate;
50			using v8::JustVoid;
51			using v8::Local;
52			using v8::Maybe;
53			using v8::MaybeLocal;
54			using v8::Nothing;
55			using v8::Object;
56			using v8::PropertyAttribute;
57			using v8::ReadOnly;
58			using v8::Signature;
59			using v8::Value;
60
61		120533	void IsConstructCallCallback(const FunctionCallbackInfo<Value>& args) {
62	✗✓	120533	CHECK(args.IsConstructCall());
63		120533	StreamReq::ResetObject(args.This());
64		120533	}
65
66		770	void LibuvStreamWrap::Initialize(Local<Object> target,
67			Local<Value> unused,
68			Local<Context> context,
69			void* priv) {
70		770	Environment* env = Environment::GetCurrent(context);
71		770	Isolate* isolate = env->isolate();
72
73			Local<FunctionTemplate> sw =
74		770	NewFunctionTemplate(isolate, IsConstructCallCallback);
75		1540	sw->InstanceTemplate()->SetInternalFieldCount(StreamReq::kInternalFieldCount);
76
77			// we need to set handle and callback to null,
78			// so that those fields are created and functions
79			// do not become megamorphic
80			// Fields:
81			// - oncomplete
82			// - callback
83			// - handle
84		3080	sw->InstanceTemplate()->Set(env->oncomplete_string(), v8::Null(isolate));
85		3080	sw->InstanceTemplate()->Set(FIXED_ONE_BYTE_STRING(isolate, "callback"),
86			v8::Null(isolate));
87		3080	sw->InstanceTemplate()->Set(FIXED_ONE_BYTE_STRING(isolate, "handle"),
88			v8::Null(isolate));
89
90		770	sw->Inherit(AsyncWrap::GetConstructorTemplate(env));
91
92		770	SetConstructorFunction(context, target, "ShutdownWrap", sw);
93		770	env->set_shutdown_wrap_template(sw->InstanceTemplate());
94
95			Local<FunctionTemplate> ww =
96		770	FunctionTemplate::New(isolate, IsConstructCallCallback);
97		1540	ww->InstanceTemplate()->SetInternalFieldCount(
98			StreamReq::kInternalFieldCount);
99		770	ww->Inherit(AsyncWrap::GetConstructorTemplate(env));
100		770	SetConstructorFunction(context, target, "WriteWrap", ww);
101		770	env->set_write_wrap_template(ww->InstanceTemplate());
102
103		2310	NODE_DEFINE_CONSTANT(target, kReadBytesOrError);
104		2310	NODE_DEFINE_CONSTANT(target, kArrayBufferOffset);
105		2310	NODE_DEFINE_CONSTANT(target, kBytesWritten);
106		2310	NODE_DEFINE_CONSTANT(target, kLastWriteWasAsync);
107			target
108		770	->Set(context,
109			FIXED_ONE_BYTE_STRING(isolate, "streamBaseState"),
110		2310	env->stream_base_state().GetJSArray())
111			.Check();
112		770	}
113
114		5338	void LibuvStreamWrap::RegisterExternalReferences(
115			ExternalReferenceRegistry* registry) {
116		5338	registry->Register(IsConstructCallCallback);
117		5338	registry->Register(GetWriteQueueSize);
118		5338	registry->Register(SetBlocking);
119		5338	StreamBase::RegisterExternalReferences(registry);
120		5338	}
121
122		19142	LibuvStreamWrap::LibuvStreamWrap(Environment* env,
123			Local<Object> object,
124			uv_stream_t* stream,
125		19142	AsyncWrap::ProviderType provider)
126			: HandleWrap(env,
127			object,
128			reinterpret_cast<uv_handle_t*>(stream),
129			provider),
130			StreamBase(env),
131		19142	stream_(stream) {
132		19142	StreamBase::AttachToObject(object);
133		19142	}
134
135
136		16878	Local<FunctionTemplate> LibuvStreamWrap::GetConstructorTemplate(
137			Environment* env) {
138		16878	Local<FunctionTemplate> tmpl = env->libuv_stream_wrap_ctor_template();
139	✓✓	16878	if (tmpl.IsEmpty()) {
140		6065	Isolate* isolate = env->isolate();
141		6065	tmpl = NewFunctionTemplate(isolate, nullptr);
142		6065	tmpl->SetClassName(FIXED_ONE_BYTE_STRING(isolate, "LibuvStreamWrap"));
143		6065	tmpl->Inherit(HandleWrap::GetConstructorTemplate(env));
144		12130	tmpl->InstanceTemplate()->SetInternalFieldCount(
145			StreamBase::kInternalFieldCount);
146			Local<FunctionTemplate> get_write_queue_size =
147			FunctionTemplate::New(isolate,
148			GetWriteQueueSize,
149			Local<Value>(),
150		6065	Signature::New(isolate, tmpl));
151		24260	tmpl->PrototypeTemplate()->SetAccessorProperty(
152			env->write_queue_size_string(),
153			get_write_queue_size,
154			Local<FunctionTemplate>(),
155			static_cast<PropertyAttribute>(ReadOnly \| DontDelete));
156		6065	SetProtoMethod(isolate, tmpl, "setBlocking", SetBlocking);
157		6065	StreamBase::AddMethods(env, tmpl);
158		6065	env->set_libuv_stream_wrap_ctor_template(tmpl);
159			}
160		16878	return tmpl;
161			}
162
163
164		4180	LibuvStreamWrap* LibuvStreamWrap::From(Environment* env, Local<Object> object) {
165		4180	Local<FunctionTemplate> sw = env->libuv_stream_wrap_ctor_template();
166	✓✗✗✓ ✗✓	8360	CHECK(!sw.IsEmpty() && sw->HasInstance(object));
167		4180	return Unwrap<LibuvStreamWrap>(object);
168			}
169
170
171		118	int LibuvStreamWrap::GetFD() {
172			#ifdef _WIN32
173			return fd_;
174			#else
175		118	int fd = -1;
176	✓✗	118	if (stream() != nullptr)
177		118	uv_fileno(reinterpret_cast<uv_handle_t*>(stream()), &fd);
178		118	return fd;
179			#endif
180			}
181
182
183		120870	bool LibuvStreamWrap::IsAlive() {
184		120870	return HandleWrap::IsAlive(this);
185			}
186
187
188		3	bool LibuvStreamWrap::IsClosing() {
189		3	return uv_is_closing(reinterpret_cast<uv_handle_t*>(stream()));
190			}
191
192
193		164181	AsyncWrap* LibuvStreamWrap::GetAsyncWrap() {
194		164181	return static_cast<AsyncWrap*>(this);
195			}
196
197
198		27532	bool LibuvStreamWrap::IsIPCPipe() {
199		27532	return is_named_pipe_ipc();
200			}
201
202		22066	int LibuvStreamWrap::ReadStart() {
203		44132	return uv_read_start(
204			stream(),
205		72787	[](uv_handle_t* handle, size_t suggested_size, uv_buf_t* buf) {
206		72787	static_cast<LibuvStreamWrap*>(handle->data)
207		72787	->OnUvAlloc(suggested_size, buf);
208		72787	},
209		73023	[](uv_stream_t* stream, ssize_t nread, const uv_buf_t* buf) {
210		73023	LibuvStreamWrap* wrap = static_cast<LibuvStreamWrap*>(stream->data);
211		145838	TryCatchScope try_catch(wrap->env());
212		73023	try_catch.SetVerbose(true);
213		73023	wrap->OnUvRead(nread, buf);
214		22066	});
215			}
216
217
218		9283	int LibuvStreamWrap::ReadStop() {
219		9283	return uv_read_stop(stream());
220			}
221
222
223		72787	void LibuvStreamWrap::OnUvAlloc(size_t suggested_size, uv_buf_t* buf) {
224		145574	HandleScope scope(env()->isolate());
225		72787	Context::Scope context_scope(env()->context());
226
227		72787	*buf = EmitAlloc(suggested_size);
228		72787	}
229
230			template <class WrapType>
231		374	static MaybeLocal<Object> AcceptHandle(Environment* env,
232			LibuvStreamWrap* parent) {
233			static_assert(std::is_base_of<LibuvStreamWrap, WrapType>::value \|\|
234			std::is_base_of<UDPWrap, WrapType>::value,
235			"Can only accept stream handles");
236
237		374	EscapableHandleScope scope(env->isolate());
238			Local<Object> wrap_obj;
239
240	✗✓	748	if (!WrapType::Instantiate(env, parent, WrapType::SOCKET).ToLocal(&wrap_obj))
241			return Local<Object>();
242
243		374	HandleWrap* wrap = Unwrap<HandleWrap>(wrap_obj);
244	✗✓	374	CHECK_NOT_NULL(wrap);
245		374	uv_stream_t* stream = reinterpret_cast<uv_stream_t*>(wrap->GetHandle());
246	✗✓	374	CHECK_NOT_NULL(stream);
247
248	✗✓	374	if (uv_accept(parent->stream(), stream))
249			ABORT();
250
251		374	return scope.Escape(wrap_obj);
252			}
253
254		73023	Maybe<void> LibuvStreamWrap::OnUvRead(ssize_t nread, const uv_buf_t* buf) {
255		145838	HandleScope scope(env()->isolate());
256		73023	Context::Scope context_scope(env()->context());
257		73023	uv_handle_type type = UV_UNKNOWN_HANDLE;
258
259	✓✓✓✓ ✓✓	77288	if (is_named_pipe_ipc() &&
260		4265	uv_pipe_pending_count(reinterpret_cast<uv_pipe_t*>(stream())) > 0) {
261		187	type = uv_pipe_pending_type(reinterpret_cast<uv_pipe_t*>(stream()));
262			}
263
264			// We should not be getting this callback if someone has already called
265			// uv_close() on the handle.
266	✗✓	73023	CHECK_EQ(persistent().IsEmpty(), false);
267
268	✓✓	73023	if (nread > 0) {
269			MaybeLocal<Object> pending_obj;
270
271	✓✓	66933	if (type == UV_TCP) {
272		130	pending_obj = AcceptHandle<TCPWrap>(env(), this);
273	✓✓	66803	} else if (type == UV_NAMED_PIPE) {
274		24	pending_obj = AcceptHandle<PipeWrap>(env(), this);
275	✓✓	66779	} else if (type == UV_UDP) {
276		33	pending_obj = AcceptHandle<UDPWrap>(env(), this);
277			} else {
278	✗✓	66746	CHECK_EQ(type, UV_UNKNOWN_HANDLE);
279			}
280
281			Local<Object> local_pending_obj;
282	✓✓	67120	if (type != UV_UNKNOWN_HANDLE &&
283	✓✗	187	(!pending_obj.ToLocal(&local_pending_obj) \|\|
284		187	object()
285		66933	->Set(env()->context(),
286			env()->pending_handle_string(),
287	✗✓	67307	local_pending_obj)
288	✗✓	187	.IsNothing())) {
289			return Nothing<void>();
290			}
291			}
292
293		73023	EmitRead(nread, *buf);
294		72815	return JustVoid();
295			}
296
297		5528	void LibuvStreamWrap::GetWriteQueueSize(
298			const FunctionCallbackInfo<Value>& info) {
299			LibuvStreamWrap* wrap;
300	✗✓	5528	ASSIGN_OR_RETURN_UNWRAP(&wrap, info.This());
301
302	✗✓	5528	if (wrap->stream() == nullptr) {
303			info.GetReturnValue().Set(0);
304			return;
305			}
306
307		5528	uint32_t write_queue_size = wrap->stream()->write_queue_size;
308	✓✗	11056	info.GetReturnValue().Set(write_queue_size);
309			}
310
311
312		32	void LibuvStreamWrap::SetBlocking(const FunctionCallbackInfo<Value>& args) {
313			LibuvStreamWrap* wrap;
314	✗✓	32	ASSIGN_OR_RETURN_UNWRAP(&wrap, args.Holder());
315
316	✗✓	32	CHECK_GT(args.Length(), 0);
317	✗✓	32	if (!wrap->IsAlive())
318			return args.GetReturnValue().Set(UV_EINVAL);
319
320		32	bool enable = args[0]->IsTrue();
321		64	args.GetReturnValue().Set(uv_stream_set_blocking(wrap->stream(), enable));
322			}
323
324			typedef SimpleShutdownWrap<ReqWrap<uv_shutdown_t>> LibuvShutdownWrap;
325			typedef SimpleWriteWrap<ReqWrap<uv_write_t>> LibuvWriteWrap;
326
327		7581	ShutdownWrap* LibuvStreamWrap::CreateShutdownWrap(Local<Object> object) {
328		7581	return new LibuvShutdownWrap(this, object);
329			}
330
331		274	WriteWrap* LibuvStreamWrap::CreateWriteWrap(Local<Object> object) {
332		274	return new LibuvWriteWrap(this, object);
333			}
334
335
336		7581	int LibuvStreamWrap::DoShutdown(ShutdownWrap* req_wrap_) {
337		7581	LibuvShutdownWrap* req_wrap = static_cast<LibuvShutdownWrap*>(req_wrap_);
338		7581	return req_wrap->Dispatch(uv_shutdown, stream(), AfterUvShutdown);
339			}
340
341
342		7578	void LibuvStreamWrap::AfterUvShutdown(uv_shutdown_t* req, int status) {
343	✓✗	7578	LibuvShutdownWrap* req_wrap = static_cast<LibuvShutdownWrap*>(
344		7578	LibuvShutdownWrap::from_req(req));
345	✗✓	7578	CHECK_NOT_NULL(req_wrap);
346		15156	HandleScope scope(req_wrap->env()->isolate());
347		7578	Context::Scope context_scope(req_wrap->env()->context());
348		7578	req_wrap->Done(status);
349		7578	}
350
351
352			// NOTE: Call to this function could change both `buf`'s and `count`'s
353			// values, shifting their base and decrementing their length. This is
354			// required in order to skip the data that was successfully written via
355			// uv_try_write().
356		120874	int LibuvStreamWrap::DoTryWrite(uv_buf_t** bufs, size_t* count) {
357			int err;
358			size_t written;
359		120874	uv_buf_t* vbufs = *bufs;
360		120874	size_t vcount = *count;
361
362		120874	err = uv_try_write(stream(), vbufs, vcount);
363	✓✗✓✓	120874	if (err == UV_ENOSYS \|\| err == UV_EAGAIN)
364		6	return 0;
365	✓✓	120868	if (err < 0)
366		2377	return err;
367
368			// Slice off the buffers: skip all written buffers and slice the one that
369			// was partially written.
370		118491	written = err;
371	✓✓	315164	for (; vcount > 0; vbufs++, vcount--) {
372			// Slice
373	✓✓	196808	if (vbufs[0].len > written) {
374		135	vbufs[0].base += written;
375		135	vbufs[0].len -= written;
376		135	written = 0;
377		135	break;
378
379			// Discard
380			} else {
381		196673	written -= vbufs[0].len;
382			}
383			}
384
385		118491	*bufs = vbufs;
386		118491	*count = vcount;
387
388		118491	return 0;
389			}
390
391
392		274	int LibuvStreamWrap::DoWrite(WriteWrap* req_wrap,
393			uv_buf_t* bufs,
394			size_t count,
395			uv_stream_t* send_handle) {
396		274	LibuvWriteWrap* w = static_cast<LibuvWriteWrap*>(req_wrap);
397		274	return w->Dispatch(uv_write2,
398			stream(),
399			bufs,
400			count,
401			send_handle,
402		274	AfterUvWrite);
403			}
404
405
406
407		272	void LibuvStreamWrap::AfterUvWrite(uv_write_t* req, int status) {
408	✓✗	272	LibuvWriteWrap* req_wrap = static_cast<LibuvWriteWrap*>(
409		272	LibuvWriteWrap::from_req(req));
410	✗✓	272	CHECK_NOT_NULL(req_wrap);
411		543	HandleScope scope(req_wrap->env()->isolate());
412		272	Context::Scope context_scope(req_wrap->env()->context());
413		272	req_wrap->Done(status);
414		271	}
415
416			} // namespace node
417
418		5410	NODE_MODULE_CONTEXT_AWARE_INTERNAL(stream_wrap,
419			node::LibuvStreamWrap::Initialize)
420		5338	NODE_MODULE_EXTERNAL_REFERENCE(
421			stream_wrap, node::LibuvStreamWrap::RegisterExternalReferences)