lib: add trace-sigint APIs

Author: theanarkh

doc/api/util.md

@@ -739,6 +739,16 @@ fs.access('file/that/does/not/exist', (err) => {
 });
 ```
 
+## `util.setTraceSigInt(enable)`
+
+<!-- YAML
+added: REPLACEME
+-->
+
+* `enable` {boolean}
+
+Enable or disable printing a stack trace on `SIGINT`. The API is only available on the main thread.
+
 ## `util.inherits(constructor, superConstructor)`
 
 <!-- YAML

lib/internal/process/pre_execution.js

@@ -386,10 +386,7 @@ function setupStacktracePrinterOnSigint() {
   if (!getOptionValue('--trace-sigint')) {
     return;
   }
-  const { SigintWatchdog } = require('internal/watchdog');
-
-  const watchdog = new SigintWatchdog();
-  watchdog.start();
+  require('internal/util/trace_sigint').setTraceSigInt(true);
 }
 
 function initializeReport() {

lib/internal/util/trace_sigint.js

@@ -0,0 +1,29 @@
+'use strict';
+
+const { isMainThread } = require('worker_threads');
+const {
+  ERR_WORKER_UNSUPPORTED_OPERATION,
+} = require('internal/errors').codes;
+
+let sigintWatchdog;
+function getSigintWatchdog() {
+  if (!sigintWatchdog) {
+    const { SigintWatchdog } = require('internal/watchdog');
+    sigintWatchdog = new SigintWatchdog();
+  }
+  return sigintWatchdog;
+}
+
+function setTraceSigInt(enable) {
+  if (!isMainThread)
+    throw new ERR_WORKER_UNSUPPORTED_OPERATION('Calling util.setTraceSigInt');
+  if (enable) {
+    getSigintWatchdog().start();
+  } else {
+    getSigintWatchdog().stop();
+  }
+};
+
+module.exports = {
+  setTraceSigInt,
+};

lib/util.js

@@ -491,3 +491,9 @@ defineLazyProperties(
   'internal/util/diff',
   ['diff'],
 );
+
+defineLazyProperties(
+  module.exports,
+  'internal/util/trace_sigint',
+  ['setTraceSigInt'],
+);

test/parallel/test-start-trace-sigint.js

@@ -0,0 +1,33 @@
+'use strict';
+
+const { mustCall } = require('../common');
+const childProcess = require('child_process');
+const util = require('util');
+const assert = require('assert');
+
+if (!process.env.CHILD) {
+  const cp = childProcess.spawn(
+    process.execPath,
+    [__filename],
+    {
+      env: { ...process.env, CHILD: '1' },
+    });
+  let data;
+  cp.stderr.on('data', (chunk) => {
+    data = data ? data + chunk : chunk;
+  });
+  cp.stderr.on('end', () => {
+    console.log(data);
+    assert.strictEqual(/SIGINT/.test(data.toString()), true);
+  });
+  cp.on('exit', mustCall((code, signal) => {
+    assert.strictEqual(signal, 'SIGINT');
+    assert.strictEqual(code, null);
+  }));
+} else {
+  util.setTraceSigInt(true);
+  // Deactivate colors even if the tty does support colors.
+  process.env.NODE_DISABLE_COLORS = '1';
+  process.kill(process.pid, 'SIGINT');
+  while (true);
+}

test/parallel/test-stop-trace-sigint.js

@@ -0,0 +1,34 @@
+'use strict';
+
+const { mustCall } = require('../common');
+const childProcess = require('child_process');
+const util = require('util');
+const assert = require('assert');
+
+if (!process.env.CHILD) {
+  {
+    const cp = childProcess.spawn(
+      process.execPath,
+      ['--trace-sigint', __filename],
+      {
+        env: { ...process.env, CHILD: '1' },
+      });
+    let data;
+    cp.stderr.on('data', (chunk) => {
+      data = data ? data + chunk : chunk;
+    });
+    cp.stderr.on('end', () => {
+      assert.strictEqual(data === undefined, true);
+    });
+    cp.on('exit', mustCall((code, signal) => {
+      assert.strictEqual(signal, 'SIGINT');
+      assert.strictEqual(code, null);
+    }));
+  }
+} else {
+  util.setTraceSigInt(false);
+  // Deactivate colors even if the tty does support colors.
+  process.env.NODE_DISABLE_COLORS = '1';
+  process.kill(process.pid, 'SIGINT');
+  while (true);
+}

test/parallel/test-trace-sigint-in-worker.js

@@ -0,0 +1,21 @@
+'use strict';
+
+const common = require('../common');
+
+const assert = require('assert');
+const util = require('util');
+const { Worker, workerData } = require('worker_threads');
+
+if (workerData?.isWorker) {
+  assert.throws(() => {
+    util.setTraceSigInt(true);
+  }, {
+    code: 'ERR_WORKER_UNSUPPORTED_OPERATION',
+  });
+  process.exit(0);
+} else {
+  const w = new Worker(__filename, { workerData: { isWorker: true } });
+  w.on('exit', common.mustCall((code) => {
+    assert.strictEqual(code, 0);
+  }));
+}