From 42266daa685442ea72e04e6f14e759281889436c Mon Sep 17 00:00:00 2001
From: legendecas <legendecas@gmail.com>
Date: Sun, 17 Nov 2019 20:02:01 +0800
Subject: [PATCH 1/3] cli: add --trace-exit cli option

It could be convenient to trace abnormal exit of the Node.js processes
that printing stacktrace on each `process.exit` call with a cli option.
This also takes effects on worker threads.
---
 doc/api/cli.md                   |  9 +++++
 src/env.cc                       | 15 ++++++++
 src/node_options.cc              |  4 +++
 src/node_options.h               |  1 +
 test/parallel/test-trace-exit.js | 59 ++++++++++++++++++++++++++++++++
 5 files changed, 88 insertions(+)
 create mode 100644 test/parallel/test-trace-exit.js

diff --git a/doc/api/cli.md b/doc/api/cli.md
index d4b4da5249a16b..c4b17570cfe8f3 100644
--- a/doc/api/cli.md
+++ b/doc/api/cli.md
@@ -774,6 +774,14 @@ added: v7.7.0
 
 Enables the collection of trace event tracing information.
 
+### `--trace-exit`
+<!-- YAML
+added: REPLACEME
+-->
+
+Prints a stack trace whenever an environment is been exited proactively, i.e.
+invoking `process.exit()`.
+
 ### `--trace-sync-io`
 <!-- YAML
 added: v2.1.0
@@ -1101,6 +1109,7 @@ Node.js options that are allowed are:
 * `--trace-event-categories`
 * `--trace-event-file-pattern`
 * `--trace-events-enabled`
+* `--trace-exit`
 * `--trace-sync-io`
 * `--trace-tls`
 * `--trace-uncaught`
diff --git a/src/env.cc b/src/env.cc
index bf3420f826f02a..f3d71c61c09fce 100644
--- a/src/env.cc
+++ b/src/env.cc
@@ -940,6 +940,21 @@ void AsyncHooks::grow_async_ids_stack() {
 uv_key_t Environment::thread_local_env = {};
 
 void Environment::Exit(int exit_code) {
+  if (options()->trace_exit) {
+    HandleScope handle_scope(isolate());
+
+    if (is_main_thread()) {
+      fprintf(stderr, "(node:%d) ", uv_os_getpid());
+    } else {
+      fprintf(stderr, "(node:%d, thread:%llu) ", uv_os_getpid(), thread_id());
+    }
+
+    fprintf(
+        stderr, "WARNING: Exited the environment with code %d\n", exit_code);
+    PrintStackTrace(
+        isolate(),
+        StackTrace::CurrentStackTrace(isolate(), 10, StackTrace::kDetailed));
+  }
   if (is_main_thread()) {
     stop_sub_worker_contexts();
     DisposePlatform();
diff --git a/src/node_options.cc b/src/node_options.cc
index abf26fb78198e8..d7f43929235a20 100644
--- a/src/node_options.cc
+++ b/src/node_options.cc
@@ -476,6 +476,10 @@ EnvironmentOptionsParser::EnvironmentOptionsParser() {
             "show stack traces on deprecations",
             &EnvironmentOptions::trace_deprecation,
             kAllowedInEnvironment);
+  AddOption("--trace-exit",
+            "show stack trace when an environment exits",
+            &EnvironmentOptions::trace_exit,
+            kAllowedInEnvironment);
   AddOption("--trace-sync-io",
             "show stack trace when use of sync IO is detected after the "
             "first tick",
diff --git a/src/node_options.h b/src/node_options.h
index c4cb5dc04f18cb..1f86ec8d2d4767 100644
--- a/src/node_options.h
+++ b/src/node_options.h
@@ -141,6 +141,7 @@ class EnvironmentOptions : public Options {
   bool test_udp_no_try_send = false;
   bool throw_deprecation = false;
   bool trace_deprecation = false;
+  bool trace_exit = false;
   bool trace_sync_io = false;
   bool trace_tls = false;
   bool trace_uncaught = false;
diff --git a/test/parallel/test-trace-exit.js b/test/parallel/test-trace-exit.js
new file mode 100644
index 00000000000000..237512ac273db1
--- /dev/null
+++ b/test/parallel/test-trace-exit.js
@@ -0,0 +1,59 @@
+'use strict';
+const common = require('../common');
+const assert = require('assert');
+const { promisify } = require('util');
+const execFile = promisify(require('child_process').execFile);
+const { Worker, isMainThread, workerData } = require('worker_threads');
+
+const variant = process.argv[process.argv.length - 1];
+switch (true) {
+  case variant === 'main-thread': {
+    return;
+  }
+  case variant === 'main-thread-exit': {
+    return process.exit(0);
+  }
+  case variant.startsWith('worker-thread'): {
+    const worker = new Worker(__filename, { workerData: variant });
+    worker.on('error', common.mustNotCall());
+    worker.on('exit', common.mustCall((code) => {
+      assert.strictEqual(code, 0);
+    }));
+    return;
+  }
+  case !isMainThread: {
+    if (workerData === 'worker-thread-exit') {
+      process.exit(0);
+    }
+    return;
+  }
+}
+
+(async function() {
+  for (const { execArgv, variant, warnings } of [
+    { execArgv: ['--trace-exit'], variant: 'main-thread-exit', warnings: 1 },
+    { execArgv: [], variant: 'main-thread-exit', warnings: 0 },
+    { execArgv: ['--trace-exit'], variant: 'main-thread', warnings: 0 },
+    { execArgv: [], variant: 'main-thread', warnings: 0 },
+    { execArgv: ['--trace-exit'], variant: 'worker-thread-exit', warnings: 1 },
+    { execArgv: [], variant: 'worker-thread-exit', warnings: 0 },
+    { execArgv: ['--trace-exit'], variant: 'worker-thread', warnings: 0 },
+    { execArgv: [], variant: 'worker-thread', warnings: 0 },
+  ]) {
+    const { stdout, stderr } =
+      await execFile(process.execPath, [...execArgv, __filename, variant]);
+    assert.strictEqual(stdout, '');
+    const actualWarnings =
+      stderr.match(/WARNING: Exited the environment with code 0/g);
+    if (warnings === 0) {
+      assert.strictEqual(actualWarnings, null);
+      return;
+    }
+    assert.strictEqual(actualWarnings.length, warnings);
+
+    if (variant.startsWith('worker')) {
+      const workerIds = stderr.match(/\(node:\d+, thread:\d+)/g);
+      assert.strictEqual(workerIds.length, warnings);
+    }
+  }
+})().then(common.mustCall());

From 1b99ae97ba44f7bd7f49d8ea48e38cb8201c06c9 Mon Sep 17 00:00:00 2001
From: legendecas <legendecas@gmail.com>
Date: Fri, 6 Dec 2019 22:26:07 +0800
Subject: [PATCH 2/3] fixup! cli: add --trace-exit cli option

---
 doc/node.1 | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/doc/node.1 b/doc/node.1
index 5f98ba709121fd..fe51c5828469c3 100644
--- a/doc/node.1
+++ b/doc/node.1
@@ -349,6 +349,10 @@ and
 .It Fl -trace-events-enabled
 Enable the collection of trace event tracing information.
 .
+.It Fl -trace-exit
+Prints a stack trace whenever an environment is been exited proactively, i.e.
+invoking `process.exit()`.
+.
 .It Fl -trace-sync-io
 Print a stack trace whenever synchronous I/O is detected after the first turn of the event loop.
 .

From 3acd073971e4dd8c7c85794b1f6a2deaf55a2172 Mon Sep 17 00:00:00 2001
From: legendecas <legendecas@gmail.com>
Date: Sun, 15 Dec 2019 15:39:06 +0800
Subject: [PATCH 3/3] fixup! cli: add --trace-exit cli option

---
 doc/api/cli.md | 4 ++--
 doc/node.1     | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/doc/api/cli.md b/doc/api/cli.md
index c4b17570cfe8f3..2bfca09bd2da1c 100644
--- a/doc/api/cli.md
+++ b/doc/api/cli.md
@@ -779,8 +779,8 @@ Enables the collection of trace event tracing information.
 added: REPLACEME
 -->
 
-Prints a stack trace whenever an environment is been exited proactively, i.e.
-invoking `process.exit()`.
+Prints a stack trace whenever an environment is exited proactively,
+i.e. invoking `process.exit()`.
 
 ### `--trace-sync-io`
 <!-- YAML
diff --git a/doc/node.1 b/doc/node.1
index fe51c5828469c3..5a7e5db8526d43 100644
--- a/doc/node.1
+++ b/doc/node.1
@@ -350,8 +350,8 @@ and
 Enable the collection of trace event tracing information.
 .
 .It Fl -trace-exit
-Prints a stack trace whenever an environment is been exited proactively, i.e.
-invoking `process.exit()`.
+Prints a stack trace whenever an environment is exited proactively,
+i.e. invoking `process.exit()`.
 .
 .It Fl -trace-sync-io
 Print a stack trace whenever synchronous I/O is detected after the first turn of the event loop.