GCC Code Coverage Report
Directory: ./ Exec Total Coverage
File: tty_wrap.cc Lines: 62 62 100.0 %
Date: 2022-08-05 04:16:08 Branches: 19 32 59.4 %

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 "tty_wrap.h"
23
24
#include "env-inl.h"
25
#include "handle_wrap.h"
26
#include "node_buffer.h"
27
#include "node_external_reference.h"
28
#include "stream_base-inl.h"
29
#include "stream_wrap.h"
30
#include "util-inl.h"
31
32
namespace node {
33
34
using v8::Array;
35
using v8::Context;
36
using v8::FunctionCallbackInfo;
37
using v8::FunctionTemplate;
38
using v8::Integer;
39
using v8::Isolate;
40
using v8::Local;
41
using v8::Object;
42
using v8::String;
43
using v8::Value;
44
45
5314
void TTYWrap::RegisterExternalReferences(ExternalReferenceRegistry* registry) {
46
5314
  registry->Register(New);
47
5314
  registry->Register(GetWindowSize);
48
5314
  registry->Register(SetRawMode);
49
5314
  registry->Register(IsTTY);
50
5314
}
51
52
5263
void TTYWrap::Initialize(Local<Object> target,
53
                         Local<Value> unused,
54
                         Local<Context> context,
55
                         void* priv) {
56
5263
  Environment* env = Environment::GetCurrent(context);
57
5263
  Isolate* isolate = env->isolate();
58
59
5263
  Local<String> ttyString = FIXED_ONE_BYTE_STRING(env->isolate(), "TTY");
60
61
5263
  Local<FunctionTemplate> t = NewFunctionTemplate(isolate, New);
62
5263
  t->SetClassName(ttyString);
63
10526
  t->InstanceTemplate()->SetInternalFieldCount(StreamBase::kInternalFieldCount);
64
5263
  t->Inherit(LibuvStreamWrap::GetConstructorTemplate(env));
65
66
5263
  SetProtoMethodNoSideEffect(
67
      isolate, t, "getWindowSize", TTYWrap::GetWindowSize);
68
5263
  SetProtoMethod(isolate, t, "setRawMode", SetRawMode);
69
70
5263
  SetMethodNoSideEffect(context, target, "isTTY", IsTTY);
71
72
  Local<Value> func;
73
15789
  if (t->GetFunction(context).ToLocal(&func) &&
74

15789
      target->Set(context, ttyString, func).IsJust()) {
75
5263
    env->set_tty_constructor_template(t);
76
  }
77
5263
}
78
79
80
362
void TTYWrap::IsTTY(const FunctionCallbackInfo<Value>& args) {
81
362
  Environment* env = Environment::GetCurrent(args);
82
  int fd;
83
724
  if (!args[0]->Int32Value(env->context()).To(&fd)) return;
84
362
  CHECK_GE(fd, 0);
85
362
  bool rc = uv_guess_handle(fd) == UV_TTY;
86
724
  args.GetReturnValue().Set(rc);
87
}
88
89
90
32
void TTYWrap::GetWindowSize(const FunctionCallbackInfo<Value>& args) {
91
32
  Environment* env = Environment::GetCurrent(args);
92
93
  TTYWrap* wrap;
94
32
  ASSIGN_OR_RETURN_UNWRAP(&wrap,
95
                          args.Holder(),
96
                          args.GetReturnValue().Set(UV_EBADF));
97
32
  CHECK(args[0]->IsArray());
98
99
  int width, height;
100
32
  int err = uv_tty_get_winsize(&wrap->handle_, &width, &height);
101
102
32
  if (err == 0) {
103
64
    Local<Array> a = args[0].As<Array>();
104
96
    a->Set(env->context(), 0, Integer::New(env->isolate(), width)).Check();
105
96
    a->Set(env->context(), 1, Integer::New(env->isolate(), height)).Check();
106
  }
107
108
64
  args.GetReturnValue().Set(err);
109
}
110
111
112
8
void TTYWrap::SetRawMode(const FunctionCallbackInfo<Value>& args) {
113
  TTYWrap* wrap;
114
8
  ASSIGN_OR_RETURN_UNWRAP(&wrap,
115
                          args.Holder(),
116
                          args.GetReturnValue().Set(UV_EBADF));
117
8
  int err = uv_tty_set_mode(&wrap->handle_, args[0]->IsTrue());
118
16
  args.GetReturnValue().Set(err);
119
}
120
121
122
48
void TTYWrap::New(const FunctionCallbackInfo<Value>& args) {
123
48
  Environment* env = Environment::GetCurrent(args);
124
125
  // This constructor should not be exposed to public javascript.
126
  // Therefore we assert that we are not trying to call this as a
127
  // normal function.
128
48
  CHECK(args.IsConstructCall());
129
130
  int fd;
131
96
  if (!args[0]->Int32Value(env->context()).To(&fd)) return;
132
48
  CHECK_GE(fd, 0);
133
134
48
  int err = 0;
135
48
  new TTYWrap(env, args.This(), fd, &err);
136
48
  if (err != 0) {
137
4
    env->CollectUVExceptionInfo(args[1], err, "uv_tty_init");
138
4
    args.GetReturnValue().SetUndefined();
139
  }
140
}
141
142
143
48
TTYWrap::TTYWrap(Environment* env,
144
                 Local<Object> object,
145
                 int fd,
146
48
                 int* init_err)
147
    : LibuvStreamWrap(env,
148
                      object,
149
48
                      reinterpret_cast<uv_stream_t*>(&handle_),
150
48
                      AsyncWrap::PROVIDER_TTYWRAP) {
151
48
  *init_err = uv_tty_init(env->event_loop(), &handle_, fd, 0);
152
48
  set_fd(fd);
153
48
  if (*init_err != 0)
154
2
    MarkAsUninitialized();
155
48
}
156
157
}  // namespace node
158
159
5386
NODE_MODULE_CONTEXT_AWARE_INTERNAL(tty_wrap, node::TTYWrap::Initialize)
160
5314
NODE_MODULE_EXTERNAL_REFERENCE(tty_wrap,
161
                               node::TTYWrap::RegisterExternalReferences)