From d9465ae6142dd43cba55a60fc27650c39d9e6c04 Mon Sep 17 00:00:00 2001
From: Erick Wendel <erick.workspace@gmail.com>
Date: Fri, 17 Dec 2021 12:04:29 -0300
Subject: [PATCH] child_process: queue pending messages

It fixes the problem of the child process not receiving messages.

Fixes: https://github.com/nodejs/node/issues/41134

PR-URL: https://github.com/nodejs/node/pull/41221
Reviewed-By: Adrian Estrada <edsadr@gmail.com>
Reviewed-By: Antoine du Hamel <duhamelantoine1995@gmail.com>
Reviewed-By: Minwoo Jung <nodecorelab@gmail.com>
---
 lib/internal/child_process.js                 | 30 ++++++++++++++++++-
 .../test-esm-child-process-fork-main.mjs      | 20 +++++++++++++
 2 files changed, 49 insertions(+), 1 deletion(-)
 create mode 100644 test/es-module/test-esm-child-process-fork-main.mjs

diff --git a/lib/internal/child_process.js b/lib/internal/child_process.js
index 40592146ada6de..0e4cfc2ead2bf4 100644
--- a/lib/internal/child_process.js
+++ b/lib/internal/child_process.js
@@ -81,6 +81,7 @@ let HTTPParser;
 const MAX_HANDLE_RETRANSMISSIONS = 3;
 const kChannelHandle = Symbol('kChannelHandle');
 const kIsUsedAsStdio = Symbol('kIsUsedAsStdio');
+const kPendingMessages = Symbol('kPendingMessages');
 
 // This object contain function to convert TCP objects to native handle objects
 // and back again.
@@ -526,6 +527,7 @@ class Control extends EventEmitter {
   constructor(channel) {
     super();
     this.#channel = channel;
+    this[kPendingMessages] = [];
   }
 
   // The methods keeping track of the counter are being used to track the
@@ -699,6 +701,24 @@ function setupChannel(target, channel, serializationMode) {
     });
   });
 
+  target.on('newListener', function() {
+
+    process.nextTick(() => {
+      if (!target.channel || !target.listenerCount('message'))
+        return;
+
+      const messages = target.channel[kPendingMessages];
+      const { length } = messages;
+      if (!length) return;
+
+      for (let i = 0; i < length; i++) {
+        ReflectApply(target.emit, target, messages[i]);
+      }
+
+      target.channel[kPendingMessages] = [];
+    });
+  });
+
   target.send = function(message, handle, options, callback) {
     if (typeof handle === 'function') {
       callback = handle;
@@ -912,7 +932,15 @@ function setupChannel(target, channel, serializationMode) {
   };
 
   function emit(event, message, handle) {
-    target.emit(event, message, handle);
+    if ('internalMessage' === event || target.listenerCount('message')) {
+      target.emit(event, message, handle);
+      return;
+    }
+
+    ArrayPrototypePush(
+      target.channel[kPendingMessages],
+      [event, message, handle]
+    );
   }
 
   function handleMessage(message, handle, internal) {
diff --git a/test/es-module/test-esm-child-process-fork-main.mjs b/test/es-module/test-esm-child-process-fork-main.mjs
new file mode 100644
index 00000000000000..4dd47a2da97b09
--- /dev/null
+++ b/test/es-module/test-esm-child-process-fork-main.mjs
@@ -0,0 +1,20 @@
+import '../common/index.mjs';
+import assert from 'assert';
+import { fork } from 'child_process';
+import { once } from 'events';
+import { fileURLToPath } from 'url';
+
+if (process.argv[2] !== 'child') {
+  const filename = fileURLToPath(import.meta.url);
+  const cp = fork(filename, ['child']);
+  const message = 'Hello World';
+  cp.send(message);
+
+  const [received] = await once(cp, 'message');
+  assert.deepStrictEqual(received, message);
+
+  cp.disconnect();
+  await once(cp, 'exit');
+} else {
+  process.on('message', (msg) => process.send(msg));
+}