Skip to content

Commit

Permalink
src: simplify handlewrap state tracking logic
Browse files Browse the repository at this point in the history
This also updates the tests to expect that a closed handle has no
reference count.

PR-URL: #6395
Reviewed-By: Anna Henningsen <anna@addaleax.net>
Reviewed-By: Colin Ihrig <cjihrig@gmail.com>
Reviewed-By: James M Snell <jasnell@gmail.com>
  • Loading branch information
bnoordhuis authored and evanlucas committed May 17, 2016
1 parent 965274d commit bbf3b3e
Show file tree
Hide file tree
Showing 4 changed files with 17 additions and 17 deletions.
8 changes: 4 additions & 4 deletions src/handle_wrap.cc
Expand Up @@ -36,10 +36,7 @@ void HandleWrap::Unref(const FunctionCallbackInfo<Value>& args) {

void HandleWrap::Unrefed(const FunctionCallbackInfo<Value>& args) {
HandleWrap* wrap = Unwrap<HandleWrap>(args.Holder());
// XXX(bnoordhuis) It's debatable whether a nullptr wrap should count
// as having a reference count but it's compatible with the logic that
// it replaces.
args.GetReturnValue().Set(wrap == nullptr || !HasRef(wrap));
args.GetReturnValue().Set(!HasRef(wrap));
}


Expand All @@ -52,6 +49,9 @@ void HandleWrap::Close(const FunctionCallbackInfo<Value>& args) {
if (!IsAlive(wrap))
return;

if (wrap->state_ != kInitialized)
return;

CHECK_EQ(false, wrap->persistent().IsEmpty());
uv_close(wrap->handle__, OnClose);
wrap->state_ = kClosing;
Expand Down
8 changes: 2 additions & 6 deletions src/handle_wrap.h
Expand Up @@ -38,15 +38,11 @@ class HandleWrap : public AsyncWrap {
static void Unrefed(const v8::FunctionCallbackInfo<v8::Value>& args);

static inline bool IsAlive(const HandleWrap* wrap) {
// XXX(bnoordhuis) It's debatable whether only kInitialized should
// count as alive but it's compatible with the check that it replaces.
return wrap != nullptr && wrap->state_ == kInitialized;
return wrap != nullptr && wrap->state_ != kClosed;
}

static inline bool HasRef(const HandleWrap* wrap) {
return wrap != nullptr &&
wrap->state_ != kClosed &&
uv_has_ref(wrap->GetHandle());
return IsAlive(wrap) && uv_has_ref(wrap->GetHandle());
}

inline uv_handle_t* GetHandle() const { return handle__; }
Expand Down
2 changes: 1 addition & 1 deletion test/parallel/test-handle-wrap-isrefed-tty.js
Expand Up @@ -19,7 +19,7 @@ if (process.argv[2] === 'child') {
assert(tty._handle.unrefed(), false);
tty.unref();
assert(tty._handle.unrefed(), true);
tty._handle.close();
tty._handle.close(common.mustCall(() => assert(tty._handle.unrefed(), true)));
assert(tty._handle.unrefed(), true);
return;
}
Expand Down
16 changes: 10 additions & 6 deletions test/parallel/test-handle-wrap-isrefed.js
Expand Up @@ -22,7 +22,7 @@ function makeAssert(message) {
assert(cp._handle.unrefed(), true);
cp.ref();
assert(cp._handle.unrefed(), false);
cp._handle.close();
cp._handle.close(common.mustCall(() => assert(cp._handle.unrefed(), true)));
assert(cp._handle.unrefed(), false);
}

Expand All @@ -39,7 +39,8 @@ function makeAssert(message) {
assert(sock4._handle.unrefed(), true);
sock4.ref();
assert(sock4._handle.unrefed(), false);
sock4._handle.close();
sock4._handle.close(
common.mustCall(() => assert(sock4._handle.unrefed(), true)));
assert(sock4._handle.unrefed(), false);

const sock6 = dgram.createSocket('udp6');
Expand All @@ -49,7 +50,8 @@ function makeAssert(message) {
assert(sock6._handle.unrefed(), true);
sock6.ref();
assert(sock6._handle.unrefed(), false);
sock6._handle.close();
sock6._handle.close(
common.mustCall(() => assert(sock6._handle.unrefed(), true)));
assert(sock6._handle.unrefed(), false);
}

Expand All @@ -65,7 +67,7 @@ function makeAssert(message) {
assert(handle.unrefed(), true);
handle.ref();
assert(handle.unrefed(), false);
handle.close();
handle.close(common.mustCall(() => assert(handle.unrefed(), true)));
assert(handle.unrefed(), false);
}

Expand All @@ -84,7 +86,8 @@ function makeAssert(message) {
server.ref();
assert(server._handle.unrefed(), false);
assert(server._unref, false);
server._handle.close();
server._handle.close(
common.mustCall(() => assert(server._handle.unrefed(), true)));
assert(server._handle.unrefed(), false);
}

Expand All @@ -98,6 +101,7 @@ function makeAssert(message) {
assert(timer._handle.unrefed(), true);
timer.ref();
assert(timer._handle.unrefed(), false);
timer.close();
timer._handle.close(
common.mustCall(() => assert(timer._handle.unrefed(), true)));
assert(timer._handle.unrefed(), false);
}

0 comments on commit bbf3b3e

Please sign in to comment.