From 2cc60777d5e463f6cc41035820c8c09fcdf89ac6 Mon Sep 17 00:00:00 2001
From: Ben Wilson <ben@registrationmax.com>
Date: Thu, 27 Jul 2023 16:19:27 -0400
Subject: [PATCH] https://github.com/benjamin-wilson/public-pool/issues/3

---
 src/models/StratumV1Client.ts | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/src/models/StratumV1Client.ts b/src/models/StratumV1Client.ts
index 3e5f3f6..1705352 100644
--- a/src/models/StratumV1Client.ts
+++ b/src/models/StratumV1Client.ts
@@ -71,7 +71,13 @@ export class StratumV1Client extends EasyUnsubscribe {
             data.toString()
                 .split('\n')
                 .filter(m => m.length > 0)
-                .forEach(m => this.handleMessage(m))
+                .forEach(async (m) => {
+                    try {
+                        await this.handleMessage(m);
+                    } catch (e) {
+                        this.promiseSocket.socket.emit('end', true);
+                    }
+                })
         });
     }
 
@@ -425,7 +431,11 @@ export class StratumV1Client extends EasyUnsubscribe {
                 'Difficulty too low').response();
             console.error(err);
             console.log(`Header: ${header.toString('hex')}`);
-            await this.promiseSocket.write(err);
+            try {
+                await this.promiseSocket.write(err);
+            } catch (e) {
+                this.promiseSocket.socket.emit('end', true);
+            }
             return false;
         }