chore: handle waking from device sleep

Author: ethanndickson

Coder Desktop/Coder Desktop/XPCInterface.swift

@@ -59,9 +59,13 @@ import VPNLib
         }
     }
 
-    func onPeerUpdate(_ data: Data) {
+    func onPeerUpdate(_ data: Data, isState: Bool) {
         Task { @MainActor in
-            svc.onExtensionPeerUpdate(data)
+            if isState {
+                svc.onExtensionPeerState(data)
+            } else {
+                svc.onExtensionPeerUpdate(data)
+            }
         }
     }
 

Coder Desktop/VPN/Manager.swift

@@ -111,7 +111,7 @@ actor Manager {
             if let conn = globalXPCListenerDelegate.conn {
                 do {
                     let data = try msg.peerUpdate.serializedData()
-                    conn.onPeerUpdate(data)
+                    conn.onPeerUpdate(data, isState: false)
                 } catch {
                     logger.error("failed to send peer update to client: \(error)")
                 }
@@ -218,6 +218,15 @@ actor Manager {
         }
         return resp.peerUpdate
     }
+
+    func forcePushPeerState() async throws {
+        let state = try await getPeerState().serializedData()
+        guard let conn = globalXPCListenerDelegate.conn else {
+            logger.info("app was not running on wake, not pushing state")
+            return
+        }
+        conn.onPeerUpdate(state, isState: true)
+    }
 }
 
 struct ManagerConfig {

Coder Desktop/VPN/PacketTunnelProvider.swift

@@ -122,14 +122,24 @@ class PacketTunnelProvider: NEPacketTunnelProvider, @unchecked Sendable {
     }
 
     override func sleep(completionHandler: @escaping () -> Void) {
-        // Add code here to get ready to sleep.
+        // Nothing we need to do on sleep right now
         logger.debug("sleep called")
         completionHandler()
     }
 
     override func wake() {
-        // Add code here to wake up.
         logger.debug("wake called")
+        guard let manager else {
+            logger.error("wake called with nil Manager")
+            return
+        }
+        Task {
+            do {
+                try await manager.forcePushPeerState()
+            } catch {
+                logger.error("error pushing peer state: \(error, privacy: .public)")
+            }
+        }
     }
 
     // Wrapper around `setTunnelNetworkSettings` that supports merging updates

Coder Desktop/VPNLib/XPC.swift

@@ -9,6 +9,7 @@ import Foundation
 @preconcurrency
 @objc public protocol VPNXPCClientCallbackProtocol {
     // data is a serialized `Vpn_PeerUpdate`
-    func onPeerUpdate(_ data: Data)
+    // If `isState` is true, the update should replace the current state.
+    func onPeerUpdate(_ data: Data, isState: Bool)
     func removeQuarantine(path: String, reply: @escaping (Bool) -> Void)
 }