From bef3eb3964acb7a9b426239a4d8368af396d8ded Mon Sep 17 00:00:00 2001
From: Riley Testut <riley@rileytestut.com>
Date: Fri, 5 Jun 2020 15:43:05 -0700
Subject: [PATCH] [AltKit] Gracefully fails if no data is received over network
 connection

---
 AltKit/Connections/NetworkConnection.swift | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/AltKit/Connections/NetworkConnection.swift b/AltKit/Connections/NetworkConnection.swift
index 041a648f..232b2a51 100644
--- a/AltKit/Connections/NetworkConnection.swift
+++ b/AltKit/Connections/NetworkConnection.swift
@@ -28,6 +28,10 @@ public class NetworkConnection: NSObject, Connection
     public func __receiveData(expectedSize: Int, completionHandler: @escaping (Data?, Error?) -> Void)
     {
         self.nwConnection.receive(minimumIncompleteLength: expectedSize, maximumLength: expectedSize) { (data, context, isComplete, error) in
+            guard data != nil || error != nil else {
+                return completionHandler(nil, ALTServerError(.lostConnection))
+            }
+            
             completionHandler(data, error)
         }
     }