server: filter out null read().

mrluanma · Mar 14, 2024 · a026833 · a026833
1 parent bfe0c20
commit a026833
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/server.js b/server.js
@@ -68,11 +68,15 @@ wsserver.on('connection', async (ws) => {
   const readable = conn.pipe(
     createTransform(encryptor.decrypt.bind(encryptor)),
   );
-  await new Promise((resolve, reject) => {
-    readable.once('readable', resolve);
-  });
 
-  const data = await readable.read();
+  let data = await readable.read();
+  while (!data) {
+    await new Promise((resolve, reject) => {
+      readable.once('readable', resolve);
+    });
+
+    data = await readable.read();
+  }
 
   let headerLength = 2;
   if (data.length < headerLength) {