From b7ed3d22f184129c94a3a631e9fb55a4febf895e Mon Sep 17 00:00:00 2001
From: theolivenbaum <rafael@curiosity.ai>
Date: Mon, 27 Jun 2022 14:53:21 +0200
Subject: [PATCH] Update main.js

---
 ElectronNET.Host/main.js | 38 ++++++++++++++++++++++++++++++++++++--
 1 file changed, 36 insertions(+), 2 deletions(-)

diff --git a/ElectronNET.Host/main.js b/ElectronNET.Host/main.js
index 69afcfc..e3e200a 100644
--- a/ElectronNET.Host/main.js
+++ b/ElectronNET.Host/main.js
@@ -242,9 +242,29 @@ function startSocketApiBridge(port) {
     server = require('http').createServer();
     io = require('socket.io')();
 
-    io.attach(server, { pingTimeout: 10000, pingInterval: 5000, maxHttpBufferSize: 1e8 });
+    let socketBufferSize = 1e8;
+    let pingTimeout = 10000;
+    let pingInterval = 5000;
+
+    if (manifestJsonFile.hasOwnProperty('socket')) {
+
+        if (manifestJsonFile.socket.hasOwnProperty('socketBufferSize') && manifestJsonFile.socket.bufferSize > 0) {
+            socketBufferSize = manifestJsonFile.bufferSize * 1e8;
+        }
+
+        if (manifestJsonFile.socket.hasOwnProperty('socketPingTimeout') && manifestJsonFile.socket.pingTimeout > 0) {
+            maxHttpBufferSize = manifestJsonFile.pingTimeout * 1000;
+        }
+
+        if (manifestJsonFile.socket.hasOwnProperty('socketPingInterval') && manifestJsonFile.socket.pingInterval > 0) {
+            pingInterval = manifestJsonFile.pingInterval * 1000;
+        }
+    }
+
+    io.attach(server, { pingTimeout: pingTimeout, pingInterval: pingInterval, maxHttpBufferSize: maxHttpBufferSize });
 
     server.listen(port, 'localhost');
+
     server.on('listening', function () {
         try { console.log('Electron Socket started on port %s at %s', server.address().port, server.address().address); } catch { }
         // Now that socket connection is established, we can guarantee port will not be open for portscanner
@@ -259,9 +279,14 @@ function startSocketApiBridge(port) {
     app['mainWindowURL'] = "";
     app['mainWindow'] = null;
 
+    let isConnected = false;
+    let checkReconnectTimeout = 0;
     // @ts-ignore
     io.on('connection', (socket) => {
 
+        isConnected = true;
+        window.clearTimeout(checkReconnectTimeout);
+
         socket.on('disconnect', function (reason) {
             try { console.log('Socket ' + socket.id + ' disconnected from .NET with reason: ' + reason); } catch { }
             try {
@@ -270,10 +295,19 @@ function startSocketApiBridge(port) {
                     delete require.cache[require.resolve(hostHookScriptFilePath)];
                     hostHook = undefined;
                 }
-
             } catch (error) {
                 try { console.error(error.message); } catch { }
             }
+
+            window.clearTimeout(checkReconnectTimeout);
+
+            //Give the server 60 seconds to reconnect, otherwise exits
+            checkReconnectTimeout = window.setTimeout((_) => {
+                if (!isConnected) {
+                    app.exit(57005); //0xDEAD
+                }
+            }, 60000);
+
         });
 
         socket.on("auth", function (authKey) {