From fe4f55ee13362c4253e78aad316eadfe08751a00 Mon Sep 17 00:00:00 2001
From: Anatoli Papirovski <apapirovski@mac.com>
Date: Mon, 22 Apr 2019 13:52:17 -0700
Subject: [PATCH] timers: fix refresh for expired timers

Expired timers were not being refresh correctly and
would always act as unrefed if refresh was called.

PR-URL: https://github.com/nodejs/node/pull/27345
Reviewed-By: Anna Henningsen <anna@addaleax.net>
Reviewed-By: Jeremiah Senkpiel <fishrock123@rocketmail.com>
Reviewed-By: Rich Trott <rtrott@gmail.com>
Reviewed-By: James M Snell <jasnell@gmail.com>
---
 lib/internal/timers.js               | 40 ++++++++++++++++------------
 lib/timers.js                        | 12 ++++-----
 test/parallel/test-timers-refresh.js | 14 ++++++++++
 3 files changed, 43 insertions(+), 23 deletions(-)

diff --git a/lib/internal/timers.js b/lib/internal/timers.js
index 7bd3f809b9bda3..b62ab9499c8fff 100644
--- a/lib/internal/timers.js
+++ b/lib/internal/timers.js
@@ -209,21 +209,23 @@ Timeout.prototype.refresh = function() {
 Timeout.prototype.unref = function() {
   if (this[kRefed]) {
     this[kRefed] = false;
-    decRefCount();
+    if (!this._destroyed)
+      decRefCount();
   }
   return this;
 };
 
 Timeout.prototype.ref = function() {
-  if (this[kRefed] === false) {
+  if (!this[kRefed]) {
     this[kRefed] = true;
-    incRefCount();
+    if (!this._destroyed)
+      incRefCount();
   }
   return this;
 };
 
 Timeout.prototype.hasRef = function() {
-  return !!this[kRefed];
+  return this[kRefed];
 };
 
 function TimersList(expiry, msecs) {
@@ -317,12 +319,16 @@ function insertGuarded(item, refed, start) {
 
   insert(item, msecs, start);
 
-  if (!item[async_id_symbol] || item._destroyed) {
+  const isDestroyed = item._destroyed;
+  if (isDestroyed || !item[async_id_symbol]) {
     item._destroyed = false;
     initAsyncResource(item, 'Timeout');
   }
 
-  if (refed === !item[kRefed]) {
+  if (isDestroyed) {
+    if (refed)
+      incRefCount();
+  } else if (refed === !item[kRefed]) {
     if (refed)
       incRefCount();
     else
@@ -519,13 +525,14 @@ function getTimerCallbacks(runNextTicks) {
       const asyncId = timer[async_id_symbol];
 
       if (!timer._onTimeout) {
-        if (timer[kRefed])
-          refCount--;
-        timer[kRefed] = null;
-
-        if (destroyHooksExist() && !timer._destroyed) {
-          emitDestroy(asyncId);
+        if (!timer._destroyed) {
           timer._destroyed = true;
+
+          if (timer[kRefed])
+            refCount--;
+
+          if (destroyHooksExist())
+            emitDestroy(asyncId);
         }
         continue;
       }
@@ -546,15 +553,14 @@ function getTimerCallbacks(runNextTicks) {
         if (timer._repeat && timer._idleTimeout !== -1) {
           timer._idleTimeout = timer._repeat;
           insert(timer, timer._idleTimeout, start);
-        } else if (!timer._idleNext && !timer._idlePrev) {
+        } else if (!timer._idleNext && !timer._idlePrev && !timer._destroyed) {
+          timer._destroyed = true;
+
           if (timer[kRefed])
             refCount--;
-          timer[kRefed] = null;
 
-          if (destroyHooksExist() && !timer._destroyed) {
+          if (destroyHooksExist())
             emitDestroy(asyncId);
-          }
-          timer._destroyed = true;
         }
       }
 
diff --git a/lib/timers.js b/lib/timers.js
index f768e1a5c90e17..5904ec9bd3b533 100644
--- a/lib/timers.js
+++ b/lib/timers.js
@@ -64,13 +64,14 @@ const {
 
 // Remove a timer. Cancels the timeout and resets the relevant timer properties.
 function unenroll(item) {
+  if (item._destroyed)
+    return;
+
+  item._destroyed = true;
+
   // Fewer checks may be possible, but these cover everything.
-  if (destroyHooksExist() &&
-      item[async_id_symbol] !== undefined &&
-      !item._destroyed) {
+  if (destroyHooksExist() && item[async_id_symbol] !== undefined)
     emitDestroy(item[async_id_symbol]);
-  }
-  item._destroyed = true;
 
   L.remove(item);
 
@@ -91,7 +92,6 @@ function unenroll(item) {
 
     decRefCount();
   }
-  item[kRefed] = null;
 
   // If active is called later, then we want to make sure not to insert again
   item._idleTimeout = -1;
diff --git a/test/parallel/test-timers-refresh.js b/test/parallel/test-timers-refresh.js
index e186c63a65ecf9..a96f5ad5368275 100644
--- a/test/parallel/test-timers-refresh.js
+++ b/test/parallel/test-timers-refresh.js
@@ -72,6 +72,20 @@ const { inspect } = require('util');
   strictEqual(timer.refresh(), timer);
 }
 
+// regular timer
+{
+  let called = false;
+  const timer = setTimeout(common.mustCall(() => {
+    if (!called) {
+      called = true;
+      process.nextTick(common.mustCall(() => {
+        timer.refresh();
+        strictEqual(timer.hasRef(), true);
+      }));
+    }
+  }, 2), 1);
+}
+
 // interval
 {
   let called = 0;