Head

GCC Code Coverage Report

Directory:	./		Exec	Total	Coverage
File:	stream_base-inl.h	Lines:	158	165	95.8 %
Date:	2022-05-22 04:15:48	Branches:	46	60	76.7 %


#ifndef SRC_STREAM_BASE_INL_H_
#define SRC_STREAM_BASE_INL_H_

#if defined(NODE_WANT_INTERNALS) && NODE_WANT_INTERNALS

#include "async_wrap-inl.h"
#include "base_object-inl.h"
#include "node.h"
#include "stream_base.h"
#include "v8.h"

namespace node {

StreamReq::StreamReq(
    StreamBase* stream,
    v8::Local<v8::Object> req_wrap_obj) : stream_(stream) {
  AttachToObject(req_wrap_obj);
}

void StreamReq::AttachToObject(v8::Local<v8::Object> req_wrap_obj) {
  CHECK_EQ(req_wrap_obj->GetAlignedPointerFromInternalField(
               StreamReq::kStreamReqField),
           nullptr);
  req_wrap_obj->SetAlignedPointerInInternalField(
      StreamReq::kStreamReqField, this);
}

StreamReq* StreamReq::FromObject(v8::Local<v8::Object> req_wrap_obj) {
  return static_cast<StreamReq*>(
      req_wrap_obj->GetAlignedPointerFromInternalField(
          StreamReq::kStreamReqField));
}

void StreamReq::Dispose() {
  BaseObjectPtr<AsyncWrap> destroy_me{GetAsyncWrap()};
  object()->SetAlignedPointerInInternalField(
      StreamReq::kStreamReqField, nullptr);
  destroy_me->Detach();
}

v8::Local<v8::Object> StreamReq::object() {
  return GetAsyncWrap()->object();
}

ShutdownWrap::ShutdownWrap(
    StreamBase* stream,
    v8::Local<v8::Object> req_wrap_obj)
    : StreamReq(stream, req_wrap_obj) { }

WriteWrap::WriteWrap(
    StreamBase* stream,
    v8::Local<v8::Object> req_wrap_obj)
    : StreamReq(stream, req_wrap_obj) { }

void StreamListener::PassReadErrorToPreviousListener(ssize_t nread) {
  CHECK_NOT_NULL(previous_listener_);
  previous_listener_->OnStreamRead(nread, uv_buf_init(nullptr, 0));
}

void StreamResource::PushStreamListener(StreamListener* listener) {
  CHECK_NOT_NULL(listener);
  CHECK_NULL(listener->stream_);

  listener->previous_listener_ = listener_;
  listener->stream_ = this;

  listener_ = listener;
}

void StreamResource::RemoveStreamListener(StreamListener* listener) {
  CHECK_NOT_NULL(listener);

  StreamListener* previous;
  StreamListener* current;

  // Remove from the linked list.
  for (current = listener_, previous = nullptr;
       /* No loop condition because we want a crash if listener is not found */
       ; previous = current, current = current->previous_listener_) {
    CHECK_NOT_NULL(current);
    if (current == listener) {
      if (previous != nullptr)
        previous->previous_listener_ = current->previous_listener_;
      else
        listener_ = listener->previous_listener_;
      break;
    }
  }

  listener->stream_ = nullptr;
  listener->previous_listener_ = nullptr;
}

uv_buf_t StreamResource::EmitAlloc(size_t suggested_size) {
  DebugSealHandleScope seal_handle_scope;
  return listener_->OnStreamAlloc(suggested_size);
}

void StreamResource::EmitRead(ssize_t nread, const uv_buf_t& buf) {
  DebugSealHandleScope seal_handle_scope;
  if (nread > 0)
    bytes_read_ += static_cast<uint64_t>(nread);
  listener_->OnStreamRead(nread, buf);
}

void StreamResource::EmitAfterWrite(WriteWrap* w, int status) {
  DebugSealHandleScope seal_handle_scope;
  listener_->OnStreamAfterWrite(w, status);
}

void StreamResource::EmitAfterShutdown(ShutdownWrap* w, int status) {
  DebugSealHandleScope seal_handle_scope;
  listener_->OnStreamAfterShutdown(w, status);
}

void StreamResource::EmitWantsWrite(size_t suggested_size) {
  DebugSealHandleScope seal_handle_scope;
  listener_->OnStreamWantsWrite(suggested_size);
}

StreamBase::StreamBase(Environment* env) : env_(env) {
  PushStreamListener(&default_listener_);
}

int StreamBase::Shutdown(v8::Local<v8::Object> req_wrap_obj) {
  Environment* env = stream_env();

  v8::HandleScope handle_scope(env->isolate());

  if (req_wrap_obj.IsEmpty()) {
    if (!env->shutdown_wrap_template()
             ->NewInstance(env->context())
             .ToLocal(&req_wrap_obj)) {
      return UV_EBUSY;
    }
    StreamReq::ResetObject(req_wrap_obj);
  }

  BaseObjectPtr<AsyncWrap> req_wrap_ptr;
  AsyncHooks::DefaultTriggerAsyncIdScope trigger_scope(GetAsyncWrap());
  ShutdownWrap* req_wrap = CreateShutdownWrap(req_wrap_obj);
  if (req_wrap != nullptr)
    req_wrap_ptr.reset(req_wrap->GetAsyncWrap());
  int err = DoShutdown(req_wrap);

  if (err != 0 && req_wrap != nullptr) {
    req_wrap->Dispose();
  }

  const char* msg = Error();
  if (msg != nullptr) {
    if (req_wrap_obj->Set(env->context(),
                          env->error_string(),
                          OneByteString(env->isolate(), msg)).IsNothing()) {
      return UV_EBUSY;
    }
    ClearError();
  }

  return err;
}

StreamWriteResult StreamBase::Write(
    uv_buf_t* bufs,
    size_t count,
    uv_stream_t* send_handle,
    v8::Local<v8::Object> req_wrap_obj) {
  Environment* env = stream_env();
  int err;

  size_t total_bytes = 0;
  for (size_t i = 0; i < count; ++i)
    total_bytes += bufs[i].len;
  bytes_written_ += total_bytes;

  if (send_handle == nullptr) {
    err = DoTryWrite(&bufs, &count);
    if (err != 0 || count == 0) {
      return StreamWriteResult { false, err, nullptr, total_bytes, {} };
    }
  }

  v8::HandleScope handle_scope(env->isolate());

  if (req_wrap_obj.IsEmpty()) {
    if (!env->write_wrap_template()
             ->NewInstance(env->context())
             .ToLocal(&req_wrap_obj)) {
      return StreamWriteResult { false, UV_EBUSY, nullptr, 0, {} };
    }
    StreamReq::ResetObject(req_wrap_obj);
  }

  AsyncHooks::DefaultTriggerAsyncIdScope trigger_scope(GetAsyncWrap());
  WriteWrap* req_wrap = CreateWriteWrap(req_wrap_obj);
  BaseObjectPtr<AsyncWrap> req_wrap_ptr(req_wrap->GetAsyncWrap());

  err = DoWrite(req_wrap, bufs, count, send_handle);
  bool async = err == 0;

  if (!async) {
    req_wrap->Dispose();
    req_wrap = nullptr;
  }

  const char* msg = Error();
  if (msg != nullptr) {
    if (req_wrap_obj->Set(env->context(),
                          env->error_string(),
                          OneByteString(env->isolate(), msg)).IsNothing()) {
      return StreamWriteResult { false, UV_EBUSY, nullptr, 0, {} };
    }
    ClearError();
  }

  return StreamWriteResult {
      async, err, req_wrap, total_bytes, std::move(req_wrap_ptr) };
}

template <typename OtherBase>
SimpleShutdownWrap<OtherBase>::SimpleShutdownWrap(
    StreamBase* stream,
    v8::Local<v8::Object> req_wrap_obj)
  : ShutdownWrap(stream, req_wrap_obj),
    OtherBase(stream->stream_env(),
              req_wrap_obj,
              AsyncWrap::PROVIDER_SHUTDOWNWRAP) {
}

template <typename OtherBase>
SimpleWriteWrap<OtherBase>::SimpleWriteWrap(
    StreamBase* stream,
    v8::Local<v8::Object> req_wrap_obj)
  : WriteWrap(stream, req_wrap_obj),
    OtherBase(stream->stream_env(),
              req_wrap_obj,
              AsyncWrap::PROVIDER_WRITEWRAP) {
}

void StreamBase::AttachToObject(v8::Local<v8::Object> obj) {
  obj->SetAlignedPointerInInternalField(
      StreamBase::kStreamBaseField, this);
}

StreamBase* StreamBase::FromObject(v8::Local<v8::Object> obj) {
  if (obj->GetAlignedPointerFromInternalField(StreamBase::kSlot) == nullptr)
    return nullptr;

  return static_cast<StreamBase*>(
      obj->GetAlignedPointerFromInternalField(
          StreamBase::kStreamBaseField));
}

WriteWrap* WriteWrap::FromObject(v8::Local<v8::Object> req_wrap_obj) {
  return static_cast<WriteWrap*>(StreamReq::FromObject(req_wrap_obj));
}

template <typename T, bool kIsWeak>
WriteWrap* WriteWrap::FromObject(
    const BaseObjectPtrImpl<T, kIsWeak>& base_obj) {
  if (!base_obj) return nullptr;
  return FromObject(base_obj->object());
}

ShutdownWrap* ShutdownWrap::FromObject(v8::Local<v8::Object> req_wrap_obj) {
  return static_cast<ShutdownWrap*>(StreamReq::FromObject(req_wrap_obj));
}

template <typename T, bool kIsWeak>
ShutdownWrap* ShutdownWrap::FromObject(
    const BaseObjectPtrImpl<T, kIsWeak>& base_obj) {
  if (!base_obj) return nullptr;
  return FromObject(base_obj->object());
}

void WriteWrap::SetBackingStore(std::unique_ptr<v8::BackingStore> bs) {
  CHECK(!backing_store_);
  backing_store_ = std::move(bs);
}

void StreamReq::Done(int status, const char* error_str) {
  AsyncWrap* async_wrap = GetAsyncWrap();
  Environment* env = async_wrap->env();
  if (error_str != nullptr) {
    v8::HandleScope handle_scope(env->isolate());
    if (async_wrap->object()->Set(
            env->context(),
            env->error_string(),
            OneByteString(env->isolate(), error_str)).IsNothing()) {
      return;
    }
  }

  OnDone(status);
}

void StreamReq::ResetObject(v8::Local<v8::Object> obj) {
  DCHECK_GT(obj->InternalFieldCount(), StreamReq::kStreamReqField);

  obj->SetAlignedPointerInInternalField(StreamReq::kSlot, nullptr);
  obj->SetAlignedPointerInInternalField(StreamReq::kStreamReqField, nullptr);
}

}  // namespace node

#endif  // defined(NODE_WANT_INTERNALS) && NODE_WANT_INTERNALS

#endif  // SRC_STREAM_BASE_INL_H_


Generated by: GCOVR (Version 4.2)

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