From e351bcea733f3199587b03f4fa9cb4b0908965ac Mon Sep 17 00:00:00 2001
From: softsimon <softsimon@protonmail.com>
Date: Fri, 13 Mar 2020 18:00:34 +0700
Subject: [PATCH] Fix for the websocket DOS bug.

---
 frontend/src/app/services/websocket.service.ts | 15 ++-------------
 1 file changed, 2 insertions(+), 13 deletions(-)

diff --git a/frontend/src/app/services/websocket.service.ts b/frontend/src/app/services/websocket.service.ts
index ab724a847..906714520 100644
--- a/frontend/src/app/services/websocket.service.ts
+++ b/frontend/src/app/services/websocket.service.ts
@@ -20,7 +20,6 @@ export class WebsocketService {
   private latestGitCommit = '';
   private onlineCheckTimeout: number;
   private onlineCheckTimeoutTwo: number;
-  private connectionCheckTimeout: number;
   private subscription: Subscription;
 
   constructor(
@@ -30,16 +29,7 @@ export class WebsocketService {
   }
 
   startSubscription(retrying = false) {
-    this.connectionCheckTimeout = window.setTimeout(() => {
-      console.log('WebSocket failed to connect, force closing, trying to reconnect');
-      this.websocketSubject.complete();
-      this.subscription.unsubscribe();
-      this.goOffline(true);
-    }, 10000);
     if (retrying) {
-      if (this.stateService.connectionState$.value === 2) {
-        return;
-      }
       this.stateService.connectionState$.next(1);
     }
     this.websocketSubject.next({'action': 'init'});
@@ -128,7 +118,6 @@ export class WebsocketService {
           this.stateService.connectionState$.next(2);
         }
 
-        clearTimeout(this.connectionCheckTimeout);
         this.startOnlineCheck();
       },
       (err: Error) => {
@@ -157,12 +146,12 @@ export class WebsocketService {
     this.lastWant = data;
   }
 
-  goOffline(instant = false) {
+  goOffline() {
     this.goneOffline = true;
     this.stateService.connectionState$.next(0);
     window.setTimeout(() => {
       this.startSubscription(true);
-    }, instant ? 10 : 10000);
+    }, 10000);
   }
 
   startOnlineCheck() {