Prevent signal and wait from racing

Signed-off-by: Jason A. Donenfeld <Jason@zx2c4.com>
2018-09-24 03:08:18 +02:00
parent 2ccbe0eb42
commit 83def0a52b
1 changed files with 3 additions and 1 deletions
@@ -181,7 +181,7 @@ class WireGuardContext {
    private var packetFlow: NEPacketTunnelFlow
    private var outboundPackets: [NEPacket] = []
    private var isTunnelClosed: Bool = false
-    private let readPacketCondition = NSCondition()
+    private var readPacketCondition = NSCondition()

    init(packetFlow: NEPacketTunnelFlow) {
        self.packetFlow = packetFlow
@@ -193,7 +193,9 @@ class WireGuardContext {
    }

    func packetsRead(packets: [NEPacket]) {
+        readPacketCondition.lock()
        outboundPackets.append(contentsOf: packets)
+        readPacketCondition.unlock()
        readPacketCondition.signal()
    }