3.0.58

package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pushrocks/smartproxy",
-  "version": "3.0.54",
+  "version": "3.0.58",
   "private": false,
   "description": "a proxy for handling high workloads of proxying",
   "main": "dist_ts/index.js",

ts/00_commitinfo_data.ts

@@ -3,6 +3,6 @@
  */
 export const commitinfo = {
   name: '@pushrocks/smartproxy',
-  version: '3.0.54',
+  version: '3.0.58',
   description: 'a proxy for handling high workloads of proxying'
 }

ts/smartproxy.classes.networkproxy.ts

@@ -247,30 +247,47 @@ JNj2Dr5H0XoLFFnvuvzcRbhlJ9J67JzR+7g=
     // Enable websockets
     const wsServer = new plugins.ws.WebSocketServer({ server: this.httpsServer  });
     wsServer.on('connection', async (wsIncoming: plugins.wsDefault, reqArg: plugins.http.IncomingMessage) => {
-      console.log(`wss: got connection for wsc for https://${reqArg.headers.host}${reqArg.url}`);
+      console.log(`wss proxy: got connection for wsc for https://${reqArg.headers.host}${reqArg.url}`);
       
       let wsOutgoing: plugins.wsDefault;
+
+      const outGoingDeferred = plugins.smartpromise.defer();
+
       try {
         wsOutgoing = new plugins.wsDefault(`ws://${this.router.routeReq(reqArg).destinationIp}:${this.router.routeReq(reqArg).destinationPort}${reqArg.url}`);
+        console.log('wss proxy: initiated outgoing proxy');
+        wsOutgoing.on('open', async () => {
+          outGoingDeferred.resolve();
+        })
       } catch (err) {
         console.log(err);
         wsIncoming.terminate();
         return;
       }
 
-      wsIncoming.on("message", (message, isBinary) => {
-        console.log("client to upstream", message);
+      wsIncoming.on("message", async (message, isBinary) => {
+        await outGoingDeferred.promise;
+        // console.log("client to upstream", message);
         wsOutgoing.send(message, { binary: isBinary });
       });
     
-      wsOutgoing.on("message", (message, isBinary) => {
-        console.log("upstream to client", message);
+      wsOutgoing.on("message", async (message, isBinary) => {
+        // console.log("upstream to client", message);
         wsIncoming.send(message, { binary: isBinary });
       });
-      wsIncoming.on("error", () => wsOutgoing.terminate());
-      wsOutgoing.on("error", () => wsIncoming.terminate());
-      wsIncoming.on("close", () => wsOutgoing.terminate());
-      wsOutgoing.on("close", () => wsIncoming.terminate());
+      const terminateWsOutgoing = () => {
+        wsOutgoing.terminate();
+        console.log('terminated outgoing ws.');
+      }
+      wsIncoming.on("error", () => terminateWsOutgoing());
+      wsIncoming.on("close", () => terminateWsOutgoing());
+
+      const terminateWsIncoming = () => {
+        wsIncoming.terminate();
+        console.log('terminated incoming ws.');
+      }
+      wsOutgoing.on("error", () => terminateWsIncoming());
+      wsOutgoing.on("close", () => terminateWsIncoming());
 
     });
     this.httpsServer.keepAliveTimeout = 600 * 1000;