Null-check for WebSocket handlers

Author: veniware

Protest/Http/KeepAlive.cs

@@ -36,6 +36,7 @@ public static async void WebSocketHandler(HttpListenerContext ctx) {
         try {
             WebSocketContext wsc = await ctx.AcceptWebSocketAsync(null);
             ws = wsc.WebSocket;
+            if (ws is null) return;
         }
         catch (WebSocketException ex) {
             ctx.Response.Close();

Protest/Protocols/Dhcp.cs

@@ -16,6 +16,7 @@ public static async void WebSocketHandler(HttpListenerContext ctx) {
         try {
             WebSocketContext wsc = await ctx.AcceptWebSocketAsync(null);
             ws = wsc.WebSocket;
+            if (ws is null) return;
         }
         catch (WebSocketException ex) {
             ctx.Response.Close();

Protest/Protocols/Icmp.cs

@@ -58,6 +58,7 @@ public static async void WebSocketHandler(HttpListenerContext ctx) {
         try {
             WebSocketContext wsc = await ctx.AcceptWebSocketAsync(null);
             ws = wsc.WebSocket;
+            if (ws is null) return;
         }
         catch (WebSocketException ex) {
             ctx.Response.Close();

Protest/Protocols/Ssh.cs

@@ -28,6 +28,7 @@ public static async void WebSocketHandler(HttpListenerContext ctx) {
         try {
             WebSocketContext wsc = await ctx.AcceptWebSocketAsync(null);
             ws = wsc.WebSocket;
+            if (ws is null) return;
         }
         catch (WebSocketException ex) {
             ctx.Response.Close();

Protest/Protocols/Telnet.cs

@@ -26,6 +26,7 @@ public static async void WebSocketHandler(HttpListenerContext ctx) {
         try {
             WebSocketContext wsc = await ctx.AcceptWebSocketAsync(null);
             ws = wsc.WebSocket;
+            if (ws is null) return;
         }
         catch (WebSocketException ex) {
             ctx.Response.Close();

Protest/Proxy/ReverseProxy.cs

@@ -86,6 +86,7 @@ public static async void WebSocketHandler(HttpListenerContext ctx) {
         try {
             WebSocketContext wsc = await ctx.AcceptWebSocketAsync(null);
             ws = wsc.WebSocket;
+            if (ws is null) return;
         }
         catch (WebSocketException ex) {
             ctx.Response.Close();

Protest/Tasks/Issues.cs

@@ -108,6 +108,7 @@ public static async void WebSocketHandler(HttpListenerContext ctx) {
         try {
             WebSocketContext wsc = await ctx.AcceptWebSocketAsync(null);
             ws = wsc.WebSocket;
+            if (ws is null) return;
         }
         catch (WebSocketException ex) {
             ctx.Response.Close();

Protest/Tools/IpDiscovery.cs

@@ -106,6 +106,7 @@ public static async void WebSocketHandler(HttpListenerContext ctx) {
         try {
             WebSocketContext wsc = await ctx.AcceptWebSocketAsync(null);
             ws = wsc.WebSocket;
+            if (ws is null) return;
         }
         catch (WebSocketException ex) {
             ctx.Response.Close();

Protest/Tools/LiveStats.cs

@@ -40,6 +40,7 @@ public static async void UserStats(HttpListenerContext ctx) {
         try {
             WebSocketContext wsc = await ctx.AcceptWebSocketAsync(null);
             ws = wsc.WebSocket;
+            if (ws is null) return;
         }
         catch (WebSocketException ex) {
             ctx.Response.Close();
@@ -100,6 +101,7 @@ public static async void DeviceStats(HttpListenerContext ctx) {
         try {
             WebSocketContext wsc = await ctx.AcceptWebSocketAsync(null);
             ws = wsc.WebSocket;
+            if (ws is null) return;
         }
         catch (WebSocketException ex) {
             ctx.Response.Close();

Protest/Tools/Monitor.cs

@@ -76,6 +76,7 @@ public static async void WebSocketHandler(HttpListenerContext ctx) {
         try {
             WebSocketContext wsc = await ctx.AcceptWebSocketAsync(null);
             ws = wsc.WebSocket;
+            if (ws is null) return;
         }
         catch (WebSocketException ex) {
             ctx.Response.Close();