Skip to content

Commit

Permalink
lib: move signal event handling into bootstrap/node.js
Browse files Browse the repository at this point in the history
Moves the `process.on()` and `promise.emit()` calls happened during
bootstrap for signal events into `bootstrap/node.js` so it's easier
to tell the side effects.

Drive-by changes:

- Moves the signal event re-arming to a point later during
  the bootstrap - as early as it were it's unlikely that there
  could be any existing signal events to re-arm for node-report.
- Use a Map instead of an Object for signal wraps since it is used
  as a deletable dictionary anyway.

PR-URL: #25859
Reviewed-By: Anna Henningsen <[email protected]>
  • Loading branch information
joyeecheung committed Feb 7, 2019
1 parent b4e670d commit 041424f
Show file tree
Hide file tree
Showing 2 changed files with 44 additions and 29 deletions.
20 changes: 17 additions & 3 deletions lib/internal/bootstrap/node.js
Original file line number Diff line number Diff line change
Expand Up @@ -67,9 +67,6 @@ process.config = JSON.parse(internalBinding('native_module').config);
const rawMethods = internalBinding('process_methods');
// Set up methods and events on the process object for the main thread
if (isMainThread) {
// This depends on process being an event emitter
mainThreadSetup.setupSignalHandlers(internalBinding);

process.abort = rawMethods.abort;
const wrapped = mainThreadSetup.wrapProcessMethods(rawMethods);
process.umask = wrapped.umask;
Expand Down Expand Up @@ -320,6 +317,23 @@ if (process.env.NODE_V8_COVERAGE) {
};
}

// Worker threads don't receive signals.
if (isMainThread) {
const {
isSignal,
startListeningIfSignal,
stopListeningIfSignal
} = mainThreadSetup.createSignalHandlers();
process.on('newListener', startListeningIfSignal);
process.on('removeListener', stopListeningIfSignal);
// re-arm pre-existing signal event registrations
// with this signal wrap capabilities.
const signalEvents = process.eventNames().filter(isSignal);
for (const ev of signalEvents) {
process.emit('newListener', ev);
}
}

if (getOptionValue('--experimental-report')) {
const {
config,
Expand Down
53 changes: 27 additions & 26 deletions lib/internal/process/main_thread_only.js
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,8 @@ const {
validateString
} = require('internal/validators');

const { signals } = internalBinding('constants').os;

// The execution of this function itself should not cause any side effects.
function wrapProcessMethods(binding) {
function chdir(directory) {
Expand Down Expand Up @@ -104,19 +106,18 @@ function wrapPosixCredentialSetters(credentials) {
};
}

// Worker threads don't receive signals.
function setupSignalHandlers(internalBinding) {
const constants = internalBinding('constants').os.signals;
const signalWraps = Object.create(null);
let Signal;
let Signal;
function isSignal(event) {
return typeof event === 'string' && signals[event] !== undefined;
}

function isSignal(event) {
return typeof event === 'string' && constants[event] !== undefined;
}
// Worker threads don't receive signals.
function createSignalHandlers() {
const signalWraps = new Map();

// Detect presence of a listener for the special signal types
process.on('newListener', function(type) {
if (isSignal(type) && signalWraps[type] === undefined) {
function startListeningIfSignal(type) {
if (isSignal(type) && !signalWraps.has(type)) {
if (Signal === undefined)
Signal = internalBinding('signal_wrap').Signal;
const wrap = new Signal();
Expand All @@ -125,30 +126,30 @@ function setupSignalHandlers(internalBinding) {

wrap.onsignal = process.emit.bind(process, type, type);

const signum = constants[type];
const signum = signals[type];
const err = wrap.start(signum);
if (err) {
wrap.close();
throw errnoException(err, 'uv_signal_start');
}

signalWraps[type] = wrap;
signalWraps.set(type, wrap);
}
});
}

process.on('removeListener', function(type) {
if (signalWraps[type] !== undefined && this.listenerCount(type) === 0) {
signalWraps[type].close();
delete signalWraps[type];
function stopListeningIfSignal(type) {
const wrap = signalWraps.get(type);
if (wrap !== undefined && process.listenerCount(type) === 0) {
wrap.close();
signalWraps.delete(type);
}
});

// re-arm pre-existing signal event registrations
// with this signal wrap capabilities.
process.eventNames().forEach((ev) => {
if (isSignal(ev))
process.emit('newListener', ev);
});
}

return {
isSignal,
startListeningIfSignal,
stopListeningIfSignal
};
}

function setupChildProcessIpcChannel() {
Expand All @@ -166,7 +167,7 @@ function setupChildProcessIpcChannel() {

module.exports = {
wrapProcessMethods,
setupSignalHandlers,
createSignalHandlers,
setupChildProcessIpcChannel,
wrapPosixCredentialSetters
};

0 comments on commit 041424f

Please sign in to comment.