Fixed a bug where Replicant was crashing due to calling invalidate on…

… a nil timer
OperatorFoundation · Feb 7, 2019 · 3ce2100 · 3ce2100
1 parent 2dae914
commit 3ce2100
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/Sources/Replicant/ReplicantConnection.swift b/Sources/Replicant/ReplicantConnection.swift
@@ -146,8 +146,11 @@ open class ReplicantConnection: Connection
         self.sendBuffer = self.sendBuffer[unencryptedChunkSize...]
 
         // Turn off the timer
-        self.sendTimer!.invalidate()
-        self.sendTimer = nil
+        if self.sendTimer != nil
+        {
+            self.sendTimer!.invalidate()
+            self.sendTimer = nil
+        }
 
         // Keep calling network.send if the leftover data is at least chunk size
         self.network.send(content: maybeEncryptedData, contentContext: contentContext, isComplete: isComplete, completion: NWConnection.SendCompletion.contentProcessed(
@@ -156,8 +159,11 @@ open class ReplicantConnection: Connection
             if let error = maybeError
             {
                 print("Received an error on Send:\(error)")
-                self.sendTimer!.invalidate()
-                self.sendTimer = nil
+                if self.sendTimer != nil
+                {
+                    self.sendTimer!.invalidate()
+                    self.sendTimer = nil
+                }
 
                 switch completion
                 {