GCC Code Coverage Report
Directory: ./ Exec Total Coverage
File: js_stream.cc Lines: 98 112 87.5 %
Date: 2022-08-02 04:16:51 Branches: 43 80 53.8 %

Line Branch Exec Source
1
#include "js_stream.h"
2
3
#include "async_wrap.h"
4
#include "env-inl.h"
5
#include "node_errors.h"
6
#include "stream_base-inl.h"
7
#include "util-inl.h"
8
#include "v8.h"
9
10
namespace node {
11
12
using errors::TryCatchScope;
13
14
using v8::Array;
15
using v8::Context;
16
using v8::FunctionCallbackInfo;
17
using v8::FunctionTemplate;
18
using v8::HandleScope;
19
using v8::Int32;
20
using v8::Isolate;
21
using v8::Local;
22
using v8::Object;
23
using v8::Value;
24
25
26
11002
JSStream::JSStream(Environment* env, Local<Object> obj)
27
    : AsyncWrap(env, obj, AsyncWrap::PROVIDER_JSSTREAM),
28
11002
      StreamBase(env) {
29
11002
  MakeWeak();
30
11002
  StreamBase::AttachToObject(obj);
31
11002
}
32
33
34
12951
AsyncWrap* JSStream::GetAsyncWrap() {
35
12951
  return static_cast<AsyncWrap*>(this);
36
}
37
38
39
11027
bool JSStream::IsAlive() {
40
11027
  return true;
41
}
42
43
44
bool JSStream::IsClosing() {
45
  HandleScope scope(env()->isolate());
46
  Context::Scope context_scope(env()->context());
47
  TryCatchScope try_catch(env());
48
  Local<Value> value;
49
  if (!MakeCallback(env()->isclosing_string(), 0, nullptr).ToLocal(&value)) {
50
    if (try_catch.HasCaught() && !try_catch.HasTerminated())
51
      errors::TriggerUncaughtException(env()->isolate(), try_catch);
52
    return true;
53
  }
54
  return value->IsTrue();
55
}
56
57
58
11907
int JSStream::ReadStart() {
59
23814
  HandleScope scope(env()->isolate());
60
11907
  Context::Scope context_scope(env()->context());
61
11907
  TryCatchScope try_catch(env());
62
  Local<Value> value;
63
11907
  int value_int = UV_EPROTO;
64
35720
  if (!MakeCallback(env()->onreadstart_string(), 0, nullptr).ToLocal(&value) ||
65

35719
      !value->Int32Value(env()->context()).To(&value_int)) {
66

1
    if (try_catch.HasCaught() && !try_catch.HasTerminated())
67
      errors::TriggerUncaughtException(env()->isolate(), try_catch);
68
  }
69
11907
  return value_int;
70
}
71
72
73
936
int JSStream::ReadStop() {
74
1872
  HandleScope scope(env()->isolate());
75
936
  Context::Scope context_scope(env()->context());
76
936
  TryCatchScope try_catch(env());
77
  Local<Value> value;
78
936
  int value_int = UV_EPROTO;
79
2807
  if (!MakeCallback(env()->onreadstop_string(), 0, nullptr).ToLocal(&value) ||
80

2806
      !value->Int32Value(env()->context()).To(&value_int)) {
81

1
    if (try_catch.HasCaught() && !try_catch.HasTerminated())
82
      errors::TriggerUncaughtException(env()->isolate(), try_catch);
83
  }
84
936
  return value_int;
85
}
86
87
88
13
int JSStream::DoShutdown(ShutdownWrap* req_wrap) {
89
26
  HandleScope scope(env()->isolate());
90
13
  Context::Scope context_scope(env()->context());
91
92
  Local<Value> argv[] = {
93
    req_wrap->object()
94
13
  };
95
96
13
  TryCatchScope try_catch(env());
97
  Local<Value> value;
98
13
  int value_int = UV_EPROTO;
99
26
  if (!MakeCallback(env()->onshutdown_string(),
100
13
                    arraysize(argv),
101
39
                    argv).ToLocal(&value) ||
102

39
      !value->Int32Value(env()->context()).To(&value_int)) {
103
    if (try_catch.HasCaught() && !try_catch.HasTerminated())
104
      errors::TriggerUncaughtException(env()->isolate(), try_catch);
105
  }
106
13
  return value_int;
107
}
108
109
110
1903
int JSStream::DoWrite(WriteWrap* w,
111
                      uv_buf_t* bufs,
112
                      size_t count,
113
                      uv_stream_t* send_handle) {
114
1903
  CHECK_NULL(send_handle);
115
116
3805
  HandleScope scope(env()->isolate());
117
1903
  Context::Scope context_scope(env()->context());
118
119
3805
  MaybeStackBuffer<Local<Value>, 16> bufs_arr(count);
120
6195
  for (size_t i = 0; i < count; i++) {
121
4292
    bufs_arr[i] =
122
8584
        Buffer::Copy(env(), bufs[i].base, bufs[i].len).ToLocalChecked();
123
  }
124
125
  Local<Value> argv[] = {
126
    w->object(),
127
    Array::New(env()->isolate(), bufs_arr.out(), count)
128
3806
  };
129
130
1903
  TryCatchScope try_catch(env());
131
  Local<Value> value;
132
1903
  int value_int = UV_EPROTO;
133
3806
  if (!MakeCallback(env()->onwrite_string(),
134
1903
                    arraysize(argv),
135
5296
                    argv).ToLocal(&value) ||
136

4883
      !value->Int32Value(env()->context()).To(&value_int)) {
137

413
    if (try_catch.HasCaught() && !try_catch.HasTerminated())
138
2
      errors::TriggerUncaughtException(env()->isolate(), try_catch);
139
  }
140
1902
  return value_int;
141
}
142
143
144
11002
void JSStream::New(const FunctionCallbackInfo<Value>& args) {
145
  // This constructor should not be exposed to public javascript.
146
  // Therefore we assert that we are not trying to call this as a
147
  // normal function.
148
11002
  CHECK(args.IsConstructCall());
149
11002
  Environment* env = Environment::GetCurrent(args);
150
11002
  new JSStream(env, args.This());
151
11002
}
152
153
154
template <class Wrap>
155
2974
void JSStream::Finish(const FunctionCallbackInfo<Value>& args) {
156
2974
  CHECK(args[0]->IsObject());
157
5948
  Wrap* w = static_cast<Wrap*>(StreamReq::FromObject(args[0].As<Object>()));
158
159
2974
  CHECK(args[1]->IsInt32());
160
8922
  w->Done(args[1].As<Int32>()->Value());
161
2974
}
162
163
164
3004
void JSStream::ReadBuffer(const FunctionCallbackInfo<Value>& args) {
165
  JSStream* wrap;
166
3004
  ASSIGN_OR_RETURN_UNWRAP(&wrap, args.Holder());
167
168
3004
  ArrayBufferViewContents<char> buffer(args[0]);
169
3004
  const char* data = buffer.data();
170
3004
  int len = buffer.length();
171
172
  // Repeatedly ask the stream's owner for memory, copy the data that we
173
  // just read from JS into those buffers and emit them as reads.
174
6007
  while (len != 0) {
175
3004
    uv_buf_t buf = wrap->EmitAlloc(len);
176
3004
    ssize_t avail = len;
177
3004
    if (static_cast<ssize_t>(buf.len) < avail)
178
      avail = buf.len;
179
180
3004
    memcpy(buf.base, data, avail);
181
3004
    data += avail;
182
3004
    len -= static_cast<int>(avail);
183
3004
    wrap->EmitRead(avail, buf);
184
  }
185
}
186
187
188
12
void JSStream::EmitEOF(const FunctionCallbackInfo<Value>& args) {
189
  JSStream* wrap;
190
12
  ASSIGN_OR_RETURN_UNWRAP(&wrap, args.Holder());
191
192
12
  wrap->EmitRead(UV_EOF);
193
}
194
195
196
766
void JSStream::Initialize(Local<Object> target,
197
                          Local<Value> unused,
198
                          Local<Context> context,
199
                          void* priv) {
200
766
  Environment* env = Environment::GetCurrent(context);
201
766
  Isolate* isolate = env->isolate();
202
203
766
  Local<FunctionTemplate> t = NewFunctionTemplate(isolate, New);
204
1532
  t->InstanceTemplate()
205
766
    ->SetInternalFieldCount(StreamBase::kInternalFieldCount);
206
766
  t->Inherit(AsyncWrap::GetConstructorTemplate(env));
207
208
766
  SetProtoMethod(isolate, t, "finishWrite", Finish<WriteWrap>);
209
766
  SetProtoMethod(isolate, t, "finishShutdown", Finish<ShutdownWrap>);
210
766
  SetProtoMethod(isolate, t, "readBuffer", ReadBuffer);
211
766
  SetProtoMethod(isolate, t, "emitEOF", EmitEOF);
212
213
766
  StreamBase::AddMethods(env, t);
214
766
  SetConstructorFunction(context, target, "JSStream", t);
215
766
}
216
217
}  // namespace node
218
219
5368
NODE_MODULE_CONTEXT_AWARE_INTERNAL(js_stream, node::JSStream::Initialize)