From 323306b8a5f6b485f2ff69af1953f77ceb8ab273 Mon Sep 17 00:00:00 2001
From: Darshan Sen <darshan.sen@postman.com>
Date: Sun, 14 Nov 2021 17:09:49 +0530
Subject: [PATCH] process: add `getActiveResourcesInfo()`

This is supposed to be a public alternative of the private APIs,
`process._getActiveResources()` and `process._getActiveHandles()`. When
called, it returns an array of objects containing the `type`, `asyncId`
and the `triggerAsyncId` of the currently active `requests`, `handles`
and `timers`.

Signed-off-by: Darshan Sen <darshan.sen@postman.com>
---
 doc/api/process.md                            | 69 +++++++++++++++++++
 lib/internal/bootstrap/node.js                | 46 ++++++++++++-
 lib/internal/timers.js                        | 11 +--
 src/node_process_methods.cc                   | 57 +++++++++++++++
 test/parallel/test-handle-wrap-isrefed.js     | 26 +++++++
 ...getactiveresources-track-active-handles.js | 44 ++++++++++++
 ...etactiveresources-track-active-requests.js | 10 +++
 .../test-process-getactiveresources.js        | 20 ++++++
 .../test-net-connect-econnrefused.js          |  5 +-
 9 files changed, 278 insertions(+), 10 deletions(-)
 create mode 100644 test/parallel/test-process-getactiveresources-track-active-handles.js
 create mode 100644 test/parallel/test-process-getactiveresources-track-active-requests.js
 create mode 100644 test/parallel/test-process-getactiveresources.js

diff --git a/doc/api/process.md b/doc/api/process.md
index 28db286bd90fd2..5d042977f96307 100644
--- a/doc/api/process.md
+++ b/doc/api/process.md
@@ -1817,6 +1817,71 @@ a code.
 Specifying a code to [`process.exit(code)`][`process.exit()`] will override any
 previous setting of `process.exitCode`.
 
+## `process.getActiveResourcesInfo()`
+
+<!-- YAML
+added: REPLACEME
+-->
+
+> Stability: 1 - Experimental
+
+* Returns: {Array} of objects containing the `type`, `asyncId` and the
+  `triggerAsyncId` of the currently active requests, handles and timers.
+
+The `process.getActiveResourcesInfo()` method can be called at any moment and it
+would return an array of objects containing some information about all the
+resources that are currently keeping the event loop alive. This can be useful
+while debugging.
+
+The same functionality can be implemented using [`async_hooks`][] by setting up
+an [`AsyncHook`][] that keeps a track of the calls made to [`init` callbacks][]
+and [`destroy` callbacks][] but this function should be preferred because it
+doesn't incur the overhead of creating any [`AsyncHook`][].
+
+```mjs
+import { getActiveResourcesInfo } from 'process';
+import { setTimeout } from 'timers';
+
+console.log('Before:', getActiveResourcesInfo());
+setTimeout(() => {}, 1000);
+console.log('After:', getActiveResourcesInfo());
+// Prints:
+//   Before: [
+//     { type: 'CloseReq', asyncId: 6, triggerAsyncId: 0 },
+//     { type: 'TTYWrap', asyncId: 7, triggerAsyncId: 0 },
+//     { type: 'TTYWrap', asyncId: 9, triggerAsyncId: 0 },
+//     { type: 'TTYWrap', asyncId: 10, triggerAsyncId: 0 }
+//   ]
+//   After: [
+//     { type: 'CloseReq', asyncId: 6, triggerAsyncId: 0 },
+//     { type: 'TTYWrap', asyncId: 7, triggerAsyncId: 0 },
+//     { type: 'TTYWrap', asyncId: 9, triggerAsyncId: 0 },
+//     { type: 'TTYWrap', asyncId: 10, triggerAsyncId: 0 },
+//     { type: 'Timeout', asyncId: 12, triggerAsyncId: 0 }
+//   ]
+```
+
+```cjs
+const { getActiveResourcesInfo } = require('process');
+const { setTimeout } = require('timers');
+
+console.log('Before:', getActiveResourcesInfo());
+setTimeout(() => {}, 1000);
+console.log('After:', getActiveResourcesInfo());
+// Prints:
+//   Before: [
+//     { type: 'TTYWrap', asyncId: 2, triggerAsyncId: 1 },
+//     { type: 'TTYWrap', asyncId: 4, triggerAsyncId: 1 },
+//     { type: 'TTYWrap', asyncId: 5, triggerAsyncId: 1 }
+//   ]
+//   After: [
+//     { type: 'TTYWrap', asyncId: 2, triggerAsyncId: 1 },
+//     { type: 'TTYWrap', asyncId: 4, triggerAsyncId: 1 },
+//     { type: 'TTYWrap', asyncId: 5, triggerAsyncId: 1 },
+//     { type: 'Timeout', asyncId: 7, triggerAsyncId: 1 }
+//   ]
+```
+
 ## `process.getegid()`
 
 <!-- YAML
@@ -3792,6 +3857,7 @@ cases:
 [`'message'`]: child_process.md#event-message
 [`'uncaughtException'`]: #event-uncaughtexception
 [`--unhandled-rejections`]: cli.md#--unhandled-rejectionsmode
+[`AsyncHook`]: async_hooks.md#class-asynchook
 [`Buffer`]: buffer.md
 [`ChildProcess.disconnect()`]: child_process.md#subprocessdisconnect
 [`ChildProcess.send()`]: child_process.md#subprocesssendmessage-sendhandle-options-callback
@@ -3802,9 +3868,12 @@ cases:
 [`Promise.race()`]: https://developer.mozilla.org/en-US/docs/Web/JavaScript/Reference/Global_Objects/Promise/race
 [`Worker`]: worker_threads.md#class-worker
 [`Worker` constructor]: worker_threads.md#new-workerfilename-options
+[`async_hooks`]: async_hooks.md
 [`console.error()`]: console.md#consoleerrordata-args
 [`console.log()`]: console.md#consolelogdata-args
+[`destroy` callbacks]: async_hooks.md#destroyasyncid
 [`domain`]: domain.md
+[`init` callbacks]: async_hooks.md#initasyncid-type-triggerasyncid-resource
 [`net.Server`]: net.md#class-netserver
 [`net.Socket`]: net.md#class-netsocket
 [`os.constants.dlopen`]: os.md#dlopen-constants
diff --git a/lib/internal/bootstrap/node.js b/lib/internal/bootstrap/node.js
index b43fb2a5eb4b3a..6bde8804dcaac8 100644
--- a/lib/internal/bootstrap/node.js
+++ b/lib/internal/bootstrap/node.js
@@ -39,6 +39,9 @@
 setupPrepareStackTrace();
 
 const {
+  ArrayPrototypeConcat,
+  ArrayPrototypeForEach,
+  ArrayPrototypePush,
   FunctionPrototypeCall,
   JSONParse,
   ObjectDefineProperty,
@@ -46,12 +49,14 @@ const {
   ObjectGetPrototypeOf,
   ObjectPreventExtensions,
   ObjectSetPrototypeOf,
+  ObjectValues,
   ReflectGet,
   ReflectSet,
   SymbolToStringTag,
   globalThis,
 } = primordials;
 const config = internalBinding('config');
+const internalTimers = require('internal/timers');
 const { deprecate, lazyDOMExceptionClass } = require('internal/util');
 
 setupProcessObject();
@@ -150,6 +155,41 @@ const rawMethods = internalBinding('process_methods');
   process._getActiveRequests = rawMethods._getActiveRequests;
   process._getActiveHandles = rawMethods._getActiveHandles;
 
+  process.getActiveResourcesInfo = function() {
+    const resources = ArrayPrototypeConcat(rawMethods._getActiveRequestsInfo(),
+                                           rawMethods._getActiveHandlesInfo());
+
+    ArrayPrototypeForEach(ObjectValues(internalTimers.timerListMap), (list) => {
+      let timeout = list._idlePrev === list ? null : list._idlePrev;
+      while (timeout !== null) {
+        if (timeout.hasRef()) {
+          ArrayPrototypePush(resources, {
+            type: 'Timeout',
+            asyncId: timeout[internalTimers.async_id_symbol],
+            triggerAsyncId: timeout[internalTimers.trigger_async_id_symbol],
+          });
+        }
+        timeout = timeout._idlePrev === list ? null : list._idlePrev;
+      }
+    });
+
+    const queue = internalTimers.outstandingQueue.head !== null ?
+      internalTimers.outstandingQueue : internalTimers.immediateQueue;
+    let immediate = queue.head;
+    while (immediate !== null) {
+      if (immediate.hasRef()) {
+        ArrayPrototypePush(resources, {
+          type: 'Immediate',
+          asyncId: immediate[internalTimers.async_id_symbol],
+          triggerAsyncId: immediate[internalTimers.trigger_async_id_symbol],
+        });
+      }
+      immediate = immediate._idleNext;
+    }
+
+    return resources;
+  };
+
   // TODO(joyeecheung): remove these
   process.reallyExit = rawMethods.reallyExit;
   process._kill = rawMethods._kill;
@@ -360,9 +400,11 @@ process.emitWarning = emitWarning;
   // TODO(joyeecheung): either remove it or make it public
   process._tickCallback = runNextTicks;
 
-  const { getTimerCallbacks } = require('internal/timers');
   const { setupTimers } = internalBinding('timers');
-  const { processImmediate, processTimers } = getTimerCallbacks(runNextTicks);
+  const {
+    processImmediate,
+    processTimers,
+  } = internalTimers.getTimerCallbacks(runNextTicks);
   // Sets two per-Environment callbacks that will be run from libuv:
   // - processImmediate will be run in the callback of the per-Environment
   //   check handle.
diff --git a/lib/internal/timers.js b/lib/internal/timers.js
index 9a07fe3348f65e..9aa572989c8cd1 100644
--- a/lib/internal/timers.js
+++ b/lib/internal/timers.js
@@ -139,6 +139,11 @@ const kRefed = Symbol('refed');
 // Create a single linked list instance only once at startup
 const immediateQueue = new ImmediateList();
 
+// If an uncaught exception was thrown during execution of immediateQueue,
+// this queue will store all remaining Immediates that need to run upon
+// resolution of all error handling (if process is still alive).
+const outstandingQueue = new ImmediateList();
+
 let nextExpiry = Infinity;
 let refCount = 0;
 
@@ -413,11 +418,6 @@ function setPosition(node, pos) {
 }
 
 function getTimerCallbacks(runNextTicks) {
-  // If an uncaught exception was thrown during execution of immediateQueue,
-  // this queue will store all remaining Immediates that need to run upon
-  // resolution of all error handling (if process is still alive).
-  const outstandingQueue = new ImmediateList();
-
   function processImmediate() {
     const queue = outstandingQueue.head !== null ?
       outstandingQueue : immediateQueue;
@@ -649,6 +649,7 @@ module.exports = {
   setUnrefTimeout,
   getTimerDuration,
   immediateQueue,
+  outstandingQueue,
   getTimerCallbacks,
   immediateInfoFields: {
     kCount,
diff --git a/src/node_process_methods.cc b/src/node_process_methods.cc
index e4b16805586cb9..b11ba4016a6ab3 100644
--- a/src/node_process_methods.cc
+++ b/src/node_process_methods.cc
@@ -1,3 +1,4 @@
+#include "async_wrap-inl.h"
 #include "base_object-inl.h"
 #include "debug_utils-inl.h"
 #include "env-inl.h"
@@ -44,6 +45,7 @@ using v8::HeapStatistics;
 using v8::Integer;
 using v8::Isolate;
 using v8::Local;
+using v8::MaybeLocal;
 using v8::NewStringType;
 using v8::Number;
 using v8::Object;
@@ -242,6 +244,26 @@ static void Uptime(const FunctionCallbackInfo<Value>& args) {
   args.GetReturnValue().Set(result);
 }
 
+static MaybeLocal<Object> GetResourceInfo(Environment* env, AsyncWrap* w) {
+  Local<Object> resource_info = Object::New(env->isolate());
+
+  if (resource_info->Set(env->context(),
+                         OneByteString(env->isolate(), "type"),
+                         OneByteString(env->isolate(),
+                                       w->MemoryInfoName().c_str()))
+                             .IsNothing() ||
+      resource_info->Set(env->context(),
+                         OneByteString(env->isolate(), "asyncId"),
+                         Number::New(env->isolate(), w->get_async_id()))
+                             .IsNothing() ||
+      resource_info->Set(env->context(),
+                         OneByteString(env->isolate(), "triggerAsyncId"),
+                         Number::New(env->isolate(), w->get_trigger_async_id()))
+                             .IsNothing()) return MaybeLocal<Object>();
+
+  return resource_info;
+}
+
 static void GetActiveRequests(const FunctionCallbackInfo<Value>& args) {
   Environment* env = Environment::GetCurrent(args);
 
@@ -257,6 +279,22 @@ static void GetActiveRequests(const FunctionCallbackInfo<Value>& args) {
       Array::New(env->isolate(), request_v.data(), request_v.size()));
 }
 
+static void GetActiveRequestsInfo(const FunctionCallbackInfo<Value>& args) {
+  Environment* env = Environment::GetCurrent(args);
+
+  std::vector<Local<Value>> requests;
+  for (ReqWrapBase* req_wrap : *env->req_wrap_queue()) {
+    AsyncWrap* w = req_wrap->GetAsyncWrap();
+    if (w->persistent().IsEmpty()) continue;
+    Local<Object> request;
+    if (!GetResourceInfo(env, w).ToLocal(&request)) return;
+    requests.emplace_back(request);
+  }
+
+  args.GetReturnValue().Set(
+      Array::New(env->isolate(), requests.data(), requests.size()));
+}
+
 // Non-static, friend of HandleWrap. Could have been a HandleWrap method but
 // implemented here for consistency with GetActiveRequests().
 void GetActiveHandles(const FunctionCallbackInfo<Value>& args) {
@@ -272,6 +310,21 @@ void GetActiveHandles(const FunctionCallbackInfo<Value>& args) {
       Array::New(env->isolate(), handle_v.data(), handle_v.size()));
 }
 
+void GetActiveHandlesInfo(const FunctionCallbackInfo<Value>& args) {
+  Environment* env = Environment::GetCurrent(args);
+
+  std::vector<Local<Value>> handles;
+  for (HandleWrap* w : *env->handle_wrap_queue()) {
+    if (w->persistent().IsEmpty() || !HandleWrap::HasRef(w)) continue;
+    Local<Object> handle;
+    if (!GetResourceInfo(env, w).ToLocal(&handle)) return;
+    handles.emplace_back(handle);
+  }
+
+  args.GetReturnValue().Set(
+      Array::New(env->isolate(), handles.data(), handles.size()));
+}
+
 static void ResourceUsage(const FunctionCallbackInfo<Value>& args) {
   Environment* env = Environment::GetCurrent(args);
 
@@ -547,7 +600,9 @@ static void Initialize(Local<Object> target,
   env->SetMethod(target, "resourceUsage", ResourceUsage);
 
   env->SetMethod(target, "_getActiveRequests", GetActiveRequests);
+  env->SetMethod(target, "_getActiveRequestsInfo", GetActiveRequestsInfo);
   env->SetMethod(target, "_getActiveHandles", GetActiveHandles);
+  env->SetMethod(target, "_getActiveHandlesInfo", GetActiveHandlesInfo);
   env->SetMethod(target, "_kill", Kill);
 
   env->SetMethodNoSideEffect(target, "cwd", Cwd);
@@ -574,7 +629,9 @@ void RegisterExternalReferences(ExternalReferenceRegistry* registry) {
   registry->Register(ResourceUsage);
 
   registry->Register(GetActiveRequests);
+  registry->Register(GetActiveRequestsInfo);
   registry->Register(GetActiveHandles);
+  registry->Register(GetActiveHandlesInfo);
   registry->Register(Kill);
 
   registry->Register(Cwd);
diff --git a/test/parallel/test-handle-wrap-isrefed.js b/test/parallel/test-handle-wrap-isrefed.js
index 2fb766ce72ea4b..c7f479e1c38ae8 100644
--- a/test/parallel/test-handle-wrap-isrefed.js
+++ b/test/parallel/test-handle-wrap-isrefed.js
@@ -106,5 +106,31 @@ const { kStateSymbol } = require('internal/dgram');
                 false, 'tcp_wrap: not unrefed on close')));
 }
 
+// timers
+{
+  strictEqual(process.getActiveResourcesInfo().filter(
+    ({ type }) => type === 'Timeout').length, 0);
+  const timeout = setTimeout(() => {}, 500);
+  strictEqual(process.getActiveResourcesInfo().filter(
+    ({ type }) => type === 'Timeout').length, 1);
+  timeout.unref();
+  strictEqual(process.getActiveResourcesInfo().filter(
+    ({ type }) => type === 'Timeout').length, 0);
+  timeout.ref();
+  strictEqual(process.getActiveResourcesInfo().filter(
+    ({ type }) => type === 'Timeout').length, 1);
+
+  strictEqual(process.getActiveResourcesInfo().filter(
+    ({ type }) => type === 'Immediate').length, 0);
+  const immediate = setImmediate(() => {});
+  strictEqual(process.getActiveResourcesInfo().filter(
+    ({ type }) => type === 'Immediate').length, 1);
+  immediate.unref();
+  strictEqual(process.getActiveResourcesInfo().filter(
+    ({ type }) => type === 'Immediate').length, 0);
+  immediate.ref();
+  strictEqual(process.getActiveResourcesInfo().filter(
+    ({ type }) => type === 'Immediate').length, 1);
+}
 
 // See also test/pseudo-tty/test-handle-wrap-isrefed-tty.js
diff --git a/test/parallel/test-process-getactiveresources-track-active-handles.js b/test/parallel/test-process-getactiveresources-track-active-handles.js
new file mode 100644
index 00000000000000..a1b6a5fbbd876d
--- /dev/null
+++ b/test/parallel/test-process-getactiveresources-track-active-handles.js
@@ -0,0 +1,44 @@
+'use strict';
+
+require('../common');
+const assert = require('assert');
+const net = require('net');
+const NUM = 8;
+const connections = [];
+const clients = [];
+let clients_counter = 0;
+
+const server = net.createServer(function listener(c) {
+  connections.push(c);
+}).listen(0, makeConnection);
+
+
+function makeConnection() {
+  if (clients_counter >= NUM) return;
+  net.connect(server.address().port, function connected() {
+    clientConnected(this);
+    makeConnection();
+  });
+}
+
+
+function clientConnected(client) {
+  clients.push(client);
+  if (++clients_counter >= NUM)
+    checkAll();
+}
+
+
+function checkAll() {
+  assert.strictEqual(process.getActiveResourcesInfo().filter(
+    ({ type }) => type === 'TCPSocketWrap').length,
+                     clients.length + connections.length);
+
+  clients.forEach((item) => item.destroy());
+  connections.forEach((item) => item.end());
+
+  assert.strictEqual(process.getActiveResourcesInfo().filter(
+    ({ type }) => type === 'TCPServerWrap').length, 1);
+
+  server.close();
+}
diff --git a/test/parallel/test-process-getactiveresources-track-active-requests.js b/test/parallel/test-process-getactiveresources-track-active-requests.js
new file mode 100644
index 00000000000000..59f9e2023c86a5
--- /dev/null
+++ b/test/parallel/test-process-getactiveresources-track-active-requests.js
@@ -0,0 +1,10 @@
+'use strict';
+
+const common = require('../common');
+const assert = require('assert');
+const fs = require('fs');
+
+for (let i = 0; i < 12; i++)
+  fs.open(__filename, 'r', common.mustCall());
+
+assert.strictEqual(process.getActiveResourcesInfo().length, 12);
diff --git a/test/parallel/test-process-getactiveresources.js b/test/parallel/test-process-getactiveresources.js
new file mode 100644
index 00000000000000..f7cb265c298d11
--- /dev/null
+++ b/test/parallel/test-process-getactiveresources.js
@@ -0,0 +1,20 @@
+'use strict';
+
+require('../common');
+
+const assert = require('assert');
+
+setTimeout(() => {}, 0);
+
+const resourcesInfo = process.getActiveResourcesInfo();
+
+assert.strictEqual(Array.isArray(resourcesInfo), true);
+assert.strictEqual(resourcesInfo.length, 1);
+
+const resourceInfo = resourcesInfo[0];
+
+assert.deepStrictEqual(Object.keys(resourceInfo),
+                       [ 'type', 'asyncId', 'triggerAsyncId' ]);
+assert.strictEqual(typeof resourceInfo.type, 'string');
+assert.strictEqual(typeof resourceInfo.asyncId, 'number');
+assert.strictEqual(typeof resourceInfo.triggerAsyncId, 'number');
diff --git a/test/sequential/test-net-connect-econnrefused.js b/test/sequential/test-net-connect-econnrefused.js
index 4fd4f8b6943e3c..5b4262902a1711 100644
--- a/test/sequential/test-net-connect-econnrefused.js
+++ b/test/sequential/test-net-connect-econnrefused.js
@@ -54,9 +54,8 @@ function pummel() {
 
 function check() {
   setTimeout(common.mustCall(function() {
-    assert.strictEqual(process._getActiveRequests().length, 0);
-    const activeHandles = process._getActiveHandles();
-    assert.ok(activeHandles.every((val) => val.constructor.name !== 'Socket'));
+    assert.strictEqual(process.getActiveResourcesInfo().filter(
+      ({ type }) => type === 'TCPWRAP').length, 0);
   }), 0);
 }