node/test/parallel/test-async-local-storage-exit-does-not-leak.js
Stephen Belanger 06f0d78eb9 async_hooks: fix leak in AsyncLocalStorage exit
If exit is called and then run or enterWith are called within the
exit function, the als instace should not be added to the storageList
additional times. The correct behaviour is to remove the instance
from the storageList before executing the exit handler and then to
restore it after.

PR-URL: https://github.com/nodejs/node/pull/35779
Reviewed-By: Vladimir de Turckheim <vlad2t@hotmail.com>
Reviewed-By: Michael Dawson <midawson@redhat.com>
Reviewed-By: Gerhard Stöbich <deb2001-github@yahoo.de>
Reviewed-By: Andrey Pechkurov <apechkurov@gmail.com>
Reviewed-By: Rich Trott <rtrott@gmail.com>
2020-11-11 11:32:50 +00:00

26 lines
754 B
JavaScript

'use strict';
const common = require('../common');
const assert = require('assert');
const { AsyncLocalStorage } = require('async_hooks');
const als = new AsyncLocalStorage();
// Make sure _propagate function exists.
assert.ok(typeof als._propagate === 'function');
// The als instance should be getting removed from the storageList in
// lib/async_hooks.js when exit(...) is called, therefore when the nested runs
// are called there should be no copy of the als in the storageList to run the
// _propagate method on.
als._propagate = common.mustNotCall('_propagate() should not be called');
const done = common.mustCall();
function run(count) {
if (count === 0) return done();
als.run({}, () => {
als.exit(run, --count);
});
}
run(100);