From 37be2433178818c5abaf3f3d36fdb3ff1fbb4767 Mon Sep 17 00:00:00 2001
From: Jeroen Leenarts <jeroen.leenarts@gmail.com>
Date: Tue, 2 Oct 2018 21:19:27 +0200
Subject: [PATCH] Handle persistentKeepAlive better.

Signed-off-by: Jason A. Donenfeld <Jason@zx2c4.com>
---
 .../TunnelConfigurationTableViewController.swift              | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/WireGuard/ViewControllers/TunnelConfigurationTableViewController.swift b/WireGuard/ViewControllers/TunnelConfigurationTableViewController.swift
index 1c11928..0d2fe2f 100644
--- a/WireGuard/ViewControllers/TunnelConfigurationTableViewController.swift
+++ b/WireGuard/ViewControllers/TunnelConfigurationTableViewController.swift
@@ -237,7 +237,7 @@ class PeerTableViewCell: UITableViewCell {
             preSharedKeyField.text = peer.presharedKey
             allowedIpsField.text = peer.allowedIPs
             endpointField.text = peer.endpoint
-            persistentKeepaliveField.text = String(peer.persistentKeepalive)
+            persistentKeepaliveField.text = peer.persistentKeepalive > 0 ? String(peer.persistentKeepalive) : nil
         }
     }
     weak var delegate: PeerTableViewCellDelegate?
@@ -269,6 +269,8 @@ extension PeerTableViewCell: UITextFieldDelegate {
         } else if sender == persistentKeepaliveField {
             if let string = string, let persistentKeepalive = Int32(string) {
                 peer.persistentKeepalive = persistentKeepalive
+            } else {
+                peer.persistentKeepalive = 0
             }
         }
     }