passepartoutvpn
/
wireguard-apple
mirror of https://github.com/passepartoutvpn/wireguard-apple.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

PacketTunnelProvider: fix locking logic

This commit is contained in:
Jason A. Donenfeld 2018-09-22 04:10:31 +02:00
parent 77d96b21a3
commit 809cb2ef78
1 changed files with 7 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
WireGuardNetworkExtension/PacketTunnelProvider.swift
View File

@ -192,22 +192,19 @@ class WireGuardContext {
readPacketCondition.signal() readPacketCondition.signal()
} }
func packetsRead(packets: [NEPacket]) {
outboundPackets.append(contentsOf: packets)
readPacketCondition.signal()
}
func readPacket(isTunnelClosed: inout Bool) -> NEPacket? { func readPacket(isTunnelClosed: inout Bool) -> NEPacket? {
if outboundPackets.isEmpty { if outboundPackets.isEmpty {
let readPacketCondition = NSCondition()
readPacketCondition.lock() readPacketCondition.lock()
var packetsObtained: [NEPacket]? packetFlow.readPacketObjects(completionHandler: packetsRead)
packetFlow.readPacketObjects { (packets: [NEPacket]) in
packetsObtained = packets
readPacketCondition.signal()
}
// Wait till the completion handler of packetFlow.readPacketObjects() finishes // Wait till the completion handler of packetFlow.readPacketObjects() finishes
while packetsObtained == nil && !self.isTunnelClosed { while outboundPackets.isEmpty && !self.isTunnelClosed {
readPacketCondition.wait() readPacketCondition.wait()
} }
if let packetsObtained = packetsObtained {
outboundPackets = packetsObtained
}
readPacketCondition.unlock() readPacketCondition.unlock()
} }
isTunnelClosed = self.isTunnelClosed isTunnelClosed = self.isTunnelClosed